From 4c30ed90547be1c3451166dd67c9d674f3f98383 Mon Sep 17 00:00:00 2001
From: Owen Mansel-Chan <owen-mc@github.com>
Date: Tue, 2 Feb 2021 11:18:31 +0000
Subject: [PATCH] Add predicate to get return statement from return instruction

---
 ql/src/semmle/go/controlflow/IR.qll | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/ql/src/semmle/go/controlflow/IR.qll b/ql/src/semmle/go/controlflow/IR.qll
index 1b1e7df9425..319d2193d00 100644
--- a/ql/src/semmle/go/controlflow/IR.qll
+++ b/ql/src/semmle/go/controlflow/IR.qll
@@ -905,6 +905,8 @@ module IR {
 
     ReturnInstruction() { this = MkReturnNode(ret) }
 
+    ReturnStmt getReturnStmt() { result = ret }
+
     /** Holds if this statement returns multiple results. */
     predicate returnsMultipleResults() { exists(MkExtractNode(ret, _)) or ret.getNumExpr() > 1 }