modules/proxy/ajp_msg.c

	ajp_msg.c revision 6121ba973fff0ef0b2d9ff6ef2a50188ab2dce34
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/* Licensed to the Apache Software Foundation (ASF) under one or more
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * contributor license agreements.  See the NOTICE file distributed with
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * this work for additional information regarding copyright ownership.
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * The ASF licenses this file to You under the Apache License, Version 2.0
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * (the "License"); you may not use this file except in compliance with
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * the License.  You may obtain a copy of the License at
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *     http://www.apache.org/licenses/LICENSE-2.0
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Unless required by applicable law or agreed to in writing, software
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * distributed under the License is distributed on an "AS IS" BASIS,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * See the License for the specific language governing permissions and
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * limitations under the License.
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg#include "ajp.h"
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgstatic char *hex_table = "0123456789ABCDEF";
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Dump up to the first 1024 bytes on an AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param pool      pool to allocate from
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to dump
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param err       error string to display
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          dump message
44f575c8cb19a7a5cd61664a7848be6bc197df02fuankg */
44f575c8cb19a7a5cd61664a7848be6bc197df02fuankgchar * ajp_msg_dump(apr_pool_t *pool, ajp_msg_t *msg, char *err)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t  i, j;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    char        line[80];
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    char        *current;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    char        *rv, *p;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t  bl = 8192;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_byte_t  x;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t  len = msg->len;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    /* Display only first 1024 bytes */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if (len > 1024)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        len = 1024;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    rv = apr_palloc(pool, bl);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_snprintf(rv, bl,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                 "ajp_msg_dump(): %s pos=%" APR_SIZE_T_FMT
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                 " len=%" APR_SIZE_T_FMT " max=%d\n",
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                 err, msg->pos, msg->len, msg->max_size);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    bl -= strlen(rv);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    p = rv + strlen(rv);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    for (i = 0; i < len; i += 16) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        current = line;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        for (j = 0; j < 16; j++) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg             x = msg->buf[i + j];
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            *current++ = hex_table[x >> 4];
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            *current++ = hex_table[x & 0x0f];
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            *current++ = ' ';
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        *current++ = ' ';
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        *current++ = '-';
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        *current++ = ' ';
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        for (j = 0; j < 16; j++) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            x = msg->buf[i + j];
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            if (x > 0x20 && x < 0x7F) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                *current++ = x;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            else {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                *current++ = '.';
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg            }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        *current++ = '\0';
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        apr_snprintf(p, bl,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                     "ajp_msg_dump(): %.4lx    %s\n",
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                     (unsigned long)i, line);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        bl -= strlen(rv);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        p = rv + strlen(rv);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return rv;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Check a new AJP Message by looking at signature and return its size
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to check
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param len       Pointer to returned len
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_check_header(ajp_msg_t *msg, apr_size_t *len)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_byte_t *head = msg->buf;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t msglen;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if (!((head[0] == 0x41 && head[1] == 0x42) ||
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg          (head[0] == 0x12 && head[1] == 0x34))) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                      "ajp_check_msg_header() got bad signature %02x%02x",
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                      head[0], head[1]);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        return AJP_EBAD_SIGNATURE;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msglen  = ((head[2] & 0xff) << 8);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msglen += (head[3] & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if (msglen > msg->max_size) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                     "ajp_check_msg_header() incoming message is "
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                     "too big %" APR_SIZE_T_FMT ", max is %d",
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                     msglen, msg->max_size);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        return AJP_ETOBIG;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->len = msglen + AJP_HEADER_LEN;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->pos = AJP_HEADER_LEN;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    *len     = msglen;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return APR_SUCCESS;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Reset an AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to reset
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_reset(ajp_msg_t *msg)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->len = AJP_HEADER_LEN;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->pos = AJP_HEADER_LEN;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return APR_SUCCESS;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Reuse an AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to reuse
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_reuse(ajp_msg_t *msg)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_byte_t *buf;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t max_size;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    buf = msg->buf;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    max_size = msg->max_size;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    memset(msg, 0, sizeof(ajp_msg_t));
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf = buf;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->max_size = max_size;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->header_len = AJP_HEADER_LEN;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    ajp_msg_reset(msg);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return APR_SUCCESS;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Mark the end of an AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to end
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_end(ajp_msg_t *msg)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t len = msg->len - AJP_HEADER_LEN;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if (msg->server_side) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        msg->buf[0] = 0x41;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        msg->buf[1] = 0x42;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    else {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        msg->buf[0] = 0x12;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        msg->buf[1] = 0x34;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[2] = (apr_byte_t)((len >> 8) & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[3] = (apr_byte_t)(len & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return APR_SUCCESS;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgstatic APR_INLINE int ajp_log_overflow(ajp_msg_t *msg, const char *context)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                 "%s(): BufferOverflowException %" APR_SIZE_T_FMT
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                 " %" APR_SIZE_T_FMT,
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg                 context, msg->pos, msg->len);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return AJP_EOVERFLOW;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Add an unsigned 32bits value to AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to get value from
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param value     value to add to AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_append_uint32(ajp_msg_t *msg, apr_uint32_t value)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t len = msg->len;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if ((len + 4) > msg->max_size) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        return ajp_log_overflow(msg, "ajp_msg_append_uint32");
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[len]     = (apr_byte_t)((value >> 24) & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[len + 1] = (apr_byte_t)((value >> 16) & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[len + 2] = (apr_byte_t)((value >> 8) & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[len + 3] = (apr_byte_t)(value & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->len += 4;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return APR_SUCCESS;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Add an unsigned 16bits value to AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to get value from
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param value     value to add to AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_append_uint16(ajp_msg_t *msg, apr_uint16_t value)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t len = msg->len;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if ((len + 2) > msg->max_size) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        return ajp_log_overflow(msg, "ajp_msg_append_uint16");
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[len]     = (apr_byte_t)((value >> 8) & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->buf[len + 1] = (apr_byte_t)(value & 0xFF);
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    msg->len += 2;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    return APR_SUCCESS;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg}
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg/**
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * Add an unsigned 8bits value to AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg *
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param msg       AJP Message to get value from
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @param value     value to add to AJP Message
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg * @return          APR_SUCCESS or error
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg */
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankgapr_status_t ajp_msg_append_uint8(ajp_msg_t *msg, apr_byte_t value)
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg{
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    apr_size_t len = msg->len;
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    if ((len + 1) > msg->max_size) {
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg        return ajp_log_overflow(msg, "ajp_msg_append_uint8");
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg    }
1850c8c89f2d775a2e905832b77b9bcc3bd46db4fuankg
    msg->buf[len] = value;
    msg->len += 1;

    return APR_SUCCESS;
}

/**
 *  Add a String in AJP message, and transform the String in ASCII
 *  if convert is set and we're on an EBCDIC machine
 *
 * @param msg       AJP Message to get value from
 * @param value     Pointer to String
 * @param convert   When set told to convert String to ASCII
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_append_string_ex(ajp_msg_t *msg, const char *value,
                                      int convert)
{
    size_t len;

    if (value == NULL) {
        return(ajp_msg_append_uint16(msg, 0xFFFF));
    }

    len = strlen(value);
    if ((msg->len + len + 3) > msg->max_size) {
        return ajp_log_overflow(msg, "ajp_msg_append_cvt_string");
    }

    /* ignore error - we checked once */
    ajp_msg_append_uint16(msg, (apr_uint16_t)len);

    /* We checked for space !!  */
    memcpy(msg->buf + msg->len, value, len + 1); /* including \0 */

    if (convert)   /* convert from EBCDIC if needed */
        ap_xlate_proto_to_ascii((char *)msg->buf + msg->len, len + 1);

    msg->len += len + 1;

    return APR_SUCCESS;
}

/**
 * Add a Byte array to AJP Message
 *
 * @param msg       AJP Message to get value from
 * @param value     Pointer to Byte array
 * @param valuelen  Byte array len
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_append_bytes(ajp_msg_t *msg, const apr_byte_t *value,
                                  apr_size_t valuelen)
{
    if (! valuelen) {
        return APR_SUCCESS; /* Shouldn't we indicate an error ? */
    }

    if ((msg->len + valuelen) > msg->max_size) {
        return ajp_log_overflow(msg, "ajp_msg_append_bytes");
    }

    /* We checked for space !!  */
    memcpy(msg->buf + msg->len, value, valuelen);
    msg->len += valuelen;

    return APR_SUCCESS;
}

/**
 * Get a 32bits unsigned value from AJP Message
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_get_uint32(ajp_msg_t *msg, apr_uint32_t *rvalue)
{
    apr_uint32_t value;

    if ((msg->pos + 3) > msg->len) {
        return ajp_log_overflow(msg, "ajp_msg_get_uint32");
    }

    value  = ((msg->buf[(msg->pos++)] & 0xFF) << 24);
    value |= ((msg->buf[(msg->pos++)] & 0xFF) << 16);
    value |= ((msg->buf[(msg->pos++)] & 0xFF) << 8);
    value |= ((msg->buf[(msg->pos++)] & 0xFF));

    *rvalue = value;
    return APR_SUCCESS;
}


/**
 * Get a 16bits unsigned value from AJP Message
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_get_uint16(ajp_msg_t *msg, apr_uint16_t *rvalue)
{
    apr_uint16_t value;

    if ((msg->pos + 1) > msg->len) {
        return ajp_log_overflow(msg, "ajp_msg_get_uint16");
    }

    value  = ((msg->buf[(msg->pos++)] & 0xFF) << 8);
    value += ((msg->buf[(msg->pos++)] & 0xFF));

    *rvalue = value;
    return APR_SUCCESS;
}

/**
 * Peek a 16bits unsigned value from AJP Message, position in message
 * is not updated
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_peek_uint16(ajp_msg_t *msg, apr_uint16_t *rvalue)
{
    apr_uint16_t value;

    if ((msg->pos + 1) > msg->len) {
        return ajp_log_overflow(msg, "ajp_msg_peek_uint16");
    }

    value = ((msg->buf[(msg->pos)] & 0xFF) << 8);
    value += ((msg->buf[(msg->pos + 1)] & 0xFF));

    *rvalue = value;
    return APR_SUCCESS;
}

/**
 * Peek a 8bits unsigned value from AJP Message, position in message
 * is not updated
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_peek_uint8(ajp_msg_t *msg, apr_byte_t *rvalue)
{
    if (msg->pos > msg->len) {
        return ajp_log_overflow(msg, "ajp_msg_peek_uint8");
    }

    *rvalue = msg->buf[msg->pos];
    return APR_SUCCESS;
}

/**
 * Get a 8bits unsigned value from AJP Message
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_get_uint8(ajp_msg_t *msg, apr_byte_t *rvalue)
{

    if (msg->pos > msg->len) {
        return ajp_log_overflow(msg, "ajp_msg_get_uint8");
    }

    *rvalue = msg->buf[msg->pos++];
    return APR_SUCCESS;
}


/**
 * Get a String value from AJP Message
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_get_string(ajp_msg_t *msg, const char **rvalue)
{
    apr_uint16_t size;
    apr_size_t   start;
    apr_status_t status;

    status = ajp_msg_get_uint16(msg, &size);
    start = msg->pos;

    if ((status != APR_SUCCESS) || (size + start > msg->max_size)) {
        return ajp_log_overflow(msg, "ajp_msg_get_string");
    }

    msg->pos += (apr_size_t)size;
    msg->pos++;                   /* a String in AJP is NULL terminated */

    *rvalue = (const char *)(msg->buf + start);
    return APR_SUCCESS;
}


/**
 * Get a Byte array from AJP Message
 *
 * @param msg       AJP Message to get value from
 * @param rvalue    Pointer where value will be returned
 * @param rvalueLen Pointer where Byte array len will be returned
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_get_bytes(ajp_msg_t *msg, apr_byte_t **rvalue,
                               apr_size_t *rvalue_len)
{
    apr_uint16_t size;
    apr_size_t   start;
    apr_status_t status;

    status = ajp_msg_get_uint16(msg, &size);
    /* save the current position */
    start = msg->pos;

    if ((status != APR_SUCCESS) || (size + start > msg->max_size)) {
        return ajp_log_overflow(msg, "ajp_msg_get_bytes");
    }
    msg->pos += (apr_size_t)size;   /* only bytes, no trailer */

    *rvalue     = msg->buf + start;
    *rvalue_len = size;

    return APR_SUCCESS;
}


/**
 * Create an AJP Message from pool
 *
 * @param pool      memory pool to allocate AJP message from
 * @param size      size of the buffer to create
 * @param rmsg      Pointer to newly created AJP message
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_create(apr_pool_t *pool, apr_size_t size, ajp_msg_t **rmsg)
{
    ajp_msg_t *msg = (ajp_msg_t *)apr_pcalloc(pool, sizeof(ajp_msg_t));

    if (!msg) {
        ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
                      "ajp_msg_create(): can't allocate AJP message memory");
        return APR_ENOPOOL;
    }

    msg->server_side = 0;

    msg->buf = (apr_byte_t *)apr_palloc(pool, size);

    /* XXX: This should never happen
     * In case if the OS cannont allocate 8K of data
     * we are in serious trouble
     * No need to check the alloc return value, cause the
     * core dump is probably the best solution anyhow.
     */
    if (msg->buf == NULL) {
        ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
                      "ajp_msg_create(): can't allocate AJP message memory");
        return APR_ENOPOOL;
    }

    msg->len = 0;
    msg->header_len = AJP_HEADER_LEN;
    msg->max_size = size;
    *rmsg = msg;

    return APR_SUCCESS;
}

/**
 * Recopy an AJP Message to another
 *
 * @param smsg      source AJP message
 * @param dmsg      destination AJP message
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_copy(ajp_msg_t *smsg, ajp_msg_t *dmsg)
{
    if (dmsg == NULL) {
        ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
                     "ajp_msg_copy(): destination msg is null");
        return AJP_EINVAL;
    }

    if (smsg->len > smsg->max_size) {
        ap_log_error(APLOG_MARK, APLOG_ERR, 0, NULL,
                     "ajp_msg_copy(): destination buffer too "
                     "small %" APR_SIZE_T_FMT ", max size is %d",
                     smsg->len, smsg->max_size);
        return  AJP_ETOSMALL;
    }

    memcpy(dmsg->buf, smsg->buf, smsg->len);
    dmsg->len = smsg->len;
    dmsg->pos = smsg->pos;

    return APR_SUCCESS;
}


/**
 * Serialize in an AJP Message a PING command
 *
 * +-----------------------+
 * | PING CMD (1 byte)     |
 * +-----------------------+
 *
 * @param smsg      AJP message to put serialized message
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_serialize_ping(ajp_msg_t *msg)
{
    apr_status_t rc;
    ajp_msg_reset(msg);

    if ((rc = ajp_msg_append_uint8(msg, CMD_AJP13_PING)) != APR_SUCCESS)
        return rc;

    return APR_SUCCESS;
}

/**
 * Serialize in an AJP Message a CPING command
 *
 * +-----------------------+
 * | CPING CMD (1 byte)    |
 * +-----------------------+
 *
 * @param smsg      AJP message to put serialized message
 * @return          APR_SUCCESS or error
 */
apr_status_t ajp_msg_serialize_cping(ajp_msg_t *msg)
{
    apr_status_t rc;
    ajp_msg_reset(msg);

    if ((rc = ajp_msg_append_uint8(msg, CMD_AJP13_CPING)) != APR_SUCCESS)
        return rc;

    return APR_SUCCESS;
}