// pkcs7padding和pkcs5padding的填充方式相同，填充字节的值都等于填充字节的个数。
// 例如需要填充4个字节，则填充的值为"4 4 4 4"。
var (
  // only difference is the block size, PKCS5 is 8 bytes, PKCS7 can be any bytes
  PKCS5          = &pkcs5{}
  PKCS7          = &pkcs5{}
  ErrPaddingSize = errors.New("padding size error")
)

// pkcs5Padding is a pkcs5 padding struct.
type pkcs5 struct{}

// Padding implements the Padding interface Padding method.
func (p *pkcs5) Padding(src []byte, blockSize int) []byte {
  srcLen := len(src)
  //注意： 当srcLen是blockSize的整倍数时，padLen等于blockSize而非0
  padLen := blockSize - (srcLen % blockSize)
  padText := bytes.Repeat([]byte{byte(padLen)}, padLen)
  return append(src, padText...)
}

// Unpadding implements the Padding interface Unpadding method.
func (p *pkcs5) Unpadding(src []byte, blockSize int) ([]byte, error) {
  srcLen := len(src)
  paddingLen := int(src[srcLen-1])
  if paddingLen >= srcLen || paddingLen > blockSize {
    return nil, ErrPaddingSize
  }
  return src[:srcLen-paddingLen], nil
}

func main() {
  o := []byte("hello world!")
  p := PKCS5.Padding(o, 8)
  u, _ := PKCS5.Unpadding(p, 8)
  fmt.Println(p)
  // [104 101 108 108 111 32 119 111 114 108 100 33 4 4 4 4]
  fmt.Println(u)
  // [104 101 108 108 111 32 119 111 114 108 100 33]
}