blob: 53f44377bcb82440f7ee5c92c3b5b28452682812 [file] [log] [blame]
Thomas Klausnerea8ba492014-09-23 16:54:47 +02001/*
2 zip_io_util.c -- I/O helper functions
Thomas Klausnere91cc0f2019-03-12 12:39:36 +01003 Copyright (C) 1999-2018 Dieter Baron and Thomas Klausner
Thomas Klausner8eab1a22018-01-15 14:10:11 +01004
Thomas Klausnerea8ba492014-09-23 16:54:47 +02005 This file is part of libzip, a library to manipulate ZIP archives.
6 The authors can be contacted at <libzip@nih.at>
Thomas Klausner8eab1a22018-01-15 14:10:11 +01007
Thomas Klausnerea8ba492014-09-23 16:54:47 +02008 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 Klausner8eab1a22018-01-15 14:10:11 +010020
Thomas Klausnerea8ba492014-09-23 16:54:47 +020021 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 Klausnerea8ba492014-09-23 16:54:47 +020034#include <stdlib.h>
35#include <string.h>
36
37#include "zipint.h"
38
Thomas Klausnerea8ba492014-09-23 16:54:47 +020039int
Thomas Klausner8eab1a22018-01-15 14:10:11 +010040_zip_read(zip_source_t *src, zip_uint8_t *b, zip_uint64_t length, zip_error_t *error) {
Thomas Klausnerea8ba492014-09-23 16:54:47 +020041 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 Klausnerd97f5442014-09-23 17:02:03 +020049 _zip_error_set_from_source(error, src);
Thomas Klausnerea8ba492014-09-23 16:54:47 +020050 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
62zip_uint8_t *
Thomas Klausner8eab1a22018-01-15 14:10:11 +010063_zip_read_data(zip_buffer_t *buffer, zip_source_t *src, size_t length, bool nulp, zip_error_t *error) {
Thomas Klausnerea8ba492014-09-23 16:54:47 +020064 zip_uint8_t *r;
Thomas Klausner8eab1a22018-01-15 14:10:11 +010065
Dieter Baronee25b7d2014-10-11 18:30:13 +020066 if (length == 0 && !nulp) {
Thomas Klausnerea8ba492014-09-23 16:54:47 +020067 return NULL;
Dieter Baronee25b7d2014-10-11 18:30:13 +020068 }
Thomas Klausnerea8ba492014-09-23 16:54:47 +020069
Dieter Baronee25b7d2014-10-11 18:30:13 +020070 r = (zip_uint8_t *)malloc(length + (nulp ? 1 : 0));
Thomas Klausnerea8ba492014-09-23 16:54:47 +020071 if (!r) {
72 zip_error_set(error, ZIP_ER_MEMORY, 0);
73 return NULL;
74 }
75
Dieter Baronee25b7d2014-10-11 18:30:13 +020076 if (buffer) {
Thomas Klausner8eab1a22018-01-15 14:10:11 +010077 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 Baronee25b7d2014-10-11 18:30:13 +020084 memcpy(r, data, length);
Thomas Klausnerea8ba492014-09-23 16:54:47 +020085 }
86 else {
Dieter Baronee25b7d2014-10-11 18:30:13 +020087 if (_zip_read(src, r, length, error) < 0) {
Thomas Klausnerea8ba492014-09-23 16:54:47 +020088 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 Baronee25b7d2014-10-11 18:30:13 +020096 r[length] = 0;
Thomas Klausner8eab1a22018-01-15 14:10:11 +010097 for (o = r; o < r + length; o++)
Thomas Klausnerea8ba492014-09-23 16:54:47 +020098 if (*o == '\0')
99 *o = ' ';
100 }
101
102 return r;
103}
104
105
Dieter Baron1d9dfeb2014-09-28 23:02:54 +0200106zip_string_t *
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100107_zip_read_string(zip_buffer_t *buffer, zip_source_t *src, zip_uint16_t len, bool nulp, zip_error_t *error) {
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200108 zip_uint8_t *raw;
Dieter Baron1d9dfeb2014-09-28 23:02:54 +0200109 zip_string_t *s;
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200110
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100111 if ((raw = _zip_read_data(buffer, src, len, nulp, error)) == NULL)
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200112 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 Klausnerea8ba492014-09-23 16:54:47 +0200120int
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100121_zip_write(zip_t *za, const void *data, zip_uint64_t length) {
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200122 zip_int64_t n;
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100123
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200124 if ((n = zip_source_write(za->src, data, length)) < 0) {
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100125 _zip_error_set_from_source(&za->error, za->src);
126 return -1;
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200127 }
128 if ((zip_uint64_t)n != length) {
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100129 zip_error_set(&za->error, ZIP_ER_WRITE, EINTR);
130 return -1;
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200131 }
Thomas Klausner8eab1a22018-01-15 14:10:11 +0100132
Thomas Klausnerea8ba492014-09-23 16:54:47 +0200133 return 0;
134}