From f5ea2848469338c936e6e04da97cc6ee4b30e92c Mon Sep 17 00:00:00 2001
From: SChernykh <sergey.v.chernykh@gmail.com>
Date: Sat, 4 Sep 2021 21:22:28 +0200
Subject: [PATCH] Added keccak tests

---
 tests/CMakeLists.txt                          |  5 +-
 ...lty_type.cpp => difficulty_type_tests.cpp} |  0
 tests/src/keccak_tests.cpp                    | 58 +++++++++++++++++++
 tests/src/{tests.cpp => main.cpp}             |  0
 4 files changed, 61 insertions(+), 2 deletions(-)
 rename tests/src/{difficulty_type.cpp => difficulty_type_tests.cpp} (100%)
 create mode 100644 tests/src/keccak_tests.cpp
 rename tests/src/{tests.cpp => main.cpp} (100%)

diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index 25d122a..791c6ae 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -51,8 +51,9 @@ set(HEADERS
 )
 
 set(SOURCES
-	src/difficulty_type.cpp
-	src/tests.cpp
+	src/difficulty_type_tests.cpp
+	src/keccak_tests.cpp
+	src/main.cpp
 	src/wallet_tests.cpp
 	../external/src/cryptonote/crypto-ops-data.c
 	../external/src/cryptonote/crypto-ops.c
diff --git a/tests/src/difficulty_type.cpp b/tests/src/difficulty_type_tests.cpp
similarity index 100%
rename from tests/src/difficulty_type.cpp
rename to tests/src/difficulty_type_tests.cpp
diff --git a/tests/src/keccak_tests.cpp b/tests/src/keccak_tests.cpp
new file mode 100644
index 0000000..4f22cca
--- /dev/null
+++ b/tests/src/keccak_tests.cpp
@@ -0,0 +1,58 @@
+/*
+ * This file is part of the Monero P2Pool <https://github.com/SChernykh/p2pool>
+ * Copyright (c) 2021 SChernykh <https://github.com/SChernykh>
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, version 3.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program. If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "common.h"
+#include "keccak.h"
+#include "gtest/gtest.h"
+
+namespace p2pool {
+
+TEST(keccak, hashing)
+{
+	auto check = [](const void* input, size_t size, const char* expected_output) {
+		hash output;
+		const uint8_t* data = reinterpret_cast<const uint8_t*>(input);
+		const int len = static_cast<int>(size);
+		keccak(data, len, output.h, HASH_SIZE);
+
+		char buf[log::Stream::BUF_SIZE + 1];
+		log::Stream s(buf);
+		s << output;
+		ASSERT_EQ(memcmp(buf, expected_output, HASH_SIZE * 2), 0);
+
+		memset(output.h, 0, HASH_SIZE);
+		memset(buf, 0, sizeof(buf));
+		s.m_pos = 0;
+
+		keccak_custom([data](int offset) { return data[offset]; }, size, output.h, HASH_SIZE);
+		s << output;
+		ASSERT_EQ(memcmp(buf, expected_output, HASH_SIZE * 2), 0);
+	};
+
+	check("", 0, "c5d2460186f7233c927e7db2dcc703c0e500b653ca82273b7bfad8045d85a470");
+	check("\xcc", 1, "eead6dbfc7340a56caedc044696a168870549a6a7f6f56961e84a54bd9970b8a");
+	check("\x41\xfb", 2, "a8eaceda4d47b3281a795ad9e1ea2122b407baf9aabcb9e18b5717b7873537d2");
+	check("\xc1\xec\xfd\xfc", 4, "b149e766d7612eaf7d55f74e1a4fdd63709a8115b14f61fcd22aa4abc8b8e122");
+
+	const uint8_t data[] = { 0x9f, 0x2f, 0xcc, 0x7c, 0x90, 0xde, 0x09, 0x0d, 0x6b, 0x87, 0xcd, 0x7e, 0x97, 0x18, 0xc1, 0xea, 0x6c, 0xb2, 0x11, 0x18, 0xfc, 0x2d, 0x5d, 0xe9, 0xf9, 0x7e, 0x5d, 0xb6, 0xac, 0x1e, 0x9c, 0x10 };
+	check(data, sizeof(data), "24dd2ee02482144f539f810d2caa8a7b75d0fa33657e47932122d273c3f6f6d1");
+
+	std::vector<uint8_t> v(1000000, 'a');
+	check(v.data(), v.size(), "fadae6b49f129bbb812be8407b7b2894f34aecf6dbd1f9b0f0c7e9853098fc96");
+}
+
+}
diff --git a/tests/src/tests.cpp b/tests/src/main.cpp
similarity index 100%
rename from tests/src/tests.cpp
rename to tests/src/main.cpp