Changeset 713 in code for trunk/cmd

Timestamp:

Nov 17, 2021, 3:18:52 PM (4 years ago)

Author:

contact

Message:

Add pprof HTTP server

This enables production debugging of the bouncer.

Closes: ​https://todo.sr.ht/~emersion/soju/155

File:

• trunk/cmd/soju/main.go (modified) (2 diffs)

trunk/cmd/soju/main.go

@@ -10 +10 @@
         "net"
         "net/http"
+        _ "net/http/pprof"
         "net/url"
         "os"
@@ -287 +288 @@
                                 }
                         }()
+                case "http+pprof":
+                        // Only allow localhost as listening host for security reasons.
+                        // Users can always explicitly setup reverse proxies if desirable.
+                        hostname, _, err := net.SplitHostPort(u.Host)
+                        if err != nil {
+                                log.Fatalf("invalid host in URI %q: %v", listen, err)
+                        } else if hostname != "localhost" {
+                                log.Fatalf("pprof listening host must be localhost")
+                        }
+
+                        // net/http/pprof registers its handlers in http.DefaultServeMux
+                        httpSrv := http.Server{
+                                Addr:    u.Host,
+                                Handler: http.DefaultServeMux,
+                        }
+                        go func() {
+                                if err := httpSrv.ListenAndServe(); err != nil {
+                                        log.Fatalf("serving %q: %v", listen, err)
+                                }
+                        }()
                 default:
                         log.Fatalf("failed to listen on %q: unsupported scheme", listen)