Initial commit

6 files changed, 771 insertions, 0 deletions

diff --git a/src/.gitignore b/src/.gitignore
new file mode 100644
index 0000000..7f6e438
--- /dev/null
+++ b/src/.gitignore
@@ -0,0 +1,2 @@
+vcc_if.c
+vcc_if.h
diff --git a/src/Makefile.am b/src/Makefile.am
new file mode 100644
index 0000000..c9daf9a
--- /dev/null
+++ b/src/Makefile.am
@@ -0,0 +1,41 @@
+# This file is part of vmod-binlog
+# Copyright (C) 2013 Sergey Poznyakoff
+#
+# Vmod-binlog is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 3, or (at your option)
+# any later version.
+#
+# Vmod-binlog 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 vmod-binlog.  If not, see <http://www.gnu.org/licenses/>.
+
+AM_CPPFLAGS = -I$(VARNISHSRC)/include -I$(VARNISHSRC)
+
+bin_PROGRAMS = binlogcat
+binlogcat_SOURCES = binlogcat.c
+
+vmoddir = $(VMODDIR)
+vmod_LTLIBRARIES = libvmod_binlog.la
+
+libvmod_binlog_la_LDFLAGS = -module -export-dynamic -avoid-version
+libvmod_binlog_la_LIBADD=
+
+libvmod_binlog_la_SOURCES = \
+        binlog.c\
+        vmod-binlog.h\
+        vcc_if.c vcc_if.h
+
+BUILT_SOURCES = vcc_if.c vcc_if.h
+
+vcc_if.c vcc_if.h: $(VARNISHSRC)/lib/libvmod_std/vmod.py $(top_srcdir)/src/vmod.vcc
+        @PYTHON@ $(VARNISHSRC)/lib/libvmod_std/vmod.py $(top_srcdir)/src/vmod.vcc
+
+EXTRA_DIST = vmod.vcc
+
+CLEANFILES = $(builddir)/vcc_if.c $(builddir)/vcc_if.h
+
diff --git a/src/binlog.c b/src/binlog.c
new file mode 100644
index 0000000..f896803
--- /dev/null
+++ b/src/binlog.c
@@ -0,0 +1,516 @@
+/* This file is part of vmod-binlog
+   Copyright (C) 2013 Sergey Poznyakoff
+
+   Vmod-binlog is free software; you can redistribute it and/or modify
+   it under the terms of the GNU General Public License as published by
+   the Free Software Foundation; either version 3, or (at your option)
+   any later version.
+
+   Vmod-binlog 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 vmod-binlog.  If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <config.h>
+#include <sys/stat.h>
+#include <sys/mman.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <errno.h>
+#include <syslog.h>
+#include <stdlib.h>
+#include <ctype.h>
+#include <time.h>
+#include "vrt.h"
+#include "vcc_if.h"
+#include "bin/varnishd/cache.h"
+#include "vmod-binlog.h"
+
+#ifndef O_SEARCH
+# define O_SEARCH 0
+#endif
+
+#define BLF_ROUNDTS 0x01
+
+struct binlog_config {
+        size_t size;         /* maximum file size */
+        unsigned interval;   /* file rotation interval */
+        char *pattern;       /* file name pattern */
+        int umask;           /* umask for new files and directories */
+        char *dir;           /* root storage directory */      
+        int dd;              /* directory descriptor */
+        char *fname;         /* current file name */
+        int fd;              /* current file descriptor */
+        union binlog_header *base;     /* mmap base */
+        struct binlog_record *recbase; /* record base */
+        size_t recnum;       /* number of records in recbase */
+        size_t recidx;       /* index of the next free entry in recbase */
+        time_t stoptime;     /* when to rotate the current file */
+        pthread_mutex_t mutex;
+        int debug;
+        int flags;
+};
+
+static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
+
+void
+binlog_error(const char *fmt, ...)
+{
+        va_list ap;
+        va_start(ap, fmt);
+        vsyslog(LOG_DAEMON|LOG_ERR, fmt, ap);
+        va_end(ap);
+}
+
+void
+binlog_debug(const char *fmt, ...)
+{
+        va_list ap;
+        va_start(ap, fmt);
+        vsyslog(LOG_DAEMON|LOG_DEBUG, fmt, ap);
+        va_end(ap);
+}
+
+#define debug(c,l,s) do { if ((c)->debug>=(l)) binlog_debug s; } while(0)
+
+int
+module_init(struct vmod_priv *priv, const struct VCL_conf *vclconf)
+{
+        struct binlog_config *conf = calloc(1, sizeof(*conf));
+        AN(conf);
+        priv->priv = conf;
+        return 0;
+}
+
+static char *
+findparam(const char *param, char *name)
+{
+        const char *p;
+        char *q;
+
+        while (*param) {
+                for (p = param, q = name; *p && *q && *p == *q; p++, q++);
+                for (param = p; *param && *param != ';'; param++);
+                if (*q == 0 && *p == '=') {
+                        size_t len = param - p - 1;
+                        q = malloc(len + 1);
+                        AN(q);
+                        memcpy(q, p + 1, len);
+                        q[len] = 0;
+                        return q;
+                }
+                if (*param)
+                        ++param;
+        }
+        return NULL;
+}
+
+static unsigned long
+getinterval(char *p, char **endp)
+{
+        int n;
+        unsigned long hours = 0, minutes = 0, seconds = 0;
+    
+        for (;;) {
+                n = 0;
+                while (*p && isdigit(*p)) {
+                        n = 10*n + *p - '0';
+                        p++;
+                }
+    
+                switch (*p) {
+                case 'h':
+                case 'H':
+                        if (hours) {
+                                *endp = p;
+                                return -1;
+                        }
+                        hours = n;
+                        break;
+                case 'm':
+                case 'M':
+                        if (minutes) {
+                                *endp = p;
+                                return -1;
+                        }
+                        minutes = n;
+                        break;
+                case 's':
+                case 'S':
+                        if (seconds) {
+                                *endp = p;
+                                return -1;
+                        }
+                        seconds = n;
+                        break;
+                default:
+                        *endp = p;
+                        if (!hours && !minutes && !seconds)
+                                return n;
+                        return (hours*60 + minutes)*60 + seconds;
+                }
+                p++;
+        }
+}
+
+void
+vmod_init(struct sess *sp, struct vmod_priv *priv, const char *param)
+{
+        struct binlog_config *conf = priv->priv;
+        struct stat st;
+        char *dir, *p, *q;
+        unsigned long n;
+
+        p = findparam(param, "debug");
+        if (p) {
+                conf->debug = atoi(p);
+                free(p);
+        }
+        
+        dir = findparam(param, "dir");
+        if (!dir) {
+                binlog_error("parameter \"dir\" not set");
+                abort();
+        }
+        if (stat(dir, &st)) {
+                if (errno == ENOENT)
+                        binlog_error("logging directory does not exist");
+                else
+                        binlog_error("cannot stat logging directory %s: %s",
+                                     dir, strerror(errno));