aboutsummaryrefslogtreecommitdiffstats
path: root/testing/rust/check-rustc
blob: 015730dc0eec967962fc972470a2b650a46fafea (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
#!/bin/sh
set -eu

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


_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 ))
}

# Checks if the given file is a statically linked binary.
is_static() {
	! readelf -l "$1" | grep -Fq INTERP \
		&& ! readelf -d "$1" | grep -Fq NEEDED
}

assert_dynamic() {
	test -f "$1" && ! is_static "$1" || {
		fail "$1 is not a dynamic executable!"
		readelf -ld "$1"
	}
}

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

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

assert_static() {
	test -f "$1" && is_static "$1" || {
		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_static 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

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