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

Change IP_Hdr copy ctor/assign to explicit method

Browse source 
Addresses BIT-1298
This commit is contained in:
Jon Siwek 2014-12-12 14:14:24 -06:00
parent c211a2c91a
commit d31b556b85
3 changed files with 14 additions and 46 deletions
Download patch file Download diff file Expand all files Collapse all files

41
src/IP.cc
View file

@ -4,8 +4,6 @@
#include "Type.h"
#include "Val.h"
#include "Var.h"
#include <algorithm>
#include <utility>
static RecordType* ip4_hdr_type = 0;
static RecordType* ip6_hdr_type = 0;
@ -639,32 +637,20 @@ VectorVal* IPv6_Hdr_Chain::BuildVal() const
return rval;
}
IP_Hdr::IP_Hdr(const IP_Hdr& other)
IP_Hdr* IP_Hdr::Copy() const
{
del = true;
char* new_hdr = new char[HdrLen()];
if ( other.ip4 )
if ( ip4 )
{
char* new_hdr = new char[other.HdrLen()];
memcpy(new_hdr, other.ip4, other.HdrLen());
ip4 = (const struct ip*)new_hdr;
ip6 = 0;
ip6_hdrs = 0;
memcpy(new_hdr, ip4, HdrLen());
return new IP_Hdr((const struct ip*) new_hdr, true);
}
else
{
ip4 = 0;
char* new_hdr = new char[other.HdrLen()];
memcpy(new_hdr, other.ip6, other.HdrLen());
ip6 = (const struct ip6_hdr*)new_hdr;
ip6_hdrs = other.ip6_hdrs->Copy(ip6);
}
}
IP_Hdr& IP_Hdr::operator=(IP_Hdr other)
{
swap(*this, other);
return *this;
memcpy(new_hdr, ip6, HdrLen());
const struct ip6_hdr* new_ip6 = (const struct ip6_hdr*)new_hdr;
IPv6_Hdr_Chain* new_ip6_hdrs = ip6_hdrs->Copy(new_ip6);
return new IP_Hdr(new_ip6, true, 0, new_ip6_hdrs);
}
IPv6_Hdr_Chain* IPv6_Hdr_Chain::Copy(const ip6_hdr* new_hdr) const
@ -698,12 +684,3 @@ IPv6_Hdr_Chain* IPv6_Hdr_Chain::Copy(const ip6_hdr* new_hdr) const
return rval;
}
void swap(IP_Hdr& a, IP_Hdr& b)
{
using std::swap;
swap(a.ip4, b.ip4);
swap(a.ip6, b.ip6);
swap(a.del, b.del);
swap(a.ip6_hdrs, b.ip6_hdrs);
}

17
src/IP.h
View file

@ -368,18 +368,11 @@ public:
}
/**
* Copy constructor. The internal buffer of \a other which contains
* the header data must not be truncated. Also not that if that buffer
* points to a full packet payload, only the IP header portion is copied.
* Copy a header. The internal buffer which contains the header data
* must not be truncated. Also note that if that buffer points to a full
* packet payload, only the IP header portion is copied.
*/
IP_Hdr(const IP_Hdr& other);
/**
* Copy assignment. The internal buffer of \a other which contains
* the header data must not be truncated. Also not that if that buffer
* points to a full packet payload, only the IP header portion is copied.
*/
IP_Hdr& operator=(IP_Hdr other);
IP_Hdr* Copy() const;
/**
* Destructor.
@ -581,8 +574,6 @@ public:
*/
RecordVal* BuildPktHdrVal() const;
friend void swap(IP_Hdr& a, IP_Hdr& b);
private:
const struct ip* ip4;

2
src/analyzer/protocol/pia/PIA.cc
View file

@ -43,7 +43,7 @@ void PIA::AddToBuffer(Buffer* buffer, uint64 seq, int len, const u_char* data,
}
DataBlock* b = new DataBlock;
b->ip = ip ? new IP_Hdr(*ip) : 0;
b->ip = ip ? ip->Copy() : 0;
b->data = tmp;
b->is_orig = is_orig;
b->len = len;