如何修复基于OCamlyacc/Menhir的OCaml解析器程序问题？

问题1：多行语句解析触发Parser.MenhirBasics.Error #

原因 #

原解析器顶层规则仅支持单个语句，文件输入一次性读取全部内容后，多语句序列无法匹配规则；或词法分析器未正确处理换行符，导致语法结构混乱。

修复步骤 #

1. 修改parser.mly的起始规则：
   将原单语句起始规则改为支持多语句序列，允许换行或分号作为分隔符：

   %start main
   %type <unit> main
   
   %%
   
   main:
     | statements EOF { () }
   
   statements:
     | statement { $1 }
     | statement NEWLINE statements { $1; $3 }
     | statement SEMI statements { $1; $3 }
   

   其中NEWLINE是词法分析器中定义的换行符token，SEMI是分号token。

2. 调整lexer.mll的换行符处理：
   在词法规则中添加换行符的匹配，将其作为有效token返回（忽略空行）：

   rule token = parse
     | ['\n'] { NEWLINE }
     | [' ' '\t'] { token lexbuf }  // 忽略空白字符
     // ... 其他token匹配规则
   

   若原词法分析器将换行符当作空白忽略，解析器无法识别语句边界，会导致多语句被连在一起解析触发错误。

问题2：执行打印语句（如a）无输出 #

原因 #

原代码仅处理赋值语句，未对顶层表达式（直接写变量或表达式）添加打印逻辑，文件输入版本遗漏了标准输入版本的逐行打印逻辑。

修复步骤 #

在parser.mly的语句规则中，为表达式语句添加打印动作：

statement:
  | IDENT '=' expr { let v = eval $3 in Hashtbl.add env $1 v }
  | expr { let v = eval $1 in print_endline (string_of_int v) }

假设eval是表达式计算函数，env是全局变量表。若变量表在calc.ml中维护，可在calc.ml中遍历解析后的语句列表，对表达式类型执行打印：

type stmt = Assign of string * expr | Expr of expr

let rec process_statement = function
  | Assign (id, e) -> let v = eval e in Hashtbl.add env id v
  | Expr e -> let v = eval e in print_endline (string_of_int v)

let () =
  let lexbuf = Lexing.from_channel (open_in Sys.argv.(1)) in
  try
    let stmts = Parser.main Lexer.token lexbuf in
    List.iter process_statement stmts
  with
  | Lexer.Error msg -> Printf.fprintf stderr "Lex error: %s\n" msg
  | Parser.Error -> Printf.fprintf stderr "Parse error at %s\n" (Lexing.lexeme lexbuf)

同时需在parser.mly中调整返回类型为stmt list，让解析器返回所有语句的列表。

问题3：引用未定义变量无报错 #

原因 #

变量查找时使用了Hashtbl.find_opt或默认值，未对未绑定变量抛出异常；或eval函数忽略了未定义变量的错误情况。

修复步骤 #

1. 修改calc.ml中eval函数的变量处理：
   使用Hashtbl.find代替容错查找，找不到变量时直接抛出Not_found异常：

   let rec eval = function
     | Var id -> Hashtbl.find env id
     | Int n -> n
     | Add (e1, e2) -> eval e1 + eval e2
     // ... 其他表达式规则
   

2. 在顶层逻辑中捕获异常并输出错误：
   在process_statement函数中捕获未定义变量的异常，给出明确错误提示：

   let rec process_statement stmt =
     try
       match stmt with
       | Assign (id, e) -> let v = eval e in Hashtbl.add env id v
       | Expr e -> let v = eval e in print_endline (string_of_int v)
     with
     | Not_found ->
         let var_name = match stmt with
           | Expr (Var id) -> id
           | Assign (_, Var id) -> id
           | _ -> "unknown"
         in
         Printf.fprintf stderr "Error: Undefined variable '%s'\n" var_name
     | _ -> Printf.fprintf stderr "Runtime error\n"
   

构建与测试建议 #

确保build.sh正确编译修改后的文件：

ocamllex lexer.mll
menhir parser.mly
ocamlc -o calc calc.ml lexer.cmo parser.cmo

使用测试文件验证修复效果，例如输入文件内容：

a=1
a
b=2
a+b
c

预期输出：

1
3
Error: Undefined variable 'c'

内容的提问来源于stack exchange，提问作者wang kai