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 48
#define PCI_SYSFS "/sys/bus/pci/"
#define PCI_ID_LEN 10   /* "XXXX XXXX" */
#define PCI_ADDR_LEN 13 /* "XXXX:XX:XX.X" */

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

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

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

69
struct _virPCIDevice {
70
    virPCIDeviceAddress address;
71 72 73

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

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

80 81
    unsigned int  pcie_cap_pos;
    unsigned int  pci_pm_cap_pos;
82 83
    bool          has_flr;
    bool          has_pm_reset;
84
    bool          managed;
85 86

    virPCIStubDriver stubDriver;
87 88

    /* used by reattach function */
89 90 91
    bool          unbind_from_stub;
    bool          remove_slot;
    bool          reprobe;
92 93
};

94
struct _virPCIDeviceList {
95 96
    virObjectLockable parent;

97
    size_t count;
98
    virPCIDevicePtr *devs;
99 100 101
};


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

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

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

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

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

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

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

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

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

198 199 200 201 202 203
static virClassPtr virPCIDeviceListClass;

static void virPCIDeviceListDispose(void *obj);

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

    return 0;
}

210
VIR_ONCE_GLOBAL_INIT(virPCI);
211

L
Laine Stump 已提交
212

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

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


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

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


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

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


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

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

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

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


294
static int
295
virPCIDeviceConfigOpen(virPCIDevicePtr dev, bool fatal)
296 297 298 299
{
    int fd;

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

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

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

318
static void
319
virPCIDeviceConfigClose(virPCIDevicePtr dev, int cfgfd)
320
{
321 322 323 324 325
    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)));
    }
326 327
}

328

329
static int
330 331
virPCIDeviceRead(virPCIDevicePtr dev,
                 int cfgfd,
332
                 unsigned int pos,
333
                 uint8_t *buf,
334
                 unsigned int buflen)
335 336 337
{
    memset(buf, 0, buflen);

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

static uint8_t
349
virPCIDeviceRead8(virPCIDevicePtr dev, int cfgfd, unsigned int pos)
350 351
{
    uint8_t buf;
352
    virPCIDeviceRead(dev, cfgfd, pos, &buf, sizeof(buf));
353 354 355 356
    return buf;
}

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

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

372 373 374
static int
virPCIDeviceReadClass(virPCIDevicePtr dev, uint16_t *device_class)
{
375 376
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) id_str = NULL;
377 378
    unsigned int value;

379
    if (!(path = virPCIFile(dev->name, "class")))
380
        return -1;
381 382 383

    /* class string is '0xNNNNNN\n' ... i.e. 9 bytes */
    if (virFileReadAll(path, 9, &id_str) < 0)
384
        return -1;
385 386 387 388 389 390

    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);
391
        return -1;
392 393 394
    }

    *device_class = (value >> 8) & 0xFFFF;
395
    return 0;
396 397
}

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

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

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

E
Eric Blake 已提交
429 430
typedef int (*virPCIDeviceIterPredicate)(virPCIDevicePtr, virPCIDevicePtr,
                                         void *);
431 432 433 434 435 436 437

/* 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
438 439 440 441
virPCIDeviceIterDevices(virPCIDeviceIterPredicate predicate,
                        virPCIDevicePtr dev,
                        virPCIDevicePtr *matched,
                        void *data)
442 443 444
{
    DIR *dir;
    struct dirent *entry;
445
    int ret = 0;
446
    int rc;
447 448 449 450 451

    *matched = NULL;

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

J
Ján Tomko 已提交
452
    if (virDirOpen(&dir, PCI_SYSFS "devices") < 0)
453 454
        return -1;

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

460 461 462 463 464 465 466 467 468
        /* 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) {
469 470 471 472
            VIR_WARN("Unusual entry in " PCI_SYSFS "devices: %s", entry->d_name);
            continue;
        }

473
        check = virPCIDeviceNew(domain, bus, slot, function);
474
        if (!check) {
475 476 477
            ret = -1;
            break;
        }
478

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

static uint8_t
496 497 498
virPCIDeviceFindCapabilityOffset(virPCIDevicePtr dev,
                                 int cfgfd,
                                 unsigned int capability)
499 500 501 502
{
    uint16_t status;
    uint8_t pos;

503
    status = virPCIDeviceRead16(dev, cfgfd, PCI_STATUS);
504 505 506
    if (!(status & PCI_STATUS_CAP_LIST))
        return 0;

507
    pos = virPCIDeviceRead8(dev, cfgfd, PCI_CAPABILITY_LIST);
508 509 510 511 512 513 514 515 516

    /* 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) {
517
        uint8_t capid = virPCIDeviceRead8(dev, cfgfd, pos);
518 519 520 521 522 523
        if (capid == capability) {
            VIR_DEBUG("%s %s: found cap 0x%.2x at 0x%.2x",
                      dev->id, dev->name, capability, pos);
            return pos;
        }

524
        pos = virPCIDeviceRead8(dev, cfgfd, pos + 1);
525 526 527 528 529 530 531
    }

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

    return 0;
}

J
Jiri Denemark 已提交
532
static unsigned int
533 534
virPCIDeviceFindExtendedCapabilityOffset(virPCIDevicePtr dev,
                                         int cfgfd,
535
                                         unsigned int capability)
J
Jiri Denemark 已提交
536 537 538 539 540 541 542 543 544 545
{
    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) {
546
        header = virPCIDeviceRead32(dev, cfgfd, pos);
J
Jiri Denemark 已提交
547 548 549 550 551 552 553 554 555 556 557

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

558 559 560 561
/* 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
562
virPCIDeviceDetectFunctionLevelReset(virPCIDevicePtr dev, int cfgfd)
563
{
M
Mark McLoughlin 已提交
564
    uint32_t caps;
565
    uint8_t pos;
566
    VIR_AUTOFREE(char *) path = NULL;
567
    int found;
568 569 570 571 572 573 574 575

    /* 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) {
576
        caps = virPCIDeviceRead32(dev, cfgfd, dev->pcie_cap_pos + PCI_EXP_DEVCAP);
577 578 579 580 581 582 583 584 585 586
        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.
     */
587
    pos = virPCIDeviceFindCapabilityOffset(dev, cfgfd, PCI_CAP_ID_AF);
588
    if (pos) {
589
        caps = virPCIDeviceRead16(dev, cfgfd, pos + PCI_AF_CAP);
590 591 592 593 594 595
        if (caps & PCI_AF_CAP_FLR) {
            VIR_DEBUG("%s %s: detected PCI FLR capability", dev->id, dev->name);
            return 1;
        }
    }

596 597 598 599 600 601
    /* 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
     */

602
    if (virAsprintf(&path, PCI_SYSFS "devices/%s/physfn", dev->name) < 0)
603 604 605 606 607 608 609 610 611
        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;
    }

612 613 614 615 616 617 618 619 620
    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.
 */
621
static unsigned int
622
virPCIDeviceDetectPowerManagementReset(virPCIDevicePtr dev, int cfgfd)
623 624 625 626 627
{
    if (dev->pci_pm_cap_pos) {
        uint32_t ctl;

        /* require the NO_SOFT_RESET bit is clear */
628
        ctl = virPCIDeviceRead32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL);
629 630 631 632 633 634 635 636 637 638 639
        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;
}

640
/* Any active devices on the same domain/bus ? */
641
static int
642
virPCIDeviceSharesBusWithActive(virPCIDevicePtr dev, virPCIDevicePtr check, void *data)
643
{
644
    virPCIDeviceList *inactiveDevs = data;
645

646
    /* Different domain, different bus, or simply identical device */
647 648 649 650
    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))
651 652
        return 0;

653
    /* same bus, but inactive, i.e. about to be assigned to guest */
654
    if (inactiveDevs && virPCIDeviceListFind(inactiveDevs, check))
655
        return 0;
656

657
    return 1;
658 659
}

660 661 662
static virPCIDevicePtr
virPCIDeviceBusContainsActiveDevices(virPCIDevicePtr dev,
                                     virPCIDeviceList *inactiveDevs)
663
{
664 665 666
    virPCIDevicePtr active = NULL;
    if (virPCIDeviceIterDevices(virPCIDeviceSharesBusWithActive,
                                dev, &active, inactiveDevs) < 0)
667 668 669 670 671
        return NULL;
    return active;
}

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

681
    if (dev->address.domain != check->address.domain)
682 683
        return 0;

684
    if ((fd = virPCIDeviceConfigOpen(check, false)) < 0)
685 686
        return 0;

687
    /* Is it a bridge? */
688 689
    ret = virPCIDeviceReadClass(check, &device_class);
    if (ret < 0 || device_class != PCI_CLASS_BRIDGE_PCI)
690
        goto cleanup;
691 692

    /* Is it a plane? */
693
    header_type = virPCIDeviceRead8(check, fd, PCI_HEADER_TYPE);
694
    if ((header_type & PCI_HEADER_TYPE_MASK) != PCI_HEADER_TYPE_BRIDGE)
695
        goto cleanup;
696

697 698
    secondary   = virPCIDeviceRead8(check, fd, PCI_SECONDARY_BUS);
    subordinate = virPCIDeviceRead8(check, fd, PCI_SUBORDINATE_BUS);
699

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

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

710
    /* otherwise, SRIOV allows VFs to be on different buses than their PFs.
711 712 713
     * 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.
     */
714
    if (dev->address.bus > secondary && dev->address.bus <= subordinate) {
715
        if (*best == NULL) {
716 717 718 719
            *best = virPCIDeviceNew(check->address.domain,
                                    check->address.bus,
                                    check->address.slot,
                                    check->address.function);
720 721 722 723 724
            if (*best == NULL) {
                ret = -1;
                goto cleanup;
            }
        } else {
725 726 727 728
            /* 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
             */
729 730 731
            int bestfd;
            uint8_t best_secondary;

732
            if ((bestfd = virPCIDeviceConfigOpen(*best, false)) < 0)
733
                goto cleanup;
734 735
            best_secondary = virPCIDeviceRead8(*best, bestfd, PCI_SECONDARY_BUS);
            virPCIDeviceConfigClose(*best, bestfd);
736 737

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

751
 cleanup:
752
    virPCIDeviceConfigClose(check, fd);
753
    return ret;
754 755
}

756
static int
757
virPCIDeviceGetParent(virPCIDevicePtr dev, virPCIDevicePtr *parent)
758
{
759
    virPCIDevicePtr best = NULL;
760 761 762
    int ret;

    *parent = NULL;
763
    ret = virPCIDeviceIterDevices(virPCIDeviceIsParent, dev, parent, &best);
764
    if (ret == 1)
765
        virPCIDeviceFree(best);
766 767 768
    else if (ret == 0)
        *parent = best;
    return ret;
769 770 771 772 773 774
}

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

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

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

    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
     */
815
    if (virPCIDeviceRead(dev, cfgfd, 0, config_space, PCI_CONF_LEN) < 0) {
816
        virReportError(VIR_ERR_INTERNAL_ERROR,
817
                       _("Failed to read PCI config space for %s"),
818
                       dev->name);
819 820 821 822 823 824
        goto out;
    }

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

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

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

832
    virPCIDeviceWrite16(parent, parentfd, PCI_BRIDGE_CONTROL, ctl);
833 834 835

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

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

844
 out:
845
    virPCIDeviceConfigClose(parent, parentfd);
846 847 848 849 850 851 852 853
    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
854
virPCIDeviceTryPowerManagementReset(virPCIDevicePtr dev, int cfgfd)
855 856 857 858 859 860 861 862
{
    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. */
863
    if (virPCIDeviceRead(dev, cfgfd, 0, &config_space[0], PCI_CONF_LEN) < 0) {
864
        virReportError(VIR_ERR_INTERNAL_ERROR,
865
                       _("Failed to read PCI config space for %s"),
866
                       dev->name);
867 868 869 870 871
        return -1;
    }

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

872
    ctl = virPCIDeviceRead32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL);
873 874
    ctl &= ~PCI_PM_CTRL_STATE_MASK;

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

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

880 881
    virPCIDeviceWrite32(dev, cfgfd, dev->pci_pm_cap_pos + PCI_PM_CTRL,
                        ctl | PCI_PM_CTRL_STATE_D0);
882 883 884

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

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

    return 0;
}

static int
896
virPCIDeviceInit(virPCIDevicePtr dev, int cfgfd)
897
{
898 899
    int flr;

900 901 902
    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);
903
    if (flr < 0)
904
        return flr;
905 906
    dev->has_flr        = !!flr;
    dev->has_pm_reset   = !!virPCIDeviceDetectPowerManagementReset(dev, cfgfd);
907

908 909 910 911
    return 0;
}

int
912 913 914
virPCIDeviceReset(virPCIDevicePtr dev,
                  virPCIDeviceList *activeDevs,
                  virPCIDeviceList *inactiveDevs)
915
{
916 917
    VIR_AUTOFREE(char *) drvPath = NULL;
    VIR_AUTOFREE(char *) drvName = NULL;
918
    int ret = -1;
919
    int fd = -1;
920 921 922 923 924 925 926 927 928 929 930 931
    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;
    }
932

933
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
934
        virReportError(VIR_ERR_INTERNAL_ERROR,
935 936 937 938
                       _("Not resetting active device %s"), dev->name);
        return -1;
    }

939 940 941 942 943 944 945 946
    /* 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;

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

955
    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
956
        goto cleanup;
957

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

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

969 970 971 972 973
    /* 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)
974
        ret = virPCIDeviceTryPowerManagementReset(dev, fd);
975

976
    /* Bus reset is not an option with the root bus */
977
    if (ret < 0 && dev->address.bus != 0)
978
        ret = virPCIDeviceTrySecondaryBusReset(dev, fd, inactiveDevs);
979

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

989
 cleanup:
990
    virPCIDeviceConfigClose(dev, fd);
991 992 993
    return ret;
}

994

995
static int
996
virPCIProbeStubDriver(virPCIStubDriver driver)
997
{
998
    const char *drvname = NULL;
999
    VIR_AUTOFREE(char *) drvpath = NULL;
1000
    bool probed = false;
1001

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

1010
 recheck:
1011
    if ((drvpath = virPCIDriverDir(drvname)) && virFileExists(drvpath))
1012 1013
        /* driver already loaded, return */
        return 0;
1014 1015

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

        goto recheck;
1024 1025
    }

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

1041
    return -1;
1042 1043
}

1044
int
1045
virPCIDeviceUnbind(virPCIDevicePtr dev)
1046
{
1047 1048 1049
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) drvpath = NULL;
    VIR_AUTOFREE(char *) driver = NULL;
1050 1051

    if (virPCIDeviceGetDriverPathAndName(dev, &drvpath, &driver) < 0)
1052
        return -1;
1053

1054
    if (!driver)
1055
        /* The device is not bound to any driver */
1056
        return 0;
1057

1058
    if (!(path = virPCIFile(dev->name, "driver/unbind")))
1059
        return -1;
1060 1061 1062 1063 1064 1065

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

1070
    return 0;
1071 1072
}

1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097

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


1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
/*
 * 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.
 */
1109
static int
1110 1111 1112
virPCIDeviceBindWithDriverOverride(virPCIDevicePtr dev,
                                   const char *driverName)
{
1113
    VIR_AUTOFREE(char *) path = NULL;
1114 1115 1116 1117 1118 1119 1120 1121 1122

    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);
1123
        return -1;
1124 1125
    }

1126
    if (virPCIDeviceRebind(dev) < 0)
1127
        return -1;
1128

1129
    return 0;
1130 1131 1132 1133
}

static int
virPCIDeviceUnbindFromStubWithNewid(virPCIDevicePtr dev)
1134 1135
{
    int result = -1;
1136 1137 1138
    VIR_AUTOFREE(char *) drvdir = NULL;
    VIR_AUTOFREE(char *) path = NULL;
    VIR_AUTOFREE(char *) driver = NULL;
1139

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

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

1152 1153
    if (!dev->unbind_from_stub) {
        VIR_DEBUG("Unbind from stub skipped for PCI device %s", dev->name);
1154
        goto remove_slot;
1155
    }
1156

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

1165 1166
    VIR_DEBUG("Unbinding PCI device %s from stub driver %s",
              dev->name, driver);
1167

1168
    if (virPCIDeviceUnbind(dev) < 0)
1169
        goto cleanup;
1170
    dev->unbind_from_stub = false;
1171

1172
 remove_slot:
1173 1174
    if (!dev->remove_slot) {
        VIR_DEBUG("Slot removal skipped for PCI device %s", dev->name);
1175
        goto reprobe;
1176 1177 1178
    }

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

    /* Xen's pciback.ko wants you to use remove_slot on the specific device */
1181
    if (!(path = virPCIDriverFile(driver, "remove_slot")))
1182 1183 1184 1185
        goto cleanup;

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

1192
 reprobe:
1193
    if (!dev->reprobe) {
1194
        VIR_DEBUG("Reprobe skipped for PCI device %s", dev->name);
1195 1196 1197
        result = 0;
        goto cleanup;
    }
1198

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

1201 1202 1203 1204 1205
    /* 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.
     */
1206 1207
    VIR_FREE(path);
    if (driver && !(path = virPCIDriverFile(driver, "remove_id")))
1208 1209
        goto cleanup;

1210
    if (!driver || !virFileExists(drvdir) || virFileExists(path)) {
1211 1212 1213 1214 1215 1216 1217 1218 1219 1220
        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;

1221
 cleanup:
1222
    /* do not do it again */
1223 1224 1225
    dev->unbind_from_stub = false;
    dev->remove_slot = false;
    dev->reprobe = false;
1226

1227 1228 1229
    return result;
}

1230 1231 1232 1233 1234 1235 1236 1237 1238 1239
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");
}
1240 1241

static int
1242 1243
virPCIDeviceUnbindFromStub(virPCIDevicePtr dev)
{
1244
    VIR_AUTOFREE(char *) path = NULL;
1245 1246 1247 1248 1249 1250 1251 1252 1253

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

1256
    return virPCIDeviceUnbindFromStubWithNewid(dev);
1257 1258 1259 1260
}

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

1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
    /* 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;
    }

1283
    if (!(stubDriverPath = virPCIDriverDir(stubDriverName))  ||
1284
        !(driverLink = virPCIFile(dev->name, "driver")))
1285 1286
        goto cleanup;

1287 1288 1289 1290 1291
    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);
1292 1293 1294
            result = 0;
            goto cleanup;
        }
1295
        reprobe = true;
1296
    }
1297 1298 1299 1300 1301 1302 1303 1304 1305

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

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

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

1326
    if (virPCIDeviceUnbind(dev) < 0)
J
Jiri Denemark 已提交
1327
        goto remove_id;
1328

1329 1330 1331
    /* If the device was bound to a driver we'll need to reprobe later */
    dev->reprobe = reprobe;

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

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

1349 1350
        VIR_FREE(path);
        if (!(path = virPCIDriverFile(stubDriverName, "bind")))
1351
            goto remove_id;
1352

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

J
Jiri Denemark 已提交
1362 1363
    result = 0;

1364
 remove_id:
J
Jiri Denemark 已提交
1365 1366
    err = virSaveLastError();

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

1382
    if (virFileExists(path) && virFileWriteStr(path, dev->id, 0) < 0) {
1383
        virReportSystemError(errno,
1384
                             _("Failed to remove PCI ID '%s' from %s"),
1385
                             dev->id, stubDriverName);
1386

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

1397
 cleanup:
1398
    if (result < 0)
J
Jiri Denemark 已提交
1399 1400 1401 1402
        virPCIDeviceUnbindFromStub(dev);

    if (err)
        virSetError(err);
1403

1404
    return result;
1405 1406
}

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

    /* 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")))
1429
        return -1;
1430 1431 1432 1433 1434 1435

    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);
1436
            return 0;
1437 1438 1439 1440
        }
    }

    if (virPCIDeviceBindWithDriverOverride(dev, stubDriverName) < 0)
1441
        return -1;
1442 1443

    dev->unbind_from_stub = true;
1444
    return 0;
1445 1446 1447 1448 1449
}

static int
virPCIDeviceBindToStub(virPCIDevicePtr dev)
{
1450
    VIR_AUTOFREE(char *) path = NULL;
1451 1452 1453 1454 1455 1456 1457 1458 1459

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

1462
    return virPCIDeviceBindToStubWithNewid(dev);
1463 1464
}

1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482
/* 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.
 */
1483
int
1484 1485
virPCIDeviceDetach(virPCIDevicePtr dev,
                   virPCIDeviceList *activeDevs,
1486
                   virPCIDeviceList *inactiveDevs)
1487
{
1488
    if (virPCIProbeStubDriver(dev->stubDriver) < 0)
1489 1490
        return -1;

1491
    if (activeDevs && virPCIDeviceListFind(activeDevs, dev)) {
1492
        virReportError(VIR_ERR_INTERNAL_ERROR,
1493 1494 1495 1496
                       _("Not detaching active device %s"), dev->name);
        return -1;
    }

1497
    if (virPCIDeviceBindToStub(dev) < 0)
1498 1499
        return -1;

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

    return 0;
1510 1511 1512
}

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

1523
    if (virPCIDeviceUnbindFromStub(dev) < 0)
1524 1525 1526
        return -1;

    /* Steal the dev from list inactiveDevs */
1527 1528
    if (inactiveDevs) {
        VIR_DEBUG("Removing PCI device %s from inactive list", dev->name);
1529
        virPCIDeviceListDel(inactiveDevs, dev);
1530
    }
1531 1532

    return 0;
1533 1534
}

1535 1536 1537 1538 1539
/* 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 已提交
1540 1541
 * 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.
1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561
 *
 * 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 已提交
1562
 * holding on to the resource.
1563 1564
 */
int
1565
virPCIDeviceWaitForCleanup(virPCIDevicePtr dev, const char *matcher)
1566 1567 1568
{
    FILE *fp;
    char line[160];
1569
    char *tmp;
1570
    unsigned long long start, end;
1571
    unsigned int domain, bus, slot, function;
1572
    bool in_matching_device;
1573 1574 1575 1576 1577 1578 1579 1580 1581
    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
         */
1582
        VIR_DEBUG("Failed to open /proc/iomem, trying to continue anyway");
1583 1584 1585 1586
        return 0;
    }

    ret = 0;
1587
    in_matching_device = false;
1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
    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))) {
1599 1600 1601 1602 1603 1604
            /* 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)
1605 1606
                continue;

1607
            if (STRPREFIX(tmp, matcher)) {
1608 1609 1610
                ret = 1;
                break;
            }
1611
        } else {
1612
            in_matching_device = false;
1613

1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627
            /* 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')
1628 1629
                continue;

1630 1631
            if (domain != dev->address.domain || bus != dev->address.bus ||
                slot != dev->address.slot || function != dev->address.function)
1632
                continue;
1633
            in_matching_device = true;
1634 1635 1636 1637
            match_depth = strspn(line, " ");
        }
    }

E
Eric Blake 已提交
1638
    VIR_FORCE_FCLOSE(fp);
1639 1640 1641 1642

    return ret;
}

1643
static char *
1644
virPCIDeviceReadID(virPCIDevicePtr dev, const char *id_name)
1645
{
1646
    VIR_AUTOFREE(char *) path = NULL;
1647 1648
    char *id_str;

1649
    if (!(path = virPCIFile(dev->name, id_name)))
1650
        return NULL;
1651 1652

    /* ID string is '0xNNNN\n' ... i.e. 7 bytes */
1653
    if (virFileReadAll(path, 7, &id_str) < 0)
1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667
        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;
}

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

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

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

1733
char *
1734
virPCIDeviceAddressAsString(virPCIDeviceAddressPtr addr)
1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745
{
    char *str;

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

1746
virPCIDevicePtr
1747 1748 1749 1750
virPCIDeviceNew(unsigned int domain,
                unsigned int bus,
                unsigned int slot,
                unsigned int function)
1751
{
1752 1753
    virPCIDevicePtr ret = NULL;
    VIR_AUTOPTR(virPCIDevice) dev = NULL;
1754 1755
    VIR_AUTOFREE(char *) vendor = NULL;
    VIR_AUTOFREE(char *) product = NULL;
1756

1757
    if (VIR_ALLOC(dev) < 0)
1758 1759
        return NULL;

1760 1761 1762 1763
    dev->address.domain = domain;
    dev->address.bus = bus;
    dev->address.slot = slot;
    dev->address.function = function;
1764

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

1776
    if (!virFileExists(dev->path)) {
1777 1778 1779
        virReportSystemError(errno,
                             _("Device %s not found: could not access %s"),
                             dev->name, dev->path);
1780
        goto cleanup;
1781 1782
    }

1783 1784
    vendor  = virPCIDeviceReadID(dev, "vendor");
    product = virPCIDeviceReadID(dev, "device");
1785 1786

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

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

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

1804
    VIR_STEAL_PTR(ret, dev);
E
Eric Blake 已提交
1805

1806 1807
 cleanup:
    return ret;
1808 1809
}

L
Laine Stump 已提交
1810 1811 1812 1813 1814 1815

virPCIDevicePtr
virPCIDeviceCopy(virPCIDevicePtr dev)
{
    virPCIDevicePtr copy;

1816
    if (VIR_ALLOC(copy) < 0)
L
Laine Stump 已提交
1817 1818 1819 1820
        return NULL;

    /* shallow copy to take care of most attributes */
    *copy = *dev;
1821
    copy->path = NULL;
C
Chunyan Liu 已提交
1822
    copy->used_by_drvname = copy->used_by_domname = NULL;
L
Laine Stump 已提交
1823
    if (VIR_STRDUP(copy->path, dev->path) < 0 ||
C
Chunyan Liu 已提交
1824 1825
        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 已提交
1826 1827 1828 1829
        goto error;
    }
    return copy;

1830
 error:
L
Laine Stump 已提交
1831 1832 1833 1834 1835
    virPCIDeviceFree(copy);
    return NULL;
}


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

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

1863
const char *
1864
virPCIDeviceGetName(virPCIDevicePtr dev)
1865 1866 1867 1868
{
    return dev->name;
}

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

1881
void virPCIDeviceSetManaged(virPCIDevicePtr dev, bool managed)
1882
{
1883
    dev->managed = managed;
1884 1885
}

1886
bool
1887
virPCIDeviceGetManaged(virPCIDevicePtr dev)
1888 1889 1890 1891
{
    return dev->managed;
}

1892 1893
void
virPCIDeviceSetStubDriver(virPCIDevicePtr dev, virPCIStubDriver driver)
1894
{
1895
    dev->stubDriver = driver;
1896 1897
}

1898
virPCIStubDriver
1899 1900 1901 1902 1903
virPCIDeviceGetStubDriver(virPCIDevicePtr dev)
{
    return dev->stubDriver;
}

1904
bool
1905
virPCIDeviceGetUnbindFromStub(virPCIDevicePtr dev)
1906 1907 1908 1909 1910
{
    return dev->unbind_from_stub;
}

void
1911
virPCIDeviceSetUnbindFromStub(virPCIDevicePtr dev, bool unbind)
1912
{
1913
    dev->unbind_from_stub = unbind;
1914 1915
}

1916
bool
1917
virPCIDeviceGetRemoveSlot(virPCIDevicePtr dev)
1918 1919 1920 1921 1922
{
    return dev->remove_slot;
}

void
1923
virPCIDeviceSetRemoveSlot(virPCIDevicePtr dev, bool remove_slot)
1924
{
1925
    dev->remove_slot = remove_slot;
1926 1927
}

1928
bool
1929
virPCIDeviceGetReprobe(virPCIDevicePtr dev)
1930 1931 1932 1933 1934
{
    return dev->reprobe;
}

void
1935
virPCIDeviceSetReprobe(virPCIDevicePtr dev, bool reprobe)
1936
{
1937
    dev->reprobe = reprobe;
1938 1939
}

C
Chunyan Liu 已提交
1940 1941 1942 1943
int
virPCIDeviceSetUsedBy(virPCIDevicePtr dev,
                      const char *drv_name,
                      const char *dom_name)
1944
{
C
Chunyan Liu 已提交
1945 1946 1947 1948 1949 1950 1951 1952
    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;
1953 1954
}

C
Chunyan Liu 已提交
1955 1956 1957 1958
void
virPCIDeviceGetUsedBy(virPCIDevicePtr dev,
                      const char **drv_name,
                      const char **dom_name)
1959
{
C
Chunyan Liu 已提交
1960 1961
    *drv_name = dev->used_by_drvname;
    *dom_name = dev->used_by_domname;
1962 1963
}

1964 1965
virPCIDeviceListPtr
virPCIDeviceListNew(void)
1966
{
1967
    virPCIDeviceListPtr list;
1968

1969 1970 1971 1972
    if (virPCIInitialize() < 0)
        return NULL;

    if (!(list = virObjectLockableNew(virPCIDeviceListClass)))
1973 1974 1975 1976 1977
        return NULL;

    return list;
}

1978 1979
static void
virPCIDeviceListDispose(void *obj)
1980
{
1981
    virPCIDeviceListPtr list = obj;
1982
    size_t i;
1983 1984

    for (i = 0; i < list->count; i++) {
1985
        virPCIDeviceFree(list->devs[i]);
1986 1987 1988 1989 1990 1991 1992 1993
        list->devs[i] = NULL;
    }

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

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

L
Laine Stump 已提交
2005 2006 2007 2008 2009

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

    if (!copy)
        return -1;
2014
    if (virPCIDeviceListAdd(list, copy) < 0)
L
Laine Stump 已提交
2015
        return -1;
2016 2017

    copy = NULL;
L
Laine Stump 已提交
2018 2019 2020 2021
    return 0;
}


2022 2023 2024
virPCIDevicePtr
virPCIDeviceListGet(virPCIDeviceListPtr list,
                    int idx)
2025 2026 2027 2028 2029 2030 2031 2032 2033
{
    if (idx >= list->count)
        return NULL;
    if (idx < 0)
        return NULL;

    return list->devs[idx];
}

2034
size_t
2035
virPCIDeviceListCount(virPCIDeviceListPtr list)
2036
{
2037 2038 2039
    return list->count;
}

2040 2041 2042
virPCIDevicePtr
virPCIDeviceListStealIndex(virPCIDeviceListPtr list,
                           int idx)
2043
{
2044
    virPCIDevicePtr ret;
2045

2046 2047
    if (idx < 0 || idx >= list->count)
        return NULL;
2048

2049
    ret = list->devs[idx];
2050
    VIR_DELETE_ELEMENT(list->devs, idx, list->count);
2051 2052 2053
    return ret;
}

2054 2055 2056
virPCIDevicePtr
virPCIDeviceListSteal(virPCIDeviceListPtr list,
                      virPCIDevicePtr dev)
2057
{
2058
    return virPCIDeviceListStealIndex(list, virPCIDeviceListFindIndex(list, dev));
2059 2060
}

2061
void
2062 2063
virPCIDeviceListDel(virPCIDeviceListPtr list,
                    virPCIDevicePtr dev)
2064
{
2065
    virPCIDeviceFree(virPCIDeviceListSteal(list, dev));
2066 2067
}

2068
int
2069
virPCIDeviceListFindIndex(virPCIDeviceListPtr list, virPCIDevicePtr dev)
2070
{
2071
    size_t i;
2072

2073 2074 2075 2076 2077 2078
    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)
2079
            return i;
2080
    }
2081 2082 2083
    return -1;
}

L
Laine Stump 已提交
2084 2085 2086 2087 2088 2089 2090 2091

virPCIDevicePtr
virPCIDeviceListFindByIDs(virPCIDeviceListPtr list,
                          unsigned int domain,
                          unsigned int bus,
                          unsigned int slot,
                          unsigned int function)
{
2092
    size_t i;
L
Laine Stump 已提交
2093 2094

    for (i = 0; i < list->count; i++) {
2095 2096 2097 2098 2099
        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 已提交
2100 2101 2102 2103 2104 2105
            return list->devs[i];
    }
    return NULL;
}


2106 2107
virPCIDevicePtr
virPCIDeviceListFind(virPCIDeviceListPtr list, virPCIDevicePtr dev)
2108
{
2109
    int idx;
2110

2111 2112
    if ((idx = virPCIDeviceListFindIndex(list, dev)) >= 0)
        return list->devs[idx];
2113 2114
    else
        return NULL;
2115
}
2116 2117


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

    if (virAsprintf(&pcidir, "/sys/bus/pci/devices/%04x:%02x:%02x.%x",
2129 2130
                    dev->address.domain, dev->address.bus,
                    dev->address.slot, dev->address.function) < 0)
2131 2132
        goto cleanup;

J
Ján Tomko 已提交
2133
    if (virDirOpen(&dir, pcidir) < 0)
2134 2135
        goto cleanup;

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

    ret = 0;

2159
 cleanup:
J
Ján Tomko 已提交
2160
    VIR_DIR_CLOSE(dir);
2161 2162
    return ret;
}
J
Jiri Denemark 已提交
2163

L
Laine Stump 已提交
2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174

/* 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)
{
2175
    VIR_AUTOFREE(char *) groupPath = NULL;
L
Laine Stump 已提交
2176 2177 2178
    DIR *groupDir = NULL;
    int ret = -1;
    struct dirent *ent;
E
Eric Blake 已提交
2179
    int direrr;
L
Laine Stump 已提交
2180 2181 2182

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

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

E
Eric Blake 已提交
2192
    while ((direrr = virDirRead(groupDir, &ent, groupPath)) > 0) {
L
Laine Stump 已提交
2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204
        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 已提交
2205
    if (direrr < 0)
L
Laine Stump 已提交
2206 2207 2208 2209
        goto cleanup;

    ret = 0;

2210
 cleanup:
J
Ján Tomko 已提交
2211
    VIR_DIR_CLOSE(groupDir);
L
Laine Stump 已提交
2212 2213 2214 2215 2216 2217 2218 2219
    return ret;
}


static int
virPCIDeviceGetIOMMUGroupAddOne(virPCIDeviceAddressPtr newDevAddr, void *opaque)
{
    virPCIDeviceListPtr groupList = opaque;
2220
    VIR_AUTOPTR(virPCIDevice) newDev = NULL;
L
Laine Stump 已提交
2221 2222 2223

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

    if (virPCIDeviceListAdd(groupList, newDev) < 0)
2227
        return -1;
L
Laine Stump 已提交
2228 2229

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


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

2248
    if (virPCIDeviceAddressIOMMUGroupIterate(&(dev->address),
L
Laine Stump 已提交
2249 2250 2251 2252 2253 2254
                                             virPCIDeviceGetIOMMUGroupAddOne,
                                             groupList) < 0)
        goto error;

    return groupList;

2255
 error:
L
Laine Stump 已提交
2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279
    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;

2280 2281
    if (VIR_APPEND_ELEMENT(*addrList->iommuGroupDevices,
                           *addrList->nIommuGroupDevices, copyAddr) < 0)
L
Laine Stump 已提交
2282 2283 2284
        goto cleanup;

    ret = 0;
2285
 cleanup:
L
Laine Stump 已提交
2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312
    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;
2313
 cleanup:
L
Laine Stump 已提交
2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324
    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)
{
2325 2326 2327
    VIR_AUTOFREE(char *) devName = NULL;
    VIR_AUTOFREE(char *) devPath = NULL;
    VIR_AUTOFREE(char *) groupPath = NULL;
L
Laine Stump 已提交
2328 2329 2330 2331
    const char *groupNumStr;
    unsigned int groupNum;

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

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

    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);
2352
        return -1;
L
Laine Stump 已提交
2353 2354
    }

2355
    return groupNum;
L
Laine Stump 已提交
2356 2357 2358
}


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

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

2387 2388 2389
    return groupDev;
}

J
Jiri Denemark 已提交
2390
static int
2391
virPCIDeviceDownstreamLacksACS(virPCIDevicePtr dev)
J
Jiri Denemark 已提交
2392 2393 2394 2395
{
    uint16_t flags;
    uint16_t ctrl;
    unsigned int pos;
2396 2397
    int fd;
    int ret = 0;
2398
    uint16_t device_class;
J
Jiri Denemark 已提交
2399

2400
    if ((fd = virPCIDeviceConfigOpen(dev, true)) < 0)
J
Jiri Denemark 已提交
2401 2402
        return -1;

2403
    if (virPCIDeviceInit(dev, fd) < 0) {
2404 2405 2406 2407
        ret = -1;
        goto cleanup;
    }

2408 2409 2410
    if (virPCIDeviceReadClass(dev, &device_class) < 0)
        goto cleanup;

J
Jiri Denemark 已提交
2411
    pos = dev->pcie_cap_pos;
2412
    if (!pos || device_class != PCI_CLASS_BRIDGE_PCI)
2413
        goto cleanup;
J
Jiri Denemark 已提交
2414

2415
    flags = virPCIDeviceRead16(dev, fd, pos + PCI_EXP_FLAGS);
J
Jiri Denemark 已提交
2416
    if (((flags & PCI_EXP_FLAGS_TYPE) >> 4) != PCI_EXP_TYPE_DOWNSTREAM)
2417
        goto cleanup;
J
Jiri Denemark 已提交
2418

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

2426
    ctrl = virPCIDeviceRead16(dev, fd, pos + PCI_EXT_ACS_CTRL);
J
Jiri Denemark 已提交
2427 2428 2429
    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);
2430 2431
        ret = 1;
        goto cleanup;
J
Jiri Denemark 已提交
2432 2433
    }

2434
 cleanup:
2435
    virPCIDeviceConfigClose(dev, fd);
2436
    return ret;
J
Jiri Denemark 已提交
2437 2438 2439
}

static int
2440
virPCIDeviceIsBehindSwitchLackingACS(virPCIDevicePtr dev)
J
Jiri Denemark 已提交
2441
{
2442
    VIR_AUTOPTR(virPCIDevice) parent = NULL;
J
Jiri Denemark 已提交
2443

2444
    if (virPCIDeviceGetParent(dev, &parent) < 0)
2445
        return -1;
2446 2447 2448 2449 2450
    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.
         */
2451
        if (dev->address.bus == 0) {
2452
            return 0;
2453
        } else {
2454
            virReportError(VIR_ERR_INTERNAL_ERROR,
2455 2456 2457 2458
                           _("Failed to find parent device for %s"),
                           dev->name);
            return -1;
        }
J
Jiri Denemark 已提交
2459 2460 2461 2462 2463 2464 2465
    }

    /* 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 {
2466
        VIR_AUTOPTR(virPCIDevice) tmp = NULL;
J
Jiri Denemark 已提交
2467
        int acs;
2468
        int ret;
J
Jiri Denemark 已提交
2469

2470
        acs = virPCIDeviceDownstreamLacksACS(parent);
J
Jiri Denemark 已提交
2471 2472 2473 2474 2475 2476 2477 2478 2479

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

        tmp = parent;
2480
        ret = virPCIDeviceGetParent(parent, &parent);
2481 2482
        if (ret < 0)
            return -1;
J
Jiri Denemark 已提交
2483 2484 2485 2486 2487
    } while (parent);

    return 0;
}

2488 2489
int virPCIDeviceIsAssignable(virPCIDevicePtr dev,
                             int strict_acs_check)
J
Jiri Denemark 已提交
2490 2491 2492 2493 2494 2495 2496 2497
{
    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.
     */

2498
    ret = virPCIDeviceIsBehindSwitchLackingACS(dev);
J
Jiri Denemark 已提交
2499 2500 2501 2502 2503 2504 2505 2506
    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 {
2507
            virReportError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
2508 2509 2510 2511 2512 2513 2514 2515 2516
                           _("Device %s is behind a switch lacking ACS and "
                             "cannot be assigned"),
                           dev->name);
            return 0;
        }
    }

    return 1;
}
2517 2518 2519 2520 2521 2522 2523 2524 2525 2526

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);
2527
    if (ret != 0)
2528 2529 2530 2531
        VIR_ERROR(_("Failed to convert '%s' to unsigned int"), s);
    return ret;
}

2532 2533
int
virPCIDeviceAddressParse(char *address,
2534
                         virPCIDeviceAddressPtr bdf)
2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560
{
    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;

2561
 out:
2562 2563 2564
    return ret;
}

2565

2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590
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);
}

2591
#ifdef __linux__
2592

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

2606
virPCIDeviceAddressPtr
2607
virPCIGetDeviceAddressFromSysfsLink(const char *device_link)
2608
{
2609
    virPCIDeviceAddressPtr bdf = NULL;
2610
    char *config_address = NULL;
2611
    VIR_AUTOFREE(char *) device_path = NULL;
2612 2613

    if (!virFileExists(device_link)) {
2614
        VIR_DEBUG("'%s' does not exist", device_link);
2615
        return NULL;
2616 2617
    }

2618
    device_path = virFileCanonicalizePath(device_link);
2619
    if (device_path == NULL) {
2620 2621 2622
        virReportSystemError(errno,
                             _("Failed to resolve device link '%s'"),
                             device_link);
2623
        return NULL;
2624 2625
    }

2626
    config_address = last_component(device_path);
2627
    if (VIR_ALLOC(bdf) < 0)
2628
        return NULL;
2629

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

2638
    return bdf;
2639 2640
}

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

2661 2662
    *pf = NULL;

2663 2664
    if (virBuildPath(&device_link, vf_sysfs_path, "physfn") == -1) {
        virReportOOMError();
2665
        return -1;
2666 2667
    }

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

2673
    return 0;
2674 2675
}

2676

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

2692 2693
    *virtual_functions = NULL;
    *num_virtual_functions = 0;
2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709
    *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;
        }
    }
2710

2711
    do {
2712
        VIR_AUTOFREE(char *) device_link = NULL;
2713 2714 2715
        /* 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;
2716

2717 2718
        if (!virFileExists(device_link))
            break;
2719

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

2727 2728
        if (VIR_APPEND_ELEMENT(*virtual_functions, *num_virtual_functions,
                               config_addr) < 0)
2729 2730
            goto error;
    } while (1);
2731

2732 2733
    VIR_DEBUG("Found %zu virtual functions for %s",
              *num_virtual_functions, sysfs_path);
2734
    ret = 0;
2735
 cleanup:
2736
    VIR_FREE(config_addr);
2737
    return ret;
2738

2739
 error:
2740 2741 2742
    for (i = 0; i < *num_virtual_functions; i++)
        VIR_FREE((*virtual_functions)[i]);
    VIR_FREE(*virtual_functions);
2743
    *num_virtual_functions = 0;
2744
    goto cleanup;
2745
}
2746

2747

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

    if (virAsprintf(&vf_sysfs_physfn_link, "%s/physfn",
2757
                    vf_sysfs_device_link) < 0)
2758
        return -1;
2759

2760
    return virFileExists(vf_sysfs_physfn_link);
2761 2762 2763 2764 2765 2766
}

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

2778
    if (!(vf_bdf = virPCIGetDeviceAddressFromSysfsLink(vf_sysfs_device_link)))
2779 2780
        return ret;

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

    for (i = 0; i < num_virt_fns; i++) {
2790 2791 2792 2793 2794
        if (virPCIDeviceAddressIsEqual(vf_bdf, virt_fns[i])) {
            *vf_index = i;
            ret = 0;
            break;
        }
2795 2796
    }

2797
 out:
2798 2799 2800

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

A
ajia@redhat.com 已提交
2803
    VIR_FREE(virt_fns);
2804 2805 2806 2807 2808
    VIR_FREE(vf_bdf);

    return ret;
}

2809 2810 2811 2812 2813
/*
 * Returns a path to the PCI sysfs file given the BDF of the PCI function
 */

int
2814
virPCIGetSysfsFile(char *virPCIDeviceName, char **pci_sysfs_device_link)
2815
{
2816 2817 2818 2819
    if (virAsprintf(pci_sysfs_device_link, PCI_SYSFS "devices/%s",
                    virPCIDeviceName) < 0)
        return -1;
    return 0;
2820 2821
}

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

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

2860 2861
    *netname = NULL;

2862 2863 2864 2865 2866 2867
    if (virBuildPath(&pcidev_sysfs_net_path, device_link_sysfs_path,
                     "net") == -1) {
        virReportOOMError();
        return -1;
    }

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

E
Eric Blake 已提交
2874
    while (virDirRead(dir, &entry, pcidev_sysfs_net_path) > 0) {
2875 2876 2877 2878 2879 2880 2881 2882 2883 2884
        /* 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);
2885 2886 2887 2888 2889 2890 2891 2892 2893 2894
                /* 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;
                }

2895 2896 2897 2898 2899 2900 2901 2902 2903 2904 2905
                continue;
            }
        } else {
            if (i++ < idx)
                continue;
        }

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

        ret = 0;
2906 2907 2908
        break;
    }

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

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

2947
    if (virPCIGetPhysicalFunction(vf_sysfs_device_path, &pf_config_address) < 0)
2948
        goto cleanup;
R
Roopa Prabhu 已提交
2949

2950
    if (!pf_config_address)
2951
        goto cleanup;
2952

2953 2954
    if (virPCIDeviceAddressGetSysfsFile(pf_config_address,
                                        &pf_sysfs_device_path) < 0) {
2955 2956
        goto cleanup;
    }
R
Roopa Prabhu 已提交
2957

2958 2959 2960
    if (virPCIGetVirtualFunctionIndex(pf_sysfs_device_path,
                                      vf_sysfs_device_path, vf_index) < 0) {
        goto cleanup;
R
Roopa Prabhu 已提交
2961 2962
    }

2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982
    /* 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 已提交
2983
        goto cleanup;
2984
    }
R
Roopa Prabhu 已提交
2985

2986 2987 2988 2989 2990 2991 2992 2993 2994
    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 已提交
2995

2996
    ret = 0;
2997
 cleanup:
R
Roopa Prabhu 已提交
2998 2999 3000 3001 3002
    VIR_FREE(pf_config_address);

    return ret;
}

3003 3004 3005 3006 3007 3008 3009 3010 3011

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

3056
#else
3057 3058
static const char *unsupported = N_("not supported on non-linux platforms");

3059 3060 3061 3062
virPCIDeviceAddressPtr
virPCIGetDeviceAddressFromSysfsLink(const char *device_link ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3063
    return NULL;
3064 3065 3066
}


3067
int
3068
virPCIGetPhysicalFunction(const char *vf_sysfs_path ATTRIBUTE_UNUSED,
3069
                          virPCIDeviceAddressPtr *pf ATTRIBUTE_UNUSED)
3070
{
3071
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3072 3073 3074 3075
    return -1;
}

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

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

int
3093 3094 3095
virPCIGetVirtualFunctionIndex(const char *pf_sysfs_device_link ATTRIBUTE_UNUSED,
                              const char *vf_sysfs_device_link ATTRIBUTE_UNUSED,
                              int *vf_index ATTRIBUTE_UNUSED)
3096
{
3097
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3098 3099 3100 3101
    return -1;

}

3102 3103 3104 3105 3106 3107 3108 3109
int
virPCIGetSysfsFile(char *virPCIDeviceName ATTRIBUTE_UNUSED,
                   char **pci_sysfs_device_link ATTRIBUTE_UNUSED)
{
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
    return -1;
}

3110
int
3111 3112
virPCIDeviceAddressGetSysfsFile(virPCIDeviceAddressPtr dev ATTRIBUTE_UNUSED,
                                char **pci_sysfs_device_link ATTRIBUTE_UNUSED)
3113
{
3114
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3115 3116 3117
    return -1;
}

3118
int
3119
virPCIGetNetName(const char *device_link_sysfs_path ATTRIBUTE_UNUSED,
3120 3121
                 size_t idx ATTRIBUTE_UNUSED,
                 char *physPortID ATTRIBUTE_UNUSED,
3122
                 char **netname ATTRIBUTE_UNUSED)
3123
{
3124
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
3125 3126
    return -1;
}
R
Roopa Prabhu 已提交
3127 3128

int
3129
virPCIGetVirtualFunctionInfo(const char *vf_sysfs_device_path ATTRIBUTE_UNUSED,
3130
                             int pfNetDevIdx ATTRIBUTE_UNUSED,
3131 3132
                             char **pfname ATTRIBUTE_UNUSED,
                             int *vf_index ATTRIBUTE_UNUSED)
R
Roopa Prabhu 已提交
3133
{
3134
    virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _(unsupported));
R
Roopa Prabhu 已提交
3135 3136
    return -1;
}
3137 3138 3139 3140 3141 3142 3143 3144 3145


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

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

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

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

    ret = dev->pcie_cap_pos != 0;

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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


3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249
int virPCIGetHeaderType(virPCIDevicePtr dev, int *hdrType)
{
    int fd;
    uint8_t type;

    *hdrType = -1;

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

    type = virPCIDeviceRead8(dev, fd, PCI_HEADER_TYPE);

    virPCIDeviceConfigClose(dev, fd);

    type &= PCI_HEADER_TYPE_MASK;
    if (type >= VIR_PCI_HEADER_LAST) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
3250 3251
                       _("Unknown PCI header type '%d' for device '%s'"),
                       type, dev->name);
3252 3253 3254 3255 3256 3257 3258 3259 3260
        return -1;
    }

    *hdrType = type;

    return 0;
}


3261 3262 3263 3264 3265 3266 3267 3268 3269 3270
void
virPCIEDeviceInfoFree(virPCIEDeviceInfoPtr dev)
{
    if (!dev)
        return;

    VIR_FREE(dev->link_cap);
    VIR_FREE(dev->link_sta);
    VIR_FREE(dev);
}
3271 3272 3273 3274 3275 3276

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