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

文件/home/weijl/workspace/nginx-1.10.3/src/http/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"

本文仅在上一篇《基于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 Allowed
		if(!(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-Type
		r->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为NULL
		out.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);
}

测试验证结果,如图:


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值