11 /* change every 1:FUZZING_PROPABILITY byte */
12 #define FUZZING_PROPABILITY 100
15 #include <linux/dvb/dmx.h>
18 #define DMX_ADD_PID _IOW('o', 51, __u16)
19 #define DMX_REMOVE_PID _IOW('o', 52, __u16)
24 #include <lib/base/eerror.h>
25 #include <lib/dvb/idvb.h>
26 #include <lib/dvb/demux.h>
27 #include <lib/dvb/esection.h>
28 #include <lib/dvb/decoder.h>
30 eDVBDemux::eDVBDemux(int adapter, int demux): adapter(adapter), demux(demux)
35 eDVBDemux::~eDVBDemux()
39 int eDVBDemux::openDemux(void)
42 snprintf(filename, 128, "/dev/dvb/adapter%d/demux%d", adapter, demux);
43 return ::open(filename, O_RDWR);
46 int eDVBDemux::openDVR(int flags)
49 snprintf(filename, 128, "/dev/dvb/adapter%d/dvr%d", adapter, demux);
50 return ::open(filename, flags);
55 RESULT eDVBDemux::setSourceFrontend(int fenum)
58 if (fd < 0) return -1;
59 int n = DMX_SOURCE_FRONT0 + fenum;
60 int res = ::ioctl(fd, DMX_SET_SOURCE, &n);
62 eDebug("DMX_SET_SOURCE failed! - %m");
69 RESULT eDVBDemux::setSourcePVR(int pvrnum)
72 if (fd < 0) return -1;
73 int n = DMX_SOURCE_DVR0 + pvrnum;
74 int res = ::ioctl(fd, DMX_SET_SOURCE, &n);
80 RESULT eDVBDemux::createSectionReader(eMainloop *context, ePtr<iDVBSectionReader> &reader)
83 reader = new eDVBSectionReader(this, context, res);
89 RESULT eDVBDemux::createPESReader(eMainloop *context, ePtr<iDVBPESReader> &reader)
92 reader = new eDVBPESReader(this, context, res);
98 RESULT eDVBDemux::createTSRecorder(ePtr<iDVBTSRecorder> &recorder)
102 recorder = new eDVBTSRecorder(this);
106 RESULT eDVBDemux::getMPEGDecoder(ePtr<iTSMPEGDecoder> &decoder, int index)
108 decoder = new eTSMPEGDecoder(this, index);
112 RESULT eDVBDemux::getSTC(pts_t &pts, int num)
114 int fd = openDemux();
123 if (ioctl(fd, DMX_GET_STC, &stc) < 0)
125 eDebug("DMX_GET_STC failed!");
132 eDebug("DMX_GET_STC - %lld", pts);
138 RESULT eDVBDemux::flush()
140 // FIXME: implement flushing the PVR queue here.
146 RESULT eDVBDemux::connectEvent(const Slot1<void,int> &event, ePtr<eConnection> &conn)
148 conn = new eConnection(this, m_event.connect(event));
152 void eDVBSectionReader::data(int)
154 __u8 data[4096]; // max. section size
156 r = ::read(fd, data, 4096);
159 for (j = 0; j < r; ++j)
161 if (!(rand()%FUZZING_PROPABILITY))
167 eWarning("ERROR reading section - %m\n");
172 // this check should never happen unless the driver is crappy!
174 if ((c = crc32((unsigned)-1, data, r)))
176 eDebug("crc32 failed! is %x\n", c);
183 eDebug("data.. but not active");
186 eDVBSectionReader::eDVBSectionReader(eDVBDemux *demux, eMainloop *context, RESULT &res): demux(demux)
189 fd = demux->openDemux();
193 notifier=eSocketNotifier::create(context, fd, eSocketNotifier::Read, false);
194 CONNECT(notifier->activated, eDVBSectionReader::data);
203 DEFINE_REF(eDVBSectionReader)
205 eDVBSectionReader::~eDVBSectionReader()
211 RESULT eDVBSectionReader::setBufferSize(int size)
213 int res=::ioctl(fd, DMX_SET_BUFFER_SIZE, size);
215 eDebug("eDVBSectionReader DMX_SET_BUFFER_SIZE failed(%m)");
219 RESULT eDVBSectionReader::start(const eDVBSectionFilterMask &mask)
226 dmx_sct_filter_params sct;
229 sct.flags = DMX_IMMEDIATE_START;
231 if (mask.flags & eDVBSectionFilterMask::rfCRC)
233 sct.flags |= DMX_CHECK_CRC;
239 memcpy(sct.filter.filter, mask.data, DMX_FILTER_SIZE);
240 memcpy(sct.filter.mask, mask.mask, DMX_FILTER_SIZE);
241 memcpy(sct.filter.mode, mask.mode, DMX_FILTER_SIZE);
242 setBufferSize(8192*8);
244 res = ::ioctl(fd, DMX_SET_FILTER, &sct);
252 RESULT eDVBSectionReader::stop()
258 ::ioctl(fd, DMX_STOP);
264 RESULT eDVBSectionReader::connectRead(const Slot1<void,const __u8*> &r, ePtr<eConnection> &conn)
266 conn = new eConnection(this, read.connect(r));
270 void eDVBPESReader::data(int)
276 r = ::read(m_fd, buffer, 16384);
281 if (errno == EAGAIN || errno == EINTR) /* ok */
283 eWarning("ERROR reading PES (fd=%d) - %m", m_fd);
290 eWarning("PES reader not active");
296 eDVBPESReader::eDVBPESReader(eDVBDemux *demux, eMainloop *context, RESULT &res): m_demux(demux)
299 m_fd = m_demux->openDemux();
303 setBufferSize(64*1024);
304 ::fcntl(m_fd, F_SETFL, O_NONBLOCK);
305 m_notifier = eSocketNotifier::create(context, m_fd, eSocketNotifier::Read, false);
306 CONNECT(m_notifier->activated, eDVBPESReader::data);
315 RESULT eDVBPESReader::setBufferSize(int size)
317 int res = ::ioctl(m_fd, DMX_SET_BUFFER_SIZE, size);
319 eDebug("eDVBPESReader DMX_SET_BUFFER_SIZE failed(%m)");
323 DEFINE_REF(eDVBPESReader)
325 eDVBPESReader::~eDVBPESReader()
331 RESULT eDVBPESReader::start(int pid)
339 dmx_pes_filter_params flt;
341 flt.pes_type = DMX_PES_OTHER;
343 flt.input = DMX_IN_FRONTEND;
344 flt.output = DMX_OUT_TAP;
346 flt.flags = DMX_IMMEDIATE_START;
348 res = ::ioctl(m_fd, DMX_SET_PES_FILTER, &flt);
351 eWarning("PES filter: DMX_SET_PES_FILTER - %m");
357 RESULT eDVBPESReader::stop()
363 ::ioctl(m_fd, DMX_STOP);
369 RESULT eDVBPESReader::connectRead(const Slot2<void,const __u8*,int> &r, ePtr<eConnection> &conn)
371 conn = new eConnection(this, m_read.connect(r));
375 eDVBRecordFileThread::eDVBRecordFileThread()
376 :eFilePushThread(IOPRIO_CLASS_RT, 7), m_ts_parser(m_stream_info)
378 m_current_offset = 0;
381 void eDVBRecordFileThread::setTimingPID(int pid, int type)
383 m_ts_parser.setPid(pid, type);
386 void eDVBRecordFileThread::startSaveMetaInformation(const std::string &filename)
388 m_stream_info.startSave(filename.c_str());
391 void eDVBRecordFileThread::stopSaveMetaInformation()
393 m_stream_info.stopSave();
396 void eDVBRecordFileThread::enableAccessPoints(bool enable)
398 m_ts_parser.enableAccessPoints(enable);
401 int eDVBRecordFileThread::getLastPTS(pts_t &pts)
403 return m_ts_parser.getLastPTS(pts);
406 int eDVBRecordFileThread::filterRecordData(const unsigned char *data, int len, size_t ¤t_span_remaining)
408 m_ts_parser.parseData(m_current_offset, data, len);
410 m_current_offset += len;
415 DEFINE_REF(eDVBTSRecorder);
417 eDVBTSRecorder::eDVBTSRecorder(eDVBDemux *demux): m_demux(demux)
421 m_thread = new eDVBRecordFileThread();
422 CONNECT(m_thread->m_event, eDVBTSRecorder::filepushEvent);
425 eDVBTSRecorder::~eDVBTSRecorder()
431 RESULT eDVBTSRecorder::start()
433 std::map<int,int>::iterator i(m_pids.begin());
438 if (m_target_fd == -1)
441 if (i == m_pids.end())
445 snprintf(filename, 128, "/dev/dvb/adapter%d/demux%d", m_demux->adapter, m_demux->demux);
447 m_source_fd = ::open(filename, O_RDONLY);
451 eDebug("FAILED to open demux (%s) in ts recoder (%m)", filename);
455 setBufferSize(1024*1024);
457 dmx_pes_filter_params flt;
458 flt.pes_type = DMX_PES_OTHER;
459 flt.output = DMX_OUT_TSDEMUX_TAP;
462 flt.input = DMX_IN_FRONTEND;
464 int res = ::ioctl(m_source_fd, DMX_SET_PES_FILTER, &flt);
467 eDebug("DMX_SET_PES_FILTER: %m");
468 ::close(m_source_fd);
473 ::ioctl(m_source_fd, DMX_START);
475 if (m_target_filename != "")
476 m_thread->startSaveMetaInformation(m_target_filename);
478 m_thread->start(m_source_fd, m_target_fd);
481 while (i != m_pids.end()) {
489 RESULT eDVBTSRecorder::setBufferSize(int size)
491 int res = ::ioctl(m_source_fd, DMX_SET_BUFFER_SIZE, size);
493 eDebug("eDVBTSRecorder DMX_SET_BUFFER_SIZE failed(%m)");
497 RESULT eDVBTSRecorder::addPID(int pid)
499 if (m_pids.find(pid) != m_pids.end())
502 m_pids.insert(std::pair<int,int>(pid, -1));
508 RESULT eDVBTSRecorder::removePID(int pid)
510 if (m_pids.find(pid) == m_pids.end())
520 RESULT eDVBTSRecorder::setTimingPID(int pid, int type)
522 m_thread->setTimingPID(pid, type);
526 RESULT eDVBTSRecorder::setTargetFD(int fd)
532 RESULT eDVBTSRecorder::setTargetFilename(const char *filename)
534 m_target_filename = filename;
536 std::string target_path = m_target_filename;
537 std::string::size_type filePos = target_path.rfind('/');
538 m_thread->setTSPath(target_path.erase(filePos));
543 RESULT eDVBTSRecorder::enableAccessPoints(bool enable)
545 m_thread->enableAccessPoints(enable);
549 RESULT eDVBTSRecorder::setBoundary(off_t max)
551 return -1; // not yet implemented
554 RESULT eDVBTSRecorder::setTimeshift(bool enable)
556 m_thread->setTimeshift(enable);
559 RESULT eDVBTSRecorder::stop()
563 for (std::map<int,int>::iterator i(m_pids.begin()); i != m_pids.end(); ++i)
569 /* workaround for record thread stop */
570 if (m_source_fd >= 0)
572 if (::ioctl(m_source_fd, DMX_STOP) < 0)
577 if (::close(m_source_fd) < 0)
588 if (m_source_fd >= 0)
590 ::close(m_source_fd);
596 m_thread->stopSaveMetaInformation();
600 RESULT eDVBTSRecorder::getCurrentPCR(pts_t &pcr)
606 /* XXX: we need a lock here */
608 /* we don't filter PCR data, so just use the last received PTS, which is not accurate, but better than nothing */
609 return m_thread->getLastPTS(pcr);
612 RESULT eDVBTSRecorder::connectEvent(const Slot1<void,int> &event, ePtr<eConnection> &conn)
614 conn = new eConnection(this, m_event.connect(event));
618 RESULT eDVBTSRecorder::startPID(int pid)
622 if (::ioctl(m_source_fd, DMX_ADD_PID, &p) < 0) {
623 perror("DMX_ADD_PID");
624 if (errno == EAGAIN || errno == EINTR) {
635 void eDVBTSRecorder::stopPID(int pid)
637 if (m_pids[pid] != -1)
641 if (::ioctl(m_source_fd, DMX_REMOVE_PID, &p) < 0) {
642 perror("DMX_REMOVE_PID");
643 if (errno == EAGAIN || errno == EINTR) {
654 void eDVBTSRecorder::filepushEvent(int event)
658 case eFilePushThread::evtWriteError:
659 m_event(eventWriteError);