beenull/crowdsec
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Merge branch 'datafiles_in_hub' of github.com:crowdsecurity/crowdsec into datafiles_in_hub

Browse source
This commit is contained in:
alteredCoder 2022-02-04 16:26:34 +01:00
parent ca224df7fa 032416ef3b
commit 78da1e77f1
1 changed files with 36 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

50
pkg/parser/enrich_geoip.go
View file

@ -87,32 +87,54 @@ func GeoIpCity(field string, p *types.Event, ctx interface{}) (map[string]string
return ret, nil
}
/*deal with data versioning without breaking : support crowdsecurity/xxx or xxx*/
func GeoIPCityInit(cfg map[string]string) (interface{}, error) {
dbCityReader, err := geoip2.Open(cfg["datadir"] + "/GeoLite2-City.mmdb")
if err != nil {
log.Debugf("couldn't open geoip : %v", err)
return nil, err
if err == nil {
return dbCityReader, nil
}
dbCityReader, err = geoip2.Open(cfg["datadir"] + "/crowdsecurity/GeoLite2-City.mmdb")
if err == nil {
return dbCityReader, nil
return dbCityReader, nil
}
log.Debugf("couldn't open geoip (%s nor %s) : %v",
cfg["datadir"]+"/crowdsecurity/GeoLite2-City.mmdb",
cfg["datadir"]+"/GeoLite2-City.mmdb",
err)
return nil, err
}
func GeoIPASNInit(cfg map[string]string) (interface{}, error) {
dbASReader, err := geoip2.Open(cfg["datadir"] + "/GeoLite2-ASN.mmdb")
if err != nil {
log.Debugf("couldn't open geoip : %v", err)
return nil, err
}
if err == nil {
return dbASReader, nil
return dbASReader, nil
}
dbASReader, err = geoip2.Open(cfg["datadir"] + "/crowdsecurity/GeoLite2-ASN.mmdb")
if err == nil {
return dbASReader, nil
}
log.Debugf("couldn't open geoip (%s nor %s) : %v",
cfg["datadir"]+"/crowdsecurity/GeoLite2-ASN.mmdb",
cfg["datadir"]+"/GeoLite2-ASN.mmdb",
err)
return nil, err
}
func IpToRangeInit(cfg map[string]string) (interface{}, error) {
ipToRangeReader, err := maxminddb.Open(cfg["datadir"] + "/GeoLite2-ASN.mmdb")
if err != nil {
log.Debugf("couldn't open geoip : %v", err)
return nil, err
if err == nil {
return ipToRangeReader, nil
}
return ipToRangeReader, nil
ipToRangeReader, err = maxminddb.Open(cfg["datadir"] + "/crowdsecurity/GeoLite2-ASN.mmdb")
if err == nil {
return ipToRangeReader, nil
}
log.Debugf("couldn't open geoip (%s nor %s) : %v",
cfg["datadir"]+"/crowdsecurity/GeoLite2-ASN.mmdb",
cfg["datadir"]+"/GeoLite2-ASN.mmdb",
err)
return nil, err
}