s3/lib/ctdbd_conn: assert hdr following read/recv
[Samba.git] / libcli / smb / read_smb.c
bloba40f702cffc353444ea2ecdec702d30fba1878c7
1 /*
2 Unix SMB/CIFS implementation.
3 Infrastructure for async SMB client requests
4 Copyright (C) Volker Lendecke 2008
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/>.
20 #include "includes.h"
21 #include "system/network.h"
22 #include "lib/async_req/async_sock.h"
23 #include "read_smb.h"
24 #include "lib/util/tevent_unix.h"
25 #include "libcli/smb/smb_constants.h"
28 * Read an smb packet asynchronously, discard keepalives
31 struct read_smb_state {
32 struct tevent_context *ev;
33 int fd;
34 uint8_t *buf;
37 static ssize_t read_smb_more(uint8_t *buf, size_t buflen, void *private_data);
38 static void read_smb_done(struct tevent_req *subreq);
40 struct tevent_req *read_smb_send(TALLOC_CTX *mem_ctx,
41 struct tevent_context *ev,
42 int fd)
44 struct tevent_req *result, *subreq;
45 struct read_smb_state *state;
47 result = tevent_req_create(mem_ctx, &state, struct read_smb_state);
48 if (result == NULL) {
49 return NULL;
51 state->ev = ev;
52 state->fd = fd;
54 subreq = read_packet_send(state, ev, fd, 4, read_smb_more, NULL);
55 if (subreq == NULL) {
56 goto fail;
58 tevent_req_set_callback(subreq, read_smb_done, result);
59 return result;
60 fail:
61 TALLOC_FREE(result);
62 return NULL;
65 static ssize_t read_smb_more(uint8_t *buf, size_t buflen, void *private_data)
67 if (buflen > 4) {
68 return 0; /* We've been here, we're done */
70 return smb_len_tcp(buf);
73 static void read_smb_done(struct tevent_req *subreq)
75 struct tevent_req *req = tevent_req_callback_data(
76 subreq, struct tevent_req);
77 struct read_smb_state *state = tevent_req_data(
78 req, struct read_smb_state);
79 ssize_t len;
80 int err;
82 len = read_packet_recv(subreq, state, &state->buf, &err);
83 TALLOC_FREE(subreq);
84 if (len == -1) {
85 tevent_req_error(req, err);
86 return;
89 if (CVAL(state->buf, 0) == NBSSkeepalive) {
90 subreq = read_packet_send(state, state->ev, state->fd, 4,
91 read_smb_more, NULL);
92 if (tevent_req_nomem(subreq, req)) {
93 return;
95 tevent_req_set_callback(subreq, read_smb_done, req);
96 return;
98 tevent_req_done(req);
101 ssize_t read_smb_recv(struct tevent_req *req, TALLOC_CTX *mem_ctx,
102 uint8_t **pbuf, int *perrno)
104 struct read_smb_state *state = tevent_req_data(
105 req, struct read_smb_state);
107 if (tevent_req_is_unix_error(req, perrno)) {
108 tevent_req_received(req);
109 return -1;
111 *pbuf = talloc_move(mem_ctx, &state->buf);
112 tevent_req_received(req);
113 return talloc_get_size(*pbuf);