def AddUser(self):
user = raw_input("USER NAME: ")
password = raw_input("PASSWORD: "******"PASSWORD SECOND TIME: ")
if password != spw:
print "WRONG PASSWORD"
return
cur = self.con.cursor()
select_user = ("SELECT * FROM `USER` WHERE `NAME` = %s")
data_user = [user]
rows = cur.execute(select_user, data_user)
if rows!=0:
print "ALREADY EXIST"
return
check = commands.getoutput("[ -d /Volumes/important\ files/keys ] && echo \"Y\" || echo \"N\"")
if check=="N":
print "PLUG IN YOU KEY DEVICE"
return
insert_user = ("INSERT INTO `USER` (`NAME`, `SALT`, `PASSWORD`, `KEY`, `IV`) VALUES (%s, %s, %s, %s, %s);")
salt = int(random.uniform(1000, 9999))
kg = keyGenerator()
kg.gen()
key = kg.getKey()
IV = kg.getIV()
commands.getoutput("echo \"%s\">/Volumes/important\ files/keys/%s.key"%(key,user))
insert_data = (user, salt, hashlib.sha224(password+str(salt)).digest(), key, IV)
cur.execute(insert_user, insert_data)
self.con.commit()
print "CREATE SUCCESS"
def main():
while True:
print "Choose from job"
print "1.Key Generation"
print "2.Encrytion"
print "3.Decryption"
print "4.Exit"
choice = sys.stdin.readline()
if (choice == '1\n'):
myKeyGenerator = keyGenerator()
myKeyGenerator.keyGen()
print "Key Generation Finished\n"
elif (choice == '2\n'):
encryptor = Encryption()
encryptor.encrypt()
print "Encryption Finished\n"
elif (choice == '3\n'):
decryptor = Decryption()
decryptor.decrypt()
print "Decryption Finished\n"
elif (choice == '4\n'):
break;
else:
print "invalid input"
choice = sys.stdin.readline()
def encrypt(self):
ctext = open("ctext.txt", "w")
totalBlocks = self.readInput()
for block in totalBlocks:
self.m = int(''.join(block), 2)
myKeyGenerator = keyGenerator()
k = random.randrange(self.p)
self.C1 = myKeyGenerator.modExp(self.g, k, self.p)
self.C2 = (myKeyGenerator.modExp(self.e2, k, self.p) *
self.m) % self.p
ctext.write(
format(self.C1, "033b") + "," + format(self.C2, "033b") + "\n")
def encrypt(self):
ctext = open("ctext.txt","w")
totalBlocks = self.readInput()
for block in totalBlocks:
self.m = int(''.join(block),2)
myKeyGenerator = keyGenerator()
k = random.randrange(self.p)
self.C1 = myKeyGenerator.modExp(self.g, k, self.p)
self.C2 = (myKeyGenerator.modExp(self.e2, k, self.p) * self.m) % self.p
ctext.write(format(self.C1, "033b") + ","+ format(self.C2,"033b") + "\n")
def decrypt(self):
decrypted = open("dtext.txt","w")
ctext = open("ctext.txt","r")
keyFile = open("prikey.txt","r")
keyString = keyFile.read()
keyList = keyString.split(',')
self.p = int(keyList[0],2)
self.d = int(keyList[2],2)
message = ''
for line in ctext:
cipherList = line.split(',')
self.C1 = int(cipherList[0],2)
self.C2 = int(cipherList[1],2)
myKeyGenerator = keyGenerator()
temp = myKeyGenerator.modExp(self.C1, self.p-1-self.d , self.p)
self.m = (temp * self.C2) % self.p
message += format(self.m,"08x").decode("hex")
decrypted.write(format(self.m,"08x").decode("hex"))
print 'Message: \n', message, '\n'
def decrypt(self):
decrypted = open("dtext.txt", "w")
ctext = open("ctext.txt", "r")
keyFile = open("prikey.txt", "r")
keyString = keyFile.read()
keyList = keyString.split(',')
self.p = int(keyList[0], 2)
self.d = int(keyList[2], 2)
message = ''
for line in ctext:
cipherList = line.split(',')
self.C1 = int(cipherList[0], 2)
self.C2 = int(cipherList[1], 2)
myKeyGenerator = keyGenerator()
temp = myKeyGenerator.modExp(self.C1, self.p - 1 - self.d, self.p)
self.m = (temp * self.C2) % self.p
message += format(self.m, "08x").decode("hex")
decrypted.write(format(self.m, "08x").decode("hex"))
print 'Message: \n', message, '\n'
def AddUser(self):
user = raw_input("USER NAME: ")
password = raw_input("PASSWORD: "******"PASSWORD SECOND TIME: ")
if password != spw:
print "WRONG PASSWORD"
return
cur = self.con.cursor()
select_user = ("SELECT * FROM `USER` WHERE `NAME` = %s")
data_user = [user]
rows = cur.execute(select_user, data_user)
if rows != 0:
print "ALREADY EXIST"
return
check = commands.getoutput(
"[ -d /Volumes/important\ files/keys ] && echo \"Y\" || echo \"N\""
)
if check == "N":
print "PLUG IN YOU KEY DEVICE"
return
insert_user = (
"INSERT INTO `USER` (`NAME`, `SALT`, `PASSWORD`, `KEY`, `IV`) VALUES (%s, %s, %s, %s, %s);"
)
salt = int(random.uniform(1000, 9999))
kg = keyGenerator()
kg.gen()
key = kg.getKey()
IV = kg.getIV()
commands.getoutput(
"echo \"%s\">/Volumes/important\ files/keys/%s.key" % (key, user))
insert_data = (user, salt,
hashlib.sha224(password + str(salt)).digest(), key, IV)
cur.execute(insert_user, insert_data)
self.con.commit()
print "CREATE SUCCESS"
import sys
import MySQLdb as mdb
import random
import hashlib
if __name__ == "__main__":
con = mdb.connect('localhost', 'CS8120', '8224074zms', 'CS8120')
user = raw_input("ADMIN NAME: ")
password = raw_input("PASSWORD: "******"PASSWORD SECOND TIME: ")
if password != spw:
print "WRONG PASSWORD"
else:
con = conDB().conDB()
with con:
cur = con.cursor()
insert_user = ("INSERT INTO `USER` (`NAME`, `SALT`, `PASSWORD`, `PRIVILEGE`, `KEY`, `IV`) VALUES (%s, %s, %s, %s, %s, %s);")
salt = int(random.uniform(1000, 9999))
kg = keyGenerator()
kg.gen()
key = kg.getKey()
iv = kg.getIV()
insert_data = (user, salt, hashlib.sha224(password+str(salt)).digest(), 1, key, IV)
cur.execute(insert_user, insert_data)
print "CREATE SUCCESS"
import hashlib
if __name__ == "__main__":
con = mdb.connect('localhost', 'CS8120', '8224074zms', 'CS8120')
user = raw_input("ADMIN NAME: ")
password = raw_input("PASSWORD: "******"PASSWORD SECOND TIME: ")
if password != spw:
print "WRONG PASSWORD"
else:
con = conDB().conDB()
with con:
cur = con.cursor()
insert_user = (
"INSERT INTO `USER` (`NAME`, `SALT`, `PASSWORD`, `PRIVILEGE`, `KEY`, `IV`) VALUES (%s, %s, %s, %s, %s, %s);"
)
salt = int(random.uniform(1000, 9999))
kg = keyGenerator()
kg.gen()
key = kg.getKey()
iv = kg.getIV()
insert_data = (user, salt,
hashlib.sha224(password + str(salt)).digest(), 1,
key, IV)
cur.execute(insert_user, insert_data)
print "CREATE SUCCESS"
