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