openethereum/ethash/src/lib.rs

76 lines
2.3 KiB
Rust
Raw Normal View History

2016-02-05 13:40:41 +01:00
// Copyright 2015, 2016 Ethcore (UK) Ltd.
// This file is part of Parity.
// Parity 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.
// Parity 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 Parity. If not, see <http://www.gnu.org/licenses/>.
2016-01-17 12:00:34 +01:00
//! Ethash implementation
//! See https://github.com/ethereum/wiki/wiki/Ethash
2016-01-18 12:49:39 +01:00
extern crate sha3;
2016-01-30 23:02:55 +01:00
extern crate lru_cache;
#[macro_use]
extern crate log;
2016-01-17 12:00:34 +01:00
mod sizes;
mod compute;
2016-01-30 23:02:55 +01:00
use lru_cache::LruCache;
2016-01-17 12:00:34 +01:00
use compute::Light;
pub use compute::{quick_get_difficulty, H256, ProofOfWork, ETHASH_EPOCH_LENGTH};
2016-01-30 23:02:55 +01:00
use std::sync::{Arc, Mutex};
2016-01-17 12:00:34 +01:00
/// Lighy/Full cache manager
pub struct EthashManager {
2016-01-30 23:02:55 +01:00
lights: Mutex<LruCache<u64, Arc<Light>>>
2016-01-17 12:00:34 +01:00
}
impl EthashManager {
/// Create a new new instance of ethash manager
pub fn new() -> EthashManager {
EthashManager {
2016-01-30 23:02:55 +01:00
lights: Mutex::new(LruCache::new(2))
2016-01-17 12:00:34 +01:00
}
}
/// Calculate the light client data
/// `block_number` - Block number to check
/// `light` - The light client handler
/// `header_hash` - The header hash to pack into the mix
/// `nonce` - The nonce to pack into the mix
pub fn compute_light(&self, block_number: u64, header_hash: &H256, nonce: u64) -> ProofOfWork {
let epoch = block_number / ETHASH_EPOCH_LENGTH;
2016-01-30 23:02:55 +01:00
let light = {
let mut lights = self.lights.lock().unwrap();
match lights.get_mut(&epoch).map(|l| l.clone()) {
None => {
let light = match Light::from_file(block_number) {
Ok(light) => Arc::new(light),
Err(e) => {
debug!("Light cache file not found for {}:{}", block_number, e);
let light = Light::new(block_number);
if let Err(e) = light.to_file() {
warn!("Light cache file write error: {}", e);
}
Arc::new(light)
}
};
lights.insert(epoch, light.clone());
light
2016-01-21 16:48:37 +01:00
}
2016-01-30 23:02:55 +01:00
Some(light) => light
2016-01-17 12:00:34 +01:00
}
2016-01-30 23:02:55 +01:00
};
light.compute(header_hash, nonce)
2016-01-17 12:00:34 +01:00
}
}