tools/hv/hv_kvp_daemon.c

   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 <ctype.h>
  35 #include <errno.h>
  36 #include <arpa/inet.h>
  37 #include <linux/connector.h>
  38 #include <linux/hyperv.h>
  39 #include <linux/netlink.h>
  40 #include <ifaddrs.h>
  41 #include <netdb.h>
  42 #include <syslog.h>
  43 #include <sys/stat.h>
  44 #include <fcntl.h>
  45 #include <dirent.h>
  46
  47 /*
  48  * KVP protocol: The user mode component first registers with the
  49  * the kernel component. Subsequently, the kernel component requests, data
  50  * for the specified keys. In response to this message the user mode component
  51  * fills in the value corresponding to the specified key. We overload the
  52  * sequence field in the cn_msg header to define our KVP message types.
  53  *
  54  * We use this infrastructure for also supporting queries from user mode
  55  * application for state that may be maintained in the KVP kernel component.
  56  *
  57  */
  58
  59
  60 enum key_index {
  61         FullyQualifiedDomainName = 0,
  62         IntegrationServicesVersion, /*This key is serviced in the kernel*/
  63         NetworkAddressIPv4,
  64         NetworkAddressIPv6,
  65         OSBuildNumber,
  66         OSName,
  67         OSMajorVersion,
  68         OSMinorVersion,
  69         OSVersion,
  70         ProcessorArchitecture
  71 };
  72
  73
  74 enum {
  75         IPADDR = 0,
  76         NETMASK,
  77         GATEWAY,
  78         DNS
  79 };
  80
  81 static char kvp_send_buffer[4096];
  82 static char kvp_recv_buffer[4096 * 2];
  83 static struct sockaddr_nl addr;
  84 static int in_hand_shake = 1;
  85
  86 static char *os_name = "";
  87 static char *os_major = "";
  88 static char *os_minor = "";
  89 static char *processor_arch;
  90 static char *os_build;
  91 static char *lic_version = "Unknown version";
  92 static struct utsname uts_buf;
  93
  94 /*
  95  * The location of the interface configuration file.
  96  */
  97
  98 #define KVP_CONFIG_LOC  "/var/opt/"
  99
 100 #define MAX_FILE_NAME 100
 101 #define ENTRIES_PER_BLOCK 50
 102
 103 struct kvp_record {
 104         char key[HV_KVP_EXCHANGE_MAX_KEY_SIZE];
 105         char value[HV_KVP_EXCHANGE_MAX_VALUE_SIZE];
 106 };
 107
 108 struct kvp_file_state {
 109         int fd;
 110         int num_blocks;
 111         struct kvp_record *records;
 112         int num_records;
 113         char fname[MAX_FILE_NAME];
 114 };
 115
 116 static struct kvp_file_state kvp_file_info[KVP_POOL_COUNT];
 117
 118 static void kvp_acquire_lock(int pool)
 119 {
 120         struct flock fl = {F_WRLCK, SEEK_SET, 0, 0, 0};
 121         fl.l_pid = getpid();
 122
 123         if (fcntl(kvp_file_info[pool].fd, F_SETLKW, &fl) == -1) {
 124                 syslog(LOG_ERR, "Failed to acquire the lock pool: %d", pool);
 125                 exit(EXIT_FAILURE);
 126         }
 127 }
 128
 129 static void kvp_release_lock(int pool)
 130 {
 131         struct flock fl = {F_UNLCK, SEEK_SET, 0, 0, 0};
 132         fl.l_pid = getpid();
 133
 134         if (fcntl(kvp_file_info[pool].fd, F_SETLK, &fl) == -1) {
 135                 perror("fcntl");
 136                 syslog(LOG_ERR, "Failed to release the lock pool: %d", pool);
 137                 exit(EXIT_FAILURE);
 138         }
 139 }
 140
 141 static void kvp_update_file(int pool)
 142 {
 143         FILE *filep;
 144         size_t bytes_written;
 145
 146         /*
 147          * We are going to write our in-memory registry out to
 148          * disk; acquire the lock first.
 149          */
 150         kvp_acquire_lock(pool);
 151
 152         filep = fopen(kvp_file_info[pool].fname, "w");
 153         if (!filep) {
 154                 kvp_release_lock(pool);
 155                 syslog(LOG_ERR, "Failed to open file, pool: %d", pool);
 156                 exit(EXIT_FAILURE);
 157         }
 158
 159         bytes_written = fwrite(kvp_file_info[pool].records,
 160                                 sizeof(struct kvp_record),
 161                                 kvp_file_info[pool].num_records, filep);
 162
 163         fclose(filep);
 164         kvp_release_lock(pool);
 165 }
 166
 167 static void kvp_update_mem_state(int pool)
 168 {
 169         FILE *filep;
 170         size_t records_read = 0;
 171         struct kvp_record *record = kvp_file_info[pool].records;
 172         struct kvp_record *readp;
 173         int num_blocks = kvp_file_info[pool].num_blocks;
 174         int alloc_unit = sizeof(struct kvp_record) * ENTRIES_PER_BLOCK;
 175
 176         kvp_acquire_lock(pool);
 177
 178         filep = fopen(kvp_file_info[pool].fname, "r");
 179         if (!filep) {
 180                 kvp_release_lock(pool);
 181                 syslog(LOG_ERR, "Failed to open file, pool: %d", pool);
 182                 exit(EXIT_FAILURE);
 183         }
 184         while (!feof(filep)) {
 185                 readp = &record[records_read];
 186                 records_read += fread(readp, sizeof(struct kvp_record),
 187                                         ENTRIES_PER_BLOCK * num_blocks,
 188                                         filep);
 189
 190                 if (!feof(filep)) {
 191                         /*
 192                          * We have more data to read.
 193                          */
 194                         num_blocks++;
 195                         record = realloc(record, alloc_unit * num_blocks);
 196
 197                         if (record == NULL) {
 198                                 syslog(LOG_ERR, "malloc failed");
 199                                 exit(EXIT_FAILURE);
 200                         }
 201                         continue;
 202                 }
 203                 break;
 204         }
 205
 206         kvp_file_info[pool].num_blocks = num_blocks;
 207         kvp_file_info[pool].records = record;
 208         kvp_file_info[pool].num_records = records_read;
 209
 210         fclose(filep);
 211         kvp_release_lock(pool);
 212 }
 213 static int kvp_file_init(void)
 214 {
 215         int  fd;
 216         FILE *filep;
 217         size_t records_read;
 218         char *fname;
 219         struct kvp_record *record;
 220         struct kvp_record *readp;
 221         int num_blocks;
 222         int i;
 223         int alloc_unit = sizeof(struct kvp_record) * ENTRIES_PER_BLOCK;
 224
 225         if (access("/var/opt/hyperv", F_OK)) {
 226                 if (mkdir("/var/opt/hyperv", S_IRUSR | S_IWUSR | S_IROTH)) {
 227                         syslog(LOG_ERR, " Failed to create /var/opt/hyperv");
 228                         exit(EXIT_FAILURE);
 229                 }
 230         }
 231
 232         for (i = 0; i < KVP_POOL_COUNT; i++) {
 233                 fname = kvp_file_info[i].fname;
 234                 records_read = 0;
 235                 num_blocks = 1;
 236                 sprintf(fname, "/var/opt/hyperv/.kvp_pool_%d", i);
 237                 fd = open(fname, O_RDWR | O_CREAT, S_IRUSR | S_IWUSR | S_IROTH);
 238
 239                 if (fd == -1)
 240                         return 1;
 241
 242
 243                 filep = fopen(fname, "r");
 244                 if (!filep)
 245                         return 1;
 246
 247                 record = malloc(alloc_unit * num_blocks);
 248                 if (record == NULL) {
 249                         fclose(filep);
 250                         return 1;
 251                 }
 252                 while (!feof(filep)) {
 253                         readp = &record[records_read];
 254                         records_read += fread(readp, sizeof(struct kvp_record),
 255                                         ENTRIES_PER_BLOCK,
 256                                         filep);
 257
 258                         if (!feof(filep)) {
 259                                 /*
 260                                  * We have more data to read.
 261                                  */
 262                                 num_blocks++;
 263                                 record = realloc(record, alloc_unit *
 264                                                 num_blocks);
 265                                 if (record == NULL) {
 266                                         fclose(filep);
 267                                         return 1;
 268                                 }
 269                                 continue;
 270                         }
 271                         break;
 272                 }
 273                 kvp_file_info[i].fd = fd;
 274                 kvp_file_info[i].num_blocks = num_blocks;
 275                 kvp_file_info[i].records = record;
 276                 kvp_file_info[i].num_records = records_read;
 277                 fclose(filep);
 278
 279         }
 280
 281         return 0;
 282 }
 283
 284 static int kvp_key_delete(int pool, __u8 *key, int key_size)
 285 {
 286         int i;
 287         int j, k;
 288         int num_records;
 289         struct kvp_record *record;
 290
 291         /*
 292          * First update the in-memory state.
 293          */
 294         kvp_update_mem_state(pool);
 295
 296         num_records = kvp_file_info[pool].num_records;
 297         record = kvp_file_info[pool].records;
 298
 299         for (i = 0; i < num_records; i++) {
 300                 if (memcmp(key, record[i].key, key_size))
 301                         continue;
 302                 /*
 303                  * Found a match; just move the remaining
 304                  * entries up.
 305                  */
 306                 if (i == num_records) {
 307                         kvp_file_info[pool].num_records--;
 308                         kvp_update_file(pool);
 309                         return 0;
 310                 }
 311
 312                 j = i;
 313                 k = j + 1;
 314                 for (; k < num_records; k++) {
 315                         strcpy(record[j].key, record[k].key);
 316                         strcpy(record[j].value, record[k].value);
 317                         j++;
 318                 }
 319
 320                 kvp_file_info[pool].num_records--;
 321                 kvp_update_file(pool);
 322                 return 0;
 323         }
 324         return 1;
 325 }
 326
 327 static int kvp_key_add_or_modify(int pool, __u8 *key, int key_size, __u8 *value,
 328                         int value_size)
 329 {
 330         int i;
 331         int num_records;
 332         struct kvp_record *record;
 333         int num_blocks;
 334
 335         if ((key_size > HV_KVP_EXCHANGE_MAX_KEY_SIZE) ||
 336                 (value_size > HV_KVP_EXCHANGE_MAX_VALUE_SIZE))
 337                 return 1;
 338
 339         /*
 340          * First update the in-memory state.
 341          */
 342         kvp_update_mem_state(pool);
 343
 344         num_records = kvp_file_info[pool].num_records;
 345         record = kvp_file_info[pool].records;
 346         num_blocks = kvp_file_info[pool].num_blocks;
 347
 348         for (i = 0; i < num_records; i++) {
 349                 if (memcmp(key, record[i].key, key_size))
 350                         continue;
 351                 /*
 352                  * Found a match; just update the value -
 353                  * this is the modify case.
 354                  */
 355                 memcpy(record[i].value, value, value_size);
 356                 kvp_update_file(pool);
 357                 return 0;
 358         }
 359
 360         /*
 361          * Need to add a new entry;
 362          */
 363         if (num_records == (ENTRIES_PER_BLOCK * num_blocks)) {
 364                 /* Need to allocate a larger array for reg entries. */
 365                 record = realloc(record, sizeof(struct kvp_record) *
 366                          ENTRIES_PER_BLOCK * (num_blocks + 1));
 367
 368                 if (record == NULL)
 369                         return 1;
 370                 kvp_file_info[pool].num_blocks++;
 371
 372         }
 373         memcpy(record[i].value, value, value_size);
 374         memcpy(record[i].key, key, key_size);
 375         kvp_file_info[pool].records = record;
 376         kvp_file_info[pool].num_records++;
 377         kvp_update_file(pool);
 378         return 0;
 379 }
 380
 381 static int kvp_get_value(int pool, __u8 *key, int key_size, __u8 *value,
 382                         int value_size)
 383 {
 384         int i;
 385         int num_records;
 386         struct kvp_record *record;
 387
 388         if ((key_size > HV_KVP_EXCHANGE_MAX_KEY_SIZE) ||
 389                 (value_size > HV_KVP_EXCHANGE_MAX_VALUE_SIZE))
 390                 return 1;
 391
 392         /*
 393          * First update the in-memory state.
 394          */
 395         kvp_update_mem_state(pool);
 396
 397         num_records = kvp_file_info[pool].num_records;
 398         record = kvp_file_info[pool].records;
 399
 400         for (i = 0; i < num_records; i++) {
 401                 if (memcmp(key, record[i].key, key_size))
 402                         continue;
 403                 /*
 404                  * Found a match; just copy the value out.
 405                  */
 406                 memcpy(value, record[i].value, value_size);
 407                 return 0;
 408         }
 409
 410         return 1;
 411 }
 412
 413 static int kvp_pool_enumerate(int pool, int index, __u8 *key, int key_size,
 414                                 __u8 *value, int value_size)
 415 {
 416         struct kvp_record *record;
 417
 418         /*
 419          * First update our in-memory database.
 420          */
 421         kvp_update_mem_state(pool);
 422         record = kvp_file_info[pool].records;
 423
 424         if (index >= kvp_file_info[pool].num_records) {
 425                 return 1;
 426         }
 427
 428         memcpy(key, record[index].key, key_size);
 429         memcpy(value, record[index].value, value_size);
 430         return 0;
 431 }
 432
 433
 434 void kvp_get_os_info(void)
 435 {
 436         FILE    *file;
 437         char    *p, buf[512];
 438
 439         uname(&uts_buf);
 440         os_build = uts_buf.release;
 441         processor_arch = uts_buf.machine;
 442
 443         /*
 444          * The current windows host (win7) expects the build
 445          * string to be of the form: x.y.z
 446          * Strip additional information we may have.
 447          */
 448         p = strchr(os_build, '-');
 449         if (p)
 450                 *p = '\0';
 451
 452         file = fopen("/etc/SuSE-release", "r");
 453         if (file != NULL)
 454                 goto kvp_osinfo_found;
 455         file  = fopen("/etc/redhat-release", "r");
 456         if (file != NULL)
 457                 goto kvp_osinfo_found;
 458         /*
 459          * Add code for other supported platforms.
 460          */
 461
 462         /*
 463          * We don't have information about the os.
 464          */
 465         os_name = uts_buf.sysname;
 466         return;
 467
 468 kvp_osinfo_found:
 469         /* up to three lines */
 470         p = fgets(buf, sizeof(buf), file);
 471         if (p) {
 472                 p = strchr(buf, '\n');
 473                 if (p)
 474                         *p = '\0';
 475                 p = strdup(buf);
 476                 if (!p)
 477                         goto done;
 478                 os_name = p;
 479
 480                 /* second line */
 481                 p = fgets(buf, sizeof(buf), file);
 482                 if (p) {
 483                         p = strchr(buf, '\n');
 484                         if (p)
 485                                 *p = '\0';
 486                         p = strdup(buf);
 487                         if (!p)
 488                                 goto done;
 489                         os_major = p;
 490
 491                         /* third line */
 492                         p = fgets(buf, sizeof(buf), file);
 493                         if (p)  {
 494                                 p = strchr(buf, '\n');
 495                                 if (p)
 496                                         *p = '\0';
 497                                 p = strdup(buf);
 498                                 if (p)
 499                                         os_minor = p;
 500                         }
 501                 }
 502         }
 503
 504 done:
 505         fclose(file);
 506         return;
 507 }
 508
 509
 510
 511 /*
 512  * Retrieve an interface name corresponding to the specified guid.
 513  * If there is a match, the function returns a pointer
 514  * to the interface name and if not, a NULL is returned.
 515  * If a match is found, the caller is responsible for
 516  * freeing the memory.
 517  */
 518
 519 static char *kvp_get_if_name(char *guid)
 520 {
 521         DIR *dir;
 522         struct dirent *entry;
 523         FILE    *file;
 524         char    *p, *q, *x;
 525         char    *if_name = NULL;
 526         char    buf[256];
 527         char *kvp_net_dir = "/sys/class/net/";
 528         char dev_id[256];
 529
 530         dir = opendir(kvp_net_dir);
 531         if (dir == NULL)
 532                 return NULL;
 533
 534         snprintf(dev_id, sizeof(dev_id), "%s", kvp_net_dir);
 535         q = dev_id + strlen(kvp_net_dir);
 536
 537         while ((entry = readdir(dir)) != NULL) {
 538                 /*
 539                  * Set the state for the next pass.
 540                  */
 541                 *q = '\0';
 542                 strcat(dev_id, entry->d_name);
 543                 strcat(dev_id, "/device/device_id");
 544
 545                 file = fopen(dev_id, "r");
 546                 if (file == NULL)
 547                         continue;
 548
 549                 p = fgets(buf, sizeof(buf), file);
 550                 if (p) {
 551                         x = strchr(p, '\n');
 552                         if (x)
 553                                 *x = '\0';
 554
 555                         if (!strcmp(p, guid)) {
 556                                 /*
 557                                  * Found the guid match; return the interface
 558                                  * name. The caller will free the memory.
 559                                  */
 560                                 if_name = strdup(entry->d_name);
 561                                 fclose(file);
 562                                 break;
 563                         }
 564                 }
 565                 fclose(file);
 566         }
 567
 568         closedir(dir);
 569         return if_name;
 570 }
 571
 572 /*
 573  * Retrieve the MAC address given the interface name.
 574  */
 575
 576 static char *kvp_if_name_to_mac(char *if_name)
 577 {
 578         FILE    *file;
 579         char    *p, *x;
 580         char    buf[256];
 581         char addr_file[256];
 582         int i;
 583         char *mac_addr = NULL;
 584
 585         snprintf(addr_file, sizeof(addr_file), "%s%s%s", "/sys/class/net/",
 586                 if_name, "/address");
 587
 588         file = fopen(addr_file, "r");
 589         if (file == NULL)
 590                 return NULL;
 591
 592         p = fgets(buf, sizeof(buf), file);
 593         if (p) {
 594                 x = strchr(p, '\n');
 595                 if (x)
 596                         *x = '\0';
 597                 for (i = 0; i < strlen(p); i++)
 598                         p[i] = toupper(p[i]);
 599                 mac_addr = strdup(p);
 600         }
 601
 602         fclose(file);
 603         return mac_addr;
 604 }
 605
 606
 607 /*
 608  * Retrieve the interface name given tha MAC address.
 609  */
 610
 611 static char *kvp_mac_to_if_name(char *mac)
 612 {
 613         DIR *dir;
 614         struct dirent *entry;
 615         FILE    *file;
 616         char    *p, *q, *x;
 617         char    *if_name = NULL;
 618         char    buf[256];
 619         char *kvp_net_dir = "/sys/class/net/";
 620         char dev_id[256];
 621         int i;
 622
 623         dir = opendir(kvp_net_dir);
 624         if (dir == NULL)
 625                 return NULL;
 626
 627         snprintf(dev_id, sizeof(dev_id), kvp_net_dir);
 628         q = dev_id + strlen(kvp_net_dir);
 629
 630         while ((entry = readdir(dir)) != NULL) {
 631                 /*
 632                  * Set the state for the next pass.
 633                  */
 634                 *q = '\0';
 635
 636                 strcat(dev_id, entry->d_name);
 637                 strcat(dev_id, "/address");
 638
 639                 file = fopen(dev_id, "r");
 640                 if (file == NULL)
 641                         continue;
 642
 643                 p = fgets(buf, sizeof(buf), file);
 644                 if (p) {
 645                         x = strchr(p, '\n');
 646                         if (x)
 647                                 *x = '\0';
 648
 649                         for (i = 0; i < strlen(p); i++)
 650                                 p[i] = toupper(p[i]);
 651
 652                         if (!strcmp(p, mac)) {
 653                                 /*
 654                                  * Found the MAC match; return the interface
 655                                  * name. The caller will free the memory.
 656                                  */
 657                                 if_name = strdup(entry->d_name);
 658                                 fclose(file);
 659                                 break;
 660                         }
 661                 }
 662                 fclose(file);
 663         }
 664
 665         closedir(dir);
 666         return if_name;
 667 }
 668
 669
 670 static void kvp_process_ipconfig_file(char *cmd,
 671                                         char *config_buf, int len,
 672                                         int element_size, int offset)
 673 {
 674         char buf[256];
 675         char *p;
 676         char *x;
 677         FILE *file;
 678
 679         /*
 680          * First execute the command.
 681          */
 682         file = popen(cmd, "r");
 683         if (file == NULL)
 684                 return;
 685
 686         if (offset == 0)
 687                 memset(config_buf, 0, len);
 688         while ((p = fgets(buf, sizeof(buf), file)) != NULL) {
 689                 if ((len - strlen(config_buf)) < (element_size + 1))
 690                         break;
 691
 692                 x = strchr(p, '\n');
 693                 *x = '\0';
 694                 strcat(config_buf, p);
 695                 strcat(config_buf, ";");
 696         }
 697         pclose(file);
 698 }
 699
 700 static void kvp_get_ipconfig_info(char *if_name,
 701                                  struct hv_kvp_ipaddr_value *buffer)
 702 {
 703         char cmd[512];
 704         char dhcp_info[128];
 705         char *p;
 706         FILE *file;
 707
 708         /*
 709          * Get the address of default gateway (ipv4).
 710          */
 711         sprintf(cmd, "%s %s", "ip route show dev", if_name);
 712         strcat(cmd, " | awk '/default/ {print $3 }'");
 713
 714         /*
 715          * Execute the command to gather gateway info.
 716          */
 717         kvp_process_ipconfig_file(cmd, (char *)buffer->gate_way,
 718                                 (MAX_GATEWAY_SIZE * 2), INET_ADDRSTRLEN, 0);
 719
 720         /*
 721          * Get the address of default gateway (ipv6).
 722          */
 723         sprintf(cmd, "%s %s", "ip -f inet6  route show dev", if_name);
 724         strcat(cmd, " | awk '/default/ {print $3 }'");
 725
 726         /*
 727          * Execute the command to gather gateway info (ipv6).
 728          */
 729         kvp_process_ipconfig_file(cmd, (char *)buffer->gate_way,
 730                                 (MAX_GATEWAY_SIZE * 2), INET6_ADDRSTRLEN, 1);
 731
 732
 733         /*
 734          * Gather the DNS  state.
 735          * Since there is no standard way to get this information
 736          * across various distributions of interest; we just invoke
 737          * an external script that needs to be ported across distros
 738          * of interest.
 739          *
 740          * Following is the expected format of the information from the script:
 741          *
 742          * ipaddr1 (nameserver1)
 743          * ipaddr2 (nameserver2)
 744          * .
 745          * .
 746          */
 747
 748         sprintf(cmd, "%s",  "hv_get_dns_info");
 749
 750         /*
 751          * Execute the command to gather DNS info.
 752          */
 753         kvp_process_ipconfig_file(cmd, (char *)buffer->dns_addr,
 754                                 (MAX_IP_ADDR_SIZE * 2), INET_ADDRSTRLEN, 0);
 755
 756         /*
 757          * Gather the DHCP state.
 758          * We will gather this state by invoking an external script.
 759          * The parameter to the script is the interface name.
 760          * Here is the expected output:
 761          *
 762          * Enabled: DHCP enabled.
 763          */
 764
 765         sprintf(cmd, "%s %s", "hv_get_dhcp_info", if_name);
 766
 767         file = popen(cmd, "r");
 768         if (file == NULL)
 769                 return;
 770
 771         p = fgets(dhcp_info, sizeof(dhcp_info), file);
 772         if (p == NULL) {
 773                 pclose(file);
 774                 return;
 775         }
 776
 777         if (!strncmp(p, "Enabled", 7))
 778                 buffer->dhcp_enabled = 1;
 779         else
 780                 buffer->dhcp_enabled = 0;
 781
 782         pclose(file);
 783 }
 784
 785
 786 static unsigned int hweight32(unsigned int *w)
 787 {
 788         unsigned int res = *w - ((*w >> 1) & 0x55555555);
 789         res = (res & 0x33333333) + ((res >> 2) & 0x33333333);
 790         res = (res + (res >> 4)) & 0x0F0F0F0F;
 791         res = res + (res >> 8);
 792         return (res + (res >> 16)) & 0x000000FF;
 793 }
 794
 795 static int kvp_process_ip_address(void *addrp,
 796                                 int family, char *buffer,
 797                                 int length,  int *offset)
 798 {
 799         struct sockaddr_in *addr;
 800         struct sockaddr_in6 *addr6;
 801         int addr_length;
 802         char tmp[50];
 803         const char *str;
 804
 805         if (family == AF_INET) {
 806                 addr = (struct sockaddr_in *)addrp;
 807                 str = inet_ntop(family, &addr->sin_addr, tmp, 50);
 808                 addr_length = INET_ADDRSTRLEN;
 809         } else {
 810                 addr6 = (struct sockaddr_in6 *)addrp;
 811                 str = inet_ntop(family, &addr6->sin6_addr.s6_addr, tmp, 50);
 812                 addr_length = INET6_ADDRSTRLEN;
 813         }
 814
 815         if ((length - *offset) < addr_length + 1)
 816                 return HV_E_FAIL;
 817         if (str == NULL) {
 818                 strcpy(buffer, "inet_ntop failed\n");
 819                 return HV_E_FAIL;
 820         }
 821         if (*offset == 0)
 822                 strcpy(buffer, tmp);
 823         else
 824                 strcat(buffer, tmp);
 825         strcat(buffer, ";");
 826
 827         *offset += strlen(str) + 1;
 828         return 0;
 829 }
 830
 831 static int
 832 kvp_get_ip_info(int family, char *if_name, int op,
 833                  void  *out_buffer, int length)
 834 {
 835         struct ifaddrs *ifap;
 836         struct ifaddrs *curp;
 837         int offset = 0;
 838         int sn_offset = 0;
 839         int error = 0;
 840         char *buffer;
 841         struct hv_kvp_ipaddr_value *ip_buffer;
 842         char cidr_mask[5]; /* /xyz */
 843         int weight;
 844         int i;
 845         unsigned int *w;
 846         char *sn_str;
 847         struct sockaddr_in6 *addr6;
 848
 849         if (op == KVP_OP_ENUMERATE) {
 850                 buffer = out_buffer;
 851         } else {
 852                 ip_buffer = out_buffer;
 853                 buffer = (char *)ip_buffer->ip_addr;
 854                 ip_buffer->addr_family = 0;
 855         }
 856         /*
 857          * On entry into this function, the buffer is capable of holding the
 858          * maximum key value.
 859          */
 860
 861         if (getifaddrs(&ifap)) {
 862                 strcpy(buffer, "getifaddrs failed\n");
 863                 return HV_E_FAIL;
 864         }
 865
 866         curp = ifap;
 867         while (curp != NULL) {
 868                 if (curp->ifa_addr == NULL) {
 869                         curp = curp->ifa_next;
 870                         continue;
 871                 }
 872
 873                 if ((if_name != NULL) &&
 874                         (strncmp(curp->ifa_name, if_name, strlen(if_name)))) {
 875                         /*
 876                          * We want info about a specific interface;
 877                          * just continue.
 878                          */
 879                         curp = curp->ifa_next;
 880                         continue;
 881                 }
 882
 883                 /*
 884                  * We only support two address families: AF_INET and AF_INET6.
 885                  * If a family value of 0 is specified, we collect both
 886                  * supported address families; if not we gather info on
 887                  * the specified address family.
 888                  */
 889                 if ((family != 0) && (curp->ifa_addr->sa_family != family)) {
 890                         curp = curp->ifa_next;
 891                         continue;
 892                 }
 893                 if ((curp->ifa_addr->sa_family != AF_INET) &&
 894                         (curp->ifa_addr->sa_family != AF_INET6)) {
 895                         curp = curp->ifa_next;
 896                         continue;
 897                 }
 898
 899                 if (op == KVP_OP_GET_IP_INFO) {
 900                         /*
 901                          * Gather info other than the IP address.
 902                          * IP address info will be gathered later.
 903                          */
 904                         if (curp->ifa_addr->sa_family == AF_INET) {
 905                                 ip_buffer->addr_family |= ADDR_FAMILY_IPV4;
 906                                 /*
 907                                  * Get subnet info.
 908                                  */
 909                                 error = kvp_process_ip_address(
 910                                                              curp->ifa_netmask,
 911                                                              AF_INET,
 912                                                              (char *)
 913                                                              ip_buffer->sub_net,
 914                                                              length,
 915                                                              &sn_offset);
 916                                 if (error)
 917                                         goto gather_ipaddr;
 918                         } else {
 919                                 ip_buffer->addr_family |= ADDR_FAMILY_IPV6;
 920
 921                                 /*
 922                                  * Get subnet info in CIDR format.
 923                                  */
 924                                 weight = 0;
 925                                 sn_str = (char *)ip_buffer->sub_net;
 926                                 addr6 = (struct sockaddr_in6 *)
 927                                         curp->ifa_netmask;
 928                                 w = addr6->sin6_addr.s6_addr32;
 929
 930                                 for (i = 0; i < 4; i++)
 931                                         weight += hweight32(&w[i]);
 932
 933                                 sprintf(cidr_mask, "/%d", weight);
 934                                 if ((length - sn_offset) <
 935                                         (strlen(cidr_mask) + 1))
 936                                         goto gather_ipaddr;
 937
 938                                 if (sn_offset == 0)
 939                                         strcpy(sn_str, cidr_mask);
 940                                 else
 941                                         strcat(sn_str, cidr_mask);
 942                                 strcat((char *)ip_buffer->sub_net, ";");
 943                                 sn_offset += strlen(sn_str) + 1;
 944                         }
 945
 946                         /*
 947                          * Collect other ip related configuration info.
 948                          */
 949
 950                         kvp_get_ipconfig_info(if_name, ip_buffer);
 951                 }
 952
 953 gather_ipaddr:
 954                 error = kvp_process_ip_address(curp->ifa_addr,
 955                                                 curp->ifa_addr->sa_family,
 956                                                 buffer,
 957                                                 length, &offset);
 958                 if (error)
 959                         goto getaddr_done;
 960
 961                 curp = curp->ifa_next;
 962         }
 963
 964 getaddr_done:
 965         freeifaddrs(ifap);
 966         return error;
 967 }
 968
 969
 970 static int expand_ipv6(char *addr, int type)
 971 {
 972         int ret;
 973         struct in6_addr v6_addr;
 974
 975         ret = inet_pton(AF_INET6, addr, &v6_addr);
 976
 977         if (ret != 1) {
 978                 if (type == NETMASK)
 979                         return 1;
 980                 return 0;
 981         }
 982
 983         sprintf(addr, "%02x%02x:%02x%02x:%02x%02x:%02x%02x:%02x%02x:"
 984                 "%02x%02x:%02x%02x:%02x%02x",
 985                 (int)v6_addr.s6_addr[0], (int)v6_addr.s6_addr[1],
 986                 (int)v6_addr.s6_addr[2], (int)v6_addr.s6_addr[3],
 987                 (int)v6_addr.s6_addr[4], (int)v6_addr.s6_addr[5],
 988                 (int)v6_addr.s6_addr[6], (int)v6_addr.s6_addr[7],
 989                 (int)v6_addr.s6_addr[8], (int)v6_addr.s6_addr[9],
 990                 (int)v6_addr.s6_addr[10], (int)v6_addr.s6_addr[11],
 991                 (int)v6_addr.s6_addr[12], (int)v6_addr.s6_addr[13],
 992                 (int)v6_addr.s6_addr[14], (int)v6_addr.s6_addr[15]);
 993
 994         return 1;
 995
 996 }
 997
 998 static int is_ipv4(char *addr)
 999 {
1000         int ret;
1001         struct in_addr ipv4_addr;
1002
1003         ret = inet_pton(AF_INET, addr, &ipv4_addr);
1004
1005         if (ret == 1)
1006                 return 1;
1007         return 0;
1008 }
1009
1010 static int parse_ip_val_buffer(char *in_buf, int *offset,
1011                                 char *out_buf, int out_len)
1012 {
1013         char *x;
1014         char *start;
1015
1016         /*
1017          * in_buf has sequence of characters that are seperated by
1018          * the character ';'. The last sequence does not have the
1019          * terminating ";" character.
1020          */
1021         start = in_buf + *offset;
1022
1023         x = strchr(start, ';');
1024         if (x)
1025                 *x = 0;
1026         else
1027                 x = start + strlen(start);
1028
1029         if (strlen(start) != 0) {
1030                 int i = 0;
1031                 /*
1032                  * Get rid of leading spaces.
1033                  */
1034                 while (start[i] == ' ')
1035                         i++;
1036
1037                 if ((x - start) <= out_len) {
1038                         strcpy(out_buf, (start + i));
1039                         *offset += (x - start) + 1;
1040                         return 1;
1041                 }
1042         }
1043         return 0;
1044 }
1045
1046 static int kvp_write_file(FILE *f, char *s1, char *s2, char *s3)
1047 {
1048         int ret;
1049
1050         ret = fprintf(f, "%s%s%s%s\n", s1, s2, "=", s3);
1051
1052         if (ret < 0)
1053                 return HV_E_FAIL;
1054
1055         return 0;
1056 }
1057
1058
1059 static int process_ip_string(FILE *f, char *ip_string, int type)
1060 {
1061         int error = 0;
1062         char addr[INET6_ADDRSTRLEN];
1063         int i = 0;
1064         int j = 0;
1065         char str[256];
1066         char sub_str[10];
1067         int offset = 0;
1068
1069         memset(addr, 0, sizeof(addr));
1070
1071         while (parse_ip_val_buffer(ip_string, &offset, addr,
1072                                         (MAX_IP_ADDR_SIZE * 2))) {
1073
1074                 sub_str[0] = 0;
1075                 if (is_ipv4(addr)) {
1076                         switch (type) {
1077                         case IPADDR:
1078                                 snprintf(str, sizeof(str), "%s", "IPADDR");
1079                                 break;
1080                         case NETMASK:
1081                                 snprintf(str, sizeof(str), "%s", "NETMASK");
1082                                 break;
1083                         case GATEWAY:
1084                                 snprintf(str, sizeof(str), "%s", "GATEWAY");
1085                                 break;
1086                         case DNS:
1087                                 snprintf(str, sizeof(str), "%s", "DNS");
1088                                 break;
1089                         }
1090                         if (i != 0) {
1091                                 if (type != DNS) {
1092                                         snprintf(sub_str, sizeof(sub_str),
1093                                                 "_%d", i++);
1094                                 } else {
1095                                         snprintf(sub_str, sizeof(sub_str),
1096                                                 "%d", ++i);
1097                                 }
1098                         } else if (type == DNS) {
1099                                 snprintf(sub_str, sizeof(sub_str), "%d", ++i);
1100                         }
1101
1102
1103                 } else if (expand_ipv6(addr, type)) {
1104                         switch (type) {
1105                         case IPADDR:
1106                                 snprintf(str, sizeof(str), "%s", "IPV6ADDR");
1107                                 break;
1108                         case NETMASK:
1109                                 snprintf(str, sizeof(str), "%s", "IPV6NETMASK");
1110                                 break;
1111                         case GATEWAY:
1112                                 snprintf(str, sizeof(str), "%s",
1113                                         "IPV6_DEFAULTGW");
1114                                 break;
1115                         case DNS:
1116                                 snprintf(str, sizeof(str), "%s",  "DNS");
1117                                 break;
1118                         }
1119                         if ((j != 0) || (type == DNS)) {
1120                                 if (type != DNS) {
1121                                         snprintf(sub_str, sizeof(sub_str),
1122                                                 "_%d", j++);
1123                                 } else {
1124                                         snprintf(sub_str, sizeof(sub_str),
1125                                                 "%d", ++i);
1126                                 }
1127                         } else if (type == DNS) {
1128                                 snprintf(sub_str, sizeof(sub_str),
1129                                         "%d", ++i);
1130                         }
1131                 } else {
1132                         return  HV_INVALIDARG;
1133                 }
1134
1135                 error = kvp_write_file(f, str, sub_str, addr);
1136                 if (error)
1137                         return error;
1138                 memset(addr, 0, sizeof(addr));
1139         }
1140
1141         return 0;
1142 }
1143
1144 static int kvp_set_ip_info(char *if_name, struct hv_kvp_ipaddr_value *new_val)
1145 {
1146         int error = 0;
1147         char if_file[128];
1148         FILE *file;
1149         char cmd[512];
1150         char *mac_addr;
1151
1152         /*
1153          * Set the configuration for the specified interface with
1154          * the information provided. Since there is no standard
1155          * way to configure an interface, we will have an external
1156          * script that does the job of configuring the interface and
1157          * flushing the configuration.
1158          *
1159          * The parameters passed to this external script are:
1160          * 1. A configuration file that has the specified configuration.
1161          *
1162          * We will embed the name of the interface in the configuration
1163          * file: ifcfg-ethx (where ethx is the interface name).
1164          *
1165          * The information provided here may be more than what is needed
1166          * in a given distro to configure the interface and so are free
1167          * ignore information that may not be relevant.
1168          *
1169          * Here is the format of the ip configuration file:
1170          *
1171          * HWADDR=macaddr
1172          * IF_NAME=interface name
1173          * DHCP=yes (This is optional; if yes, DHCP is configured)
1174          *
1175          * IPADDR=ipaddr1
1176          * IPADDR_1=ipaddr2
1177          * IPADDR_x=ipaddry (where y = x + 1)
1178          *
1179          * NETMASK=netmask1
1180          * NETMASK_x=netmasky (where y = x + 1)
1181          *
1182          * GATEWAY=ipaddr1
1183          * GATEWAY_x=ipaddry (where y = x + 1)
1184          *
1185          * DNSx=ipaddrx (where first DNS address is tagged as DNS1 etc)
1186          *
1187          * IPV6 addresses will be tagged as IPV6ADDR, IPV6 gateway will be
1188          * tagged as IPV6_DEFAULTGW and IPV6 NETMASK will be tagged as
1189          * IPV6NETMASK.
1190          *
1191          * The host can specify multiple ipv4 and ipv6 addresses to be
1192          * configured for the interface. Furthermore, the configuration
1193          * needs to be persistent. A subsequent GET call on the interface
1194          * is expected to return the configuration that is set via the SET
1195          * call.
1196          */
1197
1198         snprintf(if_file, sizeof(if_file), "%s%s%s", KVP_CONFIG_LOC,
1199                 "hyperv/ifcfg-", if_name);
1200
1201         file = fopen(if_file, "w");
1202
1203         if (file == NULL) {
1204                 syslog(LOG_ERR, "Failed to open config file");
1205                 return HV_E_FAIL;
1206         }
1207
1208         /*
1209          * First write out the MAC address.
1210          */
1211
1212         mac_addr = kvp_if_name_to_mac(if_name);
1213         if (mac_addr == NULL) {
1214                 error = HV_E_FAIL;
1215                 goto setval_error;
1216         }
1217
1218         error = kvp_write_file(file, "HWADDR", "", mac_addr);
1219         if (error)
1220                 goto setval_error;
1221
1222         error = kvp_write_file(file, "IF_NAME", "", if_name);
1223         if (error)
1224                 goto setval_error;
1225
1226         if (new_val->dhcp_enabled) {
1227                 error = kvp_write_file(file, "DHCP", "", "yes");
1228                 if (error)
1229                         goto setval_error;
1230
1231                 /*
1232                  * We are done!.
1233                  */
1234                 goto setval_done;
1235         }
1236
1237         /*
1238          * Write the configuration for ipaddress, netmask, gateway and
1239          * name servers.
1240          */
1241
1242         error = process_ip_string(file, (char *)new_val->ip_addr, IPADDR);
1243         if (error)
1244                 goto setval_error;
1245
1246         error = process_ip_string(file, (char *)new_val->sub_net, NETMASK);
1247         if (error)
1248                 goto setval_error;
1249
1250         error = process_ip_string(file, (char *)new_val->gate_way, GATEWAY);
1251         if (error)
1252                 goto setval_error;
1253
1254         error = process_ip_string(file, (char *)new_val->dns_addr, DNS);
1255         if (error)
1256                 goto setval_error;
1257
1258 setval_done:
1259         free(mac_addr);
1260         fclose(file);
1261
1262         /*
1263          * Now that we have populated the configuration file,
1264          * invoke the external script to do its magic.
1265          */
1266
1267         snprintf(cmd, sizeof(cmd), "%s %s", "hv_set_ifconfig", if_file);
1268         system(cmd);
1269         return 0;
1270
1271 setval_error:
1272         syslog(LOG_ERR, "Failed to write config file");
1273         free(mac_addr);
1274         fclose(file);
1275         return error;
1276 }
1277
1278
1279 static int
1280 kvp_get_domain_name(char *buffer, int length)
1281 {
1282         struct addrinfo hints, *info ;
1283         int error = 0;
1284
1285         gethostname(buffer, length);
1286         memset(&hints, 0, sizeof(hints));
1287         hints.ai_family = AF_INET; /*Get only ipv4 addrinfo. */
1288         hints.ai_socktype = SOCK_STREAM;
1289         hints.ai_flags = AI_CANONNAME;
1290
1291         error = getaddrinfo(buffer, NULL, &hints, &info);
1292         if (error != 0) {
1293                 strcpy(buffer, "getaddrinfo failed\n");
1294                 return error;
1295         }
1296         strcpy(buffer, info->ai_canonname);
1297         freeaddrinfo(info);
1298         return error;
1299 }
1300
1301 static int
1302 netlink_send(int fd, struct cn_msg *msg)
1303 {
1304         struct nlmsghdr *nlh;
1305         unsigned int size;
1306         struct msghdr message;
1307         char buffer[64];
1308         struct iovec iov[2];
1309
1310         size = NLMSG_SPACE(sizeof(struct cn_msg) + msg->len);
1311
1312         nlh = (struct nlmsghdr *)buffer;
1313         nlh->nlmsg_seq = 0;
1314         nlh->nlmsg_pid = getpid();
1315         nlh->nlmsg_type = NLMSG_DONE;
1316         nlh->nlmsg_len = NLMSG_LENGTH(size - sizeof(*nlh));
1317         nlh->nlmsg_flags = 0;
1318
1319         iov[0].iov_base = nlh;
1320         iov[0].iov_len = sizeof(*nlh);
1321
1322         iov[1].iov_base = msg;
1323         iov[1].iov_len = size;
1324
1325         memset(&message, 0, sizeof(message));
1326         message.msg_name = &addr;
1327         message.msg_namelen = sizeof(addr);
1328         message.msg_iov = iov;
1329         message.msg_iovlen = 2;
1330
1331         return sendmsg(fd, &message, 0);
1332 }
1333
1334 int main(void)
1335 {
1336         int fd, len, sock_opt;
1337         int error;
1338         struct cn_msg *message;
1339         struct pollfd pfd;
1340         struct nlmsghdr *incoming_msg;
1341         struct cn_msg   *incoming_cn_msg;
1342         struct hv_kvp_msg *hv_msg;
1343         char    *p;
1344         char    *key_value;
1345         char    *key_name;
1346         int     op;
1347         int     pool;
1348         char    *if_name;
1349         struct hv_kvp_ipaddr_value *kvp_ip_val;
1350
1351         daemon(1, 0);
1352         openlog("KVP", 0, LOG_USER);
1353         syslog(LOG_INFO, "KVP starting; pid is:%d", getpid());
1354         /*
1355          * Retrieve OS release information.
1356          */
1357         kvp_get_os_info();
1358
1359         if (kvp_file_init()) {
1360                 syslog(LOG_ERR, "Failed to initialize the pools");
1361                 exit(EXIT_FAILURE);
1362         }
1363
1364         fd = socket(AF_NETLINK, SOCK_DGRAM, NETLINK_CONNECTOR);
1365         if (fd < 0) {
1366                 syslog(LOG_ERR, "netlink socket creation failed; error:%d", fd);
1367                 exit(EXIT_FAILURE);
1368         }
1369         addr.nl_family = AF_NETLINK;
1370         addr.nl_pad = 0;
1371         addr.nl_pid = 0;
1372         addr.nl_groups = CN_KVP_IDX;
1373
1374
1375         error = bind(fd, (struct sockaddr *)&addr, sizeof(addr));
1376         if (error < 0) {
1377                 syslog(LOG_ERR, "bind failed; error:%d", error);
1378                 close(fd);
1379                 exit(EXIT_FAILURE);
1380         }
1381         sock_opt = addr.nl_groups;
1382         setsockopt(fd, 270, 1, &sock_opt, sizeof(sock_opt));
1383         /*
1384          * Register ourselves with the kernel.
1385          */
1386         message = (struct cn_msg *)kvp_send_buffer;
1387         message->id.idx = CN_KVP_IDX;
1388         message->id.val = CN_KVP_VAL;
1389
1390         hv_msg = (struct hv_kvp_msg *)message->data;
1391         hv_msg->kvp_hdr.operation = KVP_OP_REGISTER1;
1392         message->ack = 0;
1393         message->len = sizeof(struct hv_kvp_msg);
1394
1395         len = netlink_send(fd, message);
1396         if (len < 0) {
1397                 syslog(LOG_ERR, "netlink_send failed; error:%d", len);
1398                 close(fd);
1399                 exit(EXIT_FAILURE);
1400         }
1401
1402         pfd.fd = fd;
1403
1404         while (1) {
1405                 struct sockaddr *addr_p = (struct sockaddr *) &addr;
1406                 socklen_t addr_l = sizeof(addr);
1407                 pfd.events = POLLIN;
1408                 pfd.revents = 0;
1409                 poll(&pfd, 1, -1);
1410
1411                 len = recvfrom(fd, kvp_recv_buffer, sizeof(kvp_recv_buffer), 0,
1412                                 addr_p, &addr_l);
1413
1414                 if (len < 0 || addr.nl_pid) {
1415                         syslog(LOG_ERR, "recvfrom failed; pid:%u error:%d %s",
1416                                         addr.nl_pid, errno, strerror(errno));
1417                         close(fd);
1418                         return -1;
1419                 }
1420
1421                 incoming_msg = (struct nlmsghdr *)kvp_recv_buffer;
1422                 incoming_cn_msg = (struct cn_msg *)NLMSG_DATA(incoming_msg);
1423                 hv_msg = (struct hv_kvp_msg *)incoming_cn_msg->data;
1424
1425                 /*
1426                  * We will use the KVP header information to pass back
1427                  * the error from this daemon. So, first copy the state
1428                  * and set the error code to success.
1429                  */
1430                 op = hv_msg->kvp_hdr.operation;
1431                 pool = hv_msg->kvp_hdr.pool;
1432                 hv_msg->error = HV_S_OK;
1433
1434                 if ((in_hand_shake) && (op == KVP_OP_REGISTER1)) {
1435                         /*
1436                          * Driver is registering with us; stash away the version
1437                          * information.
1438                          */
1439                         in_hand_shake = 0;
1440                         p = (char *)hv_msg->body.kvp_register.version;
1441                         lic_version = malloc(strlen(p) + 1);
1442                         if (lic_version) {
1443                                 strcpy(lic_version, p);
1444                                 syslog(LOG_INFO, "KVP LIC Version: %s",
1445                                         lic_version);
1446                         } else {
1447                                 syslog(LOG_ERR, "malloc failed");
1448                         }
1449                         continue;
1450                 }
1451
1452                 switch (op) {
1453                 case KVP_OP_GET_IP_INFO:
1454                         kvp_ip_val = &hv_msg->body.kvp_ip_val;
1455                         if_name =
1456                         kvp_mac_to_if_name((char *)kvp_ip_val->adapter_id);
1457
1458                         if (if_name == NULL) {
1459                                 /*
1460                                  * We could not map the mac address to an
1461                                  * interface name; return error.
1462                                  */
1463                                 hv_msg->error = HV_E_FAIL;
1464                                 break;
1465                         }
1466                         error = kvp_get_ip_info(
1467                                                 0, if_name, KVP_OP_GET_IP_INFO,
1468                                                 kvp_ip_val,
1469                                                 (MAX_IP_ADDR_SIZE * 2));
1470
1471                         if (error)
1472                                 hv_msg->error = error;
1473
1474                         free(if_name);
1475                         break;
1476
1477                 case KVP_OP_SET_IP_INFO:
1478                         kvp_ip_val = &hv_msg->body.kvp_ip_val;
1479                         if_name = kvp_get_if_name(
1480                                         (char *)kvp_ip_val->adapter_id);
1481                         if (if_name == NULL) {
1482                                 /*
1483                                  * We could not map the guid to an
1484                                  * interface name; return error.
1485                                  */
1486                                 hv_msg->error = HV_GUID_NOTFOUND;
1487                                 break;
1488                         }
1489                         error = kvp_set_ip_info(if_name, kvp_ip_val);
1490                         if (error)
1491                                 hv_msg->error = error;
1492
1493                         free(if_name);
1494                         break;
1495
1496                 case KVP_OP_SET:
1497                         if (kvp_key_add_or_modify(pool,
1498                                         hv_msg->body.kvp_set.data.key,
1499                                         hv_msg->body.kvp_set.data.key_size,
1500                                         hv_msg->body.kvp_set.data.value,
1501                                         hv_msg->body.kvp_set.data.value_size))
1502                                         hv_msg->error = HV_S_CONT;
1503                         break;
1504
1505                 case KVP_OP_GET:
1506                         if (kvp_get_value(pool,
1507                                         hv_msg->body.kvp_set.data.key,
1508                                         hv_msg->body.kvp_set.data.key_size,
1509                                         hv_msg->body.kvp_set.data.value,
1510                                         hv_msg->body.kvp_set.data.value_size))
1511                                         hv_msg->error = HV_S_CONT;
1512                         break;
1513
1514                 case KVP_OP_DELETE:
1515                         if (kvp_key_delete(pool,
1516                                         hv_msg->body.kvp_delete.key,
1517                                         hv_msg->body.kvp_delete.key_size))
1518                                         hv_msg->error = HV_S_CONT;
1519                         break;
1520
1521                 default:
1522                         break;
1523                 }
1524
1525                 if (op != KVP_OP_ENUMERATE)
1526                         goto kvp_done;
1527
1528                 /*
1529                  * If the pool is KVP_POOL_AUTO, dynamically generate
1530                  * both the key and the value; if not read from the
1531                  * appropriate pool.
1532                  */
1533                 if (pool != KVP_POOL_AUTO) {
1534                         if (kvp_pool_enumerate(pool,
1535                                         hv_msg->body.kvp_enum_data.index,
1536                                         hv_msg->body.kvp_enum_data.data.key,
1537                                         HV_KVP_EXCHANGE_MAX_KEY_SIZE,
1538                                         hv_msg->body.kvp_enum_data.data.value,
1539                                         HV_KVP_EXCHANGE_MAX_VALUE_SIZE))
1540                                         hv_msg->error = HV_S_CONT;
1541                         goto kvp_done;
1542                 }
1543
1544                 hv_msg = (struct hv_kvp_msg *)incoming_cn_msg->data;
1545                 key_name = (char *)hv_msg->body.kvp_enum_data.data.key;
1546                 key_value = (char *)hv_msg->body.kvp_enum_data.data.value;
1547
1548                 switch (hv_msg->body.kvp_enum_data.index) {
1549                 case FullyQualifiedDomainName:
1550                         kvp_get_domain_name(key_value,
1551                                         HV_KVP_EXCHANGE_MAX_VALUE_SIZE);
1552                         strcpy(key_name, "FullyQualifiedDomainName");
1553                         break;
1554                 case IntegrationServicesVersion:
1555                         strcpy(key_name, "IntegrationServicesVersion");
1556                         strcpy(key_value, lic_version);
1557                         break;
1558                 case NetworkAddressIPv4:
1559                         kvp_get_ip_info(AF_INET, NULL, KVP_OP_ENUMERATE,
1560                                 key_value, HV_KVP_EXCHANGE_MAX_VALUE_SIZE);
1561                         strcpy(key_name, "NetworkAddressIPv4");
1562                         break;
1563                 case NetworkAddressIPv6:
1564                         kvp_get_ip_info(AF_INET6, NULL, KVP_OP_ENUMERATE,
1565                                 key_value, HV_KVP_EXCHANGE_MAX_VALUE_SIZE);
1566                         strcpy(key_name, "NetworkAddressIPv6");
1567                         break;
1568                 case OSBuildNumber:
1569                         strcpy(key_value, os_build);
1570                         strcpy(key_name, "OSBuildNumber");
1571                         break;
1572                 case OSName:
1573                         strcpy(key_value, os_name);
1574                         strcpy(key_name, "OSName");
1575                         break;
1576                 case OSMajorVersion:
1577                         strcpy(key_value, os_major);
1578                         strcpy(key_name, "OSMajorVersion");
1579                         break;
1580                 case OSMinorVersion:
1581                         strcpy(key_value, os_minor);
1582                         strcpy(key_name, "OSMinorVersion");
1583                         break;
1584                 case OSVersion:
1585                         strcpy(key_value, os_build);
1586                         strcpy(key_name, "OSVersion");
1587                         break;
1588                 case ProcessorArchitecture:
1589                         strcpy(key_value, processor_arch);
1590                         strcpy(key_name, "ProcessorArchitecture");
1591                         break;
1592                 default:
1593                         hv_msg->error = HV_S_CONT;
1594                         break;
1595                 }
1596                 /*
1597                  * Send the value back to the kernel. The response is
1598                  * already in the receive buffer. Update the cn_msg header to
1599                  * reflect the key value that has been added to the message
1600                  */
1601 kvp_done:
1602
1603                 incoming_cn_msg->id.idx = CN_KVP_IDX;
1604                 incoming_cn_msg->id.val = CN_KVP_VAL;
1605                 incoming_cn_msg->ack = 0;
1606                 incoming_cn_msg->len = sizeof(struct hv_kvp_msg);
1607
1608                 len = netlink_send(fd, incoming_cn_msg);
1609                 if (len < 0) {
1610                         syslog(LOG_ERR, "net_link send failed; error:%d", len);
1611                         exit(EXIT_FAILURE);
1612                 }
1613         }
1614
1615 }