From faf0718678f9497f14982b92f68f53ccf16b50a6 Mon Sep 17 00:00:00 2001
From: SunBK201 <sunbk201gm@gmail.com>
Date: Mon, 4 Dec 2023 01:08:44 +0800
Subject: [PATCH] feat: support custom bind addr

---
 main.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/main.go b/main.go
index e970ed7..135edda 100644
--- a/main.go
+++ b/main.go
@@ -15,7 +15,9 @@ var payloadByte []byte
 
 func main() {
 	var payload string
+	var addr string
 	var port int
+	flag.StringVar(&addr, "b", "127.0.0.1", "bind address (default: 127.0.0.1)")
 	flag.IntVar(&port, "p", 1080, "port")
 	flag.StringVar(&payload, "f", "FFF", "User-Agent")
 	flag.Parse()
@@ -32,12 +34,12 @@ func main() {
 
 	payloadByte = []byte(payload)
 
-	server, err := net.Listen("tcp", fmt.Sprintf("127.0.0.1:%d", port))
+	server, err := net.Listen("tcp", fmt.Sprintf("%s:%d", addr, port))
 	if err != nil {
 		printAndLog(fmt.Sprintf("Listen failed: %v", err), logger, syslog.LOG_ERR)
 		return
 	}
-	printAndLog(fmt.Sprintf("Listen on 127.0.0.1:%d", port), logger, syslog.LOG_INFO)
+	printAndLog(fmt.Sprintf("Listen on %s:%d", addr, port), logger, syslog.LOG_INFO)
 	for {
 		client, err := server.Accept()
 		if err != nil {