Browse Source

sched: Define SCHEDULER_KIST_PRIVATE for more encapsulation

Signed-off-by: David Goulet <dgoulet@torproject.org>
David Goulet 6 years ago
parent
commit
1033e14a69
4 changed files with 8 additions and 0 deletions
  1. 1 0
      src/or/scheduler.c
  2. 4 0
      src/or/scheduler.h
  3. 2 0
      src/or/scheduler_kist.c
  4. 1 0
      src/test/test_scheduler.c

+ 1 - 0
src/or/scheduler.c

@@ -6,6 +6,7 @@
 
 #include "compat_libevent.h"
 #define SCHEDULER_PRIVATE_
+#define SCHEDULER_KIST_PRIVATE
 #include "scheduler.h"
 
 #include <event2/event.h>

+ 4 - 0
src/or/scheduler.h

@@ -155,6 +155,8 @@ void scheduler_touch_channel(channel_t *chan);
  * Defined in scheduler_kist.c
  *********************************/
 
+#ifdef SCHEDULER_KIST_PRIVATE
+
 /* Socke table entry which holds information of a channel's socket and kernel
  * TCP information. Only used by KIST. */
 typedef struct socket_table_ent_s {
@@ -188,6 +190,8 @@ int32_t kist_scheduler_run_interval(const networkstatus_t *ns);
 extern int32_t sched_run_interval;
 #endif /* TOR_UNIT_TESTS */
 
+#endif /* SCHEDULER_KIST_PRIVATE */
+
 /*********************************
  * Defined in scheduler_vanilla.c
  *********************************/

+ 2 - 0
src/or/scheduler_kist.c

@@ -1,6 +1,8 @@
 /* Copyright (c) 2017, The Tor Project, Inc. */
 /* See LICENSE for licensing information */
 
+#define SCHEDULER_KIST_PRIVATE
+
 #include <event2/event.h>
 #include <netinet/tcp.h>
 

+ 1 - 0
src/test/test_scheduler.c

@@ -6,6 +6,7 @@
 #include <math.h>
 #include <event2/event.h>
 
+#define SCHEDULER_KIST_PRIVATE
 #define TOR_CHANNEL_INTERNAL_
 #define CHANNEL_PRIVATE_
 #include "or.h"