因在做license注册机制当中,有对根据本地RSA pem密钥文件判断出RSA的公私密钥长度的需求(即:根据pem内容判断出RSA密钥是1024位或者2048位,或者其他位数等),因此个人通过思考,摸索整理出一个方法,予以记录,备用,分享。
package main
import (
"crypto/rsa"
"crypto/x509"
"encoding/pem"
"errors"
"fmt"
"io/ioutil"
"os"
)
//全局变量
var privateKey, publicKey []byte
func init() {
var err error
publicKey, err = ioutil.ReadFile("public.pem")
if err != nil {
os.Exit(-1)
}
privateKey, err = ioutil.ReadFile("private.pem")
if err != nil {
os.Exit(-1)
}
}
/**
* @brief 获取RSA公钥长度
* @param[in] PubKey RSA公钥
* @return 成功返回 RSA公钥长度,失败返回error 错误信息
*/
func GetPubKeyLen(PubKey []byte) (int, error) {
if PubKey == nil {
return 0, errors.New("input arguments error")
}
block, _ := pem.Decode(PubKey)
if block == nil {
return 0, errors.New("public rsaKey error")
}
pubInterface, err := x509.ParsePKIXPublicKey(block.Bytes)
if err != nil {
return 0, err
}
pub := pubInterface.(*rsa.PublicKey)
return pub.N.BitLen(), nil
}
/**
* @brief 获取RSA私钥长度
* @param[in] PriKey RSA私钥
* @return 成功返回 RSA私钥长度,失败返回error 错误信息
*/
func GetPriKeyLen(PriKey []byte) (int, error) {
if PriKey == nil {
return 0, errors.New("input arguments error")
}
block, _ := pem.Decode(PriKey)
if block == nil {
return 0, errors.New("private rsaKey error!")
}
priv, err := x509.ParsePKCS1PrivateKey(block.Bytes)
if err != nil {
return 0, err
}
return priv.N.BitLen(), nil
}
func main() {
//获取rsa 公钥长度
PubKeyLen, _ := GetPubKeyLen(publicKey)
fmt.Println("pbulic key len is ", PubKeyLen)
//获取rsa 私钥长度
PriKeyLen, _ := GetPriKeyLen(privateKey)
fmt.Println("private key len is ", PriKeyLen)
}
附录:以下程序是产生RSA公私密钥的程序,附件,方便测试。
package main
import (
"crypto/rand"
"crypto/rsa"
"crypto/x509"
"encoding/pem"
"flag"
"log"
"os"
)
func main() {
var bits int
flag.IntVar(&bits, "b", 1024, "密钥长度,默认为1024位")
flag.Parse()
if err := GenRsaKey(bits); err != nil {
log.Fatal("密钥文件生成失败!")
}
log.Println("密钥文件生成成功!")
}
func GenRsaKey(bits int) error {
// 生成私钥文件
privateKey, err := rsa.GenerateKey(rand.Reader, bits)
if err != nil {
return err
}
derStream := x509.MarshalPKCS1PrivateKey(privateKey)
block := &pem.Block{
Type: "私钥",
Bytes: derStream,
}
file, err := os.Create("private.pem")
if err != nil {
return err
}
err = pem.Encode(file, block)
if err != nil {
return err
}
// 生成公钥文件
publicKey := &privateKey.PublicKey
derPkix, err := x509.MarshalPKIXPublicKey(publicKey)
if err != nil {
return err
}
block = &pem.Block{
Type: "公钥",
Bytes: derPkix,
}
file, err = os.Create("public.pem")
if err != nil {
return err
}
err = pem.Encode(file, block)
if err != nil {
return err
}
return nil
}