mirror of
https://github.com/wyx2685/V2bX.git
synced 2025-02-01 22:38:13 -05:00
77 lines
1.7 KiB
Go
77 lines
1.7 KiB
Go
package cmd
|
||
|
||
import (
|
||
"crypto/rand"
|
||
"encoding/base64"
|
||
"fmt"
|
||
"strings"
|
||
|
||
"github.com/Yuzuki616/V2bX/common/crypt"
|
||
|
||
"github.com/spf13/cobra"
|
||
"golang.org/x/crypto/curve25519"
|
||
)
|
||
|
||
var x25519Command = cobra.Command{
|
||
Use: "x25519",
|
||
Short: "Generate key pair for x25519 key exchange",
|
||
Run: func(cmd *cobra.Command, args []string) {
|
||
executeX25519()
|
||
},
|
||
}
|
||
|
||
func init() {
|
||
command.AddCommand(&x25519Command)
|
||
}
|
||
|
||
func executeX25519() {
|
||
var yes, key string
|
||
fmt.Println("要对私钥进行加密吗?(Y/n)")
|
||
fmt.Scan(&yes)
|
||
if strings.ToLower(yes) == "y" {
|
||
var temp string
|
||
fmt.Println("请输入Api接口地址:")
|
||
fmt.Scan(&temp)
|
||
key = temp
|
||
fmt.Println("请输入Api认证Token:")
|
||
fmt.Scan(&temp)
|
||
key += temp
|
||
key = crypt.GenShaHash([]byte(key))
|
||
}
|
||
var output string
|
||
var err error
|
||
defer func() {
|
||
fmt.Println(output)
|
||
}()
|
||
var privateKey []byte
|
||
var publicKey []byte
|
||
privateKey = make([]byte, curve25519.ScalarSize)
|
||
if _, err = rand.Read(privateKey); err != nil {
|
||
output = Err("read rand error: ", err)
|
||
return
|
||
}
|
||
|
||
// Modify random bytes using algorithm described at:
|
||
// https://cr.yp.to/ecdh.html.
|
||
privateKey[0] &= 248
|
||
privateKey[31] &= 127
|
||
privateKey[31] |= 64
|
||
|
||
if publicKey, err = curve25519.X25519(privateKey, curve25519.Basepoint); err != nil {
|
||
output = Err("gen X25519 error: ", err)
|
||
return
|
||
}
|
||
p := base64.RawURLEncoding.EncodeToString(privateKey)
|
||
output = fmt.Sprint("Private key: ",
|
||
p,
|
||
"\nPublic key: ",
|
||
base64.RawURLEncoding.EncodeToString(publicKey))
|
||
if strings.ToLower(yes) == "y" {
|
||
key, err = crypt.AesEncrypt([]byte(p), []byte(key[:32]))
|
||
if err != nil {
|
||
output = Err("encrypt private key error: ", err)
|
||
}
|
||
output += "\n加密后的Private key：" + key
|
||
}
|
||
}
|