重构 rsa 代码

Author: FishGoddess

_examples/rsa.go

@@ -7,52 +7,59 @@ package main
 import (
 	"fmt"
 
+	"github.com/FishGoddess/cryptox/bytes/encoding"
 	"github.com/FishGoddess/cryptox/hash"
 	"github.com/FishGoddess/cryptox/rsa"
 )
 
-var (
-	// Use public key to encrypt msg.
-	publicKey = rsa.MustLoadPublicKey("rsa.pub")
+func main() {
+	// Load the private key and the public key from file.
+	privateKey, err := rsa.LoadPrivateKey("rsa.key")
+	if err != nil {
+		panic(err)
+	}
 
-	// Use private key to decrypt msg.
-	privateKey = rsa.MustLoadPrivateKey("rsa.key")
-)
+	publicKey, err := rsa.LoadPublicKey("rsa.pub")
+	if err != nil {
+		panic(err)
+	}
 
-func main() {
 	msg := []byte("戴上头箍，爱不了你；不戴头箍，救不了你。")
-	fmt.Printf("Msg: %s\n", msg)
+	fmt.Printf("msg: %s\n", msg)
 
-	// Use public key to encrypt msg.
-	encrypted, err := publicKey.EncryptPKCS1v15(msg)
+	// Use the public key to encrypt msg using base64 encoding.
+	label := []byte("你好，世界")
+
+	encrypt, err := publicKey.EncryptOAEP(msg, label, encoding.Base64)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Printf("Encrypted: %s\n", encrypted.Base64())
+	fmt.Printf("encrypt: %s\n", encrypt)
 
-	// Use private key to decrypt msg.
-	decrypted, err := privateKey.DecryptPKCS1v15(encrypted)
+	// Use the private key to decrypt msg using base64 encoding.
+	decrypt, err := privateKey.DecryptOAEP(encrypt, label, encoding.Base64)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Printf("Decrypted: %s\n", decrypted)
+	fmt.Printf("decrypt: %s\n", decrypt)
+
+	// Use the private key to sign msg.
+	digest := hash.SHA256(msg, encoding.Hex)
 
-	// Use private key to sign msg.
-	msg = hash.SHA256(msg)
-	signed, err := privateKey.SignPKCS1v15(msg)
+	sign, err := privateKey.SignPSS(digest, 0, encoding.Hex)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Printf("Signed: %s\n", signed)
+	fmt.Printf("sign: %s\n", sign)
 
-	// Use public key to verify msg.
-	err = publicKey.VerifyPKCS1v15(msg, signed)
+	// Use the public key to verify the sign.
+	err = publicKey.VerifyPSS(digest, sign, 0, encoding.Hex)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Println("Verified.")
+	fmt.Println("verify: %s\n", digest)
 }

_examples/rsa.key

@@ -0,0 +1,28 @@
+-----BEGIN PRIVATE KEY-----
+MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQCnoCnYqtdcxo2Z
+zcrpkfHYg+s/gGrlt7Ww//gBJfcCZ64ls4fD4I0BD+Xe0WHuT5poow6Cyyl20yrv
+Y13oqlRr05bsnPpu2ZOiEPfiAQb898xgYYwoDwLEk75Uu1ox53unhhZbRKNne1V1
+QMYFpxAgeAMxML09ZmEqPp3q0BNZcjZqaq/2umHVOBOrqrRJZG7APZpixAvwbRKJ
+zELq40s0zD6ECovZk/3WVF8BEcrmybmnbns1Y89ptXFZIQVKPqHXUK9+VbIiKZsU
++UKpY29WKjOkELcSrG7HMZbIXivzswGn5c0DskXQ/TbzV5WNxpW07Op0s3y9G7n+
+kTcytZETAgMBAAECggEACpRE/9aBwr+0lnU9WLIW0Y47LdCk2ScChw9omhNI0cnf
+B6RwgvAW8MRi7XCeiHlVVURGjqz+ysU3YPtkT7esSKUxJxwd1sV8QqQP2PTX+ZA0
+Fw3LrYg2VFdqEZPvdICCYTIeaBfeuOcOSi5Shm8i8+xPG0QoYAnS7mUcTK0MLj65
+FYcb1p7rteeqpQwKqekaAEqOo084Nbs10++ltct3wM4xOvEFJubdMUj6kruOVwG5
+onuyONTWQ9U1akSRwlx1HnafieTG/Rsvx+6FT4wAgioZy3JKlWsLckI0K5zi/YSz
+Z0ZeqdtGLI7JbYIKv+H+zeLv0mPErwOXJk0qm3b6eQKBgQDJVMYzwB1BIVU545dz
+8+YX69WfFVjH9RSJ+9monq6fu0QW6s2qH4O8DCKqG5fuSAXPVzYvVrEDV4KGsyPG
+m0uQteM5om7ICIQVvewxBiaDX0bvFwiK7gfzdLVMgYirrkxZUQ4NgGTRScAP2ysm
+sOn52Om5ZK5JZ3v8C/aQU/D5CQKBgQDVJGcDV7XlOVMptAVICEP4G+jg2fS9ORlz
+TkgpK8sw1l1P6eO9WU2TrOh8UCQFntVe5YwgzQuBK4PJACsbHqx1Xww57MI1dlIQ
+p2vP6dFmVlneIrcHXZ4QRQug6Envp03Smnbzf9clDCAtZkt3gGDc2r0yxn8AY6rE
++q6mITHMOwKBgQC7ernmzutvDv8yHQGX9HM7q10N+u7lpQ8vPtt87edmzxekz5oc
+5aPipNpS1ccxGNhwL6JBitTja8YccQzLkSlY5EdoEB5hH60AIg+jxzpt83c2hZhq
+5yV4TCHX0HfYh0KJmbUgVYOMcMTs/wa7zNrU0m0zOtIhgMAwAWPlGoW3IQKBgHSs
+l6NRySVwiuCiRd3XgHV5ubIUPY+ziQYAjSnUakcSoUPUkbEeCIRVO3KJYB6fgseO
+unVeKPUNf/dwmygeU2Nwoz22J92iJmwtaawHn3P4wvsBX9WtXpAja6kqXwbMO6KU
+oZbLnVcPWzHe9GK3KM7dAoKf+/eXl2x6mU4hj6PvAoGBALKTzLcqAr7n+TqcRAJU
+nm44K4zj52Nj0lpritovFbP1EiVoj07AFqFULE3aHMeGKwe+aNxz5JTSHPaP6aa9
+iD4CVoJzQ41OimqHnnRSgy3g+ylk7plLk/M0rE+6Ev945Xv8vGOGci6KgkBpOx37
+/yglpQbyju+BbRvq9gDfLuKX
+-----END PRIVATE KEY-----

_examples/rsa.pub

@@ -0,0 +1,9 @@
+-----BEGIN PUBLIC KEY-----
+MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAp6Ap2KrXXMaNmc3K6ZHx
+2IPrP4Bq5be1sP/4ASX3AmeuJbOHw+CNAQ/l3tFh7k+aaKMOgsspdtMq72Nd6KpU
+a9OW7Jz6btmTohD34gEG/PfMYGGMKA8CxJO+VLtaMed7p4YWW0SjZ3tVdUDGBacQ
+IHgDMTC9PWZhKj6d6tATWXI2amqv9rph1TgTq6q0SWRuwD2aYsQL8G0SicxC6uNL
+NMw+hAqL2ZP91lRfARHK5sm5p257NWPPabVxWSEFSj6h11CvflWyIimbFPlCqWNv
+ViozpBC3EqxuxzGWyF4r87MBp+XNA7JF0P0281eVjcaVtOzqdLN8vRu5/pE3MrWR
+EwIDAQAB
+-----END PUBLIC KEY-----

_examples/rsa_key.go

@@ -5,60 +5,35 @@
 package main
 
 import (
-	"fmt"
-
 	"github.com/FishGoddess/cryptox/rsa"
 )
 
 func main() {
-	// Generate a 4096 bits key.
-	// Check rsa.KeyOption for more information of encoder and decoder.
-	privateKey, publicKey, err := rsa.GenerateKeys(4096)
+	// Generate a 2048 bits key.
+	privateKey, publicKey, err := rsa.GenerateKeys(2048)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Println(privateKey)
-	fmt.Println(publicKey)
-
-	// Try WriteToFile if you want to write your private key to file.
-	n, err := privateKey.Bytes().WriteToFile("rsa.key")
+	// Store the private key and the public key to file.
+	err = rsa.StorePrivateKey("rsa.key", privateKey)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Printf("Write %d bytes to private key file\n", n)
-
-	// Try WriteToFile if you want to write your public key to file.
-	n, err = publicKey.Bytes().WriteToFile("rsa.pub")
+	err = rsa.StorePublicKey("rsa.pub", publicKey)
 	if err != nil {
 		panic(err)
 	}
 
-	fmt.Printf("Write %d bytes to public key file\n", n)
-
-	// Load private key from file.
-	loadedPrivateKey, err := rsa.LoadPrivateKey("rsa.key")
+	// Load the private key and the public key from file.
+	privateKey, err = rsa.LoadPrivateKey("rsa.key")
 	if err != nil {
 		panic(err)
 	}
 
-	// Load public key from file.
-	loadedPublicKey, err := rsa.LoadPublicKey("rsa.pub")
+	publicKey, err = rsa.LoadPublicKey("rsa.pub")
 	if err != nil {
 		panic(err)
 	}
-
-	fmt.Println(loadedPrivateKey)
-	fmt.Println(loadedPublicKey)
-
-	// Want to load keys from file and panic if failed?
-	// Try these:
-	loadedPrivateKey = rsa.MustLoadPrivateKey("rsa.key")
-	loadedPublicKey = rsa.MustLoadPublicKey("rsa.pub")
-
-	// Already have a private or public key in bytes?
-	// Try these:
-	_, _ = rsa.ParsePrivateKey(privateKey.Bytes())
-	_, _ = rsa.ParsePublicKey(publicKey.Bytes())
 }