enable wrap around in epglist and servicelist
[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         singleLock s(cache_lock);
705         FILE *f = fopen("/hdd/epg.dat", "r");
706         if (f)
707         {
708                 int size=0;
709                 int cnt=0;
710 #if 0
711                 unsigned char md5_saved[16];
712                 unsigned char md5[16];
713                 bool md5ok=false;
714
715                 if (!md5_file("/hdd/epg.dat", 1, md5))
716                 {
717                         FILE *f = fopen("/hdd/epg.dat.md5", "r");
718                         if (f)
719                         {
720                                 fread( md5_saved, 16, 1, f);
721                                 fclose(f);
722                                 if ( !memcmp(md5_saved, md5, 16) )
723                                         md5ok=true;
724                         }
725                 }
726                 if ( md5ok )
727 #endif
728                 {
729                         unsigned int magic=0;
730                         fread( &magic, sizeof(int), 1, f);
731                         if (magic != 0x98765432)
732                         {
733                                 eDebug("epg file has incorrect byte order.. dont read it");
734                                 fclose(f);
735                                 return;
736                         }
737                         char text1[13];
738                         fread( text1, 13, 1, f);
739                         if ( !strncmp( text1, "ENIGMA_EPG_V5", 13) )
740                         {
741                                 fread( &size, sizeof(int), 1, f);
742                                 while(size--)
743                                 {
744                                         uniqueEPGKey key;
745                                         eventMap evMap;
746                                         timeMap tmMap;
747                                         int size=0;
748                                         fread( &key, sizeof(uniqueEPGKey), 1, f);
749                                         fread( &size, sizeof(int), 1, f);
750                                         while(size--)
751                                         {
752                                                 __u8 len=0;
753                                                 __u8 type=0;
754                                                 eventData *event=0;
755                                                 fread( &type, sizeof(__u8), 1, f);
756                                                 fread( &len, sizeof(__u8), 1, f);
757                                                 event = new eventData(0, len, type);
758                                                 event->EITdata = new __u8[len];
759                                                 eventData::CacheSize+=len;
760                                                 fread( event->EITdata, len, 1, f);
761                                                 evMap[ event->getEventID() ]=event;
762                                                 tmMap[ event->getStartTime() ]=event;
763                                                 ++cnt;
764                                         }
765                                         eventDB[key]=std::pair<eventMap,timeMap>(evMap,tmMap);
766                                 }
767                                 eventData::load(f);
768                                 eDebug("%d events read from /hdd/epg.dat", cnt);
769 #ifdef ENABLE_PRIVATE_EPG
770                                 char text2[11];
771                                 fread( text2, 11, 1, f);
772                                 if ( !strncmp( text2, "PRIVATE_EPG", 11) )
773                                 {
774                                         size=0;
775                                         fread( &size, sizeof(int), 1, f);
776                                         while(size--)
777                                         {
778                                                 int size=0;
779                                                 uniqueEPGKey key;
780                                                 fread( &key, sizeof(uniqueEPGKey), 1, f);
781                                                 fread( &size, sizeof(int), 1, f);
782                                                 while(size--)
783                                                 {
784                                                         int size;
785                                                         int content_id;
786                                                         fread( &content_id, sizeof(int), 1, f);
787                                                         fread( &size, sizeof(int), 1, f);
788                                                         while(size--)
789                                                         {
790                                                                 time_t time1, time2;
791                                                                 __u16 event_id;
792                                                                 fread( &time1, sizeof(time_t), 1, f);
793                                                                 fread( &time2, sizeof(time_t), 1, f);
794                                                                 fread( &event_id, sizeof(__u16), 1, f);
795                                                                 content_time_tables[key][content_id][time1]=std::pair<time_t, __u16>(time2, event_id);
796                                                         }
797                                                 }
798                                         }
799                                 }
800 #endif // ENABLE_PRIVATE_EPG
801                         }
802                         else
803                                 eDebug("[EPGC] don't read old epg database");
804                         fclose(f);
805                 }
806         }
807 }
808
809 void eEPGCache::save()
810 {
811         struct statfs s;
812         off64_t tmp;
813         if (statfs("/hdd", &s)<0)
814                 tmp=0;
815         else
816         {
817                 tmp=s.f_blocks;
818                 tmp*=s.f_bsize;
819         }
820
821         // prevent writes to builtin flash
822         if ( tmp < 1024*1024*50 ) // storage size < 50MB
823                 return;
824
825         // check for enough free space on storage
826         tmp=s.f_bfree;
827         tmp*=s.f_bsize;
828         if ( tmp < (eventData::CacheSize*12)/10 ) // 20% overhead
829                 return;
830
831         FILE *f = fopen("/hdd/epg.dat", "w");
832         int cnt=0;
833         if ( f )
834         {
835                 unsigned int magic = 0x98765432;
836                 fwrite( &magic, sizeof(int), 1, f);
837                 const char *text = "ENIGMA_EPG_V5";
838                 fwrite( text, 13, 1, f );
839                 int size = eventDB.size();
840                 fwrite( &size, sizeof(int), 1, f );
841                 for (eventCache::iterator service_it(eventDB.begin()); service_it != eventDB.end(); ++service_it)
842                 {
843                         timeMap &timemap = service_it->second.second;
844                         fwrite( &service_it->first, sizeof(uniqueEPGKey), 1, f);
845                         size = timemap.size();
846                         fwrite( &size, sizeof(int), 1, f);
847                         for (timeMap::iterator time_it(timemap.begin()); time_it != timemap.end(); ++time_it)
848                         {
849                                 __u8 len = time_it->second->ByteSize;
850                                 fwrite( &time_it->second->type, sizeof(__u8), 1, f );
851                                 fwrite( &len, sizeof(__u8), 1, f);
852                                 fwrite( time_it->second->EITdata, len, 1, f);
853                                 ++cnt;
854                         }
855                 }
856                 eDebug("%d events written to /hdd/epg.dat", cnt);
857                 eventData::save(f);
858 #ifdef ENABLE_PRIVATE_EPG
859                 const char* text3 = "PRIVATE_EPG";
860                 fwrite( text3, 11, 1, f );
861                 size = content_time_tables.size();
862                 fwrite( &size, sizeof(int), 1, f);
863                 for (contentMaps::iterator a = content_time_tables.begin(); a != content_time_tables.end(); ++a)
864                 {
865                         contentMap &content_time_table = a->second;
866                         fwrite( &a->first, sizeof(uniqueEPGKey), 1, f);
867                         int size = content_time_table.size();
868                         fwrite( &size, sizeof(int), 1, f);
869                         for (contentMap::iterator i = content_time_table.begin(); i != content_time_table.end(); ++i )
870                         {
871                                 int size = i->second.size();
872                                 fwrite( &i->first, sizeof(int), 1, f);
873                                 fwrite( &size, sizeof(int), 1, f);
874                                 for ( contentTimeMap::iterator it(i->second.begin());
875                                         it != i->second.end(); ++it )
876                                 {
877                                         fwrite( &it->first, sizeof(time_t), 1, f);
878                                         fwrite( &it->second.first, sizeof(time_t), 1, f);
879                                         fwrite( &it->second.second, sizeof(__u16), 1, f);
880                                 }
881                         }
882                 }
883 #endif
884                 fclose(f);
885 #if 0
886                 unsigned char md5[16];
887                 if (!md5_file("/hdd/epg.dat", 1, md5))
888                 {
889                         FILE *f = fopen("/hdd/epg.dat.md5", "w");
890                         if (f)
891                         {
892                                 fwrite( md5, 16, 1, f);
893                                 fclose(f);
894                         }
895                 }
896 #endif
897         }
898 }
899
900 eEPGCache::channel_data::channel_data(eEPGCache *ml)
901         :cache(ml)
902         ,abortTimer(ml), zapTimer(ml)
903         ,state(0), isRunning(0), haveData(0), can_delete(1)
904 {
905         CONNECT(zapTimer.timeout, eEPGCache::channel_data::startEPG);
906         CONNECT(abortTimer.timeout, eEPGCache::channel_data::abortNonAvail);
907 }
908
909 bool eEPGCache::channel_data::finishEPG()
910 {
911         if (!isRunning)  // epg ready
912         {
913                 eDebug("[EPGC] stop caching events(%d)", time(0)+eDVBLocalTimeHandler::getInstance()->difference());
914                 zapTimer.start(UPDATE_INTERVAL, 1);
915                 eDebug("[EPGC] next update in %i min", UPDATE_INTERVAL / 60000);
916                 for (int i=0; i < 3; ++i)
917                 {
918                         seenSections[i].clear();
919                         calcedSections[i].clear();
920                 }
921                 singleLock l(cache->cache_lock);
922                 cache->channelLastUpdated[channel->getChannelID()] = time(0)+eDVBLocalTimeHandler::getInstance()->difference();
923 #ifdef ENABLE_PRIVATE_EPG
924                 if (seenPrivateSections.empty())
925 #endif
926                 can_delete=1;
927                 return true;
928         }
929         return false;
930 }
931
932 void eEPGCache::channel_data::startEPG()
933 {
934         eDebug("[EPGC] start caching events(%d)", eDVBLocalTimeHandler::getInstance()->difference()+time(0));
935         state=0;
936         haveData=0;
937         can_delete=0;
938         for (int i=0; i < 3; ++i)
939         {
940                 seenSections[i].clear();
941                 calcedSections[i].clear();
942         }
943
944         eDVBSectionFilterMask mask;
945         memset(&mask, 0, sizeof(mask));
946         mask.pid = 0x12;
947         mask.flags = eDVBSectionFilterMask::rfCRC;
948
949         mask.data[0] = 0x4E;
950         mask.mask[0] = 0xFE;
951         m_NowNextReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_NowNextConn);
952         m_NowNextReader->start(mask);
953         isRunning |= NOWNEXT;
954
955         mask.data[0] = 0x50;
956         mask.mask[0] = 0xF0;
957         m_ScheduleReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_ScheduleConn);
958         m_ScheduleReader->start(mask);
959         isRunning |= SCHEDULE;
960
961         mask.data[0] = 0x60;
962         mask.mask[0] = 0xF0;
963         m_ScheduleOtherReader->connectRead(slot(*this, &eEPGCache::channel_data::readData), m_ScheduleOtherConn);
964         m_ScheduleOtherReader->start(mask);
965         isRunning |= SCHEDULE_OTHER;
966
967         abortTimer.start(7000,true);
968 }
969
970 void eEPGCache::channel_data::abortNonAvail()
971 {
972         if (!state)
973         {
974                 if ( !(haveData&eEPGCache::NOWNEXT) && (isRunning&eEPGCache::NOWNEXT) )
975                 {
976                         eDebug("[EPGC] abort non avail nownext reading");
977                         isRunning &= ~eEPGCache::NOWNEXT;
978                         m_NowNextReader->stop();
979                         m_NowNextConn=0;
980                 }
981                 if ( !(haveData&eEPGCache::SCHEDULE) && (isRunning&eEPGCache::SCHEDULE) )
982                 {
983                         eDebug("[EPGC] abort non avail schedule reading");
984                         isRunning &= ~SCHEDULE;
985                         m_ScheduleReader->stop();
986                         m_ScheduleConn=0;
987                 }
988                 if ( !(haveData&eEPGCache::SCHEDULE_OTHER) && (isRunning&eEPGCache::SCHEDULE_OTHER) )
989                 {
990                         eDebug("[EPGC] abort non avail schedule_other reading");
991                         isRunning &= ~SCHEDULE_OTHER;
992                         m_ScheduleOtherReader->stop();
993                         m_ScheduleOtherConn=0;
994                 }
995                 if ( isRunning )
996                         abortTimer.start(90000, true);
997                 else
998                 {
999                         ++state;
1000                         for (int i=0; i < 3; ++i)
1001                         {
1002                                 seenSections[i].clear();
1003                                 calcedSections[i].clear();
1004                         }
1005 #ifdef ENABLE_PRIVATE_EPG
1006                         if (seenPrivateSections.empty())
1007 #endif
1008                         can_delete=1;
1009                 }
1010         }
1011         ++state;
1012 }
1013
1014 void eEPGCache::channel_data::startChannel()
1015 {
1016         updateMap::iterator It = cache->channelLastUpdated.find( channel->getChannelID() );
1017
1018         int update = ( It != cache->channelLastUpdated.end() ? ( UPDATE_INTERVAL - ( (time(0)+eDVBLocalTimeHandler::getInstance()->difference()-It->second) * 1000 ) ) : ZAP_DELAY );
1019
1020         if (update < ZAP_DELAY)
1021                 update = ZAP_DELAY;
1022
1023         zapTimer.start(update, 1);
1024         if (update >= 60000)
1025                 eDebug("[EPGC] next update in %i min", update/60000);
1026         else if (update >= 1000)
1027                 eDebug("[EPGC] next update in %i sec", update/1000);
1028 }
1029
1030 void eEPGCache::channel_data::abortEPG()
1031 {
1032         for (int i=0; i < 3; ++i)
1033         {
1034                 seenSections[i].clear();
1035                 calcedSections[i].clear();
1036         }
1037         abortTimer.stop();
1038         zapTimer.stop();
1039         if (isRunning)
1040         {
1041                 eDebug("[EPGC] abort caching events !!");
1042                 if (isRunning & eEPGCache::SCHEDULE)
1043                 {
1044                         isRunning &= ~eEPGCache::SCHEDULE;
1045                         m_ScheduleReader->stop();
1046                         m_ScheduleConn=0;
1047                 }
1048                 if (isRunning & eEPGCache::NOWNEXT)
1049                 {
1050                         isRunning &= ~eEPGCache::NOWNEXT;
1051                         m_NowNextReader->stop();
1052                         m_NowNextConn=0;
1053                 }
1054                 if (isRunning & SCHEDULE_OTHER)
1055                 {
1056                         isRunning &= ~eEPGCache::SCHEDULE_OTHER;
1057                         m_ScheduleOtherReader->stop();
1058                         m_ScheduleOtherConn=0;
1059                 }
1060         }
1061 #ifdef ENABLE_PRIVATE_EPG
1062         if (m_PrivateReader)
1063                 m_PrivateReader->stop();
1064         if (m_PrivateConn)
1065                 m_PrivateConn=0;
1066 #endif
1067         can_delete=1;
1068 }
1069
1070 void eEPGCache::channel_data::readData( const __u8 *data)
1071 {
1072         if (!data)
1073                 eDebug("get Null pointer from section reader !!");
1074         else
1075         {
1076                 int source;
1077                 int map;
1078                 iDVBSectionReader *reader=NULL;
1079                 switch(data[0])
1080                 {
1081                         case 0x4E ... 0x4F:
1082                                 reader=m_NowNextReader;
1083                                 source=eEPGCache::NOWNEXT;
1084                                 map=0;
1085                                 break;
1086                         case 0x50 ... 0x5F:
1087                                 reader=m_ScheduleReader;
1088                                 source=eEPGCache::SCHEDULE;
1089                                 map=1;
1090                                 break;
1091                         case 0x60 ... 0x6F:
1092                                 reader=m_ScheduleOtherReader;
1093                                 source=eEPGCache::SCHEDULE_OTHER;
1094                                 map=2;
1095                                 break;
1096                         default:
1097                                 eDebug("[EPGC] unknown table_id !!!");
1098                                 return;
1099                 }
1100                 tidMap &seenSections = this->seenSections[map];
1101                 tidMap &calcedSections = this->calcedSections[map];
1102                 if ( state == 1 && calcedSections == seenSections || state > 1 )
1103                 {
1104                         eDebugNoNewLine("[EPGC] ");
1105                         switch (source)
1106                         {
1107                                 case eEPGCache::NOWNEXT:
1108                                         m_NowNextConn=0;
1109                                         eDebugNoNewLine("nownext");
1110                                         break;
1111                                 case eEPGCache::SCHEDULE:
1112                                         m_ScheduleConn=0;
1113                                         eDebugNoNewLine("schedule");
1114                                         break;
1115                                 case eEPGCache::SCHEDULE_OTHER:
1116                                         m_ScheduleOtherConn=0;
1117                                         eDebugNoNewLine("schedule other");
1118                                         break;
1119                                 default: eDebugNoNewLine("unknown");break;
1120                         }
1121                         eDebug(" finished(%d)", time(0)+eDVBLocalTimeHandler::getInstance()->difference());
1122                         if ( reader )
1123                                 reader->stop();
1124                         isRunning &= ~source;
1125                         if (!isRunning)
1126                                 finishEPG();
1127                 }
1128                 else
1129                 {
1130                         eit_t *eit = (eit_t*) data;
1131                         __u32 sectionNo = data[0] << 24;
1132                         sectionNo |= data[3] << 16;
1133                         sectionNo |= data[4] << 8;
1134                         sectionNo |= eit->section_number;
1135
1136                         tidMap::iterator it =
1137                                 seenSections.find(sectionNo);
1138
1139                         if ( it == seenSections.end() )
1140                         {
1141                                 seenSections.insert(sectionNo);
1142                                 calcedSections.insert(sectionNo);
1143                                 __u32 tmpval = sectionNo & 0xFFFFFF00;
1144                                 __u8 incr = source == NOWNEXT ? 1 : 8;
1145                                 for ( int i = 0; i <= eit->last_section_number; i+=incr )
1146                                 {
1147                                         if ( i == eit->section_number )
1148                                         {
1149                                                 for (int x=i; x <= eit->segment_last_section_number; ++x)
1150                                                         calcedSections.insert(tmpval|(x&0xFF));
1151                                         }
1152                                         else
1153                                                 calcedSections.insert(tmpval|(i&0xFF));
1154                                 }
1155                                 cache->sectionRead(data, source, this);
1156                         }
1157                 }
1158         }
1159 }
1160
1161 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, const eventData *&result, int direction)
1162 // if t == -1 we search the current event...
1163 {
1164         singleLock s(cache_lock);
1165         uniqueEPGKey key(service);
1166
1167         // check if EPG for this service is ready...
1168         eventCache::iterator It = eventDB.find( key );
1169         if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached ?
1170         {
1171                 if (t==-1)
1172                         t = time(0)+eDVBLocalTimeHandler::getInstance()->difference();
1173                 timeMap::iterator i = direction <= 0 ? It->second.second.lower_bound(t) :  // find > or equal
1174                         It->second.second.upper_bound(t); // just >
1175                 if ( i != It->second.second.end() )
1176                 {
1177                         if ( direction < 0 || (direction == 0 && i->second->getStartTime() > t) )
1178                         {
1179                                 timeMap::iterator x = i;
1180                                 --x;
1181                                 if ( x != It->second.second.end() )
1182                                 {
1183                                         time_t start_time = x->second->getStartTime();
1184                                         if (direction >= 0)
1185                                         {
1186                                                 if (t < start_time)
1187                                                         return -1;
1188                                                 if (t > (start_time+x->second->getDuration()))
1189                                                         return -1;
1190                                         }
1191                                         i = x;
1192                                 }
1193                                 else
1194                                         return -1;
1195                         }
1196                         result = i->second;
1197                         return 0;
1198                 }
1199         }
1200         return -1;
1201 }
1202
1203 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, const eit_event_struct *&result, int direction)
1204 {
1205         singleLock s(cache_lock);
1206         const eventData *data=0;
1207         RESULT ret = lookupEventTime(service, t, data, direction);
1208         if ( !ret && data )
1209                 result = data->get();
1210         return ret;
1211 }
1212
1213 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, Event *& result, int direction)
1214 {
1215         singleLock s(cache_lock);
1216         const eventData *data=0;
1217         RESULT ret = lookupEventTime(service, t, data, direction);
1218         if ( !ret && data )
1219                 result = new Event((uint8_t*)data->get());
1220         return ret;
1221 }
1222
1223 RESULT eEPGCache::lookupEventTime(const eServiceReference &service, time_t t, ePtr<eServiceEvent> &result, int direction)
1224 {
1225         singleLock s(cache_lock);
1226         const eventData *data=0;
1227         RESULT ret = lookupEventTime(service, t, data, direction);
1228         if ( !ret && data )
1229         {
1230                 Event ev((uint8_t*)data->get());
1231                 result = new eServiceEvent();
1232                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1233                 ret = result->parseFrom(&ev, (ref.getTransportStreamID().get()<<16)|ref.getOriginalNetworkID().get());
1234         }
1235         return ret;
1236 }
1237
1238 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eventData *&result )
1239 {
1240         singleLock s(cache_lock);
1241         uniqueEPGKey key( service );
1242
1243         eventCache::iterator It = eventDB.find( key );
1244         if ( It != eventDB.end() && !It->second.first.empty() ) // entrys cached?
1245         {
1246                 eventMap::iterator i( It->second.first.find( event_id ));
1247                 if ( i != It->second.first.end() )
1248                 {
1249                         result = i->second;
1250                         return 0;
1251                 }
1252                 else
1253                 {
1254                         result = 0;
1255                         eDebug("event %04x not found in epgcache", event_id);
1256                 }
1257         }
1258         return -1;
1259 }
1260
1261 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, const eit_event_struct *&result)
1262 {
1263         singleLock s(cache_lock);
1264         const eventData *data=0;
1265         RESULT ret = lookupEventId(service, event_id, data);
1266         if ( !ret && data )
1267                 result = data->get();
1268         return ret;
1269 }
1270
1271 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, Event *& result)
1272 {
1273         singleLock s(cache_lock);
1274         const eventData *data=0;
1275         RESULT ret = lookupEventId(service, event_id, data);
1276         if ( !ret && data )
1277                 result = new Event((uint8_t*)data->get());
1278         return ret;
1279 }
1280
1281 RESULT eEPGCache::lookupEventId(const eServiceReference &service, int event_id, ePtr<eServiceEvent> &result)
1282 {
1283         singleLock s(cache_lock);
1284         const eventData *data=0;
1285         RESULT ret = lookupEventId(service, event_id, data);
1286         if ( !ret && data )
1287         {
1288                 Event ev((uint8_t*)data->get());
1289                 result = new eServiceEvent();
1290                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1291                 ret = result->parseFrom(&ev, (ref.getTransportStreamID().get()<<16)|ref.getOriginalNetworkID().get());
1292         }
1293         return ret;
1294 }
1295
1296 RESULT eEPGCache::startTimeQuery(const eServiceReference &service, time_t begin, int minutes)
1297 {
1298         eventCache::iterator It = eventDB.find( service );
1299         if ( It != eventDB.end() && It->second.second.size() )
1300         {
1301                 m_timemap_end = minutes != -1 ? It->second.second.upper_bound(begin+minutes*60) : It->second.second.end();
1302                 if ( begin != -1 )
1303                 {
1304                         m_timemap_cursor = It->second.second.lower_bound(begin);
1305                         if ( m_timemap_cursor != It->second.second.end() )
1306                         {
1307                                 if ( m_timemap_cursor->second->getStartTime() != begin )
1308                                 {
1309                                         timeMap::iterator x = m_timemap_cursor;
1310                                         --x;
1311                                         if ( x != It->second.second.end() )
1312                                         {
1313                                                 time_t start_time = x->second->getStartTime();
1314                                                 if ( begin > start_time && begin < (start_time+x->second->getDuration()))
1315                                                         m_timemap_cursor = x;
1316                                         }
1317                                 }
1318                         }
1319                 }
1320                 else
1321                         m_timemap_cursor = It->second.second.begin();
1322                 const eServiceReferenceDVB &ref = (const eServiceReferenceDVB&)service;
1323                 currentQueryTsidOnid = (ref.getTransportStreamID().get()<<16) | ref.getOriginalNetworkID().get();
1324                 return 0;
1325         }
1326         return -1;
1327 }
1328
1329 RESULT eEPGCache::getNextTimeEntry(const eventData *& result)
1330 {
1331         if ( m_timemap_cursor != m_timemap_end )
1332         {
1333                 result = m_timemap_cursor++->second;
1334                 return 0;
1335         }
1336         return -1;
1337 }
1338
1339 RESULT eEPGCache::getNextTimeEntry(const eit_event_struct *&result)
1340 {
1341         if ( m_timemap_cursor != m_timemap_end )
1342         {
1343                 result = m_timemap_cursor++->second->get();
1344                 return 0;
1345         }
1346         return -1;
1347 }
1348
1349 RESULT eEPGCache::getNextTimeEntry(Event *&result)
1350 {
1351         if ( m_timemap_cursor != m_timemap_end )
1352         {
1353                 result = new Event((uint8_t*)m_timemap_cursor++->second->get());
1354                 return 0;
1355         }
1356         return -1;
1357 }
1358
1359 RESULT eEPGCache::getNextTimeEntry(ePtr<eServiceEvent> &result)
1360 {
1361         if ( m_timemap_cursor != m_timemap_end )
1362         {
1363                 Event ev((uint8_t*)m_timemap_cursor++->second->get());
1364                 result = new eServiceEvent();
1365                 return result->parseFrom(&ev, currentQueryTsidOnid);
1366         }
1367         return -1;
1368 }
1369
1370 void fillTuple(PyObject *tuple, char *argstring, int argcount, PyObject *service, ePtr<eServiceEvent> &ptr, PyObject *nowTime, PyObject *service_name )
1371 {
1372         PyObject *tmp=NULL;
1373         int pos=0;
1374         while(pos < argcount)
1375         {
1376                 bool inc_refcount=false;
1377                 switch(argstring[pos])
1378                 {
1379                         case 'I': // Event Id
1380                                 tmp = ptr ? PyLong_FromLong(ptr->getEventId()) : NULL;
1381                                 break;
1382                         case 'B': // Event Begin Time
1383                                 tmp = ptr ? PyLong_FromLong(ptr->getBeginTime()) : NULL;
1384                                 break;
1385                         case 'D': // Event Duration
1386                                 tmp = ptr ? PyLong_FromLong(ptr->getDuration()) : NULL;
1387                                 break;
1388                         case 'T': // Event Title
1389                                 tmp = ptr ? PyString_FromString(ptr->getEventName().c_str()) : NULL;
1390                                 break;
1391                         case 'S': // Event Short Description
1392                                 tmp = ptr ? PyString_FromString(ptr->getShortDescription().c_str()) : NULL;
1393                                 break;
1394                         case 'E': // Event Extended Description
1395                                 tmp = ptr ? PyString_FromString(ptr->getExtendedDescription().c_str()) : NULL;
1396                                 break;
1397                         case 'C': // Current Time
1398                                 tmp = nowTime;
1399                                 inc_refcount = true;
1400                                 break;
1401                         case 'R': // service reference string
1402                                 tmp = service;
1403                                 inc_refcount = true;
1404                                 break;
1405                         case 'N': // service name
1406                                 tmp = service_name;
1407                                 inc_refcount = true;
1408                 }
1409                 if (!tmp)
1410                 {
1411                         tmp = Py_None;
1412                         inc_refcount = true;
1413                 }
1414                 if (inc_refcount)
1415                         Py_INCREF(tmp);
1416                 PyTuple_SET_ITEM(tuple, pos++, tmp);
1417         }
1418 }
1419
1420 PyObject *handleEvent(ePtr<eServiceEvent> &ptr, PyObject *dest_list, char* argstring, int argcount, PyObject *service, PyObject *nowTime, PyObject *service_name, PyObject *convertFunc, PyObject *convertFuncArgs)
1421 {
1422         if (convertFunc)
1423         {
1424                 fillTuple(convertFuncArgs, argstring, argcount, service, ptr, nowTime, service_name);
1425                 PyObject *result = PyObject_CallObject(convertFunc, convertFuncArgs);
1426                 if (result == NULL)
1427                 {
1428                         if (service_name)
1429                                 Py_DECREF(service_name);
1430                         if (nowTime)
1431                                 Py_DECREF(nowTime);
1432                         Py_DECREF(convertFuncArgs);
1433                         Py_DECREF(dest_list);
1434                         return result;
1435                 }
1436                 PyList_Append(dest_list, result);
1437                 Py_DECREF(result);
1438         }
1439         else
1440         {
1441                 PyObject *tuple = PyTuple_New(argcount);
1442                 fillTuple(tuple, argstring, argcount, service, ptr, nowTime, service_name);
1443                 PyList_Append(dest_list, tuple);
1444                 Py_DECREF(tuple);
1445         }
1446         return 0;
1447 }
1448
1449 // here we get a list with tuples
1450 // first tuple entry is the servicereference
1451 // the second is the type of query (0 = time, 1 = event_id)
1452 // the third
1453 //              when type is eventid it is the event_id
1454 //              when type is time then it is the start_time ( 0 for now_time )
1455 // the fourth is the end_time .. ( optional )
1456
1457 /* argv is a python string
1458    I = Event Id
1459    B = Event Begin Time
1460    D = Event Duration
1461    T = Event Title
1462    S = Event Short Description
1463    E = Event Extended Description
1464    C = Current Time
1465    R = Service Reference
1466    N = Service Name
1467 */
1468
1469 PyObject *eEPGCache::lookupEvent(PyObject *list, PyObject *convertFunc)
1470 {
1471         PyObject *convertFuncArgs=NULL;
1472         int argcount=0;
1473         char *argstring=NULL;
1474         if (!PyList_Check(list))
1475         {
1476                 PyErr_SetString(PyExc_StandardError,
1477                         "type error");
1478                 eDebug("no list");
1479                 return NULL;
1480         }
1481         int listIt=0;
1482         int listSize=PyList_Size(list);
1483         if (!listSize)
1484         {
1485                 PyErr_SetString(PyExc_StandardError,
1486                         "not params given");
1487                 eDebug("not params given");
1488                 return NULL;
1489         }
1490         else 
1491         {
1492                 PyObject *argv=PyList_GET_ITEM(list, 0); // borrowed reference!
1493                 if (PyString_Check(argv))
1494                 {
1495                         argstring = PyString_AS_STRING(argv);
1496                         ++listIt;
1497                 }
1498                 else
1499                         argstring = "I"; // just event id as default
1500                 argcount = strlen(argstring);
1501 //              eDebug("have %d args('%s')", argcount, argstring);
1502         }
1503         if (convertFunc)
1504         {
1505                 if (!PyCallable_Check(convertFunc))
1506                 {
1507                         PyErr_SetString(PyExc_StandardError,
1508                                 "convertFunc must be callable");
1509                         eDebug("convertFunc is not callable");
1510                         return NULL;
1511                 }
1512                 convertFuncArgs = PyTuple_New(argcount);
1513         }
1514
1515         PyObject *nowTime = strchr(argstring, 'C') ?
1516                 PyLong_FromLong(time(0)+eDVBLocalTimeHandler::getInstance()->difference()) :
1517                 NULL;
1518
1519         bool must_get_service_name = strchr(argstring, 'N') ? true : false;
1520
1521         // create dest list
1522         PyObject *dest_list=PyList_New(0);
1523         while(listSize > listIt)
1524         {
1525                 PyObject *item=PyList_GET_ITEM(list, listIt++); // borrowed reference!
1526                 if (PyTuple_Check(item))
1527                 {
1528                         int type=0;
1529                         long event_id=-1;
1530                         time_t stime=-1;
1531                         int minutes=0;
1532                         int tupleSize=PyTuple_Size(item);
1533                         int tupleIt=0;
1534                         PyObject *service=NULL;
1535                         while(tupleSize > tupleIt)  // parse query args
1536                         {
1537                                 PyObject *entry=PyTuple_GET_ITEM(item, tupleIt); // borrowed reference!
1538                                 switch(tupleIt++)
1539                                 {
1540                                         case 0:
1541                                         {
1542                                                 if (!PyString_Check(entry))
1543                                                 {
1544                                                         eDebug("tuple entry 0 is no a string");
1545                                                         goto skip_entry;
1546                                                 }
1547                                                 service = entry;
1548                                                 break;
1549                                         }
1550                                         case 1:
1551                                                 type=PyInt_AsLong(entry);
1552                                                 if (type < -1 || type > 2)
1553                                                 {
1554                                                         eDebug("unknown type %d", type);
1555                                                         goto skip_entry;
1556                                                 }
1557                                                 break;
1558                                         case 2:
1559                                                 event_id=stime=PyInt_AsLong(entry);
1560                                                 break;
1561                                         case 3:
1562                                                 minutes=PyInt_AsLong(entry);
1563                                                 break;
1564                                         default:
1565                                                 eDebug("unneeded extra argument");
1566                                                 break;
1567                                 }
1568                         }
1569                         eServiceReference ref(PyString_AS_STRING(service));
1570                         if (ref.type != eServiceReference::idDVB)
1571                         {
1572                                 eDebug("service reference for epg query is not valid");
1573                                 continue;
1574                         }
1575                         PyObject *service_name=NULL;
1576                         if (must_get_service_name)
1577                         {
1578                                 ePtr<iStaticServiceInformation> sptr;
1579                                 eServiceCenterPtr service_center;
1580                                 eServiceCenter::getPrivInstance(service_center);
1581                                 if (service_center)
1582                                 {
1583                                         service_center->info(ref, sptr);
1584                                         if (sptr)
1585                                         {
1586                                                 std::string name;
1587                                                 sptr->getName(ref, name);
1588                                                 if (name.length())
1589                                                         service_name = PyString_FromString(name.c_str());
1590                                         }
1591                                 }
1592                                 if (!service_name)
1593                                         service_name = PyString_FromString("<n/a>");
1594                         }
1595                         if (minutes)
1596                         {
1597                                 Lock();
1598                                 if (!startTimeQuery(ref, stime, minutes))
1599                                 {
1600                                         ePtr<eServiceEvent> ptr;
1601                                         while (!getNextTimeEntry(ptr))
1602                                         {
1603                                                 PyObject *ret = handleEvent(ptr, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs);
1604                                                 if (ret)
1605                                                         return ret;
1606                                         }
1607                                 }
1608                                 Unlock();
1609                         }
1610                         else
1611                         {
1612                                 ePtr<eServiceEvent> ptr;
1613                                 if (stime)
1614                                 {
1615                                         if (type == 2)
1616                                                 lookupEventId(ref, event_id, ptr);
1617                                         else
1618                                                 lookupEventTime(ref, stime, ptr, type);
1619                                 }
1620                                 PyObject *ret = handleEvent(ptr, dest_list, argstring, argcount, service, nowTime, service_name, convertFunc, convertFuncArgs);
1621                                 if (ret)
1622                                         return ret;
1623                         }
1624                         if (service_name)
1625                                 Py_DECREF(service_name);
1626                 }
1627 skip_entry:
1628                 ;
1629         }
1630         if (convertFuncArgs)
1631                 Py_DECREF(convertFuncArgs);
1632         if (nowTime)
1633                 Py_DECREF(nowTime);
1634         return dest_list;
1635 }
1636
1637 #ifdef ENABLE_PRIVATE_EPG
1638 #include <dvbsi++/descriptor_tag.h>
1639 #include <dvbsi++/unknown_descriptor.h>
1640 #include <dvbsi++/private_data_specifier_descriptor.h>
1641
1642 void eEPGCache::PMTready(eDVBServicePMTHandler *pmthandler)
1643 {
1644         ePtr<eTable<ProgramMapSection> > ptr;
1645         if (!pmthandler->getPMT(ptr) && ptr)
1646         {
1647                 std::vector<ProgramMapSection*>::const_iterator i;
1648                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
1649                 {
1650                         const ProgramMapSection &pmt = **i;
1651
1652                         ElementaryStreamInfoConstIterator es;
1653                         for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
1654                         {
1655                                 int tmp=0;
1656                                 switch ((*es)->getType())
1657                                 {
1658                                 case 0x05: // private
1659                                         for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
1660                                                 desc != (*es)->getDescriptors()->end(); ++desc)
1661                                         {
1662                                                 switch ((*desc)->getTag())
1663                                                 {
1664                                                         case PRIVATE_DATA_SPECIFIER_DESCRIPTOR:
1665                                                                 if (((PrivateDataSpecifierDescriptor*)(*desc))->getPrivateDataSpecifier() == 190)
1666                                                                         tmp |= 1;
1667                                                                 break;
1668                                                         case 0x90:
1669                                                         {
1670                                                                 UnknownDescriptor *descr = (UnknownDescriptor*)*desc;
1671                                                                 int descr_len = descr->getLength();
1672                                                                 if (descr_len == 4)
1673                                                                 {
1674                                                                         uint8_t data[descr_len+2];
1675                                                                         descr->writeToBuffer(data);
1676                                                                         if ( !data[2] && !data[3] && data[4] == 0xFF && data[5] == 0xFF )
1677                                                                                 tmp |= 2;
1678                                                                 }
1679                                                                 break;
1680                                                         }
1681                                                         default:
1682                                                                 break;
1683                                                 }
1684                                         }
1685                                 default:
1686                                         break;
1687                                 }
1688                                 if (tmp==3)
1689                                 {
1690                                         eServiceReferenceDVB ref;
1691                                         if (!pmthandler->getService(ref))
1692                                         {
1693                                                 int pid = (*es)->getPid();
1694                                                 messages.send(Message(Message::got_private_pid, ref, pid));
1695                                                 return;
1696                                         }
1697                                 }
1698                         }
1699                 }
1700         }
1701         else
1702                 eDebug("PMTready but no pmt!!");
1703 }
1704
1705 struct date_time
1706 {
1707         __u8 data[5];
1708         time_t tm;
1709         date_time( const date_time &a )
1710         {
1711                 memcpy(data, a.data, 5);
1712                 tm = a.tm;
1713         }
1714         date_time( const __u8 data[5])
1715         {
1716                 memcpy(this->data, data, 5);
1717                 tm = parseDVBtime(data[0], data[1], data[2], data[3], data[4]);
1718         }
1719         date_time()
1720         {
1721         }
1722         const __u8& operator[](int pos) const
1723         {
1724                 return data[pos];
1725         }
1726 };
1727
1728 struct less_datetime
1729 {
1730         bool operator()( const date_time &a, const date_time &b ) const
1731         {
1732                 return abs(a.tm-b.tm) < 360 ? false : a.tm < b.tm;
1733         }
1734 };
1735
1736 void eEPGCache::privateSectionRead(const uniqueEPGKey &current_service, const __u8 *data)
1737 {
1738         contentMap &content_time_table = content_time_tables[current_service];
1739         singleLock s(cache_lock);
1740         std::map< date_time, std::list<uniqueEPGKey>, less_datetime > start_times;
1741         eventMap &evMap = eventDB[current_service].first;
1742         timeMap &tmMap = eventDB[current_service].second;
1743         int ptr=8;
1744         int content_id = data[ptr++] << 24;
1745         content_id |= data[ptr++] << 16;
1746         content_id |= data[ptr++] << 8;
1747         content_id |= data[ptr++];
1748
1749         contentTimeMap &time_event_map =
1750                 content_time_table[content_id];
1751         for ( contentTimeMap::iterator it( time_event_map.begin() );
1752                 it != time_event_map.end(); ++it )
1753         {
1754                 eventMap::iterator evIt( evMap.find(it->second.second) );
1755                 if ( evIt != evMap.end() )
1756                 {
1757                         delete evIt->second;
1758                         evMap.erase(evIt);
1759                 }
1760                 tmMap.erase(it->second.first);
1761         }
1762         time_event_map.clear();
1763
1764         __u8 duration[3];
1765         memcpy(duration, data+ptr, 3);
1766         ptr+=3;
1767         int duration_sec =
1768                 fromBCD(duration[0])*3600+fromBCD(duration[1])*60+fromBCD(duration[2]);
1769
1770         const __u8 *descriptors[65];
1771         const __u8 **pdescr = descriptors;
1772
1773         int descriptors_length = (data[ptr++]&0x0F) << 8;
1774         descriptors_length |= data[ptr++];
1775         while ( descriptors_length > 0 )
1776         {
1777                 int descr_type = data[ptr];
1778                 int descr_len = data[ptr+1];
1779                 descriptors_length -= (descr_len+2);
1780                 if ( descr_type == 0xf2 )
1781                 {
1782                         ptr+=2;
1783                         int tsid = data[ptr++] << 8;
1784                         tsid |= data[ptr++];
1785                         int onid = data[ptr++] << 8;
1786                         onid |= data[ptr++];
1787                         int sid = data[ptr++] << 8;
1788                         sid |= data[ptr++];
1789                         uniqueEPGKey service( sid, onid, tsid );
1790                         descr_len -= 6;
1791                         while( descr_len > 0 )
1792                         {
1793                                 __u8 datetime[5];
1794                                 datetime[0] = data[ptr++];
1795                                 datetime[1] = data[ptr++];
1796                                 int tmp_len = data[ptr++];
1797                                 descr_len -= 3;
1798                                 while( tmp_len > 0 )
1799                                 {
1800                                         memcpy(datetime+2, data+ptr, 3);
1801                                         ptr+=3;
1802                                         descr_len -= 3;
1803                                         tmp_len -= 3;
1804                                         start_times[datetime].push_back(service);
1805                                 }
1806                         }
1807                 }
1808                 else
1809                 {
1810                         *pdescr++=data+ptr;
1811                         ptr += 2;
1812                         ptr += descr_len;
1813                 }
1814         }
1815         __u8 event[4098];
1816         eit_event_struct *ev_struct = (eit_event_struct*) event;
1817         ev_struct->running_status = 0;
1818         ev_struct->free_CA_mode = 1;
1819         memcpy(event+7, duration, 3);
1820         ptr = 12;
1821         const __u8 **d=descriptors;
1822         while ( d < pdescr )
1823         {
1824                 memcpy(event+ptr, *d, ((*d)[1])+2);
1825                 ptr+=(*d++)[1];
1826                 ptr+=2;
1827         }
1828         for ( std::map< date_time, std::list<uniqueEPGKey> >::iterator it(start_times.begin()); it != start_times.end(); ++it )
1829         {
1830                 time_t now = eDVBLocalTimeHandler::getInstance()->nowTime();
1831                 if ( (it->first.tm + duration_sec) < now )
1832                         continue;
1833                 memcpy(event+2, it->first.data, 5);
1834                 int bptr = ptr;
1835                 int cnt=0;
1836                 for (std::list<uniqueEPGKey>::iterator i(it->second.begin()); i != it->second.end(); ++i)
1837                 {
1838                         event[bptr++] = 0x4A;
1839                         __u8 *len = event+(bptr++);
1840                         event[bptr++] = (i->tsid & 0xFF00) >> 8;
1841                         event[bptr++] = (i->tsid & 0xFF);
1842                         event[bptr++] = (i->onid & 0xFF00) >> 8;
1843                         event[bptr++] = (i->onid & 0xFF);
1844                         event[bptr++] = (i->sid & 0xFF00) >> 8;
1845                         event[bptr++] = (i->sid & 0xFF);
1846                         event[bptr++] = 0xB0;
1847                         bptr += sprintf((char*)(event+bptr), "Option %d", ++cnt);
1848                         *len = ((event+bptr) - len)-1;
1849                 }
1850                 int llen = bptr - 12;
1851                 ev_struct->descriptors_loop_length_hi = (llen & 0xF00) >> 8;
1852                 ev_struct->descriptors_loop_length_lo = (llen & 0xFF);
1853
1854                 time_t stime = it->first.tm;
1855                 while( tmMap.find(stime) != tmMap.end() )
1856                         ++stime;
1857                 event[6] += (stime - it->first.tm);
1858                 __u16 event_id = 0;
1859                 while( evMap.find(event_id) != evMap.end() )
1860                         ++event_id;
1861                 event[0] = (event_id & 0xFF00) >> 8;
1862                 event[1] = (event_id & 0xFF);
1863                 time_event_map[it->first.tm]=std::pair<time_t, __u16>(stime, event_id);
1864                 eventData *d = new eventData( ev_struct, bptr, eEPGCache::SCHEDULE );
1865                 evMap[event_id] = d;
1866                 tmMap[stime] = d;
1867         }
1868 }
1869
1870 void eEPGCache::channel_data::startPrivateReader(int pid, int version)
1871 {
1872         eDVBSectionFilterMask mask;
1873         memset(&mask, 0, sizeof(mask));
1874         mask.pid = pid;
1875         mask.flags = eDVBSectionFilterMask::rfCRC;
1876         mask.data[0] = 0xA0;
1877         mask.mask[0] = 0xFF;
1878         eDebug("start privatefilter for pid %04x and version %d", pid, version);
1879         if (version != -1)
1880         {
1881                 mask.data[3] = version << 1;
1882                 mask.mask[3] = 0x3E;
1883                 mask.mode[3] = 0x3E;
1884         }
1885         seenPrivateSections.clear();
1886         m_PrivateReader->connectRead(slot(*this, &eEPGCache::channel_data::readPrivateData), m_PrivateConn);
1887         m_PrivateReader->start(mask);
1888 #ifdef NEED_DEMUX_WORKAROUND
1889         m_PrevVersion=version;
1890 #endif
1891 }
1892
1893 void eEPGCache::channel_data::readPrivateData( const __u8 *data)
1894 {
1895         if (!data)
1896                 eDebug("get Null pointer from section reader !!");
1897         else
1898         {
1899                 if ( seenPrivateSections.find( data[6] ) == seenPrivateSections.end() )
1900                 {
1901 #ifdef NEED_DEMUX_WORKAROUND
1902                         int version = data[5];
1903                         version = ((version & 0x3E) >> 1);
1904                         can_delete = 0;
1905                         if ( m_PrevVersion != version )
1906                         {
1907                                 cache->privateSectionRead(m_PrivateService, data);
1908                                 seenPrivateSections.insert(data[6]);
1909                         }
1910                         else
1911                                 eDebug("ignore");
1912 #else
1913                         can_delete = 0;
1914                         cache->privateSectionRead(m_PrivateService, data);
1915                         seenPrivateSections.insert(data[6]);
1916 #endif
1917                 }
1918                 if ( seenPrivateSections.size() == (unsigned int)(data[7] + 1) )
1919                 {
1920                         eDebug("[EPGC] private finished");
1921                         if (!isRunning)
1922                                 can_delete = 1;
1923                         int version = data[5];
1924                         version = ((version & 0x3E) >> 1);
1925                         startPrivateReader(m_PrivatePid, version);
1926                 }
1927         }
1928 }
1929
1930 #endif // ENABLE_PRIVATE_EPG