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.
Comrad/komrade/backend/the_operator.py

152 lines
4.1 KiB
Python

"""
There is only one operator!
Running on node prime.
"""
# internal imports
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 *
# print(PATH_OPERATOR_WEB_KEYS_URL)
class TheOperator(Operator):
"""
The remote operator
"""
@property
def phone(self):
global TELEPHONE
from komrade.backend.the_telephone import TheTelephone
if not TELEPHONE: TELEPHONE=TheTelephone()
return TELEPHONE
def __init__(self, name = OPERATOR_NAME, passphrase='acc'):
"""
Boot up the operator. Requires knowing or setting a password of memory.
"""
super().__init__(
name,
passphrase,
path_crypt_keys=PATH_CRYPT_OP_KEYS,
path_crypt_data=PATH_CRYPT_OP_DATA
)
self._keychain = self.operator_keychain
def ring(self,
from_caller=None,
to_caller=None,
json_phone2phone={},
json_caller2phone={}, # (person) -> operator or operator -> (person)
json_caller2caller={}):
encr_msg_to_send = super().ring(
from_phone=self,
to_phone=self.phone,
from_caller=from_caller,
to_caller=to_caller,
json_phone2phone=json_phone2phone,
json_caller2phone=json_caller2phone, # (person) -> operator
json_caller2caller=json_caller2caller)
return self.send(encr_msg_to_send)
# ends the ring_ring() chain
def answer_phone(self,data_b64_str):
# route incoming call from the switchboard
self.log('Hello, this is the Operator. You said: ',data_b64_str)
# decode
data_b64 = data_b64_str.encode()
data = b64decode(data_b64)
msg_d = msg_encr_caller2caller_caller2phone_phone2phone = data
self.log('msg_encr_caller2caller_caller2phone_phone2phone incoming',msg_encr_caller2caller_caller2phone_phone2phone)
# make top-level message object, addressed to me the operator
from komrade.backend.messages import Message
msg_obj = Message(msg_d,caller=self.phone,callee=self)
msg_obj.decrypt()
# route msg back to caller
return self.route(msg_obj)
def find_pubkey(self):
return self.operator_keychain['pubkey']
def send(self,encr_data_b):
self.log(type(encr_data_b),encr_data_b,'sending!')
return encr_data_b
def route(self, msg_obj):
# get route from msg
route = msg_obj.route
# no route?
if not msg_obj.route: raise KomradeException('no route!')
# what we working with?
self.log(f'route() got incoming msg_d = {msg_d}, _msg = {_msg}, and route = {route}')
# hard code the acceptable routes
if route == 'forge_new_keys':
return self.forge_new_keys(msg)
# otherwise, hang up and try again
return OPERATOR_INTERCEPT_MESSAGE
def forge_new_keys(self,msg_obj):
data = msg_obj.msg
self.log('about to make some new keys!',data)
# get keys
forged_keys_plus_id = super().forge_new_keys(**data)
# return to Telephone/Caller
return self.ring(json_phone2phone=forged_keys_plus_id)
def test_op():
from komrade.backend.the_telephone import TheTelephone
op = TheOperator()
# op.boot()
keychain_op = op.keychain(force=True)
phone = TheTelephone()
# phone.boot()
keychain_ph = phone.keychain(force=True)
from pprint import pprint
print('REASSEMBLED OPERATOR KEYCHAIN')
pprint(keychain_op)
# stop
print('REASSEMBLED TELEPHONE KEYCHAIN')
pprint(keychain_ph)
# print(op.pubkey(keychain=keychain))
# print(op.crypt_keys.get(op.pubkey(), prefix='/privkey_encr/'))
# print(op.crypt_keys.get(op.name, prefix='/pubkey_encr/'))
# print(op.pubkey_)
# stop
# pubkey = op.keychain()['pubkey']
# pubkey_b64 = b64encode(pubkey)
# print(pubkey)
if __name__ == '__main__': test_op()