Method genFuzzVarAssign

lib/generation/ProtobufMutator.cpp:76–117

}

std::string ProtobufMutator::genFuzzVarAssign(FuzzInput &FuzzingInput) {
  std::string Code;
  // Use intermediate LocalVar for handling pointers
  auto LocalVarName = FuzzingInput.getProtoVarName();
  auto FuzzVarName = FuzzingInput.getFuzzVarName();
  const auto *MutationType = FuzzingInput.getFTGType().getRealType();

  assert(MutationType && "Unexpected Program State");

  // LocalVar declaration
  Code += SrcGenerator::genDeclStmt(*MutationType, LocalVarName);

  // Copy mutation result to LocalVar
  if (MutationType->isPointerType()) {
    Code += copyMutationPointerType(*MutationType, FuzzingInput);
  } else if (MutationType->isEnumType()) {
    Code += copyMutationEnumType(*MutationType, FuzzingInput);
  } else if (MutationType->isPrimitiveType()) {
    Code += copyMutationPrimitiveType(*MutationType, FuzzingInput);
  } else {
    assert(false && "Unsupported mutation type");
  }

  // Copy LocalVar to FuzzVar
  auto &FuzzVarType = FuzzingInput.getFTGType();
  if (&FuzzVarType == MutationType) {
    Code += SrcGenerator::genAssignStmt(FuzzVarName, LocalVarName);
  } else { // FuzzVar is pointer type
    auto PtrVarName = LocalVarName;
    auto TypeStr = SrcGenerator::genTypeStr(*MutationType);
    const auto *TypePtr = &FuzzVarType;
    while (MutationType != TypePtr) {
      Code += SrcGenerator::genAssignStmt(TypeStr + "* " + PtrVarName + "p",
                                          "&" + PtrVarName);
      TypePtr = TypePtr->getPointeeType();
      PtrVarName += "p";
      TypeStr += "*";
    }
    Code += SrcGenerator::genAssignStmt(FuzzVarName, PtrVarName);
  }
  return Code;
}

std::string ProtobufMutator::copyMutationPointerType(const Type &MutationType,
                                                     const FuzzInput &Input) {