RTMP传输

weixin_43313383

于 2022-07-11 17:38:58 发布

阅读量263

点赞数

分类专栏： ffmpeg 文章标签： ffmpeg 网络

本文链接：https://blog.csdn.net/weixin_43313383/article/details/125726336

版权

ffmpeg 专栏收录该内容

12 篇文章 3 订阅

订阅专栏

一、RTMP协议

在这里插入图片描述

1. RTMP创建流的基本流程

socket建立TCP连接
RTMP握手
建立RTMP连接
创建RTMP流

2. RTMP协议中的握手

在这里插入图片描述

3. RTMP实际的握手

在这里插入图片描述

4. 协议中建立RTMP连接

在这里插入图片描述

5. 实际建立RTMP连接

在这里插入图片描述

二、创建RTMP流

1. 协议中RTMP流的创建

在这里插入图片描述

2. 实际上创建RTMP流

在这里插入图片描述

3. 推RTMP流

在这里插入图片描述

4. 播RTMP流

在这里插入图片描述

三、RTMP消息

1. RTMP消息格式

在这里插入图片描述

2. RTMP消息类型

在这里插入图片描述

四、RTMP抓包分析

可以使用wireshark来抓取RTMP数据包
在这里插入图片描述

五、FLV协议

在这里插入图片描述
FLV 协议分析器：雷神的FLV 分析器（specialFFLV）/ FLVParser库

六、推流程序流程

1. 推流步骤

解析FLV文件
获取音视频数据
利用librtmp进行推流

2. librtmp的基本用法

初始化RTMP对象
设置RTMP URL
连接流媒体服务器
发送数据包

3. 推流的代码如下：

#include <iostream>
#include <unistd.h>

//using namespace std;

//包含ffmpeg头文件
extern "C"
{
#include "libavutil/avutil.h"
#include "libavdevice/avdevice.h"
#include "libavformat/avformat.h"
#include "libavcodec/avcodec.h"
#include "libswresample/swresample.h"
#include "libavutil/audio_fifo.h"
#include "librtmp/rtmp.h"
}

#include <windows.h>
#include <vector>
#include <string>
#include <memory>
#include<winsock2.h>

using std::vector;
using std::string;
using std::shared_ptr;

/*
 * FLV Header size: 9 byte
 * byte 1-3, signature: 'F' 'L' 'V'
 * byte 4, version: 1
 * byte 5, 0-5 bit: 保留, 必须是 0
 * byte 5, 6 bit: 是否有音频 Tag
 * byte 5, 7 bit: 保留，必须是 0
 * byte 5, 8 bit: 是否有视频 Tag
 * byte 6-9, Header Size: 必须是9
 */
static FILE *open_flv(char *flv_name)
{
    FILE *fp = nullptr;

    fp = fopen(flv_name, "rb");
    if (!fp) {
        printf("Failed to open flv: %s", flv_name);
        return fp;
    }

    fseek(fp, 9, SEEK_SET); // 9 字节的 FLV Header
    fseek(fp, 4, SEEK_CUR); // 4 字节的 PreTabSize

    return fp;
}

static RTMP *conect_rtmp_server(char *rtmpaddr)
{
    RTMP *rtmp = nullptr;

    //1. 创建RTMP对象，并进行初始化
    rtmp = RTMP_Alloc();
    if (!rtmp) {
        printf("NO Memory, Failed to alloc RTMP object!\n");
        goto __ERROR;
    }
    RTMP_Init(rtmp);

    //2. 设置RTMP服务器地址，以及连接超时时间
    rtmp->Link.timeout = 10;
    RTMP_SetupURL(rtmp, rtmpaddr);

    //3. 建立连接
//#ifdef WIN32
        // Initialize Winsock
        int iResult;
        WSADATA wsaData;
        iResult = WSAStartup(MAKEWORD(2,2), &wsaData);
        if (iResult != 0) {
            std::cout << "WSAStartup failed: " << iResult << std::endl;
            goto __ERROR;
        }
//#endif

    if (!RTMP_Connect(rtmp, nullptr)) {
        printf("Failed to Connect RTMP Server!\n");
        goto __ERROR;
    }

    //4. 设置是推流还是拉流
    //如果设置了该开关，就是推流，否则就是拉流
    RTMP_EnableWrite(rtmp);

    //5. create stream
    RTMP_ConnectStream(rtmp, 0);

    return rtmp;

__ERROR:

    //释放资源
    if (rtmp) {
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
    }

    return nullptr;
}

//分配RTMPPacket空间
static RTMPPacket *alloc_packet(void)
{
    RTMPPacket *pack = nullptr;

    pack = (RTMPPacket *)malloc(sizeof(RTMPPacket));
    if (!pack) {
        printf("No Memory, Failed to alloc RTMPPacket!\n");
        return pack;
    }

    RTMPPacket_Alloc(pack, 150 * 1024);
    RTMPPacket_Reset(pack);

    pack->m_hasAbsTimestamp = 0;
    pack->m_nChannel = 0x4;

    return pack;
}

static void read_u8(FILE *fp, unsigned int *u8)
{
    fread(u8, 1, 1, fp);
    return;
}

static void read_u24(FILE *fp, unsigned int *u24)
{
    unsigned int tmp;

    fread(&tmp, 1, 3, fp);
    *u24 = ((tmp >> 16) & 0xFF) | ((tmp << 16) & 0xFF0000) | (tmp & 0xFF00);

    return;
}

static void read_u32(FILE *fp, unsigned int *u32)
{
    unsigned int tmp;

    fread(&tmp, 1, 4, fp);
    *u32 = ((tmp >> 24) & 0xFF) | ((tmp << 24) & 0xFF000000) | ((tmp >> 8 ) & 0xFF00) | ((tmp << 8) & 0xFF0000);
    return;
}

/**
 * @brief 从flv文件中读取数据
 * 
 * @param fp [in]：flv file
 * @param packet [out]: the data from flv
 * 
 * @return 0: success, -1: failure 
 */
static int read_data(FILE *fp, RTMPPacket **packet)
{
    /*
     * tag header
     * byte 1, TT(Tag Type), 0x08 音频, 0x09 视频, 0x12 script
     * byte 2-4, Tag body的长度，preTagSize - Tag Header size
     * byte 5-7, 时间戳，单位是毫秒；script 它的时间戳是0
     * byte 8, 扩展时间戳。真正的时间戳结构 [扩展，时间戳] 一共是4字节。
     * byte 9-11，streamID，0
     */
    int ret = -1;
    size_t datasize = 0;

    unsigned int tt;
    unsigned int tag_data_size;
    unsigned int ts;
    unsigned int streamid;
    unsigned int frame_type;
    unsigned int previous_tag_len;

    read_u8(fp, &tt);
    read_u24(fp, &tag_data_size);
    read_u32(fp, &ts);
    read_u24(fp, &streamid);

    datasize = fread((*packet)->m_body, 1, tag_data_size, fp);
    if (tag_data_size != datasize) {
        printf("Failed to read tag body from flv, (datasize=%zu:tds=%d)\n",
                datasize, tag_data_size);
    }
 
    read_u32(fp, &previous_tag_len);

    //设置packet数据
    (*packet)->m_headerType = RTMP_PACKET_SIZE_LARGE;
    (*packet)->m_nTimeStamp = ts;
    (*packet)->m_packetType = tt;
    (*packet)->m_nBodySize = tag_data_size;

    ret = 0;

    return ret;
}

//向流媒体服务器推流
static void send_data(FILE *fp, RTMP *rtmp)
{
    RTMPPacket *packet = nullptr;
    unsigned int pre_ts = 0;
    unsigned int diff = 0;

    //1. 创建 RTMPPacket 对象
    packet = alloc_packet();

    while (1) {
        //2. 从flv文件中读取数据
        read_data(fp, &packet);

        //3. 判断RTMP连接是否正常
        if (!RTMP_IsConnected(rtmp)) {
            printf("Disconnect...\n");
            break;
        }

        diff = packet->m_nBodySize - pre_ts;
        //usleep(10000);

        //4. 发送数据
        int ret = RTMP_SendPacket(rtmp, packet, 0);
        printf("RTMP_SendPacket %d", ret);
        pre_ts = packet->m_nTimeStamp;
    }

    return;
}

void publish_stream()
{
    char *flv = nullptr;
    char * rtmpaddr = nullptr;

    flv = "./out.flv";
    rtmpaddr = "rtmp://106.52.30.12/live/livestream";
    //1. 读 flv 文件
    FILE *fp = open_flv(flv);

    //2. 连接 RTMP 服务器
    RTMP *rtmp = conect_rtmp_server(rtmpaddr);

    //3. publish audio/video data
    send_data(fp, rtmp);

    return;
}

int main()
{
    publish_stream();
    while (1)
    {
        /* code */
    }
    
    return 0;
}

cmake_minimum_required(VERSION 3.0.0)
project(ffmpeg VERSION 0.1.0)

include(CTest)
enable_testing()

set(CPACK_PROJECT_NAME ${PROJECT_NAME})
set(CPACK_PROJECT_VERSION ${PROJECT_VERSION})
include(CPack)

include_directories(lib/ffmpeg/include)
include_directories(lib/rtmp/include)

link_directories(${PROJECT_SOURCE_DIR}/lib/ffmpeg/lib)
link_directories(${PROJECT_SOURCE_DIR}/lib/rtmp/lib)
link_directories("D:/Software/mingw64/x86_64-w64-mingw32/lib")
#find_library(FFMPEG_LIB libavcodec.dll.a ${PROJECT_SOURCE_DIR}/lib/ffmpeg/lib NO_DEFAULT_PATH)
#MESSAGE( FATAL_ERROR "Thsi is " ${FFMPEG_LIB})

add_executable(ffmpeg main.cpp)

#TARGET_LINK_LIBRARIES(ffmpeg ${FFMPEG_LIB})
TARGET_LINK_LIBRARIES(ffmpeg libavformat.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libavcodec.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libavdevice.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libavfilter.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libavutil.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libpostproc.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libswresample.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libswscale.dll.a)
TARGET_LINK_LIBRARIES(ffmpeg libwinmm.a)