根据TLPI exercise 6-3的要求,我使用putenv()、getenv()并通过直接修改environ变量实现了setenv()和unsetenv()。
代码:
// 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指向传递给它的字符串。
发布于 2015-05-26 10:20:19
在您的情况下,如果您不打算非常频繁地设置环境变量,则没有必要导致内存资源耗尽。
但是,如果您在使用完资源(无论是文件句柄/内存/互斥文件)之后总是释放资源,那就太好了。这样,在构建服务器时,您就不会犯这种错误。
一些服务器预计将运行24x7。在这种情况下,任何类型的泄漏都意味着您的服务器最终将耗尽该资源,并以某种方式挂起/崩溃。一个短的实用程序,一个漏洞也没那么严重。任何服务器,任何泄漏都是死亡。帮你自己一个忙。你自己收拾干净。这是个好习惯
https://stackoverflow.com/questions/30455358
复制相似问题