From 40cbfd5883d834c1a81a8c1d11d9c47b0c43227b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20Silva?= Date: Fri, 15 Jun 2018 22:56:36 +0100 Subject: [PATCH] ethstore: retry deduplication of wallet file names until success --- ethstore/src/accounts_dir/disk.rs | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/ethstore/src/accounts_dir/disk.rs b/ethstore/src/accounts_dir/disk.rs index a9253cff06f..6917a6278c7 100644 --- a/ethstore/src/accounts_dir/disk.rs +++ b/ethstore/src/accounts_dir/disk.rs @@ -182,9 +182,22 @@ impl DiskDirectory where T: KeyFileManager { // check for duplicate filename and append random suffix if dedup && keyfile_path.exists() { - let suffix = ::random::random_string(4); - filename.push_str(&format!("-{}", suffix)); - keyfile_path.set_file_name(&filename); + const MAX_RETRIES: usize = 500; + let mut retries = 0; + let mut deduped_filename = filename.clone(); + + while keyfile_path.exists() { + if retries >= MAX_RETRIES { + return Err(Error::Custom(format!("Exceeded maximum retries when deduplicating account filename."))); + } + + let suffix = ::random::random_string(4); + deduped_filename = format!("{}-{}", filename, suffix); + keyfile_path.set_file_name(&deduped_filename); + retries += 1; + } + + filename = deduped_filename; } // update account filename