ngx_http_java_module.cpp

#include "request.hpp"
#include "response.hpp"
#include "utils.hpp"
#include "param.hpp"
#include "cache_t.hpp"

#include "ngx_http_java_module.hpp"

static void *ngx_http_java_create_conf(ngx_conf_t *cf);
static char *ngx_http_java_load(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
static ngx_int_t ngx_http_java_handler(ngx_http_request_t *r);

ngx_http_module_t ngx_http_java_module_ctx = {
    NULL,                      /* preconfiguration */
    NULL,                      /* postconfiguration */
    NULL,                      /* create main configuration */
    NULL,                      /* init main configuration */
    NULL,                      /* create server configuration */
    NULL,                      /* merge server configuration */
    ngx_http_java_create_conf, /* create location configuration */
    NULL                       /* merge location configuration */
};

static ngx_command_t ngx_http_java_commands[] = {
    {ngx_string("java_load"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_NOARGS,
     ngx_http_java_load,
     NGX_HTTP_LOC_CONF_OFFSET,
     0,
     NULL},
    {ngx_string("java_class_path"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_TAKE1,
     ngx_conf_set_str_array_slot,
     NGX_HTTP_LOC_CONF_OFFSET,
     offsetof(ngx_http_java_loc_conf_t, class_path),
     NULL},
    {ngx_string("java_options"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_TAKE1,
     ngx_conf_set_str_slot,
     NGX_HTTP_LOC_CONF_OFFSET,
     offsetof(ngx_http_java_loc_conf_t, options),
     NULL},
    {ngx_string("java_servlet"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_TAKE1,
     ngx_conf_set_str_slot,
     NGX_HTTP_LOC_CONF_OFFSET,
     offsetof(ngx_http_java_loc_conf_t, servlet),
     NULL},
    {ngx_string("java_uri_pattern"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_TAKE1,
     ngx_conf_set_str_slot,
     NGX_HTTP_LOC_CONF_OFFSET,
     offsetof(ngx_http_java_loc_conf_t, uri_pattern),
     NULL},
    {ngx_string("java_expires"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_TAKE1,
     ngx_conf_set_sec_slot,
     NGX_HTTP_LOC_CONF_OFFSET,
     offsetof(ngx_http_java_loc_conf_t, expires),
     NULL},
    {ngx_string("java_version"),
     NGX_HTTP_LOC_CONF | NGX_HTTP_LIF_CONF | NGX_CONF_TAKE1,
     ngx_conf_set_num_slot,
     NGX_HTTP_LOC_CONF_OFFSET,
     offsetof(ngx_http_java_loc_conf_t, version),
     NULL},
    ngx_null_command};

ngx_module_t ngx_http_java_module = {
    NGX_MODULE_V1,
    &ngx_http_java_module_ctx, /* module context */
    ngx_http_java_commands,    /* module directives */
    NGX_HTTP_MODULE,           /* module type */
    NULL,                      /* init master */
    NULL,                      /* init module */
    NULL,                      /* init process */
    NULL,                      /* init thread */
    NULL,                      /* exit thread */
    NULL,                      /* exit process */
    NULL,                      /* exit master */
    NGX_MODULE_V1_PADDING};

static void *ngx_http_java_create_conf(ngx_conf_t *cf)
{
    ngx_http_java_loc_conf_t *conf = (ngx_http_java_loc_conf_t *)ngx_pcalloc(cf->pool, sizeof(ngx_http_java_loc_conf_t));
    if (conf)
    {
        conf->expires = NGX_CONF_UNSET;
        conf->class_path = (ngx_array_t *)(NGX_CONF_UNSET_PTR);
        conf->options.data = NULL;
        conf->options.len = 0;
        conf->uri_pattern.data = NULL;
        conf->uri_pattern.len = 0;
        conf->servlet.data = NULL;
        conf->servlet.len = 0;
        conf->version = NGX_CONF_UNSET;
        return conf;
    }

    return NGX_CONF_ERROR;
}

static ngx_int_t ngx_http_java_module_handler(ngx_http_request_t *r)
{
    ngx_http_java_loc_conf_t *conf = (ngx_http_java_loc_conf_t *)ngx_http_get_module_loc_conf(r, ngx_http_java_module);

    if (r->headers_in.if_modified_since && r->headers_in.if_modified_since->value.data)
    {
        time_t now = time(NULL), old = ngx_http_parse_time(r->headers_in.if_modified_since->value.data, r->headers_in.if_modified_since->value.len);
        if (difftime(now, old) <= conf->expires)
        {
            return NGX_HTTP_NOT_MODIFIED;
        }
    }

    hi::request req;
    hi::response res;

    req.uri.assign((char *)r->uri.data, r->uri.len);

    if (java_uri_re == nullptr)
    {
        java_uri_re = std::make_shared<std::regex>(std::string((char *)conf->uri_pattern.data, conf->uri_pattern.len));
    }

    if (!std::regex_match(req.uri, *java_uri_re))
    {
        return NGX_HTTP_BAD_REQUEST;
    }

    std::string lru_cache_key;

    if (hi::set_output_headers_body_init(r, req, res, conf->expires, std::ref(lru_cache_key)) == NGX_DONE)
    {
        goto done;
    }

    if (java_engine == nullptr)
    {
        std::string cp = "-Djava.class.path=";
        ngx_str_t *ptr = (ngx_str_t *)conf->class_path->elts;
        for (ngx_uint_t i = 0; i < conf->class_path->nelts; ++i)
        {
            cp.append(std::string((char *)ptr[i].data, ptr[i].len)).append(":");
        }
        java_engine = std::make_shared<hi::java>(cp, std::string((char *)conf->options.data, conf->options.len), conf->version, std::string((char *)conf->servlet.data, conf->servlet.len));
    }

    java_engine->main(req, res);
done:
    return hi::set_output_headers_body(r, res, conf->expires, lru_cache_key);
}

static void ngx_http_java_body_handler(ngx_http_request_t *r)
{
    ngx_http_finalize_request(r, ngx_http_java_module_handler(r));
}

static ngx_int_t ngx_http_java_handler(ngx_http_request_t *r)
{
    if (r->headers_in.content_length_n > 0)
    {
        ngx_http_core_loc_conf_t *clcf = (ngx_http_core_loc_conf_t *)ngx_http_get_module_loc_conf(r, ngx_http_core_module);
        if (clcf->client_body_buffer_size < (size_t)clcf->client_max_body_size)
        {
            clcf->client_body_buffer_size = clcf->client_max_body_size;
        }
        r->request_body_in_single_buf = 1;
        r->request_body_file_log_level = 0;
        ngx_int_t rc = ngx_http_read_client_request_body(r, ngx_http_java_body_handler);
        if (rc >= NGX_HTTP_SPECIAL_RESPONSE)
        {
            return rc;
        }
        return NGX_DONE;
    }
    else
    {
        ngx_http_discard_request_body(r);
        return ngx_http_java_module_handler(r);
    }
}

static char *ngx_http_java_load(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{
    ngx_http_core_loc_conf_t *clcf;
    clcf = (ngx_http_core_loc_conf_t *)ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
    clcf->handler = ngx_http_java_handler;
    return NGX_CONF_OK;
}