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