From ad1d4d4d071240276e075a20ad761680350ba428 Mon Sep 17 00:00:00 2001
From: Kit Patella <kit@jepsen.io>
Date: Wed, 13 May 2020 15:47:05 -0700
Subject: [PATCH] http: migrate from instrumentation in s.wrap() to an
 s.enterpriseHandler()

---
 agent/http.go     | 34 ++++++++--------------------------
 agent/http_oss.go | 12 +++---------
 2 files changed, 11 insertions(+), 35 deletions(-)

diff --git a/agent/http.go b/agent/http.go
index b4c6b64ac4..c52c232f1c 100644
--- a/agent/http.go
+++ b/agent/http.go
@@ -350,6 +350,7 @@ func (s *HTTPServer) handler(enableDebug bool) http.Handler {
 	if s.agent.config.DisableHTTPUnprintableCharFilter {
 		h = mux
 	}
+	h = s.enterpriseHandler(h)
 	return &wrappedMux{
 		mux:     mux,
 		handler: h,
@@ -399,10 +400,6 @@ var (
 func (s *HTTPServer) wrap(handler endpoint, methods []string) http.HandlerFunc {
 	httpLogger := s.agent.logger.Named(logging.HTTP)
 	return func(resp http.ResponseWriter, req *http.Request) {
-
-		// Audit log the request
-		reqPayload := s.auditReq(req)
-
 		setHeaders(resp, s.agent.config.HTTPResponseHeaders)
 		setTranslateAddr(resp, s.agent.config.TranslateWANAddrs)
 
@@ -480,44 +477,33 @@ func (s *HTTPServer) wrap(handler endpoint, methods []string) http.HandlerFunc {
 				"from", req.RemoteAddr,
 				"error", err,
 			)
-			var httpCode int
 			switch {
 			case isForbidden(err):
-				httpCode = http.StatusForbidden
-				resp.WriteHeader(httpCode)
+				resp.WriteHeader(http.StatusForbidden)
 				fmt.Fprint(resp, err.Error())
 			case structs.IsErrRPCRateExceeded(err):
-				httpCode = http.StatusTooManyRequests
-				resp.WriteHeader(httpCode)
+				resp.WriteHeader(http.StatusTooManyRequests)
 			case isMethodNotAllowed(err):
 				// RFC2616 states that for 405 Method Not Allowed the response
 				// MUST include an Allow header containing the list of valid
 				// methods for the requested resource.
 				// https://www.w3.org/Protocols/rfc2616/rfc2616-sec10.html
 				addAllowHeader(err.(MethodNotAllowedError).Allow)
-				httpCode = http.StatusMethodNotAllowed
-				resp.WriteHeader(httpCode) // 405
+				resp.WriteHeader(http.StatusMethodNotAllowed) // 405
 				fmt.Fprint(resp, err.Error())
 			case isBadRequest(err):
-				httpCode = http.StatusBadRequest
-				resp.WriteHeader(httpCode)
+				resp.WriteHeader(http.StatusBadRequest)
 				fmt.Fprint(resp, err.Error())
 			case isNotFound(err):
-				httpCode = http.StatusNotFound
-				resp.WriteHeader(httpCode)
+				resp.WriteHeader(http.StatusNotFound)
 				fmt.Fprintf(resp, err.Error())
 			case isTooManyRequests(err):
-				httpCode = http.StatusTooManyRequests
-				resp.WriteHeader(httpCode)
+				resp.WriteHeader(http.StatusTooManyRequests)
 				fmt.Fprint(resp, err.Error())
 			default:
-				httpCode = http.StatusInternalServerError
-				resp.WriteHeader(httpCode)
+				resp.WriteHeader(http.StatusInternalServerError)
 				fmt.Fprint(resp, err.Error())
 			}
-
-			// Audit log the error response
-			s.auditResp(reqPayload, httpCode)
 		}
 
 		start := time.Now()
@@ -592,10 +578,6 @@ func (s *HTTPServer) wrap(handler endpoint, methods []string) http.HandlerFunc {
 		}
 		resp.Header().Set("Content-Type", contentType)
 		resp.WriteHeader(httpCode)
-
-		// Audit log the success response
-		s.auditResp(reqPayload, httpCode)
-
 		resp.Write(buf)
 	}
 }
diff --git a/agent/http_oss.go b/agent/http_oss.go
index d264bf74d5..415a8eeaf1 100644
--- a/agent/http_oss.go
+++ b/agent/http_oss.go
@@ -53,13 +53,7 @@ func parseACLAuthMethodEnterpriseMeta(req *http.Request, _ *structs.ACLAuthMetho
 	return nil
 }
 
-// auditReq is a noop stub for the corresponding func in http_ent.go
-func (s *HTTPServer) auditReq(req *http.Request) interface{} {
-	// note(kit): We return an nil here so we can pass it to auditResp. Auditing the response requires the
-	//  request object for context, so we have it pass it even when it's disabled
-	return nil
-}
-
-// auditResp is a noop stub for the corresponding func in http_ent.go
-func (s *HTTPServer) auditResp(reqPayload interface{}, httpCode int) {
+// enterpriseHandler is a noop for the enterprise implementation. we pass the original back
+func (s *HTTPServer) enterpriseHandler(next http.Handler) http.Handler {
+	return next
 }