1 // SPDX-License-Identifier: GPL-2.0-only 2 /* 3 * VMware VMCI Driver 4 * 5 * Copyright (C) 2012 VMware, Inc. All rights reserved. 6 */ 7 8 #include <linux/vmw_vmci_defs.h> 9 #include <linux/vmw_vmci_api.h> 10 #include <linux/moduleparam.h> 11 #include <linux/interrupt.h> 12 #include <linux/highmem.h> 13 #include <linux/kernel.h> 14 #include <linux/mm.h> 15 #include <linux/module.h> 16 #include <linux/sched.h> 17 #include <linux/slab.h> 18 #include <linux/init.h> 19 #include <linux/pci.h> 20 #include <linux/smp.h> 21 #include <linux/io.h> 22 #include <linux/vmalloc.h> 23 24 #include "vmci_datagram.h" 25 #include "vmci_doorbell.h" 26 #include "vmci_context.h" 27 #include "vmci_driver.h" 28 #include "vmci_event.h" 29 30 #define PCI_DEVICE_ID_VMWARE_VMCI 0x0740 31 32 #define VMCI_UTIL_NUM_RESOURCES 1 33 34 static bool vmci_disable_msi; 35 module_param_named(disable_msi, vmci_disable_msi, bool, 0); 36 MODULE_PARM_DESC(disable_msi, "Disable MSI use in driver - (default=0)"); 37 38 static bool vmci_disable_msix; 39 module_param_named(disable_msix, vmci_disable_msix, bool, 0); 40 MODULE_PARM_DESC(disable_msix, "Disable MSI-X use in driver - (default=0)"); 41 42 static u32 ctx_update_sub_id = VMCI_INVALID_ID; 43 static u32 vm_context_id = VMCI_INVALID_ID; 44 45 struct vmci_guest_device { 46 struct device *dev; /* PCI device we are attached to */ 47 void __iomem *iobase; 48 void __iomem *mmio_base; 49 50 bool exclusive_vectors; 51 52 struct tasklet_struct datagram_tasklet; 53 struct tasklet_struct bm_tasklet; 54 55 void *data_buffer; 56 void *notification_bitmap; 57 dma_addr_t notification_base; 58 }; 59 60 static bool use_ppn64; 61 62 bool vmci_use_ppn64(void) 63 { 64 return use_ppn64; 65 } 66 67 /* vmci_dev singleton device and supporting data*/ 68 struct pci_dev *vmci_pdev; 69 static struct vmci_guest_device *vmci_dev_g; 70 static DEFINE_SPINLOCK(vmci_dev_spinlock); 71 72 static atomic_t vmci_num_guest_devices = ATOMIC_INIT(0); 73 74 bool vmci_guest_code_active(void) 75 { 76 return atomic_read(&vmci_num_guest_devices) != 0; 77 } 78 79 u32 vmci_get_vm_context_id(void) 80 { 81 if (vm_context_id == VMCI_INVALID_ID) { 82 struct vmci_datagram get_cid_msg; 83 get_cid_msg.dst = 84 vmci_make_handle(VMCI_HYPERVISOR_CONTEXT_ID, 85 VMCI_GET_CONTEXT_ID); 86 get_cid_msg.src = VMCI_ANON_SRC_HANDLE; 87 get_cid_msg.payload_size = 0; 88 vm_context_id = vmci_send_datagram(&get_cid_msg); 89 } 90 return vm_context_id; 91 } 92 93 static unsigned int vmci_read_reg(struct vmci_guest_device *dev, u32 reg) 94 { 95 if (dev->mmio_base != NULL) 96 return readl(dev->mmio_base + reg); 97 return ioread32(dev->iobase + reg); 98 } 99 100 static void vmci_write_reg(struct vmci_guest_device *dev, u32 val, u32 reg) 101 { 102 if (dev->mmio_base != NULL) 103 writel(val, dev->mmio_base + reg); 104 else 105 iowrite32(val, dev->iobase + reg); 106 } 107 108 /* 109 * VM to hypervisor call mechanism. We use the standard VMware naming 110 * convention since shared code is calling this function as well. 111 */ 112 int vmci_send_datagram(struct vmci_datagram *dg) 113 { 114 unsigned long flags; 115 int result; 116 117 /* Check args. */ 118 if (dg == NULL) 119 return VMCI_ERROR_INVALID_ARGS; 120 121 /* 122 * Need to acquire spinlock on the device because the datagram 123 * data may be spread over multiple pages and the monitor may 124 * interleave device user rpc calls from multiple 125 * VCPUs. Acquiring the spinlock precludes that 126 * possibility. Disabling interrupts to avoid incoming 127 * datagrams during a "rep out" and possibly landing up in 128 * this function. 129 */ 130 spin_lock_irqsave(&vmci_dev_spinlock, flags); 131 132 if (vmci_dev_g) { 133 iowrite8_rep(vmci_dev_g->iobase + VMCI_DATA_OUT_ADDR, 134 dg, VMCI_DG_SIZE(dg)); 135 result = vmci_read_reg(vmci_dev_g, VMCI_RESULT_LOW_ADDR); 136 } else { 137 result = VMCI_ERROR_UNAVAILABLE; 138 } 139 140 spin_unlock_irqrestore(&vmci_dev_spinlock, flags); 141 142 return result; 143 } 144 EXPORT_SYMBOL_GPL(vmci_send_datagram); 145 146 /* 147 * Gets called with the new context id if updated or resumed. 148 * Context id. 149 */ 150 static void vmci_guest_cid_update(u32 sub_id, 151 const struct vmci_event_data *event_data, 152 void *client_data) 153 { 154 const struct vmci_event_payld_ctx *ev_payload = 155 vmci_event_data_const_payload(event_data); 156 157 if (sub_id != ctx_update_sub_id) { 158 pr_devel("Invalid subscriber (ID=0x%x)\n", sub_id); 159 return; 160 } 161 162 if (!event_data || ev_payload->context_id == VMCI_INVALID_ID) { 163 pr_devel("Invalid event data\n"); 164 return; 165 } 166 167 pr_devel("Updating context from (ID=0x%x) to (ID=0x%x) on event (type=%d)\n", 168 vm_context_id, ev_payload->context_id, event_data->event); 169 170 vm_context_id = ev_payload->context_id; 171 } 172 173 /* 174 * Verify that the host supports the hypercalls we need. If it does not, 175 * try to find fallback hypercalls and use those instead. Returns 176 * true if required hypercalls (or fallback hypercalls) are 177 * supported by the host, false otherwise. 178 */ 179 static int vmci_check_host_caps(struct pci_dev *pdev) 180 { 181 bool result; 182 struct vmci_resource_query_msg *msg; 183 u32 msg_size = sizeof(struct vmci_resource_query_hdr) + 184 VMCI_UTIL_NUM_RESOURCES * sizeof(u32); 185 struct vmci_datagram *check_msg; 186 187 check_msg = kzalloc(msg_size, GFP_KERNEL); 188 if (!check_msg) { 189 dev_err(&pdev->dev, "%s: Insufficient memory\n", __func__); 190 return -ENOMEM; 191 } 192 193 check_msg->dst = vmci_make_handle(VMCI_HYPERVISOR_CONTEXT_ID, 194 VMCI_RESOURCES_QUERY); 195 check_msg->src = VMCI_ANON_SRC_HANDLE; 196 check_msg->payload_size = msg_size - VMCI_DG_HEADERSIZE; 197 msg = (struct vmci_resource_query_msg *)VMCI_DG_PAYLOAD(check_msg); 198 199 msg->num_resources = VMCI_UTIL_NUM_RESOURCES; 200 msg->resources[0] = VMCI_GET_CONTEXT_ID; 201 202 /* Checks that hyper calls are supported */ 203 result = vmci_send_datagram(check_msg) == 0x01; 204 kfree(check_msg); 205 206 dev_dbg(&pdev->dev, "%s: Host capability check: %s\n", 207 __func__, result ? "PASSED" : "FAILED"); 208 209 /* We need the vector. There are no fallbacks. */ 210 return result ? 0 : -ENXIO; 211 } 212 213 /* 214 * Reads datagrams from the data in port and dispatches them. We 215 * always start reading datagrams into only the first page of the 216 * datagram buffer. If the datagrams don't fit into one page, we 217 * use the maximum datagram buffer size for the remainder of the 218 * invocation. This is a simple heuristic for not penalizing 219 * small datagrams. 220 * 221 * This function assumes that it has exclusive access to the data 222 * in port for the duration of the call. 223 */ 224 static void vmci_dispatch_dgs(unsigned long data) 225 { 226 struct vmci_guest_device *vmci_dev = (struct vmci_guest_device *)data; 227 u8 *dg_in_buffer = vmci_dev->data_buffer; 228 struct vmci_datagram *dg; 229 size_t dg_in_buffer_size = VMCI_MAX_DG_SIZE; 230 size_t current_dg_in_buffer_size = PAGE_SIZE; 231 size_t remaining_bytes; 232 233 BUILD_BUG_ON(VMCI_MAX_DG_SIZE < PAGE_SIZE); 234 235 ioread8_rep(vmci_dev->iobase + VMCI_DATA_IN_ADDR, 236 vmci_dev->data_buffer, current_dg_in_buffer_size); 237 dg = (struct vmci_datagram *)dg_in_buffer; 238 remaining_bytes = current_dg_in_buffer_size; 239 240 while (dg->dst.resource != VMCI_INVALID_ID || 241 remaining_bytes > PAGE_SIZE) { 242 unsigned dg_in_size; 243 244 /* 245 * When the input buffer spans multiple pages, a datagram can 246 * start on any page boundary in the buffer. 247 */ 248 if (dg->dst.resource == VMCI_INVALID_ID) { 249 dg = (struct vmci_datagram *)roundup( 250 (uintptr_t)dg + 1, PAGE_SIZE); 251 remaining_bytes = 252 (size_t)(dg_in_buffer + 253 current_dg_in_buffer_size - 254 (u8 *)dg); 255 continue; 256 } 257 258 dg_in_size = VMCI_DG_SIZE_ALIGNED(dg); 259 260 if (dg_in_size <= dg_in_buffer_size) { 261 int result; 262 263 /* 264 * If the remaining bytes in the datagram 265 * buffer doesn't contain the complete 266 * datagram, we first make sure we have enough 267 * room for it and then we read the reminder 268 * of the datagram and possibly any following 269 * datagrams. 270 */ 271 if (dg_in_size > remaining_bytes) { 272 if (remaining_bytes != 273 current_dg_in_buffer_size) { 274 275 /* 276 * We move the partial 277 * datagram to the front and 278 * read the reminder of the 279 * datagram and possibly 280 * following calls into the 281 * following bytes. 282 */ 283 memmove(dg_in_buffer, dg_in_buffer + 284 current_dg_in_buffer_size - 285 remaining_bytes, 286 remaining_bytes); 287 dg = (struct vmci_datagram *) 288 dg_in_buffer; 289 } 290 291 if (current_dg_in_buffer_size != 292 dg_in_buffer_size) 293 current_dg_in_buffer_size = 294 dg_in_buffer_size; 295 296 ioread8_rep(vmci_dev->iobase + 297 VMCI_DATA_IN_ADDR, 298 vmci_dev->data_buffer + 299 remaining_bytes, 300 current_dg_in_buffer_size - 301 remaining_bytes); 302 } 303 304 /* 305 * We special case event datagrams from the 306 * hypervisor. 307 */ 308 if (dg->src.context == VMCI_HYPERVISOR_CONTEXT_ID && 309 dg->dst.resource == VMCI_EVENT_HANDLER) { 310 result = vmci_event_dispatch(dg); 311 } else { 312 result = vmci_datagram_invoke_guest_handler(dg); 313 } 314 if (result < VMCI_SUCCESS) 315 dev_dbg(vmci_dev->dev, 316 "Datagram with resource (ID=0x%x) failed (err=%d)\n", 317 dg->dst.resource, result); 318 319 /* On to the next datagram. */ 320 dg = (struct vmci_datagram *)((u8 *)dg + 321 dg_in_size); 322 } else { 323 size_t bytes_to_skip; 324 325 /* 326 * Datagram doesn't fit in datagram buffer of maximal 327 * size. We drop it. 328 */ 329 dev_dbg(vmci_dev->dev, 330 "Failed to receive datagram (size=%u bytes)\n", 331 dg_in_size); 332 333 bytes_to_skip = dg_in_size - remaining_bytes; 334 if (current_dg_in_buffer_size != dg_in_buffer_size) 335 current_dg_in_buffer_size = dg_in_buffer_size; 336 337 for (;;) { 338 ioread8_rep(vmci_dev->iobase + 339 VMCI_DATA_IN_ADDR, 340 vmci_dev->data_buffer, 341 current_dg_in_buffer_size); 342 if (bytes_to_skip <= current_dg_in_buffer_size) 343 break; 344 345 bytes_to_skip -= current_dg_in_buffer_size; 346 } 347 dg = (struct vmci_datagram *)(dg_in_buffer + 348 bytes_to_skip); 349 } 350 351 remaining_bytes = 352 (size_t) (dg_in_buffer + current_dg_in_buffer_size - 353 (u8 *)dg); 354 355 if (remaining_bytes < VMCI_DG_HEADERSIZE) { 356 /* Get the next batch of datagrams. */ 357 358 ioread8_rep(vmci_dev->iobase + VMCI_DATA_IN_ADDR, 359 vmci_dev->data_buffer, 360 current_dg_in_buffer_size); 361 dg = (struct vmci_datagram *)dg_in_buffer; 362 remaining_bytes = current_dg_in_buffer_size; 363 } 364 } 365 } 366 367 /* 368 * Scans the notification bitmap for raised flags, clears them 369 * and handles the notifications. 370 */ 371 static void vmci_process_bitmap(unsigned long data) 372 { 373 struct vmci_guest_device *dev = (struct vmci_guest_device *)data; 374 375 if (!dev->notification_bitmap) { 376 dev_dbg(dev->dev, "No bitmap present in %s\n", __func__); 377 return; 378 } 379 380 vmci_dbell_scan_notification_entries(dev->notification_bitmap); 381 } 382 383 /* 384 * Interrupt handler for legacy or MSI interrupt, or for first MSI-X 385 * interrupt (vector VMCI_INTR_DATAGRAM). 386 */ 387 static irqreturn_t vmci_interrupt(int irq, void *_dev) 388 { 389 struct vmci_guest_device *dev = _dev; 390 391 /* 392 * If we are using MSI-X with exclusive vectors then we simply schedule 393 * the datagram tasklet, since we know the interrupt was meant for us. 394 * Otherwise we must read the ICR to determine what to do. 395 */ 396 397 if (dev->exclusive_vectors) { 398 tasklet_schedule(&dev->datagram_tasklet); 399 } else { 400 unsigned int icr; 401 402 /* Acknowledge interrupt and determine what needs doing. */ 403 icr = vmci_read_reg(dev, VMCI_ICR_ADDR); 404 if (icr == 0 || icr == ~0) 405 return IRQ_NONE; 406 407 if (icr & VMCI_ICR_DATAGRAM) { 408 tasklet_schedule(&dev->datagram_tasklet); 409 icr &= ~VMCI_ICR_DATAGRAM; 410 } 411 412 if (icr & VMCI_ICR_NOTIFICATION) { 413 tasklet_schedule(&dev->bm_tasklet); 414 icr &= ~VMCI_ICR_NOTIFICATION; 415 } 416 417 if (icr != 0) 418 dev_warn(dev->dev, 419 "Ignoring unknown interrupt cause (%d)\n", 420 icr); 421 } 422 423 return IRQ_HANDLED; 424 } 425 426 /* 427 * Interrupt handler for MSI-X interrupt vector VMCI_INTR_NOTIFICATION, 428 * which is for the notification bitmap. Will only get called if we are 429 * using MSI-X with exclusive vectors. 430 */ 431 static irqreturn_t vmci_interrupt_bm(int irq, void *_dev) 432 { 433 struct vmci_guest_device *dev = _dev; 434 435 /* For MSI-X we can just assume it was meant for us. */ 436 tasklet_schedule(&dev->bm_tasklet); 437 438 return IRQ_HANDLED; 439 } 440 441 /* 442 * Most of the initialization at module load time is done here. 443 */ 444 static int vmci_guest_probe_device(struct pci_dev *pdev, 445 const struct pci_device_id *id) 446 { 447 struct vmci_guest_device *vmci_dev; 448 void __iomem *iobase = NULL; 449 void __iomem *mmio_base = NULL; 450 unsigned int capabilities; 451 unsigned int caps_in_use; 452 unsigned long cmd; 453 int vmci_err; 454 int error; 455 456 dev_dbg(&pdev->dev, "Probing for vmci/PCI guest device\n"); 457 458 error = pcim_enable_device(pdev); 459 if (error) { 460 dev_err(&pdev->dev, 461 "Failed to enable VMCI device: %d\n", error); 462 return error; 463 } 464 465 /* 466 * The VMCI device with mmio access to registers requests 256KB 467 * for BAR1. If present, driver will use new VMCI device 468 * functionality for register access and datagram send/recv. 469 */ 470 471 if (pci_resource_len(pdev, 1) == VMCI_WITH_MMIO_ACCESS_BAR_SIZE) { 472 dev_info(&pdev->dev, "MMIO register access is available\n"); 473 mmio_base = pci_iomap_range(pdev, 1, VMCI_MMIO_ACCESS_OFFSET, 474 VMCI_MMIO_ACCESS_SIZE); 475 /* If the map fails, we fall back to IOIO access. */ 476 if (!mmio_base) 477 dev_warn(&pdev->dev, "Failed to map MMIO register access\n"); 478 } 479 480 if (!mmio_base) { 481 error = pcim_iomap_regions(pdev, BIT(0), KBUILD_MODNAME); 482 if (error) { 483 dev_err(&pdev->dev, "Failed to reserve/map IO regions\n"); 484 return error; 485 } 486 iobase = pcim_iomap_table(pdev)[0]; 487 } 488 489 vmci_dev = devm_kzalloc(&pdev->dev, sizeof(*vmci_dev), GFP_KERNEL); 490 if (!vmci_dev) { 491 dev_err(&pdev->dev, 492 "Can't allocate memory for VMCI device\n"); 493 return -ENOMEM; 494 } 495 496 vmci_dev->dev = &pdev->dev; 497 vmci_dev->exclusive_vectors = false; 498 vmci_dev->iobase = iobase; 499 vmci_dev->mmio_base = mmio_base; 500 501 tasklet_init(&vmci_dev->datagram_tasklet, 502 vmci_dispatch_dgs, (unsigned long)vmci_dev); 503 tasklet_init(&vmci_dev->bm_tasklet, 504 vmci_process_bitmap, (unsigned long)vmci_dev); 505 506 vmci_dev->data_buffer = vmalloc(VMCI_MAX_DG_SIZE); 507 if (!vmci_dev->data_buffer) { 508 dev_err(&pdev->dev, 509 "Can't allocate memory for datagram buffer\n"); 510 return -ENOMEM; 511 } 512 513 pci_set_master(pdev); /* To enable queue_pair functionality. */ 514 515 /* 516 * Verify that the VMCI Device supports the capabilities that 517 * we need. If the device is missing capabilities that we would 518 * like to use, check for fallback capabilities and use those 519 * instead (so we can run a new VM on old hosts). Fail the load if 520 * a required capability is missing and there is no fallback. 521 * 522 * Right now, we need datagrams. There are no fallbacks. 523 */ 524 capabilities = vmci_read_reg(vmci_dev, VMCI_CAPS_ADDR); 525 if (!(capabilities & VMCI_CAPS_DATAGRAM)) { 526 dev_err(&pdev->dev, "Device does not support datagrams\n"); 527 error = -ENXIO; 528 goto err_free_data_buffer; 529 } 530 caps_in_use = VMCI_CAPS_DATAGRAM; 531 532 /* 533 * Use 64-bit PPNs if the device supports. 534 * 535 * There is no check for the return value of dma_set_mask_and_coherent 536 * since this driver can handle the default mask values if 537 * dma_set_mask_and_coherent fails. 538 */ 539 if (capabilities & VMCI_CAPS_PPN64) { 540 dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(64)); 541 use_ppn64 = true; 542 caps_in_use |= VMCI_CAPS_PPN64; 543 } else { 544 dma_set_mask_and_coherent(&pdev->dev, DMA_BIT_MASK(44)); 545 use_ppn64 = false; 546 } 547 548 /* 549 * If the hardware supports notifications, we will use that as 550 * well. 551 */ 552 if (capabilities & VMCI_CAPS_NOTIFICATIONS) { 553 vmci_dev->notification_bitmap = dma_alloc_coherent( 554 &pdev->dev, PAGE_SIZE, &vmci_dev->notification_base, 555 GFP_KERNEL); 556 if (!vmci_dev->notification_bitmap) { 557 dev_warn(&pdev->dev, 558 "Unable to allocate notification bitmap\n"); 559 } else { 560 memset(vmci_dev->notification_bitmap, 0, PAGE_SIZE); 561 caps_in_use |= VMCI_CAPS_NOTIFICATIONS; 562 } 563 } 564 565 dev_info(&pdev->dev, "Using capabilities 0x%x\n", caps_in_use); 566 567 /* Let the host know which capabilities we intend to use. */ 568 vmci_write_reg(vmci_dev, caps_in_use, VMCI_CAPS_ADDR); 569 570 /* Set up global device so that we can start sending datagrams */ 571 spin_lock_irq(&vmci_dev_spinlock); 572 vmci_dev_g = vmci_dev; 573 vmci_pdev = pdev; 574 spin_unlock_irq(&vmci_dev_spinlock); 575 576 /* 577 * Register notification bitmap with device if that capability is 578 * used. 579 */ 580 if (caps_in_use & VMCI_CAPS_NOTIFICATIONS) { 581 unsigned long bitmap_ppn = 582 vmci_dev->notification_base >> PAGE_SHIFT; 583 if (!vmci_dbell_register_notification_bitmap(bitmap_ppn)) { 584 dev_warn(&pdev->dev, 585 "VMCI device unable to register notification bitmap with PPN 0x%lx\n", 586 bitmap_ppn); 587 error = -ENXIO; 588 goto err_remove_vmci_dev_g; 589 } 590 } 591 592 /* Check host capabilities. */ 593 error = vmci_check_host_caps(pdev); 594 if (error) 595 goto err_remove_bitmap; 596 597 /* Enable device. */ 598 599 /* 600 * We subscribe to the VMCI_EVENT_CTX_ID_UPDATE here so we can 601 * update the internal context id when needed. 602 */ 603 vmci_err = vmci_event_subscribe(VMCI_EVENT_CTX_ID_UPDATE, 604 vmci_guest_cid_update, NULL, 605 &ctx_update_sub_id); 606 if (vmci_err < VMCI_SUCCESS) 607 dev_warn(&pdev->dev, 608 "Failed to subscribe to event (type=%d): %d\n", 609 VMCI_EVENT_CTX_ID_UPDATE, vmci_err); 610 611 /* 612 * Enable interrupts. Try MSI-X first, then MSI, and then fallback on 613 * legacy interrupts. 614 */ 615 error = pci_alloc_irq_vectors(pdev, VMCI_MAX_INTRS, VMCI_MAX_INTRS, 616 PCI_IRQ_MSIX); 617 if (error < 0) { 618 error = pci_alloc_irq_vectors(pdev, 1, 1, 619 PCI_IRQ_MSIX | PCI_IRQ_MSI | PCI_IRQ_LEGACY); 620 if (error < 0) 621 goto err_remove_bitmap; 622 } else { 623 vmci_dev->exclusive_vectors = true; 624 } 625 626 /* 627 * Request IRQ for legacy or MSI interrupts, or for first 628 * MSI-X vector. 629 */ 630 error = request_irq(pci_irq_vector(pdev, 0), vmci_interrupt, 631 IRQF_SHARED, KBUILD_MODNAME, vmci_dev); 632 if (error) { 633 dev_err(&pdev->dev, "Irq %u in use: %d\n", 634 pci_irq_vector(pdev, 0), error); 635 goto err_disable_msi; 636 } 637 638 /* 639 * For MSI-X with exclusive vectors we need to request an 640 * interrupt for each vector so that we get a separate 641 * interrupt handler routine. This allows us to distinguish 642 * between the vectors. 643 */ 644 if (vmci_dev->exclusive_vectors) { 645 error = request_irq(pci_irq_vector(pdev, 1), 646 vmci_interrupt_bm, 0, KBUILD_MODNAME, 647 vmci_dev); 648 if (error) { 649 dev_err(&pdev->dev, 650 "Failed to allocate irq %u: %d\n", 651 pci_irq_vector(pdev, 1), error); 652 goto err_free_irq; 653 } 654 } 655 656 dev_dbg(&pdev->dev, "Registered device\n"); 657 658 atomic_inc(&vmci_num_guest_devices); 659 660 /* Enable specific interrupt bits. */ 661 cmd = VMCI_IMR_DATAGRAM; 662 if (caps_in_use & VMCI_CAPS_NOTIFICATIONS) 663 cmd |= VMCI_IMR_NOTIFICATION; 664 vmci_write_reg(vmci_dev, cmd, VMCI_IMR_ADDR); 665 666 /* Enable interrupts. */ 667 vmci_write_reg(vmci_dev, VMCI_CONTROL_INT_ENABLE, VMCI_CONTROL_ADDR); 668 669 pci_set_drvdata(pdev, vmci_dev); 670 671 vmci_call_vsock_callback(false); 672 return 0; 673 674 err_free_irq: 675 free_irq(pci_irq_vector(pdev, 0), vmci_dev); 676 tasklet_kill(&vmci_dev->datagram_tasklet); 677 tasklet_kill(&vmci_dev->bm_tasklet); 678 679 err_disable_msi: 680 pci_free_irq_vectors(pdev); 681 682 vmci_err = vmci_event_unsubscribe(ctx_update_sub_id); 683 if (vmci_err < VMCI_SUCCESS) 684 dev_warn(&pdev->dev, 685 "Failed to unsubscribe from event (type=%d) with subscriber (ID=0x%x): %d\n", 686 VMCI_EVENT_CTX_ID_UPDATE, ctx_update_sub_id, vmci_err); 687 688 err_remove_bitmap: 689 if (vmci_dev->notification_bitmap) { 690 vmci_write_reg(vmci_dev, VMCI_CONTROL_RESET, VMCI_CONTROL_ADDR); 691 dma_free_coherent(&pdev->dev, PAGE_SIZE, 692 vmci_dev->notification_bitmap, 693 vmci_dev->notification_base); 694 } 695 696 err_remove_vmci_dev_g: 697 spin_lock_irq(&vmci_dev_spinlock); 698 vmci_pdev = NULL; 699 vmci_dev_g = NULL; 700 spin_unlock_irq(&vmci_dev_spinlock); 701 702 err_free_data_buffer: 703 vfree(vmci_dev->data_buffer); 704 705 /* The rest are managed resources and will be freed by PCI core */ 706 return error; 707 } 708 709 static void vmci_guest_remove_device(struct pci_dev *pdev) 710 { 711 struct vmci_guest_device *vmci_dev = pci_get_drvdata(pdev); 712 int vmci_err; 713 714 dev_dbg(&pdev->dev, "Removing device\n"); 715 716 atomic_dec(&vmci_num_guest_devices); 717 718 vmci_qp_guest_endpoints_exit(); 719 720 vmci_err = vmci_event_unsubscribe(ctx_update_sub_id); 721 if (vmci_err < VMCI_SUCCESS) 722 dev_warn(&pdev->dev, 723 "Failed to unsubscribe from event (type=%d) with subscriber (ID=0x%x): %d\n", 724 VMCI_EVENT_CTX_ID_UPDATE, ctx_update_sub_id, vmci_err); 725 726 spin_lock_irq(&vmci_dev_spinlock); 727 vmci_dev_g = NULL; 728 vmci_pdev = NULL; 729 spin_unlock_irq(&vmci_dev_spinlock); 730 731 dev_dbg(&pdev->dev, "Resetting vmci device\n"); 732 vmci_write_reg(vmci_dev, VMCI_CONTROL_RESET, VMCI_CONTROL_ADDR); 733 734 /* 735 * Free IRQ and then disable MSI/MSI-X as appropriate. For 736 * MSI-X, we might have multiple vectors, each with their own 737 * IRQ, which we must free too. 738 */ 739 if (vmci_dev->exclusive_vectors) 740 free_irq(pci_irq_vector(pdev, 1), vmci_dev); 741 free_irq(pci_irq_vector(pdev, 0), vmci_dev); 742 pci_free_irq_vectors(pdev); 743 744 tasklet_kill(&vmci_dev->datagram_tasklet); 745 tasklet_kill(&vmci_dev->bm_tasklet); 746 747 if (vmci_dev->notification_bitmap) { 748 /* 749 * The device reset above cleared the bitmap state of the 750 * device, so we can safely free it here. 751 */ 752 753 dma_free_coherent(&pdev->dev, PAGE_SIZE, 754 vmci_dev->notification_bitmap, 755 vmci_dev->notification_base); 756 } 757 758 vfree(vmci_dev->data_buffer); 759 760 /* The rest are managed resources and will be freed by PCI core */ 761 } 762 763 static const struct pci_device_id vmci_ids[] = { 764 { PCI_DEVICE(PCI_VENDOR_ID_VMWARE, PCI_DEVICE_ID_VMWARE_VMCI), }, 765 { 0 }, 766 }; 767 MODULE_DEVICE_TABLE(pci, vmci_ids); 768 769 static struct pci_driver vmci_guest_driver = { 770 .name = KBUILD_MODNAME, 771 .id_table = vmci_ids, 772 .probe = vmci_guest_probe_device, 773 .remove = vmci_guest_remove_device, 774 }; 775 776 int __init vmci_guest_init(void) 777 { 778 return pci_register_driver(&vmci_guest_driver); 779 } 780 781 void __exit vmci_guest_exit(void) 782 { 783 pci_unregister_driver(&vmci_guest_driver); 784 } 785