300 lines
8.0 KiB
C++
300 lines
8.0 KiB
C++
/*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2015 K. Isom <coder@kyleisom.net>
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* copy of this software and associated documentation files (the "Software"),
|
|
* to deal in the Software without restriction, including without limitation
|
|
* the rights to use, copy, modify, merge, publish, distribute, sublicense,
|
|
* and/or sell copies of the Software, and to permit persons to whom the
|
|
* Software is furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
|
|
* FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
|
* IN THE SOFTWARE.
|
|
*/
|
|
|
|
|
|
#include <cstdint>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
#include "test_utils.h"
|
|
|
|
using std::uint8_t;
|
|
using std::uint32_t;
|
|
using std::string;
|
|
using std::cout;
|
|
using std::cerr;
|
|
using std::endl;
|
|
|
|
|
|
void
|
|
DumpHexString(std::string& hs, uint8_t *s, uint32_t sl)
|
|
{
|
|
uint32_t const bl = (2 * sl) + 1;
|
|
char *buf = new char[bl];
|
|
string tmp;
|
|
|
|
memset(buf, 0, bl);
|
|
emsha::HexString(reinterpret_cast<uint8_t *>(buf), s, sl);
|
|
tmp = string(buf);
|
|
hs.swap(tmp);
|
|
delete[] buf;
|
|
}
|
|
|
|
|
|
emsha::EMSHAResult
|
|
runHMACTest(const struct hmacTest& test, const string& label)
|
|
{
|
|
emsha::HMAC h(test.key, test.keylen);
|
|
emsha::EMSHAResult res;
|
|
uint8_t dig[emsha::SHA256_HASH_SIZE];
|
|
string hs;
|
|
|
|
res = h.Update((uint8_t *)test.input.c_str(), test.input.size());
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
for (uint32_t n = 0; n < RESULT_ITERATIONS; n++) {
|
|
res = h.Result(dig);
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
DumpHexString(hs, dig, emsha::SHA256_HASH_SIZE);
|
|
if (hs != test.output) {
|
|
res = emsha::EMSHAResult::TestFailure;
|
|
goto exit;
|
|
}
|
|
memset(dig, 0, emsha::SHA256_HASH_SIZE);
|
|
}
|
|
|
|
// Ensure that a reset and update gives the same results.
|
|
h.Reset();
|
|
|
|
res = h.Update((uint8_t *)test.input.c_str(), test.input.size());
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
for (uint32_t n = 0; n < RESULT_ITERATIONS; n++) {
|
|
res = h.Result(dig);
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
DumpHexString(hs, dig, emsha::SHA256_HASH_SIZE);
|
|
if (hs != test.output) {
|
|
res = emsha::EMSHAResult::TestFailure;
|
|
goto exit;
|
|
}
|
|
memset(dig, 0, emsha::SHA256_HASH_SIZE);
|
|
}
|
|
|
|
// Test that the single-pass function works.
|
|
res = emsha::ComputeHMAC(test.key, test.keylen,
|
|
(uint8_t *)test.input.c_str(), test.input.size(),
|
|
dig);
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
cerr << "(running single pass function test)\n";
|
|
goto exit;
|
|
}
|
|
|
|
DumpHexString(hs, dig, emsha::SHA256_HASH_SIZE);
|
|
if (hs != test.output) {
|
|
cerr << "(comparing single pass function output)\n";
|
|
res = emsha::EMSHAResult::TestFailure;
|
|
goto exit;
|
|
}
|
|
memset(dig, 0, emsha::SHA256_HASH_SIZE);
|
|
|
|
res = emsha::EMSHAResult::OK;
|
|
|
|
exit:
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
cerr << "FAILED: " << label << endl;
|
|
cerr << "\tinput: " << test.input << endl;
|
|
cerr << "\twanted: " << test.output << endl;
|
|
cerr << "\thave: " << hs << endl;
|
|
}
|
|
|
|
return res;
|
|
}
|
|
|
|
|
|
int
|
|
runHMACTests(const struct hmacTest *tests, size_t nTests, const string& label)
|
|
{
|
|
for (uint32_t i = 0; i < nTests; i++) {
|
|
if (emsha::EMSHAResult::OK != runHMACTest(*(tests + i), label)) {
|
|
return -1;
|
|
}
|
|
}
|
|
cout << "PASSED: " << label << " (" << nTests << ")" << endl;
|
|
return 0;
|
|
}
|
|
|
|
|
|
emsha::EMSHAResult
|
|
runHashTest(const struct hashTest& test, const string& label)
|
|
{
|
|
emsha::SHA256 ctx;
|
|
emsha::EMSHAResult res;
|
|
uint8_t dig[emsha::SHA256_HASH_SIZE];
|
|
string hs;
|
|
|
|
res = ctx.Update((uint8_t *)test.input.c_str(), test.input.size());
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
for (uint32_t n = 0; n < RESULT_ITERATIONS; n++) {
|
|
res = ctx.Result(dig);
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
DumpHexString(hs, dig, emsha::SHA256_HASH_SIZE);
|
|
if (hs != test.output) {
|
|
res = emsha::EMSHAResult::TestFailure;
|
|
goto exit;
|
|
}
|
|
memset(dig, 0, emsha::SHA256_HASH_SIZE);
|
|
}
|
|
|
|
// Ensure that a reset and update gives the same results.
|
|
ctx.Reset();
|
|
|
|
res = ctx.Update((uint8_t *)test.input.c_str(), test.input.size());
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
for (uint32_t n = 0; n < RESULT_ITERATIONS; n++) {
|
|
res = ctx.Result(dig);
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
goto exit;
|
|
}
|
|
|
|
DumpHexString(hs, dig, emsha::SHA256_HASH_SIZE);
|
|
if (hs != test.output) {
|
|
res = emsha::EMSHAResult::TestFailure;
|
|
goto exit;
|
|
}
|
|
memset(dig, 0, emsha::SHA256_HASH_SIZE);
|
|
}
|
|
|
|
// Test that the single-pass function works.
|
|
res = emsha::SHA256Digest((uint8_t *) test.input.c_str(),
|
|
test.input.size(), dig);
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
cerr << "(running single pass function test)\n";
|
|
goto exit;
|
|
}
|
|
|
|
DumpHexString(hs, dig, emsha::SHA256_HASH_SIZE);
|
|
if (hs != test.output) {
|
|
cerr << "(comparing single pass function output)\n";
|
|
res = emsha::EMSHAResult::TestFailure;
|
|
goto exit;
|
|
}
|
|
memset(dig, 0, emsha::SHA256_HASH_SIZE);
|
|
res = emsha::EMSHAResult::OK;
|
|
|
|
exit:
|
|
if (emsha::EMSHAResult::OK != res) {
|
|
cerr << "FAILED: " << label << endl;
|
|
cerr << "\tinput: '" << test.input << "'" << endl;
|
|
cerr << "\twanted: " << test.output << endl;
|
|
cerr << "\thave: " << hs << endl;
|
|
}
|
|
return res;
|
|
}
|
|
|
|
|
|
int
|
|
runHashTests(const struct hashTest *tests, const size_t ntests, const string& label)
|
|
{
|
|
for (uint32_t i = 0; i < ntests; i++) {
|
|
if (emsha::EMSHAResult::OK != runHashTest(*(tests + i), label)) {
|
|
return -1;
|
|
}
|
|
}
|
|
cout << "PASSED: " << label << " (" << ntests << ")" << endl;
|
|
return 0;
|
|
}
|
|
|
|
|
|
#ifdef EMSHA_NO_HEXSTRING
|
|
|
|
// If the library was built without hex string support, include it in
|
|
// the test code.
|
|
namespace emsha {
|
|
static void
|
|
write_hex_char(uint8_t *dest, uint8_t src)
|
|
{
|
|
static constexpr uint8_t lut[256][3] = {
|
|
"00", "01", "02", "03", "04", "05", "06", "07",
|
|
"08", "09", "0a", "0b", "0c", "0d", "0e", "0f",
|
|
"10", "11", "12", "13", "14", "15", "16", "17",
|
|
"18", "19", "1a", "1b", "1c", "1d", "1e", "1f",
|
|
"20", "21", "22", "23", "24", "25", "26", "27",
|
|
"28", "29", "2a", "2b", "2c", "2d", "2e", "2f",
|
|
"30", "31", "32", "33", "34", "35", "36", "37",
|
|
"38", "39", "3a", "3b", "3c", "3d", "3e", "3f",
|
|
"40", "41", "42", "43", "44", "45", "46", "47",
|
|
"48", "49", "4a", "4b", "4c", "4d", "4e", "4f",
|
|
"50", "51", "52", "53", "54", "55", "56", "57",
|
|
"58", "59", "5a", "5b", "5c", "5d", "5e", "5f",
|
|
"60", "61", "62", "63", "64", "65", "66", "67",
|
|
"68", "69", "6a", "6b", "6c", "6d", "6e", "6f",
|
|
"70", "71", "72", "73", "74", "75", "76", "77",
|
|
"78", "79", "7a", "7b", "7c", "7d", "7e", "7f",
|
|
"80", "81", "82", "83", "84", "85", "86", "87",
|
|
"88", "89", "8a", "8b", "8c", "8d", "8e", "8f",
|
|
"90", "91", "92", "93", "94", "95", "96", "97",
|
|
"98", "99", "9a", "9b", "9c", "9d", "9e", "9f",
|
|
"a0", "a1", "a2", "a3", "a4", "a5", "a6", "a7",
|
|
"a8", "a9", "aa", "ab", "ac", "ad", "ae", "af",
|
|
"b0", "b1", "b2", "b3", "b4", "b5", "b6", "b7",
|
|
"b8", "b9", "ba", "bb", "bc", "bd", "be", "bf",
|
|
"c0", "c1", "c2", "c3", "c4", "c5", "c6", "c7",
|
|
"c8", "c9", "ca", "cb", "cc", "cd", "ce", "cf",
|
|
"d0", "d1", "d2", "d3", "d4", "d5", "d6", "d7",
|
|
"d8", "d9", "da", "db", "dc", "dd", "de", "df",
|
|
"e0", "e1", "e2", "e3", "e4", "e5", "e6", "e7",
|
|
"e8", "e9", "ea", "eb", "ec", "ed", "ee", "ef",
|
|
"f0", "f1", "f2", "f3", "f4", "f5", "f6", "f7",
|
|
"f8", "f9", "fa", "fb", "fc", "fd", "fe", "ff"
|
|
};
|
|
|
|
memcpy(dest, lut[src], 2);
|
|
}
|
|
|
|
|
|
void
|
|
HexString(uint8_t *dest, uint8_t *src, uint32_t srclen)
|
|
{
|
|
uint8_t *dp = dest;
|
|
|
|
for (uint32_t i = 0; i < srclen; i++, dp += 2) {
|
|
write_hex_char(dp, src[i]);
|
|
}
|
|
}
|
|
|
|
|
|
}
|
|
|
|
#endif // EMSHA_NO_HEXSTRING
|