1 files changed, 109 insertions, 0 deletions
diff --git a/modules/es_capi/es_capi.cpp b/modules/es_capi/es_capi.cpp
new file mode 100644
index 000000000..95bc3d969
--- /dev/null
+++ b/modules/es_capi/es_capi.cpp
@@ -0,0 +1,109 @@
+/*************************************************
+* Win32 CryptoAPI EntropySource Source File      *
+* (C) 1999-2006 The Botan Project                *
+*************************************************/
+
+#include <botan/es_capi.h>
+#include <botan/parsing.h>
+#include <botan/conf.h>
+#include <windows.h>
+#include <wincrypt.h>
+
+namespace Botan {
+
+namespace {
+
+/*************************************************
+* CSP Handle                                     *
+*************************************************/
+class CSP_Handle
+   {
+   public:
+      CSP_Handle(u64bit);
+      ~CSP_Handle();
+
+      void gen_random(byte[], u32bit) const;
+      bool is_valid() const { return valid; }
+
+      HCRYPTPROV get_handle() const { return handle; }
+   private:
+      HCRYPTPROV handle;
+      bool valid;
+   };
+
+/*************************************************
+* Call CryptGenRandom                            *
+*************************************************/
+void CSP_Handle::gen_random(byte out[], u32bit n) const
+   {
+   CryptGenRandom(handle, n, out);
+   }
+
+/*************************************************
+* Initialize a CSP Handle                        *
+*************************************************/
+CSP_Handle::CSP_Handle(u64bit capi_provider)
+   {
+   valid = false;
+   DWORD prov_type = (DWORD)capi_provider;
+
+   if(CryptAcquireContext(&handle, 0, 0, prov_type, CRYPT_VERIFYCONTEXT))
+      valid = true;
+   }
+
+/*************************************************
+* Destroy a CSP Handle                           *
+*************************************************/
+CSP_Handle::~CSP_Handle()
+   {
+   if(valid)
+      CryptReleaseContext(handle, 0);
+   }
+
+}
+
+/*************************************************
+* Gather Entropy from Win32 CAPI                 *
+*************************************************/
+u32bit Win32_CAPI_EntropySource::slow_poll(byte output[], u32bit length)
+   {
+   if(length > 64)
+      length = 64;
+
+   for(u32bit j = 0; j != prov_types.size(); j++)
+      {
+      CSP_Handle csp(prov_types[j]);
+      if(!csp.is_valid())
+         continue;
+
+      csp.gen_random(output, length);
+      break;
+      }
+   return length;
+   }
+
+/*************************************************
+* Gather Entropy from Win32 CAPI                 *
+*************************************************/
+Win32_CAPI_EntropySource::Win32_CAPI_EntropySource(const std::string& provs)
+   {
+   std::vector<std::string> capi_provs;
+
+   if(provs == "")
+      capi_provs = Config::get_list("rng/ms_capi_prov_type");
+   else
+      capi_provs = split_on(provs, ':');
+
+   for(u32bit j = 0; j != capi_provs.size(); j++)
+      {
+      if(capi_provs[j] == "RSA_FULL")  prov_types.push_back(PROV_RSA_FULL);
+      if(capi_provs[j] == "INTEL_SEC") prov_types.push_back(PROV_INTEL_SEC);
+      if(capi_provs[j] == "FORTEZZA")  prov_types.push_back(PROV_FORTEZZA);
+      if(capi_provs[j] == "RNG")       prov_types.push_back(PROV_RNG);
+      }
+
+   if(prov_types.size() == 0)
+      prov_types.push_back(PROV_RSA_FULL);
+   }
+
+}