mirror of
https://github.com/zeek/zeek.git
synced 2025-10-02 06:38:20 +00:00
68 lines
1.7 KiB
C++
68 lines
1.7 KiB
C++
// See the file "COPYING" in the main distribution directory for copyright.
|
|
|
|
#include "zeek/TunnelEncapsulation.h"
|
|
|
|
#include "zeek/Conn.h"
|
|
#include "zeek/Reporter.h"
|
|
#include "zeek/util.h"
|
|
|
|
namespace zeek {
|
|
|
|
EncapsulatingConn::EncapsulatingConn(Connection* c, BifEnum::Tunnel::Type t)
|
|
: src_addr(c->OrigAddr()),
|
|
dst_addr(c->RespAddr()),
|
|
src_port(c->OrigPort()),
|
|
dst_port(c->RespPort()),
|
|
proto(c->ConnTransport()),
|
|
type(t),
|
|
uid(c->GetUID()) {
|
|
if ( ! uid ) {
|
|
uid.Set(zeek::detail::bits_per_uid);
|
|
c->SetUID(uid);
|
|
}
|
|
}
|
|
|
|
RecordValPtr EncapsulatingConn::ToVal() const {
|
|
auto rv = make_intrusive<RecordVal>(BifType::Record::Tunnel::EncapsulatingConn);
|
|
|
|
auto id_val = make_intrusive<RecordVal>(id::conn_id);
|
|
id_val->Assign(0, make_intrusive<AddrVal>(src_addr));
|
|
id_val->Assign(1, val_mgr->Port(ntohs(src_port), proto));
|
|
id_val->Assign(2, make_intrusive<AddrVal>(dst_addr));
|
|
id_val->Assign(3, val_mgr->Port(ntohs(dst_port), proto));
|
|
rv->Assign(0, std::move(id_val));
|
|
rv->Assign(1, BifType::Enum::Tunnel::Type->GetEnumVal(type));
|
|
|
|
rv->Assign(2, uid.Base62("C"));
|
|
|
|
return rv;
|
|
}
|
|
|
|
bool operator==(const EncapsulationStack& e1, const EncapsulationStack& e2) {
|
|
if ( ! e1.conns )
|
|
return e2.conns;
|
|
|
|
if ( ! e2.conns )
|
|
return false;
|
|
|
|
if ( e1.conns->size() != e2.conns->size() )
|
|
return false;
|
|
|
|
for ( size_t i = 0; i < e1.conns->size(); ++i ) {
|
|
if ( (*e1.conns)[i] != (*e2.conns)[i] )
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void EncapsulationStack::Pop() {
|
|
if ( Depth() == 0 ) {
|
|
reporter->InternalWarning("Attempted to pop from empty EncapsulationStack\n");
|
|
return;
|
|
}
|
|
|
|
conns->pop_back();
|
|
}
|
|
|
|
} // namespace zeek
|