VirtualBox

source: vbox/trunk/src/VBox/Runtime/r0drv/nt/alloc-r0drv-nt.cpp@ 94155

Last change on this file since 94155 was 93200, checked in by vboxsync, 3 years ago

IPRT/alloc-r0drv-nt.cpp: Must apply the NonPagePoolNx selection to both branches in rtR0MemAllocEx. (Yes, g_uRtNtVersion is the first thing that rtR0InitNative sets, so we shouldn't be called before it has a reasonably correct value.) bugref:10162

  • Property svn:eol-style set to native
  • Property svn:keywords set to Id Revision
File size: 4.8 KB
Line 
1/* $Id: alloc-r0drv-nt.cpp 93200 2022-01-12 13:51:29Z vboxsync $ */
2/** @file
3 * IPRT - Memory Allocation, Ring-0 Driver, NT.
4 */
5
6/*
7 * Copyright (C) 2006-2022 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 *
17 * The contents of this file may alternatively be used under the terms
18 * of the Common Development and Distribution License Version 1.0
19 * (CDDL) only, as it comes in the "COPYING.CDDL" file of the
20 * VirtualBox OSE distribution, in which case the provisions of the
21 * CDDL are applicable instead of those of the GPL.
22 *
23 * You may elect to license modified versions of this file under the
24 * terms and conditions of either the GPL or the CDDL or both.
25 */
26
27
28/*********************************************************************************************************************************
29* Header Files *
30*********************************************************************************************************************************/
31#include "the-nt-kernel.h"
32#include "internal/iprt.h"
33#include <iprt/mem.h>
34
35#include <iprt/assert.h>
36#include <iprt/errcore.h>
37#include "r0drv/alloc-r0drv.h"
38#include "internal-r0drv-nt.h"
39
40
41/**
42 * OS specific allocation function.
43 */
44DECLHIDDEN(int) rtR0MemAllocEx(size_t cb, uint32_t fFlags, PRTMEMHDR *ppHdr)
45{
46 if (!(fFlags & RTMEMHDR_FLAG_ANY_CTX))
47 {
48 PRTMEMHDR pHdr;
49 POOL_TYPE const enmPoolType = !(fFlags & RTMEMHDR_FLAG_EXEC) && g_uRtNtVersion >= RTNT_MAKE_VERSION(8,0)
50 ? NonPagedPoolNx : NonPagedPool;
51 if (g_pfnrtExAllocatePoolWithTag)
52 pHdr = (PRTMEMHDR)g_pfnrtExAllocatePoolWithTag(enmPoolType, cb + sizeof(*pHdr), IPRT_NT_POOL_TAG);
53 else
54 {
55 fFlags |= RTMEMHDR_FLAG_UNTAGGED;
56 pHdr = (PRTMEMHDR)ExAllocatePool(enmPoolType, cb + sizeof(*pHdr));
57 }
58 if (pHdr)
59 {
60 pHdr->u32Magic = RTMEMHDR_MAGIC;
61 pHdr->fFlags = fFlags;
62 pHdr->cb = (uint32_t)cb; Assert(pHdr->cb == cb);
63 pHdr->cbReq = (uint32_t)cb;
64 *ppHdr = pHdr;
65 return VINF_SUCCESS;
66 }
67 return VERR_NO_MEMORY;
68 }
69 return VERR_NOT_SUPPORTED;
70}
71
72
73/**
74 * OS specific free function.
75 */
76DECLHIDDEN(void) rtR0MemFree(PRTMEMHDR pHdr)
77{
78 pHdr->u32Magic = RTMEMHDR_MAGIC_DEAD;
79 if (g_pfnrtExFreePoolWithTag && !(pHdr->fFlags & RTMEMHDR_FLAG_UNTAGGED))
80 g_pfnrtExFreePoolWithTag(pHdr, IPRT_NT_POOL_TAG);
81 else
82 ExFreePool(pHdr);
83}
84
85
86/**
87 * Allocates physical contiguous memory (below 4GB).
88 * The allocation is page aligned and its contents is undefined.
89 *
90 * @returns Pointer to the memory block. This is page aligned.
91 * @param pPhys Where to store the physical address.
92 * @param cb The allocation size in bytes. This is always
93 * rounded up to PAGE_SIZE.
94 */
95RTR0DECL(void *) RTMemContAlloc(PRTCCPHYS pPhys, size_t cb)
96{
97 /*
98 * validate input.
99 */
100 AssertPtr(pPhys);
101 Assert(cb > 0);
102
103 /*
104 * Allocate and get physical address.
105 * Make sure the return is page aligned.
106 */
107 PHYSICAL_ADDRESS MaxPhysAddr;
108 MaxPhysAddr.HighPart = 0;
109 MaxPhysAddr.LowPart = 0xffffffff;
110 cb = RT_ALIGN_Z(cb, PAGE_SIZE);
111 void *pv = MmAllocateContiguousMemory(cb, MaxPhysAddr);
112 if (pv)
113 {
114 if (!((uintptr_t)pv & PAGE_OFFSET_MASK)) /* paranoia */
115 {
116 PHYSICAL_ADDRESS PhysAddr = MmGetPhysicalAddress(pv);
117 if (!PhysAddr.HighPart) /* paranoia */
118 {
119 *pPhys = (RTCCPHYS)PhysAddr.LowPart;
120 return pv;
121 }
122
123 /* failure */
124 AssertMsgFailed(("MMAllocContiguousMemory returned high address! PhysAddr=%RX64\n", (uint64_t)PhysAddr.QuadPart));
125 }
126 else
127 AssertMsgFailed(("MMAllocContiguousMemory didn't return a page aligned address - %p!\n", pv));
128
129 MmFreeContiguousMemory(pv);
130 }
131
132 return NULL;
133}
134
135
136/**
137 * Frees memory allocated ysing RTMemContAlloc().
138 *
139 * @param pv Pointer to return from RTMemContAlloc().
140 * @param cb The cb parameter passed to RTMemContAlloc().
141 */
142RTR0DECL(void) RTMemContFree(void *pv, size_t cb)
143{
144 if (pv)
145 {
146 Assert(cb > 0); NOREF(cb);
147 AssertMsg(!((uintptr_t)pv & PAGE_OFFSET_MASK), ("pv=%p\n", pv));
148 MmFreeContiguousMemory(pv);
149 }
150}
151
Note: See TracBrowser for help on using the repository browser.

© 2024 Oracle Support Privacy / Do Not Sell My Info Terms of Use Trademark Policy Automated Access Etiquette