mirror of
https://github.com/safing/portmaster
synced 2025-09-04 11:39:29 +00:00
Update filterlists to use new format
Also, add database reset flag
This commit is contained in:
parent
d5c00e31fd
commit
563bff1d95
3 changed files with 34 additions and 6 deletions
|
@ -4,15 +4,20 @@ import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"sync"
|
"sync"
|
||||||
|
|
||||||
|
"github.com/safing/portbase/database"
|
||||||
|
|
||||||
"github.com/hashicorp/go-version"
|
"github.com/hashicorp/go-version"
|
||||||
"github.com/safing/portbase/database/record"
|
"github.com/safing/portbase/database/record"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const resetVersion = "v0.6.0"
|
||||||
|
|
||||||
type cacheVersionRecord struct {
|
type cacheVersionRecord struct {
|
||||||
record.Base
|
record.Base
|
||||||
sync.Mutex
|
sync.Mutex
|
||||||
|
|
||||||
Version string
|
Version string
|
||||||
|
Reset string
|
||||||
}
|
}
|
||||||
|
|
||||||
// getCacheDatabaseVersion reads and returns the cache
|
// getCacheDatabaseVersion reads and returns the cache
|
||||||
|
@ -37,6 +42,10 @@ func getCacheDatabaseVersion() (*version.Version, error) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if verRecord.Reset != resetVersion {
|
||||||
|
return nil, database.ErrNotFound
|
||||||
|
}
|
||||||
|
|
||||||
ver, err := version.NewSemver(verRecord.Version)
|
ver, err := version.NewSemver(verRecord.Version)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
|
@ -50,6 +59,7 @@ func getCacheDatabaseVersion() (*version.Version, error) {
|
||||||
func setCacheDatabaseVersion(ver string) error {
|
func setCacheDatabaseVersion(ver string) error {
|
||||||
verRecord := &cacheVersionRecord{
|
verRecord := &cacheVersionRecord{
|
||||||
Version: ver,
|
Version: ver,
|
||||||
|
Reset: resetVersion,
|
||||||
}
|
}
|
||||||
|
|
||||||
verRecord.SetKey(filterListCacheVersionKey)
|
verRecord.SetKey(filterListCacheVersionKey)
|
||||||
|
|
|
@ -200,14 +200,14 @@ func processEntry(ctx context.Context, filter *scopedBloom, entry *listEntry, re
|
||||||
normalizeEntry(entry)
|
normalizeEntry(entry)
|
||||||
|
|
||||||
// Only add the entry to the bloom filter if it has any sources.
|
// Only add the entry to the bloom filter if it has any sources.
|
||||||
if len(entry.Sources) > 0 {
|
if len(entry.Resources) > 0 {
|
||||||
filter.add(entry.Type, entry.Entity)
|
filter.add(entry.Type, entry.Entity)
|
||||||
}
|
}
|
||||||
|
|
||||||
r := &entityRecord{
|
r := &entityRecord{
|
||||||
Value: entry.Entity,
|
Value: entry.Entity,
|
||||||
Type: entry.Type,
|
Type: entry.Type,
|
||||||
Sources: entry.Sources,
|
Sources: entry.getSources(),
|
||||||
UpdatedAt: time.Now().Unix(),
|
UpdatedAt: time.Now().Unix(),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -8,13 +8,31 @@ import (
|
||||||
"io"
|
"io"
|
||||||
|
|
||||||
"github.com/safing/portbase/formats/dsd"
|
"github.com/safing/portbase/formats/dsd"
|
||||||
|
"github.com/safing/portbase/utils"
|
||||||
)
|
)
|
||||||
|
|
||||||
type listEntry struct {
|
type listEntry struct {
|
||||||
Entity string `json:"entity"`
|
Type string `json:"type"`
|
||||||
Sources []string `json:"sources"`
|
Entity string `json:"entity"`
|
||||||
Whitelist bool `json:"whitelist"`
|
Whitelist bool `json:"whitelist"`
|
||||||
Type string `json:"type"`
|
Resources []entryResource `json:"resources"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type entryResource struct {
|
||||||
|
SourceID string `json:"sourceID"`
|
||||||
|
ResourceID string `json:"resourceID"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (entry *listEntry) getSources() (sourceIDs []string) {
|
||||||
|
sourceIDs = make([]string, 0, len(entry.Resources))
|
||||||
|
|
||||||
|
for _, resource := range entry.Resources {
|
||||||
|
if !utils.StringInSlice(sourceIDs, resource.SourceID) {
|
||||||
|
sourceIDs = append(sourceIDs, resource.SourceID)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// decodeFile decodes a DSDL filterlists file and sends decoded entities to
|
// decodeFile decodes a DSDL filterlists file and sends decoded entities to
|
||||||
|
|
Loading…
Add table
Reference in a new issue