1 files changed, 53 insertions, 0 deletions
diff --git a/lld/test/ELF/linkerscript/linkerscript-sizeof.s b/lld/test/ELF/linkerscript/linkerscript-sizeof.s
new file mode 100644
index 00000000000..f7420ddfe64
--- /dev/null
+++ b/lld/test/ELF/linkerscript/linkerscript-sizeof.s
@@ -0,0 +1,53 @@
+# REQUIRES: x86
+# RUN: llvm-mc -filetype=obj -triple=x86_64-unknown-linux %s -o %t
+
+# RUN: echo "SECTIONS { \
+# RUN:   .aaa         : { *(.aaa) } \
+# RUN:   .bbb         : { *(.bbb) } \
+# RUN:   .ccc         : { *(.ccc) } \
+# RUN:   _aaa = SIZEOF(.aaa); \
+# RUN:   _bbb = SIZEOF(.bbb); \
+# RUN:   _ccc = SIZEOF(.ccc); \
+# RUN: }" > %t.script
+# RUN: ld.lld -o %t1 --script %t.script %t
+# RUN: llvm-objdump -t -section-headers %t1 | FileCheck %s
+# CHECK:      Sections:
+# CHECK-NEXT:  Idx Name          Size      Address          Type
+# CHECK-NEXT:    0               00000000 0000000000000000
+# CHECK-NEXT:    1 .aaa          00000008 0000000000000120 DATA
+# CHECK-NEXT:    2 .bbb          00000010 0000000000000128 DATA
+# CHECK-NEXT:    3 .ccc          00000018 0000000000000138 DATA
+# CHECK:      SYMBOL TABLE:
+# CHECK-NEXT:  0000000000000000 *UND* 00000000
+# CHECK-NEXT:  0000000000000150 .text 00000000 _start
+# CHECK-NEXT:  0000000000000008 *ABS* 00000000 _aaa
+# CHECK-NEXT:  0000000000000010 *ABS* 00000000 _bbb
+# CHECK-NEXT:  0000000000000018 *ABS* 00000000 _ccc
+
+## Check that we error out if trying to get size of
+## section that does not exist.
+# RUN: echo "SECTIONS { \
+# RUN:   .aaa         : { *(.aaa) } \
+# RUN:   .bbb         : { *(.bbb) } \
+# RUN:   .ccc         : { *(.ccc) } \
+# RUN:   _aaa = SIZEOF(.foo); \
+# RUN: }" > %t.script
+# RUN: not ld.lld -o %t1 --script %t.script %t 2>&1 \
+# RUN:  | FileCheck -check-prefix=ERR %s
+# ERR: undefined section .foo
+
+.global _start
+_start:
+ nop
+
+.section .aaa,"a"
+ .quad 0
+
+.section .bbb,"a"
+ .quad 0
+ .quad 0
+
+.section .ccc,"a"
+ .quad 0
+ .quad 0
+ .quad 0