unittests/Support/MD5Test.cpp

   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
  19 using namespace llvm;
  20
  21 namespace {
  22 /// \brief Tests an arbitrary set of bytes passed as \p Input.
  23 void 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
  33 void 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
  43 TEST(MD5Test, MD5) {
  44   TestMD5Sum(makeArrayRef((const uint8_t *)"", (size_t) 0),
  45              "d41d8cd98f00b204e9800998ecf8427e");
  46   TestMD5Sum(makeArrayRef((const uint8_t *)"a", (size_t) 1),
  47              "0cc175b9c0f1b6a831c399e269772661");
  48   TestMD5Sum(makeArrayRef((const uint8_t *)"abcdefghijklmnopqrstuvwxyz",
  49                           (size_t) 26),
  50              "c3fcd3d76192e4007dfb496cca67e13b");
  51   TestMD5Sum(makeArrayRef((const uint8_t *)"\0", (size_t) 1),
  52              "93b885adfe0da089cdf634904fd59f71");
  53   TestMD5Sum(makeArrayRef((const uint8_t *)"a\0", (size_t) 2),
  54              "4144e195f46de78a3623da7364d04f11");
  55   TestMD5Sum(makeArrayRef((const uint8_t *)"abcdefghijklmnopqrstuvwxyz\0",
  56                           (size_t) 27),
  57              "81948d1f1554f58cd1a56ebb01f808cb");
  58   TestMD5Sum("abcdefghijklmnopqrstuvwxyz", "c3fcd3d76192e4007dfb496cca67e13b");
  59 }
  60 }