61 lines
1.6 KiB
Go
61 lines
1.6 KiB
Go
package tools
|
||
|
||
import (
|
||
"net"
|
||
"net/http"
|
||
"strings"
|
||
)
|
||
|
||
// ClientIP 首先读取 X-Forwarded-For header 中用 , 分隔的第一个ip地址，
|
||
// 如果这个地址不存在，就会从 X-Real-Ip header 中获取，如果还是不存在，
|
||
// 说明流量并非是由反向代理转发而来，而是客户端直接请求服务，这时通过 http.Request.RemoteAddr 字段
|
||
// 截取除去端口号的 ip 地址。
|
||
func ClientPublicIp(r *http.Request) string {
|
||
var ip string
|
||
for _, ip = range strings.Split(r.Header.Get("X-Forwarded-For"), ",") {
|
||
ip = strings.TrimSpace(ip)
|
||
if ip != "" && !HasLocalIPaddr(ip) {
|
||
return ip
|
||
}
|
||
}
|
||
|
||
ip = strings.TrimSpace(r.Header.Get("X-Real-Ip"))
|
||
if ip != "" && !HasLocalIPaddr(ip) {
|
||
return ip
|
||
}
|
||
|
||
if ip, _, err := net.SplitHostPort(strings.TrimSpace(r.RemoteAddr)); err == nil {
|
||
if !HasLocalIPaddr(ip) {
|
||
return ip
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 判断是地址是否来自代理
|
||
func RemoteIp(r *http.Request) string {
|
||
if ip, _, err := net.SplitHostPort(strings.TrimSpace(r.RemoteAddr)); err == nil {
|
||
return ip
|
||
}
|
||
return ""
|
||
}
|
||
|
||
// 检测IP地址字符串是否内网地址
|
||
func HasLocalIPaddr(ip string) bool {
|
||
return HasLocalIP(net.ParseIP(ip))
|
||
}
|
||
|
||
// 检测IP地址是否内网地址
|
||
// tcp/ip协议中，专门保留了三个IP地址区域作为私有地址，其地址范围如下：
|
||
// 10.0.0.0/8：10.0.0.0～10.255.255.255
|
||
// 172.16.0.0/12：172.16.0.0～172.31.255.255
|
||
// 192.168.0.0/16：192.168.0.0～192.168.255.255
|
||
func HasLocalIP(ip net.IP) bool {
|
||
// for _, network := range localNetworks {
|
||
// if network.Contains(ip) {
|
||
// return true
|
||
// }
|
||
// }
|
||
return ip.IsLoopback()
|
||
}
|