after DECLARE_REF now all is private.. not public
[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 #include <dvbsi++/teletext_descriptor.h>
17 #include <lib/base/nconfig.h> // access to python config
18
19 eDVBServicePMTHandler::eDVBServicePMTHandler()
20         :m_ca_servicePtr(0), m_dvb_scan(0), m_decode_demux_num(0xFF)
21 {
22         m_use_decode_demux = 0;
23         m_pmt_pid = -1;
24         eDVBResourceManager::getInstance(m_resourceManager);
25         CONNECT(m_PMT.tableReady, eDVBServicePMTHandler::PMTready);
26         CONNECT(m_PAT.tableReady, eDVBServicePMTHandler::PATready);
27 }
28
29 eDVBServicePMTHandler::~eDVBServicePMTHandler()
30 {
31         free();
32 }
33
34 void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
35 {
36         int state;
37         channel->getState(state);
38         
39         if ((m_last_channel_state != iDVBChannel::state_ok)
40                 && (state == iDVBChannel::state_ok) && (!m_demux))
41         {
42                 if (m_channel)
43                         if (m_channel->getDemux(m_demux, (!m_use_decode_demux) ? 0 : iDVBChannel::capDecode))
44                                 eDebug("Allocating %s-decoding a demux for now tuned-in channel failed.", m_use_decode_demux ? "" : "non-");
45                 
46                 serviceEvent(eventTuned);
47                 
48                 if (m_demux)
49                 {
50                         eDebug("ok ... now we start!!");
51
52                         if (m_pmt_pid == -1)
53                                 m_PAT.begin(eApp, eDVBPATSpec(), m_demux);
54                         else
55                                 m_PMT.begin(eApp, eDVBPMTSpec(m_pmt_pid, m_reference.getServiceID().get()), m_demux);
56
57                         if ( m_service && !m_service->cacheEmpty() )
58                                 serviceEvent(eventNewProgramInfo);
59                 }
60         } else if ((m_last_channel_state != iDVBChannel::state_failed) && 
61                         (state == iDVBChannel::state_failed))
62         {
63                 eDebug("tune failed.");
64                 serviceEvent(eventTuneFailed);
65         }
66 }
67
68 void eDVBServicePMTHandler::channelEvent(iDVBChannel *channel, int event)
69 {
70         switch (event)
71         {
72         case iDVBChannel::evtEOF:
73                 serviceEvent(eventEOF);
74                 break;
75         case iDVBChannel::evtSOF:
76                 serviceEvent(eventSOF);
77                 break;
78         default:
79                 break;
80         }
81 }
82
83 void eDVBServicePMTHandler::PMTready(int error)
84 {
85         if (error)
86                 serviceEvent(eventNoPMT);
87         else
88         {
89                 m_have_cached_program = false;
90                 serviceEvent(eventNewProgramInfo);
91                 if (!m_pvr_channel) // don't send campmt to camd.socket for playbacked services
92                 {
93                         eEPGCache::getInstance()->PMTready(this);
94                         if(!m_ca_servicePtr)
95                         {
96                                 int demuxes[2] = {0,0};
97                                 uint8_t tmp;
98                                 m_demux->getCADemuxID(tmp);
99                                 demuxes[0]=tmp;
100                                 if (m_decode_demux_num != 0xFF)
101                                         demuxes[1]=m_decode_demux_num;
102                                 else
103                                         demuxes[1]=demuxes[0];
104                                 eDVBCAService::register_service(m_reference, demuxes, m_ca_servicePtr);
105                                 eDVBCIInterfaces::getInstance()->recheckPMTHandlers();
106                         }
107                         eDVBCIInterfaces::getInstance()->gotPMT(this);
108                 }
109                 if (m_ca_servicePtr)
110                 {
111                         ePtr<eTable<ProgramMapSection> > ptr;
112                         if (!m_PMT.getCurrent(ptr))
113                                 m_ca_servicePtr->buildCAPMT(ptr);
114                         else
115                                 eDebug("eDVBServicePMTHandler cannot call buildCAPMT");
116                 }
117         }
118 }
119
120 void eDVBServicePMTHandler::PATready(int)
121 {
122         ePtr<eTable<ProgramAssociationSection> > ptr;
123         if (!m_PAT.getCurrent(ptr))
124         {
125                 int pmtpid = -1;
126                 std::vector<ProgramAssociationSection*>::const_iterator i;
127                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
128                 {
129                         const ProgramAssociationSection &pat = **i;
130                         ProgramAssociationConstIterator program;
131                         for (program = pat.getPrograms()->begin(); program != pat.getPrograms()->end(); ++program)
132                                 if (eServiceID((*program)->getProgramNumber()) == m_reference.getServiceID())
133                                         pmtpid = (*program)->getProgramMapPid();
134                 }
135                 if (pmtpid == -1)
136                         serviceEvent(eventNoPATEntry);
137                 else
138                         m_PMT.begin(eApp, eDVBPMTSpec(pmtpid, m_reference.getServiceID().get()), m_demux);
139         } else
140                 serviceEvent(eventNoPAT);
141 }
142
143 PyObject *eDVBServicePMTHandler::getCaIds()
144 {
145         ePyObject ret;
146
147         program prog;
148
149         if ( !getProgramInfo(prog) )
150         {
151                 int cnt=prog.caids.size();
152                 if (cnt)
153                 {
154                         ret=PyList_New(cnt);
155                         std::set<uint16_t>::iterator it(prog.caids.begin());
156                         while(cnt--)
157                                 PyList_SET_ITEM(ret, cnt, PyInt_FromLong(*it++));
158                 }
159         }
160
161         return ret ? (PyObject*)ret : (PyObject*)PyList_New(0);
162 }
163
164 int eDVBServicePMTHandler::getProgramInfo(struct program &program)
165 {
166         ePtr<eTable<ProgramMapSection> > ptr;
167         int cached_apid_ac3 = -1;
168         int cached_apid_mpeg = -1;
169         int cached_vpid = -1;
170         int cached_tpid = -1;
171         int ret = -1;
172
173         program.videoStreams.clear();
174         program.audioStreams.clear();
175         program.pcrPid = -1;
176         program.pmtPid = -1;
177         program.textPid = -1;
178
179         int first_ac3 = -1;
180         program.defaultAudioStream = 0;
181
182         if ( m_service && !m_service->cacheEmpty() )
183         {
184                 cached_vpid = m_service->getCacheEntry(eDVBService::cVPID);
185                 cached_apid_mpeg = m_service->getCacheEntry(eDVBService::cAPID);
186                 cached_apid_ac3 = m_service->getCacheEntry(eDVBService::cAC3PID);
187                 cached_tpid = m_service->getCacheEntry(eDVBService::cTPID);
188         }
189
190         if ( ((m_service && m_service->usePMT()) || !m_service) && !m_PMT.getCurrent(ptr))
191         {
192                 if (m_have_cached_program)
193                 {
194                         program = m_cached_program;
195                         ret = 0;
196                 }
197                 else
198                 {
199                         eDVBTableSpec table_spec;
200                         ptr->getSpec(table_spec);
201                         program.pmtPid = table_spec.pid < 0x1fff ? table_spec.pid : -1;
202                         std::vector<ProgramMapSection*>::const_iterator i;
203                         for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
204                         {
205                                 const ProgramMapSection &pmt = **i;
206                                 program.pcrPid = pmt.getPcrPid();
207                         
208                                 ElementaryStreamInfoConstIterator es;
209                                 for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
210                                 {
211                                         int isaudio = 0, isvideo = 0;
212                                         videoStream video;
213                                         audioStream audio;
214                                         audio.component_tag=video.component_tag=-1;
215                                         video.type = videoStream::vtMPEG2;
216
217                                         switch ((*es)->getType())
218                                         {
219                                         case 0x1b: // AVC Video Stream (MPEG4 H264)
220                                                 video.type = videoStream::vtMPEG4_H264;
221                                         case 0x01: // MPEG 1 video
222                                         case 0x02: // MPEG 2 video
223                                                 isvideo = 1;
224                                                 //break; fall through !!!
225                                         case 0x03: // MPEG 1 audio
226                                         case 0x04: // MPEG 2 audio:
227                                                 if (!isvideo)
228                                                 {
229                                                         isaudio = 1;
230                                                         audio.type = audioStream::atMPEG;
231                                                 }
232                                                 //break; fall through !!!
233                                         case 0x06: // PES Private
234                                         case 0x81: // user private
235                                                         /* PES private can contain AC-3, DTS or lots of other stuff.
236                                                            check descriptors to get the exact type. */
237                                                 for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
238                                                                 desc != (*es)->getDescriptors()->end(); ++desc)
239                                                 {
240                                                         switch ((*desc)->getTag())
241                                                         {
242                                                         case SUBTITLING_DESCRIPTOR:
243                                                         {
244                                                                 SubtitlingDescriptor *d = (SubtitlingDescriptor*)(*desc);
245                                                                 const SubtitlingList *list = d->getSubtitlings();
246                                                                 subtitleStream s;
247                                                                 s.pid = (*es)->getPid();
248                                                                 for (SubtitlingConstIterator it(list->begin()); it != list->end(); ++it)
249                                                                 {
250                                                                         s.subtitling_type = (*it)->getSubtitlingType();
251                                                                         s.composition_page_id = (*it)->getCompositionPageId();
252                                                                         s.ancillary_page_id = (*it)->getAncillaryPageId();
253                                                                         s.language_code = (*it)->getIso639LanguageCode();
254 //                                                                      eDebug("add dvb subtitle %s PID %04x, type %d, composition page %d, ancillary_page %d",
255 //                                                                              s.language_code.c_str(), s.pid, s.subtitling_type, s.composition_page_id, s.ancillary_page_id);
256                                                                         program.subtitleStreams.push_back(s);
257                                                                 }
258                                                                 break;
259                                                         }
260                                                         case TELETEXT_DESCRIPTOR:
261                                                                 if ( program.textPid == -1 || (*es)->getPid() == cached_tpid )
262                                                                 {
263                                                                         subtitleStream s;
264                                                                         s.subtitling_type = 0x01; // EBU TELETEXT SUBTITLES
265                                                                         s.pid = program.textPid = (*es)->getPid();
266                                                                         TeletextDescriptor *d = (TeletextDescriptor*)(*desc);
267                                                                         const VbiTeletextList *list = d->getVbiTeletexts();
268                                                                         for (VbiTeletextConstIterator it(list->begin()); it != list->end(); ++it)
269                                                                         {
270                                                                                 switch((*it)->getTeletextType())
271                                                                                 {
272                                                                                         case 0x02: // Teletext subtitle page
273                                                                                         case 0x05: // Teletext subtitle page for hearing impaired pepople
274                                                                                                 s.language_code = (*it)->getIso639LanguageCode();
275                                                                                                 s.teletext_page_number = (*it)->getTeletextPageNumber();
276                                                                                                 s.teletext_magazine_number = (*it)->getTeletextMagazineNumber();
277 //                                                                                              eDebug("add teletext subtitle %s PID %04x, page number %d, magazine number %d",
278 //                                                                                                      s.language_code.c_str(), s.pid, s.teletext_page_number, s.teletext_magazine_number);
279                                                                                                 program.subtitleStreams.push_back(s);
280                                                                                                 break;
281                                                                                 }
282                                                                         }
283                                                                 }
284                                                                 break;
285                                                         case DTS_DESCRIPTOR:
286                                                                 isaudio = 1;
287                                                                 audio.type = audioStream::atDTS;
288                                                                 break;
289                                                         case AAC_DESCRIPTOR:
290                                                                 isaudio = 1;
291                                                                 audio.type = audioStream::atAAC;
292                                                                 break;
293                                                         case AC3_DESCRIPTOR:
294                                                                 isaudio = 1;
295                                                                 audio.type = audioStream::atAC3;
296                                                                 break;
297                                                         case ISO_639_LANGUAGE_DESCRIPTOR:
298                                                                 if (!isvideo)
299                                                                 {
300                                                                         const Iso639LanguageList *languages = ((Iso639LanguageDescriptor*)*desc)->getIso639Languages();
301                                                                                 /* use last language code */
302                                                                         for (Iso639LanguageConstIterator i(languages->begin()); i != languages->end(); ++i)
303                                                                                 audio.language_code = (*i)->getIso639LanguageCode();
304                                                                 }
305                                                                 break;
306                                                         case STREAM_IDENTIFIER_DESCRIPTOR:
307                                                                 audio.component_tag =
308                                                                         video.component_tag =
309                                                                                 ((StreamIdentifierDescriptor*)*desc)->getComponentTag();
310                                                                 break;
311                                                         case CA_DESCRIPTOR:
312                                                         {
313                                                                 CaDescriptor *descr = (CaDescriptor*)(*desc);
314                                                                 program.caids.insert(descr->getCaSystemId());
315                                                                 break;
316                                                         }
317                                                         case REGISTRATION_DESCRIPTOR: /* some services don't have a separate AC3 descriptor */
318                                                         {
319                                                                         /* libdvbsi++ doesn't yet support this descriptor type, so work around. */
320                                                                 if ((*desc)->getLength() != 4)
321                                                                         break;
322                                                                 unsigned char descr[6];
323                                                                 (*desc)->writeToBuffer(descr);
324                                                                 int format_identifier = (descr[2] << 24) | (descr[3] << 16) | (descr[4] << 8) | (descr[5]);
325                                                                 switch (format_identifier)
326                                                                 {
327                                                                 case 0x41432d33:
328                                                                         isaudio = 1;
329                                                                         audio.type = audioStream::atAC3;
330                                                                         break;
331                                                                 default:
332                                                                         break;
333                                                                 }
334                                                                 break;
335                                                         }
336                                                         }
337                                                 }
338                                                 break;
339                                         }
340                                         if (isaudio)
341                                         {
342                                                 audio.pid = (*es)->getPid();
343
344                                                         /* if we find the cached pids, this will be our default stream */
345                                                 if (audio.pid == cached_apid_ac3 || audio.pid == cached_apid_mpeg)
346                                                         program.defaultAudioStream = program.audioStreams.size();
347
348                                                         /* also, we need to know the first non-mpeg (i.e. "ac3"/dts/...) stream */
349                                                 if ((audio.type != audioStream::atMPEG) && ((first_ac3 == -1) || (audio.pid == cached_apid_ac3)))
350                                                         first_ac3 = program.audioStreams.size();
351
352                                                 program.audioStreams.push_back(audio);
353                                         }
354                                         else if (isvideo)
355                                         {
356                                                 video.pid = (*es)->getPid();
357                                                 if ( !program.videoStreams.empty() && video.pid == cached_vpid )
358                                                 {
359                                                         program.videoStreams.push_back(program.videoStreams[0]);
360                                                         program.videoStreams[0] = video;
361                                                 }
362                                                 else
363                                                         program.videoStreams.push_back(video);
364                                         }
365                                         else
366                                                 continue;
367                                 }
368                                 for (DescriptorConstIterator desc = pmt.getDescriptors()->begin();
369                                         desc != pmt.getDescriptors()->end(); ++desc)
370                                 {
371                                         if ((*desc)->getTag() == CA_DESCRIPTOR)
372                                         {
373                                                 CaDescriptor *descr = (CaDescriptor*)(*desc);
374                                                 program.caids.insert(descr->getCaSystemId());
375                                         }
376                                 }
377                         }
378                         ret = 0;
379
380
381                         /* finally some fixup: if our default audio stream is an MPEG audio stream, 
382                            and we have 'defaultac3' set, use the first available ac3 stream instead.
383                            (note: if an ac3 audio stream was selected before, this will be also stored
384                            in 'fisrt_ac3', so we don't need to worry. */
385                         bool defaultac3 = false;
386                         std::string default_ac3;
387
388                         if (!ePythonConfigQuery::getConfigValue("config.av.defaultac3", default_ac3))
389                                 defaultac3 = default_ac3 == "True";
390
391                         if (defaultac3 && (first_ac3 != -1))
392                                 program.defaultAudioStream = first_ac3;
393
394                         m_cached_program = program;
395                         m_have_cached_program = true;
396                 }
397         } else if ( m_service && !m_service->cacheEmpty() )
398         {
399                 int cached_pcrpid = m_service->getCacheEntry(eDVBService::cPCRPID),
400                         vpidtype = m_service->getCacheEntry(eDVBService::cVTYPE),
401                         cnt=0;
402                 if ( vpidtype == -1 )
403                         vpidtype = videoStream::vtMPEG2;
404                 if ( cached_vpid != -1 )
405                 {
406                         videoStream s;
407                         s.pid = cached_vpid;
408                         s.type = vpidtype;
409                         program.videoStreams.push_back(s);
410                         ++cnt;
411                 }
412                 if ( cached_apid_ac3 != -1 )
413                 {
414                         audioStream s;
415                         s.type = audioStream::atAC3;
416                         s.pid = cached_apid_ac3;
417                         program.audioStreams.push_back(s);
418                         ++cnt;
419                 }
420                 if ( cached_apid_mpeg != -1 )
421                 {
422                         audioStream s;
423                         s.type = audioStream::atMPEG;
424                         s.pid = cached_apid_mpeg;
425                         program.audioStreams.push_back(s);
426                         ++cnt;
427                 }
428                 if ( cached_pcrpid != -1 )
429                 {
430                         ++cnt;
431                         program.pcrPid = cached_pcrpid;
432                 }
433                 if ( cached_tpid != -1 )
434                 {
435                         ++cnt;
436                         program.textPid = cached_tpid;
437                 }
438                 CAID_LIST &caids = m_service->m_ca;
439                 for (CAID_LIST::iterator it(caids.begin()); it != caids.end(); ++it)
440                         program.caids.insert(*it);
441                 if ( cnt )
442                         ret = 0;
443         }
444         return ret;
445 }
446
447 int eDVBServicePMTHandler::getChannel(eUsePtr<iDVBChannel> &channel)
448 {
449         channel = m_channel;
450         if (channel)
451                 return 0;
452         else
453                 return -1;
454 }
455
456 int eDVBServicePMTHandler::getDataDemux(ePtr<iDVBDemux> &demux)
457 {
458         demux = m_demux;
459         if (demux)
460                 return 0;
461         else
462                 return -1;
463 }
464
465 int eDVBServicePMTHandler::getDecodeDemux(ePtr<iDVBDemux> &demux)
466 {
467         int ret=0;
468                 /* if we're using the decoding demux as data source
469                    (for example in pvr playbacks), return that one. */
470         if (m_use_decode_demux)
471         {
472                 demux = m_demux;
473                 return ret;
474         }
475         
476         ASSERT(m_channel); /* calling without a previous ::tune is certainly bad. */
477
478         ret = m_channel->getDemux(demux, iDVBChannel::capDecode);
479         if (!ret)
480                 demux->getCADemuxID(m_decode_demux_num);
481
482         return ret;
483 }
484
485 int eDVBServicePMTHandler::getPVRChannel(ePtr<iDVBPVRChannel> &pvr_channel)
486 {
487         pvr_channel = m_pvr_channel;
488         if (pvr_channel)
489                 return 0;
490         else
491                 return -1;
492 }
493
494 void eDVBServicePMTHandler::SDTScanEvent(int event)
495 {
496         switch (event)
497         {
498                 case eDVBScan::evtFinish:
499                 {
500                         ePtr<iDVBChannelList> db;
501                         if (m_resourceManager->getChannelList(db) != 0)
502                                 eDebug("no channel list");
503                         else
504                         {
505                                 m_dvb_scan->insertInto(db, true);
506                                 eDebug("sdt update done!");
507                         }
508                         break;
509                 }
510
511                 default:
512                         break;
513         }
514 }
515
516 int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux, eCueSheet *cue)
517 {
518         RESULT res;
519         m_reference = ref;
520         
521         m_use_decode_demux = use_decode_demux;
522         
523                 /* is this a normal (non PVR) channel? */
524         if (ref.path.empty())
525         {
526                 eDVBChannelID chid;
527                 ref.getChannelID(chid);
528                 res = m_resourceManager->allocateChannel(chid, m_channel);
529                 eDebug("allocate Channel: res %d", res);
530
531                 ePtr<iDVBChannelList> db;
532                 if (!m_resourceManager->getChannelList(db))
533                                 db->getService((eServiceReferenceDVB&)m_reference, m_service);
534
535                 if (!res)
536                         eDVBCIInterfaces::getInstance()->addPMTHandler(this);
537         } else
538         {
539                 eDVBMetaParser parser;
540
541                 int ret=parser.parseFile(ref.path);
542                 if (ret || !parser.m_ref.getServiceID().get() /* incorrect sid in meta file or recordings.epl*/ )
543                 {
544                         eWarning("no .meta file found, trying to find PMT pid");
545                         eDVBTSTools tstools;
546                         if (tstools.openFile(ref.path.c_str()))
547                                 eWarning("failed to open file");
548                         else
549                         {
550                                 int service_id, pmt_pid;
551                                 if (!tstools.findPMT(pmt_pid, service_id))
552                                 {
553                                         eDebug("PMT pid found on pid %04x, service id %d", pmt_pid, service_id);
554                                         m_reference.setServiceID(service_id);
555                                         m_pmt_pid = pmt_pid;
556                                 }
557                         }
558                 } else
559                         m_reference = parser.m_ref;
560                 
561                 eDebug("alloc PVR");
562                         /* allocate PVR */
563                 res = m_resourceManager->allocatePVRChannel(m_pvr_channel);
564                 if (res)
565                         eDebug("allocatePVRChannel failed!\n");
566                 m_channel = m_pvr_channel;
567         }
568
569         if (m_channel)
570         {
571                 m_channel->connectStateChange(
572                         slot(*this, &eDVBServicePMTHandler::channelStateChanged), 
573                         m_channelStateChanged_connection);
574                 m_last_channel_state = -1;
575                 channelStateChanged(m_channel);
576
577                 m_channel->connectEvent(
578                         slot(*this, &eDVBServicePMTHandler::channelEvent), 
579                         m_channelEvent_connection);
580
581                 if (ref.path.empty())
582                 {
583                         delete m_dvb_scan;
584                         m_dvb_scan = new eDVBScan(m_channel, false, false);
585                         m_dvb_scan->connectEvent(slot(*this, &eDVBServicePMTHandler::SDTScanEvent), m_scan_event_connection);
586                 }
587         } else
588         {
589                 if (res == eDVBResourceManager::errAllSourcesBusy)
590                         serviceEvent(eventNoResources);
591                 else /* errChidNotFound, errNoChannelList, errChannelNotInList, errNoSourceFound */
592                         serviceEvent(eventMisconfiguration);
593                 return res;
594         }
595
596         if (m_pvr_channel)
597         {
598                 m_pvr_channel->setCueSheet(cue);
599                 m_pvr_channel->playFile(ref.path.c_str());
600         }
601
602         return res;
603 }
604
605 void eDVBServicePMTHandler::free()
606 {
607         m_dvb_scan = 0;
608         delete m_dvb_scan;
609
610         if (m_ca_servicePtr)
611         {
612                 int demuxes[2] = {0,0};
613                 uint8_t tmp;
614                 m_demux->getCADemuxID(tmp);
615                 demuxes[0]=tmp;
616                 if (m_decode_demux_num != 0xFF)
617                         demuxes[1]=m_decode_demux_num;
618                 else
619                         demuxes[1]=demuxes[0];
620                 ePtr<eTable<ProgramMapSection> > ptr;
621                 m_PMT.getCurrent(ptr);
622                 eDVBCAService::unregister_service(m_reference, demuxes, ptr);
623                 m_ca_servicePtr = 0;
624         }
625
626         if (m_channel)
627                 eDVBCIInterfaces::getInstance()->removePMTHandler(this);
628
629         if (m_pvr_channel)
630         {
631                 m_pvr_channel->stopFile();
632                 m_pvr_channel->setCueSheet(0);
633         }
634         m_PMT.stop();
635         m_PAT.stop();
636         m_service = 0;
637         m_channel = 0;
638         m_pvr_channel = 0;
639         m_demux = 0;
640 }
641
642 CAServiceMap eDVBCAService::exist;
643 ChannelMap eDVBCAService::exist_channels;
644 ePtr<eConnection> eDVBCAService::m_chanAddedConn;
645
646 eDVBCAService::eDVBCAService()
647         :m_sn(0), m_prev_build_hash(0), m_sendstate(0), m_retryTimer(eApp)
648 {
649         memset(m_used_demux, 0xFF, sizeof(m_used_demux));
650         CONNECT(m_retryTimer.timeout, eDVBCAService::sendCAPMT);
651         Connect();
652 }
653
654 eDVBCAService::~eDVBCAService()
655 {
656         eDebug("[eDVBCAService] free service %s", m_service.toString().c_str());
657         ::close(m_sock);
658         delete m_sn;
659 }
660
661 // begin static methods
662 RESULT eDVBCAService::register_service( const eServiceReferenceDVB &ref, int demux_nums[2], eDVBCAService *&caservice )
663 {
664         CAServiceMap::iterator it = exist.find(ref);
665         if ( it != exist.end() )
666                 caservice = it->second;
667         else
668         {
669                 caservice = (exist[ref]=new eDVBCAService());
670                 caservice->m_service = ref;
671                 eDebug("[eDVBCAService] new service %s", ref.toString().c_str() );
672         }
673
674         int loops = demux_nums[0] != demux_nums[1] ? 2 : 1;
675         for (int i=0; i < loops; ++i)
676         {
677 // search free demux entry
678                 int iter=0, max_demux_slots = sizeof(caservice->m_used_demux);
679
680                 while ( iter < max_demux_slots && caservice->m_used_demux[iter] != 0xFF )
681                         ++iter;
682
683                 if ( iter < max_demux_slots )
684                 {
685                         caservice->m_used_demux[iter] = demux_nums[i] & 0xFF;
686                         eDebug("[eDVBCAService] add demux %d to slot %d service %s", caservice->m_used_demux[iter], iter, ref.toString().c_str());
687                 }
688                 else
689                 {
690                         eDebug("[eDVBCAService] no more demux slots free for service %s!!", ref.toString().c_str());
691                         return -1;
692                 }
693         }
694         return 0;
695 }
696
697 RESULT eDVBCAService::unregister_service( const eServiceReferenceDVB &ref, int demux_nums[2], eTable<ProgramMapSection> *ptr )
698 {
699         CAServiceMap::iterator it = exist.find(ref);
700         if ( it == exist.end() )
701         {
702                 eDebug("[eDVBCAService] try to unregister non registered %s", ref.toString().c_str());
703                 return -1;
704         }
705         else
706         {
707                 eDVBCAService *caservice = it->second;
708                 int loops = demux_nums[0] != demux_nums[1] ? 2 : 1;
709                 for (int i=0; i < loops; ++i)
710                 {
711                         bool freed = false;
712                         int iter = 0,
713                                 used_demux_slots = 0,
714                                 max_demux_slots = sizeof(caservice->m_used_demux)/sizeof(int);
715                         while ( iter < max_demux_slots )
716                         {
717                                 if ( caservice->m_used_demux[iter] != 0xFF )
718                                 {
719                                         if ( !freed && caservice->m_used_demux[iter] == demux_nums[i] )
720                                         {
721                                                 eDebug("[eDVBCAService] free slot %d demux %d for service %s", iter, caservice->m_used_demux[iter], caservice->m_service.toString().c_str() );
722                                                 caservice->m_used_demux[iter] = 0xFF;
723                                                 freed=true;
724                                         }
725                                         else
726                                                 ++used_demux_slots;
727                                 }
728                                 ++iter;
729                         }
730                         if (!freed)
731                                 eDebug("[eDVBCAService] couldn't free demux slot for demux %d", demux_nums[i]);
732                         if (i || loops == 1)
733                         {
734                                 if (!used_demux_slots)  // no more used.. so we remove it
735                                 {
736                                         delete it->second;
737                                         exist.erase(it);
738                                 }
739                                 else
740                                 {
741                                         if (ptr)
742                                                 it->second->buildCAPMT(ptr);
743                                         else
744                                                 eDebug("[eDVBCAService] can not send updated demux info");
745                                 }
746                         }
747                 }
748         }
749         return 0;
750 }
751
752 void eDVBCAService::registerChannelCallback(eDVBResourceManager *res_mgr)
753 {
754         res_mgr->connectChannelAdded(slot(&DVBChannelAdded), m_chanAddedConn);
755 }
756
757 void eDVBCAService::DVBChannelAdded(eDVBChannel *chan)
758 {
759         if ( chan )
760         {
761                 eDebug("[eDVBCAService] new channel %p!", chan);
762                 channel_data *data = new channel_data();
763                 data->m_channel = chan;
764                 data->m_prevChannelState = -1;
765                 data->m_dataDemux = -1;
766                 exist_channels[chan] = data;
767                 chan->connectStateChange(slot(&DVBChannelStateChanged), data->m_stateChangedConn);
768         }
769 }
770
771 void eDVBCAService::DVBChannelStateChanged(iDVBChannel *chan)
772 {
773         ChannelMap::iterator it =
774                 exist_channels.find(chan);
775         if ( it != exist_channels.end() )
776         {
777                 int state=0;
778                 chan->getState(state);
779                 if ( it->second->m_prevChannelState != state )
780                 {
781                         switch (state)
782                         {
783                                 case iDVBChannel::state_ok:
784                                 {
785                                         eDebug("[eDVBCAService] channel %p running", chan);
786                                         break;
787                                 }
788                                 case iDVBChannel::state_release:
789                                 {
790                                         eDebug("[eDVBCAService] remove channel %p", chan);
791                                         unsigned char msg[8] = { 0x9f,0x80,0x3f,0x04,0x83,0x02,0x00,0x00 };
792                                         msg[7] = it->second->m_dataDemux & 0xFF;
793                                         int sock, clilen;
794                                         struct sockaddr_un servaddr;
795                                         memset(&servaddr, 0, sizeof(struct sockaddr_un));
796                                         servaddr.sun_family = AF_UNIX;
797                                         strcpy(servaddr.sun_path, "/tmp/camd.socket");
798                                         clilen = sizeof(servaddr.sun_family) + strlen(servaddr.sun_path);
799                                         sock = socket(PF_UNIX, SOCK_STREAM, 0);
800                                         if (sock > -1)
801                                         {
802                                                 connect(sock, (struct sockaddr *) &servaddr, clilen);
803                                                 fcntl(sock, F_SETFL, O_NONBLOCK);
804                                                 int val=1;
805                                                 setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &val, 4);
806                                                 if (write(sock, msg, 8) != 8)
807                                                         eDebug("[eDVBCAService] write leave transponder failed!!");
808                                                 close(sock);
809                                         }
810                                         exist_channels.erase(it);
811                                         delete it->second;
812                                         it->second=0;
813                                         break;
814                                 }
815                                 default: // ignore all other events
816                                         return;
817                         }
818                         if (it->second)
819                                 it->second->m_prevChannelState = state;
820                 }
821         }
822 }
823
824 channel_data *eDVBCAService::getChannelData(eDVBChannelID &chid)
825 {
826         for (ChannelMap::iterator it(exist_channels.begin()); it != exist_channels.end(); ++it)
827         {
828                 if (chid == it->second->m_channel->getChannelID())
829                         return it->second;
830         }
831         return 0;
832 }
833 // end static methods
834
835 void eDVBCAService::socketCB(int what)
836 {
837         if (what & eSocketNotifier::Read)
838                 /*eDebug("[eDVBCAService] data to read\n")*/;
839         if (what & eSocketNotifier::Priority)
840                 /*eDebug("[eDVBCAService] priority data to read\n")*/;
841         if (what & eSocketNotifier::Hungup) {
842                 /*eDebug("[eDVBCAService] connection closed\n")*/;
843                 m_sendstate=1;
844                 sendCAPMT();
845         }
846         if (what & eSocketNotifier::Error)
847                 /*eDebug("[eDVBCAService] connection error\n")*/;
848 }
849
850 void eDVBCAService::Connect()
851 {
852         if (m_sn) {
853                 delete m_sn;
854                 m_sn=0;
855         }
856         memset(&m_servaddr, 0, sizeof(struct sockaddr_un));
857         m_servaddr.sun_family = AF_UNIX;
858         strcpy(m_servaddr.sun_path, "/tmp/camd.socket");
859         m_clilen = sizeof(m_servaddr.sun_family) + strlen(m_servaddr.sun_path);
860         m_sock = socket(PF_UNIX, SOCK_STREAM, 0);
861         if (m_sock != -1)
862         {
863                 if (!connect(m_sock, (struct sockaddr *) &m_servaddr, m_clilen))
864                 {
865                         int val=1;
866                         fcntl(m_sock, F_SETFL, O_NONBLOCK);
867                         setsockopt(m_sock, SOL_SOCKET, SO_REUSEADDR, &val, 4);
868                         m_sn = new eSocketNotifier(eApp, m_sock,
869                                 eSocketNotifier::Read|eSocketNotifier::Priority|eSocketNotifier::Error|eSocketNotifier::Hungup);
870                         CONNECT(m_sn->activated, eDVBCAService::socketCB);
871                         
872                 }
873 //              else
874 //                      eDebug("[eDVBCAService] connect failed %m");
875         }
876         else
877                 eDebug("[eDVBCAService] create socket failed %m");
878 }
879
880 void eDVBCAService::buildCAPMT(eTable<ProgramMapSection> *ptr)
881 {
882         if (!ptr)
883                 return;
884
885         eDVBTableSpec table_spec;
886         ptr->getSpec(table_spec);
887
888         int pmtpid = table_spec.pid,
889                 pmt_version = table_spec.version;
890
891         uint8_t demux_mask = 0;
892         int data_demux = -1;
893
894         int iter=0, max_demux_slots = sizeof(m_used_demux);
895         while ( iter < max_demux_slots )
896         {
897                 if ( m_used_demux[iter] != 0xFF )
898                 {
899                         if ( m_used_demux[iter] > data_demux )
900                                 data_demux = m_used_demux[iter];
901                         demux_mask |= (1 << m_used_demux[iter]);
902                 }
903                 ++iter;
904         }
905
906         if ( data_demux == -1 )
907         {
908                 eDebug("[eDVBCAService] no data demux found for service %s", m_service.toString().c_str() );
909                 return;
910         }
911
912         eDebug("demux %d mask %02x prevhash %08x", data_demux, demux_mask, m_prev_build_hash);
913
914         unsigned int build_hash = ( pmtpid << 16);
915         build_hash |= (demux_mask << 8);
916         build_hash |= (pmt_version&0xFF);
917
918         if ( build_hash == m_prev_build_hash )
919         {
920                 eDebug("[eDVBCAService] don't build/send the same CA PMT twice");
921                 return;
922         }
923
924         std::vector<ProgramMapSection*>::const_iterator i=ptr->getSections().begin();
925         if ( i != ptr->getSections().end() )
926         {
927                 CaProgramMapSection capmt(*i++, m_prev_build_hash ? 0x05 /*update*/ : 0x03 /*only*/, 0x01 );
928
929                 while( i != ptr->getSections().end() )
930                 {
931 //                      eDebug("append");
932                         capmt.append(*i++);
933                 }
934
935                 // add our private descriptors to capmt
936                 uint8_t tmp[10];
937
938                 tmp[0]=0x84;  // pmt pid
939                 tmp[1]=0x02;
940                 tmp[2]=pmtpid>>8;
941                 tmp[3]=pmtpid&0xFF;
942                 capmt.injectDescriptor(tmp, false);
943
944                 tmp[0] = 0x82; // demux
945                 tmp[1] = 0x02;
946                 tmp[2] = demux_mask;    // descramble bitmask
947                 tmp[3] = data_demux&0xFF; // read section data from demux number
948                 capmt.injectDescriptor(tmp, false);
949
950                 tmp[0] = 0x81; // dvbnamespace
951                 tmp[1] = 0x08;
952                 tmp[2] = m_service.getDVBNamespace().get()>>24;
953                 tmp[3]=(m_service.getDVBNamespace().get()>>16)&0xFF;
954                 tmp[4]=(m_service.getDVBNamespace().get()>>8)&0xFF;
955                 tmp[5]=m_service.getDVBNamespace().get()&0xFF;
956                 tmp[6]=m_service.getTransportStreamID().get()>>8;
957                 tmp[7]=m_service.getTransportStreamID().get()&0xFF;
958                 tmp[8]=m_service.getOriginalNetworkID().get()>>8;
959                 tmp[9]=m_service.getOriginalNetworkID().get()&0xFF;
960                 capmt.injectDescriptor(tmp, false);
961
962                 capmt.writeToBuffer(m_capmt);
963         }
964
965         m_prev_build_hash = build_hash;
966
967         if ( m_sendstate != 0xFFFFFFFF )
968                 m_sendstate=0;
969         sendCAPMT();
970 }
971
972 void eDVBCAService::sendCAPMT()
973 {
974         if ( m_sendstate && m_sendstate != 0xFFFFFFFF ) // broken pipe retry
975         {
976                 ::close(m_sock);
977                 Connect();
978         }
979
980         int wp=0;
981         if ( m_capmt[3] & 0x80 )
982         {
983                 int i=0;
984                 int lenbytes = m_capmt[3] & ~0x80;
985                 while(i < lenbytes)
986                         wp = (wp << 8) | m_capmt[4 + i++];
987                 wp+=4;
988                 wp+=lenbytes;
989         }
990         else
991         {
992                 wp = m_capmt[3];
993                 wp+=4;
994         }
995
996         if ( write(m_sock, m_capmt, wp) == wp )
997         {
998                 m_sendstate=0xFFFFFFFF;
999                 eDebug("[eDVBCAService] send %d bytes",wp);
1000                 eDVBChannelID chid;
1001                 m_service.getChannelID(chid);
1002                 channel_data *data = getChannelData(chid);
1003                 if (data)
1004                 {
1005                         int lenbytes = m_capmt[3] & 0x80 ? m_capmt[3] & ~0x80 : 0;
1006                         data->m_dataDemux = m_capmt[24+lenbytes];
1007                 }
1008 #if 1
1009                 for(int i=0;i<wp;i++)
1010                         eDebugNoNewLine("%02x ", m_capmt[i]);
1011                 eDebug("");
1012 #endif
1013         }
1014         else
1015         {
1016                 switch(m_sendstate)
1017                 {
1018                         case 0xFFFFFFFF:
1019                                 ++m_sendstate;
1020                                 m_retryTimer.start(0,true);
1021 //                              eDebug("[eDVBCAService] send failed .. immediate retry");
1022                                 break;
1023                         default:
1024                                 m_retryTimer.start(5000,true);
1025 //                              eDebug("[eDVBCAService] send failed .. retry in 5 sec");
1026                                 break;
1027                 }
1028                 ++m_sendstate;
1029         }
1030 }
1031
1032 static PyObject *createTuple(int pid, const char *type)
1033 {
1034         PyObject *r = PyTuple_New(2);
1035         PyTuple_SetItem(r, 0, PyInt_FromLong(pid));
1036         PyTuple_SetItem(r, 1, PyString_FromString(type));
1037         return r;
1038 }
1039
1040 static inline void PyList_AppendSteal(PyObject *list, PyObject *item)
1041 {
1042         PyList_Append(list, item);
1043         Py_DECREF(item);
1044 }
1045
1046 PyObject *eDVBServicePMTHandler::program::createPythonObject()
1047 {
1048         PyObject *r = PyDict_New();
1049
1050         PyObject *l = PyList_New(0);
1051         
1052         PyList_AppendSteal(l, createTuple(0, "pat"));
1053
1054         if (pmtPid != -1)
1055                 PyList_AppendSteal(l, createTuple(pmtPid, "pmt"));
1056         
1057         for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
1058                         i(videoStreams.begin()); 
1059                         i != videoStreams.end(); ++i)
1060                 PyList_AppendSteal(l, createTuple(i->pid, "video"));
1061
1062         for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
1063                         i(audioStreams.begin()); 
1064                         i != audioStreams.end(); ++i)
1065                 PyList_AppendSteal(l, createTuple(i->pid, "audio"));
1066
1067         for (std::vector<eDVBServicePMTHandler::subtitleStream>::const_iterator
1068                         i(subtitleStreams.begin());
1069                         i != subtitleStreams.end(); ++i)
1070                 PyList_AppendSteal(l, createTuple(i->pid, "subtitle"));
1071
1072         PyList_AppendSteal(l, createTuple(pcrPid, "pcr"));
1073
1074         if (textPid != -1)
1075                 PyList_AppendSteal(l, createTuple(textPid, "text"));
1076                 
1077         PyDict_SetItemString(r, "pids", l);
1078         return r;
1079 }