From f5106362b82b035dcd8e487d755316e886de3440 Mon Sep 17 00:00:00 2001
From: gcontini <1121667+gcontini@users.noreply.github.com>
Date: 周六, 23 11月 2019 22:48:18 +0800
Subject: [PATCH] signature verified, tests

---
 test/functional/volid_test.cpp |  220 ++++++++++++++++++++++++++++++++----------------------
 1 files changed, 130 insertions(+), 90 deletions(-)

diff --git a/test/functional/volid_test.cpp b/test/functional/volid_test.cpp
index 501c815..71bcb02 100644
--- a/test/functional/volid_test.cpp
+++ b/test/functional/volid_test.cpp
@@ -1,105 +1,145 @@
-#define BOOST_TEST_MODULE standard_license_test
-//#define BOOST_TEST_MAIN
-#define BOOST_TEST_DYN_LINK
+#define BOOST_TEST_MODULE test_volid
+
 #include <boost/test/unit_test.hpp>
 #include <fstream>
+#include <iostream>
+#include <stdio.h>
 #include <cstring>
-#include "../../../src/license-generator/license-generator.h"
-#include "../../../src/library/api/license++.h"
-#include <build_properties.h>
 #include <boost/filesystem.hpp>
+#include <licensecc_properties.h>
+#include <licensecc_properties_test.h>
+
+#include <licensecc/licensecc.h>
 #include "../../src/library/ini/SimpleIni.h"
-#include "generate-license.h"
 #include "../../src/library/pc-identifiers.h"
-#include <dbus/dbus.h>
+#include "../../src/library/os/os.h"
+#include "generate-license.h"
 
 namespace fs = boost::filesystem;
 using namespace license;
 using namespace std;
-/*
- BOOST_AUTO_TEST_CASE( default_volid_lic_file ) {
- const string licLocation(PROJECT_TEST_TEMP_DIR "/volid_license.lic");
- PcSignature identifier_out;
 
- IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET;
- FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out,
- strategy);
- BOOST_ASSERT(generate_ok == FUNCTION_RETURN::OK);
- cout << "Identifier:" << identifier_out << endl;
- vector<string> extraArgs = { "-s", identifier_out };
- generate_license(licLocation, extraArgs);
+namespace license {
+namespace test {
 
- LicenseInfo license;
- LicenseLocation licenseLocation;
- licenseLocation.openFileNearModule = false;
- licenseLocation.licenseFileLocation = licLocation.c_str();
- licenseLocation.environmentVariableName = "";
- EVENT_TYPE result = acquire_license("TEST", licenseLocation, &license);
- BOOST_CHECK_EQUAL(result, LICENSE_OK);
- BOOST_CHECK_EQUAL(license.has_expiry, false);
- BOOST_CHECK_EQUAL(license.linked_to_pc, true);
- }
+BOOST_AUTO_TEST_CASE( default_volid_lic_file ) {
+	PcSignature identifier_out;
 
- static void generate_reference_file(const string& idfileLocation,
- IDENTIFICATION_STRATEGY strategies[], int num_strategies) {
- ofstream idfile(idfileLocation);
- PcSignature identifier_out;
- for (int i = 0; i < num_strategies; i++) {
- FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out,
- strategies[i]);
- BOOST_ASSERT(generate_ok == FUNCTION_RETURN::OK);
- idfile << identifier_out << endl;
- }
- idfile.close();
- }
+	const IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET;
+	BOOST_TEST_CHECKPOINT("Before generate");
+	const FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out,
+			strategy);
+	BOOST_TEST_CHECKPOINT("After generate signature");
+	BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK);
+	cout << "Identifier:" << identifier_out << endl;
+	vector<string> extraArgs;
+	extraArgs.push_back("-s");
+	extraArgs.push_back(identifier_out);
+	BOOST_TEST_CHECKPOINT("Before generate license");
+	const string licLocation = generate_license("volid_license", extraArgs);
 
- BOOST_AUTO_TEST_CASE(generated_identifiers_stability) {
- const string idfileLocation(PROJECT_TEST_TEMP_DIR "/identifiers_file");
- IDENTIFICATION_STRATEGY strategies[] =
- { IDENTIFICATION_STRATEGY::DEFAULT,
- IDENTIFICATION_STRATEGY::DISK_LABEL,
- IDENTIFICATION_STRATEGY::DISK_NUM,
- IDENTIFICATION_STRATEGY::ETHERNET
- };
- const int num_strategies = sizeof(strategies) / sizeof(strategies[0]);
- std::ifstream test_idfile_exist(idfileLocation);
- if (!test_idfile_exist.good()) {
- generate_reference_file(idfileLocation, strategies, num_strategies);
- }
- std::ifstream is(idfileLocation);
- std::istream_iterator<string> start(is), end;
- std::vector<string> reference_signatures(start, end);
- BOOST_ASSERT(reference_signatures.size() == num_strategies);
- PcSignature generated_identifier;
- BOOST_CHECKPOINT("Generating current signatures and comparing with past");
- for (int i = 0; i < num_strategies; i++) {
- FUNCTION_RETURN generate_ok = generate_user_pc_signature(
- generated_identifier, strategies[i]);
- BOOST_ASSERT(generate_ok == FUNCTION_RETURN::OK);
- if (reference_signatures[i] != generated_identifier) {
- string message = string("pc signature compare fail: strategy:")
- + to_string(strategies[i]) + " generated: ["
- + generated_identifier + "] reference: ["
- + reference_signatures[i] + "]";
- BOOST_FAIL(message);
- }
- }
-
- BOOST_CHECKPOINT("Verifying signatures");
- for (int j = 0; j < 100; j++) {
- for (unsigned int i = 0; i < reference_signatures.size(); i++) {
- PcSignature pcsig;
- strncpy(pcsig, reference_signatures[i].c_str(), sizeof(PcSignature));
- EVENT_TYPE val_result = validate_pc_signature(pcsig);
- string message = string("pc signature verification strategy:")
- + to_string(i) + " generated: [" + generated_identifier
- + "] reference: [" + reference_signatures[i] + "]";
- BOOST_CHECKPOINT("Verifying signature: ");
- BOOST_CHECK_EQUAL(val_result, LICENSE_OK);
- }
- }
- }
- */
-BOOST_AUTO_TEST_CASE( ciiii ) {
-	cout << dbus_get_local_machine_id() << endl;
+	LicenseInfo license;
+	LicenseLocation licenseLocation;
+	licenseLocation.licenseFileLocation = licLocation.c_str();
+	licenseLocation.licenseData = "";
+	const EVENT_TYPE result = acquire_license(nullptr, &licenseLocation, &license);
+	BOOST_CHECK_EQUAL(result, LICENSE_OK);
+	BOOST_CHECK_EQUAL(license.has_expiry, false);
+	BOOST_CHECK_EQUAL(license.linked_to_pc, true);
 }
+
+static void generate_reference_file(const string &idfileLocation,
+		IDENTIFICATION_STRATEGY strategies[], int num_strategies) {
+	ofstream idfile(idfileLocation);
+	PcSignature identifier_out;
+	for (int i = 0; i < num_strategies; i++) {
+		FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out,
+				strategies[i]);
+		if (generate_ok != FUNC_RET_OK) {
+			BOOST_ERROR(
+					"Generating identifier for strategy " << strategies[i] << " failed with: " << generate_ok);
+			idfile << "0000-0000-0000-0000" << endl;
+			BOOST_ASSERT(generate_ok == FUNC_RET_OK);
+		} else
+			idfile << identifier_out << endl;
+	}
+	idfile.close();
+}
+
+BOOST_AUTO_TEST_CASE(generated_identifiers_stability) {
+	const string idfileLocation(PROJECT_TEST_TEMP_DIR "/identifiers_file");
+	std::vector<IDENTIFICATION_STRATEGY> strategies;
+	size_t disk_num;
+	getDiskInfos(NULL, &disk_num);
+	if (disk_num >0) {
+		strategies = { DEFAULT, DISK_NUM, DISK_LABEL };
+	} else {
+		BOOST_TEST_CHECKPOINT("if no disk default strategy fails see #49");
+		//strategies = { DEFAULT };
+		strategies = {};
+	}
+	size_t adapters;
+	getAdapterInfos(nullptr, &adapters);
+	if(adapters > 0){
+		strategies.push_back(ETHERNET);
+	}
+
+	size_t num_strategies = strategies.size();
+	if(num_strategies == 0) {
+		//see issue #49 can't use default
+		return;
+	}
+	std::ifstream test_idfile_exist(idfileLocation);
+	if (!test_idfile_exist.good()) {
+		generate_reference_file(idfileLocation, strategies.data(),
+				strategies.size());
+	} else {
+		std::istream_iterator<string> start(test_idfile_exist), end;
+		std::vector<string> reference_signatures(start, end);
+		test_idfile_exist.close();
+		if (reference_signatures.size() != num_strategies
+				|| std::find(reference_signatures.begin(),
+						reference_signatures.end(), "0000-0000-0000-0000")
+						!= reference_signatures.end())
+			generate_reference_file(idfileLocation, strategies.data(), num_strategies);
+	}
+	std::ifstream is(idfileLocation);
+	std::istream_iterator<string> start(is), end;
+	std::vector<string> reference_signatures(start, end);
+	BOOST_TEST_CHECKPOINT(
+			"Generating current signatures and comparing with past");
+	for (int i = 0; i < num_strategies; i++) {
+		PcSignature generated_identifier;
+		FUNCTION_RETURN generate_ok = generate_user_pc_signature(
+				generated_identifier, strategies[i]);
+		BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK);
+		if (generate_ok != FUNC_RET_OK) {
+			BOOST_ERROR(
+					"Generating identifier for strategy " << strategies[i] << " failed with: " << generate_ok);
+			continue;
+		}
+		if (reference_signatures[i] != generated_identifier) {
+			string message = string("pc signature compare fail: strategy: ")
+					+ to_string(static_cast<long long>(strategies[i]))
+					+ " generated: [" + generated_identifier + "] reference: ["
+					+ reference_signatures[i] + "]";
+			BOOST_ERROR(message);
+		}
+	}
+	BOOST_TEST_CHECKPOINT("Verifying signatures");
+	for (int j = 0; j < 100; j++) {
+		for (unsigned int i = 0; i < reference_signatures.size(); i++) {
+			if (reference_signatures[i] == "0000-0000-0000-0000")
+				continue;
+			PcSignature pcsig;
+			strncpy(pcsig, reference_signatures[i].c_str(),
+					sizeof(PcSignature) - 1);
+			EVENT_TYPE val_result = validate_pc_signature(pcsig);
+			BOOST_TEST_CHECKPOINT("Verifying signature: ");
+			BOOST_CHECK_EQUAL(val_result, LICENSE_OK);
+		}
+	}
+}
+
+}  // namespace test
+}  // namespace license

--
Gitblit v1.9.1