mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 22:58:20 +00:00
f5a76c1aed
This largely copies over Spicy's `.clang-format` configuration file. The one place where we deviate is header include order since Zeek depends on headers being included in a certain order.
118 lines
3.7 KiB
C++
118 lines
3.7 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "zeek/File.h"
|
|
#include "zeek/RE.h"
|
|
#include "zeek/script_opt/CPP/Compile.h"
|
|
|
|
using namespace std;
|
|
|
|
namespace zeek::detail {
|
|
|
|
shared_ptr<CPP_InitInfo> CPPCompile::RegisterConstant(const ValPtr& vp, int& consts_offset) {
|
|
// Make sure the value pointer, which might be transient
|
|
// in construction, sticks around so we can track its
|
|
// value.
|
|
cv_indices.push_back(vp);
|
|
|
|
auto v = vp.get();
|
|
auto cv = const_vals.find(v);
|
|
|
|
if ( cv != const_vals.end() ) {
|
|
// Already did this one.
|
|
consts_offset = const_offsets[v];
|
|
return cv->second;
|
|
}
|
|
|
|
// Formulate a key that's unique per distinct constant.
|
|
|
|
const auto& t = v->GetType();
|
|
string c_desc;
|
|
|
|
if ( t->Tag() == TYPE_STRING ) {
|
|
// We can't rely on these to render with consistent
|
|
// escaping, sigh. Just use the raw string.
|
|
auto s = v->AsString();
|
|
auto b = (const char*)(s->Bytes());
|
|
c_desc = string(b, s->Len()) + "string";
|
|
}
|
|
else {
|
|
ODesc d;
|
|
v->Describe(&d);
|
|
|
|
// Don't confuse constants of different types that happen to
|
|
// render the same.
|
|
t->Describe(&d);
|
|
|
|
// Likewise, tables that have attributes.
|
|
if ( t->Tag() == TYPE_TABLE ) {
|
|
const auto& attrs = v->AsTableVal()->GetAttrs();
|
|
if ( attrs )
|
|
attrs->Describe(&d);
|
|
else
|
|
d.Add("<no-attrs>");
|
|
}
|
|
|
|
c_desc = d.Description();
|
|
}
|
|
|
|
auto c = constants.find(c_desc);
|
|
if ( c != constants.end() ) {
|
|
const_vals[v] = c->second;
|
|
consts_offset = const_offsets[v] = constants_offsets[c_desc];
|
|
return c->second;
|
|
}
|
|
|
|
auto tag = t->Tag();
|
|
auto const_name = const_info[tag]->NextName();
|
|
shared_ptr<CPP_InitInfo> gi;
|
|
|
|
switch ( tag ) {
|
|
case TYPE_BOOL: gi = make_shared<BasicConstInfo>(vp->AsBool() ? "true" : "false"); break;
|
|
|
|
case TYPE_INT: gi = make_shared<BasicConstInfo>(to_string(vp->AsInt())); break;
|
|
|
|
case TYPE_COUNT: gi = make_shared<BasicConstInfo>(to_string(vp->AsCount()) + "ULL"); break;
|
|
|
|
case TYPE_DOUBLE: gi = make_shared<BasicConstInfo>(to_string(vp->AsDouble())); break;
|
|
|
|
case TYPE_TIME: gi = make_shared<BasicConstInfo>(to_string(vp->AsDouble())); break;
|
|
|
|
case TYPE_INTERVAL: gi = make_shared<BasicConstInfo>(to_string(vp->AsDouble())); break;
|
|
|
|
case TYPE_ADDR: gi = make_shared<DescConstInfo>(this, vp); break;
|
|
|
|
case TYPE_SUBNET: gi = make_shared<DescConstInfo>(this, vp); break;
|
|
|
|
case TYPE_ENUM: gi = make_shared<EnumConstInfo>(this, vp); break;
|
|
|
|
case TYPE_STRING: gi = make_shared<StringConstInfo>(this, vp); break;
|
|
|
|
case TYPE_PATTERN: gi = make_shared<PatternConstInfo>(this, vp); break;
|
|
|
|
case TYPE_PORT: gi = make_shared<PortConstInfo>(vp); break;
|
|
|
|
case TYPE_LIST: gi = make_shared<ListConstInfo>(this, vp); break;
|
|
|
|
case TYPE_VECTOR: gi = make_shared<VectorConstInfo>(this, vp); break;
|
|
|
|
case TYPE_RECORD: gi = make_shared<RecordConstInfo>(this, vp); break;
|
|
|
|
case TYPE_TABLE: gi = make_shared<TableConstInfo>(this, vp); break;
|
|
|
|
case TYPE_FILE: gi = make_shared<FileConstInfo>(this, vp); break;
|
|
|
|
case TYPE_FUNC: gi = make_shared<FuncConstInfo>(this, vp); break;
|
|
|
|
default: reporter->InternalError("bad constant type in CPPCompile::AddConstant"); break;
|
|
}
|
|
|
|
const_info[tag]->AddInstance(gi);
|
|
const_vals[v] = constants[c_desc] = gi;
|
|
|
|
consts_offset = const_offsets[v] = constants_offsets[c_desc] = consts.size();
|
|
consts.emplace_back(tag, gi->Offset());
|
|
|
|
return gi;
|
|
}
|
|
|
|
} // namespace zeek::detail
|