bumped to 1.10.1, added pprof, updated mshark to 0.0.14

Author: shadowy-pycoder

README.md

@@ -106,7 +106,7 @@ You can download the binary for your platform from [Releases](https://github.com
 Example:
 
 ```shell
-GOHPTS_RELEASE=v1.10.0; wget -v https://github.com/shadowy-pycoder/go-http-proxy-to-socks/releases/download/$GOHPTS_RELEASE/gohpts-$GOHPTS_RELEASE-linux-amd64.tar.gz -O gohpts && tar xvzf gohpts && mv -f gohpts-$GOHPTS_RELEASE-linux-amd64 gohpts && ./gohpts -h
+GOHPTS_RELEASE=v1.10.1; wget -v https://github.com/shadowy-pycoder/go-http-proxy-to-socks/releases/download/$GOHPTS_RELEASE/gohpts-$GOHPTS_RELEASE-linux-amd64.tar.gz -O gohpts && tar xvzf gohpts && mv -f gohpts-$GOHPTS_RELEASE-linux-amd64 gohpts && ./gohpts -h
 ```
 
 Alternatively, you can install it using `go install` command (requires Go [1.24](https://go.dev/doc/install) or later):
@@ -164,6 +164,7 @@ Options:
   -j        Show logs in JSON format
   -logfile  Log file path (Default: stdout)
   -nocolor  Disable colored output for logs (no effect if -j flag specified)
+  -pprof    Address of pprof server with profiling data
 
   Sniffing:
   -sniff    Enable traffic sniffing for HTTP and TLS

cmd/gohpts/cli.go

@@ -51,6 +51,7 @@ Options:
   -j        Show logs in JSON format
   -logfile  Log file path (Default: stdout)
   -nocolor  Disable colored output for logs (no effect if -j flag specified)
+  -pprof    Address of pprof server with profiling data
 
   Sniffing:
   -sniff    Enable traffic sniffing for HTTP and TLS
@@ -140,6 +141,7 @@ func root(args []string) error {
 	flags.BoolVar(&conf.JSON, "j", false, "Show logs in JSON format")
 	flags.BoolVar(&conf.Sniff, "sniff", false, "Enable traffic sniffing for HTTP and TLS")
 	flags.StringVar(&conf.SniffLogFile, "snifflog", "", "Sniffed traffic log file path (Default: the same as -logfile)")
+	flags.StringVar(&conf.AddrPprof, "pprof", "", "Address of pprof server with profiling data")
 	flags.BoolVar(&conf.NoColor, "nocolor", false, "Disable colored output for logs (no effect if -j flag specified)")
 	flags.BoolVar(&conf.Body, "body", false, "Collect request and response body for HTTP traffic (credentials, tokens, etc)")
 	flags.BoolFunc("v", "Show version and build information", func(flagValue string) error {

go.mod

@@ -7,7 +7,7 @@ require (
 	github.com/google/uuid v1.6.0
 	github.com/rs/zerolog v1.34.0
 	github.com/shadowy-pycoder/colors v0.0.1
-	github.com/shadowy-pycoder/mshark v0.0.13
+	github.com/shadowy-pycoder/mshark v0.0.14
 	github.com/wzshiming/socks5 v0.5.2
 	golang.org/x/sys v0.33.0
 	golang.org/x/term v0.32.0

go.sum

@@ -34,8 +34,8 @@ github.com/rs/zerolog v1.34.0 h1:k43nTLIwcTVQAncfCw4KZ2VY6ukYoZaBPNOE8txlOeY=
 github.com/rs/zerolog v1.34.0/go.mod h1:bJsvje4Z08ROH4Nhs5iH600c3IkWhwp44iRc54W6wYQ=
 github.com/shadowy-pycoder/colors v0.0.1 h1:weCj/YIOupqy4BSP8KuVzr20fC+cuAv/tArz7bhhkP4=
 github.com/shadowy-pycoder/colors v0.0.1/go.mod h1:lkrJS1PY2oVigNLTT6pkbF7B/v0YcU2LD5PZnss1Q4U=
-github.com/shadowy-pycoder/mshark v0.0.13 h1:ROEuey/Th4YAmfRg8Xc17aboMs5fknQho4mNBC9h+KE=
-github.com/shadowy-pycoder/mshark v0.0.13/go.mod h1:FqbHFdsx0zMnrZZH0+oPzaFcleP4O+tUWv8i5gxo87k=
+github.com/shadowy-pycoder/mshark v0.0.14 h1:Gou+y9tIEjNSztGWzfFSYJJ8G0DDNYmkvKTQUZDYeVQ=
+github.com/shadowy-pycoder/mshark v0.0.14/go.mod h1:FqbHFdsx0zMnrZZH0+oPzaFcleP4O+tUWv8i5gxo87k=
 github.com/stretchr/testify v1.9.0 h1:HtqpIVDClZ4nwg75+f6Lvsy/wHu+3BoSGCbBAcpTsTg=
 github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY=
 github.com/wzshiming/socks5 v0.5.2 h1:LtoowVNwAmkIQSkP1r1Wg435xUmC+tfRxorNW30KtnM=

gohpts.go

@@ -17,6 +17,7 @@ import (
 	"math/rand"
 	"net"
 	"net/http"
+	"net/http/pprof"
 	"os"
 	"os/exec"
 	"os/signal"
@@ -57,6 +58,7 @@ var (
 type Config struct {
 	AddrHTTP       string
 	AddrSOCKS      string
+	AddrPprof      string
 	User           string
 	Pass           string
 	ServerUser     string
@@ -136,6 +138,7 @@ type proxyapp struct {
 	certFile       string
 	keyFile        string
 	httpServerAddr string
+	pprofAddr      string
 	iface          *net.Interface
 	tproxyAddr     string
 	tproxyAddrUDP  string
@@ -244,6 +247,12 @@ func New(conf *Config) *proxyapp {
 		lvl = zerolog.DebugLevel
 	}
 	p.debug = conf.Debug
+	if conf.AddrPprof != "" {
+		p.pprofAddr, err = getFullAddress(conf.AddrPprof, "", false)
+		if err != nil {
+			p.logger.Fatal().Err(err).Msg("")
+		}
+	}
 	// the only way I found to make debug level independent between loggers
 	l := logger.Level(lvl)
 	sl := snifflogger.Level(lvl)
@@ -504,6 +513,9 @@ func New(conf *Config) *proxyapp {
 	if p.tproxyAddrUDP != "" {
 		p.logger.Info().Msgf("TPROXY (UDP): %s", p.tproxyAddrUDP)
 	}
+	if p.pprofAddr != "" {
+		p.logger.Info().Msgf("PPROF: %s", p.pprofAddr)
+	}
 	return &p
 }
 
@@ -512,6 +524,15 @@ func (p *proxyapp) Run() {
 	quit := make(chan os.Signal, 1)
 	p.closeConn = make(chan bool)
 	signal.Notify(quit, os.Interrupt)
+	if p.pprofAddr != "" {
+		sm := http.NewServeMux()
+		sm.HandleFunc("/debug/pprof/", pprof.Index)
+		sm.HandleFunc("/debug/pprof/cmdline", pprof.Cmdline)
+		sm.HandleFunc("/debug/pprof/profile", pprof.Profile)
+		sm.HandleFunc("/debug/pprof/symbol", pprof.Symbol)
+		sm.HandleFunc("/debug/pprof/trace", pprof.Trace)
+		go http.ListenAndServe(p.pprofAddr, sm)
+	}
 	if p.arpspoofer != nil {
 		go p.arpspoofer.Start()
 	}

version.go

@@ -1,3 +1,3 @@
 package gohpts
 
-const Version string = "gohpts v1.10.0"
+const Version string = "gohpts v1.10.1"