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

ZBody/ZInst traversal

Browse source
This commit is contained in:
Vern Paxson 2024-06-11 14:36:12 -07:00
parent 9f57ac7c64
commit d5078b2722
3 changed files with 60 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/script_opt/ZAM/ZBody.cc
View file

@ -591,6 +591,11 @@ TraversalCode ZBody::Traverse(TraversalCallback* cb) const {
TraversalCode tc = cb->PreStmt(this);
HANDLE_TC_STMT_PRE(tc);
for ( size_t i = 0; i < NumInsts(); ++i ) {
tc = insts[i].Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
tc = cb->PostStmt(this);
HANDLE_TC_STMT_POST(tc);
}

50
src/script_opt/ZAM/ZInst.cc
View file

@ -298,6 +298,56 @@ string ZInst::ConstDump() const {
return d.Description();
}
TraversalCode ZInst::Traverse(TraversalCallback* cb) const {
TraversalCode tc;
if ( t ) {
tc = t->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
if ( t2 ) {
tc = t2->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
}
if ( aux ) {
tc = aux->Traverse(cb);
HANDLE_TC_STMT_POST(tc);
}
return TC_CONTINUE;
}
TraversalCode ZInstAux::Traverse(TraversalCallback* cb) const {
TraversalCode tc;
if ( id_val ) {
tc = id_val->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
if ( attrs ) {
tc = attrs->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
if ( value_var_type ) {
tc = value_var_type->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
for ( auto& lvt : loop_var_types ) {
tc = lvt->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
for ( auto& lvt : loop_var_types ) {
tc = lvt->Traverse(cb);
HANDLE_TC_STMT_PRE(tc);
}
return TC_CONTINUE;
}
void ZInstI::Dump(FILE* f, const FrameMap* frame_ids, const FrameReMap* remappings) const {
int n = NumFrameSlots();
// fprintf(f, "v%d ", n);

5
src/script_opt/ZAM/ZInst.h
View file

@ -6,6 +6,7 @@
#include "zeek/Desc.h"
#include "zeek/Func.h"
#include "zeek/TraverseTypes.h"
#include "zeek/script_opt/ZAM/BuiltInSupport.h"
#include "zeek/script_opt/ZAM/Support.h"
#include "zeek/script_opt/ZAM/ZOp.h"
@ -109,6 +110,8 @@ public:
// Returns a string describing the constant.
std::string ConstDump() const;
TraversalCode Traverse(TraversalCallback* cb) const;
ZOp op = OP_NOP;
ZAMOpType op_type = OP_X;
@ -455,6 +458,8 @@ public:
// Same but for constants.
void Add(int i, ValPtr c) { elems[i].SetConstant(c); }
TraversalCode Traverse(TraversalCallback* cb) const;
// Member variables. We could add accessors for manipulating
// these (and make the variables private), but for convenience we
// make them directly available.