2 Unix SMB/CIFS implementation.
4 Copyright (C) Volker Lendecke 2007
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/>.
22 struct packet_context {
28 * Close the underlying fd
30 static int packet_context_destructor(struct packet_context *ctx)
32 return close(ctx->fd);
36 * Initialize a packet context. The fd is given to the packet context, meaning
37 * that it is automatically closed when the packet context is freed.
39 struct packet_context *packet_init(TALLOC_CTX *mem_ctx, int fd)
41 struct packet_context *result;
43 if (!(result = TALLOC_ZERO_P(mem_ctx, struct packet_context))) {
48 talloc_set_destructor(result, packet_context_destructor);
53 * Pull data from the fd
55 NTSTATUS packet_fd_read(struct packet_context *ctx)
61 res = ioctl(ctx->fd, FIONREAD, &available);
64 DEBUG(10, ("ioctl(FIONREAD) failed: %s\n", strerror(errno)));
65 return map_nt_error_from_unix(errno);
68 SMB_ASSERT(available >= 0);
71 return NT_STATUS_END_OF_FILE;
74 new_size = ctx->in.length + available;
76 if (new_size < ctx->in.length) {
77 DEBUG(0, ("integer wrap\n"));
78 return NT_STATUS_NO_MEMORY;
81 if (!(in = TALLOC_REALLOC_ARRAY(ctx, ctx->in.data, uint8, new_size))) {
82 DEBUG(10, ("talloc failed\n"));
83 return NT_STATUS_NO_MEMORY;
88 res = recv(ctx->fd, in + ctx->in.length, available, 0);
91 DEBUG(10, ("recv failed: %s\n", strerror(errno)));
92 return map_nt_error_from_unix(errno);
96 return NT_STATUS_END_OF_FILE;
99 ctx->in.length += res;
104 NTSTATUS packet_fd_read_sync(struct packet_context *ctx,
105 struct timeval *timeout)
110 if (ctx->fd < 0 || ctx->fd >= FD_SETSIZE) {
112 return map_nt_error_from_unix(errno);
116 FD_SET(ctx->fd, &r_fds);
118 res = sys_select(ctx->fd+1, &r_fds, NULL, NULL, timeout);
121 DEBUG(10, ("select timed out\n"));
122 return NT_STATUS_IO_TIMEOUT;
126 DEBUG(10, ("select returned %s\n", strerror(errno)));
127 return map_nt_error_from_unix(errno);
130 return packet_fd_read(ctx);
133 bool packet_handler(struct packet_context *ctx,
134 bool (*full_req)(const uint8_t *buf,
138 NTSTATUS (*callback)(uint8_t *buf, size_t length,
140 void *priv, NTSTATUS *status)
145 if (!full_req(ctx->in.data, ctx->in.length, &length, priv)) {
149 if (length > ctx->in.length) {
150 *status = NT_STATUS_INTERNAL_ERROR;
154 if (length == ctx->in.length) {
159 buf = (uint8_t *)TALLOC_MEMDUP(ctx, ctx->in.data, length);
161 *status = NT_STATUS_NO_MEMORY;
165 memmove(ctx->in.data, ctx->in.data + length,
166 ctx->in.length - length);
167 ctx->in.length -= length;
170 *status = callback(buf, length, priv);
175 * How many bytes of outgoing data do we have pending?
177 size_t packet_outgoing_bytes(struct packet_context *ctx)
179 return ctx->out.length;
183 * Push data to the fd
185 NTSTATUS packet_fd_write(struct packet_context *ctx)
189 sent = send(ctx->fd, ctx->out.data, ctx->out.length, 0);
192 DEBUG(0, ("send failed: %s\n", strerror(errno)));
193 return map_nt_error_from_unix(errno);
196 memmove(ctx->out.data, ctx->out.data + sent,
197 ctx->out.length - sent);
198 ctx->out.length -= sent;
204 * Sync flush all outgoing bytes
206 NTSTATUS packet_flush(struct packet_context *ctx)
208 while (ctx->out.length != 0) {
209 NTSTATUS status = packet_fd_write(ctx);
210 if (!NT_STATUS_IS_OK(status)) {
218 * Send a list of DATA_BLOBs
220 * Example: packet_send(ctx, 2, data_blob_const(&size, sizeof(size)),
221 * data_blob_const(buf, size));
223 NTSTATUS packet_send(struct packet_context *ctx, int num_blobs, ...)
230 len = ctx->out.length;
232 va_start(ap, num_blobs);
233 for (i=0; i<num_blobs; i++) {
235 DATA_BLOB blob = va_arg(ap, DATA_BLOB);
237 tmp = len + blob.length;
239 DEBUG(0, ("integer overflow\n"));
241 return NT_STATUS_NO_MEMORY;
251 if (!(out = TALLOC_REALLOC_ARRAY(ctx, ctx->out.data, uint8, len))) {
252 DEBUG(0, ("talloc failed\n"));
253 return NT_STATUS_NO_MEMORY;
258 va_start(ap, num_blobs);
259 for (i=0; i<num_blobs; i++) {
260 DATA_BLOB blob = va_arg(ap, DATA_BLOB);
262 memcpy(ctx->out.data+ctx->out.length, blob.data, blob.length);
263 ctx->out.length += blob.length;
267 SMB_ASSERT(ctx->out.length == len);
272 * Get the packet context's file descriptor
274 int packet_get_fd(struct packet_context *ctx)