benchmark_cchain_range.sh

#!/usr/bin/env bash

set -euo pipefail

# C-Chain Re-execution Benchmark and Chaos Test Script
#
# Usage:
#   ./benchmark_cchain_range.sh [test-name]
#
# To see available tests: use `help` as the test name or invoke
# without a test name and without required env vars.
#
# Test names configure defaults for S3 sources and block ranges.
# If running in chaos mode, test names also configure defaults for the VM Config
# and min/max wait times.
# All defaults can be overridden via environment variables.
#
# Note: chaos tests can only be run with firewood VM configs.
#
# Environment variables:
#   Data sources (provide S3 sources OR local paths):
#     BLOCK_DIR_SRC: S3 object key for blocks (triggers S3 import).
#     CURRENT_STATE_DIR_SRC: S3 object key for state (triggers S3 import).
#     BLOCK_DIR: Path to local block directory.
#     CURRENT_STATE_DIR: Path to local current state directory.
#
#   Required:
#     START_BLOCK: The starting block height (inclusive).
#     END_BLOCK: The ending block height (inclusive).
#
#   Optional (reexecution tests):
#     CONFIG: VM config preset (default, archive, firewood, firewood-archive).
#     LABELS: Comma-separated key=value pairs for metric labels.
#     BENCHMARK_OUTPUT_FILE: If set, benchmark output is also written to this file.
#     METRICS_SERVER_ENABLED: If set, enables the metrics server.
#     METRICS_SERVER_PORT: If set, determines the port the metrics server will listen to.
#     METRICS_COLLECTOR_ENABLED: If set, enables the metrics collector.
#     PUSH_POST_STATE: S3 destination to push current-state after execution.
#
#   Required (chaos tests):
#     CHAOS_MODE: Set to enable chaos test mode (e.g., CHAOS_MODE=1).
#     CONFIG: VM config preset (firewood or firewood-archive only).
#     MIN_WAIT_TIME: Minimum wait before crash (e.g., 120s).
#     MAX_WAIT_TIME: Maximum wait before crash (e.g., 150s).

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"

# CI-aware error function
error() {
    if [[ "${GITHUB_ACTIONS:-}" == "true" ]]; then
        echo "::error::$1"
    else
        echo "Error: $1" >&2
    fi
    exit 1
}

show_usage() {
    cat <<EOF
Usage: $0 [test-name]

Available tests:
  help                         - Show this help message

  default                      - Quick test run (blocks 101-200, hashdb)
  hashdb-101-250k              - Blocks 101-250k with hashdb
  hashdb-archive-101-250k      - Blocks 101-250k with hashdb archive
  hashdb-33m-33m500k           - Blocks 33m-33.5m with hashdb
  firewood-101-250k            - Blocks 101-250k with firewood
  firewood-archive-101-250k    - Blocks 101-250k with firewood archive
  firewood-33m-33m500k         - Blocks 33m-33.5m with firewood
  firewood-33m-40m             - Blocks 33m-40m with firewood
EOF
}

# Set defaults based on test name (if provided)
TEST_NAME="${1:-}"
if [[ -n "$TEST_NAME" ]]; then
    shift
    case "$TEST_NAME" in
        help)
            show_usage
            exit 0
            ;;
        default)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-200-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-hashdb-full-100}"
            START_BLOCK="${START_BLOCK:-101}"
            END_BLOCK="${END_BLOCK:-200}"
            ;;
        hashdb-101-250k)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-1m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-hashdb-full-100}"
            START_BLOCK="${START_BLOCK:-101}"
            END_BLOCK="${END_BLOCK:-250000}"
            ;;
        hashdb-archive-101-250k)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-1m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-hashdb-archive-100}"
            START_BLOCK="${START_BLOCK:-101}"
            END_BLOCK="${END_BLOCK:-250000}"
            CONFIG="${CONFIG:-archive}"
            ;;
        hashdb-33m-33m500k)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-30m-40m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-hashdb-full-33m}"
            START_BLOCK="${START_BLOCK:-33000001}"
            END_BLOCK="${END_BLOCK:-33500000}"
            ;;
        firewood-101-250k)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-1m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-firewood-100}"
            START_BLOCK="${START_BLOCK:-101}"
            END_BLOCK="${END_BLOCK:-250000}"
            CONFIG="${CONFIG:-firewood}"
            ;;
        firewood-archive-101-250k)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-1m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-firewood-archive-100}"
            START_BLOCK="${START_BLOCK:-101}"
            END_BLOCK="${END_BLOCK:-250000}"
            CONFIG="${CONFIG:-firewood-archive}"
            ;;
        firewood-33m-33m500k)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-30m-40m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-firewood-33m}"
            START_BLOCK="${START_BLOCK:-33000001}"
            END_BLOCK="${END_BLOCK:-33500000}"
            CONFIG="${CONFIG:-firewood}"
            ;;
        firewood-33m-40m)
            BLOCK_DIR_SRC="${BLOCK_DIR_SRC:-cchain-mainnet-blocks-30m-40m-ldb}"
            CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC:-cchain-current-state-firewood-33m}"
            START_BLOCK="${START_BLOCK:-33000001}"
            END_BLOCK="${END_BLOCK:-40000000}"
            CONFIG="${CONFIG:-firewood}"
            ;;
        *)
            error "Unknown test '$TEST_NAME'"
            ;;
    esac
fi

# Set chaos test defaults when using a defined test with CHAOS_MODE
if [[ -n "${CHAOS_MODE:-}" && -n "${TEST_NAME:-}" ]]; then
    MIN_WAIT_TIME="${MIN_WAIT_TIME:-120s}"
    MAX_WAIT_TIME="${MAX_WAIT_TIME:-150s}"
fi

# Determine data source: S3 import or local paths
if [[ -n "${BLOCK_DIR_SRC:-}" && -n "${CURRENT_STATE_DIR_SRC:-}" ]]; then
    # S3 mode - import data
    TIMESTAMP=$(date '+%Y%m%d-%H%M%S')
    EXECUTION_DATA_DIR="${EXECUTION_DATA_DIR:-/tmp/reexec-${TEST_NAME:-custom}-${TIMESTAMP}}"

    BLOCK_DIR_SRC="${BLOCK_DIR_SRC}" \
    CURRENT_STATE_DIR_SRC="${CURRENT_STATE_DIR_SRC}" \
    EXECUTION_DATA_DIR="${EXECUTION_DATA_DIR}" \
    "${SCRIPT_DIR}/import_cchain_data.sh"

    BLOCK_DIR="${EXECUTION_DATA_DIR}/blocks"
    CURRENT_STATE_DIR="${EXECUTION_DATA_DIR}/current-state"
elif [[ -n "${BLOCK_DIR_SRC:-}" || -n "${CURRENT_STATE_DIR_SRC:-}" ]]; then
    error "Both BLOCK_DIR_SRC and CURRENT_STATE_DIR_SRC must be provided together"
elif [[ -z "${BLOCK_DIR:-}" || -z "${CURRENT_STATE_DIR:-}" ]]; then
    show_usage
    echo ""
    echo "Env vars status:"
    echo "  S3 sources:"
    [[ -n "${BLOCK_DIR_SRC:-}" ]] && echo "    BLOCK_DIR_SRC: ${BLOCK_DIR_SRC}" || echo "    BLOCK_DIR_SRC: (not set)"
    [[ -n "${CURRENT_STATE_DIR_SRC:-}" ]] && echo "    CURRENT_STATE_DIR_SRC: ${CURRENT_STATE_DIR_SRC}" || echo "    CURRENT_STATE_DIR_SRC: (not set)"
    echo "  Local paths:"
    [[ -n "${BLOCK_DIR:-}" ]] && echo "    BLOCK_DIR: ${BLOCK_DIR}" || echo "    BLOCK_DIR: (not set)"
    [[ -n "${CURRENT_STATE_DIR:-}" ]] && echo "    CURRENT_STATE_DIR: ${CURRENT_STATE_DIR}" || echo "    CURRENT_STATE_DIR: (not set)"
    echo "  Block range:"
    [[ -n "${START_BLOCK:-}" ]] && echo "    START_BLOCK: ${START_BLOCK}" || echo "    START_BLOCK: (not set)"
    [[ -n "${END_BLOCK:-}" ]] && echo "    END_BLOCK: ${END_BLOCK}" || echo "    END_BLOCK: (not set)"
    if [[ -n "${CHAOS_MODE:-}" ]]; then
        echo "  Timeouts (chaos tests):"
        [[ -n "${MIN_WAIT_TIME:-}" ]] && echo "    MIN_WAIT_TIME: ${MIN_WAIT_TIME}" || echo "    MIN_WAIT_TIME: (not set)"
        [[ -n "${MAX_WAIT_TIME:-}" ]] && echo "    MAX_WAIT_TIME: ${MAX_WAIT_TIME}" || echo "    MAX_WAIT_TIME: (not set)"
    fi
    exit 1
fi

# Validate block range
if [[ -z "${START_BLOCK:-}" || -z "${END_BLOCK:-}" ]]; then
    error "START_BLOCK and END_BLOCK are required"
fi

if [[ -n "${CHAOS_MODE:-}" ]]; then
    # Chaos tests require additional validation
    if [[ -z "${MIN_WAIT_TIME:-}" || -z "${MAX_WAIT_TIME:-}" || -z "${CONFIG:-}" ]]; then
        error "MIN_WAIT_TIME and MAX_WAIT_TIME and CONFIG are required for chaos tests"
    fi

    echo "=== Firewood Chaos Test: ${TEST_NAME:-custom} ==="
    echo "Crashing between ${MIN_WAIT_TIME} and ${MAX_WAIT_TIME}"
else
    echo "=== C-Chain Re-execution Test: ${TEST_NAME:-custom} ==="
fi

echo "Blocks: ${START_BLOCK} - ${END_BLOCK}"
echo "CONFIG: ${CONFIG:-default}"

echo "=== Running Test ==="
if [[ -n "${CHAOS_MODE:-}" ]]; then
    go run ./tests/reexecute/chaos \
        --start-block="${START_BLOCK}" \
        --end-block="${END_BLOCK}" \
        --current-state-dir="${CURRENT_STATE_DIR}" \
        --block-dir="${BLOCK_DIR}" \
        --min-wait-time="${MIN_WAIT_TIME}" \
        --max-wait-time="${MAX_WAIT_TIME}" \
        --config="${CONFIG}"
else
    go run github.com/ava-labs/avalanchego/tests/reexecute/c \
        --block-dir="${BLOCK_DIR}" \
        --current-state-dir="${CURRENT_STATE_DIR}" \
        ${RUNNER_TYPE:+--runner="${RUNNER_TYPE}"} \
        ${CONFIG:+--config="${CONFIG}"} \
        --start-block="${START_BLOCK}" \
        --end-block="${END_BLOCK}" \
        ${LABELS:+--labels="${LABELS}"} \
        ${BENCHMARK_OUTPUT_FILE:+--benchmark-output-file="${BENCHMARK_OUTPUT_FILE}"} \
        ${METRICS_SERVER_ENABLED:+--metrics-server-enabled="${METRICS_SERVER_ENABLED}"} \
        ${METRICS_SERVER_PORT:+--metrics-server-port="${METRICS_SERVER_PORT}"} \
        ${METRICS_COLLECTOR_ENABLED:+--metrics-collector-enabled="${METRICS_COLLECTOR_ENABLED}"}

        if [[ -n "${PUSH_POST_STATE:-}" ]]; then
            echo "=== Pushing post-state to S3 ==="
            "${SCRIPT_DIR}/copy_dir.sh" "${CURRENT_STATE_DIR}/" "${PUSH_POST_STATE}"
        fi
fi