Add additional context on integer tokens

3 files changed, 6 insertions, 1 deletions

diff --git a/binaryninjacore.h b/binaryninjacore.h
index 97cdf853..a78e5b3b 100644
--- a/binaryninjacore.h
+++ b/binaryninjacore.h
@@ -271,7 +271,9 @@ extern "C"
 		NoTokenContext = 0,
 		LocalVariableTokenContext = 1,
 		DataVariableTokenContext = 2,
-		FunctionReturnTokenContext = 3
+		FunctionReturnTokenContext = 3,
+		InstructionAddressTokenContext = 4,
+		ILInstructionIndexTokenContext = 5
 	};
 
 	enum BNLinearDisassemblyLineType
diff --git a/ui/flowgraphwidget.h b/ui/flowgraphwidget.h
index e90bcdc5..036d042c 100644
--- a/ui/flowgraphwidget.h
+++ b/ui/flowgraphwidget.h
@@ -176,6 +176,8 @@ protected:
 	void selectNone();
 	void navigateToHighlightedToken();
 
+	uint64_t getTokenAddress();
+
 public:
 	FlowGraphWidget(QWidget* parent, BinaryViewRef view, FlowGraphRef graph = FlowGraphRef());
 	~FlowGraphWidget();
diff --git a/ui/linearview.h b/ui/linearview.h
index 2c9c5fcc..3e089732 100644
--- a/ui/linearview.h
+++ b/ui/linearview.h
@@ -181,6 +181,7 @@ class BINARYNINJAUIAPI LinearView: public QAbstractScrollArea, public View, publ
 	void updateAnalysisRequestorsForCache();
 
 	ArchitectureRef getArchitecture(const LinearViewLine& line);
+	uint64_t getTokenAddress();
 
 	BNAnalysisWarningActionType getAnalysisWarningActionAtPos(const LinearViewLine& line, int x);