rsa: black formatting, ignore idea files
Signed-off-by: HeshamTB <hishaminv@gmail.com>
This commit is contained in:
parent
8c7bc52d9a
commit
a58f9b9337
GPG Key ID:
74876157D199B09E
2
.gitignore
vendored
2
.gitignore
vendored
@ -9,3 +9,5 @@ build/
|
||||
|
||||
#venv dir
|
||||
venv/
|
||||
|
||||
.idea/
|
||||
|
||||
132
rsa.py
132
rsa.py
@ -28,6 +28,7 @@ Q=4
|
||||
PHI = 5
|
||||
ID = 6
|
||||
|
||||
|
||||
def main():
|
||||
print("hesham-rsa version ", VERSION)
|
||||
if sys.argv[1] == "gen" and len(sys.argv) == 4: ##rsa gen <keysize> <keyname>
|
||||
@ -56,7 +57,9 @@ def main():
|
||||
print(ex)
|
||||
sys.exit(1)
|
||||
sys.exit(0)
|
||||
if sys.argv[1] == "encrypt" and len(sys.argv) == 5: ##rsa encrypt <message> <key> <signer>
|
||||
if (
|
||||
sys.argv[1] == "encrypt" and len(sys.argv) == 5
|
||||
): ##rsa encrypt <message> <key> <signer>
|
||||
msg = sys.argv[2]
|
||||
msg_list = msg.split()
|
||||
keyName = sys.argv[3]
|
||||
@ -69,7 +72,9 @@ def main():
|
||||
msg_encrypted = msg_encrypted + " " + hex(encrypt(word, key_public))
|
||||
# msg_encrypted = encrypt(msg, key_public)
|
||||
print("Encrypted msg: \n", msg_encrypted)
|
||||
print("Signed: \n", sign(msg_encrypted, signing_key)) ## Adds an encrypted sig at the end of message.
|
||||
print(
|
||||
"Signed: \n", sign(msg_encrypted, signing_key)
|
||||
) ## Adds an encrypted sig at the end of message.
|
||||
sys.exit(0)
|
||||
elif sys.argv[1] == "encrypt":
|
||||
print("Not enough arguments")
|
||||
@ -83,7 +88,9 @@ def main():
|
||||
msg_decrypted = ""
|
||||
key = readKeyFile(sys.argv[3])
|
||||
for cipher_word in cipher_list:
|
||||
msg_decrypted = msg_decrypted + " " + str(decrypt(int(cipher_word, 16),key[D],key[N]))
|
||||
msg_decrypted = (
|
||||
msg_decrypted + " " + str(decrypt(int(cipher_word, 16), key[D], key[N]))
|
||||
)
|
||||
if sig == None:
|
||||
print("\033[91mUnknown signature! \u2717" + "\033[0m")
|
||||
else:
|
||||
@ -92,7 +99,7 @@ def main():
|
||||
sys.exit(0)
|
||||
elif sys.argv[1] == "decrypt":
|
||||
print("Not enough arguments")
|
||||
print("rsa decrypt \"<cipher>\" <keyid>")
|
||||
print('rsa decrypt "<cipher>" <keyid>')
|
||||
sys.exit(1)
|
||||
if sys.argv[1] == "list":
|
||||
listKeys()
|
||||
@ -115,7 +122,9 @@ def main():
|
||||
if sys.argv[1] == "is_prime" and len(sys.argv) == 4: # rsa is_prime <base> <N>
|
||||
isPrime_cmd(0)
|
||||
sys.exit(0)
|
||||
if sys.argv[1] == "is_prime_mr" and len(sys.argv) == 4: #rsa is_prime_mr <base> <N>
|
||||
if (
|
||||
sys.argv[1] == "is_prime_mr" and len(sys.argv) == 4
|
||||
): # rsa is_prime_mr <base> <N>
|
||||
isPrime_cmd(1)
|
||||
sys.exit(0)
|
||||
if sys.argv[1] == "genrand" and len(sys.argv) == 3: # rsa genrand <bits>
|
||||
@ -124,7 +133,9 @@ def main():
|
||||
if sys.argv[1] == "genprime" and len(sys.argv) == 3: # rsa genprime <bits>
|
||||
print(getPrime(int(sys.argv[2])))
|
||||
sys.exit(0)
|
||||
if sys.argv[1] == "prime_factors" and len(sys.argv) == 4: #rsa primefactors <base> <N>
|
||||
if (
|
||||
sys.argv[1] == "prime_factors" and len(sys.argv) == 4
|
||||
): # rsa primefactors <base> <N>
|
||||
prime_factors(sys.argv[3], sys.argv[2])
|
||||
sys.exit(0)
|
||||
if sys.argv[1] == "print" and len(sys.argv) == 3: # rsa print <key>
|
||||
@ -142,9 +153,9 @@ def main():
|
||||
sys.exit(127)
|
||||
|
||||
|
||||
|
||||
def generateKeys(id, bits=64):
|
||||
from multiprocessing.pool import Pool
|
||||
|
||||
# Primes of size 32 bit random
|
||||
# resulting in a 64-bit key mod
|
||||
pool = Pool()
|
||||
@ -169,6 +180,7 @@ def generateKeys(id, bits=64):
|
||||
d = pow(e, -1, phi) # d = e^-1 mod phi
|
||||
return (n, e, d, p, q, phi, id)
|
||||
|
||||
|
||||
def encrypt(message, publicKey):
|
||||
msg_text = message
|
||||
n = publicKey[N]
|
||||
@ -181,25 +193,29 @@ def encrypt(message, publicKey):
|
||||
msg_encrypted_number_form = pow(msg_number_form, e, n) # c = msg^e mod n
|
||||
return msg_encrypted_number_form
|
||||
|
||||
|
||||
def decrypt(cipher, privateKey, n):
|
||||
msg_encrypted_number_form = cipher
|
||||
d = privateKey
|
||||
msg_decrypted_number_form = pow(msg_encrypted_number_form, d, n) # msg = c^d mod n
|
||||
msg_decrypted = int(msg_decrypted_number_form)
|
||||
try:
|
||||
msg_decrypted = str(msg_decrypted.to_bytes(msg_decrypted.bit_length(), byteOrder).decode()).strip()
|
||||
msg_decrypted = str(
|
||||
msg_decrypted.to_bytes(msg_decrypted.bit_length(), byteOrder).decode()
|
||||
).strip()
|
||||
except UnicodeDecodeError:
|
||||
# print("decrypt: Cant decrypt properly")
|
||||
return ""
|
||||
return msg_decrypted
|
||||
|
||||
|
||||
def getPrime(bits):
|
||||
while True:
|
||||
# Byte order "little" or "big" does not matter here since we want a random number from os.urandom()
|
||||
x = int.from_bytes(os.urandom(int(bits / 8)), byteOrder)
|
||||
print('Trying: ', x, end="\n")
|
||||
print("Trying: ", x, end="\n")
|
||||
if mr.is_prime(x):
|
||||
print("\nprime: ", x, '\n')
|
||||
print("\nprime: ", x, "\n")
|
||||
return x
|
||||
# backTrack(x)
|
||||
|
||||
@ -226,10 +242,12 @@ def isPrime(number):
|
||||
return False
|
||||
return True
|
||||
|
||||
|
||||
def gen_random(bits: int):
|
||||
x = int.from_bytes(os.urandom(int(bits / 8)), byteOrder)
|
||||
return x
|
||||
|
||||
|
||||
def sign(encrypted_msg, key):
|
||||
enc_msg = str(encrypted_msg)
|
||||
encrypted_msg_list = enc_msg.split()
|
||||
@ -240,6 +258,7 @@ def sign(encrypted_msg, key):
|
||||
signed_msg = str(signed_msg) + " " + str(word)
|
||||
return signed_msg.strip()
|
||||
|
||||
|
||||
def verify(cipher_list):
|
||||
local_keys = os.listdir(keysFolder)
|
||||
cipher_list.reverse() # To get last word using index 0
|
||||
@ -252,11 +271,13 @@ def verify(cipher_list):
|
||||
sig = str(decrypt(int(encrypted_sig, 16), key[E], key[N]))
|
||||
if "sig:" in sig:
|
||||
return sig.replace("sig:", "")
|
||||
else: continue
|
||||
else: return None
|
||||
else:
|
||||
continue
|
||||
else:
|
||||
return None
|
||||
|
||||
|
||||
def isPrime_cmd(func):
|
||||
|
||||
number = int_base_n_from_str(sys.argv[3], sys.argv[2])
|
||||
|
||||
if func == 0:
|
||||
@ -265,13 +286,13 @@ def isPrime_cmd(func):
|
||||
prime = mr.is_prime(number)
|
||||
|
||||
if prime:
|
||||
print('Prime')
|
||||
print("Prime")
|
||||
# print(number)
|
||||
else:
|
||||
print('Not prime')
|
||||
print("Not prime")
|
||||
|
||||
|
||||
def prime_factors(number, base):
|
||||
|
||||
num = int_base_n_from_str(number, base)
|
||||
factors = {1: 1}
|
||||
k = 0
|
||||
@ -283,7 +304,7 @@ def prime_factors(number, base):
|
||||
|
||||
for i in range(3, int(math.sqrt(num)) + 1, 2):
|
||||
j = 0
|
||||
while (num % i == 0):
|
||||
while num % i == 0:
|
||||
j += 1
|
||||
num = num / i
|
||||
if j != 0:
|
||||
@ -292,20 +313,32 @@ def prime_factors(number, base):
|
||||
factors.update({int(num): 1})
|
||||
|
||||
print(factors)
|
||||
|
||||
|
||||
def readKeyFile(keyName):
|
||||
key = tuple()
|
||||
with open(keysFolder + keyName, "r") as keyFile:
|
||||
tempkey = keyFile.readlines()
|
||||
if len(tempkey) == 3: # means it only public part (n, e, id)
|
||||
key = (int(tempkey[N].strip(), 16), int(tempkey[E].strip(), 16), 0, 0, 0, 0, tempkey[2])
|
||||
key = (
|
||||
int(tempkey[N].strip(), 16),
|
||||
int(tempkey[E].strip(), 16),
|
||||
0,
|
||||
0,
|
||||
0,
|
||||
0,
|
||||
tempkey[2],
|
||||
)
|
||||
else: # Make this a loop from 0 to 5
|
||||
key = (int(tempkey[N].strip(), 16),
|
||||
key = (
|
||||
int(tempkey[N].strip(), 16),
|
||||
int(tempkey[E].strip(), 16),
|
||||
int(tempkey[D].strip(), 16),
|
||||
int(tempkey[P].strip(), 16),
|
||||
int(tempkey[Q].strip(), 16),
|
||||
int(tempkey[PHI].strip(), 16),
|
||||
str(tempkey[ID].strip()))
|
||||
str(tempkey[ID].strip()),
|
||||
)
|
||||
return key
|
||||
|
||||
|
||||
@ -320,21 +353,24 @@ def saveKeyFile(key, fileName):
|
||||
pass
|
||||
keyFile.write(key[ID] + "\n")
|
||||
|
||||
|
||||
def printKey(key):
|
||||
n = key[N]
|
||||
e = key[E]
|
||||
d = key[D]
|
||||
id = key[ID]
|
||||
print("----------------------------------------------"+
|
||||
"\nID: {}".format(id) +
|
||||
"\n{}-BIT KEY".format(n.bit_length())+
|
||||
"\nPUBLIC PART:"+
|
||||
"\n{0}/{1}".format(hex(n), hex(e))+
|
||||
"\nPTIVATE PART:"+
|
||||
"\n{0}".format(hex(d))+
|
||||
"\n----------------------------------------------",
|
||||
print(
|
||||
"----------------------------------------------"
|
||||
+ "\nID: {}".format(id)
|
||||
+ "\n{}-BIT KEY".format(n.bit_length())
|
||||
+ "\nPUBLIC PART:"
|
||||
+ "\n{0}/{1}".format(hex(n), hex(e))
|
||||
+ "\nPTIVATE PART:"
|
||||
+ "\n{0}".format(hex(d))
|
||||
+ "\n----------------------------------------------",
|
||||
)
|
||||
|
||||
|
||||
def listKeys():
|
||||
if not os.path.isdir(keysFolder):
|
||||
os.makedirs(keysFolder)
|
||||
@ -348,15 +384,18 @@ def listKeys():
|
||||
key = readKeyFile(keyName)
|
||||
if key[D] == 0:
|
||||
check = "".strip()
|
||||
else: check = '\u2713'
|
||||
else:
|
||||
check = "\u2713"
|
||||
print("%10s%7s%7s-bit" % (key[ID].strip(), check, key[N].bit_length()))
|
||||
|
||||
|
||||
def exportKey(keyFileName):
|
||||
key = readKeyFile(keyFileName)
|
||||
public_key = (key[N], key[E], 0, 0, 0, 0, key[ID])
|
||||
saveKeyFile(public_key, key[ID] + "-public")
|
||||
print("Saved public form of key {} in keys folder".format(key[ID]))
|
||||
|
||||
|
||||
def crackKey(keyName):
|
||||
print("in crack")
|
||||
key = readKeyFile(keyName)
|
||||
@ -373,8 +412,11 @@ def crackKey(keyName):
|
||||
d = pow(e, -1, phi)
|
||||
key_cracked = (n, e, d, p, q, phi, str(keyName + "-cracked"))
|
||||
return key_cracked
|
||||
else: pass
|
||||
else: pass
|
||||
else:
|
||||
pass
|
||||
else:
|
||||
pass
|
||||
|
||||
|
||||
def crackKey2(keyName):
|
||||
print("in crack")
|
||||
@ -385,7 +427,8 @@ def crackKey2(keyName):
|
||||
print("bits: ", bits)
|
||||
while True:
|
||||
number = int.from_bytes(os.urandom(int(bits / 8)), byteOrder)
|
||||
if number == 0 or number == 1: continue
|
||||
if number == 0 or number == 1:
|
||||
continue
|
||||
print("Trying prime: ", number, end="\r")
|
||||
# if number devides n then it p or q
|
||||
if n % number == 0:
|
||||
@ -394,24 +437,28 @@ def crackKey2(keyName):
|
||||
print("p: ", p)
|
||||
q = int(n / p)
|
||||
phi = (p - 1) * (q - 1)
|
||||
if phi == 0: continue
|
||||
if phi == 0:
|
||||
continue
|
||||
e = 65537
|
||||
d = pow(e, -1, phi)
|
||||
key_cracked = (n, e, d, p, q, phi, str(keyName + "-cracked"))
|
||||
print(key_cracked)
|
||||
return key_cracked
|
||||
else: continue
|
||||
else:
|
||||
continue
|
||||
|
||||
|
||||
def printHelp():
|
||||
print("commands:")
|
||||
print("rsa gen <keysize> <keyname>")
|
||||
print("rsa encrypt <message> <key> <signer>")
|
||||
print("rsa decrypt \"<cipher>\" <key>")
|
||||
print('rsa decrypt "<cipher>" <key>')
|
||||
print("rsa export <key>")
|
||||
print("rsa crack <key>")
|
||||
print("rsa print <key>")
|
||||
print("rsa list")
|
||||
|
||||
|
||||
def backTrack(x):
|
||||
# Back track and clear terminal with length of x
|
||||
length = len(str(x))
|
||||
@ -419,25 +466,30 @@ def backTrack(x):
|
||||
print("\b", end="")
|
||||
length -= 1
|
||||
|
||||
|
||||
def keyExist(keyName):
|
||||
exist = os.path.exists(keysFolder + keyName)
|
||||
return exist
|
||||
|
||||
def int_base_n_from_str(st: str, base):
|
||||
|
||||
def int_base_n_from_str(st: str, base):
|
||||
try:
|
||||
base = int(base)
|
||||
except ValueError as e:
|
||||
print(f'Value {sys.argv[2]} is not a valid base (2, 8, 10, 16)',
|
||||
file=sys.stderr)
|
||||
print(
|
||||
f"Value {sys.argv[2]} is not a valid base (2, 8, 10, 16)", file=sys.stderr
|
||||
)
|
||||
exit(-1)
|
||||
try:
|
||||
number = int(st, base)
|
||||
except ValueError as e:
|
||||
print(f'Value {sys.argv[3]} is not valid for as a base {base} number',
|
||||
file=sys.stderr)
|
||||
print(
|
||||
f"Value {sys.argv[3]} is not valid for as a base {base} number",
|
||||
file=sys.stderr,
|
||||
)
|
||||
exit(-1)
|
||||
return number
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
|
||||
Loading…
Reference in New Issue
Block a user