From f91f5aae33291486780c9c03d4f2a1b94a89ba06 Mon Sep 17 00:00:00 2001
From: Jan Willamowius <jan@willamowius.de>
Date: 周日, 21 4月 2019 23:58:42 +0800
Subject: [PATCH] leave room fr terminating nul byte
---
test/functional/volid_test.cpp | 78 +++++++++++++++++++-------------------
1 files changed, 39 insertions(+), 39 deletions(-)
diff --git a/test/functional/volid_test.cpp b/test/functional/volid_test.cpp
index 642f361..2de9a04 100644
--- a/test/functional/volid_test.cpp
+++ b/test/functional/volid_test.cpp
@@ -5,8 +5,8 @@
#include <fstream>
#include <stdio.h>
#include <cstring>
-#include "../../../src/tools/license-generator/license-generator.h"
-#include "../../../src/library/api/license++.h"
+#include "../../src/tools/license-generator/license-generator.h"
+#include "../../src/library/api/license++.h"
#include <build_properties.h>
#include <boost/filesystem.hpp>
#include "../../src/library/ini/SimpleIni.h"
@@ -22,12 +22,14 @@
PcSignature identifier_out;
IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET;
- BOOST_CHECKPOINT("Before generate");
+ BOOST_TEST_CHECKPOINT("Before generate");
FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out,
strategy);
BOOST_ASSERT(generate_ok == FUNCTION_RETURN::FUNC_RET_OK);
cout << "Identifier:" << identifier_out << endl;
- vector<string> extraArgs = { "-s", identifier_out };
+ vector<string> extraArgs;
+ extraArgs.push_back("-s");
+ extraArgs.push_back(identifier_out);
generate_license(licLocation, extraArgs);
LicenseInfo license;
@@ -48,13 +50,13 @@
for (int i = 0; i < num_strategies; i++) {
FUNCTION_RETURN generate_ok = generate_user_pc_signature(identifier_out,
strategies[i]);
+ BOOST_ASSERT(generate_ok == FUNC_RET_OK);
if (generate_ok != FUNC_RET_OK){
- idfile.close();
- remove(idfileLocation.c_str());
- BOOST_ERROR("Generating identifier for strategy " << strategies[i] << "failed with: " << generate_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);
- idfile << identifier_out << endl;
+ else
+ idfile << identifier_out << endl;
}
idfile.close();
}
@@ -68,50 +70,48 @@
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);
- }
- //try to locate a "good" reference file.
- int tries = 2;
- std::vector<string> reference_signatures;
- do{
- std::ifstream is(idfileLocation);
- std::istream_iterator<string> start(is), end;
- reference_signatures = vector<string>(start, end);
- if (reference_signatures.size() != num_strategies){
- is.close();
- remove(idfileLocation.c_str());
- generate_reference_file(idfileLocation, strategies, num_strategies);
- }
- } while (reference_signatures.size() != num_strategies && tries-- > 0);
- BOOST_ASSERT(tries > 0);
-
- PcSignature generated_identifier;
- BOOST_CHECKPOINT("Generating current signatures and comparing with past");
+ if (!test_idfile_exist.good()){
+ generate_reference_file(idfileLocation, strategies, num_strategies);
+ }
+ 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, 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((long double) strategies[i]) + " generated: ["
+ string message = string("pc signature compare fail: strategy: ")
+ + to_string(static_cast<long long>(strategies[i])) + " generated: ["
+ generated_identifier + "] reference: ["
+ reference_signatures[i] + "]";
- BOOST_FAIL(message);
+ BOOST_ERROR(message);
}
}
-
- BOOST_CHECKPOINT("Verifying signatures");
+ 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));
+ sizeof(PcSignature)-1);
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_TEST_CHECKPOINT("Verifying signature: ");
BOOST_CHECK_EQUAL(val_result, LICENSE_OK);
}
}
--
Gitblit v1.9.1