class Peer:
def __init__(self, addr, port):
self.ip=addr
self.port=port
self.node = socket.socket()
self.crypt=RSA()
tuple=self.crypt.prime_tuple()
p, q = tuple
n=p*q
e=self.crypt.public_key(tuple)
self.pub_k=(e, n)
d=self.crypt.private_key(tuple, e)
self.priv_k=(d, n)
print(self.pub_k)
def connection(self):
self.node.connect((self.ip, self.port))
counter=0
while True:
query_en=self.node.recv(2048)
if query_en:
query=query_en.decode("utf-8")
if query in ["Enter username: "******"Enter password: "******"Enter public key: "]:
response=input(query)
type(response)
self.node.sendall(str.encode(response))
counter+=1
if counter==3:
break
message=(self.node.recv(2048)).decode("utf-8")
if message=="Name a group that you would want to create or join: ":
group=input(message)
type(group)
if group is not "":
self.node.sendall(str.encode(group))
data=self.node.recv(2048)
if data:
encoded=data.decode("utf-8")
res=self.crypt.decrypt(encoded, self.priv_k)
print(res)
self.node.close()
class Server:
def __init__(self, host, port):
self.host=host
self.port=port
self.crypt=RSA()
tuple=self.crypt.prime_tuple()
p, q = tuple
n=p*q
e=self.crypt.public_key(tuple)
self.pub_k=(e, n)
d=self.crypt.private_key(tuple, e)
self.priv_k=(d, n)
async def start(self, loop):
serv = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
try:
serv.bind((self.host,self.port))
serv.setblocking(False)
print("server started...")
except socket.error as e:
print(str(e))
serv.listen()
while True:
conn , addr = await loop.sock_accept(serv)
print(str(addr)+" connected")
loop.create_task(self.handler(conn, loop))
async def handler(self, conn, loop):
obj=Data()
hash=SHA256.new()
username=""
password=""
public_key=""
await loop.sock_sendall(conn, str.encode("Enter username: "******"utf-8")
if res:
username=res
await loop.sock_sendall(conn, str.encode("Enter password: "******"users", username, password)
if not result:
obj.user_entry(username, password)
await loop.sock_sendall(conn, str.encode("Enter public key: "))
res_en=await loop.sock_recv(conn, 2048)
res=res_en.decode("utf-8")
if res:
try:
public_key=eval(res)
except Exception as e:
print(str(e))
await loop.sock_sendall(conn, str.encode("Name a group that you would want to create or join: "))
res_en=await loop.sock_recv(conn, 2048)
res=res_en.decode("utf-8")
if res is not None:
obj.create_table(res)
key=random.randint(100000, 1000000)
key=self.crypt.encrypt(str(key), public_key)
await loop.sock_sendall(conn, str.encode(key))
result=obj.check_table(res, username, password)
if not result:
obj.group_entry(res, username, password, key)
conn.close()
#!/usr/bin/env python
from rsa import RSA
from dh import DHUser, DHSession
from elgamal import EGUser, EGSession
if __name__ == '__main__':
# RSA
# CRYPTO-19
c19 = RSA(17, 11, 7)
assert c19._d == 23
assert c19.public_key() == (7, 187)
assert c19.private_key() == (23, 187)
# CRYPTO-20
c = RSA.encrypt(c19.public_key(), 88)
m = RSA.decrypt(c19.private_key(), c)
assert c == 11
assert m == 88
# Diffie-Hellman
# CRYPTO-35
dh_session = DHSession(353, 3)
dh_user_a = DHUser(dh_session, 97)
dh_user_b = DHUser(dh_session, 233)
assert dh_user_a.y == 40
assert dh_user_b.y == 248
assert dh_session.session_key(dh_user_a, dh_user_b) == \
dh_user_a.session_key(dh_user_b.public_key()) == \
dh_user_b.session_key(dh_user_a.public_key()) == 160
