1 //===- llvm/unittest/Support/MD5Test.cpp - MD5 tests ----------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // This file implements unit tests for the MD5 functions.
10 //
11 //===----------------------------------------------------------------------===//
12
13 #include "llvm/Support/MD5.h"
14 #include "llvm/ADT/ArrayRef.h"
15 #include "llvm/ADT/SmallString.h"
16 #include "gtest/gtest.h"
17
18 using namespace llvm;
19
20 namespace {
21 /// Tests an arbitrary set of bytes passed as \p Input.
TestMD5Sum(ArrayRef<uint8_t> Input,StringRef Final)22 void TestMD5Sum(ArrayRef<uint8_t> Input, StringRef Final) {
23 MD5 Hash;
24 Hash.update(Input);
25 MD5::MD5Result MD5Res;
26 Hash.final(MD5Res);
27 SmallString<32> Res;
28 MD5::stringifyResult(MD5Res, Res);
29 EXPECT_EQ(Res, Final);
30 }
31
TestMD5Sum(StringRef Input,StringRef Final)32 void TestMD5Sum(StringRef Input, StringRef Final) {
33 MD5 Hash;
34 Hash.update(Input);
35 MD5::MD5Result MD5Res;
36 Hash.final(MD5Res);
37 SmallString<32> Res;
38 MD5::stringifyResult(MD5Res, Res);
39 EXPECT_EQ(Res, Final);
40 }
41
TEST(MD5Test,MD5)42 TEST(MD5Test, MD5) {
43 TestMD5Sum(ArrayRef((const uint8_t *)"", (size_t)0),
44 "d41d8cd98f00b204e9800998ecf8427e");
45 TestMD5Sum(ArrayRef((const uint8_t *)"a", (size_t)1),
46 "0cc175b9c0f1b6a831c399e269772661");
47 TestMD5Sum(
48 ArrayRef((const uint8_t *)"abcdefghijklmnopqrstuvwxyz", (size_t)26),
49 "c3fcd3d76192e4007dfb496cca67e13b");
50 TestMD5Sum(ArrayRef((const uint8_t *)"\0", (size_t)1),
51 "93b885adfe0da089cdf634904fd59f71");
52 TestMD5Sum(ArrayRef((const uint8_t *)"a\0", (size_t)2),
53 "4144e195f46de78a3623da7364d04f11");
54 TestMD5Sum(
55 ArrayRef((const uint8_t *)"abcdefghijklmnopqrstuvwxyz\0", (size_t)27),
56 "81948d1f1554f58cd1a56ebb01f808cb");
57 TestMD5Sum("abcdefghijklmnopqrstuvwxyz", "c3fcd3d76192e4007dfb496cca67e13b");
58 }
59
TEST(MD5HashTest,MD5)60 TEST(MD5HashTest, MD5) {
61 ArrayRef<uint8_t> Input((const uint8_t *)"abcdefghijklmnopqrstuvwxyz", 26);
62 std::array<uint8_t, 16> Vec = MD5::hash(Input);
63 MD5::MD5Result MD5Res;
64 SmallString<32> Res;
65 memcpy(MD5Res.data(), Vec.data(), Vec.size());
66 MD5::stringifyResult(MD5Res, Res);
67 EXPECT_EQ(Res, "c3fcd3d76192e4007dfb496cca67e13b");
68 EXPECT_EQ(0x3be167ca6c49fb7dULL, MD5Res.high());
69 EXPECT_EQ(0x00e49261d7d3fcc3ULL, MD5Res.low());
70 }
71
TEST(MD5Test,FinalAndResultHelpers)72 TEST(MD5Test, FinalAndResultHelpers) {
73 MD5 Hash;
74
75 Hash.update("abcd");
76
77 {
78 MD5 ReferenceHash;
79 ReferenceHash.update("abcd");
80 MD5::MD5Result ReferenceResult;
81 ReferenceHash.final(ReferenceResult);
82 EXPECT_EQ(Hash.result(), ReferenceResult);
83 }
84
85 Hash.update("xyz");
86
87 {
88 MD5 ReferenceHash;
89 ReferenceHash.update("abcd");
90 ReferenceHash.update("xyz");
91 MD5::MD5Result ReferenceResult;
92 ReferenceHash.final(ReferenceResult);
93 EXPECT_EQ(Hash.final(), ReferenceResult);
94 }
95 }
96 } // namespace
97