| ... | ... | @@ -1,6 +1,8 @@ |
| 1 | 1 | const std = @import("../../std.zig"); |
| 2 | 2 | const builtin = @import("builtin"); |
| 3 | 3 | const linux = std.os.linux; |
| 4 | const mem = std.mem; |
| 5 | const elf = std.elf; |
| 4 | 6 | const expect = std.testing.expect; |
| 5 | 7 | |
| 6 | 8 | test "getpid" { |
| ... | ... | @@ -42,3 +44,42 @@ test "timer" { |
| 42 | 44 | // TODO implicit cast from *[N]T to [*]T |
| 43 | 45 | err = linux.epoll_wait(@intCast(i32, epoll_fd), @ptrCast([*]linux.epoll_event, &events), 8, -1); |
| 44 | 46 | } |
| 47 | |
| 48 | export fn iter_fn(info: *linux.dl_phdr_info, size: usize, data: ?*usize) i32 { |
| 49 | var counter = data.?; |
| 50 | // Count how many libraries are loaded |
| 51 | counter.* += usize(1); |
| 52 | |
| 53 | // The image should contain at least a PT_LOAD segment |
| 54 | if (info.dlpi_phnum < 1) return -1; |
| 55 | |
| 56 | // Quick & dirty validation of the phdr pointers, make sure we're not |
| 57 | // pointing to some random gibberish |
| 58 | var i: usize = 0; |
| 59 | var found_load = false; |
| 60 | while (i < info.dlpi_phnum) : (i += 1) { |
| 61 | const phdr = info.dlpi_phdr[i]; |
| 62 | |
| 63 | if (phdr.p_type != elf.PT_LOAD) continue; |
| 64 | |
| 65 | // Find the ELF header |
| 66 | const elf_header = @intToPtr(*elf.Ehdr, phdr.p_vaddr - phdr.p_offset); |
| 67 | // Validate the magic |
| 68 | if (!mem.eql(u8, elf_header.e_ident[0..], "\x7fELF")) return -1; |
| 69 | // Consistency check |
| 70 | if (elf_header.e_phnum != info.dlpi_phnum) return -1; |
| 71 | |
| 72 | found_load = true; |
| 73 | break; |
| 74 | } |
| 75 | |
| 76 | if (!found_load) return -1; |
| 77 | |
| 78 | return 42; |
| 79 | } |
| 80 | |
| 81 | test "dl_iterate_phdr" { |
| 82 | var counter: usize = 0; |
| 83 | expect(linux.dl_iterate_phdr(usize, iter_fn, &counter) != 0); |
| 84 | expect(counter != 0); |
| 85 | } |
LemonBoy <thatlemon@gmail.com> 2019-05-05 13:00:32+02:00
std