From 62258ba3b4737432a95c3af8d0e03ed3fb7953e7 Mon Sep 17 00:00:00 2001
From: open-license-manager <rillf@maildrop.cc>
Date: 周一, 13 10月 2014 05:05:37 +0800
Subject: [PATCH] windows ok

---
 src/library/os/win/os-win.c |  128 ++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 128 insertions(+), 0 deletions(-)

diff --git a/src/library/os/win/os-win.c b/src/library/os/win/os-win.c
index a241734..169821b 100644
--- a/src/library/os/win/os-win.c
+++ b/src/library/os/win/os-win.c
@@ -2,9 +2,13 @@
 #include <iphlpapi.h>
 //definition of size_t
 #include <stdlib.h>
+//#include "../../base/base64.h"
 #include "../../base/logger.h"
 #include"../os.h"
+#include "public-key.h"
+
 #pragma comment(lib, "IPHLPAPI.lib")
+unsigned char* unbase64(const char* ascii, int len, int *flen);
 
 FUNCTION_RETURN getOsSpecificIdentifier(unsigned char identifier[6]){
 	return FUNC_RET_NOT_AVAIL;
@@ -170,3 +174,127 @@
 	return result;
 }
 
+static void printHash(HCRYPTHASH* hHash){
+	BYTE         *pbHash;
+	DWORD        dwHashLen;
+	DWORD        dwHashLenSize = sizeof(DWORD); 
+	char* hashStr;
+	int i;
+
+	if (CryptGetHashParam(*hHash,HP_HASHSIZE,(BYTE *)&dwHashLen, &dwHashLenSize, 0))
+	{
+		pbHash = (BYTE*)malloc(dwHashLen);
+		hashStr = (char*)malloc(dwHashLen*2 +1);
+		if (CryptGetHashParam(*hHash,HP_HASHVAL,pbHash,	&dwHashLen,	0))	{
+			for (i = 0; i < dwHashLen; i++)	{
+				sprintf(&hashStr[i * 2], "%02x", pbHash[i]);
+			}
+			LOG_DEBUG("Hash to verify: %s", hashStr);
+		}
+		free(pbHash);
+		free(hashStr);
+	}
+}
+
+FUNCTION_RETURN verifySignature(const char* stringToVerify,	const char* signatureB64) {
+	//--------------------------------------------------------------------
+	// Declare variables.
+	//
+	// hProv:           Cryptographic service provider (CSP). This example
+	//                  uses the Microsoft Enhanced Cryptographic 
+	//                  Provider.
+	// hKey:            Key to be used. In this example, you import the 
+	//                  key as a PLAINTEXTKEYBLOB.
+	// dwBlobLen:       Length of the plaintext key.
+	// pbKeyBlob:       Pointer to the exported key.
+	BYTE pubKey[] = PUBLIC_KEY;
+
+	HCRYPTPROV hProv = 0;
+	HCRYPTKEY hKey = 0;
+	HCRYPTHASH hHash = 0;
+	DWORD dwSigLen;
+	BYTE* sigBlob;
+
+	//--------------------------------------------------------------------
+	// Acquire a handle to the CSP.
+
+	if (!CryptAcquireContext(
+		&hProv,
+		NULL,
+		MS_ENHANCED_PROV,
+		PROV_RSA_FULL,
+		CRYPT_VERIFYCONTEXT))
+	{
+		// If the key container cannot be opened, try creating a new
+		// container by specifying a container name and setting the 
+		// CRYPT_NEWKEYSET flag.
+		LOG_INFO("Error in AcquireContext 0x%08x \n", GetLastError());
+		if (NTE_BAD_KEYSET == GetLastError())
+		{
+			if (!CryptAcquireContext(
+				&hProv,
+				"license++verify",
+				MS_ENHANCED_PROV,
+				PROV_RSA_FULL,
+				CRYPT_NEWKEYSET | CRYPT_VERIFYCONTEXT))
+			{
+				LOG_ERROR("Error in AcquireContext 0x%08x \n",
+					GetLastError());
+				return FUNC_RET_ERROR;
+			}
+		}
+		else
+		{
+			LOG_ERROR(" Error in AcquireContext 0x%08x \n", GetLastError());
+			return FUNC_RET_ERROR;
+		}
+	}
+
+	// Use the CryptImportKey function to import the PLAINTEXTKEYBLOB
+	// BYTE array into the key container. The function returns a 
+	// pointer to an HCRYPTKEY variable that contains the handle of
+	// the imported key.
+	if (!CryptImportKey(hProv, &pubKey[0], sizeof(pubKey), 0, 0, &hKey))
+	{
+		LOG_ERROR("Error 0x%08x in importing the PublicKey \n",
+			GetLastError());
+		return FUNC_RET_ERROR;
+	}
+
+	if (CryptCreateHash(hProv, CALG_SHA1, 0, 0, &hHash))
+	{
+		LOG_DEBUG("Hash object created.");
+	}
+	else
+	{
+		LOG_ERROR("Error in hash creation 0x%08x ", GetLastError());
+		CryptReleaseContext(hProv,0);
+		return FUNC_RET_ERROR;
+	}
+
+	if (!CryptHashData(hHash, stringToVerify, strlen(stringToVerify), 0)){
+		LOG_ERROR("Error in hashing data 0x%08x ", GetLastError());
+		CryptDestroyHash(hHash);
+		CryptReleaseContext(hProv, 0);
+		return FUNC_RET_ERROR;
+	}
+#ifdef _DEBUG
+	LOG_DEBUG("Lenght %d, hashed Data: [%s]", strlen(stringToVerify), stringToVerify);
+	printHash(&hHash);
+#endif
+	sigBlob = unbase64(signatureB64, strlen(signatureB64), &dwSigLen);
+	LOG_DEBUG("raw signature lenght %d", dwSigLen);
+	if (!CryptVerifySignature(hHash, sigBlob, dwSigLen, hKey, NULL, 0))
+	{
+		LOG_ERROR("Signature not validated!  0x%08x ", GetLastError());
+		free(sigBlob);
+		CryptDestroyHash(hHash);
+		CryptReleaseContext(hProv, 0);
+		return FUNC_RET_ERROR;
+	}
+	CryptDestroyHash(hHash);
+	free(sigBlob);
+	CryptReleaseContext(hProv, 0);
+	return FUNC_RET_OK;
+}
+

--
Gitblit v1.9.1