Yuchen Zeng
9 years ago
parent
f0cb609e21
commit
85750b08f2
24 changed files with 415 additions and 221 deletions
@ -0,0 +1,43 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2016, 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 <ares.h> |
||||
|
||||
#include "src/core/lib/iomgr/exec_ctx.h" |
||||
#include "src/core/lib/iomgr/pollset_set.h" |
||||
|
||||
typedef struct grpc_ares_ev_driver grpc_ares_ev_driver; |
||||
|
||||
void grpc_ares_notify_on_event(grpc_exec_ctx *exec_ctx, grpc_ares_ev_driver *ev_driver); |
||||
|
||||
grpc_ares_ev_driver *grpc_ares_ev_driver_create(ares_channel *channel, grpc_pollset_set *pollset_set); |
||||
@ -0,0 +1,176 @@ |
||||
/*
|
||||
* |
||||
* Copyright 2016, 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 <grpc/support/port_platform.h> |
||||
#ifdef GPR_POSIX_SOCKET |
||||
|
||||
#include "src/core/ext/resolver/dns/c_ares/grpc_ares_ev_driver.h" |
||||
|
||||
|
||||
#include "src/core/lib/iomgr/ev_posix.h" |
||||
#include "src/core/lib/iomgr/sockaddr.h" |
||||
|
||||
#include <ares.h> |
||||
#include <grpc/support/alloc.h> |
||||
#include <grpc/support/log.h> |
||||
#include <grpc/support/string_util.h> |
||||
#include <grpc/support/time.h> |
||||
#include <grpc/support/useful.h> |
||||
#include "src/core/lib/iomgr/iomgr_internal.h" |
||||
#include "src/core/lib/iomgr/sockaddr_utils.h" |
||||
#include "src/core/lib/iomgr/unix_sockets_posix.h" |
||||
#include "src/core/lib/support/block_annotate.h" |
||||
#include "src/core/lib/support/string.h" |
||||
|
||||
typedef struct fd_pair { |
||||
grpc_fd *grpc_fd; |
||||
int fd; |
||||
struct fd_pair *next; |
||||
} fd_pair; |
||||
|
||||
struct grpc_ares_ev_driver { |
||||
int id; |
||||
ares_socket_t socks[ARES_GETSOCK_MAXNUM]; |
||||
int bitmask; |
||||
grpc_closure driver_closure; |
||||
grpc_pollset_set *pollset_set; |
||||
ares_channel *channel; |
||||
fd_pair *fds; |
||||
}; |
||||
|
||||
static fd_pair *get_fd(fd_pair **head, int fd) { |
||||
fd_pair dummy_head; |
||||
fd_pair *node; |
||||
fd_pair *ret; |
||||
dummy_head.next = *head; |
||||
node = &dummy_head; |
||||
while (node->next != NULL) { |
||||
if (node->next->fd == fd) { |
||||
ret = node->next; |
||||
node->next = node->next->next; |
||||
*head = dummy_head.next; |
||||
return ret; |
||||
} |
||||
} |
||||
return NULL; |
||||
} |
||||
|
||||
static void driver_cb(grpc_exec_ctx *exec_ctx, void *arg, grpc_error *error) { |
||||
grpc_ares_ev_driver *d = arg; |
||||
size_t i; |
||||
gpr_log(GPR_ERROR, "driver_cb"); |
||||
if (error == GRPC_ERROR_NONE) { |
||||
gpr_log(GPR_ERROR, "GRPC_ERROR_NONE"); |
||||
for (i = 0; i < ARES_GETSOCK_MAXNUM; i++) { |
||||
ares_process_fd( |
||||
*d->channel, |
||||
ARES_GETSOCK_READABLE(d->bitmask, i) ? d->socks[i] : ARES_SOCKET_BAD, |
||||
ARES_GETSOCK_WRITABLE(d->bitmask, i) ? d->socks[i] : ARES_SOCKET_BAD); |
||||
} |
||||
} |
||||
grpc_ares_notify_on_event(exec_ctx, d); |
||||
grpc_exec_ctx_flush(exec_ctx); |
||||
} |
||||
|
||||
void grpc_ares_notify_on_event(grpc_exec_ctx *exec_ctx, grpc_ares_ev_driver *ev_driver) { |
||||
size_t i; |
||||
fd_pair *new_list = NULL; |
||||
ev_driver->bitmask = |
||||
ares_getsock(*ev_driver->channel, ev_driver->socks, ARES_GETSOCK_MAXNUM); |
||||
grpc_closure_init(&ev_driver->driver_closure, driver_cb, ev_driver); |
||||
for (i = 0; i < ARES_GETSOCK_MAXNUM; i++) { |
||||
char *final_name; |
||||
gpr_asprintf(&final_name, "host1%" PRIuPTR, i); |
||||
|
||||
if (ARES_GETSOCK_READABLE(ev_driver->bitmask, i) || |
||||
ARES_GETSOCK_WRITABLE(ev_driver->bitmask, i)) { |
||||
gpr_log(GPR_ERROR, "%d", ev_driver->socks[i]); |
||||
fd_pair *fdp = get_fd(&ev_driver->fds, ev_driver->socks[i]); |
||||
if (!fdp) { |
||||
gpr_log(GPR_ERROR, "new fd"); |
||||
fdp = gpr_malloc(sizeof(fd_pair)); |
||||
fdp->grpc_fd = grpc_fd_create(ev_driver->socks[i], final_name); |
||||
fdp->fd = ev_driver->socks[i]; |
||||
grpc_pollset_set_add_fd(exec_ctx, ev_driver->pollset_set, fdp->grpc_fd); |
||||
// new_fd_pair->grpc_fd = fd;
|
||||
// new_fd_pair->next = ev_driver->fds;
|
||||
} |
||||
fdp->next = new_list; |
||||
new_list = fdp; |
||||
|
||||
if (ARES_GETSOCK_READABLE(ev_driver->bitmask, i)) { |
||||
gpr_log(GPR_ERROR, "READABLE"); |
||||
|
||||
grpc_fd_notify_on_read(exec_ctx, fdp->grpc_fd, |
||||
&ev_driver->driver_closure); |
||||
} |
||||
if (ARES_GETSOCK_WRITABLE(ev_driver->bitmask, i)) { |
||||
gpr_log(GPR_ERROR, "writable"); |
||||
|
||||
grpc_fd_notify_on_write(exec_ctx, fdp->grpc_fd, |
||||
&ev_driver->driver_closure); |
||||
} |
||||
} |
||||
gpr_free(final_name); |
||||
} |
||||
|
||||
while (ev_driver->fds != NULL) { |
||||
fd_pair *cur; |
||||
// int fd;s
|
||||
cur = ev_driver->fds; |
||||
ev_driver->fds = ev_driver->fds->next; |
||||
gpr_log(GPR_ERROR, "fd in ev_driver: %d\n", cur->fd); |
||||
grpc_pollset_set_del_fd(exec_ctx, ev_driver->pollset_set, cur->grpc_fd); |
||||
gpr_log(GPR_ERROR, "grpc_pollset_set_del_fd"); |
||||
grpc_fd_shutdown(exec_ctx, cur->grpc_fd); |
||||
gpr_log(GPR_ERROR, "grpc_fd_shutdown"); |
||||
grpc_fd_orphan(exec_ctx, cur->grpc_fd, NULL, NULL, "come on.."); |
||||
gpr_log(GPR_ERROR, "grpc_fd_orphan"); |
||||
gpr_free(cur); |
||||
} |
||||
|
||||
ev_driver->fds = new_list; |
||||
|
||||
gpr_log(GPR_ERROR, "eof notify_on_event"); |
||||
} |
||||
|
||||
|
||||
|
||||
grpc_ares_ev_driver *grpc_ares_ev_driver_create(ares_channel *channel, grpc_pollset_set *pollset_set) { |
||||
grpc_ares_ev_driver *ev_driver = gpr_malloc(sizeof(grpc_ares_ev_driver)); |
||||
ev_driver->channel = channel; |
||||
ev_driver->pollset_set = pollset_set; |
||||
ev_driver->fds = NULL; |
||||
return ev_driver; |
||||
} |
||||
|
||||
#endif |
||||
Loading…
Reference in new issue