基于Getopt读取ini的C实现

freeeyes

由于工作需要，写了一个通过配置文件启动选项的 C接口。
用于自己的使用，在这里记录一下。
以下代码在linux ContOS下测试通过。

1. #include <stdio.h>  
2. #include <stdlib.h>
3. #include <unistd.h>  
4. #include <string.h>
6. #include "ini.h"
7. #include "tcpcopy.h"
9. /* 功能:从指定的配置文件中获取相关设置参数 */
10. /* 编译: gcc -o main main.c ini.c */
11. /* 测试: ./main F:tcpcopy.ini */
13. /*功能: 配置文件默认值 */
14. #define TCPCOPY_CONFIG_INI_FILE "./tcpcopy.ini"
16. char config_ini_file_name[200];
18. xcopy_clt_settings clt_settings;
20. /*功能: 显示从配置文件读取的配置项 */
21. void config_clt_setting_display()
22. {
23.         printf("clt_settings.raw_transfer=%s.\n", clt_settings.raw_transfer);
24.         printf("clt_settings.raw_clt_tf_ip=%s.\n", clt_settings.raw_clt_tf_ip);
25.         printf("clt_settings.replica_num=%d.\n", clt_settings.replica_num);
26.         printf("clt_settings.factor=%d.\n", clt_settings.replica_num);
27.         printf("clt_settings.max_rss=%d.\n", clt_settings.replica_num);
28.         printf("clt_settings.par_connections=%d.\n", clt_settings.replica_num);
29.         printf("clt_settings.log_path=%s.\n", clt_settings.log_path);
30.         printf("clt_settings.mtu=%d.\n", clt_settings.mtu);
31.         printf("clt_settings.mss=%d.\n", clt_settings.mss);
32. }
34. /*功能：清理xcopy_clt_settings的内存 */
35. void clear_clt_setting()
36. {
37.         free(clt_settings.raw_transfer);
38.         free(clt_settings.raw_clt_tf_ip);
39.         free(clt_settings.log_path);
40. }
42. /*功能: ini 文件回调，读取Ini详细信息 */
43. static int ini_read_handler(void* user, const char* section, const char* name,
44.                    const char* value)
45. {
46.     #define MATCH(s, n) strcmp(section, s) == 0 && strcmp(name, n) == 0
47.     if (MATCH("config", "x")) {
48.         clt_settings.raw_transfer = strdup(value);
49.     } else if (MATCH("config", "c")) {
50.         clt_settings.raw_clt_tf_ip = strdup(value);
51.     } else if (MATCH("config", "n")) {
52.         clt_settings.replica_num = atoi(value);
53.     } else if (MATCH("config", "f")) {
54.         clt_settings.factor = atoi(value);        
55.     } else if (MATCH("config", "m")) {
56.         clt_settings.max_rss = atoi(value);   
57.     } else if (MATCH("config", "C")) {
58.         clt_settings.par_connections = atoi(value);
59.     } else if (MATCH("config", "l")) {
60.         clt_settings.log_path = strdup(value);
61.     } else if (MATCH("config", "M")) {
62.         clt_settings.mtu = atoi(value);
63.     } else if (MATCH("config", "S")) {
64.         clt_settings.mss = atoi(value);        
65.     } else {
66.         return 0;  /* unknown section/name, error */
67.     }
68.     return 1;
69. }
71. /*功能: 从路径获取配置文件名称 */
72. void read_opt_file_name(int argc, char* argv[])
73. {
74.   int ch;   
75.   config_ini_file_name[0] = '\0';
76.   
77.   while ((ch = getopt(argc,argv,"F:"))!=-1)  
78.   {  
79.           switch(ch)  
80.     {  
81.       case 'f':  
82.               printf("option f:'%s'\n", optarg);
83.               sprintf(config_ini_file_name, "%s", optarg);
84.               break;  
85.       default:  
86.               printf("other option :%c\n",ch);  
87.     }  
88.   }         
89. }
91. int main(int argc, char* argv[])
92. {   
93.     read_opt_file_name(argc, argv);
94.    
95.     if(strlen(config_ini_file_name) == 0)
96.     {
97.             /* 没有发现设置配置文件 设置一个默认值 */
98.             sprintf(config_ini_file_name, "%s", TCPCOPY_CONFIG_INI_FILE);
99.     }
101.                 /* 读取配置文件 */
102.     if (ini_parse(config_ini_file_name, ini_read_handler, NULL) < 0) {
103.         printf("Can't load (%s)\n", config_ini_file_name);
104.         return 1;
105.     }
106.    
107.     config_clt_setting_display();
108.    
109.     clear_clt_setting();
111.     return 0;
112. }

复制代码