virpci.c 92.9 KB
Newer Older
1
/*
2 3
 * virpci.c: helper APIs for managing host PCI devices
 *
4
 * Copyright (C) 2009-2015 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
17
 * License along with this library.  If not, see
O
Osier Yang 已提交
18
 * <http://www.gnu.org/licenses/>.
19 20 21 22
 */

#include <config.h>

23
#include "virpci.h"
24
#include "virnetdev.h"
25 26 27 28 29 30 31 32

#include <dirent.h>
#include <fcntl.h>
#include <inttypes.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>

33
#include "dirname.h"
34
#include "virlog.h"
35
#include "vircommand.h"
36
#include "virerror.h"
E
Eric Blake 已提交
37
#include "virfile.h"
38
#include "virkmod.h"
39 40
#include "virstring.h"
#include "virutil.h"
41

42 43
VIR_LOG_INIT("util.pci");

44 45 46 47
#define PCI_SYSFS "/sys/bus/pci/"
#define PCI_ID_LEN 10   /* "XXXX XXXX" */
#define PCI_ADDR_LEN 13 /* "XXXX:XX:XX.X" */

48
VIR_ENUM_IMPL(virPCIELinkSpeed, VIR_PCIE_LINK_SPEED_LAST,
49 50
              "", "2.5", "5", "8", "16",
);
51

52 53 54 55 56
VIR_ENUM_IMPL(virPCIStubDriver, VIR_PCI_STUB_DRIVER_LAST,
              "none",
              "pciback", /* XEN */
              "pci-stub", /* KVM */
              "vfio-pci", /* VFIO */
57
);
58

59 60 61 62
VIR_ENUM_IMPL(virPCIHeader, VIR_PCI_HEADER_LAST,
              "endpoint",
              "pci-bridge",
              "cardbus-bridge",
63
);
64

65
struct _virPCIDevice {
66
    virPCIDeviceAddress address;
67 68 69

    char          name[PCI_ADDR_LEN]; /* domain:bus:slot.function */
    char          id[PCI_ID_LEN];     /* product vendor */
E
Eric Blake 已提交
70
    char          *path;
C
Chunyan Liu 已提交
71 72 73 74

    /* The driver:domain which uses the device */
    char          *used_by_drvname;
    char          *used_by_domname;
75

76 77
    unsigned int  pcie_cap_pos;
    unsigned int  pci_pm_cap_pos;
78 79
    bool          has_flr;
    bool          has_pm_reset;
80
    bool          managed;
81 82

    virPCIStubDriver stubDriver;
83 84

    /* used by reattach function */
85 86 87
    bool          unbind_from_stub;
    bool          remove_slot;
    bool          reprobe;
88 89
};

90
struct _virPCIDeviceList {
91 92
    virObjectLockable parent;

93
    size_t count;
94
    virPCIDevicePtr *devs;
95 96 97
};


98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114
/* For virReportOOMError()  and virReportSystemError() */
#define VIR_FROM_THIS VIR_FROM_NONE

/* Specifications referenced in comments:
 *  PCI30  - PCI Local Bus Specification 3.0
 *  PCIe20 - PCI Express Base Specification 2.0
 *  BR12   - PCI-to-PCI Bridge Architecture Specification 1.2
 *  PM12   - PCI Bus Power Management Interface Specification 1.2
 *  ECN_AF - Advanced Capabilities for Conventional PCI ECN
 */

/* Type 0 config space header length; PCI30 Section 6.1 Configuration Space Organization */
#define PCI_CONF_LEN            0x100
#define PCI_CONF_HEADER_LEN     0x40

/* PCI30 6.2.1 */
#define PCI_HEADER_TYPE         0x0e    /* Header type */
115 116 117
#define PCI_HEADER_TYPE_BRIDGE 0x1
#define PCI_HEADER_TYPE_MASK   0x7f
#define PCI_HEADER_TYPE_MULTI  0x80
118 119 120 121 122 123 124 125 126

/* PCI30 6.2.1  Device Identification */
#define PCI_CLASS_DEVICE        0x0a    /* Device class */

/* Class Code for bridge; PCI30 D.7  Base Class 06h */
#define PCI_CLASS_BRIDGE_PCI    0x0604

/* PCI30 6.2.3  Device Status */
#define PCI_STATUS              0x06    /* 16 bits */
127
#define PCI_STATUS_CAP_LIST    0x10    /* Support Capability List */
128 129 130

/* PCI30 6.7  Capabilities List */
#define PCI_CAPABILITY_LIST     0x34    /* Offset of first capability list entry */
131
#define PCI_CAP_FLAGS           2       /* Capability defined flags (16 bits) */
132 133 134 135 136 137 138 139 140 141

/* PM12 3.2.1  Capability Identifier */
#define PCI_CAP_ID_PM           0x01    /* Power Management */
/* PCI30 H Capability IDs */
#define PCI_CAP_ID_EXP          0x10    /* PCI Express */
/* ECN_AF 6.x.1.1  Capability ID for AF */
#define PCI_CAP_ID_AF           0x13    /* Advanced Features */

/* PCIe20 7.8.3  Device Capabilities Register (Offset 04h) */
#define PCI_EXP_DEVCAP          0x4     /* Device capabilities */
142 143
#define PCI_EXP_DEVCAP_FLR     (1<<28)  /* Function Level Reset */
#define PCI_EXP_LNKCAP          0xc     /* Link Capabilities */
144
#define PCI_EXP_LNKCAP_SPEED    0x0000f /* Maximum Link Speed */
145 146 147 148
#define PCI_EXP_LNKCAP_WIDTH    0x003f0 /* Maximum Link Width */
#define PCI_EXP_LNKSTA          0x12    /* Link Status */
#define PCI_EXP_LNKSTA_SPEED    0x000f  /* Negotiated Link Speed */
#define PCI_EXP_LNKSTA_WIDTH    0x03f0  /* Negotiated Link Width */
149 150 151 152 153 154 155

/* Header type 1 BR12 3.2 PCI-to-PCI Bridge Configuration Space Header Format */
#define PCI_PRIMARY_BUS         0x18    /* BR12 3.2.5.2 Primary bus number */
#define PCI_SECONDARY_BUS       0x19    /* BR12 3.2.5.3 Secondary bus number */
#define PCI_SUBORDINATE_BUS     0x1a    /* BR12 3.2.5.4 Highest bus number behind the bridge */
#define PCI_BRIDGE_CONTROL      0x3e
/* BR12 3.2.5.18  Bridge Control Register */
156
#define PCI_BRIDGE_CTL_RESET   0x40    /* Secondary bus reset */
157 158 159

/* PM12 3.2.4  Power Management Control/Status (Offset = 4) */
#define PCI_PM_CTRL                4    /* PM control and status register */
160 161 162 163
#define PCI_PM_CTRL_STATE_MASK    0x3  /* Current power state (D0 to D3) */
#define PCI_PM_CTRL_STATE_D0      0x0  /* D0 state */
#define PCI_PM_CTRL_STATE_D3hot   0x3  /* D3 state */
#define PCI_PM_CTRL_NO_SOFT_RESET 0x8  /* No reset for D3hot->D0 */
164 165 166

/* ECN_AF 6.x.1  Advanced Features Capability Structure */
#define PCI_AF_CAP              0x3     /* Advanced features capabilities */
167
#define PCI_AF_CAP_FLR         0x2     /* Function Level Reset */
168

J
Jiri Denemark 已提交
169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185
#define PCI_EXP_FLAGS           0x2
#define PCI_EXP_FLAGS_TYPE      0x00f0
#define PCI_EXP_TYPE_DOWNSTREAM 0x6

#define PCI_EXT_CAP_BASE          0x100
#define PCI_EXT_CAP_LIMIT         0x1000
#define PCI_EXT_CAP_ID_MASK       0x0000ffff
#define PCI_EXT_CAP_OFFSET_SHIFT  20
#define PCI_EXT_CAP_OFFSET_MASK   0x00000ffc

#define PCI_EXT_CAP_ID_ACS      0x000d
#define PCI_EXT_ACS_CTRL        0x06

#define PCI_EXT_CAP_ACS_SV      0x01
#define PCI_EXT_CAP_ACS_RR      0x04
#define PCI_EXT_CAP_ACS_CR      0x08
#define PCI_EXT_CAP_ACS_UF      0x10
186 187 188
#define PCI_EXT_CAP_ACS_ENABLED (PCI_EXT_CAP_ACS_SV | \
                                 PCI_EXT_CAP_ACS_RR | \
                                 PCI_EXT_CAP_ACS_CR | \
J
Jiri Denemark 已提交
189 190
                                 PCI_EXT_CAP_ACS_UF)

191 192 193
#define PCI_EXP_TYPE_ROOT_INT_EP 0x9    /* Root Complex Integrated Endpoint */
#define PCI_EXP_TYPE_ROOT_EC 0xa        /* Root Complex Event Collector */

194 195 196 197 198 199
static virClassPtr virPCIDeviceListClass;

static void virPCIDeviceListDispose(void *obj);

static int virPCIOnceInit(void)
{
200
    if (!VIR_CLASS_NEW(virPCIDeviceList, virClassForObjectLockable()))
201 202 203 204 205
        return -1;

    return 0;
}

206
VIR_ONCE_GLOBAL_INIT(virPCI);
207

L
Laine Stump 已提交
208

209 210
static char *
virPCIDriverDir(const char *driver)
L
Laine Stump 已提交
211
{
212
    char *buffer;
L
Laine Stump 已提交
213

214 215
    ignore_value(virAsprintf(&buffer, PCI_SYSFS "drivers/%s", driver));
    return buffer;
L
Laine Stump 已提交
216 217 218
}


219 220
static char *
virPCIDriverFile(const char *driver, const char *file)
L
Laine Stump 已提交
221
{
222
    char *buffer;
L
Laine Stump 已提交
223

224 225
    ignore_value(virAsprintf(&buffer, PCI_SYSFS "drivers/%s/%s", driver, file));
    return buffer;
L
Laine Stump 已提交
226 227 228
}


229 230
static char *
virPCIFile(const char *device, const char *file)
L
Laine Stump 已提交
231
{
232
    char *buffer;
L
Laine Stump 已提交
233

234 235
    ignore_value(virAsprintf(&buffer, PCI_SYSFS "devices/%s/%s", device, file));
    return buffer;
L
Laine Stump 已提交
236 237 238 239 240 241 242 243 244 245
}


/* virPCIDeviceGetDriverPathAndName - put the path to the driver
 * directory of the driver in use for this device in @path and the
 * name of the driver in @name. Both could be NULL if it's not bound
 * to any driver.
 *
 * Return 0 for success, -1 for error.
 */
246
int
L
Laine Stump 已提交
247 248 249
virPCIDeviceGetDriverPathAndName(virPCIDevicePtr dev, char **path, char **name)
{
    int ret = -1;
250
    VIR_AUTOFREE(char *) drvlink = NULL;
L
Laine Stump 已提交
251 252 253

    *path = *name = NULL;
    /* drvlink = "/sys/bus/pci/dddd:bb:ss.ff/driver" */
254
    if (!(drvlink = virPCIFile(dev->name, "driver")))
L
Laine Stump 已提交
255 256
        goto cleanup;

257 258 259 260 261
    if (!virFileExists(drvlink)) {
        ret = 0;
        goto cleanup;
    }

L
Laine Stump 已提交
262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280
    if (virFileIsLink(drvlink) != 1) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Invalid device %s driver file %s is not a symlink"),
                       dev->name, drvlink);
        goto cleanup;
    }
    if (virFileResolveLink(drvlink, path) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to resolve device %s driver symlink %s"),
                       dev->name, drvlink);
        goto cleanup;
    }
    /* path = "/sys/bus/pci/drivers/${drivername}" */

    if (VIR_STRDUP(*name, last_component(*path)) < 0)
        goto cleanup;
    /* name = "${drivername}" */

    ret = 0;
281
 cleanup:
L
Laine Stump 已提交
282 283 284 285 286 287 288 289
    if (ret < 0) {
        VIR_FREE(*path);
        VIR_FREE(*name);
    }
    return ret;
}


290
static int
291
virPCIDeviceConfigOpen(virPCIDevicePtr dev, bool fatal)
292 293 294 295
{
    int fd;

    fd = open(dev->path, O_RDWR);
296

297
    if (fd < 0) {
298 299 300 301 302 303 304 305 306
        if (fatal) {
            virReportSystemError(errno,
                                 _("Failed to open config space file '%s'"),
                                 dev->path);
        } else {
            char ebuf[1024];
            VIR_WARN("Failed to open config space file '%s': %s",
                     dev->path, virStrerror(errno, ebuf, sizeof(ebuf)));
        }
307 308
        return -1;
    }
309

310
    VIR_DEBUG("%s %s: opened %s", dev->id, dev->name, dev->path);
311
    return fd;
312 313
}

314
static void
315
virPCIDeviceConfigClose(virPCIDevicePtr dev, int cfgfd)
316
{
317 318 319 320 321
    if (VIR_CLOSE(cfgfd) < 0) {
        char ebuf[1024];
        VIR_WARN("Failed to close config space file '%s': %s",
                 dev->path, virStrerror(errno, ebuf, sizeof(ebuf)));
    }
322 323
}

324

325
static int
326 327
virPCIDeviceRead(virPCIDevicePtr dev,
                 int cfgfd,
328
                 unsigned int pos,
329
                 uint8_t *buf,
330
                 unsigned int buflen)
331 332 333
{
    memset(buf, 0, buflen);

334 335
    if (lseek(cfgfd, pos, SEEK_SET) != pos ||
        saferead(cfgfd, buf, buflen) != buflen) {
336
        char ebuf[1024];
337
        VIR_WARN("Failed to read from '%s' : %s", dev->path,
338 339 340 341 342 343 344
                 virStrerror(errno, ebuf, sizeof(ebuf)));
        return -1;
    }
    return 0;
}

static uint8_t
345
virPCIDeviceRead8(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
346 347
{
    uint8_t buf;
348
    virPCIDeviceRead(dev, cfgfd, pos, &buf, sizeof(buf));
349 350 351 352
    return buf;
}

static uint16_t
353
virPCIDeviceRead16(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
354 355
{
    uint8_t buf[2];
356
    virPCIDeviceRead(dev, cfgfd, pos, &buf[0], sizeof(buf));
357 358 359 360
    return (buf[0] << 0) | (buf[1] << 8);
}

static uint32_t
361
virPCIDeviceRead32(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
362 363
{
    uint8_t buf[4];
364
    virPCIDeviceRead(dev, cfgfd, pos, &buf[0], sizeof(buf));
365 366 367
    return (buf[0] << 0) | (buf[1] << 8) | (buf[2] << 16) | (buf[3] << 24);
}

368 369 370
static int
virPCIDeviceReadClass(virPCIDevicePtr dev, uint16_t *device_class)
{
371 372
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) id_str = NULL;
373 374
    unsigned int value;

375
    if (!(path = virPCIFile(dev->name, "class")))
376
        return -1;
377 378 379

    /* class string is '0xNNNNNN\n' ... i.e. 9 bytes */
    if (virFileReadAll(path, 9, &id_str) < 0)
380
        return -1;
381 382 383 384 385 386

    id_str[8] = '\0';
    if (virStrToLong_ui(id_str, NULL, 16, &value) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unusual value in %s/devices/%s/class: %s"),
                       PCI_SYSFS, dev->name, id_str);
387
        return -1;
388 389 390
    }

    *device_class = (value >> 8) & 0xFFFF;
391
    return 0;
392 393
}

394
static int
395 396
virPCIDeviceWrite(virPCIDevicePtr dev,
                  int cfgfd,
397
                  unsigned int pos,
398
                  uint8_t *buf,
399
                  unsigned int buflen)
400
{
401 402
    if (lseek(cfgfd, pos, SEEK_SET) != pos ||
        safewrite(cfgfd, buf, buflen) != buflen) {
403
        char ebuf[1024];
404
        VIR_WARN("Failed to write to '%s' : %s", dev->path,
405 406 407 408 409 410 411
                 virStrerror(errno, ebuf, sizeof(ebuf)));
        return -1;
    }
    return 0;
}

static void
412
virPCIDeviceWrite16(virPCIDevicePtr dev, int cfgfd, unsigned int pos, uint16_t val)
413 414
{
    uint8_t buf[2] = { (val >> 0), (val >> 8) };
415
    virPCIDeviceWrite(dev, cfgfd, pos, &buf[0], sizeof(buf));
416 417 418
}

static void
419
virPCIDeviceWrite32(virPCIDevicePtr dev, int cfgfd, unsigned int pos, uint32_t val)
420
{
421
    uint8_t buf[4] = { (val >> 0), (val >> 8), (val >> 16), (val >> 24) };
422
    virPCIDeviceWrite(dev, cfgfd, pos, &buf[0], sizeof(buf));
423 424
}

E
Eric Blake 已提交
425 426
typedef int (*virPCIDeviceIterPredicate)(virPCIDevicePtr, virPCIDevicePtr,
                                         void *);
427 428 429 430 431 432 433

/* Iterate over available PCI devices calling @predicate
 * to compare each one to @dev.
 * Return -1 on error since we don't want to assume it is
 * safe to reset if there is an error.
 */
static int
434 435 436 437
virPCIDeviceIterDevices(virPCIDeviceIterPredicate predicate,
                        virPCIDevicePtr dev,
                        virPCIDevicePtr *matched,
                        void *data)
438 439 440
{
    DIR *dir;
    struct dirent *entry;
441
    int ret = 0;
442
    int rc;
443 444 445 446 447

    *matched = NULL;

    VIR_DEBUG("%s %s: iterating over " PCI_SYSFS "devices", dev->id, dev->name);

J
Ján Tomko 已提交
448
    if (virDirOpen(&dir, PCI_SYSFS "devices") < 0)
449 450
        return -1;

E
Eric Blake 已提交
451
    while ((ret = virDirRead(dir, &entry, PCI_SYSFS "devices")) > 0) {
452
        unsigned int domain, bus, slot, function;
453
        VIR_AUTOPTR(virPCIDevice) check = NULL;
454
        char *tmp;
455

456 457 458 459 460 461 462 463 464
        /* expected format: <domain>:<bus>:<slot>.<function> */
        if (/* domain */
            virStrToLong_ui(entry->d_name, &tmp, 16, &domain) < 0 || *tmp != ':' ||
            /* bus */
            virStrToLong_ui(tmp + 1, &tmp, 16, &bus) < 0 || *tmp != ':' ||
            /* slot */
            virStrToLong_ui(tmp + 1, &tmp, 16, &slot) < 0 || *tmp != '.' ||
            /* function */
            virStrToLong_ui(tmp + 1, NULL, 16, &function) < 0) {
465 466 467 468
            VIR_WARN("Unusual entry in " PCI_SYSFS "devices: %s", entry->d_name);
            continue;
        }

469
        check = virPCIDeviceNew(domain, bus, slot, function);
470
        if (!check) {
471 472 473
            ret = -1;
            break;
        }
474

475 476 477 478 479
        rc = predicate(dev, check, data);
        if (rc < 0) {
            /* the predicate returned an error, bail */
            ret = -1;
            break;
480
        } else if (rc == 1) {
481
            VIR_DEBUG("%s %s: iter matched on %s", dev->id, dev->name, check->name);
482
            VIR_STEAL_PTR(*matched, check);
483
            ret = 1;
484 485 486
            break;
        }
    }
J
Ján Tomko 已提交
487
    VIR_DIR_CLOSE(dir);
488
    return ret;
489 490 491
}

static uint8_t
492 493 494
virPCIDeviceFindCapabilityOffset(virPCIDevicePtr dev,
                                 int cfgfd,
                                 unsigned int capability)
495 496 497 498
{
    uint16_t status;
    uint8_t pos;

499
    status = virPCIDeviceRead16(dev, cfgfd, PCI_STATUS);
500 501 502
    if (!(status & PCI_STATUS_CAP_LIST))
        return 0;

503
    pos = virPCIDeviceRead8(dev, cfgfd, PCI_CAPABILITY_LIST);
504 505 506 507 508 509 510 511 512

    /* Zero indicates last capability, capabilities can't
     * be in the config space header and 0xff is returned
     * by the kernel if we don't have access to this region
     *
     * Note: we're not handling loops or extended
     * capabilities here.
     */
    while (pos >= PCI_CONF_HEADER_LEN && pos != 0xff) {
513
        uint8_t capid = virPCIDeviceRead8(dev, cfgfd, pos);
514 515 516 517 518 519
        if (capid == capability) {
            VIR_DEBUG("%s %s: found cap 0x%.2x at 0x%.2x",
                      dev->id, dev->name, capability, pos);
            return pos;
        }

520
        pos = virPCIDeviceRead8(dev, cfgfd, pos + 1);
521 522 523 524 525 526 527
    }

    VIR_DEBUG("%s %s: failed to find cap 0x%.2x", dev->id, dev->name, capability);

    return 0;
}

J
Jiri Denemark 已提交
528
static unsigned int
529 530
virPCIDeviceFindExtendedCapabilityOffset(virPCIDevicePtr dev,
                                         int cfgfd,
531
                                         unsigned int capability)
J
Jiri Denemark 已提交
532 533 534 535 536 537 538 539 540 541
{
    int ttl;
    unsigned int pos;
    uint32_t header;

    /* minimum 8 bytes per capability */
    ttl = (PCI_EXT_CAP_LIMIT - PCI_EXT_CAP_BASE) / 8;
    pos = PCI_EXT_CAP_BASE;

    while (ttl > 0 && pos >= PCI_EXT_CAP_BASE) {
542
        header = virPCIDeviceRead32(dev, cfgfd, pos);
J
Jiri Denemark 已提交
543 544 545 546 547 548 549 550 551 552 553

        if ((header & PCI_EXT_CAP_ID_MASK) == capability)
            return pos;

        pos = (header >> PCI_EXT_CAP_OFFSET_SHIFT) & PCI_EXT_CAP_OFFSET_MASK;
        ttl--;
    }

    return 0;
}

554 555 556 557
/* detects whether this device has FLR.  Returns 0 if the device does
 * not have FLR, 1 if it does, and -1 on error
 */
static int
558
virPCIDeviceDetectFunctionLevelReset(virPCIDevicePtr dev, int cfgfd)
559
{
M
Mark McLoughlin 已提交
560
    uint32_t caps;
561
    uint8_t pos;
562
    VIR_AUTOFREE(char *) path = NULL;
563
    int found;
564 565 566 567 568 569 570 571

    /* The PCIe Function Level Reset capability allows
     * individual device functions to be reset without
     * affecting any other functions on the device or
     * any other devices on the bus. This is only common
     * on SR-IOV NICs at the moment.
     */
    if (dev->pcie_cap_pos) {
572
        caps = virPCIDeviceRead32(dev, cfgfd, dev->pcie_cap_pos + PCI_EXP_DEVCAP);
573 574 575 576 577 578 579 580 581 582
        if (caps & PCI_EXP_DEVCAP_FLR) {
            VIR_DEBUG("%s %s: detected PCIe FLR capability", dev->id, dev->name);
            return 1;
        }
    }

    /* The PCI AF Function Level Reset capability is
     * the same thing, except for conventional PCI
     * devices. This is not common yet.
     */
583
    pos = virPCIDeviceFindCapabilityOffset(dev, cfgfd, PCI_CAP_ID_AF);
584
    if (pos) {
585
        caps = virPCIDeviceRead16(dev, cfgfd, pos + PCI_AF_CAP);
586 587 588 589 590 591
        if (caps & PCI_AF_CAP_FLR) {
            VIR_DEBUG("%s %s: detected PCI FLR capability", dev->id, dev->name);
            return 1;
        }
    }

592 593 594 595 596 597
    /* there are some buggy devices that do support FLR, but forget to
     * advertise that fact in their capabilities.  However, FLR is *required*
     * to be present for virtual functions (VFs), so if we see that this
     * device is a VF, we just assume FLR works
     */

598
    if (virAsprintf(&path, PCI_SYSFS "devices/%s/physfn", dev->name) < 0)
599 600 601 602 603 604 605 606 607
        return -1;

    found = virFileExists(path);
    if (found) {
        VIR_DEBUG("%s %s: buggy device didn't advertise FLR, but is a VF; forcing flr on",
                  dev->id, dev->name);
        return 1;
    }

608 609 610 611 612 613 614 615 616
    VIR_DEBUG("%s %s: no FLR capability found", dev->id, dev->name);

    return 0;
}

/* Require the device has the PCI Power Management capability
 * and that a D3hot->D0 transition will results in a full
 * internal reset, not just a soft reset.
 */
617
static unsigned int
618
virPCIDeviceDetectPowerManagementReset(virPCIDevicePtr dev, int cfgfd)
619 620 621 622 623
{
    if (dev->pci_pm_cap_pos) {
        uint32_t ctl;

        /* require the NO_SOFT_RESET bit is clear */
624
        ctl = virPCIDeviceRead32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL);
625 626 627 628 629 630 631 632 633 634 635
        if (!(ctl & PCI_PM_CTRL_NO_SOFT_RESET)) {
            VIR_DEBUG("%s %s: detected PM reset capability", dev->id, dev->name);
            return 1;
        }
    }

    VIR_DEBUG("%s %s: no PM reset capability found", dev->id, dev->name);

    return 0;
}

636
/* Any active devices on the same domain/bus ? */
637
static int
638
virPCIDeviceSharesBusWithActive(virPCIDevicePtr dev, virPCIDevicePtr check, void *data)
639
{
640
    virPCIDeviceList *inactiveDevs = data;
641

642
    /* Different domain, different bus, or simply identical device */
643 644 645 646
    if (dev->address.domain != check->address.domain ||
        dev->address.bus != check->address.bus ||
        (dev->address.slot == check->address.slot &&
         dev->address.function == check->address.function))
647 648
        return 0;

649
    /* same bus, but inactive, i.e. about to be assigned to guest */
650
    if (inactiveDevs && virPCIDeviceListFind(inactiveDevs, check))
651
        return 0;
652

653
    return 1;
654 655
}

656 657 658
static virPCIDevicePtr
virPCIDeviceBusContainsActiveDevices(virPCIDevicePtr dev,
                                     virPCIDeviceList *inactiveDevs)
659
{
660 661 662
    virPCIDevicePtr active = NULL;
    if (virPCIDeviceIterDevices(virPCIDeviceSharesBusWithActive,
                                dev, &active, inactiveDevs) < 0)
663 664 665 666 667
        return NULL;
    return active;
}

/* Is @check the parent of @dev ? */
668
static int
669
virPCIDeviceIsParent(virPCIDevicePtr dev, virPCIDevicePtr check, void *data)
670 671 672
{
    uint16_t device_class;
    uint8_t header_type, secondary, subordinate;
673
    virPCIDevicePtr *best = data;
674 675
    int ret = 0;
    int fd;
676

677
    if (dev->address.domain != check->address.domain)
678 679
        return 0;

680
    if ((fd = virPCIDeviceConfigOpen(check, false)) < 0)
681 682
        return 0;

683
    /* Is it a bridge? */
684 685
    ret = virPCIDeviceReadClass(check, &device_class);
    if (ret < 0 || device_class != PCI_CLASS_BRIDGE_PCI)
686
        goto cleanup;
687 688

    /* Is it a plane? */
689
    header_type = virPCIDeviceRead8(check, fd, PCI_HEADER_TYPE);
690
    if ((header_type & PCI_HEADER_TYPE_MASK) != PCI_HEADER_TYPE_BRIDGE)
691
        goto cleanup;
692

693 694
    secondary   = virPCIDeviceRead8(check, fd, PCI_SECONDARY_BUS);
    subordinate = virPCIDeviceRead8(check, fd, PCI_SUBORDINATE_BUS);
695

696
    VIR_DEBUG("%s %s: found parent device %s", dev->id, dev->name, check->name);
697

698 699 700
    /* if the secondary bus exactly equals the device's bus, then we found
     * the direct parent.  No further work is necessary
     */
701
    if (dev->address.bus == secondary) {
702 703 704
        ret = 1;
        goto cleanup;
    }
705

706
    /* otherwise, SRIOV allows VFs to be on different buses than their PFs.
707 708 709
     * In this case, what we need to do is look for the "best" match; i.e.
     * the most restrictive match that still satisfies all of the conditions.
     */
710
    if (dev->address.bus > secondary && dev->address.bus <= subordinate) {
711
        if (*best == NULL) {
712 713 714 715
            *best = virPCIDeviceNew(check->address.domain,
                                    check->address.bus,
                                    check->address.slot,
                                    check->address.function);
716 717 718 719 720
            if (*best == NULL) {
                ret = -1;
                goto cleanup;
            }
        } else {
721 722 723 724
            /* OK, we had already recorded a previous "best" match for the
             * parent.  See if the current device is more restrictive than the
             * best, and if so, make it the new best
             */
725 726 727
            int bestfd;
            uint8_t best_secondary;

728
            if ((bestfd = virPCIDeviceConfigOpen(*best, false)) < 0)
729
                goto cleanup;
730 731
            best_secondary = virPCIDeviceRead8(*best, bestfd, PCI_SECONDARY_BUS);
            virPCIDeviceConfigClose(*best, bestfd);
732 733

            if (secondary > best_secondary) {
734
                virPCIDeviceFree(*best);
735 736 737 738
                *best = virPCIDeviceNew(check->address.domain,
                                        check->address.bus,
                                        check->address.slot,
                                        check->address.function);
739 740 741 742
                if (*best == NULL) {
                    ret = -1;
                    goto cleanup;
                }
743 744 745 746
            }
        }
    }

747
 cleanup:
748
    virPCIDeviceConfigClose(check, fd);
749
    return ret;
750 751
}

752
static int
753
virPCIDeviceGetParent(virPCIDevicePtr dev, virPCIDevicePtr *parent)
754
{
755
    virPCIDevicePtr best = NULL;
756 757 758
    int ret;

    *parent = NULL;
759
    ret = virPCIDeviceIterDevices(virPCIDeviceIsParent, dev, parent, &best);
760
    if (ret == 1)
761
        virPCIDeviceFree(best);
762 763 764
    else if (ret == 0)
        *parent = best;
    return ret;
765 766 767 768 769 770
}

/* Secondary Bus Reset is our sledgehammer - it resets all
 * devices behind a bus.
 */
static int
771 772 773
virPCIDeviceTrySecondaryBusReset(virPCIDevicePtr dev,
                                 int cfgfd,
                                 virPCIDeviceList *inactiveDevs)
774
{
775 776
    VIR_AUTOPTR(virPCIDevice) parent = NULL;
    VIR_AUTOPTR(virPCIDevice) conflict = NULL;
777 778 779
    uint8_t config_space[PCI_CONF_LEN];
    uint16_t ctl;
    int ret = -1;
780
    int parentfd;
781

782 783 784
    /* Refuse to do a secondary bus reset if there are other
     * devices/functions behind the bus are used by the host
     * or other guests.
785
     */
786
    if ((conflict = virPCIDeviceBusContainsActiveDevices(dev, inactiveDevs))) {
787
        virReportError(VIR_ERR_INTERNAL_ERROR,
788 789
                       _("Active %s devices on bus with %s, not doing bus reset"),
                       conflict->name, dev->name);
790 791 792 793
        return -1;
    }

    /* Find the parent bus */
794
    if (virPCIDeviceGetParent(dev, &parent) < 0)
795
        return -1;
796
    if (!parent) {
797
        virReportError(VIR_ERR_INTERNAL_ERROR,
798 799
                       _("Failed to find parent device for %s"),
                       dev->name);
800 801
        return -1;
    }
802
    if ((parentfd = virPCIDeviceConfigOpen(parent, true)) < 0)
803
        goto out;
804 805 806 807 808 809 810

    VIR_DEBUG("%s %s: doing a secondary bus reset", dev->id, dev->name);

    /* Save and restore the device's config space; we only do this
     * for the supplied device since we refuse to do a reset if there
     * are multiple devices/functions
     */
811
    if (virPCIDeviceRead(dev, cfgfd, 0, config_space, PCI_CONF_LEN) < 0) {
812
        virReportError(VIR_ERR_INTERNAL_ERROR,
813
                       _("Failed to read PCI config space for %s"),
814
                       dev->name);
815 816 817 818 819 820
        goto out;
    }

    /* Read the control register, set the reset flag, wait 200ms,
     * unset the reset flag and wait 200ms.
     */
821
    ctl = virPCIDeviceRead16(dev, cfgfd, PCI_BRIDGE_CONTROL);
822

823 824
    virPCIDeviceWrite16(parent, parentfd, PCI_BRIDGE_CONTROL,
                        ctl | PCI_BRIDGE_CTL_RESET);
825 826 827

    usleep(200 * 1000); /* sleep 200ms */

828
    virPCIDeviceWrite16(parent, parentfd, PCI_BRIDGE_CONTROL, ctl);
829 830 831

    usleep(200 * 1000); /* sleep 200ms */

832
    if (virPCIDeviceWrite(dev, cfgfd, 0, config_space, PCI_CONF_LEN) < 0) {
833
        virReportError(VIR_ERR_INTERNAL_ERROR,
834 835 836 837
                       _("Failed to restore PCI config space for %s"),
                       dev->name);
        goto out;
    }
838
    ret = 0;
839

840
 out:
841
    virPCIDeviceConfigClose(parent, parentfd);
842 843 844 845 846 847 848 849
    return ret;
}

/* Power management reset attempts to reset a device using a
 * D-state transition from D3hot to D0. Note, in detect_pm_reset()
 * above we require the device supports a full internal reset.
 */
static int
850
virPCIDeviceTryPowerManagementReset(virPCIDevicePtr dev, int cfgfd)
851 852 853 854 855 856 857 858
{
    uint8_t config_space[PCI_CONF_LEN];
    uint32_t ctl;

    if (!dev->pci_pm_cap_pos)
        return -1;

    /* Save and restore the device's config space. */
859
    if (virPCIDeviceRead(dev, cfgfd, 0, &config_space[0], PCI_CONF_LEN) < 0) {
860
        virReportError(VIR_ERR_INTERNAL_ERROR,
861
                       _("Failed to read PCI config space for %s"),
862
                       dev->name);
863 864 865 866 867
        return -1;
    }

    VIR_DEBUG("%s %s: doing a power management reset", dev->id, dev->name);

868
    ctl = virPCIDeviceRead32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL);
869 870
    ctl &= ~PCI_PM_CTRL_STATE_MASK;

871 872
    virPCIDeviceWrite32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL,
                        ctl | PCI_PM_CTRL_STATE_D3hot);
873 874 875

    usleep(10 * 1000); /* sleep 10ms */

876 877
    virPCIDeviceWrite32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL,
                        ctl | PCI_PM_CTRL_STATE_D0);
878 879 880

    usleep(10 * 1000); /* sleep 10ms */

881
    if (virPCIDeviceWrite(dev, cfgfd, 0, &config_space[0], PCI_CONF_LEN) < 0) {
882
        virReportError(VIR_ERR_INTERNAL_ERROR,
883 884 885 886
                       _("Failed to restore PCI config space for %s"),
                       dev->name);
        return -1;
    }
887 888 889 890 891

    return 0;
}

static int
892
virPCIDeviceInit(virPCIDevicePtr dev, int cfgfd)
893
{
894 895
    int flr;

896 897 898
    dev->pcie_cap_pos   = virPCIDeviceFindCapabilityOffset(dev, cfgfd, PCI_CAP_ID_EXP);
    dev->pci_pm_cap_pos = virPCIDeviceFindCapabilityOffset(dev, cfgfd, PCI_CAP_ID_PM);
    flr = virPCIDeviceDetectFunctionLevelReset(dev, cfgfd);
899
    if (flr < 0)
900
        return flr;
901 902
    dev->has_flr        = !!flr;
    dev->has_pm_reset   = !!virPCIDeviceDetectPowerManagementReset(dev, cfgfd);
903

904 905 906 907
    return 0;
}

int
908 909 910
virPCIDeviceReset(virPCIDevicePtr dev,
                  virPCIDeviceList *activeDevs,
                  virPCIDeviceList *inactiveDevs)
911
{
912 913
    VIR_AUTOFREE(char *) drvPath = NULL;
    VIR_AUTOFREE(char *) drvName = NULL;
914
    int ret = -1;
915
    int fd = -1;
916 917 918 919 920 921 922 923 924 925 926 927
    int hdrType = -1;

    if (virPCIGetHeaderType(dev, &hdrType) < 0)
        return -1;

    if (hdrType != VIR_PCI_HEADER_ENDPOINT) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Invalid attempt to reset PCI device %s. "
                         "Only PCI endpoint devices can be reset"),
                       dev->name);
        return -1;
    }
928

929
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
930
        virReportError(VIR_ERR_INTERNAL_ERROR,
931 932 933 934
                       _("Not resetting active device %s"), dev->name);
        return -1;
    }

935 936 937 938 939 940 941 942
    /* If the device is currently bound to vfio-pci, ignore all
     * requests to reset it, since the vfio-pci driver will always
     * reset it whenever appropriate, so doing it ourselves would just
     * be redundant.
     */
    if (virPCIDeviceGetDriverPathAndName(dev, &drvPath, &drvName) < 0)
        goto cleanup;

943
    if (virPCIStubDriverTypeFromString(drvName) == VIR_PCI_STUB_DRIVER_VFIO) {
944 945 946 947 948 949 950
        VIR_DEBUG("Device %s is bound to vfio-pci - skip reset",
                  dev->name);
        ret = 0;
        goto cleanup;
    }
    VIR_DEBUG("Resetting device %s", dev->name);

951
    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
952
        goto cleanup;
953

954
    if (virPCIDeviceInit(dev, fd) < 0)
955 956
        goto cleanup;

957 958 959
    /* KVM will perform FLR when starting and stopping
     * a guest, so there is no need for us to do it here.
     */
960 961 962 963
    if (dev->has_flr) {
        ret = 0;
        goto cleanup;
    }
964

965 966 967 968 969
    /* If the device supports PCI power management reset,
     * that's the next best thing because it only resets
     * the function, not the whole device.
     */
    if (dev->has_pm_reset)
970
        ret = virPCIDeviceTryPowerManagementReset(dev, fd);
971

972
    /* Bus reset is not an option with the root bus */
973
    if (ret < 0 && dev->address.bus != 0)
974
        ret = virPCIDeviceTrySecondaryBusReset(dev, fd, inactiveDevs);
975

976 977
    if (ret < 0) {
        virErrorPtr err = virGetLastError();
978
        virReportError(VIR_ERR_INTERNAL_ERROR,
979 980
                       _("Unable to reset PCI device %s: %s"),
                       dev->name,
981 982
                       err ? err->message :
                       _("no FLR, PM reset or bus reset available"));
983 984
    }

985
 cleanup:
986
    virPCIDeviceConfigClose(dev, fd);
987 988 989
    return ret;
}

990

991
static int
992
virPCIProbeStubDriver(virPCIStubDriver driver)
993
{
994
    const char *drvname = NULL;
995
    VIR_AUTOFREE(char *) drvpath = NULL;
996
    bool probed = false;
997

998 999 1000 1001 1002 1003 1004 1005
    if (driver == VIR_PCI_STUB_DRIVER_NONE ||
        !(drvname = virPCIStubDriverTypeToString(driver))) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s",
                       _("Attempting to use unknown stub driver"));
        return -1;
    }

1006
 recheck:
1007
    if ((drvpath = virPCIDriverDir(drvname)) && virFileExists(drvpath))
1008 1009
        /* driver already loaded, return */
        return 0;
1010 1011

    if (!probed) {
1012
        VIR_AUTOFREE(char *) errbuf = NULL;
1013
        probed = true;
1014 1015
        if ((errbuf = virKModLoad(drvname, true))) {
            VIR_WARN("failed to load driver %s: %s", drvname, errbuf);
1016
            goto cleanup;
1017
        }
1018 1019

        goto recheck;
1020 1021
    }

1022
 cleanup:
1023 1024 1025
    /* If we know failure was because of blacklist, let's report that;
     * otherwise, report a more generic failure message
     */
1026
    if (virKModIsBlacklisted(drvname)) {
1027 1028 1029
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to load PCI stub module %s: "
                         "administratively prohibited"),
1030
                       drvname);
1031 1032 1033
    } else {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to load PCI stub module %s"),
1034
                       drvname);
1035 1036
    }

1037
    return -1;
1038 1039
}

1040
int
1041
virPCIDeviceUnbind(virPCIDevicePtr dev)
1042
{
1043 1044 1045
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) drvpath = NULL;
    VIR_AUTOFREE(char *) driver = NULL;
1046 1047

    if (virPCIDeviceGetDriverPathAndName(dev, &drvpath, &driver) < 0)
1048
        return -1;
1049

1050
    if (!driver)
1051
        /* The device is not bound to any driver */
1052
        return 0;
1053

1054
    if (!(path = virPCIFile(dev->name, "driver/unbind")))
1055
        return -1;
1056 1057 1058 1059 1060 1061

    if (virFileExists(path)) {
        if (virFileWriteStr(path, dev->name, 0) < 0) {
            virReportSystemError(errno,
                                 _("Failed to unbind PCI device '%s' from %s"),
                                 dev->name, driver);
1062
            return -1;
1063 1064 1065
        }
    }

1066
    return 0;
1067 1068
}

1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093

/**
 * virPCIDeviceRebind:
 *  @dev: virPCIDevice object describing the device to rebind
 *
 * unbind a device from its driver, then immediately rebind it.
 *
 * Returns 0 on success, -1 on failure
 */
int virPCIDeviceRebind(virPCIDevicePtr dev)
{
    if (virPCIDeviceUnbind(dev) < 0)
        return -1;

    if (virFileWriteStr(PCI_SYSFS "drivers_probe", dev->name, 0) < 0) {
        virReportSystemError(errno,
                             _("Failed to trigger a probe for PCI device '%s'"),
                             dev->name);
        return -1;
    }

    return 0;
}


1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104
/*
 * Bind a PCI device to a driver using driver_override sysfs interface.
 * E.g.
 *
 *  echo driver-name > /sys/bus/pci/devices/0000:03:00.0/driver_override
 *  echo 0000:03:00.0 > /sys/bus/pci/devices/0000:03:00.0/driver/unbind
 *  echo 0000:03:00.0 > /sys/bus/pci/drivers_probe
 *
 * An empty driverName will cause the device to be bound to its
 * preferred driver.
 */
1105
static int
1106 1107 1108
virPCIDeviceBindWithDriverOverride(virPCIDevicePtr dev,
                                   const char *driverName)
{
1109
    VIR_AUTOFREE(char *) path = NULL;
1110 1111 1112 1113 1114 1115 1116 1117 1118

    if (!(path = virPCIFile(dev->name, "driver_override")))
        return -1;

    if (virFileWriteStr(path, driverName, 0) < 0) {
        virReportSystemError(errno,
                             _("Failed to add driver '%s' to driver_override "
                               " interface of PCI device '%s'"),
                             driverName, dev->name);
1119
        return -1;
1120 1121
    }

1122
    if (virPCIDeviceRebind(dev) < 0)
1123
        return -1;
1124

1125
    return 0;
1126 1127 1128 1129
}

static int
virPCIDeviceUnbindFromStubWithNewid(virPCIDevicePtr dev)
1130 1131
{
    int result = -1;
1132 1133 1134
    VIR_AUTOFREE(char *) drvdir = NULL;
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) driver = NULL;
1135

1136 1137 1138
    /* If the device is currently bound to one of the "well known"
     * stub drivers, then unbind it, otherwise ignore it.
     */
L
Laine Stump 已提交
1139
    if (virPCIDeviceGetDriverPathAndName(dev, &drvdir, &driver) < 0)
1140
        goto cleanup;
E
Eric Blake 已提交
1141

1142 1143
    if (!driver) {
        /* The device is not bound to any driver and we are almost done. */
1144
        VIR_DEBUG("PCI device %s is not bound to any driver", dev->name);
1145 1146 1147
        goto reprobe;
    }

1148 1149
    if (!dev->unbind_from_stub) {
        VIR_DEBUG("Unbind from stub skipped for PCI device %s", dev->name);
1150
        goto remove_slot;
1151
    }
1152

1153
    /* If the device isn't bound to a known stub, skip the unbind. */
1154
    if (virPCIStubDriverTypeFromString(driver) < 0 ||
1155 1156 1157
        virPCIStubDriverTypeFromString(driver) == VIR_PCI_STUB_DRIVER_NONE) {
        VIR_DEBUG("Unbind from stub skipped for PCI device %s because of "
                  "unknown stub driver", dev->name);
1158
        goto remove_slot;
1159
    }
1160

1161 1162
    VIR_DEBUG("Unbinding PCI device %s from stub driver %s",
              dev->name, driver);
1163

1164
    if (virPCIDeviceUnbind(dev) < 0)
1165
        goto cleanup;
1166
    dev->unbind_from_stub = false;
1167

1168
 remove_slot:
1169 1170
    if (!dev->remove_slot) {
        VIR_DEBUG("Slot removal skipped for PCI device %s", dev->name);
1171
        goto reprobe;
1172 1173 1174
    }

    VIR_DEBUG("Removing slot for PCI device %s", dev->name);
1175 1176

    /* Xen's pciback.ko wants you to use remove_slot on the specific device */
1177
    if (!(path = virPCIDriverFile(driver, "remove_slot")))
1178 1179 1180 1181
        goto cleanup;

    if (virFileExists(path) && virFileWriteStr(path, dev->name, 0) < 0) {
        virReportSystemError(errno,
1182
                             _("Failed to remove slot for PCI device '%s' from %s"),
1183 1184 1185
                             dev->name, driver);
        goto cleanup;
    }
1186
    dev->remove_slot = false;
1187

1188
 reprobe:
1189
    if (!dev->reprobe) {
1190
        VIR_DEBUG("Reprobe skipped for PCI device %s", dev->name);
1191 1192 1193
        result = 0;
        goto cleanup;
    }
1194

1195 1196
    VIR_DEBUG("Reprobing for PCI device %s", dev->name);

1197 1198 1199 1200 1201
    /* Trigger a re-probe of the device is not in the stub's dynamic
     * ID table. If the stub is available, but 'remove_id' isn't
     * available, then re-probing would just cause the device to be
     * re-bound to the stub.
     */
1202 1203
    VIR_FREE(path);
    if (driver && !(path = virPCIDriverFile(driver, "remove_id")))
1204 1205
        goto cleanup;

1206
    if (!driver || !virFileExists(drvdir) || virFileExists(path)) {
1207 1208 1209 1210 1211 1212 1213 1214 1215 1216
        if (virFileWriteStr(PCI_SYSFS "drivers_probe", dev->name, 0) < 0) {
            virReportSystemError(errno,
                                 _("Failed to trigger a re-probe for PCI device '%s'"),
                                 dev->name);
            goto cleanup;
        }
    }

    result = 0;

1217
 cleanup:
1218
    /* do not do it again */
1219 1220 1221
    dev->unbind_from_stub = false;
    dev->remove_slot = false;
    dev->reprobe = false;
1222

1223 1224 1225
    return result;
}

1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
static int
virPCIDeviceUnbindFromStubWithOverride(virPCIDevicePtr dev)
{
    if (!dev->unbind_from_stub) {
        VIR_DEBUG("Unbind from stub skipped for PCI device %s", dev->name);
        return 0;
    }

    return virPCIDeviceBindWithDriverOverride(dev, "\n");
}
1236 1237

static int
1238 1239
virPCIDeviceUnbindFromStub(virPCIDevicePtr dev)
{
1240
    VIR_AUTOFREE(char *) path = NULL;
1241 1242 1243 1244 1245 1246 1247 1248 1249

    /*
     * Prefer using the device's driver_override interface, falling back
     * to the unpleasant new_id interface.
     */
    if (!(path = virPCIFile(dev->name, "driver_override")))
        return -1;

    if (virFileExists(path))
1250
        return virPCIDeviceUnbindFromStubWithOverride(dev);
1251

1252
    return virPCIDeviceUnbindFromStubWithNewid(dev);
1253 1254 1255 1256
}

static int
virPCIDeviceBindToStubWithNewid(virPCIDevicePtr dev)
1257
{
1258
    int result = -1;
E
Eric Blake 已提交
1259
    bool reprobe = false;
1260 1261 1262
    VIR_AUTOFREE(char *) stubDriverPath = NULL;
    VIR_AUTOFREE(char *) driverLink = NULL;
    VIR_AUTOFREE(char *) path = NULL; /* reused for different purposes */
1263
    VIR_AUTOPTR(virError) err = NULL;
1264
    const char *stubDriverName = NULL;
1265

1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278
    /* Check the device is configured to use one of the known stub drivers */
    if (dev->stubDriver == VIR_PCI_STUB_DRIVER_NONE) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("No stub driver configured for PCI device %s"),
                       dev->name);
        return -1;
    } else if (!(stubDriverName = virPCIStubDriverTypeToString(dev->stubDriver))) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unknown stub driver configured for PCI device %s"),
                       dev->name);
        return -1;
    }

1279
    if (!(stubDriverPath = virPCIDriverDir(stubDriverName))  ||
1280
        !(driverLink = virPCIFile(dev->name, "driver")))
1281 1282
        goto cleanup;

1283 1284 1285 1286 1287
    if (virFileExists(driverLink)) {
        if (virFileLinkPointsTo(driverLink, stubDriverPath)) {
            /* The device is already bound to the correct driver */
            VIR_DEBUG("Device %s is already bound to %s",
                      dev->name, stubDriverName);
1288 1289 1290
            result = 0;
            goto cleanup;
        }
1291
        reprobe = true;
1292
    }
1293 1294 1295 1296 1297 1298 1299 1300 1301

    /* Add the PCI device ID to the stub's dynamic ID table;
     * this is needed to allow us to bind the device to the stub.
     * Note: if the device is not currently bound to any driver,
     * stub will immediately be bound to the device. Also, note
     * that if a new device with this ID is hotplugged, or if a probe
     * is triggered for such a device, it will also be immediately
     * bound by the stub.
     */
1302
    if (!(path = virPCIDriverFile(stubDriverName, "new_id")))
1303
        goto cleanup;
1304

1305
    if (virFileWriteStr(path, dev->id, 0) < 0) {
1306
        virReportSystemError(errno,
1307
                             _("Failed to add PCI device ID '%s' to %s"),
1308
                             dev->id, stubDriverName);
1309
        goto cleanup;
1310 1311
    }

1312
    /* check whether the device is bound to pci-stub when we write dev->id to
1313
     * ${stubDriver}/new_id.
1314
     */
1315
    if (virFileLinkPointsTo(driverLink, stubDriverPath)) {
1316 1317
        dev->unbind_from_stub = true;
        dev->remove_slot = true;
J
Jiri Denemark 已提交
1318
        result = 0;
1319 1320 1321
        goto remove_id;
    }

1322
    if (virPCIDeviceUnbind(dev) < 0)
J
Jiri Denemark 已提交
1323
        goto remove_id;
1324

1325 1326 1327
    /* If the device was bound to a driver we'll need to reprobe later */
    dev->reprobe = reprobe;

1328 1329
    /* If the device isn't already bound to pci-stub, try binding it now.
     */
1330
    if (!virFileLinkPointsTo(driverLink, stubDriverPath)) {
1331
        /* Xen's pciback.ko wants you to use new_slot first */
1332 1333
        VIR_FREE(path);
        if (!(path = virPCIDriverFile(stubDriverName, "new_slot")))
1334
            goto remove_id;
1335

1336
        if (virFileExists(path) && virFileWriteStr(path, dev->name, 0) < 0) {
1337
            virReportSystemError(errno,
1338 1339 1340
                                 _("Failed to add slot for "
                                   "PCI device '%s' to %s"),
                                 dev->name, stubDriverName);
1341
            goto remove_id;
1342
        }
1343
        dev->remove_slot = true;
1344

1345 1346
        VIR_FREE(path);
        if (!(path = virPCIDriverFile(stubDriverName, "bind")))
1347
            goto remove_id;
1348

1349
        if (virFileWriteStr(path, dev->name, 0) < 0) {
1350
            virReportSystemError(errno,
1351
                                 _("Failed to bind PCI device '%s' to %s"),
1352
                                 dev->name, stubDriverName);
1353
            goto remove_id;
1354
        }
1355
        dev->unbind_from_stub = true;
1356 1357
    }

J
Jiri Denemark 已提交
1358 1359
    result = 0;

1360
 remove_id:
J
Jiri Denemark 已提交
1361 1362
    err = virSaveLastError();

1363 1364 1365
    /* If 'remove_id' exists, remove the device id from pci-stub's dynamic
     * ID table so that 'drivers_probe' works below.
     */
1366 1367
    VIR_FREE(path);
    if (!(path = virPCIDriverFile(stubDriverName, "remove_id"))) {
E
Eric Blake 已提交
1368
        /* We do not remove PCI ID from pci-stub, and we cannot reprobe it */
1369 1370
        if (dev->reprobe) {
            VIR_WARN("Could not remove PCI ID '%s' from %s, and the device "
1371
                     "cannot be probed again.", dev->id, stubDriverName);
1372
        }
1373
        dev->reprobe = false;
J
Jiri Denemark 已提交
1374
        result = -1;
1375 1376 1377
        goto cleanup;
    }

1378
    if (virFileExists(path) && virFileWriteStr(path, dev->id, 0) < 0) {
1379
        virReportSystemError(errno,
1380
                             _("Failed to remove PCI ID '%s' from %s"),
1381
                             dev->id, stubDriverName);
1382

E
Eric Blake 已提交
1383
        /* remove PCI ID from pci-stub failed, and we cannot reprobe it */
1384 1385
        if (dev->reprobe) {
            VIR_WARN("Failed to remove PCI ID '%s' from %s, and the device "
1386
                     "cannot be probed again.", dev->id, stubDriverName);
1387
        }
1388
        dev->reprobe = false;
J
Jiri Denemark 已提交
1389
        result = -1;
1390
        goto cleanup;
1391 1392
    }

1393
 cleanup:
1394
    if (result < 0)
J
Jiri Denemark 已提交
1395 1396 1397 1398
        virPCIDeviceUnbindFromStub(dev);

    if (err)
        virSetError(err);
1399

1400
    return result;
1401 1402
}

1403 1404 1405 1406
static int
virPCIDeviceBindToStubWithOverride(virPCIDevicePtr dev)
{
    const char *stubDriverName;
1407 1408
    VIR_AUTOFREE(char *) stubDriverPath = NULL;
    VIR_AUTOFREE(char *) driverLink = NULL;
1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424

    /* Check the device is configured to use one of the known stub drivers */
    if (dev->stubDriver == VIR_PCI_STUB_DRIVER_NONE) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("No stub driver configured for PCI device %s"),
                       dev->name);
        return -1;
    } else if (!(stubDriverName = virPCIStubDriverTypeToString(dev->stubDriver))) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unknown stub driver configured for PCI device %s"),
                       dev->name);
        return -1;
    }

    if (!(stubDriverPath = virPCIDriverDir(stubDriverName))  ||
        !(driverLink = virPCIFile(dev->name, "driver")))
1425
        return -1;
1426 1427 1428 1429 1430 1431

    if (virFileExists(driverLink)) {
        if (virFileLinkPointsTo(driverLink, stubDriverPath)) {
            /* The device is already bound to the correct driver */
            VIR_DEBUG("Device %s is already bound to %s",
                      dev->name, stubDriverName);
1432
            return 0;
1433 1434 1435 1436
        }
    }

    if (virPCIDeviceBindWithDriverOverride(dev, stubDriverName) < 0)
1437
        return -1;
1438 1439

    dev->unbind_from_stub = true;
1440
    return 0;
1441 1442 1443 1444 1445
}

static int
virPCIDeviceBindToStub(virPCIDevicePtr dev)
{
1446
    VIR_AUTOFREE(char *) path = NULL;
1447 1448 1449 1450 1451 1452 1453 1454 1455

    /*
     * Prefer using the device's driver_override interface, falling back
     * to the unpleasant new_id interface.
     */
    if (!(path = virPCIFile(dev->name, "driver_override")))
        return -1;

    if (virFileExists(path))
1456
        return virPCIDeviceBindToStubWithOverride(dev);
1457

1458
    return virPCIDeviceBindToStubWithNewid(dev);
1459 1460
}

1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478
/* virPCIDeviceDetach:
 *
 * Detach this device from the host driver, attach it to the stub
 * driver (previously set with virPCIDeviceSetStubDriver(), and add *a
 * copy* of the object to the inactiveDevs list (if provided). This
 * function will *never* consume dev, so the caller should free it.
 *
 * Returns 0 on success, -1 on failure (will fail if the device is
 * already in the activeDevs list, but will be a NOP if the device is
 * already bound to the stub).
 *
 * GENERAL NOTE: activeDevs should be a list of all PCI devices
 * currently in use by a domain. inactiveDevs is a list of all PCI
 * devices that libvirt has detached from the host driver + attached
 * to the stub driver, but hasn't yet assigned to a domain. Any device
 * that is still attached to its host driver should not be on either
 * list.
 */
1479
int
1480 1481
virPCIDeviceDetach(virPCIDevicePtr dev,
                   virPCIDeviceList *activeDevs,
1482
                   virPCIDeviceList *inactiveDevs)
1483
{
1484
    if (virPCIProbeStubDriver(dev->stubDriver) < 0)
1485 1486
        return -1;

1487
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
1488
        virReportError(VIR_ERR_INTERNAL_ERROR,
1489 1490 1491 1492
                       _("Not detaching active device %s"), dev->name);
        return -1;
    }

1493
    if (virPCIDeviceBindToStub(dev) < 0)
1494 1495
        return -1;

1496 1497 1498
    /* Add *a copy of* the dev into list inactiveDevs, if
     * it's not already there.
     */
1499 1500 1501 1502
    if (inactiveDevs && !virPCIDeviceListFind(inactiveDevs, dev)) {
        VIR_DEBUG("Adding PCI device %s to inactive list", dev->name);
        if (virPCIDeviceListAddCopy(inactiveDevs, dev) < 0)
            return -1;
1503 1504 1505
    }

    return 0;
1506 1507 1508
}

int
1509 1510
virPCIDeviceReattach(virPCIDevicePtr dev,
                     virPCIDeviceListPtr activeDevs,
1511
                     virPCIDeviceListPtr inactiveDevs)
1512
{
1513
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
1514
        virReportError(VIR_ERR_INTERNAL_ERROR,
1515 1516 1517 1518
                       _("Not reattaching active device %s"), dev->name);
        return -1;
    }

1519
    if (virPCIDeviceUnbindFromStub(dev) < 0)
1520 1521 1522
        return -1;

    /* Steal the dev from list inactiveDevs */
1523 1524
    if (inactiveDevs) {
        VIR_DEBUG("Removing PCI device %s from inactive list", dev->name);
1525
        virPCIDeviceListDel(inactiveDevs, dev);
1526
    }
1527 1528

    return 0;
1529 1530
}

1531 1532 1533 1534 1535
/* Certain hypervisors (like qemu/kvm) map the PCI bar(s) on
 * the host when doing device passthrough.  This can lead to a race
 * condition where the hypervisor is still cleaning up the device while
 * libvirt is trying to re-attach it to the host device driver.  To avoid
 * this situation, we look through /proc/iomem, and if the hypervisor is
E
Eric Blake 已提交
1536 1537
 * still holding on to the bar (denoted by the string in the matcher
 * variable), then we can wait around a bit for that to clear up.
1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557
 *
 * A typical /proc/iomem looks like this (snipped for brevity):
 * 00010000-0008efff : System RAM
 * 0008f000-0008ffff : reserved
 * ...
 * 00100000-cc9fcfff : System RAM
 *   00200000-00483d3b : Kernel code
 *   00483d3c-005c88df : Kernel data
 * cc9fd000-ccc71fff : ACPI Non-volatile Storage
 * ...
 * d0200000-d02fffff : PCI Bus #05
 *   d0200000-d021ffff : 0000:05:00.0
 *     d0200000-d021ffff : e1000e
 *   d0220000-d023ffff : 0000:05:00.0
 *     d0220000-d023ffff : e1000e
 * ...
 * f0000000-f0003fff : 0000:00:1b.0
 *   f0000000-f0003fff : kvm_assigned_device
 *
 * Returns 0 if we are clear to continue, and 1 if the hypervisor is still
E
Eric Blake 已提交
1558
 * holding on to the resource.
1559 1560
 */
int
1561
virPCIDeviceWaitForCleanup(virPCIDevicePtr dev, const char *matcher)
1562 1563 1564
{
    FILE *fp;
    char line[160];
1565
    char *tmp;
1566
    unsigned long long start, end;
1567
    unsigned int domain, bus, slot, function;
1568
    bool in_matching_device;
1569 1570 1571 1572 1573 1574 1575 1576 1577
    int ret;
    size_t match_depth;

    fp = fopen("/proc/iomem", "r");
    if (!fp) {
        /* If we failed to open iomem, we just basically ignore the error.  The
         * unbind might succeed anyway, and besides, it's very likely we have
         * no way to report the error
         */
1578
        VIR_DEBUG("Failed to open /proc/iomem, trying to continue anyway");
1579 1580 1581 1582
        return 0;
    }

    ret = 0;
1583
    in_matching_device = false;
1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594
    match_depth = 0;
    while (fgets(line, sizeof(line), fp) != 0) {
        /* the logic here is a bit confusing.  For each line, we look to
         * see if it matches the domain:bus:slot.function we were given.
         * If this line matches the DBSF, then any subsequent lines indented
         * by 2 spaces are the PCI regions for this device.  It's also
         * possible that none of the PCI regions are currently mapped, in
         * which case we have no indented regions.  This code handles all
         * of these situations
         */
        if (in_matching_device && (strspn(line, " ") == (match_depth + 2))) {
1595 1596 1597 1598 1599 1600
            /* expected format: <start>-<end> : <suffix> */
            if (/* start */
                virStrToLong_ull(line, &tmp, 16, &start) < 0 || *tmp != '-' ||
                /* end */
                virStrToLong_ull(tmp + 1, &tmp, 16, &end) < 0 ||
                (tmp = STRSKIP(tmp, " : ")) == NULL)
1601 1602
                continue;

1603
            if (STRPREFIX(tmp, matcher)) {
1604 1605 1606
                ret = 1;
                break;
            }
1607
        } else {
1608
            in_matching_device = false;
1609

1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623
            /* expected format: <start>-<end> : <domain>:<bus>:<slot>.<function> */
            if (/* start */
                virStrToLong_ull(line, &tmp, 16, &start) < 0 || *tmp != '-' ||
                /* end */
                virStrToLong_ull(tmp + 1, &tmp, 16, &end) < 0 ||
                (tmp = STRSKIP(tmp, " : ")) == NULL ||
                /* domain */
                virStrToLong_ui(tmp, &tmp, 16, &domain) < 0 || *tmp != ':' ||
                /* bus */
                virStrToLong_ui(tmp + 1, &tmp, 16, &bus) < 0 || *tmp != ':' ||
                /* slot */
                virStrToLong_ui(tmp + 1, &tmp, 16, &slot) < 0 || *tmp != '.' ||
                /* function */
                virStrToLong_ui(tmp + 1, &tmp, 16, &function) < 0 || *tmp != '\n')
1624 1625
                continue;

1626 1627
            if (domain != dev->address.domain || bus != dev->address.bus ||
                slot != dev->address.slot || function != dev->address.function)
1628
                continue;
1629
            in_matching_device = true;
1630 1631 1632 1633
            match_depth = strspn(line, " ");
        }
    }

E
Eric Blake 已提交
1634
    VIR_FORCE_FCLOSE(fp);
1635 1636 1637 1638

    return ret;
}

1639
static char *
1640
virPCIDeviceReadID(virPCIDevicePtr dev, const char *id_name)
1641
{
1642
    VIR_AUTOFREE(char *) path = NULL;
1643 1644
    char *id_str;

1645
    if (!(path = virPCIFile(dev->name, id_name)))
1646
        return NULL;
1647 1648

    /* ID string is '0xNNNN\n' ... i.e. 7 bytes */
1649
    if (virFileReadAll(path, 7, &id_str) < 0)
1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663
        return NULL;

    /* Check for 0x suffix */
    if (id_str[0] != '0' || id_str[1] != 'x') {
        VIR_FREE(id_str);
        return NULL;
    }

    /* Chop off the newline; we know the string is 7 bytes */
    id_str[6] = '\0';

    return id_str;
}

1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728
bool
virPCIDeviceAddressIsValid(virPCIDeviceAddressPtr addr,
                           bool report)
{
    if (addr->domain > 0xFFFF) {
        if (report)
            virReportError(VIR_ERR_XML_ERROR,
                           _("Invalid PCI address domain='0x%x', "
                             "must be <= 0xFFFF"),
                           addr->domain);
        return false;
    }
    if (addr->bus > 0xFF) {
        if (report)
            virReportError(VIR_ERR_XML_ERROR,
                           _("Invalid PCI address bus='0x%x', "
                             "must be <= 0xFF"),
                           addr->bus);
        return false;
    }
    if (addr->slot > 0x1F) {
        if (report)
            virReportError(VIR_ERR_XML_ERROR,
                           _("Invalid PCI address slot='0x%x', "
                             "must be <= 0x1F"),
                           addr->slot);
        return false;
    }
    if (addr->function > 7) {
        if (report)
            virReportError(VIR_ERR_XML_ERROR,
                           _("Invalid PCI address function=0x%x, "
                             "must be <= 7"),
                           addr->function);
        return false;
    }
    if (virPCIDeviceAddressIsEmpty(addr)) {
        if (report)
            virReportError(VIR_ERR_XML_ERROR, "%s",
                           _("Invalid PCI address 0000:00:00, at least "
                             "one of domain, bus, or slot must be > 0"));
        return false;
    }
    return true;
}

bool
virPCIDeviceAddressIsEmpty(const virPCIDeviceAddress *addr)
{
    return !(addr->domain || addr->bus || addr->slot);
}

bool
virPCIDeviceAddressEqual(virPCIDeviceAddress *addr1,
                         virPCIDeviceAddress *addr2)
{
    if (addr1->domain == addr2->domain &&
        addr1->bus == addr2->bus &&
        addr1->slot == addr2->slot &&
        addr1->function == addr2->function) {
        return true;
    }
    return false;
}

1729
char *
1730
virPCIDeviceAddressAsString(virPCIDeviceAddressPtr addr)
1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741
{
    char *str;

    ignore_value(virAsprintf(&str, "%.4x:%.2x:%.2x.%.1x",
                             addr->domain,
                             addr->bus,
                             addr->slot,
                             addr->function));
    return str;
}

1742
virPCIDevicePtr
1743 1744 1745 1746
virPCIDeviceNew(unsigned int domain,
                unsigned int bus,
                unsigned int slot,
                unsigned int function)
1747
{
1748 1749
    virPCIDevicePtr ret = NULL;
    VIR_AUTOPTR(virPCIDevice) dev = NULL;
1750 1751
    VIR_AUTOFREE(char *) vendor = NULL;
    VIR_AUTOFREE(char *) product = NULL;
1752

1753
    if (VIR_ALLOC(dev) < 0)
1754 1755
        return NULL;

1756 1757 1758 1759
    dev->address.domain = domain;
    dev->address.bus = bus;
    dev->address.slot = slot;
    dev->address.function = function;
1760

E
Eric Blake 已提交
1761
    if (snprintf(dev->name, sizeof(dev->name), "%.4x:%.2x:%.2x.%.1x",
1762
                 domain, bus, slot, function) >= sizeof(dev->name)) {
1763
        virReportError(VIR_ERR_INTERNAL_ERROR,
E
Eric Blake 已提交
1764
                       _("dev->name buffer overflow: %.4x:%.2x:%.2x.%.1x"),
1765
                       domain, bus, slot, function);
1766
        goto cleanup;
E
Eric Blake 已提交
1767 1768
    }
    if (virAsprintf(&dev->path, PCI_SYSFS "devices/%s/config",
1769
                    dev->name) < 0)
1770
        goto cleanup;
1771

1772
    if (!virFileExists(dev->path)) {
1773 1774 1775
        virReportSystemError(errno,
                             _("Device %s not found: could not access %s"),
                             dev->name, dev->path);
1776
        goto cleanup;
1777 1778
    }

1779 1780
    vendor  = virPCIDeviceReadID(dev, "vendor");
    product = virPCIDeviceReadID(dev, "device");
1781 1782

    if (!vendor || !product) {
1783
        virReportError(VIR_ERR_INTERNAL_ERROR,
1784 1785
                       _("Failed to read product/vendor ID for %s"),
                       dev->name);
1786
        goto cleanup;
1787 1788 1789
    }

    /* strings contain '0x' prefix */
E
Eric Blake 已提交
1790 1791
    if (snprintf(dev->id, sizeof(dev->id), "%s %s", &vendor[2],
                 &product[2]) >= sizeof(dev->id)) {
1792
        virReportError(VIR_ERR_INTERNAL_ERROR,
E
Eric Blake 已提交
1793 1794
                       _("dev->id buffer overflow: %s %s"),
                       &vendor[2], &product[2]);
1795
        goto cleanup;
E
Eric Blake 已提交
1796
    }
1797 1798 1799

    VIR_DEBUG("%s %s: initialized", dev->id, dev->name);

1800
    VIR_STEAL_PTR(ret, dev);
E
Eric Blake 已提交
1801

1802 1803
 cleanup:
    return ret;
1804 1805
}

L
Laine Stump 已提交
1806 1807 1808 1809 1810 1811

virPCIDevicePtr
virPCIDeviceCopy(virPCIDevicePtr dev)
{
    virPCIDevicePtr copy;

1812
    if (VIR_ALLOC(copy) < 0)
L
Laine Stump 已提交
1813 1814 1815 1816
        return NULL;

    /* shallow copy to take care of most attributes */
    *copy = *dev;
1817
    copy->path = NULL;
C
Chunyan Liu 已提交
1818
    copy->used_by_drvname = copy->used_by_domname = NULL;
L
Laine Stump 已提交
1819
    if (VIR_STRDUP(copy->path, dev->path) < 0 ||
C
Chunyan Liu 已提交
1820 1821
        VIR_STRDUP(copy->used_by_drvname, dev->used_by_drvname) < 0 ||
        VIR_STRDUP(copy->used_by_domname, dev->used_by_domname) < 0) {
L
Laine Stump 已提交
1822 1823 1824 1825
        goto error;
    }
    return copy;

1826
 error:
L
Laine Stump 已提交
1827 1828 1829 1830 1831
    virPCIDeviceFree(copy);
    return NULL;
}


1832
void
1833
virPCIDeviceFree(virPCIDevicePtr dev)
1834
{
1835 1836
    if (!dev)
        return;
1837
    VIR_DEBUG("%s %s: freeing", dev->id, dev->name);
E
Eric Blake 已提交
1838
    VIR_FREE(dev->path);
C
Chunyan Liu 已提交
1839 1840
    VIR_FREE(dev->used_by_drvname);
    VIR_FREE(dev->used_by_domname);
1841 1842
    VIR_FREE(dev);
}
1843

1844 1845 1846 1847 1848
/**
 * virPCIDeviceGetAddress:
 * @dev: device to get address from
 *
 * Take a PCI device on input and return its PCI address. The
1849
 * returned object is owned by the device and must not be freed.
1850
 *
1851
 * Returns: a pointer to the address, which can never be NULL.
1852 1853 1854 1855
 */
virPCIDeviceAddressPtr
virPCIDeviceGetAddress(virPCIDevicePtr dev)
{
1856
    return &(dev->address);
1857 1858
}

1859
const char *
1860
virPCIDeviceGetName(virPCIDevicePtr dev)
1861 1862 1863 1864
{
    return dev->name;
}

1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876
/**
 * virPCIDeviceGetConfigPath:
 *
 * Returns a pointer to a string containing the path of @dev's PCI
 * config file.
 */
const char *
virPCIDeviceGetConfigPath(virPCIDevicePtr dev)
{
    return dev->path;
}

1877
void virPCIDeviceSetManaged(virPCIDevicePtr dev, bool managed)
1878
{
1879
    dev->managed = managed;
1880 1881
}

1882
bool
1883
virPCIDeviceGetManaged(virPCIDevicePtr dev)
1884 1885 1886 1887
{
    return dev->managed;
}

1888 1889
void
virPCIDeviceSetStubDriver(virPCIDevicePtr dev, virPCIStubDriver driver)
1890
{
1891
    dev->stubDriver = driver;
1892 1893
}

1894
virPCIStubDriver
1895 1896 1897 1898 1899
virPCIDeviceGetStubDriver(virPCIDevicePtr dev)
{
    return dev->stubDriver;
}

1900
bool
1901
virPCIDeviceGetUnbindFromStub(virPCIDevicePtr dev)
1902 1903 1904 1905 1906
{
    return dev->unbind_from_stub;
}

void
1907
virPCIDeviceSetUnbindFromStub(virPCIDevicePtr dev, bool unbind)
1908
{
1909
    dev->unbind_from_stub = unbind;
1910 1911
}

1912
bool
1913
virPCIDeviceGetRemoveSlot(virPCIDevicePtr dev)
1914 1915 1916 1917 1918
{
    return dev->remove_slot;
}

void
1919
virPCIDeviceSetRemoveSlot(virPCIDevicePtr dev, bool remove_slot)
1920
{
1921
    dev->remove_slot = remove_slot;
1922 1923
}

1924
bool
1925
virPCIDeviceGetReprobe(virPCIDevicePtr dev)
1926 1927 1928 1929 1930
{
    return dev->reprobe;
}

void
1931
virPCIDeviceSetReprobe(virPCIDevicePtr dev, bool reprobe)
1932
{
1933
    dev->reprobe = reprobe;
1934 1935
}

C
Chunyan Liu 已提交
1936 1937 1938 1939
int
virPCIDeviceSetUsedBy(virPCIDevicePtr dev,
                      const char *drv_name,
                      const char *dom_name)
1940
{
C
Chunyan Liu 已提交
1941 1942 1943 1944 1945 1946 1947 1948
    VIR_FREE(dev->used_by_drvname);
    VIR_FREE(dev->used_by_domname);
    if (VIR_STRDUP(dev->used_by_drvname, drv_name) < 0)
        return -1;
    if (VIR_STRDUP(dev->used_by_domname, dom_name) < 0)
        return -1;

    return 0;
1949 1950
}

C
Chunyan Liu 已提交
1951 1952 1953 1954
void
virPCIDeviceGetUsedBy(virPCIDevicePtr dev,
                      const char **drv_name,
                      const char **dom_name)
1955
{
C
Chunyan Liu 已提交
1956 1957
    *drv_name = dev->used_by_drvname;
    *dom_name = dev->used_by_domname;
1958 1959
}

1960 1961
virPCIDeviceListPtr
virPCIDeviceListNew(void)
1962
{
1963
    virPCIDeviceListPtr list;
1964

1965 1966 1967 1968
    if (virPCIInitialize() < 0)
        return NULL;

    if (!(list = virObjectLockableNew(virPCIDeviceListClass)))
1969 1970 1971 1972 1973
        return NULL;

    return list;
}

1974 1975
static void
virPCIDeviceListDispose(void *obj)
1976
{
1977
    virPCIDeviceListPtr list = obj;
1978
    size_t i;
1979 1980

    for (i = 0; i < list->count; i++) {
1981
        virPCIDeviceFree(list->devs[i]);
1982 1983 1984 1985 1986 1987 1988 1989
        list->devs[i] = NULL;
    }

    list->count = 0;
    VIR_FREE(list->devs);
}

int
1990 1991
virPCIDeviceListAdd(virPCIDeviceListPtr list,
                    virPCIDevicePtr dev)
1992
{
1993
    if (virPCIDeviceListFind(list, dev)) {
1994
        virReportError(VIR_ERR_INTERNAL_ERROR,
1995 1996 1997
                       _("Device %s is already in use"), dev->name);
        return -1;
    }
1998
    return VIR_APPEND_ELEMENT(list->devs, list->count, dev);
1999 2000
}

L
Laine Stump 已提交
2001 2002 2003 2004 2005

/* virPCIDeviceListAddCopy - add a *copy* of the device to this list */
int
virPCIDeviceListAddCopy(virPCIDeviceListPtr list, virPCIDevicePtr dev)
{
2006
    VIR_AUTOPTR(virPCIDevice) copy = virPCIDeviceCopy(dev);
L
Laine Stump 已提交
2007 2008 2009

    if (!copy)
        return -1;
2010
    if (virPCIDeviceListAdd(list, copy) < 0)
L
Laine Stump 已提交
2011
        return -1;
2012 2013

    copy = NULL;
L
Laine Stump 已提交
2014 2015 2016 2017
    return 0;
}


2018 2019 2020
virPCIDevicePtr
virPCIDeviceListGet(virPCIDeviceListPtr list,
                    int idx)
2021 2022 2023 2024 2025 2026 2027 2028 2029
{
    if (idx >= list->count)
        return NULL;
    if (idx < 0)
        return NULL;

    return list->devs[idx];
}

2030
size_t
2031
virPCIDeviceListCount(virPCIDeviceListPtr list)
2032
{
2033 2034 2035
    return list->count;
}

2036 2037 2038
virPCIDevicePtr
virPCIDeviceListStealIndex(virPCIDeviceListPtr list,
                           int idx)
2039
{
2040
    virPCIDevicePtr ret;
2041

2042 2043
    if (idx < 0 || idx >= list->count)
        return NULL;
2044

2045
    ret = list->devs[idx];
2046
    VIR_DELETE_ELEMENT(list->devs, idx, list->count);
2047 2048 2049
    return ret;
}

2050 2051 2052
virPCIDevicePtr
virPCIDeviceListSteal(virPCIDeviceListPtr list,
                      virPCIDevicePtr dev)
2053
{
2054
    return virPCIDeviceListStealIndex(list, virPCIDeviceListFindIndex(list, dev));
2055 2056
}

2057
void
2058 2059
virPCIDeviceListDel(virPCIDeviceListPtr list,
                    virPCIDevicePtr dev)
2060
{
2061
    virPCIDeviceFree(virPCIDeviceListSteal(list, dev));
2062 2063
}

2064
int
2065
virPCIDeviceListFindIndex(virPCIDeviceListPtr list, virPCIDevicePtr dev)
2066
{
2067
    size_t i;
2068

2069 2070 2071 2072 2073 2074
    for (i = 0; i < list->count; i++) {
        virPCIDevicePtr other = list->devs[i];
        if (other->address.domain   == dev->address.domain &&
            other->address.bus      == dev->address.bus    &&
            other->address.slot     == dev->address.slot   &&
            other->address.function == dev->address.function)
2075
            return i;
2076
    }
2077 2078 2079
    return -1;
}

L
Laine Stump 已提交
2080 2081 2082 2083 2084 2085 2086 2087

virPCIDevicePtr
virPCIDeviceListFindByIDs(virPCIDeviceListPtr list,
                          unsigned int domain,
                          unsigned int bus,
                          unsigned int slot,
                          unsigned int function)
{
2088
    size_t i;
L
Laine Stump 已提交
2089 2090

    for (i = 0; i < list->count; i++) {
2091 2092 2093 2094 2095
        virPCIDevicePtr other = list->devs[i];
        if (other->address.domain   == domain &&
            other->address.bus      == bus    &&
            other->address.slot     == slot   &&
            other->address.function == function)
L
Laine Stump 已提交
2096 2097 2098 2099 2100 2101
            return list->devs[i];
    }
    return NULL;
}


2102 2103
virPCIDevicePtr
virPCIDeviceListFind(virPCIDeviceListPtr list, virPCIDevicePtr dev)
2104
{
2105
    int idx;
2106

2107 2108
    if ((idx = virPCIDeviceListFindIndex(list, dev)) >= 0)
        return list->devs[idx];
2109 2110
    else
        return NULL;
2111
}
2112 2113


2114 2115 2116
int virPCIDeviceFileIterate(virPCIDevicePtr dev,
                            virPCIDeviceFileActor actor,
                            void *opaque)
2117
{
2118
    VIR_AUTOFREE(char *) pcidir = NULL;
2119 2120 2121
    DIR *dir = NULL;
    int ret = -1;
    struct dirent *ent;
E
Eric Blake 已提交
2122
    int direrr;
2123 2124

    if (virAsprintf(&pcidir, "/sys/bus/pci/devices/%04x:%02x:%02x.%x",
2125 2126
                    dev->address.domain, dev->address.bus,
                    dev->address.slot, dev->address.function) < 0)
2127 2128
        goto cleanup;

J
Ján Tomko 已提交
2129
    if (virDirOpen(&dir, pcidir) < 0)
2130 2131
        goto cleanup;

E
Eric Blake 已提交
2132
    while ((direrr = virDirRead(dir, &ent, pcidir)) > 0) {
2133
        VIR_AUTOFREE(char *) file = NULL;
2134
        /* Device assignment requires:
A
Alex Williamson 已提交
2135
         *   $PCIDIR/config, $PCIDIR/resource, $PCIDIR/resourceNNN,
2136
         *   $PCIDIR/rom, $PCIDIR/reset, $PCIDIR/vendor, $PCIDIR/device
2137 2138 2139
         */
        if (STREQ(ent->d_name, "config") ||
            STRPREFIX(ent->d_name, "resource") ||
A
Alex Williamson 已提交
2140
            STREQ(ent->d_name, "rom") ||
2141 2142
            STREQ(ent->d_name, "vendor") ||
            STREQ(ent->d_name, "device") ||
A
Alex Williamson 已提交
2143
            STREQ(ent->d_name, "reset")) {
2144
            if (virAsprintf(&file, "%s/%s", pcidir, ent->d_name) < 0)
2145
                goto cleanup;
2146
            if ((actor)(dev, file, opaque) < 0)
2147 2148 2149
                goto cleanup;
        }
    }
E
Eric Blake 已提交
2150 2151
    if (direrr < 0)
        goto cleanup;
2152 2153 2154

    ret = 0;

2155
 cleanup:
J
Ján Tomko 已提交
2156
    VIR_DIR_CLOSE(dir);
2157 2158
    return ret;
}
J
Jiri Denemark 已提交
2159

L
Laine Stump 已提交
2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170

/* virPCIDeviceAddressIOMMUGroupIterate:
 *   Call @actor for all devices in the same iommu_group as orig
 *   (including orig itself) Even if there is no iommu_group for the
 *   device, call @actor once for orig.
 */
int
virPCIDeviceAddressIOMMUGroupIterate(virPCIDeviceAddressPtr orig,
                                     virPCIDeviceAddressActor actor,
                                     void *opaque)
{
2171
    VIR_AUTOFREE(char *) groupPath = NULL;
L
Laine Stump 已提交
2172 2173 2174
    DIR *groupDir = NULL;
    int ret = -1;
    struct dirent *ent;
E
Eric Blake 已提交
2175
    int direrr;
L
Laine Stump 已提交
2176 2177 2178

    if (virAsprintf(&groupPath,
                    PCI_SYSFS "devices/%04x:%02x:%02x.%x/iommu_group/devices",
2179
                    orig->domain, orig->bus, orig->slot, orig->function) < 0)
L
Laine Stump 已提交
2180 2181
        goto cleanup;

J
Ján Tomko 已提交
2182
    if (virDirOpenQuiet(&groupDir, groupPath) < 0) {
L
Laine Stump 已提交
2183 2184 2185 2186 2187
        /* just process the original device, nothing more */
        ret = (actor)(orig, opaque);
        goto cleanup;
    }

E
Eric Blake 已提交
2188
    while ((direrr = virDirRead(groupDir, &ent, groupPath)) > 0) {
L
Laine Stump 已提交
2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200
        virPCIDeviceAddress newDev;

        if (virPCIDeviceAddressParse(ent->d_name, &newDev) < 0) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Found invalid device link '%s' in '%s'"),
                           ent->d_name, groupPath);
            goto cleanup;
        }

        if ((actor)(&newDev, opaque) < 0)
            goto cleanup;
    }
E
Eric Blake 已提交
2201
    if (direrr < 0)
L
Laine Stump 已提交
2202 2203 2204 2205
        goto cleanup;

    ret = 0;

2206
 cleanup:
J
Ján Tomko 已提交
2207
    VIR_DIR_CLOSE(groupDir);
L
Laine Stump 已提交
2208 2209 2210 2211 2212 2213 2214 2215
    return ret;
}


static int
virPCIDeviceGetIOMMUGroupAddOne(virPCIDeviceAddressPtr newDevAddr, void *opaque)
{
    virPCIDeviceListPtr groupList = opaque;
2216
    VIR_AUTOPTR(virPCIDevice) newDev = NULL;
L
Laine Stump 已提交
2217 2218 2219

    if (!(newDev = virPCIDeviceNew(newDevAddr->domain, newDevAddr->bus,
                                   newDevAddr->slot, newDevAddr->function)))
2220
        return -1;
L
Laine Stump 已提交
2221 2222

    if (virPCIDeviceListAdd(groupList, newDev) < 0)
2223
        return -1;
L
Laine Stump 已提交
2224 2225

    newDev = NULL; /* it's now on the list */
2226
    return 0;
L
Laine Stump 已提交
2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
}


/*
 * virPCIDeviceGetIOMMUGroupList - return a virPCIDeviceList containing
 * all of the devices in the same iommu_group as @dev.
 *
 * Return the new list, or NULL on failure
 */
virPCIDeviceListPtr
virPCIDeviceGetIOMMUGroupList(virPCIDevicePtr dev)
{
    virPCIDeviceListPtr groupList = virPCIDeviceListNew();

    if (!groupList)
        goto error;

2244
    if (virPCIDeviceAddressIOMMUGroupIterate(&(dev->address),
L
Laine Stump 已提交
2245 2246 2247 2248 2249 2250
                                             virPCIDeviceGetIOMMUGroupAddOne,
                                             groupList) < 0)
        goto error;

    return groupList;

2251
 error:
L
Laine Stump 已提交
2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275
    virObjectUnref(groupList);
    return NULL;
}


typedef struct {
    virPCIDeviceAddressPtr **iommuGroupDevices;
    size_t *nIommuGroupDevices;
} virPCIDeviceAddressList;
typedef virPCIDeviceAddressList *virPCIDeviceAddressListPtr;

static int
virPCIGetIOMMUGroupAddressesAddOne(virPCIDeviceAddressPtr newDevAddr, void *opaque)
{
    int ret = -1;
    virPCIDeviceAddressListPtr addrList = opaque;
    virPCIDeviceAddressPtr copyAddr;

    /* make a copy to insert onto the list */
    if (VIR_ALLOC(copyAddr) < 0)
        goto cleanup;

    *copyAddr = *newDevAddr;

2276 2277
    if (VIR_APPEND_ELEMENT(*addrList->iommuGroupDevices,
                           *addrList->nIommuGroupDevices, copyAddr) < 0)
L
Laine Stump 已提交
2278 2279 2280
        goto cleanup;

    ret = 0;
2281
 cleanup:
L
Laine Stump 已提交
2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308
    VIR_FREE(copyAddr);
    return ret;
}


/*
 * virPCIDeviceAddressGetIOMMUGroupAddresses - return a
 * virPCIDeviceList containing all of the devices in the same
 * iommu_group as @dev.
 *
 * Return the new list, or NULL on failure
 */
int
virPCIDeviceAddressGetIOMMUGroupAddresses(virPCIDeviceAddressPtr devAddr,
                                          virPCIDeviceAddressPtr **iommuGroupDevices,
                                          size_t *nIommuGroupDevices)
{
    int ret = -1;
    virPCIDeviceAddressList addrList = { iommuGroupDevices,
                                         nIommuGroupDevices };

    if (virPCIDeviceAddressIOMMUGroupIterate(devAddr,
                                             virPCIGetIOMMUGroupAddressesAddOne,
                                             &addrList) < 0)
        goto cleanup;

    ret = 0;
2309
 cleanup:
L
Laine Stump 已提交
2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
    return ret;
}


/* virPCIDeviceAddressGetIOMMUGroupNum - return the group number of
 * this PCI device's iommu_group, or -2 if there is no iommu_group for
 * the device (or -1 if there was any other error)
 */
int
virPCIDeviceAddressGetIOMMUGroupNum(virPCIDeviceAddressPtr addr)
{
2321 2322 2323
    VIR_AUTOFREE(char *) devName = NULL;
    VIR_AUTOFREE(char *) devPath = NULL;
    VIR_AUTOFREE(char *) groupPath = NULL;
L
Laine Stump 已提交
2324 2325 2326 2327
    const char *groupNumStr;
    unsigned int groupNum;

    if (virAsprintf(&devName, "%.4x:%.2x:%.2x.%.1x", addr->domain,
2328
                    addr->bus, addr->slot, addr->function) < 0)
2329
        return -1;
L
Laine Stump 已提交
2330

2331
    if (!(devPath = virPCIFile(devName, "iommu_group")))
2332 2333 2334
        return -1;
    if (virFileIsLink(devPath) != 1)
        return -2;
L
Laine Stump 已提交
2335 2336 2337 2338
    if (virFileResolveLink(devPath, &groupPath) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to resolve device %s iommu_group symlink %s"),
                       devName, devPath);
2339
        return -1;
L
Laine Stump 已提交
2340 2341 2342 2343 2344 2345 2346 2347
    }

    groupNumStr = last_component(groupPath);
    if (virStrToLong_ui(groupNumStr, NULL, 10, &groupNum) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("device %s iommu_group symlink %s has "
                         "invalid group number %s"),
                       devName, groupPath, groupNumStr);
2348
        return -1;
L
Laine Stump 已提交
2349 2350
    }

2351
    return groupNum;
L
Laine Stump 已提交
2352 2353 2354
}


2355 2356
/* virPCIDeviceGetIOMMUGroupDev - return the name of the device used
 * to control this PCI device's group (e.g. "/dev/vfio/15")
2357 2358
 */
char *
2359
virPCIDeviceGetIOMMUGroupDev(virPCIDevicePtr dev)
2360
{
2361 2362
    VIR_AUTOFREE(char *) devPath = NULL;
    VIR_AUTOFREE(char *) groupPath = NULL;
2363 2364
    char *groupDev = NULL;

2365
    if (!(devPath = virPCIFile(dev->name, "iommu_group")))
2366
        return NULL;
2367 2368 2369 2370
    if (virFileIsLink(devPath) != 1) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Invalid device %s iommu_group file %s is not a symlink"),
                       dev->name, devPath);
2371
        return NULL;
2372 2373 2374 2375 2376
    }
    if (virFileResolveLink(devPath, &groupPath) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to resolve device %s iommu_group symlink %s"),
                       dev->name, devPath);
2377
        return NULL;
2378 2379
    }
    if (virAsprintf(&groupDev, "/dev/vfio/%s",
2380
                    last_component(groupPath)) < 0)
2381 2382
        return NULL;

2383 2384 2385
    return groupDev;
}

J
Jiri Denemark 已提交
2386
static int
2387
virPCIDeviceDownstreamLacksACS(virPCIDevicePtr dev)
J
Jiri Denemark 已提交
2388 2389 2390 2391
{
    uint16_t flags;
    uint16_t ctrl;
    unsigned int pos;
2392 2393
    int fd;
    int ret = 0;
2394
    uint16_t device_class;
J
Jiri Denemark 已提交
2395

2396
    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
J
Jiri Denemark 已提交
2397 2398
        return -1;

2399
    if (virPCIDeviceInit(dev, fd) < 0) {
2400 2401 2402 2403
        ret = -1;
        goto cleanup;
    }

2404 2405 2406
    if (virPCIDeviceReadClass(dev, &device_class) < 0)
        goto cleanup;

J
Jiri Denemark 已提交
2407
    pos = dev->pcie_cap_pos;
2408
    if (!pos || device_class != PCI_CLASS_BRIDGE_PCI)
2409
        goto cleanup;
J
Jiri Denemark 已提交
2410

2411
    flags = virPCIDeviceRead16(dev, fd, pos + PCI_EXP_FLAGS);
J
Jiri Denemark 已提交
2412
    if (((flags & PCI_EXP_FLAGS_TYPE) >> 4) != PCI_EXP_TYPE_DOWNSTREAM)
2413
        goto cleanup;
J
Jiri Denemark 已提交
2414

2415
    pos = virPCIDeviceFindExtendedCapabilityOffset(dev, fd, PCI_EXT_CAP_ID_ACS);
J
Jiri Denemark 已提交
2416 2417
    if (!pos) {
        VIR_DEBUG("%s %s: downstream port lacks ACS", dev->id, dev->name);
2418 2419
        ret = 1;
        goto cleanup;
J
Jiri Denemark 已提交
2420 2421
    }

2422
    ctrl = virPCIDeviceRead16(dev, fd, pos + PCI_EXT_ACS_CTRL);
J
Jiri Denemark 已提交
2423 2424 2425
    if ((ctrl & PCI_EXT_CAP_ACS_ENABLED) != PCI_EXT_CAP_ACS_ENABLED) {
        VIR_DEBUG("%s %s: downstream port has ACS disabled",
                  dev->id, dev->name);
2426 2427
        ret = 1;
        goto cleanup;
J
Jiri Denemark 已提交
2428 2429
    }

2430
 cleanup:
2431
    virPCIDeviceConfigClose(dev, fd);
2432
    return ret;
J
Jiri Denemark 已提交
2433 2434 2435
}

static int
2436
virPCIDeviceIsBehindSwitchLackingACS(virPCIDevicePtr dev)
J
Jiri Denemark 已提交
2437
{
2438
    VIR_AUTOPTR(virPCIDevice) parent = NULL;
J
Jiri Denemark 已提交
2439

2440
    if (virPCIDeviceGetParent(dev, &parent) < 0)
2441
        return -1;
2442 2443 2444 2445 2446
    if (!parent) {
        /* if we have no parent, and this is the root bus, ACS doesn't come
         * into play since devices on the root bus can't P2P without going
         * through the root IOMMU.
         */
2447
        if (dev->address.bus == 0) {
2448
            return 0;
2449
        } else {
2450
            virReportError(VIR_ERR_INTERNAL_ERROR,
2451 2452 2453 2454
                           _("Failed to find parent device for %s"),
                           dev->name);
            return -1;
        }
J
Jiri Denemark 已提交
2455 2456 2457 2458 2459 2460 2461
    }

    /* XXX we should rather fail when we can't find device's parent and
     * stop the loop when we get to root instead of just stopping when no
     * parent can be found
     */
    do {
2462
        VIR_AUTOPTR(virPCIDevice) tmp = NULL;
J
Jiri Denemark 已提交
2463
        int acs;
2464
        int ret;
J
Jiri Denemark 已提交
2465

2466
        acs = virPCIDeviceDownstreamLacksACS(parent);
J
Jiri Denemark 已提交
2467 2468 2469 2470 2471 2472 2473 2474 2475

        if (acs) {
            if (acs < 0)
                return -1;
            else
                return 1;
        }

        tmp = parent;
2476
        ret = virPCIDeviceGetParent(parent, &parent);
2477 2478
        if (ret < 0)
            return -1;
J
Jiri Denemark 已提交
2479 2480 2481 2482 2483
    } while (parent);

    return 0;
}

2484 2485
int virPCIDeviceIsAssignable(virPCIDevicePtr dev,
                             int strict_acs_check)
J
Jiri Denemark 已提交
2486 2487 2488 2489 2490 2491 2492 2493
{
    int ret;

    /* XXX This could be a great place to actually check that a non-managed
     * device isn't in use, e.g. by checking that device is either un-bound
     * or bound to a stub driver.
     */

2494
    ret = virPCIDeviceIsBehindSwitchLackingACS(dev);
J
Jiri Denemark 已提交
2495 2496 2497 2498 2499 2500 2501 2502
    if (ret < 0)
        return 0;

    if (ret) {
        if (!strict_acs_check) {
            VIR_DEBUG("%s %s: strict ACS check disabled; device assignment allowed",
                      dev->id, dev->name);
        } else {
2503
            virReportError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
2504 2505 2506 2507 2508 2509 2510 2511 2512
                           _("Device %s is behind a switch lacking ACS and "
                             "cannot be assigned"),
                           dev->name);
            return 0;
        }
    }

    return 1;
}
2513 2514 2515 2516 2517 2518 2519 2520 2521 2522

static int
logStrToLong_ui(char const *s,
                char **end_ptr,
                int base,
                unsigned int *result)
{
    int ret = 0;

    ret = virStrToLong_ui(s, end_ptr, base, result);
2523
    if (ret != 0)
2524 2525 2526 2527
        VIR_ERROR(_("Failed to convert '%s' to unsigned int"), s);
    return ret;
}

2528 2529
int
virPCIDeviceAddressParse(char *address,
2530
                         virPCIDeviceAddressPtr bdf)
2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556
{
    char *p = NULL;
    int ret = -1;

    if ((address == NULL) || (logStrToLong_ui(address, &p, 16,
                                              &bdf->domain) == -1)) {
        goto out;
    }

    if ((p == NULL) || (logStrToLong_ui(p+1, &p, 16,
                                        &bdf->bus) == -1)) {
        goto out;
    }

    if ((p == NULL) || (logStrToLong_ui(p+1, &p, 16,
                                        &bdf->slot) == -1)) {
        goto out;
    }

    if ((p == NULL) || (logStrToLong_ui(p+1, &p, 16,
                                        &bdf->function) == -1)) {
        goto out;
    }

    ret = 0;

2557
 out:
2558 2559 2560
    return ret;
}

2561

2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586
bool
virZPCIDeviceAddressIsValid(virZPCIDeviceAddressPtr zpci)
{
    /* We don't need to check fid because fid covers
     * all range of uint32 type.
     */
    if (zpci->uid > VIR_DOMAIN_DEVICE_ZPCI_MAX_UID ||
        zpci->uid == 0) {
        virReportError(VIR_ERR_XML_ERROR,
                       _("Invalid PCI address uid='0x%.4x', "
                         "must be > 0x0000 and <= 0x%.4x"),
                       zpci->uid,
                       VIR_DOMAIN_DEVICE_ZPCI_MAX_UID);
        return false;
    }

    return true;
}

bool
virZPCIDeviceAddressIsEmpty(const virZPCIDeviceAddress *addr)
{
    return !(addr->uid || addr->fid);
}

2587
#ifdef __linux__
2588

2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601
/*
 * returns true if equal
 */
static bool
virPCIDeviceAddressIsEqual(virPCIDeviceAddressPtr bdf1,
                           virPCIDeviceAddressPtr bdf2)
{
    return ((bdf1->domain == bdf2->domain) &&
            (bdf1->bus == bdf2->bus) &&
            (bdf1->slot == bdf2->slot) &&
            (bdf1->function == bdf2->function));
}

2602
virPCIDeviceAddressPtr
2603
virPCIGetDeviceAddressFromSysfsLink(const char *device_link)
2604
{
2605
    virPCIDeviceAddressPtr bdf = NULL;
2606
    char *config_address = NULL;
2607
    VIR_AUTOFREE(char *) device_path = NULL;
2608 2609

    if (!virFileExists(device_link)) {
2610
        VIR_DEBUG("'%s' does not exist", device_link);
2611
        return NULL;
2612 2613
    }

2614
    device_path = virFileCanonicalizePath(device_link);
2615
    if (device_path == NULL) {
2616 2617 2618
        virReportSystemError(errno,
                             _("Failed to resolve device link '%s'"),
                             device_link);
2619
        return NULL;
2620 2621
    }

2622
    config_address = last_component(device_path);
2623
    if (VIR_ALLOC(bdf) < 0)
2624
        return NULL;
2625

2626
    if (virPCIDeviceAddressParse(config_address, bdf) < 0) {
2627
        virReportError(VIR_ERR_INTERNAL_ERROR,
2628 2629
                       _("Failed to parse PCI config address '%s'"),
                       config_address);
2630
        VIR_FREE(bdf);
2631
        return NULL;
2632 2633
    }

2634
    return bdf;
2635 2636
}

2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649
/**
 * virPCIGetPhysicalFunction:
 * @vf_sysfs_path: sysfs path for the virtual function
 * @pf: where to store the physical function's address
 *
 * Given @vf_sysfs_path, this function will store the pointer
 * to a newly-allocated virPCIDeviceAddress in @pf.
 *
 * @pf might be NULL if @vf_sysfs_path does not point to a
 * virtual function. If it's not NULL, then it should be
 * freed by the caller when no longer needed.
 *
 * Returns: >=0 on success, <0 on failure
2650 2651
 */
int
2652
virPCIGetPhysicalFunction(const char *vf_sysfs_path,
2653
                          virPCIDeviceAddressPtr *pf)
2654
{
2655
    VIR_AUTOFREE(char *) device_link = NULL;
2656

2657 2658
    *pf = NULL;

2659 2660
    if (virBuildPath(&device_link, vf_sysfs_path, "physfn") == -1) {
        virReportOOMError();
2661
        return -1;
2662 2663
    }

2664
    if ((*pf = virPCIGetDeviceAddressFromSysfsLink(device_link))) {
2665 2666 2667
        VIR_DEBUG("PF for VF device '%s': %.4x:%.2x:%.2x.%.1x", vf_sysfs_path,
                  (*pf)->domain, (*pf)->bus, (*pf)->slot, (*pf)->function);
    }
2668

2669
    return 0;
2670 2671
}

2672

2673 2674 2675 2676
/*
 * Returns virtual functions of a physical function
 */
int
2677 2678
virPCIGetVirtualFunctions(const char *sysfs_path,
                          virPCIDeviceAddressPtr **virtual_functions,
2679 2680
                          size_t *num_virtual_functions,
                          unsigned int *max_virtual_functions)
2681 2682
{
    int ret = -1;
2683
    size_t i;
2684 2685
    VIR_AUTOFREE(char *) totalvfs_file = NULL;
    VIR_AUTOFREE(char *) totalvfs_str = NULL;
2686
    virPCIDeviceAddressPtr config_addr = NULL;
2687

2688 2689
    *virtual_functions = NULL;
    *num_virtual_functions = 0;
2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705
    *max_virtual_functions = 0;

    if (virAsprintf(&totalvfs_file, "%s/sriov_totalvfs", sysfs_path) < 0)
       goto error;
    if (virFileExists(totalvfs_file)) {
        char *end = NULL; /* so that terminating \n doesn't create error */

        if (virFileReadAll(totalvfs_file, 16, &totalvfs_str) < 0)
            goto error;
        if (virStrToLong_ui(totalvfs_str, &end, 10, max_virtual_functions) < 0) {
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Unrecognized value in %s: %s"),
                           totalvfs_file, totalvfs_str);
            goto error;
        }
    }
2706

2707
    do {
2708
        VIR_AUTOFREE(char *) device_link = NULL;
2709 2710 2711
        /* look for virtfn%d links until one isn't found */
        if (virAsprintf(&device_link, "%s/virtfn%zu", sysfs_path, *num_virtual_functions) < 0)
            goto error;
2712

2713 2714
        if (!virFileExists(device_link))
            break;
2715

2716
        if (!(config_addr = virPCIGetDeviceAddressFromSysfsLink(device_link))) {
2717 2718 2719 2720 2721
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Failed to get SRIOV function from device link '%s'"),
                           device_link);
            goto error;
        }
2722

2723 2724
        if (VIR_APPEND_ELEMENT(*virtual_functions, *num_virtual_functions,
                               config_addr) < 0)
2725 2726
            goto error;
    } while (1);
2727

2728 2729
    VIR_DEBUG("Found %zu virtual functions for %s",
              *num_virtual_functions, sysfs_path);
2730
    ret = 0;
2731
 cleanup:
2732
    VIR_FREE(config_addr);
2733
    return ret;
2734

2735
 error:
2736 2737 2738
    for (i = 0; i < *num_virtual_functions; i++)
        VIR_FREE((*virtual_functions)[i]);
    VIR_FREE(*virtual_functions);
2739
    *num_virtual_functions = 0;
2740
    goto cleanup;
2741
}
2742

2743

2744 2745 2746 2747
/*
 * Returns 1 if vf device is a virtual function, 0 if not, -1 on error
 */
int
2748
virPCIIsVirtualFunction(const char *vf_sysfs_device_link)
2749
{
2750
    VIR_AUTOFREE(char *) vf_sysfs_physfn_link = NULL;
2751 2752

    if (virAsprintf(&vf_sysfs_physfn_link, "%s/physfn",
2753
                    vf_sysfs_device_link) < 0)
2754
        return -1;
2755

2756
    return virFileExists(vf_sysfs_physfn_link);
2757 2758 2759 2760 2761 2762
}

/*
 * Returns the sriov virtual function index of vf given its pf
 */
int
2763 2764 2765
virPCIGetVirtualFunctionIndex(const char *pf_sysfs_device_link,
                              const char *vf_sysfs_device_link,
                              int *vf_index)
2766
{
2767 2768
    int ret = -1;
    size_t i;
2769
    size_t num_virt_fns = 0;
2770
    unsigned int max_virt_fns = 0;
2771 2772
    virPCIDeviceAddressPtr vf_bdf = NULL;
    virPCIDeviceAddressPtr *virt_fns = NULL;
2773

2774
    if (!(vf_bdf = virPCIGetDeviceAddressFromSysfsLink(vf_sysfs_device_link)))
2775 2776
        return ret;

2777
    if (virPCIGetVirtualFunctions(pf_sysfs_device_link, &virt_fns,
2778
                                  &num_virt_fns, &max_virt_fns) < 0) {
2779
        virReportError(VIR_ERR_INTERNAL_ERROR,
2780
                       _("Error getting physical function's '%s' "
2781
                         "virtual_functions"), pf_sysfs_device_link);
2782 2783 2784 2785
        goto out;
    }

    for (i = 0; i < num_virt_fns; i++) {
2786 2787 2788 2789 2790
        if (virPCIDeviceAddressIsEqual(vf_bdf, virt_fns[i])) {
            *vf_index = i;
            ret = 0;
            break;
        }
2791 2792
    }

2793
 out:
2794 2795 2796

    /* free virtual functions */
    for (i = 0; i < num_virt_fns; i++)
2797
        VIR_FREE(virt_fns[i]);
2798

A
ajia@redhat.com 已提交
2799
    VIR_FREE(virt_fns);
2800 2801 2802 2803 2804
    VIR_FREE(vf_bdf);

    return ret;
}

2805 2806 2807 2808 2809
/*
 * Returns a path to the PCI sysfs file given the BDF of the PCI function
 */

int
2810
virPCIGetSysfsFile(char *virPCIDeviceName, char **pci_sysfs_device_link)
2811
{
2812 2813 2814 2815
    if (virAsprintf(pci_sysfs_device_link, PCI_SYSFS "devices/%s",
                    virPCIDeviceName) < 0)
        return -1;
    return 0;
2816 2817
}

R
Roopa Prabhu 已提交
2818
int
2819
virPCIDeviceAddressGetSysfsFile(virPCIDeviceAddressPtr addr,
2820
                                char **pci_sysfs_device_link)
R
Roopa Prabhu 已提交
2821
{
2822
    if (virAsprintf(pci_sysfs_device_link,
2823 2824 2825
                    PCI_SYSFS "devices/%04x:%02x:%02x.%x",
                    addr->domain, addr->bus,
                    addr->slot, addr->function) < 0)
2826 2827
        return -1;
    return 0;
R
Roopa Prabhu 已提交
2828 2829
}

2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840
/**
 * virPCIGetNetName:
 * @device_link_sysfs_path: sysfs path to the PCI device
 * @idx: used to choose which netdev when there are several
 *       (ignored if physPortID is set)
 * @physPortID: match this string in the netdev's phys_port_id
 *       (or NULL to ignore and use idx instead)
 * @netname: used to return the name of the netdev
 *       (set to NULL (but returns success) if there is no netdev)
 *
 * Returns 0 on success, -1 on error (error has been logged)
2841 2842
 */
int
2843 2844 2845 2846
virPCIGetNetName(const char *device_link_sysfs_path,
                 size_t idx,
                 char *physPortID,
                 char **netname)
2847
{
2848 2849 2850
    VIR_AUTOFREE(char *) pcidev_sysfs_net_path = NULL;
    VIR_AUTOFREE(char *) firstEntryName = NULL;
    VIR_AUTOFREE(char *) thisPhysPortID = NULL;
2851 2852 2853
    int ret = -1;
    DIR *dir = NULL;
    struct dirent *entry = NULL;
2854
    size_t i = 0;
2855

2856 2857
    *netname = NULL;

2858 2859 2860 2861 2862 2863
    if (virBuildPath(&pcidev_sysfs_net_path, device_link_sysfs_path,
                     "net") == -1) {
        virReportOOMError();
        return -1;
    }

2864 2865 2866
    if (virDirOpenQuiet(&dir, pcidev_sysfs_net_path) < 0) {
        /* this *isn't* an error - caller needs to check for netname == NULL */
        ret = 0;
2867
        goto cleanup;
2868
    }
2869

E
Eric Blake 已提交
2870
    while (virDirRead(dir, &entry, pcidev_sysfs_net_path) > 0) {
2871 2872 2873 2874 2875 2876 2877 2878 2879 2880
        /* if the caller sent a physPortID, compare it to the
         * physportID of this netdev. If not, look for entry[idx].
         */
        if (physPortID) {
            if (virNetDevGetPhysPortID(entry->d_name, &thisPhysPortID) < 0)
                goto cleanup;

            /* if this one doesn't match, keep looking */
            if (STRNEQ_NULLABLE(physPortID, thisPhysPortID)) {
                VIR_FREE(thisPhysPortID);
2881 2882 2883 2884 2885 2886 2887 2888 2889 2890
                /* save the first entry we find to use as a failsafe
                 * in case we don't match the phys_port_id. This is
                 * needed because some NIC drivers (e.g. i40e)
                 * implement phys_port_id for PFs, but not for VFs
                 */
                if (!firstEntryName &&
                    VIR_STRDUP(firstEntryName, entry->d_name) < 0) {
                    goto cleanup;
                }

2891 2892 2893 2894 2895 2896 2897 2898 2899 2900 2901
                continue;
            }
        } else {
            if (i++ < idx)
                continue;
        }

        if (VIR_STRDUP(*netname, entry->d_name) < 0)
            goto cleanup;

        ret = 0;
2902 2903 2904
        break;
    }

2905 2906
    if (ret < 0) {
        if (physPortID) {
2907 2908 2909 2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921
            if (firstEntryName) {
                /* we didn't match the provided phys_port_id, but this
                 * is probably because phys_port_id isn't implemented
                 * for this NIC driver, so just return the first
                 * (probably only) netname we found.
                 */
                *netname = firstEntryName;
                firstEntryName = NULL;
                ret = 0;
            } else {
                virReportError(VIR_ERR_INTERNAL_ERROR,
                               _("Could not find network device with "
                                 "phys_port_id '%s' under PCI device at %s"),
                               physPortID, device_link_sysfs_path);
            }
2922 2923 2924 2925 2926
        } else {
            ret = 0; /* no netdev at the given index is *not* an error */
        }
    }
 cleanup:
J
Ján Tomko 已提交
2927
    VIR_DIR_CLOSE(dir);
2928
    return ret;
2929
}
R
Roopa Prabhu 已提交
2930 2931

int
2932
virPCIGetVirtualFunctionInfo(const char *vf_sysfs_device_path,
2933 2934 2935
                             int pfNetDevIdx,
                             char **pfname,
                             int *vf_index)
R
Roopa Prabhu 已提交
2936
{
2937
    virPCIDeviceAddressPtr pf_config_address = NULL;
2938 2939 2940
    VIR_AUTOFREE(char *) pf_sysfs_device_path = NULL;
    VIR_AUTOFREE(char *) vfname = NULL;
    VIR_AUTOFREE(char *) vfPhysPortID = NULL;
R
Roopa Prabhu 已提交
2941 2942
    int ret = -1;

2943
    if (virPCIGetPhysicalFunction(vf_sysfs_device_path, &pf_config_address) < 0)
2944
        goto cleanup;
R
Roopa Prabhu 已提交
2945

2946
    if (!pf_config_address)
2947
        goto cleanup;
2948

2949 2950
    if (virPCIDeviceAddressGetSysfsFile(pf_config_address,
                                        &pf_sysfs_device_path) < 0) {
2951 2952
        goto cleanup;
    }
R
Roopa Prabhu 已提交
2953

2954 2955 2956
    if (virPCIGetVirtualFunctionIndex(pf_sysfs_device_path,
                                      vf_sysfs_device_path, vf_index) < 0) {
        goto cleanup;
R
Roopa Prabhu 已提交
2957 2958
    }

2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978
    /* If the caller hasn't asked for a specific pfNetDevIdx, and VF
     * is bound to a netdev, learn that netdev's phys_port_id (if
     * available). This can be used to disambiguate when the PF has
     * multiple netdevs. If the VF isn't bound to a netdev, then we
     * return netdev[pfNetDevIdx] on the PF, which may or may not be
     * correct.
     */
    if (pfNetDevIdx == -1) {
        if (virPCIGetNetName(vf_sysfs_device_path, 0, NULL, &vfname) < 0)
            goto cleanup;

        if (vfname) {
            if (virNetDevGetPhysPortID(vfname, &vfPhysPortID) < 0)
                goto cleanup;
        }
        pfNetDevIdx = 0;
    }

    if (virPCIGetNetName(pf_sysfs_device_path,
                         pfNetDevIdx, vfPhysPortID, pfname) < 0) {
R
Roopa Prabhu 已提交
2979
        goto cleanup;
2980
    }
R
Roopa Prabhu 已提交
2981

2982 2983 2984 2985 2986 2987 2988 2989 2990
    if (!*pfname) {
        /* this shouldn't be possible. A VF can't exist unless its
         * PF device is bound to a network driver
         */
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("The PF device for VF %s has no network device name"),
                       vf_sysfs_device_path);
        goto cleanup;
    }
R
Roopa Prabhu 已提交
2991

2992
    ret = 0;
2993
 cleanup:
R
Roopa Prabhu 已提交
2994 2995 2996 2997 2998
    VIR_FREE(pf_config_address);

    return ret;
}

2999 3000 3001 3002 3003 3004 3005 3006 3007

ssize_t
virPCIGetMdevTypes(const char *sysfspath,
                   virMediatedDeviceTypePtr **types)
{
    ssize_t ret = -1;
    int dirret = -1;
    DIR *dir = NULL;
    struct dirent *entry;
3008
    VIR_AUTOFREE(char *) types_path = NULL;
3009
    VIR_AUTOPTR(virMediatedDeviceType) mdev_type = NULL;
3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025
    virMediatedDeviceTypePtr *mdev_types = NULL;
    size_t ntypes = 0;
    size_t i;

    if (virAsprintf(&types_path, "%s/mdev_supported_types", sysfspath) < 0)
        return -1;

    if ((dirret = virDirOpenIfExists(&dir, types_path)) < 0)
        goto cleanup;

    if (dirret == 0) {
        ret = 0;
        goto cleanup;
    }

    while ((dirret = virDirRead(dir, &entry, types_path)) > 0) {
3026
        VIR_AUTOFREE(char *) tmppath = NULL;
3027 3028 3029 3030 3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051
        /* append the type id to the path and read the attributes from there */
        if (virAsprintf(&tmppath, "%s/%s", types_path, entry->d_name) < 0)
            goto cleanup;

        if (virMediatedDeviceTypeReadAttrs(tmppath, &mdev_type) < 0)
            goto cleanup;

        if (VIR_APPEND_ELEMENT(mdev_types, ntypes, mdev_type) < 0)
            goto cleanup;
    }

    if (dirret < 0)
        goto cleanup;

    VIR_STEAL_PTR(*types, mdev_types);
    ret = ntypes;
    ntypes = 0;
 cleanup:
    for (i = 0; i < ntypes; i++)
        virMediatedDeviceTypeFree(mdev_types[i]);
    VIR_FREE(mdev_types);
    VIR_DIR_CLOSE(dir);
    return ret;
}

3052
#else
3053 3054
static const char *unsupported = N_("not supported on non-linux platforms");

3055 3056 3057 3058
virPCIDeviceAddressPtr
virPCIGetDeviceAddressFromSysfsLink(const char *device_link ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3059
    return NULL;
3060 3061 3062
}


3063
int
3064
virPCIGetPhysicalFunction(const char *vf_sysfs_path ATTRIBUTE_UNUSED,
3065
                          virPCIDeviceAddressPtr *pf ATTRIBUTE_UNUSED)
3066
{
3067
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3068 3069 3070 3071
    return -1;
}

int
3072 3073
virPCIGetVirtualFunctions(const char *sysfs_path ATTRIBUTE_UNUSED,
                          virPCIDeviceAddressPtr **virtual_functions ATTRIBUTE_UNUSED,
3074 3075
                          size_t *num_virtual_functions ATTRIBUTE_UNUSED,
                          unsigned int *max_virtual_functions ATTRIBUTE_UNUSED)
3076
{
3077
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3078 3079
    return -1;
}
3080 3081

int
E
Eric Blake 已提交
3082
virPCIIsVirtualFunction(const char *vf_sysfs_device_link ATTRIBUTE_UNUSED)
3083
{
3084
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3085 3086 3087 3088
    return -1;
}

int
3089 3090 3091
virPCIGetVirtualFunctionIndex(const char *pf_sysfs_device_link ATTRIBUTE_UNUSED,
                              const char *vf_sysfs_device_link ATTRIBUTE_UNUSED,
                              int *vf_index ATTRIBUTE_UNUSED)
3092
{
3093
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3094 3095 3096 3097
    return -1;

}

3098
int
3099 3100
virPCIDeviceAddressGetSysfsFile(virPCIDeviceAddressPtr dev ATTRIBUTE_UNUSED,
                                char **pci_sysfs_device_link ATTRIBUTE_UNUSED)
3101
{
3102
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3103 3104 3105
    return -1;
}

3106
int
3107
virPCIGetNetName(const char *device_link_sysfs_path ATTRIBUTE_UNUSED,
3108 3109
                 size_t idx ATTRIBUTE_UNUSED,
                 char *physPortID ATTRIBUTE_UNUSED,
3110
                 char **netname ATTRIBUTE_UNUSED)
3111
{
3112
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3113 3114
    return -1;
}
R
Roopa Prabhu 已提交
3115 3116

int
3117
virPCIGetVirtualFunctionInfo(const char *vf_sysfs_device_path ATTRIBUTE_UNUSED,
3118
                             int pfNetDevIdx ATTRIBUTE_UNUSED,
3119 3120
                             char **pfname ATTRIBUTE_UNUSED,
                             int *vf_index ATTRIBUTE_UNUSED)
R
Roopa Prabhu 已提交
3121
{
3122
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
R
Roopa Prabhu 已提交
3123 3124
    return -1;
}
3125 3126 3127 3128 3129 3130 3131 3132 3133


ssize_t
virPCIGetMdevTypes(const char *sysfspath ATTRIBUTE_UNUSED,
                   virMediatedDeviceTypePtr **types ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
    return -1;
}
3134
#endif /* __linux__ */
3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179 3180 3181 3182 3183 3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216 3217 3218

int
virPCIDeviceIsPCIExpress(virPCIDevicePtr dev)
{
    int fd;
    int ret = -1;

    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
        return ret;

    if (virPCIDeviceInit(dev, fd) < 0)
        goto cleanup;

    ret = dev->pcie_cap_pos != 0;

 cleanup:
    virPCIDeviceConfigClose(dev, fd);
    return ret;
}

int
virPCIDeviceHasPCIExpressLink(virPCIDevicePtr dev)
{
    int fd;
    int ret = -1;
    uint16_t cap, type;

    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
        return ret;

    if (virPCIDeviceInit(dev, fd) < 0)
        goto cleanup;

    cap = virPCIDeviceRead16(dev, fd, dev->pcie_cap_pos + PCI_CAP_FLAGS);
    type = (cap & PCI_EXP_FLAGS_TYPE) >> 4;

    ret = type != PCI_EXP_TYPE_ROOT_INT_EP && type != PCI_EXP_TYPE_ROOT_EC;

 cleanup:
    virPCIDeviceConfigClose(dev, fd);
    return ret;
}

int
virPCIDeviceGetLinkCapSta(virPCIDevicePtr dev,
                          int *cap_port,
                          unsigned int *cap_speed,
                          unsigned int *cap_width,
                          unsigned int *sta_speed,
                          unsigned int *sta_width)
{
    uint32_t t;
    int fd;
    int ret = -1;

    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
        return ret;

    if (virPCIDeviceInit(dev, fd) < 0)
        goto cleanup;

    if (!dev->pcie_cap_pos) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("pci device %s is not a PCI-Express device"),
                       dev->name);
        goto cleanup;
    }

    t = virPCIDeviceRead32(dev, fd, dev->pcie_cap_pos + PCI_EXP_LNKCAP);

    *cap_port = t >> 24;
    *cap_speed = t & PCI_EXP_LNKCAP_SPEED;
    *cap_width = (t & PCI_EXP_LNKCAP_WIDTH) >> 4;

    t = virPCIDeviceRead16(dev, fd, dev->pcie_cap_pos + PCI_EXP_LNKSTA);

    *sta_speed = t & PCI_EXP_LNKSTA_SPEED;
    *sta_width = (t & PCI_EXP_LNKSTA_WIDTH) >> 4;
    ret = 0;

 cleanup:
    virPCIDeviceConfigClose(dev, fd);
    return ret;
}
3219 3220


3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247
int virPCIGetHeaderType(virPCIDevicePtr dev, int *hdrType)
{
    int fd;
    uint8_t type;

    *hdrType = -1;

    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
        return -1;

    type = virPCIDeviceRead8(dev, fd, PCI_HEADER_TYPE);

    virPCIDeviceConfigClose(dev, fd);

    type &= PCI_HEADER_TYPE_MASK;
    if (type >= VIR_PCI_HEADER_LAST) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unknown PCI header type '%d'"), type);
        return -1;
    }

    *hdrType = type;

    return 0;
}


3248 3249 3250 3251 3252 3253 3254 3255 3256 3257
void
virPCIEDeviceInfoFree(virPCIEDeviceInfoPtr dev)
{
    if (!dev)
        return;

    VIR_FREE(dev->link_cap);
    VIR_FREE(dev->link_sta);
    VIR_FREE(dev);
}
3258 3259 3260 3261 3262 3263

void
virPCIDeviceAddressFree(virPCIDeviceAddressPtr address)
{
    VIR_FREE(address);
}