Open Menu / testing / test-cases / kernel-fixups-x86_64.dtest / main.c
Loading...
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

// BOOT_ARGSxx: amfi=3 cs_enforcement_disable=1

// xxBUILDxx(macos|x86_64):  $CC main.c -o $BUILD_DIR/kernel-fixups.exe -Wl,-static -mkernel -nostdlib -Wl,-add_split_seg_info -Wl,-e,__start -Wl,-pie -Wl,-pagezero_size,0x0 -Wl,-install_name,/usr/lib/swift/split.seg.v2.hack -fno-stack-protector -fno-builtin -ffreestanding -Wl,-segprot,__HIB,rx,rx -Wl,-image_base,0x8000 -Wl,-segaddr,__HIB,0x4000
// xxBUILDxx(macos|x86_64):  $APP_CACHE_UTIL -create-kernel-collection $BUILD_DIR/kernel.kc -kernel $BUILD_DIR/kernel-fixups.exe

// BUILDxx(macos,ios,tvos,watchos,bridgeos):

// xxRUN_STATIC:    $RUN_STATIC ./kernel.kc

// This tests that unaligned fixups work in x86_64

#include "../kernel-test-runner.h"
#include "../kernel-fixups.h"
#include "../kernel-classic-relocs.h"

#define printf(...) hostFuncs->printf(__VA_ARGS__)

int x = 1;

struct __attribute__((packed)) __attribute__((aligned((4096)))) PackedS {
    int     i0; // aligned to 8
    int*    p0; // aligned to 4
    int     i1; // aligned to 4
    int*    p1; // aligned to 8
    char    i2; // aligned to 8
    int*    p2; // aligned to 1
};
struct PackedS ps = { 0, &x, 0, &x, 0, &x };

__attribute__((section(("__HIB, __text"))))
int _start(const TestRunnerFunctions* hostFuncs)
{
    const void* slideBasePointers[4];
    slideBasePointers[0] = hostFuncs->basePointers[0];
    slideBasePointers[1] = hostFuncs->basePointers[1];
    slideBasePointers[2] = hostFuncs->basePointers[2];
    slideBasePointers[3] = hostFuncs->basePointers[3];
    int slideReturnCode = slide(hostFuncs->mhs[0], slideBasePointers, hostFuncs->printf);
    if ( slideReturnCode != 0 ) {
        FAIL("slide = %d\n", slideReturnCode);
        return 0;
    }

    int slideClassicReturnCode = slideClassic(hostFuncs->mhs[0], hostFuncs->printf);
    if ( slideClassicReturnCode != 0 ) {
        FAIL("mhs[0] slide classic = %d\n", slideClassicReturnCode);
        return 0;
    }

    LOG("Done sliding");

    if ( ps.p0[0] != x ) {
    	FAIL("ps.p1[0] != x, %d != %d\n", ps.p0[0], x);
    	return 0;
    }

    if ( ps.p1[0] != x ) {
        FAIL("ps.p1[0] != x, %d != %d\n", ps.p1[0], x);
        return 0;
    }

    if ( ps.p2[0] != x ) {
        FAIL("ps.p2[0] != x, %d != %d\n", ps.p2[0], x);
        return 0;
    }

    PASS("Success");
    return 0;
}