Просмотр исходного кода

unify the order of keypair inside cashierdb

ghassmo 4 лет назад
Родитель
Сommit
e9bbba839c
2 измененных файлов с 11 добавлено и 11 удалено
  1. 5 5
      src/service/cashier.rs
  2. 6 6
      src/wallet/cashierdb.rs

+ 5 - 5
src/service/cashier.rs

@@ -53,13 +53,13 @@ impl CashierService {
 
         let cashier_secret: jubjub::Fr;
 
-        if let Ok(secret) = wallet.get_cashier_private() {
+        if let Ok(secret) = wallet.get_private() {
             cashier_secret = secret;
         } else {
             wallet.init_db()?;
-            let keys = wallet.cash_key_gen();
+            let keys = wallet.key_gen();
             wallet.put_keypair(keys.0, keys.1)?;
-            cashier_secret = wallet.get_cashier_private()?;
+            cashier_secret = wallet.get_private()?;
         }
 
         let rocks = Rocks::new(&cashier_database_path)?;
@@ -229,7 +229,7 @@ impl CashierService {
                 let cashier_public: jubjub::SubgroupPoint;
 
                 if let Some(addr) = cashier_wallet.get_address_by_btc_key(&btc_address)? {
-                    cashier_public = deserialize(&addr.1)?;
+                    cashier_public = deserialize(&addr.0)?;
                 } else {
                     let cashier_secret = jubjub::Fr::random(&mut OsRng);
                     cashier_public =
@@ -237,8 +237,8 @@ impl CashierService {
 
                     cashier_wallet.put_withdraw_keys(
                         btc_address,
-                        serialize(&cashier_secret),
                         serialize(&cashier_public),
+                        serialize(&cashier_secret),
                     )?;
                 }
 

+ 6 - 6
src/wallet/cashierdb.rs

@@ -95,7 +95,7 @@ impl CashierDb {
         Ok(())
     }
 
-    // return (private key, public key)
+    // return (public key, private key)
     pub fn get_address_by_btc_key(
         &self,
         btc_address: &Vec<u8>,
@@ -110,7 +110,7 @@ impl CashierDb {
             conn.prepare("SELECT * FROM withdraw_keypairs where btc_key_id = :btc_key_id")?;
         let addr_iter = stmt
             .query_map::<(Vec<u8>, Vec<u8>), _, _>(&[(":btc_key_id", btc_address)], |row| {
-                Ok((row.get(1)?, row.get(2)?))
+                Ok((row.get(2)?, row.get(1)?))
             })?;
 
         let mut btc_addresses = vec![];
@@ -129,8 +129,8 @@ impl CashierDb {
     pub fn put_withdraw_keys(
         &self,
         btc_key_id: Vec<u8>,
-        d_key_private: Vec<u8>,
         d_key_public: Vec<u8>,
+        d_key_private: Vec<u8>,
     ) -> Result<()> {
         debug!(target: "CASHIERDB", "Put withdraw keys");
 
@@ -151,7 +151,7 @@ impl CashierDb {
         Ok(())
     }
 
-    pub fn cash_key_gen(&self) -> (Vec<u8>, Vec<u8>) {
+    pub fn key_gen(&self) -> (Vec<u8>, Vec<u8>) {
         debug!(target: "CASHIERDB", "Generating cashier keys...");
         let secret: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
         let public = zcash_primitives::constants::SPENDING_KEY_GENERATOR * secret;
@@ -170,7 +170,7 @@ impl CashierDb {
         Ok(())
     }
 
-    pub fn get_cashier_public(&self) -> Result<jubjub::SubgroupPoint> {
+    pub fn get_public(&self) -> Result<jubjub::SubgroupPoint> {
         debug!(target: "CASHIERDB", "Returning keys...");
         let conn = Connection::open(&self.path)?;
         conn.pragma_update(None, "key", &self.password)?;
@@ -185,7 +185,7 @@ impl CashierDb {
         Ok(public)
     }
 
-    pub fn get_cashier_private(&self) -> Result<jubjub::Fr> {
+    pub fn get_private(&self) -> Result<jubjub::Fr> {
         debug!(target: "CASHIERDB", "Returning keys...");
         let conn = Connection::open(&self.path)?;
         conn.pragma_update(None, "key", &self.password)?;