| /* 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. |
| */ |
| |
| #include "ajp.h" |
| |
| APLOG_USE_MODULE(proxy_ajp); |
| |
| /* |
| * Handle the CPING/CPONG |
| */ |
| apr_status_t ajp_handle_cping_cpong(apr_socket_t *sock, |
| request_rec *r, |
| apr_interval_time_t timeout) |
| { |
| ajp_msg_t *msg; |
| apr_status_t rc, rv; |
| apr_interval_time_t org; |
| apr_byte_t result; |
| |
| ap_log_rerror(APLOG_MARK, APLOG_TRACE8, 0, r, |
| "Into ajp_handle_cping_cpong"); |
| |
| rc = ajp_msg_create(r->pool, AJP_PING_PONG_SZ, &msg); |
| if (rc != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01007) |
| "ajp_handle_cping_cpong: ajp_msg_create failed"); |
| return rc; |
| } |
| |
| rc = ajp_msg_serialize_cping(msg); |
| if (rc != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01008) |
| "ajp_handle_cping_cpong: ajp_marshal_into_msgb failed"); |
| return rc; |
| } |
| |
| rc = ajp_ilink_send(sock, msg); |
| ajp_msg_log(r, msg, "ajp_handle_cping_cpong: ajp_ilink_send packet dump"); |
| if (rc != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01009) |
| "ajp_handle_cping_cpong: ajp_ilink_send failed"); |
| return rc; |
| } |
| |
| rc = apr_socket_timeout_get(sock, &org); |
| if (rc != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01010) |
| "ajp_handle_cping_cpong: apr_socket_timeout_get failed"); |
| return rc; |
| } |
| |
| /* Set CPING/CPONG response timeout */ |
| rc = apr_socket_timeout_set(sock, timeout); |
| if (rc != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01011) |
| "ajp_handle_cping_cpong: apr_socket_timeout_set failed"); |
| return rc; |
| } |
| ajp_msg_reuse(msg); |
| |
| /* Read CPONG reply */ |
| rv = ajp_ilink_receive(sock, msg); |
| if (rv != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01012) |
| "ajp_handle_cping_cpong: ajp_ilink_receive failed"); |
| goto cleanup; |
| } |
| |
| ajp_msg_log(r, msg, "ajp_handle_cping_cpong: ajp_ilink_receive packet dump"); |
| rv = ajp_msg_get_uint8(msg, &result); |
| if (rv != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01013) |
| "ajp_handle_cping_cpong: invalid CPONG message"); |
| goto cleanup; |
| } |
| if (result != CMD_AJP13_CPONG) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01014) |
| "ajp_handle_cping_cpong: awaited CPONG, received %d ", |
| result); |
| rv = APR_EGENERAL; |
| goto cleanup; |
| } |
| |
| cleanup: |
| /* Restore original socket timeout */ |
| rc = apr_socket_timeout_set(sock, org); |
| if (rc != APR_SUCCESS) { |
| ap_log_rerror(APLOG_MARK, APLOG_ERR, 0, r, APLOGNO(01015) |
| "ajp_handle_cping_cpong: apr_socket_timeout_set failed"); |
| return rc; |
| } |
| |
| ap_log_rerror(APLOG_MARK, APLOG_TRACE8, 0, r, |
| "ajp_handle_cping_cpong: Done"); |
| return rv; |
| } |
| |
| |
| #define case_to_str(x) case CMD_AJP13_##x:\ |
| return #x;\ |
| break; |
| |
| /** |
| * Convert numeric message type into string |
| * @param type AJP message type |
| * @return AJP message type as a string |
| */ |
| const char *ajp_type_str(int type) |
| { |
| switch (type) { |
| case_to_str(FORWARD_REQUEST) |
| case_to_str(SEND_BODY_CHUNK) |
| case_to_str(SEND_HEADERS) |
| case_to_str(END_RESPONSE) |
| case_to_str(GET_BODY_CHUNK) |
| case_to_str(SHUTDOWN) |
| case_to_str(PING) |
| case_to_str(CPONG) |
| case_to_str(CPING) |
| default: |
| return "CMD_AJP13_UNKNOWN"; |
| } |
| |
| } |