Bugfix:function setenv()and unsetenv()

The original setenv() and unsetenv() function is invalid,I have rewrited to fix.
This commit is contained in:
Ray Lee 2015-08-16 16:34:35 +08:00
parent 738970c135
commit e72fe468a4

View file

@ -34,21 +34,24 @@
// Handle platform specific includes // Handle platform specific includes
#include "contrib/windows.h" #include "contrib/windows.h"
//There is no setenv()and unsetenv() in windows,but we can use putenv() instead.
int setenv(const char *name, const char *value, int overwrite) int setenv(const char *name, const char *value, int overwrite)
{ {
int exists = GetEnvironmentVariableA(name, NULL, 0); char* env = getenv(name);
if ((exists && overwrite) || (!exists)) { if ((env && overwrite) || (!env)) {
if (!SetEnvironmentVariableA(name, value)) { char* str[32];
// Set errno here correctly strcpy(str,name);
return -1; strcat(str,"=");
strcat(str,value);
return putenv(str);
} }
return 0;
}
// Exists and overwrite is 0.
return -1; return -1;
} }
void unsetenv(const char *name) void unsetenv(const char *name)
{ {
SetEnvironmentVariableA(name, NULL); char* str[32];
strcpy(str,name);
strcat(str,"=");
putenv(str);
} }