41 lines
737 B
Go
41 lines
737 B
Go
package utils
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
func GetIP(r *http.Request, fromHeader bool) (string, error) {
|
|
if fromHeader {
|
|
//Get IP from the X-REAL-IP header
|
|
ip := r.Header.Get("X-REAL-IP")
|
|
netIP := net.ParseIP(ip)
|
|
if netIP != nil {
|
|
return ip, nil
|
|
}
|
|
|
|
//Get IP from X-FORWARDED-FOR header
|
|
ips := r.Header.Get("X-FORWARDED-FOR")
|
|
splitIps := strings.Split(ips, ",")
|
|
for _, ip := range splitIps {
|
|
netIP := net.ParseIP(ip)
|
|
if netIP != nil {
|
|
return ip, nil
|
|
}
|
|
}
|
|
}
|
|
|
|
//Get IP from RemoteAddr
|
|
ip, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
netIP := net.ParseIP(ip)
|
|
if netIP != nil {
|
|
return ip, nil
|
|
}
|
|
return "", fmt.Errorf("no valid ip found")
|
|
}
|