diff options
| author | Alex Converse <alex.converse@gmail.com> | 2011-10-14 18:27:59 -0700 |
|---|---|---|
| committer | Alex Converse <alex.converse@gmail.com> | 2011-10-16 23:56:21 -0700 |
| commit | ab2940691ba76e1a9b0ce608db0dfc45021d741e (patch) | |
| tree | 1bdf5a847797658d1ca33d884ba41b4f02bcc3c3 /libavformat/aviobuf.c | |
| parent | 5a9ee3152b6e69166c7819f07a4992bd22052917 (diff) | |
avio: Check for invalid buffer length.
Diffstat (limited to 'libavformat/aviobuf.c')
| -rw-r--r-- | libavformat/aviobuf.c | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/libavformat/aviobuf.c b/libavformat/aviobuf.c index b5e9d4c61c..8f3599a9dc 100644 --- a/libavformat/aviobuf.c +++ b/libavformat/aviobuf.c @@ -769,13 +769,14 @@ int avio_get_str(AVIOContext *s, int maxlen, char *buf, int buflen) { int i; + if (buflen <= 0) + return AVERROR(EINVAL); // reserve 1 byte for terminating 0 buflen = FFMIN(buflen - 1, maxlen); for (i = 0; i < buflen; i++) if (!(buf[i] = avio_r8(s))) return i + 1; - if (buflen) - buf[i] = 0; + buf[i] = 0; for (; i < maxlen; i++) if (!avio_r8(s)) return i + 1; @@ -787,6 +788,8 @@ int avio_get_str(AVIOContext *s, int maxlen, char *buf, int buflen) {\ char* q = buf;\ int ret = 0;\ + if (buflen <= 0) \ + return AVERROR(EINVAL); \ while (ret + 1 < maxlen) {\ uint8_t tmp;\ uint32_t ch;\ |