2 Unix SMB/CIFS implementation.
3 Utilities around tsocket
4 Copyright (C) Volker Lendecke 2009
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 3 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program. If not, see <http://www.gnu.org/licenses/>.
21 #include "../lib/tsocket/tsocket.h"
22 #include "../lib/util/tevent_unix.h"
24 struct tstream_read_packet_state {
25 struct tevent_context *ev;
26 struct tstream_context *stream;
27 ssize_t (*more)(uint8_t *buf, size_t buflen, void *private_data);
33 static void tstream_read_packet_done(struct tevent_req *subreq);
35 struct tevent_req *tstream_read_packet_send(TALLOC_CTX *mem_ctx,
36 struct tevent_context *ev,
37 struct tstream_context *stream,
39 ssize_t (*more)(uint8_t *buf,
44 struct tevent_req *req, *subreq;
45 struct tstream_read_packet_state *state;
47 req = tevent_req_create(mem_ctx, &state,
48 struct tstream_read_packet_state);
52 state->buf = talloc_array(state, uint8_t, initial);
53 if (tevent_req_nomem(state->buf, req)) {
54 return tevent_req_post(req, ev);
56 state->iov.iov_base = (void *)state->buf;
57 state->iov.iov_len = initial;
60 state->stream = stream;
62 state->private_data = private_data;
64 subreq = tstream_readv_send(state, ev, stream, &state->iov, 1);
65 if (tevent_req_nomem(subreq, req)) {
66 return tevent_req_post(req, ev);
68 tevent_req_set_callback(subreq, tstream_read_packet_done, req);
73 static void tstream_read_packet_done(struct tevent_req *subreq)
75 struct tevent_req *req = tevent_req_callback_data(
76 subreq, struct tevent_req);
77 struct tstream_read_packet_state *state = tevent_req_data(
78 req, struct tstream_read_packet_state);
84 ret = tstream_readv_recv(subreq, &err);
90 tevent_req_error(req, err);
94 if (state->more == NULL) {
95 /* Nobody to ask, this is a async read_data */
99 total = talloc_array_length(state->buf);
101 more = state->more(state->buf, total, state->private_data);
103 /* We got an invalid packet, tell the caller */
104 tevent_req_error(req, EIO);
108 /* We're done, full packet received */
109 tevent_req_done(req);
113 tmp = talloc_realloc(state, state->buf, uint8_t, total+more);
114 if (tevent_req_nomem(tmp, req)) {
119 state->iov.iov_base = (void *)(state->buf + total);
120 state->iov.iov_len = more;
122 subreq = tstream_readv_send(state, state->ev, state->stream,
124 if (tevent_req_nomem(subreq, req)) {
127 tevent_req_set_callback(subreq, tstream_read_packet_done, req);
130 ssize_t tstream_read_packet_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
131 uint8_t **pbuf, int *perrno)
133 struct tstream_read_packet_state *state =
134 tevent_req_data(req, struct tstream_read_packet_state);
136 if (tevent_req_is_unix_error(req, perrno)) {
139 *pbuf = talloc_move(mem_ctx, &state->buf);
140 return talloc_array_length(*pbuf);