287 lines
11 KiB
Bash
287 lines
11 KiB
Bash
#!/bin/bash
|
||
|
||
# runit-single.sh - 用于编译和测试单个或少量 SysY 程序的脚本
|
||
# 模仿 runit.sh 的功能，但以具体文件路径作为输入。
|
||
|
||
# --- 配置区 ---
|
||
# 请根据你的环境修改这些路径
|
||
# 假设此脚本位于你的项目根目录或一个脚本目录中
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" &>/dev/null && pwd)"
|
||
# 默认寻找项目根目录下的 build 和 lib
|
||
BUILD_BIN_DIR="${SCRIPT_DIR}/../build/bin"
|
||
LIB_DIR="${SCRIPT_DIR}/../lib"
|
||
# 临时文件会存储在脚本所在目录的 tmp 子目录中
|
||
TMP_DIR="${SCRIPT_DIR}/tmp"
|
||
|
||
# 定义编译器和模拟器
|
||
SYSYC="${BUILD_BIN_DIR}/sysyc"
|
||
GCC_RISCV64="riscv64-linux-gnu-gcc"
|
||
QEMU_RISCV64="qemu-riscv64"
|
||
|
||
# --- 初始化变量 ---
|
||
EXECUTE_MODE=false
|
||
CLEAN_MODE=false
|
||
SYSYC_TIMEOUT=10 # sysyc 编译超时 (秒)
|
||
GCC_TIMEOUT=10 # gcc 编译超时 (秒)
|
||
EXEC_TIMEOUT=5 # qemu 自动化执行超时 (秒)
|
||
MAX_OUTPUT_LINES=50 # 对比失败时显示的最大行数
|
||
SY_FILES=() # 存储用户提供的 .sy 文件列表
|
||
PASSED_CASES=0
|
||
FAILED_CASES_LIST=""
|
||
|
||
# --- 函数定义 ---
|
||
show_help() {
|
||
echo "用法: $0 [文件1.sy] [文件2.sy] ... [选项]"
|
||
echo "编译并测试指定的 .sy 文件。"
|
||
echo ""
|
||
echo "如果找到对应的 .in/.out 文件，则进行自动化测试。否则，进入交互模式。"
|
||
echo ""
|
||
echo "选项:"
|
||
echo " -e, --executable 编译为可执行文件并运行测试 (必须)。"
|
||
echo " -c, --clean 清理 tmp 临时目录下的所有文件。"
|
||
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
|
||
}
|
||
|
||
# --- 本次修改点: 整个参数解析逻辑被重写 ---
|
||
# 使用标准的 while 循环来健壮地处理任意顺序的参数
|
||
while [[ "$#" -gt 0 ]]; do
|
||
case "$1" in
|
||
-e|--executable)
|
||
EXECUTE_MODE=true
|
||
shift # 消耗选项
|
||
;;
|
||
-c|--clean)
|
||
CLEAN_MODE=true
|
||
shift # 消耗选项
|
||
;;
|
||
-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
|
||
;;
|
||
*) # 其他参数被视为文件路径
|
||
if [[ -f "$1" && "$1" == *.sy ]]; then
|
||
SY_FILES+=("$1")
|
||
else
|
||
echo "警告: 无效文件或不是 .sy 文件，已忽略: $1"
|
||
fi
|
||
shift # 消耗文件参数
|
||
;;
|
||
esac
|
||
done
|
||
|
||
|
||
if ${CLEAN_MODE}; then
|
||
echo "检测到 -c/--clean 选项，正在清空 ${TMP_DIR}..."
|
||
if [ -d "${TMP_DIR}" ]; then
|
||
rm -rf "${TMP_DIR}"/* 2>/dev/null
|
||
echo "清理完成。"
|
||
else
|
||
echo "临时目录 ${TMP_DIR} 不存在，无需清理。"
|
||
fi
|
||
|
||
if [ ${#SY_FILES[@]} -eq 0 ] && ! ${EXECUTE_MODE}; then
|
||
exit 0
|
||
fi
|
||
fi
|
||
|
||
# --- 主逻辑开始 ---
|
||
if ! ${EXECUTE_MODE}; then
|
||
echo "错误: 请提供 -e 或 --executable 选项来运行测试。"
|
||
show_help
|
||
exit 1
|
||
fi
|
||
|
||
if [ ${#SY_FILES[@]} -eq 0 ]; then
|
||
echo "错误: 未提供任何 .sy 文件作为输入。"
|
||
show_help
|
||
exit 1
|
||
fi
|
||
|
||
mkdir -p "${TMP_DIR}"
|
||
TOTAL_CASES=${#SY_FILES[@]}
|
||
|
||
echo "SysY 单例测试运行器启动..."
|
||
echo "超时设置: sysyc=${SYSYC_TIMEOUT}s, gcc=${GCC_TIMEOUT}s, qemu=${EXEC_TIMEOUT}s"
|
||
echo "失败输出最大行数: ${MAX_OUTPUT_LINES}"
|
||
echo ""
|
||
|
||
for sy_file in "${SY_FILES[@]}"; do
|
||
is_passed=1
|
||
base_name=$(basename "${sy_file}" .sy)
|
||
source_dir=$(dirname "${sy_file}")
|
||
|
||
ir_file="${TMP_DIR}/${base_name}_sysyc_riscv64.ll"
|
||
assembly_file="${TMP_DIR}/${base_name}.s"
|
||
assembly_debug_file="${TMP_DIR}/${base_name}_d.s"
|
||
executable_file="${TMP_DIR}/${base_name}"
|
||
input_file="${source_dir}/${base_name}.in"
|
||
output_reference_file="${source_dir}/${base_name}.out"
|
||
output_actual_file="${TMP_DIR}/${base_name}.actual_out"
|
||
|
||
echo "======================================================================"
|
||
echo "正在处理: ${sy_file}"
|
||
|
||
# 步骤 1: sysyc 编译
|
||
echo " 使用 sysyc 编译 (超时 ${SYSYC_TIMEOUT}s)..."
|
||
timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -s ir "${sy_file}" > "${ir_file}"
|
||
SYSYC_STATUS=$?
|
||
if [ $SYSYC_STATUS -eq 124 ]; then
|
||
echo -e "\e[31m错误: SysY 编译 ${sy_file} IR超时\e[0m"
|
||
is_passed=0
|
||
elif [ $SYSYC_STATUS -ne 0 ]; then
|
||
echo -e "\e[31m错误: SysY 编译 ${sy_file} IR失败，退出码: ${SYSYC_STATUS}\e[0m"
|
||
is_passed=0
|
||
fi
|
||
timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -S "${sy_file}" -o "${assembly_file}"
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "\e[31m错误: SysY 编译失败或超时。\e[0m"
|
||
is_passed=0
|
||
fi
|
||
# timeout -s KILL ${SYSYC_TIMEOUT} "${SYSYC}" -s asmd "${sy_file}" > "${assembly_debug_file}" 2>&1
|
||
|
||
# 步骤 2: GCC 编译
|
||
if [ "$is_passed" -eq 1 ]; then
|
||
echo " 使用 gcc 编译 (超时 ${GCC_TIMEOUT}s)..."
|
||
timeout -s KILL ${GCC_TIMEOUT} "${GCC_RISCV64}" "${assembly_file}" -o "${executable_file}" -L"${LIB_DIR}" -lsysy_riscv -static
|
||
if [ $? -ne 0 ]; then
|
||
echo -e "\e[31m错误: GCC 编译失败或超时。\e[0m"
|
||
is_passed=0
|
||
fi
|
||
fi
|
||
|
||
# 步骤 3: 执行与测试
|
||
if [ "$is_passed" -eq 1 ]; then
|
||
# 检查是自动化测试还是交互模式
|
||
if [ -f "${input_file}" ] || [ -f "${output_reference_file}" ]; then
|
||
# --- 自动化测试模式 ---
|
||
echo " 检测到 .in/.out 文件，进入自动化测试模式..."
|
||
echo " 正在执行 (超时 ${EXEC_TIMEOUT}s)..."
|
||
|
||
exec_cmd="${QEMU_RISCV64} \"${executable_file}\""
|
||
[ -f "${input_file}" ] && exec_cmd+=" < \"${input_file}\""
|
||
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 执行超时。\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}/${base_name}.expected_stdout"
|
||
head -n -1 "${output_reference_file}" > "${EXPECTED_STDOUT_FILE}"
|
||
if [ "$ACTUAL_RETURN_CODE" -ne "$EXPECTED_RETURN_CODE" ]; then 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 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
|
||
else
|
||
# --- 交互模式 ---
|
||
echo -e "\e[33m"
|
||
echo " **********************************************************"
|
||
echo " ** 未找到 .in 或 .out 文件，进入交互模式。 **"
|
||
echo " ** 程序即将运行，你可以直接在终端中输入。 **"
|
||
echo " ** 按下 Ctrl+D (EOF) 或以其他方式结束程序以继续。 **"
|
||
echo " **********************************************************"
|
||
echo -e "\e[0m"
|
||
"${QEMU_RISCV64}" "${executable_file}"
|
||
INTERACTIVE_RET_CODE=$?
|
||
echo -e "\e[33m\n 交互模式执行完毕，程序返回码: ${INTERACTIVE_RET_CODE}\e[0m"
|
||
echo " 注意: 交互模式的结果未经验证。"
|
||
fi
|
||
fi
|
||
|
||
if [ "$is_passed" -eq 1 ]; then
|
||
echo -e "\e[32m状态: 通过\e[0m"
|
||
((PASSED_CASES++))
|
||
else
|
||
echo -e "\e[31m状态: 失败\e[0m"
|
||
FAILED_CASES_LIST+="${sy_file}\n"
|
||
fi
|
||
done
|
||
|
||
# --- 打印最终总结 ---
|
||
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
|