1 files changed, 103 insertions, 0 deletions
diff --git a/fluent-bit/lib/monkey/mk_bin/mk_signals.c b/fluent-bit/lib/monkey/mk_bin/mk_signals.c
new file mode 100644
index 00000000..b2c9eb6e
--- /dev/null
+++ b/fluent-bit/lib/monkey/mk_bin/mk_signals.c
@@ -0,0 +1,103 @@
+/* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+
+/*  Monkey HTTP Server
+ *  ==================
+ *  Copyright 2001-2015 Monkey Software LLC <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.
+ */
+
+#include <monkey/monkey.h>
+#include <monkey/mk_core.h>
+
+#include "monkey.h"
+#include <string.h>
+#include <signal.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+
+static struct mk_server *server_context;
+
+void mk_signal_context(struct mk_server *ctx)
+{
+    server_context = ctx;
+}
+
+/* when we catch a signal and want to exit we call this function
+   to do it gracefully */
+static void mk_signal_exit()
+{
+    /* ignore future signals to properly handle the cleanup */
+    signal(SIGTERM, SIG_IGN);
+    signal(SIGINT,  SIG_IGN);
+    signal(SIGHUP,  SIG_IGN);
+
+    mk_user_undo_uidgid(server_context);
+    mk_utils_remove_pid(server_context->path_conf_pidfile);
+    mk_exit_all(server_context);
+
+    mk_info("Exiting... >:(");
+    _exit(EXIT_SUCCESS);
+}
+
+static void mk_signal_handler(int signo, siginfo_t *si, void *context UNUSED_PARAM)
+{
+    switch (signo) {
+    case SIGTERM:
+    case SIGINT:
+        mk_signal_exit();
+        break;
+    case SIGHUP:
+        /*
+         * TODO:
+         * we should implement the httpd config reload here (not in SIGUSR2).
+         * Daemon processes “overload” this signal with a mechanism to instruct them to
+         * reload their configuration files. Sending SIGHUP to Apache, for example,
+         * instructs it to reread httpd.conf.
+         */
+        mk_signal_exit();
+        break;
+    case SIGBUS:
+    case SIGSEGV:
+#ifdef DEBUG
+        mk_utils_stacktrace();
+#endif
+        mk_err("%s (%d), code=%d, addr=%p",
+               strsignal(signo), signo, si->si_code, si->si_addr);
+        //close(sched->server_fd);
+        //pthread_exit(NULL);
+        abort();
+    default:
+        /* let the kernel handle it */
+        kill(getpid(), signo);
+    }
+
+}
+
+void mk_signal_init(void *context)
+{
+    struct sigaction act;
+    memset(&act, 0x0, sizeof(act));
+
+    /* allow signals to be handled concurrently */
+    act.sa_flags = SA_SIGINFO | SA_NODEFER;
+    act.sa_sigaction = &mk_signal_handler;
+
+    sigaction(SIGSEGV, &act, NULL);
+    sigaction(SIGBUS,  &act, NULL);
+    sigaction(SIGHUP,  &act, NULL);
+    sigaction(SIGINT,  &act, NULL);
+    sigaction(SIGTERM, &act, NULL);
+
+    mk_signal_context(context);
+}