mirror of
https://gitee.com/milvus-io/milvus.git
synced 2026-01-07 19:31:51 +08:00
131 lines
3.8 KiB
C++
131 lines
3.8 KiB
C++
#include "LicenseCheck.h"
|
|
#include <iostream>
|
|
#include <thread>
|
|
|
|
#include <boost/archive/binary_oarchive.hpp>
|
|
#include <boost/archive/binary_iarchive.hpp>
|
|
//#include <boost/foreach.hpp>
|
|
//#include <boost/serialization/vector.hpp>
|
|
#include <boost/filesystem/path.hpp>
|
|
#include <boost/serialization/map.hpp>
|
|
#include <boost/filesystem/operations.hpp>
|
|
#include <boost/thread.hpp>
|
|
#include <boost/date_time/posix_time/posix_time.hpp>
|
|
|
|
|
|
namespace zilliz {
|
|
namespace milvus {
|
|
namespace server {
|
|
|
|
LicenseCheck::LicenseCheck() {
|
|
|
|
}
|
|
|
|
LicenseCheck::~LicenseCheck() {
|
|
StopCountingDown();
|
|
}
|
|
|
|
ServerError
|
|
LicenseCheck::LegalityCheck(const std::string &license_file_path) {
|
|
|
|
int device_count;
|
|
LicenseLibrary::GetDeviceCount(device_count);
|
|
std::vector<std::string> uuid_array;
|
|
LicenseLibrary::GetUUID(device_count, uuid_array);
|
|
|
|
std::vector<std::string> sha_array;
|
|
LicenseLibrary::GetUUIDSHA256(device_count, uuid_array, sha_array);
|
|
|
|
int output_device_count;
|
|
std::map<int, std::string> uuid_encryption_map;
|
|
time_t starting_time;
|
|
time_t end_time;
|
|
ServerError err = LicenseLibrary::LicenseFileDeserialization(license_file_path,
|
|
output_device_count,
|
|
uuid_encryption_map,
|
|
starting_time,
|
|
end_time);
|
|
if(err !=SERVER_SUCCESS)
|
|
{
|
|
std::cout << "License check error: 01" << std::endl;
|
|
return SERVER_UNEXPECTED_ERROR;
|
|
}
|
|
time_t system_time;
|
|
LicenseLibrary::GetSystemTime(system_time);
|
|
|
|
if (device_count != output_device_count) {
|
|
std::cout << "License check error: 02" << std::endl;
|
|
return SERVER_UNEXPECTED_ERROR;
|
|
}
|
|
for (int i = 0; i < device_count; ++i) {
|
|
if (sha_array[i] != uuid_encryption_map[i]) {
|
|
std::cout << "License check error: 03" << std::endl;
|
|
return SERVER_UNEXPECTED_ERROR;
|
|
}
|
|
}
|
|
if (system_time < starting_time || system_time > end_time) {
|
|
std::cout << "License check error: 04" << std::endl;
|
|
return SERVER_UNEXPECTED_ERROR;
|
|
}
|
|
std::cout << "Legality Check Success" << std::endl;
|
|
return SERVER_SUCCESS;
|
|
}
|
|
|
|
// Part 2: Timing check license
|
|
|
|
ServerError
|
|
LicenseCheck::AlterFile(const std::string &license_file_path,
|
|
const boost::system::error_code &ec,
|
|
boost::asio::deadline_timer *pt) {
|
|
|
|
ServerError err = LicenseCheck::LegalityCheck(license_file_path);
|
|
if(err!=SERVER_SUCCESS) {
|
|
std::cout << "license file check error" << std::endl;
|
|
exit(1);
|
|
}
|
|
|
|
std::cout << "---runing---" << std::endl;
|
|
pt->expires_at(pt->expires_at() + boost::posix_time::hours(1));
|
|
pt->async_wait(boost::bind(LicenseCheck::AlterFile, license_file_path, boost::asio::placeholders::error, pt));
|
|
|
|
return SERVER_SUCCESS;
|
|
|
|
}
|
|
|
|
ServerError
|
|
LicenseCheck::StartCountingDown(const std::string &license_file_path) {
|
|
|
|
if (!LicenseLibrary::IsFileExistent(license_file_path)) {
|
|
std::cout << "license file not exist" << std::endl;
|
|
exit(1);
|
|
}
|
|
|
|
//create a thread to run AlterFile
|
|
if(counting_thread_ == nullptr) {
|
|
counting_thread_ = std::make_shared<std::thread>([&]() {
|
|
boost::asio::deadline_timer t(io_service_, boost::posix_time::hours(1));
|
|
t.async_wait(boost::bind(LicenseCheck::AlterFile, license_file_path, boost::asio::placeholders::error, &t));
|
|
io_service_.run();//this thread will block here
|
|
});
|
|
}
|
|
|
|
return SERVER_SUCCESS;
|
|
}
|
|
|
|
ServerError
|
|
LicenseCheck::StopCountingDown() {
|
|
if(!io_service_.stopped()) {
|
|
io_service_.stop();
|
|
}
|
|
|
|
if(counting_thread_ != nullptr) {
|
|
counting_thread_->join();
|
|
counting_thread_ = nullptr;
|
|
}
|
|
|
|
return SERVER_SUCCESS;
|
|
}
|
|
|
|
}
|
|
}
|
|
} |