Start implementing hashing for private keys

Author: bananasov

Cargo.lock

@@ -31,6 +31,7 @@ tracing-subscriber = "0.3.18"
 serde = { version = "1.0.209", features = ["derive"] }
 serde_json = "1.0.127"
 thiserror = "1.0.63"
+argon2 = "0.5.3"
 
 [dependencies]
 kromer-economy-api = { path = "api" }

Cargo.toml

@@ -14,4 +14,5 @@ kromer-economy-entity = { path = "../entity" }
 kromer-economy-migration = { path = "../migration" }
 tracing.workspace = true
 tracing-subscriber.workspace = true
-thiserror.workspace = true
+thiserror.workspace = true
+argon2.workspace = true

api/Cargo.toml

@@ -126,8 +126,11 @@ async fn register_name(
     let conn = &state.conn;
 
     // TODO: Implement proper authentication and address get from private key
-    let _private_key = &body.private_key;
-    let owner_address = "TODO_GET_ADDRESS_FROM_PRIVATE_KEY"; // TODO: Get address from private key, should return address model.
+    let private_key = &body.private_key;
+    let owner = AddressController::get_from_private_key(conn, private_key)
+        .await
+        .map_err(KromerError::Database)?
+        .ok_or_else(|| KromerError::Address(AddressError::AuthFailed))?;
 
     let name_available = NameController::is_name_available(conn, &name)
         .await
@@ -137,15 +140,9 @@ async fn register_name(
         return Err(KromerError::Name(NameError::NameTaken(name)));
     }
 
-    let owner = AddressController::fetch_address(conn, owner_address, false)
-        .await
-        .map_err(KromerError::Database)?
-        .ok_or_else(|| KromerError::Address(AddressError::NotFound(owner_address.to_string())))?;
-
-    // TODO: Check if the user has enough balance to register the name
-    // if owner.balance < state.name_cost {
-    //     return Err(KromerError::Name(NameError::InsufficientFunds));
-    // }
+    if owner.balance < state.name_cost {
+        return Err(KromerError::Name(NameError::InsufficientBalance));
+    }
 
     let registration = NameRegistration { name, owner };
 

api/src/routes/v1/names.rs

@@ -19,6 +19,7 @@ pub struct Model {
     pub private_key: Option<String>,
     pub alert: Option<String>,
     pub locked: bool,
+    pub salt: Option<String>,
 }
 
 #[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]

entity/src/addresses.rs

@@ -0,0 +1,7 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.0.1
+
+pub mod prelude;
+
+pub mod addresses;
+pub mod names;
+pub mod transactions;

entity/src/mod.rs

@@ -179,6 +179,7 @@ impl MigrationTrait for Migration {
                             .null()
                             .take(),
                     )
+                    .col(ColumnDef::new(Addresses::Salt).char_len(16).null())
                     .col(
                         ColumnDef::new(Addresses::Alert)
                             .string_len(1024)
@@ -280,6 +281,7 @@ enum Addresses {
     TotalOut,
     FirstSeen,
     PrivateKey,
+    Salt,
     Alert,
     Locked,
 }

migration/src/m20220101_000001_create_table.rs

@@ -5,4 +5,5 @@ edition = "2021"
 
 [dependencies]
 sea-orm.workspace = true
-kromer-economy-entity = { path = "../entity" }
+kromer-economy-entity = { path = "../entity" }
+argon2.workspace = true

service/Cargo.toml

@@ -151,4 +151,26 @@ impl AddressController {
             .all(conn)
             .await
     }
+
+    /// Fetches the address from the database when given a private key
+    ///
+    /// # Arguments
+    /// * `conn` - The database connection
+    /// * `private_key` - The private key to fetch the address for
+    ///
+    /// # Examples
+    /// ```
+    /// let address = AddressController::get_from_private_key(&db, "pasetorules!").await?;
+    /// ```
+    pub async fn get_from_private_key(
+        conn: &DbConn,
+        private_key: &str,
+    ) -> Result<Option<addresses::Model>, DbErr> {
+        // Address::find()
+        //     .filter(addresses::Column::PrivateKey.eq(private_key))
+        //     .one(conn)
+        //     .await
+
+        todo!()
+    }
 }