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 "test/core/end2end/end2end_tests.h"
20
21 #include <string.h>
22
23 #include <grpc/support/alloc.h>
24 #include <grpc/support/log.h>
25 #include <grpc/support/string_util.h>
26 #include <grpc/support/sync.h>
27
28 #include "src/core/ext/filters/client_channel/client_channel.h"
29 #include "src/core/ext/filters/http/server/http_server_filter.h"
30 #include "src/core/ext/transport/chttp2/transport/chttp2_transport.h"
31 #include "src/core/lib/channel/connected_channel.h"
32 #include "src/core/lib/gpr/env.h"
33 #include "src/core/lib/gpr/host_port.h"
34 #include "src/core/lib/surface/channel.h"
35 #include "src/core/lib/surface/server.h"
36 #include "test/core/end2end/fixtures/http_proxy_fixture.h"
37 #include "test/core/util/port.h"
38 #include "test/core/util/test_config.h"
39
40 typedef struct fullstack_fixture_data {
41 char* server_addr;
42 grpc_end2end_http_proxy* proxy;
43 } fullstack_fixture_data;
44
chttp2_create_fixture_fullstack(grpc_channel_args * client_args,grpc_channel_args * server_args)45 static grpc_end2end_test_fixture chttp2_create_fixture_fullstack(
46 grpc_channel_args* client_args, grpc_channel_args* server_args) {
47 grpc_end2end_test_fixture f;
48 memset(&f, 0, sizeof(f));
49 fullstack_fixture_data* ffd = static_cast<fullstack_fixture_data*>(
50 gpr_malloc(sizeof(fullstack_fixture_data)));
51 const int server_port = grpc_pick_unused_port_or_die();
52 gpr_join_host_port(&ffd->server_addr, "localhost", server_port);
53
54 /* Passing client_args to proxy_create for the case of checking for proxy auth
55 */
56 ffd->proxy = grpc_end2end_http_proxy_create(client_args);
57
58 f.fixture_data = ffd;
59 f.cq = grpc_completion_queue_create_for_next(nullptr);
60 f.shutdown_cq = grpc_completion_queue_create_for_pluck(nullptr);
61
62 return f;
63 }
64
chttp2_init_client_fullstack(grpc_end2end_test_fixture * f,grpc_channel_args * client_args)65 void chttp2_init_client_fullstack(grpc_end2end_test_fixture* f,
66 grpc_channel_args* client_args) {
67 fullstack_fixture_data* ffd =
68 static_cast<fullstack_fixture_data*>(f->fixture_data);
69 char* proxy_uri;
70
71 /* If testing for proxy auth, add credentials to proxy uri */
72 const grpc_arg* proxy_auth_arg =
73 grpc_channel_args_find(client_args, GRPC_ARG_HTTP_PROXY_AUTH_CREDS);
74 const char* proxy_auth_str = grpc_channel_arg_get_string(proxy_auth_arg);
75 if (proxy_auth_str == nullptr) {
76 gpr_asprintf(&proxy_uri, "http://%s",
77 grpc_end2end_http_proxy_get_proxy_name(ffd->proxy));
78 } else {
79 gpr_asprintf(&proxy_uri, "http://%s@%s", proxy_auth_str,
80 grpc_end2end_http_proxy_get_proxy_name(ffd->proxy));
81 }
82 gpr_setenv("http_proxy", proxy_uri);
83 gpr_free(proxy_uri);
84 f->client =
85 grpc_insecure_channel_create(ffd->server_addr, client_args, nullptr);
86 GPR_ASSERT(f->client);
87 }
88
chttp2_init_server_fullstack(grpc_end2end_test_fixture * f,grpc_channel_args * server_args)89 void chttp2_init_server_fullstack(grpc_end2end_test_fixture* f,
90 grpc_channel_args* server_args) {
91 fullstack_fixture_data* ffd =
92 static_cast<fullstack_fixture_data*>(f->fixture_data);
93 if (f->server) {
94 grpc_server_destroy(f->server);
95 }
96 f->server = grpc_server_create(server_args, nullptr);
97 grpc_server_register_completion_queue(f->server, f->cq, nullptr);
98 GPR_ASSERT(grpc_server_add_insecure_http2_port(f->server, ffd->server_addr));
99 grpc_server_start(f->server);
100 }
101
chttp2_tear_down_fullstack(grpc_end2end_test_fixture * f)102 void chttp2_tear_down_fullstack(grpc_end2end_test_fixture* f) {
103 fullstack_fixture_data* ffd =
104 static_cast<fullstack_fixture_data*>(f->fixture_data);
105 gpr_free(ffd->server_addr);
106 grpc_end2end_http_proxy_destroy(ffd->proxy);
107 gpr_free(ffd);
108 }
109
110 /* All test configurations */
111 static grpc_end2end_test_config configs[] = {
112 {"chttp2/fullstack",
113 FEATURE_MASK_SUPPORTS_DELAYED_CONNECTION |
114 FEATURE_MASK_SUPPORTS_CLIENT_CHANNEL |
115 FEATURE_MASK_SUPPORTS_AUTHORITY_HEADER,
116 nullptr, chttp2_create_fixture_fullstack, chttp2_init_client_fullstack,
117 chttp2_init_server_fullstack, chttp2_tear_down_fullstack},
118 };
119
main(int argc,char ** argv)120 int main(int argc, char** argv) {
121 size_t i;
122
123 grpc_test_init(argc, argv);
124 grpc_end2end_tests_pre_init();
125 grpc_init();
126
127 for (i = 0; i < sizeof(configs) / sizeof(*configs); i++) {
128 grpc_end2end_tests(argc, argv, configs[i]);
129 }
130
131 grpc_shutdown();
132
133 return 0;
134 }
135