cfb_test.go

Documentation: crypto/cipher

		 1  // Copyright 2010 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  package cipher_test
		 6  
		 7  import (
		 8  	"bytes"
		 9  	"crypto/aes"
		10  	"crypto/cipher"
		11  	"crypto/rand"
		12  	"encoding/hex"
		13  	"testing"
		14  )
		15  
		16  // cfbTests contains the test vectors from
		17  // https://csrc.nist.gov/publications/nistpubs/800-38a/sp800-38a.pdf, section
		18  // F.3.13.
		19  var cfbTests = []struct {
		20  	key, iv, plaintext, ciphertext string
		21  }{
		22  	{
		23  		"2b7e151628aed2a6abf7158809cf4f3c",
		24  		"000102030405060708090a0b0c0d0e0f",
		25  		"6bc1bee22e409f96e93d7e117393172a",
		26  		"3b3fd92eb72dad20333449f8e83cfb4a",
		27  	},
		28  	{
		29  		"2b7e151628aed2a6abf7158809cf4f3c",
		30  		"3B3FD92EB72DAD20333449F8E83CFB4A",
		31  		"ae2d8a571e03ac9c9eb76fac45af8e51",
		32  		"c8a64537a0b3a93fcde3cdad9f1ce58b",
		33  	},
		34  	{
		35  		"2b7e151628aed2a6abf7158809cf4f3c",
		36  		"C8A64537A0B3A93FCDE3CDAD9F1CE58B",
		37  		"30c81c46a35ce411e5fbc1191a0a52ef",
		38  		"26751f67a3cbb140b1808cf187a4f4df",
		39  	},
		40  	{
		41  		"2b7e151628aed2a6abf7158809cf4f3c",
		42  		"26751F67A3CBB140B1808CF187A4F4DF",
		43  		"f69f2445df4f9b17ad2b417be66c3710",
		44  		"c04b05357c5d1c0eeac4c66f9ff7f2e6",
		45  	},
		46  }
		47  
		48  func TestCFBVectors(t *testing.T) {
		49  	for i, test := range cfbTests {
		50  		key, err := hex.DecodeString(test.key)
		51  		if err != nil {
		52  			t.Fatal(err)
		53  		}
		54  		iv, err := hex.DecodeString(test.iv)
		55  		if err != nil {
		56  			t.Fatal(err)
		57  		}
		58  		plaintext, err := hex.DecodeString(test.plaintext)
		59  		if err != nil {
		60  			t.Fatal(err)
		61  		}
		62  		expected, err := hex.DecodeString(test.ciphertext)
		63  		if err != nil {
		64  			t.Fatal(err)
		65  		}
		66  
		67  		block, err := aes.NewCipher(key)
		68  		if err != nil {
		69  			t.Fatal(err)
		70  		}
		71  
		72  		ciphertext := make([]byte, len(plaintext))
		73  		cfb := cipher.NewCFBEncrypter(block, iv)
		74  		cfb.XORKeyStream(ciphertext, plaintext)
		75  
		76  		if !bytes.Equal(ciphertext, expected) {
		77  			t.Errorf("#%d: wrong output: got %x, expected %x", i, ciphertext, expected)
		78  		}
		79  
		80  		cfbdec := cipher.NewCFBDecrypter(block, iv)
		81  		plaintextCopy := make([]byte, len(ciphertext))
		82  		cfbdec.XORKeyStream(plaintextCopy, ciphertext)
		83  
		84  		if !bytes.Equal(plaintextCopy, plaintext) {
		85  			t.Errorf("#%d: wrong plaintext: got %x, expected %x", i, plaintextCopy, plaintext)
		86  		}
		87  	}
		88  }
		89  
		90  func TestCFBInverse(t *testing.T) {
		91  	block, err := aes.NewCipher(commonKey128)
		92  	if err != nil {
		93  		t.Error(err)
		94  		return
		95  	}
		96  
		97  	plaintext := []byte("this is the plaintext. this is the plaintext.")
		98  	iv := make([]byte, block.BlockSize())
		99  	rand.Reader.Read(iv)
	 100  	cfb := cipher.NewCFBEncrypter(block, iv)
	 101  	ciphertext := make([]byte, len(plaintext))
	 102  	copy(ciphertext, plaintext)
	 103  	cfb.XORKeyStream(ciphertext, ciphertext)
	 104  
	 105  	cfbdec := cipher.NewCFBDecrypter(block, iv)
	 106  	plaintextCopy := make([]byte, len(plaintext))
	 107  	copy(plaintextCopy, ciphertext)
	 108  	cfbdec.XORKeyStream(plaintextCopy, plaintextCopy)
	 109  
	 110  	if !bytes.Equal(plaintextCopy, plaintext) {
	 111  		t.Errorf("got: %x, want: %x", plaintextCopy, plaintext)
	 112  	}
	 113  }
	 114
View as plain text