hash_base.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <cstddef>
#include <cstring>
#include "esphome/core/helpers.h"
 
namespace esphome {
 
class HashBase {
 public:
  virtual ~HashBase() = default;
 
  virtual void init() = 0;
 
  virtual void add(const uint8_t *data, size_t len) = 0;
  void add(const char *data, size_t len) { this->add((const uint8_t *) data, len); }
 
  virtual void calculate() = 0;
 
  void get_bytes(uint8_t *output) { memcpy(output, this->digest_, this->get_size()); }
 
  void get_hex(char *output) {
    for (size_t i = 0; i < this->get_size(); i++) {
      uint8_t byte = this->digest_[i];
      output[i * 2] = format_hex_char(byte >> 4);
      output[i * 2 + 1] = format_hex_char(byte & 0x0F);
    }
  }
 
  bool equals_bytes(const uint8_t *expected) { return memcmp(this->digest_, expected, this->get_size()) == 0; }
 
  bool equals_hex(const char *expected) {
    uint8_t parsed[this->get_size()];
    if (!parse_hex(expected, parsed, this->get_size())) {
      return false;
    }
    return this->equals_bytes(parsed);
  }
 
  virtual size_t get_size() const = 0;
 
 protected:
  uint8_t digest_[32];  // Storage sized for max(MD5=16, SHA256=32) bytes
};
 
}  // namespace esphome