RPCS3/llvm-mirror
1
0
Fork 0
mirror of https://github.com/RPCS3/llvm-mirror.git synced 2024-10-18 18:42:46 +02:00
Code Issues Projects Releases Wiki Activity

Include invoke'd functions for recursive extract

Browse Source 
Summary: When recursively extracting a function from a bit code file, include functions mentioned in InvokeInst as well as CallInst

Reviewers: loladiro, espindola, volkan

Reviewed By: loladiro

Subscribers: llvm-commits

Tags: #llvm

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

llvm-svn: 357735
This commit is contained in:
David Callahan 2019-04-04 23:30:47 +00:00
parent 3deb5ec521
commit 190dbc4beb
2 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
test/tools/llvm-extract/recursive.ll
View File

@ -1,6 +1,7 @@
; RUN: llvm-extract -func=a --recursive %s -S | FileCheck --check-prefix=CHECK-AB %s
; RUN: llvm-extract -func=a --recursive --delete %s -S | FileCheck --check-prefix=CHECK-CD %s
; RUN: llvm-extract -func=d --recursive %s -S | FileCheck --check-prefix=CHECK-CD %s
; RUN: llvm-extract -func=e --recursive %s -S | FileCheck --check-prefix=CHECK-CD %s
; CHECK-AB: define void @a
; CHECK-AB: define void @b
@ -30,3 +31,10 @@ define void @d() {
call void @c()
ret void
}
define void @e() {
invoke void @c()
to label %L unwind label %L
L:
ret void
}

6
tools/llvm-extract/llvm-extract.cpp
View File

@ -270,10 +270,10 @@ int main(int argc, char **argv) {
ExitOnErr(F->materialize());
for (auto &BB : *F) {
for (auto &I : BB) {
auto *CI = dyn_cast<CallInst>(&I);
if (!CI)
CallBase *CB = dyn_cast<CallBase>(&I);
if (!CB)
continue;
Function *CF = CI->getCalledFunction();
Function *CF = CB->getCalledFunction();
if (!CF)
continue;
if (CF->isDeclaration() || GVs.count(CF))