developer.intra2net.com Git - bpdyndnsd/blob - src/util.cpp

   1 /** @file
   2  * @brief Util namespace header.
   3  *
   4  *
   5  *
   6  * @copyright Intra2net AG
   7  * @license GPLv2
   8 */
   9
  10 #include "util.hpp"
  11
  12 #include <sstream>
  13 #include <openssl/evp.h>
  14 #include <boost/algorithm/string.hpp>
  15
  16 namespace Util
  17 {
  18 /**
  19  * Computes a MD5 Digest from the given string and returns the HEX representation
  20  * @param data The string to compute the md5 for
  21  * @return The computed md5 in hex
  22  */
  23 std::string compute_md5_digest(std::string data)
  24 {
  25     // compute an MD5 digest.
  26
  27     EVP_MD_CTX mdctx;
  28     const EVP_MD *md;
  29     unsigned char md_value[EVP_MAX_MD_SIZE];
  30     unsigned int md_len = 0;
  31
  32     // Add all digest algorithms to the internal table.
  33     OpenSSL_add_all_digests();
  34
  35     // Get the md5 digest algorithm from the internal table.
  36     md = EVP_get_digestbyname("md5");
  37
  38     // Test if md is initialized.
  39     if ( (md == NULL) || (EVP_MD_size(md) == 0) )
  40         throw std::invalid_argument("NULL pointer: md");
  41
  42     // Initalize digest content mdctx.
  43     EVP_MD_CTX_init(&mdctx);
  44
  45     // Now we can call init_ex.
  46     if ( EVP_DigestInit_ex(&mdctx, md, NULL) == 0 )
  47     {
  48         EVP_MD_CTX_cleanup(&mdctx); /*lint !e534 */
  49         EVP_cleanup();  /*lint !e534 */
  50         throw std::invalid_argument("Could not set up digest context correctly");
  51     }
  52
  53     // Test if data is empty.
  54     if ( data.empty() )
  55     {
  56         EVP_MD_CTX_cleanup(&mdctx); /*lint !e534 */
  57         EVP_cleanup();  /*lint !e534 */
  58         throw std::invalid_argument("Passed data is empty");
  59     }
  60
  61     // Hash the data. At this point data is not empty and &mdctx is initialized.
  62     if ( EVP_DigestUpdate(&mdctx, data.c_str(), data.size()) == 0 )
  63     {
  64         EVP_MD_CTX_cleanup(&mdctx); /*lint !e534 */
  65         EVP_cleanup();  /*lint !e534 */
  66         throw std::invalid_argument("Could not hash data into digest context");
  67     }
  68
  69     // Retrieve the digest value from &mdctx and place it in md_value.
  70     if ( EVP_DigestFinal_ex(&mdctx, md_value, &md_len) == 0 )
  71     {
  72         EVP_MD_CTX_cleanup(&mdctx); /*lint !e534 */
  73         EVP_cleanup();  /*lint !e534 */
  74         throw std::invalid_argument("Could not retrieve digest value");
  75     }
  76
  77     // Test if md_value is filled correctly and md_len is not zero.
  78     if ( (md_len == 0) || (EVP_MD_CTX_size(&mdctx) == 0) )
  79     {
  80         EVP_MD_CTX_cleanup(&mdctx); /*lint !e534 */
  81         EVP_cleanup();  /*lint !e534 */
  82         throw std::invalid_argument("Retrieved invalid digest value");
  83     }
  84
  85     // Internal cleanup of the digest content.
  86     EVP_MD_CTX_cleanup(&mdctx); /*lint !e534 */
  87     EVP_cleanup();  /*lint !e534 */
  88
  89     // Convert md5 digest C string to hex.
  90     std::ostringstream oss_digest_md5_hex;
  91     for(unsigned int i = 0; i < md_len; ++i)
  92     {
  93         // We have to do a static cast to an decimal representation, cause otherwise ostringstream would interpret
  94         // the stream as a character and output the character representation of the hex value.
  95         oss_digest_md5_hex << std::hex << static_cast<unsigned short>(md_value[i]);
  96     }
  97
  98     return oss_digest_md5_hex.str();
  99 }
 100
 101
 102 /**
 103  * Get the status code from the given data.
 104  * @param data The data containing the status code at front, limited by delimiter.
 105  * @param delimiter The delimiter to use.
 106  * @return The parsed status code
 107  */
 108 std::string parse_status_code(std::string data, std::string delimiter)
 109 {
 110     std::list<std::string> tokens;
 111     boost::algorithm::split(tokens,data,boost::is_any_of(delimiter));
 112     if ( tokens.empty() )
 113         return "";
 114     return tokens.front();
 115 }
 116
 117 }