1 files changed, 131 insertions, 0 deletions

diff --git a/src/lib/common/timeout.c b/src/lib/common/timeout.c
new file mode 100644
index 0000000..b12780b
--- /dev/null
+++ b/src/lib/common/timeout.c
@@ -0,0 +1,131 @@
+#include "timeout.h"
+
+#include <stdbool.h>
+
+#define LOG(kind, ...) GNUNET_log_from (kind, "dbus-timeout", __VA_ARGS__)
+
+struct Timeout
+{
+  /* doubly-linked list */
+  struct Timeout *next;
+  struct Timeout *prev;
+
+  DBusTimeout *timeout;
+
+  bool scheduled;
+
+  struct GNUNET_SCHEDULER_Task *task;
+
+  unsigned ref_count;
+};
+
+struct Timeout *
+timeout_create (
+    DBusTimeout *timeout)
+{
+  struct Timeout *t = GNUNET_new (struct Timeout);
+
+  t->timeout = timeout;
+  t->scheduled = false;
+  t->ref_count = 1;
+
+  return t;
+}
+
+void
+timeout_ref (
+    struct Timeout *t)
+{
+  t->ref_count++;
+}
+
+void
+timeout_unref (
+    struct Timeout *t)
+{
+  if (0 == t->ref_count)
+  {
+    LOG (GNUNET_ERROR_TYPE_ERROR, "Tried to unref timeout with ref_count == 0\n");
+    GNUNET_abort_ ();
+  };
+
+  if (0 == --t->ref_count)
+  {
+    GNUNET_free (t);
+  }
+}
+
+static void
+handle_timeout (
+    void *cls,
+    const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  struct Timeout *t = (struct Timeout *)cls;
+
+  t->scheduled = false;
+
+  if (tc->reason & GNUNET_SCHEDULER_REASON_TIMEOUT)
+    dbus_timeout_handle (t->timeout);
+
+  if (t->ref_count > 1 && ! (tc->reason & GNUNET_SCHEDULER_REASON_SHUTDOWN))
+    timeout_schedule (t);
+  timeout_unref (t);
+};
+
+void
+timeout_schedule (
+    struct Timeout *t)
+{
+  if (! dbus_timeout_get_enabled (t->timeout))
+  {
+    LOG (GNUNET_ERROR_TYPE_WARNING, "Tried to schedule timeout that is disabled!\n");
+    return;
+  };
+
+  if (t->scheduled)
+    return;
+
+  int interval = dbus_timeout_get_interval (t->timeout);
+  struct GNUNET_TIME_Relative delay;
+  delay.rel_value_us = interval * 1000;
+  t->task = GNUNET_SCHEDULER_add_delayed (
+      delay,
+      handle_timeout,
+      t);
+  t->scheduled = true;
+  timeout_ref (t);
+}
+
+void
+do_timeout_unschedule (
+    void *cls,
+    const struct GNUNET_SCHEDULER_TaskContext *tc)
+{
+  (void)tc;
+  struct Timeout *t = (struct Timeout *)cls;
+
+  if (! t->scheduled)
+    return;
+  
+  void *ret = GNUNET_SCHEDULER_cancel (t->task);
+  if ((struct Timeout *)ret != t)
+    LOG (GNUNET_ERROR_TYPE_WARNING, "Weird result unscheduling task. t == %p, GNUNET_SCHEDULER_cancel returned %p\n", t, ret);
+
+  timeout_unref (t);
+}
+
+void
+timeout_unschedule (
+    struct Timeout *t)
+{
+  GNUNET_SCHEDULER_add_now (do_timeout_unschedule, t);
+}
+
+struct TimeoutIter *
+timeout_find (struct TimeoutIter *ti, DBusTimeout *timeout)
+{
+  while (ti && ti->t->timeout != timeout)
+    ti = ti->next;
+  return ti;
+}
+