修改状态的循环如何实现为SSA形式？以自定义strlen为例

先拆解原函数逻辑 #

原C函数的核心逻辑是：初始化长度为0，循环检查当前指针指向的字符是否非空，若是则长度加1、指针后移，直到遇到空字符后返回长度。注意*str++是先取值、再自增指针，这一点对SSA转换至关重要。

SSA转换的核心：变量版本化与phi节点 #

SSA的核心规则是每个变量仅被赋值一次，但循环会反复更新len和str，因此需要为每次迭代的变量创建不同版本，并用phi节点在循环的控制流汇合点（循环条件块）合并不同路径的变量值。

按基本块拆解的SSA实现 #

我们把函数拆成4个基本块，每个块内的变量都是唯一版本：

1. 入口块（BB0）：
   • 接收参数str，记为str_0
   • 初始化长度len_0 = 0
   • 跳转到循环条件块（BB1）
2. 循环条件块（BB1）：
   • 该块有两个前驱：第一次进入来自BB0，后续迭代来自循环体块（BB2），因此用phi节点合并变量：
     • str_phi = phi [BB0: str_0, BB2: str_2]：根据前驱块选择当前迭代的指针版本
     • len_phi = phi [BB0: len_0, BB2: len_1]：根据前驱块选择当前迭代的长度版本
   • 加载str_phi指向的字符：char_val = *str_phi
   • 判断char_val是否非空：非空则跳转到BB2，否则跳转到出口块（BB3）
3. 循环体块（BB2）：
   • 生成下一次迭代的指针版本：str_2 = str_phi + 1（对应原代码的指针自增）
   • 生成下一次迭代的长度版本：len_1 = len_phi + 1（对应原代码的长度自增）
   • 跳回BB1继续循环
4. 出口块（BB3）：
   • 返回len_phi（此时是空字符对应的最终长度）

对应LLVM IR简化版 #

Clang生成的LLVM IR本质和上述逻辑一致，简化后如下：

define i32 @strlen(i8* %str) {
entry:
  %len0 = i32 0
  br label %loop_cond

loop_cond:
  %str_phi = phi i8* [ %str, %entry ], [ %str2, %loop_body ]
  %len_phi = phi i32 [ %len0, %entry ], [ %len1, %loop_body ]
  %char_val = load i8, i8* %str_phi
  %is_nonzero = icmp ne i8 %char_val, 0
  br i1 %is_nonzero, label %loop_body, label %exit

loop_body:
  %str2 = getelementptr i8, i8* %str_phi, i32 1
  %len1 = add i32 %len_phi, 1
  br label %loop_cond

exit:
  ret i32 %len_phi
}

关于phi节点的理解 #

phi节点和数学中的欧拉函数完全无关，它是SSA专门用于控制流汇合点的指令：当一个基本块有多个前驱块时，phi节点会根据当前执行路径来自哪个前驱，选择对应的变量版本。

对于循环场景：

• 第一次进入循环条件块时，路径来自入口块，phi节点选择初始的str_0和len_0
• 后续每次迭代进入循环条件块时，路径来自循环体块，phi节点选择上一次迭代更新后的str_2和len_1

这样既满足了SSA“变量仅赋值一次”的要求，又正确实现了循环中变量的状态传递。

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