feat: add pprof endpoint option for profiling

Add --pprof and --pprof-addr flags to enable pprof HTTP endpoint
for runtime profiling and debugging.

Changes:
- Add EnablePprof and PprofAddr to AgentOptions
- Add --pprof flag to enable pprof server (default: false)
- Add --pprof-addr flag to configure listen address (default: localhost:6060)
- Implement startPprofServer() in agent.go

Usage:
  kyanos watch --pprof
  kyanos watch --pprof --pprof-addr=:9090

The pprof endpoint provides access to:
- /debug/pprof/heap - memory profiling
- /debug/pprof/profile - CPU profiling
- /debug/pprof/goroutine - goroutine dumps

Author: hengyouhai

agent/agent.go

@@ -19,6 +19,7 @@ import (
 	"kyanos/version"
 	"os"
 	"os/exec"
+	"net/http"
 	"os/signal"
 	"runtime"
 	"strings"
@@ -34,6 +35,7 @@ import (
 
 func SetupAgent(options ac.AgentOptions) {
 	startGopsServer(options)
+	startPprofServer(options)
 
 	if err := version.UpgradeDetect(); err != nil {
 		if errors.Is(err, version.ErrBehindLatest) {
@@ -252,3 +254,17 @@ func startGopsServer(opts ac.AgentOptions) {
 		}
 	}
 }
+
+func startPprofServer(opts ac.AgentOptions) {
+	if !opts.EnablePprof {
+		return
+	}
+	go func() {
+		addr := opts.GetPprofAddr()
+		common.AgentLog.Infof("Starting pprof server on http://%s/debug/pprof/", addr)
+		common.AgentLog.Info("pprof endpoints: /debug/pprof/heap, /debug/pprof/profile, /debug/pprof/goroutine")
+		if err := http.ListenAndServe(addr, nil); err != nil {
+			common.AgentLog.Errorf("Failed to start pprof server: %v", err)
+		}
+	}()
+}

agent/common/options.go

@@ -50,6 +50,8 @@ type AgentOptions struct {
 	ConntrackCloseWaitTimeMills int
 	MaxAllowStuckTimeMills      int
 	StartGopsServer             bool
+	EnablePprof                 bool
+	PprofAddr                   string
 
 	FilterComm              string
 	ProcessExecEventChannel chan *bpf.AgentProcessExecEvent
@@ -74,6 +76,13 @@ type AgentOptions struct {
 	FirstPacketEventMapPageNum int
 }
 
+func (o AgentOptions) GetPprofAddr() string {
+	if o.PprofAddr == "" {
+		return "localhost:6060"
+	}
+	return o.PprofAddr
+}
+
 func (o AgentOptions) FilterByContainer() bool {
 	return o.ContainerId != "" || o.ContainerName != "" || o.PodName != ""
 }

cmd/root.go

@@ -105,6 +105,8 @@ func init() {
 	rootCmd.PersistentFlags().IntVar(&options.MaxAllowStuckTimeMills, "max-allow-stuck-time-mills", 1000, "--max-allow-stuck-time-mills 100")
 
 	rootCmd.PersistentFlags().BoolVar(&options.StartGopsServer, "gops", false, "start gops server")
+	rootCmd.PersistentFlags().BoolVar(&options.EnablePprof, "pprof", false, "enable pprof HTTP endpoint for profiling")
+	rootCmd.PersistentFlags().StringVar(&options.PprofAddr, "pprof-addr", "localhost:6060", "pprof HTTP server address")
 
 	rootCmd.PersistentFlags().MarkHidden("default-log-level")
 	rootCmd.PersistentFlags().MarkHidden("agent-log-level")