VBoxDrvTool.cpp revision e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * Windows Driver R0 Tooling.
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * Copyright (C) 2011 Oracle Corporation
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * This file is part of VirtualBox Open Source Edition (OSE), as
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * available from http://www.virtualbox.org. This file is free software;
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * you can redistribute it and/or modify it under the terms of the GNU
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * General Public License (GPL) as published by the Free Software
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * Foundation, in version 2 as it comes in the "COPYING" file of the
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync PVOID pvMem = ExAllocatePoolWithTag(NonPagedPool, cbBytes, VBOXDRVTOOL_MEMTAG);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolRegOpenKeyU(OUT PHANDLE phKey, IN PUNICODE_STRING pName, IN ACCESS_MASK fAccess)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync InitializeObjectAttributes(&ObjAttr, pName, OBJ_CASE_INSENSITIVE | OBJ_KERNEL_HANDLE, NULL, NULL);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolRegOpenKey(OUT PHANDLE phKey, IN PWCHAR pName, IN ACCESS_MASK fAccess)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync return VBoxDrvToolRegOpenKeyU(phKey, &RtlStr, fAccess);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolRegCloseKey(IN HANDLE hKey)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolRegQueryValueDword(IN HANDLE hKey, IN PWCHAR pName, OUT PULONG pDword)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolRegSetValueDword(IN HANDLE hKey, IN PWCHAR pName, OUT ULONG val)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncstatic NTSTATUS vboxDrvToolIoCompletionSetEvent(IN PDEVICE_OBJECT pDevObj, IN PIRP pIrp, IN PVOID pvContext)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolIoPostAsync(PDEVICE_OBJECT pDevObj, PIRP pIrp, PKEVENT pEvent)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync IoSetCompletionRoutine(pIrp, vboxDrvToolIoCompletionSetEvent, pEvent, TRUE, TRUE, TRUE);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolIoPostSync(PDEVICE_OBJECT pDevObj, PIRP pIrp)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeInitializeEvent(&Event, NotificationEvent, FALSE);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync NTSTATUS Status = VBoxDrvToolIoPostAsync(pDevObj, pIrp, &Event);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, NULL);
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync/* !!!NOTE: the caller MUST be the IRP owner!!! *
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync * !! one can not post threaded IRPs this way!! */
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolIoPostSyncWithTimeout(PDEVICE_OBJECT pDevObj, PIRP pIrp, ULONG dwTimeoutMs)
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync KeInitializeEvent(&Event, NotificationEvent, FALSE);
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync NTSTATUS Status = VBoxDrvToolIoPostAsync(pDevObj, pIrp, &Event);
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Interval.QuadPart = -(int64_t) dwTimeoutMs /* ms */ * 10000;
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync Status = KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, pInterval);
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync /* debugging only */
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync /* this may happen, but this is something the caller with timeout is not expecting */
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync /* wait for the IRP to complete */
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync KeWaitForSingleObject(&Event, Executive, KernelMode, FALSE, NULL);
e04781b223e38e3b0c386d7e7c1dd7f30e9f6e90vboxsync /* by this time the IRP is completed */
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(VOID) VBoxDrvToolRefWaitEqual(PVBOXDRVTOOL_REF pRef, uint32_t u32Val)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync while ((cRefs = ASMAtomicReadU32(&pRef->cRefs)) != u32Val)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync KeDelayExecutionThread(KernelMode, TRUE, &Interval);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsyncVBOXDRVTOOL_DECL(NTSTATUS) VBoxDrvToolStrCopy(PUNICODE_STRING pDst, CONST PUNICODE_STRING pSrc)
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync USHORT cbLength = pSrc->Length + sizeof (pDst->Buffer[0]);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDst->Buffer = (PWCHAR)vboxDrvToolMemAlloc(cbLength);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync RtlMoveMemory(pDst->Buffer, pSrc->Buffer, pSrc->Length);
97b634ea021fd984782256de4ba4ff31cdb96c47vboxsync pDst->Buffer[pSrc->Length / sizeof (pDst->Buffer[0])] = L'\0';