JSON RPC: fixed shutdown on connect error

This commit is contained in:
SChernykh 2022-07-05 08:36:16 +02:00
parent 4f24f18b27
commit 64baf660bb

View file

@ -58,6 +58,8 @@ struct CurlContext
static void on_close(uv_handle_t* h); static void on_close(uv_handle_t* h);
void close_handles();
uv_poll_t m_pollHandle; uv_poll_t m_pollHandle;
curl_socket_t m_socket; curl_socket_t m_socket;
@ -235,10 +237,7 @@ int CurlContext::on_socket(CURL* /*easy*/, curl_socket_t s, int action)
case CURL_POLL_REMOVE: case CURL_POLL_REMOVE:
default: default:
curl_multi_assign(m_multiHandle, s, nullptr); curl_multi_assign(m_multiHandle, s, nullptr);
uv_poll_stop(&m_pollHandle); close_handles();
uv_close(reinterpret_cast<uv_handle_t*>(&m_async), on_close);
uv_close(reinterpret_cast<uv_handle_t*>(&m_timer), on_close);
uv_close(reinterpret_cast<uv_handle_t*>(&m_pollHandle), on_close);
break; break;
} }
@ -266,9 +265,13 @@ void CurlContext::on_timeout(uv_handle_t* req)
{ {
CurlContext* ctx = reinterpret_cast<CurlContext*>(req->data); CurlContext* ctx = reinterpret_cast<CurlContext*>(req->data);
int running_handles; int running_handles = 0;
curl_multi_socket_action(ctx->m_multiHandle, CURL_SOCKET_TIMEOUT, 0, &running_handles); curl_multi_socket_action(ctx->m_multiHandle, CURL_SOCKET_TIMEOUT, 0, &running_handles);
ctx->check_multi_info(); ctx->check_multi_info();
if (running_handles == 0) {
ctx->close_handles();
}
} }
size_t CurlContext::on_write(const void* buffer, size_t size, size_t count) size_t CurlContext::on_write(const void* buffer, size_t size, size_t count)
@ -340,6 +343,22 @@ void CurlContext::on_close(uv_handle_t* h)
delete ctx; delete ctx;
} }
void CurlContext::close_handles()
{
if (m_pollHandle.data && !uv_is_closing(reinterpret_cast<uv_handle_t*>(&m_pollHandle))) {
uv_poll_stop(&m_pollHandle);
uv_close(reinterpret_cast<uv_handle_t*>(&m_pollHandle), on_close);
}
if (m_async.data && !uv_is_closing(reinterpret_cast<uv_handle_t*>(&m_async))) {
uv_close(reinterpret_cast<uv_handle_t*>(&m_async), on_close);
}
if (m_timer.data && !uv_is_closing(reinterpret_cast<uv_handle_t*>(&m_timer))) {
uv_close(reinterpret_cast<uv_handle_t*>(&m_timer), on_close);
}
}
void Call(const std::string& address, int port, const std::string& req, const std::string& auth, CallbackBase* cb, CallbackBase* close_cb, uv_loop_t* loop) void Call(const std::string& address, int port, const std::string& req, const std::string& auth, CallbackBase* cb, CallbackBase* close_cb, uv_loop_t* loop)
{ {
if (!loop) { if (!loop) {