mirror of https://github.com/nucypher/nucypher.git
114 lines
3.5 KiB
Python
114 lines
3.5 KiB
Python
import os
|
|
|
|
from nucypher_core.ferveo import DkgPublicKey
|
|
|
|
from nucypher.blockchain.eth import domains
|
|
from nucypher.blockchain.eth.agents import CoordinatorAgent
|
|
from nucypher.blockchain.eth.registry import ContractRegistry
|
|
from nucypher.blockchain.eth.signers import InMemorySigner
|
|
from nucypher.characters.lawful import Bob, Enrico
|
|
from nucypher.policy.conditions.lingo import ConditionLingo
|
|
from nucypher.utilities.logging import GlobalLoggerSettings
|
|
from tests.constants import DEFAULT_TEST_ENRICO_PRIVATE_KEY
|
|
|
|
######################
|
|
# Boring setup stuff #
|
|
######################
|
|
|
|
LOG_LEVEL = "info"
|
|
GlobalLoggerSettings.set_log_level(log_level_name=LOG_LEVEL)
|
|
GlobalLoggerSettings.start_console_logging()
|
|
|
|
eth_endpoint = os.environ["DEMO_L1_PROVIDER_URI"]
|
|
taco_domain = domains.from_domain_name("lynx")
|
|
|
|
polygon_endpoint = os.environ["DEMO_L2_PROVIDER_URI"]
|
|
|
|
###############
|
|
# Enrico
|
|
###############
|
|
|
|
print("--------- Threshold Encryption ---------")
|
|
|
|
registry = ContractRegistry.from_latest_publication(
|
|
domain=taco_domain.name,
|
|
)
|
|
|
|
coordinator_agent = CoordinatorAgent(
|
|
blockchain_endpoint=polygon_endpoint,
|
|
registry=registry,
|
|
)
|
|
ritual_id = 1 # got this from a side channel
|
|
ritual = coordinator_agent.get_ritual(ritual_id)
|
|
signer = InMemorySigner(private_key=DEFAULT_TEST_ENRICO_PRIVATE_KEY)
|
|
enrico = Enrico(
|
|
encrypting_key=DkgPublicKey.from_bytes(bytes(ritual.public_key)), signer=signer
|
|
)
|
|
|
|
print(
|
|
f"Fetched DKG public key {bytes(enrico.policy_pubkey).hex()} "
|
|
f"for ritual #{ritual_id} "
|
|
f"from Coordinator {coordinator_agent.contract.address}"
|
|
)
|
|
|
|
conditions = {
|
|
"version": ConditionLingo.VERSION,
|
|
"condition": {
|
|
"conditionType": "compound",
|
|
"operator": "and",
|
|
"operands": [
|
|
{
|
|
"conditionType": "rpc",
|
|
"chain": 1,
|
|
"method": "eth_getBalance",
|
|
"parameters": ["0x210eeAC07542F815ebB6FD6689637D8cA2689392", "latest"],
|
|
"returnValueTest": {"comparator": "==", "value": 0},
|
|
},
|
|
{
|
|
"conditionType": "rpc",
|
|
"chain": 137,
|
|
"method": "eth_getBalance",
|
|
"parameters": ["0x210eeAC07542F815ebB6FD6689637D8cA2689392", "latest"],
|
|
"returnValueTest": {"comparator": "==", "value": 0},
|
|
},
|
|
{
|
|
"conditionType": "rpc",
|
|
"chain": 5,
|
|
"method": "eth_getBalance",
|
|
"parameters": ["0x210eeAC07542F815ebB6FD6689637D8cA2689392", "latest"],
|
|
"returnValueTest": {"comparator": ">", "value": 1},
|
|
},
|
|
{
|
|
"conditionType": "rpc",
|
|
"chain": 80001,
|
|
"method": "eth_getBalance",
|
|
"parameters": ["0x210eeAC07542F815ebB6FD6689637D8cA2689392", "latest"],
|
|
"returnValueTest": {"comparator": "==", "value": 0},
|
|
},
|
|
],
|
|
},
|
|
}
|
|
|
|
message = "hello world".encode()
|
|
threshold_message_kit = enrico.encrypt_for_dkg(plaintext=message, conditions=conditions)
|
|
|
|
print(f"\nEncrypted message:\n{bytes(threshold_message_kit).hex()}")
|
|
|
|
###############
|
|
# Bob
|
|
###############
|
|
print("--------- Threshold Decryption ---------")
|
|
|
|
bob = Bob(
|
|
domain=taco_domain.name,
|
|
eth_endpoint=eth_endpoint,
|
|
polygon_endpoint=polygon_endpoint,
|
|
registry=registry,
|
|
)
|
|
|
|
bob.start_learning_loop(now=True)
|
|
|
|
cleartext = bob.threshold_decrypt(threshold_message_kit=threshold_message_kit)
|
|
|
|
print(f"\nCleartext:{bytes(cleartext).decode()}")
|