mirror of
https://github.com/zeek/zeek.git
synced 2025-10-08 09:38:19 +00:00
87e10b5f97
frameworks. There were a number of cases that weren't thread-safe. In particular, we don't use std::string anymore for anything that's passed between threads (but instead plain old const char*, with manual memmory managmenet). This is still a check-point commit, I'll do more testing.
94 lines
2 KiB
C++
94 lines
2 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "Manager.h"
|
|
#include "ReaderFrontend.h"
|
|
#include "ReaderBackend.h"
|
|
|
|
#include "threading/MsgThread.h"
|
|
|
|
namespace input {
|
|
|
|
class InitMessage : public threading::InputMessage<ReaderBackend>
|
|
{
|
|
public:
|
|
InitMessage(ReaderBackend* backend,
|
|
const int num_fields, const threading::Field* const* fields)
|
|
: threading::InputMessage<ReaderBackend>("Init", backend),
|
|
num_fields(num_fields), fields(fields) { }
|
|
|
|
virtual bool Process()
|
|
{
|
|
return Object()->Init(num_fields, fields);
|
|
}
|
|
|
|
private:
|
|
const int num_fields;
|
|
const threading::Field* const* fields;
|
|
};
|
|
|
|
class UpdateMessage : public threading::InputMessage<ReaderBackend>
|
|
{
|
|
public:
|
|
UpdateMessage(ReaderBackend* backend)
|
|
: threading::InputMessage<ReaderBackend>("Update", backend)
|
|
{ }
|
|
|
|
virtual bool Process() { return Object()->Update(); }
|
|
};
|
|
|
|
ReaderFrontend::ReaderFrontend(const ReaderBackend::ReaderInfo& arg_info, EnumVal* type)
|
|
{
|
|
disabled = initialized = false;
|
|
info = new ReaderBackend::ReaderInfo(arg_info);
|
|
|
|
const char* t = type->Type()->AsEnumType()->Lookup(type->InternalInt());
|
|
name = copy_string(fmt("%s/%s", arg_info.source, t));
|
|
|
|
backend = input_mgr->CreateBackend(this, type->InternalInt());
|
|
assert(backend);
|
|
backend->Start();
|
|
}
|
|
|
|
ReaderFrontend::~ReaderFrontend()
|
|
{
|
|
delete [] name;
|
|
delete info;
|
|
}
|
|
|
|
void ReaderFrontend::Init(const int arg_num_fields,
|
|
const threading::Field* const* arg_fields)
|
|
{
|
|
if ( disabled )
|
|
return;
|
|
|
|
if ( initialized )
|
|
reporter->InternalError("reader initialize twice");
|
|
|
|
num_fields = arg_num_fields;
|
|
fields = arg_fields;
|
|
initialized = true;
|
|
|
|
backend->SendIn(new InitMessage(backend, num_fields, fields));
|
|
}
|
|
|
|
void ReaderFrontend::Update()
|
|
{
|
|
if ( disabled )
|
|
return;
|
|
|
|
if ( ! initialized )
|
|
{
|
|
reporter->Error("Tried to call update on uninitialized reader");
|
|
return;
|
|
}
|
|
|
|
backend->SendIn(new UpdateMessage(backend));
|
|
}
|
|
|
|
const char* ReaderFrontend::Name() const
|
|
{
|
|
return name;
|
|
}
|
|
|
|
}
|
|
|