/* Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You 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 #include #include "ap_mpm.h" #include #include #include #include #include #include #include "unixd.h" #include "scoreboard.h" #include "mpm_common.h" #include "systemd/sd-daemon.h" #include "systemd/sd-journal.h" #if APR_HAVE_UNISTD_H #include #endif static int shutdown_timer = 0; static int shutdown_counter = 0; static unsigned long bytes_served; static pid_t mainpid; static char describe_listeners[50]; static int systemd_pre_config(apr_pool_t *pconf, apr_pool_t *plog, apr_pool_t *ptemp) { sd_notify(0, "RELOADING=1\n" "STATUS=Reading configuration...\n"); ap_extended_status = 1; return OK; } static char *dump_listener(ap_listen_rec *lr, apr_pool_t *p) { apr_sockaddr_t *sa = lr->bind_addr; char addr[128]; if (apr_sockaddr_is_wildcard(sa)) { return apr_pstrcat(p, "port ", apr_itoa(p, sa->port), NULL); } apr_sockaddr_ip_getbuf(addr, sizeof addr, sa); return apr_psprintf(p, "%s port %u", addr, sa->port); } static int systemd_post_config(apr_pool_t *pconf, apr_pool_t *plog, apr_pool_t *ptemp, server_rec *s) { ap_listen_rec *lr; apr_size_t plen = sizeof describe_listeners; char *p = describe_listeners; if (ap_state_query(AP_SQ_MAIN_STATE) == AP_SQ_MS_CREATE_PRE_CONFIG) return OK; for (lr = ap_listeners; lr; lr = lr->next) { char *s = dump_listener(lr, ptemp); if (strlen(s) + 3 < plen) { char *newp = apr_cpystrn(p, s, plen); if (lr->next) newp = apr_cpystrn(newp, ", ", 3); plen -= newp - p; p = newp; } else { if (plen < 4) { p = describe_listeners + sizeof describe_listeners - 4; plen = 4; } apr_cpystrn(p, "...", plen); break; } } sd_journal_print(LOG_INFO, "Server configured, listening on: %s", describe_listeners); return OK; } /* Report the service is ready in post_config, which could be during * startup or after a reload. The server could still hit a fatal * startup error after this point during ap_run_mpm(), so this is * perhaps too early, but by post_config listen() has been called on * the TCP ports so new connections will not be rejected. There will * always be a possible async failure event simultaneous to the * service reporting "ready", so this should be good enough. */ static int systemd_post_config_last(apr_pool_t *p, apr_pool_t *plog, apr_pool_t *ptemp, server_rec *main_server) { sd_notify(0, "READY=1\n" "STATUS=Configuration loaded.\n"); return OK; } static int systemd_pre_mpm(apr_pool_t *p, ap_scoreboard_e sb_type) { int rv; mainpid = getpid(); rv = sd_notifyf(0, "READY=1\n" "STATUS=Started, listening on: %s\n" "MAINPID=%" APR_PID_T_FMT, describe_listeners, mainpid); if (rv < 0) { ap_log_perror(APLOG_MARK, APLOG_ERR, 0, p, APLOGNO(02395) "sd_notifyf returned an error %d", rv); } return OK; } static int systemd_monitor(apr_pool_t *p, server_rec *s) { ap_sload_t sload; apr_interval_time_t up_time; char bps[5]; int rv; if (!ap_extended_status) { /* Nothing useful to report if ExtendedStatus disabled. */ return DECLINED; } ap_get_sload(&sload); if (sload.access_count == 0) { rv = sd_notifyf(0, "READY=1\n" "STATUS=Running, listening on: %s\n", describe_listeners); } else { /* up_time in seconds */ up_time = (apr_uint32_t) apr_time_sec(apr_time_now() - ap_scoreboard_image->global->restart_time); apr_strfsize((unsigned long)((float) (sload.bytes_served) / (float) up_time), bps); rv = sd_notifyf(0, "READY=1\n" "STATUS=Total requests: %lu; Idle/Busy workers %d/%d;" "Requests/sec: %.3g; Bytes served/sec: %sB/sec\n", sload.access_count, sload.idle, sload.busy, ((float) sload.access_count) / (float) up_time, bps); } if (rv < 0) { ap_log_error(APLOG_MARK, APLOG_ERR, 0, s, APLOGNO(02396) "sd_notifyf returned an error %d", rv); } /* Shutdown httpd when nothing is sent for shutdown_timer seconds. */ if (sload.bytes_served == bytes_served) { /* mpm_common.c: INTERVAL_OF_WRITABLE_PROBES is 10 */ shutdown_counter += 10; if (shutdown_timer > 0 && shutdown_counter >= shutdown_timer) { rv = sd_notifyf(0, "READY=1\n" "STATUS=Stopped as result of IdleShutdown " "timeout."); if (rv < 0) { ap_log_error(APLOG_MARK, APLOG_ERR, 0, s, APLOGNO(02804) "sd_notifyf returned an error %d", rv); } kill(mainpid, AP_SIG_GRACEFUL); } } else { shutdown_counter = 0; } bytes_served = sload.bytes_served; return DECLINED; } static void systemd_register_hooks(apr_pool_t *p) { /* Enable ap_extended_status. */ ap_hook_pre_config(systemd_pre_config, NULL, NULL, APR_HOOK_LAST); /* Grab the listener config. */ ap_hook_post_config(systemd_post_config, NULL, NULL, APR_HOOK_LAST); /* Signal service is ready. */ ap_hook_post_config(systemd_post_config_last, NULL, NULL, APR_HOOK_REALLY_LAST); /* We know the PID in this hook ... */ ap_hook_pre_mpm(systemd_pre_mpm, NULL, NULL, APR_HOOK_LAST); /* Used to update httpd's status line using sd_notifyf */ ap_hook_monitor(systemd_monitor, NULL, NULL, APR_HOOK_MIDDLE); } static const char *set_shutdown_timer(cmd_parms *cmd, void *dummy, const char *arg) { const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY); if (err != NULL) { return err; } shutdown_timer = atoi(arg); return NULL; } static const command_rec systemd_cmds[] = { AP_INIT_TAKE1("IdleShutdown", set_shutdown_timer, NULL, RSRC_CONF, "Number of seconds in idle-state after which httpd is shutdown"), {NULL} }; AP_DECLARE_MODULE(systemd) = { STANDARD20_MODULE_STUFF, NULL, NULL, NULL, NULL, systemd_cmds, systemd_register_hooks, };