From c9d7bc30698555677a1922d7dfb8e3882249cbba Mon Sep 17 00:00:00 2001
From: Simon Zolin <s.zolin@adguard.com>
Date: Thu, 10 Oct 2019 17:12:32 +0300
Subject: [PATCH] * "refresh filters" HTTP handler: return the old synchronous
 behaviour

This was broken by "* dnsfilter: major refactoring"
---
 home/control_filtering.go | 10 ++++++++--
 home/filter.go            | 35 +++++++++++++++++++++--------------
 2 files changed, 29 insertions(+), 16 deletions(-)

diff --git a/home/control_filtering.go b/home/control_filtering.go
index 453e0ec4..2eddaf13 100644
--- a/home/control_filtering.go
+++ b/home/control_filtering.go
@@ -177,8 +177,14 @@ func handleFilteringSetRules(w http.ResponseWriter, r *http.Request) {
 }
 
 func handleFilteringRefresh(w http.ResponseWriter, r *http.Request) {
-	beginRefreshFilters()
-	fmt.Fprintf(w, "OK 0 filters updated\n")
+	config.controlLock.Unlock()
+	nUpdated, err := refreshFilters()
+	config.controlLock.Lock()
+	if err != nil {
+		httpError(w, http.StatusInternalServerError, "%s", err)
+		return
+	}
+	fmt.Fprintf(w, "OK %d filters updated\n", nUpdated)
 }
 
 type filterJSON struct {
diff --git a/home/filter.go b/home/filter.go
index 38425301..07a75e61 100644
--- a/home/filter.go
+++ b/home/filter.go
@@ -9,6 +9,7 @@ import (
 	"regexp"
 	"strconv"
 	"strings"
+	"sync"
 	"time"
 
 	"github.com/AdguardTeam/AdGuardHome/dnsfilter"
@@ -19,7 +20,8 @@ import (
 var (
 	nextFilterID      = time.Now().Unix() // semi-stable way to generate an unique ID
 	filterTitleRegexp = regexp.MustCompile(`^! Title: +(.*)$`)
-	forceRefresh      bool
+	refreshStatus     uint32 // 0:none; 1:in progress
+	refreshLock       sync.Mutex
 )
 
 func initFiltering() {
@@ -175,25 +177,30 @@ func assignUniqueFilterID() int64 {
 
 // Sets up a timer that will be checking for filters updates periodically
 func periodicallyRefreshFilters() {
-	nextRefresh := int64(0)
 	for {
-		if forceRefresh {
-			_ = refreshFiltersIfNecessary(true)
-			forceRefresh = false
-		}
-
-		if config.DNS.FiltersUpdateIntervalHours != 0 && nextRefresh <= time.Now().Unix() {
+		if config.DNS.FiltersUpdateIntervalHours != 0 && refreshStatus != 0 {
+			refreshStatus = 1
+			refreshLock.Lock()
 			_ = refreshFiltersIfNecessary(false)
-			nextRefresh = time.Now().Add(1 * time.Hour).Unix()
+			refreshLock.Unlock()
+			refreshStatus = 0
 		}
-		time.Sleep(1 * time.Second)
+		time.Sleep(1 * time.Hour)
 	}
 }
 
-// Schedule the procedure to refresh filters
-func beginRefreshFilters() {
-	forceRefresh = true
-	log.Debug("Filters: schedule update")
+// Refresh filters
+func refreshFilters() (int, error) {
+	if refreshStatus != 0 { // we could use atomic cmpxchg here, but it's not really required
+		return 0, fmt.Errorf("Filters update procedure is already running")
+	}
+
+	refreshStatus = 1
+	refreshLock.Lock()
+	nUpdated := refreshFiltersIfNecessary(true)
+	refreshLock.Unlock()
+	refreshStatus = 0
+	return nUpdated, nil
 }
 
 // Checks filters updates if necessary