This commit is contained in:
Aaron Paterson 2025-12-04 16:32:18 -05:00 committed by GitHub
commit 39d7d01b99
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
15 changed files with 699 additions and 468 deletions

View file

@ -222,7 +222,7 @@ func (admin *AdminConfig) newAdminHandler(addr NetworkAddress, remote bool, _ Co
muxWrap.remoteControl = admin.Remote
} else {
// see comment in allowedOrigins() as to why we disable the host check for unix/fd networks
muxWrap.enforceHost = !addr.isWildcardInterface() && !addr.IsUnixNetwork() && !addr.IsFdNetwork()
muxWrap.enforceHost = !addr.isWildcardInterface() && !addr.IsUnixNetwork() && !addr.IsFDNetwork()
muxWrap.allowedOrigins = admin.allowedOrigins(addr)
muxWrap.enforceOrigin = admin.EnforceOrigin
}
@ -342,7 +342,7 @@ func (admin AdminConfig) allowedOrigins(addr NetworkAddress) []*url.URL {
// and a false sense of security.
//
// See also the discussion in #6832.
if admin.Origins == nil && !addr.IsUnixNetwork() && !addr.IsFdNetwork() {
if admin.Origins == nil && !addr.IsUnixNetwork() && !addr.IsFDNetwork() {
if addr.isLoopback() {
uniqueOrigins[net.JoinHostPort("localhost", addr.port())] = struct{}{}
uniqueOrigins[net.JoinHostPort("::1", addr.port())] = struct{}{}

View file

@ -729,7 +729,7 @@ func AdminAPIRequest(adminAddr, method, uri string, headers http.Header, body io
return nil, err
}
parsedAddr.Host = addr
} else if parsedAddr.IsFdNetwork() {
} else if parsedAddr.IsFDNetwork() {
origin = "http://127.0.0.1"
}
@ -738,7 +738,7 @@ func AdminAPIRequest(adminAddr, method, uri string, headers http.Header, body io
if err != nil {
return nil, fmt.Errorf("making request: %v", err)
}
if parsedAddr.IsUnixNetwork() || parsedAddr.IsFdNetwork() {
if parsedAddr.IsUnixNetwork() || parsedAddr.IsFDNetwork() {
// We used to conform to RFC 2616 Section 14.26 which requires
// an empty host header when there is no host, as is the case
// with unix sockets and socket fds. However, Go required a

View file

@ -38,10 +38,6 @@ import (
"github.com/caddyserver/caddy/v2/internal"
)
// listenFdsStart is the first file descriptor number for systemd socket activation.
// File descriptors 0, 1, 2 are reserved for stdin, stdout, stderr.
const listenFdsStart = 3
// NetworkAddress represents one or more network addresses.
// It contains the individual components for a parsed network
// address of the form accepted by ParseNetworkAddress().
@ -137,42 +133,45 @@ func (na NetworkAddress) ListenAll(ctx context.Context, config net.ListenConfig)
// Listen synchronizes binds to unix domain sockets to avoid race conditions
// while an existing socket is unlinked.
func (na NetworkAddress) Listen(ctx context.Context, portOffset uint, config net.ListenConfig) (any, error) {
if na.IsUnixNetwork() {
unixSocketsMu.Lock()
defer unixSocketsMu.Unlock()
}
var (
ln any
err error
)
// check to see if plugin provides listener
if ln, err := getListenerFromPlugin(ctx, na.Network, na.Host, na.port(), portOffset, config); ln != nil || err != nil {
// check to see if plugin provides a listener
if ln, err = getListenerFromPlugin(ctx, na.Network, na.Host, na.port(), portOffset, config); ln != nil || err != nil {
return ln, err
}
// create (or reuse) the listener ourselves
return na.listen(ctx, portOffset, config)
}
func (na NetworkAddress) listen(ctx context.Context, portOffset uint, config net.ListenConfig) (any, error) {
var (
ln any
err error
address string
unixFileMode fs.FileMode
)
// lock other unix sockets from being bound and
// split unix socket addr early so lnKey
// is independent of permissions bits
if na.IsUnixNetwork() {
unixSocketsMu.Lock()
defer unixSocketsMu.Unlock()
address, unixFileMode, err = internal.SplitUnixSocketPermissionsBits(na.Host)
if err != nil {
return nil, err
}
} else if na.IsFdNetwork() {
address = na.Host
} else if na.IsFDNetwork() {
socketFd, err := strconv.ParseUint(na.Host, 0, strconv.IntSize)
if err != nil {
return nil, fmt.Errorf("invalid file descriptor: %v", err)
}
address = strconv.FormatUint(uint64(uint(socketFd)+portOffset), 10)
} else {
address = na.JoinHostPort(portOffset)
}
if strings.HasPrefix(na.Network, "ip") {
if na.IsIPNetwork() {
ln, err = config.ListenPacket(ctx, na.Network, address)
} else {
if na.IsUnixNetwork() {
@ -209,21 +208,39 @@ func (na NetworkAddress) listen(ctx context.Context, portOffset uint, config net
}
// IsUnixNetwork returns true if na.Network is
// unix, unixgram, or unixpacket.
// unix, unixgram, unixpacket, or unix+h2c.
func (na NetworkAddress) IsUnixNetwork() bool {
return IsUnixNetwork(na.Network)
}
// IsFdNetwork returns true if na.Network is
// IsTCPNetwork returns true if na.Network is
// tcp, tcp4, or tcp6.
func (na NetworkAddress) IsTCPNetwork() bool {
return IsTCPNetwork(na.Network)
}
// IsUDPNetwork returns true if na.Network is
// udp, udp4, or udp6.
func (na NetworkAddress) IsUDPNetwork() bool {
return IsUDPNetwork(na.Network)
}
// IsIPNetwork returns true if na.Network starts with
// ip: ip4: or ip6:
func (na NetworkAddress) IsIPNetwork() bool {
return IsIPNetwork(na.Network)
}
// IsFDNetwork returns true if na.Network is
// fd or fdgram.
func (na NetworkAddress) IsFdNetwork() bool {
return IsFdNetwork(na.Network)
func (na NetworkAddress) IsFDNetwork() bool {
return IsFDNetwork(na.Network)
}
// JoinHostPort is like net.JoinHostPort, but where the port
// is StartPort + offset.
func (na NetworkAddress) JoinHostPort(offset uint) string {
if na.IsUnixNetwork() || na.IsFdNetwork() {
if na.IsUnixNetwork() || na.IsFDNetwork() {
return na.Host
}
return net.JoinHostPort(na.Host, strconv.FormatUint(uint64(na.StartPort+offset), 10))
@ -260,7 +277,7 @@ func (na NetworkAddress) PortRangeSize() uint {
}
func (na NetworkAddress) isLoopback() bool {
if na.IsUnixNetwork() || na.IsFdNetwork() {
if na.IsUnixNetwork() || na.IsFDNetwork() {
return true
}
if na.Host == "localhost" {
@ -293,80 +310,12 @@ func (na NetworkAddress) port() string {
// The output can be parsed by ParseNetworkAddress(). If the
// address is a unix socket, any non-zero port will be dropped.
func (na NetworkAddress) String() string {
if na.Network == "tcp" && (na.Host != "" || na.port() != "") {
if na.Network == TCP && (na.Host != "" || na.port() != "") {
na.Network = "" // omit default network value for brevity
}
return JoinNetworkAddress(na.Network, na.Host, na.port())
}
// IsUnixNetwork returns true if the netw is a unix network.
func IsUnixNetwork(netw string) bool {
return strings.HasPrefix(netw, "unix")
}
// IsFdNetwork returns true if the netw is a fd network.
func IsFdNetwork(netw string) bool {
return strings.HasPrefix(netw, "fd")
}
// getFdByName returns the file descriptor number for the given
// socket name from systemd's LISTEN_FDNAMES environment variable.
// Socket names are provided by systemd via socket activation.
//
// The name can optionally include an index to handle multiple sockets
// with the same name: "web:0" for first, "web:1" for second, etc.
// If no index is specified, defaults to index 0 (first occurrence).
func getFdByName(nameWithIndex string) (int, error) {
if nameWithIndex == "" {
return 0, fmt.Errorf("socket name cannot be empty")
}
fdNamesStr := os.Getenv("LISTEN_FDNAMES")
if fdNamesStr == "" {
return 0, fmt.Errorf("LISTEN_FDNAMES environment variable not set")
}
// Parse name and optional index
parts := strings.Split(nameWithIndex, ":")
if len(parts) > 2 {
return 0, fmt.Errorf("invalid socket name format '%s': too many colons", nameWithIndex)
}
name := parts[0]
targetIndex := 0
if len(parts) > 1 {
var err error
targetIndex, err = strconv.Atoi(parts[1])
if err != nil {
return 0, fmt.Errorf("invalid socket index '%s': %v", parts[1], err)
}
if targetIndex < 0 {
return 0, fmt.Errorf("socket index cannot be negative: %d", targetIndex)
}
}
// Parse the socket names
names := strings.Split(fdNamesStr, ":")
// Find the Nth occurrence of the requested name
matchCount := 0
for i, fdName := range names {
if fdName == name {
if matchCount == targetIndex {
return listenFdsStart + i, nil
}
matchCount++
}
}
if matchCount == 0 {
return 0, fmt.Errorf("socket name '%s' not found in LISTEN_FDNAMES", name)
}
return 0, fmt.Errorf("socket name '%s' found %d times, but index %d requested", name, matchCount, targetIndex)
}
// ParseNetworkAddress parses addr into its individual
// components. The input string is expected to be of
// the form "network/host:port-range" where any part is
@ -397,28 +346,10 @@ func ParseNetworkAddressWithDefaults(addr, defaultNetwork string, defaultPort ui
Host: host,
}, err
}
if IsFdNetwork(network) {
fdAddr := host
// Handle named socket activation (fdname/name, fdgramname/name)
if strings.HasPrefix(network, "fdname") || strings.HasPrefix(network, "fdgramname") {
fdNum, err := getFdByName(host)
if err != nil {
return NetworkAddress{}, fmt.Errorf("named socket activation: %v", err)
}
fdAddr = strconv.Itoa(fdNum)
// Normalize network to standard fd/fdgram
if strings.HasPrefix(network, "fdname") {
network = "fd"
} else {
network = "fdgram"
}
}
if IsFDNetwork(network) {
return NetworkAddress{
Network: network,
Host: fdAddr,
Host: host,
}, nil
}
var start, end uint64
@ -460,7 +391,7 @@ func SplitNetworkAddress(a string) (network, host, port string, err error) {
if slashFound {
network = strings.ToLower(strings.TrimSpace(beforeSlash))
a = afterSlash
if IsUnixNetwork(network) || IsFdNetwork(network) {
if IsUnixNetwork(network) || IsFDNetwork(network) {
host = a
return network, host, port, err
}
@ -495,7 +426,7 @@ func JoinNetworkAddress(network, host, port string) string {
if network != "" {
a = network + "/"
}
if (host != "" && port == "") || IsUnixNetwork(network) || IsFdNetwork(network) {
if (host != "" && port == "") || IsUnixNetwork(network) || IsFDNetwork(network) {
a += host
} else if port != "" {
a += net.JoinHostPort(host, port)
@ -720,55 +651,12 @@ func (fcql *fakeCloseQuicListener) Close() error {
return nil
}
// RegisterNetwork registers a network type with Caddy so that if a listener is
// created for that network type, getListener will be invoked to get the listener.
// This should be called during init() and will panic if the network type is standard
// or reserved, or if it is already registered. EXPERIMENTAL and subject to change.
func RegisterNetwork(network string, getListener ListenerFunc) {
network = strings.TrimSpace(strings.ToLower(network))
if network == "tcp" || network == "tcp4" || network == "tcp6" ||
network == "udp" || network == "udp4" || network == "udp6" ||
network == "unix" || network == "unixpacket" || network == "unixgram" ||
strings.HasPrefix(network, "ip:") || strings.HasPrefix(network, "ip4:") || strings.HasPrefix(network, "ip6:") ||
network == "fd" || network == "fdgram" {
panic("network type " + network + " is reserved")
}
if _, ok := networkTypes[strings.ToLower(network)]; ok {
panic("network type " + network + " is already registered")
}
networkTypes[network] = getListener
}
var unixSocketsMu sync.Mutex
// getListenerFromPlugin returns a listener on the given network and address
// if a plugin has registered the network name. It may return (nil, nil) if
// no plugin can provide a listener.
func getListenerFromPlugin(ctx context.Context, network, host, port string, portOffset uint, config net.ListenConfig) (any, error) {
// get listener from plugin if network type is registered
if getListener, ok := networkTypes[network]; ok {
Log().Debug("getting listener from plugin", zap.String("network", network))
return getListener(ctx, network, host, port, portOffset, config)
}
return nil, nil
}
func listenerKey(network, addr string) string {
return network + "/" + addr
}
// ListenerFunc is a function that can return a listener given a network and address.
// The listeners must be capable of overlapping: with Caddy, new configs are loaded
// before old ones are unloaded, so listeners may overlap briefly if the configs
// both need the same listener. EXPERIMENTAL and subject to change.
type ListenerFunc func(ctx context.Context, network, host, portRange string, portOffset uint, cfg net.ListenConfig) (any, error)
var networkTypes = map[string]ListenerFunc{}
// ListenerWrapper is a type that wraps a listener
// so it can modify the input listener's methods.
// Modules that implement this interface are found

View file

@ -15,7 +15,6 @@
package caddy
import (
"os"
"reflect"
"testing"
@ -653,286 +652,3 @@ func TestSplitUnixSocketPermissionsBits(t *testing.T) {
}
}
}
// TestGetFdByName tests the getFdByName function for systemd socket activation.
func TestGetFdByName(t *testing.T) {
// Save original environment
originalFdNames := os.Getenv("LISTEN_FDNAMES")
// Restore environment after test
defer func() {
if originalFdNames != "" {
os.Setenv("LISTEN_FDNAMES", originalFdNames)
} else {
os.Unsetenv("LISTEN_FDNAMES")
}
}()
tests := []struct {
name string
fdNames string
socketName string
expectedFd int
expectError bool
}{
{
name: "simple http socket",
fdNames: "http",
socketName: "http",
expectedFd: 3,
},
{
name: "multiple different sockets - first",
fdNames: "http:https:dns",
socketName: "http",
expectedFd: 3,
},
{
name: "multiple different sockets - second",
fdNames: "http:https:dns",
socketName: "https",
expectedFd: 4,
},
{
name: "multiple different sockets - third",
fdNames: "http:https:dns",
socketName: "dns",
expectedFd: 5,
},
{
name: "duplicate names - first occurrence (no index)",
fdNames: "web:web:api",
socketName: "web",
expectedFd: 3,
},
{
name: "duplicate names - first occurrence (explicit index 0)",
fdNames: "web:web:api",
socketName: "web:0",
expectedFd: 3,
},
{
name: "duplicate names - second occurrence (index 1)",
fdNames: "web:web:api",
socketName: "web:1",
expectedFd: 4,
},
{
name: "complex duplicates - first api",
fdNames: "web:api:web:api:dns",
socketName: "api:0",
expectedFd: 4,
},
{
name: "complex duplicates - second api",
fdNames: "web:api:web:api:dns",
socketName: "api:1",
expectedFd: 6,
},
{
name: "complex duplicates - first web",
fdNames: "web:api:web:api:dns",
socketName: "web:0",
expectedFd: 3,
},
{
name: "complex duplicates - second web",
fdNames: "web:api:web:api:dns",
socketName: "web:1",
expectedFd: 5,
},
{
name: "socket not found",
fdNames: "http:https",
socketName: "missing",
expectError: true,
},
{
name: "empty socket name",
fdNames: "http",
socketName: "",
expectError: true,
},
{
name: "missing LISTEN_FDNAMES",
fdNames: "",
socketName: "http",
expectError: true,
},
{
name: "index out of range",
fdNames: "web:web",
socketName: "web:2",
expectError: true,
},
{
name: "negative index",
fdNames: "web",
socketName: "web:-1",
expectError: true,
},
{
name: "invalid index format",
fdNames: "web",
socketName: "web:abc",
expectError: true,
},
{
name: "too many colons",
fdNames: "web",
socketName: "web:0:extra",
expectError: true,
},
}
for _, tc := range tests {
t.Run(tc.name, func(t *testing.T) {
// Set up environment
if tc.fdNames != "" {
os.Setenv("LISTEN_FDNAMES", tc.fdNames)
} else {
os.Unsetenv("LISTEN_FDNAMES")
}
// Test the function
fd, err := getFdByName(tc.socketName)
if tc.expectError {
if err == nil {
t.Errorf("Expected error but got none")
}
} else {
if err != nil {
t.Errorf("Expected no error but got: %v", err)
}
if fd != tc.expectedFd {
t.Errorf("Expected FD %d but got %d", tc.expectedFd, fd)
}
}
})
}
}
// TestParseNetworkAddressFdName tests parsing of fdname and fdgramname addresses.
func TestParseNetworkAddressFdName(t *testing.T) {
// Save and restore environment
originalFdNames := os.Getenv("LISTEN_FDNAMES")
defer func() {
if originalFdNames != "" {
os.Setenv("LISTEN_FDNAMES", originalFdNames)
} else {
os.Unsetenv("LISTEN_FDNAMES")
}
}()
// Set up test environment
os.Setenv("LISTEN_FDNAMES", "http:https:dns")
tests := []struct {
input string
expectAddr NetworkAddress
expectErr bool
}{
{
input: "fdname/http",
expectAddr: NetworkAddress{
Network: "fd",
Host: "3",
},
},
{
input: "fdname/https",
expectAddr: NetworkAddress{
Network: "fd",
Host: "4",
},
},
{
input: "fdname/dns",
expectAddr: NetworkAddress{
Network: "fd",
Host: "5",
},
},
{
input: "fdname/http:0",
expectAddr: NetworkAddress{
Network: "fd",
Host: "3",
},
},
{
input: "fdname/https:0",
expectAddr: NetworkAddress{
Network: "fd",
Host: "4",
},
},
{
input: "fdgramname/http",
expectAddr: NetworkAddress{
Network: "fdgram",
Host: "3",
},
},
{
input: "fdgramname/https",
expectAddr: NetworkAddress{
Network: "fdgram",
Host: "4",
},
},
{
input: "fdgramname/http:0",
expectAddr: NetworkAddress{
Network: "fdgram",
Host: "3",
},
},
{
input: "fdname/nonexistent",
expectErr: true,
},
{
input: "fdgramname/nonexistent",
expectErr: true,
},
{
input: "fdname/http:99",
expectErr: true,
},
{
input: "fdname/invalid:abc",
expectErr: true,
},
// Test that old fd/N syntax still works
{
input: "fd/7",
expectAddr: NetworkAddress{
Network: "fd",
Host: "7",
},
},
{
input: "fdgram/8",
expectAddr: NetworkAddress{
Network: "fdgram",
Host: "8",
},
},
}
for i, tc := range tests {
actualAddr, err := ParseNetworkAddress(tc.input)
if tc.expectErr && err == nil {
t.Errorf("Test %d (%s): Expected error but got none", i, tc.input)
}
if !tc.expectErr && err != nil {
t.Errorf("Test %d (%s): Expected no error but got: %v", i, tc.input, err)
}
if !tc.expectErr && !reflect.DeepEqual(tc.expectAddr, actualAddr) {
t.Errorf("Test %d (%s): Expected %+v but got %+v", i, tc.input, tc.expectAddr, actualAddr)
}
}
}

View file

@ -604,7 +604,7 @@ func (app *App) Start() error {
// if binding to port 0, the OS chooses a port for us;
// but the user won't know the port unless we print it
if !listenAddr.IsUnixNetwork() && !listenAddr.IsFdNetwork() && listenAddr.StartPort == 0 && listenAddr.EndPort == 0 {
if !listenAddr.IsUnixNetwork() && !listenAddr.IsFDNetwork() && listenAddr.StartPort == 0 && listenAddr.EndPort == 0 {
app.logger.Info("port 0 listener",
zap.String("input_address", lnAddr),
zap.String("actual_address", ln.Addr().String()))

View file

@ -104,7 +104,7 @@ func (pp *ListenerWrapper) Provision(ctx caddy.Context) error {
pp.policy = func(options goproxy.ConnPolicyOptions) (goproxy.Policy, error) {
// trust unix sockets
if network := options.Upstream.Network(); caddy.IsUnixNetwork(network) || caddy.IsFdNetwork(network) {
if network := options.Upstream.Network(); caddy.IsUnixNetwork(network) || caddy.IsFDNetwork(network) {
return goproxy.USE, nil
}
ret := pp.FallbackPolicy

View file

@ -137,7 +137,7 @@ func parseUpstreamDialAddress(upstreamAddr string) (parsedAddr, error) {
}
// we can assume a port if only a hostname is specified, but use of a
// placeholder without a port likely means a port will be filled in
if port == "" && !strings.Contains(host, "{") && !caddy.IsUnixNetwork(network) && !caddy.IsFdNetwork(network) {
if port == "" && !strings.Contains(host, "{") && !caddy.IsUnixNetwork(network) && !caddy.IsFDNetwork(network) {
port = "80"
}
}

View file

@ -331,7 +331,7 @@ func (h *Handler) doActiveHealthCheckForAllHosts() {
return
}
if hcp := uint(upstream.activeHealthCheckPort); hcp != 0 {
if addr.IsUnixNetwork() || addr.IsFdNetwork() {
if addr.IsUnixNetwork() || addr.IsFDNetwork() {
addr.Network = "tcp" // I guess we just assume TCP since we are using a port??
}
addr.StartPort, addr.EndPort = hcp, hcp
@ -345,7 +345,7 @@ func (h *Handler) doActiveHealthCheckForAllHosts() {
return
}
hostAddr := addr.JoinHostPort(0)
if addr.IsUnixNetwork() || addr.IsFdNetwork() {
if addr.IsUnixNetwork() || addr.IsFDNetwork() {
// this will be used as the Host portion of a http.Request URL, and
// paths to socket files would produce an error when creating URL,
// so use a fake Host value instead; unix sockets are usually local

View file

@ -382,7 +382,7 @@ func cmdRespond(fl caddycmd.Flags) (int, error) {
return caddy.ExitCodeFailedStartup, err
}
if !listenAddr.IsUnixNetwork() && !listenAddr.IsFdNetwork() {
if !listenAddr.IsUnixNetwork() && !listenAddr.IsFDNetwork() {
listenAddrs := make([]string, 0, listenAddr.PortRangeSize())
for offset := uint(0); offset < listenAddr.PortRangeSize(); offset++ {
listenAddrs = append(listenAddrs, listenAddr.JoinHostPort(offset))

129
networks.go Normal file
View file

@ -0,0 +1,129 @@
// Copyright 2015 Matthew Holt and The Caddy Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package caddy
import (
"context"
"net"
"strings"
"go.uber.org/zap"
)
const (
UNIX = "unix"
UNIX_H2C = "unix+h2c"
UNIXGRAM = "unixgram"
UNIXPACKET = "unixpacket"
TCP = "tcp"
TCP4 = "tcp4"
TCP6 = "tcp6"
UDP = "udp"
UDP4 = "udp4"
UDP6 = "udp6"
IP_ = "ip:"
IP4_ = "ip4:"
IP6_ = "ip6:"
FD = "fd"
FDGRAM = "fdgram"
)
// IsUnixNetwork returns true if the netw is a unix network.
func IsUnixNetwork(netw string) bool {
return netw == UNIX || netw == UNIX_H2C || netw == UNIXGRAM || netw == UNIXPACKET
}
// IsUnixNetwork returns true if the netw is a TCP network.
func IsTCPNetwork(netw string) bool {
return netw == TCP || netw == TCP4 || netw == TCP6
}
// IsUnixNetwork returns true if the netw is a UDP network.
func IsUDPNetwork(netw string) bool {
return netw == UDP || netw == UDP4 || netw == UDP6
}
// IsIPNetwork returns true if the netw is an ip network.
func IsIPNetwork(netw string) bool {
return strings.HasPrefix(netw, IP_) || strings.HasPrefix(netw, IP4_) || strings.HasPrefix(netw, IP6_)
}
// IsFDNetwork returns true if the netw is a fd network.
func IsFDNetwork(netw string) bool {
return netw == FD || netw == FDGRAM
}
func IsReservedNetwork(network string) bool {
return IsUnixNetwork(network) ||
IsTCPNetwork(network) ||
IsUDPNetwork(network) ||
IsIPNetwork(network) ||
IsFDNetwork(network)
}
func IsIPv4Network(netw string) bool {
return netw == TCP || netw == TCP4 || netw == UDP || netw == UDP4 || strings.HasPrefix(netw, IP_) || strings.HasPrefix(netw, IP4_)
}
func IsIPv6Network(netw string) bool {
return netw == TCP || netw == TCP6 || netw == UDP || netw == UDP6 || strings.HasPrefix(netw, IP_) || strings.HasPrefix(netw, IP6_)
}
func IsStreamNetwork(netw string) bool {
return netw == UNIX || netw == UNIX_H2C || netw == UNIXPACKET || IsTCPNetwork(netw) || netw == FD
}
func IsPacketNetwork(netw string) bool {
return netw == UNIXGRAM || IsUDPNetwork(netw) || IsIPNetwork(netw) || netw == FDGRAM
}
// ListenerFunc is a function that can return a listener given a network and address.
// The listeners must be capable of overlapping: with Caddy, new configs are loaded
// before old ones are unloaded, so listeners may overlap briefly if the configs
// both need the same listener. EXPERIMENTAL and subject to change.
type ListenerFunc func(ctx context.Context, network, host, portRange string, portOffset uint, cfg net.ListenConfig) (any, error)
var networkPlugins = map[string]ListenerFunc{}
// RegisterNetwork registers a network plugin with Caddy so that if a listener is
// created for that network plugin, getListener will be invoked to get the listener.
// This should be called during init() and will panic if the network type is standard
// or reserved, or if it is already registered. EXPERIMENTAL and subject to change.
func RegisterNetwork(network string, getListener ListenerFunc) {
network = strings.TrimSpace(strings.ToLower(network))
if IsReservedNetwork(network) {
panic("network type " + network + " is reserved")
}
if _, ok := networkPlugins[strings.ToLower(network)]; ok {
panic("network type " + network + " is already registered")
}
networkPlugins[network] = getListener
}
// getListenerFromPlugin returns a listener on the given network and address
// if a plugin has registered the network name. It may return (nil, nil) if
// no plugin can provide a listener.
func getListenerFromPlugin(ctx context.Context, network, host, port string, portOffset uint, config net.ListenConfig) (any, error) {
// get listener from plugin if network is registered
if getListener, ok := networkPlugins[network]; ok {
Log().Debug("getting listener from plugin", zap.String("network", network))
return getListener(ctx, network, host, port, portOffset, config)
}
return nil, nil
}

View file

@ -36,16 +36,12 @@ func NewReplacer() *Replacer {
static: make(map[string]any),
mapMutex: &sync.RWMutex{},
}
rep.providers = []replacementProvider{
globalDefaultReplacementProvider{},
fileReplacementProvider{},
ReplacerFunc(rep.fromStatic),
}
rep.providers = append(globalReplacementProviders, ReplacerFunc(rep.fromStatic))
return rep
}
// NewEmptyReplacer returns a new Replacer,
// without the global default replacements.
// without the global replacements.
func NewEmptyReplacer() *Replacer {
rep := &Replacer{
static: make(map[string]any),
@ -360,12 +356,11 @@ func (f fileReplacementProvider) replace(key string) (any, bool) {
return string(body), true
}
// globalDefaultReplacementProvider handles replacements
// that can be used in any context, such as system variables,
// time, or environment variables.
type globalDefaultReplacementProvider struct{}
// defaultReplacementProvider handles replacements
// such as system variables, time, or environment variables.
type defaultReplacementProvider struct{}
func (f globalDefaultReplacementProvider) replace(key string) (any, bool) {
func (f defaultReplacementProvider) replace(key string) (any, bool) {
// check environment variable
const envPrefix = "env."
if strings.HasPrefix(key, envPrefix) {

8
replacer_nosystemd.go Normal file
View file

@ -0,0 +1,8 @@
//go:build !linux || nosystemd
package caddy
var globalReplacementProviders = []replacementProvider{
defaultReplacementProvider{},
fileReplacementProvider{},
}

123
replacer_systemd.go Normal file
View file

@ -0,0 +1,123 @@
//go:build linux && !nosystemd
package caddy
import (
"errors"
"fmt"
"os"
"strconv"
"strings"
"go.uber.org/zap"
)
func sdListenFds() (int, error) {
lnPid, ok := os.LookupEnv("LISTEN_PID")
if !ok {
return 0, errors.New("LISTEN_PID is unset")
}
pid, err := strconv.Atoi(lnPid)
if err != nil {
return 0, err
}
if pid != os.Getpid() {
return 0, fmt.Errorf("LISTEN_PID does not match pid: %d != %d", pid, os.Getpid())
}
lnFds, ok := os.LookupEnv("LISTEN_FDS")
if !ok {
return 0, errors.New("LISTEN_FDS is unset")
}
fds, err := strconv.Atoi(lnFds)
if err != nil {
return 0, err
}
return fds, nil
}
func sdListenFdsWithNames() (map[string][]uint, error) {
const lnFdsStart = 3
fds, err := sdListenFds()
if err != nil {
return nil, err
}
lnFdnames, ok := os.LookupEnv("LISTEN_FDNAMES")
if !ok {
return nil, errors.New("LISTEN_FDNAMES is unset")
}
fdNames := strings.Split(lnFdnames, ":")
if fds != len(fdNames) {
return nil, fmt.Errorf("LISTEN_FDS does not match LISTEN_FDNAMES length: %d != %d", fds, len(fdNames))
}
nameToFiles := make(map[string][]uint, len(fdNames))
for index, name := range fdNames {
nameToFiles[name] = append(nameToFiles[name], lnFdsStart+uint(index))
}
return nameToFiles, nil
}
func getSdListenFd(nameToFiles map[string][]uint, nameOffset string) (uint, error) {
index := uint(0)
name, offset, found := strings.Cut(nameOffset, ":")
if found {
off, err := strconv.ParseUint(offset, 0, strconv.IntSize)
if err != nil {
return 0, err
}
index += uint(off)
}
files, ok := nameToFiles[name]
if !ok {
return 0, fmt.Errorf("invalid listen fd name: %s", name)
}
if uint(len(files)) <= index {
return 0, fmt.Errorf("invalid listen fd index: %d", index)
}
return files[index], nil
}
var initNameToFiles, initNameToFilesErr = sdListenFdsWithNames()
// systemdReplacementProvider handles {systemd.*} replacements
type systemdReplacementProvider struct{}
func (f systemdReplacementProvider) replace(key string) (any, bool) {
// check environment variable
const systemdListenPrefix = "systemd.listen."
if strings.HasPrefix(key, systemdListenPrefix) {
if initNameToFilesErr != nil {
Log().Error("unable to read LISTEN_FDNAMES", zap.Error(initNameToFilesErr))
return nil, false
}
fd, err := getSdListenFd(initNameToFiles, key[len(systemdListenPrefix):])
if err != nil {
Log().Error("unable to process {"+key+"}", zap.Error(err))
return nil, false
}
return fd, true
}
// TODO const systemdCredsPrefix = "systemd.creds."
return nil, false
}
var globalReplacementProviders = []replacementProvider{
defaultReplacementProvider{},
fileReplacementProvider{},
systemdReplacementProvider{},
}

View file

@ -374,10 +374,6 @@ func TestReplacerMap(t *testing.T) {
func TestReplacerNew(t *testing.T) {
repl := NewReplacer()
if len(repl.providers) != 3 {
t.Errorf("Expected providers length '%v' got length '%v'", 3, len(repl.providers))
}
// test if default global replacements are added as the first provider
hostname, _ := os.Hostname()
wd, _ := os.Getwd()

376
replacer_test_systemd.go Normal file
View file

@ -0,0 +1,376 @@
//go:build linux && !nosystemd
package caddy
import (
"os"
"reflect"
"strconv"
"testing"
)
// TestGetSdListenFd tests the getSdListenFd function for systemd socket activation.
func TestGetSdListenFd(t *testing.T) {
// Save original environment
originalFdNames := os.Getenv("LISTEN_FDNAMES")
originalFds := os.Getenv("LISTEN_FDS")
originalPid := os.Getenv("LISTEN_PID")
// Restore environment after test
defer func() {
if originalFdNames != "" {
os.Setenv("LISTEN_FDNAMES", originalFdNames)
} else {
os.Unsetenv("LISTEN_FDNAMES")
}
if originalFds != "" {
os.Setenv("LISTEN_FDS", originalFds)
} else {
os.Unsetenv("LISTEN_FDS")
}
if originalPid != "" {
os.Setenv("LISTEN_PID", originalPid)
} else {
os.Unsetenv("LISTEN_PID")
}
}()
tests := []struct {
name string
fdNames string
fds string
socketName string
expectedFd uint
expectError bool
}{
{
name: "simple http socket",
fdNames: "http",
fds: "1",
socketName: "http",
expectedFd: 3,
},
{
name: "multiple different sockets - first",
fdNames: "http:https:dns",
fds: "3",
socketName: "http",
expectedFd: 3,
},
{
name: "multiple different sockets - second",
fdNames: "http:https:dns",
fds: "3",
socketName: "https",
expectedFd: 4,
},
{
name: "multiple different sockets - third",
fdNames: "http:https:dns",
fds: "3",
socketName: "dns",
expectedFd: 5,
},
{
name: "duplicate names - first occurrence (no index)",
fdNames: "web:web:api",
fds: "3",
socketName: "web",
expectedFd: 3,
},
{
name: "duplicate names - first occurrence (explicit index 0)",
fdNames: "web:web:api",
fds: "3",
socketName: "web:0",
expectedFd: 3,
},
{
name: "duplicate names - second occurrence (index 1)",
fdNames: "web:web:api",
fds: "3",
socketName: "web:1",
expectedFd: 4,
},
{
name: "complex duplicates - first api",
fdNames: "web:api:web:api:dns",
fds: "5",
socketName: "api:0",
expectedFd: 4,
},
{
name: "complex duplicates - second api",
fdNames: "web:api:web:api:dns",
fds: "5",
socketName: "api:1",
expectedFd: 6,
},
{
name: "complex duplicates - first web",
fdNames: "web:api:web:api:dns",
fds: "5",
socketName: "web:0",
expectedFd: 3,
},
{
name: "complex duplicates - second web",
fdNames: "web:api:web:api:dns",
fds: "5",
socketName: "web:1",
expectedFd: 5,
},
{
name: "socket not found",
fdNames: "http:https",
fds: "2",
socketName: "missing",
expectError: true,
},
{
name: "empty socket name",
fdNames: "http",
fds: "1",
socketName: "",
expectError: true,
},
{
name: "missing LISTEN_FDNAMES",
fdNames: "",
fds: "",
socketName: "http",
expectError: true,
},
{
name: "index out of range",
fdNames: "web:web",
fds: "2",
socketName: "web:2",
expectError: true,
},
{
name: "negative index",
fdNames: "web",
fds: "1",
socketName: "web:-1",
expectError: true,
},
{
name: "invalid index format",
fdNames: "web",
fds: "1",
socketName: "web:abc",
expectError: true,
},
{
name: "too many colons",
fdNames: "web",
fds: "1",
socketName: "web:0:extra",
expectError: true,
},
}
for _, tc := range tests {
t.Run(tc.name, func(t *testing.T) {
// Set up environment
if tc.fdNames != "" {
os.Setenv("LISTEN_FDNAMES", tc.fdNames)
} else {
os.Unsetenv("LISTEN_FDNAMES")
}
if tc.fds != "" {
os.Setenv("LISTEN_FDS", tc.fds)
} else {
os.Unsetenv("LISTEN_FDS")
}
os.Setenv("LISTEN_PID", strconv.Itoa(os.Getpid()))
// Test the function
var (
listenFdsWithNames map[string][]uint
err error
fd uint
)
listenFdsWithNames, err = sdListenFdsWithNames()
if err == nil {
fd, err = getSdListenFd(listenFdsWithNames, tc.socketName)
}
if tc.expectError {
if err == nil {
t.Errorf("Expected error but got none")
}
} else {
if err != nil {
t.Errorf("Expected no error but got: %v", err)
}
if fd != tc.expectedFd {
t.Errorf("Expected FD %d but got %d", tc.expectedFd, fd)
}
}
})
}
}
// TestParseSystemdListenPlaceholder tests parsing of {systemd.listen.name} placeholders.
func TestParseSystemdListenPlaceholder(t *testing.T) {
// Save and restore environment
originalFdNames := os.Getenv("LISTEN_FDNAMES")
originalFds := os.Getenv("LISTEN_FDS")
originalPid := os.Getenv("LISTEN_PID")
defer func() {
if originalFdNames != "" {
os.Setenv("LISTEN_FDNAMES", originalFdNames)
} else {
os.Unsetenv("LISTEN_FDNAMES")
}
if originalFds != "" {
os.Setenv("LISTEN_FDS", originalFds)
} else {
os.Unsetenv("LISTEN_FDS")
}
if originalPid != "" {
os.Setenv("LISTEN_PID", originalPid)
} else {
os.Unsetenv("LISTEN_PID")
}
}()
// Set up test environment
os.Setenv("LISTEN_FDNAMES", "http:https:dns")
os.Setenv("LISTEN_FDS", "3")
os.Setenv("LISTEN_PID", strconv.Itoa(os.Getpid()))
tests := []struct {
input string
expectedAddr NetworkAddress
expectedFd uint
expectErr bool
}{
{
input: "fd/{systemd.listen.http}",
expectedAddr: NetworkAddress{
Network: "fd",
Host: "{systemd.listen.http}",
},
expectedFd: 3,
},
{
input: "fd/{systemd.listen.https}",
expectedAddr: NetworkAddress{
Network: "fd",
Host: "{systemd.listen.https}",
},
expectedFd: 4,
},
{
input: "fd/{systemd.listen.dns}",
expectedAddr: NetworkAddress{
Network: "fd",
Host: "{systemd.listen.dns}",
},
expectedFd: 5,
},
{
input: "fd/{systemd.listen.http:0}",
expectedAddr: NetworkAddress{
Network: "fd",
Host: "{systemd.listen.http:0}",
},
expectedFd: 3,
},
{
input: "fd/{systemd.listen.https:0}",
expectedAddr: NetworkAddress{
Network: "fd",
Host: "{systemd.listen.https:0}",
},
expectedFd: 4,
},
{
input: "fdgram/{systemd.listen.http}",
expectedAddr: NetworkAddress{
Network: "fdgram",
Host: "{systemd.listen.http}",
},
expectedFd: 3,
},
{
input: "fdgram/{systemd.listen.https}",
expectedAddr: NetworkAddress{
Network: "fdgram",
Host: "{systemd.listen.https}",
},
expectedFd: 4,
},
{
input: "fdgram/{systemd.listen.http:0}",
expectedAddr: NetworkAddress{
Network: "fdgram",
Host: "http:0",
},
expectedFd: 3,
},
{
input: "fd/{systemd.listen.nonexistent}",
expectErr: true,
},
{
input: "fdgram/{systemd.listen.nonexistent}",
expectErr: true,
},
{
input: "fd/{systemd.listen.http:99}",
expectErr: true,
},
{
input: "fd/{systemd.listen.invalid:abc}",
expectErr: true,
},
// Test that old fd/N syntax still works
{
input: "fd/7",
expectedAddr: NetworkAddress{
Network: "fd",
Host: "7",
},
expectedFd: 7,
},
{
input: "fdgram/8",
expectedAddr: NetworkAddress{
Network: "fdgram",
Host: "8",
},
expectedFd: 8,
},
}
for i, tc := range tests {
actualAddr, err := ParseNetworkAddress(tc.input)
if err == nil {
var fd uint
fdWide, err := strconv.ParseUint(actualAddr.Host, 0, strconv.IntSize)
if err == nil {
fd = uint(fdWide)
}
if tc.expectErr && err == nil {
t.Errorf("Test %d (%s): Expected error but got none", i, tc.input)
}
if !tc.expectErr && err != nil {
t.Errorf("Test %d (%s): Expected no error but got: %v", i, tc.input, err)
}
if !tc.expectErr && !reflect.DeepEqual(tc.expectedAddr, actualAddr) {
t.Errorf("Test %d (%s): Expected %+v but got %+v", i, tc.input, tc.expectedAddr, actualAddr)
}
if !tc.expectErr && fd != tc.expectedFd {
t.Errorf("Expected FD %d but got %d", tc.expectedFd, fd)
}
}
}
}