Nucypher
/
nucypher
mirror of https://github.com/nucypher/nucypher.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Rethinking PolicyOffer more broadly - it's now a Contract, which is able to formulate its own payload.

pull/145/head
jMyles 2017-12-14 12:15:10 -08:00
parent 021f7a25d5
commit 54d593f536
1 changed files with 19 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
nkms/policy/models.py
View File

@ -18,27 +18,32 @@ group_payload_splitter = BytestringSplitter(PublicKey)
policy_payload_splitter = BytestringSplitter(KFrag) policy_payload_splitter = BytestringSplitter(KFrag)
class PolicyOffer(object): class Contract(object):
""" """
An offer from Alice to Ursula to enter into a contract for Re-Encryption services. A Policy must be implemented by agreement with n Ursulas. This class tracks the status of that implementation.
""" """
def __init__(self, n, deposit, contract_end_datetime): def __init__(self, policy, deposit, expiration):
""" """
:param n: The total number of Policies which Alice wishes to create. :param deposit: Funds which will pay for the timeframe of this Contract (not the actual re-encryptions);
:param deposit: Funds which will pay for the timeframe of the contract (not the actual re-encryptions);
a portion will be locked for each Ursula that accepts. a portion will be locked for each Ursula that accepts.
:param contract_end_datetime: The moment which Alice wants the contract to end. :param expiration: The moment which Alice wants the Contract to end.
""" """
self.n = n self.policy = policy
self.deposit = deposit self.deposit = deposit
self.contract_end_datetime = contract_end_datetime self.expiration = expiration
def activate(self, kfrag, ursula, negotiation_result): def activate(self, kfrag, ursula, negotiation_result):
self.kfrag = kfrag self.kfrag = kfrag
self.ursula = ursula self.ursula = ursula
self.negotiation_result = negotiation_result self.negotiation_result = negotiation_result
def encrypt_payload_for_ursula(self):
"""
Craft an offer to send to Ursula.
"""
return self.alice.encrypt_for(self.ursula, self.payload())[0] # We don't need the signature separately.
class PolicyOfferResponse(object): class PolicyOfferResponse(object):
pass pass
@ -201,7 +206,7 @@ class Policy(object):
self.challenge_size = challenge_size self.challenge_size = challenge_size
self.treasure_map = [] self.treasure_map = []
self.challenge_pack = [] self.challenge_pack = []
self._active_ursulas = {} self._active_contracts = {}
self._encrypted_challenge_pack = encrypted_challenge_pack self._encrypted_challenge_pack = encrypted_challenge_pack
@ -265,26 +270,20 @@ class Policy(object):
self._encrypted_challenge_pack = self.alice.encrypt_for(self.bob, msgpack.dumps(self.challenge_pack)) self._encrypted_challenge_pack = self.alice.encrypt_for(self.bob, msgpack.dumps(self.challenge_pack))
return self._encrypted_challenge_pack return self._encrypted_challenge_pack
def encrypt_payload_for_ursula(self):
"""
Craft an offer to send to Ursula.
"""
return self.alice.encrypt_for(self.ursula, self.payload())[0] # We don't need the signature separately.
def craft_offer(self, deposit, expiration): def craft_offer(self, deposit, expiration):
return PolicyOffer(self.n, deposit, expiration) return Contract(self, deposit, expiration)
def find_n_ursulas(self, networky_stuff, offer: PolicyOffer): def find_n_ursulas(self, networky_stuff, contract: Contract):
""" """
:param networky_stuff: A compliant interface (maybe a Client instance) to be used to engage the DHT swarm. :param networky_stuff: A compliant interface (maybe a Client instance) to be used to engage the DHT swarm.
""" """
for kfrag in self.kfrags: for kfrag in self.kfrags:
try: try:
ursula, result = networky_stuff.find_ursula(offer) ursula, result = networky_stuff.find_ursula(contract)
# TODO: Here, we need to assess the result and see if we're actually good to go. # TODO: Here, we need to assess the result and see if we're actually good to go.
if result.was_accepted: if result.was_accepted:
offer.activate(kfrag, ursula, result) contract.activate(kfrag, ursula, result)
self._active_ursulas[kfrag] = offer self._active_contracts[kfrag] = contract
except networky_stuff.NotEnoughQualifiedUrsulas: except networky_stuff.NotEnoughQualifiedUrsulas:
pass # TODO: Tell Alice to either wait or lower the value of n. pass # TODO: Tell Alice to either wait or lower the value of n.