legacy-libs/grpc-cloned/deps/grpc/src/core/lib/iomgr/tcp_client_windows.cc

   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 <grpc/support/port_platform.h>
  20
  21 #include "src/core/lib/iomgr/port.h"
  22
  23 #include <inttypes.h>
  24
  25 #ifdef GRPC_WINSOCK_SOCKET
  26
  27 #include "src/core/lib/iomgr/sockaddr_windows.h"
  28
  29 #include <grpc/slice_buffer.h>
  30 #include <grpc/support/alloc.h>
  31 #include <grpc/support/log.h>
  32 #include <grpc/support/log_windows.h>
  33
  34 #include "src/core/lib/channel/channel_args.h"
  35 #include "src/core/lib/iomgr/iocp_windows.h"
  36 #include "src/core/lib/iomgr/sockaddr.h"
  37 #include "src/core/lib/iomgr/sockaddr_utils.h"
  38 #include "src/core/lib/iomgr/socket_windows.h"
  39 #include "src/core/lib/iomgr/tcp_client.h"
  40 #include "src/core/lib/iomgr/tcp_windows.h"
  41 #include "src/core/lib/iomgr/timer.h"
  42
  43 typedef struct {
  44   grpc_closure* on_done;
  45   gpr_mu mu;
  46   grpc_winsocket* socket;
  47   grpc_timer alarm;
  48   grpc_closure on_alarm;
  49   char* addr_name;
  50   int refs;
  51   grpc_closure on_connect;
  52   grpc_endpoint** endpoint;
  53   grpc_channel_args* channel_args;
  54 } async_connect;
  55
  56 static void async_connect_unlock_and_cleanup(async_connect* ac,
  57                                              grpc_winsocket* socket) {
  58   int done = (--ac->refs == 0);
  59   gpr_mu_unlock(&ac->mu);
  60   if (done) {
  61     grpc_channel_args_destroy(ac->channel_args);
  62     gpr_mu_destroy(&ac->mu);
  63     gpr_free(ac->addr_name);
  64     gpr_free(ac);
  65   }
  66   if (socket != NULL) grpc_winsocket_destroy(socket);
  67 }
  68
  69 static void on_alarm(void* acp, grpc_error* error) {
  70   async_connect* ac = (async_connect*)acp;
  71   gpr_mu_lock(&ac->mu);
  72   grpc_winsocket* socket = ac->socket;
  73   ac->socket = NULL;
  74   if (socket != NULL) {
  75     grpc_winsocket_shutdown(socket);
  76   }
  77   async_connect_unlock_and_cleanup(ac, socket);
  78 }
  79
  80 static void on_connect(void* acp, grpc_error* error) {
  81   async_connect* ac = (async_connect*)acp;
  82   grpc_endpoint** ep = ac->endpoint;
  83   GPR_ASSERT(*ep == NULL);
  84   grpc_closure* on_done = ac->on_done;
  85
  86   GRPC_ERROR_REF(error);
  87
  88   gpr_mu_lock(&ac->mu);
  89   grpc_winsocket* socket = ac->socket;
  90   ac->socket = NULL;
  91   gpr_mu_unlock(&ac->mu);
  92
  93   grpc_timer_cancel(&ac->alarm);
  94
  95   gpr_mu_lock(&ac->mu);
  96
  97   if (error == GRPC_ERROR_NONE) {
  98     if (socket != NULL) {
  99       DWORD transfered_bytes = 0;
 100       DWORD flags;
 101       BOOL wsa_success =
 102           WSAGetOverlappedResult(socket->socket, &socket->write_info.overlapped,
 103                                  &transfered_bytes, FALSE, &flags);
 104       GPR_ASSERT(transfered_bytes == 0);
 105       if (!wsa_success) {
 106         error = GRPC_WSA_ERROR(WSAGetLastError(), "ConnectEx");
 107         closesocket(socket->socket);
 108       } else {
 109         *ep = grpc_tcp_create(socket, ac->channel_args, ac->addr_name);
 110         socket = NULL;
 111       }
 112     } else {
 113       error = GRPC_ERROR_CREATE_FROM_STATIC_STRING("socket is null");
 114     }
 115   }
 116
 117   async_connect_unlock_and_cleanup(ac, socket);
 118   /* If the connection was aborted, the callback was already called when
 119      the deadline was met. */
 120   GRPC_CLOSURE_SCHED(on_done, error);
 121 }
 122
 123 /* Tries to issue one async connection, then schedules both an IOCP
 124    notification request for the connection, and one timeout alert. */
 125 static void tcp_connect(grpc_closure* on_done, grpc_endpoint** endpoint,
 126                         grpc_pollset_set* interested_parties,
 127                         const grpc_channel_args* channel_args,
 128                         const grpc_resolved_address* addr,
 129                         grpc_millis deadline) {
 130   SOCKET sock = INVALID_SOCKET;
 131   BOOL success;
 132   int status;
 133   grpc_resolved_address addr6_v4mapped;
 134   grpc_resolved_address local_address;
 135   async_connect* ac;
 136   grpc_winsocket* socket = NULL;
 137   LPFN_CONNECTEX ConnectEx;
 138   GUID guid = WSAID_CONNECTEX;
 139   DWORD ioctl_num_bytes;
 140   grpc_winsocket_callback_info* info;
 141   grpc_error* error = GRPC_ERROR_NONE;
 142
 143   *endpoint = NULL;
 144
 145   /* Use dualstack sockets where available. */
 146   if (grpc_sockaddr_to_v4mapped(addr, &addr6_v4mapped)) {
 147     addr = &addr6_v4mapped;
 148   }
 149
 150   sock = WSASocket(AF_INET6, SOCK_STREAM, IPPROTO_TCP, NULL, 0,
 151                    grpc_get_default_wsa_socket_flags());
 152   if (sock == INVALID_SOCKET) {
 153     error = GRPC_WSA_ERROR(WSAGetLastError(), "WSASocket");
 154     goto failure;
 155   }
 156
 157   error = grpc_tcp_prepare_socket(sock);
 158   if (error != GRPC_ERROR_NONE) {
 159     goto failure;
 160   }
 161
 162   /* Grab the function pointer for ConnectEx for that specific socket.
 163      It may change depending on the interface. */
 164   status =
 165       WSAIoctl(sock, SIO_GET_EXTENSION_FUNCTION_POINTER, &guid, sizeof(guid),
 166                &ConnectEx, sizeof(ConnectEx), &ioctl_num_bytes, NULL, NULL);
 167
 168   if (status != 0) {
 169     error = GRPC_WSA_ERROR(WSAGetLastError(),
 170                            "WSAIoctl(SIO_GET_EXTENSION_FUNCTION_POINTER)");
 171     goto failure;
 172   }
 173
 174   grpc_sockaddr_make_wildcard6(0, &local_address);
 175
 176   status =
 177       bind(sock, (grpc_sockaddr*)&local_address.addr, (int)local_address.len);
 178   if (status != 0) {
 179     error = GRPC_WSA_ERROR(WSAGetLastError(), "bind");
 180     goto failure;
 181   }
 182
 183   socket = grpc_winsocket_create(sock, "client");
 184   info = &socket->write_info;
 185   success = ConnectEx(sock, (grpc_sockaddr*)&addr->addr, (int)addr->len, NULL,
 186                       0, NULL, &info->overlapped);
 187
 188   /* It wouldn't be unusual to get a success immediately. But we'll still get
 189      an IOCP notification, so let's ignore it. */
 190   if (!success) {
 191     int last_error = WSAGetLastError();
 192     if (last_error != ERROR_IO_PENDING) {
 193       error = GRPC_WSA_ERROR(last_error, "ConnectEx");
 194       goto failure;
 195     }
 196   }
 197
 198   ac = (async_connect*)gpr_malloc(sizeof(async_connect));
 199   ac->on_done = on_done;
 200   ac->socket = socket;
 201   gpr_mu_init(&ac->mu);
 202   ac->refs = 2;
 203   ac->addr_name = grpc_sockaddr_to_uri(addr);
 204   ac->endpoint = endpoint;
 205   ac->channel_args = grpc_channel_args_copy(channel_args);
 206   GRPC_CLOSURE_INIT(&ac->on_connect, on_connect, ac, grpc_schedule_on_exec_ctx);
 207
 208   GRPC_CLOSURE_INIT(&ac->on_alarm, on_alarm, ac, grpc_schedule_on_exec_ctx);
 209   grpc_timer_init(&ac->alarm, deadline, &ac->on_alarm);
 210   grpc_socket_notify_on_write(socket, &ac->on_connect);
 211   return;
 212
 213 failure:
 214   GPR_ASSERT(error != GRPC_ERROR_NONE);
 215   char* target_uri = grpc_sockaddr_to_uri(addr);
 216   grpc_error* final_error =
 217       grpc_error_set_str(GRPC_ERROR_CREATE_REFERENCING_FROM_STATIC_STRING(
 218                              "Failed to connect", &error, 1),
 219                          GRPC_ERROR_STR_TARGET_ADDRESS,
 220                          grpc_slice_from_copied_string(
 221                              target_uri == nullptr ? "NULL" : target_uri));
 222   GRPC_ERROR_UNREF(error);
 223   if (socket != NULL) {
 224     grpc_winsocket_destroy(socket);
 225   } else if (sock != INVALID_SOCKET) {
 226     closesocket(sock);
 227   }
 228   GRPC_CLOSURE_SCHED(on_done, final_error);
 229 }
 230
 231 grpc_tcp_client_vtable grpc_windows_tcp_client_vtable = {tcp_connect};
 232
 233 #endif /* GRPC_WINSOCK_SOCKET */