githubmirror/omim

coding/sha1.cpp
78 строк2 KB
Maxim Pimenov
[base] Replaced the namespace my with base.
18 сен 2018, 14:46
18 сен 2018, 14:46412387c
#include "coding/sha1.hpp"

#include "coding/internal/file_data.hpp"
#include "coding/file_reader.hpp"

#include "base/assert.hpp"
#include "base/logging.hpp"
#include "base/macros.hpp"

#include "3party/liboauthcpp/src/SHA1.h"
#include "3party/liboauthcpp/src/base64.h"

#include <algorithm>
#include <vector>

namespace coding
{
// static
SHA1::Hash SHA1::Calculate(std::string const & filePath)
{
  uint32_t constexpr kFileBufferSize = 8192;
  try
  {
    base::FileData file(filePath, base::FileData::OP_READ);
    uint64_t const fileSize = file.Size();

    CSHA1 sha1;
    uint64_t currSize = 0;
    unsigned char buffer[kFileBufferSize];
    while (currSize < fileSize)
    {
      auto const toRead = std::min(kFileBufferSize, static_cast<uint32_t>(fileSize - currSize));
      file.Read(currSize, buffer, toRead);
      sha1.Update(buffer, toRead);
      currSize += toRead;
    }
    sha1.Final();

    Hash result;
    ASSERT_EQUAL(result.size(), ARRAY_SIZE(sha1.m_digest), ());
    std::copy(std::begin(sha1.m_digest), std::end(sha1.m_digest), std::begin(result));
    return result;
  }
  catch (Reader::Exception const & ex)
  {
    LOG(LERROR, ("Error reading file:", filePath, ex.what()));
  }
  return {};
}

// static
std::string SHA1::CalculateBase64(std::string const & filePath)
{
  auto const sha1 = Calculate(filePath);
  return base64_encode(sha1.data(), sha1.size());
}

// static
SHA1::Hash SHA1::CalculateForString(std::string const & str)
{
  CSHA1 sha1;
  std::vector<unsigned char> dat(str.begin(), str.end());
  sha1.Update(dat.data(), static_cast<uint32_t>(dat.size()));
  sha1.Final();

  Hash result;
  ASSERT_EQUAL(result.size(), ARRAY_SIZE(sha1.m_digest), ());
  std::copy(std::begin(sha1.m_digest), std::end(sha1.m_digest), std::begin(result));
  return result;
}

// static
std::string SHA1::CalculateBase64ForString(std::string const & str)
{
  auto const sha1 = CalculateForString(str);
  return base64_encode(sha1.data(), sha1.size());
}
}  // coding