From e618abf8f3db2a94269a79c8901a51148d48fcc2 Mon Sep 17 00:00:00 2001 From: Garrett D'Amore Date: Fri, 9 Aug 2019 19:29:51 -0700 Subject: fixes #970 TCP connections through docker-proxy do not reconnect --- src/platform/posix/posix_ipcconn.c | 20 ++++++++++++++++++-- src/platform/posix/posix_tcpconn.c | 20 ++++++++++++++++++-- 2 files changed, 36 insertions(+), 4 deletions(-) (limited to 'src/platform/posix') diff --git a/src/platform/posix/posix_ipcconn.c b/src/platform/posix/posix_ipcconn.c index 07ec6213..c9faded5 100644 --- a/src/platform/posix/posix_ipcconn.c +++ b/src/platform/posix/posix_ipcconn.c @@ -186,7 +186,7 @@ ipc_doread(ipc_conn *c) // No bytes indicates a closed descriptor. // This implicitly completes this (all!) aio. nni_aio_list_remove(aio); - nni_aio_finish_error(aio, NNG_ECLOSED); + nni_aio_finish_error(aio, NNG_ECONNSHUT); continue; } @@ -201,6 +201,22 @@ ipc_doread(ipc_conn *c) } } +static void +ipc_error(void *arg, int err) +{ + ipc_conn *c = arg; + nni_aio *aio; + + nni_mtx_lock(&c->mtx); + while (((aio = nni_list_first(&c->readq)) != NULL) || + ((aio = nni_list_first(&c->writeq)) != NULL)) { + nni_aio_list_remove(aio); + nni_aio_finish_error(aio, err); + } + nni_posix_pfd_close(c->pfd); + nni_mtx_unlock(&c->mtx); +} + static void ipc_close(void *arg) { @@ -225,7 +241,7 @@ ipc_cb(nni_posix_pfd *pfd, int events, void *arg) ipc_conn *c = arg; if (events & (POLLHUP | POLLERR | POLLNVAL)) { - ipc_close(c); + ipc_error(c, NNG_ECONNSHUT); return; } nni_mtx_lock(&c->mtx); diff --git a/src/platform/posix/posix_tcpconn.c b/src/platform/posix/posix_tcpconn.c index 0d3c274d..6ca7013b 100644 --- a/src/platform/posix/posix_tcpconn.c +++ b/src/platform/posix/posix_tcpconn.c @@ -180,7 +180,7 @@ tcp_doread(nni_tcp_conn *c) // No bytes indicates a closed descriptor. // This implicitly completes this (all!) aio. nni_aio_list_remove(aio); - nni_aio_finish_error(aio, NNG_ECLOSED); + nni_aio_finish_error(aio, NNG_ECONNSHUT); continue; } @@ -195,6 +195,22 @@ tcp_doread(nni_tcp_conn *c) } } +static void +tcp_error(void *arg, int err) +{ + nni_tcp_conn *c = arg; + nni_aio *aio; + + nni_mtx_lock(&c->mtx); + while (((aio = nni_list_first(&c->readq)) != NULL) || + ((aio = nni_list_first(&c->writeq)) != NULL)) { + nni_aio_list_remove(aio); + nni_aio_finish_error(aio, err); + } + nni_posix_pfd_close(c->pfd); + nni_mtx_unlock(&c->mtx); +} + static void tcp_close(void *arg) { @@ -242,7 +258,7 @@ tcp_cb(nni_posix_pfd *pfd, int events, void *arg) nni_tcp_conn *c = arg; if (events & (POLLHUP | POLLERR | POLLNVAL)) { - tcp_close(c); + tcp_error(c, NNG_ECONNSHUT); return; } nni_mtx_lock(&c->mtx); -- cgit v1.2.3-70-g09d2