xmengnet/mihomo
1
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: add Upstream to refconn

Browse source
This commit is contained in:
wwqgtxx 2023-01-16 13:26:30 +08:00
parent e88bddc24f
commit 49a2602329
2 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
common/net/refconn.go
View file

@ -51,6 +51,10 @@ func (c *refConn) SetWriteDeadline(t time.Time) error {
return c.conn.SetWriteDeadline(t) return c.conn.SetWriteDeadline(t)
} }
func (c *refConn) Upstream() any {
return c.conn
}
func NewRefConn(conn net.Conn, ref any) net.Conn { func NewRefConn(conn net.Conn, ref any) net.Conn {
return &refConn{conn: conn, ref: ref} return &refConn{conn: conn, ref: ref}
} }

4
listener/sing/sing.go
View file

@ -48,6 +48,10 @@ func (c *waitCloseConn) RemoteAddr() net.Addr {
return c.rAddr return c.rAddr
} }
func (c *waitCloseConn) Upstream() any {
return c.Conn
}
func (h *ListenerHandler) NewConnection(ctx context.Context, conn net.Conn, metadata M.Metadata) error { func (h *ListenerHandler) NewConnection(ctx context.Context, conn net.Conn, metadata M.Metadata) error {
additions := h.Additions additions := h.Additions
if ctxAdditions := getAdditions(ctx); len(ctxAdditions) > 0 { if ctxAdditions := getAdditions(ctx); len(ctxAdditions) > 0 {