Merge: - filters: 'enable/disable filter' didn't work

Squashed commit of the following:

commit c4c616162b66f2e89932ac44253ff6bedb8692eb
Merge: 800740da 579177fc
Author: Simon Zolin <s.zolin@adguard.com>
Date:   Wed Mar 11 19:14:27 2020 +0300

    Merge remote-tracking branch 'origin/master' into apply-disable-filter

commit 800740dace018dbe58abc579c06f7c494cdd1c15
Author: Simon Zolin <s.zolin@adguard.com>
Date:   Wed Mar 11 18:40:38 2020 +0300

    improve logic

commit 2aab4fbe658de09cd4c1d0badc178e0ce9e60251
Author: Simon Zolin <s.zolin@adguard.com>
Date:   Wed Mar 11 18:23:14 2020 +0300

    - filters: 'disable filter' didn't work
This commit is contained in:
Simon Zolin 2020-03-11 19:17:46 +03:00
parent 579177fc70
commit b600d7b09d
2 changed files with 35 additions and 11 deletions

View File

@ -185,9 +185,27 @@ func handleFilteringSetURL(w http.ResponseWriter, r *http.Request) {
} }
onConfigModified() onConfigModified()
if (status&(statusURLChanged|statusEnabledChanged)) != 0 && fj.Data.Enabled { restart := false
if (status & statusEnabledChanged) != 0 {
// we must add or remove filter rules
restart = true
}
if (status&statusUpdateRequired) != 0 && fj.Data.Enabled {
// download new filter and apply its rules // download new filter and apply its rules
_, _ = refreshFilters(fj.Whitelist, true) flags := FilterRefreshBlocklists
if fj.Whitelist {
flags = FilterRefreshAllowlists
}
nUpdated, _ := refreshFilters(flags, true)
// if at least 1 filter has been updated, refreshFilters() restarts the filtering automatically
// if not - we restart the filtering ourselves
restart = false
if nUpdated == 0 {
restart = true
}
}
if restart {
enableFilters(true)
} }
} }
@ -226,7 +244,11 @@ func handleFilteringRefresh(w http.ResponseWriter, r *http.Request) {
} }
Context.controlLock.Unlock() Context.controlLock.Unlock()
resp.Updated, err = refreshFilters(req.White, false) flags := FilterRefreshBlocklists
if req.White {
flags = FilterRefreshAllowlists
}
resp.Updated, err = refreshFilters(flags|FilterRefreshForce, false)
Context.controlLock.Lock() Context.controlLock.Lock()
if err != nil { if err != nil {
httpError(w, http.StatusInternalServerError, "%s", err) httpError(w, http.StatusInternalServerError, "%s", err)

View File

@ -78,6 +78,7 @@ const (
statusEnabledChanged = 2 statusEnabledChanged = 2
statusURLChanged = 4 statusURLChanged = 4
statusURLExists = 8 statusURLExists = 8
statusUpdateRequired = 0x10
) )
// Update properties for a filter specified by its URL // Update properties for a filter specified by its URL
@ -103,13 +104,15 @@ func filterSetProperties(url string, newf filter, whitelist bool) int {
f.Name = newf.Name f.Name = newf.Name
if f.URL != newf.URL { if f.URL != newf.URL {
r |= statusURLChanged r |= statusURLChanged | statusUpdateRequired
if filterExistsNoLock(newf.URL) { if filterExistsNoLock(newf.URL) {
return statusURLExists return statusURLExists
} }
f.URL = newf.URL f.URL = newf.URL
f.unload() f.unload()
f.LastUpdated = time.Time{} f.LastUpdated = time.Time{}
f.checksum = 0
f.RulesCount = 0
} }
if f.Enabled != newf.Enabled { if f.Enabled != newf.Enabled {
@ -121,8 +124,10 @@ func filterSetProperties(url string, newf filter, whitelist bool) int {
if e != nil { if e != nil {
// This isn't a fatal error, // This isn't a fatal error,
// because it may occur when someone removes the file from disk. // because it may occur when someone removes the file from disk.
// In this case the periodic update task will try to download the file.
f.LastUpdated = time.Time{} f.LastUpdated = time.Time{}
f.checksum = 0
f.RulesCount = 0
r |= statusUpdateRequired
} }
} }
} else { } else {
@ -257,20 +262,17 @@ func periodicallyRefreshFilters() {
} }
// Refresh filters // Refresh filters
// flags: FilterRefresh*
// important: // important:
// TRUE: ignore the fact that we're currently updating the filters // TRUE: ignore the fact that we're currently updating the filters
func refreshFilters(whitelist bool, important bool) (int, error) { func refreshFilters(flags int, important bool) (int, error) {
set := atomic.CompareAndSwapUint32(&refreshStatus, 0, 1) set := atomic.CompareAndSwapUint32(&refreshStatus, 0, 1)
if !important && !set { if !important && !set {
return 0, fmt.Errorf("Filters update procedure is already running") return 0, fmt.Errorf("Filters update procedure is already running")
} }
refreshLock.Lock() refreshLock.Lock()
flags := FilterRefreshBlocklists nUpdated, _ := refreshFiltersIfNecessary(flags)
if whitelist {
flags = FilterRefreshAllowlists
}
nUpdated, _ := refreshFiltersIfNecessary(flags | FilterRefreshForce)
refreshLock.Unlock() refreshLock.Unlock()
refreshStatus = 0 refreshStatus = 0
return nUpdated, nil return nUpdated, nil