mirror of
https://github.com/safing/portmaster
synced 2025-09-02 02:29:12 +00:00
42 lines
738 B
Go
42 lines
738 B
Go
package geoip
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/safing/portbase/modules"
|
|
"github.com/safing/portmaster/updates"
|
|
)
|
|
|
|
var (
|
|
module *modules.Module
|
|
)
|
|
|
|
func init() {
|
|
module = modules.Register("geoip", prep, nil, nil, "base", "updates")
|
|
}
|
|
|
|
func prep() error {
|
|
return module.RegisterEventHook(
|
|
updates.ModuleName,
|
|
updates.ResourceUpdateEvent,
|
|
"Check for GeoIP database updates",
|
|
upgradeDatabases,
|
|
)
|
|
}
|
|
|
|
func upgradeDatabases(_ context.Context, _ interface{}) error {
|
|
dbFileLock.Lock()
|
|
reload := false
|
|
if dbCityFile != nil && dbCityFile.UpgradeAvailable() {
|
|
reload = true
|
|
}
|
|
if dbASNFile != nil && dbASNFile.UpgradeAvailable() {
|
|
reload = true
|
|
}
|
|
dbFileLock.Unlock()
|
|
|
|
if reload {
|
|
return ReloadDatabases()
|
|
}
|
|
return nil
|
|
}
|