1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include <string>
6
7 #include "remoting/base/rsa_key_pair.h"
8 #include "remoting/base/test_rsa_key_pair.h"
9 #include "testing/gmock/include/gmock/gmock.h"
10 #include "testing/gtest/include/gtest/gtest.h"
11
12 namespace remoting {
13
14 namespace {
15 const char kTestMessage[] = "Test Message";
16
17 // |kTestMessage| signed with the key from |kTestRsaKeyPair|.
18 const char kExpectedSignature[] =
19 "LfUyXU2AiKM4rpWivUR3bLiQiRt1W3iIenNfJEB8RWyoEfnvSBoD52x8q9yFvtLFDEMPWyIrwM+N2"
20 "LuaWBKG1c0R7h+twBgvpExzZneJl+lbGMRx9ba8m/KAFrUWA/NRzOen2NHCuPybOEasgrPgGWBrmf"
21 "gDcvyW8QiGuKLopGj/4c5CQT4yE8JjsyU3Qqo2ZPK4neJYQhOmAlg+Q5dAPLpzWMj5HQyOVHJaSXZ"
22 "Y8vl/LiKvbdofYLeYNVKAE4q5mfpQMrsysPYpbxBV60AhFyrvtC040MFGcflKQRZNiZwMXVb7DclC"
23 "BPgvK7rI5Y0ERtVm+yNmH7vCivfyAnDUYA==";
24
25 // Another RSA key pair, different from |kTestRsaKeyPair|
26 const char kTestRsaKeyPair2[] =
27 "MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQDLNd9bNNBFxXSasqWHv8ydItmoi"
28 "NhiWV/1Z6JI6TpKTnTm9yQSxIoCLHT8fB8QA0wzzgYEuZpDHIZDv1WxSdC3kgx84JrWe43+SPDZ2b"
29 "7ekuhXinPK+f3Nw5GjGaAUevyeWNoDD3GFka5q4zw8W1OE8E/z50FfPqOUejg+qyrgovEWQIZWvI0"
30 "CHdy8HTtxT7G0YbPJYZ8ycRkUrsXtY2RQX4IaCMEcdAmCm2Q2hYldulbZX6Bvv5GX8FOWYOerbYVK"
31 "ZsmH5II+KEP4We75ONVR2jdCCJ5L3YMXtbYtZZy1yZXcOf2fpBo6+p81M7rG9com/C75QoMVMVJJC"
32 "ahCli7/AgMBAAECggEAZNzThUC8k7UDQHmlgbCojeIraOSrin1UDMmomknxHcq9aZqHtC0LVzLbyi"
33 "qNfHQ2kYwUHqpFMERrPBsvHHVH/KWoPx94mzbubqWjrm3OuEjwu+rDuJ7G5CfLFMp2U1QMKUhuxZA"
34 "Xx7Vcfj9VuZuW4+gntyc0omLD7MGRQ0HQYXh7ZDGWrMPEs6Cjzcx9/G9AD7ysWIqk14iwJqKhztiD"
35 "NirMr64eDZFzzDvXTl3j5l+yiAHiV5LPUUKyCe+jEdZMceSKy5wSZXSkiW4zhgEzwdMN2zmxlcC59"
36 "17dw2c6xD+tKxTMwzx77sauBFNzebNU1m5hIKH+jCPiA8aQv3/l8QKBgQD8J4ilT/CV6hhw0/q99e"
37 "+54s+SIz8nYo8fvyeEBCiA4lf/OZuxl/sYWK+PthP+xzsjKTq7yoFkliXrtOioW34E0WdLv+6jEXQ"
38 "hxaXZyk0TMwxm228xMG66evXDJ8OGWCi6uiAnVWNUu6VXacQwKHf9hv6DNRNcmkQGojfcx3ZNZwKB"
39 "gQDOT0ApweRr0zA28l8C6qtNfY/NIeWTWPDTspt8zDoX1ILXc2HZHYW0QtUQ6tXNQnUW/ymavla7E"
40 "upa8AoZoqMvaUIg1BjMYIzc4yQMVf1BRCfvT8GsoymX+8Gt/DB34L7KMPOhQysMmZMNjKtozQsbZY"
41 "wcnN19pWnDv78trFZ6qQKBgHy1nMqN7+JlRjM/VCrxYOAhwiF31ztGbpz38LZFTDb6OyVau5spHKH"
42 "c8u9z0Q3YQXJRaOAJ9tblv9mEvvDNV1VQr/Lx+TejYTl2xGEjwdz2CXMxohvE5W7Lc5NSrkxae8Jm"
43 "XZK2k4sLx2mlQMfErBuy0VvZOzs4fN5/CnviFquPAoGBAI8NAI5ztPDW1L2kvSCGmxT2FTnFYSwUJ"
44 "ZiEZa/Y5AcWAUtm49fp0oW1OYuraWgTxqCVeMGlbPn2Ga3IdxhjXwdG0uV0a2V7JPEcRiiPjzUsDw"
45 "yunroXwIVzuU3saacVnPURkDynGDh6XC6u9UOLuUHb3ZURZ7rxcS6by/HdZ3FRAoGBAOhjtjyfFEn"
46 "bZtjcQd+bNtoTPV/L71+K8AYPwV0td5Qy5VbBrTIlv7pNFxE4bYNuEe6cI2cxTua4i5IoKYXyUm5u"
47 "SvUVkkz7CpoiFwMnnLsNrZmazVS2zq0Y2a2ai8C3mPgLOdroS2fBBAcuFApeq1PvISmT6ZnJJ8Yah"
48 "HQCfClh";
49
50 } // namespace
51
52 class RsaKeyPairTest : public testing::Test {
53 };
54
TEST_F(RsaKeyPairTest,ImportExportImport)55 TEST_F(RsaKeyPairTest, ImportExportImport) {
56 // Load a key from a string, export to a string, load again, and verify that
57 // we generate the same signature with both keys.
58 scoped_refptr<RsaKeyPair> exported_key = RsaKeyPair::FromString(
59 kTestRsaKeyPair);
60 scoped_refptr<RsaKeyPair> imported_key = RsaKeyPair::FromString(
61 exported_key->ToString());
62
63 ASSERT_EQ(exported_key->SignMessage(kTestMessage),
64 imported_key->SignMessage(kTestMessage));
65 }
66
TEST_F(RsaKeyPairTest,Signatures)67 TEST_F(RsaKeyPairTest, Signatures) {
68 // Sign a message and check that we get the expected signature.
69 scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::FromString(kTestRsaKeyPair);
70
71 std::string signature_base64 = key_pair->SignMessage(kTestMessage);
72 ASSERT_EQ(signature_base64, std::string(kExpectedSignature));
73 }
74
TEST_F(RsaKeyPairTest,GenerateKey)75 TEST_F(RsaKeyPairTest, GenerateKey) {
76 // Test that we can generate a valid key.
77 scoped_refptr<RsaKeyPair> key_pair = RsaKeyPair::Generate();
78 ASSERT_TRUE(key_pair.get());
79 ASSERT_NE(key_pair->ToString(), "");
80 ASSERT_NE(key_pair->GetPublicKey(), "");
81 ASSERT_NE(key_pair->SignMessage(kTestMessage), "");
82 }
83
TEST_F(RsaKeyPairTest,SignaturesDiffer)84 TEST_F(RsaKeyPairTest, SignaturesDiffer) {
85 // Sign using different keys/messages and check that signatures are different.
86 scoped_refptr<RsaKeyPair> key_pair1 = RsaKeyPair::FromString(kTestRsaKeyPair);
87 scoped_refptr<RsaKeyPair> key_pair2 = RsaKeyPair::FromString(
88 kTestRsaKeyPair2);
89
90 std::string signature_kp1_msg1_base64 = key_pair1->SignMessage(kTestMessage);
91 std::string signature_kp2_msg1_base64 = key_pair2->SignMessage(kTestMessage);
92 std::string signature_kp1_msg2_base64 = key_pair1->SignMessage("Different");
93 ASSERT_NE(signature_kp1_msg1_base64, signature_kp2_msg1_base64);
94 ASSERT_NE(signature_kp1_msg1_base64, signature_kp1_msg2_base64);
95 ASSERT_NE(key_pair1->GetPublicKey(), key_pair2->GetPublicKey());
96 }
97
98 } // namespace remoting
99