lld/test/COFF/giats.s

   1 # REQUIRES: x86
   2
   3 # Make a DLL that exports exportfn1.
   4 # RUN: yaml2obj %p/Inputs/export.yaml -o %basename_t-exp.obj
   5 # RUN: lld-link /out:%basename_t-exp.dll /dll %basename_t-exp.obj /export:exportfn1 /implib:%basename_t-exp.lib
   6
   7 # Make an object file that imports exportfn1.
   8 # RUN: llvm-mc -triple x86_64-windows-msvc %s -filetype=obj -o %basename_t.obj
   9
  10 # Check that the Guard address-taken IAT entry tables are propagated to the final executable.
  11 # RUN: lld-link %basename_t.obj -guard:cf -guard:longjmp -entry:main -out:%basename_t-nodelay.exe %basename_t-exp.lib
  12 # RUN: llvm-readobj --file-headers --coff-load-config %basename_t-nodelay.exe | FileCheck %s --check-prefix CHECK
  13
  14 # CHECK: ImageBase: 0x140000000
  15 # CHECK: LoadConfig [
  16 # CHECK:   GuardCFFunctionTable: 0x140002114
  17 # CHECK:   GuardCFFunctionCount: 1
  18 # CHECK:   GuardFlags [ (0x10500)
  19 # CHECK:     CF_FUNCTION_TABLE_PRESENT (0x400)
  20 # CHECK:     CF_INSTRUMENTED (0x100)
  21 # CHECK:     CF_LONGJUMP_TABLE_PRESENT (0x10000)
  22 # CHECK:   ]
  23 # CHECK:   GuardAddressTakenIatEntryTable: 0x140002118
  24 # CHECK:   GuardAddressTakenIatEntryCount: 1
  25 # CHECK: ]
  26 # CHECK:      GuardFidTable [
  27 # CHECK-NEXT:   0x14000{{.*}}
  28 # CHECK-NEXT: ]
  29 # CHECK:      GuardIatTable [
  30 # CHECK-NEXT:   0x14000{{.*}}
  31 # CHECK-NEXT: ]
  32
  33
  34 # Check that the additional load thunk symbol is added to the GFIDs table.
  35 # RUN: lld-link %basename_t.obj -guard:cf -guard:longjmp -entry:main -out:%basename_t-delay.exe %basename_t-exp.lib -alternatename:__delayLoadHelper2=main -delayload:%basename_t-exp.dll
  36 # RUN: llvm-readobj --file-headers --coff-load-config %basename_t-delay.exe | FileCheck %s --check-prefix DELAY-CHECK
  37
  38 # DELAY-CHECK: ImageBase: 0x140000000
  39 # DELAY-CHECK: LoadConfig [
  40 # DELAY-CHECK:   GuardCFFunctionTable: 0x140002124
  41 # DELAY-CHECK:   GuardCFFunctionCount: 2
  42 # DELAY-CHECK:   GuardFlags [ (0x10500)
  43 # DELAY-CHECK:     CF_FUNCTION_TABLE_PRESENT (0x400)
  44 # DELAY-CHECK:     CF_INSTRUMENTED (0x100)
  45 # DELAY-CHECK:     CF_LONGJUMP_TABLE_PRESENT (0x10000)
  46 # DELAY-CHECK:   ]
  47 # DELAY-CHECK:   GuardAddressTakenIatEntryTable: 0x14000212C
  48 # DELAY-CHECK:   GuardAddressTakenIatEntryCount: 1
  49 # DELAY-CHECK: ]
  50 # DELAY-CHECK:      GuardFidTable [
  51 # DELAY-CHECK-NEXT:   0x14000{{.*}}
  52 # DELAY-CHECK-NEXT:   0x14000{{.*}}
  53 # DELAY-CHECK-NEXT: ]
  54 # DELAY-CHECK:      GuardIatTable [
  55 # DELAY-CHECK-NEXT:   0x14000{{.*}}
  56 # DELAY-CHECK-NEXT: ]
  57
  58
  59 # This assembly is reduced from C code like:
  60 # __declspec(noinline)
  61 # void IndirectCall(BOOL (func)(HANDLE)) {
  62 #   (*func)(NULL);
  63 # }
  64 # int main(int argc, char** argv) {
  65 #   IndirectCall(exportfn1);
  66 # }
  67
  68         .text
  69         .def     @feat.00;
  70         .scl    3;
  71         .type   0;
  72         .endef
  73         .globl  @feat.00
  74 .set @feat.00, 2048
  75         .def     IndirectCall;  .scl    2;      .type   32;     .endef
  76         .globl  IndirectCall                    # -- Begin function IndirectCall
  77         .p2align        4, 0x90
  78 IndirectCall:                           # @IndirectCall
  79 # %bb.0:
  80         subq    $40, %rsp
  81         movq    %rcx, 32(%rsp)
  82         movq    32(%rsp), %rax
  83         movq    %rax, %rdx        # This would otherwise be: movq __guard_dispatch_icall_fptr(%rip), %rdx
  84         xorl    %ecx, %ecx
  85         callq   *%rdx
  86         nop
  87         addq    $40, %rsp
  88         retq
  89                                         # -- End function
  90         .def     main;  .scl    2;      .type   32;     .endef
  91         .globl  main                            # -- Begin function main
  92         .p2align        4, 0x90
  93 main:                                   # @main
  94 # %bb.0:
  95         subq    $56, %rsp
  96         movq    __imp_exportfn1(%rip), %rax
  97         movq    %rdx, 48(%rsp)
  98         movl    %ecx, 44(%rsp)
  99         movq    %rax, %rcx
 100         callq   IndirectCall
 101         xorl    %eax, %eax
 102         addq    $56, %rsp
 103         retq
 104                                         # -- End function
 105         .section        .gfids$y,"dr"
 106         .section        .giats$y,"dr"
 107         .symidx __imp_exportfn1
 108         .section        .gljmp$y,"dr"
 109
 110 # Load configuration directory entry (winnt.h _IMAGE_LOAD_CONFIG_DIRECTORY64).
 111 # The linker will define the __guard_* symbols.
 112         .section .rdata,"dr"
 113 .globl _load_config_used
 114 _load_config_used:
 115         .long 256
 116         .fill 124, 1, 0
 117         .quad __guard_fids_table
 118         .quad __guard_fids_count
 119         .long __guard_flags
 120         .fill 12, 1, 0
 121         .quad __guard_iat_table
 122         .quad __guard_iat_count
 123         .quad __guard_longjmp_table
 124         .quad __guard_fids_count
 125         .fill 84, 1, 0