Switch EnumType::GetVal() to return const-ref

This commit is contained in:
Jon Siwek 2020-05-14 22:15:12 -07:00
parent 902f93671c
commit 62282c0226
9 changed files with 76 additions and 90 deletions

View file

@ -200,8 +200,9 @@ static IntrusivePtr<EnumVal> lookup_enum_val(const char* module_name, const char
static void print_log(const std::vector<IntrusivePtr<Val>>& vals) static void print_log(const std::vector<IntrusivePtr<Val>>& vals)
{ {
auto plval = lookup_enum_val("Log", "PRINTLOG"); static auto plval = lookup_enum_val("Log", "PRINTLOG");
auto record = make_intrusive<RecordVal>(zeek::id::find_type<RecordType>("Log::PrintLogInfo")); static auto lpli = zeek::id::find_type<RecordType>("Log::PrintLogInfo");
auto record = make_intrusive<RecordVal>(lpli);
auto vec = make_intrusive<VectorVal>(zeek::id::string_vec); auto vec = make_intrusive<VectorVal>(zeek::id::string_vec);
for ( const auto& val : vals ) for ( const auto& val : vals )

View file

@ -1231,20 +1231,17 @@ EnumType::enum_name_list EnumType::Names() const
return n; return n;
} }
IntrusivePtr<EnumVal> EnumType::GetVal(bro_int_t i) const IntrusivePtr<EnumVal>& EnumType::GetVal(bro_int_t i)
{ {
auto it = vals.find(i); auto it = vals.find(i);
IntrusivePtr<EnumVal> rval;
if ( it == vals.end() ) if ( it == vals.end() )
{ {
rval = make_intrusive<EnumVal>(IntrusivePtr{NewRef{}, this}, i); auto ev = make_intrusive<EnumVal>(IntrusivePtr{NewRef{}, this}, i);
vals[i] = rval; return vals.emplace(i, std::move(ev)).first->second;
} }
else
rval = it->second;
return rval; return it->second;
} }
void EnumType::DescribeReST(ODesc* d, bool roles_only) const void EnumType::DescribeReST(ODesc* d, bool roles_only) const

View file

@ -723,7 +723,7 @@ public:
void DescribeReST(ODesc* d, bool roles_only = false) const override; void DescribeReST(ODesc* d, bool roles_only = false) const override;
IntrusivePtr<EnumVal> GetVal(bro_int_t i); const IntrusivePtr<EnumVal>& GetVal(bro_int_t i);
protected: protected:
void AddNameInternal(const std::string& module_name, void AddNameInternal(const std::string& module_name,

View file

@ -79,7 +79,7 @@ bool MOUNT_Interp::RPC_BuildReply(RPC_CallInfo* c, BifEnum::rpc_status rpc_statu
double last_time, int reply_len) double last_time, int reply_len)
{ {
EventHandlerPtr event = nullptr; EventHandlerPtr event = nullptr;
Val* reply = nullptr; IntrusivePtr<Val> reply;
BifEnum::MOUNT3::status_t mount_status = BifEnum::MOUNT3::MNT3_OK; BifEnum::MOUNT3::status_t mount_status = BifEnum::MOUNT3::MNT3_OK;
bool rpc_success = ( rpc_status == BifEnum::RPC_SUCCESS ); bool rpc_success = ( rpc_status == BifEnum::RPC_SUCCESS );
@ -114,19 +114,17 @@ bool MOUNT_Interp::RPC_BuildReply(RPC_CallInfo* c, BifEnum::rpc_status rpc_statu
break; break;
case BifEnum::MOUNT3::PROC_MNT: case BifEnum::MOUNT3::PROC_MNT:
reply = mount3_mnt_reply(buf, n, mount_status); reply = {AdoptRef{}, mount3_mnt_reply(buf, n, mount_status)};
event = mount_proc_mnt; event = mount_proc_mnt;
break; break;
case BifEnum::MOUNT3::PROC_UMNT: case BifEnum::MOUNT3::PROC_UMNT:
reply = nullptr;
n = 0; n = 0;
mount_status = BifEnum::MOUNT3::MNT3_OK; mount_status = BifEnum::MOUNT3::MNT3_OK;
event = mount_proc_umnt; event = mount_proc_umnt;
break; break;
case BifEnum::MOUNT3::PROC_UMNT_ALL: case BifEnum::MOUNT3::PROC_UMNT_ALL:
reply = nullptr;
n = 0; n = 0;
mount_status = BifEnum::MOUNT3::MNT3_OK; mount_status = BifEnum::MOUNT3::MNT3_OK;
event = mount_proc_umnt; event = mount_proc_umnt;
@ -139,7 +137,7 @@ bool MOUNT_Interp::RPC_BuildReply(RPC_CallInfo* c, BifEnum::rpc_status rpc_statu
// Otherwise DeliverRPC would complain about // Otherwise DeliverRPC would complain about
// excess_RPC. // excess_RPC.
n = 0; n = 0;
reply = zeek::BifType::Enum::MOUNT3::proc_t->GetVal(c->Proc()).release(); reply = zeek::BifType::Enum::MOUNT3::proc_t->GetVal(c->Proc());
event = mount_proc_not_implemented; event = mount_proc_not_implemented;
} }
else else
@ -148,9 +146,7 @@ bool MOUNT_Interp::RPC_BuildReply(RPC_CallInfo* c, BifEnum::rpc_status rpc_statu
if ( rpc_success && ! buf ) if ( rpc_success && ! buf )
{ {
// There was a parse error. We have to unref the reply. (see // There was a parse error.
// also comments in RPC_BuildCall.
Unref(reply);
reply = nullptr; reply = nullptr;
return false; return false;
} }
@ -172,12 +168,11 @@ bool MOUNT_Interp::RPC_BuildReply(RPC_CallInfo* c, BifEnum::rpc_status rpc_statu
vl.emplace_back(AdoptRef{}, request); vl.emplace_back(AdoptRef{}, request);
if ( reply ) if ( reply )
vl.emplace_back(AdoptRef{}, reply); vl.emplace_back(reply);
analyzer->EnqueueConnEvent(event, std::move(vl)); analyzer->EnqueueConnEvent(event, std::move(vl));
} }
else
Unref(reply);
return true; return true;
} }
@ -221,7 +216,8 @@ zeek::Args MOUNT_Interp::event_common_vl(RPC_CallInfo *c,
EnumVal* MOUNT_Interp::mount3_auth_flavor(const u_char*& buf, int& n) EnumVal* MOUNT_Interp::mount3_auth_flavor(const u_char*& buf, int& n)
{ {
BifEnum::MOUNT3::auth_flavor_t t = (BifEnum::MOUNT3::auth_flavor_t)extract_XDR_uint32(buf, n); BifEnum::MOUNT3::auth_flavor_t t = (BifEnum::MOUNT3::auth_flavor_t)extract_XDR_uint32(buf, n);
return zeek::BifType::Enum::MOUNT3::auth_flavor_t->GetVal(t).release(); auto rval = zeek::BifType::Enum::MOUNT3::auth_flavor_t->GetVal(t);
return rval.release();
} }
StringVal* MOUNT_Interp::mount3_fh(const u_char*& buf, int& n) StringVal* MOUNT_Interp::mount3_fh(const u_char*& buf, int& n)

View file

@ -251,7 +251,8 @@ bool NFS_Interp::RPC_BuildReply(RPC_CallInfo* c, BifEnum::rpc_status rpc_status,
// Otherwise DeliverRPC would complain about // Otherwise DeliverRPC would complain about
// excess_RPC. // excess_RPC.
n = 0; n = 0;
reply = zeek::BifType::Enum::NFS3::proc_t->GetVal(c->Proc()).release(); auto ev = zeek::BifType::Enum::NFS3::proc_t->GetVal(c->Proc());
reply = ev.release();
event = nfs_proc_not_implemented; event = nfs_proc_not_implemented;
} }
else else
@ -438,13 +439,15 @@ RecordVal* NFS_Interp::nfs3_fattr(const u_char*& buf, int& n)
EnumVal* NFS_Interp::nfs3_time_how(const u_char*& buf, int& n) EnumVal* NFS_Interp::nfs3_time_how(const u_char*& buf, int& n)
{ {
BifEnum::NFS3::time_how_t t = (BifEnum::NFS3::time_how_t)extract_XDR_uint32(buf, n); BifEnum::NFS3::time_how_t t = (BifEnum::NFS3::time_how_t)extract_XDR_uint32(buf, n);
return zeek::BifType::Enum::NFS3::time_how_t->GetVal(t).release(); auto rval = zeek::BifType::Enum::NFS3::time_how_t->GetVal(t);
return rval.release();
} }
EnumVal* NFS_Interp::nfs3_ftype(const u_char*& buf, int& n) EnumVal* NFS_Interp::nfs3_ftype(const u_char*& buf, int& n)
{ {
BifEnum::NFS3::file_type_t t = (BifEnum::NFS3::file_type_t)extract_XDR_uint32(buf, n); BifEnum::NFS3::file_type_t t = (BifEnum::NFS3::file_type_t)extract_XDR_uint32(buf, n);
return zeek::BifType::Enum::NFS3::file_type_t->GetVal(t).release(); auto rval = zeek::BifType::Enum::NFS3::file_type_t->GetVal(t);
return rval.release();
} }
RecordVal* NFS_Interp::nfs3_wcc_attr(const u_char*& buf, int& n) RecordVal* NFS_Interp::nfs3_wcc_attr(const u_char*& buf, int& n)
@ -533,7 +536,8 @@ RecordVal* NFS_Interp::nfs3_pre_op_attr(const u_char*& buf, int& n)
EnumVal *NFS_Interp::nfs3_stable_how(const u_char*& buf, int& n) EnumVal *NFS_Interp::nfs3_stable_how(const u_char*& buf, int& n)
{ {
BifEnum::NFS3::stable_how_t stable = (BifEnum::NFS3::stable_how_t)extract_XDR_uint32(buf, n); BifEnum::NFS3::stable_how_t stable = (BifEnum::NFS3::stable_how_t)extract_XDR_uint32(buf, n);
return zeek::BifType::Enum::NFS3::stable_how_t->GetVal(stable).release(); auto rval = zeek::BifType::Enum::NFS3::stable_how_t->GetVal(stable);
return rval.release();
} }
RecordVal* NFS_Interp::nfs3_lookup_reply(const u_char*& buf, int& n, BifEnum::NFS3::status_t status) RecordVal* NFS_Interp::nfs3_lookup_reply(const u_char*& buf, int& n, BifEnum::NFS3::status_t status)

View file

@ -196,7 +196,8 @@ struct val_converter {
if ( i == -1 ) if ( i == -1 )
return nullptr; return nullptr;
return etype->GetVal(i).release(); auto rval = etype->GetVal(i);
return rval.release();
} }
return nullptr; return nullptr;

View file

@ -20,7 +20,8 @@ EnumVal* query_status(bool success)
failure_val = store_query_status->Lookup("Broker", "FAILURE"); failure_val = store_query_status->Lookup("Broker", "FAILURE");
} }
return store_query_status->GetVal(success ? success_val : failure_val).release(); auto rval = store_query_status->GetVal(success ? success_val : failure_val);
return rval.release();
} }
void StoreHandleVal::ValDescribe(ODesc* d) const void StoreHandleVal::ValDescribe(ODesc* d) const

View file

@ -1072,8 +1072,8 @@ void Manager::SendEntry(ReaderFrontend* reader, Value* *vals)
else if ( i->stream_type == EVENT_STREAM ) else if ( i->stream_type == EVENT_STREAM )
{ {
EnumVal* type = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW).release(); auto type = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW);
readFields = SendEventStreamEvent(i, type, vals); readFields = SendEventStreamEvent(i, type.release(), vals);
} }
else if ( i->stream_type == ANALYSIS_STREAM ) else if ( i->stream_type == ANALYSIS_STREAM )
@ -1166,7 +1166,7 @@ int Manager::SendEntryTable(Stream* i, const Value* const *vals)
// call stream first to determine if we really add / change the entry // call stream first to determine if we really add / change the entry
if ( stream->pred && ! convert_error ) if ( stream->pred && ! convert_error )
{ {
EnumVal* ev; IntrusivePtr<EnumVal> ev;
int startpos = 0; int startpos = 0;
bool pred_convert_error = false; bool pred_convert_error = false;
predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, pred_convert_error); predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, pred_convert_error);
@ -1177,15 +1177,15 @@ int Manager::SendEntryTable(Stream* i, const Value* const *vals)
if ( ! pred_convert_error ) if ( ! pred_convert_error )
{ {
if ( updated ) if ( updated )
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED).release(); ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED);
else else
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW).release(); ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW);
bool result; bool result;
if ( stream->num_val_fields > 0 ) // we have values if ( stream->num_val_fields > 0 ) // we have values
result = CallPred(stream->pred, 3, ev, predidx->Ref(), valval->Ref()); result = CallPred(stream->pred, 3, ev.release(), predidx->Ref(), valval->Ref());
else // no values else // no values
result = CallPred(stream->pred, 2, ev, predidx->Ref()); result = CallPred(stream->pred, 2, ev.release(), predidx->Ref());
if ( result == false ) if ( result == false )
{ {
@ -1265,7 +1265,6 @@ int Manager::SendEntryTable(Stream* i, const Value* const *vals)
if ( stream->event ) if ( stream->event )
{ {
EnumVal* ev;
int startpos = 0; int startpos = 0;
Val* predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, convert_error); Val* predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, convert_error);
@ -1278,20 +1277,20 @@ int Manager::SendEntryTable(Stream* i, const Value* const *vals)
else if ( updated ) else if ( updated )
{ // in case of update send back the old value. { // in case of update send back the old value.
assert ( stream->num_val_fields > 0 ); assert ( stream->num_val_fields > 0 );
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED).release(); auto ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED);
assert ( oldval != nullptr ); assert ( oldval != nullptr );
SendEvent(stream->event, 4, stream->description->Ref(), ev, predidx, oldval.release()); SendEvent(stream->event, 4, stream->description->Ref(), ev.release(), predidx, oldval.release());
} }
else else
{ {
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW).release(); auto ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW);
if ( stream->num_val_fields == 0 ) if ( stream->num_val_fields == 0 )
{ {
Ref(stream->description); Ref(stream->description);
SendEvent(stream->event, 3, stream->description->Ref(), ev, predidx); SendEvent(stream->event, 3, stream->description->Ref(), ev.release(), predidx);
} }
else else
SendEvent(stream->event, 4, stream->description->Ref(), ev, predidx, valval->Ref()); SendEvent(stream->event, 4, stream->description->Ref(), ev.release(), predidx, valval->Ref());
} }
} }
@ -1335,9 +1334,8 @@ void Manager::EndCurrentSend(ReaderFrontend* reader)
while ( ( ih = stream->lastDict->NextEntry(lastDictIdxKey, c) ) ) while ( ( ih = stream->lastDict->NextEntry(lastDictIdxKey, c) ) )
{ {
IntrusivePtr<Val> val; IntrusivePtr<Val> val;
IntrusivePtr<Val> predidx;
Val* predidx = nullptr; IntrusivePtr<EnumVal> ev;
EnumVal* ev = nullptr;
int startpos = 0; int startpos = 0;
if ( stream->pred || stream->event ) if ( stream->pred || stream->event )
@ -1346,25 +1344,21 @@ void Manager::EndCurrentSend(ReaderFrontend* reader)
assert(idx != nullptr); assert(idx != nullptr);
val = stream->tab->Lookup(idx.get()); val = stream->tab->Lookup(idx.get());
assert(val != nullptr); assert(val != nullptr);
predidx = ListValToRecordVal(idx.get(), stream->itype, &startpos); predidx = {AdoptRef{}, ListValToRecordVal(idx.get(), stream->itype, &startpos)};
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED).release(); ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED);
} }
if ( stream->pred ) if ( stream->pred )
{ {
// ask predicate, if we want to expire this element... // ask predicate, if we want to expire this element...
Ref(ev); bool result = CallPred(stream->pred, 3, ev->Ref(), predidx->Ref(),
Ref(predidx); val->Ref());
bool result = CallPred(stream->pred, 3, ev, predidx, IntrusivePtr{val}.release());
if ( result == false ) if ( result == false )
{ {
// Keep it. Hence - we quit and simply go to the next entry of lastDict // Keep it. Hence - we quit and simply go to the next entry of lastDict
// ah well - and we have to add the entry to currDict... // ah well - and we have to add the entry to currDict...
Unref(predidx);
Unref(ev);
stream->currDict->Insert(lastDictIdxKey, stream->lastDict->RemoveEntry(lastDictIdxKey)); stream->currDict->Insert(lastDictIdxKey, stream->lastDict->RemoveEntry(lastDictIdxKey));
delete lastDictIdxKey; delete lastDictIdxKey;
continue; continue;
@ -1372,17 +1366,8 @@ void Manager::EndCurrentSend(ReaderFrontend* reader)
} }
if ( stream->event ) if ( stream->event )
{ SendEvent(stream->event, 4, stream->description->Ref(), ev->Ref(),
Ref(predidx); predidx->Ref(), val->Ref());
Ref(ev);
SendEvent(stream->event, 4, stream->description->Ref(), ev, predidx, IntrusivePtr{val}.release());
}
if ( predidx ) // if we have a stream or an event...
Unref(predidx);
if ( ev )
Unref(ev);
stream->tab->Delete(ih->idxkey); stream->tab->Delete(ih->idxkey);
stream->lastDict->Remove(lastDictIdxKey); // delete in next line stream->lastDict->Remove(lastDictIdxKey); // delete in next line
@ -1454,8 +1439,8 @@ void Manager::Put(ReaderFrontend* reader, Value* *vals)
else if ( i->stream_type == EVENT_STREAM ) else if ( i->stream_type == EVENT_STREAM )
{ {
EnumVal* type = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW).release(); auto type = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW);
readFields = SendEventStreamEvent(i, type, vals); readFields = SendEventStreamEvent(i, type.release(), vals);
} }
else if ( i->stream_type == ANALYSIS_STREAM ) else if ( i->stream_type == ANALYSIS_STREAM )
@ -1581,7 +1566,7 @@ int Manager::PutTable(Stream* i, const Value* const *vals)
// predicate if we want the update or not // predicate if we want the update or not
if ( stream->pred ) if ( stream->pred )
{ {
EnumVal* ev; IntrusivePtr<EnumVal> ev;
int startpos = 0; int startpos = 0;
bool pred_convert_error = false; bool pred_convert_error = false;
Val* predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, pred_convert_error); Val* predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, pred_convert_error);
@ -1591,18 +1576,18 @@ int Manager::PutTable(Stream* i, const Value* const *vals)
else else
{ {
if ( updated ) if ( updated )
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED).release(); ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED);
else else
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW).release(); ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW);
bool result; bool result;
if ( stream->num_val_fields > 0 ) // we have values if ( stream->num_val_fields > 0 ) // we have values
{ {
Ref(valval); Ref(valval);
result = CallPred(stream->pred, 3, ev, predidx, valval); result = CallPred(stream->pred, 3, ev.release(), predidx, valval);
} }
else // no values else // no values
result = CallPred(stream->pred, 2, ev, predidx); result = CallPred(stream->pred, 2, ev.release(), predidx);
if ( result == false ) if ( result == false )
{ {
@ -1619,7 +1604,6 @@ int Manager::PutTable(Stream* i, const Value* const *vals)
if ( stream->event ) if ( stream->event )
{ {
EnumVal* ev;
int startpos = 0; int startpos = 0;
bool event_convert_error = false; bool event_convert_error = false;
Val* predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, event_convert_error); Val* predidx = ValueToRecordVal(i, vals, stream->itype, &startpos, event_convert_error);
@ -1632,20 +1616,20 @@ int Manager::PutTable(Stream* i, const Value* const *vals)
{ {
// in case of update send back the old value. // in case of update send back the old value.
assert ( stream->num_val_fields > 0 ); assert ( stream->num_val_fields > 0 );
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED).release(); auto ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_CHANGED);
assert ( oldval != nullptr ); assert ( oldval != nullptr );
SendEvent(stream->event, 4, stream->description->Ref(), SendEvent(stream->event, 4, stream->description->Ref(),
ev, predidx, oldval.release()); ev.release(), predidx, oldval.release());
} }
else else
{ {
ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW).release(); auto ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_NEW);
if ( stream->num_val_fields == 0 ) if ( stream->num_val_fields == 0 )
SendEvent(stream->event, 4, stream->description->Ref(), SendEvent(stream->event, 4, stream->description->Ref(),
ev, predidx); ev.release(), predidx);
else else
SendEvent(stream->event, 4, stream->description->Ref(), SendEvent(stream->event, 4, stream->description->Ref(),
ev, predidx, valval->Ref()); ev.release(), predidx, valval->Ref());
} }
} }
@ -1724,9 +1708,9 @@ bool Manager::Delete(ReaderFrontend* reader, Value* *vals)
Unref(predidx); Unref(predidx);
else else
{ {
EnumVal* ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED).release(); auto ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED);
streamresult = CallPred(stream->pred, 3, ev, predidx, IntrusivePtr{val}.release()); streamresult = CallPred(stream->pred, 3, ev.release(), predidx, IntrusivePtr{val}.release());
if ( streamresult == false ) if ( streamresult == false )
{ {
@ -1743,8 +1727,8 @@ bool Manager::Delete(ReaderFrontend* reader, Value* *vals)
{ {
Ref(idxval); Ref(idxval);
assert(val != nullptr); assert(val != nullptr);
EnumVal* ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED).release(); auto ev = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED);
SendEvent(stream->event, 4, stream->description->Ref(), ev, idxval, IntrusivePtr{val}.release()); SendEvent(stream->event, 4, stream->description->Ref(), ev.release(), idxval, IntrusivePtr{val}.release());
} }
} }
@ -1758,8 +1742,8 @@ bool Manager::Delete(ReaderFrontend* reader, Value* *vals)
else if ( i->stream_type == EVENT_STREAM ) else if ( i->stream_type == EVENT_STREAM )
{ {
EnumVal* type = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED).release(); auto type = zeek::BifType::Enum::Input::Event->GetVal(BifEnum::Input::EVENT_REMOVED);
readVals = SendEventStreamEvent(i, type, vals); readVals = SendEventStreamEvent(i, type.release(), vals);
success = true; success = true;
} }
@ -2385,7 +2369,8 @@ Val* Manager::ValueToVal(const Stream* i, const Value* val, BroType* request_typ
return nullptr; return nullptr;
} }
return request_type->AsEnumType()->GetVal(index).release(); auto rval = request_type->AsEnumType()->GetVal(index);
return rval.release();
} }
default: default:
@ -2583,7 +2568,8 @@ Val* Manager::ValueToVal(const Stream* i, const Value* val, bool& have_error) co
return nullptr; return nullptr;
} }
return t->GetVal(intval).release(); auto rval = t->GetVal(intval);
return rval.release();
} }
default: default:
@ -2711,19 +2697,19 @@ void Manager::ErrorHandler(const Stream* i, ErrorType et, bool reporter_send, co
// send our script level error event // send our script level error event
if ( i->error_event ) if ( i->error_event )
{ {
EnumVal* ev; IntrusivePtr<EnumVal> ev;
switch (et) switch (et)
{ {
case ErrorType::INFO: case ErrorType::INFO:
ev = zeek::BifType::Enum::Reporter::Level->GetVal(BifEnum::Reporter::INFO).release(); ev = zeek::BifType::Enum::Reporter::Level->GetVal(BifEnum::Reporter::INFO);
break; break;
case ErrorType::WARNING: case ErrorType::WARNING:
ev = zeek::BifType::Enum::Reporter::Level->GetVal(BifEnum::Reporter::WARNING).release(); ev = zeek::BifType::Enum::Reporter::Level->GetVal(BifEnum::Reporter::WARNING);
break; break;
case ErrorType::ERROR: case ErrorType::ERROR:
ev = zeek::BifType::Enum::Reporter::Level->GetVal(BifEnum::Reporter::ERROR).release(); ev = zeek::BifType::Enum::Reporter::Level->GetVal(BifEnum::Reporter::ERROR);
break; break;
default: default:
@ -2732,7 +2718,7 @@ void Manager::ErrorHandler(const Stream* i, ErrorType et, bool reporter_send, co
} }
StringVal* message = new StringVal(buf); StringVal* message = new StringVal(buf);
SendEvent(i->error_event, 3, i->description->Ref(), message, ev); SendEvent(i->error_event, 3, i->description->Ref(), message, ev.release());
} }
if ( reporter_send ) if ( reporter_send )

View file

@ -1290,7 +1290,7 @@ void Manager::SendAllWritersTo(const broker::endpoint_info& ei)
i != stream->writers.end(); i++ ) i != stream->writers.end(); i++ )
{ {
WriterFrontend* writer = i->second->writer; WriterFrontend* writer = i->second->writer;
auto writer_val = et->GetVal(i->first.first); const auto& writer_val = et->GetVal(i->first.first);
broker_mgr->PublishLogCreate((*s)->id, broker_mgr->PublishLogCreate((*s)->id,
writer_val.get(), writer_val.get(),
*i->second->info, *i->second->info,