taoshihan
1 month ago
parent
b760cb4fcf
commit
3fd7457467
|
After Width: | Height: | Size: 58 MiB |
Binary file not shown.
@ -0,0 +1,23 @@
|
|||||||
|
package tools
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/oschwald/geoip2-golang/v2"
|
||||||
|
"net/netip"
|
||||||
|
)
|
||||||
|
|
||||||
|
func GetCity(path, ipAddress string) (string, string) {
|
||||||
|
db, err := geoip2.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
defer db.Close()
|
||||||
|
ip, err := netip.ParseAddr(ipAddress)
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
record, err := db.City(ip)
|
||||||
|
if err != nil {
|
||||||
|
return "", ""
|
||||||
|
}
|
||||||
|
return record.Country.Names.English, record.City.Names.English
|
||||||
|
}
|
||||||
@ -0,0 +1,7 @@
|
|||||||
|
package tools
|
||||||
|
|
||||||
|
import "testing"
|
||||||
|
|
||||||
|
func TestGetCity(t *testing.T) {
|
||||||
|
GetCity("../config/GeoLite2-City.mmdb", "")
|
||||||
|
}
|
||||||
@ -1,77 +0,0 @@
|
|||||||
package tools
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"github.com/ipipdotnet/ipdb-go"
|
|
||||||
"net"
|
|
||||||
)
|
|
||||||
|
|
||||||
func ParseIp(myip string) *ipdb.CityInfo {
|
|
||||||
db, err := ipdb.NewCity("./config/city.free.ipdb")
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
db.Reload("./config/city.free.ipdb")
|
|
||||||
c, err := db.FindInfo(myip, "CN")
|
|
||||||
if err != nil {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
return c
|
|
||||||
}
|
|
||||||
func GetServerIP() (net.IP, error) {
|
|
||||||
ifaces, err := net.Interfaces()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
for _, iface := range ifaces {
|
|
||||||
if iface.Flags&net.FlagUp == 0 {
|
|
||||||
continue // interface down
|
|
||||||
}
|
|
||||||
if iface.Flags&net.FlagLoopback != 0 {
|
|
||||||
continue // loopback interface
|
|
||||||
}
|
|
||||||
addrs, err := iface.Addrs()
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
for _, addr := range addrs {
|
|
||||||
ip := getIpFromAddr(addr)
|
|
||||||
if ip == nil {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
return ip, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil, errors.New("connected to the network?")
|
|
||||||
}
|
|
||||||
func getIpFromAddr(addr net.Addr) net.IP {
|
|
||||||
var ip net.IP
|
|
||||||
switch v := addr.(type) {
|
|
||||||
case *net.IPNet:
|
|
||||||
ip = v.IP
|
|
||||||
case *net.IPAddr:
|
|
||||||
ip = v.IP
|
|
||||||
}
|
|
||||||
if ip == nil || ip.IsLoopback() {
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
ip = ip.To4()
|
|
||||||
if ip == nil {
|
|
||||||
return nil // not an ipv4 address
|
|
||||||
}
|
|
||||||
|
|
||||||
return ip
|
|
||||||
}
|
|
||||||
|
|
||||||
//获取出站IP地址
|
|
||||||
func GetOutboundIP() (net.IP, error) {
|
|
||||||
conn, err := net.Dial("udp", "8.8.8.8:80")
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
defer conn.Close()
|
|
||||||
|
|
||||||
localAddr := conn.LocalAddr().(*net.UDPAddr)
|
|
||||||
|
|
||||||
return localAddr.IP, nil
|
|
||||||
}
|
|
||||||
@ -1,11 +0,0 @@
|
|||||||
package tools
|
|
||||||
|
|
||||||
import (
|
|
||||||
"log"
|
|
||||||
"testing"
|
|
||||||
)
|
|
||||||
|
|
||||||
func TestGetOutboundIP(t *testing.T) {
|
|
||||||
ip, err := GetOutboundIP()
|
|
||||||
log.Println(ip, err)
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue