1 #include <linux/dvb/frontend.h>
2 #include <linux/dvb/dmx.h>
3 #include <linux/dvb/audio.h>
4 #include <linux/dvb/version.h>
17 #define ARRAY_SIZE(arr) (sizeof(arr) / sizeof((arr)[0]))
19 #define DTV_STREAM_ID DTV_ISDBS_TS_ID
21 #define DVB_READ_BUFFER_SIZE 188 * 128
22 #define DVB_WRITE_BUFFER_SIZE 1024 * 1024
23 #define DMX_BUFFER_SIZE 1024 * 1024
24 #define RING_BUFFER_SIZE 1024 * 1024 * 16
31 pthread_mutex_t rb_mtx;
42 void * record_read(void * priv);
43 void * record_write(void * priv);
45 // TODO: switchable between VRB(virtual ring buffer) and MRB(My Ring Buffer)
47 void mrb_init(mrb* rb, size_t size) {
48 rb->rb_ptr = malloc(size);
49 rb->rb_wt = rb->rb_rt = rb->rb_ab = 0;
50 pthread_mutex_init(&rb->rb_mtx, NULL);
53 void mrb_destroy(mrb* rb) {
55 pthread_mutex_destroy(&rb->rb_mtx);
58 size_t mrb_data_len(mrb* rb) {
62 bool mrb_put(mrb* rb, char *source, size_t size) {
65 if ( rb->rb_ab + size > RING_BUFFER_SIZE ) {
66 // RingBuffer overflow.
69 if ( rb->rb_wt + size <= RING_BUFFER_SIZE ) {
70 // written bytes + newly added bytes <= buffer max size
71 memcpy(rb->rb_ptr + rb->rb_wt, source, size);
72 rb->rb_wt = rb->rb_wt + size;
75 // written bytes + newly added bytes > buffer max size
77 partial = RING_BUFFER_SIZE - rb->rb_wt;
78 memcpy(rb->rb_ptr + rb->rb_wt, source, partial);
79 memcpy(rb->rb_ptr, source + partial, size - partial);
80 rb->rb_wt = rb->rb_wt + size - RING_BUFFER_SIZE;
82 pthread_mutex_lock(&rb->rb_mtx);
84 pthread_mutex_unlock(&rb->rb_mtx);
89 bool mrb_get(mrb* rb, char *target, size_t size) {
92 if ( rb->rb_ab < size ) {
93 // RingBuffer underflow.
96 if ( rb->rb_rt + size <= rb->rb_wt ) {
97 // read bytes + newly taken bytes <= buffer max size
98 memcpy(target, rb->rb_ptr + rb->rb_rt, size);
99 rb->rb_rt = rb->rb_rt + size;
102 // read bytes + newly taken bytes > buffer max size
104 partial = RING_BUFFER_SIZE - rb->rb_rt;
105 memcpy(target, rb->rb_ptr + rb->rb_rt, partial);
106 memcpy(target + partial, rb->rb_ptr, size - partial);
107 rb->rb_rt = rb->rb_rt + size - RING_BUFFER_SIZE;
109 pthread_mutex_lock(&rb->rb_mtx);
111 pthread_mutex_unlock(&rb->rb_mtx);
116 static int search(int adapter_nr, unsigned int frequency, unsigned int ts_id)
120 struct dvb_frontend_info info;
121 struct dtv_property prop[3];
122 struct dtv_properties props;
126 sprintf(file, "/dev/dvb/adapter%d/frontend0", adapter_nr);
127 if ((fd = open(file, (O_RDWR | O_CREAT | O_TRUNC), 0666)) < 0) {
132 if (ioctl(fd, FE_GET_INFO, &info) < 0) {
133 perror("ioctl FE_GET_INFO");
137 if (info.type == FE_QPSK || info.type == FE_OFDM) {
139 fprintf(stderr, "Unknown type of adapter\n");
143 prop[0].cmd = DTV_FREQUENCY;
144 prop[0].u.data = frequency;
145 prop[1].cmd = DTV_STREAM_ID;
146 prop[1].u.data = ts_id;
147 prop[2].cmd = DTV_TUNE;
152 if ((ioctl(fd, FE_SET_PROPERTY, &props)) < 0) {
153 perror("ioctl FE_SET_PROPERTY");
157 for (i = 0; i < 4; i++) {
158 if (ioctl(fd, FE_READ_STATUS, &status) < 0) {
159 perror("ioctl FE_READ_STATUS");
162 if (status & FE_HAS_LOCK) {
163 fprintf(stderr, "Successfully tuned to %d(%d).\n",
170 fprintf(stderr, "Failed to tune (status %02x).\n", status);
177 static int track(int adapter_nr)
181 struct dmx_pes_filter_params filter;
184 filter.input = DMX_IN_FRONTEND;
185 filter.output = DMX_OUT_TS_TAP;
186 filter.pes_type = DMX_PES_VIDEO;
187 filter.flags = DMX_IMMEDIATE_START;
189 sprintf(file, "/dev/dvb/adapter%d/demux0", adapter_nr);
190 if ((fd = open(file, O_RDWR)) < 0) {
195 if (ioctl(fd, DMX_SET_PES_FILTER, &filter) < 0) {
196 perror("ioctl DMX_SET_PES_FILTER");
203 void record(int adapter_nr, char* output, int rectime) {
206 pthread_t read_thread, write_thread;
209 fout = open(output, (O_WRONLY | O_CREAT | O_TRUNC), 0666);
211 printf("output file open failed\n");
214 sprintf(input, "/dev/dvb/adapter%d/dvr0", adapter_nr);
215 fin = open(input, (O_RDONLY));
216 ioctl(fin, DMX_SET_BUFFER_SIZE, DMX_BUFFER_SIZE);
218 mrb_init(&arg.rb, RING_BUFFER_SIZE);
220 arg.rectime = rectime;
223 pthread_create(&read_thread , NULL, record_read , &arg);
224 pthread_create(&write_thread, NULL, record_write, &arg);
225 pthread_join(write_thread, NULL);
226 pthread_join(read_thread, NULL);
228 mrb_destroy(&arg.rb);
233 void * record_read(void * priv) {
235 time_t start_time, current_time;
236 char buf[DVB_READ_BUFFER_SIZE];
237 my_thread_arg* p = (my_thread_arg*)priv;
239 start_time = time(NULL);
241 rt = read(p->fin, buf, DVB_READ_BUFFER_SIZE);
242 //printf("read %d %d\n", p->fin, rt);
244 printf("dvr0 file read failed\n");
250 printf("read overflow, continue\n");
253 printf("read finished. err=%d\n", errno);
258 mrb_put(&p->rb, buf, rt);
260 // TODO: time() at each DVB_READ_BUFFER_SIZE bytes read is not efficient, reduce frequency
261 current_time = time(NULL);
262 if ( current_time - start_time > p->rectime || p->done ) {
271 void * record_write(void * priv) {
272 ssize_t rt = 0, wt, shift;
273 char buf[DVB_WRITE_BUFFER_SIZE];
274 my_thread_arg* p = (my_thread_arg*)priv;
277 if ( mrb_get(&p->rb, buf, DVB_WRITE_BUFFER_SIZE) ) {
279 rt = DVB_WRITE_BUFFER_SIZE;
281 else if ( p->done && mrb_data_len(&p->rb) > 0) {
282 // insufficient buffer, but no more buffer fill
283 rt = mrb_data_len(&p->rb);
284 mrb_get(&p->rb, buf, mrb_data_len(&p->rb));
286 else if ( p->done ) {
295 while ( ( wt = write(p->fout, buf + shift, rt) ) > 0) {
296 //printf("write %d %d\n", p->fout, rt);
298 if ( rt == 0 ) break;
302 // [buf] is not correctly written to [fout]
303 printf("output file write failed\n");
312 int main(int argc, char *argv[]) {
321 fprintf(stderr, "Usage : %s adapter_nr freq tsid rectime output\n", argv[0]);
322 fprintf(stderr, "Version: 0.0.1\n");
325 adapter_nr = strtol(argv[1], NULL, 0);
326 channel_freq = strtol(argv[2], NULL, 10);
327 channel_id = strtol(argv[3], NULL, 10);
328 rectime = atoi(argv[4]);
329 fd = search(adapter_nr, channel_freq, channel_id);
333 ret = track(adapter_nr);
334 record(adapter_nr, argv[5], rectime);