diff options
Diffstat (limited to 'gatekeeper_messages.cpp')
-rw-r--r-- | gatekeeper_messages.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/gatekeeper_messages.cpp b/gatekeeper_messages.cpp index fc76d5e..41972bb 100644 --- a/gatekeeper_messages.cpp +++ b/gatekeeper_messages.cpp @@ -52,8 +52,8 @@ static inline gatekeeper_error_t read_from_buffer(const uint8_t **buffer, const memcpy(&target->length, *buffer, sizeof(target->length)); *buffer += sizeof(target->length); if (target->length != 0) { - const uint8_t *buffer_end = *buffer + target->length; - if (buffer_end > end || buffer_end <= *buffer) return ERROR_INVALID; + const size_t buffer_size = end - *buffer; + if (buffer_size < target->length) return ERROR_INVALID; target->buffer.reset(new uint8_t[target->length]); memcpy(target->buffer.get(), *buffer, target->length); |