[GNUnet-SVN] r17318 - gnunet/src/transport

[gnunet]

Author: wachs
Date: 2011-10-10 13:33:55 +0200 (Mon, 10 Oct 2011)
New Revision: 17318

Modified:
   gnunet/src/transport/transport_api.c
Log:
quota changes


Modified: gnunet/src/transport/transport_api.c
===================================================================
--- gnunet/src/transport/transport_api.c        2011-10-10 11:31:12 UTC (rev 
17317)
+++ gnunet/src/transport/transport_api.c        2011-10-10 11:33:55 UTC (rev 
17318)
@@ -401,6 +401,7 @@
   const struct InboundMessage *im;
   const struct GNUNET_MessageHeader *imm;
   const struct SendOkMessage *okm;
+  const struct QuotaSetMessage *qm;
   struct GNUNET_TRANSPORT_GetHelloHandle *hwl;
   struct GNUNET_TRANSPORT_GetHelloHandle *next_hwl;
   struct Neighbour *n;
@@ -565,6 +566,24 @@
     if (h->rec != NULL)
       h->rec (h->cls, &im->peer, imm, &im->ats, ats_count);
     break;
+  case GNUNET_MESSAGE_TYPE_TRANSPORT_SET_QUOTA:
+#if DEBUG_TRANSPORT_API
+    GNUNET_log (GNUNET_ERROR_TYPE_DEBUG, "Receiving `%s' message.\n", 
"SET_QUOTA");
+#endif
+    if (size != sizeof (struct QuotaSetMessage))
+    {
+      GNUNET_break (0);
+      break;
+    }
+    qm = (const struct QuotaSetMessage *) msg;
+    n = neighbour_find (h, &qm->peer);
+    if (n == NULL)
+    {
+      GNUNET_break (0);
+      break;
+    }
+    GNUNET_BANDWIDTH_tracker_update_quota (&n->out_tracker, qm->quota);
+    break;
   default:
     GNUNET_log (GNUNET_ERROR_TYPE_ERROR,
                 _("Received unexpected message of type %u in %s:%u\n"),