Home | History | Annotate | Download | only in base 
      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 
     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 
     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 
     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 
     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