xref: /PHP-7.0/ext/zip/lib/zip_extra_field.c (revision 0d57c06b)
1 /*
2   zip_extra_field.c -- manipulate extra fields
3   Copyright (C) 2012-2015 Dieter Baron and Thomas Klausner
4 
5   This file is part of libzip, a library to manipulate ZIP archives.
6   The authors can be contacted at <libzip@nih.at>
7 
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.
20 
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 
34 #include <stdlib.h>
35 #include <string.h>
36 
37 #include "zipint.h"
38 
39 
40 zip_extra_field_t *
_zip_ef_clone(const zip_extra_field_t * ef,zip_error_t * error)41 _zip_ef_clone(const zip_extra_field_t *ef, zip_error_t *error)
42 {
43     zip_extra_field_t *head, *prev, *def;
44 
45     head = prev = NULL;
46 
47     while (ef) {
48         if ((def=_zip_ef_new(ef->id, ef->size, ef->data, ef->flags)) == NULL) {
49             zip_error_set(error, ZIP_ER_MEMORY, 0);
50             _zip_ef_free(head);
51             return NULL;
52         }
53 
54         if (head == NULL)
55             head = def;
56         if (prev)
57             prev->next = def;
58         prev = def;
59 
60 	ef = ef->next;
61     }
62 
63     return head;
64 }
65 
66 
67 zip_extra_field_t *
_zip_ef_delete_by_id(zip_extra_field_t * ef,zip_uint16_t id,zip_uint16_t id_idx,zip_flags_t flags)68 _zip_ef_delete_by_id(zip_extra_field_t *ef, zip_uint16_t id, zip_uint16_t id_idx, zip_flags_t flags)
69 {
70     zip_extra_field_t *head, *prev;
71     int i;
72 
73     i = 0;
74     head = ef;
75     prev = NULL;
76     for (; ef; ef=(prev ? prev->next : head)) {
77 	if ((ef->flags & flags & ZIP_EF_BOTH) && ((ef->id == id) || (id == ZIP_EXTRA_FIELD_ALL))) {
78 	    if (id_idx == ZIP_EXTRA_FIELD_ALL || i == id_idx) {
79 		ef->flags &= ~(flags & ZIP_EF_BOTH);
80 		if ((ef->flags & ZIP_EF_BOTH) == 0) {
81 		    if (prev)
82 			prev->next = ef->next;
83 		    else
84 			head = ef->next;
85 		    ef->next = NULL;
86 		    _zip_ef_free(ef);
87 
88 		    if (id_idx == ZIP_EXTRA_FIELD_ALL)
89 			continue;
90 		}
91 	    }
92 
93 	    i++;
94 	    if (i > id_idx)
95 		break;
96 	}
97 	prev = ef;
98     }
99 
100     return head;
101 }
102 
103 
104 
105 void
_zip_ef_free(zip_extra_field_t * ef)106 _zip_ef_free(zip_extra_field_t *ef)
107 {
108     zip_extra_field_t *ef2;
109 
110     while (ef) {
111 	ef2 = ef->next;
112 	free(ef->data);
113 	free(ef);
114 	ef = ef2;
115     }
116 }
117 
118 
119 const zip_uint8_t *
_zip_ef_get_by_id(const zip_extra_field_t * ef,zip_uint16_t * lenp,zip_uint16_t id,zip_uint16_t id_idx,zip_flags_t flags,zip_error_t * error)120 _zip_ef_get_by_id(const zip_extra_field_t *ef, zip_uint16_t *lenp, zip_uint16_t id, zip_uint16_t id_idx, zip_flags_t flags, zip_error_t *error)
121 {
122     static const zip_uint8_t empty[1] = { '\0' };
123 
124     int i;
125 
126     i = 0;
127     for (; ef; ef=ef->next) {
128 	if (ef->id == id && (ef->flags & flags & ZIP_EF_BOTH)) {
129 	    if (i < id_idx) {
130 		i++;
131 		continue;
132 	    }
133 
134 	    if (lenp)
135 		*lenp = ef->size;
136 	    if (ef->size > 0)
137 		return ef->data;
138 	    else
139 		return empty;
140 	}
141     }
142 
143     zip_error_set(error, ZIP_ER_NOENT, 0);
144     return NULL;
145 }
146 
147 
148 zip_extra_field_t *
_zip_ef_merge(zip_extra_field_t * to,zip_extra_field_t * from)149 _zip_ef_merge(zip_extra_field_t *to, zip_extra_field_t *from)
150 {
151     zip_extra_field_t *ef2, *tt, *tail;
152     int duplicate;
153 
154     if (to == NULL)
155 	return from;
156 
157     for (tail=to; tail->next; tail=tail->next)
158 	;
159 
160     for (; from; from=ef2) {
161 	ef2 = from->next;
162 
163 	duplicate = 0;
164 	for (tt=to; tt; tt=tt->next) {
165 	    if (tt->id == from->id && tt->size == from->size && memcmp(tt->data, from->data, tt->size) == 0) {
166 		tt->flags |= (from->flags & ZIP_EF_BOTH);
167 		duplicate = 1;
168 		break;
169 	    }
170 	}
171 
172 	from->next = NULL;
173 	if (duplicate)
174 	    _zip_ef_free(from);
175 	else
176 	    tail = tail->next = from;
177     }
178 
179     return to;
180 }
181 
182 
183 zip_extra_field_t *
_zip_ef_new(zip_uint16_t id,zip_uint16_t size,const zip_uint8_t * data,zip_flags_t flags)184 _zip_ef_new(zip_uint16_t id, zip_uint16_t size, const zip_uint8_t *data, zip_flags_t flags)
185 {
186     zip_extra_field_t *ef;
187 
188     if ((ef=(zip_extra_field_t *)malloc(sizeof(*ef))) == NULL)
189 	return NULL;
190 
191     ef->next = NULL;
192     ef->flags = flags;
193     ef->id = id;
194     ef->size = size;
195     if (size > 0) {
196 	if ((ef->data=(zip_uint8_t *)_zip_memdup(data, size, NULL)) == NULL) {
197 	    free(ef);
198 	    return NULL;
199 	}
200     }
201     else
202 	ef->data = NULL;
203 
204     return ef;
205 }
206 
207 
208 bool
_zip_ef_parse(const zip_uint8_t * data,zip_uint16_t len,zip_flags_t flags,zip_extra_field_t ** ef_head_p,zip_error_t * error)209 _zip_ef_parse(const zip_uint8_t *data, zip_uint16_t len, zip_flags_t flags, zip_extra_field_t **ef_head_p, zip_error_t *error)
210 {
211     zip_buffer_t *buffer;
212     zip_extra_field_t *ef, *ef2, *ef_head;
213 
214     if ((buffer = _zip_buffer_new((zip_uint8_t *)data, len)) == NULL) {
215         zip_error_set(error, ZIP_ER_MEMORY, 0);
216         return false;
217     }
218 
219     ef_head = ef = NULL;
220 
221     while (_zip_buffer_ok(buffer) && _zip_buffer_left(buffer) >= 4) {
222         zip_uint16_t fid, flen;
223         zip_uint8_t *ef_data;
224 
225         fid = _zip_buffer_get_16(buffer);
226 	flen = _zip_buffer_get_16(buffer);
227         ef_data = _zip_buffer_get(buffer, flen);
228 
229         if (ef_data == NULL) {
230 	    zip_error_set(error, ZIP_ER_INCONS, 0);
231             _zip_buffer_free(buffer);
232 	    _zip_ef_free(ef_head);
233 	    return false;
234         }
235 
236 	if ((ef2=_zip_ef_new(fid, flen, ef_data, flags)) == NULL) {
237 	    zip_error_set(error, ZIP_ER_MEMORY, 0);
238             _zip_buffer_free(buffer);
239 	    _zip_ef_free(ef_head);
240 	    return false;
241 	}
242 
243 	if (ef_head) {
244 	    ef->next = ef2;
245 	    ef = ef2;
246 	}
247 	else
248 	    ef_head = ef = ef2;
249     }
250 
251     if (!_zip_buffer_eof(buffer)) {
252 	/* Android APK files align stored file data with padding in extra fields; ignore. */
253 	/* see https://android.googlesource.com/platform/build/+/master/tools/zipalign/ZipAlign.cpp */
254 	size_t glen = _zip_buffer_left(buffer);
255 	zip_uint8_t *garbage;
256 	garbage = _zip_buffer_get(buffer, glen);
257 	if (glen >= 4 || garbage == NULL || memcmp(garbage, "\0\0\0", glen) != 0) {
258 	    zip_error_set(error, ZIP_ER_INCONS, 0);
259 	    _zip_buffer_free(buffer);
260 	    _zip_ef_free(ef_head);
261 	    return false;
262 	}
263     }
264 
265     _zip_buffer_free(buffer);
266 
267     if (ef_head_p) {
268 	*ef_head_p = ef_head;
269     }
270     else {
271         _zip_ef_free(ef_head);
272     }
273 
274     return true;
275 }
276 
277 
278 zip_extra_field_t *
_zip_ef_remove_internal(zip_extra_field_t * ef)279 _zip_ef_remove_internal(zip_extra_field_t *ef)
280 {
281     zip_extra_field_t *ef_head;
282     zip_extra_field_t *prev, *next;
283 
284     ef_head = ef;
285     prev = NULL;
286 
287     while (ef) {
288         if (ZIP_EF_IS_INTERNAL(ef->id)) {
289             next = ef->next;
290             if (ef_head == ef)
291                 ef_head = next;
292             ef->next = NULL;
293             _zip_ef_free(ef);
294             if (prev)
295                 prev->next = next;
296             ef = next;
297         }
298         else {
299             prev = ef;
300             ef = ef->next;
301         }
302     }
303 
304     return ef_head;
305 }
306 
307 
308 zip_uint16_t
_zip_ef_size(const zip_extra_field_t * ef,zip_flags_t flags)309 _zip_ef_size(const zip_extra_field_t *ef, zip_flags_t flags)
310 {
311     zip_uint16_t size;
312 
313     size = 0;
314     for (; ef; ef=ef->next) {
315 	if (ef->flags & flags & ZIP_EF_BOTH)
316 	    size = (zip_uint16_t)(size+4+ef->size);
317     }
318 
319     return size;
320 }
321 
322 
323 int
_zip_ef_write(zip_t * za,const zip_extra_field_t * ef,zip_flags_t flags)324 _zip_ef_write(zip_t *za, const zip_extra_field_t *ef, zip_flags_t flags)
325 {
326     zip_uint8_t b[4];
327     zip_buffer_t *buffer = _zip_buffer_new(b, sizeof(b));
328 
329     if (buffer == NULL) {
330 	return -1;
331     }
332 
333     for (; ef; ef=ef->next) {
334 	if (ef->flags & flags & ZIP_EF_BOTH) {
335             _zip_buffer_set_offset(buffer, 0);
336             _zip_buffer_put_16(buffer, ef->id);
337 	    _zip_buffer_put_16(buffer, ef->size);
338             if (!_zip_buffer_ok(buffer)) {
339                 zip_error_set(&za->error, ZIP_ER_INTERNAL, 0);
340                 _zip_buffer_free(buffer);
341                 return -1;
342             }
343 	    if (_zip_write(za, b, 4) < 0) {
344                 _zip_buffer_free(buffer);
345                 return -1;
346 	    }
347 	    if (ef->size > 0) {
348 		if (_zip_write(za, ef->data, ef->size) < 0) {
349                     _zip_buffer_free(buffer);
350                     return -1;
351 		}
352 	    }
353 	}
354     }
355 
356     _zip_buffer_free(buffer);
357     return 0;
358 }
359 
360 
361 int
_zip_read_local_ef(zip_t * za,zip_uint64_t idx)362 _zip_read_local_ef(zip_t *za, zip_uint64_t idx)
363 {
364     zip_entry_t *e;
365     unsigned char b[4];
366     zip_buffer_t *buffer;
367     zip_uint16_t fname_len, ef_len;
368 
369     if (idx >= za->nentry) {
370 	zip_error_set(&za->error, ZIP_ER_INVAL, 0);
371 	return -1;
372     }
373 
374     e = za->entry+idx;
375 
376     if (e->orig == NULL || e->orig->local_extra_fields_read)
377 	return 0;
378 
379     if (e->orig->offset + 26 > ZIP_INT64_MAX) {
380 	zip_error_set(&za->error, ZIP_ER_SEEK, EFBIG);
381 	return -1;
382     }
383 
384     if (zip_source_seek(za->src, (zip_int64_t)(e->orig->offset + 26), SEEK_SET) < 0) {
385 	_zip_error_set_from_source(&za->error, za->src);
386 	return -1;
387     }
388 
389     if ((buffer = _zip_buffer_new_from_source(za->src, sizeof(b), b, &za->error)) == NULL) {
390         return -1;
391     }
392 
393     fname_len = _zip_buffer_get_16(buffer);
394     ef_len = _zip_buffer_get_16(buffer);
395 
396     if (!_zip_buffer_eof(buffer)) {
397         _zip_buffer_free(buffer);
398         zip_error_set(&za->error, ZIP_ER_INTERNAL, 0);
399         return -1;
400     }
401 
402     _zip_buffer_free(buffer);
403 
404     if (ef_len > 0) {
405 	zip_extra_field_t *ef;
406 	zip_uint8_t *ef_raw;
407 
408 	if (zip_source_seek(za->src, fname_len, SEEK_CUR) < 0) {
409 	    zip_error_set(&za->error, ZIP_ER_SEEK, errno);
410 	    return -1;
411 	}
412 
413 	ef_raw = _zip_read_data(NULL, za->src, ef_len, 0, &za->error);
414 
415 	if (ef_raw == NULL)
416 	    return -1;
417 
418 	if (!_zip_ef_parse(ef_raw, ef_len, ZIP_EF_LOCAL, &ef, &za->error)) {
419 	    free(ef_raw);
420 	    return -1;
421 	}
422 	free(ef_raw);
423 
424 	if (ef) {
425 	    ef = _zip_ef_remove_internal(ef);
426 	    e->orig->extra_fields = _zip_ef_merge(e->orig->extra_fields, ef);
427 	}
428     }
429 
430     e->orig->local_extra_fields_read = 1;
431 
432     if (e->changes && e->changes->local_extra_fields_read == 0) {
433 	e->changes->extra_fields = e->orig->extra_fields;
434 	e->changes->local_extra_fields_read = 1;
435     }
436 
437     return 0;
438 }
439