Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 1 | #include <stdlib.h> |
| 2 | #include <string.h> |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 3 | #include "libc.h" |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 4 | |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 5 | char *__strchrnul(const char *, int); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 6 | |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 7 | static void dummy(char *old, char *new) {} |
| 8 | weak_alias(dummy, __env_rm_add); |
| 9 | |
| 10 | int __putenv(char *s, size_t l, char *r) |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 11 | { |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 12 | size_t i=0; |
| 13 | if (__environ) { |
| 14 | for (char **e = __environ; *e; e++, i++) |
| 15 | if (!strncmp(s, *e, l+1)) { |
| 16 | char *tmp = *e; |
| 17 | *e = s; |
| 18 | __env_rm_add(tmp, r); |
| 19 | return 0; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 20 | } |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 21 | } |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 22 | static char **oldenv; |
| 23 | char **newenv; |
| 24 | if (__environ == oldenv) { |
| 25 | newenv = realloc(oldenv, sizeof *newenv * (i+2)); |
| 26 | if (!newenv) goto oom; |
| 27 | } else { |
| 28 | newenv = malloc(sizeof *newenv * (i+2)); |
| 29 | if (!newenv) goto oom; |
| 30 | if (i) memcpy(newenv, __environ, sizeof *newenv * i); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 31 | free(oldenv); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 32 | } |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 33 | newenv[i] = s; |
| 34 | newenv[i+1] = 0; |
| 35 | __environ = oldenv = newenv; |
| 36 | if (r) __env_rm_add(0, r); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 37 | return 0; |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 38 | oom: |
| 39 | free(r); |
| 40 | return -1; |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 41 | } |
| 42 | |
| 43 | int putenv(char *s) |
| 44 | { |
Alexander Monakov | 8e93279 | 2017-09-03 22:12:20 +0300 | [diff] [blame] | 45 | size_t l = __strchrnul(s, '=') - s; |
| 46 | if (!l || !s[l]) return unsetenv(s); |
| 47 | return __putenv(s, l, 0); |
Rich Felker | 0b44a03 | 2011-02-12 00:22:29 -0500 | [diff] [blame] | 48 | } |