Trilltechnician/hack-house
1
1
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Updated client/client.py

Browse Source
This commit is contained in:
mirai 2022-12-01 10:21:14 +03:00
parent 749fc9c6ad
commit bc2f652d16
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
client/client.py
View File

@ -11,28 +11,25 @@ if "Linux" in str(platform.platform()):
OS = "Linux" OS = "Linux"
class Client: class Client:
def _key_gen(self) -> None: def _key_gen(self) -> None:
(pubkey, privkey) = rsa.newkeys(512) (pubkey, privkey) = rsa.newkeys(512)
with open("private.pem", "wb") as f: with open("private.pem", "wb") as f:
f.write(privkey.save_pkcs1()) f.write(privkey.save_pkcs1())
with open("public.pem", "wb") as f: with open("public.pem", "wb") as f:
f.write(pubkey.save_pkcs1()) f.write(pubkey.save_pkcs1())
def __init__(self, username: str): def __init__(self, username: str):
self.server = input("server ip: \n") self.server = input("server ip: \n")
self.port = input("server port: \n") self.port = input("server port: \n")
self.username = username self.username = username
# Urls
self.base_url = f"http://{self.server}:{self.port}" self.base_url = f"http://{self.server}:{self.port}"
self.talk_url = f"{self.base_url}/talk" self.talk_url = f"{self.base_url}/talk"
self.info_url = f"{self.base_url}/update" self.info_url = f"{self.base_url}/update"
self.key_url = f"{self.base_url}/get_key" self.key_url = f"{self.base_url}/get_key"
# Keys
self.pubkey = None self.pubkey = None
self.privkey = None self.privkey = None
self.symetric_key = None self.symetric_key = None