DrvVD.cpp revision 9daee41d5cb5a2e8390f015a486e67a6cd52b836
77b1a2d8b5dbe2c0b5200794914239fee3c8ee5dvboxsync/* $Id$ */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @file
77b1a2d8b5dbe2c0b5200794914239fee3c8ee5dvboxsync * DrvVD - Generic VBox disk media driver.
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/*
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync * Copyright (C) 2006-2010 Oracle Corporation
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync *
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * available from http://www.virtualbox.org. This file is free software;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * you can redistribute it and/or modify it under the terms of the GNU
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * General Public License (GPL) as published by the Free Software
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
a16eb14ad7a4b5ef91ddc22d3e8e92d930f736fcvboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync */
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync
1c94c0a63ba68be1a7b2c640e70d7a06464e4fcavboxsync/*******************************************************************************
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync* Header files *
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync*******************************************************************************/
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#define LOG_GROUP LOG_GROUP_DRV_VD
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <VBox/vd.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <VBox/vmm/pdmdrv.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <VBox/vmm/pdmasynccompletion.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <VBox/vmm/pdmblkcache.h>
662d52947eeb3fc8fca3b23991a5eee47077f896vboxsync#include <iprt/asm.h>
da957c069c2a3c582fe265ff88170ce4c42b499dvboxsync#include <iprt/alloc.h>
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync#include <iprt/assert.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <iprt/uuid.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <iprt/file.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <iprt/string.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <iprt/tcp.h>
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include <iprt/semaphore.h>
9496f2d398b49813176939d7a339ae513d5175efvboxsync#include <iprt/sg.h>
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#include <iprt/poll.h>
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync#include <iprt/pipe.h>
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#include <iprt/system.h>
efff36b306e370346025647a158689021df2e1d1vboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#ifdef VBOX_WITH_INIP
590bfe12ce22cd3716448fbb9f4dc51664bfe5e2vboxsync/* All lwip header files are not C++ safe. So hack around this. */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncRT_C_DECLS_BEGIN
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#include <lwip/inet.h>
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#include <lwip/tcp.h>
590bfe12ce22cd3716448fbb9f4dc51664bfe5e2vboxsync#include <lwip/sockets.h>
efff36b306e370346025647a158689021df2e1d1vboxsyncRT_C_DECLS_END
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#endif /* VBOX_WITH_INIP */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#include "VBoxDD.h"
efff36b306e370346025647a158689021df2e1d1vboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#ifdef VBOX_WITH_INIP
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/* Small hack to get at lwIP initialized status */
efff36b306e370346025647a158689021df2e1d1vboxsyncextern bool DevINIPConfigured(void);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#endif /* VBOX_WITH_INIP */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/*******************************************************************************
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync* Defined types, constants and macros *
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync*******************************************************************************/
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** Converts a pointer to VBOXDISK::IMedia to a PVBOXDISK. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#define PDMIMEDIA_2_VBOXDISK(pInterface) \
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync ( (PVBOXDISK)((uintptr_t)pInterface - RT_OFFSETOF(VBOXDISK, IMedia)) )
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** Converts a pointer to VBOXDISK::IMediaAsync to a PVBOXDISK. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync#define PDMIMEDIAASYNC_2_VBOXDISK(pInterface) \
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync ( (PVBOXDISK)((uintptr_t)pInterface - RT_OFFSETOF(VBOXDISK, IMediaAsync)) )
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/**
0abd77741a608f6c41c8dfcd4781b8b84adf1044vboxsync * VBox disk container, image information, private part.
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync */
9496f2d398b49813176939d7a339ae513d5175efvboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsynctypedef struct VBOXIMAGE
9496f2d398b49813176939d7a339ae513d5175efvboxsync{
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Pointer to next image. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync struct VBOXIMAGE *pNext;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Pointer to list of VD interfaces. Per-image. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync PVDINTERFACE pVDIfsImage;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Configuration information interface. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync VDINTERFACECONFIG VDIfConfig;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** TCP network stack interface. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync VDINTERFACETCPNET VDIfTcpNet;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** I/O interface. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync VDINTERFACEIO VDIfIo;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync} VBOXIMAGE, *PVBOXIMAGE;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync/**
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync * Storage backend data.
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsynctypedef struct DRVVDSTORAGEBACKEND
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /** PDM async completion end point. */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PPDMASYNCCOMPLETIONENDPOINT pEndpoint;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /** The template. */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PPDMASYNCCOMPLETIONTEMPLATE pTemplate;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /** Event semaphore for synchronous operations. */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync RTSEMEVENT EventSem;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /** Flag whether a synchronous operation is currently pending. */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync volatile bool fSyncIoPending;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /** Return code of the last completed request. */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync int rcReqLast;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /** Callback routine */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PFNVDCOMPLETED pfnCompleted;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync} DRVVDSTORAGEBACKEND, *PDRVVDSTORAGEBACKEND;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync/**
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * VBox disk container media main structure, private part.
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync *
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync * @implements PDMIMEDIA
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync * @implements PDMIMEDIAASYNC
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync * @implements VDINTERFACEERROR
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync * @implements VDINTERFACETCPNET
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync * @implements VDINTERFACEASYNCIO
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync * @implements VDINTERFACECONFIG
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsynctypedef struct VBOXDISK
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /** The VBox disk container. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PVBOXHDD pDisk;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** The media interface. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PDMIMEDIA IMedia;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Media port. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PPDMIMEDIAPORT pDrvMediaPort;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Pointer to the driver instance. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PPDMDRVINS pDrvIns;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Flag whether suspend has changed image open mode to read only. */
21029597fc4b76d0db0c9542daee201447281781vboxsync bool fTempReadOnly;
21029597fc4b76d0db0c9542daee201447281781vboxsync /** Flag whether to use the runtime (true) or startup error facility. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync bool fErrorUseRuntime;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Pointer to list of VD interfaces. Per-disk. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync PVDINTERFACE pVDIfsDisk;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Error interface. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync VDINTERFACEERROR VDIfError;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Thread synchronization interface. */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync VDINTERFACETHREADSYNC VDIfThreadSync;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync /** Flag whether opened disk supports async I/O operations. */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync bool fAsyncIOSupported;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** The async media interface. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync PDMIMEDIAASYNC IMediaAsync;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** The async media port interface above. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync PPDMIMEDIAASYNCPORT pDrvMediaAsyncPort;
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync /** Pointer to the list of data we need to keep per image. */
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync PVBOXIMAGE pImages;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Flag whether the media should allow concurrent open for writing. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync bool fShareable;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Flag whether a merge operation has been set up. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync bool fMergePending;
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** Synchronization to prevent destruction before merge finishes. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync RTSEMFASTMUTEX MergeCompleteMutex;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Synchronization between merge and other image accesses. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync RTSEMRW MergeLock;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /** Source image index for merging. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync unsigned uMergeSource;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Target image index for merging. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync unsigned uMergeTarget;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /** Flag whether boot acceleration is enabled. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync bool fBootAccelEnabled;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Flag whether boot acceleration is currently active. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync bool fBootAccelActive;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Size of the disk, used for read truncation. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync size_t cbDisk;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Size of the configured buffer. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync size_t cbBootAccelBuffer;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Start offset for which the buffer holds data. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync uint64_t offDisk;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Number of valid bytes in the buffer. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync size_t cbDataValid;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** The disk buffer. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync uint8_t *pbData;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Bandwidth group the disk is assigned to. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync char *pszBwGroup;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Flag whether async I/O using the host cache is enabled. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync bool fAsyncIoWithHostCache;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** I/O interface for a cache image. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync VDINTERFACEIO VDIfIoCache;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** Interface list for the cache image. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync PVDINTERFACE pVDIfsCache;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync /** The block cache handle if configured. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync PPDMBLKCACHE pBlkCache;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync} VBOXDISK, *PVBOXDISK;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync/*******************************************************************************
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync* Internal Functions *
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync*******************************************************************************/
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync/**
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync * Internal: allocate new image descriptor and put it in the list
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic PVBOXIMAGE drvvdNewImage(PVBOXDISK pThis)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync AssertPtr(pThis);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PVBOXIMAGE pImage = (PVBOXIMAGE)RTMemAllocZ(sizeof(VBOXIMAGE));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (pImage)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pImage->pVDIfsImage = NULL;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PVBOXIMAGE *pp = &pThis->pImages;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync while (*pp != NULL)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pp = &(*pp)->pNext;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync *pp = pImage;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pImage->pNext = NULL;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return pImage;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/**
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * Internal: free the list of images descriptors.
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic void drvvdFreeImages(PVBOXDISK pThis)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync while (pThis->pImages != NULL)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PVBOXIMAGE p = pThis->pImages;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pThis->pImages = pThis->pImages->pNext;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RTMemFree(p);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/**
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * Make the image temporarily read-only.
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync *
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * @returns VBox status code.
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * @param pThis The driver instance data.
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic int drvvdSetReadonly(PVBOXDISK pThis)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
a9d49c8f2b28a72e6a4db86eee91e4569290157bvboxsync int rc = VINF_SUCCESS;
a9d49c8f2b28a72e6a4db86eee91e4569290157bvboxsync if (!VDIsReadOnly(pThis->pDisk))
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
a9d49c8f2b28a72e6a4db86eee91e4569290157bvboxsync unsigned uOpenFlags;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync rc = VDGetOpenFlags(pThis->pDisk, VD_LAST_IMAGE, &uOpenFlags);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync AssertRC(rc);
9496f2d398b49813176939d7a339ae513d5175efvboxsync uOpenFlags |= VD_OPEN_FLAGS_READONLY;
9496f2d398b49813176939d7a339ae513d5175efvboxsync rc = VDSetOpenFlags(pThis->pDisk, VD_LAST_IMAGE, uOpenFlags);
9496f2d398b49813176939d7a339ae513d5175efvboxsync AssertRC(rc);
9496f2d398b49813176939d7a339ae513d5175efvboxsync pThis->fTempReadOnly = true;
59d7f5939d42ad9d344fbad8401e00a900db92c5vboxsync }
59d7f5939d42ad9d344fbad8401e00a900db92c5vboxsync return rc;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync}
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync/**
9496f2d398b49813176939d7a339ae513d5175efvboxsync * Undo the temporary read-only status of the image.
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync *
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync * @returns VBox status code.
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync * @param pThis The driver instance data.
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsyncstatic int drvvdSetWritable(PVBOXDISK pThis)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync int rc = VINF_SUCCESS;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync if (pThis->fTempReadOnly)
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync {
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync unsigned uOpenFlags;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync rc = VDGetOpenFlags(pThis->pDisk, VD_LAST_IMAGE, &uOpenFlags);
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync AssertRC(rc);
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync uOpenFlags &= ~VD_OPEN_FLAGS_READONLY;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync rc = VDSetOpenFlags(pThis->pDisk, VD_LAST_IMAGE, uOpenFlags);
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync if (RT_SUCCESS(rc))
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync pThis->fTempReadOnly = false;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync else
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync AssertRC(rc);
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync }
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync return rc;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync}
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync/*******************************************************************************
9496f2d398b49813176939d7a339ae513d5175efvboxsync* Error reporting callback *
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync*******************************************************************************/
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsyncstatic void drvvdErrorCallback(void *pvUser, int rc, RT_SRC_POS_DECL,
9496f2d398b49813176939d7a339ae513d5175efvboxsync const char *pszFormat, va_list va)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PPDMDRVINS pDrvIns = (PPDMDRVINS)pvUser;
9496f2d398b49813176939d7a339ae513d5175efvboxsync PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (pThis->fErrorUseRuntime)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync /* We must not pass VMSETRTERR_FLAGS_FATAL as it could lead to a
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync * deadlock: We are probably executed in a thread context != EMT
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync * and the EM thread would wait until every thread is suspended
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync * but we would wait for the EM thread ... */
ffbe6daf773e38167f3cabaf1f063d84ecd063e9vboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDMDrvHlpVMSetRuntimeErrorV(pDrvIns, /* fFlags=*/ 0, "DrvVD", pszFormat, va);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync else
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDMDrvHlpVMSetErrorV(pDrvIns, rc, RT_SRC_POS_ARGS, pszFormat, va);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync}
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync/*******************************************************************************
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync* VD Async I/O interface implementation *
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync*******************************************************************************/
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync#ifdef VBOX_WITH_PDM_ASYNC_COMPLETION
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsyncstatic DECLCALLBACK(void) drvvdAsyncTaskCompleted(PPDMDRVINS pDrvIns, void *pvTemplateUser, void *pvUser, int rcReq)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pvTemplateUser;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync LogFlowFunc(("pDrvIns=%#p pvTemplateUser=%#p pvUser=%#p rcReq=%d\n",
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync pDrvIns, pvTemplateUser, pvUser, rcReq));
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (pStorageBackend->fSyncIoPending)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync {
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync Assert(!pvUser);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync pStorageBackend->rcReqLast = rcReq;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync pStorageBackend->fSyncIoPending = false;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync RTSemEventSignal(pStorageBackend->EventSem);
9496f2d398b49813176939d7a339ae513d5175efvboxsync }
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync else
9496f2d398b49813176939d7a339ae513d5175efvboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync int rc;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync AssertPtr(pvUser);
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync AssertPtr(pStorageBackend->pfnCompleted);
9496f2d398b49813176939d7a339ae513d5175efvboxsync rc = pStorageBackend->pfnCompleted(pvUser, rcReq);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync AssertRC(rc);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync }
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync}
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOOpen(void *pvUser, const char *pszLocation,
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync uint32_t fOpen,
9496f2d398b49813176939d7a339ae513d5175efvboxsync PFNVDCOMPLETED pfnCompleted,
9496f2d398b49813176939d7a339ae513d5175efvboxsync void **ppStorage)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
9496f2d398b49813176939d7a339ae513d5175efvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
9496f2d398b49813176939d7a339ae513d5175efvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)RTMemAllocZ(sizeof(DRVVDSTORAGEBACKEND));
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync int rc = VINF_SUCCESS;
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (pStorageBackend)
9496f2d398b49813176939d7a339ae513d5175efvboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync pStorageBackend->fSyncIoPending = false;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync pStorageBackend->rcReqLast = VINF_SUCCESS;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync pStorageBackend->pfnCompleted = pfnCompleted;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync rc = RTSemEventCreate(&pStorageBackend->EventSem);
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync if (RT_SUCCESS(rc))
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync {
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync rc = PDMDrvHlpAsyncCompletionTemplateCreate(pThis->pDrvIns, &pStorageBackend->pTemplate,
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync drvvdAsyncTaskCompleted, pStorageBackend, "AsyncTaskCompleted");
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (RT_SUCCESS(rc))
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync uint32_t fFlags = (fOpen & RTFILE_O_ACCESS_MASK) == RTFILE_O_READ
9496f2d398b49813176939d7a339ae513d5175efvboxsync ? PDMACEP_FILE_FLAGS_READ_ONLY
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync : 0;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (pThis->fShareable)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync Assert((fOpen & RTFILE_O_DENY_MASK) == RTFILE_O_DENY_NONE);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync fFlags |= PDMACEP_FILE_FLAGS_DONT_LOCK;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync }
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (pThis->fAsyncIoWithHostCache)
9496f2d398b49813176939d7a339ae513d5175efvboxsync fFlags |= PDMACEP_FILE_FLAGS_HOST_CACHE_ENABLED;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync rc = PDMR3AsyncCompletionEpCreateForFile(&pStorageBackend->pEndpoint,
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync pszLocation, fFlags,
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync pStorageBackend->pTemplate);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync if (RT_SUCCESS(rc))
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync {
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync if (pThis->pszBwGroup)
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync rc = PDMR3AsyncCompletionEpSetBwMgr(pStorageBackend->pEndpoint, pThis->pszBwGroup);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync if (RT_SUCCESS(rc))
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync {
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync *ppStorage = pStorageBackend;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync return VINF_SUCCESS;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync }
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDMR3AsyncCompletionEpClose(pStorageBackend->pEndpoint);
9496f2d398b49813176939d7a339ae513d5175efvboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDMR3AsyncCompletionTemplateDestroy(pStorageBackend->pTemplate);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync RTSemEventDestroy(pStorageBackend->EventSem);
9496f2d398b49813176939d7a339ae513d5175efvboxsync }
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync RTMemFree(pStorageBackend);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync }
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync else
9496f2d398b49813176939d7a339ae513d5175efvboxsync rc = VERR_NO_MEMORY;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync return rc;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync}
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOClose(void *pvUser, void *pStorage)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync PDMR3AsyncCompletionEpClose(pStorageBackend->pEndpoint);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync PDMR3AsyncCompletionTemplateDestroy(pStorageBackend->pTemplate);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync RTSemEventDestroy(pStorageBackend->EventSem);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync RTMemFree(pStorageBackend);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync return VINF_SUCCESS;;
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync}
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOReadSync(void *pvUser, void *pStorage, uint64_t uOffset,
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync void *pvBuf, size_t cbRead, size_t *pcbRead)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
9496f2d398b49813176939d7a339ae513d5175efvboxsync RTSGSEG DataSeg;
9496f2d398b49813176939d7a339ae513d5175efvboxsync PPDMASYNCCOMPLETIONTASK pTask;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync Assert(!pStorageBackend->fSyncIoPending);
9496f2d398b49813176939d7a339ae513d5175efvboxsync ASMAtomicXchgBool(&pStorageBackend->fSyncIoPending, true);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync DataSeg.cbSeg = cbRead;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync DataSeg.pvSeg = pvBuf;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync int rc = PDMR3AsyncCompletionEpRead(pStorageBackend->pEndpoint, uOffset, &DataSeg, 1, cbRead, NULL, &pTask);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync if (RT_FAILURE(rc))
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync return rc;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (rc == VINF_AIO_TASK_PENDING)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync /* Wait */
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync rc = RTSemEventWait(pStorageBackend->EventSem, RT_INDEFINITE_WAIT);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync AssertRC(rc);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync }
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync else
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync ASMAtomicXchgBool(&pStorageBackend->fSyncIoPending, false);
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync if (pcbRead)
ce03ea57fdcf3d48523b1de5b894feb75e1b34davboxsync *pcbRead = cbRead;
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync return pStorageBackend->rcReqLast;
9496f2d398b49813176939d7a339ae513d5175efvboxsync}
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOWriteSync(void *pvUser, void *pStorage, uint64_t uOffset,
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync const void *pvBuf, size_t cbWrite, size_t *pcbWritten)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync{
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
9496f2d398b49813176939d7a339ae513d5175efvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
9496f2d398b49813176939d7a339ae513d5175efvboxsync RTSGSEG DataSeg;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PPDMASYNCCOMPLETIONTASK pTask;
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync Assert(!pStorageBackend->fSyncIoPending);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync ASMAtomicXchgBool(&pStorageBackend->fSyncIoPending, true);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync DataSeg.cbSeg = cbWrite;
9496f2d398b49813176939d7a339ae513d5175efvboxsync DataSeg.pvSeg = (void *)pvBuf;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync int rc = PDMR3AsyncCompletionEpWrite(pStorageBackend->pEndpoint, uOffset, &DataSeg, 1, cbWrite, NULL, &pTask);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (RT_FAILURE(rc))
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync return rc;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (rc == VINF_AIO_TASK_PENDING)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync /* Wait */
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync rc = RTSemEventWait(pStorageBackend->EventSem, RT_INDEFINITE_WAIT);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync AssertRC(rc);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync else
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync ASMAtomicXchgBool(&pStorageBackend->fSyncIoPending, false);
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync if (pcbWritten)
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync *pcbWritten = cbWrite;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync return pStorageBackend->rcReqLast;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync}
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOFlushSync(void *pvUser, void *pStorage)
9496f2d398b49813176939d7a339ae513d5175efvboxsync{
59d7f5939d42ad9d344fbad8401e00a900db92c5vboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
59d7f5939d42ad9d344fbad8401e00a900db92c5vboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync PPDMASYNCCOMPLETIONTASK pTask;
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync LogFlowFunc(("pvUser=%#p pStorage=%#p\n", pvUser, pStorage));
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync Assert(!pStorageBackend->fSyncIoPending);
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync ASMAtomicXchgBool(&pStorageBackend->fSyncIoPending, true);
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync int rc = PDMR3AsyncCompletionEpFlush(pStorageBackend->pEndpoint, NULL, &pTask);
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync if (RT_FAILURE(rc))
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync return rc;
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync if (rc == VINF_AIO_TASK_PENDING)
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync {
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync /* Wait */
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync LogFlowFunc(("Waiting for flush to complete\n"));
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync rc = RTSemEventWait(pStorageBackend->EventSem, RT_INDEFINITE_WAIT);
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync AssertRC(rc);
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync }
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync else
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync ASMAtomicXchgBool(&pStorageBackend->fSyncIoPending, false);
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync
efff36b306e370346025647a158689021df2e1d1vboxsync return pStorageBackend->rcReqLast;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOReadAsync(void *pvUser, void *pStorage, uint64_t uOffset,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PCRTSGSEG paSegments, size_t cSegments,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync size_t cbRead, void *pvCompletion,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync void **ppTask)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync int rc = PDMR3AsyncCompletionEpRead(pStorageBackend->pEndpoint, uOffset, paSegments, cSegments, cbRead,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync pvCompletion, (PPPDMASYNCCOMPLETIONTASK)ppTask);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync if (rc == VINF_AIO_TASK_PENDING)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return rc;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOWriteAsync(void *pvUser, void *pStorage, uint64_t uOffset,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PCRTSGSEG paSegments, size_t cSegments,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync size_t cbWrite, void *pvCompletion,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync void **ppTask)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync int rc = PDMR3AsyncCompletionEpWrite(pStorageBackend->pEndpoint, uOffset, paSegments, cSegments, cbWrite,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync pvCompletion, (PPPDMASYNCCOMPLETIONTASK)ppTask);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync if (rc == VINF_AIO_TASK_PENDING)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return rc;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOFlushAsync(void *pvUser, void *pStorage,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync void *pvCompletion, void **ppTask)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync int rc = PDMR3AsyncCompletionEpFlush(pStorageBackend->pEndpoint, pvCompletion,
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync (PPPDMASYNCCOMPLETIONTASK)ppTask);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync if (rc == VINF_AIO_TASK_PENDING)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return rc;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOGetSize(void *pvUser, void *pStorage, uint64_t *pcbSize)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pDrvVD = (PVBOXDISK)pvUser;
7b067f3f07310bff46d1d6a4ac94d8b9bb7ccccdvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return PDMR3AsyncCompletionEpGetSize(pStorageBackend->pEndpoint, pcbSize);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdAsyncIOSetSize(void *pvUser, void *pStorage, uint64_t cbSize)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pDrvVD = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PDRVVDSTORAGEBACKEND pStorageBackend = (PDRVVDSTORAGEBACKEND)pStorage;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return PDMR3AsyncCompletionEpSetSize(pStorageBackend->pEndpoint, cbSize);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#endif /* VBOX_WITH_PDM_ASYNC_COMPLETION */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/*******************************************************************************
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync* VD Thread Synchronization interface implementation *
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync*******************************************************************************/
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdThreadStartRead(void *pvUser)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTSemRWRequestRead(pThis->MergeLock, RT_INDEFINITE_WAIT);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdThreadFinishRead(void *pvUser)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTSemRWReleaseRead(pThis->MergeLock);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdThreadStartWrite(void *pvUser)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTSemRWRequestWrite(pThis->MergeLock, RT_INDEFINITE_WAIT);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdThreadFinishWrite(void *pvUser)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVBOXDISK pThis = (PVBOXDISK)pvUser;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTSemRWReleaseWrite(pThis->MergeLock);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/*******************************************************************************
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync* VD Configuration interface implementation *
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync*******************************************************************************/
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic bool drvvdCfgAreKeysValid(void *pvUser, const char *pszzValid)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return CFGMR3AreValuesValid((PCFGMNODE)pvUser, pszzValid);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic int drvvdCfgQuerySize(void *pvUser, const char *pszName, size_t *pcb)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return CFGMR3QuerySize((PCFGMNODE)pvUser, pszName, pcb);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic int drvvdCfgQuery(void *pvUser, const char *pszName, char *pszString, size_t cchString)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return CFGMR3QueryString((PCFGMNODE)pvUser, pszName, pszString, cchString);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync#ifdef VBOX_WITH_INIP
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/*******************************************************************************
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync* VD TCP network stack interface implementation - INIP case *
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync*******************************************************************************/
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsynctypedef union INIPSOCKADDRUNION
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync struct sockaddr Addr;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync struct sockaddr_in Ipv4;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync} INIPSOCKADDRUNION;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsynctypedef struct INIPSOCKET
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync int hSock;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync} INIPSOCKET, *PINIPSOCKET;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdINIPFlush(VDSOCKET Sock);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/** @copydoc VDINTERFACETCPNET::pfnSocketCreate */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdINIPSocketCreate(uint32_t fFlags, PVDSOCKET pSock)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PINIPSOCKET pSocketInt = NULL;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync /*
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync * The extended select method is not supported because it is impossible to wakeup
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync * the thread.
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync if (fFlags & VD_INTERFACETCPNET_CONNECT_EXTENDED_SELECT)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return VERR_NOT_SUPPORTED;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync pSocketInt = (PINIPSOCKET)RTMemAllocZ(sizeof(INIPSOCKET));
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync if (pSocketInt)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync {
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync pSocketInt->hSock = INT32_MAX;
efff36b306e370346025647a158689021df2e1d1vboxsync *pSock = (VDSOCKET)pSocketInt;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return VINF_SUCCESS;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_NO_MEMORY;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @copydoc VDINTERFACETCPNET::pfnSocketCreate */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsyncstatic DECLCALLBACK(int) drvvdINIPSocketDestroy(VDSOCKET Sock)
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync RTMemFree(pSocketInt);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync return VINF_SUCCESS;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync}
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync
ff88d4153cd65650577e8c2d1a5a3fdfa0404a80vboxsync/** @copydoc VDINTERFACETCPNET::pfnClientConnect */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdINIPClientConnect(VDSOCKET Sock, const char *pszAddress, uint32_t uPort)
fe813b3594039ba864493438e78ee0e7132bc445vboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync int rc = VINF_SUCCESS;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* Check whether lwIP is set up in this VM instance. */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync if (!DevINIPConfigured())
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync {
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync LogRelFunc(("no IP stack\n"));
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_NET_HOST_UNREACHABLE;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* Resolve hostname. As there is no standard resolver for lwIP yet,
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync * just accept numeric IP addresses for now. */
ff88d4153cd65650577e8c2d1a5a3fdfa0404a80vboxsync struct in_addr ip;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (!lwip_inet_aton(pszAddress, &ip))
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync {
fe813b3594039ba864493438e78ee0e7132bc445vboxsync LogRelFunc(("cannot resolve IP %s\n", pszAddress));
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_NET_HOST_UNREACHABLE;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* Create socket and connect. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync int iSock = lwip_socket(PF_INET, SOCK_STREAM, 0);
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync if (iSock != -1)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync struct sockaddr_in InAddr = {0};
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync InAddr.sin_family = AF_INET;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync InAddr.sin_port = htons(uPort);
fe813b3594039ba864493438e78ee0e7132bc445vboxsync InAddr.sin_addr = ip;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (!lwip_connect(iSock, (struct sockaddr *)&InAddr, sizeof(InAddr)))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSocketInt->hSock = iSock;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VINF_SUCCESS;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = VERR_NET_CONNECTION_REFUSED; /* @todo real solution needed */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync lwip_close(iSock);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync }
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync else
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = VERR_NET_CONNECTION_REFUSED; /* @todo real solution needed */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return rc;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @copydoc VDINTERFACETCPNET::pfnClientClose */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsyncstatic DECLCALLBACK(int) drvvdINIPClientClose(VDSOCKET Sock)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync lwip_close(pSocketInt->hSock);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSocketInt->hSock = INT32_MAX;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VINF_SUCCESS; /** @todo real solution needed */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync/** @copydoc VDINTERFACETCPNET::pfnIsClientConnected */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsyncstatic DECLCALLBACK(bool) drvvdINIPIsClientConnected(VDSOCKET Sock)
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync return pSocketInt->hSock != INT32_MAX;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync}
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @copydoc VDINTERFACETCPNET::pfnSelectOne */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdINIPSelectOne(VDSOCKET Sock, RTMSINTERVAL cMillies)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
fe813b3594039ba864493438e78ee0e7132bc445vboxsync fd_set fdsetR;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync FD_ZERO(&fdsetR);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync FD_SET((uintptr_t)pSocketInt->hSock, &fdsetR);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync fd_set fdsetE = fdsetR;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync int rc;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync if (cMillies == RT_INDEFINITE_WAIT)
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync rc = lwip_select(pSocketInt->hSock + 1, &fdsetR, NULL, &fdsetE, NULL);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync else
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync struct timeval timeout;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync timeout.tv_sec = cMillies / 1000;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync timeout.tv_usec = (cMillies % 1000) * 1000;
594521f7faf13f7a88f31e6cd76629bd67340229vboxsync rc = lwip_select(pSocketInt->hSock + 1, &fdsetR, NULL, &fdsetE, &timeout);
e04eeee1b306d610b0441cee9bf1c750100254d5vboxsync }
fe813b3594039ba864493438e78ee0e7132bc445vboxsync if (rc > 0)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VINF_SUCCESS;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (rc == 0)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_TIMEOUT;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync return VERR_NET_CONNECTION_REFUSED; /** @todo real solution needed */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync}
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @copydoc VDINTERFACETCPNET::pfnRead */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdINIPRead(VDSOCKET Sock, void *pvBuffer, size_t cbBuffer, size_t *pcbRead)
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync{
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync /* Do params checking */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync if (!pvBuffer || !cbBuffer)
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync {
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync AssertMsgFailed(("Invalid params\n"));
fe813b3594039ba864493438e78ee0e7132bc445vboxsync return VERR_INVALID_PARAMETER;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /*
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * Read loop.
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync * If pcbRead is NULL we have to fill the entire buffer!
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync size_t cbRead = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync size_t cbToRead = cbBuffer;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync for (;;)
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync {
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync /** @todo this clipping here is just in case (the send function
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync * needed it, so I added it here, too). Didn't investigate if this
594521f7faf13f7a88f31e6cd76629bd67340229vboxsync * really has issues. Better be safe than sorry. */
e04eeee1b306d610b0441cee9bf1c750100254d5vboxsync ssize_t cbBytesRead = lwip_recv(pSocketInt->hSock, (char *)pvBuffer + cbRead,
fe813b3594039ba864493438e78ee0e7132bc445vboxsync RT_MIN(cbToRead, 32768), 0);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (cbBytesRead < 0)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_NET_CONNECTION_REFUSED; /** @todo real solution */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (cbBytesRead == 0 && errno) /** @todo r=bird: lwip_recv will not touch errno on Windows. This may apply to other hosts as well */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_NET_CONNECTION_REFUSED; /** @todo real solution */
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync if (pcbRead)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* return partial data */
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync *pcbRead = cbBytesRead;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync break;
af90f37ee9175da3aed36bda13519a917cc4effbvboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* read more? */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync cbRead += cbBytesRead;
9496f2d398b49813176939d7a339ae513d5175efvboxsync if (cbRead == cbBuffer)
9496f2d398b49813176939d7a339ae513d5175efvboxsync break;
9496f2d398b49813176939d7a339ae513d5175efvboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync /* next */
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync cbToRead = cbBuffer - cbRead;
9496f2d398b49813176939d7a339ae513d5175efvboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync return VINF_SUCCESS;
9496f2d398b49813176939d7a339ae513d5175efvboxsync}
9496f2d398b49813176939d7a339ae513d5175efvboxsync
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync/** @copydoc VDINTERFACETCPNET::pfnWrite */
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsyncstatic DECLCALLBACK(int) drvvdINIPWrite(VDSOCKET Sock, const void *pvBuffer, size_t cbBuffer)
fe813b3594039ba864493438e78ee0e7132bc445vboxsync{
9496f2d398b49813176939d7a339ae513d5175efvboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
9496f2d398b49813176939d7a339ae513d5175efvboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync do
9496f2d398b49813176939d7a339ae513d5175efvboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync /** @todo lwip send only supports up to 65535 bytes in a single
9496f2d398b49813176939d7a339ae513d5175efvboxsync * send (stupid limitation buried in the code), so make sure we
9496f2d398b49813176939d7a339ae513d5175efvboxsync * don't get any wraparounds. This should be moved to DevINIP
9496f2d398b49813176939d7a339ae513d5175efvboxsync * stack interface once that's implemented. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync ssize_t cbWritten = lwip_send(pSocketInt->hSock, (void *)pvBuffer,
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync RT_MIN(cbBuffer, 32768), 0);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync if (cbWritten < 0)
fe813b3594039ba864493438e78ee0e7132bc445vboxsync return VERR_NET_CONNECTION_REFUSED; /** @todo real solution needed */
9496f2d398b49813176939d7a339ae513d5175efvboxsync AssertMsg(cbBuffer >= (size_t)cbWritten, ("Wrote more than we requested!!! cbWritten=%d cbBuffer=%d\n",
9496f2d398b49813176939d7a339ae513d5175efvboxsync cbWritten, cbBuffer));
9496f2d398b49813176939d7a339ae513d5175efvboxsync cbBuffer -= cbWritten;
9496f2d398b49813176939d7a339ae513d5175efvboxsync pvBuffer = (const char *)pvBuffer + cbWritten;
9496f2d398b49813176939d7a339ae513d5175efvboxsync } while (cbBuffer);
9496f2d398b49813176939d7a339ae513d5175efvboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync return VINF_SUCCESS;
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync}
9496f2d398b49813176939d7a339ae513d5175efvboxsync
5341459ca931b65de60b5af2a4cba6836b6b45cavboxsync/** @copydoc VDINTERFACETCPNET::pfnSgWrite */
9496f2d398b49813176939d7a339ae513d5175efvboxsyncstatic DECLCALLBACK(int) drvvdINIPSgWrite(VDSOCKET Sock, PCRTSGBUF pSgBuf)
9496f2d398b49813176939d7a339ae513d5175efvboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync int rc = VINF_SUCCESS;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* This is an extremely crude emulation, however it's good enough
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * for our iSCSI code. INIP has no sendmsg(). */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync for (unsigned i = 0; i < pSgBuf->cSegs; i++)
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync {
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync rc = drvvdINIPWrite(Sock, pSgBuf->paSegs[i].pvSeg,
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync pSgBuf->paSegs[i].cbSeg);
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync if (RT_FAILURE(rc))
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync break;
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync }
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync if (RT_SUCCESS(rc))
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync drvvdINIPFlush(Sock);
a39ea3668b7019c23a68936259545f9b71bce1aavboxsync
da3503c04ce76e653401396fe2795a9bc2427a1dvboxsync return rc;
da3503c04ce76e653401396fe2795a9bc2427a1dvboxsync}
ee4d840f54fd2dcea8a73b1b86d5ec0db370b05dvboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @copydoc VDINTERFACETCPNET::pfnFlush */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdINIPFlush(VDSOCKET Sock)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync int fFlag = 1;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync lwip_setsockopt(pSocketInt->hSock, IPPROTO_TCP, TCP_NODELAY,
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync (const char *)&fFlag, sizeof(fFlag));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync fFlag = 0;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync lwip_setsockopt(pSocketInt->hSock, IPPROTO_TCP, TCP_NODELAY,
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync (const char *)&fFlag, sizeof(fFlag));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VINF_SUCCESS;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** @copydoc VDINTERFACETCPNET::pfnSetSendCoalescing */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic DECLCALLBACK(int) drvvdINIPSetSendCoalescing(VDSOCKET Sock, bool fEnable)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync int fFlag = fEnable ? 0 : 1;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync lwip_setsockopt(pSocketInt->hSock, IPPROTO_TCP, TCP_NODELAY,
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync (const char *)&fFlag, sizeof(fFlag));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VINF_SUCCESS;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** @copydoc VDINTERFACETCPNET::pfnGetLocalAddress */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic DECLCALLBACK(int) drvvdINIPGetLocalAddress(VDSOCKET Sock, PRTNETADDR pAddr)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync INIPSOCKADDRUNION u;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync socklen_t cbAddr = sizeof(u);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RT_ZERO(u);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (!lwip_getsockname(pSocketInt->hSock, &u.Addr, &cbAddr))
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /*
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync * Convert the address.
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if ( cbAddr == sizeof(struct sockaddr_in)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync && u.Addr.sa_family == AF_INET)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RT_ZERO(*pAddr);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pAddr->enmType = RTNETADDRTYPE_IPV4;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pAddr->uPort = RT_N2H_U16(u.Ipv4.sin_port);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pAddr->uAddr.IPv4.u = u.Ipv4.sin_addr.s_addr;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync else
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NET_ADDRESS_FAMILY_NOT_SUPPORTED;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VINF_SUCCESS;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NET_OPERATION_NOT_SUPPORTED;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** @copydoc VDINTERFACETCPNET::pfnGetPeerAddress */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic DECLCALLBACK(int) drvvdINIPGetPeerAddress(VDSOCKET Sock, PRTNETADDR pAddr)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PINIPSOCKET pSocketInt = (PINIPSOCKET)Sock;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync INIPSOCKADDRUNION u;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync socklen_t cbAddr = sizeof(u);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RT_ZERO(u);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (!lwip_getpeername(pSocketInt->hSock, &u.Addr, &cbAddr))
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /*
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * Convert the address.
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if ( cbAddr == sizeof(struct sockaddr_in)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync && u.Addr.sa_family == AF_INET)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RT_ZERO(*pAddr);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pAddr->enmType = RTNETADDRTYPE_IPV4;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pAddr->uPort = RT_N2H_U16(u.Ipv4.sin_port);
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pAddr->uAddr.IPv4.u = u.Ipv4.sin_addr.s_addr;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync else
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NET_ADDRESS_FAMILY_NOT_SUPPORTED;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VINF_SUCCESS;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NET_OPERATION_NOT_SUPPORTED;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** @copydoc VDINTERFACETCPNET::pfnSelectOneEx */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic DECLCALLBACK(int) drvvdINIPSelectOneEx(VDSOCKET Sock, uint32_t fEvents, uint32_t *pfEvents, RTMSINTERVAL cMillies)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync AssertMsgFailed(("Not supported!\n"));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NOT_SUPPORTED;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** @copydoc VDINTERFACETCPNET::pfnPoke */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic DECLCALLBACK(int) drvvdINIPPoke(VDSOCKET Sock)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync AssertMsgFailed(("Not supported!\n"));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NOT_SUPPORTED;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync}
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync#endif /* VBOX_WITH_INIP */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/*******************************************************************************
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync* VD TCP network stack interface implementation - Host TCP case *
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync*******************************************************************************/
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/**
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * Socket data.
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsynctypedef struct VDSOCKETINT
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** IPRT socket handle. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RTSOCKET hSocket;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Pollset with the wakeup pipe and socket. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RTPOLLSET hPollSet;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Pipe endpoint - read (in the pollset). */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RTPIPE hPipeR;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Pipe endpoint - write. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync RTPIPE hPipeW;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Flag whether the thread was woken up. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync volatile bool fWokenUp;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Flag whether the thread is waiting in the select call. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync volatile bool fWaiting;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /** Old event mask. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync uint32_t fEventsOld;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync} VDSOCKETINT, *PVDSOCKETINT;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** Pollset id of the socket. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync#define VDSOCKET_POLL_ID_SOCKET 0
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** Pollset id of the pipe. */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync#define VDSOCKET_POLL_ID_PIPE 1
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync/** @copydoc VDINTERFACETCPNET::pfnSocketCreate */
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsyncstatic DECLCALLBACK(int) drvvdTcpSocketCreate(uint32_t fFlags, PVDSOCKET pSock)
e74eef731a813e4e06680c587a6759b9974b29c9vboxsync{
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync int rc = VINF_SUCCESS;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync int rc2 = VINF_SUCCESS;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync PVDSOCKETINT pSockInt = NULL;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync pSockInt = (PVDSOCKETINT)RTMemAllocZ(sizeof(VDSOCKETINT));
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (!pSockInt)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync return VERR_NO_MEMORY;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSockInt->hSocket = NIL_RTSOCKET;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSockInt->hPollSet = NIL_RTPOLLSET;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSockInt->hPipeR = NIL_RTPIPE;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSockInt->hPipeW = NIL_RTPIPE;
cba6719bd64ec749967bbe931230452664109857vboxsync pSockInt->fWokenUp = false;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSockInt->fWaiting = false;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync
c28fa006ba669ad8f26ae31d00a338379c04ea1bvboxsync if (fFlags & VD_INTERFACETCPNET_CONNECT_EXTENDED_SELECT)
cba6719bd64ec749967bbe931230452664109857vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* Init pipe and pollset. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPipeCreate(&pSockInt->hPipeR, &pSockInt->hPipeW, 0);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync if (RT_SUCCESS(rc))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync rc = RTPollSetCreate(&pSockInt->hPollSet);
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (RT_SUCCESS(rc))
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPollSetAddPipe(pSockInt->hPollSet, pSockInt->hPipeR,
e74eef731a813e4e06680c587a6759b9974b29c9vboxsync RTPOLL_EVT_READ, VDSOCKET_POLL_ID_PIPE);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (RT_SUCCESS(rc))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync *pSock = pSockInt;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VINF_SUCCESS;
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync }
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync RTPollSetRemove(pSockInt->hPollSet, VDSOCKET_POLL_ID_PIPE);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync rc2 = RTPollSetDestroy(pSockInt->hPollSet);
090d729e786b999dc285f8ea267f9effd1319544vboxsync AssertRC(rc2);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync rc2 = RTPipeClose(pSockInt->hPipeR);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync AssertRC(rc2);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync rc2 = RTPipeClose(pSockInt->hPipeW);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync AssertRC(rc2);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync }
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync }
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync else
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync {
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync *pSock = pSockInt;
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync return VINF_SUCCESS;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync RTMemFree(pSockInt);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync return rc;
9496f2d398b49813176939d7a339ae513d5175efvboxsync}
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
68a4ee3a31a0807abd03eae881c1bbaf4d42ee6dvboxsync/** @copydoc VDINTERFACETCPNET::pfnSocketDestroy */
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsyncstatic DECLCALLBACK(int) drvvdTcpSocketDestroy(VDSOCKET Sock)
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync{
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync int rc = VINF_SUCCESS;
caf54c14752060b187e3fca12a6f71f4b13126b8vboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
9496f2d398b49813176939d7a339ae513d5175efvboxsync
68a4ee3a31a0807abd03eae881c1bbaf4d42ee6dvboxsync /* Destroy the pipe and pollset if necessary. */
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync if (pSockInt->hPollSet != NIL_RTPOLLSET)
9496f2d398b49813176939d7a339ae513d5175efvboxsync {
9496f2d398b49813176939d7a339ae513d5175efvboxsync if (pSockInt->hSocket != NIL_RTSOCKET)
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync {
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync rc = RTPollSetRemove(pSockInt->hPollSet, VDSOCKET_POLL_ID_SOCKET);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync Assert(RT_SUCCESS(rc) || rc == VERR_POLL_HANDLE_ID_NOT_FOUND);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync }
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync rc = RTPollSetRemove(pSockInt->hPollSet, VDSOCKET_POLL_ID_PIPE);
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync AssertRC(rc);
975ad9d9bc9c4dc96b41d9f67a65228b1b338e2avboxsync rc = RTPollSetDestroy(pSockInt->hPollSet);
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync AssertRC(rc);
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync rc = RTPipeClose(pSockInt->hPipeR);
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync AssertRC(rc);
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync rc = RTPipeClose(pSockInt->hPipeW);
9496f2d398b49813176939d7a339ae513d5175efvboxsync AssertRC(rc);
da3503c04ce76e653401396fe2795a9bc2427a1dvboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (pSockInt->hSocket != NIL_RTSOCKET)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTTcpClientCloseEx(pSockInt->hSocket, false /*fGracefulShutdown*/);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync RTMemFree(pSockInt);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync return rc;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
c28fa006ba669ad8f26ae31d00a338379c04ea1bvboxsync
090d729e786b999dc285f8ea267f9effd1319544vboxsync/** @copydoc VDINTERFACETCPNET::pfnClientConnect */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdTcpClientConnect(VDSOCKET Sock, const char *pszAddress, uint32_t uPort)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync int rc = VINF_SUCCESS;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
c28fa006ba669ad8f26ae31d00a338379c04ea1bvboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTTcpClientConnect(pszAddress, uPort, &pSockInt->hSocket);
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (RT_SUCCESS(rc))
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* Add to the pollset if required. */
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (pSockInt->hPollSet != NIL_RTPOLLSET)
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync {
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync pSockInt->fEventsOld = RTPOLL_EVT_READ | RTPOLL_EVT_WRITE | RTPOLL_EVT_ERROR;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPollSetAddSocket(pSockInt->hPollSet, pSockInt->hSocket,
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync pSockInt->fEventsOld, VDSOCKET_POLL_ID_SOCKET);
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync }
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (RT_SUCCESS(rc))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VINF_SUCCESS;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTTcpClientCloseEx(pSockInt->hSocket, false /*fGracefulShutdown*/);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return rc;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
c28fa006ba669ad8f26ae31d00a338379c04ea1bvboxsync/** @copydoc VDINTERFACETCPNET::pfnClientClose */
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsyncstatic DECLCALLBACK(int) drvvdTcpClientClose(VDSOCKET Sock)
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync{
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync int rc = VINF_SUCCESS;
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync if (pSockInt->hPollSet != NIL_RTPOLLSET)
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync {
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync rc = RTPollSetRemove(pSockInt->hPollSet, VDSOCKET_POLL_ID_SOCKET);
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync AssertRC(rc);
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync }
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync rc = RTTcpClientCloseEx(pSockInt->hSocket, false /*fGracefulShutdown*/);
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync pSockInt->hSocket = NIL_RTSOCKET;
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync return rc;
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync}
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync/** @copydoc VDINTERFACETCPNET::pfnIsClientConnected */
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsyncstatic DECLCALLBACK(bool) drvvdTcpIsClientConnected(VDSOCKET Sock)
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync{
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync return pSockInt->hSocket != NIL_RTSOCKET;
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync}
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync/** @copydoc VDINTERFACETCPNET::pfnSelectOne */
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsyncstatic DECLCALLBACK(int) drvvdTcpSelectOne(VDSOCKET Sock, RTMSINTERVAL cMillies)
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync{
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync return RTTcpSelectOne(pSockInt->hSocket, cMillies);
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync}
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync/** @copydoc VDINTERFACETCPNET::pfnRead */
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsyncstatic DECLCALLBACK(int) drvvdTcpRead(VDSOCKET Sock, void *pvBuffer, size_t cbBuffer, size_t *pcbRead)
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync{
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync
38ceb5cfc147448ea4087a1f5971cb8fb4434bd9vboxsync return RTTcpRead(pSockInt->hSocket, pvBuffer, cbBuffer, pcbRead);
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync}
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync/** @copydoc VDINTERFACETCPNET::pfnWrite */
27b178e99b06a68ef52353b15bc647674d2006bcvboxsyncstatic DECLCALLBACK(int) drvvdTcpWrite(VDSOCKET Sock, const void *pvBuffer, size_t cbBuffer)
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync{
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync return RTTcpWrite(pSockInt->hSocket, pvBuffer, cbBuffer);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync/** @copydoc VDINTERFACETCPNET::pfnSgWrite */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdTcpSgWrite(VDSOCKET Sock, PCRTSGBUF pSgBuf)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return RTTcpSgWrite(pSockInt->hSocket, pSgBuf);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync}
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync/** @copydoc VDINTERFACETCPNET::pfnReadNB */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsyncstatic DECLCALLBACK(int) drvvdTcpReadNB(VDSOCKET Sock, void *pvBuffer, size_t cbBuffer, size_t *pcbRead)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTTcpReadNB(pSockInt->hSocket, pvBuffer, cbBuffer, pcbRead);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/** @copydoc VDINTERFACETCPNET::pfnWriteNB */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdTcpWriteNB(VDSOCKET Sock, const void *pvBuffer, size_t cbBuffer, size_t *pcbWritten)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTTcpWriteNB(pSockInt->hSocket, pvBuffer, cbBuffer, pcbWritten);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/** @copydoc VDINTERFACETCPNET::pfnSgWriteNB */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdTcpSgWriteNB(VDSOCKET Sock, PRTSGBUF pSgBuf, size_t *pcbWritten)
efff36b306e370346025647a158689021df2e1d1vboxsync{
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
efff36b306e370346025647a158689021df2e1d1vboxsync
efff36b306e370346025647a158689021df2e1d1vboxsync return RTTcpSgWriteNB(pSockInt->hSocket, pSgBuf, pcbWritten);
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync}
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync/** @copydoc VDINTERFACETCPNET::pfnFlush */
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsyncstatic DECLCALLBACK(int) drvvdTcpFlush(VDSOCKET Sock)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync{
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync return RTTcpFlush(pSockInt->hSocket);
efff36b306e370346025647a158689021df2e1d1vboxsync}
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync/** @copydoc VDINTERFACETCPNET::pfnSetSendCoalescing */
a3369a746b56a8966dd78619f4d191c9662f400dvboxsyncstatic DECLCALLBACK(int) drvvdTcpSetSendCoalescing(VDSOCKET Sock, bool fEnable)
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync{
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync return RTTcpSetSendCoalescing(pSockInt->hSocket, fEnable);
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync}
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync/** @copydoc VDINTERFACETCPNET::pfnGetLocalAddress */
27b178e99b06a68ef52353b15bc647674d2006bcvboxsyncstatic DECLCALLBACK(int) drvvdTcpGetLocalAddress(VDSOCKET Sock, PRTNETADDR pAddr)
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync{
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync return RTTcpGetLocalAddress(pSockInt->hSocket, pAddr);
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync}
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync/** @copydoc VDINTERFACETCPNET::pfnGetPeerAddress */
27b178e99b06a68ef52353b15bc647674d2006bcvboxsyncstatic DECLCALLBACK(int) drvvdTcpGetPeerAddress(VDSOCKET Sock, PRTNETADDR pAddr)
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync{
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync return RTTcpGetPeerAddress(pSockInt->hSocket, pAddr);
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync}
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
27b178e99b06a68ef52353b15bc647674d2006bcvboxsyncstatic int drvvdTcpSelectOneExPoll(VDSOCKET Sock, uint32_t fEvents,
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync uint32_t *pfEvents, RTMSINTERVAL cMillies)
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync{
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync int rc = VINF_SUCCESS;
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync uint32_t id = 0;
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync uint32_t fEventsRecv = 0;
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
27b178e99b06a68ef52353b15bc647674d2006bcvboxsync
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync *pfEvents = 0;
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync if ( pSockInt->fEventsOld != fEvents
a3369a746b56a8966dd78619f4d191c9662f400dvboxsync && pSockInt->hSocket != NIL_RTSOCKET)
51fe8789a74f6c118894aaa12eb69ec155386dbdvboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync uint32_t fPollEvents = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync if (fEvents & VD_INTERFACETCPNET_EVT_READ)
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync fPollEvents |= RTPOLL_EVT_READ;
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync if (fEvents & VD_INTERFACETCPNET_EVT_WRITE)
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync fPollEvents |= RTPOLL_EVT_WRITE;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (fEvents & VD_INTERFACETCPNET_EVT_ERROR)
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync fPollEvents |= RTPOLL_EVT_ERROR;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync rc = RTPollSetEventsChange(pSockInt->hPollSet, VDSOCKET_POLL_ID_SOCKET, fPollEvents);
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync if (RT_FAILURE(rc))
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync return rc;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync pSockInt->fEventsOld = fEvents;
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync ASMAtomicXchgBool(&pSockInt->fWaiting, true);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (ASMAtomicXchgBool(&pSockInt->fWokenUp, false))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync ASMAtomicXchgBool(&pSockInt->fWaiting, false);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync return VERR_INTERRUPTED;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPoll(pSockInt->hPollSet, cMillies, &fEventsRecv, &id);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync Assert(RT_SUCCESS(rc) || rc == VERR_TIMEOUT);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
0ccdfa1953b2f57311fb9ec01a2baf5e1e366f5avboxsync ASMAtomicXchgBool(&pSockInt->fWaiting, false);
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (RT_SUCCESS(rc))
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync {
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (id == VDSOCKET_POLL_ID_SOCKET)
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync {
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync fEventsRecv &= RTPOLL_EVT_VALID_MASK;
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync
670b83d458bceb92123155b5b47a39b9d24e3266vboxsync if (fEventsRecv & RTPOLL_EVT_READ)
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync *pfEvents |= VD_INTERFACETCPNET_EVT_READ;
670b83d458bceb92123155b5b47a39b9d24e3266vboxsync if (fEventsRecv & RTPOLL_EVT_WRITE)
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync *pfEvents |= VD_INTERFACETCPNET_EVT_WRITE;
9c149a2789022f5011e88fb62f02a1cc8068e88fvboxsync if (fEventsRecv & RTPOLL_EVT_ERROR)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync *pfEvents |= VD_INTERFACETCPNET_EVT_ERROR;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync else
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync size_t cbRead = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync uint8_t abBuf[10];
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync Assert(id == VDSOCKET_POLL_ID_PIPE);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync Assert((fEventsRecv & RTPOLL_EVT_VALID_MASK) == RTPOLL_EVT_READ);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* We got interrupted, drain the pipe. */
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync rc = RTPipeRead(pSockInt->hPipeR, abBuf, sizeof(abBuf), &cbRead);
9496f2d398b49813176939d7a339ae513d5175efvboxsync AssertRC(rc);
9496f2d398b49813176939d7a339ae513d5175efvboxsync
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync ASMAtomicXchgBool(&pSockInt->fWokenUp, false);
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync
9496f2d398b49813176939d7a339ae513d5175efvboxsync rc = VERR_INTERRUPTED;
9496f2d398b49813176939d7a339ae513d5175efvboxsync }
5d0d754550d06b7d59a935e59caaf814462d53ccvboxsync }
9496f2d398b49813176939d7a339ae513d5175efvboxsync
d80c85a1bc7317da7d0cd1254fae6a20db039c8cvboxsync return rc;
9496f2d398b49813176939d7a339ae513d5175efvboxsync}
9496f2d398b49813176939d7a339ae513d5175efvboxsync
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync/** @copydoc VDINTERFACETCPNET::pfnSelectOneEx */
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsyncstatic DECLCALLBACK(int) drvvdTcpSelectOneExNoPoll(VDSOCKET Sock, uint32_t fEvents,
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync uint32_t *pfEvents, RTMSINTERVAL cMillies)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync{
548ca31b6b47c36bacce49bed3339cb8075b9681vboxsync int rc = VINF_SUCCESS;
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
d1bffa158f98ff3c18f7d085e7372c9ea00e9a43vboxsync
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync *pfEvents = 0;
d1bffa158f98ff3c18f7d085e7372c9ea00e9a43vboxsync
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync ASMAtomicXchgBool(&pSockInt->fWaiting, true);
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync if (ASMAtomicXchgBool(&pSockInt->fWokenUp, false))
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync {
f35c44bfc9e1036d0cb376fb144cdae416c7ef3avboxsync ASMAtomicXchgBool(&pSockInt->fWaiting, false);
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync return VERR_INTERRUPTED;
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync }
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if ( pSockInt->hSocket == NIL_RTSOCKET
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync || !fEvents)
927320c7f81d3acdbccb5f3fea7548b4b7184b98vboxsync {
6b223679d40f5e57e55e867e806a9f194e2cde12vboxsync /*
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync * Only the pipe is configured or the caller doesn't wait for a socket event,
9496f2d398b49813176939d7a339ae513d5175efvboxsync * wait until there is something to read from the pipe.
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync */
9496f2d398b49813176939d7a339ae513d5175efvboxsync size_t cbRead = 0;
9496f2d398b49813176939d7a339ae513d5175efvboxsync char ch = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPipeReadBlocking(pSockInt->hPipeR, &ch, 1, &cbRead);
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync if (RT_SUCCESS(rc))
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync {
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync Assert(cbRead == 1);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = VERR_INTERRUPTED;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync ASMAtomicXchgBool(&pSockInt->fWokenUp, false);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync else
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync uint32_t fSelectEvents = 0;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (fEvents & VD_INTERFACETCPNET_EVT_READ)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync fSelectEvents |= RTSOCKET_EVT_READ;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (fEvents & VD_INTERFACETCPNET_EVT_WRITE)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync fSelectEvents |= RTSOCKET_EVT_WRITE;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (fEvents & VD_INTERFACETCPNET_EVT_ERROR)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync fSelectEvents |= RTSOCKET_EVT_ERROR;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync if (fEvents & VD_INTERFACETCPNET_HINT_INTERRUPT)
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync {
b7a5b3f9f9ecce32ddacf8404c625ce0451bbdc1vboxsync uint32_t fEventsRecv = 0;
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync /* Make sure the socket is not in the pollset. */
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync rc = RTPollSetRemove(pSockInt->hPollSet, VDSOCKET_POLL_ID_SOCKET);
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync Assert(RT_SUCCESS(rc) || rc == VERR_POLL_HANDLE_ID_NOT_FOUND);
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync
4a23807f02e9920d92c8449bd93d84501add460avboxsync for (;;)
d9d5fbda1b8f7a6f7fae555db60d0e636fd03af8vboxsync {
4a23807f02e9920d92c8449bd93d84501add460avboxsync uint32_t id = 0;
fe813b3594039ba864493438e78ee0e7132bc445vboxsync rc = RTPoll(pSockInt->hPollSet, 5, &fEvents, &id);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (rc == VERR_TIMEOUT)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* Check the socket. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTTcpSelectOneEx(pSockInt->hSocket, fSelectEvents, &fEventsRecv, 0);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (RT_SUCCESS(rc))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (fEventsRecv & RTSOCKET_EVT_READ)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync *pfEvents |= VD_INTERFACETCPNET_EVT_READ;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync if (fEventsRecv & RTSOCKET_EVT_WRITE)
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync *pfEvents |= VD_INTERFACETCPNET_EVT_WRITE;
5b465a7c1237993faf8bb50120d247f3f0319adavboxsync if (fEventsRecv & RTSOCKET_EVT_ERROR)
09127e6ed46502ff8a6a521713ee8ace53667683vboxsync *pfEvents |= VD_INTERFACETCPNET_EVT_ERROR;
ad77e3ec3cde24263bc7537575f5cae442bee3b1vboxsync break; /* Quit */
ad77e3ec3cde24263bc7537575f5cae442bee3b1vboxsync }
ad77e3ec3cde24263bc7537575f5cae442bee3b1vboxsync else if (rc != VERR_TIMEOUT)
ad77e3ec3cde24263bc7537575f5cae442bee3b1vboxsync break;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync else if (RT_SUCCESS(rc))
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync size_t cbRead = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync uint8_t abBuf[10];
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync Assert(id == VDSOCKET_POLL_ID_PIPE);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync Assert((fEventsRecv & RTPOLL_EVT_VALID_MASK) == RTPOLL_EVT_READ);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /* We got interrupted, drain the pipe. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPipeRead(pSockInt->hPipeR, abBuf, sizeof(abBuf), &cbRead);
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync AssertRC(rc);
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync ASMAtomicXchgBool(&pSockInt->fWokenUp, false);
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync
ad77e3ec3cde24263bc7537575f5cae442bee3b1vboxsync rc = VERR_INTERRUPTED;
ad77e3ec3cde24263bc7537575f5cae442bee3b1vboxsync break;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync else
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync break;
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync }
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync else /* The caller waits for a socket event. */
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync {
8083a259c13e6e26e56ca2582edbad4a8cfac25avboxsync uint32_t fEventsRecv = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync
1bf495e3eec00dd79cecb6b36ef2a97f422c3737vboxsync /* Loop until we got woken up or a socket event occurred. */
cba6719bd64ec749967bbe931230452664109857vboxsync for (;;)
cba6719bd64ec749967bbe931230452664109857vboxsync {
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync /** @todo find an adaptive wait algorithm based on the
9496f2d398b49813176939d7a339ae513d5175efvboxsync * number of wakeups in the past. */
9496f2d398b49813176939d7a339ae513d5175efvboxsync rc = RTTcpSelectOneEx(pSockInt->hSocket, fSelectEvents, &fEventsRecv, 5);
cba6719bd64ec749967bbe931230452664109857vboxsync if (rc == VERR_TIMEOUT)
cba6719bd64ec749967bbe931230452664109857vboxsync {
cba6719bd64ec749967bbe931230452664109857vboxsync /* Check if there is an event pending. */
cba6719bd64ec749967bbe931230452664109857vboxsync size_t cbRead = 0;
cba6719bd64ec749967bbe931230452664109857vboxsync char ch = 0;
a0240ff4f7663045c848fdbc192ea3d4d9f70a11vboxsync rc = RTPipeRead(pSockInt->hPipeR, &ch, 1, &cbRead);
cba6719bd64ec749967bbe931230452664109857vboxsync if (RT_SUCCESS(rc) && rc != VINF_TRY_AGAIN)
{
Assert(cbRead == 1);
rc = VERR_INTERRUPTED;
ASMAtomicXchgBool(&pSockInt->fWokenUp, false);
break; /* Quit */
}
else
Assert(rc == VINF_TRY_AGAIN);
}
else if (RT_SUCCESS(rc))
{
if (fEventsRecv & RTSOCKET_EVT_READ)
*pfEvents |= VD_INTERFACETCPNET_EVT_READ;
if (fEventsRecv & RTSOCKET_EVT_WRITE)
*pfEvents |= VD_INTERFACETCPNET_EVT_WRITE;
if (fEventsRecv & RTSOCKET_EVT_ERROR)
*pfEvents |= VD_INTERFACETCPNET_EVT_ERROR;
break; /* Quit */
}
else
break;
}
}
}
ASMAtomicXchgBool(&pSockInt->fWaiting, false);
return rc;
}
/** @copydoc VDINTERFACETCPNET::pfnPoke */
static DECLCALLBACK(int) drvvdTcpPoke(VDSOCKET Sock)
{
int rc = VINF_SUCCESS;
size_t cbWritten = 0;
PVDSOCKETINT pSockInt = (PVDSOCKETINT)Sock;
ASMAtomicXchgBool(&pSockInt->fWokenUp, true);
if (ASMAtomicReadBool(&pSockInt->fWaiting))
{
rc = RTPipeWrite(pSockInt->hPipeW, "", 1, &cbWritten);
Assert(RT_SUCCESS(rc) || cbWritten == 0);
}
return VINF_SUCCESS;
}
/*******************************************************************************
* Media interface methods *
*******************************************************************************/
/** @copydoc PDMIMEDIA::pfnRead */
static DECLCALLBACK(int) drvvdRead(PPDMIMEDIA pInterface,
uint64_t off, void *pvBuf, size_t cbRead)
{
int rc = VINF_SUCCESS;
LogFlowFunc(("off=%#llx pvBuf=%p cbRead=%d\n", off, pvBuf, cbRead));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
if (!pThis->fBootAccelActive)
rc = VDRead(pThis->pDisk, off, pvBuf, cbRead);
else
{
/* Can we serve the request from the buffer? */
if ( off >= pThis->offDisk
&& off - pThis->offDisk < pThis->cbDataValid)
{
size_t cbToCopy = RT_MIN(cbRead, pThis->offDisk + pThis->cbDataValid - off);
memcpy(pvBuf, pThis->pbData + (off - pThis->offDisk), cbToCopy);
cbRead -= cbToCopy;
off += cbToCopy;
pvBuf = (char *)pvBuf + cbToCopy;
}
if ( cbRead > 0
&& cbRead < pThis->cbBootAccelBuffer)
{
/* Increase request to the buffer size and read. */
pThis->cbDataValid = RT_MIN(pThis->cbDisk - off, pThis->cbBootAccelBuffer);
pThis->offDisk = off;
rc = VDRead(pThis->pDisk, off, pThis->pbData, pThis->cbDataValid);
if (RT_FAILURE(rc))
pThis->cbDataValid = 0;
else
memcpy(pvBuf, pThis->pbData, cbRead);
}
else if (cbRead >= pThis->cbBootAccelBuffer)
{
pThis->fBootAccelActive = false; /* Deactiviate */
}
}
if (RT_SUCCESS(rc))
Log2(("%s: off=%#llx pvBuf=%p cbRead=%d %.*Rhxd\n", __FUNCTION__,
off, pvBuf, cbRead, cbRead, pvBuf));
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc PDMIMEDIA::pfnWrite */
static DECLCALLBACK(int) drvvdWrite(PPDMIMEDIA pInterface,
uint64_t off, const void *pvBuf,
size_t cbWrite)
{
LogFlowFunc(("off=%#llx pvBuf=%p cbWrite=%d\n", off, pvBuf, cbWrite));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
Log2(("%s: off=%#llx pvBuf=%p cbWrite=%d %.*Rhxd\n", __FUNCTION__,
off, pvBuf, cbWrite, cbWrite, pvBuf));
/* Invalidate any buffer if boot acceleration is enabled. */
if (pThis->fBootAccelActive)
{
pThis->cbDataValid = 0;
pThis->offDisk = 0;
}
int rc = VDWrite(pThis->pDisk, off, pvBuf, cbWrite);
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc PDMIMEDIA::pfnFlush */
static DECLCALLBACK(int) drvvdFlush(PPDMIMEDIA pInterface)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
int rc = VDFlush(pThis->pDisk);
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc PDMIMEDIA::pfnMerge */
static DECLCALLBACK(int) drvvdMerge(PPDMIMEDIA pInterface,
PFNSIMPLEPROGRESS pfnProgress,
void *pvUser)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
int rc = VINF_SUCCESS;
/* Note: There is an unavoidable race between destruction and another
* thread invoking this function. This is handled safely and gracefully by
* atomically invalidating the lock handle in drvvdDestruct. */
int rc2 = RTSemFastMutexRequest(pThis->MergeCompleteMutex);
AssertRC(rc2);
if (RT_SUCCESS(rc2) && pThis->fMergePending)
{
/* Take shortcut: PFNSIMPLEPROGRESS is exactly the same type as
* PFNVDPROGRESS, so there's no need for a conversion function. */
/** @todo maybe introduce a conversion which limits update frequency. */
PVDINTERFACE pVDIfsOperation = NULL;
VDINTERFACEPROGRESS VDIfProgress;
VDIfProgress.pfnProgress = pfnProgress;
rc2 = VDInterfaceAdd(&VDIfProgress.Core, "DrvVD_VDIProgress", VDINTERFACETYPE_PROGRESS,
pvUser, sizeof(VDINTERFACEPROGRESS), &pVDIfsOperation);
AssertRC(rc2);
pThis->fMergePending = false;
rc = VDMerge(pThis->pDisk, pThis->uMergeSource,
pThis->uMergeTarget, pVDIfsOperation);
}
rc2 = RTSemFastMutexRelease(pThis->MergeCompleteMutex);
AssertRC(rc2);
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc PDMIMEDIA::pfnGetSize */
static DECLCALLBACK(uint64_t) drvvdGetSize(PPDMIMEDIA pInterface)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
uint64_t cb = VDGetSize(pThis->pDisk, VD_LAST_IMAGE);
LogFlowFunc(("returns %#llx (%llu)\n", cb, cb));
return cb;
}
/** @copydoc PDMIMEDIA::pfnIsReadOnly */
static DECLCALLBACK(bool) drvvdIsReadOnly(PPDMIMEDIA pInterface)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
bool f = VDIsReadOnly(pThis->pDisk);
LogFlowFunc(("returns %d\n", f));
return f;
}
/** @copydoc PDMIMEDIA::pfnBiosGetPCHSGeometry */
static DECLCALLBACK(int) drvvdBiosGetPCHSGeometry(PPDMIMEDIA pInterface,
PPDMMEDIAGEOMETRY pPCHSGeometry)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
VDGEOMETRY geo;
int rc = VDGetPCHSGeometry(pThis->pDisk, VD_LAST_IMAGE, &geo);
if (RT_SUCCESS(rc))
{
pPCHSGeometry->cCylinders = geo.cCylinders;
pPCHSGeometry->cHeads = geo.cHeads;
pPCHSGeometry->cSectors = geo.cSectors;
}
else
{
LogFunc(("geometry not available.\n"));
rc = VERR_PDM_GEOMETRY_NOT_SET;
}
LogFlowFunc(("returns %Rrc (CHS=%d/%d/%d)\n",
rc, pPCHSGeometry->cCylinders, pPCHSGeometry->cHeads, pPCHSGeometry->cSectors));
return rc;
}
/** @copydoc PDMIMEDIA::pfnBiosSetPCHSGeometry */
static DECLCALLBACK(int) drvvdBiosSetPCHSGeometry(PPDMIMEDIA pInterface,
PCPDMMEDIAGEOMETRY pPCHSGeometry)
{
LogFlowFunc(("CHS=%d/%d/%d\n",
pPCHSGeometry->cCylinders, pPCHSGeometry->cHeads, pPCHSGeometry->cSectors));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
VDGEOMETRY geo;
geo.cCylinders = pPCHSGeometry->cCylinders;
geo.cHeads = pPCHSGeometry->cHeads;
geo.cSectors = pPCHSGeometry->cSectors;
int rc = VDSetPCHSGeometry(pThis->pDisk, VD_LAST_IMAGE, &geo);
if (rc == VERR_VD_GEOMETRY_NOT_SET)
rc = VERR_PDM_GEOMETRY_NOT_SET;
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc PDMIMEDIA::pfnBiosGetLCHSGeometry */
static DECLCALLBACK(int) drvvdBiosGetLCHSGeometry(PPDMIMEDIA pInterface,
PPDMMEDIAGEOMETRY pLCHSGeometry)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
VDGEOMETRY geo;
int rc = VDGetLCHSGeometry(pThis->pDisk, VD_LAST_IMAGE, &geo);
if (RT_SUCCESS(rc))
{
pLCHSGeometry->cCylinders = geo.cCylinders;
pLCHSGeometry->cHeads = geo.cHeads;
pLCHSGeometry->cSectors = geo.cSectors;
}
else
{
LogFunc(("geometry not available.\n"));
rc = VERR_PDM_GEOMETRY_NOT_SET;
}
LogFlowFunc(("returns %Rrc (CHS=%d/%d/%d)\n",
rc, pLCHSGeometry->cCylinders, pLCHSGeometry->cHeads, pLCHSGeometry->cSectors));
return rc;
}
/** @copydoc PDMIMEDIA::pfnBiosSetLCHSGeometry */
static DECLCALLBACK(int) drvvdBiosSetLCHSGeometry(PPDMIMEDIA pInterface,
PCPDMMEDIAGEOMETRY pLCHSGeometry)
{
LogFlowFunc(("CHS=%d/%d/%d\n",
pLCHSGeometry->cCylinders, pLCHSGeometry->cHeads, pLCHSGeometry->cSectors));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
VDGEOMETRY geo;
geo.cCylinders = pLCHSGeometry->cCylinders;
geo.cHeads = pLCHSGeometry->cHeads;
geo.cSectors = pLCHSGeometry->cSectors;
int rc = VDSetLCHSGeometry(pThis->pDisk, VD_LAST_IMAGE, &geo);
if (rc == VERR_VD_GEOMETRY_NOT_SET)
rc = VERR_PDM_GEOMETRY_NOT_SET;
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc PDMIMEDIA::pfnGetUuid */
static DECLCALLBACK(int) drvvdGetUuid(PPDMIMEDIA pInterface, PRTUUID pUuid)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
int rc = VDGetUuid(pThis->pDisk, 0, pUuid);
LogFlowFunc(("returns %Rrc ({%RTuuid})\n", rc, pUuid));
return rc;
}
static DECLCALLBACK(int) drvvdDiscard(PPDMIMEDIA pInterface, PCRTRANGE paRanges, unsigned cRanges)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMIMEDIA_2_VBOXDISK(pInterface);
int rc = VDDiscardRanges(pThis->pDisk, paRanges, cRanges);
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/*******************************************************************************
* Async Media interface methods *
*******************************************************************************/
static void drvvdAsyncReqComplete(void *pvUser1, void *pvUser2, int rcReq)
{
PVBOXDISK pThis = (PVBOXDISK)pvUser1;
if (!pThis->pBlkCache)
{
int rc = pThis->pDrvMediaAsyncPort->pfnTransferCompleteNotify(pThis->pDrvMediaAsyncPort,
pvUser2, rcReq);
AssertRC(rc);
}
else
PDMR3BlkCacheIoXferComplete(pThis->pBlkCache, (PPDMBLKCACHEIOXFER)pvUser2, rcReq);
}
static DECLCALLBACK(int) drvvdStartRead(PPDMIMEDIAASYNC pInterface, uint64_t uOffset,
PCRTSGSEG paSeg, unsigned cSeg,
size_t cbRead, void *pvUser)
{
LogFlowFunc(("uOffset=%#llx paSeg=%#p cSeg=%u cbRead=%d pvUser=%#p\n",
uOffset, paSeg, cSeg, cbRead, pvUser));
int rc = VINF_SUCCESS;
PVBOXDISK pThis = PDMIMEDIAASYNC_2_VBOXDISK(pInterface);
pThis->fBootAccelActive = false;
RTSGBUF SgBuf;
RTSgBufInit(&SgBuf, paSeg, cSeg);
if (!pThis->pBlkCache)
rc = VDAsyncRead(pThis->pDisk, uOffset, cbRead, &SgBuf,
drvvdAsyncReqComplete, pThis, pvUser);
else
{
rc = PDMR3BlkCacheRead(pThis->pBlkCache, uOffset, &SgBuf, cbRead, pvUser);
if (rc == VINF_AIO_TASK_PENDING)
rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
else if (rc == VINF_SUCCESS)
rc = VINF_VD_ASYNC_IO_FINISHED;
}
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
static DECLCALLBACK(int) drvvdStartWrite(PPDMIMEDIAASYNC pInterface, uint64_t uOffset,
PCRTSGSEG paSeg, unsigned cSeg,
size_t cbWrite, void *pvUser)
{
LogFlowFunc(("uOffset=%#llx paSeg=%#p cSeg=%u cbWrite=%d pvUser=%#p\n",
uOffset, paSeg, cSeg, cbWrite, pvUser));
int rc = VINF_SUCCESS;
PVBOXDISK pThis = PDMIMEDIAASYNC_2_VBOXDISK(pInterface);
pThis->fBootAccelActive = false;
RTSGBUF SgBuf;
RTSgBufInit(&SgBuf, paSeg, cSeg);
if (!pThis->pBlkCache)
rc = VDAsyncWrite(pThis->pDisk, uOffset, cbWrite, &SgBuf,
drvvdAsyncReqComplete, pThis, pvUser);
else
{
rc = PDMR3BlkCacheWrite(pThis->pBlkCache, uOffset, &SgBuf, cbWrite, pvUser);
if (rc == VINF_AIO_TASK_PENDING)
rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
else if (rc == VINF_SUCCESS)
rc = VINF_VD_ASYNC_IO_FINISHED;
}
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
static DECLCALLBACK(int) drvvdStartFlush(PPDMIMEDIAASYNC pInterface, void *pvUser)
{
LogFlowFunc(("pvUser=%#p\n", pvUser));
int rc = VINF_SUCCESS;
PVBOXDISK pThis = PDMIMEDIAASYNC_2_VBOXDISK(pInterface);
if (!pThis->pBlkCache)
rc = VDAsyncFlush(pThis->pDisk, drvvdAsyncReqComplete, pThis, pvUser);
else
{
rc = PDMR3BlkCacheFlush(pThis->pBlkCache, pvUser);
if (rc == VINF_AIO_TASK_PENDING)
rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
else if (rc == VINF_SUCCESS)
rc = VINF_VD_ASYNC_IO_FINISHED;
}
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
static DECLCALLBACK(int) drvvdStartDiscard(PPDMIMEDIAASYNC pInterface, PCRTRANGE paRanges,
unsigned cRanges, void *pvUser)
{
int rc = VINF_SUCCESS;
PVBOXDISK pThis = PDMIMEDIAASYNC_2_VBOXDISK(pInterface);
LogFlowFunc(("paRanges=%#p cRanges=%u pvUser=%#p\n",
paRanges, cRanges, pvUser));
if (!pThis->pBlkCache)
rc = VDAsyncDiscardRanges(pThis->pDisk, paRanges, cRanges, drvvdAsyncReqComplete,
pThis, pvUser);
else
{
rc = PDMR3BlkCacheDiscard(pThis->pBlkCache, paRanges, cRanges, pvUser);
if (rc == VINF_AIO_TASK_PENDING)
rc = VERR_VD_ASYNC_IO_IN_PROGRESS;
else if (rc == VINF_SUCCESS)
rc = VINF_VD_ASYNC_IO_FINISHED;
}
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/** @copydoc FNPDMBLKCACHEXFERCOMPLETEDRV */
static void drvvdBlkCacheXferComplete(PPDMDRVINS pDrvIns, void *pvUser, int rcReq)
{
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
int rc = pThis->pDrvMediaAsyncPort->pfnTransferCompleteNotify(pThis->pDrvMediaAsyncPort,
pvUser, rcReq);
AssertRC(rc);
}
/** @copydoc FNPDMBLKCACHEXFERENQUEUEDRV */
static int drvvdBlkCacheXferEnqueue(PPDMDRVINS pDrvIns,
PDMBLKCACHEXFERDIR enmXferDir,
uint64_t off, size_t cbXfer,
PCRTSGBUF pcSgBuf, PPDMBLKCACHEIOXFER hIoXfer)
{
int rc = VINF_SUCCESS;
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
switch (enmXferDir)
{
case PDMBLKCACHEXFERDIR_READ:
rc = VDAsyncRead(pThis->pDisk, off, cbXfer, pcSgBuf, drvvdAsyncReqComplete,
pThis, hIoXfer);
break;
case PDMBLKCACHEXFERDIR_WRITE:
rc = VDAsyncWrite(pThis->pDisk, off, cbXfer, pcSgBuf, drvvdAsyncReqComplete,
pThis, hIoXfer);
break;
case PDMBLKCACHEXFERDIR_FLUSH:
rc = VDAsyncFlush(pThis->pDisk, drvvdAsyncReqComplete, pThis, hIoXfer);
break;
default:
AssertMsgFailed(("Invalid transfer type %d\n", enmXferDir));
rc = VERR_INVALID_PARAMETER;
}
if (rc == VINF_VD_ASYNC_IO_FINISHED)
PDMR3BlkCacheIoXferComplete(pThis->pBlkCache, hIoXfer, VINF_SUCCESS);
else if (RT_FAILURE(rc) && rc != VERR_VD_ASYNC_IO_IN_PROGRESS)
PDMR3BlkCacheIoXferComplete(pThis->pBlkCache, hIoXfer, rc);
return VINF_SUCCESS;
}
/** @copydoc FNPDMBLKCACHEXFERENQUEUEDISCARDDRV */
static int drvvdBlkCacheXferEnqueueDiscard(PPDMDRVINS pDrvIns, PCRTRANGE paRanges,
unsigned cRanges, PPDMBLKCACHEIOXFER hIoXfer)
{
int rc = VINF_SUCCESS;
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
rc = VDAsyncDiscardRanges(pThis->pDisk, paRanges, cRanges,
drvvdAsyncReqComplete, pThis, hIoXfer);
if (rc == VINF_VD_ASYNC_IO_FINISHED)
PDMR3BlkCacheIoXferComplete(pThis->pBlkCache, hIoXfer, VINF_SUCCESS);
else if (RT_FAILURE(rc) && rc != VERR_VD_ASYNC_IO_IN_PROGRESS)
PDMR3BlkCacheIoXferComplete(pThis->pBlkCache, hIoXfer, rc);
return VINF_SUCCESS;
}
/*******************************************************************************
* Base interface methods *
*******************************************************************************/
/**
* @interface_method_impl{PDMIBASE,pfnQueryInterface}
*/
static DECLCALLBACK(void *) drvvdQueryInterface(PPDMIBASE pInterface, const char *pszIID)
{
PPDMDRVINS pDrvIns = PDMIBASE_2_PDMDRV(pInterface);
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
PDMIBASE_RETURN_INTERFACE(pszIID, PDMIBASE, &pDrvIns->IBase);
PDMIBASE_RETURN_INTERFACE(pszIID, PDMIMEDIA, &pThis->IMedia);
PDMIBASE_RETURN_INTERFACE(pszIID, PDMIMEDIAASYNC, pThis->fAsyncIOSupported ? &pThis->IMediaAsync : NULL);
return NULL;
}
/*******************************************************************************
* Saved state notification methods *
*******************************************************************************/
/**
* Load done callback for re-opening the image writable during teleportation.
*
* This is called both for successful and failed load runs, we only care about
* successful ones.
*
* @returns VBox status code.
* @param pDrvIns The driver instance.
* @param pSSM The saved state handle.
*/
static DECLCALLBACK(int) drvvdLoadDone(PPDMDRVINS pDrvIns, PSSMHANDLE pSSM)
{
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
Assert(!pThis->fErrorUseRuntime);
/* Drop out if we don't have any work to do or if it's a failed load. */
if ( !pThis->fTempReadOnly
|| RT_FAILURE(SSMR3HandleGetStatus(pSSM)))
return VINF_SUCCESS;
int rc = drvvdSetWritable(pThis);
if (RT_FAILURE(rc)) /** @todo does the bugger set any errors? */
return SSMR3SetLoadError(pSSM, rc, RT_SRC_POS,
N_("Failed to write lock the images"));
return VINF_SUCCESS;
}
/*******************************************************************************
* Driver methods *
*******************************************************************************/
/**
* VM resume notification that we use to undo what the temporary read-only image
* mode set by drvvdSuspend.
*
* Also switch to runtime error mode if we're resuming after a state load
* without having been powered on first.
*
* @param pDrvIns The driver instance data.
*
* @todo The VMSetError vs VMSetRuntimeError mess must be fixed elsewhere,
* we're making assumptions about Main behavior here!
*/
static DECLCALLBACK(void) drvvdResume(PPDMDRVINS pDrvIns)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
drvvdSetWritable(pThis);
pThis->fErrorUseRuntime = true;
if (pThis->pBlkCache)
{
int rc = PDMR3BlkCacheResume(pThis->pBlkCache);
AssertRC(rc);
}
}
/**
* The VM is being suspended, temporarily change to read-only image mode.
*
* This is important for several reasons:
* -# It makes sure that there are no pending writes to the image. Most
* backends implements this by closing and reopening the image in read-only
* mode.
* -# It allows Main to read the images during snapshotting without having
* to account for concurrent writes.
* -# This is essential for making teleportation targets sharing images work
* right. Both with regards to caching and with regards to file sharing
* locks (RTFILE_O_DENY_*). (See also drvvdLoadDone.)
*
* @param pDrvIns The driver instance data.
*/
static DECLCALLBACK(void) drvvdSuspend(PPDMDRVINS pDrvIns)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
if (pThis->pBlkCache)
{
int rc = PDMR3BlkCacheSuspend(pThis->pBlkCache);
AssertRC(rc);
}
drvvdSetReadonly(pThis);
}
/**
* VM PowerOn notification for undoing the TempReadOnly config option and
* changing to runtime error mode.
*
* @param pDrvIns The driver instance data.
*
* @todo The VMSetError vs VMSetRuntimeError mess must be fixed elsewhere,
* we're making assumptions about Main behavior here!
*/
static DECLCALLBACK(void) drvvdPowerOn(PPDMDRVINS pDrvIns)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
drvvdSetWritable(pThis);
pThis->fErrorUseRuntime = true;
}
/**
* @copydoc FNPDMDRVRESET
*/
static DECLCALLBACK(void) drvvdReset(PPDMDRVINS pDrvIns)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
if (pThis->pBlkCache)
{
int rc = PDMR3BlkCacheClear(pThis->pBlkCache);
AssertRC(rc);
}
if (pThis->fBootAccelEnabled)
{
pThis->fBootAccelActive = true;
pThis->cbDataValid = 0;
pThis->offDisk = 0;
}
}
/**
* @copydoc FNPDMDRVDESTRUCT
*/
static DECLCALLBACK(void) drvvdDestruct(PPDMDRVINS pDrvIns)
{
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
LogFlowFunc(("\n"));
PDMDRV_CHECK_VERSIONS_RETURN_VOID(pDrvIns);
RTSEMFASTMUTEX mutex;
ASMAtomicXchgHandle(&pThis->MergeCompleteMutex, NIL_RTSEMFASTMUTEX, &mutex);
if (mutex != NIL_RTSEMFASTMUTEX)
{
/* Request the semaphore to wait until a potentially running merge
* operation has been finished. */
int rc = RTSemFastMutexRequest(mutex);
AssertRC(rc);
pThis->fMergePending = false;
rc = RTSemFastMutexRelease(mutex);
AssertRC(rc);
rc = RTSemFastMutexDestroy(mutex);
AssertRC(rc);
}
if (VALID_PTR(pThis->pBlkCache))
{
PDMR3BlkCacheRelease(pThis->pBlkCache);
pThis->pBlkCache = NULL;
}
if (VALID_PTR(pThis->pDisk))
{
VDDestroy(pThis->pDisk);
pThis->pDisk = NULL;
}
drvvdFreeImages(pThis);
if (pThis->MergeLock != NIL_RTSEMRW)
{
int rc = RTSemRWDestroy(pThis->MergeLock);
AssertRC(rc);
pThis->MergeLock = NIL_RTSEMRW;
}
if (pThis->pbData)
RTMemFree(pThis->pbData);
if (pThis->pszBwGroup)
{
MMR3HeapFree(pThis->pszBwGroup);
pThis->pszBwGroup = NULL;
}
}
/**
* Construct a VBox disk media driver instance.
*
* @copydoc FNPDMDRVCONSTRUCT
*/
static DECLCALLBACK(int) drvvdConstruct(PPDMDRVINS pDrvIns,
PCFGMNODE pCfg,
uint32_t fFlags)
{
LogFlowFunc(("\n"));
PVBOXDISK pThis = PDMINS_2_DATA(pDrvIns, PVBOXDISK);
int rc = VINF_SUCCESS;
char *pszName = NULL; /**< The path of the disk image file. */
char *pszFormat = NULL; /**< The format backed to use for this image. */
char *pszCachePath = NULL; /**< The path to the cache image. */
char *pszCacheFormat = NULL; /**< The format backend to use for the cache image. */
bool fReadOnly; /**< True if the media is read-only. */
bool fMaybeReadOnly; /**< True if the media may or may not be read-only. */
bool fHonorZeroWrites; /**< True if zero blocks should be written. */
PDMDRV_CHECK_VERSIONS_RETURN(pDrvIns);
/*
* Init the static parts.
*/
pDrvIns->IBase.pfnQueryInterface = drvvdQueryInterface;
pThis->pDrvIns = pDrvIns;
pThis->fTempReadOnly = false;
pThis->pDisk = NULL;
pThis->fAsyncIOSupported = false;
pThis->fShareable = false;
pThis->fMergePending = false;
pThis->MergeCompleteMutex = NIL_RTSEMFASTMUTEX;
pThis->uMergeSource = VD_LAST_IMAGE;
pThis->uMergeTarget = VD_LAST_IMAGE;
/* IMedia */
pThis->IMedia.pfnRead = drvvdRead;
pThis->IMedia.pfnWrite = drvvdWrite;
pThis->IMedia.pfnFlush = drvvdFlush;
pThis->IMedia.pfnMerge = drvvdMerge;
pThis->IMedia.pfnGetSize = drvvdGetSize;
pThis->IMedia.pfnIsReadOnly = drvvdIsReadOnly;
pThis->IMedia.pfnBiosGetPCHSGeometry = drvvdBiosGetPCHSGeometry;
pThis->IMedia.pfnBiosSetPCHSGeometry = drvvdBiosSetPCHSGeometry;
pThis->IMedia.pfnBiosGetLCHSGeometry = drvvdBiosGetLCHSGeometry;
pThis->IMedia.pfnBiosSetLCHSGeometry = drvvdBiosSetLCHSGeometry;
pThis->IMedia.pfnGetUuid = drvvdGetUuid;
pThis->IMedia.pfnDiscard = drvvdDiscard;
/* IMediaAsync */
pThis->IMediaAsync.pfnStartRead = drvvdStartRead;
pThis->IMediaAsync.pfnStartWrite = drvvdStartWrite;
pThis->IMediaAsync.pfnStartFlush = drvvdStartFlush;
pThis->IMediaAsync.pfnStartDiscard = drvvdStartDiscard;
/* Initialize supported VD interfaces. */
pThis->pVDIfsDisk = NULL;
pThis->VDIfError.pfnError = drvvdErrorCallback;
pThis->VDIfError.pfnMessage = NULL;
rc = VDInterfaceAdd(&pThis->VDIfError.Core, "DrvVD_VDIError", VDINTERFACETYPE_ERROR,
pDrvIns, sizeof(VDINTERFACEERROR), &pThis->pVDIfsDisk);
AssertRC(rc);
/* List of images is empty now. */
pThis->pImages = NULL;
pThis->pDrvMediaPort = PDMIBASE_QUERY_INTERFACE(pDrvIns->pUpBase, PDMIMEDIAPORT);
if (!pThis->pDrvMediaPort)
return PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_MISSING_INTERFACE_ABOVE,
N_("No media port interface above"));
/* Try to attach async media port interface above.*/
pThis->pDrvMediaAsyncPort = PDMIBASE_QUERY_INTERFACE(pDrvIns->pUpBase, PDMIMEDIAASYNCPORT);
/*
* Validate configuration and find all parent images.
* It's sort of up side down from the image dependency tree.
*/
bool fHostIP = false;
bool fUseNewIo = false;
bool fUseBlockCache = false;
bool fDiscard = false;
unsigned iLevel = 0;
PCFGMNODE pCurNode = pCfg;
VDTYPE enmType = VDTYPE_HDD;
for (;;)
{
bool fValid;
if (pCurNode == pCfg)
{
/* Toplevel configuration additionally contains the global image
* open flags. Some might be converted to per-image flags later. */
fValid = CFGMR3AreValuesValid(pCurNode,
"Format\0Path\0"
"ReadOnly\0MaybeReadOnly\0TempReadOnly\0Shareable\0HonorZeroWrites\0"
"HostIPStack\0UseNewIo\0BootAcceleration\0BootAccelerationBuffer\0"
"SetupMerge\0MergeSource\0MergeTarget\0BwGroup\0Type\0BlockCache\0"
"CachePath\0CacheFormat\0Discard\0");
}
else
{
/* All other image configurations only contain image name and
* the format information. */
fValid = CFGMR3AreValuesValid(pCurNode, "Format\0Path\0"
"MergeSource\0MergeTarget\0");
}
if (!fValid)
{
rc = PDMDrvHlpVMSetError(pDrvIns, VERR_PDM_DRVINS_UNKNOWN_CFG_VALUES,
RT_SRC_POS, N_("DrvVD: Configuration error: keys incorrect at level %d"), iLevel);
break;
}
if (pCurNode == pCfg)
{
rc = CFGMR3QueryBoolDef(pCurNode, "HostIPStack", &fHostIP, true);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"HostIPStack\" as boolean failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "HonorZeroWrites", &fHonorZeroWrites, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"HonorZeroWrites\" as boolean failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "ReadOnly", &fReadOnly, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"ReadOnly\" as boolean failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "MaybeReadOnly", &fMaybeReadOnly, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"MaybeReadOnly\" as boolean failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "TempReadOnly", &pThis->fTempReadOnly, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"TempReadOnly\" as boolean failed"));
break;
}
if (fReadOnly && pThis->fTempReadOnly)
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Both \"ReadOnly\" and \"TempReadOnly\" are set"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "Shareable", &pThis->fShareable, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"Shareable\" as boolean failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "UseNewIo", &fUseNewIo, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"UseNewIo\" as boolean failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "SetupMerge", &pThis->fMergePending, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"SetupMerge\" as boolean failed"));
break;
}
if (fReadOnly && pThis->fMergePending)
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Both \"ReadOnly\" and \"MergePending\" are set"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "BootAcceleration", &pThis->fBootAccelEnabled, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"BootAcceleration\" as boolean failed"));
break;
}
rc = CFGMR3QueryU32Def(pCurNode, "BootAccelerationBuffer", (uint32_t *)&pThis->cbBootAccelBuffer, 16 * _1K);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"BootAccelerationBuffer\" as integer failed"));
break;
}
rc = CFGMR3QueryBoolDef(pCurNode, "BlockCache", &fUseBlockCache, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"BlockCache\" as boolean failed"));
break;
}
rc = CFGMR3QueryStringAlloc(pCurNode, "BwGroup", &pThis->pszBwGroup);
if (RT_FAILURE(rc) && rc != VERR_CFGM_VALUE_NOT_FOUND)
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"BwGroup\" as string failed"));
break;
}
else
rc = VINF_SUCCESS;
rc = CFGMR3QueryBoolDef(pCurNode, "Discard", &fDiscard, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"Discard\" as boolean failed"));
break;
}
if (fReadOnly && fDiscard)
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Both \"ReadOnly\" and \"Discard\" are set"));
break;
}
char *psz;
rc = CFGMR3QueryStringAlloc(pCfg, "Type", &psz);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_BLOCK_NO_TYPE, N_("Failed to obtain the type"));
break;
}
else if (!strcmp(psz, "HardDisk"))
enmType = VDTYPE_HDD;
else if (!strcmp(psz, "DVD"))
enmType = VDTYPE_DVD;
else if (!strcmp(psz, "Floppy"))
enmType = VDTYPE_FLOPPY;
else
{
rc = PDMDrvHlpVMSetError(pDrvIns, VERR_PDM_BLOCK_UNKNOWN_TYPE, RT_SRC_POS,
N_("Unknown type \"%s\""), psz);
MMR3HeapFree(psz);
break;
}
MMR3HeapFree(psz); psz = NULL;
rc = CFGMR3QueryStringAlloc(pCurNode, "CachePath", &pszCachePath);
if (RT_FAILURE(rc) && rc != VERR_CFGM_VALUE_NOT_FOUND)
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"CachePath\" as string failed"));
break;
}
else
rc = VINF_SUCCESS;
if (pszCachePath)
{
rc = CFGMR3QueryStringAlloc(pCurNode, "CacheFormat", &pszCacheFormat);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"CacheFormat\" as string failed"));
break;
}
}
}
PCFGMNODE pParent = CFGMR3GetChild(pCurNode, "Parent");
if (!pParent)
break;
pCurNode = pParent;
iLevel++;
}
/*
* Create the image container and the necessary interfaces.
*/
if (RT_SUCCESS(rc))
{
/*
* The image has a bandwidth group but the host cache is enabled.
* Use the async I/O framework but tell it to enable the host cache.
*/
if (!fUseNewIo && pThis->pszBwGroup)
{
pThis->fAsyncIoWithHostCache = true;
fUseNewIo = true;
}
/** @todo quick hack to work around problems in the async I/O
* implementation (rw semaphore thread ownership problem)
* while a merge is running. Remove once this is fixed. */
if (pThis->fMergePending)
fUseNewIo = false;
if (RT_SUCCESS(rc) && pThis->fMergePending)
{
rc = RTSemFastMutexCreate(&pThis->MergeCompleteMutex);
if (RT_SUCCESS(rc))
rc = RTSemRWCreate(&pThis->MergeLock);
if (RT_SUCCESS(rc))
{
pThis->VDIfThreadSync.pfnStartRead = drvvdThreadStartRead;
pThis->VDIfThreadSync.pfnFinishRead = drvvdThreadFinishRead;
pThis->VDIfThreadSync.pfnStartWrite = drvvdThreadStartWrite;
pThis->VDIfThreadSync.pfnFinishWrite = drvvdThreadFinishWrite;
rc = VDInterfaceAdd(&pThis->VDIfThreadSync.Core, "DrvVD_ThreadSync", VDINTERFACETYPE_THREADSYNC,
pThis, sizeof(VDINTERFACETHREADSYNC), &pThis->pVDIfsDisk);
}
else
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Failed to create semaphores for \"MergePending\""));
}
}
if (RT_SUCCESS(rc))
{
rc = VDCreate(pThis->pVDIfsDisk, enmType, &pThis->pDisk);
/* Error message is already set correctly. */
}
}
if (pThis->pDrvMediaAsyncPort && fUseNewIo)
pThis->fAsyncIOSupported = true;
unsigned iImageIdx = 0;
while (pCurNode && RT_SUCCESS(rc))
{
/* Allocate per-image data. */
PVBOXIMAGE pImage = drvvdNewImage(pThis);
if (!pImage)
{
rc = VERR_NO_MEMORY;
break;
}
/*
* Read the image configuration.
*/
rc = CFGMR3QueryStringAlloc(pCurNode, "Path", &pszName);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"Path\" as string failed"));
break;
}
rc = CFGMR3QueryStringAlloc(pCurNode, "Format", &pszFormat);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"Format\" as string failed"));
break;
}
bool fMergeSource;
rc = CFGMR3QueryBoolDef(pCurNode, "MergeSource", &fMergeSource, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"MergeSource\" as boolean failed"));
break;
}
if (fMergeSource)
{
if (pThis->uMergeSource == VD_LAST_IMAGE)
pThis->uMergeSource = iImageIdx;
else
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Multiple \"MergeSource\" occurrences"));
break;
}
}
bool fMergeTarget;
rc = CFGMR3QueryBoolDef(pCurNode, "MergeTarget", &fMergeTarget, false);
if (RT_FAILURE(rc))
{
rc = PDMDRV_SET_ERROR(pDrvIns, rc,
N_("DrvVD: Configuration error: Querying \"MergeTarget\" as boolean failed"));
break;
}
if (fMergeTarget)
{
if (pThis->uMergeTarget == VD_LAST_IMAGE)
pThis->uMergeTarget = iImageIdx;
else
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Multiple \"MergeTarget\" occurrences"));
break;
}
}
PCFGMNODE pCfgVDConfig = CFGMR3GetChild(pCurNode, "VDConfig");
pImage->VDIfConfig.pfnAreKeysValid = drvvdCfgAreKeysValid;
pImage->VDIfConfig.pfnQuerySize = drvvdCfgQuerySize;
pImage->VDIfConfig.pfnQuery = drvvdCfgQuery;
rc = VDInterfaceAdd(&pImage->VDIfConfig.Core, "DrvVD_Config", VDINTERFACETYPE_CONFIG,
pCfgVDConfig, sizeof(VDINTERFACECONFIG), &pImage->pVDIfsImage);
AssertRC(rc);
/* Unconditionally insert the TCPNET interface, don't bother to check
* if an image really needs it. Will be ignored. Since the TCPNET
* interface is per image we could make this more flexible in the
* future if we want to. */
/* Construct TCPNET callback table depending on the config. This is
* done unconditionally, as uninterested backends will ignore it. */
if (fHostIP)
{
pImage->VDIfTcpNet.pfnSocketCreate = drvvdTcpSocketCreate;
pImage->VDIfTcpNet.pfnSocketDestroy = drvvdTcpSocketDestroy;
pImage->VDIfTcpNet.pfnClientConnect = drvvdTcpClientConnect;
pImage->VDIfTcpNet.pfnIsClientConnected = drvvdTcpIsClientConnected;
pImage->VDIfTcpNet.pfnClientClose = drvvdTcpClientClose;
pImage->VDIfTcpNet.pfnSelectOne = drvvdTcpSelectOne;
pImage->VDIfTcpNet.pfnRead = drvvdTcpRead;
pImage->VDIfTcpNet.pfnWrite = drvvdTcpWrite;
pImage->VDIfTcpNet.pfnSgWrite = drvvdTcpSgWrite;
pImage->VDIfTcpNet.pfnReadNB = drvvdTcpReadNB;
pImage->VDIfTcpNet.pfnWriteNB = drvvdTcpWriteNB;
pImage->VDIfTcpNet.pfnSgWriteNB = drvvdTcpSgWriteNB;
pImage->VDIfTcpNet.pfnFlush = drvvdTcpFlush;
pImage->VDIfTcpNet.pfnSetSendCoalescing = drvvdTcpSetSendCoalescing;
pImage->VDIfTcpNet.pfnGetLocalAddress = drvvdTcpGetLocalAddress;
pImage->VDIfTcpNet.pfnGetPeerAddress = drvvdTcpGetPeerAddress;
/*
* There is a 15ms delay between receiving the data and marking the socket
* as readable on Windows XP which hurts async I/O performance of
* TCP backends badly. Provide a different select method without
* using poll on XP.
* This is only used on XP because it is not as efficient as the one using poll
* and all other Windows versions are working fine.
*/
char szOS[64];
memset(szOS, 0, sizeof(szOS));
rc = RTSystemQueryOSInfo(RTSYSOSINFO_PRODUCT, &szOS[0], sizeof(szOS));
if (RT_SUCCESS(rc) && !strncmp(szOS, "Windows XP", 10))
{
LogRel(("VD: Detected Windows XP, disabled poll based waiting for TCP\n"));
pImage->VDIfTcpNet.pfnSelectOneEx = drvvdTcpSelectOneExNoPoll;
}
else
pImage->VDIfTcpNet.pfnSelectOneEx = drvvdTcpSelectOneExPoll;
pImage->VDIfTcpNet.pfnPoke = drvvdTcpPoke;
}
else
{
#ifndef VBOX_WITH_INIP
rc = PDMDrvHlpVMSetError(pDrvIns, VERR_PDM_DRVINS_UNKNOWN_CFG_VALUES,
RT_SRC_POS, N_("DrvVD: Configuration error: TCP over Internal Networking not compiled in"));
#else /* VBOX_WITH_INIP */
pImage->VDIfTcpNet.pfnSocketCreate = drvvdINIPSocketCreate;
pImage->VDIfTcpNet.pfnSocketDestroy = drvvdINIPSocketDestroy;
pImage->VDIfTcpNet.pfnClientConnect = drvvdINIPClientConnect;
pImage->VDIfTcpNet.pfnClientClose = drvvdINIPClientClose;
pImage->VDIfTcpNet.pfnIsClientConnected = drvvdINIPIsClientConnected;
pImage->VDIfTcpNet.pfnSelectOne = drvvdINIPSelectOne;
pImage->VDIfTcpNet.pfnRead = drvvdINIPRead;
pImage->VDIfTcpNet.pfnWrite = drvvdINIPWrite;
pImage->VDIfTcpNet.pfnSgWrite = drvvdINIPSgWrite;
pImage->VDIfTcpNet.pfnFlush = drvvdINIPFlush;
pImage->VDIfTcpNet.pfnSetSendCoalescing = drvvdINIPSetSendCoalescing;
pImage->VDIfTcpNet.pfnGetLocalAddress = drvvdINIPGetLocalAddress;
pImage->VDIfTcpNet.pfnGetPeerAddress = drvvdINIPGetPeerAddress;
pImage->VDIfTcpNet.pfnSelectOneEx = drvvdINIPSelectOneEx;
pImage->VDIfTcpNet.pfnPoke = drvvdINIPPoke;
#endif /* VBOX_WITH_INIP */
}
rc = VDInterfaceAdd(&pImage->VDIfTcpNet.Core, "DrvVD_TCPNET",
VDINTERFACETYPE_TCPNET, NULL,
sizeof(VDINTERFACETCPNET), &pImage->pVDIfsImage);
AssertRC(rc);
/* Insert the custom I/O interface only if we're told to use new IO.
* Since the I/O interface is per image we could make this more
* flexible in the future if we want to. */
if (fUseNewIo)
{
#ifdef VBOX_WITH_PDM_ASYNC_COMPLETION
pImage->VDIfIo.pfnOpen = drvvdAsyncIOOpen;
pImage->VDIfIo.pfnClose = drvvdAsyncIOClose;
pImage->VDIfIo.pfnGetSize = drvvdAsyncIOGetSize;
pImage->VDIfIo.pfnSetSize = drvvdAsyncIOSetSize;
pImage->VDIfIo.pfnReadSync = drvvdAsyncIOReadSync;
pImage->VDIfIo.pfnWriteSync = drvvdAsyncIOWriteSync;
pImage->VDIfIo.pfnFlushSync = drvvdAsyncIOFlushSync;
pImage->VDIfIo.pfnReadAsync = drvvdAsyncIOReadAsync;
pImage->VDIfIo.pfnWriteAsync = drvvdAsyncIOWriteAsync;
pImage->VDIfIo.pfnFlushAsync = drvvdAsyncIOFlushAsync;
#else /* !VBOX_WITH_PDM_ASYNC_COMPLETION */
rc = PDMDrvHlpVMSetError(pDrvIns, VERR_PDM_DRVINS_UNKNOWN_CFG_VALUES,
RT_SRC_POS, N_("DrvVD: Configuration error: Async Completion Framework not compiled in"));
#endif /* !VBOX_WITH_PDM_ASYNC_COMPLETION */
if (RT_SUCCESS(rc))
rc = VDInterfaceAdd(&pImage->VDIfIo.Core, "DrvVD_IO", VDINTERFACETYPE_IO,
pThis, sizeof(VDINTERFACEIO), &pImage->pVDIfsImage);
AssertRC(rc);
}
/*
* Open the image.
*/
unsigned uOpenFlags;
if (fReadOnly || pThis->fTempReadOnly || iLevel != 0)
uOpenFlags = VD_OPEN_FLAGS_READONLY;
else
uOpenFlags = VD_OPEN_FLAGS_NORMAL;
if (fHonorZeroWrites)
uOpenFlags |= VD_OPEN_FLAGS_HONOR_ZEROES;
if (pThis->fAsyncIOSupported)
uOpenFlags |= VD_OPEN_FLAGS_ASYNC_IO;
if (pThis->fShareable)
uOpenFlags |= VD_OPEN_FLAGS_SHAREABLE;
if (fDiscard && iLevel == 0)
uOpenFlags |= VD_OPEN_FLAGS_DISCARD;
/* Try to open backend in async I/O mode first. */
rc = VDOpen(pThis->pDisk, pszFormat, pszName, uOpenFlags, pImage->pVDIfsImage);
if (rc == VERR_NOT_SUPPORTED)
{
pThis->fAsyncIOSupported = false;
uOpenFlags &= ~VD_OPEN_FLAGS_ASYNC_IO;
rc = VDOpen(pThis->pDisk, pszFormat, pszName, uOpenFlags, pImage->pVDIfsImage);
}
if (rc == VERR_VD_DISCARD_NOT_SUPPORTED)
{
fDiscard = false;
uOpenFlags &= ~VD_OPEN_FLAGS_DISCARD;
rc = VDOpen(pThis->pDisk, pszFormat, pszName, uOpenFlags, pImage->pVDIfsImage);
}
if (!fDiscard)
{
pThis->IMedia.pfnDiscard = NULL;
pThis->IMediaAsync.pfnStartDiscard = NULL;
}
if (RT_SUCCESS(rc))
{
LogFunc(("%d - Opened '%s' in %s mode\n",
iLevel, pszName,
VDIsReadOnly(pThis->pDisk) ? "read-only" : "read-write"));
if ( VDIsReadOnly(pThis->pDisk)
&& !fReadOnly
&& !fMaybeReadOnly
&& !pThis->fTempReadOnly
&& iLevel == 0)
{
rc = PDMDrvHlpVMSetError(pDrvIns, VERR_VD_IMAGE_READ_ONLY, RT_SRC_POS,
N_("Failed to open image '%s' for writing due to wrong permissions"),
pszName);
break;
}
}
else
{
rc = PDMDrvHlpVMSetError(pDrvIns, rc, RT_SRC_POS,
N_("Failed to open image '%s' in %s mode rc=%Rrc"), pszName,
(uOpenFlags & VD_OPEN_FLAGS_READONLY) ? "read-only" : "read-write", rc);
break;
}
MMR3HeapFree(pszName);
pszName = NULL;
MMR3HeapFree(pszFormat);
pszFormat = NULL;
/* next */
iLevel--;
iImageIdx++;
pCurNode = CFGMR3GetParent(pCurNode);
}
/* Open the cache image if set. */
if ( RT_SUCCESS(rc)
&& VALID_PTR(pszCachePath))
{
/* Insert the custom I/O interface only if we're told to use new IO.
* Since the I/O interface is per image we could make this more
* flexible in the future if we want to. */
if (fUseNewIo)
{
#ifdef VBOX_WITH_PDM_ASYNC_COMPLETION
pThis->VDIfIoCache.pfnOpen = drvvdAsyncIOOpen;
pThis->VDIfIoCache.pfnClose = drvvdAsyncIOClose;
pThis->VDIfIoCache.pfnGetSize = drvvdAsyncIOGetSize;
pThis->VDIfIoCache.pfnSetSize = drvvdAsyncIOSetSize;
pThis->VDIfIoCache.pfnReadSync = drvvdAsyncIOReadSync;
pThis->VDIfIoCache.pfnWriteSync = drvvdAsyncIOWriteSync;
pThis->VDIfIoCache.pfnFlushSync = drvvdAsyncIOFlushSync;
pThis->VDIfIoCache.pfnReadAsync = drvvdAsyncIOReadAsync;
pThis->VDIfIoCache.pfnWriteAsync = drvvdAsyncIOWriteAsync;
pThis->VDIfIoCache.pfnFlushAsync = drvvdAsyncIOFlushAsync;
#else /* !VBOX_WITH_PDM_ASYNC_COMPLETION */
rc = PDMDrvHlpVMSetError(pDrvIns, VERR_PDM_DRVINS_UNKNOWN_CFG_VALUES,
RT_SRC_POS, N_("DrvVD: Configuration error: Async Completion Framework not compiled in"));
#endif /* !VBOX_WITH_PDM_ASYNC_COMPLETION */
if (RT_SUCCESS(rc))
rc = VDInterfaceAdd(&pThis->VDIfIoCache.Core, "DrvVD_IO", VDINTERFACETYPE_IO,
pThis, sizeof(VDINTERFACEIO), &pThis->pVDIfsCache);
AssertRC(rc);
}
rc = VDCacheOpen(pThis->pDisk, pszCacheFormat, pszCachePath, VD_OPEN_FLAGS_NORMAL, pThis->pVDIfsCache);
if (RT_FAILURE(rc))
rc = PDMDRV_SET_ERROR(pDrvIns, rc, N_("DrvVD: Could not open cache image"));
}
if (VALID_PTR(pszCachePath))
MMR3HeapFree(pszCachePath);
if (VALID_PTR(pszCacheFormat))
MMR3HeapFree(pszCacheFormat);
if ( RT_SUCCESS(rc)
&& pThis->fMergePending
&& ( pThis->uMergeSource == VD_LAST_IMAGE
|| pThis->uMergeTarget == VD_LAST_IMAGE))
{
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Inconsistent image merge data"));
}
/* Create the block cache if enabled. */
if ( fUseBlockCache
&& !pThis->fShareable
&& !fDiscard
&& RT_SUCCESS(rc))
{
/*
* We need a unique ID for the block cache (to identify the owner of data
* blocks in a saved state). UUIDs are not really suitable because
* there are image formats which don't support them. Furthermore it is
* possible that a new diff image was attached after a saved state
* which changes the UUID.
* However the device "name + device instance + LUN" triple the disk is
* attached to is always constant for saved states.
*/
char *pszId = NULL;
uint32_t iInstance, iLUN;
const char *pcszController;
rc = pThis->pDrvMediaPort->pfnQueryDeviceLocation(pThis->pDrvMediaPort, &pcszController,
&iInstance, &iLUN);
if (RT_FAILURE(rc))
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Configuration error: Could not query device data"));
else
{
int cbStr = RTStrAPrintf(&pszId, "%s-%d-%d", pcszController, iInstance, iLUN);
if (cbStr > 0)
{
rc = PDMDrvHlpBlkCacheRetain(pDrvIns, &pThis->pBlkCache,
drvvdBlkCacheXferComplete,
drvvdBlkCacheXferEnqueue,
drvvdBlkCacheXferEnqueueDiscard,
pszId);
if (rc == VERR_NOT_SUPPORTED)
{
LogRel(("VD: Block cache is not supported\n"));
rc = VINF_SUCCESS;
}
else
AssertRC(rc);
RTStrFree(pszId);
}
else
rc = PDMDRV_SET_ERROR(pDrvIns, VERR_PDM_DRIVER_INVALID_PROPERTIES,
N_("DrvVD: Out of memory when creating block cache"));
}
}
/*
* Register a load-done callback so we can undo TempReadOnly config before
* we get to drvvdResume. Autoamtically deregistered upon destruction.
*/
if (RT_SUCCESS(rc))
rc = PDMDrvHlpSSMRegisterEx(pDrvIns, 0 /* version */, 0 /* cbGuess */,
NULL /*pfnLivePrep*/, NULL /*pfnLiveExec*/, NULL /*pfnLiveVote*/,
NULL /*pfnSavePrep*/, NULL /*pfnSaveExec*/, NULL /*pfnSaveDone*/,
NULL /*pfnDonePrep*/, NULL /*pfnLoadExec*/, drvvdLoadDone);
/* Setup the boot acceleration stuff if enabled. */
if (RT_SUCCESS(rc) && pThis->fBootAccelEnabled)
{
pThis->cbDisk = VDGetSize(pThis->pDisk, VD_LAST_IMAGE);
Assert(pThis->cbDisk > 0);
pThis->pbData = (uint8_t *)RTMemAllocZ(pThis->cbBootAccelBuffer);
if (pThis->pbData)
{
pThis->fBootAccelActive = true;
pThis->offDisk = 0;
pThis->cbDataValid = 0;
LogRel(("VD: Boot acceleration enabled\n"));
}
else
LogRel(("VD: Boot acceleration, out of memory, disabled\n"));
}
if (RT_FAILURE(rc))
{
if (VALID_PTR(pszName))
MMR3HeapFree(pszName);
if (VALID_PTR(pszFormat))
MMR3HeapFree(pszFormat);
/* drvvdDestruct does the rest. */
}
LogFlowFunc(("returns %Rrc\n", rc));
return rc;
}
/**
* VBox disk container media driver registration record.
*/
const PDMDRVREG g_DrvVD =
{
/* u32Version */
PDM_DRVREG_VERSION,
/* szName */
"VD",
/* szRCMod */
"",
/* szR0Mod */
"",
/* pszDescription */
"Generic VBox disk media driver.",
/* fFlags */
PDM_DRVREG_FLAGS_HOST_BITS_DEFAULT,
/* fClass. */
PDM_DRVREG_CLASS_MEDIA,
/* cMaxInstances */
~0,
/* cbInstance */
sizeof(VBOXDISK),
/* pfnConstruct */
drvvdConstruct,
/* pfnDestruct */
drvvdDestruct,
/* pfnRelocate */
NULL,
/* pfnIOCtl */
NULL,
/* pfnPowerOn */
drvvdPowerOn,
/* pfnReset */
drvvdReset,
/* pfnSuspend */
drvvdSuspend,
/* pfnResume */
drvvdResume,
/* pfnAttach */
NULL,
/* pfnDetach */
NULL,
/* pfnPowerOff */
NULL,
/* pfnSoftReset */
NULL,
/* u32EndVersion */
PDM_DRVREG_VERSION
};