summaryrefslogtreecommitdiff
path: root/extra/files/rust/check-rustc
blob: d4b85be764db95005adc147d4d5af85f698f79cd (plain)
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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#!/bin/sh
# vim: set ts=4:
set -eu

RUSTC="$1"
TMPDIR="$(pwd)/.tmp-${0##*/}-$RANDOM"
failed=0

unset RUST_BACKTRACE
unset RUSTC_CRT_STATIC


_rustc() {
	printf '\n$ rustc %s\n' "$*"
	"$RUSTC" "$@"
}

die() {
	printf '\033[1;31mERROR:\033[0m %s\n' "$1" >&2  # bold red
	exit 1
}

fail() {
	printf '\033[1;31mFAIL:\033[0m %s\n' "$1" >&2  # bold red
	failed=$(( failed + 1 ))
}

assert_dynamic() {
	readelf -l "$1" | grep -Fqw INTERP \
		&& readelf -d "$1" | grep -Fqw NEEDED || {
			fail "$1 is not a dynamic executable!"
			readelf -ld "$1"
		}
}

assert_ok() {
	"$1" || fail "$1 exited with status $?"
}

assert_panic() {
	local status=0
	"$1" || status=$? && [ "$status" = 101 ] \
		|| fail "$1 exited with status $status, but expected 101"
}

assert_pie() {
	readelf -d "$1" | grep -Fw FLAGS_1 | grep -Fqw PIE || {
		fail "$1 is not a PIE executable!"
		readelf -d "$1"
	}
}

assert_static() {
	test -f "$1" \
		&& ! readelf -l "$1" | grep -Fqw INTERP \
		&& ! readelf -d "$1" | grep -Fqw NEEDED || {
			fail "$1 is not a static executable!"
			readelf -ld "$1"
		}
}


#--------------------  M a i n  --------------------

test -d "$TMPDIR" && die "$TMPDIR already exists!"
mkdir -p "$TMPDIR"
trap "rm -R '$TMPDIR'" EXIT

cd "$TMPDIR"

cat >> hello_world.rs <<-EOF
	fn main() {
	    println!("Hello, world!");
	}
EOF

_rustc hello_world.rs
assert_ok ./hello_world
assert_dynamic hello_world
assert_pie hello_world
rm -f hello_world

_rustc -C target-feature=-crt-static hello_world.rs
assert_ok ./hello_world
assert_dynamic hello_world
assert_pie hello_world
rm -f hello_world

_rustc -C target-feature=+crt-static hello_world.rs
assert_ok ./hello_world
assert_static hello_world
assert_pie hello_world
rm -f hello_world


cat >> panic.rs <<-EOF
	fn main() {
	    panic!("This should panic");
	}
EOF

_rustc -C target-feature=-crt-static panic.rs
assert_panic ./panic

_rustc -C target-feature=+crt-static panic.rs
assert_panic ./panic


[ "$failed" -eq 0 ] || die "$failed assertion(s) has failed"