Handle branch relative to pc in ARM integrated assembler.

tests_lit/assembler/arm32/branch-mult-fwd.ll

Index: tests_lit/assembler/arm32/branch-mult-fwd.ll
diff --git a/tests_lit/assembler/arm32/branch-mult-fwd.ll b/tests_lit/assembler/arm32/branch-mult-fwd.ll
new file mode 100644
index 0000000000000000000000000000000000000000..13a931556ec84269752b68075eff4a1460ab4668
--- /dev/null
+++ b/tests_lit/assembler/arm32/branch-mult-fwd.ll
@@ -0,0 +1,168 @@
+; Test that we correctly fix multiple forward branches.
+
+; Compile using standalone assembler.
+; RUN: %p2i --filetype=asm -i %s --target=arm32 --args -Om1 \
+; RUN:   | FileCheck %s --check-prefix=ASM
+
+; Show bytes in assembled standalone code.
+; RUN: %p2i --filetype=asm -i %s --target=arm32 --assemble --disassemble \
+; RUN:   --args -Om1 | FileCheck %s --check-prefix=DIS
+
+; Compile using integrated assembler.
+; RUN: %p2i --filetype=iasm -i %s --target=arm32 --args -Om1 \
+; RUN:   | FileCheck %s --check-prefix=IASM
+
+; Show bytes in assembled integrated code.
+; RUN: %p2i --filetype=iasm -i %s --target=arm32 --assemble --disassemble \
+; RUN:   --args -Om1 | FileCheck %s --check-prefix=DIS
+
+; REQUIRES: allow_dump
+
+define internal void @mult_fwd_branches(i32 %a, i32 %b) {
+; ASM-LABEL:mult_fwd_branches:
+; ASM-LABEL:.Lmult_fwd_branches$__0:
+
+; ASM-NEXT:	sub	sp, sp, #12
+; ASM-NEXT:	str	r0, [sp, #8]
+; ASM-NEXT:	str	r1, [sp, #4]
+
+; DIS-LABEL:00000000 <mult_fwd_branches>:
+
+; DIS-NEXT:   0:	e24dd00c
+; DIS-NEXT:   4:	e58d0008
+; DIS-NEXT:   8:	e58d1004
+
+; IASM-LABEL:mult_fwd_branches:
+; IASM-LABEL:.Lmult_fwd_branches$__0:
+
+; IASM-NEXT:	.byte 0xc
+; IASM-NEXT:	.byte 0xd0
+; IASM-NEXT:	.byte 0x4d
+; IASM-NEXT:	.byte 0xe2
+
+; IASM-NEXT:	.byte 0x8
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x8d
+; IASM-NEXT:	.byte 0xe5
+
+; IASM-NEXT:	.byte 0x4
+; IASM-NEXT:	.byte 0x10
+; IASM-NEXT:	.byte 0x8d
+; IASM-NEXT:	.byte 0xe5
+
+  %cmp = icmp slt i32 %a, %b
+
+; ASM-NEXT:	ldr	r0, [sp, #8]
+; ASM-NEXT:	mov	r1, #0
+; ASM-NEXT:	ldr	r2, [sp, #4]
+; ASM-NEXT:	cmp	r0, r2
+; ASM-NEXT:	movlt	r1, #1
+; ASM-NEXT:	str	r1, [sp]
+
+; DIS-NEXT:   c:	e59d0008
+; DIS-NEXT:  10:	e3a01000
+; DIS-NEXT:  14:	e59d2004
+; DIS-NEXT:  18:	e1500002
+; DIS-NEXT:  1c:	b3a01001
+; DIS-NEXT:  20:	e58d1000
+
+; IASM-NEXT:	.byte 0x8
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x9d
+; IASM-NEXT:	.byte 0xe5
+
+; IASM-NEXT:	mov	r1, #0
+
+; IASM-NEXT:	.byte 0x4
+; IASM-NEXT:	.byte 0x20
+; IASM-NEXT:	.byte 0x9d
+; IASM-NEXT:	.byte 0xe5
+
+; IASM-NEXT:	cmp	r0, r2
+; IASM-NEXT:	movlt	r1, #1
+
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x10
+; IASM-NEXT:	.byte 0x8d
+; IASM-NEXT:	.byte 0xe5
+
+  br i1 %cmp, label %then, label %else
+
+; ASM-NEXT:	ldr	r0, [sp]
+; ASM-NEXT:	uxtb	r0, r0
+; ASM-NEXT:	cmp	r0, #0
+; ASM-NEXT:	bne	.Lmult_fwd_branches$then
+; ASM-NEXT:	b	.Lmult_fwd_branches$else
+
+; DIS-NEXT:  24:	e59d0000
+; DIS-NEXT:  28:	e6ef0070
+; DIS-NEXT:  2c:	e3500000
+; DIS-NEXT:  30:	1a000000
+; DIS-NEXT:  34:	ea000000
+
+; IASM-NEXT:	ldr	r0, [sp]
+; IASM-NEXT:	uxtb	r0, r0
+; IASM-NEXT:	cmp	r0, #0
+
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x1a
+
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0xea
+
+then:
+; ASM-LABEL:.Lmult_fwd_branches$then:
+; IASM-LABEL:.Lmult_fwd_branches$then:
+
+  br label %end
+; ASM-NEXT:	b	.Lmult_fwd_branches$end
+
+; DIS-NEXT:  38:	ea000000
+
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0x0
+; IASM-NEXT:	.byte 0xea
+
+else:
+; ASM-LABEL:.Lmult_fwd_branches$else:
+; IASM-LABEL:.Lmult_fwd_branches$else:
+
+  br label %end
+; ASM-NEXT:	b	.Lmult_fwd_branches$end
+
+; DIS-NEXT:  3c:	eaffffff
+
+; IASM-NEXT:	.byte 0xff
+; IASM-NEXT:	.byte 0xff
+; IASM-NEXT:	.byte 0xff
+; IASM-NEXT:	.byte 0xea
+
+
+end:
+; ASM-LABEL:.Lmult_fwd_branches$end:
+; IASM-LABEL: .Lmult_fwd_branches$end:
+
+  ret void
+
+; ASM-NEXT:	add	sp, sp, #12
+; ASM-NEXT:	bx	lr
+
+; DIS-NEXT:  40:	e28dd00c
+; DIS-NEXT:  44:	e12fff1e
+
+; IASM-NEXT:	.byte 0xc
+; IASM-NEXT:	.byte 0xd0
+; IASM-NEXT:	.byte 0x8d
+; IASM-NEXT:	.byte 0xe2
+
+; IASM-NEXT:	.byte 0x1e
+; IASM-NEXT:	.byte 0xff
+; IASM-NEXT:	.byte 0x2f
+; IASM-NEXT:	.byte 0xe1
+
+}