legacy-libs/grpc-cloned/deps/grpc/src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver_posix.cc

   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 #include <grpc/support/port_platform.h>
  19
  20 #include "src/core/lib/iomgr/port.h"
  21 #if GRPC_ARES == 1 && defined(GRPC_POSIX_SOCKET_ARES_EV_DRIVER)
  22
  23 #include <ares.h>
  24 #include <string.h>
  25 #include <sys/ioctl.h>
  26
  27 #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_ev_driver.h"
  28
  29 #include <grpc/support/alloc.h>
  30 #include <grpc/support/log.h>
  31 #include <grpc/support/string_util.h>
  32 #include <grpc/support/time.h>
  33 #include "src/core/ext/filters/client_channel/resolver/dns/c_ares/grpc_ares_wrapper.h"
  34 #include "src/core/lib/gpr/string.h"
  35 #include "src/core/lib/iomgr/ev_posix.h"
  36 #include "src/core/lib/iomgr/iomgr_internal.h"
  37 #include "src/core/lib/iomgr/sockaddr_utils.h"
  38
  39 namespace grpc_core {
  40
  41 class GrpcPolledFdPosix : public GrpcPolledFd {
  42  public:
  43   GrpcPolledFdPosix(ares_socket_t as, grpc_pollset_set* driver_pollset_set)
  44       : as_(as) {
  45     gpr_asprintf(&name_, "c-ares fd: %d", (int)as);
  46     fd_ = grpc_fd_create((int)as, name_, false);
  47     driver_pollset_set_ = driver_pollset_set;
  48     grpc_pollset_set_add_fd(driver_pollset_set_, fd_);
  49   }
  50
  51   ~GrpcPolledFdPosix() {
  52     gpr_free(name_);
  53     grpc_pollset_set_del_fd(driver_pollset_set_, fd_);
  54     /* c-ares library will close the fd inside grpc_fd. This fd may be picked up
  55        immediately by another thread, and should not be closed by the following
  56        grpc_fd_orphan. */
  57     int dummy_release_fd;
  58     grpc_fd_orphan(fd_, nullptr, &dummy_release_fd, "c-ares query finished");
  59   }
  60
  61   void RegisterForOnReadableLocked(grpc_closure* read_closure) override {
  62     grpc_fd_notify_on_read(fd_, read_closure);
  63   }
  64
  65   void RegisterForOnWriteableLocked(grpc_closure* write_closure) override {
  66     grpc_fd_notify_on_write(fd_, write_closure);
  67   }
  68
  69   bool IsFdStillReadableLocked() override {
  70     size_t bytes_available = 0;
  71     return ioctl(grpc_fd_wrapped_fd(fd_), FIONREAD, &bytes_available) == 0 &&
  72            bytes_available > 0;
  73   }
  74
  75   void ShutdownLocked(grpc_error* error) override {
  76     grpc_fd_shutdown(fd_, error);
  77   }
  78
  79   ares_socket_t GetWrappedAresSocketLocked() override { return as_; }
  80
  81   const char* GetName() override { return name_; }
  82
  83   char* name_;
  84   ares_socket_t as_;
  85   grpc_fd* fd_;
  86   grpc_pollset_set* driver_pollset_set_;
  87 };
  88
  89 class GrpcPolledFdFactoryPosix : public GrpcPolledFdFactory {
  90  public:
  91   GrpcPolledFd* NewGrpcPolledFdLocked(ares_socket_t as,
  92                                       grpc_pollset_set* driver_pollset_set,
  93                                       grpc_combiner* combiner) override {
  94     return New<GrpcPolledFdPosix>(as, driver_pollset_set);
  95   }
  96
  97   void ConfigureAresChannelLocked(ares_channel channel) override {}
  98 };
  99
 100 UniquePtr<GrpcPolledFdFactory> NewGrpcPolledFdFactory(grpc_combiner* combiner) {
 101   return UniquePtr<GrpcPolledFdFactory>(New<GrpcPolledFdFactoryPosix>());
 102 }
 103
 104 }  // namespace grpc_core
 105
 106 #endif /* GRPC_ARES == 1 && defined(GRPC_POSIX_SOCKET_ARES_EV_DRIVER) */