clang/tools/clang-fuzzer/proto-to-cxx/proto_to_cxx.cpp

   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