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;
avformat_network_init();
for (i = 1; i < argc; i++) {
- if (!strcmp(argv[i], "-b")) {
+ 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) {
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));
goto fail;
}
- start_time = av_gettime();
+ start_time = av_gettime_relative();
while (1) {
uint8_t buf[1024];
int n;
stream_pos += n;
if (bps) {
avio_flush(output);
- while ((av_gettime() - start_time) * bps / AV_TIME_BASE < stream_pos)
+ while ((av_gettime_relative() - start_time) * bps / AV_TIME_BASE < stream_pos)
av_usleep(50 * 1000);
}
}