Golang實現ECB模式3DES算法

簡介

因項目須要使用ECB模式下的3DES算法加解密信息，golang默認只提供CBC模式，只能本身實現ECB模式。
參考https://segmentfault.com/a/11...，文章對ECB模式的DES有解釋，並實現了部分DES算法樣例。這裏把算法補全，提供3DES算法實現。

基礎

3DES
3DES算法就是採用一個長度爲24字節的密鑰，將密鑰分紅各8字節的3份子密鑰：K一、k二、k3。使用這3個密鑰對明文進行加密、解密處理，以下：
E(k,d)、D(k,d)分別表示使用密鑰k對數據d進行加密或解密，返回加密或解密後的數據。

3DES加密過程：
E(k3,D(k2,E(k1,d)))
意思爲：將明文d先用k1加密，獲得密文d1；對d1再用k2作解密處理，獲得密文d2；再對d2用k3作加密處理，獲得最終密文。
3DES解密過程與加密相反：
D(k1,E(k2,D(k3,d)))
意思爲：將密文d先用k3解密，獲得密文d1；對d1再用k2作加密處理，獲得密文d2；再對d2用k1作解密處理，獲得最終明文。

填充
填充方式採用PKCS5Padding，代碼照搬以下：

func PKCS5Padding(ciphertext []byte, blockSize int) []byte {
    padding := blockSize - len(ciphertext)%blockSize
    padtext := bytes.Repeat([]byte{byte(padding)}, padding)
    return append(ciphertext, padtext...)
}

func PKCS5Unpadding(origData []byte) []byte {
    length := len(origData)
    unpadding := int(origData[length-1])
    return origData[:(length - unpadding)]
}

填充過程就是要把明文長度湊成8的整數倍，少幾個就填充幾個對應的數字。如少4個字節才滿8的倍數，那就填充4個0x04；若是明文恰好是8的倍數，就要再填充8個0x08。
解密後，要把填充的數據刪除，就取最後一個字節的值，按值刪掉最後幾個字節。(設計的很巧妙)

代碼

完整的代碼：

package tripledesecb
import (
    "bytes"
    "crypto/des"
    "errors"
    "fmt"
    "golang.org/x/crypto/pbkdf2"
)

//ECB PKCS5Padding
func PKCS5Padding(ciphertext []byte, blockSize int) []byte {
    padding := blockSize - len(ciphertext)%blockSize
    padtext := bytes.Repeat([]byte{byte(padding)}, padding)
    return append(ciphertext, padtext...)
}

//ECB PKCS5Unpadding
func PKCS5Unpadding(origData []byte) []byte {
    length := len(origData)
    unpadding := int(origData[length-1])
    return origData[:(length - unpadding)]
}

//Des加密
func encrypt(origData, key []byte) ([]byte, error) {
    if len(origData) < 1 || len(key) < 1 {
        return nil, errors.New("wrong data or key")
    }
    block, err := des.NewCipher(key)
    if err != nil {
        return nil, err
    }
    bs := block.BlockSize()
    if len(origData)%bs != 0 {
        return nil, errors.New("wrong padding")
    }
    out := make([]byte, len(origData))
    dst := out
    for len(origData) > 0 {
        block.Encrypt(dst, origData[:bs])
        origData = origData[bs:]
        dst = dst[bs:]
    }
    return out, nil
}

//Des解密
func decrypt(crypted, key []byte) ([]byte, error) {
    if len(crypted) < 1 || len(key) < 1 {
        return nil, errors.New("wrong data or key")
    }
    block, err := des.NewCipher(key)
    if err != nil {
        return nil, err
    }
    out := make([]byte, len(crypted))
    dst := out
    bs := block.BlockSize()
    if len(crypted)%bs != 0 {
        return nil, errors.New("wrong crypted size")
    }

    for len(crypted) > 0 {
        block.Decrypt(dst, crypted[:bs])
        crypted = crypted[bs:]
        dst = dst[bs:]
    }

    return out, nil
}

//[golang ECB 3DES Encrypt]
func TripleEcbDesEncrypt(origData, key []byte) ([]byte, error) {
    tkey := make([]byte, 24, 24)
    copy(tkey, key)
    k1 := tkey[:8]
    k2 := tkey[8:16]
    k3 := tkey[16:]

    block, err := des.NewCipher(k1)
    if err != nil {
        return nil, err
    }
    bs := block.BlockSize()
    origData = PKCS5Padding(origData, bs)

    buf1, err := encrypt(origData, k1)
    if err != nil {
        return nil, err
    }
    buf2, err := decrypt(buf1, k2)
    if err != nil {
        return nil, err
    }
    out, err := encrypt(buf2, k3)
    if err != nil {
        return nil, err
    }
    return out, nil
}

//[golang ECB 3DES Decrypt]
func TripleEcbDesDecrypt(crypted, key []byte) ([]byte, error) {
    tkey := make([]byte, 24, 24)
    copy(tkey, key)
    k1 := tkey[:8]
    k2 := tkey[8:16]
    k3 := tkey[16:]
    buf1, err := decrypt(crypted, k3)
    if err != nil {
        return nil, err
    }
    buf2, err := encrypt(buf1, k2)
    if err != nil {
        return nil, err
    }
    out, err := decrypt(buf2, k1)
    if err != nil {
        return nil, err
    }
    out = PKCS5Unpadding(out)
    return out, nil
}

上面的代碼，稍微改一下就能夠開放出ecb模式的des算法，我用不上，就沒放出來。

剛開始用golang，也是第一次在segmentfault上發文，輕拍！