github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

walletdb: fix a bug in token_id_exists() function

Browse Source
This commit is contained in:
ghassmo
2021-10-07 10:18:54 +03:00
parent 25aa1200c0
commit 7c22cd645c
1 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/wallet/walletdb.rs
View File

@@ -82,8 +82,8 @@ impl WalletDb {
debug!(target: "WALLETDB", "Attempting to generate keys...");
let conn = Connection::open(&self.path)?;
conn.pragma_update(None, "key", &self.password)?;
let mut stmt = conn.prepare("SELECT * FROM keys WHERE key_id > :id")?;
let key_check = stmt.exists(&[(":id", &"0")])?;
let mut stmt = conn.prepare("SELECT * FROM keys WHERE key_id > ?")?;
let key_check = stmt.exists(params!["0"])?;
if !key_check {
let secret: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
let public = zcash_primitives::constants::SPENDING_KEY_GENERATOR * secret;
@@ -368,8 +368,8 @@ impl WalletDb {
let conn = Connection::open(&self.path)?;
conn.pragma_update(None, "key", &self.password)?;
let id = self.get_value_serialized(token_id)?;
let mut stmt = conn.prepare("SELECT * FROM coins WHERE asset_id > :id")?;
let id_check = stmt.exists([id])?;
let mut stmt = conn.prepare("SELECT * FROM coins WHERE asset_id = ?")?;
let id_check = stmt.exists(params![id])?;
Ok(id_check)
}
@@ -455,6 +455,8 @@ mod tests {
assert_eq!(token_id[0], asset_id);
assert_eq!(token_id[2], asset_id);
assert!(wallet.token_id_exists(&asset_id)?);
std::fs::remove_file(walletdb_path)?;
Ok(())