code.vuplus.com Git - vuplus_dvbapp/blob - lib/dvb/epgcache.cpp

   1 #include <lib/dvb/epgcache.h>
   2 #include <lib/dvb/dvb.h>
   3
   4 #undef EPG_DEBUG
   5
   6 #include <time.h>
   7 #include <unistd.h>  // for usleep
   8 #include <sys/vfs.h> // for statfs
   9 // #include <libmd5sum.h>
  10 #include <lib/base/eerror.h>
  11 #include <lib/dvb/pmt.h>
  12 #include <lib/dvb/db.h>
  13 #include <Python.h>
  14
  15 int eventData::CacheSize=0;
  16 descriptorMap eventData::descriptors;
  17 __u8 eventData::data[4108];
  18 extern const uint32_t crc32_table[256];
  19
  20 eventData::eventData(const eit_event_struct* e, int size, int type)
  21         :ByteSize(size&0xFF), type(type&0xFF)
  22 {
  23         if (!e)
  24                 return;
  25
  26         __u32 descr[65];
  27         __u32 *pdescr=descr;
  28
  29         __u8 *data = (__u8*)e;
  30         int ptr=10;
  31         int descriptors_length = (data[ptr++]&0x0F) << 8;
  32         descriptors_length |= data[ptr++];
  33         while ( descriptors_length > 0 )
  34         {
  35                 __u8 *descr = data+ptr;
  36                 int descr_len = descr[1]+2;
  37
  38                 __u32 crc = 0;
  39                 int cnt=0;
  40                 while(cnt++ < descr_len)
  41                         crc = (crc << 8) ^ crc32_table[((crc >> 24) ^ data[ptr++]) & 0xFF];
  42
  43                 descriptorMap::iterator it =
  44                         descriptors.find(crc);
  45                 if ( it == descriptors.end() )
  46                 {
  47                         CacheSize+=descr_len;
  48                         __u8 *d = new __u8[descr_len];
  49                         memcpy(d, descr, descr_len);
  50                         descriptors[crc] = descriptorPair(1, d);
  51                 }
  52                 else
  53                         ++it->second.first;
  54
  55                 *pdescr++=crc;
  56                 descriptors_length -= descr_len;
  57         }
  58         ByteSize = 12+((pdescr-descr)*4);
  59         EITdata = new __u8[ByteSize];
  60         CacheSize+=ByteSize;
  61         memcpy(EITdata, (__u8*) e, 12);
  62         memcpy(EITdata+12, descr, ByteSize-12);
  63 }
  64
  65 const eit_event_struct* eventData::get() const
  66 {
  67         int pos = 12;
  68         int tmp = ByteSize-12;
  69         memcpy(data, EITdata, 12);
  70         __u32 *p = (__u32*)(EITdata+12);
  71         while(tmp>0)
  72         {
  73                 descriptorMap::iterator it =
  74                         descriptors.find(*p++);
  75                 if ( it != descriptors.end() )
  76                 {
  77                         int b = it->second.second[1]+2;
  78                         memcpy(data+pos, it->second.second, b );
  79                         pos += b;
  80                 }
  81                 tmp-=4;
  82         }
  83
  84         return (const eit_event_struct*)data;
  85 }
  86
  87 eventData::~eventData()
  88 {
  89         if ( ByteSize )
  90         {
  91                 CacheSize-=ByteSize;
  92                 ByteSize-=12;
  93                 __u32 *d = (__u32*)(EITdata+12);
  94                 while(ByteSize)
  95                 {
  96                         descriptorMap::iterator it =
  97                                 descriptors.find(*d++);
  98                         if ( it != descriptors.end() )
  99                         {
 100                                 descriptorPair &p = it->second;
 101                                 if (!--p.first) // no more used descriptor
 102                                 {
 103                                         CacheSize -= it->second.second[1];
 104                                         delete [] it->second.second;    // free descriptor memory
 105                                         descriptors.erase(it);  // remove entry from descriptor map
 106                                 }
 107                         }
 108                         ByteSize-=4;
 109                 }
 110                 delete [] EITdata;
 111         }
 112 }
 113
 114 void eventData::load(FILE *f)
 115 {
 116         int size=0;
 117         int id=0;
 118         __u8 header[2];
 119         descriptorPair p;
 120         fread(&size, sizeof(int), 1, f);
 121         while(size)
 122         {
 123                 fread(&id, sizeof(__u32), 1, f);
 124                 fread(&p.first, sizeof(int), 1, f);
 125                 fread(header, 2, 1, f);
 126                 int bytes = header[1]+2;
 127                 p.second = new __u8[bytes];
 128                 p.second[0] = header[0];
 129                 p.second[1] = header[1];
 130                 fread(p.second+2, bytes-2, 1, f);
 131                 descriptors[id]=p;
 132                 --size;
 133                 CacheSize+=bytes;
 134         }
 135 }
 136
 137 void eventData::save(FILE *f)
 138 {
 139         int size=descriptors.size();
 140         descriptorMap::iterator it(descriptors.begin());
 141         fwrite(&size, sizeof(int), 1, f);
 142         while(size)
 143         {
 144                 fwrite(&it->first, sizeof(__u32), 1, f);
 145                 fwrite(&it->second.first, sizeof(int), 1, f);
 146                 fwrite(it->second.second, it->second.second[1]+2, 1, f);
 147                 ++it;
 148                 --size;
 149         }
 150 }
 151
 152 eEPGCache* eEPGCache::instance;
 153 pthread_mutex_t eEPGCache::cache_lock=
 154         PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
 155 pthread_mutex_t eEPGCache::channel_map_lock=
 156         PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
 157
 158 DEFINE_REF(eEPGCache)
 159
 160 eEPGCache::eEPGCache()
 161         :messages(this,1), cleanTimer(this)//, paused(0)
 162 {
 163         eDebug("[EPGC] Initialized EPGCache");
 164
 165         CONNECT(messages.recv_msg, eEPGCache::gotMessage);
 166         CONNECT(eDVBLocalTimeHandler::getInstance()->m_timeUpdated, eEPGCache::timeUpdated);
 167         CONNECT(cleanTimer.timeout, eEPGCache::cleanLoop);
 168
 169         ePtr<eDVBResourceManager> res_mgr;
 170         eDVBResourceManager::getInstance(res_mgr);
 171         if (!res_mgr)
 172                 eDebug("[eEPGCache] no resource manager !!!!!!!");
 173         else
 174                 res_mgr->connectChannelAdded(slot(*this,&eEPGCache::DVBChannelAdded), m_chanAddedConn);
 175         instance=this;
 176 }
 177
 178 void eEPGCache::timeUpdated()
 179 {
 180         if ( !thread_running() )
 181         {
 182                 eDebug("[EPGC] time updated.. start EPG Mainloop");
 183                 run();
 184         }
 185         else
 186                 messages.send(Message(Message::timeChanged));
 187 }
 188
 189 void eEPGCache::DVBChannelAdded(eDVBChannel *chan)
 190 {
 191         if ( chan )
 192         {
 193 //              eDebug("[eEPGCache] add channel %p", chan);
 194                 channel_data *data = new channel_data(this);
 195                 data->channel = chan;
 196                 data->prevChannelState = -1;
 197 #ifdef ENABLE_PRIVATE_EPG
 198                 data->m_PrivatePid = -1;
 199 #endif
 200                 singleLock s(channel_map_lock);
 201                 m_knownChannels.insert( std::pair<iDVBChannel*, channel_data* >(chan, data) );
 202                 chan->connectStateChange(slot(*this, &eEPGCache::DVBChannelStateChanged), data->m_stateChangedConn);
 203         }
 204 }
 205
 206 void eEPGCache::DVBChannelRunning(iDVBChannel *chan)
 207 {
 208         singleLock s(channel_map_lock);
 209         channelMapIterator it =
 210                 m_knownChannels.find(chan);
 211         if ( it == m_knownChannels.end() )
 212                 eDebug("[eEPGCache] will start non existing channel %p !!!", chan);
 213         else
 214         {
 215                 channel_data &data = *it->second;
 216                 ePtr<eDVBResourceManager> res_mgr;
 217                 if ( eDVBResourceManager::getInstance( res_mgr ) )
 218                         eDebug("[eEPGCache] no res manager!!");
 219                 else
 220                 {
 221                         ePtr<iDVBDemux> demux;
 222                         if ( data.channel->getDemux(demux, 0) )
 223                         {
 224                                 eDebug("[eEPGCache] no demux!!");
 225                                 return;
 226                         }
 227                         else
 228                         {
 229                                 RESULT res = demux->createSectionReader( this, data.m_NowNextReader );
 230                                 if ( res )
 231                                 {
 232                                         eDebug("[eEPGCache] couldnt initialize nownext reader!!");
 233                                         return;
 234                                 }
 235
 236                                 res = demux->createSectionReader( this, data.m_ScheduleReader );
 237                                 if ( res )
 238                                 {
 239                                         eDebug("[eEPGCache] couldnt initialize schedule reader!!");
 240                                         return;
 241                                 }
 242
 243                                 res = demux->createSectionReader( this, data.m_ScheduleOtherReader );
 244                                 if ( res )
 245                                 {
 246                                         eDebug("[eEPGCache] couldnt initialize schedule other reader!!");
 247                                         return;
 248                                 }
 249 #ifdef ENABLE_PRIVATE_EPG
 250                                 res = demux->createSectionReader( this, data.m_PrivateReader );
 251                                 if ( res )
 252                                 {
 253                                         eDebug("[eEPGCache] couldnt initialize private reader!!");
 254                                         return;
 255                                 }
 256 #endif
 257                                 messages.send(Message(Message::startChannel, chan));
 258                                 // -> gotMessage -> changedService
 259                         }
 260                 }
 261         }
 262 }
 263
 264 void eEPGCache::DVBChannelStateChanged(iDVBChannel *chan)
 265 {
 266         channelMapIterator it =
 267                 m_knownChannels.find(chan);
 268         if ( it != m_knownChannels.end() )
 269         {
 270                 int state=0;
 271                 chan->getState(state);
 272                 if ( it->second->prevChannelState != state )
 273                 {
 274                         switch (state)
 275                         {
 276                                 case iDVBChannel::state_ok:
 277                                 {
 278                                         eDebug("[eEPGCache] channel %p running", chan);
 279                                         DVBChannelRunning(chan);
 280                                         break;
 281                                 }
 282                                 case iDVBChannel::state_release:
 283                                 {
 284                                         eDebug("[eEPGCache] remove channel %p", chan);
 285                                         messages.send(Message(Message::leaveChannel, chan));
 286                                         while(!it->second->can_delete)
 287                                                 usleep(1000);
 288                                         delete it->second;
 289                                         m_knownChannels.erase(it);
 290                                         // -> gotMessage -> abortEPG
 291                                         break;
 292                                 }
 293                                 default: // ignore all other events
 294                                         return;
 295                         }
 296                         it->second->prevChannelState = state;
 297                 }
 298         }
 299 }
 300
 301 void eEPGCache::sectionRead(const __u8 *data, int source, channel_data *channel)
 302 {
 303         eit_t *eit = (eit_t*) data;
 304
 305         int len=HILO(eit->section_length)-1;//+3-4;
 306         int ptr=EIT_SIZE;
 307         if ( ptr >= len )
 308                 return;
 309
 310         // This fixed the EPG on the Multichoice irdeto systems
 311         // the EIT packet is non-compliant.. their EIT packet stinks
 312         if ( data[ptr-1] < 0x40 )
 313                 --ptr;
 314
 315         uniqueEPGKey service( HILO(eit->service_id), HILO(eit->original_network_id), HILO(eit->transport_stream_id) );
 316         eit_event_struct* eit_event = (eit_event_struct*) (data+ptr);
 317         int eit_event_size;
 318         int duration;
 319
 320         time_t TM = parseDVBtime( eit_event->start_time_1, eit_event->start_time_2,     eit_event->start_time_3, eit_event->start_time_4, eit_event->start_time_5);
 321         time_t now = time(0)+eDVBLocalTimeHandler::getInstance()->difference();
 322
 323         if ( TM != 3599 && TM > -1)
 324                 channel->haveData |= source;
 325
 326         singleLock s(cache_lock);
 327         // hier wird immer eine eventMap zurück gegeben.. entweder eine vorhandene..
 328         // oder eine durch [] erzeugte
 329         std::pair<eventMap,timeMap> &servicemap = eventDB[service];
 330         eventMap::iterator prevEventIt = servicemap.first.end();
 331         timeMap::iterator prevTimeIt = servicemap.second.end();
 332
 333         while (ptr<len)
 334         {
 335                 eit_event_size = HILO(eit_event->descriptors_loop_length)+EIT_LOOP_SIZE;
 336
 337                 duration = fromBCD(eit_event->duration_1)*3600+fromBCD(eit_event->duration_2)*60+fromBCD(eit_event->duration_3);
 338                 TM = parseDVBtime(
 339                         eit_event->start_time_1,
 340                         eit_event->start_time_2,
 341                         eit_event->start_time_3,
 342                         eit_event->start_time_4,
 343                         eit_event->start_time_5);
 344
 345                 if ( TM == 3599 )
 346                         goto next;
 347
 348                 if ( TM != 3599 && (TM+duration < now || TM > now+14*24*60*60) )
 349                         goto next;
 350
 351                 if ( now <= (TM+duration) || TM == 3599 /*NVOD Service*/ )  // old events should not be cached
 352                 {
 353                         __u16 event_id = HILO(eit_event->event_id);
 354 //                      eDebug("event_id is %d sid is %04x", event_id, service.sid);
 355
 356                         eventData *evt = 0;
 357                         int ev_erase_count = 0;
 358                         int tm_erase_count = 0;
 359
 360                         // search in eventmap
 361                         eventMap::iterator ev_it =
 362                                 servicemap.first.find(event_id);
 363
 364                         // entry with this event_id is already exist ?
 365                         if ( ev_it != servicemap.first.end() )
 366                         {
 367                                 if ( source > ev_it->second->type )  // update needed ?
 368                                         goto next; // when not.. the skip this entry
 369
 370                                 // search this event in timemap
 371                                 timeMap::iterator tm_it_tmp =
 372                                         servicemap.second.find(ev_it->second->getStartTime());
 373
 374                                 if ( tm_it_tmp != servicemap.second.end() )
 375                                 {
 376                                         if ( tm_it_tmp->first == TM ) // correct eventData
 377                                         {
 378                                                 // exempt memory
 379                                                 delete ev_it->second;
 380                                                 evt = new eventData(eit_event, eit_event_size, source);
 381                                                 ev_it->second=evt;
 382                                                 tm_it_tmp->second=evt;
 383                                                 goto next;
 384                                         }
 385                                         else
 386                                         {
 387                                                 tm_erase_count++;
 388                                                 // delete the found record from timemap
 389                                                 servicemap.second.erase(tm_it_tmp);
 390                                                 prevTimeIt=servicemap.second.end();
 391                                         }
 392                                 }
 393                         }
 394
 395                         // search in timemap, for check of a case if new time has coincided with time of other event
 396                         // or event was is not found in eventmap
 397                         timeMap::iterator tm_it =
 398                                 servicemap.second.find(TM);
 399
 400                         if ( tm_it != servicemap.second.end() )
 401                         {
 402                                 // i think, if event is not found on eventmap, but found on timemap updating nevertheless demands
 403 #if 0
 404                                 if ( source > tm_it->second->type && tm_erase_count == 0 ) // update needed ?
 405                                         goto next; // when not.. the skip this entry
 406 #endif
 407
 408                                 // search this time in eventmap
 409                                 eventMap::iterator ev_it_tmp =
 410                                         servicemap.first.find(tm_it->second->getEventID());
 411
 412                                 if ( ev_it_tmp != servicemap.first.end() )
 413                                 {
 414                                         ev_erase_count++;
 415                                         // delete the found record from eventmap
 416                                         servicemap.first.erase(ev_it_tmp);
 417                                         prevEventIt=servicemap.first.end();
 418                                 }
 419                         }
 420
 421                         evt = new eventData(eit_event, eit_event_size, source);
 422 #if EPG_DEBUG
 423                         bool consistencyCheck=true;
 424 #endif
 425                         if (ev_erase_count > 0 && tm_erase_count > 0) // 2 different pairs have been removed
 426                         {
 427                                 // exempt memory
 428                                 delete ev_it->second;
 429                                 delete tm_it->second;
 430                                 ev_it->second=evt;
 431                                 tm_it->second=evt;
 432                         }
 433                         else if (ev_erase_count == 0 && tm_erase_count > 0)
 434                         {
 435                                 // exempt memory
 436                                 delete ev_it->second;
 437                                 tm_it=prevTimeIt=servicemap.second.insert( prevTimeIt, std::pair<const time_t, eventData*>( TM, evt ) );
 438                                 ev_it->second=evt;
 439                         }
 440                         else if (ev_erase_count > 0 && tm_erase_count == 0)
 441                         {
 442                                 // exempt memory
 443                                 delete tm_it->second;
 444                                 ev_it=prevEventIt=servicemap.first.insert( prevEventIt, std::pair<const __u16, eventData*>( event_id, evt) );
 445                                 tm_it->second=evt;
 446                         }
 447                         else // added new eventData
 448                         {
 449 #if EPG_DEBUG
 450                                 consistencyCheck=false;
 451 #endif
 452                                 prevEventIt=servicemap.first.insert( prevEventIt, std::pair<const __u16, eventData*>( event_id, evt) );
 453                                 prevTimeIt=servicemap.second.insert( prevTimeIt, std::pair<const time_t, eventData*>( TM, evt ) );
 454                         }
 455 #if EPG_DEBUG
 456                         if ( consistencyCheck )
 457                         {
 458                                 if ( tm_it->second != evt || ev_it->second != evt )
 459                                         eFatal("tm_it->second != ev_it->second");
 460                                 else if ( tm_it->second->getStartTime() != tm_it->first )
 461                                         eFatal("event start_time(%d) non equal timemap key(%d)",
 462                                                 tm_it->second->getStartTime(), tm_it->first );
 463                                 else if ( tm_it->first != TM )
 464                                         eFatal("timemap key(%d) non equal TM(%d)",
 465                                                 tm_it->first, TM);
 466                                 else if ( ev_it->second->getEventID() != ev_it->first )
 467                                         eFatal("event_id (%d) non equal event_map key(%d)",
 468                                                 ev_it->second->getEventID(), ev_it->first);
 469                                 else if ( ev_it->first != event_id )
 470                                         eFatal("eventmap key(%d) non equal event_id(%d)",
 471                                                 ev_it->first, event_id );
 472                         }
 473 #endif
 474                 }
 475 next:
 476 #if EPG_DEBUG
 477                 if ( servicemap.first.size() != servicemap.second.size() )
 478                 {
 479                         FILE *f = fopen("/hdd/event_map.txt", "w+");
 480                         int i=0;
 481                         for (eventMap::iterator it(servicemap.first.begin())
 482                                 ; it != servicemap.first.end(); ++it )
 483                                 fprintf(f, "%d(key %d) -> time %d, event_id %d, data %p\n",
 484                                         i++, (int)it->first, (int)it->second->getStartTime(), (int)it->second->getEventID(), it->second );
 485                         fclose(f);
 486                         f = fopen("/hdd/time_map.txt", "w+");
 487                         i=0;
 488                         for (timeMap::iterator it(servicemap.second.begin())
 489                                 ; it != servicemap.second.end(); ++it )
 490                                         fprintf(f, "%d(key %d) -> time %d, event_id %d, data %p\n",
 491                                                 i++, (int)it->first, (int)it->second->getStartTime(), (int)it->second->getEventID(), it->second );
 492                         fclose(f);
 493
 494                         eFatal("(1)map sizes not equal :( sid %04x tsid %04x onid %04x size %d size2 %d",
 495                                 service.sid, service.tsid, service.onid,
 496                                 servicemap.first.size(), servicemap.second.size() );
 497                 }
 498 #endif
 499                 ptr += eit_event_size;
 500                 eit_event=(eit_event_struct*)(((__u8*)eit_event)+eit_event_size);
 501         }
 502 }
 503
 504 void eEPGCache::flushEPG(const uniqueEPGKey & s)
 505 {
 506         eDebug("[EPGC] flushEPG %d", (int)(bool)s);
 507         singleLock l(cache_lock);
 508         if (s)  // clear only this service
 509         {
 510                 eventCache::iterator it = eventDB.find(s);
 511                 if ( it != eventDB.end() )
 512                 {
 513                         eventMap &evMap = it->second.first;
 514                         timeMap &tmMap = it->second.second;
 515                         tmMap.clear();
 516                         for (eventMap::iterator i = evMap.begin(); i != evMap.end(); ++i)
 517                                 delete i->second;
 518                         evMap.clear();
 519                         eventDB.erase(it);
 520
 521                         // TODO .. search corresponding channel for removed service and remove this channel from lastupdated map
 522 #ifdef ENABLE_PRIVATE_EPG
 523                         contentMaps::iterator it =
 524                                 content_time_tables.find(s);
 525                         if ( it != content_time_tables.end() )
 526                         {
 527                                 it->second.clear();
 528                                 content_time_tables.erase(it);
 529                         }
 530 #endif
 531                 }
 532         }
 533         else // clear complete EPG Cache
 534         {
 535                 for (eventCache::iterator it(eventDB.begin());
 536                         it != eventDB.end(); ++it)
 537                 {
 538                         eventMap &evMap = it->second.first;
 539                         timeMap &tmMap = it->second.second;
 540                         for (eventMap::iterator i = evMap.begin(); i != evMap.end(); ++i)
 541                                 delete i->second;
 542                         evMap.clear();
 543                         tmMap.clear();
 544                 }
 545                 eventDB.clear();
 546 #ifdef ENABLE_PRIVATE_EPG
 547                 content_time_tables.clear();
 548 #endif
 549                 channelLastUpdated.clear();
 550                 singleLock m(channel_map_lock);
 551                 for (channelMapIterator it(m_knownChannels.begin()); it != m_knownChannels.end(); ++it)
 552                         it->second->startEPG();
 553         }
 554         eDebug("[EPGC] %i bytes for cache used", eventData::CacheSize);
 555 }
 556
 557 void eEPGCache::cleanLoop()
 558 {
 559         singleLock s(cache_lock);
 560         if (!eventDB.empty())
 561         {
 562                 eDebug("[EPGC] start cleanloop");
 563
 564                 time_t now = time(0)+eDVBLocalTimeHandler::getInstance()->difference();
 565
 566                 for (eventCache::iterator DBIt = eventDB.begin(); DBIt != eventDB.end(); DBIt++)
 567                 {
 568                         bool updated = false;
 569                         for (timeMap::iterator It = DBIt->second.second.begin(); It != DBIt->second.second.end() && It->first < now;)
 570                         {
 571                                 if ( now > (It->first+It->second->getDuration()) )  // outdated normal entry (nvod references to)
 572                                 {
 573                                         // remove entry from eventMap
 574                                         eventMap::iterator b(DBIt->second.first.find(It->second->getEventID()));
 575                                         if ( b != DBIt->second.first.end() )
 576                                         {
 577                                                 // release Heap Memory for this entry   (new ....)
 578 //                                              eDebug("[EPGC] delete old event (evmap)");
 579                                                 DBIt->second.first.erase(b);
 580                                         }
 581
 582                                         // remove entry from timeMap
 583 //                                      eDebug("[EPGC] release heap mem");
 584                                         delete It->second;
 585                                         DBIt->second.second.erase(It++);
 586 //                                      eDebug("[EPGC] delete old event (timeMap)");
 587                                         updated = true;
 588                                 }
 589                                 else
 590                                         ++It;
 591                         }
 592 #ifdef ENABLE_PRIVATE_EPG
 593                         if ( updated )
 594                         {
 595                                 contentMaps::iterator x =
 596                                         content_time_tables.find( DBIt->first );
 597                                 if ( x != content_time_tables.end() )
 598                                 {
 599                                         timeMap &tmMap = eventDB[DBIt->first].second;
 600                                         for ( contentMap::iterator i = x->second.begin(); i != x->second.end(); )
 601                                         {
 602                                                 for ( contentTimeMap::iterator it(i->second.begin());
 603                                                         it != i->second.end(); )
 604                                                 {
 605                                                         if ( tmMap.find(it->second.first) == tmMap.end() )
 606                                                                 i->second.erase(it++);
 607                                                         else
 608                                                                 ++it;
 609                                                 }
 610                                                 if ( i->second.size() )
 611                                                         ++i;
 612                                                 else
 613                                                         x->second.erase(i++);
 614                                         }
 615                                 }
 616                         }
 617 #endif
 618                 }
 619                 eDebug("[EPGC] stop cleanloop");
 620                 eDebug("[EPGC] %i bytes for cache used", eventData::CacheSize);
 621         }
 622         cleanTimer.start(CLEAN_INTERVAL,true);
 623 }
 624
 625 eEPGCache::~eEPGCache()
 626 {
 627         messages.send(Message::quit);
 628         kill(); // waiting for thread shutdown
 629         singleLock s(cache_lock);
 630         for (eventCache::iterator evIt = eventDB.begin(); evIt != eventDB.end(); evIt++)
 631                 for (eventMap::iterator It = evIt->second.first.begin(); It != evIt->second.first.end(); It++)
 632                         delete It->second;
 633 }
 634
 635 void eEPGCache::gotMessage( const Message &msg )
 636 {
 637         switch (msg.type)
 638         {
 639                 case Message::flush:
 640                         flushEPG(msg.service);
 641                         break;
 642                 case Message::startChannel:
 643                 {
 644                         singleLock s(channel_map_lock);
 645                         channelMapIterator channel =
 646                                 m_knownChannels.find(msg.channel);
 647                         if ( channel != m_knownChannels.end() )
 648                                 channel->second->startChannel();
 649                         break;
 650                 }
 651                 case Message::leaveChannel:
 652                 {
 653                         singleLock s(channel_map_lock);
 654                         channelMapIterator channel =
 655                                 m_knownChannels.find(msg.channel);
 656                         if ( channel != m_knownChannels.end() )
 657                                 channel->second->abortEPG();
 658                         break;
 659                 }
 660                 case Message::quit:
 661                         quit(0);
 662                         break;
 663 #ifdef ENABLE_PRIVATE_EPG
 664                 case Message::got_private_pid:
 665                 {
 666                         for (channelMapIterator it(m_knownChannels.begin()); it != m_knownChannels.end(); ++it)
 667                         {
 668                                 eDVBChannel *channel = (eDVBChannel*) it->first;
 669                                 channel_data *data = it->second;
 670                                 eDVBChannelID chid = channel->getChannelID();
 671                                 if ( chid.transport_stream_id.get() == msg.service.tsid &&
 672                                         chid.original_network_id.get() == msg.service.onid &&
 673                                         data->m_PrivatePid == -1 )
 674                                 {
 675                                         data->m_PrivatePid = msg.pid;
 676                                         data->m_PrivateService = msg.service;
 677                                         data->startPrivateReader(msg.pid, -1);
 678                                         break;
 679                                 }
 680                         }
 681                         break;
 682                 }
 683 #endif
 684                 case Message::timeChanged:
 685                         cleanLoop();
 686                         break;
 687                 default:
 688                         eDebug("unhandled EPGCache Message!!");
 689                         break;
 690         }
 691 }
 692
 693 void eEPGCache::thread()
 694 {
 695         nice(4);
 696         load();
 697         cleanLoop();
 698         runLoop();
 699         save();
 700 }
 701
 702 void eEPGCache::load()
 703 {
 704         singleLock s(cache_lock);
 705         FILE *f = fopen("/hdd/epg.dat", "r");
 706         if (f)
 707         {
 708                 int size=0;
 709                 int cnt=0;
 710 #if 0
 711                 unsigned char md5_saved[16];
 712                 unsigned char md5[16];
 713                 bool md5ok=false;
 714
 715                 if (!md5_file("/hdd/epg.dat", 1, md5))
 716                 {
 717                         FILE *f = fopen("/hdd/epg.dat.md5", "r");
 718                         if (f)
 719                         {
 720                                 fread( md5_saved, 16, 1, f);
 721                                 fclose(f);
 722                                 if ( !memcmp(md5_saved, md5, 16) )
 723                                         md5ok=true;
 724                         }
 725                 }
 726                 if ( md5ok )
 727 #endif
 728                 {
 729                         unsigned int magic=0;
 730                         fread( &magic, sizeof(int), 1, f);
 731                         if (magic != 0x98765432)
 732                         {
 733                                 eDebug("epg file has incorrect byte order.. dont read it");
 734                                 fclose(f);
 735                                 return;
 736                         }
 737                         char text1[13];
 738                         fread( text1, 13, 1, f);
 739                         if ( !strncmp( text1, "ENIGMA_EPG_V5", 13) )
 740                         {
 741                                 fread( &size, sizeof(int), 1, f);
 742                                 while(size--)
 743                                 {
 744                                         uniqueEPGKey key;
 745                                         eventMap evMap;
 746                                         timeMap tmMap;
 747                                         int size=0;
 748                                         fread( &key, sizeof(uniqueEPGKey), 1, f);
 749                                         fread( &size, sizeof(int), 1, f);
 750                                         while(size--)
 751                                         {
 752                                                 __u8 len=0;
 753                                                 __u8 type=0;
 754                                                 eventData *event=0;
 755                                                 fread( &type, sizeof(__u8), 1, f);
 756                                                 fread( &len, sizeof(__u8), 1, f);
 757                                                 event = new eventData(0, len, type);
 758                                                 event->EITdata = new __u8[len];
 759                                                 eventData::CacheSize+=len;
 760                                                 fread( event->EITdata, len, 1, f);
 761                                                 evMap[ event->getEventID() ]=event;
 762                                                 tmMap[ event->getStartTime() ]=event;
 763                                                 ++cnt;
 764                                         }
 765                                         eventDB[key]=std::pair<eventMap,timeMap>(evMap,tmMap);
 766                                 }
 767                                 eventData::load(f);
 768                                 eDebug("%d events read from /hdd/epg.dat", cnt);
 769 #ifdef ENABLE_PRIVATE_EPG
 770                                 char text2[11];
 771                                 fread( text2, 11, 1, f);
 772                                 if ( !strncmp( text2, "PRIVATE_EPG", 11) )
 773                                 {
 774                                         size=0;
 775                                         fread( &size, sizeof(int), 1, f);
 776                                         while(size--)
 777                                         {
 778                                                 int size=0;
 779                                                 uniqueEPGKey key;
 780                                                 fread( &key, sizeof(uniqueEPGKey), 1, f);
 781                                                 fread( &size, sizeof(int), 1, f);
 782                                                 while(size--)
 783                                                 {
 784                                                         int size;
 785                                                         int content_id;
 786                                                         fread( &content_id, sizeof(int), 1, f);
 787                                                         fread( &size, sizeof(int), 1, f);
 788                                                         while(size--)
 789                                                         {
 790                                                                 time_t time1, time2;
 791                                                                 __u16 event_id;
 792                                                                 fread( &time1, sizeof(time_t), 1, f);
 793                                                                 fread( &time2, sizeof(time_t), 1, f);
 794                                                                 fread( &event_id, sizeof(__u16), 1, f);
 795                                                                 content_time_tables[key][content_id][time1]=std::pair<time_t, __u16>(time2, event_id);
 796                                                         }
 797                                                 }
 798                                         }
 799                                 }
 800 #endif // ENABLE_PRIVATE_EPG
 801                         }
 802                         else
 803                                 eDebug("[EPGC] don't read old epg database");
 804                         fclose(f);
 805                 }
 806         }
 807 }
 808
 809 void eEPGCache::save()
 810 {
 811         struct statfs s;
 812         off64_t tmp;
 813         if (statfs("/hdd", &s)<0)
 814                 tmp=0;
 815         else
 816         {
 817                 tmp=s.f_blocks;
 818                 tmp*=s.f_bsize;
 819         }
 820
 821         // prevent writes to builtin flash
 822         if ( tmp < 1024*1024*50 ) // storage size < 50MB
 823                 return;
 824
 825         // check for enough free space on storage
 826         tmp=s.f_bfree;
 827         tmp*=s.f_bsize;
 828         if ( tmp < (eventData::CacheSize*12)/10 ) // 20% overhead
 829                 return;
 830
 831         FILE *f = fopen("/hdd/epg.dat", "w");
 832         int cnt=0;
 833         if ( f )
 834         {
 835                 unsigned int magic = 0x98765432;
 836                 fwrite( &magic, sizeof(int), 1, f);
 837                 const char *text = "ENIGMA_EPG_V5";
 838                 fwrite( text, 13, 1, f );
 839                 int size = eventDB.size();
 840                 fwrite( &size, sizeof(int), 1, f );
 841                 for (eventCache::iterator service_it(eventDB.begin()); service_it != eventDB.end(); ++service_it)
 842                 {
 843                         timeMap &timemap = service_it->second.second;
 844                         fwrite( &service_it->first, sizeof(uniqueEPGKey), 1, f);
 845                         size = timemap.size();
 846                         fwrite( &size, sizeof(int), 1, f);
 847                         for (timeMap::iterator time_it(timemap.begin()); time_it != timemap.end(); ++time_it)
 848                         {
 849                                 __u8 len = time_it->second->ByteSize;
 850                                 fwrite( &time_it->second->type, sizeof(__u8), 1, f );
 851                                 fwrite( &len, sizeof(__u8), 1, f);
 852                                 fwrite( time_it->second->EITdata, len, 1, f);
 853                                 ++cnt;
 854                         }
 855                 }
 856                 eDebug("%d events written to /hdd/epg.dat", cnt);
 857                 eventData::save(f);
 858 #ifdef ENABLE_PRIVATE_EPG
 859                 const char* text3 = "PRIVATE_EPG";
 860                 fwrite( text3, 11, 1, f );
 861                 size = content_time_tables.size();
 862                 fwrite( &size, sizeof(int), 1, f);
 863                 for (contentMaps::iterator a = content_time_tables.begin(); a != content_time_tables.end(); ++a)
 864                 {
 865                         contentMap &content_time_table = a->second;
 866                         fwrite( &a->first, sizeof(uniqueEPGKey), 1, f);
 867                         int size = content_time_table.size();
 868                         fwrite( &size, sizeof(int), 1, f);
 869                         for (contentMap::iterator i = content_time_table.begin(); i != content_time_table.end(); ++i )
 870                         {
 871                                 int size = i->second.size();
 872                                 fwrite( &i->first, sizeof(int), 1, f);
 873                                 fwrite( &size, sizeof(int), 1, f);
 874                                 for ( contentTimeMap::iterator it(i->second.begin());
 875                                         it != i->second.end(); ++it )
 876                                 {
 877                                         fwrite( &it->first, sizeof(time_t), 1, f);
 878                                         fwrite( &it->second.first, sizeof(time_t), 1, f);
 879                                         fwrite( &it->second.second, sizeof(__u16), 1, f);
 880                                 }
 881                         }
 882                 }
 883 #endif
 884                 fclose(f);
 885 #if 0
 886                 unsigned char md5[16];
 887                 if (!md5_file("/hdd/epg.dat", 1, md5))
 888                 {
 889                         FILE *f = fopen("/hdd/epg.dat.md5", "w");
 890                         if (f)
 891                         {
 892                                 fwrite( md5, 16, 1, f);
 893                                 fclose(f);
 894                         }
 895                 }
 896 #endif
 897         }
 898 }
 899
 900 eEPGCache::channel_data::channel_data(eEPGCache *ml)
 901         :cache(ml)
 902         ,abortTimer(ml), zapTimer(ml)
 903         ,state(0), isRunning(0), haveData(0), can_delete(1)
 904 {
 905         CONNECT(zapTimer.timeout, eEPGCache::channel_data::startEPG);
 906         CONNECT(abortTimer.timeout, eEPGCache::channel_data::abortNonAvail);
 907 }
 908
 909 bool eEPGCache::channel_data::finishEPG()
 910 {
 911         if (!isRunning)  // epg ready
 912         {
 913                 eDebug("[EPGC] stop caching events(%d)", time(0)+eDVBLocalTimeHandler::getInstance()->difference());
 914                 zapTimer.start(UPDATE_INTERVAL, 1);
 915                 eDebug("[EPGC] next update in %i min", UPDATE_INTERVAL / 60000);
 916                 for (int i=0; i < 3; ++i)
 917                 {
 918                         seenSections[i].clear();
 919                         calcedSections[i].clear();
 920                 }
 921                 singleLock l(cache->cache_lock);
 922                 cache->channelLastUpdated[channel->getChannelID()] = time(0)+eDVBLocalTimeHandler::getInstance()->difference();
 923 #ifdef ENABLE_PRIVATE_EPG
 924                 if (seenPrivateSections.empty())
 925 #endif
 926                 can_delete=1;
 927                 return true;
 928         }
 929         return false;
 930 }
 931
 932 void eEPGCache::channel_data::startEPG()
 933 {
 934         eDebug("[EPGC] start caching events(%d)", eDVBLocalTimeHandler::getInstance()->difference()+time(0));
 935         state=0;
 936         haveData=0;
 937         can_delete=0;
 938         for (int i=0; i < 3; ++i)
 939         {
 940                 seenSections[i].clear();
 941                 calcedSections[i].clear();
 942         }
 943
 944         eDVBSectionFilterMask mask;
 945         memset(&mask, 0, sizeof(mask));
 946         mask.pid = 0x12;
 947         mask.flags = eDVBSectionFilterMask::rfCRC;
 948
 949         mask.data[0] = 0x4E;
 950         mask.mask[0] = 0xFE;
 951         m_NowNextReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_NowNextConn);
 952         m_NowNextReader->start(mask);
 953         isRunning |= NOWNEXT;
 954
 955         mask.data[0] = 0x50;
 956         mask.mask[0] = 0xF0;
 957         m_ScheduleReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_ScheduleConn);
 958         m_ScheduleReader->start(mask);
 959         isRunning |= SCHEDULE;
 960
 961         mask.data[0] = 0x60;
 962         mask.mask[0] = 0xF0;
 963         m_ScheduleOtherReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_ScheduleOtherConn);
 964         m_ScheduleOtherReader->start(mask);
 965         isRunning |= SCHEDULE_OTHER;
 966
 967         abortTimer.start(7000,true);
 968 }
 969
 970 void eEPGCache::channel_data::abortNonAvail()
 971 {
 972         if (!state)
 973         {
 974                 if ( !(haveData&eEPGCache::NOWNEXT) && (isRunning&eEPGCache::NOWNEXT) )
 975                 {
 976                         eDebug("[EPGC] abort non avail nownext reading");
 977                         isRunning &= ~eEPGCache::NOWNEXT;
 978                         m_NowNextReader->stop();
 979                         m_NowNextConn=0;
 980                 }
 981                 if ( !(haveData&eEPGCache::SCHEDULE) && (isRunning&eEPGCache::SCHEDULE) )
 982                 {
 983                         eDebug("[EPGC] abort non avail schedule reading");
 984                         isRunning &= ~SCHEDULE;
 985                         m_ScheduleReader->stop();
 986                         m_ScheduleConn=0;
 987                 }
 988                 if ( !(haveData&eEPGCache::SCHEDULE_OTHER) && (isRunning&eEPGCache::SCHEDULE_OTHER) )
 989                 {
 990                         eDebug("[EPGC] abort non avail schedule_other reading");
 991                         isRunning &= ~SCHEDULE_OTHER;
 992                         m_ScheduleOtherReader->stop();
 993                         m_ScheduleOtherConn=0;
 994                 }
 995                 if ( isRunning )
 996                         abortTimer.start(90000, true);
 997                 else
 998                 {
 999                         ++state;
1000                         for (int i=0; i < 3; ++i)
1001                         {
1002                                 seenSections[i].clear();
1003                                 calcedSections[i].clear();
1004                         }
1005 #ifdef ENABLE_PRIVATE_EPG
1006                         if (seenPrivateSections.empty())
1007 #endif
1008                         can_delete=1;
1009                 }
1010         }
1011         ++state;
1012 }
1013
1014 void eEPGCache::channel_data::startChannel()
1015 {
1016         updateMap::iterator It = cache->channelLastUpdated.find( channel->getChannelID() );
1017
1018         int update = ( It != cache->channelLastUpdated.end() ? ( UPDATE_INTERVAL - ( (time(0)+eDVBLocalTimeHandler::getInstance()->difference()-It->second) * 1000 ) ) : ZAP_DELAY );
1019
1020         if (update < ZAP_DELAY)
1021                 update = ZAP_DELAY;
1022
1023         zapTimer.start(update, 1);
1024         if (update >= 60000)
1025                 eDebug("[EPGC] next update in %i min", update/60000);
1026         else if (update >= 1000)
1027                 eDebug("[EPGC] next update in %i sec", update/1000);
1028 }
1029
1030 void eEPGCache::channel_data::abortEPG()
1031 {
1032         for (int i=0; i < 3; ++i)
1033         {
1034                 seenSections[i].clear();
1035                 calcedSections[i].clear();
1036         }
1037         abortTimer.stop();
1038         zapTimer.stop();
1039         if (isRunning)
1040         {
1041                 eDebug("[EPGC] abort caching events !!");
1042                 if (isRunning & eEPGCache::SCHEDULE)
1043                 {
1044                         isRunning &= ~eEPGCache::SCHEDULE;
1045                         m_ScheduleReader->stop();
1046                         m_ScheduleConn=0;
1047                 }
1048                 if (isRunning & eEPGCache::NOWNEXT)
1049                 {
1050                         isRunning &= ~eEPGCache::NOWNEXT;
1051                         m_NowNextReader->stop();
1052                         m_NowNextConn=0;
1053                 }
1054                 if (isRunning & SCHEDULE_OTHER)
1055                 {
1056                         isRunning &= ~eEPGCache::SCHEDULE_OTHER;
1057                         m_ScheduleOtherReader->stop();
1058                         m_ScheduleOtherConn=0;
1059                 }
1060         }
1061 #ifdef ENABLE_PRIVATE_EPG
1062         if (m_PrivateReader)
1063                 m_PrivateReader->stop();
1064         if (m_PrivateConn)
1065                 m_PrivateConn=0;
1066 #endif
1067         can_delete=1;
1068 }
1069
1070 void eEPGCache::channel_data::readData( const __u8 *data)
1071 {
1072         if (!data)
1073                 eDebug("get Null pointer from section reader !!");
1074         else
1075         {
1076                 int source;
1077                 int map;
1078                 iDVBSectionReader *reader=NULL;
1079                 switch(data[0])
1080                 {
1081                         case 0x4E ... 0x4F:
1082                                 reader=m_NowNextReader;
1083                                 source=eEPGCache::NOWNEXT;
1084                                 map=0;
1085                                 break;
1086                         case 0x50 ... 0x5F:
1087                                 reader=m_ScheduleReader;
1088                                 source=eEPGCache::SCHEDULE;
1089                                 map=1;
1090                                 break;
1091                         case 0x60 ... 0x6F:
1092                                 reader=m_ScheduleOtherReader;
1093                                 source=eEPGCache::SCHEDULE_OTHER;
1094                                 map=2;
1095                                 break;
1096                         default:
1097                                 eDebug("[EPGC] unknown table_id !!!");
1098                                 return;
1099                 }
1100                 tidMap &seenSections = this->seenSections[map];
1101                 tidMap &calcedSections = this->calcedSections[map];
1102                 if ( state == 1 && calcedSections == seenSections || state > 1 )
1103                 {
1104                         eDebugNoNewLine("[EPGC] ");
1105                         switch (source)
1106                         {
1107                                 case eEPGCache::NOWNEXT:
1108                                         m_NowNextConn=0;
1109                                         eDebugNoNewLine("nownext");
1110                                         break;
1111                                 case eEPGCache::SCHEDULE:
1112                                         m_ScheduleConn=0;
1113                                         eDebugNoNewLine("schedule");
1114                                         break;
1115                                 case eEPGCache::SCHEDULE_OTHER:
1116                                         m_ScheduleOtherConn=0;
1117                                         eDebugNoNewLine("schedule other");
1118                                         break;
1119                                 default: eDebugNoNewLine("unknown");break;
1120                         }
1121                         eDebug(" finished(%d)", time(0)+eDVBLocalTimeHandler::getInstance()->difference());
1122                         if ( reader )
1123                                 reader->stop();
1124                         isRunning &= ~source;
1125                         if (!isRunning)
1126                                 finishEPG();
1127                 }
1128                 else
1129                 {
1130                         eit_t *eit = (eit_t*) data;
1131                         __u32 sectionNo = data[0] << 24;
1132                         sectionNo |= data[3] << 16;
1133                         sectionNo |= data[4] << 8;
1134                         sectionNo |= eit->section_number;
1135
1136                         tidMap::iterator it =
1137                                 seenSections.find(sectionNo);
1138
1139                         if ( it == seenSections.end() )
1140                         {
1141                                 seenSections.insert(sectionNo);
1142                                 calcedSections.insert(sectionNo);
1143                                 __u32 tmpval = sectionNo & 0xFFFFFF00;
1144                                 __u8 incr = source == NOWNEXT ? 1 : 8;
1145                                 for ( int i = 0; i <= eit->last_section_number; i+=incr )
1146                                 {
1147                                         if ( i == eit->section_number )
1148                                         {
1149                                                 for (int x=i; x <= eit->segment_last_section_number; ++x)
1150                                                         calcedSections.insert(tmpval|(x&0xFF));
1151                                         }
1152                                         else
1153                                                 calcedSections.insert(tmpval|(i&0xFF));
1154                                 }
1155                                 cache->sectionRead(data, source, this);
1156                         }
1157                 }
1158         }
1159 }
1160
1161 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, const eventData *&result, int direction)
1162 // if t == -1 we search the current event...
1163 {
1164         singleLock s(cache_lock);
1165         uniqueEPGKey key(service);
1166
1167         // check if EPG for this service is ready...
1168         eventCache::iterator It = eventDB.find( key );
1169         if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached ?
1170         {
1171                 if (t==-1)
1172                         t = time(0)+eDVBLocalTimeHandler::getInstance()->difference();
1173                 timeMap::iterator i = direction <= 0 ? It->second.second.lower_bound(t) :  // find > or equal
1174                         It->second.second.upper_bound(t); // just >
1175                 if ( i != It->second.second.end() )
1176                 {
1177                         if ( direction < 0 || (direction == 0 && i->second->getStartTime() > t) )
1178                         {
1179                                 timeMap::iterator x = i;
1180                                 --x;
1181                                 if ( x != It->second.second.end() )
1182                                 {
1183                                         time_t start_time = x->second->getStartTime();
1184                                         if (direction >= 0)
1185                                         {
1186                                                 if (t < start_time)
1187                                                         return -1;
1188                                                 if (t > (start_time+x->second->getDuration()))
1189                                                         return -1;
1190                                         }
1191                                         i = x;
1192                                 }
1193                                 else
1194                                         return -1;
1195                         }
1196                         result = i->second;
1197                         return 0;
1198                 }
1199         }
1200         return -1;
1201 }
1202
1203 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, const eit_event_struct *&result, int direction)
1204 {
1205         singleLock s(cache_lock);
1206         const eventData *data=0;
1207         RESULT ret = lookupEventTime(service, t, data, direction);
1208         if ( !ret && data )
1209                 result = data->get();
1210         return ret;
1211 }
1212
1213 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, Event *& result, int direction)
1214 {
1215         singleLock s(cache_lock);
1216         const eventData *data=0;
1217         RESULT ret = lookupEventTime(service, t, data, direction);
1218         if ( !ret && data )
1219                 result = new Event((uint8_t*)data->get());
1220         return ret;
1221 }
1222
1223 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, ePtr<eServiceEvent> &result, int direction)
1224 {
1225         singleLock s(cache_lock);
1226         const eventData *data=0;
1227         RESULT ret = lookupEventTime(service, t, data, direction);
1228         if ( !ret && data )
1229         {
1230                 Event ev((uint8_t*)data->get());
1231                 result = new eServiceEvent();
1232                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1233                 ret = result->parseFrom(&ev, (ref.getTransportStreamID().get()<<16)|ref.getOriginalNetworkID().get());
1234         }
1235         return ret;
1236 }
1237
1238 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eventData *&result )
1239 {
1240         singleLock s(cache_lock);
1241         uniqueEPGKey key( service );
1242
1243         eventCache::iterator It = eventDB.find( key );
1244         if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached?
1245         {
1246                 eventMap::iterator i( It->second.first.find( event_id ));
1247                 if ( i != It->second.first.end() )
1248                 {
1249                         result = i->second;
1250                         return 0;
1251                 }
1252                 else
1253                 {
1254                         result = 0;
1255                         eDebug("event %04x not found in epgcache", event_id);
1256                 }
1257         }
1258         return -1;
1259 }
1260
1261 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eit_event_struct *&result)
1262 {
1263         singleLock s(cache_lock);
1264         const eventData *data=0;
1265         RESULT ret = lookupEventId(service, event_id, data);
1266         if ( !ret && data )
1267                 result = data->get();
1268         return ret;
1269 }
1270
1271 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, Event *& result)
1272 {
1273         singleLock s(cache_lock);
1274         const eventData *data=0;
1275         RESULT ret = lookupEventId(service, event_id, data);
1276         if ( !ret && data )
1277                 result = new Event((uint8_t*)data->get());
1278         return ret;
1279 }
1280
1281 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, ePtr<eServiceEvent> &result)
1282 {
1283         singleLock s(cache_lock);
1284         const eventData *data=0;
1285         RESULT ret = lookupEventId(service, event_id, data);
1286         if ( !ret && data )
1287         {
1288                 Event ev((uint8_t*)data->get());
1289                 result = new eServiceEvent();
1290                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1291                 ret = result->parseFrom(&ev, (ref.getTransportStreamID().get()<<16)|ref.getOriginalNetworkID().get());
1292         }
1293         return ret;
1294 }
1295
1296 RESULT eEPGCache::startTimeQuery(const eServiceReference &service, time_t begin, int minutes)
1297 {
1298         eventCache::iterator It = eventDB.find( service );
1299         if ( It != eventDB.end() && It->second.second.size() )
1300         {
1301                 m_timemap_end = minutes != -1 ? It->second.second.upper_bound(begin+minutes*60) : It->second.second.end();
1302                 if ( begin != -1 )
1303                 {
1304                         m_timemap_cursor = It->second.second.lower_bound(begin);
1305                         if ( m_timemap_cursor != It->second.second.end() )
1306                         {
1307                                 if ( m_timemap_cursor->second->getStartTime() != begin )
1308                                 {
1309                                         timeMap::iterator x = m_timemap_cursor;
1310                                         --x;
1311                                         if ( x != It->second.second.end() )
1312                                         {
1313                                                 time_t start_time = x->second->getStartTime();
1314                                                 if ( begin > start_time && begin < (start_time+x->second->getDuration()))
1315                                                         m_timemap_cursor = x;
1316                                         }
1317                                 }
1318                         }
1319                 }
1320                 else
1321                         m_timemap_cursor = It->second.second.begin();
1322                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1323                 currentQueryTsidOnid = (ref.getTransportStreamID().get()<<16) | ref.getOriginalNetworkID().get();
1324                 return 0;
1325         }
1326         return -1;
1327 }
1328
1329 RESULT eEPGCache::getNextTimeEntry(const eventData *& result)
1330 {
1331         if ( m_timemap_cursor != m_timemap_end )
1332         {
1333                 result = m_timemap_cursor++->second;
1334                 return 0;
1335         }
1336         return -1;
1337 }
1338
1339 RESULT eEPGCache::getNextTimeEntry(const eit_event_struct *&result)
1340 {
1341         if ( m_timemap_cursor != m_timemap_end )
1342         {
1343                 result = m_timemap_cursor++->second->get();
1344                 return 0;
1345         }
1346         return -1;
1347 }
1348
1349 RESULT eEPGCache::getNextTimeEntry(Event *&result)
1350 {
1351         if ( m_timemap_cursor != m_timemap_end )
1352         {
1353                 result = new Event((uint8_t*)m_timemap_cursor++->second->get());
1354                 return 0;
1355         }
1356         return -1;
1357 }
1358
1359 RESULT eEPGCache::getNextTimeEntry(ePtr<eServiceEvent> &result)
1360 {
1361         if ( m_timemap_cursor != m_timemap_end )
1362         {
1363                 Event ev((uint8_t*)m_timemap_cursor++->second->get());
1364                 result = new eServiceEvent();
1365                 return result->parseFrom(&ev, currentQueryTsidOnid);
1366         }
1367         return -1;
1368 }
1369
1370 void fillTuple(PyObject *tuple, char *argstring, int argcount, PyObject *service, ePtr<eServiceEvent> &ptr, PyObject *nowTime, PyObject *service_name )
1371 {
1372         PyObject *tmp=NULL;
1373         int pos=0;
1374         while(pos < argcount)
1375         {
1376                 bool inc_refcount=false;
1377                 switch(argstring[pos])
1378                 {
1379                         case '0': // PyLong 0
1380                                 tmp = PyLong_FromLong(0);
1381                                 break;
1382                         case 'I': // Event Id
1383                                 tmp = ptr ? PyLong_FromLong(ptr->getEventId()) : NULL;
1384                                 break;
1385                         case 'B': // Event Begin Time
1386                                 tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : NULL;
1387                                 break;
1388                         case 'D': // Event Duration
1389                                 tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : NULL;
1390                                 break;
1391                         case 'T': // Event Title
1392                                 tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : NULL;
1393                                 break;
1394                         case 'S': // Event Short Description
1395                                 tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : NULL;
1396                                 break;
1397                         case 'E': // Event Extended Description
1398                                 tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : NULL;
1399                                 break;
1400                         case 'C': // Current Time
1401                                 tmp = nowTime;
1402                                 inc_refcount = true;
1403                                 break;
1404                         case 'R': // service reference string
1405                                 tmp = service;
1406                                 inc_refcount = true;
1407                                 break;
1408                         case 'N': // service name
1409                                 tmp = service_name;
1410                                 inc_refcount = true;
1411                 }
1412                 if (!tmp)
1413                 {
1414                         tmp = Py_None;
1415                         inc_refcount = true;
1416                 }
1417                 if (inc_refcount)
1418                         Py_INCREF(tmp);
1419                 PyTuple_SET_ITEM(tuple, pos++, tmp);
1420         }
1421 }
1422
1423 PyObject *handleEvent(ePtr<eServiceEvent> &ptr, PyObject *dest_list, char* argstring, int argcount, PyObject *service, PyObject *nowTime, PyObject *service_name, PyObject *convertFunc, PyObject *convertFuncArgs)
1424 {
1425         if (convertFunc)
1426         {
1427                 fillTuple(convertFuncArgs, argstring, argcount, service, ptr, nowTime, service_name);
1428                 PyObject *result = PyObject_CallObject(convertFunc, convertFuncArgs);
1429                 if (result == NULL)
1430                 {
1431                         if (service_name)
1432                                 Py_DECREF(service_name);
1433                         if (nowTime)
1434                                 Py_DECREF(nowTime);
1435                         Py_DECREF(convertFuncArgs);
1436                         Py_DECREF(dest_list);
1437                         return result;
1438                 }
1439                 PyList_Append(dest_list, result);
1440                 Py_DECREF(result);
1441         }
1442         else
1443         {
1444                 PyObject *tuple = PyTuple_New(argcount);
1445                 fillTuple(tuple, argstring, argcount, service, ptr, nowTime, service_name);
1446                 PyList_Append(dest_list, tuple);
1447                 Py_DECREF(tuple);
1448         }
1449         return 0;
1450 }
1451
1452 // here we get a python list
1453 // the first entry in the list is a python string to specify the format of the returned tuples (in a list)
1454 //   0 = PyLong(0)
1455 //   I = Event Id
1456 //   B = Event Begin Time
1457 //   D = Event Duration
1458 //   T = Event Title
1459 //   S = Event Short Description
1460 //   E = Event Extended Description
1461 //   C = Current Time
1462 //   R = Service Reference
1463 //   N = Service Name
1464 // then for each service follows a tuple
1465 //   first tuple entry is the servicereference (as string... use the ref.toString() function)
1466 //   the second is the type of query
1467 //     2 = event_id
1468 //    -1 = event before given start_time
1469 //     0 = event intersects given start_time
1470 //    +1 = event after given start_time
1471 //   the third
1472 //      when type is eventid it is the event_id
1473 //      when type is time then it is the start_time ( 0 for now_time )
1474 //   the fourth is the end_time .. ( optional .. for query all events in time range)
1475
1476 PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc)
1477 {
1478         PyObject *convertFuncArgs=NULL;
1479         int argcount=0;
1480         char *argstring=NULL;
1481         if (!PyList_Check(list))
1482         {
1483                 PyErr_SetString(PyExc_StandardError,
1484                         "type error");
1485                 eDebug("no list");
1486                 return NULL;
1487         }
1488         int listIt=0;
1489         int listSize=PyList_Size(list);
1490         if (!listSize)
1491         {
1492                 PyErr_SetString(PyExc_StandardError,
1493                         "not params given");
1494                 eDebug("not params given");
1495                 return NULL;
1496         }
1497         else
1498         {
1499                 PyObject *argv=PyList_GET_ITEM(list, 0); // borrowed reference!
1500                 if (PyString_Check(argv))
1501                 {
1502                         argstring = PyString_AS_STRING(argv);
1503                         ++listIt;
1504                 }
1505                 else
1506                         argstring = "I"; // just event id as default
1507                 argcount = strlen(argstring);
1508 //              eDebug("have %d args('%s')", argcount, argstring);
1509         }
1510         if (convertFunc)
1511         {
1512                 if (!PyCallable_Check(convertFunc))
1513                 {
1514                         PyErr_SetString(PyExc_StandardError,
1515                                 "convertFunc must be callable");
1516                         eDebug("convertFunc is not callable");
1517                         return NULL;
1518                 }
1519                 convertFuncArgs = PyTuple_New(argcount);
1520         }
1521
1522         PyObject *nowTime = strchr(argstring, 'C') ?
1523                 PyLong_FromLong(time(0)+eDVBLocalTimeHandler::getInstance()->difference()) :
1524                 NULL;
1525
1526         bool must_get_service_name = strchr(argstring, 'N') ? true : false;
1527
1528         // create dest list
1529         PyObject *dest_list=PyList_New(0);
1530         while(listSize > listIt)
1531         {
1532                 PyObject *item=PyList_GET_ITEM(list, listIt++); // borrowed reference!
1533                 if (PyTuple_Check(item))
1534                 {
1535                         int type=0;
1536                         long event_id=-1;
1537                         time_t stime=-1;
1538                         int minutes=0;
1539                         int tupleSize=PyTuple_Size(item);
1540                         int tupleIt=0;
1541                         PyObject *service=NULL;
1542                         while(tupleSize > tupleIt)  // parse query args
1543                         {
1544                                 PyObject *entry=PyTuple_GET_ITEM(item, tupleIt); // borrowed reference!
1545                                 switch(tupleIt++)
1546                                 {
1547                                         case 0:
1548                                         {
1549                                                 if (!PyString_Check(entry))
1550                                                 {
1551                                                         eDebug("tuple entry 0 is no a string");
1552                                                         goto skip_entry;
1553                                                 }
1554                                                 service = entry;
1555                                                 break;
1556                                         }
1557                                         case 1:
1558                                                 type=PyInt_AsLong(entry);
1559                                                 if (type < -1 || type > 2)
1560                                                 {
1561                                                         eDebug("unknown type %d", type);
1562                                                         goto skip_entry;
1563                                                 }
1564                                                 break;
1565                                         case 2:
1566                                                 event_id=stime=PyInt_AsLong(entry);
1567                                                 break;
1568                                         case 3:
1569                                                 minutes=PyInt_AsLong(entry);
1570                                                 break;
1571                                         default:
1572                                                 eDebug("unneeded extra argument");
1573                                                 break;
1574                                 }
1575                         }
1576                         eServiceReference ref(PyString_AS_STRING(service));
1577                         if (ref.type != eServiceReference::idDVB)
1578                         {
1579                                 eDebug("service reference for epg query is not valid");
1580                                 continue;
1581                         }
1582                         PyObject *service_name=NULL;
1583                         if (must_get_service_name)
1584                         {
1585                                 ePtr<iStaticServiceInformation> sptr;
1586                                 eServiceCenterPtr service_center;
1587                                 eServiceCenter::getPrivInstance(service_center);
1588                                 if (service_center)
1589                                 {
1590                                         service_center->info(ref, sptr);
1591                                         if (sptr)
1592                                         {
1593                                                 std::string name;
1594                                                 sptr->getName(ref, name);
1595                                                 if (name.length())
1596                                                         service_name = PyString_FromString(name.c_str());
1597                                         }
1598                                 }
1599                                 if (!service_name)
1600                                         service_name = PyString_FromString("<n/a>");
1601                         }
1602                         if (minutes)
1603                         {
1604                                 Lock();
1605                                 if (!startTimeQuery(ref, stime, minutes))
1606                                 {
1607                                         ePtr<eServiceEvent> ptr;
1608                                         while (!getNextTimeEntry(ptr))
1609                                         {
1610                                                 PyObject *ret = handleEvent(ptr, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs);
1611                                                 if (ret)
1612                                                         return ret;
1613                                         }
1614                                 }
1615                                 Unlock();
1616                         }
1617                         else
1618                         {
1619                                 ePtr<eServiceEvent> ptr;
1620                                 if (stime)
1621                                 {
1622                                         if (type == 2)
1623                                                 lookupEventId(ref, event_id, ptr);
1624                                         else
1625                                                 lookupEventTime(ref, stime, ptr, type);
1626                                 }
1627                                 PyObject *ret = handleEvent(ptr, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs);
1628                                 if (ret)
1629                                         return ret;
1630                         }
1631                         if (service_name)
1632                                 Py_DECREF(service_name);
1633                 }
1634 skip_entry:
1635                 ;
1636         }
1637         if (convertFuncArgs)
1638                 Py_DECREF(convertFuncArgs);
1639         if (nowTime)
1640                 Py_DECREF(nowTime);
1641         return dest_list;
1642 }
1643
1644 void fillTuple2(PyObject *tuple, const char *argstring, int argcount, eventData *evData, ePtr<eServiceEvent> &ptr, PyObject *service_name, PyObject *service_reference)
1645 {
1646         PyObject *tmp=NULL;
1647         int pos=0;
1648         while(pos < argcount)
1649         {
1650                 bool inc_refcount=false;
1651                 switch(argstring[pos])
1652                 {
1653                         case '0': // PyLong 0
1654                                 tmp = PyLong_FromLong(0);
1655                                 break;
1656                         case 'I': // Event Id
1657                                 tmp = PyLong_FromLong(evData->getEventID());
1658                                 break;
1659                         case 'B': // Event Begin Time
1660                                 if (ptr)
1661                                         tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : NULL;
1662                                 else
1663                                         tmp = PyLong_FromLong(evData->getStartTime());
1664                                 break;
1665                         case 'D': // Event Duration
1666                                 if (ptr)
1667                                         tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : NULL;
1668                                 else
1669                                         tmp = PyLong_FromLong(evData->getDuration());
1670                                 break;
1671                         case 'T': // Event Title
1672                                 tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : NULL;
1673                                 break;
1674                         case 'S': // Event Short Description
1675                                 tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : NULL;
1676                                 break;
1677                         case 'E': // Event Extended Description
1678                                 tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : NULL;
1679                                 break;
1680                         case 'R': // service reference string
1681                                 tmp = service_reference;
1682                                 inc_refcount = true;
1683                                 break;
1684                         case 'N': // service name
1685                                 tmp = service_name;
1686                                 inc_refcount = true;
1687                                 break;
1688                 }
1689                 if (!tmp)
1690                 {
1691                         tmp = Py_None;
1692                         inc_refcount = true;
1693                 }
1694                 if (inc_refcount)
1695                         Py_INCREF(tmp);
1696                 PyTuple_SET_ITEM(tuple, pos++, tmp);
1697         }
1698 }
1699
1700 // here we get a python tuple
1701 // the first entry in the tuple is a python string to specify the format of the returned tuples (in a list)
1702 //   I = Event Id
1703 //   B = Event Begin Time
1704 //   D = Event Duration
1705 //   T = Event Title
1706 //   S = Event Short Description
1707 //   E = Event Extended Description
1708 //   R = Service Reference
1709 //   N = Service Name
1710 //  the second tuple entry is the MAX matches value
1711 //  the third tuple entry is the type of query
1712 //     0 = search for similar broadcastings (SIMILAR_BROADCASTINGS_SEARCH)
1713 //     1 = search events with exactly title name (EXAKT_TITLE_SEARCH)
1714 //     2 = search events with text in title name (PARTIAL_TITLE_SEARCH)
1715 //  when type is 0 (SIMILAR_BROADCASTINGS_SEARCH)
1716 //   the fourth is the servicereference tring
1717 //   the fifth is the eventid
1718 //  when type is 1 or 2 (EXAKT_TITLE_SEARCH or PARTIAL_TITLE_SEARCH)
1719 //   the fourth is the search text
1720 //   the fifth is
1721 //     0 = case sensitive (CASE_CHECK)
1722 //     1 = case insensitive (NO_CASECHECK)
1723
1724 PyObject *eEPGCache::search(PyObject *arg)
1725 {
1726         PyObject *ret = 0;
1727         int descridx = -1;
1728         __u32 descr[512];
1729         int eventid = -1;
1730         const char *argstring=0;
1731         int argcount=0;
1732         int querytype=-1;
1733         bool needServiceEvent=false;
1734         int maxmatches=0;
1735
1736         if (PyTuple_Check(arg))
1737         {
1738                 int tuplesize=PyTuple_Size(arg);
1739                 if (tuplesize > 0)
1740                 {
1741                         PyObject *obj = PyTuple_GET_ITEM(arg,0);
1742                         if (PyString_Check(obj))
1743                         {
1744                                 argcount = PyString_GET_SIZE(obj);
1745                                 argstring = PyString_AS_STRING(obj);
1746                                 for (int i=0; i < argcount; ++i)
1747                                         switch(argstring[i])
1748                                         {
1749                                         case 'S':
1750                                         case 'E':
1751                                         case 'T':
1752                                                 needServiceEvent=true;
1753                                         default:
1754                                                 break;
1755                                         }
1756                         }
1757                         else
1758                         {
1759                                 PyErr_SetString(PyExc_StandardError,
1760                                         "type error");
1761                                 eDebug("tuple arg 0 is not a string");
1762                                 return NULL;
1763                         }
1764                 }
1765                 if (tuplesize > 1)
1766                         maxmatches = PyLong_AsLong(PyTuple_GET_ITEM(arg, 1));
1767                 if (tuplesize > 2)
1768                 {
1769                         querytype = PyLong_AsLong(PyTuple_GET_ITEM(arg, 2));
1770                         if (tuplesize > 4 && querytype == 0)
1771                         {
1772                                 PyObject *obj = PyTuple_GET_ITEM(arg, 3);
1773                                 if (PyString_Check(obj))
1774                                 {
1775                                         const char *refstr = PyString_AS_STRING(obj);
1776                                         eServiceReferenceDVB ref(refstr);
1777                                         if (ref.valid())
1778                                         {
1779                                                 eventid = PyLong_AsLong(PyTuple_GET_ITEM(arg, 4));
1780                                                 singleLock s(cache_lock);
1781                                                 const eventData *evData = 0;
1782                                                 lookupEventId(ref, eventid, evData);
1783                                                 if (evData)
1784                                                 {
1785                                                         __u8 *data = evData->EITdata;
1786                                                         int tmp = evData->ByteSize-12;
1787                                                         __u32 *p = (__u32*)(data+12);
1788                                                                 // search short and extended event descriptors
1789                                                         while(tmp>0)
1790                                                         {
1791                                                                 __u32 crc = *p++;
1792                                                                 descriptorMap::iterator it =
1793                                                                         eventData::descriptors.find(crc);
1794                                                                 if (it != eventData::descriptors.end())
1795                                                                 {
1796                                                                         __u8 *descr_data = it->second.second;
1797                                                                         switch(descr_data[0])
1798                                                                         {
1799                                                                         case 0x4D ... 0x4E:
1800                                                                                 descr[++descridx]=crc;
1801                                                                         default:
1802                                                                                 break;
1803                                                                         }
1804                                                                 }
1805                                                                 tmp-=4;
1806                                                         }
1807                                                 }
1808                                                 if (descridx<0)
1809                                                         eDebug("event not found");
1810                                         }
1811                                         else
1812                                         {
1813                                                 PyErr_SetString(PyExc_StandardError,
1814                                                         "type error");
1815                                                 eDebug("tuple arg 4 is not a valid service reference string");
1816                                                 return NULL;
1817                                         }
1818                                 }
1819                                 else
1820                                 {
1821                                         PyErr_SetString(PyExc_StandardError,
1822                                         "type error");
1823                                         eDebug("tuple arg 4 is not a string");
1824                                         return NULL;
1825                                 }
1826                         }
1827                         else if (tuplesize > 4 && (querytype == 1 || querytype == 2) )
1828                         {
1829                                 PyObject *obj = PyTuple_GET_ITEM(arg, 3);
1830                                 if (PyString_Check(obj))
1831                                 {
1832                                         int casetype = PyLong_AsLong(PyTuple_GET_ITEM(arg, 4));
1833                                         const char *str = PyString_AS_STRING(obj);
1834                                         int textlen = PyString_GET_SIZE(obj);
1835                                         if (querytype == 1)
1836                                                 eDebug("lookup for events with '%s' as title(%s)", str, casetype?"ignore case":"case sensitive");
1837                                         else
1838                                                 eDebug("lookup for events with '%s' in title(%s)", str, casetype?"ignore case":"case sensitive");
1839                                         singleLock s(cache_lock);
1840                                         for (descriptorMap::iterator it(eventData::descriptors.begin());
1841                                                 it != eventData::descriptors.end() && descridx < 511; ++it)
1842                                         {
1843                                                 __u8 *data = it->second.second;
1844                                                 if ( data[0] == 0x4D ) // short event descriptor
1845                                                 {
1846                                                         int title_len = data[5];
1847                                                         if ( querytype == 1 )
1848                                                         {
1849                                                                 if (title_len > textlen)
1850                                                                         continue;
1851                                                                 else if (title_len < textlen)
1852                                                                         continue;
1853                                                                 if ( casetype )
1854                                                                 {
1855                                                                         if ( !strncasecmp((const char*)data+6, str, title_len) )
1856                                                                         {
1857 //                                                                              std::string s((const char*)data+6, title_len);
1858 //                                                                              eDebug("match1 %s %s", str, s.c_str() );
1859                                                                                 descr[++descridx] = it->first;
1860                                                                         }
1861                                                                 }
1862                                                                 else if ( !strncmp((const char*)data+6, str, title_len) )
1863                                                                 {
1864 //                                                                      std::string s((const char*)data+6, title_len);
1865 //                                                                      eDebug("match2 %s %s", str, s.c_str() );
1866                                                                         descr[++descridx] = it->first;
1867                                                                 }
1868                                                         }
1869                                                         else
1870                                                         {
1871                                                                 int idx=0;
1872                                                                 while((title_len-idx) >= textlen)
1873                                                                 {
1874                                                                         if (casetype)
1875                                                                         {
1876                                                                                 if (!strncasecmp((const char*)data+6+idx, str, textlen) )
1877                                                                                 {
1878                                                                                         descr[++descridx] = it->first;
1879 //                                                                                      std::string s((const char*)data+6, title_len);
1880 //                                                                                      eDebug("match 3 %s %s", str, s.c_str() );
1881                                                                                         break;
1882                                                                                 }
1883                                                                                 else if (!strncmp((const char*)data+6+idx, str, textlen) )
1884                                                                                 {
1885                                                                                         descr[++descridx] = it->first;
1886 //                                                                                      std::string s((const char*)data+6, title_len);
1887 //                                                                                      eDebug("match 4 %s %s", str, s.c_str() );
1888                                                                                         break;
1889                                                                                 }
1890                                                                         }
1891                                                                         ++idx;
1892                                                                 }
1893                                                         }
1894                                                 }
1895                                         }
1896                                 }
1897                                 else
1898                                 {
1899                                         PyErr_SetString(PyExc_StandardError,
1900                                                 "type error");
1901                                         eDebug("tuple arg 4 is not a string");
1902                                         return NULL;
1903                                 }
1904                         }
1905                         else
1906                         {
1907                                 PyErr_SetString(PyExc_StandardError,
1908                                         "type error");
1909                                 eDebug("tuple arg 3(%d) is not a known querytype(0, 1, 2)", querytype);
1910                                 return NULL;
1911                         }
1912                 }
1913                 else
1914                 {
1915                         PyErr_SetString(PyExc_StandardError,
1916                                 "type error");
1917                         eDebug("not enough args in tuple");
1918                         return NULL;
1919                 }
1920         }
1921         else
1922         {
1923                 PyErr_SetString(PyExc_StandardError,
1924                         "type error");
1925                 eDebug("arg 0 is not a tuple");
1926                 return NULL;
1927         }
1928
1929         if (descridx > -1)
1930         {
1931                 int maxcount=maxmatches;
1932                 singleLock s(cache_lock);
1933                 // check all services
1934                 for( eventCache::iterator cit(eventDB.begin()); cit != eventDB.end() && maxcount; ++cit)
1935                 {
1936                         PyObject *service_name=0;
1937                         PyObject *service_reference=0;
1938                         eventMap &evmap = cit->second.first;
1939                         // check all events
1940                         for (eventMap::iterator evit(evmap.begin()); evit != evmap.end() && maxcount; ++evit)
1941                         {
1942                                 __u8 *data = evit->second->EITdata;
1943                                 int tmp = evit->second->ByteSize-12;
1944                                 __u32 *p = (__u32*)(data+12);
1945                                 // check if any of our descriptor used by this event
1946 //                              if (evit->first == eventid )
1947 //                                      continue;
1948                                 int cnt=-1;
1949                                 while(tmp>0)
1950                                 {
1951                                         __u32 crc32 = *p++;
1952                                         for ( int i=0; i <= descridx; ++i)
1953                                         {
1954                                                 if (descr[i] == crc32)  // found...
1955                                                         ++cnt;
1956                                         }
1957                                         tmp-=4;
1958                                 }
1959                                 if ( (querytype == 0 && cnt == descridx) ||
1960                                          ((querytype == 1 || querytype == 2) && cnt != -1) )
1961                                 {
1962                                         const uniqueEPGKey &service = cit->first;
1963                                         eServiceReference ref =
1964                                                 eDVBDB::getInstance()->searchReference(service.tsid, service.onid, service.sid);
1965                                         if (ref.valid())
1966                                         {
1967                                         // create servive event
1968                                                 ePtr<eServiceEvent> ptr;
1969                                                 if (needServiceEvent)
1970                                                 {
1971                                                         lookupEventId(ref, evit->first, ptr);
1972                                                         if (!ptr)
1973                                                                 eDebug("event not found !!!!!!!!!!!");
1974                                                 }
1975                                         // create service name
1976                                                 if (!service_name && strchr(argstring,'N'))
1977                                                 {
1978                                                         ePtr<iStaticServiceInformation> sptr;
1979                                                         eServiceCenterPtr service_center;
1980                                                         eServiceCenter::getPrivInstance(service_center);
1981                                                         if (service_center)
1982                                                         {
1983                                                                 service_center->info(ref, sptr);
1984                                                                 if (sptr)
1985                                                                 {
1986                                                                         std::string name;
1987                                                                         sptr->getName(ref, name);
1988                                                                         if (name.length())
1989                                                                                 service_name = PyString_FromString(name.c_str());
1990                                                                 }
1991                                                         }
1992                                                         if (!service_name)
1993                                                                 service_name = PyString_FromString("<n/a>");
1994                                                 }
1995                                         // create servicereference string
1996                                                 if (!service_reference && strchr(argstring,'R'))
1997                                                         service_reference = PyString_FromString(ref.toString().c_str());
1998                                         // create list
1999                                                 if (!ret)
2000                                                         ret = PyList_New(0);
2001                                         // create tuple
2002                                                 PyObject *tuple = PyTuple_New(argcount);
2003                                         // fill tuple
2004                                                 fillTuple2(tuple, argstring, argcount, evit->second, ptr, service_name, service_reference);
2005                                                 PyList_Append(ret, tuple);
2006                                                 Py_DECREF(tuple);
2007                                                 --maxcount;
2008                                         }
2009                                 }
2010                         }
2011                         if (service_name)
2012                                 Py_DECREF(service_name);
2013                         if (service_reference)
2014                                 Py_DECREF(service_reference);
2015                 }
2016         }
2017
2018         if (!ret)
2019         {
2020                 Py_INCREF(Py_None);
2021                 ret=Py_None;
2022         }
2023
2024         return ret;
2025 }
2026
2027 #ifdef ENABLE_PRIVATE_EPG
2028 #include <dvbsi++/descriptor_tag.h>
2029 #include <dvbsi++/unknown_descriptor.h>
2030 #include <dvbsi++/private_data_specifier_descriptor.h>
2031
2032 void eEPGCache::PMTready(eDVBServicePMTHandler *pmthandler)
2033 {
2034         ePtr<eTable<ProgramMapSection> > ptr;
2035         if (!pmthandler->getPMT(ptr) && ptr)
2036         {
2037                 std::vector<ProgramMapSection*>::const_iterator i;
2038                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
2039                 {
2040                         const ProgramMapSection &pmt = **i;
2041
2042                         ElementaryStreamInfoConstIterator es;
2043                         for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
2044                         {
2045                                 int tmp=0;
2046                                 switch ((*es)->getType())
2047                                 {
2048                                 case 0x05: // private
2049                                         for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
2050                                                 desc != (*es)->getDescriptors()->end(); ++desc)
2051                                         {
2052                                                 switch ((*desc)->getTag())
2053                                                 {
2054                                                         case PRIVATE_DATA_SPECIFIER_DESCRIPTOR:
2055                                                                 if (((PrivateDataSpecifierDescriptor*)(*desc))->getPrivateDataSpecifier() == 190)
2056                                                                         tmp |= 1;
2057                                                                 break;
2058                                                         case 0x90:
2059                                                         {
2060                                                                 UnknownDescriptor *descr = (UnknownDescriptor*)*desc;
2061                                                                 int descr_len = descr->getLength();
2062                                                                 if (descr_len == 4)
2063                                                                 {
2064                                                                         uint8_t data[descr_len+2];
2065                                                                         descr->writeToBuffer(data);
2066                                                                         if ( !data[2] && !data[3] && data[4] == 0xFF && data[5] == 0xFF )
2067                                                                                 tmp |= 2;
2068                                                                 }
2069                                                                 break;
2070                                                         }
2071                                                         default:
2072                                                                 break;
2073                                                 }
2074                                         }
2075                                 default:
2076                                         break;
2077                                 }
2078                                 if (tmp==3)
2079                                 {
2080                                         eServiceReferenceDVB ref;
2081                                         if (!pmthandler->getService(ref))
2082                                         {
2083                                                 int pid = (*es)->getPid();
2084                                                 messages.send(Message(Message::got_private_pid, ref, pid));
2085                                                 return;
2086                                         }
2087                                 }
2088                         }
2089                 }
2090         }
2091         else
2092                 eDebug("PMTready but no pmt!!");
2093 }
2094
2095 struct date_time
2096 {
2097         __u8 data[5];
2098         time_t tm;
2099         date_time( const date_time &a )
2100         {
2101                 memcpy(data, a.data, 5);
2102                 tm = a.tm;
2103         }
2104         date_time( const __u8 data[5])
2105         {
2106                 memcpy(this->data, data, 5);
2107                 tm = parseDVBtime(data[0], data[1], data[2], data[3], data[4]);
2108         }
2109         date_time()
2110         {
2111         }
2112         const __u8& operator[](int pos) const
2113         {
2114                 return data[pos];
2115         }
2116 };
2117
2118 struct less_datetime
2119 {
2120         bool operator()( const date_time &a, const date_time &b ) const
2121         {
2122                 return abs(a.tm-b.tm) < 360 ? false : a.tm < b.tm;
2123         }
2124 };
2125
2126 void eEPGCache::privateSectionRead(const uniqueEPGKey &current_service, const __u8 *data)
2127 {
2128         contentMap &content_time_table = content_time_tables[current_service];
2129         singleLock s(cache_lock);
2130         std::map< date_time, std::list<uniqueEPGKey>, less_datetime > start_times;
2131         eventMap &evMap = eventDB[current_service].first;
2132         timeMap &tmMap = eventDB[current_service].second;
2133         int ptr=8;
2134         int content_id = data[ptr++] << 24;
2135         content_id |= data[ptr++] << 16;
2136         content_id |= data[ptr++] << 8;
2137         content_id |= data[ptr++];
2138
2139         contentTimeMap &time_event_map =
2140                 content_time_table[content_id];
2141         for ( contentTimeMap::iterator it( time_event_map.begin() );
2142                 it != time_event_map.end(); ++it )
2143         {
2144                 eventMap::iterator evIt( evMap.find(it->second.second) );
2145                 if ( evIt != evMap.end() )
2146                 {
2147                         delete evIt->second;
2148                         evMap.erase(evIt);
2149                 }
2150                 tmMap.erase(it->second.first);
2151         }
2152         time_event_map.clear();
2153
2154         __u8 duration[3];
2155         memcpy(duration, data+ptr, 3);
2156         ptr+=3;
2157         int duration_sec =
2158                 fromBCD(duration[0])*3600+fromBCD(duration[1])*60+fromBCD(duration[2]);
2159
2160         const __u8 *descriptors[65];
2161         const __u8 **pdescr = descriptors;
2162
2163         int descriptors_length = (data[ptr++]&0x0F) << 8;
2164         descriptors_length |= data[ptr++];
2165         while ( descriptors_length > 0 )
2166         {
2167                 int descr_type = data[ptr];
2168                 int descr_len = data[ptr+1];
2169                 descriptors_length -= (descr_len+2);
2170                 if ( descr_type == 0xf2 )
2171                 {
2172                         ptr+=2;
2173                         int tsid = data[ptr++] << 8;
2174                         tsid |= data[ptr++];
2175                         int onid = data[ptr++] << 8;
2176                         onid |= data[ptr++];
2177                         int sid = data[ptr++] << 8;
2178                         sid |= data[ptr++];
2179                         uniqueEPGKey service( sid, onid, tsid );
2180                         descr_len -= 6;
2181                         while( descr_len > 0 )
2182                         {
2183                                 __u8 datetime[5];
2184                                 datetime[0] = data[ptr++];
2185                                 datetime[1] = data[ptr++];
2186                                 int tmp_len = data[ptr++];
2187                                 descr_len -= 3;
2188                                 while( tmp_len > 0 )
2189                                 {
2190                                         memcpy(datetime+2, data+ptr, 3);
2191                                         ptr+=3;
2192                                         descr_len -= 3;
2193                                         tmp_len -= 3;
2194                                         start_times[datetime].push_back(service);
2195                                 }
2196                         }
2197                 }
2198                 else
2199                 {
2200                         *pdescr++=data+ptr;
2201                         ptr += 2;
2202                         ptr += descr_len;
2203                 }
2204         }
2205         __u8 event[4098];
2206         eit_event_struct *ev_struct = (eit_event_struct*) event;
2207         ev_struct->running_status = 0;
2208         ev_struct->free_CA_mode = 1;
2209         memcpy(event+7, duration, 3);
2210         ptr = 12;
2211         const __u8 **d=descriptors;
2212         while ( d < pdescr )
2213         {
2214                 memcpy(event+ptr, *d, ((*d)[1])+2);
2215                 ptr+=(*d++)[1];
2216                 ptr+=2;
2217         }
2218         for ( std::map< date_time, std::list<uniqueEPGKey> >::iterator it(start_times.begin()); it != start_times.end(); ++it )
2219         {
2220                 time_t now = eDVBLocalTimeHandler::getInstance()->nowTime();
2221                 if ( (it->first.tm + duration_sec) < now )
2222                         continue;
2223                 memcpy(event+2, it->first.data, 5);
2224                 int bptr = ptr;
2225                 int cnt=0;
2226                 for (std::list<uniqueEPGKey>::iterator i(it->second.begin()); i != it->second.end(); ++i)
2227                 {
2228                         event[bptr++] = 0x4A;
2229                         __u8 *len = event+(bptr++);
2230                         event[bptr++] = (i->tsid & 0xFF00) >> 8;
2231                         event[bptr++] = (i->tsid & 0xFF);
2232                         event[bptr++] = (i->onid & 0xFF00) >> 8;
2233                         event[bptr++] = (i->onid & 0xFF);
2234                         event[bptr++] = (i->sid & 0xFF00) >> 8;
2235                         event[bptr++] = (i->sid & 0xFF);
2236                         event[bptr++] = 0xB0;
2237                         bptr += sprintf((char*)(event+bptr), "Option %d", ++cnt);
2238                         *len = ((event+bptr) - len)-1;
2239                 }
2240                 int llen = bptr - 12;
2241                 ev_struct->descriptors_loop_length_hi = (llen & 0xF00) >> 8;
2242                 ev_struct->descriptors_loop_length_lo = (llen & 0xFF);
2243
2244                 time_t stime = it->first.tm;
2245                 while( tmMap.find(stime) != tmMap.end() )
2246                         ++stime;
2247                 event[6] += (stime - it->first.tm);
2248                 __u16 event_id = 0;
2249                 while( evMap.find(event_id) != evMap.end() )
2250                         ++event_id;
2251                 event[0] = (event_id & 0xFF00) >> 8;
2252                 event[1] = (event_id & 0xFF);
2253                 time_event_map[it->first.tm]=std::pair<time_t, __u16>(stime, event_id);
2254                 eventData *d = new eventData( ev_struct, bptr, eEPGCache::SCHEDULE );
2255                 evMap[event_id] = d;
2256                 tmMap[stime] = d;
2257         }
2258 }
2259
2260 void eEPGCache::channel_data::startPrivateReader(int pid, int version)
2261 {
2262         eDVBSectionFilterMask mask;
2263         memset(&mask, 0, sizeof(mask));
2264         mask.pid = pid;
2265         mask.flags = eDVBSectionFilterMask::rfCRC;
2266         mask.data[0] = 0xA0;
2267         mask.mask[0] = 0xFF;
2268         eDebug("start privatefilter for pid %04x and version %d", pid, version);
2269         if (version != -1)
2270         {
2271                 mask.data[3] = version << 1;
2272                 mask.mask[3] = 0x3E;
2273                 mask.mode[3] = 0x3E;
2274         }
2275         seenPrivateSections.clear();
2276         m_PrivateReader->connectRead(slot(*this, &eEPGCache::channel_data::readPrivateData), m_PrivateConn);
2277         m_PrivateReader->start(mask);
2278 #ifdef NEED_DEMUX_WORKAROUND
2279         m_PrevVersion=version;
2280 #endif
2281 }
2282
2283 void eEPGCache::channel_data::readPrivateData( const __u8 *data)
2284 {
2285         if (!data)
2286                 eDebug("get Null pointer from section reader !!");
2287         else
2288         {
2289                 if ( seenPrivateSections.find( data[6] ) == seenPrivateSections.end() )
2290                 {
2291 #ifdef NEED_DEMUX_WORKAROUND
2292                         int version = data[5];
2293                         version = ((version & 0x3E) >> 1);
2294                         can_delete = 0;
2295                         if ( m_PrevVersion != version )
2296                         {
2297                                 cache->privateSectionRead(m_PrivateService, data);
2298                                 seenPrivateSections.insert(data[6]);
2299                         }
2300                         else
2301                                 eDebug("ignore");
2302 #else
2303                         can_delete = 0;
2304                         cache->privateSectionRead(m_PrivateService, data);
2305                         seenPrivateSections.insert(data[6]);
2306 #endif
2307                 }
2308                 if ( seenPrivateSections.size() == (unsigned int)(data[7] + 1) )
2309                 {
2310                         eDebug("[EPGC] private finished");
2311                         if (!isRunning)
2312                                 can_delete = 1;
2313                         int version = data[5];
2314                         version = ((version & 0x3E) >> 1);
2315                         startPrivateReader(m_PrivatePid, version);
2316                 }
2317         }
2318 }
2319
2320 #endif // ENABLE_PRIVATE_EPG