12.2 实现putenv_r,即putenv的可重入版本。确保你的实现是线程安全的,也是异步信号安全的。
#include <pthread.h>
#include <stdlib.h> // getenv
#include <stdio.h>
#include <string.h>
#include <unistd.h> //environ
extern char **environ;
pthread_mutex_t mutex;
static pthread_once_t init_done = PTHREAD_ONCE_INIT;
static void thread_init(void)
{
pthread_mutexattr_t attr;
pthread_mutexattr_init(&attr);
pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
pthread_mutex_init(&mutex, &attr);
pthread_mutexattr_destroy(&attr);
}
int putenv_r(char *string)
{
int i, len;
char *ptr_key = NULL;
ptr_key = strchr(string, '='); // 指向字符串中首次出现'='的地址
if (NULL == ptr_key)
{
printf("Param illegal Usage: name=value\n");
exit(0);
}
len = ptr_key - string; // 地址的差即name的长度
pthread_once(&init_done, thread_init);
/*使用互斥量保护当前线程对环境变量数组的操作*/
pthread_mutex_lock(&mutex);
for (i = 0; NULL != environ[i]; i++)
{
if (0 == strncmp(string, environ[i], len))
{
environ[i] = string; // 若已存在,覆盖
pthread_mutex_unlock(&mutex);
return 0;
}
}
environ[i] = string;
pthread_mutex_unlock(&mutex);
return 0;
}