[GNUnet-SVN] r20397 - gnunet/src/arm

[gnunet]

Author: grothoff
Date: 2012-03-09 12:39:50 +0100 (Fri, 09 Mar 2012)
New Revision: 20397

Modified:
   gnunet/src/arm/arm_api.c
Log:
-fixing #2201

Modified: gnunet/src/arm/arm_api.c
===================================================================
--- gnunet/src/arm/arm_api.c    2012-03-09 10:29:45 UTC (rev 20396)
+++ gnunet/src/arm/arm_api.c    2012-03-09 11:39:50 UTC (rev 20397)
@@ -1,6 +1,6 @@
 /*
      This file is part of GNUnet.
-     (C) 2009, 2010 Christian Grothoff (and other contributing authors)
+     (C) 2009, 2010, 2012 Christian Grothoff (and other contributing authors)
 
      GNUnet is free software; you can redistribute it and/or modify
      it under the terms of the GNU General Public License as published
@@ -83,6 +83,11 @@
   void *cont_cls;
 
   /**
+   * Handle for transmission request.
+   */
+  struct GNUNET_CLIENT_TransmitHandle *th;
+
+  /**
    * Result of the operation
    */
   enum GNUNET_ARM_ProcessStatus confirmed;
@@ -177,9 +182,10 @@
 static size_t
 write_shutdown (void *cls, size_t size, void *buf)
 {
+  struct ShutdownContext *shutdown_ctx = cls;
   struct GNUNET_MessageHeader *msg;
-  struct ShutdownContext *shutdown_ctx = cls;
 
+  shutdown_ctx->th = NULL;
   if (size < sizeof (struct GNUNET_MessageHeader))
     {
       LOG (GNUNET_ERROR_TYPE_WARNING,
@@ -229,11 +235,10 @@
   shutdown_ctx->sock = sock;
   shutdown_ctx->timeout = GNUNET_TIME_relative_to_absolute (timeout);
   shutdown_ctx->confirmed = GNUNET_ARM_PROCESS_COMMUNICATION_ERROR;    
-  /* FIXME: store return value? */
-  GNUNET_CLIENT_notify_transmit_ready (sock,
-                                      sizeof (struct GNUNET_MessageHeader),
-                                      timeout, GNUNET_YES, &write_shutdown,
-                                      shutdown_ctx);
+  shutdown_ctx->th = GNUNET_CLIENT_notify_transmit_ready (sock,
+                                                         sizeof (struct 
GNUNET_MessageHeader),
+                                                         timeout, GNUNET_NO, 
&write_shutdown,
+                                                         shutdown_ctx);
 }