ngx_hash.c分析

先分析流程,正常的话都是先添加,然后查找,按照这个流程阅读代码:
以ngx_http_module为例,ngx_http_block为其回调函数,-》ngx_http_optimize_servers-》ngx_http_server_names-》ngx_hash_add_key:




ngx_hash_add_key:
rc = ngx_hash_add_key(&ha, &name[n].name, name[n].server,
NGX_HASH_WILDCARD_KEY);即支持通配符。
先分析通配符情况:
假设addr->servers.nelts只有一个,
typedef struct {
    size_t      len;      13
    u_char     *data;  "*.example.com"
} ngx_str_t;


ngx_int_t
ngx_hash_add_key(ngx_hash_keys_arrays_t *ha, ngx_str_t *key, void *value,
    ngx_uint_t flags)
{
    size_t           len;
    u_char          *p;
    ngx_str_t       *name;
    ngx_uint_t       i, k, n, skip, last;
    ngx_array_t     *keys, *hwc;
    ngx_hash_key_t  *hk;


    last = key->len;


    if (flags & NGX_HASH_WILDCARD_KEY) {


        /*
         * supported wildcards:
         *     "*.example.com", ".example.com", and "www.example.*"
         */


        n = 0;


/*input key check*/
        for (i = 0; i < key->len; i++) {


            if (key->data[i] == '*') {
                if (++n > 1) {
                    return NGX_DECLINED;
                }
            }


            if (key->data[i] == '.' && key->data[i + 1] == '.') {
                return NGX_DECLINED;
            }


            if (key->data[i] == '\0') {
                return NGX_DECLINED;
            }
        }


        if (key->len > 1 && key->data[0] == '.') {
            skip = 1;
            goto wildcard;
        }


        if (key->len > 2) {


            if (key->data[0] == '*' && key->data[1] == '.') {
                skip = 2;
                goto wildcard;
            }
    }


wildcard:


    /* wildcard hash */


    k = ngx_hash_strlow(&key->data[skip], &key->data[skip], last - skip);


    k %= ha->hsize;


    if (skip) {


        /*
         * convert "*.example.com" to "com.example.\0"
         *      and ".example.com" to "com.example\0"
         */


        p = ngx_pnalloc(ha->temp_pool, last);
        if (p == NULL) {
            return NGX_ERROR;
        }


        len = 0;
        n = 0;


        for (i = last - 1; i; i--) {
            if (key->data[i] == '.') {
                ngx_memcpy(&p[n], &key->data[i + 1], len);
                n += len;
                p[n++] = '.';
                len = 0;
                continue;
            }


            len++;
        }


        if (len) {
            ngx_memcpy(&p[n], &key->data[1], len);
            n += len;
        }


        p[n] = '\0';


        hwc = &ha->dns_wc_head;
        keys = &ha->dns_wc_head_hash[k];


    } 


    /* check conflicts in wildcard hash */


    name = keys->elts;


            if (ngx_array_init(keys, ha->temp_pool, 4, sizeof(ngx_str_t)) != NGX_OK)
        {
            return NGX_ERROR;
        }


    name = ngx_array_push(keys);
    if (name == NULL) {
        return NGX_ERROR;
    }


    name->len = last - skip;
    name->data = ngx_pnalloc(ha->temp_pool, name->len);
    if (name->data == NULL) {
        return NGX_ERROR;
    }


    ngx_memcpy(name->data, key->data + skip, name->len);




    /* add to wildcard hash */


    hk = ngx_array_push(hwc);
    if (hk == NULL) {
        return NGX_ERROR;
    }


    hk->key.len = last - 1;
    hk->key.data = p;
    hk->key_hash = 0;
    hk->value = value;


    return NGX_OK;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值