1 #define _FILE_OFFSET_BITS 64
3 #include <linux/dvb/frontend.h>
4 #include <linux/dvb/dmx.h>
5 #include <linux/dvb/audio.h>
6 #include <linux/dvb/version.h>
20 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
22 #define DTV_STREAM_ID DTV_ISDBS_TS_ID
24 #define DVB_READ_BUFFER_SIZE 188 * 128
25 #define DVB_WRITE_BUFFER_SIZE 1024 * 1024
26 #define DMX_BUFFER_SIZE 1024 * 1024
27 #define RING_BUFFER_SIZE 1024 * 1024 * 16
34 pthread_mutex_t rb_mtx;
45 void * record_read(void * priv);
46 void * record_write(void * priv);
48 // TODO: switchable between VRB(virtual ring buffer) and MRB(My Ring Buffer)
50 void mrb_init(mrb* rb, size_t size) {
51 rb->rb_ptr = malloc(size);
52 rb->rb_wt = rb->rb_rt = rb->rb_ab = 0;
53 pthread_mutex_init(&rb->rb_mtx, NULL);
56 void mrb_destroy(mrb* rb) {
58 pthread_mutex_destroy(&rb->rb_mtx);
61 size_t mrb_data_len(mrb* rb) {
65 bool mrb_put(mrb* rb, char *source, size_t size) {
68 if ( rb->rb_ab + size > RING_BUFFER_SIZE ) {
69 // RingBuffer overflow.
72 if ( rb->rb_wt + size <= RING_BUFFER_SIZE ) {
73 // written bytes + newly added bytes <= buffer max size
74 memcpy(rb->rb_ptr + rb->rb_wt, source, size);
75 rb->rb_wt = rb->rb_wt + size;
78 // written bytes + newly added bytes > buffer max size
80 partial = RING_BUFFER_SIZE - rb->rb_wt;
81 memcpy(rb->rb_ptr + rb->rb_wt, source, partial);
82 memcpy(rb->rb_ptr, source + partial, size - partial);
83 rb->rb_wt = rb->rb_wt + size - RING_BUFFER_SIZE;
85 pthread_mutex_lock(&rb->rb_mtx);
87 pthread_mutex_unlock(&rb->rb_mtx);
92 bool mrb_get(mrb* rb, char *target, size_t size) {
95 if ( rb->rb_ab < size ) {
96 // RingBuffer underflow.
99 if ( rb->rb_rt + size <= RING_BUFFER_SIZE ) {
100 // read bytes + newly taken bytes <= buffer max size
101 memcpy(target, rb->rb_ptr + rb->rb_rt, size);
102 rb->rb_rt = rb->rb_rt + size;
105 // read bytes + newly taken bytes > buffer max size
107 partial = RING_BUFFER_SIZE - rb->rb_rt;
108 memcpy(target, rb->rb_ptr + rb->rb_rt, partial);
109 memcpy(target + partial, rb->rb_ptr, size - partial);
110 rb->rb_rt = rb->rb_rt + size - RING_BUFFER_SIZE;
112 pthread_mutex_lock(&rb->rb_mtx);
114 pthread_mutex_unlock(&rb->rb_mtx);
119 static int search(int adapter_nr, unsigned int frequency, unsigned int ts_id)
123 struct dvb_frontend_info info;
124 struct dtv_property prop[3];
125 struct dtv_properties props;
129 sprintf(file, "/dev/dvb/adapter%d/frontend0", adapter_nr);
130 if ((fd = open(file, (O_RDWR | O_CREAT | O_TRUNC), 0666)) < 0) {
135 if (ioctl(fd, FE_GET_INFO, &info) < 0) {
136 perror("ioctl FE_GET_INFO");
140 if (info.type == FE_QPSK || info.type == FE_OFDM) {
142 fprintf(stderr, "Unknown type of adapter\n");
146 prop[0].cmd = DTV_FREQUENCY;
147 prop[0].u.data = frequency;
148 prop[1].cmd = DTV_STREAM_ID;
149 prop[1].u.data = ts_id;
150 prop[2].cmd = DTV_TUNE;
155 if ((ioctl(fd, FE_SET_PROPERTY, &props)) < 0) {
156 perror("ioctl FE_SET_PROPERTY");
160 for (i = 0; i < 4; i++) {
161 if (ioctl(fd, FE_READ_STATUS, &status) < 0) {
162 perror("ioctl FE_READ_STATUS");
165 if (status & FE_HAS_LOCK) {
166 fprintf(stderr, "Successfully tuned to %d(%d).\n",
173 fprintf(stderr, "Failed to tune (status %02x).\n", status);
180 static int track(int adapter_nr)
184 struct dmx_pes_filter_params filter;
187 filter.input = DMX_IN_FRONTEND;
188 filter.output = DMX_OUT_TS_TAP;
189 filter.pes_type = DMX_PES_VIDEO;
190 filter.flags = DMX_IMMEDIATE_START;
192 sprintf(file, "/dev/dvb/adapter%d/demux0", adapter_nr);
193 if ((fd = open(file, O_RDWR)) < 0) {
198 if (ioctl(fd, DMX_SET_PES_FILTER, &filter) < 0) {
199 perror("ioctl DMX_SET_PES_FILTER");
206 void record(int adapter_nr, char* output, int rectime) {
209 pthread_t read_thread, write_thread;
212 fout = open(output, (O_WRONLY | O_CREAT | O_TRUNC), 0666);
214 printf("output file open failed\n");
217 sprintf(input, "/dev/dvb/adapter%d/dvr0", adapter_nr);
218 fin = open(input, (O_RDONLY));
219 ioctl(fin, DMX_SET_BUFFER_SIZE, DMX_BUFFER_SIZE);
221 mrb_init(&arg.rb, RING_BUFFER_SIZE);
223 arg.rectime = rectime;
226 pthread_create(&read_thread , NULL, record_read , &arg);
227 pthread_create(&write_thread, NULL, record_write, &arg);
228 pthread_join(write_thread, NULL);
229 pthread_join(read_thread, NULL);
231 mrb_destroy(&arg.rb);
236 void * record_read(void * priv) {
238 time_t start_time, current_time;
239 char* buf = malloc(DVB_READ_BUFFER_SIZE);
240 my_thread_arg* p = (my_thread_arg*)priv;
242 start_time = time(NULL);
244 rt = read(p->fin, buf, DVB_READ_BUFFER_SIZE);
245 //printf("read %d %d\n", p->fin, rt);
247 printf("dvr0 file read failed\n");
253 printf("read overflow, continue\n");
256 printf("read finished. err=%d\n", errno);
261 mrb_put(&p->rb, buf, rt);
263 // TODO: time() at each DVB_READ_BUFFER_SIZE bytes read is not efficient, reduce frequency
264 current_time = time(NULL);
265 if ( current_time - start_time > p->rectime || p->done ) {
274 void * record_write(void * priv) {
275 ssize_t rt = 0, wt, shift;
276 char* buf = malloc(DVB_WRITE_BUFFER_SIZE);
277 my_thread_arg* p = (my_thread_arg*)priv;
280 if ( mrb_get(&p->rb, buf, DVB_WRITE_BUFFER_SIZE) ) {
282 rt = DVB_WRITE_BUFFER_SIZE;
284 else if ( p->done && mrb_data_len(&p->rb) > 0) {
285 // insufficient buffer, but no more buffer fill
286 rt = mrb_data_len(&p->rb);
287 mrb_get(&p->rb, buf, mrb_data_len(&p->rb));
289 else if ( p->done ) {
298 while ( ( wt = write(p->fout, buf + shift, rt) ) > 0) {
299 //printf("write %d %d\n", p->fout, rt);
301 if ( rt == 0 ) break;
305 // [buf] is not correctly written to [fout]
306 printf("output file write failed. err=%d\n", errno);
315 int main(int argc, char *argv[]) {
325 fprintf(stderr, "Usage : %s adapter_nr channel tsid rectime output\n", argv[0]);
326 fprintf(stderr, " channel can be freqency or channel(TE1/BS1/CS1)\n");
327 fprintf(stderr, "Version: 0.0.2\n");
330 adapter_nr = strtol(argv[1], NULL, 0);
331 channel_freq = strtol(argv[2], NULL, 10);
332 channel_id = strtol(argv[3], NULL, 10);
333 rectime = atoi(argv[4]);
335 if ( channel_freq == 0 ) {
336 channel_phys = atoi(argv[2] + 2);
337 if ( toupper(*argv[2]) == 'T' && toupper(*(argv[2] + 1)) == 'E' && channel_phys != 0 ) {
338 channel_freq = ( 473 + (channel_phys - 13) * 6 ) * 1000000 + 142857;
340 else if ( toupper(*argv[2]) == 'B' && toupper(*(argv[2] + 1)) == 'S' && channel_phys != 0 ) {
341 channel_freq = (channel_phys - 1) * 38360 / 2 + 1049480;
343 else if ( toupper(*argv[2]) == 'C' && toupper(*(argv[2] + 1)) == 'S' && channel_phys != 0 ) {
344 channel_freq = (channel_phys - 2) * 40000 / 2 + 1613000;
346 fprintf(stderr, "channel_freq = %d\n", channel_freq);
349 fd = search(adapter_nr, channel_freq, channel_id);
353 ret = track(adapter_nr);
354 record(adapter_nr, argv[5], rectime);