mirror of
				https://github.com/caddyserver/caddy.git
				synced 2025-11-04 10:12:29 +08:00 
			
		
		
		
	reverseproxy: Keep path to unix socket as dial address (#4232)
This commit is contained in:
		@ -436,3 +436,57 @@ func TestReverseProxyHealthCheckUnixSocket(t *testing.T) {
 | 
			
		||||
 | 
			
		||||
	tester.AssertGetResponse("http://localhost:9080/", 200, "Hello, World!")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
func TestReverseProxyHealthCheckUnixSocketWithoutPort(t *testing.T) {
 | 
			
		||||
	if runtime.GOOS == "windows" {
 | 
			
		||||
		t.SkipNow()
 | 
			
		||||
	}
 | 
			
		||||
	tester := caddytest.NewTester(t)
 | 
			
		||||
	f, err := ioutil.TempFile("", "*.sock")
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Errorf("failed to create TempFile: %s", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	// a hack to get a file name within a valid path to use as socket
 | 
			
		||||
	socketName := f.Name()
 | 
			
		||||
	os.Remove(f.Name())
 | 
			
		||||
 | 
			
		||||
	server := http.Server{
 | 
			
		||||
		Handler: http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
 | 
			
		||||
			if strings.HasPrefix(req.URL.Path, "/health") {
 | 
			
		||||
				w.Write([]byte("ok"))
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			w.Write([]byte("Hello, World!"))
 | 
			
		||||
		}),
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	unixListener, err := net.Listen("unix", socketName)
 | 
			
		||||
	if err != nil {
 | 
			
		||||
		t.Errorf("failed to listen on the socket: %s", err)
 | 
			
		||||
		return
 | 
			
		||||
	}
 | 
			
		||||
	go server.Serve(unixListener)
 | 
			
		||||
	t.Cleanup(func() {
 | 
			
		||||
		server.Close()
 | 
			
		||||
	})
 | 
			
		||||
	runtime.Gosched() // Allow other goroutines to run
 | 
			
		||||
 | 
			
		||||
	tester.InitServer(fmt.Sprintf(`
 | 
			
		||||
	{
 | 
			
		||||
		http_port     9080
 | 
			
		||||
		https_port    9443
 | 
			
		||||
	}
 | 
			
		||||
	http://localhost:9080 {
 | 
			
		||||
		reverse_proxy {
 | 
			
		||||
			to unix/%s
 | 
			
		||||
	
 | 
			
		||||
			health_path /health
 | 
			
		||||
			health_interval 2s
 | 
			
		||||
			health_timeout 5s
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	`, socketName), "caddyfile")
 | 
			
		||||
 | 
			
		||||
	tester.AssertGetResponse("http://localhost:9080/", 200, "Hello, World!")
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -189,13 +189,14 @@ func (h *Handler) doActiveHealthCheckForAllHosts() {
 | 
			
		||||
				return
 | 
			
		||||
			}
 | 
			
		||||
			hostAddr := addr.JoinHostPort(0)
 | 
			
		||||
			dialAddr := hostAddr
 | 
			
		||||
			if addr.IsUnixNetwork() {
 | 
			
		||||
				// 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
 | 
			
		||||
				hostAddr = "localhost"
 | 
			
		||||
			}
 | 
			
		||||
			err = h.doActiveHealthCheck(DialInfo{Network: addr.Network, Address: hostAddr}, hostAddr, upstream.Host)
 | 
			
		||||
			err = h.doActiveHealthCheck(DialInfo{Network: addr.Network, Address: dialAddr}, hostAddr, upstream.Host)
 | 
			
		||||
			if err != nil {
 | 
			
		||||
				h.HealthChecks.Active.logger.Error("active health check failed",
 | 
			
		||||
					zap.String("address", hostAddr),
 | 
			
		||||
 | 
			
		||||
		Reference in New Issue
	
	Block a user