diff --git a/src/or/connection.c b/src/or/connection.c index ff5c4c04ae28aba01b6f34dfc9f0e41d86260bc7..6665809bc2a42ae842e6b97fb0f7a04d5f168d55 100644 --- a/src/or/connection.c +++ b/src/or/connection.c @@ -2661,14 +2661,17 @@ static void connection_handle_write_cb(struct bufferevent *bufev, void *arg) { connection_t *conn = arg; - (void) bufev; + struct evbuffer *output; if (connection_flushed_some(conn)<0) { connection_mark_for_close(conn); return; } - if (!connection_wants_to_flush(conn)) { + output = bufferevent_get_output(bufev); + if (!evbuffer_get_length(output)) { connection_finished_flushing(conn); + if (conn->marked_for_close && conn->hold_open_until_flushed) + conn->hold_open_until_flushed = 0; } } diff --git a/src/or/main.c b/src/or/main.c index 54c78c32dcafce211304401239ec3f0e0df02f1f..b8867852c7e11c37c45d3bc396af03f6f605c9b5 100644 --- a/src/or/main.c +++ b/src/or/main.c @@ -639,6 +639,11 @@ conn_close_if_marked(int i) assert_connection_ok(conn, now); /* assert_all_pending_dns_resolves_ok(); */ +#ifdef USE_BUFFEREVENTS + if (conn->bufev && conn->hold_open_until_flushed) + return 0; +#endif + log_debug(LD_NET,"Cleaning up connection (fd %d).",conn->s); if ((conn->s >= 0 || conn->linked_conn) && connection_wants_to_flush(conn)) { /* s == -1 means it's an incomplete edge connection, or that the socket