a5bd97054319f853ffd02d8cf7c6a21a47217f63
[enigma2.git] / lib / dvb / pmt.cpp
1 #include <lib/base/eerror.h>
2 #include <lib/dvb/pmt.h>
3 #include <lib/dvb/specs.h>
4 #include <lib/dvb/dvb.h>
5 #include <lib/dvb/metaparser.h>
6 #include <dvbsi++/ca_program_map_section.h>
7
8 eDVBServicePMTHandler::eDVBServicePMTHandler()
9         :m_pmt_pid(0xFFFF), m_ca_servicePtr(0)
10 {
11         eDVBResourceManager::getInstance(m_resourceManager);
12         CONNECT(m_PMT.tableReady, eDVBServicePMTHandler::PMTready);
13         CONNECT(m_PAT.tableReady, eDVBServicePMTHandler::PATready);
14 }
15
16 eDVBServicePMTHandler::~eDVBServicePMTHandler()
17 {
18         delete m_ca_servicePtr;
19 }
20
21 void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
22 {
23         int state;
24         channel->getState(state);
25         
26         if ((m_last_channel_state != iDVBChannel::state_ok)
27                 && (state == iDVBChannel::state_ok) && (!m_demux))
28         {
29                 if (m_channel)
30                         if (m_channel->getDemux(m_demux))
31                                 eDebug("shit it failed.. again.");
32                 
33                 serviceEvent(eventTuned);
34                 
35                 if (m_demux)
36                 {
37                         eDebug("ok ... now we start!!");
38
39                         /* emit */ m_resourceManager->m_channelRunning(channel);
40
41                         m_PAT.begin(eApp, eDVBPATSpec(), m_demux);
42
43                         if ( m_service && !m_service->cacheEmpty() )
44                                 serviceEvent(eventNewProgramInfo);
45                 }
46         }
47 }
48
49 void eDVBServicePMTHandler::PMTready(int error)
50 {
51         if (error)
52                 serviceEvent(eventNoPMT);
53         else
54         {
55                 serviceEvent(eventNewProgramInfo);
56                 if (!m_pvr_channel && !m_ca_servicePtr)   // don't send campmt to camd.socket for playbacked services
57                         m_ca_servicePtr = new eDVBCAService(*this);
58                 if (m_ca_servicePtr)
59                         m_ca_servicePtr->buildCAPMT();
60         }
61 }
62
63 void eDVBServicePMTHandler::PATready(int)
64 {
65         eDebug("got PAT");
66         ePtr<eTable<ProgramAssociationSection> > ptr;
67         if (!m_PAT.getCurrent(ptr))
68         {
69                 int pmtpid = -1;
70                 std::vector<ProgramAssociationSection*>::const_iterator i;
71                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
72                 {
73                         const ProgramAssociationSection &pat = **i;
74                         ProgramAssociationConstIterator program;
75                         for (program = pat.getPrograms()->begin(); program != pat.getPrograms()->end(); ++program)
76                                 if (eServiceID((*program)->getProgramNumber()) == m_reference.getServiceID())
77                                         pmtpid = (*program)->getProgramMapPid();
78                 }
79                 if (pmtpid == -1)
80                         serviceEvent(eventNoPATEntry);
81                 else
82                 {
83                         m_PMT.begin(eApp, eDVBPMTSpec(pmtpid, m_reference.getServiceID().get()), m_demux);
84                         m_pmt_pid = pmtpid;
85                 }
86         } else
87                 serviceEvent(eventNoPAT);
88 }
89
90 int eDVBServicePMTHandler::getProgramInfo(struct program &program)
91 {
92         eDebug("got PMT");
93         ePtr<eTable<ProgramMapSection> > ptr;
94
95         program.videoStreams.clear();
96         program.audioStreams.clear();
97         program.pcrPid = -1;
98
99         if (!m_PMT.getCurrent(ptr))
100         {
101                 std::vector<ProgramMapSection*>::const_iterator i;
102                 for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
103                 {
104                         const ProgramMapSection &pmt = **i;
105                         program.pcrPid = pmt.getPcrPid();
106                         
107                         ElementaryStreamInfoConstIterator es;
108                         for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
109                         {
110                                 int isaudio = 0, isvideo = 0;
111                                 videoStream video;
112                                 audioStream audio;
113                                 
114                                 video.pid = (*es)->getPid();
115                                 audio.pid = (*es)->getPid();
116                                 
117                                 switch ((*es)->getType())
118                                 {
119                                 case 0x01: // MPEG 1 video
120                                 case 0x02: // MPEG 2 video
121                                         isvideo = 1;
122                                         break;
123                                 case 0x03: // MPEG 1 audio
124                                 case 0x04: // MPEG 2 audio:
125                                         isaudio = 1;
126                                         audio.type = audioStream::atMPEG;
127                                         break;
128                                 }
129                                 if (isaudio)
130                                         program.audioStreams.push_back(audio);
131                                 if (isvideo)
132                                         program.videoStreams.push_back(video);
133                         }
134                 }
135                 return 0;
136         }
137         else if ( m_service && !m_service->cacheEmpty() )
138         {
139                 int vpid = m_service->getCachePID(eDVBService::cVPID),
140                         apid_ac3 = m_service->getCachePID(eDVBService::cAPID),
141                         apid_mpeg = m_service->getCachePID(eDVBService::cAC3PID),
142                         pcrpid = m_service->getCachePID(eDVBService::cPCRPID),
143                         cnt=0;
144                 if ( vpid != -1 )
145                 {
146                         videoStream s;
147                         s.pid = vpid;
148                         program.videoStreams.push_back(s);
149                         ++cnt;
150                 }
151                 if ( apid_ac3 != -1 )
152                 {
153                         audioStream s;
154                         s.type = audioStream::atAC3;
155                         s.pid = apid_ac3;
156                         program.audioStreams.push_back(s);
157                         ++cnt;
158                 }
159                 if ( apid_mpeg != -1 )
160                 {
161                         audioStream s;
162                         s.type = audioStream::atMPEG;
163                         s.pid = apid_mpeg;
164                         program.audioStreams.push_back(s);
165                         ++cnt;
166                 }
167                 if ( pcrpid != -1 )
168                 {
169                         ++cnt;
170                         program.pcrPid = pcrpid;
171                 }
172                 if ( cnt )
173                         return 0;
174         }
175         return -1;
176 }
177
178 int eDVBServicePMTHandler::getDemux(ePtr<iDVBDemux> &demux)
179 {
180         demux = m_demux;
181         if (demux)
182                 return 0;
183         else
184                 return -1;
185 }
186
187 int eDVBServicePMTHandler::getPVRChannel(ePtr<iDVBPVRChannel> &pvr_channel)
188 {
189         pvr_channel = m_pvr_channel;
190         if (pvr_channel)
191                 return 0;
192         else
193                 return -1;
194 }
195
196 int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref)
197 {
198         RESULT res;
199         m_reference = ref;
200         
201 //      ref.path = "/viva.ts"; // hrhr.
202         
203                 /* is this a normal (non PVR) channel? */
204         if (ref.path.empty())
205         {
206                 eDVBChannelID chid;
207                 ref.getChannelID(chid);
208                 res = m_resourceManager->allocateChannel(chid, m_channel);
209         } else
210         {
211                 eDVBMetaParser parser;
212                 
213                 if (parser.parseFile(ref.path))
214                         eWarning("no .meta file found, trying original service ref.");
215                 else
216                         m_reference = parser.m_ref;
217                 
218                 eDebug("alloc PVR");
219                         /* allocate PVR */
220                 res = m_resourceManager->allocatePVRChannel(m_pvr_channel);
221                 if (res)
222                         eDebug("allocatePVRChannel failed!\n");
223                 m_channel = m_pvr_channel;
224         }
225         
226         if (m_channel)
227         {
228                 m_channel->connectStateChange(
229                         slot(*this, &eDVBServicePMTHandler::channelStateChanged), 
230                         m_channelStateChanged_connection);
231                 m_last_channel_state = -1;
232                 channelStateChanged(m_channel);
233         }
234
235         if (m_pvr_channel)
236                 m_pvr_channel->playFile(ref.path.c_str());
237
238         ePtr<iDVBChannelList> db;
239         if (!m_resourceManager->getChannelList(db))
240                 db->getService((eServiceReferenceDVB&)m_reference, m_service);
241
242         return res;
243 }
244
245 void eDVBCAService::Connect()
246 {
247         memset(&m_servaddr, 0, sizeof(struct sockaddr_un));
248         m_servaddr.sun_family = AF_UNIX;
249         strcpy(m_servaddr.sun_path, "/tmp/camd.socket");
250         m_clilen = sizeof(m_servaddr.sun_family) + strlen(m_servaddr.sun_path);
251         m_sock = socket(PF_UNIX, SOCK_STREAM, 0);
252         connect(m_sock, (struct sockaddr *) &m_servaddr, m_clilen);
253         fcntl(m_sock, F_SETFL, O_NONBLOCK);
254         int val=1;
255         setsockopt(m_sock, SOL_SOCKET, SO_REUSEADDR, &val, 4);
256 }
257
258 void eDVBCAService::buildCAPMT()
259 {
260         ePtr<eTable<ProgramMapSection> > ptr;
261
262         if (m_parent.m_PMT.getCurrent(ptr))
263                 return;
264
265         std::vector<ProgramMapSection*>::const_iterator i=ptr->getSections().begin();
266         if ( i != ptr->getSections().end() )
267         {
268                 CaProgramMapSection capmt(*i++, m_capmt == NULL ? 0x03 /*only*/: 0x05 /*update*/, 0x01 );
269
270                 while( i != ptr->getSections().end() )
271                 {
272                         eDebug("append");
273                         capmt.append(*i++);
274                 }
275
276                 // add our private descriptors to capmt
277                 uint8_t tmp[10];
278
279                 tmp[0]=0x84;  // pmt pid
280                 tmp[1]=0x02;
281                 tmp[2]=m_parent.m_pmt_pid>>8;
282                 tmp[3]=m_parent.m_pmt_pid&0xFF;
283                 capmt.injectDescriptor(tmp, false);
284
285                 tmp[0] = 0x82; // demux
286                 tmp[1] = 0x02;
287                 m_parent.m_demux->getCADemuxID(tmp[3]); // read section data from demux number
288                 tmp[2] = 1 << tmp[3];                   // descramble bitmask
289                 capmt.injectDescriptor(tmp, false);
290
291                 tmp[0] = 0x81; // dvbnamespace
292                 tmp[1] = 0x08;
293                 tmp[2] = m_parent.m_reference.getDVBNamespace().get()>>24;
294                 tmp[3]=(m_parent.m_reference.getDVBNamespace().get()>>16)&0xFF;
295                 tmp[4]=(m_parent.m_reference.getDVBNamespace().get()>>8)&0xFF;
296                 tmp[5]=m_parent.m_reference.getDVBNamespace().get()&0xFF;
297                 tmp[6]=m_parent.m_reference.getTransportStreamID().get()>>8;
298                 tmp[7]=m_parent.m_reference.getTransportStreamID().get()&0xFF;
299                 tmp[8]=m_parent.m_reference.getOriginalNetworkID().get()>>8;
300                 tmp[9]=m_parent.m_reference.getOriginalNetworkID().get()&0xFF;
301                 capmt.injectDescriptor(tmp, false);
302
303                 if ( !m_capmt )
304                         m_capmt = new uint8_t[2048];
305
306                 capmt.writeToBuffer(m_capmt);
307         }
308
309         if ( m_sendstate != 0xFFFFFFFF )
310                 m_sendstate=0;
311         sendCAPMT();
312 }
313
314 void eDVBCAService::sendCAPMT()
315 {
316         if ( m_sendstate && m_sendstate != 0xFFFFFFFF ) // broken pipe retry
317         {
318                 ::close(m_sock);
319                 Connect();
320         }
321
322         int wp=0;
323         if ( m_capmt[3] & 0x80 )
324         {
325                 int i=0;
326                 int lenbytes = m_capmt[3] & ~0x80;
327                 while(i < lenbytes)
328                         wp |= (m_capmt[4+i] << (8 * i++));
329                 wp+=4;
330                 wp+=lenbytes;
331         }
332         else
333         {
334                 wp = m_capmt[3];
335                 wp+=4;
336         }
337
338         if ( write(m_sock, m_capmt, wp) == wp )
339         {
340                 m_sendstate=0xFFFFFFFF;
341                 eDebug("[eDVBCAHandler] send %d bytes",wp);
342 #if 1
343                 for(int i=0;i<wp;i++)
344                         eDebugNoNewLine("%02x ", m_capmt[i]);
345                 eDebug("");
346 #endif
347         }
348         else
349         {
350                 switch(m_sendstate)
351                 {
352                         case 0xFFFFFFFF:
353                                 ++m_sendstate;
354                                 m_retryTimer.start(0,true);
355                                 eDebug("[eDVBCAHandler] send failed .. immediate retry");
356                                 break;
357                         default:
358                                 m_retryTimer.start(5000,true);
359                                 eDebug("[eDVBCAHandler] send failed .. retry in 5 sec");
360                                 break;
361                 }
362                 ++m_sendstate;
363         }
364 }