stuff

[oweals/gnunet.git] / src / util / test_server_disconnect.c

diff --git a/src/util/test_server_disconnect.c b/src/util/test_server_disconnect.c
index 76a73d3482897e8650f9c4f52fe972f6cd55c8b8..faaca968d0db0269d02f2320e6ccdab1e366e467 100644 (file)
--- a/src/util/test_server_disconnect.c
+++ b/src/util/test_server_disconnect.c
@@ -119,7 +119,7 @@ disconnect_notify (void *cls, const struct GNUNET_MessageHeader *msg)
   GNUNET_assert (msg == NULL);
   GNUNET_assert (ok == 7);
   ok = 0;
-  GNUNET_CLIENT_disconnect (client);
+  GNUNET_CLIENT_disconnect (client, GNUNET_YES);
   client = NULL;
   GNUNET_SERVER_destroy (server);
   server = NULL;
@@ -138,6 +138,8 @@ disconnect_notify (void *cls, const struct GNUNET_MessageHeader *msg)
 static void
 notify_disconnect (void *cls, struct GNUNET_SERVER_Client *clientarg)
 {
+  if (clientarg == NULL)
+    return;
   GNUNET_assert (ok == 6);
   ok++;
   GNUNET_CLIENT_receive (client,