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