1 // Copyright 2021 Google LLC
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 // http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 //
15 ///////////////////////////////////////////////////////////////////////////////
16 #include "tink/internal/util.h"
17
18 #include <string>
19
20 #include "gmock/gmock.h"
21 #include "gtest/gtest.h"
22 #include "absl/strings/string_view.h"
23
24 namespace crypto {
25 namespace tink {
26 namespace internal {
27 namespace {
28
29 using ::testing::IsFalse;
30 using ::testing::IsTrue;
31
32 constexpr absl::string_view kLongString =
33 "a long buffer with \n several \n newlines";
34
TEST(UtilTest,EnsureStringNonNull)35 TEST(UtilTest, EnsureStringNonNull) {
36 // Purposely create a string_view from nullptr.
37 auto null_str = absl::string_view(nullptr, 0);
38 EXPECT_EQ(EnsureStringNonNull(null_str), absl::string_view(""));
39 auto uninit_str = absl::string_view();
40 EXPECT_EQ(EnsureStringNonNull(uninit_str), absl::string_view(""));
41 auto regular_str = absl::string_view("This is a non-empty non-null str");
42 EXPECT_EQ(EnsureStringNonNull(regular_str), regular_str);
43 }
44
TEST(BuffersOverlapTest,BufferOverlapEmpty)45 TEST(BuffersOverlapTest, BufferOverlapEmpty) {
46 absl::string_view empty = "";
47 EXPECT_FALSE(BuffersOverlap(empty, empty));
48 EXPECT_FALSE(BuffersOverlap(empty, ""));
49 }
50
TEST(BuffersOverlapTest,BufferOverlapSeparate)51 TEST(BuffersOverlapTest, BufferOverlapSeparate) {
52 absl::string_view first = "first";
53 absl::string_view second = "second";
54 EXPECT_FALSE(BuffersOverlap(first, second));
55 EXPECT_TRUE(BuffersOverlap(first, first));
56 }
57
TEST(BuffersOverlapTest,BufferOverlap)58 TEST(BuffersOverlapTest, BufferOverlap) {
59 absl::string_view long_buffer = kLongString;
60
61 EXPECT_TRUE(BuffersOverlap(long_buffer, long_buffer));
62
63 EXPECT_TRUE(
64 BuffersOverlap(long_buffer.substr(0, 10), long_buffer.substr(9, 5)));
65 EXPECT_FALSE(
66 BuffersOverlap(long_buffer.substr(0, 10), long_buffer.substr(10, 5)));
67
68 EXPECT_TRUE(
69 BuffersOverlap(long_buffer.substr(9, 5), long_buffer.substr(0, 10)));
70 EXPECT_FALSE(
71 BuffersOverlap(long_buffer.substr(10, 5), long_buffer.substr(0, 10)));
72 }
73
TEST(BuffersAreIdenticalTest,EmptyString)74 TEST(BuffersAreIdenticalTest, EmptyString) {
75 std::string empty_str = "";
76 absl::string_view empty = "";
77 EXPECT_FALSE(BuffersAreIdentical(empty, empty));
78 EXPECT_FALSE(BuffersAreIdentical(absl::string_view(empty_str),
79 absl::string_view(empty_str)));
80 EXPECT_FALSE(BuffersAreIdentical(empty, ""));
81 EXPECT_FALSE(BuffersAreIdentical(empty, absl::string_view(empty_str)));
82 }
83
TEST(BuffersAreIdenticalTest,BuffersAreIdentical)84 TEST(BuffersAreIdenticalTest, BuffersAreIdentical) {
85 auto some_string = std::string(kLongString);
86 auto buffer = absl::string_view(some_string);
87 EXPECT_TRUE(BuffersAreIdentical(buffer, buffer));
88 // Make sure BuffersAreIdentical is not checking for string equality.
89 std::string identical_string = some_string;
90 EXPECT_FALSE(
91 BuffersAreIdentical(buffer, absl::string_view(identical_string)));
92 }
93
TEST(BuffersAreIdenticalTest,PartialOverlapFails)94 TEST(BuffersAreIdenticalTest, PartialOverlapFails) {
95 auto some_string = std::string(kLongString);
96 auto buffer = absl::string_view(some_string);
97 EXPECT_FALSE(BuffersAreIdentical(buffer.substr(0, 10), buffer.substr(9, 5)));
98 EXPECT_FALSE(BuffersAreIdentical(buffer.substr(0, 10), buffer.substr(10, 5)));
99 EXPECT_FALSE(BuffersAreIdentical(buffer.substr(9, 5), buffer.substr(0, 10)));
100 EXPECT_FALSE(BuffersAreIdentical(buffer.substr(10, 5), buffer.substr(0, 10)));
101 }
102
TEST(UtilTest,IsPrintableAscii)103 TEST(UtilTest, IsPrintableAscii) {
104 const std::string input =
105 "!\"#$%&'()*+,-./"
106 "0123456789:;<=>?@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`"
107 "abcdefghijklmnopqrstuvwxyz{|}~";
108 EXPECT_THAT(IsPrintableAscii(input), IsTrue());
109 }
110
TEST(UtilTest,IsNotPrintableAscii)111 TEST(UtilTest, IsNotPrintableAscii) {
112 EXPECT_THAT(IsPrintableAscii("\n"), IsFalse());
113 EXPECT_THAT(IsPrintableAscii("\t"), IsFalse());
114 EXPECT_THAT(IsPrintableAscii(" "), IsFalse());
115 EXPECT_THAT(IsPrintableAscii(std::string("\x7f", 1)), IsFalse());
116 EXPECT_THAT(IsPrintableAscii("ö"), IsFalse());
117 }
118
119 } // namespace
120 } // namespace internal
121 } // namespace tink
122 } // namespace crypto
123