tea_test.go

Documentation: golang.org/x/crypto/tea

     1  // Copyright 2015 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 tea
     6  
     7  import (
     8  	"bytes"
     9  	"testing"
    10  )
    11  
    12  // A sample test key for when we just want to initialize a cipher
    13  var testKey = []byte{0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99, 0xAA, 0xBB, 0xCC, 0xDD, 0xEE, 0xFF}
    14  
    15  // Test that the block size for tea is correct
    16  func TestBlocksize(t *testing.T) {
    17  	c, err := NewCipher(testKey)
    18  	if err != nil {
    19  		t.Fatalf("NewCipher returned error: %s", err)
    20  	}
    21  
    22  	if result := c.BlockSize(); result != BlockSize {
    23  		t.Errorf("cipher.BlockSize returned %d, but expected %d", result, BlockSize)
    24  	}
    25  }
    26  
    27  // Test that invalid key sizes return an error
    28  func TestInvalidKeySize(t *testing.T) {
    29  	var key [KeySize + 1]byte
    30  
    31  	if _, err := NewCipher(key[:]); err == nil {
    32  		t.Errorf("invalid key size %d didn't result in an error.", len(key))
    33  	}
    34  
    35  	if _, err := NewCipher(key[:KeySize-1]); err == nil {
    36  		t.Errorf("invalid key size %d didn't result in an error.", KeySize-1)
    37  	}
    38  }
    39  
    40  // Test Vectors
    41  type teaTest struct {
    42  	rounds     int
    43  	key        []byte
    44  	plaintext  []byte
    45  	ciphertext []byte
    46  }
    47  
    48  var teaTests = []teaTest{
    49  	// These were sourced from https://github.com/froydnj/ironclad/blob/master/testing/test-vectors/tea.testvec
    50  	{
    51  		numRounds,
    52  		[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00},
    53  		[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00},
    54  		[]byte{0x41, 0xea, 0x3a, 0x0a, 0x94, 0xba, 0xa9, 0x40},
    55  	},
    56  	{
    57  		numRounds,
    58  		[]byte{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
    59  		[]byte{0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff},
    60  		[]byte{0x31, 0x9b, 0xbe, 0xfb, 0x01, 0x6a, 0xbd, 0xb2},
    61  	},
    62  	{
    63  		16,
    64  		[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00},
    65  		[]byte{0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00},
    66  		[]byte{0xed, 0x28, 0x5d, 0xa1, 0x45, 0x5b, 0x33, 0xc1},
    67  	},
    68  }
    69  
    70  // Test encryption
    71  func TestCipherEncrypt(t *testing.T) {
    72  	// Test encryption with standard 64 rounds
    73  	for i, test := range teaTests {
    74  		c, err := NewCipherWithRounds(test.key, test.rounds)
    75  		if err != nil {
    76  			t.Fatalf("#%d: NewCipher returned error: %s", i, err)
    77  		}
    78  
    79  		var ciphertext [BlockSize]byte
    80  		c.Encrypt(ciphertext[:], test.plaintext)
    81  
    82  		if !bytes.Equal(ciphertext[:], test.ciphertext) {
    83  			t.Errorf("#%d: incorrect ciphertext. Got %x, wanted %x", i, ciphertext, test.ciphertext)
    84  		}
    85  
    86  		var plaintext2 [BlockSize]byte
    87  		c.Decrypt(plaintext2[:], ciphertext[:])
    88  
    89  		if !bytes.Equal(plaintext2[:], test.plaintext) {
    90  			t.Errorf("#%d: incorrect plaintext. Got %x, wanted %x", i, plaintext2, test.plaintext)
    91  		}
    92  	}
    93  }
    94
View as plain text