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

track the current body with more rigor

Browse source
This commit is contained in:
Vern Paxson 2025-09-12 16:59:31 -07:00
parent 9281ffaa98
commit 16c46a3462
2 changed files with 11 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/Func.cc
View file

@ -295,10 +295,8 @@ ScriptFunc::ScriptFunc(std::string _name, FuncTypePtr ft, std::vector<StmtPtr> b
std::ranges::stable_sort(bodies, std::ranges::greater(), &Body::priority);
if ( ! bodies.empty() ) {
current_body = bodies[0].stmts;
current_priority = bodies[0].priority;
}
if ( ! bodies.empty() )
current_body = bodies[0];
}
ScriptFunc::~ScriptFunc() {
@ -562,13 +560,9 @@ void ScriptFunc::AddBody(StmtPtr new_body, const std::vector<IDPtr>& new_inits,
bodies.clear();
}
Body b;
b.stmts = new_body;
b.groups = {groups.begin(), groups.end()};
current_body = new_body;
current_priority = b.priority = priority;
current_body = Body{.stmts = new_body, .groups = {groups.begin(), groups.end()}, .priority = priority};
bodies.push_back(std::move(b));
bodies.push_back(current_body);
std::ranges::stable_sort(bodies, std::ranges::greater(), &Body::priority);
}
@ -577,18 +571,15 @@ void ScriptFunc::ReplaceBody(const StmtPtr& old_body, StmtPtr new_body) {
for ( auto body = bodies.begin(); body != bodies.end(); ++body )
if ( body->stmts.get() == old_body.get() ) {
if ( new_body ) {
if ( new_body )
body->stmts = new_body;
current_priority = body->priority;
}
else
bodies.erase(body);
found_it = true;
current_body = *body;
break;
}
current_body = new_body;
}
bool ScriptFunc::DeserializeCaptures(BrokerListView data) {

12
src/Func.h
View file

@ -265,8 +265,9 @@ public:
*/
void ReplaceBody(const detail::StmtPtr& old_body, detail::StmtPtr new_body);
StmtPtr CurrentBody() const { return current_body; }
int CurrentPriority() const { return current_priority; }
StmtPtr CurrentBody() const { return current_body.stmts; }
int CurrentPriority() const { return current_body.priority; }
auto CurrentEventGroups() const { return current_body.groups; }
/**
* Returns the function's frame size.
@ -322,11 +323,8 @@ private:
OffsetMap* captures_offset_mapping = nullptr;
// The most recently added/updated body ...
StmtPtr current_body;
// ... and its priority.
int current_priority = 0;
// A copy of the most recently added/updated Body.
Body current_body;
};
using built_in_func = ValPtr (*)(Frame* frame, const Args* args);