Model terminator instructions

Author: CodaFi

Sources/LLVM/BasicBlock.swift

@@ -40,6 +40,13 @@ public struct BasicBlock: IRValue {
     return Instruction(llvm: val)
   }
 
+  /// Returns the terminator instruction if this basic block is well formed or 
+  /// `nil` if it is not well formed.
+  public var terminator: TerminatorInstruction? {
+    guard let term = LLVMGetBasicBlockTerminator(llvm) else { return nil }
+    return TerminatorInstruction(llvm: term)
+  }
+
   /// Returns the parent function of this basic block, if it exists.
   public var parent: Function? {
     guard let functionRef = LLVMGetBasicBlockParent(llvm) else { return nil }

Sources/LLVM/Instruction.swift

@@ -51,3 +51,30 @@ public struct Instruction: IRValue {
     }
   }
 }
+
+/// A `TerminatorInstruction` represents an instruction that terminates a 
+/// basic block.
+public struct TerminatorInstruction {
+  internal let llvm: LLVMValueRef
+
+  /// Creates a `TerminatorInstruction` from an `LLVMValueRef` object.
+  public init(llvm: LLVMValueRef) {
+    self.llvm = llvm
+  }
+
+  /// Retrieves the number of successors of this terminator instruction.
+  public var successorCount: Int {
+    return Int(LLVMGetNumSuccessors(llvm))
+  }
+
+  /// Returns the successor block at the specified index, if it exists.
+  public func getSuccessor(at idx: Int) -> BasicBlock? {
+    guard let succ = LLVMGetSuccessor(llvm, UInt32(idx)) else { return nil }
+    return BasicBlock(llvm: succ)
+  }
+
+  /// Updates the successor block at the specified index.
+  public func setSuccessor(at idx: Int, to bb: BasicBlock) {
+    LLVMSetSuccessor(llvm, UInt32(idx), bb.asLLVM())
+  }
+}