def load_contract(state, contract_file) :
try :
data_directory = state.get(['Contract', 'DataDirectory'])
ledger_config = state.get(['Ledger'])
return Contract.read_from_file(ledger_config, contract_file, data_dir=data_directory)
except Exception as e :
raise Exception('unable to load the contract; {0}'.format(str(e)))
def command_create(state, bindings, pargs):
"""controller command to create a contract
"""
interp = os.environ.get("PDO_INTERPRETER", "gipsy")
parser = argparse.ArgumentParser(prog='create')
parser.add_argument('-c',
'--contract-class',
help='Name of the contract class',
required=True,
type=str)
parser.add_argument(
'-i',
'--interpreter',
help='Name of the interpreter used to evaluate the contract',
default=interp)
parser.add_argument('-s',
'--contract-source',
help='File that contains contract source code',
required=True,
type=str)
parser.add_argument('-p',
'--pservice-group',
help='Name of the provisioning service group to use',
default="default")
parser.add_argument('-e',
'--eservice-group',
help='Name of the enclave service group to use',
default="default")
parser.add_argument('-f',
'--save-file',
help='File where contract data is stored',
type=str)
parser.add_argument('--symbol', help='binding symbol for result', type=str)
options = parser.parse_args(pargs)
contract_class = options.contract_class
contract_source = options.contract_source
# ---------- load the invoker's keys ----------
try:
keyfile = state.get(['Key', 'FileName'])
keypath = state.get(['Key', 'SearchPath'])
client_keys = ServiceKeys.read_from_file(keyfile, keypath)
except Exception as e:
raise Exception('unable to load client keys; {0}'.format(str(e)))
# ---------- read the contract source code ----------
try:
source_path = state.get(['Contract', 'SourceSearchPath'])
contract_code = ContractCode.create_from_file(
contract_class,
contract_source,
source_path,
interpreter=options.interpreter)
except Exception as e:
raise Exception('unable to load contract source; {0}'.format(str(e)))
logger.debug('Loaded contract code for %s', contract_class)
# ---------- set up the enclave clients ----------
eservice_clients = get_eservice_list(state, options.eservice_group)
if len(eservice_clients) == 0:
raise Exception('unable to locate enclave services in the group %s',
options.eservice_group)
preferred_eservice_client = get_eservice(
state, eservice_group=options.eservice_group)
if preferred_eservice_client.interpreter != options.interpreter:
raise Exception(
'enclave interpreter does not match requested contract interpreter %s',
options.interpreter)
# ---------- set up the provisioning service clients ----------
pservice_clients = get_pservice_list(state, options.pservice_group)
if len(pservice_clients) == 0:
raise Exception(
'unable to locate provisioning services in the group %s',
options.pservice_group)
# ---------- register contract ----------
data_directory = state.get(['Contract', 'DataDirectory'])
ledger_config = state.get(['Sawtooth'])
try:
provisioning_service_keys = [pc.identity for pc in pservice_clients]
contract_id = register_contract(ledger_config, client_keys,
contract_code,
provisioning_service_keys)
logger.debug('Registered contract with class %s and id %s',
contract_class, contract_id)
contract_state = ContractState.create_new_state(contract_id)
contract = Contract(contract_code, contract_state, contract_id,
client_keys.identity)
# must fix this later
contract.extra_data[
'preferred-enclave'] = preferred_eservice_client.enclave_id
contract_file = "{0}_{1}.pdo".format(contract_class, contract.short_id)
if options.save_file:
contract_file = options.save_file
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e:
raise Exception('failed to register the contract; {0}'.format(str(e)))
# provision the encryption keys to all of the enclaves
try:
encrypted_state_encryption_keys = __add_enclave_secrets(
ledger_config, contract.contract_id, client_keys, eservice_clients,
pservice_clients)
for enclave_id in encrypted_state_encryption_keys:
encrypted_key = encrypted_state_encryption_keys[enclave_id]
contract.set_state_encryption_key(enclave_id, encrypted_key)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e:
raise Exception('failed to provisioning the enclaves; {0}'.format(
str(e)))
# create the initial contract state
try:
__create_contract(ledger_config, client_keys,
preferred_eservice_client, eservice_clients,
contract)
contract.contract_state.save_to_cache(data_dir=data_directory)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e:
raise Exception(
'failed to create the initial contract state; {0}'.format(str(e)))
if contract_id and options.symbol:
bindings.bind(options.symbol, contract_id)
def LocalMain(config, message):
# ---------- check the integrity of the configuration ----------
try:
ledger_config = config['Sawtooth']
contract_config = config['Contract']
service_config = config['Service']
key_config = config['Key']
except KeyError as ke:
logger.error('missing configuration section %s', str(ke))
sys.exit(-1)
# ---------- load the contract information file ----------
try:
save_file = contract_config['SaveFile']
data_directory = contract_config['DataDirectory']
logger.info('load contract from %s', save_file)
contract = Contract.read_from_file(ledger_config,
save_file,
data_dir=data_directory)
except KeyError as ke:
logger.error('missing configuration parameter %s', str(ke))
sys.exit(-1)
except Exception as e:
logger.error('failed to load the contract; %s', str(e))
sys.exit(-1)
# ---------- load the invoker's keys ----------
try:
keyfile = key_config['FileName']
keypath = key_config['SearchPath']
contract_invoker_keys = ServiceKeys.read_from_file(keyfile, keypath)
except KeyError as ke:
logger.error('missing configuration parameter %s', str(ke))
sys.exit(-1)
except Exception as e:
logger.error('unable to load client keys; %s', str(e))
sys.exit(-1)
# ---------- set up the enclave service ----------
try:
enclave_url = service_config['PreferredEnclaveService']
enclave_client = EnclaveServiceClient(enclave_url)
except KeyError as ke:
logger.error('missing configuration parameter %s', str(ke))
sys.exit(-1)
except Exception as e:
logger.error('unable to connect to enclave service; %s', str(e))
sys.exit(-1)
try:
# this is just a sanity check to make sure the selected enclave
# has actually been provisioned
contract.get_state_encryption_key(enclave_client.enclave_id)
except KeyError as ke:
logger.error('selected enclave is not provisioned')
sys.exit(-1)
# ---------- process incoming messages ----------
if message:
mlist = MessageIterator(message)
else:
mlist = InputIterator(config.get('Identity', '') + "> ")
for msg in mlist:
if not msg:
continue
logger.info('send message <%s> to contract', msg)
try:
update_request = contract.create_update_request(
contract_invoker_keys, enclave_client, msg)
update_response = update_request.evaluate()
if update_response.status:
print(update_response.result)
else:
print('ERROR: {}'.format(update_response.result))
# continue if this is an interactive session, fail
# if we are processing command line messages
if message:
sys.exit(-1)
else:
continue
except Exception as e:
logger.error('enclave failed to evaluation expression; %s', str(e))
sys.exit(-1)
# if this operation did not change state then there is nothing
# to send to the ledger or to save
if not update_response.state_changed:
continue
try:
logger.debug("sending to ledger")
txnid = update_response.submit_update_transaction(ledger_config)
except Exception as e:
logger.error('failed to save the new state; %s', str(e))
sys.exit(-1)
contract.set_state(update_response.encrypted_state)
contract.contract_state.save_to_cache(data_dir=data_directory)
sys.exit(0)
def LocalMain(commands, config):
# ---------- load the contract ----------
try:
ledger_config = config['Sawtooth']
contract_config = config['Contract']
service_config = config['Service']
key_config = config['Key']
except KeyError as ke:
logger.error('missing configuration section %s', str(ke))
sys.exit(-1)
# ---------- load the invoker's keys ----------
try:
keyfile = key_config['FileName']
keypath = key_config['SearchPath']
client_keys = ServiceKeys.read_from_file(keyfile, keypath)
except KeyError as ke:
logger.error('missing configuration parameter %s', str(ke))
sys.exit(-1)
except Exception as e:
logger.error('unable to load client keys; %s', str(e))
sys.exit(-1)
# ---------- read the contract source code ----------
try:
contract_name = contract_config['Name']
data_directory = contract_config['DataDirectory']
save_file = contract_config['SaveFile']
source_file = contract_config['SourceFile']
source_path = contract_config['SourceSearchPath']
contract_code = ContractCode.create_from_scheme_file(
contract_name, source_file, source_path)
except KeyError as ke:
logger.error('missing configuration parameter %s', str(ke))
sys.exit(-1)
except Exception as e:
logger.error('unable to load contract source; %s', str(e))
sys.exit(-1)
logger.info('Loaded contract data for %s', contract_name)
# ---------- set up the enclave clients ----------
try:
enclaveclients = []
for url in service_config['EnclaveServiceURLs']:
enclaveclients.append(EnclaveServiceClient(url))
except Exception as e:
logger.error('unable to setup enclave services; %s', str(e))
sys.exit(-1)
# ---------- set up the provisioning service clients ----------
# This is a dictionary of provisioning service public key : client pairs
try:
provclients = []
for url in service_config['ProvisioningServiceURLs']:
provclients.append(ProvisioningServiceClient(url))
except Exception as e:
logger.error('unable to setup provisioning services; %s', str(e))
sys.exit(-1)
logger.debug("All enclaveclients: %s", enclaveclients)
logger.debug("All provclients: %s", provclients)
# process the commands to create & register the contract
if 'register' in commands:
try:
provisioning_service_keys = [pc.identity for pc in provclients]
contract_id = register_contract(ledger_config, client_keys,
contract_code,
provisioning_service_keys)
logger.info('Registered contract %s with id %s', contract_name,
contract_id)
contract_state = ContractState.create_new_state(contract_id)
contract = Contract(contract_code, contract_state, contract_id,
client_keys.identity)
contract.save_to_file(save_file, data_dir=data_directory)
except Exception as e:
logger.error('failed to register the contract; %s', str(e))
sys.exit(-1)
else:
# need to read the contract from the contract file
contract = Contract.read_from_file(ledger_config, contract_name,
data_directory)
if 'addenclave' in commands:
encrypted_state_encryption_keys = AddEnclaveSecrets(
ledger_config, contract.contract_id, client_keys, enclaveclients,
provclients)
for enclave_id in encrypted_state_encryption_keys:
encrypted_key = encrypted_state_encryption_keys[enclave_id]
contract.set_state_encryption_key(enclave_id, encrypted_key)
contract.save_to_file(save_file, data_dir=data_directory)
logger.info(
'Successfully added enclave secrets to ledger for contract %s',
contract_code.name)
if 'create' in commands:
CreateContract(ledger_config, client_keys, enclaveclients, contract)
contract.contract_state.save_to_cache(data_dir=data_directory)
contract.save_to_file(save_file, data_dir=data_directory)
print('export CONTRACTID={0}'.format(contract.contract_id))
def command_create(state, bindings, pargs):
"""controller command to create a contract
"""
default_interpreter = state.get(['Contract', 'Interpreter'])
default_replicas = state.get(['Replication', 'NumProvableReplicas'], 2)
default_duration = state.get(['Replication', 'Duration'], 120)
parser = argparse.ArgumentParser(prog='create')
parser.add_argument('-c',
'--contract-class',
help='Name of the contract class',
required=False,
type=str)
parser.add_argument('-e',
'--eservice-group',
help='Name of the enclave service group to use',
default="default")
parser.add_argument('-f',
'--save-file',
help='File where contract data is stored',
type=str)
parser.add_argument('-i',
'--interpreter',
help='Interpreter used to evaluate the contract',
default=default_interpreter)
parser.add_argument('-p',
'--pservice-group',
help='Name of the provisioning service group to use',
default="default")
parser.add_argument('-s',
'--contract-source',
help='File that contains contract source code',
required=True,
type=str)
parser.add_argument('--symbol', help='binding symbol for result', type=str)
parser.add_argument('--state-replicas',
help='Number of authoritative replicas of the state',
default=default_replicas)
parser.add_argument('--state-duration',
help='Duration required for state replicas',
default=default_duration)
parser.add_argument(
'--extra-data',
help='Simple string that can save extra data with the contract file',
type=str)
options = parser.parse_args(pargs)
contract_class = options.contract_class
contract_source = options.contract_source
# ---------- load the invoker's keys ----------
try:
keyfile = state.private_key_file
keypath = state.get(['Key', 'SearchPath'])
client_keys = ServiceKeys.read_from_file(keyfile, keypath)
except Exception as e:
raise Exception('unable to load client keys; {0}'.format(str(e)))
# ---------- read the contract source code ----------
try:
source_path = state.get(['Contract', 'SourceSearchPath'])
contract_code = ContractCode.create_from_file(
contract_class,
contract_source,
source_path,
interpreter=options.interpreter)
except Exception as e:
raise Exception('unable to load contract source; {0}'.format(str(e)))
logger.debug('Loaded contract code for %s', contract_class)
# ---------- set up the enclave clients ----------
eservice_clients = get_eservice_list(state, options.eservice_group)
if len(eservice_clients) == 0:
raise Exception('unable to locate enclave services in the group %s',
options.eservice_group)
preferred_eservice_client = get_eservice(
state, eservice_group=options.eservice_group)
if preferred_eservice_client.interpreter != options.interpreter:
raise Exception(
'enclave interpreter does not match requested contract interpreter %s',
options.interpreter)
# ---------- set up the provisioning service clients ----------
pservice_clients = get_pservice_list(state, options.pservice_group)
if len(pservice_clients) == 0:
raise Exception(
'unable to locate provisioning services in the group %s',
options.pservice_group)
# ---------- register contract ----------
data_directory = state.get(['Contract', 'DataDirectory'])
ledger_config = state.get(['Ledger'])
try:
extra_params = {
'num_provable_replicas': options.state_replicas,
'availability_duration': options.state_duration,
}
if options.extra_data:
extra_params['extra_data'] = options.extra_data
provisioning_service_keys = [pc.identity for pc in pservice_clients]
contract_id = register_contract(ledger_config, client_keys,
contract_code,
provisioning_service_keys)
logger.debug('Registered contract with class %s and id %s',
contract_class, contract_id)
contract_state = ContractState.create_new_state(contract_id)
contract = Contract(contract_code, contract_state, contract_id,
client_keys.identity, **extra_params)
# must fix this later
contract.extra_data[
'preferred-enclave'] = preferred_eservice_client.enclave_id
contract_file = "{0}_{1}.pdo".format(contract_class, contract.short_id)
if options.save_file:
contract_file = options.save_file
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e:
raise Exception('failed to register the contract; {0}'.format(str(e)))
# provision the encryption keys to all of the enclaves
try:
encrypted_state_encryption_keys = __add_enclave_secrets(
ledger_config, contract.contract_id, client_keys, eservice_clients,
pservice_clients)
for enclave_id in encrypted_state_encryption_keys:
encrypted_key = encrypted_state_encryption_keys[enclave_id]
contract.set_state_encryption_key(enclave_id, encrypted_key)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e:
raise Exception('failed to provisioning the enclaves; {0}'.format(
str(e)))
# create the initial contract state
try:
__create_contract(ledger_config, client_keys,
preferred_eservice_client, eservice_clients,
contract)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e:
raise Exception(
'failed to create the initial contract state; {0}'.format(str(e)))
if contract_id and options.symbol:
bindings.bind(options.symbol, contract_id)
def CreateAndRegisterContract(config, contract_info, creator_keys):
ledger_config = config.get('Sawtooth')
contract_config = config.get('Contract')
contract_creator_id = creator_keys.identity
contract_name = contract_info['Name']
source_file = contract_info['Source']
search_path = contract_config['SourceSearchPath']
contract_code = ContractCode.create_from_scheme_file(
contract_name, source_file, search_path=search_path)
# --------------------------------------------------
logger.info('register the contract')
# --------------------------------------------------
pservice_urls = contract_info.get("ProvisioningServices")
provisioning_services = list(
map(lambda url: ProvisioningServiceClient(url), pservice_urls))
provisioning_service_keys = list(
map(lambda svc: svc.identity, provisioning_services))
contract_id = register_contract(ledger_config, creator_keys, contract_code,
provisioning_service_keys)
logger.info('registered the contract as %s', contract_id)
contract_state = ContractState.create_new_state(contract_id)
contract = Contract(contract_code, contract_state, contract_id,
contract_creator_id)
# --------------------------------------------------
logger.info('provision enclaves')
# --------------------------------------------------
eservice_urls = contract_info.get("EnclaveServices")
enclave_services = list(
map(lambda url: GetEnclaveServiceByURL(url), eservice_urls))
for eservice in enclave_services:
secret_list = []
for pservice in provisioning_services:
message = pcrypto.string_to_byte_array(eservice.enclave_id +
contract_id)
signature = creator_keys.sign(message)
secret = pservice.get_secret(eservice.enclave_id, contract_id,
creator_keys.verifying_key, signature)
secret_list.append(secret)
secretinfo = eservice.verify_secrets(contract_id, contract_creator_id,
secret_list)
encrypted_state_encryption_key = secretinfo[
'encrypted_state_encryption_key']
signature = secretinfo['signature']
txnid = add_enclave_to_contract(ledger_config, creator_keys,
contract_id, eservice.enclave_id,
secret_list,
encrypted_state_encryption_key,
signature)
contract.set_state_encryption_key(eservice.enclave_id,
encrypted_state_encryption_key)
# --------------------------------------------------
logger.info('create the initial contract state')
# --------------------------------------------------
eservice = random.choice(enclave_services)
initialize_request = contract.create_initialize_request(
creator_keys, eservice)
initialize_response = initialize_request.evaluate()
if initialize_response.status is False:
emessage = initialize_response.result
logger.warn('initialization for contract %s failed; %s', contract_name,
emessage)
raise Exception('initialization failed; {}'.format(emessage))
contract.set_state(initialize_response.encrypted_state)
logger.info('initial state created')
# --------------------------------------------------
logger.info('save the initial state in the ledger')
# --------------------------------------------------
txnid = initialize_response.submit_initialize_transaction(ledger_config,
wait=30)
return contract
def command_create(state, bindings, pargs) :
"""controller command to create a contract
"""
parser = argparse.ArgumentParser(prog='create')
parser.add_argument('-c', '--contract-class', help='Name of the contract class', required = True, type=str)
parser.add_argument('-s', '--contract-source', help='File that contains contract source code', required=True, type=str)
parser.add_argument('-f', '--save-file', help='File where contract data is stored', type=str)
options = parser.parse_args(pargs)
contract_class = options.contract_class
contract_source = options.contract_source
contract_file = "{0}.pdo".format(contract_class)
if options.save_file :
contract_file = options.save_file
# ---------- load the invoker's keys ----------
try :
keyfile = state.get(['Key', 'FileName'])
keypath = state.get(['Key', 'SearchPath'])
client_keys = ServiceKeys.read_from_file(keyfile, keypath)
except Exception as e :
raise Exception('unable to load client keys; {0}'.format(str(e)))
# ---------- read the contract source code ----------
try :
source_path = state.get(['Contract', 'SourceSearchPath'])
contract_code = ContractCode.create_from_scheme_file(contract_class, contract_source, source_path)
except Exception as e :
raise Exception('unable to load contract source; {0}'.format(str(e)))
logger.info('Loaded contract code for %s', contract_class)
# ---------- set up the enclave clients ----------
try :
eservice_urls = state.get(['Service', 'EnclaveServiceURLs'], [])
if len(eservice_urls) == 0 :
raise Exception('no enclave services specified')
enclaveclients = []
for url in eservice_urls :
enclaveclients.append(EnclaveServiceClient(url))
except Exception as e :
raise Exception('unable to contact enclave services; {0}'.format(str(e)))
# ---------- set up the provisioning service clients ----------
# This is a dictionary of provisioning service public key : client pairs
try :
pservice_urls = state.get(['Service', 'ProvisioningServiceURLs'])
if len(pservice_urls) == 0 :
raise Exception('no provisioning services specified')
provclients = []
for url in pservice_urls :
provclients.append(ProvisioningServiceClient(url))
except Exception as e :
raise Exception('unable to contact provisioning services; {0}'.format(str(e)))
# ---------- register contract ----------
data_directory = state.get(['Contract', 'DataDirectory'])
ledger_config = state.get(['Sawtooth'])
try :
provisioning_service_keys = [pc.identity for pc in provclients]
contract_id = register_contract(
ledger_config, client_keys, contract_code, provisioning_service_keys)
logger.info('Registered contract with class %s and id %s', contract_class, contract_id)
contract_state = ContractState.create_new_state(contract_id)
contract = Contract(contract_code, contract_state, contract_id, client_keys.identity)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e :
raise Exception('failed to register the contract; {0}'.format(str(e)))
# provision the encryption keys to all of the enclaves
try :
encrypted_state_encryption_keys = __add_enclave_secrets(
ledger_config, contract.contract_id, client_keys, enclaveclients, provclients)
for enclave_id in encrypted_state_encryption_keys :
encrypted_key = encrypted_state_encryption_keys[enclave_id]
contract.set_state_encryption_key(enclave_id, encrypted_key)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e :
raise Exception('failed to provisioning the enclaves; {0}'.format(str(e)))
# create the initial contract state
try :
__create_contract(ledger_config, client_keys, enclaveclients, contract)
contract.contract_state.save_to_cache(data_dir = data_directory)
contract.save_to_file(contract_file, data_dir=data_directory)
except Exception as e :
raise Exception('failed to create the initial contract state; {0}'.format(str(e)))
def Create(config, client_profile, contract_name, contract_code, eservices,
pservices):
"""
client_profile -- toxaway.models.profile.Profile
contract_code -- toxaway.models.contract_code.ContractCode
eservices -- toxaway.models.eservice.EnclaveServiceList
pservices -- toxaway.models.pservice.ProvisioningServiceList
"""
ledger_config = config['Sawtooth']
contract_config = config['ContentPaths']
state_directory = contract_config['State']
client_keys = client_profile.keys
provisioning_service_keys = list(pservices.identities())
try:
pdo_code_object = contract_code.create_pdo_contract()
except Exception as e:
logger.error('failed to create the contract object; %s', str(e))
return None
try:
pdo_contract_id = register_contract(ledger_config, client_keys,
pdo_code_object,
provisioning_service_keys)
logger.info('Registered contract %s with id %s', contract_code.name,
pdo_contract_id)
pdo_contract_state = ContractState.create_new_state(pdo_contract_id)
contract = Contract(pdo_code_object, pdo_contract_state,
pdo_contract_id, client_keys.identity)
except Exception as e:
logger.error('failed to register the contract; %s', str(e))
return None
logger.info('Contract created')
enclaveclients = []
for eservice in eservices:
enclaveclients.append(eservice.eservice_client)
provclients = []
for pservice in pservices:
provclients.append(pservice.pservice_client)
encrypted_state_encryption_keys = AddEnclaveSecrets(
ledger_config, pdo_contract_id, client_keys, enclaveclients,
provclients)
for enclave_id in encrypted_state_encryption_keys:
encrypted_key = encrypted_state_encryption_keys[enclave_id]
contract.set_state_encryption_key(enclave_id, encrypted_key)
CreateContract(ledger_config, client_keys, enclaveclients, contract)
contract.contract_state.save_to_cache(data_dir=state_directory)
logger.info('state saved to cache')
with tempfile.NamedTemporaryFile() as pdo_temp:
contract.save_to_file(pdo_temp.name)
toxaway_contract = toxaway.models.contract.Contract.import_contract(
config, pdo_temp, contract_name)
return toxaway_contract