1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
|
/* $Id: allocex.cpp $ */
/** @file
* IPRT - Memory Allocation, Extended Alloc and Free Functions for Ring-3, posix.
*/
/*
* Copyright (C) 2006-2019 Oracle Corporation
*
* This file is part of VirtualBox Open Source Edition (OSE), as
* available from http://www.virtualbox.org. This file is free software;
* you can redistribute it and/or modify it under the terms of the GNU
* General Public License (GPL) as published by the Free Software
* Foundation, in version 2 as it comes in the "COPYING" file of the
* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
*
* The contents of this file may alternatively be used under the terms
* of the Common Development and Distribution License Version 1.0
* (CDDL) only, as it comes in the "COPYING.CDDL" file of the
* VirtualBox OSE distribution, in which case the provisions of the
* CDDL are applicable instead of those of the GPL.
*
* You may elect to license modified versions of this file under the
* terms and conditions of either the GPL or the CDDL or both.
*/
/*********************************************************************************************************************************
* Header Files *
*********************************************************************************************************************************/
#define RTMEM_NO_WRAP_TO_EF_APIS
#include <iprt/mem.h>
#include "internal/iprt.h"
#include <iprt/assert.h>
#include <iprt/err.h>
#include <iprt/string.h>
#include "internal/magics.h"
#include "allocex.h"
RTDECL(int) RTMemAllocExTag(size_t cb, size_t cbAlignment, uint32_t fFlags, const char *pszTag, void **ppv) RT_NO_THROW_DEF
{
RT_NOREF_PV(pszTag);
/*
* Validate and adjust input.
*/
AssertMsgReturn(!(fFlags & ~RTMEMALLOCEX_FLAGS_VALID_MASK), ("%#x\n", fFlags), VERR_INVALID_PARAMETER);
AssertReturn(cb > 0, VERR_INVALID_PARAMETER);
AssertReturn(RT_IS_POWER_OF_TWO(cbAlignment), VERR_INVALID_PARAMETER);
AssertMsgReturn(cbAlignment <= sizeof(void *), ("%zu (%#x)\n", cbAlignment, cbAlignment), VERR_UNSUPPORTED_ALIGNMENT);
if (fFlags & RTMEMALLOCEX_FLAGS_ANY_CTX)
return VERR_NOT_SUPPORTED;
/*
* Align the request.
*/
size_t cbAligned = cb;
if (cbAlignment)
cbAligned = RT_ALIGN_Z(cb, cbAlignment);
else
cbAligned = RT_ALIGN_Z(cb, sizeof(uint64_t));
AssertMsgReturn(cbAligned >= cb && cbAligned <= ~(size_t)0, ("cbAligned=%#zx cb=%#zx", cbAligned, cb),
VERR_INVALID_PARAMETER);
/*
* Allocate the requested memory.
*/
void *pv;
if (fFlags & (RTMEMALLOCEX_FLAGS_16BIT_REACH | RTMEMALLOCEX_FLAGS_32BIT_REACH))
{
int rc;
if (fFlags & RTMEMALLOCEX_FLAGS_16BIT_REACH)
rc = rtMemAllocEx16BitReach(cbAligned + sizeof(RTMEMHDRR3), fFlags, &pv);
else
rc = rtMemAllocEx32BitReach(cbAligned + sizeof(RTMEMHDRR3), fFlags, &pv);
if (RT_FAILURE(rc))
return rc;
}
else if (fFlags & RTMEMALLOCEX_FLAGS_EXEC)
{
pv = RTMemExecAlloc(cbAligned + sizeof(RTMEMHDRR3));
if ((fFlags & RTMEMALLOCEX_FLAGS_ZEROED) && pv)
RT_BZERO(pv, cbAligned + sizeof(RTMEMHDRR3));
}
else if (fFlags & RTMEMALLOCEX_FLAGS_ZEROED)
pv = RTMemAllocZ(cbAligned + sizeof(RTMEMHDRR3));
else
pv = RTMemAlloc(cbAligned + sizeof(RTMEMHDRR3));
if (!pv)
return VERR_NO_MEMORY;
/*
* Fill in the header and return.
*/
PRTMEMHDRR3 pHdr = (PRTMEMHDRR3)pv;
pHdr->u32Magic = RTMEMHDR_MAGIC;
pHdr->fFlags = fFlags;
pHdr->cb = (uint32_t)cbAligned;
pHdr->cbReq = (uint32_t)cb;
*ppv = pHdr + 1;
return VINF_SUCCESS;
}
RT_EXPORT_SYMBOL(RTMemAllocExTag);
RTDECL(void) RTMemFreeEx(void *pv, size_t cb) RT_NO_THROW_DEF
{
if (!pv)
return;
AssertPtr(pv);
PRTMEMHDRR3 pHdr = (PRTMEMHDRR3)pv - 1;
AssertMsg(pHdr->u32Magic == RTMEMHDR_MAGIC, ("pHdr->u32Magic=%RX32 pv=%p cb=%#x\n", pHdr->u32Magic, pv, cb));
pHdr->u32Magic = RTMEMHDR_MAGIC_DEAD;
Assert(pHdr->cbReq == cb); RT_NOREF_PV(cb);
if (pHdr->fFlags & (RTMEMALLOCEX_FLAGS_16BIT_REACH | RTMEMALLOCEX_FLAGS_32BIT_REACH))
rtMemFreeExYyBitReach(pHdr, pHdr->cb + sizeof(*pHdr), pHdr->fFlags);
else if (pHdr->fFlags & RTMEMALLOCEX_FLAGS_EXEC)
RTMemExecFree(pHdr, pHdr->cb + sizeof(*pHdr));
else
RTMemFree(pHdr);
}
RT_EXPORT_SYMBOL(RTMemFreeEx);
|
