safesocket: add ConnectContext

This adds a variant for Connect that takes in a context.Context
which allows passing through cancellation etc by the caller.

Updates tailscale/corp#18266

Signed-off-by: Maisem Ali <maisem@tailscale.com>
This commit is contained in:
Maisem Ali
2024-06-10 19:38:10 -07:00
committed by Maisem Ali
parent 3672f66c74
commit 4b6a0c42c8
9 changed files with 32 additions and 17 deletions

View File

@@ -4,6 +4,7 @@
package safesocket
import (
"context"
"net"
"github.com/akutz/memconn"
@@ -15,6 +16,6 @@ func listen(path string) (net.Listener, error) {
return memconn.Listen("memu", memName)
}
func connect(_ string) (net.Conn, error) {
return memconn.Dial("memu", memName)
func connect(ctx context.Context, _ string) (net.Conn, error) {
return memconn.DialContext(ctx, "memu", memName)
}