lxc_controller.c 26.5 KB
Newer Older
1
/*
2
 * Copyright (C) 2010 Red Hat, Inc.  Copyright IBM Corp. 2008
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
 *
 * 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>
27 28
#include <sys/wait.h>
#include <sys/socket.h>
29 30
#include <sys/types.h>
#include <sys/un.h>
31 32
#include <sys/utsname.h>
#include <sys/personality.h>
33
#include <unistd.h>
34 35 36
#include <paths.h>
#include <fcntl.h>
#include <signal.h>
37
#include <getopt.h>
38
#include <sys/mount.h>
E
Eric Blake 已提交
39
#include <locale.h>
40

D
Daniel P. Berrange 已提交
41
#if HAVE_CAPNG
42
# include <cap-ng.h>
D
Daniel P. Berrange 已提交
43 44
#endif

45
#include "virterror_internal.h"
46
#include "logging.h"
47 48 49
#include "util.h"

#include "lxc_conf.h"
50 51 52 53
#include "lxc_container.h"
#include "veth.h"
#include "memory.h"
#include "util.h"
54
#include "files.h"
55

56 57
#define VIR_FROM_THIS VIR_FROM_LXC

D
Dan Smith 已提交
58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
struct cgroup_device_policy {
    char type;
    int major;
    int minor;
};

/**
 * lxcSetContainerResources
 * @def: pointer to virtual machine structure
 *
 * Creates a cgroup for the container, moves the task inside,
 * and sets resource limits
 *
 * Returns 0 on success or -1 in case of error
 */
static int lxcSetContainerResources(virDomainDefPtr def)
{
75
    virCgroupPtr driver;
D
Dan Smith 已提交
76 77 78 79 80 81 82 83 84
    virCgroupPtr cgroup;
    int rc = -1;
    int i;
    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},
85
        {'c', LXC_DEV_MAJ_TTY, LXC_DEV_MIN_TTY},
86
        {'c', LXC_DEV_MAJ_TTY, LXC_DEV_MIN_PTMX},
D
Dan Smith 已提交
87 88
        {0,   0, 0}};

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

95
        virReportSystemError(-rc, "%s",
96
                             _("Unable to get cgroup for driver"));
97 98
        return rc;
    }
D
Dan Smith 已提交
99

100
    rc = virCgroupForDomain(driver, def->name, &cgroup, 1);
D
Dan Smith 已提交
101
    if (rc != 0) {
102
        virReportSystemError(-rc,
103 104
                             _("Unable to create cgroup for domain %s"),
                             def->name);
105
        goto cleanup;
D
Dan Smith 已提交
106 107
    }

108 109 110 111 112 113 114 115 116 117
    if (def->blkio.weight) {
        rc = virCgroupSetBlkioWeight(cgroup, def->blkio.weight);
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set Blkio weight for domain %s"),
                                 def->name);
            goto cleanup;
        }
    }

118
    rc = virCgroupSetMemory(cgroup, def->mem.max_balloon);
119
    if (rc != 0) {
120
        virReportSystemError(-rc,
121 122
                             _("Unable to set memory limit for domain %s"),
                             def->name);
123 124 125
        /* Don't fail if we can't set memory due to lack of kernel support */
        if (rc != -ENOENT)
            goto cleanup;
126
    }
D
Dan Smith 已提交
127

128 129 130 131 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
    if(def->mem.hard_limit) {
        rc = virCgroupSetMemoryHardLimit(cgroup, def->mem.hard_limit);
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set memory hard limit for domain %s"),
                                 def->name);
            goto cleanup;
        }
    }

    if(def->mem.soft_limit) {
        rc = virCgroupSetMemorySoftLimit(cgroup, def->mem.soft_limit);
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set memory soft limit for domain %s"),
                                 def->name);
            goto cleanup;
        }
    }

    if(def->mem.swap_hard_limit) {
        rc = virCgroupSetSwapHardLimit(cgroup, def->mem.swap_hard_limit);
        if (rc != 0) {
            virReportSystemError(-rc,
                                 _("Unable to set swap hard limit for domain %s"),
                                 def->name);
            goto cleanup;
        }
    }

D
Dan Smith 已提交
158
    rc = virCgroupDenyAllDevices(cgroup);
159
    if (rc != 0) {
160
        virReportSystemError(-rc,
161 162 163 164
                             _("Unable to deny devices for domain %s"),
                             def->name);
        goto cleanup;
    }
D
Dan Smith 已提交
165 166 167 168 169 170 171

    for (i = 0; devices[i].type != 0; i++) {
        struct cgroup_device_policy *dev = &devices[i];
        rc = virCgroupAllowDevice(cgroup,
                                  dev->type,
                                  dev->major,
                                  dev->minor);
172
        if (rc != 0) {
173
            virReportSystemError(-rc,
174 175 176 177
                                 _("Unable to allow device %c:%d:%d for domain %s"),
                                 dev->type, dev->major, dev->minor, def->name);
            goto cleanup;
        }
D
Dan Smith 已提交
178 179
    }

180
    rc = virCgroupAllowDeviceMajor(cgroup, 'c', LXC_DEV_MAJ_PTY);
181
    if (rc != 0) {
182
        virReportSystemError(-rc,
183 184 185 186
                             _("Unable to allow PYT devices for domain %s"),
                             def->name);
        goto cleanup;
    }
187

D
Dan Smith 已提交
188 189
    rc = virCgroupAddTask(cgroup, getpid());
    if (rc != 0) {
190
        virReportSystemError(-rc,
191 192
                             _("Unable to add task %d to cgroup for domain %s"),
                             getpid(), def->name);
D
Dan Smith 已提交
193 194
    }

195 196
cleanup:
    virCgroupFree(&driver);
D
Dan Smith 已提交
197 198 199 200 201
    virCgroupFree(&cgroup);

    return rc;
}

202 203 204
static char*lxcMonitorPath(virDomainDefPtr def)
{
    char *sockpath;
205 206 207

    if (virAsprintf(&sockpath, "%s/%s.sock",
                    LXC_STATE_DIR, def->name) < 0)
208
        virReportOOMError();
209 210 211 212 213 214 215 216 217
    return sockpath;
}

static int lxcMonitorServer(const char *sockpath)
{
    int fd;
    struct sockaddr_un addr;

    if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0) {
218
        virReportSystemError(errno,
219 220
                             _("failed to create server socket '%s'"),
                             sockpath);
221 222 223 224 225 226
        goto error;
    }

    unlink(sockpath);
    memset(&addr, 0, sizeof(addr));
    addr.sun_family = AF_UNIX;
C
Chris Lalancette 已提交
227
    if (virStrcpyStatic(addr.sun_path, sockpath) == NULL) {
228
        lxcError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
229 230 231
                 _("Socket path %s too long for destination"), sockpath);
        goto error;
    }
232 233

    if (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
234
        virReportSystemError(errno,
235 236
                             _("failed to bind server socket '%s'"),
                             sockpath);
237 238 239
        goto error;
    }
    if (listen(fd, 30 /* backlog */ ) < 0) {
240
        virReportSystemError(errno,
241 242
                             _("failed to listen server socket %s"),
                             sockpath);
243 244 245 246 247 248
        goto error;
    }

    return fd;

error:
249
    VIR_FORCE_CLOSE(fd);
250 251
    return -1;
}
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272

/**
 * lxcFdForward:
 * @readFd: file descriptor to read
 * @writeFd: file desriptor to write
 *
 * Reads 1 byte of data from readFd and writes to writeFd.
 *
 * Returns 0 on success, EAGAIN if returned on read, or -1 in case of error
 */
static int lxcFdForward(int readFd, int writeFd)
{
    int rc = -1;
    char buf[2];

    if (1 != (saferead(readFd, buf, 1))) {
        if (EAGAIN == errno) {
            rc = EAGAIN;
            goto cleanup;
        }

273
        virReportSystemError(errno,
274 275
                             _("read of fd %d failed"),
                             readFd);
276 277 278 279
        goto cleanup;
    }

    if (1 != (safewrite(writeFd, buf, 1))) {
280
        virReportSystemError(errno,
281 282
                             _("write to fd %d failed"),
                             writeFd);
283 284 285 286 287 288 289 290 291
        goto cleanup;
    }

    rc = 0;

cleanup:
    return rc;
}

D
Daniel P. Berrange 已提交
292 293 294 295 296 297 298 299 300

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

    capng_clear(CAPNG_SELECT_BOTH);

    if ((ret = capng_apply(CAPNG_SELECT_BOTH)) < 0) {
301
        lxcError(VIR_ERR_INTERNAL_ERROR,
D
Daniel P. Berrange 已提交
302 303 304 305
                 _("failed to apply capabilities: %d"), ret);
        return -1;
    }
#else
306
    VIR_WARN0("libcap-ng support not compiled in, unable to clear capabilities");
D
Daniel P. Berrange 已提交
307 308 309 310
#endif
    return 0;
}

311 312 313 314 315
typedef struct _lxcTtyForwardFd_t {
    int fd;
    int active;
} lxcTtyForwardFd_t;

316 317 318 319 320 321 322 323 324 325 326
/* Return true if it is ok to ignore an accept-after-epoll syscall
   that fails with the specified errno value.  Else false.  */
static bool
ignorable_epoll_accept_errno(int errnum)
{
  return (errnum == EINVAL
          || errnum == ECONNABORTED
          || errnum == EAGAIN
          || errnum == EWOULDBLOCK);
}

327
/**
328 329 330 331 332
 * lxcControllerMain
 * @monitor: server socket fd to accept client requests
 * @client: initial client which is the libvirtd daemon
 * @appPty: open fd for application facing Pty
 * @contPty: open fd for container facing Pty
333 334 335 336 337 338 339 340
 *
 * Forwards traffic between fds.  Data read from appPty will be written to contPty
 * This process loops forever.
 * This uses epoll in edge triggered mode to avoid a hard loop on POLLHUP
 * events when the user disconnects the virsh console via ctrl-]
 *
 * Returns 0 on success or -1 in case of error
 */
341 342 343 344
static int lxcControllerMain(int monitor,
                             int client,
                             int appPty,
                             int contPty)
345 346 347 348 349 350 351 352 353 354 355 356 357 358 359
{
    int rc = -1;
    int epollFd;
    struct epoll_event epollEvent;
    int numEvents;
    int numActive = 0;
    lxcTtyForwardFd_t fdArray[2];
    int timeout = -1;
    int curFdOff = 0;
    int writeFdOff = 0;

    fdArray[0].fd = appPty;
    fdArray[0].active = 0;
    fdArray[1].fd = contPty;
    fdArray[1].active = 0;
360 361 362 363

    VIR_DEBUG("monitor=%d client=%d appPty=%d contPty=%d",
              monitor, client, appPty, contPty);

364 365 366
    /* create the epoll fild descriptor */
    epollFd = epoll_create(2);
    if (0 > epollFd) {
367
        virReportSystemError(errno, "%s",
368
                             _("epoll_create(2) failed"));
369 370 371 372 373 374 375 376
        goto cleanup;
    }

    /* add the file descriptors the epoll fd */
    memset(&epollEvent, 0x00, sizeof(epollEvent));
    epollEvent.events = EPOLLIN|EPOLLET;    /* edge triggered */
    epollEvent.data.fd = appPty;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, appPty, &epollEvent)) {
377
        virReportSystemError(errno, "%s",
378
                             _("epoll_ctl(appPty) failed"));
379 380 381 382
        goto cleanup;
    }
    epollEvent.data.fd = contPty;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, contPty, &epollEvent)) {
383
        virReportSystemError(errno, "%s",
384
                             _("epoll_ctl(contPty) failed"));
385 386 387
        goto cleanup;
    }

388 389 390
    epollEvent.events = EPOLLIN;
    epollEvent.data.fd = monitor;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, monitor, &epollEvent)) {
391
        virReportSystemError(errno, "%s",
392
                             _("epoll_ctl(monitor) failed"));
393 394 395 396 397 398
        goto cleanup;
    }

    epollEvent.events = EPOLLHUP;
    epollEvent.data.fd = client;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, client, &epollEvent)) {
399
        virReportSystemError(errno, "%s",
400
                             _("epoll_ctl(client) failed"));
401 402 403
        goto cleanup;
    }

404 405 406 407
    while (1) {
        /* if active fd's, return if no events, else wait forever */
        timeout = (numActive > 0) ? 0 : -1;
        numEvents = epoll_wait(epollFd, &epollEvent, 1, timeout);
408 409 410
        if (numEvents > 0) {
            if (epollEvent.data.fd == monitor) {
                int fd = accept(monitor, NULL, 0);
411 412 413 414 415 416 417 418 419 420 421 422
                if (fd < 0) {
                    /* First reflex may be simply to declare accept failure
                       to be a fatal error.  However, accept may fail when
                       a client quits between the above epoll_wait and here.
                       That case is not fatal, but rather to be expected,
                       if not common, so ignore it.  */
                    if (ignorable_epoll_accept_errno(errno))
                        continue;
                    virReportSystemError(errno, "%s",
                                         _("accept(monitor,...) failed"));
                    goto cleanup;
                }
423
                if (client != -1) { /* Already connected, so kick new one out */
424
                    VIR_FORCE_CLOSE(fd);
425
                    continue;
426
                }
427 428 429 430
                client = fd;
                epollEvent.events = EPOLLHUP;
                epollEvent.data.fd = client;
                if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, client, &epollEvent)) {
431
                    virReportSystemError(errno, "%s",
432
                                         _("epoll_ctl(client) failed"));
433 434 435 436
                    goto cleanup;
                }
            } else if (client != -1 && epollEvent.data.fd == client) {
                if (0 > epoll_ctl(epollFd, EPOLL_CTL_DEL, client, &epollEvent)) {
437
                    virReportSystemError(errno, "%s",
438
                                         _("epoll_ctl(client) failed"));
439 440
                    goto cleanup;
                }
441
                VIR_FORCE_CLOSE(client);
442
            } else {
443 444 445 446 447 448 449
                if (epollEvent.events & EPOLLIN) {
                    curFdOff = epollEvent.data.fd == appPty ? 0 : 1;
                    if (!fdArray[curFdOff].active) {
                        fdArray[curFdOff].active = 1;
                        ++numActive;
                    }
                } else if (epollEvent.events & EPOLLHUP) {
450
                    VIR_DEBUG("EPOLLHUP from fd %d", epollEvent.data.fd);
451 452
                    continue;
                } else {
453
                    lxcError(VIR_ERR_INTERNAL_ERROR,
454 455 456
                             _("error event %d"), epollEvent.events);
                    goto cleanup;
                }
457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473
            }
        } else if (0 == numEvents) {
            if (2 == numActive) {
                /* both fds active, toggle between the two */
                curFdOff ^= 1;
            } else {
                /* only one active, if current is active, use it, else it */
                /* must be the other one (ie. curFd just went inactive) */
                curFdOff = fdArray[curFdOff].active ? curFdOff : curFdOff ^ 1;
            }

        } else  {
            if (EINTR == errno) {
                continue;
            }

            /* error */
474
            virReportSystemError(errno, "%s",
475
                                 _("epoll_wait() failed"));
476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498
            goto cleanup;

        }

        if (0 < numActive) {
            writeFdOff = curFdOff ^ 1;
            rc = lxcFdForward(fdArray[curFdOff].fd, fdArray[writeFdOff].fd);

            if (EAGAIN == rc) {
                /* this fd no longer has data, set it as inactive */
                --numActive;
                fdArray[curFdOff].active = 0;
            } else if (-1 == rc) {
                goto cleanup;
            }

        }

    }

    rc = 0;

cleanup:
499 500 501
    VIR_FORCE_CLOSE(appPty);
    VIR_FORCE_CLOSE(contPty);
    VIR_FORCE_CLOSE(epollFd);
502 503 504
    return rc;
}

505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522


/**
 * lxcControllerMoveInterfaces
 * @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
 */
static int lxcControllerMoveInterfaces(unsigned int nveths,
                                       char **veths,
                                       pid_t container)
{
    unsigned int i;
    for (i = 0 ; i < nveths ; i++)
523
        if (moveInterfaceToNetNs(veths[i], container) < 0)
524 525 526 527 528 529 530 531
            return -1;

    return 0;
}


/**
 * lxcCleanupInterfaces:
532 533
 * @nveths: number of interfaces
 * @veths: interface names
534 535 536 537 538 539 540 541 542 543
 *
 * Cleans up the container interfaces by deleting the veth device pairs.
 *
 * Returns 0 on success or -1 in case of error
 */
static int lxcControllerCleanupInterfaces(unsigned int nveths,
                                          char **veths)
{
    unsigned int i;
    for (i = 0 ; i < nveths ; i++)
544
        vethDelete(veths[i]);
545 546 547 548

    return 0;
}

549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567
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;
}

568
#ifndef MS_REC
569
# define MS_REC          16384
570 571 572
#endif

#ifndef MS_SLAVE
573
# define MS_SLAVE              (1<<19)
574
#endif
575 576

static int
577
lxcControllerRun(virDomainDefPtr def,
578 579 580 581 582 583 584 585
                 unsigned int nveths,
                 char **veths,
                 int monitor,
                 int client,
                 int appPty)
{
    int rc = -1;
    int control[2] = { -1, -1};
586
    int containerPty = -1;
587 588
    char *containerPtyPath;
    pid_t container = -1;
589 590 591
    virDomainFSDefPtr root;
    char *devpts = NULL;
    char *devptmx = NULL;
592 593

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

599 600
    root = virDomainGetRootFilesystem(def);

601 602 603
    if (lxcSetContainerResources(def) < 0)
        goto cleanup;

604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626
    /*
     * 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
     */
    if (root) {
        VIR_DEBUG0("Setting up private /dev/pts");
        if (unshare(CLONE_NEWNS) < 0) {
627
            virReportSystemError(errno, "%s",
628
                                 _("Cannot unshare mount namespace"));
629 630 631 632
            goto cleanup;
        }

        if (mount("", "/", NULL, MS_SLAVE|MS_REC, NULL) < 0) {
633
            virReportSystemError(errno, "%s",
634
                                 _("Failed to switch root mount into slave mode"));
635 636 637 638 639
            goto cleanup;
        }

        if (virAsprintf(&devpts, "%s/dev/pts", root->src) < 0 ||
            virAsprintf(&devptmx, "%s/dev/pts/ptmx", root->src) < 0) {
640
            virReportOOMError();
641 642 643
            goto cleanup;
        }

L
Laine Stump 已提交
644
        if (virFileMakePath(devpts) != 0) {
645
            virReportSystemError(errno,
646
                                 _("Failed to make path %s"),
647 648 649 650 651
                                 devpts);
            goto cleanup;
        }

        VIR_DEBUG("Mouting 'devpts' on %s", devpts);
652 653
        if (mount("devpts", devpts, "devpts", 0,
                  "newinstance,ptmxmode=0666,mode=0620,gid=5") < 0) {
654
            virReportSystemError(errno,
655
                                 _("Failed to mount devpts on %s"),
656 657 658 659 660
                                 devpts);
            goto cleanup;
        }

        if (access(devptmx, R_OK) < 0) {
661
            VIR_WARN0("Kernel does not support private devpts, using shared devpts");
662 663
            VIR_FREE(devptmx);
        }
664 665
    }

666 667 668 669 670 671
    if (devptmx) {
        VIR_DEBUG("Opening tty on private %s", devptmx);
        if (virFileOpenTtyAt(devptmx,
                             &containerPty,
                             &containerPtyPath,
                             0) < 0) {
672
            virReportSystemError(errno, "%s",
673
                                 _("Failed to allocate tty"));
674 675 676 677 678 679 680
            goto cleanup;
        }
    } else {
        VIR_DEBUG0("Opening tty on shared /dev/ptmx");
        if (virFileOpenTty(&containerPty,
                           &containerPtyPath,
                           0) < 0) {
681
            virReportSystemError(errno, "%s",
682
                                 _("Failed to allocate tty"));
683 684 685 686
            goto cleanup;
        }
    }

687 688
    if (lxcSetPersonality(def) < 0)
        goto cleanup;
689

690 691 692 693 694 695
    if ((container = lxcContainerStart(def,
                                       nveths,
                                       veths,
                                       control[1],
                                       containerPtyPath)) < 0)
        goto cleanup;
696
    VIR_FORCE_CLOSE(control[1]);
697 698 699 700 701 702 703

    if (lxcControllerMoveInterfaces(nveths, veths, container) < 0)
        goto cleanup;

    if (lxcContainerSendContinue(control[0]) < 0)
        goto cleanup;

D
Daniel P. Berrange 已提交
704 705 706 707 708
    /* Now the container is running, there's no need for us to keep
       any elevated capabilities */
    if (lxcControllerClearCapabilities() < 0)
        goto cleanup;

709 710 711
    rc = lxcControllerMain(monitor, client, appPty, containerPty);

cleanup:
712 713
    VIR_FREE(devptmx);
    VIR_FREE(devpts);
714 715
    VIR_FORCE_CLOSE(control[0]);
    VIR_FORCE_CLOSE(control[1]);
716
    VIR_FREE(containerPtyPath);
717
    VIR_FORCE_CLOSE(containerPty);
718

719
    if (container > 1) {
720
        int status;
721
        kill(container, SIGTERM);
722 723 724
        if (!(waitpid(container, &status, WNOHANG) == 0 &&
            WIFEXITED(status)))
            kill(container, SIGKILL);
725 726
        waitpid(container, NULL, 0);
    }
727 728 729 730
    return rc;
}


731
int main(int argc, char *argv[])
732 733
{
    pid_t pid;
734
    int rc = 1;
735
    int client;
736 737 738 739 740 741 742 743 744 745
    char *name = NULL;
    int nveths = 0;
    char **veths = NULL;
    int monitor = -1;
    int appPty = -1;
    int bg = 0;
    virCapsPtr caps = NULL;
    virDomainDefPtr def = NULL;
    char *configFile = NULL;
    char *sockpath = NULL;
746
    const struct option options[] = {
747 748 749 750 751 752 753
        { "background", 0, NULL, 'b' },
        { "name",   1, NULL, 'n' },
        { "veth",   1, NULL, 'v' },
        { "console", 1, NULL, 'c' },
        { "help", 0, NULL, 'h' },
        { 0, 0, 0, 0 },
    };
754

E
Eric Blake 已提交
755 756 757 758 759 760 761
    if (setlocale(LC_ALL, "") == NULL ||
        bindtextdomain(PACKAGE, LOCALEDIR) == NULL ||
        textdomain(PACKAGE) == NULL) {
        fprintf(stderr, _("%s: initialization failed\n"), argv[0]);
        exit(EXIT_FAILURE);
    }

762 763
    while (1) {
        int c;
764

765 766 767 768 769 770 771 772 773 774 775 776 777
        c = getopt_long(argc, argv, "dn:v:m:c:h",
                       options, NULL);

        if (c == -1)
            break;

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

        case 'n':
            if ((name = strdup(optarg)) == NULL) {
778
                virReportOOMError();
779
                goto cleanup;
780
            }
781 782 783 784
            break;

        case 'v':
            if (VIR_REALLOC_N(veths, nveths+1) < 0) {
785
                virReportOOMError();
786
                goto cleanup;
787
            }
788
            if ((veths[nveths++] = strdup(optarg)) == NULL) {
789
                virReportOOMError();
790
                goto cleanup;
791
            }
792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814
            break;

        case 'c':
            if (virStrToLong_i(optarg, NULL, 10, &appPty) < 0) {
                fprintf(stderr, "malformed --console argument '%s'", optarg);
                goto cleanup;
            }
            break;

        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");
            fprintf(stderr, "  -h, --help\n");
            fprintf(stderr, "\n");
            goto cleanup;
815 816 817 818
        }
    }


819 820 821 822 823 824 825 826 827 828
    if (name == NULL) {
        fprintf(stderr, "%s: missing --name argument for configuration\n", argv[0]);
        goto cleanup;
    }

    if (appPty < 0) {
        fprintf(stderr, "%s: missing --console argument for container PTY\n", argv[0]);
        goto cleanup;
    }

829
    if (getuid() != 0) {
830 831 832
        fprintf(stderr, "%s: must be run as the 'root' user\n", argv[0]);
        goto cleanup;
    }
833

834 835
    if ((caps = lxcCapsInit()) == NULL)
        goto cleanup;
836

837
    if ((configFile = virDomainConfigFile(LXC_STATE_DIR,
838 839
                                          name)) == NULL)
        goto cleanup;
840

841
    if ((def = virDomainDefParseFile(caps, configFile,
842
                                     VIR_DOMAIN_XML_INACTIVE)) == NULL)
843
        goto cleanup;
844

845
    if (def->nnets != nveths) {
846
        fprintf(stderr, "%s: expecting %d veths, but got %d\n",
847
                argv[0], def->nnets, nveths);
848
        goto cleanup;
849 850
    }

851 852
    if ((sockpath = lxcMonitorPath(def)) == NULL)
        goto cleanup;
853

854 855
    if ((monitor = lxcMonitorServer(sockpath)) < 0)
        goto cleanup;
856

857 858 859
    if (bg) {
        if ((pid = fork()) < 0)
            goto cleanup;
860

861 862
        if (pid > 0) {
            if ((rc = virFileWritePid(LXC_STATE_DIR, name, pid)) != 0) {
863
                virReportSystemError(rc,
864 865
                                     _("Unable to write pid file '%s/%s.pid'"),
                                     LXC_STATE_DIR, name);
866 867
                _exit(1);
            }
868

869 870 871 872
            /* First child now exits, allowing original caller
             * (ie libvirtd's LXC driver to complete their
             * waitpid & continue */
            _exit(0);
873 874
        }

875 876
        /* Don't hold onto any cwd we inherit from libvirtd either */
        if (chdir("/") < 0) {
877
            virReportSystemError(errno, "%s",
878
                                 _("Unable to change to root dir"));
879 880 881 882
            goto cleanup;
        }

        if (setsid() < 0) {
883
            virReportSystemError(errno, "%s",
884
                                 _("Unable to become session leader"));
885 886 887
            goto cleanup;
        }
    }
888

A
Amy Griffis 已提交
889 890 891
    /* Initialize logging */
    virLogSetFromEnv();

892
    /* Accept initial client which is the libvirtd daemon */
893
    if ((client = accept(monitor, NULL, 0)) < 0) {
894
        virReportSystemError(errno, "%s",
895
                             _("Failed to accept a connection from driver"));
896
        goto cleanup;
897 898
    }

899
    rc = lxcControllerRun(def, nveths, veths, monitor, client, appPty);
900 901


902
cleanup:
903 904
    if (def)
        virFileDeletePid(LXC_STATE_DIR, def->name);
905
    lxcControllerCleanupInterfaces(nveths, veths);
J
Jim Meyering 已提交
906 907
    if (sockpath)
        unlink(sockpath);
908 909 910 911
    VIR_FREE(sockpath);

    return rc;
}