examples/vm_power_manager/channel_manager.c

   1 /*-
   2  *   BSD LICENSE
   3  *
   4  *   Copyright(c) 2010-2014 Intel Corporation. All rights reserved.
   5  *   All rights reserved.
   6  *
   7  *   Redistribution and use in source and binary forms, with or without
   8  *   modification, are permitted provided that the following conditions
   9  *   are met:
  10  *
  11  *     * Redistributions of source code must retain the above copyright
  12  *       notice, this list of conditions and the following disclaimer.
  13  *     * Redistributions in binary form must reproduce the above copyright
  14  *       notice, this list of conditions and the following disclaimer in
  15  *       the documentation and/or other materials provided with the
  16  *       distribution.
  17  *     * Neither the name of Intel Corporation nor the names of its
  18  *       contributors may be used to endorse or promote products derived
  19  *       from this software without specific prior written permission.
  20  *
  21  *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  *   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  *   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  24  *   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  25  *   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  26  *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  27  *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  28  *   DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  29  *   THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  30  *   (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  31  *   OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  32  */
  33
  34 #include <stdio.h>
  35 #include <stdlib.h>
  36 #include <sys/un.h>
  37 #include <fcntl.h>
  38 #include <unistd.h>
  39 #include <inttypes.h>
  40 #include <dirent.h>
  41 #include <errno.h>
  42
  43 #include <sys/queue.h>
  44 #include <sys/types.h>
  45 #include <sys/socket.h>
  46 #include <sys/select.h>
  47
  48 #include <rte_malloc.h>
  49 #include <rte_memory.h>
  50 #include <rte_mempool.h>
  51 #include <rte_log.h>
  52 #include <rte_atomic.h>
  53 #include <rte_spinlock.h>
  54
  55 #include <libvirt/libvirt.h>
  56
  57 #include "channel_manager.h"
  58 #include "channel_commands.h"
  59 #include "channel_monitor.h"
  60
  61
  62 #define RTE_LOGTYPE_CHANNEL_MANAGER RTE_LOGTYPE_USER1
  63
  64 #define ITERATIVE_BITMASK_CHECK_64(mask_u64b, i) \
  65                 for (i = 0; mask_u64b; mask_u64b &= ~(1ULL << i++)) \
  66                 if ((mask_u64b >> i) & 1) \
  67
  68 /* Global pointer to libvirt connection */
  69 static virConnectPtr global_vir_conn_ptr;
  70
  71 static unsigned char *global_cpumaps;
  72 static virVcpuInfo *global_vircpuinfo;
  73 static size_t global_maplen;
  74
  75 static unsigned global_n_host_cpus;
  76
  77 /*
  78  * Represents a single Virtual Machine
  79  */
  80 struct virtual_machine_info {
  81         char name[CHANNEL_MGR_MAX_NAME_LEN];
  82         rte_atomic64_t pcpu_mask[CHANNEL_CMDS_MAX_CPUS];
  83         struct channel_info *channels[CHANNEL_CMDS_MAX_VM_CHANNELS];
  84         uint64_t channel_mask;
  85         uint8_t num_channels;
  86         enum vm_status status;
  87         virDomainPtr domainPtr;
  88         virDomainInfo info;
  89         rte_spinlock_t config_spinlock;
  90         LIST_ENTRY(virtual_machine_info) vms_info;
  91 };
  92
  93 LIST_HEAD(, virtual_machine_info) vm_list_head;
  94
  95 static struct virtual_machine_info *
  96 find_domain_by_name(const char *name)
  97 {
  98         struct virtual_machine_info *info;
  99         LIST_FOREACH(info, &vm_list_head, vms_info) {
 100                 if (!strncmp(info->name, name, CHANNEL_MGR_MAX_NAME_LEN-1))
 101                         return info;
 102         }
 103         return NULL;
 104 }
 105
 106 static int
 107 update_pcpus_mask(struct virtual_machine_info *vm_info)
 108 {
 109         virVcpuInfoPtr cpuinfo;
 110         unsigned i, j;
 111         int n_vcpus;
 112         uint64_t mask;
 113
 114         memset(global_cpumaps, 0, CHANNEL_CMDS_MAX_CPUS*global_maplen);
 115
 116         if (!virDomainIsActive(vm_info->domainPtr)) {
 117                 n_vcpus = virDomainGetVcpuPinInfo(vm_info->domainPtr,
 118                                 vm_info->info.nrVirtCpu, global_cpumaps, global_maplen,
 119                                 VIR_DOMAIN_AFFECT_CONFIG);
 120                 if (n_vcpus < 0) {
 121                         RTE_LOG(ERR, CHANNEL_MANAGER, "Error getting vCPU info for "
 122                                         "in-active VM '%s'\n", vm_info->name);
 123                         return -1;
 124                 }
 125                 goto update_pcpus;
 126         }
 127
 128         memset(global_vircpuinfo, 0, sizeof(*global_vircpuinfo)*
 129                         CHANNEL_CMDS_MAX_CPUS);
 130
 131         cpuinfo = global_vircpuinfo;
 132
 133         n_vcpus = virDomainGetVcpus(vm_info->domainPtr, cpuinfo,
 134                         CHANNEL_CMDS_MAX_CPUS, global_cpumaps, global_maplen);
 135         if (n_vcpus < 0) {
 136                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error getting vCPU info for "
 137                                 "active VM '%s'\n", vm_info->name);
 138                 return -1;
 139         }
 140 update_pcpus:
 141         if (n_vcpus >= CHANNEL_CMDS_MAX_CPUS) {
 142                 RTE_LOG(ERR, CHANNEL_MANAGER, "Number of vCPUS(%u) is out of range "
 143                                 "0...%d\n", n_vcpus, CHANNEL_CMDS_MAX_CPUS-1);
 144                 return -1;
 145         }
 146         if (n_vcpus != vm_info->info.nrVirtCpu) {
 147                 RTE_LOG(INFO, CHANNEL_MANAGER, "Updating the number of vCPUs for VM '%s"
 148                                 " from %d -> %d\n", vm_info->name, vm_info->info.nrVirtCpu,
 149                                 n_vcpus);
 150                 vm_info->info.nrVirtCpu = n_vcpus;
 151         }
 152         for (i = 0; i < vm_info->info.nrVirtCpu; i++) {
 153                 mask = 0;
 154                 for (j = 0; j < global_n_host_cpus; j++) {
 155                         if (VIR_CPU_USABLE(global_cpumaps, global_maplen, i, j) > 0) {
 156                                 mask |= 1ULL << j;
 157                         }
 158                 }
 159                 rte_atomic64_set(&vm_info->pcpu_mask[i], mask);
 160         }
 161         return 0;
 162 }
 163
 164 int
 165 set_pcpus_mask(char *vm_name, unsigned vcpu, uint64_t core_mask)
 166 {
 167         unsigned i = 0;
 168         int flags = VIR_DOMAIN_AFFECT_LIVE|VIR_DOMAIN_AFFECT_CONFIG;
 169         struct virtual_machine_info *vm_info;
 170         uint64_t mask = core_mask;
 171
 172         if (vcpu >= CHANNEL_CMDS_MAX_CPUS) {
 173                 RTE_LOG(ERR, CHANNEL_MANAGER, "vCPU(%u) exceeds max allowable(%d)\n",
 174                                 vcpu, CHANNEL_CMDS_MAX_CPUS-1);
 175                 return -1;
 176         }
 177
 178         vm_info = find_domain_by_name(vm_name);
 179         if (vm_info == NULL) {
 180                 RTE_LOG(ERR, CHANNEL_MANAGER, "VM '%s' not found\n", vm_name);
 181                 return -1;
 182         }
 183
 184         if (!virDomainIsActive(vm_info->domainPtr)) {
 185                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to set vCPU(%u) to pCPU "
 186                                 "mask(0x%"PRIx64") for VM '%s', VM is not active\n",
 187                                 vcpu, core_mask, vm_info->name);
 188                 return -1;
 189         }
 190
 191         if (vcpu >= vm_info->info.nrVirtCpu) {
 192                 RTE_LOG(ERR, CHANNEL_MANAGER, "vCPU(%u) exceeds the assigned number of "
 193                                 "vCPUs(%u)\n", vcpu, vm_info->info.nrVirtCpu);
 194                 return -1;
 195         }
 196         memset(global_cpumaps, 0 , CHANNEL_CMDS_MAX_CPUS * global_maplen);
 197         ITERATIVE_BITMASK_CHECK_64(mask, i) {
 198                 VIR_USE_CPU(global_cpumaps, i);
 199                 if (i >= global_n_host_cpus) {
 200                         RTE_LOG(ERR, CHANNEL_MANAGER, "CPU(%u) exceeds the available "
 201                                         "number of CPUs(%u)\n", i, global_n_host_cpus);
 202                         return -1;
 203                 }
 204         }
 205         if (virDomainPinVcpuFlags(vm_info->domainPtr, vcpu, global_cpumaps,
 206                         global_maplen, flags) < 0) {
 207                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to set vCPU(%u) to pCPU "
 208                                 "mask(0x%"PRIx64") for VM '%s'\n", vcpu, core_mask,
 209                                 vm_info->name);
 210                 return -1;
 211         }
 212         rte_atomic64_set(&vm_info->pcpu_mask[vcpu], core_mask);
 213         return 0;
 214
 215 }
 216
 217 int
 218 set_pcpu(char *vm_name, unsigned vcpu, unsigned core_num)
 219 {
 220         uint64_t mask = 1ULL << core_num;
 221
 222         return set_pcpus_mask(vm_name, vcpu, mask);
 223 }
 224
 225 uint64_t
 226 get_pcpus_mask(struct channel_info *chan_info, unsigned vcpu)
 227 {
 228         struct virtual_machine_info *vm_info =
 229                         (struct virtual_machine_info *)chan_info->priv_info;
 230         return rte_atomic64_read(&vm_info->pcpu_mask[vcpu]);
 231 }
 232
 233 static inline int
 234 channel_exists(struct virtual_machine_info *vm_info, unsigned channel_num)
 235 {
 236         rte_spinlock_lock(&(vm_info->config_spinlock));
 237         if (vm_info->channel_mask & (1ULL << channel_num)) {
 238                 rte_spinlock_unlock(&(vm_info->config_spinlock));
 239                 return 1;
 240         }
 241         rte_spinlock_unlock(&(vm_info->config_spinlock));
 242         return 0;
 243 }
 244
 245
 246
 247 static int
 248 open_non_blocking_channel(struct channel_info *info)
 249 {
 250         int ret, flags;
 251         struct sockaddr_un sock_addr;
 252         fd_set soc_fd_set;
 253         struct timeval tv;
 254
 255         info->fd = socket(AF_UNIX, SOCK_STREAM, 0);
 256         if (info->fd == -1) {
 257                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error(%s) creating socket for '%s'\n",
 258                                 strerror(errno),
 259                                 info->channel_path);
 260                 return -1;
 261         }
 262         sock_addr.sun_family = AF_UNIX;
 263         memcpy(&sock_addr.sun_path, info->channel_path,
 264                         strlen(info->channel_path)+1);
 265
 266         /* Get current flags */
 267         flags = fcntl(info->fd, F_GETFL, 0);
 268         if (flags < 0) {
 269                 RTE_LOG(WARNING, CHANNEL_MANAGER, "Error(%s) fcntl get flags socket for"
 270                                 "'%s'\n", strerror(errno), info->channel_path);
 271                 return 1;
 272         }
 273         /* Set to Non Blocking */
 274         flags |= O_NONBLOCK;
 275         if (fcntl(info->fd, F_SETFL, flags) < 0) {
 276                 RTE_LOG(WARNING, CHANNEL_MANAGER, "Error(%s) setting non-blocking "
 277                                 "socket for '%s'\n", strerror(errno), info->channel_path);
 278                 return -1;
 279         }
 280         ret = connect(info->fd, (struct sockaddr *)&sock_addr,
 281                         sizeof(sock_addr));
 282         if (ret < 0) {
 283                 /* ECONNREFUSED error is given when VM is not active */
 284                 if (errno == ECONNREFUSED) {
 285                         RTE_LOG(WARNING, CHANNEL_MANAGER, "VM is not active or has not "
 286                                         "activated its endpoint to channel %s\n",
 287                                         info->channel_path);
 288                         return -1;
 289                 }
 290                 /* Wait for tv_sec if in progress */
 291                 else if (errno == EINPROGRESS) {
 292                         tv.tv_sec = 2;
 293                         tv.tv_usec = 0;
 294                         FD_ZERO(&soc_fd_set);
 295                         FD_SET(info->fd, &soc_fd_set);
 296                         if (select(info->fd+1, NULL, &soc_fd_set, NULL, &tv) > 0) {
 297                                 RTE_LOG(WARNING, CHANNEL_MANAGER, "Timeout or error on channel "
 298                                                 "'%s'\n", info->channel_path);
 299                                 return -1;
 300                         }
 301                 } else {
 302                         /* Any other error */
 303                         RTE_LOG(WARNING, CHANNEL_MANAGER, "Error(%s) connecting socket"
 304                                         " for '%s'\n", strerror(errno), info->channel_path);
 305                         return -1;
 306                 }
 307         }
 308         return 0;
 309 }
 310
 311 static int
 312 setup_channel_info(struct virtual_machine_info **vm_info_dptr,
 313                 struct channel_info **chan_info_dptr, unsigned channel_num)
 314 {
 315         struct channel_info *chan_info = *chan_info_dptr;
 316         struct virtual_machine_info *vm_info = *vm_info_dptr;
 317
 318         chan_info->channel_num = channel_num;
 319         chan_info->priv_info = (void *)vm_info;
 320         chan_info->status = CHANNEL_MGR_CHANNEL_DISCONNECTED;
 321         if (open_non_blocking_channel(chan_info) < 0) {
 322                 RTE_LOG(ERR, CHANNEL_MANAGER, "Could not open channel: "
 323                                 "'%s' for VM '%s'\n",
 324                                 chan_info->channel_path, vm_info->name);
 325                 return -1;
 326         }
 327         if (add_channel_to_monitor(&chan_info) < 0) {
 328                 RTE_LOG(ERR, CHANNEL_MANAGER, "Could add channel: "
 329                                 "'%s' to epoll ctl for VM '%s'\n",
 330                                 chan_info->channel_path, vm_info->name);
 331                 return -1;
 332
 333         }
 334         rte_spinlock_lock(&(vm_info->config_spinlock));
 335         vm_info->num_channels++;
 336         vm_info->channel_mask |= 1ULL << channel_num;
 337         vm_info->channels[channel_num] = chan_info;
 338         chan_info->status = CHANNEL_MGR_CHANNEL_CONNECTED;
 339         rte_spinlock_unlock(&(vm_info->config_spinlock));
 340         return 0;
 341 }
 342
 343 int
 344 add_all_channels(const char *vm_name)
 345 {
 346         DIR *d;
 347         struct dirent *dir;
 348         struct virtual_machine_info *vm_info;
 349         struct channel_info *chan_info;
 350         char *token, *remaining, *tail_ptr;
 351         char socket_name[PATH_MAX];
 352         unsigned channel_num;
 353         int num_channels_enabled = 0;
 354
 355         /* verify VM exists */
 356         vm_info = find_domain_by_name(vm_name);
 357         if (vm_info == NULL) {
 358                 RTE_LOG(ERR, CHANNEL_MANAGER, "VM: '%s' not found"
 359                                 " during channel discovery\n", vm_name);
 360                 return 0;
 361         }
 362         if (!virDomainIsActive(vm_info->domainPtr)) {
 363                 RTE_LOG(ERR, CHANNEL_MANAGER, "VM: '%s' is not active\n", vm_name);
 364                 vm_info->status = CHANNEL_MGR_VM_INACTIVE;
 365                 return 0;
 366         }
 367         d = opendir(CHANNEL_MGR_SOCKET_PATH);
 368         if (d == NULL) {
 369                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error opening directory '%s': %s\n",
 370                                 CHANNEL_MGR_SOCKET_PATH, strerror(errno));
 371                 return -1;
 372         }
 373         while ((dir = readdir(d)) != NULL) {
 374                 if (!strncmp(dir->d_name, ".", 1) ||
 375                                 !strncmp(dir->d_name, "..", 2))
 376                         continue;
 377
 378                 snprintf(socket_name, sizeof(socket_name), "%s", dir->d_name);
 379                 remaining = socket_name;
 380                 /* Extract vm_name from "<vm_name>.<channel_num>" */
 381                 token = strsep(&remaining, ".");
 382                 if (remaining == NULL)
 383                         continue;
 384                 if (strncmp(vm_name, token, CHANNEL_MGR_MAX_NAME_LEN))
 385                         continue;
 386
 387                 /* remaining should contain only <channel_num> */
 388                 errno = 0;
 389                 channel_num = (unsigned)strtol(remaining, &tail_ptr, 0);
 390                 if ((errno != 0) || (remaining[0] == '\0') ||
 391                                 tail_ptr == NULL || (*tail_ptr != '\0')) {
 392                         RTE_LOG(WARNING, CHANNEL_MANAGER, "Malformed channel name"
 393                                         "'%s' found it should be in the form of "
 394                                         "'<guest_name>.<channel_num>(decimal)'\n",
 395                                         dir->d_name);
 396                         continue;
 397                 }
 398                 if (channel_num >= CHANNEL_CMDS_MAX_VM_CHANNELS) {
 399                         RTE_LOG(WARNING, CHANNEL_MANAGER, "Channel number(%u) is "
 400                                         "greater than max allowable: %d, skipping '%s%s'\n",
 401                                         channel_num, CHANNEL_CMDS_MAX_VM_CHANNELS-1,
 402                                         CHANNEL_MGR_SOCKET_PATH, dir->d_name);
 403                         continue;
 404                 }
 405                 /* if channel has not been added previously */
 406                 if (channel_exists(vm_info, channel_num))
 407                         continue;
 408
 409                 chan_info = rte_malloc(NULL, sizeof(*chan_info),
 410                                 RTE_CACHE_LINE_SIZE);
 411                 if (chan_info == NULL) {
 412                         RTE_LOG(ERR, CHANNEL_MANAGER, "Error allocating memory for "
 413                                 "channel '%s%s'\n", CHANNEL_MGR_SOCKET_PATH, dir->d_name);
 414                         continue;
 415                 }
 416
 417                 snprintf(chan_info->channel_path,
 418                                 sizeof(chan_info->channel_path), "%s%s",
 419                                 CHANNEL_MGR_SOCKET_PATH, dir->d_name);
 420
 421                 if (setup_channel_info(&vm_info, &chan_info, channel_num) < 0) {
 422                         rte_free(chan_info);
 423                         continue;
 424                 }
 425
 426                 num_channels_enabled++;
 427         }
 428         closedir(d);
 429         return num_channels_enabled;
 430 }
 431
 432 int
 433 add_channels(const char *vm_name, unsigned *channel_list,
 434                 unsigned len_channel_list)
 435 {
 436         struct virtual_machine_info *vm_info;
 437         struct channel_info *chan_info;
 438         char socket_path[PATH_MAX];
 439         unsigned i;
 440         int num_channels_enabled = 0;
 441
 442         vm_info = find_domain_by_name(vm_name);
 443         if (vm_info == NULL) {
 444                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to add channels: VM '%s' "
 445                                 "not found\n", vm_name);
 446                 return 0;
 447         }
 448
 449         if (!virDomainIsActive(vm_info->domainPtr)) {
 450                 RTE_LOG(ERR, CHANNEL_MANAGER, "VM: '%s' is not active\n", vm_name);
 451                 vm_info->status = CHANNEL_MGR_VM_INACTIVE;
 452                 return 0;
 453         }
 454
 455         for (i = 0; i < len_channel_list; i++) {
 456
 457                 if (channel_list[i] >= CHANNEL_CMDS_MAX_VM_CHANNELS) {
 458                         RTE_LOG(INFO, CHANNEL_MANAGER, "Channel(%u) is out of range "
 459                                                         "0...%d\n", channel_list[i],
 460                                                         CHANNEL_CMDS_MAX_VM_CHANNELS-1);
 461                         continue;
 462                 }
 463                 if (channel_exists(vm_info, channel_list[i])) {
 464                         RTE_LOG(INFO, CHANNEL_MANAGER, "Channel already exists, skipping  "
 465                                         "'%s.%u'\n", vm_name, i);
 466                         continue;
 467                 }
 468
 469                 snprintf(socket_path, sizeof(socket_path), "%s%s.%u",
 470                                 CHANNEL_MGR_SOCKET_PATH, vm_name, channel_list[i]);
 471                 errno = 0;
 472                 if (access(socket_path, F_OK) < 0) {
 473                         RTE_LOG(ERR, CHANNEL_MANAGER, "Channel path '%s' error: "
 474                                         "%s\n", socket_path, strerror(errno));
 475                         continue;
 476                 }
 477                 chan_info = rte_malloc(NULL, sizeof(*chan_info),
 478                                 RTE_CACHE_LINE_SIZE);
 479                 if (chan_info == NULL) {
 480                         RTE_LOG(ERR, CHANNEL_MANAGER, "Error allocating memory for "
 481                                         "channel '%s'\n", socket_path);
 482                         continue;
 483                 }
 484                 snprintf(chan_info->channel_path,
 485                                 sizeof(chan_info->channel_path), "%s%s.%u",
 486                                 CHANNEL_MGR_SOCKET_PATH, vm_name, channel_list[i]);
 487                 if (setup_channel_info(&vm_info, &chan_info, channel_list[i]) < 0) {
 488                         rte_free(chan_info);
 489                         continue;
 490                 }
 491                 num_channels_enabled++;
 492
 493         }
 494         return num_channels_enabled;
 495 }
 496
 497 int
 498 remove_channel(struct channel_info **chan_info_dptr)
 499 {
 500         struct virtual_machine_info *vm_info;
 501         struct channel_info *chan_info = *chan_info_dptr;
 502
 503         close(chan_info->fd);
 504
 505         vm_info = (struct virtual_machine_info *)chan_info->priv_info;
 506
 507         rte_spinlock_lock(&(vm_info->config_spinlock));
 508         vm_info->channel_mask &= ~(1ULL << chan_info->channel_num);
 509         vm_info->num_channels--;
 510         rte_spinlock_unlock(&(vm_info->config_spinlock));
 511
 512         rte_free(chan_info);
 513         return 0;
 514 }
 515
 516 int
 517 set_channel_status_all(const char *vm_name, enum channel_status status)
 518 {
 519         struct virtual_machine_info *vm_info;
 520         unsigned i;
 521         uint64_t mask;
 522         int num_channels_changed = 0;
 523
 524         if (!(status == CHANNEL_MGR_CHANNEL_CONNECTED ||
 525                         status == CHANNEL_MGR_CHANNEL_DISABLED)) {
 526                 RTE_LOG(ERR, CHANNEL_MANAGER, "Channels can only be enabled or "
 527                                 "disabled: Unable to change status for VM '%s'\n", vm_name);
 528         }
 529         vm_info = find_domain_by_name(vm_name);
 530         if (vm_info == NULL) {
 531                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to disable channels: VM '%s' "
 532                                 "not found\n", vm_name);
 533                 return 0;
 534         }
 535
 536         rte_spinlock_lock(&(vm_info->config_spinlock));
 537         mask = vm_info->channel_mask;
 538         ITERATIVE_BITMASK_CHECK_64(mask, i) {
 539                 vm_info->channels[i]->status = status;
 540                 num_channels_changed++;
 541         }
 542         rte_spinlock_unlock(&(vm_info->config_spinlock));
 543         return num_channels_changed;
 544
 545 }
 546
 547 int
 548 set_channel_status(const char *vm_name, unsigned *channel_list,
 549                 unsigned len_channel_list, enum channel_status status)
 550 {
 551         struct virtual_machine_info *vm_info;
 552         unsigned i;
 553         int num_channels_changed = 0;
 554
 555         if (!(status == CHANNEL_MGR_CHANNEL_CONNECTED ||
 556                         status == CHANNEL_MGR_CHANNEL_DISABLED)) {
 557                 RTE_LOG(ERR, CHANNEL_MANAGER, "Channels can only be enabled or "
 558                                 "disabled: Unable to change status for VM '%s'\n", vm_name);
 559         }
 560         vm_info = find_domain_by_name(vm_name);
 561         if (vm_info == NULL) {
 562                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to add channels: VM '%s' "
 563                                 "not found\n", vm_name);
 564                 return 0;
 565         }
 566         for (i = 0; i < len_channel_list; i++) {
 567                 if (channel_exists(vm_info, channel_list[i])) {
 568                         rte_spinlock_lock(&(vm_info->config_spinlock));
 569                         vm_info->channels[channel_list[i]]->status = status;
 570                         rte_spinlock_unlock(&(vm_info->config_spinlock));
 571                         num_channels_changed++;
 572                 }
 573         }
 574         return num_channels_changed;
 575 }
 576
 577 int
 578 get_info_vm(const char *vm_name, struct vm_info *info)
 579 {
 580         struct virtual_machine_info *vm_info;
 581         unsigned i, channel_num = 0;
 582         uint64_t mask;
 583
 584         vm_info = find_domain_by_name(vm_name);
 585         if (vm_info == NULL) {
 586                 RTE_LOG(ERR, CHANNEL_MANAGER, "VM '%s' not found\n", vm_name);
 587                 return -1;
 588         }
 589         info->status = CHANNEL_MGR_VM_ACTIVE;
 590         if (!virDomainIsActive(vm_info->domainPtr))
 591                 info->status = CHANNEL_MGR_VM_INACTIVE;
 592
 593         rte_spinlock_lock(&(vm_info->config_spinlock));
 594
 595         mask = vm_info->channel_mask;
 596         ITERATIVE_BITMASK_CHECK_64(mask, i) {
 597                 info->channels[channel_num].channel_num = i;
 598                 memcpy(info->channels[channel_num].channel_path,
 599                                 vm_info->channels[i]->channel_path, UNIX_PATH_MAX);
 600                 info->channels[channel_num].status = vm_info->channels[i]->status;
 601                 info->channels[channel_num].fd = vm_info->channels[i]->fd;
 602                 channel_num++;
 603         }
 604
 605         info->num_channels = channel_num;
 606         info->num_vcpus = vm_info->info.nrVirtCpu;
 607         rte_spinlock_unlock(&(vm_info->config_spinlock));
 608
 609         memcpy(info->name, vm_info->name, sizeof(vm_info->name));
 610         for (i = 0; i < info->num_vcpus; i++) {
 611                 info->pcpu_mask[i] = rte_atomic64_read(&vm_info->pcpu_mask[i]);
 612         }
 613         return 0;
 614 }
 615
 616 int
 617 add_vm(const char *vm_name)
 618 {
 619         struct virtual_machine_info *new_domain;
 620         virDomainPtr dom_ptr;
 621         int i;
 622
 623         if (find_domain_by_name(vm_name) != NULL) {
 624                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to add VM: VM '%s' "
 625                                 "already exists\n", vm_name);
 626                 return -1;
 627         }
 628
 629         if (global_vir_conn_ptr == NULL) {
 630                 RTE_LOG(ERR, CHANNEL_MANAGER, "No connection to hypervisor exists\n");
 631                 return -1;
 632         }
 633         dom_ptr = virDomainLookupByName(global_vir_conn_ptr, vm_name);
 634         if (dom_ptr == NULL) {
 635                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error on VM lookup with libvirt: "
 636                                 "VM '%s' not found\n", vm_name);
 637                 return -1;
 638         }
 639
 640         new_domain = rte_malloc("virtual_machine_info", sizeof(*new_domain),
 641                         RTE_CACHE_LINE_SIZE);
 642         if (new_domain == NULL) {
 643                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to allocate memory for VM "
 644                                 "info\n");
 645                 return -1;
 646         }
 647         new_domain->domainPtr = dom_ptr;
 648         if (virDomainGetInfo(new_domain->domainPtr, &new_domain->info) != 0) {
 649                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to get libvirt VM info\n");
 650                 rte_free(new_domain);
 651                 return -1;
 652         }
 653         if (new_domain->info.nrVirtCpu > CHANNEL_CMDS_MAX_CPUS) {
 654                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error the number of virtual CPUs(%u) is "
 655                                 "greater than allowable(%d)\n", new_domain->info.nrVirtCpu,
 656                                 CHANNEL_CMDS_MAX_CPUS);
 657                 rte_free(new_domain);
 658                 return -1;
 659         }
 660
 661         for (i = 0; i < CHANNEL_CMDS_MAX_CPUS; i++) {
 662                 rte_atomic64_init(&new_domain->pcpu_mask[i]);
 663         }
 664         if (update_pcpus_mask(new_domain) < 0) {
 665                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error getting physical CPU pinning\n");
 666                 rte_free(new_domain);
 667                 return -1;
 668         }
 669         strncpy(new_domain->name, vm_name, sizeof(new_domain->name));
 670         new_domain->channel_mask = 0;
 671         new_domain->num_channels = 0;
 672
 673         if (!virDomainIsActive(dom_ptr))
 674                 new_domain->status = CHANNEL_MGR_VM_INACTIVE;
 675         else
 676                 new_domain->status = CHANNEL_MGR_VM_ACTIVE;
 677
 678         rte_spinlock_init(&(new_domain->config_spinlock));
 679         LIST_INSERT_HEAD(&vm_list_head, new_domain, vms_info);
 680         return 0;
 681 }
 682
 683 int
 684 remove_vm(const char *vm_name)
 685 {
 686         struct virtual_machine_info *vm_info = find_domain_by_name(vm_name);
 687
 688         if (vm_info == NULL) {
 689                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to remove VM: VM '%s' "
 690                                 "not found\n", vm_name);
 691                 return -1;
 692         }
 693         rte_spinlock_lock(&vm_info->config_spinlock);
 694         if (vm_info->num_channels != 0) {
 695                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to remove VM '%s', there are "
 696                                 "%"PRId8" channels still active\n",
 697                                 vm_name, vm_info->num_channels);
 698                 rte_spinlock_unlock(&vm_info->config_spinlock);
 699                 return -1;
 700         }
 701         LIST_REMOVE(vm_info, vms_info);
 702         rte_spinlock_unlock(&vm_info->config_spinlock);
 703         rte_free(vm_info);
 704         return 0;
 705 }
 706
 707 static void
 708 disconnect_hypervisor(void)
 709 {
 710         if (global_vir_conn_ptr != NULL) {
 711                 virConnectClose(global_vir_conn_ptr);
 712                 global_vir_conn_ptr = NULL;
 713         }
 714 }
 715
 716 static int
 717 connect_hypervisor(const char *path)
 718 {
 719         if (global_vir_conn_ptr != NULL) {
 720                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error connecting to %s, connection "
 721                                 "already established\n", path);
 722                 return -1;
 723         }
 724         global_vir_conn_ptr = virConnectOpen(path);
 725         if (global_vir_conn_ptr == NULL) {
 726                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error failed to open connection to "
 727                                 "Hypervisor '%s'\n", path);
 728                 return -1;
 729         }
 730         return 0;
 731 }
 732
 733 int
 734 channel_manager_init(const char *path)
 735 {
 736         virNodeInfo info;
 737
 738         LIST_INIT(&vm_list_head);
 739         if (connect_hypervisor(path) < 0) {
 740                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to initialize channel manager\n");
 741                 return -1;
 742         }
 743
 744         global_maplen = VIR_CPU_MAPLEN(CHANNEL_CMDS_MAX_CPUS);
 745
 746         global_vircpuinfo = rte_zmalloc(NULL, sizeof(*global_vircpuinfo) *
 747                         CHANNEL_CMDS_MAX_CPUS, RTE_CACHE_LINE_SIZE);
 748         if (global_vircpuinfo == NULL) {
 749                 RTE_LOG(ERR, CHANNEL_MANAGER, "Error allocating memory for CPU Info\n");
 750                 goto error;
 751         }
 752         global_cpumaps = rte_zmalloc(NULL, CHANNEL_CMDS_MAX_CPUS * global_maplen,
 753                         RTE_CACHE_LINE_SIZE);
 754         if (global_cpumaps == NULL) {
 755                 goto error;
 756         }
 757
 758         if (virNodeGetInfo(global_vir_conn_ptr, &info)) {
 759                 RTE_LOG(ERR, CHANNEL_MANAGER, "Unable to retrieve node Info\n");
 760                 goto error;
 761         }
 762
 763         global_n_host_cpus = (unsigned)info.cpus;
 764
 765         if (global_n_host_cpus > CHANNEL_CMDS_MAX_CPUS) {
 766                 RTE_LOG(WARNING, CHANNEL_MANAGER, "The number of host CPUs(%u) exceeds the "
 767                                 "maximum of %u. No cores over %u should be used.\n",
 768                                 global_n_host_cpus, CHANNEL_CMDS_MAX_CPUS,
 769                                 CHANNEL_CMDS_MAX_CPUS - 1);
 770                 global_n_host_cpus = CHANNEL_CMDS_MAX_CPUS;
 771         }
 772
 773         return 0;
 774 error:
 775         disconnect_hypervisor();
 776         return -1;
 777 }
 778
 779 void
 780 channel_manager_exit(void)
 781 {
 782         unsigned i;
 783         uint64_t mask;
 784         struct virtual_machine_info *vm_info;
 785
 786         LIST_FOREACH(vm_info, &vm_list_head, vms_info) {
 787
 788                 rte_spinlock_lock(&(vm_info->config_spinlock));
 789
 790                 mask = vm_info->channel_mask;
 791                 ITERATIVE_BITMASK_CHECK_64(mask, i) {
 792                         remove_channel_from_monitor(vm_info->channels[i]);
 793                         close(vm_info->channels[i]->fd);
 794                         rte_free(vm_info->channels[i]);
 795                 }
 796                 rte_spinlock_unlock(&(vm_info->config_spinlock));
 797
 798                 LIST_REMOVE(vm_info, vms_info);
 799                 rte_free(vm_info);
 800         }
 801
 802         rte_free(global_cpumaps);
 803         rte_free(global_vircpuinfo);
 804         disconnect_hypervisor();
 805 }