55 changes: 41 additions & 14 deletions llvm/test/CodeGen/X86/seh-catch-all.ll
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,6 @@
; RUN: llc -mtriple=x86_64-windows-msvc < %s | FileCheck %s
; RUN: llc -mtriple=x86_64-windows-msvc < %s | FileCheck %s --check-prefix=X64
; RUN: sed -e 's/__C_specific_handler/_except_handler3/' %s | \
; RUN: llc -mtriple=i686-windows-msvc | FileCheck %s --check-prefix=X86

@str = linkonce_odr unnamed_addr constant [27 x i8] c"GetExceptionCode(): 0x%lx\0A\00", align 1

Expand Down Expand Up @@ -29,16 +31,41 @@ eh.resume:

; Check that we can get the exception code from eax to the printf.

; CHECK-LABEL: main:
; CHECK: retq
; CHECK: # Block address taken
; CHECK: leaq str(%rip), %rcx
; CHECK: movl %eax, %edx
; CHECK: callq printf

; CHECK: .seh_handlerdata
; CHECK-NEXT: .long 1
; CHECK-NEXT: .Ltmp{{[0-9]+}}@IMGREL
; CHECK-NEXT: .Ltmp{{[0-9]+}}@IMGREL+1
; CHECK-NEXT: 1
; CHECK-NEXT: .Ltmp{{[0-9]+}}@IMGREL
; X64-LABEL: main:
; X64: callq crash
; X64: retq
; X64: # Block address taken
; X64: leaq str(%rip), %rcx
; X64: movl %eax, %edx
; X64: callq printf

; X64: .seh_handlerdata
; X64-NEXT: .long 1
; X64-NEXT: .long .Ltmp{{[0-9]+}}@IMGREL
; X64-NEXT: .long .Ltmp{{[0-9]+}}@IMGREL+1
; X64-NEXT: .long 1
; X64-NEXT: .long .Ltmp{{[0-9]+}}@IMGREL

; X86-LABEL: _main:
; The EH code load should be this offset +4.
; X86: movl %esp, -24(%ebp)
; X86: movl $L__ehtable$main,
; EH state 0
; X86: movl $0, -4(%ebp)
; X86: calll _crash
; X86: retl
; X86: # Block address taken
; X86: movl -20(%ebp), %[[ptrs:[^ ,]*]]
; X86: movl (%[[ptrs]]), %[[rec:[^ ,]*]]
; X86: movl (%[[rec]]), %[[code:[^ ,]*]]
; EH state -1
; X86: movl $-1, -4(%ebp)
; X86-DAG: movl %[[code]], 4(%esp)
; X86-DAG: movl $_str, (%esp)
; X86: calll _printf

; X86: .section .xdata,"dr"
; X86-NEXT: L__ehtable$main
; X86-NEXT: .long -1
; X86-NEXT: .long 1
; X86-NEXT: .long Ltmp{{[0-9]+}}
168 changes: 168 additions & 0 deletions llvm/test/CodeGen/X86/seh-safe-div-win32.ll
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,168 @@
; RUN: llc -mtriple i686-pc-windows-msvc < %s | FileCheck %s

; This test case is also intended to be run manually as a complete functional
; test. It should link, print something, and exit zero rather than crashing.
; It is the hypothetical lowering of a C source program that looks like:
;
; int safe_div(int *n, int *d) {
; int r;
; __try {
; __try {
; r = *n / *d;
; } __except(GetExceptionCode() == EXCEPTION_ACCESS_VIOLATION) {
; puts("EXCEPTION_ACCESS_VIOLATION");
; r = -1;
; }
; } __except(GetExceptionCode() == EXCEPTION_INT_DIVIDE_BY_ZERO) {
; puts("EXCEPTION_INT_DIVIDE_BY_ZERO");
; r = -2;
; }
; return r;
; }

@str1 = internal constant [27 x i8] c"EXCEPTION_ACCESS_VIOLATION\00"
@str2 = internal constant [29 x i8] c"EXCEPTION_INT_DIVIDE_BY_ZERO\00"

define i32 @safe_div(i32* %n, i32* %d) {
entry:
%r = alloca i32, align 4
store i32 42, i32* %r
invoke void @try_body(i32* %r, i32* %n, i32* %d)
to label %__try.cont unwind label %lpad

lpad:
%vals = landingpad { i8*, i32 } personality i8* bitcast (i32 (...)* @_except_handler3 to i8*)
catch i8* bitcast (i32 ()* @safe_div_filt0 to i8*)
catch i8* bitcast (i32 ()* @safe_div_filt1 to i8*)
%ehptr = extractvalue { i8*, i32 } %vals, 0
%sel = extractvalue { i8*, i32 } %vals, 1
%filt0_val = call i32 @llvm.eh.typeid.for(i8* bitcast (i32 ()* @safe_div_filt0 to i8*))
%is_filt0 = icmp eq i32 %sel, %filt0_val
br i1 %is_filt0, label %handler0, label %eh.dispatch1

eh.dispatch1:
%filt1_val = call i32 @llvm.eh.typeid.for(i8* bitcast (i32 ()* @safe_div_filt1 to i8*))
%is_filt1 = icmp eq i32 %sel, %filt1_val
br i1 %is_filt1, label %handler1, label %eh.resume

handler0:
call void @puts(i8* getelementptr ([27 x i8], [27 x i8]* @str1, i32 0, i32 0))
store i32 -1, i32* %r, align 4
br label %__try.cont

handler1:
call void @puts(i8* getelementptr ([29 x i8], [29 x i8]* @str2, i32 0, i32 0))
store i32 -2, i32* %r, align 4
br label %__try.cont

eh.resume:
resume { i8*, i32 } %vals

__try.cont:
%safe_ret = load i32, i32* %r, align 4
ret i32 %safe_ret
}

; Normal path code

; CHECK: {{^}}_safe_div:
; CHECK: movl $42, [[rloc:.*\(%ebp\)]]
; CHECK: leal [[rloc]],
; CHECK: calll _try_body
; CHECK: [[cont_bb:LBB0_[0-9]+]]:
; CHECK: movl [[rloc]], %eax
; CHECK: retl

; Landing pad code

; CHECK: [[handler0:Ltmp[0-9]+]]: # Block address taken
; CHECK: # %handler0
; CHECK: calll _puts
; CHECK: jmp [[cont_bb]]

; CHECK: [[handler1:Ltmp[0-9]+]]: # Block address taken
; CHECK: # %handler1
; CHECK: calll _puts
; CHECK: jmp [[cont_bb]]

; CHECK: .section .xdata,"dr"
; CHECK-NEXT: L__ehtable$safe_div:
; CHECK-NEXT: .long -1
; CHECK-NEXT: .long _safe_div_filt0
; CHECK-NEXT: .long [[handler0]]
; CHECK-NEXT: .long 0
; CHECK-NEXT: .long _safe_div_filt1
; CHECK-NEXT: .long [[handler1]]

define void @try_body(i32* %r, i32* %n, i32* %d) {
entry:
%0 = load i32, i32* %n, align 4
%1 = load i32, i32* %d, align 4
%div = sdiv i32 %0, %1
store i32 %div, i32* %r, align 4
ret void
}

; The prototype of these filter functions is:
; int filter(EXCEPTION_POINTERS *eh_ptrs, void *rbp);

; The definition of EXCEPTION_POINTERS is:
; typedef struct _EXCEPTION_POINTERS {
; EXCEPTION_RECORD *ExceptionRecord;
; CONTEXT *ContextRecord;
; } EXCEPTION_POINTERS;

; The definition of EXCEPTION_RECORD is:
; typedef struct _EXCEPTION_RECORD {
; DWORD ExceptionCode;
; ...
; } EXCEPTION_RECORD;

; FIXME: Use llvm.eh.exceptioninfo for this.
declare i32 @safe_div_filt0()
declare i32 @safe_div_filt1()
; define i32 @safe_div_filt0() {
; %eh_ptrs_c = bitcast i8* %eh_ptrs to i32**
; %eh_rec = load i32*, i32** %eh_ptrs_c
; %eh_code = load i32, i32* %eh_rec
; ; EXCEPTION_ACCESS_VIOLATION = 0xC0000005
; %cmp = icmp eq i32 %eh_code, 3221225477
; %filt.res = zext i1 %cmp to i32
; ret i32 %filt.res
; }
; define i32 @safe_div_filt1() {
; %eh_ptrs_c = bitcast i8* %eh_ptrs to i32**
; %eh_rec = load i32*, i32** %eh_ptrs_c
; %eh_code = load i32, i32* %eh_rec
; ; EXCEPTION_INT_DIVIDE_BY_ZERO = 0xC0000094
; %cmp = icmp eq i32 %eh_code, 3221225620
; %filt.res = zext i1 %cmp to i32
; ret i32 %filt.res
; }

@str_result = internal constant [21 x i8] c"safe_div result: %d\0A\00"

define i32 @main() {
%d.addr = alloca i32, align 4
%n.addr = alloca i32, align 4

store i32 10, i32* %n.addr, align 4
store i32 2, i32* %d.addr, align 4
%r1 = call i32 @safe_div(i32* %n.addr, i32* %d.addr)
call void (i8*, ...) @printf(i8* getelementptr ([21 x i8], [21 x i8]* @str_result, i32 0, i32 0), i32 %r1)

store i32 10, i32* %n.addr, align 4
store i32 0, i32* %d.addr, align 4
%r2 = call i32 @safe_div(i32* %n.addr, i32* %d.addr)
call void (i8*, ...) @printf(i8* getelementptr ([21 x i8], [21 x i8]* @str_result, i32 0, i32 0), i32 %r2)

%r3 = call i32 @safe_div(i32* %n.addr, i32* null)
call void (i8*, ...) @printf(i8* getelementptr ([21 x i8], [21 x i8]* @str_result, i32 0, i32 0), i32 %r3)
ret i32 0
}

declare i32 @_except_handler3(...)
declare i32 @llvm.eh.typeid.for(i8*) readnone nounwind
declare void @puts(i8*)
declare void @printf(i8*, ...)
declare void @abort()
16 changes: 16 additions & 0 deletions llvm/test/CodeGen/X86/win32-eh.ll
View file
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,12 @@ catchall:
; CHECK: movl %[[next]], %fs:0
; CHECK: retl

; CHECK: .section .xdata,"dr"
; CHECK-LABEL: L__ehtable$use_except_handler3:
; CHECK-NEXT: .long -1
; CHECK-NEXT: .long 1
; CHECK-NEXT: .long Ltmp{{[0-9]+}}

define void @use_except_handler4() {
invoke void @may_throw_or_crash()
to label %cont unwind label %catchall
Expand Down Expand Up @@ -64,6 +70,16 @@ catchall:
; CHECK: movl %[[next]], %fs:0
; CHECK: retl

; CHECK: .section .xdata,"dr"
; CHECK-LABEL: L__ehtable$use_except_handler4:
; CHECK-NEXT: .long -2
; CHECK-NEXT: .long 0
; CHECK-NEXT: .long 9999
; CHECK-NEXT: .long 0
; CHECK-NEXT: .long -2
; CHECK-NEXT: .long 1
; CHECK-NEXT: .long Ltmp{{[0-9]+}}

define void @use_CxxFrameHandler3() {
invoke void @may_throw_or_crash()
to label %cont unwind label %catchall
Expand Down