startup.py file runs commands on startup

1 files changed, 14 insertions, 0 deletions

diff --git a/python/scriptingprovider.py b/python/scriptingprovider.py
index 473cd36f..1bec8d61 100644
--- a/python/scriptingprovider.py
+++ b/python/scriptingprovider.py
@@ -609,6 +609,20 @@ class PythonScriptingInstance(ScriptingInstance):
 			self.completer = bncompleter.Completer(namespace = self.locals)
 
 			self.interpreter.push("from binaryninja import *")
+			
+			startup_file = os.path.join(binaryninja.user_directory(), "startup.py")
+			if os.path.isfile(startup_file):
+				with open(startup_file, 'r') as f:
+					for line in f.readlines():
+						self.interpreter.push(line)
+				# Finish input in case the file does not end with a newline
+				self.interpreter.push('\n')
+
+			else:
+				with open(startup_file, 'w') as f:
+					f.write("# Commands in this file will be run in the interactive python console on startup\n")
+					f.write("from binaryninja import *\n")
+					f.write("\n")
 
 		def execute(self, code):
 			self.code = code