Added functionality to TS Connector

This commit is contained in:
Erik Zandvliet 2011-08-01 02:33:08 +02:00
parent 50f90f15c4
commit 42052732c1
2 changed files with 100 additions and 10 deletions

View file

@ -16,22 +16,24 @@
class Transport_Packet {
public:
Transport_Packet( bool NALUStart = false, int PID = 0x100 );
Transport_Packet( bool PacketStart = false, int PID = 0x100 );
void SetPayload( char * Payload, int PayloadLen, int Offset = 13 );
void SetPesHeader( );
void Write( );
void Write( Socket::Connection conn );
void SetContinuityCounter( int Counter );
void SetMessageLength( int MsgLen );
void CreatePAT( int ContinuityCounter );
void CreatePMT( int ContinuityCounter );
private:
int PID;
char Buffer[188];
};//Transport Packet
Transport_Packet::Transport_Packet( bool NALUStart, int PID ) {
Transport_Packet::Transport_Packet( bool PacketStart, int PID ) {
(*this).PID = PID;
Buffer[0] = (char)0x47;
Buffer[1] = ( NALUStart ? (char)0x40 : 0 ) + (( PID & 0xFF00 ) >> 8 );
Buffer[1] = ( PacketStart ? (char)0x40 : 0 ) + (( PID & 0xFF00 ) >> 8 );
Buffer[2] = ( PID & 0x00FF );
Buffer[3] = (char)0x00;
}
@ -42,7 +44,7 @@ void Transport_Packet::SetMessageLength( int MsgLen ) {
}
void Transport_Packet::SetContinuityCounter( int Counter ) {
Buffer[3] = (char)0x00 + ( Counter & 0x0F );
Buffer[3] = ( Buffer[3] & 0xF0 ) + ( Counter & 0x0F );
}
void Transport_Packet::SetPesHeader( ) {
@ -62,6 +64,87 @@ void Transport_Packet::SetPayload( char * Payload, int PayloadLen, int Offset )
memcpy( &Buffer[Offset], Payload, std::min( PayloadLen, 188-Offset ) );
}
void Transport_Packet::CreatePAT( int ContinuityCounter ) {
Buffer[3] = (char)0x30 + ContinuityCounter;
Buffer[4] = (char)0x00;
Buffer[5] = (char)0x00;
Buffer[6] = (char)0xB0;
Buffer[7] = (char)0x0D;
Buffer[8] = (char)0x00;
Buffer[9] = (char)0x01;
Buffer[10] = (char)0xC1;
Buffer[11] = (char)0x00;
Buffer[12] = (char)0x00;
Buffer[13] = (char)0x00;
Buffer[14] = (char)0x01;
Buffer[15] = (char)0xF0;
Buffer[16] = (char)0x00;
Buffer[17] = (char)0x2A;
Buffer[18] = (char)0xB1;
Buffer[19] = (char)0x04;
Buffer[20] = (char)0xB2;
for(int i = 21; i < 188; i++ ) {
Buffer[i] = (char)0xFF;
}
}
void Transport_Packet::CreatePMT( int ContinuityCounter ) {
Buffer[3] = (char)0x10 + ContinuityCounter;
Buffer[4] = (char)0x00;
Buffer[5] = (char)0x02;
Buffer[6] = (char)0xB0;
Buffer[7] = (char)0x17;
Buffer[8] = (char)0x00;
Buffer[9] = (char)0x01;
Buffer[10] = (char)0xC1;
Buffer[11] = (char)0x00;
Buffer[12] = (char)0x00;
Buffer[13] = (char)0xE1;
Buffer[14] = (char)0x00;
Buffer[15] = (char)0xF0;
Buffer[16] = (char)0x00;
Buffer[17] = (char)0x1B;
Buffer[18] = (char)0xE1;
Buffer[19] = (char)0x00;
Buffer[20] = (char)0xF0;
Buffer[21] = (char)0x00;
Buffer[22] = (char)0x03;
Buffer[23] = (char)0xE1;
Buffer[24] = (char)0x01;
Buffer[25] = (char)0xF0;
Buffer[26] = (char)0x00;
Buffer[27] = (char)0x4E;
Buffer[28] = (char)0x59;
Buffer[29] = (char)0x3D;
Buffer[30] = (char)0x1E;
for(int i = 31; i < 188; i++ ) {
Buffer[i] = (char)0xFF;
}
}
void SendPAT( Socket::Connection conn ) {
static int ContinuityCounter = 0;
Transport_Packet TS;
TS = Transport_Packet( true, 0 );
TS.CreatePAT( ContinuityCounter );
TS.Write( conn );
CountinuityCounter = ( ContinuityCounter + 1 ) & 0x0F;
}
void SendPMT( Socket::Connection conn ) {
static int ContinuityCounter = 0;
Transport_Packet TS;
TS = Transport_Packet( true, 1000 );
TS.CreatePMT( ContinuityCounter );
TS.Write( conn );
CountinuityCounter = ( ContinuityCounter + 1 ) & 0x0F;
}
std::vector<Transport_Packet> WrapNalus( FLV::Tag tag ) {
static int ContinuityCounter = 0;
Transport_Packet TS;
@ -69,7 +152,7 @@ std::vector<Transport_Packet> WrapNalus( FLV::Tag tag ) {
std::cerr << "Wrapping a tag of length " << tag.len << " into " << PacketAmount << " TS Packet(s)\n";
std::vector<Transport_Packet> Result;
char LeadIn[4] = { (char)0x00, (char)0x00, (char)0x00, (char)0x01 };
TS = Transport_Packet( true );
TS = Transport_Packet( true, 101 );
TS.SetContinuityCounter( ContinuityCounter );
ContinuityCounter = ( ( ContinuityCounter + 1 ) & 0x0F );
TS.SetPesHeader( );
@ -78,7 +161,7 @@ std::vector<Transport_Packet> WrapNalus( FLV::Tag tag ) {
TS.SetPayload( &tag.data[16], 171, 17 );
Result.push_back( TS );
for( int i = 0; i < (PacketAmount - 1); i++ ) {
TS = Transport_Packet( false );
TS = Transport_Packet( false, 101 );
TS.SetContinuityCounter( ContinuityCounter );
ContinuityCounter = ( ( ContinuityCounter + 1 ) & 0x0F );
TS.SetPayload( &tag.data[187+(184*i)], 184, 4 );

View file

@ -401,12 +401,19 @@ int main( ) {
adaptation_field AF;
pes_packet PES;
int ProgramNum;
while( std::cin.good( ) && BlockNo <= 1000 ) {
std::ofstream outfile;
outfile.open( "out.ts" );
while( std::cin.good( ) ) { // && BlockNo <= 1000 ) {
for( int i = 0; i < 188; i++ ) {
if( std::cin.good( ) ){ TempChar[i] = std::cin.get(); }
}
if( ( ( TempChar[1] & 0x1F ) << 8 ) + ( TempChar[2] ) != 0x1FFF ) {
int PID = ( ( TempChar[1] & 0x1F ) << 8 ) + ( TempChar[2] );
if( PID != 100 || ( PID == 100 && ((TempChar[3] & 0x30) == 3 ) && (TempChar[4] & 0x10 )) ) {
for( int i = 0; i < 188; i++ ) {
outfile << TempChar[i];
}
/*
printf( "Block %d:\n", BlockNo );
printf( "\tSync Byte:\t\t\t%X\n", TempChar[0] );
printf( "\tTransport Error Indicator:\t%d\n", ( ( TempChar[1] & 0x80 ) != 0 ) );
@ -443,7 +450,7 @@ int main( ) {
}
BlockNo ++;
*/
} else {
EmptyBlocks ++;
}