[FFmpeg-devel] [PATCH] libRIST: allow setting fifo size and fail on overflow.
Gijs Peskens
gijs at peskens.net
Tue Jan 11 11:45:45 EET 2022
On 11-01-2022 10:43, "zhilizhao(赵志立)" wrote:
>
>> On Jan 11, 2022, at 5:23 PM, Gijs Peskens <gijs at peskens.net> wrote:
>>
>> Introduce fifo_size and overrun_nonfatal params to configure fifo buffer
>> behavior.
>>
>> Use newly introduced RIST_DATA_FLAGS_OVERFLOW flag to check for overrun
>> and error out in that case.
>> ---
>> doc/protocols.texi | 9 +++++++++
>> libavformat/librist.c | 40 +++++++++++++++++++++++++++++++++++++++-
>> 2 files changed, 48 insertions(+), 1 deletion(-)
>>
>> diff --git a/doc/protocols.texi b/doc/protocols.texi
>> index d207df0b52..f1acf0cc77 100644
>> --- a/doc/protocols.texi
>> +++ b/doc/protocols.texi
>> @@ -745,6 +745,15 @@ Set internal RIST buffer size in milliseconds for retransmission of data.
>> Default value is 0 which means the librist default (1 sec). Maximum value is 30
>> seconds.
>>
>> + at item fifo_size
>> +Size of the librist receiver output fifo in number of packets. This must be a
>> +power of 2.
>> +Defaults to 8192 (vs the libRIST default of 1024).
>> +
>> + at item overrun_nonfatal=@var{1|0}
>> +Survive in case of libRIST fifo buffer overrun. Default
>> +value is 0.
>> +
>> @item pkt_size
>> Set maximum packet size for sending data. 1316 by default.
>>
>> diff --git a/libavformat/librist.c b/libavformat/librist.c
>> index 378b635ea7..83bbf9f07a 100644
>> --- a/libavformat/librist.c
>> +++ b/libavformat/librist.c
>> @@ -43,6 +43,9 @@
>> ((patch) + ((minor)* 0x100) + ((major) *0x10000))
>> #define FF_LIBRIST_VERSION FF_LIBRIST_MAKE_VERSION(LIBRIST_API_VERSION_MAJOR, LIBRIST_API_VERSION_MINOR, LIBRIST_API_VERSION_PATCH)
>> #define FF_LIBRIST_VERSION_41 FF_LIBRIST_MAKE_VERSION(4, 1, 0)
>> +#define FF_LIBRIST_VERSION_42 FF_LIBRIST_MAKE_VERSION(4, 2, 0)
>> +
>> +#define FF_LIBRIST_FIFO_SIZE_DEFAULT (2 << 12)//8192
> It’s not easy to read without space before ‘//8192’.
> I prefer
>
> #define FF_LIBRIST_FIFO_SIZE_DEFAULT 8192
>
>> typedef struct RISTContext {
>> const AVClass *class;
>> @@ -52,6 +55,8 @@ typedef struct RISTContext {
>> int packet_size;
>> int log_level;
>> int encryption;
>> + int fifo_size;
>> + bool overrun_nonfatal;
> Should be ‘int’, please check opt.c.
>
>> char *secret;
>>
>> struct rist_logging_settings logging_settings;
>> @@ -70,6 +75,8 @@ static const AVOption librist_options[] = {
>> { "main", NULL, 0, AV_OPT_TYPE_CONST, {.i64=RIST_PROFILE_MAIN}, 0, 0, .flags = D|E, "profile" },
>> { "advanced", NULL, 0, AV_OPT_TYPE_CONST, {.i64=RIST_PROFILE_ADVANCED}, 0, 0, .flags = D|E, "profile" },
>> { "buffer_size", "set buffer_size in ms", OFFSET(buffer_size), AV_OPT_TYPE_INT, {.i64=0}, 0, 30000, .flags = D|E },
>> + { "fifo_size", "Set libRIST fifo buffer. Size must be power of 2", OFFSET(fifo_size), AV_OPT_TYPE_INT, {.i64=FF_LIBRIST_FIFO_SIZE_DEFAULT}, 2 << 9, 2 << 15, .flags = D|E },
>> + { "overrun_nonfatal", "survive in case of libRIST receiving circular buffer overrun", OFFSET(overrun_nonfatal), AV_OPT_TYPE_BOOL, {.i64 = 0}, 0, 1, D },
>> { "pkt_size", "set packet size", OFFSET(packet_size), AV_OPT_TYPE_INT, {.i64=1316}, 1, MAX_PAYLOAD_SIZE, .flags = D|E },
>> { "log_level", "set loglevel", OFFSET(log_level), AV_OPT_TYPE_INT, {.i64=RIST_LOG_INFO}, -1, INT_MAX, .flags = D|E },
>> { "secret", "set encryption secret",OFFSET(secret), AV_OPT_TYPE_STRING,{.str=NULL}, 0, 0, .flags = D|E },
>> @@ -90,7 +97,6 @@ static int risterr2ret(int err)
>> static int log_cb(void *arg, enum rist_log_level log_level, const char *msg)
>> {
>> int level;
>> -
> Unrelated changes.
>
>> switch (log_level) {
>> case RIST_LOG_ERROR: level = AV_LOG_ERROR; break;
>> case RIST_LOG_WARN: level = AV_LOG_WARNING; break;
>> @@ -139,6 +145,7 @@ static int librist_open(URLContext *h, const char *uri, int flags)
>> h->max_packet_size = s->packet_size;
>> ret = rist_sender_create(&s->ctx, s->profile, 0, logging_settings);
>> }
>> +
>> if (ret < 0)
>> goto err;
>>
>> @@ -146,6 +153,7 @@ static int librist_open(URLContext *h, const char *uri, int flags)
>> h->max_packet_size = MAX_PAYLOAD_SIZE;
>> ret = rist_receiver_create(&s->ctx, s->profile, logging_settings);
>> }
>> +
>> if (ret < 0)
>> goto err;
>>
>> @@ -161,6 +169,20 @@ static int librist_open(URLContext *h, const char *uri, int flags)
>> if (ret < 0)
>> goto err;
>>
>> + //Prior to 4.2.0 there was a bug in libRIST which made this call always fail.
>> +#if FF_LIBRIST_VERSION >= FF_LIBRIST_VERSION_42
>> + if (flags & AVIO_FLAG_READ) {
>> + ret = rist_receiver_set_output_fifo_size(s->ctx, s->fifo_size);
>> + if (ret != 0) {
>> + goto err;
>> + }
>> + }
>> +#else
>> + if (s->fifo_size != FF_LIBRIST_FIFO_SIZE_DEFAULT) {
>> + av_log(h, AV_LOG_ERROR, "libRIST prior to 0.2.7 has a bug which fails setting the fifo buffer size");
> Missing newline at the end of log message.
>
>> + }
>> +#endif
>> +
>> if (((s->encryption == 128 || s->encryption == 256) && !s->secret) ||
>> ((peer_config->key_size == 128 || peer_config->key_size == 256) && !peer_config->secret[0])) {
>> av_log(h, AV_LOG_ERROR, "secret is mandatory if encryption is enabled\n");
>> @@ -223,8 +245,24 @@ static int librist_read(URLContext *h, uint8_t *buf, int size)
>> return AVERROR_EXTERNAL;
>> }
>>
>> +#if FF_LIBRIST_VERSION >= FF_LIBRIST_VERSION_42
>> + if (data_block->flags & RIST_DATA_FLAGS_OVERFLOW == RIST_DATA_FLAGS_OVERFLOW) {
>> + if (!s->overrun_nonfatal) {
>> + av_log(h, AV_LOG_ERROR, "Fifo buffer overrun. "
>> + "To avoid, increase fifo_size URL option. "
>> + "To survive in such case, use overrun_nonfatal option\n");
>> + size = AVERROR(EIO);
>> + goto out_free;
>> + } else {
>> + av_log(h, AV_LOG_WARNING, "Fifo buffer buffer overrun. "
>> + "Surviving due to overrun_nonfatal option\n");
> 'buffer buffer' ?
>
>> + }
>> + }
>> +#endif
>> +
>> size = data_block->payload_len;
>> memcpy(buf, data_block->payload, size);
>> +out_free:
>> #if FF_LIBRIST_VERSION < FF_LIBRIST_VERSION_41
>> rist_receiver_data_block_free((struct rist_data_block**)&data_block);
>> #else
>> --
>> 2.32.0
>>
>> _______________________________________________
>> ffmpeg-devel mailing list
>> ffmpeg-devel at ffmpeg.org
>> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>>
>> To unsubscribe, visit link above, or email
>> ffmpeg-devel-request at ffmpeg.org with subject "unsubscribe".
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel at ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request at ffmpeg.org with subject "unsubscribe".
Thanks will correct!
More information about the ffmpeg-devel
mailing list