wifidog源码分析 - 初始化阶段

点击(此处)折叠或打开

1. void get_clients_from_parent(void) {
   
2.     int sock;
   
3.     struct sockaddr_un sa_un;
   
4.     s_config * config = NULL;
   
5.     char linebuffer[MAX_BUF];
   
6.     int len = 0;
   
7.     char *running1 = NULL;
   
8.     char *running2 = NULL;
   
9.     char *token1 = NULL;
   
10.     char *token2 = NULL;
    
11.     char onechar;
    
12.     char *command = NULL;
    
13.     char *key = NULL;
    
14.     char *value = NULL;
    
15.     t_client * client = NULL;
    
16.     t_client * lastclient = NULL;
    
17. 
    
18.     config = config_get_config();
    
19.     
    
20.     debug(LOG_INFO, "Connecting to parent to download clients");
    
21. 
    
22.     /* 连接socket */
    
23.     sock = socket(AF_UNIX, SOCK_STREAM, 0);
    
24.     memset(&sa_un, 0, sizeof(sa_un));
    
25.     sa_un.sun_family = AF_UNIX;
    
26.     strncpy(sa_un.sun_path, config->internal_sock, (sizeof(sa_un.sun_path) - 1)); //config->internal_sock的值为"/tmp/wifidog.sock"
    
27. 
    
28.     /* 连接已启动的wifidog */
    
29.     if (connect(sock, (struct sockaddr *)&sa_un, strlen(sa_un.sun_path) + sizeof(sa_un.sun_family))) {
    
30.         debug(LOG_ERR, "Failed to connect to parent (%s) - client list not downloaded", strerror(errno));
    
31.         return;
    
32.     }
    
33. 
    
34.     debug(LOG_INFO, "Connected to parent. Downloading clients");
    
35. 
    
36.     LOCK_CLIENT_LIST();
    
37. 
    
38.     command = NULL;
    
39.     memset(linebuffer, 0, sizeof(linebuffer));
    
40.     len = 0;
    
41.     client = NULL;
    
42.     /* 接收数据，逐个字符接收 */
    
43.     /* 数据包格式为 CLIENT|ip=%s|mac=%s|token=%s|fw_connection_state=%u|fd=%d|counters_incoming=%llu|counters_outgoing=%llu|counters_last_updated=%lun */
    
44.     while (read(sock, &onechar, 1) == 1) {
    
45.         if (onechar == 'n') {
    
46.             /* 如果接收到末尾（'n'），则转为'' */
    
47.             onechar = '';
    
48.         }
    
49.         linebuffer[len++] = onechar;
    
50.         
    
51.         if (!onechar) {
    
52.             /* 以下将数据转化为t_client结构体添加到客户端列表 */
    
53.             debug(LOG_DEBUG, "Received from parent: [%s]", linebuffer);
    
54.             running1 = linebuffer;
    
55.             while ((token1 = strsep(&running1, "|")) != NULL) {
    
56.                 if (!command) {
    
57.                     /* The first token is the command */
    
58.                     command = token1;
    
59.                 }
    
60.                 else {
    
61.                 /* Token1 has something like "foo=bar" */
    
62.                     running2 = token1;
    
63.                     key = value = NULL;
    
64.                     while ((token2 = strsep(&running2, "=")) != NULL) {
    
65.                         if (!key) {
    
66.                             key = token2;
    
67.                         }
    
68.                         else if (!value) {
    
69.                             value = token2;
    
70.                         }
    
71.                     }
    
72.                 }
    
73. 
    
74.                 if (strcmp(command, "CLIENT") == 0) {
    
75.                     /* This line has info about a client in the client list */
    
76.                     if (!client) {
    
77.                         /* Create a new client struct */
    
78.                         client = safe_malloc(sizeof(t_client));
    
79.                         memset(client, 0, sizeof(t_client));
    
80.                     }
    
81.                 }
    
82. 
    
83.                 if (key && value) {
    
84.                     if (strcmp(command, "CLIENT") == 0) {
    
85.                         /* Assign the key into the appropriate slot in the connection structure */
    
86.                         if (strcmp(key, "ip") == 0) {
    
87.                             client->ip = safe_strdup(value);
    
88.                         }
    
89.                         else if (strcmp(key, "mac") == 0) {
    
90.                             client->mac = safe_strdup(value);
    
91.                         }
    
92.                         else if (strcmp(key, "token") == 0) {
    
93.                             client->token = safe_strdup(value);
    
94.                         }
    
95.                         else if (strcmp(key, "fw_connection_state") == 0) {
    
96.                             client->fw_connection_state = atoi(value);
    
97.                         }
    
98.                         else if (strcmp(key, "fd") == 0) {
    
99.                             client->fd = atoi(value);
    
100.                         }
     
101.                         else if (strcmp(key, "counters_incoming") == 0) {
     
102.                             client->counters.incoming_history = atoll(value);
     
103.                             client->counters.incoming = client->counters.incoming_history;
     
104.                         }
     
105.                         else if (strcmp(key, "counters_outgoing") == 0) {
     
106.                             client->counters.outgoing_history = atoll(value);
     
107.                             client->counters.outgoing = client->counters.outgoing_history;
     
108.                         }
     
109.                         else if (strcmp(key, "counters_last_updated") == 0) {
     
110.                             client->counters.last_updated = atol(value);
     
111.                         }
     
112.                         else {
     
113.                             debug(LOG_NOTICE, "I don't know how to inherit key [%s] value [%s] from parent", key, value);
     
114.                         }
     
115.                     }
     
116.                 }
     
117.             }
     
118. 
     
119.             /* End of parsing this command */
     
120.             if (client) {
     
121.                 /* Add this client to the client list */
     
122.                 if (!firstclient) {
     
123.                     firstclient = client;
     
124.                     lastclient = firstclient;
     
125.                 }
     
126.                 else {
     
127.                     lastclient->next = client;
     
128.                     lastclient = client;
     
129.                 }
     
130.             }
     
131. 
     
132.             /* Clean up */
     
133.             command = NULL;
     
134.             memset(linebuffer, 0, sizeof(linebuffer));
     
135.             len = 0;
     
136.             client = NULL;
     
137.         }
     
138.     }
     
139. 
     
140.     UNLOCK_CLIENT_LIST();
     
141.     debug(LOG_INFO, "Client list downloaded successfully from parent");
     
142. 
     
143.     close(sock);
     
144. }

（编辑：淮北站长网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!