Merge pull request #19 from riklus/master

extremecoders-re · web-flow · commit 0f70e539adeb · 2022-04-03T12:27:10.000+05:30
Handling connection error #18
diff --git a/main.go b/main.go
@@ -250,7 +250,11 @@ func main() {
 
 	mutex = &sync.Mutex{}
 	for {
-		conn, _ := l.Accept()
-		go handle_connection(conn, *tunnel)
+		conn, err := l.Accept()
+		if err != nil {
+			log.Println("[WARN] Could not accept connection")
+		} else {
+			go handle_connection(conn, *tunnel)
+		}
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -250,7 +250,11 @@ func main() {`
`250`	`250`
`251`	`251`	`mutex = &sync.Mutex{}`
`252`	`252`	`for {`
`253`		`- conn, _ := l.Accept()`
`254`		`- go handle_connection(conn, *tunnel)`
	`253`	`+ conn, err := l.Accept()`
	`254`	`+ if err != nil {`
	`255`	`+ log.Println("[WARN] Could not accept connection")`
	`256`	`+ } else {`
	`257`	`+ go handle_connection(conn, *tunnel)`
	`258`	`+ }`
`255`	`259`	`}`
`256`	`260`	`}`