hash.cpp

Go to the documentation of this file.

#include "hash.h"
 
#include "esp_arduino_version.h"
 
#include "mbedtls/base64.h"
#include "mbedtls/md.h"
#include "mbedtls/md5.h"
 
namespace sensesp {

void Sha1(const String& payload_str, uint8_t *hash_output) {
  const char *payload = payload_str.c_str();
 
  const int size = 20;
 
  mbedtls_md_context_t ctx;
  mbedtls_md_type_t md_type = MBEDTLS_MD_SHA1;
 
  const size_t payload_length = payload_str.length();
 
  mbedtls_md_init(&ctx);
  mbedtls_md_setup(&ctx, mbedtls_md_info_from_type(md_type), 0);
  mbedtls_md_starts(&ctx);
  mbedtls_md_update(&ctx, reinterpret_cast<const unsigned char *>(payload),
                    payload_length);
  mbedtls_md_finish(&ctx, hash_output);
  mbedtls_md_free(&ctx);
}

String MD5(const String& payload_str) {
  const char *payload = payload_str.c_str();
  char output[33] = {0};
 
  const size_t payload_length = payload_str.length();
 
  mbedtls_md5_context ctx_;
  uint8_t i;
  uint8_t buf_[16] = {0};
  mbedtls_md5_init(&ctx_);
#if ESP_ARDUINO_VERSION_MAJOR > 2
  mbedtls_md5_starts(&ctx_);
  mbedtls_md5_update(&ctx_, (const uint8_t *)payload, payload_length);
  mbedtls_md5_finish(&ctx_, buf_);
#else
  mbedtls_md5_starts_ret(&ctx_);
  mbedtls_md5_update_ret(&ctx_, (const uint8_t *)payload, payload_length);
  mbedtls_md5_finish_ret(&ctx_, buf_);
#endif
  mbedtls_md5_free(&ctx_);
  for (i = 0; i < 16; i++) {
    sprintf(output + (i * 2), "%02x", buf_[i]);
  }
  return String(output);
}

String Base64Sha1(const String& payload_str) {
  uint8_t hash_output[20];
 
  uint8_t encoded[32];
 
  size_t output_length;
 
  sensesp::Sha1(payload_str, hash_output);
 
  int retval = mbedtls_base64_encode(encoded, sizeof(encoded), &output_length,
                                     hash_output, 20);
 
  if (retval != 0) {
    ESP_LOGE(__FILENAME__, "Base64 encoding failed");
    return "";
  }
 
  String encoded_str((char *)encoded);
 
  encoded_str.replace("/", "_");
 
  return encoded_str;
}
 
}  // namespace sensesp