Function verifyCiphertextSamples

repo/encryption/encryption_test.go:133–171

(t *testing.T, masterKey, contentID, payload []byte, samples map[string]string)

}

func verifyCiphertextSamples(t *testing.T, masterKey, contentID, payload []byte, samples map[string]string) {
	t.Helper()

	for _, encryptionAlgo := range encryption.SupportedAlgorithms(true) {
		enc, err := encryption.CreateEncryptor(parameters{encryptionAlgo, masterKey})
		if err != nil {
			t.Fatal(err)
		}

		ct := samples[encryptionAlgo]
		if ct == "" {
			func() {
				var v gather.WriteBuffer
				defer v.Close()

				require.NoError(t, enc.Encrypt(gather.FromSlice(payload), contentID, &v))

				t.Errorf("missing ciphertext sample for %q: %q,", encryptionAlgo, hex.EncodeToString(payload))
			}()
		} else {
			b, err := hex.DecodeString(ct)
			if err != nil {
				t.Errorf("invalid ciphertext for %v: %v", encryptionAlgo, err)
				continue
			}

			func() {
				var plainText gather.WriteBuffer
				defer plainText.Close()

				require.NoError(t, enc.Decrypt(gather.FromSlice(b), contentID, &plainText))

				if v := plainText.ToByteSlice(); !bytes.Equal(v, payload) {
					t.Errorf("invalid plaintext after decryption %x, want %x", v, payload)
				}
			}()
		}
	}
}

func BenchmarkEncryption(b *testing.B) {
	masterKey := make([]byte, 32)