lxc_controller.c 26.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 36 37
#include <paths.h>
#include <fcntl.h>
#include <signal.h>
38
#include <getopt.h>
39
#include <sys/mount.h>
E
Eric Blake 已提交
40
#include <locale.h>
41

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

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

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

57 58
#define VIR_FROM_THIS VIR_FROM_LXC

D
Dan Smith 已提交
59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
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)
{
76
    virCgroupPtr driver;
D
Dan Smith 已提交
77 78 79 80 81 82 83 84 85
    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},
86
        {'c', LXC_DEV_MAJ_TTY, LXC_DEV_MIN_TTY},
87
        {'c', LXC_DEV_MAJ_TTY, LXC_DEV_MIN_PTMX},
D
Dan Smith 已提交
88 89
        {0,   0, 0}};

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

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

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

109 110 111 112 113 114 115 116 117 118
    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;
        }
    }

119
    rc = virCgroupSetMemory(cgroup, def->mem.max_balloon);
120
    if (rc != 0) {
121
        virReportSystemError(-rc,
122 123
                             _("Unable to set memory limit for domain %s"),
                             def->name);
124 125 126
        /* Don't fail if we can't set memory due to lack of kernel support */
        if (rc != -ENOENT)
            goto cleanup;
127
    }
D
Dan Smith 已提交
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 158
    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 已提交
159
    rc = virCgroupDenyAllDevices(cgroup);
160
    if (rc != 0) {
161
        virReportSystemError(-rc,
162 163 164 165
                             _("Unable to deny devices for domain %s"),
                             def->name);
        goto cleanup;
    }
D
Dan Smith 已提交
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,
172 173
                                  dev->minor,
                                  VIR_CGROUP_DEVICE_RWM);
174
        if (rc != 0) {
175
            virReportSystemError(-rc,
176 177 178 179
                                 _("Unable to allow device %c:%d:%d for domain %s"),
                                 dev->type, dev->major, dev->minor, def->name);
            goto cleanup;
        }
D
Dan Smith 已提交
180 181
    }

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

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

198 199
cleanup:
    virCgroupFree(&driver);
D
Dan Smith 已提交
200 201 202 203 204
    virCgroupFree(&cgroup);

    return rc;
}

205 206 207
static char*lxcMonitorPath(virDomainDefPtr def)
{
    char *sockpath;
208 209 210

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

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

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

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

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

    return fd;

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

/**
 * 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;
        }

276
        virReportSystemError(errno,
277 278
                             _("read of fd %d failed"),
                             readFd);
279 280 281 282
        goto cleanup;
    }

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

    rc = 0;

cleanup:
    return rc;
}

D
Daniel P. Berrange 已提交
295 296 297 298 299 300 301 302 303

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

    capng_clear(CAPNG_SELECT_BOTH);

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

314 315 316 317 318
typedef struct _lxcTtyForwardFd_t {
    int fd;
    int active;
} lxcTtyForwardFd_t;

319 320 321 322 323 324 325 326 327 328 329
/* 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);
}

330
/**
331 332 333 334 335
 * 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
336 337 338 339 340 341 342 343
 *
 * 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
 */
344 345 346 347
static int lxcControllerMain(int monitor,
                             int client,
                             int appPty,
                             int contPty)
348 349 350 351 352 353 354 355 356 357 358 359 360 361 362
{
    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;
363 364 365 366

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

367 368 369
    /* create the epoll fild descriptor */
    epollFd = epoll_create(2);
    if (0 > epollFd) {
370
        virReportSystemError(errno, "%s",
371
                             _("epoll_create(2) failed"));
372 373 374 375 376 377 378 379
        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)) {
380
        virReportSystemError(errno, "%s",
381
                             _("epoll_ctl(appPty) failed"));
382 383 384 385
        goto cleanup;
    }
    epollEvent.data.fd = contPty;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, contPty, &epollEvent)) {
386
        virReportSystemError(errno, "%s",
387
                             _("epoll_ctl(contPty) failed"));
388 389 390
        goto cleanup;
    }

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

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

407 408 409 410
    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);
411 412 413
        if (numEvents > 0) {
            if (epollEvent.data.fd == monitor) {
                int fd = accept(monitor, NULL, 0);
414 415 416 417 418 419 420 421 422 423 424 425
                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;
                }
426
                if (client != -1) { /* Already connected, so kick new one out */
427
                    VIR_FORCE_CLOSE(fd);
428
                    continue;
429
                }
430 431 432 433
                client = fd;
                epollEvent.events = EPOLLHUP;
                epollEvent.data.fd = client;
                if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, client, &epollEvent)) {
434
                    virReportSystemError(errno, "%s",
435
                                         _("epoll_ctl(client) failed"));
436 437 438 439
                    goto cleanup;
                }
            } else if (client != -1 && epollEvent.data.fd == client) {
                if (0 > epoll_ctl(epollFd, EPOLL_CTL_DEL, client, &epollEvent)) {
440
                    virReportSystemError(errno, "%s",
441
                                         _("epoll_ctl(client) failed"));
442 443
                    goto cleanup;
                }
444
                VIR_FORCE_CLOSE(client);
445
            } else {
446 447 448 449 450 451 452
                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) {
453
                    VIR_DEBUG("EPOLLHUP from fd %d", epollEvent.data.fd);
454 455
                    continue;
                } else {
456
                    lxcError(VIR_ERR_INTERNAL_ERROR,
457 458 459
                             _("error event %d"), epollEvent.events);
                    goto cleanup;
                }
460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476
            }
        } 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 */
477
            virReportSystemError(errno, "%s",
478
                                 _("epoll_wait() failed"));
479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501
            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:
502 503 504
    VIR_FORCE_CLOSE(appPty);
    VIR_FORCE_CLOSE(contPty);
    VIR_FORCE_CLOSE(epollFd);
505 506 507
    return rc;
}

508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525


/**
 * 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++)
526
        if (moveInterfaceToNetNs(veths[i], container) < 0)
527 528 529 530 531 532 533 534
            return -1;

    return 0;
}


/**
 * lxcCleanupInterfaces:
535 536
 * @nveths: number of interfaces
 * @veths: interface names
537 538 539 540 541 542 543 544 545 546
 *
 * 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++)
547
        vethDelete(veths[i]);
548 549 550 551

    return 0;
}

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

571
#ifndef MS_REC
572
# define MS_REC          16384
573 574 575
#endif

#ifndef MS_SLAVE
576
# define MS_SLAVE              (1<<19)
577
#endif
578 579

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

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

602 603
    root = virDomainGetRootFilesystem(def);

604 605 606
    if (lxcSetContainerResources(def) < 0)
        goto cleanup;

607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629
    /*
     * 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) {
630
            virReportSystemError(errno, "%s",
631
                                 _("Cannot unshare mount namespace"));
632 633 634 635
            goto cleanup;
        }

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

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

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

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

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

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

690 691
    if (lxcSetPersonality(def) < 0)
        goto cleanup;
692

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

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

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

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

712 713 714
    rc = lxcControllerMain(monitor, client, appPty, containerPty);

cleanup:
715 716
    VIR_FREE(devptmx);
    VIR_FREE(devpts);
717 718
    VIR_FORCE_CLOSE(control[0]);
    VIR_FORCE_CLOSE(control[1]);
719
    VIR_FREE(containerPtyPath);
720
    VIR_FORCE_CLOSE(containerPty);
721

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


734
int main(int argc, char *argv[])
735 736
{
    pid_t pid;
737
    int rc = 1;
738
    int client;
739 740 741 742 743 744 745 746 747 748
    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;
749
    const struct option options[] = {
750 751 752 753 754 755 756
        { "background", 0, NULL, 'b' },
        { "name",   1, NULL, 'n' },
        { "veth",   1, NULL, 'v' },
        { "console", 1, NULL, 'c' },
        { "help", 0, NULL, 'h' },
        { 0, 0, 0, 0 },
    };
757

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

765 766
    while (1) {
        int c;
767

768 769 770 771 772 773 774 775 776 777 778 779 780
        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) {
781
                virReportOOMError();
782
                goto cleanup;
783
            }
784 785 786 787
            break;

        case 'v':
            if (VIR_REALLOC_N(veths, nveths+1) < 0) {
788
                virReportOOMError();
789
                goto cleanup;
790
            }
791
            if ((veths[nveths++] = strdup(optarg)) == NULL) {
792
                virReportOOMError();
793
                goto cleanup;
794
            }
795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817
            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;
818 819 820 821
        }
    }


822 823 824 825 826 827 828 829 830 831
    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;
    }

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

837 838
    if ((caps = lxcCapsInit()) == NULL)
        goto cleanup;
839

840
    if ((configFile = virDomainConfigFile(LXC_STATE_DIR,
841 842
                                          name)) == NULL)
        goto cleanup;
843

844
    if ((def = virDomainDefParseFile(caps, configFile,
845
                                     VIR_DOMAIN_XML_INACTIVE)) == NULL)
846
        goto cleanup;
847

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

854 855
    if ((sockpath = lxcMonitorPath(def)) == NULL)
        goto cleanup;
856

857 858
    if ((monitor = lxcMonitorServer(sockpath)) < 0)
        goto cleanup;
859

860 861 862
    if (bg) {
        if ((pid = fork()) < 0)
            goto cleanup;
863

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

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

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

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

A
Amy Griffis 已提交
892 893 894
    /* Initialize logging */
    virLogSetFromEnv();

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

902
    rc = lxcControllerRun(def, nveths, veths, monitor, client, appPty);
903 904


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

    return rc;
}