termCallBr missing operands

llir · Aug 21, 2021 · 2a7b338 · 2a7b338
1 parent 9363750
commit 2a7b338
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ir/ir_test.go b/ir/ir_test.go
@@ -117,6 +117,7 @@ var (
 	_ Terminator = (*TermRet)(nil)
 	_ Terminator = (*TermBr)(nil)
 	_ Terminator = (*TermCondBr)(nil)
+	_ Terminator = (*TermCallBr)(nil)
 	_ Terminator = (*TermSwitch)(nil)
 	_ Terminator = (*TermIndirectBr)(nil)
 	_ Terminator = (*TermInvoke)(nil)

diff --git a/ir/terminator.go b/ir/terminator.go
@@ -518,6 +518,10 @@ type TermCallBr struct {
 	Metadata
 }
 
+func (term *TermCallBr) Operands() []value.Value {
+	return append(append(append([]value.Value{term.Callee}, term.Args...), term.NormalRetTarget), term.OtherRetTargets...)
+}
+
 // NewCallBr returns a new callbr terminator based on the given callee, function
 // arguments and control flow return points for normal and exceptional
 // execution.