dh 算法

Author: hwholiday

encryption_algorithm/curve25519.go renamed to encryption_algorithm/curve25519_test.go

@@ -7,9 +7,10 @@ import (
 	"golang.org/x/crypto/curve25519"
 	"io"
 	"os"
+	"testing"
 )
 
-func main() {
+func Test_curve25519(t *testing.T) {
 
 	var Aprivate, Apublic [32]byte
 	//产生随机数
@@ -26,8 +27,8 @@ func main() {
 		os.Exit(0)
 	}
 	curve25519.ScalarBaseMult(&Bpublic, &Bprivate)
-	fmt.Println("B私钥",  base64.StdEncoding.EncodeToString(Bprivate[:]))
-	fmt.Println("B公钥",  base64.StdEncoding.EncodeToString(Bpublic[:])) //作为椭圆起点
+	fmt.Println("B私钥", base64.StdEncoding.EncodeToString(Bprivate[:]))
+	fmt.Println("B公钥", base64.StdEncoding.EncodeToString(Bpublic[:])) //作为椭圆起点
 
 	var Akey, Bkey [32]byte
 
@@ -37,6 +38,6 @@ func main() {
 	//B的私钥加上A的公钥计算B的key
 	curve25519.ScalarMult(&Bkey, &Bprivate, &Apublic)
 
-	fmt.Println("A交互的KEY",  base64.StdEncoding.EncodeToString(Akey[:]))
-	fmt.Println("B交互的KEY",  base64.StdEncoding.EncodeToString(Bkey[:]))
+	fmt.Println("A交互的KEY", base64.StdEncoding.EncodeToString(Akey[:]))
+	fmt.Println("B交互的KEY", base64.StdEncoding.EncodeToString(Bkey[:]))
 }

encryption_algorithm/dh_test.go

@@ -0,0 +1,25 @@
+package main
+
+import (
+	"fmt"
+	"math/big"
+	"testing"
+)
+
+func TestDh(t *testing.T) {
+	aP := big.NewInt(509)               //素数
+	aG := big.NewInt(5)                 //底数g，任选
+	aS := big.NewInt(123)               //随机数
+	AA := big.NewInt(0).Exp(aG, aS, aP) //// aG^aS mod aP
+	fmt.Println(AA)
+	//发给B aG，AA,aP
+	bS := big.NewInt(456) //随机数
+	BB := big.NewInt(0).Exp(aG, bS, aP)
+	fmt.Println(BB)
+	BK := big.NewInt(0).Exp(AA, bS, aP)
+	fmt.Println("密钥", BK)
+	//发给A BB
+	AK := big.NewInt(0).Exp(BB, aS, aP)
+	fmt.Println("密钥", AK)
+
+}

encryption_algorithm/x3curve25519.go renamed to encryption_algorithm/x3curve25519_test.go

@@ -11,6 +11,7 @@ import (
 	"hash"
 	"io"
 	"os"
+	"testing"
 )
 
 var PseronA_KDF_Prefix []byte
@@ -50,7 +51,7 @@ type PseronA struct {
 }
 
 //x3dh
-func main() {
+func Test_x3curve25519(t *testing.T) {
 	var a PseronA
 	a.IdentityPri, a.IdentityPub = GetCurve25519KeypPair()
 	a.SignedPri, a.SignedPub = GetCurve25519KeypPair()
@@ -80,15 +81,15 @@ func main() {
 
 	var bKey = bytes.Join([][]byte{b.DH1[:], b.DH2[:], b.DH3[:], b.DH4[:]}, []byte{})
 
-	fmt.Println("ADH1",  base64.StdEncoding.EncodeToString(a.DH1[:]))
-	fmt.Println("ADH2",  base64.StdEncoding.EncodeToString(a.DH2[:]))
-	fmt.Println("ADH3",  base64.StdEncoding.EncodeToString(a.DH3[:]))
-	fmt.Println("ADH4",  base64.StdEncoding.EncodeToString(a.DH4[:]))
+	fmt.Println("ADH1", base64.StdEncoding.EncodeToString(a.DH1[:]))
+	fmt.Println("ADH2", base64.StdEncoding.EncodeToString(a.DH2[:]))
+	fmt.Println("ADH3", base64.StdEncoding.EncodeToString(a.DH3[:]))
+	fmt.Println("ADH4", base64.StdEncoding.EncodeToString(a.DH4[:]))
 
-	fmt.Println("BDH1",  base64.StdEncoding.EncodeToString(b.DH1[:]))
-	fmt.Println("BDH2",  base64.StdEncoding.EncodeToString(b.DH2[:]))
-	fmt.Println("BDH3",  base64.StdEncoding.EncodeToString(b.DH3[:]))
-	fmt.Println("BDH4",  base64.StdEncoding.EncodeToString(b.DH4[:]))
+	fmt.Println("BDH1", base64.StdEncoding.EncodeToString(b.DH1[:]))
+	fmt.Println("BDH2", base64.StdEncoding.EncodeToString(b.DH2[:]))
+	fmt.Println("BDH3", base64.StdEncoding.EncodeToString(b.DH3[:]))
+	fmt.Println("BDH4", base64.StdEncoding.EncodeToString(b.DH4[:]))
 
 	fmt.Println("aKey", base64.StdEncoding.EncodeToString(aKey))
 	fmt.Println("aKey", base64.StdEncoding.EncodeToString(kdf(aKey)))
@@ -100,12 +101,12 @@ func main() {
 
 	for i := 1; i <= 3; i++ {
 		aSalt := GetCurve25519Key(a.EphemeralPri, b.EphemeralPub)
-		fmt.Println("计算Ａ的salt第 ",i," 次",  base64.StdEncoding.EncodeToString(aSalt[:]))
-		fmt.Println("计算A的KEY第 ",i," 次", base64.StdEncoding.EncodeToString(Signalkdf(aKey, aSalt, "A")))
+		fmt.Println("计算Ａ的salt第 ", i, " 次", base64.StdEncoding.EncodeToString(aSalt[:]))
+		fmt.Println("计算A的KEY第 ", i, " 次", base64.StdEncoding.EncodeToString(Signalkdf(aKey, aSalt, "A")))
 
 		bSalt := GetCurve25519Key(b.EphemeralPri, a.EphemeralPub)
-		fmt.Println("计算B的salt第 ",i," 次",  base64.StdEncoding.EncodeToString(bSalt[:]))
-		fmt.Println("计算B的KEY第 ",i," 次", base64.StdEncoding.EncodeToString(Signalkdf(bKey, bSalt, "B")))
+		fmt.Println("计算B的salt第 ", i, " 次", base64.StdEncoding.EncodeToString(bSalt[:]))
+		fmt.Println("计算B的KEY第 ", i, " 次", base64.StdEncoding.EncodeToString(Signalkdf(bKey, bSalt, "B")))
 	}
 
 }