自己动手写RTP服务器——用RTP协议传输TS流

自己动手写RTP服务器——用RTP协议传输TS流


上一篇文章我们介绍了关于RTP协议的知识,那么我们现在就自己写一个简单的传输TS流媒体的RTP服务器吧。

预备知识

关于TS流的格式:TS流封装的具体格式请参考文档ISO/IEC 13818-1。这里我们只需要了解一些简单的信息就好。首先TS流是有许多的TS Packet组成的,每个TS Packet的长度固定为188 bytes,每个packet都是以sync_byte:0x47开头。

MTU(Maximum Transmission Unit): 最大传输单元。是指一种通信协议的某一层上面所能通过的最大数据包大小(以字节为单位)。最大传输单元这个参数通常与通信接口有关(网络接口卡、串口等)。例如:以太网无法接收大于1500 字节的数据包。

参考代码

下面我会把自己写的简单的代码贴出来,并且一步步地说明。

新建main.c文件,内容如下:

[cpp]  view plain copy
  1. #include <stdio.h>  
  2. #include <string.h>  
  3. #include <sys/types.h>  
  4. #include <sys/socket.h>  
  5. #include <netinet/in.h>  
  6.   
  7. #define TS_PACKET_SIZE 188  
  8. #define MTU 1500  
说明:包含一些必要的头文件,并且定义了TS Packet的长度(188 bytes),MTU的限制(1500 bytes)。

[cpp]  view plain copy
  1. struct rtp_header{  
  2.     unsigned char cc:4;  
  3.     unsigned char x:1;  
  4.     unsigned char p:1;  
  5.     unsigned char v:2;  
  6.       
  7.     unsigned char pt:7;  
  8.     unsigned char m:1;  
  9.   
  10.     unsigned short sequence_number;  
  11.     unsigned int timestamp;  
  12.     unsigned int ssrc;  
  13. };  
  14.   
  15. void init_rtp_header(struct rtp_header *h){  
  16.     h->v = 2;  
  17.     h->p = 0;  
  18.     h->x = 0;  
  19.     h->cc = 0;  
  20.     h->m = 0;  
  21.     h->pt = 33;  
  22.     h->sequence_number = 123;  
  23.     h->timestamp = 123;  
  24.     h->ssrc = 123;  
  25. }  
说明:这里定义了RTP Header的结构体,以及初始化的方法。这里用到了位域,这是实现协议的时候常常会用到的方法。

需要注意的是:

你会发现这里定义RTP Header的时候,上一篇讲到的具体顺序不同。原因是本机和网络字节流的顺序相反,如果按照v p x cc的顺序来定义一个byte,在这个byte内部v p x cc就会按照从低位到高位的顺序放置;而在RTP流中,应该是顺序从高位到低位放置的。所以每个byte我都把顺序做了倒置。

初始化RTP Header的函数的初始化值的意义请参考rfc3550。为了实现简单,其中的sequence_number、timestamp、ssrc,都是随意填写的。在发送包的时候需要将sequence_number递增。

[cpp]  view plain copy
  1. void sequence_number_increase(struct rtp_header *header){  
  2.     unsigned short sequence = ntohs(header->sequence_number);  
  3.     sequence++;  
  4.     header->sequence_number = htons(sequence);  
  5. }  
说明:这个函数的目的就是让sequence_number加一,还是由于本机与网络字节序不同的原因,所以显得略微复杂些。

[cpp]  view plain copy
  1. int main(){  
  2.     // RTP Packet we will send  
  3.     char buf[MTU];  
  4.     unsigned int count = 0;  
  5.   
  6.     // Init RTP Header  
  7.     init_rtp_header((struct rtp_header*)buf);  
  8.     count = sizeof(struct rtp_header);  
  9.   
  10.     // Init socket  
  11.     int sock = socket(AF_INET, SOCK_DGRAM, 0);  
  12.     struct sockaddr_in dest_addr;  
  13.       
  14.     dest_addr.sin_family=AF_INET;  
  15.     dest_addr.sin_port = htons(6666);  
  16.     dest_addr.sin_addr.s_addr = INADDR_ANY;  
  17.     bzero(&(dest_addr.sin_zero),8);  
  18.       
  19.     // Open TS file  
  20.     FILE *ts_file = fopen("/home/baby/Videos/480p.ts""r+");  
说明:终于到了main函数了,main函数的开始很简单,四个部分的初始化:代表RTP Packet的buffer,RTP Header,Socket,TS流媒体文件。如果你手头没有现成的TS文件,可以用ffmpeg转码得到一个ts文件:“ffmpeg -i video.xxx video.ts”, 其中 video.xxx 表示输入的视频文件,video.ts 为输出的TS文件。

[cpp]  view plain copy
  1. while(!feof(ts_file)){  
  2.     int read_len = fread(buf+count, 1, TS_PACKET_SIZE, ts_file);  
  3.     if(*(buf+count) != 0x47){  
  4.         fprintf(stderr, "Bad sync header!\n");  
  5.         continue;  
  6.     }  
  7.     count += read_len;  
  8.       
  9.     if (count + TS_PACKET_SIZE > MTU){// We should send  
  10.         sequence_number_increase((struct rtp_header*)buf);  
  11.         sendto(sock, buf, count, 0, (const struct sockaddr*)&dest_addr, sizeof(dest_addr));  
  12.         count = sizeof(struct rtp_header);  
  13.         usleep(10000);  
  14.     }  
  15. }  
  16.   
  17. fclose(ts_file);  
说明:一切就绪后就可以不断的用UDP发送RTP Packet了。每次从ts_file中读取188 bytes,附加到buf之后,如果buf的长度还没用到达MTU的限制,那么就继续添加,否则就将buf发送出去。每次发送会将sequence_number加一,并且间隔10000微秒。当然这只是个简单的例子,实际发送视频是要根据时间戳的。

测试

短短几十行代码是否就能完成一个RTP服务器?我们需要用实验来验证。

我的测试环境是Linux,用gcc编译通过,使用VLC(MPlayer 测试也可以通过了)作为接收端。

首先启动我们的发送端程序,然后再执行“vlc rtp://127.0.0.1:6666”,等待几秒后,发现真的可以进行播放啦!


http://blog.csdn.net/baby313/article/details/7363528


二、分享一段H264视频和AAC音频的RTP封包代码

1. H264视频的RTP封包

[c-sharp]  view plain copy
  1. static int h264_parse(Track *tr, uint8_t *data, size_t len) 
  2.     h264_priv *priv = tr->private_data; 
  3. //    double nal_time; // see page 9 and 7.4.1.2 
  4.     size_t nalsize = 0, index = 0; 
  5.     uint8_t *p, *q; 
  6.     if (priv->is_avc) { 
  7.         while (1) { 
  8.             unsigned int i; 
  9.             if(index >= len) break
  10.             //get the nal size 
  11.             nalsize = 0; 
  12.             for(i = 0; i < priv->nal_length_size; i++) 
  13.                 nalsize = (nalsize << 8) | data[index++]; 
  14.             if(nalsize <= 1 || nalsize > len) { 
  15.                 if(nalsize == 1) { 
  16.                     index++; 
  17.                     continue
  18.                 } else { 
  19.                     fnc_log(FNC_LOG_VERBOSE, "[h264] AVC: nal size %d", nalsize); 
  20.                     break
  21.                 } 
  22.             } 
  23.             if (DEFAULT_MTU >= nalsize) { 
  24.                 mparser_buffer_write(tr, 
  25.                                      tr->properties.pts, 
  26.                                      tr->properties.dts, 
  27.                                      tr->properties.frame_duration, 
  28.                                      1, 
  29.                                      data + index, nalsize); 
  30.                 fnc_log(FNC_LOG_VERBOSE, "[h264] single NAL"); 
  31.             } else { 
  32.             // single NAL, to be fragmented, FU-A; 
  33.                 frag_fu_a(data + index, nalsize, DEFAULT_MTU, tr); 
  34.             } 
  35.             index += nalsize; 
  36.         } 
  37.     } else { 
  38.         //seek to the first startcode 
  39.         for (p = data; p<data + len - 3; p++) { 
  40.             if (p[0] == 0 && p[1] == 0 && p[2] == 1) { 
  41.                 break
  42.             } 
  43.         } 
  44.         if (p >= data + len) return ERR_PARSE; 
  45.         while (1) { 
  46.         //seek to the next startcode [0 0 1] 
  47.             for (q = p; q<data+len-3;q++) { 
  48.                 if (q[0] == 0 && q[1] == 0 && q[2] == 1) { 
  49.                     break
  50.                 } 
  51.             } 
  52.             if (q >= data + len) break
  53.             if (DEFAULT_MTU >= q - p) { 
  54.                 fnc_log(FNC_LOG_VERBOSE, "[h264] Sending NAL %d",p[0]&0x1f); 
  55.                 mparser_buffer_write(tr, 
  56.                                      tr->properties.pts, 
  57.                                      tr->properties.dts, 
  58.                                      tr->properties.frame_duration, 
  59.                                      1, 
  60.                                      p, q - p); 
  61.                 fnc_log(FNC_LOG_VERBOSE, "[h264] single NAL"); 
  62.             } else { 
  63.                 //FU-A 
  64.                 fnc_log(FNC_LOG_VERBOSE, "[h264] frags"); 
  65.                 frag_fu_a(p, q - p, DEFAULT_MTU, tr); 
  66.             } 
  67.             p = q; 
  68.         } 
  69.         // last NAL 
  70.         fnc_log(FNC_LOG_VERBOSE, "[h264] last NAL %d",p[0]&0x1f); 
  71.         if (DEFAULT_MTU >= len - (p - data)) { 
  72.             fnc_log(FNC_LOG_VERBOSE, "[h264] no frags"); 
  73.             mparser_buffer_write(tr, 
  74.                                  tr->properties.pts, 
  75.                                  tr->properties.dts, 
  76.                                  tr->properties.frame_duration, 
  77.                                  1, 
  78.                                  p, len - (p - data)); 
  79.         } else { 
  80.             //FU-A 
  81.             fnc_log(FNC_LOG_VERBOSE, "[h264] frags"); 
  82.             frag_fu_a(p, len - (p - data), DEFAULT_MTU, tr); 
  83.         } 
  84.     } 
  85.     fnc_log(FNC_LOG_VERBOSE, "[h264] Frame completed"); 
  86.     return ERR_NOERROR; 
[c-sharp]  view plain copy
  1. static int h264_parse(Track *tr, uint8_t *data, size_t len)  
  2. {  
  3.     h264_priv *priv = tr->private_data;  
  4. //    double nal_time; // see page 9 and 7.4.1.2  
  5.     size_t nalsize = 0, index = 0;  
  6.     uint8_t *p, *q;  
  7.     if (priv->is_avc) {  
  8.         while (1) {  
  9.             unsigned int i;  
  10.             if(index >= len) break;  
  11.             //get the nal size  
  12.             nalsize = 0;  
  13.             for(i = 0; i < priv->nal_length_size; i++)  
  14.                 nalsize = (nalsize << 8) | data[index++];  
  15.             if(nalsize <= 1 || nalsize > len) {  
  16.                 if(nalsize == 1) {  
  17.                     index++;  
  18.                     continue;  
  19.                 } else {  
  20.                     fnc_log(FNC_LOG_VERBOSE, "[h264] AVC: nal size %d", nalsize);  
  21.                     break;  
  22.                 }  
  23.             }  
  24.             if (DEFAULT_MTU >= nalsize) {  
  25.                 mparser_buffer_write(tr,  
  26.                                      tr->properties.pts,  
  27.                                      tr->properties.dts,  
  28.                                      tr->properties.frame_duration,  
  29.                                      1,  
  30.                                      data + index, nalsize);  
  31.                 fnc_log(FNC_LOG_VERBOSE, "[h264] single NAL");  
  32.             } else {  
  33.             // single NAL, to be fragmented, FU-A;  
  34.                 frag_fu_a(data + index, nalsize, DEFAULT_MTU, tr);  
  35.             }  
  36.             index += nalsize;  
  37.         }  
  38.     } else {  
  39.         //seek to the first startcode  
  40.         for (p = data; p<data + len - 3; p++) {  
  41.             if (p[0] == 0 && p[1] == 0 && p[2] == 1) {  
  42.                 break;  
  43.             }  
  44.         }  
  45.         if (p >= data + len) return ERR_PARSE;  
  46.         while (1) {  
  47.         //seek to the next startcode [0 0 1]  
  48.             for (q = p; q<data+len-3;q++) {  
  49.                 if (q[0] == 0 && q[1] == 0 && q[2] == 1) {  
  50.                     break;  
  51.                 }  
  52.             }  
  53.             if (q >= data + len) break;  
  54.             if (DEFAULT_MTU >= q - p) {  
  55.                 fnc_log(FNC_LOG_VERBOSE, "[h264] Sending NAL %d",p[0]&0x1f);  
  56.                 mparser_buffer_write(tr,  
  57.                                      tr->properties.pts,  
  58.                                      tr->properties.dts,  
  59.                                      tr->properties.frame_duration,  
  60.                                      1,  
  61.                                      p, q - p);  
  62.                 fnc_log(FNC_LOG_VERBOSE, "[h264] single NAL");  
  63.             } else {  
  64.                 //FU-A  
  65.                 fnc_log(FNC_LOG_VERBOSE, "[h264] frags");  
  66.                 frag_fu_a(p, q - p, DEFAULT_MTU, tr);  
  67.             }  
  68.             p = q;  
  69.         }  
  70.         // last NAL  
  71.         fnc_log(FNC_LOG_VERBOSE, "[h264] last NAL %d",p[0]&0x1f);  
  72.         if (DEFAULT_MTU >= len - (p - data)) {  
  73.             fnc_log(FNC_LOG_VERBOSE, "[h264] no frags");  
  74.             mparser_buffer_write(tr,  
  75.                                  tr->properties.pts,  
  76.                                  tr->properties.dts,  
  77.                                  tr->properties.frame_duration,  
  78.                                  1,  
  79.                                  p, len - (p - data));  
  80.         } else {  
  81.             //FU-A  
  82.             fnc_log(FNC_LOG_VERBOSE, "[h264] frags");  
  83.             frag_fu_a(p, len - (p - data), DEFAULT_MTU, tr);  
  84.         }  
  85.     }  
  86.     fnc_log(FNC_LOG_VERBOSE, "[h264] Frame completed");  
  87.     return ERR_NOERROR;  
  88. }  

2. AAC的RTP封包

[c-sharp]  view plain copy
  1. static int aac_parse(Track *tr, uint8_t *data, size_t len) 
  2.     //XXX handle the last packet on EOF 
  3.     int off = 0; 
  4.     uint32_t payload = DEFAULT_MTU - AU_HEADER_SIZE; 
  5.     uint8_t *packet = g_malloc0(DEFAULT_MTU); 
  6.     if(!packet) return ERR_ALLOC; 
  7. // trim away extradata 
  8. //    data += AAC_EXTRA; 
  9. //    len -= AAC_EXTRA; 
  10.     packet[0] = 0x00; 
  11.     packet[1] = 0x10; 
  12.     packet[2] = (len & 0x1fe0) >> 5; 
  13.     packet[3] = (len & 0x1f) << 3; 
  14.     if (len > payload) { 
  15.         while (len > payload) { 
  16.             memcpy(packet + AU_HEADER_SIZE, data + off, payload); 
  17.             mparser_buffer_write(tr, 
  18.                                  tr->properties.pts, 
  19.                                  tr->properties.dts, 
  20.                                  tr->properties.frame_duration, 
  21.                                  0, 
  22.                                  packet, DEFAULT_MTU); 
  23.             len -= payload; 
  24.             off += payload; 
  25.         } 
  26.     } 
  27.     memcpy(packet + AU_HEADER_SIZE, data + off, len); 
  28.     mparser_buffer_write(tr, 
  29.                          tr->properties.pts, 
  30.                          tr->properties.dts, 
  31.                          tr->properties.frame_duration, 
  32.                          1, 
  33.                          packet, len + AU_HEADER_SIZE); 
  34.     g_free(packet); 
  35.     return ERR_NOERROR; 
[c-sharp]  view plain copy
  1. static int aac_parse(Track *tr, uint8_t *data, size_t len)  
  2. {  
  3.     //XXX handle the last packet on EOF  
  4.     int off = 0;  
  5.     uint32_t payload = DEFAULT_MTU - AU_HEADER_SIZE;  
  6.     uint8_t *packet = g_malloc0(DEFAULT_MTU);  
  7.     if(!packet) return ERR_ALLOC;  
  8. // trim away extradata  
  9. //    data += AAC_EXTRA;  
  10. //    len -= AAC_EXTRA;  
  11.     packet[0] = 0x00;  
  12.     packet[1] = 0x10;  
  13.     packet[2] = (len & 0x1fe0) >> 5;  
  14.     packet[3] = (len & 0x1f) << 3;  
  15.     if (len > payload) {  
  16.         while (len > payload) {  
  17.             memcpy(packet + AU_HEADER_SIZE, data + off, payload);  
  18.             mparser_buffer_write(tr,  
  19.                                  tr->properties.pts,  
  20.                                  tr->properties.dts,  
  21.                                  tr->properties.frame_duration,  
  22.                                  0,  
  23.                                  packet, DEFAULT_MTU);  
  24.             len -= payload;  
  25.             off += payload;  
  26.         }  
  27.     }  
  28.     memcpy(packet + AU_HEADER_SIZE, data + off, len);  
  29.     mparser_buffer_write(tr,  
  30.                          tr->properties.pts,  
  31.                          tr->properties.dts,  
  32.                          tr->properties.frame_duration,  
  33.                          1,  
  34.                          packet, len + AU_HEADER_SIZE);  
  35.     g_free(packet);  
  36.     return ERR_NOERROR;  
  37. }  

上面的变量 AU_HEADER_SIZE=4

/* au header

      +---------------------------------------+

      |     AU-size                           |

      +---------------------------------------+

      |     AU-Index / AU-Index-delta         |

      +---------------------------------------+

      |     CTS-flag                          |

      +---------------------------------------+

      |     CTS-delta                         |

      +---------------------------------------+

      |     DTS-flag                          |

      +---------------------------------------+

      |     DTS-delta                         |

      +---------------------------------------+

      |     RAP-flag                          |

      +---------------------------------------+

      |     Stream-state                      |

      +---------------------------------------+

*/

http://blog.csdn.net/alangdangjia/article/details/8755608
  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值