Cloudflare header CF-Connecting-IP

mccutchen · Oct 6, 2023 · 81de5bf · 81de5bf
1 parent c2ae4ed
commit 81de5bf
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/httpbin/helpers.go b/httpbin/helpers.go
@@ -51,8 +51,8 @@ func getClientIP(r *http.Request) string {
 	}
 
 	// Special case Cloudflare for the value directly.
-	if cfClientIP := r.Header.Get("CF-Connecting-IP"); cfClientIP != "" {
-		return cfClientIP
+	if clientIP := r.Header.Get("CF-Connecting-IP"); clientIP != "" {
+		return clientIP
 	}
 
 	// Try to pull a reasonable value from the X-Forwarded-For header, if

diff --git a/httpbin/helpers_test.go b/httpbin/helpers_test.go
@@ -235,6 +235,16 @@ func TestGetClientIP(t *testing.T) {
 			},
 			want: "9.9.9.9",
 		},
+		"custom cloudflare header take precedence": {
+			given: &http.Request{
+				Header: makeHeaders(map[string]string{
+					"CF-Connecting-IP": "9.9.9.9",
+					"X-Forwarded-For":  "1.1.1.1,2.2.2.2,3.3.3.3",
+				}),
+				RemoteAddr: "0.0.0.0",
+			},
+			want: "9.9.9.9",
+		},
 		"x-forwarded-for is parsed": {
 			given: &http.Request{
 				Header: makeHeaders(map[string]string{