ayanamistandHarry Harpham
50f1c285ba
Allow `tailscale serve --tcp <port> unix:/path/to/socket` and `tailscale serve --tls-terminated-tcp <port> unix:/path/to/socket` to forward TCP connections to a Unix domain socket. Previously only host:port targets were supported for TCP serve mode. Updates #20161 Signed-off-by: ayanamist <ayanamist@gmail.com>
109 lines
2.5 KiB
Go
109 lines
2.5 KiB
Go
// Copyright (c) Tailscale Inc & contributors
|
|
// SPDX-License-Identifier: BSD-3-Clause
|
|
|
|
//go:build !ts_omit_serve
|
|
|
|
package conffile
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"tailscale.com/tailcfg"
|
|
)
|
|
|
|
func TestTargetUnixSocketRoundtrip(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
serialized string
|
|
want Target
|
|
}{
|
|
{
|
|
name: "tcp_unix_socket",
|
|
serialized: "tcp://unix:/var/run/app.sock",
|
|
want: Target{
|
|
Protocol: ProtoTCP,
|
|
Destination: "unix:/var/run/app.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "tls_terminated_tcp_unix_socket",
|
|
serialized: "tls-terminated-tcp://unix:/var/run/app.sock",
|
|
want: Target{
|
|
Protocol: ProtoTLSTerminatedTCP,
|
|
Destination: "unix:/var/run/app.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "tcp_unix_socket_relative",
|
|
serialized: "tcp://unix:relative.sock",
|
|
want: Target{
|
|
Protocol: ProtoTCP,
|
|
Destination: "unix:relative.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "http_unix_socket",
|
|
serialized: "http://unix:/var/run/app.sock",
|
|
want: Target{
|
|
Protocol: ProtoHTTP,
|
|
Destination: "unix:/var/run/app.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "https_unix_socket",
|
|
serialized: "https://unix:/var/run/app.sock",
|
|
want: Target{
|
|
Protocol: ProtoHTTPS,
|
|
Destination: "unix:/var/run/app.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "https_insecure_unix_socket",
|
|
serialized: "https+insecure://unix:/var/run/app.sock",
|
|
want: Target{
|
|
Protocol: ProtoHTTPSInsecure,
|
|
Destination: "unix:/var/run/app.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "http_unix_socket_relative",
|
|
serialized: "http://unix:relative.sock",
|
|
want: Target{
|
|
Protocol: ProtoHTTP,
|
|
Destination: "unix:relative.sock",
|
|
},
|
|
},
|
|
{
|
|
name: "tcp_host_port",
|
|
serialized: "tcp://localhost:5432",
|
|
want: Target{
|
|
Protocol: ProtoTCP,
|
|
Destination: "localhost",
|
|
DestinationPorts: tailcfg.PortRange{First: 5432, Last: 5432},
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
// Test unmarshal
|
|
var got Target
|
|
if err := got.UnmarshalJSON([]byte(`"` + tt.serialized + `"`)); err != nil {
|
|
t.Fatalf("UnmarshalJSON(%q) failed: %v", tt.serialized, err)
|
|
}
|
|
if got != tt.want {
|
|
t.Errorf("UnmarshalJSON(%q) = %+v, want %+v", tt.serialized, got, tt.want)
|
|
}
|
|
|
|
// Test marshal roundtrip
|
|
marshaled, err := tt.want.MarshalText()
|
|
if err != nil {
|
|
t.Fatalf("MarshalText() failed: %v", err)
|
|
}
|
|
if string(marshaled) != tt.serialized {
|
|
t.Errorf("MarshalText() = %q, want %q", marshaled, tt.serialized)
|
|
}
|
|
})
|
|
}
|
|
}
|