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 <dvbsi++/ca_program_map_section.h>
   8 #include <dvbsi++/descriptor_tag.h>
   9
  10 eDVBServicePMTHandler::eDVBServicePMTHandler(int record)
  11         :m_ca_servicePtr(0)
  12 {
  13         m_record = record;
  14         eDVBResourceManager::getInstance(m_resourceManager);
  15         CONNECT(m_PMT.tableReady, eDVBServicePMTHandler::PMTready);
  16         CONNECT(m_PAT.tableReady, eDVBServicePMTHandler::PATready);
  17         eDebug("new PMT handler record: %d", m_record);
  18 }
  19
  20 eDVBServicePMTHandler::~eDVBServicePMTHandler()
  21 {
  22         eDebug("delete PMT handler record: %d", m_record);
  23         if (m_ca_servicePtr)
  24         {
  25                 eDebug("unregister caservice");
  26                 uint8_t demux_num;
  27                 m_demux->getCADemuxID(demux_num);
  28                 ePtr<eTable<ProgramMapSection> > ptr;
  29                 m_PMT.getCurrent(ptr);
  30                 eDVBCAService::unregister_service(m_reference, demux_num, ptr);
  31                 eDVBCIInterfaces::getInstance()->removePMTHandler(this);
  32         }
  33 }
  34
  35 void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
  36 {
  37         int state;
  38         channel->getState(state);
  39
  40         if ((m_last_channel_state != iDVBChannel::state_ok)
  41                 && (state == iDVBChannel::state_ok) && (!m_demux))
  42         {
  43                 if (m_channel)
  44                         if (m_channel->getDemux(m_demux, m_record ? 0 : iDVBChannel::capDecode))
  45                                 eDebug("Allocating a demux for now tuned-in channel failed.");
  46
  47                 serviceEvent(eventTuned);
  48
  49                 if (m_demux)
  50                 {
  51                         eDebug("ok ... now we start!!");
  52
  53                         /* emit */ m_resourceManager->m_channelRunning(channel);
  54
  55                         m_PAT.begin(eApp, eDVBPATSpec(), 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::PMTready(int error)
  69 {
  70         if (error)
  71                 serviceEvent(eventNoPMT);
  72         else
  73         {
  74                 serviceEvent(eventNewProgramInfo);
  75                 if (!m_pvr_channel)
  76                 {
  77                         if(!m_ca_servicePtr)   // don't send campmt to camd.socket for playbacked services
  78                         {
  79                                 uint8_t demux_num;
  80                                 m_demux->getCADemuxID(demux_num);
  81                                 eDVBCAService::register_service(m_reference, demux_num, m_ca_servicePtr);
  82                                 eDVBCIInterfaces::getInstance()->addPMTHandler(this);
  83                         }
  84                         eDVBCIInterfaces::getInstance()->gotPMT(this);
  85                 }
  86                 if (m_ca_servicePtr)
  87                 {
  88                         ePtr<eTable<ProgramMapSection> > ptr;
  89                         if (!m_PMT.getCurrent(ptr))
  90                                 m_ca_servicePtr->buildCAPMT(ptr);
  91                         else
  92                                 eDebug("eDVBServicePMTHandler cannot call buildCAPMT");
  93                 }
  94         }
  95 }
  96
  97 void eDVBServicePMTHandler::PATready(int)
  98 {
  99         eDebug("got PAT");
 100         ePtr<eTable<ProgramAssociationSection> > ptr;
 101         if (!m_PAT.getCurrent(ptr))
 102         {
 103                 int pmtpid = -1;
 104                 std::vector<ProgramAssociationSection*>::const_iterator i;
 105                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
 106                 {
 107                         const ProgramAssociationSection &pat = **i;
 108                         ProgramAssociationConstIterator program;
 109                         for (program = pat.getPrograms()->begin(); program != pat.getPrograms()->end(); ++program)
 110                                 if (eServiceID((*program)->getProgramNumber()) == m_reference.getServiceID())
 111                                         pmtpid = (*program)->getProgramMapPid();
 112                 }
 113                 if (pmtpid == -1)
 114                         serviceEvent(eventNoPATEntry);
 115                 else
 116                         m_PMT.begin(eApp, eDVBPMTSpec(pmtpid, m_reference.getServiceID().get()), m_demux);
 117         } else
 118                 serviceEvent(eventNoPAT);
 119 }
 120
 121 int eDVBServicePMTHandler::getProgramInfo(struct program &program)
 122 {
 123         ePtr<eTable<ProgramMapSection> > ptr;
 124
 125         program.videoStreams.clear();
 126         program.audioStreams.clear();
 127         program.pcrPid = -1;
 128
 129         if (!m_PMT.getCurrent(ptr))
 130         {
 131                 eDVBTableSpec table_spec;
 132                 ptr->getSpec(table_spec);
 133                 program.pmtPid = table_spec.pid < 0x1fff ? table_spec.pid : -1;
 134                 std::vector<ProgramMapSection*>::const_iterator i;
 135                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
 136                 {
 137                         const ProgramMapSection &pmt = **i;
 138                         program.pcrPid = pmt.getPcrPid();
 139
 140                         ElementaryStreamInfoConstIterator es;
 141                         for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
 142                         {
 143                                 int isaudio = 0, isvideo = 0;
 144                                 videoStream video;
 145                                 audioStream audio;
 146
 147                                 video.pid = (*es)->getPid();
 148                                 audio.pid = (*es)->getPid();
 149
 150                                 switch ((*es)->getType())
 151                                 {
 152                                 case 0x01: // MPEG 1 video
 153                                 case 0x02: // MPEG 2 video
 154                                         isvideo = 1;
 155                                         break;
 156                                 case 0x03: // MPEG 1 audio
 157                                 case 0x04: // MPEG 2 audio:
 158                                         isaudio = 1;
 159                                         audio.type = audioStream::atMPEG;
 160                                         break;
 161                                 case 0x06: // PES Private
 162                                                 /* PES private can contain AC-3, DTS or lots of other stuff.
 163                                                    check descriptors to get the exact type. */
 164                                         for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
 165                                                         desc != (*es)->getDescriptors()->end(); ++desc)
 166                                         {
 167                                                 switch ((*desc)->getTag())
 168                                                 {
 169                                                 case AC3_DESCRIPTOR:
 170                                                         isaudio = 1;
 171                                                         audio.type = audioStream::atAC3;
 172                                                         break;
 173                                                 }
 174                                         }
 175                                         break;
 176                                 }
 177                                 if (isaudio)
 178                                         program.audioStreams.push_back(audio);
 179                                 if (isvideo)
 180                                         program.videoStreams.push_back(video);
 181                         }
 182                 }
 183                 return 0;
 184         }
 185         else if ( m_service && !m_service->cacheEmpty() )
 186         {
 187                 int vpid = m_service->getCachePID(eDVBService::cVPID),
 188                         apid_ac3 = m_service->getCachePID(eDVBService::cAPID),
 189                         apid_mpeg = m_service->getCachePID(eDVBService::cAC3PID),
 190                         pcrpid = m_service->getCachePID(eDVBService::cPCRPID),
 191                         cnt=0;
 192                 if ( vpid != -1 )
 193                 {
 194                         videoStream s;
 195                         s.pid = vpid;
 196                         program.videoStreams.push_back(s);
 197                         ++cnt;
 198                 }
 199                 if ( apid_ac3 != -1 )
 200                 {
 201                         audioStream s;
 202                         s.type = audioStream::atAC3;
 203                         s.pid = apid_ac3;
 204                         program.audioStreams.push_back(s);
 205                         ++cnt;
 206                 }
 207                 if ( apid_mpeg != -1 )
 208                 {
 209                         audioStream s;
 210                         s.type = audioStream::atMPEG;
 211                         s.pid = apid_mpeg;
 212                         program.audioStreams.push_back(s);
 213                         ++cnt;
 214                 }
 215                 if ( pcrpid != -1 )
 216                 {
 217                         ++cnt;
 218                         program.pcrPid = pcrpid;
 219                 }
 220                 if ( cnt )
 221                         return 0;
 222         }
 223         return -1;
 224 }
 225
 226 int eDVBServicePMTHandler::getChannel(eUsePtr<iDVBChannel> &channel)
 227 {
 228         channel = m_channel;
 229         if (channel)
 230                 return 0;
 231         else
 232                 return -1;
 233 }
 234
 235 int eDVBServicePMTHandler::getDemux(ePtr<iDVBDemux> &demux)
 236 {
 237         demux = m_demux;
 238         if (demux)
 239                 return 0;
 240         else
 241                 return -1;
 242 }
 243
 244 int eDVBServicePMTHandler::getPVRChannel(ePtr<iDVBPVRChannel> &pvr_channel)
 245 {
 246         pvr_channel = m_pvr_channel;
 247         if (pvr_channel)
 248                 return 0;
 249         else
 250                 return -1;
 251 }
 252
 253 int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref)
 254 {
 255         RESULT res;
 256         m_reference = ref;
 257
 258                 /* is this a normal (non PVR) channel? */
 259         if (ref.path.empty())
 260         {
 261                 eDVBChannelID chid;
 262                 ref.getChannelID(chid);
 263                 res = m_resourceManager->allocateChannel(chid, m_channel);
 264                 eDebug("allocate Channel: res %d", res);
 265         } else
 266         {
 267                 eDVBMetaParser parser;
 268
 269                 if (parser.parseFile(ref.path))
 270                         eWarning("no .meta file found, trying original service ref.");
 271                 else
 272                         m_reference = parser.m_ref;
 273
 274                 eDebug("alloc PVR");
 275                         /* allocate PVR */
 276                 res = m_resourceManager->allocatePVRChannel(m_pvr_channel);
 277                 if (res)
 278                         eDebug("allocatePVRChannel failed!\n");
 279                 m_channel = m_pvr_channel;
 280         }
 281
 282         if (m_channel)
 283         {
 284                 m_channel->connectStateChange(
 285                         slot(*this, &eDVBServicePMTHandler::channelStateChanged),
 286                         m_channelStateChanged_connection);
 287                 m_last_channel_state = -1;
 288                 channelStateChanged(m_channel);
 289         } else
 290         {
 291                 serviceEvent(eventTuneFailed);
 292                 return res;
 293         }
 294
 295         if (m_pvr_channel)
 296                 m_pvr_channel->playFile(ref.path.c_str());
 297
 298         ePtr<iDVBChannelList> db;
 299         if (!m_resourceManager->getChannelList(db))
 300                 db->getService((eServiceReferenceDVB&)m_reference, m_service);
 301
 302         return res;
 303 }
 304
 305 std::map<eServiceReferenceDVB, eDVBCAService*> eDVBCAService::exist;
 306
 307 eDVBCAService::eDVBCAService()
 308         :m_prev_build_hash(0), m_sendstate(0), m_retryTimer(eApp)
 309 {
 310         memset(m_used_demux, 0xFF, sizeof(m_used_demux));
 311         CONNECT(m_retryTimer.timeout, eDVBCAService::sendCAPMT);
 312         Connect();
 313 }
 314
 315 eDVBCAService::~eDVBCAService()
 316 {
 317         eDebug("[eDVBCAService] free service %s", m_service.toString().c_str());
 318         ::close(m_sock);
 319 }
 320
 321 RESULT eDVBCAService::register_service( const eServiceReferenceDVB &ref, int demux_num, eDVBCAService *&caservice )
 322 {
 323         CAServiceMap::iterator it = exist.find(ref);
 324         if ( it != exist.end() )
 325                 caservice = it->second;
 326         else
 327         {
 328                 caservice = (exist[ref]=new eDVBCAService());
 329                 caservice->m_service = ref;
 330                 eDebug("[eDVBCAHandler] new service %s", ref.toString().c_str() );
 331         }
 332 // search free demux entry
 333         int iter=0, max_demux_slots = sizeof(caservice->m_used_demux);
 334
 335         while ( iter < max_demux_slots && caservice->m_used_demux[iter] != 0xFF )
 336                 ++iter;
 337
 338         if ( iter < max_demux_slots )
 339         {
 340                 caservice->m_used_demux[iter] = demux_num & 0xFF;
 341                 eDebug("[eDVBCAHandler] add demux %d to slot %d service %s", demux_num, iter, ref.toString().c_str());
 342         }
 343         else
 344         {
 345                 eDebug("[eDVBCAHandler] no more demux slots free for service %s!!", ref.toString().c_str());
 346                 return -1;
 347         }
 348         return 0;
 349 }
 350
 351 RESULT eDVBCAService::unregister_service( const eServiceReferenceDVB &ref, int demux_num, eTable<ProgramMapSection> *ptr )
 352 {
 353         CAServiceMap::iterator it = exist.find(ref);
 354         if ( it == exist.end() )
 355         {
 356                 eDebug("[eDVBCAHandler] try to unregister non registered %s", ref.toString().c_str());
 357                 return -1;
 358         }
 359         else
 360         {
 361                 eDVBCAService *caservice = it->second;
 362                 bool freed = false;
 363                 int iter = 0,
 364                         used_demux_slots = 0,
 365                         max_demux_slots = sizeof(caservice->m_used_demux)/sizeof(int);
 366                 while ( iter < max_demux_slots )
 367                 {
 368                         if ( caservice->m_used_demux[iter] != 0xFF )
 369                         {
 370                                 if ( !freed && caservice->m_used_demux[iter] == demux_num )
 371                                 {
 372                                         eDebug("[eDVBCAService] free slot %d demux %d for service %s", iter, caservice->m_used_demux[iter], caservice->m_service.toString().c_str() );
 373                                         caservice->m_used_demux[iter] = 0xFF;
 374                                         freed=true;
 375                                 }
 376                                 else
 377                                         ++used_demux_slots;
 378                         }
 379                         ++iter;
 380                 }
 381                 if (!freed)
 382                 {
 383                         eDebug("[eDVBCAService] couldn't free demux slot for demux %d", demux_num);
 384                         return -1;
 385                 }
 386                 if (!used_demux_slots)  // no more used.. so we remove it
 387                 {
 388                         delete it->second;
 389                         exist.erase(it);
 390                 }
 391                 else
 392                 {
 393                         if (ptr)
 394                                 it->second->buildCAPMT(ptr);
 395                         else
 396                                 eDebug("[eDVBCAHandler] can not send updated demux info");
 397                 }
 398         }
 399         return 0;
 400 }
 401
 402 void eDVBCAService::Connect()
 403 {
 404         memset(&m_servaddr, 0, sizeof(struct sockaddr_un));
 405         m_servaddr.sun_family = AF_UNIX;
 406         strcpy(m_servaddr.sun_path, "/tmp/camd.socket");
 407         m_clilen = sizeof(m_servaddr.sun_family) + strlen(m_servaddr.sun_path);
 408         m_sock = socket(PF_UNIX, SOCK_STREAM, 0);
 409         connect(m_sock, (struct sockaddr *) &m_servaddr, m_clilen);
 410         fcntl(m_sock, F_SETFL, O_NONBLOCK);
 411         int val=1;
 412         setsockopt(m_sock, SOL_SOCKET, SO_REUSEADDR, &val, 4);
 413 }
 414
 415 void eDVBCAService::buildCAPMT(eTable<ProgramMapSection> *ptr)
 416 {
 417         if (!ptr)
 418                 return;
 419
 420         eDVBTableSpec table_spec;
 421         ptr->getSpec(table_spec);
 422
 423         int pmtpid = table_spec.pid,
 424                 pmt_version = table_spec.version;
 425
 426         uint8_t demux_mask = 0;
 427         uint8_t first_demux_num = 0xFF;
 428
 429 #if 1
 430         int iter=0, max_demux_slots = sizeof(m_used_demux);
 431         while ( iter < max_demux_slots )
 432         {
 433                 if ( m_used_demux[iter] != 0xFF )
 434                 {
 435                         if ( first_demux_num == 0xFF )
 436                                 first_demux_num = m_used_demux[iter];
 437                         demux_mask |= (1 << m_used_demux[iter]);
 438                 }
 439                 ++iter;
 440         }
 441 #else
 442         demux_mask = 3;
 443         first_demux_num = 0;
 444 #endif
 445
 446         if ( first_demux_num == 0xFF )
 447         {
 448                 eDebug("[eDVBCAService] no demux found for service %s", m_service.toString().c_str() );
 449                 return;
 450         }
 451
 452         eDebug("demux %d mask %02x prevhash %08x", first_demux_num, demux_mask, m_prev_build_hash);
 453
 454         unsigned int build_hash = (pmtpid << 16);
 455         build_hash |= (demux_mask << 8);
 456         build_hash |= (pmt_version&0xFF);
 457
 458         if ( build_hash == m_prev_build_hash )
 459         {
 460                 eDebug("[eDVBCAService] don't build/send the same CA PMT twice");
 461                 return;
 462         }
 463
 464         std::vector<ProgramMapSection*>::const_iterator i=ptr->getSections().begin();
 465         if ( i != ptr->getSections().end() )
 466         {
 467                 CaProgramMapSection capmt(*i++, m_prev_build_hash ? 0x05 /*update*/ : 0x03 /*only*/, 0x01 );
 468
 469                 while( i != ptr->getSections().end() )
 470                 {
 471 //                      eDebug("append");
 472                         capmt.append(*i++);
 473                 }
 474
 475                 // add our private descriptors to capmt
 476                 uint8_t tmp[10];
 477
 478                 tmp[0]=0x84;  // pmt pid
 479                 tmp[1]=0x02;
 480                 tmp[2]=pmtpid>>8;
 481                 tmp[3]=pmtpid&0xFF;
 482                 capmt.injectDescriptor(tmp, false);
 483
 484                 tmp[0] = 0x82; // demux
 485                 tmp[1] = 0x02;
 486                 tmp[2] = demux_mask;    // descramble bitmask
 487                 tmp[3] = first_demux_num; // read section data from demux number
 488                 capmt.injectDescriptor(tmp, false);
 489
 490                 tmp[0] = 0x81; // dvbnamespace
 491                 tmp[1] = 0x08;
 492                 tmp[2] = m_service.getDVBNamespace().get()>>24;
 493                 tmp[3]=(m_service.getDVBNamespace().get()>>16)&0xFF;
 494                 tmp[4]=(m_service.getDVBNamespace().get()>>8)&0xFF;
 495                 tmp[5]=m_service.getDVBNamespace().get()&0xFF;
 496                 tmp[6]=m_service.getTransportStreamID().get()>>8;
 497                 tmp[7]=m_service.getTransportStreamID().get()&0xFF;
 498                 tmp[8]=m_service.getOriginalNetworkID().get()>>8;
 499                 tmp[9]=m_service.getOriginalNetworkID().get()&0xFF;
 500                 capmt.injectDescriptor(tmp, false);
 501
 502                 capmt.writeToBuffer(m_capmt);
 503         }
 504
 505         m_prev_build_hash = build_hash;
 506
 507         if ( m_sendstate != 0xFFFFFFFF )
 508                 m_sendstate=0;
 509         sendCAPMT();
 510 }
 511
 512 void eDVBCAService::sendCAPMT()
 513 {
 514         if ( m_sendstate && m_sendstate != 0xFFFFFFFF ) // broken pipe retry
 515         {
 516                 ::close(m_sock);
 517                 Connect();
 518         }
 519
 520         int wp=0;
 521         if ( m_capmt[3] & 0x80 )
 522         {
 523                 int i=0;
 524                 int lenbytes = m_capmt[3] & ~0x80;
 525                 while(i < lenbytes)
 526                         wp = (wp << 8) | m_capmt[4 + i++];
 527                 wp+=4;
 528                 wp+=lenbytes;
 529         }
 530         else
 531         {
 532                 wp = m_capmt[3];
 533                 wp+=4;
 534         }
 535
 536         if ( write(m_sock, m_capmt, wp) == wp )
 537         {
 538                 m_sendstate=0xFFFFFFFF;
 539                 eDebug("[eDVBCAHandler] send %d bytes",wp);
 540 #if 1
 541                 for(int i=0;i<wp;i++)
 542                         eDebugNoNewLine("%02x ", m_capmt[i]);
 543                 eDebug("");
 544 #endif
 545         }
 546         else
 547         {
 548                 switch(m_sendstate)
 549                 {
 550                         case 0xFFFFFFFF:
 551                                 ++m_sendstate;
 552                                 m_retryTimer.start(0,true);
 553 //                              eDebug("[eDVBCAHandler] send failed .. immediate retry");
 554                                 break;
 555                         default:
 556                                 m_retryTimer.start(5000,true);
 557 //                              eDebug("[eDVBCAHandler] send failed .. retry in 5 sec");
 558                                 break;
 559                 }
 560                 ++m_sendstate;
 561         }
 562 }