src/cloud_print/gcp20/prototype/x_privet_token_unittest.cc

   1 // Copyright 2013 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 "cloud_print/gcp20/prototype/x_privet_token.h"
   6
   7 #include <stdio.h>
   8
   9 #include "base/base64.h"
  10 #include "base/basictypes.h"
  11 #include "base/format_macros.h"
  12 #include "base/logging.h"
  13 #include "base/sha1.h"
  14 #include "base/strings/string_number_conversions.h"
  15 #include "base/strings/stringprintf.h"
  16 #include "base/time/time.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 TEST(XPrivetTokenTest, Generation) {
  20   std::string secret = "E3E92296-E290-4E77-B678-6AEF256C30C8";
  21   uint64 gen_time = 1372444784;
  22   uint64 issue_time = gen_time;
  23
  24   XPrivetToken xtoken(secret, base::Time::FromTimeT(gen_time));
  25
  26   std::string issue_time_str = base::StringPrintf("%" PRIu64, issue_time);
  27   std::string sha1_val = base::SHA1HashString(secret + ":" + issue_time_str);
  28
  29   ASSERT_STRCASEEQ("2216828f9eefc3931c1b9a110dcca3dbec23571d",
  30                    base::HexEncode(sha1_val.data(), sha1_val.size()).c_str());
  31
  32   std::string base64_val;
  33   base::Base64Encode(sha1_val, &base64_val);
  34   std::string token = base64_val + ":" + issue_time_str;
  35
  36   ASSERT_EQ(token, xtoken.GenerateXTokenWithTime(issue_time));
  37
  38   EXPECT_NE(xtoken.GenerateXTokenWithTime(issue_time),
  39             xtoken.GenerateXTokenWithTime(issue_time + 1));
  40 }
  41
  42 TEST(XPrivetTokenTest, CheckingValid) {
  43   base::Time gen_time = base::Time::FromTimeT(1372444784);
  44   XPrivetToken xtoken("9CEEA1AD-BC24-4D5A-8AB4-A6CE3E0CC4CD", gen_time);
  45
  46   std::string token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT());
  47   EXPECT_TRUE(xtoken.CheckValidXToken(token));
  48
  49   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 1);
  50   EXPECT_TRUE(xtoken.CheckValidXToken(token));
  51
  52   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 55);
  53   EXPECT_TRUE(xtoken.CheckValidXToken(token));
  54
  55   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 60*60 - 5);
  56   EXPECT_TRUE(xtoken.CheckValidXToken(token));
  57
  58   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 60*60 + 10);
  59   EXPECT_TRUE(xtoken.CheckValidXToken(token));
  60
  61   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 24*60*60 - 1);
  62   EXPECT_TRUE(xtoken.CheckValidXToken(token));
  63 }
  64
  65 TEST(XPrivetTokenTest, CheckingInvalid) {
  66   base::Time gen_time = base::Time::FromTimeT(1372444784);
  67   XPrivetToken xtoken("9CEEA1AD-BC24-4D5A-8AB4-A6CE3E0CC4CD", gen_time);
  68
  69   // Meaningless tokens:
  70
  71   std::string token = "CEEA1AD9CEEA1AD9CEEA1AD9CEEA1AD";
  72   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  73
  74   base::Base64Encode("CEEA1AD9CEEA1AD9CEEA1AD9CEEA1AD", &token);
  75   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  76
  77   base::Base64Encode("CEEA1AD9CEEA:1AD9CEEA1AD9CEEA1AD", &token);
  78   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  79
  80   base::Base64Encode("CEEA1AD9CEEA:1AD9CEEA1AD9:CEEA1AD", &token);
  81   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  82
  83   base::Base64Encode("CEEA1AD9CEEA:123456", &token);
  84   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  85
  86   base::Base64Encode("CEEA1AD9CEEA:", &token);
  87   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  88
  89   base::Base64Encode("CEEA1AD9CEEA:1372444784", &token);
  90   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  91
  92   EXPECT_FALSE(xtoken.CheckValidXToken(""));
  93
  94   // Expired tokens:
  95
  96   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 24*60*60 + 1);
  97   EXPECT_FALSE(xtoken.CheckValidXToken(token));
  98
  99   token = xtoken.GenerateXTokenWithTime(gen_time.ToTimeT() + 7*24*60*60 - 1023);
 100   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 101
 102   // Tokens with different secret:
 103
 104   XPrivetToken another("6F02AC4E-6F37-4078-AF42-5EE5D8180284", gen_time);
 105
 106   token = another.GenerateXTokenWithTime(gen_time.ToTimeT() - 24*60*60 - 1);
 107   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 108
 109   token = another.GenerateXTokenWithTime(gen_time.ToTimeT() - 24*60*60 + 1);
 110   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 111
 112   token = another.GenerateXTokenWithTime(gen_time.ToTimeT());
 113   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 114
 115   token = another.GenerateXTokenWithTime(gen_time.ToTimeT() + 1);
 116   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 117
 118   token = another.GenerateXTokenWithTime(gen_time.ToTimeT() + 24*60*60 - 1);
 119   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 120
 121   token = another.GenerateXTokenWithTime(gen_time.ToTimeT() + 24*60*60 + 1);
 122   EXPECT_FALSE(xtoken.CheckValidXToken(token));
 123 }
 124