Question

由于需要TLPI exercise 6-3，我使用setenv()，unsetenv()和直接修改putenv()变量实现了getenv()和environ。

代码：

// setenv() / unsetenv() impl
// TLPI exercise 6-3

#include <stdio.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>

#define ENV_SEP '='
extern char **environ;

// setenv() impl using putenv() & getenv()
int setenv_impl(const char * name , const char * value , int overwrite ) {
    if(!overwrite && getenv(name)) { // exists & don't overwrite
        return 0;
    } else {
        // construct the new variable
        char *env_var = malloc(strlen(name) + strlen(value) + 2);
        strcpy(env_var, name);
        env_var[strlen(name)] = ENV_SEP;
        strcpy(env_var+(strlen(name)+1), value);

        int result = putenv(env_var);
        if(result==0) {
            return 0;
        } else {
            errno = result;
            return -1;
        }
    }
}

// unsetenv() impl via modifing environ directly,
int unsetenv_impl(const char * name ) {
    char **ep, **sp;
    size_t len;

    len = strlen(name);
    for(ep = environ; *ep != NULL;) {
        if(strncmp(*ep, name, len)==0 && (*ep)[len] == ENV_SEP) {
            // shift all successive elements back 1 step,
            for(sp=ep; *sp != NULL; sp++) {
                *sp = *(sp+1);
            }
        } else {
            ep++;
        }
    }

    return 0;
}

// setenv_impl() test
int setenv_impl_test() {
    char *key = "name";

    setenv_impl(key,"Eric", 1);
    printf("%s\n", getenv(key));

    setenv_impl(key,"Eric2", 0);
    printf("%s\n", getenv(key));

    setenv_impl(key,"Eric3", 1);
    printf("%s\n", getenv(key));

    return 0;
}

// unsetenv_impl() test
int unsetenv_impl_test() {
    char *key = "name";

    setenv_impl(key,"Eric", 1);
    printf("%s\n", getenv(key));

    unsetenv_impl(key);

    char *val = getenv(key);
    printf("%s\n", val==NULL?"NULL":getenv(key));

    return 0;
}

int main(int argc, void *argv[]) {
    // setenv_impl_test();
    unsetenv_impl_test();

    return 0;
}

在我的setevn_impl()中，我使用malloc()为新的环境变量分配内存。

但我不知道如何分配进程的默认环境的内存。

我的问题是：

在我的unsetenv_impl()实现中，是否需要free()释放已删除环境字符串的内存？
如果我不释放它，它会是一个问题，还是不会占用太多记忆因此可以被忽略？

提示

putenv()不会复制字符串，它只是使全局变量environ指向传递给它的字符串。

Answer 1

在您的情况下，如果您不打算频繁设置环境变量导致耗尽内存资源，则没有必要。

但如果您在使用完资源后总是释放资源，那就太好了，无论是文件句柄/内存/互斥锁。通过这样做，您在构建服务器时不会犯这种错误。

预计部分服务器将全天候运行。在这些情况下，任何类型的泄漏都意味着您的服务器最终会耗尽该资源并以某种方式挂起/崩溃。一个简短的实用程序，你的泄漏并不是那么糟糕。任何服务器，任何泄漏都是死亡。帮自己一个忙。自己清理干净。这是一个好习惯

c - unsetenv（）实现，是否有必要释放内存？

1 个答案: