1 //==-- proto_to_cxx.cpp - Protobuf-C++ conversion --------------------------==// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 // 9 // Implements functions for converting between protobufs and C++. 10 // 11 //===----------------------------------------------------------------------===// 12 13 #include "proto_to_cxx.h" 14 #include "cxx_proto.pb.h" 15 16 #include <ostream> 17 #include <sstream> 18 19 namespace clang_fuzzer { 20 21 // Forward decls. 22 std::ostream &operator<<(std::ostream &os, const BinaryOp &x); 23 std::ostream &operator<<(std::ostream &os, const StatementSeq &x); 24 25 // Proto to C++. 26 std::ostream &operator<<(std::ostream &os, const Const &x) { 27 return os << "(" << x.val() << ")"; 28 } 29 std::ostream &operator<<(std::ostream &os, const VarRef &x) { 30 return os << "a[" << (static_cast<uint32_t>(x.varnum()) % 100) << "]"; 31 } 32 std::ostream &operator<<(std::ostream &os, const Lvalue &x) { 33 return os << x.varref(); 34 } 35 std::ostream &operator<<(std::ostream &os, const Rvalue &x) { 36 if (x.has_varref()) return os << x.varref(); 37 if (x.has_cons()) return os << x.cons(); 38 if (x.has_binop()) return os << x.binop(); 39 return os << "1"; 40 } 41 std::ostream &operator<<(std::ostream &os, const BinaryOp &x) { 42 os << "(" << x.left(); 43 switch (x.op()) { 44 case BinaryOp::PLUS: os << "+"; break; 45 case BinaryOp::MINUS: os << "-"; break; 46 case BinaryOp::MUL: os << "*"; break; 47 case BinaryOp::DIV: os << "/"; break; 48 case BinaryOp::MOD: os << "%"; break; 49 case BinaryOp::XOR: os << "^"; break; 50 case BinaryOp::AND: os << "&"; break; 51 case BinaryOp::OR: os << "|"; break; 52 case BinaryOp::EQ: os << "=="; break; 53 case BinaryOp::NE: os << "!="; break; 54 case BinaryOp::LE: os << "<="; break; 55 case BinaryOp::GE: os << ">="; break; 56 case BinaryOp::LT: os << "<"; break; 57 case BinaryOp::GT: os << ">"; break; 58 } 59 return os << x.right() << ")"; 60 } 61 std::ostream &operator<<(std::ostream &os, const AssignmentStatement &x) { 62 return os << x.lvalue() << "=" << x.rvalue() << ";\n"; 63 } 64 std::ostream &operator<<(std::ostream &os, const IfElse &x) { 65 return os << "if (" << x.cond() << "){\n" 66 << x.if_body() << "} else { \n" 67 << x.else_body() << "}\n"; 68 } 69 std::ostream &operator<<(std::ostream &os, const While &x) { 70 return os << "while (" << x.cond() << "){\n" << x.body() << "}\n"; 71 } 72 std::ostream &operator<<(std::ostream &os, const Statement &x) { 73 if (x.has_assignment()) return os << x.assignment(); 74 if (x.has_ifelse()) return os << x.ifelse(); 75 if (x.has_while_loop()) return os << x.while_loop(); 76 return os << "(void)0;\n"; 77 } 78 std::ostream &operator<<(std::ostream &os, const StatementSeq &x) { 79 for (auto &st : x.statements()) os << st; 80 return os; 81 } 82 std::ostream &operator<<(std::ostream &os, const Function &x) { 83 return os << "void foo(int *a) {\n" << x.statements() << "}\n"; 84 } 85 86 // --------------------------------- 87 88 std::string FunctionToString(const Function &input) { 89 std::ostringstream os; 90 os << input; 91 return os.str(); 92 93 } 94 std::string ProtoToCxx(const uint8_t *data, size_t size) { 95 Function message; 96 if (!message.ParsePartialFromArray(data, size)) 97 return "#error invalid proto\n"; 98 return FunctionToString(message); 99 } 100 101 } // namespace clang_fuzzer 102