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>
19 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
21 #define DTV_STREAM_ID DTV_ISDBS_TS_ID
23 #define DVB_READ_BUFFER_SIZE 188 * 128
24 #define DVB_WRITE_BUFFER_SIZE 1024 * 1024
25 #define DMX_BUFFER_SIZE 1024 * 1024
26 #define RING_BUFFER_SIZE 1024 * 1024 * 16
33 pthread_mutex_t rb_mtx;
44 void * record_read(void * priv);
45 void * record_write(void * priv);
47 // TODO: switchable between VRB(virtual ring buffer) and MRB(My Ring Buffer)
49 void mrb_init(mrb* rb, size_t size) {
50 rb->rb_ptr = malloc(size);
51 rb->rb_wt = rb->rb_rt = rb->rb_ab = 0;
52 pthread_mutex_init(&rb->rb_mtx, NULL);
55 void mrb_destroy(mrb* rb) {
57 pthread_mutex_destroy(&rb->rb_mtx);
60 size_t mrb_data_len(mrb* rb) {
64 bool mrb_put(mrb* rb, char *source, size_t size) {
67 if ( rb->rb_ab + size > RING_BUFFER_SIZE ) {
68 // RingBuffer overflow.
71 if ( rb->rb_wt + size <= RING_BUFFER_SIZE ) {
72 // written bytes + newly added bytes <= buffer max size
73 memcpy(rb->rb_ptr + rb->rb_wt, source, size);
74 rb->rb_wt = rb->rb_wt + size;
77 // written bytes + newly added bytes > buffer max size
79 partial = RING_BUFFER_SIZE - rb->rb_wt;
80 memcpy(rb->rb_ptr + rb->rb_wt, source, partial);
81 memcpy(rb->rb_ptr, source + partial, size - partial);
82 rb->rb_wt = rb->rb_wt + size - RING_BUFFER_SIZE;
84 pthread_mutex_lock(&rb->rb_mtx);
86 pthread_mutex_unlock(&rb->rb_mtx);
91 bool mrb_get(mrb* rb, char *target, size_t size) {
94 if ( rb->rb_ab < size ) {
95 // RingBuffer underflow.
98 if ( rb->rb_rt + size <= rb->rb_wt ) {
99 // read bytes + newly taken bytes <= buffer max size
100 memcpy(target, rb->rb_ptr + rb->rb_rt, size);
101 rb->rb_rt = rb->rb_rt + size;
104 // read bytes + newly taken bytes > buffer max size
106 partial = RING_BUFFER_SIZE - rb->rb_rt;
107 memcpy(target, rb->rb_ptr + rb->rb_rt, partial);
108 memcpy(target + partial, rb->rb_ptr, size - partial);
109 rb->rb_rt = rb->rb_rt + size - RING_BUFFER_SIZE;
111 pthread_mutex_lock(&rb->rb_mtx);
113 pthread_mutex_unlock(&rb->rb_mtx);
118 static int search(int adapter_nr, unsigned int frequency, unsigned int ts_id)
122 struct dvb_frontend_info info;
123 struct dtv_property prop[3];
124 struct dtv_properties props;
128 sprintf(file, "/dev/dvb/adapter%d/frontend0", adapter_nr);
129 if ((fd = open(file, (O_RDWR | O_CREAT | O_TRUNC), 0666)) < 0) {
134 if (ioctl(fd, FE_GET_INFO, &info) < 0) {
135 perror("ioctl FE_GET_INFO");
139 if (info.type == FE_QPSK || info.type == FE_OFDM) {
141 fprintf(stderr, "Unknown type of adapter\n");
145 prop[0].cmd = DTV_FREQUENCY;
146 prop[0].u.data = frequency;
147 prop[1].cmd = DTV_STREAM_ID;
148 prop[1].u.data = ts_id;
149 prop[2].cmd = DTV_TUNE;
154 if ((ioctl(fd, FE_SET_PROPERTY, &props)) < 0) {
155 perror("ioctl FE_SET_PROPERTY");
159 for (i = 0; i < 4; i++) {
160 if (ioctl(fd, FE_READ_STATUS, &status) < 0) {
161 perror("ioctl FE_READ_STATUS");
164 if (status & FE_HAS_LOCK) {
165 fprintf(stderr, "Successfully tuned to %d(%d).\n",
172 fprintf(stderr, "Failed to tune (status %02x).\n", status);
179 static int track(int adapter_nr)
183 struct dmx_pes_filter_params filter;
186 filter.input = DMX_IN_FRONTEND;
187 filter.output = DMX_OUT_TS_TAP;
188 filter.pes_type = DMX_PES_VIDEO;
189 filter.flags = DMX_IMMEDIATE_START;
191 sprintf(file, "/dev/dvb/adapter%d/demux0", adapter_nr);
192 if ((fd = open(file, O_RDWR)) < 0) {
197 if (ioctl(fd, DMX_SET_PES_FILTER, &filter) < 0) {
198 perror("ioctl DMX_SET_PES_FILTER");
205 void record(int adapter_nr, char* output, int rectime) {
208 pthread_t read_thread, write_thread;
211 fout = open(output, (O_WRONLY | O_CREAT | O_TRUNC), 0666);
213 printf("output file open failed\n");
216 sprintf(input, "/dev/dvb/adapter%d/dvr0", adapter_nr);
217 fin = open(input, (O_RDONLY));
218 ioctl(fin, DMX_SET_BUFFER_SIZE, DMX_BUFFER_SIZE);
220 mrb_init(&arg.rb, RING_BUFFER_SIZE);
222 arg.rectime = rectime;
225 pthread_create(&read_thread , NULL, record_read , &arg);
226 pthread_create(&write_thread, NULL, record_write, &arg);
227 pthread_join(write_thread, NULL);
228 pthread_join(read_thread, NULL);
230 mrb_destroy(&arg.rb);
235 void * record_read(void * priv) {
237 time_t start_time, current_time;
238 char* buf = malloc(DVB_READ_BUFFER_SIZE);
239 my_thread_arg* p = (my_thread_arg*)priv;
241 start_time = time(NULL);
243 rt = read(p->fin, buf, DVB_READ_BUFFER_SIZE);
244 //printf("read %d %d\n", p->fin, rt);
246 printf("dvr0 file read failed\n");
252 printf("read overflow, continue\n");
255 printf("read finished. err=%d\n", errno);
260 mrb_put(&p->rb, buf, rt);
262 // TODO: time() at each DVB_READ_BUFFER_SIZE bytes read is not efficient, reduce frequency
263 current_time = time(NULL);
264 if ( current_time - start_time > p->rectime || p->done ) {
273 void * record_write(void * priv) {
274 ssize_t rt = 0, wt, shift;
275 char* buf = malloc(DVB_WRITE_BUFFER_SIZE);
276 my_thread_arg* p = (my_thread_arg*)priv;
279 if ( mrb_get(&p->rb, buf, DVB_WRITE_BUFFER_SIZE) ) {
281 rt = DVB_WRITE_BUFFER_SIZE;
283 else if ( p->done && mrb_data_len(&p->rb) > 0) {
284 // insufficient buffer, but no more buffer fill
285 rt = mrb_data_len(&p->rb);
286 mrb_get(&p->rb, buf, mrb_data_len(&p->rb));
288 else if ( p->done ) {
297 while ( ( wt = write(p->fout, buf + shift, rt) ) > 0) {
298 //printf("write %d %d\n", p->fout, rt);
300 if ( rt == 0 ) break;
304 // [buf] is not correctly written to [fout]
305 printf("output file write failed. err=%d\n", errno);
314 int main(int argc, char *argv[]) {
324 fprintf(stderr, "Usage : %s adapter_nr channel tsid rectime output\n", argv[0]);
325 fprintf(stderr, " channel can be freqency or channel(TE1/BS1/CS1)\n");
326 fprintf(stderr, "Version: 0.0.2\n");
329 adapter_nr = strtol(argv[1], NULL, 0);
330 channel_freq = strtol(argv[2], NULL, 10);
331 channel_id = strtol(argv[3], NULL, 10);
332 rectime = atoi(argv[4]);
334 if ( channel_freq == 0 ) {
335 channel_phys = atoi(argv[2] + 2);
336 if ( toupper(*argv[2]) == 'T' && toupper(*(argv[2] + 1)) == 'E' && channel_phys != 0 ) {
337 channel_freq = ( 473 + (channel_phys - 13) * 6 ) * 1000000 + 142857;
339 else if ( toupper(*argv[2]) == 'B' && toupper(*(argv[2] + 1)) == 'S' && channel_phys != 0 ) {
340 channel_freq = (channel_phys - 1) * 38360 / 2 + 1049480;
342 else if ( toupper(*argv[2]) == 'C' && toupper(*(argv[2] + 1)) == 'S' && channel_phys != 0 ) {
343 channel_freq = (channel_phys - 2) * 40000 / 2 + 1613000;
345 fprintf(stderr, "channel_freq = %d\n", channel_freq);
348 fd = search(adapter_nr, channel_freq, channel_id);
352 ret = track(adapter_nr);
353 record(adapter_nr, argv[5], rectime);