From ebd9d98fd7a2c08f57b1187a7b9efafbfbb63fe7 Mon Sep 17 00:00:00 2001
From: Matthias Schiffer <mschiffer@universe-factory.net>
Date: Fri, 2 Mar 2012 20:12:59 +0100
Subject: Delete pending tasks for deleted peers

---
 src/task.c | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

(limited to 'src/task.c')

diff --git a/src/task.c b/src/task.c
index 0d2dd6e..b565f1b 100644
--- a/src/task.c
+++ b/src/task.c
@@ -70,3 +70,32 @@ void fastd_task_schedule_handshake(fastd_context *ctx, fastd_peer *peer, int tim
 
 	fastd_queue_put(&ctx->task_queue, task, timeout);
 }
+
+static bool delete_task(void *data, void *extra) {
+	fastd_task *task = data;
+	fastd_peer *peer = extra;
+
+	if (task->any.peer != peer)
+		return true;
+
+	switch (task->any.type) {
+	case TASK_SEND:
+		fastd_buffer_free(task->send.buffer);
+		break;
+
+	case TASK_HANDLE_RECV:
+		fastd_buffer_free(task->handle_recv.buffer);
+		break;
+
+	case TASK_HANDSHAKE:
+		break;
+	}
+
+	free(task);
+
+	return false;
+}
+
+void fastd_task_delete_peer(fastd_context *ctx, fastd_peer *peer) {
+	fastd_queue_filter(&ctx->task_queue, delete_task, peer);
+}
-- 
cgit v1.2.3