|
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 "mod_proxy.h"
|
|
Packit |
90a5c9 |
#include "scoreboard.h"
|
|
Packit |
90a5c9 |
#include "ap_mpm.h"
|
|
Packit |
90a5c9 |
#include "apr_version.h"
|
|
Packit |
90a5c9 |
#include "ap_hooks.h"
|
|
Packit |
90a5c9 |
#include "ap_slotmem.h"
|
|
Packit |
90a5c9 |
#include "heartbeat.h"
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
#ifndef LBM_HEARTBEAT_MAX_LASTSEEN
|
|
Packit |
90a5c9 |
/* If we haven't seen a heartbeat in the last N seconds, don't count this IP
|
|
Packit |
90a5c9 |
* as allive.
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
#define LBM_HEARTBEAT_MAX_LASTSEEN (10)
|
|
Packit |
90a5c9 |
#endif
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
module AP_MODULE_DECLARE_DATA lbmethod_heartbeat_module;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static int (*ap_proxy_retry_worker_fn)(const char *proxy_function,
|
|
Packit |
90a5c9 |
proxy_worker *worker, server_rec *s) = NULL;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const ap_slotmem_provider_t *storage = NULL;
|
|
Packit |
90a5c9 |
static ap_slotmem_instance_t *hm_serversmem = NULL;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/*
|
|
Packit |
90a5c9 |
* configuration structure
|
|
Packit |
90a5c9 |
* path: path of the file where the heartbeat information is stored.
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
typedef struct lb_hb_ctx_t
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
const char *path;
|
|
Packit |
90a5c9 |
} lb_hb_ctx_t;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
typedef struct hb_server_t {
|
|
Packit |
90a5c9 |
const char *ip;
|
|
Packit |
90a5c9 |
int busy;
|
|
Packit |
90a5c9 |
int ready;
|
|
Packit |
90a5c9 |
int port;
|
|
Packit |
90a5c9 |
int id;
|
|
Packit |
90a5c9 |
apr_time_t seen;
|
|
Packit |
90a5c9 |
proxy_worker *worker;
|
|
Packit |
90a5c9 |
} hb_server_t;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
typedef struct ctx_servers {
|
|
Packit |
90a5c9 |
apr_time_t now;
|
|
Packit |
90a5c9 |
apr_hash_t *servers;
|
|
Packit |
90a5c9 |
} ctx_servers_t;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static void
|
|
Packit |
90a5c9 |
argstr_to_table(apr_pool_t *p, char *str, apr_table_t *parms)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
char *key;
|
|
Packit |
90a5c9 |
char *value;
|
|
Packit |
90a5c9 |
char *strtok_state;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
key = apr_strtok(str, "&", &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(03230)
|
|
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 |
static apr_status_t readfile_heartbeats(const char *path, apr_hash_t *servers,
|
|
Packit |
90a5c9 |
apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_finfo_t fi;
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
apr_file_t *fp;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!path) {
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_open(&fp, path, APR_READ|APR_BINARY|APR_BUFFERED,
|
|
Packit |
90a5c9 |
APR_OS_DEFAULT, pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_file_info_get(&fi, APR_FINFO_SIZE, fp);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
char *t;
|
|
Packit |
90a5c9 |
int lineno = 0;
|
|
Packit |
90a5c9 |
apr_bucket_alloc_t *ba = apr_bucket_alloc_create(pool);
|
|
Packit |
90a5c9 |
apr_bucket_brigade *bb = apr_brigade_create(pool, ba);
|
|
Packit |
90a5c9 |
apr_bucket_brigade *tmpbb = apr_brigade_create(pool, ba);
|
|
Packit |
90a5c9 |
apr_table_t *hbt = apr_table_make(pool, 10);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_brigade_insert_file(bb, fp, 0, fi.size, pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
do {
|
|
Packit |
90a5c9 |
hb_server_t *server;
|
|
Packit |
90a5c9 |
char buf[4096];
|
|
Packit |
90a5c9 |
apr_size_t bsize = sizeof(buf);
|
|
Packit |
90a5c9 |
const char *ip, *val;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_brigade_cleanup(tmpbb);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (APR_BRIGADE_EMPTY(bb)) {
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = apr_brigade_split_line(tmpbb, bb,
|
|
Packit |
90a5c9 |
APR_BLOCK_READ, sizeof(buf));
|
|
Packit |
90a5c9 |
lineno++;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_brigade_flatten(tmpbb, buf, &bsize);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (bsize == 0) {
|
|
Packit |
90a5c9 |
break;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
buf[bsize - 1] = 0;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* comment */
|
|
Packit |
90a5c9 |
if (buf[0] == '#') {
|
|
Packit |
90a5c9 |
continue;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* line format: <IP> <query_string>\n */
|
|
Packit |
90a5c9 |
t = strchr(buf, ' ');
|
|
Packit |
90a5c9 |
if (!t) {
|
|
Packit |
90a5c9 |
continue;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ip = apr_pstrmemdup(pool, buf, t - buf);
|
|
Packit |
90a5c9 |
t++;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
server = apr_hash_get(servers, ip, APR_HASH_KEY_STRING);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (server == NULL) {
|
|
Packit |
90a5c9 |
server = apr_pcalloc(pool, sizeof(hb_server_t));
|
|
Packit |
90a5c9 |
server->ip = ip;
|
|
Packit |
90a5c9 |
server->port = 80;
|
|
Packit |
90a5c9 |
server->seen = -1;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_hash_set(servers, server->ip, APR_HASH_KEY_STRING, server);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_table_clear(hbt);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
argstr_to_table(pool, apr_pstrdup(pool, t), hbt);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "busy"))) {
|
|
Packit |
90a5c9 |
server->busy = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "ready"))) {
|
|
Packit |
90a5c9 |
server->ready = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "lastseen"))) {
|
|
Packit |
90a5c9 |
server->seen = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if ((val = apr_table_get(hbt, "port"))) {
|
|
Packit |
90a5c9 |
server->port = atoi(val);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (server->busy == 0 && server->ready != 0) {
|
|
Packit |
90a5c9 |
/* Server has zero threads active, but lots of them ready,
|
|
Packit |
90a5c9 |
* it likely just started up, so lets /4 the number ready,
|
|
Packit |
90a5c9 |
* to prevent us from completely flooding it with all new
|
|
Packit |
90a5c9 |
* requests.
|
|
Packit |
90a5c9 |
*/
|
|
Packit |
90a5c9 |
server->ready = server->ready / 4;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
} while (1);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static apr_status_t hm_read(void* mem, void *data, apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
hm_slot_server_t *slotserver = (hm_slot_server_t *) mem;
|
|
Packit |
90a5c9 |
ctx_servers_t *ctx = (ctx_servers_t *) data;
|
|
Packit |
90a5c9 |
apr_hash_t *servers = (apr_hash_t *) ctx->servers;
|
|
Packit |
90a5c9 |
hb_server_t *server = apr_hash_get(servers, slotserver->ip, APR_HASH_KEY_STRING);
|
|
Packit |
90a5c9 |
if (server == NULL) {
|
|
Packit |
90a5c9 |
server = apr_pcalloc(pool, sizeof(hb_server_t));
|
|
Packit |
90a5c9 |
server->ip = apr_pstrdup(pool, slotserver->ip);
|
|
Packit |
90a5c9 |
server->seen = -1;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_hash_set(servers, server->ip, APR_HASH_KEY_STRING, server);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
server->busy = slotserver->busy;
|
|
Packit |
90a5c9 |
server->ready = slotserver->ready;
|
|
Packit |
90a5c9 |
server->seen = apr_time_sec(ctx->now - slotserver->seen);
|
|
Packit |
90a5c9 |
server->id = slotserver->id;
|
|
Packit |
90a5c9 |
if (server->busy == 0 && server->ready != 0) {
|
|
Packit |
90a5c9 |
server->ready = server->ready / 4;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
static apr_status_t readslot_heartbeats(ctx_servers_t *ctx,
|
|
Packit |
90a5c9 |
apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
storage->doall(hm_serversmem, hm_read, ctx, pool);
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static apr_status_t read_heartbeats(const char *path, apr_hash_t *servers,
|
|
Packit |
90a5c9 |
apr_pool_t *pool)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
if (hm_serversmem) {
|
|
Packit |
90a5c9 |
ctx_servers_t ctx;
|
|
Packit |
90a5c9 |
ctx.now = apr_time_now();
|
|
Packit |
90a5c9 |
ctx.servers = servers;
|
|
Packit |
90a5c9 |
rv = readslot_heartbeats(&ctx, pool);
|
|
Packit |
90a5c9 |
} else
|
|
Packit |
90a5c9 |
rv = readfile_heartbeats(path, servers, pool);
|
|
Packit |
90a5c9 |
return rv;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static proxy_worker *find_best_hb(proxy_balancer *balancer,
|
|
Packit |
90a5c9 |
request_rec *r)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_status_t rv;
|
|
Packit |
90a5c9 |
int i;
|
|
Packit |
90a5c9 |
apr_uint32_t openslots = 0;
|
|
Packit |
90a5c9 |
proxy_worker **worker;
|
|
Packit |
90a5c9 |
hb_server_t *server;
|
|
Packit |
90a5c9 |
apr_array_header_t *up_servers;
|
|
Packit |
90a5c9 |
proxy_worker *mycandidate = NULL;
|
|
Packit |
90a5c9 |
apr_pool_t *tpool;
|
|
Packit |
90a5c9 |
apr_hash_t *servers;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *ctx =
|
|
Packit |
90a5c9 |
ap_get_module_config(r->server->module_config,
|
|
Packit |
90a5c9 |
&lbmethod_heartbeat_module);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ap_proxy_retry_worker_fn =
|
|
Packit |
90a5c9 |
APR_RETRIEVE_OPTIONAL_FN(ap_proxy_retry_worker);
|
|
Packit |
90a5c9 |
if (!ap_proxy_retry_worker_fn) {
|
|
Packit |
90a5c9 |
/* can only happen if mod_proxy isn't loaded */
|
|
Packit |
90a5c9 |
return NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_pool_create(&tpool, r->pool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
servers = apr_hash_make(tpool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
rv = read_heartbeats(ctx->path, servers, tpool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (rv) {
|
|
Packit |
90a5c9 |
ap_log_rerror(APLOG_MARK, APLOG_ERR, rv, r, APLOGNO(01213)
|
|
Packit |
90a5c9 |
"lb_heartbeat: Unable to read heartbeats at '%s'",
|
|
Packit |
90a5c9 |
ctx->path);
|
|
Packit |
90a5c9 |
apr_pool_destroy(tpool);
|
|
Packit |
90a5c9 |
return NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
up_servers = apr_array_make(tpool, apr_hash_count(servers), sizeof(hb_server_t *));
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
for (i = 0; i < balancer->workers->nelts; i++) {
|
|
Packit |
90a5c9 |
worker = &APR_ARRAY_IDX(balancer->workers, i, proxy_worker *);
|
|
Packit |
90a5c9 |
server = apr_hash_get(servers, (*worker)->s->hostname_ex, APR_HASH_KEY_STRING);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!server) {
|
|
Packit |
90a5c9 |
ap_log_rerror(APLOG_MARK, APLOG_DEBUG, rv, r, APLOGNO(01214)
|
|
Packit |
90a5c9 |
"lb_heartbeat: No server for worker %s", (*worker)->s->name);
|
|
Packit |
90a5c9 |
continue;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (!PROXY_WORKER_IS_USABLE(*worker)) {
|
|
Packit |
90a5c9 |
ap_proxy_retry_worker_fn("BALANCER", *worker, r->server);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (PROXY_WORKER_IS_USABLE(*worker)) {
|
|
Packit |
90a5c9 |
server->worker = *worker;
|
|
Packit |
90a5c9 |
if (server->seen < LBM_HEARTBEAT_MAX_LASTSEEN) {
|
|
Packit |
90a5c9 |
openslots += server->ready;
|
|
Packit |
90a5c9 |
APR_ARRAY_PUSH(up_servers, hb_server_t *) = server;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (openslots > 0) {
|
|
Packit |
90a5c9 |
apr_uint32_t c = 0;
|
|
Packit |
90a5c9 |
apr_uint32_t pick = 0;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
pick = ap_random_pick(0, openslots);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
for (i = 0; i < up_servers->nelts; i++) {
|
|
Packit |
90a5c9 |
server = APR_ARRAY_IDX(up_servers, i, hb_server_t *);
|
|
Packit |
90a5c9 |
if (pick >= c && pick <= c + server->ready) {
|
|
Packit |
90a5c9 |
mycandidate = server->worker;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
c += server->ready;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
apr_pool_destroy(tpool);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return mycandidate;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static apr_status_t reset(proxy_balancer *balancer, server_rec *s)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static apr_status_t age(proxy_balancer *balancer, server_rec *s)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
return APR_SUCCESS;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const proxy_balancer_method heartbeat =
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
"heartbeat",
|
|
Packit |
90a5c9 |
&find_best_hb,
|
|
Packit |
90a5c9 |
NULL,
|
|
Packit |
90a5c9 |
&reset,
|
|
Packit |
90a5c9 |
&age,
|
|
Packit |
90a5c9 |
NULL
|
|
Packit |
90a5c9 |
};
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static int lb_hb_init(apr_pool_t *p, apr_pool_t *plog,
|
|
Packit |
90a5c9 |
apr_pool_t *ptemp, server_rec *s)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_size_t size;
|
|
Packit |
90a5c9 |
unsigned int num;
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *ctx = ap_get_module_config(s->module_config,
|
|
Packit |
90a5c9 |
&lbmethod_heartbeat_module);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* do nothing on first call */
|
|
Packit |
90a5c9 |
if (ap_state_query(AP_SQ_MAIN_STATE) == AP_SQ_MS_CREATE_PRE_CONFIG)
|
|
Packit |
90a5c9 |
return OK;
|
|
Packit |
90a5c9 |
|
|
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_NOTICE, 0, s, APLOGNO(02281)
|
|
Packit |
90a5c9 |
"Failed to lookup provider 'shm' for '%s'. Maybe you "
|
|
Packit |
90a5c9 |
"need to load mod_slotmem_shm?",
|
|
Packit |
90a5c9 |
AP_SLOTMEM_PROVIDER_GROUP);
|
|
Packit |
90a5c9 |
return OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
/* Try to use a slotmem created by mod_heartmonitor */
|
|
Packit |
90a5c9 |
storage->attach(&hm_serversmem, "mod_heartmonitor", &size, &num, p);
|
|
Packit |
90a5c9 |
if (!hm_serversmem)
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, s, APLOGNO(02282)
|
|
Packit |
90a5c9 |
"No slotmem from mod_heartmonitor");
|
|
Packit |
90a5c9 |
else
|
|
Packit |
90a5c9 |
ap_log_error(APLOG_MARK, APLOG_NOTICE, 0, s, APLOGNO(02283)
|
|
Packit |
90a5c9 |
"Using slotmem from mod_heartmonitor");
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (hm_serversmem)
|
|
Packit |
90a5c9 |
ctx->path = "(slotmem)";
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return OK;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static void register_hooks(apr_pool_t *p)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
static const char * const aszPred[]={ "mod_heartmonitor.c", NULL };
|
|
Packit |
90a5c9 |
ap_register_provider(p, PROXY_LBMETHOD, "heartbeat", "0", &heartbeat);
|
|
Packit |
90a5c9 |
ap_hook_post_config(lb_hb_init, aszPred, NULL, APR_HOOK_MIDDLE);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static void *lb_hb_create_config(apr_pool_t *p, server_rec *s)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *ctx = (lb_hb_ctx_t *) apr_palloc(p, sizeof(lb_hb_ctx_t));
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
ctx->path = ap_runtime_dir_relative(p, DEFAULT_HEARTBEAT_STORAGE);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return ctx;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static void *lb_hb_merge_config(apr_pool_t *p, void *basev, void *overridesv)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *ps = apr_pcalloc(p, sizeof(lb_hb_ctx_t));
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *base = (lb_hb_ctx_t *) basev;
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *overrides = (lb_hb_ctx_t *) overridesv;
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
if (overrides->path) {
|
|
Packit |
90a5c9 |
ps->path = apr_pstrdup(p, overrides->path);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
else {
|
|
Packit |
90a5c9 |
ps->path = apr_pstrdup(p, base->path);
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return ps;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const char *cmd_lb_hb_storage(cmd_parms *cmd,
|
|
Packit |
90a5c9 |
void *dconf, const char *path)
|
|
Packit |
90a5c9 |
{
|
|
Packit |
90a5c9 |
apr_pool_t *p = cmd->pool;
|
|
Packit |
90a5c9 |
lb_hb_ctx_t *ctx =
|
|
Packit |
90a5c9 |
(lb_hb_ctx_t *) ap_get_module_config(cmd->server->module_config,
|
|
Packit |
90a5c9 |
&lbmethod_heartbeat_module);
|
|
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 |
ctx->path = ap_runtime_dir_relative(p, path);
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
return NULL;
|
|
Packit |
90a5c9 |
}
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
static const command_rec cmds[] = {
|
|
Packit |
90a5c9 |
AP_INIT_TAKE1("HeartbeatStorage", cmd_lb_hb_storage, NULL, RSRC_CONF,
|
|
Packit |
90a5c9 |
"Path to read heartbeat data."),
|
|
Packit |
90a5c9 |
{NULL}
|
|
Packit |
90a5c9 |
};
|
|
Packit |
90a5c9 |
|
|
Packit |
90a5c9 |
AP_DECLARE_MODULE(lbmethod_heartbeat) = {
|
|
Packit |
90a5c9 |
STANDARD20_MODULE_STUFF,
|
|
Packit |
90a5c9 |
NULL, /* create per-directory config structure */
|
|
Packit |
90a5c9 |
NULL, /* merge per-directory config structures */
|
|
Packit |
90a5c9 |
lb_hb_create_config, /* create per-server config structure */
|
|
Packit |
90a5c9 |
lb_hb_merge_config, /* merge per-server config structures */
|
|
Packit |
90a5c9 |
cmds, /* command apr_table_t */
|
|
Packit |
90a5c9 |
register_hooks /* register hooks */
|
|
Packit |
90a5c9 |
};
|