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 | 138 +++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 137 insertions(+), 1 deletions(-) diff --git a/src/library/os/win/os-win.c b/src/library/os/win/os-win.c index e7bb77a..169821b 100644 --- a/src/library/os/win/os-win.c +++ b/src/library/os/win/os-win.c @@ -1,10 +1,14 @@ #include <Windows.h> #include <iphlpapi.h> -//definition of size_t +//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; @@ -161,4 +165,136 @@ *adapter_info_size = i; return result; } +FUNCTION_RETURN getModuleName(char buffer[MAX_PATH]) { + FUNCTION_RETURN result=FUNC_RET_OK; + DWORD wres = GetModuleFileName(NULL, buffer, MAX_PATH); + if (wres == 0){ + result = FUNC_RET_ERROR; + } + 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