如何替换yywrap定义？能否像yylex一样为其传递自定义参数？

嘿，这个问题问到点子上了！咱来一步步解决这两个需求：

Flex默认提供的yywrap签名是int yywrap(void)，但完全可以替换成自定义版本，核心是先禁用默认实现，再自己写符合需求的函数：

1. 禁用默认yywrap
   在你的Flex代码开头加上%option noyywrap，这样Flex就不会自动生成默认的yywrap函数，给自定义留足空间。

2. 自定义yywrap的参数与实现
   你可以根据需求定义任意参数的yywrap，比如带上下文指针的版本：

   %{
   #include <stdio.h>
   // 自定义上下文结构体
   typedef struct {
       int file_num;
       FILE* next_input;
   } ScanContext;
   
   // 声明自定义yywrap
   int yywrap(ScanContext* ctx);
   %}
   
   %option noyywrap
   
   %%
   // 你的词法分析规则写在这里
   "hello"   { printf("Found hello\n"); }
   .|\n      { /* 忽略其他字符 */ }
   %%
   
   int main() {
       ScanContext ctx = {0, fopen("next_file.txt", "r")};
       // 手动调用自定义yywrap（默认yylex不会主动传参数）
       while (yylex()) {
           if (yywrap(&ctx)) break;
       }
       return 0;
   }
   
   // 带参数的yywrap实现
   int yywrap(ScanContext* ctx) {
       if (ctx->next_input != NULL) {
           yyin = ctx->next_input;
           ctx->file_num++;
           ctx->next_input = NULL;
           return 0; // 返回0表示继续扫描新输入
       }
       return 1; // 返回1表示结束扫描
   }
   

   如果想让yywrap和yylex的参数机制更统一，推荐用Flex可重入模式（%option reentrant），这时候yywrap的标准签名会变成int yywrap(yyscan_t yyscanner)，你可以通过这个yyscanner指针传递自定义数据，后面会详细说。

要实现和yylex类似的自定义参数传递，最优雅的方式就是用Flex的可重入模式，它天生支持上下文传递，不用依赖全局变量：

1. 开启可重入模式并配置额外数据
   在Flex代码里加上%option reentrant，并用%option extra-type指定你要传递的自定义数据类型：

   %{
   #include <stdio.h>
   // 自定义参数结构体
   typedef struct {
       int user_id;
       char* log_path;
   } CustomArgs;
   %}
   
   %option reentrant noyywrap
   %option extra-type="CustomArgs*" // 指定额外数据的类型
   
   %%
   "user"    { 
       CustomArgs* args = yyget_extra(yyscanner);
       printf("Processing user with ID: %d\n", args->user_id);
   }
   .|\n      { /* 忽略其他内容 */ }
   %%
   
   // 带yyscan_t参数的yywrap，通过它获取自定义参数
   int yywrap(yyscan_t yyscanner) {
       CustomArgs* args = yyget_extra(yyscanner);
       printf("YYWRAP called, log path: %s\n", args->log_path);
       return 1;
   }
   
   int main() {
       yyscan_t scanner;
       CustomArgs args = {1001, "/var/log/scan.log"};
   
       // 初始化扫描器并绑定自定义参数
       yylex_init_extra(&args, &scanner);
       yylex(scanner); // yylex传递scanner参数
       yylex_destroy(scanner);
   
       return 0;
   }
   

2. 为什么这是简便方法？

   • 和yylex的参数传递逻辑完全一致，都是通过yyscan_t上下文指针传递
   • 不用手动修改yylex的调用逻辑，Flex会自动在需要时调用带yyscanner参数的yywrap
   • 避免了全局变量的线程安全问题，适合多线程场景

如果实在不想用可重入模式，也可以用全局变量传递参数，但这种方式不推荐，尤其是在多线程或者需要多个扫描实例的场景下，容易出现数据混乱。

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