From: Matthias Wachs <wachs@net.in.tum.de>
Date: Mon, 10 Oct 2011 11:33:55 +0000 (+0000)
Subject: quota changes
X-Git-Tag: initial-import-from-subversion-38251~16677
X-Git-Url: https://git.librecmc.org/?a=commitdiff_plain;h=5f939eed860c82aa347fa920b0f956fab81bd94e;p=oweals%2Fgnunet.git

quota changes
---

diff --git a/src/transport/transport_api.c b/src/transport/transport_api.c
index bf37b4ac6..b4d2ed9ff 100644
--- a/src/transport/transport_api.c
+++ b/src/transport/transport_api.c
@@ -401,6 +401,7 @@ demultiplexer (void *cls, const struct GNUNET_MessageHeader *msg)
   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 @@ demultiplexer (void *cls, const struct GNUNET_MessageHeader *msg)
     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"),