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