-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Bug 1345089 - add prng kat tests, r=ttaubert
Differential Revision: https://nss-review.dev.mozaws.net/D237 --HG-- extra : rebase_source : acfb8b83f674211a6495c0aa230c30ceca81dcc2 extra : amend_source : db848c7b99d077a9ab15fcca72c1000281035349 extra : histedit_source : f18c033c425051aa40fa98967cad794bc012c030%2Cf3e94a7be307767cbfd3465ec60dc1c9d72ff87b
- Loading branch information
1 parent
bf0c68c
commit 8533ddd
Showing
10 changed files
with
62,550 additions
and
16 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Large diffs are not rendered by default.
Oops, something went wrong.
Large diffs are not rendered by default.
Oops, something went wrong.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,195 @@ | ||
| // This Source Code Form is subject to the terms of the Mozilla Public | ||
| // License, v. 2.0. If a copy of the MPL was not distributed with this file, | ||
| // You can obtain one at http://mozilla.org/MPL/2.0/. | ||
|
|
||
| #include "nspr.h" | ||
| #include "nss.h" | ||
| #include "ssl.h" | ||
|
|
||
| #include <cstdlib> | ||
| #include <fstream> | ||
| #include <string> | ||
|
|
||
| #define GTEST_HAS_RTTI 0 | ||
| #include "gtest/gtest.h" | ||
|
|
||
| #include "blapi.h" | ||
|
|
||
| namespace nss_test { | ||
|
|
||
| typedef struct PRNGTestValuesStr { | ||
| std::vector<uint8_t> entropy; | ||
| std::vector<uint8_t> nonce; | ||
| std::vector<uint8_t> personal; | ||
| std::vector<uint8_t> expected_result; | ||
| std::vector<uint8_t> additional_entropy; | ||
| std::vector<uint8_t> additional_input_reseed; | ||
| std::vector<std::vector<uint8_t>> additional_input; | ||
| } PRNGTestValues; | ||
|
|
||
| std::vector<PRNGTestValues> test_vector; | ||
|
|
||
| bool contains(std::string& s, const char* to_find) { | ||
| return s.find(to_find) != std::string::npos; | ||
| } | ||
|
|
||
| std::string trim(std::string str) { | ||
| std::string whitespace = " \t"; | ||
| const auto strBegin = str.find_first_not_of(whitespace); | ||
| if (strBegin == std::string::npos) { | ||
| return ""; | ||
| } | ||
| const auto strEnd = str.find_last_not_of(whitespace); | ||
| const auto strRange = strEnd - strBegin + 1; | ||
| return str.substr(strBegin, strRange); | ||
| } | ||
|
|
||
| std::vector<uint8_t> hex_string_to_bytes(std::string s) { | ||
| std::vector<uint8_t> bytes; | ||
| for (size_t i = 0; i < s.length() - 1; i += 2) { | ||
| bytes.push_back(std::stoul(s.substr(i, 2), nullptr, 16)); | ||
| } | ||
| return bytes; | ||
| } | ||
|
|
||
| std::vector<uint8_t> read_option_s(std::string& s) { | ||
| size_t start = s.find("=") + 1; | ||
| assert(start > 0); | ||
| return hex_string_to_bytes(trim(s.substr(start, s.find("]", start)))); | ||
| } | ||
|
|
||
| void print_bytes(std::vector<uint8_t> bytes, std::string name) { | ||
| std::cout << name << ": "; | ||
| for (auto b : bytes) { | ||
| std::cout << std::setfill('0') << std::setw(2) << std::hex | ||
| << static_cast<int>(b); | ||
| } | ||
| std::cout << std::endl; | ||
| } | ||
|
|
||
| static void ReadFile(const std::string file_name) { | ||
| std::ifstream infile(file_name); | ||
| std::string line; | ||
|
|
||
| // Variables holding the input for each test. | ||
| bool valid_option = false; | ||
|
|
||
| // Read the file. | ||
| std::streampos pos; | ||
| while (std::getline(infile, line)) { | ||
| // We only implement SHA256. Skip all other tests. | ||
| if (contains(line, "[SHA-")) { | ||
| valid_option = contains(line, "[SHA-256]"); | ||
| } | ||
| if (!valid_option) { | ||
| continue; | ||
| } | ||
|
|
||
| // We ignore the options and infer them from the test case. | ||
|
|
||
| PRNGTestValues test; | ||
| if (line.find("COUNT =")) { | ||
| continue; | ||
| } | ||
|
|
||
| // Read test input. | ||
| do { | ||
| pos = infile.tellg(); | ||
| std::getline(infile, line); | ||
| if (contains(line, "EntropyInput ")) { | ||
| test.entropy = read_option_s(line); | ||
| continue; | ||
| } | ||
| if (contains(line, "Nonce")) { | ||
| test.nonce = read_option_s(line); | ||
| continue; | ||
| } | ||
| if (contains(line, "PersonalizationString")) { | ||
| test.personal = read_option_s(line); | ||
| continue; | ||
| } | ||
| if (contains(line, "AdditionalInput ")) { | ||
| test.additional_input.push_back(read_option_s(line)); | ||
| continue; | ||
| } | ||
| if (contains(line, "EntropyInputReseed")) { | ||
| test.additional_entropy = read_option_s(line); | ||
| continue; | ||
| } | ||
| if (contains(line, "AdditionalInputReseed")) { | ||
| test.additional_input_reseed = read_option_s(line); | ||
| continue; | ||
| } | ||
| if (contains(line, "ReturnedBits")) { | ||
| test.expected_result = read_option_s(line); | ||
| continue; | ||
| } | ||
| } while (!infile.eof() && line.find("COUNT =") && line.find("[")); | ||
|
|
||
| // Save test case. | ||
| test_vector.push_back(test); | ||
| test = {}; | ||
| infile.seekg(pos); | ||
| } | ||
| } | ||
|
|
||
| class PRNGTest : public ::testing::TestWithParam<PRNGTestValues> { | ||
| protected: | ||
| void RunTest(PRNGTestValues test) { | ||
| ASSERT_EQ(2U, test.additional_input.size()); | ||
| SECStatus rv = PRNGTEST_Instantiate_Kat( | ||
| test.entropy.data(), test.entropy.size(), test.nonce.data(), | ||
| test.nonce.size(), test.personal.data(), test.personal.size()); | ||
| ASSERT_EQ(SECSuccess, rv); | ||
| rv = PRNGTEST_Reseed(test.additional_entropy.data(), | ||
| test.additional_entropy.size(), | ||
| test.additional_input_reseed.data(), | ||
| test.additional_input_reseed.size()); | ||
| ASSERT_EQ(SECSuccess, rv); | ||
|
|
||
| // Generate bytes. | ||
| uint8_t bytes[128]; | ||
| PRNGTEST_Generate(bytes, 128, test.additional_input[0].data(), | ||
| test.additional_input[0].size()); | ||
| PRNGTEST_Generate(bytes, 128, test.additional_input[1].data(), | ||
| test.additional_input[1].size()); | ||
| std::vector<uint8_t> result(bytes, bytes + 128); | ||
| if (result != test.expected_result) { | ||
| print_bytes(result, "result "); | ||
| print_bytes(test.expected_result, "expected"); | ||
| } | ||
| ASSERT_EQ(test.expected_result, result); | ||
| rv = PRNGTEST_Uninstantiate(); | ||
| ASSERT_EQ(SECSuccess, rv); | ||
| } | ||
| }; | ||
|
|
||
| TEST_P(PRNGTest, HashDRBG) { RunTest(GetParam()); } | ||
|
|
||
| INSTANTIATE_TEST_CASE_P(NISTTestVector, PRNGTest, | ||
| ::testing::ValuesIn(test_vector)); | ||
|
|
||
| } // nss_test | ||
|
|
||
| int main(int argc, char** argv) { | ||
| if (argc < 2) { | ||
| std::cout << "usage: prng_gtest <.rsp file>" << std::endl; | ||
| return 1; | ||
| } | ||
|
|
||
| nss_test::ReadFile(argv[1]); | ||
| assert(!nss_test::test_vector.empty()); | ||
|
|
||
| ::testing::InitGoogleTest(&argc, argv); | ||
|
|
||
| if (NSS_NoDB_Init(nullptr) != SECSuccess) { | ||
| return 1; | ||
| } | ||
| int rv = RUN_ALL_TESTS(); | ||
|
|
||
| if (NSS_Shutdown() != SECSuccess) { | ||
| return 1; | ||
| } | ||
|
|
||
| return rv; | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters