X-Git-Url: https://tinc-vpn.org/git/browse?a=blobdiff_plain;f=src%2Fprotocol.c;h=7de75b931a868690861d355e194e58e46801f5b7;hb=3a149f7521dfff67e6a790c1a830afc649ae083e;hp=941459c71ccd82408e9e491be330796b569f2894;hpb=f6e87ab476a0faf8b124ecaaa27f967d825e6457;p=tinc

diff --git a/src/protocol.c b/src/protocol.c
index 941459c7..7de75b93 100644
--- a/src/protocol.c
+++ b/src/protocol.c
@@ -25,7 +25,6 @@
 #include "logger.h"
 #include "meta.h"
 #include "protocol.h"
-#include "utils.h"
 #include "xalloc.h"
 
 bool tunnelserver = false;
@@ -36,7 +35,7 @@ bool experimental = true;
 
 static bool (*request_handlers[])(connection_t *, const char *) = {
 	id_h, metakey_h, challenge_h, chal_reply_h, ack_h,
-	status_h, error_h, termreq_h,
+	NULL, NULL, termreq_h,
 	ping_h, pong_h,
 	add_subnet_h, del_subnet_h,
 	add_edge_h, del_edge_h,
@@ -57,7 +56,19 @@ static char (*request_name[]) = {
 	"REQ_PUBKEY", "ANS_PUBKEY", "SPTPS_PACKET", "UDP_INFO", "MTU_INFO",
 };
 
-static splay_tree_t *past_request_tree;
+static int past_request_compare(const past_request_t *a, const past_request_t *b) {
+	return strcmp(a->request, b->request);
+}
+
+static void free_past_request(past_request_t *r) {
+	free((char *)r->request);
+	free(r);
+}
+
+static splay_tree_t past_request_tree = {
+	.compare = (splay_compare_t) past_request_compare,
+	.delete = (splay_action_t) free_past_request,
+};
 
 /* Generic request routines - takes care of logging and error
    detection as well */
@@ -76,13 +87,14 @@ bool send_request(connection_t *c, const char *format, ...) {
 	request[sizeof(request) - 1] = 0;
 	va_end(args);
 
-	if(len < 0 || len > sizeof(request) - 1) {
+	if(len < 0 || (size_t)len > sizeof(request) - 1) {
 		logger(DEBUG_ALWAYS, LOG_ERR, "Output buffer overflow while sending request to %s (%s)",
 		       c->name, c->hostname);
 		return false;
 	}
 
-	logger(DEBUG_META, LOG_DEBUG, "Sending %s to %s (%s): %s", request_name[atoi(request)], c->name, c->hostname, request);
+	int id = atoi(request);
+	logger(DEBUG_META, LOG_DEBUG, "Sending %s to %s (%s): %s", request_name[id], c->name, c->hostname, request);
 
 	request[len++] = '\n';
 
@@ -90,7 +102,12 @@ bool send_request(connection_t *c, const char *format, ...) {
 		broadcast_meta(NULL, request, len);
 		return true;
 	} else {
-		return send_meta(c, request, len);
+		if(id) {
+			return send_meta(c, request, len);
+		} else {
+			send_meta_raw(c, request, len);
+			return true;
+		}
 	}
 }
 
@@ -98,7 +115,7 @@ void forward_request(connection_t *from, const char *request) {
 	logger(DEBUG_META, LOG_DEBUG, "Forwarding %s from %s (%s): %s", request_name[atoi(request)], from->name, from->hostname, request);
 
 	// Create a temporary newline-terminated copy of the request
-	int len = strlen(request);
+	size_t len = strlen(request);
 	char tmp[len + 1];
 	memcpy(tmp, request, len);
 	tmp[len] = '\n';
@@ -154,26 +171,15 @@ bool receive_request(connection_t *c, const char *request) {
 	return true;
 }
 
-static int past_request_compare(const past_request_t *a, const past_request_t *b) {
-	return strcmp(a->request, b->request);
-}
-
-static void free_past_request(past_request_t *r) {
-	if(r->request) {
-		free((char *)r->request);
-	}
-
-	free(r);
-}
-
 static timeout_t past_request_timeout;
 
 static void age_past_requests(void *data) {
+	(void)data;
 	int left = 0, deleted = 0;
 
-	for splay_each(past_request_t, p, past_request_tree) {
+	for splay_each(past_request_t, p, &past_request_tree) {
 		if(p->firstseen + pinginterval <= now.tv_sec) {
-			splay_delete_node(past_request_tree, node), deleted++;
+			splay_delete_node(&past_request_tree, node), deleted++;
 		} else {
 			left++;
 		}
@@ -190,18 +196,18 @@ static void age_past_requests(void *data) {
 }
 
 bool seen_request(const char *request) {
-	past_request_t *new, p = {NULL};
+	past_request_t *new, p = {0};
 
 	p.request = request;
 
-	if(splay_search(past_request_tree, &p)) {
+	if(splay_search(&past_request_tree, &p)) {
 		logger(DEBUG_SCARY_THINGS, LOG_DEBUG, "Already seen request");
 		return true;
 	} else {
 		new = xmalloc(sizeof(*new));
 		new->request = xstrdup(request);
 		new->firstseen = now.tv_sec;
-		splay_insert(past_request_tree, new);
+		splay_insert(&past_request_tree, new);
 		timeout_add(&past_request_timeout, age_past_requests, NULL, &(struct timeval) {
 			10, rand() % 100000
 		});
@@ -209,12 +215,8 @@ bool seen_request(const char *request) {
 	}
 }
 
-void init_requests(void) {
-	past_request_tree = splay_alloc_tree((splay_compare_t) past_request_compare, (splay_action_t) free_past_request);
-}
-
 void exit_requests(void) {
-	splay_delete_tree(past_request_tree);
+	splay_empty_tree(&past_request_tree);
 
 	timeout_del(&past_request_timeout);
 }