vbox_tmpl.c 317.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10
/** @file vbox_tmpl.c
 * Template File to support multiple versions of VirtualBox
 * at runtime :).
 *
 * IMPORTANT:
 * Please dont include this file in the src/Makefile.am, it
 * is automatically include by other files.
 */

/*
11
 * Copyright (C) 2010-2011 Red Hat, Inc.
12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
 * Copyright (C) 2008-2009 Sun Microsystems, Inc.
 *
 * This file is part of a free software library; you can redistribute
 * it and/or modify it under the terms of the GNU Lesser General
 * Public License version 2.1 as published by the Free Software
 * Foundation and shipped in the "COPYING" file with this library.
 * The library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY of any kind.
 *
 * Sun LGPL Disclaimer: For the avoidance of doubt, except that if
 * any license choice other than GPL or LGPL is available it will
 * apply instead, Sun elects to use only the Lesser General Public
 * License version 2.1 (LGPLv2) at this time for any software where
 * a choice of LGPL license versions is made available with the
 * language indicating that LGPLv2 or any later version may be used,
 * or where a choice of which version of the LGPL is applied is
 * otherwise unspecified.
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa
 * Clara, CA 95054 USA or visit http://www.sun.com if you need
 * additional information or have any questions.
 */

#include <config.h>

#include <sys/utsname.h>
38
#include <unistd.h>
39 40 41
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
42 43 44 45

#include "internal.h"
#include "datatypes.h"
#include "domain_conf.h"
46
#include "network_conf.h"
47
#include "virterror_internal.h"
48
#include "domain_event.h"
49
#include "storage_conf.h"
50
#include "storage_file.h"
51 52 53 54 55
#include "uuid.h"
#include "memory.h"
#include "nodeinfo.h"
#include "logging.h"
#include "vbox_driver.h"
56 57 58
#include "configmake.h"
#include "files.h"
#include "fdstream.h"
59 60 61 62

/* This one changes from version to version. */
#if VBOX_API_VERSION == 2002
# include "vbox_CAPI_v2_2.h"
63 64
#elif VBOX_API_VERSION == 3000
# include "vbox_CAPI_v3_0.h"
65 66
#elif VBOX_API_VERSION == 3001
# include "vbox_CAPI_v3_1.h"
67 68
#elif VBOX_API_VERSION == 3002
# include "vbox_CAPI_v3_2.h"
69 70
#elif VBOX_API_VERSION == 4000
# include "vbox_CAPI_v4_0.h"
71 72
#else
# error "Unsupport VBOX_API_VERSION"
73 74
#endif

75
/* Include this *last* or we'll get the wrong vbox_CAPI_*.h. */
76
#include "vbox_glue.h"
77 78


79 80 81 82 83 84 85
#define VIR_FROM_THIS                   VIR_FROM_VBOX
#define VBOX_UTF16_FREE(arg)            data->pFuncs->pfnUtf16Free(arg)
#define VBOX_UTF8_FREE(arg)             data->pFuncs->pfnUtf8Free(arg)
#define VBOX_COM_UNALLOC_MEM(arg)       data->pFuncs->pfnComUnallocMem(arg)
#define VBOX_UTF16_TO_UTF8(arg1, arg2)  data->pFuncs->pfnUtf16ToUtf8(arg1, arg2)
#define VBOX_UTF8_TO_UTF16(arg1, arg2)  data->pFuncs->pfnUtf8ToUtf16(arg1, arg2)

86 87
#define VBOX_ADDREF(arg) (arg)->vtbl->nsisupports.AddRef((nsISupports *)(arg))

88 89 90 91 92 93 94
#define VBOX_RELEASE(arg)                                                     \
    do {                                                                      \
        if (arg) {                                                            \
            (arg)->vtbl->nsisupports.Release((nsISupports *)(arg));           \
            (arg) = NULL;                                                     \
        }                                                                     \
    } while (0)
95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116

#define VBOX_OBJECT_CHECK(conn, type, value) \
vboxGlobalData *data = conn->privateData;\
type ret = value;\
if(!data->vboxObj) {\
    return ret;\
}

#define VBOX_OBJECT_HOST_CHECK(conn, type, value) \
vboxGlobalData *data = conn->privateData;\
type ret = value;\
IHost *host = NULL;\
if(!data->vboxObj) {\
    return ret;\
}\
data->vboxObj->vtbl->GetHost(data->vboxObj, &host);\
if (!host) {\
    return ret;\
}

#if VBOX_API_VERSION < 3001

117
# define VBOX_MEDIUM_RELEASE(arg) \
118 119
if(arg)\
    (arg)->vtbl->imedium.nsisupports.Release((nsISupports *)(arg))
120
# define VBOX_MEDIUM_FUNC_ARG1(object, func, arg1) \
121
    (object)->vtbl->imedium.func((IMedium *)(object), arg1)
122
# define VBOX_MEDIUM_FUNC_ARG2(object, func, arg1, arg2) \
123 124 125 126 127 128
    (object)->vtbl->imedium.func((IMedium *)(object), arg1, arg2)

#else  /* VBOX_API_VERSION >= 3001 */

typedef IMedium IHardDisk;
typedef IMediumAttachment IHardDiskAttachment;
129 130 131 132 133
# define MediaState_Inaccessible     MediumState_Inaccessible
# define HardDiskVariant_Standard    MediumVariant_Standard
# define HardDiskVariant_Fixed       MediumVariant_Fixed
# define VBOX_MEDIUM_RELEASE(arg) VBOX_RELEASE(arg)
# define VBOX_MEDIUM_FUNC_ARG1(object, func, arg1) \
134
    (object)->vtbl->func(object, arg1)
135
# define VBOX_MEDIUM_FUNC_ARG2(object, func, arg1, arg2) \
136 137 138
    (object)->vtbl->func(object, arg1, arg2)

#endif /* VBOX_API_VERSION >= 3001 */
139

140
#define vboxError(code, ...) \
141
        virReportErrorHelper(VIR_FROM_VBOX, code, __FILE__, \
142
                             __FUNCTION__, __LINE__, __VA_ARGS__)
143

144 145 146 147 148 149
#define DEBUGPRUnichar(msg, strUtf16) \
if (strUtf16) {\
    char *strUtf8 = NULL;\
\
    g_pVBoxGlobalData->pFuncs->pfnUtf16ToUtf8(strUtf16, &strUtf8);\
    if (strUtf8) {\
150
        VIR_DEBUG("%s: %s", msg, strUtf8);\
151 152 153 154 155 156
        g_pVBoxGlobalData->pFuncs->pfnUtf8Free(strUtf8);\
    }\
}

#define DEBUGUUID(msg, iid) \
{\
157
    VIR_DEBUG (msg ": {%08x-%04x-%04x-%02x%02x-%02x%02x%02x%02x%02x%02x}",\
158 159 160 161 162 163 164 165 166 167 168 169 170
          (unsigned)(iid)->m0,\
          (unsigned)(iid)->m1,\
          (unsigned)(iid)->m2,\
          (unsigned)(iid)->m3[0],\
          (unsigned)(iid)->m3[1],\
          (unsigned)(iid)->m3[2],\
          (unsigned)(iid)->m3[3],\
          (unsigned)(iid)->m3[4],\
          (unsigned)(iid)->m3[5],\
          (unsigned)(iid)->m3[6],\
          (unsigned)(iid)->m3[7]);\
}\

171 172
typedef struct {
    virMutex lock;
173
    unsigned long version;
174 175 176 177 178

    virCapsPtr caps;

    IVirtualBox *vboxObj;
    ISession *vboxSession;
179 180 181

    /** Our version specific API table pointer. */
    PCVBOXXPCOM pFuncs;
182 183 184 185 186 187 188 189 190 191 192 193 194

#if VBOX_API_VERSION == 2002

} vboxGlobalData;

#else /* !(VBOX_API_VERSION == 2002) */

    /* An array of callbacks */
    virDomainEventCallbackListPtr domainEventCallbacks;

    int fdWatch;
    int domainEventDispatching;

195 196
# if VBOX_API_VERSION <= 3002
    /* IVirtualBoxCallback is used in VirtualBox 3.x only */
197
    IVirtualBoxCallback *vboxCallback;
198
# endif /* VBOX_API_VERSION <= 3002 */
199 200 201 202 203 204 205

    nsIEventQueue  *vboxQueue;
    int volatile vboxCallBackRefCount;

    /* pointer back to the connection */
    virConnectPtr conn;

206 207
} vboxGlobalData;

208 209 210 211 212 213 214 215 216 217
/* g_pVBoxGlobalData has to be global variable,
 * there is no other way to make the callbacks
 * work other then having g_pVBoxGlobalData as
 * global, because the functions namely AddRef,
 * Release, etc consider it as global and you
 * can't change the function definition as it
 * is XPCOM nsISupport::* function and it expects
 * them that way
 */

218
static vboxGlobalData *g_pVBoxGlobalData = NULL;
219 220

#endif /* !(VBOX_API_VERSION == 2002) */
221

222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
#if VBOX_API_VERSION < 4000

# define VBOX_OBJECT_GET_MACHINE(/* in */ iid_value, /* out */ machine) \
    data->vboxObj->vtbl->GetMachine(data->vboxObj, iid_value, machine)

# define VBOX_SESSION_OPEN(/* in */ iid_value, /* unused */ machine) \
    data->vboxObj->vtbl->OpenSession(data->vboxObj, data->vboxSession, iid_value)

# define VBOX_SESSION_OPEN_EXISTING(/* in */ iid_value, /* unused */ machine) \
    data->vboxObj->vtbl->OpenExistingSession(data->vboxObj, data->vboxSession, iid_value)

# define VBOX_SESSION_CLOSE() \
    data->vboxSession->vtbl->Close(data->vboxSession)

#else /* VBOX_API_VERSION >= 4000 */

# define VBOX_OBJECT_GET_MACHINE(/* in */ iid_value, /* out */ machine) \
    data->vboxObj->vtbl->FindMachine(data->vboxObj, iid_value, machine)

# define VBOX_SESSION_OPEN(/* unused */ iid_value, /* in */ machine) \
    machine->vtbl->LockMachine(machine, data->vboxSession, LockType_Write)

# define VBOX_SESSION_OPEN_EXISTING(/* unused */ iid_value, /* in */ machine) \
    machine->vtbl->LockMachine(machine, data->vboxSession, LockType_Shared)

# define VBOX_SESSION_CLOSE() \
    data->vboxSession->vtbl->UnlockMachine(data->vboxSession)

#endif /* VBOX_API_VERSION >= 4000 */

252 253 254 255
static virDomainPtr vboxDomainDefineXML(virConnectPtr conn, const char *xml);
static int vboxDomainCreate(virDomainPtr dom);
static int vboxDomainUndefine(virDomainPtr dom);

256 257 258 259 260 261 262 263 264
static void vboxDriverLock(vboxGlobalData *data) {
    virMutexLock(&data->lock);
}

static void vboxDriverUnlock(vboxGlobalData *data) {
    virMutexUnlock(&data->lock);
}

#if VBOX_API_VERSION == 2002
265

266
static void nsIDtoChar(unsigned char *uuid, const nsID *iid) {
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305
    char uuidstrsrc[VIR_UUID_STRING_BUFLEN];
    char uuidstrdst[VIR_UUID_STRING_BUFLEN];
    unsigned char uuidinterim[VIR_UUID_BUFLEN];
    int i;

    memcpy(uuidinterim, iid, VIR_UUID_BUFLEN);
    virUUIDFormat(uuidinterim, uuidstrsrc);

    uuidstrdst[0]  = uuidstrsrc[6];
    uuidstrdst[1]  = uuidstrsrc[7];
    uuidstrdst[2]  = uuidstrsrc[4];
    uuidstrdst[3]  = uuidstrsrc[5];
    uuidstrdst[4]  = uuidstrsrc[2];
    uuidstrdst[5]  = uuidstrsrc[3];
    uuidstrdst[6]  = uuidstrsrc[0];
    uuidstrdst[7]  = uuidstrsrc[1];

    uuidstrdst[8]  = uuidstrsrc[8];

    uuidstrdst[9]  = uuidstrsrc[11];
    uuidstrdst[10] = uuidstrsrc[12];
    uuidstrdst[11] = uuidstrsrc[9];
    uuidstrdst[12] = uuidstrsrc[10];

    uuidstrdst[13] = uuidstrsrc[13];

    uuidstrdst[14] = uuidstrsrc[16];
    uuidstrdst[15] = uuidstrsrc[17];
    uuidstrdst[16] = uuidstrsrc[14];
    uuidstrdst[17] = uuidstrsrc[15];

    for(i = 18; i < VIR_UUID_STRING_BUFLEN; i++) {
        uuidstrdst[i] = uuidstrsrc[i];
    }

    uuidstrdst[VIR_UUID_STRING_BUFLEN-1] = '\0';
    virUUIDParse(uuidstrdst, uuid);
}

306
static void nsIDFromChar(nsID *iid, const unsigned char *uuid) {
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
    char uuidstrsrc[VIR_UUID_STRING_BUFLEN];
    char uuidstrdst[VIR_UUID_STRING_BUFLEN];
    unsigned char uuidinterim[VIR_UUID_BUFLEN];
    int i;

    virUUIDFormat(uuid, uuidstrsrc);

    uuidstrdst[0]  = uuidstrsrc[6];
    uuidstrdst[1]  = uuidstrsrc[7];
    uuidstrdst[2]  = uuidstrsrc[4];
    uuidstrdst[3]  = uuidstrsrc[5];
    uuidstrdst[4]  = uuidstrsrc[2];
    uuidstrdst[5]  = uuidstrsrc[3];
    uuidstrdst[6]  = uuidstrsrc[0];
    uuidstrdst[7]  = uuidstrsrc[1];

    uuidstrdst[8]  = uuidstrsrc[8];

    uuidstrdst[9]  = uuidstrsrc[11];
    uuidstrdst[10] = uuidstrsrc[12];
    uuidstrdst[11] = uuidstrsrc[9];
    uuidstrdst[12] = uuidstrsrc[10];

    uuidstrdst[13] = uuidstrsrc[13];

    uuidstrdst[14] = uuidstrsrc[16];
    uuidstrdst[15] = uuidstrsrc[17];
    uuidstrdst[16] = uuidstrsrc[14];
    uuidstrdst[17] = uuidstrsrc[15];

    for(i = 18; i < VIR_UUID_STRING_BUFLEN; i++) {
        uuidstrdst[i] = uuidstrsrc[i];
    }

    uuidstrdst[VIR_UUID_STRING_BUFLEN-1] = '\0';
    virUUIDParse(uuidstrdst, uuidinterim);
    memcpy(iid, uuidinterim, VIR_UUID_BUFLEN);
}

346
# ifdef WIN32
347

348 349 350 351 352 353 354 355 356 357 358 359 360 361 362
typedef struct _vboxIID_v2_x_WIN32 vboxIID;
typedef struct _vboxIID_v2_x_WIN32 vboxIID_v2_x_WIN32;

struct _vboxIID_v2_x_WIN32 {
    /* IID is represented by a GUID value. */
    GUID value;
};

#  define VBOX_IID_INITIALIZER { { 0, 0, 0, { 0, 0, 0, 0, 0, 0, 0, 0 } } }

static void
vboxIIDUnalloc_v2_x_WIN32(vboxGlobalData *data ATTRIBUTE_UNUSED,
                          vboxIID_v2_x_WIN32 *iid ATTRIBUTE_UNUSED)
{
    /* Nothing to free */
363
}
364

365 366 367 368 369
static void
vboxIIDToUUID_v2_x_WIN32(vboxIID_v2_x_WIN32 *iid, unsigned char *uuid)
{
    nsIDtoChar(uuid, (nsID *)&iid->value);
}
370

371 372 373 374 375
static void
vboxIIDFromUUID_v2_x_WIN32(vboxGlobalData *data, vboxIID_v2_x_WIN32 *iid,
                           const unsigned char *uuid)
{
    vboxIIDUnalloc_v2_x_WIN32(data, iid);
376

377
    nsIDFromChar((nsID *)&iid->value, uuid);
378 379
}

380 381 382 383 384
static bool
vboxIIDIsEqual_v2_x_WIN32(vboxIID_v2_x_WIN32 *iid1, vboxIID_v2_x_WIN32 *iid2)
{
    return memcmp(&iid1->value, &iid2->value, sizeof (GUID)) == 0;
}
385

386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425
static void
vboxIIDFromArrayItem_v2_x_WIN32(vboxGlobalData *data, vboxIID_v2_x_WIN32 *iid,
                                vboxArray *array, int idx)
{
    GUID *items = (GUID *)array->items;

    vboxIIDUnalloc_v2_x_WIN32(data, iid);

    memcpy(&iid->value, &items[idx], sizeof (GUID));
}

#  define vboxIIDUnalloc(iid) vboxIIDUnalloc_v2_x_WIN32(data, iid)
#  define vboxIIDToUUID(iid, uuid) vboxIIDToUUID_v2_x_WIN32(iid, uuid)
#  define vboxIIDFromUUID(iid, uuid) vboxIIDFromUUID_v2_x_WIN32(data, iid, uuid)
#  define vboxIIDIsEqual(iid1, iid2) vboxIIDIsEqual_v2_x_WIN32(iid1, iid2)
#  define vboxIIDFromArrayItem(iid, array, idx) \
    vboxIIDFromArrayItem_v2_x_WIN32(data, iid, array, idx)
#  define DEBUGIID(msg, iid) DEBUGUUID(msg, (nsID *)&(iid))

# else /* !WIN32 */

typedef struct _vboxIID_v2_x vboxIID;
typedef struct _vboxIID_v2_x vboxIID_v2_x;

struct _vboxIID_v2_x {
    /* IID is represented by a pointer to a nsID. */
    nsID *value;

    /* backing is used in cases where we need to create or copy an IID.
     * We cannot allocate memory that can be freed by ComUnallocMem.
     * Therefore, we use this stack allocated nsID instead. */
    nsID backing;
};

#  define VBOX_IID_INITIALIZER { NULL, { 0, 0, 0, { 0, 0, 0, 0, 0, 0, 0, 0 } } }

static void
vboxIIDUnalloc_v2_x(vboxGlobalData *data, vboxIID_v2_x *iid)
{
    if (iid->value == NULL) {
426 427 428
        return;
    }

429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450
    if (iid->value != &iid->backing) {
        data->pFuncs->pfnComUnallocMem(iid->value);
    }

    iid->value = NULL;
}

static void
vboxIIDToUUID_v2_x(vboxIID_v2_x *iid, unsigned char *uuid)
{
    nsIDtoChar(uuid, iid->value);
}

static void
vboxIIDFromUUID_v2_x(vboxGlobalData *data, vboxIID_v2_x *iid,
                     const unsigned char *uuid)
{
    vboxIIDUnalloc_v2_x(data, iid);

    iid->value = &iid->backing;

    nsIDFromChar(iid->value, uuid);
451 452
}

453 454 455 456 457 458 459 460 461 462 463 464 465 466 467 468 469 470 471 472 473 474 475 476 477 478 479
static bool
vboxIIDIsEqual_v2_x(vboxIID_v2_x *iid1, vboxIID_v2_x *iid2)
{
    return memcmp(iid1->value, iid2->value, sizeof (nsID)) == 0;
}

static void
vboxIIDFromArrayItem_v2_x(vboxGlobalData *data, vboxIID_v2_x *iid,
                          vboxArray *array, int idx)
{
    vboxIIDUnalloc_v2_x(data, iid);

    iid->value = &iid->backing;

    memcpy(iid->value, array->items[idx], sizeof (nsID));
}

#  define vboxIIDUnalloc(iid) vboxIIDUnalloc_v2_x(data, iid)
#  define vboxIIDToUUID(iid, uuid) vboxIIDToUUID_v2_x(iid, uuid)
#  define vboxIIDFromUUID(iid, uuid) vboxIIDFromUUID_v2_x(data, iid, uuid)
#  define vboxIIDIsEqual(iid1, iid2) vboxIIDIsEqual_v2_x(iid1, iid2)
#  define vboxIIDFromArrayItem(iid, array, idx) \
    vboxIIDFromArrayItem_v2_x(data, iid, array, idx)
#  define DEBUGIID(msg, iid) DEBUGUUID(msg, iid)

# endif /* !WIN32 */

480
#else /* VBOX_API_VERSION != 2002 */
481

482 483 484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503
typedef struct _vboxIID_v3_x vboxIID;
typedef struct _vboxIID_v3_x vboxIID_v3_x;

struct _vboxIID_v3_x {
    /* IID is represented by a UTF-16 encoded UUID in string form. */
    PRUnichar *value;

    /* owner indicates if we own the value and need to free it. */
    bool owner;
};

# define VBOX_IID_INITIALIZER { NULL, true }

static void
vboxIIDUnalloc_v3_x(vboxGlobalData *data, vboxIID_v3_x *iid)
{
    if (iid->value != NULL && iid->owner) {
        data->pFuncs->pfnUtf16Free(iid->value);
    }

    iid->value = NULL;
    iid->owner = true;
504 505
}

506 507 508 509 510 511 512 513 514 515 516
static void
vboxIIDToUUID_v3_x(vboxGlobalData *data, vboxIID_v3_x *iid,
                   unsigned char *uuid)
{
    char *utf8 = NULL;

    data->pFuncs->pfnUtf16ToUtf8(iid->value, &utf8);

    virUUIDParse(utf8, uuid);

    data->pFuncs->pfnUtf8Free(utf8);
517 518
}

519 520 521 522 523
static void
vboxIIDFromUUID_v3_x(vboxGlobalData *data, vboxIID_v3_x *iid,
                     const unsigned char *uuid)
{
    char utf8[VIR_UUID_STRING_BUFLEN];
524

525
    vboxIIDUnalloc_v3_x(data, iid);
526

527
    virUUIDFormat(uuid, utf8);
528

529 530
    data->pFuncs->pfnUtf8ToUtf16(utf8, &iid->value);
}
531

532 533 534 535 536 537
static bool
vboxIIDIsEqual_v3_x(vboxGlobalData *data, vboxIID_v3_x *iid1,
                    vboxIID_v3_x *iid2)
{
    unsigned char uuid1[VIR_UUID_BUFLEN];
    unsigned char uuid2[VIR_UUID_BUFLEN];
538 539 540 541 542

    /* Note: we can't directly compare the utf8 strings here
     * cause the two UUID's may have seperators as space or '-'
     * or mixture of both and we don't want to fail here by
     * using direct string comparison. Here virUUIDParse() takes
543 544 545
     * care of these cases. */
    vboxIIDToUUID_v3_x(data, iid1, uuid1);
    vboxIIDToUUID_v3_x(data, iid2, uuid2);
546

547 548
    return memcmp(uuid1, uuid2, VIR_UUID_BUFLEN) == 0;
}
549 550


551 552 553 554 555
static void
vboxIIDFromArrayItem_v3_x(vboxGlobalData *data, vboxIID_v3_x *iid,
                          vboxArray *array, int idx)
{
    vboxIIDUnalloc_v3_x(data, iid);
556

557 558
    iid->value = array->items[idx];
    iid->owner = false;
559 560
}

561 562 563 564 565 566 567 568

# define vboxIIDUnalloc(iid) vboxIIDUnalloc_v3_x(data, iid)
# define vboxIIDToUUID(iid, uuid) vboxIIDToUUID_v3_x(data, iid, uuid)
# define vboxIIDFromUUID(iid, uuid) vboxIIDFromUUID_v3_x(data, iid, uuid)
# define vboxIIDIsEqual(iid1, iid2) vboxIIDIsEqual_v3_x(data, iid1, iid2)
# define vboxIIDFromArrayItem(iid, array, idx) \
    vboxIIDFromArrayItem_v3_x(data, iid, array, idx)
# define DEBUGIID(msg, strUtf16) DEBUGPRUnichar(msg, strUtf16)
569

570
# if VBOX_API_VERSION >= 3001
571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586

/**
 * function to generate the name for medium,
 * for e.g: hda, sda, etc
 *
 * @returns     null terminated string with device name or NULL
 *              for failures
 * @param       conn            Input Connection Pointer
 * @param       storageBus      Input storage bus type
 * @param       deviceInst      Input device instance number
 * @param       devicePort      Input port number
 * @param       deviceSlot      Input slot number
 * @param       aMaxPortPerInst Input array of max port per device instance
 * @param       aMaxSlotPerPort Input array of max slot per device port
 *
 */
587
static char *vboxGenerateMediumName(PRUint32  storageBus,
588 589 590 591 592
                                    PRInt32   deviceInst,
                                    PRInt32   devicePort,
                                    PRInt32   deviceSlot,
                                    PRUint32 *aMaxPortPerInst,
                                    PRUint32 *aMaxSlotPerPort) {
593
    const char *prefix = NULL;
594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613
    char *name  = NULL;
    int   total = 0;
    PRUint32 maxPortPerInst = 0;
    PRUint32 maxSlotPerPort = 0;

    if (   !aMaxPortPerInst
        || !aMaxSlotPerPort)
        return NULL;

    if (   (storageBus < StorageBus_IDE)
        || (storageBus > StorageBus_Floppy))
        return NULL;

    maxPortPerInst = aMaxPortPerInst[storageBus];
    maxSlotPerPort = aMaxSlotPerPort[storageBus];
    total =   (deviceInst * maxPortPerInst * maxSlotPerPort)
            + (devicePort * maxSlotPerPort)
            + deviceSlot;

    if (storageBus == StorageBus_IDE) {
614
        prefix = "hd";
615 616
    } else if (   (storageBus == StorageBus_SATA)
               || (storageBus == StorageBus_SCSI)) {
617
        prefix = "sd";
618
    } else if (storageBus == StorageBus_Floppy) {
619
        prefix = "fd";
620 621
    }

622
    name = virIndexToDiskName(total, prefix);
623

624
    VIR_DEBUG("name=%s, total=%d, storageBus=%u, deviceInst=%d, "
625
          "devicePort=%d deviceSlot=%d, maxPortPerInst=%u maxSlotPerPort=%u",
626
          NULLSTR(name), total, storageBus, deviceInst, devicePort,
627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680 681 682 683
          deviceSlot, maxPortPerInst, maxSlotPerPort);
    return name;
}

/**
 * function to get the StorageBus, Port number
 * and Device number for the given devicename
 * e.g: hda has StorageBus = IDE, port = 0,
 *      device = 0
 *
 * @returns     true on Success, false on failure.
 * @param       deviceName      Input device name
 * @param       aMaxPortPerInst Input array of max port per device instance
 * @param       aMaxSlotPerPort Input array of max slot per device port
 * @param       storageBus      Input storage bus type
 * @param       deviceInst      Output device instance number
 * @param       devicePort      Output port number
 * @param       deviceSlot      Output slot number
 *
 */
static bool vboxGetDeviceDetails(const char *deviceName,
                                 PRUint32   *aMaxPortPerInst,
                                 PRUint32   *aMaxSlotPerPort,
                                 PRUint32    storageBus,
                                 PRInt32    *deviceInst,
                                 PRInt32    *devicePort,
                                 PRInt32    *deviceSlot) {
    int total = 0;
    PRUint32 maxPortPerInst = 0;
    PRUint32 maxSlotPerPort = 0;

    if (   !deviceName
        || !deviceInst
        || !devicePort
        || !deviceSlot
        || !aMaxPortPerInst
        || !aMaxSlotPerPort)
        return false;

    if (   (storageBus < StorageBus_IDE)
        || (storageBus > StorageBus_Floppy))
        return false;

    total = virDiskNameToIndex(deviceName);

    maxPortPerInst = aMaxPortPerInst[storageBus];
    maxSlotPerPort = aMaxSlotPerPort[storageBus];

    if (   !maxPortPerInst
        || !maxSlotPerPort
        || (total < 0))
        return false;

    *deviceInst = total / (maxPortPerInst * maxSlotPerPort);
    *devicePort = (total % (maxPortPerInst * maxSlotPerPort)) / maxSlotPerPort;
    *deviceSlot = (total % (maxPortPerInst * maxSlotPerPort)) % maxSlotPerPort;

684
    VIR_DEBUG("name=%s, total=%d, storageBus=%u, deviceInst=%d, "
685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760
          "devicePort=%d deviceSlot=%d, maxPortPerInst=%u maxSlotPerPort=%u",
          deviceName, total, storageBus, *deviceInst, *devicePort,
          *deviceSlot, maxPortPerInst, maxSlotPerPort);

    return true;
}

/**
 * function to get the values for max port per
 * instance and max slots per port for the devices
 *
 * @returns     true on Success, false on failure.
 * @param       vbox            Input IVirtualBox pointer
 * @param       maxPortPerInst  Output array of max port per instance
 * @param       maxSlotPerPort  Output array of max slot per port
 *
 */

static bool vboxGetMaxPortSlotValues(IVirtualBox *vbox,
                                     PRUint32 *maxPortPerInst,
                                     PRUint32 *maxSlotPerPort) {
    ISystemProperties *sysProps = NULL;

    if (!vbox)
        return false;

    vbox->vtbl->GetSystemProperties(vbox, &sysProps);

    if (!sysProps)
        return false;

    sysProps->vtbl->GetMaxPortCountForStorageBus(sysProps,
                                                 StorageBus_IDE,
                                                 &maxPortPerInst[StorageBus_IDE]);
    sysProps->vtbl->GetMaxPortCountForStorageBus(sysProps,
                                                 StorageBus_SATA,
                                                 &maxPortPerInst[StorageBus_SATA]);
    sysProps->vtbl->GetMaxPortCountForStorageBus(sysProps,
                                                 StorageBus_SCSI,
                                                 &maxPortPerInst[StorageBus_SCSI]);
    sysProps->vtbl->GetMaxPortCountForStorageBus(sysProps,
                                                 StorageBus_Floppy,
                                                 &maxPortPerInst[StorageBus_Floppy]);

    sysProps->vtbl->GetMaxDevicesPerPortForStorageBus(sysProps,
                                                      StorageBus_IDE,
                                                      &maxSlotPerPort[StorageBus_IDE]);
    sysProps->vtbl->GetMaxDevicesPerPortForStorageBus(sysProps,
                                                      StorageBus_SATA,
                                                      &maxSlotPerPort[StorageBus_SATA]);
    sysProps->vtbl->GetMaxDevicesPerPortForStorageBus(sysProps,
                                                      StorageBus_SCSI,
                                                      &maxSlotPerPort[StorageBus_SCSI]);
    sysProps->vtbl->GetMaxDevicesPerPortForStorageBus(sysProps,
                                                      StorageBus_Floppy,
                                                      &maxSlotPerPort[StorageBus_Floppy]);

    VBOX_RELEASE(sysProps);

    return true;
}

/**
 * Converts Utf-16 string to int
 */
static int PRUnicharToInt(PRUnichar *strUtf16) {
    char *strUtf8 = NULL;
    int ret = 0;

    if (!strUtf16)
        return -1;

    g_pVBoxGlobalData->pFuncs->pfnUtf16ToUtf8(strUtf16, &strUtf8);
    if (!strUtf8)
        return -1;

761 762 763
    if (virStrToLong_i(strUtf8, NULL, 10, &ret) < 0)
        ret = -1;

764 765 766 767 768 769 770 771 772 773 774 775 776 777 778 779 780 781 782
    g_pVBoxGlobalData->pFuncs->pfnUtf8Free(strUtf8);

    return ret;
}

/**
 * Converts int to Utf-16 string
 */
static PRUnichar *PRUnicharFromInt(int n) {
    PRUnichar *strUtf16 = NULL;
    char s[24];

    snprintf(s, sizeof(s), "%d", n);

    g_pVBoxGlobalData->pFuncs->pfnUtf8ToUtf16(s, &strUtf16);

    return strUtf16;
}

783
# endif /* VBOX_API_VERSION >= 3001 */
784

785 786
#endif /* !(VBOX_API_VERSION == 2002) */

787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825
static PRUnichar *
vboxSocketFormatAddrUtf16(vboxGlobalData *data, virSocketAddrPtr addr)
{
    char *utf8 = NULL;
    PRUnichar *utf16 = NULL;

    utf8 = virSocketFormatAddr(addr);

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

    VBOX_UTF8_TO_UTF16(utf8, &utf16);
    VIR_FREE(utf8);

    return utf16;
}

static int
vboxSocketParseAddrUtf16(vboxGlobalData *data, const PRUnichar *utf16,
                         virSocketAddrPtr addr)
{
    int result = -1;
    char *utf8 = NULL;

    VBOX_UTF16_TO_UTF8(utf16, &utf8);

    if (virSocketParseAddr(utf8, addr, AF_UNSPEC) < 0) {
        goto cleanup;
    }

    result = 0;

cleanup:
    VBOX_UTF8_FREE(utf8);

    return result;
}

826 827 828 829 830 831 832 833 834 835 836
static virCapsPtr vboxCapsInit(void) {
    struct utsname utsname;
    virCapsPtr caps;
    virCapsGuestPtr guest;

    uname(&utsname);

    if ((caps = virCapabilitiesNew(utsname.machine,
                                   0, 0)) == NULL)
        goto no_memory;

837
    if (nodeCapsInitNUMA(caps) < 0)
838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865
        goto no_memory;

    virCapabilitiesSetMacPrefix(caps, (unsigned char[]){ 0x08, 0x00, 0x27 });

    if ((guest = virCapabilitiesAddGuest(caps,
                                         "hvm",
                                         utsname.machine,
                                         sizeof(void *) * CHAR_BIT,
                                         NULL,
                                         NULL,
                                         0,
                                         NULL)) == NULL)
        goto no_memory;

    if (virCapabilitiesAddGuestDomain(guest,
                                      "vbox",
                                      NULL,
                                      NULL,
                                      0,
                                      NULL) == NULL)
        goto no_memory;
    return caps;

no_memory:
    virCapabilitiesFree(caps);
    return NULL;
}

866 867 868
static int
vboxInitialize(vboxGlobalData *data)
{
869 870 871
    data->pFuncs = g_pfnGetFunctions(VBOX_XPCOMC_VERSION);

    if (data->pFuncs == NULL)
872
        goto cleanup;
873 874 875

#if VBOX_XPCOMC_VERSION == 0x00010000U
    data->pFuncs->pfnComInitialize(&data->vboxObj, &data->vboxSession);
876
#else  /* !(VBOX_XPCOMC_VERSION == 0x00010000U) */
877 878
    data->pFuncs->pfnComInitialize(IVIRTUALBOX_IID_STR, &data->vboxObj,
                               ISESSION_IID_STR, &data->vboxSession);
879

880
# if VBOX_API_VERSION == 2002
881 882 883

    /* No event queue functionality in 2.2.* as of now */

884
# else  /* !(VBOX_API_VERSION == 2002) */
885 886 887 888 889 890 891

    /* Initial the fWatch needed for Event Callbacks */
    data->fdWatch = -1;

    data->pFuncs->pfnGetEventQueue(&data->vboxQueue);

    if (data->vboxQueue == NULL) {
892 893
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("nsIEventQueue object is null"));
894 895 896
        goto cleanup;
    }

897
# endif /* !(VBOX_API_VERSION == 2002) */
898
#endif /* !(VBOX_XPCOMC_VERSION == 0x00010000U) */
899 900

    if (data->vboxObj == NULL) {
901 902
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("IVirtualBox object is null"));
903
        goto cleanup;
904 905 906
    }

    if (data->vboxSession == NULL) {
907 908
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("ISession object is null"));
909 910
        goto cleanup;
    }
911 912 913 914 915 916 917

    return 0;

cleanup:
    return -1;
}

918
static int vboxExtractVersion(vboxGlobalData *data) {
919
    int ret = -1;
920 921 922 923 924 925 926 927 928 929
    PRUnichar *versionUtf16 = NULL;
    nsresult rc;

    if (data->version > 0)
        return 0;

    rc = data->vboxObj->vtbl->GetVersion(data->vboxObj, &versionUtf16);
    if (NS_SUCCEEDED(rc)) {
        char *vboxVersion = NULL;

930
        VBOX_UTF16_TO_UTF8(versionUtf16, &vboxVersion);
931

932
        if (virParseVersionString(vboxVersion, &data->version, false) >= 0)
933 934
            ret = 0;

935 936
        VBOX_UTF8_FREE(vboxVersion);
        VBOX_COM_UNALLOC_MEM(versionUtf16);
937 938 939
    }

    if (ret != 0)
940 941
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
                  _("Cound not extract VirtualBox version"));
942

943 944 945 946
    return ret;
}

static void vboxUninitialize(vboxGlobalData *data) {
947 948 949
    if (!data)
        return;

950 951
    if (data->pFuncs)
        data->pFuncs->pfnComUninitialize();
952 953

    virCapabilitiesFree(data->caps);
954 955 956 957 958
#if VBOX_API_VERSION == 2002
    /* No domainEventCallbacks in 2.2.* version */
#else  /* !(VBOX_API_VERSION == 2002) */
    VIR_FREE(data->domainEventCallbacks);
#endif /* !(VBOX_API_VERSION == 2002) */
959 960 961 962 963
    VIR_FREE(data);
}

static virDrvOpenStatus vboxOpen(virConnectPtr conn,
                                 virConnectAuthPtr auth ATTRIBUTE_UNUSED,
964
                                 unsigned int flags ATTRIBUTE_UNUSED) {
965
    vboxGlobalData *data = NULL;
966 967 968 969 970
    uid_t uid = getuid();

    if (conn->uri == NULL) {
        conn->uri = xmlParseURI(uid ? "vbox:///session" : "vbox:///system");
        if (conn->uri == NULL) {
971
            virReportOOMError();
972 973 974 975
            return VIR_DRV_OPEN_ERROR;
        }
    }

976 977 978 979 980 981
    if (conn->uri->scheme == NULL ||
        STRNEQ (conn->uri->scheme, "vbox"))
        return VIR_DRV_OPEN_DECLINED;

    /* Leave for remote driver */
    if (conn->uri->server != NULL)
982 983
        return VIR_DRV_OPEN_DECLINED;

984
    if (conn->uri->path == NULL || STREQ(conn->uri->path, "")) {
985
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
986 987 988 989
                  _("no VirtualBox driver path specified (try vbox:///session)"));
        return VIR_DRV_OPEN_ERROR;
    }

990
    if (uid != 0) {
991
        if (STRNEQ (conn->uri->path, "/session")) {
992
            vboxError(VIR_ERR_INTERNAL_ERROR,
993 994 995
                      _("unknown driver path '%s' specified (try vbox:///session)"), conn->uri->path);
            return VIR_DRV_OPEN_ERROR;
        }
996 997
    } else { /* root */
        if (STRNEQ (conn->uri->path, "/system") &&
998
            STRNEQ (conn->uri->path, "/session")) {
999
            vboxError(VIR_ERR_INTERNAL_ERROR,
1000 1001 1002
                      _("unknown driver path '%s' specified (try vbox:///system)"), conn->uri->path);
            return VIR_DRV_OPEN_ERROR;
        }
1003 1004 1005
    }

    if (VIR_ALLOC(data) < 0) {
1006
        virReportOOMError();
1007
        return VIR_DRV_OPEN_ERROR;
1008 1009
    }

1010
    if (!(data->caps = vboxCapsInit()) ||
1011 1012
        vboxInitialize(data) < 0 ||
        vboxExtractVersion(data) < 0) {
1013 1014 1015
        vboxUninitialize(data);
        return VIR_DRV_OPEN_ERROR;
    }
1016

1017 1018 1019 1020 1021 1022 1023
#if VBOX_API_VERSION == 2002

    /* No domainEventCallbacks in 2.2.* version */

#else  /* !(VBOX_API_VERSION == 2002) */

    if (VIR_ALLOC(data->domainEventCallbacks) < 0) {
1024
        virReportOOMError();
1025 1026 1027 1028 1029 1030 1031 1032
        return VIR_DRV_OPEN_ERROR;
    }

    data->conn = conn;
    g_pVBoxGlobalData = data;

#endif /* !(VBOX_API_VERSION == 2002) */

1033
    conn->privateData = data;
1034
    VIR_DEBUG("in vboxOpen");
1035 1036 1037 1038 1039 1040

    return VIR_DRV_OPEN_SUCCESS;
}

static int vboxClose(virConnectPtr conn) {
    vboxGlobalData *data = conn->privateData;
1041
    VIR_DEBUG("%s: in vboxClose",conn->driver->name);
1042 1043 1044 1045 1046 1047 1048 1049 1050

    vboxUninitialize(data);
    conn->privateData = NULL;

    return 0;
}

static int vboxGetVersion(virConnectPtr conn, unsigned long *version) {
    vboxGlobalData *data = conn->privateData;
1051
    VIR_DEBUG("%s: in vboxGetVersion",conn->driver->name);
1052 1053 1054 1055 1056 1057 1058 1059

    vboxDriverLock(data);
    *version = data->version;
    vboxDriverUnlock(data);

    return 0;
}

1060 1061 1062 1063 1064 1065 1066 1067 1068 1069
static int vboxIsSecure(virConnectPtr conn ATTRIBUTE_UNUSED) {
    /* Driver is using local, non-network based transport */
    return 1;
}

static int vboxIsEncrypted(virConnectPtr conn ATTRIBUTE_UNUSED) {
    /* No encryption is needed, or used on the local transport*/
    return 0;
}

1070
static int vboxGetMaxVcpus(virConnectPtr conn, const char *type ATTRIBUTE_UNUSED) {
1071
    VBOX_OBJECT_CHECK(conn, int, -1);
1072 1073 1074 1075 1076
    PRUint32 maxCPUCount = 0;

    /* VirtualBox Supports only hvm and thus the type passed to it
     * has no meaning, setting it to ATTRIBUTE_UNUSED
     */
1077
    ISystemProperties *systemProperties = NULL;
1078

1079 1080 1081 1082
    data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
    if (systemProperties) {
        systemProperties->vtbl->GetMaxGuestCPUCount(systemProperties, &maxCPUCount);
        VBOX_RELEASE(systemProperties);
1083 1084 1085 1086 1087 1088 1089 1090 1091 1092
    }

    if (maxCPUCount > 0)
        ret = maxCPUCount;

    return ret;
}


static char *vboxGetCapabilities(virConnectPtr conn) {
1093
    VBOX_OBJECT_CHECK(conn, char *, NULL);
1094 1095 1096 1097 1098 1099 1100 1101 1102

    vboxDriverLock(data);
    ret = virCapabilitiesFormatXML(data->caps);
    vboxDriverUnlock(data);

    return ret;
}

static int vboxListDomains(virConnectPtr conn, int *ids, int nids) {
1103
    VBOX_OBJECT_CHECK(conn, int, -1);
1104
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1105
    PRUint32 state;
1106
    nsresult rc;
1107 1108
    int i, j;

1109
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1110
    if (NS_FAILED(rc)) {
1111 1112
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of Domains, rc=%08x"),(unsigned)rc);
1113 1114
        goto cleanup;
    }
1115

1116 1117 1118
    ret = 0;
    for (i = 0, j = 0; (i < machines.count) && (j < nids); ++i) {
        IMachine *machine = machines.items[i];
1119 1120 1121 1122 1123 1124 1125 1126 1127 1128

        if (machine) {
            PRBool isAccessible = PR_FALSE;
            machine->vtbl->GetAccessible(machine, &isAccessible);
            if (isAccessible) {
                machine->vtbl->GetState(machine, &state);
                if (   (state >= MachineState_FirstOnline)
                    && (state <= MachineState_LastOnline) ) {
                    ret++;
                    ids[j++] = i + 1;
1129 1130 1131 1132 1133 1134
                }
            }
        }
    }

cleanup:
1135
    vboxArrayRelease(&machines);
1136 1137 1138 1139
    return ret;
}

static int vboxNumOfDomains(virConnectPtr conn) {
1140
    VBOX_OBJECT_CHECK(conn, int, -1);
1141
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1142
    PRUint32 state;
1143
    nsresult rc;
1144 1145
    int i;

1146
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1147
    if (NS_FAILED(rc)) {
1148 1149
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get number of Domains, rc=%08x"), (unsigned)rc);
1150 1151
        goto cleanup;
    }
1152

1153 1154 1155
    ret = 0;
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
1156 1157 1158 1159 1160 1161 1162 1163 1164

        if (machine) {
            PRBool isAccessible = PR_FALSE;
            machine->vtbl->GetAccessible(machine, &isAccessible);
            if (isAccessible) {
                machine->vtbl->GetState(machine, &state);
                if (   (state >= MachineState_FirstOnline)
                    && (state <= MachineState_LastOnline) )
                    ret++;
1165 1166 1167 1168 1169
            }
        }
    }

cleanup:
1170
    vboxArrayRelease(&machines);
1171 1172 1173 1174
    return ret;
}

static virDomainPtr vboxDomainCreateXML(virConnectPtr conn, const char *xml,
1175
                                        unsigned int flags) {
1176 1177 1178 1179 1180 1181 1182 1183
    /* VirtualBox currently doesn't have support for running
     * virtual machines without actually defining them and thus
     * for time being just define new machine and start it.
     *
     * TODO: After the appropriate API's are added in VirtualBox
     * change this behaviour to the expected one.
     */

1184 1185 1186 1187 1188
    virDomainPtr dom;

    virCheckFlags(0, NULL);

    dom = vboxDomainDefineXML(conn, xml);
1189 1190 1191 1192 1193
    if (dom == NULL)
        return NULL;

    if (vboxDomainCreate(dom) < 0) {
        vboxDomainUndefine(dom);
1194
        virUnrefDomain(dom);
1195
        return NULL;
1196 1197 1198 1199 1200 1201
    }

    return dom;
}

static virDomainPtr vboxDomainLookupByID(virConnectPtr conn, int id) {
1202
    VBOX_OBJECT_CHECK(conn, virDomainPtr, NULL);
1203
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1204 1205
    vboxIID iid = VBOX_IID_INITIALIZER;
    unsigned char uuid[VIR_UUID_BUFLEN];
1206
    PRUint32 state;
1207
    nsresult rc;
1208

1209 1210 1211
    /* Internal vbox IDs start from 0, the public libvirt ID
     * starts from 1, so refuse id==0, and adjust the rest*/
    if (id == 0) {
1212
        vboxError(VIR_ERR_NO_DOMAIN,
1213 1214 1215 1216 1217
                  _("no domain with matching id %d"), id);
        return NULL;
    }
    id = id - 1;

1218
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1219
    if (NS_FAILED(rc)) {
1220 1221
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of machines, rc=%08x"), (unsigned)rc);
1222 1223
        return NULL;
    }
1224

1225 1226 1227 1228
    if (id < machines.count) {
        IMachine *machine = machines.items[id];

        if (machine) {
1229
            PRBool isAccessible = PR_FALSE;
1230
            machine->vtbl->GetAccessible(machine, &isAccessible);
1231
            if (isAccessible) {
1232
                machine->vtbl->GetState(machine, &state);
1233 1234 1235 1236
                if (   (state >= MachineState_FirstOnline)
                    && (state <= MachineState_LastOnline) ) {
                    PRUnichar *machineNameUtf16 = NULL;
                    char      *machineNameUtf8  = NULL;
1237

1238
                    machine->vtbl->GetName(machine, &machineNameUtf16);
1239
                    VBOX_UTF16_TO_UTF8(machineNameUtf16, &machineNameUtf8);
1240

1241 1242 1243
                    machine->vtbl->GetId(machine, &iid.value);
                    vboxIIDToUUID(&iid, uuid);
                    vboxIIDUnalloc(&iid);
1244 1245 1246 1247 1248 1249 1250

                    /* get a new domain pointer from virGetDomain, if it fails
                     * then no need to assign the id, else assign the id, cause
                     * it is -1 by default. rest is taken care by virGetDomain
                     * itself, so need not worry.
                     */

1251
                    ret = virGetDomain(conn, machineNameUtf8, uuid);
1252 1253 1254 1255 1256 1257
                    if (ret)
                        ret->id = id + 1;

                    /* Cleanup all the XPCOM allocated stuff here */
                    VBOX_UTF8_FREE(machineNameUtf8);
                    VBOX_UTF16_FREE(machineNameUtf16);
1258 1259 1260 1261 1262
                }
            }
        }
    }

1263
    vboxArrayRelease(&machines);
1264 1265

    return ret;
1266 1267 1268
}

static virDomainPtr vboxDomainLookupByUUID(virConnectPtr conn, const unsigned char *uuid) {
1269
    VBOX_OBJECT_CHECK(conn, virDomainPtr, NULL);
1270
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1271
    vboxIID iid = VBOX_IID_INITIALIZER;
1272
    char      *machineNameUtf8  = NULL;
1273
    PRUnichar *machineNameUtf16 = NULL;
1274
    unsigned char iid_as_uuid[VIR_UUID_BUFLEN];
1275
    int i, matched = 0;
1276
    nsresult rc;
1277

1278
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1279
    if (NS_FAILED(rc)) {
1280 1281
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of machines, rc=%08x"), (unsigned)rc);
1282 1283
        return NULL;
    }
1284

1285 1286
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
1287
        PRBool isAccessible = PR_FALSE;
1288

1289 1290
        if (!machine)
            continue;
1291

1292 1293
        machine->vtbl->GetAccessible(machine, &isAccessible);
        if (isAccessible) {
1294

1295 1296
            rc = machine->vtbl->GetId(machine, &iid.value);
            if (NS_FAILED(rc))
1297
                continue;
1298 1299
            vboxIIDToUUID(&iid, iid_as_uuid);
            vboxIIDUnalloc(&iid);
1300

1301
            if (memcmp(uuid, iid_as_uuid, VIR_UUID_BUFLEN) == 0) {
1302

1303
                PRUint32 state;
1304

1305
                matched = 1;
1306

1307 1308
                machine->vtbl->GetName(machine, &machineNameUtf16);
                VBOX_UTF16_TO_UTF8(machineNameUtf16, &machineNameUtf8);
1309

1310
                machine->vtbl->GetState(machine, &state);
1311

1312 1313 1314 1315 1316
                /* get a new domain pointer from virGetDomain, if it fails
                 * then no need to assign the id, else assign the id, cause
                 * it is -1 by default. rest is taken care by virGetDomain
                 * itself, so need not worry.
                 */
1317

1318
                ret = virGetDomain(conn, machineNameUtf8, iid_as_uuid);
1319 1320 1321 1322
                if (   ret
                    && (state >= MachineState_FirstOnline)
                    && (state <= MachineState_LastOnline) )
                    ret->id = i + 1;
1323 1324
            }

1325 1326
            if (matched == 1)
                break;
1327 1328 1329
        }
    }

1330 1331 1332
    /* Do the cleanup and take care you dont leak any memory */
    VBOX_UTF8_FREE(machineNameUtf8);
    VBOX_COM_UNALLOC_MEM(machineNameUtf16);
1333
    vboxArrayRelease(&machines);
1334 1335

    return ret;
1336 1337 1338
}

static virDomainPtr vboxDomainLookupByName(virConnectPtr conn, const char *name) {
1339
    VBOX_OBJECT_CHECK(conn, virDomainPtr, NULL);
1340
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1341
    vboxIID iid = VBOX_IID_INITIALIZER;
1342
    char      *machineNameUtf8  = NULL;
1343
    PRUnichar *machineNameUtf16 = NULL;
1344
    unsigned char uuid[VIR_UUID_BUFLEN];
1345
    int i, matched = 0;
1346
    nsresult rc;
1347

1348
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1349
    if (NS_FAILED(rc)) {
1350 1351
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of machines, rc=%08x"), (unsigned)rc);
1352 1353
        return NULL;
    }
1354

1355 1356
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
1357
        PRBool isAccessible = PR_FALSE;
1358

1359 1360
        if (!machine)
            continue;
1361

1362 1363
        machine->vtbl->GetAccessible(machine, &isAccessible);
        if (isAccessible) {
1364

1365 1366
            machine->vtbl->GetName(machine, &machineNameUtf16);
            VBOX_UTF16_TO_UTF8(machineNameUtf16, &machineNameUtf8);
1367

1368
            if (STREQ(name, machineNameUtf8)) {
1369

1370
                PRUint32 state;
1371

1372
                matched = 1;
1373

1374 1375 1376
                machine->vtbl->GetId(machine, &iid.value);
                vboxIIDToUUID(&iid, uuid);
                vboxIIDUnalloc(&iid);
1377

1378
                machine->vtbl->GetState(machine, &state);
1379

1380 1381 1382 1383 1384
                /* get a new domain pointer from virGetDomain, if it fails
                 * then no need to assign the id, else assign the id, cause
                 * it is -1 by default. rest is taken care by virGetDomain
                 * itself, so need not worry.
                 */
1385

1386
                ret = virGetDomain(conn, machineNameUtf8, uuid);
1387 1388 1389 1390
                if (   ret
                    && (state >= MachineState_FirstOnline)
                    && (state <= MachineState_LastOnline) )
                    ret->id = i + 1;
1391 1392
            }

1393 1394 1395 1396 1397 1398 1399 1400 1401 1402
            if (machineNameUtf8) {
                VBOX_UTF8_FREE(machineNameUtf8);
                machineNameUtf8 = NULL;
            }
            if (machineNameUtf16) {
                VBOX_COM_UNALLOC_MEM(machineNameUtf16);
                machineNameUtf16 = NULL;
            }
            if (matched == 1)
                break;
1403 1404 1405
        }
    }

1406
    vboxArrayRelease(&machines);
1407 1408

    return ret;
1409 1410
}

1411 1412

static int vboxDomainIsActive(virDomainPtr dom) {
1413
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1414
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1415
    vboxIID iid = VBOX_IID_INITIALIZER;
1416 1417
    char      *machineNameUtf8  = NULL;
    PRUnichar *machineNameUtf16 = NULL;
1418
    unsigned char uuid[VIR_UUID_BUFLEN];
1419
    int i, matched = 0;
1420
    nsresult rc;
1421

1422
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1423
    if (NS_FAILED(rc)) {
1424 1425
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of machines, rc=%08x"), (unsigned)rc);
1426 1427
        return ret;
    }
1428

1429 1430
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
1431 1432 1433 1434 1435 1436 1437
        PRBool isAccessible = PR_FALSE;

        if (!machine)
            continue;

        machine->vtbl->GetAccessible(machine, &isAccessible);
        if (isAccessible) {
1438

1439 1440
            rc = machine->vtbl->GetId(machine, &iid.value);
            if (NS_FAILED(rc))
1441
                continue;
1442 1443
            vboxIIDToUUID(&iid, uuid);
            vboxIIDUnalloc(&iid);
1444

1445
            if (memcmp(dom->uuid, uuid, VIR_UUID_BUFLEN) == 0) {
1446

1447
                PRUint32 state;
1448

1449
                matched = 1;
1450

1451 1452
                machine->vtbl->GetName(machine, &machineNameUtf16);
                VBOX_UTF16_TO_UTF8(machineNameUtf16, &machineNameUtf8);
1453

1454
                machine->vtbl->GetState(machine, &state);
1455

1456 1457 1458 1459 1460
                if (   (state >= MachineState_FirstOnline)
                    && (state <= MachineState_LastOnline) )
                    ret = 1;
                else
                    ret = 0;
1461 1462
            }

1463 1464
            if (matched == 1)
                break;
1465 1466 1467
        }
    }

1468 1469 1470
    /* Do the cleanup and take care you dont leak any memory */
    VBOX_UTF8_FREE(machineNameUtf8);
    VBOX_COM_UNALLOC_MEM(machineNameUtf16);
1471
    vboxArrayRelease(&machines);
1472

1473 1474 1475 1476 1477
    return ret;
}


static int vboxDomainIsPersistent(virDomainPtr dom ATTRIBUTE_UNUSED) {
1478
    /* All domains are persistent. */
1479 1480 1481 1482
    return 1;
}


1483 1484 1485 1486
static int vboxDomainIsUpdated(virDomainPtr dom ATTRIBUTE_UNUSED) {
    return 0;
}

1487
static int vboxDomainSuspend(virDomainPtr dom) {
1488
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1489
    IMachine *machine    = NULL;
1490
    vboxIID iid = VBOX_IID_INITIALIZER;
1491
    IConsole *console    = NULL;
1492
    PRBool isAccessible  = PR_FALSE;
1493
    PRUint32 state;
1494
    nsresult rc;
1495

1496
    vboxIIDFromUUID(&iid, dom->uuid);
1497
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
1498
    if (NS_FAILED(rc)) {
1499
        vboxError(VIR_ERR_NO_DOMAIN,
1500
                  _("no domain with matching id %d"), dom->id);
1501 1502
        goto cleanup;
    }
1503

1504 1505
    if (!machine)
        goto cleanup;
1506

1507 1508 1509
    machine->vtbl->GetAccessible(machine, &isAccessible);
    if (isAccessible) {
        machine->vtbl->GetState(machine, &state);
1510

1511
        if (state == MachineState_Running) {
1512 1513
            /* set state pause */
            VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
1514 1515 1516 1517 1518
            data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
            if (console) {
                console->vtbl->Pause(console);
                VBOX_RELEASE(console);
                ret = 0;
1519
            } else {
1520 1521
                vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                          _("error while suspending the domain"));
1522 1523
                goto cleanup;
            }
1524
            VBOX_SESSION_CLOSE();
1525
        } else {
1526 1527
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("machine not in running state to suspend it"));
1528
            goto cleanup;
1529 1530 1531 1532
        }
    }

cleanup:
1533
    VBOX_RELEASE(machine);
1534
    vboxIIDUnalloc(&iid);
1535 1536 1537 1538
    return ret;
}

static int vboxDomainResume(virDomainPtr dom) {
1539
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1540
    IMachine *machine    = NULL;
1541
    vboxIID iid = VBOX_IID_INITIALIZER;
1542 1543
    IConsole *console    = NULL;
    PRUint32 state       = MachineState_Null;
1544
    nsresult rc;
1545

1546
    PRBool isAccessible = PR_FALSE;
1547

1548
    vboxIIDFromUUID(&iid, dom->uuid);
1549
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
1550
    if (NS_FAILED(rc)) {
1551
        vboxError(VIR_ERR_NO_DOMAIN,
1552
                  _("no domain with matching id %d"), dom->id);
1553 1554
        goto cleanup;
    }
1555

1556 1557
    if (!machine)
        goto cleanup;
1558

1559 1560 1561 1562 1563
    machine->vtbl->GetAccessible(machine, &isAccessible);
    if (isAccessible) {
        machine->vtbl->GetState(machine, &state);

        if (state == MachineState_Paused) {
1564 1565
            /* resume the machine here */
            VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
1566 1567 1568 1569 1570
            data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
            if (console) {
                console->vtbl->Resume(console);
                VBOX_RELEASE(console);
                ret = 0;
1571
            } else {
1572 1573
                vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                          _("error while resuming the domain"));
1574 1575
                goto cleanup;
            }
1576
            VBOX_SESSION_CLOSE();
1577
        } else {
1578 1579
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("machine not paused, so can't resume it"));
1580
            goto cleanup;
1581 1582 1583 1584
        }
    }

cleanup:
1585
    VBOX_RELEASE(machine);
1586
    vboxIIDUnalloc(&iid);
1587 1588 1589 1590
    return ret;
}

static int vboxDomainShutdown(virDomainPtr dom) {
1591
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1592
    IMachine *machine    = NULL;
1593
    vboxIID iid = VBOX_IID_INITIALIZER;
1594 1595
    IConsole *console    = NULL;
    PRUint32 state       = MachineState_Null;
1596 1597
    PRBool isAccessible  = PR_FALSE;
    nsresult rc;
1598

1599
    vboxIIDFromUUID(&iid, dom->uuid);
1600
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
1601
    if (NS_FAILED(rc)) {
1602
        vboxError(VIR_ERR_NO_DOMAIN,
1603
                  _("no domain with matching id %d"), dom->id);
1604 1605
        goto cleanup;
    }
1606

1607 1608
    if (!machine)
        goto cleanup;
1609

1610 1611 1612
    machine->vtbl->GetAccessible(machine, &isAccessible);
    if (isAccessible) {
        machine->vtbl->GetState(machine, &state);
1613

1614
        if (state == MachineState_Paused) {
1615 1616
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("machine paused, so can't power it down"));
1617 1618
            goto cleanup;
        } else if (state == MachineState_PoweredOff) {
1619 1620
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("machine already powered down"));
1621 1622
            goto cleanup;
        }
1623

1624
        VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
1625 1626 1627 1628 1629
        data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
        if (console) {
            console->vtbl->PowerButton(console);
            VBOX_RELEASE(console);
            ret = 0;
1630
        }
1631
        VBOX_SESSION_CLOSE();
1632 1633 1634
    }

cleanup:
1635
    VBOX_RELEASE(machine);
1636
    vboxIIDUnalloc(&iid);
1637 1638 1639 1640
    return ret;
}

static int vboxDomainReboot(virDomainPtr dom, unsigned int flags ATTRIBUTE_UNUSED) {
1641
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1642
    IMachine *machine    = NULL;
1643
    vboxIID iid = VBOX_IID_INITIALIZER;
1644 1645
    IConsole *console    = NULL;
    PRUint32 state       = MachineState_Null;
1646 1647
    PRBool isAccessible  = PR_FALSE;
    nsresult rc;
1648

1649
    vboxIIDFromUUID(&iid, dom->uuid);
1650
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
1651
    if (NS_FAILED(rc)) {
1652
        vboxError(VIR_ERR_NO_DOMAIN,
1653
                  _("no domain with matching id %d"), dom->id);
1654 1655
        goto cleanup;
    }
1656

1657 1658
    if (!machine)
        goto cleanup;
1659

1660 1661 1662
    machine->vtbl->GetAccessible(machine, &isAccessible);
    if (isAccessible) {
        machine->vtbl->GetState(machine, &state);
1663

1664
        if (state == MachineState_Running) {
1665
            VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
1666 1667 1668 1669 1670
            data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
            if (console) {
                console->vtbl->Reset(console);
                VBOX_RELEASE(console);
                ret = 0;
1671
            }
1672
            VBOX_SESSION_CLOSE();
1673
        } else {
1674 1675
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("machine not running, so can't reboot it"));
1676
            goto cleanup;
1677 1678 1679 1680
        }
    }

cleanup:
1681
    VBOX_RELEASE(machine);
1682
    vboxIIDUnalloc(&iid);
1683 1684 1685 1686
    return ret;
}

static int vboxDomainDestroy(virDomainPtr dom) {
1687
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1688
    IMachine *machine    = NULL;
1689
    vboxIID iid = VBOX_IID_INITIALIZER;
1690 1691
    IConsole *console    = NULL;
    PRUint32 state       = MachineState_Null;
1692 1693
    PRBool isAccessible  = PR_FALSE;
    nsresult rc;
1694

1695
    vboxIIDFromUUID(&iid, dom->uuid);
1696
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
1697
    if (NS_FAILED(rc)) {
1698
        vboxError(VIR_ERR_NO_DOMAIN,
1699
                  _("no domain with matching id %d"), dom->id);
1700 1701
        goto cleanup;
    }
1702

1703 1704
    if (!machine)
        goto cleanup;
1705

1706 1707 1708
    machine->vtbl->GetAccessible(machine, &isAccessible);
    if (isAccessible) {
        machine->vtbl->GetState(machine, &state);
1709

1710
        if (state == MachineState_PoweredOff) {
1711 1712
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("machine already powered down"));
1713 1714
            goto cleanup;
        }
1715

1716
        VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
1717 1718
        data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
        if (console) {
1719 1720

#if VBOX_API_VERSION == 2002
1721
            console->vtbl->PowerDown(console);
1722
#else
1723
            IProgress *progress = NULL;
1724 1725 1726 1727
            console->vtbl->PowerDown(console, &progress);
            if (progress) {
                progress->vtbl->WaitForCompletion(progress, -1);
                VBOX_RELEASE(progress);
1728
            }
1729 1730
#endif
            VBOX_RELEASE(console);
1731
            dom->id = -1;
1732
            ret = 0;
1733
        }
1734
        VBOX_SESSION_CLOSE();
1735 1736 1737
    }

cleanup:
1738
    VBOX_RELEASE(machine);
1739
    vboxIIDUnalloc(&iid);
1740 1741 1742
    return ret;
}

1743
static char *vboxDomainGetOSType(virDomainPtr dom ATTRIBUTE_UNUSED) {
1744 1745 1746 1747 1748
    /* Returning "hvm" always as suggested on list, cause
     * this functions seems to be badly named and it
     * is supposed to pass the ABI name and not the domain
     * operating system driver as I had imagined ;)
     */
1749 1750 1751
    char *osType = strdup("hvm");

    if (osType == NULL)
1752
        virReportOOMError();
1753 1754

    return osType;
1755 1756 1757
}

static int vboxDomainSetMemory(virDomainPtr dom, unsigned long memory) {
1758
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1759
    IMachine *machine    = NULL;
1760
    vboxIID iid = VBOX_IID_INITIALIZER;
1761
    PRUint32 state       = MachineState_Null;
1762 1763
    PRBool isAccessible  = PR_FALSE;
    nsresult rc;
1764

1765
    vboxIIDFromUUID(&iid, dom->uuid);
1766
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
1767
    if (NS_FAILED(rc)) {
1768
        vboxError(VIR_ERR_NO_DOMAIN,
1769
                  _("no domain with matching id %d"), dom->id);
1770 1771
        goto cleanup;
    }
1772

1773 1774
    if (!machine)
        goto cleanup;
1775

1776 1777 1778
    machine->vtbl->GetAccessible(machine, &isAccessible);
    if (isAccessible) {
        machine->vtbl->GetState(machine, &state);
1779

1780
        if (state != MachineState_PoweredOff) {
1781 1782
            vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                      _("memory size can't be changed unless domain is powered down"));
1783 1784
            goto cleanup;
        }
1785

1786
        rc = VBOX_SESSION_OPEN(iid.value, machine);
1787 1788 1789
        if (NS_SUCCEEDED(rc)) {
            rc = data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);
            if (NS_SUCCEEDED(rc) && machine) {
1790

1791 1792
                rc = machine->vtbl->SetMemorySize(machine,
                                                  VIR_DIV_UP(memory, 1024));
1793 1794 1795 1796
                if (NS_SUCCEEDED(rc)) {
                    machine->vtbl->SaveSettings(machine);
                    ret = 0;
                } else {
1797 1798 1799
                    vboxError(VIR_ERR_INTERNAL_ERROR,
                              _("could not set the memory size of the "
                                "domain to: %lu Kb, rc=%08x"),
1800
                              memory, (unsigned)rc);
1801 1802
                }
            }
1803
            VBOX_SESSION_CLOSE();
1804 1805 1806 1807
        }
    }

cleanup:
1808
    VBOX_RELEASE(machine);
1809
    vboxIIDUnalloc(&iid);
1810 1811 1812 1813
    return ret;
}

static int vboxDomainGetInfo(virDomainPtr dom, virDomainInfoPtr info) {
1814
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1815
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
1816 1817
    char *machineName    = NULL;
    PRUnichar *machineNameUtf16 = NULL;
1818 1819
    nsresult rc;
    int i = 0;
1820

1821
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
1822
    if (NS_FAILED(rc)) {
1823 1824
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of machines, rc=%08x"), (unsigned)rc);
1825 1826
        goto cleanup;
    }
1827

1828
    info->nrVirtCpu = 0;
1829 1830
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
1831
        PRBool isAccessible = PR_FALSE;
1832

1833 1834
        if (!machine)
            continue;
1835

1836 1837
        machine->vtbl->GetAccessible(machine, &isAccessible);
        if (isAccessible) {
1838

1839 1840 1841 1842 1843 1844
            machine->vtbl->GetName(machine, &machineNameUtf16);
            VBOX_UTF16_TO_UTF8(machineNameUtf16, &machineName);

            if (STREQ(dom->name, machineName)) {
                /* Get the Machine State (also match it with
                * virDomainState). Get the Machine memory and
1845
                * for time being set max_balloon and cur_balloon to same
1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862
                * Also since there is no direct way of checking
                * the cputime required (one condition being the
                * VM is remote), return zero for cputime. Get the
                * number of CPU.
                */
                PRUint32 CPUCount   = 0;
                PRUint32 memorySize = 0;
                PRUint32 state      = MachineState_Null;
                PRUint32 maxMemorySize = 4 * 1024;
                ISystemProperties *systemProperties = NULL;

                data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
                if (systemProperties) {
                    systemProperties->vtbl->GetMaxGuestRAM(systemProperties, &maxMemorySize);
                    VBOX_RELEASE(systemProperties);
                    systemProperties = NULL;
                }
1863 1864


1865 1866 1867 1868 1869 1870
                machine->vtbl->GetCPUCount(machine, &CPUCount);
                machine->vtbl->GetMemorySize(machine, &memorySize);
                machine->vtbl->GetState(machine, &state);

                info->cpuTime = 0;
                info->nrVirtCpu = CPUCount;
1871 1872
                info->memory = memorySize * 1024;
                info->maxMem = maxMemorySize * 1024;
1873 1874 1875 1876 1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895
                switch(state) {
                    case MachineState_Running:
                        info->state = VIR_DOMAIN_RUNNING;
                        break;
                    case MachineState_Stuck:
                        info->state = VIR_DOMAIN_BLOCKED;
                        break;
                    case MachineState_Paused:
                        info->state = VIR_DOMAIN_PAUSED;
                        break;
                    case MachineState_Stopping:
                        info->state = VIR_DOMAIN_SHUTDOWN;
                        break;
                    case MachineState_PoweredOff:
                        info->state = VIR_DOMAIN_SHUTOFF;
                        break;
                    case MachineState_Aborted:
                        info->state = VIR_DOMAIN_CRASHED;
                        break;
                    case MachineState_Null:
                    default:
                        info->state = VIR_DOMAIN_NOSTATE;
                        break;
1896 1897
                }

1898
                ret = 0;
1899 1900
            }

1901 1902 1903 1904 1905 1906
            if (machineName)
                VBOX_UTF8_FREE(machineName);
            if (machineNameUtf16)
                VBOX_COM_UNALLOC_MEM(machineNameUtf16);
            if (info->nrVirtCpu)
                break;
1907 1908 1909 1910
        }

    }

1911
    vboxArrayRelease(&machines);
1912 1913 1914 1915 1916

cleanup:
    return ret;
}

1917 1918 1919 1920 1921 1922 1923 1924 1925 1926 1927 1928 1929 1930 1931 1932 1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966 1967 1968 1969 1970 1971 1972 1973 1974 1975
static int
vboxDomainGetState(virDomainPtr dom,
                   int *state,
                   int *reason,
                   unsigned int flags)
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
    vboxIID domiid = VBOX_IID_INITIALIZER;
    IMachine *machine = NULL;
    PRUint32 mstate = MachineState_Null;
    nsresult rc;

    virCheckFlags(0, -1);

    vboxIIDFromUUID(&domiid, dom->uuid);
    rc = VBOX_OBJECT_GET_MACHINE(domiid.value, &machine);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    machine->vtbl->GetState(machine, &mstate);

    switch (mstate) {
    case MachineState_Running:
        *state = VIR_DOMAIN_RUNNING;
        break;
    case MachineState_Stuck:
        *state = VIR_DOMAIN_BLOCKED;
        break;
    case MachineState_Paused:
        *state = VIR_DOMAIN_PAUSED;
        break;
    case MachineState_Stopping:
        *state = VIR_DOMAIN_SHUTDOWN;
        break;
    case MachineState_PoweredOff:
        *state = VIR_DOMAIN_SHUTOFF;
        break;
    case MachineState_Aborted:
        *state = VIR_DOMAIN_CRASHED;
        break;
    case MachineState_Null:
    default:
        *state = VIR_DOMAIN_NOSTATE;
        break;
    }

    if (reason)
        *reason = 0;

    ret = 0;

cleanup:
    vboxIIDUnalloc(&domiid);
    return ret;
}

1976
static int vboxDomainSave(virDomainPtr dom, const char *path ATTRIBUTE_UNUSED) {
1977
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
1978
    IConsole *console    = NULL;
1979
    vboxIID iid = VBOX_IID_INITIALIZER;
1980
    IMachine *machine = NULL;
1981
    nsresult rc;
1982 1983 1984 1985 1986 1987 1988 1989

    /* VirtualBox currently doesn't support saving to a file
     * at a location other then the machine folder and thus
     * setting path to ATTRIBUTE_UNUSED for now, will change
     * this behaviour once get the VirtualBox API in right
     * shape to do this
     */

1990
    /* Open a Session for the machine */
1991
    vboxIIDFromUUID(&iid, dom->uuid);
1992 1993 1994 1995 1996 1997 1998 1999 2000 2001 2002
#if VBOX_API_VERSION >= 4000
    /* Get machine for the call to VBOX_SESSION_OPEN_EXISTING */
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
                  _("no domain with matching uuid"));
        return -1;
    }
#endif

    rc = VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
2003 2004 2005 2006
    if (NS_SUCCEEDED(rc)) {
        rc = data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
        if (NS_SUCCEEDED(rc) && console) {
            IProgress *progress = NULL;
2007

2008
            console->vtbl->SaveState(console, &progress);
2009

2010
            if (progress) {
2011
#if VBOX_API_VERSION == 2002
2012
                nsresult resultCode;
2013
#else
2014
                PRInt32 resultCode;
2015
#endif
2016

2017 2018 2019 2020
                progress->vtbl->WaitForCompletion(progress, -1);
                progress->vtbl->GetResultCode(progress, &resultCode);
                if (NS_SUCCEEDED(resultCode)) {
                    ret = 0;
2021
                }
2022
                VBOX_RELEASE(progress);
2023
            }
2024
            VBOX_RELEASE(console);
2025
        }
2026
        VBOX_SESSION_CLOSE();
2027 2028
    }

2029
    DEBUGIID("UUID of machine being saved:", iid.value);
2030

2031
    VBOX_RELEASE(machine);
2032
    vboxIIDUnalloc(&iid);
2033 2034 2035
    return ret;
}

2036 2037 2038 2039
static int
vboxDomainSetVcpusFlags(virDomainPtr dom, unsigned int nvcpus,
                        unsigned int flags)
{
2040
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
2041
    IMachine *machine    = NULL;
2042
    vboxIID iid = VBOX_IID_INITIALIZER;
2043
    PRUint32  CPUCount   = nvcpus;
2044
    nsresult rc;
2045

2046
    if (flags != VIR_DOMAIN_AFFECT_LIVE) {
2047 2048 2049 2050
        vboxError(VIR_ERR_INVALID_ARG, _("unsupported flags: (0x%x)"), flags);
        return -1;
    }

2051
    vboxIIDFromUUID(&iid, dom->uuid);
2052 2053 2054 2055 2056 2057 2058 2059 2060 2061 2062
#if VBOX_API_VERSION >= 4000
    /* Get machine for the call to VBOX_SESSION_OPEN */
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
                  _("no domain with matching uuid"));
        return -1;
    }
#endif

    rc = VBOX_SESSION_OPEN(iid.value, machine);
2063 2064 2065 2066 2067 2068 2069
    if (NS_SUCCEEDED(rc)) {
        data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);
        if (machine) {
            rc = machine->vtbl->SetCPUCount(machine, CPUCount);
            if (NS_SUCCEEDED(rc)) {
                machine->vtbl->SaveSettings(machine);
                ret = 0;
2070
            } else {
2071 2072 2073
                vboxError(VIR_ERR_INTERNAL_ERROR,
                          _("could not set the number of cpus of the domain "
                            "to: %u, rc=%08x"),
2074
                          CPUCount, (unsigned)rc);
2075
            }
2076
            VBOX_RELEASE(machine);
2077
        } else {
2078
            vboxError(VIR_ERR_NO_DOMAIN,
2079
                      _("no domain with matching id %d"), dom->id);
2080
        }
2081
    } else {
2082
        vboxError(VIR_ERR_NO_DOMAIN,
2083
                  _("can't open session to the domain with id %d"), dom->id);
2084
    }
2085
    VBOX_SESSION_CLOSE();
2086

2087
    vboxIIDUnalloc(&iid);
2088 2089 2090
    return ret;
}

2091 2092 2093
static int
vboxDomainSetVcpus(virDomainPtr dom, unsigned int nvcpus)
{
2094
    return vboxDomainSetVcpusFlags(dom, nvcpus, VIR_DOMAIN_AFFECT_LIVE);
2095 2096 2097 2098 2099
}

static int
vboxDomainGetVcpusFlags(virDomainPtr dom, unsigned int flags)
{
2100 2101
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
    ISystemProperties *systemProperties = NULL;
2102 2103
    PRUint32 maxCPUCount = 0;

2104
    if (flags != (VIR_DOMAIN_AFFECT_LIVE | VIR_DOMAIN_VCPU_MAXIMUM)) {
2105 2106 2107 2108
        vboxError(VIR_ERR_INVALID_ARG, _("unsupported flags: (0x%x)"), flags);
        return -1;
    }

2109 2110 2111 2112 2113
    /* Currently every domain supports the same number of max cpus
     * as that supported by vbox and thus take it directly from
     * the systemproperties.
     */

2114 2115 2116 2117
    data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
    if (systemProperties) {
        systemProperties->vtbl->GetMaxGuestCPUCount(systemProperties, &maxCPUCount);
        VBOX_RELEASE(systemProperties);
2118 2119 2120 2121 2122 2123 2124 2125
    }

    if (maxCPUCount > 0)
        ret = maxCPUCount;

    return ret;
}

2126 2127 2128
static int
vboxDomainGetMaxVcpus(virDomainPtr dom)
{
2129
    return vboxDomainGetVcpusFlags(dom, (VIR_DOMAIN_AFFECT_LIVE |
2130 2131 2132
                                         VIR_DOMAIN_VCPU_MAXIMUM));
}

2133
static char *vboxDomainGetXMLDesc(virDomainPtr dom, unsigned int flags) {
2134
    VBOX_OBJECT_CHECK(dom->conn, char *, NULL);
2135 2136
    virDomainDefPtr def  = NULL;
    IMachine *machine    = NULL;
2137
    vboxIID iid = VBOX_IID_INITIALIZER;
2138
    int gotAllABoutDef   = -1;
2139
    nsresult rc;
2140
    char *tmp;
2141 2142

    if (VIR_ALLOC(def) < 0) {
2143
        virReportOOMError();
2144 2145 2146
        goto cleanup;
    }

2147
    vboxIIDFromUUID(&iid, dom->uuid);
2148
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
2149
    if (NS_SUCCEEDED(rc)) {
2150
        PRBool accessible = PR_FALSE;
2151

2152 2153 2154 2155 2156 2157 2158
        machine->vtbl->GetAccessible(machine, &accessible);
        if (accessible) {
            int i = 0;
            struct utsname utsname;
            PRBool PAEEnabled                   = PR_FALSE;
            PRBool ACPIEnabled                  = PR_FALSE;
            PRBool IOAPICEnabled                = PR_FALSE;
2159
            PRBool VRDxEnabled                  = PR_FALSE;
2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181
            PRUint32 CPUCount                   = 0;
            PRUint32 memorySize                 = 0;
            PRUint32 netAdpCnt                  = 0;
            PRUint32 netAdpIncCnt               = 0;
            PRUint32 maxMemorySize              = 4 * 1024;
            PRUint32 USBFilterCount             = 0;
            PRUint32 maxBootPosition            = 0;
            PRUint32 serialPortCount            = 0;
            PRUint32 serialPortIncCount         = 0;
            PRUint32 parallelPortCount          = 0;
            PRUint32 parallelPortIncCount       = 0;
            IBIOSSettings *bios                 = NULL;
#if VBOX_API_VERSION < 3001
            PRInt32       hddNum                = 0;
            IDVDDrive    *dvdDrive              = NULL;
            IHardDisk    *hardDiskPM            = NULL;
            IHardDisk    *hardDiskPS            = NULL;
            IHardDisk    *hardDiskSS            = NULL;
            const char   *hddBus                = "IDE";
            PRUnichar    *hddBusUtf16           = NULL;
            IFloppyDrive *floppyDrive           = NULL;
#else  /* VBOX_API_VERSION >= 3001 */
2182
            vboxArray mediumAttachments         = VBOX_ARRAY_INITIALIZER;
2183
#endif /* VBOX_API_VERSION >= 3001 */
2184 2185 2186 2187 2188
#if VBOX_API_VERSION < 4000
            IVRDPServer *VRDxServer             = NULL;
#else  /* VBOX_API_VERSION >= 4000 */
            IVRDEServer *VRDxServer             = NULL;
#endif /* VBOX_API_VERSION >= 4000 */
2189 2190 2191
            IAudioAdapter *audioAdapter         = NULL;
            IUSBController *USBController       = NULL;
            ISystemProperties *systemProperties = NULL;
2192 2193


2194 2195 2196 2197
            def->virtType = VIR_DOMAIN_VIRT_VBOX;
            def->id = dom->id;
            memcpy(def->uuid, dom->uuid, VIR_UUID_BUFLEN);
            def->name = strdup(dom->name);
2198

2199
            machine->vtbl->GetMemorySize(machine, &memorySize);
2200
            def->mem.cur_balloon = memorySize * 1024;
2201

2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215
            data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
            if (systemProperties) {
                systemProperties->vtbl->GetMaxGuestRAM(systemProperties, &maxMemorySize);
                systemProperties->vtbl->GetMaxBootPosition(systemProperties, &maxBootPosition);
                systemProperties->vtbl->GetNetworkAdapterCount(systemProperties, &netAdpCnt);
                systemProperties->vtbl->GetSerialPortCount(systemProperties, &serialPortCount);
                systemProperties->vtbl->GetParallelPortCount(systemProperties, &parallelPortCount);
                VBOX_RELEASE(systemProperties);
                systemProperties = NULL;
            }
            /* Currently setting memory and maxMemory as same, cause
             * the notation here seems to be inconsistent while
             * reading and while dumping xml
             */
2216 2217
            /* def->mem.max_balloon = maxMemorySize * 1024; */
            def->mem.max_balloon = memorySize * 1024;
2218 2219

            machine->vtbl->GetCPUCount(machine, &CPUCount);
E
Eric Blake 已提交
2220
            def->maxvcpus = def->vcpus = CPUCount;
2221 2222 2223 2224

            /* Skip cpumasklen, cpumask, onReboot, onPoweroff, onCrash */

            def->os.type = strdup("hvm");
2225

2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250
            uname(&utsname);
            def->os.arch = strdup(utsname.machine);

            def->os.nBootDevs = 0;
            for (i = 0; (i < VIR_DOMAIN_BOOT_LAST) && (i < maxBootPosition); i++) {
                PRUint32 device = DeviceType_Null;

                machine->vtbl->GetBootOrder(machine, i+1, &device);

                if (device == DeviceType_Floppy) {
                    def->os.bootDevs[i] = VIR_DOMAIN_BOOT_FLOPPY;
                    def->os.nBootDevs++;
                } else if (device == DeviceType_DVD) {
                    def->os.bootDevs[i] = VIR_DOMAIN_BOOT_CDROM;
                    def->os.nBootDevs++;
                } else if (device == DeviceType_HardDisk) {
                    def->os.bootDevs[i] = VIR_DOMAIN_BOOT_DISK;
                    def->os.nBootDevs++;
                } else if (device == DeviceType_Network) {
                    def->os.bootDevs[i] = VIR_DOMAIN_BOOT_NET;
                    def->os.nBootDevs++;
                } else if (device == DeviceType_USB) {
                    /* Not supported by libvirt yet */
                } else if (device == DeviceType_SharedFolder) {
                    /* Not supported by libvirt yet */
2251
                }
2252
            }
2253

2254 2255 2256
            def->features = 0;
#if VBOX_API_VERSION < 3001
            machine->vtbl->GetPAEEnabled(machine, &PAEEnabled);
2257
#elif VBOX_API_VERSION == 3001
2258
            machine->vtbl->GetCpuProperty(machine, CpuPropertyType_PAE, &PAEEnabled);
2259 2260 2261
#elif VBOX_API_VERSION >= 3002
            machine->vtbl->GetCPUProperty(machine, CPUPropertyType_PAE, &PAEEnabled);
#endif
2262 2263 2264
            if (PAEEnabled) {
                def->features = def->features | (1 << VIR_DOMAIN_FEATURE_PAE);
            }
2265

2266 2267 2268 2269 2270 2271
            machine->vtbl->GetBIOSSettings(machine, &bios);
            if (bios) {
                bios->vtbl->GetACPIEnabled(bios, &ACPIEnabled);
                if (ACPIEnabled) {
                    def->features = def->features | (1 << VIR_DOMAIN_FEATURE_ACPI);
                }
2272

2273 2274 2275
                bios->vtbl->GetIOAPICEnabled(bios, &IOAPICEnabled);
                if (IOAPICEnabled) {
                    def->features = def->features | (1 << VIR_DOMAIN_FEATURE_APIC);
2276 2277
                }

2278 2279 2280 2281 2282
                VBOX_RELEASE(bios);
            }

            /* Currently VirtualBox always uses locatime
             * so locatime is always true here */
2283
            def->clock.offset = VIR_DOMAIN_CLOCK_OFFSET_LOCALTIME;
2284 2285 2286 2287 2288 2289 2290 2291

            /* dump video options vram/2d/3d/directx/etc. */
            {
                /* Currently supports only one graphics card */
                def->nvideos = 1;
                if (VIR_ALLOC_N(def->videos, def->nvideos) >= 0) {
                    if (VIR_ALLOC(def->videos[0]) >= 0) {
                        /* the default is: vram is 8MB, One monitor, 3dAccel Off */
2292
                        PRUint32 VRAMSize          = 8;
2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304
                        PRUint32 monitorCount      = 1;
                        PRBool accelerate3DEnabled = PR_FALSE;
                        PRBool accelerate2DEnabled = PR_FALSE;

                        machine->vtbl->GetVRAMSize(machine, &VRAMSize);
                        machine->vtbl->GetMonitorCount(machine, &monitorCount);
                        machine->vtbl->GetAccelerate3DEnabled(machine, &accelerate3DEnabled);
#if VBOX_API_VERSION >= 3001
                        machine->vtbl->GetAccelerate2DVideoEnabled(machine, &accelerate2DEnabled);
#endif /* VBOX_API_VERSION >= 3001 */

                        def->videos[0]->type            = VIR_DOMAIN_VIDEO_TYPE_VBOX;
2305
                        def->videos[0]->vram            = VRAMSize * 1024;
2306 2307 2308 2309
                        def->videos[0]->heads           = monitorCount;
                        if (VIR_ALLOC(def->videos[0]->accel) >= 0) {
                            def->videos[0]->accel->support3d = accelerate3DEnabled;
                            def->videos[0]->accel->support2d = accelerate2DEnabled;
2310
                        } else
2311
                            virReportOOMError();
2312
                    } else
2313
                        virReportOOMError();
2314
                } else
2315
                    virReportOOMError();
2316
            }
2317

2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328
            /* dump display options vrdp/gui/sdl */
            {
                int vrdpPresent           = 0;
                int sdlPresent            = 0;
                int guiPresent            = 0;
                int totalPresent          = 0;
                char *guiDisplay          = NULL;
                char *sdlDisplay          = NULL;
                PRUnichar *keyTypeUtf16   = NULL;
                PRUnichar *valueTypeUtf16 = NULL;
                char      *valueTypeUtf8  = NULL;
2329

2330
                def->ngraphics = 0;
2331

2332 2333 2334
                VBOX_UTF8_TO_UTF16("FRONTEND/Type", &keyTypeUtf16);
                machine->vtbl->GetExtraData(machine, keyTypeUtf16, &valueTypeUtf16);
                VBOX_UTF16_FREE(keyTypeUtf16);
2335

2336 2337 2338
                if (valueTypeUtf16) {
                    VBOX_UTF16_TO_UTF8(valueTypeUtf16, &valueTypeUtf8);
                    VBOX_UTF16_FREE(valueTypeUtf16);
2339

2340 2341 2342 2343
                    if ( STREQ(valueTypeUtf8, "sdl") || STREQ(valueTypeUtf8, "gui") ) {
                        PRUnichar *keyDislpayUtf16   = NULL;
                        PRUnichar *valueDisplayUtf16 = NULL;
                        char      *valueDisplayUtf8  = NULL;
2344

2345 2346 2347
                        VBOX_UTF8_TO_UTF16("FRONTEND/Display", &keyDislpayUtf16);
                        machine->vtbl->GetExtraData(machine, keyDislpayUtf16, &valueDisplayUtf16);
                        VBOX_UTF16_FREE(keyDislpayUtf16);
2348

2349 2350 2351
                        if (valueDisplayUtf16) {
                            VBOX_UTF16_TO_UTF8(valueDisplayUtf16, &valueDisplayUtf8);
                            VBOX_UTF16_FREE(valueDisplayUtf16);
2352

2353 2354 2355
                            if (strlen(valueDisplayUtf8) <= 0) {
                                VBOX_UTF8_FREE(valueDisplayUtf8);
                                valueDisplayUtf8 = NULL;
2356 2357
                            }
                        }
2358

2359 2360 2361 2362 2363
                        if (STREQ(valueTypeUtf8, "sdl")) {
                            sdlPresent = 1;
                            if (valueDisplayUtf8)
                                sdlDisplay = strdup(valueDisplayUtf8);
                            if (sdlDisplay == NULL) {
2364
                                virReportOOMError();
2365 2366 2367 2368 2369 2370
                                /* just don't go to cleanup yet as it is ok to have
                                 * sdlDisplay as NULL and we check it below if it
                                 * exist and then only use it there
                                 */
                            }
                            totalPresent++;
2371
                        }
2372

2373 2374 2375 2376 2377
                        if (STREQ(valueTypeUtf8, "gui")) {
                            guiPresent = 1;
                            if (valueDisplayUtf8)
                                guiDisplay = strdup(valueDisplayUtf8);
                            if (guiDisplay == NULL) {
2378
                                virReportOOMError();
2379
                                /* just don't go to cleanup yet as it is ok to have
2380 2381
                                 * guiDisplay as NULL and we check it below if it
                                 * exist and then only use it there
2382
                                 */
2383
                            }
2384 2385
                            totalPresent++;
                        }
2386 2387
                        if (valueDisplayUtf8)
                            VBOX_UTF8_FREE(valueDisplayUtf8);
2388 2389
                    }

2390 2391
                    if (STREQ(valueTypeUtf8, "vrdp"))
                        vrdpPresent = 1;
2392

2393 2394
                    VBOX_UTF8_FREE(valueTypeUtf8);
                }
2395

2396 2397 2398 2399 2400 2401 2402
                if ((totalPresent > 0) && (VIR_ALLOC_N(def->graphics, totalPresent) >= 0)) {
                    if ((guiPresent) && (VIR_ALLOC(def->graphics[def->ngraphics]) >= 0)) {
                        def->graphics[def->ngraphics]->type = VIR_DOMAIN_GRAPHICS_TYPE_DESKTOP;
                        if (guiDisplay)
                            def->graphics[def->ngraphics]->data.desktop.display = guiDisplay;
                        def->ngraphics++;
                    }
2403

2404 2405 2406 2407 2408 2409 2410 2411 2412
                    if ((sdlPresent) && (VIR_ALLOC(def->graphics[def->ngraphics]) >= 0)) {
                        def->graphics[def->ngraphics]->type = VIR_DOMAIN_GRAPHICS_TYPE_SDL;
                        if (sdlDisplay)
                            def->graphics[def->ngraphics]->data.sdl.display = sdlDisplay;
                        def->ngraphics++;
                    }
                } else if ((vrdpPresent != 1) && (totalPresent == 0) && (VIR_ALLOC_N(def->graphics, 1) >= 0)) {
                    if (VIR_ALLOC(def->graphics[def->ngraphics]) >= 0) {
                        def->graphics[def->ngraphics]->type = VIR_DOMAIN_GRAPHICS_TYPE_DESKTOP;
2413 2414 2415 2416 2417 2418 2419 2420 2421
                        tmp = getenv("DISPLAY");
                        if (tmp != NULL) {
                            def->graphics[def->ngraphics]->data.desktop.display = strdup(tmp);
                            if (def->graphics[def->ngraphics]->data.desktop.display == NULL) {
                                virReportOOMError();
                                /* just don't go to cleanup yet as it is ok to have
                                 * display as NULL
                                 */
                            }
2422 2423 2424 2425 2426
                        }
                        totalPresent++;
                        def->ngraphics++;
                    }
                }
2427

2428 2429 2430 2431 2432 2433 2434 2435
#if VBOX_API_VERSION < 4000
                machine->vtbl->GetVRDPServer(machine, &VRDxServer);
#else  /* VBOX_API_VERSION >= 4000 */
                machine->vtbl->GetVRDEServer(machine, &VRDxServer);
#endif /* VBOX_API_VERSION >= 4000 */
                if (VRDxServer) {
                    VRDxServer->vtbl->GetEnabled(VRDxServer, &VRDxEnabled);
                    if (VRDxEnabled) {
2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446

                        totalPresent++;

                        if ((VIR_REALLOC_N(def->graphics, totalPresent) >= 0) &&
                            (VIR_ALLOC(def->graphics[def->ngraphics]) >= 0)) {
                            PRUnichar *netAddressUtf16   = NULL;
                            char      *netAddressUtf8    = NULL;
                            PRBool allowMultiConnection  = PR_FALSE;
                            PRBool reuseSingleConnection = PR_FALSE;
#if VBOX_API_VERSION < 3001
                            PRUint32 VRDPport = 0;
2447
                            VRDxServer->vtbl->GetPort(VRDxServer, &VRDPport);
2448 2449
                            if (VRDPport) {
                                def->graphics[def->ngraphics]->data.rdp.port = VRDPport;
2450
#elif VBOX_API_VERSION < 4000 /* 3001 <= VBOX_API_VERSION < 4000 */
2451
                            PRUnichar *VRDPport = NULL;
2452
                            VRDxServer->vtbl->GetPorts(VRDxServer, &VRDPport);
2453 2454 2455 2456
                            if (VRDPport) {
                                /* even if vbox supports mutilpe ports, single port for now here */
                                def->graphics[def->ngraphics]->data.rdp.port = PRUnicharToInt(VRDPport);
                                VBOX_UTF16_FREE(VRDPport);
2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467
#else /* VBOX_API_VERSION >= 4000 */
                            PRUnichar *VRDEPortsKey = NULL;
                            PRUnichar *VRDEPortsValue = NULL;
                            VBOX_UTF8_TO_UTF16("TCP/Ports", &VRDEPortsKey);
                            VRDxServer->vtbl->GetVRDEProperty(VRDxServer, VRDEPortsKey, &VRDEPortsValue);
                            VBOX_UTF16_FREE(VRDEPortsKey);
                            if (VRDEPortsValue) {
                                /* even if vbox supports mutilpe ports, single port for now here */
                                def->graphics[def->ngraphics]->data.rdp.port = PRUnicharToInt(VRDEPortsValue);
                                VBOX_UTF16_FREE(VRDEPortsValue);
#endif /* VBOX_API_VERSION >= 4000 */
2468 2469 2470
                            } else {
                                def->graphics[def->ngraphics]->data.rdp.autoport = 1;
                            }
2471

2472
                            def->graphics[def->ngraphics]->type = VIR_DOMAIN_GRAPHICS_TYPE_RDP;
2473

2474 2475 2476 2477 2478 2479 2480 2481
#if VBOX_API_VERSION >= 4000
                            PRUnichar *VRDENetAddressKey = NULL;
                            VBOX_UTF8_TO_UTF16("TCP/Address", &VRDENetAddressKey);
                            VRDxServer->vtbl->GetVRDEProperty(VRDxServer, VRDENetAddressKey, &netAddressUtf16);
                            VBOX_UTF16_FREE(VRDENetAddressKey);
#else /* VBOX_API_VERSION < 4000 */
                            VRDxServer->vtbl->GetNetAddress(VRDxServer, &netAddressUtf16);
#endif /* VBOX_API_VERSION < 4000 */
2482 2483 2484 2485 2486 2487 2488
                            if (netAddressUtf16) {
                                VBOX_UTF16_TO_UTF8(netAddressUtf16, &netAddressUtf8);
                                if (STRNEQ(netAddressUtf8, ""))
                                        def->graphics[def->ngraphics]->data.rdp.listenAddr = strdup(netAddressUtf8);
                                VBOX_UTF16_FREE(netAddressUtf16);
                                VBOX_UTF8_FREE(netAddressUtf8);
                            }
2489

2490
                            VRDxServer->vtbl->GetAllowMultiConnection(VRDxServer, &allowMultiConnection);
2491 2492 2493
                            if (allowMultiConnection) {
                                def->graphics[def->ngraphics]->data.rdp.multiUser = 1;
                            }
2494

2495
                            VRDxServer->vtbl->GetReuseSingleConnection(VRDxServer, &reuseSingleConnection);
2496 2497 2498
                            if (reuseSingleConnection) {
                                def->graphics[def->ngraphics]->data.rdp.replaceUser = 1;
                            }
2499

2500
                            def->ngraphics++;
2501
                        } else
2502
                            virReportOOMError();
2503
                    }
2504
                    VBOX_RELEASE(VRDxServer);
2505
                }
2506
            }
2507

2508 2509 2510
#if VBOX_API_VERSION < 3001
            /* dump IDE hdds if present */
            VBOX_UTF8_TO_UTF16(hddBus, &hddBusUtf16);
2511

2512 2513 2514 2515
            def->ndisks = 0;
            machine->vtbl->GetHardDisk(machine, hddBusUtf16, 0, 0,  &hardDiskPM);
            if (hardDiskPM)
                def->ndisks++;
2516

2517 2518 2519
            machine->vtbl->GetHardDisk(machine, hddBusUtf16, 0, 1,  &hardDiskPS);
            if (hardDiskPS)
                def->ndisks++;
2520

2521 2522 2523
            machine->vtbl->GetHardDisk(machine, hddBusUtf16, 1, 1,  &hardDiskSS);
            if (hardDiskSS)
                def->ndisks++;
2524

2525 2526 2527 2528 2529 2530 2531 2532 2533
            VBOX_UTF16_FREE(hddBusUtf16);

            if ((def->ndisks > 0) && (VIR_ALLOC_N(def->disks, def->ndisks) >= 0)) {
                for (i = 0; i < def->ndisks; i++) {
                    if (VIR_ALLOC(def->disks[i]) >= 0) {
                        def->disks[i]->device = VIR_DOMAIN_DISK_DEVICE_DISK;
                        def->disks[i]->bus = VIR_DOMAIN_DISK_BUS_IDE;
                        def->disks[i]->type = VIR_DOMAIN_DISK_TYPE_FILE;
                    } else
2534
                        virReportOOMError();
2535
                }
2536
            }
2537

2538 2539 2540 2541
            if (hardDiskPM) {
                PRUnichar *hddlocationUtf16 = NULL;
                char *hddlocation           = NULL;
                PRUint32 hddType            = HardDiskType_Normal;
2542

2543 2544
                hardDiskPM->vtbl->imedium.GetLocation((IMedium *)hardDiskPM, &hddlocationUtf16);
                VBOX_UTF16_TO_UTF8(hddlocationUtf16, &hddlocation);
2545

2546
                hardDiskPM->vtbl->GetType(hardDiskPM, &hddType);
2547

2548 2549 2550 2551 2552
                if (hddType == HardDiskType_Immutable)
                    def->disks[hddNum]->readonly = 1;
                def->disks[hddNum]->src = strdup(hddlocation);
                def->disks[hddNum]->dst = strdup("hda");
                hddNum++;
2553

2554 2555 2556 2557
                VBOX_UTF8_FREE(hddlocation);
                VBOX_UTF16_FREE(hddlocationUtf16);
                VBOX_MEDIUM_RELEASE(hardDiskPM);
            }
2558

2559 2560 2561 2562
            if (hardDiskPS) {
                PRUnichar *hddlocationUtf16 = NULL;
                char *hddlocation           = NULL;
                PRUint32 hddType            = HardDiskType_Normal;
2563

2564 2565
                hardDiskPS->vtbl->imedium.GetLocation((IMedium *)hardDiskPS, &hddlocationUtf16);
                VBOX_UTF16_TO_UTF8(hddlocationUtf16, &hddlocation);
2566

2567
                hardDiskPS->vtbl->GetType(hardDiskPS, &hddType);
2568

2569 2570 2571 2572 2573
                if (hddType == HardDiskType_Immutable)
                    def->disks[hddNum]->readonly = 1;
                def->disks[hddNum]->src = strdup(hddlocation);
                def->disks[hddNum]->dst = strdup("hdb");
                hddNum++;
2574

2575 2576 2577 2578
                VBOX_UTF8_FREE(hddlocation);
                VBOX_UTF16_FREE(hddlocationUtf16);
                VBOX_MEDIUM_RELEASE(hardDiskPS);
            }
2579

2580 2581 2582 2583
            if (hardDiskSS) {
                PRUnichar *hddlocationUtf16 = NULL;
                char *hddlocation           = NULL;
                PRUint32 hddType            = HardDiskType_Normal;
2584

2585 2586
                hardDiskSS->vtbl->imedium.GetLocation((IMedium *)hardDiskSS, &hddlocationUtf16);
                VBOX_UTF16_TO_UTF8(hddlocationUtf16, &hddlocation);
2587

2588
                hardDiskSS->vtbl->GetType(hardDiskSS, &hddType);
2589

2590 2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607
                if (hddType == HardDiskType_Immutable)
                    def->disks[hddNum]->readonly = 1;
                def->disks[hddNum]->src = strdup(hddlocation);
                def->disks[hddNum]->dst = strdup("hdd");
                hddNum++;

                VBOX_UTF8_FREE(hddlocation);
                VBOX_UTF16_FREE(hddlocationUtf16);
                VBOX_MEDIUM_RELEASE(hardDiskSS);
            }
#else  /* VBOX_API_VERSION >= 3001 */
            /* dump IDE hdds if present */

            bool error = false;
            int diskCount = 0;
            PRUint32   maxPortPerInst[StorageBus_Floppy + 1] = {};
            PRUint32   maxSlotPerPort[StorageBus_Floppy + 1] = {};
            def->ndisks = 0;
2608
            vboxArrayGet(&mediumAttachments, machine, machine->vtbl->GetMediumAttachments);
2609 2610

            /* get the number of attachments */
2611 2612
            for (i = 0; i < mediumAttachments.count; i++) {
                IMediumAttachment *imediumattach = mediumAttachments.items[i];
2613 2614 2615 2616 2617 2618 2619
                if (imediumattach) {
                    IMedium *medium = NULL;

                    imediumattach->vtbl->GetMedium(imediumattach, &medium);
                    if (medium) {
                        def->ndisks++;
                        VBOX_RELEASE(medium);
2620 2621
                    }
                }
2622
            }
2623

2624 2625 2626 2627
            /* Allocate mem, if fails return error */
            if (VIR_ALLOC_N(def->disks, def->ndisks) >= 0) {
                for (i = 0; i < def->ndisks; i++) {
                    if (VIR_ALLOC(def->disks[i]) < 0) {
2628
                        virReportOOMError();
2629 2630
                        error = true;
                        break;
2631 2632
                    }
                }
2633
            } else {
2634
                virReportOOMError();
2635 2636 2637 2638 2639 2640 2641
                error = true;
            }

            if (!error)
                error = !vboxGetMaxPortSlotValues(data->vboxObj, maxPortPerInst, maxSlotPerPort);

            /* get the attachment details here */
2642 2643
            for (i = 0; i < mediumAttachments.count && diskCount < def->ndisks && !error; i++) {
                IMediumAttachment *imediumattach = mediumAttachments.items[i];
2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676
                IStorageController *storageController = NULL;
                PRUnichar *storageControllerName = NULL;
                PRUint32   deviceType     = DeviceType_Null;
                PRUint32   storageBus     = StorageBus_Null;
                PRBool     readOnly       = PR_FALSE;
                IMedium   *medium         = NULL;
                PRUnichar *mediumLocUtf16 = NULL;
                char      *mediumLocUtf8  = NULL;
                PRUint32   deviceInst     = 0;
                PRInt32    devicePort     = 0;
                PRInt32    deviceSlot     = 0;

                if (!imediumattach)
                    continue;

                imediumattach->vtbl->GetMedium(imediumattach, &medium);
                if (!medium)
                    continue;

                imediumattach->vtbl->GetController(imediumattach, &storageControllerName);
                if (!storageControllerName) {
                    VBOX_RELEASE(medium);
                    continue;
                }

                machine->vtbl->GetStorageControllerByName(machine,
                                                          storageControllerName,
                                                          &storageController);
                VBOX_UTF16_FREE(storageControllerName);
                if (!storageController) {
                    VBOX_RELEASE(medium);
                    continue;
                }
2677

2678 2679 2680 2681 2682 2683 2684 2685 2686
                medium->vtbl->GetLocation(medium, &mediumLocUtf16);
                VBOX_UTF16_TO_UTF8(mediumLocUtf16, &mediumLocUtf8);
                VBOX_UTF16_FREE(mediumLocUtf16);
                def->disks[diskCount]->src = strdup(mediumLocUtf8);
                VBOX_UTF8_FREE(mediumLocUtf8);

                if (!(def->disks[diskCount]->src)) {
                    VBOX_RELEASE(medium);
                    VBOX_RELEASE(storageController);
2687
                    virReportOOMError();
2688 2689 2690
                    error = true;
                    break;
                }
2691

2692 2693 2694 2695 2696 2697 2698 2699 2700 2701
                storageController->vtbl->GetBus(storageController, &storageBus);
                if (storageBus == StorageBus_IDE) {
                    def->disks[diskCount]->bus = VIR_DOMAIN_DISK_BUS_IDE;
                } else if (storageBus == StorageBus_SATA) {
                    def->disks[diskCount]->bus = VIR_DOMAIN_DISK_BUS_SATA;
                } else if (storageBus == StorageBus_SCSI) {
                    def->disks[diskCount]->bus = VIR_DOMAIN_DISK_BUS_SCSI;
                } else if (storageBus == StorageBus_Floppy) {
                    def->disks[diskCount]->bus = VIR_DOMAIN_DISK_BUS_FDC;
                }
2702

2703 2704 2705 2706 2707 2708 2709 2710 2711 2712
                imediumattach->vtbl->GetType(imediumattach, &deviceType);
                if (deviceType == DeviceType_HardDisk)
                    def->disks[diskCount]->device = VIR_DOMAIN_DISK_DEVICE_DISK;
                else if (deviceType == DeviceType_Floppy)
                    def->disks[diskCount]->device = VIR_DOMAIN_DISK_DEVICE_FLOPPY;
                else if (deviceType == DeviceType_DVD)
                    def->disks[diskCount]->device = VIR_DOMAIN_DISK_DEVICE_CDROM;

                imediumattach->vtbl->GetPort(imediumattach, &devicePort);
                imediumattach->vtbl->GetDevice(imediumattach, &deviceSlot);
2713
                def->disks[diskCount]->dst = vboxGenerateMediumName(storageBus,
2714 2715 2716 2717 2718 2719
                                                                    deviceInst,
                                                                    devicePort,
                                                                    deviceSlot,
                                                                    maxPortPerInst,
                                                                    maxSlotPerPort);
                if (!def->disks[diskCount]->dst) {
2720
                    vboxError(VIR_ERR_INTERNAL_ERROR,
2721 2722
                              _("Could not generate medium name for the disk "
                                "at: controller instance:%u, port:%d, slot:%d"),
2723 2724 2725 2726 2727 2728
                              deviceInst, devicePort, deviceSlot);
                    VBOX_RELEASE(medium);
                    VBOX_RELEASE(storageController);
                    error = true;
                    break;
                }
2729

2730 2731 2732
                medium->vtbl->GetReadOnly(medium, &readOnly);
                if (readOnly == PR_TRUE)
                    def->disks[diskCount]->readonly = 1;
2733

2734
                def->disks[diskCount]->type = VIR_DOMAIN_DISK_TYPE_FILE;
2735

2736 2737 2738 2739
                VBOX_RELEASE(medium);
                VBOX_RELEASE(storageController);
                diskCount++;
            }
2740

2741
            vboxArrayRelease(&mediumAttachments);
2742

2743 2744 2745 2746 2747 2748 2749 2750
            /* cleanup on error */
            if (error) {
                for (i = 0; i < def->ndisks; i++) {
                    VIR_FREE(def->disks[i]);
                }
                VIR_FREE(def->disks);
                def->ndisks = 0;
            }
2751

2752
#endif /* VBOX_API_VERSION >= 3001 */
2753

2754 2755 2756 2757 2758
            /* dump network cards if present */
            def->nnets = 0;
            /* Get which network cards are enabled */
            for (i = 0; i < netAdpCnt; i++) {
                INetworkAdapter *adapter = NULL;
2759

2760 2761 2762
                machine->vtbl->GetNetworkAdapter(machine, i, &adapter);
                if (adapter) {
                    PRBool enabled = PR_FALSE;
2763

2764 2765 2766 2767
                    adapter->vtbl->GetEnabled(adapter, &enabled);
                    if (enabled) {
                        def->nnets++;
                    }
2768

2769 2770 2771
                    VBOX_RELEASE(adapter);
                }
            }
2772

2773 2774 2775 2776 2777
            /* Allocate memory for the networkcards which are enabled */
            if ((def->nnets > 0) && (VIR_ALLOC_N(def->nets, def->nnets) >= 0)) {
                for (i = 0; i < def->nnets; i++) {
                    if (VIR_ALLOC(def->nets[i]) >= 0) {
                    } else
2778
                        virReportOOMError();
2779 2780
                }
            }
2781

2782 2783 2784
            /* Now get the details about the network cards here */
            for (i = 0;(netAdpIncCnt < def->nnets) && (i < netAdpCnt); i++) {
                INetworkAdapter *adapter = NULL;
2785

2786 2787 2788
                machine->vtbl->GetNetworkAdapter(machine, i, &adapter);
                if (adapter) {
                    PRBool enabled = PR_FALSE;
2789

2790 2791 2792 2793 2794 2795 2796
                    adapter->vtbl->GetEnabled(adapter, &enabled);
                    if (enabled) {
                        PRUint32 attachmentType    = NetworkAttachmentType_Null;
                        PRUint32 adapterType       = NetworkAdapterType_Null;
                        PRUnichar *MACAddressUtf16 = NULL;
                        char *MACAddress           = NULL;
                        char macaddr[VIR_MAC_STRING_BUFLEN] = {0};
2797

2798 2799
                        adapter->vtbl->GetAttachmentType(adapter, &attachmentType);
                        if (attachmentType == NetworkAttachmentType_NAT) {
2800

2801
                            def->nets[netAdpIncCnt]->type = VIR_DOMAIN_NET_TYPE_USER;
2802

2803 2804 2805
                        } else if (attachmentType == NetworkAttachmentType_Bridged) {
                            PRUnichar *hostIntUtf16 = NULL;
                            char *hostInt           = NULL;
2806

2807
                            def->nets[netAdpIncCnt]->type = VIR_DOMAIN_NET_TYPE_BRIDGE;
2808

2809
                            adapter->vtbl->GetHostInterface(adapter, &hostIntUtf16);
2810

2811 2812
                            VBOX_UTF16_TO_UTF8(hostIntUtf16, &hostInt);
                            def->nets[netAdpIncCnt]->data.bridge.brname = strdup(hostInt);
2813

2814 2815
                            VBOX_UTF8_FREE(hostInt);
                            VBOX_UTF16_FREE(hostIntUtf16);
2816

2817 2818 2819
                        } else if (attachmentType == NetworkAttachmentType_Internal) {
                            PRUnichar *intNetUtf16 = NULL;
                            char *intNet           = NULL;
2820

2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866
                            def->nets[netAdpIncCnt]->type = VIR_DOMAIN_NET_TYPE_INTERNAL;

                            adapter->vtbl->GetInternalNetwork(adapter, &intNetUtf16);

                            VBOX_UTF16_TO_UTF8(intNetUtf16, &intNet);
                            def->nets[netAdpIncCnt]->data.internal.name = strdup(intNet);

                            VBOX_UTF8_FREE(intNet);
                            VBOX_UTF16_FREE(intNetUtf16);

                        } else if (attachmentType == NetworkAttachmentType_HostOnly) {
                            PRUnichar *hostIntUtf16 = NULL;
                            char *hostInt           = NULL;

                            def->nets[netAdpIncCnt]->type = VIR_DOMAIN_NET_TYPE_NETWORK;

                            adapter->vtbl->GetHostInterface(adapter, &hostIntUtf16);

                            VBOX_UTF16_TO_UTF8(hostIntUtf16, &hostInt);
                            def->nets[netAdpIncCnt]->data.network.name = strdup(hostInt);

                            VBOX_UTF8_FREE(hostInt);
                            VBOX_UTF16_FREE(hostIntUtf16);

                        } else {
                            /* default to user type i.e. NAT in VirtualBox if this
                             * dump is ever used to create a machine.
                             */
                            def->nets[netAdpIncCnt]->type = VIR_DOMAIN_NET_TYPE_USER;
                        }

                        adapter->vtbl->GetAdapterType(adapter, &adapterType);
                        if (adapterType == NetworkAdapterType_Am79C970A) {
                            def->nets[netAdpIncCnt]->model = strdup("Am79C970A");
                        } else if (adapterType == NetworkAdapterType_Am79C973) {
                            def->nets[netAdpIncCnt]->model = strdup("Am79C973");
                        } else if (adapterType == NetworkAdapterType_I82540EM) {
                            def->nets[netAdpIncCnt]->model = strdup("82540EM");
                        } else if (adapterType == NetworkAdapterType_I82545EM) {
                            def->nets[netAdpIncCnt]->model = strdup("82545EM");
                        } else if (adapterType == NetworkAdapterType_I82543GC) {
                            def->nets[netAdpIncCnt]->model = strdup("82543GC");
#if VBOX_API_VERSION >= 3001
                        } else if (adapterType == NetworkAdapterType_Virtio) {
                            def->nets[netAdpIncCnt]->model = strdup("virtio");
#endif /* VBOX_API_VERSION >= 3001 */
2867 2868
                        }

2869 2870 2871 2872 2873 2874 2875 2876 2877 2878 2879 2880 2881 2882 2883 2884
                        adapter->vtbl->GetMACAddress(adapter, &MACAddressUtf16);
                        VBOX_UTF16_TO_UTF8(MACAddressUtf16, &MACAddress);
                        snprintf(macaddr, VIR_MAC_STRING_BUFLEN,
                                 "%c%c:%c%c:%c%c:%c%c:%c%c:%c%c",
                                 MACAddress[0], MACAddress[1], MACAddress[2], MACAddress[3],
                                 MACAddress[4], MACAddress[5], MACAddress[6], MACAddress[7],
                                 MACAddress[8], MACAddress[9], MACAddress[10], MACAddress[11]);

                        /* XXX some real error handling here some day ... */
                        if (virParseMacAddr(macaddr, def->nets[netAdpIncCnt]->mac) < 0)
                        {}

                        netAdpIncCnt++;

                        VBOX_UTF16_FREE(MACAddressUtf16);
                        VBOX_UTF8_FREE(MACAddress);
2885
                    }
2886 2887

                    VBOX_RELEASE(adapter);
2888
                }
2889
            }
2890

2891
            /* dump sound card if active */
2892

2893 2894 2895
            /* Set def->nsounds to one as VirtualBox currently supports
             * only one sound card
             */
2896

2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908 2909 2910 2911 2912 2913 2914 2915 2916
            machine->vtbl->GetAudioAdapter(machine, &audioAdapter);
            if (audioAdapter) {
                PRBool enabled = PR_FALSE;

                audioAdapter->vtbl->GetEnabled(audioAdapter, &enabled);
                if (enabled) {
                    PRUint32 audioController = AudioControllerType_AC97;

                    def->nsounds = 1;
                    if (VIR_ALLOC_N(def->sounds, def->nsounds) >= 0) {
                        if (VIR_ALLOC(def->sounds[0]) >= 0) {
                            audioAdapter->vtbl->GetAudioController(audioAdapter, &audioController);
                            if (audioController == AudioControllerType_SB16) {
                                def->sounds[0]->model = VIR_DOMAIN_SOUND_MODEL_SB16;
                            } else if (audioController == AudioControllerType_AC97) {
                                def->sounds[0]->model = VIR_DOMAIN_SOUND_MODEL_AC97;
                            }
                        } else {
                            VIR_FREE(def->sounds);
                            def->nsounds = 0;
2917
                            virReportOOMError();
2918 2919 2920
                        }
                    } else {
                        def->nsounds = 0;
2921
                        virReportOOMError();
2922 2923 2924 2925
                    }
                }
                VBOX_RELEASE(audioAdapter);
            }
2926

2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953
#if VBOX_API_VERSION < 3001
            /* dump CDROM/DVD if the drive is attached and has DVD/CD in it */
            machine->vtbl->GetDVDDrive(machine, &dvdDrive);
            if (dvdDrive) {
                PRUint32 state = DriveState_Null;

                dvdDrive->vtbl->GetState(dvdDrive, &state);
                if (state == DriveState_ImageMounted) {
                    IDVDImage *dvdImage = NULL;

                    dvdDrive->vtbl->GetImage(dvdDrive, &dvdImage);
                    if (dvdImage) {
                        PRUnichar *locationUtf16 = NULL;
                        char *location           = NULL;

                        dvdImage->vtbl->imedium.GetLocation((IMedium *)dvdImage, &locationUtf16);
                        VBOX_UTF16_TO_UTF8(locationUtf16, &location);

                        def->ndisks++;
                        if (VIR_REALLOC_N(def->disks, def->ndisks) >= 0) {
                            if (VIR_ALLOC(def->disks[def->ndisks - 1]) >= 0) {
                                def->disks[def->ndisks - 1]->device = VIR_DOMAIN_DISK_DEVICE_CDROM;
                                def->disks[def->ndisks - 1]->bus = VIR_DOMAIN_DISK_BUS_IDE;
                                def->disks[def->ndisks - 1]->type = VIR_DOMAIN_DISK_TYPE_FILE;
                                def->disks[def->ndisks - 1]->readonly = 1;
                                def->disks[def->ndisks - 1]->src = strdup(location);
                                def->disks[def->ndisks - 1]->dst = strdup("hdc");
2954
                            } else {
2955
                                def->ndisks--;
2956
                                virReportOOMError();
2957 2958
                            }
                        } else {
2959
                            def->ndisks--;
2960
                            virReportOOMError();
2961
                        }
2962 2963 2964 2965

                        VBOX_UTF8_FREE(location);
                        VBOX_UTF16_FREE(locationUtf16);
                        VBOX_MEDIUM_RELEASE(dvdImage);
2966 2967
                    }
                }
2968 2969
                VBOX_RELEASE(dvdDrive);
            }
2970

2971 2972 2973 2974 2975 2976 2977
            /* dump Floppy if the drive is attached and has floppy in it */
            machine->vtbl->GetFloppyDrive(machine, &floppyDrive);
            if (floppyDrive) {
                PRBool enabled = PR_FALSE;

                floppyDrive->vtbl->GetEnabled(floppyDrive, &enabled);
                if (enabled) {
2978 2979
                    PRUint32 state = DriveState_Null;

2980
                    floppyDrive->vtbl->GetState(floppyDrive, &state);
2981
                    if (state == DriveState_ImageMounted) {
2982
                        IFloppyImage *floppyImage = NULL;
2983

2984 2985
                        floppyDrive->vtbl->GetImage(floppyDrive, &floppyImage);
                        if (floppyImage) {
2986 2987 2988
                            PRUnichar *locationUtf16 = NULL;
                            char *location           = NULL;

2989 2990
                            floppyImage->vtbl->imedium.GetLocation((IMedium *)floppyImage, &locationUtf16);
                            VBOX_UTF16_TO_UTF8(locationUtf16, &location);
2991 2992 2993 2994

                            def->ndisks++;
                            if (VIR_REALLOC_N(def->disks, def->ndisks) >= 0) {
                                if (VIR_ALLOC(def->disks[def->ndisks - 1]) >= 0) {
2995 2996
                                    def->disks[def->ndisks - 1]->device = VIR_DOMAIN_DISK_DEVICE_FLOPPY;
                                    def->disks[def->ndisks - 1]->bus = VIR_DOMAIN_DISK_BUS_FDC;
2997
                                    def->disks[def->ndisks - 1]->type = VIR_DOMAIN_DISK_TYPE_FILE;
2998
                                    def->disks[def->ndisks - 1]->readonly = 0;
2999
                                    def->disks[def->ndisks - 1]->src = strdup(location);
3000
                                    def->disks[def->ndisks - 1]->dst = strdup("fda");
3001 3002
                                } else {
                                    def->ndisks--;
3003
                                    virReportOOMError();
3004 3005 3006
                                }
                            } else {
                                def->ndisks--;
3007
                                virReportOOMError();
3008 3009
                            }

3010 3011 3012
                            VBOX_UTF8_FREE(location);
                            VBOX_UTF16_FREE(locationUtf16);
                            VBOX_MEDIUM_RELEASE(floppyImage);
3013 3014 3015 3016
                        }
                    }
                }

3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029
                VBOX_RELEASE(floppyDrive);
            }
#else  /* VBOX_API_VERSION >= 3001 */
#endif /* VBOX_API_VERSION >= 3001 */

            /* dump serial port if active */
            def->nserials = 0;
            /* Get which serial ports are enabled/active */
            for (i = 0; i < serialPortCount; i++) {
                ISerialPort *serialPort = NULL;

                machine->vtbl->GetSerialPort(machine, i, &serialPort);
                if (serialPort) {
3030 3031
                    PRBool enabled = PR_FALSE;

3032
                    serialPort->vtbl->GetEnabled(serialPort, &enabled);
3033
                    if (enabled) {
3034
                        def->nserials++;
3035 3036
                    }

3037
                    VBOX_RELEASE(serialPort);
3038
                }
3039
            }
3040

3041 3042 3043 3044 3045
            /* Allocate memory for the serial ports which are enabled */
            if ((def->nserials > 0) && (VIR_ALLOC_N(def->serials, def->nserials) >= 0)) {
                for (i = 0; i < def->nserials; i++) {
                    if (VIR_ALLOC(def->serials[i]) >= 0) {
                    } else
3046
                        virReportOOMError();
3047 3048
                }
            }
3049

3050 3051 3052
            /* Now get the details about the serial ports here */
            for (i = 0;(serialPortIncCount < def->nserials) && (i < serialPortCount); i++) {
                ISerialPort *serialPort = NULL;
3053

3054 3055 3056
                machine->vtbl->GetSerialPort(machine, i, &serialPort);
                if (serialPort) {
                    PRBool enabled = PR_FALSE;
3057

3058 3059 3060 3061 3062 3063 3064 3065 3066 3067
                    serialPort->vtbl->GetEnabled(serialPort, &enabled);
                    if (enabled) {
                        PRUint32 hostMode    = PortMode_Disconnected;
                        PRUint32 IOBase      = 0;
                        PRUint32 IRQ         = 0;
                        PRUnichar *pathUtf16 = NULL;
                        char *path           = NULL;

                        serialPort->vtbl->GetHostMode(serialPort, &hostMode);
                        if (hostMode == PortMode_HostPipe) {
3068
                            def->serials[serialPortIncCount]->source.type = VIR_DOMAIN_CHR_TYPE_PIPE;
3069
                        } else if (hostMode == PortMode_HostDevice) {
3070
                            def->serials[serialPortIncCount]->source.type = VIR_DOMAIN_CHR_TYPE_DEV;
3071 3072
#if VBOX_API_VERSION >= 3000
                        } else if (hostMode == PortMode_RawFile) {
3073
                            def->serials[serialPortIncCount]->source.type = VIR_DOMAIN_CHR_TYPE_FILE;
3074 3075
#endif /* VBOX_API_VERSION >= 3000 */
                        } else {
3076
                            def->serials[serialPortIncCount]->source.type = VIR_DOMAIN_CHR_TYPE_NULL;
3077
                        }
3078

3079
                        def->serials[serialPortIncCount]->deviceType = VIR_DOMAIN_CHR_DEVICE_TYPE_SERIAL;
3080

3081 3082 3083 3084 3085 3086 3087
                        serialPort->vtbl->GetIRQ(serialPort, &IRQ);
                        serialPort->vtbl->GetIOBase(serialPort, &IOBase);
                        if ((IRQ == 4) && (IOBase == 1016)) {
                            def->serials[serialPortIncCount]->target.port = 0;
                        } else if ((IRQ == 3) && (IOBase == 760)) {
                            def->serials[serialPortIncCount]->target.port = 1;
                        }
3088

3089
                        serialPort->vtbl->GetPath(serialPort, &pathUtf16);
3090

3091 3092
                        if (pathUtf16) {
                            VBOX_UTF16_TO_UTF8(pathUtf16, &path);
3093
                            def->serials[serialPortIncCount]->source.data.file.path = strdup(path);
3094 3095
                        }

3096 3097 3098 3099
                        serialPortIncCount++;

                        VBOX_UTF16_FREE(pathUtf16);
                        VBOX_UTF8_FREE(path);
3100 3101
                    }

3102 3103 3104
                    VBOX_RELEASE(serialPort);
                }
            }
3105

3106 3107 3108 3109 3110
            /* dump parallel ports if active */
            def->nparallels = 0;
            /* Get which parallel ports are enabled/active */
            for (i = 0; i < parallelPortCount; i++) {
                IParallelPort *parallelPort = NULL;
3111

3112 3113 3114
                machine->vtbl->GetParallelPort(machine, i, &parallelPort);
                if (parallelPort) {
                    PRBool enabled = PR_FALSE;
3115

3116 3117 3118
                    parallelPort->vtbl->GetEnabled(parallelPort, &enabled);
                    if (enabled) {
                        def->nparallels++;
3119
                    }
3120 3121

                    VBOX_RELEASE(parallelPort);
3122
                }
3123
            }
3124

3125 3126 3127 3128 3129
            /* Allocate memory for the parallel ports which are enabled */
            if ((def->nparallels > 0) && (VIR_ALLOC_N(def->parallels, def->nparallels) >= 0)) {
                for (i = 0; i < def->nparallels; i++) {
                    if (VIR_ALLOC(def->parallels[i]) >= 0) {
                    } else
3130
                        virReportOOMError();
3131
                }
3132
            }
3133

3134 3135 3136
            /* Now get the details about the parallel ports here */
            for (i = 0;(parallelPortIncCount < def->nparallels) && (i < parallelPortCount); i++) {
                IParallelPort *parallelPort = NULL;
3137

3138 3139 3140
                machine->vtbl->GetParallelPort(machine, i, &parallelPort);
                if (parallelPort) {
                    PRBool enabled = PR_FALSE;
3141

3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155
                    parallelPort->vtbl->GetEnabled(parallelPort, &enabled);
                    if (enabled) {
                        PRUint32 IOBase      = 0;
                        PRUint32 IRQ         = 0;
                        PRUnichar *pathUtf16 = NULL;
                        char *path           = NULL;

                        parallelPort->vtbl->GetIRQ(parallelPort, &IRQ);
                        parallelPort->vtbl->GetIOBase(parallelPort, &IOBase);
                        if ((IRQ == 7) && (IOBase == 888)) {
                            def->parallels[parallelPortIncCount]->target.port = 0;
                        } else if ((IRQ == 5) && (IOBase == 632)) {
                            def->parallels[parallelPortIncCount]->target.port = 1;
                        }
3156

3157
                        def->parallels[parallelPortIncCount]->source.type = VIR_DOMAIN_CHR_TYPE_FILE;
3158
                        def->parallels[parallelPortIncCount]->deviceType = VIR_DOMAIN_CHR_DEVICE_TYPE_PARALLEL;
3159

3160
                        parallelPort->vtbl->GetPath(parallelPort, &pathUtf16);
3161

3162
                        VBOX_UTF16_TO_UTF8(pathUtf16, &path);
3163
                        def->parallels[parallelPortIncCount]->source.data.file.path = strdup(path);
3164

3165 3166 3167 3168
                        parallelPortIncCount++;

                        VBOX_UTF16_FREE(pathUtf16);
                        VBOX_UTF8_FREE(path);
3169
                    }
3170 3171

                    VBOX_RELEASE(parallelPort);
3172
                }
3173
            }
3174

3175 3176 3177 3178 3179
            /* dump USB devices/filters if active */
            def->nhostdevs = 0;
            machine->vtbl->GetUSBController(machine, &USBController);
            if (USBController) {
                PRBool enabled = PR_FALSE;
3180

3181 3182
                USBController->vtbl->GetEnabled(USBController, &enabled);
                if (enabled) {
3183
                    vboxArray deviceFilters = VBOX_ARRAY_INITIALIZER;
3184

3185 3186
                    vboxArrayGet(&deviceFilters, USBController,
                                 USBController->vtbl->GetDeviceFilters);
3187

3188
                    if (deviceFilters.count > 0) {
3189

3190 3191 3192
                        /* check if the filters are active and then only
                         * alloc mem and set def->nhostdevs
                         */
3193

3194
                        for(i = 0; i < deviceFilters.count; i++) {
3195
                            PRBool active = PR_FALSE;
3196
                            IUSBDeviceFilter *deviceFilter = deviceFilters.items[i];
3197

3198
                            deviceFilter->vtbl->GetActive(deviceFilter, &active);
3199 3200
                            if (active) {
                                def->nhostdevs++;
3201
                            }
3202
                        }
3203

3204 3205 3206
                        if (def->nhostdevs > 0) {
                            /* Alloc mem needed for the filters now */
                            if (VIR_ALLOC_N(def->hostdevs, def->nhostdevs) >= 0) {
3207

3208
                                for(i = 0; (USBFilterCount < def->nhostdevs) || (i < deviceFilters.count); i++) {
3209
                                    PRBool active = PR_FALSE;
3210
                                    IUSBDeviceFilter *deviceFilter = deviceFilters.items[i];
3211

3212
                                    deviceFilter->vtbl->GetActive(deviceFilter, &active);
3213 3214 3215 3216 3217 3218 3219 3220 3221
                                    if (active) {
                                        if (VIR_ALLOC(def->hostdevs[USBFilterCount]) >= 0) {
                                            PRUnichar *vendorIdUtf16  = NULL;
                                            char *vendorIdUtf8        = NULL;
                                            unsigned vendorId         = 0;
                                            PRUnichar *productIdUtf16 = NULL;
                                            char *productIdUtf8       = NULL;
                                            unsigned productId        = 0;
                                            char *endptr              = NULL;
3222

3223 3224 3225 3226
                                            def->hostdevs[USBFilterCount]->mode =
                                                VIR_DOMAIN_HOSTDEV_MODE_SUBSYS;
                                            def->hostdevs[USBFilterCount]->source.subsys.type =
                                                VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB;
3227

3228 3229
                                            deviceFilter->vtbl->GetVendorId(deviceFilter, &vendorIdUtf16);
                                            deviceFilter->vtbl->GetProductId(deviceFilter, &productIdUtf16);
3230

3231 3232
                                            VBOX_UTF16_TO_UTF8(vendorIdUtf16, &vendorIdUtf8);
                                            VBOX_UTF16_TO_UTF8(productIdUtf16, &productIdUtf8);
3233

3234 3235
                                            vendorId  = strtol(vendorIdUtf8, &endptr, 16);
                                            productId = strtol(productIdUtf8, &endptr, 16);
3236

3237 3238
                                            def->hostdevs[USBFilterCount]->source.subsys.u.usb.vendor  = vendorId;
                                            def->hostdevs[USBFilterCount]->source.subsys.u.usb.product = productId;
3239

3240 3241
                                            VBOX_UTF16_FREE(vendorIdUtf16);
                                            VBOX_UTF8_FREE(vendorIdUtf8);
3242

3243 3244
                                            VBOX_UTF16_FREE(productIdUtf16);
                                            VBOX_UTF8_FREE(productIdUtf8);
3245

3246 3247
                                            USBFilterCount++;
                                        } else
3248
                                            virReportOOMError();
3249
                                    }
3250 3251
                                }
                            } else
3252
                                virReportOOMError();
3253 3254 3255
                        }
                    }

3256
                    /* Cleanup */
3257
                    vboxArrayRelease(&deviceFilters);
3258 3259
                }
                VBOX_RELEASE(USBController);
3260
            }
3261 3262 3263 3264 3265

            /* all done so set gotAllABoutDef and pass def to virDomainDefFormat
             * to generate XML for it
             */
            gotAllABoutDef = 0;
3266
        }
3267 3268
        VBOX_RELEASE(machine);
        machine = NULL;
3269 3270 3271
    }

    if (gotAllABoutDef == 0)
3272
        ret = virDomainDefFormat(def, flags);
3273 3274

cleanup:
3275
    vboxIIDUnalloc(&iid);
3276 3277 3278 3279 3280
    virDomainDefFree(def);
    return ret;
}

static int vboxListDefinedDomains(virConnectPtr conn, char ** const names, int maxnames) {
3281
    VBOX_OBJECT_CHECK(conn, int, -1);
3282
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
3283 3284 3285
    char *machineName    = NULL;
    PRUnichar *machineNameUtf16 = NULL;
    PRUint32 state;
3286
    nsresult rc;
3287 3288
    int i, j;

3289
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
3290
    if (NS_FAILED(rc)) {
3291 3292 3293
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of Defined Domains, rc=%08x"),
                  (unsigned)rc);
3294 3295
        goto cleanup;
    }
3296

3297 3298 3299
    ret = 0;
    for (i = 0, j = 0; (i < machines.count) && (j < maxnames); i++) {
        IMachine *machine = machines.items[i];
3300 3301 3302 3303 3304 3305 3306 3307 3308 3309

        if (machine) {
            PRBool isAccessible = PR_FALSE;
            machine->vtbl->GetAccessible(machine, &isAccessible);
            if (isAccessible) {
                machine->vtbl->GetState(machine, &state);
                if (   (state < MachineState_FirstOnline)
                    || (state > MachineState_LastOnline) ) {
                    machine->vtbl->GetName(machine, &machineNameUtf16);
                    VBOX_UTF16_TO_UTF8(machineNameUtf16, &machineName);
3310 3311 3312 3313
                    names[j] = strdup(machineName);
                    VBOX_UTF16_FREE(machineNameUtf16);
                    VBOX_UTF8_FREE(machineName);
                    if (!names[j]) {
3314
                        virReportOOMError();
3315 3316 3317 3318
                        for ( ; j >= 0 ; j--)
                            VIR_FREE(names[j]);
                        ret = -1;
                        goto cleanup;
3319
                    }
3320
                    j++;
3321
                    ret++;
3322 3323 3324 3325 3326 3327
                }
            }
        }
    }

cleanup:
3328
    vboxArrayRelease(&machines);
3329 3330 3331 3332
    return ret;
}

static int vboxNumOfDefinedDomains(virConnectPtr conn) {
3333
    VBOX_OBJECT_CHECK(conn, int, -1);
3334
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
3335
    PRUint32 state       = MachineState_Null;
3336
    nsresult rc;
3337 3338
    int i;

3339
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
3340
    if (NS_FAILED(rc)) {
3341 3342 3343
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get number of Defined Domains, rc=%08x"),
                  (unsigned)rc);
3344 3345
        goto cleanup;
    }
3346

3347 3348 3349
    ret = 0;
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
3350 3351 3352 3353 3354 3355 3356 3357 3358

        if (machine) {
            PRBool isAccessible = PR_FALSE;
            machine->vtbl->GetAccessible(machine, &isAccessible);
            if (isAccessible) {
                machine->vtbl->GetState(machine, &state);
                if (   (state < MachineState_FirstOnline)
                    || (state > MachineState_LastOnline) ) {
                    ret++;
3359 3360 3361 3362 3363 3364
                }
            }
        }
    }

cleanup:
3365
    vboxArrayRelease(&machines);
3366 3367 3368
    return ret;
}

E
Eric Blake 已提交
3369 3370

static int
3371 3372
vboxStartMachine(virDomainPtr dom, int i, IMachine *machine,
                 vboxIID *iid ATTRIBUTE_UNUSED /* >= 4.0 */)
E
Eric Blake 已提交
3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386 3387 3388 3389 3390 3391 3392 3393 3394 3395 3396 3397 3398 3399 3400 3401 3402 3403 3404 3405 3406 3407 3408 3409 3410 3411 3412 3413 3414 3415 3416 3417 3418 3419 3420 3421 3422 3423 3424 3425 3426 3427 3428 3429 3430 3431 3432 3433 3434 3435 3436 3437 3438 3439 3440 3441 3442 3443 3444 3445 3446 3447 3448 3449 3450 3451 3452 3453 3454 3455 3456 3457 3458 3459 3460 3461 3462 3463 3464 3465
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
    int vrdpPresent              = 0;
    int sdlPresent               = 0;
    int guiPresent               = 0;
    char *guiDisplay             = NULL;
    char *sdlDisplay             = NULL;
    PRUnichar *keyTypeUtf16      = NULL;
    PRUnichar *valueTypeUtf16    = NULL;
    char      *valueTypeUtf8     = NULL;
    PRUnichar *keyDislpayUtf16   = NULL;
    PRUnichar *valueDisplayUtf16 = NULL;
    char      *valueDisplayUtf8  = NULL;
    IProgress *progress          = NULL;
    PRUnichar *env               = NULL;
    PRUnichar *sessionType       = NULL;
    nsresult rc;

    VBOX_UTF8_TO_UTF16("FRONTEND/Type", &keyTypeUtf16);
    machine->vtbl->GetExtraData(machine, keyTypeUtf16, &valueTypeUtf16);
    VBOX_UTF16_FREE(keyTypeUtf16);

    if (valueTypeUtf16) {
        VBOX_UTF16_TO_UTF8(valueTypeUtf16, &valueTypeUtf8);
        VBOX_UTF16_FREE(valueTypeUtf16);

        if ( STREQ(valueTypeUtf8, "sdl") || STREQ(valueTypeUtf8, "gui") ) {

            VBOX_UTF8_TO_UTF16("FRONTEND/Display", &keyDislpayUtf16);
            machine->vtbl->GetExtraData(machine, keyDislpayUtf16,
                                        &valueDisplayUtf16);
            VBOX_UTF16_FREE(keyDislpayUtf16);

            if (valueDisplayUtf16) {
                VBOX_UTF16_TO_UTF8(valueDisplayUtf16, &valueDisplayUtf8);
                VBOX_UTF16_FREE(valueDisplayUtf16);

                if (strlen(valueDisplayUtf8) <= 0) {
                    VBOX_UTF8_FREE(valueDisplayUtf8);
                    valueDisplayUtf8 = NULL;
                }
            }

            if (STREQ(valueTypeUtf8, "sdl")) {
                sdlPresent = 1;
                if (valueDisplayUtf8) {
                    sdlDisplay = strdup(valueDisplayUtf8);
                    if (sdlDisplay == NULL) {
                        virReportOOMError();
                        /* just don't go to cleanup yet as it is ok to have
                         * sdlDisplay as NULL and we check it below if it
                         * exist and then only use it there
                         */
                    }
                }
            }

            if (STREQ(valueTypeUtf8, "gui")) {
                guiPresent = 1;
                if (valueDisplayUtf8) {
                    guiDisplay = strdup(valueDisplayUtf8);
                    if (guiDisplay == NULL) {
                        virReportOOMError();
                        /* just don't go to cleanup yet as it is ok to have
                         * guiDisplay as NULL and we check it below if it
                         * exist and then only use it there
                         */
                    }
                }
            }
        }

        if (STREQ(valueTypeUtf8, "vrdp")) {
            vrdpPresent = 1;
        }

        if (!vrdpPresent && !sdlPresent && !guiPresent) {
            /* if nothing is selected it means either the machine xml
             * file is really old or some values are missing so fallback
             */
            guiPresent = 1;
        }

        VBOX_UTF8_FREE(valueTypeUtf8);

    } else {
        guiPresent = 1;
    }
    if (valueDisplayUtf8)
        VBOX_UTF8_FREE(valueDisplayUtf8);

    if (guiPresent) {
        if (guiDisplay) {
E
Eric Blake 已提交
3466 3467 3468 3469 3470 3471 3472
            char *displayutf8;
            if (virAsprintf(&displayutf8, "DISPLAY=%s", guiDisplay) < 0)
                virReportOOMError();
            else {
                VBOX_UTF8_TO_UTF16(displayutf8, &env);
                VIR_FREE(displayutf8);
            }
E
Eric Blake 已提交
3473 3474 3475 3476 3477 3478 3479 3480
            VIR_FREE(guiDisplay);
        }

        VBOX_UTF8_TO_UTF16("gui", &sessionType);
    }

    if (sdlPresent) {
        if (sdlDisplay) {
E
Eric Blake 已提交
3481 3482 3483 3484 3485 3486 3487
            char *displayutf8;
            if (virAsprintf(&displayutf8, "DISPLAY=%s", sdlDisplay) < 0)
                virReportOOMError();
            else {
                VBOX_UTF8_TO_UTF16(displayutf8, &env);
                VIR_FREE(displayutf8);
            }
E
Eric Blake 已提交
3488 3489 3490 3491 3492 3493 3494 3495 3496 3497
            VIR_FREE(sdlDisplay);
        }

        VBOX_UTF8_TO_UTF16("sdl", &sessionType);
    }

    if (vrdpPresent) {
        VBOX_UTF8_TO_UTF16("vrdp", &sessionType);
    }

3498
#if VBOX_API_VERSION < 4000
E
Eric Blake 已提交
3499 3500
    rc = data->vboxObj->vtbl->OpenRemoteSession(data->vboxObj,
                                                data->vboxSession,
3501
                                                iid->value,
E
Eric Blake 已提交
3502 3503 3504
                                                sessionType,
                                                env,
                                                &progress );
3505 3506 3507 3508 3509
#else /* VBOX_API_VERSION >= 4000 */
    rc = machine->vtbl->LaunchVMProcess(machine, data->vboxSession,
                                        sessionType, env, &progress);
#endif /* VBOX_API_VERSION >= 4000 */

E
Eric Blake 已提交
3510 3511
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_OPERATION_FAILED, "%s",
3512
                  _("OpenRemoteSession/LaunchVMProcess failed, domain can't be started"));
E
Eric Blake 已提交
3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532 3533 3534 3535 3536 3537 3538 3539
        ret = -1;
    } else {
        PRBool completed = 0;
#if VBOX_API_VERSION == 2002
        nsresult resultCode;
#else
        PRInt32  resultCode;
#endif
        progress->vtbl->WaitForCompletion(progress, -1);
        rc = progress->vtbl->GetCompleted(progress, &completed);
        if (NS_FAILED(rc)) {
            /* error */
            ret = -1;
        }
        progress->vtbl->GetResultCode(progress, &resultCode);
        if (NS_FAILED(resultCode)) {
            /* error */
            ret = -1;
        } else {
            /* all ok set the domid */
            dom->id = i + 1;
            ret = 0;
        }
    }

    VBOX_RELEASE(progress);

3540
    VBOX_SESSION_CLOSE();
E
Eric Blake 已提交
3541 3542 3543 3544 3545 3546 3547

    VBOX_UTF16_FREE(env);
    VBOX_UTF16_FREE(sessionType);

    return ret;
}

3548
static int vboxDomainCreateWithFlags(virDomainPtr dom, unsigned int flags) {
3549
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
3550
    vboxArray machines = VBOX_ARRAY_INITIALIZER;
3551
    unsigned char uuid[VIR_UUID_BUFLEN] = {0};
3552 3553
    nsresult rc;
    int i = 0;
3554

3555 3556
    virCheckFlags(0, -1);

3557
    if (!dom->name) {
3558
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
3559
                  _("Error while reading the domain name"));
3560 3561 3562
        goto cleanup;
    }

3563
    rc = vboxArrayGet(&machines, data->vboxObj, data->vboxObj->vtbl->GetMachines);
3564
    if (NS_FAILED(rc)) {
3565 3566
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("Could not get list of machines, rc=%08x"), (unsigned)rc);
3567 3568
        goto cleanup;
    }
3569

3570 3571
    for (i = 0; i < machines.count; ++i) {
        IMachine *machine = machines.items[i];
3572
        PRBool isAccessible = PR_FALSE;
3573

3574 3575
        if (!machine)
            continue;
3576

3577 3578
        machine->vtbl->GetAccessible(machine, &isAccessible);
        if (isAccessible) {
3579
            vboxIID iid = VBOX_IID_INITIALIZER;
3580

3581 3582
            rc = machine->vtbl->GetId(machine, &iid.value);
            if (NS_FAILED(rc))
3583
                continue;
3584
            vboxIIDToUUID(&iid, uuid);
3585

3586
            if (memcmp(dom->uuid, uuid, VIR_UUID_BUFLEN) == 0) {
3587 3588 3589 3590 3591 3592
                PRUint32 state = MachineState_Null;
                machine->vtbl->GetState(machine, &state);

                if ( (state == MachineState_PoweredOff) ||
                     (state == MachineState_Saved) ||
                     (state == MachineState_Aborted) ) {
3593
                    ret = vboxStartMachine(dom, i, machine, &iid);
3594
                } else {
3595 3596 3597
                    vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                              _("machine is not in poweroff|saved|"
                                "aborted state, so couldn't start it"));
3598
                    ret = -1;
3599 3600
                }
            }
3601
            vboxIIDUnalloc(&iid);
3602 3603
            if (ret != -1)
                break;
3604 3605 3606
        }
    }

3607
    /* Do the cleanup and take care you dont leak any memory */
3608
    vboxArrayRelease(&machines);
3609

3610 3611 3612 3613
cleanup:
    return ret;
}

3614 3615 3616 3617
static int vboxDomainCreate(virDomainPtr dom) {
    return vboxDomainCreateWithFlags(dom, 0);
}

E
Eric Blake 已提交
3618 3619 3620 3621 3622 3623 3624
static void
vboxSetBootDeviceOrder(virDomainDefPtr def, vboxGlobalData *data,
                       IMachine *machine)
{
    ISystemProperties *systemProperties = NULL;
    PRUint32 maxBootPosition            = 0;
    int i = 0;
3625

3626 3627 3628 3629 3630 3631 3632 3633 3634 3635 3636 3637 3638 3639 3640 3641
    VIR_DEBUG("def->os.type             %s", def->os.type);
    VIR_DEBUG("def->os.arch             %s", def->os.arch);
    VIR_DEBUG("def->os.machine          %s", def->os.machine);
    VIR_DEBUG("def->os.nBootDevs        %d", def->os.nBootDevs);
    VIR_DEBUG("def->os.bootDevs[0]      %d", def->os.bootDevs[0]);
    VIR_DEBUG("def->os.bootDevs[1]      %d", def->os.bootDevs[1]);
    VIR_DEBUG("def->os.bootDevs[2]      %d", def->os.bootDevs[2]);
    VIR_DEBUG("def->os.bootDevs[3]      %d", def->os.bootDevs[3]);
    VIR_DEBUG("def->os.init             %s", def->os.init);
    VIR_DEBUG("def->os.kernel           %s", def->os.kernel);
    VIR_DEBUG("def->os.initrd           %s", def->os.initrd);
    VIR_DEBUG("def->os.cmdline          %s", def->os.cmdline);
    VIR_DEBUG("def->os.root             %s", def->os.root);
    VIR_DEBUG("def->os.loader           %s", def->os.loader);
    VIR_DEBUG("def->os.bootloader       %s", def->os.bootloader);
    VIR_DEBUG("def->os.bootloaderArgs   %s", def->os.bootloaderArgs);
3642

E
Eric Blake 已提交
3643 3644 3645 3646 3647 3648
    data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
    if (systemProperties) {
        systemProperties->vtbl->GetMaxBootPosition(systemProperties,
                                                   &maxBootPosition);
        VBOX_RELEASE(systemProperties);
        systemProperties = NULL;
3649
    }
3650

E
Eric Blake 已提交
3651 3652 3653
    /* Clear the defaults first */
    for (i = 0; i < maxBootPosition; i++) {
        machine->vtbl->SetBootOrder(machine, i+1, DeviceType_Null);
3654
    }
3655

E
Eric Blake 已提交
3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666
    for (i = 0; (i < def->os.nBootDevs) && (i < maxBootPosition); i++) {
        PRUint32 device = DeviceType_Null;

        if (def->os.bootDevs[i] == VIR_DOMAIN_BOOT_FLOPPY) {
            device = DeviceType_Floppy;
        } else if (def->os.bootDevs[i] == VIR_DOMAIN_BOOT_CDROM) {
            device = DeviceType_DVD;
        } else if (def->os.bootDevs[i] == VIR_DOMAIN_BOOT_DISK) {
            device = DeviceType_HardDisk;
        } else if (def->os.bootDevs[i] == VIR_DOMAIN_BOOT_NET) {
            device = DeviceType_Network;
3667
        }
E
Eric Blake 已提交
3668
        machine->vtbl->SetBootOrder(machine, i+1, device);
3669
    }
E
Eric Blake 已提交
3670
}
3671

E
Eric Blake 已提交
3672 3673 3674 3675 3676
static void
vboxAttachDrives(virDomainDefPtr def, vboxGlobalData *data, IMachine *machine)
{
    int i;
    nsresult rc;
3677 3678

#if VBOX_API_VERSION < 3001
E
Eric Blake 已提交
3679 3680 3681 3682
    if (def->ndisks == 0)
        return;

    for (i = 0; i < def->ndisks; i++) {
3683 3684 3685 3686 3687 3688 3689 3690 3691
        VIR_DEBUG("disk(%d) type:       %d", i, def->disks[i]->type);
        VIR_DEBUG("disk(%d) device:     %d", i, def->disks[i]->device);
        VIR_DEBUG("disk(%d) bus:        %d", i, def->disks[i]->bus);
        VIR_DEBUG("disk(%d) src:        %s", i, def->disks[i]->src);
        VIR_DEBUG("disk(%d) dst:        %s", i, def->disks[i]->dst);
        VIR_DEBUG("disk(%d) driverName: %s", i, def->disks[i]->driverName);
        VIR_DEBUG("disk(%d) driverType: %s", i, def->disks[i]->driverType);
        VIR_DEBUG("disk(%d) cachemode:  %d", i, def->disks[i]->cachemode);
        VIR_DEBUG("disk(%d) readonly:   %s", i, (def->disks[i]->readonly
E
Eric Blake 已提交
3692
                                             ? "True" : "False"));
3693
        VIR_DEBUG("disk(%d) shared:     %s", i, (def->disks[i]->shared
E
Eric Blake 已提交
3694 3695 3696 3697 3698 3699 3700 3701 3702 3703 3704 3705
                                             ? "True" : "False"));

        if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_CDROM) {
            if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_FILE &&
                def->disks[i]->src != NULL) {
                IDVDDrive *dvdDrive = NULL;
                /* Currently CDROM/DVD Drive is always IDE
                 * Secondary Master so neglecting the following
                 * parameters:
                 *      def->disks[i]->bus
                 *      def->disks[i]->dst
                 */
3706

E
Eric Blake 已提交
3707 3708 3709 3710
                machine->vtbl->GetDVDDrive(machine, &dvdDrive);
                if (dvdDrive) {
                    IDVDImage *dvdImage          = NULL;
                    PRUnichar *dvdfileUtf16      = NULL;
3711 3712
                    vboxIID dvduuid = VBOX_IID_INITIALIZER;
                    vboxIID dvdemptyuuid = VBOX_IID_INITIALIZER;
3713

E
Eric Blake 已提交
3714
                    VBOX_UTF8_TO_UTF16(def->disks[i]->src, &dvdfileUtf16);
3715

E
Eric Blake 已提交
3716 3717 3718 3719 3720
                    data->vboxObj->vtbl->FindDVDImage(data->vboxObj,
                                                      dvdfileUtf16, &dvdImage);
                    if (!dvdImage) {
                        data->vboxObj->vtbl->OpenDVDImage(data->vboxObj,
                                                          dvdfileUtf16,
3721
                                                          dvdemptyuuid.value,
E
Eric Blake 已提交
3722 3723 3724 3725
                                                          &dvdImage);
                    }
                    if (dvdImage) {
                        rc = dvdImage->vtbl->imedium.GetId((IMedium *)dvdImage,
3726
                                                           &dvduuid.value);
E
Eric Blake 已提交
3727 3728 3729 3730 3731 3732
                        if (NS_FAILED(rc)) {
                            vboxError(VIR_ERR_INTERNAL_ERROR,
                                      _("can't get the uuid of the file to "
                                        "be attached to cdrom: %s, rc=%08x"),
                                      def->disks[i]->src, (unsigned)rc);
                        } else {
3733
                            rc = dvdDrive->vtbl->MountImage(dvdDrive, dvduuid.value);
E
Eric Blake 已提交
3734 3735 3736 3737 3738
                            if (NS_FAILED(rc)) {
                                vboxError(VIR_ERR_INTERNAL_ERROR,
                                          _("could not attach the file to cdrom: %s, rc=%08x"),
                                          def->disks[i]->src, (unsigned)rc);
                            } else {
3739
                                DEBUGIID("CD/DVDImage UUID:", dvduuid.value);
3740
                            }
3741
                        }
E
Eric Blake 已提交
3742 3743

                        VBOX_MEDIUM_RELEASE(dvdImage);
3744
                    }
3745
                    vboxIIDUnalloc(&dvduuid);
E
Eric Blake 已提交
3746 3747 3748 3749 3750 3751 3752 3753 3754 3755
                    VBOX_UTF16_FREE(dvdfileUtf16);
                    VBOX_RELEASE(dvdDrive);
                }
            } else if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
            }
        } else if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
            if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_FILE &&
                def->disks[i]->src != NULL) {
                IHardDisk *hardDisk     = NULL;
                PRUnichar *hddfileUtf16 = NULL;
3756
                vboxIID hdduuid = VBOX_IID_INITIALIZER;
E
Eric Blake 已提交
3757 3758 3759 3760 3761 3762
                PRUnichar *hddEmpty     = NULL;
                /* Current Limitation: Harddisk can't be connected to
                 * Secondary Master as Secondary Master is always used
                 * for CD/DVD Drive, so don't connect the harddisk if it
                 * is requested to be connected to Secondary master
                 */
3763

E
Eric Blake 已提交
3764 3765
                VBOX_UTF8_TO_UTF16(def->disks[i]->src, &hddfileUtf16);
                VBOX_UTF8_TO_UTF16("", &hddEmpty);
3766

E
Eric Blake 已提交
3767 3768
                data->vboxObj->vtbl->FindHardDisk(data->vboxObj, hddfileUtf16,
                                                  &hardDisk);
3769

E
Eric Blake 已提交
3770
                if (!hardDisk) {
3771
# if VBOX_API_VERSION == 2002
E
Eric Blake 已提交
3772 3773 3774 3775
                    data->vboxObj->vtbl->OpenHardDisk(data->vboxObj,
                                                      hddfileUtf16,
                                                      AccessMode_ReadWrite,
                                                      &hardDisk);
3776
# else
E
Eric Blake 已提交
3777 3778 3779 3780 3781 3782 3783 3784
                    data->vboxObj->vtbl->OpenHardDisk(data->vboxObj,
                                                      hddfileUtf16,
                                                      AccessMode_ReadWrite,
                                                      0,
                                                      hddEmpty,
                                                      0,
                                                      hddEmpty,
                                                      &hardDisk);
3785
# endif
E
Eric Blake 已提交
3786
                }
3787

E
Eric Blake 已提交
3788 3789
                if (hardDisk) {
                    rc = hardDisk->vtbl->imedium.GetId((IMedium *)hardDisk,
3790
                                                       &hdduuid.value);
E
Eric Blake 已提交
3791 3792 3793 3794 3795 3796 3797 3798 3799
                    if (NS_FAILED(rc)) {
                        vboxError(VIR_ERR_INTERNAL_ERROR,
                                  _("can't get the uuid of the file to be "
                                    "attached as harddisk: %s, rc=%08x"),
                                  def->disks[i]->src, (unsigned)rc);
                    } else {
                        if (def->disks[i]->readonly) {
                            hardDisk->vtbl->SetType(hardDisk,
                                                    HardDiskType_Immutable);
3800
                            VIR_DEBUG("setting harddisk to readonly");
E
Eric Blake 已提交
3801 3802 3803
                        } else if (!def->disks[i]->readonly) {
                            hardDisk->vtbl->SetType(hardDisk,
                                                    HardDiskType_Normal);
3804
                            VIR_DEBUG("setting harddisk type to normal");
E
Eric Blake 已提交
3805 3806 3807
                        }
                        if (def->disks[i]->bus == VIR_DOMAIN_DISK_BUS_IDE) {
                            if (STREQ(def->disks[i]->dst, "hdc")) {
3808
                                VIR_DEBUG("Not connecting harddisk to hdc as hdc"
E
Eric Blake 已提交
3809
                                       " is taken by CD/DVD Drive");
3810
                            } else {
E
Eric Blake 已提交
3811 3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822 3823 3824 3825 3826 3827
                                PRInt32 channel          = 0;
                                PRInt32 device           = 0;
                                PRUnichar *hddcnameUtf16 = NULL;

                                char *hddcname = strdup("IDE");
                                VBOX_UTF8_TO_UTF16(hddcname, &hddcnameUtf16);
                                VIR_FREE(hddcname);

                                if (STREQ(def->disks[i]->dst, "hda")) {
                                    channel = 0;
                                    device  = 0;
                                } else if (STREQ(def->disks[i]->dst, "hdb")) {
                                    channel = 0;
                                    device  = 1;
                                } else if (STREQ(def->disks[i]->dst, "hdd")) {
                                    channel = 1;
                                    device  = 1;
3828
                                }
E
Eric Blake 已提交
3829 3830

                                rc = machine->vtbl->AttachHardDisk(machine,
3831
                                                                   hdduuid.value,
E
Eric Blake 已提交
3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842
                                                                   hddcnameUtf16,
                                                                   channel,
                                                                   device);
                                VBOX_UTF16_FREE(hddcnameUtf16);

                                if (NS_FAILED(rc)) {
                                    vboxError(VIR_ERR_INTERNAL_ERROR,
                                              _("could not attach the file as "
                                                "harddisk: %s, rc=%08x"),
                                              def->disks[i]->src, (unsigned)rc);
                                } else {
3843
                                    DEBUGIID("Attached HDD with UUID", hdduuid.value);
3844 3845 3846
                                }
                            }
                        }
3847
                    }
E
Eric Blake 已提交
3848 3849
                    VBOX_MEDIUM_RELEASE(hardDisk);
                }
3850
                vboxIIDUnalloc(&hdduuid);
E
Eric Blake 已提交
3851 3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864
                VBOX_UTF16_FREE(hddEmpty);
                VBOX_UTF16_FREE(hddfileUtf16);
            } else if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
            }
        } else if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_FLOPPY) {
            if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_FILE &&
                def->disks[i]->src != NULL) {
                IFloppyDrive *floppyDrive;
                machine->vtbl->GetFloppyDrive(machine, &floppyDrive);
                if (floppyDrive) {
                    rc = floppyDrive->vtbl->SetEnabled(floppyDrive, 1);
                    if (NS_SUCCEEDED(rc)) {
                        IFloppyImage *floppyImage   = NULL;
                        PRUnichar *fdfileUtf16      = NULL;
3865 3866
                        vboxIID fduuid = VBOX_IID_INITIALIZER;
                        vboxIID fdemptyuuid = VBOX_IID_INITIALIZER;
3867

E
Eric Blake 已提交
3868 3869 3870 3871
                        VBOX_UTF8_TO_UTF16(def->disks[i]->src, &fdfileUtf16);
                        rc = data->vboxObj->vtbl->FindFloppyImage(data->vboxObj,
                                                                  fdfileUtf16,
                                                                  &floppyImage);
3872

E
Eric Blake 已提交
3873 3874 3875
                        if (!floppyImage) {
                            data->vboxObj->vtbl->OpenFloppyImage(data->vboxObj,
                                                                 fdfileUtf16,
3876
                                                                 fdemptyuuid.value,
E
Eric Blake 已提交
3877 3878
                                                                 &floppyImage);
                        }
3879

E
Eric Blake 已提交
3880 3881
                        if (floppyImage) {
                            rc = floppyImage->vtbl->imedium.GetId((IMedium *)floppyImage,
3882
                                                                  &fduuid.value);
E
Eric Blake 已提交
3883 3884 3885 3886 3887 3888 3889
                            if (NS_FAILED(rc)) {
                                vboxError(VIR_ERR_INTERNAL_ERROR,
                                          _("can't get the uuid of the file to "
                                            "be attached to floppy drive: %s, rc=%08x"),
                                          def->disks[i]->src, (unsigned)rc);
                            } else {
                                rc = floppyDrive->vtbl->MountImage(floppyDrive,
3890
                                                                   fduuid.value);
E
Eric Blake 已提交
3891 3892 3893 3894 3895 3896
                                if (NS_FAILED(rc)) {
                                    vboxError(VIR_ERR_INTERNAL_ERROR,
                                              _("could not attach the file to "
                                                "floppy drive: %s, rc=%08x"),
                                              def->disks[i]->src, (unsigned)rc);
                                } else {
3897
                                    DEBUGIID("floppyImage UUID", fduuid.value);
3898 3899
                                }
                            }
E
Eric Blake 已提交
3900
                            VBOX_MEDIUM_RELEASE(floppyImage);
3901
                        }
3902
                        vboxIIDUnalloc(&fduuid);
E
Eric Blake 已提交
3903
                        VBOX_UTF16_FREE(fdfileUtf16);
3904
                    }
E
Eric Blake 已提交
3905
                    VBOX_RELEASE(floppyDrive);
3906
                }
E
Eric Blake 已提交
3907
            } else if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
3908
            }
3909
        }
E
Eric Blake 已提交
3910
    }
3911
#else  /* VBOX_API_VERSION >= 3001 */
E
Eric Blake 已提交
3912 3913 3914 3915 3916 3917 3918 3919 3920 3921 3922 3923
    PRUint32 maxPortPerInst[StorageBus_Floppy + 1] = {};
    PRUint32 maxSlotPerPort[StorageBus_Floppy + 1] = {};
    PRUnichar *storageCtlName = NULL;
    bool error = false;

    /* get the max port/slots/etc for the given storage bus */
    error = !vboxGetMaxPortSlotValues(data->vboxObj, maxPortPerInst,
                                      maxSlotPerPort);

    /* add a storage controller for the mediums to be attached */
    /* this needs to change when multiple controller are supported for
     * ver > 3.1 */
3924
    {
E
Eric Blake 已提交
3925 3926 3927 3928 3929 3930 3931 3932 3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959
        IStorageController *storageCtl = NULL;
        PRUnichar *sName = NULL;

        VBOX_UTF8_TO_UTF16("IDE Controller", &sName);
        machine->vtbl->AddStorageController(machine,
                                            sName,
                                            StorageBus_IDE,
                                            &storageCtl);
        VBOX_UTF16_FREE(sName);
        VBOX_RELEASE(storageCtl);

        VBOX_UTF8_TO_UTF16("SATA Controller", &sName);
        machine->vtbl->AddStorageController(machine,
                                            sName,
                                            StorageBus_SATA,
                                            &storageCtl);
        VBOX_UTF16_FREE(sName);
        VBOX_RELEASE(storageCtl);

        VBOX_UTF8_TO_UTF16("SCSI Controller", &sName);
        machine->vtbl->AddStorageController(machine,
                                            sName,
                                            StorageBus_SCSI,
                                            &storageCtl);
        VBOX_UTF16_FREE(sName);
        VBOX_RELEASE(storageCtl);

        VBOX_UTF8_TO_UTF16("Floppy Controller", &sName);
        machine->vtbl->AddStorageController(machine,
                                            sName,
                                            StorageBus_Floppy,
                                            &storageCtl);
        VBOX_UTF16_FREE(sName);
        VBOX_RELEASE(storageCtl);
    }
3960

E
Eric Blake 已提交
3961
    for (i = 0; i < def->ndisks && !error; i++) {
3962 3963 3964 3965 3966 3967 3968 3969 3970
        VIR_DEBUG("disk(%d) type:       %d", i, def->disks[i]->type);
        VIR_DEBUG("disk(%d) device:     %d", i, def->disks[i]->device);
        VIR_DEBUG("disk(%d) bus:        %d", i, def->disks[i]->bus);
        VIR_DEBUG("disk(%d) src:        %s", i, def->disks[i]->src);
        VIR_DEBUG("disk(%d) dst:        %s", i, def->disks[i]->dst);
        VIR_DEBUG("disk(%d) driverName: %s", i, def->disks[i]->driverName);
        VIR_DEBUG("disk(%d) driverType: %s", i, def->disks[i]->driverType);
        VIR_DEBUG("disk(%d) cachemode:  %d", i, def->disks[i]->cachemode);
        VIR_DEBUG("disk(%d) readonly:   %s", i, (def->disks[i]->readonly
E
Eric Blake 已提交
3971
                                             ? "True" : "False"));
3972
        VIR_DEBUG("disk(%d) shared:     %s", i, (def->disks[i]->shared
E
Eric Blake 已提交
3973 3974 3975 3976 3977 3978 3979 3980 3981
                                             ? "True" : "False"));

        if (def->disks[i]->type == VIR_DOMAIN_DISK_TYPE_FILE &&
            def->disks[i]->src != NULL) {
            IMedium   *medium          = NULL;
            PRUnichar *mediumUUID      = NULL;
            PRUnichar *mediumFileUtf16 = NULL;
            PRUint32   storageBus      = StorageBus_Null;
            PRUint32   deviceType      = DeviceType_Null;
3982
# if VBOX_API_VERSION >= 4000
3983
            PRUint32   accessMode      = AccessMode_ReadOnly;
3984
# endif
E
Eric Blake 已提交
3985 3986 3987 3988 3989 3990 3991 3992
            PRInt32    deviceInst      = 0;
            PRInt32    devicePort      = 0;
            PRInt32    deviceSlot      = 0;

            VBOX_UTF8_TO_UTF16(def->disks[i]->src, &mediumFileUtf16);

            if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
                deviceType = DeviceType_HardDisk;
3993
# if VBOX_API_VERSION < 4000
E
Eric Blake 已提交
3994 3995
                data->vboxObj->vtbl->FindHardDisk(data->vboxObj,
                                                  mediumFileUtf16, &medium);
3996 3997
# else
                accessMode = AccessMode_ReadWrite;
3998
# endif
E
Eric Blake 已提交
3999 4000
            } else if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_CDROM) {
                deviceType = DeviceType_DVD;
4001
# if VBOX_API_VERSION < 4000
E
Eric Blake 已提交
4002 4003
                data->vboxObj->vtbl->FindDVDImage(data->vboxObj,
                                                  mediumFileUtf16, &medium);
4004 4005
# else
                accessMode = AccessMode_ReadOnly;
4006
# endif
E
Eric Blake 已提交
4007 4008
            } else if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_FLOPPY) {
                deviceType = DeviceType_Floppy;
4009
# if VBOX_API_VERSION < 4000
E
Eric Blake 已提交
4010 4011
                data->vboxObj->vtbl->FindFloppyImage(data->vboxObj,
                                                     mediumFileUtf16, &medium);
4012 4013
# else
                accessMode = AccessMode_ReadWrite;
4014
# endif
E
Eric Blake 已提交
4015 4016 4017 4018
            } else {
                VBOX_UTF16_FREE(mediumFileUtf16);
                continue;
            }
4019

4020 4021 4022 4023 4024
# if VBOX_API_VERSION >= 4000
            data->vboxObj->vtbl->FindMedium(data->vboxObj, mediumFileUtf16,
                                            deviceType, &medium);
# endif

E
Eric Blake 已提交
4025 4026
            if (!medium) {
                PRUnichar *mediumEmpty = NULL;
4027

E
Eric Blake 已提交
4028
                VBOX_UTF8_TO_UTF16("", &mediumEmpty);
4029

4030
# if VBOX_API_VERSION < 4000
4031
                if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
4032 4033 4034 4035 4036 4037 4038 4039
                    rc = data->vboxObj->vtbl->OpenHardDisk(data->vboxObj,
                                                           mediumFileUtf16,
                                                           AccessMode_ReadWrite,
                                                           false,
                                                           mediumEmpty,
                                                           false,
                                                           mediumEmpty,
                                                           &medium);
E
Eric Blake 已提交
4040 4041
                } else if (def->disks[i]->device ==
                           VIR_DOMAIN_DISK_DEVICE_CDROM) {
4042 4043 4044 4045
                    rc = data->vboxObj->vtbl->OpenDVDImage(data->vboxObj,
                                                           mediumFileUtf16,
                                                           mediumEmpty,
                                                           &medium);
E
Eric Blake 已提交
4046 4047
                } else if (def->disks[i]->device ==
                           VIR_DOMAIN_DISK_DEVICE_FLOPPY) {
4048 4049 4050 4051 4052 4053
                    rc = data->vboxObj->vtbl->OpenFloppyImage(data->vboxObj,
                                                              mediumFileUtf16,
                                                              mediumEmpty,
                                                              &medium);
                } else {
                    rc = 0;
4054
                }
4055 4056 4057 4058 4059 4060
# else /* VBOX_API_VERSION >= 4000 */
                rc = data->vboxObj->vtbl->OpenMedium(data->vboxObj,
                                                     mediumFileUtf16,
                                                     deviceType, accessMode,
                                                     &medium);
# endif /* VBOX_API_VERSION >= 4000 */
4061

E
Eric Blake 已提交
4062 4063
                VBOX_UTF16_FREE(mediumEmpty);
            }
4064

E
Eric Blake 已提交
4065 4066 4067 4068 4069 4070 4071 4072
            if (!medium) {
                vboxError(VIR_ERR_INTERNAL_ERROR,
                          _("Failed to attach the following disk/dvd/floppy "
                            "to the machine: %s, rc=%08x"),
                          def->disks[i]->src, (unsigned)rc);
                VBOX_UTF16_FREE(mediumFileUtf16);
                continue;
            }
4073

E
Eric Blake 已提交
4074 4075 4076 4077 4078 4079 4080 4081 4082 4083
            rc = medium->vtbl->GetId(medium, &mediumUUID);
            if (NS_FAILED(rc)) {
                vboxError(VIR_ERR_INTERNAL_ERROR,
                          _("can't get the uuid of the file to be attached "
                            "as harddisk/dvd/floppy: %s, rc=%08x"),
                          def->disks[i]->src, (unsigned)rc);
                VBOX_RELEASE(medium);
                VBOX_UTF16_FREE(mediumFileUtf16);
                continue;
            }
4084

E
Eric Blake 已提交
4085 4086 4087
            if (def->disks[i]->device == VIR_DOMAIN_DISK_DEVICE_DISK) {
                if (def->disks[i]->readonly) {
                    medium->vtbl->SetType(medium, MediumType_Immutable);
4088
                    VIR_DEBUG("setting harddisk to immutable");
E
Eric Blake 已提交
4089 4090
                } else if (!def->disks[i]->readonly) {
                    medium->vtbl->SetType(medium, MediumType_Normal);
4091
                    VIR_DEBUG("setting harddisk type to normal");
4092
                }
E
Eric Blake 已提交
4093
            }
4094

E
Eric Blake 已提交
4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107
            if (def->disks[i]->bus == VIR_DOMAIN_DISK_BUS_IDE) {
                VBOX_UTF8_TO_UTF16("IDE Controller", &storageCtlName);
                storageBus = StorageBus_IDE;
            } else if (def->disks[i]->bus == VIR_DOMAIN_DISK_BUS_SATA) {
                VBOX_UTF8_TO_UTF16("SATA Controller", &storageCtlName);
                storageBus = StorageBus_SATA;
            } else if (def->disks[i]->bus == VIR_DOMAIN_DISK_BUS_SCSI) {
                VBOX_UTF8_TO_UTF16("SCSI Controller", &storageCtlName);
                storageBus = StorageBus_SCSI;
            } else if (def->disks[i]->bus == VIR_DOMAIN_DISK_BUS_FDC) {
                VBOX_UTF8_TO_UTF16("Floppy Controller", &storageCtlName);
                storageBus = StorageBus_Floppy;
            }
4108

E
Eric Blake 已提交
4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120
            /* get the device details i.e instance, port and slot */
            if (!vboxGetDeviceDetails(def->disks[i]->dst,
                                      maxPortPerInst,
                                      maxSlotPerPort,
                                      storageBus,
                                      &deviceInst,
                                      &devicePort,
                                      &deviceSlot)) {
                vboxError(VIR_ERR_INTERNAL_ERROR,
                          _("can't get the port/slot number of harddisk/"
                            "dvd/floppy to be attached: %s, rc=%08x"),
                          def->disks[i]->src, (unsigned)rc);
4121 4122 4123
                VBOX_RELEASE(medium);
                VBOX_UTF16_FREE(mediumUUID);
                VBOX_UTF16_FREE(mediumFileUtf16);
E
Eric Blake 已提交
4124 4125 4126 4127 4128 4129 4130 4131 4132
                continue;
            }

            /* attach the harddisk/dvd/Floppy to the storage controller */
            rc = machine->vtbl->AttachDevice(machine,
                                             storageCtlName,
                                             devicePort,
                                             deviceSlot,
                                             deviceType,
4133
# if VBOX_API_VERSION < 4000
E
Eric Blake 已提交
4134
                                             mediumUUID);
4135 4136 4137
# else /* VBOX_API_VERSION >= 4000 */
                                             medium);
# endif /* VBOX_API_VERSION >= 4000 */
E
Eric Blake 已提交
4138 4139 4140 4141 4142 4143 4144 4145

            if (NS_FAILED(rc)) {
                vboxError(VIR_ERR_INTERNAL_ERROR,
                          _("could not attach the file as harddisk/"
                            "dvd/floppy: %s, rc=%08x"),
                          def->disks[i]->src, (unsigned)rc);
            } else {
                DEBUGIID("Attached HDD/DVD/Floppy with UUID", mediumUUID);
4146
            }
E
Eric Blake 已提交
4147 4148 4149 4150 4151

            VBOX_RELEASE(medium);
            VBOX_UTF16_FREE(mediumUUID);
            VBOX_UTF16_FREE(mediumFileUtf16);
            VBOX_UTF16_FREE(storageCtlName);
4152 4153 4154
        }
    }
#endif /* VBOX_API_VERSION >= 3001 */
E
Eric Blake 已提交
4155
}
4156

E
Eric Blake 已提交
4157 4158 4159 4160
static void
vboxAttachSound(virDomainDefPtr def, IMachine *machine)
{
    nsresult rc;
4161

E
Eric Blake 已提交
4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177
    /* Check if def->nsounds is one as VirtualBox currently supports
     * only one sound card
     */
    if (def->nsounds == 1) {
        IAudioAdapter *audioAdapter = NULL;

        machine->vtbl->GetAudioAdapter(machine, &audioAdapter);
        if (audioAdapter) {
            rc = audioAdapter->vtbl->SetEnabled(audioAdapter, 1);
            if (NS_SUCCEEDED(rc)) {
                if (def->sounds[0]->model == VIR_DOMAIN_SOUND_MODEL_SB16) {
                    audioAdapter->vtbl->SetAudioController(audioAdapter,
                                                           AudioControllerType_SB16);
                } else if (def->sounds[0]->model == VIR_DOMAIN_SOUND_MODEL_AC97) {
                    audioAdapter->vtbl->SetAudioController(audioAdapter,
                                                           AudioControllerType_AC97);
4178
                }
4179
            }
E
Eric Blake 已提交
4180
            VBOX_RELEASE(audioAdapter);
4181
        }
E
Eric Blake 已提交
4182 4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199
    }
}

static void
vboxAttachNetwork(virDomainDefPtr def, vboxGlobalData *data, IMachine *machine)
{
    ISystemProperties *systemProperties = NULL;
    PRUint32 networkAdapterCount        = 0;
    int i = 0;

    data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
    if (systemProperties) {
        systemProperties->vtbl->GetNetworkAdapterCount(systemProperties,
                                                       &networkAdapterCount);
        VBOX_RELEASE(systemProperties);
        systemProperties = NULL;
    }

4200 4201
    VIR_DEBUG("Number of Network Cards to be connected: %d", def->nnets);
    VIR_DEBUG("Number of Network Cards available: %d", networkAdapterCount);
E
Eric Blake 已提交
4202 4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213 4214 4215 4216 4217 4218 4219

    for (i = 0; (i < def->nnets) && (i < networkAdapterCount); i++) {
        INetworkAdapter *adapter = NULL;
        PRUint32 adapterType     = NetworkAdapterType_Null;
        char macaddr[VIR_MAC_STRING_BUFLEN] = {0};
        char macaddrvbox[VIR_MAC_STRING_BUFLEN - 5] = {0};

        virFormatMacAddr(def->nets[i]->mac, macaddr);
        snprintf(macaddrvbox, VIR_MAC_STRING_BUFLEN - 5,
                 "%02X%02X%02X%02X%02X%02X",
                 def->nets[i]->mac[0],
                 def->nets[i]->mac[1],
                 def->nets[i]->mac[2],
                 def->nets[i]->mac[3],
                 def->nets[i]->mac[4],
                 def->nets[i]->mac[5]);
        macaddrvbox[VIR_MAC_STRING_BUFLEN - 6] = '\0';

4220 4221 4222 4223
        VIR_DEBUG("NIC(%d): Type:   %d", i, def->nets[i]->type);
        VIR_DEBUG("NIC(%d): Model:  %s", i, def->nets[i]->model);
        VIR_DEBUG("NIC(%d): Mac:    %s", i, macaddr);
        VIR_DEBUG("NIC(%d): ifname: %s", i, def->nets[i]->ifname);
E
Eric Blake 已提交
4224
        if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_NETWORK) {
4225
            VIR_DEBUG("NIC(%d): name:    %s", i, def->nets[i]->data.network.name);
E
Eric Blake 已提交
4226
        } else if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_INTERNAL) {
4227
            VIR_DEBUG("NIC(%d): name:   %s", i, def->nets[i]->data.internal.name);
E
Eric Blake 已提交
4228
        } else if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_USER) {
4229
            VIR_DEBUG("NIC(%d): NAT.", i);
E
Eric Blake 已提交
4230
        } else if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_BRIDGE) {
4231 4232 4233
            VIR_DEBUG("NIC(%d): brname: %s", i, def->nets[i]->data.bridge.brname);
            VIR_DEBUG("NIC(%d): script: %s", i, def->nets[i]->data.bridge.script);
            VIR_DEBUG("NIC(%d): ipaddr: %s", i, def->nets[i]->data.bridge.ipaddr);
4234 4235
        }

E
Eric Blake 已提交
4236 4237 4238 4239 4240
        machine->vtbl->GetNetworkAdapter(machine, i, &adapter);
        if (adapter) {
            PRUnichar *MACAddress = NULL;

            adapter->vtbl->SetEnabled(adapter, 1);
4241

E
Eric Blake 已提交
4242 4243 4244 4245 4246 4247 4248 4249 4250 4251 4252
            if (def->nets[i]->model) {
                if (STRCASEEQ(def->nets[i]->model , "Am79C970A")) {
                    adapterType = NetworkAdapterType_Am79C970A;
                } else if (STRCASEEQ(def->nets[i]->model , "Am79C973")) {
                    adapterType = NetworkAdapterType_Am79C973;
                } else if (STRCASEEQ(def->nets[i]->model , "82540EM")) {
                    adapterType = NetworkAdapterType_I82540EM;
                } else if (STRCASEEQ(def->nets[i]->model , "82545EM")) {
                    adapterType = NetworkAdapterType_I82545EM;
                } else if (STRCASEEQ(def->nets[i]->model , "82543GC")) {
                    adapterType = NetworkAdapterType_I82543GC;
4253
#if VBOX_API_VERSION >= 3001
E
Eric Blake 已提交
4254 4255
                } else if (STRCASEEQ(def->nets[i]->model , "virtio")) {
                    adapterType = NetworkAdapterType_Virtio;
4256 4257
#endif /* VBOX_API_VERSION >= 3001 */
                }
E
Eric Blake 已提交
4258 4259 4260
            } else {
                adapterType = NetworkAdapterType_Am79C973;
            }
4261

E
Eric Blake 已提交
4262
            adapter->vtbl->SetAdapterType(adapter, adapterType);
4263

E
Eric Blake 已提交
4264 4265 4266
            if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_BRIDGE) {
                PRUnichar *hostInterface = NULL;
                /* Bridged Network */
4267

E
Eric Blake 已提交
4268
                adapter->vtbl->AttachToBridgedInterface(adapter);
4269

E
Eric Blake 已提交
4270 4271 4272 4273 4274 4275 4276 4277 4278
                if (def->nets[i]->data.bridge.brname) {
                    VBOX_UTF8_TO_UTF16(def->nets[i]->data.bridge.brname,
                                       &hostInterface);
                    adapter->vtbl->SetHostInterface(adapter, hostInterface);
                    VBOX_UTF16_FREE(hostInterface);
                }
            } else if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_INTERNAL) {
                PRUnichar *internalNetwork = NULL;
                /* Internal Network */
4279

E
Eric Blake 已提交
4280
                adapter->vtbl->AttachToInternalNetwork(adapter);
4281

E
Eric Blake 已提交
4282 4283 4284 4285 4286
                if (def->nets[i]->data.internal.name) {
                    VBOX_UTF8_TO_UTF16(def->nets[i]->data.internal.name,
                                       &internalNetwork);
                    adapter->vtbl->SetInternalNetwork(adapter, internalNetwork);
                    VBOX_UTF16_FREE(internalNetwork);
4287
                }
E
Eric Blake 已提交
4288 4289 4290 4291 4292 4293 4294
            } else if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_NETWORK) {
                PRUnichar *hostInterface = NULL;
                /* Host Only Networking (currently only vboxnet0 available
                 * on *nix and mac, on windows you can create and configure
                 * as many as you want)
                 */
                adapter->vtbl->AttachToHostOnlyInterface(adapter);
4295

E
Eric Blake 已提交
4296 4297 4298 4299 4300 4301 4302 4303 4304 4305 4306 4307 4308 4309
                if (def->nets[i]->data.network.name) {
                    VBOX_UTF8_TO_UTF16(def->nets[i]->data.network.name,
                                       &hostInterface);
                    adapter->vtbl->SetHostInterface(adapter, hostInterface);
                    VBOX_UTF16_FREE(hostInterface);
                }
            } else if (def->nets[i]->type == VIR_DOMAIN_NET_TYPE_USER) {
                /* NAT */
                adapter->vtbl->AttachToNAT(adapter);
            } else {
                /* else always default to NAT if we don't understand
                 * what option is been passed to us
                 */
                adapter->vtbl->AttachToNAT(adapter);
4310
            }
E
Eric Blake 已提交
4311 4312 4313 4314

            VBOX_UTF8_TO_UTF16(macaddrvbox, &MACAddress);
            adapter->vtbl->SetMACAddress(adapter, MACAddress);
            VBOX_UTF16_FREE(MACAddress);
4315
        }
E
Eric Blake 已提交
4316 4317 4318 4319 4320 4321 4322 4323 4324
    }
}

static void
vboxAttachSerial(virDomainDefPtr def, vboxGlobalData *data, IMachine *machine)
{
    ISystemProperties *systemProperties = NULL;
    PRUint32 serialPortCount            = 0;
    int i = 0;
4325

E
Eric Blake 已提交
4326 4327 4328 4329 4330 4331 4332
    data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
    if (systemProperties) {
        systemProperties->vtbl->GetSerialPortCount(systemProperties,
                                                   &serialPortCount);
        VBOX_RELEASE(systemProperties);
        systemProperties = NULL;
    }
4333

4334 4335
    VIR_DEBUG("Number of Serial Ports to be connected: %d", def->nserials);
    VIR_DEBUG("Number of Serial Ports available: %d", serialPortCount);
E
Eric Blake 已提交
4336 4337
    for (i = 0; (i < def->nserials) && (i < serialPortCount); i++) {
        ISerialPort *serialPort = NULL;
4338

4339 4340
        VIR_DEBUG("SerialPort(%d): Type: %d", i, def->serials[i]->source.type);
        VIR_DEBUG("SerialPort(%d): target.port: %d", i,
E
Eric Blake 已提交
4341
              def->serials[i]->target.port);
4342

E
Eric Blake 已提交
4343 4344 4345
        machine->vtbl->GetSerialPort(machine, i, &serialPort);
        if (serialPort) {
            PRUnichar *pathUtf16 = NULL;
4346

E
Eric Blake 已提交
4347
            serialPort->vtbl->SetEnabled(serialPort, 1);
4348

4349 4350 4351
            if (def->serials[i]->source.data.file.path) {
                VBOX_UTF8_TO_UTF16(def->serials[i]->source.data.file.path,
                                   &pathUtf16);
E
Eric Blake 已提交
4352 4353 4354 4355 4356 4357 4358 4359 4360 4361 4362 4363 4364 4365 4366
                serialPort->vtbl->SetPath(serialPort, pathUtf16);
            }

            /* For now hard code the serial ports to COM1 and COM2,
             * COM1 (Base Addr: 0x3F8 (decimal: 1016), IRQ: 4)
             * COM2 (Base Addr: 0x2F8 (decimal:  760), IRQ: 3)
             * TODO: make this more flexible
             */
            /* TODO: to improve the libvirt XMl handling so
             * that def->serials[i]->target.port shows real port
             * and not always start at 0
             */
            if (def->serials[i]->target.port == 0) {
                serialPort->vtbl->SetIRQ(serialPort, 4);
                serialPort->vtbl->SetIOBase(serialPort, 1016);
4367
                VIR_DEBUG(" serialPort-%d irq: %d, iobase 0x%x, path: %s",
4368
                      i, 4, 1016, def->serials[i]->source.data.file.path);
E
Eric Blake 已提交
4369 4370 4371
            } else if (def->serials[i]->target.port == 1) {
                serialPort->vtbl->SetIRQ(serialPort, 3);
                serialPort->vtbl->SetIOBase(serialPort, 760);
4372
                VIR_DEBUG(" serialPort-%d irq: %d, iobase 0x%x, path: %s",
4373
                      i, 3, 760, def->serials[i]->source.data.file.path);
E
Eric Blake 已提交
4374
            }
4375

4376
            if (def->serials[i]->source.type == VIR_DOMAIN_CHR_TYPE_DEV) {
E
Eric Blake 已提交
4377
                serialPort->vtbl->SetHostMode(serialPort, PortMode_HostDevice);
4378
            } else if (def->serials[i]->source.type == VIR_DOMAIN_CHR_TYPE_PIPE) {
E
Eric Blake 已提交
4379
                serialPort->vtbl->SetHostMode(serialPort, PortMode_HostPipe);
4380
#if VBOX_API_VERSION >= 3000
4381
            } else if (def->serials[i]->source.type == VIR_DOMAIN_CHR_TYPE_FILE) {
E
Eric Blake 已提交
4382
                serialPort->vtbl->SetHostMode(serialPort, PortMode_RawFile);
4383
#endif /* VBOX_API_VERSION >= 3000 */
E
Eric Blake 已提交
4384 4385 4386 4387
            } else {
                serialPort->vtbl->SetHostMode(serialPort,
                                              PortMode_Disconnected);
            }
4388

E
Eric Blake 已提交
4389 4390 4391 4392
            VBOX_RELEASE(serialPort);
            if (pathUtf16) {
                VBOX_UTF16_FREE(pathUtf16);
                pathUtf16 = NULL;
4393
            }
4394
        }
E
Eric Blake 已提交
4395 4396
    }
}
4397

E
Eric Blake 已提交
4398 4399 4400 4401 4402 4403
static void
vboxAttachParallel(virDomainDefPtr def, vboxGlobalData *data, IMachine *machine)
{
    ISystemProperties *systemProperties = NULL;
    PRUint32 parallelPortCount          = 0;
    int i = 0;
4404

E
Eric Blake 已提交
4405 4406 4407 4408 4409 4410 4411
    data->vboxObj->vtbl->GetSystemProperties(data->vboxObj, &systemProperties);
    if (systemProperties) {
        systemProperties->vtbl->GetParallelPortCount(systemProperties,
                                                     &parallelPortCount);
        VBOX_RELEASE(systemProperties);
        systemProperties = NULL;
    }
4412

4413 4414
    VIR_DEBUG("Number of Parallel Ports to be connected: %d", def->nparallels);
    VIR_DEBUG("Number of Parallel Ports available: %d", parallelPortCount);
E
Eric Blake 已提交
4415 4416
    for (i = 0; (i < def->nparallels) && (i < parallelPortCount); i++) {
        IParallelPort *parallelPort = NULL;
4417

4418 4419
        VIR_DEBUG("ParallelPort(%d): Type: %d", i, def->parallels[i]->source.type);
        VIR_DEBUG("ParallelPort(%d): target.port: %d", i,
E
Eric Blake 已提交
4420
              def->parallels[i]->target.port);
4421

E
Eric Blake 已提交
4422 4423 4424
        machine->vtbl->GetParallelPort(machine, i, &parallelPort);
        if (parallelPort) {
            PRUnichar *pathUtf16 = NULL;
4425

4426
            VBOX_UTF8_TO_UTF16(def->parallels[i]->source.data.file.path, &pathUtf16);
4427

E
Eric Blake 已提交
4428 4429 4430 4431 4432
            /* For now hard code the parallel ports to
             * LPT1 (Base Addr: 0x378 (decimal: 888), IRQ: 7)
             * LPT2 (Base Addr: 0x278 (decimal: 632), IRQ: 5)
             * TODO: make this more flexible
             */
4433 4434 4435 4436
            if ((def->parallels[i]->source.type == VIR_DOMAIN_CHR_TYPE_DEV)  ||
                (def->parallels[i]->source.type == VIR_DOMAIN_CHR_TYPE_PTY)  ||
                (def->parallels[i]->source.type == VIR_DOMAIN_CHR_TYPE_FILE) ||
                (def->parallels[i]->source.type == VIR_DOMAIN_CHR_TYPE_PIPE)) {
E
Eric Blake 已提交
4437 4438 4439 4440
                parallelPort->vtbl->SetPath(parallelPort, pathUtf16);
                if (i == 0) {
                    parallelPort->vtbl->SetIRQ(parallelPort, 7);
                    parallelPort->vtbl->SetIOBase(parallelPort, 888);
4441
                    VIR_DEBUG(" parallePort-%d irq: %d, iobase 0x%x, path: %s",
4442
                          i, 7, 888, def->parallels[i]->source.data.file.path);
E
Eric Blake 已提交
4443 4444 4445
                } else if (i == 1) {
                    parallelPort->vtbl->SetIRQ(parallelPort, 5);
                    parallelPort->vtbl->SetIOBase(parallelPort, 632);
4446
                    VIR_DEBUG(" parallePort-%d irq: %d, iobase 0x%x, path: %s",
4447
                          i, 5, 632, def->parallels[i]->source.data.file.path);
4448 4449
                }
            }
E
Eric Blake 已提交
4450 4451 4452 4453 4454 4455 4456 4457 4458 4459 4460

            /* like serial port, parallel port can't be enabled unless
             * correct IRQ and IOBase values are specified.
             */
            parallelPort->vtbl->SetEnabled(parallelPort, 1);

            VBOX_RELEASE(parallelPort);
            if (pathUtf16) {
                VBOX_UTF16_FREE(pathUtf16);
                pathUtf16 = NULL;
            }
4461
        }
E
Eric Blake 已提交
4462 4463 4464 4465 4466 4467 4468 4469
    }
}

static void
vboxAttachVideo(virDomainDefPtr def, IMachine *machine)
{
    if ((def->nvideos == 1) &&
        (def->videos[0]->type == VIR_DOMAIN_VIDEO_TYPE_VBOX)) {
4470 4471
        machine->vtbl->SetVRAMSize(machine,
                                   VIR_DIV_UP(def->videos[0]->vram, 1024));
E
Eric Blake 已提交
4472 4473 4474 4475
        machine->vtbl->SetMonitorCount(machine, def->videos[0]->heads);
        if (def->videos[0]->accel) {
            machine->vtbl->SetAccelerate3DEnabled(machine,
                                                  def->videos[0]->accel->support3d);
4476
#if VBOX_API_VERSION >= 3001
E
Eric Blake 已提交
4477 4478
            machine->vtbl->SetAccelerate2DVideoEnabled(machine,
                                                       def->videos[0]->accel->support2d);
4479
#endif /* VBOX_API_VERSION >= 3001 */
E
Eric Blake 已提交
4480 4481
        } else {
            machine->vtbl->SetAccelerate3DEnabled(machine, 0);
4482
#if VBOX_API_VERSION >= 3001
E
Eric Blake 已提交
4483
            machine->vtbl->SetAccelerate2DVideoEnabled(machine, 0);
4484 4485
#endif /* VBOX_API_VERSION >= 3001 */
        }
E
Eric Blake 已提交
4486 4487
    }
}
4488

E
Eric Blake 已提交
4489 4490 4491 4492 4493 4494 4495 4496 4497
static void
vboxAttachDisplay(virDomainDefPtr def, vboxGlobalData *data, IMachine *machine)
{
    int vrdpPresent  = 0;
    int sdlPresent   = 0;
    int guiPresent   = 0;
    char *guiDisplay = NULL;
    char *sdlDisplay = NULL;
    int i = 0;
4498

E
Eric Blake 已提交
4499
    for (i = 0; i < def->ngraphics; i++) {
4500 4501 4502 4503 4504
#if VBOX_API_VERSION < 4000
        IVRDPServer *VRDxServer = NULL;
#else /* VBOX_API_VERSION >= 4000 */
        IVRDEServer *VRDxServer = NULL;
#endif /* VBOX_API_VERSION >= 4000 */
4505

E
Eric Blake 已提交
4506 4507
        if ((def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_RDP) &&
            (vrdpPresent == 0)) {
4508

E
Eric Blake 已提交
4509
            vrdpPresent = 1;
4510 4511 4512 4513 4514 4515 4516
#if VBOX_API_VERSION < 4000
            machine->vtbl->GetVRDPServer(machine, &VRDxServer);
#else /* VBOX_API_VERSION >= 4000 */
            machine->vtbl->GetVRDEServer(machine, &VRDxServer);
#endif /* VBOX_API_VERSION >= 4000 */
            if (VRDxServer) {
                VRDxServer->vtbl->SetEnabled(VRDxServer, PR_TRUE);
4517
                VIR_DEBUG("VRDP Support turned ON.");
4518 4519

#if VBOX_API_VERSION < 3001
E
Eric Blake 已提交
4520
                if (def->graphics[i]->data.rdp.port) {
4521
                    VRDxServer->vtbl->SetPort(VRDxServer,
E
Eric Blake 已提交
4522
                                              def->graphics[i]->data.rdp.port);
4523
                    VIR_DEBUG("VRDP Port changed to: %d",
E
Eric Blake 已提交
4524 4525 4526 4527 4528
                          def->graphics[i]->data.rdp.port);
                } else if (def->graphics[i]->data.rdp.autoport) {
                    /* Setting the port to 0 will reset its value to
                     * the default one which is 3389 currently
                     */
4529
                    VRDxServer->vtbl->SetPort(VRDxServer, 0);
4530
                    VIR_DEBUG("VRDP Port changed to default, which is 3389 currently");
E
Eric Blake 已提交
4531
                }
4532
#elif VBOX_API_VERSION < 4000 /* 3001 <= VBOX_API_VERSION < 4000 */
E
Eric Blake 已提交
4533 4534
                PRUnichar *portUtf16 = NULL;
                portUtf16 = PRUnicharFromInt(def->graphics[i]->data.rdp.port);
4535
                VRDxServer->vtbl->SetPorts(VRDxServer, portUtf16);
E
Eric Blake 已提交
4536
                VBOX_UTF16_FREE(portUtf16);
4537 4538 4539 4540 4541 4542 4543 4544 4545 4546
#else /* VBOX_API_VERSION >= 4000 */
                PRUnichar *VRDEPortsKey = NULL;
                PRUnichar *VRDEPortsValue = NULL;
                VBOX_UTF8_TO_UTF16("TCP/Ports", &VRDEPortsKey);
                VRDEPortsValue = PRUnicharFromInt(def->graphics[i]->data.rdp.port);
                VRDxServer->vtbl->SetVRDEProperty(VRDxServer, VRDEPortsKey,
                                                  VRDEPortsValue);
                VBOX_UTF16_FREE(VRDEPortsKey);
                VBOX_UTF16_FREE(VRDEPortsValue);
#endif /* VBOX_API_VERSION >= 4000 */
4547

E
Eric Blake 已提交
4548
                if (def->graphics[i]->data.rdp.replaceUser) {
4549
                    VRDxServer->vtbl->SetReuseSingleConnection(VRDxServer,
E
Eric Blake 已提交
4550
                                                               PR_TRUE);
4551
                    VIR_DEBUG("VRDP set to reuse single connection");
E
Eric Blake 已提交
4552
                }
4553

E
Eric Blake 已提交
4554
                if (def->graphics[i]->data.rdp.multiUser) {
4555
                    VRDxServer->vtbl->SetAllowMultiConnection(VRDxServer,
E
Eric Blake 已提交
4556
                                                              PR_TRUE);
4557
                    VIR_DEBUG("VRDP set to allow multiple connection");
E
Eric Blake 已提交
4558
                }
4559

E
Eric Blake 已提交
4560
                if (def->graphics[i]->data.rdp.listenAddr) {
4561 4562 4563
#if VBOX_API_VERSION >= 4000
                    PRUnichar *netAddressKey = NULL;
#endif
E
Eric Blake 已提交
4564
                    PRUnichar *netAddressUtf16 = NULL;
4565

E
Eric Blake 已提交
4566 4567
                    VBOX_UTF8_TO_UTF16(def->graphics[i]->data.rdp.listenAddr,
                                       &netAddressUtf16);
4568 4569
#if VBOX_API_VERSION < 4000
                    VRDxServer->vtbl->SetNetAddress(VRDxServer,
E
Eric Blake 已提交
4570
                                                    netAddressUtf16);
4571 4572 4573 4574 4575 4576
#else /* VBOX_API_VERSION >= 4000 */
                    VBOX_UTF8_TO_UTF16("TCP/Address", &netAddressKey);
                    VRDxServer->vtbl->SetVRDEProperty(VRDxServer, netAddressKey,
                                                      netAddressUtf16);
                    VBOX_UTF16_FREE(netAddressKey);
#endif /* VBOX_API_VERSION >= 4000 */
4577
                    VIR_DEBUG("VRDP listen address is set to: %s",
E
Eric Blake 已提交
4578
                          def->graphics[i]->data.rdp.listenAddr);
4579

E
Eric Blake 已提交
4580
                    VBOX_UTF16_FREE(netAddressUtf16);
4581
                }
E
Eric Blake 已提交
4582

4583
                VBOX_RELEASE(VRDxServer);
4584
            }
E
Eric Blake 已提交
4585
        }
4586

E
Eric Blake 已提交
4587 4588 4589 4590 4591 4592 4593 4594 4595 4596 4597
        if ((def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_DESKTOP) &&
            (guiPresent == 0)) {
            guiPresent = 1;
            if (def->graphics[i]->data.desktop.display) {
                guiDisplay = strdup(def->graphics[i]->data.desktop.display);
                if (guiDisplay == NULL) {
                    virReportOOMError();
                    /* just don't go to cleanup yet as it is ok to have
                     * guiDisplay as NULL and we check it below if it
                     * exist and then only use it there
                     */
4598
                }
4599
            }
E
Eric Blake 已提交
4600
        }
4601

E
Eric Blake 已提交
4602 4603 4604 4605 4606 4607 4608 4609 4610 4611 4612
        if ((def->graphics[i]->type == VIR_DOMAIN_GRAPHICS_TYPE_SDL) &&
            (sdlPresent == 0)) {
            sdlPresent = 1;
            if (def->graphics[i]->data.sdl.display) {
                sdlDisplay = strdup(def->graphics[i]->data.sdl.display);
                if (sdlDisplay == NULL) {
                    virReportOOMError();
                    /* just don't go to cleanup yet as it is ok to have
                     * sdlDisplay as NULL and we check it below if it
                     * exist and then only use it there
                     */
4613
                }
4614
            }
4615
        }
E
Eric Blake 已提交
4616
    }
4617

E
Eric Blake 已提交
4618 4619 4620 4621
    if ((vrdpPresent == 1) && (guiPresent == 0) && (sdlPresent == 0)) {
        /* store extradata key that frontend is set to vrdp */
        PRUnichar *keyTypeUtf16   = NULL;
        PRUnichar *valueTypeUtf16 = NULL;
4622

E
Eric Blake 已提交
4623 4624
        VBOX_UTF8_TO_UTF16("FRONTEND/Type", &keyTypeUtf16);
        VBOX_UTF8_TO_UTF16("vrdp", &valueTypeUtf16);
4625

E
Eric Blake 已提交
4626
        machine->vtbl->SetExtraData(machine, keyTypeUtf16, valueTypeUtf16);
4627

E
Eric Blake 已提交
4628 4629
        VBOX_UTF16_FREE(keyTypeUtf16);
        VBOX_UTF16_FREE(valueTypeUtf16);
4630

E
Eric Blake 已提交
4631 4632 4633 4634 4635 4636
    } else if ((guiPresent == 0) && (sdlPresent == 1)) {
        /* store extradata key that frontend is set to sdl */
        PRUnichar *keyTypeUtf16      = NULL;
        PRUnichar *valueTypeUtf16    = NULL;
        PRUnichar *keyDislpayUtf16   = NULL;
        PRUnichar *valueDisplayUtf16 = NULL;
4637

E
Eric Blake 已提交
4638 4639
        VBOX_UTF8_TO_UTF16("FRONTEND/Type", &keyTypeUtf16);
        VBOX_UTF8_TO_UTF16("sdl", &valueTypeUtf16);
4640

E
Eric Blake 已提交
4641
        machine->vtbl->SetExtraData(machine, keyTypeUtf16, valueTypeUtf16);
4642

E
Eric Blake 已提交
4643 4644
        VBOX_UTF16_FREE(keyTypeUtf16);
        VBOX_UTF16_FREE(valueTypeUtf16);
4645

E
Eric Blake 已提交
4646 4647 4648
        if (sdlDisplay) {
            VBOX_UTF8_TO_UTF16("FRONTEND/Display", &keyDislpayUtf16);
            VBOX_UTF8_TO_UTF16(sdlDisplay, &valueDisplayUtf16);
4649

E
Eric Blake 已提交
4650 4651
            machine->vtbl->SetExtraData(machine, keyDislpayUtf16,
                                        valueDisplayUtf16);
4652

E
Eric Blake 已提交
4653 4654 4655
            VBOX_UTF16_FREE(keyDislpayUtf16);
            VBOX_UTF16_FREE(valueDisplayUtf16);
        }
4656

E
Eric Blake 已提交
4657 4658 4659 4660 4661 4662
    } else {
        /* if all are set then default is gui, with vrdp turned on */
        PRUnichar *keyTypeUtf16      = NULL;
        PRUnichar *valueTypeUtf16    = NULL;
        PRUnichar *keyDislpayUtf16   = NULL;
        PRUnichar *valueDisplayUtf16 = NULL;
4663

E
Eric Blake 已提交
4664 4665
        VBOX_UTF8_TO_UTF16("FRONTEND/Type", &keyTypeUtf16);
        VBOX_UTF8_TO_UTF16("gui", &valueTypeUtf16);
4666

E
Eric Blake 已提交
4667
        machine->vtbl->SetExtraData(machine, keyTypeUtf16, valueTypeUtf16);
4668

E
Eric Blake 已提交
4669 4670
        VBOX_UTF16_FREE(keyTypeUtf16);
        VBOX_UTF16_FREE(valueTypeUtf16);
4671

E
Eric Blake 已提交
4672 4673 4674
        if (guiDisplay) {
            VBOX_UTF8_TO_UTF16("FRONTEND/Display", &keyDislpayUtf16);
            VBOX_UTF8_TO_UTF16(guiDisplay, &valueDisplayUtf16);
4675

E
Eric Blake 已提交
4676 4677
            machine->vtbl->SetExtraData(machine, keyDislpayUtf16,
                                        valueDisplayUtf16);
4678

E
Eric Blake 已提交
4679 4680
            VBOX_UTF16_FREE(keyDislpayUtf16);
            VBOX_UTF16_FREE(valueDisplayUtf16);
4681
        }
E
Eric Blake 已提交
4682
    }
4683

E
Eric Blake 已提交
4684 4685 4686
    VIR_FREE(guiDisplay);
    VIR_FREE(sdlDisplay);
}
4687

E
Eric Blake 已提交
4688 4689 4690 4691 4692 4693 4694 4695 4696 4697 4698 4699 4700 4701 4702 4703 4704 4705 4706 4707 4708 4709
static void
vboxAttachUSB(virDomainDefPtr def, vboxGlobalData *data, IMachine *machine)
{
    IUSBController *USBController = NULL;
    int i = 0;
    bool isUSB = false;

    if (def->nhostdevs == 0)
        return;

    /* Loop through the devices first and see if you
     * have a USB Device, only if you have one then
     * start the USB controller else just proceed as
     * usual
     */
    for (i = 0; i < def->nhostdevs; i++) {
        if (def->hostdevs[i]->mode ==
            VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {
            if (def->hostdevs[i]->source.subsys.type ==
                VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
                if (def->hostdevs[i]->source.subsys.u.usb.vendor ||
                    def->hostdevs[i]->source.subsys.u.usb.product) {
4710
                    VIR_DEBUG("USB Device detected, VendorId:0x%x, ProductId:0x%x",
E
Eric Blake 已提交
4711 4712 4713 4714 4715 4716 4717 4718 4719 4720 4721 4722 4723 4724 4725 4726 4727
                          def->hostdevs[i]->source.subsys.u.usb.vendor,
                          def->hostdevs[i]->source.subsys.u.usb.product);
                    isUSB = true;
                    break;
                }
            }
        }
    }

    if (isUSB) {
        /* First Start the USB Controller and then loop
         * to attach USB Devices to it
         */
        machine->vtbl->GetUSBController(machine, &USBController);
        if (USBController) {
            USBController->vtbl->SetEnabled(USBController, 1);
            USBController->vtbl->SetEnabledEhci(USBController, 1);
4728

4729 4730
            for (i = 0; i < def->nhostdevs; i++) {
                if (def->hostdevs[i]->mode ==
E
Eric Blake 已提交
4731
                    VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {
4732
                    if (def->hostdevs[i]->source.subsys.type ==
E
Eric Blake 已提交
4733
                        VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
4734

E
Eric Blake 已提交
4735
                        char *filtername           = NULL;
E
Eric Blake 已提交
4736 4737
                        PRUnichar *filternameUtf16 = NULL;
                        IUSBDeviceFilter *filter   = NULL;
4738

E
Eric Blake 已提交
4739 4740
                        /* Zero pad for nice alignment when fewer than 9999
                         * devices.
E
Eric Blake 已提交
4741
                         */
E
Eric Blake 已提交
4742 4743 4744 4745 4746 4747 4748 4749 4750
                        if (virAsprintf(&filtername, "filter%04d", i) < 0) {
                            virReportOOMError();
                        } else {
                            VBOX_UTF8_TO_UTF16(filtername, &filternameUtf16);
                            VIR_FREE(filtername);
                            USBController->vtbl->CreateDeviceFilter(USBController,
                                                                    filternameUtf16,
                                                                    &filter);
                        }
E
Eric Blake 已提交
4751 4752 4753 4754 4755 4756 4757 4758 4759 4760 4761 4762
                        VBOX_UTF16_FREE(filternameUtf16);

                        if (filter &&
                            (def->hostdevs[i]->source.subsys.u.usb.vendor ||
                             def->hostdevs[i]->source.subsys.u.usb.product)) {

                            PRUnichar *vendorIdUtf16  = NULL;
                            char vendorId[40]         = {0};
                            PRUnichar *productIdUtf16 = NULL;
                            char productId[40]        = {0};

                            if (def->hostdevs[i]->source.subsys.u.usb.vendor) {
E
Eric Blake 已提交
4763 4764
                                snprintf(vendorId, sizeof(vendorId), "%x",
                                         def->hostdevs[i]->source.subsys.u.usb.vendor);
E
Eric Blake 已提交
4765 4766 4767 4768 4769
                                VBOX_UTF8_TO_UTF16(vendorId, &vendorIdUtf16);
                                filter->vtbl->SetVendorId(filter, vendorIdUtf16);
                                VBOX_UTF16_FREE(vendorIdUtf16);
                            }
                            if (def->hostdevs[i]->source.subsys.u.usb.product) {
E
Eric Blake 已提交
4770 4771
                                snprintf(productId, sizeof(productId), "%x",
                                         def->hostdevs[i]->source.subsys.u.usb.product);
E
Eric Blake 已提交
4772 4773 4774 4775
                                VBOX_UTF8_TO_UTF16(productId, &productIdUtf16);
                                filter->vtbl->SetProductId(filter,
                                                           productIdUtf16);
                                VBOX_UTF16_FREE(productIdUtf16);
4776
                            }
E
Eric Blake 已提交
4777 4778 4779 4780 4781
                            filter->vtbl->SetActive(filter, 1);
                            USBController->vtbl->InsertDeviceFilter(USBController,
                                                                    i,
                                                                    filter);
                            VBOX_RELEASE(filter);
4782
                        }
E
Eric Blake 已提交
4783

4784 4785 4786
                    }
                }
            }
E
Eric Blake 已提交
4787 4788 4789 4790 4791 4792 4793 4794 4795
            VBOX_RELEASE(USBController);
        }
    }
}

static virDomainPtr vboxDomainDefineXML(virConnectPtr conn, const char *xml) {
    VBOX_OBJECT_CHECK(conn, virDomainPtr, NULL);
    IMachine       *machine     = NULL;
    IBIOSSettings  *bios        = NULL;
4796 4797
    vboxIID iid = VBOX_IID_INITIALIZER;
    vboxIID mchiid = VBOX_IID_INITIALIZER;
E
Eric Blake 已提交
4798 4799 4800 4801 4802 4803 4804 4805 4806 4807 4808 4809 4810
    virDomainDefPtr def         = NULL;
    PRUnichar *machineNameUtf16 = NULL;
#if VBOX_API_VERSION >= 3002
    PRBool override             = PR_FALSE;
#endif
    nsresult rc;

    if (!(def = virDomainDefParseString(data->caps, xml,
                                        VIR_DOMAIN_XML_INACTIVE))) {
        goto cleanup;
    }

    VBOX_UTF8_TO_UTF16(def->name, &machineNameUtf16);
4811
    vboxIIDFromUUID(&iid, def->uuid);
E
Eric Blake 已提交
4812 4813 4814 4815 4816
#if VBOX_API_VERSION < 3002
    rc = data->vboxObj->vtbl->CreateMachine(data->vboxObj,
                                            machineNameUtf16,
                                            NULL,
                                            NULL,
4817
                                            iid.value,
E
Eric Blake 已提交
4818
                                            &machine);
4819
#elif VBOX_API_VERSION < 4000 /* 3002 <= VBOX_API_VERSION < 4000 */
E
Eric Blake 已提交
4820 4821 4822 4823
    rc = data->vboxObj->vtbl->CreateMachine(data->vboxObj,
                                            machineNameUtf16,
                                            NULL,
                                            NULL,
4824
                                            iid.value,
E
Eric Blake 已提交
4825 4826
                                            override,
                                            &machine);
4827 4828 4829 4830 4831 4832 4833 4834 4835
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->CreateMachine(data->vboxObj,
                                            NULL,
                                            machineNameUtf16,
                                            NULL,
                                            iid.value,
                                            override,
                                            &machine);
#endif /* VBOX_API_VERSION >= 4000 */
E
Eric Blake 已提交
4836 4837 4838 4839 4840 4841 4842 4843
    VBOX_UTF16_FREE(machineNameUtf16);

    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not define a domain, rc=%08x"), (unsigned)rc);
        goto cleanup;
    }

4844 4845
    rc = machine->vtbl->SetMemorySize(machine,
                                      VIR_DIV_UP(def->mem.cur_balloon, 1024));
E
Eric Blake 已提交
4846 4847 4848 4849
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not set the memory size of the domain to: %lu Kb, "
                    "rc=%08x"),
4850
                  def->mem.cur_balloon, (unsigned)rc);
E
Eric Blake 已提交
4851 4852
    }

E
Eric Blake 已提交
4853 4854 4855 4856 4857
    if (def->vcpus != def->maxvcpus) {
        vboxError(VIR_ERR_CONFIG_UNSUPPORTED, "%s",
                    _("current vcpu count must equal maximum"));
    }
    rc = machine->vtbl->SetCPUCount(machine, def->maxvcpus);
E
Eric Blake 已提交
4858 4859
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_INTERNAL_ERROR,
E
Eric Blake 已提交
4860 4861
                  _("could not set the number of virtual CPUs to: %u, rc=%08x"),
                  def->maxvcpus, (unsigned)rc);
E
Eric Blake 已提交
4862 4863 4864 4865 4866 4867 4868 4869 4870 4871 4872 4873 4874 4875 4876 4877 4878 4879 4880 4881 4882 4883 4884 4885 4886 4887 4888 4889 4890 4891 4892 4893 4894 4895 4896 4897 4898 4899
    }

#if VBOX_API_VERSION < 3001
    rc = machine->vtbl->SetPAEEnabled(machine, (def->features) &
                                      (1 << VIR_DOMAIN_FEATURE_PAE));
#elif VBOX_API_VERSION == 3001
    rc = machine->vtbl->SetCpuProperty(machine, CpuPropertyType_PAE,
                                       (def->features) &
                                       (1 << VIR_DOMAIN_FEATURE_PAE));
#elif VBOX_API_VERSION >= 3002
    rc = machine->vtbl->SetCPUProperty(machine, CPUPropertyType_PAE,
                                       (def->features) &
                                       (1 << VIR_DOMAIN_FEATURE_PAE));
#endif
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not change PAE status to: %s, rc=%08x"),
                  ((def->features) & (1 << VIR_DOMAIN_FEATURE_PAE))
                  ? _("Enabled") : _("Disabled"), (unsigned)rc);
    }

    machine->vtbl->GetBIOSSettings(machine, &bios);
    if (bios) {
        rc = bios->vtbl->SetACPIEnabled(bios, (def->features) &
                                        (1 << VIR_DOMAIN_FEATURE_ACPI));
        if (NS_FAILED(rc)) {
            vboxError(VIR_ERR_INTERNAL_ERROR,
                      _("could not change ACPI status to: %s, rc=%08x"),
                      ((def->features) & (1 << VIR_DOMAIN_FEATURE_ACPI))
                      ? _("Enabled") : _("Disabled"), (unsigned)rc);
        }
        rc = bios->vtbl->SetIOAPICEnabled(bios, (def->features) &
                                          (1 << VIR_DOMAIN_FEATURE_APIC));
        if (NS_FAILED(rc)) {
            vboxError(VIR_ERR_INTERNAL_ERROR,
                      _("could not change APIC status to: %s, rc=%08x"),
                      ((def->features) & (1 << VIR_DOMAIN_FEATURE_APIC))
                      ? _("Enabled") : _("Disabled"), (unsigned)rc);
4900
        }
E
Eric Blake 已提交
4901 4902 4903 4904 4905 4906 4907 4908 4909 4910 4911 4912 4913 4914 4915
        VBOX_RELEASE(bios);
    }

    /* Register the machine before attaching other devices to it */
    rc = data->vboxObj->vtbl->RegisterMachine(data->vboxObj, machine);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not define a domain, rc=%08x"), (unsigned)rc);
        goto cleanup;
    }

    /* Get the uuid of the machine, currently it is immutable
     * object so open a session to it and get it back, so that
     * you can make changes to the machine setting
     */
4916
    machine->vtbl->GetId(machine, &mchiid.value);
4917
    VBOX_SESSION_OPEN(mchiid.value, machine);
E
Eric Blake 已提交
4918 4919 4920 4921 4922 4923 4924 4925 4926 4927 4928
    data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);

    vboxSetBootDeviceOrder(def, data, machine);
    vboxAttachDrives(def, data, machine);
    vboxAttachSound(def, machine);
    vboxAttachNetwork(def, data, machine);
    vboxAttachSerial(def, data, machine);
    vboxAttachParallel(def, data, machine);
    vboxAttachVideo(def, machine);
    vboxAttachDisplay(def, data, machine);
    vboxAttachUSB(def, data, machine);
4929

4930 4931 4932 4933
    /* Save the machine settings made till now and close the
     * session. also free up the mchiid variable used.
     */
    rc = machine->vtbl->SaveSettings(machine);
4934
    VBOX_SESSION_CLOSE();
4935
    vboxIIDUnalloc(&mchiid);
4936

4937 4938
    ret = virGetDomain(conn, def->name, def->uuid);
    VBOX_RELEASE(machine);
4939

4940
    vboxIIDUnalloc(&iid);
4941 4942
    virDomainDefFree(def);

4943
    return ret;
4944 4945

cleanup:
4946
    VBOX_RELEASE(machine);
4947
    vboxIIDUnalloc(&iid);
4948 4949 4950 4951
    virDomainDefFree(def);
    return NULL;
}

4952 4953 4954
static int
vboxDomainUndefine(virDomainPtr dom)
{
4955
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
4956
    IMachine *machine    = NULL;
4957
    vboxIID iid = VBOX_IID_INITIALIZER;
4958
    nsresult rc;
4959 4960 4961
#if VBOX_API_VERSION >= 4000
    vboxArray media = VBOX_ARRAY_INITIALIZER;
#endif
4962

4963
    vboxIIDFromUUID(&iid, dom->uuid);
4964

4965
#if VBOX_API_VERSION < 4000
4966 4967 4968
    /* Block for checking if HDD's are attched to VM.
     * considering just IDE bus for now. Also skipped
     * chanel=1 and device=0 (Secondary Master) as currenlty
4969 4970 4971 4972
     * it is allocated to CD/DVD Drive by default.
     *
     * Only do this for VirtualBox 3.x and before. Since
     * VirtualBox 4.0 the Unregister method can do this for use.
4973 4974 4975
     */
    {
        PRUnichar *hddcnameUtf16 = NULL;
4976

4977 4978 4979
        char *hddcname = strdup("IDE");
        VBOX_UTF8_TO_UTF16(hddcname, &hddcnameUtf16);
        VIR_FREE(hddcname);
4980

4981
        /* Open a Session for the machine */
4982
        rc = VBOX_SESSION_OPEN(iid.value, machine);
4983 4984 4985 4986
        if (NS_SUCCEEDED(rc)) {
            rc = data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);
            if (NS_SUCCEEDED(rc) && machine) {

4987
# if VBOX_API_VERSION < 3001
4988 4989 4990 4991
                /* Disconnect all the drives if present */
                machine->vtbl->DetachHardDisk(machine, hddcnameUtf16, 0, 0);
                machine->vtbl->DetachHardDisk(machine, hddcnameUtf16, 0, 1);
                machine->vtbl->DetachHardDisk(machine, hddcnameUtf16, 1, 1);
4992
# else  /* VBOX_API_VERSION >= 3001 */
4993 4994 4995
                /* get all the controller first, then the attachments and
                 * remove them all so that the machine can be undefined
                 */
4996
                vboxArray storageControllers = VBOX_ARRAY_INITIALIZER;
4997 4998
                int i = 0, j = 0;

4999 5000
                vboxArrayGet(&storageControllers, machine,
                             machine->vtbl->GetStorageControllers);
5001

5002 5003
                for (i = 0; i < storageControllers.count; i++) {
                    IStorageController *strCtl = storageControllers.items[i];
5004
                    PRUnichar *strCtlName = NULL;
5005
                    vboxArray mediumAttachments = VBOX_ARRAY_INITIALIZER;
5006 5007 5008 5009 5010

                    if (!strCtl)
                        continue;

                    strCtl->vtbl->GetName(strCtl, &strCtlName);
5011 5012 5013
                    vboxArrayGetWithPtrArg(&mediumAttachments, machine,
                                           machine->vtbl->GetMediumAttachmentsOfController,
                                           strCtlName);
5014

5015 5016
                    for (j = 0; j < mediumAttachments.count; j++) {
                        IMediumAttachment *medAtt = mediumAttachments.items[j];
5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032
                        PRInt32 port = ~0U;
                        PRInt32 device = ~0U;

                        if (!medAtt)
                            continue;

                        medAtt->vtbl->GetPort(medAtt, &port);
                        medAtt->vtbl->GetDevice(medAtt, &device);

                        if ((port != ~0U) && (device != ~0U)) {
                            machine->vtbl->DetachDevice(machine,
                                                        strCtlName,
                                                        port,
                                                        device);
                        }
                    }
5033

5034 5035
                    vboxArrayRelease(&storageControllers);

5036 5037
                    machine->vtbl->RemoveStorageController(machine, strCtlName);
                    VBOX_UTF16_FREE(strCtlName);
5038
                }
5039 5040

                vboxArrayRelease(&storageControllers);
5041
# endif /* VBOX_API_VERSION >= 3001 */
5042 5043

                machine->vtbl->SaveSettings(machine);
5044
            }
5045
            VBOX_SESSION_CLOSE();
5046
        }
5047 5048
        VBOX_UTF16_FREE(hddcnameUtf16);
    }
5049
#endif
5050

5051
#if VBOX_API_VERSION < 4000
5052
    rc = data->vboxObj->vtbl->UnregisterMachine(data->vboxObj, iid.value, &machine);
5053 5054 5055 5056 5057 5058 5059 5060 5061 5062 5063 5064 5065 5066 5067
#else /* VBOX_API_VERSION >= 4000 */
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
                  _("no domain with matching uuid"));
        return -1;
    }

    /* We're not interested in the array returned by the Unregister method,
     * but in the side effect of unregistering the virtual machine. In order
     * to call the Unregister method correctly we need to use the vboxArray
     * wrapper here. */
    rc = vboxArrayGetWithUintArg(&media, machine, machine->vtbl->Unregister,
                                 CleanupMode_DetachAllReturnNone);
#endif /* VBOX_API_VERSION >= 4000 */
5068
    DEBUGIID("UUID of machine being undefined", iid.value);
5069

5070 5071
    if (NS_SUCCEEDED(rc)) {
#if VBOX_API_VERSION < 4000
5072
        machine->vtbl->DeleteSettings(machine);
5073 5074 5075 5076 5077 5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090 5091 5092 5093 5094 5095
#else /* VBOX_API_VERSION >= 4000 */
        IProgress *progress = NULL;

        /* The IMachine Delete method takes an array of IMedium items to be
         * deleted along with the virtual machine. We just want to pass an
         * empty array. But instead of adding a full vboxArraySetWithReturn to
         * the glue layer (in order to handle the required signature of the
         * Delete method) we use a local solution here. */
# ifdef WIN32
        SAFEARRAY *safeArray = NULL;
        typedef HRESULT __stdcall (*IMachine_Delete)(IMachine *self,
                                                     SAFEARRAY **media,
                                                     IProgress **progress);

        ((IMachine_Delete)machine->vtbl->Delete)(machine, &safeArray, &progress);
# else
        machine->vtbl->Delete(machine, 0, NULL, &progress);
# endif
        if (progress != NULL) {
            progress->vtbl->WaitForCompletion(progress, -1);
            VBOX_RELEASE(progress);
        }
#endif /* VBOX_API_VERSION >= 4000 */
5096 5097
        ret = 0;
    } else {
5098 5099
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not delete the domain, rc=%08x"), (unsigned)rc);
5100 5101
    }

5102 5103 5104
#if VBOX_API_VERSION >= 4000
    vboxArrayUnalloc(&media);
#endif
5105
    vboxIIDUnalloc(&iid);
5106
    VBOX_RELEASE(machine);
5107 5108 5109 5110

    return ret;
}

5111 5112 5113
static int vboxDomainAttachDeviceImpl(virDomainPtr dom,
                                      const char *xml,
                                      int mediaChangeOnly ATTRIBUTE_UNUSED) {
5114
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
5115
    IMachine *machine    = NULL;
5116
    vboxIID iid = VBOX_IID_INITIALIZER;
5117 5118 5119
    PRUint32 state       = MachineState_Null;
    virDomainDefPtr def  = NULL;
    virDomainDeviceDefPtr dev  = NULL;
5120
    nsresult rc;
5121 5122

    if (VIR_ALLOC(def) < 0) {
5123
        virReportOOMError();
5124 5125 5126 5127 5128
        return ret;
    }

    def->os.type = strdup("hvm");

5129
    if (def->os.type == NULL) {
5130
        virReportOOMError();
5131 5132 5133
        goto cleanup;
    }

5134
    dev = virDomainDeviceDefParse(data->caps, def, xml,
5135 5136
                                  VIR_DOMAIN_XML_INACTIVE);
    if (dev == NULL) {
5137
        virReportOOMError();
5138 5139 5140
        goto cleanup;
    }

5141
    vboxIIDFromUUID(&iid, dom->uuid);
5142
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
5143
    if (NS_FAILED(rc)) {
5144
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
5145
                  _("no domain with matching uuid"));
5146 5147
        goto cleanup;
    }
5148

5149 5150
    if (machine) {
        machine->vtbl->GetState(machine, &state);
5151

5152 5153
        if ((state == MachineState_Running) ||
            (state == MachineState_Paused)) {
5154
            rc = VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
5155
        } else {
5156
            rc = VBOX_SESSION_OPEN(iid.value, machine);
5157 5158 5159 5160 5161 5162 5163
        }
        if (NS_SUCCEEDED(rc)) {
            rc = data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);
            if (NS_SUCCEEDED(rc) && machine) {
                if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
#if VBOX_API_VERSION < 3001
                    if (dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_CDROM) {
5164 5165
                        if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_FILE &&
                            dev->data.disk->src != NULL) {
5166 5167 5168 5169 5170 5171 5172 5173 5174
                            IDVDDrive *dvdDrive = NULL;
                            /* Currently CDROM/DVD Drive is always IDE
                             * Secondary Master so neglecting the following
                             * parameter dev->data.disk->bus
                             */
                            machine->vtbl->GetDVDDrive(machine, &dvdDrive);
                            if (dvdDrive) {
                                IDVDImage *dvdImage          = NULL;
                                PRUnichar *dvdfileUtf16      = NULL;
5175 5176
                                vboxIID dvduuid = VBOX_IID_INITIALIZER;
                                vboxIID dvdemptyuuid = VBOX_IID_INITIALIZER;
5177

5178
                                VBOX_UTF8_TO_UTF16(dev->data.disk->src, &dvdfileUtf16);
5179

5180 5181
                                data->vboxObj->vtbl->FindDVDImage(data->vboxObj, dvdfileUtf16, &dvdImage);
                                if (!dvdImage) {
5182
                                    data->vboxObj->vtbl->OpenDVDImage(data->vboxObj, dvdfileUtf16, dvdemptyuuid.value, &dvdImage);
5183 5184
                                }
                                if (dvdImage) {
5185
                                    rc = dvdImage->vtbl->imedium.GetId((IMedium *)dvdImage, &dvduuid.value);
5186
                                    if (NS_FAILED(rc)) {
5187 5188 5189
                                        vboxError(VIR_ERR_INTERNAL_ERROR,
                                                  _("can't get the uuid of the file to "
                                                    "be attached to cdrom: %s, rc=%08x"),
5190 5191 5192 5193
                                                  dev->data.disk->src, (unsigned)rc);
                                    } else {
                                        /* unmount the previous mounted image */
                                        dvdDrive->vtbl->Unmount(dvdDrive);
5194
                                        rc = dvdDrive->vtbl->MountImage(dvdDrive, dvduuid.value);
5195
                                        if (NS_FAILED(rc)) {
5196 5197
                                            vboxError(VIR_ERR_INTERNAL_ERROR,
                                                      _("could not attach the file to cdrom: %s, rc=%08x"),
5198 5199
                                                      dev->data.disk->src, (unsigned)rc);
                                        } else {
5200
                                            ret = 0;
5201
                                            DEBUGIID("CD/DVD Image UUID:", dvduuid.value);
5202 5203
                                        }
                                    }
5204 5205

                                    VBOX_MEDIUM_RELEASE(dvdImage);
5206
                                }
5207
                                vboxIIDUnalloc(&dvduuid);
5208 5209
                                VBOX_UTF16_FREE(dvdfileUtf16);
                                VBOX_RELEASE(dvdDrive);
5210
                            }
5211 5212 5213
                        } else if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
                        }
                    } else if (dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_FLOPPY) {
5214 5215
                        if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_FILE &&
                            dev->data.disk->src != NULL) {
5216 5217 5218 5219 5220 5221 5222
                            IFloppyDrive *floppyDrive;
                            machine->vtbl->GetFloppyDrive(machine, &floppyDrive);
                            if (floppyDrive) {
                                rc = floppyDrive->vtbl->SetEnabled(floppyDrive, 1);
                                if (NS_SUCCEEDED(rc)) {
                                    IFloppyImage *floppyImage   = NULL;
                                    PRUnichar *fdfileUtf16      = NULL;
5223 5224
                                    vboxIID fduuid = VBOX_IID_INITIALIZER;
                                    vboxIID fdemptyuuid = VBOX_IID_INITIALIZER;
5225 5226 5227 5228 5229 5230 5231 5232
                                    VBOX_UTF8_TO_UTF16(dev->data.disk->src, &fdfileUtf16);
                                    rc = data->vboxObj->vtbl->FindFloppyImage(data->vboxObj,
                                                                              fdfileUtf16,
                                                                              &floppyImage);

                                    if (!floppyImage) {
                                        data->vboxObj->vtbl->OpenFloppyImage(data->vboxObj,
                                                                             fdfileUtf16,
5233
                                                                             fdemptyuuid.value,
5234 5235
                                                                             &floppyImage);
                                    }
5236

5237
                                    if (floppyImage) {
5238
                                        rc = floppyImage->vtbl->imedium.GetId((IMedium *)floppyImage, &fduuid.value);
5239
                                        if (NS_FAILED(rc)) {
5240 5241 5242
                                            vboxError(VIR_ERR_INTERNAL_ERROR,
                                                      _("can't get the uuid of the file to be "
                                                        "attached to floppy drive: %s, rc=%08x"),
5243 5244
                                                      dev->data.disk->src, (unsigned)rc);
                                        } else {
5245
                                            rc = floppyDrive->vtbl->MountImage(floppyDrive, fduuid.value);
5246
                                            if (NS_FAILED(rc)) {
5247 5248
                                                vboxError(VIR_ERR_INTERNAL_ERROR,
                                                          _("could not attach the file to floppy drive: %s, rc=%08x"),
5249 5250
                                                          dev->data.disk->src, (unsigned)rc);
                                            } else {
5251
                                                ret = 0;
5252
                                                DEBUGIID("attached floppy, UUID:", fduuid.value);
5253 5254
                                            }
                                        }
5255
                                        VBOX_MEDIUM_RELEASE(floppyImage);
5256
                                    }
5257
                                    vboxIIDUnalloc(&fduuid);
5258
                                    VBOX_UTF16_FREE(fdfileUtf16);
5259
                                }
5260
                                VBOX_RELEASE(floppyDrive);
5261
                            }
5262
                        } else if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
5263
                        }
5264 5265 5266 5267 5268 5269 5270
                    }
#else  /* VBOX_API_VERSION >= 3001 */
#endif /* VBOX_API_VERSION >= 3001 */
                } else if (dev->type == VIR_DOMAIN_DEVICE_NET) {
                } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
                    if (dev->data.hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {
                        if (dev->data.hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
5271 5272 5273
                        }
                    }
                }
5274 5275
                machine->vtbl->SaveSettings(machine);
                VBOX_RELEASE(machine);
5276
            }
5277
            VBOX_SESSION_CLOSE();
5278 5279 5280 5281
        }
    }

cleanup:
5282
    vboxIIDUnalloc(&iid);
5283 5284 5285 5286 5287
    virDomainDefFree(def);
    virDomainDeviceDefFree(dev);
    return ret;
}

5288 5289 5290 5291
static int vboxDomainAttachDevice(virDomainPtr dom, const char *xml) {
    return vboxDomainAttachDeviceImpl(dom, xml, 0);
}

5292 5293
static int vboxDomainAttachDeviceFlags(virDomainPtr dom, const char *xml,
                                       unsigned int flags) {
5294
    if (flags & VIR_DOMAIN_AFFECT_CONFIG) {
5295
        vboxError(VIR_ERR_OPERATION_INVALID, "%s",
5296 5297 5298 5299
                  _("cannot modify the persistent configuration of a domain"));
        return -1;
    }

5300 5301 5302 5303 5304
    return vboxDomainAttachDeviceImpl(dom, xml, 0);
}

static int vboxDomainUpdateDeviceFlags(virDomainPtr dom, const char *xml,
                                       unsigned int flags) {
5305 5306 5307
    virCheckFlags(VIR_DOMAIN_AFFECT_CURRENT |
                  VIR_DOMAIN_AFFECT_LIVE |
                  VIR_DOMAIN_AFFECT_CONFIG, -1);
5308

5309
    if (flags & VIR_DOMAIN_AFFECT_CONFIG) {
5310
        vboxError(VIR_ERR_OPERATION_INVALID, "%s",
5311 5312 5313 5314 5315
                  _("cannot modify the persistent configuration of a domain"));
        return -1;
    }

    return vboxDomainAttachDeviceImpl(dom, xml, 1);
5316 5317
}

5318
static int vboxDomainDetachDevice(virDomainPtr dom, const char *xml) {
5319
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
5320
    IMachine *machine    = NULL;
5321
    vboxIID iid = VBOX_IID_INITIALIZER;
5322 5323 5324
    PRUint32 state       = MachineState_Null;
    virDomainDefPtr def  = NULL;
    virDomainDeviceDefPtr dev  = NULL;
5325
    nsresult rc;
5326 5327

    if (VIR_ALLOC(def) < 0) {
5328
        virReportOOMError();
5329 5330 5331 5332 5333
        return ret;
    }

    def->os.type = strdup("hvm");

5334
    if (def->os.type == NULL) {
5335
        virReportOOMError();
5336 5337 5338
        goto cleanup;
    }

5339
    dev = virDomainDeviceDefParse(data->caps, def, xml,
5340 5341
                                  VIR_DOMAIN_XML_INACTIVE);
    if (dev == NULL) {
5342
        virReportOOMError();
5343 5344 5345
        goto cleanup;
    }

5346
    vboxIIDFromUUID(&iid, dom->uuid);
5347
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
5348
    if (NS_FAILED(rc)) {
5349
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
5350
                  _("no domain with matching uuid"));
5351 5352
        goto cleanup;
    }
5353

5354 5355
    if (machine) {
        machine->vtbl->GetState(machine, &state);
5356

5357 5358
        if ((state == MachineState_Running) ||
            (state == MachineState_Paused)) {
5359
            rc = VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
5360
        } else {
5361
            rc = VBOX_SESSION_OPEN(iid.value, machine);
5362
        }
5363

5364 5365 5366 5367 5368 5369 5370 5371 5372 5373 5374 5375 5376 5377 5378 5379
        if (NS_SUCCEEDED(rc)) {
            rc = data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);
            if (NS_SUCCEEDED(rc) && machine) {
                if (dev->type == VIR_DOMAIN_DEVICE_DISK) {
#if VBOX_API_VERSION < 3001
                    if (dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_CDROM) {
                        if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_FILE) {
                            IDVDDrive *dvdDrive = NULL;
                            /* Currently CDROM/DVD Drive is always IDE
                             * Secondary Master so neglecting the following
                             * parameter dev->data.disk->bus
                             */
                            machine->vtbl->GetDVDDrive(machine, &dvdDrive);
                            if (dvdDrive) {
                                rc = dvdDrive->vtbl->Unmount(dvdDrive);
                                if (NS_FAILED(rc)) {
5380 5381
                                    vboxError(VIR_ERR_INTERNAL_ERROR,
                                              _("could not de-attach the mounted ISO, rc=%08x"),
5382 5383 5384 5385 5386 5387 5388 5389 5390 5391 5392 5393 5394 5395 5396 5397 5398 5399
                                              (unsigned)rc);
                                } else {
                                    ret = 0;
                                }
                                VBOX_RELEASE(dvdDrive);
                            }
                        } else if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
                        }
                    } else if (dev->data.disk->device == VIR_DOMAIN_DISK_DEVICE_FLOPPY) {
                        if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_FILE) {
                            IFloppyDrive *floppyDrive;
                            machine->vtbl->GetFloppyDrive(machine, &floppyDrive);
                            if (floppyDrive) {
                                PRBool enabled = PR_FALSE;

                                floppyDrive->vtbl->GetEnabled(floppyDrive, &enabled);
                                if (enabled) {
                                    rc = floppyDrive->vtbl->Unmount(floppyDrive);
5400
                                    if (NS_FAILED(rc)) {
5401 5402 5403
                                        vboxError(VIR_ERR_INTERNAL_ERROR,
                                                  _("could not attach the file "
                                                    "to floppy drive, rc=%08x"),
5404 5405 5406 5407
                                                  (unsigned)rc);
                                    } else {
                                        ret = 0;
                                    }
5408 5409 5410 5411 5412
                                } else {
                                    /* If you are here means floppy drive is already unmounted
                                     * so don't flag error, just say everything is fine and quit
                                     */
                                    ret = 0;
5413
                                }
5414
                                VBOX_RELEASE(floppyDrive);
5415
                            }
5416
                        } else if (dev->data.disk->type == VIR_DOMAIN_DISK_TYPE_BLOCK) {
5417
                        }
5418 5419 5420 5421 5422 5423 5424
                    }
#else  /* VBOX_API_VERSION >= 3001 */
#endif /* VBOX_API_VERSION >= 3001 */
                } else if (dev->type == VIR_DOMAIN_DEVICE_NET) {
                } else if (dev->type == VIR_DOMAIN_DEVICE_HOSTDEV) {
                    if (dev->data.hostdev->mode == VIR_DOMAIN_HOSTDEV_MODE_SUBSYS) {
                        if (dev->data.hostdev->source.subsys.type == VIR_DOMAIN_HOSTDEV_SUBSYS_TYPE_USB) {
5425 5426 5427
                        }
                    }
                }
5428 5429
                machine->vtbl->SaveSettings(machine);
                VBOX_RELEASE(machine);
5430
            }
5431
            VBOX_SESSION_CLOSE();
5432 5433 5434 5435
        }
    }

cleanup:
5436
    vboxIIDUnalloc(&iid);
5437 5438 5439 5440 5441
    virDomainDefFree(def);
    virDomainDeviceDefFree(dev);
    return ret;
}

5442 5443
static int vboxDomainDetachDeviceFlags(virDomainPtr dom, const char *xml,
                                       unsigned int flags) {
5444
    if (flags & VIR_DOMAIN_AFFECT_CONFIG) {
5445
        vboxError(VIR_ERR_OPERATION_INVALID, "%s",
5446 5447 5448 5449 5450 5451 5452
                  _("cannot modify the persistent configuration of a domain"));
        return -1;
    }

    return vboxDomainDetachDevice(dom, xml);
}

J
Jiri Denemark 已提交
5453 5454 5455 5456 5457
static int
vboxDomainSnapshotGetAll(virDomainPtr dom,
                         IMachine *machine,
                         ISnapshot ***snapshots)
{
5458
    vboxIID empty = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5459 5460 5461 5462 5463 5464 5465 5466
    ISnapshot **list = NULL;
    PRUint32 count;
    nsresult rc;
    unsigned int next;
    unsigned int top;

    rc = machine->vtbl->GetSnapshotCount(machine, &count);
    if (NS_FAILED(rc)) {
5467
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5468 5469 5470 5471 5472 5473 5474 5475 5476 5477 5478 5479 5480
                  _("could not get snapshot count for domain %s"),
                  dom->name);
        goto error;
    }

    if (count == 0)
        goto out;

    if (VIR_ALLOC_N(list, count) < 0) {
        virReportOOMError();
        goto error;
    }

5481
#if VBOX_API_VERSION < 4000
5482
    rc = machine->vtbl->GetSnapshot(machine, empty.value, list);
5483 5484 5485
#else /* VBOX_API_VERSION >= 4000 */
    rc = machine->vtbl->FindSnapshot(machine, empty.value, list);
#endif /* VBOX_API_VERSION >= 4000 */
J
Jiri Denemark 已提交
5486
    if (NS_FAILED(rc) || !list[0]) {
5487
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5488 5489 5490 5491 5492 5493 5494 5495
                  _("could not get root snapshot for domain %s"),
                  dom->name);
        goto error;
    }

    /* BFS walk through snapshot tree */
    top = 1;
    for (next = 0; next < count; next++) {
5496
        vboxArray children = VBOX_ARRAY_INITIALIZER;
J
Jiri Denemark 已提交
5497 5498 5499
        unsigned int i;

        if (!list[next]) {
5500
            vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5501 5502 5503 5504
                      _("unexpected number of snapshots < %u"), count);
            goto error;
        }

5505 5506
        rc = vboxArrayGet(&children, list[next],
                               list[next]->vtbl->GetChildren);
J
Jiri Denemark 已提交
5507
        if (NS_FAILED(rc)) {
5508
            vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5509 5510 5511
                      "%s", _("could not get children snapshots"));
            goto error;
        }
5512 5513 5514
        for (i = 0; i < children.count; i++) {
            ISnapshot *child = children.items[i];
            if (!child)
J
Jiri Denemark 已提交
5515 5516
                continue;
            if (top == count) {
5517
                vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5518
                          _("unexpected number of snapshots > %u"), count);
5519
                vboxArrayRelease(&children);
J
Jiri Denemark 已提交
5520 5521
                goto error;
            }
5522 5523
            VBOX_ADDREF(child);
            list[top++] = child;
J
Jiri Denemark 已提交
5524
        }
5525
        vboxArrayRelease(&children);
J
Jiri Denemark 已提交
5526 5527 5528 5529 5530 5531 5532 5533 5534 5535 5536 5537 5538 5539 5540 5541 5542 5543 5544 5545 5546 5547 5548 5549 5550 5551 5552 5553 5554 5555 5556 5557 5558 5559 5560 5561 5562
    }

out:
    *snapshots = list;
    return count;

error:
    if (list) {
        for (next = 0; next < count; next++)
            VBOX_RELEASE(list[next]);
    }
    VIR_FREE(list);

    return -1;
}

static ISnapshot *
vboxDomainSnapshotGet(vboxGlobalData *data,
                      virDomainPtr dom,
                      IMachine *machine,
                      const char *name)
{
    ISnapshot **snapshots = NULL;
    ISnapshot *snapshot = NULL;
    nsresult rc;
    int count = 0;
    int i;

    if ((count = vboxDomainSnapshotGetAll(dom, machine, &snapshots)) < 0)
        goto cleanup;

    for (i = 0; i < count; i++) {
        PRUnichar *nameUtf16;
        char *nameUtf8;

        rc = snapshots[i]->vtbl->GetName(snapshots[i], &nameUtf16);
        if (NS_FAILED(rc) || !nameUtf16) {
5563
            vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5564 5565 5566 5567 5568 5569 5570 5571 5572 5573 5574 5575 5576 5577
                      "%s", _("could not get snapshot name"));
            goto cleanup;
        }
        VBOX_UTF16_TO_UTF8(nameUtf16, &nameUtf8);
        VBOX_UTF16_FREE(nameUtf16);
        if (STREQ(name, nameUtf8))
            snapshot = snapshots[i];
        VBOX_UTF8_FREE(nameUtf8);

        if (snapshot)
            break;
    }

    if (!snapshot) {
5578
        vboxError(VIR_ERR_OPERATION_INVALID,
J
Jiri Denemark 已提交
5579 5580 5581 5582 5583 5584 5585 5586 5587 5588 5589 5590 5591 5592 5593 5594 5595 5596 5597
                  _("domain %s has no snapshots with name %s"),
                  dom->name, name);
        goto cleanup;
    }

cleanup:
    if (count > 0) {
        for (i = 0; i < count; i++) {
            if (snapshots[i] != snapshot)
                VBOX_RELEASE(snapshots[i]);
        }
    }
    VIR_FREE(snapshots);
    return snapshot;
}

static virDomainSnapshotPtr
vboxDomainSnapshotCreateXML(virDomainPtr dom,
                            const char *xmlDesc,
5598
                            unsigned int flags)
J
Jiri Denemark 已提交
5599 5600 5601
{
    VBOX_OBJECT_CHECK(dom->conn, virDomainSnapshotPtr, NULL);
    virDomainSnapshotDefPtr def = NULL;
5602
    vboxIID domiid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5603 5604 5605 5606 5607 5608 5609 5610 5611 5612 5613 5614 5615 5616
    IMachine *machine = NULL;
    IConsole *console = NULL;
    IProgress *progress = NULL;
    ISnapshot *snapshot = NULL;
    PRUnichar *name = NULL;
    PRUnichar *description = NULL;
    PRUint32 state;
    nsresult rc;
#if VBOX_API_VERSION == 2002
    nsresult result;
#else
    PRInt32 result;
#endif

5617 5618
    virCheckFlags(0, NULL);

J
Jiri Denemark 已提交
5619 5620 5621
    if (!(def = virDomainSnapshotDefParseString(xmlDesc, 1)))
        goto cleanup;

5622
    vboxIIDFromUUID(&domiid, dom->uuid);
5623
    rc = VBOX_OBJECT_GET_MACHINE(domiid.value, &machine);
5624
    if (NS_FAILED(rc)) {
5625
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
5626 5627 5628 5629 5630 5631
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    rc = machine->vtbl->GetState(machine, &state);
    if (NS_FAILED(rc)) {
5632
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
5633 5634 5635 5636 5637 5638
                  _("could not get domain state"));
        goto cleanup;
    }

    if ((state >= MachineState_FirstOnline)
        && (state <= MachineState_LastOnline)) {
5639
        rc = VBOX_SESSION_OPEN_EXISTING(domiid.value, machine);
J
Jiri Denemark 已提交
5640
    } else {
5641
        rc = VBOX_SESSION_OPEN(domiid.value, machine);
J
Jiri Denemark 已提交
5642
    }
5643

J
Jiri Denemark 已提交
5644 5645 5646
    if (NS_SUCCEEDED(rc))
        rc = data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
    if (NS_FAILED(rc)) {
5647
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5648 5649 5650 5651 5652 5653 5654 5655 5656 5657 5658 5659 5660 5661 5662 5663 5664 5665 5666 5667 5668
                  _("could not open VirtualBox session with domain %s"),
                  dom->name);
        goto cleanup;
    }

    VBOX_UTF8_TO_UTF16(def->name, &name);
    if (!name) {
        virReportOOMError();
        goto cleanup;
    }

    if (def->description) {
        VBOX_UTF8_TO_UTF16(def->description, &description);
        if (!description) {
            virReportOOMError();
            goto cleanup;
        }
    }

    rc = console->vtbl->TakeSnapshot(console, name, description, &progress);
    if (NS_FAILED(rc) || !progress) {
5669
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5670 5671 5672 5673 5674 5675 5676
                  _("could not take snapshot of domain %s"), dom->name);
        goto cleanup;
    }

    progress->vtbl->WaitForCompletion(progress, -1);
    progress->vtbl->GetResultCode(progress, &result);
    if (NS_FAILED(result)) {
5677
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5678 5679 5680 5681 5682 5683
                  _("could not take snapshot of domain %s"), dom->name);
        goto cleanup;
    }

    rc = machine->vtbl->GetCurrentSnapshot(machine, &snapshot);
    if (NS_FAILED(rc)) {
5684
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5685 5686 5687 5688 5689 5690 5691 5692 5693 5694 5695 5696
                  _("could not get current snapshot of domain %s"),
                  dom->name);
        goto cleanup;
    }

    ret = virGetDomainSnapshot(dom, def->name);

cleanup:
    VBOX_RELEASE(progress);
    VBOX_UTF16_FREE(description);
    VBOX_UTF16_FREE(name);
    VBOX_RELEASE(console);
5697
    VBOX_SESSION_CLOSE();
J
Jiri Denemark 已提交
5698
    VBOX_RELEASE(machine);
5699
    vboxIIDUnalloc(&domiid);
J
Jiri Denemark 已提交
5700 5701 5702 5703 5704
    virDomainSnapshotDefFree(def);
    return ret;
}

static char *
5705 5706
vboxDomainSnapshotGetXMLDesc(virDomainSnapshotPtr snapshot,
                             unsigned int flags)
J
Jiri Denemark 已提交
5707 5708 5709
{
    virDomainPtr dom = snapshot->domain;
    VBOX_OBJECT_CHECK(dom->conn, char *, NULL);
5710
    vboxIID domiid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5711 5712 5713 5714 5715 5716 5717 5718 5719 5720 5721
    IMachine *machine = NULL;
    ISnapshot *snap = NULL;
    ISnapshot *parent = NULL;
    nsresult rc;
    virDomainSnapshotDefPtr def = NULL;
    PRUnichar *str16;
    char *str8;
    PRInt64 timestamp;
    PRBool online = PR_FALSE;
    char uuidstr[VIR_UUID_STRING_BUFLEN];

5722 5723
    virCheckFlags(0, NULL);

5724
    vboxIIDFromUUID(&domiid, dom->uuid);
5725
    rc = VBOX_OBJECT_GET_MACHINE(domiid.value, &machine);
J
Jiri Denemark 已提交
5726
    if (NS_FAILED(rc)) {
5727
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
5728 5729 5730 5731 5732 5733 5734 5735 5736 5737 5738 5739 5740
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    if (!(snap = vboxDomainSnapshotGet(data, dom, machine, snapshot->name)))
        goto cleanup;

    if (VIR_ALLOC(def) < 0
        || !(def->name = strdup(snapshot->name)))
        goto no_memory;

    rc = snap->vtbl->GetDescription(snap, &str16);
    if (NS_FAILED(rc)) {
5741
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5742 5743 5744 5745 5746 5747 5748 5749 5750 5751 5752 5753 5754 5755 5756
                  _("could not get description of snapshot %s"),
                  snapshot->name);
        goto cleanup;
    }
    if (str16) {
        VBOX_UTF16_TO_UTF8(str16, &str8);
        VBOX_UTF16_FREE(str16);
        def->description = strdup(str8);
        VBOX_UTF8_FREE(str8);
        if (!def->description)
            goto no_memory;
    }

    rc = snap->vtbl->GetTimeStamp(snap, &timestamp);
    if (NS_FAILED(rc)) {
5757
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5758 5759 5760 5761 5762 5763 5764 5765 5766
                  _("could not get creation time of snapshot %s"),
                  snapshot->name);
        goto cleanup;
    }
    /* timestamp is in milliseconds while creationTime in seconds */
    def->creationTime = timestamp / 1000;

    rc = snap->vtbl->GetParent(snap, &parent);
    if (NS_FAILED(rc)) {
5767
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5768 5769 5770 5771 5772 5773 5774
                  _("could not get parent of snapshot %s"),
                  snapshot->name);
        goto cleanup;
    }
    if (parent) {
        rc = parent->vtbl->GetName(parent, &str16);
        if (NS_FAILED(rc) || !str16) {
5775
            vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5776 5777 5778 5779 5780 5781 5782 5783 5784 5785 5786 5787 5788 5789
                      _("could not get name of parent of snapshot %s"),
                      snapshot->name);
            goto cleanup;
        }
        VBOX_UTF16_TO_UTF8(str16, &str8);
        VBOX_UTF16_FREE(str16);
        def->parent = strdup(str8);
        VBOX_UTF8_FREE(str8);
        if (!def->parent)
            goto no_memory;
    }

    rc = snap->vtbl->GetOnline(snap, &online);
    if (NS_FAILED(rc)) {
5790
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5791 5792 5793 5794 5795 5796 5797 5798 5799 5800 5801 5802 5803 5804 5805 5806 5807
                  _("could not get online state of snapshot %s"),
                  snapshot->name);
        goto cleanup;
    }
    if (online)
        def->state = VIR_DOMAIN_RUNNING;
    else
        def->state = VIR_DOMAIN_SHUTOFF;

    virUUIDFormat(dom->uuid, uuidstr);
    ret = virDomainSnapshotDefFormat(uuidstr, def, 0);

cleanup:
    virDomainSnapshotDefFree(def);
    VBOX_RELEASE(parent);
    VBOX_RELEASE(snap);
    VBOX_RELEASE(machine);
5808
    vboxIIDUnalloc(&domiid);
J
Jiri Denemark 已提交
5809 5810 5811 5812 5813 5814 5815 5816 5817
    return ret;

no_memory:
    virReportOOMError();
    goto cleanup;
}

static int
vboxDomainSnapshotNum(virDomainPtr dom,
5818
                      unsigned int flags)
J
Jiri Denemark 已提交
5819 5820
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
5821
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5822 5823 5824 5825
    IMachine *machine = NULL;
    nsresult rc;
    PRUint32 snapshotCount;

5826 5827
    virCheckFlags(0, -1);

5828
    vboxIIDFromUUID(&iid, dom->uuid);
5829
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
J
Jiri Denemark 已提交
5830
    if (NS_FAILED(rc)) {
5831
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
5832 5833 5834 5835 5836 5837
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    rc = machine->vtbl->GetSnapshotCount(machine, &snapshotCount);
    if (NS_FAILED(rc)) {
5838
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5839 5840 5841 5842 5843 5844 5845 5846 5847
                  _("could not get snapshot count for domain %s"),
                  dom->name);
        goto cleanup;
    }

    ret = snapshotCount;

cleanup:
    VBOX_RELEASE(machine);
5848
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
5849 5850 5851 5852 5853 5854 5855
    return ret;
}

static int
vboxDomainSnapshotListNames(virDomainPtr dom,
                            char **names,
                            int nameslen,
5856
                            unsigned int flags)
J
Jiri Denemark 已提交
5857 5858
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
5859
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5860 5861 5862 5863 5864 5865
    IMachine *machine = NULL;
    nsresult rc;
    ISnapshot **snapshots = NULL;
    int count = 0;
    int i;

5866 5867
    virCheckFlags(0, -1);

5868
    vboxIIDFromUUID(&iid, dom->uuid);
5869
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
J
Jiri Denemark 已提交
5870
    if (NS_FAILED(rc)) {
5871
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
5872 5873 5874 5875 5876 5877 5878 5879 5880 5881 5882 5883 5884 5885 5886 5887
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    if ((count = vboxDomainSnapshotGetAll(dom, machine, &snapshots)) < 0)
        goto cleanup;

    for (i = 0; i < nameslen; i++) {
        PRUnichar *nameUtf16;
        char *name;

        if (i >= count)
            break;

        rc = snapshots[i]->vtbl->GetName(snapshots[i], &nameUtf16);
        if (NS_FAILED(rc) || !nameUtf16) {
5888
            vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
5889 5890 5891 5892 5893 5894 5895 5896 5897 5898 5899 5900 5901 5902 5903 5904 5905 5906 5907 5908 5909 5910 5911 5912 5913
                      "%s", _("could not get snapshot name"));
            goto cleanup;
        }
        VBOX_UTF16_TO_UTF8(nameUtf16, &name);
        VBOX_UTF16_FREE(nameUtf16);
        names[i] = strdup(name);
        VBOX_UTF8_FREE(name);
        if (!names[i]) {
            virReportOOMError();
            goto cleanup;
        }
    }

    if (count <= nameslen)
        ret = count;
    else
        ret = nameslen;

cleanup:
    if (count > 0) {
        for (i = 0; i < count; i++)
            VBOX_RELEASE(snapshots[i]);
    }
    VIR_FREE(snapshots);
    VBOX_RELEASE(machine);
5914
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
5915 5916 5917 5918 5919 5920
    return ret;
}

static virDomainSnapshotPtr
vboxDomainSnapshotLookupByName(virDomainPtr dom,
                               const char *name,
5921
                               unsigned int flags)
J
Jiri Denemark 已提交
5922 5923
{
    VBOX_OBJECT_CHECK(dom->conn, virDomainSnapshotPtr, NULL);
5924
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5925 5926 5927 5928
    IMachine *machine = NULL;
    ISnapshot *snapshot = NULL;
    nsresult rc;

5929 5930
    virCheckFlags(0, NULL);

5931
    vboxIIDFromUUID(&iid, dom->uuid);
5932
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
J
Jiri Denemark 已提交
5933
    if (NS_FAILED(rc)) {
5934
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
5935 5936 5937 5938 5939 5940 5941 5942 5943 5944 5945 5946
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    if (!(snapshot = vboxDomainSnapshotGet(data, dom, machine, name)))
        goto cleanup;

    ret = virGetDomainSnapshot(dom, name);

cleanup:
    VBOX_RELEASE(snapshot);
    VBOX_RELEASE(machine);
5947
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
5948 5949 5950 5951 5952
    return ret;
}

static int
vboxDomainHasCurrentSnapshot(virDomainPtr dom,
5953
                             unsigned int flags)
J
Jiri Denemark 已提交
5954 5955
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
5956
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5957 5958 5959 5960
    IMachine *machine = NULL;
    ISnapshot *snapshot = NULL;
    nsresult rc;

5961 5962
    virCheckFlags(0, -1);

5963
    vboxIIDFromUUID(&iid, dom->uuid);
5964
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
5965
    if (NS_FAILED(rc)) {
5966
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
5967 5968 5969 5970 5971 5972
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    rc = machine->vtbl->GetCurrentSnapshot(machine, &snapshot);
    if (NS_FAILED(rc)) {
5973
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
5974 5975 5976 5977 5978 5979 5980 5981 5982 5983 5984
                  _("could not get current snapshot"));
        goto cleanup;
    }

    if (snapshot)
        ret = 1;
    else
        ret = 0;

cleanup:
    VBOX_RELEASE(machine);
5985
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
5986 5987 5988 5989 5990
    return ret;
}

static virDomainSnapshotPtr
vboxDomainSnapshotCurrent(virDomainPtr dom,
5991
                          unsigned int flags)
J
Jiri Denemark 已提交
5992 5993
{
    VBOX_OBJECT_CHECK(dom->conn, virDomainSnapshotPtr, NULL);
5994
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
5995 5996 5997 5998 5999 6000
    IMachine *machine = NULL;
    ISnapshot *snapshot = NULL;
    PRUnichar *nameUtf16 = NULL;
    char *name = NULL;
    nsresult rc;

6001 6002
    virCheckFlags(0, NULL);

6003
    vboxIIDFromUUID(&iid, dom->uuid);
6004
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
6005
    if (NS_FAILED(rc)) {
6006
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
6007 6008 6009 6010 6011 6012
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    rc = machine->vtbl->GetCurrentSnapshot(machine, &snapshot);
    if (NS_FAILED(rc)) {
6013
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6014 6015 6016 6017 6018
                  _("could not get current snapshot"));
        goto cleanup;
    }

    if (!snapshot) {
6019
        vboxError(VIR_ERR_OPERATION_INVALID, "%s",
J
Jiri Denemark 已提交
6020 6021 6022 6023 6024 6025
                  _("domain has no snapshots"));
        goto cleanup;
    }

    rc = snapshot->vtbl->GetName(snapshot, &nameUtf16);
    if (NS_FAILED(rc) || !nameUtf16) {
6026
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6027 6028 6029 6030 6031 6032 6033 6034 6035 6036 6037 6038 6039 6040 6041 6042 6043
                  _("could not get current snapshot name"));
        goto cleanup;
    }

    VBOX_UTF16_TO_UTF8(nameUtf16, &name);
    if (!name) {
        virReportOOMError();
        goto cleanup;
    }

    ret = virGetDomainSnapshot(dom, name);

cleanup:
    VBOX_UTF8_FREE(name);
    VBOX_UTF16_FREE(nameUtf16);
    VBOX_RELEASE(snapshot);
    VBOX_RELEASE(machine);
6044
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
6045 6046 6047 6048 6049 6050 6051 6052 6053 6054
    return ret;
}

#if VBOX_API_VERSION < 3001
static int
vboxDomainSnapshotRestore(virDomainPtr dom,
                          IMachine *machine,
                          ISnapshot *snapshot)
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
6055
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
6056 6057
    nsresult rc;

6058 6059
    rc = snapshot->vtbl->GetId(snapshot, &iid.value);
    if (NS_FAILED(rc)) {
6060
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6061 6062 6063 6064
                  _("could not get snapshot UUID"));
        goto cleanup;
    }

6065
    rc = machine->vtbl->SetCurrentSnapshot(machine, iid.value);
J
Jiri Denemark 已提交
6066
    if (NS_FAILED(rc)) {
6067
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6068 6069 6070 6071 6072 6073 6074
                  _("could not restore snapshot for domain %s"), dom->name);
        goto cleanup;
    }

    ret = 0;

cleanup:
6075
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
6076 6077 6078 6079 6080 6081 6082 6083 6084 6085 6086 6087 6088 6089
    return ret;
}
#else
static int
vboxDomainSnapshotRestore(virDomainPtr dom,
                          IMachine *machine,
                          ISnapshot *snapshot)
{
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
    IConsole *console = NULL;
    IProgress *progress = NULL;
    PRUint32 state;
    nsresult rc;
    PRInt32 result;
6090
    vboxIID domiid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
6091

6092 6093
    rc = machine->vtbl->GetId(machine, &domiid.value);
    if (NS_FAILED(rc)) {
6094
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6095 6096 6097 6098 6099 6100
                  _("could not get domain UUID"));
        goto cleanup;
    }

    rc = machine->vtbl->GetState(machine, &state);
    if (NS_FAILED(rc)) {
6101
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6102 6103 6104 6105 6106 6107
                  _("could not get domain state"));
        goto cleanup;
    }

    if (state >= MachineState_FirstOnline
        && state <= MachineState_LastOnline) {
6108
        vboxError(VIR_ERR_OPERATION_INVALID,
J
Jiri Denemark 已提交
6109 6110 6111 6112
                  _("domain %s is already running"), dom->name);
        goto cleanup;
    }

6113
    rc = VBOX_SESSION_OPEN(domiid.value, machine);
J
Jiri Denemark 已提交
6114 6115 6116
    if (NS_SUCCEEDED(rc))
        rc = data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
    if (NS_FAILED(rc)) {
6117
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6118 6119 6120 6121 6122 6123 6124 6125
                  _("could not open VirtualBox session with domain %s"),
                  dom->name);
        goto cleanup;
    }

    rc = console->vtbl->RestoreSnapshot(console, snapshot, &progress);
    if (NS_FAILED(rc) || !progress) {
        if (rc == VBOX_E_INVALID_VM_STATE) {
6126
            vboxError(VIR_ERR_OPERATION_INVALID, "%s",
J
Jiri Denemark 已提交
6127 6128
                      _("cannot restore domain snapshot for running domain"));
        } else {
6129
            vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6130 6131 6132 6133 6134 6135 6136 6137 6138
                      _("could not restore snapshot for domain %s"),
                      dom->name);
        }
        goto cleanup;
    }

    progress->vtbl->WaitForCompletion(progress, -1);
    progress->vtbl->GetResultCode(progress, &result);
    if (NS_FAILED(result)) {
6139
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6140 6141 6142 6143 6144 6145 6146 6147 6148
                  _("could not restore snapshot for domain %s"), dom->name);
        goto cleanup;
    }

    ret = 0;

cleanup:
    VBOX_RELEASE(progress);
    VBOX_RELEASE(console);
6149
    VBOX_SESSION_CLOSE();
6150
    vboxIIDUnalloc(&domiid);
J
Jiri Denemark 已提交
6151 6152 6153 6154 6155 6156
    return ret;
}
#endif

static int
vboxDomainRevertToSnapshot(virDomainSnapshotPtr snapshot,
6157
                           unsigned int flags)
J
Jiri Denemark 已提交
6158 6159 6160
{
    virDomainPtr dom = snapshot->domain;
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
6161
    vboxIID domiid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
6162 6163 6164 6165 6166 6167 6168
    IMachine *machine = NULL;
    ISnapshot *newSnapshot = NULL;
    ISnapshot *prevSnapshot = NULL;
    PRBool online = PR_FALSE;
    PRUint32 state;
    nsresult rc;

6169 6170
    virCheckFlags(0, -1);

6171
    vboxIIDFromUUID(&domiid, dom->uuid);
6172
    rc = VBOX_OBJECT_GET_MACHINE(domiid.value, &machine);
J
Jiri Denemark 已提交
6173
    if (NS_FAILED(rc)) {
6174
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
6175 6176 6177 6178 6179 6180 6181 6182 6183 6184
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    newSnapshot = vboxDomainSnapshotGet(data, dom, machine, snapshot->name);
    if (!newSnapshot)
        goto cleanup;

    rc = newSnapshot->vtbl->GetOnline(newSnapshot, &online);
    if (NS_FAILED(rc)) {
6185
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6186 6187 6188 6189 6190 6191 6192
                  _("could not get online state of snapshot %s"),
                  snapshot->name);
        goto cleanup;
    }

    rc = machine->vtbl->GetCurrentSnapshot(machine, &prevSnapshot);
    if (NS_FAILED(rc)) {
6193
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6194 6195 6196 6197 6198 6199 6200
                  _("could not get current snapshot of domain %s"),
                  dom->name);
        goto cleanup;
    }

    rc = machine->vtbl->GetState(machine, &state);
    if (NS_FAILED(rc)) {
6201
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6202 6203 6204 6205 6206 6207
                  _("could not get domain state"));
        goto cleanup;
    }

    if (state >= MachineState_FirstOnline
        && state <= MachineState_LastOnline) {
6208
        vboxError(VIR_ERR_OPERATION_INVALID, "%s",
J
Jiri Denemark 已提交
6209 6210 6211 6212 6213 6214 6215 6216 6217 6218 6219 6220 6221 6222 6223 6224 6225
                  _("cannot revert snapshot of running domain"));
        goto cleanup;
    }

    if (vboxDomainSnapshotRestore(dom, machine, newSnapshot))
        goto cleanup;

    if (online) {
        ret = vboxDomainCreate(dom);
        if (!ret)
            vboxDomainSnapshotRestore(dom, machine, prevSnapshot);
    } else
        ret = 0;

cleanup:
    VBOX_RELEASE(prevSnapshot);
    VBOX_RELEASE(newSnapshot);
6226
    vboxIIDUnalloc(&domiid);
J
Jiri Denemark 已提交
6227 6228 6229 6230 6231 6232 6233 6234 6235
    return ret;
}

static int
vboxDomainSnapshotDeleteSingle(vboxGlobalData *data,
                               IConsole *console,
                               ISnapshot *snapshot)
{
    IProgress *progress = NULL;
6236
    vboxIID iid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
6237 6238 6239 6240 6241 6242 6243 6244
    int ret = -1;
    nsresult rc;
#if VBOX_API_VERSION == 2002
    nsresult result;
#else
    PRInt32 result;
#endif

6245 6246
    rc = snapshot->vtbl->GetId(snapshot, &iid.value);
    if (NS_FAILED(rc)) {
6247
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6248 6249 6250 6251 6252
                  _("could not get snapshot UUID"));
        goto cleanup;
    }

#if VBOX_API_VERSION < 3001
6253
    rc = console->vtbl->DiscardSnapshot(console, iid.value, &progress);
J
Jiri Denemark 已提交
6254
#else
6255
    rc = console->vtbl->DeleteSnapshot(console, iid.value, &progress);
J
Jiri Denemark 已提交
6256 6257 6258
#endif
    if (NS_FAILED(rc) || !progress) {
        if (rc == VBOX_E_INVALID_VM_STATE) {
6259
            vboxError(VIR_ERR_OPERATION_INVALID, "%s",
J
Jiri Denemark 已提交
6260 6261
                      _("cannot delete domain snapshot for running domain"));
        } else {
6262
            vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6263 6264 6265 6266 6267 6268 6269 6270
                      _("could not delete snapshot"));
        }
        goto cleanup;
    }

    progress->vtbl->WaitForCompletion(progress, -1);
    progress->vtbl->GetResultCode(progress, &result);
    if (NS_FAILED(result)) {
6271
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6272 6273 6274 6275 6276 6277 6278 6279
                  _("could not delete snapshot"));
        goto cleanup;
    }

    ret = 0;

cleanup:
    VBOX_RELEASE(progress);
6280
    vboxIIDUnalloc(&iid);
J
Jiri Denemark 已提交
6281 6282 6283 6284 6285 6286 6287 6288
    return ret;
}

static int
vboxDomainSnapshotDeleteTree(vboxGlobalData *data,
                             IConsole *console,
                             ISnapshot *snapshot)
{
6289
    vboxArray children = VBOX_ARRAY_INITIALIZER;
J
Jiri Denemark 已提交
6290 6291 6292 6293
    int ret = -1;
    nsresult rc;
    unsigned int i;

6294
    rc = vboxArrayGet(&children, snapshot, snapshot->vtbl->GetChildren);
J
Jiri Denemark 已提交
6295
    if (NS_FAILED(rc)) {
6296
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6297 6298 6299 6300
                  _("could not get children snapshots"));
        goto cleanup;
    }

6301 6302 6303
    for (i = 0; i < children.count; i++) {
        if (vboxDomainSnapshotDeleteTree(data, console, children.items[i]))
            goto cleanup;
J
Jiri Denemark 已提交
6304 6305 6306 6307 6308
    }

    ret = vboxDomainSnapshotDeleteSingle(data, console, snapshot);

cleanup:
6309
    vboxArrayRelease(&children);
J
Jiri Denemark 已提交
6310 6311 6312 6313 6314 6315 6316 6317 6318
    return ret;
}

static int
vboxDomainSnapshotDelete(virDomainSnapshotPtr snapshot,
                         unsigned int flags)
{
    virDomainPtr dom = snapshot->domain;
    VBOX_OBJECT_CHECK(dom->conn, int, -1);
6319
    vboxIID domiid = VBOX_IID_INITIALIZER;
J
Jiri Denemark 已提交
6320 6321 6322 6323 6324 6325
    IMachine *machine = NULL;
    ISnapshot *snap = NULL;
    IConsole *console = NULL;
    PRUint32 state;
    nsresult rc;

6326 6327
    virCheckFlags(VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN, -1);

6328
    vboxIIDFromUUID(&domiid, dom->uuid);
6329
    rc = VBOX_OBJECT_GET_MACHINE(domiid.value, &machine);
J
Jiri Denemark 已提交
6330
    if (NS_FAILED(rc)) {
6331
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
J
Jiri Denemark 已提交
6332 6333 6334 6335 6336 6337 6338 6339 6340 6341
                  _("no domain with matching UUID"));
        goto cleanup;
    }

    snap = vboxDomainSnapshotGet(data, dom, machine, snapshot->name);
    if (!snap)
        goto cleanup;

    rc = machine->vtbl->GetState(machine, &state);
    if (NS_FAILED(rc)) {
6342
        vboxError(VIR_ERR_INTERNAL_ERROR, "%s",
J
Jiri Denemark 已提交
6343 6344 6345 6346 6347 6348
                  _("could not get domain state"));
        goto cleanup;
    }

    if (state >= MachineState_FirstOnline
        && state <= MachineState_LastOnline) {
6349
        vboxError(VIR_ERR_OPERATION_INVALID, "%s",
J
Jiri Denemark 已提交
6350 6351 6352 6353
                  _("cannot delete snapshots of running domain"));
        goto cleanup;
    }

6354
    rc = VBOX_SESSION_OPEN(domiid.value, machine);
J
Jiri Denemark 已提交
6355 6356 6357
    if (NS_SUCCEEDED(rc))
        rc = data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
    if (NS_FAILED(rc)) {
6358
        vboxError(VIR_ERR_INTERNAL_ERROR,
J
Jiri Denemark 已提交
6359 6360 6361 6362 6363 6364 6365 6366 6367 6368 6369 6370 6371
                  _("could not open VirtualBox session with domain %s"),
                  dom->name);
        goto cleanup;
    }

    if (flags & VIR_DOMAIN_SNAPSHOT_DELETE_CHILDREN)
        ret = vboxDomainSnapshotDeleteTree(data, console, snap);
    else
        ret = vboxDomainSnapshotDeleteSingle(data, console, snap);

cleanup:
    VBOX_RELEASE(console);
    VBOX_RELEASE(snap);
6372
    vboxIIDUnalloc(&domiid);
6373
    VBOX_SESSION_CLOSE();
J
Jiri Denemark 已提交
6374 6375 6376
    return ret;
}

6377
#if VBOX_API_VERSION == 2002 || VBOX_API_VERSION == 4000
6378
    /* No Callback support for VirtualBox 2.2.* series */
6379
#else /* !(VBOX_API_VERSION == 2002) && !(VBOX_API_VERSION == 4000) */
6380 6381

/* Functions needed for Callbacks */
6382 6383 6384 6385
static nsresult PR_COM_METHOD
vboxCallbackOnMachineStateChange(IVirtualBoxCallback *pThis,
                                 PRUnichar *machineId, PRUint32 state)
{
6386 6387 6388 6389 6390 6391 6392
    virDomainPtr dom = NULL;
    int event        = 0;
    int detail       = 0;

    g_pVBoxGlobalData->domainEventDispatching = 1;
    vboxDriverLock(g_pVBoxGlobalData);

6393
    VIR_DEBUG("IVirtualBoxCallback: %p, State: %d", pThis, state);
6394 6395 6396 6397 6398 6399 6400 6401 6402 6403 6404
    DEBUGPRUnichar("machineId", machineId);

    if (machineId) {
        char *machineIdUtf8       = NULL;
        unsigned char uuid[VIR_UUID_BUFLEN];

        g_pVBoxGlobalData->pFuncs->pfnUtf16ToUtf8(machineId, &machineIdUtf8);
        virUUIDParse(machineIdUtf8, uuid);

        dom = vboxDomainLookupByUUID(g_pVBoxGlobalData->conn, uuid);
        if (dom) {
6405
            virDomainEventPtr ev;
6406 6407 6408 6409 6410 6411 6412 6413 6414 6415 6416 6417 6418 6419 6420 6421 6422 6423 6424 6425 6426 6427 6428 6429 6430 6431 6432 6433 6434 6435

            if (state == MachineState_Starting) {
                event  = VIR_DOMAIN_EVENT_STARTED;
                detail = VIR_DOMAIN_EVENT_STARTED_BOOTED;
            } else if (state == MachineState_Restoring) {
                event  = VIR_DOMAIN_EVENT_STARTED;
                detail = VIR_DOMAIN_EVENT_STARTED_RESTORED;
            } else if (state == MachineState_Paused) {
                event  = VIR_DOMAIN_EVENT_SUSPENDED;
                detail = VIR_DOMAIN_EVENT_SUSPENDED_PAUSED;
            } else if (state == MachineState_Running) {
                event  = VIR_DOMAIN_EVENT_RESUMED;
                detail = VIR_DOMAIN_EVENT_RESUMED_UNPAUSED;
            } else if (state == MachineState_PoweredOff) {
                event  = VIR_DOMAIN_EVENT_STOPPED;
                detail = VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN;
            } else if (state == MachineState_Stopping) {
                event  = VIR_DOMAIN_EVENT_STOPPED;
                detail = VIR_DOMAIN_EVENT_STOPPED_DESTROYED;
            } else if (state == MachineState_Aborted) {
                event  = VIR_DOMAIN_EVENT_STOPPED;
                detail = VIR_DOMAIN_EVENT_STOPPED_CRASHED;
            } else if (state == MachineState_Saving) {
                event  = VIR_DOMAIN_EVENT_STOPPED;
                detail = VIR_DOMAIN_EVENT_STOPPED_SAVED;
            } else {
                event  = VIR_DOMAIN_EVENT_STOPPED;
                detail = VIR_DOMAIN_EVENT_STOPPED_SHUTDOWN;
            }

6436 6437 6438 6439 6440 6441 6442 6443
            ev = virDomainEventNewFromDom(dom, event, detail);

            if (ev) {
                virDomainEventDispatch(ev,
                                       g_pVBoxGlobalData->domainEventCallbacks,
                                       virDomainEventDispatchDefaultFunc,
                                       NULL);
                virDomainEventFree(ev);
6444 6445 6446 6447 6448 6449 6450 6451 6452 6453 6454 6455
            }
        }
    }

    virDomainEventCallbackListPurgeMarked(g_pVBoxGlobalData->domainEventCallbacks);

    vboxDriverUnlock(g_pVBoxGlobalData);
    g_pVBoxGlobalData->domainEventDispatching = 0;

    return NS_OK;
}

6456 6457 6458 6459
static nsresult PR_COM_METHOD
vboxCallbackOnMachineDataChange(IVirtualBoxCallback *pThis,
                                PRUnichar *machineId)
{
6460
    VIR_DEBUG("IVirtualBoxCallback: %p", pThis);
6461 6462 6463 6464 6465
    DEBUGPRUnichar("machineId", machineId);

    return NS_OK;
}

6466 6467 6468 6469 6470 6471 6472
static nsresult PR_COM_METHOD
vboxCallbackOnExtraDataCanChange(IVirtualBoxCallback *pThis,
                                 PRUnichar *machineId, PRUnichar *key,
                                 PRUnichar *value,
                                 PRUnichar **error ATTRIBUTE_UNUSED,
                                 PRBool *allowChange)
{
6473
    VIR_DEBUG("IVirtualBoxCallback: %p, allowChange: %s", pThis, *allowChange ? "true" : "false");
6474 6475 6476 6477 6478 6479 6480
    DEBUGPRUnichar("machineId", machineId);
    DEBUGPRUnichar("key", key);
    DEBUGPRUnichar("value", value);

    return NS_OK;
}

6481 6482 6483 6484
static nsresult PR_COM_METHOD
vboxCallbackOnExtraDataChange(IVirtualBoxCallback *pThis, PRUnichar *machineId,
                              PRUnichar *key, PRUnichar *value)
{
6485
    VIR_DEBUG("IVirtualBoxCallback: %p", pThis);
6486 6487 6488 6489 6490 6491 6492
    DEBUGPRUnichar("machineId", machineId);
    DEBUGPRUnichar("key", key);
    DEBUGPRUnichar("value", value);

    return NS_OK;
}

6493
# if VBOX_API_VERSION < 3001
6494 6495 6496 6497
static nsresult PR_COM_METHOD
vboxCallbackOnMediaRegistered(IVirtualBoxCallback *pThis, PRUnichar *mediaId,
                              PRUint32 mediaType, PRBool registered)
{
6498 6499
    VIR_DEBUG("IVirtualBoxCallback: %p, registered: %s", pThis, registered ? "true" : "false");
    VIR_DEBUG("mediaType: %d", mediaType);
6500 6501 6502 6503
    DEBUGPRUnichar("mediaId", mediaId);

    return NS_OK;
}
6504 6505
# else  /* VBOX_API_VERSION >= 3001 */
# endif /* VBOX_API_VERSION >= 3001 */
6506

6507 6508 6509 6510
static nsresult PR_COM_METHOD
vboxCallbackOnMachineRegistered(IVirtualBoxCallback *pThis,
                                PRUnichar *machineId, PRBool registered)
{
6511 6512 6513 6514 6515 6516 6517
    virDomainPtr dom = NULL;
    int event        = 0;
    int detail       = 0;

    g_pVBoxGlobalData->domainEventDispatching = 1;
    vboxDriverLock(g_pVBoxGlobalData);

6518
    VIR_DEBUG("IVirtualBoxCallback: %p, registered: %s", pThis, registered ? "true" : "false");
6519 6520 6521 6522 6523 6524 6525 6526 6527 6528 6529
    DEBUGPRUnichar("machineId", machineId);

    if (machineId) {
        char *machineIdUtf8       = NULL;
        unsigned char uuid[VIR_UUID_BUFLEN];

        g_pVBoxGlobalData->pFuncs->pfnUtf16ToUtf8(machineId, &machineIdUtf8);
        virUUIDParse(machineIdUtf8, uuid);

        dom = vboxDomainLookupByUUID(g_pVBoxGlobalData->conn, uuid);
        if (dom) {
6530
            virDomainEventPtr ev;
6531 6532 6533 6534 6535 6536 6537 6538 6539 6540 6541 6542 6543 6544 6545

            /* CURRENT LIMITATION: we never get the VIR_DOMAIN_EVENT_UNDEFINED
             * event becuase the when the machine is de-registered the call
             * to vboxDomainLookupByUUID fails and thus we don't get any
             * dom pointer which is necessary (null dom pointer doesn't work)
             * to show the VIR_DOMAIN_EVENT_UNDEFINED event
             */
            if (registered) {
                event  = VIR_DOMAIN_EVENT_DEFINED;
                detail = VIR_DOMAIN_EVENT_DEFINED_ADDED;
            } else {
                event  = VIR_DOMAIN_EVENT_UNDEFINED;
                detail = VIR_DOMAIN_EVENT_UNDEFINED_REMOVED;
            }

6546 6547 6548 6549 6550 6551 6552 6553
            ev = virDomainEventNewFromDom(dom, event, detail);

            if (ev) {
                virDomainEventDispatch(ev,
                                       g_pVBoxGlobalData->domainEventCallbacks,
                                       virDomainEventDispatchDefaultFunc,
                                       NULL);
                virDomainEventFree(ev);
6554 6555 6556 6557 6558 6559 6560 6561 6562 6563 6564 6565
            }
        }
    }

    virDomainEventCallbackListPurgeMarked(g_pVBoxGlobalData->domainEventCallbacks);

    vboxDriverUnlock(g_pVBoxGlobalData);
    g_pVBoxGlobalData->domainEventDispatching = 0;

    return NS_OK;
}

6566 6567 6568 6569
static nsresult PR_COM_METHOD
vboxCallbackOnSessionStateChange(IVirtualBoxCallback *pThis,
                                 PRUnichar *machineId, PRUint32 state)
{
6570
    VIR_DEBUG("IVirtualBoxCallback: %p, state: %d", pThis, state);
6571 6572 6573 6574 6575
    DEBUGPRUnichar("machineId", machineId);

    return NS_OK;
}

6576 6577 6578 6579
static nsresult PR_COM_METHOD
vboxCallbackOnSnapshotTaken(IVirtualBoxCallback *pThis, PRUnichar *machineId,
                            PRUnichar *snapshotId)
{
6580
    VIR_DEBUG("IVirtualBoxCallback: %p", pThis);
6581 6582 6583 6584 6585 6586
    DEBUGPRUnichar("machineId", machineId);
    DEBUGPRUnichar("snapshotId", snapshotId);

    return NS_OK;
}

6587 6588 6589 6590
static nsresult PR_COM_METHOD
vboxCallbackOnSnapshotDiscarded(IVirtualBoxCallback *pThis, PRUnichar *machineId,
                                PRUnichar *snapshotId)
{
6591
    VIR_DEBUG("IVirtualBoxCallback: %p", pThis);
6592 6593 6594 6595 6596 6597
    DEBUGPRUnichar("machineId", machineId);
    DEBUGPRUnichar("snapshotId", snapshotId);

    return NS_OK;
}

6598 6599 6600 6601
static nsresult PR_COM_METHOD
vboxCallbackOnSnapshotChange(IVirtualBoxCallback *pThis, PRUnichar *machineId,
                             PRUnichar *snapshotId)
{
6602
    VIR_DEBUG("IVirtualBoxCallback: %p", pThis);
6603 6604 6605 6606 6607 6608
    DEBUGPRUnichar("machineId", machineId);
    DEBUGPRUnichar("snapshotId", snapshotId);

    return NS_OK;
}

6609 6610 6611 6612 6613
static nsresult PR_COM_METHOD
vboxCallbackOnGuestPropertyChange(IVirtualBoxCallback *pThis,
                                  PRUnichar *machineId, PRUnichar *name,
                                  PRUnichar *value, PRUnichar *flags)
{
6614
    VIR_DEBUG("IVirtualBoxCallback: %p", pThis);
6615 6616 6617 6618 6619 6620 6621 6622
    DEBUGPRUnichar("machineId", machineId);
    DEBUGPRUnichar("name", name);
    DEBUGPRUnichar("value", value);
    DEBUGPRUnichar("flags", flags);

    return NS_OK;
}

6623 6624 6625
static nsresult PR_COM_METHOD
vboxCallbackAddRef(nsISupports *pThis)
{
6626 6627 6628 6629
    nsresult c;

    c = ++g_pVBoxGlobalData->vboxCallBackRefCount;

6630
    VIR_DEBUG("pThis: %p, vboxCallback AddRef: %d", pThis, c);
6631 6632 6633 6634

    return c;
}

6635 6636 6637
static nsresult PR_COM_METHOD
vboxCallbackRelease(nsISupports *pThis)
{
6638 6639 6640 6641 6642 6643 6644 6645 6646
    nsresult c;

    c = --g_pVBoxGlobalData->vboxCallBackRefCount;
    if (c == 0) {
        /* delete object */
        VIR_FREE(pThis->vtbl);
        VIR_FREE(pThis);
    }

6647
    VIR_DEBUG("pThis: %p, vboxCallback Release: %d", pThis, c);
6648 6649 6650 6651

    return c;
}

6652 6653 6654
static nsresult PR_COM_METHOD
vboxCallbackQueryInterface(nsISupports *pThis, const nsID *iid, void **resultp)
{
6655 6656 6657 6658 6659 6660 6661 6662 6663 6664
    IVirtualBoxCallback *that = (IVirtualBoxCallback *)pThis;
    static const nsID ivirtualboxCallbackUUID = IVIRTUALBOXCALLBACK_IID;
    static const nsID isupportIID = NS_ISUPPORTS_IID;

    /* Match UUID for IVirtualBoxCallback class */
    if (    memcmp(iid, &ivirtualboxCallbackUUID, sizeof(nsID)) == 0
        ||  memcmp(iid, &isupportIID, sizeof(nsID)) == 0) {
        g_pVBoxGlobalData->vboxCallBackRefCount++;
        *resultp = that;

6665
        VIR_DEBUG("pThis: %p, vboxCallback QueryInterface: %d", pThis, g_pVBoxGlobalData->vboxCallBackRefCount);
6666 6667 6668 6669 6670

        return NS_OK;
    }


6671
    VIR_DEBUG("pThis: %p, vboxCallback QueryInterface didn't find a matching interface", pThis);
6672 6673 6674 6675 6676 6677
    DEBUGUUID("The UUID Callback Interface expects", iid);
    DEBUGUUID("The UUID Callback Interface got", &ivirtualboxCallbackUUID);
    return NS_NOINTERFACE;
}


6678
static IVirtualBoxCallback *vboxAllocCallbackObj(void) {
6679 6680
    IVirtualBoxCallback *vboxCallback = NULL;

6681
    /* Allocate, Initialize and return a valid
6682 6683 6684
     * IVirtualBoxCallback object here
     */
    if ((VIR_ALLOC(vboxCallback) < 0) || (VIR_ALLOC(vboxCallback->vtbl) < 0)) {
6685
        VIR_FREE(vboxCallback);
6686
        virReportOOMError();
6687 6688 6689 6690 6691 6692 6693 6694 6695 6696 6697
        return NULL;
    }

    {
        vboxCallback->vtbl->nsisupports.AddRef          = &vboxCallbackAddRef;
        vboxCallback->vtbl->nsisupports.Release         = &vboxCallbackRelease;
        vboxCallback->vtbl->nsisupports.QueryInterface  = &vboxCallbackQueryInterface;
        vboxCallback->vtbl->OnMachineStateChange        = &vboxCallbackOnMachineStateChange;
        vboxCallback->vtbl->OnMachineDataChange         = &vboxCallbackOnMachineDataChange;
        vboxCallback->vtbl->OnExtraDataCanChange        = &vboxCallbackOnExtraDataCanChange;
        vboxCallback->vtbl->OnExtraDataChange           = &vboxCallbackOnExtraDataChange;
6698
# if VBOX_API_VERSION < 3001
6699
        vboxCallback->vtbl->OnMediaRegistered           = &vboxCallbackOnMediaRegistered;
6700 6701
# else  /* VBOX_API_VERSION >= 3001 */
# endif /* VBOX_API_VERSION >= 3001 */
6702 6703 6704
        vboxCallback->vtbl->OnMachineRegistered         = &vboxCallbackOnMachineRegistered;
        vboxCallback->vtbl->OnSessionStateChange        = &vboxCallbackOnSessionStateChange;
        vboxCallback->vtbl->OnSnapshotTaken             = &vboxCallbackOnSnapshotTaken;
6705
# if VBOX_API_VERSION < 3002
6706
        vboxCallback->vtbl->OnSnapshotDiscarded         = &vboxCallbackOnSnapshotDiscarded;
6707 6708 6709
# else /* VBOX_API_VERSION >= 3002 */
        vboxCallback->vtbl->OnSnapshotDeleted           = &vboxCallbackOnSnapshotDiscarded;
# endif /* VBOX_API_VERSION >= 3002 */
6710 6711 6712 6713 6714 6715 6716 6717 6718 6719 6720 6721 6722 6723 6724 6725 6726 6727 6728 6729 6730 6731 6732 6733 6734 6735 6736 6737 6738
        vboxCallback->vtbl->OnSnapshotChange            = &vboxCallbackOnSnapshotChange;
        vboxCallback->vtbl->OnGuestPropertyChange       = &vboxCallbackOnGuestPropertyChange;
        g_pVBoxGlobalData->vboxCallBackRefCount = 1;

    }

    return vboxCallback;
}

static void vboxReadCallback(int watch ATTRIBUTE_UNUSED,
                             int fd,
                             int events ATTRIBUTE_UNUSED,
                             void *opaque ATTRIBUTE_UNUSED) {
    if (fd >= 0) {
        g_pVBoxGlobalData->vboxQueue->vtbl->ProcessPendingEvents(g_pVBoxGlobalData->vboxQueue);
    } else {
        nsresult rc;
        PLEvent *pEvent = NULL;

        rc = g_pVBoxGlobalData->vboxQueue->vtbl->WaitForEvent(g_pVBoxGlobalData->vboxQueue, &pEvent);
        if (NS_SUCCEEDED(rc))
            g_pVBoxGlobalData->vboxQueue->vtbl->HandleEvent(g_pVBoxGlobalData->vboxQueue, pEvent);
    }
}

static int vboxDomainEventRegister (virConnectPtr conn,
                                    virConnectDomainEventCallback callback,
                                    void *opaque,
                                    virFreeCallback freecb) {
6739
    VBOX_OBJECT_CHECK(conn, int, -1);
6740
    int vboxRet          = -1;
6741
    nsresult rc;
6742 6743 6744 6745 6746 6747

    /* Locking has to be there as callbacks are not
     * really fully thread safe
     */
    vboxDriverLock(data);

6748
    if (data->vboxCallback == NULL) {
6749
        data->vboxCallback = vboxAllocCallbackObj();
6750 6751 6752 6753
        if (data->vboxCallback != NULL) {
            rc = data->vboxObj->vtbl->RegisterCallback(data->vboxObj, data->vboxCallback);
            if (NS_SUCCEEDED(rc)) {
                vboxRet = 0;
6754 6755
            }
        }
6756 6757 6758
    } else {
        vboxRet = 0;
    }
6759

6760 6761 6762 6763 6764 6765 6766
    /* Get the vbox file handle and add a event handle to it
     * so that the events can be passed down to the user
     */
    if (vboxRet == 0) {
        if (data->fdWatch < 0) {
            PRInt32 vboxFileHandle;
            vboxFileHandle = data->vboxQueue->vtbl->GetEventQueueSelectFD(data->vboxQueue);
6767

6768 6769
            data->fdWatch = virEventAddHandle(vboxFileHandle, VIR_EVENT_HANDLE_READABLE, vboxReadCallback, NULL, NULL);
        }
6770

6771 6772 6773 6774 6775
        if (data->fdWatch >= 0) {
            /* Once a callback is registered with virtualbox, use a list
             * to store the callbacks registered with libvirt so that
             * later you can iterate over them
             */
6776

6777 6778
            ret = virDomainEventCallbackListAdd(conn, data->domainEventCallbacks,
                                                callback, opaque, freecb);
6779
            VIR_DEBUG("virDomainEventCallbackListAdd (ret = %d) ( conn: %p, "
6780 6781 6782
                  "data->domainEventCallbacks: %p, callback: %p, opaque: %p, "
                  "freecb: %p )", ret, conn, data->domainEventCallbacks, callback,
                  opaque, freecb);
6783 6784 6785 6786 6787 6788 6789 6790 6791 6792 6793 6794 6795 6796 6797 6798 6799
        }
    }

    vboxDriverUnlock(data);

    if (ret >= 0) {
        return ret;
    } else {
        if (data->vboxObj && data->vboxCallback) {
            data->vboxObj->vtbl->UnregisterCallback(data->vboxObj, data->vboxCallback);
        }
        return -1;
    }
}

static int vboxDomainEventDeregister (virConnectPtr conn,
                                      virConnectDomainEventCallback callback) {
6800
    VBOX_OBJECT_CHECK(conn, int, -1);
6801 6802 6803 6804 6805 6806 6807 6808 6809 6810 6811 6812 6813

    /* Locking has to be there as callbacks are not
     * really fully thread safe
     */
    vboxDriverLock(data);

    if (data->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDelete(conn, data->domainEventCallbacks,
                                                   callback);
    else
        ret = virDomainEventCallbackListRemove(conn, data->domainEventCallbacks,
                                               callback);

6814
    if (data->vboxCallback) {
6815 6816 6817
        /* check count here of how many times register was called
         * and only on the last de-register do the un-register call
         */
6818
        if (data->domainEventCallbacks && virDomainEventCallbackListCount(data->domainEventCallbacks) == 0) {
6819
            data->vboxObj->vtbl->UnregisterCallback(data->vboxObj, data->vboxCallback);
6820
            VBOX_RELEASE(data->vboxCallback);
6821 6822 6823 6824 6825 6826 6827 6828 6829 6830 6831 6832

            /* Remove the Event file handle on which we are listening as well */
            virEventRemoveHandle(data->fdWatch);
            data->fdWatch = -1;
        }
    }

    vboxDriverUnlock(data);

    return ret;
}

6833 6834 6835 6836 6837 6838 6839 6840 6841 6842 6843 6844 6845 6846 6847 6848 6849 6850 6851 6852 6853 6854 6855 6856 6857 6858 6859 6860 6861 6862 6863 6864 6865 6866 6867 6868 6869 6870 6871 6872 6873 6874 6875 6876 6877 6878 6879
static int vboxDomainEventRegisterAny(virConnectPtr conn,
                                      virDomainPtr dom,
                                      int eventID,
                                      virConnectDomainEventGenericCallback callback,
                                      void *opaque,
                                      virFreeCallback freecb) {
    VBOX_OBJECT_CHECK(conn, int, -1);
    int vboxRet          = -1;
    nsresult rc;

    /* Locking has to be there as callbacks are not
     * really fully thread safe
     */
    vboxDriverLock(data);

    if (data->vboxCallback == NULL) {
        data->vboxCallback = vboxAllocCallbackObj();
        if (data->vboxCallback != NULL) {
            rc = data->vboxObj->vtbl->RegisterCallback(data->vboxObj, data->vboxCallback);
            if (NS_SUCCEEDED(rc)) {
                vboxRet = 0;
            }
        }
    } else {
        vboxRet = 0;
    }

    /* Get the vbox file handle and add a event handle to it
     * so that the events can be passed down to the user
     */
    if (vboxRet == 0) {
        if (data->fdWatch < 0) {
            PRInt32 vboxFileHandle;
            vboxFileHandle = data->vboxQueue->vtbl->GetEventQueueSelectFD(data->vboxQueue);

            data->fdWatch = virEventAddHandle(vboxFileHandle, VIR_EVENT_HANDLE_READABLE, vboxReadCallback, NULL, NULL);
        }

        if (data->fdWatch >= 0) {
            /* Once a callback is registered with virtualbox, use a list
             * to store the callbacks registered with libvirt so that
             * later you can iterate over them
             */

            ret = virDomainEventCallbackListAddID(conn, data->domainEventCallbacks,
                                                  dom, eventID,
                                                  callback, opaque, freecb);
6880
            VIR_DEBUG("virDomainEventCallbackListAddID (ret = %d) ( conn: %p, "
6881 6882 6883 6884 6885 6886 6887 6888 6889 6890 6891 6892 6893 6894 6895 6896 6897 6898 6899 6900 6901 6902 6903 6904 6905 6906 6907 6908 6909 6910 6911 6912 6913 6914 6915 6916 6917 6918 6919 6920 6921 6922 6923 6924 6925 6926 6927 6928 6929 6930 6931 6932 6933
                  "data->domainEventCallbacks: %p, callback: %p, opaque: %p, "
                  "freecb: %p )", ret, conn, data->domainEventCallbacks, callback,
                  opaque, freecb);
        }
    }

    vboxDriverUnlock(data);

    if (ret >= 0) {
        return ret;
    } else {
        if (data->vboxObj && data->vboxCallback) {
            data->vboxObj->vtbl->UnregisterCallback(data->vboxObj, data->vboxCallback);
        }
        return -1;
    }
}

static int vboxDomainEventDeregisterAny(virConnectPtr conn,
                                        int callbackID) {
    VBOX_OBJECT_CHECK(conn, int, -1);

    /* Locking has to be there as callbacks are not
     * really fully thread safe
     */
    vboxDriverLock(data);

    if (data->domainEventDispatching)
        ret = virDomainEventCallbackListMarkDeleteID(conn, data->domainEventCallbacks,
                                                     callbackID);
    else
        ret = virDomainEventCallbackListRemoveID(conn, data->domainEventCallbacks,
                                                 callbackID);

    if (data->vboxCallback) {
        /* check count here of how many times register was called
         * and only on the last de-register do the un-register call
         */
        if (data->domainEventCallbacks && virDomainEventCallbackListCount(data->domainEventCallbacks) == 0) {
            data->vboxObj->vtbl->UnregisterCallback(data->vboxObj, data->vboxCallback);
            VBOX_RELEASE(data->vboxCallback);

            /* Remove the Event file handle on which we are listening as well */
            virEventRemoveHandle(data->fdWatch);
            data->fdWatch = -1;
        }
    }

    vboxDriverUnlock(data);

    return ret;
}

6934
#endif /* !(VBOX_API_VERSION == 2002) && !(VBOX_API_VERSION == 4000) */
6935

6936 6937 6938 6939 6940
/**
 * The Network Functions here on
 */
static virDrvOpenStatus vboxNetworkOpen(virConnectPtr conn,
                                        virConnectAuthPtr auth ATTRIBUTE_UNUSED,
6941
                                        unsigned int flags ATTRIBUTE_UNUSED) {
6942 6943 6944 6945 6946 6947 6948 6949 6950 6951
    vboxGlobalData *data = conn->privateData;

    if (STRNEQ(conn->driver->name, "VBOX"))
        goto cleanup;

    if ((data->pFuncs      == NULL) ||
        (data->vboxObj     == NULL) ||
        (data->vboxSession == NULL))
        goto cleanup;

6952
    VIR_DEBUG("network initialized");
6953 6954 6955 6956 6957 6958 6959 6960
    /* conn->networkPrivateData = some network specific data */
    return VIR_DRV_OPEN_SUCCESS;

cleanup:
    return VIR_DRV_OPEN_DECLINED;
}

static int vboxNetworkClose(virConnectPtr conn) {
6961
    VIR_DEBUG("network uninitialized");
6962 6963 6964 6965 6966
    conn->networkPrivateData = NULL;
    return 0;
}

static int vboxNumOfNetworks(virConnectPtr conn) {
6967
    VBOX_OBJECT_HOST_CHECK(conn, int, 0);
6968
    vboxArray networkInterfaces = VBOX_ARRAY_INITIALIZER;
6969
    int i = 0;
6970

6971
    vboxArrayGet(&networkInterfaces, host, host->vtbl->GetNetworkInterfaces);
6972

6973 6974 6975 6976
    for (i = 0; i < networkInterfaces.count; i++) {
        IHostNetworkInterface *networkInterface = networkInterfaces.items[i];

        if (networkInterface) {
6977
            PRUint32 interfaceType = 0;
6978

6979
            networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
6980 6981
            if (interfaceType == HostNetworkInterfaceType_HostOnly) {
                PRUint32 status = HostNetworkInterfaceStatus_Unknown;
6982

6983
                networkInterface->vtbl->GetStatus(networkInterface, &status);
6984

6985 6986
                if (status == HostNetworkInterfaceStatus_Up)
                    ret++;
6987 6988 6989 6990
            }
        }
    }

6991 6992
    vboxArrayRelease(&networkInterfaces);

6993 6994
    VBOX_RELEASE(host);

6995
    VIR_DEBUG("numActive: %d", ret);
6996
    return ret;
6997 6998 6999
}

static int vboxListNetworks(virConnectPtr conn, char **const names, int nnames) {
7000
    VBOX_OBJECT_HOST_CHECK(conn, int, 0);
7001
    vboxArray networkInterfaces = VBOX_ARRAY_INITIALIZER;
7002
    int i = 0;
7003

7004 7005 7006 7007
    vboxArrayGet(&networkInterfaces, host, host->vtbl->GetNetworkInterfaces);

    for (i = 0; (ret < nnames) && (i < networkInterfaces.count); i++) {
        IHostNetworkInterface *networkInterface = networkInterfaces.items[i];
7008

7009
        if (networkInterface) {
7010
            PRUint32 interfaceType = 0;
7011

7012
            networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7013

7014 7015
            if (interfaceType == HostNetworkInterfaceType_HostOnly) {
                PRUint32 status = HostNetworkInterfaceStatus_Unknown;
7016

7017
                networkInterface->vtbl->GetStatus(networkInterface, &status);
7018

7019 7020 7021
                if (status == HostNetworkInterfaceStatus_Up) {
                    char *nameUtf8       = NULL;
                    PRUnichar *nameUtf16 = NULL;
7022

7023
                    networkInterface->vtbl->GetName(networkInterface, &nameUtf16);
7024
                    VBOX_UTF16_TO_UTF8(nameUtf16, &nameUtf8);
7025

7026
                    VIR_DEBUG("nnames[%d]: %s", ret, nameUtf8);
7027 7028
                    names[ret] = strdup(nameUtf8);
                    if (names[ret] == NULL) {
7029
                        virReportOOMError();
7030 7031
                    } else {
                        ret++;
7032 7033
                    }

7034 7035
                    VBOX_UTF8_FREE(nameUtf8);
                    VBOX_UTF16_FREE(nameUtf16);
7036 7037 7038 7039 7040
                }
            }
        }
    }

7041
    vboxArrayRelease(&networkInterfaces);
7042

7043
    VBOX_RELEASE(host);
7044

7045 7046
    return ret;
}
7047

7048 7049
static int vboxNumOfDefinedNetworks(virConnectPtr conn) {
    VBOX_OBJECT_HOST_CHECK(conn, int, 0);
7050
    vboxArray networkInterfaces = VBOX_ARRAY_INITIALIZER;
7051
    int i = 0;
7052

7053
    vboxArrayGet(&networkInterfaces, host, host->vtbl->GetNetworkInterfaces);
7054

7055 7056 7057 7058
    for (i = 0; i < networkInterfaces.count; i++) {
        IHostNetworkInterface *networkInterface = networkInterfaces.items[i];

        if (networkInterface) {
7059
            PRUint32 interfaceType = 0;
7060

7061
            networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7062 7063
            if (interfaceType == HostNetworkInterfaceType_HostOnly) {
                PRUint32 status = HostNetworkInterfaceStatus_Unknown;
7064

7065
                networkInterface->vtbl->GetStatus(networkInterface, &status);
7066

7067 7068
                if (status == HostNetworkInterfaceStatus_Down)
                    ret++;
7069 7070 7071 7072
            }
        }
    }

7073 7074
    vboxArrayRelease(&networkInterfaces);

7075 7076
    VBOX_RELEASE(host);

7077
    VIR_DEBUG("numActive: %d", ret);
7078
    return ret;
7079 7080 7081
}

static int vboxListDefinedNetworks(virConnectPtr conn, char **const names, int nnames) {
7082
    VBOX_OBJECT_HOST_CHECK(conn, int, 0);
7083
    vboxArray networkInterfaces = VBOX_ARRAY_INITIALIZER;
7084
    int i = 0;
7085

7086 7087 7088 7089
    vboxArrayGet(&networkInterfaces, host, host->vtbl->GetNetworkInterfaces);

    for (i = 0; (ret < nnames) && (i < networkInterfaces.count); i++) {
        IHostNetworkInterface *networkInterface = networkInterfaces.items[i];
7090

7091
        if (networkInterface) {
7092
            PRUint32 interfaceType = 0;
7093

7094
            networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7095

7096 7097
            if (interfaceType == HostNetworkInterfaceType_HostOnly) {
                PRUint32 status = HostNetworkInterfaceStatus_Unknown;
7098

7099
                networkInterface->vtbl->GetStatus(networkInterface, &status);
7100

7101 7102 7103
                if (status == HostNetworkInterfaceStatus_Down) {
                    char *nameUtf8       = NULL;
                    PRUnichar *nameUtf16 = NULL;
7104

7105
                    networkInterface->vtbl->GetName(networkInterface, &nameUtf16);
7106
                    VBOX_UTF16_TO_UTF8(nameUtf16, &nameUtf8);
7107

7108
                    VIR_DEBUG("nnames[%d]: %s", ret, nameUtf8);
7109 7110
                    names[ret] = strdup(nameUtf8);
                    if (names[ret] == NULL) {
7111
                        virReportOOMError();
7112 7113
                    } else {
                        ret++;
7114 7115
                    }

7116 7117
                    VBOX_UTF8_FREE(nameUtf8);
                    VBOX_UTF16_FREE(nameUtf16);
7118 7119 7120 7121 7122
                }
            }
        }
    }

7123
    vboxArrayRelease(&networkInterfaces);
7124 7125 7126 7127

    VBOX_RELEASE(host);

    return ret;
7128 7129
}

7130 7131 7132
static virNetworkPtr
vboxNetworkLookupByUUID(virConnectPtr conn, const unsigned char *uuid)
{
7133
    VBOX_OBJECT_HOST_CHECK(conn, virNetworkPtr, NULL);
7134
    vboxIID iid = VBOX_IID_INITIALIZER;
7135

7136
    vboxIIDFromUUID(&iid, uuid);
7137

7138 7139 7140
    /* TODO: "internal" networks are just strings and
     * thus can't do much with them
     */
7141
    IHostNetworkInterface *networkInterface = NULL;
7142

7143
    host->vtbl->FindHostNetworkInterfaceById(host, iid.value, &networkInterface);
7144 7145
    if (networkInterface) {
        PRUint32 interfaceType = 0;
7146

7147
        networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7148

7149 7150 7151
        if (interfaceType == HostNetworkInterfaceType_HostOnly) {
            char *nameUtf8       = NULL;
            PRUnichar *nameUtf16 = NULL;
7152

7153 7154
            networkInterface->vtbl->GetName(networkInterface, &nameUtf16);
            VBOX_UTF16_TO_UTF8(nameUtf16, &nameUtf8);
7155

7156
            ret = virGetNetwork(conn, nameUtf8, uuid);
7157

7158
            VIR_DEBUG("Network Name: %s", nameUtf8);
7159
            DEBUGIID("Network UUID", iid.value);
7160

7161 7162
            VBOX_UTF8_FREE(nameUtf8);
            VBOX_UTF16_FREE(nameUtf16);
7163
        }
7164 7165

        VBOX_RELEASE(networkInterface);
7166 7167
    }

7168 7169
    VBOX_RELEASE(host);

7170
    vboxIIDUnalloc(&iid);
7171 7172 7173 7174
    return ret;
}

static virNetworkPtr vboxNetworkLookupByName(virConnectPtr conn, const char *name) {
7175 7176 7177
    VBOX_OBJECT_HOST_CHECK(conn, virNetworkPtr, NULL);
    PRUnichar *nameUtf16                    = NULL;
    IHostNetworkInterface *networkInterface = NULL;
7178

7179
    VBOX_UTF8_TO_UTF16(name, &nameUtf16);
7180

7181
    host->vtbl->FindHostNetworkInterfaceByName(host, nameUtf16, &networkInterface);
7182

7183 7184
    if (networkInterface) {
        PRUint32 interfaceType = 0;
7185

7186
        networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7187

7188 7189
        if (interfaceType == HostNetworkInterfaceType_HostOnly) {
            unsigned char uuid[VIR_UUID_BUFLEN];
7190
            vboxIID iid = VBOX_IID_INITIALIZER;
7191

7192 7193
            networkInterface->vtbl->GetId(networkInterface, &iid.value);
            vboxIIDToUUID(&iid, uuid);
7194
            ret = virGetNetwork(conn, name, uuid);
7195
            VIR_DEBUG("Network Name: %s", name);
7196

7197 7198
            DEBUGIID("Network UUID", iid.value);
            vboxIIDUnalloc(&iid);
7199
        }
7200 7201

        VBOX_RELEASE(networkInterface);
7202 7203
    }

7204 7205 7206
    VBOX_UTF16_FREE(nameUtf16);
    VBOX_RELEASE(host);

7207 7208 7209
    return ret;
}

7210
static virNetworkPtr vboxNetworkDefineCreateXML(virConnectPtr conn, const char *xml, bool start) {
7211 7212 7213 7214
    VBOX_OBJECT_HOST_CHECK(conn, virNetworkPtr, NULL);
    PRUnichar *networkInterfaceNameUtf16    = NULL;
    char      *networkInterfaceNameUtf8     = NULL;
    IHostNetworkInterface *networkInterface = NULL;
7215
    nsresult rc;
7216

7217
    virNetworkDefPtr def = virNetworkDefParseString(xml);
7218 7219
    virNetworkIpDefPtr ipdef;
    virSocketAddr netmask;
7220

7221
    if (   (!def)
7222 7223 7224 7225 7226 7227 7228 7229 7230 7231 7232 7233 7234
        || (def->forwardType != VIR_NETWORK_FORWARD_NONE)
        || (def->nips == 0 || !def->ips))
        goto cleanup;

    /* Look for the first IPv4 IP address definition and use that.
     * If there weren't any IPv4 addresses, ignore the network (since it's
     * required below to have an IPv4 address)
    */
    ipdef = virNetworkDefGetIpByIndex(def, AF_INET, 0);
    if (!ipdef)
        goto cleanup;

    if (virNetworkIpDefNetmask(ipdef, &netmask) < 0)
7235
        goto cleanup;
7236

7237 7238 7239 7240 7241 7242
    /* the current limitation of hostonly network is that you can't
     * assign a name to it and it defaults to vboxnet*, for e.g:
     * vboxnet0, vboxnet1, etc. Also the UUID is assigned to it
     * automatically depending on the mac address and thus both
     * these paramters are ignored here for now.
     */
7243

7244
#if VBOX_API_VERSION == 2002
7245
    if (STREQ(def->name, "vboxnet0")) {
7246
        PRUint32 interfaceType = 0;
7247

7248 7249
        VBOX_UTF8_TO_UTF16(def->name, &networkInterfaceNameUtf16);
        host->vtbl->FindHostNetworkInterfaceByName(host, networkInterfaceNameUtf16, &networkInterface);
7250

7251 7252 7253 7254 7255 7256
        networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
        if (interfaceType != HostNetworkInterfaceType_HostOnly) {
            VBOX_RELEASE(networkInterface);
            networkInterface = NULL;
        }
    }
7257
#else /* VBOX_API_VERSION != 2002 */
7258 7259 7260 7261
    {
        IProgress *progress = NULL;
        host->vtbl->CreateHostOnlyNetworkInterface(host, &networkInterface,
                                                   &progress);
7262

7263 7264 7265 7266
        if (progress) {
            progress->vtbl->WaitForCompletion(progress, -1);
            VBOX_RELEASE(progress);
        }
7267
    }
7268
#endif /* VBOX_API_VERSION != 2002 */
7269

7270 7271 7272 7273
    if (networkInterface) {
        unsigned char uuid[VIR_UUID_BUFLEN];
        char      *networkNameUtf8  = NULL;
        PRUnichar *networkNameUtf16 = NULL;
7274
        vboxIID vboxnetiid = VBOX_IID_INITIALIZER;
7275 7276 7277 7278 7279 7280 7281 7282

        networkInterface->vtbl->GetName(networkInterface, &networkInterfaceNameUtf16);
        if (networkInterfaceNameUtf16) {
            VBOX_UTF16_TO_UTF8(networkInterfaceNameUtf16, &networkInterfaceNameUtf8);

            if (virAsprintf(&networkNameUtf8, "HostInterfaceNetworking-%s", networkInterfaceNameUtf8) < 0) {
                VBOX_RELEASE(host);
                VBOX_RELEASE(networkInterface);
7283
                virReportOOMError();
7284 7285 7286
                goto cleanup;
            }
        }
7287

7288
        VBOX_UTF8_TO_UTF16(networkNameUtf8 , &networkNameUtf16);
7289

7290 7291 7292
        /* Currently support only one dhcp server per network
         * with contigious address space from start to end
         */
7293 7294 7295
        if ((ipdef->nranges >= 1) &&
            VIR_SOCKET_HAS_ADDR(&ipdef->ranges[0].start) &&
            VIR_SOCKET_HAS_ADDR(&ipdef->ranges[0].end)) {
7296 7297 7298 7299 7300 7301 7302 7303 7304 7305
            IDHCPServer *dhcpServer = NULL;

            data->vboxObj->vtbl->FindDHCPServerByNetworkName(data->vboxObj,
                                                             networkNameUtf16,
                                                             &dhcpServer);
            if (!dhcpServer) {
                /* create a dhcp server */
                data->vboxObj->vtbl->CreateDHCPServer(data->vboxObj,
                                                      networkNameUtf16,
                                                      &dhcpServer);
7306
                VIR_DEBUG("couldn't find dhcp server so creating one");
7307 7308 7309 7310 7311 7312 7313 7314
            }
            if (dhcpServer) {
                PRUnichar *ipAddressUtf16     = NULL;
                PRUnichar *networkMaskUtf16   = NULL;
                PRUnichar *fromIPAddressUtf16 = NULL;
                PRUnichar *toIPAddressUtf16   = NULL;
                PRUnichar *trunkTypeUtf16     = NULL;

7315 7316 7317 7318
                ipAddressUtf16 = vboxSocketFormatAddrUtf16(data, &ipdef->address);
                networkMaskUtf16 = vboxSocketFormatAddrUtf16(data, &netmask);
                fromIPAddressUtf16 = vboxSocketFormatAddrUtf16(data, &ipdef->ranges[0].start);
                toIPAddressUtf16 = vboxSocketFormatAddrUtf16(data, &ipdef->ranges[0].end);
7319 7320 7321 7322 7323 7324 7325 7326 7327 7328

                if (ipAddressUtf16 == NULL || networkMaskUtf16 == NULL ||
                    fromIPAddressUtf16 == NULL || toIPAddressUtf16 == NULL) {
                    VBOX_UTF16_FREE(ipAddressUtf16);
                    VBOX_UTF16_FREE(networkMaskUtf16);
                    VBOX_UTF16_FREE(fromIPAddressUtf16);
                    VBOX_UTF16_FREE(toIPAddressUtf16);
                    VBOX_RELEASE(dhcpServer);
                    goto cleanup;
                }
7329 7330 7331 7332 7333 7334 7335 7336 7337 7338 7339 7340 7341 7342 7343 7344 7345 7346 7347 7348 7349 7350 7351 7352 7353

                VBOX_UTF8_TO_UTF16("netflt", &trunkTypeUtf16);

                dhcpServer->vtbl->SetEnabled(dhcpServer, PR_TRUE);

                dhcpServer->vtbl->SetConfiguration(dhcpServer,
                                                   ipAddressUtf16,
                                                   networkMaskUtf16,
                                                   fromIPAddressUtf16,
                                                   toIPAddressUtf16);

                if (start)
                    dhcpServer->vtbl->Start(dhcpServer,
                                            networkNameUtf16,
                                            networkInterfaceNameUtf16,
                                            trunkTypeUtf16);

                VBOX_UTF16_FREE(ipAddressUtf16);
                VBOX_UTF16_FREE(networkMaskUtf16);
                VBOX_UTF16_FREE(fromIPAddressUtf16);
                VBOX_UTF16_FREE(toIPAddressUtf16);
                VBOX_UTF16_FREE(trunkTypeUtf16);
                VBOX_RELEASE(dhcpServer);
            }
        }
7354

7355 7356
        if ((ipdef->nhosts >= 1) &&
            VIR_SOCKET_HAS_ADDR(&ipdef->hosts[0].ip)) {
7357 7358
            PRUnichar *ipAddressUtf16   = NULL;
            PRUnichar *networkMaskUtf16 = NULL;
7359

7360 7361
            ipAddressUtf16 = vboxSocketFormatAddrUtf16(data, &ipdef->hosts[0].ip);
            networkMaskUtf16 = vboxSocketFormatAddrUtf16(data, &netmask);
7362 7363 7364 7365 7366 7367

            if (ipAddressUtf16 == NULL || networkMaskUtf16 == NULL) {
                VBOX_UTF16_FREE(ipAddressUtf16);
                VBOX_UTF16_FREE(networkMaskUtf16);
                goto cleanup;
            }
7368

7369 7370 7371 7372 7373 7374 7375
            /* Current drawback is that since EnableStaticIpConfig() sets
             * IP and enables the interface so even if the dhcpserver is not
             * started the interface is still up and running
             */
            networkInterface->vtbl->EnableStaticIpConfig(networkInterface,
                                                         ipAddressUtf16,
                                                         networkMaskUtf16);
7376

7377 7378 7379 7380 7381 7382
            VBOX_UTF16_FREE(ipAddressUtf16);
            VBOX_UTF16_FREE(networkMaskUtf16);
        } else {
            networkInterface->vtbl->EnableDynamicIpConfig(networkInterface);
            networkInterface->vtbl->DhcpRediscover(networkInterface);
        }
7383

7384 7385 7386 7387 7388
        rc = networkInterface->vtbl->GetId(networkInterface, &vboxnetiid.value);
        if (NS_SUCCEEDED(rc)) {
            vboxIIDToUUID(&vboxnetiid, uuid);
            DEBUGIID("Real Network UUID", vboxnetiid.value);
            vboxIIDUnalloc(&vboxnetiid);
7389
            ret = virGetNetwork(conn, networkInterfaceNameUtf8, uuid);
7390
        }
7391 7392 7393 7394

        VIR_FREE(networkNameUtf8);
        VBOX_UTF16_FREE(networkNameUtf16);
        VBOX_RELEASE(networkInterface);
7395 7396
    }

7397 7398 7399 7400
    VBOX_UTF8_FREE(networkInterfaceNameUtf8);
    VBOX_UTF16_FREE(networkInterfaceNameUtf16);
    VBOX_RELEASE(host);

7401 7402 7403 7404 7405
cleanup:
    virNetworkDefFree(def);
    return ret;
}

7406 7407 7408 7409 7410 7411 7412 7413
static virNetworkPtr vboxNetworkCreateXML(virConnectPtr conn, const char *xml) {
    return vboxNetworkDefineCreateXML(conn, xml, true);
}

static virNetworkPtr vboxNetworkDefineXML(virConnectPtr conn, const char *xml) {
    return vboxNetworkDefineCreateXML(conn, xml, false);
}

7414
static int vboxNetworkUndefineDestroy(virNetworkPtr network, bool removeinterface) {
7415
    VBOX_OBJECT_HOST_CHECK(network->conn, int, -1);
7416
    char *networkNameUtf8 = NULL;
7417 7418
    PRUnichar *networkInterfaceNameUtf16    = NULL;
    IHostNetworkInterface *networkInterface = NULL;
7419 7420 7421 7422 7423 7424 7425 7426 7427 7428

    /* Current limitation of the function for VirtualBox 2.2.* is
     * that you can't delete the default hostonly adaptor namely:
     * vboxnet0 and thus all this functions does is remove the
     * dhcp server configuration, but the network can still be used
     * by giving the machine static IP and also it will still
     * show up in the net-list in virsh
     */

    if (virAsprintf(&networkNameUtf8, "HostInterfaceNetworking-%s", network->name) < 0) {
7429
        virReportOOMError();
7430 7431 7432
        goto cleanup;
    }

7433
    VBOX_UTF8_TO_UTF16(network->name, &networkInterfaceNameUtf16);
7434

7435
    host->vtbl->FindHostNetworkInterfaceByName(host, networkInterfaceNameUtf16, &networkInterface);
7436

7437 7438
    if (networkInterface) {
        PRUint32 interfaceType = 0;
7439

7440
        networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7441

7442 7443 7444
        if (interfaceType == HostNetworkInterfaceType_HostOnly) {
            PRUnichar *networkNameUtf16 = NULL;
            IDHCPServer *dhcpServer     = NULL;
7445

7446
#if VBOX_API_VERSION != 2002
7447 7448 7449
            if (removeinterface) {
                PRUnichar *iidUtf16 = NULL;
                IProgress *progress = NULL;
7450

7451
                networkInterface->vtbl->GetId(networkInterface, &iidUtf16);
7452

7453
                if (iidUtf16) {
7454
# if VBOX_API_VERSION == 3000
7455 7456 7457
                    IHostNetworkInterface *netInt = NULL;
                    host->vtbl->RemoveHostOnlyNetworkInterface(host, iidUtf16, &netInt, &progress);
                    VBOX_RELEASE(netInt);
7458
# else  /* VBOX_API_VERSION > 3000 */
7459
                    host->vtbl->RemoveHostOnlyNetworkInterface(host, iidUtf16, &progress);
7460
# endif /* VBOX_API_VERSION > 3000 */
7461 7462
                    VBOX_UTF16_FREE(iidUtf16);
                }
7463

7464 7465 7466 7467 7468
                if (progress) {
                    progress->vtbl->WaitForCompletion(progress, -1);
                    VBOX_RELEASE(progress);
                }
            }
7469 7470
#endif /* VBOX_API_VERSION != 2002 */

7471 7472 7473 7474 7475 7476 7477 7478 7479 7480 7481
            VBOX_UTF8_TO_UTF16(networkNameUtf8 , &networkNameUtf16);

            data->vboxObj->vtbl->FindDHCPServerByNetworkName(data->vboxObj,
                                                             networkNameUtf16,
                                                             &dhcpServer);
            if (dhcpServer) {
                dhcpServer->vtbl->SetEnabled(dhcpServer, PR_FALSE);
                dhcpServer->vtbl->Stop(dhcpServer);
                if (removeinterface)
                    data->vboxObj->vtbl->RemoveDHCPServer(data->vboxObj, dhcpServer);
                VBOX_RELEASE(dhcpServer);
7482 7483
            }

7484 7485
            VBOX_UTF16_FREE(networkNameUtf16);

7486
        }
7487
        VBOX_RELEASE(networkInterface);
7488 7489
    }

7490 7491 7492
    VBOX_UTF16_FREE(networkInterfaceNameUtf16);
    VBOX_RELEASE(host);

7493 7494 7495 7496 7497 7498 7499
    ret = 0;

cleanup:
    VIR_FREE(networkNameUtf8);
    return ret;
}

7500 7501 7502 7503
static int vboxNetworkUndefine(virNetworkPtr network) {
    return vboxNetworkUndefineDestroy(network, true);
}

7504
static int vboxNetworkCreate(virNetworkPtr network) {
7505
    VBOX_OBJECT_HOST_CHECK(network->conn, int, -1);
7506
    char *networkNameUtf8 = NULL;
7507 7508
    PRUnichar *networkInterfaceNameUtf16    = NULL;
    IHostNetworkInterface *networkInterface = NULL;
7509 7510 7511 7512 7513 7514 7515 7516 7517

    /* Current limitation of the function for VirtualBox 2.2.* is
     * that the default hostonly network "vboxnet0" is always active
     * and thus all this functions does is start the dhcp server,
     * but the network can still be used without starting the dhcp
     * server by giving the machine static IP
     */

    if (virAsprintf(&networkNameUtf8, "HostInterfaceNetworking-%s", network->name) < 0) {
7518
        virReportOOMError();
7519 7520 7521
        goto cleanup;
    }

7522
    VBOX_UTF8_TO_UTF16(network->name, &networkInterfaceNameUtf16);
7523

7524
    host->vtbl->FindHostNetworkInterfaceByName(host, networkInterfaceNameUtf16, &networkInterface);
7525

7526 7527
    if (networkInterface) {
        PRUint32 interfaceType = 0;
7528

7529
        networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7530

7531 7532 7533
        if (interfaceType == HostNetworkInterfaceType_HostOnly) {
            PRUnichar *networkNameUtf16 = NULL;
            IDHCPServer *dhcpServer     = NULL;
7534 7535


7536
            VBOX_UTF8_TO_UTF16(networkNameUtf8 , &networkNameUtf16);
7537

7538 7539 7540 7541 7542
            data->vboxObj->vtbl->FindDHCPServerByNetworkName(data->vboxObj,
                                                             networkNameUtf16,
                                                             &dhcpServer);
            if (dhcpServer) {
                PRUnichar *trunkTypeUtf16 = NULL;
7543

7544
                dhcpServer->vtbl->SetEnabled(dhcpServer, PR_TRUE);
7545

7546
                VBOX_UTF8_TO_UTF16("netflt", &trunkTypeUtf16);
7547

7548 7549 7550 7551
                dhcpServer->vtbl->Start(dhcpServer,
                                        networkNameUtf16,
                                        networkInterfaceNameUtf16,
                                        trunkTypeUtf16);
7552

7553 7554
                VBOX_UTF16_FREE(trunkTypeUtf16);
                VBOX_RELEASE(dhcpServer);
7555 7556
            }

7557
            VBOX_UTF16_FREE(networkNameUtf16);
7558
        }
7559 7560

        VBOX_RELEASE(networkInterface);
7561 7562
    }

7563 7564 7565
    VBOX_UTF16_FREE(networkInterfaceNameUtf16);
    VBOX_RELEASE(host);

7566 7567 7568 7569 7570 7571 7572 7573
    ret = 0;

cleanup:
    VIR_FREE(networkNameUtf8);
    return ret;
}

static int vboxNetworkDestroy(virNetworkPtr network) {
7574
    return vboxNetworkUndefineDestroy(network, false);
7575 7576
}

7577 7578
static char *vboxNetworkGetXMLDesc(virNetworkPtr network,
                                   unsigned int flags ATTRIBUTE_UNUSED) {
7579
    VBOX_OBJECT_HOST_CHECK(network->conn, char *, NULL);
7580
    virNetworkDefPtr def  = NULL;
7581
    virNetworkIpDefPtr ipdef = NULL;
7582
    char *networkNameUtf8 = NULL;
7583 7584
    PRUnichar *networkInterfaceNameUtf16    = NULL;
    IHostNetworkInterface *networkInterface = NULL;
7585 7586

    if (VIR_ALLOC(def) < 0) {
7587
        virReportOOMError();
7588 7589
        goto cleanup;
    }
7590 7591 7592 7593 7594 7595
    if (VIR_ALLOC(ipdef) < 0) {
        virReportOOMError();
        goto cleanup;
    }
    def->ips = ipdef;
    def->nips = 1;
7596 7597

    if (virAsprintf(&networkNameUtf8, "HostInterfaceNetworking-%s", network->name) < 0) {
7598
        virReportOOMError();
7599 7600 7601
        goto cleanup;
    }

7602
    VBOX_UTF8_TO_UTF16(network->name, &networkInterfaceNameUtf16);
7603

7604
    host->vtbl->FindHostNetworkInterfaceByName(host, networkInterfaceNameUtf16, &networkInterface);
7605

7606 7607
    if (networkInterface) {
        PRUint32 interfaceType = 0;
7608

7609
        networkInterface->vtbl->GetInterfaceType(networkInterface, &interfaceType);
7610

7611 7612 7613 7614 7615
        if (interfaceType == HostNetworkInterfaceType_HostOnly) {
            def->name = strdup(network->name);
            if (def->name != NULL) {
                PRUnichar *networkNameUtf16 = NULL;
                IDHCPServer *dhcpServer     = NULL;
7616
                vboxIID vboxnet0IID = VBOX_IID_INITIALIZER;
7617

7618 7619
                networkInterface->vtbl->GetId(networkInterface, &vboxnet0IID.value);
                vboxIIDToUUID(&vboxnet0IID, def->uuid);
7620

7621
                VBOX_UTF8_TO_UTF16(networkNameUtf8 , &networkNameUtf16);
7622

7623
                def->forwardType = VIR_NETWORK_FORWARD_NONE;
7624

7625 7626 7627 7628
                data->vboxObj->vtbl->FindDHCPServerByNetworkName(data->vboxObj,
                                                                 networkNameUtf16,
                                                                 &dhcpServer);
                if (dhcpServer) {
7629 7630
                    ipdef->nranges = 1;
                    if (VIR_ALLOC_N(ipdef->ranges, ipdef->nranges) >=0 ) {
7631 7632 7633 7634
                        PRUnichar *ipAddressUtf16     = NULL;
                        PRUnichar *networkMaskUtf16   = NULL;
                        PRUnichar *fromIPAddressUtf16 = NULL;
                        PRUnichar *toIPAddressUtf16   = NULL;
7635
                        bool errorOccurred = false;
7636

7637 7638 7639 7640 7641 7642 7643
                        dhcpServer->vtbl->GetIPAddress(dhcpServer, &ipAddressUtf16);
                        dhcpServer->vtbl->GetNetworkMask(dhcpServer, &networkMaskUtf16);
                        dhcpServer->vtbl->GetLowerIP(dhcpServer, &fromIPAddressUtf16);
                        dhcpServer->vtbl->GetUpperIP(dhcpServer, &toIPAddressUtf16);
                        /* Currently virtualbox supports only one dhcp server per network
                         * with contigious address space from start to end
                         */
7644
                        if (vboxSocketParseAddrUtf16(data, ipAddressUtf16,
7645
                                                     &ipdef->address) < 0 ||
7646
                            vboxSocketParseAddrUtf16(data, networkMaskUtf16,
7647
                                                     &ipdef->netmask) < 0 ||
7648
                            vboxSocketParseAddrUtf16(data, fromIPAddressUtf16,
7649
                                                     &ipdef->ranges[0].start) < 0 ||
7650
                            vboxSocketParseAddrUtf16(data, toIPAddressUtf16,
7651
                                                     &ipdef->ranges[0].end) < 0) {
7652 7653
                            errorOccurred = true;
                        }
7654 7655 7656 7657 7658

                        VBOX_UTF16_FREE(ipAddressUtf16);
                        VBOX_UTF16_FREE(networkMaskUtf16);
                        VBOX_UTF16_FREE(fromIPAddressUtf16);
                        VBOX_UTF16_FREE(toIPAddressUtf16);
7659 7660 7661 7662

                        if (errorOccurred) {
                            goto cleanup;
                        }
7663
                    } else {
7664
                        ipdef->nranges = 0;
7665
                        virReportOOMError();
7666
                    }
7667

7668 7669 7670 7671 7672 7673
                    ipdef->nhosts = 1;
                    if (VIR_ALLOC_N(ipdef->hosts, ipdef->nhosts) >=0 ) {
                        ipdef->hosts[0].name = strdup(network->name);
                        if (ipdef->hosts[0].name == NULL) {
                            VIR_FREE(ipdef->hosts);
                            ipdef->nhosts = 0;
7674
                            virReportOOMError();
7675
                        } else {
7676 7677
                            PRUnichar *macAddressUtf16 = NULL;
                            PRUnichar *ipAddressUtf16  = NULL;
7678
                            bool errorOccurred = false;
7679

7680
                            networkInterface->vtbl->GetHardwareAddress(networkInterface, &macAddressUtf16);
7681 7682
                            networkInterface->vtbl->GetIPAddress(networkInterface, &ipAddressUtf16);

7683
                            VBOX_UTF16_TO_UTF8(macAddressUtf16, &ipdef->hosts[0].mac);
7684 7685

                            if (vboxSocketParseAddrUtf16(data, ipAddressUtf16,
7686
                                                         &ipdef->hosts[0].ip) < 0) {
7687 7688
                                errorOccurred = true;
                            }
7689

7690 7691
                            VBOX_UTF16_FREE(macAddressUtf16);
                            VBOX_UTF16_FREE(ipAddressUtf16);
7692 7693 7694 7695

                            if (errorOccurred) {
                                goto cleanup;
                            }
7696 7697
                        }
                    } else {
7698
                        ipdef->nhosts = 0;
7699
                    }
7700 7701 7702 7703 7704

                    VBOX_RELEASE(dhcpServer);
                } else {
                    PRUnichar *networkMaskUtf16 = NULL;
                    PRUnichar *ipAddressUtf16   = NULL;
7705
                    bool errorOccurred = false;
7706 7707 7708 7709

                    networkInterface->vtbl->GetNetworkMask(networkInterface, &networkMaskUtf16);
                    networkInterface->vtbl->GetIPAddress(networkInterface, &ipAddressUtf16);

7710
                    if (vboxSocketParseAddrUtf16(data, networkMaskUtf16,
7711
                                                 &ipdef->netmask) < 0 ||
7712
                        vboxSocketParseAddrUtf16(data, ipAddressUtf16,
7713
                                                 &ipdef->address) < 0) {
7714 7715
                        errorOccurred = true;
                    }
7716 7717 7718

                    VBOX_UTF16_FREE(networkMaskUtf16);
                    VBOX_UTF16_FREE(ipAddressUtf16);
7719 7720 7721 7722

                    if (errorOccurred) {
                        goto cleanup;
                    }
7723 7724
                }

7725 7726
                DEBUGIID("Network UUID", vboxnet0IID.value);
                vboxIIDUnalloc(&vboxnet0IID);
7727 7728
                VBOX_UTF16_FREE(networkNameUtf16);
            } else {
7729
                virReportOOMError();
7730 7731
            }
        }
7732 7733

        VBOX_RELEASE(networkInterface);
7734 7735
    }

7736 7737 7738
    VBOX_UTF16_FREE(networkInterfaceNameUtf16);
    VBOX_RELEASE(host);

7739
    ret = virNetworkDefFormat(def);
7740 7741

cleanup:
7742
    virNetworkDefFree(def);
7743 7744 7745 7746
    VIR_FREE(networkNameUtf8);
    return ret;
}

7747 7748 7749 7750 7751 7752
/**
 * The Storage Functions here on
 */

static virDrvOpenStatus vboxStorageOpen (virConnectPtr conn,
                                         virConnectAuthPtr auth ATTRIBUTE_UNUSED,
7753
                                         unsigned int flags ATTRIBUTE_UNUSED) {
7754 7755 7756 7757 7758 7759 7760 7761 7762 7763
    vboxGlobalData *data = conn->privateData;

    if (STRNEQ(conn->driver->name, "VBOX"))
        goto cleanup;

    if ((data->pFuncs      == NULL) ||
        (data->vboxObj     == NULL) ||
        (data->vboxSession == NULL))
        goto cleanup;

7764
    VIR_DEBUG("vbox storage initialized");
7765 7766 7767 7768 7769 7770 7771 7772
    /* conn->storagePrivateData = some storage specific data */
    return VIR_DRV_OPEN_SUCCESS;

cleanup:
    return VIR_DRV_OPEN_DECLINED;
}

static int vboxStorageClose (virConnectPtr conn) {
7773
    VIR_DEBUG("vbox storage uninitialized");
7774 7775 7776 7777 7778 7779 7780 7781 7782 7783 7784 7785 7786
    conn->storagePrivateData = NULL;
    return 0;
}

static int vboxStorageNumOfPools(virConnectPtr conn ATTRIBUTE_UNUSED) {

    /** Currently only one pool supported, the default one
     * given by ISystemProperties::defaultHardDiskFolder()
     */

    return 1;
}

7787 7788
static int vboxStorageListPools(virConnectPtr conn ATTRIBUTE_UNUSED,
                                char **const names, int nnames) {
7789 7790 7791 7792 7793
    int numActive = 0;

    if (nnames == 1) {
        names[numActive] = strdup("default-pool");
        if (names[numActive] == NULL) {
7794
            virReportOOMError();
7795 7796 7797 7798 7799 7800 7801 7802 7803 7804 7805 7806 7807 7808 7809 7810 7811 7812 7813 7814 7815 7816 7817 7818 7819 7820 7821
        } else {
            numActive++;
        }
    }
    return numActive;
}

static virStoragePoolPtr vboxStoragePoolLookupByName(virConnectPtr conn, const char *name) {
    virStoragePoolPtr ret = NULL;

    /** Current limitation of the function: since
     * the default pool doesn't have UUID just assign
     * one till vbox can handle pools
     */
    if (STREQ("default-pool", name)) {
        unsigned char uuid[VIR_UUID_BUFLEN];
        const char *uuidstr = "1deff1ff-1481-464f-967f-a50fe8936cc4";

        virUUIDParse(uuidstr, uuid);

        ret = virGetStoragePool(conn, name, uuid);
    }

    return ret;
}

static int vboxStoragePoolNumOfVolumes(virStoragePoolPtr pool) {
7822
    VBOX_OBJECT_CHECK(pool->conn, int, -1);
7823
    vboxArray hardDisks = VBOX_ARRAY_INITIALIZER;
7824 7825 7826 7827
    PRUint32 hardDiskAccessible = 0;
    nsresult rc;
    int i;

7828
    rc = vboxArrayGet(&hardDisks, data->vboxObj, data->vboxObj->vtbl->GetHardDisks);
7829
    if (NS_SUCCEEDED(rc)) {
7830 7831
        for (i = 0; i < hardDisks.count; ++i) {
            IHardDisk *hardDisk = hardDisks.items[i];
7832 7833
            if (hardDisk) {
                PRUint32 hddstate;
7834

7835 7836 7837
                VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
                if (hddstate != MediaState_Inaccessible)
                    hardDiskAccessible++;
7838 7839
            }
        }
7840 7841 7842 7843

        vboxArrayRelease(&hardDisks);

        ret = hardDiskAccessible;
7844
    } else {
7845
        ret = -1;
7846 7847
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not get number of volumes in the pool: %s, rc=%08x"),
7848
                  pool->name, (unsigned)rc);
7849 7850
    }

7851
    return ret;
7852 7853 7854
}

static int vboxStoragePoolListVolumes(virStoragePoolPtr pool, char **const names, int nnames) {
7855
    VBOX_OBJECT_CHECK(pool->conn, int, -1);
7856
    vboxArray hardDisks = VBOX_ARRAY_INITIALIZER;
7857 7858 7859 7860
    PRUint32 numActive     = 0;
    nsresult rc;
    int i;

7861
    rc = vboxArrayGet(&hardDisks, data->vboxObj, data->vboxObj->vtbl->GetHardDisks);
7862
    if (NS_SUCCEEDED(rc)) {
7863 7864
        for (i = 0; i < hardDisks.count && numActive < nnames; ++i) {
            IHardDisk *hardDisk = hardDisks.items[i];
7865

7866 7867 7868 7869
            if (hardDisk) {
                PRUint32 hddstate;
                char      *nameUtf8  = NULL;
                PRUnichar *nameUtf16 = NULL;
7870

7871 7872 7873
                VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
                if (hddstate != MediaState_Inaccessible) {
                    VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetName, &nameUtf16);
7874

7875 7876
                    VBOX_UTF16_TO_UTF8(nameUtf16, &nameUtf8);
                    VBOX_UTF16_FREE(nameUtf16);
7877

7878
                    if (nameUtf8) {
7879
                        VIR_DEBUG("nnames[%d]: %s", numActive, nameUtf8);
7880 7881
                        names[numActive] = strdup(nameUtf8);
                        if (names[numActive] == NULL) {
7882
                            virReportOOMError();
7883 7884
                        } else {
                            numActive++;
7885
                        }
7886 7887

                        VBOX_UTF8_FREE(nameUtf8);
7888 7889 7890 7891
                    }
                }
            }
        }
7892 7893 7894 7895

        vboxArrayRelease(&hardDisks);

        ret = numActive;
7896
    } else {
7897
        ret = -1;
7898 7899
        vboxError(VIR_ERR_INTERNAL_ERROR,
                  _("could not get the volume list in the pool: %s, rc=%08x"),
7900
                  pool->name, (unsigned)rc);
7901 7902
    }

7903
    return ret;
7904 7905 7906
}

static virStorageVolPtr vboxStorageVolLookupByName(virStoragePoolPtr pool, const char *name) {
7907
    VBOX_OBJECT_CHECK(pool->conn, virStorageVolPtr, NULL);
7908
    vboxArray hardDisks = VBOX_ARRAY_INITIALIZER;
7909 7910 7911
    nsresult rc;
    int i;

7912 7913
    if(!name)
        return ret;
7914

7915
    rc = vboxArrayGet(&hardDisks, data->vboxObj, data->vboxObj->vtbl->GetHardDisks);
7916
    if (NS_SUCCEEDED(rc)) {
7917 7918
        for (i = 0; i < hardDisks.count; ++i) {
            IHardDisk *hardDisk = hardDisks.items[i];
7919

7920 7921 7922 7923
            if (hardDisk) {
                PRUint32 hddstate;
                char      *nameUtf8  = NULL;
                PRUnichar *nameUtf16 = NULL;
7924

7925 7926 7927
                VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
                if (hddstate != MediaState_Inaccessible) {
                    VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetName, &nameUtf16);
7928

7929 7930 7931 7932
                    if (nameUtf16) {
                        VBOX_UTF16_TO_UTF8(nameUtf16, &nameUtf8);
                        VBOX_UTF16_FREE(nameUtf16);
                    }
7933

7934
                    if (nameUtf8 && STREQ(nameUtf8, name)) {
7935 7936 7937
                        vboxIID hddIID = VBOX_IID_INITIALIZER;
                        unsigned char uuid[VIR_UUID_BUFLEN];
                        char key[VIR_UUID_STRING_BUFLEN] = "";
7938

7939 7940 7941 7942
                        rc = VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetId, &hddIID.value);
                        if (NS_SUCCEEDED(rc)) {
                            vboxIIDToUUID(&hddIID, uuid);
                            virUUIDFormat(uuid, key);
7943

7944
                            ret = virGetStorageVol(pool->conn, pool->name, name, key);
7945

7946 7947 7948 7949
                            VIR_DEBUG("virStorageVolPtr: %p", ret);
                            VIR_DEBUG("Storage Volume Name: %s", name);
                            VIR_DEBUG("Storage Volume key : %s", key);
                            VIR_DEBUG("Storage Volume Pool: %s", pool->name);
7950 7951
                        }

7952
                        vboxIIDUnalloc(&hddIID);
7953 7954
                        VBOX_UTF8_FREE(nameUtf8);
                        break;
7955
                    }
7956 7957 7958

                    if (nameUtf8)
                        VBOX_UTF8_FREE(nameUtf8);
7959 7960 7961
                }
            }
        }
7962

7963
        vboxArrayRelease(&hardDisks);
7964 7965 7966 7967 7968 7969
    }

    return ret;
}

static virStorageVolPtr vboxStorageVolLookupByKey(virConnectPtr conn, const char *key) {
7970
    VBOX_OBJECT_CHECK(conn, virStorageVolPtr, NULL);
7971 7972
    vboxIID hddIID = VBOX_IID_INITIALIZER;
    unsigned char uuid[VIR_UUID_BUFLEN];
7973 7974 7975
    IHardDisk *hardDisk  = NULL;
    nsresult rc;

7976 7977 7978
    if (!key)
        return ret;

7979 7980 7981 7982
    if (virUUIDParse(key, uuid) < 0) {
        vboxError(VIR_ERR_INVALID_ARG,
                  _("Could not parse UUID from '%s'"), key);
        return NULL;
7983 7984
    }

7985
    vboxIIDFromUUID(&hddIID, uuid);
7986
#if VBOX_API_VERSION < 4000
7987
    rc = data->vboxObj->vtbl->GetHardDisk(data->vboxObj, hddIID.value, &hardDisk);
7988 7989 7990 7991
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->FindMedium(data->vboxObj, hddIID.value,
                                         DeviceType_HardDisk, &hardDisk);
#endif /* VBOX_API_VERSION >= 4000 */
7992 7993
    if (NS_SUCCEEDED(rc)) {
        PRUint32 hddstate;
7994

7995 7996 7997 7998
        VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
        if (hddstate != MediaState_Inaccessible) {
            PRUnichar *hddNameUtf16 = NULL;
            char      *hddNameUtf8  = NULL;
7999

8000 8001
            VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetName, &hddNameUtf16);
            VBOX_UTF16_TO_UTF8(hddNameUtf16, &hddNameUtf8);
8002

8003 8004 8005
            if (hddNameUtf8) {
                if (vboxStorageNumOfPools(conn) == 1) {
                    ret = virGetStorageVol(conn, "default-pool", hddNameUtf8, key);
8006
                    VIR_DEBUG("Storage Volume Pool: %s", "default-pool");
8007 8008 8009 8010
                } else {
                    /* TODO: currently only one default pool and thus
                     * nothing here, change it when pools are supported
                     */
8011 8012
                }

8013 8014
                VIR_DEBUG("Storage Volume Name: %s", key);
                VIR_DEBUG("Storage Volume key : %s", hddNameUtf8);
8015 8016 8017

                VBOX_UTF8_FREE(hddNameUtf8);
                VBOX_UTF16_FREE(hddNameUtf16);
8018 8019
            }
        }
8020 8021

        VBOX_MEDIUM_RELEASE(hardDisk);
8022 8023
    }

8024
    vboxIIDUnalloc(&hddIID);
8025 8026 8027 8028
    return ret;
}

static virStorageVolPtr vboxStorageVolLookupByPath(virConnectPtr conn, const char *path) {
8029
    VBOX_OBJECT_CHECK(conn, virStorageVolPtr, NULL);
8030 8031 8032 8033
    PRUnichar *hddPathUtf16 = NULL;
    IHardDisk *hardDisk     = NULL;
    nsresult rc;

8034 8035
    if (!path)
        return ret;
8036

8037
    VBOX_UTF8_TO_UTF16(path, &hddPathUtf16);
8038

8039 8040
    if (!hddPathUtf16)
        return ret;
8041

8042
#if VBOX_API_VERSION < 4000
8043
    rc = data->vboxObj->vtbl->FindHardDisk(data->vboxObj, hddPathUtf16, &hardDisk);
8044 8045 8046 8047
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->FindMedium(data->vboxObj, hddPathUtf16,
                                         DeviceType_HardDisk, &hardDisk);
#endif /* VBOX_API_VERSION >= 4000 */
8048 8049
    if (NS_SUCCEEDED(rc)) {
        PRUint32 hddstate;
8050

8051 8052 8053 8054
        VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
        if (hddstate != MediaState_Inaccessible) {
            PRUnichar *hddNameUtf16 = NULL;
            char      *hddNameUtf8  = NULL;
8055

8056
            VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetName, &hddNameUtf16);
8057

8058 8059 8060 8061
            if (hddNameUtf16) {
                VBOX_UTF16_TO_UTF8(hddNameUtf16, &hddNameUtf8);
                VBOX_UTF16_FREE(hddNameUtf16);
            }
8062

8063 8064 8065 8066
            if (hddNameUtf8) {
                vboxIID hddIID = VBOX_IID_INITIALIZER;
                unsigned char uuid[VIR_UUID_BUFLEN];
                char key[VIR_UUID_STRING_BUFLEN] = "";
8067

8068 8069 8070 8071
                rc = VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetId, &hddIID.value);
                if (NS_SUCCEEDED(rc)) {
                    vboxIIDToUUID(&hddIID, uuid);
                    virUUIDFormat(uuid, key);
8072

8073 8074 8075 8076 8077
                    /* TODO: currently only one default pool and thus
                     * the check below, change it when pools are supported
                     */
                    if (vboxStorageNumOfPools(conn) == 1)
                        ret = virGetStorageVol(conn, "default-pool", hddNameUtf8, key);
8078

8079 8080 8081
                    VIR_DEBUG("Storage Volume Pool: %s", "default-pool");
                    VIR_DEBUG("Storage Volume Name: %s", hddNameUtf8);
                    VIR_DEBUG("Storage Volume key : %s", key);
8082
                }
8083

8084
                vboxIIDUnalloc(&hddIID);
8085 8086
            }

8087 8088
            if (hddNameUtf8)
                VBOX_UTF8_FREE(hddNameUtf8);
8089
        }
8090 8091

        VBOX_MEDIUM_RELEASE(hardDisk);
8092 8093
    }

8094 8095
    VBOX_UTF16_FREE(hddPathUtf16);

8096 8097 8098 8099 8100 8101
    return ret;
}

static virStorageVolPtr vboxStorageVolCreateXML(virStoragePoolPtr pool,
                                                const char *xml,
                                                unsigned int flags ATTRIBUTE_UNUSED) {
8102
    VBOX_OBJECT_CHECK(pool->conn, virStorageVolPtr, NULL);
8103
    virStorageVolDefPtr  def  = NULL;
8104 8105
    PRUnichar *hddFormatUtf16 = NULL;
    PRUnichar *hddNameUtf16   = NULL;
8106 8107 8108 8109 8110 8111 8112 8113 8114 8115 8116
    virStoragePoolDef poolDef;
    nsresult rc;

    /* since there is currently one default pool now
     * and virStorageVolDefFormat() just checks it type
     * so just assign it for now, change the behaviour
     * when vbox supports pools.
     */
    memset(&poolDef, 0, sizeof(poolDef));
    poolDef.type = VIR_STORAGE_POOL_DIR;

8117
    if ((def = virStorageVolDefParseString(&poolDef, xml)) == NULL)
8118 8119
        goto cleanup;

8120 8121 8122
    if (   !def->name
        || (def->type != VIR_STORAGE_VOL_FILE))
        goto cleanup;
8123

8124 8125 8126 8127 8128 8129
    /* TODO: for now only the vmdk, vpc and vdi type harddisk
     * variants can be created, also since there is no vdi
     * type in enum virStorageFileFormat {} the default
     * will be to create vdi if nothing is specified in
     * def->target.format
     */
8130

8131 8132 8133 8134 8135 8136 8137
    if (def->target.format == VIR_STORAGE_FILE_VMDK) {
        VBOX_UTF8_TO_UTF16("VMDK", &hddFormatUtf16);
    } else if (def->target.format == VIR_STORAGE_FILE_VPC) {
        VBOX_UTF8_TO_UTF16("VHD", &hddFormatUtf16);
    } else {
        VBOX_UTF8_TO_UTF16("VDI", &hddFormatUtf16);
    }
8138

8139
    VBOX_UTF8_TO_UTF16(def->name, &hddNameUtf16);
8140

8141 8142
    if (hddFormatUtf16 && hddNameUtf16) {
        IHardDisk *hardDisk = NULL;
8143

8144 8145 8146
        rc = data->vboxObj->vtbl->CreateHardDisk(data->vboxObj, hddFormatUtf16, hddNameUtf16, &hardDisk);
        if (NS_SUCCEEDED(rc)) {
            IProgress *progress    = NULL;
8147
            PRUint64   logicalSize = VIR_DIV_UP(def->capacity, 1024 * 1024);
8148
            PRUint32   variant     = HardDiskVariant_Standard;
8149

8150 8151
            if (def->capacity == def->allocation)
                variant = HardDiskVariant_Fixed;
8152

8153 8154
            rc = hardDisk->vtbl->CreateBaseStorage(hardDisk, logicalSize, variant, &progress);
            if (NS_SUCCEEDED(rc) && progress) {
8155
#if VBOX_API_VERSION == 2002
8156
                nsresult resultCode;
8157
#else
8158
                PRInt32  resultCode;
8159 8160
#endif

8161 8162
                progress->vtbl->WaitForCompletion(progress, -1);
                progress->vtbl->GetResultCode(progress, &resultCode);
8163

8164
                if (NS_SUCCEEDED(resultCode)) {
8165 8166 8167
                    vboxIID hddIID = VBOX_IID_INITIALIZER;
                    unsigned char uuid[VIR_UUID_BUFLEN];
                    char key[VIR_UUID_STRING_BUFLEN] = "";
8168

8169
                    rc = VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetId, &hddIID.value);
8170
                    if (NS_SUCCEEDED(rc)) {
8171 8172
                        vboxIIDToUUID(&hddIID, uuid);
                        virUUIDFormat(uuid, key);
8173

8174
                        ret = virGetStorageVol(pool->conn, pool->name, def->name, key);
8175
                    }
8176 8177

                    vboxIIDUnalloc(&hddIID);
8178 8179
                }

8180
                VBOX_RELEASE(progress);
8181
            }
8182
        }
8183 8184
    }

8185 8186 8187
    VBOX_UTF16_FREE(hddFormatUtf16);
    VBOX_UTF16_FREE(hddNameUtf16);

8188 8189 8190 8191 8192 8193 8194
cleanup:
    virStorageVolDefFree(def);
    return ret;
}

static int vboxStorageVolDelete(virStorageVolPtr vol,
                                unsigned int flags ATTRIBUTE_UNUSED) {
8195
    VBOX_OBJECT_CHECK(vol->conn, int, -1);
8196 8197
    vboxIID hddIID = VBOX_IID_INITIALIZER;
    unsigned char uuid[VIR_UUID_BUFLEN];
8198 8199 8200
    IHardDisk *hardDisk  = NULL;
    int deregister = 0;
    nsresult rc;
8201 8202
    int i = 0;
    int j = 0;
8203

8204 8205 8206 8207 8208
    if (virUUIDParse(vol->key, uuid) < 0) {
        vboxError(VIR_ERR_INVALID_ARG,
                  _("Could not parse UUID from '%s'"), vol->key);
        return -1;
    }
8209

8210
    vboxIIDFromUUID(&hddIID, uuid);
8211
#if VBOX_API_VERSION < 4000
8212
    rc = data->vboxObj->vtbl->GetHardDisk(data->vboxObj, hddIID.value, &hardDisk);
8213 8214 8215 8216
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->FindMedium(data->vboxObj, hddIID.value,
                                         DeviceType_HardDisk, &hardDisk);
#endif /* VBOX_API_VERSION >= 4000 */
8217 8218
    if (NS_SUCCEEDED(rc)) {
        PRUint32 hddstate;
8219

8220 8221 8222
        VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
        if (hddstate != MediaState_Inaccessible) {
            PRUint32  machineIdsSize = 0;
8223 8224 8225 8226 8227 8228 8229
            vboxArray machineIds = VBOX_ARRAY_INITIALIZER;

#if VBOX_API_VERSION < 3001
            vboxArrayGet(&machineIds, hardDisk, hardDisk->vtbl->imedium.GetMachineIds);
#else  /* VBOX_API_VERSION >= 3001 */
            vboxArrayGet(&machineIds, hardDisk, hardDisk->vtbl->GetMachineIds);
#endif /* VBOX_API_VERSION >= 3001 */
8230

8231 8232 8233 8234 8235 8236 8237 8238 8239 8240 8241 8242
#if VBOX_API_VERSION == 2002 && defined WIN32
            /* VirtualBox 2.2 on Windows represents IIDs as GUIDs and the
             * machineIds array contains direct instances of the GUID struct
             * instead of pointers to the actual struct instances. But there
             * is no 128bit width simple item type for a SafeArray to fit a
             * GUID in. The largest simple type it 64bit width and VirtualBox
             * uses two of this 64bit items to represents one GUID. Therefore,
             * we devide the size of the SafeArray by two, to compensate for
             * this workaround in VirtualBox */
            machineIds.count /= 2;
#endif /* VBOX_API_VERSION >= 2002 */

8243
            machineIdsSize = machineIds.count;
8244

8245
            for (i = 0; i < machineIds.count; i++) {
8246
                IMachine *machine = NULL;
8247 8248 8249
                vboxIID machineId = VBOX_IID_INITIALIZER;

                vboxIIDFromArrayItem(&machineId, &machineIds, i);
8250

8251 8252 8253 8254 8255 8256 8257 8258 8259 8260 8261
#if VBOX_API_VERSION >= 4000
                rc = VBOX_OBJECT_GET_MACHINE(machineId.value, &machine);
                if (NS_FAILED(rc)) {
                    vboxError(VIR_ERR_NO_DOMAIN, "%s",
                              _("no domain with matching uuid"));
                    break;
                }
#endif

                rc = VBOX_SESSION_OPEN(machineId.value, machine);

8262
                if (NS_SUCCEEDED(rc)) {
8263

8264 8265
                    rc = data->vboxSession->vtbl->GetMachine(data->vboxSession, &machine);
                    if (NS_SUCCEEDED(rc)) {
8266
                        vboxArray hddAttachments = VBOX_ARRAY_INITIALIZER;
8267 8268

#if VBOX_API_VERSION < 3001
8269 8270
                        vboxArrayGet(&hddAttachments, machine,
                                     machine->vtbl->GetHardDiskAttachments);
8271
#else  /* VBOX_API_VERSION >= 3001 */
8272 8273
                        vboxArrayGet(&hddAttachments, machine,
                                     machine->vtbl->GetMediumAttachments);
8274
#endif /* VBOX_API_VERSION >= 3001 */
8275 8276
                        for (j = 0; j < hddAttachments.count; j++) {
                            IHardDiskAttachment *hddAttachment = hddAttachments.items[j];
8277 8278 8279 8280 8281 8282 8283 8284 8285 8286

                            if (hddAttachment) {
                                IHardDisk *hdd = NULL;

#if VBOX_API_VERSION < 3001
                                rc = hddAttachment->vtbl->GetHardDisk(hddAttachment, &hdd);
#else  /* VBOX_API_VERSION >= 3001 */
                                rc = hddAttachment->vtbl->GetMedium(hddAttachment, &hdd);
#endif /* VBOX_API_VERSION >= 3001 */
                                if (NS_SUCCEEDED(rc) && hdd) {
8287
                                    vboxIID iid = VBOX_IID_INITIALIZER;
8288

8289 8290
                                    rc = VBOX_MEDIUM_FUNC_ARG1(hdd, GetId, &iid.value);
                                    if (NS_SUCCEEDED(rc)) {
8291

8292 8293
                                            DEBUGIID("HardDisk (to delete) UUID", hddIID.value);
                                            DEBUGIID("HardDisk (currently processing) UUID", iid.value);
8294

8295
                                        if (vboxIIDIsEqual(&hddIID, &iid)) {
8296 8297 8298 8299
                                            PRUnichar *controller = NULL;
                                            PRInt32    port       = 0;
                                            PRInt32    device     = 0;

8300
                                            DEBUGIID("Found HardDisk to delete, UUID", hddIID.value);
8301 8302 8303 8304 8305 8306 8307 8308 8309 8310 8311 8312

                                            hddAttachment->vtbl->GetController(hddAttachment, &controller);
                                            hddAttachment->vtbl->GetPort(hddAttachment, &port);
                                            hddAttachment->vtbl->GetDevice(hddAttachment, &device);

#if VBOX_API_VERSION < 3001
                                            rc = machine->vtbl->DetachHardDisk(machine, controller, port, device);
#else  /* VBOX_API_VERSION >= 3001 */
                                            rc = machine->vtbl->DetachDevice(machine, controller, port, device);
#endif /* VBOX_API_VERSION >= 3001 */
                                            if (NS_SUCCEEDED(rc)) {
                                                rc = machine->vtbl->SaveSettings(machine);
8313
                                                VIR_DEBUG("saving machine settings");
8314
                                            }
8315

8316 8317
                                            if (NS_SUCCEEDED(rc)) {
                                                deregister++;
8318
                                                VIR_DEBUG("deregistering hdd:%d", deregister);
8319
                                            }
8320 8321 8322

                                            if (controller)
                                                VBOX_UTF16_FREE(controller);
8323
                                        }
8324
                                        vboxIIDUnalloc(&iid);
8325
                                    }
8326
                                    VBOX_MEDIUM_RELEASE(hdd);
8327 8328 8329
                                }
                            }
                        }
8330
                        vboxArrayRelease(&hddAttachments);
8331
                        VBOX_RELEASE(machine);
8332
                    }
8333
                    VBOX_SESSION_CLOSE();
8334
                }
8335 8336

                vboxIIDUnalloc(&machineId);
8337
            }
8338

8339
            vboxArrayUnalloc(&machineIds);
8340

8341 8342
            if (machineIdsSize == 0 || machineIdsSize == deregister) {
                IProgress *progress = NULL;
8343

8344
                rc = hardDisk->vtbl->DeleteStorage(hardDisk, &progress);
8345

8346 8347 8348
                if (NS_SUCCEEDED(rc) && progress) {
                    progress->vtbl->WaitForCompletion(progress, -1);
                    VBOX_RELEASE(progress);
8349
                    DEBUGIID("HardDisk deleted, UUID", hddIID.value);
8350
                    ret = 0;
8351 8352 8353
                }
            }
        }
8354 8355

        VBOX_MEDIUM_RELEASE(hardDisk);
8356 8357
    }

8358
    vboxIIDUnalloc(&hddIID);
8359

8360 8361 8362 8363
    return ret;
}

static int vboxStorageVolGetInfo(virStorageVolPtr vol, virStorageVolInfoPtr info) {
8364
    VBOX_OBJECT_CHECK(vol->conn, int, -1);
8365
    IHardDisk *hardDisk  = NULL;
8366 8367
    unsigned char uuid[VIR_UUID_BUFLEN];
    vboxIID hddIID = VBOX_IID_INITIALIZER;
8368 8369
    nsresult rc;

8370
    if (!info)
8371
        return ret;
8372

8373 8374 8375
    if (virUUIDParse(vol->key, uuid) < 0) {
        vboxError(VIR_ERR_INVALID_ARG,
                  _("Could not parse UUID from '%s'"), vol->key);
8376
        return ret;
8377
    }
8378

8379
    vboxIIDFromUUID(&hddIID, uuid);
8380
#if VBOX_API_VERSION < 4000
8381
    rc = data->vboxObj->vtbl->GetHardDisk(data->vboxObj, hddIID.value, &hardDisk);
8382 8383 8384 8385
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->FindMedium(data->vboxObj, hddIID.value,
                                         DeviceType_HardDisk, &hardDisk);
#endif /* VBOX_API_VERSION >= 4000 */
8386 8387
    if (NS_SUCCEEDED(rc)) {
        PRUint32 hddstate;
8388

8389 8390
        VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
        if (hddstate != MediaState_Inaccessible) {
8391
#if VBOX_API_VERSION < 4000
8392 8393
            PRUint64 hddLogicalSize;
            PRUint64 hddActualSize;
8394 8395 8396 8397
#else /* VBOX_API_VERSION >= 4000 */
            PRInt64 hddLogicalSize;
            PRInt64 hddActualSize;
#endif /* VBOX_API_VERSION >= 4000 */
8398

8399
            info->type = VIR_STORAGE_VOL_FILE;
8400

8401
            hardDisk->vtbl->GetLogicalSize(hardDisk, &hddLogicalSize);
8402
#if VBOX_API_VERSION < 4000
8403
            info->capacity = hddLogicalSize * 1024 * 1024; /* MB => Bytes */
8404 8405 8406
#else /* VBOX_API_VERSION >= 4000 */
            info->capacity = hddLogicalSize;
#endif /* VBOX_API_VERSION >= 4000 */
8407

8408 8409
            VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetSize, &hddActualSize);
            info->allocation = hddActualSize;
8410

8411
            ret = 0;
8412

8413 8414 8415 8416
            VIR_DEBUG("Storage Volume Name: %s", vol->name);
            VIR_DEBUG("Storage Volume Type: %s", info->type == VIR_STORAGE_VOL_BLOCK ? "Block" : "File");
            VIR_DEBUG("Storage Volume Capacity: %llu", info->capacity);
            VIR_DEBUG("Storage Volume Allocation: %llu", info->allocation);
8417
        }
8418 8419

        VBOX_MEDIUM_RELEASE(hardDisk);
8420 8421
    }

8422
    vboxIIDUnalloc(&hddIID);
8423

8424 8425 8426 8427
    return ret;
}

static char *vboxStorageVolGetXMLDesc(virStorageVolPtr vol, unsigned int flags ATTRIBUTE_UNUSED) {
8428
    VBOX_OBJECT_CHECK(vol->conn, char *, NULL);
8429
    IHardDisk *hardDisk  = NULL;
8430 8431
    unsigned char uuid[VIR_UUID_BUFLEN];
    vboxIID hddIID = VBOX_IID_INITIALIZER;
8432 8433 8434 8435 8436 8437 8438 8439
    virStoragePoolDef pool;
    virStorageVolDef def;
    int defOk = 0;
    nsresult rc;

    memset(&pool, 0, sizeof(pool));
    memset(&def, 0, sizeof(def));

8440 8441 8442
    if (virUUIDParse(vol->key, uuid) < 0) {
        vboxError(VIR_ERR_INVALID_ARG,
                  _("Could not parse UUID from '%s'"), vol->key);
8443
        return ret;
8444
    }
8445

8446
    vboxIIDFromUUID(&hddIID, uuid);
8447
#if VBOX_API_VERSION < 4000
8448
    rc = data->vboxObj->vtbl->GetHardDisk(data->vboxObj, hddIID.value, &hardDisk);
8449 8450 8451 8452
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->FindMedium(data->vboxObj, hddIID.value,
                                         DeviceType_HardDisk, &hardDisk);
#endif /* VBOX_API_VERSION >= 4000 */
8453 8454 8455 8456 8457 8458
    if (NS_SUCCEEDED(rc)) {
        PRUint32 hddstate;

        VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
        if (NS_SUCCEEDED(rc) && hddstate != MediaState_Inaccessible) {
            PRUnichar *hddFormatUtf16 = NULL;
8459
#if VBOX_API_VERSION < 4000
8460 8461
            PRUint64 hddLogicalSize;
            PRUint64 hddActualSize;
8462 8463 8464 8465
#else /* VBOX_API_VERSION >= 4000 */
            PRInt64 hddLogicalSize;
            PRInt64 hddActualSize;
#endif /* VBOX_API_VERSION >= 4000 */
8466 8467 8468 8469 8470 8471 8472 8473 8474 8475 8476

            /* since there is currently one default pool now
             * and virStorageVolDefFormat() just checks it type
             * so just assign it for now, change the behaviour
             * when vbox supports pools.
             */
            pool.type = VIR_STORAGE_POOL_DIR;
            def.type = VIR_STORAGE_VOL_FILE;
            defOk = 1;

            rc = hardDisk->vtbl->GetLogicalSize(hardDisk, &hddLogicalSize);
8477 8478
            if (NS_SUCCEEDED(rc) && defOk) {
#if VBOX_API_VERSION < 4000
8479
                def.capacity = hddLogicalSize * 1024 * 1024; /* MB => Bytes */
8480 8481 8482 8483
#else /* VBOX_API_VERSION >= 4000 */
                def.capacity = hddLogicalSize;
#endif /* VBOX_API_VERSION >= 4000 */
            } else
8484 8485 8486 8487 8488 8489 8490 8491 8492 8493 8494 8495 8496 8497 8498 8499 8500 8501 8502 8503 8504 8505 8506
                defOk = 0;

            rc = VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetSize, &hddActualSize);
            if (NS_SUCCEEDED(rc) && defOk)
                def.allocation = hddActualSize;
            else
                defOk = 0;

            def.name = strdup(vol->name);
            if (!(def.name && defOk))
                defOk = 0;

            def.key = strdup(vol->key);
            if (!(def.key && defOk))
                defOk = 0;

            rc = hardDisk->vtbl->GetFormat(hardDisk, &hddFormatUtf16);
            if (NS_SUCCEEDED(rc) && defOk) {
                char *hddFormatUtf8 = NULL;

                VBOX_UTF16_TO_UTF8(hddFormatUtf16, &hddFormatUtf8);
                if (hddFormatUtf8) {

8507
                    VIR_DEBUG("Storage Volume Format: %s", hddFormatUtf8);
8508 8509 8510 8511 8512

                    if (STRCASEEQ("vmdk", hddFormatUtf8))
                        def.target.format = VIR_STORAGE_FILE_VMDK;
                    else if (STRCASEEQ("vhd", hddFormatUtf8))
                        def.target.format = VIR_STORAGE_FILE_VPC;
8513
                    else
8514
                        def.target.format = VIR_STORAGE_FILE_RAW;
8515

8516 8517 8518
                    /* TODO: need to add vdi to enum virStorageFileFormat {}
                     * and then add it here
                     */
8519

8520
                    VBOX_UTF8_FREE(hddFormatUtf8);
8521 8522
                }

8523 8524 8525
                VBOX_UTF16_FREE(hddFormatUtf16);
            } else {
                defOk = 0;
8526 8527
            }
        }
8528 8529

        VBOX_MEDIUM_RELEASE(hardDisk);
8530 8531
    }

8532
    vboxIIDUnalloc(&hddIID);
8533

8534
    if (defOk)
8535
        ret = virStorageVolDefFormat(&pool, &def);
8536 8537 8538 8539 8540

    return ret;
}

static char *vboxStorageVolGetPath(virStorageVolPtr vol) {
8541
    VBOX_OBJECT_CHECK(vol->conn, char *, NULL);
8542
    IHardDisk *hardDisk  = NULL;
8543 8544
    unsigned char uuid[VIR_UUID_BUFLEN];
    vboxIID hddIID = VBOX_IID_INITIALIZER;
8545 8546
    nsresult rc;

8547 8548 8549
    if (virUUIDParse(vol->key, uuid) < 0) {
        vboxError(VIR_ERR_INVALID_ARG,
                  _("Could not parse UUID from '%s'"), vol->key);
8550
        return ret;
8551
    }
8552

8553
    vboxIIDFromUUID(&hddIID, uuid);
8554
#if VBOX_API_VERSION < 4000
8555
    rc = data->vboxObj->vtbl->GetHardDisk(data->vboxObj, hddIID.value, &hardDisk);
8556 8557 8558 8559
#else /* VBOX_API_VERSION >= 4000 */
    rc = data->vboxObj->vtbl->FindMedium(data->vboxObj, hddIID.value,
                                         DeviceType_HardDisk, &hardDisk);
#endif /* VBOX_API_VERSION >= 4000 */
8560 8561
    if (NS_SUCCEEDED(rc)) {
        PRUint32 hddstate;
8562

8563 8564 8565 8566
        VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetState, &hddstate);
        if (hddstate != MediaState_Inaccessible) {
            PRUnichar *hddLocationUtf16 = NULL;
            char      *hddLocationUtf8  = NULL;
8567

8568
            VBOX_MEDIUM_FUNC_ARG1(hardDisk, GetLocation, &hddLocationUtf16);
8569

8570 8571
            VBOX_UTF16_TO_UTF8(hddLocationUtf16, &hddLocationUtf8);
            if (hddLocationUtf8) {
8572

8573 8574
                ret = strdup(hddLocationUtf8);
                if (!ret)
8575
                    virReportOOMError();
8576

8577 8578 8579
                VIR_DEBUG("Storage Volume Name: %s", vol->name);
                VIR_DEBUG("Storage Volume Path: %s", hddLocationUtf8);
                VIR_DEBUG("Storage Volume Pool: %s", vol->pool);
8580

8581
                VBOX_UTF8_FREE(hddLocationUtf8);
8582 8583
            }

8584
            VBOX_UTF16_FREE(hddLocationUtf16);
8585
        }
8586 8587

        VBOX_MEDIUM_RELEASE(hardDisk);
8588 8589
    }

8590
    vboxIIDUnalloc(&hddIID);
8591

8592 8593
    return ret;
}
8594

8595 8596 8597 8598 8599 8600 8601 8602 8603 8604 8605 8606 8607 8608 8609 8610 8611 8612 8613 8614 8615 8616 8617 8618 8619 8620 8621 8622 8623 8624 8625 8626 8627 8628 8629 8630 8631 8632 8633 8634 8635 8636 8637 8638 8639 8640 8641 8642 8643 8644 8645 8646 8647 8648 8649 8650 8651 8652 8653 8654 8655 8656 8657 8658 8659 8660 8661 8662 8663 8664 8665 8666 8667 8668 8669 8670 8671 8672 8673 8674 8675 8676 8677 8678 8679 8680 8681 8682 8683 8684 8685 8686 8687 8688 8689 8690 8691 8692 8693 8694 8695 8696 8697 8698 8699 8700 8701 8702 8703 8704 8705 8706 8707 8708 8709 8710 8711 8712 8713 8714 8715 8716 8717
#if VBOX_API_VERSION == 4000
static char *
vboxDomainScreenshot(virDomainPtr dom,
                     virStreamPtr st,
                     unsigned int screen,
                     unsigned int flags ATTRIBUTE_UNUSED)
{
    VBOX_OBJECT_CHECK(dom->conn, char *, NULL);
    IConsole *console = NULL;
    vboxIID iid = VBOX_IID_INITIALIZER;
    IMachine *machine = NULL;
    nsresult rc;
    char *tmp;
    int tmp_fd = -1;
    unsigned int max_screen;

    vboxIIDFromUUID(&iid, dom->uuid);
    rc = VBOX_OBJECT_GET_MACHINE(iid.value, &machine);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_NO_DOMAIN, "%s",
                  _("no domain with matching uuid"));
        return NULL;
    }

    rc = machine->vtbl->GetMonitorCount(machine, &max_screen);
    if (NS_FAILED(rc)) {
        vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                  _("unable to get monitor count"));
        VBOX_RELEASE(machine);
        return NULL;
    }

    if (screen >= max_screen) {
        vboxError(VIR_ERR_INVALID_ARG, _("screen ID higher than monitor "
                  "count (%d)"), max_screen);
        VBOX_RELEASE(machine);
        return NULL;
    }

    if (virAsprintf(&tmp, "%s/cache/libvirt/vbox.screendump.XXXXXX", LOCALSTATEDIR) < 0) {
        virReportOOMError();
        VBOX_RELEASE(machine);
        return NULL;
    }

    if ((tmp_fd = mkstemp(tmp)) == -1) {
        virReportSystemError(errno, _("mkstemp(\"%s\") failed"), tmp);
        VIR_FREE(tmp);
        VBOX_RELEASE(machine);
        return NULL;
    }


    rc = VBOX_SESSION_OPEN_EXISTING(iid.value, machine);
    if (NS_SUCCEEDED(rc)) {
        rc = data->vboxSession->vtbl->GetConsole(data->vboxSession, &console);
        if (NS_SUCCEEDED(rc) && console) {
            IDisplay *display = NULL;

            console->vtbl->GetDisplay(console, &display);

            if (display) {
                PRUint32 width, height, bitsPerPixel;
                PRUint32 screenDataSize;
                PRUint8 *screenData;

                rc = display->vtbl->GetScreenResolution(display, screen,
                                                        &width, &height,
                                                        &bitsPerPixel);

                if (NS_FAILED(rc) || !width || !height) {
                    vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                              _("unable to get screen resolution"));
                    goto endjob;
                }

                rc = display->vtbl->TakeScreenShotPNGToArray(display, screen,
                                                             width, height,
                                                             &screenDataSize,
                                                             &screenData);
                if (NS_FAILED(rc)) {
                    vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                              _("failed to take screenshot"));
                    goto endjob;
                }

                if (safewrite(tmp_fd, (char *) screenData,
                              screenDataSize) < 0) {
                    virReportSystemError(errno, _("unable to write data "
                                                  "to '%s'"), tmp);
                    goto endjob;
                }

                if (VIR_CLOSE(tmp_fd) < 0) {
                    virReportSystemError(errno, _("unable to close %s"), tmp);
                    goto endjob;
                }

                if (virFDStreamOpenFile(st, tmp, 0, 0, O_RDONLY, true) < 0) {
                    vboxError(VIR_ERR_OPERATION_FAILED, "%s",
                              _("unable to open stream"));
                    goto endjob;
                }

                ret = strdup("image/png");

endjob:
                VIR_FREE(screenData);
                VBOX_RELEASE(display);
            }
            VBOX_RELEASE(console);
        }
        VBOX_SESSION_CLOSE();
    }

    VIR_FORCE_CLOSE(tmp_fd);
    VIR_FREE(tmp);
    VBOX_RELEASE(machine);
    vboxIIDUnalloc(&iid);
    return ret;
}
#endif /* VBOX_API_VERSION == 4000 */

8718 8719 8720 8721
/**
 * Function Tables
 */

8722
virDriver NAME(Driver) = {
8723 8724
    .no = VIR_DRV_VBOX,
    .name = "VBOX",
8725 8726 8727 8728 8729 8730 8731 8732 8733 8734 8735 8736 8737 8738 8739 8740 8741 8742 8743 8744 8745 8746 8747 8748 8749 8750 8751 8752 8753 8754 8755 8756 8757 8758 8759 8760 8761 8762 8763 8764 8765
    .open = vboxOpen, /* 0.6.3 */
    .close = vboxClose, /* 0.6.3 */
    .version = vboxGetVersion, /* 0.6.3 */
    .getHostname = virGetHostname, /* 0.6.3 */
    .getMaxVcpus = vboxGetMaxVcpus, /* 0.6.3 */
    .nodeGetInfo = nodeGetInfo, /* 0.6.3 */
    .getCapabilities = vboxGetCapabilities, /* 0.6.3 */
    .listDomains = vboxListDomains, /* 0.6.3 */
    .numOfDomains = vboxNumOfDomains, /* 0.6.3 */
    .domainCreateXML = vboxDomainCreateXML, /* 0.6.3 */
    .domainLookupByID = vboxDomainLookupByID, /* 0.6.3 */
    .domainLookupByUUID = vboxDomainLookupByUUID, /* 0.6.3 */
    .domainLookupByName = vboxDomainLookupByName, /* 0.6.3 */
    .domainSuspend = vboxDomainSuspend, /* 0.6.3 */
    .domainResume = vboxDomainResume, /* 0.6.3 */
    .domainShutdown = vboxDomainShutdown, /* 0.6.3 */
    .domainReboot = vboxDomainReboot, /* 0.6.3 */
    .domainDestroy = vboxDomainDestroy, /* 0.6.3 */
    .domainGetOSType = vboxDomainGetOSType, /* 0.6.3 */
    .domainSetMemory = vboxDomainSetMemory, /* 0.6.3 */
    .domainGetInfo = vboxDomainGetInfo, /* 0.6.3 */
    .domainGetState = vboxDomainGetState, /* 0.9.2 */
    .domainSave = vboxDomainSave, /* 0.6.3 */
    .domainSetVcpus = vboxDomainSetVcpus, /* 0.7.1 */
    .domainSetVcpusFlags = vboxDomainSetVcpusFlags, /* 0.8.5 */
    .domainGetVcpusFlags = vboxDomainGetVcpusFlags, /* 0.8.5 */
    .domainGetMaxVcpus = vboxDomainGetMaxVcpus, /* 0.7.1 */
    .domainGetXMLDesc = vboxDomainGetXMLDesc, /* 0.6.3 */
    .listDefinedDomains = vboxListDefinedDomains, /* 0.6.3 */
    .numOfDefinedDomains = vboxNumOfDefinedDomains, /* 0.6.3 */
    .domainCreate = vboxDomainCreate, /* 0.6.3 */
    .domainCreateWithFlags = vboxDomainCreateWithFlags, /* 0.8.2 */
    .domainDefineXML = vboxDomainDefineXML, /* 0.6.3 */
    .domainUndefine = vboxDomainUndefine, /* 0.6.3 */
    .domainAttachDevice = vboxDomainAttachDevice, /* 0.6.3 */
    .domainAttachDeviceFlags = vboxDomainAttachDeviceFlags, /* 0.7.7 */
    .domainDetachDevice = vboxDomainDetachDevice, /* 0.6.3 */
    .domainDetachDeviceFlags = vboxDomainDetachDeviceFlags, /* 0.7.7 */
    .domainUpdateDeviceFlags = vboxDomainUpdateDeviceFlags, /* 0.8.0 */
    .nodeGetCellsFreeMemory = nodeGetCellsFreeMemory, /* 0.6.5 */
    .nodeGetFreeMemory = nodeGetFreeMemory, /* 0.6.5 */
8766
#if VBOX_API_VERSION == 4000
8767
    .domainScreenshot = vboxDomainScreenshot, /* 0.9.2 */
8768
#endif
8769
#if VBOX_API_VERSION != 2002 && VBOX_API_VERSION != 4000
8770 8771
    .domainEventRegister = vboxDomainEventRegister, /* 0.7.0 */
    .domainEventDeregister = vboxDomainEventDeregister, /* 0.7.0 */
8772
#endif
8773 8774 8775 8776 8777
    .isEncrypted = vboxIsEncrypted, /* 0.7.3 */
    .isSecure = vboxIsSecure, /* 0.7.3 */
    .domainIsActive = vboxDomainIsActive, /* 0.7.3 */
    .domainIsPersistent = vboxDomainIsPersistent, /* 0.7.3 */
    .domainIsUpdated = vboxDomainIsUpdated, /* 0.8.6 */
8778
#if VBOX_API_VERSION != 2002 && VBOX_API_VERSION != 4000
8779 8780
    .domainEventRegisterAny = vboxDomainEventRegisterAny, /* 0.8.0 */
    .domainEventDeregisterAny = vboxDomainEventDeregisterAny, /* 0.8.0 */
8781
#endif
8782 8783 8784 8785 8786 8787 8788 8789 8790
    .domainSnapshotCreateXML = vboxDomainSnapshotCreateXML, /* 0.8.0 */
    .domainSnapshotGetXMLDesc = vboxDomainSnapshotGetXMLDesc, /* 0.8.0 */
    .domainSnapshotNum = vboxDomainSnapshotNum, /* 0.8.0 */
    .domainSnapshotListNames = vboxDomainSnapshotListNames, /* 0.8.0 */
    .domainSnapshotLookupByName = vboxDomainSnapshotLookupByName, /* 0.8.0 */
    .domainHasCurrentSnapshot = vboxDomainHasCurrentSnapshot, /* 0.8.0 */
    .domainSnapshotCurrent = vboxDomainSnapshotCurrent, /* 0.8.0 */
    .domainRevertToSnapshot = vboxDomainRevertToSnapshot, /* 0.8.0 */
    .domainSnapshotDelete = vboxDomainSnapshotDelete, /* 0.8.0 */
8791
};
8792 8793 8794

virNetworkDriver NAME(NetworkDriver) = {
    "VBOX",
8795 8796 8797 8798 8799 8800 8801 8802 8803 8804 8805 8806 8807 8808
    .open                   = vboxNetworkOpen, /* 0.6.4 */
    .close                  = vboxNetworkClose, /* 0.6.4 */
    .numOfNetworks          = vboxNumOfNetworks, /* 0.6.4 */
    .listNetworks           = vboxListNetworks, /* 0.6.4 */
    .numOfDefinedNetworks   = vboxNumOfDefinedNetworks, /* 0.6.4 */
    .listDefinedNetworks    = vboxListDefinedNetworks, /* 0.6.4 */
    .networkLookupByUUID    = vboxNetworkLookupByUUID, /* 0.6.4 */
    .networkLookupByName    = vboxNetworkLookupByName, /* 0.6.4 */
    .networkCreateXML       = vboxNetworkCreateXML, /* 0.6.4 */
    .networkDefineXML       = vboxNetworkDefineXML, /* 0.6.4 */
    .networkUndefine        = vboxNetworkUndefine, /* 0.6.4 */
    .networkCreate          = vboxNetworkCreate, /* 0.6.4 */
    .networkDestroy         = vboxNetworkDestroy, /* 0.6.4 */
    .networkGetXMLDesc      = vboxNetworkGetXMLDesc, /* 0.6.4 */
8809
};
8810 8811 8812

virStorageDriver NAME(StorageDriver) = {
    .name               = "VBOX",
8813 8814 8815 8816 8817 8818 8819 8820 8821 8822 8823 8824 8825 8826 8827 8828
    .open               = vboxStorageOpen, /* 0.7.1 */
    .close              = vboxStorageClose, /* 0.7.1 */
    .numOfPools         = vboxStorageNumOfPools, /* 0.7.1 */
    .listPools          = vboxStorageListPools, /* 0.7.1 */
    .poolLookupByName   = vboxStoragePoolLookupByName, /* 0.7.1 */
    .poolNumOfVolumes   = vboxStoragePoolNumOfVolumes, /* 0.7.1 */
    .poolListVolumes    = vboxStoragePoolListVolumes, /* 0.7.1 */

    .volLookupByName    = vboxStorageVolLookupByName, /* 0.7.1 */
    .volLookupByKey     = vboxStorageVolLookupByKey, /* 0.7.1 */
    .volLookupByPath    = vboxStorageVolLookupByPath, /* 0.7.1 */
    .volCreateXML       = vboxStorageVolCreateXML, /* 0.7.1 */
    .volDelete          = vboxStorageVolDelete, /* 0.7.1 */
    .volGetInfo         = vboxStorageVolGetInfo, /* 0.7.1 */
    .volGetXMLDesc      = vboxStorageVolGetXMLDesc, /* 0.7.1 */
    .volGetPath         = vboxStorageVolGetPath /* 0.7.1 */
8829
};