#!/bin/bash # runit.sh - 用于编译和测试 SysY 程序的脚本 # 此脚本应该位于 mysysy/script/ export ASAN_OPTIONS=detect_leaks=0 # 定义相对于脚本位置的目录 SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd)" TESTDATA_DIR="${SCRIPT_DIR}/../testdata" BUILD_BIN_DIR="${SCRIPT_DIR}/../build/bin" LIB_DIR="${SCRIPT_DIR}/../lib" TMP_DIR="${SCRIPT_DIR}/tmp" # 定义编译器和模拟器 SYSYC="${BUILD_BIN_DIR}/sysyc" LLC_CMD="llc-19" GCC_RISCV64="riscv64-linux-gnu-gcc" QEMU_RISCV64="qemu-riscv64" # --- 状态变量 --- EXECUTE_MODE=false IR_EXECUTE_MODE=false OPTIMIZE_FLAG="" SYSYC_TIMEOUT=30 LLC_TIMEOUT=10 GCC_TIMEOUT=10 EXEC_TIMEOUT=30 MAX_OUTPUT_LINES=20 MAX_OUTPUT_CHARS=1000 TEST_SETS=() TOTAL_CASES=0 PASSED_CASES=0 FAILED_CASES_LIST="" INTERRUPTED=false # ================================================================= # --- 函数定义 --- # ================================================================= # 显示帮助信息的函数 show_help() { echo "用法: $0 [选项]" echo "此脚本用于按文件名前缀数字升序编译和测试 .sy 文件。" echo "" echo "选项:" echo " -e, --executable 编译为汇编并运行测试 (sysyc -> gcc -> qemu)。" echo " -eir 通过IR编译为可执行文件并运行测试 (sysyc -> llc -> gcc -> qemu)。" echo " -c, --clean 清理 'tmp' 目录下的所有生成文件。" echo " -O1 启用 sysyc 的 -O1 优化。" echo " -set [f|h|p|all]... 指定要运行的测试集 (functional, h_functional, performance)。可多选,默认为 all。" echo " -sct N 设置 sysyc 编译超时为 N 秒 (默认: 30)。" echo " -lct N 设置 llc-19 编译超时为 N 秒 (默认: 10)。" echo " -gct N 设置 gcc 交叉编译超时为 N 秒 (默认: 10)。" echo " -et N 设置 qemu 执行超时为 N 秒 (默认: 30)。" echo " -ml N, --max-lines N 当输出对比失败时,最多显示 N 行内容 (默认: 20)。" echo " -mc N, --max-chars N 当输出对比失败时,最多显示 N 个字符 (默认: 1000)。" echo " -h, --help 显示此帮助信息并退出。" echo "" echo "注意: 默认行为 (无 -e 或 -eir) 是将 .sy 文件同时编译为 .s (汇编) 和 .ll (IR),不执行。" echo " 可在任何时候按 Ctrl+C 来中断测试并显示当前已完成的测例总结。" } # 显示文件内容并根据行数和字符数截断的函数 display_file_content() { local file_path="$1" local title="$2" local max_lines="$3" local max_chars="$4" # 新增参数 if [ ! -f "$file_path" ]; then return; fi echo -e "$title" local line_count local char_count line_count=$(wc -l < "$file_path") char_count=$(wc -c < "$file_path") if [ "$line_count" -gt "$max_lines" ]; then head -n "$max_lines" "$file_path" echo -e "\e[33m[... 输出因行数过多 (共 ${line_count} 行) 而截断 ...]\e[0m" elif [ "$char_count" -gt "$max_chars" ]; then head -c "$max_chars" "$file_path" echo -e "\n\e[33m[... 输出因字符数过多 (共 ${char_count} 字符) 而截断 ...]\e[0m" else cat "$file_path" fi } # 清理临时文件的函数 clean_tmp() { echo "正在清理临时目录: ${TMP_DIR}" rm -rf "${TMP_DIR}"/* } # --- 新增:总结报告函数 --- print_summary() { echo "" # 确保从新的一行开始 echo "========================================" if [ "$INTERRUPTED" = true ]; then echo -e "\e[33m测试被中断。正在汇总已完成的结果...\e[0m" else echo "测试完成" fi local failed_count if [ -n "$FAILED_CASES_LIST" ]; then # `wc -l` 计算由换行符分隔的列表项数 failed_count=$(echo -e -n "${FAILED_CASES_LIST}" | wc -l) else failed_count=0 fi local executed_count=$((PASSED_CASES + failed_count)) echo "测试结果: [通过: ${PASSED_CASES}, 失败: ${failed_count}, 已执行: ${executed_count}/${TOTAL_CASES}]" if [ -n "$FAILED_CASES_LIST" ]; then echo "" echo -e "\e[31m未通过的测例:\e[0m" # 使用 printf 保证原样输出 printf "%b" "${FAILED_CASES_LIST}" fi echo "========================================" if [ "$failed_count" -gt 0 ]; then exit 1 else exit 0 fi } # --- 新增:SIGINT 信号处理函数 --- handle_sigint() { INTERRUPTED=true print_summary } # ================================================================= # --- 主逻辑开始 --- # ================================================================= # --- 新增:设置 trap 来捕获 SIGINT --- trap handle_sigint SIGINT mkdir -p "${TMP_DIR}" # 解析命令行参数 while [[ "$#" -gt 0 ]]; do case "$1" in -e|--executable) EXECUTE_MODE=true; shift ;; -eir) IR_EXECUTE_MODE=true; shift ;; -c|--clean) clean_tmp; exit 0 ;; -O1) OPTIMIZE_FLAG="-O1"; shift ;; -set) shift while [[ "$#" -gt 0 && ! "$1" =~ ^- ]]; do TEST_SETS+=("$1"); shift; done ;; -sct) if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then SYSYC_TIMEOUT="$2"; shift 2; else echo "错误: -sct 需要一个正整数参数。" >&2; exit 1; fi ;; -lct) if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then LLC_TIMEOUT="$2"; shift 2; else echo "错误: -lct 需要一个正整数参数。" >&2; exit 1; fi ;; -gct) if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then GCC_TIMEOUT="$2"; shift 2; else echo "错误: -gct 需要一个正整数参数。" >&2; exit 1; fi ;; -et) if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then EXEC_TIMEOUT="$2"; shift 2; else echo "错误: -et 需要一个正整数参数。" >&2; exit 1; fi ;; -ml|--max-lines) if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then MAX_OUTPUT_LINES="$2"; shift 2; else echo "错误: --max-lines 需要一个正整数参数。" >&2; exit 1; fi ;; -mc|--max-chars) if [[ -n "$2" && "$2" =~ ^[0-9]+$ ]]; then MAX_OUTPUT_CHARS="$2"; shift 2; else echo "错误: --max-chars 需要一个正整数参数。" >&2; exit 1; fi ;; -h|--help) show_help; exit 0 ;; *) echo "未知选项: $1"; show_help; exit 1 ;; esac done if ${EXECUTE_MODE} && ${IR_EXECUTE_MODE}; then echo -e "\e[31m错误: -e 和 -eir 选项不能同时使用。\e[0m" >&2 exit 1 fi declare -A SET_MAP SET_MAP[f]="functional" SET_MAP[h]="h_functional" SET_MAP[p]="performance" SEARCH_PATHS=() if [ ${#TEST_SETS[@]} -eq 0 ] || [[ " ${TEST_SETS[@]} " =~ " all " ]]; then SEARCH_PATHS+=("${TESTDATA_DIR}") else for set in "${TEST_SETS[@]}"; do if [[ -v SET_MAP[$set] ]]; then SEARCH_PATHS+=("${TESTDATA_DIR}/${SET_MAP[$set]}") else echo -e "\e[33m警告: 未知的测试集 '$set',已忽略。\e[0m" fi done fi if [ ${#SEARCH_PATHS[@]} -eq 0 ]; then echo -e "\e[31m错误: 没有找到有效的测试集目录,测试中止。\e[0m" exit 1 fi echo "SysY 测试运行器启动..." if [ -n "$OPTIMIZE_FLAG" ]; then echo "优化等级: ${OPTIMIZE_FLAG}"; fi echo "输入目录: ${SEARCH_PATHS[@]}" echo "临时目录: ${TMP_DIR}" RUN_MODE_INFO="" if ${IR_EXECUTE_MODE}; then RUN_MODE_INFO="IR执行模式 (-eir)" TIMEOUT_INFO="超时设置: sysyc=${SYSYC_TIMEOUT}s, llc=${LLC_TIMEOUT}s, gcc=${GCC_TIMEOUT}s, qemu=${EXEC_TIMEOUT}s" elif ${EXECUTE_MODE}; then RUN_MODE_INFO="直接执行模式 (-e)" TIMEOUT_INFO="超时设置: sysyc=${SYSYC_TIMEOUT}s, gcc=${GCC_TIMEOUT}s, qemu=${EXEC_TIMEOUT}s" else RUN_MODE_INFO="编译模式 (默认)" TIMEOUT_INFO="超时设置: sysyc=${SYSYC_TIMEOUT}s" fi echo "运行模式: ${RUN_MODE_INFO}" echo "${TIMEOUT_INFO}" if ${EXECUTE_MODE} || ${IR_EXECUTE_MODE}; then echo "失败输出最大行数: ${MAX_OUTPUT_LINES}" echo "失败输出最大字符数: ${MAX_OUTPUT_CHARS}" fi echo "" sy_files=$(find "${SEARCH_PATHS[@]}" -name "*.sy" | sort -V) if [ -z "$sy_files" ]; then echo "在指定目录中未找到任何 .sy 文件。" exit 0 fi TOTAL_CASES=$(echo "$sy_files" | wc -w) while IFS= read -r sy_file; do is_passed=0 # 0 表示失败, 1 表示通过 relative_path_no_ext=$(realpath --relative-to="${TESTDATA_DIR}" "${sy_file%.*}") output_base_name=$(echo "${relative_path_no_ext}" | tr '/' '_') assembly_file_S="${TMP_DIR}/${output_base_name}_sysyc_S.s" executable_file_S="${TMP_DIR}/${output_base_name}_sysyc_S" output_actual_file_S="${TMP_DIR}/${output_base_name}_sysyc_S.actual_out" ir_file="${TMP_DIR}/${output_base_name}_sysyc_ir.ll" assembly_file_from_ir="${TMP_DIR}/${output_base_name}_from_ir.s" executable_file_from_ir="${TMP_DIR}/${output_base_name}_from_ir" output_actual_file_from_ir="${TMP_DIR}/${output_base_name}_from_ir.actual_out" input_file="${sy_file%.*}.in" output_reference_file="${sy_file%.*}.out" echo "正在处理: $(basename "$sy_file") (路径: ${relative_path_no_ext}.sy)" # --- 模式 1: IR 执行模式 (-eir) --- if ${IR_EXECUTE_MODE}; then step_failed=0 test_logic_passed=0 echo " [1/4] 使用 sysyc 编译为 IR (超时 ${SYSYC_TIMEOUT}s)..." timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -s ir "${sy_file}" -o "${ir_file}" ${OPTIMIZE_FLAG} SYSYC_STATUS=$? if [ $SYSYC_STATUS -ne 0 ]; then [ $SYSYC_STATUS -eq 124 ] && echo -e "\e[31m错误: SysY (IR) 编译超时\e[0m" || echo -e "\e[31m错误: SysY (IR) 编译失败,退出码: ${SYSYC_STATUS}\e[0m" step_failed=1 fi if [ "$step_failed" -eq 0 ]; then echo " [2/4] 使用 llc-19 编译为汇编 (超时 ${LLC_TIMEOUT}s)..." timeout -s KILL ${LLC_TIMEOUT} "${LLC_CMD}" -march=riscv64 -mcpu=generic-rv64 -mattr=+m,+a,+f,+d,+c -filetype=asm "${ir_file}" -o "${assembly_file_from_ir}" LLC_STATUS=$? if [ $LLC_STATUS -ne 0 ]; then [ $LLC_STATUS -eq 124 ] && echo -e "\e[31m错误: llc-19 编译超时\e[0m" || echo -e "\e[31m错误: llc-19 编译失败,退出码: ${LLC_STATUS}\e[0m" step_failed=1 fi fi if [ "$step_failed" -eq 0 ]; then echo " [3/4] 使用 gcc 编译 (超时 ${GCC_TIMEOUT}s)..." timeout -s KILL ${GCC_TIMEOUT} "${GCC_RISCV64}" "${assembly_file_from_ir}" -o "${executable_file_from_ir}" -L"${LIB_DIR}" -lsysy_riscv -static GCC_STATUS=$? if [ $GCC_STATUS -ne 0 ]; then [ $GCC_STATUS -eq 124 ] && echo -e "\e[31m错误: GCC 编译超时\e[0m" || echo -e "\e[31m错误: GCC 编译失败,退出码: ${GCC_STATUS}\e[0m" step_failed=1 fi fi if [ "$step_failed" -eq 0 ]; then echo " [4/4] 正在执行 (超时 ${EXEC_TIMEOUT}s)..." exec_cmd="${QEMU_RISCV64} \"${executable_file_from_ir}\"" [ -f "${input_file}" ] && exec_cmd+=" < \"${input_file}\"" exec_cmd+=" > \"${output_actual_file_from_ir}\"" eval "timeout -s KILL ${EXEC_TIMEOUT} ${exec_cmd}" ACTUAL_RETURN_CODE=$? if [ "$ACTUAL_RETURN_CODE" -eq 124 ]; then echo -e "\e[31m 执行超时: 运行超过 ${EXEC_TIMEOUT} 秒\e[0m" else if [ -f "${output_reference_file}" ]; then LAST_LINE_TRIMMED=$(tail -n 1 "${output_reference_file}" | tr -d '[:space:]') test_logic_passed=1 if [[ "$LAST_LINE_TRIMMED" =~ ^[-+]?[0-9]+$ ]]; then EXPECTED_RETURN_CODE="$LAST_LINE_TRIMMED" EXPECTED_STDOUT_FILE="${TMP_DIR}/${output_base_name}_from_ir.expected_stdout" head -n -1 "${output_reference_file}" > "${EXPECTED_STDOUT_FILE}" if [ "$ACTUAL_RETURN_CODE" -eq "$EXPECTED_RETURN_CODE" ]; then echo -e "\e[32m 返回码测试成功: (${ACTUAL_RETURN_CODE}) 与期望值 (${EXPECTED_RETURN_CODE}) 匹配\e[0m" else echo -e "\e[31m 返回码测试失败: 期望: ${EXPECTED_RETURN_CODE}, 实际: ${ACTUAL_RETURN_CODE}\e[0m" test_logic_passed=0 fi if diff -q <(tr -d '[:space:]' < "${output_actual_file_from_ir}") <(tr -d '[:space:]' < "${EXPECTED_STDOUT_FILE}") >/dev/null 2>&1; then [ "$test_logic_passed" -eq 1 ] && echo -e "\e[32m 标准输出测试成功\e[0m" else echo -e "\e[31m 标准输出测试失败\e[0m" display_file_content "${EXPECTED_STDOUT_FILE}" " \e[36m---------- 期望输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" display_file_content "${output_actual_file_from_ir}" " \e[36m---------- 实际输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" test_logic_passed=0 fi else if [ $ACTUAL_RETURN_CODE -ne 0 ]; then echo -e "\e[33m警告: 程序以非零状态 ${ACTUAL_RETURN_CODE} 退出 (纯输出比较模式)。\e[0m"; fi if diff -q <(tr -d '[:space:]' < "${output_actual_file_from_ir}") <(tr -d '[:space:]' < "${output_reference_file}") >/dev/null 2>&1; then echo -e "\e[32m 成功: 输出与参考输出匹配\e[0m" else echo -e "\e[31m 失败: 输出不匹配\e[0m" display_file_content "${output_reference_file}" " \e[36m---------- 期望输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" display_file_content "${output_actual_file_from_ir}" " \e[36m---------- 实际输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" test_logic_passed=0 fi fi else echo " 无参考输出文件。程序返回码: ${ACTUAL_RETURN_CODE}" test_logic_passed=1 fi fi fi [ "$step_failed" -eq 0 ] && [ "$test_logic_passed" -eq 1 ] && is_passed=1 # --- 模式 2: 直接执行模式 (-e) --- elif ${EXECUTE_MODE}; then step_failed=0 test_logic_passed=0 echo " [1/3] 使用 sysyc 编译为汇编 (超时 ${SYSYC_TIMEOUT}s)..." timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -S "${sy_file}" -o "${assembly_file_S}" ${OPTIMIZE_FLAG} SYSYC_STATUS=$? if [ $SYSYC_STATUS -ne 0 ]; then [ $SYSYC_STATUS -eq 124 ] && echo -e "\e[31m错误: SysY (汇编) 编译超时\e[0m" || echo -e "\e[31m错误: SysY (汇编) 编译失败,退出码: ${SYSYC_STATUS}\e[0m" step_failed=1 fi if [ "$step_failed" -eq 0 ]; then echo " [2/3] 使用 gcc 编译 (超时 ${GCC_TIMEOUT}s)..." timeout -s KILL ${GCC_TIMEOUT} "${GCC_RISCV64}" "${assembly_file_S}" -o "${executable_file_S}" -L"${LIB_DIR}" -lsysy_riscv -static GCC_STATUS=$? if [ $GCC_STATUS -ne 0 ]; then [ $GCC_STATUS -eq 124 ] && echo -e "\e[31m错误: GCC 编译超时\e[0m" || echo -e "\e[31m错误: GCC 编译失败,退出码: ${GCC_STATUS}\e[0m" step_failed=1 fi fi if [ "$step_failed" -eq 0 ]; then echo " [3/3] 正在执行 (超时 ${EXEC_TIMEOUT}s)..." exec_cmd="${QEMU_RISCV64} \"${executable_file_S}\"" [ -f "${input_file}" ] && exec_cmd+=" < \"${input_file}\"" exec_cmd+=" > \"${output_actual_file_S}\"" eval "timeout -s KILL ${EXEC_TIMEOUT} ${exec_cmd}" ACTUAL_RETURN_CODE=$? if [ "$ACTUAL_RETURN_CODE" -eq 124 ]; then echo -e "\e[31m 执行超时: 运行超过 ${EXEC_TIMEOUT} 秒\e[0m" else if [ -f "${output_reference_file}" ]; then LAST_LINE_TRIMMED=$(tail -n 1 "${output_reference_file}" | tr -d '[:space:]') test_logic_passed=1 if [[ "$LAST_LINE_TRIMMED" =~ ^[-+]?[0-9]+$ ]]; then EXPECTED_RETURN_CODE="$LAST_LINE_TRIMMED" EXPECTED_STDOUT_FILE="${TMP_DIR}/${output_base_name}_sysyc_S.expected_stdout" head -n -1 "${output_reference_file}" > "${EXPECTED_STDOUT_FILE}" if [ "$ACTUAL_RETURN_CODE" -eq "$EXPECTED_RETURN_CODE" ]; then echo -e "\e[32m 返回码测试成功: (${ACTUAL_RETURN_CODE}) 与期望值 (${EXPECTED_RETURN_CODE}) 匹配\e[0m" else echo -e "\e[31m 返回码测试失败: 期望: ${EXPECTED_RETURN_CODE}, 实际: ${ACTUAL_RETURN_CODE}\e[0m" test_logic_passed=0 fi if diff -q <(tr -d '[:space:]' < "${output_actual_file_S}") <(tr -d '[:space:]' < "${EXPECTED_STDOUT_FILE}") >/dev/null 2>&1; then [ "$test_logic_passed" -eq 1 ] && echo -e "\e[32m 标准输出测试成功\e[0m" else echo -e "\e[31m 标准输出测试失败\e[0m" display_file_content "${EXPECTED_STDOUT_FILE}" " \e[36m---------- 期望输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" display_file_content "${output_actual_file_S}" " \e[36m---------- 实际输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" test_logic_passed=0 fi else if [ $ACTUAL_RETURN_CODE -ne 0 ]; then echo -e "\e[33m警告: 程序以非零状态 ${ACTUAL_RETURN_CODE} 退出 (纯输出比较模式)。\e[0m"; fi if diff -q <(tr -d '[:space:]' < "${output_actual_file_S}") <(tr -d '[:space:]' < "${output_reference_file}") >/dev/null 2>&1; then echo -e "\e[32m 成功: 输出与参考输出匹配\e[0m" else echo -e "\e[31m 失败: 输出不匹配\e[0m" display_file_content "${output_reference_file}" " \e[36m---------- 期望输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" display_file_content "${output_actual_file_S}" " \e[36m---------- 实际输出 ----------\e[0m" "${MAX_OUTPUT_LINES}" "${MAX_OUTPUT_CHARS}" test_logic_passed=0 fi fi else echo " 无参考输出文件。程序返回码: ${ACTUAL_RETURN_CODE}" test_logic_passed=1 fi fi fi [ "$step_failed" -eq 0 ] && [ "$test_logic_passed" -eq 1 ] && is_passed=1 # --- 模式 3: 默认编译模式 --- else s_compile_ok=0 ir_compile_ok=0 echo " [1/2] 使用 sysyc 编译为汇编 (超时 ${SYSYC_TIMEOUT}s)..." timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -S "${sy_file}" -o "${assembly_file_S}" ${OPTIMIZE_FLAG} SYSYC_S_STATUS=$? if [ $SYSYC_S_STATUS -eq 0 ]; then s_compile_ok=1 echo -e " \e[32m-> ${assembly_file_S} [成功]\e[0m" else [ $SYSYC_S_STATUS -eq 124 ] && echo -e " \e[31m-> [编译超时]\e[0m" || echo -e " \e[31m-> [编译失败, 退出码: ${SYSYC_S_STATUS}]\e[0m" fi echo " [2/2] 使用 sysyc 编译为 IR (超时 ${SYSYC_TIMEOUT}s)..." timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -s ir "${sy_file}" -o "${ir_file}" ${OPTIMIZE_FLAG} SYSYC_IR_STATUS=$? if [ $SYSYC_IR_STATUS -eq 0 ]; then ir_compile_ok=1 echo -e " \e[32m-> ${ir_file} [成功]\e[0m" else [ $SYSYC_IR_STATUS -eq 124 ] && echo -e " \e[31m-> [编译超时]\e[0m" || echo -e " \e[31m-> [编译失败, 退出码: ${SYSYC_IR_STATUS}]\e[0m" fi if [ "$s_compile_ok" -eq 1 ] && [ "$ir_compile_ok" -eq 1 ]; then is_passed=1 fi fi # --- 统计结果 --- if [ "$is_passed" -eq 1 ]; then ((PASSED_CASES++)) else # 确保 FAILED_CASES_LIST 的每一项都以换行符结尾 FAILED_CASES_LIST+="${relative_path_no_ext}.sy\n" fi echo "" done <<< "$sy_files" # --- 修改:调用总结函数 --- print_summary