关于《APUE》第2版 习题12.2

二熊V

题目：实现putenv_r，即putenv的可重入版本，保证即是线程安全，也是一部信号安全的

1. #include <pthread.h>
   
2. #include <stdlib.h> // getenv
   
3. #include <stdio.h>
   
4. #include <string.h>
   
5. #include <unistd.h> //environ
   
6. 
   
7. extern char **environ;
   
8. 
   
9. pthread_mutex_t mutex;
   
10. 
    
11. static pthread_once_t init_done = PTHREAD_ONCE_INIT;
    
12. 
    
13. static void thread_init(void)
    
14. {
    
15.     pthread_mutexattr_t attr;
    
16.     pthread_mutexattr_init(&attr);
    
17.     pthread_mutexattr_settype(&attr, PTHREAD_MUTEX_RECURSIVE);
    
18.     pthread_mutex_init(&mutex, &attr);
    
19.     pthread_mutexattr_destroy(&attr);
    
20. }
    
21. 
    
22. int putenv_r(char *string)
    
23. {
    
24.     int i, len;
    
25.     char *ptr_key = NULL;
    
26. 
    
27.     ptr_key = strchr(string, '='); // 指向字符串中首次出现'='的地址
    
28.     if (NULL == ptr_key)
    
29.     {
    
30.         printf("Param illegal Usage: name=value\n");
    
31.         exit(0);
    
32.     }
    
33. 
    
34.     len = ptr_key - string; // 地址的差即name的长度
    
35.     pthread_once(&init_done, thread_init);
    
36.     pthread_mutex_lock(&mutex);
    
37.     for (i = 0; NULL != environ[i]; i++)
    
38.     {
    
39.         if (0 == strncmp(string, environ[i], len))
    
40.         {
    
41.             environ[i] = string; // 若已存在，覆盖
    
42.             //printf("%d, i[%d], %s\n", __LINE__, i, environ[i]);
    
43.             pthread_mutex_unlock(&mutex);
    
44.             return 0;
    
45.         }
    
46.     }
    
47. 
    
48.     environ[i+1] = string;
    
49.     //printf("%d, i[%d], %s\n", __LINE__, i+1, environ[i+1]);
    
50.     pthread_mutex_unlock(&mutex);
    
51.     return 0;
    
52. }
    
53. 
    
54. int main(int argc, char const *argv[])
    
55. {
    
56.     const char *STRING = "EDITOR=vim";
    
57.     const char *KEY = "EDITOR";
    
58.     char *ptr = NULL;
    
59.     char *value = NULL;
    
60.     char buf[1024];
    
61. 
    
62.     ptr = malloc(sizeof(char) * strlen(STRING) + 1);
    
63.     if (NULL == ptr)
    
64.     {
    
65.         printf("malloc failed!\n");
    
66.         exit(0);
    
67.     }
    
68.     strcpy(ptr, STRING);
    
69. 
    
70.     putenv_r(ptr);
    
71.     value = getenv(KEY);
    
72.     if (NULL != value)
    
73.     {
    
74.         printf("Set %s to %s\n", buf, KEY);
    
75.     }
    
76.     else
    
77.     {
    
78.         printf("hello world\n");
    
79.     }
    
80. 
    
81.     return 0;
    
82. }

复制代码

本人尝试了如上代码，但每次getenv都返回失败，求大神们帮忙看看，问题处在哪里，谢谢！

cokeboL

直接把 修改 environ 那部分换成putenv多省事

二熊V

已解决，我把putenv_r调用前后的环境变量各打印了一边，发现新增的环境变量根本没加入数组，仔细检查问题出在

1. environ[i+1] = string;

复制代码

i 在循环结束时已经++，导致environ = NULL
看了getenv的源码，内部实现遇到environ == NULL就停止遍历返回NULL了