libvirtd.c 45.7 KB
Newer Older
D
Daniel P. Berrange 已提交
1
/*
D
Daniel P. Berrange 已提交
2
 * libvirtd.c: daemon start of day, guest process & i/o management
D
Daniel P. Berrange 已提交
3
 *
4
 * Copyright (C) 2006-2012 Red Hat, Inc.
D
Daniel P. Berrange 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17
 * Copyright (C) 2006 Daniel P. Berrange
 *
 * 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
18
 * License along with this library.  If not, see
O
Osier Yang 已提交
19
 * <http://www.gnu.org/licenses/>.
D
Daniel P. Berrange 已提交
20 21 22 23
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

24
#include <config.h>
25

D
Daniel P. Berrange 已提交
26 27
#include <unistd.h>
#include <fcntl.h>
28 29
#include <sys/wait.h>
#include <sys/stat.h>
D
Daniel P. Berrange 已提交
30
#include <getopt.h>
31
#include <stdlib.h>
32
#include <grp.h>
E
Eric Blake 已提交
33
#include <locale.h>
34

35
#include "libvirt_internal.h"
36
#include "virerror.h"
E
Eric Blake 已提交
37
#include "virfile.h"
38
#include "virpidfile.h"
39
#include "virprocess.h"
40

41 42
#define VIR_FROM_THIS VIR_FROM_QEMU

D
Daniel P. Berrange 已提交
43
#include "libvirtd.h"
44
#include "libvirtd-config.h"
45

46
#include "viruuid.h"
47
#include "remote_driver.h"
48
#include "viralloc.h"
49
#include "virconf.h"
50
#include "virnetlink.h"
51 52
#include "virnetserver.h"
#include "remote.h"
53
#include "virhook.h"
54
#include "viraudit.h"
55
#include "virstring.h"
56 57
#include "locking/lock_manager.h"
#include "viraccessmanager.h"
D
Daniel P. Berrange 已提交
58

59
#ifdef WITH_DRIVER_MODULES
60
# include "driver.h"
61
#else
62 63 64 65 66 67
# ifdef WITH_QEMU
#  include "qemu/qemu_driver.h"
# endif
# ifdef WITH_LXC
#  include "lxc/lxc_driver.h"
# endif
68 69 70
# ifdef WITH_XEN
#  include "xen/xen_driver.h"
# endif
J
Jim Fehlig 已提交
71 72 73
# ifdef WITH_LIBXL
#  include "libxl/libxl_driver.h"
# endif
74 75 76
# ifdef WITH_UML
#  include "uml/uml_driver.h"
# endif
77 78 79
# ifdef WITH_VBOX
#  include "vbox/vbox_driver.h"
# endif
80 81 82
# ifdef WITH_NETWORK
#  include "network/bridge_driver.h"
# endif
83 84
# ifdef WITH_INTERFACE
#  include "interface/interface_driver.h"
85
# endif
86
# ifdef WITH_STORAGE
87 88 89 90 91 92 93 94
#  include "storage/storage_driver.h"
# endif
# ifdef WITH_NODE_DEVICES
#  include "node_device/node_device_driver.h"
# endif
# ifdef WITH_SECRETS
#  include "secret/secret_driver.h"
# endif
95 96 97
# ifdef WITH_NWFILTER
#  include "nwfilter/nwfilter_driver.h"
# endif
98
#endif
99

100 101
#include "configmake.h"

102 103
#include "virdbus.h"

104
#if WITH_SASL
105
virNetSASLContextPtr saslCtxt = NULL;
106
#endif
107 108
virNetServerProgramPtr remoteProgram = NULL;
virNetServerProgramPtr qemuProgram = NULL;
109
virNetServerProgramPtr lxcProgram = NULL;
110

111 112 113 114 115 116 117 118 119
enum {
    VIR_DAEMON_ERR_NONE = 0,
    VIR_DAEMON_ERR_PIDFILE,
    VIR_DAEMON_ERR_RUNDIR,
    VIR_DAEMON_ERR_INIT,
    VIR_DAEMON_ERR_SIGNAL,
    VIR_DAEMON_ERR_PRIVS,
    VIR_DAEMON_ERR_NETWORK,
    VIR_DAEMON_ERR_CONFIG,
120
    VIR_DAEMON_ERR_HOOKS,
121
    VIR_DAEMON_ERR_AUDIT,
122 123 124 125 126 127 128 129 130 131 132 133 134

    VIR_DAEMON_ERR_LAST
};

VIR_ENUM_DECL(virDaemonErr)
VIR_ENUM_IMPL(virDaemonErr, VIR_DAEMON_ERR_LAST,
              "Initialization successful",
              "Unable to obtain pidfile",
              "Unable to create rundir",
              "Unable to initialize libvirt",
              "Unable to setup signal handlers",
              "Unable to drop privileges",
              "Unable to initialize network sockets",
135
              "Unable to load configuration file",
136 137
              "Unable to look for hook scripts",
              "Unable to initialize audit system")
138

139
static int daemonForkIntoBackground(const char *argv0)
140
{
141 142 143 144
    int statuspipe[2];
    if (pipe(statuspipe) < 0)
        return -1;

145
    pid_t pid = fork();
D
Daniel P. Berrange 已提交
146 147 148
    switch (pid) {
    case 0:
        {
E
Eric Blake 已提交
149
            /* intermediate child */
D
Daniel P. Berrange 已提交
150 151
            int stdinfd = -1;
            int stdoutfd = -1;
152
            int nextpid;
D
Daniel P. Berrange 已提交
153

154
            VIR_FORCE_CLOSE(statuspipe[0]);
155

156
            if ((stdinfd = open("/dev/null", O_RDONLY)) < 0)
D
Daniel P. Berrange 已提交
157
                goto cleanup;
158
            if ((stdoutfd = open("/dev/null", O_WRONLY)) < 0)
D
Daniel P. Berrange 已提交
159 160 161 162 163 164 165
                goto cleanup;
            if (dup2(stdinfd, STDIN_FILENO) != STDIN_FILENO)
                goto cleanup;
            if (dup2(stdoutfd, STDOUT_FILENO) != STDOUT_FILENO)
                goto cleanup;
            if (dup2(stdoutfd, STDERR_FILENO) != STDERR_FILENO)
                goto cleanup;
E
Eric Blake 已提交
166
            if (stdinfd > STDERR_FILENO && VIR_CLOSE(stdinfd) < 0)
D
Daniel P. Berrange 已提交
167
                goto cleanup;
E
Eric Blake 已提交
168
            if (stdoutfd > STDERR_FILENO && VIR_CLOSE(stdoutfd) < 0)
D
Daniel P. Berrange 已提交
169 170 171 172 173 174 175
                goto cleanup;

            if (setsid() < 0)
                goto cleanup;

            nextpid = fork();
            switch (nextpid) {
E
Eric Blake 已提交
176
            case 0: /* grandchild */
177
                return statuspipe[1];
E
Eric Blake 已提交
178 179 180 181
            case -1: /* error */
                goto cleanup;
            default: /* intermediate child succeeded */
                _exit(EXIT_SUCCESS);
D
Daniel P. Berrange 已提交
182 183 184
            }

        cleanup:
185 186
            VIR_FORCE_CLOSE(stdoutfd);
            VIR_FORCE_CLOSE(stdinfd);
E
Eric Blake 已提交
187 188
            VIR_FORCE_CLOSE(statuspipe[1]);
            _exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
189 190 191

        }

E
Eric Blake 已提交
192 193
    case -1: /* error in parent */
        goto error;
D
Daniel P. Berrange 已提交
194 195 196

    default:
        {
E
Eric Blake 已提交
197
            /* parent */
198 199 200
            int ret;
            char status;

201
            VIR_FORCE_CLOSE(statuspipe[1]);
202 203

            /* We wait to make sure the first child forked successfully */
204
            if (virProcessWait(pid, NULL) < 0)
E
Eric Blake 已提交
205
                goto error;
206

E
Eric Blake 已提交
207 208 209
            /* If we get here, then the grandchild was spawned, so we
             * must exit.  Block until the second child initializes
             * successfully */
210 211 212 213 214
        again:
            ret = read(statuspipe[0], &status, 1);
            if (ret == -1 && errno == EINTR)
                goto again;

E
Eric Blake 已提交
215 216 217 218 219
            VIR_FORCE_CLOSE(statuspipe[0]);

            if (ret != 1) {
                char ebuf[1024];

220
                fprintf(stderr,
E
Eric Blake 已提交
221 222 223 224 225 226 227 228
                        _("%s: error: unable to determine if daemon is "
                          "running: %s\n"), argv0,
                        virStrerror(errno, ebuf, sizeof(ebuf)));
                exit(EXIT_FAILURE);
            } else if (status != 0) {
                fprintf(stderr,
                        _("%s: error: %s. Check /var/log/messages or run "
                          "without --daemon for more info.\n"), argv0,
229
                        virDaemonErrTypeToString(status));
E
Eric Blake 已提交
230
                exit(EXIT_FAILURE);
231
            }
E
Eric Blake 已提交
232
            _exit(EXIT_SUCCESS);
D
Daniel P. Berrange 已提交
233 234
        }
    }
E
Eric Blake 已提交
235 236 237 238 239

error:
    VIR_FORCE_CLOSE(statuspipe[0]);
    VIR_FORCE_CLOSE(statuspipe[1]);
    return -1;
D
Daniel P. Berrange 已提交
240 241
}

242 243

static int
244 245
daemonPidFilePath(bool privileged,
                  char **pidfile)
246
{
247
    if (privileged) {
248 249
        if (VIR_STRDUP(*pidfile, LOCALSTATEDIR "/run/libvirtd.pid") < 0)
            goto error;
250
    } else {
251 252
        char *rundir = NULL;
        mode_t old_umask;
253

254
        if (!(rundir = virGetUserRuntimeDirectory()))
255
            goto error;
256

257 258 259 260 261 262 263 264 265
        old_umask = umask(077);
        if (virFileMakePath(rundir) < 0) {
            umask(old_umask);
            goto error;
        }
        umask(old_umask);

        if (virAsprintf(pidfile, "%s/libvirtd.pid", rundir) < 0) {
            VIR_FREE(rundir);
266
            goto error;
E
Eric Blake 已提交
267
        }
268

269
        VIR_FREE(rundir);
270 271 272
    }

    return 0;
273

274
error:
275
    return -1;
276 277
}

278 279 280 281 282
static int
daemonUnixSocketPaths(struct daemonConfig *config,
                      bool privileged,
                      char **sockfile,
                      char **rosockfile)
J
John Levon 已提交
283
{
284 285
    if (config->unix_sock_dir) {
        if (virAsprintf(sockfile, "%s/libvirt-sock", config->unix_sock_dir) < 0)
286
            goto error;
287 288
        if (privileged &&
            virAsprintf(rosockfile, "%s/libvirt-sock-ro", config->unix_sock_dir) < 0)
289
            goto error;
D
Daniel P. Berrange 已提交
290
    } else {
291
        if (privileged) {
292 293 294
            if (VIR_STRDUP(*sockfile, LOCALSTATEDIR "/run/libvirt/libvirt-sock") < 0 ||
                VIR_STRDUP(*rosockfile, LOCALSTATEDIR "/run/libvirt/libvirt-sock-ro") < 0)
                goto error;
295
        } else {
296 297
            char *rundir = NULL;
            mode_t old_umask;
298

299
            if (!(rundir = virGetUserRuntimeDirectory()))
300
                goto error;
301

302 303 304 305 306 307 308
            old_umask = umask(077);
            if (virFileMakePath(rundir) < 0) {
                umask(old_umask);
                goto error;
            }
            umask(old_umask);

309
            if (virAsprintf(sockfile, "%s/libvirt-sock", rundir) < 0) {
310
                VIR_FREE(rundir);
311
                goto error;
312
            }
313

314
            VIR_FREE(rundir);
315 316 317
        }
    }
    return 0;
318

319 320
error:
    return -1;
D
Daniel P. Berrange 已提交
321 322
}

323 324 325

static void daemonErrorHandler(void *opaque ATTRIBUTE_UNUSED,
                               virErrorPtr err ATTRIBUTE_UNUSED)
326 327 328 329 330
{
    /* Don't do anything, since logging infrastructure already
     * took care of reporting the error */
}

331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347
static int daemonErrorLogFilter(virErrorPtr err, int priority)
{
    /* These error codes don't really reflect real errors. They
     * are expected events that occur when an app tries to check
     * whether a particular guest already exists. This filters
     * them to a lower log level to prevent pollution of syslog
     */
    switch (err->code) {
    case VIR_ERR_NO_DOMAIN:
    case VIR_ERR_NO_NETWORK:
    case VIR_ERR_NO_STORAGE_POOL:
    case VIR_ERR_NO_STORAGE_VOL:
    case VIR_ERR_NO_NODE_DEVICE:
    case VIR_ERR_NO_INTERFACE:
    case VIR_ERR_NO_NWFILTER:
    case VIR_ERR_NO_SECRET:
    case VIR_ERR_NO_DOMAIN_SNAPSHOT:
348
    case VIR_ERR_OPERATION_INVALID:
349 350 351 352 353 354
        return VIR_LOG_DEBUG;
    }

    return priority;
}

355 356
static void daemonInitialize(void)
{
357 358 359 360 361 362 363
    /*
     * Note that the order is important: the first ones have a higher
     * priority when calling virStateInitialize. We must register
     * the network, storage and nodedev drivers before any domain
     * drivers, since their resources must be auto-started before
     * any domains can be auto-started.
     */
364 365 366
#ifdef WITH_DRIVER_MODULES
    /* We don't care if any of these fail, because the whole point
     * is to allow users to only install modules they want to use.
D
Dan Kenigsberg 已提交
367
     * If they try to open a connection for a module that
368 369
     * is not loaded they'll get a suitable error at that point
     */
370
# ifdef WITH_NETWORK
371
    virDriverLoadModule("network");
372 373
# endif
# ifdef WITH_STORAGE
374
    virDriverLoadModule("storage");
375 376
# endif
# ifdef WITH_NODE_DEVICES
377
    virDriverLoadModule("nodedev");
378 379
# endif
# ifdef WITH_SECRETS
380
    virDriverLoadModule("secret");
381 382 383 384
# endif
# ifdef WITH_NWFILTER
    virDriverLoadModule("nwfilter");
# endif
385
# ifdef WITH_INTERFACE
386 387
    virDriverLoadModule("interface");
# endif
388 389 390 391 392 393
# ifdef WITH_XEN
    virDriverLoadModule("xen");
# endif
# ifdef WITH_LIBXL
    virDriverLoadModule("libxl");
# endif
394
# ifdef WITH_QEMU
395
    virDriverLoadModule("qemu");
396 397
# endif
# ifdef WITH_LXC
398
    virDriverLoadModule("lxc");
399 400
# endif
# ifdef WITH_UML
401
    virDriverLoadModule("uml");
402
# endif
403 404 405
# ifdef WITH_VBOX
    virDriverLoadModule("vbox");
# endif
406
#else
407
# ifdef WITH_NETWORK
408
    networkRegister();
409
# endif
410
# ifdef WITH_INTERFACE
411
    interfaceRegister();
412
# endif
413
# ifdef WITH_STORAGE
414
    storageRegister();
415
# endif
416
# ifdef WITH_NODE_DEVICES
417
    nodedevRegister();
418 419
# endif
# ifdef WITH_SECRETS
420
    secretRegister();
421
# endif
422 423 424
# ifdef WITH_NWFILTER
    nwfilterRegister();
# endif
425 426 427
# ifdef WITH_XEN
    xenRegister();
# endif
J
Jim Fehlig 已提交
428 429 430
# ifdef WITH_LIBXL
    libxlRegister();
# endif
431
# ifdef WITH_QEMU
432
    qemuRegister();
433 434
# endif
# ifdef WITH_LXC
435
    lxcRegister();
436 437
# endif
# ifdef WITH_UML
438
    umlRegister();
439
# endif
440 441 442
# ifdef WITH_VBOX
    vboxRegister();
# endif
443
#endif
444 445 446
}


447 448 449 450 451 452 453 454 455 456
static int daemonSetupNetworking(virNetServerPtr srv,
                                 struct daemonConfig *config,
                                 const char *sock_path,
                                 const char *sock_path_ro,
                                 bool ipsock,
                                 bool privileged)
{
    virNetServerServicePtr svc = NULL;
    virNetServerServicePtr svcRO = NULL;
    virNetServerServicePtr svcTCP = NULL;
457
#if WITH_GNUTLS
458
    virNetServerServicePtr svcTLS = NULL;
459
#endif
460 461 462 463 464 465 466 467
    gid_t unix_sock_gid = 0;
    int unix_sock_ro_mask = 0;
    int unix_sock_rw_mask = 0;

    if (config->unix_sock_group) {
        if (virGetGroupID(config->unix_sock_group, &unix_sock_gid) < 0)
            return -1;
    }
468

469 470 471 472
    if (virStrToLong_i(config->unix_sock_ro_perms, NULL, 8, &unix_sock_ro_mask) != 0) {
        VIR_ERROR(_("Failed to parse mode '%s'"), config->unix_sock_ro_perms);
        goto error;
    }
473

474 475 476 477
    if (virStrToLong_i(config->unix_sock_rw_perms, NULL, 8, &unix_sock_rw_mask) != 0) {
        VIR_ERROR(_("Failed to parse mode '%s'"), config->unix_sock_rw_perms);
        goto error;
    }
478

479
    VIR_DEBUG("Registering unix socket %s", sock_path);
480 481 482 483
    if (!(svc = virNetServerServiceNewUNIX(sock_path,
                                           unix_sock_rw_mask,
                                           unix_sock_gid,
                                           config->auth_unix_rw,
484
#if WITH_GNUTLS
485 486
                                           NULL,
#endif
487
                                           false,
M
Michal Privoznik 已提交
488
                                           config->max_queued_clients,
489
                                           config->max_client_requests)))
490
        goto error;
491 492 493 494 495 496
    if (sock_path_ro) {
        VIR_DEBUG("Registering unix socket %s", sock_path_ro);
        if (!(svcRO = virNetServerServiceNewUNIX(sock_path_ro,
                                                 unix_sock_ro_mask,
                                                 unix_sock_gid,
                                                 config->auth_unix_ro,
497
#if WITH_GNUTLS
498 499
                                                 NULL,
#endif
500
                                                 true,
M
Michal Privoznik 已提交
501
                                                 config->max_queued_clients,
502
                                                 config->max_client_requests)))
503 504
            goto error;
    }
505

506 507 508 509
    if (virNetServerAddService(srv, svc,
                               config->mdns_adv && !ipsock ?
                               "_libvirt._tcp" :
                               NULL) < 0)
510
        goto error;
511

512 513 514
    if (svcRO &&
        virNetServerAddService(srv, svcRO, NULL) < 0)
        goto error;
515

516
    if (ipsock) {
517
        if (config->listen_tcp) {
518 519
            VIR_DEBUG("Registering TCP socket %s:%s",
                      config->listen_addr, config->tcp_port);
520 521 522
            if (!(svcTCP = virNetServerServiceNewTCP(config->listen_addr,
                                                     config->tcp_port,
                                                     config->auth_tcp,
523
#if WITH_GNUTLS
524 525
                                                     NULL,
#endif
526
                                                     false,
M
Michal Privoznik 已提交
527
                                                     config->max_queued_clients,
528
                                                     config->max_client_requests)))
529
                goto error;
530

531 532 533
            if (virNetServerAddService(srv, svcTCP,
                                       config->mdns_adv ? "_libvirt._tcp" : NULL) < 0)
                goto error;
534 535
        }

536
#if WITH_GNUTLS
537 538
        if (config->listen_tls) {
            virNetTLSContextPtr ctxt = NULL;
539

540 541 542 543 544 545 546 547
            if (config->ca_file ||
                config->cert_file ||
                config->key_file) {
                if (!(ctxt = virNetTLSContextNewServer(config->ca_file,
                                                       config->crl_file,
                                                       config->cert_file,
                                                       config->key_file,
                                                       (const char *const*)config->tls_allowed_dn_list,
548
                                                       config->tls_no_sanity_certificate ? false : true,
549 550 551 552 553 554
                                                       config->tls_no_verify_certificate ? false : true)))
                    goto error;
            } else {
                if (!(ctxt = virNetTLSContextNewServerPath(NULL,
                                                           !privileged,
                                                           (const char *const*)config->tls_allowed_dn_list,
555
                                                           config->tls_no_sanity_certificate ? false : true,
556 557
                                                           config->tls_no_verify_certificate ? false : true)))
                    goto error;
558 559
            }

560 561
            VIR_DEBUG("Registering TLS socket %s:%s",
                      config->listen_addr, config->tls_port);
562 563 564 565
            if (!(svcTLS =
                  virNetServerServiceNewTCP(config->listen_addr,
                                            config->tls_port,
                                            config->auth_tls,
566
                                            ctxt,
567
                                            false,
M
Michal Privoznik 已提交
568
                                            config->max_queued_clients,
569
                                            config->max_client_requests))) {
570
                virObjectUnref(ctxt);
571
                goto error;
572
            }
573 574 575 576
            if (virNetServerAddService(srv, svcTLS,
                                       config->mdns_adv &&
                                       !config->listen_tcp ? "_libvirt._tcp" : NULL) < 0)
                goto error;
577

578
            virObjectUnref(ctxt);
579
        }
580 581 582 583 584 585 586 587
#else
        (void)privileged;
        if (config->listen_tls) {
            virReportError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                           _("This libvirtd build does not support TLS"));
            goto error;
        }
#endif
588 589
    }

590
#if WITH_SASL
591 592
    if (config->auth_unix_rw == REMOTE_AUTH_SASL ||
        config->auth_unix_ro == REMOTE_AUTH_SASL ||
593
# if WITH_GNUTLS
594 595 596
        config->auth_tls == REMOTE_AUTH_SASL ||
# endif
        config->auth_tcp == REMOTE_AUTH_SASL) {
597 598 599
        saslCtxt = virNetSASLContextNewServer(
            (const char *const*)config->sasl_allowed_username_list);
        if (!saslCtxt)
600
            goto error;
D
Daniel P. Berrange 已提交
601
    }
602
#endif
D
Daniel P. Berrange 已提交
603 604

    return 0;
605

606
error:
607
#if WITH_GNUTLS
608
    virObjectUnref(svcTLS);
609
#endif
610 611 612
    virObjectUnref(svcTCP);
    virObjectUnref(svc);
    virObjectUnref(svcRO);
613
    return -1;
D
Daniel P. Berrange 已提交
614 615 616
}


617 618
/*
 * Set up the logging environment
619 620
 * By default if daemonized all errors go to the logfile libvirtd.log,
 * but if verbose or error debugging is asked for then also output
621
 * informational and debug messages. Default size if 64 kB.
622
 */
623
static int
624 625 626 627
daemonSetupLogging(struct daemonConfig *config,
                   bool privileged,
                   bool verbose,
                   bool godaemon)
628
{
629 630
    virLogReset();

631
    /*
632 633 634 635 636 637 638 639 640 641 642 643
     * Libvirtd's order of precedence is:
     * cmdline > environment > config
     *
     * In order to achieve this, we must process configuration in
     * different order for the log level versus the filters and
     * outputs. Because filters and outputs append, we have to look at
     * the environment first and then only check the config file if
     * there was no result from the environment. The default output is
     * then applied only if there was no setting from either of the
     * first two. Because we don't have a way to determine if the log
     * level has been set, we must process variables in the opposite
     * order, each one overriding the previous.
644
     */
645 646
    if (config->log_level != 0)
        virLogSetDefaultPriority(config->log_level);
647

648
    virLogSetFromEnv();
649

650
    virLogSetBufferSize(config->log_buffer_size);
651

652 653 654 655 656
    if (virLogGetNbFilters() == 0)
        virLogParseFilters(config->log_filters);

    if (virLogGetNbOutputs() == 0)
        virLogParseOutputs(config->log_outputs);
657 658

    /*
659 660 661 662
     * If no defined outputs, and either running
     * as daemon or not on a tty, then first try
     * to direct it to the systemd journal
     * (if it exists)....
663
     */
664 665
    if (virLogGetNbOutputs() == 0 &&
        (godaemon || !isatty(STDIN_FILENO))) {
666 667 668
        char *tmp;
        if (access("/run/systemd/journal/socket", W_OK) >= 0) {
            if (virAsprintf(&tmp, "%d:journald", virLogGetDefaultPriority()) < 0)
669
                goto error;
670 671 672 673 674 675 676
            virLogParseOutputs(tmp);
            VIR_FREE(tmp);
        }
    }

    /*
     * otherwise direct to libvirtd.log when running
677
     * as daemon. Otherwise the default output is stderr.
678
     */
679
    if (virLogGetNbOutputs() == 0) {
680
        char *tmp = NULL;
681

682
        if (godaemon) {
683
            if (privileged) {
684 685 686
                if (virAsprintf(&tmp, "%d:file:%s/log/libvirt/libvirtd.log",
                                virLogGetDefaultPriority(),
                                LOCALSTATEDIR) == -1)
687
                    goto error;
688
            } else {
689
                char *logdir = virGetUserCacheDirectory();
690 691 692 693 694 695 696 697
                mode_t old_umask;

                if (!logdir)
                    goto error;

                old_umask = umask(077);
                if (virFileMakePath(logdir) < 0) {
                    umask(old_umask);
698
                    goto error;
699 700
                }
                umask(old_umask);
701

702 703 704
                if (virAsprintf(&tmp, "%d:file:%s/libvirtd.log",
                                virLogGetDefaultPriority(), logdir) == -1) {
                    VIR_FREE(logdir);
705
                    goto error;
E
Eric Blake 已提交
706
                }
707
                VIR_FREE(logdir);
708
            }
709
        } else {
710
            if (virAsprintf(&tmp, "%d:stderr", virLogGetDefaultPriority()) < 0)
711
                goto error;
712
        }
713 714
        virLogParseOutputs(tmp);
        VIR_FREE(tmp);
715
    }
716 717 718 719 720 721 722

    /*
     * Command line override for --verbose
     */
    if ((verbose) && (virLogGetDefaultPriority() > VIR_LOG_INFO))
        virLogSetDefaultPriority(VIR_LOG_INFO);

723 724 725 726 727 728
    return 0;

error:
    return -1;
}

729

730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749
static int
daemonSetupAccessManager(struct daemonConfig *config)
{
    virAccessManagerPtr mgr;
    const char *none[] = { "none", NULL };
    const char **driver = (const char **)config->access_drivers;

    if (!driver ||
        !driver[0])
        driver = none;

    if (!(mgr = virAccessManagerNewStack(driver)))
        return -1;

    virAccessManagerSetDefault(mgr);
    virObjectUnref(mgr);
    return 0;
}


750 751
/* Display version information. */
static void
752
daemonVersion(const char *argv0)
753
{
754
    printf("%s (%s) %s\n", argv0, PACKAGE_NAME, PACKAGE_VERSION);
755 756
}

J
John Levon 已提交
757 758
#ifdef __sun
static int
759
daemonSetupPrivs(void)
J
John Levon 已提交
760
{
761
    chown("/var/run/libvirt", SYSTEM_UID, SYSTEM_UID);
J
John Levon 已提交
762

763 764
    if (__init_daemon_priv(PU_RESETGROUPS | PU_CLEARLIMITSET,
                           SYSTEM_UID, SYSTEM_UID, PRIV_XVM_CONTROL, NULL)) {
765
        VIR_ERROR(_("additional privileges are required"));
J
John Levon 已提交
766 767 768
        return -1;
    }

769 770
    if (priv_set(PRIV_OFF, PRIV_ALLSETS, PRIV_FILE_LINK_ANY, PRIV_PROC_INFO,
                 PRIV_PROC_SESSION, PRIV_PROC_EXEC, PRIV_PROC_FORK, NULL)) {
771
        VIR_ERROR(_("failed to set reduced privileges"));
J
John Levon 已提交
772 773 774 775 776 777
        return -1;
    }

    return 0;
}
#else
778
# define daemonSetupPrivs() 0
J
John Levon 已提交
779 780
#endif

781

782 783 784
static void daemonShutdownHandler(virNetServerPtr srv,
                                  siginfo_t *sig ATTRIBUTE_UNUSED,
                                  void *opaque ATTRIBUTE_UNUSED)
785
{
786 787
    virNetServerQuit(srv);
}
788

789 790 791 792 793 794
static void daemonReloadHandler(virNetServerPtr srv ATTRIBUTE_UNUSED,
                                siginfo_t *sig ATTRIBUTE_UNUSED,
                                void *opaque ATTRIBUTE_UNUSED)
{
        VIR_INFO("Reloading configuration on SIGHUP");
        virHookCall(VIR_HOOK_DRIVER_DAEMON, "-",
795
                    VIR_HOOK_DAEMON_OP_RELOAD, SIGHUP, "SIGHUP", NULL, NULL);
796 797 798 799
        if (virStateReload() < 0)
            VIR_WARN("Error while reloading drivers");
}

800 801 802
static int daemonSetupSignals(virNetServerPtr srv)
{
    if (virNetServerAddSignalHandler(srv, SIGINT, daemonShutdownHandler, NULL) < 0)
803
        return -1;
804 805 806 807
    if (virNetServerAddSignalHandler(srv, SIGQUIT, daemonShutdownHandler, NULL) < 0)
        return -1;
    if (virNetServerAddSignalHandler(srv, SIGTERM, daemonShutdownHandler, NULL) < 0)
        return -1;
808 809
    if (virNetServerAddSignalHandler(srv, SIGHUP, daemonReloadHandler, NULL) < 0)
        return -1;
810 811
    return 0;
}
812

813 814 815 816 817 818 819 820 821 822 823 824

static void daemonInhibitCallback(bool inhibit, void *opaque)
{
    virNetServerPtr srv = opaque;

    if (inhibit)
        virNetServerAddShutdownInhibition(srv);
    else
        virNetServerRemoveShutdownInhibition(srv);
}


825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890
#ifdef HAVE_DBUS
static DBusConnection *sessionBus;
static DBusConnection *systemBus;

static void daemonStopWorker(void *opaque)
{
    virNetServerPtr srv = opaque;

    VIR_DEBUG("Begin stop srv=%p", srv);

    ignore_value(virStateStop());

    VIR_DEBUG("Completed stop srv=%p", srv);

    /* Exit libvirtd cleanly */
    virNetServerQuit(srv);
}


/* We do this in a thread to not block the main loop */
static void daemonStop(virNetServerPtr srv)
{
    virThread thr;
    virObjectRef(srv);
    if (virThreadCreate(&thr, false, daemonStopWorker, srv) < 0)
        virObjectUnref(srv);
}


static DBusHandlerResult
handleSessionMessageFunc(DBusConnection *connection ATTRIBUTE_UNUSED,
                         DBusMessage *message,
                         void *opaque)
{
    virNetServerPtr srv = opaque;

    VIR_DEBUG("srv=%p", srv);

    if (dbus_message_is_signal(message,
                               DBUS_INTERFACE_LOCAL,
                               "Disconnected"))
        daemonStop(srv);

    return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}


static DBusHandlerResult
handleSystemMessageFunc(DBusConnection *connection ATTRIBUTE_UNUSED,
                        DBusMessage *message,
                        void *opaque)
{
    virNetServerPtr srv = opaque;

    VIR_DEBUG("srv=%p", srv);

    if (dbus_message_is_signal(message,
                               "org.freedesktop.login1.Manager",
                               "PrepareForShutdown"))
        daemonStop(srv);

    return DBUS_HANDLER_RESULT_NOT_YET_HANDLED;
}
#endif


891 892 893
static void daemonRunStateInit(void *opaque)
{
    virNetServerPtr srv = opaque;
894 895 896
    virIdentityPtr sysident = virIdentityGetSystem();

    virIdentitySetCurrent(sysident);
897

898 899 900 901
    /* Since driver initialization can take time inhibit daemon shutdown until
       we're done so clients get a chance to connect */
    daemonInhibitCallback(true, srv);

902
    /* Start the stateful HV drivers
E
Eric Blake 已提交
903
     * This is deliberately done after telling the parent process
904 905
     * we're ready, since it can take a long time and this will
     * seriously delay OS bootup process */
906 907 908
    if (virStateInitialize(virNetServerIsPrivileged(srv),
                           daemonInhibitCallback,
                           srv) < 0) {
909
        VIR_ERROR(_("Driver state initialization failed"));
910 911
        /* Ensure the main event loop quits */
        kill(getpid(), SIGTERM);
912
        goto cleanup;
913 914
    }

915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933
#ifdef HAVE_DBUS
    /* Tie the non-priviledged libvirtd to the session/shutdown lifecycle */
    if (!virNetServerIsPrivileged(srv)) {

        sessionBus = virDBusGetSessionBus();
        if (sessionBus != NULL)
            dbus_connection_add_filter(sessionBus,
                                       handleSessionMessageFunc, srv, NULL);

        systemBus = virDBusGetSystemBus();
        if (systemBus != NULL) {
            dbus_connection_add_filter(systemBus,
                                       handleSystemMessageFunc, srv, NULL);
            dbus_bus_add_match(systemBus,
                               "type='signal',sender='org.freedesktop.login1', interface='org.freedesktop.login1.Manager'",
                               NULL);
        }
    }
#endif
934 935
    /* Only now accept clients from network */
    virNetServerUpdateServices(srv, true);
936 937
cleanup:
    daemonInhibitCallback(false, srv);
938
    virObjectUnref(srv);
939 940
    virObjectUnref(sysident);
    virIdentitySetCurrent(NULL);
941
}
942

943 944 945
static int daemonStateInit(virNetServerPtr srv)
{
    virThread thr;
946
    virObjectRef(srv);
947
    if (virThreadCreate(&thr, false, daemonRunStateInit, srv) < 0) {
948
        virObjectUnref(srv);
949 950
        return -1;
    }
951 952 953
    return 0;
}

954 955 956 957 958 959 960 961 962 963 964
static int migrateProfile(void)
{
    char *old_base = NULL;
    char *updated = NULL;
    char *home = NULL;
    char *xdg_dir = NULL;
    char *config_dir = NULL;
    const char *config_home;
    int ret = -1;
    mode_t old_umask;

965 966
    VIR_DEBUG("Checking if user profile needs migrating");

967
    if (!(home = virGetUserDirectory()))
968 969 970 971 972 973 974
        goto cleanup;

    if (virAsprintf(&old_base, "%s/.libvirt", home) < 0) {
        goto cleanup;
    }

    /* if the new directory is there or the old one is not: do nothing */
975
    if (!(config_dir = virGetUserConfigDirectory()))
976 977 978
        goto cleanup;

    if (!virFileIsDir(old_base) || virFileExists(config_dir)) {
979 980 981
        VIR_DEBUG("No old profile in '%s' / "
                  "new profile directory already present '%s'",
                  old_base, config_dir);
982 983 984 985 986 987 988 989 990 991 992 993 994 995
        ret = 0;
        goto cleanup;
    }

    /* test if we already attempted to migrate first */
    if (virAsprintf(&updated, "%s/DEPRECATED-DIRECTORY", old_base) < 0) {
        goto cleanup;
    }
    if (virFileExists(updated)) {
        goto cleanup;
    }

    config_home = getenv("XDG_CONFIG_HOME");
    if (config_home && config_home[0] != '\0') {
996 997
        if (VIR_STRDUP(xdg_dir, config_home) < 0)
            goto cleanup;
998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017
    } else {
        if (virAsprintf(&xdg_dir, "%s/.config", home) < 0) {
            goto cleanup;
        }
    }

    old_umask = umask(077);
    if (virFileMakePath(xdg_dir) < 0) {
        umask(old_umask);
        goto cleanup;
    }
    umask(old_umask);

    if (rename(old_base, config_dir) < 0) {
        int fd = creat(updated, 0600);
        VIR_FORCE_CLOSE(fd);
        VIR_ERROR(_("Unable to migrate %s to %s"), old_base, config_dir);
        goto cleanup;
    }

1018
    VIR_DEBUG("Profile migrated from %s to %s", old_base, config_dir);
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030
    ret = 0;

 cleanup:
    VIR_FREE(home);
    VIR_FREE(old_base);
    VIR_FREE(xdg_dir);
    VIR_FREE(config_dir);
    VIR_FREE(updated);

    return ret;
}

1031 1032
/* Print command-line usage. */
static void
1033
daemonUsage(const char *argv0, bool privileged)
1034
{
1035 1036 1037 1038 1039 1040
    fprintf(stderr,
            _("\n"
              "Usage:\n"
              "  %s [options]\n"
              "\n"
              "Options:\n"
1041
              "  -h | --help            Display program help:\n"
1042 1043 1044 1045 1046
              "  -v | --verbose         Verbose messages.\n"
              "  -d | --daemon          Run as a daemon & write PID file.\n"
              "  -l | --listen          Listen for TCP/IP connections.\n"
              "  -t | --timeout <secs>  Exit after timeout period.\n"
              "  -f | --config <file>   Configuration file.\n"
1047
              "  -V | --version         Display version information.\n"
1048 1049 1050 1051
              "  -p | --pid-file <file> Change name of PID file.\n"
              "\n"
              "libvirt management daemon:\n"),
            argv0);
1052 1053 1054

    if (privileged) {
        fprintf(stderr,
1055 1056 1057 1058
                _("\n"
                  "  Default paths:\n"
                  "\n"
                  "    Configuration file (unless overridden by -f):\n"
1059
                  "      %s\n"
1060 1061
                  "\n"
                  "    Sockets:\n"
1062 1063
                  "      %s\n"
                  "      %s\n"
1064 1065
                  "\n"
                  "    TLS:\n"
1066 1067 1068
                  "      CA certificate:     %s\n"
                  "      Server certificate: %s\n"
                  "      Server private key: %s\n"
1069 1070 1071 1072
                  "\n"
                  "    PID file (unless overridden by -p):\n"
                  "      %s/run/libvirtd.pid\n"
                  "\n"),
1073 1074 1075
                LIBVIRTD_CONFIGURATION_FILE,
                LIBVIRTD_PRIV_UNIX_SOCKET,
                LIBVIRTD_PRIV_UNIX_SOCKET_RO,
1076 1077 1078
                LIBVIRT_CACERT,
                LIBVIRT_SERVERCERT,
                LIBVIRT_SERVERKEY,
1079 1080
                LOCALSTATEDIR);
    } else {
1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
        fprintf(stderr, "%s",
                _("\n"
                  "  Default paths:\n"
                  "\n"
                  "    Configuration file (unless overridden by -f):\n"
                  "      $XDG_CONFIG_HOME/libvirt/libvirtd.conf\n"
                  "\n"
                  "    Sockets:\n"
                  "      $XDG_RUNTIME_DIR/libvirt/libvirt-sock\n"
                  "\n"
                  "    TLS:\n"
                  "      CA certificate:     $HOME/.pki/libvirt/cacert.pem\n"
                  "      Server certificate: $HOME/.pki/libvirt/servercert.pem\n"
                  "      Server private key: $HOME/.pki/libvirt/serverkey.pem\n"
                  "\n"
                  "    PID file:\n"
                  "      $XDG_RUNTIME_DIR/libvirt/libvirtd.pid\n"
                  "\n"));
1099
    }
1100 1101
}

D
Daniel P. Berrange 已提交
1102 1103
#define MAX_LISTEN 5
int main(int argc, char **argv) {
1104 1105
    virNetServerPtr srv = NULL;
    char *remote_config_file = NULL;
1106
    int statuswrite = -1;
1107
    int ret = 1;
1108
    int pid_file_fd = -1;
1109 1110 1111 1112 1113 1114 1115 1116 1117
    char *pid_file = NULL;
    char *sock_file = NULL;
    char *sock_file_ro = NULL;
    int timeout = -1;        /* -t: Shutdown timeout */
    int verbose = 0;
    int godaemon = 0;
    int ipsock = 0;
    struct daemonConfig *config;
    bool privileged = geteuid() == 0 ? true : false;
1118
    bool implicit_conf = false;
1119 1120
    char *run_dir = NULL;
    mode_t old_umask;
D
Daniel P. Berrange 已提交
1121 1122

    struct option opts[] = {
1123 1124 1125
        { "verbose", no_argument, &verbose, 'v'},
        { "daemon", no_argument, &godaemon, 'd'},
        { "listen", no_argument, &ipsock, 'l'},
1126
        { "config", required_argument, NULL, 'f'},
1127 1128
        { "timeout", required_argument, NULL, 't'},
        { "pid-file", required_argument, NULL, 'p'},
1129 1130
        { "version", no_argument, NULL, 'V' },
        { "help", no_argument, NULL, 'h' },
D
Daniel P. Berrange 已提交
1131 1132 1133
        {0, 0, 0, 0}
    };

1134 1135
    if (setlocale(LC_ALL, "") == NULL ||
        bindtextdomain(PACKAGE, LOCALEDIR) == NULL ||
E
Eric Blake 已提交
1136 1137
        textdomain(PACKAGE) == NULL ||
        virInitialize() < 0) {
1138
        fprintf(stderr, _("%s: initialization failed\n"), argv[0]);
E
Eric Blake 已提交
1139
        exit(EXIT_FAILURE);
1140
    }
D
Daniel P. Berrange 已提交
1141

1142 1143
    if (strstr(argv[0], "lt-libvirtd") ||
        strstr(argv[0], "/daemon/.libs/libvirtd")) {
1144 1145 1146 1147 1148 1149 1150
        char *tmp = strrchr(argv[0], '/');
        if (!tmp) {
            fprintf(stderr, _("%s: cannot identify driver directory\n"), argv[0]);
            exit(EXIT_FAILURE);
        }
        *tmp = '\0';
        char *driverdir;
1151
        if (virAsprintfQuiet(&driverdir, "%s/../../src/.libs", argv[0]) < 0) {
1152 1153 1154 1155 1156 1157 1158 1159
            fprintf(stderr, _("%s: initialization failed\n"), argv[0]);
            exit(EXIT_FAILURE);
        }
        if (access(driverdir, R_OK) < 0) {
            fprintf(stderr, _("%s: expected driver directory '%s' is missing\n"),
                    argv[0], driverdir);
            exit(EXIT_FAILURE);
        }
1160 1161
        virLockManagerSetPluginDir(driverdir);
#ifdef WITH_DRIVER_MODULES
1162
        virDriverModuleInitialize(driverdir);
1163
#endif
1164 1165 1166
        *tmp = '/';
        /* Must not free 'driverdir' - it is still used */
    }
1167

D
Daniel P. Berrange 已提交
1168 1169 1170 1171 1172
    while (1) {
        int optidx = 0;
        int c;
        char *tmp;

1173
        c = getopt_long(argc, argv, "ldf:p:t:vVh", opts, &optidx);
D
Daniel P. Berrange 已提交
1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188

        if (c == -1) {
            break;
        }

        switch (c) {
        case 0:
            /* Got one of the flags */
            break;
        case 'v':
            verbose = 1;
            break;
        case 'd':
            godaemon = 1;
            break;
1189 1190
        case 'l':
            ipsock = 1;
D
Daniel P. Berrange 已提交
1191 1192 1193
            break;

        case 't':
1194
            if (virStrToLong_i(optarg, &tmp, 10, &timeout) != 0
1195 1196
                || timeout <= 0
                /* Ensure that we can multiply by 1000 without overflowing.  */
1197 1198 1199 1200
                || timeout > INT_MAX / 1000) {
                VIR_ERROR(_("Invalid value for timeout"));
                exit(EXIT_FAILURE);
            }
D
Daniel P. Berrange 已提交
1201
            break;
1202 1203

        case 'p':
1204
            VIR_FREE(pid_file);
1205
            if (VIR_STRDUP_QUIET(pid_file, optarg) < 0) {
1206
                VIR_ERROR(_("Can't allocate memory"));
1207
                exit(EXIT_FAILURE);
1208
            }
1209 1210 1211
            break;

        case 'f':
1212
            VIR_FREE(remote_config_file);
1213
            if (VIR_STRDUP_QUIET(remote_config_file, optarg) < 0) {
1214
                VIR_ERROR(_("Can't allocate memory"));
1215
                exit(EXIT_FAILURE);
1216
            }
1217 1218
            break;

1219
        case 'V':
1220
            daemonVersion(argv[0]);
1221
            exit(EXIT_SUCCESS);
1222

1223
        case 'h':
1224
            daemonUsage(argv[0], privileged);
1225
            exit(EXIT_SUCCESS);
D
Daniel P. Berrange 已提交
1226

1227
        case '?':
D
Daniel P. Berrange 已提交
1228
        default:
1229
            daemonUsage(argv[0], privileged);
1230
            exit(EXIT_FAILURE);
D
Daniel P. Berrange 已提交
1231 1232 1233
        }
    }

1234 1235 1236 1237 1238 1239
    if (optind != argc) {
        fprintf(stderr, "%s: unexpected, non-option, command line arguments\n",
                argv[0]);
        exit(EXIT_FAILURE);
    }

1240 1241
    if (!(config = daemonConfigNew(privileged))) {
        VIR_ERROR(_("Can't create initial configuration"));
1242
        exit(EXIT_FAILURE);
1243
    }
1244 1245

    /* No explicit config, so try and find a default one */
1246 1247 1248
    if (remote_config_file == NULL) {
        implicit_conf = true;
        if (daemonConfigFilePath(privileged,
1249 1250
                                 &remote_config_file) < 0) {
            VIR_ERROR(_("Can't determine config path"));
1251
            exit(EXIT_FAILURE);
1252
        }
1253
    }
1254 1255 1256

    /* Read the config file if it exists*/
    if (remote_config_file &&
1257
        daemonConfigLoadFile(config, remote_config_file, implicit_conf) < 0) {
1258 1259 1260 1261 1262 1263
        virErrorPtr err = virGetLastError();
        if (err && err->message)
            VIR_ERROR(_("Can't load config file: %s: %s"),
                      err->message, remote_config_file);
        else
            VIR_ERROR(_("Can't load config file: %s"), remote_config_file);
1264
        exit(EXIT_FAILURE);
1265
    }
1266

1267
    if (!privileged &&
1268 1269
        migrateProfile() < 0) {
        VIR_ERROR(_("Exiting due to failure to migrate profile"));
1270
        exit(EXIT_FAILURE);
1271
    }
1272

1273 1274 1275 1276
    if (config->host_uuid &&
        virSetHostUUIDStr(config->host_uuid) < 0) {
        VIR_ERROR(_("invalid host UUID: %s"), config->host_uuid);
        exit(EXIT_FAILURE);
1277 1278
    }

1279 1280
    if (daemonSetupLogging(config, privileged, verbose, godaemon) < 0) {
        VIR_ERROR(_("Can't initialize logging"));
1281
        exit(EXIT_FAILURE);
1282
    }
1283

1284 1285 1286 1287 1288
    if (daemonSetupAccessManager(config) < 0) {
        VIR_ERROR(_("Can't initialize access manager"));
        exit(EXIT_FAILURE);
    }

1289
    if (!pid_file &&
1290
        daemonPidFilePath(privileged,
1291 1292
                          &pid_file) < 0) {
        VIR_ERROR(_("Can't determine pid file path."));
1293
        exit(EXIT_FAILURE);
1294
    }
1295
    VIR_DEBUG("Decided on pid file path '%s'", NULLSTR(pid_file));
1296 1297 1298 1299

    if (daemonUnixSocketPaths(config,
                              privileged,
                              &sock_file,
1300 1301
                              &sock_file_ro) < 0) {
        VIR_ERROR(_("Can't determine socket paths"));
1302
        exit(EXIT_FAILURE);
1303
    }
1304 1305
    VIR_DEBUG("Decided on socket paths '%s' and '%s'",
              sock_file, NULLSTR(sock_file_ro));
1306

1307
    if (godaemon) {
1308
        char ebuf[1024];
1309 1310 1311 1312

        if (chdir("/") < 0) {
            VIR_ERROR(_("cannot change to root directory: %s"),
                      virStrerror(errno, ebuf, sizeof(ebuf)));
1313
            goto cleanup;
1314 1315
        }

1316
        if ((statuswrite = daemonForkIntoBackground(argv[0])) < 0) {
1317
            VIR_ERROR(_("Failed to fork as daemon: %s"),
1318
                      virStrerror(errno, ebuf, sizeof(ebuf)));
1319
            goto cleanup;
1320 1321 1322
        }
    }

J
John Levon 已提交
1323
    /* Ensure the rundir exists (on tmpfs on some systems) */
1324
    if (privileged) {
1325 1326 1327 1328
        if (VIR_STRDUP_QUIET(run_dir, LOCALSTATEDIR "/run/libvirt") < 0) {
            VIR_ERROR(_("Can't allocate memory"));
            goto cleanup;
        }
1329
    } else {
1330
        run_dir = virGetUserRuntimeDirectory();
1331

1332
        if (!run_dir) {
1333 1334
            VIR_ERROR(_("Can't determine user directory"));
            goto cleanup;
J
John Levon 已提交
1335 1336
        }
    }
1337 1338 1339 1340
    if (privileged)
        old_umask = umask(022);
    else
        old_umask = umask(077);
1341
    VIR_DEBUG("Ensuring run dir '%s' exists", run_dir);
1342 1343 1344 1345 1346 1347 1348 1349
    if (virFileMakePath(run_dir) < 0) {
        char ebuf[1024];
        VIR_ERROR(_("unable to create rundir %s: %s"), run_dir,
                  virStrerror(errno, ebuf, sizeof(ebuf)));
        ret = VIR_DAEMON_ERR_RUNDIR;
        goto cleanup;
    }
    umask(old_umask);
J
John Levon 已提交
1350

1351 1352 1353 1354 1355 1356
    /* Try to claim the pidfile, exiting if we can't */
    if ((pid_file_fd = virPidFileAcquirePath(pid_file, getpid())) < 0) {
        ret = VIR_DAEMON_ERR_PIDFILE;
        goto cleanup;
    }

1357 1358 1359 1360 1361
    if (virNetlinkStartup() < 0) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
    }

1362 1363
    if (!(srv = virNetServerNew(config->min_workers,
                                config->max_workers,
1364
                                config->prio_workers,
1365
                                config->max_clients,
1366 1367 1368
                                config->keepalive_interval,
                                config->keepalive_count,
                                !!config->keepalive_required,
1369
                                config->mdns_adv ? config->mdns_name : NULL,
1370
                                remoteClientInitHook,
1371
                                NULL,
1372
                                remoteClientFreeFunc,
1373
                                NULL))) {
1374 1375 1376 1377
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
    }

1378 1379 1380
    /* Beyond this point, nothing should rely on using
     * getuid/geteuid() == 0, for privilege level checks.
     */
1381
    VIR_DEBUG("Dropping privileges (if required)");
1382
    if (daemonSetupPrivs() < 0) {
1383
        ret = VIR_DAEMON_ERR_PRIVS;
1384
        goto cleanup;
1385
    }
J
John Levon 已提交
1386

1387
    daemonInitialize();
1388

1389 1390 1391 1392 1393 1394 1395 1396 1397
    remoteProcs[REMOTE_PROC_AUTH_LIST].needAuth = false;
    remoteProcs[REMOTE_PROC_AUTH_SASL_INIT].needAuth = false;
    remoteProcs[REMOTE_PROC_AUTH_SASL_STEP].needAuth = false;
    remoteProcs[REMOTE_PROC_AUTH_SASL_START].needAuth = false;
    remoteProcs[REMOTE_PROC_AUTH_POLKIT].needAuth = false;
    if (!(remoteProgram = virNetServerProgramNew(REMOTE_PROGRAM,
                                                 REMOTE_PROTOCOL_VERSION,
                                                 remoteProcs,
                                                 remoteNProcs))) {
1398
        ret = VIR_DAEMON_ERR_INIT;
1399 1400 1401 1402 1403
        goto cleanup;
    }
    if (virNetServerAddProgram(srv, remoteProgram) < 0) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
1404
    }
1405

1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417
    if (!(lxcProgram = virNetServerProgramNew(LXC_PROGRAM,
                                              LXC_PROTOCOL_VERSION,
                                              lxcProcs,
                                              lxcNProcs))) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
    }
    if (virNetServerAddProgram(srv, lxcProgram) < 0) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
    }

1418 1419 1420 1421 1422 1423 1424 1425 1426 1427
    if (!(qemuProgram = virNetServerProgramNew(QEMU_PROGRAM,
                                               QEMU_PROTOCOL_VERSION,
                                               qemuProcs,
                                               qemuNProcs))) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
    }
    if (virNetServerAddProgram(srv, qemuProgram) < 0) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
1428
    }
1429

1430 1431
    if (timeout != -1) {
        VIR_DEBUG("Registering shutdown timeout %d", timeout);
1432
        virNetServerAutoShutdown(srv,
1433
                                 timeout);
1434
    }
1435 1436 1437 1438

    if ((daemonSetupSignals(srv)) < 0) {
        ret = VIR_DAEMON_ERR_SIGNAL;
        goto cleanup;
1439
    }
D
Daniel P. Berrange 已提交
1440

1441
    if (config->audit_level) {
1442
        VIR_DEBUG("Attempting to configure auditing subsystem");
1443
        if (virAuditOpen() < 0) {
1444
            if (config->audit_level > 1) {
1445
                ret = VIR_DAEMON_ERR_AUDIT;
1446
                goto cleanup;
1447
            }
1448
            VIR_DEBUG("Proceeding without auditing");
1449 1450
        }
    }
1451
    virAuditLog(config->audit_logging);
1452

1453
    /* setup the hooks if any */
1454
    if (virHookInitialize() < 0) {
1455
        ret = VIR_DAEMON_ERR_HOOKS;
1456
        goto cleanup;
1457 1458
    }

1459
    /* Disable error func, now logging is setup */
1460
    virSetErrorFunc(NULL, daemonErrorHandler);
1461
    virSetErrorLogPriorityFunc(daemonErrorLogFilter);
1462

1463 1464 1465 1466 1467 1468
    /*
     * Call the daemon startup hook
     * TODO: should we abort the daemon startup if the script returned
     *       an error ?
     */
    virHookCall(VIR_HOOK_DRIVER_DAEMON, "-", VIR_HOOK_DAEMON_OP_START,
1469
                0, "start", NULL, NULL);
1470

1471 1472 1473
    if (daemonSetupNetworking(srv, config,
                              sock_file, sock_file_ro,
                              ipsock, privileged) < 0) {
1474
        ret = VIR_DAEMON_ERR_NETWORK;
1475
        goto cleanup;
1476 1477
    }

1478 1479 1480 1481 1482 1483 1484 1485 1486
    /* Tell parent of daemon that basic initialization is complete
     * In particular we're ready to accept net connections & have
     * written the pidfile
     */
    if (statuswrite != -1) {
        char status = 0;
        while (write(statuswrite, &status, 1) == -1 &&
               errno == EINTR)
            ;
1487
        VIR_FORCE_CLOSE(statuswrite);
1488 1489
    }

1490 1491 1492 1493
    /* Initialize drivers & then start accepting new clients from network */
    if (daemonStateInit(srv) < 0) {
        ret = VIR_DAEMON_ERR_INIT;
        goto cleanup;
1494
    }
D
Daniel P. Berrange 已提交
1495

1496
#if defined(__linux__) && defined(NETLINK_ROUTE)
1497 1498
    /* Register the netlink event service for NETLINK_ROUTE */
    if (virNetlinkEventServiceStart(NETLINK_ROUTE, 0) < 0) {
1499 1500 1501
        ret = VIR_DAEMON_ERR_NETWORK;
        goto cleanup;
    }
1502
#endif
1503

1504
#if defined(__linux__) && defined(NETLINK_KOBJECT_UEVENT)
T
Tang Chen 已提交
1505 1506 1507 1508 1509
    /* Register the netlink event service for NETLINK_KOBJECT_UEVENT */
    if (virNetlinkEventServiceStart(NETLINK_KOBJECT_UEVENT, 1) < 0) {
        ret = VIR_DAEMON_ERR_NETWORK;
        goto cleanup;
    }
1510
#endif
T
Tang Chen 已提交
1511

1512 1513
    /* Run event loop. */
    virNetServerRun(srv);
1514

1515 1516
    ret = 0;

1517
    virHookCall(VIR_HOOK_DRIVER_DAEMON, "-", VIR_HOOK_DAEMON_OP_SHUTDOWN,
1518
                0, "shutdown", NULL, NULL);
1519

1520
cleanup:
1521
    virNetlinkEventServiceStopAll();
1522
    virObjectUnref(remoteProgram);
1523
    virObjectUnref(lxcProgram);
1524
    virObjectUnref(qemuProgram);
1525
    virNetServerClose(srv);
1526
    virObjectUnref(srv);
1527
    virNetlinkShutdown();
1528 1529 1530 1531 1532 1533 1534 1535
    if (statuswrite != -1) {
        if (ret != 0) {
            /* Tell parent of daemon what failed */
            char status = ret;
            while (write(statuswrite, &status, 1) == -1 &&
                   errno == EINTR)
                ;
        }
1536
        VIR_FORCE_CLOSE(statuswrite);
1537
    }
1538 1539
    if (pid_file_fd != -1)
        virPidFileReleasePath(pid_file, pid_file_fd);
1540 1541 1542 1543 1544

    VIR_FREE(sock_file);
    VIR_FREE(sock_file_ro);
    VIR_FREE(pid_file);
    VIR_FREE(remote_config_file);
1545 1546
    VIR_FREE(run_dir);

1547 1548
    daemonConfigFree(config);

1549 1550
    virStateCleanup();

1551
    return ret;
D
Daniel P. Berrange 已提交
1552
}