github.com/zebozhuang/go@v0.0.0-20200207033046-f8a98f6f5c5d/src/crypto/cipher/cbc_aes_test.go (about)

     1  // Copyright 2009 The Go Authors. All rights reserved.
     2  // Use of this source code is governed by a BSD-style
     3  // license that can be found in the LICENSE file.
     4  
     5  // CBC AES test vectors.
     6  
     7  // See U.S. National Institute of Standards and Technology (NIST)
     8  // Special Publication 800-38A, ``Recommendation for Block Cipher
     9  // Modes of Operation,'' 2001 Edition, pp. 24-29.
    10  
    11  package cipher_test
    12  
    13  import (
    14  	"bytes"
    15  	"crypto/aes"
    16  	"crypto/cipher"
    17  	"testing"
    18  )
    19  
    20  var cbcAESTests = []struct {
    21  	name string
    22  	key  []byte
    23  	iv   []byte
    24  	in   []byte
    25  	out  []byte
    26  }{
    27  	// NIST SP 800-38A pp 27-29
    28  	{
    29  		"CBC-AES128",
    30  		commonKey128,
    31  		commonIV,
    32  		commonInput,
    33  		[]byte{
    34  			0x76, 0x49, 0xab, 0xac, 0x81, 0x19, 0xb2, 0x46, 0xce, 0xe9, 0x8e, 0x9b, 0x12, 0xe9, 0x19, 0x7d,
    35  			0x50, 0x86, 0xcb, 0x9b, 0x50, 0x72, 0x19, 0xee, 0x95, 0xdb, 0x11, 0x3a, 0x91, 0x76, 0x78, 0xb2,
    36  			0x73, 0xbe, 0xd6, 0xb8, 0xe3, 0xc1, 0x74, 0x3b, 0x71, 0x16, 0xe6, 0x9e, 0x22, 0x22, 0x95, 0x16,
    37  			0x3f, 0xf1, 0xca, 0xa1, 0x68, 0x1f, 0xac, 0x09, 0x12, 0x0e, 0xca, 0x30, 0x75, 0x86, 0xe1, 0xa7,
    38  		},
    39  	},
    40  	{
    41  		"CBC-AES192",
    42  		commonKey192,
    43  		commonIV,
    44  		commonInput,
    45  		[]byte{
    46  			0x4f, 0x02, 0x1d, 0xb2, 0x43, 0xbc, 0x63, 0x3d, 0x71, 0x78, 0x18, 0x3a, 0x9f, 0xa0, 0x71, 0xe8,
    47  			0xb4, 0xd9, 0xad, 0xa9, 0xad, 0x7d, 0xed, 0xf4, 0xe5, 0xe7, 0x38, 0x76, 0x3f, 0x69, 0x14, 0x5a,
    48  			0x57, 0x1b, 0x24, 0x20, 0x12, 0xfb, 0x7a, 0xe0, 0x7f, 0xa9, 0xba, 0xac, 0x3d, 0xf1, 0x02, 0xe0,
    49  			0x08, 0xb0, 0xe2, 0x79, 0x88, 0x59, 0x88, 0x81, 0xd9, 0x20, 0xa9, 0xe6, 0x4f, 0x56, 0x15, 0xcd,
    50  		},
    51  	},
    52  	{
    53  		"CBC-AES256",
    54  		commonKey256,
    55  		commonIV,
    56  		commonInput,
    57  		[]byte{
    58  			0xf5, 0x8c, 0x4c, 0x04, 0xd6, 0xe5, 0xf1, 0xba, 0x77, 0x9e, 0xab, 0xfb, 0x5f, 0x7b, 0xfb, 0xd6,
    59  			0x9c, 0xfc, 0x4e, 0x96, 0x7e, 0xdb, 0x80, 0x8d, 0x67, 0x9f, 0x77, 0x7b, 0xc6, 0x70, 0x2c, 0x7d,
    60  			0x39, 0xf2, 0x33, 0x69, 0xa9, 0xd9, 0xba, 0xcf, 0xa5, 0x30, 0xe2, 0x63, 0x04, 0x23, 0x14, 0x61,
    61  			0xb2, 0xeb, 0x05, 0xe2, 0xc3, 0x9b, 0xe9, 0xfc, 0xda, 0x6c, 0x19, 0x07, 0x8c, 0x6a, 0x9d, 0x1b,
    62  		},
    63  	},
    64  }
    65  
    66  func TestCBCEncrypterAES(t *testing.T) {
    67  	for _, test := range cbcAESTests {
    68  		c, err := aes.NewCipher(test.key)
    69  		if err != nil {
    70  			t.Errorf("%s: NewCipher(%d bytes) = %s", test.name, len(test.key), err)
    71  			continue
    72  		}
    73  
    74  		encrypter := cipher.NewCBCEncrypter(c, test.iv)
    75  
    76  		data := make([]byte, len(test.in))
    77  		copy(data, test.in)
    78  
    79  		encrypter.CryptBlocks(data, data)
    80  		if !bytes.Equal(test.out, data) {
    81  			t.Errorf("%s: CBCEncrypter\nhave %x\nwant %x", test.name, data, test.out)
    82  		}
    83  	}
    84  }
    85  
    86  func TestCBCDecrypterAES(t *testing.T) {
    87  	for _, test := range cbcAESTests {
    88  		c, err := aes.NewCipher(test.key)
    89  		if err != nil {
    90  			t.Errorf("%s: NewCipher(%d bytes) = %s", test.name, len(test.key), err)
    91  			continue
    92  		}
    93  
    94  		decrypter := cipher.NewCBCDecrypter(c, test.iv)
    95  
    96  		data := make([]byte, len(test.out))
    97  		copy(data, test.out)
    98  
    99  		decrypter.CryptBlocks(data, data)
   100  		if !bytes.Equal(test.in, data) {
   101  			t.Errorf("%s: CBCDecrypter\nhave %x\nwant %x", test.name, data, test.in)
   102  		}
   103  	}
   104  }