mturk 2004/08/09 01:06:40 Added: ajp/proxy proxy_balancer.c Log: Add proxy_balancer initial code. Revision Changes Path 1.1 jakarta-tomcat-connectors/ajp/proxy/proxy_balancer.c Index: proxy_balancer.c =================================================================== /* Copyright 1999-2004 The Apache Software Foundation * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ /* Load balancer module for Apache proxy */ #define CORE_PRIVATE #include "mod_proxy.h" #include "ap_mpm.h" #include "apr_version.h" module AP_MODULE_DECLARE_DATA proxy_balancer_module; #if APR_HAS_THREADS1 #define PROXY_BALANCER_LOCK(b) apr_thread_mutex_lock((b)->mutex) #define PROXY_BALANCER_UNLOCK(b) apr_thread_mutex_unlock((b)->mutex) #else #define PROXY_BALANCER_LOCK(b) APR_SUCCESS #define PROXY_BALANCER_UNLOCK(b) APR_SUCCESS #endif /* Retrieve the parameter with the given name */ static char *get_path_param(apr_pool_t *pool, char *url, const char *name) { char *path = NULL; for (path = strstr(url, name); path; path = strstr(path + 1, name)) { path += (strlen(name) + 1); if (*path == '=') { /* * Session path was found, get it's value */ ++path; if (strlen(path)) { char *q; path = apr_pstrdup(pool, path); if ((q = strchr(path, '?'))) *q = '\0'; return path; } } } return NULL; } static char *get_cookie_param(request_rec *r, const char *name) { const char *cookies; const char *start_cookie; if ((cookies = apr_table_get(r->headers_in, "Cookie"))) { for (start_cookie = strstr(cookies, name); start_cookie; start_cookie = strstr(start_cookie + 1, name)) { if (start_cookie == cookies || start_cookie[-1] == ';' || start_cookie[-1] == ',' || isspace(start_cookie[-1])) { start_cookie += strlen(name); while(*start_cookie && isspace(*start_cookie)) ++start_cookie; if (*start_cookie == '=' && start_cookie[1]) { /* * Session cookie was found, get it's value */ char *end_cookie, *cookie; ++start_cookie; cookie = apr_pstrdup(r->pool, start_cookie); if ((end_cookie = strchr(cookie, ';')) != NULL) *end_cookie = '\0'; if((end_cookie = strchr(cookie, ',')) != NULL) *end_cookie = '\0'; return cookie; } } } } return NULL; } static proxy_runtime_worker *find_route_worker(proxy_balancer *balancer, const char *route) { int i; proxy_runtime_worker *worker = (proxy_runtime_worker *)balancer->workers->elts; for (i = 0; i < balancer->workers->nelts; i++) { if (worker->route && strcmp(worker->route, route) == 0) { return worker; } worker++; } return NULL; } static proxy_runtime_worker *find_session_route(proxy_balancer *balancer, request_rec *r, char **route, char **url) { if (!balancer->sticky) return NULL; /* Try to find the sticky route inside url */ *route = get_path_param(r->pool, *url, balancer->sticky); if (!*route) *route = get_cookie_param(r, balancer->sticky); if (*route) { proxy_runtime_worker *worker = find_route_worker(balancer, *route); /* TODO: make worker status codes */ /* See if we have a redirection route */ if (worker && worker->w->status < 2 && worker->redirect) worker = find_route_worker(balancer, worker->redirect); else worker = NULL; return worker; } else return NULL; } static int proxy_balancer_pre_request(proxy_worker **worker, proxy_balancer **balancer, request_rec *r, proxy_server_conf *conf, char **url) { int access_status = OK; proxy_runtime_worker *runtime; char *route; apr_status_t rv; /* Spet 1: check if the url is for us */ if (!(*balancer = ap_proxy_get_balancer(r->pool, conf, *url))) return DECLINED; /* Step 2: find the session route */ runtime = find_session_route(*balancer, r, &route, url); if (!runtime) { if (route && (*balancer)->sticky_force) { ap_log_error(APLOG_MARK, APLOG_ERR, 0, r->server, "balancer: (%s). All workers in error state for route (%s)", (*balancer)->name, route); return HTTP_SERVICE_UNAVAILABLE; } } /* Lock the LoadBalancer * XXX: perhaps we need the process lock here */ if ((rv = PROXY_BALANCER_LOCK(*balancer)) != APR_SUCCESS) { ap_log_error(APLOG_MARK, APLOG_ERR, rv, r->server, "proxy_balancer_pre_request: lock"); return DECLINED; } PROXY_BALANCER_UNLOCK(*balancer); return access_status; } static int proxy_balancer_post_request(proxy_worker *worker, proxy_balancer *balancer, request_rec *r, proxy_server_conf *conf) { int access_status; if (!balancer) access_status = DECLINED; else { access_status = OK; } return access_status; } static void ap_proxy_balancer_register_hook(apr_pool_t *p) { proxy_hook_pre_request(proxy_balancer_pre_request, NULL, NULL, APR_HOOK_FIRST); proxy_hook_post_request(proxy_balancer_post_request, NULL, NULL, APR_HOOK_FIRST); } module AP_MODULE_DECLARE_DATA proxy_balancer_module = { STANDARD20_MODULE_STUFF, NULL, /* create per-directory config structure */ NULL, /* merge per-directory config structures */ NULL, /* create per-server config structure */ NULL, /* merge per-server config structures */ NULL, /* command apr_table_t */ ap_proxy_balancer_register_hook /* register hooks */ };
--------------------------------------------------------------------- To unsubscribe, e-mail: [EMAIL PROTECTED] For additional commands, e-mail: [EMAIL PROTECTED]
