From 20e318269267c681e93653f546293d8d3f2b12b0 Mon Sep 17 00:00:00 2001
From: Frank Denis <github@pureftpd.org>
Date: Wed, 10 Jan 2018 10:11:59 +0100
Subject: [PATCH] Improve the plugins interface

---
 dnscrypt-proxy/plugins.go | 37 +++++++++++++++++++++++++++++++------
 1 file changed, 31 insertions(+), 6 deletions(-)

diff --git a/dnscrypt-proxy/plugins.go b/dnscrypt-proxy/plugins.go
index c5853355..603f7047 100644
--- a/dnscrypt-proxy/plugins.go
+++ b/dnscrypt-proxy/plugins.go
@@ -9,6 +9,8 @@ type PluginsAction int
 const (
 	PluginsActionNone    = 0
 	PluginsActionForward = 1
+	PluginsActionDrop    = 2
+	PluginsActionReject  = 3
 )
 
 type PluginsState struct {
@@ -17,22 +19,35 @@ type PluginsState struct {
 	originalMaxPayloadSize int
 	maxPayloadSize         int
 	proto                  string
+	queryPlugins           *[]Plugin
+	responsePlugins        *[]Plugin
+}
+
+type Plugin interface {
+	Name() string
+	Description() string
+	Eval(pluginsState *PluginsState, msg *dns.Msg) error
 }
 
 func NewPluginsState() PluginsState {
-	return PluginsState{action: PluginsActionForward, maxPayloadSize: MaxDNSUDPPacketSize - ResponseOverhead}
+	queryPlugins := &[]Plugin{Plugin(new(PluginGetSetPayloadSize))}
+	responsePlugins := &[]Plugin{}
+	return PluginsState{action: PluginsActionForward, maxPayloadSize: MaxDNSUDPPacketSize - ResponseOverhead,
+		queryPlugins: queryPlugins, responsePlugins: responsePlugins}
 }
 
 func (pluginsState *PluginsState) ApplyQueryPlugins(packet []byte) ([]byte, error) {
+	pluginsState.action = PluginsActionForward
 	msg := dns.Msg{}
 	if err := msg.Unpack(packet); err != nil {
 		return packet, err
 	}
-
-	if ret := pluginsState.BuiltinPluginsGetSetPayloadSize(&msg); ret != nil {
-		return packet, ret
+	for _, plugin := range *pluginsState.queryPlugins {
+		if ret := plugin.Eval(pluginsState, &msg); ret != nil {
+			pluginsState.action = PluginsActionDrop
+			return packet, ret
+		}
 	}
-
 	packet2, err := msg.PackBuffer(packet)
 	if err != nil {
 		return packet, err
@@ -40,7 +55,17 @@ func (pluginsState *PluginsState) ApplyQueryPlugins(packet []byte) ([]byte, erro
 	return packet2, nil
 }
 
-func (pluginsState *PluginsState) BuiltinPluginsGetSetPayloadSize(msg *dns.Msg) error {
+type PluginGetSetPayloadSize struct{}
+
+func (plugin *PluginGetSetPayloadSize) Name() string {
+	return "get_set_payload_size"
+}
+
+func (plugin *PluginGetSetPayloadSize) Description() string {
+	return "Adjusts the maximum payload size advertised in queries sent to upstream servers."
+}
+
+func (plugin *PluginGetSetPayloadSize) Eval(pluginsState *PluginsState, msg *dns.Msg) error {
 	pluginsState.originalMaxPayloadSize = 512 - ResponseOverhead
 	opt := msg.IsEdns0()
 	dnssec := false