RSA公私密钥文件生成的代码实现

package main

import (
    "flag"
    "crypto/rsa"
    "crypto/rand"
    "crypto/x509"
    "encoding/pem"
    "os"
    "log"
)

func main() {
    var bits int
    flag.IntVar(&bits, "b", 1024, "密钥长度，默认是1024")
    flag.Parse()

    if GenRsaKey(bits) != nil {
        log.Fatalln("密钥文件生成失败！")
    }
    log.Println("密钥文件生成成功！")
}

func GenRsaKey(bits int) error {
    // 生成私钥文件
    privateKey, err := rsa.GenerateKey(rand.Reader, bits)
    if err != nil {
        return nil
    }
    derStream := x509.MarshalPKCS1PrivateKey(privateKey)
    /* 核心代码开始 */
    block := &pem.Block{
        Type:"PRIVATE KEY",
        Bytes:derStream,
    }
    fi, err := os.Create("private.pem")
    if err != nil {
        return err
    }
    err = pem.Encode(fi, block)
    if err != nil {
        return err
    }
    /* 核心代码结束 */
    // 生成公钥文件
    publicKey := &privateKey.PublicKey
    derPkix, err := x509.MarshalPKIXPublicKey(publicKey)
    if err != nil {
        return err
    }
    /* 核心代码开始 */
    block = &pem.Block{
        Type:"PUBLIC KEY",
        Bytes:derPkix,
    }
    fi, err = os.Create("public.pem")
    if err != nil {
        return err
    }
    err = pem.Encode(fi, block)
    /* 核心代码结束 */
    return err
}

package main

import (
    "io/ioutil"
    "log"
    "github.com/syndtr/goleveldb/leveldb/errors"
    "encoding/pem"
    "crypto/x509"
    "crypto/rsa"
    "fmt"
)

// 全局变量
var privateKey, publicKey []byte

func init() {
    var err error
    publicKey, err = ioutil.ReadFile("public.pem")
    if err != nil {
        log.Fatalln(err)
    }
    privateKey, err = ioutil.ReadFile("private.pem")
    if err != nil {
        log.Fatalln(err)
    }
}
/**
 * 功能：获取RSA公钥长度
 * 参数：public
 * 返回：成功则返回 RSA 公钥长度，失败返回 error 错误信息
 */
func GetPubKeyLen(pubKey []byte) (int, error) {
    if pubKey == nil {
        return 0, errors.New("input arguments error")
    }
    block, _ := pem.Decode(publicKey)
    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
}
/*
    获取RSA私钥长度
    PriKey
    成功返回 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(pubKeyLen)

    // 获取rsa 私钥长度
    privateLen, _ := GetPriKeyLen(privateKey)
    fmt.Println(privateLen)
}