openethereum/accounts/ethstore/cli/src/crack.rs

// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of OpenEthereum.

// OpenEthereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// OpenEthereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with OpenEthereum.  If not, see <http://www.gnu.org/licenses/>.

use parking_lot::Mutex;
use std::{cmp, collections::VecDeque, sync::Arc, thread};

use ethstore::{ethkey::Password, Error, PresaleWallet};
use num_cpus;

pub fn run(passwords: VecDeque<Password>, wallet_path: &str) -> Result<(), Error> {
    let passwords = Arc::new(Mutex::new(passwords));

    let mut handles = Vec::new();

    for _ in 0..num_cpus::get() {
        let passwords = passwords.clone();
        let wallet = PresaleWallet::open(&wallet_path)?;
        handles.push(thread::spawn(move || {
            look_for_password(passwords, wallet);
        }));
    }

    for handle in handles {
        handle
            .join()
            .map_err(|err| Error::Custom(format!("Error finishing thread: {:?}", err)))?;
    }

    Ok(())
}

fn look_for_password(passwords: Arc<Mutex<VecDeque<Password>>>, wallet: PresaleWallet) {
    let mut counter = 0;
    while !passwords.lock().is_empty() {
        let package = {
            let mut passwords = passwords.lock();
            let len = passwords.len();
            passwords.split_off(cmp::min(len, 32))
        };
        for pass in package {
            counter += 1;
            match wallet.decrypt(&pass) {
                Ok(_) => {
                    println!("Found password: {}", pass.as_str());
                    passwords.lock().clear();
                    return;
                }
                _ if counter % 100 == 0 => print!("."),
                _ => {}
            }
        }
    }
}