Incorporates performance improvements.
This commit is contained in:
parent
f84d54313d
commit
36a1ded4a6
|
|
@ -259,8 +259,8 @@
|
|||
},
|
||||
{
|
||||
"ImportPath": "github.com/weppos/publicsuffix-go/publicsuffix",
|
||||
"Comment": "v0.4.0-20-gcbbcd04",
|
||||
"Rev": "cbbcd048f995c801105a0083cb157c4fec9ea89c"
|
||||
"Comment": "v0.4.0-24-gf8afde6",
|
||||
"Rev": "f8afde64429e7c479c7e4f681f66fbf792dac9c6"
|
||||
},
|
||||
{
|
||||
"ImportPath": "github.com/zmap/zcrypto/json",
|
||||
|
|
|
|||
|
|
@ -11,7 +11,6 @@ import (
|
|||
"io"
|
||||
"net/http/cookiejar"
|
||||
"os"
|
||||
"regexp"
|
||||
"strings"
|
||||
|
||||
"golang.org/x/net/idna"
|
||||
|
|
@ -80,13 +79,14 @@ type FindOptions struct {
|
|||
// List represents a Public Suffix List.
|
||||
type List struct {
|
||||
// rules is kept private because you should not access rules directly
|
||||
// for lookup optimization the list will not be guaranteed to be a simple slice forever
|
||||
rules []Rule
|
||||
rules map[string]*Rule
|
||||
}
|
||||
|
||||
// NewList creates a new empty list.
|
||||
func NewList() *List {
|
||||
return &List{}
|
||||
return &List{
|
||||
rules: map[string]*Rule{},
|
||||
}
|
||||
}
|
||||
|
||||
// NewListFromString parses a string that represents a Public Suffix source
|
||||
|
|
@ -132,7 +132,7 @@ func (l *List) LoadFile(path string, options *ParserOption) ([]Rule, error) {
|
|||
// The list may be optimized internally for lookups, therefore the algorithm
|
||||
// will decide the best position for the new rule.
|
||||
func (l *List) AddRule(r *Rule) error {
|
||||
l.rules = append(l.rules, *r)
|
||||
l.rules[r.Value] = r
|
||||
return nil
|
||||
}
|
||||
|
||||
|
|
@ -195,43 +195,27 @@ Scanning:
|
|||
|
||||
// Find and returns the most appropriate rule for the domain name.
|
||||
func (l *List) Find(name string, options *FindOptions) *Rule {
|
||||
var bestRule *Rule
|
||||
|
||||
if options == nil {
|
||||
options = DefaultFindOptions
|
||||
}
|
||||
|
||||
for _, r := range l.selectRules(name, options) {
|
||||
if r.Type == ExceptionType {
|
||||
return &r
|
||||
}
|
||||
if bestRule == nil || bestRule.Length < r.Length {
|
||||
bestRule = &r
|
||||
}
|
||||
}
|
||||
|
||||
if bestRule != nil {
|
||||
return bestRule
|
||||
part := name
|
||||
for {
|
||||
rule, ok := l.rules[part]
|
||||
|
||||
if ok && rule.Match(name) && !(options.IgnorePrivate && rule.Private) {
|
||||
return rule
|
||||
}
|
||||
|
||||
i := strings.IndexRune(part, '.')
|
||||
if i < 0 {
|
||||
return options.DefaultRule
|
||||
}
|
||||
|
||||
func (l *List) selectRules(name string, options *FindOptions) []Rule {
|
||||
var found []Rule
|
||||
|
||||
// In this phase the search is a simple sequential scan
|
||||
for _, rule := range l.rules {
|
||||
if !rule.Match(name) {
|
||||
continue
|
||||
}
|
||||
if options.IgnorePrivate && rule.Private {
|
||||
continue
|
||||
}
|
||||
found = append(found, rule)
|
||||
part = part[i+1:]
|
||||
}
|
||||
|
||||
return found
|
||||
return nil
|
||||
}
|
||||
|
||||
// NewRule parses the rule content, creates and returns a Rule.
|
||||
|
|
@ -309,36 +293,46 @@ func (r *Rule) Match(name string) bool {
|
|||
|
||||
// Decompose takes a name as input and decomposes it into a tuple of <TRD+SLD, TLD>,
|
||||
// according to the rule definition and type.
|
||||
func (r *Rule) Decompose(name string) [2]string {
|
||||
var parts []string
|
||||
|
||||
func (r *Rule) Decompose(name string) (result [2]string) {
|
||||
if r == DefaultRule {
|
||||
i := strings.LastIndex(name, ".")
|
||||
if i < 0 {
|
||||
return
|
||||
}
|
||||
result[0], result[1] = name[:i], name[i+1:]
|
||||
return
|
||||
}
|
||||
switch r.Type {
|
||||
case NormalType:
|
||||
name = strings.TrimSuffix(name, r.Value)
|
||||
if len(name) == 0 {
|
||||
return
|
||||
}
|
||||
result[0], result[1] = name[:len(name)-1], r.Value
|
||||
case WildcardType:
|
||||
parts = append([]string{`.*?`}, r.parts()...)
|
||||
default:
|
||||
parts = r.parts()
|
||||
name := strings.TrimSuffix(name, r.Value)
|
||||
if len(name) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
suffix := strings.Join(parts, `\.`)
|
||||
re := regexp.MustCompile(fmt.Sprintf(`^(.+)\.(%s)$`, suffix))
|
||||
|
||||
matches := re.FindStringSubmatch(name)
|
||||
if len(matches) < 3 {
|
||||
return [2]string{"", ""}
|
||||
name = name[:len(name)-1]
|
||||
i := strings.LastIndex(name, ".")
|
||||
if i < 0 {
|
||||
return
|
||||
}
|
||||
|
||||
return [2]string{matches[1], matches[2]}
|
||||
result[0], result[1] = name[:i], name[i+1:]+"."+r.Value
|
||||
case ExceptionType:
|
||||
i := strings.IndexRune(r.Value, '.')
|
||||
if i < 0 {
|
||||
return
|
||||
}
|
||||
|
||||
func (r *Rule) parts() []string {
|
||||
labels := Labels(r.Value)
|
||||
if r.Type == ExceptionType {
|
||||
return labels[1:]
|
||||
suffix := r.Value[i+1:]
|
||||
name = strings.TrimSuffix(name, suffix)
|
||||
if len(name) == 0 {
|
||||
return
|
||||
}
|
||||
if r.Type == WildcardType && r.Value == "" {
|
||||
return []string{}
|
||||
result[0], result[1] = name[:len(name)-1], suffix
|
||||
}
|
||||
return labels
|
||||
return
|
||||
}
|
||||
|
||||
// Labels decomposes given domain name into labels,
|
||||
|
|
@ -432,7 +426,6 @@ func DomainFromListWithOptions(l *List, name string, options *FindOptions) (stri
|
|||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return dn.SLD + "." + dn.TLD, nil
|
||||
}
|
||||
|
||||
|
|
@ -458,12 +451,22 @@ func ParseFromListWithOptions(l *List, name string, options *FindOptions) (*Doma
|
|||
}
|
||||
|
||||
r := l.Find(n, options)
|
||||
if tld := r.Decompose(n)[1]; tld == "" {
|
||||
parts := r.Decompose(n)
|
||||
left, tld := parts[0], parts[1]
|
||||
if tld == "" {
|
||||
return nil, fmt.Errorf("%s is a suffix", n)
|
||||
}
|
||||
|
||||
dn := &DomainName{Rule: r}
|
||||
dn.TLD, dn.SLD, dn.TRD = decompose(r, n)
|
||||
dn := &DomainName{
|
||||
Rule: r,
|
||||
TLD: tld,
|
||||
}
|
||||
if i := strings.LastIndex(left, "."); i < 0 {
|
||||
dn.SLD = left
|
||||
} else {
|
||||
dn.TRD = left[:i]
|
||||
dn.SLD = left[i+1:]
|
||||
}
|
||||
return dn, nil
|
||||
}
|
||||
|
||||
|
|
@ -471,31 +474,15 @@ func normalize(name string) (string, error) {
|
|||
ret := strings.ToLower(name)
|
||||
|
||||
if ret == "" {
|
||||
return "", fmt.Errorf("Name is blank")
|
||||
return "", fmt.Errorf("name is blank")
|
||||
}
|
||||
if ret[0] == '.' {
|
||||
return "", fmt.Errorf("Name %s starts with a dot", ret)
|
||||
return "", fmt.Errorf("name %s starts with a dot", ret)
|
||||
}
|
||||
|
||||
return ret, nil
|
||||
}
|
||||
|
||||
func decompose(r *Rule, name string) (tld, sld, trd string) {
|
||||
parts := r.Decompose(name)
|
||||
left, tld := parts[0], parts[1]
|
||||
|
||||
dot := strings.LastIndex(left, ".")
|
||||
if dot == -1 {
|
||||
sld = left
|
||||
trd = ""
|
||||
} else {
|
||||
sld = left[dot+1:]
|
||||
trd = left[0:dot]
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
// ToASCII is a wrapper for idna.ToASCII.
|
||||
//
|
||||
// This wrapper exists because idna.ToASCII backward-compatibility was broken twice in few months
|
||||
|
|
|
|||
|
|
@ -3,10 +3,10 @@
|
|||
|
||||
package publicsuffix
|
||||
|
||||
const defaultListVersion = "PSL version f8ccab (Wed Aug 8 09:06:53 2018)"
|
||||
const defaultListVersion = "PSL version 6cca94 (Mon Aug 13 08:12:02 2018)"
|
||||
|
||||
func init() {
|
||||
r := [8618]Rule{
|
||||
r := [8619]Rule{
|
||||
{1, "ac", 1, false},
|
||||
{1, "com.ac", 2, false},
|
||||
{1, "edu.ac", 2, false},
|
||||
|
|
@ -972,6 +972,7 @@ func init() {
|
|||
{1, "my.id", 2, false},
|
||||
{1, "net.id", 2, false},
|
||||
{1, "or.id", 2, false},
|
||||
{1, "ponpes.id", 2, false},
|
||||
{1, "sch.id", 2, false},
|
||||
{1, "web.id", 2, false},
|
||||
{1, "ie", 1, false},
|
||||
|
|
@ -8626,5 +8627,7 @@ func init() {
|
|||
{1, "now.sh", 2, true},
|
||||
{1, "zone.id", 2, true},
|
||||
}
|
||||
DefaultList.rules = r[:]
|
||||
for i := range r {
|
||||
DefaultList.AddRule(&r[i])
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue