lib/silcutil/tests/test_silcschedule.c

   1 /* SilcSchedule tests */
   2
   3 #include "silc.h"
   4
   5 typedef void (*Callback)(void *context);
   6
   7 #define NUM_TTASK 200
   8 #ifdef FD_SETSIZE
   9 #define NUM_FTASK FD_SETSIZE
  10 #else
  11 #define NUM_FTASK 250
  12 #endif
  13
  14 SilcSchedule schedule;
  15 int c = 0;
  16
  17 void notify_cb(SilcSchedule schedule, SilcBool added, SilcTask task,
  18                SilcBool fd_task, SilcUInt32 fd, SilcTaskEvent event,
  19                long sec, long usec, void *context)
  20 {
  21   SILC_LOG_DEBUG(("Notify cb, %s %s task, fd %d, sec %d usec %d",
  22                   added ? "added" : "deleted", fd_task ? "fd" :"timeout",
  23                   fd, sec, usec));
  24 }
  25
  26 SILC_TASK_CALLBACK(foo)
  27 {
  28
  29 }
  30
  31 SILC_TASK_CALLBACK(timeout)
  32 {
  33   int i = (int)context;
  34   SILC_LOG_DEBUG(("Timeout task %d", i));
  35
  36   SILC_LOG_DEBUG(("Send 'timeout' signal"));
  37   if (!silc_schedule_event_signal(NULL, "timeout", NULL))
  38     SILC_LOG_DEBUG(("Error sending signal, error %d", silc_errno));
  39 }
  40
  41 SILC_TASK_CALLBACK(cont2)
  42 {
  43 #ifdef SILC_DEBUG
  44   silc_schedule_stats(schedule);
  45 #endif /* SILC_DEBUG */
  46
  47   SILC_LOG_DEBUG(("Adding %d fd tasks", NUM_FTASK - 10));
  48
  49 #if 0
  50   for (i = 0; i < NUM_FTASK - 10; i++)
  51     silc_schedule_task_add_fd(schedule, i + 5, foo, (void *)(i + 5));
  52 #endif
  53 }
  54
  55 SILC_TASK_CALLBACK(cont)
  56 {
  57   int i;
  58
  59 #ifdef SILC_DEBUG
  60   silc_schedule_stats(schedule);
  61 #endif /* SILC_DEBUG */
  62
  63   SILC_LOG_DEBUG(("Adding %d timeout tasks", NUM_TTASK / 3));
  64   for (i = 0; i < NUM_TTASK / 3; i++)
  65     silc_schedule_task_add_timeout(schedule, timeout, (void *)i, 0, 0);
  66
  67   silc_schedule_task_add_timeout(schedule, cont2, (void *)i, 0, 100);
  68 }
  69
  70 SILC_TASK_CALLBACK(start)
  71 {
  72   int i;
  73
  74   SILC_LOG_DEBUG(("Adding %d timeout tasks", NUM_TTASK));
  75
  76 #if 0
  77   for (i = 0; i < NUM_TTASK; i++)
  78     silc_schedule_task_add_timeout(schedule, timeout, (void *)i,
  79         i + (i & 9999), (i * 720391) & 999999);
  80 #endif
  81
  82   for (i = 0; i < NUM_TTASK; i++)
  83     silc_schedule_task_add_timeout(schedule, timeout, (void *)i, 0, 1);
  84
  85   silc_schedule_task_add_timeout(schedule, cont, (void *)i, 0, 100);
  86 }
  87
  88 SILC_TASK_CALLBACK(interrupt)
  89 {
  90   SILC_LOG_DEBUG(("SIGINT signal, send 'interrupted' event signal"));
  91   if (!silc_schedule_event_signal(schedule, "interrupted", NULL,
  92                                   schedule))
  93     SILC_LOG_DEBUG(("Error sending signal, error %d", silc_errno));
  94 }
  95
  96 SILC_TASK_EVENT_CALLBACK(timeout_event_cb)
  97 {
  98   SILC_LOG_DEBUG(("timeout event signalled"));
  99   if (c++ == 100) {
 100     silc_schedule_task_del_event(NULL, "timeout");
 101     return FALSE;
 102   }
 103
 104   return TRUE;
 105 }
 106
 107 SILC_TASK_EVENT_CALLBACK(interrupted_event)
 108 {
 109   SilcSchedule ptr = va_arg(va, void *);
 110   SILC_LOG_DEBUG(("interrupted event signalled, ptr %p", ptr));
 111   silc_schedule_event_disconnect(NULL, "interrupted", NULL,
 112                                  interrupted_event, NULL);
 113   silc_schedule_stop(schedule);
 114   return TRUE;
 115 }
 116
 117 int main(int argc, char **argv)
 118 {
 119   SilcBool success = FALSE;
 120   SilcSchedule child, child2;
 121   SilcTask timeout_event;
 122
 123   if (argc > 1 && !strcmp(argv[1], "-d")) {
 124     silc_log_debug(TRUE);
 125     silc_log_quick(TRUE);
 126     silc_log_debug_hexdump(TRUE);
 127     silc_log_set_debug_string("*sched*,*hash*,*errno*");
 128   }
 129
 130   SILC_LOG_DEBUG(("Allocating scheduler"));
 131   schedule = silc_schedule_init(NUM_FTASK, NULL, NULL, NULL);
 132   if (!schedule)
 133     goto err;
 134   silc_schedule_set_notify(schedule, notify_cb, NULL);
 135
 136   SILC_LOG_DEBUG(("Allocate child scheduler"));
 137   child = silc_schedule_init(0, NULL, NULL, schedule);
 138   if (!child)
 139     goto err;
 140
 141   SILC_LOG_DEBUG(("Allocate another child scheduler"));
 142   child2 = silc_schedule_init(0, NULL, NULL, child);
 143   if (!child2)
 144     goto err;
 145
 146   SILC_LOG_DEBUG(("Add 'interrupted' event to child scheduler"));
 147   if (!silc_schedule_task_add_event(child, "interrupted",
 148                                     SILC_PARAM_PTR))
 149     goto err;
 150
 151   SILC_LOG_DEBUG(("Add 'timeout' event to parent scheduler"));
 152   timeout_event =
 153     silc_schedule_task_add_event(schedule, "timeout");
 154   if (!timeout_event)
 155     goto err;
 156
 157   SILC_LOG_DEBUG(("Connect to 'interrupted' event in parent scheduler"));
 158   if (!silc_schedule_event_connect(schedule, "interrupted", NULL,
 159                                    interrupted_event, NULL))
 160     goto err;
 161
 162   SILC_LOG_DEBUG(("Connect to 'timeout' event in child2 scheduler"));
 163   if (!silc_schedule_event_connect(child2, NULL, timeout_event,
 164                                    timeout_event_cb, NULL))
 165     goto err;
 166
 167   SILC_LOG_DEBUG(("Add parent as global scheduler"));
 168   silc_schedule_set_global(schedule);
 169
 170   silc_schedule_task_add_signal(schedule, SIGINT, interrupt, NULL);
 171
 172   if (!silc_schedule_task_add_timeout(schedule, start, NULL, 1, 0))
 173     goto err;
 174
 175   SILC_LOG_DEBUG(("Running scheduler"));
 176   silc_schedule(schedule);
 177
 178   silc_schedule_uninit(schedule);
 179   silc_schedule_uninit(child);
 180   silc_schedule_uninit(child2);
 181
 182   success = TRUE;
 183
 184  err:
 185   SILC_LOG_DEBUG(("Testing was %s", success ? "SUCCESS" : "FAILURE"));
 186   fprintf(stderr, "Testing was %s\n", success ? "SUCCESS" : "FAILURE");
 187
 188   return success;
 189 }