因在做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 }