From 19523a75b5bfd1956bf415409e0dec4e98bc47f1 Mon Sep 17 00:00:00 2001 From: Serizao <11671895+Serizao@users.noreply.github.com> Date: Wed, 2 Mar 2022 08:07:20 +0100 Subject: [PATCH] Update database.py --- cme/protocols/smb/database.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/cme/protocols/smb/database.py b/cme/protocols/smb/database.py index 283a0ff5..32d9459a 100755 --- a/cme/protocols/smb/database.py +++ b/cme/protocols/smb/database.py @@ -161,11 +161,11 @@ class database: results = cur.fetchall() if not len(results): - cur.execute("INSERT INTO computers (ip, hostname, domain, os, dc) VALUES (?,?,?,?,?)", [ip, hostname, domain, os, dc]) + cur.execute("INSERT INTO computers (ip, hostname, domain, os, dc, smbv1, signing) VALUES (?,?,?,?,?,?,?)", [ip, hostname, domain, os, dc, smbv1, signing]) else: for host in results: if (hostname != host[2]) or (domain != host[3]) or (os != host[4]): - cur.execute("UPDATE computers SET hostname=?, domain=?, os=? WHERE id=?", [hostname, domain, os, host[0]]) + cur.execute("UPDATE computers SET hostname=?, domain=?, os=?, smbv1, signing WHERE id=?", [hostname, domain, os, smbv1, signing, host[0]]) if dc != None and (dc != host[5]): cur.execute("UPDATE computers SET dc=? WHERE id=?", [dc, host[0]])