Linux下librtmp使用及编程实战

　　最近想做rtmp的推流、直播的小项目，不想直接使用FFmpeg进行推流，FFmpeg进行推流特别简单，因为它已经将编码以及librtmp都集成好了，没啥意思。FFmpeg推流的例子，在雷神的博客里可以找到。这里主要是基于librmtp，结合libx264进行压缩，进行一些实验，包括三大部分：

1. rtmp流保存
2. flv文件推流
3. h264推流

首先是rtmp流保存

/*
 * RTMPRec.cpp
 *
 *  Created on: Jan 11, 2017
 *      Author: tla001
 */

#include "RTMPRec.h"
#include "sockInit.h"

RTMPRec::RTMPRec(const string url,const string filename) {
    // TODO Auto-generated constructor stub
    rtmpUrl=url;
    outFile=filename;
    bufSize=**;
    buf=new char[bufSize];
    countSize=;
    b_live_stream=true;
    rtmp=RTMP_Alloc();
}

RTMPRec::~RTMPRec() {
    // TODO Auto-generated destructor stub
     if (fp != NULL) {
            fclose(fp);
            fp = NULL;
        }

    if (buf != NULL) {
        delete[] buf;
        buf = NULL;
    }
    CleanupSockets();
    if (rtmp != NULL) {
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
        rtmp = NULL;
    }
}

int RTMPRec::init(){

    fp=fopen(outFile.c_str(),"wb");
    if(NULL==fp){
        RTMP_LogPrintf("Open File Error.\n");
        return -;
    }
    InitSockets();
    RTMP_Init(rtmp);
    //set connection timeout,default 30s
    rtmp->Link.timeout=;
    if (!RTMP_SetupURL(rtmp,const_cast<char*>(rtmpUrl.c_str()))) {
            RTMP_Log(RTMP_LOGERROR, "SetupURL Err\n");
            RTMP_Free(rtmp);
            return -;
        }
    if (b_live_stream) {
        rtmp->Link.lFlags |= RTMP_LF_LIVE;
    }

    //1hour
    RTMP_SetBufferMS(rtmp,  * );

    if (!RTMP_Connect(rtmp, NULL)) {
        RTMP_Log(RTMP_LOGERROR, "Connect Err\n");
        RTMP_Free(rtmp);
        return -;
    }

    if (!RTMP_ConnectStream(rtmp, )) {
        RTMP_Log(RTMP_LOGERROR, "ConnectStream Err\n");
        RTMP_Free(rtmp);
        RTMP_Close(rtmp);
        return -;
    }
}
void RTMPRec::run(){
    worker();
}
void RTMPRec::worker(){
    int nread;
    while ((nread = RTMP_Read(rtmp, buf, bufSize)) != ) {
        fwrite(buf, , (size_t)nread, fp);
        memset(buf,,bufSize);
        countSize += nread;
        RTMP_LogPrintf("Receive: %5dByte, Total: %5.2fkB\n", nread, countSize * 1.0 / );
    }
}
void RTMPRec::doSave(){
    this->start();
}

推送flv文件

　　主要指根据flv文件储存结构进行读取与解析的

/*
 * RTMPPushFlv.cpp
 *
 *  Created on: Jan 11, 2017
 *      Author: tla001
 */

#include "RTMPPushFlv.h"
#include "sockInit.h"
RTMPPushFlv::RTMPPushFlv(const string url) {
    // TODO Auto-generated constructor stub
    rtmpUrl=url;
    fp=NULL;
     start_time = ;
     now_time = ;

     pre_frame_time = ;
     lasttime = ;
     b_next_is_key = ;
     pre_tag_size = ;

     type = ;
     datalength = ;
     timestamp = ;

     rtmp = RTMP_Alloc();
}

RTMPPushFlv::~RTMPPushFlv() {
    // TODO Auto-generated destructor stub
    if (fp != NULL) {
        fclose(fp);
        fp = NULL;
    }
    CleanupSockets();
    if (rtmp != NULL) {
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
        rtmp = NULL;
    }
    if (p_file_buf != NULL) {
        free(p_file_buf);
        p_file_buf = NULL;
    }
}

int RTMPPushFlv::init(const string filename){
    inFile=filename;
     fp = fopen(inFile.c_str(), "rb");
    if (NULL == fp) {
        log_err("Open File Error");
        return -;
    }
    InitSockets();
    RTMP_Init(rtmp);
    //set connection timeout,default 30s
    rtmp->Link.timeout = ;
    if (!RTMP_SetupURL(rtmp, const_cast<char*>(rtmpUrl.c_str()))) {
        RTMP_Log(RTMP_LOGERROR, "SetupURL Err\n");
        //RTMP_Free(rtmp);
        return -;
    }
    RTMP_EnableWrite(rtmp);
    if (!RTMP_Connect(rtmp, NULL)) {
        RTMP_Log(RTMP_LOGERROR, "Connect Err\n");
        //RTMP_Free(rtmp);
        return -;
    }

    if (!RTMP_ConnectStream(rtmp, )) {
        RTMP_Log(RTMP_LOGERROR, "ConnectStream Err\n");
        //RTMP_Close(rtmp);
        //RTMP_Free(rtmp);
        return -;
    }

    //jump over FLV Header
    fseek(fp, , SEEK_SET);
    //jump over previousTagSizen
    fseek(fp, , SEEK_CUR);
    return ;
}

void RTMPPushFlv::run(){
    worker();
}
void RTMPPushFlv::worker(){
    log_info("Start to send data ...");
    start_time = RTMP_GetTime();
    while () {
        if ((((now_time = RTMP_GetTime()) - start_time)
                < (pre_frame_time)) && b_next_is_key) {
            //wait for 1 sec if the send process is too fast
            //this mechanism is not very good,need some improvement
            if (pre_frame_time > lasttime) {
                RTMP_LogPrintf("TimeStamp:%8lu ms\n", pre_frame_time);
                lasttime = pre_frame_time;
            }
            sleep();
            continue;
        }

        //jump over type
        fseek(fp, , SEEK_CUR);
        if (!ReadU24(&datalength, fp)) {
            break;
        }
        if (!ReadTime(&timestamp, fp)) {
            break;
        }
        //jump back
        fseek(fp, -, SEEK_CUR);

        p_file_buf = (char *) malloc( + datalength + );
        memset(p_file_buf, ,  + datalength + );
        if (fread(p_file_buf, ,  + datalength + , fp) != ( + datalength + )) {
            break;
        }

        pre_frame_time = timestamp;

        if (!RTMP_IsConnected(rtmp)) {
            RTMP_Log(RTMP_LOGERROR, "rtmp is not connect\n");
            break;
        }
        if (!RTMP_Write(rtmp, p_file_buf,  + datalength + )) {
            RTMP_Log(RTMP_LOGERROR, "Rtmp Write Error\n");
            break;
        }

        free(p_file_buf);
        p_file_buf = NULL;

        if (!PeekU8(&type, fp)) {
            break;
        }
        if (0x09 == type) {
            if (fseek(fp, , SEEK_CUR) != ) {
                break;
            }
            if (!PeekU8(&type, fp)) {
                break;
            }
            if (type == 0x17) {
                b_next_is_key = ;
            } else {
                b_next_is_key = ;
            }
            fseek(fp, -, SEEK_CUR);
        }
    }
    log_info("Send Data Over");
}
void RTMPPushFlv::doPush(){
    this->start();
}

h264数据推流

　　包括yuv数据的读入，h264编码，librtmp分包传输

　　这里对参考的库进行了必要的修改，可以实现设置参数后，不同数据格式可以自动转换为420p。

/*
 * RTMPPushH264.cpp
 *
 *  Created on: Jan 12, 2017
 *      Author: tla001
 */

#include "RTMPPushH264.h"
#include "librtmp/log.h"

int runflag=;
static void sig_user(int signo){
    if(signo==SIGINT){
        runflag=;
        printf("received SIGINT\n");
    }
}

void pushYUVByH264(){

    char url[]="rtmp://localhost/live/test1";
    int width=;
    int height=;
    int outSize=;
    int baseFrameSize=width*height;
    const long bufferSize=baseFrameSize*;
    char buffer[bufferSize];

    int fps=;
    int rate=;

    char *frame=NULL;

    if(signal(SIGINT,sig_user)==SIG_ERR)
        perror("catch SIGINT err");

    FILE* fp  = fopen("test_640x360_yuv420p.yuv", "rb");

    enum AVPixelFormat src_pix_fmt=AV_PIX_FMT_YUV420P;
    RTMP_CreatePublish(url,outSize,,RTMP_LOGINFO);
    printf("connected \n");
    RTMP_InitVideoParams(width,height,fps,rate,src_pix_fmt,false);
    printf("inited \n");
    runflag=;
    unsigned int tick = ;
    unsigned int tick_gap = /fps;
    uint32_t now=,last_update=;
    int index=;
    while(runflag){
        if(index!=){
            RTMP_SendScreenCapture((char*)buffer,height,tick);
            printf("send frame index -- %d\n",index);
        }
        last_update=RTMP_GetTime();
        switch(src_pix_fmt){
            case AV_PIX_FMT_YUV420P:
                if (fread(buffer, , baseFrameSize*/, fp) != baseFrameSize*/){
                    // Loop
                    fseek(fp, , SEEK_SET);
                    fread(buffer, , baseFrameSize*/, fp);
                    //fclose(fp);
                    //break;
                }
                printf("read file \n");
                break;
            case AV_PIX_FMT_YUV422P:
                if (fread(buffer, , baseFrameSize*, fp) != baseFrameSize*){
                    // Loop
                    fseek(fp, , SEEK_SET);
                    fread(buffer, , baseFrameSize*, fp);
                    //fclose(fp);
                    //break;
                }
                    break;
            case AV_PIX_FMT_RGB24:
                if (fread(buffer, , baseFrameSize*, fp) != baseFrameSize*){
                    // Loop
                    fseek(fp, , SEEK_SET);
                    fread(buffer, , baseFrameSize*, fp);
                    //fclose(fp);
                    //break;
                }
                    break;
            default:
                printf("Not supports this format \n");
                break;
            }
        tick +=tick_gap;
        now=RTMP_GetTime();
        usleep((tick_gap-now+last_update)*);
        index++;
    }

    RTMP_DeletePublish();
    fclose(fp);
}

完整工程

　　https://github.com/tla001/RTMPApp

相关链接

　　rmtp服务器配置

　　　　http://www.cnblogs.com/tla001/p/6263215.html

　　h264 rtmp封装

　　　　https://github.com/njk888/LibRtmpH264

　　librtmp学习

　　　　http://blog.csdn.net/leixiaohua1020/article/details/15814587