1 files changed, 133 insertions, 0 deletions
diff --git a/fluent-bit/lib/monkey/plugins/cgi/cgi.h b/fluent-bit/lib/monkey/plugins/cgi/cgi.h
new file mode 100644
index 000000000..27123b7ae
--- /dev/null
+++ b/fluent-bit/lib/monkey/plugins/cgi/cgi.h
@@ -0,0 +1,133 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/*  Monkey HTTP Server
+ *  ==================
+ *  Copyright 2001-2017 Eduardo Silva <eduardo@monkey.io>
+ *  Copyright (C) 2012-2013, Lauri Kasanen
+ *
+ *  Licensed under the Apache License, Version 2.0 (the "License");
+ *  you may not use this file except in compliance with the License.
+ *  You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ *  Unless required by applicable law or agreed to in writing, software
+ *  distributed under the License is distributed on an "AS IS" BASIS,
+ *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ *  See the License for the specific language governing permissions and
+ *  limitations under the License.
+ */
+
+#ifndef MK_CGI_H
+#define MK_CGI_H
+
+#include <monkey/mk_api.h>
+
+#include <sys/types.h>
+#include <regex.h>
+#include <signal.h>
+#include <libgen.h>
+
+enum {
+    BUFLEN  = 4096,
+    PATHLEN = 1024,
+    SHORTLEN = 64
+};
+
+regex_t match_regex;
+
+struct cgi_request **requests_by_socket;
+
+struct post_t {
+    int fd;
+    void *buf;
+    unsigned long len;
+};
+
+struct cgi_match_t {
+    regex_t match;
+    char *bin;
+    mk_ptr_t content_type;
+
+    struct mk_list _head;
+};
+
+struct cgi_vhost_t {
+    struct mk_vhost *host;
+    struct mk_list matches;
+};
+
+struct cgi_vhost_t *cgi_vhosts;
+struct mk_list cgi_global_matches;
+
+
+struct cgi_request {
+    /* Built-in reference for the event loop */
+    struct mk_event  event;
+
+    char in_buf[BUFLEN];
+
+    struct mk_list _head;
+
+    struct mk_plugin *plugin;
+    struct mk_http_request *sr;
+    struct mk_http_session *cs;
+
+    unsigned int in_len;
+
+    int   fd;			/* Pipe the CGI proc */
+    int   socket;       /* Client connection */
+    int   hangup;       /* Should close connection when done ? */
+    int   active;       /* Active session ?  */
+    pid_t child;        /* child process ID  */
+    unsigned char status_done;
+    unsigned char all_headers_done;
+    unsigned char chunked;
+};
+
+/* Global list per worker */
+pthread_key_t cgi_request_list;
+
+extern struct cgi_request **requests_by_socket;
+
+void cgi_finish(struct cgi_request *r);
+
+int swrite(const int fd, const void *buf, const size_t count);
+int channel_write(struct cgi_request *r, void *buf, size_t count);
+
+struct cgi_request *cgi_req_create(int fd, int socket,
+                                   struct mk_plugin *plugin,
+                                   struct mk_http_request *sr,
+                                   struct mk_http_session *cs);
+void cgi_req_add(struct cgi_request *r);
+int cgi_req_del(struct cgi_request *r);
+
+// Get the CGI request by the client socket
+static inline struct cgi_request *cgi_req_get(int socket)
+{
+    struct cgi_request *r = requests_by_socket[socket];
+    return r;
+}
+
+// Get the CGI request by the CGI app's fd
+static inline struct cgi_request *cgi_req_get_by_fd(int fd)
+{
+    struct mk_list *list, *node;
+    struct cgi_request *r;
+
+    list = pthread_getspecific(cgi_request_list);
+    if (mk_list_is_empty(list) == 0)
+        return NULL;
+
+    mk_list_foreach(node, list) {
+        r = mk_list_entry(node, struct cgi_request, _head);
+        if (r->fd == fd)
+            return r;
+    }
+
+    return NULL;
+}
+
+int cb_cgi_read(void *data);
+
+#endif