Include invoke'd functions for recursive extract

[david2050]

Summary

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

[loladiro]

Inline comment on implementation. The functionality looks fine to me.

tools/llvm-extract/llvm-extract.cpp

274

Just use CallBase here instead of enumerating the subclasses thereof?
Also the formatting, looks slightly off.

[david2050]

use CallBase

[loladiro]

tools/llvm-extract/llvm-extract.cpp

273–277

nit: The Function * can just remain here.

[david2050]

cleanup

test/tools/llvm-extract/recursive.ll

 ; 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
 ; CHECK-AB-NOT: define void @c
 ; CHECK-AB-NOT: define void @d
 
 ; CHECK-CD-NOT: define void @a
 ; CHECK-CD-NOT: define void @b
define void @c() {
   call void @d()
   ret void
 }
 
 define void @d() {
   call void @c()
   ret void
 }
+
+define void @e() {
+  invoke void @c()
+  to label %L unwind label %L
+L:
+  ret void
+}
 No newline at end of file

tools/llvm-extract/llvm-extract.cpp

for (GlobalValue *GV : GVs) {
       }
     }
     while (!Workqueue.empty()) {
       Function *F = &*Workqueue.back();
       Workqueue.pop_back();
       ExitOnErr(F->materialize());
       for (auto &BB : *F) {
         for (auto &I : BB) {
-          auto *CI = dyn_cast<CallInst>(&I);
-          if (!CI)
-            continue;
-          Function *CF = CI->getCalledFunction();
+          Function *CF = nullptr;
+          if(auto *CI = dyn_cast<CallInst>(&I))
+            CF = CI->getCalledFunction();
+          else if(auto *II = dyn_cast<InvokeInst>(&I))
+            CF = II->getCalledFunction();
           if (!CF)
             continue;
           if (CF->isDeclaration() || GVs.count(CF))
             continue;
           GVs.insert(CF);
           Workqueue.push_back(CF);
         }
       }