virpci.c 93.2 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
#include "viralloc.h"
42

43 44
VIR_LOG_INIT("util.pci");

45 46 47
#define PCI_SYSFS "/sys/bus/pci/"
#define PCI_ID_LEN 10   /* "XXXX XXXX" */

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

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

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

67
struct _virPCIDevice {
68
    virPCIDeviceAddress address;
69

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

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

78 79
    unsigned int  pcie_cap_pos;
    unsigned int  pci_pm_cap_pos;
80 81
    bool          has_flr;
    bool          has_pm_reset;
82
    bool          managed;
83 84

    virPCIStubDriver stubDriver;
85 86

    /* used by reattach function */
87 88 89
    bool          unbind_from_stub;
    bool          remove_slot;
    bool          reprobe;
90 91
};

92
struct _virPCIDeviceList {
93 94
    virObjectLockable parent;

95
    size_t count;
96
    virPCIDevicePtr *devs;
97 98 99
};


100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
/* 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 */
117 118 119
#define PCI_HEADER_TYPE_BRIDGE 0x1
#define PCI_HEADER_TYPE_MASK   0x7f
#define PCI_HEADER_TYPE_MULTI  0x80
120 121 122 123 124 125 126 127 128

/* 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 */
129
#define PCI_STATUS_CAP_LIST    0x10    /* Support Capability List */
130 131 132

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

/* 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 */
144 145
#define PCI_EXP_DEVCAP_FLR     (1<<28)  /* Function Level Reset */
#define PCI_EXP_LNKCAP          0xc     /* Link Capabilities */
146
#define PCI_EXP_LNKCAP_SPEED    0x0000f /* Maximum Link Speed */
147 148 149 150
#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 */
151 152 153 154 155 156 157

/* 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 */
158
#define PCI_BRIDGE_CTL_RESET   0x40    /* Secondary bus reset */
159 160 161

/* PM12 3.2.4  Power Management Control/Status (Offset = 4) */
#define PCI_PM_CTRL                4    /* PM control and status register */
162 163 164 165
#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 */
166 167 168

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

J
Jiri Denemark 已提交
171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187
#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
188 189 190
#define PCI_EXT_CAP_ACS_ENABLED (PCI_EXT_CAP_ACS_SV | \
                                 PCI_EXT_CAP_ACS_RR | \
                                 PCI_EXT_CAP_ACS_CR | \
J
Jiri Denemark 已提交
191 192
                                 PCI_EXT_CAP_ACS_UF)

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

196 197 198 199 200 201
static virClassPtr virPCIDeviceListClass;

static void virPCIDeviceListDispose(void *obj);

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

    return 0;
}

208
VIR_ONCE_GLOBAL_INIT(virPCI);
209

L
Laine Stump 已提交
210

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

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


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

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


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

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


/* 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.
 */
248
int
L
Laine Stump 已提交
249 250 251
virPCIDeviceGetDriverPathAndName(virPCIDevicePtr dev, char **path, char **name)
{
    int ret = -1;
252
    VIR_AUTOFREE(char *) drvlink = NULL;
L
Laine Stump 已提交
253 254 255

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

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

L
Laine Stump 已提交
264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282
    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;
283
 cleanup:
L
Laine Stump 已提交
284 285 286 287 288 289 290 291
    if (ret < 0) {
        VIR_FREE(*path);
        VIR_FREE(*name);
    }
    return ret;
}


292
static int
293
virPCIDeviceConfigOpenInternal(virPCIDevicePtr dev, bool readonly, bool fatal)
294 295 296
{
    int fd;

297
    fd = open(dev->path, readonly ? O_RDONLY : O_RDWR);
298

299
    if (fd < 0) {
300 301 302 303 304 305 306 307 308
        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)));
        }
309 310
        return -1;
    }
311

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

316
static int
317
virPCIDeviceConfigOpen(virPCIDevicePtr dev)
318
{
319
    return virPCIDeviceConfigOpenInternal(dev, true, true);
320 321
}

322 323 324 325 326 327
static int
virPCIDeviceConfigOpenTry(virPCIDevicePtr dev)
{
    return virPCIDeviceConfigOpenInternal(dev, true, false);
}

328 329 330
static int
virPCIDeviceConfigOpenWrite(virPCIDevicePtr dev)
{
331
    return virPCIDeviceConfigOpenInternal(dev, false, true);
332 333
}

334
static void
335
virPCIDeviceConfigClose(virPCIDevicePtr dev, int cfgfd)
336
{
337 338 339 340 341
    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)));
    }
342 343
}

344

345
static int
346 347
virPCIDeviceRead(virPCIDevicePtr dev,
                 int cfgfd,
348
                 unsigned int pos,
349
                 uint8_t *buf,
350
                 unsigned int buflen)
351 352 353
{
    memset(buf, 0, buflen);

354 355
    if (lseek(cfgfd, pos, SEEK_SET) != pos ||
        saferead(cfgfd, buf, buflen) != buflen) {
356
        char ebuf[1024];
357
        VIR_WARN("Failed to read from '%s' : %s", dev->path,
358 359 360 361 362 363 364
                 virStrerror(errno, ebuf, sizeof(ebuf)));
        return -1;
    }
    return 0;
}

static uint8_t
365
virPCIDeviceRead8(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
366 367
{
    uint8_t buf;
368
    virPCIDeviceRead(dev, cfgfd, pos, &buf, sizeof(buf));
369 370 371 372
    return buf;
}

static uint16_t
373
virPCIDeviceRead16(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
374 375
{
    uint8_t buf[2];
376
    virPCIDeviceRead(dev, cfgfd, pos, &buf[0], sizeof(buf));
377 378 379 380
    return (buf[0] << 0) | (buf[1] << 8);
}

static uint32_t
381
virPCIDeviceRead32(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
382 383
{
    uint8_t buf[4];
384
    virPCIDeviceRead(dev, cfgfd, pos, &buf[0], sizeof(buf));
385 386 387
    return (buf[0] << 0) | (buf[1] << 8) | (buf[2] << 16) | (buf[3] << 24);
}

388 389 390
static int
virPCIDeviceReadClass(virPCIDevicePtr dev, uint16_t *device_class)
{
391 392
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) id_str = NULL;
393 394
    unsigned int value;

395
    if (!(path = virPCIFile(dev->name, "class")))
396
        return -1;
397 398 399

    /* class string is '0xNNNNNN\n' ... i.e. 9 bytes */
    if (virFileReadAll(path, 9, &id_str) < 0)
400
        return -1;
401 402 403 404 405 406

    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);
407
        return -1;
408 409 410
    }

    *device_class = (value >> 8) & 0xFFFF;
411
    return 0;
412 413
}

414
static int
415 416
virPCIDeviceWrite(virPCIDevicePtr dev,
                  int cfgfd,
417
                  unsigned int pos,
418
                  uint8_t *buf,
419
                  unsigned int buflen)
420
{
421 422
    if (lseek(cfgfd, pos, SEEK_SET) != pos ||
        safewrite(cfgfd, buf, buflen) != buflen) {
423
        char ebuf[1024];
424
        VIR_WARN("Failed to write to '%s' : %s", dev->path,
425 426 427 428 429 430 431
                 virStrerror(errno, ebuf, sizeof(ebuf)));
        return -1;
    }
    return 0;
}

static void
432
virPCIDeviceWrite16(virPCIDevicePtr dev, int cfgfd, unsigned int pos, uint16_t val)
433 434
{
    uint8_t buf[2] = { (val >> 0), (val >> 8) };
435
    virPCIDeviceWrite(dev, cfgfd, pos, &buf[0], sizeof(buf));
436 437 438
}

static void
439
virPCIDeviceWrite32(virPCIDevicePtr dev, int cfgfd, unsigned int pos, uint32_t val)
440
{
441
    uint8_t buf[4] = { (val >> 0), (val >> 8), (val >> 16), (val >> 24) };
442
    virPCIDeviceWrite(dev, cfgfd, pos, &buf[0], sizeof(buf));
443 444
}

E
Eric Blake 已提交
445 446
typedef int (*virPCIDeviceIterPredicate)(virPCIDevicePtr, virPCIDevicePtr,
                                         void *);
447 448 449 450 451 452 453

/* 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
454 455 456 457
virPCIDeviceIterDevices(virPCIDeviceIterPredicate predicate,
                        virPCIDevicePtr dev,
                        virPCIDevicePtr *matched,
                        void *data)
458 459 460
{
    DIR *dir;
    struct dirent *entry;
461
    int ret = 0;
462
    int rc;
463 464 465 466 467

    *matched = NULL;

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

J
Ján Tomko 已提交
468
    if (virDirOpen(&dir, PCI_SYSFS "devices") < 0)
469 470
        return -1;

E
Eric Blake 已提交
471
    while ((ret = virDirRead(dir, &entry, PCI_SYSFS "devices")) > 0) {
472
        unsigned int domain, bus, slot, function;
473
        VIR_AUTOPTR(virPCIDevice) check = NULL;
474
        char *tmp;
475

476 477 478 479 480 481 482 483 484
        /* 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) {
485 486 487 488
            VIR_WARN("Unusual entry in " PCI_SYSFS "devices: %s", entry->d_name);
            continue;
        }

489
        check = virPCIDeviceNew(domain, bus, slot, function);
490
        if (!check) {
491 492 493
            ret = -1;
            break;
        }
494

495 496 497 498 499
        rc = predicate(dev, check, data);
        if (rc < 0) {
            /* the predicate returned an error, bail */
            ret = -1;
            break;
500
        } else if (rc == 1) {
501
            VIR_DEBUG("%s %s: iter matched on %s", dev->id, dev->name, check->name);
502
            VIR_STEAL_PTR(*matched, check);
503
            ret = 1;
504 505 506
            break;
        }
    }
J
Ján Tomko 已提交
507
    VIR_DIR_CLOSE(dir);
508
    return ret;
509 510 511
}

static uint8_t
512 513 514
virPCIDeviceFindCapabilityOffset(virPCIDevicePtr dev,
                                 int cfgfd,
                                 unsigned int capability)
515 516 517 518
{
    uint16_t status;
    uint8_t pos;

519
    status = virPCIDeviceRead16(dev, cfgfd, PCI_STATUS);
520 521 522
    if (!(status & PCI_STATUS_CAP_LIST))
        return 0;

523
    pos = virPCIDeviceRead8(dev, cfgfd, PCI_CAPABILITY_LIST);
524 525 526 527 528 529 530 531 532

    /* 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) {
533
        uint8_t capid = virPCIDeviceRead8(dev, cfgfd, pos);
534 535 536 537 538 539
        if (capid == capability) {
            VIR_DEBUG("%s %s: found cap 0x%.2x at 0x%.2x",
                      dev->id, dev->name, capability, pos);
            return pos;
        }

540
        pos = virPCIDeviceRead8(dev, cfgfd, pos + 1);
541 542 543 544 545 546 547
    }

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

    return 0;
}

J
Jiri Denemark 已提交
548
static unsigned int
549 550
virPCIDeviceFindExtendedCapabilityOffset(virPCIDevicePtr dev,
                                         int cfgfd,
551
                                         unsigned int capability)
J
Jiri Denemark 已提交
552 553 554 555 556 557 558 559 560 561
{
    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) {
562
        header = virPCIDeviceRead32(dev, cfgfd, pos);
J
Jiri Denemark 已提交
563 564 565 566 567 568 569 570 571 572 573

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

574 575 576 577
/* 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
578
virPCIDeviceDetectFunctionLevelReset(virPCIDevicePtr dev, int cfgfd)
579
{
M
Mark McLoughlin 已提交
580
    uint32_t caps;
581
    uint8_t pos;
582
    VIR_AUTOFREE(char *) path = NULL;
583
    int found;
584 585 586 587 588 589 590 591

    /* 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) {
592
        caps = virPCIDeviceRead32(dev, cfgfd, dev->pcie_cap_pos + PCI_EXP_DEVCAP);
593 594 595 596 597 598 599 600 601 602
        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.
     */
603
    pos = virPCIDeviceFindCapabilityOffset(dev, cfgfd, PCI_CAP_ID_AF);
604
    if (pos) {
605
        caps = virPCIDeviceRead16(dev, cfgfd, pos + PCI_AF_CAP);
606 607 608 609 610 611
        if (caps & PCI_AF_CAP_FLR) {
            VIR_DEBUG("%s %s: detected PCI FLR capability", dev->id, dev->name);
            return 1;
        }
    }

612 613 614 615 616 617
    /* 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
     */

618
    if (virAsprintf(&path, PCI_SYSFS "devices/%s/physfn", dev->name) < 0)
619 620 621 622 623 624 625 626 627
        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;
    }

628 629 630 631 632 633 634 635 636
    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.
 */
637
static unsigned int
638
virPCIDeviceDetectPowerManagementReset(virPCIDevicePtr dev, int cfgfd)
639 640 641 642 643
{
    if (dev->pci_pm_cap_pos) {
        uint32_t ctl;

        /* require the NO_SOFT_RESET bit is clear */
644
        ctl = virPCIDeviceRead32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL);
645 646 647 648 649 650 651 652 653 654 655
        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;
}

656
/* Any active devices on the same domain/bus ? */
657
static int
658
virPCIDeviceSharesBusWithActive(virPCIDevicePtr dev, virPCIDevicePtr check, void *data)
659
{
660
    virPCIDeviceList *inactiveDevs = data;
661

662
    /* Different domain, different bus, or simply identical device */
663 664 665 666
    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))
667 668
        return 0;

669
    /* same bus, but inactive, i.e. about to be assigned to guest */
670
    if (inactiveDevs && virPCIDeviceListFind(inactiveDevs, check))
671
        return 0;
672

673
    return 1;
674 675
}

676 677 678
static virPCIDevicePtr
virPCIDeviceBusContainsActiveDevices(virPCIDevicePtr dev,
                                     virPCIDeviceList *inactiveDevs)
679
{
680 681 682
    virPCIDevicePtr active = NULL;
    if (virPCIDeviceIterDevices(virPCIDeviceSharesBusWithActive,
                                dev, &active, inactiveDevs) < 0)
683 684 685 686 687
        return NULL;
    return active;
}

/* Is @check the parent of @dev ? */
688
static int
689
virPCIDeviceIsParent(virPCIDevicePtr dev, virPCIDevicePtr check, void *data)
690 691 692
{
    uint16_t device_class;
    uint8_t header_type, secondary, subordinate;
693
    virPCIDevicePtr *best = data;
694 695
    int ret = 0;
    int fd;
696

697
    if (dev->address.domain != check->address.domain)
698 699
        return 0;

700
    if ((fd = virPCIDeviceConfigOpenTry(check)) < 0)
701 702
        return 0;

703
    /* Is it a bridge? */
704 705
    ret = virPCIDeviceReadClass(check, &device_class);
    if (ret < 0 || device_class != PCI_CLASS_BRIDGE_PCI)
706
        goto cleanup;
707 708

    /* Is it a plane? */
709
    header_type = virPCIDeviceRead8(check, fd, PCI_HEADER_TYPE);
710
    if ((header_type & PCI_HEADER_TYPE_MASK) != PCI_HEADER_TYPE_BRIDGE)
711
        goto cleanup;
712

713 714
    secondary   = virPCIDeviceRead8(check, fd, PCI_SECONDARY_BUS);
    subordinate = virPCIDeviceRead8(check, fd, PCI_SUBORDINATE_BUS);
715

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

718 719 720
    /* if the secondary bus exactly equals the device's bus, then we found
     * the direct parent.  No further work is necessary
     */
721
    if (dev->address.bus == secondary) {
722 723 724
        ret = 1;
        goto cleanup;
    }
725

726
    /* otherwise, SRIOV allows VFs to be on different buses than their PFs.
727 728 729
     * 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.
     */
730
    if (dev->address.bus > secondary && dev->address.bus <= subordinate) {
731
        if (*best == NULL) {
732 733 734 735
            *best = virPCIDeviceNew(check->address.domain,
                                    check->address.bus,
                                    check->address.slot,
                                    check->address.function);
736 737 738 739 740
            if (*best == NULL) {
                ret = -1;
                goto cleanup;
            }
        } else {
741 742 743 744
            /* 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
             */
745 746 747
            int bestfd;
            uint8_t best_secondary;

748
            if ((bestfd = virPCIDeviceConfigOpenTry(*best)) < 0)
749
                goto cleanup;
750 751
            best_secondary = virPCIDeviceRead8(*best, bestfd, PCI_SECONDARY_BUS);
            virPCIDeviceConfigClose(*best, bestfd);
752 753

            if (secondary > best_secondary) {
754
                virPCIDeviceFree(*best);
755 756 757 758
                *best = virPCIDeviceNew(check->address.domain,
                                        check->address.bus,
                                        check->address.slot,
                                        check->address.function);
759 760 761 762
                if (*best == NULL) {
                    ret = -1;
                    goto cleanup;
                }
763 764 765 766
            }
        }
    }

767
 cleanup:
768
    virPCIDeviceConfigClose(check, fd);
769
    return ret;
770 771
}

772
static int
773
virPCIDeviceGetParent(virPCIDevicePtr dev, virPCIDevicePtr *parent)
774
{
775
    virPCIDevicePtr best = NULL;
776 777 778
    int ret;

    *parent = NULL;
779
    ret = virPCIDeviceIterDevices(virPCIDeviceIsParent, dev, parent, &best);
780
    if (ret == 1)
781
        virPCIDeviceFree(best);
782 783 784
    else if (ret == 0)
        *parent = best;
    return ret;
785 786 787 788 789 790
}

/* Secondary Bus Reset is our sledgehammer - it resets all
 * devices behind a bus.
 */
static int
791 792 793
virPCIDeviceTrySecondaryBusReset(virPCIDevicePtr dev,
                                 int cfgfd,
                                 virPCIDeviceList *inactiveDevs)
794
{
795 796
    VIR_AUTOPTR(virPCIDevice) parent = NULL;
    VIR_AUTOPTR(virPCIDevice) conflict = NULL;
797 798 799
    uint8_t config_space[PCI_CONF_LEN];
    uint16_t ctl;
    int ret = -1;
800
    int parentfd;
801

802 803 804
    /* Refuse to do a secondary bus reset if there are other
     * devices/functions behind the bus are used by the host
     * or other guests.
805
     */
806
    if ((conflict = virPCIDeviceBusContainsActiveDevices(dev, inactiveDevs))) {
807
        virReportError(VIR_ERR_INTERNAL_ERROR,
808 809
                       _("Active %s devices on bus with %s, not doing bus reset"),
                       conflict->name, dev->name);
810 811 812 813
        return -1;
    }

    /* Find the parent bus */
814
    if (virPCIDeviceGetParent(dev, &parent) < 0)
815
        return -1;
816
    if (!parent) {
817
        virReportError(VIR_ERR_INTERNAL_ERROR,
818 819
                       _("Failed to find parent device for %s"),
                       dev->name);
820 821
        return -1;
    }
822
    if ((parentfd = virPCIDeviceConfigOpenWrite(parent)) < 0)
823
        goto out;
824 825 826 827 828 829 830

    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
     */
831
    if (virPCIDeviceRead(dev, cfgfd, 0, config_space, PCI_CONF_LEN) < 0) {
832
        virReportError(VIR_ERR_INTERNAL_ERROR,
833
                       _("Failed to read PCI config space for %s"),
834
                       dev->name);
835 836 837 838 839 840
        goto out;
    }

    /* Read the control register, set the reset flag, wait 200ms,
     * unset the reset flag and wait 200ms.
     */
H
hexin 已提交
841
    ctl = virPCIDeviceRead16(dev, parentfd, PCI_BRIDGE_CONTROL);
842

843 844
    virPCIDeviceWrite16(parent, parentfd, PCI_BRIDGE_CONTROL,
                        ctl | PCI_BRIDGE_CTL_RESET);
845 846 847

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

848
    virPCIDeviceWrite16(parent, parentfd, PCI_BRIDGE_CONTROL, ctl);
849 850 851

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

852
    if (virPCIDeviceWrite(dev, cfgfd, 0, config_space, PCI_CONF_LEN) < 0) {
853
        virReportError(VIR_ERR_INTERNAL_ERROR,
854 855 856 857
                       _("Failed to restore PCI config space for %s"),
                       dev->name);
        goto out;
    }
858
    ret = 0;
859

860
 out:
861
    virPCIDeviceConfigClose(parent, parentfd);
862 863 864 865 866 867 868 869
    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
870
virPCIDeviceTryPowerManagementReset(virPCIDevicePtr dev, int cfgfd)
871 872 873 874 875 876 877 878
{
    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. */
879
    if (virPCIDeviceRead(dev, cfgfd, 0, &config_space[0], PCI_CONF_LEN) < 0) {
880
        virReportError(VIR_ERR_INTERNAL_ERROR,
881
                       _("Failed to read PCI config space for %s"),
882
                       dev->name);
883 884 885 886 887
        return -1;
    }

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

888
    ctl = virPCIDeviceRead32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL);
889 890
    ctl &= ~PCI_PM_CTRL_STATE_MASK;

891 892
    virPCIDeviceWrite32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL,
                        ctl | PCI_PM_CTRL_STATE_D3hot);
893 894 895

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

896 897
    virPCIDeviceWrite32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL,
                        ctl | PCI_PM_CTRL_STATE_D0);
898 899 900

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

901
    if (virPCIDeviceWrite(dev, cfgfd, 0, &config_space[0], PCI_CONF_LEN) < 0) {
902
        virReportError(VIR_ERR_INTERNAL_ERROR,
903 904 905 906
                       _("Failed to restore PCI config space for %s"),
                       dev->name);
        return -1;
    }
907 908 909 910 911

    return 0;
}

static int
912
virPCIDeviceInit(virPCIDevicePtr dev, int cfgfd)
913
{
914 915
    int flr;

916 917 918
    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);
919
    if (flr < 0)
920
        return flr;
921 922
    dev->has_flr        = !!flr;
    dev->has_pm_reset   = !!virPCIDeviceDetectPowerManagementReset(dev, cfgfd);
923

924 925 926 927
    return 0;
}

int
928 929 930
virPCIDeviceReset(virPCIDevicePtr dev,
                  virPCIDeviceList *activeDevs,
                  virPCIDeviceList *inactiveDevs)
931
{
932 933
    VIR_AUTOFREE(char *) drvPath = NULL;
    VIR_AUTOFREE(char *) drvName = NULL;
934
    int ret = -1;
935
    int fd = -1;
936 937 938 939 940 941 942 943 944 945 946 947
    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;
    }
948

949
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
950
        virReportError(VIR_ERR_INTERNAL_ERROR,
951 952 953 954
                       _("Not resetting active device %s"), dev->name);
        return -1;
    }

955 956 957 958 959 960 961 962
    /* 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;

963
    if (virPCIStubDriverTypeFromString(drvName) == VIR_PCI_STUB_DRIVER_VFIO) {
964 965 966 967 968 969 970
        VIR_DEBUG("Device %s is bound to vfio-pci - skip reset",
                  dev->name);
        ret = 0;
        goto cleanup;
    }
    VIR_DEBUG("Resetting device %s", dev->name);

971
    if ((fd = virPCIDeviceConfigOpenWrite(dev)) < 0)
972
        goto cleanup;
973

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

977 978 979
    /* KVM will perform FLR when starting and stopping
     * a guest, so there is no need for us to do it here.
     */
980 981 982 983
    if (dev->has_flr) {
        ret = 0;
        goto cleanup;
    }
984

985 986 987 988 989
    /* 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)
990
        ret = virPCIDeviceTryPowerManagementReset(dev, fd);
991

992
    /* Bus reset is not an option with the root bus */
993
    if (ret < 0 && dev->address.bus != 0)
994
        ret = virPCIDeviceTrySecondaryBusReset(dev, fd, inactiveDevs);
995

996 997
    if (ret < 0) {
        virErrorPtr err = virGetLastError();
998
        virReportError(VIR_ERR_INTERNAL_ERROR,
999 1000
                       _("Unable to reset PCI device %s: %s"),
                       dev->name,
1001 1002
                       err ? err->message :
                       _("no FLR, PM reset or bus reset available"));
1003 1004
    }

1005
 cleanup:
1006
    virPCIDeviceConfigClose(dev, fd);
1007 1008 1009
    return ret;
}

1010

1011
static int
1012
virPCIProbeStubDriver(virPCIStubDriver driver)
1013
{
1014
    const char *drvname = NULL;
1015
    VIR_AUTOFREE(char *) drvpath = NULL;
1016
    bool probed = false;
1017

1018 1019 1020 1021 1022 1023 1024 1025
    if (driver == VIR_PCI_STUB_DRIVER_NONE ||
        !(drvname = virPCIStubDriverTypeToString(driver))) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s",
                       _("Attempting to use unknown stub driver"));
        return -1;
    }

1026
 recheck:
1027
    if ((drvpath = virPCIDriverDir(drvname)) && virFileExists(drvpath))
1028 1029
        /* driver already loaded, return */
        return 0;
1030 1031

    if (!probed) {
1032
        VIR_AUTOFREE(char *) errbuf = NULL;
1033
        probed = true;
1034 1035
        if ((errbuf = virKModLoad(drvname, true))) {
            VIR_WARN("failed to load driver %s: %s", drvname, errbuf);
1036
            goto cleanup;
1037
        }
1038 1039

        goto recheck;
1040 1041
    }

1042
 cleanup:
1043 1044 1045
    /* If we know failure was because of blacklist, let's report that;
     * otherwise, report a more generic failure message
     */
1046
    if (virKModIsBlacklisted(drvname)) {
1047 1048 1049
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to load PCI stub module %s: "
                         "administratively prohibited"),
1050
                       drvname);
1051 1052 1053
    } else {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to load PCI stub module %s"),
1054
                       drvname);
1055 1056
    }

1057
    return -1;
1058 1059
}

1060
int
1061
virPCIDeviceUnbind(virPCIDevicePtr dev)
1062
{
1063 1064 1065
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) drvpath = NULL;
    VIR_AUTOFREE(char *) driver = NULL;
1066 1067

    if (virPCIDeviceGetDriverPathAndName(dev, &drvpath, &driver) < 0)
1068
        return -1;
1069

1070
    if (!driver)
1071
        /* The device is not bound to any driver */
1072
        return 0;
1073

1074
    if (!(path = virPCIFile(dev->name, "driver/unbind")))
1075
        return -1;
1076 1077 1078 1079 1080 1081

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

1086
    return 0;
1087 1088
}

1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113

/**
 * 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;
}


1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124
/*
 * 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.
 */
1125
static int
1126 1127 1128
virPCIDeviceBindWithDriverOverride(virPCIDevicePtr dev,
                                   const char *driverName)
{
1129
    VIR_AUTOFREE(char *) path = NULL;
1130 1131 1132 1133 1134 1135 1136 1137 1138

    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);
1139
        return -1;
1140 1141
    }

1142
    if (virPCIDeviceRebind(dev) < 0)
1143
        return -1;
1144

1145
    return 0;
1146 1147 1148 1149
}

static int
virPCIDeviceUnbindFromStubWithNewid(virPCIDevicePtr dev)
1150 1151
{
    int result = -1;
1152 1153 1154
    VIR_AUTOFREE(char *) drvdir = NULL;
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) driver = NULL;
1155

1156 1157 1158
    /* If the device is currently bound to one of the "well known"
     * stub drivers, then unbind it, otherwise ignore it.
     */
L
Laine Stump 已提交
1159
    if (virPCIDeviceGetDriverPathAndName(dev, &drvdir, &driver) < 0)
1160
        goto cleanup;
E
Eric Blake 已提交
1161

1162 1163
    if (!driver) {
        /* The device is not bound to any driver and we are almost done. */
1164
        VIR_DEBUG("PCI device %s is not bound to any driver", dev->name);
1165 1166 1167
        goto reprobe;
    }

1168 1169
    if (!dev->unbind_from_stub) {
        VIR_DEBUG("Unbind from stub skipped for PCI device %s", dev->name);
1170
        goto remove_slot;
1171
    }
1172

1173
    /* If the device isn't bound to a known stub, skip the unbind. */
1174
    if (virPCIStubDriverTypeFromString(driver) < 0 ||
1175 1176 1177
        virPCIStubDriverTypeFromString(driver) == VIR_PCI_STUB_DRIVER_NONE) {
        VIR_DEBUG("Unbind from stub skipped for PCI device %s because of "
                  "unknown stub driver", dev->name);
1178
        goto remove_slot;
1179
    }
1180

1181 1182
    VIR_DEBUG("Unbinding PCI device %s from stub driver %s",
              dev->name, driver);
1183

1184
    if (virPCIDeviceUnbind(dev) < 0)
1185
        goto cleanup;
1186
    dev->unbind_from_stub = false;
1187

1188
 remove_slot:
1189 1190
    if (!dev->remove_slot) {
        VIR_DEBUG("Slot removal skipped for PCI device %s", dev->name);
1191
        goto reprobe;
1192 1193 1194
    }

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

    /* Xen's pciback.ko wants you to use remove_slot on the specific device */
1197
    if (!(path = virPCIDriverFile(driver, "remove_slot")))
1198 1199 1200 1201
        goto cleanup;

    if (virFileExists(path) && virFileWriteStr(path, dev->name, 0) < 0) {
        virReportSystemError(errno,
1202
                             _("Failed to remove slot for PCI device '%s' from %s"),
1203 1204 1205
                             dev->name, driver);
        goto cleanup;
    }
1206
    dev->remove_slot = false;
1207

1208
 reprobe:
1209
    if (!dev->reprobe) {
1210
        VIR_DEBUG("Reprobe skipped for PCI device %s", dev->name);
1211 1212 1213
        result = 0;
        goto cleanup;
    }
1214

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

1217 1218 1219 1220 1221
    /* 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.
     */
1222 1223
    VIR_FREE(path);
    if (driver && !(path = virPCIDriverFile(driver, "remove_id")))
1224 1225
        goto cleanup;

1226
    if (!driver || !virFileExists(drvdir) || virFileExists(path)) {
1227 1228 1229 1230 1231 1232 1233 1234 1235 1236
        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;

1237
 cleanup:
1238
    /* do not do it again */
1239 1240 1241
    dev->unbind_from_stub = false;
    dev->remove_slot = false;
    dev->reprobe = false;
1242

1243 1244 1245
    return result;
}

1246 1247 1248 1249 1250 1251 1252 1253 1254 1255
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");
}
1256 1257

static int
1258 1259
virPCIDeviceUnbindFromStub(virPCIDevicePtr dev)
{
1260
    VIR_AUTOFREE(char *) path = NULL;
1261 1262 1263 1264 1265 1266 1267 1268 1269

    /*
     * 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))
1270
        return virPCIDeviceUnbindFromStubWithOverride(dev);
1271

1272
    return virPCIDeviceUnbindFromStubWithNewid(dev);
1273 1274 1275 1276
}

static int
virPCIDeviceBindToStubWithNewid(virPCIDevicePtr dev)
1277
{
1278
    int result = -1;
E
Eric Blake 已提交
1279
    bool reprobe = false;
1280 1281 1282
    VIR_AUTOFREE(char *) stubDriverPath = NULL;
    VIR_AUTOFREE(char *) driverLink = NULL;
    VIR_AUTOFREE(char *) path = NULL; /* reused for different purposes */
1283
    VIR_AUTOPTR(virError) err = NULL;
1284
    const char *stubDriverName = NULL;
1285

1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298
    /* 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;
    }

1299
    if (!(stubDriverPath = virPCIDriverDir(stubDriverName))  ||
1300
        !(driverLink = virPCIFile(dev->name, "driver")))
1301 1302
        goto cleanup;

1303 1304 1305 1306 1307
    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);
1308 1309 1310
            result = 0;
            goto cleanup;
        }
1311
        reprobe = true;
1312
    }
1313 1314 1315 1316 1317 1318 1319 1320 1321

    /* 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.
     */
1322
    if (!(path = virPCIDriverFile(stubDriverName, "new_id")))
1323
        goto cleanup;
1324

1325
    if (virFileWriteStr(path, dev->id, 0) < 0) {
1326
        virReportSystemError(errno,
1327
                             _("Failed to add PCI device ID '%s' to %s"),
1328
                             dev->id, stubDriverName);
1329
        goto cleanup;
1330 1331
    }

1332
    /* check whether the device is bound to pci-stub when we write dev->id to
1333
     * ${stubDriver}/new_id.
1334
     */
1335
    if (virFileLinkPointsTo(driverLink, stubDriverPath)) {
1336 1337
        dev->unbind_from_stub = true;
        dev->remove_slot = true;
J
Jiri Denemark 已提交
1338
        result = 0;
1339 1340 1341
        goto remove_id;
    }

1342
    if (virPCIDeviceUnbind(dev) < 0)
J
Jiri Denemark 已提交
1343
        goto remove_id;
1344

1345 1346 1347
    /* If the device was bound to a driver we'll need to reprobe later */
    dev->reprobe = reprobe;

1348 1349
    /* If the device isn't already bound to pci-stub, try binding it now.
     */
1350
    if (!virFileLinkPointsTo(driverLink, stubDriverPath)) {
1351
        /* Xen's pciback.ko wants you to use new_slot first */
1352 1353
        VIR_FREE(path);
        if (!(path = virPCIDriverFile(stubDriverName, "new_slot")))
1354
            goto remove_id;
1355

1356
        if (virFileExists(path) && virFileWriteStr(path, dev->name, 0) < 0) {
1357
            virReportSystemError(errno,
1358 1359 1360
                                 _("Failed to add slot for "
                                   "PCI device '%s' to %s"),
                                 dev->name, stubDriverName);
1361
            goto remove_id;
1362
        }
1363
        dev->remove_slot = true;
1364

1365 1366
        VIR_FREE(path);
        if (!(path = virPCIDriverFile(stubDriverName, "bind")))
1367
            goto remove_id;
1368

1369
        if (virFileWriteStr(path, dev->name, 0) < 0) {
1370
            virReportSystemError(errno,
1371
                                 _("Failed to bind PCI device '%s' to %s"),
1372
                                 dev->name, stubDriverName);
1373
            goto remove_id;
1374
        }
1375
        dev->unbind_from_stub = true;
1376 1377
    }

J
Jiri Denemark 已提交
1378 1379
    result = 0;

1380
 remove_id:
J
Jiri Denemark 已提交
1381 1382
    err = virSaveLastError();

1383 1384 1385
    /* If 'remove_id' exists, remove the device id from pci-stub's dynamic
     * ID table so that 'drivers_probe' works below.
     */
1386 1387
    VIR_FREE(path);
    if (!(path = virPCIDriverFile(stubDriverName, "remove_id"))) {
E
Eric Blake 已提交
1388
        /* We do not remove PCI ID from pci-stub, and we cannot reprobe it */
1389 1390
        if (dev->reprobe) {
            VIR_WARN("Could not remove PCI ID '%s' from %s, and the device "
1391
                     "cannot be probed again.", dev->id, stubDriverName);
1392
        }
1393
        dev->reprobe = false;
J
Jiri Denemark 已提交
1394
        result = -1;
1395 1396 1397
        goto cleanup;
    }

1398
    if (virFileExists(path) && virFileWriteStr(path, dev->id, 0) < 0) {
1399
        virReportSystemError(errno,
1400
                             _("Failed to remove PCI ID '%s' from %s"),
1401
                             dev->id, stubDriverName);
1402

E
Eric Blake 已提交
1403
        /* remove PCI ID from pci-stub failed, and we cannot reprobe it */
1404 1405
        if (dev->reprobe) {
            VIR_WARN("Failed to remove PCI ID '%s' from %s, and the device "
1406
                     "cannot be probed again.", dev->id, stubDriverName);
1407
        }
1408
        dev->reprobe = false;
J
Jiri Denemark 已提交
1409
        result = -1;
1410
        goto cleanup;
1411 1412
    }

1413
 cleanup:
1414
    if (result < 0)
J
Jiri Denemark 已提交
1415 1416 1417 1418
        virPCIDeviceUnbindFromStub(dev);

    if (err)
        virSetError(err);
1419

1420
    return result;
1421 1422
}

1423 1424 1425 1426
static int
virPCIDeviceBindToStubWithOverride(virPCIDevicePtr dev)
{
    const char *stubDriverName;
1427 1428
    VIR_AUTOFREE(char *) stubDriverPath = NULL;
    VIR_AUTOFREE(char *) driverLink = NULL;
1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444

    /* 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")))
1445
        return -1;
1446 1447 1448 1449 1450 1451

    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);
1452
            return 0;
1453 1454 1455 1456
        }
    }

    if (virPCIDeviceBindWithDriverOverride(dev, stubDriverName) < 0)
1457
        return -1;
1458 1459

    dev->unbind_from_stub = true;
1460
    return 0;
1461 1462 1463 1464 1465
}

static int
virPCIDeviceBindToStub(virPCIDevicePtr dev)
{
1466
    VIR_AUTOFREE(char *) path = NULL;
1467 1468 1469 1470 1471 1472 1473 1474 1475

    /*
     * 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))
1476
        return virPCIDeviceBindToStubWithOverride(dev);
1477

1478
    return virPCIDeviceBindToStubWithNewid(dev);
1479 1480
}

1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498
/* 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.
 */
1499
int
1500 1501
virPCIDeviceDetach(virPCIDevicePtr dev,
                   virPCIDeviceList *activeDevs,
1502
                   virPCIDeviceList *inactiveDevs)
1503
{
1504
    if (virPCIProbeStubDriver(dev->stubDriver) < 0)
1505 1506
        return -1;

1507
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
1508
        virReportError(VIR_ERR_INTERNAL_ERROR,
1509 1510 1511 1512
                       _("Not detaching active device %s"), dev->name);
        return -1;
    }

1513
    if (virPCIDeviceBindToStub(dev) < 0)
1514 1515
        return -1;

1516 1517 1518
    /* Add *a copy of* the dev into list inactiveDevs, if
     * it's not already there.
     */
1519 1520 1521 1522
    if (inactiveDevs && !virPCIDeviceListFind(inactiveDevs, dev)) {
        VIR_DEBUG("Adding PCI device %s to inactive list", dev->name);
        if (virPCIDeviceListAddCopy(inactiveDevs, dev) < 0)
            return -1;
1523 1524 1525
    }

    return 0;
1526 1527
}

1528 1529 1530 1531
/*
 * Pre-condition: inactivePCIHostdevs & activePCIHostdevs
 * are locked
 */
1532
int
1533 1534
virPCIDeviceReattach(virPCIDevicePtr dev,
                     virPCIDeviceListPtr activeDevs,
1535
                     virPCIDeviceListPtr inactiveDevs)
1536
{
1537
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
1538
        virReportError(VIR_ERR_INTERNAL_ERROR,
1539 1540 1541 1542
                       _("Not reattaching active device %s"), dev->name);
        return -1;
    }

1543
    if (virPCIDeviceUnbindFromStub(dev) < 0)
1544 1545 1546
        return -1;

    /* Steal the dev from list inactiveDevs */
1547 1548
    if (inactiveDevs) {
        VIR_DEBUG("Removing PCI device %s from inactive list", dev->name);
1549
        virPCIDeviceListDel(inactiveDevs, dev);
1550
    }
1551 1552

    return 0;
1553 1554
}

1555 1556 1557 1558 1559
/* 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 已提交
1560 1561
 * 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.
1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581
 *
 * 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 已提交
1582
 * holding on to the resource.
1583 1584
 */
int
1585
virPCIDeviceWaitForCleanup(virPCIDevicePtr dev, const char *matcher)
1586 1587 1588
{
    FILE *fp;
    char line[160];
1589
    char *tmp;
1590
    unsigned long long start, end;
1591
    unsigned int domain, bus, slot, function;
1592
    bool in_matching_device;
1593 1594 1595 1596 1597 1598 1599 1600 1601
    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
         */
1602
        VIR_DEBUG("Failed to open /proc/iomem, trying to continue anyway");
1603 1604 1605 1606
        return 0;
    }

    ret = 0;
1607
    in_matching_device = false;
1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618
    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))) {
1619 1620 1621 1622 1623 1624
            /* 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)
1625 1626
                continue;

1627
            if (STRPREFIX(tmp, matcher)) {
1628 1629 1630
                ret = 1;
                break;
            }
1631
        } else {
1632
            in_matching_device = false;
1633

1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647
            /* 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')
1648 1649
                continue;

1650 1651
            if (domain != dev->address.domain || bus != dev->address.bus ||
                slot != dev->address.slot || function != dev->address.function)
1652
                continue;
1653
            in_matching_device = true;
1654 1655 1656 1657
            match_depth = strspn(line, " ");
        }
    }

E
Eric Blake 已提交
1658
    VIR_FORCE_FCLOSE(fp);
1659 1660 1661 1662

    return ret;
}

1663
static char *
1664
virPCIDeviceReadID(virPCIDevicePtr dev, const char *id_name)
1665
{
1666
    VIR_AUTOFREE(char *) path = NULL;
1667 1668
    char *id_str;

1669
    if (!(path = virPCIFile(dev->name, id_name)))
1670
        return NULL;
1671 1672

    /* ID string is '0xNNNN\n' ... i.e. 7 bytes */
1673
    if (virFileReadAll(path, 7, &id_str) < 0)
1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687
        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;
}

1688 1689 1690 1691
bool
virPCIDeviceAddressIsValid(virPCIDeviceAddressPtr addr,
                           bool report)
{
1692
    if (addr->domain > 0xFFFFFFFF) {
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 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740
        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
1741 1742
virPCIDeviceAddressEqual(const virPCIDeviceAddress *addr1,
                         const virPCIDeviceAddress *addr2)
1743 1744 1745 1746 1747 1748 1749 1750 1751 1752
{
    if (addr1->domain == addr2->domain &&
        addr1->bus == addr2->bus &&
        addr1->slot == addr2->slot &&
        addr1->function == addr2->function) {
        return true;
    }
    return false;
}

1753
char *
1754
virPCIDeviceAddressAsString(const virPCIDeviceAddress *addr)
1755 1756 1757
{
    char *str;

1758 1759
    ignore_value(virAsprintf(&str,
                             VIR_PCI_DEVICE_ADDRESS_FMT,
1760 1761 1762 1763 1764 1765 1766
                             addr->domain,
                             addr->bus,
                             addr->slot,
                             addr->function));
    return str;
}

1767
virPCIDevicePtr
1768 1769 1770 1771
virPCIDeviceNew(unsigned int domain,
                unsigned int bus,
                unsigned int slot,
                unsigned int function)
1772
{
1773
    VIR_AUTOPTR(virPCIDevice) dev = NULL;
1774 1775
    VIR_AUTOFREE(char *) vendor = NULL;
    VIR_AUTOFREE(char *) product = NULL;
1776

1777
    if (VIR_ALLOC(dev) < 0)
1778 1779
        return NULL;

1780 1781 1782 1783
    dev->address.domain = domain;
    dev->address.bus = bus;
    dev->address.slot = slot;
    dev->address.function = function;
1784

1785
    if (virAsprintf(&dev->name,
1786
                    VIR_PCI_DEVICE_ADDRESS_FMT,
1787
                    domain, bus, slot, function) < 0)
1788
        return NULL;
1789

E
Eric Blake 已提交
1790
    if (virAsprintf(&dev->path, PCI_SYSFS "devices/%s/config",
1791
                    dev->name) < 0)
1792
        return NULL;
1793

1794
    if (!virFileExists(dev->path)) {
1795 1796 1797
        virReportSystemError(errno,
                             _("Device %s not found: could not access %s"),
                             dev->name, dev->path);
1798
        return NULL;
1799 1800
    }

1801 1802
    vendor  = virPCIDeviceReadID(dev, "vendor");
    product = virPCIDeviceReadID(dev, "device");
1803 1804

    if (!vendor || !product) {
1805
        virReportError(VIR_ERR_INTERNAL_ERROR,
1806 1807
                       _("Failed to read product/vendor ID for %s"),
                       dev->name);
1808
        return NULL;
1809 1810 1811
    }

    /* strings contain '0x' prefix */
E
Eric Blake 已提交
1812 1813
    if (snprintf(dev->id, sizeof(dev->id), "%s %s", &vendor[2],
                 &product[2]) >= sizeof(dev->id)) {
1814
        virReportError(VIR_ERR_INTERNAL_ERROR,
E
Eric Blake 已提交
1815 1816
                       _("dev->id buffer overflow: %s %s"),
                       &vendor[2], &product[2]);
1817
        return NULL;
E
Eric Blake 已提交
1818
    }
1819 1820 1821

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

1822
    VIR_RETURN_PTR(dev);
1823 1824
}

L
Laine Stump 已提交
1825 1826 1827 1828 1829 1830

virPCIDevicePtr
virPCIDeviceCopy(virPCIDevicePtr dev)
{
    virPCIDevicePtr copy;

1831
    if (VIR_ALLOC(copy) < 0)
L
Laine Stump 已提交
1832 1833 1834 1835
        return NULL;

    /* shallow copy to take care of most attributes */
    *copy = *dev;
1836
    copy->path = NULL;
C
Chunyan Liu 已提交
1837
    copy->used_by_drvname = copy->used_by_domname = NULL;
1838 1839
    if (VIR_STRDUP(copy->name, dev->name) < 0 ||
        VIR_STRDUP(copy->path, dev->path) < 0 ||
C
Chunyan Liu 已提交
1840 1841
        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 已提交
1842 1843 1844 1845
        goto error;
    }
    return copy;

1846
 error:
L
Laine Stump 已提交
1847 1848 1849 1850 1851
    virPCIDeviceFree(copy);
    return NULL;
}


1852
void
1853
virPCIDeviceFree(virPCIDevicePtr dev)
1854
{
1855 1856
    if (!dev)
        return;
1857
    VIR_DEBUG("%s %s: freeing", dev->id, dev->name);
1858
    VIR_FREE(dev->name);
E
Eric Blake 已提交
1859
    VIR_FREE(dev->path);
C
Chunyan Liu 已提交
1860 1861
    VIR_FREE(dev->used_by_drvname);
    VIR_FREE(dev->used_by_domname);
1862 1863
    VIR_FREE(dev);
}
1864

1865 1866 1867 1868 1869
/**
 * virPCIDeviceGetAddress:
 * @dev: device to get address from
 *
 * Take a PCI device on input and return its PCI address. The
1870
 * returned object is owned by the device and must not be freed.
1871
 *
1872
 * Returns: a pointer to the address, which can never be NULL.
1873 1874 1875 1876
 */
virPCIDeviceAddressPtr
virPCIDeviceGetAddress(virPCIDevicePtr dev)
{
1877
    return &(dev->address);
1878 1879
}

1880
const char *
1881
virPCIDeviceGetName(virPCIDevicePtr dev)
1882 1883 1884 1885
{
    return dev->name;
}

1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897
/**
 * virPCIDeviceGetConfigPath:
 *
 * Returns a pointer to a string containing the path of @dev's PCI
 * config file.
 */
const char *
virPCIDeviceGetConfigPath(virPCIDevicePtr dev)
{
    return dev->path;
}

1898
void virPCIDeviceSetManaged(virPCIDevicePtr dev, bool managed)
1899
{
1900
    dev->managed = managed;
1901 1902
}

1903
bool
1904
virPCIDeviceGetManaged(virPCIDevicePtr dev)
1905 1906 1907 1908
{
    return dev->managed;
}

1909 1910
void
virPCIDeviceSetStubDriver(virPCIDevicePtr dev, virPCIStubDriver driver)
1911
{
1912
    dev->stubDriver = driver;
1913 1914
}

1915
virPCIStubDriver
1916 1917 1918 1919 1920
virPCIDeviceGetStubDriver(virPCIDevicePtr dev)
{
    return dev->stubDriver;
}

1921
bool
1922
virPCIDeviceGetUnbindFromStub(virPCIDevicePtr dev)
1923 1924 1925 1926 1927
{
    return dev->unbind_from_stub;
}

void
1928
virPCIDeviceSetUnbindFromStub(virPCIDevicePtr dev, bool unbind)
1929
{
1930
    dev->unbind_from_stub = unbind;
1931 1932
}

1933
bool
1934
virPCIDeviceGetRemoveSlot(virPCIDevicePtr dev)
1935 1936 1937 1938 1939
{
    return dev->remove_slot;
}

void
1940
virPCIDeviceSetRemoveSlot(virPCIDevicePtr dev, bool remove_slot)
1941
{
1942
    dev->remove_slot = remove_slot;
1943 1944
}

1945
bool
1946
virPCIDeviceGetReprobe(virPCIDevicePtr dev)
1947 1948 1949 1950 1951
{
    return dev->reprobe;
}

void
1952
virPCIDeviceSetReprobe(virPCIDevicePtr dev, bool reprobe)
1953
{
1954
    dev->reprobe = reprobe;
1955 1956
}

C
Chunyan Liu 已提交
1957 1958 1959 1960
int
virPCIDeviceSetUsedBy(virPCIDevicePtr dev,
                      const char *drv_name,
                      const char *dom_name)
1961
{
C
Chunyan Liu 已提交
1962 1963 1964 1965 1966 1967 1968 1969
    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;
1970 1971
}

C
Chunyan Liu 已提交
1972 1973 1974 1975
void
virPCIDeviceGetUsedBy(virPCIDevicePtr dev,
                      const char **drv_name,
                      const char **dom_name)
1976
{
C
Chunyan Liu 已提交
1977 1978
    *drv_name = dev->used_by_drvname;
    *dom_name = dev->used_by_domname;
1979 1980
}

1981 1982
virPCIDeviceListPtr
virPCIDeviceListNew(void)
1983
{
1984
    virPCIDeviceListPtr list;
1985

1986 1987 1988 1989
    if (virPCIInitialize() < 0)
        return NULL;

    if (!(list = virObjectLockableNew(virPCIDeviceListClass)))
1990 1991 1992 1993 1994
        return NULL;

    return list;
}

1995 1996
static void
virPCIDeviceListDispose(void *obj)
1997
{
1998
    virPCIDeviceListPtr list = obj;
1999
    size_t i;
2000 2001

    for (i = 0; i < list->count; i++) {
2002
        virPCIDeviceFree(list->devs[i]);
2003 2004 2005 2006 2007 2008 2009 2010
        list->devs[i] = NULL;
    }

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

int
2011 2012
virPCIDeviceListAdd(virPCIDeviceListPtr list,
                    virPCIDevicePtr dev)
2013
{
2014
    if (virPCIDeviceListFind(list, dev)) {
2015
        virReportError(VIR_ERR_INTERNAL_ERROR,
2016 2017 2018
                       _("Device %s is already in use"), dev->name);
        return -1;
    }
2019
    return VIR_APPEND_ELEMENT(list->devs, list->count, dev);
2020 2021
}

L
Laine Stump 已提交
2022 2023 2024 2025 2026

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

    if (!copy)
        return -1;
2031
    if (virPCIDeviceListAdd(list, copy) < 0)
L
Laine Stump 已提交
2032
        return -1;
2033 2034

    copy = NULL;
L
Laine Stump 已提交
2035 2036 2037 2038
    return 0;
}


2039 2040 2041
virPCIDevicePtr
virPCIDeviceListGet(virPCIDeviceListPtr list,
                    int idx)
2042 2043 2044 2045 2046 2047 2048 2049 2050
{
    if (idx >= list->count)
        return NULL;
    if (idx < 0)
        return NULL;

    return list->devs[idx];
}

2051
size_t
2052
virPCIDeviceListCount(virPCIDeviceListPtr list)
2053
{
2054 2055 2056
    return list->count;
}

2057 2058 2059
virPCIDevicePtr
virPCIDeviceListStealIndex(virPCIDeviceListPtr list,
                           int idx)
2060
{
2061
    virPCIDevicePtr ret;
2062

2063 2064
    if (idx < 0 || idx >= list->count)
        return NULL;
2065

2066
    ret = list->devs[idx];
2067
    VIR_DELETE_ELEMENT(list->devs, idx, list->count);
2068 2069 2070
    return ret;
}

2071 2072 2073
virPCIDevicePtr
virPCIDeviceListSteal(virPCIDeviceListPtr list,
                      virPCIDevicePtr dev)
2074
{
2075
    return virPCIDeviceListStealIndex(list, virPCIDeviceListFindIndex(list, dev));
2076 2077
}

2078
void
2079 2080
virPCIDeviceListDel(virPCIDeviceListPtr list,
                    virPCIDevicePtr dev)
2081
{
2082
    virPCIDeviceFree(virPCIDeviceListSteal(list, dev));
2083 2084
}

2085
int
2086
virPCIDeviceListFindIndex(virPCIDeviceListPtr list, virPCIDevicePtr dev)
2087
{
2088
    size_t i;
2089

2090 2091 2092 2093 2094 2095
    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)
2096
            return i;
2097
    }
2098 2099 2100
    return -1;
}

L
Laine Stump 已提交
2101 2102 2103 2104 2105 2106 2107 2108

virPCIDevicePtr
virPCIDeviceListFindByIDs(virPCIDeviceListPtr list,
                          unsigned int domain,
                          unsigned int bus,
                          unsigned int slot,
                          unsigned int function)
{
2109
    size_t i;
L
Laine Stump 已提交
2110 2111

    for (i = 0; i < list->count; i++) {
2112 2113 2114 2115 2116
        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 已提交
2117 2118 2119 2120 2121 2122
            return list->devs[i];
    }
    return NULL;
}


2123 2124
virPCIDevicePtr
virPCIDeviceListFind(virPCIDeviceListPtr list, virPCIDevicePtr dev)
2125
{
2126
    int idx;
2127

2128 2129
    if ((idx = virPCIDeviceListFindIndex(list, dev)) >= 0)
        return list->devs[idx];
2130 2131
    else
        return NULL;
2132
}
2133 2134


2135 2136 2137
int virPCIDeviceFileIterate(virPCIDevicePtr dev,
                            virPCIDeviceFileActor actor,
                            void *opaque)
2138
{
2139
    VIR_AUTOFREE(char *) pcidir = NULL;
2140 2141 2142
    DIR *dir = NULL;
    int ret = -1;
    struct dirent *ent;
E
Eric Blake 已提交
2143
    int direrr;
2144

2145
    if (virAsprintf(&pcidir, "/sys/bus/pci/devices/" VIR_PCI_DEVICE_ADDRESS_FMT,
2146 2147
                    dev->address.domain, dev->address.bus,
                    dev->address.slot, dev->address.function) < 0)
2148 2149
        goto cleanup;

J
Ján Tomko 已提交
2150
    if (virDirOpen(&dir, pcidir) < 0)
2151 2152
        goto cleanup;

E
Eric Blake 已提交
2153
    while ((direrr = virDirRead(dir, &ent, pcidir)) > 0) {
2154
        VIR_AUTOFREE(char *) file = NULL;
2155
        /* Device assignment requires:
A
Alex Williamson 已提交
2156
         *   $PCIDIR/config, $PCIDIR/resource, $PCIDIR/resourceNNN,
2157
         *   $PCIDIR/rom, $PCIDIR/reset, $PCIDIR/vendor, $PCIDIR/device
2158 2159 2160
         */
        if (STREQ(ent->d_name, "config") ||
            STRPREFIX(ent->d_name, "resource") ||
A
Alex Williamson 已提交
2161
            STREQ(ent->d_name, "rom") ||
2162 2163
            STREQ(ent->d_name, "vendor") ||
            STREQ(ent->d_name, "device") ||
A
Alex Williamson 已提交
2164
            STREQ(ent->d_name, "reset")) {
2165
            if (virAsprintf(&file, "%s/%s", pcidir, ent->d_name) < 0)
2166
                goto cleanup;
2167
            if ((actor)(dev, file, opaque) < 0)
2168 2169 2170
                goto cleanup;
        }
    }
E
Eric Blake 已提交
2171 2172
    if (direrr < 0)
        goto cleanup;
2173 2174 2175

    ret = 0;

2176
 cleanup:
J
Ján Tomko 已提交
2177
    VIR_DIR_CLOSE(dir);
2178 2179
    return ret;
}
J
Jiri Denemark 已提交
2180

L
Laine Stump 已提交
2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191

/* 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)
{
2192
    VIR_AUTOFREE(char *) groupPath = NULL;
L
Laine Stump 已提交
2193 2194 2195
    DIR *groupDir = NULL;
    int ret = -1;
    struct dirent *ent;
E
Eric Blake 已提交
2196
    int direrr;
L
Laine Stump 已提交
2197 2198

    if (virAsprintf(&groupPath,
2199
                    PCI_SYSFS "devices/" VIR_PCI_DEVICE_ADDRESS_FMT "/iommu_group/devices",
2200
                    orig->domain, orig->bus, orig->slot, orig->function) < 0)
L
Laine Stump 已提交
2201 2202
        goto cleanup;

J
Ján Tomko 已提交
2203
    if (virDirOpenQuiet(&groupDir, groupPath) < 0) {
L
Laine Stump 已提交
2204 2205 2206 2207 2208
        /* just process the original device, nothing more */
        ret = (actor)(orig, opaque);
        goto cleanup;
    }

E
Eric Blake 已提交
2209
    while ((direrr = virDirRead(groupDir, &ent, groupPath)) > 0) {
L
Laine Stump 已提交
2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221
        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 已提交
2222
    if (direrr < 0)
L
Laine Stump 已提交
2223 2224 2225 2226
        goto cleanup;

    ret = 0;

2227
 cleanup:
J
Ján Tomko 已提交
2228
    VIR_DIR_CLOSE(groupDir);
L
Laine Stump 已提交
2229 2230 2231 2232 2233 2234 2235 2236
    return ret;
}


static int
virPCIDeviceGetIOMMUGroupAddOne(virPCIDeviceAddressPtr newDevAddr, void *opaque)
{
    virPCIDeviceListPtr groupList = opaque;
2237
    VIR_AUTOPTR(virPCIDevice) newDev = NULL;
L
Laine Stump 已提交
2238 2239 2240

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

    if (virPCIDeviceListAdd(groupList, newDev) < 0)
2244
        return -1;
L
Laine Stump 已提交
2245 2246

    newDev = NULL; /* it's now on the list */
2247
    return 0;
L
Laine Stump 已提交
2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264
}


/*
 * 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;

2265
    if (virPCIDeviceAddressIOMMUGroupIterate(&(dev->address),
L
Laine Stump 已提交
2266 2267 2268 2269 2270 2271
                                             virPCIDeviceGetIOMMUGroupAddOne,
                                             groupList) < 0)
        goto error;

    return groupList;

2272
 error:
L
Laine Stump 已提交
2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296
    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;

2297 2298
    if (VIR_APPEND_ELEMENT(*addrList->iommuGroupDevices,
                           *addrList->nIommuGroupDevices, copyAddr) < 0)
L
Laine Stump 已提交
2299 2300 2301
        goto cleanup;

    ret = 0;
2302
 cleanup:
L
Laine Stump 已提交
2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329
    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;
2330
 cleanup:
L
Laine Stump 已提交
2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341
    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)
{
2342 2343 2344
    VIR_AUTOFREE(char *) devName = NULL;
    VIR_AUTOFREE(char *) devPath = NULL;
    VIR_AUTOFREE(char *) groupPath = NULL;
L
Laine Stump 已提交
2345 2346 2347
    const char *groupNumStr;
    unsigned int groupNum;

2348 2349 2350
    if (virAsprintf(&devName,
                    VIR_PCI_DEVICE_ADDRESS_FMT,
                    addr->domain, addr->bus, addr->slot, addr->function) < 0)
2351
        return -1;
L
Laine Stump 已提交
2352

2353
    if (!(devPath = virPCIFile(devName, "iommu_group")))
2354 2355 2356
        return -1;
    if (virFileIsLink(devPath) != 1)
        return -2;
L
Laine Stump 已提交
2357 2358 2359 2360
    if (virFileResolveLink(devPath, &groupPath) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to resolve device %s iommu_group symlink %s"),
                       devName, devPath);
2361
        return -1;
L
Laine Stump 已提交
2362 2363 2364 2365 2366 2367 2368 2369
    }

    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);
2370
        return -1;
L
Laine Stump 已提交
2371 2372
    }

2373
    return groupNum;
L
Laine Stump 已提交
2374 2375 2376
}


2377 2378
/* virPCIDeviceGetIOMMUGroupDev - return the name of the device used
 * to control this PCI device's group (e.g. "/dev/vfio/15")
2379 2380
 */
char *
2381
virPCIDeviceGetIOMMUGroupDev(virPCIDevicePtr dev)
2382
{
2383 2384
    VIR_AUTOFREE(char *) devPath = NULL;
    VIR_AUTOFREE(char *) groupPath = NULL;
2385 2386
    char *groupDev = NULL;

2387
    if (!(devPath = virPCIFile(dev->name, "iommu_group")))
2388
        return NULL;
2389 2390 2391 2392
    if (virFileIsLink(devPath) != 1) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Invalid device %s iommu_group file %s is not a symlink"),
                       dev->name, devPath);
2393
        return NULL;
2394 2395 2396 2397 2398
    }
    if (virFileResolveLink(devPath, &groupPath) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Unable to resolve device %s iommu_group symlink %s"),
                       dev->name, devPath);
2399
        return NULL;
2400 2401
    }
    if (virAsprintf(&groupDev, "/dev/vfio/%s",
2402
                    last_component(groupPath)) < 0)
2403 2404
        return NULL;

2405 2406 2407
    return groupDev;
}

J
Jiri Denemark 已提交
2408
static int
2409
virPCIDeviceDownstreamLacksACS(virPCIDevicePtr dev)
J
Jiri Denemark 已提交
2410 2411 2412 2413
{
    uint16_t flags;
    uint16_t ctrl;
    unsigned int pos;
2414 2415
    int fd;
    int ret = 0;
2416
    uint16_t device_class;
J
Jiri Denemark 已提交
2417

2418
    if ((fd = virPCIDeviceConfigOpen(dev)) < 0)
J
Jiri Denemark 已提交
2419 2420
        return -1;

2421
    if (virPCIDeviceInit(dev, fd) < 0) {
2422 2423 2424 2425
        ret = -1;
        goto cleanup;
    }

2426 2427 2428
    if (virPCIDeviceReadClass(dev, &device_class) < 0)
        goto cleanup;

J
Jiri Denemark 已提交
2429
    pos = dev->pcie_cap_pos;
2430
    if (!pos || device_class != PCI_CLASS_BRIDGE_PCI)
2431
        goto cleanup;
J
Jiri Denemark 已提交
2432

2433
    flags = virPCIDeviceRead16(dev, fd, pos + PCI_EXP_FLAGS);
J
Jiri Denemark 已提交
2434
    if (((flags & PCI_EXP_FLAGS_TYPE) >> 4) != PCI_EXP_TYPE_DOWNSTREAM)
2435
        goto cleanup;
J
Jiri Denemark 已提交
2436

2437
    pos = virPCIDeviceFindExtendedCapabilityOffset(dev, fd, PCI_EXT_CAP_ID_ACS);
J
Jiri Denemark 已提交
2438 2439
    if (!pos) {
        VIR_DEBUG("%s %s: downstream port lacks ACS", dev->id, dev->name);
2440 2441
        ret = 1;
        goto cleanup;
J
Jiri Denemark 已提交
2442 2443
    }

2444
    ctrl = virPCIDeviceRead16(dev, fd, pos + PCI_EXT_ACS_CTRL);
J
Jiri Denemark 已提交
2445 2446 2447
    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);
2448 2449
        ret = 1;
        goto cleanup;
J
Jiri Denemark 已提交
2450 2451
    }

2452
 cleanup:
2453
    virPCIDeviceConfigClose(dev, fd);
2454
    return ret;
J
Jiri Denemark 已提交
2455 2456 2457
}

static int
2458
virPCIDeviceIsBehindSwitchLackingACS(virPCIDevicePtr dev)
J
Jiri Denemark 已提交
2459
{
2460
    VIR_AUTOPTR(virPCIDevice) parent = NULL;
J
Jiri Denemark 已提交
2461

2462
    if (virPCIDeviceGetParent(dev, &parent) < 0)
2463
        return -1;
2464 2465 2466 2467 2468
    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.
         */
2469
        if (dev->address.bus == 0) {
2470
            return 0;
2471
        } else {
2472
            virReportError(VIR_ERR_INTERNAL_ERROR,
2473 2474 2475 2476
                           _("Failed to find parent device for %s"),
                           dev->name);
            return -1;
        }
J
Jiri Denemark 已提交
2477 2478 2479 2480 2481 2482 2483
    }

    /* 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 {
2484
        VIR_AUTOPTR(virPCIDevice) tmp = NULL;
J
Jiri Denemark 已提交
2485
        int acs;
2486
        int ret;
J
Jiri Denemark 已提交
2487

2488
        acs = virPCIDeviceDownstreamLacksACS(parent);
J
Jiri Denemark 已提交
2489 2490 2491 2492 2493 2494 2495 2496 2497

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

        tmp = parent;
2498
        ret = virPCIDeviceGetParent(parent, &parent);
2499 2500
        if (ret < 0)
            return -1;
J
Jiri Denemark 已提交
2501 2502 2503 2504 2505
    } while (parent);

    return 0;
}

2506 2507
int virPCIDeviceIsAssignable(virPCIDevicePtr dev,
                             int strict_acs_check)
J
Jiri Denemark 已提交
2508 2509 2510 2511 2512 2513 2514 2515
{
    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.
     */

2516
    ret = virPCIDeviceIsBehindSwitchLackingACS(dev);
J
Jiri Denemark 已提交
2517 2518 2519 2520 2521 2522 2523 2524
    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 {
2525
            virReportError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
2526 2527 2528 2529 2530 2531 2532 2533 2534
                           _("Device %s is behind a switch lacking ACS and "
                             "cannot be assigned"),
                           dev->name);
            return 0;
        }
    }

    return 1;
}
2535 2536 2537 2538 2539 2540 2541 2542 2543 2544

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);
2545
    if (ret != 0)
2546 2547 2548 2549
        VIR_ERROR(_("Failed to convert '%s' to unsigned int"), s);
    return ret;
}

2550 2551
int
virPCIDeviceAddressParse(char *address,
2552
                         virPCIDeviceAddressPtr bdf)
2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578
{
    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;

2579
 out:
2580 2581 2582
    return ret;
}

2583

2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607 2608
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);
}

2609
#ifdef __linux__
2610

2611
virPCIDeviceAddressPtr
2612
virPCIGetDeviceAddressFromSysfsLink(const char *device_link)
2613
{
2614
    virPCIDeviceAddressPtr bdf = NULL;
2615
    char *config_address = NULL;
2616
    VIR_AUTOFREE(char *) device_path = NULL;
2617 2618

    if (!virFileExists(device_link)) {
2619
        VIR_DEBUG("'%s' does not exist", device_link);
2620
        return NULL;
2621 2622
    }

2623
    device_path = virFileCanonicalizePath(device_link);
2624
    if (device_path == NULL) {
2625 2626 2627
        virReportSystemError(errno,
                             _("Failed to resolve device link '%s'"),
                             device_link);
2628
        return NULL;
2629 2630
    }

2631
    config_address = last_component(device_path);
2632
    if (VIR_ALLOC(bdf) < 0)
2633
        return NULL;
2634

2635
    if (virPCIDeviceAddressParse(config_address, bdf) < 0) {
2636
        virReportError(VIR_ERR_INTERNAL_ERROR,
2637 2638
                       _("Failed to parse PCI config address '%s'"),
                       config_address);
2639
        VIR_FREE(bdf);
2640
        return NULL;
2641 2642
    }

2643
    return bdf;
2644 2645
}

2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658
/**
 * 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
2659 2660
 */
int
2661
virPCIGetPhysicalFunction(const char *vf_sysfs_path,
2662
                          virPCIDeviceAddressPtr *pf)
2663
{
2664
    VIR_AUTOFREE(char *) device_link = NULL;
2665

2666 2667
    *pf = NULL;

2668 2669
    if (virBuildPath(&device_link, vf_sysfs_path, "physfn") == -1) {
        virReportOOMError();
2670
        return -1;
2671 2672
    }

2673
    if ((*pf = virPCIGetDeviceAddressFromSysfsLink(device_link))) {
2674 2675
        VIR_DEBUG("PF for VF device '%s': " VIR_PCI_DEVICE_ADDRESS_FMT,
                  vf_sysfs_path,
2676 2677
                  (*pf)->domain, (*pf)->bus, (*pf)->slot, (*pf)->function);
    }
2678

2679
    return 0;
2680 2681
}

2682

2683 2684 2685 2686
/*
 * Returns virtual functions of a physical function
 */
int
2687 2688
virPCIGetVirtualFunctions(const char *sysfs_path,
                          virPCIDeviceAddressPtr **virtual_functions,
2689 2690
                          size_t *num_virtual_functions,
                          unsigned int *max_virtual_functions)
2691 2692
{
    int ret = -1;
2693
    size_t i;
2694 2695
    VIR_AUTOFREE(char *) totalvfs_file = NULL;
    VIR_AUTOFREE(char *) totalvfs_str = NULL;
2696
    virPCIDeviceAddressPtr config_addr = NULL;
2697

2698 2699
    *virtual_functions = NULL;
    *num_virtual_functions = 0;
2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715
    *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;
        }
    }
2716

2717
    do {
2718
        VIR_AUTOFREE(char *) device_link = NULL;
2719 2720 2721
        /* 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;
2722

2723 2724
        if (!virFileExists(device_link))
            break;
2725

2726
        if (!(config_addr = virPCIGetDeviceAddressFromSysfsLink(device_link))) {
2727 2728 2729 2730 2731
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Failed to get SRIOV function from device link '%s'"),
                           device_link);
            goto error;
        }
2732

2733 2734
        if (VIR_APPEND_ELEMENT(*virtual_functions, *num_virtual_functions,
                               config_addr) < 0)
2735 2736
            goto error;
    } while (1);
2737

2738 2739
    VIR_DEBUG("Found %zu virtual functions for %s",
              *num_virtual_functions, sysfs_path);
2740
    ret = 0;
2741
 cleanup:
2742
    VIR_FREE(config_addr);
2743
    return ret;
2744

2745
 error:
2746 2747 2748
    for (i = 0; i < *num_virtual_functions; i++)
        VIR_FREE((*virtual_functions)[i]);
    VIR_FREE(*virtual_functions);
2749
    *num_virtual_functions = 0;
2750
    goto cleanup;
2751
}
2752

2753

2754 2755 2756 2757
/*
 * Returns 1 if vf device is a virtual function, 0 if not, -1 on error
 */
int
2758
virPCIIsVirtualFunction(const char *vf_sysfs_device_link)
2759
{
2760
    VIR_AUTOFREE(char *) vf_sysfs_physfn_link = NULL;
2761 2762

    if (virAsprintf(&vf_sysfs_physfn_link, "%s/physfn",
2763
                    vf_sysfs_device_link) < 0)
2764
        return -1;
2765

2766
    return virFileExists(vf_sysfs_physfn_link);
2767 2768 2769 2770 2771 2772
}

/*
 * Returns the sriov virtual function index of vf given its pf
 */
int
2773 2774 2775
virPCIGetVirtualFunctionIndex(const char *pf_sysfs_device_link,
                              const char *vf_sysfs_device_link,
                              int *vf_index)
2776
{
2777 2778
    int ret = -1;
    size_t i;
2779
    size_t num_virt_fns = 0;
2780
    unsigned int max_virt_fns = 0;
2781 2782
    virPCIDeviceAddressPtr vf_bdf = NULL;
    virPCIDeviceAddressPtr *virt_fns = NULL;
2783

2784
    if (!(vf_bdf = virPCIGetDeviceAddressFromSysfsLink(vf_sysfs_device_link)))
2785 2786
        return ret;

2787
    if (virPCIGetVirtualFunctions(pf_sysfs_device_link, &virt_fns,
2788
                                  &num_virt_fns, &max_virt_fns) < 0) {
2789
        virReportError(VIR_ERR_INTERNAL_ERROR,
2790
                       _("Error getting physical function's '%s' "
2791
                         "virtual_functions"), pf_sysfs_device_link);
2792 2793 2794 2795
        goto out;
    }

    for (i = 0; i < num_virt_fns; i++) {
2796
        if (virPCIDeviceAddressEqual(vf_bdf, virt_fns[i])) {
2797 2798 2799 2800
            *vf_index = i;
            ret = 0;
            break;
        }
2801 2802
    }

2803
 out:
2804 2805 2806

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

A
ajia@redhat.com 已提交
2809
    VIR_FREE(virt_fns);
2810 2811 2812 2813 2814
    VIR_FREE(vf_bdf);

    return ret;
}

2815 2816 2817 2818 2819
/*
 * Returns a path to the PCI sysfs file given the BDF of the PCI function
 */

int
2820
virPCIGetSysfsFile(char *virPCIDeviceName, char **pci_sysfs_device_link)
2821
{
2822 2823 2824 2825
    if (virAsprintf(pci_sysfs_device_link, PCI_SYSFS "devices/%s",
                    virPCIDeviceName) < 0)
        return -1;
    return 0;
2826 2827
}

R
Roopa Prabhu 已提交
2828
int
2829
virPCIDeviceAddressGetSysfsFile(virPCIDeviceAddressPtr addr,
2830
                                char **pci_sysfs_device_link)
R
Roopa Prabhu 已提交
2831
{
2832
    if (virAsprintf(pci_sysfs_device_link,
2833
                    PCI_SYSFS "devices/" VIR_PCI_DEVICE_ADDRESS_FMT,
2834 2835
                    addr->domain, addr->bus,
                    addr->slot, addr->function) < 0)
2836 2837
        return -1;
    return 0;
R
Roopa Prabhu 已提交
2838 2839
}

2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850
/**
 * 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)
2851 2852
 */
int
2853 2854 2855 2856
virPCIGetNetName(const char *device_link_sysfs_path,
                 size_t idx,
                 char *physPortID,
                 char **netname)
2857
{
2858 2859 2860
    VIR_AUTOFREE(char *) pcidev_sysfs_net_path = NULL;
    VIR_AUTOFREE(char *) firstEntryName = NULL;
    VIR_AUTOFREE(char *) thisPhysPortID = NULL;
2861 2862 2863
    int ret = -1;
    DIR *dir = NULL;
    struct dirent *entry = NULL;
2864
    size_t i = 0;
2865

2866 2867
    *netname = NULL;

2868 2869 2870 2871 2872 2873
    if (virBuildPath(&pcidev_sysfs_net_path, device_link_sysfs_path,
                     "net") == -1) {
        virReportOOMError();
        return -1;
    }

2874 2875 2876
    if (virDirOpenQuiet(&dir, pcidev_sysfs_net_path) < 0) {
        /* this *isn't* an error - caller needs to check for netname == NULL */
        ret = 0;
2877
        goto cleanup;
2878
    }
2879

E
Eric Blake 已提交
2880
    while (virDirRead(dir, &entry, pcidev_sysfs_net_path) > 0) {
2881 2882 2883 2884 2885 2886 2887 2888 2889 2890
        /* 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);
2891 2892 2893 2894 2895 2896 2897 2898 2899 2900
                /* 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;
                }

2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911
                continue;
            }
        } else {
            if (i++ < idx)
                continue;
        }

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

        ret = 0;
2912 2913 2914
        break;
    }

2915 2916
    if (ret < 0) {
        if (physPortID) {
2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931
            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);
            }
2932 2933 2934 2935 2936
        } else {
            ret = 0; /* no netdev at the given index is *not* an error */
        }
    }
 cleanup:
J
Ján Tomko 已提交
2937
    VIR_DIR_CLOSE(dir);
2938
    return ret;
2939
}
R
Roopa Prabhu 已提交
2940 2941

int
2942
virPCIGetVirtualFunctionInfo(const char *vf_sysfs_device_path,
2943 2944 2945
                             int pfNetDevIdx,
                             char **pfname,
                             int *vf_index)
R
Roopa Prabhu 已提交
2946
{
2947
    virPCIDeviceAddressPtr pf_config_address = NULL;
2948 2949 2950
    VIR_AUTOFREE(char *) pf_sysfs_device_path = NULL;
    VIR_AUTOFREE(char *) vfname = NULL;
    VIR_AUTOFREE(char *) vfPhysPortID = NULL;
R
Roopa Prabhu 已提交
2951 2952
    int ret = -1;

2953
    if (virPCIGetPhysicalFunction(vf_sysfs_device_path, &pf_config_address) < 0)
2954
        goto cleanup;
R
Roopa Prabhu 已提交
2955

2956
    if (!pf_config_address)
2957
        goto cleanup;
2958

2959 2960
    if (virPCIDeviceAddressGetSysfsFile(pf_config_address,
                                        &pf_sysfs_device_path) < 0) {
2961 2962
        goto cleanup;
    }
R
Roopa Prabhu 已提交
2963

2964 2965 2966
    if (virPCIGetVirtualFunctionIndex(pf_sysfs_device_path,
                                      vf_sysfs_device_path, vf_index) < 0) {
        goto cleanup;
R
Roopa Prabhu 已提交
2967 2968
    }

2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988
    /* 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 已提交
2989
        goto cleanup;
2990
    }
R
Roopa Prabhu 已提交
2991

2992 2993 2994 2995 2996 2997 2998 2999 3000
    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 已提交
3001

3002
    ret = 0;
3003
 cleanup:
R
Roopa Prabhu 已提交
3004 3005 3006 3007 3008
    VIR_FREE(pf_config_address);

    return ret;
}

3009 3010 3011 3012 3013 3014 3015 3016 3017

ssize_t
virPCIGetMdevTypes(const char *sysfspath,
                   virMediatedDeviceTypePtr **types)
{
    ssize_t ret = -1;
    int dirret = -1;
    DIR *dir = NULL;
    struct dirent *entry;
3018
    VIR_AUTOFREE(char *) types_path = NULL;
3019
    VIR_AUTOPTR(virMediatedDeviceType) mdev_type = NULL;
3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031 3032 3033 3034 3035
    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) {
3036
        VIR_AUTOFREE(char *) tmppath = NULL;
3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061
        /* 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;
}

3062
#else
3063 3064
static const char *unsupported = N_("not supported on non-linux platforms");

3065 3066 3067 3068
virPCIDeviceAddressPtr
virPCIGetDeviceAddressFromSysfsLink(const char *device_link ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3069
    return NULL;
3070 3071 3072
}


3073
int
3074
virPCIGetPhysicalFunction(const char *vf_sysfs_path ATTRIBUTE_UNUSED,
3075
                          virPCIDeviceAddressPtr *pf ATTRIBUTE_UNUSED)
3076
{
3077
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3078 3079 3080 3081
    return -1;
}

int
3082 3083
virPCIGetVirtualFunctions(const char *sysfs_path ATTRIBUTE_UNUSED,
                          virPCIDeviceAddressPtr **virtual_functions ATTRIBUTE_UNUSED,
3084 3085
                          size_t *num_virtual_functions ATTRIBUTE_UNUSED,
                          unsigned int *max_virtual_functions ATTRIBUTE_UNUSED)
3086
{
3087
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3088 3089
    return -1;
}
3090 3091

int
E
Eric Blake 已提交
3092
virPCIIsVirtualFunction(const char *vf_sysfs_device_link ATTRIBUTE_UNUSED)
3093
{
3094
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3095 3096 3097 3098
    return -1;
}

int
3099 3100 3101
virPCIGetVirtualFunctionIndex(const char *pf_sysfs_device_link ATTRIBUTE_UNUSED,
                              const char *vf_sysfs_device_link ATTRIBUTE_UNUSED,
                              int *vf_index ATTRIBUTE_UNUSED)
3102
{
3103
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3104 3105 3106 3107
    return -1;

}

3108 3109 3110 3111 3112 3113 3114 3115
int
virPCIGetSysfsFile(char *virPCIDeviceName ATTRIBUTE_UNUSED,
                   char **pci_sysfs_device_link ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
    return -1;
}

3116
int
3117 3118
virPCIDeviceAddressGetSysfsFile(virPCIDeviceAddressPtr dev ATTRIBUTE_UNUSED,
                                char **pci_sysfs_device_link ATTRIBUTE_UNUSED)
3119
{
3120
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3121 3122 3123
    return -1;
}

3124
int
3125
virPCIGetNetName(const char *device_link_sysfs_path ATTRIBUTE_UNUSED,
3126 3127
                 size_t idx ATTRIBUTE_UNUSED,
                 char *physPortID ATTRIBUTE_UNUSED,
3128
                 char **netname ATTRIBUTE_UNUSED)
3129
{
3130
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3131 3132
    return -1;
}
R
Roopa Prabhu 已提交
3133 3134

int
3135
virPCIGetVirtualFunctionInfo(const char *vf_sysfs_device_path ATTRIBUTE_UNUSED,
3136
                             int pfNetDevIdx ATTRIBUTE_UNUSED,
3137 3138
                             char **pfname ATTRIBUTE_UNUSED,
                             int *vf_index ATTRIBUTE_UNUSED)
R
Roopa Prabhu 已提交
3139
{
3140
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
R
Roopa Prabhu 已提交
3141 3142
    return -1;
}
3143 3144 3145 3146 3147 3148 3149 3150 3151


ssize_t
virPCIGetMdevTypes(const char *sysfspath ATTRIBUTE_UNUSED,
                   virMediatedDeviceTypePtr **types ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
    return -1;
}
3152
#endif /* __linux__ */
3153 3154 3155 3156 3157 3158 3159

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

3160
    if ((fd = virPCIDeviceConfigOpen(dev)) < 0)
3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172 3173 3174 3175 3176 3177 3178 3179
        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;

3180
    if ((fd = virPCIDeviceConfigOpen(dev)) < 0)
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
        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;

3208
    if ((fd = virPCIDeviceConfigOpen(dev)) < 0)
3209 3210 3211 3212 3213 3214 3215 3216 3217 3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236
        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;
}
3237 3238


3239 3240 3241 3242 3243 3244 3245
int virPCIGetHeaderType(virPCIDevicePtr dev, int *hdrType)
{
    int fd;
    uint8_t type;

    *hdrType = -1;

3246
    if ((fd = virPCIDeviceConfigOpen(dev)) < 0)
3247 3248 3249 3250 3251 3252 3253 3254 3255
        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,
3256 3257
                       _("Unknown PCI header type '%d' for device '%s'"),
                       type, dev->name);
3258 3259 3260 3261 3262 3263 3264 3265 3266
        return -1;
    }

    *hdrType = type;

    return 0;
}


3267 3268 3269 3270 3271 3272 3273 3274 3275 3276
void
virPCIEDeviceInfoFree(virPCIEDeviceInfoPtr dev)
{
    if (!dev)
        return;

    VIR_FREE(dev->link_cap);
    VIR_FREE(dev->link_sta);
    VIR_FREE(dev);
}
3277 3278 3279 3280 3281 3282

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