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

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

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

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

55 56
#define VIR_FROM_THIS VIR_FROM_LXC

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

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

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

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

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

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

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

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

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

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

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

    return rc;
}

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

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

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

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

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

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

    return fd;

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

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

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

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

    rc = 0;

cleanup:
    return rc;
}

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

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

    capng_clear(CAPNG_SELECT_BOTH);

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

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

315 316 317 318 319 320 321 322 323 324 325
/* 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);
}

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

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

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

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

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

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

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


/**
 * 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++)
522
        if (moveInterfaceToNetNs(veths[i], container) < 0)
523 524 525 526 527 528 529 530
            return -1;

    return 0;
}


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

    return 0;
}

548
#ifndef MS_REC
549
# define MS_REC          16384
550 551 552
#endif

#ifndef MS_SLAVE
553
# define MS_SLAVE              (1<<19)
554
#endif
555 556

static int
557
lxcControllerRun(virDomainDefPtr def,
558 559 560 561 562 563 564 565
                 unsigned int nveths,
                 char **veths,
                 int monitor,
                 int client,
                 int appPty)
{
    int rc = -1;
    int control[2] = { -1, -1};
566
    int containerPty = -1;
567 568
    char *containerPtyPath;
    pid_t container = -1;
569 570 571
    virDomainFSDefPtr root;
    char *devpts = NULL;
    char *devptmx = NULL;
572 573

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

579 580
    root = virDomainGetRootFilesystem(def);

581 582 583
    if (lxcSetContainerResources(def) < 0)
        goto cleanup;

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606
    /*
     * 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) {
607
            virReportSystemError(errno, "%s",
608
                                 _("Cannot unshare mount namespace"));
609 610 611 612
            goto cleanup;
        }

        if (mount("", "/", NULL, MS_SLAVE|MS_REC, NULL) < 0) {
613
            virReportSystemError(errno, "%s",
614
                                 _("Failed to switch root mount into slave mode"));
615 616 617 618 619
            goto cleanup;
        }

        if (virAsprintf(&devpts, "%s/dev/pts", root->src) < 0 ||
            virAsprintf(&devptmx, "%s/dev/pts/ptmx", root->src) < 0) {
620
            virReportOOMError();
621 622 623
            goto cleanup;
        }

L
Laine Stump 已提交
624
        if (virFileMakePath(devpts) != 0) {
625
            virReportSystemError(errno,
626
                                 _("Failed to make path %s"),
627 628 629 630 631
                                 devpts);
            goto cleanup;
        }

        VIR_DEBUG("Mouting 'devpts' on %s", devpts);
632 633
        if (mount("devpts", devpts, "devpts", 0,
                  "newinstance,ptmxmode=0666,mode=0620,gid=5") < 0) {
634
            virReportSystemError(errno,
635
                                 _("Failed to mount devpts on %s"),
636 637 638 639 640
                                 devpts);
            goto cleanup;
        }

        if (access(devptmx, R_OK) < 0) {
641
            VIR_WARN0("Kernel does not support private devpts, using shared devpts");
642 643
            VIR_FREE(devptmx);
        }
644 645
    }

646 647 648 649 650 651
    if (devptmx) {
        VIR_DEBUG("Opening tty on private %s", devptmx);
        if (virFileOpenTtyAt(devptmx,
                             &containerPty,
                             &containerPtyPath,
                             0) < 0) {
652
            virReportSystemError(errno, "%s",
653
                                 _("Failed to allocate tty"));
654 655 656 657 658 659 660
            goto cleanup;
        }
    } else {
        VIR_DEBUG0("Opening tty on shared /dev/ptmx");
        if (virFileOpenTty(&containerPty,
                           &containerPtyPath,
                           0) < 0) {
661
            virReportSystemError(errno, "%s",
662
                                 _("Failed to allocate tty"));
663 664 665 666 667
            goto cleanup;
        }
    }


668 669 670 671 672 673
    if ((container = lxcContainerStart(def,
                                       nveths,
                                       veths,
                                       control[1],
                                       containerPtyPath)) < 0)
        goto cleanup;
674
    VIR_FORCE_CLOSE(control[1]);
675 676 677 678 679 680 681

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

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

D
Daniel P. Berrange 已提交
682 683 684 685 686
    /* Now the container is running, there's no need for us to keep
       any elevated capabilities */
    if (lxcControllerClearCapabilities() < 0)
        goto cleanup;

687 688 689
    rc = lxcControllerMain(monitor, client, appPty, containerPty);

cleanup:
690 691
    VIR_FREE(devptmx);
    VIR_FREE(devpts);
692 693
    VIR_FORCE_CLOSE(control[0]);
    VIR_FORCE_CLOSE(control[1]);
694
    VIR_FREE(containerPtyPath);
695
    VIR_FORCE_CLOSE(containerPty);
696

697
    if (container > 1) {
698
        int status;
699
        kill(container, SIGTERM);
700 701 702
        if (!(waitpid(container, &status, WNOHANG) == 0 &&
            WIFEXITED(status)))
            kill(container, SIGKILL);
703 704
        waitpid(container, NULL, 0);
    }
705 706 707 708
    return rc;
}


709
int main(int argc, char *argv[])
710 711
{
    pid_t pid;
712
    int rc = 1;
713
    int client;
714 715 716 717 718 719 720 721 722 723
    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;
724
    const struct option options[] = {
725 726 727 728 729 730 731
        { "background", 0, NULL, 'b' },
        { "name",   1, NULL, 'n' },
        { "veth",   1, NULL, 'v' },
        { "console", 1, NULL, 'c' },
        { "help", 0, NULL, 'h' },
        { 0, 0, 0, 0 },
    };
732

E
Eric Blake 已提交
733 734 735 736 737 738 739
    if (setlocale(LC_ALL, "") == NULL ||
        bindtextdomain(PACKAGE, LOCALEDIR) == NULL ||
        textdomain(PACKAGE) == NULL) {
        fprintf(stderr, _("%s: initialization failed\n"), argv[0]);
        exit(EXIT_FAILURE);
    }

740 741
    while (1) {
        int c;
742

743 744 745 746 747 748 749 750 751 752 753 754 755
        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) {
756
                virReportOOMError();
757
                goto cleanup;
758
            }
759 760 761 762
            break;

        case 'v':
            if (VIR_REALLOC_N(veths, nveths+1) < 0) {
763
                virReportOOMError();
764
                goto cleanup;
765
            }
766
            if ((veths[nveths++] = strdup(optarg)) == NULL) {
767
                virReportOOMError();
768
                goto cleanup;
769
            }
770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789 790 791 792
            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;
793 794 795 796
        }
    }


797 798 799 800 801 802 803 804 805 806
    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;
    }

807
    if (getuid() != 0) {
808 809 810
        fprintf(stderr, "%s: must be run as the 'root' user\n", argv[0]);
        goto cleanup;
    }
811

812 813
    if ((caps = lxcCapsInit()) == NULL)
        goto cleanup;
814

815
    if ((configFile = virDomainConfigFile(LXC_STATE_DIR,
816 817
                                          name)) == NULL)
        goto cleanup;
818

819
    if ((def = virDomainDefParseFile(caps, configFile,
820
                                     VIR_DOMAIN_XML_INACTIVE)) == NULL)
821
        goto cleanup;
822

823
    if (def->nnets != nveths) {
824
        fprintf(stderr, "%s: expecting %d veths, but got %d\n",
825
                argv[0], def->nnets, nveths);
826
        goto cleanup;
827 828
    }

829 830
    if ((sockpath = lxcMonitorPath(def)) == NULL)
        goto cleanup;
831

832 833
    if ((monitor = lxcMonitorServer(sockpath)) < 0)
        goto cleanup;
834

835 836 837
    if (bg) {
        if ((pid = fork()) < 0)
            goto cleanup;
838

839 840
        if (pid > 0) {
            if ((rc = virFileWritePid(LXC_STATE_DIR, name, pid)) != 0) {
841
                virReportSystemError(rc,
842 843
                                     _("Unable to write pid file '%s/%s.pid'"),
                                     LXC_STATE_DIR, name);
844 845
                _exit(1);
            }
846

847 848 849 850
            /* First child now exits, allowing original caller
             * (ie libvirtd's LXC driver to complete their
             * waitpid & continue */
            _exit(0);
851 852
        }

853 854
        /* Don't hold onto any cwd we inherit from libvirtd either */
        if (chdir("/") < 0) {
855
            virReportSystemError(errno, "%s",
856
                                 _("Unable to change to root dir"));
857 858 859 860
            goto cleanup;
        }

        if (setsid() < 0) {
861
            virReportSystemError(errno, "%s",
862
                                 _("Unable to become session leader"));
863 864 865
            goto cleanup;
        }
    }
866

A
Amy Griffis 已提交
867 868 869
    /* Initialize logging */
    virLogSetFromEnv();

870
    /* Accept initial client which is the libvirtd daemon */
871
    if ((client = accept(monitor, NULL, 0)) < 0) {
872
        virReportSystemError(errno, "%s",
873
                             _("Failed to accept a connection from driver"));
874
        goto cleanup;
875 876
    }

877
    rc = lxcControllerRun(def, nveths, veths, monitor, client, appPty);
878 879


880
cleanup:
881 882
    if (def)
        virFileDeletePid(LXC_STATE_DIR, def->name);
883
    lxcControllerCleanupInterfaces(nveths, veths);
J
Jim Meyering 已提交
884 885
    if (sockpath)
        unlink(sockpath);
886 887 888 889
    VIR_FREE(sockpath);

    return rc;
}