201 lines
No EOL
6.4 KiB
Bash
Executable file
201 lines
No EOL
6.4 KiB
Bash
Executable file
#!/bin/bash
|
|
|
|
# Change to the analysis directory (where this script is located)
|
|
cd "$(dirname "$0")"
|
|
|
|
# Default values
|
|
OPTIMIZER_CLASS=""
|
|
TOTAL_RUNS=50
|
|
TRADES_PER_RUN=20
|
|
|
|
# Function to show usage
|
|
show_usage() {
|
|
echo "Usage: $0 <optimizer_class> [runs=N] [trades=N]"
|
|
echo ""
|
|
echo "Parameters:"
|
|
echo " optimizer_class Required. The optimizer class to use"
|
|
echo " runs=N Optional. Number of fuzzing runs (default: 50)"
|
|
echo " trades=N Optional. Trades per run (default: 20, actual will be ±5)"
|
|
echo ""
|
|
echo "Examples:"
|
|
echo " $0 BullMarketOptimizer"
|
|
echo " $0 WhaleOptimizer runs=100"
|
|
echo " $0 BearMarketOptimizer runs=10 trades=50"
|
|
echo " $0 NeutralMarketOptimizer trades=30 runs=25"
|
|
echo ""
|
|
echo "Available optimizers:"
|
|
echo " - BullMarketOptimizer"
|
|
echo " - NeutralMarketOptimizer"
|
|
echo " - BearMarketOptimizer"
|
|
echo " - WhaleOptimizer"
|
|
echo " - MockOptimizer"
|
|
echo " - RandomScenarioOptimizer"
|
|
}
|
|
|
|
# Parse arguments
|
|
if [ $# -eq 0 ]; then
|
|
echo "Error: No optimizer class specified"
|
|
show_usage
|
|
exit 1
|
|
fi
|
|
|
|
# First argument is always the optimizer class
|
|
OPTIMIZER_CLASS=$1
|
|
shift
|
|
|
|
# Parse named parameters
|
|
for arg in "$@"; do
|
|
case $arg in
|
|
runs=*)
|
|
TOTAL_RUNS="${arg#*=}"
|
|
if ! [[ "$TOTAL_RUNS" =~ ^[0-9]+$ ]] || [ "$TOTAL_RUNS" -eq 0 ]; then
|
|
echo "Error: Invalid value for runs. Must be a positive integer."
|
|
exit 1
|
|
fi
|
|
;;
|
|
trades=*)
|
|
TRADES_PER_RUN="${arg#*=}"
|
|
if ! [[ "$TRADES_PER_RUN" =~ ^[0-9]+$ ]] || [ "$TRADES_PER_RUN" -eq 0 ]; then
|
|
echo "Error: Invalid value for trades. Must be a positive integer."
|
|
exit 1
|
|
fi
|
|
;;
|
|
*)
|
|
echo "Error: Unknown parameter '$arg'"
|
|
show_usage
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# Colors for output
|
|
GREEN='\033[0;32m'
|
|
RED='\033[0;31m'
|
|
YELLOW='\033[1;33m'
|
|
NC='\033[0m' # No Color
|
|
|
|
OUTPUT_DIR="fuzzing_results_${OPTIMIZER_CLASS}_$(date +%Y%m%d_%H%M%S)"
|
|
MERGED_CSV="$OUTPUT_DIR/merged_profitable_scenarios.csv"
|
|
|
|
echo -e "${GREEN}=== Fuzzing Campaign ===${NC}"
|
|
echo "Optimizer: $OPTIMIZER_CLASS"
|
|
echo "Total runs: $TOTAL_RUNS"
|
|
echo "Trades per run: $TRADES_PER_RUN (±5)"
|
|
echo "Output directory: $OUTPUT_DIR"
|
|
echo ""
|
|
|
|
# Validate that the optimizer class exists by doing a dry run
|
|
echo "Validating optimizer class..."
|
|
OPTIMIZER_CLASS="$OPTIMIZER_CLASS" FUZZING_RUNS=0 forge script FuzzingAnalysis.s.sol --ffi --via-ir > /tmp/optimizer_check.log 2>&1
|
|
if [ $? -ne 0 ]; then
|
|
echo -e "${RED}Error: Invalid optimizer class '${OPTIMIZER_CLASS}'${NC}"
|
|
echo -e "${RED}Check the error:${NC}"
|
|
grep -E "(Unknown optimizer|revert)" /tmp/optimizer_check.log
|
|
echo ""
|
|
show_usage
|
|
exit 1
|
|
fi
|
|
|
|
# Create output directory
|
|
mkdir -p "$OUTPUT_DIR"
|
|
|
|
# Initialize merged CSV with header
|
|
echo "Scenario,Seed,Initial Balance,Final Balance,Profit,Profit %" > "$MERGED_CSV"
|
|
|
|
# Track statistics
|
|
TOTAL_PROFITABLE=0
|
|
FAILED_RUNS=0
|
|
|
|
# Save configuration
|
|
CONFIG_FILE="$OUTPUT_DIR/config.txt"
|
|
{
|
|
echo "Fuzzing Configuration"
|
|
echo "===================="
|
|
echo "Optimizer: $OPTIMIZER_CLASS"
|
|
echo "Total runs: $TOTAL_RUNS"
|
|
echo "Trades per run: $TRADES_PER_RUN (±5)"
|
|
echo "Start time: $(date)"
|
|
} > "$CONFIG_FILE"
|
|
|
|
# Run fuzzing analysis multiple times
|
|
for i in $(seq 1 $TOTAL_RUNS); do
|
|
echo -e "${YELLOW}Running fuzzing iteration $i/$TOTAL_RUNS...${NC}"
|
|
|
|
# Run single fuzzing iteration with specified optimizer and trades
|
|
OPTIMIZER_CLASS="$OPTIMIZER_CLASS" TRADES_PER_RUN="$TRADES_PER_RUN" FUZZING_RUNS=1 forge script FuzzingAnalysis.s.sol --ffi --via-ir --gas-limit 200000000 > "$OUTPUT_DIR/run_$i.log" 2>&1
|
|
|
|
if [ $? -eq 0 ]; then
|
|
echo -e "${GREEN}✓ Run $i completed${NC}"
|
|
|
|
# Check if profitable scenarios were found
|
|
if grep -q "PROFITABLE!" "$OUTPUT_DIR/run_$i.log"; then
|
|
echo -e "${GREEN} Found profitable scenario!${NC}"
|
|
((TOTAL_PROFITABLE++))
|
|
|
|
# Extract profit percentage
|
|
PROFIT_PCT=$(grep "PROFITABLE!" "$OUTPUT_DIR/run_$i.log" | grep -oE "Profit: [0-9]+%" | grep -oE "[0-9]+")
|
|
echo -e "${GREEN} Profit: ${PROFIT_PCT}%${NC}"
|
|
|
|
# Extract CSV file path if generated
|
|
CSV_FILE=$(grep "Profitable scenarios written to:" "$OUTPUT_DIR/run_$i.log" | awk '{print $NF}')
|
|
if [ -n "$CSV_FILE" ] && [ -f "$CSV_FILE" ]; then
|
|
# Append data rows (skip header) to merged CSV
|
|
tail -n +2 "$CSV_FILE" >> "$MERGED_CSV"
|
|
# Move individual CSV to output directory
|
|
mv "$CSV_FILE" "$OUTPUT_DIR/"
|
|
fi
|
|
fi
|
|
else
|
|
echo -e "${RED}✗ Run $i failed${NC}"
|
|
((FAILED_RUNS++))
|
|
# Show last few lines of error
|
|
echo -e "${RED}Error details:${NC}"
|
|
tail -5 "$OUTPUT_DIR/run_$i.log"
|
|
fi
|
|
|
|
# Small delay to avoid overwhelming the system
|
|
sleep 0.5
|
|
done
|
|
|
|
# Update config with end time
|
|
echo "End time: $(date)" >> "$CONFIG_FILE"
|
|
|
|
echo ""
|
|
echo -e "${GREEN}=== FUZZING CAMPAIGN COMPLETE ===${NC}"
|
|
echo "Optimizer: $OPTIMIZER_CLASS"
|
|
echo "Total runs: $TOTAL_RUNS"
|
|
echo "Trades per run: $TRADES_PER_RUN (±5)"
|
|
echo "Successful runs: $((TOTAL_RUNS - FAILED_RUNS))"
|
|
echo "Failed runs: $FAILED_RUNS"
|
|
echo "Total profitable scenarios: $TOTAL_PROFITABLE"
|
|
echo ""
|
|
echo "Results saved in: $OUTPUT_DIR"
|
|
echo "Merged CSV: $MERGED_CSV"
|
|
|
|
# Generate summary report
|
|
SUMMARY="$OUTPUT_DIR/summary.txt"
|
|
{
|
|
echo "Fuzzing Campaign Summary"
|
|
echo "========================"
|
|
echo "Date: $(date)"
|
|
echo "Optimizer: $OPTIMIZER_CLASS"
|
|
echo "Total runs: $TOTAL_RUNS"
|
|
echo "Trades per run: $TRADES_PER_RUN (±5)"
|
|
echo ""
|
|
echo "Results:"
|
|
echo "--------"
|
|
echo "Successful runs: $((TOTAL_RUNS - FAILED_RUNS)) / $TOTAL_RUNS"
|
|
echo "Failed runs: $FAILED_RUNS"
|
|
echo "Total profitable scenarios: $TOTAL_PROFITABLE / $((TOTAL_RUNS - FAILED_RUNS))"
|
|
echo "Success rate: $(awk "BEGIN {if ($TOTAL_RUNS - $FAILED_RUNS > 0) printf \"%.2f\", $TOTAL_PROFITABLE/($TOTAL_RUNS-$FAILED_RUNS)*100; else print \"0.00\"}")%"
|
|
} > "$SUMMARY"
|
|
|
|
echo ""
|
|
echo "Summary report: $SUMMARY"
|
|
|
|
# If there were profitable scenarios, show a sample
|
|
if [ $TOTAL_PROFITABLE -gt 0 ]; then
|
|
echo ""
|
|
echo -e "${GREEN}Sample profitable scenarios:${NC}"
|
|
head -5 "$MERGED_CSV"
|
|
fi |