jk_service_apache2.c

Tomcat 4.1与WebServer集成组件的源代码包.

/* ========================================================================= *
 *                                                                           *
 *                 The Apache Software License,  Version 1.1                 *
 *                                                                           *
 *          Copyright (c) 1999-2002 The Apache Software Foundation.          *
 *                           All rights reserved.                            *
 *                                                                           *
 * ========================================================================= *
 *                                                                           *
 * Redistribution and use in source and binary forms,  with or without modi- *
 * fication, are permitted provided that the following conditions are met:   *
 *                                                                           *
 * 1. Redistributions of source code  must retain the above copyright notice *
 *    notice, this list of conditions and the following disclaimer.          *
 *                                                                           *
 * 2. Redistributions  in binary  form  must  reproduce the  above copyright *
 *    notice,  this list of conditions  and the following  disclaimer in the *
 *    documentation and/or other materials provided with the distribution.   *
 *                                                                           *
 * 3. The end-user documentation  included with the redistribution,  if any, *
 *    must include the following acknowlegement:                             *
 *                                                                           *
 *       "This product includes  software developed  by the Apache  Software *
 *        Foundation <http://www.apache.org/>."                              *
 *                                                                           *
 *    Alternately, this acknowlegement may appear in the software itself, if *
 *    and wherever such third-party acknowlegements normally appear.         *
 *                                                                           *
 * 4. The names  "The  Jakarta  Project",  "Jk",  and  "Apache  Software     *
 *    Foundation"  must not be used  to endorse or promote  products derived *
 *    from this  software without  prior  written  permission.  For  written *
 *    permission, please contact <[e-mail]>.                        *
 *                                                                           *
 * 5. Products derived from this software may not be called "Apache" nor may *
 *    "Apache" appear in their names without prior written permission of the *
 *    Apache Software Foundation.                                            *
 *                                                                           *
 * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED WARRANTIES *
 * INCLUDING, BUT NOT LIMITED TO,  THE IMPLIED WARRANTIES OF MERCHANTABILITY *
 * AND FITNESS FOR  A PARTICULAR PURPOSE  ARE DISCLAIMED.  IN NO EVENT SHALL *
 * THE APACHE  SOFTWARE  FOUNDATION OR  ITS CONTRIBUTORS  BE LIABLE  FOR ANY *
 * DIRECT,  INDIRECT,   INCIDENTAL,  SPECIAL,  EXEMPLARY,  OR  CONSEQUENTIAL *
 * DAMAGES (INCLUDING,  BUT NOT LIMITED TO,  PROCUREMENT OF SUBSTITUTE GOODS *
 * OR SERVICES;  LOSS OF USE,  DATA,  OR PROFITS;  OR BUSINESS INTERRUPTION) *
 * HOWEVER CAUSED AND  ON ANY  THEORY  OF  LIABILITY,  WHETHER IN  CONTRACT, *
 * STRICT LIABILITY, OR TORT  (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN *
 * ANY  WAY  OUT OF  THE  USE OF  THIS  SOFTWARE,  EVEN  IF  ADVISED  OF THE *
 * POSSIBILITY OF SUCH DAMAGE.                                               *
 *                                                                           *
 * ========================================================================= *
 *                                                                           *
 * This software  consists of voluntary  contributions made  by many indivi- *
 * duals on behalf of the  Apache Software Foundation.  For more information *
 * on the Apache Software Foundation, please see <http://www.apache.org/>.   *
 *                                                                           *
 * ========================================================================= */

/**
 * Description: Apache 2 plugin for Jakarta/Tomcat                         
 * Author:      Gal Shachor <[e-mail]>                           
 *                 Henri Gomez <[e-mail]>                            
 * Version:     $Revision: 1.33 $                                           
 */

/*
 * Jakarta (jk_) include files
 */
#include "jk_apache2.h"

#include "util_script.h"

/* #define USE_APRTABLES  */

#define NULL_FOR_EMPTY(x)   ((((x)!=NULL) && (strlen((x))!=0)) ? (x) : NULL ) 

static int JK_METHOD jk2_service_apache2_head(jk_env_t *env, jk_ws_service_t *s )
{
    int h;
    int numheaders;
    request_rec *r;
    jk_map_t *headers;
    int debug=1;
    
    if(s==NULL ||  s->ws_private==NULL ) {
        env->l->jkLog(env, env->l, JK_LOG_ERROR, 
                      "service.head() NullPointerException\n");
        return JK_ERR;
    }
    
    if( s->uriEnv != NULL )
        debug=s->uriEnv->mbean->debug;

    r = (request_rec *)s->ws_private;  
        
    if(s->msg==NULL) {
        s->msg = "";
    }
    r->status = s->status;
    r->status_line = apr_psprintf(r->pool, "%d %s", s->status, s->msg);
    headers=s->headers_out;

#ifdef USE_APRTABLES
    {
        char *val= headers->get( env, headers, "Content-Type");
        if( val!=NULL ) {
            char *tmp = apr_pstrdup(r->pool, val);
            ap_content_type_tolower(tmp); 
            /* It should be done like this in Apache 2.0 */
            /* This way, Apache 2.0 will be able to set the output filter */
            /* and it make jk useable with deflate using AddOutputFilterByType DEFLATE text/html */
            ap_set_content_type(r, tmp);
        }
        val= headers->get( env, headers, "Last-Modified");
        if( val!=NULL ) {
            /*
             * If the script gave us a Last-Modified header, we can't just
             * pass it on blindly because of restrictions on future values.
             */
            ap_update_mtime(r, ap_parseHTTPdate(val));
            ap_set_last_modified(r);
        }

        /* No other change required - headers is the same as req->headers_out,
           just with a different interface
        */
    }
#else
    numheaders = headers->size(env, headers);
    /* XXX As soon as we switch to jk_map_apache2, this will not be needed ! */
    if( debug > 0 )
        env->l->jkLog(env, env->l, JK_LOG_DEBUG, 
                      "service.head() %d %d %#lx\n", s->status,
                      numheaders, s->uriEnv);
    
    for(h = 0 ; h < numheaders; h++) {
        char *name=headers->nameAt( env, headers, h );
        char *val=headers->valueAt( env, headers, h );
        name=s->pool->pstrdup( env, s->pool, name );
        val=s->pool->pstrdup( env, s->pool, val );

        if( debug > 0 ) 
            env->l->jkLog(env, env->l, JK_LOG_DEBUG, 
                          "service.head() %s: %s %d %d\n", name, val, h, headers->size( env, headers ));

        /* the cmp can also be avoided in we do this earlier and use
           the header id */
        if(!strcasecmp(name, "Content-type")) {
            /* XXX should be done in handler ! */
            char *tmp = apr_pstrdup(r->pool, val);
            ap_content_type_tolower(tmp); 
            r->content_type = tmp;
            apr_table_set(r->headers_out, name, val);
        } else if(!strcasecmp(name, "Location")) {
            /* XXX setn */
            apr_table_set(r->headers_out, name, val);
        } else if(!strcasecmp(name, "Content-Length")) {
            apr_table_set(r->headers_out, name, val );
        } else if(!strcasecmp(name, "Transfer-Encoding")) {
            apr_table_set(r->headers_out,name, val);
        } else if(!strcasecmp(name, "Last-Modified")) {
            /*
             * If the script gave us a Last-Modified header, we can't just
             * pass it on blindly because of restrictions on future values.
             */
            ap_update_mtime(r, ap_parseHTTPdate(val));
            ap_set_last_modified(r);
            apr_table_set(r->headers_out, name, val);
        } else {     
            /* All other headers may have multiple values like
             * Set-Cookie, so use the table_add to allow that.
             */
             apr_table_add(r->headers_out, name, val);
            /* apr_table_set(r->headers_out, name, val); */
        }
    }
#endif
    
    /* this NOP function was removed in apache 2.0 alpha14 */
    /* ap_send_http_header(r); */
    s->response_started = JK_TRUE;
    
    return JK_OK;
}

/*
 * Read a chunk of the request body into a buffer.  Attempt to read len
 * bytes into the buffer.  Write the number of bytes actually read into
 * actually_read.
 *
 * Think of this function as a method of the apache1.3-specific subclass of
 * the jk_ws_service class.  Think of the *s param as a "this" or "self"
 * pointer.
 */
static int JK_METHOD jk2_service_apache2_read(jk_env_t *env, jk_ws_service_t *s,
                                              void *b, unsigned len,
                                              unsigned *actually_read)
{
    if(s==NULL || s->ws_private==NULL ||  b==NULL || actually_read==NULL ) {
        env->l->jkLog(env, env->l, JK_LOG_ERROR, 
                      "service.read() NullPointerException\n");
        return JK_ERR;
    }

    if(!s->read_body_started) {
        if(ap_should_client_block(s->ws_private)) {
            s->read_body_started = JK_TRUE;
        }
    }

    if(s->read_body_started) {
        long rv;
        if ((rv = ap_get_client_block(s->ws_private, b, len)) < 0) {
            *actually_read = 0;
        } else {
            *actually_read = (unsigned) rv;
        }
    }
    return JK_OK;
}

/*
 * Write a chunk of response data back to the browser.  If the headers
 * haven't yet been sent over, send over default header values (Status =
 * 200, basically).
 *
 * Write len bytes from buffer b.
 *
 * Think of this function as a method of the apache1.3-specific subclass of
 * the jk_ws_service class.  Think of the *s param as a "this" or "self"
 * pointer.
 */
/* Works with 4096, fails with 8192 */
#ifndef CHUNK_SIZE
#define CHUNK_SIZE 4096
#endif

static int JK_METHOD jk2_service_apache2_write(jk_env_t *env, jk_ws_service_t *s,
                                               const void *b, unsigned int len)
{
    size_t r = 0;
    long ll=len;
    char *bb=(char *)b;
    request_rec *rr;
    int debug=1;
    int rc;
    
    if(s==NULL  || s->ws_private == NULL ||  b==NULL) {
        env->l->jkLog(env, env->l, JK_LOG_ERROR, 
                      "service.write() NullPointerException\n");
        return JK_ERR;
    }
    if( s->uriEnv != NULL )
        debug=s->uriEnv->mbean->debug;
    
    if(len==0 ) {
        return JK_OK;
    }

    /* BUFF *bf = p->r->connection->client; */
    /* size_t w = (size_t)l; */
    rr=s->ws_private;
    
    if(!s->response_started) {
        if( debug > 0 )
            env->l->jkLog(env, env->l, JK_LOG_DEBUG, 
                          "service.write() default head\n");
        rc=s->head(env, s);
        if( rc != JK_OK) {
            return rc;
        }
        
        {
            const apr_array_header_t *t = apr_table_elts(rr->headers_out);
            if(t && t->nelts) {
                int i;
                
                apr_table_entry_t *elts = (apr_table_entry_t *)t->elts;
                
                if( debug > 0 ) {
                    for(i = 0 ; i < t->nelts ; i++) {
                        env->l->jkLog(env, env->l, JK_LOG_DEBUG, "OutHeaders %s: %s\n",
                                      elts[i].key, elts[i].val);
                    }
                }
            }
        }
    }

    if (rr->header_only) {
        ap_rflush(rr);
        return JK_OK;
    }
    
    /* Debug - try to get around rwrite */
    while( ll > 0 ) {
        unsigned long toSend=(ll>CHUNK_SIZE) ? CHUNK_SIZE : ll;
        r = ap_rwrite((const char *)bb, toSend, rr );
        /*  env->l->jkLog(env, env->l, JK_LOG_INFO,  */