|
Packit |
90a5c9 |
/* Licensed to the Apache Software Foundation (ASF) under one or more
|
|
Packit |
90a5c9 |
* contributor license agreements. See the NOTICE file distributed with
|
|
Packit |
90a5c9 |
* this work for additional information regarding copyright ownership.
|
|
Packit |
90a5c9 |
* The ASF licenses this file to You under the Apache License, Version 2.0
|
|
Packit |
90a5c9 |
* (the "License"); you may not use this file except in compliance with
|
|
Packit |
90a5c9 |
* the License. You may obtain a copy of the License at
|
|
Packit |
90a5c9 |
*
|
|
Packit |
90a5c9 |
* http://www.apache.org/licenses/LICENSE-2.0
|
|
Packit |
90a5c9 |
*
|
|
Packit |
90a5c9 |
* Unless required by applicable law or agreed to in writing, software
|
|
Packit |
90a5c9 |
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
Packit |
90a5c9 |
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
Packit |
90a5c9 |
* See the License for the specific language governing permissions and
|
|
Packit |
90a5c9 |
* limitations under the License.
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
#include "httpd.h"
|
|
Packit |
90a5c9 |
#include "http_config.h"
|
|
Packit |
90a5c9 |
#include "http_log.h"
|
|
Packit |
90a5c9 |
#include "http_core.h"
|
|
Packit |
90a5c9 |
#include "http_protocol.h"
|
|
Packit |
90a5c9 |
#include "apr_strings.h"
|
|
Packit |
90a5c9 |
#include "apr_hash.h"
|
|
Packit |
90a5c9 |
#include "apr_time.h"
|
|
Packit |
90a5c9 |
#include "ap_mpm.h"
|
|
Packit |
90a5c9 |
#include "scoreboard.h"
|
|
Packit |
90a5c9 |
#include "mod_watchdog.h"
|
|
Packit |
90a5c9 |
#include "ap_slotmem.h"
|
|
Packit |
90a5c9 |
#include "heartbeat.h"
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
#ifndef HM_UPDATE_SEC
|
|
Packit |
90a5c9 |
/* How often we update the stats file */
|
|
Packit |
90a5c9 |
/* TODO: Make a runtime config */
|
|
Packit |
90a5c9 |
#define HM_UPDATE_SEC (5)
|
|
Packit |
90a5c9 |
#endif
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
#define HM_WATHCHDOG_NAME ("_heartmonitor_")
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const ap_slotmem_provider_t *storage = NULL;
|
|
Packit |
90a5c9 |
static ap_slotmem_instance_t *slotmem = NULL;
|
|
Packit |
90a5c9 |
static int maxworkers = 0;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
module AP_MODULE_DECLARE_DATA heartmonitor_module;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
typedef struct hm_server_t
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
const char *ip;
|
|
Packit |
90a5c9 |
int busy;
|
|
Packit |
90a5c9 |
int ready;
|
|
Packit |
90a5c9 |
unsigned int port;
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
} hm_server_t;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
typedef struct hm_ctx_t
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
int active;
|
|
Packit |
90a5c9 |
const char *storage_path;
|
|
Packit |
90a5c9 |
ap_watchdog_t *watchdog;
|
|
Packit |
90a5c9 |
apr_interval_time_t interval;
|
|
Packit |
90a5c9 |
apr_sockaddr_t *mcast_addr;
|
|
Packit |
90a5c9 |
apr_status_t status;
|
|
Packit |
90a5c9 |
volatile int keep_running;
|
|
Packit |
90a5c9 |
apr_socket_t *sock;
|
|
Packit |
90a5c9 |
apr_pool_t *p;
|
|
Packit |
90a5c9 |
apr_hash_t *servers;
|
|
Packit |
90a5c9 |
server_rec *s;
|
|
Packit |
90a5c9 |
} hm_ctx_t;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
typedef struct hm_slot_server_ctx_t {
|
|
Packit |
90a5c9 |
hm_server_t *s;
|
|
Packit |
90a5c9 |
int found;
|
|
Packit |
90a5c9 |
unsigned int item_id;
|
|
Packit |
90a5c9 |
} hm_slot_server_ctx_t;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static apr_status_t hm_listen(hm_ctx_t *ctx)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_socket_create(&ctx->sock, ctx->mcast_addr->family,
|
|
Packit |
90a5c9 |
SOCK_DGRAM, APR_PROTO_UDP, ctx->p);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02068)
|
|
Packit |
90a5c9 |
"Failed to create listening socket.");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_socket_opt_set(ctx->sock, APR_SO_REUSEADDR, 1);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02069)
|
|
Packit |
90a5c9 |
"Failed to set APR_SO_REUSEADDR to 1 on socket.");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_socket_opt_set(ctx->sock, APR_SO_NONBLOCK, 1);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02070)
|
|
Packit |
90a5c9 |
"Failed to set APR_SO_NONBLOCK to 1 on socket.");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_socket_bind(ctx->sock, ctx->mcast_addr);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02071)
|
|
Packit |
90a5c9 |
"Failed to bind on socket.");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_mcast_join(ctx->sock, ctx->mcast_addr, NULL, NULL);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02072)
|
|
Packit |
90a5c9 |
"Failed to join multicast group");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_mcast_loopback(ctx->sock, 1);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02073)
|
|
Packit |
90a5c9 |
"Failed to accept localhost mulitcast on socket.");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* XXX: The same exists in mod_lbmethod_heartbeat.c where it is named argstr_to_table */
|
|
Packit |
90a5c9 |
static void qs_to_table(const char *input, apr_table_t *parms,
|
|
Packit |
90a5c9 |
apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
char *key;
|
|
Packit |
90a5c9 |
char *value;
|
|
Packit |
90a5c9 |
char *query_string;
|
|
Packit |
90a5c9 |
char *strtok_state;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (input == NULL) {
|
|
Packit |
90a5c9 |
return;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
query_string = apr_pstrdup(p, input);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
key = apr_strtok(query_string, "&", &strtok_state);
|
|
Packit |
90a5c9 |
while (key) {
|
|
Packit |
90a5c9 |
value = strchr(key, '=');
|
|
Packit |
90a5c9 |
if (value) {
|
|
Packit |
90a5c9 |
*value = '\0'; /* Split the string in two */
|
|
Packit |
90a5c9 |
value++; /* Skip passed the = */
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
value = "1";
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
ap_unescape_url(key);
|
|
Packit |
90a5c9 |
ap_unescape_url(value);
|
|
Packit |
90a5c9 |
apr_table_set(parms, key, value);
|
|
Packit |
90a5c9 |
/*
|
|
Packit |
90a5c9 |
ap_log_rerror(APLOG_MARK, APLOG_DEBUG, 0, r, APLOGNO(03182)
|
|
Packit |
90a5c9 |
"Found query arg: %s = %s", key, value);
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
key = apr_strtok(NULL, "&", &strtok_state);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
#define SEEN_TIMEOUT (30)
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Store in the slotmem */
|
|
Packit |
90a5c9 |
static apr_status_t hm_update(void* mem, void *data, apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
hm_slot_server_t *old = (hm_slot_server_t *) mem;
|
|
Packit |
90a5c9 |
hm_slot_server_ctx_t *s = (hm_slot_server_ctx_t *) data;
|
|
Packit |
90a5c9 |
hm_server_t *new = s->s;
|
|
Packit |
90a5c9 |
if (strncmp(old->ip, new->ip, MAXIPSIZE)==0) {
|
|
Packit |
90a5c9 |
s->found = 1;
|
|
Packit |
90a5c9 |
old->busy = new->busy;
|
|
Packit |
90a5c9 |
old->ready = new->ready;
|
|
Packit |
90a5c9 |
old->seen = new->seen;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
/* Read the id corresponding to the entry in the slotmem */
|
|
Packit |
90a5c9 |
static apr_status_t hm_readid(void* mem, void *data, apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
hm_slot_server_t *old = (hm_slot_server_t *) mem;
|
|
Packit |
90a5c9 |
hm_slot_server_ctx_t *s = (hm_slot_server_ctx_t *) data;
|
|
Packit |
90a5c9 |
hm_server_t *new = s->s;
|
|
Packit |
90a5c9 |
if (strncmp(old->ip, new->ip, MAXIPSIZE)==0) {
|
|
Packit |
90a5c9 |
s->found = 1;
|
|
Packit |
90a5c9 |
s->item_id = old->id;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
/* update the entry or create it if not existing */
|
|
Packit |
90a5c9 |
static apr_status_t hm_slotmem_update_stat(hm_server_t *s, apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
/* We call do_all (to try to update) otherwise grab + put */
|
|
Packit |
90a5c9 |
hm_slot_server_ctx_t ctx;
|
|
Packit |
90a5c9 |
ctx.s = s;
|
|
Packit |
90a5c9 |
ctx.found = 0;
|
|
Packit |
90a5c9 |
storage->doall(slotmem, hm_update, &ctx, pool);
|
|
Packit |
90a5c9 |
if (!ctx.found) {
|
|
Packit |
90a5c9 |
unsigned int i;
|
|
Packit |
90a5c9 |
hm_slot_server_t hmserver;
|
|
Packit |
90a5c9 |
memcpy(hmserver.ip, s->ip, MAXIPSIZE);
|
|
Packit |
90a5c9 |
hmserver.busy = s->busy;
|
|
Packit |
90a5c9 |
hmserver.ready = s->ready;
|
|
Packit |
90a5c9 |
hmserver.seen = s->seen;
|
|
Packit |
90a5c9 |
/* XXX locking for grab() / put() */
|
|
Packit |
90a5c9 |
storage->grab(slotmem, &i);
|
|
Packit |
90a5c9 |
hmserver.id = i;
|
|
Packit |
90a5c9 |
storage->put(slotmem, i, (unsigned char *)&hmserver, sizeof(hmserver));
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
static apr_status_t hm_slotmem_remove_stat(hm_server_t *s, apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
hm_slot_server_ctx_t ctx;
|
|
Packit |
90a5c9 |
ctx.s = s;
|
|
Packit |
90a5c9 |
ctx.found = 0;
|
|
Packit |
90a5c9 |
storage->doall(slotmem, hm_readid, &ctx, pool);
|
|
Packit |
90a5c9 |
if (ctx.found) {
|
|
Packit |
90a5c9 |
storage->release(slotmem, ctx.item_id);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
static apr_status_t hm_file_update_stat(hm_ctx_t *ctx, hm_server_t *s, apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
apr_file_t *fp;
|
|
Packit |
90a5c9 |
apr_file_t *fpin;
|
|
Packit |
90a5c9 |
apr_time_t now;
|
|
Packit |
90a5c9 |
apr_time_t fage;
|
|
Packit |
90a5c9 |
apr_finfo_t fi;
|
|
Packit |
90a5c9 |
int updated = 0;
|
|
Packit |
90a5c9 |
char *path = apr_pstrcat(pool, ctx->storage_path, ".tmp.XXXXXX", NULL);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* TODO: Update stats file (!) */
|
|
Packit |
90a5c9 |
rv = apr_file_mktemp(&fp, path, APR_CREATE | APR_WRITE, pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02074)
|
|
Packit |
90a5c9 |
"Unable to open tmp file: %s", path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
rv = apr_file_open(&fpin, ctx->storage_path, APR_READ|APR_BINARY|APR_BUFFERED,
|
|
Packit |
90a5c9 |
APR_OS_DEFAULT, pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
now = apr_time_now();
|
|
Packit |
90a5c9 |
if (rv == APR_SUCCESS) {
|
|
Packit |
90a5c9 |
char *t;
|
|
Packit |
90a5c9 |
apr_table_t *hbt = apr_table_make(pool, 10);
|
|
Packit |
90a5c9 |
apr_bucket_alloc_t *ba;
|
|
Packit |
90a5c9 |
apr_bucket_brigade *bb;
|
|
Packit |
90a5c9 |
apr_bucket_brigade *tmpbb;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_info_get(&fi, APR_FINFO_SIZE | APR_FINFO_MTIME, fpin);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02075)
|
|
Packit |
90a5c9 |
"Unable to read file: %s", ctx->storage_path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Read the file and update the line corresponding to the node */
|
|
Packit |
90a5c9 |
ba = apr_bucket_alloc_create(pool);
|
|
Packit |
90a5c9 |
bb = apr_brigade_create(pool, ba);
|
|
Packit |
90a5c9 |
apr_brigade_insert_file(bb, fpin, 0, fi.size, pool);
|
|
Packit |
90a5c9 |
tmpbb = apr_brigade_create(pool, ba);
|
|
Packit |
90a5c9 |
fage = apr_time_sec(now - fi.mtime);
|
|
Packit |
90a5c9 |
do {
|
|
Packit |
90a5c9 |
char buf[4096];
|
|
Packit |
90a5c9 |
const char *ip;
|
|
Packit |
90a5c9 |
apr_size_t bsize = sizeof(buf);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_brigade_cleanup(tmpbb);
|
|
Packit |
90a5c9 |
if (APR_BRIGADE_EMPTY(bb)) {
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
rv = apr_brigade_split_line(tmpbb, bb,
|
|
Packit |
90a5c9 |
APR_BLOCK_READ, sizeof(buf));
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02076)
|
|
Packit |
90a5c9 |
"Unable to read from file: %s", ctx->storage_path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_brigade_flatten(tmpbb, buf, &bsize);
|
|
Packit |
90a5c9 |
if (bsize == 0) {
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
buf[bsize - 1] = 0;
|
|
Packit |
90a5c9 |
t = strchr(buf, ' ');
|
|
Packit |
90a5c9 |
if (t) {
|
|
Packit |
90a5c9 |
ip = apr_pstrmemdup(pool, buf, t - buf);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
ip = NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!ip || buf[0] == '#') {
|
|
Packit |
90a5c9 |
/* copy things we can't process */
|
|
Packit |
90a5c9 |
apr_file_printf(fp, "%s\n", buf);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else if (strcmp(ip, s->ip) != 0 ) {
|
|
Packit |
90a5c9 |
hm_server_t node;
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
const char *val;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Update seen time according to the last file modification */
|
|
Packit |
90a5c9 |
apr_table_clear(hbt);
|
|
Packit |
90a5c9 |
qs_to_table(apr_pstrdup(pool, t), hbt, pool);
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "busy"))) {
|
|
Packit |
90a5c9 |
node.busy = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
node.busy = 0;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "ready"))) {
|
|
Packit |
90a5c9 |
node.ready = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
node.ready = 0;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "lastseen"))) {
|
|
Packit |
90a5c9 |
node.seen = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
node.seen = SEEN_TIMEOUT;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
seen = fage + node.seen;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "port"))) {
|
|
Packit |
90a5c9 |
node.port = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
node.port = 80;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
apr_file_printf(fp, "%s &ready=%u&busy=%u&lastseen=%u&port=%u\n",
|
|
Packit |
90a5c9 |
ip, node.ready, node.busy, (unsigned int) seen, node.port);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
seen = apr_time_sec(now - s->seen);
|
|
Packit |
90a5c9 |
apr_file_printf(fp, "%s &ready=%u&busy=%u&lastseen=%u&port=%u\n",
|
|
Packit |
90a5c9 |
s->ip, s->ready, s->busy, (unsigned int) seen, s->port);
|
|
Packit |
90a5c9 |
updated = 1;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
} while (1);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!updated) {
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
seen = apr_time_sec(now - s->seen);
|
|
Packit |
90a5c9 |
apr_file_printf(fp, "%s &ready=%u&busy=%u&lastseen=%u&port=%u\n",
|
|
Packit |
90a5c9 |
s->ip, s->ready, s->busy, (unsigned int) seen, s->port);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_flush(fp);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02077)
|
|
Packit |
90a5c9 |
"Unable to flush file: %s", path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_close(fp);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02078)
|
|
Packit |
90a5c9 |
"Unable to close file: %s", path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_perms_set(path,
|
|
Packit |
90a5c9 |
APR_FPROT_UREAD | APR_FPROT_GREAD |
|
|
Packit |
90a5c9 |
APR_FPROT_WREAD);
|
|
Packit |
90a5c9 |
if (rv && rv != APR_INCOMPLETE && rv != APR_ENOTIMPL) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02079)
|
|
Packit |
90a5c9 |
"Unable to set file permissions on %s",
|
|
Packit |
90a5c9 |
path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_rename(path, ctx->storage_path, pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02080)
|
|
Packit |
90a5c9 |
"Unable to move file: %s -> %s", path,
|
|
Packit |
90a5c9 |
ctx->storage_path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
static apr_status_t hm_update_stat(hm_ctx_t *ctx, hm_server_t *s, apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
if (slotmem)
|
|
Packit |
90a5c9 |
return hm_slotmem_update_stat(s, pool);
|
|
Packit |
90a5c9 |
else
|
|
Packit |
90a5c9 |
return hm_file_update_stat(ctx, s, pool);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Store in a file */
|
|
Packit |
90a5c9 |
static apr_status_t hm_file_update_stats(hm_ctx_t *ctx, apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
apr_file_t *fp;
|
|
Packit |
90a5c9 |
apr_hash_index_t *hi;
|
|
Packit |
90a5c9 |
apr_time_t now;
|
|
Packit |
90a5c9 |
char *path = apr_pstrcat(p, ctx->storage_path, ".tmp.XXXXXX", NULL);
|
|
Packit |
90a5c9 |
/* TODO: Update stats file (!) */
|
|
Packit |
90a5c9 |
rv = apr_file_mktemp(&fp, path, APR_CREATE | APR_WRITE, p);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02081)
|
|
Packit |
90a5c9 |
"Unable to open tmp file: %s", path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
now = apr_time_now();
|
|
Packit |
90a5c9 |
for (hi = apr_hash_first(p, ctx->servers);
|
|
Packit |
90a5c9 |
hi != NULL; hi = apr_hash_next(hi)) {
|
|
Packit |
90a5c9 |
hm_server_t *s = NULL;
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
apr_hash_this(hi, NULL, NULL, (void **) &s);
|
|
Packit |
90a5c9 |
seen = apr_time_sec(now - s->seen);
|
|
Packit |
90a5c9 |
if (seen > SEEN_TIMEOUT) {
|
|
Packit |
90a5c9 |
/*
|
|
Packit |
90a5c9 |
* Skip this entry from the heartbeat file -- when it comes back,
|
|
Packit |
90a5c9 |
* we will reuse the memory...
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
apr_file_printf(fp, "%s &ready=%u&busy=%u&lastseen=%u&port=%u\n",
|
|
Packit |
90a5c9 |
s->ip, s->ready, s->busy, (unsigned int) seen, s->port);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_flush(fp);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02082)
|
|
Packit |
90a5c9 |
"Unable to flush file: %s", path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_close(fp);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02083)
|
|
Packit |
90a5c9 |
"Unable to close file: %s", path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_perms_set(path,
|
|
Packit |
90a5c9 |
APR_FPROT_UREAD | APR_FPROT_GREAD |
|
|
Packit |
90a5c9 |
APR_FPROT_WREAD);
|
|
Packit |
90a5c9 |
if (rv && rv != APR_INCOMPLETE && rv != APR_ENOTIMPL) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02084)
|
|
Packit |
90a5c9 |
"Unable to set file permissions on %s",
|
|
Packit |
90a5c9 |
path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_rename(path, ctx->storage_path, p);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02085)
|
|
Packit |
90a5c9 |
"Unable to move file: %s -> %s", path,
|
|
Packit |
90a5c9 |
ctx->storage_path);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
/* Store in a slotmem */
|
|
Packit |
90a5c9 |
static apr_status_t hm_slotmem_update_stats(hm_ctx_t *ctx, apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
apr_time_t now;
|
|
Packit |
90a5c9 |
apr_hash_index_t *hi;
|
|
Packit |
90a5c9 |
now = apr_time_now();
|
|
Packit |
90a5c9 |
for (hi = apr_hash_first(p, ctx->servers);
|
|
Packit |
90a5c9 |
hi != NULL; hi = apr_hash_next(hi)) {
|
|
Packit |
90a5c9 |
hm_server_t *s = NULL;
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
apr_hash_this(hi, NULL, NULL, (void **) &s);
|
|
Packit |
90a5c9 |
seen = apr_time_sec(now - s->seen);
|
|
Packit |
90a5c9 |
if (seen > SEEN_TIMEOUT) {
|
|
Packit |
90a5c9 |
/* remove it */
|
|
Packit |
90a5c9 |
rv = hm_slotmem_remove_stat(s, p);
|
|
Packit |
90a5c9 |
} else {
|
|
Packit |
90a5c9 |
/* update it */
|
|
Packit |
90a5c9 |
rv = hm_slotmem_update_stat(s, p);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
if (rv !=APR_SUCCESS)
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
/* Store/update the stats */
|
|
Packit |
90a5c9 |
static apr_status_t hm_update_stats(hm_ctx_t *ctx, apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
if (slotmem)
|
|
Packit |
90a5c9 |
return hm_slotmem_update_stats(ctx, p);
|
|
Packit |
90a5c9 |
else
|
|
Packit |
90a5c9 |
return hm_file_update_stats(ctx, p);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static hm_server_t *hm_get_server(hm_ctx_t *ctx, const char *ip, const int port)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
hm_server_t *s;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
s = apr_hash_get(ctx->servers, ip, APR_HASH_KEY_STRING);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (s == NULL) {
|
|
Packit |
90a5c9 |
s = apr_palloc(ctx->p, sizeof(hm_server_t));
|
|
Packit |
90a5c9 |
s->ip = apr_pstrdup(ctx->p, ip);
|
|
Packit |
90a5c9 |
s->port = port;
|
|
Packit |
90a5c9 |
s->ready = 0;
|
|
Packit |
90a5c9 |
s->busy = 0;
|
|
Packit |
90a5c9 |
s->seen = 0;
|
|
Packit |
90a5c9 |
apr_hash_set(ctx->servers, s->ip, APR_HASH_KEY_STRING, s);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return s;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Process a message received from a backend node */
|
|
Packit |
90a5c9 |
static void hm_processmsg(hm_ctx_t *ctx, apr_pool_t *p,
|
|
Packit |
90a5c9 |
apr_sockaddr_t *from, char *buf, int len)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_table_t *tbl;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
buf[len] = '\0';
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
tbl = apr_table_make(p, 10);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
qs_to_table(buf, tbl, p);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (apr_table_get(tbl, "v") != NULL &&
|
|
Packit |
90a5c9 |
apr_table_get(tbl, "busy") != NULL &&
|
|
Packit |
90a5c9 |
apr_table_get(tbl, "ready") != NULL) {
|
|
Packit |
90a5c9 |
char *ip;
|
|
Packit |
90a5c9 |
int port = 80;
|
|
Packit |
90a5c9 |
hm_server_t *s;
|
|
Packit |
90a5c9 |
/* TODO: REMOVE ME BEFORE PRODUCTION (????) */
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, ctx->s, APLOGNO(02086)
|
|
Packit |
90a5c9 |
"%pI busy=%s ready=%s", from,
|
|
Packit |
90a5c9 |
apr_table_get(tbl, "busy"), apr_table_get(tbl, "ready"));
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_sockaddr_ip_get(&ip, from);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (apr_table_get(tbl, "port") != NULL)
|
|
Packit |
90a5c9 |
port = atoi(apr_table_get(tbl, "port"));
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
s = hm_get_server(ctx, ip, port);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
s->busy = atoi(apr_table_get(tbl, "busy"));
|
|
Packit |
90a5c9 |
s->ready = atoi(apr_table_get(tbl, "ready"));
|
|
Packit |
90a5c9 |
s->seen = apr_time_now();
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, 0, ctx->s, APLOGNO(02087)
|
|
Packit |
90a5c9 |
"malformed message from %pI",
|
|
Packit |
90a5c9 |
from);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
/* Read message from multicast socket */
|
|
Packit |
90a5c9 |
#define MAX_MSG_LEN (1000)
|
|
Packit |
90a5c9 |
static apr_status_t hm_recv(hm_ctx_t *ctx, apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
char buf[MAX_MSG_LEN + 1];
|
|
Packit |
90a5c9 |
apr_sockaddr_t from;
|
|
Packit |
90a5c9 |
apr_size_t len = MAX_MSG_LEN;
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
from.pool = p;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_socket_recvfrom(&from, ctx->sock, 0, buf, &len;;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (APR_STATUS_IS_EAGAIN(rv)) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02088) "would block");
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02089) "recvfrom failed");
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
hm_processmsg(ctx, p, &from, buf, len);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static apr_status_t hm_watchdog_callback(int state, void *data,
|
|
Packit |
90a5c9 |
apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv = APR_SUCCESS;
|
|
Packit |
90a5c9 |
apr_time_t cur, now;
|
|
Packit |
90a5c9 |
hm_ctx_t *ctx = (hm_ctx_t *)data;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!ctx->active) {
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
switch (state) {
|
|
Packit |
90a5c9 |
case AP_WATCHDOG_STATE_STARTING:
|
|
Packit |
90a5c9 |
rv = hm_listen(ctx);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ctx->status = rv;
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, ctx->s, APLOGNO(02090)
|
|
Packit |
90a5c9 |
"Unable to listen for connections!");
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
ctx->keep_running = 1;
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, ctx->s, APLOGNO(02091)
|
|
Packit |
90a5c9 |
"%s listener started.",
|
|
Packit |
90a5c9 |
HM_WATHCHDOG_NAME);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
case AP_WATCHDOG_STATE_RUNNING:
|
|
Packit |
90a5c9 |
/* store in the slotmem or in the file depending on configuration */
|
|
Packit |
90a5c9 |
hm_update_stats(ctx, pool);
|
|
Packit |
90a5c9 |
cur = now = apr_time_sec(apr_time_now());
|
|
Packit |
90a5c9 |
/* TODO: Insted HN_UPDATE_SEC use
|
|
Packit |
90a5c9 |
* the ctx->interval
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
while ((now - cur) < apr_time_sec(ctx->interval)) {
|
|
Packit |
90a5c9 |
int n;
|
|
Packit |
90a5c9 |
apr_status_t rc;
|
|
Packit |
90a5c9 |
apr_pool_t *p;
|
|
Packit |
90a5c9 |
apr_pollfd_t pfd;
|
|
Packit |
90a5c9 |
apr_interval_time_t timeout;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_pool_create(&p, pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
pfd.desc_type = APR_POLL_SOCKET;
|
|
Packit |
90a5c9 |
pfd.desc.s = ctx->sock;
|
|
Packit |
90a5c9 |
pfd.p = p;
|
|
Packit |
90a5c9 |
pfd.reqevents = APR_POLLIN;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
timeout = apr_time_from_sec(1);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rc = apr_poll(&pfd, 1, &n, timeout);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!ctx->keep_running) {
|
|
Packit |
90a5c9 |
apr_pool_destroy(p);
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
if (rc == APR_SUCCESS && (pfd.rtnevents & APR_POLLIN)) {
|
|
Packit |
90a5c9 |
hm_recv(ctx, p);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
now = apr_time_sec(apr_time_now());
|
|
Packit |
90a5c9 |
apr_pool_destroy(p);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
case AP_WATCHDOG_STATE_STOPPING:
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, ctx->s, APLOGNO(02092)
|
|
Packit |
90a5c9 |
"stopping %s listener.",
|
|
Packit |
90a5c9 |
HM_WATHCHDOG_NAME);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ctx->keep_running = 0;
|
|
Packit |
90a5c9 |
if (ctx->sock) {
|
|
Packit |
90a5c9 |
apr_socket_close(ctx->sock);
|
|
Packit |
90a5c9 |
ctx->sock = NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static int hm_post_config(apr_pool_t *p, apr_pool_t *plog,
|
|
Packit |
90a5c9 |
apr_pool_t *ptemp, server_rec *s)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
hm_ctx_t *ctx = ap_get_module_config(s->module_config,
|
|
Packit |
90a5c9 |
&heartmonitor_module);
|
|
Packit |
90a5c9 |
APR_OPTIONAL_FN_TYPE(ap_watchdog_get_instance) *hm_watchdog_get_instance;
|
|
Packit |
90a5c9 |
APR_OPTIONAL_FN_TYPE(ap_watchdog_register_callback) *hm_watchdog_register_callback;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
hm_watchdog_get_instance = APR_RETRIEVE_OPTIONAL_FN(ap_watchdog_get_instance);
|
|
Packit |
90a5c9 |
hm_watchdog_register_callback = APR_RETRIEVE_OPTIONAL_FN(ap_watchdog_register_callback);
|
|
Packit |
90a5c9 |
if (!hm_watchdog_get_instance || !hm_watchdog_register_callback) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, 0, s, APLOGNO(02093)
|
|
Packit |
90a5c9 |
"mod_watchdog is required");
|
|
Packit |
90a5c9 |
return !OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Create the slotmem */
|
|
Packit |
90a5c9 |
if (ap_state_query(AP_SQ_MAIN_STATE) == AP_SQ_MS_CREATE_CONFIG) {
|
|
Packit |
90a5c9 |
/* this is the real thing */
|
|
Packit |
90a5c9 |
if (maxworkers) {
|
|
Packit |
90a5c9 |
storage = ap_lookup_provider(AP_SLOTMEM_PROVIDER_GROUP, "shm",
|
|
Packit |
90a5c9 |
AP_SLOTMEM_PROVIDER_VERSION);
|
|
Packit |
90a5c9 |
if (!storage) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_EMERG, 0, s, APLOGNO(02284)
|
|
Packit |
90a5c9 |
"failed to lookup provider 'shm' for '%s', "
|
|
Packit |
90a5c9 |
"maybe you need to load mod_slotmem_shm?",
|
|
Packit |
90a5c9 |
AP_SLOTMEM_PROVIDER_GROUP);
|
|
Packit |
90a5c9 |
return !OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
storage->create(&slotmem, "mod_heartmonitor", sizeof(hm_slot_server_t), maxworkers, AP_SLOTMEM_TYPE_PREGRAB, p);
|
|
Packit |
90a5c9 |
if (!slotmem) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_EMERG, 0, s, APLOGNO(02285)
|
|
Packit |
90a5c9 |
"slotmem_create for status failed");
|
|
Packit |
90a5c9 |
return !OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!ctx->active) {
|
|
Packit |
90a5c9 |
return OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
rv = hm_watchdog_get_instance(&ctx->watchdog,
|
|
Packit |
90a5c9 |
HM_WATHCHDOG_NAME,
|
|
Packit |
90a5c9 |
0, 1, p);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(02094)
|
|
Packit |
90a5c9 |
"Failed to create watchdog instance (%s)",
|
|
Packit |
90a5c9 |
HM_WATHCHDOG_NAME);
|
|
Packit |
90a5c9 |
return !OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
/* Register a callback with zero interval. */
|
|
Packit |
90a5c9 |
rv = hm_watchdog_register_callback(ctx->watchdog,
|
|
Packit |
90a5c9 |
0,
|
|
Packit |
90a5c9 |
ctx,
|
|
Packit |
90a5c9 |
hm_watchdog_callback);
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_CRIT, rv, s, APLOGNO(02095)
|
|
Packit |
90a5c9 |
"Failed to register watchdog callback (%s)",
|
|
Packit |
90a5c9 |
HM_WATHCHDOG_NAME);
|
|
Packit |
90a5c9 |
return !OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_DEBUG, 0, s, APLOGNO(02096)
|
|
Packit |
90a5c9 |
"wd callback %s", HM_WATHCHDOG_NAME);
|
|
Packit |
90a5c9 |
return OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static int hm_handler(request_rec *r)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_bucket_brigade *input_brigade;
|
|
Packit |
90a5c9 |
apr_size_t len;
|
|
Packit |
90a5c9 |
char *buf;
|
|
Packit |
90a5c9 |
apr_status_t status;
|
|
Packit |
90a5c9 |
apr_table_t *tbl;
|
|
Packit |
90a5c9 |
hm_server_t hmserver;
|
|
Packit |
90a5c9 |
char *ip;
|
|
Packit |
90a5c9 |
hm_ctx_t *ctx;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (strcmp(r->handler, "heartbeat")) {
|
|
Packit |
90a5c9 |
return DECLINED;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
if (r->method_number != M_POST) {
|
|
Packit |
90a5c9 |
return HTTP_METHOD_NOT_ALLOWED;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
len = MAX_MSG_LEN;
|
|
Packit |
90a5c9 |
ctx = ap_get_module_config(r->server->module_config,
|
|
Packit |
90a5c9 |
&heartmonitor_module);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
buf = apr_pcalloc(r->pool, MAX_MSG_LEN);
|
|
Packit |
90a5c9 |
input_brigade = apr_brigade_create(r->connection->pool, r->connection->bucket_alloc);
|
|
Packit |
90a5c9 |
status = ap_get_brigade(r->input_filters, input_brigade, AP_MODE_READBYTES, APR_BLOCK_READ, MAX_MSG_LEN);
|
|
Packit |
90a5c9 |
if (status != APR_SUCCESS) {
|
|
Packit |
90a5c9 |
return ap_map_http_request_error(status, HTTP_BAD_REQUEST);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
apr_brigade_flatten(input_brigade, buf, &len;;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* we can't use hm_processmsg because it uses hm_get_server() */
|
|
Packit |
90a5c9 |
buf[len] = '\0';
|
|
Packit |
90a5c9 |
tbl = apr_table_make(r->pool, 10);
|
|
Packit |
90a5c9 |
qs_to_table(buf, tbl, r->pool);
|
|
Packit |
90a5c9 |
apr_sockaddr_ip_get(&ip, r->connection->client_addr);
|
|
Packit |
90a5c9 |
hmserver.ip = ip;
|
|
Packit |
90a5c9 |
hmserver.port = 80;
|
|
Packit |
90a5c9 |
if (apr_table_get(tbl, "port") != NULL)
|
|
Packit |
90a5c9 |
hmserver.port = atoi(apr_table_get(tbl, "port"));
|
|
Packit |
90a5c9 |
hmserver.busy = atoi(apr_table_get(tbl, "busy"));
|
|
Packit |
90a5c9 |
hmserver.ready = atoi(apr_table_get(tbl, "ready"));
|
|
Packit |
90a5c9 |
hmserver.seen = apr_time_now();
|
|
Packit |
90a5c9 |
hm_update_stat(ctx, &hmserver, r->pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ap_set_content_type(r, "text/plain");
|
|
Packit |
90a5c9 |
ap_set_content_length(r, 2);
|
|
Packit |
90a5c9 |
ap_rputs("OK", r);
|
|
Packit |
90a5c9 |
ap_rflush(r);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static void hm_register_hooks(apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
static const char * const aszSucc[]={ "mod_proxy.c", NULL };
|
|
Packit |
90a5c9 |
ap_hook_post_config(hm_post_config, NULL, NULL, APR_HOOK_MIDDLE);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ap_hook_handler(hm_handler, NULL, aszSucc, APR_HOOK_FIRST);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static void *hm_create_config(apr_pool_t *p, server_rec *s)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
hm_ctx_t *ctx = (hm_ctx_t *) apr_palloc(p, sizeof(hm_ctx_t));
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ctx->active = 0;
|
|
Packit |
90a5c9 |
ctx->storage_path = ap_runtime_dir_relative(p, DEFAULT_HEARTBEAT_STORAGE);
|
|
Packit |
90a5c9 |
/* TODO: Add directive for tuning the update interval
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
ctx->interval = apr_time_from_sec(HM_UPDATE_SEC);
|
|
Packit |
90a5c9 |
ctx->s = s;
|
|
Packit |
90a5c9 |
apr_pool_create(&ctx->p, p);
|
|
Packit |
90a5c9 |
ctx->servers = apr_hash_make(ctx->p);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return ctx;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const char *cmd_hm_storage(cmd_parms *cmd,
|
|
Packit |
90a5c9 |
void *dconf, const char *path)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_pool_t *p = cmd->pool;
|
|
Packit |
90a5c9 |
hm_ctx_t *ctx =
|
|
Packit |
90a5c9 |
(hm_ctx_t *) ap_get_module_config(cmd->server->module_config,
|
|
Packit |
90a5c9 |
&heartmonitor_module);
|
|
Packit |
90a5c9 |
const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (err != NULL) {
|
|
Packit |
90a5c9 |
return err;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ctx->storage_path = ap_runtime_dir_relative(p, path);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const char *cmd_hm_listen(cmd_parms *cmd,
|
|
Packit |
90a5c9 |
void *dconf, const char *mcast_addr)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
char *host_str;
|
|
Packit |
90a5c9 |
char *scope_id;
|
|
Packit |
90a5c9 |
apr_port_t port = 0;
|
|
Packit |
90a5c9 |
apr_pool_t *p = cmd->pool;
|
|
Packit |
90a5c9 |
hm_ctx_t *ctx =
|
|
Packit |
90a5c9 |
(hm_ctx_t *) ap_get_module_config(cmd->server->module_config,
|
|
Packit |
90a5c9 |
&heartmonitor_module);
|
|
Packit |
90a5c9 |
const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (err != NULL) {
|
|
Packit |
90a5c9 |
return err;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!ctx->active) {
|
|
Packit |
90a5c9 |
ctx->active = 1;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
return "HeartbeatListen: May only be specified once.";
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_parse_addr_port(&host_str, &scope_id, &port, mcast_addr, cmd->temp_pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
return "HeartbeatListen: Unable to parse multicast address.";
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (host_str == NULL) {
|
|
Packit |
90a5c9 |
return "HeartbeatListen: No host provided in multicast address";
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (port == 0) {
|
|
Packit |
90a5c9 |
return "HeartbeatListen: No port provided in multicast address";
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_sockaddr_info_get(&ctx->mcast_addr, host_str, APR_INET, port, 0,
|
|
Packit |
90a5c9 |
p);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
return
|
|
Packit |
90a5c9 |
"HeartbeatListen: apr_sockaddr_info_get failed on multicast address";
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const char *cmd_hm_maxworkers(cmd_parms *cmd,
|
|
Packit |
90a5c9 |
void *dconf, const char *data)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
const char *err = ap_check_cmd_context(cmd, GLOBAL_ONLY);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (err != NULL) {
|
|
Packit |
90a5c9 |
return err;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
maxworkers = atoi(data);
|
|
Packit |
90a5c9 |
if (maxworkers <= 10)
|
|
Packit |
90a5c9 |
return "HeartbeatMaxServers: Should be bigger than 10";
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const command_rec hm_cmds[] = {
|
|
Packit |
90a5c9 |
AP_INIT_TAKE1("HeartbeatListen", cmd_hm_listen, NULL, RSRC_CONF,
|
|
Packit |
90a5c9 |
"Address to listen for heartbeat requests"),
|
|
Packit |
90a5c9 |
AP_INIT_TAKE1("HeartbeatStorage", cmd_hm_storage, NULL, RSRC_CONF,
|
|
Packit |
90a5c9 |
"Path to store heartbeat data."),
|
|
Packit |
90a5c9 |
AP_INIT_TAKE1("HeartbeatMaxServers", cmd_hm_maxworkers, NULL, RSRC_CONF,
|
|
Packit |
90a5c9 |
"Max number of servers when using slotmem (instead file) to store heartbeat data."),
|
|
Packit |
90a5c9 |
{NULL}
|
|
Packit |
90a5c9 |
};
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
AP_DECLARE_MODULE(heartmonitor) = {
|
|
Packit |
90a5c9 |
STANDARD20_MODULE_STUFF,
|
|
Packit |
90a5c9 |
NULL, /* create per-directory config structure */
|
|
Packit |
90a5c9 |
NULL, /* merge per-directory config structures */
|
|
Packit |
90a5c9 |
hm_create_config, /* create per-server config structure */
|
|
Packit |
90a5c9 |
NULL, /* merge per-server config structures */
|
|
Packit |
90a5c9 |
hm_cmds, /* command apr_table_t */
|
|
Packit |
90a5c9 |
hm_register_hooks
|
|
Packit |
90a5c9 |
};
|