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
19namespace clang_fuzzer {
20
21// Forward decls.
22std::ostream &operator<<(std::ostream &os, const BinaryOp &x);
23std::ostream &operator<<(std::ostream &os, const StatementSeq &x);
24
25// Proto to C++.
26std::ostream &operator<<(std::ostream &os, const Const &x) {
27  return os << "(" << x.val() << ")";
28}
29std::ostream &operator<<(std::ostream &os, const VarRef &x) {
30  return os << "a[" << (static_cast<uint32_t>(x.varnum()) % 100) << "]";
31}
32std::ostream &operator<<(std::ostream &os, const Lvalue &x) {
33  return os << x.varref();
34}
35std::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}
41std::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}
61std::ostream &operator<<(std::ostream &os, const AssignmentStatement &x) {
62  return os << x.lvalue() << "=" << x.rvalue() << ";\n";
63}
64std::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}
69std::ostream &operator<<(std::ostream &os, const While &x) {
70  return os << "while (" << x.cond() << "){\n" << x.body() << "}\n";
71}
72std::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}
78std::ostream &operator<<(std::ostream &os, const StatementSeq &x) {
79  for (auto &st : x.statements()) os << st;
80  return os;
81}
82std::ostream &operator<<(std::ostream &os, const Function &x) {
83  return os << "void foo(int *a) {\n" << x.statements() << "}\n";
84}
85
86// ---------------------------------
87
88std::string FunctionToString(const Function &input) {
89  std::ostringstream os;
90  os << input;
91  return os.str();
92
93}
94std::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