qemu_driver.c 123.9 KB
Newer Older
D
Daniel P. Berrange 已提交
1 2 3
/*
 * driver.c: core driver methods for managing qemu guests
 *
4
 * Copyright (C) 2006, 2007, 2008 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 36
#include <sys/types.h>
#include <sys/poll.h>
#include <dirent.h>
#include <limits.h>
#include <string.h>
#include <stdio.h>
#include <strings.h>
#include <stdarg.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
37
#include <sys/utsname.h>
38 39 40 41
#include <sys/stat.h>
#include <fcntl.h>
#include <signal.h>
#include <paths.h>
42 43
#include <pwd.h>
#include <stdio.h>
44
#include <sys/wait.h>
D
Daniel P. Berrange 已提交
45

46 47 48 49
#if HAVE_NUMACTL
#include <numa.h>
#endif

50 51 52 53
#if HAVE_SCHED_H
#include <sched.h>
#endif

54 55
#include "qemu_driver.h"
#include "qemu_conf.h"
J
Jim Meyering 已提交
56
#include "c-ctype.h"
57
#include "event.h"
58
#include "buf.h"
59
#include "util.h"
60
#include "nodeinfo.h"
61
#include "stats_linux.h"
62
#include "capabilities.h"
63
#include "memory.h"
64
#include "uuid.h"
65

R
Richard W.M. Jones 已提交
66 67 68
/* For storing short-lived temporary files. */
#define TEMPDIR LOCAL_STATE_DIR "/cache/libvirt"

A
Atsushi SAKAI 已提交
69
#ifdef WITH_LIBVIRTD
70
static int qemudShutdown(void);
A
Atsushi SAKAI 已提交
71
#endif
72

73 74 75
/* qemudDebug statements should be changed to use this macro instead. */
#define DEBUG(fmt,...) VIR_DEBUG(__FILE__, fmt, __VA_ARGS__)
#define DEBUG0(msg) VIR_DEBUG(__FILE__, "%s", msg)
76

77 78
#define qemudLog(level, msg...) fprintf(stderr, msg)

79 80 81 82 83 84 85 86 87
static int qemudSetCloseExec(int fd) {
    int flags;
    if ((flags = fcntl(fd, F_GETFD)) < 0)
        goto error;
    flags |= FD_CLOEXEC;
    if ((fcntl(fd, F_SETFD, flags)) < 0)
        goto error;
    return 0;
 error:
J
Jim Meyering 已提交
88
    qemudLog(QEMUD_ERR,
89
             "%s", _("Failed to set close-on-exec file descriptor flag\n"));
90 91 92 93 94 95 96 97 98 99 100 101 102
    return -1;
}


static int qemudSetNonBlock(int fd) {
    int flags;
    if ((flags = fcntl(fd, F_GETFL)) < 0)
        goto error;
    flags |= O_NONBLOCK;
    if ((fcntl(fd, F_SETFL, flags)) < 0)
        goto error;
    return 0;
 error:
J
Jim Meyering 已提交
103
    qemudLog(QEMUD_ERR,
104
             "%s", _("Failed to set non-blocking file descriptor flag\n"));
105 106 107 108
    return -1;
}


109
static void qemudDispatchVMEvent(int fd, int events, void *opaque);
110 111
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
112 113
                              virDomainObjPtr vm,
                              const char *migrateFrom);
114

115 116
static void qemudShutdownVMDaemon(virConnectPtr conn,
                                  struct qemud_driver *driver,
117
                                  virDomainObjPtr vm);
118

119 120
static int qemudStartNetworkDaemon(virConnectPtr conn,
                                   struct qemud_driver *driver,
121
                                   virNetworkObjPtr network);
122

123 124
static int qemudShutdownNetworkDaemon(virConnectPtr conn,
                                      struct qemud_driver *driver,
125
                                      virNetworkObjPtr network);
126

127
static int qemudDomainGetMaxVcpus(virDomainPtr dom);
128
static int qemudMonitorCommand (const struct qemud_driver *driver,
129
                                const virDomainObjPtr vm,
130 131
                                const char *cmd,
                                char **reply);
132

J
Jim Meyering 已提交
133
static struct qemud_driver *qemu_driver = NULL;
134 135 136 137


static
void qemudAutostartConfigs(struct qemud_driver *driver) {
138
    virNetworkObjPtr network;
139
    virDomainObjPtr vm;
140 141 142

    network = driver->networks;
    while (network != NULL) {
143
        virNetworkObjPtr next = network->next;
144 145

        if (network->autostart &&
146
            !virNetworkIsActive(network) &&
147
            qemudStartNetworkDaemon(NULL, driver, network) < 0) {
148
            virErrorPtr err = virGetLastError();
149
            qemudLog(QEMUD_ERR, _("Failed to autostart network '%s': %s\n"),
150 151 152 153 154 155
                     network->def->name, err->message);
        }

        network = next;
    }

156
    vm = driver->domains;
157
    while (vm != NULL) {
158
        virDomainObjPtr next = vm->next;
159 160

        if (vm->autostart &&
161 162
            !virDomainIsActive(vm) &&
            qemudStartVMDaemon(NULL, driver, vm, NULL) < 0) {
163
            virErrorPtr err = virGetLastError();
164
            qemudLog(QEMUD_ERR, _("Failed to autostart VM '%s': %s\n"),
165 166 167 168 169
                     vm->def->name, err->message);
        }

        vm = next;
    }
170 171
}

A
Atsushi SAKAI 已提交
172
#ifdef WITH_LIBVIRTD
173 174 175 176 177 178 179
/**
 * qemudStartup:
 *
 * Initialization function for the QEmu daemon
 */
static int
qemudStartup(void) {
180 181 182
    uid_t uid = geteuid();
    struct passwd *pw;
    char *base = NULL;
D
Daniel P. Berrange 已提交
183
    char driverConf[PATH_MAX];
184

185
    if (VIR_ALLOC(qemu_driver) < 0)
186 187 188 189 190 191
        return -1;

    /* Don't have a dom0 so start from 1 */
    qemu_driver->nextvmid = 1;

    if (!uid) {
192 193 194
        if (asprintf(&qemu_driver->logDir,
                     "%s/log/libvirt/qemu", LOCAL_STATE_DIR) == -1)
            goto out_of_memory;
195

D
Daniel P. Berrange 已提交
196
        if ((base = strdup (SYSCONF_DIR "/libvirt")) == NULL)
197 198 199
            goto out_of_memory;
    } else {
        if (!(pw = getpwuid(uid))) {
200
            qemudLog(QEMUD_ERR, _("Failed to find user record for uid '%d': %s\n"),
201 202 203 204
                     uid, strerror(errno));
            goto out_of_memory;
        }

205 206 207
        if (asprintf(&qemu_driver->logDir,
                     "%s/.libvirt/qemu/log", pw->pw_dir) == -1)
            goto out_of_memory;
208

D
Daniel P. Berrange 已提交
209
        if (asprintf (&base, "%s/.libvirt", pw->pw_dir) == -1) {
J
Jim Meyering 已提交
210
            qemudLog (QEMUD_ERR,
211
                      "%s", _("out of memory in asprintf\n"));
212 213 214 215 216 217 218
            goto out_of_memory;
        }
    }

    /* Configuration paths are either ~/.libvirt/qemu/... (session) or
     * /etc/libvirt/qemu/... (system).
     */
D
Daniel P. Berrange 已提交
219
    if (snprintf (driverConf, sizeof(driverConf), "%s/qemu.conf", base) == -1)
220
        goto out_of_memory;
D
Daniel P. Berrange 已提交
221
    driverConf[sizeof(driverConf)-1] = '\0';
222

D
Daniel P. Berrange 已提交
223
    if (asprintf (&qemu_driver->configDir, "%s/qemu", base) == -1)
224 225
        goto out_of_memory;

D
Daniel P. Berrange 已提交
226
    if (asprintf (&qemu_driver->autostartDir, "%s/qemu/autostart", base) == -1)
227 228
        goto out_of_memory;

D
Daniel P. Berrange 已提交
229 230 231 232
    if (asprintf (&qemu_driver->networkConfigDir, "%s/qemu/networks", base) == -1)
        goto out_of_memory;

    if (asprintf (&qemu_driver->networkAutostartDir, "%s/qemu/networks/autostart",
233 234 235
                  base) == -1)
        goto out_of_memory;

236
    VIR_FREE(base);
237 238 239

    if ((qemu_driver->caps = qemudCapsInit()) == NULL)
        goto out_of_memory;
D
Daniel P. Berrange 已提交
240 241

    if (qemudLoadDriverConfig(qemu_driver, driverConf) < 0) {
242
        qemudShutdown();
D
Daniel P. Berrange 已提交
243 244 245
        return -1;
    }

246 247 248 249 250
    if (virDomainLoadAllConfigs(NULL,
                                qemu_driver->caps,
                                &qemu_driver->domains,
                                qemu_driver->configDir,
                                qemu_driver->autostartDir) < 0) {
D
Daniel P. Berrange 已提交
251 252 253
        qemudShutdown();
        return -1;
    }
254 255 256 257 258 259 260
    if (virNetworkLoadAllConfigs(NULL,
                                 &qemu_driver->networks,
                                 qemu_driver->networkConfigDir,
                                 qemu_driver->networkAutostartDir) < 0) {
        qemudShutdown();
        return -1;
    }
261 262 263 264 265
    qemudAutostartConfigs(qemu_driver);

    return 0;

 out_of_memory:
J
Jim Meyering 已提交
266
    qemudLog (QEMUD_ERR,
267
              "%s", _("qemudStartup: out of memory\n"));
268 269
    VIR_FREE(base);
    VIR_FREE(qemu_driver);
270 271 272
    return -1;
}

273 274 275 276 277 278 279 280
/**
 * qemudReload:
 *
 * Function to restart the QEmu daemon, it will recheck the configuration
 * files and update its state and the networking
 */
static int
qemudReload(void) {
281 282 283 284 285 286
    virDomainLoadAllConfigs(NULL,
                            qemu_driver->caps,
                            &qemu_driver->domains,
                            qemu_driver->configDir,
                            qemu_driver->autostartDir);

287 288 289 290
    virNetworkLoadAllConfigs(NULL,
                             &qemu_driver->networks,
                             qemu_driver->networkConfigDir,
                             qemu_driver->networkAutostartDir);
291 292

     if (qemu_driver->iptables) {
J
Jim Meyering 已提交
293
        qemudLog(QEMUD_INFO,
294
                 "%s", _("Reloading iptables rules\n"));
295
        iptablesReloadRules(qemu_driver->iptables);
296
    }
297 298

    qemudAutostartConfigs(qemu_driver);
299 300

    return 0;
301 302
}

303 304 305 306 307 308 309 310 311 312
/**
 * qemudActive:
 *
 * Checks if the QEmu daemon is active, i.e. has an active domain or
 * an active network
 *
 * Returns 1 if active, 0 otherwise
 */
static int
qemudActive(void) {
313
    virDomainObjPtr dom = qemu_driver->domains;
314
    virNetworkObjPtr net = qemu_driver->networks;
315

316 317 318 319 320 321
    while (dom) {
        if (virDomainIsActive(dom))
            return 1;
        dom = dom->next;
    }

322
    while (net) {
323
        if (virNetworkIsActive(net))
324 325 326 327
            return 1;
        net = net->next;
    }

328 329 330 331
    /* Otherwise we're happy to deal with a shutdown */
    return 0;
}

332 333 334 335 336 337 338
/**
 * qemudShutdown:
 *
 * Shutdown the QEmu daemon, it will stop all active domains and networks
 */
static int
qemudShutdown(void) {
339
    virDomainObjPtr vm;
340
    virNetworkObjPtr network;
341

342
    if (!qemu_driver)
343
        return -1;
344

345 346
    virCapabilitiesFree(qemu_driver->caps);

347
    /* shutdown active VMs */
348
    vm = qemu_driver->domains;
349
    while (vm) {
350 351
        virDomainObjPtr next = vm->next;
        if (virDomainIsActive(vm))
352
            qemudShutdownVMDaemon(NULL, qemu_driver, vm);
353
        if (!vm->persistent)
354 355
            virDomainRemoveInactive(&qemu_driver->domains,
                                    vm);
356 357
        vm = next;
    }
358

359
    /* free inactive VMs */
360
    vm = qemu_driver->domains;
361
    while (vm) {
362 363
        virDomainObjPtr next = vm->next;
        virDomainObjFree(vm);
364 365
        vm = next;
    }
366
    qemu_driver->domains = NULL;
367 368

    /* shutdown active networks */
369
    network = qemu_driver->networks;
370
    while (network) {
371 372
        virNetworkObjPtr next = network->next;
        if (virNetworkIsActive(network))
373
            qemudShutdownNetworkDaemon(NULL, qemu_driver, network);
374 375
        network = next;
    }
376

377
    /* free inactive networks */
378
    network = qemu_driver->networks;
379
    while (network) {
380 381
        virNetworkObjPtr next = network->next;
        virNetworkObjFree(network);
382 383
        network = next;
    }
384 385
    qemu_driver->networks = NULL;

386
    VIR_FREE(qemu_driver->logDir);
387 388 389 390 391
    VIR_FREE(qemu_driver->configDir);
    VIR_FREE(qemu_driver->autostartDir);
    VIR_FREE(qemu_driver->networkConfigDir);
    VIR_FREE(qemu_driver->networkAutostartDir);
    VIR_FREE(qemu_driver->vncTLSx509certdir);
D
Daniel P. Berrange 已提交
392

393 394 395 396 397
    if (qemu_driver->brctl)
        brShutdown(qemu_driver->brctl);
    if (qemu_driver->iptables)
        iptablesContextFree(qemu_driver->iptables);

398
    VIR_FREE(qemu_driver);
399 400

    return 0;
401
}
A
Atsushi SAKAI 已提交
402
#endif
403 404

/* Return -1 for error, 1 to continue reading and 0 for success */
405 406
typedef int qemudHandlerMonitorOutput(virConnectPtr conn,
                                      struct qemud_driver *driver,
407
                                      virDomainObjPtr vm,
408 409 410 411
                                      const char *output,
                                      int fd);

static int
412 413
qemudReadMonitorOutput(virConnectPtr conn,
                       struct qemud_driver *driver,
414
                       virDomainObjPtr vm,
415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
                       int fd,
                       char *buf,
                       int buflen,
                       qemudHandlerMonitorOutput func,
                       const char *what)
{
#define MONITOR_TIMEOUT 3000
    int got = 0;
    buf[0] = '\0';

   /* Consume & discard the initial greeting */
    while (got < (buflen-1)) {
        int ret;

        ret = read(fd, buf+got, buflen-got-1);
        if (ret == 0) {
431
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
432
                             _("QEMU quit during %s startup\n%s"), what, buf);
433 434 435 436 437 438 439 440
            return -1;
        }
        if (ret < 0) {
            struct pollfd pfd = { .fd = fd, .events = POLLIN };
            if (errno == EINTR)
                continue;

            if (errno != EAGAIN) {
441
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
442
                                 _("Failure while reading %s startup output: %s"),
443 444 445 446 447 448
                                 what, strerror(errno));
                return -1;
            }

            ret = poll(&pfd, 1, MONITOR_TIMEOUT);
            if (ret == 0) {
449
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
450
                                 _("Timed out while reading %s startup output"), what);
451 452 453
                return -1;
            } else if (ret == -1) {
                if (errno != EINTR) {
454
                    qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
455
                                     _("Failure while reading %s startup output: %s"),
456 457 458 459 460 461 462 463 464
                                     what, strerror(errno));
                    return -1;
                }
            } else {
                /* Make sure we continue loop & read any further data
                   available before dealing with EOF */
                if (pfd.revents & (POLLIN | POLLHUP))
                    continue;

465
                qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
466
                                 _("Failure while reading %s startup output"), what);
467 468 469 470 471
                return -1;
            }
        } else {
            got += ret;
            buf[got] = '\0';
472
            if ((ret = func(conn, driver, vm, buf, fd)) != 1)
473 474 475 476
                return ret;
        }
    }

477
    qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
478
                     _("Out of space while reading %s startup output"), what);
479 480 481 482 483 484
    return -1;

#undef MONITOR_TIMEOUT
}

static int
485 486
qemudCheckMonitorPrompt(virConnectPtr conn ATTRIBUTE_UNUSED,
                        struct qemud_driver *driver ATTRIBUTE_UNUSED,
487
                        virDomainObjPtr vm,
488 489 490 491 492 493 494 495 496 497 498
                        const char *output,
                        int fd)
{
    if (strstr(output, "(qemu) ") == NULL)
        return 1; /* keep reading */

    vm->monitor = fd;

    return 0;
}

499 500
static int qemudOpenMonitor(virConnectPtr conn,
                            struct qemud_driver *driver,
501
                            virDomainObjPtr vm,
502
                            const char *monitor) {
503 504 505 506
    int monfd;
    char buf[1024];
    int ret = -1;

D
Daniel Veillard 已提交
507
    if ((monfd = open(monitor, O_RDWR)) < 0) {
508
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
509
                         _("Unable to open monitor path %s"), monitor);
510 511 512
        return -1;
    }
    if (qemudSetCloseExec(monfd) < 0) {
513
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
514
                         "%s", _("Unable to set monitor close-on-exec flag"));
515 516 517
        goto error;
    }
    if (qemudSetNonBlock(monfd) < 0) {
518
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
519
                         "%s", _("Unable to put monitor into non-blocking mode"));
520 521 522
        goto error;
    }

523 524
    ret = qemudReadMonitorOutput(conn,
                                 driver, vm, monfd,
525 526 527
                                 buf, sizeof(buf),
                                 qemudCheckMonitorPrompt,
                                 "monitor");
528 529 530 531 532

    /* Keep monitor open upon success */
    if (ret == 0)
        return ret;

533 534 535 536 537
 error:
    close(monfd);
    return ret;
}

538 539
static int qemudExtractMonitorPath(virConnectPtr conn,
                                   const char *haystack,
540
                                   size_t *offset,
541
                                   char **path) {
542
    static const char needle[] = "char device redirected to";
543
    char *tmp, *dev;
544

545
    VIR_FREE(*path);
546
    /* First look for our magic string */
547 548 549 550 551
    if (!(tmp = strstr(haystack + *offset, needle))) {
        return 1;
    }
    tmp += sizeof(needle);
    dev = tmp;
552

553 554 555 556 557
    /*
     * And look for first whitespace character and nul terminate
     * to mark end of the pty path
     */
    while (*tmp) {
558
        if (c_isspace(*tmp)) {
559 560 561 562 563 564 565
            if (VIR_ALLOC_N(*path, (tmp-dev)+1) < 0) {
                qemudReportError(conn, NULL, NULL,
                                 VIR_ERR_NO_MEMORY, NULL);
                return -1;
            }
            strncpy(*path, dev, (tmp-dev));
            (*path)[(tmp-dev)] = '\0';
566
            /* ... now further update offset till we get EOL */
567
            *offset += tmp - haystack;
568 569
            return 0;
        }
570
        tmp++;
571 572 573 574 575
    }

    /*
     * We found a path, but didn't find any whitespace,
     * so it must be still incomplete - we should at
576 577
     * least see a \n - indicate that we want to carry
     * on trying again
578
     */
579
    return 1;
580 581 582
}

static int
583 584
qemudFindCharDevicePTYs(virConnectPtr conn,
                        struct qemud_driver *driver,
585
                        virDomainObjPtr vm,
586 587
                        const char *output,
                        int fd ATTRIBUTE_UNUSED)
588
{
589
    char *monitor = NULL;
590
    size_t offset = 0;
591 592
    virDomainChrDefPtr chr;
    int ret;
593 594 595 596 597

    /* The order in which QEMU prints out the PTY paths is
       the order in which it procsses its monitor, serial
       and parallel device args. This code must match that
       ordering.... */
598

599
    /* So first comes the monitor device */
600 601
    if ((ret = qemudExtractMonitorPath(conn, output, &offset, &monitor)) != 0)
        goto cleanup;
602

603 604 605
    /* then the serial devices */
    chr = vm->def->serials;
    while (chr) {
606 607 608 609
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
            if ((ret = qemudExtractMonitorPath(conn, output, &offset,
                                               &chr->data.file.path)) != 0)
                goto cleanup;
610 611 612 613 614 615 616
        }
        chr = chr->next;
    }

    /* and finally the parallel devices */
    chr = vm->def->parallels;
    while (chr) {
617 618 619 620
        if (chr->type == VIR_DOMAIN_CHR_TYPE_PTY) {
            if ((ret = qemudExtractMonitorPath(conn, output, &offset,
                                               &chr->data.file.path)) != 0)
                goto cleanup;
621 622 623 624 625
        }
        chr = chr->next;
    }

    /* Got them all, so now open the monitor console */
626 627 628 629 630
    ret = qemudOpenMonitor(conn, driver, vm, monitor);

cleanup:
    VIR_FREE(monitor);
    return ret;
631 632
}

633 634
static int qemudWaitForMonitor(virConnectPtr conn,
                               struct qemud_driver *driver,
635
                               virDomainObjPtr vm) {
636
    char buf[1024]; /* Plenty of space to get startup greeting */
637
    int ret = qemudReadMonitorOutput(conn,
638
                                     driver, vm, vm->stderr_fd,
639
                                     buf, sizeof(buf),
640
                                     qemudFindCharDevicePTYs,
641 642 643
                                     "console");

    buf[sizeof(buf)-1] = '\0';
644 645

    if (safewrite(vm->logfile, buf, strlen(buf)) < 0) {
646
        /* Log, but ignore failures to write logfile for VM */
647
        qemudLog(QEMUD_WARN, _("Unable to log VM console data: %s\n"),
648 649 650 651 652
                 strerror(errno));
    }
    return ret;
}

653 654 655
static int
qemudDetectVcpuPIDs(virConnectPtr conn,
                    struct qemud_driver *driver,
656
                    virDomainObjPtr vm) {
657 658 659 660 661 662
    char *qemucpus = NULL;
    char *line;
    int lastVcpu = -1;

    /* Only KVM has seperate threads for CPUs,
       others just use main QEMU process for CPU */
663
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM)
664 665 666 667 668 669 670 671 672 673
        vm->nvcpupids = 1;
    else
        vm->nvcpupids = vm->def->vcpus;

    if (VIR_ALLOC_N(vm->vcpupids, vm->nvcpupids) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                         "%s", _("allocate cpumap"));
        return -1;
    }

674
    if (vm->def->virtType != VIR_DOMAIN_VIRT_KVM) {
675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737
        vm->vcpupids[0] = vm->pid;
        return 0;
    }

    if (qemudMonitorCommand(driver, vm, "info cpus", &qemucpus) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot run monitor command to fetch CPU thread info"));
        VIR_FREE(vm->vcpupids);
        vm->nvcpupids = 0;
        return -1;
    }

    /*
     * This is the gross format we're about to parse :-{
     *
     * (qemu) info cpus
     * * CPU #0: pc=0x00000000000f0c4a thread_id=30019
     *   CPU #1: pc=0x00000000fffffff0 thread_id=30020
     *   CPU #2: pc=0x00000000fffffff0 thread_id=30021
     *
     */
    line = qemucpus;
    do {
        char *offset = strchr(line, '#');
        char *end = NULL;
        int vcpu = 0, tid = 0;

        /* See if we're all done */
        if (offset == NULL)
            break;

        /* Extract VCPU number */
        if (virStrToLong_i(offset + 1, &end, 10, &vcpu) < 0)
            goto error;
        if (end == NULL || *end != ':')
            goto error;

        /* Extract host Thread ID */
        if ((offset = strstr(line, "thread_id=")) == NULL)
            goto error;
        if (virStrToLong_i(offset + strlen("thread_id="), &end, 10, &tid) < 0)
            goto error;
        if (end == NULL || !c_isspace(*end))
            goto error;

        /* Validate the VCPU is in expected range & order */
        if (vcpu > vm->nvcpupids ||
            vcpu != (lastVcpu + 1))
            goto error;

        lastVcpu = vcpu;
        vm->vcpupids[vcpu] = tid;

        /* Skip to next data line */
        line = strchr(offset, '\r');
        if (line == NULL)
            line = strchr(offset, '\n');
    } while (line != NULL);

    /* Validate we got data for all VCPUs we expected */
    if (lastVcpu != (vm->def->vcpus - 1))
        goto error;

738
    VIR_FREE(qemucpus);
739 740 741 742
    return 0;

error:
    VIR_FREE(vm->vcpupids);
743 744
    vm->nvcpupids = 0;
    VIR_FREE(qemucpus);
745 746 747 748 749 750 751 752

    /* Explicitly return success, not error. Older KVM does
       not have vCPU -> Thread mapping info and we don't
       want to break its use. This merely disables ability
       to pin vCPUS with libvirt */
    return 0;
}

753 754 755
static int
qemudInitCpus(virConnectPtr conn,
              struct qemud_driver *driver,
756
              virDomainObjPtr vm) {
757 758 759 760 761 762 763 764 765 766 767 768 769 770 771
    char *info = NULL;
#if HAVE_SCHED_GETAFFINITY
    cpu_set_t mask;
    int i, maxcpu = QEMUD_CPUMASK_LEN;
    virNodeInfo nodeinfo;

    if (virNodeInfoPopulate(conn, &nodeinfo) < 0)
        return -1;

    /* setaffinity fails if you set bits for CPUs which
     * aren't present, so we have to limit ourselves */
    if (maxcpu > nodeinfo.cpus)
        maxcpu = nodeinfo.cpus;

    CPU_ZERO(&mask);
D
Daniel P. Berrange 已提交
772 773 774 775 776 777
    if (vm->def->cpumask) {
        for (i = 0 ; i < maxcpu ; i++)
            if (vm->def->cpumask[i])
                CPU_SET(i, &mask);
    } else {
        for (i = 0 ; i < maxcpu ; i++)
778
            CPU_SET(i, &mask);
D
Daniel P. Berrange 已提交
779
    }
780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797

    for (i = 0 ; i < vm->nvcpupids ; i++) {
        if (sched_setaffinity(vm->vcpupids[i],
                              sizeof(mask), &mask) < 0) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             _("failed to set CPU affinity %s"),
                             strerror(errno));
            return -1;
        }
    }
#endif /* HAVE_SCHED_GETAFFINITY */

    /* Allow the CPUS to start executing */
    if (qemudMonitorCommand(driver, vm, "cont", &info) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("resume operation failed"));
        return -1;
    }
798
    VIR_FREE(info);
799 800 801 802 803

    return 0;
}


804
static int qemudNextFreeVNCPort(struct qemud_driver *driver ATTRIBUTE_UNUSED) {
805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839
    int i;

    for (i = 5900 ; i < 6000 ; i++) {
        int fd;
        int reuse = 1;
        struct sockaddr_in addr;
        addr.sin_family = AF_INET;
        addr.sin_port = htons(i);
        addr.sin_addr.s_addr = htonl(INADDR_ANY);
        fd = socket(PF_INET, SOCK_STREAM, 0);
        if (fd < 0)
            return -1;

        if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (void*)&reuse, sizeof(reuse)) < 0) {
            close(fd);
            break;
        }

        if (bind(fd, (struct sockaddr*)&addr, sizeof(addr)) == 0) {
            /* Not in use, lets grab it */
            close(fd);
            return i;
        }
        close(fd);

        if (errno == EADDRINUSE) {
            /* In use, try next */
            continue;
        }
        /* Some other bad failure, get out.. */
        break;
    }
    return -1;
}

840 841
static int qemudStartVMDaemon(virConnectPtr conn,
                              struct qemud_driver *driver,
842 843
                              virDomainObjPtr vm,
                              const char *migrateFrom) {
844
    const char **argv = NULL, **tmp;
845
    int i, ret;
846
    char logfile[PATH_MAX];
847
    struct stat sb;
848 849
    int *tapfds = NULL;
    int ntapfds = 0;
850
    unsigned int qemuCmdFlags;
851 852 853
    fd_set keepfd;

    FD_ZERO(&keepfd);
854

855
    if (virDomainIsActive(vm)) {
856
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
857
                         "%s", _("VM is already active"));
858 859 860
        return -1;
    }

861 862 863
    if (vm->def->graphics &&
        vm->def->graphics->type == VIR_DOMAIN_GRAPHICS_TYPE_VNC &&
        vm->def->graphics->data.vnc.autoport) {
864
        int port = qemudNextFreeVNCPort(driver);
865
        if (port < 0) {
866
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
867
                             "%s", _("Unable to find an unused VNC port"));
868 869
            return -1;
        }
870 871
        vm->def->graphics->data.vnc.port = port;
    }
872

873
    if ((strlen(driver->logDir) + /* path */
874 875 876 877
         1 + /* Separator */
         strlen(vm->def->name) + /* basename */
         4 + /* suffix .log */
         1 /* NULL */) > PATH_MAX) {
878
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
879
                         _("config file path too long: %s/%s.log"),
880
                         driver->logDir, vm->def->name);
881 882
        return -1;
    }
883
    strcpy(logfile, driver->logDir);
884 885 886 887
    strcat(logfile, "/");
    strcat(logfile, vm->def->name);
    strcat(logfile, ".log");

888
    if (virFileMakePath(driver->logDir) < 0) {
889
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
890
                         _("cannot create log directory %s: %s"),
891
                         driver->logDir, strerror(errno));
892 893 894 895 896
        return -1;
    }

    if ((vm->logfile = open(logfile, O_CREAT | O_TRUNC | O_WRONLY,
                            S_IRUSR | S_IWUSR)) < 0) {
897
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
898
                         _("failed to create logfile %s: %s"),
899 900 901
                         logfile, strerror(errno));
        return -1;
    }
902 903
    if (qemudSetCloseExec(vm->logfile) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
904
                         _("Unable to set VM logfile close-on-exec flag %s"),
905 906 907 908 909
                         strerror(errno));
        close(vm->logfile);
        vm->logfile = -1;
        return -1;
    }
910

911 912 913 914
    /* Make sure the binary we are about to try exec'ing exists.
     * Technically we could catch the exec() failure, but that's
     * in a sub-process so its hard to feed back a useful error
     */
915
    if (stat(vm->def->emulator, &sb) < 0) {
916 917
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("Cannot find QEMU binary %s: %s"),
918
                         vm->def->emulator,
919 920 921 922
                         strerror(errno));
        return -1;
    }

923 924 925 926 927 928 929 930
    if (qemudExtractVersionInfo(vm->def->emulator,
                                NULL,
                                &qemuCmdFlags) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("Cannot determine QEMU argv syntax %s"),
                         vm->def->emulator);
        return -1;
    }
931

932 933 934
    if (qemudBuildCommandLine(conn, driver, vm,
                              qemuCmdFlags, &argv,
                              &tapfds, &ntapfds, migrateFrom) < 0) {
935 936 937 938 939 940 941
        close(vm->logfile);
        vm->logfile = -1;
        return -1;
    }

    tmp = argv;
    while (*tmp) {
942
        if (safewrite(vm->logfile, *tmp, strlen(*tmp)) < 0)
943
            qemudLog(QEMUD_WARN, _("Unable to write argv to logfile %d: %s\n"),
944
                     errno, strerror(errno));
945
        if (safewrite(vm->logfile, " ", 1) < 0)
946
            qemudLog(QEMUD_WARN, _("Unable to write argv to logfile %d: %s\n"),
947 948 949
                     errno, strerror(errno));
        tmp++;
    }
950
    if (safewrite(vm->logfile, "\n", 1) < 0)
951
        qemudLog(QEMUD_WARN, _("Unable to write argv to logfile %d: %s\n"),
952 953
                 errno, strerror(errno));

954 955 956
    vm->stdout_fd = -1;
    vm->stderr_fd = -1;

957 958 959 960
    for (i = 0 ; i < ntapfds ; i++)
        FD_SET(tapfds[i], &keepfd);

    ret = virExec(conn, argv, NULL, &keepfd, &vm->pid,
961 962
                  vm->stdin_fd, &vm->stdout_fd, &vm->stderr_fd,
                  VIR_EXEC_NONBLOCK);
963
    if (ret == 0) {
964 965
        vm->def->id = driver->nextvmid++;
        vm->state = migrateFrom ? VIR_DOMAIN_PAUSED : VIR_DOMAIN_RUNNING;
966 967
    }

968
    for (i = 0 ; argv[i] ; i++)
969 970
        VIR_FREE(argv[i]);
    VIR_FREE(argv);
971

972 973 974
    if (tapfds) {
        for (i = 0 ; i < ntapfds ; i++) {
            close(tapfds[i]);
975
        }
976
        VIR_FREE(tapfds);
977 978
    }

979
    if (ret == 0) {
980
        if ((virEventAddHandle(vm->stdout_fd,
981 982 983
                               POLLIN | POLLERR | POLLHUP,
                               qemudDispatchVMEvent,
                               driver) < 0) ||
984
            (virEventAddHandle(vm->stderr_fd,
985 986 987 988 989 990
                               POLLIN | POLLERR | POLLHUP,
                               qemudDispatchVMEvent,
                               driver) < 0) ||
            (qemudWaitForMonitor(conn, driver, vm) < 0) ||
            (qemudDetectVcpuPIDs(conn, driver, vm) < 0) ||
            (qemudInitCpus(conn, driver, vm) < 0)) {
991 992 993
            qemudShutdownVMDaemon(conn, driver, vm);
            return -1;
        }
994 995
    }

996
    return ret;
997 998
}

999
static int qemudVMData(struct qemud_driver *driver ATTRIBUTE_UNUSED,
1000
                       virDomainObjPtr vm, int fd) {
1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016
    char buf[4096];
    if (vm->pid < 0)
        return 0;

    for (;;) {
        int ret = read(fd, buf, sizeof(buf)-1);
        if (ret < 0) {
            if (errno == EAGAIN)
                return 0;
            return -1;
        }
        if (ret == 0) {
            return 0;
        }
        buf[ret] = '\0';

1017
        if (safewrite(vm->logfile, buf, ret) < 0) {
1018
            /* Log, but ignore failures to write logfile for VM */
1019
            qemudLog(QEMUD_WARN, _("Unable to log VM console data: %s\n"),
1020 1021 1022
                     strerror(errno));
        }
    }
1023 1024

    qemudAutostartConfigs(qemu_driver);
1025 1026 1027
}


1028
static void qemudShutdownVMDaemon(virConnectPtr conn ATTRIBUTE_UNUSED,
1029 1030
                                  struct qemud_driver *driver, virDomainObjPtr vm) {
    if (!virDomainIsActive(vm))
1031
        return;
1032

1033
    qemudLog(QEMUD_INFO, _("Shutting down VM '%s'\n"), vm->def->name);
1034 1035 1036

    kill(vm->pid, SIGTERM);

1037 1038
    qemudVMData(driver, vm, vm->stdout_fd);
    qemudVMData(driver, vm, vm->stderr_fd);
1039

1040 1041
    virEventRemoveHandle(vm->stdout_fd);
    virEventRemoveHandle(vm->stderr_fd);
1042 1043

    if (close(vm->logfile) < 0)
1044
        qemudLog(QEMUD_WARN, _("Unable to close logfile %d: %s\n"),
1045
                 errno, strerror(errno));
1046 1047
    close(vm->stdout_fd);
    close(vm->stderr_fd);
1048 1049 1050
    if (vm->monitor != -1)
        close(vm->monitor);
    vm->logfile = -1;
1051 1052
    vm->stdout_fd = -1;
    vm->stderr_fd = -1;
1053 1054 1055 1056 1057
    vm->monitor = -1;

    if (waitpid(vm->pid, NULL, WNOHANG) != vm->pid) {
        kill(vm->pid, SIGKILL);
        if (waitpid(vm->pid, NULL, 0) != vm->pid) {
J
Jim Meyering 已提交
1058
            qemudLog(QEMUD_WARN,
1059
                     "%s", _("Got unexpected pid, damn\n"));
1060 1061 1062 1063
        }
    }

    vm->pid = -1;
1064
    vm->def->id = -1;
1065
    vm->state = VIR_DOMAIN_SHUTOFF;
1066
    VIR_FREE(vm->vcpupids);
1067
    vm->nvcpupids = 0;
1068 1069

    if (vm->newDef) {
1070
        virDomainDefFree(vm->def);
1071
        vm->def = vm->newDef;
1072
        vm->def->id = -1;
1073 1074 1075 1076
        vm->newDef = NULL;
    }
}

1077
static int qemudDispatchVMLog(struct qemud_driver *driver, virDomainObjPtr vm, int fd) {
1078 1079
    if (qemudVMData(driver, vm, fd) < 0) {
        qemudShutdownVMDaemon(NULL, driver, vm);
1080
        if (!vm->persistent)
1081 1082
            virDomainRemoveInactive(&driver->domains,
                                    vm);
1083
    }
1084 1085 1086
    return 0;
}

1087
static int qemudDispatchVMFailure(struct qemud_driver *driver, virDomainObjPtr vm,
1088
                                  int fd ATTRIBUTE_UNUSED) {
1089
    qemudShutdownVMDaemon(NULL, driver, vm);
1090
    if (!vm->persistent)
1091 1092
        virDomainRemoveInactive(&driver->domains,
                                vm);
1093 1094 1095 1096
    return 0;
}

static int
1097
qemudBuildDnsmasqArgv(virConnectPtr conn,
1098
                      virNetworkObjPtr network,
1099
                      const char ***argv) {
1100
    int i, len, r;
1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
    char buf[PATH_MAX];

    len =
        1 + /* dnsmasq */
        1 + /* --keep-in-foreground */
        1 + /* --strict-order */
        1 + /* --bind-interfaces */
        2 + /* --pid-file "" */
        2 + /* --conf-file "" */
        /*2 + *//* --interface virbr0 */
        2 + /* --except-interface lo */
        2 + /* --listen-address 10.0.0.1 */
        1 + /* --dhcp-leasefile=path */
        (2 * network->def->nranges) + /* --dhcp-range 10.0.0.2,10.0.0.254 */
1115 1116
        /*  --dhcp-host 01:23:45:67:89:0a,hostname,10.0.0.3 */
        (2 * network->def->nhosts) +
1117 1118
        1;  /* NULL */

1119
    if (VIR_ALLOC_N(*argv, len) < 0)
1120 1121 1122 1123 1124 1125 1126 1127 1128
        goto no_memory;

#define APPEND_ARG(v, n, s) do {     \
        if (!((v)[(n)] = strdup(s))) \
            goto no_memory;          \
    } while (0)

    i = 0;

1129
    APPEND_ARG(*argv, i++, DNSMASQ);
1130 1131 1132 1133

    APPEND_ARG(*argv, i++, "--keep-in-foreground");
    /*
     * Needed to ensure dnsmasq uses same algorithm for processing
1134
     * multiple namedriver entries in /etc/resolv.conf as GLibC.
1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164 1165 1166 1167 1168
     */
    APPEND_ARG(*argv, i++, "--strict-order");
    APPEND_ARG(*argv, i++, "--bind-interfaces");

    APPEND_ARG(*argv, i++, "--pid-file");
    APPEND_ARG(*argv, i++, "");

    APPEND_ARG(*argv, i++, "--conf-file");
    APPEND_ARG(*argv, i++, "");

    /*
     * XXX does not actually work, due to some kind of
     * race condition setting up ipv6 addresses on the
     * interface. A sleep(10) makes it work, but that's
     * clearly not practical
     *
     * APPEND_ARG(*argv, i++, "--interface");
     * APPEND_ARG(*argv, i++, network->def->bridge);
     */
    APPEND_ARG(*argv, i++, "--listen-address");
    APPEND_ARG(*argv, i++, network->def->ipAddress);

    APPEND_ARG(*argv, i++, "--except-interface");
    APPEND_ARG(*argv, i++, "lo");

    /*
     * NB, dnsmasq command line arg bug means we need to
     * use a single arg '--dhcp-leasefile=path' rather than
     * two separate args in '--dhcp-leasefile path' style
     */
    snprintf(buf, sizeof(buf), "--dhcp-leasefile=%s/lib/libvirt/dhcp-%s.leases",
             LOCAL_STATE_DIR, network->def->name);
    APPEND_ARG(*argv, i++, buf);

1169
    for (r = 0 ; r < network->def->nranges ; r++) {
1170
        snprintf(buf, sizeof(buf), "%s,%s",
1171 1172
                 network->def->ranges[r].start,
                 network->def->ranges[r].end);
1173 1174 1175 1176 1177

        APPEND_ARG(*argv, i++, "--dhcp-range");
        APPEND_ARG(*argv, i++, buf);
    }

1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195
    for (r = 0 ; r < network->def->nhosts ; r++) {
        virNetworkDHCPHostDefPtr host = &(network->def->hosts[r]);
        if ((host->mac) && (host->name)) {
            snprintf(buf, sizeof(buf), "%s,%s,%s",
                     host->mac, host->name, host->ip);
        } else if (host->mac) {
            snprintf(buf, sizeof(buf), "%s,%s",
                     host->mac, host->ip);
        } else if (host->name) {
            snprintf(buf, sizeof(buf), "%s,%s",
                     host->name, host->ip);
        } else
            continue;

        APPEND_ARG(*argv, i++, "--dhcp-host");
        APPEND_ARG(*argv, i++, buf);
    }

1196 1197 1198 1199 1200 1201 1202
#undef APPEND_ARG

    return 0;

 no_memory:
    if (argv) {
        for (i = 0; (*argv)[i]; i++)
1203 1204
            VIR_FREE((*argv)[i]);
        VIR_FREE(*argv);
1205
    }
1206 1207
    qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                     "%s", _("failed to allocate space for dnsmasq argv"));
1208 1209 1210 1211 1212
    return -1;
}


static int
1213
dhcpStartDhcpDaemon(virConnectPtr conn,
1214
                    virNetworkObjPtr network)
1215
{
1216
    const char **argv;
1217 1218
    int ret, i;

1219
    if (network->def->ipAddress == NULL) {
1220
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1221
                         "%s", _("cannot start dhcp daemon without IP address for server"));
1222 1223 1224 1225
        return -1;
    }

    argv = NULL;
1226
    if (qemudBuildDnsmasqArgv(conn, network, &argv) < 0)
1227 1228
        return -1;

1229 1230
    ret = virExec(conn, argv, NULL, NULL,
                  &network->dnsmasqPid, -1, NULL, NULL, VIR_EXEC_NONBLOCK);
1231 1232

    for (i = 0; argv[i]; i++)
1233 1234
        VIR_FREE(argv[i]);
    VIR_FREE(argv);
1235 1236 1237 1238

    return ret;
}

1239 1240 1241
static int
qemudAddMasqueradingIptablesRules(virConnectPtr conn,
                      struct qemud_driver *driver,
1242
                      virNetworkObjPtr network) {
1243 1244 1245 1246
    int err;
    /* allow forwarding packets from the bridge interface */
    if ((err = iptablesAddForwardAllowOut(driver->iptables,
                                          network->def->network,
1247
                                          network->def->bridge,
1248 1249 1250
                                          network->def->forwardDev))) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add iptables rule to allow forwarding from '%s' : %s\n"),
1251
                         network->def->bridge, strerror(err));
1252 1253 1254 1255 1256 1257
        goto masqerr1;
    }

    /* allow forwarding packets to the bridge interface if they are part of an existing connection */
    if ((err = iptablesAddForwardAllowRelatedIn(driver->iptables,
                                         network->def->network,
1258
                                         network->def->bridge,
1259 1260 1261
                                         network->def->forwardDev))) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add iptables rule to allow forwarding to '%s' : %s\n"),
1262
                         network->def->bridge, strerror(err));
1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280
        goto masqerr2;
    }

    /* enable masquerading */
    if ((err = iptablesAddForwardMasquerade(driver->iptables,
                                            network->def->network,
                                            network->def->forwardDev))) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add iptables rule to enable masquerading : %s\n"),
                         strerror(err));
        goto masqerr3;
    }

    return 1;

 masqerr3:
    iptablesRemoveForwardAllowRelatedIn(driver->iptables,
                                 network->def->network,
1281
                                 network->def->bridge,
1282 1283 1284 1285
                                 network->def->forwardDev);
 masqerr2:
    iptablesRemoveForwardAllowOut(driver->iptables,
                                  network->def->network,
1286
                                  network->def->bridge,
1287 1288 1289 1290 1291 1292 1293 1294
                                  network->def->forwardDev);
 masqerr1:
    return 0;
}

static int
qemudAddRoutingIptablesRules(virConnectPtr conn,
                      struct qemud_driver *driver,
1295
                      virNetworkObjPtr network) {
1296 1297 1298 1299
    int err;
    /* allow routing packets from the bridge interface */
    if ((err = iptablesAddForwardAllowOut(driver->iptables,
                                          network->def->network,
1300
                                          network->def->bridge,
1301 1302 1303
                                          network->def->forwardDev))) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add iptables rule to allow routing from '%s' : %s\n"),
1304
                         network->def->bridge, strerror(err));
1305 1306 1307 1308 1309 1310
        goto routeerr1;
    }

    /* allow routing packets to the bridge interface */
    if ((err = iptablesAddForwardAllowIn(driver->iptables,
                                         network->def->network,
1311
                                         network->def->bridge,
1312 1313 1314
                                         network->def->forwardDev))) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("failed to add iptables rule to allow routing to '%s' : %s\n"),
1315
                         network->def->bridge, strerror(err));
1316 1317 1318 1319 1320 1321 1322 1323 1324
        goto routeerr2;
    }

    return 1;


 routeerr2:
    iptablesRemoveForwardAllowOut(driver->iptables,
                                  network->def->network,
1325
                                  network->def->bridge,
1326 1327 1328 1329 1330
                                  network->def->forwardDev);
 routeerr1:
    return 0;
}

1331
static int
1332 1333
qemudAddIptablesRules(virConnectPtr conn,
                      struct qemud_driver *driver,
1334
                      virNetworkObjPtr network) {
1335 1336
    int err;

1337
    if (!driver->iptables && !(driver->iptables = iptablesContextNew())) {
1338 1339
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                     "%s", _("failed to allocate space for IP tables support"));
1340
        return 0;
1341 1342 1343 1344
    }


    /* allow DHCP requests through to dnsmasq */
1345
    if ((err = iptablesAddTcpInput(driver->iptables, network->def->bridge, 67))) {
1346
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1347
                         _("failed to add iptables rule to allow DHCP requests from '%s' : %s"),
1348
                         network->def->bridge, strerror(err));
1349 1350 1351
        goto err1;
    }

1352
    if ((err = iptablesAddUdpInput(driver->iptables, network->def->bridge, 67))) {
1353
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1354
                         _("failed to add iptables rule to allow DHCP requests from '%s' : %s"),
1355
                         network->def->bridge, strerror(err));
1356 1357 1358 1359
        goto err2;
    }

    /* allow DNS requests through to dnsmasq */
1360
    if ((err = iptablesAddTcpInput(driver->iptables, network->def->bridge, 53))) {
1361
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1362
                         _("failed to add iptables rule to allow DNS requests from '%s' : %s"),
1363
                         network->def->bridge, strerror(err));
1364 1365 1366
        goto err3;
    }

1367
    if ((err = iptablesAddUdpInput(driver->iptables, network->def->bridge, 53))) {
1368
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1369
                         _("failed to add iptables rule to allow DNS requests from '%s' : %s"),
1370
                         network->def->bridge, strerror(err));
1371 1372 1373 1374 1375 1376
        goto err4;
    }


    /* Catch all rules to block forwarding to/from bridges */

1377
    if ((err = iptablesAddForwardRejectOut(driver->iptables, network->def->bridge))) {
1378
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1379
                         _("failed to add iptables rule to block outbound traffic from '%s' : %s"),
1380
                         network->def->bridge, strerror(err));
1381 1382 1383
        goto err5;
    }

1384
    if ((err = iptablesAddForwardRejectIn(driver->iptables, network->def->bridge))) {
1385
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1386
                         _("failed to add iptables rule to block inbound traffic to '%s' : %s"),
1387
                         network->def->bridge, strerror(err));
1388 1389 1390 1391
        goto err6;
    }

    /* Allow traffic between guests on the same bridge */
1392
    if ((err = iptablesAddForwardAllowCross(driver->iptables, network->def->bridge))) {
1393
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1394
                         _("failed to add iptables rule to allow cross bridge traffic on '%s' : %s"),
1395
                         network->def->bridge, strerror(err));
1396 1397 1398 1399
        goto err7;
    }


1400 1401 1402 1403 1404 1405 1406 1407
    /* If masquerading is enabled, set up the rules*/
    if (network->def->forwardType == VIR_NETWORK_FORWARD_NAT &&
        !qemudAddMasqueradingIptablesRules(conn, driver, network))
        goto err8;
    /* else if routing is enabled, set up the rules*/
    else if (network->def->forwardType == VIR_NETWORK_FORWARD_ROUTE &&
             !qemudAddRoutingIptablesRules(conn, driver, network))
        goto err8;
1408

1409 1410 1411
    iptablesSaveRules(driver->iptables);

    return 1;
1412

1413
 err8:
1414
    iptablesRemoveForwardAllowCross(driver->iptables,
1415
                                    network->def->bridge);
1416
 err7:
1417
    iptablesRemoveForwardRejectIn(driver->iptables,
1418
                                  network->def->bridge);
1419
 err6:
1420
    iptablesRemoveForwardRejectOut(driver->iptables,
1421
                                   network->def->bridge);
1422
 err5:
1423
    iptablesRemoveUdpInput(driver->iptables, network->def->bridge, 53);
1424
 err4:
1425
    iptablesRemoveTcpInput(driver->iptables, network->def->bridge, 53);
1426
 err3:
1427
    iptablesRemoveUdpInput(driver->iptables, network->def->bridge, 67);
1428
 err2:
1429
    iptablesRemoveTcpInput(driver->iptables, network->def->bridge, 67);
1430 1431 1432 1433 1434
 err1:
    return 0;
}

static void
1435
qemudRemoveIptablesRules(struct qemud_driver *driver,
1436 1437
                         virNetworkObjPtr network) {
    if (network->def->forwardType != VIR_NETWORK_FORWARD_NONE) {
1438
        iptablesRemoveForwardMasquerade(driver->iptables,
1439 1440 1441
                                        network->def->network,
                                        network->def->forwardDev);

1442
        if (network->def->forwardType == VIR_NETWORK_FORWARD_NAT)
1443 1444
            iptablesRemoveForwardAllowRelatedIn(driver->iptables,
                                                network->def->network,
1445
                                                network->def->bridge,
1446
                                                network->def->forwardDev);
1447
        else if (network->def->forwardType == VIR_NETWORK_FORWARD_ROUTE)
1448 1449
            iptablesRemoveForwardAllowIn(driver->iptables,
                                         network->def->network,
1450
                                         network->def->bridge,
1451 1452
                                         network->def->forwardDev);

1453
        iptablesRemoveForwardAllowOut(driver->iptables,
1454
                                      network->def->network,
1455
                                      network->def->bridge,
1456 1457
                                      network->def->forwardDev);
    }
1458 1459 1460 1461 1462 1463 1464
    iptablesRemoveForwardAllowCross(driver->iptables, network->def->bridge);
    iptablesRemoveForwardRejectIn(driver->iptables, network->def->bridge);
    iptablesRemoveForwardRejectOut(driver->iptables, network->def->bridge);
    iptablesRemoveUdpInput(driver->iptables, network->def->bridge, 53);
    iptablesRemoveTcpInput(driver->iptables, network->def->bridge, 53);
    iptablesRemoveUdpInput(driver->iptables, network->def->bridge, 67);
    iptablesRemoveTcpInput(driver->iptables, network->def->bridge, 67);
1465
    iptablesSaveRules(driver->iptables);
1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477
}

static int
qemudEnableIpForwarding(void)
{
#define PROC_IP_FORWARD "/proc/sys/net/ipv4/ip_forward"

    int fd, ret;

    if ((fd = open(PROC_IP_FORWARD, O_WRONLY|O_TRUNC)) == -1)
        return 0;

1478
    if (safewrite(fd, "1\n", 2) < 0)
1479 1480 1481 1482 1483 1484 1485 1486 1487
        ret = 0;

    close (fd);

    return 1;

#undef PROC_IP_FORWARD
}

1488 1489
static int qemudStartNetworkDaemon(virConnectPtr conn,
                                   struct qemud_driver *driver,
1490
                                   virNetworkObjPtr network) {
1491 1492
    int err;

1493
    if (virNetworkIsActive(network)) {
1494
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1495
                         "%s", _("network is already active"));
1496 1497 1498
        return -1;
    }

1499
    if (!driver->brctl && (err = brInit(&driver->brctl))) {
1500
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1501
                         _("cannot initialize bridge support: %s"), strerror(err));
1502 1503 1504
        return -1;
    }

1505
    if ((err = brAddBridge(driver->brctl, &network->def->bridge))) {
1506
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1507 1508
                         _("cannot create bridge '%s' : %s"),
                         network->def->bridge, strerror(err));
1509 1510 1511
        return -1;
    }

1512

1513
    if (brSetForwardDelay(driver->brctl, network->def->bridge, network->def->delay) < 0)
1514 1515
        goto err_delbr;

1516
    if (brSetEnableSTP(driver->brctl, network->def->bridge, network->def->stp ? 1 : 0) < 0)
1517 1518
        goto err_delbr;

1519 1520
    if (network->def->ipAddress &&
        (err = brSetInetAddress(driver->brctl, network->def->bridge, network->def->ipAddress))) {
1521
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1522
                         _("cannot set IP address on bridge '%s' to '%s' : %s"),
1523
                         network->def->bridge, network->def->ipAddress, strerror(err));
1524 1525 1526
        goto err_delbr;
    }

1527 1528
    if (network->def->netmask &&
        (err = brSetInetNetmask(driver->brctl, network->def->bridge, network->def->netmask))) {
1529
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1530
                         _("cannot set netmask on bridge '%s' to '%s' : %s"),
1531
                         network->def->bridge, network->def->netmask, strerror(err));
1532 1533 1534
        goto err_delbr;
    }

1535 1536
    if (network->def->ipAddress &&
        (err = brSetInterfaceUp(driver->brctl, network->def->bridge, 1))) {
1537
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1538
                         _("failed to bring the bridge '%s' up : %s"),
1539
                         network->def->bridge, strerror(err));
1540 1541 1542
        goto err_delbr;
    }

1543
    if (!qemudAddIptablesRules(conn, driver, network))
1544 1545
        goto err_delbr1;

1546
    if (network->def->forwardType != VIR_NETWORK_FORWARD_NONE &&
1547
        !qemudEnableIpForwarding()) {
1548
        qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
1549
                         _("failed to enable IP forwarding : %s"), strerror(err));
1550 1551 1552
        goto err_delbr2;
    }

1553
    if (network->def->nranges &&
1554
        dhcpStartDhcpDaemon(conn, network) < 0)
1555 1556 1557 1558 1559 1560 1561
        goto err_delbr2;

    network->active = 1;

    return 0;

 err_delbr2:
1562
    qemudRemoveIptablesRules(driver, network);
1563 1564

 err_delbr1:
1565 1566
    if (network->def->ipAddress &&
        (err = brSetInterfaceUp(driver->brctl, network->def->bridge, 0))) {
1567
        qemudLog(QEMUD_WARN, _("Failed to bring down bridge '%s' : %s\n"),
1568
                 network->def->bridge, strerror(err));
1569 1570 1571
    }

 err_delbr:
1572
    if ((err = brDeleteBridge(driver->brctl, network->def->bridge))) {
1573
        qemudLog(QEMUD_WARN, _("Failed to delete bridge '%s' : %s\n"),
1574
                 network->def->bridge, strerror(err));
1575 1576 1577 1578 1579 1580
    }

    return -1;
}


1581 1582
static int qemudShutdownNetworkDaemon(virConnectPtr conn ATTRIBUTE_UNUSED,
                                      struct qemud_driver *driver,
1583
                                      virNetworkObjPtr network) {
1584 1585
    int err;

1586
    qemudLog(QEMUD_INFO, _("Shutting down network '%s'\n"), network->def->name);
1587

1588
    if (!virNetworkIsActive(network))
1589 1590 1591 1592 1593
        return 0;

    if (network->dnsmasqPid > 0)
        kill(network->dnsmasqPid, SIGTERM);

1594
    qemudRemoveIptablesRules(driver, network);
1595

1596 1597
    if (network->def->ipAddress &&
        (err = brSetInterfaceUp(driver->brctl, network->def->bridge, 0))) {
1598
        qemudLog(QEMUD_WARN, _("Failed to bring down bridge '%s' : %s\n"),
1599
                 network->def->bridge, strerror(err));
1600 1601
    }

1602
    if ((err = brDeleteBridge(driver->brctl, network->def->bridge))) {
1603
        qemudLog(QEMUD_WARN, _("Failed to delete bridge '%s' : %s\n"),
1604
                 network->def->bridge, strerror(err));
1605 1606 1607 1608 1609 1610
    }

    if (network->dnsmasqPid > 0 &&
        waitpid(network->dnsmasqPid, NULL, WNOHANG) != network->dnsmasqPid) {
        kill(network->dnsmasqPid, SIGKILL);
        if (waitpid(network->dnsmasqPid, NULL, 0) != network->dnsmasqPid)
J
Jim Meyering 已提交
1611
            qemudLog(QEMUD_WARN,
1612
                     "%s", _("Got unexpected pid for dnsmasq\n"));
1613 1614 1615 1616 1617 1618
    }

    network->dnsmasqPid = -1;
    network->active = 0;

    if (network->newDef) {
1619
        virNetworkDefFree(network->def);
1620 1621 1622 1623
        network->def = network->newDef;
        network->newDef = NULL;
    }

1624 1625 1626
    if (!network->configFile)
        virNetworkRemoveInactive(&driver->networks,
                                 network);
1627

1628 1629 1630 1631 1632
    return 0;
}


static void qemudDispatchVMEvent(int fd, int events, void *opaque) {
1633
    struct qemud_driver *driver = (struct qemud_driver *)opaque;
1634
    virDomainObjPtr vm = driver->domains;
1635 1636

    while (vm) {
1637
        if (virDomainIsActive(vm) &&
1638 1639
            (vm->stdout_fd == fd ||
             vm->stderr_fd == fd))
1640 1641 1642 1643 1644 1645 1646 1647
            break;

        vm = vm->next;
    }

    if (!vm)
        return;

1648
    if (events == POLLIN)
1649
        qemudDispatchVMLog(driver, vm, fd);
1650
    else
1651
        qemudDispatchVMFailure(driver, vm, fd);
1652 1653
}

1654 1655
static int
qemudMonitorCommand (const struct qemud_driver *driver ATTRIBUTE_UNUSED,
1656
                     const virDomainObjPtr vm,
1657 1658
                     const char *cmd,
                     char **reply) {
D
Daniel P. Berrange 已提交
1659 1660
    int size = 0;
    char *buf = NULL;
1661
    size_t cmdlen = strlen(cmd);
D
Daniel P. Berrange 已提交
1662

1663 1664 1665
    if (safewrite(vm->monitor, cmd, cmdlen) != cmdlen)
        return -1;
    if (safewrite(vm->monitor, "\r", 1) != 1)
D
Daniel P. Berrange 已提交
1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677
        return -1;

    *reply = NULL;

    for (;;) {
        struct pollfd fd = { vm->monitor, POLLIN | POLLERR | POLLHUP, 0 };
        char *tmp;

        /* Read all the data QEMU has sent thus far */
        for (;;) {
            char data[1024];
            int got = read(vm->monitor, data, sizeof(data));
D
Daniel P. Berrange 已提交
1678

1679 1680
            if (got == 0)
                goto error;
D
Daniel P. Berrange 已提交
1681 1682 1683 1684 1685
            if (got < 0) {
                if (errno == EINTR)
                    continue;
                if (errno == EAGAIN)
                    break;
1686
                goto error;
1687
            }
1688
            if (VIR_REALLOC_N(buf, size+got+1) < 0)
1689 1690
                goto error;

D
Daniel P. Berrange 已提交
1691 1692 1693 1694
            memmove(buf+size, data, got);
            buf[size+got] = '\0';
            size += got;
        }
1695

D
Daniel P. Berrange 已提交
1696
        /* Look for QEMU prompt to indicate completion */
D
Daniel P. Berrange 已提交
1697
        if (buf && ((tmp = strstr(buf, "\n(qemu) ")) != NULL)) {
D
Daniel P. Berrange 已提交
1698 1699 1700 1701 1702 1703 1704 1705
            tmp[0] = '\0';
            break;
        }
    pollagain:
        /* Need to wait for more data */
        if (poll(&fd, 1, -1) < 0) {
            if (errno == EINTR)
                goto pollagain;
1706
            goto error;
D
Daniel P. Berrange 已提交
1707 1708 1709
        }
    }

1710 1711
    /* Log, but ignore failures to write logfile for VM */
    if (safewrite(vm->logfile, buf, strlen(buf)) < 0)
1712
        qemudLog(QEMUD_WARN, _("Unable to log VM console data: %s\n"),
1713 1714
                 strerror(errno));

D
Daniel P. Berrange 已提交
1715 1716
    *reply = buf;
    return 0;
1717 1718 1719 1720 1721

 error:
    if (buf) {
        /* Log, but ignore failures to write logfile for VM */
        if (safewrite(vm->logfile, buf, strlen(buf)) < 0)
1722
            qemudLog(QEMUD_WARN, _("Unable to log VM console data: %s\n"),
1723
                     strerror(errno));
1724
        VIR_FREE(buf);
1725 1726
    }
    return -1;
D
Daniel P. Berrange 已提交
1727 1728
}

1729 1730 1731 1732 1733 1734 1735 1736 1737 1738
/**
 * qemudProbe:
 *
 * Probe for the availability of the qemu driver, assume the
 * presence of QEmu emulation if the binaries are installed
 */
static const char *qemudProbe(void)
{
    if ((virFileExists("/usr/bin/qemu")) ||
        (virFileExists("/usr/bin/qemu-kvm")) ||
1739
        (virFileExists("/usr/bin/xenner"))) {
1740
        if (getuid() == 0) {
1741 1742 1743 1744
            return("qemu:///system");
        } else {
            return("qemu:///session");
        }
1745 1746 1747
    }
    return(NULL);
}
1748

1749
static virDrvOpenStatus qemudOpen(virConnectPtr conn,
1750
                                  xmlURIPtr uri,
1751
                                  virConnectAuthPtr auth ATTRIBUTE_UNUSED,
1752
                                  int flags ATTRIBUTE_UNUSED) {
1753 1754 1755
    uid_t uid = getuid();

    if (qemu_driver == NULL)
1756
        goto decline;
1757

1758 1759 1760 1761 1762 1763
    if (uri == NULL || uri->scheme == NULL || uri->path == NULL)
        goto decline;

    if (STRNEQ (uri->scheme, "qemu"))
        goto decline;

1764
    if (uid != 0) {
1765 1766
        if (STRNEQ (uri->path, "/session"))
            goto decline;
1767
    } else { /* root */
1768 1769
        if (STRNEQ (uri->path, "/system") &&
            STRNEQ (uri->path, "/session"))
1770
            goto decline;
1771 1772 1773 1774 1775
    }

    conn->privateData = qemu_driver;

    return VIR_DRV_OPEN_SUCCESS;
1776 1777

 decline:
1778
    return VIR_DRV_OPEN_DECLINED;
1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789
}

static int qemudClose(virConnectPtr conn) {
    /*struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;*/

    conn->privateData = NULL;

    return 0;
}

static const char *qemudGetType(virConnectPtr conn ATTRIBUTE_UNUSED) {
1790
    return "QEMU";
1791 1792
}

1793
static int qemudGetMaxVCPUs(virConnectPtr conn, const char *type) {
1794 1795 1796
    if (!type)
        return 16;

1797
    if (STRCASEEQ(type, "qemu"))
1798 1799 1800 1801
        return 16;

    /* XXX future KVM will support SMP. Need to probe
       kernel to figure out KVM module version i guess */
1802
    if (STRCASEEQ(type, "kvm"))
1803 1804
        return 1;

1805
    if (STRCASEEQ(type, "kqemu"))
1806
        return 1;
1807 1808 1809

    qemudReportError(conn, NULL, NULL, VIR_ERR_INVALID_ARG,
                     _("unknown type '%s'"), type);
1810 1811 1812
    return -1;
}

1813 1814 1815
static int qemudGetNodeInfo(virConnectPtr conn,
                            virNodeInfoPtr nodeinfo) {
    return virNodeInfoPopulate(conn, nodeinfo);
1816 1817 1818
}


1819 1820 1821
static char *qemudGetCapabilities(virConnectPtr conn) {
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
    char *xml;
1822

1823
    if ((xml = virCapabilitiesFormatXML(driver->caps)) == NULL) {
1824 1825
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                 "%s", _("failed to allocate space for capabilities support"));
1826 1827 1828
        return NULL;
    }

1829
    return xml;
1830 1831 1832
}


1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887
#if HAVE_NUMACTL
static int
qemudNodeGetCellsFreeMemory(virConnectPtr conn,
                            unsigned long long *freeMems,
                            int startCell,
                            int maxCells)
{
    int n, lastCell, numCells;

    if (numa_available() < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("NUMA not supported on this host"));
        return -1;
    }
    lastCell = startCell + maxCells - 1;
    if (lastCell > numa_max_node())
        lastCell = numa_max_node();

    for (numCells = 0, n = startCell ; n <= lastCell ; n++) {
        long long mem;
        if (numa_node_size64(n, &mem) < 0) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("Failed to query NUMA free memory"));
            return -1;
        }
        freeMems[numCells++] = mem;
    }
    return numCells;
}

static unsigned long long
qemudNodeGetFreeMemory (virConnectPtr conn)
{
    unsigned long long freeMem = 0;
    int n;
    if (numa_available() < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("NUMA not supported on this host"));
        return -1;
    }

    for (n = 0 ; n <= numa_max_node() ; n++) {
        long long mem;
        if (numa_node_size64(n, &mem) < 0) {
            qemudReportError(conn, NULL, NULL, VIR_ERR_INTERNAL_ERROR,
                             "%s", _("Failed to query NUMA free memory"));
            return -1;
        }
        freeMem += mem;
    }

    return freeMem;
}

#endif
1888

D
Daniel P. Berrange 已提交
1889 1890 1891
static int qemudGetProcessInfo(unsigned long long *cpuTime, int pid) {
    char proc[PATH_MAX];
    FILE *pidinfo;
1892
    unsigned long long usertime, systime;
D
Daniel P. Berrange 已提交
1893 1894 1895 1896 1897 1898

    if (snprintf(proc, sizeof(proc), "/proc/%d/stat", pid) >= (int)sizeof(proc)) {
        return -1;
    }

    if (!(pidinfo = fopen(proc, "r"))) {
1899
        /*printf("cannot read pid info");*/
D
Daniel P. Berrange 已提交
1900 1901 1902 1903 1904
        /* VM probably shut down, so fake 0 */
        *cpuTime = 0;
        return 0;
    }

1905
    if (fscanf(pidinfo, "%*d %*s %*c %*d %*d %*d %*d %*d %*u %*u %*u %*u %*u %llu %llu", &usertime, &systime) != 2) {
1906
        qemudDebug("not enough arg");
D
Daniel P. Berrange 已提交
1907 1908 1909 1910 1911 1912 1913 1914
        return -1;
    }

    /* We got jiffies
     * We want nanoseconds
     * _SC_CLK_TCK is jiffies per second
     * So calulate thus....
     */
1915
    *cpuTime = 1000ull * 1000ull * 1000ull * (usertime + systime) / (unsigned long long)sysconf(_SC_CLK_TCK);
D
Daniel P. Berrange 已提交
1916

1917
    qemudDebug("Got %llu %llu %llu", usertime, systime, *cpuTime);
D
Daniel P. Berrange 已提交
1918 1919 1920 1921 1922 1923 1924

    fclose(pidinfo);

    return 0;
}


1925
static virDomainPtr qemudDomainLookupByID(virConnectPtr conn,
1926
                                          int id) {
1927
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
1928
    virDomainObjPtr vm = virDomainFindByID(driver->domains, id);
1929 1930 1931
    virDomainPtr dom;

    if (!vm) {
1932
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1933 1934 1935
        return NULL;
    }

1936
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1937
    if (dom) dom->id = vm->def->id;
1938 1939
    return dom;
}
1940
static virDomainPtr qemudDomainLookupByUUID(virConnectPtr conn,
1941
                                            const unsigned char *uuid) {
1942
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
1943
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, uuid);
1944 1945 1946
    virDomainPtr dom;

    if (!vm) {
1947
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1948 1949 1950
        return NULL;
    }

1951
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1952
    if (dom) dom->id = vm->def->id;
1953 1954
    return dom;
}
1955
static virDomainPtr qemudDomainLookupByName(virConnectPtr conn,
1956
                                            const char *name) {
1957
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
1958
    virDomainObjPtr vm = virDomainFindByName(driver->domains, name);
1959 1960 1961
    virDomainPtr dom;

    if (!vm) {
1962
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_DOMAIN, NULL);
1963 1964 1965
        return NULL;
    }

1966
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
1967
    if (dom) dom->id = vm->def->id;
1968 1969 1970
    return dom;
}

1971
static int qemudGetVersion(virConnectPtr conn, unsigned long *version) {
1972
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
1973
    if (qemudExtractVersion(conn, driver) < 0)
1974 1975
        return -1;

1976 1977
    *version = qemu_driver->qemuVersion;
    return 0;
D
Daniel P. Berrange 已提交
1978 1979
}

1980 1981 1982 1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995 1996 1997 1998 1999 2000 2001
static char *
qemudGetHostname (virConnectPtr conn)
{
    int r;
    char hostname[HOST_NAME_MAX+1], *str;

    r = gethostname (hostname, HOST_NAME_MAX+1);
    if (r == -1) {
        qemudReportError (conn, NULL, NULL, VIR_ERR_SYSTEM_ERROR,
                          "%s", strerror (errno));
        return NULL;
    }
    /* Caller frees this string. */
    str = strdup (hostname);
    if (str == NULL) {
        qemudReportError (conn, NULL, NULL, VIR_ERR_SYSTEM_ERROR,
                         "%s", strerror (errno));
        return NULL;
    }
    return str;
}

2002
static int qemudListDomains(virConnectPtr conn, int *ids, int nids) {
2003
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
2004
    virDomainObjPtr vm = driver->domains;
D
Daniel P. Berrange 已提交
2005 2006
    int got = 0;
    while (vm && got < nids) {
2007 2008
        if (virDomainIsActive(vm)) {
            ids[got] = vm->def->id;
2009 2010
            got++;
        }
D
Daniel P. Berrange 已提交
2011 2012 2013 2014
        vm = vm->next;
    }
    return got;
}
2015
static int qemudNumDomains(virConnectPtr conn) {
2016
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
2017 2018 2019 2020 2021 2022 2023 2024
    int n = 0;
    virDomainObjPtr dom = driver->domains;
    while (dom) {
        if (virDomainIsActive(dom))
            n++;
        dom = dom->next;
    }
    return n;
D
Daniel P. Berrange 已提交
2025
}
2026
static virDomainPtr qemudDomainCreate(virConnectPtr conn, const char *xml,
2027
                                      unsigned int flags ATTRIBUTE_UNUSED) {
2028 2029
    virDomainDefPtr def;
    virDomainObjPtr vm;
2030 2031
    virDomainPtr dom;
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
D
Daniel P. Berrange 已提交
2032

2033
    if (!(def = virDomainDefParseString(conn, driver->caps, xml)))
2034 2035
        return NULL;

2036
    vm = virDomainFindByName(driver->domains, def->name);
2037
    if (vm) {
2038
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2039
                         _("domain '%s' is already defined"),
2040 2041 2042 2043
                         def->name);
        virDomainDefFree(def);
        return NULL;
    }
2044 2045 2046 2047 2048 2049
    vm = virDomainFindByUUID(driver->domains, def->uuid);
    if (vm) {
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(def->uuid, uuidstr);
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2050
                         _("domain with uuid '%s' is already defined"),
2051 2052 2053 2054
                         uuidstr);
        virDomainDefFree(def);
        return NULL;
    }
2055

2056 2057 2058 2059
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
        virDomainDefFree(def);
D
Daniel P. Berrange 已提交
2060 2061 2062
        return NULL;
    }

2063 2064 2065
    if (qemudStartVMDaemon(conn, driver, vm, NULL) < 0) {
        virDomainRemoveInactive(&driver->domains,
                                vm);
D
Daniel P. Berrange 已提交
2066 2067 2068
        return NULL;
    }

2069
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2070
    if (dom) dom->id = vm->def->id;
2071
    return dom;
D
Daniel P. Berrange 已提交
2072 2073 2074
}


2075
static int qemudDomainSuspend(virDomainPtr dom) {
2076
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
D
Daniel P. Berrange 已提交
2077
    char *info;
2078
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
D
Daniel P. Berrange 已提交
2079
    if (!vm) {
2080
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN, _("no domain with matching id %d"), dom->id);
D
Daniel P. Berrange 已提交
2081 2082
        return -1;
    }
2083
    if (!virDomainIsActive(vm)) {
2084 2085
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
D
Daniel P. Berrange 已提交
2086 2087
        return -1;
    }
2088
    if (vm->state == VIR_DOMAIN_PAUSED)
D
Daniel P. Berrange 已提交
2089
        return 0;
D
Daniel P. Berrange 已提交
2090

2091
    if (qemudMonitorCommand(driver, vm, "stop", &info) < 0) {
2092 2093
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("suspend operation failed"));
D
Daniel P. Berrange 已提交
2094 2095
        return -1;
    }
2096
    vm->state = VIR_DOMAIN_PAUSED;
2097
    qemudDebug("Reply %s", info);
2098
    VIR_FREE(info);
D
Daniel P. Berrange 已提交
2099 2100 2101 2102
    return 0;
}


2103
static int qemudDomainResume(virDomainPtr dom) {
2104
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
D
Daniel P. Berrange 已提交
2105
    char *info;
2106
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
D
Daniel P. Berrange 已提交
2107
    if (!vm) {
2108 2109
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         _("no domain with matching id %d"), dom->id);
D
Daniel P. Berrange 已提交
2110 2111
        return -1;
    }
2112
    if (!virDomainIsActive(vm)) {
2113 2114
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
D
Daniel P. Berrange 已提交
2115 2116
        return -1;
    }
2117
    if (vm->state == VIR_DOMAIN_RUNNING)
D
Daniel P. Berrange 已提交
2118
        return 0;
2119
    if (qemudMonitorCommand(driver, vm, "cont", &info) < 0) {
2120 2121
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("resume operation failed"));
D
Daniel P. Berrange 已提交
2122 2123
        return -1;
    }
2124
    vm->state = VIR_DOMAIN_RUNNING;
2125
    qemudDebug("Reply %s", info);
2126
    VIR_FREE(info);
D
Daniel P. Berrange 已提交
2127
    return 0;
D
Daniel P. Berrange 已提交
2128 2129 2130
}


2131 2132
static int qemudDomainShutdown(virDomainPtr dom) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2133
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
2134 2135 2136 2137
    char* info;

    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2138
                         _("no domain with matching id %d"), dom->id);
2139 2140 2141 2142 2143
        return -1;
    }

    if (qemudMonitorCommand(driver, vm, "system_powerdown", &info) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2144
                         "%s", _("shutdown operation failed"));
2145 2146 2147 2148 2149 2150 2151
        return -1;
    }
    return 0;

}


2152
static int qemudDomainDestroy(virDomainPtr dom) {
2153
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2154
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
2155

D
Daniel P. Berrange 已提交
2156
    if (!vm) {
2157
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2158
                         _("no domain with matching id %d"), dom->id);
D
Daniel P. Berrange 已提交
2159 2160
        return -1;
    }
2161

2162
    qemudShutdownVMDaemon(dom->conn, driver, vm);
2163
    if (!vm->persistent)
2164 2165
        virDomainRemoveInactive(&driver->domains,
                                vm);
2166

2167
    return 0;
D
Daniel P. Berrange 已提交
2168 2169 2170
}


2171 2172
static char *qemudDomainGetOSType(virDomainPtr dom) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2173
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2174 2175 2176 2177
    char *type;

    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2178
                         "%s", _("no domain with matching uuid"));
2179 2180 2181 2182
        return NULL;
    }

    if (!(type = strdup(vm->def->os.type))) {
2183 2184
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_MEMORY,
                         "%s", _("failed to allocate space for ostype"));
2185 2186 2187 2188 2189
        return NULL;
    }
    return type;
}

2190 2191 2192
/* Returns max memory in kb, 0 if error */
static unsigned long qemudDomainGetMaxMemory(virDomainPtr dom) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2193
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2194 2195

    if (!vm) {
2196 2197 2198
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2199
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2200
                         _("no domain with matching uuid '%s'"), uuidstr);
2201 2202 2203 2204 2205 2206 2207 2208
        return 0;
    }

    return vm->def->maxmem;
}

static int qemudDomainSetMaxMemory(virDomainPtr dom, unsigned long newmax) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2209
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2210 2211

    if (!vm) {
2212 2213 2214
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2215
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2216
                         _("no domain with matching uuid '%s'"), uuidstr);
2217 2218 2219 2220 2221
        return -1;
    }

    if (newmax < vm->def->memory) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
2222
                         "%s", _("cannot set max memory lower than current memory"));
2223 2224 2225 2226 2227 2228 2229 2230 2231
        return -1;
    }

    vm->def->maxmem = newmax;
    return 0;
}

static int qemudDomainSetMemory(virDomainPtr dom, unsigned long newmem) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2232
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2233 2234

    if (!vm) {
2235 2236 2237
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2238
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2239
                         _("no domain with matching uuid '%s'"), uuidstr);
2240 2241 2242
        return -1;
    }

2243
    if (virDomainIsActive(vm)) {
2244
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
2245
                         "%s", _("cannot set memory of an active domain"));
2246 2247 2248 2249 2250
        return -1;
    }

    if (newmem > vm->def->maxmem) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
2251
                         "%s", _("cannot set memory higher than max memory"));
2252 2253 2254 2255 2256 2257 2258
        return -1;
    }

    vm->def->memory = newmem;
    return 0;
}

2259
static int qemudDomainGetInfo(virDomainPtr dom,
2260
                              virDomainInfoPtr info) {
2261
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2262
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
2263
    if (!vm) {
2264 2265
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
D
Daniel P. Berrange 已提交
2266 2267 2268
        return -1;
    }

2269
    info->state = vm->state;
D
Daniel P. Berrange 已提交
2270

2271
    if (!virDomainIsActive(vm)) {
2272
        info->cpuTime = 0;
D
Daniel P. Berrange 已提交
2273
    } else {
2274
        if (qemudGetProcessInfo(&(info->cpuTime), vm->pid) < 0) {
2275
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED, ("cannot read cputime for domain"));
D
Daniel P. Berrange 已提交
2276 2277 2278 2279
            return -1;
        }
    }

2280 2281 2282
    info->maxMem = vm->def->maxmem;
    info->memory = vm->def->memory;
    info->nrVirtCpu = vm->def->vcpus;
D
Daniel P. Berrange 已提交
2283 2284 2285 2286
    return 0;
}


D
Daniel P. Berrange 已提交
2287
static char *qemudEscape(const char *in, int shell)
2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308
{
    int len = 0;
    int i, j;
    char *out;

    /* To pass through the QEMU monitor, we need to use escape
       sequences: \r, \n, \", \\

       To pass through both QEMU + the shell, we need to escape
       the single character ' as the five characters '\\''
    */

    for (i = 0; in[i] != '\0'; i++) {
        switch(in[i]) {
        case '\r':
        case '\n':
        case '"':
        case '\\':
            len += 2;
            break;
        case '\'':
D
Daniel P. Berrange 已提交
2309 2310 2311 2312
            if (shell)
                len += 5;
            else
                len += 1;
2313 2314 2315 2316 2317 2318 2319
            break;
        default:
            len += 1;
            break;
        }
    }

2320
    if (VIR_ALLOC_N(out, len + 1) < 0)
2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338
        return NULL;

    for (i = j = 0; in[i] != '\0'; i++) {
        switch(in[i]) {
        case '\r':
            out[j++] = '\\';
            out[j++] = 'r';
            break;
        case '\n':
            out[j++] = '\\';
            out[j++] = 'n';
            break;
        case '"':
        case '\\':
            out[j++] = '\\';
            out[j++] = in[i];
            break;
        case '\'':
D
Daniel P. Berrange 已提交
2339 2340 2341 2342 2343 2344 2345 2346 2347
            if (shell) {
                out[j++] = '\'';
                out[j++] = '\\';
                out[j++] = '\\';
                out[j++] = '\'';
                out[j++] = '\'';
            } else {
                out[j++] = in[i];
            }
2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358
            break;
        default:
            out[j++] = in[i];
            break;
        }
    }
    out[j] = '\0';

    return out;
}

2359 2360 2361 2362 2363
static char *qemudEscapeMonitorArg(const char *in)
{
    return qemudEscape(in, 0);
}

D
Daniel P. Berrange 已提交
2364 2365 2366 2367
static char *qemudEscapeShellArg(const char *in)
{
    return qemudEscape(in, 1);
}
2368

2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379
#define QEMUD_SAVE_MAGIC "LibvirtQemudSave"
#define QEMUD_SAVE_VERSION 1

struct qemud_save_header {
    char magic[sizeof(QEMUD_SAVE_MAGIC)-1];
    int version;
    int xml_len;
    int was_running;
    int unused[16];
};

2380
static int qemudDomainSave(virDomainPtr dom,
2381
                           const char *path) {
2382
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2383
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
2384 2385 2386 2387 2388 2389 2390 2391 2392 2393
    char *command, *info;
    int fd;
    char *safe_path;
    char *xml;
    struct qemud_save_header header;

    memset(&header, 0, sizeof(header));
    memcpy(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic));
    header.version = QEMUD_SAVE_VERSION;

D
Daniel P. Berrange 已提交
2394
    if (!vm) {
2395
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2396
                         _("no domain with matching id %d"), dom->id);
D
Daniel P. Berrange 已提交
2397 2398
        return -1;
    }
2399

2400
    if (!virDomainIsActive(vm)) {
2401
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2402
                         "%s", _("domain is not running"));
D
Daniel P. Berrange 已提交
2403 2404
        return -1;
    }
2405 2406 2407 2408 2409 2410

    /* Pause */
    if (vm->state == VIR_DOMAIN_RUNNING) {
        header.was_running = 1;
        if (qemudDomainSuspend(dom) != 0) {
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2411
                             "%s", _("failed to pause domain"));
2412 2413 2414 2415 2416
            return -1;
        }
    }

    /* Get XML for the domain */
2417
    xml = virDomainDefFormat(dom->conn, vm->def, VIR_DOMAIN_XML_SECURE);
2418 2419
    if (!xml) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2420
                         "%s", _("failed to get domain xml"));
2421 2422 2423 2424 2425 2426 2427
        return -1;
    }
    header.xml_len = strlen(xml) + 1;

    /* Write header to file, followed by XML */
    if ((fd = open(path, O_CREAT|O_TRUNC|O_WRONLY, S_IRUSR|S_IWUSR)) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2428
                         _("failed to create '%s'"), path);
2429
        VIR_FREE(xml);
2430 2431 2432 2433 2434
        return -1;
    }

    if (safewrite(fd, &header, sizeof(header)) != sizeof(header)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2435
                         "%s", _("failed to write save header"));
2436
        close(fd);
2437
        VIR_FREE(xml);
2438 2439 2440 2441 2442
        return -1;
    }

    if (safewrite(fd, xml, header.xml_len) != header.xml_len) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2443
                         "%s", _("failed to write xml"));
2444
        close(fd);
2445
        VIR_FREE(xml);
2446 2447 2448 2449
        return -1;
    }

    close(fd);
2450
    VIR_FREE(xml);
2451 2452 2453 2454 2455

    /* Migrate to file */
    safe_path = qemudEscapeShellArg(path);
    if (!safe_path) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2456
                         "%s", _("out of memory"));
2457 2458 2459 2460
        return -1;
    }
    if (asprintf (&command, "migrate \"exec:"
                  "dd of='%s' oflag=append conv=notrunc 2>/dev/null"
2461
                  "\"", safe_path) == -1) {
2462
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2463
                         "%s", _("out of memory"));
2464
        VIR_FREE(safe_path);
2465 2466 2467 2468 2469 2470
        return -1;
    }
    free(safe_path);

    if (qemudMonitorCommand(driver, vm, command, &info) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2471
                         "%s", _("migrate operation failed"));
2472
        VIR_FREE(command);
2473 2474 2475
        return -1;
    }

2476 2477 2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488
    DEBUG ("migrate reply: %s", info);

    /* If the command isn't supported then qemu prints:
     * unknown command: migrate" */
    if (strstr(info, "unknown command:")) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                          "%s",
                          _("'migrate' not supported by this qemu"));
        VIR_FREE(info);
        VIR_FREE(command);
        return -1;
    }

2489 2490
    VIR_FREE(info);
    VIR_FREE(command);
2491 2492 2493

    /* Shut it down */
    qemudShutdownVMDaemon(dom->conn, driver, vm);
2494
    if (!vm->persistent)
2495 2496
        virDomainRemoveInactive(&driver->domains,
                                vm);
2497 2498

    return 0;
D
Daniel P. Berrange 已提交
2499 2500 2501
}


2502 2503
static int qemudDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus) {
    const struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2504
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2505 2506 2507
    int max;

    if (!vm) {
2508 2509 2510
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2511
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2512
                         _("no domain with matching uuid '%s'"), uuidstr);
2513 2514 2515
        return -1;
    }

2516
    if (virDomainIsActive(vm)) {
2517
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT, "%s",
2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538
                         _("cannot change vcpu count of an active domain"));
        return -1;
    }

    if ((max = qemudDomainGetMaxVcpus(dom)) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR, "%s",
                         _("could not determine max vcpus for the domain"));
        return -1;
    }

    if (nvcpus > max) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         _("requested vcpus is greater than max allowable"
                           " vcpus for the domain: %d > %d"), nvcpus, max);
        return -1;
    }

    vm->def->vcpus = nvcpus;
    return 0;
}

2539 2540 2541 2542 2543 2544 2545 2546

#if HAVE_SCHED_GETAFFINITY
static int
qemudDomainPinVcpu(virDomainPtr dom,
                   unsigned int vcpu,
                   unsigned char *cpumap,
                   int maplen) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2547
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2548 2549 2550 2551
    cpu_set_t mask;
    int i, maxcpu;
    virNodeInfo nodeinfo;

2552
    if (!virDomainIsActive(vm)) {
2553 2554 2555 2556 2557 2558 2559 2560 2561 2562 2563 2564 2565 2566 2567 2568 2569 2570 2571 2572 2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         "%s",_("cannot pin vcpus on an inactive domain"));
        return -1;
    }

    if (vcpu > (vm->nvcpupids-1)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         _("vcpu number out of range %d > %d"),
                         vcpu, vm->nvcpupids);
        return -1;
    }

    if (virNodeInfoPopulate(dom->conn, &nodeinfo) < 0)
        return -1;

    maxcpu = maplen * 8;
    if (maxcpu > nodeinfo.cpus)
        maxcpu = nodeinfo.cpus;

    CPU_ZERO(&mask);
    for (i = 0 ; i < maxcpu ; i++) {
        if ((cpumap[i/8] >> (i % 8)) & 1)
            CPU_SET(i, &mask);
    }

    if (vm->vcpupids != NULL) {
        if (sched_setaffinity(vm->vcpupids[vcpu], sizeof(mask), &mask) < 0) {
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                             _("cannot set affinity: %s"), strerror(errno));
            return -1;
        }
    } else {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("cpu affinity is not supported"));
        return -1;
    }

    return 0;
}

static int
qemudDomainGetVcpus(virDomainPtr dom,
                    virVcpuInfoPtr info,
                    int maxinfo,
                    unsigned char *cpumaps,
                    int maplen) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2600
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2601 2602 2603
    virNodeInfo nodeinfo;
    int i, v, maxcpu;

2604
    if (!virDomainIsActive(vm)) {
2605 2606 2607 2608 2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639 2640 2641 2642 2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         "%s",_("cannot pin vcpus on an inactive domain"));
        return -1;
    }

    if (virNodeInfoPopulate(dom->conn, &nodeinfo) < 0)
        return -1;

    maxcpu = maplen * 8;
    if (maxcpu > nodeinfo.cpus)
        maxcpu = nodeinfo.cpus;

    /* Clamp to actual number of vcpus */
    if (maxinfo > vm->nvcpupids)
        maxinfo = vm->nvcpupids;

    if (maxinfo < 1)
        return 0;

    if (info != NULL) {
        memset(info, 0, sizeof(*info) * maxinfo);
        for (i = 0 ; i < maxinfo ; i++) {
            info[i].number = i;
            info[i].state = VIR_VCPU_RUNNING;
            /* XXX cpu time, current pCPU mapping */
        }
    }

    if (cpumaps != NULL) {
        memset(cpumaps, 0, maplen * maxinfo);
        if (vm->vcpupids != NULL) {
            for (v = 0 ; v < maxinfo ; v++) {
                cpu_set_t mask;
                unsigned char *cpumap = VIR_GET_CPUMAP(cpumaps, maplen, v);
                CPU_ZERO(&mask);

                if (sched_getaffinity(vm->vcpupids[v], sizeof(mask), &mask) < 0) {
                    qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                                     _("cannot get affinity: %s"), strerror(errno));
                    return -1;
                }

                for (i = 0 ; i < maxcpu ; i++)
                    if (CPU_ISSET(i, &mask))
                        VIR_USE_CPU(cpumap, i);
            }
        } else {
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                             "%s", _("cpu affinity is not available"));
            return -1;
        }
    }

    return maxinfo;
}
#endif /* HAVE_SCHED_GETAFFINITY */


2663 2664
static int qemudDomainGetMaxVcpus(virDomainPtr dom) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2665
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2666 2667 2668 2669
    const char *type;
    int ret;

    if (!vm) {
2670 2671 2672
        char uuidstr[VIR_UUID_STRING_BUFLEN];

        virUUIDFormat(dom->uuid, uuidstr);
2673
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2674
                         _("no domain with matching uuid '%s'"), uuidstr);
2675 2676 2677
        return -1;
    }

2678
    if (!(type = virDomainVirtTypeToString(vm->def->virtType))) {
2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690 2691 2692
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         _("unknown virt type in domain definition '%d'"),
                         vm->def->virtType);
        return -1;
    }

    if ((ret = qemudGetMaxVCPUs(dom->conn, type)) < 0) {
        return -1;
    }

    return ret;
}


2693
static int qemudDomainRestore(virConnectPtr conn,
2694 2695
                       const char *path) {
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
2696 2697
    virDomainDefPtr def;
    virDomainObjPtr vm;
2698
    int fd;
2699
    int ret;
2700 2701 2702 2703 2704 2705
    char *xml;
    struct qemud_save_header header;

    /* Verify the header and read the XML */
    if ((fd = open(path, O_RDONLY)) < 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2706
                         "%s", _("cannot read domain image"));
2707 2708 2709 2710 2711
        return -1;
    }

    if (saferead(fd, &header, sizeof(header)) != sizeof(header)) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2712
                         "%s", _("failed to read qemu header"));
2713 2714 2715 2716 2717 2718
        close(fd);
        return -1;
    }

    if (memcmp(header.magic, QEMUD_SAVE_MAGIC, sizeof(header.magic)) != 0) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2719
                         "%s", _("image magic is incorrect"));
2720 2721 2722 2723 2724 2725
        close(fd);
        return -1;
    }

    if (header.version > QEMUD_SAVE_VERSION) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2726
                         _("image version is not supported (%d > %d)"),
2727 2728 2729 2730 2731
                         header.version, QEMUD_SAVE_VERSION);
        close(fd);
        return -1;
    }

2732
    if (VIR_ALLOC_N(xml, header.xml_len) < 0) {
2733
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2734
                         "%s", _("out of memory"));
2735 2736 2737 2738 2739 2740
        close(fd);
        return -1;
    }

    if (saferead(fd, xml, header.xml_len) != header.xml_len) {
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2741
                         "%s", _("failed to read XML"));
2742
        close(fd);
2743
        VIR_FREE(xml);
2744 2745 2746 2747
        return -1;
    }

    /* Create a domain from this XML */
2748
    if (!(def = virDomainDefParseString(conn, driver->caps, xml))) {
2749
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2750
                         "%s", _("failed to parse XML"));
2751
        close(fd);
2752
        VIR_FREE(xml);
2753 2754
        return -1;
    }
2755
    VIR_FREE(xml);
2756 2757

    /* Ensure the name and UUID don't already exist in an active VM */
2758 2759 2760 2761
    vm = virDomainFindByUUID(driver->domains, def->uuid);
    if (!vm)
        vm = virDomainFindByName(driver->domains, def->name);
    if (vm && virDomainIsActive(vm)) {
2762
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2763
                         _("domain is already active as '%s'"), vm->def->name);
2764 2765 2766 2767
        close(fd);
        return -1;
    }

2768 2769 2770
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
2771
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2772
                         "%s", _("failed to assign new VM"));
2773
        virDomainDefFree(def);
2774 2775 2776 2777 2778
        close(fd);
        return -1;
    }

    /* Set the migration source and start it up. */
2779
    vm->stdin_fd = fd;
2780
    ret = qemudStartVMDaemon(conn, driver, vm, "stdio");
2781
    close(fd);
2782
    vm->stdin_fd = -1;
2783
    if (ret < 0) {
2784
        qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2785
                         "%s", _("failed to start VM"));
2786
        if (!vm->persistent)
2787 2788
            virDomainRemoveInactive(&driver->domains,
                                    vm);
2789 2790 2791 2792 2793 2794
        return -1;
    }

    /* If it was running before, resume it now. */
    if (header.was_running) {
        char *info;
2795
        if (qemudMonitorCommand(driver, vm, "cont", &info) < 0) {
2796
            qemudReportError(conn, NULL, NULL, VIR_ERR_OPERATION_FAILED,
2797
                             "%s", _("failed to resume domain"));
2798 2799
            return -1;
        }
2800
        VIR_FREE(info);
2801 2802 2803 2804
        vm->state = VIR_DOMAIN_RUNNING;
    }

    return 0;
D
Daniel P. Berrange 已提交
2805 2806 2807
}


2808
static char *qemudDomainDumpXML(virDomainPtr dom,
2809
                                int flags ATTRIBUTE_UNUSED) {
2810
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2811
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
2812
    if (!vm) {
2813 2814
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
2815
        return NULL;
D
Daniel P. Berrange 已提交
2816 2817
    }

2818 2819 2820 2821
    return virDomainDefFormat(dom->conn,
                              (flags & VIR_DOMAIN_XML_INACTIVE) && vm->newDef ?
                              vm->newDef : vm->def,
                              flags);
D
Daniel P. Berrange 已提交
2822 2823 2824
}


2825
static int qemudListDefinedDomains(virConnectPtr conn,
2826 2827
                            char **const names, int nnames) {
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
2828
    virDomainObjPtr vm = driver->domains;
2829
    int got = 0, i;
D
Daniel P. Berrange 已提交
2830
    while (vm && got < nnames) {
2831
        if (!virDomainIsActive(vm)) {
2832
            if (!(names[got] = strdup(vm->def->name))) {
2833 2834
                qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                     "%s", _("failed to allocate space for VM name string"));
2835 2836
                goto cleanup;
            }
2837 2838
            got++;
        }
D
Daniel P. Berrange 已提交
2839 2840 2841
        vm = vm->next;
    }
    return got;
2842 2843 2844

 cleanup:
    for (i = 0 ; i < got ; i++)
2845
        VIR_FREE(names[i]);
2846
    return -1;
D
Daniel P. Berrange 已提交
2847 2848 2849
}


2850
static int qemudNumDefinedDomains(virConnectPtr conn) {
2851
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
2852 2853 2854 2855 2856 2857 2858 2859
    int n = 0;
    virDomainObjPtr dom = driver->domains;
    while (dom) {
        if (!virDomainIsActive(dom))
            n++;
        dom = dom->next;
    }
    return n;
D
Daniel P. Berrange 已提交
2860 2861 2862
}


2863
static int qemudDomainStart(virDomainPtr dom) {
2864
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2865
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
2866 2867

    if (!vm) {
2868
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
2869
                         "%s", _("no domain with matching uuid"));
2870
        return -1;
2871 2872
    }

2873
    return qemudStartVMDaemon(dom->conn, driver, vm, NULL);
D
Daniel P. Berrange 已提交
2874 2875 2876
}


2877
static virDomainPtr qemudDomainDefine(virConnectPtr conn, const char *xml) {
2878
    struct qemud_driver *driver = (struct qemud_driver *)conn->privateData;
2879 2880
    virDomainDefPtr def;
    virDomainObjPtr vm;
2881
    virDomainPtr dom;
2882

2883
    if (!(def = virDomainDefParseString(conn, driver->caps, xml)))
2884 2885
        return NULL;

2886 2887 2888 2889
    if (!(vm = virDomainAssignDef(conn,
                                  &driver->domains,
                                  def))) {
        virDomainDefFree(def);
2890 2891
        return NULL;
    }
2892
    vm->persistent = 1;
2893

2894 2895
    if (virDomainSaveConfig(conn,
                            driver->configDir,
2896
                            vm->newDef ? vm->newDef : vm->def) < 0) {
2897 2898
        virDomainRemoveInactive(&driver->domains,
                                vm);
2899 2900 2901
        return NULL;
    }

2902
    dom = virGetDomain(conn, vm->def->name, vm->def->uuid);
2903
    if (dom) dom->id = vm->def->id;
2904
    return dom;
D
Daniel P. Berrange 已提交
2905 2906
}

2907
static int qemudDomainUndefine(virDomainPtr dom) {
2908
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2909
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
D
Daniel P. Berrange 已提交
2910 2911

    if (!vm) {
2912 2913
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
D
Daniel P. Berrange 已提交
2914 2915 2916
        return -1;
    }

2917
    if (virDomainIsActive(vm)) {
2918 2919
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot delete active domain"));
D
Daniel P. Berrange 已提交
2920 2921 2922
        return -1;
    }

2923 2924 2925 2926 2927 2928 2929
    if (!vm->persistent) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot undefine transient domain"));
        return -1;
    }

    if (virDomainDeleteConfig(dom->conn, driver->configDir, driver->autostartDir, vm) < 0)
D
Daniel P. Berrange 已提交
2930 2931
        return -1;

2932 2933
    virDomainRemoveInactive(&driver->domains,
                            vm);
D
Daniel P. Berrange 已提交
2934 2935 2936 2937

    return 0;
}

2938
static int qemudDomainChangeCDROM(virDomainPtr dom,
2939 2940 2941
                                  virDomainObjPtr vm,
                                  virDomainDiskDefPtr olddisk,
                                  virDomainDiskDefPtr newdisk) {
2942 2943 2944
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
    char *cmd, *reply, *safe_path;

2945
    if (newdisk->src) {
2946 2947 2948
        safe_path = qemudEscapeMonitorArg(newdisk->src);
        if (!safe_path) {
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2949
                             "%s", _("out of memory"));
2950 2951 2952 2953 2954 2955 2956
            return -1;
        }
        if (asprintf (&cmd, "change %s \"%s\"",
                      /* XXX qemu may support multiple CDROM in future */
                      /* olddisk->dst */ "cdrom",
                      safe_path) == -1) {
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2957
                             "%s", _("out of memory"));
2958
            VIR_FREE(safe_path);
2959 2960
            return -1;
        }
2961
        VIR_FREE(safe_path);
2962 2963

    } else if (asprintf(&cmd, "eject cdrom") == -1) {
2964
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
2965
                         "%s", _("out of memory"));
2966 2967 2968 2969
        return -1;
    }

    if (qemudMonitorCommand(driver, vm, cmd, &reply) < 0) {
2970 2971
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("cannot change cdrom media"));
2972
        VIR_FREE(cmd);
2973 2974
        return -1;
    }
2975 2976
    VIR_FREE(reply);
    VIR_FREE(cmd);
2977
    strcpy(olddisk->src, newdisk->src);
2978 2979 2980 2981
    olddisk->type = newdisk->type;
    return 0;
}

2982 2983 2984
static int qemudDomainAttachCdromDevice(virDomainPtr dom,
                                        virDomainDeviceDefPtr dev)
{
2985
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
2986 2987
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
    virDomainDiskDefPtr disk;
2988 2989

    if (!vm) {
2990 2991
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
2992 2993 2994 2995 2996
        return -1;
    }

    disk = vm->def->disks;
    while (disk) {
2997 2998
        if (disk->device == VIR_DOMAIN_DISK_DEVICE_CDROM &&
            STREQ(disk->dst, dev->data.disk->dst))
2999 3000 3001 3002 3003
            break;
        disk = disk->next;
    }

    if (!disk) {
3004 3005
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                         "%s", _("CDROM not attached, cannot change media"));
3006 3007 3008
        return -1;
    }

3009
    if (qemudDomainChangeCDROM(dom, vm, disk, dev->data.disk) < 0) {
3010 3011
        return -1;
    }
3012 3013
    return 0;
}
3014

3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 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
static int qemudDomainAttachUsbMassstorageDevice(virDomainPtr dom, virDomainDeviceDefPtr dev)
{
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
    int ret;
    char *cmd, *reply;

    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
        return -1;
    }

    ret = asprintf(&cmd, "usb_add disk:%s", dev->data.disk->src);
    if (ret == -1) {
        qemudReportError(dom->conn, NULL, NULL, VIR_ERR_NO_MEMORY, NULL);
        return ret;
    }

    if (qemudMonitorCommand(driver, vm, cmd, &reply) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("cannot attach usb device"));
        VIR_FREE(cmd);
        return -1;
    }

    DEBUG ("attach_usb reply: %s", reply);
    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s",
                          _("adding usb device failed"));
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
    VIR_FREE(reply);
    VIR_FREE(cmd);
    return 0;
}

3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103
static int qemudDomainAttachHostDevice(virDomainPtr dom, virDomainDeviceDefPtr dev)
{
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
    int ret;
    char *cmd, *reply;

    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
        return -1;
    }

    if (dev->data.hostdev->source.subsys.usb.vendor) {
        ret = asprintf(&cmd, "usb_add host:%.4x:%.4x",
                       dev->data.hostdev->source.subsys.usb.vendor,
                       dev->data.hostdev->source.subsys.usb.product);
    } else {
        ret = asprintf(&cmd, "usb_add host:%.3d.%.3d",
                       dev->data.hostdev->source.subsys.usb.bus,
                       dev->data.hostdev->source.subsys.usb.device);
    }
    if (ret == -1) {
        qemudReportError(dom->conn, NULL, NULL, VIR_ERR_NO_MEMORY, NULL);
        return -1;
    }

    if (qemudMonitorCommand(driver, vm, cmd, &reply) < 0) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("cannot attach usb device"));
        VIR_FREE(cmd);
        return -1;
    }

    DEBUG ("attach_usb reply: %s", reply);
    /* If the command failed qemu prints:
     * Could not add ... */
    if (strstr(reply, "Could not add ")) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s",
                          _("adding usb device failed"));
        VIR_FREE(reply);
        VIR_FREE(cmd);
        return -1;
    }
    VIR_FREE(reply);
    VIR_FREE(cmd);
3104 3105 3106
    return 0;
}

3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132 3133
static int qemudDomainAttachDevice(virDomainPtr dom,
                                   const char *xml) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
    virDomainDeviceDefPtr dev;
    int ret = 0;

    if (!vm) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
        return -1;
    }

    if (!virDomainIsActive(vm)) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot attach device on inactive domain"));
        return -1;
    }

    dev = virDomainDeviceDefParse(dom->conn, vm->def, xml);
    if (dev == NULL) {
        return -1;
    }

    if (dev->type == VIR_DOMAIN_DEVICE_DISK &&
        dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_CDROM) {
                ret = qemudDomainAttachCdromDevice(dom, dev);
3134 3135 3136 3137
    } else if (dev->type == VIR_DOMAIN_DEVICE_DISK &&
        dev->data.disk->device == VIR_DOMAIN_DEVICE_DISK &&
        dev->data.disk->bus == VIR_DOMAIN_DISK_BUS_USB) {
                ret = qemudDomainAttachUsbMassstorageDevice(dom, dev);
3138 3139 3140 3141 3142 3143
    } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV &&
        dev->data.hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS &&
        dev->data.hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
                ret = qemudDomainAttachHostDevice(dom, dev);
    } else {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
3144
                         "%s", _("this devicetype cannot be attached"));
3145 3146 3147 3148 3149 3150 3151
        ret = -1;
    }

    VIR_FREE(dev);
    return ret;
}

3152
static int qemudDomainGetAutostart(virDomainPtr dom,
3153 3154
                            int *autostart) {
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
3155
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
3156 3157

    if (!vm) {
3158 3159
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3160 3161 3162 3163 3164 3165 3166 3167
        return -1;
    }

    *autostart = vm->autostart;

    return 0;
}

3168
static int qemudDomainSetAutostart(virDomainPtr dom,
3169
                                   int autostart) {
3170
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
3171
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
3172 3173
    char *configFile = NULL, *autostartLink = NULL;
    int ret = -1;
3174 3175

    if (!vm) {
3176 3177
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no domain with matching uuid"));
3178 3179 3180
        return -1;
    }

3181 3182 3183 3184 3185 3186
    if (!vm->persistent) {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("cannot set autostart for transient domain"));
        return -1;
    }

3187 3188 3189 3190 3191
    autostart = (autostart != 0);

    if (vm->autostart == autostart)
        return 0;

3192 3193 3194 3195 3196
    if ((configFile = virDomainConfigFile(dom->conn, driver->configDir, vm->def->name)) == NULL)
        goto cleanup;
    if ((autostartLink = virDomainConfigFile(dom->conn, driver->autostartDir, vm->def->name)) == NULL)
        goto cleanup;

3197 3198 3199
    if (autostart) {
        int err;

3200
        if ((err = virFileMakePath(driver->autostartDir))) {
3201
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3202
                             _("cannot create autostart directory %s: %s"),
3203
                             driver->autostartDir, strerror(err));
3204
            goto cleanup;
3205 3206
        }

3207
        if (symlink(configFile, autostartLink) < 0) {
3208
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3209 3210 3211
                             _("Failed to create symlink '%s to '%s': %s"),
                             autostartLink, configFile, strerror(errno));
            goto cleanup;
3212 3213
        }
    } else {
3214
        if (unlink(autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
3215
            qemudReportError(dom->conn, dom, NULL, VIR_ERR_INTERNAL_ERROR,
3216
                             _("Failed to delete symlink '%s': %s"),
3217 3218
                             autostartLink, strerror(errno));
            goto cleanup;
3219 3220 3221
        }
    }

3222
    vm->autostart = autostart;
3223
    ret = 0;
3224

3225 3226 3227 3228 3229
cleanup:
    VIR_FREE(configFile);
    VIR_FREE(autostartLink);

    return ret;
3230 3231
}

3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246
/* This uses the 'info blockstats' monitor command which was
 * integrated into both qemu & kvm in late 2007.  If the command is
 * not supported we detect this and return the appropriate error.
 */
static int
qemudDomainBlockStats (virDomainPtr dom,
                       const char *path,
                       struct _virDomainBlockStats *stats)
{
    const struct qemud_driver *driver =
        (struct qemud_driver *)dom->conn->privateData;
    char *dummy, *info;
    const char *p, *eol;
    char qemu_dev_name[32];
    size_t len;
3247
    const virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
3248 3249 3250 3251 3252 3253

    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching id %d"), dom->id);
        return -1;
    }
3254
    if (!virDomainIsActive (vm)) {
3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("domain is not running"));
        return -1;
    }

    /*
     * QEMU internal block device names are different from the device
     * names we use in libvirt, so we need to map between them:
     *
     *   hd[a-]   to  ide0-hd[0-]
     *   cdrom    to  ide1-cd0
     *   fd[a-]   to  floppy[0-]
     */
J
Jim Meyering 已提交
3268
    if (STRPREFIX (path, "hd") && c_islower(path[2]))
3269 3270 3271 3272
        snprintf (qemu_dev_name, sizeof (qemu_dev_name),
                  "ide0-hd%d", path[2] - 'a');
    else if (STREQ (path, "cdrom"))
        strcpy (qemu_dev_name, "ide1-cd0");
J
Jim Meyering 已提交
3273
    else if (STRPREFIX (path, "fd") && c_islower(path[2]))
3274 3275 3276 3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287 3288 3289 3290 3291 3292 3293 3294 3295 3296
        snprintf (qemu_dev_name, sizeof (qemu_dev_name),
                  "floppy%d", path[2] - 'a');
    else {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          _("invalid path: %s"), path);
        return -1;
    }

    len = strlen (qemu_dev_name);

    if (qemudMonitorCommand (driver, vm, "info blockstats", &info) < 0) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("'info blockstats' command failed"));
        return -1;
    }

    DEBUG ("info blockstats reply: %s", info);

    /* If the command isn't supported then qemu prints the supported
     * info commands, so the output starts "info ".  Since this is
     * unlikely to be the name of a block device, we can use this
     * to detect if qemu supports the command.
     */
3297
    if (STRPREFIX (info, "info ")) {
3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328
        free (info);
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                          "%s",
                          _("'info blockstats' not supported by this qemu"));
        return -1;
    }

    stats->rd_req = -1;
    stats->rd_bytes = -1;
    stats->wr_req = -1;
    stats->wr_bytes = -1;
    stats->errs = -1;

    /* The output format for both qemu & KVM is:
     *   blockdevice: rd_bytes=% wr_bytes=% rd_operations=% wr_operations=%
     *   (repeated for each block device)
     * where '%' is a 64 bit number.
     */
    p = info;

    while (*p) {
        if (STREQLEN (p, qemu_dev_name, len)
            && p[len] == ':' && p[len+1] == ' ') {

            eol = strchr (p, '\n');
            if (!eol)
                eol = p + strlen (p);

            p += len+2;         /* Skip to first label. */

            while (*p) {
3329
                if (STRPREFIX (p, "rd_bytes=")) {
3330 3331 3332
                    p += 9;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->rd_bytes) == -1)
                        DEBUG ("error reading rd_bytes: %s", p);
3333
                } else if (STRPREFIX (p, "wr_bytes=")) {
3334 3335 3336
                    p += 9;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->wr_bytes) == -1)
                        DEBUG ("error reading wr_bytes: %s", p);
3337
                } else if (STRPREFIX (p, "rd_operations=")) {
3338 3339 3340
                    p += 14;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->rd_req) == -1)
                        DEBUG ("error reading rd_req: %s", p);
3341
                } else if (STRPREFIX (p, "wr_operations=")) {
3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373
                    p += 14;
                    if (virStrToLong_ll (p, &dummy, 10, &stats->wr_req) == -1)
                        DEBUG ("error reading wr_req: %s", p);
                } else
                    DEBUG ("unknown block stat near %s", p);

                /* Skip to next label. */
                p = strchr (p, ' ');
                if (!p || p >= eol) break;
                p++;
            }

            goto done;
        }

        /* Skip to next line. */
        p = strchr (p, '\n');
        if (!p) break;
        p++;
    }

    /* If we reach here then the device was not found. */
    free (info);
    qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                      _("device not found: %s (%s)"), path, qemu_dev_name);
    return -1;

 done:
    free (info);
    return 0;
}

3374 3375 3376 3377 3378 3379 3380
static int
qemudDomainInterfaceStats (virDomainPtr dom,
                           const char *path,
                           struct _virDomainInterfaceStats *stats)
{
#ifdef __linux__
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
3381 3382
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
    virDomainNetDefPtr net;
3383 3384 3385

    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
3386
                          _("no domain with matching id %d"), dom->id);
3387 3388 3389
        return -1;
    }

3390
    if (!virDomainIsActive(vm)) {
3391
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
3392
                         "%s", _("domain is not running"));
3393 3394 3395 3396 3397
        return -1;
    }

    if (!path || path[0] == '\0') {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
3398
                         "%s", _("NULL or empty path"));
3399 3400 3401 3402 3403
        return -1;
    }

    /* Check the path is one of the domain's network interfaces. */
    for (net = vm->def->nets; net; net = net->next) {
3404
        if (net->ifname && STREQ (net->ifname, path))
3405
            goto ok;
3406 3407 3408
    }

    qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
3409
                      _("invalid path, '%s' is not a known interface"), path);
3410 3411 3412 3413 3414 3415 3416 3417 3418 3419 3420
    return -1;
 ok:

    return linuxDomainInterfaceStats (dom->conn, path, stats);
#else
    qemudReportError (dom->conn, dom, NULL, VIR_ERR_NO_SUPPORT,
                      "%s", __FUNCTION__);
    return -1;
#endif
}

3421 3422 3423 3424 3425 3426 3427 3428
static int
qemudDomainBlockPeek (virDomainPtr dom,
                      const char *path,
                      unsigned long long offset, size_t size,
                      void *buffer,
                      unsigned int flags ATTRIBUTE_UNUSED)
{
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
3429 3430
    virDomainObjPtr vm = virDomainFindByUUID(driver->domains, dom->uuid);
    virDomainDiskDefPtr disk;
3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445
    int fd, ret = -1;

    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching uuid"));
        return -1;
    }

    if (!path || path[0] == '\0') {
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                         _("NULL or empty path"));
        return -1;
    }

    /* Check the path belongs to this domain. */
3446 3447 3448
    for (disk = vm->def->disks ; disk != NULL ; disk = disk->next) {
        if (disk->src != NULL &&
            STREQ (disk->src, path)) goto found;
3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465 3466 3467 3468 3469 3470 3471 3472 3473 3474 3475 3476 3477 3478 3479
    }
    qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                      _("invalid path"));
    return -1;

found:
    /* The path is correct, now try to open it and get its size. */
    fd = open (path, O_RDONLY);
    if (fd == -1) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_SYSTEM_ERROR,
                          "%s", strerror (errno));
        goto done;
    }

    /* Seek and read. */
    /* NB. Because we configure with AC_SYS_LARGEFILE, off_t should
     * be 64 bits on all platforms.
     */
    if (lseek (fd, offset, SEEK_SET) == (off_t) -1 ||
        saferead (fd, buffer, size) == (ssize_t) -1) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_SYSTEM_ERROR,
                          "%s", strerror (errno));
        goto done;
    }

    ret = 0;
 done:
    if (fd >= 0) close (fd);
    return ret;
}

R
Richard W.M. Jones 已提交
3480 3481 3482 3483 3484 3485 3486
static int
qemudDomainMemoryPeek (virDomainPtr dom,
                       unsigned long long offset, size_t size,
                       void *buffer,
                       unsigned int flags)
{
    struct qemud_driver *driver = (struct qemud_driver *)dom->conn->privateData;
3487
    virDomainObjPtr vm = virDomainFindByID(driver->domains, dom->id);
R
Richard W.M. Jones 已提交
3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499 3500 3501 3502 3503
    char cmd[256], *info;
    char tmp[] = TEMPDIR "/qemu.mem.XXXXXX";
    int fd = -1, ret = -1;

    if (flags != VIR_MEMORY_VIRTUAL) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_ARG,
                          _("QEMU driver only supports virtual memory addrs"));
        return -1;
    }

    if (!vm) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_INVALID_DOMAIN,
                          _("no domain with matching id %d"), dom->id);
        return -1;
    }

3504
    if (!virDomainIsActive(vm)) {
R
Richard W.M. Jones 已提交
3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539 3540 3541
        qemudReportError(dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                         "%s", _("domain is not running"));
        return -1;
    }

    /* Create a temporary filename. */
    if ((fd = mkstemp (tmp)) == -1) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_SYSTEM_ERROR,
                          "%s", strerror (errno));
        return -1;
    }

    /* Issue the memsave command. */
    snprintf (cmd, sizeof cmd, "memsave %llu %zi \"%s\"", offset, size, tmp);
    if (qemudMonitorCommand (driver, vm, cmd, &info) < 0) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_OPERATION_FAILED,
                          "%s", _("'info blockstats' command failed"));
        goto done;
    }

    DEBUG ("memsave reply: %s", info);
    free (info);

    /* Read the memory file into buffer. */
    if (saferead (fd, buffer, size) == (ssize_t) -1) {
        qemudReportError (dom->conn, dom, NULL, VIR_ERR_SYSTEM_ERROR,
                          "%s", strerror (errno));
        goto done;
    }

    ret = 0;
done:
    if (fd >= 0) close (fd);
    unlink (tmp);
    return ret;
}

3542
static virNetworkPtr qemudNetworkLookupByUUID(virConnectPtr conn ATTRIBUTE_UNUSED,
3543
                                              const unsigned char *uuid) {
3544
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3545
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, uuid);
3546 3547 3548
    virNetworkPtr net;

    if (!network) {
3549 3550
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_NETWORK,
                         "%s", _("no network with matching uuid"));
3551 3552 3553
        return NULL;
    }

3554
    net = virGetNetwork(conn, network->def->name, network->def->uuid);
3555 3556
    return net;
}
3557
static virNetworkPtr qemudNetworkLookupByName(virConnectPtr conn ATTRIBUTE_UNUSED,
3558
                                              const char *name) {
3559
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3560
    virNetworkObjPtr network = virNetworkFindByName(driver->networks, name);
3561 3562 3563
    virNetworkPtr net;

    if (!network) {
3564 3565
        qemudReportError(conn, NULL, NULL, VIR_ERR_NO_NETWORK,
                         "%s", _("no network with matching name"));
3566 3567 3568
        return NULL;
    }

3569
    net = virGetNetwork(conn, network->def->name, network->def->uuid);
3570 3571 3572 3573
    return net;
}

static virDrvOpenStatus qemudOpenNetwork(virConnectPtr conn,
3574
                                         xmlURIPtr uri ATTRIBUTE_UNUSED,
3575
                                         virConnectAuthPtr auth ATTRIBUTE_UNUSED,
3576 3577 3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588
                                         int flags ATTRIBUTE_UNUSED) {
    if (!qemu_driver)
        return VIR_DRV_OPEN_DECLINED;

    conn->networkPrivateData = qemu_driver;
    return VIR_DRV_OPEN_SUCCESS;
}

static int qemudCloseNetwork(virConnectPtr conn) {
    conn->networkPrivateData = NULL;
    return 0;
}

3589
static int qemudNumNetworks(virConnectPtr conn) {
3590
    int nactive = 0;
3591
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3592 3593 3594 3595 3596 3597 3598
    virNetworkObjPtr net = driver->networks;
    while (net) {
        if (virNetworkIsActive(net))
            nactive++;
        net = net->next;
    }
    return nactive;
3599 3600
}

3601
static int qemudListNetworks(virConnectPtr conn, char **const names, int nnames) {
3602
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3603
    virNetworkObjPtr network = driver->networks;
3604
    int got = 0, i;
3605
    while (network && got < nnames) {
3606
        if (virNetworkIsActive(network)) {
3607
            if (!(names[got] = strdup(network->def->name))) {
3608 3609
                qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                     "%s", _("failed to allocate space for VM name string"));
3610 3611
                goto cleanup;
            }
3612 3613
            got++;
        }
3614 3615 3616
        network = network->next;
    }
    return got;
3617 3618 3619

 cleanup:
    for (i = 0 ; i < got ; i++)
3620
        VIR_FREE(names[i]);
3621
    return -1;
3622 3623
}

3624
static int qemudNumDefinedNetworks(virConnectPtr conn) {
3625
    int ninactive = 0;
3626
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3627 3628 3629 3630 3631 3632 3633
    virNetworkObjPtr net = driver->networks;
    while (net) {
        if (!virNetworkIsActive(net))
            ninactive++;
        net = net->next;
    }
    return ninactive;
3634 3635
}

3636
static int qemudListDefinedNetworks(virConnectPtr conn, char **const names, int nnames) {
3637
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3638
    virNetworkObjPtr network = driver->networks;
3639
    int got = 0, i;
3640
    while (network && got < nnames) {
3641
        if (!virNetworkIsActive(network)) {
3642
            if (!(names[got] = strdup(network->def->name))) {
3643 3644
                qemudReportError(conn, NULL, NULL, VIR_ERR_NO_MEMORY,
                     "%s", _("failed to allocate space for VM name string"));
3645 3646
                goto cleanup;
            }
3647 3648
            got++;
        }
3649 3650 3651
        network = network->next;
    }
    return got;
3652 3653 3654

 cleanup:
    for (i = 0 ; i < got ; i++)
3655
        VIR_FREE(names[i]);
3656
    return -1;
3657 3658
}

3659
static virNetworkPtr qemudNetworkCreate(virConnectPtr conn, const char *xml) {
3660
 struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3661 3662
    virNetworkDefPtr def;
    virNetworkObjPtr network;
3663
    virNetworkPtr net;
3664

3665
    if (!(def = virNetworkDefParseString(conn, xml)))
3666 3667
        return NULL;

3668 3669 3670 3671
    if (!(network = virNetworkAssignDef(conn,
                                        &driver->networks,
                                        def))) {
        virNetworkDefFree(def);
3672 3673 3674
        return NULL;
    }

3675
    if (qemudStartNetworkDaemon(conn, driver, network) < 0) {
3676 3677
        virNetworkRemoveInactive(&driver->networks,
                                 network);
3678 3679 3680
        return NULL;
    }

3681
    net = virGetNetwork(conn, network->def->name, network->def->uuid);
3682
    return net;
3683 3684
}

3685
static virNetworkPtr qemudNetworkDefine(virConnectPtr conn, const char *xml) {
3686
    struct qemud_driver *driver = (struct qemud_driver *)conn->networkPrivateData;
3687 3688
    virNetworkDefPtr def;
    virNetworkObjPtr network;
3689

3690
    if (!(def = virNetworkDefParseString(conn, xml)))
3691 3692
        return NULL;

3693 3694 3695 3696
    if (!(network = virNetworkAssignDef(conn,
                                        &driver->networks,
                                        def))) {
        virNetworkDefFree(def);
3697 3698 3699
        return NULL;
    }

3700 3701 3702 3703 3704 3705
    if (virNetworkSaveConfig(conn,
                             driver->networkConfigDir,
                             driver->networkAutostartDir,
                             network) < 0) {
        virNetworkRemoveInactive(&driver->networks,
                                 network);
3706 3707 3708
        return NULL;
    }

3709
    return virGetNetwork(conn, network->def->name, network->def->uuid);
3710 3711
}

3712
static int qemudNetworkUndefine(virNetworkPtr net) {
3713
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3714
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3715 3716

    if (!network) {
3717 3718
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_DOMAIN,
                         "%s", _("no network with matching uuid"));
3719 3720 3721
        return -1;
    }

3722 3723 3724
    if (virNetworkIsActive(network)) {
        qemudReportError(net->conn, NULL, net, VIR_ERR_INTERNAL_ERROR,
                         "%s", _("network is still active"));
3725
        return -1;
3726
    }
3727

3728 3729
    if (virNetworkDeleteConfig(net->conn, network) < 0)
        return -1;
3730

3731 3732
    virNetworkRemoveInactive(&driver->networks,
                             network);
3733 3734

    return 0;
3735 3736
}

3737
static int qemudNetworkStart(virNetworkPtr net) {
3738
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3739
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3740 3741

    if (!network) {
3742
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_NETWORK,
3743
                         "%s", _("no network with matching uuid"));
3744
        return -1;
3745 3746
    }

3747
    return qemudStartNetworkDaemon(net->conn, driver, network);
3748 3749
}

3750
static int qemudNetworkDestroy(virNetworkPtr net) {
3751
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3752
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3753
    int ret;
3754

3755
    if (!network) {
3756
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_NETWORK,
3757
                         "%s", _("no network with matching uuid"));
3758 3759 3760
        return -1;
    }

3761
    ret = qemudShutdownNetworkDaemon(net->conn, driver, network);
3762 3763

    return ret;
3764 3765
}

3766
static char *qemudNetworkDumpXML(virNetworkPtr net, int flags ATTRIBUTE_UNUSED) {
3767
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3768
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3769

3770
    if (!network) {
3771
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_NETWORK,
3772
                         "%s", _("no network with matching uuid"));
3773
        return NULL;
3774 3775
    }

3776
    return virNetworkDefFormat(net->conn, network->def);
3777
}
D
Daniel P. Berrange 已提交
3778

3779
static char *qemudNetworkGetBridgeName(virNetworkPtr net) {
3780
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3781
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3782
    char *bridge;
3783
    if (!network) {
3784 3785
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_NETWORK,
                         "%s", _("no network with matching id"));
3786
        return NULL;
3787 3788
    }

3789
    bridge = strdup(network->def->bridge);
3790
    if (!bridge) {
3791 3792
        qemudReportError(net->conn, NULL, net, VIR_ERR_NO_MEMORY,
                 "%s", _("failed to allocate space for network bridge string"));
3793 3794 3795
        return NULL;
    }
    return bridge;
3796 3797
}

3798
static int qemudNetworkGetAutostart(virNetworkPtr net,
3799
                             int *autostart) {
3800
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3801
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3802 3803

    if (!network) {
3804 3805
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_NETWORK,
                         "%s", _("no network with matching uuid"));
3806 3807 3808 3809 3810 3811 3812 3813
        return -1;
    }

    *autostart = network->autostart;

    return 0;
}

3814
static int qemudNetworkSetAutostart(virNetworkPtr net,
3815
                             int autostart) {
3816
    struct qemud_driver *driver = (struct qemud_driver *)net->conn->networkPrivateData;
3817
    virNetworkObjPtr network = virNetworkFindByUUID(driver->networks, net->uuid);
3818 3819

    if (!network) {
3820 3821
        qemudReportError(net->conn, NULL, net, VIR_ERR_INVALID_NETWORK,
                         "%s", _("no network with matching uuid"));
3822 3823 3824 3825 3826 3827 3828 3829
        return -1;
    }

    autostart = (autostart != 0);

    if (network->autostart == autostart)
        return 0;

3830 3831 3832
    if (autostart) {
        int err;

3833
        if ((err = virFileMakePath(driver->networkAutostartDir))) {
3834
            qemudReportError(net->conn, NULL, net, VIR_ERR_INTERNAL_ERROR,
3835
                             _("cannot create autostart directory %s: %s"),
3836
                             driver->networkAutostartDir, strerror(err));
3837 3838 3839 3840
            return -1;
        }

        if (symlink(network->configFile, network->autostartLink) < 0) {
3841
            qemudReportError(net->conn, NULL, net, VIR_ERR_INTERNAL_ERROR,
3842
                             _("Failed to create symlink '%s' to '%s': %s"),
3843 3844 3845 3846 3847
                             network->autostartLink, network->configFile, strerror(errno));
            return -1;
        }
    } else {
        if (unlink(network->autostartLink) < 0 && errno != ENOENT && errno != ENOTDIR) {
3848
            qemudReportError(net->conn, NULL, net, VIR_ERR_INTERNAL_ERROR,
3849
                             _("Failed to delete symlink '%s': %s"),
3850 3851 3852 3853 3854
                             network->autostartLink, strerror(errno));
            return -1;
        }
    }

3855 3856 3857 3858 3859
    network->autostart = autostart;

    return 0;
}

3860 3861 3862 3863
static virDriver qemuDriver = {
    VIR_DRV_QEMU,
    "QEMU",
    LIBVIR_VERSION_NUMBER,
3864
    qemudProbe, /* probe */
3865 3866
    qemudOpen, /* open */
    qemudClose, /* close */
3867
    NULL, /* supports_feature */
3868 3869
    qemudGetType, /* type */
    qemudGetVersion, /* version */
3870
    qemudGetHostname, /* hostname */
3871
    NULL, /* URI  */
3872 3873 3874 3875 3876 3877 3878 3879 3880 3881 3882
    qemudGetMaxVCPUs, /* getMaxVcpus */
    qemudGetNodeInfo, /* nodeGetInfo */
    qemudGetCapabilities, /* getCapabilities */
    qemudListDomains, /* listDomains */
    qemudNumDomains, /* numOfDomains */
    qemudDomainCreate, /* domainCreateLinux */
    qemudDomainLookupByID, /* domainLookupByID */
    qemudDomainLookupByUUID, /* domainLookupByUUID */
    qemudDomainLookupByName, /* domainLookupByName */
    qemudDomainSuspend, /* domainSuspend */
    qemudDomainResume, /* domainResume */
3883
    qemudDomainShutdown, /* domainShutdown */
3884 3885 3886
    NULL, /* domainReboot */
    qemudDomainDestroy, /* domainDestroy */
    qemudDomainGetOSType, /* domainGetOSType */
3887 3888 3889
    qemudDomainGetMaxMemory, /* domainGetMaxMemory */
    qemudDomainSetMaxMemory, /* domainSetMaxMemory */
    qemudDomainSetMemory, /* domainSetMemory */
3890 3891 3892 3893
    qemudDomainGetInfo, /* domainGetInfo */
    qemudDomainSave, /* domainSave */
    qemudDomainRestore, /* domainRestore */
    NULL, /* domainCoreDump */
3894
    qemudDomainSetVcpus, /* domainSetVcpus */
3895 3896 3897 3898
#if HAVE_SCHED_GETAFFINITY
    qemudDomainPinVcpu, /* domainPinVcpu */
    qemudDomainGetVcpus, /* domainGetVcpus */
#else
3899 3900
    NULL, /* domainPinVcpu */
    NULL, /* domainGetVcpus */
3901
#endif
3902
    qemudDomainGetMaxVcpus, /* domainGetMaxVcpus */
3903 3904 3905 3906 3907 3908
    qemudDomainDumpXML, /* domainDumpXML */
    qemudListDefinedDomains, /* listDomains */
    qemudNumDefinedDomains, /* numOfDomains */
    qemudDomainStart, /* domainCreate */
    qemudDomainDefine, /* domainDefineXML */
    qemudDomainUndefine, /* domainUndefine */
3909
    qemudDomainAttachDevice, /* domainAttachDevice */
3910 3911 3912 3913 3914 3915
    NULL, /* domainDetachDevice */
    qemudDomainGetAutostart, /* domainGetAutostart */
    qemudDomainSetAutostart, /* domainSetAutostart */
    NULL, /* domainGetSchedulerType */
    NULL, /* domainGetSchedulerParameters */
    NULL, /* domainSetSchedulerParameters */
3916 3917 3918
    NULL, /* domainMigratePrepare */
    NULL, /* domainMigratePerform */
    NULL, /* domainMigrateFinish */
3919
    qemudDomainBlockStats, /* domainBlockStats */
3920
    qemudDomainInterfaceStats, /* domainInterfaceStats */
3921
    qemudDomainBlockPeek, /* domainBlockPeek */
R
Richard W.M. Jones 已提交
3922
    qemudDomainMemoryPeek, /* domainMemoryPeek */
3923 3924 3925 3926
#if HAVE_NUMACTL
    qemudNodeGetCellsFreeMemory, /* nodeGetCellsFreeMemory */
    qemudNodeGetFreeMemory,  /* getFreeMemory */
#else
3927
    NULL, /* nodeGetCellsFreeMemory */
3928
    NULL, /* getFreeMemory */
3929
#endif
3930 3931 3932
};

static virNetworkDriver qemuNetworkDriver = {
3933
    "QEMU",
3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952
    qemudOpenNetwork, /* open */
    qemudCloseNetwork, /* close */
    qemudNumNetworks, /* numOfNetworks */
    qemudListNetworks, /* listNetworks */
    qemudNumDefinedNetworks, /* numOfDefinedNetworks */
    qemudListDefinedNetworks, /* listDefinedNetworks */
    qemudNetworkLookupByUUID, /* networkLookupByUUID */
    qemudNetworkLookupByName, /* networkLookupByName */
    qemudNetworkCreate, /* networkCreateXML */
    qemudNetworkDefine, /* networkDefineXML */
    qemudNetworkUndefine, /* networkUndefine */
    qemudNetworkStart, /* networkCreate */
    qemudNetworkDestroy, /* networkDestroy */
    qemudNetworkDumpXML, /* networkDumpXML */
    qemudNetworkGetBridgeName, /* networkGetBridgeName */
    qemudNetworkGetAutostart, /* networkGetAutostart */
    qemudNetworkSetAutostart, /* networkSetAutostart */
};

A
Atsushi SAKAI 已提交
3953
#ifdef WITH_LIBVIRTD
3954 3955 3956 3957 3958
static virStateDriver qemuStateDriver = {
    qemudStartup,
    qemudShutdown,
    qemudReload,
    qemudActive,
3959
    NULL
3960
};
A
Atsushi SAKAI 已提交
3961
#endif
3962

3963 3964 3965
int qemudRegister(void) {
    virRegisterDriver(&qemuDriver);
    virRegisterNetworkDriver(&qemuNetworkDriver);
A
Atsushi SAKAI 已提交
3966
#ifdef WITH_LIBVIRTD
3967
    virRegisterStateDriver(&qemuStateDriver);
A
Atsushi SAKAI 已提交
3968
#endif
3969 3970 3971
    return 0;
}