clib_MD5.cpp

Go to the documentation of this file.

#include "clib_MD5.h"

#include <iomanip>
#include <sstream>

#include "logfacility.h"
#include "pol_global_config.h"
#include "stlutil.h"

#ifdef WINDOWS
#include "Header_Windows.h"
#include <wincrypt.h>

namespace Pol
{
namespace Clib
{
static HCRYPTPROV hProv = NULL;

bool MD5_Encrypt( const std::string& in, std::string& out )
{
  // bool bResult = true;

  // HCRYPTKEY hKey = NULL;
  // HCRYPTKEY hXchgKey = NULL;
  HCRYPTHASH hHash = NULL;

  if ( !hProv )
  {
    if ( !CryptAcquireContext( &hProv, NULL, MS_DEF_PROV, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT ) )
    {
      if ( !CryptAcquireContext( &hProv, NULL, NULL, PROV_RSA_FULL, 0 ) )
      {
        ERROR_PRINT << "Error " << GetLastError() << " acquiring crypt context\n";
        return false;
      }
    }
  }

  if ( !CryptCreateHash( hProv, CALG_MD5, 0, 0, &hHash ) )
  {
    ERROR_PRINT << "Error " << GetLastError() << " creating hash\n";
    return false;
  }

  // Hash password string.
  if ( !CryptHashData( hHash, (const unsigned char*)( in.data() ),
                       static_cast<unsigned long>( in.length() ), 0 ) )
  {
    ERROR_PRINT << "Error " << GetLastError() << " adding data to hash\n";
    return false;
  }
  unsigned long len = 16;
  unsigned char buf[16];
  if ( !CryptGetHashParam( hHash, HP_HASHVAL, buf, &len, 0 ) )
  {
    ERROR_PRINT << "Error " << GetLastError() << " getting hash value\n";
    return false;
  }

  std::ostringstream os;
  for ( unsigned int i = 0; i < len; i++ )
  {
    os << std::setfill( '0' ) << std::setw( 2 ) << std::hex << (int)buf[i];
  }
  out = os.str();

  CryptDestroyHash( hHash );
  return true;
}

void MD5_Cleanup()
{
  if ( hProv )
  {
    CryptReleaseContext( hProv, 0 );
  }
}

#else

#include <openssl/md5.h>

namespace Pol
{
namespace Clib
{
bool MD5_Encrypt( const std::string& in, std::string& out )
{
  unsigned char sum[16];

  MD5( reinterpret_cast<const unsigned char*>( in.c_str() ), in.length(), sum );

  std::ostringstream os;
  for ( auto& elem : sum )
  {
    os << std::setfill( '0' ) << std::setw( 2 ) << std::hex << (int)elem;
  }
  out = os.str();

  return true;
}
void MD5_Cleanup()
{
  // OpenSSL cleanup, if any
}

#endif

bool MD5_Compare( const std::string& a, const std::string& b )
{
  bool ret = false;
  if ( stringicmp( a, b ) == 0 )
    ret = true;

  return ret;
}
}
}