boringssl/crypto/evp/scrypt_test.cc
David Benjamin b529253bea Implement scrypt from RFC 7914.
This imports upstream's scrypt implementation, though it's been heavily
revised. I lost track of words vs. blocks vs. bigger blocks too many
times in the original code and introduced a typedef for the fixed-width
Salsa20 blocks. The downside is going from bytes to blocks is a bit
trickier, so I took advantage of our little-endian assumption.

This also adds an missing check for N < 2^32. Upstream's code is making
this assumption in Integerify. I'll send that change back upstream. I've
also removed the weird edge case where a NULL out_key parameter means to
validate N/r/p against max_mem and nothing else. That's just in there to
get a different error code out of their PKCS#12 code.

Performance-wise, the cleanup appears to be the same (up to what little
precision I was able to get here), but an optimization to use bitwise
AND rather than modulus makes us measurably faster. Though scrypt isn't
a fast operation to begin with, so hopefully it isn't anyone's
bottleneck.

This CL does not route scrypt up to the PKCS#12 code, though we could
write our own version of that if we need to later.

BUG=chromium:731993

Change-Id: Ib2f43344017ed37b6bafd85a2c2b103d695020b8
Reviewed-on: https://boringssl-review.googlesource.com/17084
Reviewed-by: Adam Langley <agl@google.com>
2017-06-12 20:32:21 +00:00

104 lines
4.2 KiB
C++

/* Copyright (c) 2017, Google Inc.
*
* Permission to use, copy, modify, and/or distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
* SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
* OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
* CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
#include <stdlib.h>
#include <string.h>
#include <vector>
#include <gtest/gtest.h>
#include <openssl/err.h>
#include <openssl/evp.h>
#include "../test/file_test.h"
#include "../test/test_util.h"
static bool GetUint64(FileTest *t, uint64_t *out, const char *name) {
std::string str;
if (!t->GetAttribute(&str, name)) {
return false;
}
char *endptr;
*out = strtoull(str.data(), &endptr, 10);
return !str.empty() && *endptr == '\0';
}
TEST(ScryptTest, TestVectors) {
FileTestGTest("crypto/evp/scrypt_tests.txt", [](FileTest *t) {
std::vector<uint8_t> password, salt, key;
uint64_t N, r, p, max_mem = 0;
ASSERT_TRUE(t->GetBytes(&password, "Password"));
ASSERT_TRUE(t->GetBytes(&salt, "Salt"));
ASSERT_TRUE(t->GetBytes(&key, "Key"));
ASSERT_TRUE(GetUint64(t, &N, "N"));
ASSERT_TRUE(GetUint64(t, &r, "r"));
ASSERT_TRUE(GetUint64(t, &p, "p"));
if (t->HasAttribute("MaxMemory")) {
ASSERT_TRUE(GetUint64(t, &max_mem, "MaxMemory"));
}
std::vector<uint8_t> result(key.size());
ASSERT_TRUE(EVP_PBE_scrypt(reinterpret_cast<const char *>(password.data()),
password.size(), salt.data(), salt.size(), N, r,
p, max_mem, result.data(), result.size()));
EXPECT_EQ(Bytes(key), Bytes(result));
});
}
TEST(ScryptTest, MemoryLimit) {
static const char kPassword[] = "pleaseletmein";
static const char kSalt[] = "SodiumChloride";
// This test requires more than 1GB to run.
uint8_t key[64];
EXPECT_FALSE(EVP_PBE_scrypt(kPassword, strlen(kPassword),
reinterpret_cast<const uint8_t *>(kSalt),
strlen(kSalt), 1048576 /* N */, 8 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
uint32_t err = ERR_get_error();
EXPECT_EQ(ERR_LIB_EVP, ERR_GET_LIB(err));
EXPECT_EQ(EVP_R_MEMORY_LIMIT_EXCEEDED, ERR_GET_REASON(err));
}
TEST(ScryptTest, InvalidParameters) {
uint8_t key[64];
// p and r are non-zero.
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 1024 /* N */, 0 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 1024 /* N */, 8 /* r */,
0 /* p */, 0 /* max_mem */, key, sizeof(key)));
// N must be a power of 2 > 1.
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 0 /* N */, 8 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 1 /* N */, 8 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 1023 /* N */, 8 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
EXPECT_TRUE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 1024 /* N */, 8 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 1025 /* N */, 8 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
// N must be below 2^(128 * r / 8).
EXPECT_FALSE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 65536 /* N */, 1 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
EXPECT_TRUE(EVP_PBE_scrypt(nullptr, 0, nullptr, 0, 32768 /* N */, 1 /* r */,
1 /* p */, 0 /* max_mem */, key, sizeof(key)));
}