1 files changed, 97 insertions, 0 deletions
diff --git a/ngx_postgres-1.0/src/ngx_postgres_escape.c b/ngx_postgres-1.0/src/ngx_postgres_escape.c
new file mode 100644
index 0000000..dd78194
--- /dev/null
+++ b/ngx_postgres-1.0/src/ngx_postgres_escape.c
@@ -0,0 +1,97 @@
+/*
+ * Copyright (c) 2010, FRiCKLE Piotr Sikora <info@frickle.com>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+ * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+ * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+ * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+ * HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
+ * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#ifndef DDEBUG
+#define DDEBUG 0
+#endif
+
+#include "ngx_postgres_ddebug.h"
+#include "ngx_postgres_escape.h"
+#include "ngx_postgres_module.h"
+
+#include <libpq-fe.h>
+
+
+uintptr_t ngx_postgres_script_exit_code = (uintptr_t) NULL;
+
+
+void
+ngx_postgres_escape_string(ngx_http_script_engine_t *e)
+{
+    ngx_postgres_escape_t      *pge;
+    ngx_http_variable_value_t  *v;
+    u_char                     *p, *s;
+
+    v = e->sp - 1;
+
+    dd("entering: \"%.*s\"", (int) v->len, v->data);
+
+    pge = (ngx_postgres_escape_t *) e->ip;
+    e->ip += sizeof(ngx_postgres_escape_t);
+
+    if ((v == NULL) || (v->not_found)) {
+        v->data = (u_char *) "NULL";
+        v->len = sizeof("NULL") - 1;
+        dd("returning (NULL)");
+        goto done;
+    }
+
+    if (v->len == 0) {
+        if (pge->empty) {
+            v->data = (u_char *) "''";
+            v->len = 2;
+            dd("returning (empty/empty)");
+            goto done;
+        } else {
+            v->data = (u_char *) "NULL";
+            v->len = sizeof("NULL") - 1;
+            dd("returning (empty/NULL)");
+            goto done;
+        }
+    }
+
+    s = p = ngx_pnalloc(e->request->pool, 2 * v->len + 2);
+    if (p == NULL) {
+        e->ip = (u_char *) &ngx_postgres_script_exit_code;
+        e->status = NGX_HTTP_INTERNAL_SERVER_ERROR;
+        dd("returning (NGX_HTTP_INTERNAL_SERVER_ERROR)");
+        return;
+    }
+
+    *p++ = '\'';
+    v->len = PQescapeString((char *) p, (const char *) v->data, v->len);
+    p[v->len] = '\'';
+    v->len += 2;
+    v->data = s;
+
+    dd("returning");
+
+done:
+
+    v->valid = 1;
+    v->no_cacheable = 0;
+    v->not_found = 0;
+}