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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
|
--!A cross-platform build utility based on Lua
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
-- http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.
--
-- Copyright (C) 2015-present, Xmake Open Source Community.
--
-- @author ruki
-- @file libc.lua
--
-- define module: libc
local libc = libc or {}
-- save original interfaces
libc._malloc = libc._malloc or libc.malloc
libc._free = libc._free or libc.free
libc._memcpy = libc._memcpy or libc.memcpy
libc._memmov = libc._memmov or libc.memmov
libc._memset = libc._memset or libc.memset
libc._strndup = libc._strndup or libc.strndup
libc._dataptr = libc._dataptr or libc.dataptr
libc._byteof = libc._byteof or libc.byteof
libc._setbyte = libc._setbyte or libc.setbyte
-- load modules
local ffi = xmake._LUAJIT and require("ffi")
-- define ffi interfaces
if ffi then
ffi.cdef[[
void* malloc(size_t size);
void free(void* data);
void* memmove(void* dest, const void* src, size_t n);
]]
end
-- allocate memory
--
-- @param size the memory size in bytes
-- @param opt the options, e.g. {zeroed = true}
-- @return the cdata pointer
--
function libc.malloc(size, opt)
if ffi then
if opt and opt.gc then
return ffi.gc(ffi.cast("unsigned char*", ffi.C.malloc(size)), ffi.C.free)
else
return ffi.cast("unsigned char*", ffi.C.malloc(size))
end
else
local data, errors = libc._malloc(size)
if not data then
os.raise(errors)
end
return data
end
end
-- free memory
--
-- @param data the cdata pointer to free
--
function libc.free(data)
if ffi then
return ffi.C.free(data)
else
return libc._free(data)
end
end
-- copy memory
--
-- @param dst the destination pointer
-- @param src the source pointer
-- @param size the copy size in bytes
--
function libc.memcpy(dst, src, size)
if ffi then
return ffi.copy(dst, src, size)
else
return libc._memcpy(dst, src, size)
end
end
-- move memory (handles overlapping regions)
--
-- @param dst the destination pointer
-- @param src the source pointer
-- @param size the move size in bytes
--
function libc.memmov(dst, src, size)
if ffi then
return ffi.C.memmove(dst, src, size)
else
return libc._memmov(dst, src, size)
end
end
-- fill memory with a byte value
--
-- @param data the memory pointer
-- @param ch the byte value
-- @param size the fill size in bytes
--
function libc.memset(data, ch, size)
if ffi then
return ffi.fill(data, size, ch)
else
libc._memset(data, ch, size)
end
end
-- duplicate a string with length limit
--
-- @param s the source string
-- @param n the maximum length
-- @return the new cdata string pointer
--
function libc.strndup(s, n)
if ffi then
return ffi.string(s, n)
else
local s, errors = libc._strndup(s, n)
if not s then
os.raise(errors)
end
return s
end
end
-- get byte value at the given offset
--
-- @param data the memory pointer
-- @param offset the byte offset
-- @return the byte value
--
function libc.byteof(data, offset)
if ffi then
return data[offset]
else
return libc._byteof(data, offset)
end
end
-- set byte value at the given offset
--
-- @param data the memory pointer
-- @param offset the byte offset
-- @param value the byte value
--
function libc.setbyte(data, offset, value)
if ffi then
data[offset] = value
else
return libc._setbyte(data, offset, value)
end
end
-- get cdata pointer from string or bytes
--
-- @param data the string or bytes data
-- @param opt the options (optional)
-- @return the cdata pointer
--
function libc.dataptr(data, opt)
opt = opt or {}
if ffi and opt.ffi ~= false then
if opt.gc then
return ffi.gc(ffi.cast("unsigned char*", data), ffi.C.free)
else
return ffi.cast("unsigned char*", data)
end
else
return type(data) == "number" and data or libc._dataptr(data)
end
end
-- get the numeric address of a cdata pointer
--
-- @param data the cdata pointer
-- @param opt the options (optional)
-- @return the address as number
--
function libc.ptraddr(data, opt)
opt = opt or {}
if ffi and opt.ffi ~= false then
return tonumber(ffi.cast('unsigned long long', data))
else
return data
end
end
-- return module: libc
return libc
|
