You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
156 lines
4.7 KiB
Python
156 lines
4.7 KiB
Python
import os,sys; sys.path.append(os.path.abspath(os.path.join(os.path.abspath(os.path.join(os.path.dirname(__file__),'..')),'..')))
|
|
from komrade import *
|
|
from komrade.backend import *
|
|
|
|
|
|
|
|
|
|
def create_phonelines():
|
|
# crypt
|
|
keycrypt = Crypt(fn=PATH_CRYPT_OP_KEYS)
|
|
|
|
# Operator
|
|
op_keypair = KomradeAsymmetricKey()
|
|
op_pubkey,op_privkey = op_keypair.pubkey_obj,op_keypair.privkey_obj
|
|
op_privkey_decr = KomradeSymmetricKeyWithPassphrase()
|
|
op_privkey_encr_b = op_privkey_decr.encrypt(op_privkey.data)
|
|
op_privkey_encr = KomradeEncryptedAsymmetricPrivateKey(op_privkey_encr_b)
|
|
|
|
# save pub and priv
|
|
op_uri = op_pubkey.data_b64
|
|
keycrypt.set(OPERATOR_NAME,op_pubkey.data,prefix='/pubkey/')
|
|
keycrypt.set(op_uri,op_privkey_encr.data,prefix='/privkey_encr/')
|
|
|
|
|
|
## Telephone
|
|
phone_keypair = KomradeAsymmetricKey()
|
|
phone_pubkey,phone_privkey = phone_keypair.pubkey_obj,phone_keypair.privkey_obj
|
|
phone_privkey_decr = KomradeSymmetricKeyWithoutPassphrase()
|
|
phone_privkey_encr_b = phone_privkey_decr.encrypt(phone_privkey.data)
|
|
phone_privkey_encr = KomradeEncryptedAsymmetricPrivateKey(phone_privkey_encr_b)
|
|
|
|
# save pub and priv
|
|
keycrypt.set(TELEPHONE_NAME,phone_pubkey.data,prefix='/pubkey/')
|
|
|
|
## world
|
|
world_keypair = KomradeAsymmetricKey()
|
|
world_pubkey,world_privkey = world_keypair.pubkey_obj,world_keypair.privkey_obj
|
|
world_privkey_decr = KomradeSymmetricKeyWithPassphrase()
|
|
world_privkey_encr_b = world_privkey_decr.encrypt(world_privkey.data)
|
|
world_privkey_encr = KomradeEncryptedAsymmetricPrivateKey(world_privkey_encr_b)
|
|
|
|
# save pub and priv
|
|
world_uri = world_pubkey.data_b64
|
|
keycrypt.set(WORLD_NAME,world_pubkey.data,prefix='/pubkey/')
|
|
keycrypt.set(world_uri,world_privkey_encr.data,prefix='/privkey_encr/')
|
|
|
|
|
|
|
|
|
|
# keys to save built in
|
|
builtin_keys = {
|
|
OPERATOR_NAME: {
|
|
'pubkey':op_pubkey.data,
|
|
},
|
|
TELEPHONE_NAME: {
|
|
'pubkey':phone_pubkey.data,
|
|
'privkey':phone_privkey.data,
|
|
},
|
|
WORLD_NAME: {
|
|
'pubkey':world_pubkey.data,
|
|
}
|
|
}
|
|
|
|
import pickle
|
|
from base64 import b64encode
|
|
builtin_keys_b = pickle.dumps(builtin_keys)
|
|
|
|
omega_key = KomradeSymmetricKeyWithoutPassphrase()
|
|
builtin_keys_b_encr = omega_key.encrypt(builtin_keys_b)
|
|
builtin_keys_b_encr_with_key = omega_key.data + BSEP + builtin_keys_b_encr
|
|
builtin_keys_b_encr_with_key_b64 = b64encode(builtin_keys_b_encr_with_key)
|
|
|
|
with open(PATH_BUILTIN_KEYCHAIN,'wb') as of:
|
|
of.write(builtin_keys_b_encr_with_key_b64)
|
|
print('>> saved:',PATH_BUILTIN_KEYCHAIN)
|
|
print(builtin_keys_b_encr_with_key_b64)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def create_secret():
|
|
if not os.path.exists(PATH_CRYPT_SECRET):
|
|
secret = get_random_binary_id()
|
|
from komrade.backend.keymaker import make_key_discreet
|
|
print('shhh! creating secret:',make_key_discreet(secret))
|
|
with open(PATH_CRYPT_SECRET,'wb') as of:
|
|
of.write(secret)
|
|
|
|
|
|
|
|
def check_phonelines():
|
|
# if needed
|
|
create_secret()
|
|
|
|
# is world there?
|
|
keycrypt = Crypt(fn=PATH_CRYPT_OP_KEYS)
|
|
|
|
# builtins
|
|
with open(PATH_BUILTIN_KEYCHAIN,'rb') as f:
|
|
builtin_keys_encr_b64 = f.read()
|
|
builtin_keys_encr = b64decode(builtin_keys_encr_b64)
|
|
omega_key_b,builtin_keys_encr = builtin_keys_encr.split(BSEP)
|
|
omega_key = KomradeSymmetricKeyWithoutPassphrase(omega_key_b)
|
|
builtin_keys_b = omega_key.decrypt(builtin_keys_encr)
|
|
builtin_keys = pickle.loads(builtin_keys_b)
|
|
# print(builtin_keys)
|
|
|
|
for keyring in builtin_keys:
|
|
name = keyring.get('name')
|
|
keychain = dict((k,v) for k,v in keyring.items() if k!='name')
|
|
print(name,keychain)
|
|
if not 'pubkey' in keyring: continue
|
|
uri = b64encode(keychain.get('pubkey')
|
|
|
|
if not keycrypt.has(name,prefix='/pubkey/')
|
|
keycrypt.set(name,keychain['pubkey'],prefix='/pubkey/')
|
|
|
|
for key in [k for k in keychain if k!='pubkey']:
|
|
if not keycrypt.has(uri,prefix=f'/{key}/'):
|
|
keycrypt.set(uri,keychain[key],prefix=f'/{key}/')
|
|
|
|
# make sure world's qr is there too
|
|
if name==WORLD_NAME:
|
|
import pyqrcode
|
|
qr = pyqrcode.create(uri)
|
|
ofnfn = os.path.join(PATH_QRCODES,name+'.png')
|
|
qr.png(ofnfn,scale=5)
|
|
# print('>> saved:',ofnfn)
|
|
|
|
|
|
return builtin_keys
|
|
|
|
|
|
def test_phonelines():
|
|
from komrade.backend.the_telephone import TheTelephone
|
|
from komrade.backend.the_operator import TheOperator
|
|
|
|
phone = TheTelephone()
|
|
op = TheOperator()
|
|
|
|
print('phone',dict_format(phone._keychain))
|
|
print('op',dict_format(op._keychain))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
test_phonelines() |