refactor authenticators

transport/internet/conn_authenticator.go

@@ -1,28 +0,0 @@
-package internet
-
-import (
-	"errors"
-	"net"
-
-	"context"
-
-	"v2ray.com/core/common"
-)
-
-type ConnectionAuthenticator interface {
-	Client(net.Conn) net.Conn
-	Server(net.Conn) net.Conn
-}
-
-func CreateConnectionAuthenticator(config interface{}) (ConnectionAuthenticator, error) {
-	auth, err := common.CreateObject(context.Background(), config)
-	if err != nil {
-		return nil, err
-	}
-	switch a := auth.(type) {
-	case ConnectionAuthenticator:
-		return a, nil
-	default:
-		return nil, errors.New("Internet: Not a ConnectionAuthenticator.")
-	}
-}

transport/internet/header.go

@@ -3,6 +3,7 @@ package internet
 import (
 	"context"
 	"errors"
+	"net"
 
 	"v2ray.com/core/common"
 )
@@ -17,10 +18,24 @@ func CreatePacketHeader(config interface{}) (PacketHeader, error) {
 	if err != nil {
 		return nil, err
 	}
-	switch h := header.(type) {
+	if h, ok := header.(PacketHeader); ok {
-	case PacketHeader:
 		return h, nil
-	default:
+	}
 	return nil, errors.New("Internet: Not a packet header.")
 }
+
+type ConnectionAuthenticator interface {
+	Client(net.Conn) net.Conn
+	Server(net.Conn) net.Conn
+}
+
+func CreateConnectionAuthenticator(config interface{}) (ConnectionAuthenticator, error) {
+	auth, err := common.CreateObject(context.Background(), config)
+	if err != nil {
+		return nil, err
+	}
+	if a, ok := auth.(ConnectionAuthenticator); ok {
+		return a, nil
+	}
+	return nil, errors.New("Internet: Not a ConnectionAuthenticator.")
 }