use crate::{
chain_tip::ChainTip,
extranonce_manager::{AllocatedExtranoncePrefix, ExtranoncePrefix},
server::{
error::StandardChannelError,
jobs::{
extended::ExtendedJob, factory::JobFactory, job_store::JobStore, standard::StandardJob,
},
share_accounting::{ShareAccounting, ShareValidationError, ShareValidationResult},
},
target::{bytes_to_hex, hash_rate_to_target, u256_to_block_hash},
MAX_EXTRANONCE_LEN,
};
use bitcoin::{
absolute::LockTime,
blockdata::{
block::{Header, Version},
witness::Witness,
},
consensus::Encodable,
hashes::sha256d::Hash,
transaction::{OutPoint, Transaction, TxIn, TxOut, Version as TxVersion},
CompactTarget, Sequence, Target,
};
use mining_sv2::{
SubmitSharesStandard, ERROR_CODE_OPEN_MINING_CHANNEL_INVALID_NOMINAL_HASHRATE,
ERROR_CODE_SUBMIT_SHARES_DIFFICULTY_TOO_LOW, ERROR_CODE_SUBMIT_SHARES_DUPLICATE_SHARE,
ERROR_CODE_SUBMIT_SHARES_INVALID_JOB_ID, ERROR_CODE_SUBMIT_SHARES_STALE_SHARE,
ERROR_CODE_UPDATE_CHANNEL_INVALID_NOMINAL_HASHRATE,
};
use std::{collections::HashMap, convert::TryInto, marker::PhantomData};
use template_distribution_sv2::{NewTemplate, SetNewPrevHash};
use tracing::debug;
#[derive(Debug)]
pub struct StandardChannel<'a, J>
where
J: JobStore<StandardJob<'a>>,
{
pub channel_id: u32,
user_identity: String,
extranonce_prefix: ExtranoncePrefix,
requested_max_target: Target,
target: Target,
job_id_to_target: HashMap<u32, Target>,
nominal_hashrate: f32,
stable_hashrate: bool,
share_accounting: ShareAccounting,
expected_share_per_minute: f32,
job_store: J,
job_factory: JobFactory,
chain_tip: Option<ChainTip>,
phantom: PhantomData<&'a ()>,
}
impl<'a, J> StandardChannel<'a, J>
where
J: JobStore<StandardJob<'a>>,
{
#[allow(clippy::too_many_arguments)]
pub fn new_for_pool(
channel_id: u32,
user_identity: String,
extranonce_prefix: AllocatedExtranoncePrefix,
requested_max_target: Target,
nominal_hashrate: f32,
share_batch_size: usize,
expected_share_per_minute: f32,
job_store: J,
pool_tag_string: String,
) -> Result<Self, StandardChannelError> {
Self::new(
channel_id,
user_identity,
extranonce_prefix.into(),
requested_max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
Some(pool_tag_string),
None,
)
}
#[allow(clippy::too_many_arguments)]
pub fn new_for_job_declaration_client(
channel_id: u32,
user_identity: String,
extranonce_prefix: AllocatedExtranoncePrefix,
requested_max_target: Target,
nominal_hashrate: f32,
share_batch_size: usize,
expected_share_per_minute: f32,
job_store: J,
pool_tag_string: Option<String>,
miner_tag_string: String,
) -> Result<Self, StandardChannelError> {
Self::new(
channel_id,
user_identity,
extranonce_prefix.into(),
requested_max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
pool_tag_string,
Some(miner_tag_string),
)
}
#[allow(clippy::too_many_arguments)]
fn new(
channel_id: u32,
user_identity: String,
extranonce_prefix: ExtranoncePrefix,
requested_max_target: Target,
nominal_hashrate: f32,
share_batch_size: usize,
expected_share_per_minute: f32,
job_store: J,
pool_tag_string: Option<String>,
miner_tag_string: Option<String>,
) -> Result<Self, StandardChannelError> {
let calculated_target =
match hash_rate_to_target(nominal_hashrate.into(), expected_share_per_minute.into()) {
Ok(target_u256) => target_u256,
Err(_) => {
return Err(StandardChannelError::OpenChannelInvalidNominalHashrate(
ERROR_CODE_OPEN_MINING_CHANNEL_INVALID_NOMINAL_HASHRATE,
));
}
};
let target = calculated_target.min(requested_max_target);
if extranonce_prefix.len() > MAX_EXTRANONCE_LEN as usize {
return Err(StandardChannelError::ExtranoncePrefixTooLarge);
}
let script_sig_size = 5 + 1 + 3 + pool_tag_string.as_ref().map_or(0, |s| s.len()) +
miner_tag_string.as_ref().map_or(0, |s| s.len()) +
1 + extranonce_prefix.len();
if script_sig_size > 100 {
return Err(StandardChannelError::ScriptSigSizeTooLarge);
}
Ok(Self {
channel_id,
user_identity,
extranonce_prefix,
requested_max_target,
target,
job_id_to_target: HashMap::new(),
nominal_hashrate,
stable_hashrate: false,
share_accounting: ShareAccounting::new(share_batch_size),
expected_share_per_minute,
job_factory: JobFactory::new(true, pool_tag_string, miner_tag_string),
chain_tip: None,
job_store,
phantom: PhantomData,
})
}
pub fn get_channel_id(&self) -> u32 {
self.channel_id
}
pub fn get_user_identity(&self) -> &str {
&self.user_identity
}
pub fn get_extranonce_prefix(&self) -> &[u8] {
self.extranonce_prefix.as_bytes()
}
pub fn set_extranonce_prefix(
&mut self,
extranonce_prefix: AllocatedExtranoncePrefix,
) -> Result<(), StandardChannelError> {
if extranonce_prefix.len() > MAX_EXTRANONCE_LEN as usize {
return Err(StandardChannelError::ExtranoncePrefixTooLarge);
}
self.extranonce_prefix = extranonce_prefix.into();
Ok(())
}
pub fn set_target(&mut self, target: Target) {
self.target = target;
}
pub fn set_nominal_hashrate(&mut self, nominal_hashrate: f32) {
self.nominal_hashrate = nominal_hashrate;
}
pub fn set_stable_hashrate(&mut self, stable_hashrate: bool) {
self.stable_hashrate = stable_hashrate;
}
pub fn get_stable_hashrate(&self) -> bool {
self.stable_hashrate
}
pub fn get_requested_max_target(&self) -> &Target {
&self.requested_max_target
}
pub fn get_target(&self) -> &Target {
&self.target
}
pub fn get_nominal_hashrate(&self) -> f32 {
self.nominal_hashrate
}
pub fn update_channel(
&mut self,
nominal_hashrate: f32,
requested_max_target: Option<Target>,
) -> Result<(), StandardChannelError> {
let target = match hash_rate_to_target(
nominal_hashrate.into(),
self.expected_share_per_minute.into(),
) {
Ok(target) => target,
Err(_) => {
return Err(StandardChannelError::UpdateChannelInvalidNominalHashrate(
ERROR_CODE_UPDATE_CHANNEL_INVALID_NOMINAL_HASHRATE,
));
}
};
let requested_max_target = match requested_max_target {
Some(ref requested_max_target) => *requested_max_target,
None => self.requested_max_target,
};
let target_bytes = target.to_be_bytes();
let max_target_bytes = requested_max_target.to_be_bytes();
let old_target = self.target;
let old_target_bytes = old_target.to_be_bytes();
debug!(
"updating channel target \nold target:\t{}\nnew target:\t{}\nmax_target:\t{}",
bytes_to_hex(&old_target_bytes),
bytes_to_hex(&target_bytes),
bytes_to_hex(&max_target_bytes)
);
let new_target = target.min(requested_max_target);
self.target = new_target;
self.nominal_hashrate = nominal_hashrate;
self.requested_max_target = requested_max_target;
Ok(())
}
pub fn get_active_job(&self) -> Option<StandardJob<'a>> {
self.job_store.get_active_job()
}
pub fn get_future_job_id_from_template_id(&self, template_id: u64) -> Option<u32> {
self.job_store
.get_future_job_id_from_template_id(template_id)
}
pub fn get_future_job(&self, job_id: u32) -> Option<StandardJob<'a>> {
self.job_store.get_future_job(job_id)
}
pub fn get_past_job(&self, job_id: u32) -> Option<StandardJob<'a>> {
self.job_store.get_past_job(job_id)
}
pub fn get_stale_job(&self, job_id: u32) -> Option<StandardJob<'a>> {
self.job_store.get_stale_job(job_id)
}
pub fn get_shares_per_minute(&self) -> f32 {
self.expected_share_per_minute
}
pub fn get_chain_tip(&self) -> Option<&ChainTip> {
self.chain_tip.as_ref()
}
#[cfg(test)]
fn set_chain_tip(&mut self, chain_tip: ChainTip) {
self.chain_tip = Some(chain_tip);
}
pub fn get_share_accounting(&self) -> &ShareAccounting {
&self.share_accounting
}
pub fn on_new_template(
&mut self,
template: NewTemplate<'a>,
coinbase_reward_outputs: Vec<TxOut>,
) -> Result<(), StandardChannelError> {
match template.future_template {
true => {
let new_job = self
.job_factory
.new_standard_job(
self.channel_id,
None,
self.extranonce_prefix.as_bytes().to_vec(),
template.clone(),
coinbase_reward_outputs,
)
.map_err(StandardChannelError::JobFactoryError)?;
self.job_store.add_future_job(template.template_id, new_job);
}
false => {
match self.chain_tip.clone() {
None => return Err(StandardChannelError::ChainTipNotSet),
Some(chain_tip) => {
let new_job = self
.job_factory
.new_standard_job(
self.channel_id,
Some(chain_tip),
self.extranonce_prefix.as_bytes().to_vec(),
template.clone(),
coinbase_reward_outputs,
)
.map_err(StandardChannelError::JobFactoryError)?;
self.job_id_to_target
.insert(new_job.get_job_id(), self.target);
self.job_store.add_active_job(new_job);
}
}
}
}
Ok(())
}
pub fn on_group_channel_job(
&mut self,
extended_job: ExtendedJob<'a>,
) -> Result<(), StandardChannelError> {
let standard_job = extended_job
.into_standard_job(self.channel_id, self.extranonce_prefix.as_bytes().to_vec())
.map_err(|_| StandardChannelError::FailedToConvertToStandardJob)?;
match standard_job.is_future() {
true => {
self.job_store
.add_future_job(standard_job.get_template().template_id, standard_job);
}
false => {
self.job_id_to_target
.insert(standard_job.get_job_id(), self.target);
self.job_store.add_active_job(standard_job);
}
}
Ok(())
}
pub fn on_set_new_prev_hash(
&mut self,
set_new_prev_hash: SetNewPrevHash<'a>,
) -> Result<(), StandardChannelError> {
match self.job_store.has_future_jobs() {
false => {
return Err(StandardChannelError::TemplateIdNotFound);
}
true => {
if !self.job_store.activate_future_job(
set_new_prev_hash.template_id,
set_new_prev_hash.header_timestamp,
) {
return Err(StandardChannelError::TemplateIdNotFound);
}
self.job_id_to_target.clear();
let job_id = self
.job_store
.get_active_job()
.expect("active job must exist")
.get_job_id();
self.job_id_to_target.insert(job_id, self.target);
}
}
self.share_accounting.flush_seen_shares();
self.chain_tip = Some(set_new_prev_hash.into());
Ok(())
}
pub fn validate_share(
&mut self,
share: SubmitSharesStandard,
) -> Result<ShareValidationResult, ShareValidationError> {
let job_id = share.job_id;
let is_active_job = self
.job_store
.get_active_job()
.is_some_and(|job| job.get_job_id() == job_id);
let is_past_job = self.job_store.get_past_job(job_id).is_some();
let is_stale_job = self.job_store.get_stale_job(job_id).is_some();
if is_stale_job {
self.share_accounting
.increment_rejected_shares(ERROR_CODE_SUBMIT_SHARES_STALE_SHARE);
return Err(ShareValidationError::Stale(
ERROR_CODE_SUBMIT_SHARES_STALE_SHARE,
));
}
if !is_active_job && !is_past_job && !is_stale_job {
self.share_accounting
.increment_rejected_shares(ERROR_CODE_SUBMIT_SHARES_INVALID_JOB_ID);
return Err(ShareValidationError::InvalidJobId(
ERROR_CODE_SUBMIT_SHARES_INVALID_JOB_ID,
));
}
let job = if is_active_job {
self.job_store
.get_active_job()
.expect("active job must exist")
} else if is_past_job {
self.job_store
.get_past_job(job_id)
.expect("past job must exist")
} else {
self.job_store
.get_stale_job(job_id)
.expect("stale job must exist")
};
let Some(job_target) = self.job_id_to_target.get(&job_id) else {
self.share_accounting
.increment_rejected_shares(ERROR_CODE_SUBMIT_SHARES_INVALID_JOB_ID);
return Err(ShareValidationError::InvalidJobId(
ERROR_CODE_SUBMIT_SHARES_INVALID_JOB_ID,
));
};
let merkle_root: [u8; 32] = job
.get_merkle_root()
.inner_as_ref()
.try_into()
.expect("merkle root must be 32 bytes");
let chain_tip = self
.chain_tip
.as_ref()
.ok_or(ShareValidationError::NoChainTip)?;
let prev_hash = chain_tip.prev_hash();
let nbits = CompactTarget::from_consensus(chain_tip.nbits());
let header = Header {
version: Version::from_consensus(share.version as i32),
prev_blockhash: u256_to_block_hash(prev_hash.clone()),
merkle_root: (*Hash::from_bytes_ref(&merkle_root)).into(),
time: share.ntime,
bits: nbits,
nonce: share.nonce,
};
let share_hash = header.block_hash();
let share_raw_hash: [u8; 32] = *share_hash.to_raw_hash().as_ref();
let share_hash_target = Target::from_le_bytes(share_raw_hash);
let share_hash_as_diff = share_hash_target.difficulty_float();
let network_target = Target::from_compact(nbits);
let share_hash_target_bytes = share_hash_target.to_be_bytes();
let job_target_bytes = job_target.to_be_bytes();
debug!(
"share validation \nshare:\t\t{}\njob target:\t{}\nnetwork target:\t{}",
bytes_to_hex(&share_hash_target_bytes),
bytes_to_hex(&job_target_bytes),
format!("{:x}", network_target)
);
if network_target.is_met_by(share_hash) {
if self
.share_accounting
.is_share_seen(share_hash.to_raw_hash())
{
self.share_accounting
.increment_rejected_shares(ERROR_CODE_SUBMIT_SHARES_DUPLICATE_SHARE);
return Err(ShareValidationError::DuplicateShare(
ERROR_CODE_SUBMIT_SHARES_DUPLICATE_SHARE,
));
}
self.share_accounting.update_share_accounting(
job_target.difficulty_float(),
share.sequence_number,
share_hash.to_raw_hash(),
);
self.share_accounting.increment_blocks_found();
self.share_accounting.mark_batch_acknowledged();
let op_pushbytes_pool_miner_tag = self
.job_factory
.op_pushbytes_pool_miner_tag()
.map_err(|_| ShareValidationError::InvalidCoinbase)?;
let mut script_sig = job.get_template().coinbase_prefix.to_vec();
script_sig.extend(op_pushbytes_pool_miner_tag);
script_sig.push(self.extranonce_prefix.len() as u8); script_sig.extend(job.get_extranonce_prefix());
let tx_in = TxIn {
previous_output: OutPoint::null(),
script_sig: script_sig.into(),
sequence: Sequence(job.get_template().coinbase_tx_input_sequence),
witness: Witness::from(vec![vec![0; 32]]),
};
let coinbase = Transaction {
version: TxVersion::non_standard(job.get_template().coinbase_tx_version as i32),
lock_time: LockTime::from_consensus(job.get_template().coinbase_tx_locktime),
input: vec![tx_in],
output: job.get_coinbase_outputs().to_vec(),
};
let mut serialized_coinbase = Vec::new();
coinbase
.consensus_encode(&mut serialized_coinbase)
.map_err(|_| ShareValidationError::InvalidCoinbase)?;
return Ok(ShareValidationResult::BlockFound(
share_hash.to_raw_hash(),
Some(job.get_template().template_id),
serialized_coinbase,
));
}
if share_hash_target <= *job_target {
if self
.share_accounting
.is_share_seen(share_hash.to_raw_hash())
{
self.share_accounting
.increment_rejected_shares(ERROR_CODE_SUBMIT_SHARES_DUPLICATE_SHARE);
return Err(ShareValidationError::DuplicateShare(
ERROR_CODE_SUBMIT_SHARES_DUPLICATE_SHARE,
));
}
self.share_accounting.update_share_accounting(
job_target.difficulty_float(),
share.sequence_number,
share_hash.to_raw_hash(),
);
self.share_accounting.update_best_diff(share_hash_as_diff);
Ok(ShareValidationResult::Valid(share_hash.to_raw_hash()))
} else {
self.share_accounting
.increment_rejected_shares(ERROR_CODE_SUBMIT_SHARES_DIFFICULTY_TOO_LOW);
Err(ShareValidationError::DoesNotMeetTarget(
ERROR_CODE_SUBMIT_SHARES_DIFFICULTY_TOO_LOW,
))
}
}
}
#[cfg(test)]
mod tests {
use crate::{
chain_tip::ChainTip,
extranonce_manager::{AllocatedExtranoncePrefix, ExtranoncePrefix, ExtranoncePrefixError},
server::{
error::StandardChannelError,
jobs::{
job_store::{DefaultJobStore, JobStore},
standard::StandardJob,
},
share_accounting::{ShareValidationError, ShareValidationResult},
standard::StandardChannel,
},
};
use binary_sv2::Sv2Option;
use bitcoin::{transaction::TxOut, Amount, ScriptBuf, Target};
use mining_sv2::{
NewMiningJob, SubmitSharesStandard, ERROR_CODE_SUBMIT_SHARES_DIFFICULTY_TOO_LOW,
};
use std::convert::TryInto;
use template_distribution_sv2::{NewTemplate, SetNewPrevHash as SetNewPrevHashTdp};
const SATS_AVAILABLE_IN_TEMPLATE: u64 = 5000000000;
#[test]
fn test_future_job_activation_flow() {
let standard_channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let nominal_hashrate = 10.0;
let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut standard_channel = StandardChannel::new(
standard_channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let template = NewTemplate {
template_id: 1,
future_template: true,
version: 536870912,
coinbase_tx_version: 2,
coinbase_prefix: vec![2, 159, 0, 0].try_into().unwrap(),
coinbase_tx_input_sequence: 4294967294,
coinbase_tx_value_remaining: SATS_AVAILABLE_IN_TEMPLATE,
coinbase_tx_outputs_count: 1,
coinbase_tx_outputs: vec![
0, 0, 0, 0, 0, 0, 0, 0, 38, 106, 36, 170, 33, 169, 237, 226, 246, 28, 63, 113, 209,
222, 253, 63, 169, 153, 223, 163, 105, 83, 117, 92, 105, 6, 137, 121, 153, 98, 180,
139, 235, 216, 54, 151, 78, 140, 249,
]
.try_into()
.unwrap(),
coinbase_tx_locktime: 158,
merkle_path: vec![].try_into().unwrap(),
};
let pubkey_hash = [
235, 225, 183, 220, 194, 147, 204, 170, 14, 231, 67, 168, 111, 137, 223, 130, 88, 194,
8, 252,
];
let mut script_bytes = vec![0]; script_bytes.push(20); script_bytes.extend_from_slice(&pubkey_hash);
let script = ScriptBuf::from(script_bytes);
let coinbase_reward_outputs = vec![TxOut {
value: Amount::from_sat(SATS_AVAILABLE_IN_TEMPLATE),
script_pubkey: script,
}];
assert!(!standard_channel.job_store.has_future_jobs());
standard_channel
.on_new_template(template.clone(), coinbase_reward_outputs)
.unwrap();
let expected_future_standard_job = NewMiningJob {
channel_id: standard_channel_id,
job_id: 1,
merkle_root: [
213, 241, 108, 144, 69, 96, 29, 8, 222, 2, 135, 14, 213, 87, 81, 21, 140, 98, 42,
221, 221, 174, 219, 248, 106, 52, 168, 88, 18, 146, 186, 71,
]
.into(),
version: 536870912,
min_ntime: Sv2Option::new(None),
};
let future_standard_job_from_channel = standard_channel.get_future_job(1).unwrap();
assert_eq!(
future_standard_job_from_channel.get_job_message(),
&expected_future_standard_job
);
let ntime = 1747092633;
let set_new_prev_hash = SetNewPrevHashTdp {
template_id: template.template_id,
prev_hash: [
200, 53, 253, 129, 214, 31, 43, 84, 179, 58, 58, 76, 128, 213, 24, 53, 38, 144,
205, 88, 172, 20, 251, 22, 217, 141, 21, 221, 21, 0, 0, 0,
]
.into(),
header_timestamp: ntime,
n_bits: 503543726,
target: [
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
174, 119, 3, 0, 0,
]
.into(),
};
standard_channel
.on_set_new_prev_hash(set_new_prev_hash)
.unwrap();
let mut previously_future_job = future_standard_job_from_channel.clone();
previously_future_job.activate(ntime);
let activated_job = standard_channel.get_active_job().unwrap();
assert_eq!(
activated_job.get_job_message(),
previously_future_job.get_job_message()
);
}
#[test]
fn test_non_future_job_creation_flow() {
let standard_channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let nominal_hashrate = 10.0;
let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut standard_channel = StandardChannel::new(
standard_channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let ntime = 1747092633;
let prev_hash = [
200, 53, 253, 129, 214, 31, 43, 84, 179, 58, 58, 76, 128, 213, 24, 53, 38, 144, 205,
88, 172, 20, 251, 22, 217, 141, 21, 221, 21, 0, 0, 0,
]
.into();
let nbits = 503543726;
let chain_tip = ChainTip::new(prev_hash, nbits, ntime);
let template = NewTemplate {
template_id: 1,
future_template: false,
version: 536870912,
coinbase_tx_version: 2,
coinbase_prefix: vec![2, 159, 0, 0].try_into().unwrap(),
coinbase_tx_input_sequence: 4294967294,
coinbase_tx_value_remaining: SATS_AVAILABLE_IN_TEMPLATE,
coinbase_tx_outputs_count: 1,
coinbase_tx_outputs: vec![
0, 0, 0, 0, 0, 0, 0, 0, 38, 106, 36, 170, 33, 169, 237, 226, 246, 28, 63, 113, 209,
222, 253, 63, 169, 153, 223, 163, 105, 83, 117, 92, 105, 6, 137, 121, 153, 98, 180,
139, 235, 216, 54, 151, 78, 140, 249,
]
.try_into()
.unwrap(),
coinbase_tx_locktime: 158,
merkle_path: vec![].try_into().unwrap(),
};
let pubkey_hash = [
235, 225, 183, 220, 194, 147, 204, 170, 14, 231, 67, 168, 111, 137, 223, 130, 88, 194,
8, 252,
];
let mut script_bytes = vec![0]; script_bytes.push(20); script_bytes.extend_from_slice(&pubkey_hash);
let script = ScriptBuf::from(script_bytes);
let coinbase_reward_outputs = vec![TxOut {
value: Amount::from_sat(SATS_AVAILABLE_IN_TEMPLATE),
script_pubkey: script,
}];
standard_channel.set_chain_tip(chain_tip);
standard_channel
.on_new_template(template.clone(), coinbase_reward_outputs)
.unwrap();
let expected_active_standard_job = NewMiningJob {
channel_id: standard_channel_id,
job_id: 1,
merkle_root: [
213, 241, 108, 144, 69, 96, 29, 8, 222, 2, 135, 14, 213, 87, 81, 21, 140, 98, 42,
221, 221, 174, 219, 248, 106, 52, 168, 88, 18, 146, 186, 71,
]
.into(),
version: 536870912,
min_ntime: Sv2Option::new(Some(ntime)),
};
let active_standard_job_from_channel = standard_channel.get_active_job().unwrap().clone();
assert_eq!(
active_standard_job_from_channel.get_job_message(),
&expected_active_standard_job
);
}
#[test]
fn test_share_validation_block_found() {
let standard_channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let nominal_hashrate = 1.0;
let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut standard_channel = StandardChannel::new(
standard_channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let template = NewTemplate {
template_id: 1,
future_template: false,
version: 536870912,
coinbase_tx_version: 2,
coinbase_prefix: vec![2, 159, 0, 0].try_into().unwrap(),
coinbase_tx_input_sequence: 4294967294,
coinbase_tx_value_remaining: SATS_AVAILABLE_IN_TEMPLATE,
coinbase_tx_outputs_count: 1,
coinbase_tx_outputs: vec![
0, 0, 0, 0, 0, 0, 0, 0, 38, 106, 36, 170, 33, 169, 237, 226, 246, 28, 63, 113, 209,
222, 253, 63, 169, 153, 223, 163, 105, 83, 117, 92, 105, 6, 137, 121, 153, 98, 180,
139, 235, 216, 54, 151, 78, 140, 249,
]
.try_into()
.unwrap(),
coinbase_tx_locktime: 158,
merkle_path: vec![].try_into().unwrap(),
};
let pubkey_hash = [
235, 225, 183, 220, 194, 147, 204, 170, 14, 231, 67, 168, 111, 137, 223, 130, 88, 194,
8, 252,
];
let mut script_bytes = vec![0]; script_bytes.push(20); script_bytes.extend_from_slice(&pubkey_hash);
let script = ScriptBuf::from(script_bytes);
let coinbase_reward_outputs = vec![TxOut {
value: Amount::from_sat(SATS_AVAILABLE_IN_TEMPLATE),
script_pubkey: script,
}];
let ntime = 1745596910;
let prev_hash = [
251, 175, 106, 40, 35, 87, 122, 90, 58, 51, 78, 32, 202, 236, 228, 36, 154, 174, 206,
144, 147, 195, 21, 224, 195, 103, 214, 189, 51, 190, 24, 98,
]
.into();
let n_bits = 545259519;
let chain_tip = ChainTip::new(prev_hash, n_bits, ntime);
standard_channel.set_chain_tip(chain_tip);
standard_channel
.on_new_template(template.clone(), coinbase_reward_outputs)
.unwrap();
let active_standard_job = standard_channel.get_active_job().unwrap();
let share_valid_block = SubmitSharesStandard {
channel_id: standard_channel_id,
sequence_number: 0,
job_id: active_standard_job.get_job_id(),
nonce: 0,
ntime: 1745596932,
version: 536870912,
};
let res = standard_channel.validate_share(share_valid_block.clone());
assert!(matches!(
res,
Ok(ShareValidationResult::BlockFound(_, _, _))
));
assert_eq!(
standard_channel.get_share_accounting().get_blocks_found(),
1
);
let res = standard_channel.validate_share(share_valid_block);
assert!(matches!(
res.unwrap_err(),
ShareValidationError::DuplicateShare(_)
));
assert_eq!(
standard_channel.get_share_accounting().get_blocks_found(),
1
);
}
#[test]
fn test_share_validation_does_not_meet_target() {
let standard_channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let nominal_hashrate = 100.0; let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut standard_channel = StandardChannel::new(
standard_channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let template = NewTemplate {
template_id: 1,
future_template: false,
version: 536870912,
coinbase_tx_version: 2,
coinbase_prefix: vec![2, 159, 0, 0].try_into().unwrap(),
coinbase_tx_input_sequence: 4294967294,
coinbase_tx_value_remaining: SATS_AVAILABLE_IN_TEMPLATE,
coinbase_tx_outputs_count: 1,
coinbase_tx_outputs: vec![
0, 0, 0, 0, 0, 0, 0, 0, 38, 106, 36, 170, 33, 169, 237, 226, 246, 28, 63, 113, 209,
222, 253, 63, 169, 153, 223, 163, 105, 83, 117, 92, 105, 6, 137, 121, 153, 98, 180,
139, 235, 216, 54, 151, 78, 140, 249,
]
.try_into()
.unwrap(),
coinbase_tx_locktime: 158,
merkle_path: vec![].try_into().unwrap(),
};
let pubkey_hash = [
235, 225, 183, 220, 194, 147, 204, 170, 14, 231, 67, 168, 111, 137, 223, 130, 88, 194,
8, 252,
];
let mut script_bytes = vec![0]; script_bytes.push(20); script_bytes.extend_from_slice(&pubkey_hash);
let script = ScriptBuf::from(script_bytes);
let coinbase_reward_outputs = vec![TxOut {
value: Amount::from_sat(SATS_AVAILABLE_IN_TEMPLATE),
script_pubkey: script,
}];
let ntime = 1745596910;
let prev_hash = [
154, 124, 239, 231, 221, 122, 160, 173, 164, 175, 87, 33, 74, 214, 191, 107, 73, 34, 0,
162, 227, 16, 44, 40, 33, 73, 0, 0, 0, 0, 0, 0,
]
.into();
let n_bits = 453040064;
let chain_tip = ChainTip::new(prev_hash, n_bits, ntime);
standard_channel.set_chain_tip(chain_tip);
standard_channel
.on_new_template(template.clone(), coinbase_reward_outputs)
.unwrap();
let active_standard_job = standard_channel.get_active_job().unwrap();
let share_low_diff = SubmitSharesStandard {
channel_id: standard_channel_id,
sequence_number: 0,
job_id: active_standard_job.get_job_id(),
nonce: 3,
ntime: 1745596932,
version: 536870912,
};
let res = standard_channel.validate_share(share_low_diff);
assert!(matches!(
res.unwrap_err(),
ShareValidationError::DoesNotMeetTarget(_)
));
assert_eq!(
standard_channel
.get_share_accounting()
.get_rejected_shares_error_count(ERROR_CODE_SUBMIT_SHARES_DIFFICULTY_TOO_LOW),
1
);
}
#[test]
fn test_share_validation_valid_share() {
let standard_channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let nominal_hashrate = 1_000.0; let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut standard_channel = StandardChannel::new(
standard_channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let template = NewTemplate {
template_id: 1,
future_template: false,
version: 536870912,
coinbase_tx_version: 2,
coinbase_prefix: vec![2, 159, 0, 0].try_into().unwrap(),
coinbase_tx_input_sequence: 4294967294,
coinbase_tx_value_remaining: SATS_AVAILABLE_IN_TEMPLATE,
coinbase_tx_outputs_count: 1,
coinbase_tx_outputs: vec![
0, 0, 0, 0, 0, 0, 0, 0, 38, 106, 36, 170, 33, 169, 237, 226, 246, 28, 63, 113, 209,
222, 253, 63, 169, 153, 223, 163, 105, 83, 117, 92, 105, 6, 137, 121, 153, 98, 180,
139, 235, 216, 54, 151, 78, 140, 249,
]
.try_into()
.unwrap(),
coinbase_tx_locktime: 158,
merkle_path: vec![].try_into().unwrap(),
};
let pubkey_hash = [
235, 225, 183, 220, 194, 147, 204, 170, 14, 231, 67, 168, 111, 137, 223, 130, 88, 194,
8, 252,
];
let mut script_bytes = vec![0]; script_bytes.push(20); script_bytes.extend_from_slice(&pubkey_hash);
let script = ScriptBuf::from(script_bytes);
let coinbase_reward_outputs = vec![TxOut {
value: Amount::from_sat(SATS_AVAILABLE_IN_TEMPLATE),
script_pubkey: script,
}];
let ntime = 1745596910;
let prev_hash = [
154, 124, 239, 231, 221, 122, 160, 173, 164, 175, 87, 33, 74, 214, 191, 107, 73, 34, 0,
162, 227, 16, 44, 40, 33, 73, 0, 0, 0, 0, 0, 0,
]
.into();
let n_bits = 453040064;
let chain_tip = ChainTip::new(prev_hash, n_bits, ntime);
standard_channel.set_chain_tip(chain_tip);
standard_channel
.on_new_template(template.clone(), coinbase_reward_outputs)
.unwrap();
let valid_share = SubmitSharesStandard {
channel_id: standard_channel_id,
sequence_number: 1,
job_id: 1,
nonce: 134870,
ntime: 1745611105,
version: 536870912,
};
let res = standard_channel.validate_share(valid_share);
assert!(matches!(res, Ok(ShareValidationResult::Valid(_))));
}
#[test]
fn test_new_clamps_target_to_max_target() {
let channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [0, 0, 0, 1].to_vec();
let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let very_small_hashrate = 0.1;
let job_store = DefaultJobStore::<StandardJob>::new();
let not_so_permissive_max_target = Target::from_le_bytes([
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0x00,
]);
let channel = StandardChannel::new(
channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix).unwrap(),
not_so_permissive_max_target,
very_small_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
assert_eq!(
channel.get_requested_max_target(),
¬_so_permissive_max_target
);
assert_eq!(channel.get_target(), ¬_so_permissive_max_target);
}
#[test]
fn test_update_channel() {
let channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let expected_share_per_minute = 1.0;
let initial_hashrate = 10.0;
let share_batch_size = 100;
let job_store = DefaultJobStore::<StandardJob>::new();
let max_target = Target::from_le_bytes([0xff; 32]);
let mut channel = StandardChannel::new(
channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix).unwrap(),
max_target,
initial_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let initial_target = channel.get_target().clone();
let new_hashrate = 100.0;
channel
.update_channel(new_hashrate, Some(max_target))
.unwrap();
let new_target = channel.get_target().clone();
assert_ne!(initial_target, new_target);
assert_eq!(channel.get_nominal_hashrate(), new_hashrate);
let result = channel.update_channel(-1.0, Some(max_target));
assert!(result.is_err());
assert!(matches!(
result,
Err(StandardChannelError::UpdateChannelInvalidNominalHashrate(_))
));
let not_so_permissive_max_target = Target::from_le_bytes([
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff,
0xff, 0xff, 0xff, 0x00,
]);
let very_small_hashrate = 0.1;
let result =
channel.update_channel(very_small_hashrate, Some(not_so_permissive_max_target));
assert!(result.is_ok());
assert_eq!(channel.get_target(), ¬_so_permissive_max_target);
let sufficiently_big_hashrate = 1000.0;
let result = channel.update_channel(
sufficiently_big_hashrate,
Some(not_so_permissive_max_target),
);
assert!(result.is_ok());
}
#[test]
fn test_update_extranonce_prefix() {
let channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let expected_share_per_minute = 1.0;
let nominal_hashrate = 1_000.0;
let share_batch_size = 100;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut channel = StandardChannel::new(
channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let current_extranonce_prefix = channel.get_extranonce_prefix();
assert_eq!(current_extranonce_prefix, extranonce_prefix.as_slice());
let new_extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2,
]
.to_vec();
channel
.set_extranonce_prefix(
AllocatedExtranoncePrefix::for_test(new_extranonce_prefix.clone()).unwrap(),
)
.unwrap();
let current_extranonce_prefix = channel.get_extranonce_prefix();
assert_eq!(current_extranonce_prefix, new_extranonce_prefix.as_slice());
let new_extranonce_prefix_too_long = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 2, 1,
]
.to_vec();
assert!(matches!(
ExtranoncePrefix::from_wire(new_extranonce_prefix_too_long),
Err(ExtranoncePrefixError::ExceedsMaxLength)
));
}
#[test]
fn test_set_new_prev_hash_without_future_jobs_preserves_state() {
let standard_channel_id = 1;
let user_identity = "user_identity".to_string();
let extranonce_prefix = [
83, 116, 114, 97, 116, 117, 109, 32, 86, 50, 32, 83, 82, 73, 32, 80, 111, 111, 108, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1,
]
.to_vec();
let max_target = Target::from_le_bytes([0xff; 32]);
let nominal_hashrate = 100.0;
let share_batch_size = 100;
let expected_share_per_minute = 1.0;
let job_store = DefaultJobStore::<StandardJob>::new();
let mut standard_channel = StandardChannel::new(
standard_channel_id,
user_identity,
ExtranoncePrefix::from_wire(extranonce_prefix.clone()).unwrap(),
max_target,
nominal_hashrate,
share_batch_size,
expected_share_per_minute,
job_store,
None,
None,
)
.unwrap();
let template = NewTemplate {
template_id: 1,
future_template: false,
version: 536870912,
coinbase_tx_version: 2,
coinbase_prefix: vec![2, 159, 0, 0].try_into().unwrap(),
coinbase_tx_input_sequence: 4294967294,
coinbase_tx_value_remaining: SATS_AVAILABLE_IN_TEMPLATE,
coinbase_tx_outputs_count: 1,
coinbase_tx_outputs: vec![
0, 0, 0, 0, 0, 0, 0, 0, 38, 106, 36, 170, 33, 169, 237, 226, 246, 28, 63, 113, 209,
222, 253, 63, 169, 153, 223, 163, 105, 83, 117, 92, 105, 6, 137, 121, 153, 98, 180,
139, 235, 216, 54, 151, 78, 140, 249,
]
.try_into()
.unwrap(),
coinbase_tx_locktime: 158,
merkle_path: vec![].try_into().unwrap(),
};
let pubkey_hash = [
235, 225, 183, 220, 194, 147, 204, 170, 14, 231, 67, 168, 111, 137, 223, 130, 88, 194,
8, 252,
];
let mut script_bytes = vec![0];
script_bytes.push(20);
script_bytes.extend_from_slice(&pubkey_hash);
let script = ScriptBuf::from(script_bytes);
let coinbase_reward_outputs = vec![TxOut {
value: Amount::from_sat(SATS_AVAILABLE_IN_TEMPLATE),
script_pubkey: script,
}];
let ntime = 1745596910;
let prev_hash = [
154, 124, 239, 231, 221, 122, 160, 173, 164, 175, 87, 33, 74, 214, 191, 107, 73, 34, 0,
162, 227, 16, 44, 40, 33, 73, 0, 0, 0, 0, 0, 0,
]
.into();
let n_bits = 453040064;
let chain_tip = ChainTip::new(prev_hash, n_bits, ntime);
standard_channel.set_chain_tip(chain_tip);
standard_channel
.on_new_template(template.clone(), coinbase_reward_outputs)
.unwrap();
let active_standard_job = standard_channel.get_active_job().unwrap();
let active_job_id = active_standard_job.get_job_id();
assert!(!standard_channel.job_store.has_future_jobs());
let snph = SetNewPrevHashTdp {
template_id: 999,
prev_hash: [
200, 53, 253, 129, 214, 31, 43, 84, 179, 58, 58, 76, 128, 213, 24, 53, 38, 144,
205, 88, 172, 20, 251, 22, 217, 141, 21, 221, 21, 0, 0, 0,
]
.into(),
header_timestamp: ntime + 600,
n_bits,
target: [0xff; 32].into(),
};
let res = standard_channel.on_set_new_prev_hash(snph);
assert!(matches!(res, Err(StandardChannelError::TemplateIdNotFound)));
let share_low_diff = SubmitSharesStandard {
channel_id: standard_channel_id,
sequence_number: 0,
job_id: active_job_id,
nonce: 3,
ntime: 1745596932,
version: 536870912,
};
let res = standard_channel.validate_share(share_low_diff);
assert!(matches!(
res.unwrap_err(),
ShareValidationError::DoesNotMeetTarget(_)
));
}
}