Digest.h

/*
 *  Copyright (C) 2018 Team Kodi
 *  This file is part of Kodi - https://kodi.tv
 *
 *  SPDX-License-Identifier: GPL-2.0-or-later
 *  See LICENSES/README.md for more information.
 */

#pragma once

#include "StringUtils.h"

#include <iostream>
#include <memory>
#include <stdexcept>
#include <string>

#include <openssl/evp.h>

namespace KODI
{
namespace UTILITY
{

class CDigest
{
public:
  enum class Type
  {
    MD5,
    SHA1,
    SHA256,
    SHA512,
    INVALID
  };

  static std::string TypeToString(Type type);
  static Type TypeFromString(std::string const& type);

  CDigest(Type type);
  void Update(std::string const& data);
  void Update(void const* data, std::size_t size);
  std::string Finalize();
  std::string FinalizeRaw();

  static std::string Calculate(Type type, std::string const& data);
  static std::string Calculate(Type type, void const* data, std::size_t size);

private:
  struct MdCtxDeleter
  {
    void operator()(EVP_MD_CTX* context);
  };

  bool m_finalized{false};
  std::unique_ptr<EVP_MD_CTX, MdCtxDeleter> m_context;
  EVP_MD const* m_md;
};

struct TypedDigest
{
  CDigest::Type type{CDigest::Type::INVALID};
  std::string value;

  TypedDigest() = default;

  TypedDigest(CDigest::Type type, std::string const& value)
  : type(type), value(value)
  {}

  bool Empty() const
  {
    return (type == CDigest::Type::INVALID || value.empty());
  }
};

inline bool operator==(TypedDigest const& left, TypedDigest const& right)
{
  if (left.type != right.type)
  {
    throw std::logic_error("Cannot compare digests of different type");
  }
  return StringUtils::EqualsNoCase(left.value, right.value);
}

inline bool operator!=(TypedDigest const& left, TypedDigest const& right)
{
  return !(left == right);
}

std::ostream& operator<<(std::ostream& os, TypedDigest const& digest);

}
}