diff --git a/limiter/quota/wrapper/conn.go b/limiter/quota/wrapper/conn.go index 0c7d4c76..dbe97e8c 100644 --- a/limiter/quota/wrapper/conn.go +++ b/limiter/quota/wrapper/conn.go @@ -76,3 +76,9 @@ func (c *quotaConn) Context() context.Context { } return nil } + +// UnwrapConn returns the underlying connection, allowing type assertions +// in handlers that iterate wrapper layers (e.g., SSH handler's unwrapConn). +func (c *quotaConn) UnwrapConn() net.Conn { + return c.Conn +}