301 lines
11 KiB
Bash
301 lines
11 KiB
Bash
#!/bin/bash
|
|
|
|
# runit.sh - 用于编译和测试 SysY 程序的脚本
|
|
# 此脚本应该位于 mysysy/test_script/
|
|
|
|
# 定义相对于脚本位置的目录
|
|
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"
|
|
TMP_DIR="${SCRIPT_DIR}/tmp"
|
|
|
|
# 定义编译器和模拟器
|
|
SYSYC="${BUILD_BIN_DIR}/sysyc"
|
|
GCC_RISCV64="riscv64-linux-gnu-gcc"
|
|
QEMU_RISCV64="qemu-riscv64"
|
|
|
|
# --- 新增功能: 初始化变量 ---
|
|
EXECUTE_MODE=false
|
|
SYSYC_TIMEOUT=10 # sysyc 编译超时 (秒)
|
|
GCC_TIMEOUT=10 # gcc 编译超时 (秒)
|
|
EXEC_TIMEOUT=5 # qemu 执行超时 (秒)
|
|
MAX_OUTPUT_LINES=50 # 对比失败时显示的最大行数
|
|
TEST_SETS=() # 用于存储要运行的测试集
|
|
TOTAL_CASES=0
|
|
PASSED_CASES=0
|
|
FAILED_CASES_LIST="" # 用于存储未通过的测例列表
|
|
|
|
# 显示帮助信息的函数
|
|
show_help() {
|
|
echo "用法: $0 [选项]"
|
|
echo "此脚本用于按文件名前缀数字升序编译和测试 .sy 文件。"
|
|
echo ""
|
|
echo "选项:"
|
|
echo " -e, --executable 编译为可执行文件并运行测试。"
|
|
echo " -c, --clean 清理 'tmp' 目录下的所有生成文件。"
|
|
echo " -set [f|h|p|all]... 指定要运行的测试集 (functional, h_functional, performance)。可多选，默认为 all。"
|
|
echo " -sct N 设置 sysyc 编译超时为 N 秒 (默认: 10)。"
|
|
echo " -gct N 设置 gcc 交叉编译超时为 N 秒 (默认: 10)。"
|
|
echo " -et N 设置 qemu 执行超时为 N 秒 (默认: 5)。"
|
|
echo " -ml N, --max-lines N 当输出对比失败时，最多显示 N 行内容 (默认: 50)。"
|
|
echo " -h, --help 显示此帮助信息并退出。"
|
|
}
|
|
|
|
# 显示文件内容并根据行数截断的函数
|
|
display_file_content() {
|
|
local file_path="$1"
|
|
local title="$2"
|
|
local max_lines="$3"
|
|
|
|
if [ ! -f "$file_path" ]; then
|
|
return
|
|
fi
|
|
|
|
echo -e "$title"
|
|
local line_count
|
|
line_count=$(wc -l < "$file_path")
|
|
|
|
if [ "$line_count" -gt "$max_lines" ]; then
|
|
head -n "$max_lines" "$file_path"
|
|
echo -e "\e[33m[... 输出已截断，共 ${line_count} 行 ...]\e[0m"
|
|
else
|
|
cat "$file_path"
|
|
fi
|
|
}
|
|
|
|
# 清理临时文件的函数
|
|
clean_tmp() {
|
|
echo "正在清理临时目录: ${TMP_DIR}"
|
|
rm -rf "${TMP_DIR}"/*
|
|
}
|
|
|
|
# 如果临时目录不存在，则创建它
|
|
mkdir -p "${TMP_DIR}"
|
|
|
|
# 解析命令行参数
|
|
while [[ "$#" -gt 0 ]]; do
|
|
case "$1" in
|
|
-e|--executable)
|
|
EXECUTE_MODE=true
|
|
shift
|
|
;;
|
|
-c|--clean)
|
|
clean_tmp
|
|
exit 0
|
|
;;
|
|
-set)
|
|
shift # 移过 '-set'
|
|
# 消耗所有后续参数直到遇到下一个选项
|
|
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
|
|
;;
|
|
-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
|
|
;;
|
|
-h|--help)
|
|
show_help
|
|
exit 0
|
|
;;
|
|
*)
|
|
echo "未知选项: $1"
|
|
show_help
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
# --- 本次修改点: 根据 -set 参数构建查找路径 ---
|
|
declare -A SET_MAP
|
|
SET_MAP[f]="functional"
|
|
SET_MAP[h]="h_functional"
|
|
SET_MAP[p]="performance"
|
|
|
|
SEARCH_PATHS=()
|
|
|
|
# 如果未指定测试集，或指定了 'all'，则搜索所有目录
|
|
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 测试运行器启动..."
|
|
echo "输入目录: ${SEARCH_PATHS[@]}"
|
|
echo "临时目录: ${TMP_DIR}"
|
|
echo "执行模式: ${EXECUTE_MODE}"
|
|
if ${EXECUTE_MODE}; then
|
|
echo "超时设置: sysyc=${SYSYC_TIMEOUT}s, gcc=${GCC_TIMEOUT}s, qemu=${EXEC_TIMEOUT}s"
|
|
echo "失败输出最大行数: ${MAX_OUTPUT_LINES}"
|
|
fi
|
|
echo ""
|
|
|
|
# 使用构建好的路径查找 .sy 文件并排序
|
|
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)
|
|
|
|
# --- 修复: 使用 here-string (<<<) 代替管道 (|) 来避免子 shell 问题 ---
|
|
while IFS= read -r sy_file; do
|
|
is_passed=1 # 1 表示通过, 0 表示失败
|
|
|
|
relative_path_no_ext=$(realpath --relative-to="${TESTDATA_DIR}" "${sy_file%.*}")
|
|
output_base_name=$(echo "${relative_path_no_ext}" | tr '/' '_')
|
|
|
|
assembly_file="${TMP_DIR}/${output_base_name}_sysyc_riscv64.s"
|
|
executable_file="${TMP_DIR}/${output_base_name}_sysyc_riscv64"
|
|
input_file="${sy_file%.*}.in"
|
|
output_reference_file="${sy_file%.*}.out"
|
|
output_actual_file="${TMP_DIR}/${output_base_name}_sysyc_riscv64.actual_out"
|
|
|
|
echo "正在处理: $(basename "$sy_file") (路径: ${relative_path_no_ext}.sy)"
|
|
|
|
# 步骤 1: 使用 sysyc 编译 .sy 到 .s
|
|
echo " 使用 sysyc 编译 (超时 ${SYSYC_TIMEOUT}s)..."
|
|
timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -S "${sy_file}" -o "${assembly_file}"
|
|
SYSYC_STATUS=$?
|
|
if [ $SYSYC_STATUS -eq 124 ]; then
|
|
echo -e "\e[31m错误: SysY 编译 ${sy_file} 超时\e[0m"
|
|
is_passed=0
|
|
elif [ $SYSYC_STATUS -ne 0 ]; then
|
|
echo -e "\e[31m错误: SysY 编译 ${sy_file} 失败，退出码: ${SYSYC_STATUS}\e[0m"
|
|
is_passed=0
|
|
fi
|
|
|
|
# 只有当 EXECUTE_MODE 为 true 且上一步成功时才继续
|
|
if ${EXECUTE_MODE} && [ "$is_passed" -eq 1 ]; then
|
|
# 步骤 2: 使用 riscv64-linux-gnu-gcc 编译 .s 到可执行文件
|
|
echo " 使用 gcc 编译 (超时 ${GCC_TIMEOUT}s)..."
|
|
timeout -s KILL ${GCC_TIMEOUT} "${GCC_RISCV64}" "${assembly_file}" -o "${executable_file}" -L"${LIB_DIR}" -lsysy_riscv -static
|
|
GCC_STATUS=$?
|
|
if [ $GCC_STATUS -eq 124 ]; then
|
|
echo -e "\e[31m错误: GCC 编译 ${assembly_file} 超时\e[0m"
|
|
is_passed=0
|
|
elif [ $GCC_STATUS -ne 0 ]; then
|
|
echo -e "\e[31m错误: GCC 编译 ${assembly_file} 失败，退出码: ${GCC_STATUS}\e[0m"
|
|
is_passed=0
|
|
fi
|
|
elif ! ${EXECUTE_MODE}; then
|
|
echo " 跳过执行模式。仅生成汇编文件。"
|
|
if [ "$is_passed" -eq 1 ]; then
|
|
((PASSED_CASES++))
|
|
else
|
|
FAILED_CASES_LIST+="${relative_path_no_ext}.sy\n"
|
|
fi
|
|
echo ""
|
|
continue
|
|
fi
|
|
|
|
# 步骤 3, 4, 5: 只有当编译都成功时才执行
|
|
if [ "$is_passed" -eq 1 ]; then
|
|
echo " 正在执行 (超时 ${EXEC_TIMEOUT}s)..."
|
|
|
|
exec_cmd="${QEMU_RISCV64} \"${executable_file}\""
|
|
if [ -f "${input_file}" ]; then
|
|
exec_cmd+=" < \"${input_file}\""
|
|
fi
|
|
exec_cmd+=" > \"${output_actual_file}\""
|
|
|
|
eval "timeout -s KILL ${EXEC_TIMEOUT} ${exec_cmd}"
|
|
ACTUAL_RETURN_CODE=$?
|
|
|
|
if [ "$ACTUAL_RETURN_CODE" -eq 124 ]; then
|
|
echo -e "\e[31m 执行超时: ${sy_file} 运行超过 ${EXEC_TIMEOUT} 秒\e[0m"
|
|
is_passed=0
|
|
else
|
|
if [ -f "${output_reference_file}" ]; then
|
|
LAST_LINE_TRIMMED=$(tail -n 1 "${output_reference_file}" | tr -d '[:space:]')
|
|
|
|
if [[ "$LAST_LINE_TRIMMED" =~ ^[-+]?[0-9]+$ ]]; then
|
|
EXPECTED_RETURN_CODE="$LAST_LINE_TRIMMED"
|
|
EXPECTED_STDOUT_FILE="${TMP_DIR}/${output_base_name}_sysyc_riscv64.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"
|
|
is_passed=0
|
|
fi
|
|
|
|
if ! diff -q <(tr -d '[:space:]' < "${output_actual_file}") <(tr -d '[:space:]' < "${EXPECTED_STDOUT_FILE}") >/dev/null 2>&1; then
|
|
echo -e "\e[31m 标准输出测试失败\e[0m"
|
|
is_passed=0
|
|
display_file_content "${EXPECTED_STDOUT_FILE}" " \e[36m---------- 期望输出 ----------\e[0m" "${MAX_OUTPUT_LINES}"
|
|
display_file_content "${output_actual_file}" " \e[36m---------- 实际输出 ----------\e[0m" "${MAX_OUTPUT_LINES}"
|
|
echo -e " \e[36m------------------------------\e[0m"
|
|
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}") <(tr -d '[:space:]' < "${output_reference_file}") >/dev/null 2>&1; then
|
|
echo -e "\e[32m 成功: 输出与参考输出匹配\e[0m"
|
|
else
|
|
echo -e "\e[31m 失败: 输出不匹配\e[0m"
|
|
is_passed=0
|
|
display_file_content "${output_reference_file}" " \e[36m---------- 期望输出 ----------\e[0m" "${MAX_OUTPUT_LINES}"
|
|
display_file_content "${output_actual_file}" " \e[36m---------- 实际输出 ----------\e[0m" "${MAX_OUTPUT_LINES}"
|
|
echo -e " \e[36m------------------------------\e[0m"
|
|
fi
|
|
fi
|
|
else
|
|
echo " 无参考输出文件。程序返回码: ${ACTUAL_RETURN_CODE}"
|
|
fi
|
|
fi
|
|
fi
|
|
|
|
if [ "$is_passed" -eq 1 ]; then
|
|
((PASSED_CASES++))
|
|
else
|
|
FAILED_CASES_LIST+="${relative_path_no_ext}.sy\n"
|
|
fi
|
|
echo ""
|
|
done <<< "$sy_files"
|
|
|
|
echo "========================================"
|
|
echo "测试完成"
|
|
echo "测试通过率: [${PASSED_CASES}/${TOTAL_CASES}]"
|
|
|
|
if [ -n "$FAILED_CASES_LIST" ]; then
|
|
echo ""
|
|
echo -e "\e[31m未通过的测例:\e[0m"
|
|
echo -e "${FAILED_CASES_LIST}"
|
|
fi
|
|
|
|
echo "========================================"
|
|
|
|
if [ "$PASSED_CASES" -eq "$TOTAL_CASES" ]; then
|
|
exit 0
|
|
else
|
|
exit 1
|
|
fi
|