From cc56100f159b6bddee810713e7c42e7f886a488d Mon Sep 17 00:00:00 2001 From: xjasonlyu Date: Sat, 5 Feb 2022 20:14:05 +0800 Subject: [PATCH] Chore(metadata): add TCPAddr method --- metadata/metadata.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/metadata/metadata.go b/metadata/metadata.go index 16125ca..2205dd4 100644 --- a/metadata/metadata.go +++ b/metadata/metadata.go @@ -35,6 +35,16 @@ func (m *Metadata) Addr() net.Addr { return &Addr{metadata: m} } +func (m *Metadata) TCPAddr() *net.TCPAddr { + if m.Network != TCP || m.DstIP == nil { + return nil + } + return &net.TCPAddr{ + IP: m.DstIP, + Port: int(m.DstPort), + } +} + func (m *Metadata) UDPAddr() *net.UDPAddr { if m.Network != UDP || m.DstIP == nil { return nil