This commit is contained in:
parent
a2e09b3656
commit
be21d4100d
@ -3,6 +3,7 @@ import logging
|
||||
import os
|
||||
import shutil
|
||||
import sys
|
||||
import binascii
|
||||
from base64 import b64encode
|
||||
from Crypto.PublicKey import RSA
|
||||
from Crypto.Hash import SHA256
|
||||
@ -19,6 +20,7 @@ class Authetication:
|
||||
PRIVATE_KEY_DIRECTORY_LOCATION = CONFIG_DIRECTORY_LOCATION + os.path.sep + "private_keys"
|
||||
USER_INDEX = 0
|
||||
|
||||
|
||||
def __init__(self):
|
||||
if not os.path.isdir(self.HOME_DIRECTORY_LOCATION):
|
||||
os.mkdir(self.HOME_DIRECTORY_LOCATION)
|
||||
@ -35,6 +37,7 @@ class Authetication:
|
||||
with open(self.CONFIG_FILE_LOCATION, 'w+') as outfile:
|
||||
json.dump(data, outfile)
|
||||
|
||||
|
||||
def login(self, username: str, password: str) -> str:
|
||||
with open(self.CONFIG_FILE_LOCATION) as json_file:
|
||||
data = json.load(json_file)
|
||||
@ -51,6 +54,7 @@ class Authetication:
|
||||
auth_logger.debug("User logged in: " + username)
|
||||
return user['homeDir']
|
||||
|
||||
|
||||
def checkUserExists(self, username: str) -> bool:
|
||||
with open(self.CONFIG_FILE_LOCATION) as json_file:
|
||||
data = json.load(json_file)
|
||||
@ -60,6 +64,7 @@ class Authetication:
|
||||
return True
|
||||
return False
|
||||
|
||||
|
||||
def initConfig(self):
|
||||
data = {'index': 0, 'user': []}
|
||||
with open(self.CONFIG_FILE_LOCATION, 'w+') as outfile:
|
||||
@ -78,13 +83,15 @@ class Authetication:
|
||||
|
||||
private_key = RSA.generate(2048)
|
||||
public_key = private_key.publickey()
|
||||
private_key_value = str(private_key.export_key('DER', passphrase=user_passphrase, pkcs=8))
|
||||
public_key_value = str(public_key.export_key('DER', pkcs=8))
|
||||
private_key_value = bytes.hex(private_key.exportKey('DER', passphrase=user_passphrase, pkcs=8))
|
||||
public_key_value = bytes.hex(public_key.exportKey('DER', pkcs=8))
|
||||
|
||||
|
||||
##Save private key in separate file
|
||||
user_privatekey = {'passphrase': user_passphrase, 'privateClientKey': private_key_value, 'publicServerKey': public_server_key}
|
||||
with open(self.PRIVATE_KEY_DIRECTORY_LOCATION + os.path.sep + str(data['index']) + '.txt', 'w+') as outfile:
|
||||
json.dump(user_privatekey, outfile)
|
||||
outfile.close()
|
||||
|
||||
##Save public key in users
|
||||
for user in data['user']:
|
||||
@ -93,11 +100,15 @@ class Authetication:
|
||||
with open(self.CONFIG_FILE_LOCATION, 'w') as outfile:
|
||||
json.dump(data, outfile)
|
||||
break
|
||||
outfile.close()
|
||||
return True
|
||||
else:
|
||||
|
||||
return False
|
||||
|
||||
|
||||
|
||||
|
||||
def saveUser(self, username: str, password: str) -> bool:
|
||||
bytePass = password.encode('utf-8')
|
||||
b64pwd = b64encode(SHA256.new(bytePass).digest())
|
||||
|
Loading…
Reference in New Issue
Block a user