客户端使用librtmp推流实践-阿里云开发者社区

开发者社区> 人工智能> 正文

客户端使用librtmp推流实践

简介: 直播推流一般都采用RTMP协议,它是一种基于TCP的应用层协议。librtmp是RTMP协议的一个实现库,它封装了协议细节,方便我们调用。 本文主要介绍如何利用librtmp库实现音视频推流,不涉及RTMP协议的细节解析。 # 建立连接 首先我们需要引入rtmp.h头文件,开始建立连接: ``` // 为RTMP对象分配内存 _rtmp = RTMP_Alloc

直播推流一般都采用RTMP协议,它是一种基于TCP的应用层协议。librtmp是RTMP协议的一个实现库,它封装了协议细节,方便我们调用。
本文主要介绍如何利用librtmp库实现音视频推流,不涉及RTMP协议的细节解析。

建立连接

首先我们需要引入rtmp.h头文件,开始建立连接:

    // 为RTMP对象分配内存
    _rtmp = RTMP_Alloc();
    // 初始化_rtmp
    RTMP_Init(_rtmp);
    // 设置超时时间
    RTMP_SetSocketTimeout(_rtmp, 30);
    // 设置推流地址
    RTMP_SetupURL(_rtmp, "rtmp://xxx");
    // 设置可写状态,推流关键,一定要设置
    RTMP_EnableWrite(_rtmp);
    // 建立连接
    if (RTMP_Connect(_rtmp, NULL) == false) {
        return -1;
    }
    // 建立Stream连接
    if (RTMP_ConnectStream(_rtmp, 0) == false) {
        return -1;
    }

发送音视频编码信息

在发送音视频数据包之前,我们需要首先发送音视频流的编码信息。这些编码信息至关重要,没有它们,解码器将无法解码。

视频描述信息

常见H264格式编码的视频流,描述信息被称为AVCDecoderConfigurationRecord,该结构在“ISO-14496-15 AVC file format”中有详细说明。
需要注意,如果我们推的是视频文件,可以使用ffmpeg直接从文件中读取,它被存放在extradata中。如果是编码器输出,那么需要根据编码器输出的SPS、PPS重新配置AVCDecoderConfigurationRecord结构,构造示例代码如下:

- (void)sendAudioData:(char*)data size:(int)size pts:(long)pts {
    char* body = malloc(1024);
    int i = 0;
    /*AVCDecoderConfigurationRecord*/
    body[i++] = 0x01;
    body[i++] = sps[1];
    body[i++] = sps[2];
    body[i++] = sps[3];
    body[i++] = 0xff;
    /*sps*/
    body[i++] = 0xe1;
    body[i++] = (sLen >> 8) & 0xff;
    body[i++] = sLen & 0xff;
    memcpy(&body[i], sps, sLen);
    i += sLen;
    /*pps*/
    body[i++] = 0x01;
    body[i++] = (plen >> 8) & 0xff;
    body[i++] = (plen) & 0xff;
    memcpy(&body[i], pps, plen);
}

视频描述信息构造完成后就可以封包推流了。RTMP封包使用FLV tag格式,所以还需要加上tag header。示例代码如下:

- (void)sendVideoInfo:(void*)data size:(int)size {
    RTMPPacket *packet = (RTMPPacket *)malloc(RTMP_HEAD_SIZE + size + 5);
    memset(packet, 0, RTMP_HEAD_SIZE);
    packet->m_body = (char *)packet + RTMP_HEAD_SIZE;
    packet->m_nBodySize = size + 5;
    char *body = (char *)packet->m_body;
    memset(body,0,size + 5);
    int i = 0;
    // 前四位1表示关键帧,后四位7表示AVC格式
    body[i++] = 0x17;
    // 0x00 表示AVC Sequence Header
    body[i++] = 0x00;
    body[i++] = 0x00;
    body[i++] = 0x00;
    body[i++] = 0x00;
    // 拷贝AVCDecoderConfigurationRecord
    memcpy(&body[i], data, size);
    
    packet->m_body = body;
    packet->m_packetType = RTMP_PACKET_TYPE_VIDEO;
    packet->m_nBodySize = size + 5;
    packet->m_nChannel = 0x04;
    packet->m_nTimeStamp = 0;
    packet->m_hasAbsTimestamp = 0;
    packet->m_headerType = RTMP_PACKET_SIZE_LARGE;
    packet->m_nInfoField2 = _rtmp->m_stream_id;
    RTMP_SendPacket(_rtmp, packet, true);
    free(packet);
}

音频描述信息

AAC格式编码的音频流,描述信息被称为AudioSpecificConfig,该结构在“ISO-14496-3 Audio”中有详细说明。
同样的,如果推的是文件文件,可以使用ffmpeg直接读取。如果是编码器输出,那么需要根据采样率、通道数等重新配置AudioSpecificConfig结构。

接下来就是封包推流,示例代码如下:

- (void)sendAudioInfo:(void*)data size:(int)size {
    RTMPPacket *packet = (RTMPPacket *)malloc(RTMP_HEAD_SIZE + size + 2);
    memset(packet, 0, RTMP_HEAD_SIZE);
    packet->m_body = (char *)packet + RTMP_HEAD_SIZE;
    packet->m_nBodySize = size + 2;
    char *body = (char *)packet->m_body;
    memset(body,0,size + 2);
    int i = 0;
    body[i++] = 0xAF;
    body[i++] = 0x00;
    // 拷贝AudioSpecificConfig
    memcpy(&body[i], data, size);
    
    packet->m_body = body;
    packet->m_packetType = RTMP_PACKET_TYPE_AUDIO;
    packet->m_nBodySize = size + 2;
    packet->m_nChannel = 0x04;
    packet->m_nTimeStamp = 0;
    packet->m_hasAbsTimestamp = 0;
    packet->m_headerType = RTMP_PACKET_SIZE_LARGE;
    packet->m_nInfoField2 = _rtmp->m_stream_id;
    RTMP_SendPacket(_rtmp, packet, true);
    free(packet);
}

发送音视频数据包

视频数据包

发送视频数据时,需要注意一点:一般视频编码器输出的视频包前四个字节已经就是NALU SIZE了,如果是这种情况,需要跳过前四个字节。

-(void)sendVideoData:(char*)data size:(int)size pts:(long)pts key:(BOOL)isKey {    
    // skip 4 bytes
//    data += 4;
//    size -= 4;
    RTMPPacket *packet = (RTMPPacket *)malloc(RTMP_HEAD_SIZE + size + 9);
    memset(packet, 0, RTMP_HEAD_SIZE);
    packet->m_body = (char *)packet + RTMP_HEAD_SIZE;
    packet->m_nBodySize = size + 9;
    char *body = (char *)packet->m_body;
    memset(body,0,size + 9);
    int i = 0;
    // header
    // 前四位1表示关键帧,前四位2表示非关键帧
    body[i++] = isKey ? 0x17 : 0x27;
    // 0x01 表示AVC NALU
    body[i++] = 0x01;
    body[i++] = 0x00;
    body[i++] = 0x00;
    body[i++] = 0x00;
    // NALU SIZE
    body[i++] = size >> 24 & 0xff;
    body[i++] = size >> 16 & 0xff;
    body[i++] = size >> 8 & 0xff;
    body[i++] = size & 0xff;
    memcpy(&body[i], data, size);
    
    packet->m_body = body;
    packet->m_nBodySize = size + 9;
    packet->m_hasAbsTimestamp = 0;
    packet->m_packetType = RTMP_PACKET_TYPE_VIDEO;
    packet->m_nInfoField2 = _rtmp->m_stream_id;
    packet->m_nChannel = 0x04;
    packet->m_headerType = RTMP_PACKET_SIZE_LARGE;
    packet->m_nTimeStamp = pts;
    RTMP_SendPacket(_rtmp, packet, true);
    free(packet);
}

音频数据包

发送音频数据包的方式和发送描述信息的方式类似,示例代码如下:

- (void)sendAudioData:(char*)data size:(int)size pts:(long)pts {
    RTMPPacket *packet = (RTMPPacket *)malloc(RTMP_HEAD_SIZE+size+2);
    memset(packet, 0, RTMP_HEAD_SIZE);
    packet->m_body = (char *)packet + RTMP_HEAD_SIZE;
    packet->m_nBodySize = size + 2;
    char *body = (char *)packet->m_body;
    memset(body,0,size + 2);
    int i = 0;
    body[i++] = 0xAF;
    body[i++] = 0x01;
    memcpy(&body[i], data, size);
    
    packet->m_body = body;
    packet->m_packetType = RTMP_PACKET_TYPE_AUDIO;
    packet->m_nBodySize = size + 2;
    packet->m_nChannel = 0x04;
    packet->m_nTimeStamp = pts;
    packet->m_hasAbsTimestamp = 0;
    packet->m_headerType = RTMP_PACKET_SIZE_LARGE;
    packet->m_nInfoField2 = _rtmp->m_stream_id;
    RTMP_SendPacket(_rtmp, packet, true);
    free(packet);
}

关闭连接

推流结束时,我们需要关闭连接:

    // 关闭连接
    RTMP_Close(_rtmp);
    // 释放内存
    RTMP_Free(_rtmp);
    _rtmp = NULL;

版权声明:如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件至:developerteam@list.alibaba-inc.com 进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容。

分享:
+ 订阅

了解行业+人工智能最先进的技术和实践,参与行业+人工智能实践项目

其他文章