+++ /dev/null
-/*
- * Licensed to the Apache Software Foundation (ASF) under one or more
- * contributor license agreements. See the NOTICE file distributed with
- * this work for additional information regarding copyright ownership.
- * The ASF licenses this file to You under the Apache License, Version 2.0
- * (the "License"); you may not use this file except in compliance with
- * the License. You may obtain a copy of the License at
- *
- * http://www.apache.org/licenses/LICENSE-2.0
- *
- * Unless required by applicable law or agreed to in writing, software
- * distributed under the License is distributed on an "AS IS" BASIS,
- * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- * See the License for the specific language governing permissions and
- * limitations under the License.
- */
-
-/***************************************************************************
- * Description: common stuff for bi-directional protocols ajp13/ajp14. *
- * Author: Gal Shachor <shachor@il.ibm.com> *
- * Author: Henri Gomez <hgomez@apache.org> *
- * Version: $Revision: 1137200 $ *
- ***************************************************************************/
-
-
-#include "jk_global.h"
-#include "jk_util.h"
-#include "jk_ajp13.h"
-#include "jk_ajp14.h"
-#include "jk_ajp_common.h"
-#include "jk_connect.h"
-#if defined(AS400) && !defined(AS400_UTF8)
-#include "util_ebcdic.h"
-#endif
-#if defined(NETWARE) && defined(__NOVELL_LIBC__)
-#include "novsock2.h"
-#endif
-
-const char *response_trans_headers[] = {
- "Content-Type",
- "Content-Language",
- "Content-Length",
- "Date",
- "Last-Modified",
- "Location",
- "Set-Cookie",
- "Set-Cookie2",
- "Servlet-Engine",
- "Status",
- "WWW-Authenticate"
-};
-
-static const char *long_res_header_for_sc(int sc)
-{
- const char *rc = NULL;
- sc = sc & 0X00FF;
- if (sc <= SC_RES_HEADERS_NUM && sc > 0) {
- rc = response_trans_headers[sc - 1];
- }
-
- return rc;
-}
-
-static const char *ajp_state_type[] = {
- JK_AJP_STATE_TEXT_IDLE,
- JK_AJP_STATE_TEXT_OK,
- JK_AJP_STATE_TEXT_ERROR,
- JK_AJP_STATE_TEXT_PROBE,
- "unknown",
- NULL
-};
-
-#define UNKNOWN_METHOD (-1)
-
-static int sc_for_req_method(const char *method, size_t len)
-{
- /* Note: the following code was generated by the "shilka" tool from
- the "cocom" parsing/compilation toolkit. It is an optimized lookup
- based on analysis of the input keywords. Postprocessing was done
- on the shilka output, but the basic structure and analysis is
- from there. Should new HTTP methods be added, then manual insertion
- into this code is fine, or simply re-running the shilka tool on
- the appropriate input. */
-
- /* Note: it is also quite reasonable to just use our method_registry,
- but I'm assuming (probably incorrectly) we want more speed here
- (based on the optimizations the previous code was doing). */
-
- switch (len)
- {
- case 3:
- switch (method[0])
- {
- case 'A':
- return (method[1] == 'C'
- && method[2] == 'L'
- ? SC_M_ACL : UNKNOWN_METHOD);
- case 'P':
- return (method[1] == 'U'
- && method[2] == 'T'
- ? SC_M_PUT : UNKNOWN_METHOD);
- case 'G':
- return (method[1] == 'E'
- && method[2] == 'T'
- ? SC_M_GET : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 4:
- switch (method[0])
- {
- case 'H':
- return (method[1] == 'E'
- && method[2] == 'A'
- && method[3] == 'D'
- ? SC_M_HEAD : UNKNOWN_METHOD);
- case 'P':
- return (method[1] == 'O'
- && method[2] == 'S'
- && method[3] == 'T'
- ? SC_M_POST : UNKNOWN_METHOD);
- case 'M':
- return (method[1] == 'O'
- && method[2] == 'V'
- && method[3] == 'E'
- ? SC_M_MOVE : UNKNOWN_METHOD);
- case 'L':
- return (method[1] == 'O'
- && method[2] == 'C'
- && method[3] == 'K'
- ? SC_M_LOCK : UNKNOWN_METHOD);
- case 'C':
- return (method[1] == 'O'
- && method[2] == 'P'
- && method[3] == 'Y'
- ? SC_M_COPY : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 5:
- switch (method[2])
- {
- case 'R':
- return (memcmp(method, "MERGE", 5) == 0
- ? SC_M_MERGE : UNKNOWN_METHOD);
- case 'C':
- return (memcmp(method, "MKCOL", 5) == 0
- ? SC_M_MKCOL : UNKNOWN_METHOD);
- case 'B':
- return (memcmp(method, "LABEL", 5) == 0
- ? SC_M_LABEL : UNKNOWN_METHOD);
- case 'A':
- return (memcmp(method, "TRACE", 5) == 0
- ? SC_M_TRACE : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 6:
- switch (method[0])
- {
- case 'U':
- switch (method[5])
- {
- case 'K':
- return (memcmp(method, "UNLOCK", 6) == 0
- ? SC_M_UNLOCK : UNKNOWN_METHOD);
- case 'E':
- return (memcmp(method, "UPDATE", 6) == 0
- ? SC_M_UPDATE : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
- case 'R':
- return (memcmp(method, "REPORT", 6) == 0
- ? SC_M_REPORT : UNKNOWN_METHOD);
- case 'S':
- return (memcmp(method, "SEARCH", 6) == 0
- ? SC_M_SEARCH : UNKNOWN_METHOD);
- case 'D':
- return (memcmp(method, "DELETE", 6) == 0
- ? SC_M_DELETE : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 7:
- switch (method[1])
- {
- case 'P':
- return (memcmp(method, "OPTIONS", 7) == 0
- ? SC_M_OPTIONS : UNKNOWN_METHOD);
- case 'H':
- return (memcmp(method, "CHECKIN", 7) == 0
- ? SC_M_CHECKIN : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 8:
- switch (method[0])
- {
- case 'P':
- return (memcmp(method, "PROPFIND", 8) == 0
- ? SC_M_PROPFIND : UNKNOWN_METHOD);
- case 'C':
- return (memcmp(method, "CHECKOUT", 8) == 0
- ? SC_M_CHECKOUT : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 9:
- return (memcmp(method, "PROPPATCH", 9) == 0
- ? SC_M_PROPPATCH : UNKNOWN_METHOD);
-
- case 10:
- switch (method[0])
- {
- case 'U':
- return (memcmp(method, "UNCHECKOUT", 10) == 0
- ? SC_M_UNCHECKOUT : UNKNOWN_METHOD);
- case 'M':
- return (memcmp(method, "MKACTIVITY", 10) == 0
- ? SC_M_MKACTIVITY : UNKNOWN_METHOD);
- default:
- return UNKNOWN_METHOD;
- }
-
- case 11:
- return (memcmp(method, "MKWORKSPACE", 11) == 0
- ? SC_M_MKWORKSPACE : UNKNOWN_METHOD);
-
- case 15:
- return (memcmp(method, "VERSION-CONTROL", 15) == 0
- ? SC_M_VERSION_CONTROL : UNKNOWN_METHOD);
-
- case 16:
- return (memcmp(method, "BASELINE-CONTROL", 16) == 0
- ? SC_M_BASELINE_CONTROL : UNKNOWN_METHOD);
-
- default:
- return UNKNOWN_METHOD;
- }
-
- /* NOTREACHED */
-}
-
-static int sc_for_req_header(const char *header_name)
-{
- char header[16];
- size_t len = strlen(header_name);
- const char *p = header_name;
- int i = 0;
-
- /* ACCEPT-LANGUAGE is the longest header
- * that is of interest.
- */
- if (len < 4 || len > 15)
- return UNKNOWN_METHOD;
-
- while (*p) {
- header[i++] = toupper((unsigned char)*p);
- p++;
- }
-
- header[i] = '\0';
- p = &header[1];
-
-/* Always do memcmp including the final \0-termination character.
- */
- switch (header[0]) {
- case 'A':
- if (memcmp(p, "CCEPT", 6) == 0) {
- if (!header[6])
- return SC_ACCEPT;
- else if (header[6] == '-') {
- p += 6;
- if (memcmp(p, "CHARSET", 8) == 0)
- return SC_ACCEPT_CHARSET;
- else if (memcmp(p, "ENCODING", 9) == 0)
- return SC_ACCEPT_ENCODING;
- else if (memcmp(p, "LANGUAGE", 9) == 0)
- return SC_ACCEPT_LANGUAGE;
- else
- return UNKNOWN_METHOD;
- }
- else
- return UNKNOWN_METHOD;
- }
- else if (memcmp(p, "UTHORIZATION", 13) == 0)
- return SC_AUTHORIZATION;
- else
- return UNKNOWN_METHOD;
- break;
- case 'C':
- if(memcmp(p, "OOKIE2", 7) == 0)
- return SC_COOKIE2;
- else if (memcmp(p, "OOKIE", 6) == 0)
- return SC_COOKIE;
- else if(memcmp(p, "ONNECTION", 10) == 0)
- return SC_CONNECTION;
- else if(memcmp(p, "ONTENT-TYPE", 12) == 0)
- return SC_CONTENT_TYPE;
- else if(memcmp(p, "ONTENT-LENGTH", 14) == 0)
- return SC_CONTENT_LENGTH;
- else
- return UNKNOWN_METHOD;
- break;
- case 'H':
- if(memcmp(p, "OST", 4) == 0)
- return SC_HOST;
- else
- return UNKNOWN_METHOD;
- break;
- case 'P':
- if(memcmp(p, "RAGMA", 6) == 0)
- return SC_PRAGMA;
- else
- return UNKNOWN_METHOD;
- break;
- case 'R':
- if(memcmp(p, "EFERER", 7) == 0)
- return SC_REFERER;
- else
- return UNKNOWN_METHOD;
- break;
- case 'U':
- if(memcmp(p, "SER-AGENT", 10) == 0)
- return SC_USER_AGENT;
- else
- return UNKNOWN_METHOD;
- break;
- default:
- return UNKNOWN_METHOD;
- }
- /* NOTREACHED */
-}
-
-/* Return the string representation of the worker state */
-const char *jk_ajp_get_state(ajp_worker_t *aw, jk_logger_t *l)
-{
- return ajp_state_type[aw->s->state];
-}
-
-/* Return the int representation of the worker state */
-int jk_ajp_get_state_code(const char *v)
-{
- if (!v)
- return JK_AJP_STATE_DEF;
- else if (*v == 'i' || *v == 'I' || *v == 'n' || *v == 'N' || *v == '0')
- return JK_AJP_STATE_IDLE;
- else if (*v == 'o' || *v == 'O' || *v == '1')
- return JK_AJP_STATE_OK;
- else if (*v == 'e' || *v == 'E' || *v == '4')
- return JK_AJP_STATE_ERROR;
- else if (*v == 'p' || *v == 'P' || *v == '6')
- return JK_AJP_STATE_PROBE;
- else
- return JK_AJP_STATE_DEF;
-}
-
-int jk_ajp_get_cping_mode(const char *m, int def)
-{
- int mv = def;
- if (!m)
- return mv;
- while (*m != '\0') {
- if (*m == 'C' || *m == 'c')
- mv |= AJP_CPING_CONNECT;
- else if (*m == 'P' || *m == 'p')
- mv |= AJP_CPING_PREPOST;
- else if (*m == 'I' || *m == 'i')
- mv |= AJP_CPING_INTERVAL;
- else if (*m == 'A' || *m == 'a') {
- mv = AJP_CPING_CONNECT | AJP_CPING_PREPOST | AJP_CPING_INTERVAL;
- break;
- }
- m++;
- }
- return mv;
-}
-
-/*
- * Message structure
- *
- *
-AJPV13_REQUEST/AJPV14_REQUEST=
- request_prefix (1) (byte)
- method (byte)
- protocol (string)
- req_uri (string)
- remote_addr (string)
- remote_host (string)
- server_name (string)
- server_port (short)
- is_ssl (boolean)
- num_headers (short)
- num_headers*(req_header_name header_value)
-
- ?context (byte)(string)
- ?servlet_path (byte)(string)
- ?remote_user (byte)(string)
- ?auth_type (byte)(string)
- ?query_string (byte)(string)
- ?route (byte)(string)
- ?ssl_cert (byte)(string)
- ?ssl_cipher (byte)(string)
- ?ssl_session (byte)(string)
- ?ssl_key_size (byte)(int) via JkOptions +ForwardKeySize
- request_terminator (byte)
- ?body content_length*(var binary)
-
- */
-
-static int ajp_marshal_into_msgb(jk_msg_buf_t *msg,
- jk_ws_service_t *s,
- jk_logger_t *l, ajp_endpoint_t * ae)
-{
- int method;
- unsigned int i;
-
- JK_TRACE_ENTER(l);
-
- if ((method = sc_for_req_method(s->method,
- strlen(s->method))) == UNKNOWN_METHOD)
- method = SC_M_JK_STORED;
-
- if (jk_b_append_byte(msg, JK_AJP13_FORWARD_REQUEST) ||
- jk_b_append_byte(msg, (unsigned char)method) ||
- jk_b_append_string(msg, s->protocol) ||
- jk_b_append_string(msg, s->req_uri) ||
- jk_b_append_string(msg, s->remote_addr) ||
- jk_b_append_string(msg, s->remote_host) ||
- jk_b_append_string(msg, s->server_name) ||
- jk_b_append_int(msg, (unsigned short)s->server_port) ||
- jk_b_append_byte(msg, (unsigned char)(s->is_ssl)) ||
- jk_b_append_int(msg, (unsigned short)(s->num_headers))) {
-
- jk_log(l, JK_LOG_ERROR,
- "failed appending the message begining");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- for (i = 0; i < s->num_headers; i++) {
- int sc;
-
- if ((sc = sc_for_req_header(s->headers_names[i])) != UNKNOWN_METHOD) {
- if (jk_b_append_int(msg, (unsigned short)sc)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the header name");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- else {
- if (jk_b_append_string(msg, s->headers_names[i])) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the header name");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- if (jk_b_append_string(msg, s->headers_values[i])) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the header value");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- if (s->secret) {
- if (jk_b_append_byte(msg, SC_A_SECRET) ||
- jk_b_append_string(msg, s->secret)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending secret");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- if (s->remote_user) {
- if (jk_b_append_byte(msg, SC_A_REMOTE_USER) ||
- jk_b_append_string(msg, s->remote_user)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the remote user");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- if (s->auth_type) {
- if (jk_b_append_byte(msg, SC_A_AUTH_TYPE) ||
- jk_b_append_string(msg, s->auth_type)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the auth type");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- if (s->query_string) {
- if (jk_b_append_byte(msg, SC_A_QUERY_STRING) ||
-#if defined(AS400) && !defined(AS400_UTF8)
- jk_b_append_asciistring(msg, s->query_string)) {
-#else
- jk_b_append_string(msg, s->query_string)) {
-#endif
- jk_log(l, JK_LOG_ERROR,
- "failed appending the query string");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- if (s->route) {
- if (jk_b_append_byte(msg, SC_A_ROUTE) ||
- jk_b_append_string(msg, s->route)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the route");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- if (s->ssl_cert_len) {
- if (jk_b_append_byte(msg, SC_A_SSL_CERT) ||
- jk_b_append_string(msg, s->ssl_cert)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the SSL certificates");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- if (s->ssl_cipher) {
- if (jk_b_append_byte(msg, SC_A_SSL_CIPHER) ||
- jk_b_append_string(msg, s->ssl_cipher)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the SSL ciphers");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- if (s->ssl_session) {
- if (jk_b_append_byte(msg, SC_A_SSL_SESSION) ||
- jk_b_append_string(msg, s->ssl_session)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the SSL session");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- /*
- * ssl_key_size is required by Servlet 2.3 API
- * added support only in ajp14 mode
- * JFC removed: ae->proto == AJP14_PROTO
- */
- if (s->ssl_key_size != -1) {
- if (jk_b_append_byte(msg, SC_A_SSL_KEY_SIZE) ||
- jk_b_append_int(msg, (unsigned short)s->ssl_key_size)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the SSL key size");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- /* If the method was unrecognized, encode it as an attribute */
- if (method == SC_M_JK_STORED) {
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG, "unknown method %s", s->method);
- if (jk_b_append_byte(msg, SC_A_STORED_METHOD) ||
- jk_b_append_string(msg, s->method)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the request method");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- /* Forward the remote port information, which was forgotten
- * from the builtin data of the AJP 13 protocol.
- * Since the servlet spec allows to retrieve it via getRemotePort(),
- * we provide the port to the Tomcat connector as a request
- * attribute. Modern Tomcat versions know how to retrieve
- * the remote port from this attribute.
- */
- {
- if (jk_b_append_byte(msg, SC_A_REQ_ATTRIBUTE) ||
- jk_b_append_string(msg, SC_A_REQ_REMOTE_PORT) ||
- jk_b_append_string(msg, s->remote_port)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the remote port %s",
- s->remote_port);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- /* Forward activation information from the load balancer.
- * It can be used by the backend to deny access by requests,
- * which come with a session id but for an invalid session.
- * Such requests get forwarded to backends even if they
- * are disabled" in the load balancer, because the balancer
- * does not know, which sessions are valid.
- * If the backend can check, that is was "disabled" it can
- * delete the session cookie and respond with a self-referential
- * redirect. The new request will then be balanced to some
- * other node that is not disabled.
- */
- {
- if (jk_b_append_byte(msg, SC_A_REQ_ATTRIBUTE) ||
- jk_b_append_string(msg, SC_A_JK_LB_ACTIVATION) ||
- jk_b_append_string(msg, s->activation)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the activation state %s",
- s->activation);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
-
- if (s->num_attributes > 0) {
- for (i = 0; i < s->num_attributes; i++) {
- if (jk_b_append_byte(msg, SC_A_REQ_ATTRIBUTE) ||
- jk_b_append_string(msg, s->attributes_names[i]) ||
- jk_b_append_string(msg, s->attributes_values[i])) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending attribute %s=%s",
- s->attributes_names[i], s->attributes_values[i]);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- }
-
- if (jk_b_append_byte(msg, SC_A_ARE_DONE)) {
- jk_log(l, JK_LOG_ERROR,
- "failed appending the message end");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG, "ajp marshaling done");
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-/*
-AJPV13_RESPONSE/AJPV14_RESPONSE:=
- response_prefix (2)
- status (short)
- status_msg (short)
- num_headers (short)
- num_headers*(res_header_name header_value)
- *body_chunk
- terminator boolean <! -- recycle connection or not -->
-
-req_header_name :=
- sc_req_header_name | (string)
-
-res_header_name :=
- sc_res_header_name | (string)
-
-header_value :=
- (string)
-
-body_chunk :=
- length (short)
- body length*(var binary)
-
- */
-
-
-static int ajp_unmarshal_response(jk_msg_buf_t *msg,
- jk_res_data_t * d,
- ajp_endpoint_t * ae, jk_logger_t *l)
-{
- jk_pool_t *p = &ae->pool;
-
- JK_TRACE_ENTER(l);
-
- d->status = jk_b_get_int(msg);
- if (!d->status) {
- jk_log(l, JK_LOG_ERROR,
- "NULL status");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- d->msg = (char *)jk_b_get_string(msg);
- if (d->msg) {
-#if (defined(AS400) && !defined(AS400_UTF8)) || defined(_OSD_POSIX)
- jk_xlate_from_ascii(d->msg, strlen(d->msg));
-#endif
- }
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "status = %d", d->status);
-
- d->num_headers = jk_b_get_int(msg);
- d->header_names = d->header_values = NULL;
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "Number of headers is = %d",
- d->num_headers);
-
- if (d->num_headers) {
- d->header_names = jk_pool_alloc(p, sizeof(char *) * d->num_headers);
- d->header_values = jk_pool_alloc(p, sizeof(char *) * d->num_headers);
-
- if (d->header_names && d->header_values) {
- unsigned int i;
- for (i = 0; i < d->num_headers; i++) {
- unsigned short name = jk_b_pget_int(msg, msg->pos);
-
- if ((name & 0XFF00) == 0XA000) {
- jk_b_get_int(msg);
- name = name & 0X00FF;
- if (name <= SC_RES_HEADERS_NUM) {
- d->header_names[i] =
- (char *)long_res_header_for_sc(name);
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "No such sc (%d)", name);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- else {
- d->header_names[i] = (char *)jk_b_get_string(msg);
- if (!d->header_names[i]) {
- jk_log(l, JK_LOG_ERROR,
- "NULL header name");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-#if (defined(AS400) && !defined(AS400_UTF8)) || defined(_OSD_POSIX)
- jk_xlate_from_ascii(d->header_names[i],
- strlen(d->header_names[i]));
-#endif
-
- }
-
- d->header_values[i] = (char *)jk_b_get_string(msg);
- if (!d->header_values[i]) {
- jk_log(l, JK_LOG_ERROR,
- "NULL header value");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
-#if (defined(AS400) && !defined(AS400_UTF8)) || defined(_OSD_POSIX)
- jk_xlate_from_ascii(d->header_values[i],
- strlen(d->header_values[i]));
-#endif
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "Header[%d] [%s] = [%s]",
- i, d->header_names[i], d->header_values[i]);
- }
- }
- }
-
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-/*
- * Abort endpoint use
- */
-static void ajp_abort_endpoint(ajp_endpoint_t * ae, int shutdown, jk_logger_t *l)
-{
- JK_TRACE_ENTER(l);
- if (shutdown == JK_TRUE && IS_VALID_SOCKET(ae->sd)) {
- if (ae->hard_close) {
- /* Force unclean connection close to communicate client write errors
- * back to Tomcat by aborting AJP response writes.
- */
- jk_close_socket(ae->sd, l);
- }
- else {
- jk_shutdown_socket(ae->sd, l);
- }
- }
- ae->worker->s->connected--;
- ae->sd = JK_INVALID_SOCKET;
- ae->last_op = JK_AJP13_END_RESPONSE;
- JK_TRACE_EXIT(l);
-}
-
-/*
- * Reset the endpoint (clean buf and close socket)
- */
-static void ajp_reset_endpoint(ajp_endpoint_t * ae, jk_logger_t *l)
-{
- JK_TRACE_ENTER(l);
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) resetting endpoint with socket %d%s",
- ae->worker->name, ae->sd, ae->reuse? "" : " (socket shutdown)");
- if (!ae->reuse) {
- ajp_abort_endpoint(ae, JK_TRUE, l);
- }
- jk_reset_pool(&(ae->pool));
- JK_TRACE_EXIT(l);
-}
-
-/*
- * Close the endpoint (close pool and close socket)
- */
-void ajp_close_endpoint(ajp_endpoint_t * ae, jk_logger_t *l)
-{
- JK_TRACE_ENTER(l);
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) closing endpoint with socket %d%s",
- ae->worker->name, ae->sd, ae->reuse ? "" : " (socket shutdown)");
- if (IS_VALID_SOCKET(ae->sd)) {
- jk_shutdown_socket(ae->sd, l);
- }
- ae->sd = JK_INVALID_SOCKET;
- jk_close_pool(&(ae->pool));
- free(ae);
- JK_TRACE_EXIT(l);
-}
-
-
-/** Steal a connection from an idle cache endpoint
- * @param ae endpoint that needs a new connection
- * @param l logger
- * @return JK_FALSE: failure
- * JK_TRUE: success
- * @remark Always closes old socket endpoint
- */
-static int ajp_next_connection(ajp_endpoint_t *ae, jk_logger_t *l)
-{
- int rc;
- int ret = JK_FALSE;
- ajp_worker_t *aw = ae->worker;
-
- JK_TRACE_ENTER(l);
-
- /* Close previous socket */
- if (IS_VALID_SOCKET(ae->sd))
- jk_shutdown_socket(ae->sd, l);
- /* Mark existing endpoint socket as closed */
- ae->sd = JK_INVALID_SOCKET;
- JK_ENTER_CS(&aw->cs, rc);
- if (rc) {
- unsigned int i;
- for (i = 0; i < aw->ep_cache_sz; i++) {
- /* Find cache slot with usable socket */
- if (aw->ep_cache[i] && IS_VALID_SOCKET(aw->ep_cache[i]->sd)) {
- ae->sd = aw->ep_cache[i]->sd;
- aw->ep_cache[i]->sd = JK_INVALID_SOCKET;
- break;
- }
- }
- JK_LEAVE_CS(&aw->cs, rc);
- if (IS_VALID_SOCKET(ae->sd)) {
- ret = JK_TRUE;
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) Will try pooled connection socket %d from slot %d",
- ae->worker->name, ae->sd, i);
- }
- }
- JK_TRACE_EXIT(l);
- return ret;
-}
-
-/** Handle the cping/cpong query
- * @param ae endpoint
- * @param timeout wait timeout in milliseconds
- * @param l logger
- * @return JK_FALSE: failure
- * JK_TRUE: success
- * @remark Always closes socket in case of
- * a socket error
- */
-static int ajp_handle_cping_cpong(ajp_endpoint_t * ae, int timeout, jk_logger_t *l)
-{
- int i;
- int cmd;
- jk_msg_buf_t *msg;
-
- JK_TRACE_ENTER(l);
-
- ae->last_errno = 0;
- msg = jk_b_new(&ae->pool);
- if (!msg) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- if (jk_b_set_buffer_size(msg, 16)) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message buffer");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- jk_b_reset(msg);
- jk_b_append_byte(msg, AJP13_CPING_REQUEST);
-
- /* Send CPing query */
- if (ajp_connection_tcp_send_message(ae, msg, l) != JK_TRUE) {
- jk_log(l, JK_LOG_INFO,
- "can't send cping query");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- for (i = 0; i < 2; i++) {
- /* wait for Pong reply for timeout milliseconds
- */
- if (jk_is_input_event(ae->sd, timeout, l) == JK_FALSE) {
- ae->last_errno = errno;
- jk_log(l, JK_LOG_INFO, "timeout in reply cpong");
- /* We can't trust this connection any more. */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- /* Read and check for Pong reply
- */
- if (ajp_connection_tcp_get_message(ae, msg, l) != JK_TRUE) {
- jk_log(l, JK_LOG_INFO,
- "awaited reply cpong, not received");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- if ((cmd = jk_b_get_byte(msg)) != AJP13_CPONG_REPLY) {
- /* If the respose was not CPONG it means that
- * the previous response was not consumed by the
- * client but the AJP messages was already in
- * the network buffer.
- * silently drop this single extra packet instead
- * recycling the connection
- */
- if (i || ae->last_op == JK_AJP13_END_RESPONSE ||
- cmd < JK_AJP13_SEND_BODY_CHUNK ||
- cmd > AJP13_CPONG_REPLY) {
- jk_log(l, JK_LOG_WARNING,
- "awaited reply cpong, received %d instead. "
- "Closing connection",
- cmd);
- /* We can't trust this connection any more. */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- else {
- jk_log(l, JK_LOG_INFO,
- "awaited reply cpong, received %d instead. "
- "Retrying next packet",
- cmd);
-
- }
- }
- else {
- ae->last_op = AJP13_CPONG_REPLY;
- /* We have received Pong reply */
- break;
- }
- }
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-/** Connect an endpoint to a backend
- * @param ae endpoint
- * @param l logger
- * @return JK_FALSE: failure
- * JK_TRUE: success
- * @remark Always closes socket in case of
- * a socket error
- * @remark Cares about ae->last_errno
- */
-int ajp_connect_to_endpoint(ajp_endpoint_t * ae, jk_logger_t *l)
-{
- char buf[32];
- int rc = JK_TRUE;
-
- JK_TRACE_ENTER(l);
-
- ae->last_errno = 0;
- ae->sd = jk_open_socket(&ae->worker->worker_inet_addr,
- ae->worker->keepalive,
- ae->worker->socket_timeout,
- ae->worker->socket_connect_timeout,
- ae->worker->socket_buf, l);
-
- if (!IS_VALID_SOCKET(ae->sd)) {
- ae->last_errno = errno;
- jk_log(l, JK_LOG_INFO,
- "Failed opening socket to (%s) (errno=%d)",
- jk_dump_hinfo(&ae->worker->worker_inet_addr, buf), ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- ae->worker->s->connected++;
- /* set last_access only if needed */
- if (ae->worker->cache_timeout > 0)
- ae->last_access = time(NULL);
- /* Check if we must execute a logon after the physical connect */
- /* XXX: Not sure, if we really should do logon before cping/cpong */
- /* XXX: and if no cping/cpong is allowed before or after logon. */
- if (ae->worker->logon != NULL) {
- rc = ae->worker->logon(ae, l);
- if (rc == JK_FALSE) {
- jk_log(l, JK_LOG_ERROR,
- "(%s) ajp14 worker logon to the backend server failed",
- ae->worker->name);
- /* Close the socket if unable to logon */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- }
- }
- /* XXX: Should we send a cping also after logon to validate the connection? */
- else if (ae->worker->connect_timeout > 0) {
- rc = ajp_handle_cping_cpong(ae, ae->worker->connect_timeout, l);
- if (rc == JK_FALSE)
- jk_log(l, JK_LOG_ERROR,
- "(%s) cping/cpong after connecting to the backend server failed (errno=%d)",
- ae->worker->name, ae->last_errno);
- }
- JK_TRACE_EXIT(l);
- return rc;
-}
-
-/* Syncing config values from shm */
-void jk_ajp_pull(ajp_worker_t * aw, int locked, jk_logger_t *l)
-{
- int address_change = JK_FALSE;
- int port = 0;
- char host[JK_SHM_STR_SIZ+1];
- struct sockaddr_in inet_addr;
- JK_TRACE_ENTER(l);
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "syncing mem for ajp worker '%s' from shm (%u -> %u) [%u->%u]",
- aw->name, aw->sequence, aw->s->h.sequence, aw->addr_sequence, aw->s->addr_sequence);
- if (locked == JK_FALSE)
- jk_shm_lock();
-
- aw->cache_timeout = aw->s->cache_timeout;
- aw->connect_timeout = aw->s->connect_timeout;
- aw->ping_timeout = aw->s->ping_timeout;
- aw->reply_timeout = aw->s->reply_timeout;
- aw->prepost_timeout = aw->s->prepost_timeout;
- aw->recovery_opts = aw->s->recovery_opts;
- aw->retries = aw->s->retries;
- aw->retry_interval = aw->s->retry_interval;
- aw->max_packet_size = aw->s->max_packet_size;
- aw->sequence = aw->s->h.sequence;
- if (aw->addr_sequence != aw->s->addr_sequence) {
- address_change = JK_TRUE;
- aw->addr_sequence = aw->s->addr_sequence;
- strncpy(host, aw->s->host, JK_SHM_STR_SIZ);
- port = aw->s->port;
- }
- if (locked == JK_FALSE)
- jk_shm_unlock();
-
- if (address_change == JK_TRUE) {
- if (!jk_resolve(host, port, &inet_addr,
- aw->worker.we->pool, l)) {
- jk_log(l, JK_LOG_ERROR,
- "Failed resolving address '%s:%d' for worker '%s'.",
- host, port, aw->name);
- }
- else {
- int rc;
- JK_ENTER_CS(&aw->cs, rc);
- if (rc) {
- unsigned int i;
- for (i = 0; i < aw->ep_cache_sz; i++) {
- /* Close all connections in the cache */
- if (aw->ep_cache[i] && IS_VALID_SOCKET(aw->ep_cache[i]->sd)) {
- int sd = aw->ep_cache[i]->sd;
- aw->ep_cache[i]->sd = JK_INVALID_SOCKET;
- aw->ep_cache[i]->addr_sequence = aw->addr_sequence;
- jk_shutdown_socket(sd, l);
- aw->s->connected--;
- }
- }
- }
- aw->port = port;
- strncpy(aw->host, host, JK_SHM_STR_SIZ);
- memcpy(&(aw->worker_inet_addr), &inet_addr, sizeof(inet_addr));
- if (rc) {
- JK_LEAVE_CS(&aw->cs, rc);
- } else {
- jk_log(l, JK_LOG_ERROR,
- "locking thread (errno=%d)", errno);
- }
- }
- }
-
- JK_TRACE_EXIT(l);
-}
-
-/* Syncing config values to shm */
-void jk_ajp_push(ajp_worker_t * aw, int locked, jk_logger_t *l)
-{
- int address_change = JK_FALSE;
-
- JK_TRACE_ENTER(l);
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "syncing shm for ajp worker '%s' from mem (%u -> %u) [%u->%u]",
- aw->name, aw->s->h.sequence, aw->sequence, aw->s->addr_sequence, aw->addr_sequence);
- if (locked == JK_FALSE)
- jk_shm_lock();
-
- aw->s->cache_timeout = aw->cache_timeout;
- aw->s->connect_timeout = aw->connect_timeout;
- aw->s->ping_timeout = aw->ping_timeout;
- aw->s->reply_timeout = aw->reply_timeout;
- aw->s->prepost_timeout = aw->prepost_timeout;
- aw->s->recovery_opts = aw->recovery_opts;
- aw->s->retries = aw->retries;
- aw->s->retry_interval = aw->retry_interval;
- aw->s->max_packet_size = aw->max_packet_size;
- aw->s->h.sequence = aw->sequence;
- if (aw->s->addr_sequence != aw->addr_sequence) {
- address_change = JK_TRUE;
- strncpy(aw->s->host, aw->host, JK_SHM_STR_SIZ);
- aw->s->port = aw->port;
- aw->s->addr_sequence = aw->addr_sequence;
- }
- if (locked == JK_FALSE)
- jk_shm_unlock();
-
- if (address_change == JK_TRUE) {
- int rc;
- JK_ENTER_CS(&aw->cs, rc);
- if (rc) {
- unsigned int i;
- for (i = 0; i < aw->ep_cache_sz; i++) {
- /* Close all connections in the cache */
- if (aw->ep_cache[i] && IS_VALID_SOCKET(aw->ep_cache[i]->sd)) {
- int sd = aw->ep_cache[i]->sd;
- aw->ep_cache[i]->sd = JK_INVALID_SOCKET;
- aw->ep_cache[i]->addr_sequence = aw->addr_sequence;
- jk_shutdown_socket(sd, l);
- aw->s->connected--;
- }
- }
- JK_LEAVE_CS(&aw->cs, rc);
- } else {
- jk_log(l, JK_LOG_ERROR,
- "locking thread (errno=%d)", errno);
- }
- }
- JK_TRACE_EXIT(l);
-}
-
-/** Send a message to an endpoint, using corresponding PROTO HEADER
- * @param ae endpoint
- * @param msg message to send
- * @param l logger
- * @return JK_FATAL_ERROR: endpoint contains unknown protocol
- * JK_FALSE: other failure
- * JK_TRUE: success
- * @remark Always closes socket in case of
- * a socket error, or JK_FATAL_ERROR
- * @remark Cares about ae->last_errno
- */
-int ajp_connection_tcp_send_message(ajp_endpoint_t * ae,
- jk_msg_buf_t *msg, jk_logger_t *l)
-{
- int rc;
-
- JK_TRACE_ENTER(l);
-
- ae->last_errno = 0;
- if (ae->proto == AJP13_PROTO) {
- jk_b_end(msg, AJP13_WS_HEADER);
- if (JK_IS_DEBUG_LEVEL(l))
- jk_dump_buff(l, JK_LOG_DEBUG, "sending to ajp13", msg);
- }
- else if (ae->proto == AJP14_PROTO) {
- jk_b_end(msg, AJP14_WS_HEADER);
- if (JK_IS_DEBUG_LEVEL(l))
- jk_dump_buff(l, JK_LOG_DEBUG, "sending to ajp14", msg);
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "(%s) unknown protocol %d, supported are AJP13/AJP14",
- ae->worker->name, ae->proto);
- /* We've got a protocol error. */
- /* We can't trust this connection any more, */
- /* because we might have send already parts of the request. */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
-
- /* This is the only place in this function where we use the socket. */
- /* If sendfull gets an error, it implicitely closes the socket. */
- /* So any socket error inside ajp_connection_tcp_send_message */
- /* results in a socket close and invalidated endpoint connection. */
- if ((rc = jk_tcp_socket_sendfull(ae->sd, msg->buf,
- msg->len, l)) > 0) {
- ae->endpoint.wr += (jk_uint64_t)rc;
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- ae->last_errno = errno;
- jk_log(l, JK_LOG_INFO,
- "sendfull for socket %d returned %d (errno=%d)",
- ae->sd, rc, ae->last_errno);
- ajp_abort_endpoint(ae, JK_FALSE, l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-/** Receive a message from an endpoint, checking PROTO HEADER
- * @param ae endpoint
- * @param msg message to send
- * @param l logger
- * @return JK_TRUE: success
- * JK_FALSE: could not read the AJP packet header
- * JK_AJP_PROTOCOL_ERROR: failure after reading
- * the AJP packet header
- * @remark Always closes socket in case of
- * a socket error
- * @remark Cares about ae->last_errno
- */
-int ajp_connection_tcp_get_message(ajp_endpoint_t * ae,
- jk_msg_buf_t *msg, jk_logger_t *l)
-{
- unsigned char head[AJP_HEADER_LEN];
- int rc;
- int msglen;
- unsigned int header;
- char buf[32];
-
- JK_TRACE_ENTER(l);
-
- ae->last_errno = 0;
- /* If recvfull gets an error, it implicitely closes the socket. */
- /* We will invalidate the endpoint connection. */
- rc = jk_tcp_socket_recvfull(ae->sd, head, AJP_HEADER_LEN, l);
-
- /* If the return code is not negative */
- /* then we always get back the correct number of bytes. */
- if (rc < 0) {
- if (rc == JK_SOCKET_EOF) {
- ae->last_errno = EPIPE;
- jk_log(l, JK_LOG_INFO,
- "(%s) can't receive the response header message from tomcat, "
- "tomcat (%s) has forced a connection close for socket %d",
- ae->worker->name, jk_dump_hinfo(&ae->worker->worker_inet_addr, buf),
- ae->sd);
- }
- else {
- ae->last_errno = -rc;
- jk_log(l, JK_LOG_INFO,
- "(%s) can't receive the response header message from tomcat, "
- "network problems or tomcat (%s) is down (errno=%d)",
- ae->worker->name, jk_dump_hinfo(&ae->worker->worker_inet_addr, buf),
- ae->last_errno);
- }
- ajp_abort_endpoint(ae, JK_FALSE, l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- ae->endpoint.rd += (jk_uint64_t)rc;
- header = ((unsigned int)head[0] << 8) | head[1];
-
- if (ae->proto == AJP13_PROTO) {
- if (header != AJP13_SW_HEADER) {
-
- if (header == AJP14_SW_HEADER) {
- jk_log(l, JK_LOG_ERROR,
- "received AJP14 reply on an AJP13 connection from %s",
- jk_dump_hinfo(&ae->worker->worker_inet_addr, buf));
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "wrong message format 0x%04x from %s",
- header, jk_dump_hinfo(&ae->worker->worker_inet_addr,
- buf));
- }
- /* We've got a protocol error. */
- /* We can't trust this connection any more. */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_AJP_PROTOCOL_ERROR;
- }
- }
- else if (ae->proto == AJP14_PROTO) {
- if (header != AJP14_SW_HEADER) {
-
- if (header == AJP13_SW_HEADER) {
- jk_log(l, JK_LOG_ERROR,
- "received AJP13 reply on an AJP14 connection from %s",
- jk_dump_hinfo(&ae->worker->worker_inet_addr, buf));
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "wrong message format 0x%04x from %s",
- header, jk_dump_hinfo(&ae->worker->worker_inet_addr,
- buf));
- }
- /* We've got a protocol error. */
- /* We can't trust this connection any more. */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_AJP_PROTOCOL_ERROR;
- }
- }
-
- msglen = ((head[2] & 0xff) << 8);
- msglen += (head[3] & 0xFF);
-
- if (msglen > msg->maxlen) {
- jk_log(l, JK_LOG_ERROR,
- "wrong message size %d %d from %s",
- msglen, msg->maxlen,
- jk_dump_hinfo(&ae->worker->worker_inet_addr, buf));
- /* We've got a protocol error. */
- /* We can't trust this connection any more. */
- ajp_abort_endpoint(ae, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_AJP_PROTOCOL_ERROR;
- }
-
- msg->len = msglen;
- msg->pos = 0;
-
- /* If recvfull gets an error, it implicitely closes the socket. */
- /* We will invalidate the endpoint connection. */
- rc = jk_tcp_socket_recvfull(ae->sd, msg->buf, msglen, l);
- /* If the return code is not negative */
- /* then we always get back the correct number of bytes. */
- if (rc < 0) {
- if (rc == JK_SOCKET_EOF) {
- ae->last_errno = EPIPE;
- jk_log(l, JK_LOG_ERROR,
- "(%s) can't receive the response body message from tomcat, "
- "tomcat (%s) has forced a connection close for socket %d",
- ae->worker->name, jk_dump_hinfo(&ae->worker->worker_inet_addr, buf),
- ae->sd);
- }
- else {
- ae->last_errno = -rc;
- jk_log(l, JK_LOG_ERROR,
- "(%s) can't receive the response body message from tomcat, "
- "network problems or tomcat (%s) is down (errno=%d)",
- ae->worker->name, jk_dump_hinfo(&ae->worker->worker_inet_addr, buf),
- ae->last_errno);
- }
- ajp_abort_endpoint(ae, JK_FALSE, l);
- JK_TRACE_EXIT(l);
- /* Although we have a connection, this is effectively a protocol error.
- * We received the AJP header packet, but not the packet payload
- */
- return JK_AJP_PROTOCOL_ERROR;
- }
- ae->endpoint.rd += (jk_uint64_t)rc;
-
- if (JK_IS_DEBUG_LEVEL(l)) {
- if (ae->proto == AJP13_PROTO)
- jk_dump_buff(l, JK_LOG_DEBUG, "received from ajp13", msg);
- else if (ae->proto == AJP14_PROTO)
- jk_dump_buff(l, JK_LOG_DEBUG, "received from ajp14", msg);
- }
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-/*
- * Read all the data from the socket.
- *
- * Socket API doesn't guaranty that all the data will be kept in a
- * single read, so we must loop until all awaited data is received
- */
-
-static int ajp_read_fully_from_server(jk_ws_service_t *s, jk_logger_t *l,
- unsigned char *buf, unsigned int len)
-{
- unsigned int rdlen = 0;
- unsigned int padded_len = len;
-
- JK_TRACE_ENTER(l);
-
- if (s->is_chunked && s->no_more_chunks) {
- JK_TRACE_EXIT(l);
- return 0;
- }
- if (s->is_chunked) {
- /* Corner case: buf must be large enough to hold next
- * chunk size (if we're on or near a chunk border).
- * Pad the length to a reasonable value, otherwise the
- * read fails and the remaining chunks are tossed.
- */
- padded_len = (len < CHUNK_BUFFER_PAD) ? len : len - CHUNK_BUFFER_PAD;
- }
-
- while (rdlen < padded_len) {
- unsigned int this_time = 0;
- if (!s->read(s, buf + rdlen, len - rdlen, &this_time)) {
- /* Remote Client read failed. */
- JK_TRACE_EXIT(l);
- return JK_CLIENT_RD_ERROR;
- }
-
- if (0 == this_time) {
- if (s->is_chunked) {
- s->no_more_chunks = 1; /* read no more */
- }
- break;
- }
- rdlen += this_time;
- }
-
- JK_TRACE_EXIT(l);
- return (int)rdlen;
-}
-
-
-/*
- * Read data from AJP13/AJP14 protocol
- * Returns -1 on error, else number of bytes read
- */
-
-static int ajp_read_into_msg_buff(ajp_endpoint_t * ae,
- jk_ws_service_t *r,
- jk_msg_buf_t *msg, int len, jk_logger_t *l)
-{
- unsigned char *read_buf = msg->buf;
-
- JK_TRACE_ENTER(l);
-
- jk_b_reset(msg);
-
- read_buf += AJP_HEADER_LEN; /* leave some space for the buffer headers */
- read_buf += AJP_HEADER_SZ_LEN; /* leave some space for the read length */
-
- /* Pick the max size since we don't know the content_length */
- if (r->is_chunked && len == 0) {
- len = AJP13_MAX_SEND_BODY_SZ;
- }
-
- if ((len = ajp_read_fully_from_server(r, l, read_buf, len)) < 0) {
- jk_log(l, JK_LOG_INFO,
- "(%s) receiving data from client failed. "
- "Connection aborted or network problems",
- ae->worker->name);
- JK_TRACE_EXIT(l);
- return JK_CLIENT_RD_ERROR;
- }
-
- if (!r->is_chunked) {
- ae->left_bytes_to_send -= len;
- }
-
- if (len > 0) {
- /* Recipient recognizes empty packet as end of stream, not
- an empty body packet */
- if (0 != jk_b_append_int(msg, (unsigned short)len)) {
- jk_log(l, JK_LOG_INFO,
- "Failed appending message length");
- JK_TRACE_EXIT(l);
- return JK_CLIENT_RD_ERROR;
- }
- }
-
- msg->len += len;
-
- JK_TRACE_EXIT(l);
- return len;
-}
-
-
-/*
- * send request to Tomcat via Ajp13
- * - first try to find reuseable socket
- * - if no such available, try to connect
- * - send request, but send must be seen as asynchronous,
- * since send() call will return noerror about 95% of time
- * Hopefully we'll get more information on next read.
- *
- * nb: op->request is the original request msg buffer
- * op->reply is the reply msg buffer which could be scratched
- *
- * Return values of ajp_send_request() function:
- * return value op->recoverable reason
- * JK_FATAL_ERROR JK_FALSE ajp_connection_tcp_send_message() returns JK_FATAL_ERROR
- * Endpoint belongs to unknown protocol.
- * JK_FATAL_ERROR JK_TRUE ajp_connection_tcp_send_message() returns JK_FALSE
- * Sending request or request body in jk_tcp_socket_sendfull() returns with error.
- * JK_FATAL_ERROR JK_TRUE Could not connect to backend
- * JK_CLIENT_RD_ERROR JK_FALSE Error during reading parts of POST body from client
- * JK_TRUE JK_TRUE All other cases (OK)
- */
-static int ajp_send_request(jk_endpoint_t *e,
- jk_ws_service_t *s,
- jk_logger_t *l,
- ajp_endpoint_t * ae, ajp_operation_t * op)
-{
- int err_conn = 0;
- int err_cping = 0;
- int err_send = 0;
- int rc;
- int postlen;
-
- JK_TRACE_ENTER(l);
-
- ae->last_errno = 0;
- /* Up to now, we can recover */
- op->recoverable = JK_TRUE;
-
- /* Check if the previous request really ended
- */
- if (ae->last_op != JK_AJP13_END_RESPONSE &&
- ae->last_op != AJP13_CPONG_REPLY) {
- jk_log(l, JK_LOG_INFO,
- "(%s) did not receive END_RESPONSE, "
- "closing socket %d",
- ae->worker->name, ae->sd);
- ajp_abort_endpoint(ae, JK_TRUE, l);
- }
- /*
- * First try to check open connections...
- */
- while (IS_VALID_SOCKET(ae->sd)) {
- int err = JK_FALSE;
- if (jk_is_socket_connected(ae->sd, l) == JK_FALSE) {
- ae->last_errno = errno;
- jk_log(l, JK_LOG_DEBUG,
- "(%s) failed sending request, "
- "socket %d is not connected any more (errno=%d)",
- ae->worker->name, ae->sd, ae->last_errno);
- ajp_abort_endpoint(ae, JK_FALSE, l);
- err = JK_TRUE;
- err_conn++;
- }
- if (ae->worker->prepost_timeout > 0 && !err) {
- /* handle cping/cpong if prepost_timeout is set
- * If the socket is disconnected no need to handle
- * the cping/cpong
- */
- if (ajp_handle_cping_cpong(ae,
- ae->worker->prepost_timeout, l) == JK_FALSE) {
- jk_log(l, JK_LOG_INFO,
- "(%s) failed sending request, "
- "socket %d prepost cping/cpong failure (errno=%d)",
- ae->worker->name, ae->sd, ae->last_errno);
- /* XXX: Is there any reason to try other
- * connections to the node if one of them fails
- * the cping/cpong heartbeat?
- * Tomcat can be either too busy or simply dead, so
- * there is a chance that all other connections would
- * fail as well.
- */
- err = JK_TRUE;
- err_cping++;
- }
- }
-
- /* We've got a connected socket and the optional
- * cping/cpong worked, so let's send the request now.
- */
- if (err == JK_FALSE) {
- rc = ajp_connection_tcp_send_message(ae, op->request, l);
- /* If this worked, we can break out of the loop
- * and proceed with the request.
- */
- if (rc == JK_TRUE) {
- ae->last_op = JK_AJP13_FORWARD_REQUEST;
- break;
- }
- /* Error during sending the request.
- */
- err_send++;
- if (rc == JK_FATAL_ERROR)
- op->recoverable = JK_FALSE;
- jk_log(l, JK_LOG_INFO,
- "(%s) failed sending request (%srecoverable) "
- "(errno=%d)",
- ae->worker->name,
- op->recoverable ? "" : "un",
- ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- /* If we got an error or can't send data, then try to steal another pooled
- * connection and try again. If we are not successful, break out of this
- * loop and try to open a new connection after the loop.
- */
- if (ajp_next_connection(ae, l) == JK_FALSE)
- break;
- }
-
- /*
- * If we failed to reuse a connection, try to reconnect.
- */
- if (!IS_VALID_SOCKET(ae->sd)) {
- /* Could not steal any connection from an endpoint - backend is disconnected */
- if (err_conn + err_cping + err_send > 0)
- jk_log(l, JK_LOG_INFO,
- "(%s) all endpoints are disconnected, "
- "detected by connect check (%d), cping (%d), send (%d)",
- ae->worker->name, err_conn, err_cping, err_send);
- else if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) all endpoints are disconnected.",
- ae->worker->name);
- /* Connect to the backend.
- */
- if (ajp_connect_to_endpoint(ae, l) != JK_TRUE) {
- jk_log(l, JK_LOG_ERROR,
- "(%s) connecting to backend failed. Tomcat is probably not started "
- "or is listening on the wrong port (errno=%d)",
- ae->worker->name, ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- if (ae->worker->connect_timeout <= 0 &&
- ae->worker->prepost_timeout > 0) {
- /* handle cping/cpong if prepost_timeout is set
- * and we didn't already do a connect cping/cpong.
- */
- if (ajp_handle_cping_cpong(ae,
- ae->worker->prepost_timeout, l) == JK_FALSE) {
- jk_log(l, JK_LOG_INFO,
- "(%s) failed sending request, "
- "socket %d prepost cping/cpong failure (errno=%d)",
- ae->worker->name, ae->sd, ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- }
-
- /* We've got a connected socket and the optional
- * cping/cpong worked, so let's send the request now.
- */
- rc = ajp_connection_tcp_send_message(ae, op->request, l);
- /* Error during sending the request.
- */
- if (rc != JK_TRUE) {
- if (rc == JK_FATAL_ERROR)
- op->recoverable = JK_FALSE;
- jk_log(l, JK_LOG_ERROR,
- "(%s) failed sending request on a fresh connection (%srecoverable), "
- "socket %d (errno=%d)",
- ae->worker->name, op->recoverable ? "" : "un",
- ae->sd, ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- ae->last_op = JK_AJP13_FORWARD_REQUEST;
- }
- else if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) Statistics about invalid connections: "
- "connect check (%d), cping (%d), send (%d)",
- ae->worker->name, err_conn, err_cping, err_send);
-
- /*
- * From now on an error means that we have an internal server error
- * or Tomcat crashed.
- */
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) request body to send %" JK_UINT64_T_FMT " - request body to resend %d",
- ae->worker->name, ae->left_bytes_to_send,
- op->reply->len - AJP_HEADER_LEN);
-
- /*
- * POST recovery job is done here and will work when data to
- * POST are less than 8k, since it's the maximum size of op-post buffer.
- * We send here the first part of data which was sent previously to the
- * remote Tomcat
- */
-
- /* Did we have something to resend (ie the op-post has been feeded previously */
-
- postlen = op->post->len;
- if (postlen > AJP_HEADER_LEN) {
- rc = ajp_connection_tcp_send_message(ae, op->post, l);
- /* Error during sending the request body.
- */
- if (rc != JK_TRUE) {
- if (rc == JK_FATAL_ERROR)
- op->recoverable = JK_FALSE;
- jk_log(l, JK_LOG_ERROR,
- "(%s) failed sending request body of size %d (%srecoverable), "
- "socket %d (errno=%d)",
- ae->worker->name, postlen, op->recoverable ? "" : "un",
- ae->sd, ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- else {
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG, "Resent the request body (%d)",
- postlen);
- }
- }
- else if (s->reco_status == RECO_FILLED) {
- /* Recovery in LB MODE */
- postlen = s->reco_buf->len;
-
- if (postlen > AJP_HEADER_LEN) {
- rc = ajp_connection_tcp_send_message(ae, s->reco_buf, l);
- /* Error during sending the request body.
- */
- if (rc != JK_TRUE) {
- if (rc == JK_FATAL_ERROR)
- op->recoverable = JK_FALSE;
- jk_log(l, JK_LOG_ERROR,
- "(%s) failed sending request body of size %d (lb mode) (%srecoverable), "
- "socket %d (errno=%d)",
- ae->worker->name, postlen, op->recoverable ? "" : "un",
- ae->sd, ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- }
- else {
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "Resent the request body (lb mode) (%d)", postlen);
- }
- }
- else {
- /* We never sent any POST data and we check if we have to send at
- * least one block of data (max 8k). These data will be kept in reply
- * for resend if the remote Tomcat is down, a fact we will learn only
- * doing a read (not yet)
- */
- /* || s->is_chunked - this can't be done here. The original protocol
- sends the first chunk of post data ( based on Content-Length ),
- and that's what the java side expects.
- Sending this data for chunked would break other ajp13 servers.
-
- Note that chunking will continue to work - using the normal read.
- */
-
- if (ae->left_bytes_to_send > 0) {
- int len = AJP13_MAX_SEND_BODY_SZ;
- if (ae->left_bytes_to_send < (jk_uint64_t)AJP13_MAX_SEND_BODY_SZ) {
- len = (int)ae->left_bytes_to_send;
- }
- if ((len = ajp_read_into_msg_buff(ae, s, op->post, len, l)) <= 0) {
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) browser stop sending data, no need to recover",
- ae->worker->name);
- op->recoverable = JK_FALSE;
- /* Send an empty POST message since per AJP protocol
- * spec whenever we have content length the message
- * packet must be followed with initial POST packet.
- * Size zero will be handled as error in container.
- */
- jk_b_reset(op->post);
- jk_b_append_int(op->post, 0);
- ajp_connection_tcp_send_message(ae, op->post, l);
- JK_TRACE_EXIT(l);
- return JK_CLIENT_RD_ERROR;
- }
-
- /* If a RECOVERY buffer is available in LB mode, fill it */
- if (s->reco_status == RECO_INITED) {
- jk_b_copy(op->post, s->reco_buf);
- s->reco_status = RECO_FILLED;
- }
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "(%s) sending %d bytes of request body",
- ae->worker->name, len);
-
- s->content_read = (jk_uint64_t)len;
- rc = ajp_connection_tcp_send_message(ae, op->post, l);
- /* Error during sending the request body.
- */
- if (rc != JK_TRUE) {
- if (rc == JK_FATAL_ERROR)
- op->recoverable = JK_FALSE;
- jk_log(l, JK_LOG_ERROR,
- "(%s) failed sending request body of size %d (%srecoverable), "
- "socket %d (errno=%d)",
- ae->worker->name, len, op->recoverable ? "" : "un",
- ae->sd, ae->last_errno);
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- }
- }
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-
-/*
- * What to do with incoming data (dispatcher)
- */
-
-static int ajp_process_callback(jk_msg_buf_t *msg,
- jk_msg_buf_t *pmsg,
- ajp_endpoint_t * ae,
- jk_ws_service_t *r, jk_logger_t *l)
-{
- int code = (int)jk_b_get_byte(msg);
-
- JK_TRACE_ENTER(l);
-
- switch (code) {
- case JK_AJP13_SEND_HEADERS:
- {
- int rc;
- jk_res_data_t res;
- if (ae->last_op == JK_AJP13_SEND_HEADERS) {
- /* Do not send anything to the client.
- * Backend already send us the headers.
- */
- if (JK_IS_DEBUG_LEVEL(l)) {
- jk_log(l, JK_LOG_DEBUG,
- "Already received AJP13_SEND HEADERS");
- }
- JK_TRACE_EXIT(l);
- return JK_AJP13_ERROR;
- }
- if (!ajp_unmarshal_response(msg, &res, ae, l)) {
- jk_log(l, JK_LOG_ERROR,
- "ajp_unmarshal_response failed");
- JK_TRACE_EXIT(l);
- return JK_AJP13_ERROR;
- }
- r->http_response_status = res.status;
- if (r->extension.fail_on_status_size > 0)
- rc = is_http_status_fail(r->extension.fail_on_status_size,
- r->extension.fail_on_status, res.status);
- else
- rc = is_http_status_fail(ae->worker->http_status_fail_num,
- ae->worker->http_status_fail, res.status);
- if (rc > 0) {
- JK_TRACE_EXIT(l);
- return JK_STATUS_FATAL_ERROR;
- }
- else if (rc < 0) {
- JK_TRACE_EXIT(l);
- return JK_STATUS_ERROR;
- }
-
- if (r->extension.use_server_error_pages &&
- r->http_response_status >= r->extension.use_server_error_pages)
- r->response_blocked = JK_TRUE;
-
- /*
- * Call even if response is blocked, since it also handles
- * forwarding some headers for special http status codes
- * even if the server uses an own error page.
- * Example: The WWW-Authenticate header in case of
- * HTTP_UNAUTHORIZED (401).
- */
- r->start_response(r, res.status, res.msg,
- (const char *const *)res.header_names,
- (const char *const *)res.header_values,
- res.num_headers);
-
- if (!r->response_blocked) {
- if (r->flush && r->flush_header)
- r->flush(r);
- }
- }
- return JK_AJP13_SEND_HEADERS;
-
- case JK_AJP13_SEND_BODY_CHUNK:
- if (ae->last_op == JK_AJP13_FORWARD_REQUEST) {
- /* AJP13_SEND_BODY_CHUNK with length 0 is
- * explicit flush packet message.
- * Ignore those if they are left over from previous responses.
- * Reportedly some versions of JBoss suffer from that problem.
- */
- if (jk_b_get_int(msg) == 0) {
- jk_log(l, JK_LOG_DEBUG,
- "Ignoring flush message received while sending the request");
- return ae->last_op;
- }
- /* We have just send a request but received something
- * that probably originates from buffered response.
- */
- if (JK_IS_DEBUG_LEVEL(l)) {
- jk_log(l, JK_LOG_DEBUG,
- "Unexpected AJP13_SEND_BODY_CHUNK");
- }
- JK_TRACE_EXIT(l);
- return JK_AJP13_ERROR;
- }
- if (!r->response_blocked) {
- unsigned int len = (unsigned int)jk_b_get_int(msg);
- /*
- * Do a sanity check on len to prevent write reading beyond buffer
- * boundaries and thus revealing possible sensitive memory
- * contents to the client.
- * len cannot be larger than msg->len - 3 because the ajp message
- * contains the magic byte for JK_AJP13_SEND_BODY_CHUNK (1 byte)
- * and the length of the chunk (2 bytes). The remaining part of
- * the message is the chunk.
- */
- if (len > (unsigned int)(msg->len - 3)) {
- jk_log(l, JK_LOG_ERROR,
- "Chunk length too large. Length of AJP message is %i,"
- " chunk length is %i.", msg->len, len);
- JK_TRACE_EXIT(l);
- return JK_INTERNAL_ERROR;
- }
- if (len == 0) {
- /* AJP13_SEND_BODY_CHUNK with length 0 is
- * explicit flush packet message.
- */
- if (r->response_started) {
- if (r->flush) {
- r->flush(r);
- }
- }
- else {
- jk_log(l, JK_LOG_DEBUG,
- "Ignoring flush message received before headers");
- }
- }
- else {
- if (!r->write(r, msg->buf + msg->pos, len)) {
- jk_log(l, JK_LOG_INFO,
- "Writing to client aborted or client network problems");
- JK_TRACE_EXIT(l);
- return JK_CLIENT_WR_ERROR;
- }
- if (r->flush && r->flush_packets)
- r->flush(r);
- }
- }
- break;
-
- case JK_AJP13_GET_BODY_CHUNK:
- {
- int len = (int)jk_b_get_int(msg);
-
- if (len < 0) {
- len = 0;
- }
- if (len > AJP13_MAX_SEND_BODY_SZ) {
- len = AJP13_MAX_SEND_BODY_SZ;
- }
- if ((jk_uint64_t)len > ae->left_bytes_to_send) {
- len = (int)ae->left_bytes_to_send;
- }
-
- /* the right place to add file storage for upload */
- if ((len = ajp_read_into_msg_buff(ae, r, pmsg, len, l)) >= 0) {
- r->content_read += (jk_uint64_t)len;
- JK_TRACE_EXIT(l);
- return JK_AJP13_HAS_RESPONSE;
- }
-
- jk_log(l, JK_LOG_INFO,
- "Reading from client aborted or client network problems");
-
- JK_TRACE_EXIT(l);
- return JK_CLIENT_RD_ERROR;
- }
- break;
-
- case JK_AJP13_END_RESPONSE:
- ae->reuse = (int)jk_b_get_byte(msg);
- if (!ae->reuse) {
- /*
- * AJP13 protocol reuse flag set to false.
- * Tomcat will close its side of the connection.
- */
- jk_log(l, JK_LOG_WARNING, "AJP13 protocol: Reuse is set to false");
- }
- else if (r->disable_reuse) {
- if (JK_IS_DEBUG_LEVEL(l)) {
- jk_log(l, JK_LOG_DEBUG, "AJP13 protocol: Reuse is disabled");
- }
- ae->reuse = JK_FALSE;
- }
- else {
- /* Reuse in all cases */
- if (JK_IS_DEBUG_LEVEL(l)) {
- jk_log(l, JK_LOG_DEBUG, "AJP13 protocol: Reuse is OK");
- }
- ae->reuse = JK_TRUE;
- }
- if (!r->response_blocked) {
- if (r->done) {
- /* Done with response */
- r->done(r);
- }
- else if (r->flush && !r->flush_packets) {
- /* Flush after the last write */
- r->flush(r);
- }
- }
- JK_TRACE_EXIT(l);
- return JK_AJP13_END_RESPONSE;
- break;
-
- default:
- jk_log(l, JK_LOG_ERROR,
- "Unknown AJP protocol code: %02X", code);
- JK_TRACE_EXIT(l);
- return JK_AJP13_ERROR;
- }
-
- JK_TRACE_EXIT(l);
- return JK_AJP13_NO_RESPONSE;
-}
-
-/*
- * get replies from Tomcat via Ajp13/Ajp14
- * ajp13/ajp14 is async but handling read/send this way prevent nice recovery
- * In fact if tomcat link is broken during upload (browser -> apache -> tomcat)
- * we'll loose data and we'll have to abort the whole request.
- *
- * Return values of ajp_get_reply() function:
- * return value op->recoverable reason
- * JK_REPLY_TIMEOUT ?recovery_options Reply timeout while waiting for response packet
- * JK_FALSE ?recovery_options Error during ajp_connection_tcp_get_message()
- * Could not read the AJP packet header
- * JK_AJP_PROTOCOL_ERROR: ?recovery_options Error during ajp_connection_tcp_get_message()
- * Failure after reading the AJP packet header
- * JK_STATUS_ERROR mostly JK_TRUE ajp_process_callback() returns JK_STATUS_ERROR
- * Recoverable, if callback didn't return with a JK_HAS_RESPONSE before.
- * JK_HAS_RESPONSE: parts of the post buffer are consumed.
- * JK_STATUS_FATAL_ERROR mostly JK_TRUE ajp_process_callback() returns JK_STATUS_FATAL_ERROR
- * Recoverable, if callback didn't return with a JK_HAS_RESPONSE before.
- * JK_HAS_RESPONSE: parts of the post buffer are consumed.
- * JK_FATAL_ERROR ? ajp_process_callback() returns JK_AJP13_ERROR
- * JK_AJP13_ERROR: protocol error, or JK_INTERNAL_ERROR: chunk size to large
- * JK_CLIENT_RD_ERROR ? ajp_process_callback() returns JK_CLIENT_RD_ERROR
- * JK_CLIENT_RD_ERROR: could not read post from client.
- * JK_CLIENT_WR_ERROR ? ajp_process_callback() returns JK_CLIENT_WR_ERROR
- * JK_CLIENT_WR_ERROR: could not write back result to client
- * JK_TRUE ? ajp_process_callback() returns JK_AJP13_END_RESPONSE
- * JK_FALSE ? Other unhandled cases (unknown return codes)
- */
-static int ajp_get_reply(jk_endpoint_t *e,
- jk_ws_service_t *s,
- jk_logger_t *l,
- ajp_endpoint_t * p, ajp_operation_t * op)
-{
- /* Don't get header from tomcat yet */
- int headeratclient = JK_FALSE;
-
- JK_TRACE_ENTER(l);
-
- p->last_errno = 0;
- /* Start read all reply message */
- while (1) {
- int rc = 0;
- /* Allow to overwrite reply_timeout on a per URL basis via service struct */
- int reply_timeout = s->extension.reply_timeout;
- if (reply_timeout < 0)
- reply_timeout = p->worker->reply_timeout;
- /* If we set a reply timeout, check if something is available */
- if (reply_timeout > 0) {
- if (jk_is_input_event(p->sd, reply_timeout, l) ==
- JK_FALSE) {
- p->last_errno = errno;
- jk_log(l, JK_LOG_ERROR,
- "(%s) Timeout with waiting reply from tomcat. "
- "Tomcat is down, stopped or network problems (errno=%d)",
- p->worker->name, p->last_errno);
- /* We can't trust this connection any more. */
- ajp_abort_endpoint(p, JK_TRUE, l);
- if (headeratclient == JK_FALSE) {
- if (p->worker->recovery_opts & RECOVER_ABORT_IF_TCGETREQUEST)
- op->recoverable = JK_FALSE;
- /*
- * We revert back to recoverable, if recovery_opts allow it for GET or HEAD
- */
- if (op->recoverable == JK_FALSE) {
- if (p->worker->recovery_opts & RECOVER_ALWAYS_HTTP_HEAD) {
- if (!strcmp(s->method, "HEAD"))
- op->recoverable = JK_TRUE;
- }
- else if (p->worker->recovery_opts & RECOVER_ALWAYS_HTTP_GET) {
- if (!strcmp(s->method, "GET"))
- op->recoverable = JK_TRUE;
- }
- }
- }
- else {
- if (p->worker->recovery_opts & RECOVER_ABORT_IF_TCSENDHEADER)
- op->recoverable = JK_FALSE;
- }
-
- JK_TRACE_EXIT(l);
- return JK_REPLY_TIMEOUT;
- }
- }
-
- if ((rc = ajp_connection_tcp_get_message(p, op->reply, l)) != JK_TRUE) {
- if (headeratclient == JK_FALSE) {
- jk_log(l, JK_LOG_ERROR,
- "(%s) Tomcat is down or refused connection. "
- "No response has been sent to the client (yet)",
- p->worker->name);
- /*
- * communication with tomcat has been interrupted BEFORE
- * headers have been sent to the client.
- */
-
- /*
- * We mark it unrecoverable if recovery_opts set to RECOVER_ABORT_IF_TCGETREQUEST
- */
- if (p->worker->recovery_opts & RECOVER_ABORT_IF_TCGETREQUEST)
- op->recoverable = JK_FALSE;
- /*
- * We revert back to recoverable, if recovery_opts allow it for GET or HEAD
- */
- if (op->recoverable == JK_FALSE) {
- if (p->worker->recovery_opts & RECOVER_ALWAYS_HTTP_HEAD) {
- if (!strcmp(s->method, "HEAD"))
- op->recoverable = JK_TRUE;
- }
- else if (p->worker->recovery_opts & RECOVER_ALWAYS_HTTP_GET) {
- if (!strcmp(s->method, "GET"))
- op->recoverable = JK_TRUE;
- }
- }
-
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "(%s) Tomcat is down or network problems. "
- "Part of the response has already been sent to the client",
- p->worker->name);
-
- /* communication with tomcat has been interrupted AFTER
- * headers have been sent to the client.
- * headers (and maybe parts of the body) have already been
- * sent, therefore the response is "complete" in a sense
- * that nobody should append any data, especially no 500 error
- * page of the webserver!
- */
-
- /*
- * We mark it unrecoverable if recovery_opts set to RECOVER_ABORT_IF_TCSENDHEADER
- */
- if (p->worker->recovery_opts & RECOVER_ABORT_IF_TCSENDHEADER)
- op->recoverable = JK_FALSE;
-
- }
-
- JK_TRACE_EXIT(l);
- return rc;
- }
-
- rc = ajp_process_callback(op->reply, op->post, p, s, l);
- p->last_op = rc;
- /* no more data to be sent, fine we have finish here */
- if (JK_AJP13_END_RESPONSE == rc) {
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- else if (JK_AJP13_SEND_HEADERS == rc) {
- if (headeratclient == JK_FALSE)
- headeratclient = JK_TRUE;
- else {
- /* Backend send headers twice?
- * This is protocol violation
- */
- jk_log(l, JK_LOG_ERROR,
- "(%s) Tomcat already send headers",
- p->worker->name);
- op->recoverable = JK_FALSE;
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- else if (JK_STATUS_ERROR == rc || JK_STATUS_FATAL_ERROR == rc) {
- jk_log(l, JK_LOG_INFO,
- "(%s) request failed%s, "
- "because of response status %d, ",
- p->worker->name,
- rc == JK_STATUS_FATAL_ERROR ? "" : " (soft)",
- s->http_response_status);
- JK_TRACE_EXIT(l);
- return rc;
- }
- else if (JK_AJP13_HAS_RESPONSE == rc) {
- /*
- * in upload-mode there is no second chance since
- * we may have already sent part of the uploaded data
- * to Tomcat.
- * In this case if Tomcat connection is broken we must
- * abort request and indicate error.
- * A possible work-around could be to store the uploaded
- * data to file and replay for it
- */
- op->recoverable = JK_FALSE;
- rc = ajp_connection_tcp_send_message(p, op->post, l);
- if (rc < 0) {
- jk_log(l, JK_LOG_ERROR,
- "(%s) Tomcat is down or network problems",
- p->worker->name);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- else if (JK_AJP13_ERROR == rc) {
- /*
- * Tomcat has send invalid AJP message.
- * Loadbalancer if present will decide if
- * failover is possible.
- */
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- else if (JK_CLIENT_RD_ERROR == rc) {
- /*
- * Client has stop sending to us, so get out.
- * We assume this isn't our fault, so just a normal exit.
- */
- JK_TRACE_EXIT(l);
- return JK_CLIENT_RD_ERROR;
- }
- else if (JK_CLIENT_WR_ERROR == rc) {
- /*
- * Client has stop receiving to us, so get out.
- * We assume this isn't our fault, so just a normal exit.
- */
- JK_TRACE_EXIT(l);
- return JK_CLIENT_WR_ERROR;
- }
- else if (JK_INTERNAL_ERROR == rc) {
- /*
- * Internal error, like memory allocation or invalid packet lengths.
- */
- JK_TRACE_EXIT(l);
- return JK_FATAL_ERROR;
- }
- else if (JK_AJP13_NO_RESPONSE == rc) {
- /*
- * This is fine, loop again, more data to send.
- */
- continue;
- }
- else if (rc < 0) {
- op->recoverable = JK_FALSE;
- jk_log(l, JK_LOG_ERROR,
- "(%s) Callback returns with unknown value %d",
- p->worker->name, rc);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- /* XXX: Not reached? */
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-static void ajp_update_stats(jk_endpoint_t *e, ajp_worker_t *aw, int rc, jk_logger_t *l)
-{
- aw->s->readed += e->rd;
- aw->s->transferred += e->wr;
- if (aw->s->busy)
- aw->s->busy--;
- if (rc == JK_TRUE) {
- aw->s->state = JK_AJP_STATE_OK;
- }
- else if (rc == JK_CLIENT_ERROR) {
- aw->s->state = JK_AJP_STATE_OK;
- aw->s->client_errors++;
- }
- else {
- aw->s->state = JK_AJP_STATE_ERROR;
- aw->s->errors++;
- aw->s->error_time = time(NULL);
- }
-}
-
-/*
- * service is now splitted in ajp_send_request and ajp_get_reply
- * much more easier to do errors recovery
- *
- * We serve here the request, using AJP13/AJP14 (e->proto)
- *
- * Return values of service() method for ajp13/ajp14 worker:
- * return value is_error e->recoverable reason
- * JK_FALSE JK_HTTP_SERVER_ERROR TRUE Invalid Parameters (null values)
- * JK_SERVER_ERROR JK_HTTP_SERVER_ERROR TRUE Error during initializing empty request, response or post body objects
- * JK_CLIENT_ERROR JK_HTTP_REQUEST_TOO_LARGE JK_TRUE Request doesn't fit into buffer (error during ajp_marshal_into_msgb())
- * JK_CLIENT_ERROR JK_HTTP_BAD_REQUEST JK_FALSE Error during reading parts of POST body from client
- * JK_FATAL_ERROR JK_HTTP_SERVER_ERROR JK_FALSE If ajp_send_request() returns JK_FATAL_ERROR and !op->recoverable.
- * JK_FATAL_ERROR JK_HTTP_SERVER_ERROR JK_FALSE If ajp_send_request() returns JK_TRUE but !op->recoverable.
- * This should never happen.
- * JK_CLIENT_ERROR JK_HTTP_BAD_REQUEST ? ajp_get_reply() returns JK_CLIENT_RD_ERROR
- * JK_CLIENT_ERROR JK_HTTP_OK ? ajp_get_reply() returns JK_CLIENT_WR_ERROR
- * JK_FATAL_ERROR JK_HTTP_SERVER_ERROR JK_TRUE ajp_get_reply() returns JK_FATAL_ERROR
- * JK_FATAL_ERROR: protocol error or internal error
- * JK_FATAL_ERROR JK_HTTP_SERVER_ERROR JK_FALSE ajp_get_reply() returns JK_FATAL_ERROR
- * JK_FATAL_ERROR: protocol error or internal error
- * JK_STATUS_ERROR JK_HTTP_SERVER_BUSY JK_TRUE ajp_get_reply() returns JK_STATUS_ERROR
- * Only if op->recoverable and no more ajp13/ajp14 direct retries
- * JK_STATUS_ERROR JK_HTTP_SERVER_BUSY JK_FALSE ajp_get_reply() returns JK_STATUS_ERROR
- * Only if !op->recoverable
- * JK_STATUS_FATAL_ERROR JK_HTTP_SERVER_BUSY JK_TRUE ajp_get_reply() returns JK_STATUS_ERROR
- * Only if op->recoverable and no more ajp13/ajp14 direct retries
- * JK_STATUS_FATAL_ERROR JK_HTTP_SERVER_BUSY JK_FALSE ajp_get_reply() returns JK_STATUS_FATAL_ERROR
- * Only if !op->recoverable
- * JK_REPLY_TIMEOUT JK_HTTP_GATEWAY_TIME_OUT JK_TRUE ajp_get_reply() returns JK_REPLY_TIMEOUT
- * JK_AJP_PROTOCOL_ERROR JK_HTTP_GATEWAY_TIME_OUT ? ajp_get_reply() returns JK_AJP_PROTOCOL_ERROR
- * ??? JK_FATAL_ERROR JK_HTTP_GATEWAY_TIME_OUT JK_FALSE ajp_get_reply() returns something else
- * Only if !op->recoverable
- * ??? JK_FALSE JK_HTTP_SERVER_BUSY JK_TRUE ajp_get_reply() returns JK_FALSE
- * Only if op->recoverable and no more ajp13/ajp14 direct retries
- * JK_TRUE JK_HTTP_OK ? OK
- */
-static int JK_METHOD ajp_service(jk_endpoint_t *e,
- jk_ws_service_t *s,
- jk_logger_t *l, int *is_error)
-{
- int i;
- int err = JK_TRUE;
- ajp_operation_t oper;
- ajp_operation_t *op = &oper;
- ajp_endpoint_t *p;
- ajp_worker_t *aw;
- int log_error;
- int rc = JK_UNSET;
- char *msg = "";
- int retry_interval;
-
- JK_TRACE_ENTER(l);
-
- if (!e || !e->endpoint_private || !s || !is_error) {
- JK_LOG_NULL_PARAMS(l);
- if (is_error)
- *is_error = JK_HTTP_SERVER_ERROR;
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- p = e->endpoint_private;
- aw = p->worker;
-
- if (aw->sequence != aw->s->h.sequence)
- jk_ajp_pull(aw, JK_FALSE, l);
-
- aw->s->used++;
-
- /* Set returned error to SERVER ERROR */
- *is_error = JK_HTTP_SERVER_ERROR;
-
- op->request = jk_b_new(&(p->pool));
- if (!op->request) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message");
- JK_TRACE_EXIT(l);
- return JK_SERVER_ERROR;
- }
- if (jk_b_set_buffer_size(op->request, aw->max_packet_size)) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message buffer");
- JK_TRACE_EXIT(l);
- return JK_SERVER_ERROR;
- }
- jk_b_reset(op->request);
-
- op->reply = jk_b_new(&(p->pool));
- if (!op->reply) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message");
- JK_TRACE_EXIT(l);
- return JK_SERVER_ERROR;
- }
- if (jk_b_set_buffer_size(op->reply, aw->max_packet_size)) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message buffer");
- JK_TRACE_EXIT(l);
- return JK_SERVER_ERROR;
- }
-
- op->post = jk_b_new(&(p->pool));
- if (!op->post) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message");
- JK_TRACE_EXIT(l);
- return JK_SERVER_ERROR;
- }
- if (jk_b_set_buffer_size(op->post, aw->max_packet_size)) {
- jk_log(l, JK_LOG_ERROR,
- "Failed allocating AJP message buffer");
- JK_TRACE_EXIT(l);
- return JK_SERVER_ERROR;
- }
- jk_b_reset(op->post);
-
- /* Set returned error to OK */
- *is_error = JK_HTTP_OK;
-
- op->recoverable = JK_TRUE;
- op->uploadfd = -1; /* not yet used, later ;) */
-
- p->left_bytes_to_send = s->content_length;
- p->reuse = JK_FALSE;
- p->hard_close = JK_FALSE;
-
- s->secret = aw->secret;
-
- /*
- * We get here initial request (in op->request)
- */
- if (!ajp_marshal_into_msgb(op->request, s, l, p)) {
- *is_error = JK_HTTP_REQUEST_TOO_LARGE;
- jk_log(l, JK_LOG_INFO,
- "Creating AJP message failed, "
- "without recovery");
- aw->s->client_errors++;
- JK_TRACE_EXIT(l);
- return JK_CLIENT_ERROR;
- }
-
- if (JK_IS_DEBUG_LEVEL(l)) {
- jk_log(l, JK_LOG_DEBUG, "processing %s with %d retries",
- aw->name, aw->retries);
- }
- aw->s->busy++;
- if (aw->s->state == JK_AJP_STATE_ERROR)
- aw->s->state = JK_AJP_STATE_PROBE;
- if (aw->s->busy > aw->s->max_busy)
- aw->s->max_busy = aw->s->busy;
- retry_interval = p->worker->retry_interval;
- for (i = 0; i < aw->retries; i++) {
- /* Reset reply message buffer for each retry */
- jk_b_reset(op->reply);
-
- /*
- * ajp_send_request() already locally handles
- * reconnecting and broken connection detection.
- * So if we already failed in it, wait a bit before
- * retrying the same backend.
- */
- if (i > 0 && retry_interval >= 0) {
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "retry %d, sleeping for %d ms before retrying",
- i, retry_interval);
- jk_sleep(retry_interval);
- /* Pull shared memory if something changed during sleep */
- if (aw->sequence != aw->s->h.sequence)
- jk_ajp_pull(aw, JK_FALSE, l);
- }
- /*
- * We're using op->request which hold initial request
- * if Tomcat is stopped or restarted, we will pass op->request
- * to next valid tomcat.
- */
- log_error = JK_TRUE;
- rc = JK_UNSET;
- msg = "";
- err = ajp_send_request(e, s, l, p, op);
- e->recoverable = op->recoverable;
- if (err == JK_CLIENT_RD_ERROR) {
- *is_error = JK_HTTP_BAD_REQUEST;
- msg = "because of client read error";
- aw->s->client_errors++;
- rc = JK_CLIENT_ERROR;
- log_error = JK_FALSE;
- e->recoverable = JK_FALSE;
- /* Ajp message set reuse to TRUE in END_REQUEST message
- * However due to client bad request if the recovery
- * RECOVER_ABORT_IF_CLIENTERROR is set the physical connection
- * will be closed and application in Tomcat can catch that
- * generated exception, knowing the client aborted the
- * connection. This AJP protocol limitation, where we
- * should actually send some packet informing the backend
- * that client broke the connection in a middle of
- * request/response cycle.
- */
- if (aw->recovery_opts & RECOVER_ABORT_IF_CLIENTERROR) {
- /* Mark the endpoint for shutdown */
- p->reuse = JK_FALSE;
- p->hard_close = JK_TRUE;
- }
- }
- else if (err == JK_FATAL_ERROR) {
- *is_error = JK_HTTP_SERVER_BUSY;
- msg = "because of error during request sending";
- rc = err;
- if (!op->recoverable) {
- *is_error = JK_HTTP_SERVER_ERROR;
- msg = "because of protocol error during request sending";
- }
- }
- else if (err == JK_TRUE && op->recoverable) {
- /* Up to there we can recover */
-
- err = ajp_get_reply(e, s, l, p, op);
- e->recoverable = op->recoverable;
- if (err == JK_TRUE) {
- *is_error = JK_HTTP_OK;
- /* Done with the request */
- ajp_update_stats(e, aw, JK_TRUE, l);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
-
- if (err == JK_CLIENT_RD_ERROR) {
- *is_error = JK_HTTP_BAD_REQUEST;
- msg = "because of client read error";
- aw->s->client_errors++;
- rc = JK_CLIENT_ERROR;
- log_error = JK_FALSE;
- e->recoverable = JK_FALSE;
- op->recoverable = JK_FALSE;
- if (aw->recovery_opts & RECOVER_ABORT_IF_CLIENTERROR) {
- /* Mark the endpoint for shutdown */
- p->reuse = JK_FALSE;
- p->hard_close = JK_TRUE;
- }
- }
- else if (err == JK_CLIENT_WR_ERROR) {
- /* XXX: Is this correct to log this as 200? */
- *is_error = JK_HTTP_OK;
- msg = "because of client write error";
- aw->s->client_errors++;
- rc = JK_CLIENT_ERROR;
- log_error = JK_FALSE;
- e->recoverable = JK_FALSE;
- op->recoverable = JK_FALSE;
- if (aw->recovery_opts & RECOVER_ABORT_IF_CLIENTERROR) {
- /* Mark the endpoint for shutdown */
- p->reuse = JK_FALSE;
- p->hard_close = JK_TRUE;
- }
- }
- else if (err == JK_FATAL_ERROR) {
- *is_error = JK_HTTP_SERVER_ERROR;
- msg = "because of server error";
- rc = err;
- }
- else if (err == JK_REPLY_TIMEOUT) {
- *is_error = JK_HTTP_GATEWAY_TIME_OUT;
- msg = "because of reply timeout";
- aw->s->reply_timeouts++;
- rc = err;
- }
- else if (err == JK_STATUS_ERROR || err == JK_STATUS_FATAL_ERROR) {
- *is_error = JK_HTTP_SERVER_BUSY;
- msg = "because of response status";
- rc = err;
- }
- else if (err == JK_AJP_PROTOCOL_ERROR) {
- *is_error = JK_HTTP_BAD_GATEWAY;
- msg = "because of protocol error";
- rc = err;
- }
- /* This should only be the cases err == JK_FALSE */
- else {
- /* if we can't get reply, check if unrecoverable flag was set
- * if is_recoverable_error is cleared, we have started
- * receiving upload data and we must consider that
- * operation is no more recoverable
- */
- *is_error = JK_HTTP_BAD_GATEWAY;
- msg = "";
- rc = JK_FALSE;
- }
- }
- else {
- /* XXX: this should never happen:
- * ajp_send_request() never returns JK_TRUE if !op->recoverable.
- * and all other return values have already been handled.
- */
- e->recoverable = JK_FALSE;
- *is_error = JK_HTTP_SERVER_ERROR;
- msg = "because of an unknown reason";
- rc = JK_FATAL_ERROR;
- jk_log(l, JK_LOG_ERROR,
- "(%s) unexpected condition err=%d recoverable=%d",
- aw->name, err, op->recoverable);
- }
- if (!op->recoverable && log_error == JK_TRUE) {
- jk_log(l, JK_LOG_ERROR,
- "(%s) sending request to tomcat failed (unrecoverable), "
- "%s "
- "(attempt=%d)",
- aw->name, msg, i + 1);
- }
- else {
- jk_log(l, JK_LOG_INFO,
- "(%s) sending request to tomcat failed (%srecoverable), "
- "%s "
- "(attempt=%d)",
- aw->name,
- op->recoverable ? "" : "un",
- msg, i + 1);
- }
- if (!op->recoverable) {
- ajp_update_stats(e, aw, rc, l);
- JK_TRACE_EXIT(l);
- return rc;
- }
- /* Get another connection from the pool and try again.
- * Note: All sockets are probably closed already.
- */
- ajp_next_connection(p, l);
- }
- /* Log the error only once per failed request. */
- jk_log(l, JK_LOG_ERROR,
- "(%s) connecting to tomcat failed.",
- aw->name);
-
- ajp_update_stats(e, aw, rc, l);
- JK_TRACE_EXIT(l);
- return rc;
-}
-
-/*
- * Validate the worker (ajp13/ajp14)
- */
-
-int ajp_validate(jk_worker_t *pThis,
- jk_map_t *props,
- jk_worker_env_t *we, jk_logger_t *l, int proto)
-{
- int port;
- const char *host;
-
- JK_TRACE_ENTER(l);
-
- if (proto == AJP13_PROTO) {
- port = AJP13_DEF_PORT;
- host = AJP13_DEF_HOST;
- }
- else if (proto == AJP14_PROTO) {
- port = AJP14_DEF_PORT;
- host = AJP14_DEF_HOST;
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "unknown protocol %d", proto);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- if (pThis && pThis->worker_private) {
- ajp_worker_t *p = pThis->worker_private;
- p->port = jk_get_worker_port(props, p->name, port);
- if (!host) {
- host = "undefined";
- }
- strncpy(p->host, jk_get_worker_host(props, p->name, host), JK_SHM_STR_SIZ);
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "worker %s contact is '%s:%d'",
- p->name, p->host, p->port);
- /* Copy the contact to shm */
- strncpy(p->s->host, p->host, JK_SHM_STR_SIZ);
- p->s->port = p->port;
- p->s->addr_sequence = p->addr_sequence = 0;
- /* Resolve if port > 0.
- */
- if (p->port > 0) {
- if (jk_resolve(p->host, p->port, &p->worker_inet_addr, we->pool, l)) {
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- jk_log(l, JK_LOG_ERROR,
- "worker %s can't resolve tomcat address %s",
- p->name, p->host);
- p->s->port = p->port = 0;
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "worker %s contact is disabled",
- p->name);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- else {
- p->s->port = p->port = 0;
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "worker %s contact is disabled",
- p->name);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- }
- else {
- JK_LOG_NULL_PARAMS(l);
- }
-
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-static int ajp_create_endpoint_cache(ajp_worker_t *p, int proto, jk_logger_t *l)
-{
- unsigned int i;
- time_t now = time(NULL);
-
- JK_TRACE_ENTER(l);
-
- p->ep_cache = (ajp_endpoint_t **)calloc(1, sizeof(ajp_endpoint_t *) *
- p->ep_cache_sz);
- if (!p->ep_cache) {
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "setting connection pool size to %u with min %u and acquire timeout %d",
- p->ep_cache_sz, p->ep_mincache_sz, p->cache_acquire_timeout);
- for (i = 0; i < p->ep_cache_sz; i++) {
- p->ep_cache[i] = (ajp_endpoint_t *)calloc(1, sizeof(ajp_endpoint_t));
- if (!p->ep_cache[i]) {
- jk_log(l, JK_LOG_ERROR,
- "allocating endpoint slot %d (errno=%d)",
- i, errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- p->ep_cache[i]->sd = JK_INVALID_SOCKET;
- p->ep_cache[i]->reuse = JK_FALSE;
- p->ep_cache[i]->hard_close = JK_FALSE;
- p->ep_cache[i]->last_access = now;
- jk_open_pool(&(p->ep_cache[i]->pool), p->ep_cache[i]->buf,
- sizeof(p->ep_cache[i]->buf));
- p->ep_cache[i]->worker = p;
- p->ep_cache[i]->endpoint.endpoint_private = p->ep_cache[i];
- p->ep_cache[i]->proto = proto;
- p->ep_cache[i]->endpoint.service = ajp_service;
- p->ep_cache[i]->endpoint.done = ajp_done;
- p->ep_cache[i]->last_op = JK_AJP13_END_RESPONSE;
- p->ep_cache[i]->addr_sequence = 0;
- }
-
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-int ajp_init(jk_worker_t *pThis,
- jk_map_t *props, jk_worker_env_t *we, jk_logger_t *l, int proto)
-{
- int rc = JK_FALSE;
- int cache;
- /*
- * start the connection cache
- */
- JK_TRACE_ENTER(l);
-
- cache = jk_get_worker_def_cache_size(proto);
-
- if (pThis && pThis->worker_private) {
- ajp_worker_t *p = pThis->worker_private;
- p->worker.we = we;
- p->ep_cache_sz = jk_get_worker_cache_size(props, p->name, cache);
- p->ep_mincache_sz = jk_get_worker_cache_size_min(props, p->name,
- (p->ep_cache_sz+1) / 2);
- p->socket_timeout =
- jk_get_worker_socket_timeout(props, p->name, AJP_DEF_SOCKET_TIMEOUT);
-
- p->socket_connect_timeout =
- jk_get_worker_socket_connect_timeout(props, p->name,
- p->socket_timeout * 1000);
-
- p->keepalive =
- jk_get_worker_socket_keepalive(props, p->name, JK_FALSE);
-
- p->cache_timeout =
- jk_get_worker_cache_timeout(props, p->name,
- AJP_DEF_CACHE_TIMEOUT);
-
- p->ping_timeout =
- jk_get_worker_ping_timeout(props, p->name,
- AJP_DEF_PING_TIMEOUT);
- p->ping_mode =
- jk_get_worker_ping_mode(props, p->name,
- AJP_CPING_NONE);
-
- p->connect_timeout =
- jk_get_worker_connect_timeout(props, p->name,
- AJP_DEF_CONNECT_TIMEOUT);
-
- p->prepost_timeout =
- jk_get_worker_prepost_timeout(props, p->name,
- AJP_DEF_PREPOST_TIMEOUT);
-
- if ((p->ping_mode & AJP_CPING_CONNECT) &&
- p->connect_timeout == AJP_DEF_CONNECT_TIMEOUT)
- p->connect_timeout = p->ping_timeout;
-
- if ((p->ping_mode & AJP_CPING_PREPOST) &&
- p->prepost_timeout == AJP_DEF_PREPOST_TIMEOUT)
- p->prepost_timeout = p->ping_timeout;
-
- p->conn_ping_interval =
- jk_get_worker_conn_ping_interval(props, p->name, 0);
- if ((p->ping_mode & AJP_CPING_INTERVAL) &&
- p->conn_ping_interval == 0) {
- /* XXX: Ping timeout is in miliseconds
- * and ping_interval is in seconds.
- * Use 10 times larger value for ping interval
- * (ping_timeout / 1000) * 10
- */
- p->conn_ping_interval = p->ping_timeout / 100;
- }
- p->reply_timeout =
- jk_get_worker_reply_timeout(props, p->name,
- AJP_DEF_REPLY_TIMEOUT);
-
- p->recovery_opts =
- jk_get_worker_recovery_opts(props, p->name,
- AJP_DEF_RECOVERY_OPTS);
-
- p->retries =
- jk_get_worker_retries(props, p->name,
- JK_RETRIES);
-
- p->max_packet_size =
- jk_get_max_packet_size(props, p->name);
-
- p->socket_buf =
- jk_get_worker_socket_buffer(props, p->name, p->max_packet_size);
-
- p->retry_interval =
- jk_get_worker_retry_interval(props, p->name,
- JK_SLEEP_DEF);
- p->cache_acquire_timeout = jk_get_worker_cache_acquire_timeout(props,
- p->name, p->retries * p->retry_interval);
- p->http_status_fail_num = jk_get_worker_fail_on_status(props, p->name,
- &p->http_status_fail[0],
- JK_MAX_HTTP_STATUS_FAILS);
-
- if (p->retries < 1) {
- jk_log(l, JK_LOG_INFO,
- "number of retries must be greater then 1. Setting to default=%d",
- JK_RETRIES);
- p->retries = JK_RETRIES;
- }
-
- p->maintain_time = jk_get_worker_maintain_time(props);
- if(p->maintain_time < 0)
- p->maintain_time = 0;
- p->s->last_maintain_time = time(NULL);
- p->s->last_reset = p->s->last_maintain_time;
-
- if (JK_IS_DEBUG_LEVEL(l)) {
-
- jk_log(l, JK_LOG_DEBUG,
- "setting endpoint options:",
- p->keepalive);
- jk_log(l, JK_LOG_DEBUG,
- "keepalive: %d",
- p->keepalive);
-
- jk_log(l, JK_LOG_DEBUG,
- "socket timeout: %d",
- p->socket_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "socket connect timeout: %d",
- p->socket_connect_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "buffer size: %d",
- p->socket_buf);
-
- jk_log(l, JK_LOG_DEBUG,
- "pool timeout: %d",
- p->cache_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "ping timeout: %d",
- p->ping_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "connect timeout: %d",
- p->connect_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "reply timeout: %d",
- p->reply_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "prepost timeout: %d",
- p->prepost_timeout);
-
- jk_log(l, JK_LOG_DEBUG,
- "recovery options: %d",
- p->recovery_opts);
-
- jk_log(l, JK_LOG_DEBUG,
- "retries: %d",
- p->retries);
-
- jk_log(l, JK_LOG_DEBUG,
- "max packet size: %d",
- p->max_packet_size);
-
- jk_log(l, JK_LOG_DEBUG,
- "retry interval: %d",
- p->retry_interval);
- }
- /*
- * Need to initialize secret here since we could return from inside
- * of the following loop
- */
-
- p->secret = jk_get_worker_secret(props, p->name);
- /* Initialize cache slots */
- JK_INIT_CS(&(p->cs), rc);
- if (!rc) {
- jk_log(l, JK_LOG_ERROR,
- "creating thread lock (errno=%d)",
- errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- if (!ajp_create_endpoint_cache(p, proto, l)) {
- jk_log(l, JK_LOG_ERROR,
- "allocating connection pool of size %u",
- p->ep_cache_sz);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- rc = JK_TRUE;
- }
- else {
- JK_LOG_NULL_PARAMS(l);
- }
-
- JK_TRACE_EXIT(l);
- return rc;
-}
-
-int JK_METHOD ajp_worker_factory(jk_worker_t **w,
- const char *name, jk_logger_t *l)
-{
- ajp_worker_t *aw;
-
- JK_TRACE_ENTER(l);
- if (name == NULL || w == NULL) {
- JK_LOG_NULL_PARAMS(l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- aw = (ajp_worker_t *) calloc(1, sizeof(ajp_worker_t));
- if (!aw) {
- jk_log(l, JK_LOG_ERROR,
- "malloc of private_data failed");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- jk_open_pool(&aw->p,
- aw->buf,
- sizeof(jk_pool_atom_t) * TINY_POOL_SIZE);
-
- strncpy(aw->name, name, JK_SHM_STR_SIZ);
- aw->login = NULL;
-
- aw->ep_cache_sz = 0;
- aw->ep_cache = NULL;
- aw->connect_retry_attempts = AJP_DEF_RETRY_ATTEMPTS;
- aw->worker.worker_private = aw;
-
- aw->worker.maintain = ajp_maintain;
-
- aw->logon = NULL;
-
- *w = &aw->worker;
-
- aw->s = jk_shm_alloc_ajp_worker(&aw->p);
- if (!aw->s) {
- jk_close_pool(&aw->p);
- free(aw);
- jk_log(l, JK_LOG_ERROR,
- "allocating ajp worker record from shared memory");
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- JK_TRACE_EXIT(l);
- return JK_TRUE;
-}
-
-int ajp_destroy(jk_worker_t **pThis, jk_logger_t *l, int proto)
-{
- JK_TRACE_ENTER(l);
-
- if (pThis && *pThis && (*pThis)->worker_private) {
- unsigned int i;
- ajp_worker_t *aw = (*pThis)->worker_private;
-
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "up to %u endpoints to close",
- aw->ep_cache_sz);
-
- for (i = 0; i < aw->ep_cache_sz; i++) {
- if (aw->ep_cache[i])
- ajp_close_endpoint(aw->ep_cache[i], l);
- }
- free(aw->ep_cache);
- JK_DELETE_CS(&(aw->cs), i);
-
- if (aw->login) {
- /* take care of removing previously allocated data */
- if (aw->login->servlet_engine_name)
- free(aw->login->servlet_engine_name);
-
- free(aw->login);
- aw->login = NULL;
- }
-
- jk_close_pool(&aw->p);
- free(aw);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
-
- JK_LOG_NULL_PARAMS(l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-int JK_METHOD ajp_done(jk_endpoint_t **e, jk_logger_t *l)
-{
- JK_TRACE_ENTER(l);
-
- if (e && *e && (*e)->endpoint_private) {
- ajp_endpoint_t *p = (*e)->endpoint_private;
- int rc;
- ajp_worker_t *w = p->worker;
-
- /* set last_access only if needed */
- if (w->cache_timeout > 0)
- p->last_access = time(NULL);
- if (w->s->addr_sequence != p->addr_sequence) {
- p->reuse = JK_FALSE;
- p->addr_sequence = w->s->addr_sequence;
- }
- ajp_reset_endpoint(p, l);
- *e = NULL;
- JK_ENTER_CS(&w->cs, rc);
- if (rc) {
- int i;
-
- for (i = w->ep_cache_sz - 1; i >= 0; i--) {
- if (w->ep_cache[i] == NULL) {
- w->ep_cache[i] = p;
- break;
- }
- }
- JK_LEAVE_CS(&w->cs, rc);
-
- if (i >= 0) {
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "recycling connection pool slot=%u for worker %s",
- i, p->worker->name);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- /* This should never hapen because
- * there is always free empty cache slot
- */
- jk_log(l, JK_LOG_ERROR,
- "could not find empty connection pool slot from %u for worker %s",
- w->ep_cache_sz, w->name);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- jk_log(l, JK_LOG_ERROR,
- "locking thread (errno=%d)", errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
-
- JK_LOG_NULL_PARAMS(l);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-int ajp_get_endpoint(jk_worker_t *pThis,
- jk_endpoint_t **je, jk_logger_t *l, int proto)
-{
- JK_TRACE_ENTER(l);
-
- if (pThis && pThis->worker_private && je) {
- ajp_worker_t *aw = pThis->worker_private;
- ajp_endpoint_t *ae = NULL;
- int rc;
- int retry = 0;
-
- *je = NULL;
- /* Loop until cache_acquire_timeout interval elapses */
- while ((retry * JK_SLEEP_DEF) < aw->cache_acquire_timeout) {
-
- JK_ENTER_CS(&aw->cs, rc);
- if (rc) {
- unsigned int slot;
- /* Try to find connected socket cache entry */
- for (slot = 0; slot < aw->ep_cache_sz; slot++) {
- if (aw->ep_cache[slot] &&
- IS_VALID_SOCKET(aw->ep_cache[slot]->sd)) {
- ae = aw->ep_cache[slot];
- if (ae->reuse) {
- aw->ep_cache[slot] = NULL;
- break;
- }
- else {
- /* XXX: We shouldn't have non reusable
- * opened socket in the cache
- */
- ajp_reset_endpoint(ae, l);
- ae = NULL;
- jk_log(l, JK_LOG_WARNING,
- "closing non reusable pool slot=%d", slot);
- }
- }
- }
- if (!ae) {
- /* No connected cache entry found.
- * Use the first free one.
- */
- for (slot = 0; slot < aw->ep_cache_sz; slot++) {
- if (aw->ep_cache[slot]) {
- ae = aw->ep_cache[slot];
- aw->ep_cache[slot] = NULL;
- break;
- }
- }
- }
- JK_LEAVE_CS(&aw->cs, rc);
- if (ae) {
- if (aw->cache_timeout > 0)
- ae->last_access = time(NULL);
- *je = &ae->endpoint;
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "acquired connection pool slot=%u after %d retries",
- slot, retry);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- else {
- retry++;
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "could not get free endpoint for worker %s"
- " (retry %d, sleeping for %d ms)",
- aw->name, retry, JK_SLEEP_DEF);
- jk_sleep(JK_SLEEP_DEF);
- }
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "locking thread (errno=%d)", errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-
- }
- }
- jk_log(l, JK_LOG_WARNING,
- "Unable to get the free endpoint for worker %s from %u slots",
- aw->name, aw->ep_cache_sz);
- }
- else {
- JK_LOG_NULL_PARAMS(l);
- }
-
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-int JK_METHOD ajp_maintain(jk_worker_t *pThis, time_t mstarted, jk_logger_t *l)
-{
- JK_TRACE_ENTER(l);
-
- if (pThis && pThis->worker_private) {
- ajp_worker_t *aw = pThis->worker_private;
- time_t now = mstarted;
- int rc;
- long delta;
-
- jk_shm_lock();
-
- /* Now we check for global maintenance (once for all processes).
- * Checking workers for idleness.
- * Therefore we globally sync and we use a global timestamp.
- * Since it's possible that we come here a few milliseconds
- * before the interval has passed, we allow a little tolerance.
- */
- delta = (long)difftime(mstarted, aw->s->last_maintain_time) + JK_AJP_MAINTAIN_TOLERANCE;
- if (delta >= aw->maintain_time) {
- aw->s->last_maintain_time = mstarted;
- if (aw->s->state == JK_AJP_STATE_OK &&
- aw->s->used == aw->s->used_snapshot)
- aw->s->state = JK_AJP_STATE_IDLE;
- aw->s->used_snapshot = aw->s->used;
- }
-
- jk_shm_unlock();
-
- /* Do connection pool maintenance only if timeouts or keepalives are set */
- if (aw->cache_timeout <= 0 &&
- aw->conn_ping_interval <= 0) {
- /* Nothing to do. */
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
-
- JK_ENTER_CS(&aw->cs, rc);
- if (rc) {
- unsigned int n = 0, k = 0, cnt = 0;
- int i;
- unsigned int m, m_count = 0;
- jk_sock_t *m_sock;
- /* Count open slots */
- for (i = (int)aw->ep_cache_sz - 1; i >= 0; i--) {
- if (aw->ep_cache[i] && IS_VALID_SOCKET(aw->ep_cache[i]->sd))
- cnt++;
- }
- m_sock = (jk_sock_t *)malloc((cnt + 1) * sizeof(jk_sock_t));
- /* Handle worker cache timeouts */
- if (aw->cache_timeout > 0) {
- for (i = (int)aw->ep_cache_sz - 1;
- i >= 0; i--) {
- /* Skip the closed sockets */
- if (aw->ep_cache[i] && IS_VALID_SOCKET(aw->ep_cache[i]->sd)) {
- int elapsed = (int)difftime(mstarted, aw->ep_cache[i]->last_access);
- if (elapsed > aw->cache_timeout) {
- time_t rt = 0;
- n++;
- if (JK_IS_DEBUG_LEVEL(l))
- rt = time(NULL);
- aw->ep_cache[i]->reuse = JK_FALSE;
- m_sock[m_count++] = aw->ep_cache[i]->sd;
- aw->ep_cache[i]->sd = JK_INVALID_SOCKET;
- ajp_reset_endpoint(aw->ep_cache[i], l);
- if (JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "cleaning pool slot=%d elapsed %d in %d",
- i, elapsed, (int)(difftime(time(NULL), rt)));
- }
- }
- if (cnt <= aw->ep_mincache_sz + n) {
- if (JK_IS_DEBUG_LEVEL(l)) {
- jk_log(l, JK_LOG_DEBUG,
- "reached pool min size %u from %u cache slots",
- aw->ep_mincache_sz, aw->ep_cache_sz);
- }
- break;
- }
- }
- }
- /* Handle worker connection keepalive */
- if (aw->conn_ping_interval > 0 && aw->ping_timeout > 0) {
- for (i = (int)aw->ep_cache_sz - 1; i >= 0; i--) {
- /* Skip the closed sockets */
- if (aw->ep_cache[i] && IS_VALID_SOCKET(aw->ep_cache[i]->sd)) {
- int elapsed = (int)difftime(now, aw->ep_cache[i]->last_access);
- if (elapsed > aw->conn_ping_interval) {
- k++;
- /* handle cping/cpong.
- */
- if (ajp_handle_cping_cpong(aw->ep_cache[i],
- aw->ping_timeout, l) == JK_FALSE) {
- jk_log(l, JK_LOG_INFO,
- "(%s) failed sending request, "
- "socket %d keepalive cping/cpong "
- "failure (errno=%d)",
- aw->name,
- aw->ep_cache[i]->sd,
- aw->ep_cache[i]->last_errno);
- aw->ep_cache[i]->reuse = JK_FALSE;
- m_sock[m_count++] = aw->ep_cache[i]->sd;
- aw->ep_cache[i]->sd = JK_INVALID_SOCKET;
- ajp_reset_endpoint(aw->ep_cache[i], l);
- }
- else {
- now = time(NULL);
- aw->ep_cache[i]->last_access = now;
- }
- }
- }
- }
- }
- JK_LEAVE_CS(&aw->cs, rc);
- /* Shutdown sockets outside of the lock.
- * This has benefits only if maintain was
- * called from the watchdog thread.
- */
- for (m = 0; m < m_count; m++) {
- jk_shutdown_socket(m_sock[m], l);
- }
- free(m_sock);
- if (n && JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "recycled %u sockets in %d seconds from %u pool slots",
- n, (int)(difftime(time(NULL), mstarted)),
- aw->ep_cache_sz);
- if (k && JK_IS_DEBUG_LEVEL(l))
- jk_log(l, JK_LOG_DEBUG,
- "pinged %u sockets in %d seconds from %u pool slots",
- k, (int)(difftime(time(NULL), mstarted)),
- aw->ep_cache_sz);
- JK_TRACE_EXIT(l);
- return JK_TRUE;
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "locking thread (errno=%d)",
- errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-
- }
- }
- else {
- JK_LOG_NULL_PARAMS(l);
- }
-
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
-
-int ajp_has_endpoint(jk_worker_t *pThis,
- jk_logger_t *l)
-{
- JK_TRACE_ENTER(l);
-
- if (pThis && pThis->worker_private) {
- ajp_worker_t *aw = pThis->worker_private;
- int rc;
-
- JK_ENTER_CS(&aw->cs, rc);
- if (rc) {
- unsigned int slot;
- /* Try to find connected socket cache entry */
- for (slot = 0; slot < aw->ep_cache_sz; slot++) {
- if (aw->ep_cache[slot]) {
- JK_LEAVE_CS(&aw->cs, rc);
- return JK_TRUE;
- }
- }
- JK_LEAVE_CS(&aw->cs, rc);
- }
- else {
- jk_log(l, JK_LOG_ERROR,
- "locking thread (errno=%d)", errno);
- JK_TRACE_EXIT(l);
- return JK_FALSE;
- }
- }
- else {
- JK_LOG_NULL_PARAMS(l);
- }
-
- JK_TRACE_EXIT(l);
- return JK_FALSE;
-}
Code Review / bottlenecks.git / blobdiff