node_device_udev.c 54.6 KB
Newer Older
1 2 3
/*
 * node_device_udev.c: node device enumeration - libudev implementation
 *
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>
#include <libudev.h>
23
#include <pciaccess.h>
24 25 26
#include <scsi/scsi.h>
#include <c-ctype.h>

27
#include "dirname.h"
28
#include "node_device_conf.h"
29
#include "node_device_event.h"
30
#include "node_device_driver.h"
31 32
#include "node_device_udev.h"
#include "virerror.h"
33 34
#include "driver.h"
#include "datatypes.h"
35
#include "virlog.h"
36
#include "viralloc.h"
37
#include "viruuid.h"
38
#include "virbuffer.h"
39
#include "virfile.h"
40
#include "virpci.h"
41
#include "virpidfile.h"
42
#include "virstring.h"
43
#include "virnetdev.h"
44
#include "virmdev.h"
45

46 47
#include "configmake.h"

48 49
#define VIR_FROM_THIS VIR_FROM_NODEDEV

50 51
VIR_LOG_INIT("node_device.node_device_udev");

52 53 54 55
#ifndef TYPE_RAID
# define TYPE_RAID 12
#endif

56 57 58 59 60 61
typedef struct _udevEventData udevEventData;
typedef udevEventData *udevEventDataPtr;

struct _udevEventData {
    virObjectLockable parent;

62 63
    struct udev_monitor *udev_monitor;
    int watch;
64 65 66 67 68 69

    /* Thread data */
    virThread th;
    virCond threadCond;
    bool threadQuit;
    bool dataReady;
70 71
};

72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
static virClassPtr udevEventDataClass;

static void
udevEventDataDispose(void *obj)
{
    struct udev *udev = NULL;
    udevEventDataPtr priv = obj;

    if (priv->watch != -1)
        virEventRemoveHandle(priv->watch);

    if (!priv->udev_monitor)
        return;

    udev = udev_monitor_get_udev(priv->udev_monitor);
    udev_monitor_unref(priv->udev_monitor);
    udev_unref(udev);
89 90

    virCondDestroy(&priv->threadCond);
91 92 93 94 95 96
}


static int
udevEventDataOnceInit(void)
{
97
    if (!VIR_CLASS_NEW(udevEventData, virClassForObjectLockable()))
98 99 100 101 102
        return -1;

    return 0;
}

103
VIR_ONCE_GLOBAL_INIT(udevEventData);
104 105 106 107 108 109 110 111 112 113 114 115

static udevEventDataPtr
udevEventDataNew(void)
{
    udevEventDataPtr ret = NULL;

    if (udevEventDataInitialize() < 0)
        return NULL;

    if (!(ret = virObjectLockableNew(udevEventDataClass)))
        return NULL;

116 117 118 119 120
    if (virCondInit(&ret->threadCond) < 0) {
        virObjectUnref(ret);
        return NULL;
    }

121 122 123 124
    ret->watch = -1;
    return ret;
}

125

J
Ján Tomko 已提交
126 127 128 129 130 131 132 133 134 135 136
static bool
udevHasDeviceProperty(struct udev_device *dev,
                      const char *key)
{
    if (udev_device_get_property_value(dev, key))
        return true;

    return false;
}


137 138 139
static const char *
udevGetDeviceProperty(struct udev_device *udev_device,
                      const char *property_key)
140
{
141
    const char *ret = NULL;
142

143
    ret = udev_device_get_property_value(udev_device, property_key);
144

145 146
    VIR_DEBUG("Found property key '%s' value '%s' for device with sysname '%s'",
              property_key, NULLSTR(ret), udev_device_get_sysname(udev_device));
147 148 149 150 151

    return ret;
}


152 153 154 155
static int
udevGetStringProperty(struct udev_device *udev_device,
                      const char *property_key,
                      char **value)
156
{
157
    *value = g_strdup(udevGetDeviceProperty(udev_device, property_key));
158

159
    return 0;
160 161 162
}


163 164 165 166 167
static int
udevGetIntProperty(struct udev_device *udev_device,
                   const char *property_key,
                   int *value,
                   int base)
168
{
169
    const char *str = NULL;
170

171
    str = udevGetDeviceProperty(udev_device, property_key);
172

173
    if (str && virStrToLong_i(str, NULL, base, value) < 0) {
174 175
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to int"), str);
176
        return -1;
177
    }
178
    return 0;
179 180 181
}


182 183 184 185 186
static int
udevGetUintProperty(struct udev_device *udev_device,
                    const char *property_key,
                    unsigned int *value,
                    int base)
187
{
188
    const char *str = NULL;
189

190
    str = udevGetDeviceProperty(udev_device, property_key);
191

192
    if (str && virStrToLong_ui(str, NULL, base, value) < 0) {
193 194
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to int"), str);
195
        return -1;
196
    }
197
    return 0;
198 199 200
}


201 202 203
static const char *
udevGetDeviceSysfsAttr(struct udev_device *udev_device,
                       const char *attr_name)
204
{
205
    const char *ret = NULL;
206

207
    ret = udev_device_get_sysattr_value(udev_device, attr_name);
208 209

    VIR_DEBUG("Found sysfs attribute '%s' value '%s' "
210
              "for device with sysname '%s'",
211
              attr_name, NULLSTR(ret),
212 213 214 215 216
              udev_device_get_sysname(udev_device));
    return ret;
}


217 218 219 220
static int
udevGetStringSysfsAttr(struct udev_device *udev_device,
                       const char *attr_name,
                       char **value)
221
{
222
    *value = g_strdup(udevGetDeviceSysfsAttr(udev_device, attr_name));
223

224
    virStringStripControlChars(*value);
225

226 227
    if (*value != NULL && (STREQ(*value, "")))
        VIR_FREE(*value);
228

229
    return 0;
230 231 232
}


233 234 235 236 237
static int
udevGetIntSysfsAttr(struct udev_device *udev_device,
                    const char *attr_name,
                    int *value,
                    int base)
238
{
239
    const char *str = NULL;
240

241
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
242

243
    if (str && virStrToLong_i(str, NULL, base, value) < 0) {
244 245
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to int"), str);
246
        return -1;
247 248
    }

249
    return 0;
250 251 252
}


253 254 255 256 257
static int
udevGetUintSysfsAttr(struct udev_device *udev_device,
                     const char *attr_name,
                     unsigned int *value,
                     int base)
258
{
259
    const char *str = NULL;
260

261
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
262

263
    if (str && virStrToLong_ui(str, NULL, base, value) < 0) {
264 265
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to unsigned int"), str);
266
        return -1;
267 268
    }

269
    return 0;
270 271 272
}


273 274 275 276
static int
udevGetUint64SysfsAttr(struct udev_device *udev_device,
                       const char *attr_name,
                       unsigned long long *value)
277
{
278
    const char *str = NULL;
279

280
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
281

282
    if (str && virStrToLong_ull(str, NULL, 0, value) < 0) {
283 284
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to unsigned long long"), str);
J
Ján Tomko 已提交
285
        return -1;
286 287
    }

J
Ján Tomko 已提交
288
    return 0;
289 290 291
}


292 293 294 295
static int
udevGenerateDeviceName(struct udev_device *device,
                       virNodeDeviceDefPtr def,
                       const char *s)
296
{
297
    size_t i;
298 299
    virBuffer buf = VIR_BUFFER_INITIALIZER;

300
    virBufferAsprintf(&buf, "%s_%s",
301 302 303
                      udev_device_get_subsystem(device),
                      udev_device_get_sysname(device));

304
    if (s != NULL)
305
        virBufferAsprintf(&buf, "_%s", s);
306 307 308

    def->name = virBufferContentAndReset(&buf);

309
    for (i = 0; i < strlen(def->name); i++) {
310
        if (!(c_isalnum(*(def->name + i))))
311 312 313
            *(def->name + i) = '_';
    }

314
    return 0;
315 316
}

317 318 319 320 321 322

static int
udevTranslatePCIIds(unsigned int vendor,
                    unsigned int product,
                    char **vendor_string,
                    char **product_string)
323 324 325 326 327 328 329 330 331 332 333 334 335 336 337
{
    struct pci_id_match m;
    const char *vendor_name = NULL, *device_name = NULL;

    m.vendor_id = vendor;
    m.device_id = product;
    m.subvendor_id = PCI_MATCH_ANY;
    m.subdevice_id = PCI_MATCH_ANY;
    m.device_class = 0;
    m.device_class_mask = 0;
    m.match_data = 0;

    /* pci_get_strings returns void */
    pci_get_strings(&m,
                    &device_name,
338
                    &vendor_name,
339 340 341
                    NULL,
                    NULL);

342 343
    *vendor_string = g_strdup(vendor_name);
    *product_string = g_strdup(device_name);
344

345
    return 0;
346 347 348
}


349 350 351
static int
udevProcessPCI(struct udev_device *device,
               virNodeDeviceDefPtr def)
352
{
353
    virNodeDevCapPCIDevPtr pci_dev = &def->caps->data.pci_dev;
354 355
    virPCIEDeviceInfoPtr pci_express = NULL;
    virPCIDevicePtr pciDev = NULL;
356
    int ret = -1;
357
    char *p;
358 359 360 361 362
    bool privileged;

    nodeDeviceLock();
    privileged = driver->privileged;
    nodeDeviceUnlock();
363

364 365
    pci_dev->klass = -1;
    if (udevGetIntProperty(device, "PCI_CLASS", &pci_dev->klass, 16) < 0)
366
        goto cleanup;
367

368
    if ((p = strrchr(def->sysfs_path, '/')) == NULL ||
369 370 371 372
        virStrToLong_ui(p + 1, &p, 16, &pci_dev->domain) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &pci_dev->bus) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &pci_dev->slot) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &pci_dev->function) < 0) {
373 374
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the PCI address from sysfs path: '%s'"),
375
                       def->sysfs_path);
376
        goto cleanup;
377 378
    }

379
    if (udevGetUintSysfsAttr(device, "vendor", &pci_dev->vendor, 16) < 0)
380
        goto cleanup;
381

382
    if (udevGetUintSysfsAttr(device, "device", &pci_dev->product, 16) < 0)
383
        goto cleanup;
384

385 386 387 388
    if (udevTranslatePCIIds(pci_dev->vendor,
                            pci_dev->product,
                            &pci_dev->vendor_name,
                            &pci_dev->product_name) != 0) {
389
        goto cleanup;
390
    }
391

392
    if (udevGenerateDeviceName(device, def, NULL) != 0)
393
        goto cleanup;
394

395 396
    /* The default value is -1, because it can't be 0
     * as zero is valid node number. */
397
    pci_dev->numa_node = -1;
398
    if (udevGetIntSysfsAttr(device, "numa_node",
399
                            &pci_dev->numa_node, 10) < 0)
400
        goto cleanup;
401

402
    if (virNodeDeviceGetPCIDynamicCaps(def->sysfs_path, pci_dev) < 0)
403
        goto cleanup;
404

405 406 407 408
    if (!(pciDev = virPCIDeviceNew(pci_dev->domain,
                                   pci_dev->bus,
                                   pci_dev->slot,
                                   pci_dev->function)))
409
        goto cleanup;
410

411
    /* We need to be root to read PCI device configs */
412
    if (privileged) {
413
        if (virPCIGetHeaderType(pciDev, &pci_dev->hdrType) < 0)
414
            goto cleanup;
415

416 417
        if (virPCIDeviceIsPCIExpress(pciDev) > 0) {
            if (VIR_ALLOC(pci_express) < 0)
418
                goto cleanup;
419

420 421 422
            if (virPCIDeviceHasPCIExpressLink(pciDev) > 0) {
                if (VIR_ALLOC(pci_express->link_cap) < 0 ||
                    VIR_ALLOC(pci_express->link_sta) < 0)
423
                    goto cleanup;
424 425 426 427 428 429 430

                if (virPCIDeviceGetLinkCapSta(pciDev,
                                              &pci_express->link_cap->port,
                                              &pci_express->link_cap->speed,
                                              &pci_express->link_cap->width,
                                              &pci_express->link_sta->speed,
                                              &pci_express->link_sta->width) < 0)
431
                    goto cleanup;
432 433 434

                pci_express->link_sta->port = -1; /* PCIe can't negotiate port. Yet :) */
            }
435 436
            pci_dev->flags |= VIR_NODE_DEV_CAP_FLAG_PCIE;
            pci_dev->pci_express = pci_express;
437
            pci_express = NULL;
438 439 440
        }
    }

441 442
    ret = 0;

443
 cleanup:
444
    virPCIDeviceFree(pciDev);
445
    virPCIEDeviceInfoFree(pci_express);
446 447 448
    return ret;
}

449 450 451

static int
drmGetMinorType(int minor)
M
Marc-André Lureau 已提交
452 453 454 455 456 457 458 459 460 461 462 463 464 465 466 467
{
    int type = minor >> 6;

    if (minor < 0)
        return -1;

    switch (type) {
    case VIR_NODE_DEV_DRM_PRIMARY:
    case VIR_NODE_DEV_DRM_CONTROL:
    case VIR_NODE_DEV_DRM_RENDER:
        return type;
    default:
        return -1;
    }
}

468 469 470 471

static int
udevProcessDRMDevice(struct udev_device *device,
                     virNodeDeviceDefPtr def)
M
Marc-André Lureau 已提交
472
{
473
    virNodeDevCapDRMPtr drm = &def->caps->data.drm;
M
Marc-André Lureau 已提交
474 475 476 477 478 479 480 481 482 483 484
    int minor;

    if (udevGenerateDeviceName(device, def, NULL) != 0)
        return -1;

    if (udevGetIntProperty(device, "MINOR", &minor, 10) < 0)
        return -1;

    if ((minor = drmGetMinorType(minor)) == -1)
        return -1;

485
    drm->type = minor;
M
Marc-André Lureau 已提交
486 487 488

    return 0;
}
489

490 491 492 493

static int
udevProcessUSBDevice(struct udev_device *device,
                     virNodeDeviceDefPtr def)
494
{
495
    virNodeDevCapUSBDevPtr usb_dev = &def->caps->data.usb_dev;
496

497
    if (udevGetUintProperty(device, "BUSNUM", &usb_dev->bus, 10) < 0)
498
        return -1;
499
    if (udevGetUintProperty(device, "DEVNUM", &usb_dev->device, 10) < 0)
500
        return -1;
501
    if (udevGetUintProperty(device, "ID_VENDOR_ID", &usb_dev->vendor, 16) < 0)
502
        return -1;
503

504 505
    if (udevGetStringProperty(device,
                              "ID_VENDOR_FROM_DATABASE",
506
                              &usb_dev->vendor_name) < 0)
507
        return -1;
508

509
    if (!usb_dev->vendor_name &&
510
        udevGetStringSysfsAttr(device, "manufacturer",
511
                               &usb_dev->vendor_name) < 0)
512
        return -1;
513

514
    if (udevGetUintProperty(device, "ID_MODEL_ID", &usb_dev->product, 16) < 0)
515
        return -1;
516

517 518
    if (udevGetStringProperty(device,
                              "ID_MODEL_FROM_DATABASE",
519
                              &usb_dev->product_name) < 0)
520
        return -1;
521

522
    if (!usb_dev->product_name &&
523
        udevGetStringSysfsAttr(device, "product",
524
                               &usb_dev->product_name) < 0)
525
        return -1;
526

527
    if (udevGenerateDeviceName(device, def, NULL) != 0)
528
        return -1;
529

530
    return 0;
531 532 533
}


534 535 536
static int
udevProcessUSBInterface(struct udev_device *device,
                        virNodeDeviceDefPtr def)
537
{
538
    virNodeDevCapUSBIfPtr usb_if = &def->caps->data.usb_if;
539

540
    if (udevGetUintSysfsAttr(device, "bInterfaceNumber",
541
                             &usb_if->number, 16) < 0)
542
        return -1;
543

544
    if (udevGetUintSysfsAttr(device, "bInterfaceClass",
545
                             &usb_if->klass, 16) < 0)
546
        return -1;
547

548
    if (udevGetUintSysfsAttr(device, "bInterfaceSubClass",
549
                             &usb_if->subclass, 16) < 0)
550
        return -1;
551

552
    if (udevGetUintSysfsAttr(device, "bInterfaceProtocol",
553
                             &usb_if->protocol, 16) < 0)
554
        return -1;
555

556
    if (udevGenerateDeviceName(device, def, NULL) != 0)
557
        return -1;
558

559
    return 0;
560 561 562
}


563 564 565
static int
udevProcessNetworkInterface(struct udev_device *device,
                            virNodeDeviceDefPtr def)
566
{
D
David Allan 已提交
567
    const char *devtype = udev_device_get_devtype(device);
568
    virNodeDevCapNetPtr net = &def->caps->data.net;
569

D
David Allan 已提交
570
    if (devtype && STREQ(devtype, "wlan")) {
571
        net->subtype = VIR_NODE_DEV_CAP_NET_80211;
D
David Allan 已提交
572
    } else {
573
        net->subtype = VIR_NODE_DEV_CAP_NET_80203;
D
David Allan 已提交
574 575
    }

576 577
    if (udevGetStringProperty(device,
                              "INTERFACE",
578
                              &net->ifname) < 0)
579
        return -1;
580

581
    if (udevGetStringSysfsAttr(device, "address",
582
                               &net->address) < 0)
583
        return -1;
584

585
    if (udevGetUintSysfsAttr(device, "addr_len", &net->address_len, 0) < 0)
586
        return -1;
587

588
    if (udevGenerateDeviceName(device, def, net->address) != 0)
589
        return -1;
590

591
    if (virNetDevGetLinkInfo(net->ifname, &net->lnk) < 0)
592
        return -1;
593

594
    if (virNetDevGetFeatures(net->ifname, &net->features) < 0)
595
        return -1;
596

597
    return 0;
598 599 600
}


601
static int
J
Ján Tomko 已提交
602
udevProcessSCSIHost(struct udev_device *device G_GNUC_UNUSED,
603
                    virNodeDeviceDefPtr def)
604
{
605
    virNodeDevCapSCSIHostPtr scsi_host = &def->caps->data.scsi_host;
606
    char *filename = NULL;
J
Ján Tomko 已提交
607
    char *str;
608

609
    filename = last_component(def->sysfs_path);
610

611
    if (!(str = STRSKIP(filename, "host")) ||
612
        virStrToLong_ui(str, NULL, 0, &scsi_host->host) < 0) {
613 614 615
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse SCSI host '%s'"),
                       filename);
616
        return -1;
617 618
    }

619
    virNodeDeviceGetSCSIHostCaps(&def->caps->data.scsi_host);
620

621
    if (udevGenerateDeviceName(device, def, NULL) != 0)
622
        return -1;
623

624
    return 0;
625 626 627
}


628 629 630
static int
udevProcessSCSITarget(struct udev_device *device,
                      virNodeDeviceDefPtr def)
D
David Allan 已提交
631 632
{
    const char *sysname = NULL;
633
    virNodeDevCapSCSITargetPtr scsi_target = &def->caps->data.scsi_target;
D
David Allan 已提交
634 635 636

    sysname = udev_device_get_sysname(device);

637
    scsi_target->name = g_strdup(sysname);
D
David Allan 已提交
638

639
    virNodeDeviceGetSCSITargetCaps(def->sysfs_path, &def->caps->data.scsi_target);
640

641
    if (udevGenerateDeviceName(device, def, NULL) != 0)
642
        return -1;
D
David Allan 已提交
643

644
    return 0;
D
David Allan 已提交
645 646 647
}


648
static int
J
Ján Tomko 已提交
649
udevGetSCSIType(virNodeDeviceDefPtr def G_GNUC_UNUSED,
650 651
                unsigned int type,
                char **typestring)
652 653 654 655 656 657 658 659
{
    int ret = 0;
    int foundtype = 1;

    *typestring = NULL;

    switch (type) {
    case TYPE_DISK:
660
        *typestring = g_strdup("disk");
661 662
        break;
    case TYPE_TAPE:
663
        *typestring = g_strdup("tape");
664 665
        break;
    case TYPE_PROCESSOR:
666
        *typestring = g_strdup("processor");
667 668
        break;
    case TYPE_WORM:
669
        *typestring = g_strdup("worm");
670 671
        break;
    case TYPE_ROM:
672
        *typestring = g_strdup("cdrom");
673 674
        break;
    case TYPE_SCANNER:
675
        *typestring = g_strdup("scanner");
676 677
        break;
    case TYPE_MOD:
678
        *typestring = g_strdup("mod");
679 680
        break;
    case TYPE_MEDIUM_CHANGER:
681
        *typestring = g_strdup("changer");
682 683
        break;
    case TYPE_ENCLOSURE:
684
        *typestring = g_strdup("enclosure");
685
        break;
686
    case TYPE_RAID:
687
        *typestring = g_strdup("raid");
688
        break;
689 690 691 692 693 694 695 696 697 698
    case TYPE_NO_LUN:
    default:
        foundtype = 0;
        break;
    }

    if (*typestring == NULL) {
        if (foundtype == 1) {
            ret = -1;
        } else {
699 700
            VIR_DEBUG("Failed to find SCSI device type %d for %s",
                      type, def->sysfs_path);
701 702 703 704 705 706 707
        }
    }

    return ret;
}


708
static int
J
Ján Tomko 已提交
709
udevProcessSCSIDevice(struct udev_device *device G_GNUC_UNUSED,
710
                      virNodeDeviceDefPtr def)
711 712 713
{
    int ret = -1;
    unsigned int tmp = 0;
714
    virNodeDevCapSCSIPtr scsi = &def->caps->data.scsi;
715 716
    char *filename = NULL, *p = NULL;

717
    filename = last_component(def->sysfs_path);
718

719 720 721 722
    if (virStrToLong_ui(filename, &p, 10, &scsi->host) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 10, &scsi->bus) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 10, &scsi->target) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 10, &scsi->lun) < 0) {
723 724 725 726
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the SCSI address from filename: '%s'"),
                       filename);
        return -1;
727 728
    }

729 730
    if (udev_device_get_sysattr_value(device, "type")) {
        if (udevGetUintSysfsAttr(device, "type", &tmp, 0) < 0)
731
            goto cleanup;
732

733
        if (udevGetSCSIType(def, tmp, &scsi->type) < 0)
734
            goto cleanup;
735 736
    }

737
    if (udevGenerateDeviceName(device, def, NULL) != 0)
738
        goto cleanup;
739 740 741

    ret = 0;

742
 cleanup:
743
    if (ret != 0) {
744 745 746
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to process SCSI device with sysfs path '%s'"),
                       def->sysfs_path);
747 748 749 750 751
    }
    return ret;
}


752 753 754
static int
udevProcessDisk(struct udev_device *device,
                virNodeDeviceDefPtr def)
755
{
756
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
757

758
    if (udevGetUint64SysfsAttr(device, "size", &storage->num_blocks) < 0)
759
        return -1;
760

J
Ján Tomko 已提交
761
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
762
                               &storage->logical_block_size) < 0)
763
        return -1;
764

765
    storage->size = storage->num_blocks * storage->logical_block_size;
766

767
    return 0;
768 769 770
}


771 772 773 774
static int
udevProcessRemoveableMedia(struct udev_device *device,
                           virNodeDeviceDefPtr def,
                           int has_media)
775
{
776
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
J
Ján Tomko 已提交
777
    int is_removable = 0;
778

779 780 781
    if (udevGetIntSysfsAttr(device, "removable", &is_removable, 0) < 0)
        return -1;
    if (is_removable == 1)
782 783
        def->caps->data.storage.flags |= VIR_NODE_DEV_CAP_STORAGE_REMOVABLE;

J
Ján Tomko 已提交
784 785
    if (!has_media)
        return 0;
786

J
Ján Tomko 已提交
787 788
    def->caps->data.storage.flags |=
        VIR_NODE_DEV_CAP_STORAGE_REMOVABLE_MEDIA_AVAILABLE;
789

J
Ján Tomko 已提交
790
    if (udevGetStringProperty(device, "ID_FS_LABEL",
791
                              &storage->media_label) < 0)
J
Ján Tomko 已提交
792
        return -1;
793

J
Ján Tomko 已提交
794
    if (udevGetUint64SysfsAttr(device, "size",
795
                               &storage->num_blocks) < 0)
J
Ján Tomko 已提交
796
        return -1;
797

J
Ján Tomko 已提交
798
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
799
                               &storage->logical_block_size) < 0)
J
Ján Tomko 已提交
800
        return -1;
801

J
Ján Tomko 已提交
802 803 804 805 806 807 808
    /* XXX This calculation is wrong for the qemu virtual cdrom
     * which reports the size in 512 byte blocks, but the logical
     * block size as 2048.  I don't have a physical cdrom on a
     * devel system to see how they behave. */
    def->caps->data.storage.removable_media_size =
        def->caps->data.storage.num_blocks *
        def->caps->data.storage.logical_block_size;
809

J
Ján Tomko 已提交
810
    return 0;
811 812
}

813 814 815 816

static int
udevProcessCDROM(struct udev_device *device,
                 virNodeDeviceDefPtr def)
817 818 819 820 821 822 823 824
{
    int has_media = 0;

    /* NB: the drive_type string provided by udev is different from
     * that provided by HAL; now it's "cd" instead of "cdrom" We
     * change it to cdrom to preserve compatibility with earlier
     * versions of libvirt.  */
    VIR_FREE(def->caps->data.storage.drive_type);
825
    def->caps->data.storage.drive_type = g_strdup("cdrom");
826

827 828
    if (udevHasDeviceProperty(device, "ID_CDROM_MEDIA") &&
        udevGetIntProperty(device, "ID_CDROM_MEDIA", &has_media, 0) < 0)
829
        return -1;
830

831
    return udevProcessRemoveableMedia(device, def, has_media);
832 833
}

834 835 836 837

static int
udevProcessFloppy(struct udev_device *device,
                  virNodeDeviceDefPtr def)
838 839 840
{
    int has_media = 0;

841
    if (udevHasDeviceProperty(device, "ID_CDROM_MEDIA")) {
842
        /* USB floppy */
843 844
        if (udevGetIntProperty(device, "DKD_MEDIA_AVAILABLE", &has_media, 0) < 0)
            return -1;
845
    } else if (udevHasDeviceProperty(device, "ID_FS_LABEL")) {
846 847 848 849 850 851
        /* Legacy floppy */
        has_media = 1;
    }

    return udevProcessRemoveableMedia(device, def, has_media);
}
852

853

854 855 856
static int
udevProcessSD(struct udev_device *device,
              virNodeDeviceDefPtr def)
857
{
858
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
859

J
Ján Tomko 已提交
860
    if (udevGetUint64SysfsAttr(device, "size",
861
                               &storage->num_blocks) < 0)
862
        return -1;
863

J
Ján Tomko 已提交
864
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
865
                               &storage->logical_block_size) < 0)
866
        return -1;
867

868
    storage->size = storage->num_blocks * storage->logical_block_size;
869

870
    return 0;
871 872 873
}


874 875 876 877
/* This function exists to deal with the case in which a driver does
 * not provide a device type in the usual place, but udev told us it's
 * a storage device, and we can make a good guess at what kind of
 * storage device it is from other information that is provided. */
878 879
static int
udevKludgeStorageType(virNodeDeviceDefPtr def)
880
{
881 882 883
    VIR_DEBUG("Could not find definitive storage type for device "
              "with sysfs path '%s', trying to guess it",
              def->sysfs_path);
884

885
    /* virtio disk */
886 887
    if (STRPREFIX(def->caps->data.storage.block, "/dev/vd")) {
        def->caps->data.storage.drive_type = g_strdup("disk");
888
        VIR_DEBUG("Found storage type '%s' for device "
889
                  "with sysfs path '%s'",
890 891
                  def->caps->data.storage.drive_type,
                  def->sysfs_path);
892
        return 0;
893
    }
894 895 896
    VIR_DEBUG("Could not determine storage type "
              "for device with sysfs path '%s'", def->sysfs_path);
    return -1;
897 898 899
}


900 901 902
static int
udevProcessStorage(struct udev_device *device,
                   virNodeDeviceDefPtr def)
903
{
904
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
905
    int ret = -1;
906
    const char* devnode;
907

908
    devnode = udev_device_get_devnode(device);
909
    if (!devnode) {
910
        VIR_DEBUG("No devnode for '%s'", udev_device_get_devpath(device));
911
        goto cleanup;
912
    }
913

914
    storage->block = g_strdup(devnode);
915

916
    if (udevGetStringProperty(device, "ID_BUS", &storage->bus) < 0)
917
        goto cleanup;
918
    if (udevGetStringProperty(device, "ID_SERIAL", &storage->serial) < 0)
919
        goto cleanup;
920

921
    if (udevGetStringSysfsAttr(device, "device/vendor", &storage->vendor) < 0)
922
        goto cleanup;
923 924 925
    if (def->caps->data.storage.vendor)
        virTrimSpaces(def->caps->data.storage.vendor, NULL);

926
    if (udevGetStringSysfsAttr(device, "device/model", &storage->model) < 0)
927
        goto cleanup;
928 929
    if (def->caps->data.storage.model)
        virTrimSpaces(def->caps->data.storage.model, NULL);
930 931 932 933 934
    /* There is no equivalent of the hotpluggable property in libudev,
     * but storage is going toward a world in which hotpluggable is
     * expected, so I don't see a problem with not having a property
     * for it. */

935
    if (udevGetStringProperty(device, "ID_TYPE", &storage->drive_type) < 0)
936
        goto cleanup;
937

938
    if (!storage->drive_type ||
939
        STREQ(def->caps->data.storage.drive_type, "generic")) {
940 941
        int val = 0;
        const char *str = NULL;
942 943 944

        /* All floppy drives have the ID_DRIVE_FLOPPY prop. This is
         * needed since legacy floppies don't have a drive_type */
945
        if (udevGetIntProperty(device, "ID_DRIVE_FLOPPY", &val, 0) < 0)
946
            goto cleanup;
947 948
        else if (val == 1)
            str = "floppy";
949

950
        if (!str) {
951
            if (udevGetIntProperty(device, "ID_CDROM", &val, 0) < 0)
952
                goto cleanup;
953 954 955
            else if (val == 1)
                str = "cd";
        }
956

957
        if (!str) {
958
            if (udevGetIntProperty(device, "ID_DRIVE_FLASH_SD", &val, 0) < 0)
959
                goto cleanup;
960 961 962
            if (val == 1)
                str = "sd";
        }
963

964
        if (str) {
965
            storage->drive_type = g_strdup(str);
966 967
        } else {
            /* If udev doesn't have it, perhaps we can guess it. */
968
            if (udevKludgeStorageType(def) != 0)
969
                goto cleanup;
970 971 972 973 974 975 976
        }
    }

    if (STREQ(def->caps->data.storage.drive_type, "cd")) {
        ret = udevProcessCDROM(device, def);
    } else if (STREQ(def->caps->data.storage.drive_type, "disk")) {
        ret = udevProcessDisk(device, def);
977 978
    } else if (STREQ(def->caps->data.storage.drive_type, "floppy")) {
        ret = udevProcessFloppy(device, def);
979 980
    } else if (STREQ(def->caps->data.storage.drive_type, "sd")) {
        ret = udevProcessSD(device, def);
981
    } else {
982 983
        VIR_DEBUG("Unsupported storage type '%s'",
                  def->caps->data.storage.drive_type);
984
        goto cleanup;
985 986
    }

987
    if (udevGenerateDeviceName(device, def, storage->serial) != 0)
988
        goto cleanup;
989

990
 cleanup:
991
    VIR_DEBUG("Storage ret=%d", ret);
992 993 994
    return ret;
}

995

996
static int
997
udevProcessSCSIGeneric(struct udev_device *dev,
998 999
                       virNodeDeviceDefPtr def)
{
1000 1001
    if (udevGetStringProperty(dev, "DEVNAME", &def->caps->data.sg.path) < 0 ||
        !def->caps->data.sg.path)
1002 1003 1004 1005 1006 1007 1008 1009
        return -1;

    if (udevGenerateDeviceName(dev, def, NULL) != 0)
        return -1;

    return 0;
}

1010

1011 1012 1013 1014 1015 1016 1017 1018
static int
udevProcessMediatedDevice(struct udev_device *dev,
                          virNodeDeviceDefPtr def)
{
    int ret = -1;
    const char *uuidstr = NULL;
    int iommugrp = -1;
    char *linkpath = NULL;
1019
    char *canonicalpath = NULL;
1020 1021
    virNodeDevCapMdevPtr data = &def->caps->data.mdev;

1022 1023 1024 1025 1026 1027
    /* Because of a kernel uevent race, we might get the 'add' event prior to
     * the sysfs tree being ready, so any attempt to access any sysfs attribute
     * would result in ENOENT and us dropping the device, so let's work around
     * it by waiting for the attributes to become available.
     */

1028
    linkpath = g_strdup_printf("%s/mdev_type", udev_device_get_syspath(dev));
1029

1030 1031 1032 1033 1034 1035 1036
    if (virFileWaitForExists(linkpath, 1, 100) < 0) {
        virReportSystemError(errno,
                             _("failed to wait for file '%s' to appear"),
                             linkpath);
        goto cleanup;
    }

1037 1038
    if (virFileResolveLink(linkpath, &canonicalpath) < 0) {
        virReportSystemError(errno, _("failed to resolve '%s'"), linkpath);
1039
        goto cleanup;
1040
    }
1041

1042
    data->type = g_strdup(last_component(canonicalpath));
1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055

    uuidstr = udev_device_get_sysname(dev);
    if ((iommugrp = virMediatedDeviceGetIOMMUGroupNum(uuidstr)) < 0)
        goto cleanup;

    if (udevGenerateDeviceName(dev, def, NULL) != 0)
        goto cleanup;

    data->iommuGroupNumber = iommugrp;

    ret = 0;
 cleanup:
    VIR_FREE(linkpath);
1056
    VIR_FREE(canonicalpath);
1057 1058 1059
    return ret;
}

B
Bjoern Walk 已提交
1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089

static int
udevProcessCCW(struct udev_device *device,
               virNodeDeviceDefPtr def)
{
    int online;
    char *p;
    virNodeDevCapDataPtr data = &def->caps->data;

    /* process only online devices to keep the list sane */
    if (udevGetIntSysfsAttr(device, "online", &online, 0) < 0 || online != 1)
        return -1;

    if ((p = strrchr(def->sysfs_path, '/')) == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->ccw_dev.cssid) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->ccw_dev.ssid) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->ccw_dev.devno) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the CCW address from sysfs path: '%s'"),
                       def->sysfs_path);
        return -1;
    }

    if (udevGenerateDeviceName(device, def, NULL) != 0)
        return -1;

    return 0;
}


1090 1091 1092 1093 1094 1095 1096 1097 1098 1099
static int
udevGetDeviceNodes(struct udev_device *device,
                   virNodeDeviceDefPtr def)
{
    const char *devnode = NULL;
    struct udev_list_entry *list_entry = NULL;
    int n = 0;

    devnode = udev_device_get_devnode(device);

1100
    def->devnode = g_strdup(devnode);
1101 1102 1103 1104 1105 1106 1107 1108 1109

    udev_list_entry_foreach(list_entry, udev_device_get_devlinks_list_entry(device))
        n++;

    if (VIR_ALLOC_N(def->devlinks, n + 1) < 0)
        return -1;

    n = 0;
    udev_list_entry_foreach(list_entry, udev_device_get_devlinks_list_entry(device)) {
1110
        def->devlinks[n++] = g_strdup(udev_list_entry_get_name(list_entry));
1111 1112 1113 1114 1115
    }

    return 0;
}

1116

1117 1118
static int
udevGetDeviceType(struct udev_device *device,
1119
                  virNodeDevCapType *type)
1120 1121
{
    const char *devtype = NULL;
1122
    char *subsystem = NULL;
1123
    int ret = -1;
D
David Allan 已提交
1124

1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141 1142
    devtype = udev_device_get_devtype(device);
    *type = 0;

    if (devtype) {
        if (STREQ(devtype, "usb_device"))
            *type = VIR_NODE_DEV_CAP_USB_DEV;
        else if (STREQ(devtype, "usb_interface"))
            *type = VIR_NODE_DEV_CAP_USB_INTERFACE;
        else if (STREQ(devtype, "scsi_host"))
            *type = VIR_NODE_DEV_CAP_SCSI_HOST;
        else if (STREQ(devtype, "scsi_target"))
            *type = VIR_NODE_DEV_CAP_SCSI_TARGET;
        else if (STREQ(devtype, "scsi_device"))
            *type = VIR_NODE_DEV_CAP_SCSI;
        else if (STREQ(devtype, "disk"))
            *type = VIR_NODE_DEV_CAP_STORAGE;
        else if (STREQ(devtype, "wlan"))
            *type = VIR_NODE_DEV_CAP_NET;
M
Marc-André Lureau 已提交
1143 1144
        else if (STREQ(devtype, "drm_minor"))
            *type = VIR_NODE_DEV_CAP_DRM;
1145 1146
    } else {
        /* PCI devices don't set the DEVTYPE property. */
1147
        if (udevHasDeviceProperty(device, "PCI_CLASS"))
1148
            *type = VIR_NODE_DEV_CAP_PCI_DEV;
1149

1150 1151 1152 1153
        /* Wired network interfaces don't set the DEVTYPE property,
         * USB devices also have an INTERFACE property, but they do
         * set DEVTYPE, so if devtype is NULL and the INTERFACE
         * property exists, we have a network device. */
1154
        if (udevHasDeviceProperty(device, "INTERFACE"))
1155
            *type = VIR_NODE_DEV_CAP_NET;
1156

B
Bjoern Walk 已提交
1157 1158
        /* The following devices do not set the DEVTYPE property, therefore
         * we need to rely on the SUBSYSTEM property */
1159 1160 1161 1162
        if (udevGetStringProperty(device, "SUBSYSTEM", &subsystem) < 0)
            return -1;

        if (STREQ_NULLABLE(subsystem, "scsi_generic"))
1163
            *type = VIR_NODE_DEV_CAP_SCSI_GENERIC;
1164 1165
        else if (STREQ_NULLABLE(subsystem, "mdev"))
            *type = VIR_NODE_DEV_CAP_MDEV;
B
Bjoern Walk 已提交
1166 1167
        else if (STREQ_NULLABLE(subsystem, "ccw"))
            *type = VIR_NODE_DEV_CAP_CCW_DEV;
1168

1169
        VIR_FREE(subsystem);
1170 1171
    }

1172 1173 1174 1175 1176 1177
    if (!*type)
        VIR_DEBUG("Could not determine device type for device "
                  "with sysfs name '%s'",
                  udev_device_get_sysname(device));
    else
        ret = 0;
1178 1179 1180 1181 1182

    return ret;
}


1183 1184 1185
static int
udevGetDeviceDetails(struct udev_device *device,
                     virNodeDeviceDefPtr def)
1186
{
1187
    switch (def->caps->data.type) {
1188
    case VIR_NODE_DEV_CAP_PCI_DEV:
1189
        return udevProcessPCI(device, def);
1190
    case VIR_NODE_DEV_CAP_USB_DEV:
1191
        return udevProcessUSBDevice(device, def);
1192
    case VIR_NODE_DEV_CAP_USB_INTERFACE:
1193
        return udevProcessUSBInterface(device, def);
1194
    case VIR_NODE_DEV_CAP_NET:
1195
        return udevProcessNetworkInterface(device, def);
1196
    case VIR_NODE_DEV_CAP_SCSI_HOST:
1197
        return udevProcessSCSIHost(device, def);
D
David Allan 已提交
1198
    case VIR_NODE_DEV_CAP_SCSI_TARGET:
1199
        return udevProcessSCSITarget(device, def);
1200
    case VIR_NODE_DEV_CAP_SCSI:
1201
        return udevProcessSCSIDevice(device, def);
1202
    case VIR_NODE_DEV_CAP_STORAGE:
1203
        return udevProcessStorage(device, def);
1204
    case VIR_NODE_DEV_CAP_SCSI_GENERIC:
1205
        return udevProcessSCSIGeneric(device, def);
M
Marc-André Lureau 已提交
1206
    case VIR_NODE_DEV_CAP_DRM:
1207
        return udevProcessDRMDevice(device, def);
1208
    case VIR_NODE_DEV_CAP_MDEV:
1209
        return udevProcessMediatedDevice(device, def);
B
Bjoern Walk 已提交
1210 1211
    case VIR_NODE_DEV_CAP_CCW_DEV:
        return udevProcessCCW(device, def);
1212
    case VIR_NODE_DEV_CAP_MDEV_TYPES:
1213 1214 1215 1216
    case VIR_NODE_DEV_CAP_SYSTEM:
    case VIR_NODE_DEV_CAP_FC_HOST:
    case VIR_NODE_DEV_CAP_VPORTS:
    case VIR_NODE_DEV_CAP_LAST:
1217 1218 1219
        break;
    }

1220
    return 0;
1221 1222 1223
}


1224 1225
static int
udevRemoveOneDevice(struct udev_device *device)
1226
{
1227
    virNodeDeviceObjPtr obj = NULL;
1228
    virNodeDeviceDefPtr def;
1229
    virObjectEventPtr event = NULL;
1230 1231 1232
    const char *name = NULL;

    name = udev_device_get_syspath(device);
1233
    if (!(obj = virNodeDeviceObjListFindBySysfsPath(driver->devs, name))) {
1234 1235
        VIR_DEBUG("Failed to find device to remove that has udev name '%s'",
                  name);
1236
        return -1;
1237
    }
1238
    def = virNodeDeviceObjGetDef(obj);
1239

1240
    event = virNodeDeviceEventLifecycleNew(def->name,
1241 1242 1243 1244
                                           VIR_NODE_DEVICE_EVENT_DELETED,
                                           0);

    VIR_DEBUG("Removing device '%s' with sysfs path '%s'",
1245
              def->name, name);
1246
    virNodeDeviceObjListRemove(driver->devs, obj);
1247
    virObjectUnref(obj);
1248

1249
    virObjectEventStateQueue(driver->nodeDeviceEventState, event);
1250
    return 0;
1251 1252 1253
}


1254 1255 1256
static int
udevSetParent(struct udev_device *device,
              virNodeDeviceDefPtr def)
1257 1258 1259
{
    struct udev_device *parent_device = NULL;
    const char *parent_sysfs_path = NULL;
1260
    virNodeDeviceObjPtr obj = NULL;
1261
    virNodeDeviceDefPtr objdef;
1262 1263
    int ret = -1;

1264 1265
    parent_device = device;
    do {
1266

1267
        parent_device = udev_device_get_parent(parent_device);
1268
        if (parent_device == NULL)
1269
            break;
1270

1271 1272
        parent_sysfs_path = udev_device_get_syspath(parent_device);
        if (parent_sysfs_path == NULL) {
1273 1274 1275
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Could not get syspath for parent of '%s'"),
                           udev_device_get_syspath(parent_device));
1276
            goto cleanup;
1277 1278
        }

1279 1280
        if ((obj = virNodeDeviceObjListFindBySysfsPath(driver->devs,
                                                       parent_sysfs_path))) {
1281
            objdef = virNodeDeviceObjGetDef(obj);
1282
            def->parent = g_strdup(objdef->name);
1283
            virNodeDeviceObjEndAPI(&obj);
1284

1285
            def->parent_sysfs_path = g_strdup(parent_sysfs_path);
1286 1287 1288 1289
        }

    } while (def->parent == NULL && parent_device != NULL);

1290 1291
    if (!def->parent)
        def->parent = g_strdup("computer");
1292 1293 1294

    ret = 0;

1295
 cleanup:
1296 1297 1298 1299
    return ret;
}


1300 1301
static int
udevAddOneDevice(struct udev_device *device)
1302 1303
{
    virNodeDeviceDefPtr def = NULL;
1304
    virNodeDeviceObjPtr obj = NULL;
1305
    virNodeDeviceDefPtr objdef;
1306 1307
    virObjectEventPtr event = NULL;
    bool new_device = true;
1308 1309
    int ret = -1;

1310
    if (VIR_ALLOC(def) != 0)
1311
        goto cleanup;
1312

1313
    def->sysfs_path = g_strdup(udev_device_get_syspath(device));
1314

1315
    if (udevGetStringProperty(device, "DRIVER", &def->driver) < 0)
1316
        goto cleanup;
1317

1318
    if (VIR_ALLOC(def->caps) != 0)
1319
        goto cleanup;
1320

1321
    if (udevGetDeviceType(device, &def->caps->data.type) != 0)
1322
        goto cleanup;
1323

1324 1325 1326
    if (udevGetDeviceNodes(device, def) != 0)
        goto cleanup;

1327
    if (udevGetDeviceDetails(device, def) != 0)
1328
        goto cleanup;
1329

1330
    if (udevSetParent(device, def) != 0)
1331
        goto cleanup;
1332

1333
    if ((obj = virNodeDeviceObjListFindByName(driver->devs, def->name))) {
1334
        virNodeDeviceObjEndAPI(&obj);
1335 1336 1337
        new_device = false;
    }

1338 1339
    /* If this is a device change, the old definition will be freed
     * and the current definition will take its place. */
1340
    if (!(obj = virNodeDeviceObjListAssignDef(driver->devs, def)))
1341
        goto cleanup;
1342
    objdef = virNodeDeviceObjGetDef(obj);
1343

1344
    if (new_device)
1345
        event = virNodeDeviceEventLifecycleNew(objdef->name,
1346 1347
                                               VIR_NODE_DEVICE_EVENT_CREATED,
                                               0);
1348
    else
1349
        event = virNodeDeviceEventUpdateNew(objdef->name);
1350

1351
    virNodeDeviceObjEndAPI(&obj);
1352 1353 1354

    ret = 0;

1355
 cleanup:
1356
    virObjectEventStateQueue(driver->nodeDeviceEventState, event);
1357

1358
    if (ret != 0) {
1359
        VIR_DEBUG("Discarding device %d %p %s", ret, def,
1360
                  def ? NULLSTR(def->sysfs_path) : "");
1361 1362 1363
        virNodeDeviceDefFree(def);
    }

1364 1365 1366 1367
    return ret;
}


1368 1369 1370
static int
udevProcessDeviceListEntry(struct udev *udev,
                           struct udev_list_entry *list_entry)
1371 1372 1373 1374 1375 1376 1377 1378
{
    struct udev_device *device;
    const char *name = NULL;
    int ret = -1;

    name = udev_list_entry_get_name(list_entry);

    device = udev_device_new_from_syspath(udev, name);
1379

1380 1381
    if (device != NULL) {
        if (udevAddOneDevice(device) != 0) {
1382 1383
            VIR_DEBUG("Failed to create node device for udev device '%s'",
                      name);
1384 1385 1386 1387
        }
        ret = 0;
    }

1388 1389
    udev_device_unref(device);

1390 1391 1392 1393
    return ret;
}


1394 1395 1396 1397 1398 1399 1400 1401
/* We do not care about every device (see udevGetDeviceType).
 * Do not bother enumerating over subsystems that do not
 * contain interesting devices.
 */
const char *subsystem_blacklist[] = {
    "acpi", "tty", "vc", "i2c",
};

1402 1403
static int
udevEnumerateAddMatches(struct udev_enumerate *udev_enumerate)
1404 1405 1406
{
    size_t i;

1407
    for (i = 0; i < G_N_ELEMENTS(subsystem_blacklist); i++) {
1408 1409 1410 1411 1412 1413 1414 1415 1416 1417
        const char *s = subsystem_blacklist[i];
        if (udev_enumerate_add_nomatch_subsystem(udev_enumerate, s) < 0) {
            virReportSystemError(errno, "%s", _("failed to add susbsystem filter"));
            return -1;
        }
    }
    return 0;
}


1418 1419
static int
udevEnumerateDevices(struct udev *udev)
1420 1421 1422
{
    struct udev_enumerate *udev_enumerate = NULL;
    struct udev_list_entry *list_entry = NULL;
1423
    int ret = -1;
1424 1425

    udev_enumerate = udev_enumerate_new(udev);
1426 1427
    if (udevEnumerateAddMatches(udev_enumerate) < 0)
        goto cleanup;
1428

1429 1430
    if (udev_enumerate_scan_devices(udev_enumerate) < 0)
        VIR_WARN("udev scan devices failed");
1431 1432 1433 1434 1435 1436 1437

    udev_list_entry_foreach(list_entry,
                            udev_enumerate_get_list_entry(udev_enumerate)) {

        udevProcessDeviceListEntry(udev, list_entry);
    }

1438
    ret = 0;
1439
 cleanup:
1440 1441 1442 1443 1444
    udev_enumerate_unref(udev_enumerate);
    return ret;
}


1445 1446
static void
udevPCITranslateDeinit(void)
1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457
{
#if defined __s390__ || defined __s390x_
    /* Nothing was initialized, nothing needs to be cleaned up */
#else
    /* pci_system_cleanup returns void */
    pci_system_cleanup();
#endif
    return;
}


1458 1459
static int
nodeStateCleanup(void)
1460
{
1461 1462
    udevEventDataPtr priv = NULL;

J
Ján Tomko 已提交
1463 1464
    if (!driver)
        return -1;
1465

1466 1467 1468 1469 1470 1471 1472 1473 1474 1475
    priv = driver->privateData;
    if (priv) {
        virObjectLock(priv);
        priv->threadQuit = true;
        virCondSignal(&priv->threadCond);
        virObjectUnlock(priv);
        virThreadJoin(&priv->th);
    }

    virObjectUnref(priv);
1476
    virObjectUnref(driver->nodeDeviceEventState);
1477

1478
    virNodeDeviceObjListFree(driver->devs);
1479 1480 1481 1482 1483

    if (driver->lockFD != -1)
        virPidFileRelease(driver->stateDir, "driver", driver->lockFD);

    VIR_FREE(driver->stateDir);
J
Ján Tomko 已提交
1484 1485
    virMutexDestroy(&driver->lock);
    VIR_FREE(driver);
1486

J
Ján Tomko 已提交
1487 1488
    udevPCITranslateDeinit();
    return 0;
1489 1490 1491
}


1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508
static int
udevHandleOneDevice(struct udev_device *device)
{
    const char *action = udev_device_get_action(device);

    VIR_DEBUG("udev action: '%s'", action);

    if (STREQ(action, "add") || STREQ(action, "change"))
        return udevAddOneDevice(device);

    if (STREQ(action, "remove"))
        return udevRemoveOneDevice(device);

    return 0;
}


1509 1510 1511
/* the caller must be holding the udevEventData object lock prior to calling
 * this function
 */
1512
static bool
1513
udevEventMonitorSanityCheck(udevEventDataPtr priv,
1514
                            int fd)
1515
{
1516
    int rc = -1;
1517

1518
    rc = udev_monitor_get_fd(priv->udev_monitor);
1519
    if (fd != rc) {
1520 1521 1522
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("File descriptor returned by udev %d does not "
                         "match node device file descriptor %d"),
1523
                       fd, rc);
1524 1525 1526 1527 1528 1529 1530 1531

        /* this is a non-recoverable error, let's remove the handle, so that we
         * don't get in here again because of some spurious behaviour and report
         * the same error multiple times
         */
        virEventRemoveHandle(priv->watch);
        priv->watch = -1;

1532
        return false;
1533 1534
    }

1535 1536 1537 1538
    return true;
}


1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558
/**
 * udevEventHandleThread
 * @opaque: unused
 *
 * Thread to handle the udevEventHandleCallback processing when udev
 * tells us there's a device change for us (add, modify, delete, etc).
 *
 * Once notified there is data to be processed, the actual @device
 * data retrieval by libudev may be delayed due to how threads are
 * scheduled. In fact, the event loop could be scheduled earlier than
 * the handler thread, thus potentially emitting the very same event
 * the handler thread is currently trying to process, simply because
 * the data hadn't been retrieved from the socket.
 *
 * NB: Some older distros, such as CentOS 6, libudev opens sockets
 * without the NONBLOCK flag which might cause issues with event
 * based algorithm. Although the issue can be mitigated by resetting
 * priv->dataReady for each event found; however, the scheduler issues
 * would still come into play.
 */
1559
static void
J
Ján Tomko 已提交
1560
udevEventHandleThread(void *opaque G_GNUC_UNUSED)
1561
{
1562
    udevEventDataPtr priv = driver->privateData;
1563 1564
    struct udev_device *device = NULL;

1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580
    /* continue rather than break from the loop on non-fatal errors */
    while (1) {
        virObjectLock(priv);
        while (!priv->dataReady && !priv->threadQuit) {
            if (virCondWait(&priv->threadCond, &priv->parent.lock)) {
                virReportSystemError(errno, "%s",
                                     _("handler failed to wait on condition"));
                virObjectUnlock(priv);
                return;
            }
        }

        if (priv->threadQuit) {
            virObjectUnlock(priv);
            return;
        }
1581

1582 1583
        errno = 0;
        device = udev_monitor_receive_device(priv->udev_monitor);
1584 1585
        virObjectUnlock(priv);

1586 1587 1588 1589 1590 1591
        if (!device) {
            if (errno == 0) {
                virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                               _("failed to receive device from udev monitor"));
                return;
            }
1592

1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603
            /* POSIX allows both EAGAIN and EWOULDBLOCK to be used
             * interchangeably when the read would block or timeout was fired
             */
            VIR_WARNINGS_NO_WLOGICALOP_EQUAL_EXPR
            if (errno != EAGAIN && errno != EWOULDBLOCK) {
            VIR_WARNINGS_RESET
                virReportSystemError(errno, "%s",
                                     _("failed to receive device from udev "
                                       "monitor"));
                return;
            }
1604

1605 1606 1607
            /* Trying to move the reset of the @priv->dataReady flag to
             * after the udev_monitor_receive_device wouldn't help much
             * due to event mgmt and scheduler timing. */
1608 1609 1610 1611 1612 1613 1614 1615 1616
            virObjectLock(priv);
            priv->dataReady = false;
            virObjectUnlock(priv);

            continue;
        }

        udevHandleOneDevice(device);
        udev_device_unref(device);
1617 1618 1619 1620 1621

        /* Instead of waiting for the next event after processing @device
         * data, let's keep reading from the udev monitor and only wait
         * for the next event once either a EAGAIN or a EWOULDBLOCK error
         * is encountered. */
1622
    }
1623 1624 1625
}


1626
static void
J
Ján Tomko 已提交
1627
udevEventHandleCallback(int watch G_GNUC_UNUSED,
1628
                        int fd,
J
Ján Tomko 已提交
1629 1630
                        int events G_GNUC_UNUSED,
                        void *data G_GNUC_UNUSED)
1631 1632 1633 1634 1635
{
    udevEventDataPtr priv = driver->privateData;

    virObjectLock(priv);

1636 1637 1638 1639 1640 1641 1642
    if (!udevEventMonitorSanityCheck(priv, fd))
        priv->threadQuit = true;
    else
        priv->dataReady = true;

    virCondSignal(&priv->threadCond);
    virObjectUnlock(priv);
1643 1644 1645
}


1646 1647
/* DMI is intel-compatible specific */
#if defined(__x86_64__) || defined(__i386__) || defined(__amd64__)
1648
static void
1649
udevGetDMIData(virNodeDevCapSystemPtr syscap)
1650
{
1651
    udevEventDataPtr priv = driver->privateData;
1652 1653
    struct udev *udev = NULL;
    struct udev_device *device = NULL;
1654 1655
    virNodeDevCapSystemHardwarePtr hardware = &syscap->hardware;
    virNodeDevCapSystemFirmwarePtr firmware = &syscap->firmware;
1656

1657
    virObjectLock(priv);
1658
    udev = udev_monitor_get_udev(priv->udev_monitor);
1659

1660 1661
    device = udev_device_new_from_syspath(udev, DMI_DEVPATH);
    if (device == NULL) {
1662 1663
        device = udev_device_new_from_syspath(udev, DMI_DEVPATH_FALLBACK);
        if (device == NULL) {
1664 1665 1666
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Failed to get udev device for syspath '%s' or '%s'"),
                           DMI_DEVPATH, DMI_DEVPATH_FALLBACK);
1667
            virObjectUnlock(priv);
1668
            return;
1669
        }
1670
    }
1671
    virObjectUnlock(priv);
1672

1673
    if (udevGetStringSysfsAttr(device, "product_name",
1674
                               &syscap->product_name) < 0)
1675
        goto cleanup;
1676
    if (udevGetStringSysfsAttr(device, "sys_vendor",
1677
                               &hardware->vendor_name) < 0)
1678
        goto cleanup;
1679
    if (udevGetStringSysfsAttr(device, "product_version",
1680
                               &hardware->version) < 0)
1681
        goto cleanup;
1682
    if (udevGetStringSysfsAttr(device, "product_serial",
1683
                               &hardware->serial) < 0)
1684
        goto cleanup;
1685

1686
    if (virGetHostUUID(hardware->uuid))
1687
        goto cleanup;
1688

1689
    if (udevGetStringSysfsAttr(device, "bios_vendor",
1690
                               &firmware->vendor_name) < 0)
1691
        goto cleanup;
1692
    if (udevGetStringSysfsAttr(device, "bios_version",
1693
                               &firmware->version) < 0)
1694
        goto cleanup;
1695
    if (udevGetStringSysfsAttr(device, "bios_date",
1696
                               &firmware->release_date) < 0)
1697
        goto cleanup;
1698

1699
 cleanup:
1700
    if (device != NULL)
1701 1702 1703
        udev_device_unref(device);
    return;
}
1704
#endif
1705 1706


1707 1708
static int
udevSetupSystemDev(void)
1709 1710
{
    virNodeDeviceDefPtr def = NULL;
1711
    virNodeDeviceObjPtr obj = NULL;
1712 1713
    int ret = -1;

1714 1715
    if (VIR_ALLOC(def) < 0)
        return -1;
1716

1717
    def->name = g_strdup("computer");
1718

1719
    if (VIR_ALLOC(def->caps) != 0)
1720
        goto cleanup;
1721

1722
#if defined(__x86_64__) || defined(__i386__) || defined(__amd64__)
1723
    udevGetDMIData(&def->caps->data.system);
1724
#endif
1725

1726
    if (!(obj = virNodeDeviceObjListAssignDef(driver->devs, def)))
1727
        goto cleanup;
1728

1729
    virNodeDeviceObjEndAPI(&obj);
1730 1731 1732

    ret = 0;

1733
 cleanup:
1734
    if (ret == -1)
1735 1736
        virNodeDeviceDefFree(def);

1737 1738 1739
    return ret;
}

1740

1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754
static void
nodeStateInitializeEnumerate(void *opaque)
{
    struct udev *udev = opaque;
    udevEventDataPtr priv = driver->privateData;

    /* Populate with known devices */
    if (udevEnumerateDevices(udev) != 0)
        goto error;

    return;

 error:
    virObjectLock(priv);
1755 1756
    ignore_value(virEventRemoveHandle(priv->watch));
    priv->watch = -1;
1757
    priv->threadQuit = true;
1758
    virCondSignal(&priv->threadCond);
1759 1760 1761 1762
    virObjectUnlock(priv);
}


1763
static int
J
Ján Tomko 已提交
1764
udevPCITranslateInit(bool privileged G_GNUC_UNUSED)
1765
{
1766 1767 1768 1769
#if defined __s390__ || defined __s390x_
    /* On s390(x) system there is no PCI bus.
     * Therefore there is nothing to initialize here. */
#else
1770
    int rc;
1771

1772
    if ((rc = pci_system_init()) != 0) {
1773 1774 1775
        /* Ignore failure as non-root; udev is not as helpful in that
         * situation, but a non-privileged user won't benefit much
         * from udev in the first place.  */
1776
        if (errno != ENOENT && (privileged  || errno != EACCES)) {
1777 1778
            virReportSystemError(rc, "%s",
                                 _("Failed to initialize libpciaccess"));
1779
            return -1;
1780
        }
1781
    }
1782
#endif
1783 1784 1785
    return 0;
}

1786 1787 1788

static int
nodeStateInitialize(bool privileged,
J
Ján Tomko 已提交
1789 1790
                    virStateInhibitCallback callback G_GNUC_UNUSED,
                    void *opaque G_GNUC_UNUSED)
1791
{
1792
    udevEventDataPtr priv = NULL;
1793
    struct udev *udev = NULL;
1794
    virThread enumThread;
1795

1796
    if (VIR_ALLOC(driver) < 0)
1797
        return VIR_DRV_STATE_INIT_ERROR;
1798

1799
    driver->lockFD = -1;
1800
    if (virMutexInit(&driver->lock) < 0) {
1801 1802
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Unable to initialize mutex"));
1803
        VIR_FREE(driver);
1804
        return VIR_DRV_STATE_INIT_ERROR;
1805 1806
    }

1807 1808
    driver->privileged = privileged;

1809
    if (privileged) {
1810
        driver->stateDir = g_strdup_printf("%s/libvirt/nodedev", RUNSTATEDIR);
1811
    } else {
1812
        g_autofree char *rundir = NULL;
1813 1814 1815

        if (!(rundir = virGetUserRuntimeDirectory()))
            goto cleanup;
1816
        driver->stateDir = g_strdup_printf("%s/nodedev/run", rundir);
1817 1818 1819 1820 1821 1822 1823 1824 1825
    }

    if (virFileMakePathWithMode(driver->stateDir, S_IRWXU) < 0) {
        virReportSystemError(errno, _("cannot create state directory '%s'"),
                             driver->stateDir);
        goto cleanup;
    }

    if ((driver->lockFD =
1826
         virPidFileAcquire(driver->stateDir, "driver", false, getpid())) < 0)
1827 1828
        goto cleanup;

1829 1830
    if (!(driver->devs = virNodeDeviceObjListNew()) ||
        !(priv = udevEventDataNew()))
1831
        goto cleanup;
1832

1833
    driver->privateData = priv;
1834
    driver->nodeDeviceEventState = virObjectEventStateNew();
1835

1836
    if (udevPCITranslateInit(privileged) < 0)
1837
        goto cleanup;
1838

1839
    udev = udev_new();
1840 1841 1842
    if (!udev) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("failed to create udev context"));
1843
        goto cleanup;
1844
    }
1845

1846 1847
    virObjectLock(priv);

1848
    priv->udev_monitor = udev_monitor_new_from_netlink(udev, "udev");
1849
    if (!priv->udev_monitor) {
1850 1851
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("udev_monitor_new_from_netlink returned NULL"));
1852
        goto unlock;
1853 1854
    }

1855
    udev_monitor_enable_receiving(priv->udev_monitor);
1856

1857 1858 1859 1860 1861 1862 1863
    /* mimic udevd's behaviour and override the systems rmem_max limit in case
     * there's a significant number of device 'add' events
     */
    if (geteuid() == 0)
        udev_monitor_set_receive_buffer_size(priv->udev_monitor,
                                             128 * 1024 * 1024);

1864 1865 1866 1867 1868 1869
    if (virThreadCreate(&priv->th, true, udevEventHandleThread, NULL) < 0) {
        virReportSystemError(errno, "%s",
                             _("failed to create udev handler thread"));
        goto unlock;
    }

1870 1871 1872 1873 1874 1875 1876 1877
    /* We register the monitor with the event callback so we are
     * notified by udev of device changes before we enumerate existing
     * devices because libvirt will simply recreate the device if we
     * try to register it twice, i.e., if the device appears between
     * the time we register the callback and the time we begin
     * enumeration.  The alternative is to register the callback after
     * we enumerate, in which case we will fail to create any devices
     * that appear while the enumeration is taking place.  */
1878 1879 1880
    priv->watch = virEventAddHandle(udev_monitor_get_fd(priv->udev_monitor),
                                    VIR_EVENT_HANDLE_READABLE,
                                    udevEventHandleCallback, NULL, NULL);
1881
    if (priv->watch == -1)
1882
        goto unlock;
1883

1884 1885
    virObjectUnlock(priv);

1886
    /* Create a fictional 'computer' device to root the device tree. */
1887
    if (udevSetupSystemDev() != 0)
1888
        goto cleanup;
1889

1890 1891 1892 1893
    if (virThreadCreate(&enumThread, false, nodeStateInitializeEnumerate,
                        udev) < 0) {
        virReportSystemError(errno, "%s",
                             _("failed to create udev enumerate thread"));
1894
        goto cleanup;
1895
    }
1896

1897
    return VIR_DRV_STATE_INIT_COMPLETE;
1898

1899
 cleanup:
1900
    nodeStateCleanup();
1901
    return VIR_DRV_STATE_INIT_ERROR;
1902 1903

 unlock:
1904
    virObjectUnlock(priv);
1905
    goto cleanup;
1906 1907 1908
}


1909 1910
static int
nodeStateReload(void)
1911 1912 1913 1914 1915
{
    return 0;
}


1916
static virNodeDeviceDriver udevNodeDeviceDriver = {
1917
    .name = "udev",
1918 1919
    .nodeNumOfDevices = nodeNumOfDevices, /* 0.7.3 */
    .nodeListDevices = nodeListDevices, /* 0.7.3 */
1920
    .connectListAllNodeDevices = nodeConnectListAllNodeDevices, /* 0.10.2 */
1921 1922
    .connectNodeDeviceEventRegisterAny = nodeConnectNodeDeviceEventRegisterAny, /* 2.2.0 */
    .connectNodeDeviceEventDeregisterAny = nodeConnectNodeDeviceEventDeregisterAny, /* 2.2.0 */
1923 1924 1925 1926 1927 1928 1929 1930
    .nodeDeviceLookupByName = nodeDeviceLookupByName, /* 0.7.3 */
    .nodeDeviceLookupSCSIHostByWWN = nodeDeviceLookupSCSIHostByWWN, /* 1.0.2 */
    .nodeDeviceGetXMLDesc = nodeDeviceGetXMLDesc, /* 0.7.3 */
    .nodeDeviceGetParent = nodeDeviceGetParent, /* 0.7.3 */
    .nodeDeviceNumOfCaps = nodeDeviceNumOfCaps, /* 0.7.3 */
    .nodeDeviceListCaps = nodeDeviceListCaps, /* 0.7.3 */
    .nodeDeviceCreateXML = nodeDeviceCreateXML, /* 0.7.3 */
    .nodeDeviceDestroy = nodeDeviceDestroy, /* 0.7.3 */
1931 1932
};

1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944

static virHypervisorDriver udevHypervisorDriver = {
    .name = "nodedev",
    .connectOpen = nodeConnectOpen, /* 4.1.0 */
    .connectClose = nodeConnectClose, /* 4.1.0 */
    .connectIsEncrypted = nodeConnectIsEncrypted, /* 4.1.0 */
    .connectIsSecure = nodeConnectIsSecure, /* 4.1.0 */
    .connectIsAlive = nodeConnectIsAlive, /* 4.1.0 */
};


static virConnectDriver udevConnectDriver = {
1945
    .localOnly = true,
1946
    .uriSchemes = (const char *[]){ "nodedev", NULL },
1947 1948 1949 1950 1951
    .hypervisorDriver = &udevHypervisorDriver,
    .nodeDeviceDriver = &udevNodeDeviceDriver,
};


1952
static virStateDriver udevStateDriver = {
M
Matthias Bolte 已提交
1953
    .name = "udev",
1954 1955 1956
    .stateInitialize = nodeStateInitialize, /* 0.7.3 */
    .stateCleanup = nodeStateCleanup, /* 0.7.3 */
    .stateReload = nodeStateReload, /* 0.7.3 */
1957 1958
};

1959 1960 1961

int
udevNodeRegister(void)
1962
{
1963
    VIR_DEBUG("Registering udev node device backend");
1964

1965 1966
    if (virRegisterConnectDriver(&udevConnectDriver, false) < 0)
        return -1;
1967
    if (virSetSharedNodeDeviceDriver(&udevNodeDeviceDriver) < 0)
1968 1969 1970 1971
        return -1;

    return virRegisterStateDriver(&udevStateDriver);
}