Added SILC Thread Pool API.

[silc.git] / lib / silcutil / tests / test_silcthread.c

diff --git a/lib/silcutil/tests/test_silcthread.c b/lib/silcutil/tests/test_silcthread.c
new file mode 100644 (file)
index 0000000..920cb6f
--- /dev/null
+++ b/lib/silcutil/tests/test_silcthread.c
@@ -0,0 +1,103 @@
+/* SilcThreadPool tests */
+
+#include "silc.h"
+
+SilcSchedule schedule;
+
+static void func(SilcSchedule schedule, void *context)
+{
+  SILC_LOG_DEBUG(("func: %d", (int)context));
+  sleep(1);
+}
+
+static void compl(SilcSchedule schedule, void *context)
+{
+  SILC_LOG_DEBUG(("completion: %d", (int)context));
+  if ((int)context == 0xff)
+    silc_schedule_stop(schedule);
+}
+
+int main(int argc, char **argv)
+{
+  SilcBool success = FALSE;
+  SilcThreadPool tp;
+  int i;
+
+  if (argc > 1 && !strcmp(argv[1], "-d")) {
+    silc_log_debug(TRUE);
+    silc_log_quick(TRUE);
+    silc_log_debug_hexdump(TRUE);
+    silc_log_set_debug_string("*thread*");
+  }
+
+  schedule = silc_schedule_init(0, NULL);
+  if (!schedule)
+    goto err;
+
+  SILC_LOG_DEBUG(("Allocate thread pool"));
+  tp = silc_thread_pool_alloc(NULL, 2, 4, TRUE);
+  if (!tp)
+    goto err;
+  SILC_LOG_DEBUG(("Stop thread pool"));
+  silc_thread_pool_free(tp, TRUE);
+
+
+  SILC_LOG_DEBUG(("Allocate thread pool"));
+  tp = silc_thread_pool_alloc(NULL, 0, 2, TRUE);
+  if (!tp)
+    goto err;
+  for (i = 0; i < 4; i++) {
+    SILC_LOG_DEBUG(("Run thread %d", i + 1));
+    if (!silc_thread_pool_run(tp, TRUE, NULL, func, (void *) i + 1,
+                             compl, (void *)i + 1))
+      goto err;
+  }
+  sleep(3);
+  SILC_LOG_DEBUG(("Stop thread pool"));
+  silc_thread_pool_free(tp, TRUE);
+
+  SILC_LOG_DEBUG(("Allocate thread pool"));
+  tp = silc_thread_pool_alloc(NULL, 0, 2, TRUE);
+  if (!tp)
+    goto err;
+  for (i = 0; i < 2; i++) {
+    SILC_LOG_DEBUG(("Run thread %d", i + 1));
+    if (!silc_thread_pool_run(tp, FALSE, NULL, func, (void *) i + 1,
+                             compl, (void *)i + 1))
+      goto err;
+  }
+  if (silc_thread_pool_run(tp, FALSE, NULL, func, (void *)3,
+                          compl, (void *)3))
+    goto err;
+  sleep(3);
+  SILC_LOG_DEBUG(("Stop thread pool"));
+  silc_thread_pool_free(tp, TRUE);
+
+  SILC_LOG_DEBUG(("Allocate thread pool"));
+  tp = silc_thread_pool_alloc(NULL, 8, 16, FALSE);
+  if (!tp)
+    goto err;
+  for (i = 0; i < 8; i++) {
+    SILC_LOG_DEBUG(("Run thread %d", i + 1));
+    if (!silc_thread_pool_run(tp, FALSE, schedule, func, (void *) i + 1,
+                             compl, (void *)i + 1))
+      goto err;
+  }
+  if (!silc_thread_pool_run(tp, FALSE, schedule, func, (void *)0xff,
+                           compl, (void *)0xff))
+    goto err;
+
+  silc_schedule(schedule);
+
+  SILC_LOG_DEBUG(("Stop thread pool"));
+  silc_thread_pool_free(tp, TRUE);
+
+  silc_schedule_uninit(schedule);
+  success = TRUE;
+
+ err:
+  SILC_LOG_DEBUG(("Testing was %s", success ? "SUCCESS" : "FAILURE"));
+  fprintf(stderr, "Testing was %s\n", success ? "SUCCESS" : "FAILURE");
+
+  return success;
+}