lxc_controller.c 25.5 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
    rc = virCgroupSetMemory(cgroup, def->mem.max_balloon);
108
    if (rc != 0) {
109
        virReportSystemError(-rc,
110 111
                             _("Unable to set memory limit for domain %s"),
                             def->name);
112 113 114
        /* Don't fail if we can't set memory due to lack of kernel support */
        if (rc != -ENOENT)
            goto cleanup;
115
    }
D
Dan Smith 已提交
116

117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146
    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 已提交
147
    rc = virCgroupDenyAllDevices(cgroup);
148
    if (rc != 0) {
149
        virReportSystemError(-rc,
150 151 152 153
                             _("Unable to deny devices for domain %s"),
                             def->name);
        goto cleanup;
    }
D
Dan Smith 已提交
154 155 156 157 158 159 160

    for (i = 0; devices[i].type != 0; i++) {
        struct cgroup_device_policy *dev = &devices[i];
        rc = virCgroupAllowDevice(cgroup,
                                  dev->type,
                                  dev->major,
                                  dev->minor);
161
        if (rc != 0) {
162
            virReportSystemError(-rc,
163 164 165 166
                                 _("Unable to allow device %c:%d:%d for domain %s"),
                                 dev->type, dev->major, dev->minor, def->name);
            goto cleanup;
        }
D
Dan Smith 已提交
167 168
    }

169
    rc = virCgroupAllowDeviceMajor(cgroup, 'c', LXC_DEV_MAJ_PTY);
170
    if (rc != 0) {
171
        virReportSystemError(-rc,
172 173 174 175
                             _("Unable to allow PYT devices for domain %s"),
                             def->name);
        goto cleanup;
    }
176

D
Dan Smith 已提交
177 178
    rc = virCgroupAddTask(cgroup, getpid());
    if (rc != 0) {
179
        virReportSystemError(-rc,
180 181
                             _("Unable to add task %d to cgroup for domain %s"),
                             getpid(), def->name);
D
Dan Smith 已提交
182 183
    }

184 185
cleanup:
    virCgroupFree(&driver);
D
Dan Smith 已提交
186 187 188 189 190
    virCgroupFree(&cgroup);

    return rc;
}

191 192 193
static char*lxcMonitorPath(virDomainDefPtr def)
{
    char *sockpath;
194 195 196

    if (virAsprintf(&sockpath, "%s/%s.sock",
                    LXC_STATE_DIR, def->name) < 0)
197
        virReportOOMError();
198 199 200 201 202 203 204 205 206
    return sockpath;
}

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

    if ((fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0) {
207
        virReportSystemError(errno,
208 209
                             _("failed to create server socket '%s'"),
                             sockpath);
210 211 212 213 214 215
        goto error;
    }

    unlink(sockpath);
    memset(&addr, 0, sizeof(addr));
    addr.sun_family = AF_UNIX;
C
Chris Lalancette 已提交
216
    if (virStrcpyStatic(addr.sun_path, sockpath) == NULL) {
217
        lxcError(VIR_ERR_INTERNAL_ERROR,
C
Chris Lalancette 已提交
218 219 220
                 _("Socket path %s too long for destination"), sockpath);
        goto error;
    }
221 222

    if (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
223
        virReportSystemError(errno,
224 225
                             _("failed to bind server socket '%s'"),
                             sockpath);
226 227 228
        goto error;
    }
    if (listen(fd, 30 /* backlog */ ) < 0) {
229
        virReportSystemError(errno,
230 231
                             _("failed to listen server socket %s"),
                             sockpath);
232 233 234 235 236 237
        goto error;
    }

    return fd;

error:
238
    VIR_FORCE_CLOSE(fd);
239 240
    return -1;
}
241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261

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

262
        virReportSystemError(errno,
263 264
                             _("read of fd %d failed"),
                             readFd);
265 266 267 268
        goto cleanup;
    }

    if (1 != (safewrite(writeFd, buf, 1))) {
269
        virReportSystemError(errno,
270 271
                             _("write to fd %d failed"),
                             writeFd);
272 273 274 275 276 277 278 279 280
        goto cleanup;
    }

    rc = 0;

cleanup:
    return rc;
}

D
Daniel P. Berrange 已提交
281 282 283 284 285 286 287 288 289

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

    capng_clear(CAPNG_SELECT_BOTH);

    if ((ret = capng_apply(CAPNG_SELECT_BOTH)) < 0) {
290
        lxcError(VIR_ERR_INTERNAL_ERROR,
D
Daniel P. Berrange 已提交
291 292 293 294
                 _("failed to apply capabilities: %d"), ret);
        return -1;
    }
#else
295
    VIR_WARN0("libcap-ng support not compiled in, unable to clear capabilities");
D
Daniel P. Berrange 已提交
296 297 298 299
#endif
    return 0;
}

300 301 302 303 304
typedef struct _lxcTtyForwardFd_t {
    int fd;
    int active;
} lxcTtyForwardFd_t;

305 306 307 308 309 310 311 312 313 314 315
/* 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);
}

316
/**
317 318 319 320 321
 * 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
322 323 324 325 326 327 328 329
 *
 * 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
 */
330 331 332 333
static int lxcControllerMain(int monitor,
                             int client,
                             int appPty,
                             int contPty)
334 335 336 337 338 339 340 341 342 343 344 345 346 347 348
{
    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;
349 350 351 352

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

353 354 355
    /* create the epoll fild descriptor */
    epollFd = epoll_create(2);
    if (0 > epollFd) {
356
        virReportSystemError(errno, "%s",
357
                             _("epoll_create(2) failed"));
358 359 360 361 362 363 364 365
        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)) {
366
        virReportSystemError(errno, "%s",
367
                             _("epoll_ctl(appPty) failed"));
368 369 370 371
        goto cleanup;
    }
    epollEvent.data.fd = contPty;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, contPty, &epollEvent)) {
372
        virReportSystemError(errno, "%s",
373
                             _("epoll_ctl(contPty) failed"));
374 375 376
        goto cleanup;
    }

377 378 379
    epollEvent.events = EPOLLIN;
    epollEvent.data.fd = monitor;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, monitor, &epollEvent)) {
380
        virReportSystemError(errno, "%s",
381
                             _("epoll_ctl(monitor) failed"));
382 383 384 385 386 387
        goto cleanup;
    }

    epollEvent.events = EPOLLHUP;
    epollEvent.data.fd = client;
    if (0 > epoll_ctl(epollFd, EPOLL_CTL_ADD, client, &epollEvent)) {
388
        virReportSystemError(errno, "%s",
389
                             _("epoll_ctl(client) failed"));
390 391 392
        goto cleanup;
    }

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

494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511


/**
 * 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++)
512
        if (moveInterfaceToNetNs(veths[i], container) < 0)
513 514 515 516 517 518 519 520
            return -1;

    return 0;
}


/**
 * lxcCleanupInterfaces:
521 522
 * @nveths: number of interfaces
 * @veths: interface names
523 524 525 526 527 528 529 530 531 532
 *
 * 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++)
533
        vethDelete(veths[i]);
534 535 536 537

    return 0;
}

538
#ifndef MS_REC
539
# define MS_REC          16384
540 541 542
#endif

#ifndef MS_SLAVE
543
# define MS_SLAVE              (1<<19)
544
#endif
545 546

static int
547
lxcControllerRun(virDomainDefPtr def,
548 549 550 551 552 553 554 555
                 unsigned int nveths,
                 char **veths,
                 int monitor,
                 int client,
                 int appPty)
{
    int rc = -1;
    int control[2] = { -1, -1};
556
    int containerPty = -1;
557 558
    char *containerPtyPath;
    pid_t container = -1;
559 560 561
    virDomainFSDefPtr root;
    char *devpts = NULL;
    char *devptmx = NULL;
562 563

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

569 570
    root = virDomainGetRootFilesystem(def);

571 572 573
    if (lxcSetContainerResources(def) < 0)
        goto cleanup;

574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596
    /*
     * 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) {
597
            virReportSystemError(errno, "%s",
598
                                 _("Cannot unshare mount namespace"));
599 600 601 602
            goto cleanup;
        }

        if (mount("", "/", NULL, MS_SLAVE|MS_REC, NULL) < 0) {
603
            virReportSystemError(errno, "%s",
604
                                 _("Failed to switch root mount into slave mode"));
605 606 607 608 609
            goto cleanup;
        }

        if (virAsprintf(&devpts, "%s/dev/pts", root->src) < 0 ||
            virAsprintf(&devptmx, "%s/dev/pts/ptmx", root->src) < 0) {
610
            virReportOOMError();
611 612 613
            goto cleanup;
        }

L
Laine Stump 已提交
614
        if (virFileMakePath(devpts) != 0) {
615
            virReportSystemError(errno,
616
                                 _("Failed to make path %s"),
617 618 619 620 621 622
                                 devpts);
            goto cleanup;
        }

        VIR_DEBUG("Mouting 'devpts' on %s", devpts);
        if (mount("devpts", devpts, "devpts", 0, "newinstance,ptmxmode=0666") < 0) {
623
            virReportSystemError(errno,
624
                                 _("Failed to mount devpts on %s"),
625 626 627 628 629
                                 devpts);
            goto cleanup;
        }

        if (access(devptmx, R_OK) < 0) {
630
            VIR_WARN0("Kernel does not support private devpts, using shared devpts");
631 632
            VIR_FREE(devptmx);
        }
633 634
    }

635 636 637 638 639 640
    if (devptmx) {
        VIR_DEBUG("Opening tty on private %s", devptmx);
        if (virFileOpenTtyAt(devptmx,
                             &containerPty,
                             &containerPtyPath,
                             0) < 0) {
641
            virReportSystemError(errno, "%s",
642
                                 _("Failed to allocate tty"));
643 644 645 646 647 648 649
            goto cleanup;
        }
    } else {
        VIR_DEBUG0("Opening tty on shared /dev/ptmx");
        if (virFileOpenTty(&containerPty,
                           &containerPtyPath,
                           0) < 0) {
650
            virReportSystemError(errno, "%s",
651
                                 _("Failed to allocate tty"));
652 653 654 655 656
            goto cleanup;
        }
    }


657 658 659 660 661 662
    if ((container = lxcContainerStart(def,
                                       nveths,
                                       veths,
                                       control[1],
                                       containerPtyPath)) < 0)
        goto cleanup;
663
    VIR_FORCE_CLOSE(control[1]);
664 665 666 667 668 669 670

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

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

D
Daniel P. Berrange 已提交
671 672 673 674 675
    /* Now the container is running, there's no need for us to keep
       any elevated capabilities */
    if (lxcControllerClearCapabilities() < 0)
        goto cleanup;

676 677 678
    rc = lxcControllerMain(monitor, client, appPty, containerPty);

cleanup:
679 680
    VIR_FREE(devptmx);
    VIR_FREE(devpts);
681 682
    VIR_FORCE_CLOSE(control[0]);
    VIR_FORCE_CLOSE(control[1]);
683
    VIR_FREE(containerPtyPath);
684
    VIR_FORCE_CLOSE(containerPty);
685

686
    if (container > 1) {
687
        int status;
688
        kill(container, SIGTERM);
689 690 691
        if (!(waitpid(container, &status, WNOHANG) == 0 &&
            WIFEXITED(status)))
            kill(container, SIGKILL);
692 693
        waitpid(container, NULL, 0);
    }
694 695 696 697
    return rc;
}


698
int main(int argc, char *argv[])
699 700
{
    pid_t pid;
701
    int rc = 1;
702
    int client;
703 704 705 706 707 708 709 710 711 712
    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;
713
    const struct option options[] = {
714 715 716 717 718 719 720
        { "background", 0, NULL, 'b' },
        { "name",   1, NULL, 'n' },
        { "veth",   1, NULL, 'v' },
        { "console", 1, NULL, 'c' },
        { "help", 0, NULL, 'h' },
        { 0, 0, 0, 0 },
    };
721

E
Eric Blake 已提交
722 723 724 725 726 727 728
    if (setlocale(LC_ALL, "") == NULL ||
        bindtextdomain(PACKAGE, LOCALEDIR) == NULL ||
        textdomain(PACKAGE) == NULL) {
        fprintf(stderr, _("%s: initialization failed\n"), argv[0]);
        exit(EXIT_FAILURE);
    }

729 730
    while (1) {
        int c;
731

732 733 734 735 736 737 738 739 740 741 742 743 744
        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) {
745
                virReportOOMError();
746
                goto cleanup;
747
            }
748 749 750 751
            break;

        case 'v':
            if (VIR_REALLOC_N(veths, nveths+1) < 0) {
752
                virReportOOMError();
753
                goto cleanup;
754
            }
755
            if ((veths[nveths++] = strdup(optarg)) == NULL) {
756
                virReportOOMError();
757
                goto cleanup;
758
            }
759 760 761 762 763 764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781
            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;
782 783 784 785
        }
    }


786 787 788 789 790 791 792 793 794 795
    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;
    }

796
    if (getuid() != 0) {
797 798 799
        fprintf(stderr, "%s: must be run as the 'root' user\n", argv[0]);
        goto cleanup;
    }
800

801 802
    if ((caps = lxcCapsInit()) == NULL)
        goto cleanup;
803

804
    if ((configFile = virDomainConfigFile(LXC_STATE_DIR,
805 806
                                          name)) == NULL)
        goto cleanup;
807

808
    if ((def = virDomainDefParseFile(caps, configFile,
809
                                     VIR_DOMAIN_XML_INACTIVE)) == NULL)
810
        goto cleanup;
811

812
    if (def->nnets != nveths) {
813
        fprintf(stderr, "%s: expecting %d veths, but got %d\n",
814
                argv[0], def->nnets, nveths);
815
        goto cleanup;
816 817
    }

818 819
    if ((sockpath = lxcMonitorPath(def)) == NULL)
        goto cleanup;
820

821 822
    if ((monitor = lxcMonitorServer(sockpath)) < 0)
        goto cleanup;
823

824 825 826
    if (bg) {
        if ((pid = fork()) < 0)
            goto cleanup;
827

828 829
        if (pid > 0) {
            if ((rc = virFileWritePid(LXC_STATE_DIR, name, pid)) != 0) {
830
                virReportSystemError(rc,
831 832
                                     _("Unable to write pid file '%s/%s.pid'"),
                                     LXC_STATE_DIR, name);
833 834
                _exit(1);
            }
835

836 837 838 839
            /* First child now exits, allowing original caller
             * (ie libvirtd's LXC driver to complete their
             * waitpid & continue */
            _exit(0);
840 841
        }

842 843
        /* Don't hold onto any cwd we inherit from libvirtd either */
        if (chdir("/") < 0) {
844
            virReportSystemError(errno, "%s",
845
                                 _("Unable to change to root dir"));
846 847 848 849
            goto cleanup;
        }

        if (setsid() < 0) {
850
            virReportSystemError(errno, "%s",
851
                                 _("Unable to become session leader"));
852 853 854
            goto cleanup;
        }
    }
855

A
Amy Griffis 已提交
856 857 858
    /* Initialize logging */
    virLogSetFromEnv();

859
    /* Accept initial client which is the libvirtd daemon */
860
    if ((client = accept(monitor, NULL, 0)) < 0) {
861
        virReportSystemError(errno, "%s",
862
                             _("Failed to accept a connection from driver"));
863
        goto cleanup;
864 865
    }

866
    rc = lxcControllerRun(def, nveths, veths, monitor, client, appPty);
867 868


869
cleanup:
870 871
    if (def)
        virFileDeletePid(LXC_STATE_DIR, def->name);
872
    lxcControllerCleanupInterfaces(nveths, veths);
J
Jim Meyering 已提交
873 874
    if (sockpath)
        unlink(sockpath);
875 876 877 878
    VIR_FREE(sockpath);

    return rc;
}