Add support for matching continents
This commit is contained in:
63
profile/endpoints/endpoint-continent.go
Normal file
63
profile/endpoints/endpoint-continent.go
Normal file
@@ -0,0 +1,63 @@
|
||||
package endpoints
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"github.com/safing/portmaster/intel"
|
||||
)
|
||||
|
||||
var (
|
||||
continentCodePrefix = "C:"
|
||||
continentRegex = regexp.MustCompile(`^C:[A-Z]{2}$`)
|
||||
)
|
||||
|
||||
// EndpointContinent matches countries.
|
||||
type EndpointContinent struct {
|
||||
EndpointBase
|
||||
|
||||
ContinentCode string
|
||||
}
|
||||
|
||||
// Matches checks whether the given entity matches this endpoint definition.
|
||||
func (ep *EndpointContinent) Matches(ctx context.Context, entity *intel.Entity) (EPResult, Reason) {
|
||||
if entity.IP == nil {
|
||||
return NoMatch, nil
|
||||
}
|
||||
|
||||
if !entity.IPScope.IsGlobal() {
|
||||
return NoMatch, nil
|
||||
}
|
||||
|
||||
countryInfo := entity.GetCountryInfo(ctx)
|
||||
if countryInfo == nil {
|
||||
return MatchError, ep.makeReason(ep, "", "country data not available to match")
|
||||
}
|
||||
|
||||
if ep.ContinentCode == countryInfo.Continent.Code {
|
||||
return ep.match(
|
||||
ep, entity,
|
||||
fmt.Sprintf("%s (%s)", countryInfo.Continent.Name, countryInfo.Continent.Code),
|
||||
"IP is located in",
|
||||
)
|
||||
}
|
||||
|
||||
return NoMatch, nil
|
||||
}
|
||||
|
||||
func (ep *EndpointContinent) String() string {
|
||||
return ep.renderPPP(continentCodePrefix + ep.ContinentCode)
|
||||
}
|
||||
|
||||
func parseTypeContinent(fields []string) (Endpoint, error) {
|
||||
if continentRegex.MatchString(fields[1]) {
|
||||
ep := &EndpointContinent{
|
||||
ContinentCode: strings.TrimPrefix(strings.ToUpper(fields[1]), continentCodePrefix),
|
||||
}
|
||||
return ep.parsePPP(ep, fields)
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
@@ -232,6 +232,10 @@ func parseEndpoint(value string) (endpoint Endpoint, err error) { //nolint:gocog
|
||||
if endpoint, err = parseTypeCountry(fields); endpoint != nil || err != nil {
|
||||
return
|
||||
}
|
||||
// continent
|
||||
if endpoint, err = parseTypeContinent(fields); endpoint != nil || err != nil {
|
||||
return
|
||||
}
|
||||
// asn
|
||||
if endpoint, err = parseTypeASN(fields); endpoint != nil || err != nil {
|
||||
return
|
||||
|
||||
Reference in New Issue
Block a user