Function TestHttpOverProxyProtocol

server/server_test.go:925–992

(t *testing.T)

}

func TestHttpOverProxyProtocol(t *testing.T) {
	prometheus.DefaultRegisterer = prometheus.NewRegistry()

	var cfg Config
	cfg.RegisterFlags(flag.NewFlagSet("", flag.ExitOnError))
	cfg.ProxyProtocolEnabled = true
	setAutoAssignedPorts(DefaultNetwork, &cfg)

	server, err := New(cfg)
	require.NoError(t, err)
	defer server.Shutdown()

	server.HTTP.HandleFunc("/test-proxy-proto", func(w http.ResponseWriter, r *http.Request) {
		ip, _, err := net.SplitHostPort(r.RemoteAddr)
		require.NoError(t, err)
		_, err = w.Write([]byte(ip))
		require.NoError(t, err)
	})

	go func() {
		require.NoError(t, server.Run())
	}()

	t.Run("good PROXY header", func(t *testing.T) {
		fakeSourceIP := "1.2.3.4"
		proxyHeader := fmt.Sprintf("PROXY TCP4 %s 192.168.0.1 51234 80\r\n", fakeSourceIP)
		client := &http.Client{
			Transport: &http.Transport{
				DialContext: proxyDialer(proxyHeader),
			},
		}

		res, err := client.Get(httpTarget(server, "/test-proxy-proto"))
		require.NoError(t, err)
		defer res.Body.Close()
		require.Equal(t, res.StatusCode, http.StatusOK)

		body, err := io.ReadAll(res.Body)
		require.NoError(t, err)
		require.Equal(t, []byte(fakeSourceIP), body)
	})

	t.Run("malformed PROXY header", func(t *testing.T) {
		proxyHeader := "badPROXY TCP4 1.2.3.4 192.168.0.1 51234 80\r\n"
		client := &http.Client{
			Transport: &http.Transport{
				DialContext: proxyDialer(proxyHeader),
			},
		}

		res, err := client.Get(httpTarget(server, "/test-proxy-proto"))
		require.NoError(t, err)
		defer res.Body.Close()
		require.Equal(t, http.StatusBadRequest, res.StatusCode)
	})

	t.Run("no PROXY header", func(t *testing.T) {
		client := &http.Client{}