Merge commit '87acd33c092ab9e7d73686627e9105d99c1e4928'

* commit '87acd33c092ab9e7d73686627e9105d99c1e4928':
  aviocat: Add support for specifying the input duration

Merged-by: Michael Niedermayer <michaelni@gmx.at>

1 files changed, 14 insertions, 2 deletions

diff --git a/tools/aviocat.c b/tools/aviocat.c
index 42560cbc41..56b918e44d 100644
--- a/tools/aviocat.c
+++ b/tools/aviocat.c
@@ -26,13 +26,13 @@
 
 static int usage(const char *argv0, int ret)
 {
-    fprintf(stderr, "%s [-b bytespersec] input_url output_url\n", argv0);
+    fprintf(stderr, "%s [-b bytespersec] [-d duration] input_url output_url\n", argv0);
     return ret;
 }
 
 int main(int argc, char **argv)
 {
-    int bps = 0, ret, i;
+    int bps = 0, duration = 0, ret, i;
     const char *input_url = NULL, *output_url = NULL;
     int64_t stream_pos = 0;
     int64_t start_time;
@@ -46,6 +46,9 @@ int main(int argc, char **argv)
         if (!strcmp(argv[i], "-b") && i + 1 < argc) {
             bps = atoi(argv[i + 1]);
             i++;
+        } else if (!strcmp(argv[i], "-d") && i + 1 < argc) {
+            duration = atoi(argv[i + 1]);
+            i++;
         } else if (!input_url) {
             input_url = argv[i];
         } else if (!output_url) {
@@ -63,6 +66,15 @@ int main(int argc, char **argv)
         fprintf(stderr, "Unable to open %s: %s\n", input_url, errbuf);
         return 1;
     }
+    if (duration && !bps) {
+        int64_t size = avio_size(input);
+        if (size < 0) {
+            av_strerror(ret, errbuf, sizeof(errbuf));
+            fprintf(stderr, "Unable to get size of %s: %s\n", input_url, errbuf);
+            goto fail;
+        }
+        bps = size / duration;
+    }
     ret = avio_open2(&output, output_url, AVIO_FLAG_WRITE, NULL, NULL);
     if (ret) {
         av_strerror(ret, errbuf, sizeof(errbuf));