Add cancelScript api for scriptingProvider

1 files changed, 23 insertions, 0 deletions

diff --git a/python/scriptingprovider.py b/python/scriptingprovider.py
index 21a68fcd..690bd6cc 100644
--- a/python/scriptingprovider.py
+++ b/python/scriptingprovider.py
@@ -107,6 +107,7 @@ class ScriptingInstance(object):
 			self._cb.context = 0
 			self._cb.destroyInstance = self._cb.destroyInstance.__class__(self._destroy_instance)
 			self._cb.executeScriptInput = self._cb.executeScriptInput.__class__(self._execute_script_input)
+			self._cb.cancelScriptInput = self._cb.cancelScriptInput.__class__(self._cancel_script_input)
 			self._cb.setCurrentBinaryView = self._cb.setCurrentBinaryView.__class__(self._set_current_binary_view)
 			self._cb.setCurrentFunction = self._cb.setCurrentFunction.__class__(self._set_current_function)
 			self._cb.setCurrentBasicBlock = self._cb.setCurrentBasicBlock.__class__(self._set_current_basic_block)
@@ -133,6 +134,13 @@ class ScriptingInstance(object):
 			log.log_error(traceback.format_exc())
 			return ScriptingProviderExecuteResult.InvalidScriptInput
 
+	def _cancel_script_input(self, ctxt):
+		try:
+			return self.perform_cancel_script_input()
+		except:
+			log.log_error(traceback.format_exc())
+			return ScriptingProviderExecuteResult.ScriptExecutionCancelled
+
 	def _set_current_binary_view(self, ctxt, view):
 		try:
 			if view:
@@ -189,6 +197,10 @@ class ScriptingInstance(object):
 		return ScriptingProviderExecuteResult.InvalidScriptInput
 
 	@abc.abstractmethod
+	def perform_cancel_script_input(self):
+		return ScriptingProviderExecuteResult.ScriptExecutionCancelled
+
+	@abc.abstractmethod
 	def perform_set_current_binary_view(self, view):
 		raise NotImplementedError
 
@@ -225,6 +237,9 @@ class ScriptingInstance(object):
 	def execute_script_input(self, text):
 		return core.BNExecuteScriptInput(self.handle, text)
 
+	def cancel_script_input(self, text):
+		return core.BNCancelScriptInput(self.handle)
+
 	def set_current_binary_view(self, view):
 		if view is not None:
 			view = view.handle
@@ -636,6 +651,14 @@ class PythonScriptingInstance(ScriptingInstance):
 		return ScriptingProviderExecuteResult.SuccessfulScriptExecution
 
 	@abc.abstractmethod
+	def perform_cancel_script_input(self):
+		for tid, tobj in threading._active.items():
+			if tobj is self.interpreter:
+				if ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid), ctypes.py_object(KeyboardInterrupt)) != 1:
+					ctypes.pythonapi.PyThreadState_SetAsyncExc(ctypes.c_long(tid), None)
+				break
+
+	@abc.abstractmethod
 	def perform_set_current_binary_view(self, view):
 		self.interpreter.current_view = view