mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 06:38:20 +00:00
Reformat Zeek in Spicy style
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.
This commit is contained in:
Benjamin Bannier
parent
7b8e7ed72c
commit
f5a76c1aed
786 changed files with 131714 additions and 153609 deletions
|
|
@ -2,277 +2,254 @@
|
|||
|
||||
#include "zeek/script_opt/CPP/Compile.h"
|
||||
|
||||
namespace zeek::detail
|
||||
{
|
||||
namespace zeek::detail {
|
||||
|
||||
using namespace std;
|
||||
|
||||
void CPPCompile::CompileFunc(const FuncInfo& func)
|
||||
{
|
||||
if ( ! IsCompilable(func) )
|
||||
return;
|
||||
void CPPCompile::CompileFunc(const FuncInfo& func) {
|
||||
if ( ! IsCompilable(func) )
|
||||
return;
|
||||
|
||||
auto fname = Canonicalize(BodyName(func).c_str()) + "_zf";
|
||||
auto pf = func.Profile();
|
||||
auto f = func.Func();
|
||||
const auto& body = func.Body();
|
||||
auto fname = Canonicalize(BodyName(func).c_str()) + "_zf";
|
||||
auto pf = func.Profile();
|
||||
auto f = func.Func();
|
||||
const auto& body = func.Body();
|
||||
|
||||
DefineBody(f->GetType(), pf, fname, body, nullptr, f->Flavor());
|
||||
}
|
||||
DefineBody(f->GetType(), pf, fname, body, nullptr, f->Flavor());
|
||||
}
|
||||
|
||||
void CPPCompile::CompileLambda(const LambdaExpr* l, const ProfileFunc* pf)
|
||||
{
|
||||
auto lname = Canonicalize(l->Name().c_str()) + "_lb";
|
||||
auto body = l->Ingredients()->Body();
|
||||
auto l_id = l->Ingredients()->GetID();
|
||||
auto& ids = l->OuterIDs();
|
||||
void CPPCompile::CompileLambda(const LambdaExpr* l, const ProfileFunc* pf) {
|
||||
auto lname = Canonicalize(l->Name().c_str()) + "_lb";
|
||||
auto body = l->Ingredients()->Body();
|
||||
auto l_id = l->Ingredients()->GetID();
|
||||
auto& ids = l->OuterIDs();
|
||||
|
||||
DefineBody(l_id->GetType<FuncType>(), pf, lname, body, &ids, FUNC_FLAVOR_FUNCTION);
|
||||
}
|
||||
DefineBody(l_id->GetType<FuncType>(), pf, lname, body, &ids, FUNC_FLAVOR_FUNCTION);
|
||||
}
|
||||
|
||||
void CPPCompile::GenInvokeBody(const string& call, const TypePtr& t)
|
||||
{
|
||||
if ( ! t || t->Tag() == TYPE_VOID )
|
||||
{
|
||||
Emit("%s;", call);
|
||||
Emit("return nullptr;");
|
||||
}
|
||||
else
|
||||
Emit("return %s;", NativeToGT(call, t, GEN_VAL_PTR));
|
||||
}
|
||||
void CPPCompile::GenInvokeBody(const string& call, const TypePtr& t) {
|
||||
if ( ! t || t->Tag() == TYPE_VOID ) {
|
||||
Emit("%s;", call);
|
||||
Emit("return nullptr;");
|
||||
}
|
||||
else
|
||||
Emit("return %s;", NativeToGT(call, t, GEN_VAL_PTR));
|
||||
}
|
||||
|
||||
void CPPCompile::DefineBody(const FuncTypePtr& ft, const ProfileFunc* pf, const string& fname,
|
||||
const StmtPtr& body, const IDPList* lambda_ids, FunctionFlavor flavor)
|
||||
{
|
||||
locals.clear();
|
||||
params.clear();
|
||||
void CPPCompile::DefineBody(const FuncTypePtr& ft, const ProfileFunc* pf, const string& fname, const StmtPtr& body,
|
||||
const IDPList* lambda_ids, FunctionFlavor flavor) {
|
||||
locals.clear();
|
||||
params.clear();
|
||||
|
||||
body_name = fname;
|
||||
body_name = fname;
|
||||
|
||||
ret_type = ft->Yield();
|
||||
in_hook = flavor == FUNC_FLAVOR_HOOK;
|
||||
auto ret_type_str = in_hook ? "bool" : FullTypeName(ret_type);
|
||||
ret_type = ft->Yield();
|
||||
in_hook = flavor == FUNC_FLAVOR_HOOK;
|
||||
auto ret_type_str = in_hook ? "bool" : FullTypeName(ret_type);
|
||||
|
||||
for ( const auto& p : pf->Params() )
|
||||
params.emplace(p);
|
||||
for ( const auto& p : pf->Params() )
|
||||
params.emplace(p);
|
||||
|
||||
NL();
|
||||
NL();
|
||||
|
||||
Emit("%s %s(%s)", ret_type_str, fname, ParamDecl(ft, lambda_ids, pf));
|
||||
Emit("%s %s(%s)", ret_type_str, fname, ParamDecl(ft, lambda_ids, pf));
|
||||
|
||||
StartBlock();
|
||||
StartBlock();
|
||||
|
||||
// Deal with "any" parameters, if any.
|
||||
TranslateAnyParams(ft, pf);
|
||||
// Deal with "any" parameters, if any.
|
||||
TranslateAnyParams(ft, pf);
|
||||
|
||||
// Make sure that any events referred to in this function have
|
||||
// been initialized.
|
||||
InitializeEvents(pf);
|
||||
// Make sure that any events referred to in this function have
|
||||
// been initialized.
|
||||
InitializeEvents(pf);
|
||||
|
||||
// Create the local variables.
|
||||
DeclareLocals(pf, lambda_ids);
|
||||
// Create the local variables.
|
||||
DeclareLocals(pf, lambda_ids);
|
||||
|
||||
GenStmt(body);
|
||||
GenStmt(body);
|
||||
|
||||
if ( in_hook )
|
||||
{
|
||||
Emit("return true;");
|
||||
in_hook = false;
|
||||
}
|
||||
if ( in_hook ) {
|
||||
Emit("return true;");
|
||||
in_hook = false;
|
||||
}
|
||||
|
||||
// Seatbelts for running off the end of a function that's supposed
|
||||
// to return a non-native type.
|
||||
if ( ! IsNativeType(ret_type) )
|
||||
Emit("return nullptr;");
|
||||
// Seatbelts for running off the end of a function that's supposed
|
||||
// to return a non-native type.
|
||||
if ( ! IsNativeType(ret_type) )
|
||||
Emit("return nullptr;");
|
||||
|
||||
EndBlock();
|
||||
}
|
||||
EndBlock();
|
||||
}
|
||||
|
||||
void CPPCompile::TranslateAnyParams(const FuncTypePtr& ft, const ProfileFunc* pf)
|
||||
{
|
||||
const auto& formals = ft->Params();
|
||||
int n = formals->NumFields();
|
||||
void CPPCompile::TranslateAnyParams(const FuncTypePtr& ft, const ProfileFunc* pf) {
|
||||
const auto& formals = ft->Params();
|
||||
int n = formals->NumFields();
|
||||
|
||||
for ( auto i = 0; i < n; ++i )
|
||||
{
|
||||
const auto& t = formals->GetFieldType(i);
|
||||
if ( t->Tag() != TYPE_ANY )
|
||||
// Not a relevant parameter.
|
||||
continue;
|
||||
for ( auto i = 0; i < n; ++i ) {
|
||||
const auto& t = formals->GetFieldType(i);
|
||||
if ( t->Tag() != TYPE_ANY )
|
||||
// Not a relevant parameter.
|
||||
continue;
|
||||
|
||||
auto param_id = FindParam(i, pf);
|
||||
if ( ! param_id )
|
||||
// Parameter isn't used, skip it.
|
||||
continue;
|
||||
auto param_id = FindParam(i, pf);
|
||||
if ( ! param_id )
|
||||
// Parameter isn't used, skip it.
|
||||
continue;
|
||||
|
||||
const auto& pt = param_id->GetType();
|
||||
if ( pt->Tag() == TYPE_ANY )
|
||||
// It's already "any", nothing more to do.
|
||||
continue;
|
||||
const auto& pt = param_id->GetType();
|
||||
if ( pt->Tag() == TYPE_ANY )
|
||||
// It's already "any", nothing more to do.
|
||||
continue;
|
||||
|
||||
auto any_i = string("any_param__CPP_") + Fmt(i);
|
||||
auto any_i = string("any_param__CPP_") + Fmt(i);
|
||||
|
||||
Emit("%s %s = %s;", FullTypeName(pt), LocalName(param_id),
|
||||
GenericValPtrToGT(any_i, pt, GEN_NATIVE));
|
||||
}
|
||||
}
|
||||
Emit("%s %s = %s;", FullTypeName(pt), LocalName(param_id), GenericValPtrToGT(any_i, pt, GEN_NATIVE));
|
||||
}
|
||||
}
|
||||
|
||||
void CPPCompile::InitializeEvents(const ProfileFunc* pf)
|
||||
{
|
||||
// Make sure that any events referred to in this function have
|
||||
// been initialized. We have to do this dynamically because it
|
||||
// depends on whether the final script using the compiled code
|
||||
// happens to load the associated event handler
|
||||
for ( const auto& e : pf->Events() )
|
||||
{
|
||||
auto ev_name = globals[e] + "_ev";
|
||||
void CPPCompile::InitializeEvents(const ProfileFunc* pf) {
|
||||
// Make sure that any events referred to in this function have
|
||||
// been initialized. We have to do this dynamically because it
|
||||
// depends on whether the final script using the compiled code
|
||||
// happens to load the associated event handler
|
||||
for ( const auto& e : pf->Events() ) {
|
||||
auto ev_name = globals[e] + "_ev";
|
||||
|
||||
// Create a scope so we don't have to individualize the
|
||||
// variables.
|
||||
Emit("{");
|
||||
Emit("static bool did_init = false;");
|
||||
Emit("if ( ! did_init )");
|
||||
StartBlock();
|
||||
// Create a scope so we don't have to individualize the
|
||||
// variables.
|
||||
Emit("{");
|
||||
Emit("static bool did_init = false;");
|
||||
Emit("if ( ! did_init )");
|
||||
StartBlock();
|
||||
|
||||
// We do both a Lookup and a Register because only the latter
|
||||
// returns an EventHandlerPtr, sigh.
|
||||
Emit("if ( event_registry->Lookup(\"%s\") )", e);
|
||||
StartBlock();
|
||||
Emit("%s = event_registry->Register(\"%s\");", ev_name, e);
|
||||
EndBlock();
|
||||
Emit("did_init = true;");
|
||||
EndBlock();
|
||||
Emit("}");
|
||||
}
|
||||
}
|
||||
// We do both a Lookup and a Register because only the latter
|
||||
// returns an EventHandlerPtr, sigh.
|
||||
Emit("if ( event_registry->Lookup(\"%s\") )", e);
|
||||
StartBlock();
|
||||
Emit("%s = event_registry->Register(\"%s\");", ev_name, e);
|
||||
EndBlock();
|
||||
Emit("did_init = true;");
|
||||
EndBlock();
|
||||
Emit("}");
|
||||
}
|
||||
}
|
||||
|
||||
void CPPCompile::DeclareLocals(const ProfileFunc* pf, const IDPList* lambda_ids)
|
||||
{
|
||||
// It's handy to have a set of the lambda captures rather than a list.
|
||||
IDSet lambda_set;
|
||||
if ( lambda_ids )
|
||||
for ( auto li : *lambda_ids )
|
||||
lambda_set.insert(li);
|
||||
void CPPCompile::DeclareLocals(const ProfileFunc* pf, const IDPList* lambda_ids) {
|
||||
// It's handy to have a set of the lambda captures rather than a list.
|
||||
IDSet lambda_set;
|
||||
if ( lambda_ids )
|
||||
for ( auto li : *lambda_ids )
|
||||
lambda_set.insert(li);
|
||||
|
||||
const auto& ls = pf->Locals();
|
||||
const auto& ls = pf->Locals();
|
||||
|
||||
// Track whether we generated a declaration. This is just for
|
||||
// tidiness in the output.
|
||||
bool did_decl = false;
|
||||
// Track whether we generated a declaration. This is just for
|
||||
// tidiness in the output.
|
||||
bool did_decl = false;
|
||||
|
||||
for ( const auto& l : ls )
|
||||
{
|
||||
auto ln = LocalName(l);
|
||||
for ( const auto& l : ls ) {
|
||||
auto ln = LocalName(l);
|
||||
|
||||
if ( lambda_set.count(l) > 0 )
|
||||
// No need to declare these, they're passed in as
|
||||
// parameters.
|
||||
ln = lambda_names[l];
|
||||
if ( lambda_set.count(l) > 0 )
|
||||
// No need to declare these, they're passed in as
|
||||
// parameters.
|
||||
ln = lambda_names[l];
|
||||
|
||||
else if ( params.count(l) == 0 )
|
||||
{ // Not a parameter, so must be a local.
|
||||
Emit("%s %s;", FullTypeName(l->GetType()), ln);
|
||||
did_decl = true;
|
||||
}
|
||||
else if ( params.count(l) == 0 ) { // Not a parameter, so must be a local.
|
||||
Emit("%s %s;", FullTypeName(l->GetType()), ln);
|
||||
did_decl = true;
|
||||
}
|
||||
|
||||
locals.emplace(l, ln);
|
||||
}
|
||||
locals.emplace(l, ln);
|
||||
}
|
||||
|
||||
if ( did_decl )
|
||||
NL();
|
||||
}
|
||||
if ( did_decl )
|
||||
NL();
|
||||
}
|
||||
|
||||
string CPPCompile::BodyName(const FuncInfo& func)
|
||||
{
|
||||
const auto& f = func.Func();
|
||||
const auto& body = func.Body();
|
||||
string fname = f->Name();
|
||||
string CPPCompile::BodyName(const FuncInfo& func) {
|
||||
const auto& f = func.Func();
|
||||
const auto& body = func.Body();
|
||||
string fname = f->Name();
|
||||
|
||||
// Extend name with location information.
|
||||
auto loc = body->GetLocationInfo();
|
||||
if ( loc->filename )
|
||||
{
|
||||
auto fn = loc->filename;
|
||||
// Extend name with location information.
|
||||
auto loc = body->GetLocationInfo();
|
||||
if ( loc->filename ) {
|
||||
auto fn = loc->filename;
|
||||
|
||||
// Skip leading goop that gets added by search paths.
|
||||
while ( *fn == '.' || *fn == '/' )
|
||||
++fn;
|
||||
// Skip leading goop that gets added by search paths.
|
||||
while ( *fn == '.' || *fn == '/' )
|
||||
++fn;
|
||||
|
||||
auto canonicalize = [](char c) -> char
|
||||
{
|
||||
return isalnum(c) ? c : '_';
|
||||
};
|
||||
auto canonicalize = [](char c) -> char { return isalnum(c) ? c : '_'; };
|
||||
|
||||
string fns = fn;
|
||||
transform(fns.begin(), fns.end(), fns.begin(), canonicalize);
|
||||
string fns = fn;
|
||||
transform(fns.begin(), fns.end(), fns.begin(), canonicalize);
|
||||
|
||||
if ( ! isalpha(fns[0]) )
|
||||
// This can happen for filenames beginning with numbers.
|
||||
fns = "_" + fns;
|
||||
if ( ! isalpha(fns[0]) )
|
||||
// This can happen for filenames beginning with numbers.
|
||||
fns = "_" + fns;
|
||||
|
||||
fname = fns + "__" + fname;
|
||||
}
|
||||
fname = fns + "__" + fname;
|
||||
}
|
||||
|
||||
const auto& bodies = f->GetBodies();
|
||||
const auto& bodies = f->GetBodies();
|
||||
|
||||
if ( bodies.size() == 1 )
|
||||
return fname;
|
||||
if ( bodies.size() == 1 )
|
||||
return fname;
|
||||
|
||||
// Make the name distinct-per-body.
|
||||
// Make the name distinct-per-body.
|
||||
|
||||
size_t i;
|
||||
for ( i = 0; i < bodies.size(); ++i )
|
||||
if ( bodies[i].stmts == body )
|
||||
break;
|
||||
size_t i;
|
||||
for ( i = 0; i < bodies.size(); ++i )
|
||||
if ( bodies[i].stmts == body )
|
||||
break;
|
||||
|
||||
if ( i >= bodies.size() )
|
||||
reporter->InternalError("can't find body in CPPCompile::BodyName");
|
||||
if ( i >= bodies.size() )
|
||||
reporter->InternalError("can't find body in CPPCompile::BodyName");
|
||||
|
||||
return fname + "__" + Fmt(static_cast<int>(i));
|
||||
}
|
||||
return fname + "__" + Fmt(static_cast<int>(i));
|
||||
}
|
||||
|
||||
p_hash_type CPPCompile::BodyHash(const Stmt* body)
|
||||
{
|
||||
auto bn = body_names.find(body);
|
||||
ASSERT(bn != body_names.end());
|
||||
p_hash_type CPPCompile::BodyHash(const Stmt* body) {
|
||||
auto bn = body_names.find(body);
|
||||
ASSERT(bn != body_names.end());
|
||||
|
||||
auto& body_name = bn->second;
|
||||
auto bh = body_hashes.find(body_name);
|
||||
ASSERT(bh != body_hashes.end());
|
||||
auto& body_name = bn->second;
|
||||
auto bh = body_hashes.find(body_name);
|
||||
ASSERT(bh != body_hashes.end());
|
||||
|
||||
return bh->second;
|
||||
}
|
||||
return bh->second;
|
||||
}
|
||||
|
||||
string CPPCompile::GenArgs(const RecordTypePtr& params, const Expr* e)
|
||||
{
|
||||
const auto& exprs = e->AsListExpr()->Exprs();
|
||||
string gen;
|
||||
string CPPCompile::GenArgs(const RecordTypePtr& params, const Expr* e) {
|
||||
const auto& exprs = e->AsListExpr()->Exprs();
|
||||
string gen;
|
||||
|
||||
int n = exprs.size();
|
||||
int n = exprs.size();
|
||||
|
||||
for ( auto i = 0; i < n; ++i )
|
||||
{
|
||||
auto e_i = exprs[i];
|
||||
auto gt = GEN_NATIVE;
|
||||
for ( auto i = 0; i < n; ++i ) {
|
||||
auto e_i = exprs[i];
|
||||
auto gt = GEN_NATIVE;
|
||||
|
||||
const auto& param_t = params->GetFieldType(i);
|
||||
bool param_any = param_t->Tag() == TYPE_ANY;
|
||||
bool arg_any = e_i->GetType()->Tag() == TYPE_ANY;
|
||||
const auto& param_t = params->GetFieldType(i);
|
||||
bool param_any = param_t->Tag() == TYPE_ANY;
|
||||
bool arg_any = e_i->GetType()->Tag() == TYPE_ANY;
|
||||
|
||||
if ( param_any && ! arg_any )
|
||||
gt = GEN_VAL_PTR;
|
||||
if ( param_any && ! arg_any )
|
||||
gt = GEN_VAL_PTR;
|
||||
|
||||
auto expr_gen = GenExpr(e_i, gt);
|
||||
auto expr_gen = GenExpr(e_i, gt);
|
||||
|
||||
if ( ! param_any && arg_any )
|
||||
expr_gen = GenericValPtrToGT(expr_gen, param_t, GEN_NATIVE);
|
||||
if ( ! param_any && arg_any )
|
||||
expr_gen = GenericValPtrToGT(expr_gen, param_t, GEN_NATIVE);
|
||||
|
||||
gen += expr_gen;
|
||||
if ( i < n - 1 )
|
||||
gen += ", ";
|
||||
}
|
||||
gen += expr_gen;
|
||||
if ( i < n - 1 )
|
||||
gen += ", ";
|
||||
}
|
||||
|
||||
return gen;
|
||||
}
|
||||
return gen;
|
||||
}
|
||||
|
||||
} // zeek::detail
|
||||
} // namespace zeek::detail
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue