| File: | util/envlist.c |
| Location: | line 48, column 3 |
| Description: | Use of memory after it is freed |
| 1 | #include "qemu-common.h" | |||||
| 2 | #include "qemu/queue.h" | |||||
| 3 | #include "qemu/envlist.h" | |||||
| 4 | ||||||
| 5 | struct envlist_entry { | |||||
| 6 | const char *ev_var; /* actual env value */ | |||||
| 7 | QLIST_ENTRY(envlist_entry)struct { struct envlist_entry *le_next; struct envlist_entry * *le_prev; } ev_link; | |||||
| 8 | }; | |||||
| 9 | ||||||
| 10 | struct envlist { | |||||
| 11 | QLIST_HEAD(, envlist_entry)struct { struct envlist_entry *lh_first; } el_entries; /* actual entries */ | |||||
| 12 | size_t el_count; /* number of entries */ | |||||
| 13 | }; | |||||
| 14 | ||||||
| 15 | static int envlist_parse(envlist_t *envlist, | |||||
| 16 | const char *env, int (*)(envlist_t *, const char *)); | |||||
| 17 | ||||||
| 18 | /* | |||||
| 19 | * Allocates new envlist and returns pointer to that or | |||||
| 20 | * NULL in case of error. | |||||
| 21 | */ | |||||
| 22 | envlist_t * | |||||
| 23 | envlist_create(void) | |||||
| 24 | { | |||||
| 25 | envlist_t *envlist; | |||||
| 26 | ||||||
| 27 | if ((envlist = malloc(sizeof (*envlist))) == NULL((void*)0)) | |||||
| 28 | return (NULL((void*)0)); | |||||
| 29 | ||||||
| 30 | QLIST_INIT(&envlist->el_entries)do { (&envlist->el_entries)->lh_first = ((void*)0); } while ( 0); | |||||
| 31 | envlist->el_count = 0; | |||||
| 32 | ||||||
| 33 | return (envlist); | |||||
| 34 | } | |||||
| 35 | ||||||
| 36 | /* | |||||
| 37 | * Releases given envlist and its entries. | |||||
| 38 | */ | |||||
| 39 | void | |||||
| 40 | envlist_free(envlist_t *envlist) | |||||
| 41 | { | |||||
| 42 | struct envlist_entry *entry; | |||||
| 43 | ||||||
| 44 | assert(envlist != NULL)((envlist != ((void*)0)) ? (void) (0) : __assert_fail ("envlist != ((void*)0)" , "/home/stefan/src/qemu/qemu.org/qemu/util/envlist.c", 44, __PRETTY_FUNCTION__ )); | |||||
| 45 | ||||||
| 46 | while (envlist->el_entries.lh_first != NULL((void*)0)) { | |||||
| ||||||
| 47 | entry = envlist->el_entries.lh_first; | |||||
| 48 | QLIST_REMOVE(entry, ev_link)do { if ((entry)->ev_link.le_next != ((void*)0)) (entry)-> ev_link.le_next->ev_link.le_prev = (entry)->ev_link.le_prev ; *(entry)->ev_link.le_prev = (entry)->ev_link.le_next; } while ( 0); | |||||
| ||||||
| 49 | ||||||
| 50 | free((char *)entry->ev_var); | |||||
| 51 | free(entry); | |||||
| 52 | } | |||||
| 53 | free(envlist); | |||||
| 54 | } | |||||
| 55 | ||||||
| 56 | /* | |||||
| 57 | * Parses comma separated list of set/modify environment | |||||
| 58 | * variable entries and updates given enlist accordingly. | |||||
| 59 | * | |||||
| 60 | * For example: | |||||
| 61 | * envlist_parse(el, "HOME=foo,SHELL=/bin/sh"); | |||||
| 62 | * | |||||
| 63 | * inserts/sets environment variables HOME and SHELL. | |||||
| 64 | * | |||||
| 65 | * Returns 0 on success, errno otherwise. | |||||
| 66 | */ | |||||
| 67 | int | |||||
| 68 | envlist_parse_set(envlist_t *envlist, const char *env) | |||||
| 69 | { | |||||
| 70 | return (envlist_parse(envlist, env, &envlist_setenv)); | |||||
| 71 | } | |||||
| 72 | ||||||
| 73 | /* | |||||
| 74 | * Parses comma separated list of unset environment variable | |||||
| 75 | * entries and removes given variables from given envlist. | |||||
| 76 | * | |||||
| 77 | * Returns 0 on success, errno otherwise. | |||||
| 78 | */ | |||||
| 79 | int | |||||
| 80 | envlist_parse_unset(envlist_t *envlist, const char *env) | |||||
| 81 | { | |||||
| 82 | return (envlist_parse(envlist, env, &envlist_unsetenv)); | |||||
| 83 | } | |||||
| 84 | ||||||
| 85 | /* | |||||
| 86 | * Parses comma separated list of set, modify or unset entries | |||||
| 87 | * and calls given callback for each entry. | |||||
| 88 | * | |||||
| 89 | * Returns 0 in case of success, errno otherwise. | |||||
| 90 | */ | |||||
| 91 | static int | |||||
| 92 | envlist_parse(envlist_t *envlist, const char *env, | |||||
| 93 | int (*callback)(envlist_t *, const char *)) | |||||
| 94 | { | |||||
| 95 | char *tmpenv, *envvar; | |||||
| 96 | char *envsave = NULL((void*)0); | |||||
| 97 | ||||||
| 98 | assert(callback != NULL)((callback != ((void*)0)) ? (void) (0) : __assert_fail ("callback != ((void*)0)" , "/home/stefan/src/qemu/qemu.org/qemu/util/envlist.c", 98, __PRETTY_FUNCTION__ )); | |||||
| 99 | ||||||
| 100 | if ((envlist == NULL((void*)0)) || (env == NULL((void*)0))) | |||||
| 101 | return (EINVAL22); | |||||
| 102 | ||||||
| 103 | /* | |||||
| 104 | * We need to make temporary copy of the env string | |||||
| 105 | * as strtok_r(3) modifies it while it tokenizes. | |||||
| 106 | */ | |||||
| 107 | if ((tmpenv = strdup(env)) == NULL((void*)0)) | |||||
| 108 | return (errno(*__errno_location ())); | |||||
| 109 | ||||||
| 110 | envvar = strtok_r(tmpenv, ",", &envsave); | |||||
| 111 | while (envvar != NULL((void*)0)) { | |||||
| 112 | if ((*callback)(envlist, envvar) != 0) { | |||||
| 113 | free(tmpenv); | |||||
| 114 | return (errno(*__errno_location ())); | |||||
| 115 | } | |||||
| 116 | envvar = strtok_r(NULL((void*)0), ",", &envsave); | |||||
| 117 | } | |||||
| 118 | ||||||
| 119 | free(tmpenv); | |||||
| 120 | return (0); | |||||
| 121 | } | |||||
| 122 | ||||||
| 123 | /* | |||||
| 124 | * Sets environment value to envlist in similar manner | |||||
| 125 | * than putenv(3). | |||||
| 126 | * | |||||
| 127 | * Returns 0 in success, errno otherwise. | |||||
| 128 | */ | |||||
| 129 | int | |||||
| 130 | envlist_setenv(envlist_t *envlist, const char *env) | |||||
| 131 | { | |||||
| 132 | struct envlist_entry *entry = NULL((void*)0); | |||||
| 133 | const char *eq_sign; | |||||
| 134 | size_t envname_len; | |||||
| 135 | ||||||
| 136 | if ((envlist == NULL((void*)0)) || (env == NULL((void*)0))) | |||||
| 137 | return (EINVAL22); | |||||
| 138 | ||||||
| 139 | /* find out first equals sign in given env */ | |||||
| 140 | if ((eq_sign = strchr(env, '=')) == NULL((void*)0)) | |||||
| 141 | return (EINVAL22); | |||||
| 142 | envname_len = eq_sign - env + 1; | |||||
| 143 | ||||||
| 144 | /* | |||||
| 145 | * If there already exists variable with given name | |||||
| 146 | * we remove and release it before allocating a whole | |||||
| 147 | * new entry. | |||||
| 148 | */ | |||||
| 149 | for (entry = envlist->el_entries.lh_first; entry != NULL((void*)0); | |||||
| 150 | entry = entry->ev_link.le_next) { | |||||
| 151 | if (strncmp(entry->ev_var, env, envname_len) == 0) | |||||
| 152 | break; | |||||
| 153 | } | |||||
| 154 | ||||||
| 155 | if (entry != NULL((void*)0)) { | |||||
| 156 | QLIST_REMOVE(entry, ev_link)do { if ((entry)->ev_link.le_next != ((void*)0)) (entry)-> ev_link.le_next->ev_link.le_prev = (entry)->ev_link.le_prev ; *(entry)->ev_link.le_prev = (entry)->ev_link.le_next; } while ( 0); | |||||
| 157 | free((char *)entry->ev_var); | |||||
| 158 | free(entry); | |||||
| 159 | } else { | |||||
| 160 | envlist->el_count++; | |||||
| 161 | } | |||||
| 162 | ||||||
| 163 | if ((entry = malloc(sizeof (*entry))) == NULL((void*)0)) | |||||
| 164 | return (errno(*__errno_location ())); | |||||
| 165 | if ((entry->ev_var = strdup(env)) == NULL((void*)0)) { | |||||
| 166 | free(entry); | |||||
| 167 | return (errno(*__errno_location ())); | |||||
| 168 | } | |||||
| 169 | QLIST_INSERT_HEAD(&envlist->el_entries, entry, ev_link)do { if (((entry)->ev_link.le_next = (&envlist->el_entries )->lh_first) != ((void*)0)) (&envlist->el_entries)-> lh_first->ev_link.le_prev = &(entry)->ev_link.le_next ; (&envlist->el_entries)->lh_first = (entry); (entry )->ev_link.le_prev = &(&envlist->el_entries)-> lh_first; } while ( 0); | |||||
| 170 | ||||||
| 171 | return (0); | |||||
| 172 | } | |||||
| 173 | ||||||
| 174 | /* | |||||
| 175 | * Removes given env value from envlist in similar manner | |||||
| 176 | * than unsetenv(3). Returns 0 in success, errno otherwise. | |||||
| 177 | */ | |||||
| 178 | int | |||||
| 179 | envlist_unsetenv(envlist_t *envlist, const char *env) | |||||
| 180 | { | |||||
| 181 | struct envlist_entry *entry; | |||||
| 182 | size_t envname_len; | |||||
| 183 | ||||||
| 184 | if ((envlist == NULL((void*)0)) || (env == NULL((void*)0))) | |||||
| 185 | return (EINVAL22); | |||||
| 186 | ||||||
| 187 | /* env is not allowed to contain '=' */ | |||||
| 188 | if (strchr(env, '=') != NULL((void*)0)) | |||||
| 189 | return (EINVAL22); | |||||
| 190 | ||||||
| 191 | /* | |||||
| 192 | * Find out the requested entry and remove | |||||
| 193 | * it from the list. | |||||
| 194 | */ | |||||
| 195 | envname_len = strlen(env); | |||||
| 196 | for (entry = envlist->el_entries.lh_first; entry != NULL((void*)0); | |||||
| 197 | entry = entry->ev_link.le_next) { | |||||
| 198 | if (strncmp(entry->ev_var, env, envname_len) == 0) | |||||
| 199 | break; | |||||
| 200 | } | |||||
| 201 | if (entry != NULL((void*)0)) { | |||||
| 202 | QLIST_REMOVE(entry, ev_link)do { if ((entry)->ev_link.le_next != ((void*)0)) (entry)-> ev_link.le_next->ev_link.le_prev = (entry)->ev_link.le_prev ; *(entry)->ev_link.le_prev = (entry)->ev_link.le_next; } while ( 0); | |||||
| 203 | free((char *)entry->ev_var); | |||||
| 204 | free(entry); | |||||
| 205 | ||||||
| 206 | envlist->el_count--; | |||||
| 207 | } | |||||
| 208 | return (0); | |||||
| 209 | } | |||||
| 210 | ||||||
| 211 | /* | |||||
| 212 | * Returns given envlist as array of strings (in same form that | |||||
| 213 | * global variable environ is). Caller must free returned memory | |||||
| 214 | * by calling free(3) for each element and for the array. Returned | |||||
| 215 | * array and given envlist are not related (no common references). | |||||
| 216 | * | |||||
| 217 | * If caller provides count pointer, number of items in array is | |||||
| 218 | * stored there. In case of error, NULL is returned and no memory | |||||
| 219 | * is allocated. | |||||
| 220 | */ | |||||
| 221 | char ** | |||||
| 222 | envlist_to_environ(const envlist_t *envlist, size_t *count) | |||||
| 223 | { | |||||
| 224 | struct envlist_entry *entry; | |||||
| 225 | char **env, **penv; | |||||
| 226 | ||||||
| 227 | penv = env = malloc((envlist->el_count + 1) * sizeof (char *)); | |||||
| 228 | if (env == NULL((void*)0)) | |||||
| 229 | return (NULL((void*)0)); | |||||
| 230 | ||||||
| 231 | for (entry = envlist->el_entries.lh_first; entry != NULL((void*)0); | |||||
| 232 | entry = entry->ev_link.le_next) { | |||||
| 233 | *(penv++) = strdup(entry->ev_var); | |||||
| 234 | } | |||||
| 235 | *penv = NULL((void*)0); /* NULL terminate the list */ | |||||
| 236 | ||||||
| 237 | if (count != NULL((void*)0)) | |||||
| 238 | *count = envlist->el_count; | |||||
| 239 | ||||||
| 240 | return (env); | |||||
| 241 | } |