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