output endpoint address when stream opened

xiyan130227 · Jun 11, 2019 · 45e58e6 · 45e58e6
1 parent 7382d41
commit 45e58e6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/client/main.go b/client/main.go
@@ -43,8 +43,8 @@ func handleClient(sess *smux.Session, p1 io.ReadWriteCloser, quiet bool) {
 
 	defer p2.Close()
 
-	logln("stream opened", p2.ID())
-	defer logln("stream closed", p2.ID())
+	logln("stream opened", p2.ID(), "@", p2.RemoteAddr())
+	defer logln("stream closed", p2.ID(), "@", p2.RemoteAddr())
 
 	// start tunnel & wait for tunnel termination
 	streamCopy := func(dst io.Writer, src io.ReadCloser) chan struct{} {

diff --git a/server/main.go b/server/main.go
@@ -71,8 +71,8 @@ func handleClient(p1 *smux.Stream, p2 io.ReadWriteCloser, quiet bool) {
 	defer p1.Close()
 	defer p2.Close()
 
-	logln("stream opened", p1.ID())
-	defer logln("stream closed", p1.ID())
+	logln("stream opened", p1.ID(), "@", p1.RemoteAddr())
+	defer logln("stream closed", p1.ID(), "@", p1.RemoteAddr())
 
 	// start tunnel & wait for tunnel termination
 	streamCopy := func(dst io.Writer, src io.ReadCloser) chan struct{} {