cmake+librtmp实现推拉流

作者: 何亮hook_8285 | 来源:发表于2023-01-28 21:13 被阅读0次

微信原生组件｜基于小程序实现音视频通话
Nginx搭建RTMP推拉流服务器
直播推拉流
视频推拉流
从零开始使用nginx-rtmp-module搭建RTMP普通推
1.推拉流
ffmpeg直播推拉流
java实现直播（推拉流）和保存直播视频
ffmpeg+nginx+rtmp实现推拉流，搭建实时视频流场景
iOS-直播推拉流

cmake_minimum_required(VERSION 3.16)
#使用静态编译环境
set(VCPKG_TARGET_TRIPLET x64-windows-static)
set(VCPKG_ROOT D:/vcpkg/scripts/buildsystems/vcpkg.cmake)
set(CMAKE_TOOLCHAIN_FILE ${VCPKG_ROOT})
project(demo_rtmp)
if(MSVC)
    set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} /utf-8")
endif()
set(CMAKE_CXX_STANDARD 11)

#查找openssl 加解密库
find_package(OpenSSL REQUIRED)
#查找zlib库
find_package(ZLIB REQUIRED)

#查找librtmp头文件目录
find_path(LIBRTMP_INCLUDE_DIR NAMES rtmp.h rtmp_sys.h PATH_SUFFIXES librtmp)
include_directories(${LIBRTMP_INCLUDE_DIR})
#查询librtmp库文件路径
find_library(LIBRTMP_LIBRARIES librtmpd)

#link_libraries用来链接静态库
link_libraries(OpenSSL::SSL OpenSSL::Crypto)
link_libraries(ZLIB::ZLIB)
link_libraries(${LIBRTMP_LIBRARIES})

add_executable(${PROJECT_NAME} main.cpp)
target_link_libraries(${PROJECT_NAME} ws2_32 winmm)

librtmp推送FLV文件

#include <iostream>

#include <librtmp/rtmp_sys.h>
#include <librtmp/log.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdint.h>
#ifndef WIN32
#include <unistd.h>
#endif



#define HTON16(x)  ((x>>8&0xff)|(x<<8&0xff00))
#define HTON24(x)  ((x>>16&0xff)|(x<<16&0xff0000)|(x&0xff00))
#define HTON32(x)  ((x>>24&0xff)|(x>>8&0xff00)|\
   (x<<8&0xff0000)|(x<<24&0xff000000))
#define HTONTIME(x) ((x>>16&0xff)|(x<<16&0xff0000)|(x&0xff00)|(x&0xff000000))

/*read 1 byte*/
int ReadU8(uint32_t *u8,FILE*fp){
    if(fread(u8,1,1,fp)!=1)
        return 0;
    return 1;
}
/*read 2 byte*/
int ReadU16(uint32_t *u16,FILE*fp){
    if(fread(u16,2,1,fp)!=1)
        return 0;
    *u16=HTON16(*u16);
    return 1;
}
/*read 3 byte*/
int ReadU24(uint32_t *u24,FILE*fp){
    if(fread(u24,3,1,fp)!=1)
        return 0;
    *u24=HTON24(*u24);
    return 1;
}
/*read 4 byte*/
int ReadU32(uint32_t *u32,FILE*fp){
    if(fread(u32,4,1,fp)!=1)
        return 0;
    *u32=HTON32(*u32);
    return 1;
}
/*read 1 byte,and loopback 1 byte at once*/
int PeekU8(uint32_t *u8,FILE*fp){
    if(fread(u8,1,1,fp)!=1)
        return 0;
    fseek(fp,-1,SEEK_CUR);
    return 1;
}
/*read 4 byte and convert to time format*/
int ReadTime(uint32_t *utime,FILE*fp){
    if(fread(utime,4,1,fp)!=1)
        return 0;
    *utime=HTONTIME(*utime);
    return 1;
}

int InitSockets()
{
#ifdef WIN32
    WORD version;
    WSADATA wsaData;
    version=MAKEWORD(2,2);
    return (WSAStartup(version, &wsaData) == 0);
#endif
}

void CleanupSockets()
{
#ifdef WIN32
    WSACleanup();
#endif
}

////使用RTMP_SendPacket()函数推流
int publish_using_packet(){
    RTMP *rtmp=NULL;
    RTMPPacket *packet=NULL;
    uint32_t start_time=0;
    uint32_t now_time=0;
    //the timestamp of the previous frame
    long pre_frame_time=0;
    long lasttime=0;
    int bNextIsKey=1;
    uint32_t preTagsize=0;

    //packet attributes
    uint32_t type=0;
    uint32_t datalength=0;
    uint32_t timestamp=0;
    uint32_t streamid=0;

    FILE*fp=NULL;
    fp=fopen("d:\\destination.flv","rb");
    if (!fp){
        RTMP_LogPrintf("Open File Error.\n");
        CleanupSockets();
        return -1;
    }

    /* set log level */
    //RTMP_LogLevel loglvl=RTMP_LOGDEBUG;
    //RTMP_LogSetLevel(loglvl);

    if (!InitSockets()){
        RTMP_LogPrintf("Init Socket Err\n");
        return -1;
    }

    rtmp=RTMP_Alloc();
    RTMP_Init(rtmp);
    //set connection timeout,default 30s
    rtmp->Link.timeout=5;
    if(!RTMP_SetupURL(rtmp,"rtmp://127.0.0.1/live/test"))
    {
        RTMP_Log(RTMP_LOGERROR,"SetupURL Err\n");
        RTMP_Free(rtmp);
        CleanupSockets();
        return -1;
    }

    //if unable,the AMF command would be 'play' instead of 'publish'
    RTMP_EnableWrite(rtmp);

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

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

    packet=(RTMPPacket*)malloc(sizeof(RTMPPacket));
    RTMPPacket_Alloc(packet,1024*64);
    RTMPPacket_Reset(packet);

    packet->m_hasAbsTimestamp = 0;
    packet->m_nChannel = 0x04;
    packet->m_nInfoField2 = rtmp->m_stream_id;

    RTMP_LogPrintf("Start to send data ...\n");

    //jump over FLV Header
    fseek(fp,9,SEEK_SET);
    //jump over previousTagSizen
    fseek(fp,4,SEEK_CUR);
    start_time=RTMP_GetTime();
    while(1)
    {
        if((((now_time=RTMP_GetTime())-start_time)
            <(pre_frame_time)) && bNextIsKey){
            //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(1000);
            continue;
        }

        //not quite the same as FLV spec
        if(!ReadU8(&type,fp))
            break;
        if(!ReadU24(&datalength,fp))
            break;
        if(!ReadTime(&timestamp,fp))
            break;
        if(!ReadU24(&streamid,fp))
            break;

        if (type!=0x08&&type!=0x09){
            //jump over non_audio and non_video frame，
            //jump over next previousTagSizen at the same time
            fseek(fp,datalength+4,SEEK_CUR);
            continue;
        }

        if(fread(packet->m_body,1,datalength,fp)!=datalength)
            break;

        packet->m_headerType = RTMP_PACKET_SIZE_LARGE;
        packet->m_nTimeStamp = timestamp;
        packet->m_packetType = type;
        packet->m_nBodySize  = datalength;
        pre_frame_time=timestamp;

        if (!RTMP_IsConnected(rtmp)){
            RTMP_Log(RTMP_LOGERROR,"rtmp is not connect\n");
            break;
        }
        if (!RTMP_SendPacket(rtmp,packet,0)){
            RTMP_Log(RTMP_LOGERROR,"Send Error\n");
            break;
        }

        if(!ReadU32(&preTagsize,fp))
            break;

        if(!PeekU8(&type,fp))
            break;
        if(type==0x09){
            if(fseek(fp,11,SEEK_CUR)!=0)
                break;
            if(!PeekU8(&type,fp)){
                break;
            }
            if(type==0x17)
                bNextIsKey=1;
            else
                bNextIsKey=0;

            fseek(fp,-11,SEEK_CUR);
        }
    }

    RTMP_LogPrintf("\nSend Data Over\n");

    if(fp)
        fclose(fp);

    if (rtmp!=NULL){
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
        rtmp=NULL;
    }
    if (packet!=NULL){
        RTMPPacket_Free(packet);
        free(packet);
        packet=NULL;
    }

    CleanupSockets();
    return 0;
}

//使用RTMP_Write()函数推流
int publish_using_write(){
    uint32_t start_time=0;
    uint32_t now_time=0;
    uint32_t pre_frame_time=0;
    uint32_t lasttime=0;
    int bNextIsKey=0;
    char* pFileBuf=NULL;

    //read from tag header
    uint32_t type=0;
    uint32_t datalength=0;
    uint32_t timestamp=0;

    RTMP *rtmp=NULL;

    FILE*fp=NULL;
    fp=fopen("d:\\destination.flv","rb");
    if (!fp){
        RTMP_LogPrintf("Open File Error.\n");
        CleanupSockets();
        return -1;
    }

    /* set log level */
    //RTMP_LogLevel loglvl=RTMP_LOGDEBUG;
    //RTMP_LogSetLevel(loglvl);

    if (!InitSockets()){
        RTMP_LogPrintf("Init Socket Err\n");
        return -1;
    }

    rtmp=RTMP_Alloc();
    RTMP_Init(rtmp);
    //set connection timeout,default 30s
    rtmp->Link.timeout=5;
    if(!RTMP_SetupURL(rtmp,"rtmp://127.0.0.1/live/test"))
    {
        RTMP_Log(RTMP_LOGERROR,"SetupURL Err\n");
        RTMP_Free(rtmp);
        CleanupSockets();
        return -1;
    }

    RTMP_EnableWrite(rtmp);
    //1hour
    RTMP_SetBufferMS(rtmp, 3600*1000);
    if (!RTMP_Connect(rtmp,NULL)){
        RTMP_Log(RTMP_LOGERROR,"Connect Err\n");
        RTMP_Free(rtmp);
        CleanupSockets();
        return -1;
    }

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

    printf("Start to send data ...\n");

    //jump over FLV Header
    fseek(fp,9,SEEK_SET);
    //jump over previousTagSizen
    fseek(fp,4,SEEK_CUR);
    start_time=RTMP_GetTime();
    while(1)
    {
        if((((now_time=RTMP_GetTime())-start_time)
            <(pre_frame_time)) && bNextIsKey){
            //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(1000);
            continue;
        }

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

        pFileBuf=(char*)malloc(11+datalength+4);
        memset(pFileBuf,0,11+datalength+4);
        if(fread(pFileBuf,1,11+datalength+4,fp)!=(11+datalength+4))
            break;

        pre_frame_time=timestamp;

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

        free(pFileBuf);
        pFileBuf=NULL;

        if(!PeekU8(&type,fp))
            break;
        if(type==0x09){
            if(fseek(fp,11,SEEK_CUR)!=0)
                break;
            if(!PeekU8(&type,fp)){
                break;
            }
            if(type==0x17)
                bNextIsKey=1;
            else
                bNextIsKey=0;
            fseek(fp,-11,SEEK_CUR);
        }
    }

    RTMP_LogPrintf("\nSend Data Over\n");

    if(fp)
        fclose(fp);

    if (rtmp!=NULL){
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
        rtmp=NULL;
    }

    if(pFileBuf){
        free(pFileBuf);
        pFileBuf=NULL;
    }

    CleanupSockets();
    return 0;
}
//heliang
int main(int argc, char* argv[]){
   
    publish_using_packet();
//    publish_using_write();
    return 0;
}

librtmp拉流并保存FLV文件

#include <iostream>

#include <librtmp/rtmp_sys.h>
#include <librtmp/log.h>
int InitSockets()
{
#ifdef WIN32
    WORD version;
    WSADATA wsaData;
    version = MAKEWORD(1, 1);
    return (WSAStartup(version, &wsaData) == 0);
#endif
}

void CleanupSockets()
{
#ifdef WIN32
    WSACleanup();
#endif
}

//heliang
//ffmpeg -re -stream_loop -1 -i d:\\destination.flv  -vcodec h264 -acodec aac -f flv rtmp://127.0.0.1/live/test
int main(int argc, char* argv[])
{
    InitSockets();

    double duration=-1;
    int nRead;
    //is live stream ?
    bool bLiveStream=true;


    int bufsize=1024*1024*10;
    char *buf=(char*)malloc(bufsize);
    memset(buf,0,bufsize);
    long countbufsize=0;

    FILE *fp=fopen("d:\\receive.flv","wb");
    if (!fp){
        RTMP_LogPrintf("Open File Error.\n");
        CleanupSockets();
        return -1;
    }

    /* set log level */
    //RTMP_LogLevel loglvl=RTMP_LOGDEBUG;
    //RTMP_LogSetLevel(loglvl);

    RTMP *rtmp=RTMP_Alloc();
    RTMP_Init(rtmp);
    //set connection timeout,default 30s
    rtmp->Link.timeout=10;
    // HKS's live URL
    if(!RTMP_SetupURL(rtmp,"rtmp://127.0.0.1/live/test"))
    {
        RTMP_Log(RTMP_LOGERROR,"SetupURL Err\n");
        RTMP_Free(rtmp);
        CleanupSockets();
        return -1;
    }
    if (bLiveStream){
        rtmp->Link.lFlags|=RTMP_LF_LIVE;
    }

    //1hour
    RTMP_SetBufferMS(rtmp, 3600*1000);

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

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

    while(nRead=RTMP_Read(rtmp,buf,bufsize)){
        fwrite(buf,1,nRead,fp);

        countbufsize+=nRead;
        RTMP_LogPrintf("Receive: %5dByte, Total: %5.2fkB\n",nRead,countbufsize*1.0/1024);
    }

    if(fp)
        fclose(fp);

    if(buf){
        free(buf);
    }

    if(rtmp){
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
        CleanupSockets();
        rtmp=NULL;
    }
    return 0;
}

微信原生组件｜基于小程序实现音视频通话
1 微信小程序原生推拉流组件功能简介本文将介绍如何使用微信小程序原生推拉流组件和进行推拉流，快速实现...
Nginx搭建RTMP推拉流服务器
如题，今天就来实现一个推拉流服务器，模拟下推流后被客户端拉流看到效果。详细步骤如下安装Nginx 安装FFmp...
直播推拉流
https://blog.csdn.net/sunxiaopengsun/article/details/6288...
视频推拉流
推流，指的是把采集阶段封包好的内容传输到服务器的过程。拉流是指服务器已有直播内容，用指定地址进行拉取的过程数据...
从零开始使用nginx-rtmp-module搭建RTMP普通推
整体步骤如下：1. 安装nginx-rtmp-module。2. 修改nginx配置，普通推拉流设置和权限验证推拉...
1.推拉流
前提： 1.已安装nginx 2.下载安装xtmp模块 nginx源码目录下 ./configure --add-...
ffmpeg直播推拉流
推流 -re减慢帧率速率-c 音视频编码方式-f flv 已flv文件格式拉流
java实现直播（推拉流）和保存直播视频
首先墨迹一下前提：公司接了个电商的项目，然后甲方要求有类似于淘宝的直播的功能。重点是！！！甲方爸爸不愿意除阿里云直...
ffmpeg+nginx+rtmp实现推拉流，搭建实时视频流场景
一、准备 1.环境：windows2.开发工具：ffmpeg、nginx（代理服务器）nginx-rmtp-mod...
iOS-直播推拉流
1 . 音视频处理的一般流程：数据采集→数据编码→数据传输(流媒体服务器) →解码数据→播放显示 1、数据采集：...