1 /*
2  *  Copyright 2012 The WebRTC Project Authors. All rights reserved.
3  *
4  *  Use of this source code is governed by a BSD-style license
5  *  that can be found in the LICENSE file in the root of the source
6  *  tree. An additional intellectual property rights grant can be found
7  *  in the file PATENTS.  All contributing project authors may
8  *  be found in the AUTHORS file in the root of the source tree.
9  */
10 
11 #include "rtc_base/message_digest.h"
12 
13 #include "rtc_base/string_encode.h"
14 #include "test/gtest.h"
15 
16 namespace rtc {
17 
18 // Test vectors from RFC 1321.
TEST(MessageDigestTest,TestMd5Digest)19 TEST(MessageDigestTest, TestMd5Digest) {
20   // Test the string versions of the APIs.
21   EXPECT_EQ("d41d8cd98f00b204e9800998ecf8427e", ComputeDigest(DIGEST_MD5, ""));
22   EXPECT_EQ("900150983cd24fb0d6963f7d28e17f72",
23             ComputeDigest(DIGEST_MD5, "abc"));
24   EXPECT_EQ("c3fcd3d76192e4007dfb496cca67e13b",
25             ComputeDigest(DIGEST_MD5, "abcdefghijklmnopqrstuvwxyz"));
26 
27   // Test the raw buffer versions of the APIs; also check output buffer size.
28   char output[16];
29   EXPECT_EQ(sizeof(output),
30             ComputeDigest(DIGEST_MD5, "abc", 3, output, sizeof(output)));
31   EXPECT_EQ("900150983cd24fb0d6963f7d28e17f72",
32             hex_encode(output, sizeof(output)));
33   EXPECT_EQ(0U,
34             ComputeDigest(DIGEST_MD5, "abc", 3, output, sizeof(output) - 1));
35 }
36 
37 // Test vectors from RFC 3174.
TEST(MessageDigestTest,TestSha1Digest)38 TEST(MessageDigestTest, TestSha1Digest) {
39   // Test the string versions of the APIs.
40   EXPECT_EQ("da39a3ee5e6b4b0d3255bfef95601890afd80709",
41             ComputeDigest(DIGEST_SHA_1, ""));
42   EXPECT_EQ("a9993e364706816aba3e25717850c26c9cd0d89d",
43             ComputeDigest(DIGEST_SHA_1, "abc"));
44   EXPECT_EQ("84983e441c3bd26ebaae4aa1f95129e5e54670f1",
45             ComputeDigest(
46                 DIGEST_SHA_1,
47                 "abcdbcdecdefdefgefghfghighijhijkijkljklmklmnlmnomnopnopq"));
48 
49   // Test the raw buffer versions of the APIs; also check output buffer size.
50   char output[20];
51   EXPECT_EQ(sizeof(output),
52             ComputeDigest(DIGEST_SHA_1, "abc", 3, output, sizeof(output)));
53   EXPECT_EQ("a9993e364706816aba3e25717850c26c9cd0d89d",
54             hex_encode(output, sizeof(output)));
55   EXPECT_EQ(0U,
56             ComputeDigest(DIGEST_SHA_1, "abc", 3, output, sizeof(output) - 1));
57 }
58 
59 // Test that we fail properly if a bad digest algorithm is specified.
TEST(MessageDigestTest,TestBadDigest)60 TEST(MessageDigestTest, TestBadDigest) {
61   std::string output;
62   EXPECT_FALSE(ComputeDigest("sha-9000", "abc", &output));
63   EXPECT_EQ("", ComputeDigest("sha-9000", "abc"));
64 }
65 
66 // Test vectors from RFC 2202.
TEST(MessageDigestTest,TestMd5Hmac)67 TEST(MessageDigestTest, TestMd5Hmac) {
68   // Test the string versions of the APIs.
69   EXPECT_EQ("9294727a3638bb1c13f48ef8158bfc9d",
70             ComputeHmac(DIGEST_MD5, std::string(16, '\x0b'), "Hi There"));
71   EXPECT_EQ("750c783e6ab0b503eaa86e310a5db738",
72             ComputeHmac(DIGEST_MD5, "Jefe", "what do ya want for nothing?"));
73   EXPECT_EQ("56be34521d144c88dbb8c733f0e8b3f6",
74             ComputeHmac(DIGEST_MD5, std::string(16, '\xaa'),
75                         std::string(50, '\xdd')));
76   EXPECT_EQ(
77       "697eaf0aca3a3aea3a75164746ffaa79",
78       ComputeHmac(DIGEST_MD5,
79                   "\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
80                   "\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19",
81                   std::string(50, '\xcd')));
82   EXPECT_EQ(
83       "56461ef2342edc00f9bab995690efd4c",
84       ComputeHmac(DIGEST_MD5, std::string(16, '\x0c'), "Test With Truncation"));
85   EXPECT_EQ(
86       "6b1ab7fe4bd7bf8f0b62e6ce61b9d0cd",
87       ComputeHmac(DIGEST_MD5, std::string(80, '\xaa'),
88                   "Test Using Larger Than Block-Size Key - Hash Key First"));
89   EXPECT_EQ("6f630fad67cda0ee1fb1f562db3aa53e",
90             ComputeHmac(DIGEST_MD5, std::string(80, '\xaa'),
91                         "Test Using Larger Than Block-Size Key and Larger "
92                         "Than One Block-Size Data"));
93 
94   // Test the raw buffer versions of the APIs; also check output buffer size.
95   std::string key(16, '\x0b');
96   std::string input("Hi There");
97   char output[16];
98   EXPECT_EQ(sizeof(output),
99             ComputeHmac(DIGEST_MD5, key.c_str(), key.size(), input.c_str(),
100                         input.size(), output, sizeof(output)));
101   EXPECT_EQ("9294727a3638bb1c13f48ef8158bfc9d",
102             hex_encode(output, sizeof(output)));
103   EXPECT_EQ(0U, ComputeHmac(DIGEST_MD5, key.c_str(), key.size(), input.c_str(),
104                             input.size(), output, sizeof(output) - 1));
105 }
106 
107 // Test vectors from RFC 2202.
TEST(MessageDigestTest,TestSha1Hmac)108 TEST(MessageDigestTest, TestSha1Hmac) {
109   // Test the string versions of the APIs.
110   EXPECT_EQ("b617318655057264e28bc0b6fb378c8ef146be00",
111             ComputeHmac(DIGEST_SHA_1, std::string(20, '\x0b'), "Hi There"));
112   EXPECT_EQ("effcdf6ae5eb2fa2d27416d5f184df9c259a7c79",
113             ComputeHmac(DIGEST_SHA_1, "Jefe", "what do ya want for nothing?"));
114   EXPECT_EQ("125d7342b9ac11cd91a39af48aa17b4f63f175d3",
115             ComputeHmac(DIGEST_SHA_1, std::string(20, '\xaa'),
116                         std::string(50, '\xdd')));
117   EXPECT_EQ(
118       "4c9007f4026250c6bc8414f9bf50c86c2d7235da",
119       ComputeHmac(DIGEST_SHA_1,
120                   "\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
121                   "\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19",
122                   std::string(50, '\xcd')));
123   EXPECT_EQ("4c1a03424b55e07fe7f27be1d58bb9324a9a5a04",
124             ComputeHmac(DIGEST_SHA_1, std::string(20, '\x0c'),
125                         "Test With Truncation"));
126   EXPECT_EQ(
127       "aa4ae5e15272d00e95705637ce8a3b55ed402112",
128       ComputeHmac(DIGEST_SHA_1, std::string(80, '\xaa'),
129                   "Test Using Larger Than Block-Size Key - Hash Key First"));
130   EXPECT_EQ("e8e99d0f45237d786d6bbaa7965c7808bbff1a91",
131             ComputeHmac(DIGEST_SHA_1, std::string(80, '\xaa'),
132                         "Test Using Larger Than Block-Size Key and Larger "
133                         "Than One Block-Size Data"));
134 
135   // Test the raw buffer versions of the APIs; also check output buffer size.
136   std::string key(20, '\x0b');
137   std::string input("Hi There");
138   char output[20];
139   EXPECT_EQ(sizeof(output),
140             ComputeHmac(DIGEST_SHA_1, key.c_str(), key.size(), input.c_str(),
141                         input.size(), output, sizeof(output)));
142   EXPECT_EQ("b617318655057264e28bc0b6fb378c8ef146be00",
143             hex_encode(output, sizeof(output)));
144   EXPECT_EQ(0U,
145             ComputeHmac(DIGEST_SHA_1, key.c_str(), key.size(), input.c_str(),
146                         input.size(), output, sizeof(output) - 1));
147 }
148 
TEST(MessageDigestTest,TestBadHmac)149 TEST(MessageDigestTest, TestBadHmac) {
150   std::string output;
151   EXPECT_FALSE(ComputeHmac("sha-9000", "key", "abc", &output));
152   EXPECT_EQ("", ComputeHmac("sha-9000", "key", "abc"));
153 }
154 
155 }  // namespace rtc
156