diff options
Diffstat (limited to 'src/fluent-bit/lib/monkey/plugins/cheetah/cmd.c')
-rw-r--r-- | src/fluent-bit/lib/monkey/plugins/cheetah/cmd.c | 472 |
1 files changed, 472 insertions, 0 deletions
diff --git a/src/fluent-bit/lib/monkey/plugins/cheetah/cmd.c b/src/fluent-bit/lib/monkey/plugins/cheetah/cmd.c new file mode 100644 index 000000000..6f02dad23 --- /dev/null +++ b/src/fluent-bit/lib/monkey/plugins/cheetah/cmd.c @@ -0,0 +1,472 @@ +/* -*- 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. + */ + +#include <monkey/mk_api.h> + +#include <pwd.h> +#include <ctype.h> + +#include "cheetah.h" +#include "cutils.h" +#include "cmd.h" + +/* strip leading and trailing space from input command line. */ +static char *strip_whitespace(char *cmd) +{ + char *end; + while (isspace(*cmd)) + cmd++; + if (*cmd == 0) + return cmd; + end = cmd + strlen(cmd) - 1; + while (end > cmd && isspace(*end)) + end--; + end++; + *end = '\0'; + return cmd; +} + +int mk_cheetah_cmd(char *raw_cmd, struct mk_server *server) +{ + char *cmd = strip_whitespace(raw_cmd); + if (strcmp(cmd, MK_CHEETAH_CONFIG) == 0 || + strcmp(cmd, MK_CHEETAH_CONFIG_SC) == 0) { + mk_cheetah_cmd_config(server); + } + else if (strcmp(cmd, MK_CHEETAH_STATUS) == 0 || + strcmp(cmd, MK_CHEETAH_STATUS_SC) == 0) { + mk_cheetah_cmd_status(server); + } + else if (strcmp(cmd, MK_CHEETAH_CLEAR) == 0 || + strcmp(cmd, MK_CHEETAH_CLEAR_SC) == 0) { + mk_cheetah_cmd_clear(); + } + else if (strcmp(cmd, MK_CHEETAH_UPTIME) == 0 || + strcmp(cmd, MK_CHEETAH_UPTIME_SC) == 0) { + mk_cheetah_cmd_uptime(server); + } + else if (strcmp(cmd, MK_CHEETAH_PLUGINS) == 0 || + strcmp(cmd, MK_CHEETAH_PLUGINS_SC) == 0) { + mk_cheetah_cmd_plugins(server); + } + else if (strcmp(cmd, MK_CHEETAH_WORKERS) == 0 || + strcmp(cmd, MK_CHEETAH_WORKERS_SC) == 0) { + mk_cheetah_cmd_workers(server); + } + else if (strcmp(cmd, MK_CHEETAH_VHOSTS) == 0 || + strcmp(cmd, MK_CHEETAH_VHOSTS_SC) == 0) { + mk_cheetah_cmd_vhosts(server); + } + else if (strcmp(cmd, MK_CHEETAH_HELP) == 0 || + strcmp(cmd, MK_CHEETAH_HELP_SC) == 0 || + strcmp(cmd, MK_CHEETAH_SHELP) == 0 || + strcmp(cmd, MK_CHEETAH_SHELP_SC) == 0) { + mk_cheetah_cmd_help(); + } + else if (strcmp(cmd, MK_CHEETAH_QUIT) == 0 || + strcmp(cmd, MK_CHEETAH_QUIT_SC) == 0) { + return mk_cheetah_cmd_quit(); + } + else if (strlen(cmd) == 0) { + return 0; + } + else { + CHEETAH_WRITE("Invalid command, type 'help' for a list of available commands\n"); + } + + CHEETAH_FLUSH(); + return 0; +} + +void mk_cheetah_cmd_clear() +{ + CHEETAH_WRITE("\033[2J\033[1;1H"); +} + +void mk_cheetah_cmd_uptime(struct mk_server *server) +{ + int days; + int hours; + int minutes; + int seconds; + long int upmind; + long int upminh; + long int uptime; + (void) server; + + /* uptime in seconds */ + uptime = time(NULL) - init_time; + + /* days */ + days = uptime / MK_CHEETAH_ONEDAY; + upmind = uptime - (days * MK_CHEETAH_ONEDAY); + + /* hours */ + hours = upmind / MK_CHEETAH_ONEHOUR; + upminh = upmind - hours * MK_CHEETAH_ONEHOUR; + + /* minutes */ + minutes = upminh / MK_CHEETAH_ONEMINUTE; + seconds = upminh - minutes * MK_CHEETAH_ONEMINUTE; + + CHEETAH_WRITE + ("Server has been running: %i day%s, %i hour%s, %i minute%s and %i second%s\n\n", + days, (days > 1) ? "s" : "", hours, (hours > 1) ? "s" : "", minutes, + (minutes > 1) ? "s" : "", seconds, (seconds > 1) ? "s" : ""); +} + +void mk_cheetah_cmd_plugins_print_core(struct mk_list *list) +{ + struct mk_plugin *p; + struct mk_list *head; + + CHEETAH_WRITE("\n%s[MASTER INIT]%s", ANSI_BOLD ANSI_BLUE, ANSI_RESET); + + mk_list_foreach(head, list) { + p = mk_list_entry(head, struct mk_plugin, _head); + + if (p->master_init) { + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + CHEETAH_WRITE("\n\n%s[WORKER INIT]%s", ANSI_BOLD ANSI_BLUE, ANSI_RESET); + + mk_list_foreach(head, list) { + p = mk_list_entry(head, struct mk_plugin, _head); + + if (p->worker_init) { + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + CHEETAH_WRITE("\n\n"); +} + +void mk_cheetah_cmd_plugins_print_network(struct mk_list *list) +{ + struct mk_plugin *p; + struct mk_list *head; + + CHEETAH_WRITE("%s[NETWORK I/O]%s", ANSI_BOLD ANSI_RED, ANSI_RESET); + + mk_list_foreach(head, list) { + p = mk_list_entry(head, struct mk_plugin, _head); + if (p->hooks & MK_PLUGIN_NETWORK_LAYER) { + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + CHEETAH_WRITE("\n"); +} + +void mk_cheetah_cmd_plugins(struct mk_server *server) +{ + struct mk_plugin *p; + struct mk_plugin_stage *s; + struct mk_list *head; + + if (mk_list_is_empty(&server->stage10_handler)) { + CHEETAH_WRITE("%s[%sSTAGE_10%s]%s", + ANSI_BOLD, ANSI_YELLOW, ANSI_WHITE, ANSI_RESET); + mk_list_foreach(head, &mk_api->config->stage10_handler) { + s = mk_list_entry(head, struct mk_plugin_stage, _head); + p = s->plugin; + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + if (mk_list_is_empty(&mk_api->config->stage20_handler)) { + CHEETAH_WRITE("%s[%sSTAGE_20%s]%s", + ANSI_BOLD, ANSI_YELLOW, ANSI_WHITE, ANSI_RESET); + mk_list_foreach(head, &mk_api->config->stage20_handler) { + s = mk_list_entry(head, struct mk_plugin_stage, _head); + p = s->plugin; + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + if (mk_list_is_empty(&mk_api->config->stage30_handler)) { + CHEETAH_WRITE("%s[%sSTAGE_30%s]%s", + ANSI_BOLD, ANSI_YELLOW, ANSI_WHITE, ANSI_RESET); + mk_list_foreach(head, &mk_api->config->stage30_handler) { + s = mk_list_entry(head, struct mk_plugin_stage, _head); + p = s->plugin; + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + if (mk_list_is_empty(&mk_api->config->stage40_handler)) { + CHEETAH_WRITE("%s[%sSTAGE_40%s]%s", + ANSI_BOLD, ANSI_YELLOW, ANSI_WHITE, ANSI_RESET); + mk_list_foreach(head, &mk_api->config->stage40_handler) { + s = mk_list_entry(head, struct mk_plugin_stage, _head); + p = s->plugin; + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + if (mk_list_is_empty(&mk_api->config->stage50_handler)) { + CHEETAH_WRITE("%s[%sSTAGE_50%s]%s", + ANSI_BOLD, ANSI_YELLOW, ANSI_WHITE, ANSI_RESET); + mk_list_foreach(head, &mk_api->config->stage50_handler) { + s = mk_list_entry(head, struct mk_plugin_stage, _head); + p = s->plugin; + CHEETAH_WRITE("\n [%s] %s v%s on \"%s\"", + p->shortname, p->name, p->version, p->path); + } + } + + CHEETAH_WRITE("\n\n"); +} + +void mk_cheetah_cmd_vhosts(struct mk_server *server) +{ + struct mk_vhost *entry_host; + struct mk_vhost_alias *entry_alias; + struct mk_rconf_section *section; + struct mk_rconf_entry *entry; + struct mk_list *hosts = &server->hosts; + struct mk_list *aliases; + struct mk_list *head_host; + struct mk_list *head_alias; + struct mk_list *head_sections; + struct mk_list *head_entries; + + mk_list_foreach(head_host, hosts) { + entry_host = mk_list_entry(head_host, struct mk_vhost, _head); + + aliases = &entry_host->server_names; + entry_alias = mk_list_entry_first(aliases, struct mk_vhost_alias, _head); + CHEETAH_WRITE("%s[%sVHost '%s'%s%s]%s\n", + ANSI_BOLD, ANSI_YELLOW, + entry_alias->name, ANSI_BOLD, ANSI_WHITE, ANSI_RESET); + + CHEETAH_WRITE(" - Names : "); + mk_list_foreach(head_alias, aliases) { + entry_alias = mk_list_entry(head_alias, struct mk_vhost_alias, _head); + CHEETAH_WRITE("%s ", entry_alias->name); + } + CHEETAH_WRITE("\n"); + + CHEETAH_WRITE(" - Document root : %s\n", entry_host->documentroot.data); + CHEETAH_WRITE(" - Config file : %s\n", entry_host->file); + + if (!entry_host->config) { + continue; + } + + mk_list_foreach(head_sections, &entry_host->config->sections) { + section = mk_list_entry(head_sections, struct mk_rconf_section, _head); + CHEETAH_WRITE(" %s+%s [%s]\n", ANSI_GREEN, ANSI_RESET, + section->name); + + mk_list_foreach(head_entries, §ion->entries) { + entry = mk_list_entry(head_entries, struct mk_rconf_entry, _head); + CHEETAH_WRITE(" - %11.10s : %s\n", entry->key, entry->val); + } + } + } + + CHEETAH_WRITE("\n"); +} + +void mk_cheetah_cmd_workers(struct mk_server *server) +{ + int i; + unsigned long long active_connections; + struct mk_sched_worker *node; + struct mk_sched_ctx *ctx; + + ctx = server->sched_ctx; + node = ctx->workers; + for (i=0; i < server->workers; i++) { + active_connections = (node[i].accepted_connections - node[i].closed_connections); + + CHEETAH_WRITE("* Worker %i\n", node[i].idx); + CHEETAH_WRITE(" - Task ID : %i\n", node[i].pid); + CHEETAH_WRITE(" - Active Connections: %llu\n", active_connections); + } + + CHEETAH_WRITE("\n"); +} + +int mk_cheetah_cmd_quit() +{ + CHEETAH_WRITE("Cheeta says: Good Bye!\n"); + if (listen_mode == LISTEN_STDIN) { + pthread_exit(NULL); + return 0; + } + else { + return -1; + } +} + +void mk_cheetah_cmd_help() +{ + CHEETAH_WRITE("List of available commands for Cheetah Shell\n"); + CHEETAH_WRITE("\ncommand shortcut description"); + CHEETAH_WRITE("\n----------------------------------------------------"); + CHEETAH_WRITE("\n? (\\?) Synonym for 'help'"); + CHEETAH_WRITE("\nconfig (\\f) Display global configuration"); + CHEETAH_WRITE("\nplugins (\\g) List loaded plugins and associated stages"); + CHEETAH_WRITE("\nstatus (\\s) Display general web server information"); + CHEETAH_WRITE("\nuptime (\\u) Display how long the web server has been running"); + CHEETAH_WRITE("\nvhosts (\\v) List virtual hosts configured"); + CHEETAH_WRITE("\nworkers (\\w) Show thread workers information\n"); + CHEETAH_WRITE("\nclear (\\c) Clear screen"); + CHEETAH_WRITE("\nhelp (\\h) Print this help"); + CHEETAH_WRITE("\nquit (\\q) Exit Cheetah shell :_(\n\n"); +} + +static void mk_cheetah_listen_config(struct mk_server *server) +{ + struct mk_list *head; + struct mk_config_listener *listener; + + mk_list_foreach(head, &server->listeners) { + listener = mk_list_entry(head, struct mk_config_listener, _head); + CHEETAH_WRITE("\nListen on : %s:%s", + listener->address, + listener->port); + } +} + +void mk_cheetah_cmd_config(struct mk_server *server) +{ + struct mk_string_line *entry; + struct mk_list *head; + struct mk_config_listener *listener; + + listener = mk_list_entry_first(&server->listeners, + struct mk_config_listener, + _head); + + CHEETAH_WRITE("Basic configuration"); + CHEETAH_WRITE("\n-------------------"); + mk_cheetah_listen_config(server); + CHEETAH_WRITE("\nWorkers : %i threads", mk_api->config->workers); + CHEETAH_WRITE("\nTimeout : %i seconds", mk_api->config->timeout); + CHEETAH_WRITE("\nPidFile : %s.%s", + mk_api->config->path_conf_pidfile, + listener->port); + CHEETAH_WRITE("\nUserDir : %s", + mk_api->config->conf_user_pub); + + + if (mk_list_is_empty(mk_api->config->index_files) == 0) { + CHEETAH_WRITE("\nIndexFile : No index files defined"); + } + else { + CHEETAH_WRITE("\nIndexFile : "); + mk_list_foreach(head, mk_api->config->index_files) { + entry = mk_list_entry(head, struct mk_string_line, _head); + CHEETAH_WRITE("%s ", entry->val); + } + + } + + CHEETAH_WRITE("\nHideVersion : "); + if (mk_api->config->hideversion == MK_TRUE) { + CHEETAH_WRITE("On"); + } + else { + CHEETAH_WRITE("Off"); + } + + CHEETAH_WRITE("\nResume : "); + if (mk_api->config->resume == MK_TRUE) { + CHEETAH_WRITE("On"); + } + else { + CHEETAH_WRITE("Off"); + } + + CHEETAH_WRITE("\nUser : %s", mk_api->config->user); + CHEETAH_WRITE("\n\nAdvanced configuration"); + CHEETAH_WRITE("\n----------------------"); + CHEETAH_WRITE("\nKeepAlive : "); + if (mk_api->config->keep_alive == MK_TRUE) { + CHEETAH_WRITE("On"); + } + else { + CHEETAH_WRITE("Off"); + } + CHEETAH_WRITE("\nMaxKeepAliveRequest : %i req/connection", + mk_api->config->max_keep_alive_request); + CHEETAH_WRITE("\nKeepAliveTimeout : %i seconds", mk_api->config->keep_alive_timeout); + CHEETAH_WRITE("\nMaxRequestSize : %i KB", + mk_api->config->max_request_size/1024); + CHEETAH_WRITE("\nSymLink : "); + if (mk_api->config->symlink == MK_TRUE) { + CHEETAH_WRITE("On"); + } + else { + CHEETAH_WRITE("Off"); + } + CHEETAH_WRITE("\n\n"); +} + +void mk_cheetah_cmd_status(struct mk_server *server) +{ + int nthreads = server->workers; + char tmp[64]; + + CHEETAH_WRITE("Monkey Version : %s\n", MK_VERSION_STR); + CHEETAH_WRITE("Configuration path : %s\n", server->path_conf_root); + + CHEETAH_WRITE("Cheetah! mode : "); + if (listen_mode == LISTEN_STDIN) { + CHEETAH_WRITE("STDIN\n"); + } + else { + CHEETAH_WRITE("SERVER @ %s\n", cheetah_server); + } + + CHEETAH_WRITE("Process ID : %i\n", getpid()); + CHEETAH_WRITE("Process User : "); + mk_cheetah_print_running_user(); + mk_cheetah_listen_config(server); + + CHEETAH_WRITE("\n"); + CHEETAH_WRITE("Worker Threads : %i (per configuration: %i)\n", + nthreads, server->workers); + + CHEETAH_WRITE("Memory Allocator : "); +#ifdef MALLOC_LIBC + CHEETAH_WRITE("libc, system default\n"); +#else + CHEETAH_WRITE("Jemalloc\n"); +#endif + + if (mk_api->kernel_features_print(tmp, sizeof(tmp), server) > 0) { + CHEETAH_WRITE("Kernel Features : %s\n", tmp); + } + + CHEETAH_WRITE("Events backend : %s\n", mk_api->ev_backend()); + CHEETAH_WRITE("\n"); +} |