Golang1.7.3使用標準庫的AES加密解密不實用擴展協議

package main

import (
    "crypto/aes"
    "crypto/cipher"
    "fmt"
)

type tobytes struct {
    cip     cipher.Block
    pdgtext []byte
}

func main() {
    cip, _ := aes.NewCipher([]byte("1234567891234567"))
    enc := &tobytes{cip: cip, pdgtext: make([]byte, cip.BlockSize())}
    src := enc.Encrypt([]byte("czxichen"))
    dst := enc.Decrypt(src)
    fmt.Println(string(dst))
}

//使用AES加密文本,加密的文本不能爲空
func (a *tobytes) Encrypt(src []byte) (dst []byte) {
    src = a.padding(src)
    dst = make([]byte, len(src))
    var index int = 0
    for len(src) > 0 {
        a.cip.Encrypt(dst[index:index+a.cip.BlockSize()], src)
        index += a.cip.BlockSize()
        src = src[a.cip.BlockSize():]
    }
    return dst
}

//使用AES解密文本
func (a *tobytes) Decrypt(src []byte) (dst []byte) {
    if len(src)%a.cip.BlockSize() != 0 {
        return src
    }
    dst = make([]byte, len(src))
    var index int = 0
    for len(src) > 0 {
        a.cip.Decrypt(dst[index:index+a.cip.BlockSize()], src)
        index += a.cip.BlockSize()
        src = src[a.cip.BlockSize():]
    }
    return a.unpadding(dst)
}

//使用AES加密文本的時候文本必須定長,即必須是16,24,32的整數倍,
func (a *tobytes) padding(src []byte) (dst []byte) {
    pdg := a.cip.BlockSize() - len(src)%a.cip.BlockSize()
    p := a.pdgtext[:pdg]
    p[pdg-1] = byte(pdg)
    return append(src, p...)
}

//使用AES解密文本,解密收刪除padding的文本
func (a *tobytes) unpadding(src []byte) (dst []byte) {
    length := len(src)
    if length <= 0 {
        return src
    }
    return src[:(length - int(src[length-1]))]
}