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

diff --git a/src/protocol.c b/src/protocol.c
index 3a0ae1d7..7de75b93 100644
--- a/src/protocol.c
+++ b/src/protocol.c
@@ -56,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 */
@@ -103,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';
@@ -159,24 +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) {
-	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++;
 		}
@@ -197,14 +200,14 @@ bool seen_request(const char *request) {
 
 	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
 		});
@@ -212,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);
 }