Change BroValUnion to use IntrusivePtr for record field storage

This also changes the AsRecord() and AsNonConstRecord() accessors
to return std::vector<IntrusivePtr<Val>>* instead of val_list*
This commit is contained in:
Jon Siwek 2020-05-19 17:15:13 -07:00
parent 9583873936
commit 377779bb2a
5 changed files with 38 additions and 36 deletions

3
NEWS
View file

@ -89,6 +89,9 @@ Changed Functionality
- ``TypeList::Types()`` and ``IndexType::IndexTypes()`` now return an
``std::vector`` instead of ``type_list*``
- ``AsRecord()`` and ``AsNonConstRecord()`` have changed to return
``std::vector<IntrusivePtr<Val>>*``.
Removed Functionality
---------------------

View file

@ -910,7 +910,7 @@ Connection* NetSessions::FindConnection(Val* v)
return nullptr;
RecordType* vr = vt->AsRecordType();
const val_list* vl = v->AsRecord();
auto vl = v->AsRecord();
int orig_h, orig_p; // indices into record's value list
int resp_h, resp_p;

View file

@ -2673,7 +2673,8 @@ RecordVal::RecordVal(IntrusivePtr<RecordType> t, bool init_fields) : Val(std::mo
origin = nullptr;
auto rt = GetType()->AsRecordType();
int n = rt->NumFields();
val_list* vl = val.val_list_val = new val_list(n);
auto vl = val.record_val = new std::vector<IntrusivePtr<Val>>;
vl->reserve(n);
if ( is_parsing )
parse_time_records[rt].emplace_back(NewRef{}, this);
@ -2715,13 +2716,13 @@ RecordVal::RecordVal(IntrusivePtr<RecordType> t, bool init_fields) : Val(std::mo
def = make_intrusive<VectorVal>(cast_intrusive<VectorType>(type));
}
vl->push_back(def.release());
vl->emplace_back(std::move(def));
}
}
RecordVal::~RecordVal()
{
delete_vals(AsNonConstRecord());
delete AsNonConstRecord();
}
IntrusivePtr<Val> RecordVal::SizeVal() const
@ -2731,8 +2732,7 @@ IntrusivePtr<Val> RecordVal::SizeVal() const
void RecordVal::Assign(int field, IntrusivePtr<Val> new_val)
{
Val* old_val = AsNonConstRecord()->replace(field, new_val.release());
Unref(old_val);
(*AsNonConstRecord())[field] = std::move(new_val);
Modified();
}
@ -2743,15 +2743,15 @@ void RecordVal::Assign(int field, Val* new_val)
Val* RecordVal::Lookup(int field) const
{
return (*AsRecord())[field];
return (*AsRecord())[field].get();
}
IntrusivePtr<Val> RecordVal::LookupWithDefault(int field) const
{
Val* val = (*AsRecord())[field];
const auto& val = (*AsRecord())[field];
if ( val )
return {NewRef{}, val};
return val;
return GetType()->AsRecordType()->FieldDefault(field);
}
@ -2767,16 +2767,16 @@ void RecordVal::ResizeParseTimeRecords(RecordType* rt)
for ( auto& rv : rvs )
{
auto vs = rv->val.val_list_val;
auto current_length = vs->length();
auto vs = rv->val.record_val;
int current_length = vs->size();
auto required_length = rt->NumFields();
if ( required_length > current_length )
{
vs->resize(required_length);
vs->reserve(required_length);
for ( auto i = current_length; i < required_length; ++i )
vs->replace(i, rt->FieldDefault(i).release());
vs->emplace_back(rt->FieldDefault(i));
}
}
}
@ -2876,8 +2876,8 @@ IntrusivePtr<TableVal> RecordVal::GetRecordFieldsVal() const
void RecordVal::Describe(ODesc* d) const
{
const val_list* vl = AsRecord();
int n = vl->length();
auto vl = AsRecord();
auto n = vl->size();
auto record_type = GetType()->AsRecordType();
if ( d->IsBinary() || d->IsPortable() )
@ -2890,7 +2890,7 @@ void RecordVal::Describe(ODesc* d) const
else
d->Add("[");
loop_over_list(*vl, i)
for ( size_t i = 0; i < n; ++i )
{
if ( ! d->IsBinary() && i > 0 )
d->Add(", ");
@ -2900,7 +2900,8 @@ void RecordVal::Describe(ODesc* d) const
if ( ! d->IsBinary() )
d->Add("=");
Val* v = (*vl)[i];
const auto& v = (*vl)[i];
if ( v )
v->Describe(d);
else
@ -2913,14 +2914,14 @@ void RecordVal::Describe(ODesc* d) const
void RecordVal::DescribeReST(ODesc* d) const
{
const val_list* vl = AsRecord();
int n = vl->length();
auto vl = AsRecord();
auto n = vl->size();
auto record_type = GetType()->AsRecordType();
d->Add("{");
d->PushIndent();
loop_over_list(*vl, i)
for ( size_t i = 0; i < n; ++i )
{
if ( i > 0 )
d->NL();
@ -2928,7 +2929,7 @@ void RecordVal::DescribeReST(ODesc* d) const
d->Add(record_type->FieldName(i));
d->Add("=");
Val* v = (*vl)[i];
const auto& v = (*vl)[i];
if ( v )
v->Describe(d);
@ -2951,10 +2952,10 @@ IntrusivePtr<Val> RecordVal::DoClone(CloneState* state)
rv->origin = nullptr;
state->NewClone(this, rv);
for ( const auto& vlv : *val.val_list_val )
for ( const auto& vlv : *val.record_val)
{
auto v = vlv ? vlv->Clone(state) : nullptr;
rv->val.val_list_val->push_back(v.release());
rv->val.record_val->emplace_back(std::move(v));
}
return rv;
@ -2963,15 +2964,17 @@ IntrusivePtr<Val> RecordVal::DoClone(CloneState* state)
unsigned int RecordVal::MemoryAllocation() const
{
unsigned int size = 0;
const val_list* vl = AsRecord();
const auto& vl = *AsRecord();
for ( const auto& v : *vl )
for ( const auto& v : vl )
{
if ( v )
size += v->MemoryAllocation();
}
return size + padded_sizeof(*this) + val.val_list_val->MemoryAllocation();
size += pad_size(vl.capacity() * sizeof(IntrusivePtr<Val>));
size += padded_sizeof(vl);
return size + padded_sizeof(*this);
}
IntrusivePtr<Val> EnumVal::SizeVal() const

View file

@ -80,8 +80,7 @@ union BroValUnion {
BroFile* file_val;
RE_Matcher* re_val;
PDict<TableEntryVal>* table_val;
val_list* val_list_val;
std::vector<IntrusivePtr<Val>>* record_val;
std::vector<Val*>* vector_val;
BroValUnion() = default;
@ -116,9 +115,6 @@ union BroValUnion {
constexpr BroValUnion(PDict<TableEntryVal>* value) noexcept
: table_val(value) {}
constexpr BroValUnion(val_list* value) noexcept
: val_list_val(value) {}
constexpr BroValUnion(std::vector<Val*> *value) noexcept
: vector_val(value) {}
};
@ -222,7 +218,7 @@ public:
CONST_ACCESSOR(TYPE_STRING, BroString*, string_val, AsString)
CONST_ACCESSOR(TYPE_FUNC, Func*, func_val, AsFunc)
CONST_ACCESSOR(TYPE_TABLE, PDict<TableEntryVal>*, table_val, AsTable)
CONST_ACCESSOR(TYPE_RECORD, val_list*, val_list_val, AsRecord)
CONST_ACCESSOR(TYPE_RECORD, std::vector<IntrusivePtr<Val>>*, record_val, AsRecord)
CONST_ACCESSOR(TYPE_FILE, BroFile*, file_val, AsFile)
CONST_ACCESSOR(TYPE_PATTERN, RE_Matcher*, re_val, AsPattern)
CONST_ACCESSOR(TYPE_VECTOR, std::vector<Val*>*, vector_val, AsVector)
@ -368,7 +364,7 @@ protected:
{}
ACCESSOR(TYPE_TABLE, PDict<TableEntryVal>*, table_val, AsNonConstTable)
ACCESSOR(TYPE_RECORD, val_list*, val_list_val, AsNonConstRecord)
ACCESSOR(TYPE_RECORD, std::vector<IntrusivePtr<Val>>*, record_val, AsNonConstRecord)
// For internal use by the Val::Clone() methods.
struct CloneState {

View file

@ -3333,8 +3333,8 @@ function lookup_connection%(cid: conn_id%): connection
%%{
const char* conn_id_string(Val* c)
{
Val* id = (*(c->AsRecord()))[0];
const val_list* vl = id->AsRecord();
const auto& id = (*(c->AsRecord()))[0];
auto vl = id->AsRecord();
const IPAddr& orig_h = (*vl)[0]->AsAddr();
uint32_t orig_p = (*vl)[1]->AsPortVal()->Port();
@ -3459,7 +3459,7 @@ function dump_packet%(pkt: pcap_packet, file_name: string%) : bool
uint32_t caplen, len, link_type;
u_char *data;
const val_list* pkt_vl = pkt->AsRecord();
auto pkt_vl = pkt->AsRecord();
ts.tv_sec = (*pkt_vl)[0]->AsCount();
ts.tv_usec = (*pkt_vl)[1]->AsCount();