1 files changed, 58 insertions, 0 deletions
diff --git a/keyconf.py b/keyconf.py
new file mode 100755
index 0000000..78d3af1
--- /dev/null
+++ b/keyconf.py
@@ -0,0 +1,58 @@
+#!/usr/bin/python
+import sqlite, time
+from sys import argv
+
+con = sqlite.connect('yubikeys.sqlite')
+cur = con.cursor()
+
+if (len(argv)<3):
+	print "YubiServe Key Management Tool 1.0\n"
+	print " -a <publicname> <internalname> <aeskey>\tAdd a new key"
+	print " -k <publicname>\t\t\t\tDelete a key from database"
+	print " -d <publicname>\t\t\t\tDisable a key from database (it will be kept)"
+	print " -e <publicname>\t\t\t\tEnable a key within the database\n"
+else:
+	if ((argv[1] == '-d') and (len(argv)>2)):
+		cur.execute("SELECT * FROM yubikeys WHERE publicname = '" + argv[2] + "'")
+		if (cur.rowcount == 0):
+			print "Key not found."
+		else:
+			cur.execute("SELECT * FROM yubikeys WHERE publicname = '" + argv[2] + "' AND active = 'true'")
+			if (cur.rowcount == 1):
+				cur.execute("UPDATE yubikeys SET active = 'false' WHERE publicname = '" + argv[2] + "'")
+				print "Key " + argv[2] + " disabled."
+				con.commit()
+			else:
+				print "Key is already disabled."
+
+	elif ((argv[1] == '-e') and (len(argv)>2)):
+		cur.execute("SELECT * FROM yubikeys WHERE publicname = '" + argv[2] + "'")
+		if (cur.rowcount == 0):
+			print "Key not found."
+		else:
+			cur.execute("SELECT * FROM yubikeys WHERE publicname = '" + argv[2] + "' AND active = 'false'")
+			if (cur.rowcount == 1):
+				cur.execute("UPDATE yubikeys SET active = 'true' WHERE publicname = '" + argv[2] + "'")
+				print "Key " + argv[2] + " enabled."
+				con.commit()
+			else:
+				print "Key is already enabled."
+	elif ((argv[1] == '-k') and (len(argv)>2)):
+		cur.execute("SELECT * FROM yubikeys WHERE publicname = '" + argv[2] + "'")
+		if (cur.rowcount == 0):
+			print "Key not found."
+		else:
+			cur.execute("DELETE FROM yubikeys WHERE publicname = '" + argv[2] + "'")
+			print "Key " + argv[2] + " deleted."
+			con.commit()
+	elif ((argv[1] == '-a') and (len(argv)>3)):
+		if ((len(argv[2])<=16) and (len(argv[3]) <= 12) and (len(argv[4])<=32)):
+			cur.execute("SELECT * FROM yubikeys WHERE publicname = '" + argv[2] + "'")
+			if (cur.rowcount == 0):
+				cur.execute("INSERT INTO yubikeys VALUES ('" + argv[2] + "', '" + time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()) + "', '" + argv[3] + "', '" + argv[4] + "', 'true', 1, 1)")
+				con.commit()
+				print "Key " + argv[2] + " added to database."
+			else:
+				print "Key is already into database. Delete it before adding the same key!"
+	else:
+		print "Not enough parameters. Try looking at " + argv[0] + " --help"