diff options
| author | Quentin Young <qlyoung@nvidia.com> | 2020-09-02 16:43:29 -0400 |
|---|---|---|
| committer | Quentin Young <qlyoung@nvidia.com> | 2020-09-03 14:23:54 -0400 |
| commit | 06cf2c0c36e044dcdc4cdd5f7d6e971bc07a294c (patch) | |
| tree | 745d1fe8f9b565bfc8af41319cefdffa4f008054 /lib/stream.h | |
| parent | e6464fdc184e4b0a4032286c8fae245dcdb03d8f (diff) | |
lib: add stream_rewind_getp()
stream_forward_getp() cannot be used with negative numbers due to the
size_t argument, we'll end up doing overflow arithmetic.
Signed-off-by: Quentin Young <qlyoung@nvidia.com>
Diffstat (limited to 'lib/stream.h')
| -rw-r--r-- | lib/stream.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/lib/stream.h b/lib/stream.h index f2c16b3486..e78006fbcc 100644 --- a/lib/stream.h +++ b/lib/stream.h @@ -174,6 +174,8 @@ extern void stream_set_getp(struct stream *, size_t); extern void stream_set_endp(struct stream *, size_t); extern void stream_forward_getp(struct stream *, size_t); extern bool stream_forward_getp2(struct stream *, size_t); +extern void stream_rewind_getp(struct stream *s, size_t size); +extern bool stream_rewind_getp2(struct stream *s, size_t size); extern void stream_forward_endp(struct stream *, size_t); extern bool stream_forward_endp2(struct stream *, size_t); @@ -461,6 +463,12 @@ static inline const uint8_t *ptr_get_be32(const uint8_t *ptr, uint32_t *out) goto stream_failure; \ } while (0) +#define STREAM_REWIND_GETP(STR, SIZE) \ + do { \ + if (!stream_rewind_getp2((STR), (SIZE))) \ + goto stream_failure; \ + } while (0) + #define STREAM_FORWARD_ENDP(STR, SIZE) \ do { \ if (!stream_forward_endp2((STR), (SIZE))) \ |