Gabriele Contini
2019-09-07 7b1950f29fa4a73a43b71ee3c1e444fa38e296f0
test/functional/volid_test.cpp
@@ -1,105 +1,126 @@
#define BOOST_TEST_MODULE standard_license_test
//#define BOOST_TEST_MAIN
#define BOOST_TEST_DYN_LINK
//#undef BOOST_TEST_DYN_LINK
#include <boost/test/unit_test.hpp>
#include <fstream>
#include <stdio.h>
#include <cstring>
#include "../../../src/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"
#include "generate-license.h"
#include "../../src/library/pc-identifiers.h"
#include <dbus/dbus.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);
BOOST_AUTO_TEST_CASE( default_volid_lic_file ) {
   const string licLocation(PROJECT_TEST_TEMP_DIR "/volid_license.lic");
   PcSignature identifier_out;
 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);
 }
   IDENTIFICATION_STRATEGY strategy = IDENTIFICATION_STRATEGY::ETHERNET;
   BOOST_TEST_CHECKPOINT("Before generate");
   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");
   generate_license(licLocation, extraArgs);
 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();
 }
 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.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);
}
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 == FUNC_RET_OK);
      if (generate_ok != FUNC_RET_OK) {
         BOOST_ERROR(
               "Generating identifier for strategy " << strategies[i] << " failed with: " << generate_ok);
         idfile << "0000-0000-0000-0000" << endl;
      } 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;
   if (getVirtualization() != CONTAINER) {
      strategies = { DEFAULT, DISK_LABEL, DISK_NUM, ETHERNET };
   } else {
      strategies = { DEFAULT, ETHERNET };
   }
   int num_strategies = strategies.size();
   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);
      }
   }
}