Codebase list golang-github-twstrike-otr3 / HEAD wipe_test.go
HEAD

Tree @HEAD (Download .tar.gz)

wipe_test.go @HEAD — raw · history · blame 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
package otr3

import (
	"math/big"
	"testing"
)

func Test_zeroes_generateZeroes(t *testing.T) {
	z := zeroes(5)
	assertDeepEquals(t, z, []byte{0, 0, 0, 0, 0})
}

func Test_wipeBytes_zeroesTheSlice(t *testing.T) {
	b := []byte{1, 2, 3, 4, 5}
	wipeBytes(b)

	assertDeepEquals(t, b, zeroes(len(b)))
}

func Test_wipeBigInt_numberIsZeroed(t *testing.T) {
	n := big.NewInt(3)
	wipeBigInt(n)
	assertEquals(t, n.Cmp(big.NewInt(0)), 0)
}

func Test_setBigInt_numberIsSet(t *testing.T) {
	n := big.NewInt(3)
	n = setBigInt(n, big.NewInt(5))
	assertEquals(t, n.Cmp(big.NewInt(5)), 0)
}

func Test_setBigInt_setWhenSourceIsNull(t *testing.T) {
	var n *big.Int
	n = setBigInt(n, big.NewInt(5))
	assertEquals(t, n.Cmp(big.NewInt(5)), 0)
}

func Test_wipe_macKey(t *testing.T) {
	k := macKey{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 7, 6, 5}
	k.wipe()

	assertDeepEquals(t, k, macKey{})
}

func Test_wipe_keyManagementContext(t *testing.T) {
	keys := keyManagementContext{
		ourKeyID:   2,
		theirKeyID: 3,
		ourCurrentDHKeys: dhKeyPair{
			priv: big.NewInt(1),
			pub:  big.NewInt(2),
		},
		ourPreviousDHKeys: dhKeyPair{
			priv: big.NewInt(3),
			pub:  big.NewInt(4),
		},
		theirCurrentDHPubKey:  big.NewInt(5),
		theirPreviousDHPubKey: big.NewInt(6),
		counterHistory: counterHistory{
			counters: []*keyPairCounter{
				&keyPairCounter{1, 1, 1, 1},
			},
		},
		macKeyHistory: macKeyHistory{
			items: []macKeyUsage{
				macKeyUsage{
					ourKeyID:     2,
					theirKeyID:   3,
					receivingKey: macKey{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4},
				},
			},
		},
		oldMACKeys: []macKey{
			macKey{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 8, 7, 6, 5},
		},
	}

	keys.wipe()

	assertDeepEquals(t, keys, keyManagementContext{})
}

func Test_dhKeyPair_wipe_HandlesNilWell(t *testing.T) {
	(*dhKeyPair)(nil).wipe()
}

func Test_akeKeys_wipe_HandlesNilWell(t *testing.T) {
	(*akeKeys)(nil).wipe()
}

func Test_ake_wipe_HandlesNilWell(t *testing.T) {
	(*ake)(nil).wipe(true)
}

func Test_ake_wipeGX_HandlesNilWell(t *testing.T) {
	(*ake)(nil).wipeGX()
}

func Test_keyManagementContext_wipeKeys_HandlesNilWell(t *testing.T) {
	(*keyManagementContext)(nil).wipeKeys()
}

func Test_keyManagementContext_wipe_HandlesNilWell(t *testing.T) {
	(*keyManagementContext)(nil).wipe()
}

func Test_counterHistory_wipe_HandlesNilWell(t *testing.T) {
	(*counterHistory)(nil).wipe()
}

func Test_keyPairCounter_wipe_HandlesNilWell(t *testing.T) {
	(*keyPairCounter)(nil).wipe()
}

func Test_macKeyHistory_wipe_HandlesNilWell(t *testing.T) {
	(*macKeyHistory)(nil).wipe()
}

func Test_macKeyUsage_wipe_HandlesNilWell(t *testing.T) {
	(*macKeyUsage)(nil).wipe()
}

func Test_macKey_wipe_HandlesNilWell(t *testing.T) {
	(*macKey)(nil).wipe()
}