客户端使用librtmp推流实践客户端使用librtmp推流实践

【客户端使用librtmp推流实践】播推流一般都采用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); }

关闭连接推流结束时，我们需要关闭连接：