基于Nginx实现一个自己的HTTP模块--发送磁盘中的文件

来源:互联网 发布:网络水晶头接线图 编辑:程序博客网 时间:2024/06/11 04:38

文件/home/weijl/workspace/nginx-1.10.3/src/http/config内容如下:

ngx_addon_name=ngx_http_mytest_moduleHTTP_MODULES="$HTTP_MODULES ngx_http_mytest_module"NGX_ADDON_SRCS="$NGX_ADDON_SRCS $ngx_addon_dir/ngx_http_mytest_module.c"本文仅在上一篇《基于Nginx实现一个自己的HTTP模块》基础上修改只是发送响应内容是磁盘中的文件,所以仅修改方法static ngx_int_t ngx_http_mytest_handler(ngx_http_request_t *r),修改后的代码如下:
//HTTP的HTTP_CONTENT_PHASE阶段mytest模块介入处理http请求内容static ngx_int_t ngx_http_mytest_handler(ngx_http_request_t *r){//必须时GET或者HEAD方法,否则返回405 Not Allowedif(!(r->method &(NGX_HTTP_GET | NGX_HTTP_HEAD))){    ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "weijl NGX_HTTP_NOT_ALLOWED");return NGX_HTTP_NOT_ALLOWED;}//丢弃请求中的包体ngx_int_t rc = ngx_http_discard_request_body(r);if(rc != NGX_OK){ ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "weijl rc=%d", rc);return rc;}/*设置返回的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;//发送HTTP头部rc = ngx_http_send_header(r);if(rc == NGX_ERROR || rc > NGX_OK || r->header_only){ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "weijl rc=%d", rc);return rc;}//构造ngx_buf_t结构体准备发送包体ngx_buf_t *b;b = ngx_palloc(r->pool, sizeof(ngx_buf_t));if(NULL == b){    ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "weijl b=NULL");return NGX_HTTP_INTERNAL_SERVER_ERROR;}u_char* filename = (u_char*)"/home/weijl/workspace/nginx-1.10.3/src/http/config";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 = strlen((const char*)filename);if(b->file->fd <= 0){return NGX_HTTP_NOT_FOUND;}if(ngx_file_info(filename, &b->file->info) == NGX_FILE_ERROR){return NGX_HTTP_INTERNAL_SERVER_ERROR;}//响应包是由包体内容的,需要设置Conten-Length长度r->headers_out.content_length_n = b->file->info.st_size;//设置Content-Typer->headers_out.content_type = type;b->file_pos = 0;b->file_last = b->file->info.st_size;//声明这是最后一块缓冲区b->last_buf =1;//构造发送时的ngx_chain_t结构体ngx_chain_t out;out.buf = b;//设置next为NULLout.next = NULL;//清理文件句柄////在请求结束时调用cln的handler方法清理资源ngx_pool_cleanup_t* cln = ngx_pool_cleanup_add(r->pool, sizeof(ngx_pool_cleanup_file_t));if(NULL == cln){return NGX_ERROR;}//将Nginx提供的ngx_pool_cleanup_file函数设置为回调方法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;ngx_log_error(NGX_LOG_ERR, r->connection->log, 0, "weijl 构造包体完成,开始发送包体\n");//最后一步为发送包体,发送结束后HTTP框架会调用ngx_http_finalize_request方法结束请求return ngx_http_output_filter(r, &out);}

测试验证结果,如图:


0 0
原创粉丝点击