55 lines
1 KiB
Go
55 lines
1 KiB
Go
package filters
|
|
|
|
import (
|
|
"bufio"
|
|
"net"
|
|
"os"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// IsAkamai checks for the presence of the given IP in the Akamain CIDR.
|
|
func IsAkamai(ip net.IP) bool {
|
|
var file *os.File
|
|
var err error
|
|
if ip.To4() != nil {
|
|
file, err = os.Open("filters/data/akamai.cidr")
|
|
if err != nil {
|
|
logrus.Fatal("filters#IsAkamai", err)
|
|
}
|
|
} else if ip.To16() != nil {
|
|
file, err = os.Open("filters/data/akamaiv6.cidr")
|
|
if err != nil {
|
|
logrus.Fatal("filters#IsAkamai", err)
|
|
}
|
|
} else {
|
|
logrus.Error("filters#IsAkamai#invalid ip format")
|
|
return false
|
|
}
|
|
defer file.Close()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
for scanner.Scan() {
|
|
_, ipNet, err := net.ParseCIDR(scanner.Text())
|
|
if err != nil {
|
|
continue
|
|
}
|
|
if ipNet.Contains(ip) {
|
|
return true
|
|
}
|
|
|
|
}
|
|
return false
|
|
}
|
|
|
|
// Check the version of the IP address (IPv4 or IPv6).
|
|
func checkIPversion(ip string) (string, bool) {
|
|
if net.ParseIP(ip).To4() != nil {
|
|
return "IPv4", true
|
|
} else if net.ParseIP(ip).To16() != nil {
|
|
return "IPv6", true
|
|
} else {
|
|
return "", false
|
|
}
|
|
}
|