[GNUnet-SVN] r21157 - gnunet/src/util

[gnunet]

Author: grothoff
Date: 2012-04-25 16:54:46 +0200 (Wed, 25 Apr 2012)
New Revision: 21157

Modified:
   gnunet/src/util/test_service.c
Log:
-handle shutdown better

Modified: gnunet/src/util/test_service.c
===================================================================
--- gnunet/src/util/test_service.c      2012-04-25 14:45:39 UTC (rev 21156)
+++ gnunet/src/util/test_service.c      2012-04-25 14:54:46 UTC (rev 21157)
@@ -49,10 +49,16 @@
 static void
 do_stop (void *cls, const struct GNUNET_SCHEDULER_TaskContext *tc)
 {
-  GNUNET_CLIENT_disconnect (client);
-  client = NULL;
-  GNUNET_SERVICE_stop (sctx);
-  sctx = NULL;
+  if (NULL != client)
+  {
+    GNUNET_CLIENT_disconnect (client);
+    client = NULL;
+  }
+  if (NULL != sctx)
+  {
+    GNUNET_SERVICE_stop (sctx);
+    sctx = NULL;
+  }
 }
 
 
@@ -99,12 +105,11 @@
 recv_cb (void *cls, struct GNUNET_SERVER_Client *sc,
          const struct GNUNET_MessageHeader *message)
 {
+  if (NULL == message)
+    return;
   GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Receiving client message...\n");
   GNUNET_SERVER_receive_done (sc, GNUNET_OK);
-  if (sctx != NULL)
-    GNUNET_SCHEDULER_add_now (&do_stop, NULL);
-  else
-    GNUNET_SCHEDULER_shutdown ();
+  GNUNET_SCHEDULER_add_now (&do_stop, NULL);
   ok = 0;
 }