1 /*
2 * This code is derived from (original license follows):
3 *
4 * This is an OpenSSL-compatible implementation of the RSA Data Security, Inc.
5 * MD5 Message-Digest Algorithm (RFC 1321).
6 *
7 * Homepage:
8 * http://openwall.info/wiki/people/solar/software/public-domain-source-code/md5
9 *
10 * Author:
11 * Alexander Peslyak, better known as Solar Designer <solar at openwall.com>
12 *
13 * This software was written by Alexander Peslyak in 2001. No copyright is
14 * claimed, and the software is hereby placed in the public domain.
15 * In case this attempt to disclaim copyright and place the software in the
16 * public domain is deemed null and void, then the software is
17 * Copyright (c) 2001 Alexander Peslyak and it is hereby released to the
18 * general public under the following terms:
19 *
20 * Redistribution and use in source and binary forms, with or without
21 * modification, are permitted.
22 *
23 * There's ABSOLUTELY NO WARRANTY, express or implied.
24 *
25 * See md5.c for more information.
26 */
27
28 #ifndef LLVM_SUPPORT_MD5_H
29 #define LLVM_SUPPORT_MD5_H
30
31 #include "llvm/ADT/SmallString.h"
32 #include "llvm/Support/DataTypes.h"
33 #include "llvm/Support/Endian.h"
34 #include <array>
35
36 namespace llvm {
37 template <typename T> class ArrayRef;
38
39 class MD5 {
40 // Any 32-bit or wider unsigned integer data type will do.
41 typedef uint32_t MD5_u32plus;
42
43 MD5_u32plus a, b, c, d;
44 MD5_u32plus hi, lo;
45 uint8_t buffer[64];
46 MD5_u32plus block[16];
47
48 public:
49 typedef uint8_t MD5Result[16];
50
51 MD5();
52
53 /// \brief Updates the hash for the byte stream provided.
54 void update(ArrayRef<uint8_t> Data);
55
56 /// \brief Updates the hash for the StringRef provided.
57 void update(StringRef Str);
58
59 /// \brief Finishes off the hash and puts the result in result.
60 void final(MD5Result &Result);
61
62 /// \brief Translates the bytes in \p Res to a hex string that is
63 /// deposited into \p Str. The result will be of length 32.
64 static void stringifyResult(MD5Result &Result, SmallString<32> &Str);
65
66 /// \brief Computes the hash for a given bytes.
67 static std::array<uint8_t, 16> hash(ArrayRef<uint8_t> Data);
68
69 private:
70 const uint8_t *body(ArrayRef<uint8_t> Data);
71 };
72
73 /// Helper to compute and return lower 64 bits of the given string's MD5 hash.
MD5Hash(StringRef Str)74 inline uint64_t MD5Hash(StringRef Str) {
75 MD5 Hash;
76 Hash.update(Str);
77 llvm::MD5::MD5Result Result;
78 Hash.final(Result);
79 // Return the least significant 8 bytes. Our MD5 implementation returns the
80 // result in little endian, so we may need to swap bytes.
81 using namespace llvm::support;
82 return endian::read<uint64_t, little, unaligned>(Result);
83 }
84
85 }
86
87 #endif
88