1 files changed, 99 insertions, 0 deletions
diff --git a/fluent-bit/lib/monkey/include/monkey/mk_plugin_stage.h b/fluent-bit/lib/monkey/include/monkey/mk_plugin_stage.h
new file mode 100644
index 000000000..5ad4b73ec
--- /dev/null
+++ b/fluent-bit/lib/monkey/include/monkey/mk_plugin_stage.h
@@ -0,0 +1,99 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/*  Monkey HTTP Server
+ *  ==================
+ *  Copyright 2001-2017 Eduardo Silva <eduardo@monkey.io>
+ *
+ *  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_PLUGIN_STAGE_H
+#define MK_PLUGIN_STAGE_H
+
+static inline int mk_plugin_stage_run_10(int socket, struct mk_server *server)
+{
+    int ret;
+    struct mk_list *head;
+    struct mk_plugin_stage *stage;
+
+    mk_list_foreach(head, &server->stage10_handler) {
+        stage = mk_list_entry(head, struct mk_plugin_stage, _head);
+        ret = stage->stage10(socket);
+        switch (ret) {
+        case MK_PLUGIN_RET_CLOSE_CONX:
+            MK_TRACE("return MK_PLUGIN_RET_CLOSE_CONX");
+            return MK_PLUGIN_RET_CLOSE_CONX;
+        }
+    }
+
+    return -1;
+}
+
+static inline int mk_plugin_stage_run_20(struct mk_http_session *cs,
+                                         struct mk_http_request *sr,
+                                         struct mk_server *server)
+{
+    int ret;
+    struct mk_list *head;
+    struct mk_plugin_stage *stage;
+
+    mk_list_foreach(head, &server->stage20_handler) {
+        stage = mk_list_entry(head, struct mk_plugin_stage, _head);
+        ret = stage->stage20(cs, sr);
+        switch (ret) {
+        case MK_PLUGIN_RET_CLOSE_CONX:
+            MK_TRACE("return MK_PLUGIN_RET_CLOSE_CONX");
+            return MK_PLUGIN_RET_CLOSE_CONX;
+        }
+    }
+
+    return -1;
+}
+
+static inline int mk_plugin_stage_run_40(struct mk_http_session *cs,
+                                         struct mk_http_request *sr,
+                                         struct mk_server *server)
+{
+    struct mk_list *head;
+    struct mk_plugin_stage *stage;
+
+    mk_list_foreach(head, &server->stage40_handler) {
+        stage = mk_list_entry(head, struct mk_plugin_stage, _head);
+        stage->stage40(cs, sr);
+    }
+
+    return -1;
+}
+
+static inline int mk_plugin_stage_run_50(int socket, struct mk_server *server)
+{
+    int ret;
+
+    struct mk_list *head;
+    struct mk_plugin_stage *stage;
+
+    mk_list_foreach(head, &server->stage50_handler) {
+        stage = mk_list_entry(head, struct mk_plugin_stage, _head);
+        ret = stage->stage50(socket);
+        switch (ret) {
+        case MK_PLUGIN_RET_NOT_ME:
+            break;
+        case MK_PLUGIN_RET_CONTINUE:
+            return MK_PLUGIN_RET_CONTINUE;
+        }
+    }
+
+    return -1;
+}
+
+#endif