add aes-128 aes-192 cipher

xtaci · Aug 16, 2016 · a04c959 · a04c959
1 parent 4a79d26
commit a04c959
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/client/main.go b/client/main.go
@@ -115,7 +115,7 @@ func main() {
 		cli.StringFlag{
 			Name:  "crypt",
 			Value: "aes",
-			Usage: "methods for encryption: aes, tea, xor, none",
+			Usage: "methods for encryption: aes, aes-128, aes-192, tea, xor, none",
 		},
 		cli.StringFlag{
 			Name:  "mode",
@@ -228,6 +228,10 @@ func main() {
 			block, _ = kcp.NewSimpleXORBlockCrypt(pass)
 		case "none":
 			block, _ = kcp.NewNoneBlockCrypt(pass)
+		case "aes-128":
+			block, _ = kcp.NewAESBlockCrypt(pass[:16])
+		case "aes-192":
+			block, _ = kcp.NewAESBlockCrypt(pass[:24])
 		default:
 			block, _ = kcp.NewAESBlockCrypt(pass)
 		}

diff --git a/server/main.go b/server/main.go
@@ -133,7 +133,7 @@ func main() {
 		cli.StringFlag{
 			Name:  "crypt",
 			Value: "aes",
-			Usage: "methods for encryption: aes, tea, xor, none",
+			Usage: "methods for encryption: aes, aes-128, aes-192, tea, xor, none",
 		},
 		cli.StringFlag{
 			Name:  "mode",
@@ -234,6 +234,10 @@ func main() {
 			block, _ = kcp.NewSimpleXORBlockCrypt(pass)
 		case "none":
 			block, _ = kcp.NewNoneBlockCrypt(pass)
+		case "aes-128":
+			block, _ = kcp.NewAESBlockCrypt(pass[:16])
+		case "aes-192":
+			block, _ = kcp.NewAESBlockCrypt(pass[:24])
 		default:
 			block, _ = kcp.NewAESBlockCrypt(pass)
 		}