1: config init in separate file
All checks were successful
continuous-integration/drone/push Build is passing
All checks were successful
continuous-integration/drone/push Build is passing
2: auth_test doesnt generate production state
This commit is contained in:
44
server/config_init.py
Normal file
44
server/config_init.py
Normal file
@@ -0,0 +1,44 @@
|
||||
from authentication import Authetication
|
||||
from Crypto.PublicKey import RSA
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
|
||||
|
||||
def generatePrivateKeyForUser(auth: Authetication, username: str, user_passphrase: str, public_server_key='') -> bool:
|
||||
if auth.checkUserExists(username):
|
||||
with open(auth.CONFIG_FILE_LOCATION) as json_file:
|
||||
data = json.load(json_file)
|
||||
|
||||
private_key = RSA.generate(2048)
|
||||
public_key = private_key.publickey()
|
||||
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(auth.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']:
|
||||
if username == user['username']:
|
||||
user['publicKey'] = public_key_value
|
||||
with open(auth.CONFIG_FILE_LOCATION, 'w') as outfile:
|
||||
json.dump(data, outfile)
|
||||
break
|
||||
outfile.close()
|
||||
return True
|
||||
else:
|
||||
return False
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
auth = Authetication()
|
||||
auth.initConfig()
|
||||
auth.saveUser('alma' ,'alma')
|
||||
generatePrivateKeyForUser('alma', 'amla')
|
||||
auth.saveUser('citrom' ,'citrom')
|
||||
generatePrivateKeyForUser('citrom', 'mortic')
|
||||
Reference in New Issue
Block a user