openresty bundle

1 files changed, 195 insertions, 0 deletions

diff --git a/ngx_stream_lua-0.0.16/src/ngx_stream_lua_pcrefix.c b/ngx_stream_lua-0.0.16/src/ngx_stream_lua_pcrefix.c
new file mode 100644
index 0000000..2e8a9f8
--- /dev/null
+++ b/ngx_stream_lua-0.0.16/src/ngx_stream_lua_pcrefix.c
@@ -0,0 +1,195 @@
+
+/*
+ * !!! DO NOT EDIT DIRECTLY !!!
+ * This file was automatically generated from the following template:
+ *
+ * src/subsys/ngx_subsys_lua_pcrefix.c.tt2
+ */
+
+
+/*
+ * Copyright (C) Xiaozhe Wang (chaoslawful)
+ * Copyright (C) Yichun Zhang (agentzh)
+ */
+
+
+#ifndef DDEBUG
+#define DDEBUG 0
+#endif
+#include "ddebug.h"
+
+
+#include "ngx_stream_lua_pcrefix.h"
+#include "stdio.h"
+
+#if (NGX_PCRE)
+
+static ngx_pool_t *ngx_stream_lua_pcre_pool = NULL;
+
+#if (NGX_PCRE2)
+static ngx_uint_t  ngx_regex_direct_alloc;
+#else
+static void *(*old_pcre_malloc)(size_t);
+static void (*old_pcre_free)(void *ptr);
+#endif
+
+
+/* XXX: work-around to nginx regex subsystem, must init a memory pool
+ * to use PCRE functions. As PCRE still has memory-leaking problems,
+ * and nginx overwrote pcre_malloc/free hooks with its own static
+ * functions, so nobody else can reuse nginx regex subsystem... */
+#if (NGX_PCRE2)
+
+void *
+ngx_stream_lua_pcre_malloc(size_t size, void *data)
+{
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    if (ngx_stream_lua_pcre_pool) {
+        return ngx_palloc(ngx_stream_lua_pcre_pool, size);
+    }
+
+    if (ngx_regex_direct_alloc) {
+        return ngx_alloc(size, ngx_cycle->log);
+    }
+
+    fprintf(stderr, "error: lua pcre malloc failed due to empty pcre pool");
+
+    return NULL;
+}
+
+
+void
+ngx_stream_lua_pcre_free(void *ptr, void *data)
+{
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    if (ngx_stream_lua_pcre_pool) {
+        ngx_pfree(ngx_stream_lua_pcre_pool, ptr);
+        return;
+    }
+
+    if (ngx_regex_direct_alloc) {
+        ngx_free(ptr);
+        return;
+    }
+
+    fprintf(stderr, "error: lua pcre free failed due to empty pcre pool");
+}
+
+#else
+
+static void *
+ngx_stream_lua_pcre_malloc(size_t size)
+{
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    if (ngx_stream_lua_pcre_pool) {
+        return ngx_palloc(ngx_stream_lua_pcre_pool, size);
+    }
+
+    fprintf(stderr, "error: lua pcre malloc failed due to empty pcre pool");
+
+    return NULL;
+}
+
+
+static void
+ngx_stream_lua_pcre_free(void *ptr)
+{
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    if (ngx_stream_lua_pcre_pool) {
+        ngx_pfree(ngx_stream_lua_pcre_pool, ptr);
+        return;
+    }
+
+    fprintf(stderr, "error: lua pcre free failed due to empty pcre pool");
+}
+
+#endif
+
+
+#if (NGX_PCRE2)
+
+ngx_pool_t *
+ngx_stream_lua_pcre_malloc_init(ngx_pool_t *pool)
+{
+    ngx_pool_t          *old_pool;
+
+    dd("lua pcre pool was %p", ngx_stream_lua_pcre_pool);
+
+    ngx_regex_direct_alloc = (pool == NULL) ? 1 : 0;
+
+    old_pool = ngx_stream_lua_pcre_pool;
+    ngx_stream_lua_pcre_pool = pool;
+
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    return old_pool;
+}
+
+
+void
+ngx_stream_lua_pcre_malloc_done(ngx_pool_t *old_pool)
+{
+    dd("lua pcre pool was %p", ngx_stream_lua_pcre_pool);
+
+    ngx_stream_lua_pcre_pool = old_pool;
+    ngx_regex_direct_alloc = 0;
+
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+}
+
+#else
+
+ngx_pool_t *
+ngx_stream_lua_pcre_malloc_init(ngx_pool_t *pool)
+{
+    ngx_pool_t          *old_pool;
+
+    if (pcre_malloc != ngx_stream_lua_pcre_malloc) {
+
+        dd("overriding nginx pcre malloc and free");
+
+        ngx_stream_lua_pcre_pool = pool;
+
+        old_pcre_malloc = pcre_malloc;
+        old_pcre_free = pcre_free;
+
+        pcre_malloc = ngx_stream_lua_pcre_malloc;
+        pcre_free = ngx_stream_lua_pcre_free;
+
+        return NULL;
+    }
+
+    dd("lua pcre pool was %p", ngx_stream_lua_pcre_pool);
+
+    old_pool = ngx_stream_lua_pcre_pool;
+    ngx_stream_lua_pcre_pool = pool;
+
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    return old_pool;
+}
+
+
+void
+ngx_stream_lua_pcre_malloc_done(ngx_pool_t *old_pool)
+{
+    dd("lua pcre pool was %p", ngx_stream_lua_pcre_pool);
+
+    ngx_stream_lua_pcre_pool = old_pool;
+
+    dd("lua pcre pool is %p", ngx_stream_lua_pcre_pool);
+
+    if (old_pool == NULL) {
+        pcre_malloc = old_pcre_malloc;
+        pcre_free = old_pcre_free;
+    }
+}
+
+#endif
+#endif /* NGX_PCRE */
+
+/* vi:set ft=c ts=4 sw=4 et fdm=marker: */