VBoxServicePageSharing.cpp revision 1a9a40add2c78bba007e3f5c876859967a8e9edf
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/* $Id$ */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/** @file
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * VBoxService - Guest page sharing.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Copyright (C) 2006-2012 Oracle Corporation
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync *
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * available from http://www.virtualbox.org. This file is free software;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * you can redistribute it and/or modify it under the terms of the GNU
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * General Public License (GPL) as published by the Free Software
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/*******************************************************************************
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync* Header Files *
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync*******************************************************************************/
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/assert.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/avl.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/asm.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/mem.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/ldr.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/process.h>
359f2915f947abde1bba4358eada89941ee87cccvboxsync#include <iprt/env.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/stream.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/file.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/string.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/semaphore.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/system.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/thread.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <iprt/time.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <VBox/VBoxGuestLib.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include "VBoxServiceInternal.h"
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include "VBoxServiceUtils.h"
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/*******************************************************************************
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync* Global Variables *
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync*******************************************************************************/
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/** The semaphore we're blocking on. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic RTSEMEVENTMULTI g_PageSharingEvent = NIL_RTSEMEVENTMULTI;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#if defined(RT_OS_WINDOWS) && !defined(TARGET_NT4)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <tlhelp32.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <psapi.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#include <winternl.h>
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsynctypedef struct
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync AVLPVNODECORE Core;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync HMODULE hModule;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char szFileVersion[16];
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync MODULEENTRY32 Info;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync} KNOWN_MODULE, *PKNOWN_MODULE;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#define SystemModuleInformation 11
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsynctypedef struct _RTL_PROCESS_MODULE_INFORMATION
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ULONG Section;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PVOID MappedBase;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PVOID ImageBase;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ULONG ImageSize;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ULONG Flags;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync USHORT LoadOrderIndex;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync USHORT InitOrderIndex;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync USHORT LoadCount;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync USHORT OffsetToFileName;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync CHAR FullPathName[256];
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync} RTL_PROCESS_MODULE_INFORMATION, *PRTL_PROCESS_MODULE_INFORMATION;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsynctypedef struct _RTL_PROCESS_MODULES
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ULONG NumberOfModules;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTL_PROCESS_MODULE_INFORMATION Modules[1];
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync} RTL_PROCESS_MODULES, *PRTL_PROCESS_MODULES;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsynctypedef NTSTATUS (WINAPI *PFNZWQUERYSYSTEMINFORMATION)(ULONG, PVOID, ULONG, PULONG);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic PFNZWQUERYSYSTEMINFORMATION g_pfnZwQuerySystemInformation = NULL;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic DECLCALLBACK(int) VBoxServicePageSharingEmptyTreeCallback(PAVLPVNODECORE pNode, void *pvUser);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic PAVLPVNODECORE g_pKnownModuleTree = NULL;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic uint64_t g_idSession = 0;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/**
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Registers a new module with the VMM
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * @param pModule Module ptr
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * @param fValidateMemory Validate/touch memory pages or not
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncvoid VBoxServicePageSharingRegisterModule(PKNOWN_MODULE pModule, bool fValidateMemory)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VMMDEVSHAREDREGIONDESC aRegions[VMMDEVSHAREDREGIONDESC_MAX];
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync DWORD dwModuleSize = pModule->Info.modBaseSize;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync BYTE *pBaseAddress = pModule->Info.modBaseAddr;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync DWORD cbVersionSize, dummy;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync BYTE *pVersionInfo;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync cbVersionSize = GetFileVersionInfoSize(pModule->Info.szExePath, &dummy);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!cbVersionSize)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: GetFileVersionInfoSize failed with %d\n", GetLastError());
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pVersionInfo = (BYTE *)RTMemAlloc(cbVersionSize);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!pVersionInfo)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!GetFileVersionInfo(pModule->Info.szExePath, 0, cbVersionSize, pVersionInfo))
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: GetFileVersionInfo failed with %d\n", GetLastError());
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync goto end;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Fetch default code page. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync struct LANGANDCODEPAGE {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync WORD wLanguage;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync WORD wCodePage;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync } *lpTranslate;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync UINT cbTranslate;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync BOOL ret = VerQueryValue(pVersionInfo, TEXT("\\VarFileInfo\\Translation"), (LPVOID *)&lpTranslate, &cbTranslate);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if ( !ret
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync || cbTranslate < 4)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: VerQueryValue failed with %d (cb=%d)\n", GetLastError(), cbTranslate);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync goto end;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync unsigned i;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync UINT cbFileVersion;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char *lpszFileVersion;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync unsigned cTranslationBlocks = cbTranslate/sizeof(struct LANGANDCODEPAGE);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync for(i = 0; i < cTranslationBlocks; i++)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Fetch file version string. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char szFileVersionLocation[256];
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync sprintf(szFileVersionLocation, TEXT("\\StringFileInfo\\%04x%04x\\FileVersion"), lpTranslate[i].wLanguage, lpTranslate[i].wCodePage);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ret = VerQueryValue(pVersionInfo, szFileVersionLocation, (LPVOID *)&lpszFileVersion, &cbFileVersion);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (ret)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (i == cTranslationBlocks)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: no file version found!\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync goto end;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync _snprintf(pModule->szFileVersion, sizeof(pModule->szFileVersion), "%s", lpszFileVersion);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pModule->szFileVersion[RT_ELEMENTS(pModule->szFileVersion) - 1] = 0;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync unsigned idxRegion = 0;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (fValidateMemory)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync do
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync MEMORY_BASIC_INFORMATION MemInfo;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync SIZE_T ret = VirtualQuery(pBaseAddress, &MemInfo, sizeof(MemInfo));
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync Assert(ret);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!ret)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: VirtualQueryEx failed with %d\n", GetLastError());
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if ( MemInfo.State == MEM_COMMIT
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync && MemInfo.Type == MEM_IMAGE)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync switch (MemInfo.Protect)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync case PAGE_EXECUTE:
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync case PAGE_EXECUTE_READ:
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync case PAGE_READONLY:
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char *pRegion = (char *)MemInfo.BaseAddress;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Skip the first region as it only contains the image file header. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (pRegion != (char *)pModule->Info.modBaseAddr)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Touch all pages. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync while (pRegion < (char *)MemInfo.BaseAddress + MemInfo.RegionSize)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Try to trick the optimizer to leave the page touching code in place. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ASMProbeReadByte(pRegion);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pRegion += PAGE_SIZE;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#ifdef RT_ARCH_X86
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync aRegions[idxRegion].GCRegionAddr = (RTGCPTR32)MemInfo.BaseAddress;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#else
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync aRegions[idxRegion].GCRegionAddr = (RTGCPTR64)MemInfo.BaseAddress;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#endif
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync aRegions[idxRegion].cbRegion = MemInfo.RegionSize;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync idxRegion++;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync default:
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break; /* ignore */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pBaseAddress = (BYTE *)MemInfo.BaseAddress + MemInfo.RegionSize;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (dwModuleSize > MemInfo.RegionSize)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync dwModuleSize -= MemInfo.RegionSize;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync else
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync dwModuleSize = 0;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync break;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (idxRegion >= RT_ELEMENTS(aRegions))
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break; /* out of room */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync while (dwModuleSize);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync else
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* We can't probe kernel memory ranges, so pretend it's one big region. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#ifdef RT_ARCH_X86
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync aRegions[idxRegion].GCRegionAddr = (RTGCPTR32)pBaseAddress;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#else
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync aRegions[idxRegion].GCRegionAddr = (RTGCPTR64)pBaseAddress;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#endif
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync aRegions[idxRegion].cbRegion = dwModuleSize;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync idxRegion++;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: VbglR3RegisterSharedModule %s %s base=%p size=%x cregions=%d\n", pModule->Info.szModule, pModule->szFileVersion, pModule->Info.modBaseAddr, pModule->Info.modBaseSize, idxRegion);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync int rc = VbglR3RegisterSharedModule(pModule->Info.szModule, pModule->szFileVersion, (uintptr_t)pModule->Info.modBaseAddr,
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pModule->Info.modBaseSize, idxRegion, aRegions);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (RT_FAILURE(rc))
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingRegisterModule: VbglR3RegisterSharedModule failed with %Rrc\n", rc);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncend:
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTMemFree(pVersionInfo);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/**
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Inspect all loaded modules for the specified process
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * @param dwProcessId Process id
359f2915f947abde1bba4358eada89941ee87cccvboxsync */
359f2915f947abde1bba4358eada89941ee87cccvboxsyncvoid VBoxServicePageSharingInspectModules(DWORD dwProcessId, PAVLPVNODECORE *ppNewTree)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync HANDLE hProcess, hSnapshot;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Get a list of all the modules in this process. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync hProcess = OpenProcess(PROCESS_QUERY_INFORMATION,
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync FALSE /* no child process handle inheritance */, dwProcessId);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (hProcess == NULL)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInspectModules: OpenProcess %x failed with %d\n", dwProcessId, GetLastError());
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPMODULE, dwProcessId);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (hSnapshot == INVALID_HANDLE_VALUE)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInspectModules: CreateToolhelp32Snapshot failed with %d\n", GetLastError());
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync CloseHandle(hProcess);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInspectModules\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync MODULEENTRY32 ModuleInfo;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync BOOL bRet;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ModuleInfo.dwSize = sizeof(ModuleInfo);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync bRet = Module32First(hSnapshot, &ModuleInfo);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync do
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /** @todo when changing this make sure VBoxService.exe is excluded! */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char *pszDot = strrchr(ModuleInfo.szModule, '.');
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if ( pszDot
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync && (pszDot[1] == 'e' || pszDot[1] == 'E'))
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync continue; /* ignore executables for now. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Found it before? */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PAVLPVNODECORE pRec = RTAvlPVGet(ppNewTree, ModuleInfo.modBaseAddr);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!pRec)
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync {
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync pRec = RTAvlPVRemove(&g_pKnownModuleTree, ModuleInfo.modBaseAddr);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync if (!pRec)
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync {
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync /* New module; register it. */
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync PKNOWN_MODULE pModule = (PKNOWN_MODULE)RTMemAllocZ(sizeof(*pModule));
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync Assert(pModule);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync if (!pModule)
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync break;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync pModule->Info = ModuleInfo;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync pModule->Core.Key = ModuleInfo.modBaseAddr;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync pModule->hModule = LoadLibraryEx(ModuleInfo.szExePath, 0, DONT_RESOLVE_DLL_REFERENCES);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync if (pModule->hModule)
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServicePageSharingRegisterModule(pModule, true /* validate pages */);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "\n\n MODULE NAME: %s", ModuleInfo.szModule );
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "\n executable = %s", ModuleInfo.szExePath );
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "\n process ID = 0x%08X", ModuleInfo.th32ProcessID );
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "\n base address = 0x%08X", (DWORD) ModuleInfo.modBaseAddr );
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "\n base size = %d", ModuleInfo.modBaseSize );
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync pRec = &pModule->Core;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync bool ret = RTAvlPVInsert(ppNewTree, pRec);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync Assert(ret); NOREF(ret);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync while (Module32Next(hSnapshot, &ModuleInfo));
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync CloseHandle(hSnapshot);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync CloseHandle(hProcess);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/**
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Inspect all running processes for executables and dlls that might be worth sharing
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * with other VMs.
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync *
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync */
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsyncvoid VBoxServicePageSharingInspectGuest()
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync{
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync HANDLE hSnapshot;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync PAVLPVNODECORE pNewTree = NULL;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync DWORD dwProcessId = GetCurrentProcessId();
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInspectGuest\n");
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync hSnapshot = CreateToolhelp32Snapshot(TH32CS_SNAPPROCESS, 0);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync if (hSnapshot == INVALID_HANDLE_VALUE)
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync {
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInspectGuest: CreateToolhelp32Snapshot failed with %d\n", GetLastError());
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync return;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync /* Check loaded modules for all running processes. */
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync PROCESSENTRY32 ProcessInfo;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync ProcessInfo.dwSize = sizeof(ProcessInfo);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync Process32First(hSnapshot, &ProcessInfo);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync do
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Skip our own process. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (ProcessInfo.th32ProcessID != dwProcessId)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServicePageSharingInspectModules(ProcessInfo.th32ProcessID, &pNewTree);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync while (Process32Next(hSnapshot, &ProcessInfo));
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync CloseHandle(hSnapshot);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Check all loaded kernel modules. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (g_pfnZwQuerySystemInformation)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ULONG cbBuffer = 0;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PVOID pBuffer = NULL;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PRTL_PROCESS_MODULES pSystemModules;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync NTSTATUS ret = g_pfnZwQuerySystemInformation(SystemModuleInformation, (PVOID)&cbBuffer, 0, &cbBuffer);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!cbBuffer)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(1, "ZwQuerySystemInformation returned length 0\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync goto skipkernelmodules;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pBuffer = RTMemAllocZ(cbBuffer);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!pBuffer)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync goto skipkernelmodules;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync ret = g_pfnZwQuerySystemInformation(SystemModuleInformation, pBuffer, cbBuffer, &cbBuffer);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (ret != STATUS_SUCCESS)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(1, "ZwQuerySystemInformation returned %x (1)\n", ret);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync goto skipkernelmodules;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pSystemModules = (PRTL_PROCESS_MODULES)pBuffer;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync for (unsigned i = 0; i < pSystemModules->NumberOfModules; i++)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(4, "\n\n KERNEL MODULE NAME: %s", pSystemModules->Modules[i].FullPathName[pSystemModules->Modules[i].OffsetToFileName] );
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(4, "\n executable = %s", pSystemModules->Modules[i].FullPathName );
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(4, "\n flags = 0x%08X\n", pSystemModules->Modules[i].Flags);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* User-mode modules seem to have no flags set; skip them as we detected them above. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (pSystemModules->Modules[i].Flags == 0)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync continue;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Found it before? */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PAVLPVNODECORE pRec = RTAvlPVGet(&pNewTree, pSystemModules->Modules[i].ImageBase);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!pRec)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync pRec = RTAvlPVRemove(&g_pKnownModuleTree, pSystemModules->Modules[i].ImageBase);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!pRec)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* New module; register it. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char szFullFilePath[512];
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync PKNOWN_MODULE pModule = (PKNOWN_MODULE)RTMemAllocZ(sizeof(*pModule));
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync Assert(pModule);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!pModule)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync strcpy(pModule->Info.szModule, &pSystemModules->Modules[i].FullPathName[pSystemModules->Modules[i].OffsetToFileName]);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync GetSystemDirectoryA(szFullFilePath, sizeof(szFullFilePath));
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* skip \Systemroot\system32 */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char *lpPath = strchr(&pSystemModules->Modules[i].FullPathName[1], '\\');
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (!lpPath)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* Seen just file names in XP; try to locate the file in the system32 and system32\drivers directories. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync strcat(szFullFilePath, "\\");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync strcat(szFullFilePath, pSystemModules->Modules[i].FullPathName);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "Unexpected kernel module name try %s\n", szFullFilePath);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (RTFileExists(szFullFilePath) == false)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync GetSystemDirectoryA(szFullFilePath, sizeof(szFullFilePath));
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync strcat(szFullFilePath, "\\drivers\\");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync strcat(szFullFilePath, pSystemModules->Modules[i].FullPathName);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "Unexpected kernel module name try %s\n", szFullFilePath);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (RTFileExists(szFullFilePath) == false)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceVerbose(1, "Unexpected kernel module name %s\n", pSystemModules->Modules[i].FullPathName);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTMemFree(pModule);
359f2915f947abde1bba4358eada89941ee87cccvboxsync continue;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync else
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
359f2915f947abde1bba4358eada89941ee87cccvboxsync lpPath = strchr(lpPath+1, '\\');
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (!lpPath)
359f2915f947abde1bba4358eada89941ee87cccvboxsync {
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceVerbose(1, "Unexpected kernel module name %s (2)\n", pSystemModules->Modules[i].FullPathName);
359f2915f947abde1bba4358eada89941ee87cccvboxsync RTMemFree(pModule);
359f2915f947abde1bba4358eada89941ee87cccvboxsync continue;
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync strcat(szFullFilePath, lpPath);
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync strcpy(pModule->Info.szExePath, szFullFilePath);
359f2915f947abde1bba4358eada89941ee87cccvboxsync pModule->Info.modBaseAddr = (BYTE *)pSystemModules->Modules[i].ImageBase;
359f2915f947abde1bba4358eada89941ee87cccvboxsync pModule->Info.modBaseSize = pSystemModules->Modules[i].ImageSize;
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync pModule->Core.Key = pSystemModules->Modules[i].ImageBase;
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServicePageSharingRegisterModule(pModule, false /* don't check memory pages */);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "\n\n KERNEL MODULE NAME: %s", pModule->Info.szModule );
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "\n executable = %s", pModule->Info.szExePath );
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceVerbose(3, "\n base address = 0x%08X", (DWORD) pModule->Info.modBaseAddr );
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceVerbose(3, "\n flags = 0x%08X", pSystemModules->Modules[i].Flags);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "\n base size = %d", pModule->Info.modBaseSize );
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync pRec = &pModule->Core;
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync bool ret = RTAvlPVInsert(&pNewTree, pRec);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync Assert(ret); NOREF(ret);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsyncskipkernelmodules:
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (pBuffer)
359f2915f947abde1bba4358eada89941ee87cccvboxsync RTMemFree(pBuffer);
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync /* Delete leftover modules in the old tree. */
359f2915f947abde1bba4358eada89941ee87cccvboxsync RTAvlPVDestroy(&g_pKnownModuleTree, VBoxServicePageSharingEmptyTreeCallback, NULL);
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync /* Check all registered modules. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VbglR3CheckSharedModules();
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync /* Activate new module tree. */
359f2915f947abde1bba4358eada89941ee87cccvboxsync g_pKnownModuleTree = pNewTree;
359f2915f947abde1bba4358eada89941ee87cccvboxsync}
359f2915f947abde1bba4358eada89941ee87cccvboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/**
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * RTAvlPVDestroy callback.
359f2915f947abde1bba4358eada89941ee87cccvboxsync */
359f2915f947abde1bba4358eada89941ee87cccvboxsyncstatic DECLCALLBACK(int) VBoxServicePageSharingEmptyTreeCallback(PAVLPVNODECORE pNode, void *pvUser)
359f2915f947abde1bba4358eada89941ee87cccvboxsync{
359f2915f947abde1bba4358eada89941ee87cccvboxsync PKNOWN_MODULE pModule = (PKNOWN_MODULE)pNode;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync bool *pfUnregister = (bool *)pvUser;
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingEmptyTreeCallback %s %s\n", pModule->Info.szModule, pModule->szFileVersion);
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync /* Dereference module in the hypervisor. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if ( !pfUnregister
359f2915f947abde1bba4358eada89941ee87cccvboxsync || *pfUnregister)
359f2915f947abde1bba4358eada89941ee87cccvboxsync {
359f2915f947abde1bba4358eada89941ee87cccvboxsync int rc = VbglR3UnregisterSharedModule(pModule->Info.szModule, pModule->szFileVersion,
359f2915f947abde1bba4358eada89941ee87cccvboxsync (uintptr_t)pModule->Info.modBaseAddr, pModule->Info.modBaseSize);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync AssertRC(rc);
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (pModule->hModule)
359f2915f947abde1bba4358eada89941ee87cccvboxsync FreeLibrary(pModule->hModule);
359f2915f947abde1bba4358eada89941ee87cccvboxsync RTMemFree(pNode);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return 0;
359f2915f947abde1bba4358eada89941ee87cccvboxsync}
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync#elif TARGET_NT4
359f2915f947abde1bba4358eada89941ee87cccvboxsyncvoid VBoxServicePageSharingInspectGuest()
359f2915f947abde1bba4358eada89941ee87cccvboxsync{
359f2915f947abde1bba4358eada89941ee87cccvboxsync /* not implemented */
359f2915f947abde1bba4358eada89941ee87cccvboxsync}
359f2915f947abde1bba4358eada89941ee87cccvboxsync#else
359f2915f947abde1bba4358eada89941ee87cccvboxsyncvoid VBoxServicePageSharingInspectGuest()
359f2915f947abde1bba4358eada89941ee87cccvboxsync{
359f2915f947abde1bba4358eada89941ee87cccvboxsync /* @todo other platforms */
359f2915f947abde1bba4358eada89941ee87cccvboxsync}
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#endif
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/** @copydoc VBOXSERVICE::pfnPreInit */
359f2915f947abde1bba4358eada89941ee87cccvboxsyncstatic DECLCALLBACK(int) VBoxServicePageSharingPreInit(void)
359f2915f947abde1bba4358eada89941ee87cccvboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return VINF_SUCCESS;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync/** @copydoc VBOXSERVICE::pfnOption */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic DECLCALLBACK(int) VBoxServicePageSharingOption(const char **ppszShort, int argc, char **argv, int *pi)
359f2915f947abde1bba4358eada89941ee87cccvboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync NOREF(ppszShort);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync NOREF(argc);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync NOREF(argv);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync NOREF(pi);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return -1;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/** @copydoc VBOXSERVICE::pfnInit */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncstatic DECLCALLBACK(int) VBoxServicePageSharingInit(void)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInit\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync int rc = RTSemEventMultiCreate(&g_PageSharingEvent);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync AssertRCReturn(rc, rc);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#if defined(RT_OS_WINDOWS) && !defined(TARGET_NT4)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync g_pfnZwQuerySystemInformation = (PFNZWQUERYSYSTEMINFORMATION)RTLdrGetSystemSymbol("ntdll.dll", "ZwQuerySystemInformation");
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync rc = VbglR3GetSessionId(&g_idSession);
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (RT_FAILURE(rc))
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (rc == VERR_IO_GEN_FAILURE)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(0, "PageSharing: Page sharing support is not available by the host\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync else
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceError("VBoxServicePageSharingInit: Failed with rc=%Rrc\n", rc);
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync rc = VERR_SERVICE_DISABLED;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTSemEventMultiDestroy(g_PageSharingEvent);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync g_PageSharingEvent = NIL_RTSEMEVENTMULTI;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#endif
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return rc;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/** @copydoc VBOXSERVICE::pfnWorker */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncDECLCALLBACK(int) VBoxServicePageSharingWorker(bool volatile *pfShutdown)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Tell the control thread that it can continue
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * spawning services.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTThreadUserSignal(RTThreadSelf());
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Now enter the loop retrieving runtime data continuously.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync for (;;)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync BOOL fEnabled = VbglR3PageSharingIsEnabled();
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingWorker: enabled=%d\n", fEnabled);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (fEnabled)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServicePageSharingInspectGuest();
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Block for a minute.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync *
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * The event semaphore takes care of ignoring interruptions and it
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * allows us to implement service wakeup later.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (*pfShutdown)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync int rc = RTSemEventMultiWait(g_PageSharingEvent, 60000);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (*pfShutdown)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (rc != VERR_TIMEOUT && RT_FAILURE(rc))
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceError("VBoxServicePageSharingWorker: RTSemEventMultiWait failed; rc=%Rrc\n", rc);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync#if defined(RT_OS_WINDOWS) && !defined(TARGET_NT4)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync uint64_t idNewSession = g_idSession;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync rc = VbglR3GetSessionId(&idNewSession);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync AssertRC(rc);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (idNewSession != g_idSession)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync bool fUnregister = false;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingWorker: VM was restored!!\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /* The VM was restored, so reregister all modules the next time. */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTAvlPVDestroy(&g_pKnownModuleTree, VBoxServicePageSharingEmptyTreeCallback, &fUnregister);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync g_pKnownModuleTree = NULL;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync g_idSession = idNewSession;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync#endif
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTSemEventMultiDestroy(g_PageSharingEvent);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync g_PageSharingEvent = NIL_RTSEMEVENTMULTI;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingWorker: finished thread\n");
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return 0;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync#ifdef RT_OS_WINDOWS
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/**
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * This gets control when VBoxService is launched with -pagefusionfork by
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * VBoxServicePageSharingWorkerProcess().
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync *
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * @returns RTEXITCODE_SUCCESS.
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync *
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * @remarks It won't normally return since the parent drops the shutdown hint
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * via RTProcTerminate().
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync */
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsyncRTEXITCODE VBoxServicePageSharingInitFork(void)
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync{
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingInitFork\n");
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync bool fShutdown = false;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServicePageSharingInit();
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync VBoxServicePageSharingWorker(&fShutdown);
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync return RTEXITCODE_SUCCESS;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync}
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync/** @copydoc VBOXSERVICE::pfnWorker */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsyncDECLCALLBACK(int) VBoxServicePageSharingWorkerProcess(bool volatile *pfShutdown)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync{
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync RTPROCESS hProcess = NIL_RTPROCESS;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync int rc;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Tell the control thread that it can continue
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * spawning services.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync RTThreadUserSignal(RTThreadSelf());
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync /*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Now enter the loop retrieving runtime data continuously.
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync for (;;)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync BOOL fEnabled = VbglR3PageSharingIsEnabled();
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync VBoxServiceVerbose(3, "VBoxServicePageSharingWorkerProcess: enabled=%d\n", fEnabled);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync /*
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * Start a 2nd VBoxService process to deal with page fusion as we do
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * not wish to dummy load dlls into this process. (First load with
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync * DONT_RESOLVE_DLL_REFERENCES, 2nd normal -> dll init routines not called!)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync */
359f2915f947abde1bba4358eada89941ee87cccvboxsync if ( fEnabled
359f2915f947abde1bba4358eada89941ee87cccvboxsync && hProcess == NIL_RTPROCESS)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync {
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync char szExeName[256];
359f2915f947abde1bba4358eada89941ee87cccvboxsync char *pszExeName = RTProcGetExecutablePath(szExeName, sizeof(szExeName));
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (pszExeName)
359f2915f947abde1bba4358eada89941ee87cccvboxsync {
359f2915f947abde1bba4358eada89941ee87cccvboxsync char const *papszArgs[3];
359f2915f947abde1bba4358eada89941ee87cccvboxsync papszArgs[0] = pszExeName;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync papszArgs[1] = "pagefusion";
359f2915f947abde1bba4358eada89941ee87cccvboxsync papszArgs[2] = NULL;
359f2915f947abde1bba4358eada89941ee87cccvboxsync rc = RTProcCreate(pszExeName, papszArgs, RTENV_DEFAULT, 0 /* normal child */, &hProcess);
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (RT_FAILURE(rc))
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceError("VBoxServicePageSharingWorkerProcess: RTProcCreate %s failed; rc=%Rrc\n", pszExeName, rc);
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync
359f2915f947abde1bba4358eada89941ee87cccvboxsync /*
359f2915f947abde1bba4358eada89941ee87cccvboxsync * Block for a minute.
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync *
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync * The event semaphore takes care of ignoring interruptions and it
359f2915f947abde1bba4358eada89941ee87cccvboxsync * allows us to implement service wakeup later.
359f2915f947abde1bba4358eada89941ee87cccvboxsync */
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (*pfShutdown)
359f2915f947abde1bba4358eada89941ee87cccvboxsync break;
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync rc = RTSemEventMultiWait(g_PageSharingEvent, 60000);
359f2915f947abde1bba4358eada89941ee87cccvboxsync if (*pfShutdown)
359f2915f947abde1bba4358eada89941ee87cccvboxsync break;
0a7c9cc36ee176228eab4150144ad2561b0c4647vboxsync if (rc != VERR_TIMEOUT && RT_FAILURE(rc))
359f2915f947abde1bba4358eada89941ee87cccvboxsync {
359f2915f947abde1bba4358eada89941ee87cccvboxsync VBoxServiceError("VBoxServicePageSharingWorkerProcess: RTSemEventMultiWait failed; rc=%Rrc\n", rc);
359f2915f947abde1bba4358eada89941ee87cccvboxsync break;
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
359f2915f947abde1bba4358eada89941ee87cccvboxsync }
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync if (hProcess != NIL_RTPROCESS)
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync RTProcTerminate(hProcess);
2b73ea648457a8fb1ec027d1eeea63f879dad507vboxsync
RTSemEventMultiDestroy(g_PageSharingEvent);
g_PageSharingEvent = NIL_RTSEMEVENTMULTI;
VBoxServiceVerbose(3, "VBoxServicePageSharingWorkerProcess: finished thread\n");
return 0;
}
#endif /* RT_OS_WINDOWS */
/** @copydoc VBOXSERVICE::pfnTerm */
static DECLCALLBACK(void) VBoxServicePageSharingTerm(void)
{
VBoxServiceVerbose(3, "VBoxServicePageSharingTerm\n");
}
/** @copydoc VBOXSERVICE::pfnStop */
static DECLCALLBACK(void) VBoxServicePageSharingStop(void)
{
RTSemEventMultiSignal(g_PageSharingEvent);
}
/**
* The 'pagesharing' service description.
*/
VBOXSERVICE g_PageSharing =
{
/* pszName. */
"pagesharing",
/* pszDescription. */
"Page Sharing",
/* pszUsage. */
NULL,
/* pszOptions. */
NULL,
/* methods */
VBoxServicePageSharingPreInit,
VBoxServicePageSharingOption,
VBoxServicePageSharingInit,
#ifdef RT_OS_WINDOWS
VBoxServicePageSharingWorkerProcess,
#else
VBoxServicePageSharingWorker,
#endif
VBoxServicePageSharingStop,
VBoxServicePageSharingTerm
};