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 } 806