qemu_conf.c 44.8 KB
Newer Older
D
Daniel P. Berrange 已提交
1
/*
2
 * qemu_conf.c: QEMU configuration management
D
Daniel P. Berrange 已提交
3
 *
4
 * Copyright (C) 2006-2013 Red Hat, Inc.
D
Daniel P. Berrange 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17
 * Copyright (C) 2006 Daniel P. Berrange
 *
 * 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
18
 * License along with this library.  If not, see
O
Osier Yang 已提交
19
 * <http://www.gnu.org/licenses/>.
D
Daniel P. Berrange 已提交
20 21 22 23
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

24
#include <config.h>
25

D
Daniel P. Berrange 已提交
26 27 28 29
#include <string.h>
#include <limits.h>
#include <sys/types.h>
#include <sys/stat.h>
30
#include <stdlib.h>
D
Daniel P. Berrange 已提交
31 32 33
#include <unistd.h>
#include <errno.h>
#include <fcntl.h>
34
#include <sys/wait.h>
35
#include <arpa/inet.h>
D
Daniel P. Berrange 已提交
36

37
#include "virerror.h"
38
#include "qemu_conf.h"
39
#include "qemu_command.h"
40
#include "qemu_capabilities.h"
41
#include "qemu_bridge_filter.h"
42
#include "viruuid.h"
43
#include "virbuffer.h"
44
#include "virconf.h"
45
#include "viralloc.h"
46
#include "datatypes.h"
47
#include "virxml.h"
48
#include "nodeinfo.h"
49
#include "virlog.h"
50
#include "cpu/cpu.h"
51
#include "domain_nwfilter.h"
E
Eric Blake 已提交
52
#include "virfile.h"
53
#include "virstring.h"
54
#include "viratomic.h"
55
#include "configmake.h"
56

57 58
#define VIR_FROM_THIS VIR_FROM_QEMU

59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
typedef struct _qemuDriverCloseDef qemuDriverCloseDef;
typedef qemuDriverCloseDef *qemuDriverCloseDefPtr;
struct _qemuDriverCloseDef {
    virConnectPtr conn;
    virQEMUCloseCallback cb;
};

struct _virQEMUCloseCallbacks {
    virObjectLockable parent;

    /* UUID string to qemuDriverCloseDef mapping */
    virHashTablePtr list;
};


74
static virClassPtr virQEMUDriverConfigClass;
75
static virClassPtr virQEMUCloseCallbacksClass;
76
static void virQEMUDriverConfigDispose(void *obj);
77
static void virQEMUCloseCallbacksDispose(void *obj);
78 79 80

static int virQEMUConfigOnceInit(void)
{
81 82 83 84
    virQEMUDriverConfigClass = virClassNew(virClassForObject(),
                                           "virQEMUDriverConfig",
                                           sizeof(virQEMUDriverConfig),
                                           virQEMUDriverConfigDispose);
85

86 87 88 89 90 91 92 93 94
    virQEMUCloseCallbacksClass = virClassNew(virClassForObjectLockable(),
                                             "virQEMUCloseCallbacks",
                                             sizeof(virQEMUCloseCallbacks),
                                             virQEMUCloseCallbacksDispose);

    if (!virQEMUDriverConfigClass || !virQEMUCloseCallbacksClass)
        return -1;
    else
        return 0;
95 96 97 98 99
}

VIR_ONCE_GLOBAL_INIT(virQEMUConfig)


100 101
static void
qemuDriverLock(virQEMUDriverPtr driver)
102 103 104
{
    virMutexLock(&driver->lock);
}
105 106
static void
qemuDriverUnlock(virQEMUDriverPtr driver)
107 108 109 110 111
{
    virMutexUnlock(&driver->lock);
}


112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
virQEMUDriverConfigPtr virQEMUDriverConfigNew(bool privileged)
{
    virQEMUDriverConfigPtr cfg;

    if (virQEMUConfigInitialize() < 0)
        return NULL;

    if (!(cfg = virObjectNew(virQEMUDriverConfigClass)))
        return NULL;

    cfg->privileged = privileged;
    cfg->uri = privileged ? "qemu:///system" : "qemu:///session";

    if (privileged) {
        if (virGetUserID(QEMU_USER, &cfg->user) < 0)
            goto error;
        if (virGetGroupID(QEMU_GROUP, &cfg->group) < 0)
            goto error;
    } else {
131 132
        cfg->user = (uid_t)-1;
        cfg->group = (gid_t)-1;
133 134 135
    }
    cfg->dynamicOwnership = privileged;

136
    cfg->cgroupControllers = -1; /* -1 == auto-detect */
137 138 139 140 141 142

    if (privileged) {
        if (virAsprintf(&cfg->logDir,
                        "%s/log/libvirt/qemu", LOCALSTATEDIR) < 0)
            goto no_memory;

143 144
        if (VIR_STRDUP(cfg->configBaseDir, SYSCONFDIR "/libvirt") < 0)
            goto error;
D
Daniel P. Berrange 已提交
145

146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
        if (virAsprintf(&cfg->stateDir,
                      "%s/run/libvirt/qemu", LOCALSTATEDIR) < 0)
            goto no_memory;

        if (virAsprintf(&cfg->libDir,
                      "%s/lib/libvirt/qemu", LOCALSTATEDIR) < 0)
            goto no_memory;

        if (virAsprintf(&cfg->cacheDir,
                      "%s/cache/libvirt/qemu", LOCALSTATEDIR) < 0)
            goto no_memory;
        if (virAsprintf(&cfg->saveDir,
                      "%s/lib/libvirt/qemu/save", LOCALSTATEDIR) < 0)
            goto no_memory;
        if (virAsprintf(&cfg->snapshotDir,
                        "%s/lib/libvirt/qemu/snapshot", LOCALSTATEDIR) < 0)
            goto no_memory;
        if (virAsprintf(&cfg->autoDumpPath,
                        "%s/lib/libvirt/qemu/dump", LOCALSTATEDIR) < 0)
            goto no_memory;
    } else {
        char *rundir;
        char *cachedir;

        cachedir = virGetUserCacheDirectory();
        if (!cachedir)
            goto error;

        if (virAsprintf(&cfg->logDir,
                        "%s/qemu/log", cachedir) < 0) {
            VIR_FREE(cachedir);
            goto no_memory;
        }
        if (virAsprintf(&cfg->cacheDir, "%s/qemu/cache", cachedir) < 0) {
            VIR_FREE(cachedir);
            goto no_memory;
        }
        VIR_FREE(cachedir);
184

185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
        rundir = virGetUserRuntimeDirectory();
        if (!rundir)
            goto error;
        if (virAsprintf(&cfg->stateDir, "%s/qemu/run", rundir) < 0) {
            VIR_FREE(rundir);
            goto no_memory;
        }
        VIR_FREE(rundir);

        if (!(cfg->configBaseDir = virGetUserConfigDirectory()))
            goto error;

        if (virAsprintf(&cfg->libDir, "%s/qemu/lib", cfg->configBaseDir) < 0)
            goto no_memory;
        if (virAsprintf(&cfg->saveDir, "%s/qemu/save", cfg->configBaseDir) < 0)
            goto no_memory;
        if (virAsprintf(&cfg->snapshotDir, "%s/qemu/snapshot", cfg->configBaseDir) < 0)
            goto no_memory;
        if (virAsprintf(&cfg->autoDumpPath, "%s/qemu/dump", cfg->configBaseDir) < 0)
            goto no_memory;
    }

    if (virAsprintf(&cfg->configDir, "%s/qemu", cfg->configBaseDir) < 0)
        goto no_memory;
    if (virAsprintf(&cfg->autostartDir, "%s/qemu/autostart", cfg->configBaseDir) < 0)
210
        goto no_memory;
211 212


213 214
    if (VIR_STRDUP(cfg->vncListen, "127.0.0.1") < 0)
        goto error;
215

216 217
    if (VIR_STRDUP(cfg->vncTLSx509certdir, SYSCONFDIR "/pki/libvirt-vnc") < 0)
        goto error;
D
Daniel P. Berrange 已提交
218

219 220
    if (VIR_STRDUP(cfg->spiceListen, "127.0.0.1") < 0)
        goto error;
221

222 223
    if (VIR_STRDUP(cfg->spiceTLSx509certdir , SYSCONFDIR "/pki/libvirt-spice") < 0)
        goto error;
224

225 226 227
    cfg->remotePortMin = QEMU_REMOTE_PORT_MIN;
    cfg->remotePortMax = QEMU_REMOTE_PORT_MAX;

228 229 230
    cfg->webSocketPortMin = QEMU_WEBSOCKET_PORT_MIN;
    cfg->webSocketPortMax = QEMU_WEBSOCKET_PORT_MAX;

231
#if defined HAVE_MNTENT_H && defined HAVE_GETMNTENT_R
232 233 234
    /* For privileged driver, try and find hugepage mount automatically.
     * Non-privileged driver requires admin to create a dir for the
     * user, chown it, and then let user configure it manually */
235 236
    if (privileged &&
        !(cfg->hugetlbfsMount = virFileFindMountPoint("hugetlbfs"))) {
237
        if (errno != ENOENT) {
238
            virReportSystemError(errno, "%s",
239
                                 _("unable to find hugetlbfs mountpoint"));
240
            goto error;
241 242 243
        }
    }
#endif
244 245
    if (VIR_STRDUP(cfg->bridgeHelperName, "/usr/libexec/qemu-bridge-helper") < 0)
        goto error;
246

247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294
    cfg->clearEmulatorCapabilities = true;

    cfg->securityDefaultConfined = true;
    cfg->securityRequireConfined = false;

    cfg->keepAliveInterval = 5;
    cfg->keepAliveCount = 5;
    cfg->seccompSandbox = -1;

    return cfg;

no_memory:
    virReportOOMError();
error:
    virObjectUnref(cfg);
    return NULL;
}


static void virQEMUDriverConfigDispose(void *obj)
{
    virQEMUDriverConfigPtr cfg = obj;


    virStringFreeList(cfg->cgroupDeviceACL);

    VIR_FREE(cfg->configBaseDir);
    VIR_FREE(cfg->configDir);
    VIR_FREE(cfg->autostartDir);
    VIR_FREE(cfg->logDir);
    VIR_FREE(cfg->stateDir);

    VIR_FREE(cfg->libDir);
    VIR_FREE(cfg->cacheDir);
    VIR_FREE(cfg->saveDir);
    VIR_FREE(cfg->snapshotDir);

    VIR_FREE(cfg->vncTLSx509certdir);
    VIR_FREE(cfg->vncListen);
    VIR_FREE(cfg->vncPassword);
    VIR_FREE(cfg->vncSASLdir);

    VIR_FREE(cfg->spiceTLSx509certdir);
    VIR_FREE(cfg->spiceListen);
    VIR_FREE(cfg->spicePassword);

    VIR_FREE(cfg->hugetlbfsMount);
    VIR_FREE(cfg->hugepagePath);
295
    VIR_FREE(cfg->bridgeHelperName);
296 297 298 299 300 301 302 303 304 305

    VIR_FREE(cfg->saveImageFormat);
    VIR_FREE(cfg->dumpImageFormat);
    VIR_FREE(cfg->autoDumpPath);

    virStringFreeList(cfg->securityDriverNames);

    VIR_FREE(cfg->lockManagerName);
}

306

307 308 309 310 311 312 313
int virQEMUDriverConfigLoadFile(virQEMUDriverConfigPtr cfg,
                                const char *filename)
{
    virConfPtr conf = NULL;
    virConfValuePtr p;
    int ret = -1;
    int i;
314

D
Daniel P. Berrange 已提交
315 316 317
    /* Just check the file is readable before opening it, otherwise
     * libvirt emits an error.
     */
318
    if (access(filename, R_OK) == -1) {
319
        VIR_INFO("Could not read qemu config file %s", filename);
320
        return 0;
321
    }
D
Daniel P. Berrange 已提交
322

323 324
    if (!(conf = virConfReadFile(filename, 0)))
        goto cleanup;
D
Daniel P. Berrange 已提交
325

326 327 328 329 330
#define CHECK_TYPE(name,typ)                          \
    if (p && p->type != (typ)) {                      \
        virReportError(VIR_ERR_INTERNAL_ERROR,        \
                       "%s: %s: expected type " #typ, \
                       filename, (name));             \
331
        goto cleanup;                                 \
332 333 334 335 336 337 338 339
    }

#define GET_VALUE_LONG(NAME, VAR)     \
    p = virConfGetValue(conf, NAME);  \
    CHECK_TYPE(NAME, VIR_CONF_LONG);  \
    if (p)                            \
        VAR = p->l;

340 341 342 343 344 345
#define GET_VALUE_BOOL(NAME, VAR)     \
    p = virConfGetValue(conf, NAME);  \
    CHECK_TYPE(NAME, VIR_CONF_LONG);  \
    if (p)                            \
        VAR = p->l != 0;

346 347 348 349 350
#define GET_VALUE_STR(NAME, VAR)           \
    p = virConfGetValue(conf, NAME);       \
    CHECK_TYPE(NAME, VIR_CONF_STRING);     \
    if (p && p->str) {                     \
        VIR_FREE(VAR);                     \
351 352
        if (VIR_STRDUP(VAR, p->str) < 0)   \
            goto cleanup;                  \
353 354
    }

355 356 357 358 359 360 361 362 363
    GET_VALUE_BOOL("vnc_auto_unix_socket", cfg->vncAutoUnixSocket);
    GET_VALUE_BOOL("vnc_tls", cfg->vncTLS);
    GET_VALUE_BOOL("vnc_tls_x509_verify", cfg->vncTLSx509verify);
    GET_VALUE_STR("vnc_tls_x509_cert_dir", cfg->vncTLSx509certdir);
    GET_VALUE_STR("vnc_listen", cfg->vncListen);
    GET_VALUE_STR("vnc_password", cfg->vncPassword);
    GET_VALUE_BOOL("vnc_sasl", cfg->vncSASL);
    GET_VALUE_STR("vnc_sasl_dir", cfg->vncSASLdir);
    GET_VALUE_BOOL("vnc_allow_host_audio", cfg->vncAllowHostAudio);
364

365
    p = virConfGetValue(conf, "security_driver");
366 367 368 369
    if (p && p->type == VIR_CONF_LIST) {
        size_t len;
        virConfValuePtr pp;

J
Ján Tomko 已提交
370
        /* Calc length and check items */
371 372
        for (len = 0, pp = p->list; pp; len++, pp = pp->next) {
            if (pp->type != VIR_CONF_STRING) {
373 374 375
                virReportError(VIR_ERR_CONF_SYNTAX, "%s",
                               _("security_driver must be a list of strings"));
                goto cleanup;
376 377 378
            }
        }

379
        if (VIR_ALLOC_N(cfg->securityDriverNames, len + 1) < 0)
380
            goto no_memory;
381 382

        for (i = 0, pp = p->list; pp; i++, pp = pp->next) {
383 384
            if (VIR_STRDUP(cfg->securityDriverNames[i], pp->str) < 0)
                goto cleanup;
385
        }
386
        cfg->securityDriverNames[len] = NULL;
387
    } else {
388
        CHECK_TYPE("security_driver", VIR_CONF_STRING);
389
        if (p && p->str) {
390
            if (VIR_ALLOC_N(cfg->securityDriverNames, 2) < 0)
391
                goto no_memory;
392 393
            if (VIR_STRDUP(cfg->securityDriverNames[0], p->str) < 0)
                goto cleanup;
394

395
            cfg->securityDriverNames[1] = NULL;
396
        }
397 398
    }

399 400
    GET_VALUE_BOOL("security_default_confined", cfg->securityDefaultConfined);
    GET_VALUE_BOOL("security_require_confined", cfg->securityRequireConfined);
401

402 403 404 405
    GET_VALUE_BOOL("spice_tls", cfg->spiceTLS);
    GET_VALUE_STR("spice_tls_x509_cert_dir", cfg->spiceTLSx509certdir);
    GET_VALUE_STR("spice_listen", cfg->spiceListen);
    GET_VALUE_STR("spice_password", cfg->spicePassword);
406 407


408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436
    GET_VALUE_LONG("remote_websocket_port_min", cfg->webSocketPortMin);
    if (cfg->webSocketPortMin < QEMU_WEBSOCKET_PORT_MIN) {
        /* if the port is too low, we can't get the display name
         * to tell to vnc (usually subtract 5700, e.g. localhost:1
         * for port 5701) */
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("%s: remote_websocket_port_min: port must be greater "
                         "than or equal to %d"),
                        filename, QEMU_WEBSOCKET_PORT_MIN);
        goto cleanup;
    }

    GET_VALUE_LONG("remote_websocket_port_max", cfg->webSocketPortMax);
    if (cfg->webSocketPortMax > QEMU_WEBSOCKET_PORT_MAX ||
        cfg->webSocketPortMax < cfg->webSocketPortMin) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                        _("%s: remote_websocket_port_max: port must be between "
                          "the minimal port and %d"),
                       filename, QEMU_WEBSOCKET_PORT_MAX);
        goto cleanup;
    }

    if (cfg->webSocketPortMin > cfg->webSocketPortMax) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                        _("%s: remote_websocket_port_min: min port must not be "
                          "greater than max port"), filename);
        goto cleanup;
    }

437 438
    GET_VALUE_LONG("remote_display_port_min", cfg->remotePortMin);
    if (cfg->remotePortMin < QEMU_REMOTE_PORT_MIN) {
439 440 441 442 443 444 445
        /* if the port is too low, we can't get the display name
         * to tell to vnc (usually subtract 5900, e.g. localhost:1
         * for port 5901) */
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("%s: remote_display_port_min: port must be greater "
                         "than or equal to %d"),
                        filename, QEMU_REMOTE_PORT_MIN);
446
        goto cleanup;
447 448
    }

449 450 451
    GET_VALUE_LONG("remote_display_port_max", cfg->remotePortMax);
    if (cfg->remotePortMax > QEMU_REMOTE_PORT_MAX ||
        cfg->remotePortMax < cfg->remotePortMin) {
452 453 454 455
        virReportError(VIR_ERR_INTERNAL_ERROR,
                        _("%s: remote_display_port_max: port must be between "
                          "the minimal port and %d"),
                       filename, QEMU_REMOTE_PORT_MAX);
456
        goto cleanup;
457 458
    }

459
    if (cfg->remotePortMin > cfg->remotePortMax) {
460
        virReportError(VIR_ERR_INTERNAL_ERROR,
461 462
                        _("%s: remote_display_port_min: min port must not be "
                          "greater than max port"), filename);
463
        goto cleanup;
464 465
    }

466 467
    p = virConfGetValue(conf, "user");
    CHECK_TYPE("user", VIR_CONF_STRING);
468 469
    if (p && p->str &&
        virGetUserID(p->str, &cfg->user) < 0)
470
        goto cleanup;
471

472 473
    p = virConfGetValue(conf, "group");
    CHECK_TYPE("group", VIR_CONF_STRING);
474 475
    if (p && p->str &&
        virGetGroupID(p->str, &cfg->group) < 0)
476
        goto cleanup;
477

478
    GET_VALUE_BOOL("dynamic_ownership", cfg->dynamicOwnership);
479

480 481
    p = virConfGetValue(conf, "cgroup_controllers");
    CHECK_TYPE("cgroup_controllers", VIR_CONF_LIST);
482
    if (p) {
483
        cfg->cgroupControllers = 0;
484 485 486 487
        virConfValuePtr pp;
        for (i = 0, pp = p->list; pp; ++i, pp = pp->next) {
            int ctl;
            if (pp->type != VIR_CONF_STRING) {
488 489 490 491
                virReportError(VIR_ERR_CONF_SYNTAX, "%s",
                               _("cgroup_controllers must be a "
                                 "list of strings"));
                goto cleanup;
492
            }
493 494 495 496 497

            if ((ctl = virCgroupControllerTypeFromString(pp->str)) < 0) {
                virReportError(VIR_ERR_CONF_SYNTAX,
                               _("Unknown cgroup controller '%s'"), pp->str);
                goto cleanup;
498
            }
499
            cfg->cgroupControllers |= (1 << ctl);
500 501 502
        }
    }

503 504
    p = virConfGetValue(conf, "cgroup_device_acl");
    CHECK_TYPE("cgroup_device_acl", VIR_CONF_LIST);
505 506 507 508 509
    if (p) {
        int len = 0;
        virConfValuePtr pp;
        for (pp = p->list; pp; pp = pp->next)
            len++;
510
        if (VIR_ALLOC_N(cfg->cgroupDeviceACL, 1+len) < 0)
511 512
            goto no_memory;

513 514
        for (i = 0, pp = p->list; pp; ++i, pp = pp->next) {
            if (pp->type != VIR_CONF_STRING) {
515 516 517 518
                virReportError(VIR_ERR_CONF_SYNTAX, "%s",
                               _("cgroup_device_acl must be a "
                                 "list of strings"));
                goto cleanup;
519
            }
520 521
            if (VIR_STRDUP(cfg->cgroupDeviceACL[i], pp->str) < 0)
                goto cleanup;
522
        }
523
        cfg->cgroupDeviceACL[i] = NULL;
524 525
    }

526 527 528 529 530
    GET_VALUE_STR("save_image_format", cfg->saveImageFormat);
    GET_VALUE_STR("dump_image_format", cfg->dumpImageFormat);
    GET_VALUE_STR("auto_dump_path", cfg->autoDumpPath);
    GET_VALUE_BOOL("auto_dump_bypass_cache", cfg->autoDumpBypassCache);
    GET_VALUE_BOOL("auto_start_bypass_cache", cfg->autoStartBypassCache);
531

532
    GET_VALUE_STR("hugetlbfs_mount", cfg->hugetlbfsMount);
533
    GET_VALUE_STR("bridge_helper", cfg->bridgeHelperName);
534

535 536 537 538 539 540 541 542
    GET_VALUE_BOOL("mac_filter", cfg->macFilter);

    GET_VALUE_BOOL("relaxed_acs_check", cfg->relaxedACS);
    GET_VALUE_BOOL("clear_emulator_capabilities", cfg->clearEmulatorCapabilities);
    GET_VALUE_BOOL("allow_disk_format_probing", cfg->allowDiskFormatProbing);
    GET_VALUE_BOOL("set_process_name", cfg->setProcessName);
    GET_VALUE_LONG("max_processes", cfg->maxProcesses);
    GET_VALUE_LONG("max_files", cfg->maxFiles);
543

544 545 546 547 548 549 550 551
    GET_VALUE_STR("lock_manager", cfg->lockManagerName);

    GET_VALUE_LONG("max_queued", cfg->maxQueuedJobs);

    GET_VALUE_LONG("keepalive_interval", cfg->keepAliveInterval);
    GET_VALUE_LONG("keepalive_count", cfg->keepAliveCount);

    GET_VALUE_LONG("seccomp_sandbox", cfg->seccompSandbox);
552

553 554 555
    ret = 0;

cleanup:
556
    virConfFree(conf);
557 558 559 560 561
    return ret;

no_memory:
    virReportOOMError();
    goto cleanup;
D
Daniel P. Berrange 已提交
562
}
563
#undef GET_VALUE_BOOL
564 565
#undef GET_VALUE_LONG
#undef GET_VALUE_STRING
566

567 568
virQEMUDriverConfigPtr virQEMUDriverGetConfig(virQEMUDriverPtr driver)
{
569 570 571 572 573
    virQEMUDriverConfigPtr conf;
    qemuDriverLock(driver);
    conf = virObjectRef(driver->config);
    qemuDriverUnlock(driver);
    return conf;
574 575
}

576
virDomainXMLOptionPtr
577
virQEMUDriverCreateXMLConf(virQEMUDriverPtr driver)
578
{
579
    virQEMUDriverDomainDefParserConfig.priv = driver;
580
    return virDomainXMLOptionNew(&virQEMUDriverDomainDefParserConfig,
581 582
                                 &virQEMUDriverPrivateDataCallbacks,
                                 &virQEMUDriverDomainXMLNamespace);
583 584
}

585 586 587 588 589 590 591 592 593 594 595

virCapsPtr virQEMUDriverCreateCapabilities(virQEMUDriverPtr driver)
{
    size_t i;
    virCapsPtr caps;
    virSecurityManagerPtr *sec_managers = NULL;
    /* Security driver data */
    const char *doi, *model;
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);

    /* Basic host arch / guest machine capabilities */
596 597
    if (!(caps = virQEMUCapsInit(driver->qemuCapsCache)))
        goto no_memory;
598 599 600 601

    if (virGetHostUUID(caps->host.host_uuid)) {
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("cannot get the host uuid"));
602
        goto error;
603 604 605
    }

    /* access sec drivers and create a sec model for each one */
606 607
    if (!(sec_managers = virSecurityManagerGetNested(driver->securityManager)))
        goto error;
608 609 610 611 612 613 614 615 616 617 618 619

    /* calculate length */
    for (i = 0; sec_managers[i]; i++)
        ;
    caps->host.nsecModels = i;

    if (VIR_ALLOC_N(caps->host.secModels, caps->host.nsecModels) < 0)
        goto no_memory;

    for (i = 0; sec_managers[i]; i++) {
        doi = virSecurityManagerGetDOI(sec_managers[i]);
        model = virSecurityManagerGetModel(sec_managers[i]);
620 621 622
        if (VIR_STRDUP(caps->host.secModels[i].model, model) < 0 ||
            VIR_STRDUP(caps->host.secModels[i].doi, doi) < 0)
            goto error;
623 624 625 626 627 628 629 630 631 632
        VIR_DEBUG("Initialized caps for security driver \"%s\" with "
                  "DOI \"%s\"", model, doi);
    }
    VIR_FREE(sec_managers);

    virObjectUnref(cfg);
    return caps;

no_memory:
    virReportOOMError();
633
error:
634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
    VIR_FREE(sec_managers);
    virObjectUnref(caps);
    virObjectUnref(cfg);
    return NULL;
}


/**
 * virQEMUDriverGetCapabilities:
 *
 * Get a reference to the virCapsPtr instance for the
 * driver. If @refresh is true, the capabilities will be
 * rebuilt first
 *
 * The caller must release the reference with virObjetUnref
 *
 * Returns: a reference to a virCapsPtr instance or NULL
 */
virCapsPtr virQEMUDriverGetCapabilities(virQEMUDriverPtr driver,
                                        bool refresh)
{
655
    virCapsPtr ret = NULL;
656 657 658 659 660
    if (refresh) {
        virCapsPtr caps = NULL;
        if ((caps = virQEMUDriverCreateCapabilities(driver)) == NULL)
            return NULL;

661
        qemuDriverLock(driver);
662 663
        virObjectUnref(driver->caps);
        driver->caps = caps;
664 665
    } else {
        qemuDriverLock(driver);
666 667
    }

668 669 670
    ret = virObjectRef(driver->caps);
    qemuDriverUnlock(driver);
    return ret;
671 672 673
}


674
static void
675 676
virQEMUCloseCallbacksFreeData(void *payload,
                              const void *name ATTRIBUTE_UNUSED)
677 678 679 680
{
    VIR_FREE(payload);
}

681 682
virQEMUCloseCallbacksPtr
virQEMUCloseCallbacksNew(void)
683
{
684
    virQEMUCloseCallbacksPtr closeCallbacks;
685

686 687 688 689 690 691 692 693 694 695 696 697 698
    if (virQEMUConfigInitialize() < 0)
        return NULL;

    if (!(closeCallbacks = virObjectLockableNew(virQEMUCloseCallbacksClass)))
        return NULL;

    closeCallbacks->list = virHashCreate(5, virQEMUCloseCallbacksFreeData);
    if (!closeCallbacks->list) {
        virObjectUnref(closeCallbacks);
        return NULL;
    }

    return closeCallbacks;
699 700
}

701 702
static void
virQEMUCloseCallbacksDispose(void *obj)
703
{
704 705 706
    virQEMUCloseCallbacksPtr closeCallbacks = obj;

    virHashFree(closeCallbacks->list);
707 708 709
}

int
710 711 712 713
virQEMUCloseCallbacksSet(virQEMUCloseCallbacksPtr closeCallbacks,
                         virDomainObjPtr vm,
                         virConnectPtr conn,
                         virQEMUCloseCallback cb)
714 715 716
{
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    qemuDriverCloseDefPtr closeDef;
717
    int ret = -1;
718 719 720 721 722

    virUUIDFormat(vm->def->uuid, uuidstr);
    VIR_DEBUG("vm=%s, uuid=%s, conn=%p, cb=%p",
              vm->def->name, uuidstr, conn, cb);

723 724 725
    virObjectLock(closeCallbacks);

    closeDef = virHashLookup(closeCallbacks->list, uuidstr);
726 727
    if (closeDef) {
        if (closeDef->conn != conn) {
728 729 730 731
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Close callback for domain %s already registered"
                             " with another connection %p"),
                           vm->def->name, closeDef->conn);
732
            goto cleanup;
733 734
        }
        if (closeDef->cb && closeDef->cb != cb) {
735 736 737
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Another close callback is already defined for"
                             " domain %s"), vm->def->name);
738
            goto cleanup;
739 740 741 742 743 744
        }

        closeDef->cb = cb;
    } else {
        if (VIR_ALLOC(closeDef) < 0) {
            virReportOOMError();
745
            goto cleanup;
746 747 748 749
        }

        closeDef->conn = conn;
        closeDef->cb = cb;
750
        if (virHashAddEntry(closeCallbacks->list, uuidstr, closeDef) < 0) {
751
            VIR_FREE(closeDef);
752
            goto cleanup;
753 754
        }
    }
755 756 757

    ret = 0;
cleanup:
758
    virObjectUnlock(closeCallbacks);
759
    return ret;
760 761 762
}

int
763 764 765
virQEMUCloseCallbacksUnset(virQEMUCloseCallbacksPtr closeCallbacks,
                           virDomainObjPtr vm,
                           virQEMUCloseCallback cb)
766 767 768
{
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    qemuDriverCloseDefPtr closeDef;
769
    int ret = -1;
770 771 772 773 774

    virUUIDFormat(vm->def->uuid, uuidstr);
    VIR_DEBUG("vm=%s, uuid=%s, cb=%p",
              vm->def->name, uuidstr, cb);

775 776 777
    virObjectLock(closeCallbacks);

    closeDef = virHashLookup(closeCallbacks->list, uuidstr);
778
    if (!closeDef)
779
        goto cleanup;
780 781

    if (closeDef->cb && closeDef->cb != cb) {
782 783 784
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       _("Trying to remove mismatching close callback for"
                         " domain %s"), vm->def->name);
785
        goto cleanup;
786 787
    }

788
    ret = virHashRemoveEntry(closeCallbacks->list, uuidstr);
789
cleanup:
790
    virObjectUnlock(closeCallbacks);
791
    return ret;
792 793
}

794 795 796 797
virQEMUCloseCallback
virQEMUCloseCallbacksGet(virQEMUCloseCallbacksPtr closeCallbacks,
                         virDomainObjPtr vm,
                         virConnectPtr conn)
798 799 800
{
    char uuidstr[VIR_UUID_STRING_BUFLEN];
    qemuDriverCloseDefPtr closeDef;
801
    virQEMUCloseCallback cb = NULL;
802 803 804 805 806

    virUUIDFormat(vm->def->uuid, uuidstr);
    VIR_DEBUG("vm=%s, uuid=%s, conn=%p",
              vm->def->name, uuidstr, conn);

807 808 809
    virObjectLock(closeCallbacks);

    closeDef = virHashLookup(closeCallbacks->list, uuidstr);
810 811 812
    if (closeDef && (!conn || closeDef->conn == conn))
        cb = closeDef->cb;

813 814
    virObjectUnlock(closeCallbacks);

815 816 817 818
    VIR_DEBUG("cb=%p", cb);
    return cb;
}

819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835

typedef struct _virQEMUCloseCallbacksListEntry virQEMUCloseCallbacksListEntry;
typedef virQEMUCloseCallbacksListEntry *virQEMUCloseCallbacksListEntryPtr;
struct _virQEMUCloseCallbacksListEntry {
    unsigned char uuid[VIR_UUID_BUFLEN];
    virQEMUCloseCallback callback;
};


typedef struct _virQEMUCloseCallbacksList virQEMUCloseCallbacksList;
typedef virQEMUCloseCallbacksList *virQEMUCloseCallbacksListPtr;
struct _virQEMUCloseCallbacksList {
    size_t nentries;
    virQEMUCloseCallbacksListEntryPtr entries;
};


836
struct virQEMUCloseCallbacksData {
837
    virConnectPtr conn;
838 839
    virQEMUCloseCallbacksListPtr list;
    bool oom;
840 841
};

842

843
static void
844
virQEMUCloseCallbacksGetOne(void *payload,
845
                            const void *key,
846
                            void *opaque)
847
{
848
    struct virQEMUCloseCallbacksData *data = opaque;
849
    qemuDriverCloseDefPtr closeDef = payload;
850 851 852 853 854
    const char *uuidstr = key;
    unsigned char uuid[VIR_UUID_BUFLEN];

    if (virUUIDParse(uuidstr, uuid) < 0)
        return;
855 856

    VIR_DEBUG("conn=%p, thisconn=%p, uuid=%s, cb=%p",
857
              closeDef->conn, data->conn, uuidstr, closeDef->cb);
858 859 860 861

    if (data->conn != closeDef->conn || !closeDef->cb)
        return;

862 863 864
    if (VIR_EXPAND_N(data->list->entries,
                     data->list->nentries, 1) < 0) {
        data->oom = true;
865 866 867
        return;
    }

868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890
    memcpy(data->list->entries[data->list->nentries - 1].uuid,
           uuid, VIR_UUID_BUFLEN);
    data->list->entries[data->list->nentries - 1].callback = closeDef->cb;
}


static virQEMUCloseCallbacksListPtr
virQEMUCloseCallbacksGetForConn(virQEMUCloseCallbacksPtr closeCallbacks,
                                virConnectPtr conn)
{
    virQEMUCloseCallbacksListPtr list = NULL;
    struct virQEMUCloseCallbacksData data;

    if (VIR_ALLOC(list) < 0) {
        virReportOOMError();
        return NULL;
    }

    data.conn = conn;
    data.list = list;
    data.oom = false;

    virHashForEach(closeCallbacks->list, virQEMUCloseCallbacksGetOne, &data);
891

892 893 894 895 896 897 898 899
    if (data.oom) {
        VIR_FREE(list->entries);
        VIR_FREE(list);
        virReportOOMError();
        return NULL;
    }

    return list;
900 901
}

902

903
void
904 905 906
virQEMUCloseCallbacksRun(virQEMUCloseCallbacksPtr closeCallbacks,
                         virConnectPtr conn,
                         virQEMUDriverPtr driver)
907
{
908 909
    virQEMUCloseCallbacksListPtr list;
    size_t i;
910

911
    VIR_DEBUG("conn=%p", conn);
912

913 914 915 916
    /* We must not hold the lock while running the callbacks,
     * so first we obtain the list of callbacks, then remove
     * them all from the hash. At that point we can release
     * the lock and run the callbacks safely. */
917

918 919 920 921 922
    virObjectLock(closeCallbacks);
    list = virQEMUCloseCallbacksGetForConn(closeCallbacks, conn);
    if (!list)
        return;

923
    for (i = 0; i < list->nentries; i++) {
924 925 926
        virHashRemoveEntry(closeCallbacks->list,
                           list->entries[i].uuid);
    }
927
    virObjectUnlock(closeCallbacks);
928

929
    for (i = 0; i < list->nentries; i++) {
930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945
        virDomainObjPtr vm;

        if (!(vm = virDomainObjListFindByUUID(driver->domains,
                                              list->entries[i].uuid))) {
            char uuidstr[VIR_UUID_STRING_BUFLEN];
            virUUIDFormat(list->entries[i].uuid, uuidstr);
            VIR_DEBUG("No domain object with UUID %s", uuidstr);
            continue;
        }

        vm = list->entries[i].callback(driver, vm, conn);
        if (vm)
            virObjectUnlock(vm);
    }
    VIR_FREE(list->entries);
    VIR_FREE(list);
946
}
947

948
struct _qemuSharedDeviceEntry {
949 950 951 952
    size_t ref;
    char **domains; /* array of domain names */
};

953
/* Construct the hash key for sharedDevices as "major:minor" */
954
char *
955
qemuGetSharedDeviceKey(const char *device_path)
956 957 958 959 960
{
    int maj, min;
    char *key = NULL;
    int rc;

961
    if ((rc = virGetDeviceID(device_path, &maj, &min)) < 0) {
962 963
        virReportSystemError(-rc,
                             _("Unable to get minor number of device '%s'"),
964
                             device_path);
965 966 967 968 969 970 971 972 973 974 975
        return NULL;
    }

    if (virAsprintf(&key, "%d:%d", maj, min) < 0) {
        virReportOOMError();
        return NULL;
    }

    return key;
}

976
/* Check if a shared device's setting conflicts with the conf
977 978
 * used by other domain(s). Currently only checks the sgio
 * setting. Note that this should only be called for disk with
979
 * block source if the device type is disk.
980 981 982 983
 *
 * Returns 0 if no conflicts, otherwise returns -1.
 */
static int
984 985
qemuCheckSharedDevice(virHashTablePtr sharedDevices,
                      virDomainDeviceDefPtr dev)
986
{
987 988
    virDomainDiskDefPtr disk = NULL;
    virDomainHostdevDefPtr hostdev = NULL;
989 990
    char *sysfs_path = NULL;
    char *key = NULL;
991 992 993
    char *hostdev_name = NULL;
    char *hostdev_path = NULL;
    char *device_path = NULL;
994
    int val;
995 996
    int ret = 0;

997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        disk = dev->data.disk;

        /* The only conflicts between shared disk we care about now
         * is sgio setting, which is only valid for device='lun'.
         */
        if (disk->device != VIR_DOMAIN_DISK_DEVICE_LUN)
            return 0;

        device_path = disk->src;
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
        hostdev = dev->data.hostdev;

        if (!(hostdev_name = virSCSIDeviceGetDevName(hostdev->source.subsys.u.scsi.adapter,
                                                     hostdev->source.subsys.u.scsi.bus,
                                                     hostdev->source.subsys.u.scsi.target,
                                                     hostdev->source.subsys.u.scsi.unit)))
            goto cleanup;

        if (virAsprintf(&hostdev_path, "/dev/%s", hostdev_name) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        device_path = hostdev_path;
    } else {
1023
        return 0;
1024
    }
1025

1026
    if (!(sysfs_path = virGetUnprivSGIOSysfsPath(device_path, NULL))) {
1027 1028 1029 1030 1031 1032 1033 1034 1035 1036
        ret = -1;
        goto cleanup;
    }

    /* It can't be conflict if unpriv_sgio is not supported
     * by kernel.
     */
    if (!virFileExists(sysfs_path))
        goto cleanup;

1037
    if (!(key = qemuGetSharedDeviceKey(device_path))) {
1038 1039 1040 1041 1042 1043 1044
        ret = -1;
        goto cleanup;
    }

    /* It can't be conflict if no other domain is
     * is sharing it.
     */
1045
    if (!(virHashLookup(sharedDevices, key)))
1046 1047
        goto cleanup;

1048
    if (virGetDeviceUnprivSGIO(device_path, NULL, &val) < 0) {
1049 1050 1051 1052 1053
        ret = -1;
        goto cleanup;
    }

    if ((val == 0 &&
1054 1055
         (disk->sgio == VIR_DOMAIN_DEVICE_SGIO_FILTERED ||
          disk->sgio == VIR_DOMAIN_DEVICE_SGIO_DEFAULT)) ||
1056
        (val == 1 &&
1057
         disk->sgio == VIR_DOMAIN_DEVICE_SGIO_UNFILTERED))
1058 1059
        goto cleanup;

1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        if (disk->type == VIR_DOMAIN_DISK_TYPE_VOLUME) {
            virReportError(VIR_ERR_OPERATION_INVALID,
                           _("sgio of shared disk 'pool=%s' 'volume=%s' conflicts "
                             "with other active domains"),
                           disk->srcpool->pool,
                           disk->srcpool->volume);
        } else {
            virReportError(VIR_ERR_OPERATION_INVALID,
                           _("sgio of shared disk '%s' conflicts with other "
                             "active domains"), disk->src);
        }
1072 1073
    } else {
        virReportError(VIR_ERR_OPERATION_INVALID,
1074 1075 1076 1077 1078 1079
                       _("sgio of shared scsi host device '%s-%d-%d-%d' conflicts "
                          "with other active domains"),
                       hostdev->source.subsys.u.scsi.adapter,
                       hostdev->source.subsys.u.scsi.bus,
                       hostdev->source.subsys.u.scsi.target,
                       hostdev->source.subsys.u.scsi.unit);
1080 1081
    }

1082 1083
    ret = -1;
cleanup:
1084 1085
    VIR_FREE(hostdev_name);
    VIR_FREE(hostdev_path);
1086 1087 1088 1089
    VIR_FREE(sysfs_path);
    VIR_FREE(key);
    return ret;
}
1090
bool
1091 1092 1093
qemuSharedDeviceEntryDomainExists(qemuSharedDeviceEntryPtr entry,
                                  const char *name,
                                  int *idx)
1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108
{
    size_t i;

    for (i = 0; i < entry->ref; i++) {
        if (STREQ(entry->domains[i], name)) {
            if (idx)
                *idx = i;
            return true;
        }
    }

    return false;
}

void
1109
qemuSharedDeviceEntryFree(void *payload, const void *name ATTRIBUTE_UNUSED)
1110
{
1111
    qemuSharedDeviceEntryPtr entry = payload;
1112 1113
    size_t i;

1114 1115 1116
    if (!entry)
        return;

1117 1118 1119 1120 1121 1122 1123
    for (i = 0; i < entry->ref; i++) {
        VIR_FREE(entry->domains[i]);
    }
    VIR_FREE(entry->domains);
    VIR_FREE(entry);
}

1124 1125
static qemuSharedDeviceEntryPtr
qemuSharedDeviceEntryCopy(const qemuSharedDeviceEntryPtr entry)
1126
{
1127
    qemuSharedDeviceEntryPtr ret = NULL;
1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140
    size_t i;

    if (VIR_ALLOC(ret) < 0) {
        virReportOOMError();
        return NULL;
    }

    if (VIR_ALLOC_N(ret->domains, entry->ref) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    for (i = 0; i < entry->ref; i++) {
1141
        if (VIR_STRDUP(ret->domains[i], entry->domains[i]) < 0)
1142 1143 1144 1145 1146 1147 1148
            goto cleanup;
        ret->ref++;
    }

    return ret;

cleanup:
1149
    qemuSharedDeviceEntryFree(ret, NULL);
1150 1151 1152
    return NULL;
}

1153
/* qemuAddSharedDevice:
1154
 * @driver: Pointer to qemu driver struct
1155
 * @dev: The device def
1156 1157 1158
 * @name: The domain name
 *
 * Increase ref count and add the domain name into the list which
1159
 * records all the domains that use the shared device if the entry
1160
 * already exists, otherwise add a new entry.
1161 1162
 */
int
1163 1164 1165
qemuAddSharedDevice(virQEMUDriverPtr driver,
                    virDomainDeviceDefPtr dev,
                    const char *name)
1166
{
1167 1168
    qemuSharedDeviceEntry *entry = NULL;
    qemuSharedDeviceEntry *new_entry = NULL;
1169 1170 1171 1172
    virDomainDiskDefPtr disk = NULL;
    virDomainHostdevDefPtr hostdev = NULL;
    char *dev_name = NULL;
    char *dev_path = NULL;
1173
    char *key = NULL;
1174
    int ret = -1;
1175

1176 1177 1178
    /* Currently the only conflicts we have to care about for
     * the shared disk and shared host device is "sgio" setting,
     * which is only valid for block disk and scsi host device.
1179
     */
1180 1181 1182
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        disk = dev->data.disk;

1183
        if (!disk->shared ||
1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197
            !disk->src ||
            (disk->type != VIR_DOMAIN_DISK_TYPE_BLOCK &&
             !(disk->type == VIR_DOMAIN_DISK_TYPE_VOLUME &&
               disk->srcpool &&
               disk->srcpool->voltype == VIR_STORAGE_VOL_BLOCK)))
            return 0;
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
        hostdev = dev->data.hostdev;

        if (!hostdev->shareable ||
            !(hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS &&
              hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI))
            return 0;
    } else {
1198
        return 0;
1199
    }
1200

1201
    qemuDriverLock(driver);
1202 1203
    if (qemuCheckSharedDevice(driver->sharedDevices, dev) < 0)
        goto cleanup;
1204

1205
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222
        if (!(key = qemuGetSharedDeviceKey(disk->src)))
            goto cleanup;
    } else {
        if (!(dev_name = virSCSIDeviceGetDevName(hostdev->source.subsys.u.scsi.adapter,
                                                 hostdev->source.subsys.u.scsi.bus,
                                                 hostdev->source.subsys.u.scsi.target,
                                                 hostdev->source.subsys.u.scsi.unit)))
            goto cleanup;

        if (virAsprintf(&dev_path, "/dev/%s", dev_name) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        if (!(key = qemuGetSharedDeviceKey(dev_path)))
            goto cleanup;
    }
1223

1224
    if ((entry = virHashLookup(driver->sharedDevices, key))) {
1225 1226
        /* Nothing to do if the shared scsi host device is already
         * recorded in the table.
1227
         */
1228
        if (qemuSharedDeviceEntryDomainExists(entry, name, NULL)) {
1229 1230 1231 1232
            ret = 0;
            goto cleanup;
        }

1233
        if (!(new_entry = qemuSharedDeviceEntryCopy(entry)))
1234 1235
            goto cleanup;

1236 1237
        if (VIR_EXPAND_N(new_entry->domains, new_entry->ref, 1) < 0 ||
            VIR_STRDUP(new_entry->domains[new_entry->ref - 1], name) < 0) {
1238
            qemuSharedDeviceEntryFree(new_entry, NULL);
1239 1240 1241 1242
            virReportOOMError();
            goto cleanup;
        }

1243 1244
        if (virHashUpdateEntry(driver->sharedDevices, key, new_entry) < 0) {
            qemuSharedDeviceEntryFree(new_entry, NULL);
1245 1246
            goto cleanup;
        }
1247
    } else {
1248 1249 1250
        if (VIR_ALLOC(entry) < 0 ||
            VIR_ALLOC_N(entry->domains, 1) < 0 ||
            VIR_STRDUP(entry->domains[0], name) < 0) {
1251
            qemuSharedDeviceEntryFree(entry, NULL);
1252 1253 1254 1255 1256 1257
            virReportOOMError();
            goto cleanup;
        }

        entry->ref = 1;

1258
        if (virHashAddEntry(driver->sharedDevices, key, entry))
1259
            goto cleanup;
1260 1261
    }

1262 1263
    ret = 0;
cleanup:
1264
    qemuDriverUnlock(driver);
1265 1266
    VIR_FREE(dev_name);
    VIR_FREE(dev_path);
1267
    VIR_FREE(key);
1268
    return ret;
1269 1270
}

1271
/* qemuRemoveSharedDevice:
1272
 * @driver: Pointer to qemu driver struct
1273
 * @device: The device def
1274 1275 1276
 * @name: The domain name
 *
 * Decrease ref count and remove the domain name from the list which
1277 1278
 * records all the domains that use the shared device if ref is not
 * 1, otherwise remove the entry.
1279 1280
 */
int
1281 1282 1283
qemuRemoveSharedDevice(virQEMUDriverPtr driver,
                       virDomainDeviceDefPtr dev,
                       const char *name)
1284
{
1285 1286
    qemuSharedDeviceEntryPtr entry = NULL;
    qemuSharedDeviceEntryPtr new_entry = NULL;
1287 1288
    virDomainDiskDefPtr disk = NULL;
    virDomainHostdevDefPtr hostdev = NULL;
1289
    char *key = NULL;
1290 1291
    char *dev_name = NULL;
    char *dev_path = NULL;
1292
    int ret = -1;
1293
    int idx;
1294

1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        disk = dev->data.disk;

        if (!disk->shared ||
            !disk->src ||
            (disk->type != VIR_DOMAIN_DISK_TYPE_BLOCK &&
             !(disk->type == VIR_DOMAIN_DISK_TYPE_VOLUME &&
               disk->srcpool &&
               disk->srcpool->voltype == VIR_STORAGE_VOL_BLOCK)))
            return 0;
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
        hostdev = dev->data.hostdev;

        if (!hostdev->shareable ||
            !(hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS &&
              hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI))
            return 0;
    } else {
1313
        return 0;
1314
    }
1315

1316
    qemuDriverLock(driver);
1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335

    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        if (!(key = qemuGetSharedDeviceKey(disk->src)))
            goto cleanup;
    } else {
        if (!(dev_name = virSCSIDeviceGetDevName(hostdev->source.subsys.u.scsi.adapter,
                                                 hostdev->source.subsys.u.scsi.bus,
                                                 hostdev->source.subsys.u.scsi.target,
                                                 hostdev->source.subsys.u.scsi.unit)))
            goto cleanup;

        if (virAsprintf(&dev_path, "/dev/%s", dev_name) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        if (!(key = qemuGetSharedDeviceKey(dev_path)))
            goto cleanup;
    }
1336

1337
    if (!(entry = virHashLookup(driver->sharedDevices, key)))
1338
        goto cleanup;
1339

1340 1341 1342
    /* Nothing to do if the shared disk is not recored in
     * the table.
     */
1343
    if (!qemuSharedDeviceEntryDomainExists(entry, name, &idx)) {
1344 1345 1346 1347 1348
        ret = 0;
        goto cleanup;
    }

    if (entry->ref != 1) {
1349
        if (!(new_entry = qemuSharedDeviceEntryCopy(entry)))
1350 1351 1352 1353 1354 1355 1356 1357 1358
            goto cleanup;

        if (idx != new_entry->ref - 1)
            memmove(&new_entry->domains[idx],
                    &new_entry->domains[idx + 1],
                    sizeof(*new_entry->domains) * (new_entry->ref - idx - 1));

        VIR_SHRINK_N(new_entry->domains, new_entry->ref, 1);

1359 1360
        if (virHashUpdateEntry(driver->sharedDevices, key, new_entry) < 0){
            qemuSharedDeviceEntryFree(new_entry, NULL);
1361
            goto cleanup;
1362
        }
1363
    } else {
1364
        if (virHashRemoveEntry(driver->sharedDevices, key) < 0)
1365
            goto cleanup;
1366 1367
    }

1368 1369
    ret = 0;
cleanup:
1370
    qemuDriverUnlock(driver);
1371 1372
    VIR_FREE(dev_name);
    VIR_FREE(dev_path);
1373
    VIR_FREE(key);
1374
    return ret;
1375
}
1376

1377
int
1378
qemuSetUnprivSGIO(virDomainDeviceDefPtr dev)
1379
{
1380 1381
    virDomainDiskDefPtr disk = NULL;
    virDomainHostdevDefPtr hostdev = NULL;
1382
    char *sysfs_path = NULL;
1383 1384 1385
    char *path = NULL;
    char *hostdev_name = NULL;
    char *hostdev_path = NULL;
1386 1387 1388 1389 1390 1391
    int val = -1;
    int ret = 0;

    /* "sgio" is only valid for block disk; cdrom
     * and floopy disk can have empty source.
     */
1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424
    if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
        disk = dev->data.disk;

        if (!disk->src ||
            disk->device != VIR_DOMAIN_DISK_DEVICE_LUN ||
            (disk->type != VIR_DOMAIN_DISK_TYPE_BLOCK &&
             !(disk->type == VIR_DOMAIN_DISK_TYPE_VOLUME &&
               disk->srcpool &&
               disk->srcpool->voltype == VIR_STORAGE_VOL_BLOCK)))
            return 0;

        path = disk->src;
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
        hostdev = dev->data.hostdev;

        if (!hostdev->shareable ||
            !(hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS &&
              hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI))
            return 0;

        if (!(hostdev_name = virSCSIDeviceGetDevName(hostdev->source.subsys.u.scsi.adapter,
                                                     hostdev->source.subsys.u.scsi.bus,
                                                     hostdev->source.subsys.u.scsi.target,
                                                     hostdev->source.subsys.u.scsi.unit)))
            goto cleanup;

        if (virAsprintf(&hostdev_path, "/dev/%s", hostdev_name) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        path = hostdev_path;
    } else {
1425
        return 0;
1426
    }
1427

1428 1429 1430 1431 1432
    sysfs_path = virGetUnprivSGIOSysfsPath(path, NULL);
    if (sysfs_path == NULL) {
        ret = -1;
        goto cleanup;
    }
1433 1434

    /* By default, filter the SG_IO commands, i.e. set unpriv_sgio to 0.  */
1435 1436 1437 1438 1439 1440

    if (dev->type == VIR_DOMAIN_DEVICE_DISK)
        val = (disk->sgio == VIR_DOMAIN_DEVICE_SGIO_UNFILTERED);
    else
        val = (hostdev->source.subsys.u.scsi.sgio ==
               VIR_DOMAIN_DEVICE_SGIO_UNFILTERED);
1441 1442 1443 1444 1445 1446

    /* Do not do anything if unpriv_sgio is not supported by the kernel and the
     * whitelist is enabled.  But if requesting unfiltered access, always call
     * virSetDeviceUnprivSGIO, to report an error for unsupported unpriv_sgio.
     */
    if ((virFileExists(sysfs_path) || val == 1) &&
1447
        virSetDeviceUnprivSGIO(path, NULL, val) < 0)
1448 1449
        ret = -1;

1450
cleanup:
1451
    VIR_FREE(sysfs_path);
1452 1453
    VIR_FREE(hostdev_name);
    VIR_FREE(hostdev_path);
1454 1455
    return ret;
}
1456

1457 1458 1459 1460
int qemuDriverAllocateID(virQEMUDriverPtr driver)
{
    return virAtomicIntInc(&driver->nextvmid);
}
1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512

int
qemuTranslateDiskSourcePool(virConnectPtr conn,
                            virDomainDiskDefPtr def)
{
    virStoragePoolPtr pool = NULL;
    virStorageVolPtr vol = NULL;
    virStorageVolInfo info;
    int ret = -1;

    if (def->type != VIR_DOMAIN_DISK_TYPE_VOLUME)
        return 0;

    if (!def->srcpool)
        return 0;

    if (!(pool = virStoragePoolLookupByName(conn, def->srcpool->pool)))
        return -1;

    if (!(vol = virStorageVolLookupByName(pool, def->srcpool->volume)))
        goto cleanup;

    if (virStorageVolGetInfo(vol, &info) < 0)
        goto cleanup;

    if (def->startupPolicy &&
        info.type != VIR_STORAGE_VOL_FILE) {
        virReportError(VIR_ERR_XML_ERROR, "%s",
                       _("'startupPolicy' is only valid for 'file' type volume"));
        goto cleanup;
    }

    switch (info.type) {
    case VIR_STORAGE_VOL_FILE:
    case VIR_STORAGE_VOL_BLOCK:
    case VIR_STORAGE_VOL_DIR:
        if (!(def->src = virStorageVolGetPath(vol)))
            goto cleanup;
        break;
    case VIR_STORAGE_VOL_NETWORK:
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("Using network volume as disk source is not supported"));
        goto cleanup;
    }

    def->srcpool->voltype = info.type;
    ret = 0;
cleanup:
    virStoragePoolFree(pool);
    virStorageVolFree(vol);
    return ret;
}