学习笔记（新手）：从PS封装格式的视频文件中提取H.264及Nalu(方法+资料+代码)

本文主要是介绍学习笔记（新手）：从PS封装格式的视频文件中提取H.264及Nalu(方法+资料+代码)，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

学习笔记（新手）：从PS封装格式的视频文件中提取H.264及Nalu(方法+资料+代码)

目的

从一个.ps文件中提取出H.264裸码，博主此前从未了解过音视频的相关知识，以下内容适合新手理解这些概念和过程，并可以进行相关实验。
初学，以下内容根据我的理解来表述，希望大家能够指出我的错误！

Nalu

每个视频帧分为若干NAL单元(NALU)。视频PS格式码流以NALU为单位进行打包。若当前为I帧或P帧的第一个NALU则需加PSH头部。若当前为I帧的第一个NALU还需要加PSM头部。每个NALU分为若干段，每段前需加PES头部,每段数据与PES头部组成PES包。

我们在提取出H.264裸码之后，再做一步，就是找出其中每一个Nalu的起始位置、长度和类别。
Nalu头的标志有两种 0x000001或者0x00000001，这只是标志，不是种类。
标志后的一位的值 0x67代表SPS，0x68代表PPS
我认为这篇博客已经写得非常详细和易懂了，除了0x67，0x68还有很多，了解Nalu的规则有利于我们之后学习RTP。
编写代码的时候，就只要标志头就可以了，但要小心两种标志都是可以的，不要出bug。
这里就不放这部分代码了，只放提取H.264的代码。
链接: https://blog.csdn.net/qq_29350001/article/details/78226286.

总结

对于码流的提取，最关键的就是搞清楚实际的码流数据从哪一位开始，到哪一位结束，因此就要了解清楚头信息里的关于其他信息的长度，才能给准确定位。在编写代码时，我的思路就是：
1、遍历每一字节，先找到PSH头，然后确定PSH头的长度
2、取紧接的4个字节，判断是系统头还是PSM头还是PES头
3、根据上面说的规则确定长度
4、提取数据，检测下面4个字节是哪一种头，直到遍历结束。

代码

代码没有经过优化和简化，作为参考，如果看完上面的内容知道了，那就不需要看代码也行。
当然，为了达到我们要的目的，可以直接遍历，然后找PES头的标志E0，C0，这样子代码三十行就可以搞定了。
（不建议大家直接看代码，先要理解，再看代码）

// An highlighted block
#include <stdio.h>
#pragma pack(1)
#include <WINSOCK2.H> 
#pragma comment(lib, "ws2_32.lib")
#define SEEK_CUR 1
#define PRINTF_DEBUG#define SEEK_CUR2 2
#define MAX_PS_STARTCODE_LEN 4
#define MAX_PDTS_LEN 5
#define MAX_ES_NUMS 6
#define MAX_PDTS_STRING_LEN 12
#define MMIN_PS_HEADER_LEN 14
#define MAX_PES_PACKET_LEN 65496#define SCODE_PS_END 0x000001B9
#define SCODE_PS_HEADER 0x000001BA
#define SCODE_PES_E0 0x000001E0
#define SCODE_PES_C0 0x000001C0
#define SCODE_PES_BD 0x000001BD
#define SCODE_PS_SYSTEM_HEADER 0x000001BB
#define SCODE_PS_SYSTEM_MAP_HEADER 0x000001BC
unsigned char *p = NULL;
int psheader(FILE *fp, char *p)
{//printf("!!!!!!!!1");unsigned char PS_header_lencode;//unsigned char F_lencode;//unsigned int PS_header_systemcode = 0;unsigned int pes_header_code = 0;unsigned long long data = 0;unsigned short stream_map_lencode = 0;unsigned short int PS_system_header_lencode = 0;int PS_header_len = 0;int PS_system_header_len = 0;int stream_map_len = 0;int F_len = 0;int count = 1;fseek(fp, 9, SEEK_CUR);//指向PS头第14个字节if (fread(&PS_header_lencode, 1, 1, fp))//读取第14个字节（1个字节）{//PS_header_len = PS_header_lencode & 0x07;//第14个字节最后3位说明了包头14字节后填充数据的长度printf("5找到了PS_header_lencode+PS_header_lencode: 0x%8X\n", PS_header_lencode);printf("6  %d\n", PS_header_len);fseek(fp, PS_header_len, SEEK_CUR);//跳过填充数据的长度fread(&PS_header_systemcode, 4, 1, fp);//读取系统头的标识（4个字节）BB或BCPS_header_systemcode = htonl(PS_header_systemcode);printf("7找到了PS_header_systemcode+PS_header_systemcode: 0x%8X\n", PS_header_systemcode);//判断是否有系统头if (SCODE_PS_SYSTEM_HEADER == PS_header_systemcode){printf("8BB");fseek(fp, 4, SEEK_CUR);//跳过系统头fread(&PS_system_header_lencode, 2, 1, fp);PS_system_header_len = PS_system_header_lencode;printf("9  %d", PS_system_header_lencode);fseek(fp, PS_system_header_len, SEEK_CUR);}if (SCODE_PS_SYSTEM_MAP_HEADER == PS_header_systemcode){fread(&stream_map_lencode, 2, 1, fp);stream_map_lencode = htons(stream_map_lencode);stream_map_len = stream_map_lencode;printf("11找到了stream_map_lencode+stream_map_lencode: 0x%8X\n", stream_map_lencode);printf("12  %d\n", stream_map_len);fseek(fp, stream_map_len, SEEK_CUR);fread(&pes_header_code, 4, 1, fp);//接下来就是pes包  000001E0/C0pes_header_code = htonl(pes_header_code);printf("13找到了pes_header_code+pes_header_code: 0x%8X +%d\n", pes_header_code,count);count++;}else{pes_header_code = PS_header_systemcode;}while (pes_header_code == SCODE_PES_E0 || pes_header_code == SCODE_PES_C0 || pes_header_code == SCODE_PES_BD){unsigned short data_lencode;int data_len;fread(&data_lencode, 2, 1, fp);data_lencode = htons(data_lencode);data_len = data_lencode;//Yprintf("14  %d\n", data_lencode);//00 00 01 E0 Y Y 8c 80 X (x个)+h264数据//E0后3+X是要跳过的数据fseek(fp, 2, SEEK_CUR);fread(&F_lencode, 1, 1, fp);//取F_lencodeF_len = F_lencode;fseek(fp, F_len, SEEK_CUR);printf("15  %d\n", F_len);fread(p, data_len - 3 - F_len, 1, fp);//读出264数据//写入if (SCODE_PES_E0 == pes_header_code){FILE *fp2;fp2 = fopen("C:\\Users\\zhangjindong\\Desktop\\Vdata.txt", "ab+");fwrite(p, data_len - 3 - F_len, 1, fp2);fclose(fp2);printf("写入成功\n");}if (pes_header_code == SCODE_PES_C0){FILE *fp2;fp2 = fopen("C:\\Users\\zhangjindong\\Desktop\\newA.txt", "ab+");fwrite(p, data_len - 3 - F_len, 1, fp2);fclose(fp2);printf("写入成功\n");}fread(&pes_header_code, 4, 1, fp);pes_header_code = htonl(pes_header_code);//更新pes标识printf("16找到了pes_header_code+pes_header_code: 0x%8X +%d\n", pes_header_code, count);count++;//fseek(fp, 4, SEEK_CUR);//break;调试用中断}}return pes_header_code;
}int main()
{int readLen = 0;int t = 2;int h264Data;unsigned int startCode = 0;p = malloc(MAX_PES_PACKET_LEN);//char *data;FILE *fp;if (fp = fopen("C:\\Users\\zhangjindong\\Desktop\\first.mp4", "rb"))printf("1打开成功\n");else printf("2打开失败\n");while (1){readLen = fread(&startCode, 4, 1, fp);startCode = htonl(startCode);if (SCODE_PS_HEADER == startCode){break;}}while (SCODE_PS_HEADER == startCode ){printf("3找到了PS_header+startCode: 0x%08X\n", startCode);startCode = psheader(fp,p);//break;//向后解析函数调用}if(SCODE_PS_HEADER != startCode){printf("4不是: 0x%08X\n", startCode);fseek(fp, 0, SEEK_CUR);		}free(p);fclose(fp);printf("OVER!");
}