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

转自：  http://www.devdiv.com/iOS_iPhone-_rtp_-thread-118648-1-1.html

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

1. #include "rtpsession.h"
   
2. #include "rtppacket.h"
   
3. #include "rtpudpv4transmitter.h"
   
4. #include "rtpipv4address.h"
   
5. #include "rtpsessionparams.h"
   
6. #include "rtperrors.h"
   
7. 
   
8. @interface VideoRTPPack : NSObject<SocketHandlerDelegate> 
   
9. {
   
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

复制代码

1. #import "VideoRTPPack.h"
   
2. 
   
3. @implementation VideoRTPPack
   
4. 
   
5. @synthesize dicDataBuffer;
   
6. 
   
7. - (id)initWithVideoRTP:(NSString *)hostIP onPort:(unsigned short)port
   
8. {
   
9.         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