[+] log layer [+]

krishpranav · krishpranav · commit 30120fad2563 · 2021-06-26T16:20:06.000+05:30
diff --git a/pkg/httpserver/loglayer.go b/pkg/httpserver/loglayer.go
@@ -1,6 +1,44 @@
 package httpserver
 
+import (
+	"bytes"
+	"io/ioutil"
+	"net/http"
+	"net/http/httputil"
+	"path"
+
+	"github.com/projectdiscovery/gologger"
+)
+
 var (
 	EnableUpload  bool
 	EnableVerbose bool
 )
+
+func (t *HTTPServer) loglayer(handler http.Handler) http.Handler {
+	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		fullRequest, _ := httputil.DumpRequest(r, true)
+		lrw := newLoggingResponseWriter(w)
+		handler.ServeHTTP(lrw, r)
+
+		// Handles file write if enabled
+		if EnableUpload && r.Method == http.MethodPut {
+			data, err := ioutil.ReadAll(r.Body)
+			if err != nil {
+				gologger.Print().Msgf("%s\n", err)
+			}
+			err = handleUpload(path.Base(r.URL.Path), data)
+			if err != nil {
+				gologger.Print().Msgf("%s\n", err)
+			}
+		}
+
+		if EnableVerbose {
+			headers := new(bytes.Buffer)
+			lrw.Header().Write(headers) //nolint
+			gologger.Print().Msgf("\nRemote Address: %s\n%s\n%s %d %s\n%s\n%s\n", r.RemoteAddr, string(fullRequest), r.Proto, lrw.statusCode, http.StatusText(lrw.statusCode), headers.String(), string(lrw.Data))
+		} else {
+			gologger.Print().Msgf("%s \"%s %s %s\" %d %d", r.RemoteAddr, r.Method, r.URL, r.Proto, lrw.statusCode, len(lrw.Data))
+		}
+	})
+}
