shorten: Fix out-of-array read

[ffmpeg] / tools / aviocat.c

diff --git a/tools/aviocat.c b/tools/aviocat.c
index c43c69d1be5be80fb260bae5744d94ad2fb980bb..52a96bde2f85d8f580360b151dd2e2a417984b24 100644 (file)
--- a/tools/aviocat.c
+++ b/tools/aviocat.c
@@ -19,13 +19,10 @@
  */
 
 #include <stdio.h>
-#include <string.h>
-#include <unistd.h>
-#include <sys/stat.h>
+#include <stdlib.h>
+
+#include "libavutil/time.h"
 #include "libavformat/avformat.h"
-#include "libavformat/riff.h"
-#include "libavutil/intreadwrite.h"
-#include "libavutil/mathematics.h"
 
 static int usage(const char *argv0, int ret)
 {
@@ -84,12 +81,14 @@ int main(int argc, char **argv)
         stream_pos += n;
         if (bps) {
             avio_flush(output);
-            while ((av_gettime() - start_time)*bps/AV_TIME_BASE < stream_pos)
-                usleep(50*1000);
+            while ((av_gettime() - start_time) * bps / AV_TIME_BASE < stream_pos)
+                av_usleep(50 * 1000);
         }
     }
 
+    avio_flush(output);
     avio_close(output);
+
 fail:
     avio_close(input);
     avformat_network_deinit();