123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321 |
- /*
- zip_buffer.c -- bounds checked access to memory buffer
- Copyright (C) 2014 Dieter Baron and Thomas Klausner
- This file is part of libzip, a library to manipulate ZIP archives.
- The authors can be contacted at <libzip@nih.at>
- Redistribution and use in source and binary forms, with or without
- modification, are permitted provided that the following conditions
- are met:
- 1. Redistributions of source code must retain the above copyright
- notice, this list of conditions and the following disclaimer.
- 2. Redistributions in binary form must reproduce the above copyright
- notice, this list of conditions and the following disclaimer in
- the documentation and/or other materials provided with the
- distribution.
- 3. The names of the authors may not be used to endorse or promote
- products derived from this software without specific prior
- written permission.
- THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS
- OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
- WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
- ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY
- DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE
- GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
- INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER
- IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
- OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
- IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- #include <stdlib.h>
- #include <string.h>
- #include "zipint.h"
- zip_uint8_t *
- _zip_buffer_data(zip_buffer_t *buffer)
- {
- return buffer->data;
- }
- void
- _zip_buffer_free(zip_buffer_t *buffer)
- {
- if (buffer == NULL) {
- return;
- }
- if (buffer->free_data) {
- free(buffer->data);
- }
- free(buffer);
- }
- bool
- _zip_buffer_eof(zip_buffer_t *buffer)
- {
- return buffer->ok && buffer->offset == buffer->size;
- }
- zip_uint8_t *
- _zip_buffer_get(zip_buffer_t *buffer, zip_uint64_t length)
- {
- zip_uint8_t *data;
- if (!buffer->ok || buffer->offset + length < length || buffer->offset + length > buffer->size) {
- buffer->ok = false;
- return NULL;
- }
- data = buffer->data + buffer->offset;
- buffer->offset += length;
- return data;
- }
- zip_uint16_t
- _zip_buffer_get_16(zip_buffer_t *buffer)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 2);
- if (data == NULL) {
- return 0;
- }
- return (zip_uint16_t)(data[0] + (data[1] << 8));
- }
- zip_uint32_t
- _zip_buffer_get_32(zip_buffer_t *buffer)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 4);
- if (data == NULL) {
- return 0;
- }
- return ((((((zip_uint32_t)data[3] << 8) + data[2]) << 8) + data[1]) << 8) + data[0];
- }
- zip_uint64_t
- _zip_buffer_get_64(zip_buffer_t *buffer)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 8);
- if (data == NULL) {
- return 0;
- }
- return ((zip_uint64_t)data[7] << 56) + ((zip_uint64_t)data[6] << 48) + ((zip_uint64_t)data[5] << 40) + ((zip_uint64_t)data[4] << 32) + ((zip_uint64_t)data[3] << 24) + ((zip_uint64_t)data[2] << 16) + ((zip_uint64_t)data[1] << 8) + (zip_uint64_t)data[0];
- }
- zip_uint8_t
- _zip_buffer_get_8(zip_buffer_t *buffer)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 1);
- if (data == NULL) {
- return 0;
- }
- return data[0];
- }
- zip_uint64_t
- _zip_buffer_left(zip_buffer_t *buffer)
- {
- return buffer->ok ? buffer->size - buffer->offset : 0;
- }
- zip_buffer_t *
- _zip_buffer_new(zip_uint8_t *data, zip_uint64_t size)
- {
- bool free_data = (data == NULL);
- zip_buffer_t *buffer;
- if (data == NULL) {
- if ((data = (zip_uint8_t *)malloc(size)) == NULL) {
- return NULL;
- }
- }
- if ((buffer = (zip_buffer_t *)malloc(sizeof(*buffer))) == NULL) {
- if (free_data) {
- free(data);
- }
- return NULL;
- }
- buffer->ok = true;
- buffer->data = data;
- buffer->size = size;
- buffer->offset = 0;
- buffer->free_data = free_data;
- return buffer;
- }
- zip_buffer_t *
- _zip_buffer_new_from_source(zip_source_t *src, zip_uint64_t size, zip_uint8_t *buf, zip_error_t *error)
- {
- zip_buffer_t *buffer;
- if ((buffer = _zip_buffer_new(buf, size)) == NULL) {
- zip_error_set(error, ZIP_ER_MEMORY, 0);
- return NULL;
- }
- if (_zip_read(src, buffer->data, size, error) < 0) {
- _zip_buffer_free(buffer);
- return NULL;
- }
- return buffer;
- }
- zip_uint64_t
- _zip_buffer_offset(zip_buffer_t *buffer)
- {
- return buffer->ok ? buffer->offset : 0;
- }
- bool
- _zip_buffer_ok(zip_buffer_t *buffer)
- {
- return buffer->ok;
- }
- int
- _zip_buffer_put(zip_buffer_t *buffer, const void *src, size_t length)
- {
- zip_uint8_t *dst = _zip_buffer_get(buffer, length);
- if (dst == NULL) {
- return -1;
- }
- memcpy(dst, src, length);
- return 0;
- }
- int
- _zip_buffer_put_16(zip_buffer_t *buffer, zip_uint16_t i)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 2);
- if (data == NULL) {
- return -1;
- }
- data[0] = (zip_uint8_t)(i & 0xff);
- data[1] = (zip_uint8_t)((i >> 8) & 0xff);
- return 0;
- }
- int
- _zip_buffer_put_32(zip_buffer_t *buffer, zip_uint32_t i)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 4);
- if (data == NULL) {
- return -1;
- }
- data[0] = (zip_uint8_t)(i & 0xff);
- data[1] = (zip_uint8_t)((i >> 8) & 0xff);
- data[2] = (zip_uint8_t)((i >> 16) & 0xff);
- data[3] = (zip_uint8_t)((i >> 24) & 0xff);
- return 0;
- }
- int
- _zip_buffer_put_64(zip_buffer_t *buffer, zip_uint64_t i)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 8);
- if (data == NULL) {
- return -1;
- }
- data[0] = (zip_uint8_t)(i & 0xff);
- data[1] = (zip_uint8_t)((i >> 8) & 0xff);
- data[2] = (zip_uint8_t)((i >> 16) & 0xff);
- data[3] = (zip_uint8_t)((i >> 24) & 0xff);
- data[4] = (zip_uint8_t)((i >> 32) & 0xff);
- data[5] = (zip_uint8_t)((i >> 40) & 0xff);
- data[6] = (zip_uint8_t)((i >> 48) & 0xff);
- data[7] = (zip_uint8_t)((i >> 56) & 0xff);
- return 0;
- }
- int
- _zip_buffer_put_8(zip_buffer_t *buffer, zip_uint8_t i)
- {
- zip_uint8_t *data = _zip_buffer_get(buffer, 1);
- if (data == NULL) {
- return -1;
- }
- data[0] = i;
- return 0;
- }
- int
- _zip_buffer_set_offset(zip_buffer_t *buffer, zip_uint64_t offset)
- {
- if (offset > buffer->size) {
- buffer->ok = false;
- return -1;
- }
- buffer->ok = true;
- buffer->offset = offset;
- return 0;
- }
- int
- _zip_buffer_skip(zip_buffer_t *buffer, zip_uint64_t length) {
- zip_uint64_t offset = buffer->offset + length;
- if (offset < buffer->offset) {
- buffer->ok = false;
- return -1;
- }
- return _zip_buffer_set_offset(buffer, offset);
- }
- zip_uint64_t
- _zip_buffer_size(zip_buffer_t *buffer)
- {
- return buffer->size;
- }
|