Signed-off-by: Steven Liu <[email protected]>
---
libavformat/libsrt.c | 15 ++++++++++++++-
1 file changed, 14 insertions(+), 1 deletion(-)
diff --git a/libavformat/libsrt.c b/libavformat/libsrt.c
index b5568089fa..91e8c4b55b 100644
--- a/libavformat/libsrt.c
+++ b/libavformat/libsrt.c
@@ -478,6 +478,7 @@ static int libsrt_open(URLContext *h, const char *uri, int
flags)
SRTContext *s = h->priv_data;
const char * p;
char buf[256];
+ int ret = 0;
if (srt_startup() < 0) {
return AVERROR_UNKNOWN;
@@ -564,10 +565,18 @@ static int libsrt_open(URLContext *h, const char *uri,
int flags)
if (av_find_info_tag(buf, sizeof(buf), "streamid", p)) {
av_freep(&s->streamid);
s->streamid = av_strdup(buf);
+ if (!s->streamid) {
+ ret = AVERROR(ENOMEM);
+ goto err;
+ }
}
if (av_find_info_tag(buf, sizeof(buf), "smoother", p)) {
av_freep(&s->smoother);
s->smoother = av_strdup(buf);
+ if(!s->smoother) {
+ ret = AVERROR(ENOMEM);
+ goto err;
+ }
}
if (av_find_info_tag(buf, sizeof(buf), "messageapi", p)) {
s->messageapi = strtol(buf, NULL, 10);
@@ -578,11 +587,15 @@ static int libsrt_open(URLContext *h, const char *uri,
int flags)
} else if (!strcmp(buf, "file")) {
s->transtype = SRTT_FILE;
} else {
- return AVERROR(EINVAL);
+ ret = AVERROR(EINVAL);
+ goto err;
}
}
}
return libsrt_setup(h, uri, flags);
+err:
+ av_freep(&s->smoother);
+ av_freep(&s->streamid);
}
static int libsrt_read(URLContext *h, uint8_t *buf, int size)
--
2.15.1
_______________________________________________
ffmpeg-devel mailing list
[email protected]
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
[email protected] with subject "unsubscribe".