封装音视频流为RTP包来网络传输

ios没有直接得api用来封装传输rtp，但是经常有用到得接收过来得音视频数据，经过X264处理转换便可以封装为Rtp包进行直接发送。大家可以看代码：

 

#include "rtpsession.h"

02. #include "rtppacket.h"

03. #include "rtpudpv4transmitter.h"

04. #include "rtpipv4address.h"

05. #include "rtpsessionparams.h"

06. #include "rtperrors.h"

07. 

08. @interface VideoRTPPack : NSObject<SocketHandlerDelegate> 

09. {

10.         RTPSession          m_RTPSession;

11.         

12.         NSMutableDictionary *dicDataBuffer;

13. }

14. 

15. @property(nonatomic, retain) NSMutableDictionary *dicDataBuffer;

16. 

17. - (id)initWithVideoRTP:(NSString *)hostIP onPort:(unsigned short)port;

18. 

19. - (void)createVideoRTP:(NSString *)host onPort:(unsigned short)port;

20. 

21. - (void)SendVideoPacket:(unsigned char*)videoData 

22.                                         Nal:(int)_nal

23.                                    Size:(size_t)videoDataSize 

24.                                    User:(unsigned long)user_id

25.                                 FrameID:(int)video_frame_id;

26. 

27. - (void)resendSubPacketData:(int)frameID withSubPack:(int)packID;

28. 

29. - (void)closeVideoRTP;

30. 

31. @end

 

#import "VideoRTPPack.h"

02. 

03. @implementation VideoRTPPack

04. 

05. @synthesize dicDataBuffer;

06. 

07. - (id)initWithVideoRTP:(NSString *)hostIP onPort:(unsigned short)port

08. {

09.         if (self = [super init]) 

10.         {

11.                 [self createVideoRTP:hostIP onPort:port];

12.                 

13.                 dicDataBuffer = [[NSMutableDictionary alloc] init];

14.         }

15.         

16.         return self;

17. }

18. 

19. - (void)dealloc

20. {

21.         KLC_RELEASE_SAFELY(self.dicDataBuffer)

22.         

23.         [super dealloc];

24. }

25. 

26. - (void)createVideoRTP:(NSString *)host onPort:(unsigned short)port

27. {

28.         RTPUDPv4TransmissionParams m_VideoTransparams;

29.         RTPSessionParams m_VideoSessionparams;

30.         

31.         m_VideoSessionparams.SetOwnTimestampUnit(1.0/10.0);

32.         m_VideoSessionparams.SetAcceptOwnPackets(TRUE);

33.         m_VideoSessionparams.SetUsePollThread(0);

34.         m_VideoTransparams.SetPortbase(port);

35.         

36.         int nStatus = m_RTPSession.Create(m_VideoSessionparams, &m_VideoTransparams);

37.         if(nStatus < 0) 

38.         {

39.                 NSLog(@"create rtp faild!");

40.                 return;

41.         }

42.         

43.         unsigned long ipAddress = ntohl(inet_addr([host UTF8String]));

44.         

45.         RTPIPv4Address JRTPVideoAddr(ipAddress, port);

46.         nStatus = m_RTPSession.AddDestination(JRTPVideoAddr);

47.         if(nStatus < 0) 

48.         {

49.                 NSLog(@"add destination faild!");

50.                 return;

51.         }

52.         

53.         m_RTPSession.SetDefaultPayloadType(96);

54.         m_RTPSession.SetDefaultMark(FALSE);

55.         m_RTPSession.SetDefaultTimestampIncrement(160);

56. }

57. 

58. - (void)SendVideoPacket:(unsigned char*)videoData 

59.                                         Nal:(int)_nal

60.                                    Size:(size_t)videoDataSize 

61.                                    User:(unsigned long)user_id

62.                                 FrameID:(int)video_frame_id

63. {

64.         NSLog(@"video_frame_id = %d", video_frame_id);

65.         

66.         unsigned short length = 0;

67.         int subPackIndex = 1;

68.         int i_video_loop = 0, offset = 0, video_pack_len = 0;

69.         NSMutableArray *subPackArray = [[NSMutableArray alloc] init];

70.         

71.         MSQ_PACK_MID msq_pack_mid;

72.         strcpy(msq_pack_mid.msq_head.AppName, GAppName);

73.         msq_pack_mid.msq_head.sTransType  = htons(TRANSTYPE_CLIENTTOSERVER);

74.         msq_pack_mid.msq_head.sDataType   = htons(SENDSTREAM_VIDEO_DATATYPE);

75.         msq_pack_mid.msq_head.lSendID     = htonl(user_id);

76.         

77.         i_video_loop = videoDataSize / VIDEO_PACKET_SIZE;

78.         if ((videoDataSize % VIDEO_PACKET_SIZE) > 0) i_video_loop ++;

79.         

80.         while (videoDataSize)

81.         {

82.                 VideoPacket *pVideoPacket = (VideoPacket*)msq_pack_mid.msq_data;

83.                 

84.                 pVideoPacket->PacketHead.TotalPackets   = i_video_loop;

85.                 pVideoPacket->PacketHead.subpacketindex = subPackIndex;

86.                 pVideoPacket->PacketHead.sOffset        = htons(offset);

87.                 

88.                 video_pack_len = (videoDataSize < VIDEO_PACKET_SIZE) ? videoDataSize:VIDEO_PACKET_SIZE;

89.                 pVideoPacket->PacketHead.sDataLength    = htons(video_pack_len);

90.                 videoDataSize = videoDataSize - video_pack_len;

91.                 

92.                 pVideoPacket->PacketHead.lFrameID       = htonl(video_frame_id);

93.                 pVideoPacket->PacketHead.timeStart      = 0;

94.                 pVideoPacket->PacketHead.timeEnd        = 0;

95.                 pVideoPacket->PacketHead.s_nal          = _nal;

96.                 

97.                 memcpy(pVideoPacket->rawData, videoData+offset, video_pack_len);

98.                 offset = offset + video_pack_len;

99.                 

100.                 length = sizeof(VideoPacketHead) + video_pack_len;

101.                 

102.                 msq_pack_mid.msq_head.sTCPLength = htons(length);

103.                 msq_pack_mid.msq_head.sDataLength = msq_pack_mid.msq_head.sTCPLength;

104.                 

105.                 NSData *subData = [NSData dataWithBytes:(void *)&msq_pack_mid length:sizeof(MSQ_HEAD)+length];

106.                 [subPackArray addObject:subData];

107.                 

108.                 m_RTPSession.SendPacket(&msq_pack_mid, sizeof(MSQ_HEAD)+length);

109.                 

110.                 subPackIndex ++;

111.         }

112.         

113.         // 把视频数据存入缓冲区

114.         if ([[self.dicDataBuffer allKeys] count] < 50) 

115.         {

116.                 [self.dicDataBuffer setObject:subPackArray 

117.                                                            forKey:[NSNumber numberWithInt:video_frame_id]];

118.         }

119.         else 

120.         {

121.                 [self.dicDataBuffer removeObjectForKey:[NSNumber numberWithInt:video_frame_id-51]];

122.                 [self.dicDataBuffer setObject:subPackArray 

123.                                                            forKey:[NSNumber numberWithInt:video_frame_id]];

124.         }

125.         

126.         KLC_RELEASE_SAFELY(subPackArray)

127. }

128. 

129. - (void)resendSubPacketData:(int)frameID withSubPack:(int)packID

130. {

131.         NSArray *lostArray = (NSArray *)[self.dicDataBuffer objectForKey:[NSNumber numberWithInt:frameID]];

132.         NSData *lostData = [lostArray objectAtIndex:packID-1];

133.         

134.         int nStatus = m_RTPSession.SendPacket([lostData bytes], [lostData length]);

135.         if (nStatus == 0) NSLog(@"ReSend Packet Data Succeed!");

136. }

137. 

138. - (void)closeVideoRTP

139. {

140.         if (m_RTPSession.IsActive()) 

141.         {

142.                 m_RTPSession.Destroy();

143.         }

144. }

145. 

146. 

147. @end