Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-02 06:38:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

fix to -O gen-C++ for recent AST profiling changes for identifying function parameters

Browse source
This commit is contained in:
Vern Paxson 2024-09-11 16:25:21 +02:00 • committed by Christian Kreibich
parent 2b64e3b05e
commit d1c31927c4
2 changed files with 27 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

3
src/script_opt/CPP/GenFunc.cc
View file

@ -149,6 +149,7 @@ void CPPCompile::DeclareLocals(const ProfileFunc* pf, const IDPList* lambda_ids)
capture_names.insert(CaptureName(li));
const auto& ls = pf->Locals();
int num_params = static_cast<int>(pf->Params().size());
// Track whether we generated a declaration. This is just for
// tidiness in the output.
@ -162,7 +163,7 @@ void CPPCompile::DeclareLocals(const ProfileFunc* pf, const IDPList* lambda_ids)
// No need to declare these, they're passed in as parameters.
ln = cn;
else if ( params.count(l) == 0 ) { // Not a parameter, so must be a local.
else if ( params.count(l) == 0 && l->Offset() >= num_params ) { // Not a parameter, so must be a local.
Emit("%s %s;", FullTypeName(l->GetType()), ln);
did_decl = true;
}

34
src/script_opt/ProfileFunc.cc
View file

@ -41,7 +41,20 @@ ProfileFunc::ProfileFunc(const Func* func, const StmtPtr& body, bool _abs_rec_fi
}
}
Profile(profiled_func_t.get(), body);
TrackType(profiled_func_t);
body->Traverse(this);
// Examine the locals and identify the parameters based on their offsets
// (being careful not to be fooled by captures that incidentally have low
// offsets). This approach allows us to accommodate function definitions
// that use different parameter names than appear in the original
// declaration.
num_params = profiled_func_t->Params()->NumFields();
for ( auto l : locals ) {
if ( captures.count(l) == 0 && l->Offset() < num_params )
params.insert(l);
}
}
ProfileFunc::ProfileFunc(const Stmt* s, bool _abs_rec_fields) {
@ -68,20 +81,11 @@ ProfileFunc::ProfileFunc(const Expr* e, bool _abs_rec_fields) {
captures_offsets[oid] = offset++;
}
Profile(func->GetType()->AsFuncType(), func->Ingredients()->Body());
}
auto ft = func->GetType()->AsFuncType();
auto& body = func->Ingredients()->Body();
else
// We don't have a function type, so do the traversal
// directly.
e->Traverse(this);
}
void ProfileFunc::Profile(const FuncType* ft, const StmtPtr& body) {
num_params = ft->Params()->NumFields();
assert(profiled_scope != nullptr);
auto& ov = profiled_scope->OrderedVars();
for ( int i = 0; i < num_params; ++i )
params.insert(ov[i].get());
@ -90,6 +94,12 @@ void ProfileFunc::Profile(const FuncType* ft, const StmtPtr& body) {
body->Traverse(this);
}
else
// We don't have a function type, so do the traversal
// directly.
e->Traverse(this);
}
TraversalCode ProfileFunc::PreStmt(const Stmt* s) {
stmts.push_back(s);