git.sesse.net Git - ffmpeg/blob - tools/aviocat.c

   1 /*
   2  * Copyright (c) 2012 Martin Storsjo
   3  *
   4  * This file is part of Libav.
   5  *
   6  * Libav is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * Libav is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with Libav; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  19  */
  20
  21 #include <stdio.h>
  22 #include <string.h>
  23 #include <unistd.h>
  24 #include <sys/stat.h>
  25 #include "libavformat/avformat.h"
  26 #include "libavformat/riff.h"
  27 #include "libavutil/intreadwrite.h"
  28 #include "libavutil/mathematics.h"
  29
  30 static int usage(const char *argv0, int ret)
  31 {
  32     fprintf(stderr, "%s [-b bytespersec] input_url output_url\n", argv0);
  33     return ret;
  34 }
  35
  36 int main(int argc, char **argv)
  37 {
  38     int bps = 0, ret, i;
  39     const char *input_url = NULL, *output_url = NULL;
  40     int64_t stream_pos = 0;
  41     int64_t start_time;
  42     char errbuf[50];
  43     AVIOContext *input, *output;
  44
  45     av_register_all();
  46     avformat_network_init();
  47
  48     for (i = 1; i < argc; i++) {
  49         if (!strcmp(argv[i], "-b")) {
  50             bps = atoi(argv[i + 1]);
  51             i++;
  52         } else if (!input_url) {
  53             input_url = argv[i];
  54         } else if (!output_url) {
  55             output_url = argv[i];
  56         } else {
  57             return usage(argv[0], 1);
  58         }
  59     }
  60     if (!output_url)
  61         return usage(argv[0], 1);
  62
  63     ret = avio_open2(&input, input_url, AVIO_FLAG_READ, NULL, NULL);
  64     if (ret) {
  65         av_strerror(ret, errbuf, sizeof(errbuf));
  66         fprintf(stderr, "Unable to open %s: %s\n", input_url, errbuf);
  67         return 1;
  68     }
  69     ret = avio_open2(&output, output_url, AVIO_FLAG_WRITE, NULL, NULL);
  70     if (ret) {
  71         av_strerror(ret, errbuf, sizeof(errbuf));
  72         fprintf(stderr, "Unable to open %s: %s\n", output_url, errbuf);
  73         goto fail;
  74     }
  75
  76     start_time = av_gettime();
  77     while (1) {
  78         uint8_t buf[1024];
  79         int n;
  80         n = avio_read(input, buf, sizeof(buf));
  81         if (n <= 0)
  82             break;
  83         avio_write(output, buf, n);
  84         stream_pos += n;
  85         if (bps) {
  86             avio_flush(output);
  87             while ((av_gettime() - start_time)*bps/AV_TIME_BASE < stream_pos)
  88                 usleep(50*1000);
  89         }
  90     }
  91
  92     avio_close(output);
  93 fail:
  94     avio_close(input);
  95     avformat_network_deinit();
  96     return ret ? 1 : 0;
  97 }