node_device_udev.c 47.1 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 23 24
 *
 * Author: Dave Allan <dallan@redhat.com>
 */

#include <config.h>
#include <libudev.h>
25
#include <pciaccess.h>
26 27 28
#include <scsi/scsi.h>
#include <c-ctype.h>

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

#define VIR_FROM_THIS VIR_FROM_NODEDEV

49 50
VIR_LOG_INIT("node_device.node_device_udev");

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

55 56 57
struct _udevPrivate {
    struct udev_monitor *udev_monitor;
    int watch;
58
    bool privileged;
59 60
};

61

J
Ján Tomko 已提交
62 63 64 65 66 67 68 69 70 71 72
static bool
udevHasDeviceProperty(struct udev_device *dev,
                      const char *key)
{
    if (udev_device_get_property_value(dev, key))
        return true;

    return false;
}


73 74
static const char *udevGetDeviceProperty(struct udev_device *udev_device,
                                         const char *property_key)
75
{
76
    const char *ret = NULL;
77

78
    ret = udev_device_get_property_value(udev_device, property_key);
79

80 81
    VIR_DEBUG("Found property key '%s' value '%s' for device with sysname '%s'",
              property_key, NULLSTR(ret), udev_device_get_sysname(udev_device));
82 83 84 85 86 87 88 89 90

    return ret;
}


static int udevGetStringProperty(struct udev_device *udev_device,
                                 const char *property_key,
                                 char **value)
{
91 92
    if (VIR_STRDUP(*value,
                   udevGetDeviceProperty(udev_device, property_key)) < 0)
93
        return -1;
94

95
    return 0;
96 97 98 99 100 101 102 103
}


static int udevGetIntProperty(struct udev_device *udev_device,
                              const char *property_key,
                              int *value,
                              int base)
{
104
    const char *str = NULL;
105

106
    str = udevGetDeviceProperty(udev_device, property_key);
107

108
    if (str && virStrToLong_i(str, NULL, base, value) < 0) {
109 110
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to int"), str);
111
        return -1;
112
    }
113
    return 0;
114 115 116 117 118 119 120 121
}


static int udevGetUintProperty(struct udev_device *udev_device,
                               const char *property_key,
                               unsigned int *value,
                               int base)
{
122
    const char *str = NULL;
123

124
    str = udevGetDeviceProperty(udev_device, property_key);
125

126
    if (str && virStrToLong_ui(str, NULL, base, value) < 0) {
127 128
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to int"), str);
129
        return -1;
130
    }
131
    return 0;
132 133 134
}


135 136
static const char *udevGetDeviceSysfsAttr(struct udev_device *udev_device,
                                          const char *attr_name)
137
{
138
    const char *ret = NULL;
139

140
    ret = udev_device_get_sysattr_value(udev_device, attr_name);
141 142

    VIR_DEBUG("Found sysfs attribute '%s' value '%s' "
143
              "for device with sysname '%s'",
144
              attr_name, NULLSTR(ret),
145 146 147 148 149 150 151 152 153
              udev_device_get_sysname(udev_device));
    return ret;
}


static int udevGetStringSysfsAttr(struct udev_device *udev_device,
                                  const char *attr_name,
                                  char **value)
{
154
    if (VIR_STRDUP(*value, udevGetDeviceSysfsAttr(udev_device, attr_name)) < 0)
155
        return -1;
156

157
    virStringStripControlChars(*value);
158

159 160
    if (*value != NULL && (STREQ(*value, "")))
        VIR_FREE(*value);
161

162
    return 0;
163 164 165 166 167 168 169 170
}


static int udevGetIntSysfsAttr(struct udev_device *udev_device,
                               const char *attr_name,
                               int *value,
                               int base)
{
171
    const char *str = NULL;
172

173
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
174

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

181
    return 0;
182 183 184 185 186 187 188 189
}


static int udevGetUintSysfsAttr(struct udev_device *udev_device,
                                const char *attr_name,
                                unsigned int *value,
                                int base)
{
190
    const char *str = NULL;
191

192
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
193

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

200
    return 0;
201 202 203 204 205 206 207
}


static int udevGetUint64SysfsAttr(struct udev_device *udev_device,
                                  const char *attr_name,
                                  unsigned long long *value)
{
208
    const char *str = NULL;
209

210
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
211

212
    if (str && virStrToLong_ull(str, NULL, 0, value) < 0) {
213 214
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to convert '%s' to unsigned long long"), str);
J
Ján Tomko 已提交
215
        return -1;
216 217
    }

J
Ján Tomko 已提交
218
    return 0;
219 220 221 222 223 224 225
}


static int udevGenerateDeviceName(struct udev_device *device,
                                  virNodeDeviceDefPtr def,
                                  const char *s)
{
226
    size_t i;
227 228
    virBuffer buf = VIR_BUFFER_INITIALIZER;

229
    virBufferAsprintf(&buf, "%s_%s",
230 231 232
                      udev_device_get_subsystem(device),
                      udev_device_get_sysname(device));

233
    if (s != NULL)
234
        virBufferAsprintf(&buf, "_%s", s);
235

236 237
    if (virBufferCheckError(&buf) < 0)
        return -1;
238 239 240

    def->name = virBufferContentAndReset(&buf);

241
    for (i = 0; i < strlen(def->name); i++) {
242
        if (!(c_isalnum(*(def->name + i))))
243 244 245
            *(def->name + i) = '_';
    }

246
    return 0;
247 248
}

249
#if HAVE_UDEV_LOGGING
J
Ján Tomko 已提交
250 251 252 253 254 255 256 257 258
typedef void (*udevLogFunctionPtr)(struct udev *udev,
                                   int priority,
                                   const char *file,
                                   int line,
                                   const char *fn,
                                   const char *format,
                                   va_list args);

static void
259
ATTRIBUTE_FMT_PRINTF(6, 0)
J
Ján Tomko 已提交
260 261 262 263 264 265 266
udevLogFunction(struct udev *udev ATTRIBUTE_UNUSED,
                int priority,
                const char *file,
                int line,
                const char *fn,
                const char *fmt,
                va_list args)
267
{
J
Ján Tomko 已提交
268
    virBuffer buf = VIR_BUFFER_INITIALIZER;
269
    char *format = NULL;
J
Ján Tomko 已提交
270 271 272 273 274 275

    virBufferAdd(&buf, fmt, -1);
    virBufferTrim(&buf, "\n", -1);

    format = virBufferContentAndReset(&buf);

276
    virLogVMessage(&virLogSelf,
J
Ján Tomko 已提交
277 278 279 280
                   virLogPriorityFromSyslog(priority),
                   file, line, fn, NULL, format ? format : fmt, args);

    VIR_FREE(format);
281
}
282
#endif
283 284


285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
static int udevTranslatePCIIds(unsigned int vendor,
                               unsigned int product,
                               char **vendor_string,
                               char **product_string)
{
    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,
304
                    &vendor_name,
305 306 307
                    NULL,
                    NULL);

308
    if (VIR_STRDUP(*vendor_string, vendor_name) < 0 ||
309
        VIR_STRDUP(*product_string, device_name) < 0)
310
        return -1;
311

312
    return 0;
313 314 315
}


316 317 318
static int udevProcessPCI(struct udev_device *device,
                          virNodeDeviceDefPtr def)
{
319
    virNodeDevCapPCIDevPtr pci_dev = &def->caps->data.pci_dev;
320 321
    virPCIEDeviceInfoPtr pci_express = NULL;
    virPCIDevicePtr pciDev = NULL;
322
    udevPrivate *priv = driver->privateData;
323
    int ret = -1;
324
    char *p;
325

326
    if (udevGetUintProperty(device, "PCI_CLASS", &pci_dev->class, 16) < 0)
327
        goto cleanup;
328

329
    if ((p = strrchr(def->sysfs_path, '/')) == NULL ||
330 331 332 333
        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) {
334 335
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the PCI address from sysfs path: '%s'"),
336
                       def->sysfs_path);
337
        goto cleanup;
338 339
    }

340
    if (udevGetUintSysfsAttr(device, "vendor", &pci_dev->vendor, 16) < 0)
341
        goto cleanup;
342

343
    if (udevGetUintSysfsAttr(device, "device", &pci_dev->product, 16) < 0)
344
        goto cleanup;
345

346 347 348 349
    if (udevTranslatePCIIds(pci_dev->vendor,
                            pci_dev->product,
                            &pci_dev->vendor_name,
                            &pci_dev->product_name) != 0) {
350
        goto cleanup;
351
    }
352

353
    if (udevGenerateDeviceName(device, def, NULL) != 0)
354
        goto cleanup;
355

356 357
    /* The default value is -1, because it can't be 0
     * as zero is valid node number. */
358
    pci_dev->numa_node = -1;
359
    if (udevGetIntSysfsAttr(device, "numa_node",
360
                            &pci_dev->numa_node, 10) < 0)
361
        goto cleanup;
362

363
    if (nodeDeviceSysfsGetPCIRelatedDevCaps(def->sysfs_path, pci_dev) < 0)
364
        goto cleanup;
365

366 367 368 369
    if (!(pciDev = virPCIDeviceNew(pci_dev->domain,
                                   pci_dev->bus,
                                   pci_dev->slot,
                                   pci_dev->function)))
370
        goto cleanup;
371

372
    /* We need to be root to read PCI device configs */
373
    if (priv->privileged) {
374
        if (virPCIGetHeaderType(pciDev, &pci_dev->hdrType) < 0)
375
            goto cleanup;
376

377 378
        if (virPCIDeviceIsPCIExpress(pciDev) > 0) {
            if (VIR_ALLOC(pci_express) < 0)
379
                goto cleanup;
380

381 382 383
            if (virPCIDeviceHasPCIExpressLink(pciDev) > 0) {
                if (VIR_ALLOC(pci_express->link_cap) < 0 ||
                    VIR_ALLOC(pci_express->link_sta) < 0)
384
                    goto cleanup;
385 386 387 388 389 390 391

                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)
392
                    goto cleanup;
393 394 395

                pci_express->link_sta->port = -1; /* PCIe can't negotiate port. Yet :) */
            }
396 397
            pci_dev->flags |= VIR_NODE_DEV_CAP_FLAG_PCIE;
            pci_dev->pci_express = pci_express;
398
            pci_express = NULL;
399 400 401
        }
    }

402 403
    ret = 0;

404
 cleanup:
405
    virPCIDeviceFree(pciDev);
406
    virPCIEDeviceInfoFree(pci_express);
407 408 409
    return ret;
}

M
Marc-André Lureau 已提交
410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429
static int drmGetMinorType(int minor)
{
    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;
    }
}

static int udevProcessDRMDevice(struct udev_device *device,
                                virNodeDeviceDefPtr def)
{
430
    virNodeDevCapDRMPtr drm = &def->caps->data.drm;
M
Marc-André Lureau 已提交
431 432 433 434 435 436 437 438 439 440 441
    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;

442
    drm->type = minor;
M
Marc-André Lureau 已提交
443 444 445

    return 0;
}
446 447 448 449

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

452
    if (udevGetUintProperty(device, "BUSNUM", &usb_dev->bus, 10) < 0)
453
        return -1;
454
    if (udevGetUintProperty(device, "DEVNUM", &usb_dev->device, 10) < 0)
455
        return -1;
456
    if (udevGetUintProperty(device, "ID_VENDOR_ID", &usb_dev->vendor, 16) < 0)
457
        return -1;
458

459 460
    if (udevGetStringProperty(device,
                              "ID_VENDOR_FROM_DATABASE",
461
                              &usb_dev->vendor_name) < 0)
462
        return -1;
463

464
    if (!usb_dev->vendor_name &&
465
        udevGetStringSysfsAttr(device, "manufacturer",
466
                               &usb_dev->vendor_name) < 0)
467
        return -1;
468

469
    if (udevGetUintProperty(device, "ID_MODEL_ID", &usb_dev->product, 16) < 0)
470
        return -1;
471

472 473
    if (udevGetStringProperty(device,
                              "ID_MODEL_FROM_DATABASE",
474
                              &usb_dev->product_name) < 0)
475
        return -1;
476

477
    if (!usb_dev->product_name &&
478
        udevGetStringSysfsAttr(device, "product",
479
                               &usb_dev->product_name) < 0)
480
        return -1;
481

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

485
    return 0;
486 487 488 489 490 491
}


static int udevProcessUSBInterface(struct udev_device *device,
                                   virNodeDeviceDefPtr def)
{
492
    virNodeDevCapUSBIfPtr usb_if = &def->caps->data.usb_if;
493

494
    if (udevGetUintSysfsAttr(device, "bInterfaceNumber",
495
                             &usb_if->number, 16) < 0)
496
        return -1;
497

498
    if (udevGetUintSysfsAttr(device, "bInterfaceClass",
499
                             &usb_if->_class, 16) < 0)
500
        return -1;
501

502
    if (udevGetUintSysfsAttr(device, "bInterfaceSubClass",
503
                             &usb_if->subclass, 16) < 0)
504
        return -1;
505

506
    if (udevGetUintSysfsAttr(device, "bInterfaceProtocol",
507
                             &usb_if->protocol, 16) < 0)
508
        return -1;
509

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

513
    return 0;
514 515 516 517 518 519
}


static int udevProcessNetworkInterface(struct udev_device *device,
                                       virNodeDeviceDefPtr def)
{
D
David Allan 已提交
520
    const char *devtype = udev_device_get_devtype(device);
521
    virNodeDevCapNetPtr net = &def->caps->data.net;
522

D
David Allan 已提交
523
    if (devtype && STREQ(devtype, "wlan")) {
524
        net->subtype = VIR_NODE_DEV_CAP_NET_80211;
D
David Allan 已提交
525
    } else {
526
        net->subtype = VIR_NODE_DEV_CAP_NET_80203;
D
David Allan 已提交
527 528
    }

529 530
    if (udevGetStringProperty(device,
                              "INTERFACE",
531
                              &net->ifname) < 0)
532
        return -1;
533

534
    if (udevGetStringSysfsAttr(device, "address",
535
                               &net->address) < 0)
536
        return -1;
537

538
    if (udevGetUintSysfsAttr(device, "addr_len", &net->address_len, 0) < 0)
539
        return -1;
540

541
    if (udevGenerateDeviceName(device, def, net->address) != 0)
542
        return -1;
543

544
    if (virNetDevGetLinkInfo(net->ifname, &net->lnk) < 0)
545
        return -1;
546

547
    if (virNetDevGetFeatures(net->ifname, &net->features) < 0)
548
        return -1;
549

550
    return 0;
551 552 553 554 555 556
}


static int udevProcessSCSIHost(struct udev_device *device ATTRIBUTE_UNUSED,
                               virNodeDeviceDefPtr def)
{
557
    virNodeDevCapSCSIHostPtr scsi_host = &def->caps->data.scsi_host;
558
    char *filename = NULL;
J
Ján Tomko 已提交
559
    char *str;
560

561
    filename = last_component(def->sysfs_path);
562

563
    if (!(str = STRSKIP(filename, "host")) ||
564
        virStrToLong_ui(str, NULL, 0, &scsi_host->host) < 0) {
565 566 567
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse SCSI host '%s'"),
                       filename);
568
        return -1;
569 570
    }

571
    nodeDeviceSysfsGetSCSIHostCaps(&def->caps->data.scsi_host);
572

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

576
    return 0;
577 578 579
}


D
David Allan 已提交
580 581 582 583
static int udevProcessSCSITarget(struct udev_device *device ATTRIBUTE_UNUSED,
                                 virNodeDeviceDefPtr def)
{
    const char *sysname = NULL;
584
    virNodeDevCapSCSITargetPtr scsi_target = &def->caps->data.scsi_target;
D
David Allan 已提交
585 586 587

    sysname = udev_device_get_sysname(device);

588
    if (VIR_STRDUP(scsi_target->name, sysname) < 0)
589
        return -1;
D
David Allan 已提交
590

591
    if (udevGenerateDeviceName(device, def, NULL) != 0)
592
        return -1;
D
David Allan 已提交
593

594
    return 0;
D
David Allan 已提交
595 596 597
}


598
static int udevGetSCSIType(virNodeDeviceDefPtr def ATTRIBUTE_UNUSED,
599
                           unsigned int type, char **typestring)
600 601 602 603 604 605 606 607
{
    int ret = 0;
    int foundtype = 1;

    *typestring = NULL;

    switch (type) {
    case TYPE_DISK:
608
        ignore_value(VIR_STRDUP(*typestring, "disk"));
609 610
        break;
    case TYPE_TAPE:
611
        ignore_value(VIR_STRDUP(*typestring, "tape"));
612 613
        break;
    case TYPE_PROCESSOR:
614
        ignore_value(VIR_STRDUP(*typestring, "processor"));
615 616
        break;
    case TYPE_WORM:
617
        ignore_value(VIR_STRDUP(*typestring, "worm"));
618 619
        break;
    case TYPE_ROM:
620
        ignore_value(VIR_STRDUP(*typestring, "cdrom"));
621 622
        break;
    case TYPE_SCANNER:
623
        ignore_value(VIR_STRDUP(*typestring, "scanner"));
624 625
        break;
    case TYPE_MOD:
626
        ignore_value(VIR_STRDUP(*typestring, "mod"));
627 628
        break;
    case TYPE_MEDIUM_CHANGER:
629
        ignore_value(VIR_STRDUP(*typestring, "changer"));
630 631
        break;
    case TYPE_ENCLOSURE:
632
        ignore_value(VIR_STRDUP(*typestring, "enclosure"));
633
        break;
634
    case TYPE_RAID:
635
        ignore_value(VIR_STRDUP(*typestring, "raid"));
636
        break;
637 638 639 640 641 642 643 644 645 646
    case TYPE_NO_LUN:
    default:
        foundtype = 0;
        break;
    }

    if (*typestring == NULL) {
        if (foundtype == 1) {
            ret = -1;
        } else {
647 648
            VIR_DEBUG("Failed to find SCSI device type %d for %s",
                      type, def->sysfs_path);
649 650 651 652 653 654 655 656 657 658 659 660
        }
    }

    return ret;
}


static int udevProcessSCSIDevice(struct udev_device *device ATTRIBUTE_UNUSED,
                                 virNodeDeviceDefPtr def)
{
    int ret = -1;
    unsigned int tmp = 0;
661
    virNodeDevCapSCSIPtr scsi = &def->caps->data.scsi;
662 663
    char *filename = NULL, *p = NULL;

664
    filename = last_component(def->sysfs_path);
665

666 667 668 669
    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) {
670 671 672 673
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the SCSI address from filename: '%s'"),
                       filename);
        return -1;
674 675
    }

676 677
    if (udev_device_get_sysattr_value(device, "type")) {
        if (udevGetUintSysfsAttr(device, "type", &tmp, 0) < 0)
678
            goto cleanup;
679

680
        if (udevGetSCSIType(def, tmp, &scsi->type) < 0)
681
            goto cleanup;
682 683
    }

684
    if (udevGenerateDeviceName(device, def, NULL) != 0)
685
        goto cleanup;
686 687 688

    ret = 0;

689
 cleanup:
690
    if (ret != 0) {
691 692 693
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Failed to process SCSI device with sysfs path '%s'"),
                       def->sysfs_path);
694 695 696 697 698 699 700 701
    }
    return ret;
}


static int udevProcessDisk(struct udev_device *device,
                           virNodeDeviceDefPtr def)
{
702
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
703

704
    if (udevGetUint64SysfsAttr(device, "size", &storage->num_blocks) < 0)
705
        return -1;
706

J
Ján Tomko 已提交
707
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
708
                               &storage->logical_block_size) < 0)
709
        return -1;
710

711
    storage->size = storage->num_blocks * storage->logical_block_size;
712

713
    return 0;
714 715 716
}


717 718 719
static int udevProcessRemoveableMedia(struct udev_device *device,
                                      virNodeDeviceDefPtr def,
                                      int has_media)
720
{
721
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
J
Ján Tomko 已提交
722
    int is_removable = 0;
723

724 725 726
    if (udevGetIntSysfsAttr(device, "removable", &is_removable, 0) < 0)
        return -1;
    if (is_removable == 1)
727 728
        def->caps->data.storage.flags |= VIR_NODE_DEV_CAP_STORAGE_REMOVABLE;

J
Ján Tomko 已提交
729 730
    if (!has_media)
        return 0;
731

J
Ján Tomko 已提交
732 733
    def->caps->data.storage.flags |=
        VIR_NODE_DEV_CAP_STORAGE_REMOVABLE_MEDIA_AVAILABLE;
734

J
Ján Tomko 已提交
735
    if (udevGetStringProperty(device, "ID_FS_LABEL",
736
                              &storage->media_label) < 0)
J
Ján Tomko 已提交
737
        return -1;
738

J
Ján Tomko 已提交
739
    if (udevGetUint64SysfsAttr(device, "size",
740
                               &storage->num_blocks) < 0)
J
Ján Tomko 已提交
741
        return -1;
742

J
Ján Tomko 已提交
743
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
744
                               &storage->logical_block_size) < 0)
J
Ján Tomko 已提交
745
        return -1;
746

J
Ján Tomko 已提交
747 748 749 750 751 752 753
    /* 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;
754

J
Ján Tomko 已提交
755
    return 0;
756 757
}

758 759 760 761 762 763 764 765 766 767
static int udevProcessCDROM(struct udev_device *device,
                            virNodeDeviceDefPtr def)
{
    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);
768
    if (VIR_STRDUP(def->caps->data.storage.drive_type, "cdrom") < 0)
769
        return -1;
770

771 772
    if (udevHasDeviceProperty(device, "ID_CDROM_MEDIA") &&
        udevGetIntProperty(device, "ID_CDROM_MEDIA", &has_media, 0) < 0)
773
        return -1;
774

775
    return udevProcessRemoveableMedia(device, def, has_media);
776 777 778 779 780 781 782
}

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

783
    if (udevHasDeviceProperty(device, "ID_CDROM_MEDIA")) {
784
        /* USB floppy */
785 786
        if (udevGetIntProperty(device, "DKD_MEDIA_AVAILABLE", &has_media, 0) < 0)
            return -1;
787
    } else if (udevHasDeviceProperty(device, "ID_FS_LABEL")) {
788 789 790 791 792 793
        /* Legacy floppy */
        has_media = 1;
    }

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

795 796 797 798

static int udevProcessSD(struct udev_device *device,
                         virNodeDeviceDefPtr def)
{
799
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
800

J
Ján Tomko 已提交
801
    if (udevGetUint64SysfsAttr(device, "size",
802
                               &storage->num_blocks) < 0)
803
        return -1;
804

J
Ján Tomko 已提交
805
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
806
                               &storage->logical_block_size) < 0)
807
        return -1;
808

809
    storage->size = storage->num_blocks * storage->logical_block_size;
810

811
    return 0;
812 813 814 815
}



816 817 818 819 820 821
/* 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. */
static int udevKludgeStorageType(virNodeDeviceDefPtr def)
{
822 823 824
    VIR_DEBUG("Could not find definitive storage type for device "
              "with sysfs path '%s', trying to guess it",
              def->sysfs_path);
825

826 827 828
    /* virtio disk */
    if (STRPREFIX(def->caps->data.storage.block, "/dev/vd") &&
        VIR_STRDUP(def->caps->data.storage.drive_type, "disk") > 0) {
829
        VIR_DEBUG("Found storage type '%s' for device "
830
                  "with sysfs path '%s'",
831 832
                  def->caps->data.storage.drive_type,
                  def->sysfs_path);
833
        return 0;
834
    }
835 836 837
    VIR_DEBUG("Could not determine storage type "
              "for device with sysfs path '%s'", def->sysfs_path);
    return -1;
838 839 840 841 842 843
}


static int udevProcessStorage(struct udev_device *device,
                              virNodeDeviceDefPtr def)
{
844
    virNodeDevCapStoragePtr storage = &def->caps->data.storage;
845
    int ret = -1;
846
    const char* devnode;
847

848
    devnode = udev_device_get_devnode(device);
849
    if (!devnode) {
850
        VIR_DEBUG("No devnode for '%s'", udev_device_get_devpath(device));
851
        goto cleanup;
852
    }
853

854
    if (VIR_STRDUP(storage->block, devnode) < 0)
855
        goto cleanup;
856

857
    if (udevGetStringProperty(device, "ID_BUS", &storage->bus) < 0)
858
        goto cleanup;
859
    if (udevGetStringProperty(device, "ID_SERIAL", &storage->serial) < 0)
860
        goto cleanup;
861

862
    if (udevGetStringSysfsAttr(device, "device/vendor", &storage->vendor) < 0)
863
        goto cleanup;
864 865 866
    if (def->caps->data.storage.vendor)
        virTrimSpaces(def->caps->data.storage.vendor, NULL);

867
    if (udevGetStringSysfsAttr(device, "device/model", &storage->model) < 0)
868
        goto cleanup;
869 870
    if (def->caps->data.storage.model)
        virTrimSpaces(def->caps->data.storage.model, NULL);
871 872 873 874 875
    /* 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. */

876
    if (udevGetStringProperty(device, "ID_TYPE", &storage->drive_type) < 0)
877
        goto cleanup;
878

879
    if (!storage->drive_type ||
880
        STREQ(def->caps->data.storage.drive_type, "generic")) {
881 882
        int val = 0;
        const char *str = NULL;
883 884 885

        /* All floppy drives have the ID_DRIVE_FLOPPY prop. This is
         * needed since legacy floppies don't have a drive_type */
886
        if (udevGetIntProperty(device, "ID_DRIVE_FLOPPY", &val, 0) < 0)
887
            goto cleanup;
888 889
        else if (val == 1)
            str = "floppy";
890

891
        if (!str) {
892
            if (udevGetIntProperty(device, "ID_CDROM", &val, 0) < 0)
893
                goto cleanup;
894 895 896
            else if (val == 1)
                str = "cd";
        }
897

898
        if (!str) {
899
            if (udevGetIntProperty(device, "ID_DRIVE_FLASH_SD", &val, 0) < 0)
900
                goto cleanup;
901 902 903
            if (val == 1)
                str = "sd";
        }
904

905
        if (str) {
906
            if (VIR_STRDUP(storage->drive_type, str) < 0)
907
                goto cleanup;
908 909
        } else {
            /* If udev doesn't have it, perhaps we can guess it. */
910
            if (udevKludgeStorageType(def) != 0)
911
                goto cleanup;
912 913 914 915 916 917 918
        }
    }

    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);
919 920
    } else if (STREQ(def->caps->data.storage.drive_type, "floppy")) {
        ret = udevProcessFloppy(device, def);
921 922
    } else if (STREQ(def->caps->data.storage.drive_type, "sd")) {
        ret = udevProcessSD(device, def);
923
    } else {
924 925
        VIR_DEBUG("Unsupported storage type '%s'",
                  def->caps->data.storage.drive_type);
926
        goto cleanup;
927 928
    }

929
    if (udevGenerateDeviceName(device, def, storage->serial) != 0)
930
        goto cleanup;
931

932
 cleanup:
933
    VIR_DEBUG("Storage ret=%d", ret);
934 935 936
    return ret;
}

937
static int
938
udevProcessSCSIGeneric(struct udev_device *dev,
939 940
                       virNodeDeviceDefPtr def)
{
941 942
    if (udevGetStringProperty(dev, "DEVNAME", &def->caps->data.sg.path) < 0 ||
        !def->caps->data.sg.path)
943 944 945 946 947 948 949 950
        return -1;

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

    return 0;
}

951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978
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);

    if (VIR_STRDUP(def->devnode, devnode) < 0)
        return -1;

    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)) {
        if (VIR_STRDUP(def->devlinks[n++], udev_list_entry_get_name(list_entry)) < 0)
            return -1;
    }

    return 0;
}

979 980
static int
udevGetDeviceType(struct udev_device *device,
981
                  virNodeDevCapType *type)
982 983
{
    const char *devtype = NULL;
984
    char *subsystem = NULL;
985
    int ret = -1;
D
David Allan 已提交
986

987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004
    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 已提交
1005 1006
        else if (STREQ(devtype, "drm_minor"))
            *type = VIR_NODE_DEV_CAP_DRM;
1007 1008
    } else {
        /* PCI devices don't set the DEVTYPE property. */
1009
        if (udevHasDeviceProperty(device, "PCI_CLASS"))
1010
            *type = VIR_NODE_DEV_CAP_PCI_DEV;
1011

1012 1013 1014 1015
        /* 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. */
1016
        if (udevHasDeviceProperty(device, "INTERFACE"))
1017
            *type = VIR_NODE_DEV_CAP_NET;
1018 1019

        /* SCSI generic device doesn't set DEVTYPE property */
1020 1021 1022 1023
        if (udevGetStringProperty(device, "SUBSYSTEM", &subsystem) < 0)
            return -1;

        if (STREQ_NULLABLE(subsystem, "scsi_generic"))
1024 1025
            *type = VIR_NODE_DEV_CAP_SCSI_GENERIC;
        VIR_FREE(subsystem);
1026 1027
    }

1028 1029 1030 1031 1032 1033
    if (!*type)
        VIR_DEBUG("Could not determine device type for device "
                  "with sysfs name '%s'",
                  udev_device_get_sysname(device));
    else
        ret = 0;
1034 1035 1036 1037 1038 1039 1040 1041

    return ret;
}


static int udevGetDeviceDetails(struct udev_device *device,
                                virNodeDeviceDefPtr def)
{
1042
    switch (def->caps->data.type) {
1043
    case VIR_NODE_DEV_CAP_PCI_DEV:
1044
        return udevProcessPCI(device, def);
1045
    case VIR_NODE_DEV_CAP_USB_DEV:
1046
        return udevProcessUSBDevice(device, def);
1047
    case VIR_NODE_DEV_CAP_USB_INTERFACE:
1048
        return udevProcessUSBInterface(device, def);
1049
    case VIR_NODE_DEV_CAP_NET:
1050
        return udevProcessNetworkInterface(device, def);
1051
    case VIR_NODE_DEV_CAP_SCSI_HOST:
1052
        return udevProcessSCSIHost(device, def);
D
David Allan 已提交
1053
    case VIR_NODE_DEV_CAP_SCSI_TARGET:
1054
        return udevProcessSCSITarget(device, def);
1055
    case VIR_NODE_DEV_CAP_SCSI:
1056
        return udevProcessSCSIDevice(device, def);
1057
    case VIR_NODE_DEV_CAP_STORAGE:
1058
        return udevProcessStorage(device, def);
1059
    case VIR_NODE_DEV_CAP_SCSI_GENERIC:
1060
        return udevProcessSCSIGeneric(device, def);
M
Marc-André Lureau 已提交
1061
    case VIR_NODE_DEV_CAP_DRM:
1062 1063 1064 1065 1066
        return udevProcessDRMDevice(device, def);
    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:
1067 1068 1069
        break;
    }

1070
    return 0;
1071 1072 1073 1074 1075 1076
}


static int udevRemoveOneDevice(struct udev_device *device)
{
    virNodeDeviceObjPtr dev = NULL;
1077
    virObjectEventPtr event = NULL;
1078
    const char *name = NULL;
1079
    int ret = -1;
1080 1081

    name = udev_device_get_syspath(device);
1082
    dev = virNodeDeviceObjFindBySysfsPath(&driver->devs, name);
1083

1084
    if (!dev) {
1085 1086
        VIR_DEBUG("Failed to find device to remove that has udev name '%s'",
                  name);
1087
        goto cleanup;
1088 1089
    }

1090 1091 1092 1093 1094 1095
    event = virNodeDeviceEventLifecycleNew(dev->def->name,
                                           VIR_NODE_DEVICE_EVENT_DELETED,
                                           0);

    VIR_DEBUG("Removing device '%s' with sysfs path '%s'",
              dev->def->name, name);
1096
    virNodeDeviceObjRemove(&driver->devs, &dev);
1097 1098 1099 1100 1101

    ret = 0;
 cleanup:
    if (event)
        virObjectEventStateQueue(driver->nodeDeviceEventState, event);
1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
    return ret;
}


static int udevSetParent(struct udev_device *device,
                         virNodeDeviceDefPtr def)
{
    struct udev_device *parent_device = NULL;
    const char *parent_sysfs_path = NULL;
    virNodeDeviceObjPtr dev = NULL;
    int ret = -1;

1114 1115
    parent_device = device;
    do {
1116

1117
        parent_device = udev_device_get_parent(parent_device);
1118
        if (parent_device == NULL)
1119
            break;
1120

1121 1122
        parent_sysfs_path = udev_device_get_syspath(parent_device);
        if (parent_sysfs_path == NULL) {
1123 1124 1125
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Could not get syspath for parent of '%s'"),
                           udev_device_get_syspath(parent_device));
1126
            goto cleanup;
1127 1128
        }

1129 1130
        dev = virNodeDeviceObjFindBySysfsPath(&driver->devs,
                                              parent_sysfs_path);
1131
        if (dev != NULL) {
1132 1133
            if (VIR_STRDUP(def->parent, dev->def->name) < 0) {
                virNodeDeviceObjUnlock(dev);
1134
                goto cleanup;
1135
            }
1136
            virNodeDeviceObjUnlock(dev);
1137

1138
            if (VIR_STRDUP(def->parent_sysfs_path, parent_sysfs_path) < 0)
1139
                goto cleanup;
1140 1141 1142 1143
        }

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

1144
    if (!def->parent && VIR_STRDUP(def->parent, "computer") < 0)
1145
        goto cleanup;
1146 1147 1148

    ret = 0;

1149
 cleanup:
1150 1151 1152 1153 1154 1155 1156 1157
    return ret;
}


static int udevAddOneDevice(struct udev_device *device)
{
    virNodeDeviceDefPtr def = NULL;
    virNodeDeviceObjPtr dev = NULL;
1158 1159
    virObjectEventPtr event = NULL;
    bool new_device = true;
1160 1161
    int ret = -1;

1162
    if (VIR_ALLOC(def) != 0)
1163
        goto cleanup;
1164

1165
    if (VIR_STRDUP(def->sysfs_path, udev_device_get_syspath(device)) < 0)
1166
        goto cleanup;
1167

1168
    if (udevGetStringProperty(device, "DRIVER", &def->driver) < 0)
1169
        goto cleanup;
1170

1171
    if (VIR_ALLOC(def->caps) != 0)
1172
        goto cleanup;
1173

1174
    if (udevGetDeviceType(device, &def->caps->data.type) != 0)
1175
        goto cleanup;
1176

1177 1178 1179
    if (udevGetDeviceNodes(device, def) != 0)
        goto cleanup;

1180
    if (udevGetDeviceDetails(device, def) != 0)
1181
        goto cleanup;
1182

1183
    if (udevSetParent(device, def) != 0)
1184
        goto cleanup;
1185

1186
    dev = virNodeDeviceObjFindByName(&driver->devs, def->name);
1187 1188 1189 1190 1191
    if (dev) {
        virNodeDeviceObjUnlock(dev);
        new_device = false;
    }

1192 1193
    /* If this is a device change, the old definition will be freed
     * and the current definition will take its place. */
1194
    dev = virNodeDeviceObjAssignDef(&driver->devs, def);
1195
    if (dev == NULL)
1196
        goto cleanup;
1197

1198 1199 1200 1201
    if (new_device)
        event = virNodeDeviceEventLifecycleNew(dev->def->name,
                                               VIR_NODE_DEVICE_EVENT_CREATED,
                                               0);
1202 1203
    else
        event = virNodeDeviceEventUpdateNew(dev->def->name);
1204

1205 1206 1207 1208
    virNodeDeviceObjUnlock(dev);

    ret = 0;

1209
 cleanup:
1210 1211 1212
    if (event)
        virObjectEventStateQueue(driver->nodeDeviceEventState, event);

1213
    if (ret != 0) {
1214
        VIR_DEBUG("Discarding device %d %p %s", ret, def,
1215
                  def ? NULLSTR(def->sysfs_path) : "");
1216 1217 1218
        virNodeDeviceDefFree(def);
    }

1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232
    return ret;
}


static int udevProcessDeviceListEntry(struct udev *udev,
                                      struct udev_list_entry *list_entry)
{
    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);
1233

1234 1235
    if (device != NULL) {
        if (udevAddOneDevice(device) != 0) {
1236 1237
            VIR_DEBUG("Failed to create node device for udev device '%s'",
                      name);
1238 1239 1240 1241
        }
        ret = 0;
    }

1242 1243
    udev_device_unref(device);

1244 1245 1246 1247
    return ret;
}


1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270
/* 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",
};

static int udevEnumerateAddMatches(struct udev_enumerate *udev_enumerate)
{
    size_t i;

    for (i = 0; i < ARRAY_CARDINALITY(subsystem_blacklist); i++) {
        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;
}


1271 1272 1273 1274
static int udevEnumerateDevices(struct udev *udev)
{
    struct udev_enumerate *udev_enumerate = NULL;
    struct udev_list_entry *list_entry = NULL;
1275
    int ret = -1;
1276 1277

    udev_enumerate = udev_enumerate_new(udev);
1278 1279
    if (udevEnumerateAddMatches(udev_enumerate) < 0)
        goto cleanup;
1280 1281

    ret = udev_enumerate_scan_devices(udev_enumerate);
1282
    if (ret != 0) {
1283 1284 1285
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("udev scan devices returned %d"),
                       ret);
1286
        goto cleanup;
1287 1288 1289 1290 1291 1292 1293 1294
    }

    udev_list_entry_foreach(list_entry,
                            udev_enumerate_get_list_entry(udev_enumerate)) {

        udevProcessDeviceListEntry(udev, list_entry);
    }

1295
 cleanup:
1296 1297 1298 1299 1300
    udev_enumerate_unref(udev_enumerate);
    return ret;
}


1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312
static void udevPCITranslateDeinit(void)
{
#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;
}


1313
static int nodeStateCleanup(void)
1314
{
1315
    udevPrivate *priv = NULL;
1316 1317 1318
    struct udev_monitor *udev_monitor = NULL;
    struct udev *udev = NULL;

J
Ján Tomko 已提交
1319 1320
    if (!driver)
        return -1;
1321

J
Ján Tomko 已提交
1322
    nodeDeviceLock();
1323

1324
    virObjectUnref(driver->nodeDeviceEventState);
1325

J
Ján Tomko 已提交
1326
    priv = driver->privateData;
1327

1328 1329 1330
    if (priv) {
        if (priv->watch != -1)
            virEventRemoveHandle(priv->watch);
1331

1332
        udev_monitor = DRV_STATE_UDEV_MONITOR(driver);
1333

1334 1335 1336 1337
        if (udev_monitor != NULL) {
            udev = udev_monitor_get_udev(udev_monitor);
            udev_monitor_unref(udev_monitor);
        }
J
Ján Tomko 已提交
1338
    }
1339

J
Ján Tomko 已提交
1340 1341
    if (udev != NULL)
        udev_unref(udev);
1342

J
Ján Tomko 已提交
1343 1344 1345 1346 1347
    virNodeDeviceObjListFree(&driver->devs);
    nodeDeviceUnlock();
    virMutexDestroy(&driver->lock);
    VIR_FREE(driver);
    VIR_FREE(priv);
1348

J
Ján Tomko 已提交
1349 1350
    udevPCITranslateDeinit();
    return 0;
1351 1352 1353 1354 1355 1356 1357 1358 1359
}


static void udevEventHandleCallback(int watch ATTRIBUTE_UNUSED,
                                    int fd,
                                    int events ATTRIBUTE_UNUSED,
                                    void *data ATTRIBUTE_UNUSED)
{
    struct udev_device *device = NULL;
1360
    struct udev_monitor *udev_monitor = DRV_STATE_UDEV_MONITOR(driver);
1361 1362 1363
    const char *action = NULL;
    int udev_fd = -1;

1364
    nodeDeviceLock();
1365 1366
    udev_fd = udev_monitor_get_fd(udev_monitor);
    if (fd != udev_fd) {
1367 1368 1369 1370
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("File descriptor returned by udev %d does not "
                         "match node device file descriptor %d"),
                       fd, udev_fd);
1371
        goto cleanup;
1372 1373 1374 1375
    }

    device = udev_monitor_receive_device(udev_monitor);
    if (device == NULL) {
1376 1377
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("udev_monitor_receive_device returned NULL"));
1378
        goto cleanup;
1379 1380 1381
    }

    action = udev_device_get_action(device);
1382
    VIR_DEBUG("udev action: '%s'", action);
1383 1384 1385

    if (STREQ(action, "add") || STREQ(action, "change")) {
        udevAddOneDevice(device);
1386
        goto cleanup;
1387 1388 1389 1390
    }

    if (STREQ(action, "remove")) {
        udevRemoveOneDevice(device);
1391
        goto cleanup;
1392 1393
    }

1394
 cleanup:
1395
    udev_device_unref(device);
1396
    nodeDeviceUnlock();
1397 1398 1399 1400
    return;
}


1401 1402
/* DMI is intel-compatible specific */
#if defined(__x86_64__) || defined(__i386__) || defined(__amd64__)
1403
static void
1404
udevGetDMIData(virNodeDevCapSystemPtr syscap)
1405 1406 1407
{
    struct udev *udev = NULL;
    struct udev_device *device = NULL;
1408 1409
    virNodeDevCapSystemHardwarePtr hardware = &syscap->hardware;
    virNodeDevCapSystemFirmwarePtr firmware = &syscap->firmware;
1410

1411
    udev = udev_monitor_get_udev(DRV_STATE_UDEV_MONITOR(driver));
1412

1413 1414
    device = udev_device_new_from_syspath(udev, DMI_DEVPATH);
    if (device == NULL) {
1415 1416
        device = udev_device_new_from_syspath(udev, DMI_DEVPATH_FALLBACK);
        if (device == NULL) {
1417 1418 1419
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Failed to get udev device for syspath '%s' or '%s'"),
                           DMI_DEVPATH, DMI_DEVPATH_FALLBACK);
1420
            return;
1421
        }
1422 1423
    }

1424
    if (udevGetStringSysfsAttr(device, "product_name",
1425
                               &syscap->product_name) < 0)
1426
        goto cleanup;
1427
    if (udevGetStringSysfsAttr(device, "sys_vendor",
1428
                               &hardware->vendor_name) < 0)
1429
        goto cleanup;
1430
    if (udevGetStringSysfsAttr(device, "product_version",
1431
                               &hardware->version) < 0)
1432
        goto cleanup;
1433
    if (udevGetStringSysfsAttr(device, "product_serial",
1434
                               &hardware->serial) < 0)
1435
        goto cleanup;
1436

1437
    if (virGetHostUUID(hardware->uuid))
1438
        goto cleanup;
1439

1440
    if (udevGetStringSysfsAttr(device, "bios_vendor",
1441
                               &firmware->vendor_name) < 0)
1442
        goto cleanup;
1443
    if (udevGetStringSysfsAttr(device, "bios_version",
1444
                               &firmware->version) < 0)
1445
        goto cleanup;
1446
    if (udevGetStringSysfsAttr(device, "bios_date",
1447
                               &firmware->release_date) < 0)
1448
        goto cleanup;
1449

1450
 cleanup:
1451
    if (device != NULL)
1452 1453 1454
        udev_device_unref(device);
    return;
}
1455
#endif
1456 1457 1458 1459 1460 1461 1462 1463


static int udevSetupSystemDev(void)
{
    virNodeDeviceDefPtr def = NULL;
    virNodeDeviceObjPtr dev = NULL;
    int ret = -1;

1464 1465
    if (VIR_ALLOC(def) < 0)
        return -1;
1466

1467
    if (VIR_STRDUP(def->name, "computer") < 0)
1468
        goto cleanup;
1469

1470
    if (VIR_ALLOC(def->caps) != 0)
1471
        goto cleanup;
1472

1473
#if defined(__x86_64__) || defined(__i386__) || defined(__amd64__)
1474
    udevGetDMIData(&def->caps->data.system);
1475
#endif
1476

1477
    dev = virNodeDeviceObjAssignDef(&driver->devs, def);
1478
    if (dev == NULL)
1479
        goto cleanup;
1480 1481 1482 1483 1484

    virNodeDeviceObjUnlock(dev);

    ret = 0;

1485
 cleanup:
1486
    if (ret == -1)
1487 1488
        virNodeDeviceDefFree(def);

1489 1490 1491
    return ret;
}

1492
static int udevPCITranslateInit(bool privileged ATTRIBUTE_UNUSED)
1493
{
1494 1495 1496 1497
#if defined __s390__ || defined __s390x_
    /* On s390(x) system there is no PCI bus.
     * Therefore there is nothing to initialize here. */
#else
1498
    int rc;
1499

1500
    if ((rc = pci_system_init()) != 0) {
1501 1502 1503
        /* 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.  */
1504
        if (errno != ENOENT && (privileged  || errno != EACCES)) {
1505 1506
            virReportSystemError(rc, "%s",
                                 _("Failed to initialize libpciaccess"));
1507
            return -1;
1508
        }
1509
    }
1510
#endif
1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521
    return 0;
}

static int nodeStateInitialize(bool privileged,
                               virStateInhibitCallback callback ATTRIBUTE_UNUSED,
                               void *opaque ATTRIBUTE_UNUSED)
{
    udevPrivate *priv = NULL;
    struct udev *udev = NULL;
    int ret = -1;

1522
    if (VIR_ALLOC(priv) < 0)
1523
        return -1;
1524 1525

    priv->watch = -1;
1526
    priv->privileged = privileged;
1527

1528
    if (VIR_ALLOC(driver) < 0) {
1529
        VIR_FREE(priv);
1530
        return -1;
1531 1532
    }

1533
    if (virMutexInit(&driver->lock) < 0) {
1534 1535
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Unable to initialize mutex"));
1536
        VIR_FREE(priv);
1537
        VIR_FREE(driver);
1538
        return -1;
1539 1540
    }

1541
    driver->privateData = priv;
1542
    nodeDeviceLock();
1543
    driver->nodeDeviceEventState = virObjectEventStateNew();
1544

1545
    if (udevPCITranslateInit(privileged) < 0)
1546
        goto cleanup;
1547

1548
    udev = udev_new();
1549 1550 1551 1552 1553
    if (!udev) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("failed to create udev context"));
        goto cleanup;
    }
1554
#if HAVE_UDEV_LOGGING
J
Ján Tomko 已提交
1555 1556
    /* cast to get rid of missing-format-attribute warning */
    udev_set_log_fn(udev, (udevLogFunctionPtr) udevLogFunction);
1557
#endif
1558

1559 1560
    priv->udev_monitor = udev_monitor_new_from_netlink(udev, "udev");
    if (priv->udev_monitor == NULL) {
1561 1562
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("udev_monitor_new_from_netlink returned NULL"));
1563
        goto cleanup;
1564 1565
    }

1566
    udev_monitor_enable_receiving(priv->udev_monitor);
1567 1568 1569 1570 1571 1572 1573 1574 1575

    /* 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.  */
1576 1577 1578
    priv->watch = virEventAddHandle(udev_monitor_get_fd(priv->udev_monitor),
                                    VIR_EVENT_HANDLE_READABLE,
                                    udevEventHandleCallback, NULL, NULL);
1579
    if (priv->watch == -1)
1580
        goto cleanup;
1581 1582

    /* Create a fictional 'computer' device to root the device tree. */
1583
    if (udevSetupSystemDev() != 0)
1584
        goto cleanup;
1585 1586 1587

    /* Populate with known devices */

1588
    if (udevEnumerateDevices(udev) != 0)
1589
        goto cleanup;
1590 1591

    ret = 0;
1592

1593
 cleanup:
1594
    nodeDeviceUnlock();
1595

1596
    if (ret == -1)
1597
        nodeStateCleanup();
1598 1599 1600 1601
    return ret;
}


1602
static int nodeStateReload(void)
1603 1604 1605 1606 1607
{
    return 0;
}


1608
static virNodeDeviceDriver udevNodeDeviceDriver = {
1609
    .name = "udev",
1610 1611
    .nodeNumOfDevices = nodeNumOfDevices, /* 0.7.3 */
    .nodeListDevices = nodeListDevices, /* 0.7.3 */
1612
    .connectListAllNodeDevices = nodeConnectListAllNodeDevices, /* 0.10.2 */
1613 1614
    .connectNodeDeviceEventRegisterAny = nodeConnectNodeDeviceEventRegisterAny, /* 2.2.0 */
    .connectNodeDeviceEventDeregisterAny = nodeConnectNodeDeviceEventDeregisterAny, /* 2.2.0 */
1615 1616 1617 1618 1619 1620 1621 1622
    .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 */
1623 1624 1625
};

static virStateDriver udevStateDriver = {
M
Matthias Bolte 已提交
1626
    .name = "udev",
1627 1628 1629
    .stateInitialize = nodeStateInitialize, /* 0.7.3 */
    .stateCleanup = nodeStateCleanup, /* 0.7.3 */
    .stateReload = nodeStateReload, /* 0.7.3 */
1630 1631 1632 1633
};

int udevNodeRegister(void)
{
1634
    VIR_DEBUG("Registering udev node device backend");
1635

1636
    if (virSetSharedNodeDeviceDriver(&udevNodeDeviceDriver) < 0)
1637 1638 1639 1640
        return -1;

    return virRegisterStateDriver(&udevStateDriver);
}