From d8e3969640380de4d2f19bf3527d28ad5b5b4500 Mon Sep 17 00:00:00 2001
From: James Swineson <github@public.swineson.me>
Date: Mon, 11 Nov 2019 10:11:25 +0800
Subject: [PATCH] add no_ecs argument in query string support

---
 doh-server/server.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/doh-server/server.go b/doh-server/server.go
index 24113eb..5a1b0d2 100644
--- a/doh-server/server.go
+++ b/doh-server/server.go
@@ -234,6 +234,11 @@ func (s *Server) handlerFunc(w http.ResponseWriter, r *http.Request) {
 }
 
 func (s *Server) findClientIP(r *http.Request) net.IP {
+	noEcs := r.URL.Query().Get("no_ecs")
+	if strings.ToLower(noEcs) == "true" {
+		return nil
+	}
+
 	XForwardedFor := r.Header.Get("X-Forwarded-For")
 	if XForwardedFor != "" {
 		for _, addr := range strings.Split(XForwardedFor, ",") {