gox/cryptox/rsax/rsax_test.go at main · soyacen/gox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
package rsax

import (
	"crypto/rand"
	"crypto/rsa"
	"crypto/x509"
	"encoding/pem"
	"os"
	"path/filepath"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestSignHex(t *testing.T) {
	privateKey, publicKey, err := GenerateKeyHex(1024)
	assert.NoError(t, err)

	t.Log(privateKey)
	t.Log(publicKey)

	rawData := []byte("he is hello kitty")
	signHex, err := SignWithSha256Hex(rawData, privateKey)
	assert.NoError(t, err)

	err = VerifySignWithSha256Hex(rawData, signHex, publicKey)
	assert.NoError(t, err)
}

func TestSignBase64(t *testing.T) {
	privateKey, publicKey, err := GenerateKeyBase64(1024)
	assert.NoError(t, err)

	t.Log(privateKey)
	t.Log(publicKey)

	rawData := []byte("he is hello kitty")
	signHex, err := SignWithSha256Base64(rawData, privateKey)
	assert.NoError(t, err)

	err = VerifySignWithSha256Base64(rawData, signHex, publicKey)
	assert.NoError(t, err)
}

func TestCrypt(t *testing.T) {
	privateKey, publicKey, err := GenerateKeyHex(1024)
	assert.NoError(t, err)

	rawData := []byte("he is hello kitty")
	encryptDate, err := EncryptToHex(rawData, publicKey)
	assert.NoError(t, err)

	data, err := DecryptByHex(encryptDate, privateKey)
	assert.NoError(t, err)

	assert.Equal(t, rawData, data)
}

func TestLoad(t *testing.T) {
	// 临时目录生成 PEM 私钥文件,避免依赖 /tmp/priv.pem
	privateKey, err := rsa.GenerateKey(rand.Reader, 1024)
	assert.NoError(t, err)

	pemBytes := pem.EncodeToMemory(&pem.Block{
		Type:  "RSA PRIVATE KEY",
		Bytes: x509.MarshalPKCS1PrivateKey(privateKey),
	})

	file := filepath.Join(t.TempDir(), "priv.pem")
	assert.NoError(t, os.WriteFile(file, pemBytes, 0o600))

	priv, pub, err := LoadKeyBase64(file)
	assert.NoError(t, err)
	assert.NotEmpty(t, priv)
	assert.NotEmpty(t, pub)
	t.Log(priv)
	t.Log(pub)

	privHex, pubHex, err := LoadKeyHex(file)
	assert.NoError(t, err)
	assert.NotEmpty(t, privHex)
	assert.NotEmpty(t, pubHex)
}

func TestLoad_MissingFile(t *testing.T) {
	_, _, err := LoadKeyBase64(filepath.Join(t.TempDir(), "does-not-exist.pem"))
	assert.Error(t, err)
}