lib-smtp: test-smtp-server-errors: Add support for client-side connection context...

diff --git a/src/lib-smtp/test-smtp-server-errors.c b/src/lib-smtp/test-smtp-server-errors.c
index d2569e76575bc6515c913916ab7c6b9d91114d34..1407242bca5b4d93bfcebe4d680c7765d192ecd1 100644 (file)
--- a/src/lib-smtp/test-smtp-server-errors.c
+++ b/src/lib-smtp/test-smtp-server-errors.c
@@ -30,6 +30,7 @@ struct server_connection {
 
 struct client_connection {
        struct connection conn;
+       void *context;
 
        pool_t pool;
 };
@@ -61,6 +62,7 @@ static unsigned int client_pids_count = 0;
 static unsigned int client_index;
 static void (*test_client_connected)(struct client_connection *conn);
 static void (*test_client_input)(struct client_connection *conn);
+static void (*test_client_deinit)(struct client_connection *conn);
 
 /*
  * Forward declarations
@@ -1332,6 +1334,8 @@ server_connection_deinit(struct client_connection **_conn)
 
        *_conn = NULL;
 
+       if (test_client_deinit != NULL)
+               test_client_deinit(conn);
        connection_deinit(&conn->conn);
        pool_unref(&conn->pool);
 }