mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
33 lines
693 B
Go
33 lines
693 B
Go
package geoip
|
|
|
|
import (
|
|
"fmt"
|
|
"net"
|
|
|
|
"github.com/oschwald/maxminddb-golang"
|
|
)
|
|
|
|
func getReader(ip net.IP) *maxminddb.Reader {
|
|
isV6 := ip.To4() == nil
|
|
return worker.GetReader(isV6, true)
|
|
}
|
|
|
|
// GetLocation returns Location data of an IP address.
|
|
func GetLocation(ip net.IP) (*Location, error) {
|
|
db := getReader(ip)
|
|
if db == nil {
|
|
return nil, fmt.Errorf("geoip database not available")
|
|
}
|
|
record := &Location{}
|
|
if err := db.Lookup(ip, record); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
record.AddCountryInfo()
|
|
return record, nil
|
|
}
|
|
|
|
// IsInitialized returns whether the geoip database has been initialized.
|
|
func IsInitialized(v6, wait bool) bool {
|
|
return worker.GetReader(v6, wait) != nil
|
|
}
|