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 <stddef.h>
22
23 #include <memory>
24 #include <utility>
25
26 #include "gtest/gtest.h"
27
28 #include <grpc/support/alloc.h>
29 #include <grpc/support/log.h>
30
31 #include "src/core/lib/gpr/string.h"
32 #include "src/core/lib/slice/slice_string_helpers.h"
33 #include "test/core/util/test_config.h"
34
35 #define TEST_VECTOR(raw, encoded, dict) \
36 test_vector(raw, sizeof(raw) - 1, encoded, sizeof(encoded) - 1, dict)
37
38 #define TEST_NONCONFORMANT_VECTOR(encoded, permissive_unencoded) \
39 test_nonconformant_vector(encoded, sizeof(encoded) - 1, \
40 permissive_unencoded, \
41 sizeof(permissive_unencoded) - 1)
42
test_vector(const char * raw,size_t raw_length,const char * encoded,size_t encoded_length,grpc_core::PercentEncodingType type)43 static void test_vector(const char* raw, size_t raw_length, const char* encoded,
44 size_t encoded_length,
45 grpc_core::PercentEncodingType type) {
46 char* raw_msg = gpr_dump(raw, raw_length, GPR_DUMP_HEX | GPR_DUMP_ASCII);
47 char* encoded_msg =
48 gpr_dump(encoded, encoded_length, GPR_DUMP_HEX | GPR_DUMP_ASCII);
49 gpr_log(GPR_DEBUG, "Trial:\nraw = %s\nencoded = %s", raw_msg, encoded_msg);
50 gpr_free(raw_msg);
51 gpr_free(encoded_msg);
52
53 auto raw_slice = grpc_core::Slice::FromCopiedBuffer(raw, raw_length);
54 auto encoded_slice =
55 grpc_core::Slice::FromCopiedBuffer(encoded, encoded_length);
56 auto raw2encoded_slice = grpc_core::PercentEncodeSlice(raw_slice.Ref(), type);
57 auto encoded2raw_permissive_slice =
58 grpc_core::PermissivePercentDecodeSlice(encoded_slice.Ref());
59
60 char* raw2encoded_msg = grpc_dump_slice(raw2encoded_slice.c_slice(),
61 GPR_DUMP_HEX | GPR_DUMP_ASCII);
62 char* encoded2raw_permissive_msg = grpc_dump_slice(
63 encoded2raw_permissive_slice.c_slice(), GPR_DUMP_HEX | GPR_DUMP_ASCII);
64 gpr_log(GPR_DEBUG,
65 "Result:\nraw2encoded = %s\nencoded2raw_permissive "
66 "= %s",
67 raw2encoded_msg, encoded2raw_permissive_msg);
68 gpr_free(raw2encoded_msg);
69 gpr_free(encoded2raw_permissive_msg);
70
71 ASSERT_EQ(raw_slice, encoded2raw_permissive_slice);
72 ASSERT_EQ(encoded_slice, raw2encoded_slice);
73 }
74
test_nonconformant_vector(const char * encoded,size_t encoded_length,const char * permissive_unencoded,size_t permissive_unencoded_length)75 static void test_nonconformant_vector(const char* encoded,
76 size_t encoded_length,
77 const char* permissive_unencoded,
78 size_t permissive_unencoded_length) {
79 char* permissive_unencoded_msg =
80 gpr_dump(permissive_unencoded, permissive_unencoded_length,
81 GPR_DUMP_HEX | GPR_DUMP_ASCII);
82 char* encoded_msg =
83 gpr_dump(encoded, encoded_length, GPR_DUMP_HEX | GPR_DUMP_ASCII);
84 gpr_log(GPR_DEBUG, "Trial:\nraw = %s\nencoded = %s", permissive_unencoded_msg,
85 encoded_msg);
86 gpr_free(permissive_unencoded_msg);
87 gpr_free(encoded_msg);
88
89 auto permissive_unencoded_slice = grpc_core::Slice::FromCopiedBuffer(
90 permissive_unencoded, permissive_unencoded_length);
91 auto encoded_slice =
92 grpc_core::Slice::FromCopiedBuffer(encoded, encoded_length);
93 auto encoded2raw_permissive_slice =
94 grpc_core::PermissivePercentDecodeSlice(std::move(encoded_slice));
95
96 char* encoded2raw_permissive_msg = grpc_dump_slice(
97 encoded2raw_permissive_slice.c_slice(), GPR_DUMP_HEX | GPR_DUMP_ASCII);
98 gpr_log(GPR_DEBUG, "Result:\nencoded2raw_permissive = %s",
99 encoded2raw_permissive_msg);
100 gpr_free(encoded2raw_permissive_msg);
101
102 ASSERT_EQ(permissive_unencoded_slice, encoded2raw_permissive_slice);
103 }
104
TEST(PercentEncodingTest,MainTest)105 TEST(PercentEncodingTest, MainTest) {
106 TEST_VECTOR(
107 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_.~",
108 "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789-_.~",
109 grpc_core::PercentEncodingType::URL);
110 TEST_VECTOR("\x00", "%00", grpc_core::PercentEncodingType::URL);
111 TEST_VECTOR("\x01", "%01", grpc_core::PercentEncodingType::URL);
112 TEST_VECTOR("a b", "a%20b", grpc_core::PercentEncodingType::URL);
113 TEST_VECTOR(" b", "%20b", grpc_core::PercentEncodingType::URL);
114 TEST_VECTOR("a b", "a b", grpc_core::PercentEncodingType::Compatible);
115 TEST_VECTOR(" b", " b", grpc_core::PercentEncodingType::Compatible);
116 TEST_VECTOR("\x0f", "%0F", grpc_core::PercentEncodingType::URL);
117 TEST_VECTOR("\xff", "%FF", grpc_core::PercentEncodingType::URL);
118 TEST_VECTOR("\xee", "%EE", grpc_core::PercentEncodingType::URL);
119 TEST_VECTOR("%2", "%252", grpc_core::PercentEncodingType::URL);
120 TEST_NONCONFORMANT_VECTOR("%", "%");
121 TEST_NONCONFORMANT_VECTOR("%A", "%A");
122 TEST_NONCONFORMANT_VECTOR("%AG", "%AG");
123 TEST_NONCONFORMANT_VECTOR("\0", "\0");
124 }
125
main(int argc,char ** argv)126 int main(int argc, char** argv) {
127 grpc::testing::TestEnvironment env(&argc, argv);
128 ::testing::InitGoogleTest(&argc, argv);
129 grpc::testing::TestGrpcScope grpc_scope;
130 return RUN_ALL_TESTS();
131 }
132