qemu_cgroup.c 38.6 KB
Newer Older
1 2 3
/*
 * qemu_cgroup.c: QEMU cgroup management
 *
4
 * Copyright (C) 2006-2015 Red Hat, Inc.
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/>.
20 21 22 23 24 25 26
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

#include <config.h>

#include "qemu_cgroup.h"
27
#include "qemu_domain.h"
28
#include "qemu_process.h"
29
#include "vircgroup.h"
30
#include "virlog.h"
31
#include "viralloc.h"
32
#include "virerror.h"
33
#include "domain_audit.h"
34
#include "virscsi.h"
35
#include "virstring.h"
36
#include "virfile.h"
37
#include "virtypedparam.h"
38
#include "virnuma.h"
39
#include "virsystemd.h"
40 41 42

#define VIR_FROM_THIS VIR_FROM_QEMU

43 44
VIR_LOG_INIT("qemu.qemu_cgroup");

45
const char *const defaultDeviceACL[] = {
46 47 48
    "/dev/null", "/dev/full", "/dev/zero",
    "/dev/random", "/dev/urandom",
    "/dev/ptmx", "/dev/kvm", "/dev/kqemu",
49
    "/dev/rtc", "/dev/hpet", "/dev/vfio/vfio",
50 51 52 53 54
    NULL,
};
#define DEVICE_PTY_MAJOR 136
#define DEVICE_SND_MAJOR 116

55

56
static int
57 58 59
qemuSetupImagePathCgroup(virDomainObjPtr vm,
                         const char *path,
                         bool readonly)
60
{
61
    qemuDomainObjPrivatePtr priv = vm->privateData;
62
    int perms = VIR_CGROUP_DEVICE_READ;
63
    int ret;
64

65
    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_DEVICES))
66 67
        return 0;

68
    if (!readonly)
69
        perms |= VIR_CGROUP_DEVICE_WRITE;
70

71
    VIR_DEBUG("Allow path %s, perms: %s",
72
              path, virCgroupGetDevicePermsString(perms));
73

74
    ret = virCgroupAllowDevicePath(priv->cgroup, path, perms, true);
75

76
    virDomainAuditCgroupPath(vm, priv->cgroup, "allow", path,
77 78
                             virCgroupGetDevicePermsString(perms),
                             ret == 0);
79 80

    return ret;
81 82 83
}


84 85 86 87 88 89 90 91 92 93 94 95 96 97 98
static int
qemuSetupImageCgroupInternal(virDomainObjPtr vm,
                             virStorageSourcePtr src,
                             bool forceReadonly)
{
    if (!src->path || !virStorageSourceIsLocalStorage(src)) {
        VIR_DEBUG("Not updating cgroups for disk path '%s', type: %s",
                  NULLSTR(src->path), virStorageTypeToString(src->type));
        return 0;
    }

    return qemuSetupImagePathCgroup(vm, src->path, src->readonly || forceReadonly);
}


99
int
100 101
qemuSetupImageCgroup(virDomainObjPtr vm,
                     virStorageSourcePtr src)
102
{
103
    return qemuSetupImageCgroupInternal(vm, src, false);
104 105 106 107 108 109 110
}


int
qemuTeardownImageCgroup(virDomainObjPtr vm,
                        virStorageSourcePtr src)
{
111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int perms = VIR_CGROUP_DEVICE_READ |
                VIR_CGROUP_DEVICE_WRITE |
                VIR_CGROUP_DEVICE_MKNOD;
    int ret;

    if (!virCgroupHasController(priv->cgroup,
                                VIR_CGROUP_CONTROLLER_DEVICES))
        return 0;

    if (!src->path || !virStorageSourceIsLocalStorage(src)) {
        VIR_DEBUG("Not updating cgroups for disk path '%s', type: %s",
                  NULLSTR(src->path), virStorageTypeToString(src->type));
        return 0;
    }

    VIR_DEBUG("Deny path %s", src->path);

    ret = virCgroupDenyDevicePath(priv->cgroup, src->path, perms, true);

    virDomainAuditCgroupPath(vm, priv->cgroup, "deny", src->path,
                             virCgroupGetDevicePermsString(perms), ret == 0);

    return ret;
135 136 137
}


138 139 140
int
qemuSetupDiskCgroup(virDomainObjPtr vm,
                    virDomainDiskDefPtr disk)
141
{
142
    virStorageSourcePtr next;
143
    bool forceReadonly = false;
144

145
    for (next = disk->src; next; next = next->backingStore) {
146
        if (qemuSetupImageCgroupInternal(vm, next, forceReadonly) < 0)
147
            return -1;
148 149 150

        /* setup only the top level image for read-write */
        forceReadonly = true;
151
    }
152 153

    return 0;
154 155 156
}


157 158 159
int
qemuTeardownDiskCgroup(virDomainObjPtr vm,
                       virDomainDiskDefPtr disk)
160
{
161
    virStorageSourcePtr next;
162

163
    for (next = disk->src; next; next = next->backingStore) {
164
        if (qemuTeardownImageCgroup(vm, next) < 0)
165 166
            return -1;
    }
167

168
    return 0;
169 170
}

171

172
static int
173
qemuSetupChrSourceCgroup(virDomainObjPtr vm,
174
                         virDomainChrSourceDefPtr source)
175
{
176
    qemuDomainObjPrivatePtr priv = vm->privateData;
177
    int ret;
178

179
    if (source->type != VIR_DOMAIN_CHR_TYPE_DEV)
180 181
        return 0;

182
    VIR_DEBUG("Process path '%s' for device", source->data.file.path);
183

184
    ret = virCgroupAllowDevicePath(priv->cgroup, source->data.file.path,
185
                                   VIR_CGROUP_DEVICE_RW, false);
186
    virDomainAuditCgroupPath(vm, priv->cgroup, "allow",
187
                             source->data.file.path, "rw", ret == 0);
188

189
    return ret;
190 191
}

192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213

static int
qemuTeardownChrSourceCgroup(virDomainObjPtr vm,
                            virDomainChrSourceDefPtr source)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret;

    if (source->type != VIR_DOMAIN_CHR_TYPE_DEV)
        return 0;

    VIR_DEBUG("Process path '%s' for device", source->data.file.path);

    ret = virCgroupDenyDevicePath(priv->cgroup, source->data.file.path,
                                  VIR_CGROUP_DEVICE_RW, false);
    virDomainAuditCgroupPath(vm, priv->cgroup, "deny",
                             source->data.file.path, "rw", ret == 0);

    return ret;
}


214
static int
215 216 217
qemuSetupChardevCgroupCB(virDomainDefPtr def ATTRIBUTE_UNUSED,
                         virDomainChrDefPtr dev,
                         void *opaque)
218
{
219 220
    virDomainObjPtr vm = opaque;

221
    return qemuSetupChrSourceCgroup(vm, dev->source);
222 223 224 225
}


static int
226
qemuSetupTPMCgroup(virDomainObjPtr vm)
227
{
228
    int ret = 0;
229
    virDomainTPMDefPtr dev = vm->def->tpm;
230 231 232

    switch (dev->type) {
    case VIR_DOMAIN_TPM_TYPE_PASSTHROUGH:
233
        ret = qemuSetupChrSourceCgroup(vm, &dev->data.passthrough.source);
234 235 236 237 238
        break;
    case VIR_DOMAIN_TPM_TYPE_LAST:
        break;
    }

239
    return ret;
240 241
}

242

243 244 245 246 247 248 249 250 251 252 253
static int
qemuSetupInputCgroup(virDomainObjPtr vm,
                     virDomainInputDefPtr dev)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = 0;

    switch (dev->type) {
    case VIR_DOMAIN_INPUT_TYPE_PASSTHROUGH:
        VIR_DEBUG("Process path '%s' for input device", dev->source.evdev);
        ret = virCgroupAllowDevicePath(priv->cgroup, dev->source.evdev,
254
                                       VIR_CGROUP_DEVICE_RW, false);
255 256 257 258 259 260 261 262
        virDomainAuditCgroupPath(vm, priv->cgroup, "allow", dev->source.evdev, "rw", ret == 0);
        break;
    }

    return ret;
}


263
static int
264
qemuSetupHostSCSIDeviceCgroup(virSCSIDevicePtr dev ATTRIBUTE_UNUSED,
265 266 267 268 269
                              const char *path,
                              void *opaque)
{
    virDomainObjPtr vm = opaque;
    qemuDomainObjPrivatePtr priv = vm->privateData;
270
    int ret;
271 272 273

    VIR_DEBUG("Process path '%s' for SCSI device", path);

274 275 276
    ret = virCgroupAllowDevicePath(priv->cgroup, path,
                                   virSCSIDeviceGetReadonly(dev) ?
                                   VIR_CGROUP_DEVICE_READ :
277
                                   VIR_CGROUP_DEVICE_RW, false);
278 279

    virDomainAuditCgroupPath(vm, priv->cgroup, "allow", path,
280
                             virSCSIDeviceGetReadonly(dev) ? "r" : "rw", ret == 0);
281

282
    return ret;
283
}
284

285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
static int
qemuSetupHostSCSIVHostDeviceCgroup(virSCSIVHostDevicePtr dev ATTRIBUTE_UNUSED,
                                   const char *path,
                                   void *opaque)
{
    virDomainObjPtr vm = opaque;
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret;

    VIR_DEBUG("Process path '%s' for scsi_host device", path);

    ret = virCgroupAllowDevicePath(priv->cgroup, path,
                                   VIR_CGROUP_DEVICE_RW, false);

    virDomainAuditCgroupPath(vm, priv->cgroup, "allow", path, "rw", ret == 0);

    return ret;
}

304
int
305
qemuSetupHostdevCgroup(virDomainObjPtr vm,
306 307 308 309
                       virDomainHostdevDefPtr dev)
{
    int ret = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;
310
    virDomainHostdevSubsysUSBPtr usbsrc = &dev->source.subsys.u.usb;
311
    virDomainHostdevSubsysPCIPtr pcisrc = &dev->source.subsys.u.pci;
312
    virDomainHostdevSubsysSCSIPtr scsisrc = &dev->source.subsys.u.scsi;
313
    virDomainHostdevSubsysSCSIVHostPtr hostsrc = &dev->source.subsys.u.scsi_host;
314
    virPCIDevicePtr pci = NULL;
315
    virUSBDevicePtr usb = NULL;
316
    virSCSIDevicePtr scsi = NULL;
317
    virSCSIVHostDevicePtr host = NULL;
318
    char *path = NULL;
319
    int rv;
320 321 322 323 324 325 326 327 328 329 330

    /* currently this only does something for PCI devices using vfio
     * for device assignment, but it is called for *all* hostdev
     * devices.
     */

    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_DEVICES))
        return 0;

    if (dev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {

331
        switch ((virDomainHostdevSubsysType) dev->source.subsys.type) {
332
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI:
333 334 335 336 337
            if (pcisrc->backend == VIR_DOMAIN_HOSTDEV_PCI_BACKEND_VFIO) {
                pci = virPCIDeviceNew(pcisrc->addr.domain,
                                      pcisrc->addr.bus,
                                      pcisrc->addr.slot,
                                      pcisrc->addr.function);
338 339 340
                if (!pci)
                    goto cleanup;

341
                if (!(path = virPCIDeviceGetIOMMUGroupDev(pci)))
342 343 344
                    goto cleanup;

                VIR_DEBUG("Cgroup allow %s for PCI device assignment", path);
345
                rv = virCgroupAllowDevicePath(priv->cgroup, path,
346
                                              VIR_CGROUP_DEVICE_RW, false);
347
                virDomainAuditCgroupPath(vm, priv->cgroup,
348 349
                                         "allow", path, "rw", rv == 0);
                if (rv < 0)
350 351 352
                    goto cleanup;
            }
            break;
353 354 355 356 357 358 359 360

        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB:
            /* NB: hostdev->missing wasn't previously checked in the
             * case of hotplug, only when starting a domain. Now it is
             * always checked, and the cgroup setup skipped if true.
             */
            if (dev->missing)
                break;
361
            if ((usb = virUSBDeviceNew(usbsrc->bus, usbsrc->device,
362 363 364 365
                                       NULL)) == NULL) {
                goto cleanup;
            }

366 367 368 369 370 371 372 373
            if (VIR_STRDUP(path, virUSBDeviceGetPath(usb)) < 0)
                goto cleanup;

            VIR_DEBUG("Process path '%s' for USB device", path);
            rv = virCgroupAllowDevicePath(priv->cgroup, path,
                                          VIR_CGROUP_DEVICE_RW, false);
            virDomainAuditCgroupPath(vm, priv->cgroup, "allow", path, "rw", rv == 0);
            if (rv < 0)
374 375
                goto cleanup;
            break;
376

377
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI: {
378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396
            if (scsisrc->protocol ==
                VIR_DOMAIN_HOSTDEV_SCSI_PROTOCOL_TYPE_ISCSI) {
                virDomainHostdevSubsysSCSIiSCSIPtr iscsisrc = &scsisrc->u.iscsi;
                /* Follow qemuSetupDiskCgroup() and qemuSetImageCgroupInternal()
                 * which does nothing for non local storage
                 */
                VIR_DEBUG("Not updating cgroups for hostdev iSCSI path '%s'",
                          iscsisrc->path);
            } else {
                virDomainHostdevSubsysSCSIHostPtr scsihostsrc =
                    &scsisrc->u.host;
                if ((scsi = virSCSIDeviceNew(NULL,
                                             scsihostsrc->adapter,
                                             scsihostsrc->bus,
                                             scsihostsrc->target,
                                             scsihostsrc->unit,
                                             dev->readonly,
                                             dev->shareable)) == NULL)
                    goto cleanup;
397

398 399 400 401 402
                if (virSCSIDeviceFileIterate(scsi,
                                             qemuSetupHostSCSIDeviceCgroup,
                                             vm) < 0)
                    goto cleanup;
            }
403 404
            break;
        }
405

406
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI_HOST: {
407 408 409 410 411 412 413 414 415 416
            if (hostsrc->protocol ==
                VIR_DOMAIN_HOSTDEV_SUBSYS_SCSI_HOST_PROTOCOL_TYPE_VHOST) {
                if (!(host = virSCSIVHostDeviceNew(hostsrc->wwpn)))
                    goto cleanup;

                if (virSCSIVHostDeviceFileIterate(host,
                                             qemuSetupHostSCSIVHostDeviceCgroup,
                                             vm) < 0)
                    goto cleanup;
            }
417 418 419
            break;
        }

420
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_LAST:
421 422 423 424 425
            break;
        }
    }

    ret = 0;
426
 cleanup:
427
    virPCIDeviceFree(pci);
428
    virUSBDeviceFree(usb);
429
    virSCSIDeviceFree(scsi);
430
    virSCSIVHostDeviceFree(host);
431 432 433 434 435 436 437 438 439 440
    VIR_FREE(path);
    return ret;
}

int
qemuTeardownHostdevCgroup(virDomainObjPtr vm,
                       virDomainHostdevDefPtr dev)
{
    int ret = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;
441
    virDomainHostdevSubsysPCIPtr pcisrc = &dev->source.subsys.u.pci;
442 443 444 445 446 447 448 449 450 451 452 453 454
    virPCIDevicePtr pci = NULL;
    char *path = NULL;

    /* currently this only does something for PCI devices using vfio
     * for device assignment, but it is called for *all* hostdev
     * devices.
     */

    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_DEVICES))
        return 0;

    if (dev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {

455
        switch ((virDomainHostdevSubsysType) dev->source.subsys.type) {
456
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_PCI:
457
            if (pcisrc->backend == VIR_DOMAIN_HOSTDEV_PCI_BACKEND_VFIO) {
458
                int rv;
459

460 461 462 463
                pci = virPCIDeviceNew(pcisrc->addr.domain,
                                      pcisrc->addr.bus,
                                      pcisrc->addr.slot,
                                      pcisrc->addr.function);
464 465 466
                if (!pci)
                    goto cleanup;

467
                if (!(path = virPCIDeviceGetIOMMUGroupDev(pci)))
468 469 470
                    goto cleanup;

                VIR_DEBUG("Cgroup deny %s for PCI device assignment", path);
471
                rv = virCgroupDenyDevicePath(priv->cgroup, path,
472
                                             VIR_CGROUP_DEVICE_RWM, false);
473
                virDomainAuditCgroupPath(vm, priv->cgroup,
474 475
                                         "deny", path, "rwm", rv == 0);
                if (rv < 0)
476 477 478
                    goto cleanup;
            }
            break;
479 480 481
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB:
            /* nothing to tear down for USB */
            break;
482 483 484
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI:
            /* nothing to tear down for SCSI */
            break;
485 486 487
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_SCSI_HOST:
            /* nothing to tear down for scsi_host */
            break;
488
        case VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_LAST:
489 490 491 492 493
            break;
        }
    }

    ret = 0;
494
 cleanup:
495 496 497 498 499
    virPCIDeviceFree(pci);
    VIR_FREE(path);
    return ret;
}

500 501 502 503
static int
qemuSetupBlkioCgroup(virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
504
    size_t i;
505 506 507 508 509 510 511 512 513 514 515 516

    if (!virCgroupHasController(priv->cgroup,
                                VIR_CGROUP_CONTROLLER_BLKIO)) {
        if (vm->def->blkio.weight || vm->def->blkio.ndevices) {
            virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                           _("Block I/O tuning is not available on this host"));
            return -1;
        } else {
            return 0;
        }
    }

517 518 519
    if (vm->def->blkio.weight != 0 &&
        virCgroupSetBlkioWeight(priv->cgroup, vm->def->blkio.weight) < 0)
        return -1;
520 521 522

    if (vm->def->blkio.ndevices) {
        for (i = 0; i < vm->def->blkio.ndevices; i++) {
523
            virBlkioDevicePtr dev = &vm->def->blkio.devices[i];
524 525
            if (dev->weight &&
                (virCgroupSetBlkioDeviceWeight(priv->cgroup, dev->path,
526 527 528
                                               dev->weight) < 0 ||
                 virCgroupGetBlkioDeviceWeight(priv->cgroup, dev->path,
                                               &dev->weight) < 0))
529 530 531 532
                return -1;

            if (dev->riops &&
                (virCgroupSetBlkioDeviceReadIops(priv->cgroup, dev->path,
533 534 535
                                                 dev->riops) < 0 ||
                 virCgroupGetBlkioDeviceReadIops(priv->cgroup, dev->path,
                                                 &dev->riops) < 0))
536 537 538 539
                return -1;

            if (dev->wiops &&
                (virCgroupSetBlkioDeviceWriteIops(priv->cgroup, dev->path,
540 541 542
                                                  dev->wiops) < 0 ||
                 virCgroupGetBlkioDeviceWriteIops(priv->cgroup, dev->path,
                                                  &dev->wiops) < 0))
543 544 545 546
                return -1;

            if (dev->rbps &&
                (virCgroupSetBlkioDeviceReadBps(priv->cgroup, dev->path,
547 548 549
                                                dev->rbps) < 0 ||
                 virCgroupGetBlkioDeviceReadBps(priv->cgroup, dev->path,
                                                &dev->rbps) < 0))
550 551 552 553
                return -1;

            if (dev->wbps &&
                (virCgroupSetBlkioDeviceWriteBps(priv->cgroup, dev->path,
554 555 556
                                                 dev->wbps) < 0 ||
                 virCgroupGetBlkioDeviceWriteBps(priv->cgroup, dev->path,
                                                 &dev->wbps) < 0))
557 558 559 560 561 562 563
                return -1;
        }
    }

    return 0;
}

564

565 566 567 568 569
static int
qemuSetupMemoryCgroup(virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

E
Eric Blake 已提交
570
    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_MEMORY)) {
571 572 573
        if (virMemoryLimitIsSet(vm->def->mem.hard_limit) ||
            virMemoryLimitIsSet(vm->def->mem.soft_limit) ||
            virMemoryLimitIsSet(vm->def->mem.swap_hard_limit)) {
574 575 576
            virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                           _("Memory cgroup is not available on this host"));
            return -1;
O
Osier Yang 已提交
577 578
        } else {
            return 0;
579 580 581
        }
    }

582 583 584
    if (virMemoryLimitIsSet(vm->def->mem.hard_limit))
        if (virCgroupSetMemoryHardLimit(priv->cgroup, vm->def->mem.hard_limit) < 0)
            return -1;
585

586 587 588
    if (virMemoryLimitIsSet(vm->def->mem.soft_limit))
        if (virCgroupSetMemorySoftLimit(priv->cgroup, vm->def->mem.soft_limit) < 0)
            return -1;
589

590 591 592
    if (virMemoryLimitIsSet(vm->def->mem.swap_hard_limit))
        if (virCgroupSetMemSwapHardLimit(priv->cgroup, vm->def->mem.swap_hard_limit) < 0)
            return -1;
593 594 595 596 597

    return 0;
}


598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616
static int
qemuSetupFirmwareCgroup(virDomainObjPtr vm)
{
    if (!vm->def->os.loader)
        return 0;

    if (vm->def->os.loader->path &&
        qemuSetupImagePathCgroup(vm, vm->def->os.loader->path,
                                 vm->def->os.loader->readonly == VIR_TRISTATE_BOOL_YES) < 0)
        return -1;

    if (vm->def->os.loader->nvram &&
        qemuSetupImagePathCgroup(vm, vm->def->os.loader->nvram, false) < 0)
        return -1;

    return 0;
}


617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664
int
qemuSetupRNGCgroup(virDomainObjPtr vm,
                   virDomainRNGDefPtr rng)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int rv;

    if (rng->backend == VIR_DOMAIN_RNG_BACKEND_RANDOM) {
        VIR_DEBUG("Setting Cgroup ACL for RNG device");
        rv = virCgroupAllowDevicePath(priv->cgroup,
                                      rng->source.file,
                                      VIR_CGROUP_DEVICE_RW, false);
        virDomainAuditCgroupPath(vm, priv->cgroup, "allow",
                                 rng->source.file,
                                 "rw", rv == 0);
        if (rv < 0 &&
            !virLastErrorIsSystemErrno(ENOENT))
            return -1;
    }

    return 0;
}


int
qemuTeardownRNGCgroup(virDomainObjPtr vm,
                      virDomainRNGDefPtr rng)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int rv;

    if (rng->backend == VIR_DOMAIN_RNG_BACKEND_RANDOM) {
        VIR_DEBUG("Tearing down Cgroup ACL for RNG device");
        rv = virCgroupDenyDevicePath(priv->cgroup,
                                     rng->source.file,
                                     VIR_CGROUP_DEVICE_RW, false);
        virDomainAuditCgroupPath(vm, priv->cgroup, "deny",
                                 rng->source.file,
                                 "rw", rv == 0);
        if (rv < 0 &&
            !virLastErrorIsSystemErrno(ENOENT))
            return -1;
    }

    return 0;
}


665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680
int
qemuSetupChardevCgroup(virDomainObjPtr vm,
                       virDomainChrDefPtr dev)
{
    return qemuSetupChrSourceCgroup(vm, dev->source);
}


int
qemuTeardownChardevCgroup(virDomainObjPtr vm,
                          virDomainChrDefPtr dev)
{
    return qemuTeardownChrSourceCgroup(vm, dev->source);
}


681 682 683 684 685 686 687
static int
qemuSetupDevicesCgroup(virQEMUDriverPtr driver,
                       virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    virQEMUDriverConfigPtr cfg = NULL;
    const char *const *deviceACL = NULL;
688
    int rv = -1;
689
    int ret = -1;
690
    size_t i;
691 692 693 694

    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_DEVICES))
        return 0;

695 696 697 698 699
    rv = virCgroupDenyAllDevices(priv->cgroup);
    virDomainAuditCgroup(vm, priv->cgroup, "deny", "all", rv == 0);
    if (rv < 0) {
        if (virLastErrorIsSystemErrno(EPERM)) {
            virResetLastError();
700 701 702 703 704 705 706
            VIR_WARN("Group devices ACL is not accessible, disabling whitelisting");
            return 0;
        }

        goto cleanup;
    }

707 708 709
    if (qemuSetupFirmwareCgroup(vm) < 0)
        goto cleanup;

710
    for (i = 0; i < vm->def->ndisks; i++) {
711 712 713 714
        if (qemuSetupDiskCgroup(vm, vm->def->disks[i]) < 0)
            goto cleanup;
    }

715 716
    rv = virCgroupAllowDevice(priv->cgroup, 'c', DEVICE_PTY_MAJOR, -1,
                              VIR_CGROUP_DEVICE_RW);
717
    virDomainAuditCgroupMajor(vm, priv->cgroup, "allow", DEVICE_PTY_MAJOR,
718 719
                              "pty", "rw", rv == 0);
    if (rv < 0)
720 721 722 723 724 725 726 727
        goto cleanup;

    cfg = virQEMUDriverGetConfig(driver);
    deviceACL = cfg->cgroupDeviceACL ?
                (const char *const *)cfg->cgroupDeviceACL :
                defaultDeviceACL;

    if (vm->def->nsounds &&
728
        ((!vm->def->ngraphics && cfg->nogfxAllowHostAudio) ||
729 730
         (vm->def->graphics &&
          ((vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
731
           cfg->vncAllowHostAudio) ||
732
           (vm->def->graphics[0]->type == VIR_DOMAIN_GRAPHICS_TYPE_SDL))))) {
733 734
        rv = virCgroupAllowDevice(priv->cgroup, 'c', DEVICE_SND_MAJOR, -1,
                                  VIR_CGROUP_DEVICE_RW);
735
        virDomainAuditCgroupMajor(vm, priv->cgroup, "allow", DEVICE_SND_MAJOR,
736 737
                                  "sound", "rw", rv == 0);
        if (rv < 0)
738 739 740
            goto cleanup;
    }

741
    for (i = 0; deviceACL[i] != NULL; i++) {
742
        if (!virFileExists(deviceACL[i])) {
N
Nehal J Wani 已提交
743
            VIR_DEBUG("Ignoring non-existent device %s", deviceACL[i]);
744 745 746
            continue;
        }

747
        rv = virCgroupAllowDevicePath(priv->cgroup, deviceACL[i],
748
                                      VIR_CGROUP_DEVICE_RW, false);
749 750 751
        virDomainAuditCgroupPath(vm, priv->cgroup, "allow", deviceACL[i], "rw", rv == 0);
        if (rv < 0 &&
            !virLastErrorIsSystemErrno(ENOENT))
752 753 754 755 756
            goto cleanup;
    }

    if (virDomainChrDefForeach(vm->def,
                               true,
757
                               qemuSetupChardevCgroupCB,
758 759 760
                               vm) < 0)
        goto cleanup;

761
    if (vm->def->tpm && qemuSetupTPMCgroup(vm) < 0)
762 763 764
        goto cleanup;

    for (i = 0; i < vm->def->nhostdevs; i++) {
765
        if (qemuSetupHostdevCgroup(vm, vm->def->hostdevs[i]) < 0)
766 767 768
            goto cleanup;
    }

769 770 771 772 773
    for (i = 0; i < vm->def->ninputs; i++) {
        if (qemuSetupInputCgroup(vm, vm->def->inputs[i]) < 0)
            goto cleanup;
    }

774
    for (i = 0; i < vm->def->nrngs; i++) {
775 776
        if (qemuSetupRNGCgroup(vm, vm->def->rngs[i]) < 0)
            goto cleanup;
777 778
    }

779
    ret = 0;
780
 cleanup:
781 782 783 784 785
    virObjectUnref(cfg);
    return ret;
}


786
int
787
qemuSetupCpusetMems(virDomainObjPtr vm)
788
{
789
    virCgroupPtr cgroup_temp = NULL;
790
    qemuDomainObjPrivatePtr priv = vm->privateData;
791
    virDomainNumatuneMemMode mode;
792
    char *mem_mask = NULL;
793 794 795 796 797
    int ret = -1;

    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPUSET))
        return 0;

798 799
    if (virDomainNumatuneGetMode(vm->def->numa, -1, &mode) < 0 ||
        mode != VIR_DOMAIN_NUMATUNE_MEM_STRICT)
800 801
        return 0;

802
    if (virDomainNumatuneMaybeFormatNodeset(vm->def->numa,
803
                                            priv->autoNodeset,
804
                                            &mem_mask, -1) < 0)
805
        goto cleanup;
806

807
    if (mem_mask)
J
John Ferlan 已提交
808 809
        if (virCgroupNewThread(priv->cgroup, VIR_CGROUP_THREAD_EMULATOR, 0,
                               false, &cgroup_temp) < 0 ||
810
            virCgroupSetCpusetMems(cgroup_temp, mem_mask) < 0)
811
            goto cleanup;
812

813 814 815
    ret = 0;
 cleanup:
    VIR_FREE(mem_mask);
816
    virCgroupFree(&cgroup_temp);
817 818 819 820 821
    return ret;
}


static int
822
qemuSetupCpusetCgroup(virDomainObjPtr vm)
823 824 825 826 827 828
{
    qemuDomainObjPrivatePtr priv = vm->privateData;

    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPUSET))
        return 0;

829 830 831
    if (virCgroupSetCpusetMemoryMigrate(priv->cgroup, true) < 0)
        return -1;

832
    return 0;
833 834 835
}


836
static int
837 838
qemuSetupCpuCgroup(virQEMUDriverPtr driver,
                   virDomainObjPtr vm)
839 840
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
841 842 843 844
    virObjectEventPtr event = NULL;
    virTypedParameterPtr eventParams = NULL;
    int eventNparams = 0;
    int eventMaxparams = 0;
845 846

    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPU)) {
847
       if (vm->def->cputune.sharesSpecified) {
848 849 850 851 852 853 854 855
           virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                          _("CPU tuning is not available on this host"));
           return -1;
       } else {
           return 0;
       }
    }

856 857 858 859 860 861 862
    if (vm->def->cputune.sharesSpecified) {
        unsigned long long val;
        if (virCgroupSetCpuShares(priv->cgroup, vm->def->cputune.shares) < 0)
            return -1;

        if (virCgroupGetCpuShares(priv->cgroup, &val) < 0)
            return -1;
863 864 865 866
        if (vm->def->cputune.shares != val) {
            vm->def->cputune.shares = val;
            if (virTypedParamsAddULLong(&eventParams, &eventNparams,
                                        &eventMaxparams,
867
                                        VIR_DOMAIN_TUNABLE_CPU_CPU_SHARES,
868 869 870 871 872 873
                                        val) < 0)
                return -1;

            event = virDomainEventTunableNewFromObj(vm, eventParams, eventNparams);
        }

874
        qemuDomainEventQueue(driver, event);
875
    }
876 877 878 879 880

    return 0;
}


881
static int
882
qemuInitCgroup(virQEMUDriverPtr driver,
883 884 885
               virDomainObjPtr vm,
               size_t nnicindexes,
               int *nicindexes)
886
{
887
    int ret = -1;
888 889 890
    qemuDomainObjPrivatePtr priv = vm->privateData;
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);

891
    if (!virQEMUDriverIsPrivileged(driver))
892 893
        goto done;

894 895 896
    if (!virCgroupAvailable())
        goto done;

897 898
    virCgroupFree(&priv->cgroup);

899
    if (!vm->def->resource) {
900 901
        virDomainResourceDefPtr res;

902
        if (VIR_ALLOC(res) < 0)
903
            goto cleanup;
904

905
        if (VIR_STRDUP(res->partition, "/machine") < 0) {
906 907 908 909 910
            VIR_FREE(res);
            goto cleanup;
        }

        vm->def->resource = res;
911 912
    }

913 914 915 916 917 918
    if (vm->def->resource->partition[0] != '/') {
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED,
                       _("Resource partition '%s' must start with '/'"),
                       vm->def->resource->partition);
        goto cleanup;
    }
919

920 921 922 923 924 925 926 927 928 929 930 931
    /*
     * We need to do this because of systemd-machined, because
     * CreateMachine requires the name to be a valid hostname.
     */
    priv->machineName = virSystemdMakeMachineName("qemu",
                                                  vm->def->id,
                                                  vm->def->name,
                                                  virQEMUDriverIsPrivileged(driver));
    if (!priv->machineName)
        goto cleanup;

    if (virCgroupNewMachine(priv->machineName,
932 933 934 935 936
                            "qemu",
                            vm->def->uuid,
                            NULL,
                            vm->pid,
                            false,
937
                            nnicindexes, nicindexes,
938 939 940
                            vm->def->resource->partition,
                            cfg->cgroupControllers,
                            &priv->cgroup) < 0) {
941 942
        if (virCgroupNewIgnoreError())
            goto done;
943

944 945
        goto cleanup;
    }
946

947
 done:
948
    ret = 0;
949
 cleanup:
950 951 952
    virObjectUnref(cfg);
    return ret;
}
953

954 955 956
static void
qemuRestoreCgroupState(virDomainObjPtr vm)
{
957
    char *mem_mask = NULL;
958
    char *nodeset = NULL;
959 960
    int empty = -1;
    qemuDomainObjPrivatePtr priv = vm->privateData;
961
    size_t i = 0;
962
    virBitmapPtr all_nodes;
963
    virCgroupPtr cgroup_temp = NULL;
964

965 966 967 968
    if (!virNumaIsAvailable() ||
        !virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPUSET))
        return;

969
    if (!(all_nodes = virNumaGetHostMemoryNodeset()))
970 971 972 973 974 975
        goto error;

    if (!(mem_mask = virBitmapFormat(all_nodes)))
        goto error;

    if ((empty = virCgroupHasEmptyTasks(priv->cgroup,
976
                                        VIR_CGROUP_CONTROLLER_CPUSET)) <= 0)
977 978 979 980 981
        goto error;

    if (virCgroupSetCpusetMems(priv->cgroup, mem_mask) < 0)
        goto error;

982
    for (i = 0; i < virDomainDefGetVcpusMax(vm->def); i++) {
983
        virDomainVcpuDefPtr vcpu = virDomainDefGetVcpu(vm->def, i);
984 985 986 987

        if (!vcpu->online)
            continue;

J
John Ferlan 已提交
988 989
        if (virCgroupNewThread(priv->cgroup, VIR_CGROUP_THREAD_VCPU, i,
                               false, &cgroup_temp) < 0 ||
990 991 992 993 994
            virCgroupSetCpusetMemoryMigrate(cgroup_temp, true) < 0 ||
            virCgroupGetCpusetMems(cgroup_temp, &nodeset) < 0 ||
            virCgroupSetCpusetMems(cgroup_temp, nodeset) < 0)
            goto cleanup;

995
        VIR_FREE(nodeset);
996 997 998
        virCgroupFree(&cgroup_temp);
    }

999 1000 1001
    for (i = 0; i < vm->def->niothreadids; i++) {
        if (virCgroupNewThread(priv->cgroup, VIR_CGROUP_THREAD_IOTHREAD,
                               vm->def->iothreadids[i]->iothread_id,
J
John Ferlan 已提交
1002
                               false, &cgroup_temp) < 0 ||
1003 1004 1005 1006 1007
            virCgroupSetCpusetMemoryMigrate(cgroup_temp, true) < 0 ||
            virCgroupGetCpusetMems(cgroup_temp, &nodeset) < 0 ||
            virCgroupSetCpusetMems(cgroup_temp, nodeset) < 0)
            goto cleanup;

1008
        VIR_FREE(nodeset);
1009 1010 1011
        virCgroupFree(&cgroup_temp);
    }

J
John Ferlan 已提交
1012 1013
    if (virCgroupNewThread(priv->cgroup, VIR_CGROUP_THREAD_EMULATOR, 0,
                           false, &cgroup_temp) < 0 ||
1014 1015 1016 1017 1018
        virCgroupSetCpusetMemoryMigrate(cgroup_temp, true) < 0 ||
        virCgroupGetCpusetMems(cgroup_temp, &nodeset) < 0 ||
        virCgroupSetCpusetMems(cgroup_temp, nodeset) < 0)
        goto cleanup;

1019 1020
 cleanup:
    VIR_FREE(mem_mask);
1021
    VIR_FREE(nodeset);
1022
    virBitmapFree(all_nodes);
1023
    virCgroupFree(&cgroup_temp);
1024 1025 1026 1027 1028 1029 1030
    return;

 error:
    virResetLastError();
    VIR_DEBUG("Couldn't restore cgroups to meaningful state");
    goto cleanup;
}
1031 1032 1033 1034 1035 1036 1037 1038 1039

int
qemuConnectCgroup(virQEMUDriverPtr driver,
                  virDomainObjPtr vm)
{
    virQEMUDriverConfigPtr cfg = virQEMUDriverGetConfig(driver);
    qemuDomainObjPrivatePtr priv = vm->privateData;
    int ret = -1;

1040
    if (!virQEMUDriverIsPrivileged(driver))
1041 1042 1043 1044 1045 1046 1047
        goto done;

    if (!virCgroupAvailable())
        goto done;

    virCgroupFree(&priv->cgroup);

1048 1049
    if (virCgroupNewDetectMachine(vm->def->name,
                                  "qemu",
1050 1051
                                  vm->def->id,
                                  virQEMUDriverIsPrivileged(driver),
1052
                                  vm->pid,
1053
                                  cfg->cgroupControllers,
1054
                                  &priv->cgroup) < 0)
1055
        goto cleanup;
1056

1057 1058 1059 1060
    priv->machineName = virSystemdGetMachineNameByPID(vm->pid);
    if (!priv->machineName)
        virResetLastError();

1061 1062
    qemuRestoreCgroupState(vm);

1063
 done:
1064
    ret = 0;
1065
 cleanup:
1066
    virObjectUnref(cfg);
1067
    return ret;
1068 1069
}

1070 1071
int
qemuSetupCgroup(virQEMUDriverPtr driver,
1072 1073 1074
                virDomainObjPtr vm,
                size_t nnicindexes,
                int *nicindexes)
1075
{
1076
    qemuDomainObjPrivatePtr priv = vm->privateData;
1077
    int ret = -1;
1078

1079 1080 1081 1082 1083 1084
    if (!vm->pid) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("Cannot setup cgroups until process is started"));
        return -1;
    }

1085
    if (qemuInitCgroup(driver, vm, nnicindexes, nicindexes) < 0)
1086
        return -1;
1087

1088
    if (!priv->cgroup)
1089
        return 0;
1090

1091 1092
    if (qemuSetupDevicesCgroup(driver, vm) < 0)
        goto cleanup;
1093

1094 1095
    if (qemuSetupBlkioCgroup(vm) < 0)
        goto cleanup;
1096

1097 1098
    if (qemuSetupMemoryCgroup(vm) < 0)
        goto cleanup;
1099

1100
    if (qemuSetupCpuCgroup(driver, vm) < 0)
1101
        goto cleanup;
1102

1103
    if (qemuSetupCpusetCgroup(vm) < 0)
1104
        goto cleanup;
1105

1106
    ret = 0;
1107
 cleanup:
1108
    return ret;
1109 1110
}

1111 1112 1113 1114
int
qemuSetupCgroupVcpuBW(virCgroupPtr cgroup,
                      unsigned long long period,
                      long long quota)
1115 1116 1117 1118 1119 1120 1121 1122
{
    unsigned long long old_period;

    if (period == 0 && quota == 0)
        return 0;

    if (period) {
        /* get old period, and we can rollback if set quota failed */
1123
        if (virCgroupGetCpuCfsPeriod(cgroup, &old_period) < 0)
1124 1125
            return -1;

1126
        if (virCgroupSetCpuCfsPeriod(cgroup, period) < 0)
1127 1128 1129
            return -1;
    }

1130 1131 1132
    if (quota &&
        virCgroupSetCpuCfsQuota(cgroup, quota) < 0)
        goto error;
1133 1134 1135

    return 0;

1136
 error:
1137
    if (period) {
1138 1139 1140 1141 1142 1143
        virErrorPtr saved = virSaveLastError();
        ignore_value(virCgroupSetCpuCfsPeriod(cgroup, old_period));
        if (saved) {
            virSetError(saved);
            virFreeError(saved);
        }
1144 1145 1146 1147 1148
    }

    return -1;
}

1149

1150
int
1151 1152
qemuSetupCgroupCpusetCpus(virCgroupPtr cgroup,
                          virBitmapPtr cpumask)
1153
{
1154
    int ret = -1;
1155 1156
    char *new_cpus = NULL;

1157
    if (!(new_cpus = virBitmapFormat(cpumask)))
1158 1159
        goto cleanup;

1160
    if (virCgroupSetCpusetCpus(cgroup, new_cpus) < 0)
1161 1162
        goto cleanup;

1163
    ret = 0;
1164
 cleanup:
1165
    VIR_FREE(new_cpus);
1166
    return ret;
1167 1168
}

1169

1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218
int
qemuSetupGlobalCpuCgroup(virDomainObjPtr vm)
{
    qemuDomainObjPrivatePtr priv = vm->privateData;
    unsigned long long period = vm->def->cputune.global_period;
    long long quota = vm->def->cputune.global_quota;
    char *mem_mask = NULL;
    virDomainNumatuneMemMode mem_mode;

    if ((period || quota) &&
        !virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPU)) {
        virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                       _("cgroup cpu is required for scheduler tuning"));
        return -1;
    }

    /*
     * If CPU cgroup controller is not initialized here, then we need
     * neither period nor quota settings.  And if CPUSET controller is
     * not initialized either, then there's nothing to do anyway.
     */
    if (!virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPU) &&
        !virCgroupHasController(priv->cgroup, VIR_CGROUP_CONTROLLER_CPUSET))
        return 0;


    if (virDomainNumatuneGetMode(vm->def->numa, -1, &mem_mode) == 0 &&
        mem_mode == VIR_DOMAIN_NUMATUNE_MEM_STRICT &&
        virDomainNumatuneMaybeFormatNodeset(vm->def->numa,
                                            priv->autoNodeset,
                                            &mem_mask, -1) < 0)
        goto cleanup;

    if (period || quota) {
        if (qemuSetupCgroupVcpuBW(priv->cgroup, period, quota) < 0)
            goto cleanup;
    }

    VIR_FREE(mem_mask);

    return 0;

 cleanup:
    VIR_FREE(mem_mask);

    return -1;
}


1219
int
1220
qemuRemoveCgroup(virDomainObjPtr vm)
1221
{
1222
    qemuDomainObjPrivatePtr priv = vm->privateData;
1223

1224
    if (priv->cgroup == NULL)
1225 1226
        return 0; /* Not supported, so claim success */

1227
    if (virCgroupTerminateMachine(priv->machineName) < 0) {
1228 1229 1230 1231
        if (!virCgroupNewIgnoreError())
            VIR_DEBUG("Failed to terminate cgroup for %s", vm->def->name);
    }

M
Martin Kletzander 已提交
1232 1233
    VIR_FREE(priv->machineName);

1234
    return virCgroupRemove(priv->cgroup);
1235
}
1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257


static void
qemuCgroupEmulatorAllNodesDataFree(qemuCgroupEmulatorAllNodesDataPtr data)
{
    if (!data)
        return;

    virCgroupFree(&data->emulatorCgroup);
    VIR_FREE(data->emulatorMemMask);
    VIR_FREE(data);
}


/**
 * qemuCgroupEmulatorAllNodesAllow:
 * @cgroup: domain cgroup pointer
 * @retData: filled with structure used to roll back the operation
 *
 * Allows all NUMA nodes for the qemu emulator thread temporarily. This is
 * necessary when hotplugging cpus since it requires memory allocated in the
 * DMA region. Afterwards the operation can be reverted by
1258
 * qemuCgroupEmulatorAllNodesRestore.
1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274
 *
 * Returns 0 on success -1 on error
 */
int
qemuCgroupEmulatorAllNodesAllow(virCgroupPtr cgroup,
                                qemuCgroupEmulatorAllNodesDataPtr *retData)
{
    qemuCgroupEmulatorAllNodesDataPtr data = NULL;
    char *all_nodes_str = NULL;
    virBitmapPtr all_nodes = NULL;
    int ret = -1;

    if (!virNumaIsAvailable() ||
        !virCgroupHasController(cgroup, VIR_CGROUP_CONTROLLER_CPUSET))
        return 0;

1275
    if (!(all_nodes = virNumaGetHostMemoryNodeset()))
1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
        goto cleanup;

    if (!(all_nodes_str = virBitmapFormat(all_nodes)))
        goto cleanup;

    if (VIR_ALLOC(data) < 0)
        goto cleanup;

    if (virCgroupNewThread(cgroup, VIR_CGROUP_THREAD_EMULATOR, 0,
                           false, &data->emulatorCgroup) < 0)
        goto cleanup;

    if (virCgroupGetCpusetMems(data->emulatorCgroup, &data->emulatorMemMask) < 0 ||
        virCgroupSetCpusetMems(data->emulatorCgroup, all_nodes_str) < 0)
        goto cleanup;

    VIR_STEAL_PTR(*retData, data);
    ret = 0;

 cleanup:
    VIR_FREE(all_nodes_str);
    virBitmapFree(all_nodes);
    qemuCgroupEmulatorAllNodesDataFree(data);

    return ret;
}


/**
1305
 * qemuCgroupEmulatorAllNodesRestore:
1306 1307 1308 1309 1310 1311
 * @data: data structure created by qemuCgroupEmulatorAllNodesAllow
 *
 * Rolls back the setting done by qemuCgroupEmulatorAllNodesAllow and frees the
 * associated data.
 */
void
1312
qemuCgroupEmulatorAllNodesRestore(qemuCgroupEmulatorAllNodesDataPtr data)
1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325
{
    virErrorPtr err;

    if (!data)
        return;

    err = virSaveLastError();
    virCgroupSetCpusetMems(data->emulatorCgroup, data->emulatorMemMask);
    virSetError(err);
    virFreeError(err);

    qemuCgroupEmulatorAllNodesDataFree(data);
}