diff options
Diffstat (limited to '')
| -rw-r--r-- | libkeccak_state_unmarshal.c | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/libkeccak_state_unmarshal.c b/libkeccak_state_unmarshal.c index af1f7de..e46db54 100644 --- a/libkeccak_state_unmarshal.c +++ b/libkeccak_state_unmarshal.c @@ -5,7 +5,8 @@ /** * Unmarshal a `struct libkeccak_state` from a buffer * - * @param state The slot for the unmarshalled state, must not be initialised (memory leak otherwise) + * @param state The slot for the unmarshalled state, must not be + * initialised (memory leak otherwise), can be `NULL` * @param data The input buffer * @return The number of bytes read from `data`, 0 on error */ @@ -14,6 +15,10 @@ libkeccak_state_unmarshal(struct libkeccak_state *restrict state, const void *re { #define get(type, var) state->var = *((const type *)data), data += sizeof(type) / sizeof(char) const unsigned char *restrict data = data_; + if (!state) { + data += (7 * sizeof(long int) + 26 * sizeof(int64_t)) / sizeof(char); + return sizeof(struct libkeccak_state) - sizeof(char *) + *(const size_t *)data * sizeof(char); + } get(long int, r); get(long int, c); get(long int, n); |