esx_driver.c 118.4 KB
Newer Older
1 2

/*
3
 * esx_driver.c: core driver functions for managing VMware ESX hosts
4
 *
E
Eric Blake 已提交
5
 * Copyright (C) 2010 Red Hat, Inc.
6
 * Copyright (C) 2009-2010 Matthias Bolte <matthias.bolte@googlemail.com>
7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
 * Copyright (C) 2009 Maximilian Wilhelm <max@rfc2324.org>
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307  USA
 *
 */

#include <config.h>

#include <netdb.h>

#include "internal.h"
#include "domain_conf.h"
31
#include "authhelper.h"
32 33 34 35 36
#include "util.h"
#include "memory.h"
#include "logging.h"
#include "uuid.h"
#include "esx_driver.h"
37 38 39 40 41
#include "esx_interface_driver.h"
#include "esx_network_driver.h"
#include "esx_storage_driver.h"
#include "esx_device_monitor.h"
#include "esx_secret_driver.h"
M
Matthias Bolte 已提交
42
#include "esx_nwfilter_driver.h"
43
#include "esx_private.h"
44 45 46 47 48 49 50 51 52 53 54
#include "esx_vi.h"
#include "esx_vi_methods.h"
#include "esx_util.h"
#include "esx_vmx.h"

#define VIR_FROM_THIS VIR_FROM_ESX

static int esxDomainGetMaxVcpus(virDomainPtr domain);



55
static esxVI_Boolean
56
esxSupportsLongMode(esxPrivate *priv)
57 58 59 60 61 62
{
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    esxVI_HostCpuIdInfo *hostCpuIdInfoList = NULL;
    esxVI_HostCpuIdInfo *hostCpuIdInfo = NULL;
63
    esxVI_ParsedHostCpuIdInfo parsedHostCpuIdInfo;
64 65 66 67 68 69
    char edxLongModeBit = '?';

    if (priv->supportsLongMode != esxVI_Boolean_Undefined) {
        return priv->supportsLongMode;
    }

70
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
71
        return esxVI_Boolean_Undefined;
72 73
    }

74
    if (esxVI_String_AppendValueToList(&propertyNameList,
75
                                       "hardware.cpuFeature") < 0 ||
76
        esxVI_LookupObjectContentByType(priv->host, priv->host->hostFolder,
77 78
                                        "HostSystem", propertyNameList,
                                        esxVI_Boolean_True, &hostSystem) < 0) {
M
Matthias Bolte 已提交
79
        goto cleanup;
80 81 82
    }

    if (hostSystem == NULL) {
83 84
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve the HostSystem object"));
M
Matthias Bolte 已提交
85
        goto cleanup;
86 87 88 89 90 91
    }

    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "hardware.cpuFeature")) {
            if (esxVI_HostCpuIdInfo_CastListFromAnyType
92
                  (dynamicProperty->val, &hostCpuIdInfoList) < 0) {
M
Matthias Bolte 已提交
93
                goto cleanup;
94 95 96 97 98
            }

            for (hostCpuIdInfo = hostCpuIdInfoList; hostCpuIdInfo != NULL;
                 hostCpuIdInfo = hostCpuIdInfo->_next) {
                if (hostCpuIdInfo->level->value == -2147483647) { /* 0x80000001 */
99 100
                    if (esxVI_ParseHostCpuIdInfo(&parsedHostCpuIdInfo,
                                                 hostCpuIdInfo) < 0) {
M
Matthias Bolte 已提交
101
                        goto cleanup;
102 103
                    }

104
                    edxLongModeBit = parsedHostCpuIdInfo.edx[29];
105 106 107 108 109 110

                    if (edxLongModeBit == '1') {
                        priv->supportsLongMode = esxVI_Boolean_True;
                    } else if (edxLongModeBit == '0') {
                        priv->supportsLongMode = esxVI_Boolean_False;
                    } else {
111
                        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
112 113 114 115
                                  _("Bit 29 (Long Mode) of HostSystem property "
                                    "'hardware.cpuFeature[].edx' with value '%s' "
                                    "has unexpected value '%c', expecting '0' "
                                    "or '1'"), hostCpuIdInfo->edx, edxLongModeBit);
M
Matthias Bolte 已提交
116
                        goto cleanup;
117 118 119 120 121 122 123 124 125 126 127 128 129
                    }

                    break;
                }
            }

            break;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

  cleanup:
M
Matthias Bolte 已提交
130 131 132 133
    /*
     * If we goto cleanup in case of an error then priv->supportsLongMode
     * is still esxVI_Boolean_Undefined, therefore we don't need to set it.
     */
134 135 136 137 138 139 140 141 142
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);
    esxVI_HostCpuIdInfo_Free(&hostCpuIdInfoList);

    return priv->supportsLongMode;
}



143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
static int
esxLookupHostSystemBiosUuid(esxPrivate *priv, unsigned char *uuid)
{
    int result = -1;
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;

    if (esxVI_EnsureSession(priv->host) < 0) {
        return -1;
    }

    if (esxVI_String_AppendValueToList(&propertyNameList,
                                       "hardware.systemInfo.uuid") < 0 ||
        esxVI_LookupObjectContentByType(priv->host, priv->host->hostFolder,
                                        "HostSystem", propertyNameList,
                                        esxVI_Boolean_True, &hostSystem) < 0) {
        goto cleanup;
    }

    if (hostSystem == NULL) {
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve the HostSystem object"));
        goto cleanup;
    }

    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "hardware.systemInfo.uuid")) {
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
                                         esxVI_Type_String) < 0) {
                goto cleanup;
            }

            if (strlen(dynamicProperty->val->string) > 0) {
                if (virUUIDParse(dynamicProperty->val->string, uuid) < 0) {
                    ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
                              _("Could not parse UUID from string '%s'"),
                              dynamicProperty->val->string);
                    goto cleanup;
                }
            } else {
                /* HostSystem has an empty UUID */
                memset(uuid, 0, VIR_UUID_BUFLEN);
            }

            break;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

    result = 0;

  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);

    return result;
}



206
static virCapsPtr
207
esxCapsInit(esxPrivate *priv)
208
{
209
    esxVI_Boolean supportsLongMode = esxSupportsLongMode(priv);
210 211 212
    virCapsPtr caps = NULL;
    virCapsGuestPtr guest = NULL;

213 214 215 216 217 218 219 220 221
    if (supportsLongMode == esxVI_Boolean_Undefined) {
        return NULL;
    }

    if (supportsLongMode == esxVI_Boolean_True) {
        caps = virCapabilitiesNew("x86_64", 1, 1);
    } else {
        caps = virCapabilitiesNew("i686", 1, 1);
    }
222 223

    if (caps == NULL) {
224
        virReportOOMError();
225 226 227
        return NULL;
    }

228
    virCapabilitiesSetMacPrefix(caps, (unsigned char[]){ 0x00, 0x0c, 0x29 });
229 230
    virCapabilitiesAddHostMigrateTransport(caps, "esx");

231 232 233 234
    if (esxLookupHostSystemBiosUuid(priv, caps->host.host_uuid) < 0) {
        goto failure;
    }

235 236 237
    /* i686 */
    guest = virCapabilitiesAddGuest(caps, "hvm", "i686", 32, NULL, NULL, 0,
                                    NULL);
238 239 240 241 242 243 244 245 246 247 248 249 250 251

    if (guest == NULL) {
        goto failure;
    }

    /*
     * FIXME: Maybe distinguish betwen ESX and GSX here, see
     * esxVMX_ParseConfig() and VIR_DOMAIN_VIRT_VMWARE
     */
    if (virCapabilitiesAddGuestDomain(guest, "vmware", NULL, NULL, 0,
                                      NULL) == NULL) {
        goto failure;
    }

252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270
    /* x86_64 */
    if (supportsLongMode == esxVI_Boolean_True) {
        guest = virCapabilitiesAddGuest(caps, "hvm", "x86_64", 64, NULL, NULL,
                                        0, NULL);

        if (guest == NULL) {
            goto failure;
        }

        /*
         * FIXME: Maybe distinguish betwen ESX and GSX here, see
         * esxVMX_ParseConfig() and VIR_DOMAIN_VIRT_VMWARE
         */
        if (virCapabilitiesAddGuestDomain(guest, "vmware", NULL, NULL, 0,
                                          NULL) == NULL) {
            goto failure;
        }
    }

271 272 273 274 275 276 277 278 279 280
    return caps;

  failure:
    virCapabilitiesFree(caps);

    return NULL;
}



281
/*
M
Matthias Bolte 已提交
282
 * URI format: {esx|gsx}://[<username>@]<hostname>[:<port>]/[<query parameter> ...]
283
 *
284 285 286
 * If no port is specified the default port is set dependent on the scheme and
 * transport parameter:
 * - esx+http  80
287
 * - esx+https 443
288 289 290
 * - gsx+http  8222
 * - gsx+https 8333
 *
291 292 293 294 295
 * Optional query parameters:
 * - transport={http|https}
 * - vcenter={<vcenter>|*}
 * - no_verify={0|1}
 * - auto_answer={0|1}
M
Matthias Bolte 已提交
296
 * - proxy=[{http|socks|socks4|socks4a|socks5}://]<hostname>[:<port>]
297
 *
298 299 300
 * If no transport parameter is specified https is used.
 *
 * The vcenter parameter is only necessary for migration, because the vCenter
301 302 303 304
 * server is in charge to initiate a migration between two ESX hosts. The
 * vcenter parameter can be set to an explicity hostname or to *. If set to *,
 * the driver will check if the ESX host is managed by a vCenter and connect to
 * it. If the ESX host is not managed by a vCenter an error is reported.
305 306
 *
 * If the no_verify parameter is set to 1, this disables libcurl client checks
307
 * of the server's certificate. The default value it 0.
308 309 310 311
 *
 * If the auto_answer parameter is set to 1, the driver will respond to all
 * virtual machine questions with the default answer, otherwise virtual machine
 * questions will be reported as errors. The default value it 0.
M
Matthias Bolte 已提交
312 313 314 315
 *
 * The proxy parameter allows to specify a proxy for to be used by libcurl.
 * The default for the optional <type> part is http and socks is synonymous for
 * socks5. The optional <port> part allows to override the default port 1080.
316 317 318 319
 */
static virDrvOpenStatus
esxOpen(virConnectPtr conn, virConnectAuthPtr auth, int flags ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
320
    virDrvOpenStatus result = VIR_DRV_OPEN_ERROR;
321
    esxPrivate *priv = NULL;
322
    esxUtil_ParsedQuery *parsedQuery = NULL;
M
Matthias Bolte 已提交
323 324
    char hostIpAddress[NI_MAXHOST] = "";
    char vCenterIpAddress[NI_MAXHOST] = "";
325
    char *url = NULL;
M
Matthias Bolte 已提交
326
    char *vCenter = NULL;
327 328
    char *username = NULL;
    char *password = NULL;
329 330 331
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
332

333
    /* Decline if the URI is NULL or the scheme is neither 'esx' nor 'gsx' */
334
    if (conn->uri == NULL || conn->uri->scheme == NULL ||
335 336
        (STRCASENEQ(conn->uri->scheme, "esx") &&
         STRCASENEQ(conn->uri->scheme, "gsx"))) {
337 338 339
        return VIR_DRV_OPEN_DECLINED;
    }

M
Matthias Bolte 已提交
340 341 342
    /* Decline URIs without server part, or missing auth */
    if (conn->uri->server == NULL || auth == NULL || auth->cb == NULL) {
        return VIR_DRV_OPEN_DECLINED;
343 344
    }

345 346
    if (conn->uri->path != NULL && STRNEQ(conn->uri->path, "") &&
        STRNEQ(conn->uri->path, "/")) {
M
Matthias Bolte 已提交
347
        VIR_WARN("Ignoring unexpected path '%s' in URI", conn->uri->path);
348 349 350 351
    }

    /* Allocate per-connection private data */
    if (VIR_ALLOC(priv) < 0) {
352
        virReportOOMError();
M
Matthias Bolte 已提交
353
        goto cleanup;
354 355
    }

356 357 358 359 360 361 362
    if (esxUtil_ParseQuery(&parsedQuery, conn->uri) < 0) {
        goto cleanup;
    }

    priv->transport = parsedQuery->transport;
    parsedQuery->transport = NULL;

M
Matthias Bolte 已提交
363 364
    priv->maxVcpus = -1;
    priv->supportsVMotion = esxVI_Boolean_Undefined;
365
    priv->supportsLongMode = esxVI_Boolean_Undefined;
366 367
    priv->autoAnswer = parsedQuery->autoAnswer ? esxVI_Boolean_True
                                               : esxVI_Boolean_False;
368 369
    priv->usedCpuTimeCounterId = -1;

M
Matthias Bolte 已提交
370 371 372 373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388
    /*
     * Set the port dependent on the transport protocol if no port is
     * specified. This allows us to rely on the port parameter being
     * correctly set when building URIs later on, without the need to
     * distinguish between the situations port == 0 and port != 0
     */
    if (conn->uri->port == 0) {
        if (STRCASEEQ(conn->uri->scheme, "esx")) {
            if (STRCASEEQ(priv->transport, "https")) {
                conn->uri->port = 443;
            } else {
                conn->uri->port = 80;
            }
        } else { /* GSX */
            if (STRCASEEQ(priv->transport, "https")) {
                conn->uri->port = 8333;
            } else {
                conn->uri->port = 8222;
            }
389
        }
M
Matthias Bolte 已提交
390
    }
391

M
Matthias Bolte 已提交
392
    /* Login to host */
393
    if (esxUtil_ResolveHostname(conn->uri->server, hostIpAddress,
394
                                NI_MAXHOST) < 0) {
M
Matthias Bolte 已提交
395
        goto cleanup;
396 397
    }

M
Matthias Bolte 已提交
398 399
    if (virAsprintf(&url, "%s://%s:%d/sdk", priv->transport,
                    conn->uri->server, conn->uri->port) < 0) {
400
        virReportOOMError();
M
Matthias Bolte 已提交
401
        goto cleanup;
M
Matthias Bolte 已提交
402 403 404 405 406 407
    }

    if (conn->uri->user != NULL) {
        username = strdup(conn->uri->user);

        if (username == NULL) {
408
            virReportOOMError();
M
Matthias Bolte 已提交
409
            goto cleanup;
410
        }
M
Matthias Bolte 已提交
411
    } else {
412
        username = virRequestUsername(auth, "root", conn->uri->server);
413

M
Matthias Bolte 已提交
414
        if (username == NULL) {
415
            ESX_ERROR(VIR_ERR_AUTH_FAILED, "%s", _("Username request failed"));
M
Matthias Bolte 已提交
416
            goto cleanup;
417
        }
M
Matthias Bolte 已提交
418
    }
419

420
    password = virRequestPassword(auth, username, conn->uri->server);
M
Matthias Bolte 已提交
421 422

    if (password == NULL) {
423
        ESX_ERROR(VIR_ERR_AUTH_FAILED, "%s", _("Password request failed"));
M
Matthias Bolte 已提交
424
        goto cleanup;
M
Matthias Bolte 已提交
425 426
    }

427 428
    if (esxVI_Context_Alloc(&priv->host) < 0 ||
        esxVI_Context_Connect(priv->host, url, hostIpAddress, username,
M
Matthias Bolte 已提交
429
                              password, parsedQuery) < 0) {
M
Matthias Bolte 已提交
430
        goto cleanup;
M
Matthias Bolte 已提交
431 432 433 434 435
    }

    if (STRCASEEQ(conn->uri->scheme, "esx")) {
        if (priv->host->productVersion != esxVI_ProductVersion_ESX35 &&
            priv->host->productVersion != esxVI_ProductVersion_ESX40) {
436
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
437
                      _("%s is neither an ESX 3.5 host nor an ESX 4.0 host"),
M
Matthias Bolte 已提交
438
                      conn->uri->server);
M
Matthias Bolte 已提交
439
            goto cleanup;
440
        }
M
Matthias Bolte 已提交
441 442
    } else { /* GSX */
        if (priv->host->productVersion != esxVI_ProductVersion_GSX20) {
443
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
444
                      _("%s isn't a GSX 2.0 host"), conn->uri->server);
M
Matthias Bolte 已提交
445
            goto cleanup;
M
Matthias Bolte 已提交
446 447
        }
    }
448

449
    /* Query the host for maintenance mode and vCenter IP address */
450
    if (esxVI_String_AppendValueListToList(&propertyNameList,
451 452
                                           "runtime.inMaintenanceMode\0"
                                           "summary.managementServerIp\0") < 0 ||
453 454
        esxVI_LookupHostSystemByIp(priv->host, hostIpAddress, propertyNameList,
                                   &hostSystem) < 0) {
M
Matthias Bolte 已提交
455
        goto cleanup;
456 457 458 459 460 461
    }

    /* Warn if host is in maintenance mode */
    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "runtime.inMaintenanceMode")) {
462
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
463
                                         esxVI_Type_Boolean) < 0) {
M
Matthias Bolte 已提交
464
                goto cleanup;
465 466 467 468 469 470 471 472 473 474
            }

            if (dynamicProperty->val->boolean == esxVI_Boolean_True) {
                VIR_WARN0("The server is in maintenance mode");
            }

            break;
        }
    }

M
Matthias Bolte 已提交
475 476
    /* Login to vCenter */
    if (vCenter != NULL) {
477 478 479 480 481 482
        VIR_FREE(url);
        VIR_FREE(password);
        VIR_FREE(username);

        /* If a vCenter is specified resolve the hostname */
        if (STRNEQ(vCenter, "*") &&
483
            esxUtil_ResolveHostname(vCenter, vCenterIpAddress,
484
                                    NI_MAXHOST) < 0) {
M
Matthias Bolte 已提交
485
            goto cleanup;
486 487 488 489 490
        }

        /* Lookup the vCenter from the ESX host */
        for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
             dynamicProperty = dynamicProperty->_next) {
491
            if (STREQ(dynamicProperty->name, "summary.managementServerIp")) {
492
                if (esxVI_AnyType_ExpectType(dynamicProperty->val,
493
                                             esxVI_Type_String) < 0) {
M
Matthias Bolte 已提交
494
                    goto cleanup;
495 496 497 498 499 500 501 502 503
                }

                /* Get the vCenter IP address or verify the specified one */
                if (STREQ(vCenter, "*")) {
                    VIR_FREE(vCenter);

                    vCenter = strdup(dynamicProperty->val->string);

                    if (vCenter == NULL) {
504
                        virReportOOMError();
M
Matthias Bolte 已提交
505
                        goto cleanup;
506 507 508 509
                    }

                    if (virStrcpyStatic(vCenterIpAddress,
                                        dynamicProperty->val->string) == NULL) {
510
                        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
511 512 513
                                  _("vCenter IP address %s too big for "
                                    "destination"),
                                  dynamicProperty->val->string);
M
Matthias Bolte 已提交
514
                        goto cleanup;
515 516 517
                    }
                } else if (STRNEQ(vCenterIpAddress,
                           dynamicProperty->val->string)) {
518
                    ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
519 520 521
                              _("This host is managed by a vCenter with IP "
                                "address %s, but a mismachting vCenter '%s' "
                                "(%s) has been specified"),
522 523
                              dynamicProperty->val->string, vCenter,
                              vCenterIpAddress);
M
Matthias Bolte 已提交
524
                    goto cleanup;
525 526 527 528 529 530 531
                }

                break;
            }
        }

        if (STREQ(vCenter, "*")) {
532 533
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                      _("This host is not managed by a vCenter"));
M
Matthias Bolte 已提交
534
            goto cleanup;
535 536
        }

M
Matthias Bolte 已提交
537 538
        if (virAsprintf(&url, "%s://%s/sdk", priv->transport,
                        vCenter) < 0) {
539
            virReportOOMError();
M
Matthias Bolte 已提交
540
            goto cleanup;
M
Matthias Bolte 已提交
541
        }
542

543
        if (esxVI_Context_Alloc(&priv->vCenter) < 0) {
M
Matthias Bolte 已提交
544
            goto cleanup;
545 546
        }

547
        username = virRequestUsername(auth, "administrator", vCenter);
M
Matthias Bolte 已提交
548 549

        if (username == NULL) {
550
            ESX_ERROR(VIR_ERR_AUTH_FAILED, "%s", _("Username request failed"));
M
Matthias Bolte 已提交
551
            goto cleanup;
552 553
        }

554
        password = virRequestPassword(auth, username, vCenter);
555 556

        if (password == NULL) {
557
            ESX_ERROR(VIR_ERR_AUTH_FAILED, "%s", _("Password request failed"));
M
Matthias Bolte 已提交
558
            goto cleanup;
559 560
        }

561
        if (esxVI_Context_Connect(priv->vCenter, url, vCenterIpAddress,
M
Matthias Bolte 已提交
562
                                  username, password, parsedQuery) < 0) {
M
Matthias Bolte 已提交
563
            goto cleanup;
564 565
        }

M
Matthias Bolte 已提交
566 567
        if (priv->vCenter->productVersion != esxVI_ProductVersion_VPX25 &&
            priv->vCenter->productVersion != esxVI_ProductVersion_VPX40) {
568
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
569 570
                      _("%s is neither a vCenter 2.5 server nor a vCenter "
                        "4.0 server"), conn->uri->server);
M
Matthias Bolte 已提交
571
            goto cleanup;
572
        }
573 574 575
    }

    conn->privateData = priv;
576

M
Matthias Bolte 已提交
577
    /* Setup capabilities */
578
    priv->caps = esxCapsInit(priv);
579

M
Matthias Bolte 已提交
580
    if (priv->caps == NULL) {
M
Matthias Bolte 已提交
581
        goto cleanup;
582 583
    }

M
Matthias Bolte 已提交
584
    result = VIR_DRV_OPEN_SUCCESS;
585

M
Matthias Bolte 已提交
586 587
  cleanup:
    if (result == VIR_DRV_OPEN_ERROR && priv != NULL) {
588
        esxVI_Context_Free(&priv->host);
M
Matthias Bolte 已提交
589
        esxVI_Context_Free(&priv->vCenter);
590

591 592
        virCapabilitiesFree(priv->caps);

M
Matthias Bolte 已提交
593
        VIR_FREE(priv->transport);
594 595 596
        VIR_FREE(priv);
    }

597
    esxUtil_FreeParsedQuery(&parsedQuery);
M
Matthias Bolte 已提交
598 599 600 601 602 603
    VIR_FREE(url);
    VIR_FREE(vCenter);
    VIR_FREE(password);
    VIR_FREE(username);
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);
604

M
Matthias Bolte 已提交
605
    return result;
606 607 608 609 610 611 612
}



static int
esxClose(virConnectPtr conn)
{
M
Matthias Bolte 已提交
613
    esxPrivate *priv = conn->privateData;
E
Eric Blake 已提交
614
    int result = 0;
615

E
Eric Blake 已提交
616 617 618 619
    if (esxVI_EnsureSession(priv->host) < 0 ||
        esxVI_Logout(priv->host) < 0) {
        result = -1;
    }
620

M
Matthias Bolte 已提交
621
    esxVI_Context_Free(&priv->host);
622

M
Matthias Bolte 已提交
623
    if (priv->vCenter != NULL) {
E
Eric Blake 已提交
624 625 626 627
        if (esxVI_EnsureSession(priv->vCenter) < 0 ||
            esxVI_Logout(priv->vCenter) < 0) {
            result = -1;
        }
628

M
Matthias Bolte 已提交
629
        esxVI_Context_Free(&priv->vCenter);
630 631
    }

632 633
    virCapabilitiesFree(priv->caps);

634 635 636 637 638
    VIR_FREE(priv->transport);
    VIR_FREE(priv);

    conn->privateData = NULL;

E
Eric Blake 已提交
639
    return result;
640 641 642 643 644
}



static esxVI_Boolean
645
esxSupportsVMotion(esxPrivate *priv)
646 647 648 649 650
{
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;

M
Matthias Bolte 已提交
651 652
    if (priv->supportsVMotion != esxVI_Boolean_Undefined) {
        return priv->supportsVMotion;
653 654
    }

655
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
656
        return esxVI_Boolean_Undefined;
657 658
    }

659
    if (esxVI_String_AppendValueToList(&propertyNameList,
660
                                       "capability.vmotionSupported") < 0 ||
661
        esxVI_LookupObjectContentByType(priv->host, priv->host->hostFolder,
662 663
                                        "HostSystem", propertyNameList,
                                        esxVI_Boolean_True, &hostSystem) < 0) {
M
Matthias Bolte 已提交
664
        goto cleanup;
665 666 667
    }

    if (hostSystem == NULL) {
668 669
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve the HostSystem object"));
M
Matthias Bolte 已提交
670
        goto cleanup;
671 672 673 674 675
    }

    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "capability.vmotionSupported")) {
676
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
677
                                         esxVI_Type_Boolean) < 0) {
M
Matthias Bolte 已提交
678
                goto cleanup;
679 680
            }

M
Matthias Bolte 已提交
681
            priv->supportsVMotion = dynamicProperty->val->boolean;
682 683 684 685 686 687 688
            break;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

  cleanup:
M
Matthias Bolte 已提交
689 690 691 692
    /*
     * If we goto cleanup in case of an error then priv->supportsVMotion is
     * still esxVI_Boolean_Undefined, therefore we don't need to set it.
     */
693 694 695
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);

M
Matthias Bolte 已提交
696
    return priv->supportsVMotion;
697 698 699 700 701 702 703
}



static int
esxSupportsFeature(virConnectPtr conn, int feature)
{
M
Matthias Bolte 已提交
704
    esxPrivate *priv = conn->privateData;
M
Matthias Bolte 已提交
705
    esxVI_Boolean supportsVMotion = esxVI_Boolean_Undefined;
706 707 708

    switch (feature) {
      case VIR_DRV_FEATURE_MIGRATION_V1:
709
        supportsVMotion = esxSupportsVMotion(priv);
710

M
Matthias Bolte 已提交
711
        if (supportsVMotion == esxVI_Boolean_Undefined) {
712 713 714
            return -1;
        }

M
Matthias Bolte 已提交
715 716 717
        /* Migration is only possible via a vCenter and if VMotion is enabled */
        return priv->vCenter != NULL &&
               supportsVMotion == esxVI_Boolean_True ? 1 : 0;
718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736

      default:
        return 0;
    }
}



static const char *
esxGetType(virConnectPtr conn ATTRIBUTE_UNUSED)
{
    return "ESX";
}



static int
esxGetVersion(virConnectPtr conn, unsigned long *version)
{
M
Matthias Bolte 已提交
737
    esxPrivate *priv = conn->privateData;
738

739 740 741
    if (virParseVersionString(priv->host->service->about->version,
                              version) < 0) {
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
742
                  _("Could not parse version number from '%s'"),
743
                  priv->host->service->about->version);
744

745
        return -1;
746 747 748 749 750 751 752 753 754 755
    }

    return 0;
}



static char *
esxGetHostname(virConnectPtr conn)
{
M
Matthias Bolte 已提交
756
    esxPrivate *priv = conn->privateData;
757 758 759 760 761 762 763
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    const char *hostName = NULL;
    const char *domainName = NULL;
    char *complete = NULL;

764
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
765
        return NULL;
766 767 768
    }

    if (esxVI_String_AppendValueListToList
769
          (&propertyNameList,
770 771
           "config.network.dnsConfig.hostName\0"
           "config.network.dnsConfig.domainName\0") < 0 ||
772
        esxVI_LookupObjectContentByType(priv->host, priv->host->hostFolder,
773 774
                                        "HostSystem", propertyNameList,
                                        esxVI_Boolean_True, &hostSystem) < 0) {
M
Matthias Bolte 已提交
775
        goto cleanup;
776 777 778
    }

    if (hostSystem == NULL) {
779 780
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve the HostSystem object"));
M
Matthias Bolte 已提交
781
        goto cleanup;
782 783 784 785 786 787
    }

    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name,
                  "config.network.dnsConfig.hostName")) {
788
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
789
                                         esxVI_Type_String) < 0) {
M
Matthias Bolte 已提交
790
                goto cleanup;
791 792 793 794 795
            }

            hostName = dynamicProperty->val->string;
        } else if (STREQ(dynamicProperty->name,
                         "config.network.dnsConfig.domainName")) {
796
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
797
                                         esxVI_Type_String) < 0) {
M
Matthias Bolte 已提交
798
                goto cleanup;
799 800 801 802 803 804 805 806
            }

            domainName = dynamicProperty->val->string;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

M
Matthias Bolte 已提交
807
    if (hostName == NULL || strlen(hostName) < 1) {
808 809
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Missing or empty 'hostName' property"));
M
Matthias Bolte 已提交
810
        goto cleanup;
811 812
    }

M
Matthias Bolte 已提交
813
    if (domainName == NULL || strlen(domainName) < 1) {
814
        complete = strdup(hostName);
815

816
        if (complete == NULL) {
817
            virReportOOMError();
M
Matthias Bolte 已提交
818
            goto cleanup;
819 820 821
        }
    } else {
        if (virAsprintf(&complete, "%s.%s", hostName, domainName) < 0) {
822
            virReportOOMError();
M
Matthias Bolte 已提交
823
            goto cleanup;
824
        }
825 826 827
    }

  cleanup:
M
Matthias Bolte 已提交
828 829 830 831 832
    /*
     * If we goto cleanup in case of an error then complete is still NULL,
     * either strdup returned NULL or virAsprintf failed. When virAsprintf
     * fails it guarantees setting complete to NULL
     */
833 834 835 836 837 838 839 840 841 842 843
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);

    return complete;
}



static int
esxNodeGetInfo(virConnectPtr conn, virNodeInfoPtr nodeinfo)
{
M
Matthias Bolte 已提交
844
    int result = -1;
M
Matthias Bolte 已提交
845
    esxPrivate *priv = conn->privateData;
846 847 848 849 850 851 852 853 854 855 856
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    int64_t cpuInfo_hz = 0;
    int16_t cpuInfo_numCpuCores = 0;
    int16_t cpuInfo_numCpuPackages = 0;
    int16_t cpuInfo_numCpuThreads = 0;
    int64_t memorySize = 0;
    int32_t numaInfo_numNodes = 0;
    char *ptr = NULL;

M
Matthias Bolte 已提交
857
    memset(nodeinfo, 0, sizeof (*nodeinfo));
858

859
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
860
        return -1;
861 862
    }

863
    if (esxVI_String_AppendValueListToList(&propertyNameList,
864 865 866 867 868 869 870
                                           "hardware.cpuInfo.hz\0"
                                           "hardware.cpuInfo.numCpuCores\0"
                                           "hardware.cpuInfo.numCpuPackages\0"
                                           "hardware.cpuInfo.numCpuThreads\0"
                                           "hardware.memorySize\0"
                                           "hardware.numaInfo.numNodes\0"
                                           "summary.hardware.cpuModel\0") < 0 ||
871
        esxVI_LookupObjectContentByType(priv->host, priv->host->hostFolder,
872 873
                                        "HostSystem", propertyNameList,
                                        esxVI_Boolean_True, &hostSystem) < 0) {
M
Matthias Bolte 已提交
874
        goto cleanup;
875 876 877
    }

    if (hostSystem == NULL) {
878 879
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve the HostSystem object"));
M
Matthias Bolte 已提交
880
        goto cleanup;
881 882 883 884 885
    }

    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "hardware.cpuInfo.hz")) {
886
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
887
                                         esxVI_Type_Long) < 0) {
M
Matthias Bolte 已提交
888
                goto cleanup;
889 890 891 892 893
            }

            cpuInfo_hz = dynamicProperty->val->int64;
        } else if (STREQ(dynamicProperty->name,
                         "hardware.cpuInfo.numCpuCores")) {
894
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
895
                                         esxVI_Type_Short) < 0) {
M
Matthias Bolte 已提交
896
                goto cleanup;
897 898 899 900 901
            }

            cpuInfo_numCpuCores = dynamicProperty->val->int16;
        } else if (STREQ(dynamicProperty->name,
                         "hardware.cpuInfo.numCpuPackages")) {
902
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
903
                                         esxVI_Type_Short) < 0) {
M
Matthias Bolte 已提交
904
                goto cleanup;
905 906 907 908 909
            }

            cpuInfo_numCpuPackages = dynamicProperty->val->int16;
        } else if (STREQ(dynamicProperty->name,
                         "hardware.cpuInfo.numCpuThreads")) {
910
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
911
                                         esxVI_Type_Short) < 0) {
M
Matthias Bolte 已提交
912
                goto cleanup;
913 914 915 916
            }

            cpuInfo_numCpuThreads = dynamicProperty->val->int16;
        } else if (STREQ(dynamicProperty->name, "hardware.memorySize")) {
917
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
918
                                         esxVI_Type_Long) < 0) {
M
Matthias Bolte 已提交
919
                goto cleanup;
920 921 922 923 924
            }

            memorySize = dynamicProperty->val->int64;
        } else if (STREQ(dynamicProperty->name,
                         "hardware.numaInfo.numNodes")) {
925
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
926
                                         esxVI_Type_Int) < 0) {
M
Matthias Bolte 已提交
927
                goto cleanup;
928 929 930 931 932
            }

            numaInfo_numNodes = dynamicProperty->val->int32;
        } else if (STREQ(dynamicProperty->name,
                         "summary.hardware.cpuModel")) {
933
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
934
                                         esxVI_Type_String) < 0) {
M
Matthias Bolte 已提交
935
                goto cleanup;
936 937 938 939 940 941
            }

            ptr = dynamicProperty->val->string;

            /* Strip the string to fit more relevant information in 32 chars */
            while (*ptr != '\0') {
M
Matthias Bolte 已提交
942 943
                if (STRPREFIX(ptr, "  ")) {
                    memmove(ptr, ptr + 1, strlen(ptr + 1) + 1);
944
                    continue;
945
                } else if (STRPREFIX(ptr, "(R)") || STRPREFIX(ptr, "(C)")) {
M
Matthias Bolte 已提交
946
                    memmove(ptr, ptr + 3, strlen(ptr + 3) + 1);
947
                    continue;
948 949 950
                } else if (STRPREFIX(ptr, "(TM)")) {
                    memmove(ptr, ptr + 4, strlen(ptr + 4) + 1);
                    continue;
951 952 953 954 955
                }

                ++ptr;
            }

C
Chris Lalancette 已提交
956 957 958
            if (virStrncpy(nodeinfo->model, dynamicProperty->val->string,
                           sizeof(nodeinfo->model) - 1,
                           sizeof(nodeinfo->model)) == NULL) {
959
                ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
960
                          _("CPU Model %s too long for destination"),
C
Chris Lalancette 已提交
961
                          dynamicProperty->val->string);
M
Matthias Bolte 已提交
962
                goto cleanup;
C
Chris Lalancette 已提交
963
            }
964 965 966 967 968 969 970
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

    nodeinfo->memory = memorySize / 1024; /* Scale from bytes to kilobytes */
    nodeinfo->cpus = cpuInfo_numCpuCores;
971
    nodeinfo->mhz = cpuInfo_hz / (1000 * 1000); /* Scale from hz to mhz */
972 973 974 975 976 977 978 979 980
    nodeinfo->nodes = numaInfo_numNodes;
    nodeinfo->sockets = cpuInfo_numCpuPackages;
    nodeinfo->cores = cpuInfo_numCpuPackages > 0
                        ? cpuInfo_numCpuCores / cpuInfo_numCpuPackages
                        : 0;
    nodeinfo->threads = cpuInfo_numCpuCores > 0
                          ? cpuInfo_numCpuThreads / cpuInfo_numCpuCores
                          : 0;

M
Matthias Bolte 已提交
981 982
    result = 0;

983 984 985 986 987 988 989 990 991
  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);

    return result;
}



992 993 994
static char *
esxGetCapabilities(virConnectPtr conn)
{
M
Matthias Bolte 已提交
995
    esxPrivate *priv = conn->privateData;
M
Matthias Bolte 已提交
996
    char *xml = virCapabilitiesFormatXML(priv->caps);
997 998

    if (xml == NULL) {
999
        virReportOOMError();
1000 1001 1002 1003 1004 1005 1006 1007
        return NULL;
    }

    return xml;
}



1008 1009 1010
static int
esxListDomains(virConnectPtr conn, int *ids, int maxids)
{
M
Matthias Bolte 已提交
1011
    bool success = false;
M
Matthias Bolte 已提交
1012
    esxPrivate *priv = conn->privateData;
1013 1014 1015 1016 1017 1018 1019
    esxVI_ObjectContent *virtualMachineList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;
    int count = 0;

    if (ids == NULL || maxids < 0) {
1020 1021
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s", _("Invalid argument"));
        return -1;
1022 1023 1024 1025 1026 1027
    }

    if (maxids == 0) {
        return 0;
    }

1028
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1029
        return -1;
1030 1031
    }

1032
    if (esxVI_String_AppendValueToList(&propertyNameList,
1033
                                       "runtime.powerState") < 0 ||
1034
        esxVI_LookupObjectContentByType(priv->host, priv->host->vmFolder,
1035 1036 1037
                                        "VirtualMachine", propertyNameList,
                                        esxVI_Boolean_True,
                                        &virtualMachineList) < 0) {
M
Matthias Bolte 已提交
1038
        goto cleanup;
1039 1040 1041 1042
    }

    for (virtualMachine = virtualMachineList; virtualMachine != NULL;
         virtualMachine = virtualMachine->_next) {
1043
        if (esxVI_GetVirtualMachinePowerState(virtualMachine,
1044
                                              &powerState) < 0) {
M
Matthias Bolte 已提交
1045
            goto cleanup;
1046 1047 1048 1049 1050 1051 1052 1053 1054
        }

        if (powerState != esxVI_VirtualMachinePowerState_PoweredOn) {
            continue;
        }

        if (esxUtil_ParseVirtualMachineIDString(virtualMachine->obj->value,
                                                &ids[count]) < 0 ||
            ids[count] <= 0) {
1055
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
1056
                      _("Failed to parse positive integer from '%s'"),
1057
                      virtualMachine->obj->value);
M
Matthias Bolte 已提交
1058
            goto cleanup;
1059 1060 1061 1062 1063 1064 1065 1066 1067
        }

        count++;

        if (count >= maxids) {
            break;
        }
    }

M
Matthias Bolte 已提交
1068 1069
    success = true;

1070 1071 1072 1073
  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachineList);

M
Matthias Bolte 已提交
1074
    return success ? count : -1;
1075 1076 1077 1078 1079 1080 1081
}



static int
esxNumberOfDomains(virConnectPtr conn)
{
M
Matthias Bolte 已提交
1082
    esxPrivate *priv = conn->privateData;
1083

1084
    if (esxVI_EnsureSession(priv->host) < 0) {
1085 1086 1087
        return -1;
    }

1088
    return esxVI_LookupNumberOfDomainsByPowerState
1089
             (priv->host, esxVI_VirtualMachinePowerState_PoweredOn,
1090 1091 1092 1093 1094 1095 1096 1097
              esxVI_Boolean_False);
}



static virDomainPtr
esxDomainLookupByID(virConnectPtr conn, int id)
{
M
Matthias Bolte 已提交
1098
    esxPrivate *priv = conn->privateData;
1099 1100 1101 1102
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachineList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachinePowerState powerState;
M
Matthias Bolte 已提交
1103 1104 1105
    int id_candidate = -1;
    char *name_candidate = NULL;
    unsigned char uuid_candidate[VIR_UUID_BUFLEN];
1106 1107
    virDomainPtr domain = NULL;

1108
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1109
        return NULL;
1110 1111
    }

1112
    if (esxVI_String_AppendValueListToList(&propertyNameList,
1113
                                           "configStatus\0"
1114 1115
                                           "name\0"
                                           "runtime.powerState\0"
1116
                                           "config.uuid\0") < 0 ||
1117
        esxVI_LookupObjectContentByType(priv->host, priv->host->vmFolder,
1118 1119 1120
                                        "VirtualMachine", propertyNameList,
                                        esxVI_Boolean_True,
                                        &virtualMachineList) < 0) {
M
Matthias Bolte 已提交
1121
        goto cleanup;
1122 1123 1124 1125
    }

    for (virtualMachine = virtualMachineList; virtualMachine != NULL;
         virtualMachine = virtualMachine->_next) {
1126
        if (esxVI_GetVirtualMachinePowerState(virtualMachine,
1127
                                              &powerState) < 0) {
M
Matthias Bolte 已提交
1128
            goto cleanup;
1129 1130 1131 1132 1133 1134 1135
        }

        /* Only running/suspended domains have an ID != -1 */
        if (powerState == esxVI_VirtualMachinePowerState_PoweredOff) {
            continue;
        }

M
Matthias Bolte 已提交
1136
        VIR_FREE(name_candidate);
1137

1138
        if (esxVI_GetVirtualMachineIdentity(virtualMachine,
M
Matthias Bolte 已提交
1139 1140
                                            &id_candidate, &name_candidate,
                                            uuid_candidate) < 0) {
M
Matthias Bolte 已提交
1141
            goto cleanup;
1142 1143
        }

M
Matthias Bolte 已提交
1144
        if (id != id_candidate) {
1145 1146 1147
            continue;
        }

M
Matthias Bolte 已提交
1148
        domain = virGetDomain(conn, name_candidate, uuid_candidate);
1149 1150

        if (domain == NULL) {
M
Matthias Bolte 已提交
1151
            goto cleanup;
1152 1153 1154 1155 1156 1157 1158 1159
        }

        domain->id = id;

        break;
    }

    if (domain == NULL) {
1160
        ESX_ERROR(VIR_ERR_NO_DOMAIN, _("No domain with ID %d"), id);
1161 1162 1163 1164 1165
    }

  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachineList);
M
Matthias Bolte 已提交
1166
    VIR_FREE(name_candidate);
1167 1168 1169 1170 1171 1172 1173 1174 1175

    return domain;
}



static virDomainPtr
esxDomainLookupByUUID(virConnectPtr conn, const unsigned char *uuid)
{
M
Matthias Bolte 已提交
1176
    esxPrivate *priv = conn->privateData;
1177 1178 1179
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachinePowerState powerState;
1180 1181
    int id = -1;
    char *name = NULL;
1182 1183
    virDomainPtr domain = NULL;

1184
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1185
        return NULL;
1186 1187
    }

1188
    if (esxVI_String_AppendValueListToList(&propertyNameList,
1189
                                           "name\0"
1190
                                           "runtime.powerState\0") < 0 ||
1191 1192
        esxVI_LookupVirtualMachineByUuid(priv->host, uuid, propertyNameList,
                                         &virtualMachine,
M
Matthias Bolte 已提交
1193
                                         esxVI_Occurrence_RequiredItem) < 0 ||
1194 1195
        esxVI_GetVirtualMachineIdentity(virtualMachine, &id, &name, NULL) < 0 ||
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
1196
        goto cleanup;
1197 1198
    }

1199
    domain = virGetDomain(conn, name, uuid);
1200 1201

    if (domain == NULL) {
M
Matthias Bolte 已提交
1202
        goto cleanup;
1203
    }
1204

1205 1206 1207 1208 1209
    /* Only running/suspended virtual machines have an ID != -1 */
    if (powerState != esxVI_VirtualMachinePowerState_PoweredOff) {
        domain->id = id;
    } else {
        domain->id = -1;
1210 1211 1212 1213
    }

  cleanup:
    esxVI_String_Free(&propertyNameList);
1214 1215
    esxVI_ObjectContent_Free(&virtualMachine);
    VIR_FREE(name);
1216 1217 1218 1219 1220 1221 1222 1223 1224

    return domain;
}



static virDomainPtr
esxDomainLookupByName(virConnectPtr conn, const char *name)
{
M
Matthias Bolte 已提交
1225
    esxPrivate *priv = conn->privateData;
1226 1227 1228
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachinePowerState powerState;
1229 1230
    int id = -1;
    unsigned char uuid[VIR_UUID_BUFLEN];
1231 1232
    virDomainPtr domain = NULL;

1233
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1234
        return NULL;
1235 1236
    }

1237
    if (esxVI_String_AppendValueListToList(&propertyNameList,
1238
                                           "configStatus\0"
1239
                                           "runtime.powerState\0"
1240
                                           "config.uuid\0") < 0 ||
1241 1242 1243
        esxVI_LookupVirtualMachineByName(priv->host, name, propertyNameList,
                                         &virtualMachine,
                                         esxVI_Occurrence_OptionalItem) < 0) {
M
Matthias Bolte 已提交
1244
        goto cleanup;
1245 1246
    }

1247
    if (virtualMachine == NULL) {
1248
        ESX_ERROR(VIR_ERR_NO_DOMAIN, _("No domain with name '%s'"), name);
M
Matthias Bolte 已提交
1249
        goto cleanup;
1250
    }
1251 1252


M
Matthias Bolte 已提交
1253 1254 1255
    if (esxVI_GetVirtualMachineIdentity(virtualMachine, &id, NULL, uuid) < 0 ||
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
        goto cleanup;
1256
    }
1257

1258
    domain = virGetDomain(conn, name, uuid);
1259

1260
    if (domain == NULL) {
M
Matthias Bolte 已提交
1261
        goto cleanup;
1262 1263
    }

1264 1265 1266 1267 1268
    /* Only running/suspended virtual machines have an ID != -1 */
    if (powerState != esxVI_VirtualMachinePowerState_PoweredOff) {
        domain->id = id;
    } else {
        domain->id = -1;
1269 1270 1271 1272
    }

  cleanup:
    esxVI_String_Free(&propertyNameList);
1273
    esxVI_ObjectContent_Free(&virtualMachine);
1274 1275 1276 1277 1278 1279 1280 1281 1282

    return domain;
}



static int
esxDomainSuspend(virDomainPtr domain)
{
M
Matthias Bolte 已提交
1283
    int result = -1;
M
Matthias Bolte 已提交
1284
    esxPrivate *priv = domain->conn->privateData;
1285 1286 1287 1288 1289 1290
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

1291
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1292
        return -1;
1293 1294
    }

1295
    if (esxVI_String_AppendValueToList(&propertyNameList,
1296
                                       "runtime.powerState") < 0 ||
1297
        esxVI_LookupVirtualMachineByUuidAndPrepareForTask
1298 1299 1300
          (priv->host, domain->uuid, propertyNameList, &virtualMachine,
           priv->autoAnswer) < 0 ||
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
1301
        goto cleanup;
1302 1303 1304
    }

    if (powerState != esxVI_VirtualMachinePowerState_PoweredOn) {
1305 1306
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s",
                  _("Domain is not powered on"));
M
Matthias Bolte 已提交
1307
        goto cleanup;
1308 1309
    }

1310 1311 1312
    if (esxVI_SuspendVM_Task(priv->host, virtualMachine->obj, &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
1313
        goto cleanup;
1314 1315 1316
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
1317
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s", _("Could not suspend domain"));
M
Matthias Bolte 已提交
1318
        goto cleanup;
1319 1320
    }

M
Matthias Bolte 已提交
1321 1322
    result = 0;

1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainResume(virDomainPtr domain)
{
M
Matthias Bolte 已提交
1336
    int result = -1;
M
Matthias Bolte 已提交
1337
    esxPrivate *priv = domain->conn->privateData;
1338 1339 1340 1341 1342 1343
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

1344
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1345
        return -1;
1346 1347
    }

1348
    if (esxVI_String_AppendValueToList(&propertyNameList,
1349
                                       "runtime.powerState") < 0 ||
1350
        esxVI_LookupVirtualMachineByUuidAndPrepareForTask
1351 1352 1353
          (priv->host, domain->uuid, propertyNameList, &virtualMachine,
           priv->autoAnswer) < 0 ||
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
1354
        goto cleanup;
1355 1356 1357
    }

    if (powerState != esxVI_VirtualMachinePowerState_Suspended) {
1358
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s", _("Domain is not suspended"));
M
Matthias Bolte 已提交
1359
        goto cleanup;
1360 1361
    }

1362 1363
    if (esxVI_PowerOnVM_Task(priv->host, virtualMachine->obj, NULL,
                             &task) < 0 ||
1364 1365
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
1366
        goto cleanup;
1367 1368 1369
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
1370
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s", _("Could not resume domain"));
M
Matthias Bolte 已提交
1371
        goto cleanup;
1372 1373
    }

M
Matthias Bolte 已提交
1374 1375
    result = 0;

1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainShutdown(virDomainPtr domain)
{
M
Matthias Bolte 已提交
1389
    int result = -1;
M
Matthias Bolte 已提交
1390
    esxPrivate *priv = domain->conn->privateData;
1391 1392 1393 1394
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;

1395
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1396
        return -1;
1397 1398
    }

1399
    if (esxVI_String_AppendValueToList(&propertyNameList,
1400
                                       "runtime.powerState") < 0 ||
1401 1402
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
1403
                                         esxVI_Occurrence_RequiredItem) < 0 ||
1404
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
1405
        goto cleanup;
1406 1407 1408
    }

    if (powerState != esxVI_VirtualMachinePowerState_PoweredOn) {
1409 1410
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s",
                  _("Domain is not powered on"));
M
Matthias Bolte 已提交
1411
        goto cleanup;
1412 1413
    }

1414
    if (esxVI_ShutdownGuest(priv->host, virtualMachine->obj) < 0) {
M
Matthias Bolte 已提交
1415
        goto cleanup;
1416 1417
    }

M
Matthias Bolte 已提交
1418 1419
    result = 0;

1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);

    return result;
}



static int
esxDomainReboot(virDomainPtr domain, unsigned int flags ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
1432
    int result = -1;
M
Matthias Bolte 已提交
1433
    esxPrivate *priv = domain->conn->privateData;
1434 1435 1436 1437
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;

1438
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1439
        return -1;
1440 1441
    }

1442
    if (esxVI_String_AppendValueToList(&propertyNameList,
1443
                                       "runtime.powerState") < 0 ||
1444 1445
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
1446
                                         esxVI_Occurrence_RequiredItem) < 0 ||
1447
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
1448
        goto cleanup;
1449 1450 1451
    }

    if (powerState != esxVI_VirtualMachinePowerState_PoweredOn) {
1452 1453
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s",
                  _("Domain is not powered on"));
M
Matthias Bolte 已提交
1454
        goto cleanup;
1455 1456
    }

1457
    if (esxVI_RebootGuest(priv->host, virtualMachine->obj) < 0) {
M
Matthias Bolte 已提交
1458
        goto cleanup;
1459 1460
    }

M
Matthias Bolte 已提交
1461 1462
    result = 0;

1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);

    return result;
}



static int
esxDomainDestroy(virDomainPtr domain)
{
M
Matthias Bolte 已提交
1475
    int result = -1;
M
Matthias Bolte 已提交
1476
    esxPrivate *priv = domain->conn->privateData;
1477
    esxVI_Context *ctx = NULL;
1478 1479 1480 1481 1482 1483
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

1484 1485 1486 1487 1488 1489
    if (priv->vCenter != NULL) {
        ctx = priv->vCenter;
    } else {
        ctx = priv->host;
    }

1490
    if (esxVI_EnsureSession(ctx) < 0) {
M
Matthias Bolte 已提交
1491
        return -1;
1492 1493
    }

1494
    if (esxVI_String_AppendValueToList(&propertyNameList,
1495
                                       "runtime.powerState") < 0 ||
1496
        esxVI_LookupVirtualMachineByUuidAndPrepareForTask
1497
          (ctx, domain->uuid, propertyNameList, &virtualMachine,
1498
           priv->autoAnswer) < 0 ||
1499
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
1500
        goto cleanup;
1501 1502 1503
    }

    if (powerState != esxVI_VirtualMachinePowerState_PoweredOn) {
1504 1505
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s",
                  _("Domain is not powered on"));
M
Matthias Bolte 已提交
1506
        goto cleanup;
1507 1508
    }

1509 1510 1511
    if (esxVI_PowerOffVM_Task(ctx, virtualMachine->obj, &task) < 0 ||
        esxVI_WaitForTaskCompletion(ctx, task, domain->uuid, priv->autoAnswer,
                                    &taskInfoState) < 0) {
M
Matthias Bolte 已提交
1512
        goto cleanup;
1513 1514 1515
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
1516
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s", _("Could not destroy domain"));
M
Matthias Bolte 已提交
1517
        goto cleanup;
1518 1519
    }

M
Matthias Bolte 已提交
1520 1521
    result = 0;

1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static char *
1533
esxDomainGetOSType(virDomainPtr domain ATTRIBUTE_UNUSED)
1534
{
1535 1536 1537
    char *osType = strdup("hvm");

    if (osType == NULL) {
1538
        virReportOOMError();
1539 1540 1541 1542
        return NULL;
    }

    return osType;
1543 1544 1545 1546 1547 1548 1549
}



static unsigned long
esxDomainGetMaxMemory(virDomainPtr domain)
{
M
Matthias Bolte 已提交
1550
    esxPrivate *priv = domain->conn->privateData;
1551 1552 1553 1554 1555
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    unsigned long memoryMB = 0;

1556
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1557
        return 0;
1558 1559
    }

1560
    if (esxVI_String_AppendValueToList(&propertyNameList,
1561
                                       "config.hardware.memoryMB") < 0 ||
1562 1563
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
1564
                                         esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
1565
        goto cleanup;
1566 1567 1568 1569 1570
    }

    for (dynamicProperty = virtualMachine->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "config.hardware.memoryMB")) {
1571
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
1572
                                         esxVI_Type_Int) < 0) {
M
Matthias Bolte 已提交
1573
                goto cleanup;
1574 1575 1576
            }

            if (dynamicProperty->val->int32 < 0) {
1577 1578
                ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
                          _("Got invalid memory size %d"),
1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601
                          dynamicProperty->val->int32);
            } else {
                memoryMB = dynamicProperty->val->int32;
            }

            break;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachine);

    return memoryMB * 1024; /* Scale from megabyte to kilobyte */
}



static int
esxDomainSetMaxMemory(virDomainPtr domain, unsigned long memory)
{
M
Matthias Bolte 已提交
1602
    int result = -1;
M
Matthias Bolte 已提交
1603
    esxPrivate *priv = domain->conn->privateData;
1604 1605 1606 1607 1608
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachineConfigSpec *spec = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

1609
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1610
        return -1;
1611 1612
    }

1613
    if (esxVI_LookupVirtualMachineByUuidAndPrepareForTask
1614
          (priv->host, domain->uuid, NULL, &virtualMachine,
1615
           priv->autoAnswer) < 0 ||
1616 1617
        esxVI_VirtualMachineConfigSpec_Alloc(&spec) < 0 ||
        esxVI_Long_Alloc(&spec->memoryMB) < 0) {
M
Matthias Bolte 已提交
1618
        goto cleanup;
1619 1620 1621 1622 1623
    }

    spec->memoryMB->value =
      memory / 1024; /* Scale from kilobytes to megabytes */

1624 1625 1626 1627
    if (esxVI_ReconfigVM_Task(priv->host, virtualMachine->obj, spec,
                              &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
1628
        goto cleanup;
1629 1630 1631
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
1632
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
1633
                  _("Could not set max-memory to %lu kilobytes"), memory);
M
Matthias Bolte 已提交
1634
        goto cleanup;
1635 1636
    }

M
Matthias Bolte 已提交
1637 1638
    result = 0;

1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_VirtualMachineConfigSpec_Free(&spec);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainSetMemory(virDomainPtr domain, unsigned long memory)
{
M
Matthias Bolte 已提交
1652
    int result = -1;
M
Matthias Bolte 已提交
1653
    esxPrivate *priv = domain->conn->privateData;
1654 1655 1656 1657 1658
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachineConfigSpec *spec = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

1659
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1660
        return -1;
1661 1662
    }

1663
    if (esxVI_LookupVirtualMachineByUuidAndPrepareForTask
1664
          (priv->host, domain->uuid, NULL, &virtualMachine,
1665
           priv->autoAnswer) < 0 ||
1666 1667 1668
        esxVI_VirtualMachineConfigSpec_Alloc(&spec) < 0 ||
        esxVI_ResourceAllocationInfo_Alloc(&spec->memoryAllocation) < 0 ||
        esxVI_Long_Alloc(&spec->memoryAllocation->limit) < 0) {
M
Matthias Bolte 已提交
1669
        goto cleanup;
1670 1671 1672 1673 1674
    }

    spec->memoryAllocation->limit->value =
      memory / 1024; /* Scale from kilobytes to megabytes */

1675 1676 1677 1678
    if (esxVI_ReconfigVM_Task(priv->host, virtualMachine->obj, spec,
                              &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
1679
        goto cleanup;
1680 1681 1682
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
1683
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
1684
                  _("Could not set memory to %lu kilobytes"), memory);
M
Matthias Bolte 已提交
1685
        goto cleanup;
1686 1687
    }

M
Matthias Bolte 已提交
1688 1689
    result = 0;

1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_VirtualMachineConfigSpec_Free(&spec);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainGetInfo(virDomainPtr domain, virDomainInfoPtr info)
{
M
Matthias Bolte 已提交
1703
    int result = -1;
M
Matthias Bolte 已提交
1704
    esxPrivate *priv = domain->conn->privateData;
1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    esxVI_VirtualMachinePowerState powerState;
    int64_t memory_limit = -1;
    esxVI_PerfMetricId *perfMetricId = NULL;
    esxVI_PerfMetricId *perfMetricIdList = NULL;
    esxVI_Int *counterId = NULL;
    esxVI_Int *counterIdList = NULL;
    esxVI_PerfCounterInfo *perfCounterInfo = NULL;
    esxVI_PerfCounterInfo *perfCounterInfoList = NULL;
    esxVI_PerfQuerySpec *querySpec = NULL;
1717 1718
    esxVI_PerfEntityMetricBase *perfEntityMetricBase = NULL;
    esxVI_PerfEntityMetricBase *perfEntityMetricBaseList = NULL;
1719 1720 1721 1722
    esxVI_PerfEntityMetric *perfEntityMetric = NULL;
    esxVI_PerfMetricIntSeries *perfMetricIntSeries = NULL;
    esxVI_Long *value = NULL;

M
Matthias Bolte 已提交
1723 1724
    memset(info, 0, sizeof (*info));

1725
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1726
        return -1;
1727 1728
    }

1729
    if (esxVI_String_AppendValueListToList(&propertyNameList,
1730 1731 1732 1733
                                           "runtime.powerState\0"
                                           "config.hardware.memoryMB\0"
                                           "config.hardware.numCPU\0"
                                           "config.memoryAllocation.limit\0") < 0 ||
1734 1735
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
1736
                                         esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
1737
        goto cleanup;
1738 1739 1740 1741 1742 1743 1744 1745
    }

    info->state = VIR_DOMAIN_NOSTATE;

    for (dynamicProperty = virtualMachine->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "runtime.powerState")) {
            if (esxVI_VirtualMachinePowerState_CastFromAnyType
1746
                  (dynamicProperty->val, &powerState) < 0) {
M
Matthias Bolte 已提交
1747
                goto cleanup;
1748 1749
            }

1750 1751
            info->state = esxVI_VirtualMachinePowerState_ConvertToLibvirt
                            (powerState);
1752
        } else if (STREQ(dynamicProperty->name, "config.hardware.memoryMB")) {
1753
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
1754
                                         esxVI_Type_Int) < 0) {
M
Matthias Bolte 已提交
1755
                goto cleanup;
1756 1757 1758 1759
            }

            info->maxMem = dynamicProperty->val->int32 * 1024; /* Scale from megabyte to kilobyte */
        } else if (STREQ(dynamicProperty->name, "config.hardware.numCPU")) {
1760
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
1761
                                         esxVI_Type_Int) < 0) {
M
Matthias Bolte 已提交
1762
                goto cleanup;
1763 1764 1765 1766 1767
            }

            info->nrVirtCpu = dynamicProperty->val->int32;
        } else if (STREQ(dynamicProperty->name,
                         "config.memoryAllocation.limit")) {
1768
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
1769
                                         esxVI_Type_Long) < 0) {
M
Matthias Bolte 已提交
1770
                goto cleanup;
1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787
            }

            memory_limit = dynamicProperty->val->int64;

            if (memory_limit > 0) {
                memory_limit *= 1024; /* Scale from megabyte to kilobyte */
            }
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

    /* memory_limit < 0 means no memory limit is set */
    info->memory = memory_limit < 0 ? info->maxMem : memory_limit;

    /* Verify the cached 'used CPU time' performance counter ID */
    if (info->state == VIR_DOMAIN_RUNNING && priv->usedCpuTimeCounterId >= 0) {
1788
        if (esxVI_Int_Alloc(&counterId) < 0) {
M
Matthias Bolte 已提交
1789
            goto cleanup;
1790 1791 1792 1793
        }

        counterId->value = priv->usedCpuTimeCounterId;

1794
        if (esxVI_Int_AppendToList(&counterIdList, counterId) < 0) {
M
Matthias Bolte 已提交
1795
            goto cleanup;
1796 1797
        }

1798
        if (esxVI_QueryPerfCounter(priv->host, counterIdList,
1799
                                   &perfCounterInfo) < 0) {
M
Matthias Bolte 已提交
1800
            goto cleanup;
1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820
        }

        if (STRNEQ(perfCounterInfo->groupInfo->key, "cpu") ||
            STRNEQ(perfCounterInfo->nameInfo->key, "used") ||
            STRNEQ(perfCounterInfo->unitInfo->key, "millisecond")) {
            VIR_DEBUG("Cached usedCpuTimeCounterId %d is invalid",
                      priv->usedCpuTimeCounterId);

            priv->usedCpuTimeCounterId = -1;
        }

        esxVI_Int_Free(&counterIdList);
        esxVI_PerfCounterInfo_Free(&perfCounterInfo);
    }

    /*
     * Query the PerformanceManager for the 'used CPU time' performance
     * counter ID and cache it, if it's not already cached.
     */
    if (info->state == VIR_DOMAIN_RUNNING && priv->usedCpuTimeCounterId < 0) {
1821 1822 1823
        if (esxVI_QueryAvailablePerfMetric(priv->host, virtualMachine->obj,
                                           NULL, NULL, NULL,
                                           &perfMetricIdList) < 0) {
M
Matthias Bolte 已提交
1824
            goto cleanup;
1825 1826 1827 1828 1829 1830 1831 1832 1833
        }

        for (perfMetricId = perfMetricIdList; perfMetricId != NULL;
             perfMetricId = perfMetricId->_next) {
            VIR_DEBUG("perfMetricId counterId %d, instance '%s'",
                      perfMetricId->counterId->value, perfMetricId->instance);

            counterId = NULL;

1834 1835
            if (esxVI_Int_DeepCopy(&counterId, perfMetricId->counterId) < 0 ||
                esxVI_Int_AppendToList(&counterIdList, counterId) < 0) {
M
Matthias Bolte 已提交
1836
                goto cleanup;
1837 1838 1839
            }
        }

1840
        if (esxVI_QueryPerfCounter(priv->host, counterIdList,
1841
                                   &perfCounterInfoList) < 0) {
M
Matthias Bolte 已提交
1842
            goto cleanup;
1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869 1870 1871 1872 1873 1874 1875
        }

        for (perfCounterInfo = perfCounterInfoList; perfCounterInfo != NULL;
             perfCounterInfo = perfCounterInfo->_next) {
            VIR_DEBUG("perfCounterInfo key %d, nameInfo '%s', groupInfo '%s', "
                      "unitInfo '%s', rollupType %d, statsType %d",
                      perfCounterInfo->key->value,
                      perfCounterInfo->nameInfo->key,
                      perfCounterInfo->groupInfo->key,
                      perfCounterInfo->unitInfo->key,
                      perfCounterInfo->rollupType,
                      perfCounterInfo->statsType);

            if (STREQ(perfCounterInfo->groupInfo->key, "cpu") &&
                STREQ(perfCounterInfo->nameInfo->key, "used") &&
                STREQ(perfCounterInfo->unitInfo->key, "millisecond")) {
                priv->usedCpuTimeCounterId = perfCounterInfo->key->value;
                break;
            }
        }

        if (priv->usedCpuTimeCounterId < 0) {
            VIR_WARN0("Could not find 'used CPU time' performance counter");
        }
    }

    /*
     * Query the PerformanceManager for the 'used CPU time' performance
     * counter value.
     */
    if (info->state == VIR_DOMAIN_RUNNING && priv->usedCpuTimeCounterId >= 0) {
        VIR_DEBUG("usedCpuTimeCounterId %d BEGIN", priv->usedCpuTimeCounterId);

1876 1877 1878 1879
        if (esxVI_PerfQuerySpec_Alloc(&querySpec) < 0 ||
            esxVI_Int_Alloc(&querySpec->maxSample) < 0 ||
            esxVI_PerfMetricId_Alloc(&querySpec->metricId) < 0 ||
            esxVI_Int_Alloc(&querySpec->metricId->counterId) < 0) {
M
Matthias Bolte 已提交
1880
            goto cleanup;
1881 1882 1883 1884 1885 1886 1887 1888
        }

        querySpec->entity = virtualMachine->obj;
        querySpec->maxSample->value = 1;
        querySpec->metricId->counterId->value = priv->usedCpuTimeCounterId;
        querySpec->metricId->instance = (char *)"";
        querySpec->format = (char *)"normal";

1889 1890
        if (esxVI_QueryPerf(priv->host, querySpec,
                            &perfEntityMetricBaseList) < 0) {
1891 1892 1893
            querySpec->entity = NULL;
            querySpec->metricId->instance = NULL;
            querySpec->format = NULL;
M
Matthias Bolte 已提交
1894
            goto cleanup;
1895 1896
        }

1897 1898 1899
        for (perfEntityMetricBase = perfEntityMetricBaseList;
             perfEntityMetricBase != NULL;
             perfEntityMetricBase = perfEntityMetricBase->_next) {
1900 1901
            VIR_DEBUG0("perfEntityMetric ...");

1902 1903 1904 1905
            perfEntityMetric =
              esxVI_PerfEntityMetric_DynamicCast(perfEntityMetricBase);

            if (perfMetricIntSeries == NULL) {
1906
                VIR_ERROR0(_("QueryPerf returned object with unexpected type"));
1907 1908
            }

1909 1910 1911 1912
            perfMetricIntSeries =
              esxVI_PerfMetricIntSeries_DynamicCast(perfEntityMetric->value);

            if (perfMetricIntSeries == NULL) {
1913
                VIR_ERROR0(_("QueryPerf returned object with unexpected type"));
1914 1915 1916
            }

            for (; perfMetricIntSeries != NULL;
1917 1918 1919 1920 1921 1922
                 perfMetricIntSeries = perfMetricIntSeries->_next) {
                VIR_DEBUG0("perfMetricIntSeries ...");

                for (value = perfMetricIntSeries->value;
                     value != NULL;
                     value = value->_next) {
1923
                    VIR_DEBUG("value %lld", (long long int)value->value);
1924 1925 1926 1927 1928 1929 1930 1931 1932
                }
            }
        }

        querySpec->entity = NULL;
        querySpec->metricId->instance = NULL;
        querySpec->format = NULL;

        VIR_DEBUG("usedCpuTimeCounterId %d END", priv->usedCpuTimeCounterId);
M
Matthias Bolte 已提交
1933 1934 1935 1936 1937

        /*
         * FIXME: Cannot map between realtive used-cpu-time and absolute
         *        info->cpuTime
         */
1938 1939
    }

M
Matthias Bolte 已提交
1940 1941
    result = 0;

1942 1943 1944 1945 1946 1947 1948
  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_PerfMetricId_Free(&perfMetricIdList);
    esxVI_Int_Free(&counterIdList);
    esxVI_PerfCounterInfo_Free(&perfCounterInfoList);
    esxVI_PerfQuerySpec_Free(&querySpec);
1949
    esxVI_PerfEntityMetricBase_Free(&perfEntityMetricBaseList);
1950 1951 1952 1953 1954 1955 1956 1957 1958

    return result;
}



static int
esxDomainSetVcpus(virDomainPtr domain, unsigned int nvcpus)
{
M
Matthias Bolte 已提交
1959
    int result = -1;
M
Matthias Bolte 已提交
1960
    esxPrivate *priv = domain->conn->privateData;
M
Matthias Bolte 已提交
1961
    int maxVcpus;
1962 1963 1964 1965 1966 1967
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachineConfigSpec *spec = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

    if (nvcpus < 1) {
1968 1969
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s",
                  _("Requested number of virtual CPUs must at least be 1"));
M
Matthias Bolte 已提交
1970
        return -1;
1971 1972
    }

1973
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
1974
        return -1;
1975 1976
    }

M
Matthias Bolte 已提交
1977
    maxVcpus = esxDomainGetMaxVcpus(domain);
1978

M
Matthias Bolte 已提交
1979
    if (maxVcpus < 0) {
M
Matthias Bolte 已提交
1980
        return -1;
1981 1982
    }

M
Matthias Bolte 已提交
1983
    if (nvcpus > maxVcpus) {
1984
        ESX_ERROR(VIR_ERR_INVALID_ARG,
1985 1986
                  _("Requested number of virtual CPUs is greater than max "
                    "allowable number of virtual CPUs for the domain: %d > %d"),
M
Matthias Bolte 已提交
1987
                  nvcpus, maxVcpus);
M
Matthias Bolte 已提交
1988
        return -1;
1989 1990
    }

1991
    if (esxVI_LookupVirtualMachineByUuidAndPrepareForTask
1992
          (priv->host, domain->uuid, NULL, &virtualMachine,
1993
           priv->autoAnswer) < 0 ||
1994 1995
        esxVI_VirtualMachineConfigSpec_Alloc(&spec) < 0 ||
        esxVI_Int_Alloc(&spec->numCPUs) < 0) {
M
Matthias Bolte 已提交
1996
        goto cleanup;
1997 1998 1999 2000
    }

    spec->numCPUs->value = nvcpus;

2001 2002 2003 2004
    if (esxVI_ReconfigVM_Task(priv->host, virtualMachine->obj, spec,
                              &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
2005
        goto cleanup;
2006 2007 2008
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
2009
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
2010
                  _("Could not set number of virtual CPUs to %d"), nvcpus);
M
Matthias Bolte 已提交
2011
        goto cleanup;
2012 2013
    }

M
Matthias Bolte 已提交
2014 2015
    result = 0;

2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_VirtualMachineConfigSpec_Free(&spec);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainGetMaxVcpus(virDomainPtr domain)
{
M
Matthias Bolte 已提交
2029
    esxPrivate *priv = domain->conn->privateData;
2030 2031 2032 2033
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;

M
Matthias Bolte 已提交
2034 2035
    if (priv->maxVcpus > 0) {
        return priv->maxVcpus;
2036 2037
    }

M
Matthias Bolte 已提交
2038 2039
    priv->maxVcpus = -1;

2040
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2041
        return -1;
2042 2043
    }

2044
    if (esxVI_String_AppendValueToList(&propertyNameList,
2045
                                       "capability.maxSupportedVcpus") < 0 ||
2046 2047 2048
        esxVI_LookupObjectContentByType(priv->host, priv->host->hostFolder,
                                        "HostSystem", propertyNameList,
                                        esxVI_Boolean_True, &hostSystem) < 0) {
M
Matthias Bolte 已提交
2049
        goto cleanup;
2050 2051 2052
    }

    if (hostSystem == NULL) {
2053 2054
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve the HostSystem object"));
M
Matthias Bolte 已提交
2055
        goto cleanup;
2056 2057 2058 2059 2060
    }

    for (dynamicProperty = hostSystem->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "capability.maxSupportedVcpus")) {
2061
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
2062
                                         esxVI_Type_Int) < 0) {
M
Matthias Bolte 已提交
2063
                goto cleanup;
2064 2065
            }

M
Matthias Bolte 已提交
2066
            priv->maxVcpus = dynamicProperty->val->int32;
2067 2068 2069 2070 2071 2072 2073 2074 2075 2076
            break;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);

M
Matthias Bolte 已提交
2077
    return priv->maxVcpus;
2078 2079 2080 2081 2082 2083 2084
}



static char *
esxDomainDumpXML(virDomainPtr domain, int flags)
{
M
Matthias Bolte 已提交
2085
    esxPrivate *priv = domain->conn->privateData;
2086 2087 2088 2089 2090
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    const char *vmPathName = NULL;
    char *datastoreName = NULL;
M
Matthias Bolte 已提交
2091 2092
    char *directoryName = NULL;
    char *fileName = NULL;
2093
    virBuffer buffer = VIR_BUFFER_INITIALIZER;
2094 2095 2096 2097 2098
    char *url = NULL;
    char *vmx = NULL;
    virDomainDefPtr def = NULL;
    char *xml = NULL;

2099
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2100
        return NULL;
2101 2102
    }

2103
    if (esxVI_String_AppendValueToList(&propertyNameList,
2104
                                       "config.files.vmPathName") < 0 ||
2105 2106
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
2107
                                         esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
2108
        goto cleanup;
2109 2110 2111 2112 2113
    }

    for (dynamicProperty = virtualMachine->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "config.files.vmPathName")) {
2114
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
2115
                                         esxVI_Type_String) < 0) {
M
Matthias Bolte 已提交
2116
                goto cleanup;
2117 2118 2119 2120 2121 2122 2123
            }

            vmPathName = dynamicProperty->val->string;
            break;
        }
    }

2124 2125
    if (esxUtil_ParseDatastoreRelatedPath(vmPathName, &datastoreName,
                                          &directoryName, &fileName) < 0) {
M
Matthias Bolte 已提交
2126
        goto cleanup;
2127 2128
    }

2129 2130
    virBufferVSprintf(&buffer, "%s://%s:%d/folder/", priv->transport,
                      domain->conn->uri->server, domain->conn->uri->port);
M
Matthias Bolte 已提交
2131 2132 2133 2134 2135 2136 2137

    if (directoryName != NULL) {
        virBufferURIEncodeString(&buffer, directoryName);
        virBufferAddChar(&buffer, '/');
    }

    virBufferURIEncodeString(&buffer, fileName);
2138 2139 2140 2141 2142 2143
    virBufferAddLit(&buffer, "?dcPath=");
    virBufferURIEncodeString(&buffer, priv->host->datacenter->value);
    virBufferAddLit(&buffer, "&dsName=");
    virBufferURIEncodeString(&buffer, datastoreName);

    if (virBufferError(&buffer)) {
2144
        virReportOOMError();
M
Matthias Bolte 已提交
2145
        goto cleanup;
2146 2147
    }

2148 2149
    url = virBufferContentAndReset(&buffer);

2150
    if (esxVI_Context_DownloadFile(priv->host, url, &vmx) < 0) {
M
Matthias Bolte 已提交
2151
        goto cleanup;
2152 2153
    }

2154
    def = esxVMX_ParseConfig(priv->host, vmx, datastoreName, directoryName,
2155
                             priv->host->productVersion);
2156 2157

    if (def != NULL) {
2158
        xml = virDomainDefFormat(def, flags);
2159 2160 2161
    }

  cleanup:
M
Matthias Bolte 已提交
2162 2163 2164 2165
    if (url == NULL) {
        virBufferFreeAndReset(&buffer);
    }

2166 2167
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachine);
2168
    VIR_FREE(datastoreName);
M
Matthias Bolte 已提交
2169 2170
    VIR_FREE(directoryName);
    VIR_FREE(fileName);
2171 2172
    VIR_FREE(url);
    VIR_FREE(vmx);
2173
    virDomainDefFree(def);
2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184

    return xml;
}



static char *
esxDomainXMLFromNative(virConnectPtr conn, const char *nativeFormat,
                       const char *nativeConfig,
                       unsigned int flags ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
2185
    esxPrivate *priv = conn->privateData;
2186 2187 2188 2189
    virDomainDefPtr def = NULL;
    char *xml = NULL;

    if (STRNEQ(nativeFormat, "vmware-vmx")) {
2190
        ESX_ERROR(VIR_ERR_INVALID_ARG,
2191
                  _("Unsupported config format '%s'"), nativeFormat);
2192
        return NULL;
2193 2194
    }

2195
    def = esxVMX_ParseConfig(priv->host, nativeConfig, "?", "?",
2196
                             priv->host->productVersion);
2197 2198

    if (def != NULL) {
2199
        xml = virDomainDefFormat(def, VIR_DOMAIN_XML_INACTIVE);
2200 2201 2202 2203 2204 2205 2206 2207 2208
    }

    virDomainDefFree(def);

    return xml;
}



M
Matthias Bolte 已提交
2209 2210 2211 2212 2213
static char *
esxDomainXMLToNative(virConnectPtr conn, const char *nativeFormat,
                     const char *domainXml,
                     unsigned int flags ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
2214
    esxPrivate *priv = conn->privateData;
M
Matthias Bolte 已提交
2215 2216 2217 2218
    virDomainDefPtr def = NULL;
    char *vmx = NULL;

    if (STRNEQ(nativeFormat, "vmware-vmx")) {
2219
        ESX_ERROR(VIR_ERR_INVALID_ARG,
2220
                  _("Unsupported config format '%s'"), nativeFormat);
M
Matthias Bolte 已提交
2221 2222 2223
        return NULL;
    }

2224
    def = virDomainDefParseString(priv->caps, domainXml, 0);
M
Matthias Bolte 已提交
2225 2226 2227 2228 2229

    if (def == NULL) {
        return NULL;
    }

2230
    vmx = esxVMX_FormatConfig(priv->host, def, priv->host->productVersion);
M
Matthias Bolte 已提交
2231 2232 2233 2234 2235 2236 2237 2238

    virDomainDefFree(def);

    return vmx;
}



2239 2240 2241
static int
esxListDefinedDomains(virConnectPtr conn, char **const names, int maxnames)
{
M
Matthias Bolte 已提交
2242
    bool success = false;
M
Matthias Bolte 已提交
2243
    esxPrivate *priv = conn->privateData;
2244 2245 2246 2247 2248 2249
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachineList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    esxVI_VirtualMachinePowerState powerState;
    int count = 0;
2250
    int i;
2251 2252

    if (names == NULL || maxnames < 0) {
2253 2254
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s", _("Invalid argument"));
        return -1;
2255 2256 2257 2258 2259 2260
    }

    if (maxnames == 0) {
        return 0;
    }

2261
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2262
        return -1;
2263 2264
    }

2265
    if (esxVI_String_AppendValueListToList(&propertyNameList,
2266 2267
                                           "name\0"
                                           "runtime.powerState\0") < 0 ||
2268
        esxVI_LookupObjectContentByType(priv->host, priv->host->vmFolder,
2269 2270 2271
                                        "VirtualMachine", propertyNameList,
                                        esxVI_Boolean_True,
                                        &virtualMachineList) < 0) {
M
Matthias Bolte 已提交
2272
        goto cleanup;
2273 2274 2275 2276
    }

    for (virtualMachine = virtualMachineList; virtualMachine != NULL;
         virtualMachine = virtualMachine->_next) {
2277
        if (esxVI_GetVirtualMachinePowerState(virtualMachine,
2278
                                              &powerState) < 0) {
M
Matthias Bolte 已提交
2279
            goto cleanup;
2280 2281 2282 2283 2284 2285 2286 2287 2288 2289
        }

        if (powerState == esxVI_VirtualMachinePowerState_PoweredOn) {
            continue;
        }

        for (dynamicProperty = virtualMachine->propSet;
             dynamicProperty != NULL;
             dynamicProperty = dynamicProperty->_next) {
            if (STREQ(dynamicProperty->name, "name")) {
2290
                if (esxVI_AnyType_ExpectType(dynamicProperty->val,
2291
                                             esxVI_Type_String) < 0) {
M
Matthias Bolte 已提交
2292
                    goto cleanup;
2293 2294 2295 2296 2297
                }

                names[count] = strdup(dynamicProperty->val->string);

                if (names[count] == NULL) {
2298
                    virReportOOMError();
M
Matthias Bolte 已提交
2299
                    goto cleanup;
2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311
                }

                count++;
                break;
            }
        }

        if (count >= maxnames) {
            break;
        }
    }

M
Matthias Bolte 已提交
2312
    success = true;
2313

M
Matthias Bolte 已提交
2314 2315 2316 2317 2318
  cleanup:
    if (! success) {
        for (i = 0; i < count; ++i) {
            VIR_FREE(names[i]);
        }
2319

M
Matthias Bolte 已提交
2320
        count = -1;
2321 2322
    }

M
Matthias Bolte 已提交
2323 2324
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachineList);
2325

M
Matthias Bolte 已提交
2326
    return count;
2327 2328 2329 2330 2331 2332 2333
}



static int
esxNumberOfDefinedDomains(virConnectPtr conn)
{
M
Matthias Bolte 已提交
2334
    esxPrivate *priv = conn->privateData;
2335

2336
    if (esxVI_EnsureSession(priv->host) < 0) {
2337 2338 2339
        return -1;
    }

2340
    return esxVI_LookupNumberOfDomainsByPowerState
2341
             (priv->host, esxVI_VirtualMachinePowerState_PoweredOn,
2342 2343 2344 2345 2346 2347 2348 2349
              esxVI_Boolean_True);
}



static int
esxDomainCreate(virDomainPtr domain)
{
M
Matthias Bolte 已提交
2350
    int result = -1;
M
Matthias Bolte 已提交
2351
    esxPrivate *priv = domain->conn->privateData;
2352 2353 2354 2355 2356 2357
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

2358
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2359
        return -1;
2360 2361
    }

2362
    if (esxVI_String_AppendValueToList(&propertyNameList,
2363
                                       "runtime.powerState") < 0 ||
2364
        esxVI_LookupVirtualMachineByUuidAndPrepareForTask
2365 2366 2367
          (priv->host, domain->uuid, propertyNameList, &virtualMachine,
           priv->autoAnswer) < 0 ||
        esxVI_GetVirtualMachinePowerState(virtualMachine,
2368
                                          &powerState) < 0) {
M
Matthias Bolte 已提交
2369
        goto cleanup;
2370 2371 2372
    }

    if (powerState != esxVI_VirtualMachinePowerState_PoweredOff) {
2373 2374
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s",
                  _("Domain is not powered off"));
M
Matthias Bolte 已提交
2375
        goto cleanup;
2376 2377
    }

2378 2379
    if (esxVI_PowerOnVM_Task(priv->host, virtualMachine->obj, NULL,
                             &task) < 0 ||
2380 2381
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
2382
        goto cleanup;
2383 2384 2385
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
2386
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s", _("Could not start domain"));
M
Matthias Bolte 已提交
2387
        goto cleanup;
2388 2389
    }

M
Matthias Bolte 已提交
2390 2391
    result = 0;

2392 2393 2394 2395 2396 2397 2398 2399 2400 2401
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



M
Matthias Bolte 已提交
2402 2403 2404
static virDomainPtr
esxDomainDefineXML(virConnectPtr conn, const char *xml ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
2405
    esxPrivate *priv = conn->privateData;
M
Matthias Bolte 已提交
2406 2407
    virDomainDefPtr def = NULL;
    char *vmx = NULL;
2408 2409
    int i;
    virDomainDiskDefPtr disk = NULL;
M
Matthias Bolte 已提交
2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423
    esxVI_ObjectContent *virtualMachine = NULL;
    char *datastoreName = NULL;
    char *directoryName = NULL;
    char *fileName = NULL;
    virBuffer buffer = VIR_BUFFER_INITIALIZER;
    char *url = NULL;
    char *datastoreRelatedPath = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_ManagedObjectReference *resourcePool = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;
    virDomainPtr domain = NULL;

2424
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2425
        return NULL;
M
Matthias Bolte 已提交
2426 2427 2428
    }

    /* Parse domain XML */
2429
    def = virDomainDefParseString(priv->caps, xml,
M
Matthias Bolte 已提交
2430 2431 2432
                                  VIR_DOMAIN_XML_INACTIVE);

    if (def == NULL) {
M
Matthias Bolte 已提交
2433
        return NULL;
M
Matthias Bolte 已提交
2434 2435 2436
    }

    /* Check if an existing domain should be edited */
2437
    if (esxVI_LookupVirtualMachineByUuid(priv->host, def->uuid, NULL,
M
Matthias Bolte 已提交
2438
                                         &virtualMachine,
M
Matthias Bolte 已提交
2439
                                         esxVI_Occurrence_OptionalItem) < 0) {
M
Matthias Bolte 已提交
2440
        goto cleanup;
M
Matthias Bolte 已提交
2441 2442 2443 2444
    }

    if (virtualMachine != NULL) {
        /* FIXME */
2445 2446 2447
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Domain already exists, editing existing domains is not "
                    "supported yet"));
M
Matthias Bolte 已提交
2448
        goto cleanup;
M
Matthias Bolte 已提交
2449 2450 2451
    }

    /* Build VMX from domain XML */
2452
    vmx = esxVMX_FormatConfig(priv->host, def, priv->host->productVersion);
M
Matthias Bolte 已提交
2453 2454

    if (vmx == NULL) {
M
Matthias Bolte 已提交
2455
        goto cleanup;
M
Matthias Bolte 已提交
2456 2457
    }

2458 2459 2460 2461 2462 2463 2464
    /*
     * Build VMX datastore URL. Use the source of the first file-based harddisk
     * to deduce the datastore and path for the VMX file. Don't just use the
     * first disk, because it may be CDROM disk and ISO images are normaly not
     * located in the virtual machine's directory. This approach to deduce the
     * datastore isn't perfect but should work in the majority of cases.
     */
M
Matthias Bolte 已提交
2465
    if (def->ndisks < 1) {
2466 2467 2468
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Domain XML doesn't contain any disks, cannot deduce "
                    "datastore and path for VMX file"));
M
Matthias Bolte 已提交
2469
        goto cleanup;
2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480
    }

    for (i = 0; i < def->ndisks; ++i) {
        if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK &&
            def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_FILE) {
            disk = def->disks[i];
            break;
        }
    }

    if (disk == NULL) {
2481 2482 2483
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Domain XML doesn't contain any file-based harddisks, "
                    "cannot deduce datastore and path for VMX file"));
M
Matthias Bolte 已提交
2484
        goto cleanup;
M
Matthias Bolte 已提交
2485 2486
    }

2487
    if (disk->src == NULL) {
2488 2489 2490
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("First file-based harddisk has no source, cannot deduce "
                    "datastore and path for VMX file"));
M
Matthias Bolte 已提交
2491
        goto cleanup;
M
Matthias Bolte 已提交
2492 2493
    }

2494
    if (esxUtil_ParseDatastoreRelatedPath(disk->src, &datastoreName,
2495
                                          &directoryName, &fileName) < 0) {
M
Matthias Bolte 已提交
2496
        goto cleanup;
M
Matthias Bolte 已提交
2497 2498
    }

2499
    if (! virFileHasSuffix(fileName, ".vmdk")) {
2500
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
2501 2502
                  _("Expecting source '%s' of first file-based harddisk to "
                    "be a VMDK image"), disk->src);
M
Matthias Bolte 已提交
2503
        goto cleanup;
M
Matthias Bolte 已提交
2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520
    }

    virBufferVSprintf(&buffer, "%s://%s:%d/folder/", priv->transport,
                      conn->uri->server, conn->uri->port);

    if (directoryName != NULL) {
        virBufferURIEncodeString(&buffer, directoryName);
        virBufferAddChar(&buffer, '/');
    }

    virBufferURIEncodeString(&buffer, def->name);
    virBufferAddLit(&buffer, ".vmx?dcPath=");
    virBufferURIEncodeString(&buffer, priv->host->datacenter->value);
    virBufferAddLit(&buffer, "&dsName=");
    virBufferURIEncodeString(&buffer, datastoreName);

    if (virBufferError(&buffer)) {
2521
        virReportOOMError();
M
Matthias Bolte 已提交
2522
        goto cleanup;
M
Matthias Bolte 已提交
2523 2524 2525 2526 2527 2528 2529
    }

    url = virBufferContentAndReset(&buffer);

    if (directoryName != NULL) {
        if (virAsprintf(&datastoreRelatedPath, "[%s] %s/%s.vmx", datastoreName,
                        directoryName, def->name) < 0) {
2530
            virReportOOMError();
M
Matthias Bolte 已提交
2531
            goto cleanup;
M
Matthias Bolte 已提交
2532 2533 2534 2535
        }
    } else {
        if (virAsprintf(&datastoreRelatedPath, "[%s] %s.vmx", datastoreName,
                        def->name) < 0) {
2536
            virReportOOMError();
M
Matthias Bolte 已提交
2537
            goto cleanup;
M
Matthias Bolte 已提交
2538 2539 2540 2541
        }
    }

    /* Get resource pool */
2542 2543
    if (esxVI_String_AppendValueToList(&propertyNameList, "parent") < 0 ||
        esxVI_LookupHostSystemByIp(priv->host, priv->host->ipAddress,
M
Matthias Bolte 已提交
2544
                                   propertyNameList, &hostSystem) < 0) {
M
Matthias Bolte 已提交
2545
        goto cleanup;
M
Matthias Bolte 已提交
2546 2547
    }

2548
    if (esxVI_LookupResourcePoolByHostSystem(priv->host, hostSystem,
2549
                                             &resourcePool) < 0) {
M
Matthias Bolte 已提交
2550
        goto cleanup;
M
Matthias Bolte 已提交
2551 2552 2553 2554 2555 2556
    }

    /* Check, if VMX file already exists */
    /* FIXME */

    /* Upload VMX file */
2557
    if (esxVI_Context_UploadFile(priv->host, url, vmx) < 0) {
M
Matthias Bolte 已提交
2558
        goto cleanup;
M
Matthias Bolte 已提交
2559 2560 2561
    }

    /* Register the domain */
2562
    if (esxVI_RegisterVM_Task(priv->host, priv->host->vmFolder,
M
Matthias Bolte 已提交
2563 2564
                              datastoreRelatedPath, NULL, esxVI_Boolean_False,
                              resourcePool, hostSystem->obj, &task) < 0 ||
2565 2566
        esxVI_WaitForTaskCompletion(priv->host, task, def->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
2567
        goto cleanup;
M
Matthias Bolte 已提交
2568 2569 2570
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
2571
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s", _("Could not define domain"));
M
Matthias Bolte 已提交
2572
        goto cleanup;
M
Matthias Bolte 已提交
2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583
    }

    domain = virGetDomain(conn, def->name, def->uuid);

    if (domain != NULL) {
        domain->id = -1;
    }

    /* FIXME: Add proper rollback in case of an error */

  cleanup:
M
Matthias Bolte 已提交
2584 2585 2586 2587
    if (url == NULL) {
        virBufferFreeAndReset(&buffer);
    }

M
Matthias Bolte 已提交
2588 2589 2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605
    virDomainDefFree(def);
    VIR_FREE(vmx);
    VIR_FREE(datastoreName);
    VIR_FREE(directoryName);
    VIR_FREE(fileName);
    VIR_FREE(url);
    VIR_FREE(datastoreRelatedPath);
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);
    esxVI_ManagedObjectReference_Free(&resourcePool);
    esxVI_ManagedObjectReference_Free(&task);

    return domain;
}



2606 2607 2608
static int
esxDomainUndefine(virDomainPtr domain)
{
M
Matthias Bolte 已提交
2609
    int result = -1;
M
Matthias Bolte 已提交
2610
    esxPrivate *priv = domain->conn->privateData;
2611
    esxVI_Context *ctx = NULL;
2612 2613 2614 2615
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;

2616 2617 2618 2619 2620 2621
    if (priv->vCenter != NULL) {
        ctx = priv->vCenter;
    } else {
        ctx = priv->host;
    }

2622
    if (esxVI_EnsureSession(ctx) < 0) {
M
Matthias Bolte 已提交
2623
        return -1;
2624 2625
    }

2626
    if (esxVI_String_AppendValueToList(&propertyNameList,
2627
                                       "runtime.powerState") < 0 ||
2628 2629
        esxVI_LookupVirtualMachineByUuid(ctx, domain->uuid, propertyNameList,
                                         &virtualMachine,
M
Matthias Bolte 已提交
2630
                                         esxVI_Occurrence_RequiredItem) < 0 ||
2631
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
2632
        goto cleanup;
2633 2634 2635 2636
    }

    if (powerState != esxVI_VirtualMachinePowerState_Suspended &&
        powerState != esxVI_VirtualMachinePowerState_PoweredOff) {
2637 2638
        ESX_ERROR(VIR_ERR_OPERATION_INVALID, "%s",
                  _("Domain is not suspended or powered off"));
M
Matthias Bolte 已提交
2639
        goto cleanup;
2640 2641
    }

2642
    if (esxVI_UnregisterVM(ctx, virtualMachine->obj) < 0) {
M
Matthias Bolte 已提交
2643
        goto cleanup;
2644 2645
    }

M
Matthias Bolte 已提交
2646 2647
    result = 0;

2648 2649 2650 2651 2652 2653 2654 2655 2656
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);

    return result;
}



2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668
/*
 * The scheduler interface exposes basically the CPU ResourceAllocationInfo:
 *
 * - http://www.vmware.com/support/developer/vc-sdk/visdk25pubs/ReferenceGuide/vim.ResourceAllocationInfo.html
 * - http://www.vmware.com/support/developer/vc-sdk/visdk25pubs/ReferenceGuide/vim.SharesInfo.html
 * - http://www.vmware.com/support/developer/vc-sdk/visdk25pubs/ReferenceGuide/vim.SharesInfo.Level.html
 *
 *
 * Available parameters:
 *
 * - reservation (VIR_DOMAIN_SCHED_FIELD_LLONG >= 0, in megaherz)
 *
2669
 *   The amount of CPU resource that is guaranteed to be available to the domain.
2670 2671 2672 2673
 *
 *
 * - limit (VIR_DOMAIN_SCHED_FIELD_LLONG >= 0, or -1, in megaherz)
 *
2674 2675
 *   The CPU utilization of the domain will be limited to this value, even if
 *   more CPU resources are available. If the limit is set to -1, the CPU
2676 2677 2678 2679 2680 2681 2682 2683 2684 2685 2686
 *   utilization of the domain is unlimited. If the limit is not set to -1, it
 *   must be greater than or equal to the reservation.
 *
 *
 * - shares (VIR_DOMAIN_SCHED_FIELD_INT >= 0, or in {-1, -2, -3}, no unit)
 *
 *   Shares are used to determine relative CPU allocation between domains. In
 *   general, a domain with more shares gets proportionally more of the CPU
 *   resource. The special values -1, -2 and -3 represent the predefined
 *   SharesLevel 'low', 'normal' and 'high'.
 */
2687
static char *
2688
esxDomainGetSchedulerType(virDomainPtr domain ATTRIBUTE_UNUSED, int *nparams)
2689 2690 2691 2692
{
    char *type = strdup("allocation");

    if (type == NULL) {
2693
        virReportOOMError();
2694
        return NULL;
2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707
    }

    *nparams = 3; /* reservation, limit, shares */

    return type;
}



static int
esxDomainGetSchedulerParameters(virDomainPtr domain,
                                virSchedParameterPtr params, int *nparams)
{
M
Matthias Bolte 已提交
2708
    int result = -1;
M
Matthias Bolte 已提交
2709
    esxPrivate *priv = domain->conn->privateData;
2710 2711 2712 2713 2714 2715 2716 2717
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    esxVI_SharesInfo *sharesInfo = NULL;
    unsigned int mask = 0;
    int i = 0;

    if (*nparams < 3) {
2718 2719
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s",
                  _("Parameter array must have space for 3 items"));
M
Matthias Bolte 已提交
2720
        return -1;
2721 2722
    }

2723
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2724
        return -1;
2725 2726
    }

2727
    if (esxVI_String_AppendValueListToList(&propertyNameList,
2728 2729 2730
                                           "config.cpuAllocation.reservation\0"
                                           "config.cpuAllocation.limit\0"
                                           "config.cpuAllocation.shares\0") < 0 ||
2731 2732
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
2733
                                         esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
2734
        goto cleanup;
2735 2736 2737 2738 2739 2740
    }

    for (dynamicProperty = virtualMachine->propSet;
         dynamicProperty != NULL && mask != 7 && i < 3;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "config.cpuAllocation.reservation") &&
M
Matthias Bolte 已提交
2741
            ! (mask & (1 << 0))) {
2742 2743 2744 2745 2746
            snprintf (params[i].field, VIR_DOMAIN_SCHED_FIELD_LENGTH, "%s",
                      "reservation");

            params[i].type = VIR_DOMAIN_SCHED_FIELD_LLONG;

2747
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
2748
                                         esxVI_Type_Long) < 0) {
M
Matthias Bolte 已提交
2749
                goto cleanup;
2750 2751 2752 2753 2754 2755 2756
            }

            params[i].value.l = dynamicProperty->val->int64;
            mask |= 1 << 0;
            ++i;
        } else if (STREQ(dynamicProperty->name,
                         "config.cpuAllocation.limit") &&
M
Matthias Bolte 已提交
2757
                   ! (mask & (1 << 1))) {
2758 2759 2760 2761 2762
            snprintf (params[i].field, VIR_DOMAIN_SCHED_FIELD_LENGTH, "%s",
                      "limit");

            params[i].type = VIR_DOMAIN_SCHED_FIELD_LLONG;

2763
            if (esxVI_AnyType_ExpectType(dynamicProperty->val,
2764
                                         esxVI_Type_Long) < 0) {
M
Matthias Bolte 已提交
2765
                goto cleanup;
2766 2767 2768 2769 2770 2771 2772
            }

            params[i].value.l = dynamicProperty->val->int64;
            mask |= 1 << 1;
            ++i;
        } else if (STREQ(dynamicProperty->name,
                         "config.cpuAllocation.shares") &&
M
Matthias Bolte 已提交
2773
                   ! (mask & (1 << 2))) {
2774 2775 2776 2777 2778
            snprintf (params[i].field, VIR_DOMAIN_SCHED_FIELD_LENGTH, "%s",
                      "shares");

            params[i].type = VIR_DOMAIN_SCHED_FIELD_INT;

2779
            if (esxVI_SharesInfo_CastFromAnyType(dynamicProperty->val,
2780
                                                 &sharesInfo) < 0) {
M
Matthias Bolte 已提交
2781
                goto cleanup;
2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801
            }

            switch (sharesInfo->level) {
              case esxVI_SharesLevel_Custom:
                params[i].value.i = sharesInfo->shares->value;
                break;

              case esxVI_SharesLevel_Low:
                params[i].value.i = -1;
                break;

              case esxVI_SharesLevel_Normal:
                params[i].value.i = -2;
                break;

              case esxVI_SharesLevel_High:
                params[i].value.i = -3;
                break;

              default:
2802
                ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
2803
                          _("Shares level has unknown value %d"),
2804
                          (int)sharesInfo->level);
M
Matthias Bolte 已提交
2805
                goto cleanup;
2806 2807 2808 2809 2810 2811 2812 2813 2814 2815 2816 2817
            }

            esxVI_SharesInfo_Free(&sharesInfo);

            mask |= 1 << 2;
            ++i;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

    *nparams = i;
M
Matthias Bolte 已提交
2818
    result = 0;
2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832

  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&virtualMachine);

    return result;
}



static int
esxDomainSetSchedulerParameters(virDomainPtr domain,
                                virSchedParameterPtr params, int nparams)
{
M
Matthias Bolte 已提交
2833
    int result = -1;
M
Matthias Bolte 已提交
2834
    esxPrivate *priv = domain->conn->privateData;
2835 2836 2837 2838 2839 2840 2841
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachineConfigSpec *spec = NULL;
    esxVI_SharesInfo *sharesInfo = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;
    int i;

2842
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
2843
        return -1;
2844 2845
    }

2846
    if (esxVI_LookupVirtualMachineByUuidAndPrepareForTask
2847
          (priv->host, domain->uuid, NULL, &virtualMachine,
2848
           priv->autoAnswer) < 0 ||
2849 2850
        esxVI_VirtualMachineConfigSpec_Alloc(&spec) < 0 ||
        esxVI_ResourceAllocationInfo_Alloc(&spec->cpuAllocation) < 0) {
M
Matthias Bolte 已提交
2851
        goto cleanup;
2852 2853 2854 2855 2856
    }

    for (i = 0; i < nparams; ++i) {
        if (STREQ (params[i].field, "reservation") &&
            params[i].type == VIR_DOMAIN_SCHED_FIELD_LLONG) {
2857
            if (esxVI_Long_Alloc(&spec->cpuAllocation->reservation) < 0) {
M
Matthias Bolte 已提交
2858
                goto cleanup;
2859 2860 2861
            }

            if (params[i].value.l < 0) {
2862
                ESX_ERROR(VIR_ERR_INVALID_ARG,
2863 2864
                          _("Could not set reservation to %lld MHz, expecting "
                            "positive value"), params[i].value.l);
M
Matthias Bolte 已提交
2865
                goto cleanup;
2866 2867 2868 2869 2870
            }

            spec->cpuAllocation->reservation->value = params[i].value.l;
        } else if (STREQ (params[i].field, "limit") &&
                   params[i].type == VIR_DOMAIN_SCHED_FIELD_LLONG) {
2871
            if (esxVI_Long_Alloc(&spec->cpuAllocation->limit) < 0) {
M
Matthias Bolte 已提交
2872
                goto cleanup;
2873 2874 2875
            }

            if (params[i].value.l < -1) {
2876
                ESX_ERROR(VIR_ERR_INVALID_ARG,
2877 2878
                          _("Could not set limit to %lld MHz, expecting "
                            "positive value or -1 (unlimited)"),
2879
                          params[i].value.l);
M
Matthias Bolte 已提交
2880
                goto cleanup;
2881 2882 2883 2884
            }

            spec->cpuAllocation->limit->value = params[i].value.l;
        } else if (STREQ (params[i].field, "shares") &&
2885
                   params[i].type == VIR_DOMAIN_SCHED_FIELD_INT) {
2886 2887
            if (esxVI_SharesInfo_Alloc(&sharesInfo) < 0 ||
                esxVI_Int_Alloc(&sharesInfo->shares) < 0) {
M
Matthias Bolte 已提交
2888
                goto cleanup;
2889 2890 2891 2892
            }

            spec->cpuAllocation->shares = sharesInfo;

2893
            if (params[i].value.i >= 0) {
2894
                spec->cpuAllocation->shares->level = esxVI_SharesLevel_Custom;
2895
                spec->cpuAllocation->shares->shares->value = params[i].value.i;
2896
            } else {
2897
                switch (params[i].value.i) {
2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915
                  case -1:
                    spec->cpuAllocation->shares->level = esxVI_SharesLevel_Low;
                    spec->cpuAllocation->shares->shares->value = -1;
                    break;

                  case -2:
                    spec->cpuAllocation->shares->level =
                      esxVI_SharesLevel_Normal;
                    spec->cpuAllocation->shares->shares->value = -1;
                    break;

                  case -3:
                    spec->cpuAllocation->shares->level =
                      esxVI_SharesLevel_High;
                    spec->cpuAllocation->shares->shares->value = -1;
                    break;

                  default:
2916
                    ESX_ERROR(VIR_ERR_INVALID_ARG,
2917 2918
                              _("Could not set shares to %d, expecting positive "
                                "value or -1 (low), -2 (normal) or -3 (high)"),
2919
                              params[i].value.i);
M
Matthias Bolte 已提交
2920
                    goto cleanup;
2921 2922 2923
                }
            }
        } else {
2924
            ESX_ERROR(VIR_ERR_INVALID_ARG, _("Unknown field '%s'"),
2925
                      params[i].field);
M
Matthias Bolte 已提交
2926
            goto cleanup;
2927 2928 2929
        }
    }

2930 2931 2932 2933
    if (esxVI_ReconfigVM_Task(priv->host, virtualMachine->obj, spec,
                              &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
2934
        goto cleanup;
2935 2936 2937
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
2938 2939
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not change scheduler parameters"));
M
Matthias Bolte 已提交
2940
        goto cleanup;
2941 2942
    }

M
Matthias Bolte 已提交
2943 2944
    result = 0;

2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961 2962 2963
  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_VirtualMachineConfigSpec_Free(&spec);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainMigratePrepare(virConnectPtr dconn,
                        char **cookie ATTRIBUTE_UNUSED,
                        int *cookielen ATTRIBUTE_UNUSED,
                        const char *uri_in, char **uri_out,
                        unsigned long flags ATTRIBUTE_UNUSED,
                        const char *dname ATTRIBUTE_UNUSED,
                        unsigned long resource ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
2964
    int result = -1;
2965
    esxUtil_ParsedQuery *parsedQuery = NULL;
2966 2967

    if (uri_in == NULL) {
2968
        if (esxUtil_ParseQuery(&parsedQuery, dconn->uri) < 0) {
2969 2970 2971
            return -1;
        }

2972
        if (virAsprintf(uri_out, "%s://%s:%d/sdk", parsedQuery->transport,
2973
                        dconn->uri->server, dconn->uri->port) < 0) {
2974
            virReportOOMError();
M
Matthias Bolte 已提交
2975
            goto cleanup;
2976 2977 2978
        }
    }

M
Matthias Bolte 已提交
2979 2980
    result = 0;

2981
  cleanup:
2982
    esxUtil_FreeParsedQuery(&parsedQuery);
2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997

    return result;
}



static int
esxDomainMigratePerform(virDomainPtr domain,
                        const char *cookie ATTRIBUTE_UNUSED,
                        int cookielen ATTRIBUTE_UNUSED,
                        const char *uri,
                        unsigned long flags ATTRIBUTE_UNUSED,
                        const char *dname,
                        unsigned long bandwidth ATTRIBUTE_UNUSED)
{
M
Matthias Bolte 已提交
2998
    int result = -1;
M
Matthias Bolte 已提交
2999
    esxPrivate *priv = domain->conn->privateData;
3000
    xmlURIPtr xmlUri = NULL;
M
Matthias Bolte 已提交
3001
    char hostIpAddress[NI_MAXHOST] = "";
3002 3003 3004 3005 3006 3007 3008 3009
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_ManagedObjectReference *resourcePool = NULL;
    esxVI_Event *eventList = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

M
Matthias Bolte 已提交
3010
    if (priv->vCenter == NULL) {
3011 3012
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s",
                  _("Migration not possible without a vCenter"));
M
Matthias Bolte 已提交
3013
        return -1;
3014 3015 3016
    }

    if (dname != NULL) {
3017 3018
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s",
                  _("Renaming domains on migration not supported"));
M
Matthias Bolte 已提交
3019
        return -1;
3020 3021
    }

3022
    if (esxVI_EnsureSession(priv->vCenter) < 0) {
M
Matthias Bolte 已提交
3023
        return -1;
3024 3025 3026 3027 3028 3029
    }

    /* Parse the destination URI and resolve the hostname */
    xmlUri = xmlParseURI(uri);

    if (xmlUri == NULL) {
3030
        virReportOOMError();
M
Matthias Bolte 已提交
3031
        return -1;
3032 3033
    }

3034
    if (esxUtil_ResolveHostname(xmlUri->server, hostIpAddress,
3035
                                NI_MAXHOST) < 0) {
M
Matthias Bolte 已提交
3036
        goto cleanup;
3037 3038 3039
    }

    /* Lookup VirtualMachine, HostSystem and ResourcePool */
3040
    if (esxVI_LookupVirtualMachineByUuidAndPrepareForTask
3041
          (priv->vCenter, domain->uuid, NULL, &virtualMachine,
3042
           priv->autoAnswer) < 0 ||
3043 3044
        esxVI_String_AppendValueToList(&propertyNameList, "parent") < 0 ||
        esxVI_LookupHostSystemByIp(priv->vCenter, hostIpAddress,
M
Matthias Bolte 已提交
3045
                                   propertyNameList, &hostSystem) < 0) {
M
Matthias Bolte 已提交
3046
        goto cleanup;
3047 3048
    }

3049 3050
    if (esxVI_LookupResourcePoolByHostSystem(priv->vCenter, hostSystem,
                                             &resourcePool) < 0) {
M
Matthias Bolte 已提交
3051
        goto cleanup;
3052 3053 3054
    }

    /* Validate the purposed migration */
3055
    if (esxVI_ValidateMigration(priv->vCenter, virtualMachine->obj,
3056 3057 3058
                                esxVI_VirtualMachinePowerState_Undefined,
                                NULL, resourcePool, hostSystem->obj,
                                &eventList) < 0) {
M
Matthias Bolte 已提交
3059
        goto cleanup;
3060 3061 3062 3063 3064 3065 3066 3067
    }

    if (eventList != NULL) {
        /*
         * FIXME: Need to report the complete list of events. Limit reporting
         *        to the first event for now.
         */
        if (eventList->fullFormattedMessage != NULL) {
3068
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
3069 3070
                      _("Could not migrate domain, validation reported a "
                        "problem: %s"), eventList->fullFormattedMessage);
3071
        } else {
3072 3073 3074
            ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                      _("Could not migrate domain, validation reported a "
                        "problem"));
3075 3076
        }

M
Matthias Bolte 已提交
3077
        goto cleanup;
3078 3079 3080
    }

    /* Perform the purposed migration */
3081
    if (esxVI_MigrateVM_Task(priv->vCenter, virtualMachine->obj, resourcePool,
3082 3083 3084 3085
                             hostSystem->obj,
                             esxVI_VirtualMachineMovePriority_DefaultPriority,
                             esxVI_VirtualMachinePowerState_Undefined,
                             &task) < 0 ||
3086 3087
        esxVI_WaitForTaskCompletion(priv->vCenter, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
3088
        goto cleanup;
3089 3090 3091
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
3092 3093 3094
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not migrate domain, migration task finished with "
                    "an error"));
M
Matthias Bolte 已提交
3095
        goto cleanup;
3096 3097
    }

M
Matthias Bolte 已提交
3098 3099
    result = 0;

3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125
  cleanup:
    xmlFreeURI(xmlUri);
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);
    esxVI_ManagedObjectReference_Free(&resourcePool);
    esxVI_Event_Free(&eventList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static virDomainPtr
esxDomainMigrateFinish(virConnectPtr dconn, const char *dname,
                       const char *cookie ATTRIBUTE_UNUSED,
                       int cookielen ATTRIBUTE_UNUSED,
                       const char *uri ATTRIBUTE_UNUSED,
                       unsigned long flags ATTRIBUTE_UNUSED)
{
    return esxDomainLookupByName(dconn, dname);
}



M
Matthias Bolte 已提交
3126 3127 3128 3129
static unsigned long long
esxNodeGetFreeMemory(virConnectPtr conn)
{
    unsigned long long result = 0;
M
Matthias Bolte 已提交
3130
    esxPrivate *priv = conn->privateData;
M
Matthias Bolte 已提交
3131 3132 3133 3134 3135 3136 3137
    esxVI_String *propertyNameList = NULL;
    esxVI_ObjectContent *hostSystem = NULL;
    esxVI_ManagedObjectReference *managedObjectReference = NULL;
    esxVI_ObjectContent *resourcePool = NULL;
    esxVI_DynamicProperty *dynamicProperty = NULL;
    esxVI_ResourcePoolResourceUsage *resourcePoolResourceUsage = NULL;

3138
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3139
        return 0;
M
Matthias Bolte 已提交
3140 3141 3142
    }

    /* Lookup host system with its resource pool */
3143 3144
    if (esxVI_String_AppendValueToList(&propertyNameList, "parent") < 0 ||
        esxVI_LookupHostSystemByIp(priv->host, priv->host->ipAddress,
M
Matthias Bolte 已提交
3145
                                   propertyNameList, &hostSystem) < 0) {
M
Matthias Bolte 已提交
3146
        goto cleanup;
M
Matthias Bolte 已提交
3147 3148
    }

3149
    if (esxVI_LookupResourcePoolByHostSystem(priv->host, hostSystem,
3150
                                             &managedObjectReference) < 0) {
M
Matthias Bolte 已提交
3151
        goto cleanup;
M
Matthias Bolte 已提交
3152 3153 3154 3155 3156
    }

    esxVI_String_Free(&propertyNameList);

    /* Get memory usage of resource pool */
3157
    if (esxVI_String_AppendValueToList(&propertyNameList,
M
Matthias Bolte 已提交
3158
                                       "runtime.memory") < 0 ||
3159
        esxVI_LookupObjectContentByType(priv->host, managedObjectReference,
3160 3161 3162
                                        "ResourcePool", propertyNameList,
                                        esxVI_Boolean_False,
                                        &resourcePool) < 0) {
M
Matthias Bolte 已提交
3163
        goto cleanup;
M
Matthias Bolte 已提交
3164 3165 3166 3167 3168 3169
    }

    for (dynamicProperty = resourcePool->propSet; dynamicProperty != NULL;
         dynamicProperty = dynamicProperty->_next) {
        if (STREQ(dynamicProperty->name, "runtime.memory")) {
            if (esxVI_ResourcePoolResourceUsage_CastFromAnyType
3170
                  (dynamicProperty->val, &resourcePoolResourceUsage) < 0) {
M
Matthias Bolte 已提交
3171
                goto cleanup;
M
Matthias Bolte 已提交
3172 3173 3174 3175 3176 3177 3178 3179 3180
            }

            break;
        } else {
            VIR_WARN("Unexpected '%s' property", dynamicProperty->name);
        }
    }

    if (resourcePoolResourceUsage == NULL) {
3181 3182
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Could not retrieve memory usage of resource pool"));
M
Matthias Bolte 已提交
3183
        goto cleanup;
M
Matthias Bolte 已提交
3184 3185 3186 3187 3188 3189 3190 3191 3192 3193 3194 3195 3196 3197 3198 3199
    }

    result = resourcePoolResourceUsage->unreservedForVm->value;

  cleanup:
    esxVI_String_Free(&propertyNameList);
    esxVI_ObjectContent_Free(&hostSystem);
    esxVI_ManagedObjectReference_Free(&managedObjectReference);
    esxVI_ObjectContent_Free(&resourcePool);
    esxVI_ResourcePoolResourceUsage_Free(&resourcePoolResourceUsage);

    return result;
}



3200 3201 3202
static int
esxIsEncrypted(virConnectPtr conn)
{
M
Matthias Bolte 已提交
3203
    esxPrivate *priv = conn->privateData;
3204 3205 3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216

    if (STRCASEEQ(priv->transport, "https")) {
        return 1;
    } else {
        return 0;
    }
}



static int
esxIsSecure(virConnectPtr conn)
{
M
Matthias Bolte 已提交
3217
    esxPrivate *priv = conn->privateData;
3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229 3230

    if (STRCASEEQ(priv->transport, "https")) {
        return 1;
    } else {
        return 0;
    }
}



static int
esxDomainIsActive(virDomainPtr domain)
{
M
Matthias Bolte 已提交
3231
    int result = -1;
M
Matthias Bolte 已提交
3232
    esxPrivate *priv = domain->conn->privateData;
3233 3234 3235 3236
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_String *propertyNameList = NULL;
    esxVI_VirtualMachinePowerState powerState;

3237
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3238
        return -1;
3239 3240
    }

3241
    if (esxVI_String_AppendValueToList(&propertyNameList,
3242
                                       "runtime.powerState") < 0 ||
3243 3244
        esxVI_LookupVirtualMachineByUuid(priv->host, domain->uuid,
                                         propertyNameList, &virtualMachine,
M
Matthias Bolte 已提交
3245
                                         esxVI_Occurrence_RequiredItem) < 0 ||
3246
        esxVI_GetVirtualMachinePowerState(virtualMachine, &powerState) < 0) {
M
Matthias Bolte 已提交
3247
        goto cleanup;
3248 3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273
    }

    if (powerState != esxVI_VirtualMachinePowerState_PoweredOff) {
        result = 1;
    } else {
        result = 0;
    }

  cleanup:
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_String_Free(&propertyNameList);

    return result;
}



static int
esxDomainIsPersistent(virDomainPtr domain ATTRIBUTE_UNUSED)
{
    /* ESX has no concept of transient domains, so all of them are persistent */
    return 1;
}



3274 3275
static virDomainSnapshotPtr
esxDomainSnapshotCreateXML(virDomainPtr domain, const char *xmlDesc,
3276
                           unsigned int flags)
3277 3278 3279 3280 3281 3282 3283 3284 3285 3286 3287
{
    esxPrivate *priv = domain->conn->privateData;
    virDomainSnapshotDefPtr def = NULL;
    esxVI_ObjectContent *virtualMachine = NULL;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotList = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTree = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTreeParent = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;
    virDomainSnapshotPtr snapshot = NULL;

3288 3289
    virCheckFlags(0, NULL);

3290
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3291
        return NULL;
3292 3293 3294 3295 3296
    }

    def = virDomainSnapshotDefParseString(xmlDesc, 1);

    if (def == NULL) {
M
Matthias Bolte 已提交
3297
        return NULL;
3298 3299 3300 3301 3302 3303 3304 3305 3306 3307
    }

    if (esxVI_LookupVirtualMachineByUuidAndPrepareForTask
          (priv->host, domain->uuid, NULL, &virtualMachine,
           priv->autoAnswer) < 0 ||
        esxVI_LookupRootSnapshotTreeList(priv->host, domain->uuid,
                                         &rootSnapshotList) < 0 ||
        esxVI_GetSnapshotTreeByName(rootSnapshotList, def->name,
                                    &snapshotTree, &snapshotTreeParent,
                                    esxVI_Occurrence_OptionalItem) < 0) {
M
Matthias Bolte 已提交
3308
        goto cleanup;
3309 3310 3311 3312 3313
    }

    if (snapshotTree != NULL) {
        ESX_ERROR(VIR_ERR_OPERATION_INVALID,
                  _("Snapshot '%s' already exists"), def->name);
M
Matthias Bolte 已提交
3314
        goto cleanup;
3315 3316 3317 3318 3319 3320 3321 3322
    }

    if (esxVI_CreateSnapshot_Task(priv->host, virtualMachine->obj,
                                  def->name, def->description,
                                  esxVI_Boolean_True,
                                  esxVI_Boolean_False, &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
3323
        goto cleanup;
3324 3325 3326 3327
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR, "%s", _("Could not create snapshot"));
M
Matthias Bolte 已提交
3328
        goto cleanup;
3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345
    }

    snapshot = virGetDomainSnapshot(domain, def->name);

  cleanup:
    virDomainSnapshotDefFree(def);
    esxVI_ObjectContent_Free(&virtualMachine);
    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotList);
    esxVI_ManagedObjectReference_Free(&task);

    return snapshot;
}



static char *
esxDomainSnapshotDumpXML(virDomainSnapshotPtr snapshot,
3346
                         unsigned int flags)
3347 3348 3349 3350 3351 3352 3353 3354 3355
{
    esxPrivate *priv = snapshot->domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotList = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTree = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTreeParent = NULL;
    virDomainSnapshotDef def;
    char uuid_string[VIR_UUID_STRING_BUFLEN] = "";
    char *xml = NULL;

3356 3357
    virCheckFlags(0, NULL);

M
Matthias Bolte 已提交
3358
    memset(&def, 0, sizeof (def));
3359 3360

    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3361
        return NULL;
3362 3363 3364 3365 3366 3367 3368
    }

    if (esxVI_LookupRootSnapshotTreeList(priv->host, snapshot->domain->uuid,
                                         &rootSnapshotList) < 0 ||
        esxVI_GetSnapshotTreeByName(rootSnapshotList, snapshot->name,
                                    &snapshotTree, &snapshotTreeParent,
                                    esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
3369
        goto cleanup;
3370 3371 3372 3373 3374 3375 3376 3377
    }

    def.name = snapshot->name;
    def.description = snapshotTree->description;
    def.parent = snapshotTreeParent != NULL ? snapshotTreeParent->name : NULL;

    if (esxVI_DateTime_ConvertToCalendarTime(snapshotTree->createTime,
                                             &def.creationTime) < 0) {
M
Matthias Bolte 已提交
3378
        goto cleanup;
3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396
    }

    def.state = esxVI_VirtualMachinePowerState_ConvertToLibvirt
                  (snapshotTree->state);

    virUUIDFormat(snapshot->domain->uuid, uuid_string);

    xml = virDomainSnapshotDefFormat(uuid_string, &def, 0);

  cleanup:
    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotList);

    return xml;
}



static int
3397
esxDomainSnapshotNum(virDomainPtr domain, unsigned int flags)
3398
{
M
Matthias Bolte 已提交
3399
    int count;
3400 3401 3402
    esxPrivate *priv = domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotTreeList = NULL;

3403 3404
    virCheckFlags(0, -1);

3405
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3406
        return -1;
3407 3408 3409 3410
    }

    if (esxVI_LookupRootSnapshotTreeList(priv->host, domain->uuid,
                                         &rootSnapshotTreeList) < 0) {
M
Matthias Bolte 已提交
3411
        return -1;
3412 3413
    }

M
Matthias Bolte 已提交
3414
    count = esxVI_GetNumberOfSnapshotTrees(rootSnapshotTreeList);
3415 3416 3417

    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotTreeList);

M
Matthias Bolte 已提交
3418
    return count;
3419 3420 3421 3422 3423 3424
}



static int
esxDomainSnapshotListNames(virDomainPtr domain, char **names, int nameslen,
3425
                           unsigned int flags)
3426
{
M
Matthias Bolte 已提交
3427
    int result;
3428 3429 3430
    esxPrivate *priv = domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotTreeList = NULL;

3431 3432
    virCheckFlags(0, -1);

3433 3434 3435 3436 3437 3438 3439 3440 3441 3442
    if (names == NULL || nameslen < 0) {
        ESX_ERROR(VIR_ERR_INVALID_ARG, "%s", _("Invalid argument"));
        return -1;
    }

    if (nameslen == 0) {
        return 0;
    }

    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3443
        return -1;
3444 3445 3446 3447
    }

    if (esxVI_LookupRootSnapshotTreeList(priv->host, domain->uuid,
                                         &rootSnapshotTreeList) < 0) {
M
Matthias Bolte 已提交
3448
        return -1;
3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461
    }

    result = esxVI_GetSnapshotTreeNames(rootSnapshotTreeList, names, nameslen);

    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotTreeList);

    return result;
}



static virDomainSnapshotPtr
esxDomainSnapshotLookupByName(virDomainPtr domain, const char *name,
3462
                              unsigned int flags)
3463 3464 3465 3466 3467 3468 3469
{
    esxPrivate *priv = domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotTreeList = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTree = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTreeParent = NULL;
    virDomainSnapshotPtr snapshot = NULL;

3470 3471
    virCheckFlags(0, NULL);

3472
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3473
        return NULL;
3474 3475 3476 3477 3478 3479 3480 3481 3482 3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498 3499
    }

    if (esxVI_LookupRootSnapshotTreeList(priv->host, domain->uuid,
                                         &rootSnapshotTreeList) < 0 ||
        esxVI_GetSnapshotTreeByName(rootSnapshotTreeList, name, &snapshotTree,
                                    &snapshotTreeParent,
                                    esxVI_Occurrence_RequiredItem) < 0) {
        goto cleanup;
    }

    snapshot = virGetDomainSnapshot(domain, name);

  cleanup:
    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotTreeList);

    return snapshot;
}



static int
esxDomainHasCurrentSnapshot(virDomainPtr domain, unsigned int flags)
{
    esxPrivate *priv = domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *currentSnapshotTree = NULL;

3500
    virCheckFlags(0, -1);
3501 3502

    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3503
        return -1;
3504 3505 3506 3507 3508
    }

    if (esxVI_LookupCurrentSnapshotTree(priv->host, domain->uuid,
                                        &currentSnapshotTree,
                                        esxVI_Occurrence_OptionalItem) < 0) {
M
Matthias Bolte 已提交
3509
        return -1;
3510 3511 3512
    }

    if (currentSnapshotTree != NULL) {
M
Matthias Bolte 已提交
3513 3514
        esxVI_VirtualMachineSnapshotTree_Free(&currentSnapshotTree);
        return 1;
3515 3516
    }

M
Matthias Bolte 已提交
3517
    return 0;
3518 3519 3520 3521 3522 3523 3524 3525 3526
}



static virDomainSnapshotPtr
esxDomainSnapshotCurrent(virDomainPtr domain, unsigned int flags)
{
    esxPrivate *priv = domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *currentSnapshotTree = NULL;
M
Matthias Bolte 已提交
3527
    virDomainSnapshotPtr snapshot = NULL;
3528

3529
    virCheckFlags(0, NULL);
3530 3531

    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3532
        return NULL;
3533 3534 3535 3536 3537
    }

    if (esxVI_LookupCurrentSnapshotTree(priv->host, domain->uuid,
                                        &currentSnapshotTree,
                                        esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
3538
        return NULL;
3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552
    }

    snapshot = virGetDomainSnapshot(domain, currentSnapshotTree->name);

    esxVI_VirtualMachineSnapshotTree_Free(&currentSnapshotTree);

    return snapshot;
}



static int
esxDomainRevertToSnapshot(virDomainSnapshotPtr snapshot, unsigned int flags)
{
M
Matthias Bolte 已提交
3553
    int result = -1;
3554 3555 3556 3557 3558 3559 3560
    esxPrivate *priv = snapshot->domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotList = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTree = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTreeParent = NULL;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

3561
    virCheckFlags(0, -1);
3562 3563

    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3564
        return -1;
3565 3566 3567 3568 3569 3570 3571
    }

    if (esxVI_LookupRootSnapshotTreeList(priv->host, snapshot->domain->uuid,
                                         &rootSnapshotList) < 0 ||
        esxVI_GetSnapshotTreeByName(rootSnapshotList, snapshot->name,
                                    &snapshotTree, &snapshotTreeParent,
                                    esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
3572
        goto cleanup;
3573 3574 3575 3576 3577 3578
    }

    if (esxVI_RevertToSnapshot_Task(priv->host, snapshotTree->snapshot, NULL,
                                    &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, snapshot->domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
3579
        goto cleanup;
3580 3581 3582 3583 3584
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
                  _("Could not revert to snapshot '%s'"), snapshot->name);
M
Matthias Bolte 已提交
3585
        goto cleanup;
3586 3587
    }

M
Matthias Bolte 已提交
3588 3589
    result = 0;

3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600 3601
  cleanup:
    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



static int
esxDomainSnapshotDelete(virDomainSnapshotPtr snapshot, unsigned int flags)
{
M
Matthias Bolte 已提交
3602
    int result = -1;
3603 3604 3605 3606 3607 3608 3609 3610
    esxPrivate *priv = snapshot->domain->conn->privateData;
    esxVI_VirtualMachineSnapshotTree *rootSnapshotList = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTree = NULL;
    esxVI_VirtualMachineSnapshotTree *snapshotTreeParent = NULL;
    esxVI_Boolean removeChildren = esxVI_Boolean_False;
    esxVI_ManagedObjectReference *task = NULL;
    esxVI_TaskInfoState taskInfoState;

3611 3612
    virCheckFlags(VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN, -1);

3613
    if (esxVI_EnsureSession(priv->host) < 0) {
M
Matthias Bolte 已提交
3614
        return -1;
3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625
    }

    if (flags & VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN) {
        removeChildren = esxVI_Boolean_True;
    }

    if (esxVI_LookupRootSnapshotTreeList(priv->host, snapshot->domain->uuid,
                                         &rootSnapshotList) < 0 ||
        esxVI_GetSnapshotTreeByName(rootSnapshotList, snapshot->name,
                                    &snapshotTree, &snapshotTreeParent,
                                    esxVI_Occurrence_RequiredItem) < 0) {
M
Matthias Bolte 已提交
3626
        goto cleanup;
3627 3628 3629 3630 3631 3632
    }

    if (esxVI_RemoveSnapshot_Task(priv->host, snapshotTree->snapshot,
                                  removeChildren, &task) < 0 ||
        esxVI_WaitForTaskCompletion(priv->host, task, snapshot->domain->uuid,
                                    priv->autoAnswer, &taskInfoState) < 0) {
M
Matthias Bolte 已提交
3633
        goto cleanup;
3634 3635 3636 3637 3638
    }

    if (taskInfoState != esxVI_TaskInfoState_Success) {
        ESX_ERROR(VIR_ERR_INTERNAL_ERROR,
                  _("Could not delete snapshot '%s'"), snapshot->name);
M
Matthias Bolte 已提交
3639
        goto cleanup;
3640 3641
    }

M
Matthias Bolte 已提交
3642 3643
    result = 0;

3644 3645 3646 3647 3648 3649 3650 3651 3652
  cleanup:
    esxVI_VirtualMachineSnapshotTree_Free(&rootSnapshotList);
    esxVI_ManagedObjectReference_Free(&task);

    return result;
}



3653 3654 3655 3656 3657 3658 3659 3660
static virDriver esxDriver = {
    VIR_DRV_ESX,
    "ESX",
    esxOpen,                         /* open */
    esxClose,                        /* close */
    esxSupportsFeature,              /* supports_feature */
    esxGetType,                      /* type */
    esxGetVersion,                   /* version */
3661
    NULL,                            /* libvirtVersion (impl. in libvirt.c) */
3662 3663 3664
    esxGetHostname,                  /* hostname */
    NULL,                            /* getMaxVcpus */
    esxNodeGetInfo,                  /* nodeGetInfo */
3665
    esxGetCapabilities,              /* getCapabilities */
3666 3667 3668 3669 3670 3671 3672 3673 3674 3675 3676 3677 3678 3679 3680 3681 3682 3683 3684 3685 3686 3687 3688 3689 3690 3691
    esxListDomains,                  /* listDomains */
    esxNumberOfDomains,              /* numOfDomains */
    NULL,                            /* domainCreateXML */
    esxDomainLookupByID,             /* domainLookupByID */
    esxDomainLookupByUUID,           /* domainLookupByUUID */
    esxDomainLookupByName,           /* domainLookupByName */
    esxDomainSuspend,                /* domainSuspend */
    esxDomainResume,                 /* domainResume */
    esxDomainShutdown,               /* domainShutdown */
    esxDomainReboot,                 /* domainReboot */
    esxDomainDestroy,                /* domainDestroy */
    esxDomainGetOSType,              /* domainGetOSType */
    esxDomainGetMaxMemory,           /* domainGetMaxMemory */
    esxDomainSetMaxMemory,           /* domainSetMaxMemory */
    esxDomainSetMemory,              /* domainSetMemory */
    esxDomainGetInfo,                /* domainGetInfo */
    NULL,                            /* domainSave */
    NULL,                            /* domainRestore */
    NULL,                            /* domainCoreDump */
    esxDomainSetVcpus,               /* domainSetVcpus */
    NULL,                            /* domainPinVcpu */
    NULL,                            /* domainGetVcpus */
    esxDomainGetMaxVcpus,            /* domainGetMaxVcpus */
    NULL,                            /* domainGetSecurityLabel */
    NULL,                            /* nodeGetSecurityModel */
    esxDomainDumpXML,                /* domainDumpXML */
3692
    esxDomainXMLFromNative,          /* domainXMLFromNative */
M
Matthias Bolte 已提交
3693
    esxDomainXMLToNative,            /* domainXMLToNative */
3694 3695 3696
    esxListDefinedDomains,           /* listDefinedDomains */
    esxNumberOfDefinedDomains,       /* numOfDefinedDomains */
    esxDomainCreate,                 /* domainCreate */
M
Matthias Bolte 已提交
3697
    esxDomainDefineXML,              /* domainDefineXML */
3698
    esxDomainUndefine,               /* domainUndefine */
3699
    NULL,                            /* domainAttachDevice */
3700
    NULL,                            /* domainAttachDeviceFlags */
3701
    NULL,                            /* domainDetachDevice */
3702
    NULL,                            /* domainDetachDeviceFlags */
3703
    NULL,                            /* domainUpdateDeviceFlags */
3704 3705 3706 3707 3708 3709 3710 3711 3712 3713
    NULL,                            /* domainGetAutostart */
    NULL,                            /* domainSetAutostart */
    esxDomainGetSchedulerType,       /* domainGetSchedulerType */
    esxDomainGetSchedulerParameters, /* domainGetSchedulerParameters */
    esxDomainSetSchedulerParameters, /* domainSetSchedulerParameters */
    esxDomainMigratePrepare,         /* domainMigratePrepare */
    esxDomainMigratePerform,         /* domainMigratePerform */
    esxDomainMigrateFinish,          /* domainMigrateFinish */
    NULL,                            /* domainBlockStats */
    NULL,                            /* domainInterfaceStats */
3714
    NULL,                            /* domainMemoryStats */
3715 3716
    NULL,                            /* domainBlockPeek */
    NULL,                            /* domainMemoryPeek */
3717
    NULL,                            /* domainGetBlockInfo */
3718
    NULL,                            /* nodeGetCellsFreeMemory */
M
Matthias Bolte 已提交
3719
    esxNodeGetFreeMemory,            /* nodeGetFreeMemory */
3720 3721 3722 3723 3724 3725 3726
    NULL,                            /* domainEventRegister */
    NULL,                            /* domainEventDeregister */
    NULL,                            /* domainMigratePrepare2 */
    NULL,                            /* domainMigrateFinish2 */
    NULL,                            /* nodeDeviceDettach */
    NULL,                            /* nodeDeviceReAttach */
    NULL,                            /* nodeDeviceReset */
C
Chris Lalancette 已提交
3727
    NULL,                            /* domainMigratePrepareTunnel */
3728 3729 3730 3731
    esxIsEncrypted,                  /* isEncrypted */
    esxIsSecure,                     /* isSecure */
    esxDomainIsActive,               /* domainIsActive */
    esxDomainIsPersistent,           /* domainIsPersistent */
J
Jiri Denemark 已提交
3732
    NULL,                            /* cpuCompare */
3733
    NULL,                            /* cpuBaseline */
3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750
    NULL,                            /* domainGetJobInfo */
    NULL,                            /* domainAbortJob */
    NULL,                            /* domainMigrateSetMaxDowntime */
    NULL,                            /* domainEventRegisterAny */
    NULL,                            /* domainEventDeregisterAny */
    NULL,                            /* domainManagedSave */
    NULL,                            /* domainHasManagedSaveImage */
    NULL,                            /* domainManagedSaveRemove */
    esxDomainSnapshotCreateXML,      /* domainSnapshotCreateXML */
    esxDomainSnapshotDumpXML,        /* domainSnapshotDumpXML */
    esxDomainSnapshotNum,            /* domainSnapshotNum */
    esxDomainSnapshotListNames,      /* domainSnapshotListNames */
    esxDomainSnapshotLookupByName,   /* domainSnapshotLookupByName */
    esxDomainHasCurrentSnapshot,     /* domainHasCurrentSnapshot */
    esxDomainSnapshotCurrent,        /* domainSnapshotCurrent */
    esxDomainRevertToSnapshot,       /* domainRevertToSnapshot */
    esxDomainSnapshotDelete,         /* domainSnapshotDelete */
3751 3752 3753 3754 3755 3756 3757
};



int
esxRegister(void)
{
3758 3759 3760 3761 3762
    if (virRegisterDriver(&esxDriver) < 0 ||
        esxInterfaceRegister() < 0 ||
        esxNetworkRegister() < 0 ||
        esxStorageRegister() < 0 ||
        esxDeviceRegister() < 0 ||
M
Matthias Bolte 已提交
3763 3764
        esxSecretRegister() < 0 ||
        esxNWFilterRegister() < 0) {
3765 3766
        return -1;
    }
3767 3768 3769

    return 0;
}