Nginx编写Http模块

HTTP模块的调用

  • worker 进程会在一个for 循环里面反复调用事件模块检测网络事件。


基本数据结构

  • 对整形的封装
    ngx_int_t 封装了有符号整形, 而 ngx_uint_t 封装了无符号整形
  • ngx_str_t
    定义:
typedef struct {
    size_t len;
    u_char * data;
}ngx_str_t;
  • ngx_list_t
    链表容器,链表的元素是一个个数组
typedef struct ngx_list_part_s ngx_list_part_t;
struct ngx_list_part_s{
    void             * elts;   // 指向数组的起始位置
    ngx_unit_t         nelts;  // 标记数组中已经使用的容量
    ngx_list_part_t  * next;
};
typedef struct {
    ngx_list_part_t    * last;  // 指向链表的最后一个数组的位置
    ngx_list_part_t      part;  // 指向链表头
    size_t               size;  // 一个链表元素(ie, 数组节点)所包含的元素大小
    ngx_uint_t           nalloc; // 一个链表元素(ie, 数组节点)所包含的元素个数
    ngx_pool_t         * pool;  // 内存池地址
}ngx_list_t;

其内存分布图如下所示


  • ngx_table_elt_t
    本质是一个键值对结构

  • ngx_buf_t
    , 是处理大数据的关键数据结构, 既应用于内存数据也应用于磁盘数据

  • ngx_chain_t
    一般与ngx_buf_t配合来使用,用来发送包体

将自己的HTTP模块编译到Nginx中

主要是配置好config文件

 ngx_addon_name=ngx_http_mytest_module
 HTTP_MODULES="$HTTP_MODULES ngx_http_mytest_module"
 NGX_ADDON_SRCS="$NGX_ADDON_SRCS $ngx_addon_dir/ngx_http_mytest_module.c"

HTTP模块数据结构

  • ngx_module_t:这个数据结构中, 我们最需要关注的是下面3个变量
    ctx: 用于指向一类模块的上下文结构体 (对于HTTP 模块, ctx 指针必须指向 ngx_http_module_t 接口)
    commands:,用于处理nginx.conf 文件中的配置项
    type:用来表示模块的类型(定义HTTP 模块, 这个值应该设置为: NGX_HTTP_MODULE)

ngx_http_module_t接口

HTTP 框架在启动的过程中会在每个阶段中调用ngx_http_module_t 中相应的方法


commands 数组

用来定义模块的配置文件参数, 数组中的每个元素都是 ngx_command_t 类型, 数组的结尾使用 ngx_null_command 来标记.
Nginx 解析配置文件的流程: 解析每个配置项的时候, 首先会遍历所有的模块,对于每个模块而言, 通过遍历commands 数组进行, 查找他所感兴趣的配置项,个ngx_command_t 结构体定义了自己感兴趣的一个配置项

定义一个Http模块

这里有一个hello_world的例子
http://www.jianshu.com/p/d7a783de09dd

处理用户请求

在出现mytest 配置项的时候, ngx_http_mytest 方法被调用, 这时候, 将ngx_http_core_loc_conf_t 结构的handler 成员指定为ngx_http_mytest_handler。 在HTTP 框架接收完HTTP请求的头部之后, 会调用handler指向的方法, 用来处理用户请求。

  • 处理方法的返回值
  • 主要包含RFC 2616 规范中定义的返回码, 以及Nginx 自身定义的HTTP返回码
  • 主要需要了解4个通用的返回码:
    • NGX_OK: 表示成功
    • NGX_DECLINED: 继续在NGX_HTTP_CONTENT_PHASE 阶段, 寻找感兴趣的HTTP 模块来再次处理这个请求
    • NGX_DONE: 表示到此为止, 同时HTTP 框架将暂时不再继续执行这个请求的后续部分。事实上, 如果是keepalive类型的用户请求, 就会保持这个http 连接, 然后把控制权返回给Nginx
    • NGX_ERROR: 表示错误, 将终止请求
  • 获取URI 和 参数
    • 请求的所有信息(方法, URI, 协议版本号和头部等)都可以在传入的ngx_http_request_t 类型的参数 r 中获取。
    • 可以通过r->method 这个整形成员与预定义的几个宏进行比较, 从而获取方法名
    • URI, uri 成员指向用户请求的 URI
    • args, 指向用户请求中的URI参数
    • http_protocol 的data 成员指向用户请求中HTTP 协议版本字符串的起始地址, len 成员为其字符串长度
  • 获取HTTP 头部
    • ngx_http_headers_in_t 类型的headers_in存储已经解析过的HTTP 头部
  • 获取HTTP 包体
    • HTTP 包体的长度有可能非常大, 如果试图一次性调用并读完所有的包体, 可能导致nginx 的阻塞
    • nginx 提供一个叫做 ngx_http_read_client_request_body的异步方法, 调用它只是说明要求Nginx开始连接请求的包体, 并不表示是否已经接收完毕。
    • 如果 ngx_http_read_client_request_body是在ngx_http_mytest_handler处理方法中调用的话, 一般后者需要返回 NGX_DONE
    • 而如果不想处理请求中的包体的时候, 可以调用 ngx_http_discard_request_body 方法将接收自客户端的HTTP 包体丢弃掉
  • 发送响应
    • 使用api: ngx_http_send_header 即可发送响应
    • 发送之前, 需要预先指定一下 headers_out 中的成员,如下
    r->headers_out.status = NGX_HTTP_OK;
    //响应包是有包体内容的,所以需要设置Content-Length长度
    r->headers_out.content_length_n = response.len;
    //设置Content-Type
    r->headers_out.content_type = type;
  • 将内存中的字符串作为包体发送
    • 调用 ngx_http_output_filter 可以实现向客户端发送HTTP 响应包体
    • 发送的内容放在 ngx_buf_t 结构中, 并使用 ngx_chain_t 传递给 ngx_http_output_filter 方法,如下所示
    //构造ngx_buf_t结构准备发送包体
    ngx_buf_t                 *b;
    b = ngx_create_temp_buf(r->pool, response.len);
    if (b == NULL)
    {
        return NGX_HTTP_INTERNAL_SERVER_ERROR;
    }
    //将Hello World拷贝到ngx_buf_t指向的内存中
    ngx_memcpy(b->pos, response.data, response.len);
    //注意,一定要设置好last指针
    b->last = b->pos + response.len;
    //声明这是最后一块缓冲区
    b->last_buf = 1;

    //构造发送时的ngx_chain_t结构体
    ngx_chain_t     out;
    //赋值ngx_buf_t
    out.buf = b;
    //设置next为NULL
    out.next = NULL;

    //最后一步发送包体,http框架会调用ngx_http_finalize_request方法
//结束请求
    return ngx_http_output_filter(r, &out);
  • 清理文件句柄
    • nginx 通过异步方式将整个文件高效的发送给用户, 但是需要在发送完毕之后, 关闭打开的文件句柄, 避免句柄泄露
    • ngx_pool_cleanup_file 用于关闭文件句柄
    • ngx_pool_cleanup_t 结构


    • ngx_pool_cleanup_file_t 结构


    • 使用ngx_pool_cleanup_add 添加, 并设置完上面两个结构即可
  • 支持用户多线程下载和断点续传
r->allow_ranges = 1;

一个包含文件传输的代码例子

#include <ngx_config.h>
#include <ngx_core.h>
#include <ngx_http.h>
static char *
ngx_http_mytest(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
static ngx_int_t ngx_http_mytest_handler(ngx_http_request_t *r);
static ngx_command_t  ngx_http_mytest_commands[] =
{
    {
        ngx_string("mytest"),
        NGX_HTTP_MAIN_CONF | NGX_HTTP_SRV_CONF | NGX_HTTP_LOC_CONF | NGX_HTTP_LMT_CONF | NGX_CONF_NOARGS,
        ngx_http_mytest,
        NGX_HTTP_LOC_CONF_OFFSET,
        0,
        NULL
    },
    ngx_null_command
};
static ngx_http_module_t  ngx_http_mytest_module_ctx =
{
    NULL,                              /* preconfiguration */
    NULL,                       /* postconfiguration */
    NULL,                              /* create main configuration */
    NULL,                              /* init main configuration */
    NULL,                              /* create server configuration */
    NULL,                              /* merge server configuration */
    NULL,                   /* create location configuration */
    NULL                    /* merge location configuration */
};
ngx_module_t  ngx_http_mytest_module =
{
    NGX_MODULE_V1,
    &ngx_http_mytest_module_ctx,           /* module context */
    ngx_http_mytest_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 char *
ngx_http_mytest(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{
    ngx_http_core_loc_conf_t  *clcf;
    //首先找到mytest配置项所属的配置块,clcf貌似是location块内的数据
//结构,其实不然,它可以是main、srv或者loc级别配置项,也就是说在每个
//http{}和server{}内也都有一个ngx_http_core_loc_conf_t结构体
    clcf = ngx_http_conf_get_module_loc_conf(cf, ngx_http_core_module);
    //http框架在处理用户请求进行到NGX_HTTP_CONTENT_PHASE阶段时,如果
//请求的主机域名、URI与mytest配置项所在的配置块相匹配,就将调用我们
//实现的ngx_http_mytest_handler方法处理这个请求
    clcf->handler = ngx_http_mytest_handler;
    return NGX_CONF_OK;
}
static ngx_int_t ngx_http_mytest_handler(ngx_http_request_t *r)
{
    //必须是GET或者HEAD方法,否则返回405 Not Allowed
    if (!(r->method & (NGX_HTTP_GET | NGX_HTTP_HEAD)))
    {
        return NGX_HTTP_NOT_ALLOWED;
    }
    //丢弃请求中的包体
    ngx_int_t rc = ngx_http_discard_request_body(r);
    if (rc != NGX_OK)
    {
        return rc;
    }
    ngx_buf_t *b;
    b = ngx_palloc(r->pool, sizeof(ngx_buf_t));
    //要打开的文件
    u_char* filename = (u_char*)"/tmp/test.txt";
    b->in_file = 1;
    b->file = ngx_pcalloc(r->pool, sizeof(ngx_file_t));
    b->file->fd = ngx_open_file(filename, NGX_FILE_RDONLY | NGX_FILE_NONBLOCK, NGX_FILE_OPEN, 0);
    b->file->log = r->connection->log;
    b->file->name.data = filename;
    b->file->name.len = sizeof(filename) - 1;
    if (b->file->fd <= 0)
    {
        return NGX_HTTP_NOT_FOUND;
    }
    //支持断点续传
    r->allow_ranges = 1;
    //获取文件长度
    if (ngx_file_info(filename, &b->file->info) == NGX_FILE_ERROR)
    {
        return NGX_HTTP_INTERNAL_SERVER_ERROR;
    }
    //设置缓冲区指向的文件块
    b->file_pos = 0;
    b->file_last = b->file->info.st_size;
    ngx_pool_cleanup_t* cln = ngx_pool_cleanup_add(r->pool, sizeof(ngx_pool_cleanup_file_t));
    if (cln == NULL)
    {
        return NGX_ERROR;
    }
    cln->handler = ngx_pool_cleanup_file;
    ngx_pool_cleanup_file_t  *clnf = cln->data;
    clnf->fd = b->file->fd;
    clnf->name = b->file->name.data;
    clnf->log = r->pool->log;
    //设置返回的Content-Type。注意,ngx_str_t有一个很方便的初始化宏
//ngx_string,它可以把ngx_str_t的data和len成员都设置好
    ngx_str_t type = ngx_string("text/plain");
    //设置返回状态码
    r->headers_out.status = NGX_HTTP_OK;
    //响应包是有包体内容的,所以需要设置Content-Length长度
    r->headers_out.content_length_n = b->file->info.st_size;
    //设置Content-Type
    r->headers_out.content_type = type;
    //发送http头部
    rc = ngx_http_send_header(r);
    if (rc == NGX_ERROR || rc > NGX_OK || r->header_only)
    {
        return rc;
    }
    //构造发送时的ngx_chain_t结构体
    ngx_chain_t     out;
    //赋值ngx_buf_t
    out.buf = b;
    //设置next为NULL
    out.next = NULL;
    //最后一步发送包体,http框架会调用ngx_http_finalize_request方法
//结束请求
    return ngx_http_output_filter(r, &out);
}

注意点

处理方法必须是快速, 无阻塞的, ngx_http_mytest_handler 或者类似的处理方法中是不可以有耗时很长的操作的

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容