simplify dial/listen

client/dial.go

@@ -1,11 +1,18 @@
-// +build !linux
-
 package main
 
 import (
+	"github.com/pkg/errors"
 	kcp "github.com/xtaci/kcp-go"
+	"github.com/xtaci/tcpraw"
 )
 
 func dial(config *Config, block kcp.BlockCrypt) (*kcp.UDPSession, error) {
+	if config.TCP {
+		conn, err := tcpraw.Dial("tcp", config.RemoteAddr)
+		if err != nil {
+			return nil, errors.Wrap(err, "tcpraw.Dial()")
+		}
+		return kcp.NewConn(config.RemoteAddr, block, config.DataShard, config.ParityShard, conn)
+	}
 	return kcp.DialWithOptions(config.RemoteAddr, block, config.DataShard, config.ParityShard)
 }

client/main.go

@@ -223,7 +223,7 @@ func main() {
 		},
 		cli.BoolFlag{
 			Name:  "tcp",
-			Usage: "to emulate a TCP connection",
+			Usage: "to emulate a TCP connection(linux)",
 		},
 		cli.StringFlag{
 			Name:  "c",

server/listen.go

@@ -1,9 +1,18 @@
-// +build !linux
-
 package main
 
-import kcp "github.com/xtaci/kcp-go"
+import (
+	"github.com/pkg/errors"
+	kcp "github.com/xtaci/kcp-go"
+	"github.com/xtaci/tcpraw"
+)
 
 func listen(config *Config, block kcp.BlockCrypt) (*kcp.Listener, error) {
+	if config.TCP {
+		conn, err := tcpraw.Listen("tcp", config.Listen)
+		if err != nil {
+			return nil, errors.Wrap(err, "tcpraw.Listen()")
+		}
+		return kcp.ServeConn(block, config.DataShard, config.ParityShard, conn)
+	}
 	return kcp.ListenWithOptions(config.Listen, block, config.DataShard, config.ParityShard)
 }

server/main.go

@@ -240,7 +240,7 @@ func main() {
 		},
 		cli.BoolFlag{
 			Name:  "tcp",
-			Usage: "to emulate a TCP connection",
+			Usage: "to emulate a TCP connection(linux)",
 		},
 		cli.StringFlag{
 			Name:  "c",