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