1 //
2 //
3 // Copyright 2016 gRPC authors.
4 //
5 // Licensed under the Apache License, Version 2.0 (the "License");
6 // you may not use this file except in compliance with the License.
7 // You may obtain a copy of the License at
8 //
9 // http://www.apache.org/licenses/LICENSE-2.0
10 //
11 // Unless required by applicable law or agreed to in writing, software
12 // distributed under the License is distributed on an "AS IS" BASIS,
13 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 // See the License for the specific language governing permissions and
15 // limitations under the License.
16 //
17 //
18
19 #include "src/core/lib/slice/percent_encoding.h"
20
21 #include <grpc/support/alloc.h>
22 #include <stddef.h>
23
24 #include <memory>
25 #include <utility>
26
27 #include "absl/log/log.h"
28 #include "gtest/gtest.h"
29 #include "src/core/lib/slice/slice_string_helpers.h"
30 #include "src/core/util/string.h"
31 #include "test/core/test_util/test_config.h"
32
33 #define TEST_VECTOR(raw, encoded, dict) \
34 test_vector(raw, sizeof(raw) - 1, encoded, sizeof(encoded) - 1, dict)
35
36 #define TEST_NONCONFORMANT_VECTOR(encoded, permissive_unencoded) \
37 test_nonconformant_vector(encoded, sizeof(encoded) - 1, \
38 permissive_unencoded, \
39 sizeof(permissive_unencoded) - 1)
40
test_vector(const char * raw,size_t raw_length,const char * encoded,size_t encoded_length,grpc_core::PercentEncodingType type)41 static void test_vector(const char* raw, size_t raw_length, const char* encoded,
42 size_t encoded_length,
43 grpc_core::PercentEncodingType type) {
44 char* raw_msg = gpr_dump(raw, raw_length, GPR_DUMP_HEX | GPR_DUMP_ASCII);
45 char* encoded_msg =
46 gpr_dump(encoded, encoded_length, GPR_DUMP_HEX | GPR_DUMP_ASCII);
47 VLOG(2) << "Trial:\nraw = " << raw_msg << "\nencoded = " << encoded_msg;
48 gpr_free(raw_msg);
49 gpr_free(encoded_msg);
50
51 auto raw_slice = grpc_core::Slice::FromCopiedBuffer(raw, raw_length);
52 auto encoded_slice =
53 grpc_core::Slice::FromCopiedBuffer(encoded, encoded_length);
54 auto raw2encoded_slice = grpc_core::PercentEncodeSlice(raw_slice.Ref(), type);
55 auto encoded2raw_permissive_slice =
56 grpc_core::PermissivePercentDecodeSlice(encoded_slice.Ref());
57
58 char* raw2encoded_msg = grpc_dump_slice(raw2encoded_slice.c_slice(),
59 GPR_DUMP_HEX | GPR_DUMP_ASCII);
60 char* encoded2raw_permissive_msg = grpc_dump_slice(
61 encoded2raw_permissive_slice.c_slice(), GPR_DUMP_HEX | GPR_DUMP_ASCII);
62 VLOG(2) << "Result:\nraw2encoded = " << raw2encoded_msg
63 << "\nencoded2raw_permissive = " << encoded2raw_permissive_msg;
64 gpr_free(raw2encoded_msg);
65 gpr_free(encoded2raw_permissive_msg);
66
67 ASSERT_EQ(raw_slice, encoded2raw_permissive_slice);
68 ASSERT_EQ(encoded_slice, raw2encoded_slice);
69 }
70
test_nonconformant_vector(const char * encoded,size_t encoded_length,const char * permissive_unencoded,size_t permissive_unencoded_length)71 static void test_nonconformant_vector(const char* encoded,
72 size_t encoded_length,
73 const char* permissive_unencoded,
74 size_t permissive_unencoded_length) {
75 char* permissive_unencoded_msg =
76 gpr_dump(permissive_unencoded, permissive_unencoded_length,
77 GPR_DUMP_HEX | GPR_DUMP_ASCII);
78 char* encoded_msg =
79 gpr_dump(encoded, encoded_length, GPR_DUMP_HEX | GPR_DUMP_ASCII);
80 VLOG(2) << "Trial:\nraw = " << permissive_unencoded_msg
81 << "\nencoded = " << encoded_msg;
82 gpr_free(permissive_unencoded_msg);
83 gpr_free(encoded_msg);
84
85 auto permissive_unencoded_slice = grpc_core::Slice::FromCopiedBuffer(
86 permissive_unencoded, permissive_unencoded_length);
87 auto encoded_slice =
88 grpc_core::Slice::FromCopiedBuffer(encoded, encoded_length);
89 auto encoded2raw_permissive_slice =
90 grpc_core::PermissivePercentDecodeSlice(std::move(encoded_slice));
91
92 char* encoded2raw_permissive_msg = grpc_dump_slice(
93 encoded2raw_permissive_slice.c_slice(), GPR_DUMP_HEX | GPR_DUMP_ASCII);
94 VLOG(2) << "Result:\nencoded2raw_permissive = " << encoded2raw_permissive_msg;
95 gpr_free(encoded2raw_permissive_msg);
96
97 ASSERT_EQ(permissive_unencoded_slice, encoded2raw_permissive_slice);
98 }
99
TEST(PercentEncodingTest,MainTest)100 TEST(PercentEncodingTest, MainTest) {
101 TEST_VECTOR(
102 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_.~",
103 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_.~",
104 grpc_core::PercentEncodingType::URL);
105 TEST_VECTOR("\x00", "%00", grpc_core::PercentEncodingType::URL);
106 TEST_VECTOR("\x01", "%01", grpc_core::PercentEncodingType::URL);
107 TEST_VECTOR("a b", "a%20b", grpc_core::PercentEncodingType::URL);
108 TEST_VECTOR(" b", "%20b", grpc_core::PercentEncodingType::URL);
109 TEST_VECTOR("a b", "a b", grpc_core::PercentEncodingType::Compatible);
110 TEST_VECTOR(" b", " b", grpc_core::PercentEncodingType::Compatible);
111 TEST_VECTOR("\x0f", "%0F", grpc_core::PercentEncodingType::URL);
112 TEST_VECTOR("\xff", "%FF", grpc_core::PercentEncodingType::URL);
113 TEST_VECTOR("\xee", "%EE", grpc_core::PercentEncodingType::URL);
114 TEST_VECTOR("%2", "%252", grpc_core::PercentEncodingType::URL);
115 TEST_NONCONFORMANT_VECTOR("%", "%");
116 TEST_NONCONFORMANT_VECTOR("%A", "%A");
117 TEST_NONCONFORMANT_VECTOR("%AG", "%AG");
118 TEST_NONCONFORMANT_VECTOR("\0", "\0");
119 }
120
main(int argc,char ** argv)121 int main(int argc, char** argv) {
122 grpc::testing::TestEnvironment env(&argc, argv);
123 ::testing::InitGoogleTest(&argc, argv);
124 grpc::testing::TestGrpcScope grpc_scope;
125 return RUN_ALL_TESTS();
126 }
127