/** @brief This function is called when a new EIT packet for all channels is there and we asked for rewrite
* this function save the full EIT for each service.
* @return return 1 when the packet is updated
*/
int eit_rewrite_new_global_packet(unsigned char *ts_packet, rewrite_parameters_t *rewrite_vars)
{
eit_t *eit=NULL;
/*Check the version before getting the full packet*/
if(!rewrite_vars->eit_needs_update)
rewrite_vars->eit_needs_update=eit_need_update(rewrite_vars,ts_packet,1);
/*We need to update the full packet, we download it*/
if(rewrite_vars->eit_needs_update )
{
if(get_ts_packet(ts_packet,rewrite_vars->full_eit))
{
//We check if we have to store this new EIT packet (CRC32 can make false alarms)
if(!eit_need_update(rewrite_vars,rewrite_vars->full_eit->data_full,0))
{
return 0;
}
log_message( log_module, MSG_DETAIL,"New full EIT for update");
//For debugging purposes
eit=(eit_t*)(rewrite_vars->full_eit->data_full);
eit_display_header(eit);
eit_packet_t *eit_packet;
eit_packet=eit_new_packet(rewrite_vars,HILO(eit->service_id));
if(NULL==eit_packet)
return 0;
if(eit->version_number!=eit_packet->version)
{
log_message( log_module, MSG_DETAIL,"New version for EIT sid %d need update. stored version : %d, new: %d",
HILO(eit->service_id),
eit_packet->version,
eit->version_number);
//New version so we clear all contents
eit_free_packet_contents(eit_packet);
}
eit_packet->last_section_number = eit->last_section_number;
eit_packet->version=eit->version_number;
eit_packet->service_id = HILO(eit->service_id);
eit_packet->full_eit_ok=1;
/*We've got the FULL EIT packet*/
//we copy the data to the right section
if(eit_packet->full_eit_sections[eit->section_number]==NULL)
eit_packet->full_eit_sections[eit->section_number]=calloc(1,sizeof(mumudvb_ts_packet_t));
if(eit_packet->full_eit_sections[eit->section_number]==NULL)
{
log_message( log_module, MSG_ERROR,"Problem with calloc : %s file : %s line %d\n",strerror(errno),__FILE__,__LINE__);
return 0;
}
memcpy(eit_packet->full_eit_sections[eit->section_number], rewrite_vars->full_eit,sizeof( mumudvb_ts_packet_t));
//We store that we saw this section number
eit_packet->sections_stored[eit->section_number]=1;
log_message( log_module, MSG_DETAIL,"Full EIT updated. sid %d section number %d, last_section_number %d\n",
eit_packet->service_id,
eit->section_number,
eit_packet->last_section_number);
rewrite_vars->eit_needs_update = 0;
log_message( log_module, MSG_WARN,"!!Stored after update");
eit_show_stored(rewrite_vars);
}
}
return 0;
}
int main(void)
{
int press_enter = PRESS_ENTER;
//We initalise the logging parameters
log_params.verbosity = 999;
log_params.log_type=LOGGING_CONSOLE;
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing program for MuMuDVB\n");
log_message( log_module, MSG_INFO,"===================================================================\n");
/**************************** Very basic test ****************************************************/
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Display Ca system id 1\n" );
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"%s\n\n" ,ca_sys_id_to_str(1));
/**************************** Testing string compute *********************************************/
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing string compute %s\n",TEST_STRING_COMPUT );
log_message( log_module, MSG_INFO,"===================================================================\n");
int resultat;
resultat=string_comput(TEST_STRING_COMPUT);
if(resultat==TEST_STRING_COMPUT_RES)
log_message( log_module, MSG_INFO,"%d -- PASS\n\n" ,resultat);
else
log_message( log_module, MSG_INFO,"%d -- FAIL\n\n" ,resultat);
/************************************* Testing the SDT parser *************************************/
char *files_sdt[NUM_FILES_TEST_READ_SDT]={FILES_TEST_READ_SDT_TS};
for(int i_file=0;i_file<NUM_FILES_TEST_READ_SDT;i_file++)
{
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing TS read for SDT - test %d on %d file %s\n", i_file+1, NUM_FILES_TEST_READ_SDT , files_sdt[i_file] );
press_enter_func(press_enter);
FILE *testfile;
testfile=fopen (files_sdt[i_file], "r");
if(testfile!=NULL)
{
//We read all the packets contained in the file
unsigned char ts_packet_raw[TS_PACKET_SIZE];
int num_sdt_read=0;
mumudvb_ts_packet_t ts_packet_mumu;
memset(&ts_packet_mumu, 0, sizeof(mumudvb_ts_packet_t));
mumudvb_service_t services;
memset(&services, 0, sizeof(mumudvb_service_t));
//Just to make pthread happy
pthread_mutex_init(&ts_packet_mumu.packetmutex,NULL);
int iRet,pid;
log_message( log_module, MSG_INFO,"File opened, reading packets\n" );
while(fread(ts_packet_raw,TS_PACKET_SIZE,1, testfile) && num_sdt_read<NUM_READ_SDT)
{
/************ SYNC *************/
if(ts_packet_raw[0] != 0x47)
{
log_message( log_module, MSG_INFO," !!!!!!!!!!!!! Sync error, we search for a new sync byte !!!!!!!!!!!!!!\n");
unsigned char sync;
//We search the next sync byte
while(fread(&sync,1,1, testfile) && sync!=0x47);
ts_packet_raw[0]=sync;
//We found a "sync byte" we get the rest of the packet
if(!fread(ts_packet_raw-1,TS_PACKET_SIZE-1,1, testfile))
continue;
else
log_message( log_module, MSG_INFO," sync byte found :) \n");
}
// Get the PID of the received TS packet
pid = HILO(((ts_header_t *)ts_packet_raw)->pid);
if(pid != 17) continue;
log_message( log_module, MSG_INFO,"New elementary (188bytes TS packet) pid %d continuity_counter %d",
pid,
((ts_header_t *)ts_packet_raw)->continuity_counter );
iRet=get_ts_packet(ts_packet_raw, &ts_packet_mumu);
//If it's the beginning of a new packet we display some information
if(((ts_header_t *)ts_packet_raw)->payload_unit_start_indicator)
log_message(log_module, MSG_INFO, "First two bytes of the packet 0x%02x %02x",
ts_packet_mumu.data_partial[0],
ts_packet_mumu.data_partial[1]);
if(iRet==1)//packet is parsed
{
log_message( log_module, MSG_INFO,"New packet -- parsing\n" );
num_sdt_read++;
autoconf_read_sdt(ts_packet_mumu.data_full,ts_packet_mumu.len_full,&services);
}
}
log_message( log_module, MSG_INFO,"Final services list .... \n");
autoconf_print_services(&services);
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing service sorting on this list\n" );
press_enter_func(press_enter);
autoconf_sort_services(&services);
autoconf_print_services(&services);
//We free starting at the next since the first is not malloc'ed
autoconf_free_services(services.next);
fclose(testfile);
}
else
log_message( log_module, MSG_INFO,"Test file %s cannot be open : %s\n", files_sdt[i_file],strerror(errno) );
}
/************************************* Testing the NIT parser *************************************/
char *files_nit[NUM_FILES_TEST_READ_NIT]={FILES_TEST_READ_NIT_TS};
for(int i_file=0;i_file<NUM_FILES_TEST_READ_NIT;i_file++)
{
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing TS read for NIT - test %d on %d file %s\n", i_file+1, NUM_FILES_TEST_READ_NIT , files_sdt[i_file] );
press_enter_func(press_enter);
FILE *testfile;
testfile=fopen (files_nit[i_file], "r");
if(testfile!=NULL)
{
//We read all the packets contained in the file
unsigned char ts_packet_raw[TS_PACKET_SIZE];
int num_nit_read=0;
mumudvb_ts_packet_t ts_packet_mumu;
memset(&ts_packet_mumu, 0, sizeof(mumudvb_ts_packet_t));
mumudvb_service_t services;
memset(&services, 0, sizeof(mumudvb_service_t));
//Just to make pthread happy
pthread_mutex_init(&ts_packet_mumu.packetmutex,NULL);
int iRet,pid;
log_message( log_module, MSG_INFO,"File opened, reading packets\n" );
while(fread(ts_packet_raw,TS_PACKET_SIZE,1, testfile) && num_nit_read<NUM_READ_NIT)
{
/************ SYNC *************/
if(ts_packet_raw[0] != 0x47)
{
log_message( log_module, MSG_INFO," !!!!!!!!!!!!! Sync error, we search for a new sync byte !!!!!!!!!!!!!!\n");
unsigned char sync;
//We search the next sync byte
while(fread(&sync,1,1, testfile) && sync!=0x47);
ts_packet_raw[0]=sync;
//We found a "sync byte" we get the rest of the packet
if(!fread(ts_packet_raw-1,TS_PACKET_SIZE-1,1, testfile))
continue;
else
log_message( log_module, MSG_INFO," sync byte found :) \n");
}
// Get the PID of the received TS packet
pid = HILO(((ts_header_t *)ts_packet_raw)->pid);
if(pid != 16) continue;
log_message( log_module, MSG_INFO,"New elementary (188bytes TS packet) pid %d continuity_counter %d",
pid,
((ts_header_t *)ts_packet_raw)->continuity_counter );
iRet=get_ts_packet(ts_packet_raw, &ts_packet_mumu);
//If it's the beginning of a new packet we display some information
if(((ts_header_t *)ts_packet_raw)->payload_unit_start_indicator)
log_message(log_module, MSG_INFO, "First two bytes of the packet 0x%02x %02x",
ts_packet_mumu.data_partial[0],
ts_packet_mumu.data_partial[1]);
if(iRet==1)//packet is parsed
{
log_message( log_module, MSG_INFO,"New packet -- parsing\n" );
num_nit_read++;
//autoconf_read_sdt(ts_packet_mumu.data_full,ts_packet_mumu.len_full,&services);
parse_nit_descriptors(ts_packet_mumu.data_full,ts_packet_mumu.len_full);
}
}
log_message( log_module, MSG_INFO,"===================================================================\n");
fclose(testfile);
}
else
log_message( log_module, MSG_INFO,"Test file %s cannot be open : %s\n", files_sdt[i_file],strerror(errno) );
}
/************************************* Testing the resistance to strange data *********************/
log_message( log_module, MSG_INFO,"===================================================================\n");
char *files_rand[NUM_FILES_TEST_READ_RAND]={FILES_TEST_READ_RAND};
for(int i_file=0;i_file<NUM_FILES_TEST_READ_RAND;i_file++)
{
log_message( log_module, MSG_INFO,"Testing Resistance to bad packets file %d on %d : %s\n", i_file+1, NUM_FILES_TEST_READ_RAND, files_rand[i_file] );
press_enter_func(press_enter);
FILE *testfile;
testfile=fopen (files_rand[i_file] , "r");
if(testfile!=NULL)
{
//We read all the packets contained in the file
unsigned char ts_packet_raw[TS_PACKET_SIZE];
int num_rand_read=0;
mumudvb_ts_packet_t ts_packet_mumu;
//Just to make pthread happy
pthread_mutex_init(&ts_packet_mumu.packetmutex,NULL);
int iRet;
log_message( log_module, MSG_INFO,"File opened, reading packets\n" );
while(fread(ts_packet_raw,TS_PACKET_SIZE,1, testfile))
{
num_rand_read++;
log_message( log_module, MSG_INFO,"Position %d packets\n", num_rand_read);
iRet=get_ts_packet(ts_packet_raw, &ts_packet_mumu);
if(iRet==1)//packet is parsed
{
log_message( log_module, MSG_INFO,"New VALID packet\n" );
}
}
fclose(testfile);
}
else
log_message( log_module, MSG_INFO,"Test file %s cannot be open : %s\n", files_rand[i_file], strerror(errno) );
}
/************************ Testing autoconfiguration with a dump **********************************/
char *files_autoconf[NUM_FILES_TEST_AUTOCONF]={FILES_TEST_AUTOCONF};
for(int i_file=0;i_file<NUM_FILES_TEST_AUTOCONF;i_file++)
{
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing autoconfiguration file %d on %d %s\n",i_file+1, NUM_FILES_TEST_AUTOCONF, files_autoconf[i_file]);
press_enter_func(press_enter);
FILE *testfile;
testfile=fopen (files_autoconf[i_file] , "r");
if(testfile!=NULL)
{
int iret;
mumudvb_chan_and_pids_t chan_and_pids;
memset(&chan_and_pids,0,sizeof(mumudvb_chan_and_pids_t));
chan_and_pids.number_of_channels=0;
//autoconfiguration
autoconf_parameters_t autoconf_vars;
memset(&autoconf_vars,0,sizeof(autoconf_parameters_t));
autoconf_vars.autoconfiguration=AUTOCONF_MODE_FULL;
autoconf_vars.autoconf_radios=1;
autoconf_vars.autoconf_scrambled=1;
strcpy (autoconf_vars.autoconf_ip4,"239.100.%card.%number");
autoconf_vars.transport_stream_id=-1;
unicast_parameters_t unicast_vars;
memset(&unicast_vars,0,sizeof(unicast_parameters_t));
unicast_vars.unicast=0;
//multicast parameters
multicast_parameters_t multicast_vars;
memset(&multicast_vars,0,sizeof(multicast_parameters_t));
multicast_vars.multicast=0;
fds_t fds;
memset(&fds,0,sizeof(fds_t));
tuning_parameters_t tuneparams;
memset(&tuneparams,0,sizeof(tuning_parameters_t));
iret=autoconf_init(&autoconf_vars, chan_and_pids.channels,chan_and_pids.number_of_channels);
if(iret)
{
log_message( log_module, MSG_INFO,"error with autoconfiguration init\n");
}
unsigned char actual_ts_packet[TS_PACKET_SIZE];
int pid;
int packet_count=0;
while(fread(actual_ts_packet,TS_PACKET_SIZE,1, testfile))
{
// get the pid of the received ts packet
pid = ((actual_ts_packet[1] & 0x1f) << 8) | (actual_ts_packet[2]);
packet_count++;
if(!(packet_count %100))
log_message( log_module, MSG_INFO,"Packet count %d", packet_count);
iret = autoconf_new_packet(pid, actual_ts_packet, &autoconf_vars, &fds, &chan_and_pids, &tuneparams, &multicast_vars, &unicast_vars, 0);
}
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Forcing to finish and re read");
press_enter_func(press_enter);
//if there is a partial service list, we force autoconf to go to the next step
autoconf_vars.time_start_autoconfiguration=1;
autoconf_poll(100000, &autoconf_vars, &chan_and_pids, &tuneparams, &multicast_vars, &fds, &unicast_vars, 0);
press_enter_func(press_enter);
rewind(testfile);
while(fread(actual_ts_packet,TS_PACKET_SIZE,1, testfile))
{
// get the pid of the received ts packet
pid = ((actual_ts_packet[1] & 0x1f) << 8) | (actual_ts_packet[2]);
log_message( log_module, MSG_INFO,"PID %d", pid);
packet_count++;
if(!(packet_count %100))
log_message( log_module, MSG_INFO,"Packet count %d", packet_count);
iret = autoconf_new_packet(pid, actual_ts_packet, &autoconf_vars, &fds, &chan_and_pids, &tuneparams, &multicast_vars, &unicast_vars, 0);
}
fclose(testfile);
log_message( log_module, MSG_INFO,"============ DONE we display the services ========\n");
if(autoconf_vars.services)
{
autoconf_sort_services(autoconf_vars.services);
autoconf_print_services(autoconf_vars.services);
}
else
log_message( log_module, MSG_INFO,"No services or services freed\n");
log_message( log_module, MSG_INFO,"============ DONE we display the channels ========\n");
if(chan_and_pids.number_of_channels)
{
log_streamed_channels(log_module,chan_and_pids.number_of_channels, chan_and_pids.channels, 1, 0, 0, 0,"");
//We can tests other things here like REWRITE etc ....
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"Testing SDT rewrite on this file (%s)\n", files_autoconf[i_file]);
press_enter_func(press_enter);
FILE *testfile;
testfile=fopen (files_autoconf[i_file] , "r");
if(testfile!=NULL)
{
//Parameters for rewriting
rewrite_parameters_t rewrite_vars={
.rewrite_pat = OPTION_OFF,
.pat_version=-1,
.full_pat=NULL,
.pat_needs_update=1,
.full_pat_ok=0,
.pat_continuity_counter=0,
.rewrite_sdt = OPTION_ON,
.sdt_version=-1,
.full_sdt=NULL,
.sdt_needs_update=1,
.full_sdt_ok=0,
.sdt_continuity_counter=0,
.eit_sort=OPTION_OFF,
};
for (int curr_channel = 0; curr_channel < MAX_CHANNELS; curr_channel++)
chan_and_pids.channels[curr_channel].generated_sdt_version=-1;
rewrite_vars.full_sdt=malloc(sizeof(mumudvb_ts_packet_t));
if(rewrite_vars.full_sdt==NULL)
{
log_message( log_module, MSG_ERROR,"Problem with malloc : %s file : %s line %d\n",strerror(errno),__FILE__,__LINE__);
}
memset (rewrite_vars.full_sdt, 0, sizeof( mumudvb_ts_packet_t));//we clear it
pthread_mutex_init(&rewrite_vars.full_sdt->packetmutex,NULL);
while(fread(actual_ts_packet,TS_PACKET_SIZE,1, testfile))
{
// get the pid of the received ts packet
pid = ((actual_ts_packet[1] & 0x1f) << 8) | (actual_ts_packet[2]);
if( (pid == 17) ) //This is a SDT PID
{
//we check the new packet and if it's fully updated we set the skip to 0
if(sdt_rewrite_new_global_packet(actual_ts_packet, &rewrite_vars)==1)
{
log_message( log_module, MSG_DETAIL,"The SDT version changed, we force the update of all the channels.\n");
for (int curr_channel = 0; curr_channel < chan_and_pids.number_of_channels; curr_channel++)
chan_and_pids.channels[curr_channel].sdt_rewrite_skip=0;
}
for (int curr_channel = 0; curr_channel < chan_and_pids.number_of_channels; curr_channel++)
{
if(!chan_and_pids.channels[curr_channel].sdt_rewrite_skip ) //AND the generation was successful
sdt_rewrite_new_channel_packet(actual_ts_packet, &rewrite_vars, &chan_and_pids.channels[curr_channel], curr_channel);
}
}
}
fclose(testfile);
log_message( log_module, MSG_INFO,"===================================================================\n");
log_message( log_module, MSG_INFO,"End of testing SDT rewritef\n");
log_message( log_module, MSG_INFO,"===================================================================\n");
}
}
else