Line data    Source code

       1             : /*
       2             :  *
       3             :  * Copyright 2015, Google Inc.
       4             :  * All rights reserved.
       5             :  *
       6             :  * Redistribution and use in source and binary forms, with or without
       7             :  * modification, are permitted provided that the following conditions are
       8             :  * met:
       9             :  *
      10             :  *     * Redistributions of source code must retain the above copyright
      11             :  * notice, this list of conditions and the following disclaimer.
      12             :  *     * Redistributions in binary form must reproduce the above
      13             :  * copyright notice, this list of conditions and the following disclaimer
      14             :  * in the documentation and/or other materials provided with the
      15             :  * distribution.
      16             :  *     * Neither the name of Google Inc. nor the names of its
      17             :  * contributors may be used to endorse or promote products derived from
      18             :  * this software without specific prior written permission.
      19             :  *
      20             :  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
      21             :  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
      22             :  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
      23             :  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
      24             :  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
      25             :  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
      26             :  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
      27             :  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
      28             :  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
      29             :  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
      30             :  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
      31             :  *
      32             :  */
      33             : 
      34             : #include "test/core/end2end/end2end_tests.h"
      35             : 
      36             : #include <string.h>
      37             : 
      38             : #include "src/core/channel/client_channel.h"
      39             : #include "src/core/channel/connected_channel.h"
      40             : #include "src/core/channel/http_server_filter.h"
      41             : #include "src/core/surface/channel.h"
      42             : #include "src/core/surface/server.h"
      43             : #include "src/core/transport/chttp2_transport.h"
      44             : #include <grpc/support/alloc.h>
      45             : #include <grpc/support/host_port.h>
      46             : #include <grpc/support/log.h>
      47             : #include <grpc/support/sync.h>
      48             : #include <grpc/support/thd.h>
      49             : #include <grpc/support/useful.h>
      50             : #include "test/core/end2end/fixtures/proxy.h"
      51             : #include "test/core/util/port.h"
      52             : #include "test/core/util/test_config.h"
      53             : 
      54             : typedef struct fullstack_fixture_data {
      55             :   grpc_end2end_proxy *proxy;
      56             : } fullstack_fixture_data;
      57             : 
      58         126 : static grpc_server *create_proxy_server(const char *port) {
      59         126 :   grpc_server *s = grpc_server_create(NULL, NULL);
      60         126 :   GPR_ASSERT(grpc_server_add_insecure_http2_port(s, port));
      61         126 :   return s;
      62             : }
      63             : 
      64         126 : static grpc_channel *create_proxy_client(const char *target) {
      65         126 :   return grpc_insecure_channel_create(target, NULL, NULL);
      66             : }
      67             : 
      68             : static const grpc_end2end_proxy_def proxy_def = {create_proxy_server,
      69             :                                                  create_proxy_client};
      70             : 
      71         126 : static grpc_end2end_test_fixture chttp2_create_fixture_fullstack(
      72             :     grpc_channel_args *client_args, grpc_channel_args *server_args) {
      73             :   grpc_end2end_test_fixture f;
      74         126 :   fullstack_fixture_data *ffd = gpr_malloc(sizeof(fullstack_fixture_data));
      75         126 :   memset(&f, 0, sizeof(f));
      76             : 
      77         126 :   ffd->proxy = grpc_end2end_proxy_create(&proxy_def);
      78             : 
      79         126 :   f.fixture_data = ffd;
      80         126 :   f.cq = grpc_completion_queue_create(NULL);
      81             : 
      82         126 :   return f;
      83             : }
      84             : 
      85         126 : void chttp2_init_client_fullstack(grpc_end2end_test_fixture *f,
      86             :                                   grpc_channel_args *client_args) {
      87         126 :   fullstack_fixture_data *ffd = f->fixture_data;
      88         126 :   f->client = grpc_insecure_channel_create(
      89             :       grpc_end2end_proxy_get_client_target(ffd->proxy), client_args, NULL);
      90         126 :   GPR_ASSERT(f->client);
      91         126 : }
      92             : 
      93         128 : void chttp2_init_server_fullstack(grpc_end2end_test_fixture *f,
      94             :                                   grpc_channel_args *server_args) {
      95         128 :   fullstack_fixture_data *ffd = f->fixture_data;
      96         128 :   if (f->server) {
      97           2 :     grpc_server_destroy(f->server);
      98             :   }
      99         128 :   f->server = grpc_server_create(server_args, NULL);
     100         128 :   grpc_server_register_completion_queue(f->server, f->cq, NULL);
     101         128 :   GPR_ASSERT(grpc_server_add_insecure_http2_port(
     102             :       f->server, grpc_end2end_proxy_get_server_port(ffd->proxy)));
     103         128 :   grpc_server_start(f->server);
     104         128 : }
     105             : 
     106         126 : void chttp2_tear_down_fullstack(grpc_end2end_test_fixture *f) {
     107         126 :   fullstack_fixture_data *ffd = f->fixture_data;
     108         126 :   grpc_end2end_proxy_destroy(ffd->proxy);
     109         126 :   gpr_free(ffd);
     110         126 : }
     111             : 
     112             : /* All test configurations */
     113             : static grpc_end2end_test_config configs[] = {
     114             :     {"chttp2/fullstack+proxy", FEATURE_MASK_SUPPORTS_DELAYED_CONNECTION,
     115             :      chttp2_create_fixture_fullstack, chttp2_init_client_fullstack,
     116             :      chttp2_init_server_fullstack, chttp2_tear_down_fullstack},
     117             : };
     118             : 
     119          57 : int main(int argc, char **argv) {
     120             :   size_t i;
     121             : 
     122          57 :   grpc_test_init(argc, argv);
     123          57 :   grpc_init();
     124             : 
     125         114 :   for (i = 0; i < sizeof(configs) / sizeof(*configs); i++) {
     126          57 :     grpc_end2end_tests(configs[i]);
     127             :   }
     128             : 
     129          57 :   grpc_shutdown();
     130             : 
     131          57 :   return 0;
     132             : }