libvirtd.c 98.1 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-2011 Red Hat, Inc.
D
Daniel P. Berrange 已提交
5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23
 * 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
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
 *
 * Author: Daniel P. Berrange <berrange@redhat.com>
 */

24
#include <config.h>
25

D
Daniel P. Berrange 已提交
26 27 28 29 30 31 32 33 34 35
#include <sys/types.h>
#include <sys/wait.h>
#include <sys/stat.h>
#include <unistd.h>
#include <fcntl.h>
#include <limits.h>
#include <sys/socket.h>
#include <sys/un.h>
#include <sys/poll.h>
#include <netinet/in.h>
36
#include <netinet/tcp.h>
D
Daniel P. Berrange 已提交
37 38 39 40
#include <netdb.h>
#include <stdlib.h>
#include <pwd.h>
#include <stdio.h>
41 42
#include <stdarg.h>
#include <syslog.h>
D
Daniel P. Berrange 已提交
43 44 45
#include <string.h>
#include <errno.h>
#include <getopt.h>
46
#include <fnmatch.h>
47
#include <grp.h>
48
#include <signal.h>
J
Jim Meyering 已提交
49
#include <netdb.h>
E
Eric Blake 已提交
50
#include <locale.h>
51

52
#include "libvirt_internal.h"
53
#include "virterror_internal.h"
54
#include "files.h"
55

56 57
#define VIR_FROM_THIS VIR_FROM_QEMU

D
Daniel P. Berrange 已提交
58
#include "libvirtd.h"
59 60
#include "dispatch.h"

61
#include "util.h"
62
#include "uuid.h"
63
#include "remote_driver.h"
64
#include "conf.h"
D
Daniel P. Berrange 已提交
65
#include "event.h"
66
#include "event_poll.h"
67
#include "memory.h"
68
#include "stream.h"
69
#include "hooks.h"
70
#include "virtaudit.h"
71
#ifdef HAVE_AVAHI
72
# include "mdns.h"
73
#endif
D
Daniel P. Berrange 已提交
74

75
#ifdef WITH_DRIVER_MODULES
76
# include "driver.h"
77
#else
78 79 80 81 82 83
# ifdef WITH_QEMU
#  include "qemu/qemu_driver.h"
# endif
# ifdef WITH_LXC
#  include "lxc/lxc_driver.h"
# endif
J
Jim Fehlig 已提交
84 85 86
# ifdef WITH_LIBXL
#  include "libxl/libxl_driver.h"
# endif
87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104
# ifdef WITH_UML
#  include "uml/uml_driver.h"
# endif
# ifdef WITH_NETWORK
#  include "network/bridge_driver.h"
# endif
# ifdef WITH_NETCF
#  include "interface/netcf_driver.h"
# endif
# ifdef WITH_STORAGE_DIR
#  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
105 106 107
# ifdef WITH_NWFILTER
#  include "nwfilter/nwfilter_driver.h"
# endif
108
#endif
109 110


J
John Levon 已提交
111
#ifdef __sun
112 113
# include <ucred.h>
# include <priv.h>
J
John Levon 已提交
114

115 116 117
# ifndef PRIV_VIRT_MANAGE
#  define PRIV_VIRT_MANAGE ((const char *)"virt_manage")
# endif
J
John Levon 已提交
118

119 120 121
# ifndef PRIV_XVM_CONTROL
#  define PRIV_XVM_CONTROL ((const char *)"xvm_control")
# endif
J
John Levon 已提交
122

123 124
# define PU_RESETGROUPS          0x0001  /* Remove supplemental groups */
# define PU_CLEARLIMITSET        0x0008  /* L=0 */
J
John Levon 已提交
125 126 127

extern int __init_daemon_priv(int, uid_t, gid_t, ...);

128
# define SYSTEM_UID 60
J
John Levon 已提交
129 130 131 132 133 134 135 136 137 138 139 140 141

static gid_t unix_sock_gid = 60; /* Not used */
static int unix_sock_rw_mask = 0666;
static int unix_sock_ro_mask = 0666;

#else

static gid_t unix_sock_gid = 0; /* Only root by default */
static int unix_sock_rw_mask = 0700; /* Allow user only */
static int unix_sock_ro_mask = 0777; /* Allow world */

#endif /* __sun */

142 143
#include "configmake.h"

144 145
static int godaemon = 0;        /* -d: Be a daemon */
static int verbose = 0;         /* -v: Verbose mode */
146
static int timeout = -1;        /* -t: Shutdown timeout */
147
static int sigwrite = -1;       /* Signal handler pipe */
148
static int ipsock = 0;          /* -l  Listen for TCP/IP */
149

150
/* Defaults for configuration file elements */
151 152
static int listen_tls = 1;
static int listen_tcp = 0;
153
static char *listen_addr  = (char *) LIBVIRTD_LISTEN_ADDR;
154 155
static char *tls_port = (char *) LIBVIRTD_TLS_PORT;
static char *tcp_port = (char *) LIBVIRTD_TCP_PORT;
156

157 158
static char *unix_sock_dir = NULL;

159 160 161 162
#if HAVE_POLKIT
static int auth_unix_rw = REMOTE_AUTH_POLKIT;
static int auth_unix_ro = REMOTE_AUTH_POLKIT;
#else
163 164
static int auth_unix_rw = REMOTE_AUTH_NONE;
static int auth_unix_ro = REMOTE_AUTH_NONE;
165
#endif /* HAVE_POLKIT */
166 167 168 169 170 171 172
#if HAVE_SASL
static int auth_tcp = REMOTE_AUTH_SASL;
#else
static int auth_tcp = REMOTE_AUTH_NONE;
#endif
static int auth_tls = REMOTE_AUTH_NONE;

173
static int mdns_adv = 1;
174
static char *mdns_name = NULL;
175

176
static int tls_no_verify_certificate = 0;
177
static char **tls_allowed_dn_list = NULL;
178

179 180 181 182
static char *key_file = (char *) LIBVIRT_SERVERKEY;
static char *cert_file = (char *) LIBVIRT_SERVERCERT;
static char *ca_file = (char *) LIBVIRT_CACERT;
static char *crl_file = (char *) "";
183 184 185 186

static gnutls_certificate_credentials_t x509_cred;
static gnutls_dh_params_t dh_params;

187 188 189 190
static int min_workers = 5;
static int max_workers = 20;
static int max_clients = 20;

191 192 193 194 195
/* Total number of 'in-process' RPC calls allowed across all clients */
static int max_requests = 20;
/* Total number of 'in-process' RPC calls allowed by a single client*/
static int max_client_requests = 5;

196 197 198
static int audit_level = 1;
static int audit_logging = 0;

199
#define DH_BITS 1024
200

201 202
static sig_atomic_t sig_errors = 0;
static int sig_lasterrno = 0;
203
static const char *argv0;
204

205 206 207 208 209 210 211 212 213
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,
214
    VIR_DAEMON_ERR_HOOKS,
215
    VIR_DAEMON_ERR_AUDIT,
216 217 218 219 220 221 222 223 224 225 226 227 228

    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",
229
              "Unable to load configuration file",
230 231
              "Unable to look for hook scripts",
              "Unable to initialize audit system")
232

233 234
static void sig_handler(int sig, siginfo_t * siginfo,
                        void* context ATTRIBUTE_UNUSED) {
235
    int origerrno;
236
    int r;
237

238 239
    /* set the sig num in the struct */
    siginfo->si_signo = sig;
240 241

    origerrno = errno;
242
    r = safewrite(sigwrite, siginfo, sizeof(*siginfo));
243 244 245 246
    if (r == -1) {
        sig_errors++;
        sig_lasterrno = errno;
    }
247
    errno = origerrno;
D
Daniel P. Berrange 已提交
248
}
249

250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269
static void sig_fatal(int sig, siginfo_t * siginfo ATTRIBUTE_UNUSED,
                      void* context ATTRIBUTE_UNUSED) {
    struct sigaction sig_action;
    int origerrno;

    origerrno = errno;
    virLogEmergencyDumpAll(sig);

    /*
     * If the signal is fatal, avoid looping over this handler
     * by desactivating it
     */
    if (sig != SIGUSR2) {
        sig_action.sa_flags = SA_SIGINFO;
        sig_action.sa_handler = SIG_IGN;
        sigaction(sig, &sig_action, NULL);
    }
    errno = origerrno;
}

270 271
static void qemudDispatchClientEvent(int watch, int fd, int events, void *opaque);
static void qemudDispatchServerEvent(int watch, int fd, int events, void *opaque);
272
static int qemudStartWorker(struct qemud_server *server, struct qemud_worker *worker);
273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288

void
qemudClientMessageQueuePush(struct qemud_client_message **queue,
                            struct qemud_client_message *msg)
{
    struct qemud_client_message *tmp = *queue;

    if (tmp) {
        while (tmp->next)
            tmp = tmp->next;
        tmp->next = msg;
    } else {
        *queue = msg;
    }
}

289
struct qemud_client_message *
290 291 292 293 294 295 296 297 298 299 300
qemudClientMessageQueueServe(struct qemud_client_message **queue)
{
    struct qemud_client_message *tmp = *queue;

    if (tmp) {
        *queue = tmp->next;
        tmp->next = NULL;
    }

    return tmp;
}
D
Daniel P. Berrange 已提交
301

302 303 304 305 306
static int
remoteCheckCertFile(const char *type, const char *file)
{
    struct stat sb;
    if (stat(file, &sb) < 0) {
307 308 309
        char ebuf[1024];
        VIR_ERROR(_("Cannot access %s '%s': %s"),
                  type, file, virStrerror(errno, ebuf, sizeof ebuf));
310 311 312 313 314
        return -1;
    }
    return 0;
}

315 316 317 318 319 320 321 322 323 324
static int
remoteInitializeGnuTLS (void)
{
    int err;

    /* Initialise GnuTLS. */
    gnutls_global_init ();

    err = gnutls_certificate_allocate_credentials (&x509_cred);
    if (err) {
325
        VIR_ERROR(_("gnutls_certificate_allocate_credentials: %s"),
326 327 328 329 330
                  gnutls_strerror (err));
        return -1;
    }

    if (ca_file && ca_file[0] != '\0') {
331 332 333
        if (remoteCheckCertFile("CA certificate", ca_file) < 0)
            return -1;

334
        VIR_DEBUG("loading CA cert from %s", ca_file);
335 336 337
        err = gnutls_certificate_set_x509_trust_file (x509_cred, ca_file,
                                                      GNUTLS_X509_FMT_PEM);
        if (err < 0) {
338
            VIR_ERROR(_("gnutls_certificate_set_x509_trust_file: %s"),
339 340 341 342 343 344
                      gnutls_strerror (err));
            return -1;
        }
    }

    if (crl_file && crl_file[0] != '\0') {
J
Jim Meyering 已提交
345
        if (remoteCheckCertFile("CA revocation list", crl_file) < 0)
346 347
            return -1;

348
        VIR_DEBUG("loading CRL from %s", crl_file);
349 350 351
        err = gnutls_certificate_set_x509_crl_file (x509_cred, crl_file,
                                                    GNUTLS_X509_FMT_PEM);
        if (err < 0) {
352
            VIR_ERROR(_("gnutls_certificate_set_x509_crl_file: %s"),
353 354 355 356 357 358
                      gnutls_strerror (err));
            return -1;
        }
    }

    if (cert_file && cert_file[0] != '\0' && key_file && key_file[0] != '\0') {
359 360 361 362
        if (remoteCheckCertFile("server certificate", cert_file) < 0)
            return -1;
        if (remoteCheckCertFile("server key", key_file) < 0)
            return -1;
363
        VIR_DEBUG("loading cert and key from %s and %s", cert_file, key_file);
364 365 366 367 368
        err =
            gnutls_certificate_set_x509_key_file (x509_cred,
                                                  cert_file, key_file,
                                                  GNUTLS_X509_FMT_PEM);
        if (err < 0) {
369
            VIR_ERROR(_("gnutls_certificate_set_x509_key_file: %s"),
370 371 372 373 374 375 376 377 378 379 380 381
                      gnutls_strerror (err));
            return -1;
        }
    }

    /* Generate Diffie Hellman parameters - for use with DHE
     * kx algorithms. These should be discarded and regenerated
     * once a day, once a week or once a month. Depending on the
     * security requirements.
     */
    err = gnutls_dh_params_init (&dh_params);
    if (err < 0) {
382
        VIR_ERROR(_("gnutls_dh_params_init: %s"), gnutls_strerror (err));
383 384 385 386
        return -1;
    }
    err = gnutls_dh_params_generate2 (dh_params, DH_BITS);
    if (err < 0) {
387
        VIR_ERROR(_("gnutls_dh_params_generate2: %s"), gnutls_strerror (err));
388 389 390 391 392 393 394 395
        return -1;
    }

    gnutls_certificate_set_dh_params (x509_cred, dh_params);

    return 0;
}

396
static void
397 398
qemudDispatchSignalEvent(int watch ATTRIBUTE_UNUSED,
                         int fd ATTRIBUTE_UNUSED,
399 400
                         int events ATTRIBUTE_UNUSED,
                         void *opaque) {
D
Daniel P. Berrange 已提交
401
    struct qemud_server *server = (struct qemud_server *)opaque;
402
    siginfo_t siginfo;
403

404
    virMutexLock(&server->lock);
405

406
    if (saferead(server->sigread, &siginfo, sizeof(siginfo)) != sizeof(siginfo)) {
407 408 409
        char ebuf[1024];
        VIR_ERROR(_("Failed to read from signal pipe: %s"),
                  virStrerror(errno, ebuf, sizeof ebuf));
410
        virMutexUnlock(&server->lock);
D
Daniel P. Berrange 已提交
411
        return;
412 413
    }

414
    switch (siginfo.si_signo) {
415
    case SIGHUP:
416
        VIR_INFO0(_("Reloading configuration on SIGHUP"));
417 418
        virHookCall(VIR_HOOK_DRIVER_DAEMON, "-",
                    VIR_HOOK_DAEMON_OP_RELOAD, SIGHUP, "SIGHUP", NULL);
419
        if (virStateReload() < 0)
420
            VIR_WARN0("Error while reloading drivers");
421

422 423 424
        break;

    case SIGINT:
425
    case SIGQUIT:
426
    case SIGTERM:
427
        VIR_WARN("Shutting down on signal %d", siginfo.si_signo);
428
        server->quitEventThread = 1;
429 430 431
        break;

    default:
432
        VIR_INFO(_("Received unexpected signal %d"), siginfo.si_signo);
433 434 435
        break;
    }

436
    virMutexUnlock(&server->lock);
437 438
}

439

440 441 442 443 444
static int daemonForkIntoBackground(void) {
    int statuspipe[2];
    if (pipe(statuspipe) < 0)
        return -1;

D
Daniel P. Berrange 已提交
445 446 447 448 449 450
    int pid = fork();
    switch (pid) {
    case 0:
        {
            int stdinfd = -1;
            int stdoutfd = -1;
451
            int nextpid;
D
Daniel P. Berrange 已提交
452

453
            VIR_FORCE_CLOSE(statuspipe[0]);
454

455
            if ((stdinfd = open("/dev/null", O_RDONLY)) < 0)
D
Daniel P. Berrange 已提交
456
                goto cleanup;
457
            if ((stdoutfd = open("/dev/null", O_WRONLY)) < 0)
D
Daniel P. Berrange 已提交
458 459 460 461 462 463 464
                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;
465
            if (VIR_CLOSE(stdinfd) < 0)
D
Daniel P. Berrange 已提交
466
                goto cleanup;
467
            if (VIR_CLOSE(stdoutfd) < 0)
D
Daniel P. Berrange 已提交
468 469 470 471 472 473 474 475
                goto cleanup;

            if (setsid() < 0)
                goto cleanup;

            nextpid = fork();
            switch (nextpid) {
            case 0:
476
                return statuspipe[1];
D
Daniel P. Berrange 已提交
477 478 479
            case -1:
                return -1;
            default:
480
                _exit(0);
D
Daniel P. Berrange 已提交
481 482 483
            }

        cleanup:
484 485
            VIR_FORCE_CLOSE(stdoutfd);
            VIR_FORCE_CLOSE(stdinfd);
D
Daniel P. Berrange 已提交
486 487 488 489 490 491 492 493 494
            return -1;

        }

    case -1:
        return -1;

    default:
        {
495 496 497 498
            int got, exitstatus = 0;
            int ret;
            char status;

499
            VIR_FORCE_CLOSE(statuspipe[1]);
500 501 502

            /* We wait to make sure the first child forked successfully */
            if ((got = waitpid(pid, &exitstatus, 0)) < 0 ||
D
Daniel P. Berrange 已提交
503
                got != pid ||
504
                exitstatus != 0) {
D
Daniel P. Berrange 已提交
505 506
                return -1;
            }
507 508 509 510 511 512 513 514

            /* Now block until the second child initializes successfully */
        again:
            ret = read(statuspipe[0], &status, 1);
            if (ret == -1 && errno == EINTR)
                goto again;

            if (ret == 1 && status != 0) {
515
                fprintf(stderr,
516 517
                        _("%s: error: %s. Check /var/log/messages or run without "
                          "--daemon for more info.\n"), argv0,
518
                        virDaemonErrTypeToString(status));
519 520
            }
            _exit(ret == 1 && status == 0 ? 0 : 1);
D
Daniel P. Berrange 已提交
521 522 523 524
        }
    }
}

525 526 527
static int qemudWritePidFile(const char *pidFile) {
    int fd;
    FILE *fh;
528
    char ebuf[1024];
529 530 531 532 533

    if (pidFile[0] == '\0')
        return 0;

    if ((fd = open(pidFile, O_WRONLY|O_CREAT|O_EXCL, 0644)) < 0) {
534
        VIR_ERROR(_("Failed to open pid file '%s' : %s"),
535
                  pidFile, virStrerror(errno, ebuf, sizeof ebuf));
536 537 538
        return -1;
    }

539
    if (!(fh = VIR_FDOPEN(fd, "w"))) {
540
        VIR_ERROR(_("Failed to fdopen pid file '%s' : %s"),
541
                  pidFile, virStrerror(errno, ebuf, sizeof ebuf));
542
        VIR_FORCE_CLOSE(fd);
543 544 545 546
        return -1;
    }

    if (fprintf(fh, "%lu\n", (unsigned long)getpid()) < 0) {
547 548
        VIR_ERROR(_("%s: Failed to write to pid file '%s' : %s"),
                  argv0, pidFile, virStrerror(errno, ebuf, sizeof ebuf));
549
        VIR_FORCE_FCLOSE(fh);
550 551 552
        return -1;
    }

553
    if (VIR_FCLOSE(fh) == EOF) {
554 555
        VIR_ERROR(_("%s: Failed to close pid file '%s' : %s"),
                  argv0, pidFile, virStrerror(errno, ebuf, sizeof ebuf));
556 557 558 559 560 561
        return -1;
    }

    return 0;
}

D
Daniel P. Berrange 已提交
562
static int qemudListenUnix(struct qemud_server *server,
563
                           char *path, int readonly, int auth) {
564
    struct qemud_socket *sock;
D
Daniel P. Berrange 已提交
565
    mode_t oldmask;
566
    char ebuf[1024];
D
Daniel P. Berrange 已提交
567

568
    if (VIR_ALLOC(sock) < 0) {
569
        VIR_ERROR0(_("Failed to allocate memory for struct qemud_socket"));
D
Daniel P. Berrange 已提交
570
        return -1;
571
    }
D
Daniel P. Berrange 已提交
572 573

    sock->readonly = readonly;
574
    sock->type = QEMUD_SOCK_TYPE_UNIX;
575
    sock->auth = auth;
576
    sock->path = path;
577 578 579 580 581 582
    sock->addr.len = sizeof(sock->addr.data.un);
    if (!(sock->addrstr = strdup(path))) {
        VIR_ERROR(_("Failed to copy socket address: %s"),
                  virStrerror(errno, ebuf, sizeof ebuf));
        goto cleanup;
    }
D
Daniel P. Berrange 已提交
583

584
    if ((sock->fd = socket(PF_UNIX, SOCK_STREAM, 0)) < 0) {
585
        VIR_ERROR(_("Failed to create socket: %s"),
586
                  virStrerror(errno, ebuf, sizeof ebuf));
D
Daniel P. Berrange 已提交
587
        goto cleanup;
588
    }
D
Daniel P. Berrange 已提交
589

590 591
    if (virSetCloseExec(sock->fd) < 0 ||
        virSetNonBlock(sock->fd) < 0)
D
Daniel P. Berrange 已提交
592
        goto cleanup;
D
Daniel P. Berrange 已提交
593

594 595
    sock->addr.data.un.sun_family = AF_UNIX;
    if (virStrcpyStatic(sock->addr.data.un.sun_path, path) == NULL) {
C
Chris Lalancette 已提交
596 597 598
        VIR_ERROR(_("Path %s too long for unix socket"), path);
        goto cleanup;
    }
599 600
    if (sock->addr.data.un.sun_path[0] == '@')
        sock->addr.data.un.sun_path[0] = '\0';
D
Daniel P. Berrange 已提交
601

602
    oldmask = umask(readonly ? ~unix_sock_ro_mask : ~unix_sock_rw_mask);
603
    if (bind(sock->fd, &sock->addr.data.sa, sock->addr.len) < 0) {
604
        VIR_ERROR(_("Failed to bind socket to '%s': %s"),
605
                  path, virStrerror(errno, ebuf, sizeof ebuf));
606
        umask(oldmask);
D
Daniel P. Berrange 已提交
607
        goto cleanup;
608
    }
D
Daniel P. Berrange 已提交
609
    umask(oldmask);
610 611 612 613 614 615 616 617

    /* chown() doesn't work for abstract sockets but we use them only
     * if libvirtd runs unprivileged
     */
    if (server->privileged && chown(path, -1, unix_sock_gid)) {
        VIR_ERROR(_("Failed to change group ID of '%s' to %d: %s"),
                  path, unix_sock_gid,
                  virStrerror(errno, ebuf, sizeof ebuf));
618 619
        goto cleanup;
    }
D
Daniel P. Berrange 已提交
620

621
    if (listen(sock->fd, 30) < 0) {
622
        VIR_ERROR(_("Failed to listen for connections on '%s': %s"),
623
                  path, virStrerror(errno, ebuf, sizeof ebuf));
D
Daniel P. Berrange 已提交
624 625 626 627 628 629 630
        goto cleanup;
    }

    sock->next = server->sockets;
    server->sockets = sock;
    server->nsockets++;

D
Daniel P. Berrange 已提交
631
    return 0;
D
Daniel P. Berrange 已提交
632 633

 cleanup:
634
    VIR_FORCE_CLOSE(sock->fd);
635
    VIR_FREE(sock);
D
Daniel P. Berrange 已提交
636
    return -1;
D
Daniel P. Berrange 已提交
637 638
}

639
/* See: http://people.redhat.com/drepper/userapi-ipv6.html */
640
static int
641
remoteMakeSockets (int *fds, int max_fds, int *nfds_r, const char *node, const char *service)
642 643 644 645 646 647 648
{
    struct addrinfo *ai;
    struct addrinfo hints;
    memset (&hints, 0, sizeof hints);
    hints.ai_flags = AI_PASSIVE | AI_ADDRCONFIG;
    hints.ai_socktype = SOCK_STREAM;

649
    int e = getaddrinfo (node, service, &hints, &ai);
650
    if (e != 0) {
651
        VIR_ERROR(_("getaddrinfo: %s"), gai_strerror (e));
652 653 654 655 656
        return -1;
    }

    struct addrinfo *runp = ai;
    while (runp && *nfds_r < max_fds) {
657
        char ebuf[1024];
658 659 660
        fds[*nfds_r] = socket (runp->ai_family, runp->ai_socktype,
                               runp->ai_protocol);
        if (fds[*nfds_r] == -1) {
661
            VIR_ERROR(_("socket: %s"), virStrerror (errno, ebuf, sizeof ebuf));
662 663 664 665 666 667
            return -1;
        }

        int opt = 1;
        setsockopt (fds[*nfds_r], SOL_SOCKET, SO_REUSEADDR, &opt, sizeof opt);

668 669 670 671 672 673 674 675 676 677 678 679 680 681 682
#ifdef IPV6_V6ONLY
        if (runp->ai_family == PF_INET6) {
            int on = 1;
            /*
             * Normally on Linux an INET6 socket will bind to the INET4
             * address too. If getaddrinfo returns results with INET4
             * first though, this will result in INET6 binding failing.
             * We can trivially cope with multiple server sockets, so
             * we force it to only listen on IPv6
             */
            setsockopt(fds[*nfds_r], IPPROTO_IPV6,IPV6_V6ONLY,
                       (void*)&on, sizeof on);
        }
#endif

683 684
        if (bind (fds[*nfds_r], runp->ai_addr, runp->ai_addrlen) == -1) {
            if (errno != EADDRINUSE) {
685
                VIR_ERROR(_("bind: %s"), virStrerror (errno, ebuf, sizeof ebuf));
686 687
                return -1;
            }
688
            VIR_FORCE_CLOSE(fds[*nfds_r]);
689
        } else {
690 691 692 693
            ++*nfds_r;
        }
        runp = runp->ai_next;
    }
694

695 696 697 698 699 700 701 702 703
    freeaddrinfo (ai);
    return 0;
}

/* Listen on the named/numbered TCP port.  On a machine with IPv4 and
 * IPv6 interfaces this may generate several sockets.
 */
static int
remoteListenTCP (struct qemud_server *server,
704
                 const char *addr,
705
                 const char *port,
706
                 int type,
707
                 int auth)
708 709 710
{
    int fds[2];
    int nfds = 0;
711
    int i;
712 713
    struct qemud_socket *sock;

714
    if (remoteMakeSockets (fds, 2, &nfds, addr, port) == -1)
715 716 717
        return -1;

    for (i = 0; i < nfds; ++i) {
718
        char ebuf[1024];
719

720
        if (VIR_ALLOC(sock) < 0) {
721 722
            VIR_ERROR(_("remoteListenTCP: calloc: %s"),
                      virStrerror (errno, ebuf, sizeof ebuf));
723
            goto cleanup;
724 725
        }

726
        sock->addr.len = sizeof(sock->addr.data.stor);
727 728 729 730 731 732
        sock->readonly = 0;
        sock->next = server->sockets;
        server->sockets = sock;
        server->nsockets++;

        sock->fd = fds[i];
733
        sock->type = type;
734
        sock->auth = auth;
D
Daniel P. Berrange 已提交
735

736
        if (getsockname(sock->fd, &sock->addr.data.sa, &sock->addr.len) < 0)
737
            goto cleanup;
738

739 740
        if (!(sock->addrstr = virSocketFormatAddrFull(&sock->addr, true, ";")))
            goto cleanup;
741

742 743
        if (virSetCloseExec(sock->fd) < 0 ||
            virSetNonBlock(sock->fd) < 0)
744
            goto cleanup;
745 746

        if (listen (sock->fd, 30) < 0) {
747 748
            VIR_ERROR(_("remoteListenTCP: listen: %s"),
                      virStrerror (errno, ebuf, sizeof ebuf));
749
            goto cleanup;
750
        }
751 752 753
    }

    return 0;
754 755 756

cleanup:
    for (i = 0; i < nfds; ++i)
757
        VIR_FORCE_CLOSE(fds[i]);
758
    return -1;
759 760 761
}

static int qemudInitPaths(struct qemud_server *server,
762 763
                          char **sockname,
                          char **roSockname)
J
John Levon 已提交
764
{
765
    char *base_dir_prefix = NULL;
766
    char *sock_dir_prefix = NULL;
767 768 769 770 771
    int ret = -1;

    /* The base_dir_prefix is the base under which all libvirtd
     * files live */
    if (server->privileged) {
772
        if (!(base_dir_prefix = strdup (LOCALSTATEDIR)))
773 774 775 776 777 778
            goto no_memory;
    } else {
        uid_t uid = geteuid();
        if (!(base_dir_prefix = virGetUserDirectory(uid)))
            goto cleanup;
    }
779

780 781
    /* The unix_sock_dir is the location under which all
     * unix domain sockets live */
D
Daniel P. Berrange 已提交
782
    if (unix_sock_dir) {
783 784 785
        if (!(sock_dir_prefix = strdup(unix_sock_dir)))
            goto no_memory;

D
Daniel P. Berrange 已提交
786 787 788 789 790 791 792
        /* Change the group ownership of /var/run/libvirt to unix_sock_gid */
        if (server->privileged) {
            if (chown(unix_sock_dir, -1, unix_sock_gid) < 0)
                VIR_ERROR(_("Failed to change group ownership of %s"),
                          unix_sock_dir);
        }
    } else {
793
        if (server->privileged) {
794 795 796
            if (virAsprintf(&sock_dir_prefix, "%s/run/libvirt",
                            base_dir_prefix) < 0)
                goto no_memory;
797
        } else {
798 799 800
            if (virAsprintf(&sock_dir_prefix, "%s/.libvirt",
                            base_dir_prefix) < 0)
                goto no_memory;
801 802
        }
    }
803

804
    if (server->privileged) {
805 806 807 808 809 810 811 812
        if (virAsprintf(sockname, "%s/libvirt-sock",
                        sock_dir_prefix) < 0)
            goto no_memory;
        if (virAsprintf(roSockname, "%s/libvirt-sock-ro",
                        sock_dir_prefix) < 0)
            goto no_memory;
        unlink(*sockname);
        unlink(*roSockname);
D
Daniel P. Berrange 已提交
813
    } else {
814 815 816 817 818 819
        if (virAsprintf(sockname, "@%s/libvirt-sock",
                        sock_dir_prefix) < 0)
            goto no_memory;
        /* There is no RO socket in unprivileged mode,
         * since the user always has full RW access
         * to their private instance */
820
    }
821

822
    if (server->privileged) {
823 824 825
        if (virAsprintf(&server->logDir, "%s/log/libvirt",
                        base_dir_prefix) < 0)
            goto no_memory;
826
    } else {
827 828 829
        if (virAsprintf(&server->logDir, "%s/.libvirt/log",
                        base_dir_prefix) < 0)
            goto no_memory;
830
    }
831

832
    ret = 0;
833

834 835 836
no_memory:
    if (ret != 0)
        virReportOOMError();
837 838

 cleanup:
839
    VIR_FREE(base_dir_prefix);
840
    VIR_FREE(sock_dir_prefix);
841
    return ret;
D
Daniel P. Berrange 已提交
842 843
}

844 845 846 847 848 849
static void virshErrorHandler(void *opaque ATTRIBUTE_UNUSED, virErrorPtr err ATTRIBUTE_UNUSED)
{
    /* Don't do anything, since logging infrastructure already
     * took care of reporting the error */
}

850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873
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:
        return VIR_LOG_DEBUG;
    }

    return priority;
}


874
static struct qemud_server *qemudInitialize(void) {
D
Daniel P. Berrange 已提交
875 876
    struct qemud_server *server;

877
    if (VIR_ALLOC(server) < 0) {
878
        VIR_ERROR0(_("Failed to allocate struct qemud_server"));
D
Daniel P. Berrange 已提交
879
        return NULL;
880
    }
D
Daniel P. Berrange 已提交
881

882 883 884
    server->privileged = geteuid() == 0 ? 1 : 0;
    server->sigread = server->sigwrite = -1;

885
    if (virMutexInit(&server->lock) < 0) {
886
        VIR_ERROR0(_("cannot initialize mutex"));
887
        VIR_FREE(server);
888
        return NULL;
889 890
    }
    if (virCondInit(&server->job) < 0) {
891
        VIR_ERROR0(_("cannot initialize condition variable"));
892
        virMutexDestroy(&server->lock);
893
        VIR_FREE(server);
894
        return NULL;
895 896
    }

897
    if (virEventRegisterDefaultImpl() < 0) {
898 899 900
        virMutexDestroy(&server->lock);
        if (virCondDestroy(&server->job) < 0)
        {}
901 902 903 904
        VIR_FREE(server);
        return NULL;
    }

905 906 907 908 909 910 911
    /*
     * 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.
     */
912 913 914
#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 已提交
915
     * If they try to open a connection for a module that
916 917 918 919
     * is not loaded they'll get a suitable error at that point
     */
    virDriverLoadModule("network");
    virDriverLoadModule("storage");
920
    virDriverLoadModule("nodedev");
921
    virDriverLoadModule("secret");
922 923 924
    virDriverLoadModule("qemu");
    virDriverLoadModule("lxc");
    virDriverLoadModule("uml");
925
    virDriverLoadModule("nwfilter");
926
#else
927
# ifdef WITH_NETWORK
928
    networkRegister();
929 930
# endif
# ifdef WITH_NETCF
931
    interfaceRegister();
932 933
# endif
# ifdef WITH_STORAGE_DIR
934
    storageRegister();
935 936
# endif
# if defined(WITH_NODE_DEVICES)
937
    nodedevRegister();
938 939
# endif
# ifdef WITH_SECRETS
940
    secretRegister();
941
# endif
942 943 944
# ifdef WITH_NWFILTER
    nwfilterRegister();
# endif
J
Jim Fehlig 已提交
945 946 947
# ifdef WITH_LIBXL
    libxlRegister();
# endif
948
# ifdef WITH_QEMU
949
    qemuRegister();
950 951
# endif
# ifdef WITH_LXC
952
    lxcRegister();
953 954
# endif
# ifdef WITH_UML
955
    umlRegister();
956
# endif
957 958
#endif

959 960 961
    return server;
}

962
static int qemudNetworkInit(struct qemud_server *server) {
963 964
    char *sockname = NULL;
    char *roSockname = NULL;
965
#if HAVE_SASL
966 967 968
    int err;
#endif /* HAVE_SASL */

969
    if (qemudInitPaths(server, &sockname, &roSockname) < 0)
970 971 972
        goto cleanup;

    if (qemudListenUnix(server, sockname, 0, auth_unix_rw) < 0)
973
        goto cleanup;
974
    sockname = NULL;
975

976
    if (roSockname != NULL && qemudListenUnix(server, roSockname, 1, auth_unix_ro) < 0)
977
        goto cleanup;
978
    roSockname = NULL;
979 980 981 982 983 984 985

#if HAVE_SASL
    if (auth_unix_rw == REMOTE_AUTH_SASL ||
        auth_unix_ro == REMOTE_AUTH_SASL ||
        auth_tcp == REMOTE_AUTH_SASL ||
        auth_tls == REMOTE_AUTH_SASL) {
        if ((err = sasl_server_init(NULL, "libvirt")) != SASL_OK) {
986 987
            VIR_ERROR(_("Failed to initialize SASL authentication %s"),
                      sasl_errstring(err, NULL, NULL));
988 989
            goto cleanup;
        }
990 991 992
    }
#endif

993
#if HAVE_POLKIT0
994 995 996
    if (auth_unix_rw == REMOTE_AUTH_POLKIT ||
        auth_unix_ro == REMOTE_AUTH_POLKIT) {
        DBusError derr;
997 998 999 1000

        dbus_connection_set_change_sigpipe(FALSE);
        dbus_threads_init_default();

1001 1002 1003
        dbus_error_init(&derr);
        server->sysbus = dbus_bus_get(DBUS_BUS_SYSTEM, &derr);
        if (!(server->sysbus)) {
1004 1005
            VIR_ERROR(_("Failed to connect to system bus for PolicyKit auth: %s"),
                      derr.message);
1006 1007 1008
            dbus_error_free(&derr);
            goto cleanup;
        }
1009
        dbus_connection_set_exit_on_disconnect(server->sysbus, FALSE);
1010 1011 1012
    }
#endif

1013
    if (ipsock) {
1014
        if (listen_tcp && remoteListenTCP (server, listen_addr, tcp_port, QEMUD_SOCK_TYPE_TCP, auth_tcp) < 0)
1015 1016 1017 1018 1019 1020
            goto cleanup;

        if (listen_tls) {
            if (remoteInitializeGnuTLS () < 0)
                goto cleanup;

1021
            if (remoteListenTCP (server, listen_addr, tls_port, QEMUD_SOCK_TYPE_TLS, auth_tls) < 0)
1022 1023
                goto cleanup;
        }
D
Daniel P. Berrange 已提交
1024 1025
    }

1026
#ifdef HAVE_AVAHI
1027
    if (server->privileged && mdns_adv) {
1028
        struct libvirtd_mdns_group *group;
1029
        struct qemud_socket *sock;
1030
        int port = 0;
1031
        int ret;
1032 1033 1034 1035

        server->mdns = libvirtd_mdns_new();

        if (!mdns_name) {
1036 1037
            char *groupname, *localhost, *tmp;

1038 1039
            localhost = virGetHostname(NULL);
            if (localhost == NULL)
1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054
                /* we couldn't resolve the hostname; assume that we are
                 * running in disconnected operation, and report a less
                 * useful Avahi string
                 */
                ret = virAsprintf(&groupname, "Virtualization Host");
            else {
                /* Extract the host part of the potentially FQDN */
                if ((tmp = strchr(localhost, '.')))
                    *tmp = '\0';
                ret = virAsprintf(&groupname, "Virtualization Host %s",
                                  localhost);
            }
            VIR_FREE(localhost);
            if (ret < 0) {
                virReportOOMError();
1055
                goto cleanup;
1056
            }
1057
            group = libvirtd_mdns_add_group(server->mdns, groupname);
1058
            VIR_FREE(groupname);
1059 1060 1061 1062
        } else {
            group = libvirtd_mdns_add_group(server->mdns, mdns_name);
        }

1063
        /*
1064 1065 1066 1067 1068 1069
         * See if there's a TLS enabled port we can advertise. Cowardly
         * don't bother to advertise TCP since we don't want people using
         * them for real world apps
         */
        sock = server->sockets;
        while (sock) {
1070 1071 1072
            if (virSocketGetPort(&sock->addr) != -1 &&
                sock->type == QEMUD_SOCK_TYPE_TLS) {
                port = virSocketGetPort(&sock->addr);
1073 1074 1075 1076
                break;
            }
            sock = sock->next;
        }
1077

1078 1079 1080 1081 1082 1083 1084 1085 1086
        /*
         * Add the primary entry - we choose SSH because its most likely to always
         * be available
         */
        libvirtd_mdns_add_entry(group, "_libvirt._tcp", port);
        libvirtd_mdns_start(server->mdns);
    }
#endif

1087
    return 0;
D
Daniel P. Berrange 已提交
1088 1089

 cleanup:
1090 1091
    VIR_FREE(sockname);
    VIR_FREE(roSockname);
1092
    return -1;
D
Daniel P. Berrange 已提交
1093 1094
}

1095 1096 1097 1098 1099
static int qemudNetworkEnable(struct qemud_server *server) {
    struct qemud_socket *sock;

    sock = server->sockets;
    while (sock) {
1100 1101 1102 1103 1104 1105
        if ((sock->watch = virEventAddHandle(sock->fd,
                                             VIR_EVENT_HANDLE_READABLE |
                                             VIR_EVENT_HANDLE_ERROR |
                                             VIR_EVENT_HANDLE_HANGUP,
                                             qemudDispatchServerEvent,
                                             server, NULL)) < 0) {
1106 1107 1108 1109 1110 1111 1112 1113
            VIR_ERROR0(_("Failed to add server event callback"));
            return -1;
        }

        sock = sock->next;
    }
    return 0;
}
1114

1115

1116 1117 1118
static gnutls_session_t
remoteInitializeTLSSession (void)
{
1119 1120
    gnutls_session_t session;
    int err;
1121

1122 1123
    err = gnutls_init (&session, GNUTLS_SERVER);
    if (err != 0) goto failed;
1124

1125 1126 1127 1128 1129
    /* avoid calling all the priority functions, since the defaults
     * are adequate.
     */
    err = gnutls_set_default_priority (session);
    if (err != 0) goto failed;
1130

1131 1132
    err = gnutls_credentials_set (session, GNUTLS_CRD_CERTIFICATE, x509_cred);
    if (err != 0) goto failed;
1133

1134 1135 1136
    /* request client certificate if any.
     */
    gnutls_certificate_server_set_request (session, GNUTLS_CERT_REQUEST);
1137

1138
    gnutls_dh_set_prime_bits (session, DH_BITS);
1139

1140
    return session;
1141

1142 1143 1144 1145
failed:
    VIR_ERROR(_("remoteInitializeTLSSession: %s"),
              gnutls_strerror (err));
    return NULL;
1146 1147 1148 1149
}

/* Check DN is on tls_allowed_dn_list. */
static int
1150
remoteCheckDN (const char *dname)
1151
{
1152
    char **wildcards;
1153 1154 1155 1156 1157 1158 1159

    /* If the list is not set, allow any DN. */
    wildcards = tls_allowed_dn_list;
    if (!wildcards)
        return 1;

    while (*wildcards) {
1160
        if (fnmatch (*wildcards, dname, 0) == 0)
1161 1162 1163 1164 1165
            return 1;
        wildcards++;
    }

    /* Print the client's DN. */
1166
    VIR_DEBUG("remoteCheckDN: failed: client DN is %s", dname);
1167

1168
    return 0; /* Not found. */
1169 1170 1171
}

static int
1172
remoteCheckCertificate(struct qemud_client *client)
1173 1174 1175 1176 1177 1178
{
    int ret;
    unsigned int status;
    const gnutls_datum_t *certs;
    unsigned int nCerts, i;
    time_t now;
1179 1180 1181 1182
    char name[256];
    size_t namesize = sizeof name;

    memset(name, 0, namesize);
1183

1184 1185
    if ((ret = gnutls_certificate_verify_peers2 (client->tlssession, &status)) < 0){
        VIR_ERROR(_("Failed to verify certificate peers: %s"),
1186
                  gnutls_strerror (ret));
1187
        goto authdeny;
1188 1189 1190 1191
    }

    if (status != 0) {
        if (status & GNUTLS_CERT_INVALID)
1192
            VIR_ERROR0(_("The client certificate is not trusted."));
1193 1194

        if (status & GNUTLS_CERT_SIGNER_NOT_FOUND)
1195
            VIR_ERROR0(_("The client certificate has unknown issuer."));
1196 1197

        if (status & GNUTLS_CERT_REVOKED)
1198
            VIR_ERROR0(_("The client certificate has been revoked."));
1199

1200
#ifndef GNUTLS_1_0_COMPAT
1201
        if (status & GNUTLS_CERT_INSECURE_ALGORITHM)
1202
            VIR_ERROR0(_("The client certificate uses an insecure algorithm."));
1203
#endif
1204

1205
        goto authdeny;
1206 1207
    }

1208 1209 1210
    if (gnutls_certificate_type_get(client->tlssession) != GNUTLS_CRT_X509) {
        VIR_ERROR0(_("Only x509 certificates are supported"));
        goto authdeny;
1211 1212
    }

1213 1214 1215
    if (!(certs = gnutls_certificate_get_peers(client->tlssession, &nCerts))) {
        VIR_ERROR0(_("The certificate has no peers"));
        goto authdeny;
1216 1217 1218 1219 1220 1221 1222 1223
    }

    now = time (NULL);

    for (i = 0; i < nCerts; i++) {
        gnutls_x509_crt_t cert;

        if (gnutls_x509_crt_init (&cert) < 0) {
1224 1225
            VIR_ERROR0(_("Unable to initialize certificate"));
            goto authfail;
1226 1227 1228
        }

        if (gnutls_x509_crt_import(cert, &certs[i], GNUTLS_X509_FMT_DER) < 0) {
1229
            VIR_ERROR0(_("Unable to load certificate"));
1230
            gnutls_x509_crt_deinit (cert);
1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252
            goto authfail;
        }

        if (i == 0) {
            ret = gnutls_x509_crt_get_dn (cert, name, &namesize);
            if (ret != 0) {
                VIR_ERROR(_("Failed to get certificate distinguished name: %s"),
                          gnutls_strerror(ret));
                gnutls_x509_crt_deinit (cert);
                goto authfail;
            }

            if (!remoteCheckDN (name)) {
                /* This is the most common error: make it informative. */
                VIR_ERROR0(_("Client's Distinguished Name is not on the list "
                             "of allowed clients (tls_allowed_dn_list).  Use "
                             "'certtool -i --infile clientcert.pem' to view the"
                             "Distinguished Name field in the client certificate,"
                             "or run this daemon with --verbose option."));
                gnutls_x509_crt_deinit (cert);
                goto authdeny;
            }
1253
        }
1254

1255
        if (gnutls_x509_crt_get_expiration_time (cert) < now) {
1256
            VIR_ERROR0(_("The client certificate has expired"));
1257
            gnutls_x509_crt_deinit (cert);
1258
            goto authdeny;
1259
        }
1260

1261
        if (gnutls_x509_crt_get_activation_time (cert) > now) {
1262
            VIR_ERROR0(_("The client certificate is not yet active"));
1263
            gnutls_x509_crt_deinit (cert);
1264
            goto authdeny;
1265 1266 1267
        }
    }

M
Matthias Bolte 已提交
1268
    PROBE(CLIENT_TLS_ALLOW, "fd=%d, name=%s", client->fd, (char *)name);
1269
    return 0;
1270 1271

authdeny:
M
Matthias Bolte 已提交
1272
    PROBE(CLIENT_TLS_DENY, "fd=%d, name=%s", client->fd, (char *)name);
1273 1274 1275
    return -1;

authfail:
1276
    PROBE(CLIENT_TLS_FAIL, "fd=%d", client->fd);
1277
    return -1;
1278 1279 1280 1281 1282 1283
}

/* Check the client's access. */
static int
remoteCheckAccess (struct qemud_client *client)
{
1284 1285
    struct qemud_client_message *confirm;

1286
    /* Verify client certificate. */
1287
    if (remoteCheckCertificate (client) == -1) {
1288 1289
        VIR_ERROR0(_("remoteCheckCertificate: "
                     "failed to verify client's certificate"));
1290
        if (!tls_no_verify_certificate) return -1;
1291 1292
        else VIR_INFO0(_("remoteCheckCertificate: tls_no_verify_certificate "
                          "is set so the bad certificate is ignored"));
1293 1294
    }

1295 1296 1297 1298 1299 1300 1301 1302 1303
    if (client->tx) {
        VIR_INFO("%s",
                 _("client had unexpected data pending tx after access check"));
        return -1;
    }

    if (VIR_ALLOC(confirm) < 0)
        return -1;

1304 1305 1306 1307
    /* Checks have succeeded.  Write a '\1' byte back to the client to
     * indicate this (otherwise the socket is abruptly closed).
     * (NB. The '\1' byte is sent in an encrypted record).
     */
1308 1309 1310 1311 1312 1313
    confirm->async = 1;
    confirm->bufferLength = 1;
    confirm->bufferOffset = 0;
    confirm->buffer[0] = '\1';

    client->tx = confirm;
1314 1315
    return 0;
}
D
Daniel P. Berrange 已提交
1316

1317 1318
#if HAVE_POLKIT
int qemudGetSocketIdentity(int fd, uid_t *uid, pid_t *pid) {
1319
# ifdef SO_PEERCRED
1320 1321 1322 1323
    struct ucred cr;
    unsigned int cr_len = sizeof (cr);

    if (getsockopt (fd, SOL_SOCKET, SO_PEERCRED, &cr, &cr_len) < 0) {
1324
        char ebuf[1024];
1325
        VIR_ERROR(_("Failed to verify client credentials: %s"),
1326
                  virStrerror(errno, ebuf, sizeof ebuf));
1327 1328 1329 1330 1331
        return -1;
    }

    *pid = cr.pid;
    *uid = cr.uid;
1332
# else
1333
    /* XXX Many more OS support UNIX socket credentials we could port to. See dbus ....*/
1334 1335
#  error "UNIX socket credentials not supported/implemented on this platform yet..."
# endif
1336 1337 1338 1339
    return 0;
}
#endif

1340

D
Daniel P. Berrange 已提交
1341 1342
static int qemudDispatchServer(struct qemud_server *server, struct qemud_socket *sock) {
    int fd;
1343 1344
    virSocketAddr addr;
    char *addrstr = NULL;
1345
    struct qemud_client *client = NULL;
1346
    int no_slow_start = 1;
1347
    int i;
D
Daniel P. Berrange 已提交
1348

1349 1350
    addr.len = sizeof(addr.data.stor);
    if ((fd = accept(sock->fd, &addr.data.sa, &addr.len)) < 0) {
1351
        char ebuf[1024];
D
Daniel P. Berrange 已提交
1352 1353
        if (errno == EAGAIN)
            return 0;
1354 1355
        VIR_ERROR(_("Failed to accept connection: %s"),
                  virStrerror(errno, ebuf, sizeof ebuf));
D
Daniel P. Berrange 已提交
1356 1357
        return -1;
    }
1358 1359 1360 1361
    if (!(addrstr = virSocketFormatAddrFull(&addr, true, ";"))) {
        VIR_ERROR0(_("Failed to format addresss: out of memory"));
        goto error;
    }
D
Daniel P. Berrange 已提交
1362

1363 1364
    PROBE(CLIENT_CONNECT, "fd=%d, readonly=%d localAddr=%s remoteAddr=%s",
          fd, sock->readonly, sock->addrstr, addrstr);
1365

1366
    if (server->nclients >= max_clients) {
1367 1368
        VIR_ERROR(_("Too many active clients (%d), dropping connection from %s"),
                  max_clients, addrstr);
1369
        goto error;
1370 1371
    }

1372 1373
    if (VIR_RESIZE_N(server->clients, server->nclients_max,
                     server->nclients, 1) < 0) {
1374
        VIR_ERROR0(_("Out of memory allocating clients"));
1375
        goto error;
1376 1377
    }

J
John Levon 已提交
1378 1379 1380 1381 1382 1383 1384 1385 1386
#ifdef __sun
    {
        ucred_t *ucred = NULL;
        const priv_set_t *privs;

        if (getpeerucred (fd, &ucred) == -1 ||
            (privs = ucred_getprivset (ucred, PRIV_EFFECTIVE)) == NULL) {
            if (ucred != NULL)
                ucred_free (ucred);
1387
            goto error;
J
John Levon 已提交
1388 1389 1390 1391
        }

        if (!priv_ismember (privs, PRIV_VIRT_MANAGE)) {
            ucred_free (ucred);
1392
            goto error;
J
John Levon 已提交
1393 1394 1395 1396 1397 1398
        }

        ucred_free (ucred);
    }
#endif /* __sun */

1399 1400 1401 1402
    /* Disable Nagle.  Unix sockets will ignore this. */
    setsockopt (fd, IPPROTO_TCP, TCP_NODELAY, (void *)&no_slow_start,
                sizeof no_slow_start);

1403 1404
    if (virSetCloseExec(fd) < 0 ||
        virSetNonBlock(fd) < 0) {
1405
        goto error;
D
Daniel P. Berrange 已提交
1406 1407
    }

1408
    if (VIR_ALLOC(client) < 0)
1409
        goto error;
1410
    if (virMutexInit(&client->lock) < 0) {
1411
        VIR_ERROR0(_("cannot initialize mutex"));
1412
        goto error;
1413
    }
1414

1415
    client->magic = QEMUD_CLIENT_MAGIC;
D
Daniel P. Berrange 已提交
1416 1417
    client->fd = fd;
    client->readonly = sock->readonly;
1418
    client->type = sock->type;
1419
    client->auth = sock->auth;
1420 1421 1422
    client->addr = addr;
    client->addrstr = addrstr;
    addrstr = NULL;
1423

1424 1425 1426 1427
    for (i = 0 ; i < VIR_DOMAIN_EVENT_ID_LAST ; i++) {
        client->domainEventCallbackID[i] = -1;
    }

1428 1429
    /* Prepare one for packet receive */
    if (VIR_ALLOC(client->rx) < 0)
1430
        goto error;
1431 1432 1433
    client->rx->bufferLength = REMOTE_MESSAGE_HEADER_XDR_LEN;


1434 1435 1436 1437 1438 1439 1440 1441 1442 1443
#if HAVE_POLKIT
    /* Only do policy checks for non-root - allow root user
       through with no checks, as a fail-safe - root can easily
       change policykit policy anyway, so its pointless trying
       to restrict root */
    if (client->auth == REMOTE_AUTH_POLKIT) {
        uid_t uid;
        pid_t pid;

        if (qemudGetSocketIdentity(client->fd, &uid, &pid) < 0)
1444
            goto error;
1445

1446
        /* Client is running as root, so disable auth */
1447
        if (uid == 0) {
1448
            VIR_INFO(_("Turn off polkit auth for privileged client pid %d from %s"),
E
Eric Blake 已提交
1449
                     pid, client->addrstr);
1450 1451 1452 1453 1454
            client->auth = REMOTE_AUTH_NONE;
        }
    }
#endif

1455
    if (client->type != QEMUD_SOCK_TYPE_TLS) {
1456
        /* Plain socket, so prepare to read first message */
1457
        if (qemudRegisterClientEvent (server, client) < 0)
1458
            goto error;
1459 1460 1461
    } else {
        int ret;

1462 1463
        client->tlssession = remoteInitializeTLSSession ();
        if (client->tlssession == NULL)
1464
            goto error;
1465

1466
        gnutls_transport_set_ptr (client->tlssession,
1467 1468 1469
                                  (gnutls_transport_ptr_t) (long) fd);

        /* Begin the TLS handshake. */
1470
        ret = gnutls_handshake (client->tlssession);
1471
        if (ret == 0) {
1472 1473
            client->handshake = 0;

1474 1475
            /* Unlikely, but ...  Next step is to check the certificate. */
            if (remoteCheckAccess (client) == -1)
1476
                goto error;
D
Daniel P. Berrange 已提交
1477

1478
            /* Handshake & cert check OK,  so prepare to read first message */
1479
            if (qemudRegisterClientEvent(server, client) < 0)
1480
                goto error;
1481
        } else if (ret == GNUTLS_E_INTERRUPTED || ret == GNUTLS_E_AGAIN) {
1482 1483
            /* Most likely, need to do more handshake data */
            client->handshake = 1;
D
Daniel P. Berrange 已提交
1484

1485
            if (qemudRegisterClientEvent (server, client) < 0)
1486
                goto error;
1487
        } else {
1488
            PROBE(CLIENT_TLS_FAIL, "fd=%d", client->fd);
1489
            VIR_ERROR(_("TLS handshake failed for client %s: %s"),
E
Eric Blake 已提交
1490
                      client->addrstr, gnutls_strerror (ret));
1491
            goto error;
1492 1493
        }
    }
D
Daniel P. Berrange 已提交
1494

1495
    server->clients[server->nclients++] = client;
D
Daniel P. Berrange 已提交
1496

1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509
    if (server->nclients > server->nactiveworkers &&
        server->nactiveworkers < server->nworkers) {
        for (i = 0 ; i < server->nworkers ; i++) {
            if (!server->workers[i].hasThread) {
                if (qemudStartWorker(server, &server->workers[i]) < 0)
                    return -1;
                server->nactiveworkers++;
                break;
            }
        }
    }


D
Daniel P. Berrange 已提交
1510
    return 0;
1511

1512 1513 1514
error:
    if (client) {
        if (client->tlssession) gnutls_deinit (client->tlssession);
1515 1516
        if (client) {
            VIR_FREE(client->addrstr);
1517
            VIR_FREE(client->rx);
1518
        }
1519 1520
        VIR_FREE(client);
    }
1521
    VIR_FREE(addrstr);
1522
    VIR_FORCE_CLOSE(fd);
1523
    PROBE(CLIENT_DISCONNECT, "fd=%d", fd);
1524
    return -1;
D
Daniel P. Berrange 已提交
1525 1526 1527
}


1528 1529 1530 1531 1532 1533 1534
/*
 * You must hold lock for at least the client
 * We don't free stuff here, merely disconnect the client's
 * network socket & resources.
 * We keep the libvirt connection open until any async
 * jobs have finished, then clean it up elsehwere
 */
1535
void qemudDispatchClientFailure(struct qemud_client *client) {
1536
    if (client->watch != -1) {
1537
        virEventRemoveHandle(client->watch);
1538 1539
        client->watch = -1;
    }
D
Daniel P. Berrange 已提交
1540

1541
    /* Deregister event delivery callback */
1542 1543 1544 1545 1546
    if (client->conn) {
        int i;

        for (i = 0 ; i < VIR_DOMAIN_EVENT_ID_LAST ; i++) {
            if (client->domainEventCallbackID[i] != -1) {
1547
                VIR_DEBUG("Deregistering to relay remote events %d", i);
1548 1549 1550 1551 1552
                virConnectDomainEventDeregisterAny(client->conn,
                                                   client->domainEventCallbackID[i]);
            }
            client->domainEventCallbackID[i] = -1;
        }
1553 1554
    }

1555
#if HAVE_SASL
1556 1557 1558 1559
    if (client->saslconn) {
        sasl_dispose(&client->saslconn);
        client->saslconn = NULL;
    }
1560
    VIR_FREE(client->saslUsername);
1561
#endif
1562 1563 1564 1565 1566
    if (client->tlssession) {
        gnutls_deinit (client->tlssession);
        client->tlssession = NULL;
    }
    if (client->fd != -1) {
1567
        PROBE(CLIENT_DISCONNECT, "fd=%d", client->fd);
1568
        VIR_FORCE_CLOSE(client->fd);
1569
    }
1570
    VIR_FREE(client->addrstr);
1571 1572 1573 1574 1575 1576 1577 1578
}


/* Caller must hold server lock */
static struct qemud_client *qemudPendingJob(struct qemud_server *server)
{
    int i;
    for (i = 0 ; i < server->nclients ; i++) {
1579
        virMutexLock(&server->clients[i]->lock);
1580
        if (server->clients[i]->dx) {
1581 1582 1583
            /* Delibrately don't unlock client - caller wants the lock */
            return server->clients[i];
        }
1584
        virMutexUnlock(&server->clients[i]->lock);
1585 1586
    }
    return NULL;
D
Daniel P. Berrange 已提交
1587 1588
}

1589 1590
static void *qemudWorker(void *data)
{
1591 1592
    struct qemud_worker *worker = data;
    struct qemud_server *server = worker->server;
1593 1594

    while (1) {
1595
        struct qemud_client *client = NULL;
1596
        struct qemud_client_message *msg;
1597

1598
        virMutexLock(&server->lock);
1599 1600 1601
        while ((client = qemudPendingJob(server)) == NULL) {
            if (worker->quitRequest ||
                virCondWait(&server->job, &server->lock) < 0) {
1602 1603 1604 1605
                virMutexUnlock(&server->lock);
                return NULL;
            }
        }
1606
        if (worker->quitRequest) {
1607
            virMutexUnlock(&client->lock);
1608 1609 1610 1611
            virMutexUnlock(&server->lock);
            return NULL;
        }
        worker->processingCall = 1;
1612
        virMutexUnlock(&server->lock);
1613 1614 1615 1616

        /* We own a locked client now... */
        client->refs++;

1617
        /* Remove our message from dispatch queue while we use it */
1618
        msg = qemudClientMessageQueueServe(&client->dx);
1619 1620 1621

        /* This function drops the lock during dispatch,
         * and re-acquires it before returning */
1622
        if (remoteDispatchClientRequest (server, client, msg) < 0) {
1623
            VIR_FREE(msg);
1624 1625 1626 1627 1628
            qemudDispatchClientFailure(client);
            client->refs--;
            virMutexUnlock(&client->lock);
            continue;
        }
1629 1630

        client->refs--;
1631
        virMutexUnlock(&client->lock);
1632 1633 1634 1635

        virMutexLock(&server->lock);
        worker->processingCall = 0;
        virMutexUnlock(&server->lock);
1636 1637
    }
}
D
Daniel P. Berrange 已提交
1638

E
Eric Blake 已提交
1639 1640 1641 1642
static int
qemudStartWorker(struct qemud_server *server,
                 struct qemud_worker *worker)
{
1643
    pthread_attr_t attr;
E
Eric Blake 已提交
1644 1645 1646 1647
    int ret = -1;

    if (pthread_attr_init(&attr) != 0)
        return -1;
1648 1649 1650 1651
    /* We want to join workers, so don't detach them */
    /*pthread_attr_setdetachstate(&attr, 1);*/

    if (worker->hasThread)
E
Eric Blake 已提交
1652
        goto cleanup;
1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664

    worker->server = server;
    worker->hasThread = 1;
    worker->quitRequest = 0;
    worker->processingCall = 0;

    if (pthread_create(&worker->thread,
                       &attr,
                       qemudWorker,
                       worker) != 0) {
        worker->hasThread = 0;
        worker->server = NULL;
E
Eric Blake 已提交
1665
        goto cleanup;
1666 1667
    }

E
Eric Blake 已提交
1668 1669 1670 1671
    ret = 0;
cleanup:
    pthread_attr_destroy(&attr);
    return ret;
1672 1673
}

D
Daniel P. Berrange 已提交
1674

1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687
/*
 * Read data into buffer using wire decoding (plain or TLS)
 *
 * Returns:
 *   -1 on error or EOF
 *    0 on EAGAIN
 *    n number of bytes
 */
static ssize_t qemudClientReadBuf(struct qemud_client *client,
                                  char *data, ssize_t len) {
    ssize_t ret;

    if (len < 0) {
1688 1689
        VIR_ERROR(_("unexpected negative length request %lld"),
                  (long long int) len);
1690 1691 1692
        qemudDispatchClientFailure(client);
        return -1;
    }
1693

1694
    /* VIR_DEBUG("qemudClientRead: len = %d", len);*/
1695

1696
    if (!client->tlssession) {
1697
        char ebuf[1024];
1698 1699 1700 1701 1702 1703
        ret = read (client->fd, data, len);
        if (ret == -1 && (errno == EAGAIN ||
                          errno == EINTR))
            return 0;
        if (ret <= 0) {
            if (ret != 0)
1704 1705
                VIR_ERROR(_("read: %s"),
                          virStrerror (errno, ebuf, sizeof ebuf));
1706
            qemudDispatchClientFailure(client);
1707 1708 1709
            return -1;
        }
    } else {
1710
        ret = gnutls_record_recv (client->tlssession, data, len);
1711 1712 1713 1714 1715 1716 1717 1718 1719

        if (ret < 0 && (ret == GNUTLS_E_AGAIN ||
                        ret == GNUTLS_E_INTERRUPTED))
            return 0;
        if (ret <= 0) {
            if (ret != 0)
                VIR_ERROR(_("gnutls_record_recv: %s"),
                          gnutls_strerror (ret));
            qemudDispatchClientFailure(client);
1720 1721
            return -1;
        }
D
Daniel P. Berrange 已提交
1722
    }
1723

1724 1725 1726
    return ret;
}

1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744
/*
 * Read data into buffer without decoding
 *
 * Returns:
 *   -1 on error or EOF
 *    0 on EAGAIN
 *    n number of bytes
 */
static ssize_t qemudClientReadPlain(struct qemud_client *client) {
    ssize_t ret;
    ret = qemudClientReadBuf(client,
                             client->rx->buffer + client->rx->bufferOffset,
                             client->rx->bufferLength - client->rx->bufferOffset);
    if (ret <= 0)
        return ret; /* -1 error, 0 eagain */

    client->rx->bufferOffset += ret;
    return ret;
D
Daniel P. Berrange 已提交
1745 1746
}

1747
#if HAVE_SASL
1748 1749 1750 1751 1752 1753 1754 1755 1756 1757
/*
 * Read data into buffer decoding with SASL
 *
 * Returns:
 *   -1 on error or EOF
 *    0 on EAGAIN
 *    n number of bytes
 */
static ssize_t qemudClientReadSASL(struct qemud_client *client) {
    ssize_t got, want;
1758 1759 1760 1761 1762 1763 1764 1765 1766 1767

    /* We're doing a SSF data read, so now its times to ensure
     * future writes are under SSF too.
     *
     * cf remoteSASLCheckSSF in remote.c
     */
    client->saslSSF |= QEMUD_SASL_SSF_WRITE;

    /* Need to read some more data off the wire */
    if (client->saslDecoded == NULL) {
1768
        int ret;
1769 1770 1771 1772
        ssize_t encodedLen;

        encodedLen = qemudClientReadBuf(client, client->saslTemporary,
                                        sizeof(client->saslTemporary));
1773 1774 1775 1776

        if (encodedLen <= 0)
            return encodedLen;

1777
        ret = sasl_decode(client->saslconn, client->saslTemporary, encodedLen,
1778
                          &client->saslDecoded, &client->saslDecodedLength);
1779

1780 1781 1782 1783
        if (ret != SASL_OK) {
            VIR_ERROR(_("failed to decode SASL data %s"),
                      sasl_errstring(ret, NULL, NULL));
            qemudDispatchClientFailure(client);
1784
            return -1;
1785
        }
1786 1787 1788 1789 1790 1791

        client->saslDecodedOffset = 0;
    }

    /* Some buffered decoded data to return now */
    got = client->saslDecodedLength - client->saslDecodedOffset;
1792
    want = client->rx->bufferLength - client->rx->bufferOffset;
1793 1794 1795 1796

    if (want > got)
        want = got;

1797
    memcpy(client->rx->buffer + client->rx->bufferOffset,
1798 1799
           client->saslDecoded + client->saslDecodedOffset, want);
    client->saslDecodedOffset += want;
1800
    client->rx->bufferOffset += want;
1801 1802 1803 1804 1805 1806

    if (client->saslDecodedOffset == client->saslDecodedLength) {
        client->saslDecoded = NULL;
        client->saslDecodedOffset = client->saslDecodedLength = 0;
    }

1807
    return want;
1808 1809 1810
}
#endif

1811 1812 1813 1814 1815
/*
 * Read as much data off wire as possible till we fill our
 * buffer, or would block on I/O
 */
static ssize_t qemudClientRead(struct qemud_client *client) {
1816 1817
#if HAVE_SASL
    if (client->saslSSF & QEMUD_SASL_SSF_READ)
1818
        return qemudClientReadSASL(client);
1819 1820
    else
#endif
1821
        return qemudClientReadPlain(client);
1822 1823 1824
}


1825 1826 1827 1828 1829
/*
 * Read data until we get a complete message to process
 */
static void qemudDispatchClientRead(struct qemud_server *server,
                                    struct qemud_client *client) {
1830
    /* VIR_DEBUG("qemudDispatchClientRead: mode = %d", client->mode);*/
D
Daniel P. Berrange 已提交
1831

1832 1833 1834
readmore:
    if (qemudClientRead(client) < 0)
        return; /* Error */
1835

1836 1837
    if (client->rx->bufferOffset < client->rx->bufferLength)
        return; /* Still not read enough */
1838

1839 1840 1841 1842
    /* Either done with length word header */
    if (client->rx->bufferLength == REMOTE_MESSAGE_HEADER_XDR_LEN) {
        unsigned int len;
        XDR x;
1843

1844
        xdrmem_create(&x, client->rx->buffer, client->rx->bufferLength, XDR_DECODE);
1845

1846 1847
        if (!xdr_u_int(&x, &len)) {
            xdr_destroy (&x);
1848
            VIR_DEBUG0("Failed to decode packet length");
1849
            qemudDispatchClientFailure(client);
D
Daniel P. Berrange 已提交
1850 1851
            return;
        }
1852
        xdr_destroy (&x);
1853

1854
        if (len < REMOTE_MESSAGE_HEADER_XDR_LEN) {
1855
            VIR_DEBUG("Packet length %u too small", len);
1856
            qemudDispatchClientFailure(client);
1857
            return;
D
Daniel P. Berrange 已提交
1858
        }
1859

1860 1861 1862 1863
        /* Length includes the size of the length word itself */
        len -= REMOTE_MESSAGE_HEADER_XDR_LEN;

        if (len > REMOTE_MESSAGE_MAX) {
1864
            VIR_DEBUG("Packet length %u too large", len);
1865
            qemudDispatchClientFailure(client);
1866 1867 1868
            return;
        }

1869 1870
        /* Prepare to read rest of message */
        client->rx->bufferLength += len;
1871

1872
        qemudUpdateClientEvent(client);
D
Daniel P. Berrange 已提交
1873

1874 1875 1876 1877 1878
        /* Try and read payload immediately instead of going back
           into poll() because chances are the data is already
           waiting for us */
        goto readmore;
    } else {
1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891
        /* Grab the completed message */
        struct qemud_client_message *msg = qemudClientMessageQueueServe(&client->rx);
        struct qemud_client_filter *filter;

        /* Decode the header so we can use it for routing decisions */
        if (remoteDecodeClientMessageHeader(msg) < 0) {
            VIR_FREE(msg);
            qemudDispatchClientFailure(client);
        }

        /* Check if any filters match this message */
        filter = client->filters;
        while (filter) {
1892 1893 1894
            int ret;
            ret = (filter->query)(client, msg, filter->opaque);
            if (ret == 1) {
1895 1896
                msg = NULL;
                break;
1897 1898 1899 1900
            } else if (ret == -1) {
                VIR_FREE(msg);
                qemudDispatchClientFailure(client);
                return;
1901 1902 1903 1904
            }
            filter = filter->next;
        }

1905
        /* Move completed message to the end of the dispatch queue */
1906 1907
        if (msg)
            qemudClientMessageQueuePush(&client->dx, msg);
1908 1909 1910 1911 1912 1913
        client->nrequests++;

        /* Possibly need to create another receive buffer */
        if ((client->nrequests < max_client_requests &&
             VIR_ALLOC(client->rx) < 0)) {
            qemudDispatchClientFailure(client);
D
Daniel P. Berrange 已提交
1914
        } else {
1915 1916 1917
            if (client->rx)
                client->rx->bufferLength = REMOTE_MESSAGE_HEADER_XDR_LEN;

1918 1919 1920 1921
            qemudUpdateClientEvent(client);

            /* Tell one of the workers to get on with it... */
            virCondSignal(&server->job);
D
Daniel P. Berrange 已提交
1922
        }
D
Daniel P. Berrange 已提交
1923 1924 1925 1926
    }
}


1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939
/*
 * Send a chunk of data using wire encoding (plain or TLS)
 *
 * Returns:
 *   -1 on error
 *    0 on EAGAIN
 *    n number of bytes
 */
static ssize_t qemudClientWriteBuf(struct qemud_client *client,
                                   const char *data, ssize_t len) {
    ssize_t ret;

    if (len < 0) {
1940 1941
        VIR_ERROR(_("unexpected negative length request %lld"),
                  (long long int) len);
1942 1943 1944 1945
        qemudDispatchClientFailure(client);
        return -1;
    }

1946
    if (!client->tlssession) {
1947
        char ebuf[1024];
1948 1949 1950
        if ((ret = write(client->fd, data, len)) == -1) {
            if (errno == EAGAIN || errno == EINTR)
                return 0;
1951
            VIR_ERROR(_("write: %s"), virStrerror (errno, ebuf, sizeof ebuf));
1952
            qemudDispatchClientFailure(client);
1953 1954 1955
            return -1;
        }
    } else {
1956
        ret = gnutls_record_send (client->tlssession, data, len);
1957 1958 1959 1960 1961 1962 1963
        if (ret < 0) {
            if (ret == GNUTLS_E_INTERRUPTED ||
                ret == GNUTLS_E_AGAIN)
                return 0;

            VIR_ERROR(_("gnutls_record_send: %s"), gnutls_strerror (ret));
            qemudDispatchClientFailure(client);
1964 1965
            return -1;
        }
D
Daniel P. Berrange 已提交
1966
    }
1967 1968
    return ret;
}
1969

1970

1971 1972 1973 1974 1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986
/*
 * Send client->tx using no encoding
 *
 * Returns:
 *   -1 on error or EOF
 *    0 on EAGAIN
 *    n number of bytes
 */
static int qemudClientWritePlain(struct qemud_client *client) {
    int ret = qemudClientWriteBuf(client,
                                  client->tx->buffer + client->tx->bufferOffset,
                                  client->tx->bufferLength - client->tx->bufferOffset);
    if (ret <= 0)
        return ret; /* -1 error, 0 = egain */
    client->tx->bufferOffset += ret;
    return ret;
D
Daniel P. Berrange 已提交
1987 1988 1989
}


1990
#if HAVE_SASL
1991 1992 1993 1994 1995 1996 1997 1998 1999
/*
 * Send client->tx using SASL encoding
 *
 * Returns:
 *   -1 on error
 *    0 on EAGAIN
 *    n number of bytes
 */
static int qemudClientWriteSASL(struct qemud_client *client) {
2000 2001 2002 2003
    int ret;

    /* Not got any pending encoded data, so we need to encode raw stuff */
    if (client->saslEncoded == NULL) {
2004 2005 2006
        ret = sasl_encode(client->saslconn,
                          client->tx->buffer + client->tx->bufferOffset,
                          client->tx->bufferLength - client->tx->bufferOffset,
2007 2008 2009
                          &client->saslEncoded,
                          &client->saslEncodedLength);

2010 2011 2012 2013 2014 2015 2016
        if (ret != SASL_OK) {
            VIR_ERROR(_("failed to encode SASL data %s"),
                      sasl_errstring(ret, NULL, NULL));
            qemudDispatchClientFailure(client);
            return -1;
        }

2017 2018 2019 2020
        client->saslEncodedOffset = 0;
    }

    /* Send some of the encoded stuff out on the wire */
2021
    ret = qemudClientWriteBuf(client,
2022 2023 2024
                              client->saslEncoded + client->saslEncodedOffset,
                              client->saslEncodedLength - client->saslEncodedOffset);

2025 2026
    if (ret <= 0)
        return ret; /* -1 error, 0 == egain */
2027 2028 2029 2030 2031 2032 2033 2034

    /* Note how much we sent */
    client->saslEncodedOffset += ret;

    /* Sent all encoded, so update raw buffer to indicate completion */
    if (client->saslEncodedOffset == client->saslEncodedLength) {
        client->saslEncoded = NULL;
        client->saslEncodedOffset = client->saslEncodedLength = 0;
2035 2036 2037

        /* Mark as complete, so caller detects completion */
        client->tx->bufferOffset = client->tx->bufferLength;
2038 2039
    }

2040
    return ret;
2041 2042 2043
}
#endif

2044 2045 2046 2047 2048 2049 2050 2051 2052
/*
 * Send as much data in the client->tx as possible
 *
 * Returns:
 *   -1 on error or EOF
 *    0 on EAGAIN
 *    n number of bytes
 */
static ssize_t qemudClientWrite(struct qemud_client *client) {
2053 2054
#if HAVE_SASL
    if (client->saslSSF & QEMUD_SASL_SSF_WRITE)
2055
        return qemudClientWriteSASL(client);
2056 2057
    else
#endif
2058
        return qemudClientWritePlain(client);
2059 2060 2061
}


2062 2063 2064 2065
void
qemudClientMessageRelease(struct qemud_client *client,
                          struct qemud_client_message *msg)
{
2066 2067 2068
    if (msg->streamTX) {
        remoteStreamMessageFinished(client, msg);
    } else if (!msg->async)
2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083 2084 2085 2086
        client->nrequests--;

    /* See if the recv queue is currently throttled */
    if (!client->rx &&
        client->nrequests < max_client_requests) {
        /* Reset message record for next RX attempt */
        memset(msg, 0, sizeof(*msg));
        client->rx = msg;
        /* Get ready to receive next message */
        client->rx->bufferLength = REMOTE_MESSAGE_HEADER_XDR_LEN;
    } else {
        VIR_FREE(msg);
    }

    qemudUpdateClientEvent(client);
}


2087 2088 2089 2090 2091
/*
 * Process all queued client->tx messages until
 * we would block on I/O
 */
static void
2092
qemudDispatchClientWrite(struct qemud_client *client) {
2093 2094
    while (client->tx) {
        ssize_t ret;
2095

2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109
        ret = qemudClientWrite(client);
        if (ret < 0) {
            qemudDispatchClientFailure(client);
            return;
        }
        if (ret == 0)
            return; /* Would block on write EAGAIN */

        if (client->tx->bufferOffset == client->tx->bufferLength) {
            struct qemud_client_message *reply;

            /* Get finished reply from head of tx queue */
            reply = qemudClientMessageQueueServe(&client->tx);

2110
            qemudClientMessageRelease(client, reply);
2111

2112 2113
            if (client->closing)
                qemudDispatchClientFailure(client);
2114
         }
2115
    }
2116
}
2117

2118
static void
2119
qemudDispatchClientHandshake(struct qemud_client *client) {
2120 2121 2122 2123
    int ret;
    /* Continue the handshake. */
    ret = gnutls_handshake (client->tlssession);
    if (ret == 0) {
2124 2125
        client->handshake = 0;

2126 2127 2128
        /* Finished.  Next step is to check the certificate. */
        if (remoteCheckAccess (client) == -1)
            qemudDispatchClientFailure(client);
2129 2130
        else
            qemudUpdateClientEvent(client);
2131 2132 2133 2134 2135
    } else if (ret == GNUTLS_E_AGAIN ||
               ret == GNUTLS_E_INTERRUPTED) {
        /* Carry on waiting for more handshake. Update
           the events just in case handshake data flow
           direction has changed */
2136
        qemudUpdateClientEvent (client);
2137
    } else {
2138
        PROBE(CLIENT_TLS_FAIL, "fd=%d", client->fd);
2139 2140 2141 2142
        /* Fatal error in handshake */
        VIR_ERROR(_("TLS handshake failed: %s"),
                  gnutls_strerror (ret));
        qemudDispatchClientFailure(client);
D
Daniel P. Berrange 已提交
2143 2144 2145
    }
}

2146
static void
2147
qemudDispatchClientEvent(int watch, int fd, int events, void *opaque) {
D
Daniel P. Berrange 已提交
2148
    struct qemud_server *server = (struct qemud_server *)opaque;
2149 2150
    struct qemud_client *client = NULL;
    int i;
2151

2152
    virMutexLock(&server->lock);
2153

2154
    for (i = 0 ; i < server->nclients ; i++) {
2155
        virMutexLock(&server->clients[i]->lock);
2156 2157
        if (server->clients[i]->watch == watch) {
            client = server->clients[i];
D
Daniel P. Berrange 已提交
2158
            break;
2159
        }
2160
        virMutexUnlock(&server->clients[i]->lock);
D
Daniel P. Berrange 已提交
2161
    }
2162

2163 2164
    virMutexUnlock(&server->lock);

2165
    if (!client) {
D
Daniel P. Berrange 已提交
2166
        return;
2167 2168
    }

2169 2170
    if (client->fd != fd) {
        virMutexUnlock(&client->lock);
2171
        return;
2172 2173 2174 2175 2176
    }

    if (events & (VIR_EVENT_HANDLE_WRITABLE |
                  VIR_EVENT_HANDLE_READABLE)) {
        if (client->handshake) {
2177
            qemudDispatchClientHandshake(client);
2178 2179
        } else {
            if (events & VIR_EVENT_HANDLE_WRITABLE)
2180
                qemudDispatchClientWrite(client);
2181 2182 2183 2184 2185 2186 2187 2188 2189 2190
            if (events & VIR_EVENT_HANDLE_READABLE)
                qemudDispatchClientRead(server, client);
        }
    }

    /* NB, will get HANGUP + READABLE at same time upon
     * disconnect */
    if (events & (VIR_EVENT_HANDLE_ERROR |
                  VIR_EVENT_HANDLE_HANGUP))
        qemudDispatchClientFailure(client);
2191

2192
    virMutexUnlock(&client->lock);
D
Daniel P. Berrange 已提交
2193 2194
}

2195 2196 2197 2198 2199 2200

/*
 * @client: a locked client object
 */
static int
qemudCalculateHandleMode(struct qemud_client *client) {
2201 2202 2203
    int mode = 0;

    if (client->handshake) {
2204
        if (gnutls_record_get_direction (client->tlssession) == 0)
2205
            mode |= VIR_EVENT_HANDLE_READABLE;
2206
        else
2207 2208 2209 2210 2211 2212
            mode |= VIR_EVENT_HANDLE_WRITABLE;
    } else {
        /* If there is a message on the rx queue then
         * we're wanting more input */
        if (client->rx)
            mode |= VIR_EVENT_HANDLE_READABLE;
2213

2214 2215 2216 2217
        /* If there are one or more messages to send back to client,
           then monitor for writability on socket */
        if (client->tx)
            mode |= VIR_EVENT_HANDLE_WRITABLE;
2218 2219
    }

2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232
    return mode;
}

/*
 * @server: a locked or unlocked server object
 * @client: a locked client object
 */
int qemudRegisterClientEvent(struct qemud_server *server,
                             struct qemud_client *client) {
    int mode;

    mode = qemudCalculateHandleMode(client);

2233 2234 2235 2236
    if ((client->watch = virEventAddHandle(client->fd,
                                           mode,
                                           qemudDispatchClientEvent,
                                           server, NULL)) < 0)
2237
        return -1;
D
Daniel P. Berrange 已提交
2238 2239 2240 2241

    return 0;
}

2242 2243 2244 2245 2246 2247 2248 2249
/*
 * @client: a locked client object
 */
void qemudUpdateClientEvent(struct qemud_client *client) {
    int mode;

    mode = qemudCalculateHandleMode(client);

2250
    virEventUpdateHandle(client->watch, mode);
2251 2252 2253
}


2254
static void
2255
qemudDispatchServerEvent(int watch, int fd, int events, void *opaque) {
D
Daniel P. Berrange 已提交
2256
    struct qemud_server *server = (struct qemud_server *)opaque;
2257 2258
    struct qemud_socket *sock;

2259
    virMutexLock(&server->lock);
2260 2261

    sock = server->sockets;
D
Daniel P. Berrange 已提交
2262

D
Daniel P. Berrange 已提交
2263
    while (sock) {
2264
        if (sock->watch == watch)
D
Daniel P. Berrange 已提交
2265 2266 2267
            break;

        sock = sock->next;
D
Daniel P. Berrange 已提交
2268
    }
D
Daniel P. Berrange 已提交
2269

2270
    if (sock && sock->fd == fd && events)
D
Daniel P. Berrange 已提交
2271
        qemudDispatchServer(server, sock);
2272

2273
    virMutexUnlock(&server->lock);
D
Daniel P. Berrange 已提交
2274 2275 2276
}


2277
static int qemudOneLoop(void) {
2278
    sig_atomic_t errors;
D
Daniel P. Berrange 已提交
2279

2280
    if (virEventRunDefaultImpl() < 0)
D
Daniel P. Berrange 已提交
2281 2282
        return -1;

2283 2284 2285
    /* Check for any signal handling errors and log them. */
    errors = sig_errors;
    if (errors) {
2286
        char ebuf[1024];
2287
        sig_errors -= errors;
2288
        VIR_ERROR(_("Signal handler reported %d errors: last error: %s"),
2289
                  errors, virStrerror (sig_lasterrno, ebuf, sizeof ebuf));
2290 2291 2292
        return -1;
    }

D
Daniel P. Berrange 已提交
2293 2294 2295
    return 0;
}

2296
static void qemudInactiveTimer(int timerid, void *data) {
2297
    struct qemud_server *server = (struct qemud_server *)data;
2298 2299 2300

    if (virStateActive() ||
        server->clients) {
2301
        VIR_DEBUG0("Timer expired but still active, not shutting down");
2302
        virEventUpdateTimeout(timerid, -1);
2303
    } else {
2304
        VIR_DEBUG0("Timer expired and inactive, shutting down");
2305
        server->quitEventThread = 1;
2306 2307 2308
    }
}

2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325
static void qemudFreeClient(struct qemud_client *client) {
    while (client->rx) {
        struct qemud_client_message *msg
            = qemudClientMessageQueueServe(&client->rx);
        VIR_FREE(msg);
    }
    while (client->dx) {
        struct qemud_client_message *msg
            = qemudClientMessageQueueServe(&client->dx);
        VIR_FREE(msg);
    }
    while (client->tx) {
        struct qemud_client_message *msg
            = qemudClientMessageQueueServe(&client->tx);
        VIR_FREE(msg);
    }

2326 2327 2328
    while (client->streams)
        remoteRemoveClientStream(client, client->streams);

2329 2330 2331
    if (client->conn)
        virConnectClose(client->conn);
    virMutexDestroy(&client->lock);
E
Eric Blake 已提交
2332
    VIR_FREE(client->addrstr);
2333 2334 2335
    VIR_FREE(client);
}

2336 2337
static void *qemudRunLoop(void *opaque) {
    struct qemud_server *server = opaque;
2338
    int timerid = -1;
2339
    int i;
2340
    int timerActive = 0;
2341

2342
    virMutexLock(&server->lock);
2343

2344
    if (timeout > 0 &&
2345 2346 2347
        (timerid = virEventAddTimeout(-1,
                                      qemudInactiveTimer,
                                      server, NULL)) < 0) {
2348
        VIR_ERROR0(_("Failed to register shutdown timeout"));
2349
        return NULL;
2350 2351
    }

2352 2353 2354 2355
    if (min_workers > max_workers)
        max_workers = min_workers;

    server->nworkers = max_workers;
2356
    if (VIR_ALLOC_N(server->workers, server->nworkers) < 0) {
2357
        VIR_ERROR0(_("Failed to allocate workers"));
2358
        return NULL;
2359 2360
    }

2361 2362 2363 2364
    for (i = 0 ; i < min_workers ; i++) {
        if (qemudStartWorker(server, &server->workers[i]) < 0)
            goto cleanup;
        server->nactiveworkers++;
2365
    }
2366

2367
    for (;!server->quitEventThread;) {
2368 2369 2370 2371
        /* A shutdown timeout is specified, so check
         * if any drivers have active state, if not
         * shutdown after timeout seconds
         */
2372 2373 2374
        if (timeout > 0) {
            if (timerActive) {
                if (server->clients) {
2375
                    VIR_DEBUG("Deactivating shutdown timer %d", timerid);
2376
                    virEventUpdateTimeout(timerid, -1);
2377 2378 2379 2380 2381
                    timerActive = 0;
                }
            } else {
                if (!virStateActive() &&
                    !server->clients) {
2382
                    VIR_DEBUG("Activating shutdown timer %d", timerid);
2383
                    virEventUpdateTimeout(timerid, timeout * 1000);
2384 2385 2386
                    timerActive = 1;
                }
            }
2387 2388
        }

2389
        virMutexUnlock(&server->lock);
2390 2391
        if (qemudOneLoop() < 0) {
            virMutexLock(&server->lock);
2392
            VIR_DEBUG0("Loop iteration error, exiting");
2393
            break;
2394
        }
2395
        virMutexLock(&server->lock);
2396 2397 2398 2399

    reprocess:
        for (i = 0 ; i < server->nclients ; i++) {
            int inactive;
2400
            virMutexLock(&server->clients[i]->lock);
2401 2402
            inactive = server->clients[i]->fd == -1
                && server->clients[i]->refs == 0;
2403
            virMutexUnlock(&server->clients[i]->lock);
2404
            if (inactive) {
2405
                qemudFreeClient(server->clients[i]);
2406
                server->nclients--;
2407
                if (i < server->nclients)
2408 2409
                    memmove(server->clients + i,
                            server->clients + i + 1,
2410 2411
                            sizeof (*server->clients) * (server->nclients - i));

2412 2413
                VIR_SHRINK_N(server->clients, server->nclients_max,
                             server->nclients_max - server->nclients);
2414
                goto reprocess;
2415 2416
            }
        }
D
Daniel P. Berrange 已提交
2417

2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436
        /* If number of active workers exceeds both the min_workers
         * threshold and the number of clients, then kill some
         * off */
        for (i = 0 ; (i < server->nworkers &&
                      server->nactiveworkers > server->nclients &&
                      server->nactiveworkers > min_workers) ; i++) {

            if (server->workers[i].hasThread &&
                !server->workers[i].processingCall) {
                server->workers[i].quitRequest = 1;

                virCondBroadcast(&server->job);
                virMutexUnlock(&server->lock);
                pthread_join(server->workers[i].thread, NULL);
                virMutexLock(&server->lock);
                server->workers[i].hasThread = 0;
                server->nactiveworkers--;
            }
        }
2437 2438
    }

2439
cleanup:
2440
    for (i = 0 ; i < server->nworkers ; i++) {
2441 2442 2443 2444 2445 2446
        if (!server->workers[i].hasThread)
            continue;

        server->workers[i].quitRequest = 1;
        virCondBroadcast(&server->job);

2447
        virMutexUnlock(&server->lock);
2448
        pthread_join(server->workers[i].thread, NULL);
2449
        virMutexLock(&server->lock);
2450
        server->workers[i].hasThread = 0;
2451
    }
2452
    VIR_FREE(server->workers);
2453 2454 2455 2456
    for (i = 0; i < server->nclients; i++)
        qemudFreeClient(server->clients[i]);
    server->nclients = 0;
    VIR_SHRINK_N(server->clients, server->nclients_max, server->nclients_max);
2457

2458
    virMutexUnlock(&server->lock);
2459
    return NULL;
D
Daniel P. Berrange 已提交
2460 2461
}

2462

E
Eric Blake 已提交
2463 2464 2465
static int
qemudStartEventLoop(struct qemud_server *server)
{
2466
    pthread_attr_t attr;
E
Eric Blake 已提交
2467 2468 2469 2470
    int ret = -1;

    if (pthread_attr_init(&attr) != 0)
        return -1;
2471 2472 2473 2474 2475 2476 2477
    /* We want to join the eventloop, so don't detach it */
    /*pthread_attr_setdetachstate(&attr, 1);*/

    if (pthread_create(&server->eventThread,
                       &attr,
                       qemudRunLoop,
                       server) != 0)
E
Eric Blake 已提交
2478
        goto cleanup;
2479 2480 2481

    server->hasEventThread = 1;

E
Eric Blake 已提交
2482 2483 2484 2485
    ret = 0;
cleanup:
    pthread_attr_destroy(&attr);
    return ret;
2486 2487 2488
}


D
Daniel P. Berrange 已提交
2489
static void qemudCleanup(struct qemud_server *server) {
2490 2491
    struct qemud_socket *sock;

2492 2493
    VIR_FORCE_CLOSE(server->sigread);
    VIR_FORCE_CLOSE(server->sigwrite);
2494 2495

    sock = server->sockets;
D
Daniel P. Berrange 已提交
2496
    while (sock) {
2497
        struct qemud_socket *next = sock->next;
2498
        if (sock->watch)
2499
            virEventRemoveHandle(sock->watch);
2500
        VIR_FORCE_CLOSE(sock->fd);
2501 2502 2503 2504 2505 2506 2507

        /* Unlink unix domain sockets which are not in
         * the abstract namespace */
        if (sock->path &&
            sock->path[0] != '@')
            unlink(sock->path);
        VIR_FREE(sock->path);
2508
        VIR_FREE(sock->addrstr);
2509

2510
        VIR_FREE(sock);
2511
        sock = next;
D
Daniel P. Berrange 已提交
2512
    }
2513
    VIR_FREE(server->logDir);
2514

2515
#ifdef HAVE_SASL
2516 2517 2518
    if (server->saslUsernameWhitelist) {
        char **list = server->saslUsernameWhitelist;
        while (*list) {
2519
            VIR_FREE(*list);
2520 2521
            list++;
        }
2522
        VIR_FREE(server->saslUsernameWhitelist);
2523
    }
2524
#endif
2525

2526 2527 2528 2529 2530
#if HAVE_POLKIT0
        if (server->sysbus)
            dbus_connection_unref(server->sysbus);
#endif

2531
    virStateCleanup();
2532

2533 2534 2535 2536 2537 2538
    if (virCondDestroy(&server->job) < 0) {
        ;
    }
    virMutexDestroy(&server->lock);

    VIR_FREE(server);
D
Daniel P. Berrange 已提交
2539 2540
}

2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556
/* Allocate an array of malloc'd strings from the config file, filename
 * (used only in diagnostics), using handle "conf".  Upon error, return -1
 * and free any allocated memory.  Otherwise, save the array in *list_arg
 * and return 0.
 */
static int
remoteConfigGetStringList(virConfPtr conf, const char *key, char ***list_arg,
                          const char *filename)
{
    char **list;
    virConfValuePtr p = virConfGetValue (conf, key);
    if (!p)
        return 0;

    switch (p->type) {
    case VIR_CONF_STRING:
2557
        if (VIR_ALLOC_N(list, 2) < 0) {
2558
            VIR_ERROR(_("failed to allocate memory for %s config list"), key);
2559 2560 2561 2562 2563
            return -1;
        }
        list[0] = strdup (p->str);
        list[1] = NULL;
        if (list[0] == NULL) {
2564
            VIR_ERROR(_("failed to allocate memory for %s config list value"),
2565
                      key);
2566
            VIR_FREE(list);
2567 2568 2569 2570 2571 2572 2573 2574 2575
            return -1;
        }
        break;

    case VIR_CONF_LIST: {
        int i, len = 0;
        virConfValuePtr pp;
        for (pp = p->list; pp; pp = pp->next)
            len++;
2576
        if (VIR_ALLOC_N(list, 1+len) < 0) {
2577
            VIR_ERROR(_("failed to allocate memory for %s config list"), key);
2578 2579 2580 2581
            return -1;
        }
        for (i = 0, pp = p->list; pp; ++i, pp = pp->next) {
            if (pp->type != VIR_CONF_STRING) {
2582
                VIR_ERROR(_("remoteReadConfigFile: %s: %s:"
2583
                            " must be a string or list of strings"),
2584
                          filename, key);
2585
                VIR_FREE(list);
2586 2587 2588 2589 2590 2591
                return -1;
            }
            list[i] = strdup (pp->str);
            if (list[i] == NULL) {
                int j;
                for (j = 0 ; j < i ; j++)
2592 2593
                    VIR_FREE(list[j]);
                VIR_FREE(list);
2594 2595
                VIR_ERROR(_("failed to allocate memory for %s config list value"),
                          key);
2596 2597 2598 2599 2600 2601 2602 2603 2604
                return -1;
            }

        }
        list[i] = NULL;
        break;
    }

    default:
2605
        VIR_ERROR(_("remoteReadConfigFile: %s: %s:"
2606
                    " must be a string or list of strings"),
2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620
                  filename, key);
        return -1;
    }

    *list_arg = list;
    return 0;
}

/* A helper function used by each of the following macros.  */
static int
checkType (virConfValuePtr p, const char *filename,
           const char *key, virConfType required_type)
{
    if (p->type != required_type) {
2621
        VIR_ERROR(_("remoteReadConfigFile: %s: %s: invalid type:"
2622
                    " got %s; expected %s"), filename, key,
2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641
                  virConfTypeName (p->type),
                  virConfTypeName (required_type));
        return -1;
    }
    return 0;
}

/* If there is no config data for the key, #var_name, then do nothing.
   If there is valid data of type VIR_CONF_STRING, and strdup succeeds,
   store the result in var_name.  Otherwise, (i.e. invalid type, or strdup
   failure), give a diagnostic and "goto" the cleanup-and-fail label.  */
#define GET_CONF_STR(conf, filename, var_name)                          \
    do {                                                                \
        virConfValuePtr p = virConfGetValue (conf, #var_name);          \
        if (p) {                                                        \
            if (checkType (p, filename, #var_name, VIR_CONF_STRING) < 0) \
                goto free_and_fail;                                     \
            (var_name) = strdup (p->str);                               \
            if ((var_name) == NULL) {                                   \
2642
                char ebuf[1024];                                        \
2643
                VIR_ERROR(_("remoteReadConfigFile: %s"),		\
2644
                          virStrerror(errno, ebuf, sizeof ebuf));       \
2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660
                goto free_and_fail;                                     \
            }                                                           \
        }                                                               \
    } while (0)

/* Like GET_CONF_STR, but for integral values.  */
#define GET_CONF_INT(conf, filename, var_name)                          \
    do {                                                                \
        virConfValuePtr p = virConfGetValue (conf, #var_name);          \
        if (p) {                                                        \
            if (checkType (p, filename, #var_name, VIR_CONF_LONG) < 0)  \
                goto free_and_fail;                                     \
            (var_name) = p->l;                                          \
        }                                                               \
    } while (0)

2661 2662 2663 2664 2665 2666 2667 2668

static int remoteConfigGetAuth(virConfPtr conf, const char *key, int *auth, const char *filename) {
    virConfValuePtr p;

    p = virConfGetValue (conf, key);
    if (!p)
        return 0;

2669
    if (checkType (p, filename, key, VIR_CONF_STRING) < 0)
2670 2671 2672 2673 2674 2675 2676 2677 2678 2679
        return -1;

    if (!p->str)
        return 0;

    if (STREQ(p->str, "none")) {
        *auth = REMOTE_AUTH_NONE;
#if HAVE_SASL
    } else if (STREQ(p->str, "sasl")) {
        *auth = REMOTE_AUTH_SASL;
2680 2681 2682 2683
#endif
#if HAVE_POLKIT
    } else if (STREQ(p->str, "polkit")) {
        *auth = REMOTE_AUTH_POLKIT;
2684 2685
#endif
    } else {
2686
        VIR_ERROR(_("remoteReadConfigFile: %s: %s: unsupported auth %s"),
2687
                  filename, key, p->str);
2688 2689 2690 2691 2692 2693
        return -1;
    }

    return 0;
}

2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713
#ifdef HAVE_SASL
static inline int
remoteReadSaslAllowedUsernameList (virConfPtr conf,
                                   struct qemud_server *server,
                                   const char *filename)
{
    return
        remoteConfigGetStringList (conf, "sasl_allowed_username_list",
                                   &server->saslUsernameWhitelist, filename);
}
#else
static inline int
remoteReadSaslAllowedUsernameList (virConfPtr conf ATTRIBUTE_UNUSED,
                                   struct qemud_server *server ATTRIBUTE_UNUSED,
                                   const char *filename ATTRIBUTE_UNUSED)
{
    return 0;
}
#endif

2714 2715
/*
 * Set up the logging environment
2716 2717
 * By default if daemonized all errors go to the logfile libvirtd.log,
 * but if verbose or error debugging is asked for then also output
2718
 * informational and debug messages. Default size if 64 kB.
2719
 */
2720
static int
2721 2722
qemudSetLogging(struct qemud_server *server, virConfPtr conf,
                const char *filename)
2723 2724
{
    int log_level = 0;
2725
    int log_buffer_size = 64;
2726 2727
    char *log_filters = NULL;
    char *log_outputs = NULL;
2728
    char *log_file = NULL;
2729 2730
    int ret = -1;

2731 2732 2733
    GET_CONF_INT (conf, filename, log_buffer_size);
    virLogSetBufferSize(log_buffer_size);

2734 2735
    virLogReset();

2736
    /*
2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748
     * 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.
2749
     */
2750
    GET_CONF_INT (conf, filename, log_level);
2751 2752 2753
    if (log_level != 0)
        virLogSetDefaultPriority(log_level);

2754
    virLogSetFromEnv();
2755 2756 2757 2758

    if (virLogGetNbFilters() == 0) {
        GET_CONF_STR (conf, filename, log_filters);
        virLogParseFilters(log_filters);
2759
    }
2760

2761 2762 2763
    if (virLogGetNbOutputs() == 0) {
        GET_CONF_STR (conf, filename, log_outputs);
        virLogParseOutputs(log_outputs);
2764
    }
2765 2766

    /*
2767
     * If no defined outputs, then direct to libvirtd.log when running
2768
     * as daemon. Otherwise the default output is stderr.
2769
     */
2770
    if (virLogGetNbOutputs() == 0) {
2771
        char *tmp = NULL;
2772

2773
        if (godaemon) {
2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787
            if (server->privileged) {
                if (virAsprintf(&tmp, "%d:file:%s/log/libvirt/libvirtd.log",
                                virLogGetDefaultPriority(),
                                LOCALSTATEDIR) == -1)
                    goto out_of_memory;
            } else {
                char *userdir = virGetUserDirectory(geteuid());
                if (!userdir)
                    goto free_and_fail;

                if (virAsprintf(&tmp, "%d:file:%s/.libvirt/libvirtd.log",
                                virLogGetDefaultPriority(), userdir) == -1)
                    goto out_of_memory;
            }
2788
        } else {
2789 2790
            if (virAsprintf(&tmp, "%d:stderr", virLogGetDefaultPriority()) < 0)
                goto out_of_memory;
2791
        }
2792 2793
        virLogParseOutputs(tmp);
        VIR_FREE(tmp);
2794
    }
2795 2796 2797 2798 2799 2800 2801

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

2802 2803
    ret = 0;

2804 2805 2806
free_and_fail:
    VIR_FREE(log_filters);
    VIR_FREE(log_outputs);
2807
    VIR_FREE(log_file);
2808
    return(ret);
2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821

out_of_memory:
    virReportOOMError();
    goto free_and_fail;
}

/*
 * Stop logging
 */
static void
qemudStopLogging(void)
{
    virLogShutdown();
2822
}
2823

2824 2825 2826 2827
/* Read the config file if it exists.
 * Only used in the remote case, hence the name.
 */
static int
2828
remoteReadConfigFile (struct qemud_server *server, const char *filename)
2829 2830 2831
{
    virConfPtr conf;

2832 2833 2834 2835 2836
    /* The following variable names must match the corresponding
       configuration strings.  */
    char *unix_sock_ro_perms = NULL;
    char *unix_sock_rw_perms = NULL;
    char *unix_sock_group = NULL;
2837
    char *buf = NULL;
2838
    char *host_uuid = NULL;
2839

2840 2841
#if HAVE_POLKIT
    /* Change the default back to no auth for non-root */
2842
    if (!server->privileged && auth_unix_rw == REMOTE_AUTH_POLKIT)
2843
        auth_unix_rw = REMOTE_AUTH_NONE;
2844
    if (!server->privileged && auth_unix_ro == REMOTE_AUTH_POLKIT)
2845 2846 2847
        auth_unix_ro = REMOTE_AUTH_NONE;
#endif

2848
    conf = virConfReadFile (filename, 0);
2849
    if (!conf) return -1;
2850

2851 2852 2853
    /*
     * First get all the logging settings and activate them
     */
2854
    if (qemudSetLogging(server, conf, filename) < 0)
2855
        goto free_and_fail;
2856

2857 2858
    GET_CONF_INT (conf, filename, listen_tcp);
    GET_CONF_INT (conf, filename, listen_tls);
2859 2860
    GET_CONF_STR (conf, filename, tls_port);
    GET_CONF_STR (conf, filename, tcp_port);
2861
    GET_CONF_STR (conf, filename, listen_addr);
J
Jim Meyering 已提交
2862

2863
    if (remoteConfigGetAuth(conf, "auth_unix_rw", &auth_unix_rw, filename) < 0)
J
Jim Meyering 已提交
2864
        goto free_and_fail;
2865 2866 2867 2868 2869 2870 2871
#if HAVE_POLKIT
    /* Change default perms to be wide-open if PolicyKit is enabled.
     * Admin can always override in config file
     */
    if (auth_unix_rw == REMOTE_AUTH_POLKIT)
        unix_sock_rw_mask = 0777;
#endif
2872
    if (remoteConfigGetAuth(conf, "auth_unix_ro", &auth_unix_ro, filename) < 0)
J
Jim Meyering 已提交
2873
        goto free_and_fail;
2874
    if (remoteConfigGetAuth(conf, "auth_tcp", &auth_tcp, filename) < 0)
J
Jim Meyering 已提交
2875
        goto free_and_fail;
2876
    if (remoteConfigGetAuth(conf, "auth_tls", &auth_tls, filename) < 0)
J
Jim Meyering 已提交
2877
        goto free_and_fail;
2878

2879 2880
    GET_CONF_STR (conf, filename, unix_sock_group);
    if (unix_sock_group) {
2881
        if (!server->privileged) {
2882
            VIR_WARN0("Cannot set group when not running as root");
2883
        } else {
2884
            int ret;
2885
            struct group grpdata, *grp;
2886 2887 2888 2889 2890 2891
            size_t maxbuf = sysconf(_SC_GETGR_R_SIZE_MAX);

            if (maxbuf == -1)
                maxbuf = 1024;

            if (VIR_ALLOC_N(buf, maxbuf) < 0) {
2892
                VIR_ERROR0(_("Failed to allocate memory for buffer"));
2893 2894 2895 2896 2897 2898 2899 2900
                goto free_and_fail;
            }

            while ((ret = getgrnam_r(unix_sock_group, &grpdata,
                                     buf, maxbuf,
                                     &grp)) == ERANGE) {
                    maxbuf *= 2;
                    if (maxbuf > 65536 || VIR_REALLOC_N(buf, maxbuf) < 0) {
2901
                        VIR_ERROR0(_("Failed to reallocate enough memory for buffer"));
2902 2903 2904 2905 2906
                        goto free_and_fail;
                    }
            }

            if (ret != 0 || !grp) {
2907
                VIR_ERROR(_("Failed to lookup group '%s'"), unix_sock_group);
2908
                goto free_and_fail;
2909 2910
            }
            unix_sock_gid = grp->gr_gid;
2911
            VIR_FREE(buf);
2912
        }
2913
        VIR_FREE(unix_sock_group);
2914 2915
    }

2916 2917
    GET_CONF_STR (conf, filename, unix_sock_ro_perms);
    if (unix_sock_ro_perms) {
2918
        if (virStrToLong_i (unix_sock_ro_perms, NULL, 8, &unix_sock_ro_mask) != 0) {
2919
            VIR_ERROR(_("Failed to parse mode '%s'"), unix_sock_ro_perms);
2920
            goto free_and_fail;
2921
        }
2922
        VIR_FREE(unix_sock_ro_perms);
2923 2924
    }

2925 2926
    GET_CONF_STR (conf, filename, unix_sock_rw_perms);
    if (unix_sock_rw_perms) {
2927
        if (virStrToLong_i (unix_sock_rw_perms, NULL, 8, &unix_sock_rw_mask) != 0) {
2928
            VIR_ERROR(_("Failed to parse mode '%s'"), unix_sock_rw_perms);
2929
            goto free_and_fail;
2930
        }
2931
        VIR_FREE(unix_sock_rw_perms);
2932 2933
    }

2934 2935
    GET_CONF_STR (conf, filename, unix_sock_dir);

2936 2937
    GET_CONF_INT (conf, filename, mdns_adv);
    GET_CONF_STR (conf, filename, mdns_name);
2938

2939
    GET_CONF_INT (conf, filename, tls_no_verify_certificate);
2940

2941 2942 2943 2944
    GET_CONF_STR (conf, filename, key_file);
    GET_CONF_STR (conf, filename, cert_file);
    GET_CONF_STR (conf, filename, ca_file);
    GET_CONF_STR (conf, filename, crl_file);
2945

2946 2947 2948
    if (remoteConfigGetStringList (conf, "tls_allowed_dn_list",
                                   &tls_allowed_dn_list, filename) < 0)
        goto free_and_fail;
2949

2950
    if (remoteReadSaslAllowedUsernameList (conf, server, filename) < 0)
2951
        goto free_and_fail;
2952

2953 2954 2955 2956 2957

    GET_CONF_INT (conf, filename, min_workers);
    GET_CONF_INT (conf, filename, max_workers);
    GET_CONF_INT (conf, filename, max_clients);

2958 2959 2960
    GET_CONF_INT (conf, filename, max_requests);
    GET_CONF_INT (conf, filename, max_client_requests);

2961 2962 2963
    GET_CONF_INT (conf, filename, audit_level);
    GET_CONF_INT (conf, filename, audit_logging);

2964
    GET_CONF_STR (conf, filename, host_uuid);
2965 2966
    if (virSetHostUUIDStr(host_uuid)) {
        VIR_ERROR(_("invalid host UUID: %s"), host_uuid);
2967
        goto free_and_fail;
2968
    }
2969 2970 2971

    VIR_FREE(host_uuid);

2972 2973
    virConfFree (conf);
    return 0;
2974

2975 2976
 free_and_fail:
    virConfFree (conf);
2977
    VIR_FREE(host_uuid);
2978 2979 2980 2981 2982
    VIR_FREE(mdns_name);
    VIR_FREE(unix_sock_ro_perms);
    VIR_FREE(unix_sock_rw_perms);
    VIR_FREE(unix_sock_group);
    VIR_FREE(buf);
2983

2984 2985 2986 2987
    /* Don't bother trying to free listen_addr, tcp_port, tls_port, key_file,
       cert_file, ca_file, or crl_file, since they are initialized to
       non-malloc'd strings.  Besides, these are static variables, and callers
       are unlikely to call this function more than once, so there wouldn't
2988 2989 2990 2991 2992
       even be a real leak.  */

    if (tls_allowed_dn_list) {
        int i;
        for (i = 0; tls_allowed_dn_list[i]; i++)
2993 2994
            VIR_FREE(tls_allowed_dn_list[i]);
        VIR_FREE(tls_allowed_dn_list);
2995 2996 2997
    }

    return -1;
2998 2999
}

3000 3001
/* Display version information. */
static void
3002
version (void)
3003 3004 3005 3006
{
    printf ("%s (%s) %s\n", argv0, PACKAGE_NAME, PACKAGE_VERSION);
}

J
John Levon 已提交
3007 3008 3009 3010 3011 3012 3013 3014
#ifdef __sun
static int
qemudSetupPrivs (void)
{
    chown ("/var/run/libvirt", SYSTEM_UID, SYSTEM_UID);

    if (__init_daemon_priv (PU_RESETGROUPS | PU_CLEARLIMITSET,
        SYSTEM_UID, SYSTEM_UID, PRIV_XVM_CONTROL, NULL)) {
3015
        VIR_ERROR0(_("additional privileges are required"));
J
John Levon 已提交
3016 3017 3018 3019 3020
        return -1;
    }

    if (priv_set (PRIV_OFF, PRIV_ALLSETS, PRIV_FILE_LINK_ANY, PRIV_PROC_INFO,
        PRIV_PROC_SESSION, PRIV_PROC_EXEC, PRIV_PROC_FORK, NULL)) {
3021
        VIR_ERROR0(_("failed to set reduced privileges"));
J
John Levon 已提交
3022 3023 3024 3025 3026 3027
        return -1;
    }

    return 0;
}
#else
3028
# define qemudSetupPrivs() 0
J
John Levon 已提交
3029 3030
#endif

3031 3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067

/*
 * Doing anything non-trivial in signal handlers is pretty dangerous,
 * since there are very few async-signal safe POSIX funtions. To
 * deal with this we setup a very simple signal handler. It simply
 * writes the signal number to a pipe. The main event loop then sees
 * the signal on the pipe and can safely do the processing from
 * event loop context
 */
static int
daemonSetupSignals(struct qemud_server *server)
{
    struct sigaction sig_action;
    int sigpipe[2];

    if (pipe(sigpipe) < 0)
        return -1;

    if (virSetNonBlock(sigpipe[0]) < 0 ||
        virSetNonBlock(sigpipe[1]) < 0 ||
        virSetCloseExec(sigpipe[0]) < 0 ||
        virSetCloseExec(sigpipe[1]) < 0) {
        char ebuf[1024];
        VIR_ERROR(_("Failed to create pipe: %s"),
                  virStrerror(errno, ebuf, sizeof ebuf));
        goto error;
    }

    sig_action.sa_sigaction = sig_handler;
    sig_action.sa_flags = SA_SIGINFO;
    sigemptyset(&sig_action.sa_mask);

    sigaction(SIGHUP, &sig_action, NULL);
    sigaction(SIGINT, &sig_action, NULL);
    sigaction(SIGQUIT, &sig_action, NULL);
    sigaction(SIGTERM, &sig_action, NULL);

3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079
    /*
     * catch fatal errors to dump a log, also hook to USR2 for dynamic
     * debugging purposes or testing
     */
    sig_action.sa_sigaction = sig_fatal;
    sigaction(SIGFPE, &sig_action, NULL);
    sigaction(SIGSEGV, &sig_action, NULL);
    sigaction(SIGILL, &sig_action, NULL);
    sigaction(SIGABRT, &sig_action, NULL);
    sigaction(SIGBUS, &sig_action, NULL);
    sigaction(SIGUSR2, &sig_action, NULL);

3080 3081 3082
    sig_action.sa_handler = SIG_IGN;
    sigaction(SIGPIPE, &sig_action, NULL);

3083 3084 3085 3086
    if (virEventAddHandle(sigpipe[0],
                          VIR_EVENT_HANDLE_READABLE,
                          qemudDispatchSignalEvent,
                          server, NULL) < 0) {
3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097
        VIR_ERROR0(_("Failed to register callback for signal pipe"));
        goto error;
    }

    server->sigread = sigpipe[0];
    server->sigwrite = sigpipe[1];
    sigwrite = sigpipe[1];

    return 0;

error:
3098 3099
    VIR_FORCE_CLOSE(sigpipe[0]);
    VIR_FORCE_CLOSE(sigpipe[1]);
3100 3101 3102
    return -1;
}

3103 3104
/* Print command-line usage. */
static void
3105
usage (void)
3106 3107
{
    fprintf (stderr,
3108
             _("\n\
3109 3110 3111 3112 3113 3114
Usage:\n\
  %s [options]\n\
\n\
Options:\n\
  -v | --verbose         Verbose messages.\n\
  -d | --daemon          Run as a daemon & write PID file.\n\
3115 3116
  -l | --listen          Listen for TCP/IP connections.\n\
  -t | --timeout <secs>  Exit after timeout period.\n\
3117
  -f | --config <file>   Configuration file.\n\
3118
     | --version         Display version information.\n\
3119 3120
  -p | --pid-file <file> Change name of PID file.\n\
\n\
3121
libvirt management daemon:\n\
3122 3123 3124
\n\
  Default paths:\n\
\n\
W
Wolfgang Mauerer 已提交
3125
    Configuration file (unless overridden by -f):\n\
3126
      %s/libvirt/libvirtd.conf\n\
3127
\n\
3128
    Sockets (as root):\n\
3129 3130
      %s/run/libvirt/libvirt-sock\n\
      %s/run/libvirt/libvirt-sock-ro\n\
3131 3132 3133
\n\
    Sockets (as non-root):\n\
      $HOME/.libvirt/libvirt-sock (in UNIX abstract namespace)\n\
3134 3135
\n\
    TLS:\n\
3136 3137 3138
      CA certificate:     %s\n\
      Server certificate: %s\n\
      Server private key: %s\n\
3139 3140 3141
\n\
    PID file (unless overridden by --pid-file):\n\
      %s\n\
3142 3143
\n"),
               argv0,
3144 3145 3146
               SYSCONFDIR,
               LOCALSTATEDIR,
               LOCALSTATEDIR,
3147 3148 3149 3150 3151 3152
               LIBVIRT_CACERT,
               LIBVIRT_SERVERCERT,
               LIBVIRT_SERVERKEY,
               (REMOTE_PID_FILE[0] != '\0'
                ? REMOTE_PID_FILE
                : _("(disabled in ./configure)")));
3153 3154
}

3155 3156 3157 3158
enum {
    OPT_VERSION = 129
};

D
Daniel P. Berrange 已提交
3159 3160
#define MAX_LISTEN 5
int main(int argc, char **argv) {
3161
    struct qemud_server *server = NULL;
3162
    const char *pid_file = NULL;
3163
    const char *remote_config_file = NULL;
3164
    int statuswrite = -1;
3165
    int ret = 1;
3166
    argv0 = argv[0];
D
Daniel P. Berrange 已提交
3167 3168 3169 3170

    struct option opts[] = {
        { "verbose", no_argument, &verbose, 1},
        { "daemon", no_argument, &godaemon, 1},
3171
        { "listen", no_argument, &ipsock, 1},
3172
        { "config", required_argument, NULL, 'f'},
3173 3174
        { "timeout", required_argument, NULL, 't'},
        { "pid-file", required_argument, NULL, 'p'},
3175
        { "version", no_argument, NULL, OPT_VERSION },
3176
        { "help", no_argument, NULL, '?' },
D
Daniel P. Berrange 已提交
3177 3178 3179
        {0, 0, 0, 0}
    };

E
Eric Blake 已提交
3180 3181 3182 3183 3184 3185
    if (setlocale (LC_ALL, "") == NULL ||
        bindtextdomain (PACKAGE, LOCALEDIR) == NULL ||
        textdomain(PACKAGE) == NULL ||
        virInitialize() < 0) {
        fprintf(stderr, _("%s: initialization failed\n"), argv0);
        exit(EXIT_FAILURE);
3186
    }
D
Daniel P. Berrange 已提交
3187

D
Daniel P. Berrange 已提交
3188 3189 3190 3191 3192
    while (1) {
        int optidx = 0;
        int c;
        char *tmp;

3193
        c = getopt_long(argc, argv, "ldf:p:t:v", opts, &optidx);
D
Daniel P. Berrange 已提交
3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204 3205 3206 3207 3208

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

        switch (c) {
        case 0:
            /* Got one of the flags */
            break;
        case 'v':
            verbose = 1;
            break;
        case 'd':
            godaemon = 1;
            break;
3209 3210
        case 'l':
            ipsock = 1;
D
Daniel P. Berrange 已提交
3211 3212 3213
            break;

        case 't':
3214
            if (virStrToLong_i(optarg, &tmp, 10, &timeout) != 0
3215 3216 3217
                || timeout <= 0
                /* Ensure that we can multiply by 1000 without overflowing.  */
                || timeout > INT_MAX / 1000)
D
Daniel P. Berrange 已提交
3218 3219
                timeout = -1;
            break;
3220 3221

        case 'p':
3222 3223 3224 3225 3226
            pid_file = optarg;
            break;

        case 'f':
            remote_config_file = optarg;
3227 3228
            break;

3229
        case OPT_VERSION:
3230
            version ();
3231 3232
            return 0;

D
Daniel P. Berrange 已提交
3233
        case '?':
3234
            usage ();
D
Daniel P. Berrange 已提交
3235 3236 3237
            return 2;

        default:
3238 3239
            fprintf (stderr, _("%s: internal error: unknown flag: %c\n"),
                     argv0, c);
3240
            exit (EXIT_FAILURE);
D
Daniel P. Berrange 已提交
3241 3242 3243
        }
    }

3244 3245
    if (remote_config_file == NULL) {
        static const char *default_config_file
3246
            = SYSCONFDIR "/libvirt/libvirtd.conf";
3247
        remote_config_file =
3248
            (access(default_config_file, R_OK) == 0
3249 3250 3251 3252
             ? default_config_file
             : "/dev/null");
    }

3253
    if (godaemon) {
3254
        char ebuf[1024];
3255 3256 3257 3258 3259 3260 3261

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

3262
        if ((statuswrite = daemonForkIntoBackground()) < 0) {
3263 3264
            VIR_ERROR(_("Failed to fork as daemon: %s"),
                      virStrerror(errno, ebuf, sizeof ebuf));
3265
            goto error;
3266 3267 3268 3269 3270
        }
    }

    /* If running as root and no PID file is set, use the default */
    if (pid_file == NULL &&
3271
        geteuid() == 0 &&
3272 3273 3274 3275 3276
        REMOTE_PID_FILE[0] != '\0')
        pid_file = REMOTE_PID_FILE;

    /* If we have a pidfile set, claim it now, exiting if already taken */
    if (pid_file != NULL &&
3277 3278 3279
        qemudWritePidFile (pid_file) < 0) {
        pid_file = NULL; /* Prevent unlinking of someone else's pid ! */
        ret = VIR_DAEMON_ERR_PIDFILE;
3280
        goto error;
3281
    }
3282

J
John Levon 已提交
3283
    /* Ensure the rundir exists (on tmpfs on some systems) */
3284
    if (geteuid() == 0) {
3285
        const char *rundir = LOCALSTATEDIR "/run/libvirt";
3286
        mode_t old_umask;
J
John Levon 已提交
3287

3288
        old_umask = umask(022);
J
John Levon 已提交
3289 3290
        if (mkdir (rundir, 0755)) {
            if (errno != EEXIST) {
3291 3292 3293
                char ebuf[1024];
                VIR_ERROR(_("unable to create rundir %s: %s"), rundir,
                          virStrerror(errno, ebuf, sizeof(ebuf)));
3294
                ret = VIR_DAEMON_ERR_RUNDIR;
3295
                umask(old_umask);
3296
                goto error;
J
John Levon 已提交
3297 3298
            }
        }
3299
        umask(old_umask);
J
John Levon 已提交
3300 3301
    }

3302 3303 3304 3305 3306 3307
    /* Beyond this point, nothing should rely on using
     * getuid/geteuid() == 0, for privilege level checks.
     * It must all use the flag 'server->privileged'
     * which is also passed into all libvirt stateful
     * drivers
     */
3308 3309
    if (qemudSetupPrivs() < 0) {
        ret = VIR_DAEMON_ERR_PRIVS;
3310
        goto error;
3311
    }
J
John Levon 已提交
3312

3313
    if (!(server = qemudInitialize())) {
3314
        ret = VIR_DAEMON_ERR_INIT;
3315
        goto error;
3316
    }
3317

3318 3319
    if ((daemonSetupSignals(server)) < 0) {
        ret = VIR_DAEMON_ERR_SIGNAL;
3320
        goto error;
3321
    }
3322

3323
    /* Read the config file (if it exists). */
3324 3325
    if (remoteReadConfigFile (server, remote_config_file) < 0) {
        ret = VIR_DAEMON_ERR_CONFIG;
3326
        goto error;
3327
    }
D
Daniel P. Berrange 已提交
3328

3329 3330 3331 3332 3333 3334 3335 3336 3337 3338
    if (audit_level) {
        if (virAuditOpen() < 0) {
            if (audit_level > 1) {
                ret = VIR_DAEMON_ERR_AUDIT;
                goto error;
            }
        }
    }
    virAuditLog(audit_logging);

3339
    /* setup the hooks if any */
3340
    if (virHookInitialize() < 0) {
3341 3342 3343 3344
        ret = VIR_DAEMON_ERR_HOOKS;
        goto error;
    }

3345 3346
    /* Disable error func, now logging is setup */
    virSetErrorFunc(NULL, virshErrorHandler);
3347
    virSetErrorLogPriorityFunc(daemonErrorLogFilter);
3348

3349 3350 3351 3352 3353 3354 3355 3356
    /*
     * 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,
                0, "start", NULL);

3357
    if (qemudNetworkInit(server) < 0) {
3358
        ret = VIR_DAEMON_ERR_NETWORK;
3359
        goto error;
3360 3361
    }

3362 3363 3364 3365 3366 3367 3368 3369 3370
    /* 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)
            ;
3371
        VIR_FORCE_CLOSE(statuswrite);
3372 3373
    }

3374 3375 3376
    /* Start the event loop in a background thread, since
     * state initialization needs events to be being processed */
    if (qemudStartEventLoop(server) < 0) {
3377
        VIR_ERROR0(_("Event thread startup failed"));
3378 3379 3380
        goto error;
    }

3381 3382 3383 3384 3385
    /* Start the stateful HV drivers
     * This is delibrately done after telling the parent process
     * we're ready, since it can take a long time and this will
     * seriously delay OS bootup process */
    if (virStateInitialize(server->privileged) < 0) {
3386
        VIR_ERROR0(_("Driver state initialization failed"));
3387
        goto shutdown;
3388
    }
D
Daniel P. Berrange 已提交
3389

3390 3391 3392
    /* Start accepting new clients from network */
    virMutexLock(&server->lock);
    if (qemudNetworkEnable(server) < 0) {
3393
        VIR_ERROR0(_("Network event loop enablement failed"));
3394 3395 3396 3397
        goto shutdown;
    }
    virMutexUnlock(&server->lock);

3398 3399
    ret = 0;

3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415
shutdown:
    /* In a non-0 shutdown scenario we need to tell event loop
     * to quit immediately. Otherwise in normal case we just
     * sit in the thread join forever. Sure this means the
     * main thread doesn't do anything useful ever, but that's
     * not too much of drain on resources
     */
    if (ret != 0) {
        virMutexLock(&server->lock);
        if (server->hasEventThread)
            /* This SIGQUIT triggers the shutdown process */
            kill(getpid(), SIGQUIT);
        virMutexUnlock(&server->lock);
    }
    pthread_join(server->eventThread, NULL);

3416 3417 3418
    virHookCall(VIR_HOOK_DRIVER_DAEMON, "-", VIR_HOOK_DAEMON_OP_SHUTDOWN,
                0, "shutdown", NULL);

3419
error:
3420 3421 3422 3423 3424 3425 3426 3427
    if (statuswrite != -1) {
        if (ret != 0) {
            /* Tell parent of daemon what failed */
            char status = ret;
            while (write(statuswrite, &status, 1) == -1 &&
                   errno == EINTR)
                ;
        }
3428
        VIR_FORCE_CLOSE(statuswrite);
3429
    }
3430 3431
    if (server)
        qemudCleanup(server);
3432 3433
    if (pid_file)
        unlink (pid_file);
3434
    qemudStopLogging();
3435
    return ret;
D
Daniel P. Berrange 已提交
3436
}