1
0

setenv.c 899 B

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. #include <stdlib.h>
  2. #include <string.h>
  3. #include <errno.h>
  4. void __env_rm_add(char *old, char *new)
  5. {
  6. static char **env_alloced;
  7. static size_t env_alloced_n;
  8. for (size_t i=0; i < env_alloced_n; i++)
  9. if (env_alloced[i] == old) {
  10. env_alloced[i] = new;
  11. free(old);
  12. return;
  13. } else if (!env_alloced[i] && new) {
  14. env_alloced[i] = new;
  15. new = 0;
  16. }
  17. if (!new) return;
  18. char **t = realloc(env_alloced, sizeof *t * (env_alloced_n+1));
  19. if (!t) return;
  20. (env_alloced = t)[env_alloced_n++] = new;
  21. }
  22. int setenv(const char *var, const char *value, int overwrite)
  23. {
  24. char *s;
  25. size_t l1, l2;
  26. if (!var || !(l1 = __strchrnul(var, '=') - var) || var[l1]) {
  27. errno = EINVAL;
  28. return -1;
  29. }
  30. if (!overwrite && getenv(var)) return 0;
  31. l2 = strlen(value);
  32. s = malloc(l1+l2+2);
  33. if (!s) return -1;
  34. memcpy(s, var, l1);
  35. s[l1] = '=';
  36. memcpy(s+l1+1, value, l2+1);
  37. return __putenv(s, l1, s);
  38. }