| 1 | #include <mach-o/loader.h> |
| 2 | #include <stdio.h> |
| 3 | #include <stdlib.h> |
| 4 | #include <string> |
| 5 | #include <vector> |
| 6 | |
| 7 | |
| 8 | // Normally these are picked up by including <mach/thread_status.h> |
| 9 | // but that does a compile time check for the build host arch and |
| 10 | // only defines the ARM register context constants when building on |
| 11 | // an arm system. We're creating fake corefiles, and might be |
| 12 | // creating them on an intel system. |
| 13 | #define ARM_THREAD_STATE 1 |
| 14 | #define ARM_THREAD_STATE_COUNT 17 |
| 15 | #define ARM_EXCEPTION_STATE 3 |
| 16 | #define ARM_EXCEPTION_STATE_COUNT 3 |
| 17 | #define ARM_THREAD_STATE64 6 |
| 18 | #define ARM_THREAD_STATE64_COUNT 68 |
| 19 | #define ARM_EXCEPTION_STATE64 7 |
| 20 | #define ARM_EXCEPTION_STATE64_COUNT 4 |
| 21 | |
| 22 | |
| 23 | union uint32_buf { |
| 24 | uint8_t bytebuf[4]; |
| 25 | uint32_t val; |
| 26 | }; |
| 27 | |
| 28 | union uint64_buf { |
| 29 | uint8_t bytebuf[8]; |
| 30 | uint64_t val; |
| 31 | }; |
| 32 | |
| 33 | void add_uint64(std::vector<uint8_t> &buf, uint64_t val) { |
| 34 | uint64_buf conv; |
| 35 | conv.val = val; |
| 36 | for (int i = 0; i < 8; i++) |
| 37 | buf.push_back(x: conv.bytebuf[i]); |
| 38 | } |
| 39 | |
| 40 | void add_uint32(std::vector<uint8_t> &buf, uint32_t val) { |
| 41 | uint32_buf conv; |
| 42 | conv.val = val; |
| 43 | for (int i = 0; i < 4; i++) |
| 44 | buf.push_back(x: conv.bytebuf[i]); |
| 45 | } |
| 46 | |
| 47 | std::vector<uint8_t> armv7_lc_thread_load_command() { |
| 48 | std::vector<uint8_t> data; |
| 49 | add_uint32(data, LC_THREAD); // thread_command.cmd |
| 50 | add_uint32(buf&: data, val: 104); // thread_command.cmdsize |
| 51 | add_uint32(buf&: data, ARM_THREAD_STATE); // thread_command.flavor |
| 52 | add_uint32(buf&: data, ARM_THREAD_STATE_COUNT); // thread_command.count |
| 53 | add_uint32(buf&: data, val: 0x00010000); // r0 |
| 54 | add_uint32(buf&: data, val: 0x00020000); // r1 |
| 55 | add_uint32(buf&: data, val: 0x00030000); // r2 |
| 56 | add_uint32(buf&: data, val: 0x00040000); // r3 |
| 57 | add_uint32(buf&: data, val: 0x00050000); // r4 |
| 58 | add_uint32(buf&: data, val: 0x00060000); // r5 |
| 59 | add_uint32(buf&: data, val: 0x00070000); // r6 |
| 60 | add_uint32(buf&: data, val: 0x00080000); // r7 |
| 61 | add_uint32(buf&: data, val: 0x00090000); // r8 |
| 62 | add_uint32(buf&: data, val: 0x000a0000); // r9 |
| 63 | add_uint32(buf&: data, val: 0x000b0000); // r10 |
| 64 | add_uint32(buf&: data, val: 0x000c0000); // r11 |
| 65 | add_uint32(buf&: data, val: 0x000d0000); // r12 |
| 66 | add_uint32(buf&: data, val: 0x000e0000); // sp |
| 67 | add_uint32(buf&: data, val: 0x000f0000); // lr |
| 68 | add_uint32(buf&: data, val: 0x00100000); // pc |
| 69 | add_uint32(buf&: data, val: 0x00110000); // cpsr |
| 70 | |
| 71 | add_uint32(buf&: data, ARM_EXCEPTION_STATE); // thread_command.flavor |
| 72 | add_uint32(buf&: data, ARM_EXCEPTION_STATE_COUNT); // thread_command.count |
| 73 | add_uint32(buf&: data, val: 0x00003f5c); // far |
| 74 | add_uint32(buf&: data, val: 0xf2000000); // esr |
| 75 | add_uint32(buf&: data, val: 0x00000000); // exception |
| 76 | |
| 77 | return data; |
| 78 | } |
| 79 | |
| 80 | std::vector<uint8_t> arm64_lc_thread_load_command() { |
| 81 | std::vector<uint8_t> data; |
| 82 | add_uint32(data, LC_THREAD); // thread_command.cmd |
| 83 | add_uint32(buf&: data, val: 312); // thread_command.cmdsize |
| 84 | add_uint32(buf&: data, ARM_THREAD_STATE64); // thread_command.flavor |
| 85 | add_uint32(buf&: data, ARM_THREAD_STATE64_COUNT); // thread_command.count |
| 86 | add_uint64(buf&: data, val: 0x0000000000000001); // x0 |
| 87 | add_uint64(buf&: data, val: 0x000000016fdff3c0); // x1 |
| 88 | add_uint64(buf&: data, val: 0x000000016fdff3d0); // x2 |
| 89 | add_uint64(buf&: data, val: 0x000000016fdff510); // x3 |
| 90 | add_uint64(buf&: data, val: 0x0000000000000000); // x4 |
| 91 | add_uint64(buf&: data, val: 0x0000000000000000); // x5 |
| 92 | add_uint64(buf&: data, val: 0x0000000000000000); // x6 |
| 93 | add_uint64(buf&: data, val: 0x0000000000000000); // x7 |
| 94 | add_uint64(buf&: data, val: 0x000000010000d910); // x8 |
| 95 | add_uint64(buf&: data, val: 0x0000000000000001); // x9 |
| 96 | add_uint64(buf&: data, val: 0xe1e88de000000000); // x10 |
| 97 | add_uint64(buf&: data, val: 0x0000000000000003); // x11 |
| 98 | add_uint64(buf&: data, val: 0x0000000000000148); // x12 |
| 99 | add_uint64(buf&: data, val: 0x0000000000004000); // x13 |
| 100 | add_uint64(buf&: data, val: 0x0000000000000008); // x14 |
| 101 | add_uint64(buf&: data, val: 0x0000000000000000); // x15 |
| 102 | add_uint64(buf&: data, val: 0x0000000000000000); // x16 |
| 103 | add_uint64(buf&: data, val: 0x0000000100003f5c); // x17 |
| 104 | add_uint64(buf&: data, val: 0x0000000000000000); // x18 |
| 105 | add_uint64(buf&: data, val: 0x0000000100003f5c); // x19 |
| 106 | add_uint64(buf&: data, val: 0x000000010000c000); // x20 |
| 107 | add_uint64(buf&: data, val: 0x000000010000d910); // x21 |
| 108 | add_uint64(buf&: data, val: 0x000000016fdff250); // x22 |
| 109 | add_uint64(buf&: data, val: 0x000000018ce12366); // x23 |
| 110 | add_uint64(buf&: data, val: 0x000000016fdff1d0); // x24 |
| 111 | add_uint64(buf&: data, val: 0x0000000000000001); // x25 |
| 112 | add_uint64(buf&: data, val: 0x0000000000000000); // x26 |
| 113 | add_uint64(buf&: data, val: 0x0000000000000000); // x27 |
| 114 | add_uint64(buf&: data, val: 0x0000000000000000); // x28 |
| 115 | add_uint64(buf&: data, val: 0x000000016fdff3a0); // fp |
| 116 | add_uint64(buf&: data, val: 0x000000018cd97f28); // lr |
| 117 | add_uint64(buf&: data, val: 0x000000016fdff140); // sp |
| 118 | add_uint64(buf&: data, val: 0x0000000100003f5c); // pc |
| 119 | add_uint32(buf&: data, val: 0x80001000); // cpsr |
| 120 | |
| 121 | add_uint32(buf&: data, val: 0x00000000); // padding |
| 122 | |
| 123 | add_uint32(buf&: data, ARM_EXCEPTION_STATE64); // thread_command.flavor |
| 124 | add_uint32(buf&: data, ARM_EXCEPTION_STATE64_COUNT); // thread_command.count |
| 125 | add_uint64(buf&: data, val: 0x0000000100003f5c); // far |
| 126 | add_uint32(buf&: data, val: 0xf2000000); // esr |
| 127 | add_uint32(buf&: data, val: 0x00000000); // exception |
| 128 | |
| 129 | return data; |
| 130 | } |
| 131 | |
| 132 | enum arch { unspecified, armv7, arm64 }; |
| 133 | |
| 134 | int main(int argc, char **argv) { |
| 135 | if (argc != 3) { |
| 136 | fprintf(stderr, |
| 137 | format: "usage: create-arm-corefiles [armv7|arm64] <output-core-name>\n" ); |
| 138 | exit(status: 1); |
| 139 | } |
| 140 | |
| 141 | arch arch = unspecified; |
| 142 | |
| 143 | if (strcmp(argv[1], "armv7" ) == 0) |
| 144 | arch = armv7; |
| 145 | else if (strcmp(argv[1], "arm64" ) == 0) |
| 146 | arch = arm64; |
| 147 | else { |
| 148 | fprintf(stderr, format: "unrecognized architecture %s\n" , argv[1]); |
| 149 | exit(status: 1); |
| 150 | } |
| 151 | |
| 152 | // An array of load commands (in the form of byte arrays) |
| 153 | std::vector<std::vector<uint8_t>> load_commands; |
| 154 | |
| 155 | // An array of corefile contents (page data, lc_note data, etc) |
| 156 | std::vector<uint8_t> payload; |
| 157 | |
| 158 | // First add all the load commands / payload so we can figure out how large |
| 159 | // the load commands will actually be. |
| 160 | if (arch == armv7) |
| 161 | load_commands.push_back(x: armv7_lc_thread_load_command()); |
| 162 | else if (arch == arm64) |
| 163 | load_commands.push_back(x: arm64_lc_thread_load_command()); |
| 164 | |
| 165 | int size_of_load_commands = 0; |
| 166 | for (const auto &lc : load_commands) |
| 167 | size_of_load_commands += lc.size(); |
| 168 | |
| 169 | int header_and_load_cmd_room = |
| 170 | sizeof(struct mach_header_64) + size_of_load_commands; |
| 171 | |
| 172 | // Erase the load commands / payload now that we know how much space is |
| 173 | // needed, redo it. |
| 174 | load_commands.clear(); |
| 175 | payload.clear(); |
| 176 | |
| 177 | if (arch == armv7) |
| 178 | load_commands.push_back(x: armv7_lc_thread_load_command()); |
| 179 | else if (arch == arm64) |
| 180 | load_commands.push_back(x: arm64_lc_thread_load_command()); |
| 181 | |
| 182 | struct mach_header_64 mh; |
| 183 | mh.magic = MH_MAGIC_64; |
| 184 | if (arch == armv7) { |
| 185 | mh.cputype = CPU_TYPE_ARM; |
| 186 | mh.cpusubtype = CPU_SUBTYPE_ARM_V7M; |
| 187 | } else if (arch == arm64) { |
| 188 | mh.cputype = CPU_TYPE_ARM64; |
| 189 | mh.cpusubtype = CPU_SUBTYPE_ARM64_ALL; |
| 190 | } |
| 191 | mh.filetype = MH_CORE; |
| 192 | mh.ncmds = load_commands.size(); |
| 193 | mh.sizeofcmds = size_of_load_commands; |
| 194 | mh.flags = 0; |
| 195 | mh.reserved = 0; |
| 196 | |
| 197 | FILE *f = fopen(filename: argv[2], modes: "w" ); |
| 198 | |
| 199 | if (f == nullptr) { |
| 200 | fprintf(stderr, format: "Unable to open file %s for writing\n" , argv[2]); |
| 201 | exit(status: 1); |
| 202 | } |
| 203 | |
| 204 | fwrite(&mh, sizeof(struct mach_header_64), 1, f); |
| 205 | |
| 206 | for (const auto &lc : load_commands) |
| 207 | fwrite(ptr: lc.data(), size: lc.size(), n: 1, s: f); |
| 208 | |
| 209 | fseek(stream: f, off: header_and_load_cmd_room, SEEK_SET); |
| 210 | |
| 211 | fwrite(ptr: payload.data(), size: payload.size(), n: 1, s: f); |
| 212 | |
| 213 | fclose(stream: f); |
| 214 | } |
| 215 | |