libcruft-crypto/tools/hash.cpp

138 lines
3.3 KiB
C++
Raw Normal View History

2018-01-14 17:17:34 +11:00
/*
2018-08-04 15:18:16 +10:00
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
2018-01-14 17:17:34 +11:00
*
* Copyright:
* 2014-2016, Danny Robson <danny@nerdcruft.net>
*/
#include <cruft/crypto/hash/md2.hpp>
#include <cruft/crypto/hash/md4.hpp>
#include <cruft/crypto/hash/md5.hpp>
#include <cruft/crypto/hash/ripemd.hpp>
#include <cruft/crypto/hash/sha1.hpp>
#include <cruft/crypto/hash/sha2.hpp>
2018-01-14 17:17:34 +11:00
#include <cruft/util/hash/adler.hpp>
#include <cruft/util/hash/bsdsum.cpp>
#include <cruft/util/hash/crc.hpp>
#include <cruft/util/io.hpp>
#include <cruft/util/stream.hpp>
#include <iostream>
#include <cstdlib>
#include <cstring>
///////////////////////////////////////////////////////////////////////////////
static
const char* NAMES[] = {
"adler32",
"bsdsum",
"crc",
"MD2",
"MD4",
"MD5",
"RIPEMD",
"SHA1",
"SHA256",
};
///////////////////////////////////////////////////////////////////////////////
std::ostream&
print_digest (std::ostream &os, uint32_t t)
{
return os << std::hex << t << std::dec;
}
//-----------------------------------------------------------------------------
template <size_t S>
std::ostream&
print_digest (std::ostream &os, std::array<uint8_t,S> digest)
{
2018-08-05 14:51:17 +10:00
cruft::stream::scoped::flags f (os);
2018-01-14 17:17:34 +11:00
os << std::hex;
for (auto c: digest)
os << +(c >> 4) << +(c & 0x0F);
return os;
}
///////////////////////////////////////////////////////////////////////////////
static void
2018-08-05 14:51:17 +10:00
compute (const std::string &name, const cruft::view<const uint8_t*> data)
2018-01-14 17:17:34 +11:00
{
#define stream(TYPE, ...) do { \
if (name != #TYPE) \
break; \
\
print_digest ( \
std::cout, \
2018-08-05 14:51:17 +10:00
cruft::hash::TYPE{} (data) \
2018-01-14 17:17:34 +11:00
) << '\n'; \
return; \
} while (0);
stream (adler32);
//stream (bsdsum);
//stream (crc32);
//stream (MD2);
//stream (MD4);
//stream (MD5);
//stream (RIPEMD);
//stream (SHA1);
//stream (SHA256);
#undef stream
}
///////////////////////////////////////////////////////////////////////////////
enum {
ARG_CMD,
ARG_HASH,
ARG_INPUT,
NUM_ARGS
};
//-----------------------------------------------------------------------------
void
print_usage (int argc, char **argv)
{
(void)argc;
std::cerr << argv[ARG_CMD] << " [";
for (auto name: NAMES)
std::cerr << name << "|";
std::cerr << "] <input>\n";
}
//-----------------------------------------------------------------------------
int
main (int argc, char **argv)
{
if (argc != NUM_ARGS) {
print_usage (argc, argv);
return EXIT_FAILURE;
}
if (strcmp (argv[ARG_INPUT], "-")) {
2018-08-05 14:51:17 +10:00
const cruft::mapped_file src (argv[ARG_INPUT]);
compute (argv[ARG_HASH], cruft::view{src});
2018-01-14 17:17:34 +11:00
return EXIT_SUCCESS;
} else {
//compute (argv[ARG_HASH], std::cin);
return EXIT_SUCCESS;
}
}