Mirror/zeek
1
0
Fork 0
mirror of https://github.com/zeek/zeek.git synced 2025-10-02 06:38:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Switch IPv6_Hdr_Chain to a vector of objects instead of pointers

Browse source
This commit is contained in:
Tim Wojtulewicz 2025-06-06 12:10:49 -07:00
parent 33b23ef4b1
commit d6e25a2e22
2 changed files with 17 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/IP.cc
View file

@ -457,8 +457,6 @@ static inline bool isIPv6ExtHeader(uint8_t type) {
}
IPv6_Hdr_Chain::~IPv6_Hdr_Chain() {
for ( auto& c : chain )
delete c;
delete homeAddr;
delete finalDst;
}
@ -481,23 +479,21 @@ void IPv6_Hdr_Chain::Init(const struct ip6_hdr* ip6, uint64_t total_len, bool se
return;
current_type = next_type;
IPv6_Hdr* p = new IPv6_Hdr(current_type, hdrs);
IPv6_Hdr p{current_type, hdrs};
next_type = p->NextHdr();
uint16_t cur_len = p->Length();
next_type = p.NextHdr();
uint16_t cur_len = p.Length();
// If this header is truncated, don't add it to chain, don't go further.
if ( cur_len > total_len ) {
delete p;
if ( cur_len > total_len )
return;
}
if ( set_next && next_type == IPPROTO_FRAGMENT ) {
p->ChangeNext(next);
p.ChangeNext(next);
next_type = next;
}
chain.push_back(p);
chain.emplace_back(std::move(p));
// Check for routing headers and remember final destination address.
if ( current_type == IPPROTO_ROUTING )
@ -521,7 +517,7 @@ bool IPv6_Hdr_Chain::IsFragment() const {
return false;
}
return chain[chain.size() - 1]->Type() == IPPROTO_FRAGMENT;
return chain.back().Type() == IPPROTO_FRAGMENT;
}
IPAddr IPv6_Hdr_Chain::SrcAddr() const {
@ -533,7 +529,7 @@ IPAddr IPv6_Hdr_Chain::SrcAddr() const {
return {};
}
return IPAddr{((const struct ip6_hdr*)(chain[0]->Data()))->ip6_src};
return IPAddr{((const struct ip6_hdr*)(chain[0].Data()))->ip6_src};
}
IPAddr IPv6_Hdr_Chain::DstAddr() const {
@ -545,7 +541,7 @@ IPAddr IPv6_Hdr_Chain::DstAddr() const {
return {};
}
return IPAddr{((const struct ip6_hdr*)(chain[0]->Data()))->ip6_dst};
return IPAddr{((const struct ip6_hdr*)(chain[0].Data()))->ip6_dst};
}
void IPv6_Hdr_Chain::ProcessRoutingHeader(const struct ip6_rthdr* r, uint16_t len) {
@ -647,9 +643,9 @@ VectorValPtr IPv6_Hdr_Chain::ToVal() const {
auto rval = make_intrusive<VectorVal>(ip6_ext_hdr_chain_type);
for ( size_t i = 1; i < chain.size(); ++i ) {
auto v = chain[i]->ToVal();
auto v = chain[i].ToVal();
auto ext_hdr = make_intrusive<RecordVal>(ip6_ext_hdr_type);
uint8_t type = chain[i]->Type();
uint8_t type = chain[i].Type();
ext_hdr->Assign(0, type);
switch ( type ) {
@ -700,11 +696,11 @@ IPv6_Hdr_Chain* IPv6_Hdr_Chain::Copy(const ip6_hdr* new_hdr) const {
}
const u_char* new_data = (const u_char*)new_hdr;
const u_char* old_data = chain[0]->Data();
const u_char* old_data = chain[0].Data();
for ( const auto& c : chain ) {
int off = c->Data() - old_data;
rval->chain.push_back(new IPv6_Hdr(c->Type(), new_data + off));
int off = c.Data() - old_data;
rval->chain.emplace_back(c.Type(), new_data + off);
}
return rval;

6
src/IP.h
View file

@ -163,7 +163,7 @@ public:
/**
* Accesses the header at the given location in the chain.
*/
const IPv6_Hdr* operator[](const size_t i) const { return chain[i]; }
const IPv6_Hdr* operator[](const size_t i) const { return &chain[i]; }
/**
* Returns whether the header chain indicates a fragmented packet.
@ -174,7 +174,7 @@ public:
* Returns pointer to fragment header structure if the chain contains one.
*/
const struct ip6_frag* GetFragHdr() const {
return IsFragment() ? (const struct ip6_frag*)chain[chain.size() - 1]->Data() : nullptr;
return IsFragment() ? (const struct ip6_frag*)chain[chain.size() - 1].Data() : nullptr;
}
/**
@ -245,7 +245,7 @@ protected:
*/
void ProcessDstOpts(const struct ip6_dest* d, uint16_t len);
std::vector<IPv6_Hdr*> chain;
std::vector<IPv6_Hdr> chain;
/**
* The summation of all header lengths in the chain in bytes.