[Bug Fix] Propagate query args during URL redirect to preserve heap allocation type (#445)

Propagate query args during URL redirect to preserve heap allocation type

internal/driver/webui.go

@@ -179,11 +179,18 @@ func defaultWebServer(args *plugin.HTTPServerArgs) error {
 	// https://github.com/google/pprof/pull/348
 	mux := http.NewServeMux()
 	mux.Handle("/ui/", http.StripPrefix("/ui", handler))
-	mux.Handle("/", http.RedirectHandler("/ui/", http.StatusTemporaryRedirect))
+	mux.Handle("/", redirectWithQuery("/ui"))
 	s := &http.Server{Handler: mux}
 	return s.Serve(ln)
 }
 
+func redirectWithQuery(path string) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		pathWithQuery := &gourl.URL{Path: path, RawQuery: r.URL.RawQuery}
+		http.Redirect(w, r, pathWithQuery.String(), http.StatusTemporaryRedirect)
+	}
+}
+
 func isLocalhost(host string) bool {
 	for _, v := range []string{"localhost", "127.0.0.1", "[::1]", "::1"} {
 		if host == v {