lxc_controller.c 49.6 KB
Newer Older
1
/*
2 3
 * Copyright (C) 2010-2011 Red Hat, Inc.
 * Copyright IBM Corp. 2008
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27
 *
 * lxc_controller.c: linux container process controller
 *
 * Authors:
 *  David L. Leskovec <dlesko at linux.vnet.ibm.com>
 *
 * 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
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
 */

#include <config.h>

#include <sys/epoll.h>
28 29
#include <sys/wait.h>
#include <sys/socket.h>
30 31
#include <sys/types.h>
#include <sys/un.h>
32 33
#include <sys/utsname.h>
#include <sys/personality.h>
34
#include <unistd.h>
35
#include <paths.h>
36
#include <errno.h>
37 38
#include <fcntl.h>
#include <signal.h>
39
#include <getopt.h>
40
#include <sys/mount.h>
E
Eric Blake 已提交
41
#include <locale.h>
42 43
#include <grp.h>
#include <sys/stat.h>
44
#include <time.h>
45

D
Daniel P. Berrange 已提交
46
#if HAVE_CAPNG
47
# include <cap-ng.h>
D
Daniel P. Berrange 已提交
48 49
#endif

50 51 52 53 54
#if HAVE_NUMACTL
# define NUMA_VERSION1_COMPATIBILITY 1
# include <numa.h>
#endif

55
#include "virterror_internal.h"
56
#include "logging.h"
57 58 59
#include "util.h"

#include "lxc_conf.h"
60
#include "lxc_container.h"
61 62
#include "virnetdev.h"
#include "virnetdevveth.h"
63 64
#include "memory.h"
#include "util.h"
E
Eric Blake 已提交
65
#include "virfile.h"
66
#include "virpidfile.h"
67
#include "command.h"
68 69
#include "processinfo.h"
#include "nodeinfo.h"
70
#include "virrandom.h"
71
#include "rpc/virnetserver.h"
72

73 74
#define VIR_FROM_THIS VIR_FROM_LXC

D
Dan Smith 已提交
75 76 77 78 79 80
struct cgroup_device_policy {
    char type;
    int major;
    int minor;
};

81

82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
typedef struct _virLXCControllerConsole virLXCControllerConsole;
typedef virLXCControllerConsole *virLXCControllerConsolePtr;
struct _virLXCControllerConsole {
    int hostWatch;
    int hostFd;  /* PTY FD in the host OS */
    bool hostClosed;
    int hostEpoll;
    bool hostBlocking;

    int contWatch;
    int contFd;  /* PTY FD in the container */
    bool contClosed;
    int contEpoll;
    bool contBlocking;

    int epollWatch;
    int epollFd; /* epoll FD for dealing with EOF */

    size_t fromHostLen;
    char fromHostBuf[1024];
    size_t fromContLen;
    char fromContBuf[1024];
};

106 107 108 109 110
typedef struct _virLXCController virLXCController;
typedef virLXCController *virLXCControllerPtr;
struct _virLXCController {
    char *name;
    virDomainDefPtr def;
111

112 113
    int handshakeFd;

114 115
    pid_t initpid;

116 117
    size_t nveths;
    char **veths;
118 119 120

    size_t nconsoles;
    virLXCControllerConsolePtr consoles;
121
    char *devptmx;
122 123 124

    size_t nloopDevs;
    int *loopDevFds;
125 126

    virSecurityManagerPtr securityManager;
127

128 129
    /* Server socket */
    virNetServerPtr server;
130 131

    virCgroupPtr cgroup;
132 133 134 135 136 137 138 139 140 141 142 143 144 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
};

static void virLXCControllerFree(virLXCControllerPtr ctrl);

static virLXCControllerPtr virLXCControllerNew(const char *name)
{
    virLXCControllerPtr ctrl = NULL;
    virCapsPtr caps = NULL;
    char *configFile = NULL;

    if (VIR_ALLOC(ctrl) < 0)
        goto no_memory;

    if (!(ctrl->name = strdup(name)))
        goto no_memory;

    if ((caps = lxcCapsInit(NULL)) == NULL)
        goto error;

    if ((configFile = virDomainConfigFile(LXC_STATE_DIR,
                                          ctrl->name)) == NULL)
        goto error;

    if ((ctrl->def = virDomainDefParseFile(caps,
                                           configFile,
                                           1 << VIR_DOMAIN_VIRT_LXC,
                                           0)) == NULL)
        goto error;

cleanup:
    VIR_FREE(configFile);
    virCapabilitiesFree(caps);
    return ctrl;

no_memory:
    virReportOOMError();
error:
    virLXCControllerFree(ctrl);
    ctrl = NULL;
    goto cleanup;
}

174

175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
static int virLXCControllerCloseLoopDevices(virLXCControllerPtr ctrl,
                                            bool force)
{
    size_t i;

    for (i = 0 ; i < ctrl->nloopDevs ; i++) {
        if (force) {
            VIR_FORCE_CLOSE(ctrl->loopDevFds[i]);
        } else {
            if (VIR_CLOSE(ctrl->loopDevFds[i]) < 0) {
                virReportSystemError(errno, "%s",
                                     _("Unable to close loop device"));
                return -1;
            }
        }
    }

    return 0;
}


196 197 198 199 200
static void virLXCControllerStopInit(virLXCControllerPtr ctrl)
{
    if (ctrl->initpid == 0)
        return;

201
    virLXCControllerCloseLoopDevices(ctrl, true);
202 203
    virPidAbort(ctrl->initpid);
    ctrl->initpid = 0;
204 205

    virCgroupFree(&ctrl->cgroup);
206 207 208
}


209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
static void virLXCControllerConsoleClose(virLXCControllerConsolePtr console)
{
    if (console->hostWatch != -1)
        virEventRemoveHandle(console->hostWatch);
    VIR_FORCE_CLOSE(console->hostFd);

    if (console->contWatch != -1)
        virEventRemoveHandle(console->contWatch);
    VIR_FORCE_CLOSE(console->contFd);

    if (console->epollWatch != -1)
        virEventRemoveHandle(console->epollWatch);
    VIR_FORCE_CLOSE(console->epollFd);
}


225 226
static void virLXCControllerFree(virLXCControllerPtr ctrl)
{
227 228
    size_t i;

229 230 231
    if (!ctrl)
        return;

232 233
    virLXCControllerStopInit(ctrl);

234 235
    virSecurityManagerFree(ctrl->securityManager);

236 237 238 239
    for (i = 0 ; i < ctrl->nveths ; i++)
        VIR_FREE(ctrl->veths[i]);
    VIR_FREE(ctrl->veths);

240 241 242 243
    for (i = 0 ; i < ctrl->nconsoles ; i++)
        virLXCControllerConsoleClose(&(ctrl->consoles[i]));
    VIR_FREE(ctrl->consoles);

244 245
    VIR_FORCE_CLOSE(ctrl->handshakeFd);

246 247
    VIR_FREE(ctrl->devptmx);

248 249 250
    virDomainDefFree(ctrl->def);
    VIR_FREE(ctrl->name);

251
    virNetServerFree(ctrl->server);
252

253 254 255
    VIR_FREE(ctrl);
}

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
static int virLXCControllerAddConsole(virLXCControllerPtr ctrl,
                                      int hostFd)
{
    if (VIR_EXPAND_N(ctrl->consoles, ctrl->nconsoles, 1) < 0) {
        virReportOOMError();
        return -1;
    }
    ctrl->consoles[ctrl->nconsoles-1].hostFd = hostFd;
    ctrl->consoles[ctrl->nconsoles-1].hostWatch = -1;

    ctrl->consoles[ctrl->nconsoles-1].contFd = -1;
    ctrl->consoles[ctrl->nconsoles-1].contWatch = -1;

    ctrl->consoles[ctrl->nconsoles-1].epollFd = -1;
    ctrl->consoles[ctrl->nconsoles-1].epollWatch = -1;
    return 0;
}


static int virLXCControllerConsoleSetNonblocking(virLXCControllerConsolePtr console)
{
    if (virSetBlocking(console->hostFd, false) < 0 ||
        virSetBlocking(console->contFd, false) < 0) {
        virReportSystemError(errno, "%s",
                             _("Unable to set console file descriptor non-blocking"));
        return -1;
    }

    return 0;
}


289 290 291 292 293 294 295 296 297 298 299 300
static int virLXCControllerDaemonHandshake(virLXCControllerPtr ctrl)
{
    if (lxcContainerSendContinue(ctrl->handshakeFd) < 0) {
        virReportSystemError(errno, "%s",
                             _("error sending continue signal to daemon"));
        return -1;
    }
    VIR_FORCE_CLOSE(ctrl->handshakeFd);
    return 0;
}


301 302 303 304 305 306 307 308 309 310 311 312 313
static int virLXCControllerValidateNICs(virLXCControllerPtr ctrl)
{
    if (ctrl->def->nnets != ctrl->nveths) {
        lxcError(VIR_ERR_INTERNAL_ERROR,
                 _("expecting %d veths, but got %zu"),
                 ctrl->def->nnets, ctrl->nveths);
        return -1;
    }

    return 0;
}


314 315 316 317 318 319 320 321 322 323 324 325 326
static int virLXCControllerValidateConsoles(virLXCControllerPtr ctrl)
{
    if (ctrl->def->nconsoles != ctrl->nconsoles) {
        lxcError(VIR_ERR_INTERNAL_ERROR,
                 _("expecting %d consoles, but got %zu tty file handlers"),
                 ctrl->def->nconsoles, ctrl->nconsoles);
        return -1;
    }

    return 0;
}


327
static int virLXCControllerSetupLoopDevice(virDomainFSDefPtr fs)
328
{
329
    int lofd;
330 331
    char *loname = NULL;

332
    if ((lofd = virFileLoopDeviceAssociate(fs->src, &loname)) < 0)
333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
        return -1;

    /*
     * We now change it into a block device type, so that
     * the rest of container setup 'just works'
     */
    fs->type = VIR_DOMAIN_FS_TYPE_BLOCK;
    VIR_FREE(fs->src);
    fs->src = loname;
    loname = NULL;

    return lofd;
}


348
static int virLXCControllerSetupLoopDevices(virLXCControllerPtr ctrl)
349 350 351 352
{
    size_t i;
    int ret = -1;

353
    for (i = 0 ; i < ctrl->def->nfss ; i++) {
354 355
        int fd;

356
        if (ctrl->def->fss[i]->type != VIR_DOMAIN_FS_TYPE_FILE)
357 358
            continue;

359
        fd = virLXCControllerSetupLoopDevice(ctrl->def->fss[i]);
360 361 362 363
        if (fd < 0)
            goto cleanup;

        VIR_DEBUG("Saving loop fd %d", fd);
364
        if (VIR_EXPAND_N(ctrl->loopDevFds, ctrl->nloopDevs, 1) < 0) {
365 366 367 368
            VIR_FORCE_CLOSE(fd);
            virReportOOMError();
            goto cleanup;
        }
369
        ctrl->loopDevFds[ctrl->nloopDevs - 1] = fd;
370 371 372 373 374 375 376 377 378
    }

    VIR_DEBUG("Setup all loop devices");
    ret = 0;

cleanup:
    return ret;
}

379
#if HAVE_NUMACTL
380
static int virLXCControllerSetupNUMAPolicy(virLXCControllerPtr ctrl)
381 382 383 384 385 386 387 388 389
{
    nodemask_t mask;
    int mode = -1;
    int node = -1;
    int ret = -1;
    int i = 0;
    int maxnode = 0;
    bool warned = false;

390
    if (!ctrl->def->numatune.memory.nodemask)
391 392 393 394 395 396 397 398 399 400 401 402 403 404 405
        return 0;

    VIR_DEBUG("Setting NUMA memory policy");

    if (numa_available() < 0) {
        lxcError(VIR_ERR_CONFIG_UNSUPPORTED,
                 "%s", _("Host kernel is not aware of NUMA."));
        return -1;
    }

    maxnode = numa_max_node() + 1;

    /* Convert nodemask to NUMA bitmask. */
    nodemask_zero(&mask);
    for (i = 0; i < VIR_DOMAIN_CPUMASK_LEN; i++) {
406
        if (ctrl->def->numatune.memory.nodemask[i]) {
407 408 409 410 411 412 413 414 415 416 417 418 419 420
            if (i > NUMA_NUM_NODES) {
                lxcError(VIR_ERR_CONFIG_UNSUPPORTED,
                         _("Host cannot support NUMA node %d"), i);
                return -1;
            }
            if (i > maxnode && !warned) {
                VIR_WARN("nodeset is out of range, there is only %d NUMA "
                         "nodes on host", maxnode);
                warned = true;
            }
            nodemask_set(&mask, i);
        }
    }

421
    mode = ctrl->def->numatune.memory.mode;
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459

    if (mode == VIR_DOMAIN_NUMATUNE_MEM_STRICT) {
        numa_set_bind_policy(1);
        numa_set_membind(&mask);
        numa_set_bind_policy(0);
    } else if (mode == VIR_DOMAIN_NUMATUNE_MEM_PREFERRED) {
        int nnodes = 0;
        for (i = 0; i < NUMA_NUM_NODES; i++) {
            if (nodemask_isset(&mask, i)) {
                node = i;
                nnodes++;
            }
        }

        if (nnodes != 1) {
            lxcError(VIR_ERR_CONFIG_UNSUPPORTED,
                     "%s", _("NUMA memory tuning in 'preferred' mode "
                             "only supports single node"));
            goto cleanup;
        }

        numa_set_bind_policy(0);
        numa_set_preferred(node);
    } else if (mode == VIR_DOMAIN_NUMATUNE_MEM_INTERLEAVE) {
        numa_set_interleave_mask(&mask);
    } else {
        lxcError(VIR_ERR_CONFIG_UNSUPPORTED,
                 _("Unable to set NUMA policy %s"),
                 virDomainNumatuneMemModeTypeToString(mode));
        goto cleanup;
    }

    ret = 0;

cleanup:
    return ret;
}
#else
460
static int virLXCControllerSetupNUMAPolicy(virLXCControllerPtr ctrl)
461
{
462
    if (ctrl->def->numatune.memory.nodemask) {
463 464 465 466 467 468 469 470 471
        lxcError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                 _("NUMA policy is not available on this platform"));
        return -1;
    }

    return 0;
}
#endif

472 473 474 475

/*
 * To be run while still single threaded
 */
476
static int virLXCControllerSetupCpuAffinity(virLXCControllerPtr ctrl)
477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499
{
    int i, hostcpus, maxcpu = CPU_SETSIZE;
    virNodeInfo nodeinfo;
    unsigned char *cpumap;
    int cpumaplen;

    VIR_DEBUG("Setting CPU affinity");

    if (nodeGetInfo(NULL, &nodeinfo) < 0)
        return -1;

    /* setaffinity fails if you set bits for CPUs which
     * aren't present, so we have to limit ourselves */
    hostcpus = VIR_NODEINFO_MAXCPUS(nodeinfo);
    if (maxcpu > hostcpus)
        maxcpu = hostcpus;

    cpumaplen = VIR_CPU_MAPLEN(maxcpu);
    if (VIR_ALLOC_N(cpumap, cpumaplen) < 0) {
        virReportOOMError();
        return -1;
    }

500
    if (ctrl->def->cpumask) {
501 502
        /* XXX why don't we keep 'cpumask' in the libvirt cpumap
         * format to start with ?!?! */
503 504
        for (i = 0 ; i < maxcpu && i < ctrl->def->cpumasklen ; i++)
            if (ctrl->def->cpumask[i])
505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527 528 529
                VIR_USE_CPU(cpumap, i);
    } else {
        /* You may think this is redundant, but we can't assume libvirtd
         * itself is running on all pCPUs, so we need to explicitly set
         * the spawned LXC instance to all pCPUs if no map is given in
         * its config file */
        for (i = 0 ; i < maxcpu ; i++)
            VIR_USE_CPU(cpumap, i);
    }

    /* We are pressuming we are running between fork/exec of LXC
     * so use '0' to indicate our own process ID. No threads are
     * running at this point
     */
    if (virProcessInfoSetAffinity(0, /* Self */
                                  cpumap, cpumaplen, maxcpu) < 0) {
        VIR_FREE(cpumap);
        return -1;
    }
    VIR_FREE(cpumap);

    return 0;
}


530
static int virLXCControllerSetupCgroupsCpuTune(virLXCControllerPtr ctrl)
531 532
{
    int ret = -1;
533 534
    if (ctrl->def->cputune.shares != 0) {
        int rc = virCgroupSetCpuShares(ctrl->cgroup, ctrl->def->cputune.shares);
535 536 537
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set io cpu shares for domain %s"),
538
                                 ctrl->def->name);
539 540 541
            goto cleanup;
        }
    }
542 543
    if (ctrl->def->cputune.quota != 0) {
        int rc = virCgroupSetCpuCfsQuota(ctrl->cgroup, ctrl->def->cputune.quota);
544 545 546
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set io cpu quota for domain %s"),
547
                                 ctrl->def->name);
548 549 550
            goto cleanup;
        }
    }
551 552
    if (ctrl->def->cputune.period != 0) {
        int rc = virCgroupSetCpuCfsPeriod(ctrl->cgroup, ctrl->def->cputune.period);
553 554 555
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set io cpu period for domain %s"),
556
                                 ctrl->def->name);
557 558 559 560 561 562 563 564 565
            goto cleanup;
        }
    }
    ret = 0;
cleanup:
    return ret;
}


566
static int virLXCControllerSetupCgroupsBlkioTune(virLXCControllerPtr ctrl)
567 568 569
{
    int ret = -1;

570 571
    if (ctrl->def->blkio.weight) {
        int rc = virCgroupSetBlkioWeight(ctrl->cgroup, ctrl->def->blkio.weight);
572 573 574
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set Blkio weight for domain %s"),
575
                                 ctrl->def->name);
576 577 578 579 580 581 582 583 584 585
            goto cleanup;
        }
    }

    ret = 0;
cleanup:
    return ret;
}


586
static int virLXCControllerSetupCgroupsMemTune(virLXCControllerPtr ctrl)
D
Dan Smith 已提交
587
{
588 589
    int ret = -1;
    int rc;
590

591
    rc = virCgroupSetMemory(ctrl->cgroup, ctrl->def->mem.max_balloon);
592
    if (rc != 0) {
593
        virReportSystemError(-rc,
594
                             _("Unable to set memory limit for domain %s"),
595
                             ctrl->def->name);
596
        goto cleanup;
597
    }
D
Dan Smith 已提交
598

599 600
    if (ctrl->def->mem.hard_limit) {
        rc = virCgroupSetMemoryHardLimit(ctrl->cgroup, ctrl->def->mem.hard_limit);
601 602 603
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set memory hard limit for domain %s"),
604
                                 ctrl->def->name);
605 606 607 608
            goto cleanup;
        }
    }

609 610
    if (ctrl->def->mem.soft_limit) {
        rc = virCgroupSetMemorySoftLimit(ctrl->cgroup, ctrl->def->mem.soft_limit);
611 612 613
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set memory soft limit for domain %s"),
614
                                 ctrl->def->name);
615 616 617 618
            goto cleanup;
        }
    }

619 620
    if (ctrl->def->mem.swap_hard_limit) {
        rc = virCgroupSetMemSwapHardLimit(ctrl->cgroup, ctrl->def->mem.swap_hard_limit);
621 622 623
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set swap hard limit for domain %s"),
624
                                 ctrl->def->name);
625 626 627 628
            goto cleanup;
        }
    }

629 630 631 632 633 634
    ret = 0;
cleanup:
    return ret;
}


635
static int virLXCControllerSetupCgroupsDeviceACL(virLXCControllerPtr ctrl)
636 637 638 639 640 641 642 643 644 645 646 647 648 649
{
    int ret = -1;
    int rc;
    size_t i;
    static const struct cgroup_device_policy devices[] = {
        {'c', LXC_DEV_MAJ_MEMORY, LXC_DEV_MIN_NULL},
        {'c', LXC_DEV_MAJ_MEMORY, LXC_DEV_MIN_ZERO},
        {'c', LXC_DEV_MAJ_MEMORY, LXC_DEV_MIN_FULL},
        {'c', LXC_DEV_MAJ_MEMORY, LXC_DEV_MIN_RANDOM},
        {'c', LXC_DEV_MAJ_MEMORY, LXC_DEV_MIN_URANDOM},
        {'c', LXC_DEV_MAJ_TTY, LXC_DEV_MIN_TTY},
        {'c', LXC_DEV_MAJ_TTY, LXC_DEV_MIN_PTMX},
        {0,   0, 0}};

650
    rc = virCgroupDenyAllDevices(ctrl->cgroup);
651
    if (rc != 0) {
652
        virReportSystemError(-rc,
653
                             _("Unable to deny devices for domain %s"),
654
                             ctrl->def->name);
655 656
        goto cleanup;
    }
D
Dan Smith 已提交
657 658

    for (i = 0; devices[i].type != 0; i++) {
659
        const struct cgroup_device_policy *dev = &devices[i];
660
        rc = virCgroupAllowDevice(ctrl->cgroup,
D
Dan Smith 已提交
661 662
                                  dev->type,
                                  dev->major,
663 664
                                  dev->minor,
                                  VIR_CGROUP_DEVICE_RWM);
665
        if (rc != 0) {
666
            virReportSystemError(-rc,
667
                                 _("Unable to allow device %c:%d:%d for domain %s"),
668
                                 dev->type, dev->major, dev->minor, ctrl->def->name);
669 670
            goto cleanup;
        }
D
Dan Smith 已提交
671 672
    }

673 674
    for (i = 0 ; i < ctrl->def->nfss ; i++) {
        if (ctrl->def->fss[i]->type != VIR_DOMAIN_FS_TYPE_BLOCK)
675 676
            continue;

677 678 679
        rc = virCgroupAllowDevicePath(ctrl->cgroup,
                                      ctrl->def->fss[i]->src,
                                      ctrl->def->fss[i]->readonly ?
680 681 682 683 684
                                      VIR_CGROUP_DEVICE_READ :
                                      VIR_CGROUP_DEVICE_RW);
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to allow device %s for domain %s"),
685
                                 ctrl->def->fss[i]->src, ctrl->def->name);
686 687 688 689
            goto cleanup;
        }
    }

690
    rc = virCgroupAllowDeviceMajor(ctrl->cgroup, 'c', LXC_DEV_MAJ_PTY,
691
                                   VIR_CGROUP_DEVICE_RWM);
692
    if (rc != 0) {
693
        virReportSystemError(-rc,
694
                             _("Unable to allow PTY devices for domain %s"),
695
                             ctrl->def->name);
696 697
        goto cleanup;
    }
698

699 700 701 702 703 704 705
    ret = 0;
cleanup:
    return ret;
}


/**
706 707
 * virLXCControllerSetupResourceLimits
 * @ctrl: the controller state
708 709 710 711 712 713
 *
 * Creates a cgroup for the container, moves the task inside,
 * and sets resource limits
 *
 * Returns 0 on success or -1 in case of error
 */
714
static int virLXCControllerSetupResourceLimits(virLXCControllerPtr ctrl)
715 716 717 718
{
    virCgroupPtr driver;
    int rc = -1;

719
    if (virLXCControllerSetupCpuAffinity(ctrl) < 0)
720 721
        return -1;

722
    if (virLXCControllerSetupNUMAPolicy(ctrl) < 0)
723 724 725 726 727 728 729 730 731 732 733 734 735
        return -1;

    rc = virCgroupForDriver("lxc", &driver, 1, 0);
    if (rc != 0) {
        /* Skip all if no driver cgroup is configured */
        if (rc == -ENXIO || rc == -ENOENT)
            return 0;

        virReportSystemError(-rc, "%s",
                             _("Unable to get cgroup for driver"));
        return rc;
    }

736
    rc = virCgroupForDomain(driver, ctrl->def->name, &ctrl->cgroup, 1);
737 738 739
    if (rc != 0) {
        virReportSystemError(-rc,
                             _("Unable to create cgroup for domain %s"),
740
                             ctrl->def->name);
741 742 743
        goto cleanup;
    }

744
    if (virLXCControllerSetupCgroupsCpuTune(ctrl) < 0)
745 746
        goto cleanup;

747
    if (virLXCControllerSetupCgroupsBlkioTune(ctrl) < 0)
748 749
        goto cleanup;

750
    if (virLXCControllerSetupCgroupsMemTune(ctrl) < 0)
751 752
        goto cleanup;

753
    if (virLXCControllerSetupCgroupsDeviceACL(ctrl) < 0)
754 755
        goto cleanup;

756
    rc = virCgroupAddTask(ctrl->cgroup, getpid());
D
Dan Smith 已提交
757
    if (rc != 0) {
758
        virReportSystemError(-rc,
759
                             _("Unable to add task %d to cgroup for domain %s"),
760
                             getpid(), ctrl->def->name);
D
Dan Smith 已提交
761 762
    }

763 764
cleanup:
    virCgroupFree(&driver);
D
Dan Smith 已提交
765 766 767 768

    return rc;
}

769

770 771 772 773 774 775 776
static int virLXCControllerClientHook(virNetServerPtr server ATTRIBUTE_UNUSED,
                                      virNetServerClientPtr client,
                                      void *opaque)
{
    virLXCControllerPtr ctrl = opaque;
    virNetServerClientSetPrivateData(client, ctrl, NULL);
    return 0;
777 778
}

779 780

static int virLXCControllerSetupServer(virLXCControllerPtr ctrl)
781
{
782 783
    virNetServerServicePtr svc = NULL;
    char *sockpath;
784

785 786 787 788
    if (virAsprintf(&sockpath, "%s/%s.sock",
                    LXC_STATE_DIR, ctrl->name) < 0) {
        virReportOOMError();
        return -1;
789 790
    }

791 792 793 794 795
    if (!(ctrl->server = virNetServerNew(0, 0, 0, 1,
                                         -1, 0, false,
                                         NULL,
                                         virLXCControllerClientHook,
                                         ctrl)))
C
Chris Lalancette 已提交
796
        goto error;
797

798 799 800 801 802 803 804
    if (!(svc = virNetServerServiceNewUNIX(sockpath,
                                           0700,
                                           0,
                                           0,
                                           false,
                                           5,
                                           NULL)))
805
        goto error;
806 807

    if (virNetServerAddService(ctrl->server, svc, NULL) < 0)
808
        goto error;
809 810
    virNetServerServiceFree(svc);
    svc = NULL;
811

812 813 814
    virNetServerUpdateServices(ctrl->server, true);
    VIR_FREE(sockpath);
    return 0;
815 816

error:
817 818 819 820
    VIR_FREE(sockpath);
    virNetServerFree(ctrl->server);
    ctrl->server = NULL;
    virNetServerServiceFree(svc);
821 822
    return -1;
}
823

D
Daniel P. Berrange 已提交
824 825 826 827 828 829 830 831 832

static int lxcControllerClearCapabilities(void)
{
#if HAVE_CAPNG
    int ret;

    capng_clear(CAPNG_SELECT_BOTH);

    if ((ret = capng_apply(CAPNG_SELECT_BOTH)) < 0) {
833
        lxcError(VIR_ERR_INTERNAL_ERROR,
D
Daniel P. Berrange 已提交
834 835 836 837
                 _("failed to apply capabilities: %d"), ret);
        return -1;
    }
#else
838
    VIR_WARN("libcap-ng support not compiled in, unable to clear capabilities");
D
Daniel P. Berrange 已提交
839 840 841 842
#endif
    return 0;
}

843 844 845 846
static bool quit = false;
static virMutex lock;


847 848
static void virLXCControllerSignalChildIO(virNetServerPtr server ATTRIBUTE_UNUSED,
                                          siginfo_t *info ATTRIBUTE_UNUSED,
849
                                          void *opaque)
850
{
851
    virLXCControllerPtr ctrl = opaque;
852
    int ret;
853 854

    ret = waitpid(-1, NULL, WNOHANG);
855 856 857 858 859
    if (ret == ctrl->initpid) {
        virMutexLock(&lock);
        quit = true;
        virMutexUnlock(&lock);
    }
860 861 862
}


863
static void virLXCControllerConsoleUpdateWatch(virLXCControllerConsolePtr console)
864 865 866 867
{
    int hostEvents = 0;
    int contEvents = 0;

868
    if (!console->hostClosed || (!console->hostBlocking && console->fromContLen)) {
869 870 871 872 873
        if (console->fromHostLen < sizeof(console->fromHostBuf))
            hostEvents |= VIR_EVENT_HANDLE_READABLE;
        if (console->fromContLen)
            hostEvents |= VIR_EVENT_HANDLE_WRITABLE;
    }
874
    if (!console->contClosed || (!console->contBlocking && console->fromHostLen)) {
875 876 877 878 879 880
        if (console->fromContLen < sizeof(console->fromContBuf))
            contEvents |= VIR_EVENT_HANDLE_READABLE;
        if (console->fromHostLen)
            contEvents |= VIR_EVENT_HANDLE_WRITABLE;
    }

881 882 883
    VIR_DEBUG("Container watch %d=%d host watch %d=%d",
              console->contWatch, contEvents,
              console->hostWatch, hostEvents);
884 885
    virEventUpdateHandle(console->contWatch, contEvents);
    virEventUpdateHandle(console->hostWatch, hostEvents);
886

887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922
    if (console->hostClosed) {
        int events = EPOLLIN | EPOLLET;
        if (console->hostBlocking)
            events |= EPOLLOUT;

        if (events != console->hostEpoll) {
            struct epoll_event event;
            int action = EPOLL_CTL_ADD;
            if (console->hostEpoll)
                action = EPOLL_CTL_MOD;

            VIR_DEBUG("newHostEvents=%x oldHostEvents=%x", events, console->hostEpoll);

            event.events = events;
            event.data.fd = console->hostFd;
            if (epoll_ctl(console->epollFd, action, console->hostFd, &event) < 0) {
                VIR_DEBUG(":fail");
                virReportSystemError(errno, "%s",
                                     _("Unable to add epoll fd"));
                quit = true;
                goto cleanup;
            }
            console->hostEpoll = events;
            VIR_DEBUG("newHostEvents=%x oldHostEvents=%x", events, console->hostEpoll);
        }
    } else if (console->hostEpoll) {
        VIR_DEBUG("Stop epoll oldContEvents=%x", console->hostEpoll);
        if (epoll_ctl(console->epollFd, EPOLL_CTL_DEL, console->hostFd, NULL) < 0) {
            virReportSystemError(errno, "%s",
                                 _("Unable to remove epoll fd"));
                VIR_DEBUG(":fail");
            quit = true;
            goto cleanup;
        }
        console->hostEpoll = 0;
    }
923

924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962
    if (console->contClosed) {
        int events = EPOLLIN | EPOLLET;
        if (console->contBlocking)
            events |= EPOLLOUT;

        if (events != console->contEpoll) {
            struct epoll_event event;
            int action = EPOLL_CTL_ADD;
            if (console->contEpoll)
                action = EPOLL_CTL_MOD;

            VIR_DEBUG("newContEvents=%x oldContEvents=%x", events, console->contEpoll);

            event.events = events;
            event.data.fd = console->contFd;
            if (epoll_ctl(console->epollFd, action, console->contFd, &event) < 0) {
                virReportSystemError(errno, "%s",
                                     _("Unable to add epoll fd"));
                VIR_DEBUG(":fail");
                quit = true;
                goto cleanup;
            }
            console->contEpoll = events;
            VIR_DEBUG("newHostEvents=%x oldHostEvents=%x", events, console->contEpoll);
        }
    } else if (console->contEpoll) {
        VIR_DEBUG("Stop epoll oldContEvents=%x", console->contEpoll);
        if (epoll_ctl(console->epollFd, EPOLL_CTL_DEL, console->contFd, NULL) < 0) {
            virReportSystemError(errno, "%s",
                                 _("Unable to remove epoll fd"));
                VIR_DEBUG(":fail");
            quit = true;
            goto cleanup;
        }
        console->contEpoll = 0;
    }
cleanup:
    return;
}
963 964


965
static void virLXCControllerConsoleEPoll(int watch, int fd, int events, void *opaque)
966
{
967
    virLXCControllerConsolePtr console = opaque;
968

969 970 971 972 973 974 975 976 977 978
    virMutexLock(&lock);
    VIR_DEBUG("IO event watch=%d fd=%d events=%d fromHost=%zu fromcont=%zu",
              watch, fd, events,
              console->fromHostLen,
              console->fromContLen);

    while (1) {
        struct epoll_event event;
        int ret;
        ret = epoll_wait(console->epollFd, &event, 1, 0);
979
        if (ret < 0) {
S
Stefan Berger 已提交
980
            if (errno == EINTR)
981 982 983 984 985 986 987
                continue;
            virReportSystemError(errno, "%s",
                                 _("Unable to wait on epoll"));
            quit = true;
            goto cleanup;
        }

988 989 990 991 992 993 994
        if (ret == 0)
            break;

        VIR_DEBUG("fd=%d hostFd=%d contFd=%d hostEpoll=%x contEpoll=%x",
                  event.data.fd, console->hostFd, console->contFd,
                  console->hostEpoll, console->contEpoll);

995 996 997
        /* If we get HUP+dead PID, we just re-enable the main loop
         * which will see the PID has died and exit */
        if ((event.events & EPOLLIN)) {
998 999
            if (event.data.fd == console->hostFd) {
                console->hostClosed = false;
1000
            } else {
1001
                console->contClosed = false;
1002
            }
1003
            virLXCControllerConsoleUpdateWatch(console);
1004 1005 1006 1007 1008
            break;
        }
    }

cleanup:
1009
    virMutexUnlock(&lock);
1010 1011
}

1012
static void virLXCControllerConsoleIO(int watch, int fd, int events, void *opaque)
1013
{
1014
    virLXCControllerConsolePtr console = opaque;
1015 1016

    virMutexLock(&lock);
1017 1018 1019 1020
    VIR_DEBUG("IO event watch=%d fd=%d events=%d fromHost=%zu fromcont=%zu",
              watch, fd, events,
              console->fromHostLen,
              console->fromContLen);
1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076
    if (events & VIR_EVENT_HANDLE_READABLE) {
        char *buf;
        size_t *len;
        size_t avail;
        ssize_t done;
        if (watch == console->hostWatch) {
            buf = console->fromHostBuf;
            len = &console->fromHostLen;
            avail = sizeof(console->fromHostBuf) - *len;
        } else {
            buf = console->fromContBuf;
            len = &console->fromContLen;
            avail = sizeof(console->fromContBuf) - *len;
        }
    reread:
        done = read(fd, buf + *len, avail);
        if (done == -1 && errno == EINTR)
            goto reread;
        if (done == -1 && errno != EAGAIN) {
            virReportSystemError(errno, "%s",
                                 _("Unable to read container pty"));
            goto error;
        }
        if (done > 0) {
            *len += done;
        } else {
            VIR_DEBUG("Read fd %d done %d errno %d", fd, (int)done, errno);
        }
    }

    if (events & VIR_EVENT_HANDLE_WRITABLE) {
        char *buf;
        size_t *len;
        ssize_t done;
        if (watch == console->hostWatch) {
            buf = console->fromContBuf;
            len = &console->fromContLen;
        } else {
            buf = console->fromHostBuf;
            len = &console->fromHostLen;
        }

    rewrite:
        done = write(fd, buf, *len);
        if (done == -1 && errno == EINTR)
            goto rewrite;
        if (done == -1 && errno != EAGAIN) {
            virReportSystemError(errno, "%s",
                                 _("Unable to write to container pty"));
            goto error;
        }
        if (done > 0) {
            memmove(buf, buf + done, (*len - done));
            *len -= done;
        } else {
            VIR_DEBUG("Write fd %d done %d errno %d", fd, (int)done, errno);
1077 1078 1079 1080
            if (watch == console->hostWatch)
                console->hostBlocking = true;
            else
                console->contBlocking = true;
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
        }
    }

    if (events & VIR_EVENT_HANDLE_HANGUP) {
        if (watch == console->hostWatch) {
            console->hostClosed = true;
        } else {
            console->contClosed = true;
        }
        VIR_DEBUG("Got EOF on %d %d", watch, fd);
    }

1093
    virLXCControllerConsoleUpdateWatch(console);
1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105
    virMutexUnlock(&lock);
    return;

error:
    virEventRemoveHandle(console->contWatch);
    virEventRemoveHandle(console->hostWatch);
    console->contWatch = console->hostWatch = -1;
    quit = true;
    virMutexUnlock(&lock);
}


1106
/**
1107
 * lxcControllerMain
1108 1109
 * @serverFd: server socket fd to accept client requests
 * @clientFd: initial client which is the libvirtd daemon
1110
 *
1111
 * Processes I/O on consoles and the monitor
1112 1113 1114
 *
 * Returns 0 on success or -1 in case of error
 */
1115
static int virLXCControllerMain(virLXCControllerPtr ctrl)
1116
{
1117
    virErrorPtr err;
1118
    int rc = -1;
1119
    size_t i;
1120 1121 1122 1123

    if (virMutexInit(&lock) < 0)
        goto cleanup2;

1124 1125 1126 1127
    if (virNetServerAddSignalHandler(ctrl->server,
                                     SIGCHLD,
                                     virLXCControllerSignalChildIO,
                                     ctrl) < 0)
1128 1129
        goto cleanup;

1130 1131
    virResetLastError();

1132 1133
    for (i = 0 ; i < ctrl->nconsoles ; i++) {
        if ((ctrl->consoles[i].epollFd = epoll_create1(EPOLL_CLOEXEC)) < 0) {
1134 1135 1136 1137 1138
            virReportSystemError(errno, "%s",
                                 _("Unable to create epoll fd"));
            goto cleanup;
        }

1139 1140 1141 1142 1143
        if ((ctrl->consoles[i].epollWatch = virEventAddHandle(ctrl->consoles[i].epollFd,
                                                              VIR_EVENT_HANDLE_READABLE,
                                                              virLXCControllerConsoleEPoll,
                                                              &(ctrl->consoles[i]),
                                                              NULL)) < 0) {
1144 1145 1146 1147 1148
            lxcError(VIR_ERR_INTERNAL_ERROR, "%s",
                     _("Unable to watch epoll FD"));
            goto cleanup;
        }

1149 1150 1151 1152 1153
        if ((ctrl->consoles[i].hostWatch = virEventAddHandle(ctrl->consoles[i].hostFd,
                                                             VIR_EVENT_HANDLE_READABLE,
                                                             virLXCControllerConsoleIO,
                                                             &(ctrl->consoles[i]),
                                                             NULL)) < 0) {
1154 1155 1156 1157 1158
            lxcError(VIR_ERR_INTERNAL_ERROR, "%s",
                     _("Unable to watch host console PTY"));
            goto cleanup;
        }

1159 1160 1161 1162 1163
        if ((ctrl->consoles[i].contWatch = virEventAddHandle(ctrl->consoles[i].contFd,
                                                             VIR_EVENT_HANDLE_READABLE,
                                                             virLXCControllerConsoleIO,
                                                             &(ctrl->consoles[i]),
                                                             NULL)) < 0) {
1164 1165 1166 1167
            lxcError(VIR_ERR_INTERNAL_ERROR, "%s",
                     _("Unable to watch host console PTY"));
            goto cleanup;
        }
1168
    }
1169

1170 1171 1172 1173
    virMutexLock(&lock);
    while (!quit) {
        virMutexUnlock(&lock);
        if (virEventRunDefaultImpl() < 0)
1174
            goto cleanup;
1175
        virMutexLock(&lock);
1176
    }
1177
    virMutexUnlock(&lock);
1178

1179 1180 1181
    err = virGetLastError();
    if (!err || err->code == VIR_ERR_OK)
        rc = 0;
1182 1183

cleanup:
1184 1185
    virMutexDestroy(&lock);
cleanup2:
1186

1187 1188
    for (i = 0 ; i < ctrl->nconsoles ; i++)
        virLXCControllerConsoleClose(&(ctrl->consoles[i]));
1189

1190 1191 1192
    return rc;
}

1193 1194 1195


/**
1196
 * virLXCControllerMoveInterfaces
1197 1198 1199 1200 1201 1202 1203 1204
 * @nveths: number of interfaces
 * @veths: interface names
 * @container: pid of container
 *
 * Moves network interfaces into a container's namespace
 *
 * Returns 0 on success or -1 in case of error
 */
1205
static int virLXCControllerMoveInterfaces(virLXCControllerPtr ctrl)
1206
{
1207 1208 1209
    size_t i;

    for (i = 0 ; i < ctrl->nveths ; i++) {
1210
        if (virNetDevSetNamespace(ctrl->veths[i], ctrl->initpid) < 0)
1211
            return -1;
1212
    }
1213 1214 1215 1216 1217 1218

    return 0;
}


/**
1219 1220
 * virLXCControllerDeleteInterfaces:
 * @ctrl: the LXC controller
1221 1222 1223 1224 1225
 *
 * Cleans up the container interfaces by deleting the veth device pairs.
 *
 * Returns 0 on success or -1 in case of error
 */
1226
static int virLXCControllerDeleteInterfaces(virLXCControllerPtr ctrl)
1227
{
1228 1229
    size_t i;
    int ret = 0;
1230

1231 1232 1233 1234 1235 1236
    for (i = 0 ; i < ctrl->nveths ; i++) {
        if (virNetDevVethDelete(ctrl->veths[i]) < 0)
            ret = -1;
    }

    return ret;
1237 1238
}

1239

1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258
static int lxcSetPersonality(virDomainDefPtr def)
{
    struct utsname utsname;
    const char *altArch;

    uname(&utsname);

    altArch = lxcContainerGetAlt32bitArch(utsname.machine);
    if (altArch &&
        STREQ(def->os.arch, altArch)) {
        if (personality(PER_LINUX32) < 0) {
            virReportSystemError(errno, _("Unable to request personality for %s on %s"),
                                 altArch, utsname.machine);
            return -1;
        }
    }
    return 0;
}

1259
#ifndef MS_REC
1260
# define MS_REC          16384
1261 1262 1263
#endif

#ifndef MS_SLAVE
1264
# define MS_SLAVE              (1<<19)
1265
#endif
1266

1267 1268 1269 1270 1271 1272 1273 1274 1275 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 1305 1306 1307 1308 1309 1310
/* Create a private tty using the private devpts at PTMX, returning
 * the master in *TTYMASTER and the name of the slave, _from the
 * perspective of the guest after remounting file systems_, in
 * *TTYNAME.  Heavily borrowed from glibc, but doesn't require that
 * devpts == "/dev/pts" */
static int
lxcCreateTty(char *ptmx, int *ttymaster, char **ttyName)
{
    int ret = -1;
    int ptyno;
    int unlock = 0;

    if ((*ttymaster = open(ptmx, O_RDWR|O_NOCTTY|O_NONBLOCK)) < 0)
        goto cleanup;

    if (ioctl(*ttymaster, TIOCSPTLCK, &unlock) < 0)
        goto cleanup;

    if (ioctl(*ttymaster, TIOCGPTN, &ptyno) < 0)
        goto cleanup;

    /* If mount() succeeded at honoring newinstance, then the kernel
     * was new enough to also honor the mode=0620,gid=5 options, which
     * guarantee that the new pty already has correct permissions; so
     * while glibc has to fstat(), fchmod(), and fchown() for older
     * kernels, we can skip those steps.  ptyno shouldn't currently be
     * anything other than 0, but let's play it safe.  */
    if (virAsprintf(ttyName, "/dev/pts/%d", ptyno) < 0) {
        virReportOOMError();
        errno = ENOMEM;
        goto cleanup;
    }

    ret = 0;

cleanup:
    if (ret != 0) {
        VIR_FORCE_CLOSE(*ttymaster);
        VIR_FREE(*ttyName);
    }

    return ret;
}

1311

1312
static int
1313
virLXCControllerSetupDevPTS(virLXCControllerPtr ctrl)
1314
{
1315
    virDomainFSDefPtr root = virDomainGetRootFilesystem(ctrl->def);
1316
    char *mount_options = NULL;
1317 1318 1319
    char *opts;
    char *devpts = NULL;
    int ret = -1;
1320

1321 1322 1323 1324 1325 1326 1327 1328
    if (!root) {
        if (ctrl->nconsoles != 1) {
            lxcError(VIR_ERR_CONFIG_UNSUPPORTED,
                     _("Expected exactly one console, but got %zu"),
                     ctrl->nconsoles);
            return -1;
        }
        return 0;
1329 1330
    }

1331
    VIR_DEBUG("Setting up private /dev/pts");
1332

1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352
    /*
     * If doing a chroot style setup, we need to prepare
     * a private /dev/pts for the child now, which they
     * will later move into position.
     *
     * This is complex because 'virsh console' needs to
     * use /dev/pts from the host OS, and the guest OS
     * needs to use /dev/pts from the guest.
     *
     * This means that we (libvirt_lxc) need to see and
     * use both /dev/pts instances. We're running in the
     * host OS context though and don't want to expose
     * the guest OS /dev/pts there.
     *
     * Thus we call unshare(CLONE_NS) so that we can see
     * the guest's new /dev/pts, without it becoming
     * visible to the host OS. We also put the root FS
     * into slave mode, just in case it was currently
     * marked as shared
     */
1353 1354
    mount_options = virSecurityManagerGetMountOptions(ctrl->securityManager,
                                                      ctrl->def);
1355

1356 1357 1358 1359 1360 1361
    if (!virFileExists(root->src)) {
        virReportSystemError(errno,
                             _("root source %s does not exist"),
                             root->src);
        goto cleanup;
    }
1362

1363 1364 1365 1366 1367
    if (unshare(CLONE_NEWNS) < 0) {
        virReportSystemError(errno, "%s",
                             _("Cannot unshare mount namespace"));
        goto cleanup;
    }
1368

1369 1370 1371 1372 1373
    if (mount("", "/", NULL, MS_SLAVE|MS_REC, NULL) < 0) {
        virReportSystemError(errno, "%s",
                             _("Failed to switch root mount into slave mode"));
        goto cleanup;
    }
1374

1375 1376 1377 1378 1379
    if (virAsprintf(&devpts, "%s/dev/pts", root->src) < 0 ||
        virAsprintf(&ctrl->devptmx, "%s/dev/pts/ptmx", root->src) < 0) {
        virReportOOMError();
        goto cleanup;
    }
1380

1381 1382 1383 1384 1385 1386
    if (virFileMakePath(devpts) < 0) {
        virReportSystemError(errno,
                             _("Failed to make path %s"),
                             devpts);
        goto cleanup;
    }
1387

1388 1389 1390 1391 1392 1393 1394
    /* XXX should we support gid=X for X!=5 for distros which use
     * a different gid for tty?  */
    if (virAsprintf(&opts, "newinstance,ptmxmode=0666,mode=0620,gid=5%s",
                    (mount_options ? mount_options : "")) < 0) {
        virReportOOMError();
        goto cleanup;
    }
1395

1396 1397 1398 1399 1400 1401 1402 1403
    VIR_DEBUG("Mount devpts on %s type=tmpfs flags=%x, opts=%s",
              devpts, MS_NOSUID, opts);
    if (mount("devpts", devpts, "devpts", MS_NOSUID, opts) < 0) {
        virReportSystemError(errno,
                             _("Failed to mount devpts on %s"),
                             devpts);
        goto cleanup;
    }
1404

1405 1406 1407
    if (access(ctrl->devptmx, R_OK) < 0) {
        VIR_WARN("Kernel does not support private devpts, using shared devpts");
        VIR_FREE(ctrl->devptmx);
1408 1409
    }

1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424
    ret = 0;

cleanup:
    VIR_FREE(opts);
    VIR_FREE(devpts);
    return ret;
}


static int
virLXCControllerSetupConsoles(virLXCControllerPtr ctrl,
                              char **containerTTYPaths)
{
    size_t i;

1425
    for (i = 0 ; i < ctrl->nconsoles ; i++) {
1426 1427 1428
        if (ctrl->devptmx) {
            VIR_DEBUG("Opening tty on private %s", ctrl->devptmx);
            if (lxcCreateTty(ctrl->devptmx,
1429 1430
                             &ctrl->consoles[i].contFd,
                             &containerTTYPaths[i]) < 0) {
1431 1432
                virReportSystemError(errno, "%s",
                                     _("Failed to allocate tty"));
1433
                return -1;
1434 1435 1436
            }
        } else {
            VIR_DEBUG("Opening tty on shared /dev/ptmx");
1437 1438
            if (virFileOpenTty(&ctrl->consoles[i].contFd,
                               &containerTTYPaths[i],
1439 1440 1441
                               0) < 0) {
                virReportSystemError(errno, "%s",
                                     _("Failed to allocate tty"));
1442
                return -1;
1443 1444 1445
            }
        }
    }
1446 1447 1448 1449 1450
    return 0;
}


static int
1451
virLXCControllerRun(virLXCControllerPtr ctrl)
1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478
{
    int rc = -1;
    int control[2] = { -1, -1};
    int containerhandshake[2] = { -1, -1 };
    char **containerTTYPaths = NULL;
    size_t i;

    if (VIR_ALLOC_N(containerTTYPaths, ctrl->nconsoles) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    if (socketpair(PF_UNIX, SOCK_STREAM, 0, control) < 0) {
        virReportSystemError(errno, "%s",
                             _("sockpair failed"));
        goto cleanup;
    }

    if (socketpair(PF_UNIX, SOCK_STREAM, 0, containerhandshake) < 0) {
        virReportSystemError(errno, "%s",
                             _("socketpair failed"));
        goto cleanup;
    }

    if (virLXCControllerSetupLoopDevices(ctrl) < 0)
        goto cleanup;

1479
    if (virLXCControllerSetupResourceLimits(ctrl) < 0)
1480 1481 1482 1483 1484 1485 1486
        goto cleanup;

    if (virLXCControllerSetupDevPTS(ctrl) < 0)
        goto cleanup;

    if (virLXCControllerSetupConsoles(ctrl, containerTTYPaths) < 0)
        goto cleanup;
1487

1488
    if (lxcSetPersonality(ctrl->def) < 0)
1489
        goto cleanup;
1490

1491
    if ((ctrl->initpid = lxcContainerStart(ctrl->def,
1492
                                           ctrl->securityManager,
1493 1494 1495 1496
                                           ctrl->nveths,
                                           ctrl->veths,
                                           control[1],
                                           containerhandshake[1],
1497 1498
                                           containerTTYPaths,
                                           ctrl->nconsoles)) < 0)
1499
        goto cleanup;
1500
    VIR_FORCE_CLOSE(control[1]);
1501
    VIR_FORCE_CLOSE(containerhandshake[1]);
1502

1503
    if (virLXCControllerMoveInterfaces(ctrl) < 0)
1504 1505
        goto cleanup;

1506 1507 1508
    if (lxcContainerSendContinue(control[0]) < 0) {
        virReportSystemError(errno, "%s",
                             _("Unable to send container continue message"));
1509
        goto cleanup;
1510
    }
1511

1512 1513 1514 1515 1516 1517
    if (lxcContainerWaitForContinue(containerhandshake[0]) < 0) {
        virReportSystemError(errno, "%s",
                             _("error receiving signal from container"));
        goto cleanup;
    }

1518 1519 1520
    /* Now the container is fully setup... */

    /* ...we can close the loop devices... */
1521 1522
    if (virLXCControllerCloseLoopDevices(ctrl, false) < 0)
        goto cleanup;
1523 1524

    /* ...and reduce our privileges */
D
Daniel P. Berrange 已提交
1525 1526 1527
    if (lxcControllerClearCapabilities() < 0)
        goto cleanup;

1528
    if (virLXCControllerDaemonHandshake(ctrl) < 0)
1529 1530
        goto cleanup;

1531 1532
    for (i = 0 ; i < ctrl->nconsoles ; i++)
        if (virLXCControllerConsoleSetNonblocking(&(ctrl->consoles[i])) < 0)
1533
            goto cleanup;
1534

1535
    rc = virLXCControllerMain(ctrl);
1536 1537

cleanup:
1538 1539
    VIR_FORCE_CLOSE(control[0]);
    VIR_FORCE_CLOSE(control[1]);
1540 1541
    VIR_FORCE_CLOSE(containerhandshake[0]);
    VIR_FORCE_CLOSE(containerhandshake[1]);
1542

1543 1544 1545
    for (i = 0 ; i < ctrl->nconsoles ; i++)
        VIR_FREE(containerTTYPaths[i]);
    VIR_FREE(containerTTYPaths);
1546

1547
    virLXCControllerStopInit(ctrl);
1548

1549 1550 1551 1552
    return rc;
}


1553
int main(int argc, char *argv[])
1554 1555
{
    pid_t pid;
1556 1557
    int rc = 1;
    char *name = NULL;
1558
    size_t nveths = 0;
1559
    char **veths = NULL;
1560
    int handshakeFd = -1;
1561
    int bg = 0;
1562
    const struct option options[] = {
1563 1564 1565 1566
        { "background", 0, NULL, 'b' },
        { "name",   1, NULL, 'n' },
        { "veth",   1, NULL, 'v' },
        { "console", 1, NULL, 'c' },
1567
        { "handshakefd", 1, NULL, 's' },
1568
        { "security", 1, NULL, 'S' },
1569 1570 1571
        { "help", 0, NULL, 'h' },
        { 0, 0, 0, 0 },
    };
1572 1573
    int *ttyFDs = NULL;
    size_t nttyFDs = 0;
1574
    virLXCControllerPtr ctrl = NULL;
1575
    size_t i;
1576
    const char *securityDriver = "none";
1577

E
Eric Blake 已提交
1578 1579
    if (setlocale(LC_ALL, "") == NULL ||
        bindtextdomain(PACKAGE, LOCALEDIR) == NULL ||
1580 1581
        textdomain(PACKAGE) == NULL ||
        virRandomInitialize(time(NULL) ^ getpid())) {
E
Eric Blake 已提交
1582 1583 1584 1585
        fprintf(stderr, _("%s: initialization failed\n"), argv[0]);
        exit(EXIT_FAILURE);
    }

1586 1587 1588
    /* Initialize logging */
    virLogSetFromEnv();

1589 1590
    while (1) {
        int c;
1591

1592
        c = getopt_long(argc, argv, "dn:v:m:c:s:h:S:",
1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604
                       options, NULL);

        if (c == -1)
            break;

        switch (c) {
        case 'b':
            bg = 1;
            break;

        case 'n':
            if ((name = strdup(optarg)) == NULL) {
1605
                virReportOOMError();
1606
                goto cleanup;
1607
            }
1608 1609 1610 1611
            break;

        case 'v':
            if (VIR_REALLOC_N(veths, nveths+1) < 0) {
1612
                virReportOOMError();
1613
                goto cleanup;
1614
            }
1615
            if ((veths[nveths++] = strdup(optarg)) == NULL) {
1616
                virReportOOMError();
1617
                goto cleanup;
1618
            }
1619 1620 1621
            break;

        case 'c':
1622 1623 1624 1625 1626
            if (VIR_REALLOC_N(ttyFDs, nttyFDs + 1) < 0) {
                virReportOOMError();
                goto cleanup;
            }
            if (virStrToLong_i(optarg, NULL, 10, &ttyFDs[nttyFDs++]) < 0) {
1627 1628 1629 1630 1631
                fprintf(stderr, "malformed --console argument '%s'", optarg);
                goto cleanup;
            }
            break;

1632
        case 's':
1633
            if (virStrToLong_i(optarg, NULL, 10, &handshakeFd) < 0) {
1634 1635 1636 1637 1638 1639
                fprintf(stderr, "malformed --handshakefd argument '%s'",
                        optarg);
                goto cleanup;
            }
            break;

1640
        case 'S':
1641
            securityDriver = optarg;
1642 1643
            break;

1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654
        case 'h':
        case '?':
            fprintf(stderr, "\n");
            fprintf(stderr, "syntax: %s [OPTIONS]\n", argv[0]);
            fprintf(stderr, "\n");
            fprintf(stderr, "Options\n");
            fprintf(stderr, "\n");
            fprintf(stderr, "  -b, --background\n");
            fprintf(stderr, "  -n NAME, --name NAME\n");
            fprintf(stderr, "  -c FD, --console FD\n");
            fprintf(stderr, "  -v VETH, --veth VETH\n");
1655
            fprintf(stderr, "  -s FD, --handshakefd FD\n");
1656
            fprintf(stderr, "  -S NAME, --security NAME\n");
1657 1658 1659
            fprintf(stderr, "  -h, --help\n");
            fprintf(stderr, "\n");
            goto cleanup;
1660 1661 1662
        }
    }

1663 1664 1665 1666 1667
    if (name == NULL) {
        fprintf(stderr, "%s: missing --name argument for configuration\n", argv[0]);
        goto cleanup;
    }

1668
    if (handshakeFd < 0) {
1669 1670 1671 1672 1673
        fprintf(stderr, "%s: missing --handshake argument for container PTY\n",
                argv[0]);
        goto cleanup;
    }

1674
    if (getuid() != 0) {
1675 1676 1677
        fprintf(stderr, "%s: must be run as the 'root' user\n", argv[0]);
        goto cleanup;
    }
1678

1679 1680
    virEventRegisterDefaultImpl();

1681
    if (!(ctrl = virLXCControllerNew(name)))
1682
        goto cleanup;
1683

1684 1685
    ctrl->handshakeFd = handshakeFd;

1686 1687 1688 1689 1690
    if (!(ctrl->securityManager = virSecurityManagerNew(securityDriver,
                                                        LXC_DRIVER_NAME,
                                                        false, false, false)))
        goto cleanup;

1691
    VIR_DEBUG("Security model %s type %s label %s imagelabel %s",
1692 1693 1694 1695
              NULLSTR(ctrl->def->seclabel.model),
              virDomainSeclabelTypeToString(ctrl->def->seclabel.type),
              NULLSTR(ctrl->def->seclabel.label),
              NULLSTR(ctrl->def->seclabel.imagelabel));
1696

1697 1698 1699
    ctrl->veths = veths;
    ctrl->nveths = nveths;

1700 1701 1702 1703 1704 1705
    for (i = 0 ; i < nttyFDs ; i++) {
        if (virLXCControllerAddConsole(ctrl, ttyFDs[i]) < 0)
            goto cleanup;
        ttyFDs[i] = -1;
    }

1706
    if (virLXCControllerValidateNICs(ctrl) < 0)
1707
        goto cleanup;
1708

1709 1710 1711
    if (virLXCControllerValidateConsoles(ctrl) < 0)
        goto cleanup;

1712
    if (virLXCControllerSetupServer(ctrl) < 0)
1713
        goto cleanup;
1714

1715 1716 1717
    if (bg) {
        if ((pid = fork()) < 0)
            goto cleanup;
1718

1719
        if (pid > 0) {
1720
            if ((rc = virPidFileWrite(LXC_STATE_DIR, name, pid)) < 0) {
1721
                virReportSystemError(-rc,
1722 1723
                                     _("Unable to write pid file '%s/%s.pid'"),
                                     LXC_STATE_DIR, name);
1724 1725
                _exit(1);
            }
1726

1727 1728 1729 1730
            /* First child now exits, allowing original caller
             * (ie libvirtd's LXC driver to complete their
             * waitpid & continue */
            _exit(0);
1731 1732
        }

1733 1734
        /* Don't hold onto any cwd we inherit from libvirtd either */
        if (chdir("/") < 0) {
1735
            virReportSystemError(errno, "%s",
1736
                                 _("Unable to change to root dir"));
1737 1738 1739 1740
            goto cleanup;
        }

        if (setsid() < 0) {
1741
            virReportSystemError(errno, "%s",
1742
                                 _("Unable to become session leader"));
1743 1744 1745
            goto cleanup;
        }
    }
1746

1747
    rc = virLXCControllerRun(ctrl);
1748

1749
cleanup:
1750
    virPidFileDelete(LXC_STATE_DIR, name);
1751
    virLXCControllerDeleteInterfaces(ctrl);
1752 1753 1754 1755
    for (i = 0 ; i < nttyFDs ; i++)
        VIR_FORCE_CLOSE(ttyFDs[i]);
    VIR_FREE(ttyFDs);

1756
    virLXCControllerFree(ctrl);
1757

1758
    return rc ? EXIT_FAILURE : EXIT_SUCCESS;
1759
}