TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2024-11-25 19:15:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

tsnet: add Start method to allow connecting to the tailnet without

Browse Source 
requiring a call to Dial/Listen.

Signed-off-by: Maisem Ali <maisem@tailscale.com>
This commit is contained in:
Maisem Ali 2022-01-03 11:20:40 -08:00 committed by Maisem Ali
parent ebdd25920e
commit dd45bba76b
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tsnet/tsnet.go
View File

@ -67,8 +67,9 @@ type Server struct {
}
// Dial connects to the address on the tailnet.
// It will start the server if it has not been started yet.
func (s *Server) Dial(ctx context.Context, network, address string) (net.Conn, error) {
if err := s.init(); err != nil {
if err := s.Start(); err != nil {
return nil, err
}
return s.dialer.UserDial(ctx, network, address)
@ -80,7 +81,9 @@ func (s *Server) doInit() {
}
}
func (s *Server) init() error {
// Start connects the server to the tailnet.
// Optional: any calls to Dial/Listen will also call Start.
func (s *Server) Start() error {
s.initOnce.Do(s.doInit)
return s.initErr
}
@ -232,13 +235,14 @@ func (s *Server) forwardTCP(c net.Conn, port uint16) {
}
// Listen announces only on the Tailscale network.
// It will start the server if it has not been started yet.
func (s *Server) Listen(network, addr string) (net.Listener, error) {
host, port, err := net.SplitHostPort(addr)
if err != nil {
return nil, fmt.Errorf("tsnet: %w", err)
}
if err := s.init(); err != nil {
if err := s.Start(); err != nil {
return nil, err
}