node_device_udev.c 43.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 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 31
#include "node_device_conf.h"
#include "node_device_driver.h"
32 33 34
#include "node_device_linux_sysfs.h"
#include "node_device_udev.h"
#include "virerror.h"
35 36
#include "driver.h"
#include "datatypes.h"
37
#include "virlog.h"
38
#include "viralloc.h"
39
#include "viruuid.h"
40
#include "virbuffer.h"
41
#include "virfile.h"
42
#include "virpci.h"
43
#include "virstring.h"
44
#include "virnetdev.h"
45 46 47

#define VIR_FROM_THIS VIR_FROM_NODEDEV

48 49
VIR_LOG_INIT("node_device.node_device_udev");

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

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

60

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

    return false;
}


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

77
    ret = udev_device_get_property_value(udev_device, property_key);
78

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

    return ret;
}


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

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


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

105
    str = udevGetDeviceProperty(udev_device, property_key);
106

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


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

122
    str = udevGetDeviceProperty(udev_device, property_key);
123

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


132 133
static const char *udevGetDeviceSysfsAttr(struct udev_device *udev_device,
                                          const char *attr_name)
134
{
135
    const char *ret = NULL;
136

137
    ret = udev_device_get_sysattr_value(udev_device, attr_name);
138 139

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


static int udevGetStringSysfsAttr(struct udev_device *udev_device,
                                  const char *attr_name,
                                  char **value)
{
151
    if (VIR_STRDUP(*value, udevGetDeviceSysfsAttr(udev_device, attr_name)) < 0)
152
        return -1;
153

154
    virStringStripControlChars(*value);
155

156 157
    if (*value != NULL && (STREQ(*value, "")))
        VIR_FREE(*value);
158

159
    return 0;
160 161 162 163 164 165 166 167
}


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

170
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
171

172 173
    if (str && virStrToLong_i(str, NULL, base, value) < 0) {
        VIR_ERROR(_("Failed to convert '%s' to int"), str);
174
        return -1;
175 176
    }

177
    return 0;
178 179 180 181 182 183 184 185
}


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

188
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
189

190 191
    if (str && virStrToLong_ui(str, NULL, base, value) < 0) {
        VIR_ERROR(_("Failed to convert '%s' to unsigned int"), str);
192
        return -1;
193 194
    }

195
    return 0;
196 197 198 199 200 201 202
}


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

205
    str = udevGetDeviceSysfsAttr(udev_device, attr_name);
206

207 208
    if (str && virStrToLong_ull(str, NULL, 0, value) < 0) {
        VIR_ERROR(_("Failed to convert '%s' to unsigned long long"), str);
J
Ján Tomko 已提交
209
        return -1;
210 211
    }

J
Ján Tomko 已提交
212
    return 0;
213 214 215 216 217 218 219
}


static int udevGenerateDeviceName(struct udev_device *device,
                                  virNodeDeviceDefPtr def,
                                  const char *s)
{
220 221
    int ret = 0;
    size_t i;
222 223
    virBuffer buf = VIR_BUFFER_INITIALIZER;

224
    virBufferAsprintf(&buf, "%s_%s",
225 226 227
                      udev_device_get_subsystem(device),
                      udev_device_get_sysname(device));

228
    if (s != NULL)
229
        virBufferAsprintf(&buf, "_%s", s);
230

231 232
    if (virBufferCheckError(&buf) < 0)
        return -1;
233 234 235

    def->name = virBufferContentAndReset(&buf);

236
    for (i = 0; i < strlen(def->name); i++) {
237
        if (!(c_isalnum(*(def->name + i))))
238 239 240 241 242 243
            *(def->name + i) = '_';
    }

    return ret;
}

244
#if HAVE_UDEV_LOGGING
J
Ján Tomko 已提交
245 246 247 248 249 250 251 252 253
typedef void (*udevLogFunctionPtr)(struct udev *udev,
                                   int priority,
                                   const char *file,
                                   int line,
                                   const char *fn,
                                   const char *format,
                                   va_list args);

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

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

    format = virBufferContentAndReset(&buf);

271
    virLogVMessage(&virLogSelf,
J
Ján Tomko 已提交
272 273 274 275
                   virLogPriorityFromSyslog(priority),
                   file, line, fn, NULL, format ? format : fmt, args);

    VIR_FREE(format);
276
}
277
#endif
278 279


280 281 282 283 284
static int udevTranslatePCIIds(unsigned int vendor,
                               unsigned int product,
                               char **vendor_string,
                               char **product_string)
{
285
    int ret = -1;
286 287 288 289 290 291 292 293 294 295 296 297 298 299
    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,
300
                    &vendor_name,
301 302 303
                    NULL,
                    NULL);

304 305 306
    if (VIR_STRDUP(*vendor_string, vendor_name) < 0||
        VIR_STRDUP(*product_string, device_name) < 0)
        goto out;
307 308 309

    ret = 0;

310
 out:
311 312 313 314
    return ret;
}


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

326
    syspath = udev_device_get_syspath(device);
327

328
    if (udevGetUintProperty(device, "PCI_CLASS", &data->pci_dev.class, 16) < 0)
329 330
        goto out;

331 332 333 334 335 336 337 338
    if ((p = strrchr(syspath, '/')) == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->pci_dev.domain) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->pci_dev.bus) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->pci_dev.slot) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 16, &data->pci_dev.function) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the PCI address from sysfs path: '%s'"),
                       syspath);
339 340 341
        goto out;
    }

342
    if (udevGetUintSysfsAttr(device, "vendor", &data->pci_dev.vendor, 16) < 0)
343 344
        goto out;

345
    if (udevGetUintSysfsAttr(device, "device", &data->pci_dev.product, 16) < 0)
346 347
        goto out;

348 349 350 351 352 353
    if (udevTranslatePCIIds(data->pci_dev.vendor,
                            data->pci_dev.product,
                            &data->pci_dev.vendor_name,
                            &data->pci_dev.product_name) != 0) {
        goto out;
    }
354

355
    if (udevGenerateDeviceName(device, def, NULL) != 0)
356 357
        goto out;

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

365
    if (nodeDeviceSysfsGetPCIRelatedDevCaps(syspath, data) < 0)
366 367
        goto out;

368 369 370 371 372 373
    if (!(pciDev = virPCIDeviceNew(data->pci_dev.domain,
                                   data->pci_dev.bus,
                                   data->pci_dev.slot,
                                   data->pci_dev.function)))
        goto out;

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

379 380
        if (virPCIDeviceIsPCIExpress(pciDev) > 0) {
            if (VIR_ALLOC(pci_express) < 0)
381 382
                goto out;

383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400
            if (virPCIDeviceHasPCIExpressLink(pciDev) > 0) {
                if (VIR_ALLOC(pci_express->link_cap) < 0 ||
                    VIR_ALLOC(pci_express->link_sta) < 0)
                    goto out;

                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)
                    goto out;

                pci_express->link_sta->port = -1; /* PCIe can't negotiate port. Yet :) */
            }
            data->pci_dev.flags |= VIR_NODE_DEV_CAP_FLAG_PCIE;
            data->pci_dev.pci_express = pci_express;
            pci_express = NULL;
401 402 403
        }
    }

404 405
    ret = 0;

406
 out:
407
    virPCIDeviceFree(pciDev);
408
    virPCIEDeviceInfoFree(pci_express);
409 410 411 412 413 414 415
    return ret;
}


static int udevProcessUSBDevice(struct udev_device *device,
                                virNodeDeviceDefPtr def)
{
416
    virNodeDevCapDataPtr data = &def->caps->data;
417 418
    int ret = -1;

419
    if (udevGetUintProperty(device, "BUSNUM", &data->usb_dev.bus, 10) < 0)
420
        goto out;
421
    if (udevGetUintProperty(device, "DEVNUM", &data->usb_dev.device, 10) < 0)
422
        goto out;
423
    if (udevGetUintProperty(device, "ID_VENDOR_ID", &data->usb_dev.vendor, 16) < 0)
424 425
        goto out;

426 427 428 429 430 431
    if (udevGetStringProperty(device,
                              "ID_VENDOR_FROM_DATABASE",
                              &data->usb_dev.vendor_name) < 0)
        goto out;

    if (!data->usb_dev.vendor_name &&
432 433
        udevGetStringSysfsAttr(device, "manufacturer",
                               &data->usb_dev.vendor_name) < 0)
434 435
        goto out;

436
    if (udevGetUintProperty(device, "ID_MODEL_ID", &data->usb_dev.product, 16) < 0)
437 438
        goto out;

439 440 441 442 443 444
    if (udevGetStringProperty(device,
                              "ID_MODEL_FROM_DATABASE",
                              &data->usb_dev.product_name) < 0)
        goto out;

    if (!data->usb_dev.product_name &&
445 446
        udevGetStringSysfsAttr(device, "product",
                               &data->usb_dev.product_name) < 0)
447 448
        goto out;

449
    if (udevGenerateDeviceName(device, def, NULL) != 0)
450 451 452 453
        goto out;

    ret = 0;

454
 out:
455 456 457 458 459 460 461 462
    return ret;
}


static int udevProcessUSBInterface(struct udev_device *device,
                                   virNodeDeviceDefPtr def)
{
    int ret = -1;
463
    virNodeDevCapDataPtr data = &def->caps->data;
464

465 466
    if (udevGetUintSysfsAttr(device, "bInterfaceNumber",
                             &data->usb_if.number, 16) < 0)
467 468
        goto out;

469 470
    if (udevGetUintSysfsAttr(device, "bInterfaceClass",
                             &data->usb_if._class, 16) < 0)
471 472
        goto out;

473 474
    if (udevGetUintSysfsAttr(device, "bInterfaceSubClass",
                             &data->usb_if.subclass, 16) < 0)
475 476
        goto out;

477 478
    if (udevGetUintSysfsAttr(device, "bInterfaceProtocol",
                             &data->usb_if.protocol, 16) < 0)
479 480
        goto out;

481
    if (udevGenerateDeviceName(device, def, NULL) != 0)
482 483 484 485
        goto out;

    ret = 0;

486
 out:
487 488 489 490 491 492 493 494
    return ret;
}


static int udevProcessNetworkInterface(struct udev_device *device,
                                       virNodeDeviceDefPtr def)
{
    int ret = -1;
D
David Allan 已提交
495
    const char *devtype = udev_device_get_devtype(device);
496
    virNodeDevCapDataPtr data = &def->caps->data;
497

D
David Allan 已提交
498 499 500 501 502 503
    if (devtype && STREQ(devtype, "wlan")) {
        data->net.subtype = VIR_NODE_DEV_CAP_NET_80211;
    } else {
        data->net.subtype = VIR_NODE_DEV_CAP_NET_80203;
    }

504 505
    if (udevGetStringProperty(device,
                              "INTERFACE",
506
                              &data->net.ifname) < 0)
507 508
        goto out;

509 510
    if (udevGetStringSysfsAttr(device, "address",
                               &data->net.address) < 0)
511 512
        goto out;

513
    if (udevGetUintSysfsAttr(device, "addr_len", &data->net.address_len, 0) < 0)
514 515
        goto out;

516
    if (udevGenerateDeviceName(device, def, data->net.address) != 0)
517 518
        goto out;

519 520 521
    if (virNetDevGetLinkInfo(data->net.ifname, &data->net.lnk) < 0)
        goto out;

522 523 524
    if (virNetDevGetFeatures(data->net.ifname, &data->net.features) < 0)
        goto out;

525 526
    ret = 0;

527
 out:
528 529 530 531 532 533 534 535
    return ret;
}


static int udevProcessSCSIHost(struct udev_device *device ATTRIBUTE_UNUSED,
                               virNodeDeviceDefPtr def)
{
    int ret = -1;
536
    virNodeDevCapDataPtr data = &def->caps->data;
537
    char *filename = NULL;
J
Ján Tomko 已提交
538
    char *str;
539

540
    filename = last_component(def->sysfs_path);
541

542 543 544 545 546
    if (!(str = STRSKIP(filename, "host")) ||
        virStrToLong_ui(str, NULL, 0, &data->scsi_host.host) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse SCSI host '%s'"),
                       filename);
547 548 549
        goto out;
    }

550
    nodeDeviceSysfsGetSCSIHostCaps(&def->caps->data);
551

552
    if (udevGenerateDeviceName(device, def, NULL) != 0)
553 554 555 556
        goto out;

    ret = 0;

557
 out:
558 559 560 561
    return ret;
}


D
David Allan 已提交
562 563 564 565 566
static int udevProcessSCSITarget(struct udev_device *device ATTRIBUTE_UNUSED,
                                 virNodeDeviceDefPtr def)
{
    int ret = -1;
    const char *sysname = NULL;
567
    virNodeDevCapDataPtr data = &def->caps->data;
D
David Allan 已提交
568 569 570

    sysname = udev_device_get_sysname(device);

571
    if (VIR_STRDUP(data->scsi_target.name, sysname) < 0)
D
David Allan 已提交
572 573
        goto out;

574
    if (udevGenerateDeviceName(device, def, NULL) != 0)
D
David Allan 已提交
575 576 577 578
        goto out;

    ret = 0;

579
 out:
D
David Allan 已提交
580 581 582 583
    return ret;
}


584
static int udevGetSCSIType(virNodeDeviceDefPtr def ATTRIBUTE_UNUSED,
585
                           unsigned int type, char **typestring)
586 587 588 589 590 591 592 593
{
    int ret = 0;
    int foundtype = 1;

    *typestring = NULL;

    switch (type) {
    case TYPE_DISK:
594
        ignore_value(VIR_STRDUP(*typestring, "disk"));
595 596
        break;
    case TYPE_TAPE:
597
        ignore_value(VIR_STRDUP(*typestring, "tape"));
598 599
        break;
    case TYPE_PROCESSOR:
600
        ignore_value(VIR_STRDUP(*typestring, "processor"));
601 602
        break;
    case TYPE_WORM:
603
        ignore_value(VIR_STRDUP(*typestring, "worm"));
604 605
        break;
    case TYPE_ROM:
606
        ignore_value(VIR_STRDUP(*typestring, "cdrom"));
607 608
        break;
    case TYPE_SCANNER:
609
        ignore_value(VIR_STRDUP(*typestring, "scanner"));
610 611
        break;
    case TYPE_MOD:
612
        ignore_value(VIR_STRDUP(*typestring, "mod"));
613 614
        break;
    case TYPE_MEDIUM_CHANGER:
615
        ignore_value(VIR_STRDUP(*typestring, "changer"));
616 617
        break;
    case TYPE_ENCLOSURE:
618
        ignore_value(VIR_STRDUP(*typestring, "enclosure"));
619
        break;
620
    case TYPE_RAID:
621
        ignore_value(VIR_STRDUP(*typestring, "raid"));
622
        break;
623 624 625 626 627 628 629 630 631 632
    case TYPE_NO_LUN:
    default:
        foundtype = 0;
        break;
    }

    if (*typestring == NULL) {
        if (foundtype == 1) {
            ret = -1;
        } else {
633 634
            VIR_DEBUG("Failed to find SCSI device type %d for %s",
                      type, def->sysfs_path);
635 636 637 638 639 640 641 642 643 644 645 646
        }
    }

    return ret;
}


static int udevProcessSCSIDevice(struct udev_device *device ATTRIBUTE_UNUSED,
                                 virNodeDeviceDefPtr def)
{
    int ret = -1;
    unsigned int tmp = 0;
647
    virNodeDevCapDataPtr data = &def->caps->data;
648 649
    char *filename = NULL, *p = NULL;

650
    filename = last_component(def->sysfs_path);
651

652 653 654 655 656 657 658 659
    if (virStrToLong_ui(filename, &p, 10, &data->scsi.host) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 10, &data->scsi.bus) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 10, &data->scsi.target) < 0 || p == NULL ||
        virStrToLong_ui(p + 1, &p, 10, &data->scsi.lun) < 0) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("failed to parse the SCSI address from filename: '%s'"),
                       filename);
        return -1;
660 661
    }

662 663 664 665 666
    if (udev_device_get_sysattr_value(device, "type")) {
        if (udevGetUintSysfsAttr(device, "type", &tmp, 0) < 0)
            goto out;

        if (udevGetSCSIType(def, tmp, &data->scsi.type) < 0)
667 668 669
            goto out;
    }

670
    if (udevGenerateDeviceName(device, def, NULL) != 0)
671 672 673 674
        goto out;

    ret = 0;

675
 out:
676
    if (ret != 0) {
677
        VIR_ERROR(_("Failed to process SCSI device with sysfs path '%s'"),
678 679 680 681 682 683 684 685 686
                  def->sysfs_path);
    }
    return ret;
}


static int udevProcessDisk(struct udev_device *device,
                           virNodeDeviceDefPtr def)
{
687
    virNodeDevCapDataPtr data = &def->caps->data;
688 689
    int ret = 0;

J
Ján Tomko 已提交
690
    if (udevGetUint64SysfsAttr(device, "size", &data->storage.num_blocks) < 0)
691 692
        goto out;

J
Ján Tomko 已提交
693 694
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
                               &data->storage.logical_block_size) < 0)
695 696 697 698 699
        goto out;

    data->storage.size = data->storage.num_blocks *
        data->storage.logical_block_size;

700
 out:
701 702 703 704
    return ret;
}


705 706 707
static int udevProcessRemoveableMedia(struct udev_device *device,
                                      virNodeDeviceDefPtr def,
                                      int has_media)
708
{
709
    virNodeDevCapDataPtr data = &def->caps->data;
710
    int is_removable = 0, ret = 0;
711

712 713 714
    if (udevGetIntSysfsAttr(device, "removable", &is_removable, 0) < 0)
        return -1;
    if (is_removable == 1)
715 716
        def->caps->data.storage.flags |= VIR_NODE_DEV_CAP_STORAGE_REMOVABLE;

717
    if (has_media) {
718 719 720 721

        def->caps->data.storage.flags |=
            VIR_NODE_DEV_CAP_STORAGE_REMOVABLE_MEDIA_AVAILABLE;

722
        if (udevGetStringProperty(device, "ID_FS_LABEL",
723
                                  &data->storage.media_label) < 0)
724 725
            goto out;

J
Ján Tomko 已提交
726 727
        if (udevGetUint64SysfsAttr(device, "size",
                                   &data->storage.num_blocks) < 0)
728 729
            goto out;

J
Ján Tomko 已提交
730 731
        if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
                                   &data->storage.logical_block_size) < 0)
732 733 734 735 736 737 738 739 740 741 742
            goto out;

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

743
 out:
744 745 746
    return ret;
}

747 748 749 750 751 752 753 754 755 756 757
static int udevProcessCDROM(struct udev_device *device,
                            virNodeDeviceDefPtr def)
{
    int ret = -1;
    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);
758
    if (VIR_STRDUP(def->caps->data.storage.drive_type, "cdrom") < 0)
759 760
        goto out;

761 762 763
    if (udevHasDeviceProperty(device, "ID_CDROM_MEDIA") &&
        udevGetIntProperty(device, "ID_CDROM_MEDIA", &has_media, 0) < 0)
        goto out;
764 765

    ret = udevProcessRemoveableMedia(device, def, has_media);
766
 out:
767 768 769 770 771 772 773 774
    return ret;
}

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

775
    if (udevHasDeviceProperty(device, "ID_CDROM_MEDIA")) {
776
        /* USB floppy */
777 778
        if (udevGetIntProperty(device, "DKD_MEDIA_AVAILABLE", &has_media, 0) < 0)
            return -1;
779
    } else if (udevHasDeviceProperty(device, "ID_FS_LABEL")) {
780 781 782 783 784 785
        /* Legacy floppy */
        has_media = 1;
    }

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

787 788 789 790

static int udevProcessSD(struct udev_device *device,
                         virNodeDeviceDefPtr def)
{
791
    virNodeDevCapDataPtr data = &def->caps->data;
792 793
    int ret = 0;

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

J
Ján Tomko 已提交
798 799
    if (udevGetUint64SysfsAttr(device, "queue/logical_block_size",
                               &data->storage.logical_block_size) < 0)
800 801 802 803 804
        goto out;

    data->storage.size = data->storage.num_blocks *
        data->storage.logical_block_size;

805
 out:
806 807 808 809 810
    return ret;
}



811 812 813 814 815 816
/* 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)
{
817 818 819
    VIR_DEBUG("Could not find definitive storage type for device "
              "with sysfs path '%s', trying to guess it",
              def->sysfs_path);
820

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


static void udevStripSpaces(char *s)
{
838
    if (s == NULL)
839 840 841 842 843 844 845 846 847 848 849 850 851 852
        return;

    while (virFileStripSuffix(s, " ")) {
        /* do nothing */
        ;
    }

    return;
}


static int udevProcessStorage(struct udev_device *device,
                              virNodeDeviceDefPtr def)
{
853
    virNodeDevCapDataPtr data = &def->caps->data;
854
    int ret = -1;
855
    const char* devnode;
856

857
    devnode = udev_device_get_devnode(device);
858
    if (!devnode) {
859
        VIR_DEBUG("No devnode for '%s'", udev_device_get_devpath(device));
860 861
        goto out;
    }
862 863 864

    if (VIR_STRDUP(data->storage.block, devnode) < 0)
        goto out;
865

866
    if (udevGetStringProperty(device, "ID_BUS", &data->storage.bus) < 0)
867
        goto out;
868
    if (udevGetStringProperty(device, "ID_SERIAL", &data->storage.serial) < 0)
869
        goto out;
870

871
    if (udevGetStringSysfsAttr(device, "device/vendor", &data->storage.vendor) < 0)
872 873
        goto out;
    udevStripSpaces(def->caps->data.storage.vendor);
874
    if (udevGetStringSysfsAttr(device, "device/model", &data->storage.model) < 0)
875 876 877 878 879 880 881
        goto out;
    udevStripSpaces(def->caps->data.storage.model);
    /* 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. */

882 883 884 885
    if (udevGetStringProperty(device, "ID_TYPE", &data->storage.drive_type) < 0)
        goto out;

    if (!data->storage.drive_type ||
886
        STREQ(def->caps->data.storage.drive_type, "generic")) {
887 888
        int val = 0;
        const char *str = NULL;
889 890 891

        /* All floppy drives have the ID_DRIVE_FLOPPY prop. This is
         * needed since legacy floppies don't have a drive_type */
892
        if (udevGetIntProperty(device, "ID_DRIVE_FLOPPY", &val, 0) < 0)
893 894 895
            goto out;
        else if (val == 1)
            str = "floppy";
896

897
        if (!str) {
898
            if (udevGetIntProperty(device, "ID_CDROM", &val, 0) < 0)
899
                goto out;
900 901 902
            else if (val == 1)
                str = "cd";
        }
903

904
        if (!str) {
905
            if (udevGetIntProperty(device, "ID_DRIVE_FLASH_SD", &val, 0) < 0)
906
                goto out;
907 908 909
            if (val == 1)
                str = "sd";
        }
910

911 912
        if (str) {
            if (VIR_STRDUP(data->storage.drive_type, str) < 0)
913
                goto out;
914 915
        } else {
            /* If udev doesn't have it, perhaps we can guess it. */
916
            if (udevKludgeStorageType(def) != 0)
917
                goto out;
918 919 920 921 922 923 924
        }
    }

    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);
925 926
    } else if (STREQ(def->caps->data.storage.drive_type, "floppy")) {
        ret = udevProcessFloppy(device, def);
927 928
    } else if (STREQ(def->caps->data.storage.drive_type, "sd")) {
        ret = udevProcessSD(device, def);
929
    } else {
930 931
        VIR_DEBUG("Unsupported storage type '%s'",
                  def->caps->data.storage.drive_type);
932 933 934
        goto out;
    }

935
    if (udevGenerateDeviceName(device, def, data->storage.serial) != 0)
936 937
        goto out;

938
 out:
939
    VIR_DEBUG("Storage ret=%d", ret);
940 941 942
    return ret;
}

943
static int
944
udevProcessSCSIGeneric(struct udev_device *dev,
945 946
                       virNodeDeviceDefPtr def)
{
947 948
    if (udevGetStringProperty(dev, "DEVNAME", &def->caps->data.sg.path) < 0 ||
        !def->caps->data.sg.path)
949 950 951 952 953 954 955 956
        return -1;

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

    return 0;
}

957 958
static int
udevGetDeviceType(struct udev_device *device,
959
                  virNodeDevCapType *type)
960 961
{
    const char *devtype = NULL;
962
    char *subsystem = NULL;
963
    int ret = -1;
D
David Allan 已提交
964

965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984
    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;
    } else {
        /* PCI devices don't set the DEVTYPE property. */
985
        if (udevHasDeviceProperty(device, "PCI_CLASS"))
986
            *type = VIR_NODE_DEV_CAP_PCI_DEV;
987

988 989 990 991
        /* 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. */
992
        if (udevHasDeviceProperty(device, "INTERFACE"))
993
            *type = VIR_NODE_DEV_CAP_NET;
994 995

        /* SCSI generic device doesn't set DEVTYPE property */
996 997 998 999
        if (udevGetStringProperty(device, "SUBSYSTEM", &subsystem) < 0)
            return -1;

        if (STREQ_NULLABLE(subsystem, "scsi_generic"))
1000 1001
            *type = VIR_NODE_DEV_CAP_SCSI_GENERIC;
        VIR_FREE(subsystem);
1002 1003
    }

1004 1005 1006 1007 1008 1009
    if (!*type)
        VIR_DEBUG("Could not determine device type for device "
                  "with sysfs name '%s'",
                  udev_device_get_sysname(device));
    else
        ret = 0;
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019

    return ret;
}


static int udevGetDeviceDetails(struct udev_device *device,
                                virNodeDeviceDefPtr def)
{
    int ret = 0;

1020
    switch (def->caps->data.type) {
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038
    case VIR_NODE_DEV_CAP_SYSTEM:
        /* There's no libudev equivalent of system, so ignore it. */
        break;
    case VIR_NODE_DEV_CAP_PCI_DEV:
        ret = udevProcessPCI(device, def);
        break;
    case VIR_NODE_DEV_CAP_USB_DEV:
        ret = udevProcessUSBDevice(device, def);
        break;
    case VIR_NODE_DEV_CAP_USB_INTERFACE:
        ret = udevProcessUSBInterface(device, def);
        break;
    case VIR_NODE_DEV_CAP_NET:
        ret = udevProcessNetworkInterface(device, def);
        break;
    case VIR_NODE_DEV_CAP_SCSI_HOST:
        ret = udevProcessSCSIHost(device, def);
        break;
D
David Allan 已提交
1039 1040 1041
    case VIR_NODE_DEV_CAP_SCSI_TARGET:
        ret = udevProcessSCSITarget(device, def);
        break;
1042 1043 1044 1045 1046 1047
    case VIR_NODE_DEV_CAP_SCSI:
        ret = udevProcessSCSIDevice(device, def);
        break;
    case VIR_NODE_DEV_CAP_STORAGE:
        ret = udevProcessStorage(device, def);
        break;
1048
    case VIR_NODE_DEV_CAP_SCSI_GENERIC:
1049
        ret = udevProcessSCSIGeneric(device, def);
1050
        break;
1051
    default:
1052
        VIR_ERROR(_("Unknown device type %d"), def->caps->data.type);
1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
        ret = -1;
        break;
    }

    return ret;
}


static int udevRemoveOneDevice(struct udev_device *device)
{
    virNodeDeviceObjPtr dev = NULL;
    const char *name = NULL;
    int ret = 0;

    name = udev_device_get_syspath(device);
1068
    dev = virNodeDeviceFindBySysfsPath(&driver->devs, name);
1069

1070
    if (dev != NULL) {
1071
        VIR_DEBUG("Removing device '%s' with sysfs path '%s'",
1072
                  dev->def->name, name);
1073
        virNodeDeviceObjRemove(&driver->devs, dev);
1074
    } else {
1075 1076
        VIR_DEBUG("Failed to find device to remove that has udev name '%s'",
                  name);
1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091
        ret = -1;
    }

    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;

1092 1093
    parent_device = device;
    do {
1094

1095
        parent_device = udev_device_get_parent(parent_device);
1096
        if (parent_device == NULL)
1097
            break;
1098

1099 1100
        parent_sysfs_path = udev_device_get_syspath(parent_device);
        if (parent_sysfs_path == NULL) {
1101 1102 1103
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Could not get syspath for parent of '%s'"),
                           udev_device_get_syspath(parent_device));
1104
            goto out;
1105 1106
        }

1107
        dev = virNodeDeviceFindBySysfsPath(&driver->devs,
1108 1109
                                           parent_sysfs_path);
        if (dev != NULL) {
1110 1111
            if (VIR_STRDUP(def->parent, dev->def->name) < 0) {
                virNodeDeviceObjUnlock(dev);
1112 1113
                goto out;
            }
1114
            virNodeDeviceObjUnlock(dev);
1115

1116
            if (VIR_STRDUP(def->parent_sysfs_path, parent_sysfs_path) < 0)
1117 1118 1119 1120 1121
                goto out;
        }

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

1122
    if (!def->parent && VIR_STRDUP(def->parent, "computer") < 0)
1123 1124 1125 1126
        goto out;

    ret = 0;

1127
 out:
1128 1129 1130 1131 1132 1133 1134 1135 1136 1137
    return ret;
}


static int udevAddOneDevice(struct udev_device *device)
{
    virNodeDeviceDefPtr def = NULL;
    virNodeDeviceObjPtr dev = NULL;
    int ret = -1;

1138
    if (VIR_ALLOC(def) != 0)
1139 1140
        goto out;

1141 1142 1143
    if (VIR_STRDUP(def->sysfs_path, udev_device_get_syspath(device)) < 0)
        goto out;

1144
    if (udevGetStringProperty(device, "DRIVER", &def->driver) < 0)
1145 1146
        goto out;

1147
    if (VIR_ALLOC(def->caps) != 0)
1148 1149
        goto out;

1150
    if (udevGetDeviceType(device, &def->caps->data.type) != 0)
1151 1152
        goto out;

1153
    if (udevGetDeviceDetails(device, def) != 0)
1154 1155
        goto out;

1156
    if (udevSetParent(device, def) != 0)
1157 1158
        goto out;

1159 1160
    /* If this is a device change, the old definition will be freed
     * and the current definition will take its place. */
1161
    dev = virNodeDeviceAssignDef(&driver->devs, def);
1162
    if (dev == NULL)
1163 1164 1165 1166 1167 1168
        goto out;

    virNodeDeviceObjUnlock(dev);

    ret = 0;

1169
 out:
1170
    if (ret != 0) {
1171
        VIR_DEBUG("Discarding device %d %p %s", ret, def,
1172
                  def ? NULLSTR(def->sysfs_path) : "");
1173 1174 1175
        virNodeDeviceDefFree(def);
    }

1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189
    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);
1190

1191 1192
    if (device != NULL) {
        if (udevAddOneDevice(device) != 0) {
1193 1194
            VIR_DEBUG("Failed to create node device for udev device '%s'",
                      name);
1195 1196 1197 1198
        }
        ret = 0;
    }

1199 1200
    udev_device_unref(device);

1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
    return ret;
}


static int udevEnumerateDevices(struct udev *udev)
{
    struct udev_enumerate *udev_enumerate = NULL;
    struct udev_list_entry *list_entry = NULL;
    int ret = 0;

    udev_enumerate = udev_enumerate_new(udev);

    ret = udev_enumerate_scan_devices(udev_enumerate);
    if (0 != ret) {
1215
        VIR_ERROR(_("udev scan devices returned %d"), ret);
1216 1217 1218 1219 1220 1221 1222 1223 1224
        goto out;
    }

    udev_list_entry_foreach(list_entry,
                            udev_enumerate_get_list_entry(udev_enumerate)) {

        udevProcessDeviceListEntry(udev, list_entry);
    }

1225
 out:
1226 1227 1228 1229 1230
    udev_enumerate_unref(udev_enumerate);
    return ret;
}


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


1243
static int nodeStateCleanup(void)
1244
{
1245
    udevPrivate *priv = NULL;
1246 1247 1248
    struct udev_monitor *udev_monitor = NULL;
    struct udev *udev = NULL;

J
Ján Tomko 已提交
1249 1250
    if (!driver)
        return -1;
1251

J
Ján Tomko 已提交
1252
    nodeDeviceLock();
1253

J
Ján Tomko 已提交
1254
    priv = driver->privateData;
1255

1256 1257 1258
    if (priv) {
        if (priv->watch != -1)
            virEventRemoveHandle(priv->watch);
1259

1260
        udev_monitor = DRV_STATE_UDEV_MONITOR(driver);
1261

1262 1263 1264 1265
        if (udev_monitor != NULL) {
            udev = udev_monitor_get_udev(udev_monitor);
            udev_monitor_unref(udev_monitor);
        }
J
Ján Tomko 已提交
1266
    }
1267

J
Ján Tomko 已提交
1268 1269
    if (udev != NULL)
        udev_unref(udev);
1270

J
Ján Tomko 已提交
1271 1272 1273 1274 1275
    virNodeDeviceObjListFree(&driver->devs);
    nodeDeviceUnlock();
    virMutexDestroy(&driver->lock);
    VIR_FREE(driver);
    VIR_FREE(priv);
1276

J
Ján Tomko 已提交
1277 1278
    udevPCITranslateDeinit();
    return 0;
1279 1280 1281 1282 1283 1284 1285 1286 1287
}


static void udevEventHandleCallback(int watch ATTRIBUTE_UNUSED,
                                    int fd,
                                    int events ATTRIBUTE_UNUSED,
                                    void *data ATTRIBUTE_UNUSED)
{
    struct udev_device *device = NULL;
1288
    struct udev_monitor *udev_monitor = DRV_STATE_UDEV_MONITOR(driver);
1289 1290 1291
    const char *action = NULL;
    int udev_fd = -1;

1292
    nodeDeviceLock();
1293 1294
    udev_fd = udev_monitor_get_fd(udev_monitor);
    if (fd != udev_fd) {
1295 1296
        VIR_ERROR(_("File descriptor returned by udev %d does not "
                    "match node device file descriptor %d"), fd, udev_fd);
1297 1298 1299 1300 1301
        goto out;
    }

    device = udev_monitor_receive_device(udev_monitor);
    if (device == NULL) {
1302
        VIR_ERROR(_("udev_monitor_receive_device returned NULL"));
1303 1304 1305 1306
        goto out;
    }

    action = udev_device_get_action(device);
1307
    VIR_DEBUG("udev action: '%s'", action);
1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318

    if (STREQ(action, "add") || STREQ(action, "change")) {
        udevAddOneDevice(device);
        goto out;
    }

    if (STREQ(action, "remove")) {
        udevRemoveOneDevice(device);
        goto out;
    }

1319
 out:
1320
    udev_device_unref(device);
1321
    nodeDeviceUnlock();
1322 1323 1324 1325
    return;
}


1326 1327
/* DMI is intel-compatible specific */
#if defined(__x86_64__) || defined(__i386__) || defined(__amd64__)
1328
static void
1329
udevGetDMIData(virNodeDevCapDataPtr data)
1330 1331 1332 1333
{
    struct udev *udev = NULL;
    struct udev_device *device = NULL;

1334
    udev = udev_monitor_get_udev(DRV_STATE_UDEV_MONITOR(driver));
1335

1336 1337
    device = udev_device_new_from_syspath(udev, DMI_DEVPATH);
    if (device == NULL) {
1338 1339
        device = udev_device_new_from_syspath(udev, DMI_DEVPATH_FALLBACK);
        if (device == NULL) {
1340
            VIR_ERROR(_("Failed to get udev device for syspath '%s' or '%s'"),
1341
                      DMI_DEVPATH, DMI_DEVPATH_FALLBACK);
1342 1343
            goto out;
        }
1344 1345
    }

1346 1347
    if (udevGetStringSysfsAttr(device, "product_name",
                               &data->system.product_name) < 0)
1348
        goto out;
1349 1350
    if (udevGetStringSysfsAttr(device, "sys_vendor",
                               &data->system.hardware.vendor_name) < 0)
1351
        goto out;
1352 1353
    if (udevGetStringSysfsAttr(device, "product_version",
                               &data->system.hardware.version) < 0)
1354
        goto out;
1355 1356
    if (udevGetStringSysfsAttr(device, "product_serial",
                               &data->system.hardware.serial) < 0)
1357 1358
        goto out;

1359
    if (virGetHostUUID(data->system.hardware.uuid))
1360 1361
        goto out;

1362 1363
    if (udevGetStringSysfsAttr(device, "bios_vendor",
                               &data->system.firmware.vendor_name) < 0)
1364
        goto out;
1365 1366
    if (udevGetStringSysfsAttr(device, "bios_version",
                               &data->system.firmware.version) < 0)
1367
        goto out;
1368 1369
    if (udevGetStringSysfsAttr(device, "bios_date",
                               &data->system.firmware.release_date) < 0)
1370 1371
        goto out;

1372
 out:
1373
    if (device != NULL)
1374 1375 1376
        udev_device_unref(device);
    return;
}
1377
#endif
1378 1379 1380 1381 1382 1383 1384 1385


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

1386
    if (VIR_ALLOC(def) != 0)
1387 1388
        goto out;

1389
    if (VIR_STRDUP(def->name, "computer") < 0)
1390 1391
        goto out;

1392
    if (VIR_ALLOC(def->caps) != 0)
1393 1394
        goto out;

1395
#if defined(__x86_64__) || defined(__i386__) || defined(__amd64__)
1396
    udevGetDMIData(&def->caps->data);
1397
#endif
1398

1399
    dev = virNodeDeviceAssignDef(&driver->devs, def);
1400
    if (dev == NULL)
1401 1402 1403 1404 1405 1406
        goto out;

    virNodeDeviceObjUnlock(dev);

    ret = 0;

1407
 out:
1408
    if (ret == -1)
1409 1410
        virNodeDeviceDefFree(def);

1411 1412 1413
    return ret;
}

1414
static int udevPCITranslateInit(bool privileged ATTRIBUTE_UNUSED)
1415
{
1416 1417 1418 1419
#if defined __s390__ || defined __s390x_
    /* On s390(x) system there is no PCI bus.
     * Therefore there is nothing to initialize here. */
#else
1420 1421 1422
    int pciret;

    if ((pciret = pci_system_init()) != 0) {
1423 1424 1425
        /* 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.  */
1426
        if (errno != ENOENT && (privileged  || errno != EACCES)) {
1427 1428
            char ebuf[256];
            VIR_ERROR(_("Failed to initialize libpciaccess: %s"),
1429
                      virStrerror(pciret, ebuf, sizeof(ebuf)));
1430
            return -1;
1431
        }
1432
    }
1433
#endif
1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444
    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;

1445
    if (VIR_ALLOC(priv) < 0)
1446
        return -1;
1447 1448

    priv->watch = -1;
1449
    priv->privileged = privileged;
1450

1451
    if (VIR_ALLOC(driver) < 0) {
1452
        VIR_FREE(priv);
1453
        return -1;
1454 1455
    }

1456 1457
    if (virMutexInit(&driver->lock) < 0) {
        VIR_ERROR(_("Failed to initialize mutex for driver"));
1458
        VIR_FREE(priv);
1459
        VIR_FREE(driver);
1460
        return -1;
1461 1462
    }

1463
    driver->privateData = priv;
1464
    nodeDeviceLock();
1465

1466
    if (udevPCITranslateInit(privileged) < 0)
1467 1468
        goto out_unlock;

1469 1470 1471 1472 1473 1474 1475
    /*
     * http://www.kernel.org/pub/linux/utils/kernel/hotplug/libudev/libudev-udev.html#udev-new
     *
     * indicates no return value other than success, so we don't check
     * its return value.
     */
    udev = udev_new();
1476
#if HAVE_UDEV_LOGGING
J
Ján Tomko 已提交
1477 1478
    /* cast to get rid of missing-format-attribute warning */
    udev_set_log_fn(udev, (udevLogFunctionPtr) udevLogFunction);
1479
#endif
1480

1481 1482
    priv->udev_monitor = udev_monitor_new_from_netlink(udev, "udev");
    if (priv->udev_monitor == NULL) {
1483
        VIR_ERROR(_("udev_monitor_new_from_netlink returned NULL"));
1484
        goto out_unlock;
1485 1486
    }

1487
    udev_monitor_enable_receiving(priv->udev_monitor);
1488 1489 1490 1491 1492 1493 1494 1495 1496

    /* 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.  */
1497 1498 1499
    priv->watch = virEventAddHandle(udev_monitor_get_fd(priv->udev_monitor),
                                    VIR_EVENT_HANDLE_READABLE,
                                    udevEventHandleCallback, NULL, NULL);
1500
    if (priv->watch == -1)
1501
        goto out_unlock;
1502 1503

    /* Create a fictional 'computer' device to root the device tree. */
1504
    if (udevSetupSystemDev() != 0)
1505
        goto out_unlock;
1506 1507 1508

    /* Populate with known devices */

1509
    if (udevEnumerateDevices(udev) != 0)
1510
        goto out_unlock;
1511 1512

    ret = 0;
1513

1514
 out_unlock:
1515
    nodeDeviceUnlock();
1516

1517
    if (ret == -1)
1518
        nodeStateCleanup();
1519 1520 1521 1522
    return ret;
}


1523
static int nodeStateReload(void)
1524 1525 1526 1527 1528
{
    return 0;
}


1529
static virNodeDeviceDriver udevNodeDeviceDriver = {
1530
    .name = "udev",
1531 1532
    .nodeNumOfDevices = nodeNumOfDevices, /* 0.7.3 */
    .nodeListDevices = nodeListDevices, /* 0.7.3 */
1533
    .connectListAllNodeDevices = nodeConnectListAllNodeDevices, /* 0.10.2 */
1534 1535 1536 1537 1538 1539 1540 1541
    .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 */
1542 1543 1544
};

static virStateDriver udevStateDriver = {
M
Matthias Bolte 已提交
1545
    .name = "udev",
1546 1547 1548
    .stateInitialize = nodeStateInitialize, /* 0.7.3 */
    .stateCleanup = nodeStateCleanup, /* 0.7.3 */
    .stateReload = nodeStateReload, /* 0.7.3 */
1549 1550 1551 1552
};

int udevNodeRegister(void)
{
1553
    VIR_DEBUG("Registering udev node device backend");
1554

1555
    if (virSetSharedNodeDeviceDriver(&udevNodeDeviceDriver) < 0)
1556 1557 1558 1559
        return -1;

    return virRegisterStateDriver(&udevStateDriver);
}