more work on dvb subtitles (not finished/working yet)
[enigma2.git] / lib / dvb / pmt.cpp
1 #include <lib/base/eerror.h>
2 #include <lib/dvb/pmt.h>
3 #include <lib/dvb/specs.h>
4 #include <lib/dvb/dvb.h>
5 #include <lib/dvb/metaparser.h>
6 #include <lib/dvb_ci/dvbci.h>
7 #include <lib/dvb/epgcache.h>
8 #include <lib/dvb/scan.h>
9 #include <dvbsi++/ca_descriptor.h>
10 #include <dvbsi++/ca_program_map_section.h>
11 #include <dvbsi++/teletext_descriptor.h>
12 #include <dvbsi++/descriptor_tag.h>
13 #include <dvbsi++/iso639_language_descriptor.h>
14 #include <dvbsi++/stream_identifier_descriptor.h>
15 #include <dvbsi++/subtitling_descriptor.h>
16
17 eDVBServicePMTHandler::eDVBServicePMTHandler()
18         :m_ca_servicePtr(0), m_dvb_scan(0), m_decode_demux_num(0xFF)
19 {
20         m_use_decode_demux = 0;
21         m_pmt_pid = -1;
22         eDVBResourceManager::getInstance(m_resourceManager);
23         CONNECT(m_PMT.tableReady, eDVBServicePMTHandler::PMTready);
24         CONNECT(m_PAT.tableReady, eDVBServicePMTHandler::PATready);
25 }
26
27 eDVBServicePMTHandler::~eDVBServicePMTHandler()
28 {
29         free();
30 }
31
32 void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
33 {
34         int state;
35         channel->getState(state);
36         
37         if ((m_last_channel_state != iDVBChannel::state_ok)
38                 && (state == iDVBChannel::state_ok) && (!m_demux))
39         {
40                 if (m_channel)
41                         if (m_channel->getDemux(m_demux, (!m_use_decode_demux) ? 0 : iDVBChannel::capDecode))
42                                 eDebug("Allocating %s-decoding a demux for now tuned-in channel failed.", m_use_decode_demux ? "" : "non-");
43                 
44                 serviceEvent(eventTuned);
45                 
46                 if (m_demux)
47                 {
48                         eDebug("ok ... now we start!!");
49
50                         if (m_pmt_pid == -1)
51                                 m_PAT.begin(eApp, eDVBPATSpec(), m_demux);
52                         else
53                                 m_PMT.begin(eApp, eDVBPMTSpec(m_pmt_pid, m_reference.getServiceID().get()), m_demux);
54
55                         if ( m_service && !m_service->cacheEmpty() )
56                                 serviceEvent(eventNewProgramInfo);
57                 }
58         } else if ((m_last_channel_state != iDVBChannel::state_failed) && 
59                         (state == iDVBChannel::state_failed))
60         {
61                 eDebug("tune failed.");
62                 serviceEvent(eventTuneFailed);
63         }
64 }
65
66 void eDVBServicePMTHandler::channelEvent(iDVBChannel *channel, int event)
67 {
68         switch (event)
69         {
70         case iDVBChannel::evtEOF:
71                 serviceEvent(eventEOF);
72                 break;
73         case iDVBChannel::evtSOF:
74                 serviceEvent(eventSOF);
75                 break;
76         default:
77                 break;
78         }
79 }
80
81 void eDVBServicePMTHandler::PMTready(int error)
82 {
83         if (error)
84                 serviceEvent(eventNoPMT);
85         else
86         {
87                 m_have_cached_program = false;
88                 serviceEvent(eventNewProgramInfo);
89                 eEPGCache::getInstance()->PMTready(this);
90                 if (!m_pvr_channel) // don't send campmt to camd.socket for playbacked services
91                 {
92                         if(!m_ca_servicePtr)
93                         {
94                                 int demuxes[2] = {0,0};
95                                 uint8_t tmp;
96                                 m_demux->getCADemuxID(tmp);
97                                 demuxes[0]=tmp;
98                                 if (m_decode_demux_num != 0xFF)
99                                         demuxes[1]=m_decode_demux_num;
100                                 else
101                                         demuxes[1]=demuxes[0];
102                                 eDVBCAService::register_service(m_reference, demuxes, m_ca_servicePtr);
103                                 eDVBCIInterfaces::getInstance()->recheckPMTHandlers();
104                         }
105                         eDVBCIInterfaces::getInstance()->gotPMT(this);
106                 }
107                 if (m_ca_servicePtr)
108                 {
109                         ePtr<eTable<ProgramMapSection> > ptr;
110                         if (!m_PMT.getCurrent(ptr))
111                                 m_ca_servicePtr->buildCAPMT(ptr);
112                         else
113                                 eDebug("eDVBServicePMTHandler cannot call buildCAPMT");
114                 }
115         }
116 }
117
118 void eDVBServicePMTHandler::PATready(int)
119 {
120         ePtr<eTable<ProgramAssociationSection> > ptr;
121         if (!m_PAT.getCurrent(ptr))
122         {
123                 int pmtpid = -1;
124                 std::vector<ProgramAssociationSection*>::const_iterator i;
125                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
126                 {
127                         const ProgramAssociationSection &pat = **i;
128                         ProgramAssociationConstIterator program;
129                         for (program = pat.getPrograms()->begin(); program != pat.getPrograms()->end(); ++program)
130                                 if (eServiceID((*program)->getProgramNumber()) == m_reference.getServiceID())
131                                         pmtpid = (*program)->getProgramMapPid();
132                 }
133                 if (pmtpid == -1)
134                         serviceEvent(eventNoPATEntry);
135                 else
136                         m_PMT.begin(eApp, eDVBPMTSpec(pmtpid, m_reference.getServiceID().get()), m_demux);
137         } else
138                 serviceEvent(eventNoPAT);
139 }
140
141 PyObject *eDVBServicePMTHandler::getCaIds()
142 {
143         PyObject *ret=0;
144
145         program prog;
146
147         if ( !getProgramInfo(prog) )
148         {
149                 int cnt=prog.caids.size();
150                 if (cnt)
151                 {
152                         ret=PyList_New(cnt);
153                         std::set<uint16_t>::iterator it(prog.caids.begin());
154                         while(cnt--)
155                                 PyList_SET_ITEM(ret, cnt, PyInt_FromLong(*it++));
156                 }
157         }
158
159         return ret ? ret : PyList_New(0);
160 }
161
162 int eDVBServicePMTHandler::getProgramInfo(struct program &program)
163 {
164         ePtr<eTable<ProgramMapSection> > ptr;
165         int cached_apid_ac3 = -1;
166         int cached_apid_mpeg = -1;
167         int cached_vpid = -1;
168         int cached_tpid = -1;
169         int ret = -1;
170
171         program.videoStreams.clear();
172         program.audioStreams.clear();
173         program.pcrPid = -1;
174         program.pmtPid = -1;
175         program.textPid = -1;
176
177         if ( m_service && !m_service->cacheEmpty() )
178         {
179                 cached_vpid = m_service->getCacheEntry(eDVBService::cVPID);
180                 cached_apid_mpeg = m_service->getCacheEntry(eDVBService::cAC3PID);
181                 cached_apid_ac3 = m_service->getCacheEntry(eDVBService::cAPID);
182                 cached_tpid = m_service->getCacheEntry(eDVBService::cTPID);
183         }
184
185         if ( ((m_service && m_service->usePMT()) || !m_service) && !m_PMT.getCurrent(ptr))
186         {
187                 if (m_have_cached_program)
188                 {
189                         program = m_cached_program;
190                         ret = 0;
191                 }
192                 else
193                 {
194                         eDVBTableSpec table_spec;
195                         ptr->getSpec(table_spec);
196                         program.pmtPid = table_spec.pid < 0x1fff ? table_spec.pid : -1;
197                         std::vector<ProgramMapSection*>::const_iterator i;
198                         for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
199                         {
200                                 const ProgramMapSection &pmt = **i;
201                                 program.pcrPid = pmt.getPcrPid();
202                         
203                                 ElementaryStreamInfoConstIterator es;
204                                 for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
205                                 {
206                                         int isaudio = 0, isvideo = 0;
207                                         videoStream video;
208                                         audioStream audio;
209                                         audio.component_tag=video.component_tag=-1;
210                                         video.type = videoStream::vtMPEG2;
211
212                                         switch ((*es)->getType())
213                                         {
214                                         case 0x1b: // AVC Video Stream (MPEG4 H264)
215                                                 video.type = videoStream::vtMPEG4_H264;
216                                         case 0x01: // MPEG 1 video
217                                         case 0x02: // MPEG 2 video
218                                                 isvideo = 1;
219                                                 //break; fall through !!!
220                                         case 0x03: // MPEG 1 audio
221                                         case 0x04: // MPEG 2 audio:
222                                                 if (!isvideo)
223                                                 {
224                                                         isaudio = 1;
225                                                         audio.type = audioStream::atMPEG;
226                                                 }
227                                                 //break; fall through !!!
228                                         case 0x06: // PES Private
229                                                         /* PES private can contain AC-3, DTS or lots of other stuff.
230                                                            check descriptors to get the exact type. */
231                                                 for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
232                                                                 desc != (*es)->getDescriptors()->end(); ++desc)
233                                                 {
234                                                         switch ((*desc)->getTag())
235                                                         {
236                                                         case SUBTITLING_DESCRIPTOR:
237                                                         {
238                                                                 SubtitlingDescriptor *d = (SubtitlingDescriptor*)(*desc);
239                                                                 const SubtitlingList *list = d->getSubtitlings();
240                                                                 subtitleStream s;
241                                                                 s.pid = (*es)->getPid();
242                                                                 for (SubtitlingConstIterator it(list->begin()); it != list->end(); ++it)
243                                                                 {
244                                                                         s.subtitling_type = (*it)->getSubtitlingType();
245                                                                         s.composition_page_id = (*it)->getCompositionPageId();
246                                                                         s.ancillary_page_id = (*it)->getAncillaryPageId();
247                                                                         s.language_code = (*it)->getIso639LanguageCode();
248                                                                         eDebug("add subtitle %s PID %04x, type %d, composition page %d, ancillary_page %d",
249                                                                                 s.language_code.c_str(), s.pid, s.subtitling_type, s.composition_page_id, s.ancillary_page_id);
250                                                                         program.subtitleStreams.push_back(s);
251                                                                 }
252                                                                 break;
253                                                         }
254                                                         case TELETEXT_DESCRIPTOR:
255                                                                 if ( program.textPid == -1 || (*es)->getPid() == cached_tpid )
256                                                                         program.textPid = (*es)->getPid();
257                                                                 break;
258                                                         case DTS_DESCRIPTOR:
259                                                                 isaudio = 1;
260                                                                 audio.type = audioStream::atDTS;
261                                                                 break;
262                                                         case AAC_DESCRIPTOR:
263                                                                 isaudio = 1;
264                                                                 audio.type = audioStream::atAAC;
265                                                                 break;
266                                                         case AC3_DESCRIPTOR:
267                                                                 isaudio = 1;
268                                                                 audio.type = audioStream::atAC3;
269                                                                 break;
270                                                         case ISO_639_LANGUAGE_DESCRIPTOR:
271                                                                 if (!isvideo)
272                                                                 {
273                                                                         const Iso639LanguageList *languages = ((Iso639LanguageDescriptor*)*desc)->getIso639Languages();
274                                                                                 /* use last language code */
275                                                                         for (Iso639LanguageConstIterator i(languages->begin()); i != languages->end(); ++i)
276                                                                                 audio.language_code = (*i)->getIso639LanguageCode();
277                                                                 }
278                                                                 break;
279                                                         case STREAM_IDENTIFIER_DESCRIPTOR:
280                                                                 audio.component_tag =
281                                                                         video.component_tag =
282                                                                                 ((StreamIdentifierDescriptor*)*desc)->getComponentTag();
283                                                                 break;
284                                                         case CA_DESCRIPTOR:
285                                                         {
286                                                                 CaDescriptor *descr = (CaDescriptor*)(*desc);
287                                                                 program.caids.insert(descr->getCaSystemId());
288                                                                 break;
289                                                         }
290                                                         }
291                                                 }
292                                                 break;
293                                         }
294                                         if (isaudio)
295                                         {
296                                                 audio.pid = (*es)->getPid();
297                                                 if ( !program.audioStreams.empty() &&
298                                                         ( audio.pid == cached_apid_ac3 || audio.pid == cached_apid_mpeg) )
299                                                 {
300                                                         program.audioStreams.push_back(program.audioStreams[0]);
301                                                         program.audioStreams[0] = audio;
302                                                 }
303                                                 else
304                                                         program.audioStreams.push_back(audio);
305                                         }
306                                         else if (isvideo)
307                                         {
308                                                 video.pid = (*es)->getPid();
309                                                 if ( !program.videoStreams.empty() && video.pid == cached_vpid )
310                                                 {
311                                                         program.videoStreams.push_back(program.videoStreams[0]);
312                                                         program.videoStreams[0] = video;
313                                                 }
314                                                 else
315                                                         program.videoStreams.push_back(video);
316                                         }
317                                         else
318                                                 continue;
319                                 }
320                                 for (DescriptorConstIterator desc = pmt.getDescriptors()->begin();
321                                         desc != pmt.getDescriptors()->end(); ++desc)
322                                 {
323                                         if ((*desc)->getTag() == CA_DESCRIPTOR)
324                                         {
325                                                 CaDescriptor *descr = (CaDescriptor*)(*desc);
326                                                 program.caids.insert(descr->getCaSystemId());
327                                         }
328                                 }
329                         }
330                         ret = 0;
331                         m_cached_program = program;
332                         m_have_cached_program = true;
333                 }
334         } else if ( m_service && !m_service->cacheEmpty() )
335         {
336                 int cached_pcrpid = m_service->getCacheEntry(eDVBService::cPCRPID),
337                         vpidtype = m_service->getCacheEntry(eDVBService::cVTYPE),
338                         cnt=0;
339                 if ( vpidtype == -1 )
340                         vpidtype = videoStream::vtMPEG2;
341                 if ( cached_vpid != -1 )
342                 {
343                         videoStream s;
344                         s.pid = cached_vpid;
345                         s.type = vpidtype;
346                         program.videoStreams.push_back(s);
347                         ++cnt;
348                 }
349                 if ( cached_apid_ac3 != -1 )
350                 {
351                         audioStream s;
352                         s.type = audioStream::atAC3;
353                         s.pid = cached_apid_ac3;
354                         program.audioStreams.push_back(s);
355                         ++cnt;
356                 }
357                 if ( cached_apid_mpeg != -1 )
358                 {
359                         audioStream s;
360                         s.type = audioStream::atMPEG;
361                         s.pid = cached_apid_mpeg;
362                         program.audioStreams.push_back(s);
363                         ++cnt;
364                 }
365                 if ( cached_pcrpid != -1 )
366                 {
367                         ++cnt;
368                         program.pcrPid = cached_pcrpid;
369                 }
370                 if ( cached_tpid != -1 )
371                 {
372                         ++cnt;
373                         program.textPid = cached_tpid;
374                 }
375                 CAID_LIST &caids = m_service->m_ca;
376                 for (CAID_LIST::iterator it(caids.begin()); it != caids.end(); ++it)
377                         program.caids.insert(*it);
378                 if ( cnt )
379                         ret = 0;
380         }
381         return ret;
382 }
383
384 int eDVBServicePMTHandler::getChannel(eUsePtr<iDVBChannel> &channel)
385 {
386         channel = m_channel;
387         if (channel)
388                 return 0;
389         else
390                 return -1;
391 }
392
393 int eDVBServicePMTHandler::getDataDemux(ePtr<iDVBDemux> &demux)
394 {
395         demux = m_demux;
396         if (demux)
397                 return 0;
398         else
399                 return -1;
400 }
401
402 int eDVBServicePMTHandler::getDecodeDemux(ePtr<iDVBDemux> &demux)
403 {
404         int ret=0;
405                 /* if we're using the decoding demux as data source
406                    (for example in pvr playbacks), return that one. */
407         if (m_use_decode_demux)
408         {
409                 demux = m_demux;
410                 return ret;
411         }
412         
413         ASSERT(m_channel); /* calling without a previous ::tune is certainly bad. */
414
415         ret = m_channel->getDemux(demux, iDVBChannel::capDecode);
416         if (!ret)
417                 demux->getCADemuxID(m_decode_demux_num);
418
419         return ret;
420 }
421
422 int eDVBServicePMTHandler::getPVRChannel(ePtr<iDVBPVRChannel> &pvr_channel)
423 {
424         pvr_channel = m_pvr_channel;
425         if (pvr_channel)
426                 return 0;
427         else
428                 return -1;
429 }
430
431 void eDVBServicePMTHandler::SDTScanEvent(int event)
432 {
433         switch (event)
434         {
435                 case eDVBScan::evtFinish:
436                 {
437                         ePtr<iDVBChannelList> db;
438                         if (m_resourceManager->getChannelList(db) != 0)
439                                 eDebug("no channel list");
440                         else
441                         {
442                                 m_dvb_scan->insertInto(db, true);
443                                 eDebug("sdt update done!");
444                         }
445                         break;
446                 }
447
448                 default:
449                         break;
450         }
451 }
452
453 int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux, eCueSheet *cue)
454 {
455         RESULT res;
456         m_reference = ref;
457         
458         m_use_decode_demux = use_decode_demux;
459         
460                 /* is this a normal (non PVR) channel? */
461         if (ref.path.empty())
462         {
463                 eDVBChannelID chid;
464                 ref.getChannelID(chid);
465                 res = m_resourceManager->allocateChannel(chid, m_channel);
466                 eDebug("allocate Channel: res %d", res);
467
468                 ePtr<iDVBChannelList> db;
469                 if (!m_resourceManager->getChannelList(db))
470                                 db->getService((eServiceReferenceDVB&)m_reference, m_service);
471
472                 if (!res)
473                         eDVBCIInterfaces::getInstance()->addPMTHandler(this);
474         } else
475         {
476                 eDVBMetaParser parser;
477                 
478                 if (parser.parseFile(ref.path))
479                 {
480                         eWarning("no .meta file found, trying to find PMT pid");
481                         eDVBTSTools tstools;
482                         if (tstools.openFile(ref.path.c_str()))
483                                 eWarning("failed to open file");
484                         else
485                         {
486                                 int service_id, pmt_pid;
487                                 if (!tstools.findPMT(pmt_pid, service_id))
488                                 {
489                                         eDebug("PMT pid found on pid %04x, service id %d", pmt_pid, service_id);
490                                         m_reference.setServiceID(service_id);
491                                         m_pmt_pid = pmt_pid;
492                                 }
493                         }
494                 } else
495                         m_reference = parser.m_ref;
496                 
497                 eDebug("alloc PVR");
498                         /* allocate PVR */
499                 res = m_resourceManager->allocatePVRChannel(m_pvr_channel);
500                 if (res)
501                         eDebug("allocatePVRChannel failed!\n");
502                 m_channel = m_pvr_channel;
503         }
504
505         if (m_channel)
506         {
507                 m_channel->connectStateChange(
508                         slot(*this, &eDVBServicePMTHandler::channelStateChanged), 
509                         m_channelStateChanged_connection);
510                 m_last_channel_state = -1;
511                 channelStateChanged(m_channel);
512
513                 m_channel->connectEvent(
514                         slot(*this, &eDVBServicePMTHandler::channelEvent), 
515                         m_channelEvent_connection);
516
517                 if (ref.path.empty())
518                 {
519                         delete m_dvb_scan;
520                         m_dvb_scan = new eDVBScan(m_channel, false, false);
521                         m_dvb_scan->connectEvent(slot(*this, &eDVBServicePMTHandler::SDTScanEvent), m_scan_event_connection);
522                 }
523         } else
524         {
525                 serviceEvent(eventTuneFailed);
526                 return res;
527         }
528
529         if (m_pvr_channel)
530         {
531                 m_pvr_channel->setCueSheet(cue);
532                 m_pvr_channel->playFile(ref.path.c_str());
533         }
534
535         return res;
536 }
537
538 void eDVBServicePMTHandler::free()
539 {
540         m_dvb_scan = 0;
541         delete m_dvb_scan;
542
543         if (m_ca_servicePtr)
544         {
545                 int demuxes[2] = {0,0};
546                 uint8_t tmp;
547                 m_demux->getCADemuxID(tmp);
548                 demuxes[0]=tmp;
549                 if (m_decode_demux_num != 0xFF)
550                         demuxes[1]=m_decode_demux_num;
551                 else
552                         demuxes[1]=demuxes[0];
553                 ePtr<eTable<ProgramMapSection> > ptr;
554                 m_PMT.getCurrent(ptr);
555                 eDVBCAService::unregister_service(m_reference, demuxes, ptr);
556                 m_ca_servicePtr = 0;
557         }
558
559         if (m_channel)
560                 eDVBCIInterfaces::getInstance()->removePMTHandler(this);
561
562         if (m_pvr_channel)
563         {
564                 m_pvr_channel->stopFile();
565                 m_pvr_channel->setCueSheet(0);
566         }
567         m_PMT.stop();
568         m_PAT.stop();
569         m_service = 0;
570         m_channel = 0;
571         m_pvr_channel = 0;
572         m_demux = 0;
573 }
574
575 std::map<eServiceReferenceDVB, eDVBCAService*> eDVBCAService::exist;
576
577 eDVBCAService::eDVBCAService()
578         :m_prev_build_hash(0), m_sendstate(0), m_retryTimer(eApp)
579 {
580         memset(m_used_demux, 0xFF, sizeof(m_used_demux));
581         CONNECT(m_retryTimer.timeout, eDVBCAService::sendCAPMT);
582         Connect();
583 }
584
585 eDVBCAService::~eDVBCAService()
586 {
587         eDebug("[eDVBCAService] free service %s", m_service.toString().c_str());
588         ::close(m_sock);
589 }
590
591 RESULT eDVBCAService::register_service( const eServiceReferenceDVB &ref, int demux_nums[2], eDVBCAService *&caservice )
592 {
593         CAServiceMap::iterator it = exist.find(ref);
594         if ( it != exist.end() )
595                 caservice = it->second;
596         else
597         {
598                 caservice = (exist[ref]=new eDVBCAService());
599                 caservice->m_service = ref;
600                 eDebug("[eDVBCAService] new service %s", ref.toString().c_str() );
601         }
602
603         int loops = demux_nums[0] != demux_nums[1] ? 2 : 1;
604         for (int i=0; i < loops; ++i)
605         {
606 // search free demux entry
607                 int iter=0, max_demux_slots = sizeof(caservice->m_used_demux);
608
609                 while ( iter < max_demux_slots && caservice->m_used_demux[iter] != 0xFF )
610                         ++iter;
611
612                 if ( iter < max_demux_slots )
613                 {
614                         caservice->m_used_demux[iter] = demux_nums[i] & 0xFF;
615                         eDebug("[eDVBCAService] add demux %d to slot %d service %s", caservice->m_used_demux[iter], iter, ref.toString().c_str());
616                 }
617                 else
618                 {
619                         eDebug("[eDVBCAService] no more demux slots free for service %s!!", ref.toString().c_str());
620                         return -1;
621                 }
622         }
623         return 0;
624 }
625
626 RESULT eDVBCAService::unregister_service( const eServiceReferenceDVB &ref, int demux_nums[2], eTable<ProgramMapSection> *ptr )
627 {
628         CAServiceMap::iterator it = exist.find(ref);
629         if ( it == exist.end() )
630         {
631                 eDebug("[eDVBCAService] try to unregister non registered %s", ref.toString().c_str());
632                 return -1;
633         }
634         else
635         {
636                 eDVBCAService *caservice = it->second;
637                 int loops = demux_nums[0] != demux_nums[1] ? 2 : 1;
638                 for (int i=0; i < loops; ++i)
639                 {
640                         bool freed = false;
641                         int iter = 0,
642                                 used_demux_slots = 0,
643                                 max_demux_slots = sizeof(caservice->m_used_demux)/sizeof(int);
644                         while ( iter < max_demux_slots )
645                         {
646                                 if ( caservice->m_used_demux[iter] != 0xFF )
647                                 {
648                                         if ( !freed && caservice->m_used_demux[iter] == demux_nums[i] )
649                                         {
650                                                 eDebug("[eDVBCAService] free slot %d demux %d for service %s", iter, caservice->m_used_demux[iter], caservice->m_service.toString().c_str() );
651                                                 caservice->m_used_demux[iter] = 0xFF;
652                                                 freed=true;
653                                         }
654                                         else
655                                                 ++used_demux_slots;
656                                 }
657                                 ++iter;
658                         }
659                         if (!freed)
660                                 eDebug("[eDVBCAService] couldn't free demux slot for demux %d", demux_nums[i]);
661                         if (i || loops == 1)
662                         {
663                                 if (!used_demux_slots)  // no more used.. so we remove it
664                                 {
665                                         delete it->second;
666                                         exist.erase(it);
667                                 }
668                                 else
669                                 {
670                                         if (ptr)
671                                                 it->second->buildCAPMT(ptr);
672                                         else
673                                                 eDebug("[eDVBCAService] can not send updated demux info");
674                                 }
675                         }
676                 }
677         }
678         return 0;
679 }
680
681 void eDVBCAService::Connect()
682 {
683         memset(&m_servaddr, 0, sizeof(struct sockaddr_un));
684         m_servaddr.sun_family = AF_UNIX;
685         strcpy(m_servaddr.sun_path, "/tmp/camd.socket");
686         m_clilen = sizeof(m_servaddr.sun_family) + strlen(m_servaddr.sun_path);
687         m_sock = socket(PF_UNIX, SOCK_STREAM, 0);
688         connect(m_sock, (struct sockaddr *) &m_servaddr, m_clilen);
689         fcntl(m_sock, F_SETFL, O_NONBLOCK);
690         int val=1;
691         setsockopt(m_sock, SOL_SOCKET, SO_REUSEADDR, &val, 4);
692 }
693
694 void eDVBCAService::buildCAPMT(eTable<ProgramMapSection> *ptr)
695 {
696         if (!ptr)
697                 return;
698
699         eDVBTableSpec table_spec;
700         ptr->getSpec(table_spec);
701
702         int pmtpid = table_spec.pid,
703                 pmt_version = table_spec.version;
704
705         uint8_t demux_mask = 0;
706         int data_demux = -1;
707
708         int iter=0, max_demux_slots = sizeof(m_used_demux);
709         while ( iter < max_demux_slots )
710         {
711                 if ( m_used_demux[iter] != 0xFF )
712                 {
713                         if ( m_used_demux[iter] > data_demux )
714                                 data_demux = m_used_demux[iter];
715                         demux_mask |= (1 << m_used_demux[iter]);
716                 }
717                 ++iter;
718         }
719
720         if ( data_demux == -1 )
721         {
722                 eDebug("[eDVBCAService] no data demux found for service %s", m_service.toString().c_str() );
723                 return;
724         }
725
726         eDebug("demux %d mask %02x prevhash %08x", data_demux, demux_mask, m_prev_build_hash);
727
728         unsigned int build_hash = (pmtpid << 16);
729         build_hash |= (demux_mask << 8);
730         build_hash |= (pmt_version&0xFF);
731
732         if ( build_hash == m_prev_build_hash )
733         {
734                 eDebug("[eDVBCAService] don't build/send the same CA PMT twice");
735                 return;
736         }
737
738         std::vector<ProgramMapSection*>::const_iterator i=ptr->getSections().begin();
739         if ( i != ptr->getSections().end() )
740         {
741                 CaProgramMapSection capmt(*i++, m_prev_build_hash ? 0x05 /*update*/ : 0x03 /*only*/, 0x01 );
742
743                 while( i != ptr->getSections().end() )
744                 {
745 //                      eDebug("append");
746                         capmt.append(*i++);
747                 }
748
749                 // add our private descriptors to capmt
750                 uint8_t tmp[10];
751
752                 tmp[0]=0x84;  // pmt pid
753                 tmp[1]=0x02;
754                 tmp[2]=pmtpid>>8;
755                 tmp[3]=pmtpid&0xFF;
756                 capmt.injectDescriptor(tmp, false);
757
758                 tmp[0] = 0x82; // demux
759                 tmp[1] = 0x02;
760                 tmp[2] = demux_mask;    // descramble bitmask
761                 tmp[3] = data_demux&0xFF; // read section data from demux number
762                 capmt.injectDescriptor(tmp, false);
763
764                 tmp[0] = 0x81; // dvbnamespace
765                 tmp[1] = 0x08;
766                 tmp[2] = m_service.getDVBNamespace().get()>>24;
767                 tmp[3]=(m_service.getDVBNamespace().get()>>16)&0xFF;
768                 tmp[4]=(m_service.getDVBNamespace().get()>>8)&0xFF;
769                 tmp[5]=m_service.getDVBNamespace().get()&0xFF;
770                 tmp[6]=m_service.getTransportStreamID().get()>>8;
771                 tmp[7]=m_service.getTransportStreamID().get()&0xFF;
772                 tmp[8]=m_service.getOriginalNetworkID().get()>>8;
773                 tmp[9]=m_service.getOriginalNetworkID().get()&0xFF;
774                 capmt.injectDescriptor(tmp, false);
775
776                 capmt.writeToBuffer(m_capmt);
777         }
778
779         m_prev_build_hash = build_hash;
780
781         if ( m_sendstate != 0xFFFFFFFF )
782                 m_sendstate=0;
783         sendCAPMT();
784 }
785
786 void eDVBCAService::sendCAPMT()
787 {
788         if ( m_sendstate && m_sendstate != 0xFFFFFFFF ) // broken pipe retry
789         {
790                 ::close(m_sock);
791                 Connect();
792         }
793
794         int wp=0;
795         if ( m_capmt[3] & 0x80 )
796         {
797                 int i=0;
798                 int lenbytes = m_capmt[3] & ~0x80;
799                 while(i < lenbytes)
800                         wp = (wp << 8) | m_capmt[4 + i++];
801                 wp+=4;
802                 wp+=lenbytes;
803         }
804         else
805         {
806                 wp = m_capmt[3];
807                 wp+=4;
808         }
809
810         if ( write(m_sock, m_capmt, wp) == wp )
811         {
812                 m_sendstate=0xFFFFFFFF;
813                 eDebug("[eDVBCAService] send %d bytes",wp);
814 #if 1
815                 for(int i=0;i<wp;i++)
816                         eDebugNoNewLine("%02x ", m_capmt[i]);
817                 eDebug("");
818 #endif
819         }
820         else
821         {
822                 switch(m_sendstate)
823                 {
824                         case 0xFFFFFFFF:
825                                 ++m_sendstate;
826                                 m_retryTimer.start(0,true);
827 //                              eDebug("[eDVBCAService] send failed .. immediate retry");
828                                 break;
829                         default:
830                                 m_retryTimer.start(5000,true);
831 //                              eDebug("[eDVBCAService] send failed .. retry in 5 sec");
832                                 break;
833                 }
834                 ++m_sendstate;
835         }
836 }