FFmpeg 获取h264裸码流

原帖链接如下：http://blog.youkuaiyun.com/ren65432/article/details/43449391

有时候我们需要获取h264裸码流进行分析。本文介绍如何通过FFmpeg 获取h264 码流。获取到的h264码流文件 可以直接通过vlc 等播放器直接播放。

一、 .h264文件数据流

如下图 是通过WinHex工具 分析的一个.h264文件数据：

ffmpeg 获取h264 思路如下:

1，写同步码，4字节(00,00,00,01)

2，写sps

3，写同步码，4字节(00,00,00,01)

4，写pps

5，将读到的AVPacket.data 的前4位替换成（00,00,00,01）写文件。

二、sps pps的获取

H.264的SPS和PPS串，包含了初始化H.264解码器所需要的信息参数，包括编码所用的profile，level，图像的宽和高，deblock滤波器等。

（1）avcC的数据结构：

1. aligned(8) class AVCDecoderConfigurationRecord {     
2.  unsigned int(8) configurationVersion = 1;     
3.  unsigned int(8) AVCProfileIndication;     
4.  unsigned int(8) profile_compatibility;     
5.  unsigned int(8) AVCLevelIndication;     
6.  bit(6) reserved = '111111'b;     
7.  unsigned int(2) lengthSizeMinusOne;     
8.  bit(3) reserved = '111'b;     
9.  unsigned int(5) numOfSequenceParameterSets;     
10. for (i=0; i< numOfSequenceParameterSetsispan>    
11.  unsigned int(16) sequenceParameterSetLength ;     
12.  bit(8*sequenceParameterSetLength) sequenceParameterSetNALUnit;     
13.  }     
14.  unsigned int(8) numOfPictureParameterSets;     
15. for (i=0; i< numOfPictureParameterSetsispan>    
16.  unsigned int(16) pictureParameterSetLength;     
17.  bit(8*pictureParameterSetLength) pictureParameterSetNALUnit;     
18.  }     
19. }     

avcC的数据结构对应sps和pps流。

(2) FFmpeg 如何获取sps和pps

ffmpeg获取sps和pps非常简单。avcC数据结构对应于AVFormatContext->streams[H264Index]->codec->extradata。

代码如下:

1.     if ((ret = avformat_open_input(&ic, InputFileName, NULL, NULL)) < 0)  
2.     {  
3.         xprintf->Trace(0,"******** Decode avformat_open_input() Function result=%d",ret);  
4.         return ret;  
5.     }  
6.   
7.     if ((ret = avformat_find_stream_info(ic, NULL)) < 0)   
8.     {    
9.         xprintf->Trace(0,"******** Decode avformat_find_stream_info() Function result=%d ",ret);  
10.         avformat_close_input(&ic);    
11.         return ret;    
12.     }    
13.   
14.     for (int i=0;i<ic->streams[0]->codec->extradata_size;i++)  
15.     {  
16.         printf("%x ",ic->streams[0]->codec->extradata[i]);  
17.     }  

输出打印结果如下：

对应上面的avcC结构体我们知道:

第7,8位, 为sps长度（0,18）即为24。

接下来的24位为sps数据，(67,64,0,20,ac,b2,0,a0,b,76,2,20,0,0,3,0,20,0,0,c,81,e3,6,49)。

接下来的1表示后面跟着PPS:

1. numOfPictureParameterSets, 为1。  

接下来2位为pps长度:(0,6) 即为6。

接下来6位为pps数据，（68，eb,c3,cb,22,c0）。

三、以上各部分的详细代码如下：

1. int GetH264Stream()  
2. {  
3.     int ret;  
4.     AVFormatContext *ic=NULL;  
5.     AVFormatContext *oc=NULL;  
6.   
7.     uint8_t sps[100];  
8.     uint8_t pps[100];  
9.     int spsLength=0;  
10.     int ppsLength=0;  
11.     uint8_t startcode[4]={00,00,00,01};  
12.     FILE *fp;  
13.   
14.     fp=fopen("123.h264","wb+");   
15.   
16.     char *InputFileName="11111.mp4";  
17.   
18.     if ((ret = avformat_open_input(&ic, InputFileName, NULL, NULL)) < 0)  
19.     {  
20.         return ret;  
21.     }  
22.   
23.     if ((ret = avformat_find_stream_info(ic, NULL)) < 0)   
24.     {    
25.         avformat_close_input(&ic);    
26.         return ret;    
27.     }    
28.   
29.     spsLength=ic->streams[0]->codec->extradata[6]*0xFF+ic->streams[0]->codec->extradata[7];  
30.   
31.     ppsLength=ic->streams[0]->codec->extradata[8+spsLength+1]*0xFF+ic->streams[0]->codec->extradata[8+spsLength+2];  
32.   
33.     for (int i=0;i<spsLength;i++)  
34.     {  
35.         sps[i]=ic->streams[0]->codec->extradata[i+8];  
36.     }  
37.       
38.     for (int i=0;i<ppsLength;i++)  
39.     {  
40.         pps[i]=ic->streams[0]->codec->extradata[i+8+2+1+spsLength];  
41.     }  
42.   
43.   
44.     for(int i=0;i<ic->nb_streams;i++)  
45.     {   
46.         if(ic->streams[i]->codec->codec_type==AVMEDIA_TYPE_VIDEO)  
47.         {  
48.             videoindex=i;  
49.         }  
50.         else if(ic->streams[i]->codec->codec_type==AVMEDIA_TYPE_AUDIO)  
51.         {  
52.             audioindex=i;  
53.         }  
54.     }  
55.   
56.     AVOutputFormat *ofmt = NULL;  
57.     AVPacket pkt;  
58.       
59.     avformat_alloc_output_context2(&oc, NULL, NULL, OutPutPath);  
60.   
61.     if (!oc)   
62.     {  
63.         printf( "Could not create output context\n");  
64.         ret = AVERROR_UNKNOWN;  
65.     }  
66.     ofmt = oc->oformat;  
67.     int i;  
68.   
69.     for (i = 0; i < ic->nb_streams; i++)   
70.     {  
71.         AVStream *in_stream = ic->streams[i];  
72.         AVStream *out_stream = avformat_new_stream(oc, in_stream->codec->codec);  
73.   
74.         if (!out_stream)   
75.         {  
76.             printf( "Failed allocating output stream\n");  
77.             ret = AVERROR_UNKNOWN;  
78.         }  
79.         ret = avcodec_copy_context(out_stream->codec, in_stream->codec);  
80.         if (ret < 0)   
81.         {  
82.             printf( "Failed to copy context from input to output stream codec context\n");  
83.         }  
84.         out_stream->codec->codec_tag = 0;  
85.         if (oc->oformat->flags & AVFMT_GLOBALHEADER)  
86.             out_stream->codec->flags |= CODEC_FLAG_GLOBAL_HEADER;  
87.     }  
88.   
89.     if (!(ofmt->flags & AVFMT_NOFILE))   
90.     {  
91.         ret = avio_open(&oc->pb, OutPutPath, AVIO_FLAG_WRITE);  
92.         if (ret < 0)   
93.         {  
94.             printf( "Could not open output file '%s'", OutPutPath);  
95.   
96.         }  
97.     }  
98.     ret = avformat_write_header(oc, NULL);  
99.   
100.     int frame_index=0;  
101.     int flag=1;  
102.   
103.     av_init_packet(&pkt);  
104.     pkt.data = NULL;  
105.     pkt.size = 0;  
106.   
107.     while (1)   
108.     {  
109.         AVStream *in_stream, *out_stream;  
110.   
111.         ret = av_read_frame(ic, &pkt);  
112.   
113.         if (ret < 0)  
114.             break;  
115.         in_stream  = ic->streams[pkt.stream_index];  
116.         out_stream = oc->streams[pkt.stream_index];  
117.   
118.         AVPacket tmppkt;  
119.         if (in_stream->codec->codec_type==AVMEDIA_TYPE_VIDEO )  
120.         {  
121.   
122.             if (flag)  
123.             {  
124.                 fwrite(startcode,4,1,fp);  
125.                 fwrite(sps,spsLength,1,fp);  
126.                 fwrite(startcode,4,1,fp);  
127.                 fwrite(pps,ppsLength,1,fp);  
128.   
129.                 pkt.data[0]=0x00;  
130.                 pkt.data[1]=0x00;  
131.                 pkt.data[2]=0x00;  
132.                 pkt.data[3]=0x01;  
133.                 fwrite(pkt.data,pkt.size,1,fp);  
134.   
135.                 flag=0;  
136.             }  
137.             else  
138.             {  
139.                 pkt.data[0]=0x00;  
140.                 pkt.data[1]=0x00;  
141.                 pkt.data[2]=0x00;  
142.                 pkt.data[3]=0x01;  
143.                 fwrite(pkt.data,pkt.size,1,fp);  
144.             }  
145.   
146.             pkt.pts = av_rescale_q_rnd(pkt.pts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
147.             pkt.dts = av_rescale_q_rnd(pkt.dts, in_stream->time_base, out_stream->time_base, (AVRounding)(AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX));  
148.             pkt.duration = av_rescale_q(pkt.duration, in_stream->time_base, out_stream->time_base);  
149.             pkt.pos = -1;  
150.               
151.             pkt.stream_index=0;  
152.             ret = av_interleaved_write_frame(oc, &pkt);  
153.         }  
154.   
155.         av_free_packet(&pkt);  
156.     }  
157.   
158.     fclose(fp);  
159.     fp=NULL;  
160.   
161.     av_write_trailer(oc);  
162.     return 0;  
163. }