音视频

实现了从一个 FLV 文件读取音视频数据，并通过 RTMP 协议进行推流。核心步骤包括：

FLV 文件头解析 (openfile)

  • 功能: 打开并读取 FLV 文件，跳过文件头（9 字节）。
  • 步骤:
    • 打开指定的 FLV 文件。
    • 检查文件是否成功打开。如果打开失败，则返回 nullptr。
    • 跳过文件头的 9 字节，定位到数据部分。
1
2
3
4
5
6
7
8
9
static QFile *openfile(char *flv_name) {
    QFile *file = new QFile(flv_name);
    if(!file->open(QFile::ReadOnly)){
        qDebug() << "文件打开失败!";
        return nullptr;
    }
    file->seek(9); //跳过 9字节 header
    return file;
}

RTMP 连接 (connect_rtmp_server)

  • 功能: 初始化 RTMP 连接到 RTMP 服务器。
  • 步骤:
    • 检查 RTMP 地址是否为空。
    • 创建并初始化 RTMP 对象。
    • 设置 RTMP 服务器地址和连接超时时间。
    • 进行连接，并设置为推流模式（调用 RTMP_EnableWrite）。
    • 创建并连接流。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
static RTMP *connect_rtmp_server(char *rtmpaddr) {
    if(rtmpaddr == NULL){
        qDebug() << "连接的RTMP地址为空!";
        return NULL;
    }
    RTMP *rtmp = nullptr;
    rtmp = RTMP_Alloc();
    RTMP_Init(rtmp);
    if(!rtmp){
        qDebug() << "Failed to alloc RTMP object!";
        goto __ERROR;
    }

    RTMP_SetupURL(rtmp, rtmpaddr);
    rtmp->Link.timeout = 10;

    if(!RTMP_Connect(rtmp, NULL)){
        qDebug() << "Failed to connect RTMP Server!";
        goto __ERROR;
    }

    RTMP_EnableWrite(rtmp);
    RTMP_ConnectStream(rtmp, 0);

    return rtmp;
__ERROR:
    if(rtmp){
        RTMP_Close(rtmp);
        RTMP_Free(rtmp);
    }
    return NULL;
}

RTMP 数据包分配 (alloc_packet)

  • 功能: 为 RTMP 数据包分配内存并初始化。
  • 步骤:
    • 分配 RTMPPacket 内存空间。
    • 分配 64KB 内存空间用于数据包。
    • 重置数据包并设置初始值。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
static RTMPPacket *alloc_packet() {
    RTMPPacket *packet = NULL;
    packet = (RTMPPacket *)malloc(sizeof(RTMPPacket));
    if(!packet){
        qDebug() << "Failed to alloc RTMPPacket!";
        return NULL;
    }

    RTMPPacket_Alloc(packet, 64 * 1024);
    RTMPPacket_Reset(packet);

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

    return packet;
}

读取 FLV 数据并填充 RTMP 数据包 (read_data)

  • 功能: 逐个读取 FLV 文件中的数据块并填充 RTMP 数据包。
  • 步骤:
    • 跳过前 4 字节的 pre-tag size。
    • 读取 tag type、tag data size、时间戳和流 ID。
    • 读取实际的 tag 数据体，并将其填充到 RTMPPacket 中。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
static int read_data(QFile *file, RTMPPacket **packet) {
    if (!file || !packet) {
        qDebug() << "Error: Invalid arguments.";
        return 0;
    }

    int ret = 0;
    QByteArray tt, tag_data_size, ts, streamid;

    file->read(4);  // 跳过 pre-tag size
    tt = file->read(1);  // 读取 tag type
    tag_data_size = file->read(3);  // 读取 tag data size

    unsigned int tagDataSize = ((unsigned char)tag_data_size[0] << 16)
                               | ((unsigned char)tag_data_size[1] << 8)
                               | (unsigned char)tag_data_size[2];

    if (file->bytesAvailable() < tagDataSize) {
        qDebug() << "Error: Insufficient data for tag body.";
        return 0;
    }

    ts = file->read(4);  // 读取时间戳
    unsigned int timestamp = ((unsigned char)ts[0] << 16)
                             | ((unsigned char)ts[1] << 8)
                             | ((unsigned char)ts[2]);

    streamid = file->read(3);  // 读取流 ID
    QByteArray bodyData = file->read(tagDataSize);  // 读取 tag body

    (*packet)->m_packetType = static_cast<unsigned char>(tt[0]);
    std::memcpy((*packet)->m_body, bodyData.constData(), tagDataSize);
    (*packet)->m_headerType = RTMP_PACKET_SIZE_LARGE;
    (*packet)->m_nTimeStamp = timestamp;
    (*packet)->m_nBodySize = tagDataSize;

    qDebug() << "tt:" << static_cast<unsigned char>(tt[0])
             << "ts:" << timestamp
             << "datasize:" << tagDataSize;

    ret = 1; // 成功
    return ret;
}

推流到 RTMP 服务器 (send_data)

  • 功能: 从 FLV 文件中读取数据并通过 RTMP 协议发送。
  • 步骤:
    • 创建 RTMPPacket 对象。
    • 逐个读取 FLV 数据包并发送到 RTMP 服务器。
    • 如果连接断开，则中断发送。
    • 使用 RTMP_SendPacket 发送数据包到 RTMP 服务器。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
static void send_data(QFile *file, RTMP *rtmp) {
    RTMPPacket *packet = alloc_packet();

    while(read_data(file, &packet)) {
        if(!RTMP_IsConnected(rtmp)) {
            qDebug() << "Disconnect...";
            break;
        }

        if(!RTMP_SendPacket(rtmp, packet, 0)) {
            qDebug() << "Failed to send packet!";
        }
    }

    qDebug() << "发送结束";
}

RTMP 推流整体的封装 (publish_stream)

  • 功能: 完成 FLV 文件读取、RTMP 连接、数据发送等操作。
  • 步骤:
    • 打开 FLV 文件并连接到 RTMP 服务器。
    • 将 FLV 文件中的音视频数据逐个发送到 RTMP 服务器。
    • 完成推流后，关闭文件。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
void publish_stream() {
    char *rtmpaddr = (char *)"rtmp://127.0.0.1:1935/live/stream01";
    QFile *file = openfile((char *)"output.flv");
    if(!file) {
        return;
    }

    RTMP *rtmp = connect_rtmp_server(rtmpaddr);
    if(!rtmp) {
        return;
    }

    send_data(file, rtmp);
    file->close();
}

线程运行入口 (run)

  • 功能: 在线程中执行发布流的操作。
  • 步骤:
    • 初始化 Windows 套接字库 (WSAStartup)。
    • 调用 publish_stream 完成推流任务。
    • 调用 WSACleanup 关闭套接字库。
1
2
3
4
5
6
7
8
9
10
void myRtmp::run() {
    WSADATA wsaData;
    int result = WSAStartup(MAKEWORD(2, 2), &wsaData);
    if (result != 0) {
        qDebug() << "WSAStartup failed: " << result;
        return;
    }
    publish_stream();
    WSACleanup();
}

主要结构和功能：

  • RTMP 流传输: 使用 RTMP 协议推送 FLV 文件数据到服务器。
  • 文件处理: 从 FLV 文件中提取音视频数据并填充到 RTMP 数据包中。
  • 多线程支持: 使用 QThread 在独立线程中执行推流任务。

说些什么吧！

valine