fix potential panic due to nil interface conversion (#6)

samber · Aug 28, 2023 · cb9170e · cb9170e
1 parent afd2ac6
commit cb9170e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/middleware.go b/middleware.go
@@ -80,8 +80,8 @@ func NewWithConfig(logger *slog.Logger, config Config) echo.MiddlewareFunc {
 				slog.String("user-agent", userAgent),
 			}
 
-			xForwardedFor := c.Get(echo.HeaderXForwardedFor).(string)
-			if len(xForwardedFor) > 0 {
+			xForwardedFor, ok := c.Get(echo.HeaderXForwardedFor).(string)
+			if ok && len(xForwardedFor) > 0 {
 				ips := lo.Map(strings.Split(xForwardedFor, ","), func(ip string, _ int) string {
 					return strings.TrimSpace(ip)
 				})