authorgravatar for emekankurumeh@outlook.comemekoi <emekankurumeh@outlook.com> 2019-05-30 20:03:56-05:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2019-12-10 11:41:54-05:00
log29fd727b79682a0f3e8ad7340cf86db989b9109e
tree4614be3e4449817c7d508e70858aa77a8fbaf1c8
parent9561e7c6b9fb2d9ebcbfd611196db698372ae7bd
signature Commit is signed but in an unrecognized format.

fixed windows dynamic library loading and added loading for darwin

4 files changed, 103 insertions(+), 14 deletions(-)

lib/std/c/darwin.zig+4
......@@ -128,3 +128,7 @@ pub const pthread_attr_t = extern struct {
128128 __sig: c_long,
129129 __opaque: [56]u8,
130130};
131
132pub extern "c" fn dlopen(path: [*]const u8, mode: c_int) ?*c_void;
133pub extern "c" fn dlclose(handle: *c_void) c_int;
134pub extern "c" fn dlsym(handle: ?*c_void, symbol: [*]const u8) ?*c_void;
lib/std/dynamic_library.zig+80-10
......@@ -7,11 +7,13 @@ const assert = std.debug.assert;
77const testing = std.testing;
88const elf = std.elf;
99const windows = std.os.windows;
10const system = std.os.system;
1011const maxInt = std.math.maxInt;
1112
1213pub const DynLib = switch (builtin.os) {
13 .linux => LinuxDynLib,
14 .linux => if (builtin.link_libc) DlDynlib else LinuxDynLib,
1415 .windows => WindowsDynLib,
16 .macosx, .tvos, .watchos, .ios => DlDynlib,
1517 else => void,
1618};
1719
......@@ -99,12 +101,14 @@ pub fn linkmap_iterator(phdrs: []elf.Phdr) !LinkMap.Iterator {
99101}
100102
101103pub const LinuxDynLib = struct {
104 pub const Error = ElfLib.Error;
105
102106 elf_lib: ElfLib,
103107 fd: i32,
104108 memory: []align(mem.page_size) u8,
105109
106110 /// Trusts the file
107 pub fn open(path: []const u8) !DynLib {
111 pub fn open(path: []const u8) !LinuxDynLib {
108112 const fd = try os.open(path, 0, os.O_RDONLY | os.O_CLOEXEC);
109113 errdefer os.close(fd);
110114
......@@ -121,26 +125,42 @@ pub const LinuxDynLib = struct {
121125 );
122126 errdefer os.munmap(bytes);
123127
124 return DynLib{
128 return LinuxDynLib{
125129 .elf_lib = try ElfLib.init(bytes),
126130 .fd = fd,
127131 .memory = bytes,
128132 };
129133 }
130134
131 pub fn close(self: *DynLib) void {
135 pub fn close(self: *LinuxDynLib) void {
132136 os.munmap(self.memory);
133137 os.close(self.fd);
134138 self.* = undefined;
135139 }
136140
137 pub fn lookup(self: *DynLib, name: []const u8) ?usize {
138 return self.elf_lib.lookup("", name);
141 pub fn lookup(self: *LinuxDynLib, comptime T: type, name: []const u8) ?T {
142 if (self.elf_lib.lookup("", name)) |symbol| {
143 return @ptrCast(T, symbol);
144 } else {
145 return null;
146 }
139147 }
140148};
141149
142150pub const ElfLib = struct {
143151 strings: [*:0]u8,
152
153 pub const Error = error{
154 NotElfFile,
155 NotDynamicLibrary,
156 MissingDynamicLinkingInformation,
157 BaseNotFound,
158 ElfStringSectionNotFound,
159 ElfSymSectionNotFound,
160 ElfHashTableNotFound,
161 };
162
163 strings: [*]u8,
144164 syms: [*]elf.Sym,
145165 hashtab: [*]os.Elf_Symndx,
146166 versym: ?[*]u16,
......@@ -245,11 +265,12 @@ fn checkver(def_arg: *elf.Verdef, vsym_arg: i32, vername: []const u8, strings: [
245265}
246266
247267pub const WindowsDynLib = struct {
268 pub const Error = error{FileNotFound};
269
248270 dll: windows.HMODULE,
249271
250272 pub fn open(path: []const u8) !WindowsDynLib {
251273 const wpath = try windows.sliceToPrefixedFileW(path);
252
253274 return WindowsDynLib{
254275 .dll = try windows.LoadLibraryW(&wpath),
255276 };
......@@ -260,8 +281,57 @@ pub const WindowsDynLib = struct {
260281 self.* = undefined;
261282 }
262283
263 pub fn lookup(self: *WindowsDynLib, name: []const u8) ?usize {
264 return @ptrToInt(windows.kernel32.GetProcAddress(self.dll, name.ptr));
284 pub fn lookupC(self: *WindowsDynLib, comptime T: type, name: [*:0]const u8) ?T {
285 if (windows.kernel32.GetProcAddress(self.dll, name)) |addr| {
286 return @ptrCast(T, addr);
287 } else {
288 return null;
289 }
290 }
291
292 pub fn lookup(self: *DlDynlib, comptime T: type, comptime max_name_len: usize, name: []const u8) ?T {
293 const c_name: [max_name_len]u8 = undefined;
294 mem.copy(&c_name, name);
295 c_name[name.len] = 0;
296 return self.lookupC(T, &c_name);
297 }
298};
299
300pub const DlDynlib = struct {
301 pub const Error = error{FileNotFound};
302
303 handle: *c_void,
304
305 pub fn open(path: []const u8) !DlDynlib {
306 if (!builtin.link_libc and !os.darwin.is_the_target) {
307 @compileError("DlDynlib requires libc");
308 }
309
310 return DlDynlib{
311 .handle = system.dlopen(path.ptr, system.RTLD_LAZY) orelse {
312 return error.FileNotFound;
313 },
314 };
315 }
316
317 pub fn close(self: *DlDynlib) void {
318 _ = system.dlclose(self.handle);
319 self.* = undefined;
320 }
321
322 pub fn lookupC(self: *DlDynlib, comptime T: type, name: [*:0]const u8) ?T {
323 if (system.dlsym(self.handle, name)) |symbol| {
324 return @ptrCast(T, symbol);
325 } else {
326 return null;
327 }
328 }
329
330 pub fn lookup(self: *DlDynlib, comptime T: type, comptime max_name_len: usize, name: []const u8) ?T {
331 const c_name: [max_name_len]u8 = undefined;
332 mem.copy(&c_name, name);
333 c_name[name.len] = 0;
334 return self.lookupC(T, &c_name);
265335 }
266336};
267337
......@@ -269,6 +339,7 @@ test "dynamic_library" {
269339 const libname = switch (builtin.os) {
270340 .linux => "invalid_so.so",
271341 .windows => "invalid_dll.dll",
342 .macosx, .tvos, .watchos, .ios => "invalid_dylib.dylib",
272343 else => return,
273344 };
274345
......@@ -276,5 +347,4 @@ test "dynamic_library" {
276347 testing.expect(err == error.FileNotFound);
277348 return;
278349 };
279 @panic("Expected error from function");
280350}
lib/std/os/bits/darwin.zig+14
......@@ -1183,6 +1183,7 @@ pub fn S_ISSOCK(m: u32) bool {
11831183pub fn S_IWHT(m: u32) bool {
11841184 return m & S_IFMT == S_IFWHT;
11851185}
1186
11861187pub const HOST_NAME_MAX = 72;
11871188
11881189pub const AT_FDCWD = -2;
......@@ -1209,3 +1210,16 @@ pub const addrinfo = extern struct {
12091210 addr: ?*sockaddr,
12101211 next: ?*addrinfo,
12111212};
1213
1214pub const RTLD_LAZY = 0x1;
1215pub const RTLD_NOW = 0x2;
1216pub const RTLD_LOCAL = 0x4;
1217pub const RTLD_GLOBAL = 0x8;
1218pub const RTLD_NOLOAD = 0x10;
1219pub const RTLD_NODELETE = 0x80;
1220pub const RTLD_FIRST = 0x100;
1221
1222pub const RTLD_NEXT = @intToPtr(*c_void, ~maxInt(usize));
1223pub const RTLD_DEFAULT = @intToPtr(*c_void, ~maxInt(usize) - 1);
1224pub const RTLD_SELF = @intToPtr(*c_void, ~maxInt(usize) - 2);
1225pub const RTLD_MAIN_ONLY = @intToPtr(*c_void, ~maxInt(usize) - 4);
test/standalone.zig+5-4
......@@ -19,10 +19,11 @@ pub fn addCases(cases: *tests.StandaloneContext) void {
1919 cases.addBuildFile("test/standalone/use_alias/build.zig");
2020 cases.addBuildFile("test/standalone/brace_expansion/build.zig");
2121 cases.addBuildFile("test/standalone/empty_env/build.zig");
22 if (builtin.os == builtin.Os.linux) {
23 // TODO hook up the DynLib API for windows using LoadLibraryA
24 // TODO figure out how to make this work on darwin - probably libSystem has dlopen/dlsym in it
25 cases.addBuildFile("test/standalone/load_dynamic_library/build.zig");
22 switch (builtin.os) {
23 .linux, .windows, .macosx, .tvos, .watchos, .ios => {
24 cases.addBuildFile("test/standalone/load_dynamic_library/build.zig");
25 },
26 else => {},
2627 }
2728
2829 if (builtin.arch == builtin.Arch.x86_64) { // TODO add C ABI support for other architectures