mirror of https://github.com/grpc/grpc.git
Adam Michalik
9 years ago
parent
d9b8db6c17
commit
46cc2acc3e
10 changed files with 2872 additions and 1304 deletions
@ -0,0 +1,160 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2015, Google Inc. |
||||
* All rights reserved. |
||||
* |
||||
* Redistribution and use in source and binary forms, with or without |
||||
* modification, are permitted provided that the following conditions are |
||||
* met: |
||||
* |
||||
* * Redistributions of source code must retain the above copyright |
||||
* notice, this list of conditions and the following disclaimer. |
||||
* * Redistributions in binary form must reproduce the above |
||||
* copyright notice, this list of conditions and the following disclaimer |
||||
* in the documentation and/or other materials provided with the |
||||
* distribution. |
||||
* * Neither the name of Google Inc. nor the names of its |
||||
* contributors may be used to endorse or promote products derived from |
||||
* this software without specific prior written permission. |
||||
* |
||||
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
||||
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
||||
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
||||
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
||||
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
||||
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
||||
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
||||
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
||||
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
||||
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
||||
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
||||
* |
||||
*/ |
||||
|
||||
#include "test/core/end2end/end2end_tests.h" |
||||
|
||||
#include <fcntl.h> |
||||
#include <string.h> |
||||
#include <sys/socket.h> |
||||
#include <sys/types.h> |
||||
|
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <grpc/support/sync.h> |
||||
#include <grpc/support/thd.h> |
||||
#include <grpc/support/useful.h> |
||||
#include "src/core/ext/client_config/client_channel.h" |
||||
#include "src/core/ext/transport/chttp2/transport/chttp2_transport.h" |
||||
#include "src/core/lib/channel/compress_filter.h" |
||||
#include "src/core/lib/channel/connected_channel.h" |
||||
#include "src/core/lib/channel/http_client_filter.h" |
||||
#include "src/core/lib/channel/http_server_filter.h" |
||||
#include "src/core/lib/iomgr/endpoint_pair.h" |
||||
#include "src/core/lib/iomgr/iomgr.h" |
||||
#include "src/core/lib/iomgr/socket_utils_posix.h" |
||||
#include "src/core/lib/iomgr/tcp_posix.h" |
||||
#include "src/core/lib/iomgr/unix_sockets_posix.h" |
||||
#include "src/core/lib/surface/channel.h" |
||||
#include "src/core/lib/surface/server.h" |
||||
#include "test/core/util/port.h" |
||||
#include "test/core/util/test_config.h" |
||||
|
||||
/* chttp2 transport that is immediately available (used for testing
|
||||
connected_channel without a client_channel */ |
||||
|
||||
static void server_setup_transport(void *ts, grpc_transport *transport) { |
||||
grpc_end2end_test_fixture *f = ts; |
||||
grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
||||
grpc_server_setup_transport(&exec_ctx, f->server, transport, |
||||
grpc_server_get_channel_args(f->server)); |
||||
grpc_exec_ctx_finish(&exec_ctx); |
||||
} |
||||
|
||||
typedef struct { |
||||
int fd_pair[2]; |
||||
} sp_fixture_data; |
||||
|
||||
|
||||
static void create_sockets(int sv[2]) { |
||||
int flags; |
||||
grpc_create_socketpair_if_unix(sv); |
||||
flags = fcntl(sv[0], F_GETFL, 0); |
||||
GPR_ASSERT(fcntl(sv[0], F_SETFL, flags | O_NONBLOCK) == 0); |
||||
flags = fcntl(sv[1], F_GETFL, 0); |
||||
GPR_ASSERT(fcntl(sv[1], F_SETFL, flags | O_NONBLOCK) == 0); |
||||
GPR_ASSERT(grpc_set_socket_no_sigpipe_if_possible(sv[0])); |
||||
GPR_ASSERT(grpc_set_socket_no_sigpipe_if_possible(sv[1])); |
||||
} |
||||
|
||||
static grpc_end2end_test_fixture chttp2_create_fixture_socketpair( |
||||
grpc_channel_args *client_args, grpc_channel_args *server_args) { |
||||
sp_fixture_data* fixture_data = gpr_malloc(sizeof(*fixture_data)); |
||||
|
||||
grpc_end2end_test_fixture f; |
||||
memset(&f, 0, sizeof(f)); |
||||
f.fixture_data = fixture_data; |
||||
f.cq = grpc_completion_queue_create(NULL); |
||||
|
||||
create_sockets(fixture_data->fd_pair); |
||||
|
||||
return f; |
||||
} |
||||
|
||||
static void chttp2_init_client_socketpair(grpc_end2end_test_fixture *f, |
||||
grpc_channel_args *client_args) { |
||||
grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
||||
sp_fixture_data *sfd = f->fixture_data; |
||||
|
||||
f->client = grpc_insecure_channel_create_from_fd( |
||||
"fixture_client", sfd->fd_pair[0], client_args); |
||||
GPR_ASSERT(f->client); |
||||
|
||||
grpc_exec_ctx_finish(&exec_ctx); |
||||
} |
||||
|
||||
static void chttp2_init_server_socketpair(grpc_end2end_test_fixture *f, |
||||
grpc_channel_args *server_args) { |
||||
grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT; |
||||
sp_fixture_data *sfd = f->fixture_data; |
||||
grpc_transport *transport; |
||||
GPR_ASSERT(!f->server); |
||||
f->server = grpc_server_create(server_args, NULL); |
||||
grpc_server_register_completion_queue(f->server, f->cq, NULL); |
||||
grpc_server_start(f->server); |
||||
grpc_endpoint *server_endpoint = grpc_tcp_create( |
||||
grpc_fd_create(sfd->fd_pair[1], "fixture_server"), |
||||
65536 /* read_slice_size */, "fixture_server"); |
||||
|
||||
transport = |
||||
grpc_create_chttp2_transport(&exec_ctx, server_args, server_endpoint, 0); |
||||
server_setup_transport(f, transport); |
||||
grpc_chttp2_transport_start_reading(&exec_ctx, transport, NULL, 0); |
||||
grpc_exec_ctx_finish(&exec_ctx); |
||||
} |
||||
|
||||
static void chttp2_tear_down_socketpair(grpc_end2end_test_fixture *f) { |
||||
gpr_free(f->fixture_data); |
||||
} |
||||
|
||||
/* All test configurations */ |
||||
static grpc_end2end_test_config configs[] = { |
||||
{"chttp2/socketpair", 0, chttp2_create_fixture_socketpair, |
||||
chttp2_init_client_socketpair, chttp2_init_server_socketpair, |
||||
chttp2_tear_down_socketpair}, |
||||
}; |
||||
|
||||
int main(int argc, char **argv) { |
||||
size_t i; |
||||
|
||||
grpc_test_init(argc, argv); |
||||
grpc_end2end_tests_pre_init(); |
||||
grpc_init(); |
||||
|
||||
for (i = 0; i < sizeof(configs) / sizeof(*configs); i++) { |
||||
grpc_end2end_tests(argc, argv, configs[i]); |
||||
} |
||||
|
||||
grpc_shutdown(); |
||||
|
||||
return 0; |
||||
} |
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in new issue