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 #ifndef GRPC_TEST_CORE_UTIL_TEST_TCP_SERVER_H 20 #define GRPC_TEST_CORE_UTIL_TEST_TCP_SERVER_H 21 22 #include <vector> 23 24 #include <grpc/support/sync.h> 25 #include "src/core/lib/iomgr/tcp_server.h" 26 27 // test_tcp_server should be stack-allocated or new'ed, never gpr_malloc'ed 28 // since it contains C++ objects. 29 struct test_tcp_server { 30 grpc_tcp_server* tcp_server = nullptr; 31 grpc_closure shutdown_complete; 32 bool shutdown = false; 33 // mu is filled in by grpc_pollset_init and controlls the pollset. 34 // TODO: Switch this to a Mutex once pollset_init can provide a Mutex 35 gpr_mu* mu; 36 std::vector<grpc_pollset*> pollset; 37 grpc_tcp_server_cb on_connect; 38 void* cb_data; 39 }; 40 41 void test_tcp_server_init(test_tcp_server* server, 42 grpc_tcp_server_cb on_connect, void* user_data); 43 void test_tcp_server_start(test_tcp_server* server, int port); 44 void test_tcp_server_poll(test_tcp_server* server, int milliseconds); 45 void test_tcp_server_destroy(test_tcp_server* server); 46 47 #endif /* GRPC_TEST_CORE_UTIL_TEST_TCP_SERVER_H */ 48