contrib/fwd: add utility functions

2 files changed, 128 insertions, 0 deletions

diff --git a/contrib/fwd/src/fwd_utils.c b/contrib/fwd/src/fwd_utils.c
new file mode 100644
index 000000000..170ecf468
--- /dev/null
+++ b/contrib/fwd/src/fwd_utils.c
@@ -0,0 +1,78 @@
+/*
+ * fwd - OpenWrt firewall daemon - commmon utility functions
+ *
+ *   Copyright (C) 2009 Jo-Philipp Wich <xm@subsignal.org>
+ *
+ * The fwd program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License version 2
+ * as published by the Free Software Foundation.
+ *
+ * The fwd program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+ * See the GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with the fwd program. If not, see http://www.gnu.org/licenses/.
+ */
+
+
+#include "fwd_utils.h"
+
+
+void fwd_log_init(void)
+{
+	openlog("Firewall", 0, LOG_DAEMON | LOG_PERROR);
+}
+
+void __fwd_log(int prio, const char *msg, ...)
+{
+	va_list ap;
+	va_start(ap, msg);
+	vsyslog(prio, msg, ap);
+	va_end(ap);
+}
+
+
+
+int fwd_empty_cidr(struct fwd_cidr *c)
+{
+	if( (c == NULL) || ((c->addr.s_addr == 0) && (c->prefix == 0)) )
+		return 1;
+
+	return 0;
+}
+
+int fwd_equal_cidr(struct fwd_cidr *a, struct fwd_cidr *b)
+{
+	if( fwd_empty_cidr(a) && fwd_empty_cidr(b) )
+		return 1;
+	else if( (a->addr.s_addr == b->addr.s_addr) && (a->prefix == b->prefix) )
+		return 1;
+
+	return 0;
+}
+
+void fwd_update_cidr(struct fwd_cidr *a, struct fwd_cidr *b)
+{
+	if( a != NULL )
+	{
+		a->addr.s_addr = b ? b->addr.s_addr : 0;
+		a->prefix = b ? b->prefix : 0;
+	}
+}
+
+
+/* fwd_zmalloc(size_t)
+ * Allocates a zeroed buffer of the given size. */
+void * fwd_zmalloc(size_t s)
+{
+	void *b = malloc(s);
+
+	if( b != NULL )
+		memset(b, 0, s);
+
+	return b;
+}
+
+
diff --git a/contrib/fwd/src/fwd_utils.h b/contrib/fwd/src/fwd_utils.h
new file mode 100644
index 000000000..618a53784
--- /dev/null
+++ b/contrib/fwd/src/fwd_utils.h
@@ -0,0 +1,50 @@
+/*
+ * fwd - OpenWrt firewall daemon - commmon utility header
+ *
+ *   Copyright (C) 2009 Jo-Philipp Wich <xm@subsignal.org>
+ *
+ * The fwd program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License version 2
+ * as published by the Free Software Foundation.
+ *
+ * The fwd program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+ * See the GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with the fwd program. If not, see http://www.gnu.org/licenses/.
+ */
+
+#ifndef __FWD_UTILS_H__
+#define __FWD_UTILS_H__
+
+#include <syslog.h>
+
+#include "fwd.h"
+
+void fwd_log_init(void);
+void __fwd_log(int, const char *, ...);
+#define fwd_log_info(...) __fwd_log(LOG_INFO, __VA_ARGS__)
+#define fwd_log_err(...) __fwd_log(LOG_ERR, __VA_ARGS__)
+
+int fwd_empty_cidr(struct fwd_cidr *);
+int fwd_equal_cidr(struct fwd_cidr *, struct fwd_cidr *);
+void fwd_update_cidr(struct fwd_cidr *, struct fwd_cidr *);
+
+/* fwd_zmalloc(size_t)
+ * Allocates a zeroed buffer of the given size. */
+void * fwd_zmalloc(size_t);
+
+/* fwd_alloc_ptr(type)
+ * Allocates a buffer with the size of the given datatype
+ * and returns a pointer to it. */
+#define fwd_alloc_ptr(t) (t *) fwd_zmalloc(sizeof(t))
+
+/* fwd_free_ptr(void *)
+ * Frees the given pointer and sets it to NULL.
+ * Safe for NULL values. */
+#define fwd_free_ptr(x) do { if(x != NULL) free(x); x = NULL; } while(0)
+
+#endif
+