Skip to content

Commit ff4f23d

Browse files
committed
fix(HttpMessage): remove extra space in key string
and rollback forwarded address header check (already happens in fluke)
1 parent a42c0de commit ff4f23d

File tree

1 file changed

+7
-22
lines changed

1 file changed

+7
-22
lines changed

HttpMessage.go

Lines changed: 7 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -164,34 +164,19 @@ func appendRequestParams(message *[][]string, req *http.Request) {
164164
* Adds request headers to message.
165165
*/
166166
func appendRequestHeaders(message *[][]string, req *http.Request) {
167-
addressInHeaders := false
168167
reqHeaders := req.Header
169168
for headerName, headerValues := range reqHeaders {
170-
lowerHeaderName := strings.ToLower(headerName)
171-
name := "request_header:" + lowerHeaderName
169+
name := "request_header:" + strings.ToLower(headerName)
172170
for _, value := range headerValues {
173171
*message = append(*message, []string{name, value})
174172
}
173+
}
175174

176-
if !addressInHeaders {
177-
if lowerHeaderName == "x-forwarded-for" ||
178-
lowerHeaderName == "forwarded-for" ||
179-
lowerHeaderName == "forwarded" ||
180-
lowerHeaderName == "cf-connecting-ip" ||
181-
lowerHeaderName == "fastly-client-ip" ||
182-
lowerHeaderName == "true-client-ip" {
183-
addressInHeaders = true
184-
} else if lowerHeaderName == "x-forwarded" ||
185-
lowerHeaderName == "x-client-ip" ||
186-
lowerHeaderName == "x-real-ip" ||
187-
lowerHeaderName == "x-cluster-client-ip" {
188-
*message = append(*message, []string{"request_header: x-forwarded-for", reqHeaders.Get(headerName)})
189-
addressInHeaders = true
190-
}
175+
if req.RemoteAddr != "" {
176+
addr := req.RemoteAddr
177+
if idx := strings.Index(addr, ":"); idx != -1 {
178+
addr = addr[:idx]
191179
}
180+
*message = append(*message, []string{"request_header:x-forwarded-for", addr})
192181
}
193-
if !addressInHeaders && req.RemoteAddr != "" {
194-
*message = append(*message, []string{"request_header: x-forwarded-for", req.RemoteAddr})
195-
}
196-
197182
}

0 commit comments

Comments
 (0)