Nginx模块开发（2）————下载文件
2024-09-06 22:51:46
Nginx的HTTP模块下载文件和传送缓冲区的字符串差不多，只需将文件标志置为1即可，我转送的文件是mp3的，所以HTTP的那个mine 类型要写为audio/mp3，二话不说了，贴代码，代码和之前那个helloworld差不多。
//start from the very beginning,and to create greatness
//@author: Chuangwei Lin
//@E-mail：979951191@qq.com
//@brief：nginx的HTTP模块:发送文件
#include <ngx_config.h>//包含必要的头文件
#include <ngx_core.h>
#include <ngx_http.h>
//先声明函数
static char *ngx_http_lcwsendfile(ngx_conf_t *cf, ngx_command_t *cmd, void *conf);
static ngx_int_t ngx_http_lcwsendfile_handler(ngx_http_request_t *r);
//ngx_command_t定义模块的配置文件参数
static ngx_command_t ngx_http_lcwsendfile_commands[] =
{
{
//配置项名称，就是配置文件里面的那个名称
ngx_string("lcwsendfile"),
//配置项类型，将指定配置项可以出现的位置
//例如出现在server{}或location{}中，以及他可以携带的参数个数
NGX_HTTP_MAIN_CONF | NGX_HTTP_SRV_CONF | NGX_HTTP_LOC_CONF | NGX_HTTP_LMT_CONF | NGX_CONF_NOARGS,
//ngx_command_t结构体中的set成员，
//当在某个配置块中出现lcwsendfile配置项时，Nginx将会调用ngx_http_lcwsendfile方法
//ngx_http_lcwsendfile方法将在下面实现
ngx_http_lcwsendfile,
//在配置文件中的偏移量conf
NGX_HTTP_LOC_CONF_OFFSET,
//offset通常用于使用预设的解析方法解析配置项，需要与conf配合使用
0,
//配置项读取后的处理方法，必须是ngx_conf_post_t结构的指针
NULL
},
//ngx_null_command是一个空的ngx_command_t结构，用来表示数组的结尾
ngx_null_command
};
//ngx_http_module_t的8个回调方法，因为目前没有什么工作是必须在HTTP框架初始化
//时完成的，所以暂时不必实现ngx_http_module_t的8个回调方法
static ngx_http_module_t ngx_http_lcwsendfile_module_ctx =
{
NULL, // preconfiguration解析配置文件前调用
NULL, // postconfiguration 完成配置文件解析后调用
NULL, // create main configuration当需要创建数据结构用于存储main级别的
//(直属于http{}块的配置项)的全局配置项时
NULL, // init main configuration常用于初始化main级别的配置项
NULL, // create server configuration当需要创建数据结构用于存储srv级别的
//(直属于server{}块的配置项)的配置项时
NULL, // merge server configuration用于合并main级别和srv级别下的同名配置项
NULL, // create location configuration 当需要创建数据结构用于存储loc级别的
//(直属于location{}块的配置项)的配置项时
NULL // merge location configuration 用于合并srv和loc级别下的同名配置项
};
//定义lcwsendfile模块
//lcwtest模块在编译时会被加入到ngx_modules全局数组中
//Nginx在启动时，会调用所有模块的初始化回调方法
//HTTP框架初始化时会调用ngx_http_module_t中的8个方法
//HTTP模块数据结构
ngx_module_t ngx_http_lcwsendfile_module =
{
NGX_MODULE_V1,//该宏为下面的ctx_index,index，spare0，spare1，spare2，spare3，version变量
//提供了初始化的值：0,0,0,0,0,0,1
//ctx_index表示当前模块在这类模块中的序号
//index表示当前模块在所有模块中的序号，Nginx启动时会根据ngx_modules数组设置各模块的index值
//spare0 spare系列的保留变量，暂未使用
//spare1
//spare2
//spare3
//version模块的版本，便于将来的扩展，目前只有一种，默认为1
&ngx_http_lcwsendfile_module_ctx, //ctx用于指向一类模块的上下文结构
ngx_http_lcwsendfile_commands, //commands将处理nginx.conf中的配置项
NGX_HTTP_MODULE, //模块的类型，与ctx指针紧密相关，取值范围是以下5种：
//NGX_HTTP_MODULE,NGX_CORE_MODULE,NGX_CONF_MODULE,NGX_EVENT_MODULE,NGX_MAIL_MODULE
//以下7个函数指针表示有7个执行点会分别调用这7种方法，对于任一个方法而言，如果不需要nginx在某个是可执行它
//那么简单地将他设为空指针即可
NULL, //master进程启动时回调init_master
NULL, //init_module回调方法在初始化所有模块时被调用，在master/worker模式下，
//这个阶段将在启动worker子进程前完成
NULL, //init_process回调方法在正常服务前被调用，在master/worker模式下，
//多个worker子进程已经产生，在每个worker子进程的初始化过程会调用所有模块的init_process函数
NULL, //由于nginx暂不支持多线程模式，所以init thread在框架代码中没有被调用过
NULL, // exit thread,也不支持
NULL, //exit process回调方法将在服务停止前调用，在master/worker模式下，worker进程会在退出前调用它
NULL, //exit master回调方法将在master进程退出前被调用
NGX_MODULE_V1_PADDING //这里是8个spare_hook变量，是保留字段，目前没有使用，Nginx提供了NGX_MODULE_V1_PADDING宏来填充
};
/******************************************************
函数名：ngx_http_lcwsendfile(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
参数：
功能：lcwsendfile方法的实现
*******************************************************/
static char* ngx_http_lcwsendfile(ngx_conf_t *cf, ngx_command_t *cmd, void *conf)
{
ngx_http_core_loc_conf_t *clcf;
//首先找到lcwtest配置项所属的配置块，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与lcwsendfile配置项所在的配置块相匹配，就将调用我们
//实现的ngx_http_lcwsendfile_handler方法处理这个请求
//ngx_http_lcwsendfile_handler将在下面实现
clcf->handler = ngx_http_lcwsendfile_handler;
return NGX_CONF_OK;
}
/******************************************************
函数名：ngx_http_lcwsendfile_handler(ngx_http_request_t *r)
参数：ngx_http_request_t结构体
功能：ngx_http_lcwtest_handler方法的实现
*******************************************************/
static ngx_int_t ngx_http_lcwsendfile_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结构准备发送包体
ngx_buf_t *b;
//开辟ngx_buf_t缓冲区
b = ngx_palloc(r->pool, sizeof(ngx_buf_t));
//要打开的文件
u_char* filename = (u_char*)"/mnt/hgfs/lcw_program/张杰 - My Sunshine.mp3";
b->in_file = 1;//将in_file设置为1就表示ngx_buf_t缓冲区传送的是文件而不是内存
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;
//获取文件长度ngx_file_info是linux系统中stat的一个宏
if (ngx_file_info(filename, &b->file->info) == NGX_FILE_ERROR)
{
return NGX_HTTP_INTERNAL_SERVER_ERROR;
}
//设置缓冲区指向的文件块，这里是告诉Nginx从文件的file_pos偏移量开始发送文件
//一直到达file_last偏移处截止
b->file_pos = 0;
b->file_last = b->file->info.st_size;
//接下来要关闭文件句柄，否则会出现句柄泄露问题，于是定义如下结构体
//ngx_pool_cleanup_add用于告诉HTTP框架，在请求结束时调用cln的handler方法清理资源
ngx_pool_cleanup_t* cln = ngx_pool_cleanup_add(r->pool, sizeof(ngx_pool_cleanup_file_t));
if (cln == NULL)
{
return NGX_ERROR;
}
//赋值执行实际清理资源工作的回调方法，ngx_pool_cleanup_file方法的作用是把文件句柄关闭
//ngx_pool_cleanup_file方法需要一个ngx_pool_cleanup_file_t类型的参数
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_pool_cleanup_file_t参数传给ngx_pool_cleanup_file方法？
//设置返回的Content-Type。注意，ngx_str_t有一个很方便的初始化宏
//ngx_string，它可以把ngx_str_t的data和len成员都设置好
ngx_str_t type = ngx_string("audio/mp3");
//设置返回状态码
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);
}配置文件也是差不多，编译方法也一样。不过我总是遇到一个问题，就是在windows新建的config文件有时候好像被识别不了，最后我在linux下写这个文件就可以了有点奇怪
还有编译之后把模块加进nginx里，然后好像上次的那个就没有了，难道以后都得全部加进去吗？
运行可以得到下面的结果：
下载成功，
是张杰的MySunshine，文件没有损坏，可以听
最新文章
- Eclipse插件开发中对于Jar包和类文件引用的处理（彻底解决插件开发中的NoClassDefFoundError问题）（转）
- ORA-00988: missing or invalid password(s)
- python实用笔记,加快编程速度，lamdba,三元运算,open.
- VC++ 线程同步 总结
- UML类图关系大全
- php大力力 [010节]PHP常量
- WPF、Windows Forms和Silverlight间的联系和区别（转）
- acess() 判断目录是否存在
- HDU 3487 Splay
- 然爸读书笔记（2013-5）----Rework(重来)
- Injection Attacks-XML注入
- redis 中文手册
- Mac os 进行Android开发笔记（2）
- Excel 帮助类
- 一起来学linux：SSH远程登陆
- 摆脱命令行，Ubuntu下配置Android开发环境
- sublime实现背景透明化
- windows 安装xadmin
- windows环境下python编码问题
- mybatis入门篇：存储过程的操作