Add Python bindings to chiavdf and chiapos

Cargo.toml

@@ -139,3 +139,5 @@ zstd = "0.13.2"
 blocking-threadpool = "1.0.1"
 libfuzzer-sys = "0.4"
 wasm-bindgen = "0.2.93"
+chiavdf = "1.1.5"
+chiapos = "2.0.5"

crates/chia-consensus/Cargo.toml

@@ -29,6 +29,8 @@ chia-puzzles = { workspace = true }
 chia-bls = { workspace = true }
 hex-literal = { workspace = true }
 thiserror = { workspace = true }
+chiavdf = { workspace = true }
+chiapos = { workspace = true }
 
 [dev-dependencies]
 num-traits = { workspace = true }

crates/chia-consensus/src/lib.rs

@@ -8,5 +8,7 @@ pub mod gen;
 pub mod generator_rom;
 pub mod merkle_set;
 pub mod merkle_tree;
+pub mod proof_of_space;
 pub mod spendbundle_conditions;
 pub mod spendbundle_validation;
+pub mod vdf;

crates/chia-consensus/src/proof_of_space.rs

@@ -0,0 +1,17 @@
+use chia_protocol::{Bytes32, ProofOfSpace};
+use chiapos::validate_proof;
+
+pub fn get_quality_string(pos: &ProofOfSpace, plot_id: Bytes32) -> Option<Bytes32> {
+    let mut quality = [0; 32];
+    if validate_proof(
+        &plot_id.to_bytes(),
+        pos.size,
+        &pos.challenge.to_bytes(),
+        &pos.proof,
+        &mut quality,
+    ) {
+        Some(Bytes32::from(quality))
+    } else {
+        None
+    }
+}

crates/chia-consensus/src/vdf.rs

@@ -0,0 +1,28 @@
+use chia_protocol::{ClassgroupElement, VDFInfo, VDFProof};
+use chiavdf::{create_discriminant, verify_n_wesolowski};
+
+use crate::consensus_constants::ConsensusConstants;
+
+pub fn validate_vdf_proof(
+    proof: &VDFProof,
+    input_el: &ClassgroupElement,
+    info: &VDFInfo,
+    constants: &ConsensusConstants,
+) -> bool {
+    if proof.witness_type + 1 > constants.max_vdf_witness_size {
+        return false;
+    }
+
+    let mut discriminant = vec![0; constants.discriminant_size_bits as usize / 8];
+    create_discriminant(&info.challenge, &mut discriminant);
+
+    let proof_buf = [info.output.data.as_slice(), proof.witness.as_slice()].concat();
+
+    verify_n_wesolowski(
+        &discriminant,
+        &input_el.data,
+        &proof_buf,
+        info.number_of_iterations,
+        proof.witness_type as u64,
+    )
+}

wheel/generate_type_stubs.py

@@ -267,6 +267,9 @@ class _Unspec:
 def solution_generator(spends: Sequence[Tuple[Coin, bytes, bytes]]) -> bytes: ...
 def solution_generator_backrefs(spends: Sequence[Tuple[Coin, bytes, bytes]]) -> bytes: ...
 
+def validate_vdf_proof(proof: VDFProof, input_el: ClassgroupElement, info: VDFInfo, constants: ConsensusConstants) -> bool: ...
+def get_quality_string(pos: ProofOfSpace, plot_id: bytes32) -> Optional[bytes32]: ...
+
 def compute_merkle_set_root(items: Sequence[bytes]) -> bytes: ...
 
 def supports_fast_forward(spend: CoinSpend) -> bool : ...

wheel/python/chia_rs/chia_rs.pyi

@@ -16,6 +16,9 @@ class _Unspec:
 def solution_generator(spends: Sequence[Tuple[Coin, bytes, bytes]]) -> bytes: ...
 def solution_generator_backrefs(spends: Sequence[Tuple[Coin, bytes, bytes]]) -> bytes: ...
 
+def validate_vdf_proof(proof: VDFProof, input_el: ClassgroupElement, info: VDFInfo, constants: ConsensusConstants) -> bool: ...
+def get_quality_string(pos: ProofOfSpace, plot_id: bytes32) -> Optional[bytes32]: ...
+
 def compute_merkle_set_root(items: Sequence[bytes]) -> bytes: ...
 
 def supports_fast_forward(spend: CoinSpend) -> bool : ...