package main

import (
    "crypto/ecdsa"
    "crypto/elliptic"
    "crypto/rand"
    "crypto/sha1"
    "crypto/x509"
    "encoding/pem"
    "fmt"
    "math/big"
    "os"
)

func main() {
    GenerateEccKey()
    src := []byte("橢圓曲線進(jìn)行數(shù)字簽名")
    rText,sText := EccSignature(src,"eccPrivate.pem")
    bl := EccVerify(src,rText,sText,"eccPublic.pem")
    fmt.Println(bl)
}

// 生成密鑰對(duì)
func GenerateEccKey() {
    // --------私鑰寫入磁盤--------
    // 1 使用ecdsa生成密鑰對(duì)
    privateKey,err := ecdsa.GenerateKey(elliptic.P521(),rand.Reader)
    if err != nil {
        panic(err)
    }
    // 2 使用x509將私鑰進(jìn)行序列化
    derText,err := x509.MarshalECPrivateKey(privateKey)
    if err != nil {
        panic(err)
    }
    // 3 將序列化后的字符串放入pem.Block結(jié)構(gòu)體中
    block := pem.Block{
        Type:"ecdsa private key",
        Bytes:derText,
    }
    // 4 使用pem編碼
    file,err := os.Create("eccPrivate.pem")
    if err != nil {
        panic(err)
    }
    pem.Encode(file,&block)
    file.Close()

    // --------公鑰寫入磁盤--------
    // 1 從私鑰中獲取公鑰
    publicKey := privateKey.PublicKey
    // 2 x509序列化公鑰
    derText,err = x509.MarshalPKIXPublicKey(&publicKey)
    if err != nil {
        panic(err)
    }
    // 3 將序列化后的字符串放入pem.Block 中
    block = pem.Block{
        Type:"ecdsa public key",
        Bytes:derText,
    }
    // 4 使用pem編碼
    file,err = os.Create("eccPublic.pem")
    if err != nil {
        panic(err)
    }
    pem.Encode(file,&block)
    file.Close()
}

// ecdsa 簽名
func EccSignature(plainText []byte,privName string) (rText,sText []byte) {
    // 1 打開私鑰將內(nèi)容讀取出來(lái)
    file,err := os.Open(privName)
    if err != nil {
        panic(err)
    }
    fileInfo,err := file.Stat()
    if err != nil {
        panic(err)
    }
    buf := make([]byte,fileInfo.Size())
    file.Read(buf)
    file.Close()
    // 2 pem解碼
    block,_ := pem.Decode(buf)
    // 3 x509解析
    privateKey,err := x509.ParseECPrivateKey(block.Bytes)
    if err != nil {
        panic(err)
    }
    // 4 對(duì)原始數(shù)據(jù)進(jìn)行hash運(yùn)算得到散列值
    hashText := sha1.Sum(plainText)
    // 5 進(jìn)行數(shù)據(jù)簽名
    r,s,err := ecdsa.Sign(rand.Reader,privateKey,hashText[:])
    if err != nil {
        panic(err)
    }
    // 6 對(duì)r贾惦，s內(nèi)存中的數(shù)據(jù)進(jìn)行格式化
    rText,err = r.MarshalText()
    if err != nil {
        panic(err)
    }
    sText,err = s.MarshalText()
    if err != nil {
        panic(err)
    }
    return
}

// ecdsa簽名認(rèn)證
func EccVerify(plainText,rText,sText []byte,pubFile string) bool {
    // 1 打開公鑰文件逝薪，將內(nèi)容讀取出來(lái)
    file,err := os.Open(pubFile)
    if err != nil {
        panic(err)
    }
    fileInfo,err := file.Stat()
    if err != nil {
        panic(err)
    }
    buf := make([]byte,fileInfo.Size())
    file.Read(buf)
    file.Close()
    // 2 pem解碼
    block,_ := pem.Decode(buf)
    // 3 x509解析
    pubInterface,err := x509.ParsePKIXPublicKey(block.Bytes)
    if err != nil {
        panic(err)
    }
    // 4 類型斷言獲取公鑰
    publicKey := pubInterface.(*ecdsa.PublicKey)
    // 5 對(duì)原始數(shù)據(jù)進(jìn)行hash運(yùn)算得到散列值
    hashText := sha1.Sum(plainText)
    // 6 簽名認(rèn)證
    // 將rText,sText 轉(zhuǎn)換成 int
    var r,s big.Int
    r.UnmarshalText(rText)
    s.UnmarshalText(sText)
    bl := ecdsa.Verify(publicKey,hashText[:],&r,&s)
    return bl
}