]>
Commit | Line | Data |
---|---|---|
cc04acf5 KS |
1 | /* |
2 | * An implementation of key value pair (KVP) functionality for Linux. | |
3 | * | |
4 | * | |
5 | * Copyright (C) 2010, Novell, Inc. | |
6 | * Author : K. Y. Srinivasan <ksrinivasan@novell.com> | |
7 | * | |
8 | * This program is free software; you can redistribute it and/or modify it | |
9 | * under the terms of the GNU General Public License version 2 as published | |
10 | * by the Free Software Foundation. | |
11 | * | |
12 | * This program is distributed in the hope that it will be useful, but | |
13 | * WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 | * MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE, GOOD TITLE or | |
15 | * NON INFRINGEMENT. See the GNU General Public License for more | |
16 | * details. | |
17 | * | |
18 | * You should have received a copy of the GNU General Public License | |
19 | * along with this program; if not, write to the Free Software | |
20 | * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
21 | * | |
22 | */ | |
23 | ||
24 | ||
25 | #include <sys/types.h> | |
26 | #include <sys/socket.h> | |
27 | #include <sys/poll.h> | |
28 | #include <sys/utsname.h> | |
29 | #include <linux/types.h> | |
30 | #include <stdio.h> | |
31 | #include <stdlib.h> | |
32 | #include <unistd.h> | |
33 | #include <string.h> | |
34 | #include <errno.h> | |
35 | #include <arpa/inet.h> | |
36 | #include <linux/connector.h> | |
eab6af71 | 37 | #include <linux/hyperv.h> |
cc04acf5 | 38 | #include <linux/netlink.h> |
cc04acf5 KS |
39 | #include <ifaddrs.h> |
40 | #include <netdb.h> | |
41 | #include <syslog.h> | |
db425334 S |
42 | #include <sys/stat.h> |
43 | #include <fcntl.h> | |
cc04acf5 KS |
44 | |
45 | /* | |
46 | * KVP protocol: The user mode component first registers with the | |
47 | * the kernel component. Subsequently, the kernel component requests, data | |
48 | * for the specified keys. In response to this message the user mode component | |
49 | * fills in the value corresponding to the specified key. We overload the | |
50 | * sequence field in the cn_msg header to define our KVP message types. | |
51 | * | |
52 | * We use this infrastructure for also supporting queries from user mode | |
53 | * application for state that may be maintained in the KVP kernel component. | |
54 | * | |
cc04acf5 KS |
55 | */ |
56 | ||
cc04acf5 KS |
57 | |
58 | enum key_index { | |
59 | FullyQualifiedDomainName = 0, | |
60 | IntegrationServicesVersion, /*This key is serviced in the kernel*/ | |
61 | NetworkAddressIPv4, | |
62 | NetworkAddressIPv6, | |
63 | OSBuildNumber, | |
64 | OSName, | |
65 | OSMajorVersion, | |
66 | OSMinorVersion, | |
67 | OSVersion, | |
68 | ProcessorArchitecture | |
69 | }; | |
70 | ||
cc04acf5 | 71 | static char kvp_send_buffer[4096]; |
9b595780 | 72 | static char kvp_recv_buffer[4096 * 2]; |
cc04acf5 | 73 | static struct sockaddr_nl addr; |
b47a81dc | 74 | static int in_hand_shake = 1; |
cc04acf5 | 75 | |
7989f7d5 OH |
76 | static char *os_name = ""; |
77 | static char *os_major = ""; | |
78 | static char *os_minor = ""; | |
79 | static char *processor_arch; | |
80 | static char *os_build; | |
b47a81dc | 81 | static char *lic_version = "Unknown version"; |
7989f7d5 | 82 | static struct utsname uts_buf; |
cc04acf5 | 83 | |
db425334 S |
84 | |
85 | #define MAX_FILE_NAME 100 | |
86 | #define ENTRIES_PER_BLOCK 50 | |
87 | ||
88 | struct kvp_record { | |
d0cbc156 S |
89 | char key[HV_KVP_EXCHANGE_MAX_KEY_SIZE]; |
90 | char value[HV_KVP_EXCHANGE_MAX_VALUE_SIZE]; | |
db425334 S |
91 | }; |
92 | ||
93 | struct kvp_file_state { | |
94 | int fd; | |
95 | int num_blocks; | |
96 | struct kvp_record *records; | |
97 | int num_records; | |
d0cbc156 | 98 | char fname[MAX_FILE_NAME]; |
db425334 S |
99 | }; |
100 | ||
101 | static struct kvp_file_state kvp_file_info[KVP_POOL_COUNT]; | |
102 | ||
103 | static void kvp_acquire_lock(int pool) | |
104 | { | |
105 | struct flock fl = {F_WRLCK, SEEK_SET, 0, 0, 0}; | |
106 | fl.l_pid = getpid(); | |
107 | ||
108 | if (fcntl(kvp_file_info[pool].fd, F_SETLKW, &fl) == -1) { | |
109 | syslog(LOG_ERR, "Failed to acquire the lock pool: %d", pool); | |
110 | exit(-1); | |
111 | } | |
112 | } | |
113 | ||
114 | static void kvp_release_lock(int pool) | |
115 | { | |
116 | struct flock fl = {F_UNLCK, SEEK_SET, 0, 0, 0}; | |
117 | fl.l_pid = getpid(); | |
118 | ||
119 | if (fcntl(kvp_file_info[pool].fd, F_SETLK, &fl) == -1) { | |
120 | perror("fcntl"); | |
121 | syslog(LOG_ERR, "Failed to release the lock pool: %d", pool); | |
122 | exit(-1); | |
123 | } | |
124 | } | |
125 | ||
126 | static void kvp_update_file(int pool) | |
127 | { | |
128 | FILE *filep; | |
129 | size_t bytes_written; | |
130 | ||
131 | /* | |
132 | * We are going to write our in-memory registry out to | |
133 | * disk; acquire the lock first. | |
134 | */ | |
135 | kvp_acquire_lock(pool); | |
136 | ||
137 | filep = fopen(kvp_file_info[pool].fname, "w"); | |
138 | if (!filep) { | |
139 | kvp_release_lock(pool); | |
140 | syslog(LOG_ERR, "Failed to open file, pool: %d", pool); | |
141 | exit(-1); | |
142 | } | |
143 | ||
144 | bytes_written = fwrite(kvp_file_info[pool].records, | |
145 | sizeof(struct kvp_record), | |
146 | kvp_file_info[pool].num_records, filep); | |
147 | ||
148 | fflush(filep); | |
149 | kvp_release_lock(pool); | |
150 | } | |
151 | ||
adc80ae6 S |
152 | static void kvp_update_mem_state(int pool) |
153 | { | |
154 | FILE *filep; | |
155 | size_t records_read = 0; | |
156 | struct kvp_record *record = kvp_file_info[pool].records; | |
157 | struct kvp_record *readp; | |
158 | int num_blocks = kvp_file_info[pool].num_blocks; | |
159 | int alloc_unit = sizeof(struct kvp_record) * ENTRIES_PER_BLOCK; | |
160 | ||
161 | kvp_acquire_lock(pool); | |
162 | ||
163 | filep = fopen(kvp_file_info[pool].fname, "r"); | |
164 | if (!filep) { | |
165 | kvp_release_lock(pool); | |
166 | syslog(LOG_ERR, "Failed to open file, pool: %d", pool); | |
167 | exit(-1); | |
168 | } | |
169 | while (!feof(filep)) { | |
170 | readp = &record[records_read]; | |
171 | records_read += fread(readp, sizeof(struct kvp_record), | |
172 | ENTRIES_PER_BLOCK * num_blocks, | |
173 | filep); | |
174 | ||
175 | if (!feof(filep)) { | |
176 | /* | |
177 | * We have more data to read. | |
178 | */ | |
179 | num_blocks++; | |
180 | record = realloc(record, alloc_unit * num_blocks); | |
181 | ||
182 | if (record == NULL) { | |
183 | syslog(LOG_ERR, "malloc failed"); | |
184 | exit(-1); | |
185 | } | |
186 | continue; | |
187 | } | |
188 | break; | |
189 | } | |
190 | ||
191 | kvp_file_info[pool].num_blocks = num_blocks; | |
192 | kvp_file_info[pool].records = record; | |
193 | kvp_file_info[pool].num_records = records_read; | |
194 | ||
195 | kvp_release_lock(pool); | |
196 | } | |
db425334 S |
197 | static int kvp_file_init(void) |
198 | { | |
00b83355 | 199 | int fd; |
db425334 S |
200 | FILE *filep; |
201 | size_t records_read; | |
d0cbc156 | 202 | char *fname; |
db425334 S |
203 | struct kvp_record *record; |
204 | struct kvp_record *readp; | |
205 | int num_blocks; | |
206 | int i; | |
207 | int alloc_unit = sizeof(struct kvp_record) * ENTRIES_PER_BLOCK; | |
208 | ||
209 | if (access("/var/opt/hyperv", F_OK)) { | |
210 | if (mkdir("/var/opt/hyperv", S_IRUSR | S_IWUSR | S_IROTH)) { | |
211 | syslog(LOG_ERR, " Failed to create /var/opt/hyperv"); | |
212 | exit(-1); | |
213 | } | |
214 | } | |
215 | ||
216 | for (i = 0; i < KVP_POOL_COUNT; i++) { | |
217 | fname = kvp_file_info[i].fname; | |
218 | records_read = 0; | |
219 | num_blocks = 1; | |
220 | sprintf(fname, "/var/opt/hyperv/.kvp_pool_%d", i); | |
221 | fd = open(fname, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR | S_IROTH); | |
222 | ||
223 | if (fd == -1) | |
224 | return 1; | |
225 | ||
226 | ||
227 | filep = fopen(fname, "r"); | |
228 | if (!filep) | |
229 | return 1; | |
230 | ||
231 | record = malloc(alloc_unit * num_blocks); | |
232 | if (record == NULL) { | |
233 | fclose(filep); | |
234 | return 1; | |
235 | } | |
236 | while (!feof(filep)) { | |
237 | readp = &record[records_read]; | |
238 | records_read += fread(readp, sizeof(struct kvp_record), | |
239 | ENTRIES_PER_BLOCK, | |
240 | filep); | |
241 | ||
242 | if (!feof(filep)) { | |
243 | /* | |
244 | * We have more data to read. | |
245 | */ | |
246 | num_blocks++; | |
247 | record = realloc(record, alloc_unit * | |
248 | num_blocks); | |
249 | if (record == NULL) { | |
250 | fclose(filep); | |
251 | return 1; | |
252 | } | |
253 | continue; | |
254 | } | |
255 | break; | |
256 | } | |
257 | kvp_file_info[i].fd = fd; | |
258 | kvp_file_info[i].num_blocks = num_blocks; | |
259 | kvp_file_info[i].records = record; | |
260 | kvp_file_info[i].num_records = records_read; | |
261 | fclose(filep); | |
262 | ||
263 | } | |
264 | ||
265 | return 0; | |
266 | } | |
267 | ||
268 | static int kvp_key_delete(int pool, __u8 *key, int key_size) | |
269 | { | |
270 | int i; | |
271 | int j, k; | |
adc80ae6 S |
272 | int num_records; |
273 | struct kvp_record *record; | |
274 | ||
275 | /* | |
276 | * First update the in-memory state. | |
277 | */ | |
278 | kvp_update_mem_state(pool); | |
279 | ||
280 | num_records = kvp_file_info[pool].num_records; | |
281 | record = kvp_file_info[pool].records; | |
db425334 S |
282 | |
283 | for (i = 0; i < num_records; i++) { | |
284 | if (memcmp(key, record[i].key, key_size)) | |
285 | continue; | |
286 | /* | |
287 | * Found a match; just move the remaining | |
288 | * entries up. | |
289 | */ | |
290 | if (i == num_records) { | |
291 | kvp_file_info[pool].num_records--; | |
292 | kvp_update_file(pool); | |
293 | return 0; | |
294 | } | |
295 | ||
296 | j = i; | |
297 | k = j + 1; | |
298 | for (; k < num_records; k++) { | |
299 | strcpy(record[j].key, record[k].key); | |
300 | strcpy(record[j].value, record[k].value); | |
301 | j++; | |
302 | } | |
303 | ||
304 | kvp_file_info[pool].num_records--; | |
305 | kvp_update_file(pool); | |
306 | return 0; | |
307 | } | |
308 | return 1; | |
309 | } | |
310 | ||
311 | static int kvp_key_add_or_modify(int pool, __u8 *key, int key_size, __u8 *value, | |
312 | int value_size) | |
313 | { | |
314 | int i; | |
adc80ae6 S |
315 | int num_records; |
316 | struct kvp_record *record; | |
317 | int num_blocks; | |
db425334 S |
318 | |
319 | if ((key_size > HV_KVP_EXCHANGE_MAX_KEY_SIZE) || | |
320 | (value_size > HV_KVP_EXCHANGE_MAX_VALUE_SIZE)) | |
321 | return 1; | |
322 | ||
adc80ae6 S |
323 | /* |
324 | * First update the in-memory state. | |
325 | */ | |
326 | kvp_update_mem_state(pool); | |
327 | ||
328 | num_records = kvp_file_info[pool].num_records; | |
329 | record = kvp_file_info[pool].records; | |
330 | num_blocks = kvp_file_info[pool].num_blocks; | |
331 | ||
db425334 S |
332 | for (i = 0; i < num_records; i++) { |
333 | if (memcmp(key, record[i].key, key_size)) | |
334 | continue; | |
335 | /* | |
336 | * Found a match; just update the value - | |
337 | * this is the modify case. | |
338 | */ | |
339 | memcpy(record[i].value, value, value_size); | |
340 | kvp_update_file(pool); | |
341 | return 0; | |
342 | } | |
343 | ||
344 | /* | |
345 | * Need to add a new entry; | |
346 | */ | |
347 | if (num_records == (ENTRIES_PER_BLOCK * num_blocks)) { | |
348 | /* Need to allocate a larger array for reg entries. */ | |
349 | record = realloc(record, sizeof(struct kvp_record) * | |
350 | ENTRIES_PER_BLOCK * (num_blocks + 1)); | |
351 | ||
352 | if (record == NULL) | |
353 | return 1; | |
354 | kvp_file_info[pool].num_blocks++; | |
355 | ||
356 | } | |
357 | memcpy(record[i].value, value, value_size); | |
358 | memcpy(record[i].key, key, key_size); | |
359 | kvp_file_info[pool].records = record; | |
360 | kvp_file_info[pool].num_records++; | |
361 | kvp_update_file(pool); | |
362 | return 0; | |
363 | } | |
364 | ||
365 | static int kvp_get_value(int pool, __u8 *key, int key_size, __u8 *value, | |
366 | int value_size) | |
367 | { | |
368 | int i; | |
adc80ae6 S |
369 | int num_records; |
370 | struct kvp_record *record; | |
db425334 S |
371 | |
372 | if ((key_size > HV_KVP_EXCHANGE_MAX_KEY_SIZE) || | |
373 | (value_size > HV_KVP_EXCHANGE_MAX_VALUE_SIZE)) | |
374 | return 1; | |
375 | ||
adc80ae6 S |
376 | /* |
377 | * First update the in-memory state. | |
378 | */ | |
379 | kvp_update_mem_state(pool); | |
380 | ||
381 | num_records = kvp_file_info[pool].num_records; | |
382 | record = kvp_file_info[pool].records; | |
383 | ||
db425334 S |
384 | for (i = 0; i < num_records; i++) { |
385 | if (memcmp(key, record[i].key, key_size)) | |
386 | continue; | |
387 | /* | |
388 | * Found a match; just copy the value out. | |
389 | */ | |
390 | memcpy(value, record[i].value, value_size); | |
391 | return 0; | |
392 | } | |
393 | ||
394 | return 1; | |
395 | } | |
396 | ||
b47a81dc | 397 | static int kvp_pool_enumerate(int pool, int index, __u8 *key, int key_size, |
adc80ae6 S |
398 | __u8 *value, int value_size) |
399 | { | |
400 | struct kvp_record *record; | |
401 | ||
402 | /* | |
403 | * First update our in-memory database. | |
404 | */ | |
405 | kvp_update_mem_state(pool); | |
406 | record = kvp_file_info[pool].records; | |
407 | ||
408 | if (index >= kvp_file_info[pool].num_records) { | |
b47a81dc | 409 | return 1; |
adc80ae6 S |
410 | } |
411 | ||
412 | memcpy(key, record[index].key, key_size); | |
413 | memcpy(value, record[index].value, value_size); | |
b47a81dc | 414 | return 0; |
adc80ae6 S |
415 | } |
416 | ||
417 | ||
cc04acf5 KS |
418 | void kvp_get_os_info(void) |
419 | { | |
420 | FILE *file; | |
7989f7d5 | 421 | char *p, buf[512]; |
cc04acf5 | 422 | |
7989f7d5 OH |
423 | uname(&uts_buf); |
424 | os_build = uts_buf.release; | |
064931d0 | 425 | processor_arch = uts_buf.machine; |
cc04acf5 | 426 | |
e54bbc64 S |
427 | /* |
428 | * The current windows host (win7) expects the build | |
429 | * string to be of the form: x.y.z | |
430 | * Strip additional information we may have. | |
431 | */ | |
432 | p = strchr(os_build, '-'); | |
433 | if (p) | |
434 | *p = '\0'; | |
435 | ||
cc04acf5 KS |
436 | file = fopen("/etc/SuSE-release", "r"); |
437 | if (file != NULL) | |
438 | goto kvp_osinfo_found; | |
439 | file = fopen("/etc/redhat-release", "r"); | |
440 | if (file != NULL) | |
441 | goto kvp_osinfo_found; | |
442 | /* | |
443 | * Add code for other supported platforms. | |
444 | */ | |
445 | ||
446 | /* | |
447 | * We don't have information about the os. | |
448 | */ | |
7989f7d5 | 449 | os_name = uts_buf.sysname; |
cc04acf5 KS |
450 | return; |
451 | ||
452 | kvp_osinfo_found: | |
7989f7d5 OH |
453 | /* up to three lines */ |
454 | p = fgets(buf, sizeof(buf), file); | |
455 | if (p) { | |
456 | p = strchr(buf, '\n'); | |
457 | if (p) | |
458 | *p = '\0'; | |
459 | p = strdup(buf); | |
460 | if (!p) | |
461 | goto done; | |
462 | os_name = p; | |
463 | ||
464 | /* second line */ | |
465 | p = fgets(buf, sizeof(buf), file); | |
466 | if (p) { | |
467 | p = strchr(buf, '\n'); | |
468 | if (p) | |
469 | *p = '\0'; | |
470 | p = strdup(buf); | |
471 | if (!p) | |
472 | goto done; | |
473 | os_major = p; | |
474 | ||
475 | /* third line */ | |
476 | p = fgets(buf, sizeof(buf), file); | |
477 | if (p) { | |
478 | p = strchr(buf, '\n'); | |
479 | if (p) | |
480 | *p = '\0'; | |
481 | p = strdup(buf); | |
482 | if (p) | |
483 | os_minor = p; | |
484 | } | |
485 | } | |
486 | } | |
487 | ||
488 | done: | |
cc04acf5 KS |
489 | fclose(file); |
490 | return; | |
491 | } | |
492 | ||
4a52c4af S |
493 | static void kvp_process_ipconfig_file(char *cmd, |
494 | char *config_buf, int len, | |
495 | int element_size, int offset) | |
496 | { | |
497 | char buf[256]; | |
498 | char *p; | |
499 | char *x; | |
500 | FILE *file; | |
501 | ||
502 | /* | |
503 | * First execute the command. | |
504 | */ | |
505 | file = popen(cmd, "r"); | |
506 | if (file == NULL) | |
507 | return; | |
508 | ||
509 | if (offset == 0) | |
510 | memset(config_buf, 0, len); | |
511 | while ((p = fgets(buf, sizeof(buf), file)) != NULL) { | |
512 | if ((len - strlen(config_buf)) < (element_size + 1)) | |
513 | break; | |
514 | ||
515 | x = strchr(p, '\n'); | |
516 | *x = '\0'; | |
517 | strcat(config_buf, p); | |
518 | strcat(config_buf, ";"); | |
519 | } | |
520 | pclose(file); | |
521 | } | |
522 | ||
523 | static void kvp_get_ipconfig_info(char *if_name, | |
524 | struct hv_kvp_ipaddr_value *buffer) | |
525 | { | |
526 | char cmd[512]; | |
527 | ||
528 | /* | |
529 | * Get the address of default gateway (ipv4). | |
530 | */ | |
531 | sprintf(cmd, "%s %s", "ip route show dev", if_name); | |
532 | strcat(cmd, " | awk '/default/ {print $3 }'"); | |
533 | ||
534 | /* | |
535 | * Execute the command to gather gateway info. | |
536 | */ | |
537 | kvp_process_ipconfig_file(cmd, (char *)buffer->gate_way, | |
538 | (MAX_GATEWAY_SIZE * 2), INET_ADDRSTRLEN, 0); | |
539 | ||
540 | /* | |
541 | * Get the address of default gateway (ipv6). | |
542 | */ | |
543 | sprintf(cmd, "%s %s", "ip -f inet6 route show dev", if_name); | |
544 | strcat(cmd, " | awk '/default/ {print $3 }'"); | |
545 | ||
546 | /* | |
547 | * Execute the command to gather gateway info (ipv6). | |
548 | */ | |
549 | kvp_process_ipconfig_file(cmd, (char *)buffer->gate_way, | |
550 | (MAX_GATEWAY_SIZE * 2), INET6_ADDRSTRLEN, 1); | |
551 | ||
96929887 S |
552 | |
553 | /* | |
554 | * Gather the DNS state. | |
555 | * Since there is no standard way to get this information | |
556 | * across various distributions of interest; we just invoke | |
557 | * an external script that needs to be ported across distros | |
558 | * of interest. | |
559 | * | |
560 | * Following is the expected format of the information from the script: | |
561 | * | |
562 | * ipaddr1 (nameserver1) | |
563 | * ipaddr2 (nameserver2) | |
564 | * . | |
565 | * . | |
566 | */ | |
567 | ||
568 | sprintf(cmd, "%s", "hv_get_dns_info"); | |
569 | ||
570 | /* | |
571 | * Execute the command to gather DNS info. | |
572 | */ | |
573 | kvp_process_ipconfig_file(cmd, (char *)buffer->dns_addr, | |
574 | (MAX_IP_ADDR_SIZE * 2), INET_ADDRSTRLEN, 0); | |
4a52c4af S |
575 | } |
576 | ||
577 | ||
6a60a6a8 S |
578 | static unsigned int hweight32(unsigned int *w) |
579 | { | |
580 | unsigned int res = *w - ((*w >> 1) & 0x55555555); | |
581 | res = (res & 0x33333333) + ((res >> 2) & 0x33333333); | |
582 | res = (res + (res >> 4)) & 0x0F0F0F0F; | |
583 | res = res + (res >> 8); | |
584 | return (res + (res >> 16)) & 0x000000FF; | |
585 | } | |
586 | ||
af733015 S |
587 | static int kvp_process_ip_address(void *addrp, |
588 | int family, char *buffer, | |
589 | int length, int *offset) | |
590 | { | |
591 | struct sockaddr_in *addr; | |
592 | struct sockaddr_in6 *addr6; | |
593 | int addr_length; | |
594 | char tmp[50]; | |
595 | const char *str; | |
596 | ||
597 | if (family == AF_INET) { | |
598 | addr = (struct sockaddr_in *)addrp; | |
599 | str = inet_ntop(family, &addr->sin_addr, tmp, 50); | |
600 | addr_length = INET_ADDRSTRLEN; | |
601 | } else { | |
602 | addr6 = (struct sockaddr_in6 *)addrp; | |
603 | str = inet_ntop(family, &addr6->sin6_addr.s6_addr, tmp, 50); | |
604 | addr_length = INET6_ADDRSTRLEN; | |
605 | } | |
606 | ||
607 | if ((length - *offset) < addr_length + 1) | |
608 | return 1; | |
609 | if (str == NULL) { | |
610 | strcpy(buffer, "inet_ntop failed\n"); | |
611 | return 1; | |
612 | } | |
613 | if (*offset == 0) | |
614 | strcpy(buffer, tmp); | |
615 | else | |
616 | strcat(buffer, tmp); | |
617 | strcat(buffer, ";"); | |
618 | ||
619 | *offset += strlen(str) + 1; | |
620 | return 0; | |
621 | } | |
622 | ||
cc04acf5 | 623 | static int |
0ecaa198 S |
624 | kvp_get_ip_address(int family, char *if_name, int op, |
625 | void *out_buffer, int length) | |
cc04acf5 KS |
626 | { |
627 | struct ifaddrs *ifap; | |
628 | struct ifaddrs *curp; | |
cc04acf5 | 629 | int offset = 0; |
04405784 | 630 | int sn_offset = 0; |
cc04acf5 | 631 | int error = 0; |
0ecaa198 S |
632 | char *buffer; |
633 | struct hv_kvp_ipaddr_value *ip_buffer; | |
6a60a6a8 S |
634 | char cidr_mask[5]; /* /xyz */ |
635 | int weight; | |
636 | int i; | |
637 | unsigned int *w; | |
638 | char *sn_str; | |
639 | struct sockaddr_in6 *addr6; | |
0ecaa198 S |
640 | |
641 | if (op == KVP_OP_ENUMERATE) { | |
642 | buffer = out_buffer; | |
643 | } else { | |
644 | ip_buffer = out_buffer; | |
645 | buffer = (char *)ip_buffer->ip_addr; | |
646 | ip_buffer->addr_family = 0; | |
647 | } | |
cc04acf5 KS |
648 | /* |
649 | * On entry into this function, the buffer is capable of holding the | |
0ecaa198 | 650 | * maximum key value. |
cc04acf5 KS |
651 | */ |
652 | ||
653 | if (getifaddrs(&ifap)) { | |
654 | strcpy(buffer, "getifaddrs failed\n"); | |
655 | return 1; | |
656 | } | |
657 | ||
658 | curp = ifap; | |
659 | while (curp != NULL) { | |
0ecaa198 S |
660 | if (curp->ifa_addr == NULL) { |
661 | curp = curp->ifa_next; | |
662 | continue; | |
663 | } | |
cc04acf5 | 664 | |
0ecaa198 S |
665 | if ((if_name != NULL) && |
666 | (strncmp(curp->ifa_name, if_name, strlen(if_name)))) { | |
667 | /* | |
668 | * We want info about a specific interface; | |
669 | * just continue. | |
670 | */ | |
671 | curp = curp->ifa_next; | |
672 | continue; | |
673 | } | |
cc04acf5 | 674 | |
0ecaa198 S |
675 | /* |
676 | * We only support two address families: AF_INET and AF_INET6. | |
677 | * If a family value of 0 is specified, we collect both | |
678 | * supported address families; if not we gather info on | |
679 | * the specified address family. | |
680 | */ | |
681 | if ((family != 0) && (curp->ifa_addr->sa_family != family)) { | |
682 | curp = curp->ifa_next; | |
683 | continue; | |
684 | } | |
685 | if ((curp->ifa_addr->sa_family != AF_INET) && | |
686 | (curp->ifa_addr->sa_family != AF_INET6)) { | |
687 | curp = curp->ifa_next; | |
688 | continue; | |
689 | } | |
690 | ||
0d5b6b19 S |
691 | if (op == KVP_OP_GET_IP_INFO) { |
692 | /* | |
693 | * Gather info other than the IP address. | |
694 | * IP address info will be gathered later. | |
695 | */ | |
04405784 | 696 | if (curp->ifa_addr->sa_family == AF_INET) { |
0d5b6b19 | 697 | ip_buffer->addr_family |= ADDR_FAMILY_IPV4; |
04405784 S |
698 | /* |
699 | * Get subnet info. | |
700 | */ | |
701 | error = kvp_process_ip_address( | |
702 | curp->ifa_netmask, | |
703 | AF_INET, | |
704 | (char *) | |
705 | ip_buffer->sub_net, | |
706 | length, | |
707 | &sn_offset); | |
708 | if (error) | |
709 | goto gather_ipaddr; | |
710 | } else { | |
0d5b6b19 | 711 | ip_buffer->addr_family |= ADDR_FAMILY_IPV6; |
6a60a6a8 | 712 | |
04405784 | 713 | /* |
6a60a6a8 | 714 | * Get subnet info in CIDR format. |
04405784 | 715 | */ |
6a60a6a8 S |
716 | weight = 0; |
717 | sn_str = (char *)ip_buffer->sub_net; | |
718 | addr6 = (struct sockaddr_in6 *) | |
719 | curp->ifa_netmask; | |
720 | w = addr6->sin6_addr.s6_addr32; | |
721 | ||
722 | for (i = 0; i < 4; i++) | |
723 | weight += hweight32(&w[i]); | |
724 | ||
725 | sprintf(cidr_mask, "/%d", weight); | |
726 | if ((length - sn_offset) < | |
727 | (strlen(cidr_mask) + 1)) | |
04405784 | 728 | goto gather_ipaddr; |
6a60a6a8 S |
729 | |
730 | if (sn_offset == 0) | |
731 | strcpy(sn_str, cidr_mask); | |
732 | else | |
733 | strcat(sn_str, cidr_mask); | |
734 | strcat((char *)ip_buffer->sub_net, ";"); | |
735 | sn_offset += strlen(sn_str) + 1; | |
04405784 | 736 | } |
4a52c4af S |
737 | |
738 | /* | |
739 | * Collect other ip related configuration info. | |
740 | */ | |
741 | ||
742 | kvp_get_ipconfig_info(if_name, ip_buffer); | |
0d5b6b19 S |
743 | } |
744 | ||
04405784 | 745 | gather_ipaddr: |
af733015 S |
746 | error = kvp_process_ip_address(curp->ifa_addr, |
747 | curp->ifa_addr->sa_family, | |
748 | buffer, | |
749 | length, &offset); | |
750 | if (error) | |
751 | goto getaddr_done; | |
0ecaa198 | 752 | |
cc04acf5 KS |
753 | curp = curp->ifa_next; |
754 | } | |
755 | ||
756 | getaddr_done: | |
757 | freeifaddrs(ifap); | |
758 | return error; | |
759 | } | |
760 | ||
761 | ||
762 | static int | |
763 | kvp_get_domain_name(char *buffer, int length) | |
764 | { | |
765 | struct addrinfo hints, *info ; | |
cc04acf5 KS |
766 | int error = 0; |
767 | ||
5be528c2 | 768 | gethostname(buffer, length); |
cc04acf5 KS |
769 | memset(&hints, 0, sizeof(hints)); |
770 | hints.ai_family = AF_INET; /*Get only ipv4 addrinfo. */ | |
771 | hints.ai_socktype = SOCK_STREAM; | |
772 | hints.ai_flags = AI_CANONNAME; | |
773 | ||
5be528c2 | 774 | error = getaddrinfo(buffer, NULL, &hints, &info); |
cc04acf5 KS |
775 | if (error != 0) { |
776 | strcpy(buffer, "getaddrinfo failed\n"); | |
5be528c2 | 777 | return error; |
cc04acf5 KS |
778 | } |
779 | strcpy(buffer, info->ai_canonname); | |
cc04acf5 KS |
780 | freeaddrinfo(info); |
781 | return error; | |
782 | } | |
783 | ||
784 | static int | |
785 | netlink_send(int fd, struct cn_msg *msg) | |
786 | { | |
787 | struct nlmsghdr *nlh; | |
788 | unsigned int size; | |
789 | struct msghdr message; | |
790 | char buffer[64]; | |
791 | struct iovec iov[2]; | |
792 | ||
793 | size = NLMSG_SPACE(sizeof(struct cn_msg) + msg->len); | |
794 | ||
795 | nlh = (struct nlmsghdr *)buffer; | |
796 | nlh->nlmsg_seq = 0; | |
797 | nlh->nlmsg_pid = getpid(); | |
798 | nlh->nlmsg_type = NLMSG_DONE; | |
799 | nlh->nlmsg_len = NLMSG_LENGTH(size - sizeof(*nlh)); | |
800 | nlh->nlmsg_flags = 0; | |
801 | ||
802 | iov[0].iov_base = nlh; | |
803 | iov[0].iov_len = sizeof(*nlh); | |
804 | ||
805 | iov[1].iov_base = msg; | |
806 | iov[1].iov_len = size; | |
807 | ||
808 | memset(&message, 0, sizeof(message)); | |
809 | message.msg_name = &addr; | |
810 | message.msg_namelen = sizeof(addr); | |
811 | message.msg_iov = iov; | |
812 | message.msg_iovlen = 2; | |
813 | ||
814 | return sendmsg(fd, &message, 0); | |
815 | } | |
816 | ||
7989f7d5 | 817 | int main(void) |
cc04acf5 KS |
818 | { |
819 | int fd, len, sock_opt; | |
820 | int error; | |
821 | struct cn_msg *message; | |
822 | struct pollfd pfd; | |
823 | struct nlmsghdr *incoming_msg; | |
824 | struct cn_msg *incoming_cn_msg; | |
26403354 | 825 | struct hv_kvp_msg *hv_msg; |
7989f7d5 | 826 | char *p; |
cc04acf5 KS |
827 | char *key_value; |
828 | char *key_name; | |
b47a81dc S |
829 | int op; |
830 | int pool; | |
cc04acf5 KS |
831 | |
832 | daemon(1, 0); | |
833 | openlog("KVP", 0, LOG_USER); | |
834 | syslog(LOG_INFO, "KVP starting; pid is:%d", getpid()); | |
835 | /* | |
836 | * Retrieve OS release information. | |
837 | */ | |
838 | kvp_get_os_info(); | |
839 | ||
db425334 S |
840 | if (kvp_file_init()) { |
841 | syslog(LOG_ERR, "Failed to initialize the pools"); | |
842 | exit(-1); | |
843 | } | |
844 | ||
cc04acf5 KS |
845 | fd = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_CONNECTOR); |
846 | if (fd < 0) { | |
847 | syslog(LOG_ERR, "netlink socket creation failed; error:%d", fd); | |
848 | exit(-1); | |
849 | } | |
850 | addr.nl_family = AF_NETLINK; | |
851 | addr.nl_pad = 0; | |
852 | addr.nl_pid = 0; | |
853 | addr.nl_groups = CN_KVP_IDX; | |
854 | ||
855 | ||
856 | error = bind(fd, (struct sockaddr *)&addr, sizeof(addr)); | |
857 | if (error < 0) { | |
858 | syslog(LOG_ERR, "bind failed; error:%d", error); | |
859 | close(fd); | |
860 | exit(-1); | |
861 | } | |
862 | sock_opt = addr.nl_groups; | |
863 | setsockopt(fd, 270, 1, &sock_opt, sizeof(sock_opt)); | |
864 | /* | |
865 | * Register ourselves with the kernel. | |
866 | */ | |
867 | message = (struct cn_msg *)kvp_send_buffer; | |
868 | message->id.idx = CN_KVP_IDX; | |
869 | message->id.val = CN_KVP_VAL; | |
26403354 S |
870 | |
871 | hv_msg = (struct hv_kvp_msg *)message->data; | |
b47a81dc | 872 | hv_msg->kvp_hdr.operation = KVP_OP_REGISTER1; |
cc04acf5 | 873 | message->ack = 0; |
26403354 | 874 | message->len = sizeof(struct hv_kvp_msg); |
cc04acf5 KS |
875 | |
876 | len = netlink_send(fd, message); | |
877 | if (len < 0) { | |
878 | syslog(LOG_ERR, "netlink_send failed; error:%d", len); | |
879 | close(fd); | |
880 | exit(-1); | |
881 | } | |
882 | ||
883 | pfd.fd = fd; | |
884 | ||
885 | while (1) { | |
bcc2c9c3 OH |
886 | struct sockaddr *addr_p = (struct sockaddr *) &addr; |
887 | socklen_t addr_l = sizeof(addr); | |
cc04acf5 KS |
888 | pfd.events = POLLIN; |
889 | pfd.revents = 0; | |
890 | poll(&pfd, 1, -1); | |
891 | ||
bcc2c9c3 OH |
892 | len = recvfrom(fd, kvp_recv_buffer, sizeof(kvp_recv_buffer), 0, |
893 | addr_p, &addr_l); | |
cc04acf5 | 894 | |
bcc2c9c3 OH |
895 | if (len < 0 || addr.nl_pid) { |
896 | syslog(LOG_ERR, "recvfrom failed; pid:%u error:%d %s", | |
897 | addr.nl_pid, errno, strerror(errno)); | |
cc04acf5 KS |
898 | close(fd); |
899 | return -1; | |
900 | } | |
901 | ||
902 | incoming_msg = (struct nlmsghdr *)kvp_recv_buffer; | |
903 | incoming_cn_msg = (struct cn_msg *)NLMSG_DATA(incoming_msg); | |
26403354 | 904 | hv_msg = (struct hv_kvp_msg *)incoming_cn_msg->data; |
cc04acf5 | 905 | |
b47a81dc S |
906 | /* |
907 | * We will use the KVP header information to pass back | |
908 | * the error from this daemon. So, first copy the state | |
909 | * and set the error code to success. | |
910 | */ | |
911 | op = hv_msg->kvp_hdr.operation; | |
912 | pool = hv_msg->kvp_hdr.pool; | |
913 | hv_msg->error = HV_S_OK; | |
914 | ||
915 | if ((in_hand_shake) && (op == KVP_OP_REGISTER1)) { | |
cc04acf5 KS |
916 | /* |
917 | * Driver is registering with us; stash away the version | |
918 | * information. | |
919 | */ | |
b47a81dc | 920 | in_hand_shake = 0; |
e485ceac | 921 | p = (char *)hv_msg->body.kvp_register.version; |
7989f7d5 | 922 | lic_version = malloc(strlen(p) + 1); |
cc04acf5 | 923 | if (lic_version) { |
7989f7d5 | 924 | strcpy(lic_version, p); |
cc04acf5 KS |
925 | syslog(LOG_INFO, "KVP LIC Version: %s", |
926 | lic_version); | |
927 | } else { | |
928 | syslog(LOG_ERR, "malloc failed"); | |
929 | } | |
930 | continue; | |
b47a81dc | 931 | } |
cc04acf5 | 932 | |
b47a81dc | 933 | switch (op) { |
fa3d5b85 | 934 | case KVP_OP_SET: |
b47a81dc | 935 | if (kvp_key_add_or_modify(pool, |
db425334 S |
936 | hv_msg->body.kvp_set.data.key, |
937 | hv_msg->body.kvp_set.data.key_size, | |
938 | hv_msg->body.kvp_set.data.value, | |
939 | hv_msg->body.kvp_set.data.value_size)) | |
b47a81dc | 940 | hv_msg->error = HV_S_CONT; |
db425334 S |
941 | break; |
942 | ||
fa3d5b85 | 943 | case KVP_OP_GET: |
b47a81dc | 944 | if (kvp_get_value(pool, |
db425334 S |
945 | hv_msg->body.kvp_set.data.key, |
946 | hv_msg->body.kvp_set.data.key_size, | |
947 | hv_msg->body.kvp_set.data.value, | |
948 | hv_msg->body.kvp_set.data.value_size)) | |
b47a81dc | 949 | hv_msg->error = HV_S_CONT; |
db425334 S |
950 | break; |
951 | ||
fa3d5b85 | 952 | case KVP_OP_DELETE: |
b47a81dc | 953 | if (kvp_key_delete(pool, |
db425334 S |
954 | hv_msg->body.kvp_delete.key, |
955 | hv_msg->body.kvp_delete.key_size)) | |
b47a81dc | 956 | hv_msg->error = HV_S_CONT; |
db425334 S |
957 | break; |
958 | ||
cc04acf5 | 959 | default: |
26403354 | 960 | break; |
cc04acf5 KS |
961 | } |
962 | ||
b47a81dc | 963 | if (op != KVP_OP_ENUMERATE) |
fa3d5b85 S |
964 | goto kvp_done; |
965 | ||
adc80ae6 S |
966 | /* |
967 | * If the pool is KVP_POOL_AUTO, dynamically generate | |
968 | * both the key and the value; if not read from the | |
969 | * appropriate pool. | |
970 | */ | |
b47a81dc S |
971 | if (pool != KVP_POOL_AUTO) { |
972 | if (kvp_pool_enumerate(pool, | |
adc80ae6 S |
973 | hv_msg->body.kvp_enum_data.index, |
974 | hv_msg->body.kvp_enum_data.data.key, | |
975 | HV_KVP_EXCHANGE_MAX_KEY_SIZE, | |
976 | hv_msg->body.kvp_enum_data.data.value, | |
b47a81dc S |
977 | HV_KVP_EXCHANGE_MAX_VALUE_SIZE)) |
978 | hv_msg->error = HV_S_CONT; | |
adc80ae6 S |
979 | goto kvp_done; |
980 | } | |
981 | ||
26403354 S |
982 | hv_msg = (struct hv_kvp_msg *)incoming_cn_msg->data; |
983 | key_name = (char *)hv_msg->body.kvp_enum_data.data.key; | |
984 | key_value = (char *)hv_msg->body.kvp_enum_data.data.value; | |
cc04acf5 | 985 | |
26403354 | 986 | switch (hv_msg->body.kvp_enum_data.index) { |
cc04acf5 KS |
987 | case FullyQualifiedDomainName: |
988 | kvp_get_domain_name(key_value, | |
989 | HV_KVP_EXCHANGE_MAX_VALUE_SIZE); | |
990 | strcpy(key_name, "FullyQualifiedDomainName"); | |
991 | break; | |
992 | case IntegrationServicesVersion: | |
993 | strcpy(key_name, "IntegrationServicesVersion"); | |
994 | strcpy(key_value, lic_version); | |
995 | break; | |
996 | case NetworkAddressIPv4: | |
0ecaa198 S |
997 | kvp_get_ip_address(AF_INET, NULL, KVP_OP_ENUMERATE, |
998 | key_value, HV_KVP_EXCHANGE_MAX_VALUE_SIZE); | |
cc04acf5 KS |
999 | strcpy(key_name, "NetworkAddressIPv4"); |
1000 | break; | |
1001 | case NetworkAddressIPv6: | |
0ecaa198 S |
1002 | kvp_get_ip_address(AF_INET6, NULL, KVP_OP_ENUMERATE, |
1003 | key_value, HV_KVP_EXCHANGE_MAX_VALUE_SIZE); | |
cc04acf5 KS |
1004 | strcpy(key_name, "NetworkAddressIPv6"); |
1005 | break; | |
1006 | case OSBuildNumber: | |
1007 | strcpy(key_value, os_build); | |
1008 | strcpy(key_name, "OSBuildNumber"); | |
1009 | break; | |
1010 | case OSName: | |
1011 | strcpy(key_value, os_name); | |
1012 | strcpy(key_name, "OSName"); | |
1013 | break; | |
1014 | case OSMajorVersion: | |
1015 | strcpy(key_value, os_major); | |
1016 | strcpy(key_name, "OSMajorVersion"); | |
1017 | break; | |
1018 | case OSMinorVersion: | |
1019 | strcpy(key_value, os_minor); | |
1020 | strcpy(key_name, "OSMinorVersion"); | |
1021 | break; | |
1022 | case OSVersion: | |
1023 | strcpy(key_value, os_build); | |
1024 | strcpy(key_name, "OSVersion"); | |
1025 | break; | |
1026 | case ProcessorArchitecture: | |
1027 | strcpy(key_value, processor_arch); | |
1028 | strcpy(key_name, "ProcessorArchitecture"); | |
1029 | break; | |
1030 | default: | |
b47a81dc | 1031 | hv_msg->error = HV_S_CONT; |
cc04acf5 KS |
1032 | break; |
1033 | } | |
1034 | /* | |
1035 | * Send the value back to the kernel. The response is | |
1036 | * already in the receive buffer. Update the cn_msg header to | |
1037 | * reflect the key value that has been added to the message | |
1038 | */ | |
fa3d5b85 | 1039 | kvp_done: |
cc04acf5 KS |
1040 | |
1041 | incoming_cn_msg->id.idx = CN_KVP_IDX; | |
1042 | incoming_cn_msg->id.val = CN_KVP_VAL; | |
cc04acf5 | 1043 | incoming_cn_msg->ack = 0; |
26403354 | 1044 | incoming_cn_msg->len = sizeof(struct hv_kvp_msg); |
cc04acf5 KS |
1045 | |
1046 | len = netlink_send(fd, incoming_cn_msg); | |
1047 | if (len < 0) { | |
1048 | syslog(LOG_ERR, "net_link send failed; error:%d", len); | |
1049 | exit(-1); | |
1050 | } | |
1051 | } | |
1052 | ||
1053 | } |