1 files changed, 32 insertions, 0 deletions

diff --git a/trackerlogic.c b/trackerlogic.c
index 477c4a6..b0b7f4e 100644
--- a/trackerlogic.c
+++ b/trackerlogic.c
@@ -20,6 +20,7 @@
 #include "ot_ip.h"
 #include "ot_livesync.h"
 #include "ot_mutex.h"
+#include "ot_workqueue.h"
 #include "ot_random.h"
 #include "ot_stats.h"
 #include "ot_udp.h"
@@ -560,6 +561,7 @@ void trackerlogic_init(void) {
 
   /* Initialise background worker threads */
   mutex_init();
+  workqueue_init();
   clean_init();
   fullscrape_init();
   accesslist_init();
@@ -567,6 +569,35 @@ void trackerlogic_init(void) {
   stats_init();
 }
 
+int workstruct_init(struct ot_workstruct *ws) {
+  memset(ws, 0, sizeof(*ws));
+  ws->inbuf  = malloc(G_INBUF_SIZE);
+  ws->outbuf = malloc(G_OUTBUF_SIZE);
+#ifdef _DEBUG_HTTPERROR
+  ws->debugbuf = malloc(G_DEBUGBUF_SIZE);
+#endif
+  if (!ws->inbuf || !ws->outbuf)
+    return -1;
+
+  /* Picking which peers to hand out starts here, so two workers seeded alike
+     would hand out alike */
+  random_bytes(ws->rand48_state, sizeof(ws->rand48_state));
+
+  stats_counters_register(ws);
+  return 0;
+}
+
+void workstruct_deinit(struct ot_workstruct *ws) {
+  /* The tally goes to the global before this struct goes out of scope */
+  stats_counters_retire(ws);
+
+  free(ws->inbuf);
+  free(ws->outbuf);
+#ifdef _DEBUG_HTTPERROR
+  free(ws->debugbuf);
+#endif
+}
+
 void trackerlogic_deinit(void) {
   int    delta_torrentcount = 0;
   size_t bucket, j;
@@ -594,5 +625,6 @@ void trackerlogic_deinit(void) {
   }
 
   /* Release mutexes */
+  workqueue_deinit();
   mutex_deinit();
 }