mirror of
https://github.com/zeek/zeek.git
synced 2025-10-10 02:28:21 +00:00
107 lines
1.7 KiB
C++
107 lines
1.7 KiB
C++
#include "Event.h"
|
|
#include "EventHandler.h"
|
|
#include "Func.h"
|
|
#include "Scope.h"
|
|
#include "RemoteSerializer.h"
|
|
|
|
EventHandler::EventHandler(const char* arg_name)
|
|
{
|
|
name = copy_string(arg_name);
|
|
used = false;
|
|
local = 0;
|
|
type = 0;
|
|
error_handler = false;
|
|
enabled = true;
|
|
}
|
|
|
|
EventHandler::~EventHandler()
|
|
{
|
|
Unref(local);
|
|
delete [] name;
|
|
}
|
|
|
|
EventHandler::operator bool() const
|
|
{
|
|
return enabled && ((local && local->HasBodies()) || receivers.length());
|
|
}
|
|
|
|
FuncType* EventHandler::FType()
|
|
{
|
|
if ( type )
|
|
return type;
|
|
|
|
ID* id = lookup_ID(name, current_module.c_str());
|
|
|
|
if ( ! id )
|
|
return 0;
|
|
|
|
if ( id->Type()->Tag() != TYPE_FUNC )
|
|
return 0;
|
|
|
|
return type = id->Type()->AsFuncType();
|
|
}
|
|
|
|
void EventHandler::SetLocalHandler(Func* f)
|
|
{
|
|
if ( local )
|
|
Unref(local);
|
|
|
|
Ref(f);
|
|
local = f;
|
|
}
|
|
|
|
void EventHandler::Call(val_list* vl, bool no_remote)
|
|
{
|
|
#ifdef PROFILE_BRO_FUNCTIONS
|
|
DEBUG_MSG("Event: %s\n", Name());
|
|
#endif
|
|
|
|
if ( ! no_remote )
|
|
{
|
|
loop_over_list(receivers, i)
|
|
{
|
|
SerialInfo info(remote_serializer);
|
|
remote_serializer->SendCall(&info, receivers[i], name, vl);
|
|
}
|
|
}
|
|
|
|
if ( local )
|
|
// No try/catch here; we pass exceptions upstream.
|
|
Unref(local->Call(vl));
|
|
else
|
|
{
|
|
loop_over_list(*vl, i)
|
|
Unref((*vl)[i]);
|
|
}
|
|
}
|
|
|
|
void EventHandler::AddRemoteHandler(SourceID peer)
|
|
{
|
|
receivers.append(peer);
|
|
}
|
|
|
|
void EventHandler::RemoveRemoteHandler(SourceID peer)
|
|
{
|
|
receivers.remove(peer);
|
|
}
|
|
|
|
bool EventHandler::Serialize(SerialInfo* info) const
|
|
{
|
|
return SERIALIZE(name);
|
|
}
|
|
|
|
EventHandler* EventHandler::Unserialize(UnserialInfo* info)
|
|
{
|
|
char* name;
|
|
if ( ! UNSERIALIZE_STR(&name, 0) )
|
|
return 0;
|
|
|
|
EventHandler* h = event_registry->Lookup(name);
|
|
if ( ! h )
|
|
{
|
|
h = new EventHandler(name);
|
|
event_registry->Register(h);
|
|
}
|
|
|
|
return h;
|
|
}
|