mirror of
https://github.com/zeek/zeek.git
synced 2025-10-11 02:58:20 +00:00
Initial import of svn+ssh:://svn.icir.org/bro/trunk/bro as of r7088
This commit is contained in:
Robin Sommer
commit
61757ac78b
1383 changed files with 380824 additions and 0 deletions
498
src/Stmt.h
Normal file
498
src/Stmt.h
Normal file
|
|
@ -0,0 +1,498 @@
|
|||
// $Id: Stmt.h 6916 2009-09-24 20:48:36Z vern $
|
||||
//
|
||||
// See the file "COPYING" in the main distribution directory for copyright.
|
||||
|
||||
#ifndef stmt_h
|
||||
#define stmt_h
|
||||
|
||||
// BRO statements.
|
||||
|
||||
#include "BroList.h"
|
||||
#include "Obj.h"
|
||||
#include "Expr.h"
|
||||
|
||||
#include "StmtEnums.h"
|
||||
|
||||
#include "TraverseTypes.h"
|
||||
|
||||
class StmtList;
|
||||
class ForStmt;
|
||||
|
||||
class Stmt : public BroObj {
|
||||
public:
|
||||
BroStmtTag Tag() const { return tag; }
|
||||
|
||||
virtual ~Stmt();
|
||||
|
||||
virtual Val* Exec(Frame* f, stmt_flow_type& flow) const = 0;
|
||||
|
||||
Stmt* Ref() { ::Ref(this); return this; }
|
||||
|
||||
bool SetLocationInfo(const Location* loc)
|
||||
{ return Stmt::SetLocationInfo(loc, loc); }
|
||||
bool SetLocationInfo(const Location* start, const Location* end);
|
||||
|
||||
// Returns a fully simplified version of the statement (this
|
||||
// may be the same statement, or a newly created one).
|
||||
virtual Stmt* Simplify();
|
||||
|
||||
// True if the statement has no side effects, false otherwise.
|
||||
virtual int IsPure() const;
|
||||
|
||||
StmtList* AsStmtList()
|
||||
{
|
||||
CHECK_TAG(tag, STMT_LIST, "Stmt::AsStmtList", stmt_name)
|
||||
return (StmtList*) this;
|
||||
}
|
||||
|
||||
ForStmt* AsForStmt()
|
||||
{
|
||||
CHECK_TAG(tag, STMT_FOR, "Stmt::AsForStmt", stmt_name)
|
||||
return (ForStmt*) this;
|
||||
}
|
||||
|
||||
void RegisterAccess() const { last_access = network_time; access_count++; }
|
||||
void AccessStats(ODesc* d) const;
|
||||
|
||||
virtual void Describe(ODesc* d) const;
|
||||
|
||||
virtual void IncrBPCount() { ++breakpoint_count; }
|
||||
virtual void DecrBPCount()
|
||||
{
|
||||
if ( breakpoint_count )
|
||||
--breakpoint_count;
|
||||
else
|
||||
internal_error("breakpoint count decremented below 0");
|
||||
}
|
||||
|
||||
virtual unsigned int BPCount() const { return breakpoint_count; }
|
||||
|
||||
bool Serialize(SerialInfo* info) const;
|
||||
static Stmt* Unserialize(UnserialInfo* info, BroStmtTag want = STMT_ANY);
|
||||
|
||||
virtual TraversalCode Traverse(TraversalCallback* cb) const = 0;
|
||||
|
||||
protected:
|
||||
Stmt() {}
|
||||
Stmt(BroStmtTag arg_tag);
|
||||
|
||||
void AddTag(ODesc* d) const;
|
||||
void DescribeDone(ODesc* d) const;
|
||||
|
||||
DECLARE_ABSTRACT_SERIAL(Stmt);
|
||||
|
||||
BroStmtTag tag;
|
||||
int breakpoint_count; // how many breakpoints on this statement
|
||||
|
||||
// FIXME: Learn the exact semantics of mutable.
|
||||
mutable double last_access; // time of last execution
|
||||
mutable uint32 access_count; // number of executions
|
||||
};
|
||||
|
||||
class ExprListStmt : public Stmt {
|
||||
public:
|
||||
const ListExpr* ExprList() const { return l; }
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
ExprListStmt() { l = 0; }
|
||||
ExprListStmt(BroStmtTag t, ListExpr* arg_l);
|
||||
|
||||
virtual ~ExprListStmt();
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
virtual Val* DoExec(val_list* vals, stmt_flow_type& flow) const = 0;
|
||||
|
||||
Stmt* Simplify();
|
||||
virtual Stmt* DoSimplify();
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
void PrintVals(ODesc* d, val_list* vals, int offset) const;
|
||||
|
||||
DECLARE_ABSTRACT_SERIAL(ExprListStmt);
|
||||
|
||||
ListExpr* l;
|
||||
};
|
||||
|
||||
class AlarmStmt : public ExprListStmt {
|
||||
public:
|
||||
AlarmStmt(ListExpr* l) : ExprListStmt(STMT_ALARM, l) { }
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
AlarmStmt() {}
|
||||
|
||||
Val* DoExec(val_list* vals, stmt_flow_type& flow) const;
|
||||
|
||||
DECLARE_SERIAL(AlarmStmt);
|
||||
};
|
||||
|
||||
class PrintStmt : public ExprListStmt {
|
||||
public:
|
||||
PrintStmt(ListExpr* l) : ExprListStmt(STMT_PRINT, l) { }
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
PrintStmt() {}
|
||||
|
||||
Val* DoExec(val_list* vals, stmt_flow_type& flow) const;
|
||||
|
||||
DECLARE_SERIAL(PrintStmt);
|
||||
};
|
||||
|
||||
class ExprStmt : public Stmt {
|
||||
public:
|
||||
ExprStmt(Expr* e);
|
||||
virtual ~ExprStmt();
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
const Expr* StmtExpr() const { return e; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
ExprStmt() { e = 0; }
|
||||
ExprStmt(BroStmtTag t, Expr* e);
|
||||
|
||||
virtual Val* DoExec(Frame* f, Val* v, stmt_flow_type& flow) const;
|
||||
|
||||
Stmt* Simplify();
|
||||
int IsPure() const;
|
||||
|
||||
// Called by Simplify(), after the expression's been simplified.
|
||||
virtual Stmt* DoSimplify();
|
||||
|
||||
DECLARE_SERIAL(ExprStmt);
|
||||
|
||||
Expr* e;
|
||||
};
|
||||
|
||||
class IfStmt : public ExprStmt {
|
||||
public:
|
||||
IfStmt(Expr* test, Stmt* s1, Stmt* s2);
|
||||
~IfStmt();
|
||||
|
||||
const Stmt* TrueBranch() const { return s1; }
|
||||
const Stmt* FalseBranch() const { return s2; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
IfStmt() { s1 = s2 = 0; }
|
||||
|
||||
Val* DoExec(Frame* f, Val* v, stmt_flow_type& flow) const;
|
||||
Stmt* DoSimplify();
|
||||
int IsPure() const;
|
||||
|
||||
DECLARE_SERIAL(IfStmt);
|
||||
|
||||
Stmt* s1;
|
||||
Stmt* s2;
|
||||
};
|
||||
|
||||
class Case : public BroObj {
|
||||
public:
|
||||
Case(ListExpr* c, Stmt* arg_s) { cases = c; s = arg_s; }
|
||||
~Case();
|
||||
|
||||
const ListExpr* Cases() const { return cases; }
|
||||
ListExpr* Cases() { return cases; }
|
||||
|
||||
const Stmt* Body() const { return s; }
|
||||
Stmt* Body() { return s; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
bool Serialize(SerialInfo* info) const;
|
||||
static Case* Unserialize(UnserialInfo* info);
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
Case() { cases = 0; s = 0; }
|
||||
|
||||
DECLARE_SERIAL(Case);
|
||||
|
||||
ListExpr* cases;
|
||||
Stmt* s;
|
||||
};
|
||||
|
||||
class SwitchStmt : public ExprStmt {
|
||||
public:
|
||||
SwitchStmt(Expr* index, case_list* cases);
|
||||
~SwitchStmt();
|
||||
|
||||
const case_list* Cases() const { return cases; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
SwitchStmt() { cases = 0; }
|
||||
|
||||
Val* DoExec(Frame* f, Val* v, stmt_flow_type& flow) const;
|
||||
Stmt* DoSimplify();
|
||||
int IsPure() const;
|
||||
|
||||
DECLARE_SERIAL(SwitchStmt);
|
||||
|
||||
case_list* cases;
|
||||
};
|
||||
|
||||
class AddStmt : public ExprStmt {
|
||||
public:
|
||||
AddStmt(Expr* e);
|
||||
|
||||
int IsPure() const;
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
AddStmt() {}
|
||||
|
||||
DECLARE_SERIAL(AddStmt);
|
||||
};
|
||||
|
||||
class DelStmt : public ExprStmt {
|
||||
public:
|
||||
DelStmt(Expr* e);
|
||||
|
||||
int IsPure() const;
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
DelStmt() {}
|
||||
|
||||
DECLARE_SERIAL(DelStmt);
|
||||
};
|
||||
|
||||
class EventStmt : public ExprStmt {
|
||||
public:
|
||||
EventStmt(EventExpr* e);
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
EventStmt() { event_expr = 0; }
|
||||
|
||||
DECLARE_SERIAL(EventStmt);
|
||||
|
||||
EventExpr* event_expr;
|
||||
};
|
||||
|
||||
class ForStmt : public ExprStmt {
|
||||
public:
|
||||
ForStmt(id_list* loop_vars, Expr* loop_expr);
|
||||
~ForStmt();
|
||||
|
||||
void AddBody(Stmt* arg_body) { body = arg_body; }
|
||||
|
||||
const id_list* LoopVar() const { return loop_vars; }
|
||||
const Expr* LoopExpr() const { return e; }
|
||||
const Stmt* LoopBody() const { return body; }
|
||||
|
||||
int IsPure() const;
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
ForStmt() { loop_vars = 0; body = 0; }
|
||||
|
||||
Val* DoExec(Frame* f, Val* v, stmt_flow_type& flow) const;
|
||||
Stmt* DoSimplify();
|
||||
|
||||
DECLARE_SERIAL(ForStmt);
|
||||
|
||||
id_list* loop_vars;
|
||||
Stmt* body;
|
||||
};
|
||||
|
||||
class NextStmt : public Stmt {
|
||||
public:
|
||||
NextStmt() : Stmt(STMT_NEXT) { }
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
int IsPure() const;
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
DECLARE_SERIAL(NextStmt);
|
||||
};
|
||||
|
||||
class BreakStmt : public Stmt {
|
||||
public:
|
||||
BreakStmt() : Stmt(STMT_BREAK) { }
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
int IsPure() const;
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
DECLARE_SERIAL(BreakStmt);
|
||||
};
|
||||
|
||||
class ReturnStmt : public ExprStmt {
|
||||
public:
|
||||
ReturnStmt(Expr* e);
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
ReturnStmt() {}
|
||||
|
||||
DECLARE_SERIAL(ReturnStmt);
|
||||
};
|
||||
|
||||
class StmtList : public Stmt {
|
||||
public:
|
||||
StmtList();
|
||||
~StmtList();
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
const stmt_list& Stmts() const { return stmts; }
|
||||
stmt_list& Stmts() { return stmts; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
Stmt* Simplify();
|
||||
int IsPure() const;
|
||||
|
||||
DECLARE_SERIAL(StmtList);
|
||||
|
||||
stmt_list stmts;
|
||||
};
|
||||
|
||||
class EventBodyList : public StmtList {
|
||||
public:
|
||||
EventBodyList() : StmtList()
|
||||
{ topmost = false; tag = STMT_EVENT_BODY_LIST; }
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
// "Topmost" means that this is the main body of a function or event.
|
||||
// void SetTopmost(bool is_topmost) { topmost = is_topmost; }
|
||||
// bool IsTopmost() { return topmost; }
|
||||
|
||||
protected:
|
||||
Stmt* Simplify();
|
||||
|
||||
DECLARE_SERIAL(EventBodyList);
|
||||
|
||||
bool topmost;
|
||||
};
|
||||
|
||||
class InitStmt : public Stmt {
|
||||
public:
|
||||
InitStmt(id_list* arg_inits) : Stmt(STMT_INIT)
|
||||
{
|
||||
inits = arg_inits;
|
||||
if ( arg_inits && arg_inits->length() )
|
||||
SetLocationInfo((*arg_inits)[0]->GetLocationInfo());
|
||||
}
|
||||
|
||||
~InitStmt();
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
|
||||
const id_list* Inits() const { return inits; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
friend class Stmt;
|
||||
InitStmt() { inits = 0; }
|
||||
|
||||
DECLARE_SERIAL(InitStmt);
|
||||
|
||||
id_list* inits;
|
||||
};
|
||||
|
||||
class NullStmt : public Stmt {
|
||||
public:
|
||||
NullStmt() : Stmt(STMT_NULL) { }
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
int IsPure() const;
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
DECLARE_SERIAL(NullStmt);
|
||||
};
|
||||
|
||||
class WhenStmt : public Stmt {
|
||||
public:
|
||||
// s2 is null if no timeout block given.
|
||||
WhenStmt(Expr* cond, Stmt* s1, Stmt* s2, Expr* timeout, bool is_return);
|
||||
~WhenStmt();
|
||||
|
||||
Val* Exec(Frame* f, stmt_flow_type& flow) const;
|
||||
int IsPure() const;
|
||||
Stmt* Simplify();
|
||||
|
||||
const Expr* Cond() const { return cond; }
|
||||
const Stmt* Body() const { return s1; }
|
||||
const Expr* TimeoutExpr() const { return timeout; }
|
||||
const Stmt* TimeoutBody() const { return s2; }
|
||||
|
||||
void Describe(ODesc* d) const;
|
||||
|
||||
TraversalCode Traverse(TraversalCallback* cb) const;
|
||||
|
||||
protected:
|
||||
WhenStmt() { cond = 0; s1 = s2 = 0; timeout = 0; is_return = 0; }
|
||||
|
||||
DECLARE_SERIAL(WhenStmt);
|
||||
|
||||
Expr* cond;
|
||||
Stmt* s1;
|
||||
Stmt* s2;
|
||||
Expr* timeout;
|
||||
bool is_return;
|
||||
};
|
||||
|
||||
extern Stmt* simplify_stmt(Stmt* s);
|
||||
extern int same_stmt(const Stmt* s1, const Stmt* s2);
|
||||
|
||||
#endif
|
||||
Loading…
Add table
Add a link
Reference in a new issue