-
Notifications
You must be signed in to change notification settings - Fork 8
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
27 changed files
with
538 additions
and
90 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,2 +1,2 @@ | ||
| [workspace] | ||
| members = ["testing-utils/test-moat-cli", "moat-cli-user", "moat-cli-lp", "moat-cli-sp", "testing-utils/test-moat-request", "moat-core", "wallet-accessor", "integration-tests", "license-provider", "macros/code-hasher"] | ||
| members = ["testing-utils/test-moat-cli", "moat-cli-user", "moat-cli-lp", "moat-cli-sp", "testing-utils/test-moat-request", "moat-core", "wallet-accessor", "integration-tests", "macros/code-hasher", "moat-example"] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file was deleted.
Oops, something went wrong.
This file was deleted.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,224 @@ | ||
| // This Source Code Form is subject to the terms of the Mozilla Public | ||
| // License, v. 2.0. If a copy of the MPL was not distributed with this | ||
| // file, You can obtain one at http://mozilla.org/MPL/2.0/. | ||
| // | ||
| // Copyright (c) DUSK NETWORK. All rights reserved. | ||
|
|
||
| use dusk_jubjub::{JubJubAffine, JubJubScalar}; | ||
| use dusk_pki::{PublicSpendKey, SecretSpendKey}; | ||
| use dusk_wallet::{RuskHttpClient, Wallet, WalletPath}; | ||
|
|
||
| use zk_citadel::license::{Session, SessionCookie}; | ||
|
|
||
| use crate::license_provider::{LicenseIssuer, ReferenceLP}; | ||
| use crate::utils::MoatCoreUtils; | ||
| use crate::{ | ||
| CitadelInquirer, LicenseSessionId, RequestCreator, RequestSender, TxAwaiter, | ||
| }; | ||
| use wallet_accessor::Password::{self, Pwd}; | ||
| use wallet_accessor::{BlockchainAccessConfig, WalletAccessor}; | ||
|
|
||
| use rand::rngs::OsRng; | ||
| use std::path::Path; | ||
| use toml_base_config::BaseConfig; | ||
|
|
||
| pub struct MoatCore {} | ||
|
|
||
| impl MoatCore { | ||
| /// Retrieve the keypair (psk, ssk) from the installed wallet, given the | ||
| /// Moat Context | ||
| pub fn get_wallet_keypair( | ||
| moat_context: &MoatContext, | ||
| ) -> (PublicSpendKey, SecretSpendKey) { | ||
| // We access the wallet to get the key pair | ||
| let wallet_accessor = WalletAccessor::create( | ||
| moat_context.wallet_path.clone(), | ||
| moat_context.wallet_password.clone(), | ||
| ) | ||
| .unwrap(); | ||
| let wallet = Wallet::from_file(wallet_accessor).unwrap(); | ||
|
|
||
| wallet.spending_keys(wallet.default_address()).unwrap() | ||
| } | ||
|
|
||
| /// Create and send a transaction containing a license request | ||
| pub async fn request_license( | ||
| ssk_user: &SecretSpendKey, | ||
| psk_lp: &PublicSpendKey, | ||
| moat_context: &MoatContext, | ||
| rng: &mut OsRng, | ||
| ) -> String { | ||
| let request = RequestCreator::create(ssk_user, psk_lp, rng).unwrap(); | ||
| let request_hash = MoatCoreUtils::to_hash_hex(&request); | ||
|
|
||
| let tx_id = RequestSender::send_request( | ||
| request, | ||
| &moat_context.blockchain_access_config, | ||
| &moat_context.wallet_path, | ||
| &moat_context.wallet_password, | ||
| moat_context.gas_limit, | ||
| moat_context.gas_price, | ||
| ) | ||
| .await | ||
| .unwrap(); | ||
|
|
||
| let client = RuskHttpClient::new( | ||
| moat_context.blockchain_access_config.rusk_address.clone(), | ||
| ); | ||
| TxAwaiter::wait_for(&client, tx_id).await.unwrap(); | ||
|
|
||
| request_hash | ||
| } | ||
|
|
||
| /// Create and send a transaction containing a license for a given request | ||
| pub async fn issue_license( | ||
| request_hash: &String, | ||
| ssk_lp: &SecretSpendKey, | ||
| moat_context: &MoatContext, | ||
| attr_data: &JubJubScalar, | ||
| rng: &mut OsRng, | ||
| ) -> String { | ||
| let mut reference_lp = ReferenceLP::create_with_ssk(ssk_lp).unwrap(); | ||
|
|
||
| let (_total_count, _this_lp_count) = reference_lp | ||
| .scan(&moat_context.blockchain_access_config) | ||
| .await | ||
| .unwrap(); | ||
|
|
||
| let request = reference_lp.get_request(request_hash); | ||
|
|
||
| match request { | ||
| Some(request) => { | ||
| let license_issuer = LicenseIssuer::new( | ||
| moat_context.blockchain_access_config.clone(), | ||
| moat_context.wallet_path.clone(), | ||
| moat_context.wallet_password.clone(), | ||
| moat_context.gas_limit, | ||
| moat_context.gas_price, | ||
| ); | ||
| let (_tx_id, license_blob) = license_issuer | ||
| .issue_license( | ||
| rng, | ||
| &request, | ||
| &reference_lp.ssk_lp, | ||
| attr_data, | ||
| ) | ||
| .await | ||
| .unwrap(); | ||
| MoatCoreUtils::blob_to_hash_hex(&license_blob) | ||
| } | ||
| _ => "nothing to do".to_string(), | ||
| } | ||
| } | ||
|
|
||
| /// Create and send a transaction containing a proof that uses a given | ||
| /// license | ||
| pub async fn use_license( | ||
| moat_context: &MoatContext, | ||
| psk_lp: &PublicSpendKey, | ||
| psk_sp: &PublicSpendKey, | ||
| ssk: &SecretSpendKey, | ||
| challenge: &JubJubScalar, | ||
| license_hash: &str, | ||
| rng: &mut OsRng, | ||
| ) -> Option<SessionCookie> { | ||
| let pos_license = MoatCoreUtils::get_license_to_use( | ||
| &moat_context.blockchain_access_config, | ||
| ssk, | ||
| license_hash.to_owned(), | ||
| ) | ||
| .await | ||
| .unwrap(); | ||
|
|
||
| match pos_license { | ||
| Some((pos, license)) => { | ||
| println!( | ||
| "using license: {}", | ||
| MoatCoreUtils::to_hash_hex(&license) | ||
| ); | ||
|
|
||
| let (_tx_id, session_cookie) = | ||
| MoatCoreUtils::prove_and_send_use_license( | ||
| &moat_context.blockchain_access_config, | ||
| &moat_context.wallet_path, | ||
| &moat_context.wallet_password, | ||
| psk_lp, | ||
| psk_sp, | ||
| ssk, | ||
| challenge, | ||
| &license, | ||
| pos, | ||
| moat_context.gas_limit, | ||
| moat_context.gas_price, | ||
| &mut None, | ||
| rng, | ||
| ) | ||
| .await | ||
| .unwrap(); | ||
|
|
||
| Some(session_cookie) | ||
| } | ||
| _ => None, | ||
| } | ||
| } | ||
|
|
||
| /// Given a session cookie, verify that it corresponds to an existing | ||
| /// session in the Blockchain | ||
| pub async fn verify_requested_service( | ||
| moat_context: &MoatContext, | ||
| psk_lp: &PublicSpendKey, | ||
| psk_sp: &PublicSpendKey, | ||
| session_cookie: &SessionCookie, | ||
| ) -> bool { | ||
| let client = RuskHttpClient::new( | ||
| moat_context.blockchain_access_config.rusk_address.clone(), | ||
| ); | ||
|
|
||
| let pk_lp = JubJubAffine::from(*psk_lp.A()); | ||
| let pk_sp = JubJubAffine::from(*psk_sp.A()); | ||
|
|
||
| let session_id = LicenseSessionId { | ||
| id: session_cookie.session_id, | ||
| }; | ||
| let session = CitadelInquirer::get_session(&client, session_id) | ||
| .await | ||
| .unwrap() | ||
| .unwrap(); | ||
|
|
||
| let session = Session::from(&session.public_inputs); | ||
| session.verifies_ok(*session_cookie, pk_lp, pk_sp) | ||
| } | ||
| } | ||
|
|
||
| pub struct MoatContext { | ||
| blockchain_access_config: BlockchainAccessConfig, | ||
| wallet_path: WalletPath, | ||
| wallet_password: Password, | ||
| gas_limit: u64, | ||
| gas_price: u64, | ||
| } | ||
|
|
||
| impl MoatContext { | ||
| /// Create a new Moat Context given the required configurations | ||
| pub fn new( | ||
| config_path: &String, | ||
| wallet_path: &String, | ||
| wallet_password: &String, | ||
| gas_limit: u64, | ||
| gas_price: u64, | ||
| ) -> Self { | ||
| let wallet_path = WalletPath::from(Path::new(&wallet_path)); | ||
| let wallet_password = Pwd(wallet_password.to_string()); | ||
|
|
||
| let blockchain_access_config = | ||
| BlockchainAccessConfig::load_path(config_path).unwrap(); | ||
|
|
||
| Self { | ||
| blockchain_access_config, | ||
| wallet_path, | ||
| wallet_password, | ||
| gas_limit, | ||
| gas_price, | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.