use crate::{Opcode, Operand, Registers, Stack};
use console::{
network::prelude::*,
program::{Literal, LiteralType, Plaintext, PlaintextType, Register, RegisterType, Value},
};
pub type HashBHP256<N> = HashInstruction<N, { Hasher::BHP256 as u8 }>;
pub type HashBHP512<N> = HashInstruction<N, { Hasher::BHP512 as u8 }>;
pub type HashBHP768<N> = HashInstruction<N, { Hasher::BHP768 as u8 }>;
pub type HashBHP1024<N> = HashInstruction<N, { Hasher::BHP1024 as u8 }>;
pub type HashPED64<N> = HashInstruction<N, { Hasher::PED64 as u8 }>;
pub type HashPED128<N> = HashInstruction<N, { Hasher::PED128 as u8 }>;
pub type HashPSD2<N> = HashInstruction<N, { Hasher::PSD2 as u8 }>;
pub type HashPSD4<N> = HashInstruction<N, { Hasher::PSD4 as u8 }>;
pub type HashPSD8<N> = HashInstruction<N, { Hasher::PSD8 as u8 }>;
enum Hasher {
BHP256,
BHP512,
BHP768,
BHP1024,
PED64,
PED128,
PSD2,
PSD4,
PSD8,
}
#[derive(Clone, PartialEq, Eq, Hash)]
pub struct HashInstruction<N: Network, const VARIANT: u8> {
operands: Vec<Operand<N>>,
destination: Register<N>,
}
impl<N: Network, const VARIANT: u8> HashInstruction<N, VARIANT> {
#[inline]
pub const fn opcode() -> Opcode {
match VARIANT {
0 => Opcode::Hash("hash.bhp256"),
1 => Opcode::Hash("hash.bhp512"),
2 => Opcode::Hash("hash.bhp768"),
3 => Opcode::Hash("hash.bhp1024"),
4 => Opcode::Hash("hash.ped64"),
5 => Opcode::Hash("hash.ped128"),
6 => Opcode::Hash("hash.psd2"),
7 => Opcode::Hash("hash.psd4"),
8 => Opcode::Hash("hash.psd8"),
_ => panic!("Invalid 'hash' instruction opcode"),
}
}
#[inline]
pub fn operands(&self) -> &[Operand<N>] {
debug_assert!(self.operands.len() == 1, "Hash operation must have one operand");
&self.operands
}
#[inline]
pub fn destinations(&self) -> Vec<Register<N>> {
vec![self.destination.clone()]
}
}
impl<N: Network, const VARIANT: u8> HashInstruction<N, VARIANT> {
#[inline]
pub fn evaluate<A: circuit::Aleo<Network = N>>(
&self,
stack: &Stack<N>,
registers: &mut Registers<N, A>,
) -> Result<()> {
if self.operands.len() != 1 {
bail!("Instruction '{}' expects 1 operands, found {} operands", Self::opcode(), self.operands.len())
}
let input = registers.load(stack, &self.operands[0])?;
let output = match VARIANT {
0 => N::hash_bhp256(&input.to_bits_le())?,
1 => N::hash_bhp512(&input.to_bits_le())?,
2 => N::hash_bhp768(&input.to_bits_le())?,
3 => N::hash_bhp1024(&input.to_bits_le())?,
4 => N::hash_ped64(&input.to_bits_le())?,
5 => N::hash_ped128(&input.to_bits_le())?,
6 => N::hash_psd2(&input.to_fields()?)?,
7 => N::hash_psd4(&input.to_fields()?)?,
8 => N::hash_psd8(&input.to_fields()?)?,
_ => bail!("Invalid 'hash' variant: {VARIANT}"),
};
registers.store(stack, &self.destination, Value::Plaintext(Plaintext::from(Literal::Field(output))))
}
#[inline]
pub fn execute<A: circuit::Aleo<Network = N>>(
&self,
stack: &Stack<N>,
registers: &mut Registers<N, A>,
) -> Result<()> {
use circuit::{ToBits, ToFields};
if self.operands.len() != 1 {
bail!("Instruction '{}' expects 1 operands, found {} operands", Self::opcode(), self.operands.len())
}
let input = registers.load_circuit(stack, &self.operands[0])?;
let output = match VARIANT {
0 => A::hash_bhp256(&input.to_bits_le()),
1 => A::hash_bhp512(&input.to_bits_le()),
2 => A::hash_bhp768(&input.to_bits_le()),
3 => A::hash_bhp1024(&input.to_bits_le()),
4 => A::hash_ped64(&input.to_bits_le()),
5 => A::hash_ped128(&input.to_bits_le()),
6 => A::hash_psd2(&input.to_fields()),
7 => A::hash_psd4(&input.to_fields()),
8 => A::hash_psd8(&input.to_fields()),
_ => bail!("Invalid 'hash' variant: {VARIANT}"),
};
let output =
circuit::Value::Plaintext(circuit::Plaintext::Literal(circuit::Literal::Field(output), Default::default()));
registers.store_circuit(stack, &self.destination, output)
}
#[inline]
pub fn output_types(&self, _stack: &Stack<N>, input_types: &[RegisterType<N>]) -> Result<Vec<RegisterType<N>>> {
if input_types.len() != 1 {
bail!("Instruction '{}' expects 1 inputs, found {} inputs", Self::opcode(), input_types.len())
}
if self.operands.len() != 1 {
bail!("Instruction '{}' expects 1 operands, found {} operands", Self::opcode(), self.operands.len())
}
match VARIANT {
0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 => {
Ok(vec![RegisterType::Plaintext(PlaintextType::Literal(LiteralType::Field))])
}
_ => bail!("Invalid 'hash' variant: {VARIANT}"),
}
}
}
impl<N: Network, const VARIANT: u8> Parser for HashInstruction<N, VARIANT> {
#[inline]
fn parse(string: &str) -> ParserResult<Self> {
let (string, _) = tag(*Self::opcode())(string)?;
let (string, _) = Sanitizer::parse_whitespaces(string)?;
let (string, operand) = Operand::parse(string)?;
let (string, _) = Sanitizer::parse_whitespaces(string)?;
let (string, _) = tag("into")(string)?;
let (string, _) = Sanitizer::parse_whitespaces(string)?;
let (string, destination) = Register::parse(string)?;
Ok((string, Self { operands: vec![operand], destination }))
}
}
impl<N: Network, const VARIANT: u8> FromStr for HashInstruction<N, VARIANT> {
type Err = Error;
#[inline]
fn from_str(string: &str) -> Result<Self> {
match Self::parse(string) {
Ok((remainder, object)) => {
ensure!(remainder.is_empty(), "Failed to parse string. Found invalid character in: \"{remainder}\"");
Ok(object)
}
Err(error) => bail!("Failed to parse string. {error}"),
}
}
}
impl<N: Network, const VARIANT: u8> Debug for HashInstruction<N, VARIANT> {
fn fmt(&self, f: &mut Formatter) -> fmt::Result {
Display::fmt(self, f)
}
}
impl<N: Network, const VARIANT: u8> Display for HashInstruction<N, VARIANT> {
fn fmt(&self, f: &mut Formatter) -> fmt::Result {
if self.operands.len() != 1 {
eprintln!("The number of operands must be 1, found {}", self.operands.len());
return Err(fmt::Error);
}
write!(f, "{} ", Self::opcode())?;
self.operands.iter().try_for_each(|operand| write!(f, "{} ", operand))?;
write!(f, "into {}", self.destination)
}
}
impl<N: Network, const VARIANT: u8> FromBytes for HashInstruction<N, VARIANT> {
fn read_le<R: Read>(mut reader: R) -> IoResult<Self> {
let operands = vec![Operand::read_le(&mut reader)?];
let destination = Register::read_le(&mut reader)?;
Ok(Self { operands, destination })
}
}
impl<N: Network, const VARIANT: u8> ToBytes for HashInstruction<N, VARIANT> {
fn write_le<W: Write>(&self, mut writer: W) -> IoResult<()> {
if self.operands.len() != 1 {
return Err(error(format!("The number of operands must be 1, found {}", self.operands.len())));
}
self.operands[0].write_le(&mut writer)?;
self.destination.write_le(&mut writer)
}
}
#[cfg(test)]
mod tests {
use super::*;
use console::network::Testnet3;
type CurrentNetwork = Testnet3;
#[test]
fn test_parse() {
let (string, hash) = HashBHP512::<CurrentNetwork>::parse("hash.bhp512 r0 into r1").unwrap();
assert!(string.is_empty(), "Parser did not consume all of the string: '{string}'");
assert_eq!(hash.operands.len(), 1, "The number of operands is incorrect");
assert_eq!(hash.operands[0], Operand::Register(Register::Locator(0)), "The first operand is incorrect");
assert_eq!(hash.destination, Register::Locator(1), "The destination register is incorrect");
}
}