nginx之rtmp模块引用计数设计

对于没有垃圾自动回收机制的语言（如C/C++)来说，引用计数提供了一种简洁高效的内存管理方法， 

nginx-rtmp模块为方便管理内存，在给ngx_chain_t分配内存的时候，多分配了sizeof(uint32_t)类型的字节数,用来方便管理内存

在ngx_rtmp_shared.c文件中

#define NGX_RTMP_REFCOUNT_TYPE              uint32_t
#define NGX_RTMP_REFCOUNT_BYTES            sizeof(NGX_RTMP_REFCOUNT_TYPE)

#define ngx_rtmp_ref(b)                     *((NGX_RTMP_REFCOUNT_TYPE*)(b) - 1)
#define ngx_rtmp_ref_set(b, v)              ngx_rtmp_ref(b) = v
#define ngx_rtmp_ref_get(b)                 ++ngx_rtmp_ref(b)
#define ngx_rtmp_ref_put(b)                --ngx_rtmp_ref(b)

//给chain分配内存

ngx_chain_t * ngx_rtmp_alloc_shared_buf(ngx_rtmp_core_srv_conf_t *cscf)
{
    u_char                     *p;
    ngx_chain_t                *out;
    ngx_buf_t                  *b;
    size_t                      size;

    if (cscf->free) {
        out = cscf->free;
        cscf->free = out->next;

    } else {
        size = cscf->chunk_size + NGX_RTMP_MAX_CHUNK_HEADER;
        //在给chain分配内存的时候，在chain前面多分配了NGX_RTMP_REFCOUNT_BYTES字节的内存，也就是4个字节，这4字节的值初始化的时候为0
        p = ngx_pcalloc(cscf->pool, NGX_RTMP_REFCOUNT_BYTES 
                + sizeof(ngx_chain_t)
                + sizeof(ngx_buf_t)
                + size);
        if (p == NULL) {
            return NULL;
        }

//p指针后移NGX_RTMP_REFCOUNT_BYTES 个字节，也就是然后再覆盖out即分配的chain
        p += NGX_RTMP_REFCOUNT_BYTES;
        out = (ngx_chain_t *)p;
        p += sizeof(ngx_chain_t);
        out->buf = (ngx_buf_t *)p;

        p += sizeof(ngx_buf_t);
        out->buf->start = p;
        out->buf->end = p + size;
    }

    out->next = NULL;
    b = out->buf;
    b->pos = b->last = b->start + NGX_RTMP_MAX_CHUNK_HEADER;
    b->memory = 1;

    //刚分配的时候，给引用计数置1
    ngx_rtmp_ref_set(out, 1);

    return out;
}

void ngx_rtmp_free_shared_chain(ngx_rtmp_core_srv_conf_t *cscf, ngx_chain_t *in)
{
    ngx_chain_t        *cl;

    //在释放内存的时候，先判断引用计数是否为0，不为0直接返回
    if (ngx_rtmp_ref_put(in)) {
        return;
    }

   //为0 的时候，将当前chain放到需要释放的链表当中，下次循环利用
    for (cl = in; ; cl = cl->next) {
        if (cl->next == NULL) {
            cl->next = cscf->free;
            cscf->free = in;
            return;
        }
    }
}