Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 1 | /* |
| 2 | zip_io_util.c -- I/O helper functions |
Thomas Klausner | e91cc0f | 2019-03-12 12:39:36 +0100 | [diff] [blame] | 3 | Copyright (C) 1999-2018 Dieter Baron and Thomas Klausner |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 4 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 5 | This file is part of libzip, a library to manipulate ZIP archives. |
| 6 | The authors can be contacted at <libzip@nih.at> |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 7 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 8 | Redistribution and use in source and binary forms, with or without |
| 9 | modification, are permitted provided that the following conditions |
| 10 | are met: |
| 11 | 1. Redistributions of source code must retain the above copyright |
| 12 | notice, this list of conditions and the following disclaimer. |
| 13 | 2. Redistributions in binary form must reproduce the above copyright |
| 14 | notice, this list of conditions and the following disclaimer in |
| 15 | the documentation and/or other materials provided with the |
| 16 | distribution. |
| 17 | 3. The names of the authors may not be used to endorse or promote |
| 18 | products derived from this software without specific prior |
| 19 | written permission. |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 20 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 21 | THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS |
| 22 | OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
| 23 | WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| 24 | ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY |
| 25 | DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| 26 | DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE |
| 27 | GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS |
| 28 | INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER |
| 29 | IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR |
| 30 | OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN |
| 31 | IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
| 32 | */ |
| 33 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 34 | #include <stdlib.h> |
| 35 | #include <string.h> |
| 36 | |
| 37 | #include "zipint.h" |
| 38 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 39 | int |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 40 | _zip_read(zip_source_t *src, zip_uint8_t *b, zip_uint64_t length, zip_error_t *error) { |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 41 | zip_int64_t n; |
| 42 | |
| 43 | if (length > ZIP_INT64_MAX) { |
| 44 | zip_error_set(error, ZIP_ER_INTERNAL, 0); |
| 45 | return -1; |
| 46 | } |
| 47 | |
| 48 | if ((n = zip_source_read(src, b, length)) < 0) { |
Thomas Klausner | d97f544 | 2014-09-23 17:02:03 +0200 | [diff] [blame] | 49 | _zip_error_set_from_source(error, src); |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 50 | return -1; |
| 51 | } |
| 52 | |
| 53 | if (n < (zip_int64_t)length) { |
| 54 | zip_error_set(error, ZIP_ER_EOF, 0); |
| 55 | return -1; |
| 56 | } |
| 57 | |
| 58 | return 0; |
| 59 | } |
| 60 | |
| 61 | |
| 62 | zip_uint8_t * |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 63 | _zip_read_data(zip_buffer_t *buffer, zip_source_t *src, size_t length, bool nulp, zip_error_t *error) { |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 64 | zip_uint8_t *r; |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 65 | |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 66 | if (length == 0 && !nulp) { |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 67 | return NULL; |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 68 | } |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 69 | |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 70 | r = (zip_uint8_t *)malloc(length + (nulp ? 1 : 0)); |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 71 | if (!r) { |
| 72 | zip_error_set(error, ZIP_ER_MEMORY, 0); |
| 73 | return NULL; |
| 74 | } |
| 75 | |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 76 | if (buffer) { |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 77 | zip_uint8_t *data = _zip_buffer_get(buffer, length); |
| 78 | |
| 79 | if (data == NULL) { |
| 80 | zip_error_set(error, ZIP_ER_MEMORY, 0); |
| 81 | free(r); |
| 82 | return NULL; |
| 83 | } |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 84 | memcpy(r, data, length); |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 85 | } |
| 86 | else { |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 87 | if (_zip_read(src, r, length, error) < 0) { |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 88 | free(r); |
| 89 | return NULL; |
| 90 | } |
| 91 | } |
| 92 | |
| 93 | if (nulp) { |
| 94 | zip_uint8_t *o; |
| 95 | /* replace any in-string NUL characters with spaces */ |
Dieter Baron | ee25b7d | 2014-10-11 18:30:13 +0200 | [diff] [blame] | 96 | r[length] = 0; |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 97 | for (o = r; o < r + length; o++) |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 98 | if (*o == '\0') |
| 99 | *o = ' '; |
| 100 | } |
| 101 | |
| 102 | return r; |
| 103 | } |
| 104 | |
| 105 | |
Dieter Baron | 1d9dfeb | 2014-09-28 23:02:54 +0200 | [diff] [blame] | 106 | zip_string_t * |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 107 | _zip_read_string(zip_buffer_t *buffer, zip_source_t *src, zip_uint16_t len, bool nulp, zip_error_t *error) { |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 108 | zip_uint8_t *raw; |
Dieter Baron | 1d9dfeb | 2014-09-28 23:02:54 +0200 | [diff] [blame] | 109 | zip_string_t *s; |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 110 | |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 111 | if ((raw = _zip_read_data(buffer, src, len, nulp, error)) == NULL) |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 112 | return NULL; |
| 113 | |
| 114 | s = _zip_string_new(raw, len, ZIP_FL_ENC_GUESS, error); |
| 115 | free(raw); |
| 116 | return s; |
| 117 | } |
| 118 | |
| 119 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 120 | int |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 121 | _zip_write(zip_t *za, const void *data, zip_uint64_t length) { |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 122 | zip_int64_t n; |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 123 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 124 | if ((n = zip_source_write(za->src, data, length)) < 0) { |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 125 | _zip_error_set_from_source(&za->error, za->src); |
| 126 | return -1; |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 127 | } |
| 128 | if ((zip_uint64_t)n != length) { |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 129 | zip_error_set(&za->error, ZIP_ER_WRITE, EINTR); |
| 130 | return -1; |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 131 | } |
Thomas Klausner | 8eab1a2 | 2018-01-15 14:10:11 +0100 | [diff] [blame] | 132 | |
Thomas Klausner | ea8ba49 | 2014-09-23 16:54:47 +0200 | [diff] [blame] | 133 | return 0; |
| 134 | } |