From fd4a6bc9794c3138b322753c8ce1b628c7b3e86b Mon Sep 17 00:00:00 2001
From: zhangzengfei <zhangzengfei@iotlink.com>
Date: 星期四, 21 十一月 2019 18:27:33 +0800
Subject: [PATCH] fix: add prefix on create key file
---
README.md | 4 ++--
rsa.go | 6 +++---
2 files changed, 5 insertions(+), 5 deletions(-)
diff --git a/README.md b/README.md
index 4533534..e46bd00 100644
--- a/README.md
+++ b/README.md
@@ -8,9 +8,9 @@
)
func main() {
- // 鐢熸垚绉橀挜鍜屽叕閽�
+ // 鐢熸垚绉橀挜鍜屽叕閽�, 鍙傛暟: 鏂囦欢鍓嶇紑
if len(os.Args) > 1 && os.Args[1] == "-g" {
- licence.GetRsaKey()
+ licence.GetRsaKey("test")
return
}
diff --git a/rsa.go b/rsa.go
index 9391146..433e017 100644
--- a/rsa.go
+++ b/rsa.go
@@ -28,13 +28,13 @@
prikey *rsa.PrivateKey //绉侀挜
}
-func GetRsaKey() error {
+func GetRsaKey(prefix string) error {
privateKey, err := rsa.GenerateKey(rand.Reader, 4096)
if err != nil {
return err
}
x509PrivateKey := x509.MarshalPKCS1PrivateKey(privateKey)
- privateFile, err := os.Create(privateFileName)
+ privateFile, err := os.Create(prefix + "-" + privateFileName)
if err != nil {
return err
}
@@ -52,7 +52,7 @@
if err != nil {
panic(err)
}
- publicFile, _ := os.Create(publicFileName)
+ publicFile, _ := os.Create(prefix + "-" + publicFileName)
defer publicFile.Close()
publicBlock := pem.Block{
Type: publicKeyPrefix,
--
Gitblit v1.8.0