1 /*******************************************************************************
2 VLC Player Plugin by A. Lätsch 2007
4 This is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
8 ********************************************************************************/
10 #include <sys/types.h>
14 #include <sys/socket.h>
18 #include "servicets.h"
19 #include <lib/base/eerror.h>
20 #include <lib/base/object.h>
21 #include <lib/base/ebase.h>
22 #include <servicets.h>
23 #include <lib/service/service.h>
24 #include <lib/base/init_num.h>
25 #include <lib/base/init.h>
26 #include <lib/dvb/decoder.h>
28 #include <lib/dvb/pmt.h>
30 #define MAX(a,b) ((a) > (b) ? (a) : (b))
32 /********************************************************************/
33 /* eServiceFactoryTS */
34 /********************************************************************/
36 eServiceFactoryTS::eServiceFactoryTS()
38 ePtr<eServiceCenter> sc;
40 eServiceCenter::getPrivInstance(sc);
43 std::list<std::string> extensions;
44 sc->addServiceFactory(eServiceFactoryTS::id, this, extensions);
48 eServiceFactoryTS::~eServiceFactoryTS()
50 ePtr<eServiceCenter> sc;
52 eServiceCenter::getPrivInstance(sc);
54 sc->removeServiceFactory(eServiceFactoryTS::id);
57 DEFINE_REF(eServiceFactoryTS)
60 RESULT eServiceFactoryTS::play(const eServiceReference &ref, ePtr<iPlayableService> &ptr)
62 ptr = new eServiceTS(ref);
66 RESULT eServiceFactoryTS::record(const eServiceReference &ref, ePtr<iRecordableService> &ptr)
72 RESULT eServiceFactoryTS::list(const eServiceReference &, ePtr<iListableService> &ptr)
78 RESULT eServiceFactoryTS::info(const eServiceReference &ref, ePtr<iStaticServiceInformation> &ptr)
84 RESULT eServiceFactoryTS::offlineOperations(const eServiceReference &, ePtr<iServiceOfflineOperations> &ptr)
91 /********************************************************************/
93 /********************************************************************/
94 DEFINE_REF(TSAudioInfo);
96 void TSAudioInfo::addAudio(int pid, std::string lang, std::string desc, int type) {
98 as.description = desc;
102 audioStreams.push_back(as);
106 /********************************************************************/
108 /********************************************************************/
110 eServiceTS::eServiceTS(const eServiceReference &url): m_pump(eApp, 1)
112 eDebug("ServiceTS construct!");
113 m_filename = url.path.c_str();
114 m_vpid = url.getData(0) == 0 ? 0x44 : url.getData(0);
115 m_apid = url.getData(1) == 0 ? 0x45 : url.getData(1);
121 eServiceTS::~eServiceTS()
123 eDebug("ServiceTS destruct!");
127 DEFINE_REF(eServiceTS);
129 size_t crop(char *buf)
131 size_t len = strlen(buf) - 1;
132 while (len > 0 && (buf[len] == '\r' || buf[len] == '\n')) {
138 static int getline(char** pbuffer, size_t* pbufsize, int fd)
143 if (i >= *pbufsize) {
144 char *newbuf = (char*)realloc(*pbuffer, (*pbufsize)+1024);
148 *pbufsize = (*pbufsize)+1024;
150 rc = ::read(fd, (*pbuffer)+i, 1);
151 if (rc <= 0 || (*pbuffer)[i] == '\n')
153 (*pbuffer)[i] = '\0';
154 return rc <= 0 ? -1 : i;
156 if ((*pbuffer)[i] != '\r') i++;
160 int eServiceTS::openHttpConnection(std::string url)
166 int slash = url.find("/", 7);
168 host = url.substr(7, slash-7);
169 uri = url.substr(slash, url.length()-slash);
171 host = url.substr(7, url.length()-7);
174 int dp = host.find(":");
176 port = atoi(host.substr(1, host.length()-1).c_str());
179 port = atoi(host.substr(dp+1, host.length()-dp-1).c_str());
180 host = host.substr(0, dp);
183 struct hostent* h = gethostbyname(host.c_str());
184 if (h == NULL || h->h_addr_list == NULL)
186 int fd = socket(PF_INET, SOCK_STREAM, 0);
190 struct sockaddr_in addr;
191 addr.sin_family = AF_INET;
192 addr.sin_addr.s_addr = *((in_addr_t*)h->h_addr_list[0]);
193 addr.sin_port = htons(port);
195 eDebug("connecting to %s", url.c_str());
197 if (connect(fd, (sockaddr*)&addr, sizeof(addr)) == -1) {
198 std::string msg = "connect failed for: " + url;
203 std::string request = "GET ";
204 request.append(uri).append(" HTTP/1.1\n");
205 request.append("Host: ").append(host).append("\n");
206 request.append("Accept: */*\n");
207 request.append("Connection: close\n");
208 request.append("\n");
209 //eDebug(request.c_str());
210 write(fd, request.c_str(), request.length());
213 size_t buflen = 1000;
214 char* linebuf = (char*)malloc(1000);
216 rc = getline(&linebuf, &buflen, fd);
217 //eDebug("RECV(%d): %s", rc, linebuf);
228 rc = sscanf(linebuf, "%99s %d %99s", proto, &statuscode, statusmsg);
229 if (rc != 3 || statuscode != 200) {
230 eDebug("wrong response: \"200 OK\" expected.");
235 eDebug("proto=%s, code=%d, msg=%s", proto, statuscode, statusmsg);
238 rc = getline(&linebuf, &buflen, fd);
239 //eDebug("RECV(%d): %s", rc, linebuf);
246 RESULT eServiceTS::connectEvent(const Slot2<void,iPlayableService*,int> &event, ePtr<eConnection> &connection)
248 connection = new eConnection((iPlayableService*)this, m_event.connect(event));
252 RESULT eServiceTS::start()
254 ePtr<eDVBResourceManager> rmgr;
255 eDVBResourceManager::getInstance(rmgr);
256 eDVBChannel dvbChannel(rmgr, 0);
259 m_destfd = ::open("/dev/misc/pvr", O_WRONLY);
261 eDebug("Cannot open /dev/misc/pvr");
265 if (dvbChannel.getDemux(m_decodedemux, iDVBChannel::capDecode) != 0) {
266 eDebug("Cannot allocate decode-demux");
269 if (m_decodedemux->getMPEGDecoder(m_decoder, 1) != 0) {
270 eDebug("Cannot allocate MPEGDecoder");
273 m_decoder->setVideoPID(m_vpid, eDVBVideo::MPEG2);
274 m_decoder->setAudioPID(m_apid, eDVBAudio::aMPEG);
275 m_streamthread = new eStreamThread();
276 CONNECT(m_streamthread->m_event, eServiceTS::recv_event);
278 if (unpause() != 0) return -1;
280 m_event(this, evStart);
284 RESULT eServiceTS::stop()
288 if (m_state != stRunning)
290 printf("TS: %s stop\n", m_filename.c_str());
291 m_streamthread->stop();
292 m_decodedemux->flush();
298 void eServiceTS::recv_event(int evt)
300 eDebug("eServiceTS::recv_event: %d", evt);
302 case eStreamThread::evtEOS:
303 m_decodedemux->flush();
305 m_event((iPlayableService*)this, evEOF);
307 case eStreamThread::evtReadError:
308 case eStreamThread::evtWriteError:
311 m_event((iPlayableService*)this, evEOF);
313 case eStreamThread::evtSOS:
314 m_event((iPlayableService*)this, evSOF);
316 case eStreamThread::evtStreamInfo:
317 bool wasnull = !m_audioInfo;
318 m_streamthread->getAudioInfo(m_audioInfo);
320 eDebug("[servicets] %d audiostreams found", m_audioInfo->audioStreams.size());
321 if (m_audioInfo && wasnull) {
322 int sel = getCurrentTrack();
325 else if (m_audioInfo->audioStreams[sel].type != eDVBAudio::aMPEG)
332 RESULT eServiceTS::pause(ePtr<iPauseableService> &ptr)
339 RESULT eServiceTS::pause()
341 m_streamthread->stop();
346 RESULT eServiceTS::unpause()
348 int is_streaming = !strncmp(m_filename.c_str(), "http://", 7);
351 srcfd = openHttpConnection(m_filename);
353 srcfd = ::open(m_filename.c_str(), O_RDONLY);
356 eDebug("Cannot open source stream: %s", m_filename.c_str());
359 m_decodedemux->flush();
360 m_streamthread->start(srcfd, m_destfd);
366 RESULT eServiceTS::seek(ePtr<iSeekableService> &ptr)
372 RESULT eServiceTS::getLength(pts_t &pts)
377 RESULT eServiceTS::seekTo(pts_t to)
382 RESULT eServiceTS::seekRelative(int direction, pts_t to)
387 RESULT eServiceTS::getPlayPosition(pts_t &pts)
392 RESULT eServiceTS::setTrickmode(int trick)
397 RESULT eServiceTS::isCurrentlySeekable()
402 RESULT eServiceTS::info(ePtr<iServiceInformation>&i)
408 RESULT eServiceTS::getName(std::string &name)
411 size_t n = name.rfind('/');
412 if (n != std::string::npos)
413 name = name.substr(n + 1);
417 int eServiceTS::getInfo(int w)
422 std::string eServiceTS::getInfoString(int w)
427 int eServiceTS::getNumberOfTracks() {
429 return (int)m_audioInfo->audioStreams.size();
434 RESULT eServiceTS::selectTrack(unsigned int i) {
436 m_apid = m_audioInfo->audioStreams[i].pid;
437 eDebug("[servicets] audio track %d PID 0x%02x type %d\n", i, m_apid, m_audioInfo->audioStreams[i].type);
438 m_decoder->setAudioPID(m_apid, m_audioInfo->audioStreams[i].type);
439 if (m_state == stRunning)
447 RESULT eServiceTS::getTrackInfo(struct iAudioTrackInfo &info, unsigned int n) {
449 info.m_pid = m_audioInfo->audioStreams[n].pid;
450 info.m_description = m_audioInfo->audioStreams[n].description;
451 info.m_language = m_audioInfo->audioStreams[n].language;
458 int eServiceTS::getCurrentTrack() {
460 for (size_t i = 0; i < m_audioInfo->audioStreams.size(); i++) {
461 if (m_apid == m_audioInfo->audioStreams[i].pid) {
469 /********************************************************************/
471 /********************************************************************/
473 DEFINE_REF(eStreamThread)
475 eStreamThread::eStreamThread(): m_messagepump(eApp, 0) {
476 CONNECT(m_messagepump.recv_msg, eStreamThread::recvEvent);
478 eStreamThread::~eStreamThread() {
481 void eStreamThread::start(int srcfd, int destfd) {
486 run(IOPRIO_CLASS_RT);
489 void eStreamThread::stop() {
494 void eStreamThread::recvEvent(const int &evt)
499 RESULT eStreamThread::getAudioInfo(ePtr<TSAudioInfo> &ptr)
505 #define REGISTRATION_DESCRIPTOR 5
506 #define LANGUAGE_DESCRIPTOR 10
508 std::string eStreamThread::getDescriptor(unsigned char buf[], int buflen, int type)
513 if (buf[0] == type) {
515 if (desc_len > 20) desc_len = 20;
516 strncpy(str, (char*)buf+2, desc_len);
517 str[desc_len] = '\0';
518 return std::string(str);
520 buflen -= desc_len+2;
527 bool eStreamThread::scanAudioInfo(unsigned char buf[], int len)
532 int adaptfield, pmtpid, offset;
533 unsigned char pmt[1188];
536 for (int a=0; a < len - 188*4; a++) {
537 if ( buf[a] != 0x47 || buf[a + 188] != 0x47 || buf[a + 376] != 0x47 )
538 continue; // TS Header
540 if ((0x40 & buf[a + 1]) == 0) // start
543 if ((0xC0 & buf[a + 3]) != 0) // scrambling
546 adaptfield = (0x30 & buf[a + 3]) >> 4;
548 if ((adaptfield & 1) == 0) // adapt - no payload
551 offset = adaptfield == 3 ? 1 + (0xFF & buf[a + 4]) : 0; //adaptlength
553 if (buf[a + offset + 4] != 0 || buf[a + offset + 5] != 2 || (0xF0 & buf[a + offset + 6]) != 0xB0)
559 pmtpid = (0x1F & buf[a + 1])<<8 | (0xFF & buf[a + 2]);
560 memcpy(pmt + pmtsize, buf + a + 4 + offset, 184 - offset);
561 pmtsize += 184 - offset;
567 if (pmtsize == 0) return false;
569 int pmtlen = (0x0F & pmt[2]) << 8 | (0xFF & pmt[3]);
572 ePtr<TSAudioInfo> ainfo = new TSAudioInfo();
574 for (int b=8; b < pmtlen-4 && b < pmtsize-6; b++)
576 if ( (0xe0 & pmt[b+1]) != 0xe0 )
579 int pid = (0x1F & pmt[b+1])<<8 | (0xFF & pmt[b+2]);
584 case 2: // MPEG Video
585 //addVideo(pid, "MPEG2");
588 case 0x1B: // H.264 Video
589 //addVideo(pid, "H.264");
593 case 4: // MPEG Audio
594 lang = getDescriptor(pmt+b+5, pmt[b+4], LANGUAGE_DESCRIPTOR);
595 ainfo->addAudio(pid, lang, "MPEG", eDVBAudio::aMPEG);
599 case 0x81: //private data of AC3 in ATSC
603 lang = getDescriptor(pmt+b+5, pmt[b+4], LANGUAGE_DESCRIPTOR);
604 pd_type = getDescriptor(pmt+b+5, pmt[b+4], REGISTRATION_DESCRIPTOR);
605 if (pd_type == "AC-3")
606 ainfo->addAudio(pid, lang, pd_type, eDVBAudio::aAC3);
611 if (ainfo->audioStreams.size() > 0) {
619 void eStreamThread::thread() {
620 const int bufsize = 40000;
621 unsigned char buf[bufsize];
625 struct timeval timeout;
627 time_t next_scantime = 0;
628 bool sosSend = false;
632 eDebug("eStreamThread started");
634 pthread_testcancel();
641 FD_SET(m_srcfd, &rfds);
642 maxfd = MAX(maxfd, m_srcfd);
645 FD_SET(m_destfd, &wfds);
646 maxfd = MAX(maxfd, m_destfd);
648 rc = select(maxfd+1, &rfds, &wfds, NULL, &timeout);
650 eDebug("eStreamThread::thread: timeout!");
654 eDebug("eStreamThread::thread: error in select (%d)", errno);
657 if (FD_ISSET(m_srcfd, &rfds)) {
658 rc = ::read(m_srcfd, buf+r, bufsize - r);
660 eDebug("eStreamThread::thread: error in read (%d)", errno);
661 m_messagepump.send(evtReadError);
663 } else if (rc == 0) {
668 m_messagepump.send(evtSOS);
671 if (r == bufsize) eDebug("eStreamThread::thread: buffer full");
674 if (FD_ISSET(m_destfd, &wfds) && (w < r) && ((r > bufsize/4) || eof)) {
675 rc = ::write(m_destfd, buf+w, r-w);
677 eDebug("eStreamThread::thread: error in write (%d)", errno);
678 m_messagepump.send(evtWriteError);
682 //eDebug("eStreamThread::thread: buffer r=%d w=%d",r,w);
684 if (time(0) >= next_scantime) {
685 if (scanAudioInfo(buf, r)) {
686 m_messagepump.send(evtStreamInfo);
687 next_scantime = time(0) + 1;
696 m_messagepump.send(evtEOS);
700 eDebug("eStreamThread end");
703 void eStreamThread::thread_finished() {
704 if (m_srcfd >= 0) ::close(m_srcfd);
705 eDebug("eStreamThread closed");
708 eAutoInitPtr<eServiceFactoryTS> init_eServiceFactoryTS(eAutoInitNumbers::service+1, "eServiceFactoryTS");
713 Py_InitModule("servicets", NULL);