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);
262 eDebug("Cannot open /dev/misc/pvr");
266 if (dvbChannel.getDemux(m_decodedemux, iDVBChannel::capDecode) != 0) {
267 eDebug("Cannot allocate decode-demux");
270 if (m_decodedemux->getMPEGDecoder(m_decoder, 1) != 0) {
271 eDebug("Cannot allocate MPEGDecoder");
274 m_decoder->setVideoPID(m_vpid, eDVBVideo::MPEG2);
275 m_decoder->setAudioPID(m_apid, eDVBAudio::aMPEG);
276 m_streamthread = new eStreamThread();
277 CONNECT(m_streamthread->m_event, eServiceTS::recv_event);
279 if (unpause() != 0) return -1;
281 m_event(this, evStart);
285 RESULT eServiceTS::stop()
292 if (m_state != stRunning)
294 printf("TS: %s stop\n", m_filename.c_str());
295 m_streamthread->stop();
296 m_decodedemux->flush();
302 void eServiceTS::recv_event(int evt)
304 eDebug("eServiceTS::recv_event: %d", evt);
306 case eStreamThread::evtEOS:
307 m_decodedemux->flush();
309 m_event((iPlayableService*)this, evEOF);
311 case eStreamThread::evtReadError:
312 case eStreamThread::evtWriteError:
315 m_event((iPlayableService*)this, evEOF);
317 case eStreamThread::evtSOS:
318 m_event((iPlayableService*)this, evSOF);
320 case eStreamThread::evtStreamInfo:
321 bool wasnull = !m_audioInfo;
322 m_streamthread->getAudioInfo(m_audioInfo);
324 eDebug("[servicets] %d audiostreams found", m_audioInfo->audioStreams.size());
325 if (m_audioInfo && wasnull) {
326 int sel = getCurrentTrack();
329 else if (m_audioInfo->audioStreams[sel].type != eDVBAudio::aMPEG)
336 RESULT eServiceTS::pause(ePtr<iPauseableService> &ptr)
343 RESULT eServiceTS::pause()
345 m_streamthread->stop();
350 RESULT eServiceTS::unpause()
352 int is_streaming = !strncmp(m_filename.c_str(), "http://", 7);
355 srcfd = openHttpConnection(m_filename);
357 srcfd = ::open(m_filename.c_str(), O_RDONLY);
360 eDebug("Cannot open source stream: %s", m_filename.c_str());
363 m_decodedemux->flush();
364 m_streamthread->start(srcfd, m_destfd);
370 RESULT eServiceTS::seek(ePtr<iSeekableService> &ptr)
376 RESULT eServiceTS::getLength(pts_t &pts)
381 RESULT eServiceTS::seekTo(pts_t to)
386 RESULT eServiceTS::seekRelative(int direction, pts_t to)
391 RESULT eServiceTS::getPlayPosition(pts_t &pts)
396 RESULT eServiceTS::setTrickmode(int trick)
401 RESULT eServiceTS::isCurrentlySeekable()
406 RESULT eServiceTS::info(ePtr<iServiceInformation>&i)
412 RESULT eServiceTS::getName(std::string &name)
415 size_t n = name.rfind('/');
416 if (n != std::string::npos)
417 name = name.substr(n + 1);
421 int eServiceTS::getInfo(int w)
426 std::string eServiceTS::getInfoString(int w)
431 int eServiceTS::getNumberOfTracks() {
433 return (int)m_audioInfo->audioStreams.size();
438 RESULT eServiceTS::selectTrack(unsigned int i) {
440 m_apid = m_audioInfo->audioStreams[i].pid;
441 eDebug("[servicets] audio track %d PID 0x%02x type %d\n", i, m_apid, m_audioInfo->audioStreams[i].type);
442 m_decoder->setAudioPID(m_apid, m_audioInfo->audioStreams[i].type);
443 if (m_state == stRunning)
451 RESULT eServiceTS::getTrackInfo(struct iAudioTrackInfo &info, unsigned int n) {
453 info.m_pid = m_audioInfo->audioStreams[n].pid;
454 info.m_description = m_audioInfo->audioStreams[n].description;
455 info.m_language = m_audioInfo->audioStreams[n].language;
462 int eServiceTS::getCurrentTrack() {
464 for (size_t i = 0; i < m_audioInfo->audioStreams.size(); i++) {
465 if (m_apid == m_audioInfo->audioStreams[i].pid) {
473 /********************************************************************/
475 /********************************************************************/
477 DEFINE_REF(eStreamThread)
479 eStreamThread::eStreamThread(): m_messagepump(eApp, 0) {
480 CONNECT(m_messagepump.recv_msg, eStreamThread::recvEvent);
482 eStreamThread::~eStreamThread() {
485 void eStreamThread::start(int srcfd, int destfd) {
490 run(IOPRIO_CLASS_RT);
493 void eStreamThread::stop() {
498 void eStreamThread::recvEvent(const int &evt)
503 RESULT eStreamThread::getAudioInfo(ePtr<TSAudioInfo> &ptr)
509 #define REGISTRATION_DESCRIPTOR 5
510 #define LANGUAGE_DESCRIPTOR 10
512 std::string eStreamThread::getDescriptor(unsigned char buf[], int buflen, int type)
517 if (buf[0] == type) {
519 if (desc_len > 20) desc_len = 20;
520 strncpy(str, (char*)buf+2, desc_len);
521 str[desc_len] = '\0';
522 return std::string(str);
524 buflen -= desc_len+2;
531 bool eStreamThread::scanAudioInfo(unsigned char buf[], int len)
536 int adaptfield, pmtpid, offset;
537 unsigned char pmt[1188];
540 for (int a=0; a < len - 188*4; a++) {
541 if ( buf[a] != 0x47 || buf[a + 188] != 0x47 || buf[a + 376] != 0x47 )
542 continue; // TS Header
544 if ((0x40 & buf[a + 1]) == 0) // start
547 if ((0xC0 & buf[a + 3]) != 0) // scrambling
550 adaptfield = (0x30 & buf[a + 3]) >> 4;
552 if ((adaptfield & 1) == 0) // adapt - no payload
555 offset = adaptfield == 3 ? 1 + (0xFF & buf[a + 4]) : 0; //adaptlength
557 if (buf[a + offset + 4] != 0 || buf[a + offset + 5] != 2 || (0xF0 & buf[a + offset + 6]) != 0xB0)
563 pmtpid = (0x1F & buf[a + 1])<<8 | (0xFF & buf[a + 2]);
564 memcpy(pmt + pmtsize, buf + a + 4 + offset, 184 - offset);
565 pmtsize += 184 - offset;
571 if (pmtsize == 0) return false;
573 int pmtlen = (0x0F & pmt[2]) << 8 | (0xFF & pmt[3]);
576 ePtr<TSAudioInfo> ainfo = new TSAudioInfo();
578 for (int b=8; b < pmtlen-4 && b < pmtsize-6; b++)
580 if ( (0xe0 & pmt[b+1]) != 0xe0 )
583 int pid = (0x1F & pmt[b+1])<<8 | (0xFF & pmt[b+2]);
588 case 2: // MPEG Video
589 //addVideo(pid, "MPEG2");
592 case 0x1B: // H.264 Video
593 //addVideo(pid, "H.264");
597 case 4: // MPEG Audio
598 lang = getDescriptor(pmt+b+5, pmt[b+4], LANGUAGE_DESCRIPTOR);
599 ainfo->addAudio(pid, lang, "MPEG", eDVBAudio::aMPEG);
603 case 0x81: //private data of AC3 in ATSC
607 lang = getDescriptor(pmt+b+5, pmt[b+4], LANGUAGE_DESCRIPTOR);
608 pd_type = getDescriptor(pmt+b+5, pmt[b+4], REGISTRATION_DESCRIPTOR);
609 if (pd_type == "AC-3")
610 ainfo->addAudio(pid, lang, pd_type, eDVBAudio::aAC3);
615 if (ainfo->audioStreams.size() > 0) {
623 void eStreamThread::thread() {
624 const int bufsize = 40000;
625 unsigned char buf[bufsize];
629 struct timeval timeout;
631 time_t next_scantime = 0;
632 bool sosSend = false;
636 eDebug("eStreamThread started");
638 pthread_testcancel();
645 FD_SET(m_srcfd, &rfds);
646 maxfd = MAX(maxfd, m_srcfd);
649 FD_SET(m_destfd, &wfds);
650 maxfd = MAX(maxfd, m_destfd);
652 rc = select(maxfd+1, &rfds, &wfds, NULL, &timeout);
654 eDebug("eStreamThread::thread: timeout!");
658 eDebug("eStreamThread::thread: error in select (%d)", errno);
661 if (FD_ISSET(m_srcfd, &rfds)) {
662 rc = ::read(m_srcfd, buf+r, bufsize - r);
664 eDebug("eStreamThread::thread: error in read (%d)", errno);
665 m_messagepump.send(evtReadError);
667 } else if (rc == 0) {
672 m_messagepump.send(evtSOS);
675 if (r == bufsize) eDebug("eStreamThread::thread: buffer full");
678 if (FD_ISSET(m_destfd, &wfds) && (w < r) && ((r > bufsize/4) || eof)) {
679 rc = ::write(m_destfd, buf+w, r-w);
681 eDebug("eStreamThread::thread: error in write (%d)", errno);
682 m_messagepump.send(evtWriteError);
686 //eDebug("eStreamThread::thread: buffer r=%d w=%d",r,w);
688 if (time(0) >= next_scantime) {
689 if (scanAudioInfo(buf, r)) {
690 m_messagepump.send(evtStreamInfo);
691 next_scantime = time(0) + 1;
700 m_messagepump.send(evtEOS);
704 eDebug("eStreamThread end");
707 void eStreamThread::thread_finished() {
708 if (m_srcfd >= 0) ::close(m_srcfd);
709 eDebug("eStreamThread closed");
712 eAutoInitPtr<eServiceFactoryTS> init_eServiceFactoryTS(eAutoInitNumbers::service+1, "eServiceFactoryTS");
717 Py_InitModule("servicets", NULL);