(转)FLV文件格式解析部分代码

前面在FLV文件格式大揭秘中介绍了FLV文件格式，其中也谈到了曾经编写的代码解析flv文件。应网友要求，现在把代码贴出来，仅供大家参考。因为这代码还是在2006年写的，现在看来有些地方还不够准确，解析的也不够完全，所以仅供参考。

 

flvheadpacket.h

#include "packet.h"

class CFlvHeadPacket : public CPacket
{
public:
CFlvHeadPacket(void);
CFlvHeadPacket(UInt8* pContent, UInt32 nLength);
~CFlvHeadPacket(void);

};
flvheadpacket.cpp

#include "stdafx.h"
#include "flvheadpacket.h"

CFlvHeadPacket::CFlvHeadPacket(void)
{
m_nType = FLV_HEAD;
}

CFlvHeadPacket::CFlvHeadPacket(UInt8* pContent, UInt32 nLength)
{
m_nType = FLV_HEAD;
m_pContent = pContent;
m_nLength = nLength;
}

CFlvHeadPacket::~CFlvHeadPacket(void)
{
}

flvmetapacket.h

#include "packet.h"
class CFlvMetaPacket : public CPacket
{
public:
CFlvMetaPacket(void);
CFlvMetaPacket(UInt8* pContent, UInt32 nLength);
~CFlvMetaPacket(void);
};

flvmetapacket.cpp

#include "stdafx.h"
#include "flvmetapacket.h"

CFlvMetaPacket::CFlvMetaPacket(void)
{
m_nType = FLV_META;
}

CFlvMetaPacket::CFlvMetaPacket(UInt8* pContent, UInt32 nLength)
{
m_nType = FLV_META;
m_pContent = pContent;
m_nLength = nLength;
}

CFlvMetaPacket::~CFlvMetaPacket(void)
{
}

flvdatapacket.h

#include "packet.h"

enum{
FLV_UNKNOWN_FRAME,
FLV_VIDEO_FRAME,
FLV_AUDIO_FRAME
};

class CFlvDataPacket : public CPacket
{
public:
CFlvDataPacket(void);
CFlvDataPacket(UInt8* pContent, UInt32 nLength);
~CFlvDataPacket(void);

void    Print();
UInt8   m_nDataType;
UInt32  m_nTime;
UInt8   m_nFlag;
};

flvdatapacket.cpp

#include "stdafx.h"
#include "flvdatapacket.h"

CFlvDataPacket::CFlvDataPacket(void)
{
m_nType = FLV_DATA;
m_nDataType = FLV_UNKNOWN_FRAME;
}

CFlvDataPacket::CFlvDataPacket(UInt8* pContent, UInt32 nLength)
{
m_nType = FLV_DATA;
m_pContent = pContent;
m_nLength = nLength;

if ( m_pContent[0] == 8 )
m_nDataType = FLV_AUDIO_FRAME;
else if ( m_pContent[0] == 9 )
m_nDataType = FLV_VIDEO_FRAME;
else
m_nDataType = FLV_UNKNOWN_FRAME;

m_nTime = FOUR_CHARS_TO_INT(0,m_pContent[4],m_pContent[5],m_pContent[6]);

m_nFlag = m_pContent[11];
}

CFlvDataPacket::~CFlvDataPacket(void)
{
}

void CFlvDataPacket::Print()
{
printf("Type:%d,DataType=%d,Time=%d,Flag=%d,Length=%d/n",m_nType,m_nDataType,m_nTime,m_nFlag,m_nLength);
}

flvreader.h

#ifndef __FLVREADER_H__
#define __FLVREADER_H__

#define FLV_HEAD_LENGTH 13
#define FLV_DATA_HEAD_LENGTH 11

class CFlvDataPacket;
class CFlvHeadPacket;
class CFlvMetaPacket;

class CFlvReader
{
public:
CFlvReader(void);
~CFlvReader(void);

public:
SInt8*  m_sFile;
FILE*   m_pFile;
UInt32  m_nOffset;
BOOL                SetFile(SInt8* sFile);
FILESTAT64          GetFileStat64();
FILESTAT            GetFileStat();
CFlvHeadPacket*     ReadHeader();
CFlvMetaPacket*     ReadMeta();
CFlvDataPacket*     GetNextDataPacket();
};
#endif

flvreader.cpp

#include "stdafx.h"
#include "flvreader.h"
#include "flvdatapacket.h"
#include "flvheadpacket.h"
#include "flvmetapacket.h"

CFlvReader::CFlvReader(void)
{
m_sFile = NULL;
m_pFile = NULL;
m_nOffset = 0;
}

CFlvReader::~CFlvReader(void)
{
if ( m_pFile != NULL )
{
fclose( m_pFile );
m_pFile = NULL;
}
}

BOOL CFlvReader::SetFile(SInt8* sFile)
{
if ( m_pFile != NULL )
{
fclose( m_pFile );
m_pFile = NULL;
}
m_sFile     = sFile;
m_pFile     = fopen( m_sFile, "rb" );
return m_pFile != NULL;
}

CFlvHeadPacket* CFlvReader::ReadHeader()
{
UInt8*  pContent    = new UInt8[FLV_HEAD_LENGTH];
UInt32  nRead       = fread( pContent, sizeof(SInt8), FLV_HEAD_LENGTH, m_pFile );
if ( nRead == FLV_HEAD_LENGTH )
{
return new CFlvHeadPacket(pContent, FLV_HEAD_LENGTH);
}
else
{
delete pContent;
return NULL;
}
}

CFlvMetaPacket* CFlvReader::ReadMeta()
{
UInt8   buf[FLV_DATA_HEAD_LENGTH];
UInt32  nRead   = fread( buf, sizeof(UInt8), FLV_DATA_HEAD_LENGTH, m_pFile );
if ( nRead == FLV_DATA_HEAD_LENGTH )
{
if ( buf[0] == 18 )//meta
{
UInt32 nSize = FOUR_CHARS_TO_INT(0,buf[1],buf[2],buf[3]);
UInt8*  pContent = new  UInt8[nSize+FLV_DATA_HEAD_LENGTH+4];
memcpy(pContent,buf,FLV_DATA_HEAD_LENGTH);
nRead = fread(&pContent[FLV_DATA_HEAD_LENGTH],sizeof(UInt8), nSize+4, m_pFile);
if ( nRead = (nSize+4) )
{
return new CFlvMetaPacket(pContent,nSize+FLV_DATA_HEAD_LENGTH+4);
}
else
{
delete pContent;
return NULL;
}
}
}
return NULL;
}

CFlvDataPacket* CFlvReader::GetNextDataPacket()
{
UInt8   buf[FLV_DATA_HEAD_LENGTH];
UInt32  nRead   = fread( buf, sizeof(UInt8), FLV_DATA_HEAD_LENGTH, m_pFile );
if ( nRead == FLV_DATA_HEAD_LENGTH )
{
if ( buf[0] == 8 || buf[0] == 9 )
{
UInt32 nSize = FOUR_CHARS_TO_INT(0,buf[1],buf[2],buf[3]);
UInt8*  pContent = new  UInt8[nSize+FLV_DATA_HEAD_LENGTH+4];
memcpy(pContent,buf,FLV_DATA_HEAD_LENGTH);
nRead = fread(&pContent[FLV_DATA_HEAD_LENGTH],sizeof(UInt8), nSize+4,m_pFile);
if ( nRead = (nSize+4) )
{
return new CFlvDataPacket(pContent,nSize+FLV_DATA_HEAD_LENGTH+4);
}
else
{
delete pContent;
return NULL;
}
}
}
return NULL;
}

FILESTAT CFlvReader::GetFileStat()
{
FILESTAT    fileStat;
memset(&fileStat,0,sizeof(FILESTAT));

if ( m_pFile != NULL )
{
if ( _stat(m_sFile,&fileStat) != 0 )
{
printf( "/nGetFileStat Error!/n" );
}
}

return fileStat;
}

FILESTAT64      CFlvReader::GetFileStat64()
{
FILESTAT64  fileStat;
memset(&fileStat,0,sizeof(FILESTAT64));

if ( m_pFile != NULL )
{
if ( _stat64(m_sFile,&fileStat) != 0 )
{
printf( "/nGetFileStat64 Error!/n" );
}
}

return fileStat;
}

转自：http://blog.csdn.net/b4362928/article/details/4970249