
/*
 * Copyright (C) Igor Sysoev
 */


#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_event.h>
#include <ngx_http.h>


#define NGX_HTTP_VARS_HASH_PRIME  29

#define ngx_http_vars_hash_key(key, vn)                                      \
    {                                                                        \
        ngx_uint_t  n;                                                       \
        for (key = 0, n = 0; n < (vn)->len; n++) {                           \
            key += (vn)->data[n];                                            \
        }                                                                    \
        key %= NGX_HTTP_VARS_HASH_PRIME;                                     \
    }


static ngx_http_variable_value_t *
    ngx_http_variable_header(ngx_http_request_t *r, uintptr_t data);
static ngx_http_variable_value_t *
    ngx_http_variable_unknown_header(ngx_http_request_t *r, uintptr_t data);
static ngx_http_variable_value_t *
    ngx_http_variable_remote_addr(ngx_http_request_t *r, uintptr_t data);
static ngx_http_variable_value_t *
    ngx_http_variable_uri(ngx_http_request_t *r, uintptr_t data);
static ngx_http_variable_value_t *
    ngx_http_variable_query_string(ngx_http_request_t *r, uintptr_t data);


static ngx_array_t  *ngx_http_core_variables_hash;


static ngx_http_core_variable_t  ngx_http_core_variables[] = {

    { ngx_string("HTTP_HOST"), ngx_http_variable_header,
      offsetof(ngx_http_headers_in_t, host) },

    { ngx_string("HTTP_USER_AGENT"), ngx_http_variable_header,
      offsetof(ngx_http_headers_in_t, user_agent) },

    { ngx_string("HTTP_REFERER"), ngx_http_variable_header,
      offsetof(ngx_http_headers_in_t, referer) },

#if (NGX_HTTP_GZIP)
    { ngx_string("HTTP_VIA"), ngx_http_variable_header,
      offsetof(ngx_http_headers_in_t, via) },
#endif

#if (NGX_HTTP_PROXY)
    { ngx_string("HTTP_X_FORWARDED_FOR"), ngx_http_variable_header,
      offsetof(ngx_http_headers_in_t, x_forwarded_for) },
#endif

    { ngx_string("REMOTE_ADDR"), ngx_http_variable_remote_addr, 0 },

    { ngx_string("DOCUMENT_URI"), ngx_http_variable_uri, 0 },

    { ngx_string("QUERY_STRING"), ngx_http_variable_query_string, 0 },

    { ngx_null_string, NULL, 0 }
};


ngx_http_variable_t *
ngx_http_add_variable(ngx_conf_t *cf, ngx_str_t *name, ngx_uint_t set)
{
    ngx_uint_t                  i;
    ngx_http_variable_t        *v;
    ngx_http_core_main_conf_t  *cmcf;

    cmcf = ngx_http_conf_get_module_main_conf(cf, ngx_http_core_module);

    v = cmcf->variables.elts;

    if (v == NULL) {
        if (ngx_array_init(&cmcf->variables, cf->pool, 4,
                           sizeof(ngx_http_variable_t)) == NGX_ERROR)
        {
            return NULL;
        }

    } else {
        for (i = 0; i < cmcf->variables.nelts; i++) {
            if (name->len != v[i].name.len
                || ngx_strncasecmp(name->data, v[i].name.data, name->len) != 0)
            {
                continue;
            }

            if (set && v[i].handler) {
                ngx_conf_log_error(NGX_LOG_EMERG, cf, 0,
                                   "the duplicate \"%V\" variable", name);
                return NULL;
            }

            return &v[i];
        }
    }

    v = ngx_array_push(&cmcf->variables);
    if (v == NULL) {
        return NULL;
    }

    v->name.len = name->len;
    v->name.data = ngx_palloc(cf->pool, name->len);
    if (v->name.data == NULL) {
        return NULL;
    }

    for (i = 0; i < name->len; i++) {
        v->name.data[i] = ngx_toupper(name->data[i]);
    }

    v->index = cmcf->variables.nelts - 1;
    v->handler = NULL;
    v->data = 0;

    return v;
}


ngx_http_variable_value_t *
ngx_http_get_indexed_variable(ngx_http_request_t *r, ngx_uint_t index)
{
    ngx_http_variable_t        *v;
    ngx_http_variable_value_t  *vv;
    ngx_http_core_main_conf_t  *cmcf;

    cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);

    if (cmcf->variables.elts == NULL || cmcf->variables.nelts <= index) {
        ngx_log_error(NGX_LOG_ALERT, r->connection->log, 0,
                      "unknown variable index: %d", index);
        return NULL;
    }

    if (r->variables && r->variables[index]) {
        return r->variables[index];
    }

    v = cmcf->variables.elts;

    vv = v[index].handler(r, v[index].data);

    if (r->variables == NULL) {
        r->variables = ngx_pcalloc(r->pool, cmcf->variables.nelts
                                        * sizeof(ngx_http_variable_value_t *));
        if (r->variables == NULL) {
            return NULL;
        }
    }

    r->variables[index] = vv;

    return vv;
}


ngx_http_variable_value_t *
ngx_http_get_variable(ngx_http_request_t *r, ngx_str_t *name)
{
    ngx_uint_t                  i, key;
    ngx_http_variable_t        *v;
    ngx_http_core_variable_t   *cv;
    ngx_http_core_main_conf_t  *cmcf;

    cmcf = ngx_http_get_module_main_conf(r, ngx_http_core_module);

    v = cmcf->variables.elts;
    for (i = 0; i < cmcf->variables.nelts; i++) {
        if (v[i].name.len != name->len) {
            continue;
        }

        if (ngx_strncmp(v[i].name.data, name->data, name->len) == 0) {
            return ngx_http_get_indexed_variable(r, v[i].index);
        }
    }

    ngx_http_vars_hash_key(key, name);

    cv = ngx_http_core_variables_hash[key].elts;
    for (i = 0; i < ngx_http_core_variables_hash[key].nelts; i++) {
        if (cv[i].name.len != name->len) {
            continue;
        }

        if (ngx_strncmp(cv[i].name.data, name->data, name->len) == 0) {
            return cv[i].handler(r, cv[i].data);
        }
    }

    if (ngx_strncmp(name->data, "HTTP_", 5) == 0) {
        return ngx_http_variable_unknown_header(r, (uintptr_t) name);
    }

    ngx_log_error(NGX_LOG_ERR, r->connection->log, 0,
                  "unknown \"%V\" variable", name);

    return NGX_HTTP_VARIABLE_NOT_FOUND;
}


static ngx_http_variable_value_t *
ngx_http_variable_header(ngx_http_request_t *r, uintptr_t data)
{
    ngx_table_elt_t            *h;
    ngx_http_variable_value_t  *vv;

    h = *(ngx_table_elt_t **) ((char *) &r->headers_in + data);

    if (h == NULL) {
        return NGX_HTTP_VARIABLE_NOT_FOUND;
    }

    vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
    if (vv == NULL) {
        return NULL;
    }

    vv->value = 0;
    vv->text = h->value;

    return vv;
}


static ngx_http_variable_value_t *
ngx_http_variable_unknown_header(ngx_http_request_t *r, uintptr_t data)
{
    ngx_str_t  *var = (ngx_str_t *) data;

    u_char                      ch;
    ngx_uint_t                  i, n;
    ngx_list_part_t            *part;
    ngx_table_elt_t            *header;
    ngx_http_variable_value_t  *vv;

    part = &r->headers_in.headers.part;
    header = part->elts;

    for (i = 0; /* void */ ; i++) {

        if (i >= part->nelts) {
            if (part->next == NULL) {
                break;
            }

            part = part->next;
            header = part->elts;
            i = 0;
        }

        for (n = 0; n + 5 < var->len && n < header[i].key.len; n++)
        {
            ch = header[i].key.data[n];

            if (ch >= 'a' && ch <= 'z') {
                ch &= ~0x20;

            } else if (ch == '-') {
                ch = '_';
            }

            if (var->data[n + 5] != ch) {
                break;
            }
        }

        if (n + 5 == var->len) {
            vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
            if (vv == NULL) {
                return NULL;
            }

            vv->value = 0;
            vv->text = header[i].value;
            return vv;
        }
    }

    return NGX_HTTP_VARIABLE_NOT_FOUND;
}


static ngx_http_variable_value_t *
ngx_http_variable_remote_addr(ngx_http_request_t *r, uintptr_t data)
{
    ngx_http_variable_value_t  *vv;

    vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
    if (vv == NULL) {
        return NULL;
    }

    vv->value = 0;
    vv->text = r->connection->addr_text;

    return vv;
}


static ngx_http_variable_value_t *
ngx_http_variable_uri(ngx_http_request_t *r, uintptr_t data)
{
    ngx_http_variable_value_t  *vv;

    vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
    if (vv == NULL) {
        return NULL;
    }

    vv->value = 0;
    vv->text = r->uri;

    return vv;
}


static ngx_http_variable_value_t *
ngx_http_variable_query_string(ngx_http_request_t *r, uintptr_t data)
{
    ngx_http_variable_value_t  *vv;

    vv = ngx_palloc(r->pool, sizeof(ngx_http_variable_value_t));
    if (vv == NULL) {
        return NULL;
    }

    vv->value = 0;
    vv->text = r->args;

    return vv;
}


ngx_int_t
ngx_http_variables_init(ngx_cycle_t *cycle)
{
    ngx_uint_t                  i, j, key;
    ngx_http_variable_t        *v;
    ngx_http_core_variable_t   *cv, *vp;
    ngx_http_core_main_conf_t  *cmcf;

    ngx_http_core_variables_hash = ngx_palloc(cycle->pool,
                                              NGX_HTTP_VARS_HASH_PRIME
                                              * sizeof(ngx_array_t));
    if (ngx_http_core_variables_hash == NULL) {
        return NGX_ERROR;
    }

    for (i = 0; i < NGX_HTTP_VARS_HASH_PRIME; i++) {
        if (ngx_array_init(&ngx_http_core_variables_hash[i], cycle->pool, 4,
                           sizeof(ngx_http_core_variable_t)) == NGX_ERROR)
        {
            return NGX_ERROR;
        }
    }

    for (cv = ngx_http_core_variables; cv->name.len; cv++) {
        ngx_http_vars_hash_key(key, &cv->name);

        vp = ngx_array_push(&ngx_http_core_variables_hash[key]);
        if (vp == NULL) {
            return NGX_ERROR;
        }

        *vp = *cv;
    }


    cmcf = ngx_http_cycle_get_module_main_conf(cycle, ngx_http_core_module);

    v = cmcf->variables.elts;
    for (i = 0; i < cmcf->variables.nelts; i++) {

        if (v[i].handler) {
            continue;
        }

        ngx_http_vars_hash_key(key, &v[i].name);

        cv = ngx_http_core_variables_hash[key].elts;
        for (j = 0; j < ngx_http_core_variables_hash[key].nelts; j++) {
            if (cv[j].name.len != v[i].name.len) {
                continue;
            }

            if (ngx_strncmp(cv[j].name.data, v[i].name.data, v[i].name.len)
                == 0)
            {
                v[i].handler = cv[j].handler;
                v[i].data = cv[j].data;
                continue;
            }
        }

        if (ngx_strncmp(v[i].name.data, "HTTP_", 5) == 0) {
            v[i].handler = ngx_http_variable_unknown_header;
            v[i].data = (uintptr_t) &v[i].name;
            continue;
        }

        ngx_log_error(NGX_LOG_ERR, cycle->log, 0,
                      "unknown \"%V\" variable", &v[i].name);

        return NGX_ERROR;
    }

    return NGX_OK;
}