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
TEST(XPrivetTokenTest,Generation)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
TEST(XPrivetTokenTest,CheckingValid)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
TEST(XPrivetTokenTest,CheckingInvalid)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
125