node_device_hal.c 23.9 KB
Newer Older
1 2 3
/*
 * node_device_hal.c: node device enumeration - HAL-based implementation
 *
4
 * Copyright (C) 2011, 2013 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16 17 18
 * Copyright (C) 2008 Virtual Iron Software, Inc.
 * Copyright (C) 2008 David F. Lively
 *
 * 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
19
 * License along with this library.  If not, see
O
Osier Yang 已提交
20
 * <http://www.gnu.org/licenses/>.
21 22 23 24
 *
 * Author: David F. Lively <dlively@virtualiron.com>
 */

25 26
#include <config.h>

27 28 29 30 31
#include <stdio.h>
#include <stdlib.h>
#include <libhal.h>

#include "node_device_conf.h"
32
#include "node_device_hal.h"
33
#include "virerror.h"
34 35
#include "driver.h"
#include "datatypes.h"
36
#include "viralloc.h"
37
#include "viruuid.h"
38
#include "virpci.h"
39
#include "virlog.h"
40
#include "node_device_driver.h"
41
#include "virdbus.h"
42
#include "virstring.h"
43

44 45
#define VIR_FROM_THIS VIR_FROM_NODEDEV

46 47
VIR_LOG_INIT("node_device.node_device_hal");

48 49 50 51
/*
 * Host device enumeration (HAL implementation)
 */

52
static virNodeDeviceDriverStatePtr driverState;
53 54

#define CONN_DRV_STATE(conn) \
55
        ((virNodeDeviceDriverStatePtr)((conn)->nodeDevicePrivateData))
56 57 58 59 60 61
#define DRV_STATE_HAL_CTX(ds) ((LibHalContext *)((ds)->privateData))
#define CONN_HAL_CTX(conn) DRV_STATE_HAL_CTX(CONN_DRV_STATE(conn))

#define NODE_DEV_UDI(obj) ((const char *)((obj)->privateData)


62 63
static const char *
hal_name(const char *udi)
64 65 66
{
    const char *name = strrchr(udi, '/');
    if (name)
67
        return name + 1;
68 69 70 71
    return udi;
}


72 73 74
static int
get_str_prop(LibHalContext *ctxt, const char *udi,
             const char *prop, char **val_p)
75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
{
    char *val = libhal_device_get_property_string(ctxt, udi, prop, NULL);

    if (val) {
        if (*val) {
            *val_p = val;
            return 0;
        } else {
            /* Treat empty strings as NULL values */
            VIR_FREE(val);
        }
    }

    return -1;
}

91 92 93
static int
get_int_prop(LibHalContext *ctxt, const char *udi,
             const char *prop, int *val_p)
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108
{
    DBusError err;
    int val;
    int rv;

    dbus_error_init(&err);
    val = libhal_device_get_property_int(ctxt, udi, prop, &err);
    rv = dbus_error_is_set(&err);
    dbus_error_free(&err);
    if (rv == 0)
        *val_p = val;

    return rv;
}

109 110 111
static int
get_bool_prop(LibHalContext *ctxt, const char *udi,
              const char *prop, int *val_p)
112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
{
    DBusError err;
    int val;
    int rv;

    dbus_error_init(&err);
    val = libhal_device_get_property_bool(ctxt, udi, prop, &err);
    rv = dbus_error_is_set(&err);
    dbus_error_free(&err);
    if (rv == 0)
        *val_p = val;

    return rv;
}

127 128 129
static int
get_uint64_prop(LibHalContext *ctxt, const char *udi,
                const char *prop, unsigned long long *val_p)
130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
{
    DBusError err;
    unsigned long long val;
    int rv;

    dbus_error_init(&err);
    val = libhal_device_get_property_uint64(ctxt, udi, prop, &err);
    rv = dbus_error_is_set(&err);
    dbus_error_free(&err);
    if (rv == 0)
        *val_p = val;

    return rv;
}

145 146 147
static int
gather_pci_cap(LibHalContext *ctx, const char *udi,
               union _virNodeDevCapData *d)
148 149 150 151 152 153 154 155 156 157 158
{
    char *sysfs_path;

    if (get_str_prop(ctx, udi, "pci.linux.sysfs_path", &sysfs_path) == 0) {
        char *p = strrchr(sysfs_path, '/');
        if (p) {
            (void)virStrToLong_ui(p+1, &p, 16, &d->pci_dev.domain);
            (void)virStrToLong_ui(p+1, &p, 16, &d->pci_dev.bus);
            (void)virStrToLong_ui(p+1, &p, 16, &d->pci_dev.slot);
            (void)virStrToLong_ui(p+1, &p, 16, &d->pci_dev.function);
        }
159

160 161
        if (!virPCIGetPhysicalFunction(sysfs_path,
                                       &d->pci_dev.physical_function))
162 163
            d->pci_dev.flags |= VIR_NODE_DEV_CAP_FLAG_PCI_PHYSICAL_FUNCTION;

164 165 166 167 168 169 170
        int ret = virPCIGetVirtualFunctions(sysfs_path,
                                            &d->pci_dev.virtual_functions,
                                            &d->pci_dev.num_virtual_functions);
        if (ret < 0) {
            VIR_FREE(sysfs_path);
            return -1;
        }
171

172 173
        if (d->pci_dev.num_virtual_functions > 0)
            d->pci_dev.flags |= VIR_NODE_DEV_CAP_FLAG_PCI_VIRTUAL_FUNCTION;
174 175
        VIR_FREE(sysfs_path);
    }
176

177 178 179 180 181 182
    (void)get_int_prop(ctx, udi, "pci.vendor_id", (int *)&d->pci_dev.vendor);
    if (get_str_prop(ctx, udi, "pci.vendor", &d->pci_dev.vendor_name) != 0)
        (void)get_str_prop(ctx, udi, "info.vendor", &d->pci_dev.vendor_name);
    (void)get_int_prop(ctx, udi, "pci.product_id", (int *)&d->pci_dev.product);
    if (get_str_prop(ctx, udi, "pci.product", &d->pci_dev.product_name) != 0)
        (void)get_str_prop(ctx, udi, "info.product", &d->pci_dev.product_name);
183

184 185 186 187
    return 0;
}


188 189 190
static int
gather_usb_cap(LibHalContext *ctx, const char *udi,
               union _virNodeDevCapData *d)
191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
{
    (void)get_int_prop(ctx, udi, "usb.interface.number",
                       (int *)&d->usb_if.number);
    (void)get_int_prop(ctx, udi, "usb.interface.class",
                       (int *)&d->usb_if._class);
    (void)get_int_prop(ctx, udi, "usb.interface.subclass",
                       (int *)&d->usb_if.subclass);
    (void)get_int_prop(ctx, udi, "usb.interface.protocol",
                       (int *)&d->usb_if.protocol);
    (void)get_str_prop(ctx, udi, "usb.interface.description",
                       &d->usb_if.description);
    return 0;
}


206 207 208
static int
gather_usb_device_cap(LibHalContext *ctx, const char *udi,
                      union _virNodeDevCapData *d)
209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227
{
    (void)get_int_prop(ctx, udi, "usb_device.bus_number",
                       (int *)&d->usb_dev.bus);
    (void)get_int_prop(ctx, udi, "usb_device.linux.device_number",
                       (int *)&d->usb_dev.device);
    (void)get_int_prop(ctx, udi, "usb_device.vendor_id",
                       (int *)&d->usb_dev.vendor);
    if (get_str_prop(ctx, udi, "usb_device.vendor",
                     &d->usb_dev.vendor_name) != 0)
        (void)get_str_prop(ctx, udi, "info.vendor", &d->usb_dev.vendor_name);
    (void)get_int_prop(ctx, udi, "usb_device.product_id",
                       (int *)&d->usb_dev.product);
    if (get_str_prop(ctx, udi, "usb_device.product",
                     &d->usb_dev.product_name) != 0)
        (void)get_str_prop(ctx, udi, "info.product", &d->usb_dev.product_name);
    return 0;
}


228 229 230
static int
gather_net_cap(LibHalContext *ctx, const char *udi,
               union _virNodeDevCapData *d)
231 232
{
    unsigned long long dummy;
233
    (void)get_str_prop(ctx, udi, "net.interface", &d->net.ifname);
234 235 236 237 238 239 240 241 242 243 244 245 246 247
    (void)get_str_prop(ctx, udi, "net.address", &d->net.address);
    if (get_uint64_prop(ctx, udi, "net.80203.mac_address",
                        &dummy) == 0)
        d->net.subtype = VIR_NODE_DEV_CAP_NET_80203;
    else if (get_uint64_prop(ctx, udi, "net.80211.mac_address",
                             &dummy) == 0)
        d->net.subtype = VIR_NODE_DEV_CAP_NET_80211;
    else
        d->net.subtype = VIR_NODE_DEV_CAP_NET_LAST;

    return 0;
}


248 249 250
static int
gather_scsi_host_cap(LibHalContext *ctx, const char *udi,
                     union _virNodeDevCapData *d)
251
{
252 253
    int retval = 0;

254
    (void)get_int_prop(ctx, udi, "scsi_host.host", (int *)&d->scsi_host.host);
255

O
Osier Yang 已提交
256
    retval = detect_scsi_host_caps(d);
257 258 259 260 261 262 263

    if (retval == -1) {
        goto out;
    }

out:
    return retval;
264 265 266
}


267 268 269
static int
gather_scsi_cap(LibHalContext *ctx, const char *udi,
                union _virNodeDevCapData *d)
270 271 272 273 274 275 276 277 278 279
{
    (void)get_int_prop(ctx, udi, "scsi.host", (int *)&d->scsi.host);
    (void)get_int_prop(ctx, udi, "scsi.bus", (int *)&d->scsi.bus);
    (void)get_int_prop(ctx, udi, "scsi.target", (int *)&d->scsi.target);
    (void)get_int_prop(ctx, udi, "scsi.lun", (int *)&d->scsi.lun);
    (void)get_str_prop(ctx, udi, "scsi.type", &d->scsi.type);
    return 0;
}


280 281 282
static int
gather_storage_cap(LibHalContext *ctx, const char *udi,
                   union _virNodeDevCapData *d)
283 284
{
    int val;
285
    (void)get_str_prop(ctx, udi, "block.device", &d->storage.block);
286 287 288 289
    (void)get_str_prop(ctx, udi, "storage.bus", &d->storage.bus);
    (void)get_str_prop(ctx, udi, "storage.drive_type", &d->storage.drive_type);
    (void)get_str_prop(ctx, udi, "storage.model", &d->storage.model);
    (void)get_str_prop(ctx, udi, "storage.vendor", &d->storage.vendor);
290
    (void)get_str_prop(ctx, udi, "storage.serial", &d->storage.serial);
291 292
    if (get_bool_prop(ctx, udi, "storage.removable", &val) == 0 && val) {
        d->storage.flags |= VIR_NODE_DEV_CAP_STORAGE_REMOVABLE;
293 294
        if (get_bool_prop(ctx, udi, "storage.removable.media_available",
                          &val) == 0 && val) {
295 296 297 298 299 300 301 302 303 304 305 306 307
            d->storage.flags |=
                VIR_NODE_DEV_CAP_STORAGE_REMOVABLE_MEDIA_AVAILABLE;
            (void)get_uint64_prop(ctx, udi, "storage.removable.media_size",
                                  &d->storage.removable_media_size);
        }
    } else {
        (void)get_uint64_prop(ctx, udi, "storage.size", &d->storage.size);
    }
    if (get_bool_prop(ctx, udi, "storage.hotpluggable", &val) == 0 && val)
        d->storage.flags |= VIR_NODE_DEV_CAP_STORAGE_HOTPLUGGABLE;
    return 0;
}

308 309 310 311 312 313 314 315
static int
gather_scsi_generic_cap(LibHalContext *ctx, const char *udi,
                        union _virNodeDevCapData *d)
{
    (void)get_str_prop(ctx, udi, "scsi_generic.device", &d->sg.path);
    return 0;
}

316

317 318 319
static int
gather_system_cap(LibHalContext *ctx, const char *udi,
                  union _virNodeDevCapData *d)
320 321 322 323 324 325 326 327 328 329 330
{
    char *uuidstr;

    (void)get_str_prop(ctx, udi, "system.product", &d->system.product_name);
    (void)get_str_prop(ctx, udi, "system.hardware.vendor",
                       &d->system.hardware.vendor_name);
    (void)get_str_prop(ctx, udi, "system.hardware.version",
                       &d->system.hardware.version);
    (void)get_str_prop(ctx, udi, "system.hardware.serial",
                       &d->system.hardware.serial);
    if (get_str_prop(ctx, udi, "system.hardware.uuid", &uuidstr) == 0) {
331
        ignore_value(virUUIDParse(uuidstr, d->system.hardware.uuid));
332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359 360 361 362
        VIR_FREE(uuidstr);
    }
    (void)get_str_prop(ctx, udi, "system.firmware.vendor",
                       &d->system.firmware.vendor_name);
    (void)get_str_prop(ctx, udi, "system.firmware.version",
                       &d->system.firmware.version);
    (void)get_str_prop(ctx, udi, "system.firmware.release_date",
                       &d->system.firmware.release_date);
    return 0;
}


struct _caps_tbl_entry {
    const char *cap_name;
    enum virNodeDevCapType type;
    int (*gather_fn)(LibHalContext *ctx,
                     const char *udi,
                     union _virNodeDevCapData *data);
};

typedef struct _caps_tbl_entry caps_tbl_entry;

static caps_tbl_entry caps_tbl[] = {
    { "system",     VIR_NODE_DEV_CAP_SYSTEM,        gather_system_cap },
    { "pci",        VIR_NODE_DEV_CAP_PCI_DEV,       gather_pci_cap },
    { "usb",        VIR_NODE_DEV_CAP_USB_INTERFACE, gather_usb_cap },
    { "usb_device", VIR_NODE_DEV_CAP_USB_DEV,       gather_usb_device_cap },
    { "net",        VIR_NODE_DEV_CAP_NET,           gather_net_cap },
    { "scsi_host",  VIR_NODE_DEV_CAP_SCSI_HOST,     gather_scsi_host_cap },
    { "scsi",       VIR_NODE_DEV_CAP_SCSI,          gather_scsi_cap },
    { "storage",    VIR_NODE_DEV_CAP_STORAGE,       gather_storage_cap },
363
    { "scsi_generic", VIR_NODE_DEV_CAP_SCSI_GENERIC, gather_scsi_generic_cap },
364 365 366 367
};


/* qsort/bsearch string comparator */
368 369
static int
cmpstringp(const void *p1, const void *p2)
370 371 372 373 374 375
{
    /* from man 3 qsort */
    return strcmp(* (char * const *) p1, * (char * const *) p2);
}


376 377 378 379
static int
gather_capability(LibHalContext *ctx, const char *udi,
                  const char *cap_name,
                  virNodeDevCapsDefPtr *caps_p)
380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405
{
    caps_tbl_entry *entry;

    entry = bsearch(&cap_name, caps_tbl, ARRAY_CARDINALITY(caps_tbl),
                    sizeof(caps_tbl[0]), cmpstringp);

    if (entry) {
        virNodeDevCapsDefPtr caps;
        if (VIR_ALLOC(caps) < 0)
            return ENOMEM;
        caps->type = entry->type;
        if (entry->gather_fn) {
            int rv = (*entry->gather_fn)(ctx, udi, &caps->data);
            if (rv != 0) {
                virNodeDevCapsDefFree(caps);
                return rv;
            }
        }
        caps->next = *caps_p;
        *caps_p = caps;
    }

    return 0;
}


406 407 408
static int
gather_capabilities(LibHalContext *ctx, const char *udi,
                    virNodeDevCapsDefPtr *caps_p)
409 410 411
{
    char *bus_name = NULL;
    virNodeDevCapsDefPtr caps = NULL;
412
    char **hal_cap_names = NULL;
413 414
    int rv;
    size_t i;
415 416 417 418 419 420 421

    if (STREQ(udi, "/org/freedesktop/Hal/devices/computer")) {
        rv = gather_capability(ctx, udi, "system", &caps);
        if (rv != 0)
            goto failure;
    }

422 423
    if (get_str_prop(ctx, udi, "info.subsystem", &bus_name) == 0 ||
        get_str_prop(ctx, udi, "linux.subsystem", &bus_name) == 0) {
424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462 463
        rv = gather_capability(ctx, udi, bus_name, &caps);
        if (rv != 0)
            goto failure;
    }

    hal_cap_names = libhal_device_get_property_strlist(ctx, udi,
                                                       "info.capabilities",
                                                       NULL);
    if (hal_cap_names) {
        for (i = 0; hal_cap_names[i]; i++) {
            if (! (bus_name && STREQ(hal_cap_names[i], bus_name))) {
                rv = gather_capability(ctx, udi, hal_cap_names[i], &caps);
                if (rv != 0)
                    goto failure;
            }
        }
        for (i = 0; hal_cap_names[i]; i++)
            VIR_FREE(hal_cap_names[i]);
        VIR_FREE(hal_cap_names);
    }
    VIR_FREE(bus_name);

    *caps_p = caps;
    return 0;

 failure:
    VIR_FREE(bus_name);
    if (hal_cap_names) {
        for (i = 0; hal_cap_names[i]; i++)
            VIR_FREE(hal_cap_names[i]);
        VIR_FREE(hal_cap_names);
    }
    while (caps) {
        virNodeDevCapsDefPtr next = caps->next;
        virNodeDevCapsDefFree(caps);
        caps = next;
    }
    return rv;
}

464 465
static void
free_udi(void *udi)
466 467 468 469
{
    VIR_FREE(udi);
}

470 471
static void
dev_create(const char *udi)
472
{
473
    LibHalContext *ctx;
474
    char *parent_key = NULL;
475 476
    virNodeDeviceObjPtr dev = NULL;
    virNodeDeviceDefPtr def = NULL;
477 478
    const char *name = hal_name(udi);
    int rv;
479
    char *privData;
480
    char *devicePath = NULL;
481

482
    if (VIR_STRDUP(privData, udi) < 0)
483
        return;
484

485 486 487 488 489
    nodeDeviceLock(driverState);
    ctx = DRV_STATE_HAL_CTX(driverState);

    if (VIR_ALLOC(def) < 0)
        goto failure;
490

491
    if (VIR_STRDUP(def->name, name) < 0)
492 493 494
        goto failure;

    if (get_str_prop(ctx, udi, "info.parent", &parent_key) == 0) {
495 496
        if (VIR_STRDUP(def->parent, hal_name(parent_key)) < 0) {
            VIR_FREE(parent_key);
497
            goto failure;
498 499
        }
        VIR_FREE(parent_key);
500 501
    }

502
    rv = gather_capabilities(ctx, udi, &def->caps);
503 504
    if (rv != 0) goto failure;

505 506
    if (def->caps == NULL)
        goto cleanup;
507

508
    /* Some devices don't have a path in sysfs, so ignore failure */
509
    (void)get_str_prop(ctx, udi, "linux.sysfs_path", &devicePath);
510

511
    dev = virNodeDeviceAssignDef(&driverState->devs,
512 513
                                 def);

514 515
    if (!dev) {
        VIR_FREE(devicePath);
516
        goto failure;
517
    }
518

519 520
    dev->privateData = privData;
    dev->privateFree = free_udi;
521
    dev->def->sysfs_path = devicePath;
522

523
    virNodeDeviceObjUnlock(dev);
524

525
    nodeDeviceUnlock(driverState);
526 527 528
    return;

 failure:
529
    VIR_DEBUG("FAILED TO ADD dev %s", name);
530 531
cleanup:
    VIR_FREE(privData);
532
    virNodeDeviceDefFree(def);
533
    nodeDeviceUnlock(driverState);
534 535
}

536 537
static void
dev_refresh(const char *udi)
538 539 540 541 542 543 544 545 546 547 548 549
{
    const char *name = hal_name(udi);
    virNodeDeviceObjPtr dev;

    nodeDeviceLock(driverState);
    dev = virNodeDeviceFindByName(&driverState->devs, name);
    if (dev) {
        /* Simply "rediscover" device -- incrementally handling changes
         * to sub-capabilities (like net.80203) is nasty ... so avoid it.
         */
        virNodeDeviceObjRemove(&driverState->devs, dev);
    } else
550
        VIR_DEBUG("no device named %s", name);
551 552 553 554 555 556
    nodeDeviceUnlock(driverState);

    if (dev) {
        dev_create(udi);
    }
}
557

558 559 560
static void
device_added(LibHalContext *ctx ATTRIBUTE_UNUSED,
             const char *udi)
561
{
562
    VIR_DEBUG("%s", hal_name(udi));
563
    dev_create(udi);
564 565 566
}


567 568 569
static void
device_removed(LibHalContext *ctx ATTRIBUTE_UNUSED,
               const char *udi)
570 571
{
    const char *name = hal_name(udi);
572 573 574
    virNodeDeviceObjPtr dev;

    nodeDeviceLock(driverState);
575
    dev = virNodeDeviceFindByName(&driverState->devs, name);
576
    VIR_DEBUG("%s", name);
577 578 579
    if (dev)
        virNodeDeviceObjRemove(&driverState->devs, dev);
    else
580
        VIR_DEBUG("no device named %s", name);
581
    nodeDeviceUnlock(driverState);
582 583 584
}


585 586 587
static void
device_cap_added(LibHalContext *ctx,
                 const char *udi, const char *cap)
588 589
{
    const char *name = hal_name(udi);
590 591 592
    virNodeDeviceObjPtr dev;

    nodeDeviceLock(driverState);
593
    dev = virNodeDeviceFindByName(&driverState->devs, name);
594
    nodeDeviceUnlock(driverState);
595
    VIR_DEBUG("%s %s", cap, name);
596
    if (dev) {
597
        (void)gather_capability(ctx, udi, cap, &dev->def->caps);
598 599
        virNodeDeviceObjUnlock(dev);
    } else {
600
        VIR_DEBUG("no device named %s", name);
601
    }
602 603 604
}


605 606 607 608
static void
device_cap_lost(LibHalContext *ctx ATTRIBUTE_UNUSED,
                const char *udi,
                const char *cap)
609 610
{
    const char *name = hal_name(udi);
611
    VIR_DEBUG("%s %s", cap, name);
612 613

    dev_refresh(udi);
614 615 616
}


617 618 619 620 621 622
static void
device_prop_modified(LibHalContext *ctx ATTRIBUTE_UNUSED,
                     const char *udi,
                     const char *key,
                     dbus_bool_t is_removed ATTRIBUTE_UNUSED,
                     dbus_bool_t is_added ATTRIBUTE_UNUSED)
623 624
{
    const char *name = hal_name(udi);
625
    VIR_DEBUG("%s %s", name, key);
626

627
    dev_refresh(udi);
628 629 630
}


631 632 633 634
static int
nodeStateInitialize(bool privileged ATTRIBUTE_UNUSED,
                    virStateInhibitCallback callback ATTRIBUTE_UNUSED,
                    void *opaque ATTRIBUTE_UNUSED)
635 636 637
{
    LibHalContext *hal_ctx = NULL;
    char **udi = NULL;
638 639
    int num_devs;
    size_t i;
640
    int ret = -1;
641 642
    DBusConnection *sysbus;
    DBusError err;
643 644 645 646 647 648 649 650

    /* Ensure caps_tbl is sorted by capability name */
    qsort(caps_tbl, ARRAY_CARDINALITY(caps_tbl), sizeof(caps_tbl[0]),
          cmpstringp);

    if (VIR_ALLOC(driverState) < 0)
        return -1;

651 652 653 654
    if (virMutexInit(&driverState->lock) < 0) {
        VIR_FREE(driverState);
        return -1;
    }
655 656
    nodeDeviceLock(driverState);

657
    dbus_error_init(&err);
658 659 660 661 662 663 664
    if (!(sysbus = virDBusGetSystemBus())) {
        virErrorPtr verr = virGetLastError();
        VIR_ERROR(_("DBus not available, disabling HAL driver: %s"),
                    verr->message);
        ret = 0;
        goto failure;
    }
665

666 667
    hal_ctx = libhal_ctx_new();
    if (hal_ctx == NULL) {
668
        VIR_ERROR(_("libhal_ctx_new returned NULL"));
669 670
        goto failure;
    }
671

672
    if (!libhal_ctx_set_dbus_connection(hal_ctx, sysbus)) {
673
        VIR_ERROR(_("libhal_ctx_set_dbus_connection failed"));
674 675 676
        goto failure;
    }
    if (!libhal_ctx_init(hal_ctx, &err)) {
677
        VIR_ERROR(_("libhal_ctx_init failed, haldaemon is probably not running"));
678 679 680 681
        /* We don't want to show a fatal error here,
           otherwise entire libvirtd shuts down when
           hald isn't running */
        ret = 0;
682 683 684
        goto failure;
    }

685 686 687 688 689 690 691 692 693 694
    /* Populate with known devices */
    driverState->privateData = hal_ctx;

    /* We need to unlock state now, since setting these callbacks cause
     * a dbus RPC call, and while this call is waiting for the reply,
     * a signal may already arrive, triggering the callback and thus
     * requiring the lock !
     */
    nodeDeviceUnlock(driverState);

695 696 697 698 699
    /* Register HAL event callbacks */
    if (!libhal_ctx_set_device_added(hal_ctx, device_added) ||
        !libhal_ctx_set_device_removed(hal_ctx, device_removed) ||
        !libhal_ctx_set_device_new_capability(hal_ctx, device_cap_added) ||
        !libhal_ctx_set_device_lost_capability(hal_ctx, device_cap_lost) ||
700
        !libhal_ctx_set_device_property_modified(hal_ctx, device_prop_modified) ||
701
        !libhal_device_property_watch_all(hal_ctx, &err)) {
702
        VIR_ERROR(_("setting up HAL callbacks failed"));
703 704 705 706 707
        goto failure;
    }

    udi = libhal_get_all_devices(hal_ctx, &num_devs, &err);
    if (udi == NULL) {
708
        VIR_ERROR(_("libhal_get_all_devices failed"));
709 710
        goto failure;
    }
711
    for (i = 0; i < num_devs; i++) {
712
        dev_create(udi[i]);
713 714 715
        VIR_FREE(udi[i]);
    }
    VIR_FREE(udi);
716 717 718 719 720

    return 0;

 failure:
    if (dbus_error_is_set(&err)) {
721
        VIR_ERROR(_("%s: %s"), err.name, err.message);
722 723 724 725 726
        dbus_error_free(&err);
    }
    virNodeDeviceObjListFree(&driverState->devs);
    if (hal_ctx)
        (void)libhal_ctx_free(hal_ctx);
727
    nodeDeviceUnlock(driverState);
728 729
    VIR_FREE(driverState);

730
    return ret;
731 732 733
}


734 735
static int
nodeStateCleanup(void)
736 737
{
    if (driverState) {
738
        nodeDeviceLock(driverState);
739 740 741 742
        LibHalContext *hal_ctx = DRV_STATE_HAL_CTX(driverState);
        virNodeDeviceObjListFree(&driverState->devs);
        (void)libhal_ctx_shutdown(hal_ctx, NULL);
        (void)libhal_ctx_free(hal_ctx);
743
        nodeDeviceUnlock(driverState);
744
        virMutexDestroy(&driverState->lock);
745 746 747 748 749 750 751
        VIR_FREE(driverState);
        return 0;
    }
    return -1;
}


752 753
static int
nodeStateReload(void)
754
{
755 756
    DBusError err;
    char **udi = NULL;
757 758
    int num_devs;
    size_t i;
759 760
    LibHalContext *hal_ctx;

761
    VIR_INFO("Reloading HAL device state");
762
    nodeDeviceLock(driverState);
763
    VIR_INFO("Removing existing objects");
764 765 766 767
    virNodeDeviceObjListFree(&driverState->devs);
    nodeDeviceUnlock(driverState);

    hal_ctx = DRV_STATE_HAL_CTX(driverState);
768
    VIR_INFO("Creating new objects");
769 770 771
    dbus_error_init(&err);
    udi = libhal_get_all_devices(hal_ctx, &num_devs, &err);
    if (udi == NULL) {
772
        VIR_ERROR(_("libhal_get_all_devices failed"));
773 774 775 776 777 778 779
        return -1;
    }
    for (i = 0; i < num_devs; i++) {
        dev_create(udi[i]);
        VIR_FREE(udi[i]);
    }
    VIR_FREE(udi);
780
    VIR_INFO("HAL device reload complete");
781 782

    return 0;
783 784 785
}


786 787 788 789
static virDrvOpenStatus
nodeDeviceOpen(virConnectPtr conn,
               virConnectAuthPtr auth ATTRIBUTE_UNUSED,
               unsigned int flags)
790
{
E
Eric Blake 已提交
791 792
    virCheckFlags(VIR_CONNECT_RO, VIR_DRV_OPEN_ERROR);

793 794 795
    if (driverState == NULL)
        return VIR_DRV_OPEN_DECLINED;

796
    conn->nodeDevicePrivateData = driverState;
797 798 799 800

    return VIR_DRV_OPEN_SUCCESS;
}

801 802
static int
nodeDeviceClose(virConnectPtr conn ATTRIBUTE_UNUSED)
803
{
804
    conn->nodeDevicePrivateData = NULL;
805 806 807 808
    return 0;
}


809 810
static virNodeDeviceDriver halNodeDeviceDriver = {
    .name = "halNodeDeviceDriver",
811
    .nodeDeviceOpen = nodeDeviceOpen, /* 0.5.0 */
812
    .nodeDeviceClose = nodeDeviceClose, /* 0.5.0 */
813 814
    .nodeNumOfDevices = nodeNumOfDevices, /* 0.5.0 */
    .nodeListDevices = nodeListDevices, /* 0.5.0 */
815
    .connectListAllNodeDevices = nodeConnectListAllNodeDevices, /* 0.10.2 */
816 817 818 819 820 821 822 823
    .nodeDeviceLookupByName = nodeDeviceLookupByName, /* 0.5.0 */
    .nodeDeviceLookupSCSIHostByWWN = nodeDeviceLookupSCSIHostByWWN, /* 1.0.2 */
    .nodeDeviceGetXMLDesc = nodeDeviceGetXMLDesc, /* 0.5.0 */
    .nodeDeviceGetParent = nodeDeviceGetParent, /* 0.5.0 */
    .nodeDeviceNumOfCaps = nodeDeviceNumOfCaps, /* 0.5.0 */
    .nodeDeviceListCaps = nodeDeviceListCaps, /* 0.5.0 */
    .nodeDeviceCreateXML = nodeDeviceCreateXML, /* 0.6.5 */
    .nodeDeviceDestroy = nodeDeviceDestroy, /* 0.6.5 */
824 825 826 827
};


static virStateDriver halStateDriver = {
828
    .name = "HAL",
829 830 831
    .stateInitialize = nodeStateInitialize, /* 0.5.0 */
    .stateCleanup = nodeStateCleanup, /* 0.5.0 */
    .stateReload = nodeStateReload, /* 0.5.0 */
832 833
};

834 835
int
halNodeRegister(void)
836
{
837
    if (virRegisterNodeDeviceDriver(&halNodeDeviceDriver) < 0)
838 839 840
        return -1;
    return virRegisterStateDriver(&halStateDriver);
}