open-license-manager

2014-08-06 425c196d0762424675120d92f9e4105d13af0f1a

 src/bootstrap/win/Main.cpp

@@ -1,13 +1,36 @@
#include <stdio.h>
#include "CryptoHelper.h"
#include <string>

void main() {
using namespace std;

void write_pubkey_file(string public_fname, BYTE *pbPublicKey, DWORD dwPublicKeySize){
   FILE* fp = fopen(public_fname.c_str(), "w");
   fprintf(fp, "//file generated by bootstrap.cpp, do not edit.\n\n");
   fprintf(fp, "#ifndef PUBLIC_KEY_H_\n#define PUBLIC_KEY_H_\n");
   fprintf(fp, "static BYTE PUBLIC_KEY[] = {");
   for (int i = 0; i < dwPublicKeySize; i++){
      if (i != 0){
         fprintf(fp, ",");
      }
      if (i % 15 == 0){
         fprintf(fp, "\n        ");
      }
      fprintf(fp, "%d", pbPublicKey[i]);
   }
   fprintf(fp, "\n};\n\n");
   fprintf(fp, "#define SHARED_RANDOM %d;\n", 12345);
   fprintf(fp, "#endif\n");
   fclose(fp);
}

void generatePk(string private_include, string public_include) {
   CryptoHelper cryptoHlpr;
   BYTE *pbPublicKey = NULL, *pbPrivateKey = NULL;
   DWORD dwPublicKeySize = 0, dwPrivateKeySize = 0;
   HRESULT hr = S_OK;
   // Get the key container context.
   if (FAILED(hr = cryptoHlpr.AcquireContext((L"TestContainer")))) {
   if (FAILED(hr = cryptoHlpr.AcquireContext(_T("TestContainer")))) {
// Call FormatMessage to display the error returned in hr.
      return;
   }
@@ -22,6 +45,9 @@
// Call FormatMessage to display the error returned in hr.
      return;
   }
   else {
      write_pubkey_file(public_include, pbPublicKey, dwPublicKeySize);
   }
   // Print out the public key to console as a
   // hexadecimal string.
   wprintf(L"\n\nPublicKey = \"");
@@ -34,7 +60,7 @@
   // Print out the private key to console as a
   // hexadecimal string.
   wprintf(L"\n\nPrivateKey = \"");
   for (i = 0; i < dwPrivateKeySize; i++) {
   for (DWORD i = 0; i < dwPrivateKeySize; i++) {
      wprintf(L"%02x", pbPrivateKey[i]);
   }
   wprintf(L"\"\n");
@@ -48,3 +74,23 @@
      delete[] pbPrivateKey;
   return;
}


int main(int argc, char** argv) {

   if (argc != 3) {
      //print_usage();
      exit(2);
   }
   else {
      printf("********************************************\n");
      printf("*  Bootstrap!!!                            *\n");
      printf("********************************************\n");

   }
   string private_fname = string(argv[1]);
   string public_fname(argv[2]);

   generatePk(private_fname, public_fname);
   return 0;
}