lib/async_req: remove the tevent_fd as early as possible via a read_packet_cleanup...

BUG: https://bugzilla.samba.org/show_bug.cgi?id=11316

Signed-off-by: Stefan Metzmacher <metze@samba.org>
Reviewed-by: Volker Lendecke <vl@samba.org>

diff --git a/lib/async_req/async_sock.c b/lib/async_req/async_sock.c
index 19278f652e0220dc0e4648983889dbdd51dd7dc3..adb8f87d0ccf2e78b777b21560e1f84538e014f6 100644 (file)
--- a/lib/async_req/async_sock.c
+++ b/lib/async_req/async_sock.c
@@ -378,12 +378,15 @@ ssize_t writev_recv(struct tevent_req *req, int *perrno)
 
 struct read_packet_state {
        int fd;
+       struct tevent_fd *fde;
        uint8_t *buf;
        size_t nread;
        ssize_t (*more)(uint8_t *buf, size_t buflen, void *private_data);
        void *private_data;
 };
 
+static void read_packet_cleanup(struct tevent_req *req,
+                                enum tevent_req_state req_state);
 static void read_packet_handler(struct tevent_context *ev,
                                struct tevent_fd *fde,
                                uint16_t flags, void *private_data);
@@ -398,7 +401,6 @@ struct tevent_req *read_packet_send(TALLOC_CTX *mem_ctx,
 {
        struct tevent_req *req;
        struct read_packet_state *state;
-       struct tevent_fd *fde;
 
        req = tevent_req_create(mem_ctx, &state, struct read_packet_state);
        if (req == NULL) {
@@ -409,19 +411,31 @@ struct tevent_req *read_packet_send(TALLOC_CTX *mem_ctx,
        state->more = more;
        state->private_data = private_data;
 
+       tevent_req_set_cleanup_fn(req, read_packet_cleanup);
+
        state->buf = talloc_array(state, uint8_t, initial);
        if (tevent_req_nomem(state->buf, req)) {
                return tevent_req_post(req, ev);
        }
 
-       fde = tevent_add_fd(ev, state, fd, TEVENT_FD_READ, read_packet_handler,
-                           req);
-       if (tevent_req_nomem(fde, req)) {
+       state->fde = tevent_add_fd(ev, state, fd,
+                                  TEVENT_FD_READ, read_packet_handler,
+                                  req);
+       if (tevent_req_nomem(state->fde, req)) {
                return tevent_req_post(req, ev);
        }
        return req;
 }
 
+static void read_packet_cleanup(struct tevent_req *req,
+                          enum tevent_req_state req_state)
+{
+       struct read_packet_state *state =
+               tevent_req_data(req, struct read_packet_state);
+
+       TALLOC_FREE(state->fde);
+}
+
 static void read_packet_handler(struct tevent_context *ev,
                                struct tevent_fd *fde,
                                uint16_t flags, void *private_data)
@@ -499,9 +513,11 @@ ssize_t read_packet_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
                tevent_req_data(req, struct read_packet_state);
 
        if (tevent_req_is_unix_error(req, perrno)) {
+               tevent_req_received(req);
                return -1;
        }
        *pbuf = talloc_move(mem_ctx, &state->buf);
+       tevent_req_received(req);
        return talloc_get_size(*pbuf);
 }