Control Flow Flattening

Description

The purpose of this pass is to completely flatten the control flow graph of a program.

Available Compiler Options

• -mllvm -fla: activates code flattening
• -mllvm -funcFLA="func1,func2,func3": if code flattening is activated, apply it only on functions func1, func2 and func3
• -mllvm -perFLA=20: if code flattening is activated, apply it with a probability of 20% on each function

Implemented Technique

Here is an example. Consider this very simple C program :

#include <stdlib.h>
int main(int argc, char** argv) {
  int a = atoi(argv[1]);
  if(a == 0)
    return 1;
  else
    return 10;
  return 0;
}

The flattening pass will transform this code into this one :

#include <stdlib.h>
int main(int argc, char** argv) {
  int a = atoi(argv[1]);
  int b = 0;
  while(1) {
    switch(b) {
      case 0:
        if(a == 0)
          b == 1;
        else
          b == 2;
        break;
      case 1:
        return 1;
      case 2:
        return 10;
      default:
        break;
    }
  }
  return 0;
}

As one can see, all basic blocks are split and put into an infinite loop and the program flow is controlled by a switch and the variable b.

Here, it's what it looks like for the control flow generated before the flattening :

After the flattening, we get the following instruction flow:

As one can see, the main difference between the example in pure C, the IR version is completely flattened.