blob: 11e7af9a0194eca36a79451a09139f5f4f4a032e [file] [log] [blame]
// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package subtle_test
import (
"bytes"
"encoding/hex"
"fmt"
"strings"
"testing"
"github.com/google/tink/go/prf/subtle"
"github.com/google/tink/go/testutil"
)
type rfc4868test struct {
key string
data string
prf map[string]string
}
func TestVectorsRFC4868(t *testing.T) {
// Test vectors from RFC 4868.
testvectors := []*rfc4868test{
{
key: "0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b0b",
data: "4869205468657265",
prf: map[string]string{
"SHA256": "b0344c61d8db38535ca8afceaf0bf12b881dc200c9833da726e9376c2e32cff7",
"SHA512": "87aa7cdea5ef619d4ff0b4241a1d6cb02379f4e2ce4ec2787ad0b30545e17cdedaa833b7d6b8a702038b274eaea3f4e4be9d914eeb61f1702e696c203a126854",
},
},
{
key: "4a656665",
data: "7768617420646f2079612077616e7420666f72206e6f7468696e673f",
prf: map[string]string{
"SHA256": "5bdcc146bf60754e6a042426089575c75a003f089d2739839dec58b964ec3843",
"SHA512": "164b7a7bfcf819e2e395fbe73b56e0a387bd64222e831fd610270cd7ea2505549758bf75c05a994a6d034f65f8f0e6fdcaeab1a34d4a6b4b636e070a38bce737",
},
},
{
key: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
data: "dddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddddd",
prf: map[string]string{
"SHA256": "773ea91e36800e46854db8ebd09181a72959098b3ef8c122d9635514ced565fe",
"SHA512": "fa73b0089d56a284efb0f0756c890be9b1b5dbdd8ee81a3655f83e33b2279d39bf3e848279a722c806b485a47e67c807b946a337bee8942674278859e13292fb",
},
},
{
key: "0102030405060708090a0b0c0d0e0f10111213141516171819",
data: "cdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcdcd",
prf: map[string]string{
"SHA256": "82558a389a443c0ea4cc819899f2083a85f0faa3e578f8077a2e3ff46729665b",
"SHA512": "b0ba465637458c6990e5a8c5f61d4af7e576d97ff94b872de76f8050361ee3dba91ca5c11aa25eb4d679275cc5788063a5f19741120c4f2de2adebeb10a298dd",
},
},
{
key: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
data: "54657374205573696e67204c6172676572205468616e20426c6f636b2d53697a65204b6579202d2048617368204b6579204669727374",
prf: map[string]string{
"SHA256": "60e431591ee0b67f0d8a26aacbf5b77f8e0bc6213728c5140546040f0ee37f54",
"SHA512": "80b24263c7c1a3ebb71493c1dd7be8b49b46d1f41b4aeec1121b013783f8f3526b56d037e05f2598bd0fd2215d6a1e5295e64f73f63f0aec8b915a985d786598",
},
},
{
key: "aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa",
data: "5468697320697320612074657374207573696e672061206c6172676572207468616e20626c6f636b2d73697a65206b657920616e642061206c6172676572207468616e20626c6f636b2d73697a6520646174612e20546865206b6579206e6565647320746f20626520686173686564206265666f7265206265696e6720757365642062792074686520484d414320616c676f726974686d2e",
prf: map[string]string{
"SHA256": "9b09ffa71b942fcb27635fbcd5b0e944bfdc63644f0713938a7f51535c3a35e2",
"SHA512": "e37b6a775dc87dbaa4dfa9f96e5e3ffddebd71f8867289865df5a32d20cdc944b6022cac3c4982b10d5eeb55c3e4de15134676fb6de0446065c97440fa8c6a58",
},
},
}
for _, v := range testvectors {
key, err := hex.DecodeString(v.key)
if err != nil {
t.Errorf("Could not decode key: %v", err)
}
data, err := hex.DecodeString(v.data)
if err != nil {
t.Errorf("Could not decode data: %v", err)
}
for hash, e := range v.prf {
h, err := subtle.NewHMACPRF(hash, key)
if err != nil {
t.Errorf("Could not create HMAC PRF object: %v", err)
}
output, err := h.ComputePRF(data, uint32(len(e)/2))
if err != nil {
t.Errorf("Error computing HMAC: %v", err)
}
if hex.EncodeToString(output) != e {
t.Errorf("Computation and test vector differ. Computation: %q, Test Vector %q", hex.EncodeToString(output), e)
}
}
}
}
func TestHMACPRFWycheproofCases(t *testing.T) {
testutil.SkipTestIfTestSrcDirIsNotSet(t)
for _, hash := range []string{"SHA1", "SHA256", "SHA512"} {
filename := fmt.Sprintf("hmac_%s_test.json", strings.ToLower(hash))
suite := new(macSuite)
if err := testutil.PopulateSuite(suite, filename); err != nil {
t.Fatalf("Failed populating suite: %s", err)
}
for _, group := range suite.TestGroups {
groupName := fmt.Sprintf("%s-%s-%s(%d)", suite.Algorithm, group.Type, hash, group.KeySize)
if group.TagSize%8 != 0 {
t.Errorf("For %s, requested tag size is not a multiple of 8, but %d", groupName, group.TagSize)
}
for _, test := range group.Tests {
caseName := fmt.Sprintf("%s:Case-%d", groupName, test.CaseID)
t.Run(caseName, func(t *testing.T) {
h, err := subtle.NewHMACPRF(hash, test.Key)
switch test.Result {
case "valid":
if err != nil {
t.Fatalf("NewHMACPRF() failed: %v", err)
}
res, err := h.ComputePRF(test.Message, group.TagSize/8)
if err != nil {
t.Fatalf("ComputePRF() failed: %v", err)
}
if !bytes.Equal(res, test.Tag) {
t.Errorf("ComputePRF() result and expected result do not match:\nComputed: %q\nExpected: %q", hex.EncodeToString(res), hex.EncodeToString(test.Tag))
}
case "invalid":
if err != nil {
return
}
res, err := h.ComputePRF(test.Message, group.TagSize/8)
if err != nil {
return
}
if bytes.Equal(res, test.Tag) {
t.Errorf("ComputePRF() result and invalid expected result match:\nComputed: %q\nExpected: %q", hex.EncodeToString(res), hex.EncodeToString(test.Tag))
}
default:
t.Fatalf("Unsupported test result: %q", test.Result)
}
})
}
}
}
}
func TestHMACPRFHash(t *testing.T) {
if _, err := subtle.NewHMACPRF("SHA256", []byte{
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10}); err != nil {
t.Errorf("Expected NewHMACPRF to work with SHA256: %v", err)
}
if _, err := subtle.NewHMACPRF("SHA512", []byte{
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10}); err != nil {
t.Errorf("Expected NewHMACPRF to work with SHA512: %v", err)
}
if _, err := subtle.NewHMACPRF("SHA1", []byte{
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10}); err != nil {
t.Errorf("Expected NewHMACPRF to work with SHA1: %v", err)
}
if _, err := subtle.NewHMACPRF("md5", []byte{
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10}); err == nil {
t.Errorf("Expected NewHMACPRF to fail with md5")
}
}
func TestHMACPRFOutputLength(t *testing.T) {
for hash, length := range map[string]int{"SHA1": 20, "SHA256": 32, "SHA512": 64} {
prf, err := subtle.NewHMACPRF(hash, []byte{
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10,
0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08,
0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f, 0x10})
if err != nil {
t.Errorf("Expected NewHMACPRF to work on 32 byte key with hash %s", hash)
}
for i := 0; i <= length; i++ {
output, err := prf.ComputePRF([]byte{0x01, 0x02}, uint32(i))
if err != nil {
t.Errorf("Expected to be able to compute HMAC %s PRF with %d output length", hash, i)
}
if len(output) != i {
t.Errorf("Expected HMAC %s PRF to compute %d bytes, got %d", hash, i, len(output))
}
}
for i := length + 1; i < 100; i++ {
_, err := prf.ComputePRF([]byte{0x01, 0x02}, uint32(i))
if err == nil {
t.Errorf("Expected to not be able to compute HMAC %s PRF with %d output length", hash, i)
}
}
}
}
func TestValidateHMACPRFParams(t *testing.T) {
if err := subtle.ValidateHMACPRFParams("SHA256", 32); err != nil {
t.Errorf("Unexpected error for valid HMAC PRF params: %v", err)
}
if err := subtle.ValidateHMACPRFParams("SHA256", 4); err == nil {
t.Errorf("Short key size not detected for HMAC PRF params")
}
if err := subtle.ValidateHMACPRFParams("md5", 32); err == nil {
t.Errorf("Weak hash function not detected for HMAC PRF params")
}
}