Fix for PS2 function name issue #169

BC-SECURITY · Apr 15, 2020 · f833dfb · f833dfb
1 parent 79e77d2
commit f833dfb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/setup/setup_database.py b/setup/setup_database.py
@@ -188,8 +188,8 @@
     "Invoke_Mimikatz" text 
 )''')
 
-rand1 = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(5))
-rand2 = ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(5))
+rand1 = random.choice(string.ascii_uppercase) + ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(4))
+rand2 = random.choice(string.ascii_uppercase) + ''.join(random.choice(string.ascii_uppercase + string.digits) for _ in range(4))
 c.execute("INSERT INTO functions VALUES(?,?)", (rand1, rand2))
 
 # commit the changes and close everything off