1 /*
2   zip_set_name.c -- rename helper function
3   Copyright (C) 1999-2019 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 
35 #include <stdlib.h>
36 #include <string.h>
37 
38 #include "zipint.h"
39 
40 
41 int
_zip_set_name(zip_t * za,zip_uint64_t idx,const char * name,zip_flags_t flags)42 _zip_set_name(zip_t *za, zip_uint64_t idx, const char *name, zip_flags_t flags) {
43     zip_entry_t *e;
44     zip_string_t *str;
45     bool same_as_orig;
46     zip_int64_t i;
47     const zip_uint8_t *old_name, *new_name;
48     zip_string_t *old_str;
49 
50     if (idx >= za->nentry) {
51 	zip_error_set(&za->error, ZIP_ER_INVAL, 0);
52 	return -1;
53     }
54 
55     if (ZIP_IS_RDONLY(za)) {
56 	zip_error_set(&za->error, ZIP_ER_RDONLY, 0);
57 	return -1;
58     }
59 
60     if (name && name[0] != '\0') {
61 	/* TODO: check for string too long */
62 	if ((str = _zip_string_new((const zip_uint8_t *)name, (zip_uint16_t)strlen(name), flags, &za->error)) == NULL)
63 	    return -1;
64 	if ((flags & ZIP_FL_ENCODING_ALL) == ZIP_FL_ENC_GUESS && _zip_guess_encoding(str, ZIP_ENCODING_UNKNOWN) == ZIP_ENCODING_UTF8_GUESSED)
65 	    str->encoding = ZIP_ENCODING_UTF8_KNOWN;
66     }
67     else
68 	str = NULL;
69 
70     /* TODO: encoding flags needed for CP437? */
71     if ((i = _zip_name_locate(za, name, 0, NULL)) >= 0 && (zip_uint64_t)i != idx) {
72 	_zip_string_free(str);
73 	zip_error_set(&za->error, ZIP_ER_EXISTS, 0);
74 	return -1;
75     }
76 
77     /* no effective name change */
78     if (i >= 0 && (zip_uint64_t)i == idx) {
79 	_zip_string_free(str);
80 	return 0;
81     }
82 
83     e = za->entry + idx;
84 
85     if (e->orig)
86 	same_as_orig = _zip_string_equal(e->orig->filename, str);
87     else
88 	same_as_orig = false;
89 
90     if (!same_as_orig && e->changes == NULL) {
91 	if ((e->changes = _zip_dirent_clone(e->orig)) == NULL) {
92 	    zip_error_set(&za->error, ZIP_ER_MEMORY, 0);
93 	    _zip_string_free(str);
94 	    return -1;
95 	}
96     }
97 
98     if ((new_name = _zip_string_get(same_as_orig ? e->orig->filename : str, NULL, 0, &za->error)) == NULL) {
99 	_zip_string_free(str);
100 	return -1;
101     }
102 
103     if (e->changes) {
104 	old_str = e->changes->filename;
105     }
106     else if (e->orig) {
107 	old_str = e->orig->filename;
108     }
109     else {
110 	old_str = NULL;
111     }
112 
113     if (old_str) {
114 	if ((old_name = _zip_string_get(old_str, NULL, 0, &za->error)) == NULL) {
115 	    _zip_string_free(str);
116 	    return -1;
117 	}
118     }
119     else {
120 	old_name = NULL;
121     }
122 
123     if (_zip_hash_add(za->names, new_name, idx, 0, &za->error) == false) {
124 	_zip_string_free(str);
125 	return -1;
126     }
127     if (old_name) {
128 	_zip_hash_delete(za->names, old_name, NULL);
129     }
130 
131     if (same_as_orig) {
132 	if (e->changes) {
133 	    if (e->changes->changed & ZIP_DIRENT_FILENAME) {
134 		_zip_string_free(e->changes->filename);
135 		e->changes->changed &= ~ZIP_DIRENT_FILENAME;
136 		if (e->changes->changed == 0) {
137 		    _zip_dirent_free(e->changes);
138 		    e->changes = NULL;
139 		}
140 		else {
141 		    /* TODO: what if not cloned? can that happen? */
142 		    e->changes->filename = e->orig->filename;
143 		}
144 	    }
145 	}
146 	_zip_string_free(str);
147     }
148     else {
149 	if (e->changes->changed & ZIP_DIRENT_FILENAME) {
150 	    _zip_string_free(e->changes->filename);
151 	}
152 	e->changes->changed |= ZIP_DIRENT_FILENAME;
153 	e->changes->filename = str;
154     }
155 
156     return 0;
157 }
158