Add support for setting fwmarks per-connection. Useful for shaping individual streams...

[cubemap] / markpool.cpp

diff --git a/markpool.cpp b/markpool.cpp
new file mode 100644 (file)
index 0000000..196952e
--- /dev/null
+++ b/markpool.cpp
@@ -0,0 +1,43 @@
+#include "markpool.h"
+#include "mutexlock.h"
+#include <stdio.h>
+#include <assert.h>
+#include <pthread.h>
+#include <queue>
+
+MarkPool::MarkPool(int start, int end)
+       : start(start), end(end)
+{
+       assert(start > 0 && start < 65536);
+       assert(end > 0 && end < 65536);
+
+       for (int i = start; i < end; ++i) {
+               free_marks.push(i);
+       }
+
+       pthread_mutex_init(&mutex, NULL);
+}
+
+int MarkPool::get_mark()
+{
+       MutexLock lock(&mutex);
+       if (free_marks.empty()) {
+               fprintf(stderr, "WARNING: Out of free marks in mark pool %d-%d, session will not be marked.\n",
+                       start, end);
+               fprintf(stderr, "         To fix, increase the pool size and HUP the server.\n");
+               return 0;
+       }
+       int mark = free_marks.front();
+       free_marks.pop();
+       return mark;
+}
+
+void MarkPool::release_mark(int mark)
+{
+       if (mark == 0) {
+               return;
+       }
+
+       MutexLock lock(&mutex);
+       free_marks.push(mark);
+}