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/>.
21 #include "../lib/util/select.h"
22 #include "system/select.h"
24 struct packet_context
{
30 * Close the underlying fd
32 static int packet_context_destructor(struct packet_context
*ctx
)
34 return close(ctx
->fd
);
38 * Initialize a packet context. The fd is given to the packet context, meaning
39 * that it is automatically closed when the packet context is freed.
41 struct packet_context
*packet_init(TALLOC_CTX
*mem_ctx
, int fd
)
43 struct packet_context
*result
;
45 if (!(result
= TALLOC_ZERO_P(mem_ctx
, struct packet_context
))) {
50 talloc_set_destructor(result
, packet_context_destructor
);
55 * Pull data from the fd
57 NTSTATUS
packet_fd_read(struct packet_context
*ctx
)
63 res
= ioctl(ctx
->fd
, FIONREAD
, &available
);
66 DEBUG(10, ("ioctl(FIONREAD) failed: %s\n", strerror(errno
)));
67 return map_nt_error_from_unix(errno
);
70 SMB_ASSERT(available
>= 0);
73 return NT_STATUS_END_OF_FILE
;
76 new_size
= ctx
->in
.length
+ available
;
78 if (new_size
< ctx
->in
.length
) {
79 DEBUG(0, ("integer wrap\n"));
80 return NT_STATUS_NO_MEMORY
;
83 if (!(in
= TALLOC_REALLOC_ARRAY(ctx
, ctx
->in
.data
, uint8
, new_size
))) {
84 DEBUG(10, ("talloc failed\n"));
85 return NT_STATUS_NO_MEMORY
;
90 res
= recv(ctx
->fd
, in
+ ctx
->in
.length
, available
, 0);
93 DEBUG(10, ("recv failed: %s\n", strerror(errno
)));
94 return map_nt_error_from_unix(errno
);
98 return NT_STATUS_END_OF_FILE
;
101 ctx
->in
.length
+= res
;
106 NTSTATUS
packet_fd_read_sync(struct packet_context
*ctx
, int timeout
)
110 res
= poll_one_fd(ctx
->fd
, POLLIN
|POLLHUP
, timeout
, &revents
);
112 DEBUG(10, ("poll timed out\n"));
113 return NT_STATUS_IO_TIMEOUT
;
117 DEBUG(10, ("poll returned %s\n", strerror(errno
)));
118 return map_nt_error_from_unix(errno
);
120 if ((revents
& (POLLIN
|POLLHUP
|POLLERR
)) == 0) {
121 DEBUG(10, ("socket not readable\n"));
122 return NT_STATUS_IO_TIMEOUT
;
125 return packet_fd_read(ctx
);
128 bool packet_handler(struct packet_context
*ctx
,
129 bool (*full_req
)(const uint8_t *buf
,
133 NTSTATUS (*callback
)(uint8_t *buf
, size_t length
,
135 void *priv
, NTSTATUS
*status
)
140 if (!full_req(ctx
->in
.data
, ctx
->in
.length
, &length
, priv
)) {
144 if (length
> ctx
->in
.length
) {
145 *status
= NT_STATUS_INTERNAL_ERROR
;
149 if (length
== ctx
->in
.length
) {
154 buf
= (uint8_t *)TALLOC_MEMDUP(ctx
, ctx
->in
.data
, length
);
156 *status
= NT_STATUS_NO_MEMORY
;
160 memmove(ctx
->in
.data
, ctx
->in
.data
+ length
,
161 ctx
->in
.length
- length
);
162 ctx
->in
.length
-= length
;
165 *status
= callback(buf
, length
, priv
);
170 * How many bytes of outgoing data do we have pending?
172 size_t packet_outgoing_bytes(struct packet_context
*ctx
)
174 return ctx
->out
.length
;
178 * Push data to the fd
180 NTSTATUS
packet_fd_write(struct packet_context
*ctx
)
184 sent
= send(ctx
->fd
, ctx
->out
.data
, ctx
->out
.length
, 0);
187 DEBUG(0, ("send failed: %s\n", strerror(errno
)));
188 return map_nt_error_from_unix(errno
);
191 memmove(ctx
->out
.data
, ctx
->out
.data
+ sent
,
192 ctx
->out
.length
- sent
);
193 ctx
->out
.length
-= sent
;
199 * Sync flush all outgoing bytes
201 NTSTATUS
packet_flush(struct packet_context
*ctx
)
203 while (ctx
->out
.length
!= 0) {
204 NTSTATUS status
= packet_fd_write(ctx
);
205 if (!NT_STATUS_IS_OK(status
)) {
213 * Send a list of DATA_BLOBs
215 * Example: packet_send(ctx, 2, data_blob_const(&size, sizeof(size)),
216 * data_blob_const(buf, size));
218 NTSTATUS
packet_send(struct packet_context
*ctx
, int num_blobs
, ...)
225 len
= ctx
->out
.length
;
227 va_start(ap
, num_blobs
);
228 for (i
=0; i
<num_blobs
; i
++) {
230 DATA_BLOB blob
= va_arg(ap
, DATA_BLOB
);
232 tmp
= len
+ blob
.length
;
234 DEBUG(0, ("integer overflow\n"));
236 return NT_STATUS_NO_MEMORY
;
246 if (!(out
= TALLOC_REALLOC_ARRAY(ctx
, ctx
->out
.data
, uint8
, len
))) {
247 DEBUG(0, ("talloc failed\n"));
248 return NT_STATUS_NO_MEMORY
;
253 va_start(ap
, num_blobs
);
254 for (i
=0; i
<num_blobs
; i
++) {
255 DATA_BLOB blob
= va_arg(ap
, DATA_BLOB
);
257 memcpy(ctx
->out
.data
+ctx
->out
.length
, blob
.data
, blob
.length
);
258 ctx
->out
.length
+= blob
.length
;
262 SMB_ASSERT(ctx
->out
.length
== len
);
267 * Get the packet context's file descriptor
269 int packet_get_fd(struct packet_context
*ctx
)