From 18a3d3bc354e667bc58385e59745b82b53695139 Mon Sep 17 00:00:00 2001 From: AL-LCL Date: Fri, 19 May 2023 11:06:25 +0200 Subject: NexRAT --- Utilities/db_queries.py | 36 ++++++++++++++++++++++++++++++++++++ 1 file changed, 36 insertions(+) create mode 100644 Utilities/db_queries.py (limited to 'Utilities/db_queries.py') diff --git a/Utilities/db_queries.py b/Utilities/db_queries.py new file mode 100644 index 0000000..db4a08e --- /dev/null +++ b/Utilities/db_queries.py @@ -0,0 +1,36 @@ +import sqlite3 + + +def update_db(use_latest, ip, port, module_ports): + conn = sqlite3.connect('Settings.db') + c = conn.cursor() + + if use_latest is False: + try: + c.execute("""CREATE TABLE settings ( + ID INTEGER PRIMARY KEY AUTOINCREMENT, + ip TEXT, + port INTEGER, + module_ports TEXT + )""") + conn.commit() + except: + c.execute("UPDATE settings SET ip=?, port=?, module_ports=? WHERE ID=1;", (ip, port, module_ports)) + conn.commit() + else: + c.execute("INSERT INTO settings (ip, port, module_ports) VALUES (?, ?, ?);", (ip, port, module_ports)) + conn.commit() + + conn.close() + + +def get_module_data(): + conn = sqlite3.connect('Settings.db') + c = conn.cursor() + + c.execute("SELECT * FROM settings WHERE ID=1") + conn.commit() + db_data = c.fetchone() + conn.close() + + return db_data \ No newline at end of file -- cgit v1.2.3