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. 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
