1 /* 2 * Test if our calling convention gymnastics actually work 3 */ 4 5 #include <efi.h> 6 #include <efilib.h> 7 8 #ifdef __x86_64__ 9 #include <x86_64/pe.h> 10 #include <x86_64/efibind.h> 11 #endif 12 13 #if 0 14 asm volatile("out %0,%1" : : "a" ((uint8_t)a), "dN" (0x80)); 15 16 extern void dump_stack(void); 17 asm( ".globl dump_stack\n" 18 "dump_stack:\n" 19 " movq %rsp, %rdi\n" 20 " jmp *dump_stack_helper@GOTPCREL(%rip)\n" 21 ".size dump_stack, .-dump_stack"); 22 23 void dump_stack_helper(uint64_t rsp_val) 24 { 25 uint64_t *rsp = (uint64_t *)rsp_val; 26 int x; 27 28 Print(L"%%rsp: 0x%08x%08x stack:\r\n", 29 (rsp_val & 0xffffffff00000000) >>32, 30 rsp_val & 0xffffffff); 31 for (x = 0; x < 8; x++) { 32 Print(L"%08x: ", ((uint64_t)rsp) & 0xffffffff); 33 Print(L"%016x ", *rsp++); 34 Print(L"%016x ", *rsp++); 35 Print(L"%016x ", *rsp++); 36 Print(L"%016x\r\n", *rsp++); 37 } 38 } 39 #endif 40 41 EFI_STATUS EFI_FUNCTION test_failure_callback(void) 42 { 43 return EFI_UNSUPPORTED; 44 } 45 46 EFI_STATUS test_failure(void) 47 { 48 return uefi_call_wrapper(test_failure_callback, 0); 49 } 50 51 EFI_STATUS EFI_FUNCTION test_call0_callback(void) 52 { 53 return EFI_SUCCESS; 54 } 55 56 EFI_STATUS test_call0(void) 57 { 58 return uefi_call_wrapper(test_call0_callback, 0); 59 } 60 61 EFI_STATUS EFI_FUNCTION test_call1_callback(UINT32 a) 62 { 63 if (a != 0x12345678) { 64 return EFI_LOAD_ERROR; 65 } 66 return EFI_SUCCESS; 67 } 68 69 EFI_STATUS test_call1(void) 70 { 71 return uefi_call_wrapper(test_call1_callback, 1,0x12345678); 72 } 73 74 EFI_STATUS EFI_FUNCTION test_call2_callback(UINT32 a, UINT32 b) 75 { 76 if (a != 0x12345678) { 77 return EFI_LOAD_ERROR; 78 } 79 if (b != 0x23456789) { 80 return EFI_INVALID_PARAMETER; 81 } 82 return EFI_SUCCESS; 83 } 84 85 EFI_STATUS test_call2(void) 86 { 87 return uefi_call_wrapper(test_call2_callback, 2, 88 0x12345678, 0x23456789); 89 } 90 91 EFI_STATUS EFI_FUNCTION test_call3_callback(UINT32 a, UINT32 b, 92 UINT32 c) 93 { 94 if (a != 0x12345678) 95 return EFI_LOAD_ERROR; 96 if (b != 0x23456789) 97 return EFI_INVALID_PARAMETER; 98 if (c != 0x3456789a) 99 return EFI_UNSUPPORTED; 100 return EFI_SUCCESS; 101 } 102 103 EFI_STATUS test_call3(void) 104 { 105 return uefi_call_wrapper(test_call3_callback, 3, 106 0x12345678, 0x23456789, 0x3456789a); 107 } 108 109 EFI_STATUS EFI_FUNCTION test_call4_callback(UINT32 a, UINT32 b, 110 UINT32 c, UINT32 d) 111 { 112 if (a != 0x12345678) 113 return EFI_LOAD_ERROR; 114 if (b != 0x23456789) 115 return EFI_INVALID_PARAMETER; 116 if (c != 0x3456789a) 117 return EFI_UNSUPPORTED; 118 if (d != 0x456789ab) 119 return EFI_BAD_BUFFER_SIZE; 120 121 return EFI_SUCCESS; 122 } 123 124 EFI_STATUS test_call4(void) 125 { 126 return uefi_call_wrapper(test_call4_callback, 4, 127 0x12345678, 0x23456789, 0x3456789a, 0x456789ab); 128 } 129 130 EFI_STATUS EFI_FUNCTION test_call5_callback(UINT32 a, UINT32 b, 131 UINT32 c, UINT32 d, UINT32 e) 132 { 133 if (a != 0x12345678) 134 return EFI_LOAD_ERROR; 135 if (b != 0x23456789) 136 return EFI_INVALID_PARAMETER; 137 if (c != 0x3456789a) 138 return EFI_UNSUPPORTED; 139 if (d != 0x456789ab) 140 return EFI_BAD_BUFFER_SIZE; 141 if (e != 0x56789abc) 142 return EFI_BUFFER_TOO_SMALL; 143 144 return EFI_SUCCESS; 145 } 146 147 EFI_STATUS test_call5(void) 148 { 149 return uefi_call_wrapper(test_call5_callback, 5, 150 0x12345678, 0x23456789, 0x3456789a, 0x456789ab, 0x56789abc); 151 } 152 153 EFI_STATUS EFI_FUNCTION test_call6_callback(UINT32 a, UINT32 b, 154 UINT32 c, UINT32 d, UINT32 e, UINT32 f) 155 { 156 if (a != 0x12345678) 157 return EFI_LOAD_ERROR; 158 if (b != 0x23456789) 159 return EFI_INVALID_PARAMETER; 160 if (c != 0x3456789a) 161 return EFI_UNSUPPORTED; 162 if (d != 0x456789ab) 163 return EFI_BAD_BUFFER_SIZE; 164 if (e != 0x56789abc) 165 return EFI_BUFFER_TOO_SMALL; 166 if (f != 0x6789abcd) 167 return EFI_NOT_READY; 168 169 return EFI_SUCCESS; 170 } 171 172 EFI_STATUS test_call6(void) 173 { 174 return uefi_call_wrapper(test_call6_callback, 6, 175 0x12345678, 0x23456789, 0x3456789a, 0x456789ab, 0x56789abc, 176 0x6789abcd); 177 } 178 179 EFI_STATUS EFI_FUNCTION test_call7_callback(UINT32 a, UINT32 b, 180 UINT32 c, UINT32 d, UINT32 e, UINT32 f, UINT32 g) 181 { 182 if (a != 0x12345678) 183 return EFI_LOAD_ERROR; 184 if (b != 0x23456789) 185 return EFI_INVALID_PARAMETER; 186 if (c != 0x3456789a) 187 return EFI_UNSUPPORTED; 188 if (d != 0x456789ab) 189 return EFI_BAD_BUFFER_SIZE; 190 if (e != 0x56789abc) 191 return EFI_BUFFER_TOO_SMALL; 192 if (f != 0x6789abcd) 193 return EFI_NOT_READY; 194 if (g != 0x789abcde) 195 return EFI_DEVICE_ERROR; 196 197 return EFI_SUCCESS; 198 } 199 200 EFI_STATUS test_call7(void) 201 { 202 return uefi_call_wrapper(test_call7_callback, 7, 203 0x12345678, 0x23456789, 0x3456789a, 0x456789ab, 204 0x56789abc, 0x6789abcd, 0x789abcde); 205 } 206 207 EFI_STATUS EFI_FUNCTION test_call8_callback(UINT32 a, UINT32 b, 208 UINT32 c, UINT32 d, UINT32 e, UINT32 f, UINT32 g, UINT32 h) 209 { 210 if (a != 0x12345678) 211 return EFI_LOAD_ERROR; 212 if (b != 0x23456789) 213 return EFI_INVALID_PARAMETER; 214 if (c != 0x3456789a) 215 return EFI_UNSUPPORTED; 216 if (d != 0x456789ab) 217 return EFI_BAD_BUFFER_SIZE; 218 if (e != 0x56789abc) 219 return EFI_BUFFER_TOO_SMALL; 220 if (f != 0x6789abcd) 221 return EFI_NOT_READY; 222 if (g != 0x789abcde) 223 return EFI_DEVICE_ERROR; 224 if (h != 0x89abcdef) 225 return EFI_WRITE_PROTECTED; 226 227 return EFI_SUCCESS; 228 } 229 230 EFI_STATUS test_call8(void) 231 { 232 return uefi_call_wrapper(test_call8_callback, 8, 233 0x12345678, 234 0x23456789, 235 0x3456789a, 236 0x456789ab, 237 0x56789abc, 238 0x6789abcd, 239 0x789abcde, 240 0x89abcdef); 241 } 242 243 EFI_STATUS EFI_FUNCTION test_call9_callback(UINT32 a, UINT32 b, 244 UINT32 c, UINT32 d, UINT32 e, UINT32 f, UINT32 g, UINT32 h, UINT32 i) 245 { 246 if (a != 0x12345678) 247 return EFI_LOAD_ERROR; 248 if (b != 0x23456789) 249 return EFI_INVALID_PARAMETER; 250 if (c != 0x3456789a) 251 return EFI_UNSUPPORTED; 252 if (d != 0x456789ab) 253 return EFI_BAD_BUFFER_SIZE; 254 if (e != 0x56789abc) 255 return EFI_BUFFER_TOO_SMALL; 256 if (f != 0x6789abcd) 257 return EFI_NOT_READY; 258 if (g != 0x789abcde) 259 return EFI_DEVICE_ERROR; 260 if (h != 0x89abcdef) 261 return EFI_WRITE_PROTECTED; 262 if (i != 0x9abcdef0) 263 return EFI_OUT_OF_RESOURCES; 264 265 return EFI_SUCCESS; 266 } 267 268 EFI_STATUS test_call9(void) 269 { 270 return uefi_call_wrapper(test_call9_callback, 9, 271 0x12345678, 272 0x23456789, 273 0x3456789a, 274 0x456789ab, 275 0x56789abc, 276 0x6789abcd, 277 0x789abcde, 278 0x89abcdef, 279 0x9abcdef0); 280 } 281 282 extern EFI_STATUS test_call10(void); 283 EFI_STATUS EFI_FUNCTION test_call10_callback(UINT32 a, UINT32 b, 284 UINT32 c, UINT32 d, UINT32 e, UINT32 f, UINT32 g, UINT32 h, UINT32 i, 285 UINT32 j) 286 { 287 if (a != 0x12345678) 288 return EFI_LOAD_ERROR; 289 if (b != 0x23456789) 290 return EFI_INVALID_PARAMETER; 291 if (c != 0x3456789a) 292 return EFI_UNSUPPORTED; 293 if (d != 0x456789ab) 294 return EFI_BAD_BUFFER_SIZE; 295 if (e != 0x56789abc) 296 return EFI_BUFFER_TOO_SMALL; 297 if (f != 0x6789abcd) 298 return EFI_NOT_READY; 299 if (g != 0x789abcde) 300 return EFI_DEVICE_ERROR; 301 if (h != 0x89abcdef) 302 return EFI_WRITE_PROTECTED; 303 if (i != 0x9abcdef0) 304 return EFI_OUT_OF_RESOURCES; 305 if (j != 0xabcdef01) 306 return EFI_VOLUME_CORRUPTED; 307 308 return EFI_SUCCESS; 309 } 310 311 EFI_STATUS test_call10(void) 312 { 313 return uefi_call_wrapper(test_call10_callback, 10, 314 0x12345678, 315 0x23456789, 316 0x3456789a, 317 0x456789ab, 318 0x56789abc, 319 0x6789abcd, 320 0x789abcde, 321 0x89abcdef, 322 0x9abcdef0, 323 0xabcdef01); 324 } 325 326 EFI_STATUS 327 efi_main (EFI_HANDLE *image, EFI_SYSTEM_TABLE *systab) 328 { 329 EFI_STATUS rc = EFI_SUCCESS; 330 331 InitializeLib(image, systab); 332 PoolAllocationType = 2; /* klooj */ 333 334 #ifndef __x86_64__ 335 uefi_call_wrapper(systab->ConOut->OutputString, 2, systab->ConOut, 336 L"This test is only valid on x86_64\n"); 337 return EFI_UNSUPPORTED; 338 #endif 339 340 __asm__ volatile("out %0,%1" : : "a" ((uint8_t)0x14), "dN" (0x80)); 341 342 Print(L"Hello\r\n"); 343 rc = test_failure(); 344 if (EFI_ERROR(rc)) { 345 Print(L"Returning Failure works\n"); 346 } else { 347 Print(L"Returning failure doesn't work.\r\n"); 348 Print(L"%%rax was 0x%016x, should have been 0x%016x\n", 349 rc, EFI_UNSUPPORTED); 350 return EFI_INVALID_PARAMETER; 351 } 352 353 rc = test_call0(); 354 if (!EFI_ERROR(rc)) { 355 Print(L"0 args works just fine here.\r\n"); 356 } else { 357 Print(L"0 args failed: 0x%016x\n", rc); 358 return rc; 359 } 360 361 rc = test_call1(); 362 if (!EFI_ERROR(rc)) { 363 Print(L"1 arg works just fine here.\r\n"); 364 } else { 365 Print(L"1 arg failed: 0x%016x\n", rc); 366 return rc; 367 } 368 369 rc = test_call2(); 370 if (!EFI_ERROR(rc)) { 371 Print(L"2 args works just fine here.\r\n"); 372 } else { 373 Print(L"2 args failed: 0x%016x\n", rc); 374 return rc; 375 } 376 377 rc = test_call3(); 378 if (!EFI_ERROR(rc)) { 379 Print(L"3 args works just fine here.\r\n"); 380 } else { 381 Print(L"3 args failed: 0x%016x\n", rc); 382 return rc; 383 } 384 385 rc = test_call4(); 386 if (!EFI_ERROR(rc)) { 387 Print(L"4 args works just fine here.\r\n"); 388 } else { 389 Print(L"4 args failed: 0x%016x\n", rc); 390 return rc; 391 } 392 393 rc = test_call5(); 394 if (!EFI_ERROR(rc)) { 395 Print(L"5 args works just fine here.\r\n"); 396 } else { 397 Print(L"5 args failed: 0x%016x\n", rc); 398 return rc; 399 } 400 401 rc = test_call6(); 402 if (!EFI_ERROR(rc)) { 403 Print(L"6 args works just fine here.\r\n"); 404 } else { 405 Print(L"6 args failed: 0x%016x\n", rc); 406 return rc; 407 } 408 409 rc = test_call7(); 410 if (!EFI_ERROR(rc)) { 411 Print(L"7 args works just fine here.\r\n"); 412 } else { 413 Print(L"7 args failed: 0x%016x\n", rc); 414 return rc; 415 } 416 417 rc = test_call8(); 418 if (!EFI_ERROR(rc)) { 419 Print(L"8 args works just fine here.\r\n"); 420 } else { 421 Print(L"8 args failed: 0x%016x\n", rc); 422 return rc; 423 } 424 425 rc = test_call9(); 426 if (!EFI_ERROR(rc)) { 427 Print(L"9 args works just fine here.\r\n"); 428 } else { 429 Print(L"9 args failed: 0x%016x\n", rc); 430 return rc; 431 } 432 433 rc = test_call10(); 434 if (!EFI_ERROR(rc)) { 435 Print(L"10 args works just fine here.\r\n"); 436 } else { 437 Print(L"10 args failed: 0x%016x\n", rc); 438 return rc; 439 } 440 441 return rc; 442 } 443
