1 files changed, 71 insertions, 0 deletions
diff --git a/lib/hook.c b/lib/hook.c
new file mode 100644
index 0000000..895243a
--- /dev/null
+++ b/lib/hook.c
@@ -0,0 +1,71 @@
+/*
+ * Copyright (c) 2016  David Lamparter, for NetDEF, Inc.
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#ifdef HAVE_CONFIG_H
+#include "config.h"
+#endif
+
+#include <string.h>
+
+#include "memory.h"
+#include "hook.h"
+
+DEFINE_MTYPE_STATIC(LIB, HOOK_ENTRY, "Hook entry");
+
+void _hook_register(struct hook *hook, struct hookent *stackent, void *funcptr,
+		    void *arg, bool has_arg, struct frrmod_runtime *module,
+		    const char *funcname, int priority)
+{
+	struct hookent *he, **pos;
+
+	if (!stackent->ent_used)
+		he = stackent;
+	else {
+		he = XCALLOC(MTYPE_HOOK_ENTRY, sizeof(*he));
+		he->ent_on_heap = true;
+	}
+	he->ent_used = true;
+	he->hookfn = funcptr;
+	he->hookarg = arg;
+	he->has_arg = has_arg;
+	he->module = module;
+	he->fnname = funcname;
+	he->priority = priority;
+
+	for (pos = &hook->entries; *pos; pos = &(*pos)->next)
+		if (hook->reverse ? (*pos)->priority < priority
+				  : (*pos)->priority >= priority)
+			break;
+
+	he->next = *pos;
+	*pos = he;
+}
+
+void _hook_unregister(struct hook *hook, void *funcptr, void *arg, bool has_arg)
+{
+	struct hookent *he, **prev;
+
+	for (prev = &hook->entries; (he = *prev) != NULL; prev = &he->next)
+		if (he->hookfn == funcptr && he->hookarg == arg
+		    && he->has_arg == has_arg) {
+			*prev = he->next;
+			if (he->ent_on_heap)
+				XFREE(MTYPE_HOOK_ENTRY, he);
+			else
+				memset(he, 0, sizeof(*he));
+			break;
+		}
+}