mold/test/elf/reloc.sh

#!/bin/bash
export LANG=
set -e
testname=$(basename -s .sh "$0")
echo -n "Testing $testname ... "
cd "$(dirname "$0")"/../..
mold="$(pwd)/mold"
t="$(pwd)/out/test/elf/$testname"
mkdir -p "$t"

[ "$(uname -m)" = x86_64 ] || { echo skipped; exit; }

cat <<'EOF' | cc -fPIC -c -o "$t"/a.o -x assembler -
.data
.globl ext_var
.type ext_var, @object
.size ext_var, 56
ext_var:
  .long 56
EOF

cat <<'EOF' | cc -fPIC -c -o "$t"/b.o -xc -
#include <stdio.h>

int print(int x) {
  printf("%d\n", x);
  return 0;
}

int print64(long x) {
  printf("%ld\n", x);
  return 0;
}
EOF

cc -shared -o "$t"/c.so "$t"/a.o "$t"/b.o

# Absolute symbol
cat <<'EOF' > "$t"/d.s
.globl abs_sym
.set abs_sym, 42

.globl main
main:
  sub $8, %rsp
  lea abs_sym, %edi
  call print
  add $8, %rsp
  ret
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -no-pie
"$t"/exe | grep -q '^42$'
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -pie
"$t"/exe | grep -q '^42$'

# GOT
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  mov ext_var@GOTPCREL(%rip), %rdi
  mov (%rdi), %edi
  call print
  add $8, %rsp
  ret
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -no-pie
"$t"/exe | grep -q '^56$'
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -pie
"$t"/exe | grep -q '^56$'

# Copyrel
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  mov ext_var(%rip), %edi
  call print
  add $8, %rsp
  ret
EOF

clang -c -o "$t"/d.o "$t"/d.s
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.o -no-pie
"$t"/exe | grep -q '^56$'
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -pie
"$t"/exe | grep -q '^56$'

# Copyrel
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  mov foo(%rip), %rdi
  mov (%rdi), %edi
  call print
  add $8, %rsp
  ret

.data
foo:
  .quad ext_var
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -no-pie
"$t"/exe | grep -q '^56$'
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -pie
"$t"/exe | grep -q '^56$'

# PLT
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  mov $76, %edi
  call print@PLT
  add $8, %rsp
  ret
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -no-pie
"$t"/exe | grep -q '^76$'
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -pie
"$t"/exe | grep -q '^76$'

# PLT
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  mov $76, %edi
  lea print(%rip), %rax
  call *%rax
  add $8, %rsp
  ret
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -no-pie
"$t"/exe | grep -q '^76$'
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s -pie
"$t"/exe | grep -q '^76$'

# SIZE32
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  mov $foo+2@SIZE, %edi
  call print@PLT
  add $8, %rsp
  ret

.data
.globl foo
.type foo, %object
.size foo, 24
foo:
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s
"$t"/exe | grep -q '^26$'

# SIZE64
cat <<'EOF' > "$t"/d.s
.globl main
main:
  sub $8, %rsp
  movabs $foo+5@SIZE, %rdi
  call print64@PLT
  add $8, %rsp
  ret

.data
.globl foo
.type foo, %object
.size foo, 56
foo:
EOF

clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/d.s
"$t"/exe | grep -q '^61$'

# GOTPCREL64
cat <<'EOF' > "$t"/e.c
extern long ext_var;
void print64(long);

int main() {
  print64(ext_var);
}
EOF

clang -c -o "$t"/e.o "$t"/e.c -mcmodel=large -fPIC
clang -fuse-ld="$mold" -o "$t"/exe "$t"/c.so "$t"/e.o
"$t"/exe | grep -q '^56$'

# R_X86_64_32 against non-alloc section
cat <<'EOF' > "$t"/f.s
.globl main
main:
  sub $8, %rsp
  add $8, %rsp
  ret

.section .foo, "", @progbits
.zero 16
foo:
.quad bar

.section .bar, "", @progbits
.zero 24
bar:
.quad foo
EOF

clang -c -o "$t"/f.o "$t"/f.s
clang -fuse-ld="$mold" -o "$t"/exe "$t"/f.o
readelf -x .foo -x .bar "$t"/exe > "$t"/log

fgrep -q '0x00000010 00000000 00000000 10000000 00000000' "$t"/log
fgrep -q '0x00000010 18000000 00000000' "$t"/log

echo OK