mod_heartmonitor.c revision eec239e376d6338d3e89c39ad3c1f8934ff09af3
0922cbe8300e97215564748d449824f458196335Lennart Poettering/* Licensed to the Apache Software Foundation (ASF) under one or more
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * contributor license agreements. See the NOTICE file distributed with
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * this work for additional information regarding copyright ownership.
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * The ASF licenses this file to You under the Apache License, Version 2.0
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * (the "License"); you may not use this file except in compliance with
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * the License. You may obtain a copy of the License at
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering *
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * http://www.apache.org/licenses/LICENSE-2.0
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering *
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * Unless required by applicable law or agreed to in writing, software
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * distributed under the License is distributed on an "AS IS" BASIS,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * See the License for the specific language governing permissions and
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * limitations under the License.
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering */
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "httpd.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "http_config.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "http_log.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "http_core.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "apr_strings.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "apr_hash.h"
ef7963b561adec267dbc7dbdfc0d29105f2ca872Lennart Poettering#include "apr_time.h"
ef7963b561adec267dbc7dbdfc0d29105f2ca872Lennart Poettering#include "ap_mpm.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "scoreboard.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#include "mod_watchdog.h"
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#ifndef HM_UPDATE_SEC
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering/* How often we update the stats file */
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering/* TODO: Make a runtime config */
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#define HM_UPDATE_SEC (5)
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#endif
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering#define HM_WATHCHDOG_NAME ("_heartmonitor_")
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringmodule AP_MODULE_DECLARE_DATA heartmonitor_module;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringtypedef struct hm_server_t
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering const char *ip;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering int busy;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering int ready;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_time_t seen;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering} hm_server_t;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringtypedef struct hm_ctx_t
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering int active;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering const char *storage_path;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_watchdog_t *watchdog;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_interval_time_t interval;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_sockaddr_t *mcast_addr;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_status_t status;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering volatile int keep_running;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_socket_t *sock;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_pool_t *p;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_hash_t *servers;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering server_rec *s;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering} hm_ctx_t;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringstatic apr_status_t hm_listen(hm_ctx_t *ctx)
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_status_t rv;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering rv = apr_socket_create(&ctx->sock, ctx->mcast_addr->family,
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering SOCK_DGRAM, APR_PROTO_UDP, ctx->p);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering "Heartmonitor: Failed to create listening socket.");
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering return rv;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering rv = apr_socket_opt_set(ctx->sock, APR_SO_REUSEADDR, 1);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering if (rv) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering "Heartmonitor: Failed to set APR_SO_REUSEADDR to 1 on socket.");
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering rv = apr_socket_opt_set(ctx->sock, APR_SO_NONBLOCK, 1);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering if (rv) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Failed to set APR_SO_REUSEADDR to 1 on socket.");
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering rv = apr_socket_bind(ctx->sock, ctx->mcast_addr);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Failed to bind on socket.");
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering rv = apr_mcast_join(ctx->sock, ctx->mcast_addr, NULL, NULL);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Failed to join multicast group");
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering rv = apr_mcast_loopback(ctx->sock, 1);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Failed to accept localhost mulitcast on socket.");
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return APR_SUCCESS;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering}
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringstatic void qs_to_table(const char *input, apr_table_t *parms,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_pool_t *p)
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering char *key;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering char *value;
5190bbb240ab3360f7d37714d1c877858aef8c12Thomas Hindoe Paaboel Andersen char *query_string;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering char *strtok_state;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering if (input == NULL) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering return;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
772c552f763a125d17d814c4e5a35bbba49fc27aNis Martensen query_string = apr_pstrdup(p, input);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering key = apr_strtok(query_string, "&", &strtok_state);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering while (key) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering value = strchr(key, '=');
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering if (value) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering *value = '\0'; /* Split the string in two */
772c552f763a125d17d814c4e5a35bbba49fc27aNis Martensen value++; /* Skip passed the = */
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering else {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering value = "1";
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_unescape_url(key);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_unescape_url(value);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_table_set(parms, key, value);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering /*
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_rerror(APLOG_MARK, APLOG_DEBUG, 0, r,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Found query arg: %s = %s", key, value);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering */
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering key = apr_strtok(NULL, "&", &strtok_state);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering}
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering#define SEEN_TIMEOUT (30)
772c552f763a125d17d814c4e5a35bbba49fc27aNis Martensen
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringstatic apr_status_t hm_update_stats(hm_ctx_t *ctx, apr_pool_t *p)
772c552f763a125d17d814c4e5a35bbba49fc27aNis Martensen{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_status_t rv;
ba9904e9ce0628cce3bbd8106f6dc6914e418edaLennart Poettering apr_file_t *fp;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_hash_index_t *hi;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_time_t now;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering char *path = apr_pstrcat(p, ctx->storage_path, ".tmp.XXXXXX", NULL);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering /* TODO: Update stats file (!) */
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering rv = apr_file_mktemp(&fp, path, APR_CREATE | APR_WRITE, p);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Unable to open tmp file: %s", path);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering now = apr_time_now();
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering for (hi = apr_hash_first(p, ctx->servers);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering hi != NULL; hi = apr_hash_next(hi)) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering hm_server_t *s = NULL;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_uint32_t seen;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering apr_hash_this(hi, NULL, NULL, (void **) &s);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering seen = apr_time_sec(now - s->seen);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering if (seen > SEEN_TIMEOUT) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering /*
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering * Skip this entry from the heartbeat file -- when it comes back,
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering * we will reuse the memory...
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering */
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering else {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_file_printf(fp, "%s &ready=%u&busy=%u&lastseen=%u\n",
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering s->ip, s->ready, s->busy, seen);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering rv = apr_file_flush(fp);
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering if (rv) {
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering "Heartmonitor: Unable to flush file: %s", path);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
d4205751d4643c272059a3728045929dd0e5e800Lennart Poettering rv = apr_file_close(fp);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Unable to close file: %s", path);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering rv = apr_file_perms_set(path,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering APR_FPROT_UREAD | APR_FPROT_GREAD |
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering APR_FPROT_WREAD);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv && rv != APR_INCOMPLETE) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Unable to set file permssions on %s",
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering path);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering rv = apr_file_rename(path, ctx->storage_path, p);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (rv) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering "Heartmonitor: Unable to move file: %s -> %s", path,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering ctx->storage_path);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return rv;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return APR_SUCCESS;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering}
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringstatic hm_server_t *hm_get_server(hm_ctx_t *ctx, const char *ip)
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering hm_server_t *s;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering s = apr_hash_get(ctx->servers, ip, APR_HASH_KEY_STRING);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (s == NULL) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering s = apr_palloc(ctx->p, sizeof(hm_server_t));
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering s->ip = apr_pstrdup(ctx->p, ip);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering s->ready = 0;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering s->busy = 0;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering s->seen = 0;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_hash_set(ctx->servers, s->ip, APR_HASH_KEY_STRING, s);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering }
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering return s;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering}
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering/* Process a message receive from a backend node */
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poetteringstatic void hm_processmsg(hm_ctx_t *ctx, apr_pool_t *p,
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_sockaddr_t *from, char *buf, int len)
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering{
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_table_t *tbl;
5190bbb240ab3360f7d37714d1c877858aef8c12Thomas Hindoe Paaboel Andersen
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering buf[len] = '\0';
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5190bbb240ab3360f7d37714d1c877858aef8c12Thomas Hindoe Paaboel Andersen tbl = apr_table_make(p, 10);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering qs_to_table(buf, tbl, p);
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering if (apr_table_get(tbl, "v") != NULL &&
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_table_get(tbl, "busy") != NULL &&
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering apr_table_get(tbl, "ready") != NULL) {
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering char *ip;
5d6a86d7a034a1fb3d6e3f1b58e2c13739270894Lennart Poettering hm_server_t *s;
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek /* TODO: REMOVE ME BEFORE PRODUCTION (????) */
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, ctx->s,
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek "Heartmonitor: %pI busy=%s ready=%s", from,
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek apr_table_get(tbl, "busy"), apr_table_get(tbl, "ready"));
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek apr_sockaddr_ip_get(&ip, from);
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek s = hm_get_server(ctx, ip);
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek s->busy = atoi(apr_table_get(tbl, "busy"));
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek s->ready = atoi(apr_table_get(tbl, "ready"));
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek s->seen = apr_time_now();
20ad4cfd8e5592f634f20468798cbc1055ab9fafZbigniew Jędrzejewski-Szmek }
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering else {
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering ap_log_error(APLOG_MARK, APLOG_CRIT, 0, ctx->s,
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering "Heartmonitor: malformed message from %pI",
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering from);
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering }
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering}
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering/* Read message from multicast socket */
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering#define MAX_MSG_LEN (1000)
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poetteringstatic apr_status_t hm_recv(hm_ctx_t *ctx, apr_pool_t *p)
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering{
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering char buf[MAX_MSG_LEN + 1];
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering apr_sockaddr_t from;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering apr_size_t len = MAX_MSG_LEN;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering apr_status_t rv;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering from.pool = p;
9444b1f20e311f073864d81e913bd4f32fe95cfdLennart Poettering
rv = apr_socket_recvfrom(&from, ctx->sock, 0, buf, &len);
if (APR_STATUS_IS_EAGAIN(rv)) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
"Heartmonitor: would block");
return APR_SUCCESS;
}
else if (rv) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
"Heartmonitor: recvfrom failed");
return rv;
}
hm_processmsg(ctx, p, &from, buf, len);
return rv;
}
static apr_status_t hm_watchdog_callback(int state, void *data,
apr_pool_t *pool)
{
apr_status_t rv = APR_SUCCESS;
apr_time_t cur, now;
hm_ctx_t *ctx = (hm_ctx_t *)data;
if (!ctx->active) {
return rv;
}
switch (state) {
case AP_WATCHDOG_STATE_STARTING:
rv = hm_listen(ctx);
if (rv) {
ctx->status = rv;
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s,
"Heartmonitor: Unable to listen for connections!");
}
else {
ctx->keep_running = 1;
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, ctx->s,
"Heartmonitor: %s listener started.",
HM_WATHCHDOG_NAME);
}
break;
case AP_WATCHDOG_STATE_RUNNING:
hm_update_stats(ctx, pool);
cur = now = apr_time_sec(apr_time_now());
/* TODO: Insted HN_UPDATE_SEC use
* the ctx->interval
*/
while ((now - cur) < apr_time_sec(ctx->interval)) {
int n;
apr_status_t rc;
apr_pool_t *p;
apr_pollfd_t pfd;
apr_interval_time_t timeout;
apr_pool_create(&p, pool);
pfd.desc_type = APR_POLL_SOCKET;
pfd.desc.s = ctx->sock;
pfd.p = p;
pfd.reqevents = APR_POLLIN;
timeout = apr_time_from_sec(1);
rc = apr_poll(&pfd, 1, &n, timeout);
if (!ctx->keep_running) {
apr_pool_destroy(p);
break;
}
if (rc == APR_SUCCESS && (pfd.rtnevents & APR_POLLIN)) {
hm_recv(ctx, p);
}
now = apr_time_sec(apr_time_now());
apr_pool_destroy(p);
}
break;
case AP_WATCHDOG_STATE_STOPPING:
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, ctx->s,
"Heartmonitor: stopping %s listener.",
HM_WATHCHDOG_NAME);
ctx->keep_running = 0;
if (ctx->sock) {
apr_socket_close(ctx->sock);
ctx->sock = NULL;
}
break;
}
return rv;
}
static int hm_post_config(apr_pool_t *p, apr_pool_t *plog,
apr_pool_t *ptemp, server_rec *s)
{
apr_status_t rv;
hm_ctx_t *ctx = ap_get_module_config(s->module_config,
&heartmonitor_module);
if (!ctx->active) {
return OK;
}
rv = ap_watchdog_get_instance(&ctx->watchdog,
HM_WATHCHDOG_NAME,
0, 1, p);
if (rv) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s,
"Heartmonitor: Failed to create watchdog "
"instance (%s)", HM_WATHCHDOG_NAME);
return !OK;
}
/* Register a callback with zero interval. */
rv = ap_watchdog_register_callback(ctx->watchdog,
0,
ctx,
hm_watchdog_callback);
if (rv) {
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s,
"Heartmonitor: Failed to register watchdog "
"callback (%s)", HM_WATHCHDOG_NAME);
return !OK;
}
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, s,
"Heartmonitor: wd callback %s", HM_WATHCHDOG_NAME);
return OK;
}
static int hm_handler(request_rec *r)
{
apr_bucket_brigade *input_brigade;
apr_size_t len=MAX_MSG_LEN;
char *buf;
apr_status_t status;
hm_ctx_t *ctx = ap_get_module_config(r->server->module_config,
&heartmonitor_module);
if (strcmp(r->handler, "hearthbeat")) {
return DECLINED;
}
if (r->method_number != M_POST) {
return HTTP_METHOD_NOT_ALLOWED;
}
buf = apr_pcalloc(r->pool, MAX_MSG_LEN);
input_brigade = apr_brigade_create(r->connection->pool, r->connection->bucket_alloc);
status = ap_get_brigade(r->input_filters, input_brigade, AP_MODE_READBYTES, APR_BLOCK_READ, MAX_MSG_LEN);
if (status != APR_SUCCESS) {
return HTTP_INTERNAL_SERVER_ERROR;
}
apr_brigade_flatten(input_brigade, buf, &len);
hm_processmsg(ctx, r->pool, r->connection->remote_addr, buf, len);
return HTTP_OK;
}
static void hm_register_hooks(apr_pool_t *p)
{
static const char * const aszSucc[]={ "mod_proxy.c", NULL };
ap_hook_post_config(hm_post_config, NULL, NULL, APR_HOOK_MIDDLE);
ap_hook_handler(hm_handler, NULL, aszSucc, APR_HOOK_FIRST);
}
static void *hm_create_config(apr_pool_t *p, server_rec *s)
{
hm_ctx_t *ctx = (hm_ctx_t *) apr_palloc(p, sizeof(hm_ctx_t));
ctx->active = 0;
ctx->storage_path = ap_server_root_relative(p, "logs/hb.dat");
/* TODO: Add directive for tuning the update interval
*/
ctx->interval = apr_time_from_sec(HM_UPDATE_SEC);
ctx->s = s;
apr_pool_create(&ctx->p, p);
ctx->servers = apr_hash_make(ctx->p);
return ctx;
}
static const char *cmd_hm_storage(cmd_parms *cmd,
void *dconf, const char *path)
{
apr_pool_t *p = cmd->pool;
hm_ctx_t *ctx =
(hm_ctx_t *) ap_get_module_config(cmd->server->module_config,
&heartmonitor_module);
const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
if (err != NULL) {
return err;
}
ctx->storage_path = ap_server_root_relative(p, path);
return NULL;
}
static const char *cmd_hm_listen(cmd_parms *cmd,
void *dconf, const char *mcast_addr)
{
apr_status_t rv;
char *host_str;
char *scope_id;
apr_port_t port = 0;
apr_pool_t *p = cmd->pool;
hm_ctx_t *ctx =
(hm_ctx_t *) ap_get_module_config(cmd->server->module_config,
&heartmonitor_module);
const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
if (err != NULL) {
return err;
}
if (!ctx->active) {
ctx->active = 1;
}
else {
return "HeartbeatListen: May only be specified once.";
}
rv = apr_parse_addr_port(&host_str, &scope_id, &port, mcast_addr, cmd->temp_pool);
if (rv) {
return "HeartbeatListen: Unable to parse multicast address.";
}
if (host_str == NULL) {
return "HeartbeatListen: No host provided in multicast address";
}
if (port == 0) {
return "HeartbeatListen: No port provided in multicast address";
}
rv = apr_sockaddr_info_get(&ctx->mcast_addr, host_str, APR_INET, port, 0,
p);
if (rv) {
return
"HeartbeatListen: apr_sockaddr_info_get failed on multicast address";
}
return NULL;
}
static const command_rec hm_cmds[] = {
AP_INIT_TAKE1("HeartbeatListen", cmd_hm_listen, NULL, RSRC_CONF,
"Address to listen for heartbeat requests"),
AP_INIT_TAKE1("HeartbeatStorage", cmd_hm_storage, NULL, RSRC_CONF,
"Path to store heartbeat data."),
{NULL}
};
module AP_MODULE_DECLARE_DATA heartmonitor_module = {
STANDARD20_MODULE_STUFF,
NULL, /* create per-directory config structure */
NULL, /* merge per-directory config structures */
hm_create_config, /* create per-server config structure */
NULL, /* merge per-server config structures */
hm_cmds, /* command apr_table_t */
hm_register_hooks
};