remote.c 151.3 KB
Newer Older
1
/*
2
 * remote.c: handlers for RPC method calls
3
 *
4
 * Copyright (C) 2007-2012 Red Hat, Inc.
5 6 7 8 9 10 11 12 13 14 15 16
 *
 * 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
17
 * License along with this library.  If not, see
O
Osier Yang 已提交
18
 * <http://www.gnu.org/licenses/>.
19 20 21 22 23 24
 *
 * Author: Richard W.M. Jones <rjones@redhat.com>
 */

#include <config.h>

25
#include "virerror.h"
26

27
#if WITH_POLKIT0
28 29
# include <polkit/polkit.h>
# include <polkit-dbus/polkit-dbus.h>
30 31
#endif

32
#include "remote.h"
33
#include "libvirtd.h"
34 35
#include "libvirt_internal.h"
#include "datatypes.h"
36
#include "viralloc.h"
37
#include "virlog.h"
C
Chris Lalancette 已提交
38
#include "stream.h"
39
#include "viruuid.h"
40
#include "vircommand.h"
41
#include "intprops.h"
42
#include "virnetserverservice.h"
J
Jim Fehlig 已提交
43
#include "virnetserver.h"
44
#include "virfile.h"
45
#include "virtypedparam.h"
46
#include "virdbus.h"
47
#include "virprocess.h"
48 49
#include "remote_protocol.h"
#include "qemu_protocol.h"
50
#include "lxc_protocol.h"
51
#include "virstring.h"
52 53

#define VIR_FROM_THIS VIR_FROM_RPC
54

55
#if SIZEOF_LONG < 8
E
Eric Blake 已提交
56 57 58
# define HYPER_TO_TYPE(_type, _to, _from)                               \
    do {                                                                \
        if ((_from) != (_type)(_from)) {                                \
59 60 61
            virReportError(VIR_ERR_OVERFLOW,                            \
                           _("conversion from hyper to %s overflowed"), \
                           #_type);                                     \
E
Eric Blake 已提交
62 63 64
            goto cleanup;                                               \
        }                                                               \
        (_to) = (_from);                                                \
65 66 67 68 69 70 71 72 73
    } while (0)

# define HYPER_TO_LONG(_to, _from) HYPER_TO_TYPE(long, _to, _from)
# define HYPER_TO_ULONG(_to, _from) HYPER_TO_TYPE(unsigned long, _to, _from)
#else
# define HYPER_TO_LONG(_to, _from) (_to) = (_from)
# define HYPER_TO_ULONG(_to, _from) (_to) = (_from)
#endif

74 75 76 77 78 79 80
static virDomainPtr get_nonnull_domain(virConnectPtr conn, remote_nonnull_domain domain);
static virNetworkPtr get_nonnull_network(virConnectPtr conn, remote_nonnull_network network);
static virInterfacePtr get_nonnull_interface(virConnectPtr conn, remote_nonnull_interface iface);
static virStoragePoolPtr get_nonnull_storage_pool(virConnectPtr conn, remote_nonnull_storage_pool pool);
static virStorageVolPtr get_nonnull_storage_vol(virConnectPtr conn, remote_nonnull_storage_vol vol);
static virSecretPtr get_nonnull_secret(virConnectPtr conn, remote_nonnull_secret secret);
static virNWFilterPtr get_nonnull_nwfilter(virConnectPtr conn, remote_nonnull_nwfilter nwfilter);
81
static virDomainSnapshotPtr get_nonnull_domain_snapshot(virDomainPtr dom, remote_nonnull_domain_snapshot snapshot);
82 83 84 85 86 87 88 89 90
static void make_nonnull_domain(remote_nonnull_domain *dom_dst, virDomainPtr dom_src);
static void make_nonnull_network(remote_nonnull_network *net_dst, virNetworkPtr net_src);
static void make_nonnull_interface(remote_nonnull_interface *interface_dst, virInterfacePtr interface_src);
static void make_nonnull_storage_pool(remote_nonnull_storage_pool *pool_dst, virStoragePoolPtr pool_src);
static void make_nonnull_storage_vol(remote_nonnull_storage_vol *vol_dst, virStorageVolPtr vol_src);
static void make_nonnull_node_device(remote_nonnull_node_device *dev_dst, virNodeDevicePtr dev_src);
static void make_nonnull_secret(remote_nonnull_secret *secret_dst, virSecretPtr secret_src);
static void make_nonnull_nwfilter(remote_nonnull_nwfilter *net_dst, virNWFilterPtr nwfilter_src);
static void make_nonnull_domain_snapshot(remote_nonnull_domain_snapshot *snapshot_dst, virDomainSnapshotPtr snapshot_src);
91

92 93 94 95 96
static virTypedParameterPtr
remoteDeserializeTypedParameters(remote_typed_param *args_params_val,
                                 u_int args_params_len,
                                 int limit,
                                 int *nparams);
97

98 99 100 101 102 103
static int
remoteSerializeDomainDiskErrors(virDomainDiskErrorPtr errors,
                                int nerrors,
                                remote_domain_disk_error **ret_errors_val,
                                u_int *ret_errors_len);

104 105
#include "remote_dispatch.h"
#include "qemu_dispatch.h"
106
#include "lxc_dispatch.h"
C
Chris Lalancette 已提交
107 108


109 110
/* Prototypes */
static void
111 112
remoteDispatchDomainEventSend(virNetServerClientPtr client,
                              virNetServerProgramPtr program,
113 114 115
                              int procnr,
                              xdrproc_t proc,
                              void *data);
116

117 118 119 120 121
static int remoteRelayDomainEventLifecycle(virConnectPtr conn ATTRIBUTE_UNUSED,
                                           virDomainPtr dom,
                                           int event,
                                           int detail,
                                           void *opaque)
122
{
123
    virNetServerClientPtr client = opaque;
124
    remote_domain_event_lifecycle_msg data;
125

126 127 128
    if (!client)
        return -1;

129
    VIR_DEBUG("Relaying domain lifecycle event %d %d", event, detail);
130

131
    /* build return data */
132
    memset(&data, 0, sizeof(data));
133
    make_nonnull_domain(&data.dom, dom);
134 135
    data.event = event;
    data.detail = detail;
136

137
    remoteDispatchDomainEventSend(client, remoteProgram,
138 139
                                  REMOTE_PROC_DOMAIN_EVENT_LIFECYCLE,
                                  (xdrproc_t)xdr_remote_domain_event_lifecycle_msg, &data);
140

141 142
    return 0;
}
143

144 145 146 147
static int remoteRelayDomainEventReboot(virConnectPtr conn ATTRIBUTE_UNUSED,
                                        virDomainPtr dom,
                                        void *opaque)
{
148
    virNetServerClientPtr client = opaque;
149 150 151 152 153
    remote_domain_event_reboot_msg data;

    if (!client)
        return -1;

154
    VIR_DEBUG("Relaying domain reboot event %s %d", dom->name, dom->id);
155 156

    /* build return data */
157
    memset(&data, 0, sizeof(data));
158
    make_nonnull_domain(&data.dom, dom);
159

160
    remoteDispatchDomainEventSend(client, remoteProgram,
161 162
                                  REMOTE_PROC_DOMAIN_EVENT_REBOOT,
                                  (xdrproc_t)xdr_remote_domain_event_reboot_msg, &data);
163 164 165 166

    return 0;
}

167

168 169 170 171 172
static int remoteRelayDomainEventRTCChange(virConnectPtr conn ATTRIBUTE_UNUSED,
                                           virDomainPtr dom,
                                           long long offset,
                                           void *opaque)
{
173
    virNetServerClientPtr client = opaque;
174 175 176 177 178
    remote_domain_event_rtc_change_msg data;

    if (!client)
        return -1;

179
    VIR_DEBUG("Relaying domain rtc change event %s %d %lld", dom->name, dom->id, offset);
180 181

    /* build return data */
182
    memset(&data, 0, sizeof(data));
183
    make_nonnull_domain(&data.dom, dom);
184 185
    data.offset = offset;

186
    remoteDispatchDomainEventSend(client, remoteProgram,
187 188
                                  REMOTE_PROC_DOMAIN_EVENT_RTC_CHANGE,
                                  (xdrproc_t)xdr_remote_domain_event_rtc_change_msg, &data);
189 190 191 192 193

    return 0;
}


194 195 196 197 198
static int remoteRelayDomainEventWatchdog(virConnectPtr conn ATTRIBUTE_UNUSED,
                                          virDomainPtr dom,
                                          int action,
                                          void *opaque)
{
199
    virNetServerClientPtr client = opaque;
200 201 202 203 204
    remote_domain_event_watchdog_msg data;

    if (!client)
        return -1;

205
    VIR_DEBUG("Relaying domain watchdog event %s %d %d", dom->name, dom->id, action);
206 207

    /* build return data */
208
    memset(&data, 0, sizeof(data));
209
    make_nonnull_domain(&data.dom, dom);
210 211
    data.action = action;

212
    remoteDispatchDomainEventSend(client, remoteProgram,
213 214
                                  REMOTE_PROC_DOMAIN_EVENT_WATCHDOG,
                                  (xdrproc_t)xdr_remote_domain_event_watchdog_msg, &data);
215 216 217 218 219

    return 0;
}


220 221 222 223 224 225 226
static int remoteRelayDomainEventIOError(virConnectPtr conn ATTRIBUTE_UNUSED,
                                         virDomainPtr dom,
                                         const char *srcPath,
                                         const char *devAlias,
                                         int action,
                                         void *opaque)
{
227
    virNetServerClientPtr client = opaque;
228 229 230 231 232
    remote_domain_event_io_error_msg data;

    if (!client)
        return -1;

233
    VIR_DEBUG("Relaying domain io error %s %d %s %s %d", dom->name, dom->id, srcPath, devAlias, action);
234 235

    /* build return data */
236
    memset(&data, 0, sizeof(data));
237 238 239 240 241 242
    data.srcPath = strdup(srcPath);
    if (data.srcPath == NULL)
        goto mem_error;
    data.devAlias = strdup(devAlias);
    if (data.devAlias == NULL)
        goto mem_error;
243
    make_nonnull_domain(&data.dom, dom);
244 245
    data.action = action;

246
    remoteDispatchDomainEventSend(client, remoteProgram,
247 248
                                  REMOTE_PROC_DOMAIN_EVENT_IO_ERROR,
                                  (xdrproc_t)xdr_remote_domain_event_io_error_msg, &data);
249 250

    return 0;
251 252
mem_error:
    virReportOOMError();
E
Eric Blake 已提交
253 254
    VIR_FREE(data.srcPath);
    VIR_FREE(data.devAlias);
255
    return -1;
256 257 258
}


259 260 261 262 263 264 265 266
static int remoteRelayDomainEventIOErrorReason(virConnectPtr conn ATTRIBUTE_UNUSED,
                                               virDomainPtr dom,
                                               const char *srcPath,
                                               const char *devAlias,
                                               int action,
                                               const char *reason,
                                               void *opaque)
{
267
    virNetServerClientPtr client = opaque;
268 269 270 271 272
    remote_domain_event_io_error_reason_msg data;

    if (!client)
        return -1;

273 274
    VIR_DEBUG("Relaying domain io error %s %d %s %s %d %s",
              dom->name, dom->id, srcPath, devAlias, action, reason);
275 276

    /* build return data */
277
    memset(&data, 0, sizeof(data));
278 279 280 281 282 283
    data.srcPath = strdup(srcPath);
    if (data.srcPath == NULL)
        goto mem_error;
    data.devAlias = strdup(devAlias);
    if (data.devAlias == NULL)
        goto mem_error;
284
    data.action = action;
285 286 287 288 289
    data.reason = strdup(reason);
    if (data.reason == NULL)
        goto mem_error;

    make_nonnull_domain(&data.dom, dom);
290

291
    remoteDispatchDomainEventSend(client, remoteProgram,
292 293
                                  REMOTE_PROC_DOMAIN_EVENT_IO_ERROR_REASON,
                                  (xdrproc_t)xdr_remote_domain_event_io_error_reason_msg, &data);
294 295

    return 0;
296 297 298

mem_error:
    virReportOOMError();
E
Eric Blake 已提交
299 300 301
    VIR_FREE(data.srcPath);
    VIR_FREE(data.devAlias);
    VIR_FREE(data.reason);
302
    return -1;
303 304 305
}


306 307 308 309 310 311 312 313 314
static int remoteRelayDomainEventGraphics(virConnectPtr conn ATTRIBUTE_UNUSED,
                                          virDomainPtr dom,
                                          int phase,
                                          virDomainEventGraphicsAddressPtr local,
                                          virDomainEventGraphicsAddressPtr remote,
                                          const char *authScheme,
                                          virDomainEventGraphicsSubjectPtr subject,
                                          void *opaque)
{
315
    virNetServerClientPtr client = opaque;
316 317 318 319 320 321
    remote_domain_event_graphics_msg data;
    int i;

    if (!client)
        return -1;

322 323 324 325
    VIR_DEBUG("Relaying domain graphics event %s %d %d - %d %s %s  - %d %s %s - %s", dom->name, dom->id, phase,
              local->family, local->service, local->node,
              remote->family, remote->service, remote->node,
              authScheme);
326

327
    VIR_DEBUG("Subject %d", subject->nidentity);
328
    for (i = 0 ; i < subject->nidentity ; i++) {
329
        VIR_DEBUG("  %s=%s", subject->identities[i].type, subject->identities[i].name);
330 331 332
    }

    /* build return data */
333
    memset(&data, 0, sizeof(data));
334 335 336
    data.phase = phase;
    data.local.family = local->family;
    data.remote.family = remote->family;
337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353
    data.authScheme = strdup(authScheme);
    if (data.authScheme == NULL)
        goto mem_error;

    data.local.node = strdup(local->node);
    if (data.local.node == NULL)
        goto mem_error;
    data.local.service = strdup(local->service);
    if (data.local.service == NULL)
        goto mem_error;

    data.remote.node = strdup(remote->node);
    if (data.remote.node == NULL)
        goto mem_error;
    data.remote.service = strdup(remote->service);
    if (data.remote.service == NULL)
        goto mem_error;
354 355

    data.subject.subject_len = subject->nidentity;
356 357 358
    if (VIR_ALLOC_N(data.subject.subject_val, data.subject.subject_len) < 0)
        goto mem_error;

359
    for (i = 0 ; i < data.subject.subject_len ; i++) {
360 361 362 363 364 365
        data.subject.subject_val[i].type = strdup(subject->identities[i].type);
        if (data.subject.subject_val[i].type == NULL)
            goto mem_error;
        data.subject.subject_val[i].name = strdup(subject->identities[i].name);
        if (data.subject.subject_val[i].name == NULL)
            goto mem_error;
366
    }
367
    make_nonnull_domain(&data.dom, dom);
368

369
    remoteDispatchDomainEventSend(client, remoteProgram,
370 371
                                  REMOTE_PROC_DOMAIN_EVENT_GRAPHICS,
                                  (xdrproc_t)xdr_remote_domain_event_graphics_msg, &data);
372 373

    return 0;
374 375 376

mem_error:
    virReportOOMError();
E
Eric Blake 已提交
377 378 379 380 381
    VIR_FREE(data.authScheme);
    VIR_FREE(data.local.node);
    VIR_FREE(data.local.service);
    VIR_FREE(data.remote.node);
    VIR_FREE(data.remote.service);
382 383
    if (data.subject.subject_val != NULL) {
        for (i = 0 ; i < data.subject.subject_len ; i++) {
E
Eric Blake 已提交
384 385
            VIR_FREE(data.subject.subject_val[i].type);
            VIR_FREE(data.subject.subject_val[i].name);
386
        }
E
Eric Blake 已提交
387
        VIR_FREE(data.subject.subject_val);
388 389
    }
    return -1;
390 391
}

392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408
static int remoteRelayDomainEventBlockJob(virConnectPtr conn ATTRIBUTE_UNUSED,
                                          virDomainPtr dom,
                                          const char *path,
                                          int type,
                                          int status,
                                          void *opaque)
{
    virNetServerClientPtr client = opaque;
    remote_domain_event_block_job_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain block job event %s %d %s %i, %i",
              dom->name, dom->id, path, type, status);

    /* build return data */
409
    memset(&data, 0, sizeof(data));
410 411 412
    data.path = strdup(path);
    if (data.path == NULL)
        goto mem_error;
413 414
    data.type = type;
    data.status = status;
415
    make_nonnull_domain(&data.dom, dom);
416 417 418 419 420 421

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_BLOCK_JOB,
                                  (xdrproc_t)xdr_remote_domain_event_block_job_msg, &data);

    return 0;
422 423 424

mem_error:
    virReportOOMError();
E
Eric Blake 已提交
425
    VIR_FREE(data.path);
426
    return -1;
427 428
}

429

430 431 432 433
static int remoteRelayDomainEventControlError(virConnectPtr conn ATTRIBUTE_UNUSED,
                                              virDomainPtr dom,
                                              void *opaque)
{
434
    virNetServerClientPtr client = opaque;
435 436 437 438 439 440 441 442
    remote_domain_event_control_error_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain control error %s %d", dom->name, dom->id);

    /* build return data */
443
    memset(&data, 0, sizeof(data));
444 445
    make_nonnull_domain(&data.dom, dom);

446
    remoteDispatchDomainEventSend(client, remoteProgram,
447 448 449 450 451 452 453
                                  REMOTE_PROC_DOMAIN_EVENT_CONTROL_ERROR,
                                  (xdrproc_t)xdr_remote_domain_event_control_error_msg, &data);

    return 0;
}


454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472
static int remoteRelayDomainEventDiskChange(virConnectPtr conn ATTRIBUTE_UNUSED,
                                            virDomainPtr dom,
                                            const char *oldSrcPath,
                                            const char *newSrcPath,
                                            const char *devAlias,
                                            int reason,
                                            void *opaque)
{
    virNetServerClientPtr client = opaque;
    remote_domain_event_disk_change_msg data;
    char **oldSrcPath_p = NULL, **newSrcPath_p = NULL;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain %s %d disk change %s %s %s %d",
              dom->name, dom->id, oldSrcPath, newSrcPath, devAlias, reason);

    /* build return data */
473
    memset(&data, 0, sizeof(data));
474 475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498
    if (oldSrcPath &&
        ((VIR_ALLOC(oldSrcPath_p) < 0) ||
         !(*oldSrcPath_p = strdup(oldSrcPath))))
        goto mem_error;

    if (newSrcPath &&
        ((VIR_ALLOC(newSrcPath_p) < 0) ||
         !(*newSrcPath_p = strdup(newSrcPath))))
        goto mem_error;

    data.oldSrcPath = oldSrcPath_p;
    data.newSrcPath = newSrcPath_p;
    if (!(data.devAlias = strdup(devAlias)))
        goto mem_error;
    data.reason = reason;

    make_nonnull_domain(&data.dom, dom);

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_DISK_CHANGE,
                                  (xdrproc_t)xdr_remote_domain_event_disk_change_msg, &data);

    return 0;

mem_error:
M
Michal Privoznik 已提交
499 500
    VIR_FREE(oldSrcPath_p);
    VIR_FREE(newSrcPath_p);
501 502 503 504 505
    virReportOOMError();
    return -1;
}


506 507 508 509 510 511 512 513 514 515 516 517 518 519 520
static int remoteRelayDomainEventTrayChange(virConnectPtr conn ATTRIBUTE_UNUSED,
                                            virDomainPtr dom,
                                            const char *devAlias,
                                            int reason,
                                            void *opaque) {
    virNetServerClientPtr client = opaque;
    remote_domain_event_tray_change_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain %s %d tray change devAlias: %s reason: %d",
              dom->name, dom->id, devAlias, reason);

    /* build return data */
521
    memset(&data, 0, sizeof(data));
522 523 524 525 526 527 528 529 530 531 532 533 534 535 536 537

    if (!(data.devAlias = strdup(devAlias))) {
        virReportOOMError();
        return -1;
    }
    data.reason = reason;

    make_nonnull_domain(&data.dom, dom);

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_TRAY_CHANGE,
                                  (xdrproc_t)xdr_remote_domain_event_tray_change_msg, &data);

    return 0;
}

O
Osier Yang 已提交
538 539
static int remoteRelayDomainEventPMWakeup(virConnectPtr conn ATTRIBUTE_UNUSED,
                                          virDomainPtr dom,
540
                                          int reason ATTRIBUTE_UNUSED,
O
Osier Yang 已提交
541 542 543 544 545 546 547 548 549 550
                                          void *opaque) {
    virNetServerClientPtr client = opaque;
    remote_domain_event_pmwakeup_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain %s %d system pmwakeup", dom->name, dom->id);

    /* build return data */
551
    memset(&data, 0, sizeof(data));
O
Osier Yang 已提交
552 553 554 555 556 557 558 559 560
    make_nonnull_domain(&data.dom, dom);

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_PMWAKEUP,
                                  (xdrproc_t)xdr_remote_domain_event_pmwakeup_msg, &data);

    return 0;
}

O
Osier Yang 已提交
561 562
static int remoteRelayDomainEventPMSuspend(virConnectPtr conn ATTRIBUTE_UNUSED,
                                           virDomainPtr dom,
563
                                           int reason ATTRIBUTE_UNUSED,
O
Osier Yang 已提交
564 565 566 567 568 569 570 571 572 573
                                           void *opaque) {
    virNetServerClientPtr client = opaque;
    remote_domain_event_pmsuspend_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain %s %d system pmsuspend", dom->name, dom->id);

    /* build return data */
574
    memset(&data, 0, sizeof(data));
O
Osier Yang 已提交
575 576 577 578 579 580 581 582 583
    make_nonnull_domain(&data.dom, dom);

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_PMSUSPEND,
                                  (xdrproc_t)xdr_remote_domain_event_pmsuspend_msg, &data);

    return 0;
}

584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610
static int
remoteRelayDomainEventBalloonChange(virConnectPtr conn ATTRIBUTE_UNUSED,
                                    virDomainPtr dom,
                                    unsigned long long actual,
                                    void *opaque)
{
    virNetServerClientPtr client = opaque;
    remote_domain_event_balloon_change_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain balloon change event %s %d %lld", dom->name, dom->id, actual);

    /* build return data */
    memset(&data, 0, sizeof(data));
    make_nonnull_domain(&data.dom, dom);
    data.actual = actual;

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_BALLOON_CHANGE,
                                  (xdrproc_t)xdr_remote_domain_event_balloon_change_msg, &data);

    return 0;
}


611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634
static int remoteRelayDomainEventPMSuspendDisk(virConnectPtr conn ATTRIBUTE_UNUSED,
                                               virDomainPtr dom,
                                               int reason ATTRIBUTE_UNUSED,
                                               void *opaque) {
    virNetServerClientPtr client = opaque;
    remote_domain_event_pmsuspend_disk_msg data;

    if (!client)
        return -1;

    VIR_DEBUG("Relaying domain %s %d system pmsuspend-disk", dom->name, dom->id);

    /* build return data */
    memset(&data, 0, sizeof(data));
    make_nonnull_domain(&data.dom, dom);

    remoteDispatchDomainEventSend(client, remoteProgram,
                                  REMOTE_PROC_DOMAIN_EVENT_PMSUSPEND_DISK,
                                  (xdrproc_t)xdr_remote_domain_event_pmsuspend_disk_msg, &data);

    return 0;
}


635
static virConnectDomainEventGenericCallback domainEventCallbacks[] = {
636
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventLifecycle),
637
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventReboot),
638
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventRTCChange),
639
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventWatchdog),
640
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventIOError),
641
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventGraphics),
642
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventIOErrorReason),
643
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventControlError),
644
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventBlockJob),
645
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventDiskChange),
646
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventTrayChange),
O
Osier Yang 已提交
647
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventPMWakeup),
O
Osier Yang 已提交
648
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventPMSuspend),
649
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventBalloonChange),
650
    VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventPMSuspendDisk),
651 652 653 654
};

verify(ARRAY_CARDINALITY(domainEventCallbacks) == VIR_DOMAIN_EVENT_ID_LAST);

655 656 657 658 659 660 661
/*
 * You must hold lock for at least the client
 * We don't free stuff here, merely disconnect the client's
 * network socket & resources.
 * We keep the libvirt connection open until any async
 * jobs have finished, then clean it up elsewhere
 */
662
void remoteClientFreeFunc(void *data)
663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683 684 685
{
    struct daemonClientPrivate *priv = data;

    /* Deregister event delivery callback */
    if (priv->conn) {
        int i;

        for (i = 0 ; i < VIR_DOMAIN_EVENT_ID_LAST ; i++) {
            if (priv->domainEventCallbackID[i] != -1) {
                VIR_DEBUG("Deregistering to relay remote events %d", i);
                virConnectDomainEventDeregisterAny(priv->conn,
                                                   priv->domainEventCallbackID[i]);
            }
            priv->domainEventCallbackID[i] = -1;
        }

        virConnectClose(priv->conn);
    }

    VIR_FREE(priv);
}


686 687 688 689 690 691 692
static void remoteClientCloseFunc(virNetServerClientPtr client)
{
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    daemonRemoveAllClientStreams(priv->streams);
}

693

694 695
void *remoteClientInitHook(virNetServerClientPtr client,
                           void *opaque ATTRIBUTE_UNUSED)
696 697 698 699 700 701
{
    struct daemonClientPrivate *priv;
    int i;

    if (VIR_ALLOC(priv) < 0) {
        virReportOOMError();
702
        return NULL;
703 704 705 706 707
    }

    if (virMutexInit(&priv->lock) < 0) {
        VIR_FREE(priv);
        virReportOOMError();
708
        return NULL;
709 710 711 712 713
    }

    for (i = 0 ; i < VIR_DOMAIN_EVENT_ID_LAST ; i++)
        priv->domainEventCallbackID[i] = -1;

714
    virNetServerClientSetCloseHook(client, remoteClientCloseFunc);
715
    return priv;
716 717
}

718 719 720
/*----- Functions. -----*/

static int
721 722 723 724 725
remoteDispatchConnectOpen(virNetServerPtr server,
                          virNetServerClientPtr client,
                          virNetMessagePtr msg ATTRIBUTE_UNUSED,
                          virNetMessageErrorPtr rerr,
                          struct remote_connect_open_args *args)
726 727
{
    const char *name;
728
    unsigned int flags;
729
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);
730
    int rv = -1;
731

732 733 734 735
    VIR_DEBUG("priv=%p conn=%p", priv, priv->conn);
    virMutexLock(&priv->lock);
    /* Already opened? */
    if (priv->conn) {
736
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection already open"));
737 738 739
        goto cleanup;
    }

740
    if (virNetServerKeepAliveRequired(server) && !priv->keepalive_supported) {
741 742
        virReportError(VIR_ERR_OPERATION_FAILED, "%s",
                       _("keepalive support is required to connect"));
743 744 745
        goto cleanup;
    }

746 747 748 749 750 751
    name = args->name ? *args->name : NULL;

    /* If this connection arrived on a readonly socket, force
     * the connection to be readonly.
     */
    flags = args->flags;
752 753
    if (virNetServerClientGetReadonly(client))
        flags |= VIR_CONNECT_RO;
754

755
    priv->conn =
756
        flags & VIR_CONNECT_RO
757 758
        ? virConnectOpenReadOnly(name)
        : virConnectOpen(name);
759

760
    if (priv->conn == NULL)
761 762 763
        goto cleanup;

    rv = 0;
764

765 766
cleanup:
    if (rv < 0)
767 768
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
769
    return rv;
770 771 772 773
}


static int
774 775 776 777
remoteDispatchConnectClose(virNetServerPtr server ATTRIBUTE_UNUSED,
                           virNetServerClientPtr client ATTRIBUTE_UNUSED,
                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
                           virNetMessageErrorPtr rerr ATTRIBUTE_UNUSED)
778
{
779
    virNetServerClientDelayedClose(client);
780
    return 0;
781 782
}

783

784
static int
785 786
remoteDispatchDomainGetSchedulerType(virNetServerPtr server ATTRIBUTE_UNUSED,
                                     virNetServerClientPtr client ATTRIBUTE_UNUSED,
787
                                     virNetMessagePtr msg ATTRIBUTE_UNUSED,
788
                                     virNetMessageErrorPtr rerr,
789 790
                                     remote_domain_get_scheduler_type_args *args,
                                     remote_domain_get_scheduler_type_ret *ret)
791
{
792
    virDomainPtr dom = NULL;
793 794
    char *type;
    int nparams;
795
    int rv = -1;
796 797
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
798

799
    if (!priv->conn) {
800
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
801
        goto cleanup;
802 803
    }

804
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
805
        goto cleanup;
806

807
    if (!(type = virDomainGetSchedulerType(dom, &nparams)))
808
        goto cleanup;
809 810 811

    ret->type = type;
    ret->nparams = nparams;
812 813 814 815
    rv = 0;

cleanup:
    if (rv < 0)
816
        virNetMessageSaveError(rerr);
817 818 819
    if (dom)
        virDomainFree(dom);
    return rv;
820 821
}

822 823
/* Helper to serialize typed parameters. This also filters out any string
 * parameters that must not be returned to older clients.  */
824 825 826
static int
remoteSerializeTypedParameters(virTypedParameterPtr params,
                               int nparams,
827
                               remote_typed_param **ret_params_val,
828 829
                               u_int *ret_params_len,
                               unsigned int flags)
830 831
{
    int i;
832
    int j;
833 834 835 836 837 838 839 840 841
    int rv = -1;
    remote_typed_param *val;

    *ret_params_len = nparams;
    if (VIR_ALLOC_N(val, nparams) < 0) {
        virReportOOMError();
        goto cleanup;
    }

842
    for (i = 0, j = 0; i < nparams; ++i) {
843 844 845 846 847
        /* virDomainGetCPUStats can return a sparse array; also, we
         * can't pass back strings to older clients.  */
        if (!params[i].type ||
            (!(flags & VIR_TYPED_PARAM_STRING_OKAY) &&
             params[i].type == VIR_TYPED_PARAM_STRING)) {
848 849 850 851
            --*ret_params_len;
            continue;
        }

852
        /* remoteDispatchClientRequest will free this: */
853 854
        val[j].field = strdup(params[i].field);
        if (val[j].field == NULL) {
855 856 857
            virReportOOMError();
            goto cleanup;
        }
858
        val[j].value.type = params[i].type;
859
        switch (params[i].type) {
860
        case VIR_TYPED_PARAM_INT:
861
            val[j].value.remote_typed_param_value_u.i = params[i].value.i;
862 863
            break;
        case VIR_TYPED_PARAM_UINT:
864
            val[j].value.remote_typed_param_value_u.ui = params[i].value.ui;
865 866
            break;
        case VIR_TYPED_PARAM_LLONG:
867
            val[j].value.remote_typed_param_value_u.l = params[i].value.l;
868 869
            break;
        case VIR_TYPED_PARAM_ULLONG:
870
            val[j].value.remote_typed_param_value_u.ul = params[i].value.ul;
871 872
            break;
        case VIR_TYPED_PARAM_DOUBLE:
873
            val[j].value.remote_typed_param_value_u.d = params[i].value.d;
874 875
            break;
        case VIR_TYPED_PARAM_BOOLEAN:
876 877 878 879 880 881 882 883 884
            val[j].value.remote_typed_param_value_u.b = params[i].value.b;
            break;
        case VIR_TYPED_PARAM_STRING:
            val[j].value.remote_typed_param_value_u.s =
                strdup(params[i].value.s);
            if (val[j].value.remote_typed_param_value_u.s == NULL) {
                virReportOOMError();
                goto cleanup;
            }
885 886
            break;
        default:
887 888
            virReportError(VIR_ERR_RPC, _("unknown parameter type: %d"),
                           params[i].type);
889 890
            goto cleanup;
        }
891
        j++;
892 893 894 895 896 897 898 899
    }

    *ret_params_val = val;
    val = NULL;
    rv = 0;

cleanup:
    if (val) {
900
        for (i = 0; i < nparams; i++) {
901
            VIR_FREE(val[i].field);
902
            if (val[i].value.type == VIR_TYPED_PARAM_STRING)
903 904
                VIR_FREE(val[i].value.remote_typed_param_value_u.s);
        }
905 906 907 908 909 910 911
        VIR_FREE(val);
    }
    return rv;
}

/* Helper to deserialize typed parameters. */
static virTypedParameterPtr
912 913
remoteDeserializeTypedParameters(remote_typed_param *args_params_val,
                                 u_int args_params_len,
914 915 916
                                 int limit,
                                 int *nparams)
{
917
    int i = 0;
918 919 920 921 922
    int rv = -1;
    virTypedParameterPtr params = NULL;

    /* Check the length of the returned list carefully. */
    if (args_params_len > limit) {
923
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
924 925 926 927 928 929 930 931 932 933 934 935 936
        goto cleanup;
    }
    if (VIR_ALLOC_N(params, args_params_len) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    *nparams = args_params_len;

    /* Deserialise the result. */
    for (i = 0; i < args_params_len; ++i) {
        if (virStrcpyStatic(params[i].field,
                            args_params_val[i].field) == NULL) {
937 938 939
            virReportError(VIR_ERR_INTERNAL_ERROR,
                           _("Parameter %s too big for destination"),
                           args_params_val[i].field);
940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967
            goto cleanup;
        }
        params[i].type = args_params_val[i].value.type;
        switch (params[i].type) {
        case VIR_TYPED_PARAM_INT:
            params[i].value.i =
                args_params_val[i].value.remote_typed_param_value_u.i;
            break;
        case VIR_TYPED_PARAM_UINT:
            params[i].value.ui =
                args_params_val[i].value.remote_typed_param_value_u.ui;
            break;
        case VIR_TYPED_PARAM_LLONG:
            params[i].value.l =
                args_params_val[i].value.remote_typed_param_value_u.l;
            break;
        case VIR_TYPED_PARAM_ULLONG:
            params[i].value.ul =
                args_params_val[i].value.remote_typed_param_value_u.ul;
            break;
        case VIR_TYPED_PARAM_DOUBLE:
            params[i].value.d =
                args_params_val[i].value.remote_typed_param_value_u.d;
            break;
        case VIR_TYPED_PARAM_BOOLEAN:
            params[i].value.b =
                args_params_val[i].value.remote_typed_param_value_u.b;
            break;
968 969 970 971 972 973 974 975
        case VIR_TYPED_PARAM_STRING:
            params[i].value.s =
                strdup(args_params_val[i].value.remote_typed_param_value_u.s);
            if (params[i].value.s == NULL) {
                virReportOOMError();
                goto cleanup;
            }
            break;
976
        default:
977 978
            virReportError(VIR_ERR_INTERNAL_ERROR, _("unknown parameter type: %d"),
                           params[i].type);
979 980 981 982 983 984 985
            goto cleanup;
        }
    }

    rv = 0;

cleanup:
986
    if (rv < 0) {
987 988
        virTypedParamsFree(params, i);
        params = NULL;
989
    }
990 991 992
    return params;
}

993
static int
994 995
remoteDispatchDomainGetSchedulerParameters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                           virNetServerClientPtr client ATTRIBUTE_UNUSED,
996
                                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
997
                                           virNetMessageErrorPtr rerr,
998 999
                                           remote_domain_get_scheduler_parameters_args *args,
                                           remote_domain_get_scheduler_parameters_ret *ret)
1000
{
1001
    virDomainPtr dom = NULL;
1002
    virTypedParameterPtr params = NULL;
1003
    int nparams = 0;
1004
    int rv = -1;
1005 1006
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1007

1008
    if (!priv->conn) {
1009
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1010
        goto cleanup;
1011 1012
    }

1013
    if (args->nparams > REMOTE_DOMAIN_SCHEDULER_PARAMETERS_MAX) {
1014
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
1015
        goto cleanup;
1016
    }
1017
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0)
1018
        goto no_memory;
1019
    nparams = args->nparams;
1020

1021
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1022
        goto cleanup;
1023

1024
    if (virDomainGetSchedulerParameters(dom, params, &nparams) < 0)
1025
        goto cleanup;
1026

1027
    if (remoteSerializeTypedParameters(params, nparams,
1028
                                       &ret->params.params_val,
1029 1030
                                       &ret->params.params_len,
                                       0) < 0)
1031 1032 1033 1034 1035 1036
        goto cleanup;

    rv = 0;

cleanup:
    if (rv < 0)
1037
        virNetMessageSaveError(rerr);
1038
    virTypedParamsFree(params, nparams);
1039 1040 1041 1042 1043 1044 1045 1046 1047
    if (dom)
        virDomainFree(dom);
    return rv;

no_memory:
    virReportOOMError();
    goto cleanup;
}

1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062
static int
remoteDispatchConnectListAllDomains(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client,
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                    virNetMessageErrorPtr rerr,
                                    remote_connect_list_all_domains_args *args,
                                    remote_connect_list_all_domains_ret *ret)
{
    virDomainPtr *doms = NULL;
    int ndomains = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
1063
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
        goto cleanup;
    }

    if ((ndomains = virConnectListAllDomains(priv->conn,
                                             args->need_results ? &doms : NULL,
                                             args->flags)) < 0)
        goto cleanup;

    if (doms && ndomains) {
        if (VIR_ALLOC_N(ret->domains.domains_val, ndomains) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->domains.domains_len = ndomains;

        for (i = 0; i < ndomains; i++)
            make_nonnull_domain(ret->domains.domains_val + i, doms[i]);
    } else {
        ret->domains.domains_len = 0;
        ret->domains.domains_val = NULL;
    }

    ret->ret = ndomains;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (doms) {
        for (i = 0; i < ndomains; i++)
            virDomainFree(doms[i]);
        VIR_FREE(doms);
    }
    return rv;
}

1102
static int
1103 1104
remoteDispatchDomainGetSchedulerParametersFlags(virNetServerPtr server ATTRIBUTE_UNUSED,
                                                virNetServerClientPtr client ATTRIBUTE_UNUSED,
1105
                                                virNetMessagePtr msg ATTRIBUTE_UNUSED,
1106
                                                virNetMessageErrorPtr rerr,
1107 1108 1109 1110 1111
                                                remote_domain_get_scheduler_parameters_flags_args *args,
                                                remote_domain_get_scheduler_parameters_flags_ret *ret)
{
    virDomainPtr dom = NULL;
    virTypedParameterPtr params = NULL;
1112
    int nparams = 0;
1113
    int rv = -1;
1114 1115
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1116

1117
    if (!priv->conn) {
1118
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1119 1120 1121
        goto cleanup;
    }

1122
    if (args->nparams > REMOTE_DOMAIN_SCHEDULER_PARAMETERS_MAX) {
1123
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
1124 1125
        goto cleanup;
    }
1126
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0)
1127
        goto no_memory;
1128
    nparams = args->nparams;
1129

1130
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1131 1132 1133 1134 1135 1136 1137
        goto cleanup;

    if (virDomainGetSchedulerParametersFlags(dom, params, &nparams,
                                             args->flags) < 0)
        goto cleanup;

    if (remoteSerializeTypedParameters(params, nparams,
1138
                                       &ret->params.params_val,
1139 1140
                                       &ret->params.params_len,
                                       args->flags) < 0)
1141
        goto cleanup;
1142

1143
    rv = 0;
1144 1145

cleanup:
1146
    if (rv < 0)
1147
        virNetMessageSaveError(rerr);
1148
    virTypedParamsFree(params, nparams);
1149 1150 1151 1152 1153 1154 1155
    if (dom)
        virDomainFree(dom);
    return rv;

no_memory:
    virReportOOMError();
    goto cleanup;
1156 1157
}

1158
static int
1159 1160
remoteDispatchDomainMemoryStats(virNetServerPtr server ATTRIBUTE_UNUSED,
                                virNetServerClientPtr client ATTRIBUTE_UNUSED,
1161
                                virNetMessagePtr msg ATTRIBUTE_UNUSED,
1162
                                virNetMessageErrorPtr rerr,
1163 1164
                                remote_domain_memory_stats_args *args,
                                remote_domain_memory_stats_ret *ret)
1165
{
1166
    virDomainPtr dom = NULL;
1167
    struct _virDomainMemoryStat *stats;
1168
    int nr_stats, i;
1169
    int rv = -1;
1170 1171
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1172

1173
    if (!priv->conn) {
1174
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1175
        goto cleanup;
1176 1177
    }

1178
    if (args->maxStats > REMOTE_DOMAIN_MEMORY_STATS_MAX) {
1179 1180
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("maxStats > REMOTE_DOMAIN_MEMORY_STATS_MAX"));
1181
        goto cleanup;
1182 1183
    }

1184
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1185
        goto cleanup;
1186 1187 1188

    /* Allocate stats array for making dispatch call */
    if (VIR_ALLOC_N(stats, args->maxStats) < 0) {
1189 1190
        virReportOOMError();
        goto cleanup;
1191
    }
1192

1193
    nr_stats = virDomainMemoryStats(dom, stats, args->maxStats, args->flags);
1194
    if (nr_stats < 0)
1195
        goto cleanup;
1196 1197 1198

    /* Allocate return buffer */
    if (VIR_ALLOC_N(ret->stats.stats_val, args->maxStats) < 0) {
1199 1200
        virReportOOMError();
        goto cleanup;
1201 1202 1203 1204 1205 1206 1207 1208
    }

    /* Copy the stats into the xdr return structure */
    for (i = 0; i < nr_stats; i++) {
        ret->stats.stats_val[i].tag = stats[i].tag;
        ret->stats.stats_val[i].val = stats[i].val;
    }
    ret->stats.stats_len = nr_stats;
1209 1210 1211 1212
    rv = 0;

cleanup:
    if (rv < 0)
1213
        virNetMessageSaveError(rerr);
1214 1215
    if (dom)
        virDomainFree(dom);
1216
    VIR_FREE(stats);
1217
    return rv;
1218 1219
}

1220
static int
1221 1222
remoteDispatchDomainBlockPeek(virNetServerPtr server ATTRIBUTE_UNUSED,
                              virNetServerClientPtr client ATTRIBUTE_UNUSED,
1223
                              virNetMessagePtr msg ATTRIBUTE_UNUSED,
1224
                              virNetMessageErrorPtr rerr,
1225 1226
                              remote_domain_block_peek_args *args,
                              remote_domain_block_peek_ret *ret)
1227
{
1228
    virDomainPtr dom = NULL;
1229 1230 1231 1232
    char *path;
    unsigned long long offset;
    size_t size;
    unsigned int flags;
1233
    int rv = -1;
1234 1235
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1236

1237
    if (!priv->conn) {
1238
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1239
        goto cleanup;
1240 1241
    }

1242
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1243
        goto cleanup;
1244 1245 1246 1247 1248 1249
    path = args->path;
    offset = args->offset;
    size = args->size;
    flags = args->flags;

    if (size > REMOTE_DOMAIN_BLOCK_PEEK_BUFFER_MAX) {
1250 1251
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("size > maximum buffer size"));
1252
        goto cleanup;
1253 1254 1255
    }

    ret->buffer.buffer_len = size;
1256
    if (VIR_ALLOC_N(ret->buffer.buffer_val, size) < 0) {
1257 1258
        virReportOOMError();
        goto cleanup;
1259 1260
    }

1261
    if (virDomainBlockPeek(dom, path, offset, size,
1262
                           ret->buffer.buffer_val, flags) < 0)
1263
        goto cleanup;
1264

1265 1266 1267 1268
    rv = 0;

cleanup:
    if (rv < 0) {
1269
        virNetMessageSaveError(rerr);
1270 1271 1272 1273 1274
        VIR_FREE(ret->buffer.buffer_val);
    }
    if (dom)
        virDomainFree(dom);
    return rv;
1275 1276
}

1277 1278 1279
static int
remoteDispatchDomainBlockStatsFlags(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client ATTRIBUTE_UNUSED,
1280
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
1281 1282 1283 1284 1285 1286 1287
                                    virNetMessageErrorPtr rerr,
                                    remote_domain_block_stats_flags_args *args,
                                    remote_domain_block_stats_flags_ret *ret)
{
    virTypedParameterPtr params = NULL;
    virDomainPtr dom = NULL;
    const char *path = args->path;
1288
    int nparams = 0;
1289 1290 1291 1292 1293 1294
    unsigned int flags;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
1295
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1296 1297 1298 1299 1300 1301 1302
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;
    flags = args->flags;

1303
    if (args->nparams > REMOTE_DOMAIN_BLOCK_STATS_PARAMETERS_MAX) {
1304
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
1305 1306
        goto cleanup;
    }
1307
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
1308 1309 1310
        virReportOOMError();
        goto cleanup;
    }
1311
    nparams = args->nparams;
1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326

    if (virDomainBlockStatsFlags(dom, path, params, &nparams, flags) < 0)
        goto cleanup;

    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    /* Serialise the block stats. */
    if (remoteSerializeTypedParameters(params, nparams,
                                       &ret->params.params_val,
1327 1328
                                       &ret->params.params_len,
                                       args->flags) < 0)
1329 1330 1331 1332 1333 1334
        goto cleanup;

success:
    rv = 0;

cleanup:
1335
    if (rv < 0)
1336
        virNetMessageSaveError(rerr);
1337
    virTypedParamsFree(params, nparams);
A
ajia@redhat.com 已提交
1338 1339
    if (dom)
        virDomainFree(dom);
1340 1341 1342
    return rv;
}

R
Richard W.M. Jones 已提交
1343
static int
1344 1345
remoteDispatchDomainMemoryPeek(virNetServerPtr server ATTRIBUTE_UNUSED,
                               virNetServerClientPtr client ATTRIBUTE_UNUSED,
1346
                               virNetMessagePtr msg ATTRIBUTE_UNUSED,
1347
                               virNetMessageErrorPtr rerr,
1348 1349
                               remote_domain_memory_peek_args *args,
                               remote_domain_memory_peek_ret *ret)
R
Richard W.M. Jones 已提交
1350
{
1351
    virDomainPtr dom = NULL;
R
Richard W.M. Jones 已提交
1352 1353 1354
    unsigned long long offset;
    size_t size;
    unsigned int flags;
1355
    int rv = -1;
1356 1357
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
R
Richard W.M. Jones 已提交
1358

1359
    if (!priv->conn) {
1360
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1361
        goto cleanup;
1362 1363
    }

1364
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1365
        goto cleanup;
R
Richard W.M. Jones 已提交
1366 1367 1368 1369 1370
    offset = args->offset;
    size = args->size;
    flags = args->flags;

    if (size > REMOTE_DOMAIN_MEMORY_PEEK_BUFFER_MAX) {
1371 1372
        virReportError(VIR_ERR_INTERNAL_ERROR,
                       "%s", _("size > maximum buffer size"));
1373
        goto cleanup;
R
Richard W.M. Jones 已提交
1374 1375 1376
    }

    ret->buffer.buffer_len = size;
1377
    if (VIR_ALLOC_N(ret->buffer.buffer_val, size) < 0) {
1378 1379
        virReportOOMError();
        goto cleanup;
R
Richard W.M. Jones 已提交
1380 1381
    }

1382
    if (virDomainMemoryPeek(dom, offset, size,
1383
                            ret->buffer.buffer_val, flags) < 0)
1384
        goto cleanup;
R
Richard W.M. Jones 已提交
1385

1386 1387 1388 1389
    rv = 0;

cleanup:
    if (rv < 0) {
1390
        virNetMessageSaveError(rerr);
1391 1392 1393 1394 1395
        VIR_FREE(ret->buffer.buffer_val);
    }
    if (dom)
        virDomainFree(dom);
    return rv;
R
Richard W.M. Jones 已提交
1396 1397
}

1398
static int
1399 1400
remoteDispatchDomainGetSecurityLabel(virNetServerPtr server ATTRIBUTE_UNUSED,
                                     virNetServerClientPtr client ATTRIBUTE_UNUSED,
1401
                                     virNetMessagePtr msg ATTRIBUTE_UNUSED,
1402
                                     virNetMessageErrorPtr rerr,
1403 1404
                                     remote_domain_get_security_label_args *args,
                                     remote_domain_get_security_label_ret *ret)
1405
{
1406 1407
    virDomainPtr dom = NULL;
    virSecurityLabelPtr seclabel = NULL;
1408
    int rv = -1;
1409 1410
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1411

1412
    if (!priv->conn) {
1413
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1414
        goto cleanup;
1415 1416
    }

1417
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430
        goto cleanup;

    if (VIR_ALLOC(seclabel) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    if (virDomainGetSecurityLabel(dom, seclabel) < 0)
        goto cleanup;

    ret->label.label_len = strlen(seclabel->label) + 1;
    if (VIR_ALLOC_N(ret->label.label_val, ret->label.label_len) < 0) {
        virReportOOMError();
1431
        goto cleanup;
1432 1433 1434
    }
    strcpy(ret->label.label_val, seclabel->label);
    ret->enforcing = seclabel->enforcing;
1435

1436 1437 1438 1439
    rv = 0;

cleanup:
    if (rv < 0)
1440
        virNetMessageSaveError(rerr);
1441 1442 1443
    if (dom)
        virDomainFree(dom);
    VIR_FREE(seclabel);
1444
    return rv;
1445 1446
}

M
Marcelo Cerri 已提交
1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509
static int
remoteDispatchDomainGetSecurityLabelList(virNetServerPtr server ATTRIBUTE_UNUSED,
                                         virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                         virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                         virNetMessageErrorPtr rerr,
                                         remote_domain_get_security_label_list_args *args,
                                         remote_domain_get_security_label_list_ret *ret)
{
    virDomainPtr dom = NULL;
    virSecurityLabelPtr seclabels = NULL;
    int i, len, rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if ((len = virDomainGetSecurityLabelList(dom, &seclabels)) < 0) {
        ret->ret = len;
        ret->labels.labels_len = 0;
        ret->labels.labels_val = NULL;
        goto done;
    }

    if (VIR_ALLOC_N(ret->labels.labels_val, len) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    for (i = 0; i < len; i++) {
        size_t label_len = strlen(seclabels[i].label) + 1;
        remote_domain_get_security_label_ret *cur = &ret->labels.labels_val[i];
        if (VIR_ALLOC_N(cur->label.label_val, label_len) < 0) {
            virReportOOMError();
            goto cleanup;
        }
        if (virStrcpy(cur->label.label_val, seclabels[i].label, label_len) == NULL) {
            virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                           _("failed to copy security label"));
            goto cleanup;
        }
        cur->label.label_len = label_len;
        cur->enforcing = seclabels[i].enforcing;
    }
    ret->labels.labels_len = ret->ret = len;

done:
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (dom)
        virDomainFree(dom);
    VIR_FREE(seclabels);
    return rv;
}

1510
static int
1511 1512
remoteDispatchNodeGetSecurityModel(virNetServerPtr server ATTRIBUTE_UNUSED,
                                   virNetServerClientPtr client ATTRIBUTE_UNUSED,
1513
                                   virNetMessagePtr msg ATTRIBUTE_UNUSED,
1514
                                   virNetMessageErrorPtr rerr,
1515
                                   remote_node_get_security_model_ret *ret)
1516
{
1517
    virSecurityModel secmodel;
1518
    int rv = -1;
1519 1520
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1521

1522
    if (!priv->conn) {
1523
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1524
        goto cleanup;
1525 1526
    }

1527
    memset(&secmodel, 0, sizeof(secmodel));
1528
    if (virNodeGetSecurityModel(priv->conn, &secmodel) < 0)
1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540
        goto cleanup;

    ret->model.model_len = strlen(secmodel.model) + 1;
    if (VIR_ALLOC_N(ret->model.model_val, ret->model.model_len) < 0) {
        virReportOOMError();
        goto cleanup;
    }
    strcpy(ret->model.model_val, secmodel.model);

    ret->doi.doi_len = strlen(secmodel.doi) + 1;
    if (VIR_ALLOC_N(ret->doi.doi_val, ret->doi.doi_len) < 0) {
        virReportOOMError();
1541
        goto cleanup;
1542 1543
    }
    strcpy(ret->doi.doi_val, secmodel.doi);
1544

1545 1546 1547 1548
    rv = 0;

cleanup:
    if (rv < 0)
1549
        virNetMessageSaveError(rerr);
1550
    return rv;
1551 1552
}

1553
static int
1554 1555
remoteDispatchDomainGetVcpuPinInfo(virNetServerPtr server ATTRIBUTE_UNUSED,
                                   virNetServerClientPtr client ATTRIBUTE_UNUSED,
1556
                                   virNetMessagePtr msg ATTRIBUTE_UNUSED,
1557
                                   virNetMessageErrorPtr rerr,
E
Eric Blake 已提交
1558 1559
                                   remote_domain_get_vcpu_pin_info_args *args,
                                   remote_domain_get_vcpu_pin_info_ret *ret)
1560 1561 1562 1563 1564
{
    virDomainPtr dom = NULL;
    unsigned char *cpumaps = NULL;
    int num;
    int rv = -1;
1565 1566
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1567

1568
    if (!priv->conn) {
1569
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1570 1571 1572
        goto cleanup;
    }

1573
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1574 1575 1576
        goto cleanup;

    if (args->ncpumaps > REMOTE_VCPUINFO_MAX) {
1577
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("ncpumaps > REMOTE_VCPUINFO_MAX"));
1578 1579 1580 1581 1582
        goto cleanup;
    }

    if (INT_MULTIPLY_OVERFLOW(args->ncpumaps, args->maplen) ||
        args->ncpumaps * args->maplen > REMOTE_CPUMAPS_MAX) {
1583
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("maxinfo * maplen > REMOTE_CPUMAPS_MAX"));
1584 1585 1586 1587 1588 1589 1590 1591
        goto cleanup;
    }

    /* Allocate buffers to take the results. */
    if (args->maplen > 0 &&
        VIR_ALLOC_N(cpumaps, args->ncpumaps * args->maplen) < 0)
        goto no_memory;

E
Eric Blake 已提交
1592
    if ((num = virDomainGetVcpuPinInfo(dom,
1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605
                                       args->ncpumaps,
                                       cpumaps,
                                       args->maplen,
                                       args->flags)) < 0)
        goto cleanup;

    ret->num = num;
    /* Don't need to allocate/copy the cpumaps if we make the reasonable
     * assumption that unsigned char and char are the same size.
     * Note that remoteDispatchClientRequest will free.
     */
    ret->cpumaps.cpumaps_len = args->ncpumaps * args->maplen;
    ret->cpumaps.cpumaps_val = (char *) cpumaps;
1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696
    cpumaps = NULL;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    VIR_FREE(cpumaps);
    if (dom)
        virDomainFree(dom);
    return rv;

no_memory:
    virReportOOMError();
    goto cleanup;
}

static int
remoteDispatchDomainPinEmulator(virNetServerPtr server ATTRIBUTE_UNUSED,
                                virNetServerClientPtr client,
                                virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                virNetMessageErrorPtr rerr,
                                remote_domain_pin_emulator_args *args)
{
    int rv = -1;
    virDomainPtr dom = NULL;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if (virDomainPinEmulator(dom,
                             (unsigned char *) args->cpumap.cpumap_val,
                             args->cpumap.cpumap_len,
                             args->flags) < 0)
        goto cleanup;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (dom)
        virDomainFree(dom);
    return rv;
}


static int
remoteDispatchDomainGetEmulatorPinInfo(virNetServerPtr server ATTRIBUTE_UNUSED,
                                       virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                       virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                       virNetMessageErrorPtr rerr,
                                       remote_domain_get_emulator_pin_info_args *args,
                                       remote_domain_get_emulator_pin_info_ret *ret)
{
    virDomainPtr dom = NULL;
    unsigned char *cpumaps = NULL;
    int r;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    /* Allocate buffers to take the results */
    if (args->maplen > 0 &&
        VIR_ALLOC_N(cpumaps, args->maplen) < 0)
        goto no_memory;

    if ((r = virDomainGetEmulatorPinInfo(dom,
                                         cpumaps,
                                         args->maplen,
                                         args->flags)) < 0)
        goto cleanup;

    ret->ret = r;
    ret->cpumaps.cpumaps_len = args->maplen;
    ret->cpumaps.cpumaps_val = (char *) cpumaps;
1697 1698 1699 1700 1701 1702
    cpumaps = NULL;

    rv = 0;

cleanup:
    if (rv < 0)
1703
        virNetMessageSaveError(rerr);
1704 1705 1706 1707 1708 1709 1710 1711 1712 1713
    VIR_FREE(cpumaps);
    if (dom)
        virDomainFree(dom);
    return rv;

no_memory:
    virReportOOMError();
    goto cleanup;
}

1714
static int
1715 1716
remoteDispatchDomainGetVcpus(virNetServerPtr server ATTRIBUTE_UNUSED,
                             virNetServerClientPtr client ATTRIBUTE_UNUSED,
1717
                             virNetMessagePtr msg ATTRIBUTE_UNUSED,
1718
                             virNetMessageErrorPtr rerr,
1719 1720
                             remote_domain_get_vcpus_args *args,
                             remote_domain_get_vcpus_ret *ret)
1721
{
1722
    virDomainPtr dom = NULL;
1723 1724 1725
    virVcpuInfoPtr info = NULL;
    unsigned char *cpumaps = NULL;
    int info_len, i;
1726
    int rv = -1;
1727 1728
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1729

1730
    if (!priv->conn) {
1731
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1732
        goto cleanup;
1733 1734
    }

1735
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1736
        goto cleanup;
1737

1738
    if (args->maxinfo > REMOTE_VCPUINFO_MAX) {
1739
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("maxinfo > REMOTE_VCPUINFO_MAX"));
1740
        goto cleanup;
1741
    }
1742

1743 1744
    if (INT_MULTIPLY_OVERFLOW(args->maxinfo, args->maplen) ||
        args->maxinfo * args->maplen > REMOTE_CPUMAPS_MAX) {
1745
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("maxinfo * maplen > REMOTE_CPUMAPS_MAX"));
1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781
        goto cleanup;
    }

    /* Allocate buffers to take the results. */
    if (VIR_ALLOC_N(info, args->maxinfo) < 0)
        goto no_memory;
    if (args->maplen > 0 &&
        VIR_ALLOC_N(cpumaps, args->maxinfo * args->maplen) < 0)
        goto no_memory;

    if ((info_len = virDomainGetVcpus(dom,
                                      info, args->maxinfo,
                                      cpumaps, args->maplen)) < 0)
        goto cleanup;

    /* Allocate the return buffer for info. */
    ret->info.info_len = info_len;
    if (VIR_ALLOC_N(ret->info.info_val, info_len) < 0)
        goto no_memory;

    for (i = 0; i < info_len; ++i) {
        ret->info.info_val[i].number = info[i].number;
        ret->info.info_val[i].state = info[i].state;
        ret->info.info_val[i].cpu_time = info[i].cpuTime;
        ret->info.info_val[i].cpu = info[i].cpu;
    }

    /* Don't need to allocate/copy the cpumaps if we make the reasonable
     * assumption that unsigned char and char are the same size.
     * Note that remoteDispatchClientRequest will free.
     */
    ret->cpumaps.cpumaps_len = args->maxinfo * args->maplen;
    ret->cpumaps.cpumaps_val = (char *) cpumaps;
    cpumaps = NULL;

    rv = 0;
1782 1783

cleanup:
1784
    if (rv < 0) {
1785
        virNetMessageSaveError(rerr);
1786 1787 1788 1789
        VIR_FREE(ret->info.info_val);
    }
    VIR_FREE(cpumaps);
    VIR_FREE(info);
1790 1791 1792
    if (dom)
        virDomainFree(dom);
    return rv;
1793 1794 1795 1796

no_memory:
    virReportOOMError();
    goto cleanup;
1797 1798 1799
}

static int
1800 1801
remoteDispatchDomainMigratePrepare(virNetServerPtr server ATTRIBUTE_UNUSED,
                                   virNetServerClientPtr client ATTRIBUTE_UNUSED,
1802
                                   virNetMessagePtr msg ATTRIBUTE_UNUSED,
1803
                                   virNetMessageErrorPtr rerr,
1804 1805
                                   remote_domain_migrate_prepare_args *args,
                                   remote_domain_migrate_prepare_ret *ret)
1806
{
1807 1808 1809 1810 1811
    char *cookie = NULL;
    int cookielen = 0;
    char *uri_in;
    char **uri_out;
    char *dname;
1812
    int rv = -1;
1813 1814
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1815

1816
    if (!priv->conn) {
1817
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1818
        goto cleanup;
1819 1820
    }

1821 1822 1823 1824 1825 1826
    uri_in = args->uri_in == NULL ? NULL : *args->uri_in;
    dname = args->dname == NULL ? NULL : *args->dname;

    /* Wacky world of XDR ... */
    if (VIR_ALLOC(uri_out) < 0) {
        virReportOOMError();
1827
        goto cleanup;
1828
    }
1829

1830
    if (virDomainMigratePrepare(priv->conn, &cookie, &cookielen,
1831 1832
                                uri_in, uri_out,
                                args->flags, dname, args->resource) < 0)
1833
        goto cleanup;
1834

1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845
    /* remoteDispatchClientRequest will free cookie, uri_out and
     * the string if there is one.
     */
    ret->cookie.cookie_len = cookielen;
    ret->cookie.cookie_val = cookie;
    if (*uri_out == NULL) {
        ret->uri_out = NULL;
    } else {
        ret->uri_out = uri_out;
        uri_out = NULL;
    }
1846

1847
    rv = 0;
1848

1849 1850
cleanup:
    if (rv < 0)
1851
        virNetMessageSaveError(rerr);
1852
    VIR_FREE(uri_out);
1853
    return rv;
1854 1855
}

1856
static int
1857 1858
remoteDispatchDomainMigratePrepare2(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client ATTRIBUTE_UNUSED,
1859
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
1860
                                    virNetMessageErrorPtr rerr,
1861 1862
                                    remote_domain_migrate_prepare2_args *args,
                                    remote_domain_migrate_prepare2_ret *ret)
1863
{
1864 1865 1866 1867 1868
    char *cookie = NULL;
    int cookielen = 0;
    char *uri_in;
    char **uri_out;
    char *dname;
1869
    int rv = -1;
1870 1871
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1872

1873
    if (!priv->conn) {
1874
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1875
        goto cleanup;
1876 1877
    }

1878 1879
    uri_in = args->uri_in == NULL ? NULL : *args->uri_in;
    dname = args->dname == NULL ? NULL : *args->dname;
1880

1881 1882
    /* Wacky world of XDR ... */
    if (VIR_ALLOC(uri_out) < 0) {
1883 1884
        virReportOOMError();
        goto cleanup;
1885 1886
    }

1887
    if (virDomainMigratePrepare2(priv->conn, &cookie, &cookielen,
1888 1889 1890
                                 uri_in, uri_out,
                                 args->flags, dname, args->resource,
                                 args->dom_xml) < 0)
1891
        goto cleanup;
1892

1893 1894 1895 1896 1897 1898
    /* remoteDispatchClientRequest will free cookie, uri_out and
     * the string if there is one.
     */
    ret->cookie.cookie_len = cookielen;
    ret->cookie.cookie_val = cookie;
    ret->uri_out = *uri_out == NULL ? NULL : uri_out;
1899

1900 1901 1902 1903
    rv = 0;

cleanup:
    if (rv < 0)
1904
        virNetMessageSaveError(rerr);
1905
    return rv;
1906 1907
}

C
Chris Lalancette 已提交
1908
static int
1909 1910
remoteDispatchDomainGetMemoryParameters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                        virNetServerClientPtr client ATTRIBUTE_UNUSED,
1911
                                        virNetMessagePtr msg ATTRIBUTE_UNUSED,
1912 1913 1914
                                        virNetMessageErrorPtr rerr,
                                        remote_domain_get_memory_parameters_args *args,
                                        remote_domain_get_memory_parameters_ret *ret)
C
Chris Lalancette 已提交
1915
{
1916
    virDomainPtr dom = NULL;
1917
    virTypedParameterPtr params = NULL;
1918
    int nparams = 0;
1919
    unsigned int flags;
1920
    int rv = -1;
1921 1922
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
1923

1924
    if (!priv->conn) {
1925
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1926
        goto cleanup;
1927
    }
C
Chris Lalancette 已提交
1928

1929
    flags = args->flags;
C
Chris Lalancette 已提交
1930

1931
    if (args->nparams > REMOTE_DOMAIN_MEMORY_PARAMETERS_MAX) {
1932
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
1933 1934
        goto cleanup;
    }
1935
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
1936 1937
        virReportOOMError();
        goto cleanup;
C
Chris Lalancette 已提交
1938
    }
1939
    nparams = args->nparams;
C
Chris Lalancette 已提交
1940

1941
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
1942
        goto cleanup;
C
Chris Lalancette 已提交
1943

1944
    if (virDomainGetMemoryParameters(dom, params, &nparams, flags) < 0)
1945
        goto cleanup;
C
Chris Lalancette 已提交
1946

1947 1948 1949 1950 1951 1952
    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
1953 1954
    }

1955
    if (remoteSerializeTypedParameters(params, nparams,
1956
                                       &ret->params.params_val,
1957 1958
                                       &ret->params.params_len,
                                       args->flags) < 0)
1959
        goto cleanup;
1960

1961
success:
1962 1963 1964
    rv = 0;

cleanup:
1965
    if (rv < 0)
1966
        virNetMessageSaveError(rerr);
1967
    virTypedParamsFree(params, nparams);
1968 1969 1970
    if (dom)
        virDomainFree(dom);
    return rv;
1971 1972
}

1973 1974 1975 1976 1977 1978 1979 1980 1981 1982
static int
remoteDispatchDomainGetNumaParameters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                      virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                      virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                      virNetMessageErrorPtr rerr,
                                      remote_domain_get_numa_parameters_args *args,
                                      remote_domain_get_numa_parameters_ret *ret)
{
    virDomainPtr dom = NULL;
    virTypedParameterPtr params = NULL;
1983
    int nparams = 0;
1984 1985 1986 1987 1988 1989
    unsigned int flags;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
1990
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
1991 1992 1993 1994 1995
        goto cleanup;
    }

    flags = args->flags;

1996
    if (args->nparams > REMOTE_DOMAIN_NUMA_PARAMETERS_MAX) {
1997
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
1998 1999
        goto cleanup;
    }
2000
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
2001 2002 2003
        virReportOOMError();
        goto cleanup;
    }
2004
    nparams = args->nparams;
2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if (virDomainGetNumaParameters(dom, params, &nparams, flags) < 0)
        goto cleanup;

    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    if (remoteSerializeTypedParameters(params, nparams,
                                       &ret->params.params_val,
                                       &ret->params.params_len,
                                       flags) < 0)
        goto cleanup;

success:
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
2032
    virTypedParamsFree(params, nparams);
2033 2034 2035 2036 2037
    if (dom)
        virDomainFree(dom);
    return rv;
}

2038
static int
2039 2040
remoteDispatchDomainGetBlkioParameters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                       virNetServerClientPtr client ATTRIBUTE_UNUSED,
2041
                                       virNetMessagePtr msg ATTRIBUTE_UNUSED,
2042 2043 2044
                                       virNetMessageErrorPtr rerr,
                                       remote_domain_get_blkio_parameters_args *args,
                                       remote_domain_get_blkio_parameters_ret *ret)
2045
{
2046
    virDomainPtr dom = NULL;
2047
    virTypedParameterPtr params = NULL;
2048
    int nparams = 0;
2049
    unsigned int flags;
2050
    int rv = -1;
2051 2052
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2053

2054
    if (!priv->conn) {
2055
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
2056
        goto cleanup;
2057 2058
    }

2059 2060
    flags = args->flags;

2061
    if (args->nparams > REMOTE_DOMAIN_BLKIO_PARAMETERS_MAX) {
2062
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
2063
        goto cleanup;
2064
    }
2065
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
2066 2067
        virReportOOMError();
        goto cleanup;
2068
    }
2069
    nparams = args->nparams;
2070

2071
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
2072
        goto cleanup;
2073

2074
    if (virDomainGetBlkioParameters(dom, params, &nparams, flags) < 0)
2075
        goto cleanup;
2076

2077 2078 2079 2080 2081 2082 2083 2084
    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

2085
    if (remoteSerializeTypedParameters(params, nparams,
2086
                                       &ret->params.params_val,
2087 2088
                                       &ret->params.params_len,
                                       args->flags) < 0)
2089
        goto cleanup;
2090

2091 2092
success:
    rv = 0;
2093

2094
cleanup:
2095
    if (rv < 0)
2096
        virNetMessageSaveError(rerr);
2097
    virTypedParamsFree(params, nparams);
2098 2099 2100
    if (dom)
        virDomainFree(dom);
    return rv;
2101 2102
}

2103
static int
2104 2105
remoteDispatchNodeGetCPUStats(virNetServerPtr server ATTRIBUTE_UNUSED,
                              virNetServerClientPtr client ATTRIBUTE_UNUSED,
2106
                              virNetMessagePtr msg ATTRIBUTE_UNUSED,
2107 2108 2109
                              virNetMessageErrorPtr rerr,
                              remote_node_get_cpu_stats_args *args,
                              remote_node_get_cpu_stats_ret *ret)
2110
{
2111
    virNodeCPUStatsPtr params = NULL;
2112 2113
    int i;
    int cpuNum = args->cpuNum;
2114
    int nparams = 0;
2115 2116
    unsigned int flags;
    int rv = -1;
2117 2118
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2119

2120
    if (!priv->conn) {
2121
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
2122 2123 2124 2125 2126
        goto cleanup;
    }

    flags = args->flags;

2127
    if (args->nparams > REMOTE_NODE_CPU_STATS_MAX) {
2128
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
2129 2130
        goto cleanup;
    }
2131
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
2132 2133 2134
        virReportOOMError();
        goto cleanup;
    }
2135
    nparams = args->nparams;
2136

2137
    if (virNodeGetCPUStats(priv->conn, cpuNum, params, &nparams, flags) < 0)
2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164
        goto cleanup;

    /* In this case, we need to send back the number of stats
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    /* Serialise the memory parameters. */
    ret->params.params_len = nparams;
    if (VIR_ALLOC_N(ret->params.params_val, nparams) < 0)
        goto no_memory;

    for (i = 0; i < nparams; ++i) {
        /* remoteDispatchClientRequest will free this: */
        ret->params.params_val[i].field = strdup(params[i].field);
        if (ret->params.params_val[i].field == NULL)
            goto no_memory;

        ret->params.params_val[i].value = params[i].value;
    }

success:
    rv = 0;

2165 2166
cleanup:
    if (rv < 0) {
2167
        virNetMessageSaveError(rerr);
2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182
        if (ret->params.params_val) {
            for (i = 0; i < nparams; i++)
                VIR_FREE(ret->params.params_val[i].field);
            VIR_FREE(ret->params.params_val);
        }
    }
    VIR_FREE(params);
    return rv;

no_memory:
    virReportOOMError();
    goto cleanup;
}

static int
2183 2184
remoteDispatchNodeGetMemoryStats(virNetServerPtr server ATTRIBUTE_UNUSED,
                                 virNetServerClientPtr client ATTRIBUTE_UNUSED,
2185
                                 virNetMessagePtr msg ATTRIBUTE_UNUSED,
2186 2187 2188
                                 virNetMessageErrorPtr rerr,
                                 remote_node_get_memory_stats_args *args,
                                 remote_node_get_memory_stats_ret *ret)
2189
{
2190
    virNodeMemoryStatsPtr params = NULL;
2191 2192
    int i;
    int cellNum = args->cellNum;
2193
    int nparams = 0;
2194 2195
    unsigned int flags;
    int rv = -1;
2196 2197
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2198

2199
    if (!priv->conn) {
2200
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
2201 2202 2203 2204 2205
        goto cleanup;
    }

    flags = args->flags;

2206
    if (args->nparams > REMOTE_NODE_MEMORY_STATS_MAX) {
2207
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
2208 2209
        goto cleanup;
    }
2210
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
2211 2212 2213
        virReportOOMError();
        goto cleanup;
    }
2214
    nparams = args->nparams;
2215

2216
    if (virNodeGetMemoryStats(priv->conn, cellNum, params, &nparams, flags) < 0)
2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243
        goto cleanup;

    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    /* Serialise the memory parameters. */
    ret->params.params_len = nparams;
    if (VIR_ALLOC_N(ret->params.params_val, nparams) < 0)
        goto no_memory;

    for (i = 0; i < nparams; ++i) {
        /* remoteDispatchClientRequest will free this: */
        ret->params.params_val[i].field = strdup(params[i].field);
        if (ret->params.params_val[i].field == NULL)
            goto no_memory;

        ret->params.params_val[i].value = params[i].value;
    }

success:
    rv = 0;

2244 2245
cleanup:
    if (rv < 0) {
2246
        virNetMessageSaveError(rerr);
2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260
        if (ret->params.params_val) {
            for (i = 0; i < nparams; i++)
                VIR_FREE(ret->params.params_val[i].field);
            VIR_FREE(ret->params.params_val);
        }
    }
    VIR_FREE(params);
    return rv;

no_memory:
    virReportOOMError();
    goto cleanup;
}

2261 2262 2263
static int
remoteDispatchDomainGetBlockJobInfo(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client ATTRIBUTE_UNUSED,
2264
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275
                                    virNetMessageErrorPtr rerr,
                                    remote_domain_get_block_job_info_args *args,
                                    remote_domain_get_block_job_info_ret *ret)
{
    virDomainPtr dom = NULL;
    virDomainBlockJobInfo tmp;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
2276
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    rv = virDomainGetBlockJobInfo(dom, args->path, &tmp, args->flags);
    if (rv <= 0)
        goto cleanup;

    ret->type = tmp.type;
    ret->bandwidth = tmp.bandwidth;
    ret->cur = tmp.cur;
    ret->end = tmp.end;
    ret->found = 1;
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (dom)
        virDomainFree(dom);
    return rv;
}

2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312
static int
remoteDispatchDomainGetBlockIoTune(virNetServerPtr server ATTRIBUTE_UNUSED,
                                   virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                   virNetMessagePtr hdr ATTRIBUTE_UNUSED,
                                   virNetMessageErrorPtr rerr,
                                   remote_domain_get_block_io_tune_args *args,
                                   remote_domain_get_block_io_tune_ret *ret)
{
    virDomainPtr dom = NULL;
    int rv = -1;
    virTypedParameterPtr params = NULL;
2313
    int nparams = 0;
2314 2315 2316 2317
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
2318
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
2319 2320 2321
        goto cleanup;
    }

2322
    if (args->nparams > REMOTE_DOMAIN_BLOCK_IO_TUNE_PARAMETERS_MAX) {
2323
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
2324 2325 2326
        goto cleanup;
    }

2327
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
2328 2329 2330
        virReportOOMError();
        goto cleanup;
    }
2331
    nparams = args->nparams;
2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356 2357 2358 2359 2360

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if (virDomainGetBlockIoTune(dom, args->disk ? *args->disk : NULL,
                                params, &nparams, args->flags) < 0)
        goto cleanup;

    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    /* Serialise the block I/O tuning parameters. */
    if (remoteSerializeTypedParameters(params, nparams,
                                       &ret->params.params_val,
                                       &ret->params.params_len,
                                       args->flags) < 0)
        goto cleanup;

success:
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
2361
    virTypedParamsFree(params, nparams);
2362 2363 2364 2365
    if (dom)
        virDomainFree(dom);
    return rv;
}
2366

D
Daniel Veillard 已提交
2367 2368
/*-------------------------------------------------------------*/

2369
static int
2370 2371
remoteDispatchAuthList(virNetServerPtr server ATTRIBUTE_UNUSED,
                       virNetServerClientPtr client,
2372
                       virNetMessagePtr msg ATTRIBUTE_UNUSED,
2373
                       virNetMessageErrorPtr rerr,
2374
                       remote_auth_list_ret *ret)
2375
{
2376
    int rv = -1;
2377 2378
    int auth = virNetServerClientGetAuth(client);
    uid_t callerUid;
2379
    gid_t callerGid;
2380 2381 2382 2383 2384 2385 2386
    pid_t callerPid;

    /* If the client is root then we want to bypass the
     * policykit auth to avoid root being denied if
     * some piece of polkit isn't present/running
     */
    if (auth == VIR_NET_SERVER_SERVICE_AUTH_POLKIT) {
2387 2388
        if (virNetServerClientGetUNIXIdentity(client, &callerUid, &callerGid,
                                              &callerPid) < 0) {
2389 2390 2391 2392
            /* Don't do anything on error - it'll be validated at next
             * phase of auth anyway */
            virResetLastError();
        } else if (callerUid == 0) {
2393 2394
            char *ident;
            if (virAsprintf(&ident, "pid:%lld,uid:%d",
J
Jim Fehlig 已提交
2395 2396 2397
                            (long long) callerPid, callerUid) < 0) {
                virReportOOMError();
                goto cleanup;
2398
            }
J
Jim Fehlig 已提交
2399
            VIR_INFO("Bypass polkit auth for privileged client %s", ident);
2400 2401
            virNetServerClientSetAuth(client, 0);
            auth = VIR_NET_SERVER_SERVICE_AUTH_NONE;
J
Jim Fehlig 已提交
2402
            VIR_FREE(ident);
2403 2404
        }
    }
2405

2406
    ret->types.types_len = 1;
2407
    if (VIR_ALLOC_N(ret->types.types_val, ret->types.types_len) < 0) {
2408 2409
        virReportOOMError();
        goto cleanup;
2410
    }
2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424

    switch (auth) {
    case VIR_NET_SERVER_SERVICE_AUTH_NONE:
        ret->types.types_val[0] = REMOTE_AUTH_NONE;
        break;
    case VIR_NET_SERVER_SERVICE_AUTH_POLKIT:
        ret->types.types_val[0] = REMOTE_AUTH_POLKIT;
        break;
    case VIR_NET_SERVER_SERVICE_AUTH_SASL:
        ret->types.types_val[0] = REMOTE_AUTH_SASL;
        break;
    default:
        ret->types.types_val[0] = REMOTE_AUTH_NONE;
    }
2425

2426 2427 2428 2429
    rv = 0;

cleanup:
    if (rv < 0)
2430
        virNetMessageSaveError(rerr);
2431
    return rv;
2432 2433 2434
}


2435
#ifdef WITH_SASL
2436 2437
/*
 * Initializes the SASL session in prepare for authentication
2438
 * and gives the client a list of allowed mechanisms to choose
2439 2440
 */
static int
2441 2442
remoteDispatchAuthSaslInit(virNetServerPtr server ATTRIBUTE_UNUSED,
                           virNetServerClientPtr client,
2443
                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
2444
                           virNetMessageErrorPtr rerr,
2445
                           remote_auth_sasl_init_ret *ret)
2446
{
2447 2448 2449
    virNetSASLSessionPtr sasl = NULL;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2450

2451
    virMutexLock(&priv->lock);
2452

2453 2454 2455
    VIR_DEBUG("Initialize SASL auth %d", virNetServerClientGetFD(client));
    if (virNetServerClientGetAuth(client) != VIR_NET_SERVER_SERVICE_AUTH_SASL ||
        priv->sasl != NULL) {
2456
        VIR_ERROR(_("client tried invalid SASL init request"));
2457
        goto authfail;
2458 2459
    }

2460 2461 2462 2463 2464
    sasl = virNetSASLSessionNewServer(saslCtxt,
                                      "libvirt",
                                      virNetServerClientLocalAddrString(client),
                                      virNetServerClientRemoteAddrString(client));
    if (!sasl)
2465
        goto authfail;
2466

2467
# if WITH_GNUTLS
2468
    /* Inform SASL that we've got an external SSF layer from TLS */
2469 2470 2471 2472
    if (virNetServerClientHasTLSSession(client)) {
        int ssf;

        if ((ssf = virNetServerClientGetTLSKeySize(client)) < 0)
2473
            goto authfail;
2474 2475 2476 2477 2478

        ssf *= 8; /* key size is bytes, sasl wants bits */

        VIR_DEBUG("Setting external SSF %d", ssf);
        if (virNetSASLSessionExtKeySize(sasl, ssf) < 0)
2479
            goto authfail;
2480
    }
2481
# endif
2482

2483
    if (virNetServerClientIsSecure(client))
2484
        /* If we've got TLS or UNIX domain sock, we don't care about SSF */
2485 2486
        virNetSASLSessionSecProps(sasl, 0, 0, true);
    else
2487
        /* Plain TCP, better get an SSF layer */
2488 2489 2490 2491
        virNetSASLSessionSecProps(sasl,
                                  56,  /* Good enough to require kerberos */
                                  100000,  /* Arbitrary big number */
                                  false); /* No anonymous */
2492

2493
    if (!(ret->mechlist = virNetSASLSessionListMechanisms(sasl)))
2494
        goto authfail;
2495
    VIR_DEBUG("Available mechanisms for client: '%s'", ret->mechlist);
2496

2497 2498
    priv->sasl = sasl;
    virMutexUnlock(&priv->lock);
2499
    return 0;
2500 2501

authfail:
2502
    virResetLastError();
2503 2504
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
2505
    virNetMessageSaveError(rerr);
2506 2507 2508
    PROBE(RPC_SERVER_CLIENT_AUTH_FAIL,
          "client=%p auth=%d",
          client, REMOTE_AUTH_SASL);
2509
    virObjectUnref(sasl);
2510
    virMutexUnlock(&priv->lock);
2511
    return -1;
2512 2513
}

2514
/*
2515 2516
 * Returns 0 if ok, -1 on error, -2 if rejected
 */
2517
static int
2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533
remoteSASLFinish(virNetServerClientPtr client)
{
    const char *identity;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);
    int ssf;

    /* TLS or UNIX domain sockets trivially OK */
    if (!virNetServerClientIsSecure(client)) {
        if ((ssf = virNetSASLSessionGetKeySize(priv->sasl)) < 0)
            goto error;

        VIR_DEBUG("negotiated an SSF of %d", ssf);
        if (ssf < 56) { /* 56 is good for Kerberos */
            VIR_ERROR(_("negotiated SSF %d was not strong enough"), ssf);
            return -2;
        }
2534
    }
2535 2536

    if (!(identity = virNetSASLSessionGetIdentity(priv->sasl)))
2537
        return -2;
2538

2539 2540
    if (!virNetSASLContextCheckIdentity(saslCtxt, identity))
        return -2;
2541

2542
    virNetServerClientSetAuth(client, 0);
2543
    virNetServerClientSetSASLSession(client, priv->sasl);
2544

2545
    VIR_DEBUG("Authentication successful %d", virNetServerClientGetFD(client));
2546

2547 2548 2549
    PROBE(RPC_SERVER_CLIENT_AUTH_ALLOW,
          "client=%p auth=%d identity=%s",
          client, REMOTE_AUTH_SASL, identity);
2550

2551
    virObjectUnref(priv->sasl);
2552
    priv->sasl = NULL;
2553

2554
    return 0;
2555

2556 2557 2558
error:
    return -1;
}
2559

2560 2561 2562 2563
/*
 * This starts the SASL authentication negotiation.
 */
static int
2564 2565
remoteDispatchAuthSaslStart(virNetServerPtr server ATTRIBUTE_UNUSED,
                            virNetServerClientPtr client,
2566
                            virNetMessagePtr msg ATTRIBUTE_UNUSED,
2567
                            virNetMessageErrorPtr rerr,
2568 2569
                            remote_auth_sasl_start_args *args,
                            remote_auth_sasl_start_ret *ret)
2570 2571
{
    const char *serverout;
2572
    size_t serveroutlen;
2573
    int err;
2574 2575 2576
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2577
    const char *identity;
2578

2579
    virMutexLock(&priv->lock);
2580

2581 2582 2583
    VIR_DEBUG("Start SASL auth %d", virNetServerClientGetFD(client));
    if (virNetServerClientGetAuth(client) != VIR_NET_SERVER_SERVICE_AUTH_SASL ||
        priv->sasl == NULL) {
2584
        VIR_ERROR(_("client tried invalid SASL start request"));
2585
        goto authfail;
2586 2587
    }

2588 2589
    VIR_DEBUG("Using SASL mechanism %s. Data %d bytes, nil: %d",
              args->mech, args->data.data_len, args->nil);
2590 2591 2592 2593 2594 2595 2596 2597 2598
    err = virNetSASLSessionServerStart(priv->sasl,
                                       args->mech,
                                       /* NB, distinction of NULL vs "" is *critical* in SASL */
                                       args->nil ? NULL : args->data.data_val,
                                       args->data.data_len,
                                       &serverout,
                                       &serveroutlen);
    if (err != VIR_NET_SASL_COMPLETE &&
        err != VIR_NET_SASL_CONTINUE)
2599
        goto authfail;
2600

2601
    if (serveroutlen > REMOTE_AUTH_SASL_DATA_MAX) {
2602
        VIR_ERROR(_("sasl start reply data too long %d"), (int)serveroutlen);
2603
        goto authfail;
2604 2605 2606 2607
    }

    /* NB, distinction of NULL vs "" is *critical* in SASL */
    if (serverout) {
2608 2609
        if (VIR_ALLOC_N(ret->data.data_val, serveroutlen) < 0)
            goto authfail;
2610 2611 2612 2613 2614 2615 2616
        memcpy(ret->data.data_val, serverout, serveroutlen);
    } else {
        ret->data.data_val = NULL;
    }
    ret->nil = serverout ? 0 : 1;
    ret->data.data_len = serveroutlen;

2617
    VIR_DEBUG("SASL return data %d bytes, nil; %d", ret->data.data_len, ret->nil);
2618
    if (err == VIR_NET_SASL_CONTINUE) {
2619 2620
        ret->complete = 0;
    } else {
2621
        /* Check username whitelist ACL */
2622
        if ((err = remoteSASLFinish(client)) < 0) {
2623 2624 2625 2626 2627
            if (err == -2)
                goto authdeny;
            else
                goto authfail;
        }
2628

2629 2630 2631
        ret->complete = 1;
    }

2632
    virMutexUnlock(&priv->lock);
2633
    return 0;
2634 2635

authfail:
2636 2637 2638
    PROBE(RPC_SERVER_CLIENT_AUTH_FAIL,
          "client=%p auth=%d",
          client, REMOTE_AUTH_SASL);
2639 2640 2641
    goto error;

authdeny:
2642
    identity = virNetSASLSessionGetIdentity(priv->sasl);
2643 2644 2645
    PROBE(RPC_SERVER_CLIENT_AUTH_DENY,
          "client=%p auth=%d identity=%s",
          client, REMOTE_AUTH_SASL, identity);
2646 2647
    goto error;

2648
error:
2649
    virObjectUnref(priv->sasl);
2650 2651
    priv->sasl = NULL;
    virResetLastError();
2652 2653
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
2654 2655 2656
    if (rv < 0)
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
2657
    return -1;
2658 2659 2660 2661
}


static int
2662 2663
remoteDispatchAuthSaslStep(virNetServerPtr server ATTRIBUTE_UNUSED,
                           virNetServerClientPtr client,
2664
                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
2665
                           virNetMessageErrorPtr rerr,
2666 2667
                           remote_auth_sasl_step_args *args,
                           remote_auth_sasl_step_ret *ret)
2668 2669
{
    const char *serverout;
2670
    size_t serveroutlen;
2671
    int err;
2672 2673 2674
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2675
    const char *identity;
2676

2677 2678 2679 2680 2681
    virMutexLock(&priv->lock);

    VIR_DEBUG("Step SASL auth %d", virNetServerClientGetFD(client));
    if (virNetServerClientGetAuth(client) != VIR_NET_SERVER_SERVICE_AUTH_SASL ||
        priv->sasl == NULL) {
2682
        VIR_ERROR(_("client tried invalid SASL start request"));
2683
        goto authfail;
2684 2685
    }

2686
    VIR_DEBUG("Step using SASL Data %d bytes, nil: %d",
2687
              args->data.data_len, args->nil);
2688 2689 2690 2691 2692 2693 2694 2695
    err = virNetSASLSessionServerStep(priv->sasl,
                                      /* NB, distinction of NULL vs "" is *critical* in SASL */
                                      args->nil ? NULL : args->data.data_val,
                                      args->data.data_len,
                                      &serverout,
                                      &serveroutlen);
    if (err != VIR_NET_SASL_COMPLETE &&
        err != VIR_NET_SASL_CONTINUE)
2696
        goto authfail;
2697 2698

    if (serveroutlen > REMOTE_AUTH_SASL_DATA_MAX) {
2699
        VIR_ERROR(_("sasl step reply data too long %d"),
2700
                  (int)serveroutlen);
2701
        goto authfail;
2702 2703 2704 2705
    }

    /* NB, distinction of NULL vs "" is *critical* in SASL */
    if (serverout) {
2706 2707
        if (VIR_ALLOC_N(ret->data.data_val, serveroutlen) < 0)
            goto authfail;
2708 2709 2710 2711 2712 2713 2714
        memcpy(ret->data.data_val, serverout, serveroutlen);
    } else {
        ret->data.data_val = NULL;
    }
    ret->nil = serverout ? 0 : 1;
    ret->data.data_len = serveroutlen;

2715
    VIR_DEBUG("SASL return data %d bytes, nil; %d", ret->data.data_len, ret->nil);
2716
    if (err == VIR_NET_SASL_CONTINUE) {
2717 2718
        ret->complete = 0;
    } else {
2719
        /* Check username whitelist ACL */
2720
        if ((err = remoteSASLFinish(client)) < 0) {
2721 2722 2723 2724 2725
            if (err == -2)
                goto authdeny;
            else
                goto authfail;
        }
2726

2727 2728 2729
        ret->complete = 1;
    }

2730
    virMutexUnlock(&priv->lock);
2731
    return 0;
2732 2733

authfail:
2734 2735 2736
    PROBE(RPC_SERVER_CLIENT_AUTH_FAIL,
          "client=%p auth=%d",
          client, REMOTE_AUTH_SASL);
2737 2738 2739
    goto error;

authdeny:
2740
    identity = virNetSASLSessionGetIdentity(priv->sasl);
2741 2742 2743
    PROBE(RPC_SERVER_CLIENT_AUTH_DENY,
          "client=%p auth=%d identity=%s",
          client, REMOTE_AUTH_SASL, identity);
2744 2745
    goto error;

2746
error:
2747
    virObjectUnref(priv->sasl);
2748 2749
    priv->sasl = NULL;
    virResetLastError();
2750 2751
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
2752 2753 2754
    if (rv < 0)
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
2755 2756
    return -1;
}
2757 2758 2759 2760
#else
static int
remoteDispatchAuthSaslInit(virNetServerPtr server ATTRIBUTE_UNUSED,
                           virNetServerClientPtr client ATTRIBUTE_UNUSED,
2761
                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
2762 2763 2764 2765
                           virNetMessageErrorPtr rerr,
                           remote_auth_sasl_init_ret *ret ATTRIBUTE_UNUSED)
{
    VIR_WARN("Client tried unsupported SASL auth");
2766 2767
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
2768 2769 2770 2771 2772 2773
    virNetMessageSaveError(rerr);
    return -1;
}
static int
remoteDispatchAuthSaslStart(virNetServerPtr server ATTRIBUTE_UNUSED,
                            virNetServerClientPtr client ATTRIBUTE_UNUSED,
2774
                            virNetMessagePtr msg ATTRIBUTE_UNUSED,
2775 2776 2777 2778 2779
                            virNetMessageErrorPtr rerr,
                            remote_auth_sasl_start_args *args ATTRIBUTE_UNUSED,
                            remote_auth_sasl_start_ret *ret ATTRIBUTE_UNUSED)
{
    VIR_WARN("Client tried unsupported SASL auth");
2780 2781
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
2782 2783 2784 2785 2786 2787
    virNetMessageSaveError(rerr);
    return -1;
}
static int
remoteDispatchAuthSaslStep(virNetServerPtr server ATTRIBUTE_UNUSED,
                           virNetServerClientPtr client ATTRIBUTE_UNUSED,
2788
                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
2789 2790 2791 2792 2793
                           virNetMessageErrorPtr rerr,
                           remote_auth_sasl_step_args *args ATTRIBUTE_UNUSED,
                           remote_auth_sasl_step_ret *ret ATTRIBUTE_UNUSED)
{
    VIR_WARN("Client tried unsupported SASL auth");
2794 2795
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
2796 2797 2798 2799
    virNetMessageSaveError(rerr);
    return -1;
}
#endif
2800 2801 2802



2803
#if WITH_POLKIT1
2804
static int
2805 2806
remoteDispatchAuthPolkit(virNetServerPtr server ATTRIBUTE_UNUSED,
                         virNetServerClientPtr client,
2807
                         virNetMessagePtr msg ATTRIBUTE_UNUSED,
2808
                         virNetMessageErrorPtr rerr,
2809
                         remote_auth_polkit_ret *ret)
2810
{
2811
    pid_t callerPid = -1;
2812
    gid_t callerGid = -1;
2813
    uid_t callerUid = -1;
2814 2815
    const char *action;
    int status = -1;
2816
    char *ident = NULL;
2817
    bool authdismissed = 0;
2818
    char *pkout = NULL;
2819 2820
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2821
    virCommandPtr cmd = NULL;
2822

2823 2824
    virMutexLock(&priv->lock);
    action = virNetServerClientGetReadonly(client) ?
2825 2826 2827
        "org.libvirt.unix.monitor" :
        "org.libvirt.unix.manage";

2828
    cmd = virCommandNewArgList(PKCHECK_PATH, "--action-id", action, NULL);
2829
    virCommandSetOutputBuffer(cmd, &pkout);
2830
    virCommandSetErrorBuffer(cmd, &pkout);
2831

2832 2833
    VIR_DEBUG("Start PolicyKit auth %d", virNetServerClientGetFD(client));
    if (virNetServerClientGetAuth(client) != VIR_NET_SERVER_SERVICE_AUTH_POLKIT) {
2834
        VIR_ERROR(_("client tried invalid PolicyKit init request"));
2835 2836 2837
        goto authfail;
    }

2838 2839
    if (virNetServerClientGetUNIXIdentity(client, &callerUid, &callerGid,
                                          &callerPid) < 0) {
2840 2841 2842
        goto authfail;
    }

2843 2844
    VIR_INFO("Checking PID %lld running as %d",
             (long long) callerPid, callerUid);
2845

2846
    virCommandAddArg(cmd, "--process");
2847
    virCommandAddArgFormat(cmd, "%lld", (long long) callerPid);
2848
    virCommandAddArg(cmd, "--allow-user-interaction");
2849

2850 2851
    if (virAsprintf(&ident, "pid:%lld,uid:%d",
                    (long long) callerPid, callerUid) < 0) {
2852
        virReportOOMError();
2853 2854 2855
        goto authfail;
    }

2856
    if (virCommandRun(cmd, &status) < 0)
2857
        goto authfail;
2858

2859
    authdismissed = (pkout && strstr(pkout, "dismissed=true"));
2860
    if (status != 0) {
2861
        char *tmp = virProcessTranslateStatus(status);
2862 2863
        VIR_ERROR(_("Policy kit denied action %s from pid %lld, uid %d: %s"),
                  action, (long long) callerPid, callerUid, NULLSTR(tmp));
2864
        VIR_FREE(tmp);
2865
        goto authdeny;
2866
    }
2867 2868 2869
    PROBE(RPC_SERVER_CLIENT_AUTH_ALLOW,
          "client=%p auth=%d identity=%s",
          client, REMOTE_AUTH_POLKIT, ident);
2870 2871
    VIR_INFO("Policy allowed action %s from pid %lld, uid %d",
             action, (long long) callerPid, callerUid);
2872 2873
    ret->complete = 1;

2874
    virNetServerClientSetAuth(client, 0);
2875
    virMutexUnlock(&priv->lock);
2876
    virCommandFree(cmd);
E
Eric Blake 已提交
2877
    VIR_FREE(pkout);
2878
    VIR_FREE(ident);
2879

2880 2881
    return 0;

2882
error:
2883 2884
    virCommandFree(cmd);
    VIR_FREE(ident);
2885
    virResetLastError();
2886

2887
    if (authdismissed) {
2888 2889
        virReportError(VIR_ERR_AUTH_CANCELLED, "%s",
                       _("authentication cancelled by user"));
2890 2891
    } else if (pkout && *pkout) {
        virReportError(VIR_ERR_AUTH_FAILED, _("polkit: %s"), pkout);
2892
    } else {
2893
        virReportError(VIR_ERR_AUTH_FAILED, "%s", _("authentication failed"));
2894
    }
2895 2896

    VIR_FREE(pkout);
2897 2898 2899 2900
    virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
    return -1;

2901
authfail:
2902 2903 2904
    PROBE(RPC_SERVER_CLIENT_AUTH_FAIL,
          "client=%p auth=%d",
          client, REMOTE_AUTH_POLKIT);
2905 2906 2907
    goto error;

authdeny:
2908 2909
    PROBE(RPC_SERVER_CLIENT_AUTH_DENY,
          "client=%p auth=%d identity=%s",
2910
          client, REMOTE_AUTH_POLKIT, ident);
2911
    goto error;
2912
}
2913
#elif WITH_POLKIT0
2914
static int
2915
remoteDispatchAuthPolkit(virNetServerPtr server ATTRIBUTE_UNUSED,
2916
                         virNetServerClientPtr client,
2917
                         virNetMessagePtr msg ATTRIBUTE_UNUSED,
2918
                         virNetMessageErrorPtr rerr,
2919
                         remote_auth_polkit_ret *ret)
2920 2921
{
    pid_t callerPid;
2922
    gid_t callerGid;
2923
    uid_t callerUid;
2924 2925 2926 2927 2928 2929
    PolKitCaller *pkcaller = NULL;
    PolKitAction *pkaction = NULL;
    PolKitContext *pkcontext = NULL;
    PolKitError *pkerr = NULL;
    PolKitResult pkresult;
    DBusError err;
2930
    const char *action;
2931
    char *ident = NULL;
2932 2933
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
2934
    DBusConnection *sysbus;
2935

J
Jim Fehlig 已提交
2936
    virMutexLock(&priv->lock);
2937

J
Jim Fehlig 已提交
2938
    action = virNetServerClientGetReadonly(client) ?
2939 2940
        "org.libvirt.unix.monitor" :
        "org.libvirt.unix.manage";
2941

2942
    VIR_DEBUG("Start PolicyKit auth %d", virNetServerClientGetFD(client));
J
Jim Fehlig 已提交
2943
    if (virNetServerClientGetAuth(client) != VIR_NET_SERVER_SERVICE_AUTH_POLKIT) {
2944
        VIR_ERROR(_("client tried invalid PolicyKit init request"));
2945
        goto authfail;
2946 2947
    }

2948 2949
    if (virNetServerClientGetUNIXIdentity(client, &callerUid, &callerGid,
                                          &callerPid) < 0) {
2950
        VIR_ERROR(_("cannot get peer socket identity"));
2951
        goto authfail;
2952 2953
    }

2954 2955 2956
    if (virAsprintf(&ident, "pid:%lld,uid:%d",
                    (long long) callerPid, callerUid) < 0) {
        virReportOOMError();
2957 2958 2959
        goto authfail;
    }

2960 2961 2962
    if (!(sysbus = virDBusGetSystemBus()))
        goto authfail;

2963 2964
    VIR_INFO("Checking PID %lld running as %d",
             (long long) callerPid, callerUid);
2965
    dbus_error_init(&err);
2966
    if (!(pkcaller = polkit_caller_new_from_pid(sysbus,
2967
                                                callerPid, &err))) {
2968
        VIR_ERROR(_("Failed to lookup policy kit caller: %s"), err.message);
2969
        dbus_error_free(&err);
2970
        goto authfail;
2971
    }
2972

2973
    if (!(pkaction = polkit_action_new())) {
2974
        char ebuf[1024];
2975
        VIR_ERROR(_("Failed to create polkit action %s"),
2976
                  virStrerror(errno, ebuf, sizeof(ebuf)));
2977
        polkit_caller_unref(pkcaller);
2978
        goto authfail;
2979 2980 2981 2982 2983
    }
    polkit_action_set_action_id(pkaction, action);

    if (!(pkcontext = polkit_context_new()) ||
        !polkit_context_init(pkcontext, &pkerr)) {
2984
        char ebuf[1024];
2985
        VIR_ERROR(_("Failed to create polkit context %s"),
2986
                  (pkerr ? polkit_error_get_error_message(pkerr)
2987
                   : virStrerror(errno, ebuf, sizeof(ebuf))));
2988 2989 2990 2991 2992
        if (pkerr)
            polkit_error_free(pkerr);
        polkit_caller_unref(pkcaller);
        polkit_action_unref(pkaction);
        dbus_error_free(&err);
2993
        goto authfail;
2994
    }
2995

2996
# if HAVE_POLKIT_CONTEXT_IS_CALLER_AUTHORIZED
2997 2998 2999 3000 3001 3002
    pkresult = polkit_context_is_caller_authorized(pkcontext,
                                                   pkaction,
                                                   pkcaller,
                                                   0,
                                                   &pkerr);
    if (pkerr && polkit_error_is_set(pkerr)) {
3003 3004 3005
        VIR_ERROR(_("Policy kit failed to check authorization %d %s"),
                  polkit_error_get_error_code(pkerr),
                  polkit_error_get_error_message(pkerr));
3006
        goto authfail;
3007
    }
3008
# else
3009 3010 3011
    pkresult = polkit_context_can_caller_do_action(pkcontext,
                                                   pkaction,
                                                   pkcaller);
3012
# endif
3013 3014 3015 3016
    polkit_context_unref(pkcontext);
    polkit_caller_unref(pkcaller);
    polkit_action_unref(pkaction);
    if (pkresult != POLKIT_RESULT_YES) {
3017 3018
        VIR_ERROR(_("Policy kit denied action %s from pid %lld, uid %d, result: %s"),
                  action, (long long) callerPid, callerUid,
3019
                  polkit_result_to_string_representation(pkresult));
3020
        goto authdeny;
3021
    }
3022 3023 3024
    PROBE(RPC_SERVER_CLIENT_AUTH_ALLOW,
          "client=%p auth=%d identity=%s",
          client, REMOTE_AUTH_POLKIT, ident);
3025 3026
    VIR_INFO("Policy allowed action %s from pid %lld, uid %d, result %s",
             action, (long long) callerPid, callerUid,
3027 3028
             polkit_result_to_string_representation(pkresult));
    ret->complete = 1;
3029

3030
    virNetServerClientSetAuth(client, 0);
J
Jim Fehlig 已提交
3031
    virMutexUnlock(&priv->lock);
3032
    VIR_FREE(ident);
3033
    return 0;
3034

3035
error:
3036
    VIR_FREE(ident);
3037
    virResetLastError();
3038 3039
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
3040
    virNetMessageSaveError(rerr);
J
Jim Fehlig 已提交
3041
    virMutexUnlock(&priv->lock);
3042 3043
    return -1;

3044
authfail:
3045 3046 3047
    PROBE(RPC_SERVER_CLIENT_AUTH_FAIL,
          "client=%p auth=%d",
          client, REMOTE_AUTH_POLKIT);
3048
    goto error;
3049

3050
authdeny:
3051 3052 3053
    PROBE(RPC_SERVER_CLIENT_AUTH_DENY,
          "client=%p auth=%d identity=%s",
          client, REMOTE_AUTH_POLKIT, ident);
3054
    goto error;
3055 3056
}

3057
#else /* !WITH_POLKIT0 & !HAVE_POLKIT1*/
3058 3059

static int
3060
remoteDispatchAuthPolkit(virNetServerPtr server ATTRIBUTE_UNUSED,
J
Jiri Denemark 已提交
3061
                         virNetServerClientPtr client ATTRIBUTE_UNUSED,
3062
                         virNetMessagePtr msg ATTRIBUTE_UNUSED,
3063
                         virNetMessageErrorPtr rerr,
3064
                         remote_auth_polkit_ret *ret ATTRIBUTE_UNUSED)
3065
{
3066
    VIR_ERROR(_("client tried unsupported PolicyKit init request"));
3067 3068
    virReportError(VIR_ERR_AUTH_FAILED, "%s",
                   _("authentication failed"));
3069
    virNetMessageSaveError(rerr);
3070 3071
    return -1;
}
3072
#endif /* WITH_POLKIT1 */
3073 3074


3075 3076 3077
/***************************************************************
 *     NODE INFO APIS
 **************************************************************/
3078

3079
static int
3080 3081
remoteDispatchNodeDeviceGetParent(virNetServerPtr server ATTRIBUTE_UNUSED,
                                  virNetServerClientPtr client ATTRIBUTE_UNUSED,
3082
                                  virNetMessagePtr msg ATTRIBUTE_UNUSED,
3083
                                  virNetMessageErrorPtr rerr,
3084 3085
                                  remote_node_device_get_parent_args *args,
                                  remote_node_device_get_parent_ret *ret)
3086
{
3087 3088
    virNodeDevicePtr dev = NULL;
    const char *parent = NULL;
3089
    int rv = -1;
3090 3091
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3092

3093
    if (!priv->conn) {
3094
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3095
        goto cleanup;
3096 3097
    }

3098
    if (!(dev = virNodeDeviceLookupByName(priv->conn, args->name)))
3099 3100
        goto cleanup;

3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119
    parent = virNodeDeviceGetParent(dev);

    if (parent == NULL) {
        ret->parent = NULL;
    } else {
        /* remoteDispatchClientRequest will free this. */
        char **parent_p;
        if (VIR_ALLOC(parent_p) < 0) {
            virReportOOMError();
            goto cleanup;
        }
        if (!(*parent_p = strdup(parent))) {
            VIR_FREE(parent_p);
            virReportOOMError();
            goto cleanup;
        }
        ret->parent = parent_p;
    }

3120 3121 3122 3123
    rv = 0;

cleanup:
    if (rv < 0)
3124
        virNetMessageSaveError(rerr);
3125 3126
    if (dev)
        virNodeDeviceFree(dev);
3127
    return rv;
3128 3129
}

3130 3131 3132 3133 3134

/***************************
 * Register / deregister events
 ***************************/
static int
3135 3136 3137 3138 3139
remoteDispatchConnectDomainEventRegister(virNetServerPtr server ATTRIBUTE_UNUSED,
                                         virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                         virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                         virNetMessageErrorPtr rerr ATTRIBUTE_UNUSED,
                                         remote_connect_domain_event_register_ret *ret ATTRIBUTE_UNUSED)
O
Osier Yang 已提交
3140
{
3141
    int callbackID;
3142
    int rv = -1;
3143 3144
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
O
Osier Yang 已提交
3145

3146
    if (!priv->conn) {
3147
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3148
        goto cleanup;
3149 3150
    }

3151 3152 3153
    virMutexLock(&priv->lock);

    if (priv->domainEventCallbackID[VIR_DOMAIN_EVENT_ID_LIFECYCLE] != -1) {
3154
        virReportError(VIR_ERR_INTERNAL_ERROR, _("domain event %d already registered"), VIR_DOMAIN_EVENT_ID_LIFECYCLE);
3155
        goto cleanup;
3156
    }
O
Osier Yang 已提交
3157

3158
    if ((callbackID = virConnectDomainEventRegisterAny(priv->conn,
3159 3160 3161 3162
                                                       NULL,
                                                       VIR_DOMAIN_EVENT_ID_LIFECYCLE,
                                                       VIR_DOMAIN_EVENT_CALLBACK(remoteRelayDomainEventLifecycle),
                                                       client, NULL)) < 0)
3163
        goto cleanup;
O
Osier Yang 已提交
3164

3165
    priv->domainEventCallbackID[VIR_DOMAIN_EVENT_ID_LIFECYCLE] = callbackID;
3166

3167 3168 3169 3170
    rv = 0;

cleanup:
    if (rv < 0)
3171 3172
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
3173
    return rv;
O
Osier Yang 已提交
3174 3175
}

3176
static int
3177 3178 3179 3180 3181
remoteDispatchConnectDomainEventDeregister(virNetServerPtr server ATTRIBUTE_UNUSED,
                                           virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                           virNetMessageErrorPtr rerr ATTRIBUTE_UNUSED,
                                           remote_connect_domain_event_deregister_ret *ret ATTRIBUTE_UNUSED)
3182
{
3183
    int rv = -1;
3184 3185
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3186

3187
    if (!priv->conn) {
3188
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3189
        goto cleanup;
3190 3191
    }

3192 3193 3194
    virMutexLock(&priv->lock);

    if (priv->domainEventCallbackID[VIR_DOMAIN_EVENT_ID_LIFECYCLE] < 0) {
3195
        virReportError(VIR_ERR_INTERNAL_ERROR, _("domain event %d not registered"), VIR_DOMAIN_EVENT_ID_LIFECYCLE);
3196
        goto cleanup;
3197
    }
3198

3199 3200
    if (virConnectDomainEventDeregisterAny(priv->conn,
                                           priv->domainEventCallbackID[VIR_DOMAIN_EVENT_ID_LIFECYCLE]) < 0)
3201
        goto cleanup;
3202

3203 3204
    priv->domainEventCallbackID[VIR_DOMAIN_EVENT_ID_LIFECYCLE] = -1;

3205 3206 3207
    rv = 0;

cleanup:
3208
    if (rv < 0)
3209 3210
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
3211 3212 3213 3214
    return rv;
}

static void
3215 3216
remoteDispatchDomainEventSend(virNetServerClientPtr client,
                              virNetServerProgramPtr program,
3217 3218 3219 3220
                              int procnr,
                              xdrproc_t proc,
                              void *data)
{
3221
    virNetMessagePtr msg;
3222

3223
    if (!(msg = virNetMessageNew(false)))
3224
        goto cleanup;
3225

3226 3227 3228 3229 3230 3231
    msg->header.prog = virNetServerProgramGetID(program);
    msg->header.vers = virNetServerProgramGetVersion(program);
    msg->header.proc = procnr;
    msg->header.type = VIR_NET_MESSAGE;
    msg->header.serial = 1;
    msg->header.status = VIR_NET_OK;
3232

3233
    if (virNetMessageEncodeHeader(msg) < 0)
3234 3235
        goto cleanup;

3236 3237
    if (virNetMessageEncodePayload(msg, proc, data) < 0)
        goto cleanup;
3238

3239 3240
    VIR_DEBUG("Queue event %d %zu", procnr, msg->bufferLength);
    virNetServerClientSendMessage(client, msg);
3241

3242
    xdr_free(proc, data);
3243 3244 3245
    return;

cleanup:
3246
    virNetMessageFree(msg);
3247
    xdr_free(proc, data);
3248 3249
}

3250
static int
3251 3252
remoteDispatchSecretGetValue(virNetServerPtr server ATTRIBUTE_UNUSED,
                             virNetServerClientPtr client ATTRIBUTE_UNUSED,
3253
                             virNetMessagePtr msg ATTRIBUTE_UNUSED,
3254
                             virNetMessageErrorPtr rerr,
3255 3256
                             remote_secret_get_value_args *args,
                             remote_secret_get_value_ret *ret)
3257
{
3258 3259 3260
    virSecretPtr secret = NULL;
    size_t value_size;
    unsigned char *value;
3261
    int rv = -1;
3262 3263
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3264

3265
    if (!priv->conn) {
3266
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3267
        goto cleanup;
3268 3269
    }

3270
    if (!(secret = get_nonnull_secret(priv->conn, args->secret)))
3271
        goto cleanup;
3272

3273
    if (!(value = virSecretGetValue(secret, &value_size, args->flags)))
3274
        goto cleanup;
3275

3276 3277 3278
    ret->value.value_len = value_size;
    ret->value.value_val = (char *)value;

3279 3280 3281 3282
    rv = 0;

cleanup:
    if (rv < 0)
3283
        virNetMessageSaveError(rerr);
3284 3285
    if (secret)
        virSecretFree(secret);
3286
    return rv;
3287 3288
}

3289
static int
3290 3291
remoteDispatchDomainGetState(virNetServerPtr server ATTRIBUTE_UNUSED,
                             virNetServerClientPtr client ATTRIBUTE_UNUSED,
3292
                             virNetMessagePtr msg ATTRIBUTE_UNUSED,
3293
                             virNetMessageErrorPtr rerr,
3294 3295 3296 3297 3298
                             remote_domain_get_state_args *args,
                             remote_domain_get_state_ret *ret)
{
    virDomainPtr dom = NULL;
    int rv = -1;
3299 3300
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3301

3302
    if (!priv->conn) {
3303
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3304 3305 3306
        goto cleanup;
    }

3307
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
3308 3309 3310 3311 3312 3313 3314 3315 3316
        goto cleanup;

    if (virDomainGetState(dom, &ret->state, &ret->reason, args->flags) < 0)
        goto cleanup;

    rv = 0;

cleanup:
    if (rv < 0)
3317
        virNetMessageSaveError(rerr);
3318 3319 3320 3321 3322
    if (dom)
        virDomainFree(dom);
    return rv;
}

3323
static int
3324 3325 3326 3327 3328
remoteDispatchConnectDomainEventRegisterAny(virNetServerPtr server ATTRIBUTE_UNUSED,
                                            virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                            virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                            virNetMessageErrorPtr rerr ATTRIBUTE_UNUSED,
                                            remote_connect_domain_event_register_any_args *args)
3329 3330
{
    int callbackID;
3331
    int rv = -1;
3332 3333
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3334

3335
    if (!priv->conn) {
3336
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3337
        goto cleanup;
3338 3339
    }

3340 3341
    virMutexLock(&priv->lock);

3342 3343
    if (args->eventID >= VIR_DOMAIN_EVENT_ID_LAST ||
        args->eventID < 0) {
3344
        virReportError(VIR_ERR_INTERNAL_ERROR, _("unsupported event ID %d"), args->eventID);
3345
        goto cleanup;
3346 3347
    }

3348
    if (priv->domainEventCallbackID[args->eventID] != -1)  {
3349
        virReportError(VIR_ERR_INTERNAL_ERROR, _("domain event %d already registered"), args->eventID);
3350
        goto cleanup;
3351 3352
    }

3353
    if ((callbackID = virConnectDomainEventRegisterAny(priv->conn,
3354 3355 3356
                                                       NULL,
                                                       args->eventID,
                                                       domainEventCallbacks[args->eventID],
3357
                                                       client, NULL)) < 0)
3358
        goto cleanup;
3359

3360
    priv->domainEventCallbackID[args->eventID] = callbackID;
3361

3362 3363 3364 3365
    rv = 0;

cleanup:
    if (rv < 0)
3366 3367
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
3368
    return rv;
3369 3370 3371 3372
}


static int
3373 3374 3375 3376 3377
remoteDispatchConnectDomainEventDeregisterAny(virNetServerPtr server ATTRIBUTE_UNUSED,
                                              virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                              virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                              virNetMessageErrorPtr rerr ATTRIBUTE_UNUSED,
                                              remote_connect_domain_event_deregister_any_args *args)
3378 3379
{
    int callbackID = -1;
3380
    int rv = -1;
3381 3382
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3383

3384
    if (!priv->conn) {
3385
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3386
        goto cleanup;
3387 3388
    }

3389 3390
    virMutexLock(&priv->lock);

3391 3392
    if (args->eventID >= VIR_DOMAIN_EVENT_ID_LAST ||
        args->eventID < 0) {
3393
        virReportError(VIR_ERR_INTERNAL_ERROR, _("unsupported event ID %d"), args->eventID);
3394
        goto cleanup;
3395 3396
    }

3397 3398
    callbackID = priv->domainEventCallbackID[args->eventID];
    if (callbackID < 0) {
3399
        virReportError(VIR_ERR_INTERNAL_ERROR, _("domain event %d not registered"), args->eventID);
3400
        goto cleanup;
3401 3402
    }

3403
    if (virConnectDomainEventDeregisterAny(priv->conn, callbackID) < 0)
3404
        goto cleanup;
3405

3406 3407
    priv->domainEventCallbackID[args->eventID] = -1;

3408 3409 3410 3411
    rv = 0;

cleanup:
    if (rv < 0)
3412 3413
        virNetMessageSaveError(rerr);
    virMutexUnlock(&priv->lock);
3414
    return rv;
3415 3416
}

C
Chris Lalancette 已提交
3417
static int
3418 3419 3420 3421 3422 3423
qemuDispatchDomainMonitorCommand(virNetServerPtr server ATTRIBUTE_UNUSED,
                                 virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                 virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                 virNetMessageErrorPtr rerr,
                                 qemu_domain_monitor_command_args *args,
                                 qemu_domain_monitor_command_ret *ret)
C
Chris Lalancette 已提交
3424
{
3425
    virDomainPtr dom = NULL;
3426
    int rv = -1;
3427 3428
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
C
Chris Lalancette 已提交
3429

3430
    if (!priv->conn) {
3431
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3432
        goto cleanup;
3433 3434
    }

3435
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
3436
        goto cleanup;
C
Chris Lalancette 已提交
3437

3438
    if (virDomainQemuMonitorCommand(dom, args->cmd, &ret->result,
3439
                                    args->flags) < 0)
3440
        goto cleanup;
C
Chris Lalancette 已提交
3441

3442
    rv = 0;
C
Chris Lalancette 已提交
3443

3444 3445
cleanup:
    if (rv < 0)
3446
        virNetMessageSaveError(rerr);
3447 3448
    if (dom)
        virDomainFree(dom);
3449
    return rv;
C
Chris Lalancette 已提交
3450 3451
}

3452

3453
static int
3454 3455
remoteDispatchDomainMigrateBegin3(virNetServerPtr server ATTRIBUTE_UNUSED,
                                  virNetServerClientPtr client ATTRIBUTE_UNUSED,
3456
                                  virNetMessagePtr msg ATTRIBUTE_UNUSED,
3457
                                  virNetMessageErrorPtr rerr,
3458 3459 3460 3461 3462 3463
                                  remote_domain_migrate_begin3_args *args,
                                  remote_domain_migrate_begin3_ret *ret)
{
    char *xml = NULL;
    virDomainPtr dom = NULL;
    char *dname;
3464
    char *xmlin;
3465 3466 3467
    char *cookieout = NULL;
    int cookieoutlen = 0;
    int rv = -1;
3468 3469
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3470

3471
    if (!priv->conn) {
3472
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3473 3474 3475
        goto cleanup;
    }

3476
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
3477 3478
        goto cleanup;

3479
    xmlin = args->xmlin == NULL ? NULL : *args->xmlin;
3480 3481
    dname = args->dname == NULL ? NULL : *args->dname;

3482
    if (!(xml = virDomainMigrateBegin3(dom, xmlin,
3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497
                                       &cookieout, &cookieoutlen,
                                       args->flags, dname, args->resource)))
        goto cleanup;

    /* remoteDispatchClientRequest will free cookie and
     * the xml string if there is one.
     */
    ret->cookie_out.cookie_out_len = cookieoutlen;
    ret->cookie_out.cookie_out_val = cookieout;
    ret->xml = xml;

    rv = 0;

cleanup:
    if (rv < 0)
3498
        virNetMessageSaveError(rerr);
3499 3500 3501 3502 3503 3504 3505
    if (dom)
        virDomainFree(dom);
    return rv;
}


static int
3506 3507
remoteDispatchDomainMigratePrepare3(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client ATTRIBUTE_UNUSED,
3508
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
3509
                                    virNetMessageErrorPtr rerr,
3510 3511 3512 3513 3514 3515 3516 3517 3518
                                    remote_domain_migrate_prepare3_args *args,
                                    remote_domain_migrate_prepare3_ret *ret)
{
    char *cookieout = NULL;
    int cookieoutlen = 0;
    char *uri_in;
    char **uri_out;
    char *dname;
    int rv = -1;
3519 3520
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3521

3522
    if (!priv->conn) {
3523
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535
        goto cleanup;
    }

    uri_in = args->uri_in == NULL ? NULL : *args->uri_in;
    dname = args->dname == NULL ? NULL : *args->dname;

    /* Wacky world of XDR ... */
    if (VIR_ALLOC(uri_out) < 0) {
        virReportOOMError();
        goto cleanup;
    }

3536
    if (virDomainMigratePrepare3(priv->conn,
3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555
                                 args->cookie_in.cookie_in_val,
                                 args->cookie_in.cookie_in_len,
                                 &cookieout, &cookieoutlen,
                                 uri_in, uri_out,
                                 args->flags, dname, args->resource,
                                 args->dom_xml) < 0)
        goto cleanup;

    /* remoteDispatchClientRequest will free cookie, uri_out and
     * the string if there is one.
     */
    ret->cookie_out.cookie_out_len = cookieoutlen;
    ret->cookie_out.cookie_out_val = cookieout;
    ret->uri_out = *uri_out == NULL ? NULL : uri_out;

    rv = 0;

cleanup:
    if (rv < 0) {
3556
        virNetMessageSaveError(rerr);
3557 3558 3559 3560 3561
        VIR_FREE(uri_out);
    }
    return rv;
}

3562

3563
static int
3564 3565
remoteDispatchDomainMigratePerform3(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client ATTRIBUTE_UNUSED,
3566
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
3567
                                    virNetMessageErrorPtr rerr,
3568 3569 3570 3571
                                    remote_domain_migrate_perform3_args *args,
                                    remote_domain_migrate_perform3_ret *ret)
{
    virDomainPtr dom = NULL;
3572
    char *xmlin;
3573
    char *dname;
3574 3575
    char *uri;
    char *dconnuri;
3576 3577 3578
    char *cookieout = NULL;
    int cookieoutlen = 0;
    int rv = -1;
3579 3580
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3581

3582
    if (!priv->conn) {
3583
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3584 3585 3586
        goto cleanup;
    }

3587
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
3588 3589
        goto cleanup;

3590
    xmlin = args->xmlin == NULL ? NULL : *args->xmlin;
3591
    dname = args->dname == NULL ? NULL : *args->dname;
3592 3593
    uri = args->uri == NULL ? NULL : *args->uri;
    dconnuri = args->dconnuri == NULL ? NULL : *args->dconnuri;
3594

3595
    if (virDomainMigratePerform3(dom, xmlin,
3596 3597 3598
                                 args->cookie_in.cookie_in_val,
                                 args->cookie_in.cookie_in_len,
                                 &cookieout, &cookieoutlen,
3599
                                 dconnuri, uri,
3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611
                                 args->flags, dname, args->resource) < 0)
        goto cleanup;

    /* remoteDispatchClientRequest will free cookie
     */
    ret->cookie_out.cookie_out_len = cookieoutlen;
    ret->cookie_out.cookie_out_val = cookieout;

    rv = 0;

cleanup:
    if (rv < 0)
3612
        virNetMessageSaveError(rerr);
3613 3614 3615 3616 3617 3618 3619
    if (dom)
        virDomainFree(dom);
    return rv;
}


static int
3620 3621
remoteDispatchDomainMigrateFinish3(virNetServerPtr server ATTRIBUTE_UNUSED,
                                   virNetServerClientPtr client ATTRIBUTE_UNUSED,
3622
                                   virNetMessagePtr msg ATTRIBUTE_UNUSED,
3623
                                   virNetMessageErrorPtr rerr,
3624 3625 3626 3627 3628 3629
                                   remote_domain_migrate_finish3_args *args,
                                   remote_domain_migrate_finish3_ret *ret)
{
    virDomainPtr dom = NULL;
    char *cookieout = NULL;
    int cookieoutlen = 0;
3630 3631
    char *uri;
    char *dconnuri;
3632
    int rv = -1;
3633 3634
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3635

3636
    if (!priv->conn) {
3637
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3638 3639 3640
        goto cleanup;
    }

3641 3642 3643
    uri = args->uri == NULL ? NULL : *args->uri;
    dconnuri = args->dconnuri == NULL ? NULL : *args->dconnuri;

3644
    if (!(dom = virDomainMigrateFinish3(priv->conn, args->dname,
3645 3646 3647 3648 3649 3650
                                        args->cookie_in.cookie_in_val,
                                        args->cookie_in.cookie_in_len,
                                        &cookieout, &cookieoutlen,
                                        dconnuri, uri,
                                        args->flags,
                                        args->cancelled)))
3651 3652
        goto cleanup;

3653
    make_nonnull_domain(&ret->dom, dom);
3654 3655 3656 3657 3658 3659 3660 3661 3662 3663

    /* remoteDispatchClientRequest will free cookie
     */
    ret->cookie_out.cookie_out_len = cookieoutlen;
    ret->cookie_out.cookie_out_val = cookieout;

    rv = 0;

cleanup:
    if (rv < 0) {
3664
        virNetMessageSaveError(rerr);
3665 3666 3667 3668 3669 3670 3671 3672 3673
        VIR_FREE(cookieout);
    }
    if (dom)
        virDomainFree(dom);
    return rv;
}


static int
3674 3675
remoteDispatchDomainMigrateConfirm3(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client ATTRIBUTE_UNUSED,
3676
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
3677 3678
                                    virNetMessageErrorPtr rerr,
                                    remote_domain_migrate_confirm3_args *args)
3679 3680 3681
{
    virDomainPtr dom = NULL;
    int rv = -1;
3682 3683
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
3684

3685
    if (!priv->conn) {
3686
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3687 3688 3689
        goto cleanup;
    }

3690
    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
3691 3692 3693 3694 3695 3696 3697 3698 3699 3700 3701 3702
        goto cleanup;

    if (virDomainMigrateConfirm3(dom,
                                 args->cookie_in.cookie_in_val,
                                 args->cookie_in.cookie_in_len,
                                 args->flags, args->cancelled) < 0)
        goto cleanup;

    rv = 0;

cleanup:
    if (rv < 0)
3703
        virNetMessageSaveError(rerr);
3704 3705 3706 3707 3708 3709
    if (dom)
        virDomainFree(dom);
    return rv;
}


3710 3711 3712 3713 3714 3715
static int remoteDispatchConnectSupportsFeature(virNetServerPtr server ATTRIBUTE_UNUSED,
                                                virNetServerClientPtr client,
                                                virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                                virNetMessageErrorPtr rerr,
                                                remote_connect_supports_feature_args *args,
                                                remote_connect_supports_feature_ret *ret)
3716 3717 3718 3719 3720 3721
{
    int rv = -1;
    int supported;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

3722 3723 3724 3725 3726 3727 3728 3729 3730 3731
    /* This feature is checked before opening the connection, thus we must
     * check it first.
     */
    if (args->feature == VIR_DRV_FEATURE_PROGRAM_KEEPALIVE) {
        if (virNetServerClientStartKeepAlive(client) < 0)
            goto cleanup;
        supported = 1;
        goto done;
    }

3732
    if (!priv->conn) {
3733
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3734 3735 3736 3737 3738 3739 3740 3741 3742
        goto cleanup;
    }

    switch (args->feature) {
    case VIR_DRV_FEATURE_FD_PASSING:
        supported = 1;
        break;

    default:
3743
        if ((supported = virConnectSupportsFeature(priv->conn, args->feature)) < 0)
3744 3745 3746 3747
            goto cleanup;
        break;
    }

3748
done:
3749 3750 3751 3752 3753 3754 3755 3756 3757 3758
    ret->supported = supported;
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    return rv;
}


3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772
static int
remoteDispatchDomainOpenGraphics(virNetServerPtr server ATTRIBUTE_UNUSED,
                                 virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                 virNetMessagePtr msg,
                                 virNetMessageErrorPtr rerr,
                                 remote_domain_open_graphics_args *args)
{
    virDomainPtr dom = NULL;
    int rv = -1;
    int fd = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
3773
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if ((fd = virNetMessageDupFD(msg, 0)) < 0)
        goto cleanup;

    if (virDomainOpenGraphics(dom,
                              args->idx,
                              fd,
                              args->flags) < 0)
        goto cleanup;

    rv = 0;

cleanup:
    VIR_FORCE_CLOSE(fd);
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (dom)
        virDomainFree(dom);
    return rv;
}

3800 3801 3802 3803 3804 3805 3806 3807 3808 3809 3810
static int
remoteDispatchDomainGetInterfaceParameters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                           virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                           virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                           virNetMessageErrorPtr rerr,
                                           remote_domain_get_interface_parameters_args *args,
                                           remote_domain_get_interface_parameters_ret *ret)
{
    virDomainPtr dom = NULL;
    virTypedParameterPtr params = NULL;
    const char *device = args->device;
3811
    int nparams = 0;
3812 3813 3814 3815 3816 3817
    unsigned int flags;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
3818
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3819 3820 3821 3822 3823
        goto cleanup;
    }

    flags = args->flags;

3824
    if (args->nparams > REMOTE_DOMAIN_INTERFACE_PARAMETERS_MAX) {
3825
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
3826 3827
        goto cleanup;
    }
3828
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
3829 3830 3831
        virReportOOMError();
        goto cleanup;
    }
3832
    nparams = args->nparams;
3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846 3847 3848 3849 3850 3851 3852 3853 3854 3855 3856 3857 3858 3859

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if (virDomainGetInterfaceParameters(dom, device, params, &nparams, flags) < 0)
        goto cleanup;

    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    if (remoteSerializeTypedParameters(params, nparams,
                                       &ret->params.params_val,
                                       &ret->params.params_len,
                                       flags) < 0)
        goto cleanup;

success:
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
3860
    virTypedParamsFree(params, nparams);
3861 3862 3863 3864
    if (dom)
        virDomainFree(dom);
    return rv;
}
3865

3866 3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877 3878 3879 3880 3881
static int
remoteDispatchDomainGetCPUStats(virNetServerPtr server ATTRIBUTE_UNUSED,
                                virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                virNetMessagePtr hdr ATTRIBUTE_UNUSED,
                                virNetMessageErrorPtr rerr,
                                remote_domain_get_cpu_stats_args *args,
                                remote_domain_get_cpu_stats_ret *ret)
{
    virDomainPtr dom = NULL;
    struct daemonClientPrivate *priv;
    virTypedParameterPtr params = NULL;
    int rv = -1;
    int percpu_len = 0;

    priv = virNetServerClientGetPrivateData(client);
    if (!priv->conn) {
3882
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3883 3884 3885 3886
        goto cleanup;
    }

    if (args->nparams > REMOTE_NODE_CPU_STATS_MAX) {
3887
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
3888 3889 3890
        goto cleanup;
    }
    if (args->ncpus > REMOTE_DOMAIN_GET_CPU_STATS_NCPUS_MAX) {
3891
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("ncpus too large"));
3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909 3910 3911 3912 3913 3914 3915 3916 3917 3918 3919 3920 3921
        goto cleanup;
    }

    if (args->nparams > 0 &&
        VIR_ALLOC_N(params, args->ncpus * args->nparams) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    percpu_len = virDomainGetCPUStats(dom, params, args->nparams,
                                      args->start_cpu, args->ncpus,
                                      args->flags);
    if (percpu_len < 0)
        goto cleanup;
    /* If nparams == 0, the function returns a single value */
    if (args->nparams == 0)
        goto success;

    if (remoteSerializeTypedParameters(params, args->nparams * args->ncpus,
                                       &ret->params.params_val,
                                       &ret->params.params_len,
                                       args->flags) < 0)
        goto cleanup;

success:
    rv = 0;
    ret->nparams = percpu_len;
3922 3923 3924 3925 3926 3927 3928 3929
    if (args->nparams && !(args->flags & VIR_TYPED_PARAM_STRING_OKAY)) {
        int i;

        for (i = 0; i < percpu_len; i++) {
            if (params[i].type == VIR_TYPED_PARAM_STRING)
                ret->nparams--;
        }
    }
3930 3931 3932 3933

cleanup:
    if (rv < 0)
         virNetMessageSaveError(rerr);
3934
    virTypedParamsFree(params, args->ncpus * args->nparams);
3935 3936 3937 3938 3939
    if (dom)
        virDomainFree(dom);
    return rv;
}

3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950
static int remoteDispatchDomainGetDiskErrors(
    virNetServerPtr server ATTRIBUTE_UNUSED,
    virNetServerClientPtr client,
    virNetMessagePtr msg ATTRIBUTE_UNUSED,
    virNetMessageErrorPtr rerr,
    remote_domain_get_disk_errors_args *args,
    remote_domain_get_disk_errors_ret *ret)
{
    int rv = -1;
    virDomainPtr dom = NULL;
    virDomainDiskErrorPtr errors = NULL;
3951
    int len = 0;
3952 3953 3954 3955
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
3956
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
3957 3958 3959 3960 3961 3962 3963
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if (args->maxerrors > REMOTE_DOMAIN_DISK_ERRORS_MAX) {
3964 3965
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s",
                       _("maxerrors too large"));
3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002
        goto cleanup;
    }

    if (args->maxerrors &&
        VIR_ALLOC_N(errors, args->maxerrors) < 0) {
        virReportOOMError();
        goto cleanup;
    }

    if ((len = virDomainGetDiskErrors(dom, errors,
                                      args->maxerrors,
                                      args->flags)) < 0)
        goto cleanup;

    ret->nerrors = len;
    if (errors &&
        remoteSerializeDomainDiskErrors(errors, len,
                                        &ret->errors.errors_val,
                                        &ret->errors.errors_len) < 0)
        goto cleanup;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (dom)
        virDomainFree(dom);
    if (errors) {
        int i;
        for (i = 0; i < len; i++)
            VIR_FREE(errors[i].disk);
    }
    VIR_FREE(errors);
    return rv;
}

4003 4004 4005 4006 4007 4008 4009 4010 4011 4012 4013 4014 4015 4016 4017 4018
static int
remoteDispatchDomainListAllSnapshots(virNetServerPtr server ATTRIBUTE_UNUSED,
                                     virNetServerClientPtr client,
                                     virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                     virNetMessageErrorPtr rerr,
                                     remote_domain_list_all_snapshots_args *args,
                                     remote_domain_list_all_snapshots_ret *ret)
{
    virDomainSnapshotPtr *snaps = NULL;
    int nsnaps = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);
    virDomainPtr dom = NULL;

    if (!priv->conn) {
4019
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045 4046 4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076 4077 4078 4079
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if ((nsnaps = virDomainListAllSnapshots(dom,
                                            args->need_results ? &snaps : NULL,
                                            args->flags)) < 0)
        goto cleanup;

    if (snaps && nsnaps) {
        if (VIR_ALLOC_N(ret->snapshots.snapshots_val, nsnaps) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->snapshots.snapshots_len = nsnaps;

        for (i = 0; i < nsnaps; i++)
            make_nonnull_domain_snapshot(ret->snapshots.snapshots_val + i,
                                         snaps[i]);
    } else {
        ret->snapshots.snapshots_len = 0;
        ret->snapshots.snapshots_val = NULL;
    }

    ret->ret = nsnaps;
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (dom)
        virDomainFree(dom);
    if (snaps) {
        for (i = 0; i < nsnaps; i++)
            virDomainSnapshotFree(snaps[i]);
        VIR_FREE(snaps);
    }
    return rv;
}

static int
remoteDispatchDomainSnapshotListAllChildren(virNetServerPtr server ATTRIBUTE_UNUSED,
                                            virNetServerClientPtr client,
                                            virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                            virNetMessageErrorPtr rerr,
                                            remote_domain_snapshot_list_all_children_args *args,
                                            remote_domain_snapshot_list_all_children_ret *ret)
{
    virDomainSnapshotPtr *snaps = NULL;
    int nsnaps = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);
    virDomainPtr dom = NULL;
    virDomainSnapshotPtr snapshot = NULL;

    if (!priv->conn) {
4080
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->snapshot.dom)))
        goto cleanup;

    if (!(snapshot = get_nonnull_domain_snapshot(dom, args->snapshot)))
        goto cleanup;

    if ((nsnaps = virDomainSnapshotListAllChildren(snapshot,
                                                   args->need_results ? &snaps : NULL,
                                                   args->flags)) < 0)
        goto cleanup;

    if (snaps && nsnaps) {
        if (VIR_ALLOC_N(ret->snapshots.snapshots_val, nsnaps) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->snapshots.snapshots_len = nsnaps;

        for (i = 0; i < nsnaps; i++)
            make_nonnull_domain_snapshot(ret->snapshots.snapshots_val + i,
                                         snaps[i]);
    } else {
        ret->snapshots.snapshots_len = 0;
        ret->snapshots.snapshots_val = NULL;
    }

    ret->ret = nsnaps;
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (snapshot)
        virDomainSnapshotFree(snapshot);
    if (dom)
        virDomainFree(dom);
    if (snaps) {
        for (i = 0; i < nsnaps; i++)
            virDomainSnapshotFree(snaps[i]);
        VIR_FREE(snaps);
    }
    return rv;
}

4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182
static int
remoteDispatchConnectListAllStoragePools(virNetServerPtr server ATTRIBUTE_UNUSED,
                                         virNetServerClientPtr client,
                                         virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                         virNetMessageErrorPtr rerr,
                                         remote_connect_list_all_storage_pools_args *args,
                                         remote_connect_list_all_storage_pools_ret *ret)
{
    virStoragePoolPtr *pools = NULL;
    int npools = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if ((npools = virConnectListAllStoragePools(priv->conn,
                                                args->need_results ? &pools : NULL,
                                                args->flags)) < 0)
        goto cleanup;

    if (pools && npools) {
        if (VIR_ALLOC_N(ret->pools.pools_val, npools) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->pools.pools_len = npools;

        for (i = 0; i < npools; i++)
            make_nonnull_storage_pool(ret->pools.pools_val + i, pools[i]);
    } else {
        ret->pools.pools_len = 0;
        ret->pools.pools_val = NULL;
    }

    ret->ret = npools;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (pools) {
        for (i = 0; i < npools; i++)
            virStoragePoolFree(pools[i]);
        VIR_FREE(pools);
    }
    return rv;
}

4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199 4200 4201 4202 4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213 4214 4215 4216 4217 4218 4219 4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230 4231 4232 4233 4234 4235 4236 4237 4238 4239 4240
static int
remoteDispatchStoragePoolListAllVolumes(virNetServerPtr server ATTRIBUTE_UNUSED,
                                        virNetServerClientPtr client,
                                        virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                        virNetMessageErrorPtr rerr,
                                        remote_storage_pool_list_all_volumes_args *args,
                                        remote_storage_pool_list_all_volumes_ret *ret)
{
    virStorageVolPtr *vols = NULL;
    virStoragePoolPtr pool = NULL;
    int nvols = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if (!(pool = get_nonnull_storage_pool(priv->conn, args->pool)))
        goto cleanup;

    if ((nvols = virStoragePoolListAllVolumes(pool,
                                              args->need_results ? &vols : NULL,
                                              args->flags)) < 0)
        goto cleanup;

    if (vols && nvols) {
        if (VIR_ALLOC_N(ret->vols.vols_val, nvols) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->vols.vols_len = nvols;

        for (i = 0; i < nvols; i++)
            make_nonnull_storage_vol(ret->vols.vols_val + i, vols[i]);
    } else {
        ret->vols.vols_len = 0;
        ret->vols.vols_val = NULL;
    }

    ret->ret = nvols;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (vols) {
        for (i = 0; i < nvols; i++)
            virStorageVolFree(vols[i]);
        VIR_FREE(vols);
    }
    return rv;
}

4241 4242 4243 4244 4245 4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272 4273 4274 4275 4276 4277 4278 4279 4280 4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294
static int
remoteDispatchConnectListAllNetworks(virNetServerPtr server ATTRIBUTE_UNUSED,
                                     virNetServerClientPtr client,
                                     virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                     virNetMessageErrorPtr rerr,
                                     remote_connect_list_all_networks_args *args,
                                     remote_connect_list_all_networks_ret *ret)
{
    virNetworkPtr *nets = NULL;
    int nnets = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if ((nnets = virConnectListAllNetworks(priv->conn,
                                           args->need_results ? &nets : NULL,
                                           args->flags)) < 0)
        goto cleanup;

    if (nets && nnets) {
        if (VIR_ALLOC_N(ret->nets.nets_val, nnets) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->nets.nets_len = nnets;

        for (i = 0; i < nnets; i++)
            make_nonnull_network(ret->nets.nets_val + i, nets[i]);
    } else {
        ret->nets.nets_len = 0;
        ret->nets.nets_val = NULL;
    }

    ret->ret = nnets;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (nets) {
        for (i = 0; i < nnets; i++)
            virNetworkFree(nets[i]);
        VIR_FREE(nets);
    }
    return rv;
}

4295 4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348
static int
remoteDispatchConnectListAllInterfaces(virNetServerPtr server ATTRIBUTE_UNUSED,
                                       virNetServerClientPtr client,
                                       virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                       virNetMessageErrorPtr rerr,
                                       remote_connect_list_all_interfaces_args *args,
                                       remote_connect_list_all_interfaces_ret *ret)
{
    virInterfacePtr *ifaces = NULL;
    int nifaces = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if ((nifaces = virConnectListAllInterfaces(priv->conn,
                                               args->need_results ? &ifaces : NULL,
                                               args->flags)) < 0)
        goto cleanup;

    if (ifaces && nifaces) {
        if (VIR_ALLOC_N(ret->ifaces.ifaces_val, nifaces) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->ifaces.ifaces_len = nifaces;

        for (i = 0; i < nifaces; i++)
            make_nonnull_interface(ret->ifaces.ifaces_val + i, ifaces[i]);
    } else {
        ret->ifaces.ifaces_len = 0;
        ret->ifaces.ifaces_val = NULL;
    }

    ret->ret = nifaces;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (ifaces) {
        for (i = 0; i < nifaces; i++)
            virInterfaceFree(ifaces[i]);
        VIR_FREE(ifaces);
    }
    return rv;
}

4349 4350 4351 4352 4353 4354 4355 4356 4357 4358 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372 4373 4374 4375 4376 4377 4378 4379 4380 4381 4382 4383 4384 4385 4386 4387 4388 4389 4390 4391 4392 4393 4394 4395 4396 4397 4398 4399 4400 4401
static int
remoteDispatchConnectListAllNodeDevices(virNetServerPtr server ATTRIBUTE_UNUSED,
                                        virNetServerClientPtr client,
                                        virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                        virNetMessageErrorPtr rerr,
                                        remote_connect_list_all_node_devices_args *args,
                                        remote_connect_list_all_node_devices_ret *ret)
{
    virNodeDevicePtr *devices = NULL;
    int ndevices = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if ((ndevices = virConnectListAllNodeDevices(priv->conn,
                                                 args->need_results ? &devices : NULL,
                                                 args->flags)) < 0)
        goto cleanup;

    if (devices && ndevices) {
        if (VIR_ALLOC_N(ret->devices.devices_val, ndevices) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->devices.devices_len = ndevices;

        for (i = 0; i < ndevices; i++)
            make_nonnull_node_device(ret->devices.devices_val + i, devices[i]);
    } else {
        ret->devices.devices_len = 0;
        ret->devices.devices_val = NULL;
    }

    ret->ret = ndevices;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (devices) {
        for (i = 0; i < ndevices; i++)
            virNodeDeviceFree(devices[i]);
        VIR_FREE(devices);
    }
    return rv;
}
4402

4403 4404 4405 4406 4407 4408 4409 4410 4411 4412 4413 4414 4415 4416 4417 4418 4419 4420 4421 4422 4423 4424 4425 4426 4427 4428 4429 4430 4431 4432 4433 4434 4435 4436 4437 4438 4439 4440 4441 4442 4443 4444 4445 4446 4447 4448 4449 4450 4451 4452 4453 4454 4455 4456
static int
remoteDispatchConnectListAllNWFilters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                      virNetServerClientPtr client,
                                      virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                      virNetMessageErrorPtr rerr,
                                      remote_connect_list_all_nwfilters_args *args,
                                      remote_connect_list_all_nwfilters_ret *ret)
{
    virNWFilterPtr *filters = NULL;
    int nfilters = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if ((nfilters = virConnectListAllNWFilters(priv->conn,
                                               args->need_results ? &filters : NULL,
                                               args->flags)) < 0)
        goto cleanup;

    if (filters && nfilters) {
        if (VIR_ALLOC_N(ret->filters.filters_val, nfilters) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->filters.filters_len = nfilters;

        for (i = 0; i < nfilters; i++)
            make_nonnull_nwfilter(ret->filters.filters_val + i, filters[i]);
    } else {
        ret->filters.filters_len = 0;
        ret->filters.filters_val = NULL;
    }

    ret->ret = nfilters;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (filters) {
        for (i = 0; i < nfilters; i++)
            virNWFilterFree(filters[i]);
        VIR_FREE(filters);
    }
    return rv;
}

4457 4458 4459 4460 4461 4462 4463 4464 4465 4466 4467 4468 4469 4470 4471 4472 4473 4474 4475 4476 4477 4478 4479 4480 4481 4482 4483 4484 4485 4486 4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503 4504 4505 4506 4507 4508 4509 4510
static int
remoteDispatchConnectListAllSecrets(virNetServerPtr server ATTRIBUTE_UNUSED,
                                    virNetServerClientPtr client,
                                    virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                    virNetMessageErrorPtr rerr,
                                    remote_connect_list_all_secrets_args *args,
                                    remote_connect_list_all_secrets_ret *ret)
{
    virSecretPtr *secrets = NULL;
    int nsecrets = 0;
    int i;
    int rv = -1;
    struct daemonClientPrivate *priv = virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if ((nsecrets = virConnectListAllSecrets(priv->conn,
                                             args->need_results ? &secrets : NULL,
                                             args->flags)) < 0)
        goto cleanup;

    if (secrets && nsecrets) {
        if (VIR_ALLOC_N(ret->secrets.secrets_val, nsecrets) < 0) {
            virReportOOMError();
            goto cleanup;
        }

        ret->secrets.secrets_len = nsecrets;

        for (i = 0; i < nsecrets; i++)
            make_nonnull_secret(ret->secrets.secrets_val + i, secrets[i]);
    } else {
        ret->secrets.secrets_len = 0;
        ret->secrets.secrets_val = NULL;
    }

    ret->ret = nsecrets;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    if (secrets) {
        for (i = 0; i < nsecrets; i++)
            virSecretFree(secrets[i]);
        VIR_FREE(secrets);
    }
    return rv;
}

4511 4512 4513 4514 4515 4516 4517 4518 4519
static int
remoteDispatchNodeGetMemoryParameters(virNetServerPtr server ATTRIBUTE_UNUSED,
                                      virNetServerClientPtr client ATTRIBUTE_UNUSED,
                                      virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                      virNetMessageErrorPtr rerr,
                                      remote_node_get_memory_parameters_args *args,
                                      remote_node_get_memory_parameters_ret *ret)
{
    virTypedParameterPtr params = NULL;
4520
    int nparams = 0;
4521 4522 4523 4524 4525 4526 4527 4528 4529 4530 4531 4532
    unsigned int flags;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    flags = args->flags;

4533
    if (args->nparams > REMOTE_NODE_MEMORY_PARAMETERS_MAX) {
4534 4535 4536
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("nparams too large"));
        goto cleanup;
    }
4537
    if (args->nparams && VIR_ALLOC_N(params, args->nparams) < 0) {
4538 4539 4540
        virReportOOMError();
        goto cleanup;
    }
4541
    nparams = args->nparams;
4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560 4561 4562 4563 4564 4565

    if (virNodeGetMemoryParameters(priv->conn, params, &nparams, flags) < 0)
        goto cleanup;

    /* In this case, we need to send back the number of parameters
     * supported
     */
    if (args->nparams == 0) {
        ret->nparams = nparams;
        goto success;
    }

    if (remoteSerializeTypedParameters(params, nparams,
                                       &ret->params.params_val,
                                       &ret->params.params_len,
                                       args->flags) < 0)
        goto cleanup;

success:
    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
4566
    virTypedParamsFree(params, nparams);
4567 4568 4569
    return rv;
}

4570 4571 4572 4573 4574 4575 4576 4577 4578
static int
remoteDispatchNodeGetCPUMap(virNetServerPtr server ATTRIBUTE_UNUSED,
                            virNetServerClientPtr client ATTRIBUTE_UNUSED,
                            virNetMessagePtr msg ATTRIBUTE_UNUSED,
                            virNetMessageErrorPtr rerr,
                            remote_node_get_cpu_map_args *args,
                            remote_node_get_cpu_map_ret *ret)
{
    unsigned char *cpumap = NULL;
4579
    unsigned int online = 0;
4580 4581 4582 4583 4584 4585 4586 4587 4588 4589 4590 4591 4592
    unsigned int flags;
    int cpunum;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    flags = args->flags;

4593 4594
    cpunum = virNodeGetCPUMap(priv->conn, args->need_map ? &cpumap : NULL,
                              args->need_online ? &online : NULL, flags);
4595 4596 4597 4598
    if (cpunum < 0)
        goto cleanup;

    /* 'serialize' return cpumap */
4599
    if (args->need_map) {
4600 4601 4602 4603 4604 4605 4606 4607 4608 4609 4610 4611 4612 4613 4614 4615 4616
        ret->cpumap.cpumap_len = VIR_CPU_MAPLEN(cpunum);
        ret->cpumap.cpumap_val = (char *) cpumap;
        cpumap = NULL;
    }

    ret->online = online;
    ret->ret = cpunum;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    VIR_FREE(cpumap);
    return rv;
}

4617 4618 4619 4620 4621 4622 4623 4624 4625 4626 4627 4628 4629 4630 4631 4632 4633 4634 4635 4636 4637 4638 4639 4640 4641 4642 4643 4644 4645 4646 4647 4648 4649 4650 4651 4652 4653 4654 4655 4656 4657 4658 4659 4660 4661 4662 4663 4664 4665 4666
static int
lxcDispatchDomainOpenNamespace(virNetServerPtr server ATTRIBUTE_UNUSED,
                               virNetServerClientPtr client ATTRIBUTE_UNUSED,
                               virNetMessagePtr msg ATTRIBUTE_UNUSED,
                               virNetMessageErrorPtr rerr,
                               lxc_domain_open_namespace_args *args)
{
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);
    int *fdlist = NULL;
    int ret;
    virDomainPtr dom = NULL;
    size_t i;

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    ret = virDomainLxcOpenNamespace(dom,
                                    &fdlist,
                                    args->flags);
    if (ret < 0)
        goto cleanup;

    /* We shouldn't have received any from the client,
     * but in case they're playing games with us, prevent
     * a resource leak
     */
    for (i = 0 ; i < msg->nfds ; i++)
        VIR_FORCE_CLOSE(msg->fds[i]);
    VIR_FREE(msg->fds);
    msg->nfds = 0;

    msg->fds = fdlist;
    msg->nfds = ret;

    rv = 1;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    virDomainFree(dom);
    return rv;
}

4667 4668 4669 4670 4671 4672 4673 4674 4675 4676 4677 4678 4679 4680 4681 4682 4683 4684 4685 4686 4687 4688 4689 4690 4691 4692 4693 4694 4695 4696 4697 4698 4699 4700 4701 4702 4703 4704 4705 4706 4707 4708 4709 4710
static int
remoteDispatchDomainGetJobStats(virNetServerPtr server ATTRIBUTE_UNUSED,
                                virNetServerClientPtr client,
                                virNetMessagePtr msg ATTRIBUTE_UNUSED,
                                virNetMessageErrorPtr rerr,
                                remote_domain_get_job_stats_args *args,
                                remote_domain_get_job_stats_ret *ret)
{
    virDomainPtr dom = NULL;
    virTypedParameterPtr params = NULL;
    int nparams = 0;
    int rv = -1;
    struct daemonClientPrivate *priv =
        virNetServerClientGetPrivateData(client);

    if (!priv->conn) {
        virReportError(VIR_ERR_INTERNAL_ERROR, "%s", _("connection not open"));
        goto cleanup;
    }

    if (!(dom = get_nonnull_domain(priv->conn, args->dom)))
        goto cleanup;

    if (virDomainGetJobStats(dom, &ret->type, &params,
                             &nparams, args->flags) < 0)
        goto cleanup;

    if (remoteSerializeTypedParameters(params, nparams,
                                       &ret->params.params_val,
                                       &ret->params.params_len,
                                       0) < 0)
        goto cleanup;

    rv = 0;

cleanup:
    if (rv < 0)
        virNetMessageSaveError(rerr);
    virTypedParamsFree(params, nparams);
    if (dom)
        virDomainFree(dom);
    return rv;
}

4711 4712 4713 4714 4715 4716 4717 4718 4719
/*----- Helpers. -----*/

/* get_nonnull_domain and get_nonnull_network turn an on-wire
 * (name, uuid) pair into virDomainPtr or virNetworkPtr object.
 * virDomainPtr or virNetworkPtr cannot be NULL.
 *
 * NB. If these return NULL then the caller must return an error.
 */
static virDomainPtr
4720
get_nonnull_domain(virConnectPtr conn, remote_nonnull_domain domain)
4721 4722
{
    virDomainPtr dom;
4723
    dom = virGetDomain(conn, domain.name, BAD_CAST domain.uuid);
4724 4725 4726 4727 4728 4729 4730 4731
    /* Should we believe the domain.id sent by the client?  Maybe
     * this should be a check rather than an assignment? XXX
     */
    if (dom) dom->id = domain.id;
    return dom;
}

static virNetworkPtr
4732
get_nonnull_network(virConnectPtr conn, remote_nonnull_network network)
4733
{
4734
    return virGetNetwork(conn, network.name, BAD_CAST network.uuid);
4735 4736
}

D
Daniel Veillard 已提交
4737
static virInterfacePtr
4738
get_nonnull_interface(virConnectPtr conn, remote_nonnull_interface iface)
D
Daniel Veillard 已提交
4739
{
4740
    return virGetInterface(conn, iface.name, iface.mac);
D
Daniel Veillard 已提交
4741 4742
}

4743
static virStoragePoolPtr
4744
get_nonnull_storage_pool(virConnectPtr conn, remote_nonnull_storage_pool pool)
4745
{
4746 4747
    return virGetStoragePool(conn, pool.name, BAD_CAST pool.uuid,
                             NULL, NULL);
4748 4749 4750
}

static virStorageVolPtr
4751
get_nonnull_storage_vol(virConnectPtr conn, remote_nonnull_storage_vol vol)
4752 4753
{
    virStorageVolPtr ret;
4754 4755
    ret = virGetStorageVol(conn, vol.pool, vol.name, vol.key,
                           NULL, NULL);
4756 4757 4758
    return ret;
}

4759
static virSecretPtr
4760
get_nonnull_secret(virConnectPtr conn, remote_nonnull_secret secret)
4761
{
4762
    return virGetSecret(conn, BAD_CAST secret.uuid, secret.usageType, secret.usageID);
4763 4764
}

4765
static virNWFilterPtr
4766
get_nonnull_nwfilter(virConnectPtr conn, remote_nonnull_nwfilter nwfilter)
4767
{
4768
    return virGetNWFilter(conn, nwfilter.name, BAD_CAST nwfilter.uuid);
4769 4770
}

C
Chris Lalancette 已提交
4771
static virDomainSnapshotPtr
4772
get_nonnull_domain_snapshot(virDomainPtr dom, remote_nonnull_domain_snapshot snapshot)
C
Chris Lalancette 已提交
4773
{
4774
    return virGetDomainSnapshot(dom, snapshot.name);
C
Chris Lalancette 已提交
4775 4776
}

4777 4778
/* Make remote_nonnull_domain and remote_nonnull_network. */
static void
4779
make_nonnull_domain(remote_nonnull_domain *dom_dst, virDomainPtr dom_src)
4780 4781
{
    dom_dst->id = dom_src->id;
4782 4783
    dom_dst->name = strdup(dom_src->name);
    memcpy(dom_dst->uuid, dom_src->uuid, VIR_UUID_BUFLEN);
4784 4785 4786
}

static void
4787
make_nonnull_network(remote_nonnull_network *net_dst, virNetworkPtr net_src)
4788
{
4789 4790
    net_dst->name = strdup(net_src->name);
    memcpy(net_dst->uuid, net_src->uuid, VIR_UUID_BUFLEN);
4791 4792
}

D
Daniel Veillard 已提交
4793
static void
4794 4795
make_nonnull_interface(remote_nonnull_interface *interface_dst,
                       virInterfacePtr interface_src)
D
Daniel Veillard 已提交
4796
{
4797 4798
    interface_dst->name = strdup(interface_src->name);
    interface_dst->mac = strdup(interface_src->mac);
D
Daniel Veillard 已提交
4799 4800
}

4801
static void
4802
make_nonnull_storage_pool(remote_nonnull_storage_pool *pool_dst, virStoragePoolPtr pool_src)
4803
{
4804 4805
    pool_dst->name = strdup(pool_src->name);
    memcpy(pool_dst->uuid, pool_src->uuid, VIR_UUID_BUFLEN);
4806 4807 4808
}

static void
4809
make_nonnull_storage_vol(remote_nonnull_storage_vol *vol_dst, virStorageVolPtr vol_src)
4810
{
4811 4812 4813
    vol_dst->pool = strdup(vol_src->pool);
    vol_dst->name = strdup(vol_src->name);
    vol_dst->key = strdup(vol_src->key);
4814
}
4815 4816

static void
4817
make_nonnull_node_device(remote_nonnull_node_device *dev_dst, virNodeDevicePtr dev_src)
4818 4819 4820
{
    dev_dst->name = strdup(dev_src->name);
}
4821 4822

static void
4823
make_nonnull_secret(remote_nonnull_secret *secret_dst, virSecretPtr secret_src)
4824
{
4825
    memcpy(secret_dst->uuid, secret_src->uuid, VIR_UUID_BUFLEN);
4826
    secret_dst->usageType = secret_src->usageType;
4827
    secret_dst->usageID = strdup(secret_src->usageID);
4828
}
4829 4830

static void
4831
make_nonnull_nwfilter(remote_nonnull_nwfilter *nwfilter_dst, virNWFilterPtr nwfilter_src)
4832
{
4833 4834
    nwfilter_dst->name = strdup(nwfilter_src->name);
    memcpy(nwfilter_dst->uuid, nwfilter_src->uuid, VIR_UUID_BUFLEN);
4835
}
C
Chris Lalancette 已提交
4836 4837

static void
4838
make_nonnull_domain_snapshot(remote_nonnull_domain_snapshot *snapshot_dst, virDomainSnapshotPtr snapshot_src)
C
Chris Lalancette 已提交
4839 4840
{
    snapshot_dst->name = strdup(snapshot_src->name);
4841
    make_nonnull_domain(&snapshot_dst->dom, snapshot_src->domain);
C
Chris Lalancette 已提交
4842
}
4843 4844 4845 4846 4847 4848 4849 4850 4851 4852 4853 4854 4855 4856 4857 4858 4859 4860 4861 4862 4863 4864 4865 4866 4867 4868 4869 4870 4871 4872 4873 4874 4875 4876

static int
remoteSerializeDomainDiskErrors(virDomainDiskErrorPtr errors,
                                int nerrors,
                                remote_domain_disk_error **ret_errors_val,
                                u_int *ret_errors_len)
{
    remote_domain_disk_error *val = NULL;
    int i = 0;

    if (VIR_ALLOC_N(val, nerrors) < 0)
        goto no_memory;

    for (i = 0; i < nerrors; i++) {
        if (!(val[i].disk = strdup(errors[i].disk)))
            goto no_memory;
        val[i].error = errors[i].error;
    }

    *ret_errors_len = nerrors;
    *ret_errors_val = val;

    return 0;

no_memory:
    if (val) {
        int j;
        for (j = 0; j < i; j++)
            VIR_FREE(val[j].disk);
        VIR_FREE(val);
    }
    virReportOOMError();
    return -1;
}