64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package filters
|
|
|
|
import (
|
|
"bufio"
|
|
"io/ioutil"
|
|
"net"
|
|
"os"
|
|
"path/filepath"
|
|
"runtime"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
_, b, _, _ = runtime.Caller(0)
|
|
basepath = filepath.Dir(b)
|
|
)
|
|
|
|
// RunIPFilters runs the battery of filters for an IP.
|
|
func RunIPFilters(ip net.IP) bool {
|
|
if ip.To4() != nil {
|
|
path := basepath + "/data/ipv4/"
|
|
sliceIPv4, err := ioutil.ReadDir(path)
|
|
if err != nil {
|
|
logrus.Warn("filters#ReadDir#ipv4", err)
|
|
}
|
|
|
|
for _, name := range sliceIPv4 {
|
|
f, err := os.OpenFile(path+name.Name(), 1, 0644)
|
|
if err != nil {
|
|
logrus.Warn("filters#OpenFile#", err)
|
|
}
|
|
scanner := bufio.NewScanner(f)
|
|
for scanner.Scan() {
|
|
_, ipNet, err := net.ParseCIDR(scanner.Text())
|
|
if err != nil {
|
|
continue
|
|
}
|
|
if ipNet.Contains(ip) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
} else if ip.To16() != nil {
|
|
// run ipv6 filter battery
|
|
} else {
|
|
logrus.Error("filters#invalid IP format")
|
|
return false
|
|
}
|
|
|
|
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
|
|
}
|
|
}
|