From 1922d8c2cf7dcaacafd17394849577794b1f10eb Mon Sep 17 00:00:00 2001 From: open-license-manager <rillf@maildrop.cc> Date: 周六, 13 9月 2014 06:24:55 +0800 Subject: [PATCH] signature windows --- src/license-generator/license-generator.cpp | 56 ++++++++++++++++++++++++++++++++++++++++---------------- 1 files changed, 40 insertions(+), 16 deletions(-) diff --git a/src/license-generator/license-generator.cpp b/src/license-generator/license-generator.cpp index f6e3e80..846d5d9 100644 --- a/src/license-generator/license-generator.cpp +++ b/src/license-generator/license-generator.cpp @@ -50,12 +50,12 @@ ("begin_date,b", po::value<string>(), "Specify the start of the validity for this license. " " Format YYYYMMDD. If not specified defaults to today") // - ("end_date,e", po::value<string>(), + ("expire_date,e", po::value<string>(), "Specify the expire date for this license. " " Format YYYYMMDD. If not specified the license won't expire") // ("client_signature,s", po::value<string>(), "The signature of the pc that requires the license. " - "It should be in the format XXX-XXXX-XXXX." + "It should be in the format XXXX-XXXX-XXXX-XXXX." " If not specified the license " "won't be linked to a specific pc.") // ("start_version,t", po::value<unsigned int>()->default_value(0 @@ -71,14 +71,17 @@ vector<FullLicenseInfo> LicenseGenerator::parseLicenseInfo( po::variables_map vm) { - time_t begin_date = FullLicenseInfo::UNUSED_TIME; - time_t end_date = FullLicenseInfo::UNUSED_TIME; - if (vm.count("end_date")) { - const std::string dt_end = vm["end_date"].as<string>(); + string begin_date = FullLicenseInfo::UNUSED_TIME; + string end_date = FullLicenseInfo::UNUSED_TIME; + if (vm.count("expire_date")) { + const std::string dt_end = vm["expire_date"].as<string>(); try { - end_date = seconds_from_epoch(dt_end.c_str()); - begin_date = time(NULL); - } catch (invalid_argument &e) { + end_date = normalize_date(dt_end.c_str()); + char curdate[20]; + time_t curtime = time(NULL); + strftime(curdate, 20, "%Y-%m-%d", localtime(&curtime)); + begin_date.assign(curdate); + } catch (const invalid_argument &e) { cerr << endl << "End date not recognized: " << dt_end << " Please enter a valid date in format YYYYMMDD" << endl; exit(2); @@ -87,7 +90,7 @@ if (vm.count("begin_date")) { const std::string begin_date_str = vm["begin_date"].as<string>(); try { - begin_date = seconds_from_epoch(begin_date_str.c_str()); + begin_date = normalize_date(begin_date_str.c_str()); } catch (invalid_argument &e) { cerr << endl << "Begin date not recognized: " << begin_date_str << " Please enter a valid date in format YYYYMMDD" << endl; @@ -98,14 +101,15 @@ string client_signature = ""; if (vm.count("client_signature")) { client_signature = vm["client_signature"].as<string>(); - regex e("\\d{3}-[:alnum:]{4}-[:alnum:]{4}"); + //fixme match + and / + /*regex e("(A-Za-z0-9){4}-(A-Za-z0-9){4}-(A-Za-z0-9){4}-(A-Za-z0-9){4}"); if (!regex_match(client_signature, e)) { cerr << endl << "Client signature not recognized: " << client_signature - << " Please enter a valid signature in format XXX-XXXX-XXX" + << " Please enter a valid signature in format XXXX-XXXX-XXXX-XXXX" << endl; exit(2); - } + }*/ } string extra_data = ""; if (vm.count("extra_data")) { @@ -133,7 +137,7 @@ return licInfo; } -void LicenseGenerator::generateAndOutptuLicenses(const po::variables_map& vm, +void LicenseGenerator::generateAndOutputLicenses(const po::variables_map& vm, ostream& outputFile) { vector<FullLicenseInfo> licenseInfo = parseLicenseInfo(vm); license::LicenseSigner licSigner = @@ -176,14 +180,34 @@ << " error: " << strerror( errno); exit(3); } - generateAndOutptuLicenses(vm, ofstream); + generateAndOutputLicenses(vm, ofstream); ofstream.close(); } else { - generateAndOutptuLicenses(vm, cout); + generateAndOutputLicenses(vm, cout); } return 0; } +const std::string formats[] = { "%4u-%2u-%2u","%4u/%2u/%2u","%4u%2u%2u" }; +const size_t formats_n = 3; +string LicenseGenerator::normalize_date(const char * s) { + unsigned int year, month, day; + int chread; + bool found = false; + for (size_t i = 0; i < formats_n &&!found; ++i) { + chread = sscanf(s, formats[i].c_str(),&year, &month, &day); + if (chread == 3){ + found = true; + } + } + if (!found){ + throw invalid_argument("Date not recognized."); + } + ostringstream oss; + oss << year << "-"<<setfill('0') << std::setw(2)<< month<<"-"<<day; + //delete (facet); + return oss.str(); +} } -- Gitblit v1.9.1