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