1//===- llvm/unittest/Support/MD5Test.cpp - MD5 tests ----------------------===//
2//
3//                     The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This file implements unit tests for the MD5 functions.
11//
12//===----------------------------------------------------------------------===//
13
14#include "llvm/ADT/ArrayRef.h"
15#include "llvm/ADT/SmallString.h"
16#include "llvm/Support/MD5.h"
17#include "gtest/gtest.h"
18
19using namespace llvm;
20
21namespace {
22/// \brief Tests an arbitrary set of bytes passed as \p Input.
23void TestMD5Sum(ArrayRef<uint8_t> Input, StringRef Final) {
24  MD5 Hash;
25  Hash.update(Input);
26  MD5::MD5Result MD5Res;
27  Hash.final(MD5Res);
28  SmallString<32> Res;
29  MD5::stringifyResult(MD5Res, Res);
30  EXPECT_EQ(Res, Final);
31}
32
33void TestMD5Sum(StringRef Input, StringRef Final) {
34  MD5 Hash;
35  Hash.update(Input);
36  MD5::MD5Result MD5Res;
37  Hash.final(MD5Res);
38  SmallString<32> Res;
39  MD5::stringifyResult(MD5Res, Res);
40  EXPECT_EQ(Res, Final);
41}
42
43TEST(MD5Test, MD5) {
44  TestMD5Sum(ArrayRef<uint8_t>((const uint8_t *)"", (size_t) 0),
45             "d41d8cd98f00b204e9800998ecf8427e");
46  TestMD5Sum(ArrayRef<uint8_t>((const uint8_t *)"a", (size_t) 1),
47             "0cc175b9c0f1b6a831c399e269772661");
48  TestMD5Sum(ArrayRef<uint8_t>((const uint8_t *)"abcdefghijklmnopqrstuvwxyz",
49                               (size_t) 26),
50             "c3fcd3d76192e4007dfb496cca67e13b");
51  TestMD5Sum(ArrayRef<uint8_t>((const uint8_t *)"\0", (size_t) 1),
52             "93b885adfe0da089cdf634904fd59f71");
53  TestMD5Sum(ArrayRef<uint8_t>((const uint8_t *)"a\0", (size_t) 2),
54             "4144e195f46de78a3623da7364d04f11");
55  TestMD5Sum(ArrayRef<uint8_t>((const uint8_t *)"abcdefghijklmnopqrstuvwxyz\0",
56                               (size_t) 27),
57             "81948d1f1554f58cd1a56ebb01f808cb");
58  TestMD5Sum("abcdefghijklmnopqrstuvwxyz", "c3fcd3d76192e4007dfb496cca67e13b");
59}
60}
61