Skip to content

[CHIA-3701] Check plots v2 #19982

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 2 commits into from
Sep 9, 2025
Merged

[CHIA-3701] Check plots v2 #19982

Changes from 1 commit
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
d483b7f
support v2 plots in check-plots
arvidn Aug 21, 2025
ad4ace9
review comments
arvidn Sep 4, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
33 changes: 27 additions & 6 deletions chia/plotting/check_plots.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,16 +3,20 @@
import concurrent.futures
import logging
from collections import Counter
from collections.abc import Sequence
from pathlib import Path
from threading import Lock
from time import monotonic, sleep
from typing import Optional
from typing import Optional, Union

from chia_rs import G1Element
from chia_rs.sized_bytes import bytes32
from chia_rs.sized_ints import uint8, uint32
from chiapos import Verifier

from chia.consensus.default_constants import DEFAULT_CONSTANTS
from chia.plotting.manager import PlotManager
from chia.plotting.prover import PlotVersion
from chia.plotting.util import (
PlotInfo,
PlotRefreshEvents,
Expand All @@ -22,6 +26,10 @@
get_plot_filenames,
parse_plot_info,
)
from chia.types.blockchain_format.proof_of_space import (
quality_for_partial_proof,
solve_proof,
)
from chia.util.bech32m import encode_puzzle_hash
from chia.util.config import load_config
from chia.util.cpu import available_logical_cores
Expand Down Expand Up @@ -168,12 +176,20 @@ def process_plot(plot_path: Path, plot_info: PlotInfo, num_start: int, num_end:

total_proofs = 0
caught_exception: bool = False
version = pr.get_version()
for i in range(num_start, num_end):
challenge = std_hash(i.to_bytes(32, "big"))
# these are either qualities (v1) or partial proofs (v2)
proofs: Sequence[Union[bytes32, bytes]]
# Some plot errors cause get_qualities_for_challenge to throw a RuntimeError
try:
quality_start_time = round(monotonic() * 1000)
qualities = pr.get_qualities_for_challenge(challenge)
if version == PlotVersion.V1:
proofs = pr.get_qualities_for_challenge(challenge)
else:
proofs = pr.get_partial_proofs_for_challenge(
challenge, DEFAULT_CONSTANTS.PLOT_DIFFICULTY_INITIAL
)
quality_spent_time = round(monotonic() * 1000) - quality_start_time
if quality_spent_time > 8000:
log.warning(
Expand Down Expand Up @@ -201,12 +217,17 @@ def process_plot(plot_path: Path, plot_info: PlotInfo, num_start: int, num_end:
caught_exception = True
break

for index, quality_str in enumerate(qualities):
for index, proof in enumerate(proofs):
# Other plot errors cause get_full_proof or validate_proof to throw an AssertionError
try:
proof_start_time = round(monotonic() * 1000)
# TODO : todo_v2_plots handle v2 plots
proof = pr.get_full_proof(challenge, index, parallel_read)
if version == PlotVersion.V1:
quality_str = bytes32(proof)
full_proof = pr.get_full_proof(challenge, index, parallel_read)
else:
quality_str = quality_for_partial_proof(proof, challenge)
full_proof = solve_proof(proof)

proof_spent_time = round(monotonic() * 1000) - proof_start_time
if proof_spent_time > 15000:
log.warning(
Expand All @@ -216,7 +237,7 @@ def process_plot(plot_path: Path, plot_info: PlotInfo, num_start: int, num_end:
else:
log.info(f"\tFinding proof took: {proof_spent_time} ms. Filepath: {plot_path}")

ver_quality_str = v.validate_proof(pr.get_id(), pr.get_size().size_v1, challenge, proof)
ver_quality_str = v.validate_proof(pr.get_id(), pr.get_size().size_v1, challenge, full_proof)
if quality_str == ver_quality_str:
total_proofs += 1
else:
Expand Down
Loading