Sam Clegg 492f752969 [WebAssembly] Initial implementation of PIC code generation
This change implements lowering of references global symbols in PIC
mode.

This change implements lowering of global references in PIC mode using a
new @GOT reference type. @GOT references can be used with function or
data symbol names combined with the get_global instruction. In this case
the linker will insert the wasm global that stores the address of the
symbol (either in memory for data symbols or in the wasm table for
function symbols).

For now I'm continuing to use the R_WASM_GLOBAL_INDEX_LEB relocation
type for this type of reference which means that this relocation type
can refer to either a global or a function or data symbol. We could
choose to introduce specific relocation types for GOT entries in the
future.  See the current dynamic linking proposal:

https://github.com/WebAssembly/tool-conventions/blob/master/DynamicLinking.md

Differential Revision: https://reviews.llvm.org/D54647

llvm-svn: 357022
2019-03-26 19:46:15 +00:00

154 lines
5.7 KiB
LLVM

; RUN: llc < %s -asm-verbose=false -wasm-disable-explicit-locals -wasm-keep-registers | FileCheck %s -check-prefixes=NON-PIC,CHECK
; RUN: llc < %s -asm-verbose=false -relocation-model=pic -fast-isel -wasm-disable-explicit-locals -wasm-keep-registers | FileCheck %s -check-prefixes=PIC,CHECK
; RUN: llc < %s -asm-verbose=false -relocation-model=pic -fast-isel=false -wasm-disable-explicit-locals -wasm-keep-registers | FileCheck %s -check-prefixes=PIC,CHECK
; Test that globals assemble as expected with -fPIC.
; We test here both with and without fast-isel.
target datalayout = "e-m:e-p:32:32-i64:64-n32:64-S128"
target triple = "wasm32-unknown-unknown"
@hidden_global = external hidden global i32
@hidden_global_array = external hidden global [10 x i32]
@external_global = external global i32
@external_global_array = external global [10 x i32]
declare i32 @foo();
; For hidden symbols PIC code needs to offset all loads and stores
; by the value of the __memory_base global
define i32 @load_hidden_global() {
; CHECK-LABEL: load_hidden_global:
; PIC: global.get $push[[L0:[0-9]+]]=, __memory_base{{$}}
; PIC-NEXT: i32.const $push[[L1:[0-9]+]]=, hidden_global{{$}}
; PIC-NEXT: i32.add $push[[L2:[0-9]+]]=, $pop[[L0]], $pop[[L1]]{{$}}
; PIC-NEXT: i32.load $push[[L3:[0-9]+]]=, 0($pop[[L2]]){{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.load $push1=, hidden_global($pop0){{$}}
; CHECK-NEXT: end_function
%1 = load i32, i32* @hidden_global
ret i32 %1
}
define i32 @load_hidden_global_offset() {
; CHECK-LABEL: load_hidden_global_offset:
; PIC: global.get $push[[L0:[0-9]+]]=, __memory_base{{$}}
; PIC-NEXT: i32.const $push[[L1:[0-9]+]]=, hidden_global_array{{$}}
; PIC-NEXT: i32.add $push[[L2:[0-9]+]]=, $pop[[L0]], $pop[[L1:[0-9]+]]{{$}}
; PIC-NEXT: i32.const $push[[L3:[0-9]+]]=, 20{{$}}
; PIC-NEXT: i32.add $push[[L4:[0-9]+]]=, $pop[[L2]], $pop[[L3]]{{$}}
; PIC-NEXT: i32.load $push{{[0-9]+}}=, 0($pop[[L4]]){{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT:i32.load $push1=, hidden_global_array+20($pop0){{$}}
; CHECK-NEXT: end_function
%1 = getelementptr [10 x i32], [10 x i32]* @hidden_global_array, i32 0, i32 5
%2 = load i32, i32* %1
ret i32 %2
}
; Store to a hidden global
define void @store_hidden_global(i32 %n) {
; CHECK-LABEL: store_hidden_global:
; PIC: global.get $push[[L0:[0-9]+]]=, __memory_base{{$}}
; PIC-NEXT: i32.const $push[[L1:[0-9]+]]=, hidden_global{{$}}
; PIC-NEXT: i32.add $push[[L2:[0-9]+]]=, $pop[[L0]], $pop[[L1]]{{$}}
; PIC-NEXT: i32.store 0($pop[[L2]]), $0{{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.store hidden_global($pop0), $0{{$}}
; CHECK-NEXT: end_function
store i32 %n, i32* @hidden_global
ret void
}
define void @store_hidden_global_offset(i32 %n) {
; CHECK-LABEL: store_hidden_global_offset:
; PIC: global.get $push[[L0:[0-9]+]]=, __memory_base{{$}}
; PIC-NEXT: i32.const $push[[L1:[0-9]+]]=, hidden_global_array{{$}}
; PIC-NEXT: i32.add $push[[L2:[0-9]+]]=, $pop[[L0]], $pop[[L1]]{{$}}
; PIC-NEXT: i32.const $push[[L3:[0-9]+]]=, 20{{$}}
; PIC-NEXT: i32.add $push[[L4:[0-9]+]]=, $pop[[L2]], $pop[[L3]]{{$}}
; PIC-NEXT: i32.store 0($pop[[L4]]), $0{{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.store hidden_global_array+20($pop0), $0{{$}}
; CHECK-NEXT: end_function
%1 = getelementptr [10 x i32], [10 x i32]* @hidden_global_array, i32 0, i32 5
store i32 %n, i32* %1
ret void
}
; For non-hidden globals PIC code has to load the address from a wasm global
; using the @GOT relocation type.
define i32 @load_external_global() {
; CHECK-LABEL: load_external_global:
; PIC: global.get $push[[L0:[0-9]+]]=, external_global@GOT{{$}}
; PIC-NEXT: i32.load $push{{[0-9]+}}=, 0($pop[[L0]]){{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.load $push1=, external_global($pop0){{$}}
; CHECK-NEXT: end_function
%1 = load i32, i32* @external_global
ret i32 %1
}
define i32 @load_external_global_offset() {
; CHECK-LABEL: load_external_global_offset:
; PIC: global.get $push[[L0:[0-9]+]]=, external_global_array@GOT{{$}}
; PIC-NEXT: i32.const $push[[L1:[0-9]+]]=, 20{{$}}
; PIC-NEXT: i32.add $push[[L2:[0-9]+]]=, $pop[[L0]], $pop[[L1]]{{$}}
; PIC-NEXT: i32.load $push{{[0-9]+}}=, 0($pop[[L2]]){{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.load $push1=, external_global_array+20($pop0){{$}}
; CHECK-NEXT: end_function
%1 = getelementptr [10 x i32], [10 x i32]* @external_global_array, i32 0, i32 5
%2 = load i32, i32* %1
ret i32 %2
}
; Store to a non-hidden global via the wasm global.
define void @store_external_global(i32 %n) {
; CHECK-LABEL: store_external_global:
; PIC: global.get $push[[L0:[0-9]+]]=, external_global@GOT{{$}}
; PIC-NEXT: i32.store 0($pop[[L0]]), $0{{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.store external_global($pop0), $0{{$}}
; CHECK-NEXT: end_function
store i32 %n, i32* @external_global
ret void
}
define void @store_external_global_offset(i32 %n) {
; CHECK-LABEL: store_external_global_offset:
; PIC: global.get $push[[L0:[0-9]+]]=, external_global_array@GOT{{$}}
; PIC-NEXT: i32.const $push[[L1:[0-9]+]]=, 20{{$}}
; PIC-NEXT: i32.add $push[[L2:[0-9]+]]=, $pop[[L0]], $pop[[L1]]{{$}}
; PIC-NEXT: i32.store 0($pop[[L2]]), $0{{$}}
; NON-PIC: i32.const $push0=, 0{{$}}
; NON-PIC-NEXT: i32.store external_global_array+20($pop0), $0{{$}}
; CHECK-NEXT: end_function
%1 = getelementptr [10 x i32], [10 x i32]* @external_global_array, i32 0, i32 5
store i32 %n, i32* %1
ret void
}
; PIC: .globaltype __memory_base, i32