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

ZAM support for "when" statements

Browse source
This commit is contained in:
Vern Paxson 2023-06-16 16:12:56 -07:00 • committed by Arne Welzel
parent 558cec58ed
commit d89b39a0bb
7 changed files with 246 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

57
src/script_opt/ZAM/Stmt.cc
View file

@ -60,6 +60,9 @@ const ZAMStmt ZAMCompiler::CompileStmt(const Stmt* s)
case STMT_INIT:
return CompileInit(static_cast<const InitStmt*>(s));
case STMT_WHEN:
return CompileWhen(static_cast<const WhenStmt*>(s));
case STMT_NULL:
return EmptyStmt();
@ -1093,6 +1096,60 @@ const ZAMStmt ZAMCompiler::CompileInit(const InitStmt* is)
return last;
}
const ZAMStmt ZAMCompiler::CompileWhen(const WhenStmt* ws)
{
auto wi = ws->Info();
auto timeout = wi->TimeoutExpr();
auto lambda = NewSlot(true);
(void)BuildLambda(lambda, wi->Lambda().get());
std::vector<IDPtr> local_aggr_slots;
for ( auto& l : wi->WhenExprLocals() )
if ( IsAggr(l->GetType()->Tag()) )
local_aggr_slots.push_back(l);
int n = local_aggr_slots.size();
auto aux = new ZInstAux(n);
aux->wi = wi;
for ( auto i = 0; i < n; ++i )
{
auto la = local_aggr_slots[i];
aux->Add(i, FrameSlot(la), la->GetType());
}
ZInstI z;
if ( timeout )
{
if ( timeout->Tag() == EXPR_NAME )
{
auto ns = FrameSlot(timeout->AsNameExpr());
z = ZInstI(OP_WHEN_TIMEOUT_VV, lambda, ns);
}
else
{
ASSERT(timeout->Tag() == EXPR_CONST);
z = ZInstI(OP_WHEN_TIMEOUT_VC, lambda, timeout->AsConstExpr());
}
}
else
z = ZInstI(OP_WHEN_V, lambda);
z.aux = aux;
if ( ws->IsReturn() )
{
(void)AddInst(z);
z = ZInstI(OP_RETURN_C);
z.c = ZVal();
}
return AddInst(z);
}
const ZAMStmt ZAMCompiler::InitRecord(IDPtr id, RecordType* rt)
{
auto z = ZInstI(OP_INIT_RECORD_V, FrameSlot(id));