1 //
2 //
3 // Copyright 2015 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/ext/transport/chttp2/transport/bin_encoder.h"
20
21 #include <grpc/grpc.h>
22 #include <grpc/support/alloc.h>
23 #include <gtest/gtest.h>
24 #include <string.h>
25
26 #include <memory>
27
28 #include "absl/log/log.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 static int all_ok = 1;
34
expect_slice_eq(grpc_slice expected,grpc_slice slice,const char * debug,int line)35 static void expect_slice_eq(grpc_slice expected, grpc_slice slice,
36 const char* debug, int line) {
37 if (!grpc_slice_eq(slice, expected)) {
38 char* hs = grpc_dump_slice(slice, GPR_DUMP_HEX | GPR_DUMP_ASCII);
39 char* he = grpc_dump_slice(expected, GPR_DUMP_HEX | GPR_DUMP_ASCII);
40 LOG(ERROR) << "FAILED:" << line << ": " << debug << "\ngot: " << hs
41 << "\nwant: " << he;
42 gpr_free(hs);
43 gpr_free(he);
44 all_ok = 0;
45 }
46 grpc_slice_unref(expected);
47 grpc_slice_unref(slice);
48 }
49
B64(const char * s)50 static grpc_slice B64(const char* s) {
51 grpc_slice ss = grpc_slice_from_copied_string(s);
52 grpc_slice out = grpc_chttp2_base64_encode(ss);
53 grpc_slice_unref(ss);
54 return out;
55 }
56
HUFF(const char * s)57 static grpc_slice HUFF(const char* s) {
58 grpc_slice ss = grpc_slice_from_copied_string(s);
59 grpc_slice out = grpc_chttp2_huffman_compress(ss);
60 grpc_slice_unref(ss);
61 return out;
62 }
63
64 #define EXPECT_SLICE_EQ(expected, slice) \
65 expect_slice_eq( \
66 grpc_slice_from_copied_buffer(expected, sizeof(expected) - 1), slice, \
67 #slice, __LINE__);
68
expect_combined_equiv(const char * s,size_t len,int line)69 static void expect_combined_equiv(const char* s, size_t len, int line) {
70 grpc_slice input = grpc_slice_from_copied_buffer(s, len);
71 grpc_slice base64 = grpc_chttp2_base64_encode(input);
72 grpc_slice expect = grpc_chttp2_huffman_compress(base64);
73 uint32_t wire_size;
74 grpc_slice got =
75 grpc_chttp2_base64_encode_and_huffman_compress(input, &wire_size);
76 if (!grpc_slice_eq(expect, got)) {
77 char* t = grpc_dump_slice(input, GPR_DUMP_HEX | GPR_DUMP_ASCII);
78 char* e = grpc_dump_slice(expect, GPR_DUMP_HEX | GPR_DUMP_ASCII);
79 char* g = grpc_dump_slice(got, GPR_DUMP_HEX | GPR_DUMP_ASCII);
80 LOG(ERROR) << "FAILED:" << line << ":\ntest: " << t << "\ngot: " << g
81 << "\nwant: " << e;
82 gpr_free(t);
83 gpr_free(e);
84 gpr_free(g);
85 all_ok = 0;
86 }
87 grpc_slice_unref(input);
88 grpc_slice_unref(base64);
89 grpc_slice_unref(expect);
90 grpc_slice_unref(got);
91 }
92
93 #define EXPECT_COMBINED_EQUIV(x) \
94 expect_combined_equiv(x, sizeof(x) - 1, __LINE__)
95
expect_binary_header(const char * hdr,int binary)96 static void expect_binary_header(const char* hdr, int binary) {
97 if (grpc_is_binary_header(grpc_slice_from_static_string(hdr)) != binary) {
98 LOG(ERROR) << "FAILED: expected header '" << hdr << "' to be "
99 << (binary ? "binary" : "not binary");
100 all_ok = 0;
101 }
102 }
103
TEST(BinEncoderTest,MainTest)104 TEST(BinEncoderTest, MainTest) {
105 // Base64 test vectors from RFC 4648, with padding removed
106 // BASE64("") = ""
107 EXPECT_SLICE_EQ("", B64(""));
108 // BASE64("f") = "Zg"
109 EXPECT_SLICE_EQ("Zg", B64("f"));
110 // BASE64("fo") = "Zm8"
111 EXPECT_SLICE_EQ("Zm8", B64("fo"));
112 // BASE64("foo") = "Zm9v"
113 EXPECT_SLICE_EQ("Zm9v", B64("foo"));
114 // BASE64("foob") = "Zm9vYg"
115 EXPECT_SLICE_EQ("Zm9vYg", B64("foob"));
116 // BASE64("fooba") = "Zm9vYmE"
117 EXPECT_SLICE_EQ("Zm9vYmE", B64("fooba"));
118 // BASE64("foobar") = "Zm9vYmFy"
119 EXPECT_SLICE_EQ("Zm9vYmFy", B64("foobar"));
120
121 EXPECT_SLICE_EQ("wMHCw8TF", B64("\xc0\xc1\xc2\xc3\xc4\xc5"));
122
123 // Huffman encoding tests
124 EXPECT_SLICE_EQ("\xf1\xe3\xc2\xe5\xf2\x3a\x6b\xa0\xab\x90\xf4\xff",
125 HUFF("www.example.com"));
126 EXPECT_SLICE_EQ("\xa8\xeb\x10\x64\x9c\xbf", HUFF("no-cache"));
127 EXPECT_SLICE_EQ("\x25\xa8\x49\xe9\x5b\xa9\x7d\x7f", HUFF("custom-key"));
128 EXPECT_SLICE_EQ("\x25\xa8\x49\xe9\x5b\xb8\xe8\xb4\xbf", HUFF("custom-value"));
129 EXPECT_SLICE_EQ("\xae\xc3\x77\x1a\x4b", HUFF("private"));
130 EXPECT_SLICE_EQ(
131 "\xd0\x7a\xbe\x94\x10\x54\xd4\x44\xa8\x20\x05\x95\x04\x0b\x81\x66\xe0\x82"
132 "\xa6\x2d\x1b\xff",
133 HUFF("Mon, 21 Oct 2013 20:13:21 GMT"));
134 EXPECT_SLICE_EQ(
135 "\x9d\x29\xad\x17\x18\x63\xc7\x8f\x0b\x97\xc8\xe9\xae\x82\xae\x43\xd3",
136 HUFF("https://www.example.com"));
137
138 // Various test vectors for combined encoding
139 EXPECT_COMBINED_EQUIV("");
140 EXPECT_COMBINED_EQUIV("f");
141 EXPECT_COMBINED_EQUIV("fo");
142 EXPECT_COMBINED_EQUIV("foo");
143 EXPECT_COMBINED_EQUIV("foob");
144 EXPECT_COMBINED_EQUIV("fooba");
145 EXPECT_COMBINED_EQUIV("foobar");
146 EXPECT_COMBINED_EQUIV("www.example.com");
147 EXPECT_COMBINED_EQUIV("no-cache");
148 EXPECT_COMBINED_EQUIV("custom-key");
149 EXPECT_COMBINED_EQUIV("custom-value");
150 EXPECT_COMBINED_EQUIV("private");
151 EXPECT_COMBINED_EQUIV("Mon, 21 Oct 2013 20:13:21 GMT");
152 EXPECT_COMBINED_EQUIV("https://www.example.com");
153 EXPECT_COMBINED_EQUIV(
154 "\x00\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0a\x0b\x0c\x0d\x0e\x0f"
155 "\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19\x1a\x1b\x1c\x1d\x1e\x1f"
156 "\x20\x21\x22\x23\x24\x25\x26\x27\x28\x29\x2a\x2b\x2c\x2d\x2e\x2f"
157 "\x30\x31\x32\x33\x34\x35\x36\x37\x38\x39\x3a\x3b\x3c\x3d\x3e\x3f"
158 "\x40\x41\x42\x43\x44\x45\x46\x47\x48\x49\x4a\x4b\x4c\x4d\x4e\x4f"
159 "\x50\x51\x52\x53\x54\x55\x56\x57\x58\x59\x5a\x5b\x5c\x5d\x5e\x5f"
160 "\x60\x61\x62\x63\x64\x65\x66\x67\x68\x69\x6a\x6b\x6c\x6d\x6e\x6f"
161 "\x70\x71\x72\x73\x74\x75\x76\x77\x78\x79\x7a\x7b\x7c\x7d\x7e\x7f"
162 "\x80\x81\x82\x83\x84\x85\x86\x87\x88\x89\x8a\x8b\x8c\x8d\x8e\x8f"
163 "\x90\x91\x92\x93\x94\x95\x96\x97\x98\x99\x9a\x9b\x9c\x9d\x9e\x9f"
164 "\xa0\xa1\xa2\xa3\xa4\xa5\xa6\xa7\xa8\xa9\xaa\xab\xac\xad\xae\xaf"
165 "\xb0\xb1\xb2\xb3\xb4\xb5\xb6\xb7\xb8\xb9\xba\xbb\xbc\xbd\xbe\xbf"
166 "\xc0\xc1\xc2\xc3\xc4\xc5\xc6\xc7\xc8\xc9\xca\xcb\xcc\xcd\xce\xcf"
167 "\xd0\xd1\xd2\xd3\xd4\xd5\xd6\xd7\xd8\xd9\xda\xdb\xdc\xdd\xde\xdf"
168 "\xe0\xe1\xe2\xe3\xe4\xe5\xe6\xe7\xe8\xe9\xea\xeb\xec\xed\xee\xef"
169 "\xf0\xf1\xf2\xf3\xf4\xf5\xf6\xf7\xf8\xf9\xfa\xfb\xfc\xfd\xfe\xff");
170
171 expect_binary_header("foo-bin", 1);
172 expect_binary_header("foo-bar", 0);
173 expect_binary_header("-bin", 0);
174 }
175
main(int argc,char ** argv)176 int main(int argc, char** argv) {
177 grpc::testing::TestEnvironment env(&argc, argv);
178 ::testing::InitGoogleTest(&argc, argv);
179 grpc::testing::TestGrpcScope grpc_scope;
180 return RUN_ALL_TESTS();
181 }
182