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()
|
|||
|
|
}
|