【Go】Goで学ぶ暗号技術 ~ ECBモード~

元のデータ:        [H][e][l][l][o]
パディング後:      [H][e][l][l][o][0B][0B][0B][0B][0B][0B][0B][0B][0B][0B][0B]
（16バイトに調整）

package main

import (
    "bytes"
    "crypto/aes"
    "fmt"
    "log"
)

// PKCS#7パディング
func pkcs7Padding(originalMessage []byte, blockSize int) []byte {
    padding := blockSize - len(originalMessage)%blockSize
    padtext := bytes.Repeat([]byte{byte(padding)}, padding)
    return append(originalMessage, padtext...)
}

// PKCS#7アンパディング
func pkcs7UnPadding(data []byte) ([]byte, error) {
    length := len(data)
    if length == 0 {
        return nil, fmt.Errorf("data is empty")
    }
    padding := int(data[length-1])
    if padding > length {
        return nil, fmt.Errorf("invalid padding size")
    }
    return data[:length-padding], nil
}

// 暗号化
func encryptECB(originalMessage, key []byte) ([]byte, error) {
    block, err := aes.NewCipher(key)
    if err != nil {
        return nil, err
    }
    blockSize := block.BlockSize()
    plaintext := pkcs7Padding(originalMessage, blockSize)
    cipherText := make([]byte, len(plaintext))
    for start := 0; start < len(plaintext); start += blockSize {
        block.Encrypt(cipherText[start:start+blockSize], plaintext[start:start+blockSize])
    }
    return cipherText, nil
}

// 復号化
func decryptECB(ciphertext, key []byte) ([]byte, error) {
    block, err := aes.NewCipher(key)
    if err != nil {
        return nil, err
    }
    blockSize := block.BlockSize()
    if len(ciphertext)%blockSize != 0 {
        return nil, fmt.Errorf("ciphertext is not a multiple of the block size")
    }
    plaintext := make([]byte, len(ciphertext))
    for start := 0; start < len(ciphertext); start += blockSize {
        block.Decrypt(plaintext[start:start+blockSize], ciphertext[start:start+blockSize])
    }
    return pkcs7UnPadding(plaintext)
}

func main() {
    key := []byte("abcdefghijklmnop")
    originalMessage := []byte("Hello, ECB mode!")
    
    // 暗号化
    encrypted, err := encryptECB(originalMessage, key)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Encrypted: %x\n", encrypted)
    
    // 復号化
    decrypted, err := decryptECB(encrypted, key)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Decrypted: %s\n", decrypted)
}