iOS音视频封装与传输demo

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. @interface VideoRTPPack : NSObject
  8. {
  9.         RTPSession          m_RTPSession;
  10.        
  11.         NSMutableDictionary *dicDataBuffer;
  12. }

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

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

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

  16. - (void)SendVideoPacket:(unsigned char*)videoData
  17.                                         Nal:(int)_nal
  18.                                    Size:(size_t)videoDataSize
  19.                                    User:(unsigned long)user_id
  20.                                 FrameID:(int)video_frame_id;

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

  22. - (void)closeVideoRTP;

  23. @end
复制代码
  1. #import "VideoRTPPack.h"

  2. @implementation VideoRTPPack

  3. @synthesize dicDataBuffer;

  4. - (id)initWithVideoRTP:(NSString *)hostIP onPort:(unsigned short)port
  5. {
  6.         if (self = [super init])
  7.         {
  8.                 [self createVideoRTP:hostIP onPort:port];
  9.                
  10.                 dicDataBuffer = [[NSMutableDictionary alloc] init];
  11.         }
  12.        
  13.         return self;
  14. }

  15. - (void)dealloc
  16. {
  17.         KLC_RELEASE_SAFELY(self.dicDataBuffer)
  18.        
  19.         [super dealloc];
  20. }

  21. - (void)createVideoRTP:(NSString *)host onPort:(unsigned short)port
  22. {
  23.         RTPUDPv4TransmissionParams m_VideoTransparams;
  24.         RTPSessionParams m_VideoSessionparams;
  25.        
  26.         m_VideoSessionparams.SetOwnTimestampUnit(1.0/10.0);
  27.         m_VideoSessionparams.SetAcceptOwnPackets(TRUE);
  28.         m_VideoSessionparams.SetUsePollThread(0);
  29.         m_VideoTransparams.SetPortbase(port);
  30.        
  31.         int nStatus = m_RTPSession.Create(m_VideoSessionparams, &m_VideoTransparams);
  32.         if(nStatus < 0)
  33.         {
  34.                 NSLog(@"create rtp faild!");
  35.                 return;
  36.         }
  37.        
  38.         unsigned long ipAddress = ntohl(inet_addr([host UTF8String]));
  39.        
  40.         RTPIPv4Address JRTPVideoAddr(ipAddress, port);
  41.         nStatus = m_RTPSession.AddDestination(JRTPVideoAddr);
  42.         if(nStatus < 0)
  43.         {
  44.                 NSLog(@"add destination faild!");
  45.                 return;
  46.         }
  47.        
  48.         m_RTPSession.SetDefaultPayloadType(96);
  49.         m_RTPSession.SetDefaultMark(FALSE);
  50.         m_RTPSession.SetDefaultTimestampIncrement(160);
  51. }

  52. - (void)SendVideoPacket:(unsigned char*)videoData
  53.                                         Nal:(int)_nal
  54.                                    Size:(size_t)videoDataSize
  55.                                    User:(unsigned long)user_id
  56.                                 FrameID:(int)video_frame_id
  57. {
  58.         NSLog(@"video_frame_id = %d", video_frame_id);
  59.        
  60.         unsigned short length = 0;
  61.         int subPackIndex = 1;
  62.         int i_video_loop = 0, offset = 0, video_pack_len = 0;
  63.         NSMutableArray *subPackArray = [[NSMutableArray alloc] init];
  64.        
  65.         MSQ_PACK_MID msq_pack_mid;
  66.         strcpy(msq_pack_mid.msq_head.AppName, GAppName);
  67.         msq_pack_mid.msq_head.sTransType  = htons(TRANSTYPE_CLIENTTOSERVER);
  68.         msq_pack_mid.msq_head.sDataType   = htons(SENDSTREAM_VIDEO_DATATYPE);
  69.         msq_pack_mid.msq_head.lSendID     = htonl(user_id);
  70.        
  71.         i_video_loop = videoDataSize / VIDEO_PACKET_SIZE;
  72.         if ((videoDataSize % VIDEO_PACKET_SIZE) > 0) i_video_loop ++;
  73.        
  74.         while (videoDataSize)
  75.         {
  76.                 VideoPacket *pVideoPacket = (VideoPacket*)msq_pack_mid.msq_data;
  77.                
  78.                 pVideoPacket->PacketHead.TotalPackets   = i_video_loop;
  79.                 pVideoPacket->PacketHead.subpacketindex = subPackIndex;
  80.                 pVideoPacket->PacketHead.sOffset        = htons(offset);
  81.                
  82.                 video_pack_len = (videoDataSize < VIDEO_PACKET_SIZE) ? videoDataSize:VIDEO_PACKET_SIZE;
  83.                 pVideoPacket->PacketHead.sDataLength    = htons(video_pack_len);
  84.                 videoDataSize = videoDataSize - video_pack_len;
  85.                
  86.                 pVideoPacket->PacketHead.lFrameID       = htonl(video_frame_id);
  87.                 pVideoPacket->PacketHead.timeStart      = 0;
  88.                 pVideoPacket->PacketHead.timeEnd        = 0;
  89.                 pVideoPacket->PacketHead.s_nal          = _nal;
  90.                
  91.                 memcpy(pVideoPacket->rawData, videoData+offset, video_pack_len);
  92.                 offset = offset + video_pack_len;
  93.                
  94.                 length = sizeof(VideoPacketHead) + video_pack_len;
  95.                
  96.                 msq_pack_mid.msq_head.sTCPLength = htons(length);
  97.                 msq_pack_mid.msq_head.sDataLength = msq_pack_mid.msq_head.sTCPLength;
  98.                
  99.                 NSData *subData = [NSData dataWithBytes:(void *)&msq_pack_mid length:sizeof(MSQ_HEAD)+length];
  100.                 [subPackArray addObject:subData];
  101.                
  102.                 m_RTPSession.SendPacket(&msq_pack_mid, sizeof(MSQ_HEAD)+length);
  103.                
  104.                 subPackIndex ++;
  105.         }
  106.        
  107.         // 把视频数据存入缓冲区
  108.         if ([[self.dicDataBuffer allKeys] count] < 50)
  109.         {
  110.                 [self.dicDataBuffer setObject:subPackArray
  111.                                                            forKey:[NSNumber numberWithInt:video_frame_id]];
  112.         }
  113.         else
  114.         {
  115.                 [self.dicDataBuffer removeObjectForKey:[NSNumber numberWithInt:video_frame_id-51]];
  116.                 [self.dicDataBuffer setObject:subPackArray
  117.                                                            forKey:[NSNumber numberWithInt:video_frame_id]];
  118.         }
  119.        
  120.         KLC_RELEASE_SAFELY(subPackArray)
  121. }

  122. - (void)resendSubPacketData:(int)frameID withSubPack:(int)packID
  123. {
  124.         NSArray *lostArray = (NSArray *)[self.dicDataBuffer objectForKey:[NSNumber numberWithInt:frameID]];
  125.         NSData *lostData = [lostArray objectAtIndex:packID-1];
  126.        
  127.         int nStatus = m_RTPSession.SendPacket([lostData bytes], [lostData length]);
  128.         if (nStatus == 0) NSLog(@"ReSend Packet Data Succeed!");
  129. }

  130. - (void)closeVideoRTP
  131. {
  132.         if (m_RTPSession.IsActive())
  133.         {
  134.                 m_RTPSession.Destroy();
  135.         }
  136. }


  137. @end
demo程序如附件中

 

你可能感兴趣的:(嵌入式与语音视频系统)