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
12namespace remoting {
13
14namespace {
15const char kTestMessage[] = "Test Message";
16
17// |kTestMessage| signed with the key from |kTestRsaKeyPair|.
18const 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|
26const 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
52class RsaKeyPairTest : public testing::Test {
53};
54
55TEST_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
67TEST_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
75TEST_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
84TEST_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