Как исправить эту `FunctionPass` так, чтобы она не переходила в бесконечный цикл?

Question

Следующий код представляет собой попытку создать FunctionPass, который выполняет итерации по всем BasicBlock, и сменил a + b на (a xor b) + 2 * (a and b) для обфускации.

Теперь, когда я использую ReplaceInstWithValue, итератор становится недействительным, и программа переходит в бесконечный цикл.

Я пробовал несколько способов исправить это, но ни один из них не оказался полезным.

Как я могу изменить программу, чтобы она выполняла итерации по всем инструкциям в программе без перехода в бесконечный цикл в первой инструкции add?

#include "llvm/Pass.h" 
#include "llvm/IR/Function.h" 
#include "llvm/IR/BasicBlock.h" 
#include "llvm/IR/Instructions.h" 
#include "llvm/IR/IRBuilder.h" 
#include "llvm/Support/raw_ostream.h" 
#include "llvm/Transforms/Utils/BasicBlockUtils.h" 
#include <map> 
#include <string> 

using namespace llvm; 

namespace { 
struct CountOp : public FunctionPass { 
    std::map<std::string, int> bbNameToId; 

    static char ID; 

    CountOp() : FunctionPass(ID) {} 

    virtual bool runOnFunction(Function &F) { 

     for (Function::iterator bs = F.begin(), be = F.end(); bs != be; ++bs) { 
      for (BasicBlock::iterator is = bs->begin(), ie = be->end(); is != ie; ++is) { 
       Instruction& inst = *is; 
       BinaryOperator* binop = dyn_cast<BinaryOperator>(&inst); 

       if (!binop) { 
        continue; 
       } 

       unsigned opcode = binop->getOpcode(); 
       errs() << binop->getOpcodeName() << "\n"; 

       if (opcode != Instruction::Add) { 
        continue; 
       } 

       IRBuilder<> builder(binop); 
       Value* v = builder.CreateAdd(builder.CreateXor(binop->getOperand(0), binop->getOperand(1)), 
              builder.CreateMul(ConstantInt::get(binop->getType(), 2), 
                   builder.CreateAnd(binop->getOperand(0), binop->getOperand(1)))); 

       ReplaceInstWithValue(bs->getInstList(), is, v); 
      } 
     } 


     return true; 
    } 
}; 
} 

char CountOp::ID = 0; 
static RegisterPass<CountOp> X("opChanger", "Change add operations", false, false); 

Answer 1

На самом деле, проблема заключается в for заявлении:

for (BasicBlock::iterator is = bs->begin(), ie = be->end(); is != ie; ++is).

ie должен быть инициализирован до bs->end(), а не be->end().

Тогда пример хорошо работает.