print address on compressed connection

xtaci · Jul 11, 2019 · b4473a5 · b4473a5
1 parent 3b01e69
commit b4473a5
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/generic/comp.go b/generic/comp.go
@@ -32,6 +32,24 @@ func (c *CompStream) Close() error {
 	return c.conn.Close()
 }
 
+func (c *CompStream) LocalAddr() net.Addr {
+	if ts, ok := c.conn.(interface {
+		LocalAddr() net.Addr
+	}); ok {
+		return ts.LocalAddr()
+	}
+	return nil
+}
+
+func (c *CompStream) RemoteAddr() net.Addr {
+	if ts, ok := c.conn.(interface {
+		RemoteAddr() net.Addr
+	}); ok {
+		return ts.RemoteAddr()
+	}
+	return nil
+}
+
 func NewCompStream(conn net.Conn) *CompStream {
 	c := new(CompStream)
 	c.conn = conn