120 lines
3.3 KiB
Go
120 lines
3.3 KiB
Go
package home
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/AdguardTeam/AdGuardHome/dnsfilter"
|
|
"github.com/AdguardTeam/golibs/log"
|
|
"github.com/AdguardTeam/urlfilter"
|
|
)
|
|
|
|
var serviceRules map[string][]*urlfilter.NetworkRule // service name -> filtering rules
|
|
|
|
type svc struct {
|
|
name string
|
|
rules []string
|
|
}
|
|
|
|
// Keep in sync with:
|
|
// client/src/helpers/constants.js
|
|
// client/src/components/ui/Icons.js
|
|
var serviceRulesArray = []svc{
|
|
{"whatsapp", []string{"||whatsapp.net^"}},
|
|
{"facebook", []string{"||facebook.com^"}},
|
|
{"twitter", []string{"||twitter.com^", "||t.co^", "||twimg.com^"}},
|
|
{"youtube", []string{"||youtube.com^", "||ytimg.com^"}},
|
|
{"messenger", []string{"||fb.com^", "||facebook.com^"}},
|
|
{"twitch", []string{"||twitch.tv^", "||ttvnw.net^"}},
|
|
{"netflix", []string{"||nflxext.com^", "||netflix.com^"}},
|
|
{"instagram", []string{"||instagram.com^"}},
|
|
{"snapchat", []string{"||snapchat.com^"}},
|
|
{"discord", []string{"||discord.gg^", "||discordapp.net^", "||discordapp.com^"}},
|
|
{"ok", []string{"||ok.ru^"}},
|
|
{"skype", []string{"||skype.com^"}},
|
|
{"vk", []string{"||vk.com^"}},
|
|
{"steam", []string{"||steam.com^"}},
|
|
{"mail_ru", []string{"||mail.ru^"}},
|
|
}
|
|
|
|
// convert array to map
|
|
func initServices() {
|
|
serviceRules = make(map[string][]*urlfilter.NetworkRule)
|
|
for _, s := range serviceRulesArray {
|
|
rules := []*urlfilter.NetworkRule{}
|
|
for _, text := range s.rules {
|
|
rule, err := urlfilter.NewNetworkRule(text, 0)
|
|
if err != nil {
|
|
log.Error("urlfilter.NewNetworkRule: %s rule: %s", err, text)
|
|
continue
|
|
}
|
|
rules = append(rules, rule)
|
|
}
|
|
serviceRules[s.name] = rules
|
|
}
|
|
}
|
|
|
|
// ApplyBlockedServices - set blocked services settings for this DNS request
|
|
func ApplyBlockedServices(setts *dnsfilter.RequestFilteringSettings, list []string) {
|
|
setts.ServicesRules = []dnsfilter.ServiceEntry{}
|
|
for _, name := range list {
|
|
rules, ok := serviceRules[name]
|
|
|
|
if !ok {
|
|
log.Error("unknown service name: %s", name)
|
|
continue
|
|
}
|
|
|
|
s := dnsfilter.ServiceEntry{}
|
|
s.Name = name
|
|
s.Rules = rules
|
|
setts.ServicesRules = append(setts.ServicesRules, s)
|
|
}
|
|
}
|
|
|
|
func handleBlockedServicesList(w http.ResponseWriter, r *http.Request) {
|
|
log.Tracef("%s %v", r.Method, r.URL)
|
|
|
|
config.RLock()
|
|
list := config.DNS.BlockedServices
|
|
config.RUnlock()
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
err := json.NewEncoder(w).Encode(list)
|
|
if err != nil {
|
|
httpError(w, http.StatusInternalServerError, "json.Encode: %s", err)
|
|
return
|
|
}
|
|
}
|
|
|
|
func handleBlockedServicesSet(w http.ResponseWriter, r *http.Request) {
|
|
log.Tracef("%s %v", r.Method, r.URL)
|
|
|
|
list := []string{}
|
|
err := json.NewDecoder(r.Body).Decode(&list)
|
|
if err != nil {
|
|
httpError(w, http.StatusBadRequest, "json.Decode: %s", err)
|
|
return
|
|
}
|
|
|
|
config.Lock()
|
|
config.DNS.BlockedServices = list
|
|
config.Unlock()
|
|
|
|
log.Debug("Updated blocked services list: %d", len(list))
|
|
|
|
err = writeAllConfigsAndReloadDNS()
|
|
if err != nil {
|
|
httpError(w, http.StatusBadRequest, "%s", err)
|
|
return
|
|
}
|
|
|
|
returnOK(w)
|
|
}
|
|
|
|
// RegisterBlockedServicesHandlers - register HTTP handlers
|
|
func RegisterBlockedServicesHandlers() {
|
|
http.HandleFunc("/control/blocked_services/list", postInstall(optionalAuth(ensureGET(handleBlockedServicesList))))
|
|
http.HandleFunc("/control/blocked_services/set", postInstall(optionalAuth(ensurePOST(handleBlockedServicesSet))))
|
|
}
|