https://github.com/akkartik/mu/blob/main/012elf.cc
1
2
3
4
5 :(before "End Main")
6 assert(argc > 1);
7 if (is_equal(argv[1], "run")) {
8
9 if (Trace_file.is_open()) Trace_stream = new trace_stream;
10 trace(2, "run") << "=== Starting to run" << end();
11 if (argc <= 2) {
12 raise << "Not enough arguments provided.\n" << die();
13 }
14 reset();
15 cerr << std::hex;
16 load_elf(argv[2], argc, argv);
17 while (EIP < End_of_program)
18 run_one_instruction();
19 raise << "executed past end of the world: " << EIP << " vs " << End_of_program << '\n' << end();
20 return 1;
21 }
22
23 :(code)
24 void load_elf(const string& filename, int argc, char* argv[]) {
25 int fd = open(filename.c_str(), O_RDONLY);
26 if (fd < 0) raise << filename.c_str() << ": open" << perr() << '\n' << die();
27 off_t size = lseek(fd, 0, SEEK_END);
28 lseek(fd, 0, SEEK_SET);
29 uint8_t* elf_contents = static_cast<uint8_t*>(malloc(size));
30 if (elf_contents == NULL) raise << "malloc(" << size << ')' << perr() << '\n' << die();
31 ssize_t read_size = read(fd, elf_contents, size);
32 if (size != read_size) raise << "read → " << size << " (!= " << read_size << ')' << perr() << '\n' << die();
33 load_elf_contents(elf_contents, size, argc, argv);
34 free(elf_contents);
35 }
36
37 void load_elf_contents(uint8_t* elf_contents, size_t size, int argc, char* argv[]) {
38 uint8_t magic[5] = {0};
39 memcpy(magic, elf_contents, 4);
40 if (memcmp(magic, "\177ELF", 4) != 0)
41 raise << "Invalid ELF file; starts with \"" << magic << '"' << die();
42 if # Copyright (C) 2009, 2010 Roman Zimbelmann <romanz@lavabit.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""Workaround to allow running single test cases directly"""
try:
from __init__ import init, Fake, OK, raise_ok, TODO
except:
from test import init, Fake, OK, raise_ok, TODO