Jędrzej Stuczyński
12637d93ff
* Sqlx struct stub * Initial schema * Initial error enum * Managed for persisted shared keys * Initial inbox manager * Comments * Using new database in clients handler * Extending gateway storage API * tokio::main + placeholder values * Removed old client store * Simplified logic of async packet processing * Renamed table + not null restriction * BandwidthManager * Removed sled dependency * Using centralised storage for bandwidth * Dead code removal * WIP connection_handler split and simplification Maybe it doesn't look like it right now, but once completed it will remove bunch of redundant checks for Nones etc * Further more explicit clients handler split * Minor cleanup * Temporary store for active client handles * Fixed error types * Error trait on iv and encrypted address * Authentication and registration moved to the handler * Removal of clients handler * Further logic simplification + returned explicit bandwidth values * Further cleanup and comments * Updated config with relevant changes * Basic bandwidth tracking in client * FreshHandle doc comments + fixed stagger issue * Removed side-effects from .map * More doc comments * Database migration on build * Increased default claimed bandwidth * Renaming * Fixed client determining available bandwidth * Removed dead sql table that might be used in the future * Windows workaround * Comment * Return error rather than cap credential
77 lines
2.6 KiB
Rust
77 lines
2.6 KiB
Rust
// Copyright 2021 - Nym Technologies SA <contact@nymtech.net>
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use crate::node::storage::models::PersistedSharedKeys;
|
|
|
|
#[derive(Clone)]
|
|
pub(crate) struct SharedKeysManager {
|
|
connection_pool: sqlx::SqlitePool,
|
|
}
|
|
|
|
impl SharedKeysManager {
|
|
/// Creates new instance of the `SharedKeysManager` with the provided sqlite connection pool.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `connection_pool`: database connection pool to use.
|
|
pub(crate) fn new(connection_pool: sqlx::SqlitePool) -> Self {
|
|
SharedKeysManager { connection_pool }
|
|
}
|
|
|
|
/// Inserts provided derived shared keys into the database.
|
|
/// If keys previously existed for the provided client, they are overwritten with the new data.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `shared_keys`: shared encryption (AES128CTR) and mac (hmac-blake3) derived shared keys to store.
|
|
pub(crate) async fn insert_shared_keys(
|
|
&self,
|
|
shared_keys: PersistedSharedKeys,
|
|
) -> Result<(), sqlx::Error> {
|
|
sqlx::query!("INSERT OR REPLACE INTO shared_keys(client_address_bs58, derived_aes128_ctr_blake3_hmac_keys_bs58) VALUES (?, ?)",
|
|
shared_keys.client_address_bs58,
|
|
shared_keys.derived_aes128_ctr_blake3_hmac_keys_bs58,
|
|
).execute(&self.connection_pool).await?;
|
|
Ok(())
|
|
}
|
|
|
|
/// Tries to retrieve shared keys stored for the particular client.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `client_address_bs58`: base58-encoded address of the client
|
|
pub(crate) async fn get_shared_keys(
|
|
&self,
|
|
client_address_bs58: &str,
|
|
) -> Result<Option<PersistedSharedKeys>, sqlx::Error> {
|
|
sqlx::query_as!(
|
|
PersistedSharedKeys,
|
|
"SELECT * FROM shared_keys WHERE client_address_bs58 = ?",
|
|
client_address_bs58
|
|
)
|
|
.fetch_optional(&self.connection_pool)
|
|
.await
|
|
}
|
|
|
|
/// Removes from the database shared keys derived with the particular client.
|
|
///
|
|
/// # Arguments
|
|
///
|
|
/// * `client_address_bs58`: base58-encoded address of the client
|
|
// currently there is no code flow that causes removal (not overwriting)
|
|
// of the stored keys. However, retain the function for consistency and completion sake
|
|
#[allow(dead_code)]
|
|
pub(crate) async fn remove_shared_keys(
|
|
&self,
|
|
client_address_bs58: &str,
|
|
) -> Result<(), sqlx::Error> {
|
|
sqlx::query!(
|
|
"DELETE FROM shared_keys WHERE client_address_bs58 = ?",
|
|
client_address_bs58
|
|
)
|
|
.execute(&self.connection_pool)
|
|
.await?;
|
|
Ok(())
|
|
}
|
|
}
|