lib/dvb/epgcache.cpp

   1 #include <lib/dvb/epgcache.h>
   2 #include <lib/dvb/dvb.h>
   3
   4 #undef EPG_DEBUG
   5
   6 #ifdef EPG_DEBUG
   7 #include <lib/service/event.h>
   8 #endif
   9
  10 #include <time.h>
  11 #include <unistd.h>  // for usleep
  12 #include <sys/vfs.h> // for statfs
  13 // #include <libmd5sum.h>
  14 #include <lib/base/eerror.h>
  15 #include <lib/base/estring.h>
  16 #include <lib/dvb/pmt.h>
  17 #include <lib/dvb/db.h>
  18 #include <lib/python/python.h>
  19 #include <dvbsi++/descriptor_tag.h>
  20
  21 int eventData::CacheSize=0;
  22 descriptorMap eventData::descriptors;
  23 __u8 eventData::data[4108];
  24 extern const uint32_t crc32_table[256];
  25
  26 const eServiceReference &handleGroup(const eServiceReference &ref)
  27 {
  28         if (ref.flags & eServiceReference::isGroup)
  29         {
  30                 ePtr<eDVBResourceManager> res;
  31                 if (!eDVBResourceManager::getInstance(res))
  32                 {
  33                         ePtr<iDVBChannelList> db;
  34                         if (!res->getChannelList(db))
  35                         {
  36                                 eBouquet *bouquet=0;
  37                                 if (!db->getBouquet(ref, bouquet))
  38                                 {
  39                                         std::list<eServiceReference>::iterator it(bouquet->m_services.begin());
  40                                         if (it != bouquet->m_services.end())
  41                                                 return *it;
  42                                 }
  43                         }
  44                 }
  45         }
  46         return ref;
  47 }
  48
  49 eventData::eventData(const eit_event_struct* e, int size, int type)
  50         :ByteSize(size&0xFF), type(type&0xFF)
  51 {
  52         if (!e)
  53                 return;
  54
  55         __u32 descr[65];
  56         __u32 *pdescr=descr;
  57
  58         __u8 *data = (__u8*)e;
  59         int ptr=12;
  60         size -= 12;
  61
  62         while(size > 1)
  63         {
  64                 __u8 *descr = data+ptr;
  65                 int descr_len = descr[1];
  66                 descr_len += 2;
  67                 if (size >= descr_len)
  68                 {
  69                         switch (descr[0])
  70                         {
  71                                 case EXTENDED_EVENT_DESCRIPTOR:
  72                                 case SHORT_EVENT_DESCRIPTOR:
  73                                 case LINKAGE_DESCRIPTOR:
  74                                 case COMPONENT_DESCRIPTOR:
  75                                 {
  76                                         __u32 crc = 0;
  77                                         int cnt=0;
  78                                         while(cnt++ < descr_len)
  79                                                 crc = (crc << 8) ^ crc32_table[((crc >> 24) ^ data[ptr++]) & 0xFF];
  80
  81                                         descriptorMap::iterator it =
  82                                                 descriptors.find(crc);
  83                                         if ( it == descriptors.end() )
  84                                         {
  85                                                 CacheSize+=descr_len;
  86                                                 __u8 *d = new __u8[descr_len];
  87                                                 memcpy(d, descr, descr_len);
  88                                                 descriptors[crc] = descriptorPair(1, d);
  89                                         }
  90                                         else
  91                                                 ++it->second.first;
  92                                         *pdescr++=crc;
  93                                         break;
  94                                 }
  95                                 default: // do not cache all other descriptors
  96                                         ptr += descr_len;
  97                                         break;
  98                         }
  99                         size -= descr_len;
 100                 }
 101                 else
 102                         break;
 103         }
 104         ASSERT(pdescr <= &descr[65]);
 105         ByteSize = 10+((pdescr-descr)*4);
 106         EITdata = new __u8[ByteSize];
 107         CacheSize+=ByteSize;
 108         memcpy(EITdata, (__u8*) e, 10);
 109         memcpy(EITdata+10, descr, ByteSize-10);
 110 }
 111
 112 const eit_event_struct* eventData::get() const
 113 {
 114         int pos = 12;
 115         int tmp = ByteSize-10;
 116         memcpy(data, EITdata, 10);
 117         int descriptors_length=0;
 118         __u32 *p = (__u32*)(EITdata+10);
 119         while(tmp>3)
 120         {
 121                 descriptorMap::iterator it =
 122                         descriptors.find(*p++);
 123                 if ( it != descriptors.end() )
 124                 {
 125                         int b = it->second.second[1]+2;
 126                         memcpy(data+pos, it->second.second, b );
 127                         pos += b;
 128                         descriptors_length += b;
 129                 }
 130                 else
 131                         eFatal("LINE %d descriptor not found in descriptor cache %08x!!!!!!", __LINE__, *(p-1));
 132                 tmp-=4;
 133         }
 134         ASSERT(pos <= 4108);
 135         data[10] = (descriptors_length >> 8) & 0x0F;
 136         data[11] = descriptors_length & 0xFF;
 137         return (eit_event_struct*)data;
 138 }
 139
 140 eventData::~eventData()
 141 {
 142         if ( ByteSize )
 143         {
 144                 CacheSize -= ByteSize;
 145                 __u32 *d = (__u32*)(EITdata+10);
 146                 ByteSize -= 10;
 147                 while(ByteSize>3)
 148                 {
 149                         descriptorMap::iterator it =
 150                                 descriptors.find(*d++);
 151                         if ( it != descriptors.end() )
 152                         {
 153                                 descriptorPair &p = it->second;
 154                                 if (!--p.first) // no more used descriptor
 155                                 {
 156                                         CacheSize -= it->second.second[1];
 157                                         delete [] it->second.second;    // free descriptor memory
 158                                         descriptors.erase(it);  // remove entry from descriptor map
 159                                 }
 160                         }
 161                         else
 162                                 eFatal("LINE %d descriptor not found in descriptor cache %08x!!!!!!", __LINE__, *(d-1));
 163                         ByteSize -= 4;
 164                 }
 165                 delete [] EITdata;
 166         }
 167 }
 168
 169 void eventData::load(FILE *f)
 170 {
 171         int size=0;
 172         int id=0;
 173         __u8 header[2];
 174         descriptorPair p;
 175         fread(&size, sizeof(int), 1, f);
 176         while(size)
 177         {
 178                 fread(&id, sizeof(__u32), 1, f);
 179                 fread(&p.first, sizeof(int), 1, f);
 180                 fread(header, 2, 1, f);
 181                 int bytes = header[1]+2;
 182                 p.second = new __u8[bytes];
 183                 p.second[0] = header[0];
 184                 p.second[1] = header[1];
 185                 fread(p.second+2, bytes-2, 1, f);
 186                 descriptors[id]=p;
 187                 --size;
 188                 CacheSize+=bytes;
 189         }
 190 }
 191
 192 void eventData::save(FILE *f)
 193 {
 194         int size=descriptors.size();
 195         descriptorMap::iterator it(descriptors.begin());
 196         fwrite(&size, sizeof(int), 1, f);
 197         while(size)
 198         {
 199                 fwrite(&it->first, sizeof(__u32), 1, f);
 200                 fwrite(&it->second.first, sizeof(int), 1, f);
 201                 fwrite(it->second.second, it->second.second[1]+2, 1, f);
 202                 ++it;
 203                 --size;
 204         }
 205 }
 206
 207 eEPGCache* eEPGCache::instance;
 208 pthread_mutex_t eEPGCache::cache_lock=
 209         PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
 210 pthread_mutex_t eEPGCache::channel_map_lock=
 211         PTHREAD_RECURSIVE_MUTEX_INITIALIZER_NP;
 212
 213 DEFINE_REF(eEPGCache)
 214
 215 eEPGCache::eEPGCache()
 216         :messages(this,1), cleanTimer(this)//, paused(0)
 217 {
 218         eDebug("[EPGC] Initialized EPGCache");
 219
 220         CONNECT(messages.recv_msg, eEPGCache::gotMessage);
 221         CONNECT(eDVBLocalTimeHandler::getInstance()->m_timeUpdated, eEPGCache::timeUpdated);
 222         CONNECT(cleanTimer.timeout, eEPGCache::cleanLoop);
 223
 224         ePtr<eDVBResourceManager> res_mgr;
 225         eDVBResourceManager::getInstance(res_mgr);
 226         if (!res_mgr)
 227                 eDebug("[eEPGCache] no resource manager !!!!!!!");
 228         else
 229         {
 230                 res_mgr->connectChannelAdded(slot(*this,&eEPGCache::DVBChannelAdded), m_chanAddedConn);
 231                 if (eDVBLocalTimeHandler::getInstance()->ready())
 232                         timeUpdated();
 233         }
 234         instance=this;
 235 }
 236
 237 void eEPGCache::timeUpdated()
 238 {
 239         if (!sync())
 240         {
 241                 eDebug("[EPGC] time updated.. start EPG Mainloop");
 242                 run();
 243         } else
 244                 messages.send(Message(Message::timeChanged));
 245 }
 246
 247 void eEPGCache::DVBChannelAdded(eDVBChannel *chan)
 248 {
 249         if ( chan )
 250         {
 251 //              eDebug("[eEPGCache] add channel %p", chan);
 252                 channel_data *data = new channel_data(this);
 253                 data->channel = chan;
 254                 data->prevChannelState = -1;
 255 #ifdef ENABLE_PRIVATE_EPG
 256                 data->m_PrivatePid = -1;
 257 #endif
 258                 singleLock s(channel_map_lock);
 259                 m_knownChannels.insert( std::pair<iDVBChannel*, channel_data* >(chan, data) );
 260                 chan->connectStateChange(slot(*this, &eEPGCache::DVBChannelStateChanged), data->m_stateChangedConn);
 261         }
 262 }
 263
 264 void eEPGCache::DVBChannelRunning(iDVBChannel *chan)
 265 {
 266         channelMapIterator it =
 267                 m_knownChannels.find(chan);
 268         if ( it == m_knownChannels.end() )
 269                 eDebug("[eEPGCache] will start non existing channel %p !!!", chan);
 270         else
 271         {
 272                 channel_data &data = *it->second;
 273                 ePtr<eDVBResourceManager> res_mgr;
 274                 if ( eDVBResourceManager::getInstance( res_mgr ) )
 275                         eDebug("[eEPGCache] no res manager!!");
 276                 else
 277                 {
 278                         ePtr<iDVBDemux> demux;
 279                         if ( data.channel->getDemux(demux, 0) )
 280                         {
 281                                 eDebug("[eEPGCache] no demux!!");
 282                                 return;
 283                         }
 284                         else
 285                         {
 286                                 RESULT res = demux->createSectionReader( this, data.m_NowNextReader );
 287                                 if ( res )
 288                                 {
 289                                         eDebug("[eEPGCache] couldnt initialize nownext reader!!");
 290                                         return;
 291                                 }
 292
 293                                 res = demux->createSectionReader( this, data.m_ScheduleReader );
 294                                 if ( res )
 295                                 {
 296                                         eDebug("[eEPGCache] couldnt initialize schedule reader!!");
 297                                         return;
 298                                 }
 299
 300                                 res = demux->createSectionReader( this, data.m_ScheduleOtherReader );
 301                                 if ( res )
 302                                 {
 303                                         eDebug("[eEPGCache] couldnt initialize schedule other reader!!");
 304                                         return;
 305                                 }
 306 #ifdef ENABLE_PRIVATE_EPG
 307                                 res = demux->createSectionReader( this, data.m_PrivateReader );
 308                                 if ( res )
 309                                 {
 310                                         eDebug("[eEPGCache] couldnt initialize private reader!!");
 311                                         return;
 312                                 }
 313 #endif
 314 #ifdef ENABLE_MHW_EPG
 315                                 res = demux->createSectionReader( this, data.m_MHWReader );
 316                                 if ( res )
 317                                 {
 318                                         eDebug("[eEPGCache] couldnt initialize mhw reader!!");
 319                                         return;
 320                                 }
 321                                 res = demux->createSectionReader( this, data.m_MHWReader2 );
 322                                 if ( res )
 323                                 {
 324                                         eDebug("[eEPGCache] couldnt initialize mhw reader!!");
 325                                         return;
 326                                 }
 327 #endif
 328                                 messages.send(Message(Message::startChannel, chan));
 329                                 // -> gotMessage -> changedService
 330                         }
 331                 }
 332         }
 333 }
 334
 335 void eEPGCache::DVBChannelStateChanged(iDVBChannel *chan)
 336 {
 337         channelMapIterator it =
 338                 m_knownChannels.find(chan);
 339         if ( it != m_knownChannels.end() )
 340         {
 341                 int state=0;
 342                 chan->getState(state);
 343                 if ( it->second->prevChannelState != state )
 344                 {
 345                         switch (state)
 346                         {
 347                                 case iDVBChannel::state_ok:
 348                                 {
 349                                         eDebug("[eEPGCache] channel %p running", chan);
 350                                         DVBChannelRunning(chan);
 351                                         break;
 352                                 }
 353                                 case iDVBChannel::state_release:
 354                                 {
 355                                         eDebug("[eEPGCache] remove channel %p", chan);
 356                                         messages.send(Message(Message::leaveChannel, chan));
 357                                         pthread_mutex_lock(&it->second->channel_active);
 358                                         singleLock s(channel_map_lock);
 359                                         m_knownChannels.erase(it);
 360                                         pthread_mutex_unlock(&it->second->channel_active);
 361                                         delete it->second;
 362                                         it->second=0;
 363                                         // -> gotMessage -> abortEPG
 364                                         break;
 365                                 }
 366                                 default: // ignore all other events
 367                                         return;
 368                         }
 369                         if (it->second)
 370                                 it->second->prevChannelState = state;
 371                 }
 372         }
 373 }
 374
 375 void eEPGCache::FixOverlapping(std::pair<eventMap,timeMap> &servicemap, time_t TM, int duration, const timeMap::iterator &tm_it, const uniqueEPGKey &service)
 376 {
 377         timeMap::iterator tmp = tm_it;
 378         while ((tmp->first+tmp->second->getDuration()-300) > TM)
 379         {
 380                 if(tmp->first != TM
 381 #ifdef ENABLE_PRIVATE_EPG
 382                         && tmp->second->type != PRIVATE
 383 #endif
 384 #ifdef ENABLE_MHW
 385                         && tmp->second->type != MHW
 386 #endif
 387                         )
 388                 {
 389                         __u16 event_id = tmp->second->getEventID();
 390                         servicemap.first.erase(event_id);
 391 #ifdef EPG_DEBUG
 392                         Event evt((uint8_t*)tmp->second->get());
 393                         eServiceEvent event;
 394                         event.parseFrom(&evt, service.sid<<16|service.onid);
 395                         eDebug("(1)erase no more used event %04x %d\n%s %s\n%s",
 396                                 service.sid, event_id,
 397                                 event.getBeginTimeString().c_str(),
 398                                 event.getEventName().c_str(),
 399                                 event.getExtendedDescription().c_str());
 400 #endif
 401                         delete tmp->second;
 402                         if (tmp == servicemap.second.begin())
 403                         {
 404                                 servicemap.second.erase(tmp);
 405                                 break;
 406                         }
 407                         else
 408                                 servicemap.second.erase(tmp--);
 409                 }
 410                 else
 411                 {
 412                         if (tmp == servicemap.second.begin())
 413                                 break;
 414                         --tmp;
 415                 }
 416         }
 417
 418         tmp = tm_it;
 419         while(tmp->first < (TM+duration-300))
 420         {
 421                 if (tmp->first != TM && tmp->second->type != PRIVATE)
 422                 {
 423                         __u16 event_id = tmp->second->getEventID();
 424                         servicemap.first.erase(event_id);
 425 #ifdef EPG_DEBUG
 426                         Event evt((uint8_t*)tmp->second->get());
 427                         eServiceEvent event;
 428                         event.parseFrom(&evt, service.sid<<16|service.onid);
 429                         eDebug("(2)erase no more used event %04x %d\n%s %s\n%s",
 430                                 service.sid, event_id,
 431                                 event.getBeginTimeString().c_str(),
 432                                 event.getEventName().c_str(),
 433                                 event.getExtendedDescription().c_str());
 434 #endif
 435                         delete tmp->second;
 436                         servicemap.second.erase(tmp++);
 437                 }
 438                 else
 439                         ++tmp;
 440                 if (tmp == servicemap.second.end())
 441                         break;
 442         }
 443 }
 444
 445 void eEPGCache::sectionRead(const __u8 *data, int source, channel_data *channel)
 446 {
 447         eit_t *eit = (eit_t*) data;
 448
 449         int len=HILO(eit->section_length)-1;//+3-4;
 450         int ptr=EIT_SIZE;
 451         if ( ptr >= len )
 452                 return;
 453
 454         // This fixed the EPG on the Multichoice irdeto systems
 455         // the EIT packet is non-compliant.. their EIT packet stinks
 456         if ( data[ptr-1] < 0x40 )
 457                 --ptr;
 458
 459         uniqueEPGKey service( HILO(eit->service_id), HILO(eit->original_network_id), HILO(eit->transport_stream_id) );
 460         eit_event_struct* eit_event = (eit_event_struct*) (data+ptr);
 461         int eit_event_size;
 462         int duration;
 463
 464         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);
 465         time_t now = eDVBLocalTimeHandler::getInstance()->nowTime();
 466
 467         if ( TM != 3599 && TM > -1)
 468                 channel->haveData |= source;
 469
 470         singleLock s(cache_lock);
 471         // hier wird immer eine eventMap zurück gegeben.. entweder eine vorhandene..
 472         // oder eine durch [] erzeugte
 473         std::pair<eventMap,timeMap> &servicemap = eventDB[service];
 474         eventMap::iterator prevEventIt = servicemap.first.end();
 475         timeMap::iterator prevTimeIt = servicemap.second.end();
 476
 477         while (ptr<len)
 478         {
 479                 eit_event_size = HILO(eit_event->descriptors_loop_length)+EIT_LOOP_SIZE;
 480
 481                 duration = fromBCD(eit_event->duration_1)*3600+fromBCD(eit_event->duration_2)*60+fromBCD(eit_event->duration_3);
 482                 TM = parseDVBtime(
 483                         eit_event->start_time_1,
 484                         eit_event->start_time_2,
 485                         eit_event->start_time_3,
 486                         eit_event->start_time_4,
 487                         eit_event->start_time_5);
 488
 489                 if ( TM == 3599 )
 490                         goto next;
 491
 492                 if ( TM != 3599 && (TM+duration < now || TM > now+14*24*60*60) )
 493                         goto next;
 494
 495                 if ( now <= (TM+duration) || TM == 3599 /*NVOD Service*/ )  // old events should not be cached
 496                 {
 497                         __u16 event_id = HILO(eit_event->event_id);
 498 //                      eDebug("event_id is %d sid is %04x", event_id, service.sid);
 499
 500                         eventData *evt = 0;
 501                         int ev_erase_count = 0;
 502                         int tm_erase_count = 0;
 503
 504                         // search in eventmap
 505                         eventMap::iterator ev_it =
 506                                 servicemap.first.find(event_id);
 507
 508                         // entry with this event_id is already exist ?
 509                         if ( ev_it != servicemap.first.end() )
 510                         {
 511                                 if ( source > ev_it->second->type )  // update needed ?
 512                                         goto next; // when not.. then skip this entry
 513
 514                                 // search this event in timemap
 515                                 timeMap::iterator tm_it_tmp =
 516                                         servicemap.second.find(ev_it->second->getStartTime());
 517
 518                                 if ( tm_it_tmp != servicemap.second.end() )
 519                                 {
 520                                         if ( tm_it_tmp->first == TM ) // just update eventdata
 521                                         {
 522                                                 // exempt memory
 523                                                 eventData *tmp = ev_it->second;
 524                                                 ev_it->second = tm_it_tmp->second =
 525                                                         new eventData(eit_event, eit_event_size, source);
 526                                                 FixOverlapping(servicemap, TM, duration, tm_it_tmp, service);
 527                                                 delete tmp;
 528                                                 goto next;
 529                                         }
 530                                         else  // event has new event begin time
 531                                         {
 532                                                 tm_erase_count++;
 533                                                 // delete the found record from timemap
 534                                                 servicemap.second.erase(tm_it_tmp);
 535                                                 prevTimeIt=servicemap.second.end();
 536                                         }
 537                                 }
 538                         }
 539
 540                         // search in timemap, for check of a case if new time has coincided with time of other event
 541                         // or event was is not found in eventmap
 542                         timeMap::iterator tm_it =
 543                                 servicemap.second.find(TM);
 544
 545                         if ( tm_it != servicemap.second.end() )
 546                         {
 547                                 // event with same start time but another event_id...
 548                                 if ( source > tm_it->second->type &&
 549                                         ev_it == servicemap.first.end() )
 550                                         goto next; // when not.. then skip this entry
 551
 552                                 // search this time in eventmap
 553                                 eventMap::iterator ev_it_tmp =
 554                                         servicemap.first.find(tm_it->second->getEventID());
 555
 556                                 if ( ev_it_tmp != servicemap.first.end() )
 557                                 {
 558                                         ev_erase_count++;
 559                                         // delete the found record from eventmap
 560                                         servicemap.first.erase(ev_it_tmp);
 561                                         prevEventIt=servicemap.first.end();
 562                                 }
 563                         }
 564                         evt = new eventData(eit_event, eit_event_size, source);
 565 #ifdef EPG_DEBUG
 566                         bool consistencyCheck=true;
 567 #endif
 568                         if (ev_erase_count > 0 && tm_erase_count > 0) // 2 different pairs have been removed
 569                         {
 570                                 // exempt memory
 571                                 delete ev_it->second;
 572                                 delete tm_it->second;
 573                                 ev_it->second=evt;
 574                                 tm_it->second=evt;
 575                         }
 576                         else if (ev_erase_count == 0 && tm_erase_count > 0)
 577                         {
 578                                 // exempt memory
 579                                 delete ev_it->second;
 580                                 tm_it=prevTimeIt=servicemap.second.insert( prevTimeIt, std::pair<const time_t, eventData*>( TM, evt ) );
 581                                 ev_it->second=evt;
 582                         }
 583                         else if (ev_erase_count > 0 && tm_erase_count == 0)
 584                         {
 585                                 // exempt memory
 586                                 delete tm_it->second;
 587                                 ev_it=prevEventIt=servicemap.first.insert( prevEventIt, std::pair<const __u16, eventData*>( event_id, evt) );
 588                                 tm_it->second=evt;
 589                         }
 590                         else // added new eventData
 591                         {
 592 #ifdef EPG_DEBUG
 593                                 consistencyCheck=false;
 594 #endif
 595                                 ev_it=prevEventIt=servicemap.first.insert( prevEventIt, std::pair<const __u16, eventData*>( event_id, evt) );
 596                                 tm_it=prevTimeIt=servicemap.second.insert( prevTimeIt, std::pair<const time_t, eventData*>( TM, evt ) );
 597                         }
 598
 599                         FixOverlapping(servicemap, TM, duration, tm_it, service);
 600
 601 #ifdef EPG_DEBUG
 602                         if ( consistencyCheck )
 603                         {
 604                                 if ( tm_it->second != evt || ev_it->second != evt )
 605                                         eFatal("tm_it->second != ev_it->second");
 606                                 else if ( tm_it->second->getStartTime() != tm_it->first )
 607                                         eFatal("event start_time(%d) non equal timemap key(%d)",
 608                                                 tm_it->second->getStartTime(), tm_it->first );
 609                                 else if ( tm_it->first != TM )
 610                                         eFatal("timemap key(%d) non equal TM(%d)",
 611                                                 tm_it->first, TM);
 612                                 else if ( ev_it->second->getEventID() != ev_it->first )
 613                                         eFatal("event_id (%d) non equal event_map key(%d)",
 614                                                 ev_it->second->getEventID(), ev_it->first);
 615                                 else if ( ev_it->first != event_id )
 616                                         eFatal("eventmap key(%d) non equal event_id(%d)",
 617                                                 ev_it->first, event_id );
 618                         }
 619 #endif
 620                 }
 621 next:
 622 #ifdef EPG_DEBUG
 623                 if ( servicemap.first.size() != servicemap.second.size() )
 624                 {
 625                         FILE *f = fopen("/hdd/event_map.txt", "w+");
 626                         int i=0;
 627                         for (eventMap::iterator it(servicemap.first.begin())
 628                                 ; it != servicemap.first.end(); ++it )
 629                                 fprintf(f, "%d(key %d) -> time %d, event_id %d, data %p\n",
 630                                         i++, (int)it->first, (int)it->second->getStartTime(), (int)it->second->getEventID(), it->second );
 631                         fclose(f);
 632                         f = fopen("/hdd/time_map.txt", "w+");
 633                         i=0;
 634                         for (timeMap::iterator it(servicemap.second.begin())
 635                                 ; it != servicemap.second.end(); ++it )
 636                                         fprintf(f, "%d(key %d) -> time %d, event_id %d, data %p\n",
 637                                                 i++, (int)it->first, (int)it->second->getStartTime(), (int)it->second->getEventID(), it->second );
 638                         fclose(f);
 639
 640                         eFatal("(1)map sizes not equal :( sid %04x tsid %04x onid %04x size %d size2 %d",
 641                                 service.sid, service.tsid, service.onid,
 642                                 servicemap.first.size(), servicemap.second.size() );
 643                 }
 644 #endif
 645                 ptr += eit_event_size;
 646                 eit_event=(eit_event_struct*)(((__u8*)eit_event)+eit_event_size);
 647         }
 648 }
 649
 650 void eEPGCache::flushEPG(const uniqueEPGKey & s)
 651 {
 652         eDebug("[EPGC] flushEPG %d", (int)(bool)s);
 653         singleLock l(cache_lock);
 654         if (s)  // clear only this service
 655         {
 656                 eventCache::iterator it = eventDB.find(s);
 657                 if ( it != eventDB.end() )
 658                 {
 659                         eventMap &evMap = it->second.first;
 660                         timeMap &tmMap = it->second.second;
 661                         tmMap.clear();
 662                         for (eventMap::iterator i = evMap.begin(); i != evMap.end(); ++i)
 663                                 delete i->second;
 664                         evMap.clear();
 665                         eventDB.erase(it);
 666
 667                         // TODO .. search corresponding channel for removed service and remove this channel from lastupdated map
 668 #ifdef ENABLE_PRIVATE_EPG
 669                         contentMaps::iterator it =
 670                                 content_time_tables.find(s);
 671                         if ( it != content_time_tables.end() )
 672                         {
 673                                 it->second.clear();
 674                                 content_time_tables.erase(it);
 675                         }
 676 #endif
 677                 }
 678         }
 679         else // clear complete EPG Cache
 680         {
 681                 for (eventCache::iterator it(eventDB.begin());
 682                         it != eventDB.end(); ++it)
 683                 {
 684                         eventMap &evMap = it->second.first;
 685                         timeMap &tmMap = it->second.second;
 686                         for (eventMap::iterator i = evMap.begin(); i != evMap.end(); ++i)
 687                                 delete i->second;
 688                         evMap.clear();
 689                         tmMap.clear();
 690                 }
 691                 eventDB.clear();
 692 #ifdef ENABLE_PRIVATE_EPG
 693                 content_time_tables.clear();
 694 #endif
 695                 channelLastUpdated.clear();
 696                 singleLock m(channel_map_lock);
 697                 for (channelMapIterator it(m_knownChannels.begin()); it != m_knownChannels.end(); ++it)
 698                         it->second->startEPG();
 699         }
 700         eDebug("[EPGC] %i bytes for cache used", eventData::CacheSize);
 701 }
 702
 703 void eEPGCache::cleanLoop()
 704 {
 705         singleLock s(cache_lock);
 706         if (!eventDB.empty())
 707         {
 708                 eDebug("[EPGC] start cleanloop");
 709
 710                 time_t now = eDVBLocalTimeHandler::getInstance()->nowTime();
 711
 712                 for (eventCache::iterator DBIt = eventDB.begin(); DBIt != eventDB.end(); DBIt++)
 713                 {
 714                         bool updated = false;
 715                         for (timeMap::iterator It = DBIt->second.second.begin(); It != DBIt->second.second.end() && It->first < now;)
 716                         {
 717                                 if ( now > (It->first+It->second->getDuration()) )  // outdated normal entry (nvod references to)
 718                                 {
 719                                         // remove entry from eventMap
 720                                         eventMap::iterator b(DBIt->second.first.find(It->second->getEventID()));
 721                                         if ( b != DBIt->second.first.end() )
 722                                         {
 723                                                 // release Heap Memory for this entry   (new ....)
 724 //                                              eDebug("[EPGC] delete old event (evmap)");
 725                                                 DBIt->second.first.erase(b);
 726                                         }
 727
 728                                         // remove entry from timeMap
 729 //                                      eDebug("[EPGC] release heap mem");
 730                                         delete It->second;
 731                                         DBIt->second.second.erase(It++);
 732 //                                      eDebug("[EPGC] delete old event (timeMap)");
 733                                         updated = true;
 734                                 }
 735                                 else
 736                                         ++It;
 737                         }
 738 #ifdef ENABLE_PRIVATE_EPG
 739                         if ( updated )
 740                         {
 741                                 contentMaps::iterator x =
 742                                         content_time_tables.find( DBIt->first );
 743                                 if ( x != content_time_tables.end() )
 744                                 {
 745                                         timeMap &tmMap = DBIt->second.second;
 746                                         for ( contentMap::iterator i = x->second.begin(); i != x->second.end(); )
 747                                         {
 748                                                 for ( contentTimeMap::iterator it(i->second.begin());
 749                                                         it != i->second.end(); )
 750                                                 {
 751                                                         if ( tmMap.find(it->second.first) == tmMap.end() )
 752                                                                 i->second.erase(it++);
 753                                                         else
 754                                                                 ++it;
 755                                                 }
 756                                                 if ( i->second.size() )
 757                                                         ++i;
 758                                                 else
 759                                                         x->second.erase(i++);
 760                                         }
 761                                 }
 762                         }
 763 #endif
 764                 }
 765                 eDebug("[EPGC] stop cleanloop");
 766                 eDebug("[EPGC] %i bytes for cache used", eventData::CacheSize);
 767         }
 768         cleanTimer.start(CLEAN_INTERVAL,true);
 769 }
 770
 771 eEPGCache::~eEPGCache()
 772 {
 773         messages.send(Message::quit);
 774         kill(); // waiting for thread shutdown
 775         singleLock s(cache_lock);
 776         for (eventCache::iterator evIt = eventDB.begin(); evIt != eventDB.end(); evIt++)
 777                 for (eventMap::iterator It = evIt->second.first.begin(); It != evIt->second.first.end(); It++)
 778                         delete It->second;
 779 }
 780
 781 void eEPGCache::gotMessage( const Message &msg )
 782 {
 783         switch (msg.type)
 784         {
 785                 case Message::flush:
 786                         flushEPG(msg.service);
 787                         break;
 788                 case Message::startChannel:
 789                 {
 790                         singleLock s(channel_map_lock);
 791                         channelMapIterator channel =
 792                                 m_knownChannels.find(msg.channel);
 793                         if ( channel != m_knownChannels.end() )
 794                                 channel->second->startChannel();
 795                         break;
 796                 }
 797                 case Message::leaveChannel:
 798                 {
 799                         singleLock s(channel_map_lock);
 800                         channelMapIterator channel =
 801                                 m_knownChannels.find(msg.channel);
 802                         if ( channel != m_knownChannels.end() )
 803                                 channel->second->abortEPG();
 804                         break;
 805                 }
 806                 case Message::quit:
 807                         quit(0);
 808                         break;
 809 #ifdef ENABLE_PRIVATE_EPG
 810                 case Message::got_private_pid:
 811                 {
 812                         singleLock s(channel_map_lock);
 813                         for (channelMapIterator it(m_knownChannels.begin()); it != m_knownChannels.end(); ++it)
 814                         {
 815                                 eDVBChannel *channel = (eDVBChannel*) it->first;
 816                                 channel_data *data = it->second;
 817                                 eDVBChannelID chid = channel->getChannelID();
 818                                 if ( chid.transport_stream_id.get() == msg.service.tsid &&
 819                                         chid.original_network_id.get() == msg.service.onid &&
 820                                         data->m_PrivatePid == -1 )
 821                                 {
 822                                         data->m_PrevVersion = -1;
 823                                         data->m_PrivatePid = msg.pid;
 824                                         data->m_PrivateService = msg.service;
 825                                         int onid = chid.original_network_id.get();
 826                                         onid |= 0x80000000;  // we use highest bit as private epg indicator
 827                                         chid.original_network_id = onid;
 828                                         updateMap::iterator It = channelLastUpdated.find( chid );
 829                                         int update = ( It != channelLastUpdated.end() ? ( UPDATE_INTERVAL - ( (eDVBLocalTimeHandler::getInstance()->nowTime()-It->second) * 1000 ) ) : ZAP_DELAY );
 830                                         if (update < ZAP_DELAY)
 831                                                 update = ZAP_DELAY;
 832                                         data->startPrivateTimer.start(update, 1);
 833                                         if (update >= 60000)
 834                                                 eDebug("[EPGC] next private update in %i min", update/60000);
 835                                         else if (update >= 1000)
 836                                                 eDebug("[EPGC] next private update in %i sec", update/1000);
 837                                         break;
 838                                 }
 839                         }
 840                         break;
 841                 }
 842 #endif
 843                 case Message::timeChanged:
 844                         cleanLoop();
 845                         break;
 846                 default:
 847                         eDebug("unhandled EPGCache Message!!");
 848                         break;
 849         }
 850 }
 851
 852 void eEPGCache::thread()
 853 {
 854         hasStarted();
 855         nice(4);
 856         load();
 857         cleanLoop();
 858         runLoop();
 859         save();
 860 }
 861
 862 void eEPGCache::load()
 863 {
 864         FILE *f = fopen("/hdd/epg.dat", "r");
 865         if (f)
 866         {
 867                 unlink("/hdd/epg.dat");
 868                 int size=0;
 869                 int cnt=0;
 870 #if 0
 871                 unsigned char md5_saved[16];
 872                 unsigned char md5[16];
 873                 bool md5ok=false;
 874
 875                 if (!md5_file("/hdd/epg.dat", 1, md5))
 876                 {
 877                         FILE *f = fopen("/hdd/epg.dat.md5", "r");
 878                         if (f)
 879                         {
 880                                 fread( md5_saved, 16, 1, f);
 881                                 fclose(f);
 882                                 if ( !memcmp(md5_saved, md5, 16) )
 883                                         md5ok=true;
 884                         }
 885                 }
 886                 if ( md5ok )
 887 #endif
 888                 {
 889                         unsigned int magic=0;
 890                         fread( &magic, sizeof(int), 1, f);
 891                         if (magic != 0x98765432)
 892                         {
 893                                 eDebug("[EPGC] epg file has incorrect byte order.. dont read it");
 894                                 fclose(f);
 895                                 return;
 896                         }
 897                         char text1[13];
 898                         fread( text1, 13, 1, f);
 899                         if ( !strncmp( text1, "ENIGMA_EPG_V7", 13) )
 900                         {
 901                                 singleLock s(cache_lock);
 902                                 fread( &size, sizeof(int), 1, f);
 903                                 while(size--)
 904                                 {
 905                                         uniqueEPGKey key;
 906                                         eventMap evMap;
 907                                         timeMap tmMap;
 908                                         int size=0;
 909                                         fread( &key, sizeof(uniqueEPGKey), 1, f);
 910                                         fread( &size, sizeof(int), 1, f);
 911                                         while(size--)
 912                                         {
 913                                                 __u8 len=0;
 914                                                 __u8 type=0;
 915                                                 eventData *event=0;
 916                                                 fread( &type, sizeof(__u8), 1, f);
 917                                                 fread( &len, sizeof(__u8), 1, f);
 918                                                 event = new eventData(0, len, type);
 919                                                 event->EITdata = new __u8[len];
 920                                                 eventData::CacheSize+=len;
 921                                                 fread( event->EITdata, len, 1, f);
 922                                                 evMap[ event->getEventID() ]=event;
 923                                                 tmMap[ event->getStartTime() ]=event;
 924                                                 ++cnt;
 925                                         }
 926                                         eventDB[key]=std::pair<eventMap,timeMap>(evMap,tmMap);
 927                                 }
 928                                 eventData::load(f);
 929                                 eDebug("[EPGC] %d events read from /hdd/epg.dat", cnt);
 930 #ifdef ENABLE_PRIVATE_EPG
 931                                 char text2[11];
 932                                 fread( text2, 11, 1, f);
 933                                 if ( !strncmp( text2, "PRIVATE_EPG", 11) )
 934                                 {
 935                                         size=0;
 936                                         fread( &size, sizeof(int), 1, f);
 937                                         while(size--)
 938                                         {
 939                                                 int size=0;
 940                                                 uniqueEPGKey key;
 941                                                 fread( &key, sizeof(uniqueEPGKey), 1, f);
 942                                                 eventMap &evMap=eventDB[key].first;
 943                                                 fread( &size, sizeof(int), 1, f);
 944                                                 while(size--)
 945                                                 {
 946                                                         int size;
 947                                                         int content_id;
 948                                                         fread( &content_id, sizeof(int), 1, f);
 949                                                         fread( &size, sizeof(int), 1, f);
 950                                                         while(size--)
 951                                                         {
 952                                                                 time_t time1, time2;
 953                                                                 __u16 event_id;
 954                                                                 fread( &time1, sizeof(time_t), 1, f);
 955                                                                 fread( &time2, sizeof(time_t), 1, f);
 956                                                                 fread( &event_id, sizeof(__u16), 1, f);
 957                                                                 content_time_tables[key][content_id][time1]=std::pair<time_t, __u16>(time2, event_id);
 958                                                                 eventMap::iterator it =
 959                                                                         evMap.find(event_id);
 960                                                                 if (it != evMap.end())
 961                                                                         it->second->type = PRIVATE;
 962                                                         }
 963                                                 }
 964                                         }
 965                                 }
 966 #endif // ENABLE_PRIVATE_EPG
 967                         }
 968                         else
 969                                 eDebug("[EPGC] don't read old epg database");
 970                         fclose(f);
 971                 }
 972         }
 973 }
 974
 975 void eEPGCache::save()
 976 {
 977         struct statfs s;
 978         off64_t tmp;
 979         if (statfs("/hdd", &s)<0)
 980                 tmp=0;
 981         else
 982         {
 983                 tmp=s.f_blocks;
 984                 tmp*=s.f_bsize;
 985         }
 986
 987         // prevent writes to builtin flash
 988         if ( tmp < 1024*1024*50 ) // storage size < 50MB
 989                 return;
 990
 991         // check for enough free space on storage
 992         tmp=s.f_bfree;
 993         tmp*=s.f_bsize;
 994         if ( tmp < (eventData::CacheSize*12)/10 ) // 20% overhead
 995                 return;
 996
 997         FILE *f = fopen("/hdd/epg.dat", "w");
 998         int cnt=0;
 999         if ( f )
1000         {
1001                 unsigned int magic = 0x98765432;
1002                 fwrite( &magic, sizeof(int), 1, f);
1003                 const char *text = "ENIGMA_EPG_V7";
1004                 fwrite( text, 13, 1, f );
1005                 int size = eventDB.size();
1006                 fwrite( &size, sizeof(int), 1, f );
1007                 for (eventCache::iterator service_it(eventDB.begin()); service_it != eventDB.end(); ++service_it)
1008                 {
1009                         timeMap &timemap = service_it->second.second;
1010                         fwrite( &service_it->first, sizeof(uniqueEPGKey), 1, f);
1011                         size = timemap.size();
1012                         fwrite( &size, sizeof(int), 1, f);
1013                         for (timeMap::iterator time_it(timemap.begin()); time_it != timemap.end(); ++time_it)
1014                         {
1015                                 __u8 len = time_it->second->ByteSize;
1016                                 fwrite( &time_it->second->type, sizeof(__u8), 1, f );
1017                                 fwrite( &len, sizeof(__u8), 1, f);
1018                                 fwrite( time_it->second->EITdata, len, 1, f);
1019                                 ++cnt;
1020                         }
1021                 }
1022                 eDebug("[EPGC] %d events written to /hdd/epg.dat", cnt);
1023                 eventData::save(f);
1024 #ifdef ENABLE_PRIVATE_EPG
1025                 const char* text3 = "PRIVATE_EPG";
1026                 fwrite( text3, 11, 1, f );
1027                 size = content_time_tables.size();
1028                 fwrite( &size, sizeof(int), 1, f);
1029                 for (contentMaps::iterator a = content_time_tables.begin(); a != content_time_tables.end(); ++a)
1030                 {
1031                         contentMap &content_time_table = a->second;
1032                         fwrite( &a->first, sizeof(uniqueEPGKey), 1, f);
1033                         int size = content_time_table.size();
1034                         fwrite( &size, sizeof(int), 1, f);
1035                         for (contentMap::iterator i = content_time_table.begin(); i != content_time_table.end(); ++i )
1036                         {
1037                                 int size = i->second.size();
1038                                 fwrite( &i->first, sizeof(int), 1, f);
1039                                 fwrite( &size, sizeof(int), 1, f);
1040                                 for ( contentTimeMap::iterator it(i->second.begin());
1041                                         it != i->second.end(); ++it )
1042                                 {
1043                                         fwrite( &it->first, sizeof(time_t), 1, f);
1044                                         fwrite( &it->second.first, sizeof(time_t), 1, f);
1045                                         fwrite( &it->second.second, sizeof(__u16), 1, f);
1046                                 }
1047                         }
1048                 }
1049 #endif
1050                 fclose(f);
1051 #if 0
1052                 unsigned char md5[16];
1053                 if (!md5_file("/hdd/epg.dat", 1, md5))
1054                 {
1055                         FILE *f = fopen("/hdd/epg.dat.md5", "w");
1056                         if (f)
1057                         {
1058                                 fwrite( md5, 16, 1, f);
1059                                 fclose(f);
1060                         }
1061                 }
1062 #endif
1063         }
1064 }
1065
1066 eEPGCache::channel_data::channel_data(eEPGCache *ml)
1067         :cache(ml)
1068         ,abortTimer(ml), zapTimer(ml), state(0)
1069         ,isRunning(0), haveData(0)
1070 #ifdef ENABLE_PRIVATE_EPG
1071         ,startPrivateTimer(ml)
1072 #endif
1073 #ifdef ENABLE_MHW_EPG
1074         ,m_MHWTimeoutTimer(ml)
1075 #endif
1076 {
1077 #ifdef ENABLE_MHW_EPG
1078         CONNECT(m_MHWTimeoutTimer.timeout, eEPGCache::channel_data::MHWTimeout);
1079 #endif
1080         CONNECT(zapTimer.timeout, eEPGCache::channel_data::startEPG);
1081         CONNECT(abortTimer.timeout, eEPGCache::channel_data::abortNonAvail);
1082 #ifdef ENABLE_PRIVATE_EPG
1083         CONNECT(startPrivateTimer.timeout, eEPGCache::channel_data::startPrivateReader);
1084 #endif
1085         pthread_mutex_init(&channel_active, 0);
1086 }
1087
1088 bool eEPGCache::channel_data::finishEPG()
1089 {
1090         if (!isRunning)  // epg ready
1091         {
1092                 eDebug("[EPGC] stop caching events(%ld)", eDVBLocalTimeHandler::getInstance()->nowTime());
1093                 zapTimer.start(UPDATE_INTERVAL, 1);
1094                 eDebug("[EPGC] next update in %i min", UPDATE_INTERVAL / 60000);
1095                 for (int i=0; i < 3; ++i)
1096                 {
1097                         seenSections[i].clear();
1098                         calcedSections[i].clear();
1099                 }
1100                 singleLock l(cache->cache_lock);
1101                 cache->channelLastUpdated[channel->getChannelID()] = eDVBLocalTimeHandler::getInstance()->nowTime();
1102 #ifdef ENABLE_MHW_EPG
1103                 cleanup();
1104 #endif
1105                 return true;
1106         }
1107         return false;
1108 }
1109
1110 void eEPGCache::channel_data::startEPG()
1111 {
1112         eDebug("[EPGC] start caching events(%ld)", eDVBLocalTimeHandler::getInstance()->nowTime());
1113         state=0;
1114         haveData=0;
1115         for (int i=0; i < 3; ++i)
1116         {
1117                 seenSections[i].clear();
1118                 calcedSections[i].clear();
1119         }
1120
1121         eDVBSectionFilterMask mask;
1122         memset(&mask, 0, sizeof(mask));
1123
1124 #ifdef ENABLE_MHW_EPG
1125         mask.pid = 0xD3;
1126         mask.data[0] = 0x91;
1127         mask.mask[0] = 0xFF;
1128         m_MHWReader->connectRead(slot(*this, &eEPGCache::channel_data::readMHWData), m_MHWConn);
1129         m_MHWReader->start(mask);
1130         isRunning |= MHW;
1131         memcpy(&m_MHWFilterMask, &mask, sizeof(eDVBSectionFilterMask));
1132
1133         mask.pid = 0x231;
1134         mask.data[0] = 0xC8;
1135         mask.mask[0] = 0xFF;
1136         mask.data[1] = 0;
1137         mask.mask[1] = 0xFF;
1138         m_MHWReader2->connectRead(slot(*this, &eEPGCache::channel_data::readMHWData2), m_MHWConn2);
1139         m_MHWReader2->start(mask);
1140         isRunning |= MHW;
1141         memcpy(&m_MHWFilterMask2, &mask, sizeof(eDVBSectionFilterMask));
1142         mask.data[1] = 0;
1143         mask.mask[1] = 0;
1144 #endif
1145
1146         mask.pid = 0x12;
1147         mask.flags = eDVBSectionFilterMask::rfCRC;
1148
1149         mask.data[0] = 0x4E;
1150         mask.mask[0] = 0xFE;
1151         m_NowNextReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_NowNextConn);
1152         m_NowNextReader->start(mask);
1153         isRunning |= NOWNEXT;
1154
1155         mask.data[0] = 0x50;
1156         mask.mask[0] = 0xF0;
1157         m_ScheduleReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_ScheduleConn);
1158         m_ScheduleReader->start(mask);
1159         isRunning |= SCHEDULE;
1160
1161         mask.data[0] = 0x60;
1162         m_ScheduleOtherReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_ScheduleOtherConn);
1163         m_ScheduleOtherReader->start(mask);
1164         isRunning |= SCHEDULE_OTHER;
1165
1166         abortTimer.start(7000,true);
1167 }
1168
1169 void eEPGCache::channel_data::abortNonAvail()
1170 {
1171         if (!state)
1172         {
1173                 if ( !(haveData&NOWNEXT) && (isRunning&NOWNEXT) )
1174                 {
1175                         eDebug("[EPGC] abort non avail nownext reading");
1176                         isRunning &= ~NOWNEXT;
1177                         m_NowNextReader->stop();
1178                         m_NowNextConn=0;
1179                 }
1180                 if ( !(haveData&SCHEDULE) && (isRunning&SCHEDULE) )
1181                 {
1182                         eDebug("[EPGC] abort non avail schedule reading");
1183                         isRunning &= ~SCHEDULE;
1184                         m_ScheduleReader->stop();
1185                         m_ScheduleConn=0;
1186                 }
1187                 if ( !(haveData&SCHEDULE_OTHER) && (isRunning&SCHEDULE_OTHER) )
1188                 {
1189                         eDebug("[EPGC] abort non avail schedule_other reading");
1190                         isRunning &= ~SCHEDULE_OTHER;
1191                         m_ScheduleOtherReader->stop();
1192                         m_ScheduleOtherConn=0;
1193                 }
1194 #ifdef ENABLE_MHW_EPG
1195                 if ( !(haveData&MHW) && (isRunning&MHW) )
1196                 {
1197                         eDebug("[EPGC] abort non avail mhw reading");
1198                         isRunning &= ~MHW;
1199                         m_MHWReader->stop();
1200                         m_MHWConn=0;
1201                         m_MHWReader2->stop();
1202                         m_MHWConn2=0;
1203                 }
1204 #endif
1205                 if ( isRunning )
1206                         abortTimer.start(90000, true);
1207                 else
1208                 {
1209                         ++state;
1210                         for (int i=0; i < 3; ++i)
1211                         {
1212                                 seenSections[i].clear();
1213                                 calcedSections[i].clear();
1214                         }
1215                 }
1216         }
1217         ++state;
1218 }
1219
1220 void eEPGCache::channel_data::startChannel()
1221 {
1222         pthread_mutex_lock(&channel_active);
1223         updateMap::iterator It = cache->channelLastUpdated.find( channel->getChannelID() );
1224
1225         int update = ( It != cache->channelLastUpdated.end() ? ( UPDATE_INTERVAL - ( (eDVBLocalTimeHandler::getInstance()->nowTime()-It->second) * 1000 ) ) : ZAP_DELAY );
1226
1227         if (update < ZAP_DELAY)
1228                 update = ZAP_DELAY;
1229
1230         zapTimer.start(update, 1);
1231         if (update >= 60000)
1232                 eDebug("[EPGC] next update in %i min", update/60000);
1233         else if (update >= 1000)
1234                 eDebug("[EPGC] next update in %i sec", update/1000);
1235 }
1236
1237 void eEPGCache::channel_data::abortEPG()
1238 {
1239         for (int i=0; i < 3; ++i)
1240         {
1241                 seenSections[i].clear();
1242                 calcedSections[i].clear();
1243         }
1244         abortTimer.stop();
1245         zapTimer.stop();
1246         if (isRunning)
1247         {
1248                 eDebug("[EPGC] abort caching events !!");
1249                 if (isRunning & SCHEDULE)
1250                 {
1251                         isRunning &= ~SCHEDULE;
1252                         m_ScheduleReader->stop();
1253                         m_ScheduleConn=0;
1254                 }
1255                 if (isRunning & NOWNEXT)
1256                 {
1257                         isRunning &= ~NOWNEXT;
1258                         m_NowNextReader->stop();
1259                         m_NowNextConn=0;
1260                 }
1261                 if (isRunning & SCHEDULE_OTHER)
1262                 {
1263                         isRunning &= ~SCHEDULE_OTHER;
1264                         m_ScheduleOtherReader->stop();
1265                         m_ScheduleOtherConn=0;
1266                 }
1267 #ifdef ENABLE_MHW_EPG
1268                 if (isRunning & MHW)
1269                 {
1270                         isRunning &= ~MHW;
1271                         m_MHWReader->stop();
1272                         m_MHWConn=0;
1273                         m_MHWReader2->stop();
1274                         m_MHWConn2=0;
1275                 }
1276 #endif
1277         }
1278 #ifdef ENABLE_PRIVATE_EPG
1279         if (m_PrivateReader)
1280                 m_PrivateReader->stop();
1281         if (m_PrivateConn)
1282                 m_PrivateConn=0;
1283 #endif
1284         pthread_mutex_unlock(&channel_active);
1285 }
1286
1287 void eEPGCache::channel_data::readData( const __u8 *data)
1288 {
1289         int source;
1290         int map;
1291         iDVBSectionReader *reader=NULL;
1292         switch(data[0])
1293         {
1294                 case 0x4E ... 0x4F:
1295                         reader=m_NowNextReader;
1296                         source=NOWNEXT;
1297                         map=0;
1298                         break;
1299                 case 0x50 ... 0x5F:
1300                         reader=m_ScheduleReader;
1301                         source=SCHEDULE;
1302                         map=1;
1303                         break;
1304                 case 0x60 ... 0x6F:
1305                         reader=m_ScheduleOtherReader;
1306                         source=SCHEDULE_OTHER;
1307                         map=2;
1308                         break;
1309                 default:
1310                         eDebug("[EPGC] unknown table_id !!!");
1311                         return;
1312         }
1313         tidMap &seenSections = this->seenSections[map];
1314         tidMap &calcedSections = this->calcedSections[map];
1315         if ( state == 1 && calcedSections == seenSections || state > 1 )
1316         {
1317                 eDebugNoNewLine("[EPGC] ");
1318                 switch (source)
1319                 {
1320                         case NOWNEXT:
1321                                 m_NowNextConn=0;
1322                                 eDebugNoNewLine("nownext");
1323                                 break;
1324                         case SCHEDULE:
1325                                 m_ScheduleConn=0;
1326                                 eDebugNoNewLine("schedule");
1327                                 break;
1328                         case SCHEDULE_OTHER:
1329                                 m_ScheduleOtherConn=0;
1330                                 eDebugNoNewLine("schedule other");
1331                                 break;
1332                         default: eDebugNoNewLine("unknown");break;
1333                 }
1334                 eDebug(" finished(%ld)", eDVBLocalTimeHandler::getInstance()->nowTime());
1335                 if ( reader )
1336                         reader->stop();
1337                 isRunning &= ~source;
1338                 if (!isRunning)
1339                         finishEPG();
1340         }
1341         else
1342         {
1343                 eit_t *eit = (eit_t*) data;
1344                 __u32 sectionNo = data[0] << 24;
1345                 sectionNo |= data[3] << 16;
1346                 sectionNo |= data[4] << 8;
1347                 sectionNo |= eit->section_number;
1348
1349                 tidMap::iterator it =
1350                         seenSections.find(sectionNo);
1351
1352                 if ( it == seenSections.end() )
1353                 {
1354                         seenSections.insert(sectionNo);
1355                         calcedSections.insert(sectionNo);
1356                         __u32 tmpval = sectionNo & 0xFFFFFF00;
1357                         __u8 incr = source == NOWNEXT ? 1 : 8;
1358                         for ( int i = 0; i <= eit->last_section_number; i+=incr )
1359                         {
1360                                 if ( i == eit->section_number )
1361                                 {
1362                                         for (int x=i; x <= eit->segment_last_section_number; ++x)
1363                                                 calcedSections.insert(tmpval|(x&0xFF));
1364                                 }
1365                                 else
1366                                         calcedSections.insert(tmpval|(i&0xFF));
1367                         }
1368                         cache->sectionRead(data, source, this);
1369                 }
1370         }
1371 }
1372
1373 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, const eventData *&result, int direction)
1374 // if t == -1 we search the current event...
1375 {
1376         singleLock s(cache_lock);
1377         uniqueEPGKey key(handleGroup(service));
1378
1379         // check if EPG for this service is ready...
1380         eventCache::iterator It = eventDB.find( key );
1381         if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached ?
1382         {
1383                 if (t==-1)
1384                         t = eDVBLocalTimeHandler::getInstance()->nowTime();
1385                 timeMap::iterator i = direction <= 0 ? It->second.second.lower_bound(t) :  // find > or equal
1386                         It->second.second.upper_bound(t); // just >
1387                 if ( i != It->second.second.end() )
1388                 {
1389                         if ( direction < 0 || (direction == 0 && i->first > t) )
1390                         {
1391                                 timeMap::iterator x = i;
1392                                 --x;
1393                                 if ( x != It->second.second.end() )
1394                                 {
1395                                         time_t start_time = x->first;
1396                                         if (direction >= 0)
1397                                         {
1398                                                 if (t < start_time)
1399                                                         return -1;
1400                                                 if (t > (start_time+x->second->getDuration()))
1401                                                         return -1;
1402                                         }
1403                                         i = x;
1404                                 }
1405                                 else
1406                                         return -1;
1407                         }
1408                         result = i->second;
1409                         return 0;
1410                 }
1411         }
1412         return -1;
1413 }
1414
1415 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, const eit_event_struct *&result, int direction)
1416 {
1417         singleLock s(cache_lock);
1418         const eventData *data=0;
1419         RESULT ret = lookupEventTime(service, t, data, direction);
1420         if ( !ret && data )
1421                 result = data->get();
1422         return ret;
1423 }
1424
1425 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, Event *& result, int direction)
1426 {
1427         singleLock s(cache_lock);
1428         const eventData *data=0;
1429         RESULT ret = lookupEventTime(service, t, data, direction);
1430         if ( !ret && data )
1431                 result = new Event((uint8_t*)data->get());
1432         return ret;
1433 }
1434
1435 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, ePtr<eServiceEvent> &result, int direction)
1436 {
1437         singleLock s(cache_lock);
1438         const eventData *data=0;
1439         RESULT ret = lookupEventTime(service, t, data, direction);
1440         if ( !ret && data )
1441         {
1442                 Event ev((uint8_t*)data->get());
1443                 result = new eServiceEvent();
1444                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1445                 ret = result->parseFrom(&ev, (ref.getTransportStreamID().get()<<16)|ref.getOriginalNetworkID().get());
1446         }
1447         return ret;
1448 }
1449
1450 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eventData *&result )
1451 {
1452         singleLock s(cache_lock);
1453         uniqueEPGKey key(handleGroup(service));
1454
1455         eventCache::iterator It = eventDB.find( key );
1456         if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached?
1457         {
1458                 eventMap::iterator i( It->second.first.find( event_id ));
1459                 if ( i != It->second.first.end() )
1460                 {
1461                         result = i->second;
1462                         return 0;
1463                 }
1464                 else
1465                 {
1466                         result = 0;
1467                         eDebug("[EPGC] event %04x not found in epgcache", event_id);
1468                 }
1469         }
1470         return -1;
1471 }
1472
1473 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eit_event_struct *&result)
1474 {
1475         singleLock s(cache_lock);
1476         const eventData *data=0;
1477         RESULT ret = lookupEventId(service, event_id, data);
1478         if ( !ret && data )
1479                 result = data->get();
1480         return ret;
1481 }
1482
1483 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, Event *& result)
1484 {
1485         singleLock s(cache_lock);
1486         const eventData *data=0;
1487         RESULT ret = lookupEventId(service, event_id, data);
1488         if ( !ret && data )
1489                 result = new Event((uint8_t*)data->get());
1490         return ret;
1491 }
1492
1493 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, ePtr<eServiceEvent> &result)
1494 {
1495         singleLock s(cache_lock);
1496         const eventData *data=0;
1497         RESULT ret = lookupEventId(service, event_id, data);
1498         if ( !ret && data )
1499         {
1500                 Event ev((uint8_t*)data->get());
1501                 result = new eServiceEvent();
1502                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1503                 ret = result->parseFrom(&ev, (ref.getTransportStreamID().get()<<16)|ref.getOriginalNetworkID().get());
1504         }
1505         return ret;
1506 }
1507
1508 RESULT eEPGCache::startTimeQuery(const eServiceReference &service, time_t begin, int minutes)
1509 {
1510         const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)handleGroup(service);
1511         if (begin == -1)
1512                 begin = eDVBLocalTimeHandler::getInstance()->nowTime();
1513         Lock();
1514         eventCache::iterator It = eventDB.find(ref);
1515         if ( It != eventDB.end() && It->second.second.size() )
1516         {
1517                 m_timemap_cursor = It->second.second.lower_bound(begin);
1518                 if ( m_timemap_cursor != It->second.second.end() )
1519                 {
1520                         if ( m_timemap_cursor->first != begin )
1521                         {
1522                                 timeMap::iterator x = m_timemap_cursor;
1523                                 --x;
1524                                 if ( x != It->second.second.end() )
1525                                 {
1526                                         time_t start_time = x->first;
1527                                         if ( begin > start_time && begin < (start_time+x->second->getDuration()))
1528                                                 m_timemap_cursor = x;
1529                                 }
1530                         }
1531                 }
1532
1533                 if (minutes != -1)
1534                         m_timemap_end = It->second.second.upper_bound(begin+minutes*60);
1535                 else
1536                         m_timemap_end = It->second.second.end();
1537
1538                 currentQueryTsidOnid = (ref.getTransportStreamID().get()<<16) | ref.getOriginalNetworkID().get();
1539                 Unlock();
1540                 return 0;
1541         }
1542         Unlock();
1543         return -1;
1544 }
1545
1546 RESULT eEPGCache::getNextTimeEntry(const eventData *& result)
1547 {
1548         if ( m_timemap_cursor != m_timemap_end )
1549         {
1550                 result = m_timemap_cursor++->second;
1551                 return 0;
1552         }
1553         return -1;
1554 }
1555
1556 RESULT eEPGCache::getNextTimeEntry(const eit_event_struct *&result)
1557 {
1558         if ( m_timemap_cursor != m_timemap_end )
1559         {
1560                 result = m_timemap_cursor++->second->get();
1561                 return 0;
1562         }
1563         return -1;
1564 }
1565
1566 RESULT eEPGCache::getNextTimeEntry(Event *&result)
1567 {
1568         if ( m_timemap_cursor != m_timemap_end )
1569         {
1570                 result = new Event((uint8_t*)m_timemap_cursor++->second->get());
1571                 return 0;
1572         }
1573         return -1;
1574 }
1575
1576 RESULT eEPGCache::getNextTimeEntry(ePtr<eServiceEvent> &result)
1577 {
1578         if ( m_timemap_cursor != m_timemap_end )
1579         {
1580                 Event ev((uint8_t*)m_timemap_cursor++->second->get());
1581                 result = new eServiceEvent();
1582                 return result->parseFrom(&ev, currentQueryTsidOnid);
1583         }
1584         return -1;
1585 }
1586
1587 void fillTuple(ePyObject tuple, char *argstring, int argcount, ePyObject service, eServiceEvent *ptr, ePyObject nowTime, ePyObject service_name )
1588 {
1589         ePyObject tmp;
1590         int pos=0;
1591         while(pos < argcount)
1592         {
1593                 bool inc_refcount=false;
1594                 switch(argstring[pos])
1595                 {
1596                         case '0': // PyLong 0
1597                                 tmp = PyLong_FromLong(0);
1598                                 break;
1599                         case 'I': // Event Id
1600                                 tmp = ptr ? PyLong_FromLong(ptr->getEventId()) : ePyObject();
1601                                 break;
1602                         case 'B': // Event Begin Time
1603                                 tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : ePyObject();
1604                                 break;
1605                         case 'D': // Event Duration
1606                                 tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : ePyObject();
1607                                 break;
1608                         case 'T': // Event Title
1609                                 tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : ePyObject();
1610                                 break;
1611                         case 'S': // Event Short Description
1612                                 tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : ePyObject();
1613                                 break;
1614                         case 'E': // Event Extended Description
1615                                 tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : ePyObject();
1616                                 break;
1617                         case 'C': // Current Time
1618                                 tmp = nowTime;
1619                                 inc_refcount = true;
1620                                 break;
1621                         case 'R': // service reference string
1622                                 tmp = service;
1623                                 inc_refcount = true;
1624                                 break;
1625                         case 'n': // short service name
1626                         case 'N': // service name
1627                                 tmp = service_name;
1628                                 inc_refcount = true;
1629                 }
1630                 if (!tmp)
1631                 {
1632                         tmp = Py_None;
1633                         inc_refcount = true;
1634                 }
1635                 if (inc_refcount)
1636                         Py_INCREF(tmp);
1637                 PyTuple_SET_ITEM(tuple, pos++, tmp);
1638         }
1639 }
1640
1641 int handleEvent(eServiceEvent *ptr, ePyObject dest_list, char* argstring, int argcount, ePyObject service, ePyObject nowTime, ePyObject service_name, ePyObject convertFunc, ePyObject convertFuncArgs)
1642 {
1643         if (convertFunc)
1644         {
1645                 fillTuple(convertFuncArgs, argstring, argcount, service, ptr, nowTime, service_name);
1646                 ePyObject result = PyObject_CallObject(convertFunc, convertFuncArgs);
1647                 if (result)
1648                 {
1649                         if (service_name)
1650                                 Py_DECREF(service_name);
1651                         if (nowTime)
1652                                 Py_DECREF(nowTime);
1653                         Py_DECREF(convertFuncArgs);
1654                         Py_DECREF(dest_list);
1655                         PyErr_SetString(PyExc_StandardError,
1656                                 "error in convertFunc execute");
1657                         eDebug("error in convertFunc execute");
1658                         return -1;
1659                 }
1660                 PyList_Append(dest_list, result);
1661                 Py_DECREF(result);
1662         }
1663         else
1664         {
1665                 ePyObject tuple = PyTuple_New(argcount);
1666                 fillTuple(tuple, argstring, argcount, service, ptr, nowTime, service_name);
1667                 PyList_Append(dest_list, tuple);
1668                 Py_DECREF(tuple);
1669         }
1670         return 0;
1671 }
1672
1673 // here we get a python list
1674 // the first entry in the list is a python string to specify the format of the returned tuples (in a list)
1675 //   0 = PyLong(0)
1676 //   I = Event Id
1677 //   B = Event Begin Time
1678 //   D = Event Duration
1679 //   T = Event Title
1680 //   S = Event Short Description
1681 //   E = Event Extended Description
1682 //   C = Current Time
1683 //   R = Service Reference
1684 //   N = Service Name
1685 //   n = Short Service Name
1686 // then for each service follows a tuple
1687 //   first tuple entry is the servicereference (as string... use the ref.toString() function)
1688 //   the second is the type of query
1689 //     2 = event_id
1690 //    -1 = event before given start_time
1691 //     0 = event intersects given start_time
1692 //    +1 = event after given start_time
1693 //   the third
1694 //      when type is eventid it is the event_id
1695 //      when type is time then it is the start_time ( -1 for now_time )
1696 //   the fourth is the end_time .. ( optional .. for query all events in time range)
1697
1698 PyObject *eEPGCache::lookupEvent(ePyObject list, ePyObject convertFunc)
1699 {
1700         ePyObject convertFuncArgs;
1701         int argcount=0;
1702         char *argstring=NULL;
1703         if (!PyList_Check(list))
1704         {
1705                 PyErr_SetString(PyExc_StandardError,
1706                         "type error");
1707                 eDebug("no list");
1708                 return NULL;
1709         }
1710         int listIt=0;
1711         int listSize=PyList_Size(list);
1712         if (!listSize)
1713         {
1714                 PyErr_SetString(PyExc_StandardError,
1715                         "not params given");
1716                 eDebug("not params given");
1717                 return NULL;
1718         }
1719         else
1720         {
1721                 ePyObject argv=PyList_GET_ITEM(list, 0); // borrowed reference!
1722                 if (PyString_Check(argv))
1723                 {
1724                         argstring = PyString_AS_STRING(argv);
1725                         ++listIt;
1726                 }
1727                 else
1728                         argstring = "I"; // just event id as default
1729                 argcount = strlen(argstring);
1730 //              eDebug("have %d args('%s')", argcount, argstring);
1731         }
1732         if (convertFunc)
1733         {
1734                 if (!PyCallable_Check(convertFunc))
1735                 {
1736                         PyErr_SetString(PyExc_StandardError,
1737                                 "convertFunc must be callable");
1738                         eDebug("convertFunc is not callable");
1739                         return NULL;
1740                 }
1741                 convertFuncArgs = PyTuple_New(argcount);
1742         }
1743
1744         ePyObject nowTime = strchr(argstring, 'C') ?
1745                 PyLong_FromLong(eDVBLocalTimeHandler::getInstance()->nowTime()) :
1746                 ePyObject();
1747
1748         int must_get_service_name = strchr(argstring, 'N') ? 1 : strchr(argstring, 'n') ? 2 : 0;
1749
1750         // create dest list
1751         ePyObject dest_list=PyList_New(0);
1752         while(listSize > listIt)
1753         {
1754                 ePyObject item=PyList_GET_ITEM(list, listIt++); // borrowed reference!
1755                 if (PyTuple_Check(item))
1756                 {
1757                         bool service_changed=false;
1758                         int type=0;
1759                         long event_id=-1;
1760                         time_t stime=-1;
1761                         int minutes=0;
1762                         int tupleSize=PyTuple_Size(item);
1763                         int tupleIt=0;
1764                         ePyObject service;
1765                         while(tupleSize > tupleIt)  // parse query args
1766                         {
1767                                 ePyObject entry=PyTuple_GET_ITEM(item, tupleIt); // borrowed reference!
1768                                 switch(tupleIt++)
1769                                 {
1770                                         case 0:
1771                                         {
1772                                                 if (!PyString_Check(entry))
1773                                                 {
1774                                                         eDebug("tuple entry 0 is no a string");
1775                                                         goto skip_entry;
1776                                                 }
1777                                                 service = entry;
1778                                                 break;
1779                                         }
1780                                         case 1:
1781                                                 type=PyInt_AsLong(entry);
1782                                                 if (type < -1 || type > 2)
1783                                                 {
1784                                                         eDebug("unknown type %d", type);
1785                                                         goto skip_entry;
1786                                                 }
1787                                                 break;
1788                                         case 2:
1789                                                 event_id=stime=PyInt_AsLong(entry);
1790                                                 break;
1791                                         case 3:
1792                                                 minutes=PyInt_AsLong(entry);
1793                                                 break;
1794                                         default:
1795                                                 eDebug("unneeded extra argument");
1796                                                 break;
1797                                 }
1798                         }
1799
1800                         if (minutes && stime == -1)
1801                                 stime = eDVBLocalTimeHandler::getInstance()->nowTime();
1802
1803                         eServiceReference ref(handleGroup(eServiceReference(PyString_AS_STRING(service))));
1804                         if (ref.type != eServiceReference::idDVB)
1805                         {
1806                                 eDebug("service reference for epg query is not valid");
1807                                 continue;
1808                         }
1809
1810                         // redirect subservice querys to parent service
1811                         eServiceReferenceDVB &dvb_ref = (eServiceReferenceDVB&)ref;
1812                         if (dvb_ref.getParentTransportStreamID().get()) // linkage subservice
1813                         {
1814                                 eServiceCenterPtr service_center;
1815                                 if (!eServiceCenter::getPrivInstance(service_center))
1816                                 {
1817                                         dvb_ref.setTransportStreamID( dvb_ref.getParentTransportStreamID() );
1818                                         dvb_ref.setServiceID( dvb_ref.getParentServiceID() );
1819                                         dvb_ref.setParentTransportStreamID(eTransportStreamID(0));
1820                                         dvb_ref.setParentServiceID(eServiceID(0));
1821                                         dvb_ref.name="";
1822                                         service = PyString_FromString(dvb_ref.toString().c_str());
1823                                         service_changed = true;
1824                                 }
1825                         }
1826
1827                         ePyObject service_name;
1828                         if (must_get_service_name)
1829                         {
1830                                 ePtr<iStaticServiceInformation> sptr;
1831                                 eServiceCenterPtr service_center;
1832                                 eServiceCenter::getPrivInstance(service_center);
1833                                 if (service_center)
1834                                 {
1835                                         service_center->info(ref, sptr);
1836                                         if (sptr)
1837                                         {
1838                                                 std::string name;
1839                                                 sptr->getName(ref, name);
1840
1841                                                 if (must_get_service_name == 1)
1842                                                 {
1843                                                         unsigned int pos;
1844                                                         // filter short name brakets
1845                                                         while((pos = name.find("\xc2\x86")) != std::string::npos)
1846                                                                 name.erase(pos,2);
1847                                                         while((pos = name.find("\xc2\x87")) != std::string::npos)
1848                                                                 name.erase(pos,2);
1849                                                 }
1850                                                 else
1851                                                         name = buildShortName(name);
1852
1853                                                 if (name.length())
1854                                                         service_name = PyString_FromString(name.c_str());
1855                                         }
1856                                 }
1857                                 if (!service_name)
1858                                         service_name = PyString_FromString("<n/a>");
1859                         }
1860                         if (minutes)
1861                         {
1862                                 Lock();
1863                                 if (!startTimeQuery(ref, stime, minutes))
1864                                 {
1865                                         while ( m_timemap_cursor != m_timemap_end )
1866                                         {
1867                                                 Event ev((uint8_t*)m_timemap_cursor++->second->get());
1868                                                 eServiceEvent evt;
1869                                                 evt.parseFrom(&ev, currentQueryTsidOnid);
1870                                                 if (handleEvent(&evt, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs))
1871                                                 {
1872                                                         Unlock();
1873                                                         return 0;  // error
1874                                                 }
1875                                         }
1876                                 }
1877                                 else
1878                                         handleEvent(0, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs);
1879                                 Unlock();
1880                         }
1881                         else
1882                         {
1883                                 eServiceEvent evt;
1884                                 Event *ev=0;
1885                                 if (stime)
1886                                 {
1887                                         if (type == 2)
1888                                                 lookupEventId(ref, event_id, ev);
1889                                         else
1890                                                 lookupEventTime(ref, stime, ev, type);
1891                                         if (ev)
1892                                         {
1893                                                 const eServiceReferenceDVB &dref = (const eServiceReferenceDVB&)ref;
1894                                                 evt.parseFrom(ev, (dref.getTransportStreamID().get()<<16)|dref.getOriginalNetworkID().get());
1895                                         }
1896                                 }
1897                                 if (handleEvent(ev ? &evt : 0, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs))
1898                                         return 0; // error
1899                         }
1900                         if (service_changed)
1901                                 Py_DECREF(service);
1902                         if (service_name)
1903                                 Py_DECREF(service_name);
1904                 }
1905 skip_entry:
1906                 ;
1907         }
1908         if (convertFuncArgs)
1909                 Py_DECREF(convertFuncArgs);
1910         if (nowTime)
1911                 Py_DECREF(nowTime);
1912         return dest_list;
1913 }
1914
1915 void fillTuple2(ePyObject tuple, const char *argstring, int argcount, eventData *evData, eServiceEvent *ptr, ePyObject service_name, ePyObject service_reference)
1916 {
1917         ePyObject tmp;
1918         int pos=0;
1919         while(pos < argcount)
1920         {
1921                 bool inc_refcount=false;
1922                 switch(argstring[pos])
1923                 {
1924                         case '0': // PyLong 0
1925                                 tmp = PyLong_FromLong(0);
1926                                 break;
1927                         case 'I': // Event Id
1928                                 tmp = PyLong_FromLong(evData->getEventID());
1929                                 break;
1930                         case 'B': // Event Begin Time
1931                                 if (ptr)
1932                                         tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : ePyObject();
1933                                 else
1934                                         tmp = PyLong_FromLong(evData->getStartTime());
1935                                 break;
1936                         case 'D': // Event Duration
1937                                 if (ptr)
1938                                         tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : ePyObject();
1939                                 else
1940                                         tmp = PyLong_FromLong(evData->getDuration());
1941                                 break;
1942                         case 'T': // Event Title
1943                                 tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : ePyObject();
1944                                 break;
1945                         case 'S': // Event Short Description
1946                                 tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : ePyObject();
1947                                 break;
1948                         case 'E': // Event Extended Description
1949                                 tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : ePyObject();
1950                                 break;
1951                         case 'R': // service reference string
1952                                 tmp = service_reference;
1953                                 inc_refcount = true;
1954                                 break;
1955                         case 'n': // short service name
1956                         case 'N': // service name
1957                                 tmp = service_name;
1958                                 inc_refcount = true;
1959                                 break;
1960                 }
1961                 if (!tmp)
1962                 {
1963                         tmp = Py_None;
1964                         inc_refcount = true;
1965                 }
1966                 if (inc_refcount)
1967                         Py_INCREF(tmp);
1968                 PyTuple_SET_ITEM(tuple, pos++, tmp);
1969         }
1970 }
1971
1972 // here we get a python tuple
1973 // the first entry in the tuple is a python string to specify the format of the returned tuples (in a list)
1974 //   I = Event Id
1975 //   B = Event Begin Time
1976 //   D = Event Duration
1977 //   T = Event Title
1978 //   S = Event Short Description
1979 //   E = Event Extended Description
1980 //   R = Service Reference
1981 //   N = Service Name
1982 //   n = Short Service Name
1983 //  the second tuple entry is the MAX matches value
1984 //  the third tuple entry is the type of query
1985 //     0 = search for similar broadcastings (SIMILAR_BROADCASTINGS_SEARCH)
1986 //     1 = search events with exactly title name (EXAKT_TITLE_SEARCH)
1987 //     2 = search events with text in title name (PARTIAL_TITLE_SEARCH)
1988 //  when type is 0 (SIMILAR_BROADCASTINGS_SEARCH)
1989 //   the fourth is the servicereference string
1990 //   the fifth is the eventid
1991 //  when type is 1 or 2 (EXAKT_TITLE_SEARCH or PARTIAL_TITLE_SEARCH)
1992 //   the fourth is the search text
1993 //   the fifth is
1994 //     0 = case sensitive (CASE_CHECK)
1995 //     1 = case insensitive (NO_CASECHECK)
1996
1997 PyObject *eEPGCache::search(ePyObject arg)
1998 {
1999         ePyObject ret;
2000         int descridx = -1;
2001         __u32 descr[512];
2002         int eventid = -1;
2003         const char *argstring=0;
2004         char *refstr=0;
2005         int argcount=0;
2006         int querytype=-1;
2007         bool needServiceEvent=false;
2008         int maxmatches=0;
2009
2010         if (PyTuple_Check(arg))
2011         {
2012                 int tuplesize=PyTuple_Size(arg);
2013                 if (tuplesize > 0)
2014                 {
2015                         ePyObject obj = PyTuple_GET_ITEM(arg,0);
2016                         if (PyString_Check(obj))
2017                         {
2018                                 argcount = PyString_GET_SIZE(obj);
2019                                 argstring = PyString_AS_STRING(obj);
2020                                 for (int i=0; i < argcount; ++i)
2021                                         switch(argstring[i])
2022                                         {
2023                                         case 'S':
2024                                         case 'E':
2025                                         case 'T':
2026                                                 needServiceEvent=true;
2027                                         default:
2028                                                 break;
2029                                         }
2030                         }
2031                         else
2032                         {
2033                                 PyErr_SetString(PyExc_StandardError,
2034                                         "type error");
2035                                 eDebug("tuple arg 0 is not a string");
2036                                 return NULL;
2037                         }
2038                 }
2039                 if (tuplesize > 1)
2040                         maxmatches = PyLong_AsLong(PyTuple_GET_ITEM(arg, 1));
2041                 if (tuplesize > 2)
2042                 {
2043                         querytype = PyLong_AsLong(PyTuple_GET_ITEM(arg, 2));
2044                         if (tuplesize > 4 && querytype == 0)
2045                         {
2046                                 ePyObject obj = PyTuple_GET_ITEM(arg, 3);
2047                                 if (PyString_Check(obj))
2048                                 {
2049                                         refstr = PyString_AS_STRING(obj);
2050                                         eServiceReferenceDVB ref(refstr);
2051                                         if (ref.valid())
2052                                         {
2053                                                 eventid = PyLong_AsLong(PyTuple_GET_ITEM(arg, 4));
2054                                                 singleLock s(cache_lock);
2055                                                 const eventData *evData = 0;
2056                                                 lookupEventId(ref, eventid, evData);
2057                                                 if (evData)
2058                                                 {
2059                                                         __u8 *data = evData->EITdata;
2060                                                         int tmp = evData->ByteSize-10;
2061                                                         __u32 *p = (__u32*)(data+10);
2062                                                                 // search short and extended event descriptors
2063                                                         while(tmp>3)
2064                                                         {
2065                                                                 __u32 crc = *p++;
2066                                                                 descriptorMap::iterator it =
2067                                                                         eventData::descriptors.find(crc);
2068                                                                 if (it != eventData::descriptors.end())
2069                                                                 {
2070                                                                         __u8 *descr_data = it->second.second;
2071                                                                         switch(descr_data[0])
2072                                                                         {
2073                                                                         case 0x4D ... 0x4E:
2074                                                                                 descr[++descridx]=crc;
2075                                                                         default:
2076                                                                                 break;
2077                                                                         }
2078                                                                 }
2079                                                                 tmp-=4;
2080                                                         }
2081                                                 }
2082                                                 if (descridx<0)
2083                                                         eDebug("event not found");
2084                                         }
2085                                         else
2086                                         {
2087                                                 PyErr_SetString(PyExc_StandardError,
2088                                                         "type error");
2089                                                 eDebug("tuple arg 4 is not a valid service reference string");
2090                                                 return NULL;
2091                                         }
2092                                 }
2093                                 else
2094                                 {
2095                                         PyErr_SetString(PyExc_StandardError,
2096                                         "type error");
2097                                         eDebug("tuple arg 4 is not a string");
2098                                         return NULL;
2099                                 }
2100                         }
2101                         else if (tuplesize > 4 && (querytype == 1 || querytype == 2) )
2102                         {
2103                                 ePyObject obj = PyTuple_GET_ITEM(arg, 3);
2104                                 if (PyString_Check(obj))
2105                                 {
2106                                         int casetype = PyLong_AsLong(PyTuple_GET_ITEM(arg, 4));
2107                                         const char *str = PyString_AS_STRING(obj);
2108                                         int textlen = PyString_GET_SIZE(obj);
2109                                         if (querytype == 1)
2110                                                 eDebug("lookup for events with '%s' as title(%s)", str, casetype?"ignore case":"case sensitive");
2111                                         else
2112                                                 eDebug("lookup for events with '%s' in title(%s)", str, casetype?"ignore case":"case sensitive");
2113                                         singleLock s(cache_lock);
2114                                         for (descriptorMap::iterator it(eventData::descriptors.begin());
2115                                                 it != eventData::descriptors.end() && descridx < 511; ++it)
2116                                         {
2117                                                 __u8 *data = it->second.second;
2118                                                 if ( data[0] == 0x4D ) // short event descriptor
2119                                                 {
2120                                                         int title_len = data[5];
2121                                                         if ( querytype == 1 )
2122                                                         {
2123                                                                 if (title_len > textlen)
2124                                                                         continue;
2125                                                                 else if (title_len < textlen)
2126                                                                         continue;
2127                                                                 if ( casetype )
2128                                                                 {
2129                                                                         if ( !strncasecmp((const char*)data+6, str, title_len) )
2130                                                                         {
2131 //                                                                              std::string s((const char*)data+6, title_len);
2132 //                                                                              eDebug("match1 %s %s", str, s.c_str() );
2133                                                                                 descr[++descridx] = it->first;
2134                                                                         }
2135                                                                 }
2136                                                                 else if ( !strncmp((const char*)data+6, str, title_len) )
2137                                                                 {
2138 //                                                                      std::string s((const char*)data+6, title_len);
2139 //                                                                      eDebug("match2 %s %s", str, s.c_str() );
2140                                                                         descr[++descridx] = it->first;
2141                                                                 }
2142                                                         }
2143                                                         else
2144                                                         {
2145                                                                 int idx=0;
2146                                                                 while((title_len-idx) >= textlen)
2147                                                                 {
2148                                                                         if (casetype)
2149                                                                         {
2150                                                                                 if (!strncasecmp((const char*)data+6+idx, str, textlen) )
2151                                                                                 {
2152                                                                                         descr[++descridx] = it->first;
2153 //                                                                                      std::string s((const char*)data+6, title_len);
2154 //                                                                                      eDebug("match 3 %s %s", str, s.c_str() );
2155                                                                                         break;
2156                                                                                 }
2157                                                                                 else if (!strncmp((const char*)data+6+idx, str, textlen) )
2158                                                                                 {
2159                                                                                         descr[++descridx] = it->first;
2160 //                                                                                      std::string s((const char*)data+6, title_len);
2161 //                                                                                      eDebug("match 4 %s %s", str, s.c_str() );
2162                                                                                         break;
2163                                                                                 }
2164                                                                         }
2165                                                                         ++idx;
2166                                                                 }
2167                                                         }
2168                                                 }
2169                                         }
2170                                 }
2171                                 else
2172                                 {
2173                                         PyErr_SetString(PyExc_StandardError,
2174                                                 "type error");
2175                                         eDebug("tuple arg 4 is not a string");
2176                                         return NULL;
2177                                 }
2178                         }
2179                         else
2180                         {
2181                                 PyErr_SetString(PyExc_StandardError,
2182                                         "type error");
2183                                 eDebug("tuple arg 3(%d) is not a known querytype(0, 1, 2)", querytype);
2184                                 return NULL;
2185                         }
2186                 }
2187                 else
2188                 {
2189                         PyErr_SetString(PyExc_StandardError,
2190                                 "type error");
2191                         eDebug("not enough args in tuple");
2192                         return NULL;
2193                 }
2194         }
2195         else
2196         {
2197                 PyErr_SetString(PyExc_StandardError,
2198                         "type error");
2199                 eDebug("arg 0 is not a tuple");
2200                 return NULL;
2201         }
2202
2203         if (descridx > -1)
2204         {
2205                 int maxcount=maxmatches;
2206                 eServiceReferenceDVB ref(refstr?(const eServiceReferenceDVB&)handleGroup(eServiceReference(refstr)):eServiceReferenceDVB(""));
2207                 // ref is only valid in SIMILAR_BROADCASTING_SEARCH
2208                 // in this case we start searching with the base service
2209                 bool first = ref.valid() ? true : false;
2210                 singleLock s(cache_lock);
2211                 eventCache::iterator cit(ref.valid() ? eventDB.find(ref) : eventDB.begin());
2212                 while(cit != eventDB.end() && maxcount)
2213                 {
2214                         if ( ref.valid() && !first && cit->first == ref )
2215                         {
2216                                 // do not scan base service twice ( only in SIMILAR BROADCASTING SEARCH )
2217                                 ++cit;
2218                                 continue;
2219                         }
2220                         ePyObject service_name;
2221                         ePyObject service_reference;
2222                         timeMap &evmap = cit->second.second;
2223                         // check all events
2224                         for (timeMap::iterator evit(evmap.begin()); evit != evmap.end() && maxcount; ++evit)
2225                         {
2226                                 int evid = evit->second->getEventID();
2227                                 if ( evid == eventid)
2228                                         continue;
2229                                 __u8 *data = evit->second->EITdata;
2230                                 int tmp = evit->second->ByteSize-10;
2231                                 __u32 *p = (__u32*)(data+10);
2232                                 // check if any of our descriptor used by this event
2233                                 int cnt=-1;
2234                                 while(tmp>3)
2235                                 {
2236                                         __u32 crc32 = *p++;
2237                                         for ( int i=0; i <= descridx; ++i)
2238                                         {
2239                                                 if (descr[i] == crc32)  // found...
2240                                                         ++cnt;
2241                                         }
2242                                         tmp-=4;
2243                                 }
2244                                 if ( (querytype == 0 && cnt == descridx) ||
2245                                          ((querytype == 1 || querytype == 2) && cnt != -1) )
2246                                 {
2247                                         const uniqueEPGKey &service = cit->first;
2248                                         eServiceReference ref =
2249                                                 eDVBDB::getInstance()->searchReference(service.tsid, service.onid, service.sid);
2250                                         if (ref.valid())
2251                                         {
2252                                         // create servive event
2253                                                 eServiceEvent ptr;
2254                                                 Event *ev=0;
2255                                                 if (needServiceEvent)
2256                                                 {
2257                                                         if (lookupEventId(ref, evid, ev))
2258                                                                 eDebug("event not found !!!!!!!!!!!");
2259                                                         else
2260                                                         {
2261                                                                 const eServiceReferenceDVB &dref = (const eServiceReferenceDVB&)ref;
2262                                                                 ptr.parseFrom(ev, (dref.getTransportStreamID().get()<<16)|dref.getOriginalNetworkID().get());
2263                                                         }
2264                                                 }
2265                                         // create service name
2266                                                 if (!service_name)
2267                                                 {
2268                                                         int must_get_service_name = strchr(argstring, 'N') ? 1 : strchr(argstring, 'n') ? 2 : 0;
2269                                                         if (must_get_service_name)
2270                                                         {
2271                                                                 ePtr<iStaticServiceInformation> sptr;
2272                                                                 eServiceCenterPtr service_center;
2273                                                                 eServiceCenter::getPrivInstance(service_center);
2274                                                                 if (service_center)
2275                                                                 {
2276                                                                         service_center->info(ref, sptr);
2277                                                                         if (sptr)
2278                                                                         {
2279                                                                                 std::string name;
2280                                                                                 sptr->getName(ref, name);
2281
2282                                                                                 if (must_get_service_name == 1)
2283                                                                                 {
2284                                                                                         unsigned int pos;
2285                                                                                         // filter short name brakets
2286                                                                                         while((pos = name.find("\xc2\x86")) != std::string::npos)
2287                                                                                                 name.erase(pos,2);
2288                                                                                         while((pos = name.find("\xc2\x87")) != std::string::npos)
2289                                                                                                 name.erase(pos,2);
2290                                                                                 }
2291                                                                                 else
2292                                                                                         name = buildShortName(name);
2293
2294                                                                                 if (name.length())
2295                                                                                         service_name = PyString_FromString(name.c_str());
2296                                                                         }
2297                                                                 }
2298                                                                 if (!service_name)
2299                                                                         service_name = PyString_FromString("<n/a>");
2300                                                         }
2301                                                 }
2302                                         // create servicereference string
2303                                                 if (!service_reference && strchr(argstring,'R'))
2304                                                         service_reference = PyString_FromString(ref.toString().c_str());
2305                                         // create list
2306                                                 if (!ret)
2307                                                         ret = PyList_New(0);
2308                                         // create tuple
2309                                                 ePyObject tuple = PyTuple_New(argcount);
2310                                         // fill tuple
2311                                                 fillTuple2(tuple, argstring, argcount, evit->second, ev ? &ptr : 0, service_name, service_reference);
2312                                                 PyList_Append(ret, tuple);
2313                                                 Py_DECREF(tuple);
2314                                                 --maxcount;
2315                                         }
2316                                 }
2317                         }
2318                         if (service_name)
2319                                 Py_DECREF(service_name);
2320                         if (service_reference)
2321                                 Py_DECREF(service_reference);
2322                         if (first)
2323                         {
2324                                 // now start at first service in epgcache database ( only in SIMILAR BROADCASTING SEARCH )
2325                                 first=false;
2326                                 cit=eventDB.begin();
2327                         }
2328                         else
2329                                 ++cit;
2330                 }
2331         }
2332
2333         if (!ret)
2334                 Py_RETURN_NONE;
2335
2336         return ret;
2337 }
2338
2339 #ifdef ENABLE_PRIVATE_EPG
2340 #include <dvbsi++/descriptor_tag.h>
2341 #include <dvbsi++/unknown_descriptor.h>
2342 #include <dvbsi++/private_data_specifier_descriptor.h>
2343
2344 void eEPGCache::PMTready(eDVBServicePMTHandler *pmthandler)
2345 {
2346         ePtr<eTable<ProgramMapSection> > ptr;
2347         if (!pmthandler->getPMT(ptr) && ptr)
2348         {
2349                 std::vector<ProgramMapSection*>::const_iterator i;
2350                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
2351                 {
2352                         const ProgramMapSection &pmt = **i;
2353
2354                         ElementaryStreamInfoConstIterator es;
2355                         for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
2356                         {
2357                                 int tmp=0;
2358                                 switch ((*es)->getType())
2359                                 {
2360                                 case 0x05: // private
2361                                         for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
2362                                                 desc != (*es)->getDescriptors()->end(); ++desc)
2363                                         {
2364                                                 switch ((*desc)->getTag())
2365                                                 {
2366                                                         case PRIVATE_DATA_SPECIFIER_DESCRIPTOR:
2367                                                                 if (((PrivateDataSpecifierDescriptor*)(*desc))->getPrivateDataSpecifier() == 190)
2368                                                                         tmp |= 1;
2369                                                                 break;
2370                                                         case 0x90:
2371                                                         {
2372                                                                 UnknownDescriptor *descr = (UnknownDescriptor*)*desc;
2373                                                                 int descr_len = descr->getLength();
2374                                                                 if (descr_len == 4)
2375                                                                 {
2376                                                                         uint8_t data[descr_len+2];
2377                                                                         descr->writeToBuffer(data);
2378                                                                         if ( !data[2] && !data[3] && data[4] == 0xFF && data[5] == 0xFF )
2379                                                                                 tmp |= 2;
2380                                                                 }
2381                                                                 break;
2382                                                         }
2383                                                         default:
2384                                                                 break;
2385                                                 }
2386                                         }
2387                                 default:
2388                                         break;
2389                                 }
2390                                 if (tmp==3)
2391                                 {
2392                                         eServiceReferenceDVB ref;
2393                                         if (!pmthandler->getServiceReference(ref))
2394                                         {
2395                                                 int pid = (*es)->getPid();
2396                                                 messages.send(Message(Message::got_private_pid, ref, pid));
2397                                                 return;
2398                                         }
2399                                 }
2400                         }
2401                 }
2402         }
2403         else
2404                 eDebug("PMTready but no pmt!!");
2405 }
2406
2407 struct date_time
2408 {
2409         __u8 data[5];
2410         time_t tm;
2411         date_time( const date_time &a )
2412         {
2413                 memcpy(data, a.data, 5);
2414                 tm = a.tm;
2415         }
2416         date_time( const __u8 data[5])
2417         {
2418                 memcpy(this->data, data, 5);
2419                 tm = parseDVBtime(data[0], data[1], data[2], data[3], data[4]);
2420         }
2421         date_time()
2422         {
2423         }
2424         const __u8& operator[](int pos) const
2425         {
2426                 return data[pos];
2427         }
2428 };
2429
2430 struct less_datetime
2431 {
2432         bool operator()( const date_time &a, const date_time &b ) const
2433         {
2434                 return abs(a.tm-b.tm) < 360 ? false : a.tm < b.tm;
2435         }
2436 };
2437
2438 void eEPGCache::privateSectionRead(const uniqueEPGKey &current_service, const __u8 *data)
2439 {
2440         contentMap &content_time_table = content_time_tables[current_service];
2441         singleLock s(cache_lock);
2442         std::map< date_time, std::list<uniqueEPGKey>, less_datetime > start_times;
2443         eventMap &evMap = eventDB[current_service].first;
2444         timeMap &tmMap = eventDB[current_service].second;
2445         int ptr=8;
2446         int content_id = data[ptr++] << 24;
2447         content_id |= data[ptr++] << 16;
2448         content_id |= data[ptr++] << 8;
2449         content_id |= data[ptr++];
2450
2451         contentTimeMap &time_event_map =
2452                 content_time_table[content_id];
2453         for ( contentTimeMap::iterator it( time_event_map.begin() );
2454                 it != time_event_map.end(); ++it )
2455         {
2456                 eventMap::iterator evIt( evMap.find(it->second.second) );
2457                 if ( evIt != evMap.end() )
2458                 {
2459                         delete evIt->second;
2460                         evMap.erase(evIt);
2461                 }
2462                 tmMap.erase(it->second.first);
2463         }
2464         time_event_map.clear();
2465
2466         __u8 duration[3];
2467         memcpy(duration, data+ptr, 3);
2468         ptr+=3;
2469         int duration_sec =
2470                 fromBCD(duration[0])*3600+fromBCD(duration[1])*60+fromBCD(duration[2]);
2471
2472         const __u8 *descriptors[65];
2473         const __u8 **pdescr = descriptors;
2474
2475         int descriptors_length = (data[ptr++]&0x0F) << 8;
2476         descriptors_length |= data[ptr++];
2477         while ( descriptors_length > 1 )
2478         {
2479                 int descr_type = data[ptr];
2480                 int descr_len = data[ptr+1];
2481                 descriptors_length -= 2;
2482                 if (descriptors_length >= descr_len)
2483                 {
2484                         descriptors_length -= descr_len;
2485                         if ( descr_type == 0xf2 && descr_len > 5)
2486                         {
2487                                 ptr+=2;
2488                                 int tsid = data[ptr++] << 8;
2489                                 tsid |= data[ptr++];
2490                                 int onid = data[ptr++] << 8;
2491                                 onid |= data[ptr++];
2492                                 int sid = data[ptr++] << 8;
2493                                 sid |= data[ptr++];
2494
2495 // WORKAROUND for wrong transmitted epg data (01.08.2006)
2496                                 if ( onid == 0x85 )
2497                                 {
2498                                         switch( (tsid << 16) | sid )
2499                                         {
2500                                                 case 0x01030b: sid = 0x1b; tsid = 4; break;  // Premiere Win
2501                                                 case 0x0300f0: sid = 0xe0; tsid = 2; break;
2502                                                 case 0x0300f1: sid = 0xe1; tsid = 2; break;
2503                                                 case 0x0300f5: sid = 0xdc; break;
2504                                                 case 0x0400d2: sid = 0xe2; tsid = 0x11; break;
2505                                                 case 0x1100d3: sid = 0xe3; break;
2506                                         }
2507                                 }
2508 ////////////////////////////////////////////
2509
2510                                 uniqueEPGKey service( sid, onid, tsid );
2511                                 descr_len -= 6;
2512                                 while( descr_len > 2 )
2513                                 {
2514                                         __u8 datetime[5];
2515                                         datetime[0] = data[ptr++];
2516                                         datetime[1] = data[ptr++];
2517                                         int tmp_len = data[ptr++];
2518                                         descr_len -= 3;
2519                                         if (descr_len >= tmp_len)
2520                                         {
2521                                                 descr_len -= tmp_len;
2522                                                 while( tmp_len > 2 )
2523                                                 {
2524                                                         memcpy(datetime+2, data+ptr, 3);
2525                                                         ptr += 3;
2526                                                         tmp_len -= 3;
2527                                                         start_times[datetime].push_back(service);
2528                                                 }
2529                                         }
2530                                 }
2531                         }
2532                         else
2533                         {
2534                                 *pdescr++=data+ptr;
2535                                 ptr += 2;
2536                                 ptr += descr_len;
2537                         }
2538                 }
2539         }
2540         ASSERT(pdescr <= &descriptors[65])
2541         __u8 event[4098];
2542         eit_event_struct *ev_struct = (eit_event_struct*) event;
2543         ev_struct->running_status = 0;
2544         ev_struct->free_CA_mode = 1;
2545         memcpy(event+7, duration, 3);
2546         ptr = 12;
2547         const __u8 **d=descriptors;
2548         while ( d < pdescr )
2549         {
2550                 memcpy(event+ptr, *d, ((*d)[1])+2);
2551                 ptr+=(*d++)[1];
2552                 ptr+=2;
2553         }
2554         ASSERT(ptr <= 4098);
2555         for ( std::map< date_time, std::list<uniqueEPGKey> >::iterator it(start_times.begin()); it != start_times.end(); ++it )
2556         {
2557                 time_t now = eDVBLocalTimeHandler::getInstance()->nowTime();
2558                 if ( (it->first.tm + duration_sec) < now )
2559                         continue;
2560                 memcpy(event+2, it->first.data, 5);
2561                 int bptr = ptr;
2562                 int cnt=0;
2563                 for (std::list<uniqueEPGKey>::iterator i(it->second.begin()); i != it->second.end(); ++i)
2564                 {
2565                         event[bptr++] = 0x4A;
2566                         __u8 *len = event+(bptr++);
2567                         event[bptr++] = (i->tsid & 0xFF00) >> 8;
2568                         event[bptr++] = (i->tsid & 0xFF);
2569                         event[bptr++] = (i->onid & 0xFF00) >> 8;
2570                         event[bptr++] = (i->onid & 0xFF);
2571                         event[bptr++] = (i->sid & 0xFF00) >> 8;
2572                         event[bptr++] = (i->sid & 0xFF);
2573                         event[bptr++] = 0xB0;
2574                         bptr += sprintf((char*)(event+bptr), "Option %d", ++cnt);
2575                         *len = ((event+bptr) - len)-1;
2576                 }
2577                 int llen = bptr - 12;
2578                 ev_struct->descriptors_loop_length_hi = (llen & 0xF00) >> 8;
2579                 ev_struct->descriptors_loop_length_lo = (llen & 0xFF);
2580
2581                 time_t stime = it->first.tm;
2582                 while( tmMap.find(stime) != tmMap.end() )
2583                         ++stime;
2584                 event[6] += (stime - it->first.tm);
2585                 __u16 event_id = 0;
2586                 while( evMap.find(event_id) != evMap.end() )
2587                         ++event_id;
2588                 event[0] = (event_id & 0xFF00) >> 8;
2589                 event[1] = (event_id & 0xFF);
2590                 time_event_map[it->first.tm]=std::pair<time_t, __u16>(stime, event_id);
2591                 eventData *d = new eventData( ev_struct, bptr, PRIVATE );
2592                 evMap[event_id] = d;
2593                 tmMap[stime] = d;
2594                 ASSERT(bptr <= 4098);
2595         }
2596 }
2597
2598 void eEPGCache::channel_data::startPrivateReader()
2599 {
2600         eDVBSectionFilterMask mask;
2601         memset(&mask, 0, sizeof(mask));
2602         mask.pid = m_PrivatePid;
2603         mask.flags = eDVBSectionFilterMask::rfCRC;
2604         mask.data[0] = 0xA0;
2605         mask.mask[0] = 0xFF;
2606         eDebug("[EPGC] start privatefilter for pid %04x and version %d", m_PrivatePid, m_PrevVersion);
2607         if (m_PrevVersion != -1)
2608         {
2609                 mask.data[3] = m_PrevVersion << 1;
2610                 mask.mask[3] = 0x3E;
2611                 mask.mode[3] = 0x3E;
2612         }
2613         seenPrivateSections.clear();
2614         if (!m_PrivateConn)
2615                 m_PrivateReader->connectRead(slot(*this, &eEPGCache::channel_data::readPrivateData), m_PrivateConn);
2616         m_PrivateReader->start(mask);
2617 }
2618
2619 void eEPGCache::channel_data::readPrivateData( const __u8 *data)
2620 {
2621         if ( seenPrivateSections.find(data[6]) == seenPrivateSections.end() )
2622         {
2623                 cache->privateSectionRead(m_PrivateService, data);
2624                 seenPrivateSections.insert(data[6]);
2625         }
2626         if ( seenPrivateSections.size() == (unsigned int)(data[7] + 1) )
2627         {
2628                 eDebug("[EPGC] private finished");
2629                 eDVBChannelID chid = channel->getChannelID();
2630                 int tmp = chid.original_network_id.get();
2631                 tmp |= 0x80000000; // we use highest bit as private epg indicator
2632                 chid.original_network_id = tmp;
2633                 cache->channelLastUpdated[chid] = eDVBLocalTimeHandler::getInstance()->nowTime();
2634                 m_PrevVersion = (data[5] & 0x3E) >> 1;
2635                 startPrivateReader();
2636         }
2637 }
2638
2639 #endif // ENABLE_PRIVATE_EPG
2640
2641 #ifdef ENABLE_MHW_EPG
2642 void eEPGCache::channel_data::cleanup()
2643 {
2644         m_channels.clear();
2645         m_themes.clear();
2646         m_titles.clear();
2647         m_program_ids.clear();
2648 }
2649
2650 __u8 *eEPGCache::channel_data::delimitName( __u8 *in, __u8 *out, int len_in )
2651 {
2652         // Names in mhw structs are not strings as they are not '\0' terminated.
2653         // This function converts the mhw name into a string.
2654         // Constraint: "length of out" = "length of in" + 1.
2655         int i;
2656         for ( i=0; i < len_in; i++ )
2657                 out[i] = in[i];
2658
2659         i = len_in - 1;
2660         while ( ( i >=0 ) && ( out[i] == 0x20 ) )
2661                 i--;
2662
2663         out[i+1] = 0;
2664         return out;
2665 }
2666
2667 void eEPGCache::channel_data::timeMHW2DVB( u_char hours, u_char minutes, u_char *return_time)
2668 // For time of day
2669 {
2670         return_time[0] = toBCD( hours );
2671         return_time[1] = toBCD( minutes );
2672         return_time[2] = 0;
2673 }
2674
2675 void eEPGCache::channel_data::timeMHW2DVB( int minutes, u_char *return_time)
2676 {
2677         timeMHW2DVB( int(minutes/60), minutes%60, return_time );
2678 }
2679
2680 void eEPGCache::channel_data::timeMHW2DVB( u_char day, u_char hours, u_char minutes, u_char *return_time)
2681 // For date plus time of day
2682 {
2683         // Remove offset in mhw time.
2684         __u8 local_hours = hours;
2685         if ( hours >= 16 )
2686                 local_hours -= 4;
2687         else if ( hours >= 8 )
2688                 local_hours -= 2;
2689
2690         // As far as we know all mhw time data is sent in central Europe time zone.
2691         // So, temporarily set timezone to western europe
2692         time_t dt = eDVBLocalTimeHandler::getInstance()->nowTime();
2693
2694         char *old_tz = getenv( "TZ" );
2695         putenv("TZ=CET-1CEST,M3.5.0/2,M10.5.0/3");
2696         tzset();
2697
2698         tm localnow;
2699         localtime_r(&dt, &localnow);
2700
2701         if (day == 7)
2702                 day = 0;
2703         if ( day + 1 < localnow.tm_wday )               // day + 1 to prevent old events to show for next week.
2704                 day += 7;
2705         if (local_hours <= 5)
2706                 day++;
2707
2708         dt += 3600*24*(day - localnow.tm_wday); // Shift dt to the recording date (local time zone).
2709         dt += 3600*(local_hours - localnow.tm_hour);  // Shift dt to the recording hour.
2710
2711         tm recdate;
2712         gmtime_r( &dt, &recdate );   // This will also take care of DST.
2713
2714         if ( old_tz == NULL )
2715                 unsetenv( "TZ" );
2716         else
2717                 putenv( old_tz );
2718         tzset();
2719
2720         // Calculate MJD according to annex in ETSI EN 300 468
2721         int l=0;
2722         if ( recdate.tm_mon <= 1 )      // Jan or Feb
2723                 l=1;
2724         int mjd = 14956 + recdate.tm_mday + int( (recdate.tm_year - l) * 365.25) +
2725                 int( (recdate.tm_mon + 2 + l * 12) * 30.6001);
2726
2727         return_time[0] = (mjd & 0xFF00)>>8;
2728         return_time[1] = mjd & 0xFF;
2729
2730         timeMHW2DVB( recdate.tm_hour, minutes, return_time+2 );
2731 }
2732
2733 void eEPGCache::channel_data::storeTitle(std::map<__u32, mhw_title_t>::iterator itTitle, std::string sumText, const __u8 *data)
2734 // data is borrowed from calling proc to save memory space.
2735 {
2736         __u8 name[34];
2737
2738         // For each title a separate EIT packet will be sent to eEPGCache::sectionRead()
2739         bool isMHW2 = itTitle->second.mhw2_mjd_hi || itTitle->second.mhw2_mjd_lo ||
2740                 itTitle->second.mhw2_duration_hi || itTitle->second.mhw2_duration_lo;
2741
2742         eit_t *packet = (eit_t *) data;
2743         packet->table_id = 0x50;
2744         packet->section_syntax_indicator = 1;
2745
2746         packet->service_id_hi = m_channels[ itTitle->second.channel_id - 1 ].channel_id_hi;
2747         packet->service_id_lo = m_channels[ itTitle->second.channel_id - 1 ].channel_id_lo;
2748         packet->version_number = 0;     // eEPGCache::sectionRead() will dig this for the moment
2749         packet->current_next_indicator = 0;
2750         packet->section_number = 0;     // eEPGCache::sectionRead() will dig this for the moment
2751         packet->last_section_number = 0;        // eEPGCache::sectionRead() will dig this for the moment
2752         packet->transport_stream_id_hi = m_channels[ itTitle->second.channel_id - 1 ].transport_stream_id_hi;
2753         packet->transport_stream_id_lo = m_channels[ itTitle->second.channel_id - 1 ].transport_stream_id_lo;
2754         packet->original_network_id_hi = m_channels[ itTitle->second.channel_id - 1 ].network_id_hi;
2755         packet->original_network_id_lo = m_channels[ itTitle->second.channel_id - 1 ].network_id_lo;
2756         packet->segment_last_section_number = 0; // eEPGCache::sectionRead() will dig this for the moment
2757         packet->segment_last_table_id = 0x50;
2758
2759         __u8 *title = isMHW2 ? ((__u8*)(itTitle->second.title))-4 : (__u8*)itTitle->second.title;
2760         std::string prog_title = (char *) delimitName( title, name, isMHW2 ? 33 : 23 );
2761         int prog_title_length = prog_title.length();
2762
2763         int packet_length = EIT_SIZE + EIT_LOOP_SIZE + EIT_SHORT_EVENT_DESCRIPTOR_SIZE +
2764                 prog_title_length + 1;
2765
2766         eit_event_t *event_data = (eit_event_t *) (data + EIT_SIZE);
2767         event_data->event_id_hi = (( itTitle->first ) >> 8 ) & 0xFF;
2768         event_data->event_id_lo = ( itTitle->first ) & 0xFF;
2769
2770         if (isMHW2)
2771         {
2772                 u_char *data = (u_char*) event_data;
2773                 data[2] = itTitle->second.mhw2_mjd_hi;
2774                 data[3] = itTitle->second.mhw2_mjd_lo;
2775                 data[4] = itTitle->second.mhw2_hours;
2776                 data[5] = itTitle->second.mhw2_minutes;
2777                 data[6] = itTitle->second.mhw2_seconds;
2778                 timeMHW2DVB( HILO(itTitle->second.mhw2_duration), data+7 );
2779         }
2780         else
2781         {
2782                 timeMHW2DVB( itTitle->second.dh.day, itTitle->second.dh.hours, itTitle->second.ms.minutes,
2783                 (u_char *) event_data + 2 );
2784                 timeMHW2DVB( HILO(itTitle->second.duration), (u_char *) event_data+7 );
2785         }
2786
2787         event_data->running_status = 0;
2788         event_data->free_CA_mode = 0;
2789         int descr_ll = EIT_SHORT_EVENT_DESCRIPTOR_SIZE + 1 + prog_title_length;
2790
2791         eit_short_event_descriptor_struct *short_event_descriptor =
2792                 (eit_short_event_descriptor_struct *) ( (u_char *) event_data + EIT_LOOP_SIZE);
2793         short_event_descriptor->descriptor_tag = EIT_SHORT_EVENT_DESCRIPTOR;
2794         short_event_descriptor->descriptor_length = EIT_SHORT_EVENT_DESCRIPTOR_SIZE +
2795                 prog_title_length - 1;
2796         short_event_descriptor->language_code_1 = 'e';
2797         short_event_descriptor->language_code_2 = 'n';
2798         short_event_descriptor->language_code_3 = 'g';
2799         short_event_descriptor->event_name_length = prog_title_length;
2800         u_char *event_name = (u_char *) short_event_descriptor + EIT_SHORT_EVENT_DESCRIPTOR_SIZE;
2801         memcpy(event_name, prog_title.c_str(), prog_title_length);
2802
2803         // Set text length
2804         event_name[prog_title_length] = 0;
2805
2806         if ( sumText.length() > 0 )
2807         // There is summary info
2808         {
2809                 unsigned int sum_length = sumText.length();
2810                 if ( sum_length + short_event_descriptor->descriptor_length <= 0xff )
2811                 // Store summary in short event descriptor
2812                 {
2813                         // Increase all relevant lengths
2814                         event_name[prog_title_length] = sum_length;
2815                         short_event_descriptor->descriptor_length += sum_length;
2816                         packet_length += sum_length;
2817                         descr_ll += sum_length;
2818                         sumText.copy( (char *) event_name+prog_title_length+1, sum_length );
2819                 }
2820                 else
2821                 // Store summary in extended event descriptors
2822                 {
2823                         int remaining_sum_length = sumText.length();
2824                         int nbr_descr = int(remaining_sum_length/247) + 1;
2825                         for ( int i=0; i < nbr_descr; i++)
2826                         // Loop once per extended event descriptor
2827                         {
2828                                 eit_extended_descriptor_struct *ext_event_descriptor = (eit_extended_descriptor_struct *) (data + packet_length);
2829                                 sum_length = remaining_sum_length > 247 ? 247 : remaining_sum_length;
2830                                 remaining_sum_length -= sum_length;
2831                                 packet_length += 8 + sum_length;
2832                                 descr_ll += 8 + sum_length;
2833
2834                                 ext_event_descriptor->descriptor_tag = EIT_EXTENDED_EVENT_DESCRIPOR;
2835                                 ext_event_descriptor->descriptor_length = sum_length + 6;
2836                                 ext_event_descriptor->descriptor_number = i;
2837                                 ext_event_descriptor->last_descriptor_number = nbr_descr - 1;
2838                                 ext_event_descriptor->iso_639_2_language_code_1 = 'e';
2839                                 ext_event_descriptor->iso_639_2_language_code_2 = 'n';
2840                                 ext_event_descriptor->iso_639_2_language_code_3 = 'g';
2841                                 u_char *the_text = (u_char *) ext_event_descriptor + 8;
2842                                 the_text[-2] = 0;
2843                                 the_text[-1] = sum_length;
2844                                 sumText.copy( (char *) the_text, sum_length, sumText.length() - sum_length - remaining_sum_length );
2845                         }
2846                 }
2847         }
2848
2849         if (!isMHW2)
2850         {
2851                 // Add content descriptor
2852                 u_char *descriptor = (u_char *) data + packet_length;
2853                 packet_length += 4;
2854                 descr_ll += 4;
2855
2856                 int content_id = 0;
2857                 std::string content_descr = (char *) delimitName( m_themes[itTitle->second.theme_id].name, name, 15 );
2858                 if ( content_descr.find( "FILM" ) != std::string::npos )
2859                         content_id = 0x10;
2860                 else if ( content_descr.find( "SPORT" ) != std::string::npos )
2861                         content_id = 0x40;
2862
2863                 descriptor[0] = 0x54;
2864                 descriptor[1] = 2;
2865                 descriptor[2] = content_id;
2866                 descriptor[3] = 0;
2867         }
2868
2869         event_data->descriptors_loop_length_hi = (descr_ll & 0xf00)>>8;
2870         event_data->descriptors_loop_length_lo = (descr_ll & 0xff);
2871
2872         packet->section_length_hi =  ((packet_length - 3)&0xf00)>>8;
2873         packet->section_length_lo =  (packet_length - 3)&0xff;
2874
2875         // Feed the data to eEPGCache::sectionRead()
2876         cache->sectionRead( data, MHW, this );
2877 }
2878
2879 void eEPGCache::channel_data::startTimeout(int msec)
2880 {
2881         m_MHWTimeoutTimer.start(msec,true);
2882         m_MHWTimeoutet=false;
2883 }
2884
2885 void eEPGCache::channel_data::startMHWReader(__u16 pid, __u8 tid)
2886 {
2887         m_MHWFilterMask.pid = pid;
2888         m_MHWFilterMask.data[0] = tid;
2889         m_MHWReader->start(m_MHWFilterMask);
2890 //      eDebug("start 0x%02x 0x%02x", pid, tid);
2891 }
2892
2893 void eEPGCache::channel_data::startMHWReader2(__u16 pid, __u8 tid, int ext)
2894 {
2895         m_MHWFilterMask2.pid = pid;
2896         m_MHWFilterMask2.data[0] = tid;
2897         if (ext != -1)
2898         {
2899                 m_MHWFilterMask2.data[1] = ext;
2900                 m_MHWFilterMask2.mask[1] = 0xFF;
2901 //              eDebug("start 0x%03x 0x%02x 0x%02x", pid, tid, ext);
2902         }
2903         else
2904         {
2905                 m_MHWFilterMask2.data[1] = 0;
2906                 m_MHWFilterMask2.mask[1] = 0;
2907 //              eDebug("start 0x%02x 0x%02x", pid, tid);
2908         }
2909         m_MHWReader2->start(m_MHWFilterMask2);
2910 }
2911
2912 void eEPGCache::channel_data::readMHWData(const __u8 *data)
2913 {
2914         if ( m_MHWReader2 )
2915                 m_MHWReader2->stop();
2916
2917         if ( state > 1 || // aborted
2918                 // have si data.. so we dont read mhw data
2919                 (haveData & (SCHEDULE|SCHEDULE_OTHER)) )
2920         {
2921                 eDebug("[EPGC] mhw aborted %d", state);
2922         }
2923         else if (m_MHWFilterMask.pid == 0xD3 && m_MHWFilterMask.data[0] == 0x91)
2924         // Channels table
2925         {
2926                 int len = ((data[1]&0xf)<<8) + data[2] - 1;
2927                 int record_size = sizeof( mhw_channel_name_t );
2928                 int nbr_records = int (len/record_size);
2929
2930                 m_channels.resize(nbr_records);
2931                 for ( int i = 0; i < nbr_records; i++ )
2932                 {
2933                         mhw_channel_name_t *channel = (mhw_channel_name_t*) &data[4 + i*record_size];
2934                         m_channels[i]=*channel;
2935                 }
2936                 haveData |= MHW;
2937
2938                 eDebug("[EPGC] mhw %d channels found", m_channels.size());
2939
2940                 // Channels table has been read, start reading the themes table.
2941                 startMHWReader(0xD3, 0x92);
2942                 return;
2943         }
2944         else if (m_MHWFilterMask.pid == 0xD3 && m_MHWFilterMask.data[0] == 0x92)
2945         // Themes table
2946         {
2947                 int len = ((data[1]&0xf)<<8) + data[2] - 16;
2948                 int record_size = sizeof( mhw_theme_name_t );
2949                 int nbr_records = int (len/record_size);
2950                 int idx_ptr = 0;
2951                 __u8 next_idx = (__u8) *(data + 3 + idx_ptr);
2952                 __u8 idx = 0;
2953                 __u8 sub_idx = 0;
2954                 for ( int i = 0; i < nbr_records; i++ )
2955                 {
2956                         mhw_theme_name_t *theme = (mhw_theme_name_t*) &data[19 + i*record_size];
2957                         if ( i >= next_idx )
2958                         {
2959                                 idx = (idx_ptr<<4);
2960                                 idx_ptr++;
2961                                 next_idx = (__u8) *(data + 3 + idx_ptr);
2962                                 sub_idx = 0;
2963                         }
2964                         else
2965                                 sub_idx++;
2966
2967                         m_themes[idx+sub_idx] = *theme;
2968                 }
2969                 eDebug("[EPGC] mhw %d themes found", m_themes.size());
2970                 // Themes table has been read, start reading the titles table.
2971                 startMHWReader(0xD2, 0x90);
2972                 startTimeout(4000);
2973                 return;
2974         }
2975         else if (m_MHWFilterMask.pid == 0xD2 && m_MHWFilterMask.data[0] == 0x90)
2976         // Titles table
2977         {
2978                 mhw_title_t *title = (mhw_title_t*) data;
2979
2980                 if ( title->channel_id == 0xFF )        // Separator
2981                         return; // Continue reading of the current table.
2982                 else
2983                 {
2984                         // Create unique key per title
2985                         __u32 title_id = ((title->channel_id)<<16)|((title->dh.day)<<13)|((title->dh.hours)<<8)|
2986                                 (title->ms.minutes);
2987                         __u32 program_id = ((title->program_id_hi)<<24)|((title->program_id_mh)<<16)|
2988                                 ((title->program_id_ml)<<8)|(title->program_id_lo);
2989
2990                         if ( m_titles.find( title_id ) == m_titles.end() )
2991                         {
2992                                 startTimeout(4000);
2993                                 title->mhw2_mjd_hi = 0;
2994                                 title->mhw2_mjd_lo = 0;
2995                                 title->mhw2_duration_hi = 0;
2996                                 title->mhw2_duration_lo = 0;
2997                                 m_titles[ title_id ] = *title;
2998                                 if ( (title->ms.summary_available) && (m_program_ids.find(program_id) == m_program_ids.end()) )
2999                                         // program_ids will be used to gather summaries.
3000                                         m_program_ids.insert(std::pair<__u32,__u32>(program_id,title_id));
3001                                 return; // Continue reading of the current table.
3002                         }
3003                         else if (!checkTimeout())
3004                                 return;
3005                 }
3006                 if ( !m_program_ids.empty())
3007                 {
3008                         // Titles table has been read, there are summaries to read.
3009                         // Start reading summaries, store corresponding titles on the fly.
3010                         startMHWReader(0xD3, 0x90);
3011                         eDebug("[EPGC] mhw %d titles(%d with summary) found",
3012                                 m_titles.size(),
3013                                 m_program_ids.size());
3014                         startTimeout(4000);
3015                         return;
3016                 }
3017         }
3018         else if (m_MHWFilterMask.pid == 0xD3 && m_MHWFilterMask.data[0] == 0x90)
3019         // Summaries table
3020         {
3021                 mhw_summary_t *summary = (mhw_summary_t*) data;
3022
3023                 // Create unique key per record
3024                 __u32 program_id = ((summary->program_id_hi)<<24)|((summary->program_id_mh)<<16)|
3025                         ((summary->program_id_ml)<<8)|(summary->program_id_lo);
3026                 int len = ((data[1]&0xf)<<8) + data[2];
3027
3028                 // ugly workaround to convert const __u8* to char*
3029                 char *tmp=0;
3030                 memcpy(&tmp, &data, sizeof(void*));
3031                 tmp[len+3] = 0; // Terminate as a string.
3032
3033                 std::multimap<__u32, __u32>::iterator itProgid( m_program_ids.find( program_id ) );
3034                 if ( itProgid == m_program_ids.end() )
3035                 { /*    This part is to prevent to looping forever if some summaries are not received yet.
3036                         There is a timeout of 4 sec. after the last successfully read summary. */
3037                         if (!m_program_ids.empty() && !checkTimeout())
3038                                 return; // Continue reading of the current table.
3039                 }
3040                 else
3041                 {
3042                         std::string the_text = (char *) (data + 11 + summary->nb_replays * 7);
3043
3044                         unsigned int pos=0;
3045                         while((pos = the_text.find("\r\n")) != std::string::npos)
3046                                 the_text.replace(pos, 2, " ");
3047
3048                         // Find corresponding title, store title and summary in epgcache.
3049                         std::map<__u32, mhw_title_t>::iterator itTitle( m_titles.find( itProgid->second ) );
3050                         if ( itTitle != m_titles.end() )
3051                         {
3052                                 startTimeout(4000);
3053                                 storeTitle( itTitle, the_text, data );
3054                                 m_titles.erase( itTitle );
3055                         }
3056                         m_program_ids.erase( itProgid );
3057                         if ( !m_program_ids.empty() )
3058                                 return; // Continue reading of the current table.
3059                 }
3060         }
3061         eDebug("[EPGC] mhw finished(%ld) %d summaries not found",
3062                 eDVBLocalTimeHandler::getInstance()->nowTime(),
3063                 m_program_ids.size());
3064         // Summaries have been read, titles that have summaries have been stored.
3065         // Now store titles that do not have summaries.
3066         for (std::map<__u32, mhw_title_t>::iterator itTitle(m_titles.begin()); itTitle != m_titles.end(); itTitle++)
3067                 storeTitle( itTitle, "", data );
3068         isRunning &= ~MHW;
3069         m_MHWConn=0;
3070         if ( m_MHWReader )
3071                 m_MHWReader->stop();
3072         if (haveData)
3073                 finishEPG();
3074 }
3075
3076 void eEPGCache::channel_data::readMHWData2(const __u8 *data)
3077 {
3078         int dataLen = (((data[1]&0xf) << 8) | data[2]) + 3;
3079
3080         if ( m_MHWReader )
3081                 m_MHWReader->stop();
3082
3083         if ( state > 1 || // aborted
3084                 // have si data.. so we dont read mhw data
3085                 (haveData & (eEPGCache::SCHEDULE|eEPGCache::SCHEDULE_OTHER)) )
3086         {
3087                 eDebug("[EPGC] mhw2 aborted %d", state);
3088         }
3089         else if (m_MHWFilterMask2.pid == 0x231 && m_MHWFilterMask2.data[0] == 0xC8 && m_MHWFilterMask2.data[1] == 0)
3090         // Channels table
3091         {
3092                 int num_channels = data[119];
3093                 m_channels.resize(num_channels);
3094                 if(dataLen > 119)
3095                 {
3096                         int ptr = 120 + 8 * num_channels;
3097                         if( dataLen > ptr )
3098                         {
3099                                 for( int chid = 0; chid < num_channels; ++chid )
3100                                 {
3101                                         ptr += ( data[ptr] & 0x0f ) + 1;
3102                                         if( dataLen < ptr )
3103                                                 goto abort;
3104                                 }
3105                         }
3106                         else
3107                                 goto abort;
3108                 }
3109                 else
3110                         goto abort;
3111                 // data seems consistent...
3112                 const __u8 *tmp = data+120;
3113                 for (int i=0; i < num_channels; ++i)
3114                 {
3115                         mhw_channel_name_t channel;
3116                         channel.network_id_hi = *(tmp++);
3117                         channel.network_id_lo = *(tmp++);
3118                         channel.transport_stream_id_hi = *(tmp++);
3119                         channel.transport_stream_id_lo = *(tmp++);
3120                         channel.channel_id_hi = *(tmp++);
3121                         channel.channel_id_lo = *(tmp++);
3122                         m_channels[i]=channel;
3123                         tmp+=2;
3124                 }
3125                 for (int i=0; i < num_channels; ++i)
3126                 {
3127                         mhw_channel_name_t &channel = m_channels[i];
3128                         int channel_name_len=*(tmp++)&0x0f;
3129                         int x=0;
3130                         for (; x < channel_name_len; ++x)
3131                                 channel.name[x]=*(tmp++);
3132                         channel.name[x+1]=0;
3133                 }
3134                 haveData |= MHW;
3135                 eDebug("[EPGC] mhw2 %d channels found", m_channels.size());
3136         }
3137         else if (m_MHWFilterMask2.pid == 0x231 && m_MHWFilterMask2.data[0] == 0xC8 && m_MHWFilterMask2.data[1] == 1)
3138         {
3139                 // Themes table
3140                 eDebug("[EPGC] mhw2 themes nyi");
3141         }
3142         else if (m_MHWFilterMask2.pid == 0x234 && m_MHWFilterMask2.data[0] == 0xe6)
3143         // Titles table
3144         {
3145                 int pos=18;
3146                 bool valid=true;
3147                 int len = ((data[1]&0xf)<<8) + data[2] - 16;
3148                 bool finish=false;
3149                 if(data[dataLen-1] != 0xff)
3150                         return;
3151                 while( pos < dataLen )
3152                 {
3153                         valid = false;
3154                         pos += 7;
3155                         if( pos < dataLen )
3156                         {
3157                                 pos += 3;
3158                                 if( pos < dataLen )
3159                                 {
3160                                         if( data[pos] > 0xc0 )
3161                                         {
3162                                                 pos += ( data[pos] - 0xc0 );
3163                                                 pos += 4;
3164                                                 if( pos < dataLen )
3165                                                 {
3166                                                         if( data[pos] == 0xff )
3167                                                         {
3168                                                                 ++pos;
3169                                                                 valid = true;
3170                                                         }
3171                                                 }
3172                                         }
3173                                 }
3174                         }
3175                         if( !valid )
3176                         {
3177                                 if (checkTimeout())
3178                                         goto start_summary;
3179                                 return;
3180                         }
3181                 }
3182                 // data seems consistent...
3183                 mhw_title_t title;
3184                 pos = 18;
3185                 while (pos < len)
3186                 {
3187                         title.channel_id = data[pos]+1;
3188                         title.program_id_ml = data[pos+1];
3189                         title.program_id_lo = data[pos+2];
3190                         title.mhw2_mjd_hi = data[pos+3];
3191                         title.mhw2_mjd_lo = data[pos+4];
3192                         title.mhw2_hours = data[pos+5];
3193                         title.mhw2_minutes = data[pos+6];
3194                         title.mhw2_seconds = data[pos+7];
3195                         int duration = ((data[pos+8] << 8)|data[pos+9]) >> 4;
3196                         title.mhw2_duration_hi = (duration&0xFF00) >> 8;
3197                         title.mhw2_duration_lo = duration&0xFF;
3198                         __u8 slen = data[pos+10] & 0x3f;
3199                         __u8 *dest = ((__u8*)title.title)-4;
3200                         memcpy(dest, &data[pos+11], slen>33 ? 33 : slen);
3201                         memset(dest+slen, 0, 33-slen);
3202                         pos += 11 + slen;
3203 //                      not used theme id (data[7] & 0x3f) + (data[pos] & 0x3f);
3204                         __u32 summary_id = (data[pos+1] << 8) | data[pos+2];
3205
3206                         // Create unique key per title
3207                         __u32 title_id = (title.channel_id<<16) | (title.program_id_ml<<8) | title.program_id_lo;
3208
3209                         pos += 4;
3210
3211                         std::map<__u32, mhw_title_t>::iterator it = m_titles.find( title_id );
3212                         if ( it == m_titles.end() )
3213                         {
3214                                 startTimeout(5000);
3215                                 m_titles[ title_id ] = title;
3216                                 if (summary_id != 0xFFFF)
3217                                 {
3218                                         bool add=true;
3219                                         std::multimap<__u32, __u32>::iterator it(m_program_ids.lower_bound(summary_id));
3220                                         while (it != m_program_ids.end() && it->first == summary_id)
3221                                         {
3222                                                 if (it->second == title_id) {
3223                                                         add=false;
3224                                                         break;
3225                                                 }
3226                                                 ++it;
3227                                         }
3228                                         if (add)
3229                                                 m_program_ids.insert(std::pair<__u32,__u32>(summary_id,title_id));
3230                                 }
3231                         }
3232                         else
3233                         {
3234                                 if ( !checkTimeout() )
3235                                         continue;       // Continue reading of the current table.
3236                                 finish=true;
3237                                 break;
3238                         }
3239                 }
3240 start_summary:
3241                 if (finish)
3242                 {
3243                         eDebug("[EPGC] mhw2 %d titles(%d with summary) found", m_titles.size(), m_program_ids.size());
3244                         if (!m_program_ids.empty())
3245                         {
3246                                 // Titles table has been read, there are summaries to read.
3247                                 // Start reading summaries, store corresponding titles on the fly.
3248                                 startMHWReader2(0x236, 0x96);
3249                                 startTimeout(15000);
3250                                 return;
3251                         }
3252                 }
3253                 else
3254                         return;
3255         }
3256         else if (m_MHWFilterMask2.pid == 0x236 && m_MHWFilterMask2.data[0] == 0x96)
3257         // Summaries table
3258         {
3259                 if (!checkTimeout())
3260                 {
3261                         int len, loop, pos, lenline;
3262                         bool valid;
3263                         valid = true;
3264                         if( dataLen > 15 )
3265                         {
3266                                 loop = data[14];
3267                                 pos = 15 + loop;
3268                                 if( dataLen > pos )
3269                                 {
3270                                         loop = data[pos] & 0x0f;
3271                                         pos += 1;
3272                                         if( dataLen > pos )
3273                                         {
3274                                                 len = 0;
3275                                                 for( ; loop > 0; --loop )
3276                                                 {
3277                                                         if( dataLen > (pos+len) )
3278                                                         {
3279                                                                 lenline = data[pos+len];
3280                                                                 len += lenline + 1;
3281                                                         }
3282                                                         else
3283                                                                 valid=false;
3284                                                 }
3285                                         }
3286                                 }
3287                         }
3288                         else
3289                                 return;  // continue reading
3290                         if (valid)
3291                         {
3292                                 // data seems consistent...
3293                                 __u32 summary_id = (data[3]<<8)|data[4];
3294
3295                                 // ugly workaround to convert const __u8* to char*
3296                                 char *tmp=0;
3297                                 memcpy(&tmp, &data, sizeof(void*));
3298
3299                                 len = 0;
3300                                 loop = data[14];
3301                                 pos = 15 + loop;
3302                                 loop = tmp[pos] & 0x0f;
3303                                 pos += 1;
3304                                 for( ; loop > 0; loop -- )
3305                                 {
3306                                         lenline = tmp[pos+len];
3307                                         tmp[pos+len] = ' ';
3308                                         len += lenline + 1;
3309                                 }
3310                                 if( len > 0 )
3311                                     tmp[pos+len] = 0;
3312                                 else
3313                                         tmp[pos+1] = 0;
3314
3315                                 std::multimap<__u32, __u32>::iterator itProgId( m_program_ids.lower_bound(summary_id) );
3316                                 if ( itProgId == m_program_ids.end() || itProgId->first != summary_id)
3317                                 { /*    This part is to prevent to looping forever if some summaries are not received yet.
3318                                         There is a timeout of 4 sec. after the last successfully read summary. */
3319                                         if ( !m_program_ids.empty() )
3320                                                 return; // Continue reading of the current table.
3321                                 }
3322                                 else
3323                                 {
3324                                         startTimeout(15000);
3325                                         std::string the_text = (char *) (data + pos + 1);
3326
3327                                         while( itProgId != m_program_ids.end() && itProgId->first == summary_id )
3328                                         {
3329                                                 // Find corresponding title, store title and summary in epgcache.
3330                                                 std::map<__u32, mhw_title_t>::iterator itTitle( m_titles.find( itProgId->second ) );
3331                                                 if ( itTitle != m_titles.end() )
3332                                                 {
3333                                                         storeTitle( itTitle, the_text, data );
3334                                                         m_titles.erase( itTitle );
3335                                                 }
3336                                                 m_program_ids.erase( itProgId++ );
3337                                         }
3338                                         if ( !m_program_ids.empty() )
3339                                                 return; // Continue reading of the current table.
3340                                 }
3341                         }
3342                         else
3343                                 return;  // continue reading
3344                 }
3345         }
3346         if (isRunning & eEPGCache::MHW)
3347         {
3348                 if ( m_MHWFilterMask2.pid == 0x231 && m_MHWFilterMask2.data[0] == 0xC8 && m_MHWFilterMask2.data[1] == 0)
3349                 {
3350                         // Channels table has been read, start reading the themes table.
3351                         startMHWReader2(0x231, 0xC8, 1);
3352                         return;
3353                 }
3354                 else if ( m_MHWFilterMask2.pid == 0x231 && m_MHWFilterMask2.data[0] == 0xC8 && m_MHWFilterMask2.data[1] == 1)
3355                 {
3356                         // Themes table has been read, start reading the titles table.
3357                         startMHWReader2(0x234, 0xe6);
3358                         return;
3359                 }
3360                 else
3361                 {
3362                         // Summaries have been read, titles that have summaries have been stored.
3363                         // Now store titles that do not have summaries.
3364                         for (std::map<__u32, mhw_title_t>::iterator itTitle(m_titles.begin()); itTitle != m_titles.end(); itTitle++)
3365                                 storeTitle( itTitle, "", data );
3366                         eDebug("[EPGC] mhw2 finished(%ld) %d summaries not found",
3367                                 eDVBLocalTimeHandler::getInstance()->nowTime(),
3368                                 m_program_ids.size());
3369                 }
3370         }
3371 abort:
3372         isRunning &= ~MHW;
3373         m_MHWConn2=0;
3374         if ( m_MHWReader2 )
3375                 m_MHWReader2->stop();
3376         if (haveData)
3377                 finishEPG();
3378 }
3379 #endif