comparison recpt1/recpt1.c @ 8:6da603afd363

added udp capability

7:407af34cfbd9

#include <string.h>
#include <pthread.h>
#include <unistd.h>
#include <getopt.h>

#include <sys/ioctl.h>
#include "pt1_ioctl.h"

#include "recpt1.h"
#include "decoder.h"

/* globals */
int f_exit = FALSE;

typedef struct thread_data {
    QUEUE_T *queue;
    decoder *decoder;
    int fd;
} thread_data;

/* lookup frequency conversion table*/
ISDB_T_FREQ_CONV_TABLE *
searchrecoff(char *channel)

QUEUE_T *
create_queue(size_t size)
{
    QUEUE_T *p_queue;
    int     memsize = sizeof(QUEUE_T) + size * sizeof(BUFSZ);

    p_queue = (QUEUE_T*)calloc(memsize, sizeof(char));

    if(p_queue != NULL) {
        p_queue->size = size;

/* dequeue data. this function will block if queue is empty. */
BUFSZ *
dequeue(QUEUE_T *p_queue)
{
    BUFSZ	*buffer;

    pthread_mutex_lock(&p_queue->mutex);
    /* entered the critical section*/

    /* wait until queue is filled */

write_func(void *p)
{
    thread_data *data = (thread_data *)p;
    QUEUE_T *p_queue = data->queue;
    decoder *dec = data->decoder;
    int wfd = data->fd;
    int use_b25 = dec ? 1 : 0;
    BUFSZ	*buf;
    ARIB_STD_B25_BUFFER sbuf, dbuf;
    int code;

    while(1) {
        buf = dequeue(p_queue);
        /* no entry in the queue */
        if(buf == NULL){

                fprintf(stderr, "b25_decode failed\n");
                close(wfd);
                break;
            }
            write(wfd, dbuf.data, dbuf.size);
            free(buf);
        } else {
            write(wfd, sbuf.data, sbuf.size);
            free(buf);
        }

        /* normal exit */
        if((f_exit) && (!p_queue->no_empty)) {
            if(use_b25) {
                code = b25_finish(dec, &sbuf, &dbuf);
                if(code < 0) {
                    fprintf(stderr, "b25_finish failed\n");
                    close(wfd);
                    break;
                }
                write(wfd, dbuf.data, dbuf.size);
            }
            close(wfd);
            break;
        }
    }

    return NULL;
}

int
main(int argc, char **argv)
{
    int fd, wfd;
    int lp;
    int recsec;
    time_t start_time, cur_time;
    FREQUENCY freq;
    ISDB_T_FREQ_CONV_TABLE *ptr;
    pthread_t dequeue_threads;
    QUEUE_T *p_queue = create_queue(MAX_QUEUE);
    BUFSZ   *bufptr;
    decoder *dec = NULL;
    thread_data tdata;
    decoder_options dopt;

    int use_b25 = 0;
    int result;
    int option_index;
    struct option long_options[] = {
        { "b25",       0, NULL, 'b'},
        { "B25",       0, NULL, 'b'},
        { "round",     1, NULL, 'r'},
        { "strip",     0, NULL, 's'},
        { "emm",       0, NULL, 'm'},
        { "EMM",       0, NULL, 'm'}
    };

    dopt.round = 4;
    dopt.strip = 0;
    dopt.emm = 0;

    while((result = getopt_long(argc, argv, "br:sm", long_options, &option_index)) != -1) {
        switch(result) {
        case 'b':
			use_b25 = 1;
            fprintf(stderr, "using B25...\n");
			break;
        case 's':
			dopt.strip = 1;
            fprintf(stderr, "enable B25 strip\n");
			break;
        case 'm':
			dopt.emm = 1;
            fprintf(stderr, "enable B25 emm processing\n");
			break;
        case 'r':
			dopt.round = atoi(optarg);
            fprintf(stderr, "set round %d\n", dopt.round);
            break;
        case ':':
            fprintf(stderr, "%c needs value\n", result);
            break;
        case '?':
            fprintf(stderr, "Usage: [--b25 [--round N] [--strip] [--EMM]] channel recsec destfile\n");
            break;
        }
    }

	if(argc - optind < 3) {
        printf("Usage %s: [--b25 [--round N] [--strip] [--EMM]] channel recsec destfile\n", argv[0]);
        printf("channel =\n");
        printf("151ch：BS朝日\n");
        printf("161ch：BS-i\n");
        printf("191ch：WOWOW\n");
        printf("171ch：BSジャパン\n");

        printf("102ch：NHK衛星第2放送(BS2)\n");
        printf("103ch：NHKハイビジョン(BShi)\n");
        printf("CS2-CS24：CSチャンネル\n");
        return 1;
    }
	ptr = searchrecoff(argv[optind]);
	if(ptr == NULL){
		printf("Channel Select Error(%s)\n", argv[optind]);
		return 1;
	}

    freq.frequencyno = ptr->set_freq;
    freq.slot = ptr->add_freq;

    if(ptr->type == CHTYPE_SATELLITE) {
        for(lp = 0; lp < 2; lp++) {
            fd = open(bsdev[lp], O_RDONLY);
            if(fd >= 0) {
                break;
            }
        }
        if(fd < 0) {
            fprintf(stderr, "Device Open Error\n");
            return 1;
        }
    } else {
        for(lp = 0; lp < 2; lp++) {
            fd = open(isdb_t_dev[lp], O_RDONLY);
            if(fd >= 0) {
                break;
            }
        }
        if(fd < 0) {
            fprintf(stderr, "Device Open Error\n");
            return 1;
        }
    }
	recsec = atoi(argv[optind + 1]);

    /* initialize decoder */
    if(use_b25) {
        dec = b25_startup(&dopt);
        if(!dec) {

            fprintf(stderr, "fall back to encrypted recording\n");
            use_b25 = 0;
        }
    }

    /* open output file */
    wfd = open(argv[optind + 2], (O_RDWR | O_CREAT | O_TRUNC), 0666);
    if(wfd < 0) {
        fprintf(stderr, "Output File Open Error(%s)\n", argv[optind + 2]);
        return 1;
    }

    if(ioctl(fd, SET_CHANNEL, &freq) < 0) {
        fprintf(stderr, "Tuner Select Error\n");
        return 1;
    }

    /* make reader thread */
    tdata.queue = p_queue;
    tdata.decoder = dec;
    tdata.fd = wfd;
    pthread_create(&dequeue_threads, NULL, write_func, &tdata);

    /* start recording */
    if(ioctl(fd, START_REC, 0) < 0) {
        fprintf(stderr, "Tuner Start Error\n");
        return 1;
    }

    time(&start_time);

    /* read from tuner */
    while(1) {
        time(&cur_time);
        bufptr = malloc(sizeof(BUFSZ));
        bufptr->size = read(fd, bufptr->buffer, MAX_READ_SIZE);
        if(bufptr->size <= 0) {
            if((cur_time - start_time) >= recsec) {
                f_exit = TRUE;
                enqueue(p_queue, NULL);
                break;

        }
        enqueue(p_queue, bufptr);

        /* stop recording */
        if((cur_time - start_time) >= recsec) {
            ioctl(fd, STOP_REC, 0);
            /* read remaining data */
            while(1) {
                bufptr = malloc(sizeof(BUFSZ));
                bufptr->size = read(fd, bufptr->buffer, MAX_READ_SIZE);
                if(bufptr->size <= 0) {
                    f_exit = TRUE;
                    enqueue(p_queue, NULL);
                    break;
                }

            }
            break;
        }
    }
    /* close tuner */
    close(fd);

    /* wait reader thread */
    pthread_join(dequeue_threads, NULL);
    destroy_queue(p_queue);

    /* release decoder */
    if(use_b25) {
        b25_shutdown(dec);
    }

8:6da603afd363

#include <string.h>
#include <pthread.h>
#include <unistd.h>
#include <getopt.h>

#include <netdb.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#include <sys/ioctl.h>
#include "pt1_ioctl.h"

#include "recpt1.h"
#include "decoder.h"

/* globals */
int f_exit = FALSE;

typedef struct sock_data {
    int sfd;    /* socket fd */
    struct sockaddr_in addr;
} sock_data;

typedef struct thread_data {
    QUEUE_T *queue;
    decoder *decoder;
    int wfd;    /* output file fd */
    sock_data *sock_data;
} thread_data;

/* lookup frequency conversion table*/
ISDB_T_FREQ_CONV_TABLE *
searchrecoff(char *channel)

QUEUE_T *
create_queue(size_t size)
{
    QUEUE_T *p_queue;
    int memsize = sizeof(QUEUE_T) + size * sizeof(BUFSZ);

    p_queue = (QUEUE_T*)calloc(memsize, sizeof(char));

    if(p_queue != NULL) {
        p_queue->size = size;

/* dequeue data. this function will block if queue is empty. */
BUFSZ *
dequeue(QUEUE_T *p_queue)
{
    BUFSZ *buffer;

    pthread_mutex_lock(&p_queue->mutex);
    /* entered the critical section*/

    /* wait until queue is filled */

write_func(void *p)
{
    thread_data *data = (thread_data *)p;
    QUEUE_T *p_queue = data->queue;
    decoder *dec = data->decoder;
    int wfd = data->wfd;
    int use_b25 = dec ? 1 : 0;
    int use_udp = data->sock_data ? 1 : 0;
    int sfd = 0;
    struct sockaddr *addr = NULL;
    BUFSZ *buf;
    ARIB_STD_B25_BUFFER sbuf, dbuf;
    int code;

    if(use_udp) {
        sfd = data->sock_data->sfd;
        addr = (struct sockaddr *)&data->sock_data->addr;
    }

    while(1) {
        buf = dequeue(p_queue);
        /* no entry in the queue */
        if(buf == NULL){

                fprintf(stderr, "b25_decode failed\n");
                close(wfd);
                break;
            }
            write(wfd, dbuf.data, dbuf.size);

            if(use_udp && sfd) {
                sendto(sfd, dbuf.data, dbuf.size, 0,
                       addr, sizeof(struct sockaddr_in));
            }
            free(buf);
        } else {
            write(wfd, sbuf.data, sbuf.size);

            if(use_udp && sfd) {
                sendto(sfd, sbuf.data, sbuf.size, 0,
                       addr, sizeof(struct sockaddr_in));
            }
            free(buf);
        }

        /* normal exit */
        if((f_exit) && (!p_queue->no_empty)) {
            if(use_b25) {
                code = b25_finish(dec, &sbuf, &dbuf);
                if(code < 0) {
                    fprintf(stderr, "b25_finish failed\n");
                    close(wfd);
                    close(sfd);
                    break;
                }
                write(wfd, dbuf.data, dbuf.size);

                if(use_udp && sfd) {
                    sendto(sfd, dbuf.data, dbuf.size, 0,
                           addr, sizeof(struct sockaddr_in));
                }
            }
            close(wfd);
            close(sfd);
            break;
        }
    }

    return NULL;
}

void
show_usage(char *cmd)
{
    fprintf(stderr, "Usage: %s [--b25 [--round N] [--strip] [--EMM]] [--udp hostname [--port port]] channel recsec destfile\n", cmd);
}

int
main(int argc, char **argv)
{
    int tfd, wfd;
    int lp;
    int recsec;
    time_t start_time, cur_time;
    FREQUENCY freq;
    ISDB_T_FREQ_CONV_TABLE *ptr;
    pthread_t dequeue_threads;
    QUEUE_T *p_queue = create_queue(MAX_QUEUE);
    BUFSZ   *bufptr;
    decoder *dec = NULL;
    thread_data tdata;
    decoder_options dopt = {
        4,  /* round */
        0,  /* strip */
        0   /* emm */
    };

    int result;
    int option_index;
    struct option long_options[] = {
        { "b25",       0, NULL, 'b'},
        { "B25",       0, NULL, 'b'},
        { "round",     1, NULL, 'r'},
        { "strip",     0, NULL, 's'},
        { "emm",       0, NULL, 'm'},
        { "EMM",       0, NULL, 'm'},
        { "udp",       1, NULL, 'u'},
        { "port",      1, NULL, 'p'}
    };

    int use_b25 = 0;
    int use_udp = 0;
    char *host_to = NULL;
    int port_to = 1234;
    sock_data *sdata = NULL;

    while((result = getopt_long(argc, argv, "br:sm" "u:p:", long_options, &option_index)) != -1) {
        switch(result) {
        case 'b':
            use_b25 = 1;
            fprintf(stderr, "using B25...\n");
            break;
        case 's':
            dopt.strip = 1;
            fprintf(stderr, "enable B25 strip\n");
            break;
        case 'm':
            dopt.emm = 1;
            fprintf(stderr, "enable B25 emm processing\n");
            break;
        case 'r':
            dopt.round = atoi(optarg);
            fprintf(stderr, "set round %d\n", dopt.round);
            break;
        case ':':
            fprintf(stderr, "%c needs value\n", result);
            break;
        case '?':
            show_usage(argv[0]);
            break;
        case 'u':
            use_udp = 1;
            host_to = optarg;
            fprintf(stderr, "UDP destination address: %s\n", host_to);
            break;
        case 'p':
            port_to = atoi(optarg);
            fprintf(stderr, "UDP port: %d\n", port_to);
            break;
        }
    }

    if(argc - optind < 3) {
        show_usage(argv[0]);
        printf("channel =\n");
        printf("151ch：BS朝日\n");
        printf("161ch：BS-i\n");
        printf("191ch：WOWOW\n");
        printf("171ch：BSジャパン\n");

        printf("102ch：NHK衛星第2放送(BS2)\n");
        printf("103ch：NHKハイビジョン(BShi)\n");
        printf("CS2-CS24：CSチャンネル\n");
        return 1;
    }
    ptr = searchrecoff(argv[optind]);
    if(ptr == NULL){
        printf("Channel Select Error(%s)\n", argv[optind]);
        return 1;
    }

    freq.frequencyno = ptr->set_freq;
    freq.slot = ptr->add_freq;

    if(ptr->type == CHTYPE_SATELLITE) {
        for(lp = 0; lp < 2; lp++) {
            tfd = open(bsdev[lp], O_RDONLY);
            if(tfd >= 0) {
                break;
            }
        }
        if(tfd < 0) {
            fprintf(stderr, "Device Open Error\n");
            return 1;
        }
    } else {
        for(lp = 0; lp < 2; lp++) {
            tfd = open(isdb_t_dev[lp], O_RDONLY);
            if(tfd >= 0) {
                break;
            }
        }
        if(tfd < 0) {
            fprintf(stderr, "Device Open Error\n");
            return 1;
        }
    }
    recsec = atoi(argv[optind + 1]);

    /* initialize decoder */
    if(use_b25) {
        dec = b25_startup(&dopt);
        if(!dec) {

            fprintf(stderr, "fall back to encrypted recording\n");
            use_b25 = 0;
        }
    }

    /* initialize udp connection */
    if(use_udp) {
      sdata = calloc(1, sizeof(sock_data));
      struct in_addr ia;
      ia.s_addr = inet_addr(host_to);
      if(ia.s_addr == INADDR_NONE) {
            struct hostent *hoste = gethostbyname(host_to);
            if(!hoste) {
                perror("failed to get host by name");
                return 1;
            }
            ia.s_addr = *(in_addr_t*) (hoste->h_addr_list[0]);
        }
        sdata->addr.sin_family = AF_INET;
        sdata->addr.sin_port = htons (port_to);
        sdata->addr.sin_addr.s_addr = ia.s_addr;
        if((sdata->sfd = socket(PF_INET, SOCK_DGRAM, 0)) < 0) {
            perror("socket");
            return 1;
        }
    }

    /* open output file */
    wfd = open(argv[optind + 2], (O_RDWR | O_CREAT | O_TRUNC), 0666);
    if(wfd < 0) {
        fprintf(stderr, "Output File Open Error(%s)\n", argv[optind + 2]);
        return 1;
    }

    if(ioctl(tfd, SET_CHANNEL, &freq) < 0) {
        fprintf(stderr, "Tuner Select Error\n");
        return 1;
    }

    /* make reader thread */
    tdata.queue = p_queue;
    tdata.decoder = dec;
    tdata.wfd = wfd;
    tdata.sock_data = sdata;
    pthread_create(&dequeue_threads, NULL, write_func, &tdata);

    /* start recording */
    if(ioctl(tfd, START_REC, 0) < 0) {
        fprintf(stderr, "Tuner Start Error\n");
        return 1;
    }

    time(&start_time);

    /* read from tuner */
    while(1) {
        time(&cur_time);
        bufptr = malloc(sizeof(BUFSZ));
        bufptr->size = read(tfd, bufptr->buffer, MAX_READ_SIZE);
        if(bufptr->size <= 0) {
            if((cur_time - start_time) >= recsec) {
                f_exit = TRUE;
                enqueue(p_queue, NULL);
                break;

        }
        enqueue(p_queue, bufptr);

        /* stop recording */
        if((cur_time - start_time) >= recsec) {
            ioctl(tfd, STOP_REC, 0);
            /* read remaining data */
            while(1) {
                bufptr = malloc(sizeof(BUFSZ));
                bufptr->size = read(tfd, bufptr->buffer, MAX_READ_SIZE);
                if(bufptr->size <= 0) {
                    f_exit = TRUE;
                    enqueue(p_queue, NULL);
                    break;
                }

            }
            break;
        }
    }
    /* close tuner */
    close(tfd);

    /* wait reader thread */
    pthread_join(dequeue_threads, NULL);
    destroy_queue(p_queue);

    /* close socket */
    if(use_udp) {
        close(sdata->sfd);
        free(sdata);
    }

    /* release decoder */
    if(use_b25) {
        b25_shutdown(dec);
    }