首页 > Nginx > Nginx断点续传功能代码浅析-Range模块

Nginx断点续传功能代码浅析-Range模块

2013年10月3日 发表评论 阅读评论 20443次阅读    

HTTP 的Content-Range支持对于一般的网页处理没啥重要的作用,但是对于大文件的下载,CDN回源,点续传功能的作用是非常重要的。

Content-Range允许一次只下载一个文件的一部分,后面再分批次下载文件的其他部分,或者并发下载,提高下载速度,这样如果在下载一个文件的过程中,网络断开了,恢复后不需要重新下载。

nginx 对Content-Range的支持包括header处理和body处理,分别用来解析客户端发送过来的Range header 和裁剪返回给客户端的请求数据Body。其实现分别由2个filter过滤模块完成,分别是ngx_http_range_header_filter_module和ngx_http_range_body_filter_module。下面分别介绍。

零、解析客户端发送的请求Range头

这部分很简单,nginx通过在ngx_http_headers_in请求头处理数组中增加对Range,If-Range的处理函数,从而读取到客户端发送的HEADER的时候,调用对应的函数,将请求数据设置到对应的字段上去。比如对于Range,处理函数为ngx_http_process_header_line ,Range值会放入ngx_http_headers_in_t 的 range字段上面。以便后续filter模块去处理。

ngx_http_headers_in
    { ngx_string("Range"), offsetof(ngx_http_headers_in_t, range),
                 ngx_http_process_header_line },

    { ngx_string("If-Range"),
                 offsetof(ngx_http_headers_in_t, if_range),
                 ngx_http_process_unique_header_line },

一、Header 过滤函数

ngx_http_range_header_filter_module模块回调函数中,只设置了一个ngx_http_range_header_filter_init,用来在nginx filter链表中挂载当前模块的header filter函数。这里说一下nginx对于filter函数链表的组织形式,相当巧妙:利用static 文件局部变量来保存下一个链表节点元素,类似堆栈的方式一个个将后面链接的文件组织起来,达到代码非常简单优美的效果。看下面就知道了。

ngx_http_top_header_filter为全局变量,对于ngx_http_range_filter_module.c文件来说,是static 全局变量,也就是只在本文件有效。多个filter里面各有一份这样的static变量。

//静态局部变量,用来保存下一个过滤器的函数指针。nginx通常做法,用这种方式来建立一调filter链
static ngx_http_output_header_filter_pt  ngx_http_next_header_filter;

static ngx_int_t
ngx_http_range_header_filter_init(ngx_conf_t *cf)
{//组成filter链表
    ngx_http_next_header_filter = ngx_http_top_header_filter;
    ngx_http_top_header_filter = ngx_http_range_header_filter;
    return NGX_OK;
}

ngx_int_t
ngx_http_send_header(ngx_http_request_t *r)
{
    if (r->err_status) {//如果有错误发生,则发送错误状态。
        r->headers_out.status = r->err_status;
        r->headers_out.status_line.len = 0;
    }
	//第一个是ngx_http_not_modified_header_filter,后面一个个往后掉用。每个函数独立一个文件,那里面有个静态变量记录它的上一个filter是谁。
    return ngx_http_top_header_filter(r);//最后一个是ngx_http_header_filter_module模块里面的发送模块
}

这样,在nginx需要给客户端发送头部header的时候,会调用ngx_http_send_header函数,而其代码很简单,只是调用了ngx_http_top_header_filter,因此,如果header range filter是最后一个挂载的header filter模块的话,ngx_http_top_header_filter链表头节点就等于ngx_http_range_header_filter,也就是最先调用ngx_http_range_header_filter函数进行header过滤处理。

ngx_http_range_header_filter函数用来给客户端返回的header中插入"Accept-Ranges:bytes"头,以及解析客户端发送过来的Range头,其首先检查一下客户端是否支持Range,如果发送了If-Range,则需要检查一下last_modified_time的时间,是否相等,如果不相等,表示服务器的文件比客户端已经下载的不一样了,需要重新下载全部文件。

static ngx_int_t
ngx_http_range_header_filter(ngx_http_request_t *r)
{//ngx_http_top_header_filter链的函数指针,用来过滤发送给客户端的头部数据。
//ngx_http_send_header函数会调用这里。
    time_t                        if_range;
    ngx_int_t                     rc;
    ngx_http_range_filter_ctx_t  *ctx;

    if (r->http_version < NGX_HTTP_VERSION_10
        || r->headers_out.status != NGX_HTTP_OK
        || r != r->main
        || r->headers_out.content_length_n == -1
        || !r->allow_ranges)
    {//不处理非200的请求,否则直接调用下一个。
        return ngx_http_next_header_filter(r);
    }

    if (r->headers_in.range == NULL
        || r->headers_in.range->value.len < 7
        || ngx_strncasecmp(r->headers_in.range->value.data,
                           (u_char *) "bytes=", 6)
           != 0)
    {//如果客户端发送过来的头部数据中没有range字段,则我们也不需要处理range。直接返回即可
        goto next_filter;
    }

    if (r->headers_in.if_range && r->headers_out.last_modified_time != -1) {
		//语法为If-Range = "If-Range" ":" ( entity-tag | HTTP-date )  后面带的是时间,也就如果修改时间等于XX的话,就给我我没有的,否则给我所有的。
        if_range = ngx_http_parse_time(r->headers_in.if_range->value.data, r->headers_in.if_range->value.len);
        ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0,
                       "http ir:%d lm:%d", if_range, r->headers_out.last_modified_time);
//If-Range的意思是:“如果entity没有发生变化,那么把我缺失的部分发送给我。
//如果entity发生了变化,那么把整个entity发送给我”。
        if (if_range != r->headers_out.last_modified_time) {
            goto next_filter;//时间跟服务器上的时间不相等,需要返回所有的。所以略过
        }
    }

之后便需要解析Range头了。解析后的开始-结束区块会放入ctx->ranges数组中,用来在发送body的时候拼接http multipart 数据。这个工作主要由ngx_http_range_parse函数进行,其会解析字符串的Range格式数据,"Range: bytes=0-1024"请求一块 , “Range:bytes=500-600,601-999”代表客户端请求2块。ngx_http_range_parse不多介绍了,其主要就是做字符串解析。结果会放到ctx->ranges里面。后面拼接返回数据的时候会需要的。

解析请求头完成后,需要准备返回给客户端的头部数据,其实这里还不能拼接返回数据,只是准备了一下相关结构。如果客户端只请求了一块,则由ngx_http_range_singlepart_header完成,否则由ngx_http_range_multipart_header完成。

//解析range的语法,在本文件开头有介绍。解析出开始,结束等。结果range放入ctx->ranges里面
    rc = ngx_http_range_parse(r, ctx);

    if (rc == NGX_OK) {
        ngx_http_set_ctx(r, ctx, ngx_http_range_body_filter_module);
		//修改发送的头部字段为206
        r->headers_out.status = NGX_HTTP_PARTIAL_CONTENT;
        r->headers_out.status_line.len = 0;

        if (ctx->ranges.nelts == 1) {//单个RANGE
            return ngx_http_range_singlepart_header(r, ctx);
        }
		//多个range,比如"Range: bytes=500-600,601-999"
        return ngx_http_range_multipart_header(r, ctx);
    }

    if (rc == NGX_HTTP_RANGE_NOT_SATISFIABLE) {
        return ngx_http_range_not_satisfiable(r);
    }

对于单个range请求,比较简单,只需要在返回的头部增加"Content-Range: bytes START-END/SIZE"头,数据就放在body里面即可,不需要用multipart的方式组织数据。单个range请求的返回数据形如:

* "HTTP/1.0 206 Partial Content" CRLF
* ... header ...
* "Content-Type: image/jpeg" CRLF
* "Content-Length: SIZE" CRLF
* "Content-Range: bytes START-END/SIZE" CRLF
* CRLF
* ... data ...

static ngx_int_t
ngx_http_range_singlepart_header(ngx_http_request_t *r, ngx_http_range_filter_ctx_t *ctx)
{//客户端只请求了一个range,那么我们返回的数据格式为: Content-Range: bytes START-END/SIZE" CRLF
    ngx_table_elt_t   *content_range;
    ngx_http_range_t  *range;

    content_range = ngx_list_push(&r->headers_out.headers);//在输出头里面申请一个header line
    if (content_range == NULL) {
        return NGX_ERROR;
    }
    r->headers_out.content_range = content_range;
    content_range->hash = 1;
    ngx_str_set(&content_range->key, "Content-Range");

    content_range->value.data = ngx_pnalloc(r->pool, sizeof("bytes -/") - 1 + 3 * NGX_OFF_T_LEN);//申请三个足够长度的数字的字符串
    if (content_range->value.data == NULL) {
        return NGX_ERROR;
    }
    /* "Content-Range: bytes SSSS-EEEE/TTTT" header */
    range = ctx->ranges.elts;
//设置START-END/SIZE格式。
    content_range->value.len = ngx_sprintf(content_range->value.data,
                                           "bytes %O-%O/%O",
                                           range->start, range->end - 1, r->headers_out.content_length_n) - content_range->value.data;
    r->headers_out.content_length_n = range->end - range->start;//本次返回的数据长度。总长度在SIZE上面
//·····
    return ngx_http_next_header_filter(r);
}

但是对于一次请求多个区块的请求,其处理相对复杂。因为返回的body数据无法直接放在BODY里面发送,需要组织成multipart 的形式才行了。这个由ngx_http_range_multipart_header函数搞定。先看一下多个range请求时返回数据的格式:

"HTTP/1.0 206 Partial Content" CRLF
... header ...
"Content-Type: multipart/byteranges; boundary=0123456789" CRLF
CRLF//头部结束
CRLF
"--0123456789" CRLF
"Content-Type: image/jpeg" CRLF
"Content-Range: bytes START0-END0/SIZE" CRLF
CRLF
... data ...
CRLF
"--0123456789" CRLF
"Content-Type: image/jpeg" CRLF
"Content-Range: bytes START1-END1/SIZE" CRLF
CRLF
... data ...
CRLF
"--0123456789--" CRLF

ngx_http_range_multipart_header函数首先调用ngx_next_temp_number获取一个临时数字当做boundary,然后组成boundary_header,也就是上面黄色背景的部分,这部分是通用的,可以在组成给客户端的返回数据的时候重复使用。

static ngx_int_t
ngx_http_range_multipart_header(ngx_http_request_t *r,
    ngx_http_range_filter_ctx_t *ctx)
{
    size_t              len;
    ngx_uint_t          i;
    ngx_http_range_t   *range;
    ngx_atomic_uint_t   boundary;

    len = sizeof(CRLF "--") - 1 + NGX_ATOMIC_T_LEN
          + sizeof(CRLF "Content-Type: ") - 1
          + r->headers_out.content_type.len
          + sizeof(CRLF "Content-Range: bytes ") - 1;

    if (r->headers_out.charset.len) {
        len += sizeof("; charset=") - 1 + r->headers_out.charset.len;
    }

    ctx->boundary_header.data = ngx_pnalloc(r->pool, len);
    if (ctx->boundary_header.data == NULL) {
        return NGX_ERROR;
    }
    boundary = ngx_next_temp_number(0);
    /*
     * The boundary header of the range:
     * CRLF
     * "--0123456789" CRLF
     * "Content-Type: image/jpeg" CRLF
     * "Content-Range: bytes "
     */
//拼接bondery头,也就是--xxxxx以及后面的那几行头部数据。头部数据之后就是bonder的数据部分了。
    if (r->headers_out.charset.len) {
        ctx->boundary_header.len = ngx_sprintf(ctx->boundary_header.data,
                                           CRLF "--%0muA" CRLF
                                           "Content-Type: %V; charset=%V" CRLF
                                           "Content-Range: bytes ",
                                           boundary,
                                           &r->headers_out.content_type,
                                           &r->headers_out.charset)
                                   - ctx->boundary_header.data;

        r->headers_out.charset.len = 0;

    }
//···

然后组成"Content-Type: multipart/byteranges; boundary="这一行头部数据,放到r->headers_out.content_type上面。

//这是最开始那一行声明下面是mutipart 格式数据的行:"Content-Type: multipart/byteranges; boundary=0123456789" CRLF
//下面的sizeof其实没必要用全部字符串,Content-Type:不需要。
    r->headers_out.content_type.data =
        ngx_pnalloc(r->pool, sizeof("Content-Type: multipart/byteranges; boundary=") - 1 + NGX_ATOMIC_T_LEN);

    if (r->headers_out.content_type.data == NULL) {
        return NGX_ERROR;
    }
    r->headers_out.content_type_lowcase = NULL;
    /* "Content-Type: multipart/byteranges; boundary=0123456789" */
    r->headers_out.content_type.len =
                           ngx_sprintf(r->headers_out.content_type.data,
                                       "multipart/byteranges; boundary=%0muA", boundary)
                           - r->headers_out.content_type.data;

    r->headers_out.content_type_len = r->headers_out.content_type.len;

然后就需要一个个将range的开始,结束部分字符串进行拼接了,也就是"SSSS-EEEE/TTTT" CRLF CRLF",结果放到range[i].content_range.data上面,用来在后面body过滤函数里面组成返回body。之后就是调用下一个header filter过滤哈数ngx_http_next_header_filter。

到这里头部数据的过滤函数完毕了,总结一下就是:

1.调用ngx_http_range_parse解析客户端请求头:“Range: bytes=0-1024”。
2.如果是单个Range,调用ngx_http_range_singlepart_header拼接"Content-Range: bytes START-END/SIZE"头。
3.如果是多个Range请求,调用ngx_http_range_multipart_header准备multipart数据,以便后面body filter使用。

二、BODY 过滤函数

类似header过滤函数的处理方式,nginx挂载了body过滤函数ngx_http_range_body_filter,用来在ngx_http_output_filter里面需要给客户端发送HTTP BODY的时候调用。

当该函数很简单,就是判断一下ctx->ranges.nelts的数量,如果是单个Range请求,就调用ngx_http_range_singlepart_body完成工作,否则判断一下是否所有的ranges都在第一块buf里面,nginx目前只支持所有的ranges都在一块buf里面的情况,并且也只支持所有的都在buf链表的的一块buf里面。所以支持的比较鸡肋。如果条件满足,就调用ngx_http_range_body_filter去处理。

static ngx_int_t
ngx_http_range_body_filter(ngx_http_request_t *r, ngx_chain_t *in)
{//这是第一个BODY过滤函数
    ngx_http_range_filter_ctx_t  *ctx;

    if (in == NULL) {
        return ngx_http_next_body_filter(r, in);
    }
    ctx = ngx_http_get_module_ctx(r, ngx_http_range_body_filter_module);
    if (ctx == NULL) {
        return ngx_http_next_body_filter(r, in);
    }
    if (ctx->ranges.nelts == 1) {//如果只有一个range
        return ngx_http_range_singlepart_body(r, ctx, in);
    }
    /*下面说明,NGINX只支持整个数据都在一个buffer里面的情况也就是多个body它不好处理
     * multipart ranges are supported only if whole body is in a single buffer
     */
    if (ngx_buf_special(in->buf)) {
        return ngx_http_next_body_filter(r, in);
    }
	//检查是否所有的RANGE都在第一块buf里面。那么在后面的buf里面不行么,不行,nginx不支持
    if (ngx_http_range_test_overlapped(r, ctx, in) != NGX_OK) {
        return NGX_ERROR;
    }

    return ngx_http_range_multipart_body(r, ctx, in);
}

单个range的情况由ngx_http_range_singlepart_body完成,其相对也简单,循环扫描每个in参数的buf链表,如果其在range里面,就将其放到out的链表中,组成一个新的链表,然后调用下一个filter。下面看一下有改动的函数。

static ngx_int_t ngx_http_range_singlepart_body(ngx_http_request_t *r, ngx_http_range_filter_ctx_t *ctx, ngx_chain_t *in)
{//处理客户端只发送一个range过来的情况
    off_t              start, last;
    ngx_buf_t         *buf;
    ngx_chain_t       *out, *cl, **ll;
    ngx_http_range_t  *range;

    out = NULL;
    ll = &out;
    range = ctx->ranges.elts;
//对in参数的要发送出去的缓冲数据链表,一一检查其内容是否在range的start-end之间,
//不在的就丢弃,只剪裁出区间之内的,发送给客户端。
    for (cl = in; cl; cl = cl->next) {

        buf = cl->buf;

        start = ctx->offset;//在所有数据中的位置。
        last = ctx->offset + ngx_buf_size(buf);

        ctx->offset = last;
        ngx_log_debug2(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, "http range body buf: %O-%O", start, last);
        if (range->end <= start || range->start >= last) {//丢弃这个。将其pos指向last从而略过。
            ngx_log_debug0(NGX_LOG_DEBUG_HTTP, r->connection->log, 0, "http range body skip");
            if (buf->in_file) {
                buf->file_pos = buf->file_last;
            }
            buf->pos = buf->last;
            buf->sync = 1;
            continue;
        }
        if (range->start > start) {//开头重合,去掉重合部分
            if (buf->in_file) {
                buf->file_pos += range->start - start;//向后移动开始部分
            }
            if (ngx_buf_in_memory(buf)) {//如果在内存的话就移动指针
                buf->pos += (size_t) (range->start - start);
            }
        }

        if (range->end <= last) {//尾部重合,去掉尾部
            if (buf->in_file) {
                buf->file_last -= last - range->end;
            }
            if (ngx_buf_in_memory(buf)) {
                buf->last -= (size_t) (last - range->end);
            }
            buf->last_buf = 1;//标记为最后一块内存,后面的都不需要了。
            *ll = cl;
            cl->next = NULL;//直接剪断,后面可能还有输出链的,但是没事,等这个连接关闭后,数据都回收了的。
            break;
        }
        *ll = cl;//后移动。
        ll = &cl->next;
    }
	//到这里后,out变量所指向的链表里面的数据都是range之间的,需要发送给客户端的数据。于是调用下一个filter进行发送。
    if (out == NULL) {
        return NGX_OK;
    }
    return ngx_http_next_body_filter(r, out);
}

对于多个range的情况,首先由ngx_http_range_test_overlapped函数判断一下是否in参数的buf是最后一块,如果是最后一块那可以支持,否则,如果所有的ranges都在第一块buf里面,那也支持。其他都不能处理。
ngx_http_range_multipart_body的工作是组成mutipart格式的数据,调用下一个过滤器发送给客户端。具体不啰嗦了。
总结一下,nginx对于多个ranges的请求支持目前还不是让人满意,所以需要使用的同学建议只使用单个Range请求的方式。否则就会出现诡异的500错误而不得善终。不过好消息是:PHP没有cached的请求是不支持range的,image不支持,目前只有static请求是支持的。但是,static请求静态文件的话,一般都是整个文件都在一块buf结构里面,所以多个range也就一定在一块也是第一块buf里面,所以结论是:nginx 对range的支持能够满足大部分需求。

Share
分类: Nginx 标签: ,
  1. 青山画卷
    2014年10月13日17:51 | #1

    请问下,这些代码是用的什么语言写的?

  1. 本文目前尚无任何 trackbacks 和 pingbacks.

注意: 评论者允许使用'@user空格'的方式将自己的评论通知另外评论者。例如, ABC是本文的评论者之一,则使用'@ABC '(不包括单引号)将会自动将您的评论发送给ABC。使用'@all ',将会将评论发送给之前所有其它评论者。请务必注意user必须和评论者名相匹配(大小写一致)。