You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
146 lines
4.5 KiB
146 lines
4.5 KiB
import os
|
|
import multiprocessing
|
|
import importlib
|
|
import tempfile
|
|
from collections import OrderedDict
|
|
|
|
# per-instance globals
|
|
pEp = None
|
|
handler = None
|
|
own_addresses = []
|
|
indent = 0
|
|
i_name = ""
|
|
|
|
def create_account(address, name):
|
|
global own_addresses
|
|
i = pEp.Identity(address, name)
|
|
pEp.myself(i)
|
|
own_addresses.append(address)
|
|
|
|
def printi(*args):
|
|
global indent
|
|
print(i_name + ">" * indent, *args)
|
|
|
|
def printheader(blah=None):
|
|
global indent
|
|
if blah is None:
|
|
printi("-" * 80)
|
|
indent = indent - 1
|
|
else:
|
|
indent = indent + 1
|
|
printi("-" * (39 - int(len(blah)/2)) +
|
|
" " + blah + " " +
|
|
"-" * (39 - len(blah) + int(len(blah)/2)))
|
|
|
|
def printmsg(msg):
|
|
printi("from :", msg.from_)
|
|
printi("to :", msg.to)
|
|
printi("short :", msg.shortmsg)
|
|
printi("opt_fields :", msg.opt_fields)
|
|
lng = msg.longmsg.splitlines()
|
|
lngcut = lng[:20]+["[...]"] if len(lng)>20 else lng
|
|
pfx = "long : "
|
|
for l in lngcut :
|
|
printi(pfx + l)
|
|
pfx = " "
|
|
printi("attachments : ", msg.attachments)
|
|
|
|
def pEp_instance_run(iname, conn, msgs_folders):
|
|
global pEp, handler, own_addresses, i_name
|
|
|
|
i_name = iname
|
|
|
|
pEp = importlib.import_module("pEp")
|
|
|
|
class Handler(pEp.SyncMixIn):
|
|
def messageToSend(self, msg):
|
|
printheader("SYNC MESSAGE to send")
|
|
printmsg(msg)
|
|
printheader()
|
|
for rcpt in msg.to + msg.cc + msg.bcc:
|
|
# list inside dict from MP manager are not proxified.
|
|
msgs = msgs_folders.get(rcpt.address,[])
|
|
msgs.append(str(msg))
|
|
msgs_folders[rcpt.address] = msgs
|
|
|
|
def showHandshake(self, me, partner):
|
|
printheader("show HANDSHAKE dialog")
|
|
printi("handshake needed between " + repr(me) + " and " + repr(partner))
|
|
printheader()
|
|
# TODO: accept or reject
|
|
|
|
handler = Handler()
|
|
|
|
while True:
|
|
order = conn.recv()
|
|
if order is None:
|
|
break
|
|
|
|
func = order[0]
|
|
|
|
printheader("DECRYPT messages")
|
|
# decrypt every non-consumed message for all instance accounts
|
|
for own_address in own_addresses:
|
|
msgs_for_me = msgs_folders[own_address]
|
|
for msgstr in msgs_for_me:
|
|
msg = pEp.incoming_message(msgstr)
|
|
printi("--- decrypt()")
|
|
printmsg(msg)
|
|
msg2, keys, rating, consumed, flags = msg.decrypt()
|
|
|
|
if consumed: #PEP_MESSAGE_CONSUMED
|
|
printi("--- PEP_MESSAGE_CONSUMED")
|
|
# folder may have changed in the meantime,
|
|
# remove item directly from latest version of it.
|
|
folder = msgs_folders[own_address]
|
|
folder.remove(msgstr)
|
|
msgs_folders[own_address] = folder
|
|
else :
|
|
printi("->-")
|
|
printmsg(msg2)
|
|
printi("---")
|
|
printheader()
|
|
|
|
res = None
|
|
if func is not None:
|
|
printheader("Executing function " + func.__name__)
|
|
args, kwargs = order[1:] + [[], {}][len(order) - 1:]
|
|
printi("args :", args)
|
|
printi("kwargs :", kwargs)
|
|
res = func(*args,**kwargs)
|
|
printi("function " + func.__name__ + " returned :", res)
|
|
printheader()
|
|
|
|
conn.send(res)
|
|
|
|
def pEp_instance_main(iname, conn, msgs_folders):
|
|
# run with a dispensable $HOME to get fresh DB and PGP keyrings
|
|
with tempfile.TemporaryDirectory() as tmpdirname:
|
|
print("Instance " + iname + " runs into " + tmpdirname)
|
|
os.environ['HOME'] = tmpdirname
|
|
pEp_instance_run(iname, conn, msgs_folders)
|
|
print(iname + " exiting")
|
|
|
|
def run_scenario(scenario):
|
|
instances = OrderedDict()
|
|
with multiprocessing.Manager() as manager:
|
|
msgs_folders = manager.dict()
|
|
for iname, order in scenario:
|
|
if iname not in instances:
|
|
conn, child_conn = multiprocessing.Pipe()
|
|
proc = multiprocessing.Process(
|
|
target=pEp_instance_main,
|
|
args=(iname,child_conn,msgs_folders))
|
|
proc.start()
|
|
instances[iname] = (proc, conn)
|
|
else:
|
|
proc, conn = instances[iname]
|
|
|
|
conn.send(order)
|
|
res = conn.recv()
|
|
|
|
for iname, (proc, conn) in instances.items():
|
|
# tell process to terminate
|
|
conn.send(None)
|
|
proc.join()
|
|
|
|
|