Nginx 获取自定义 HTTP 头的值 C 代码

要在 nginx 中获取数据包中自定义的 HTTP 头的值，可以使用以下 C 代码：

// 包含必要的头文件
#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_http.h>

// 自定义 HTTP 模块的配置结构
typedef struct {
    ngx_str_t custom_header;
} ngx_http_custom_module_loc_conf_t;

// 解析配置文件中的自定义 HTTP 头
static char *ngx_http_custom_module_custom_header(ngx_conf_t *cf, ngx_command_t *cmd, void *conf) {
    ngx_http_custom_module_loc_conf_t *clcf = conf;
    ngx_str_t *value;

    // 获取配置文件中的值
    value = cf->args->elts;

    // 设置自定义 HTTP 头的值
    clcf->custom_header = value[1];

    return NGX_CONF_OK;
}

// 处理请求的回调函数
static ngx_int_t ngx_http_custom_module_handler(ngx_http_request_t *r) {
    ngx_http_custom_module_loc_conf_t *clcf;
    ngx_table_elt_t *h;
    ngx_str_t header_name = ngx_string('X-Custom-Header');

    // 获取自定义 HTTP 模块的配置
    clcf = ngx_http_get_module_loc_conf(r, ngx_http_custom_module);

    // 创建 HTTP 头
    h = ngx_list_push(&r->headers_out.headers);
    if (h == NULL) {
        return NGX_HTTP_INTERNAL_SERVER_ERROR;
    }

    // 设置 HTTP 头的名字和值
    h->hash = 1;
    h->key = header_name;
    h->value = clcf->custom_header;
    h->lowcase_key = header_name.data;

    // 设置响应的状态码和响应头
    r->headers_out.status = NGX_HTTP_OK;
    r->headers_out.content_length_n = clcf->custom_header.len;
    ngx_str_set(&r->headers_out.content_type, 'text/plain');

    // 发送响应
    ngx_http_send_header(r);
    ngx_http_output_filter(r, &r->headers_out);

    // 返回 OK 标识
    return NGX_OK;
}

// 创建 HTTP 模块的配置结构
static void *ngx_http_custom_module_create_loc_conf(ngx_conf_t *cf) {
    ngx_http_custom_module_loc_conf_t *conf;

    conf = ngx_pcalloc(cf->pool, sizeof(ngx_http_custom_module_loc_conf_t));
    if (conf == NULL) {
        return NGX_CONF_ERROR;
    }

    conf->custom_header.len = 0;
    conf->custom_header.data = NULL;

    return conf;
}

// 定义模块的指令
static ngx_command_t ngx_http_custom_module_commands[] = {
    {
        ngx_string('custom_header'),
        NGX_HTTP_LOC_CONF | NGX_CONF_TAKE1,
        ngx_http_custom_module_custom_header,
        NGX_HTTP_LOC_CONF_OFFSET,
        offsetof(ngx_http_custom_module_loc_conf_t, custom_header),
        NULL
    },
    ngx_null_command
};

// 定义模块的上下文
static ngx_http_module_t ngx_http_custom_module_ctx = {
    NULL,                                // preconfiguration
    NULL,                                // postconfiguration
    NULL,                                // create main configuration
    NULL,                                // init main configuration
    NULL,                                // create server configuration
    NULL,                                // merge server configuration
    ngx_http_custom_module_create_loc_conf,       // create location configuration
    NULL                                 // merge location configuration
};

// 定义模块
ngx_module_t ngx_http_custom_module = {
    NGX_MODULE_V1,
    &ngx_http_custom_module_ctx,          // module context
    ngx_http_custom_module_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
};

上述代码定义了一个自定义的 nginx 模块，用于获取数据包中自定义的 HTTP 头的值。需要在 nginx 的配置文件中进行相应的配置，例如：

http {
    ...
    server {
        ...
        location /custom {
            custom_header 'Custom Value';
        }
    }
}

这样，在访问 http://localhost/custom 时，会返回自定义的 HTTP 头 X-Custom-Header: Custom Value。