| OLD | NEW |
| (Empty) |
| 1 // Copyright 2008-2009 Google Inc. | |
| 2 // | |
| 3 // Licensed under the Apache License, Version 2.0 (the "License"); | |
| 4 // you may not use this file except in compliance with the License. | |
| 5 // You may obtain a copy of the License at | |
| 6 // | |
| 7 // http://www.apache.org/licenses/LICENSE-2.0 | |
| 8 // | |
| 9 // Unless required by applicable law or agreed to in writing, software | |
| 10 // distributed under the License is distributed on an "AS IS" BASIS, | |
| 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| 12 // See the License for the specific language governing permissions and | |
| 13 // limitations under the License. | |
| 14 // ======================================================================== | |
| 15 | |
| 16 #include <windows.h> | |
| 17 #include <cstring> | |
| 18 #include <vector> | |
| 19 #include "omaha/base/security/sha.h" | |
| 20 #include "omaha/base/string.h" | |
| 21 #include "omaha/base/utils.h" | |
| 22 #include "omaha/net/cup_utils.h" | |
| 23 #include "omaha/testing/unit_test.h" | |
| 24 | |
| 25 namespace omaha { | |
| 26 | |
| 27 namespace cup_utils { | |
| 28 | |
| 29 namespace { | |
| 30 | |
| 31 const char kPlainText[] = "The quick brown fox jumps over the lazy dog"; | |
| 32 const int kRsaKeySize = 128; // 128 bytes. | |
| 33 | |
| 34 // Test padding for a random message of num_bytes length. | |
| 35 void TestPadding(size_t num_bytes) { | |
| 36 std::vector<uint8> data(num_bytes); | |
| 37 EXPECT_TRUE(GenRandom(&data.front(), data.size())); | |
| 38 std::vector<uint8> m = RsaPad(kRsaKeySize, &data.front(), data.size()); | |
| 39 EXPECT_EQ(m.size(), kRsaKeySize); | |
| 40 | |
| 41 uint8 hash[SHA_DIGEST_SIZE] = {0}; | |
| 42 SHA(&m.front(), m.size() - SHA_DIGEST_SIZE, hash); | |
| 43 EXPECT_EQ(memcmp(hash, &m[kRsaKeySize - SHA_DIGEST_SIZE], sizeof(hash)), 0); | |
| 44 | |
| 45 EXPECT_FALSE(m[0] & 0x80); // msb is always reset. | |
| 46 EXPECT_TRUE(m[0] & 0x60); // bit next to msb is always set. | |
| 47 } | |
| 48 | |
| 49 } // namespace | |
| 50 | |
| 51 | |
| 52 TEST(CupUtilsTest, RsaPad) { | |
| 53 EXPECT_GE(kRsaKeySize, SHA_DIGEST_SIZE); | |
| 54 | |
| 55 TestPadding(1); // 1 byte. | |
| 56 TestPadding(SHA_DIGEST_SIZE); // 20 bytes. | |
| 57 TestPadding(kRsaKeySize); // 128 bytes. | |
| 58 TestPadding(1000); // 1000 bytes. | |
| 59 } | |
| 60 | |
| 61 // The underlying B64_encode function in common/security does not pad | |
| 62 // with '='. | |
| 63 TEST(CupUtilsTest, B64Encode) { | |
| 64 const char* bytes = "1"; | |
| 65 EXPECT_STREQ(B64Encode(bytes, strlen(bytes)), "MQ"); | |
| 66 bytes = "12"; | |
| 67 EXPECT_STREQ(B64Encode(bytes, strlen(bytes)), "MTI"); | |
| 68 bytes = "123"; | |
| 69 EXPECT_STREQ(B64Encode(bytes, strlen(bytes)), "MTIz"); | |
| 70 bytes = "Google Inc"; | |
| 71 EXPECT_STREQ(B64Encode(bytes, strlen(bytes)), "R29vZ2xlIEluYw"); | |
| 72 | |
| 73 const uint8* first = reinterpret_cast<const uint8*>(bytes); | |
| 74 const uint8* last = first + strlen(bytes); | |
| 75 EXPECT_STREQ(B64Encode(std::vector<uint8>(first, last)), "R29vZ2xlIEluYw"); | |
| 76 } | |
| 77 | |
| 78 TEST(CupUtilsTest, Hash) { | |
| 79 // Empty vector. | |
| 80 std::vector<uint8> data; | |
| 81 std::vector<uint8> hash = Hash(data); | |
| 82 EXPECT_STREQ(BytesToHex(hash), | |
| 83 _T("da39a3ee5e6b4b0d3255bfef95601890afd80709")); | |
| 84 | |
| 85 // Non-empty vector. | |
| 86 const uint8* first = reinterpret_cast<const uint8*>(kPlainText); | |
| 87 const uint8* last = first + strlen(kPlainText); | |
| 88 data.clear(); | |
| 89 data.insert(data.begin(), first, last); | |
| 90 hash = Hash(data); | |
| 91 EXPECT_STREQ(BytesToHex(hash), | |
| 92 _T("2fd4e1c67a2d28fced849ee1bb76e7391b93eb12")); | |
| 93 | |
| 94 // Empty CString. | |
| 95 hash = Hash(CStringA()); | |
| 96 EXPECT_STREQ(BytesToHex(hash), | |
| 97 _T("da39a3ee5e6b4b0d3255bfef95601890afd80709")); | |
| 98 | |
| 99 // Non-empty CString. | |
| 100 hash = Hash(CStringA(kPlainText)); | |
| 101 EXPECT_STREQ(BytesToHex(hash), | |
| 102 _T("2fd4e1c67a2d28fced849ee1bb76e7391b93eb12")); | |
| 103 | |
| 104 // Hash of strings | |
| 105 CStringA arg(kPlainText); | |
| 106 hash = HashBuffers(arg.GetString(), arg.GetLength(), NULL, 0, NULL, 0); | |
| 107 EXPECT_STREQ(BytesToHex(hash), | |
| 108 _T("64eb91d899d68d2af394fbfe8f7c3b2884055ddb")); | |
| 109 | |
| 110 hash = HashBuffers(NULL, 0, arg.GetString(), arg.GetLength(), NULL, 0); | |
| 111 EXPECT_STREQ(BytesToHex(hash), | |
| 112 _T("6c9f398d556ffe64fe441ca5491e25fd67d2ab65")); | |
| 113 | |
| 114 hash = HashBuffers(NULL, 0, NULL, 0, arg.GetString(), arg.GetLength()); | |
| 115 EXPECT_STREQ(BytesToHex(hash), | |
| 116 _T("9a7e591318dc169fc023414d8d25321bcb11121e")); | |
| 117 | |
| 118 hash = HashBuffers(NULL, 0, NULL, 0, NULL, 0); | |
| 119 EXPECT_STREQ(BytesToHex(hash), | |
| 120 _T("d0dc1cf9bf61884f8e7982e0b1b87954bd9ee9c7")); | |
| 121 } | |
| 122 | |
| 123 TEST(CupUtilsTest, SymSign) { | |
| 124 // Sign of vectors: ignore NULL vectors. | |
| 125 std::vector<uint8> key; // This is the signing key. | |
| 126 key.push_back(7); | |
| 127 | |
| 128 CStringA arg(kPlainText); | |
| 129 std::vector<uint8> hash = Hash(arg); | |
| 130 std::vector<uint8> sym_sign = SymSign(key, 1, &hash, NULL, NULL); | |
| 131 EXPECT_STREQ(BytesToHex(sym_sign), | |
| 132 _T("1c6fa359c0a7b11421ae4d31a92f9de8a4ef8d2d")); | |
| 133 | |
| 134 sym_sign = SymSign(key, 1, NULL, &hash, NULL); | |
| 135 EXPECT_STREQ(BytesToHex(sym_sign), | |
| 136 _T("1c6fa359c0a7b11421ae4d31a92f9de8a4ef8d2d")); | |
| 137 | |
| 138 sym_sign = SymSign(key, 1, NULL, NULL, &hash); | |
| 139 EXPECT_STREQ(BytesToHex(sym_sign), | |
| 140 _T("1c6fa359c0a7b11421ae4d31a92f9de8a4ef8d2d")); | |
| 141 | |
| 142 sym_sign = SymSign(key, 1, &hash, &hash, &hash); | |
| 143 EXPECT_STREQ(BytesToHex(sym_sign), | |
| 144 _T("fef3e343795946cd47ff4e07eca5f3f09b051d86")); | |
| 145 } | |
| 146 | |
| 147 TEST(CupUtilsTest, ParseCupCookie) { | |
| 148 EXPECT_STREQ(_T(""), ParseCupCookie(_T(""))); | |
| 149 EXPECT_STREQ(_T(""), ParseCupCookie(_T("foo; bar;"))); | |
| 150 EXPECT_STREQ(_T("c="), ParseCupCookie(_T("c="))); | |
| 151 EXPECT_STREQ(_T("c="), ParseCupCookie(_T("c= ;"))); | |
| 152 EXPECT_STREQ(_T("c=foo"), ParseCupCookie(_T("c=foo"))); | |
| 153 EXPECT_STREQ(_T("c=foo"), ParseCupCookie(_T(";c=foo"))); | |
| 154 EXPECT_STREQ(_T("c=foo"), ParseCupCookie(_T("c=foo ; bar;"))); | |
| 155 EXPECT_STREQ(_T("c=foo"), ParseCupCookie(_T("b=bar; c=foo;"))); | |
| 156 EXPECT_STREQ(_T("c=foo"), ParseCupCookie(_T("b=bar; c=foo ; foobar"))); | |
| 157 } | |
| 158 | |
| 159 } // namespace cup_utils | |
| 160 | |
| 161 } // namespace omaha | |
| OLD | NEW |