s3 swat: Add XSRF protection to viewconfig page
[Samba.git] / source4 / libcli / raw / clitransport.c
blob5cf0272e88469b774a898c25ba43127aaf42db4a
1 /*
2 Unix SMB/CIFS implementation.
3 SMB client transport context management functions
5 Copyright (C) Andrew Tridgell 1994-2005
6 Copyright (C) James Myers 2003 <myersjj@samba.org>
8 This program is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3 of the License, or
11 (at your option) any later version.
13 This program is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "libcli/raw/libcliraw.h"
24 #include "libcli/raw/raw_proto.h"
25 #include "lib/socket/socket.h"
26 #include "../lib/util/dlinklist.h"
27 #include "lib/events/events.h"
28 #include "lib/stream/packet.h"
29 #include "librpc/gen_ndr/ndr_nbt.h"
30 #include "../libcli/nbt/libnbt.h"
34 an event has happened on the socket
36 static void smbcli_transport_event_handler(struct tevent_context *ev,
37 struct tevent_fd *fde,
38 uint16_t flags, void *private_data)
40 struct smbcli_transport *transport = talloc_get_type(private_data,
41 struct smbcli_transport);
42 if (flags & EVENT_FD_READ) {
43 packet_recv(transport->packet);
44 return;
46 if (flags & EVENT_FD_WRITE) {
47 packet_queue_run(transport->packet);
52 destroy a transport
54 static int transport_destructor(struct smbcli_transport *transport)
56 smbcli_transport_dead(transport, NT_STATUS_LOCAL_DISCONNECT);
57 return 0;
62 handle receive errors
64 static void smbcli_transport_error(void *private_data, NTSTATUS status)
66 struct smbcli_transport *transport = talloc_get_type(private_data, struct smbcli_transport);
67 smbcli_transport_dead(transport, status);
70 static NTSTATUS smbcli_transport_finish_recv(void *private_data, DATA_BLOB blob);
73 create a transport structure based on an established socket
75 struct smbcli_transport *smbcli_transport_init(struct smbcli_socket *sock,
76 TALLOC_CTX *parent_ctx,
77 bool primary,
78 struct smbcli_options *options,
79 struct smb_iconv_convenience *iconv_convenience)
81 struct smbcli_transport *transport;
83 transport = talloc_zero(parent_ctx, struct smbcli_transport);
84 if (!transport) return NULL;
86 if (primary) {
87 transport->socket = talloc_steal(transport, sock);
88 } else {
89 transport->socket = talloc_reference(transport, sock);
91 transport->negotiate.protocol = PROTOCOL_NT1;
92 transport->options = *options;
93 transport->negotiate.max_xmit = transport->options.max_xmit;
94 transport->iconv_convenience = iconv_convenience;
96 /* setup the stream -> packet parser */
97 transport->packet = packet_init(transport);
98 if (transport->packet == NULL) {
99 talloc_free(transport);
100 return NULL;
102 packet_set_private(transport->packet, transport);
103 packet_set_socket(transport->packet, transport->socket->sock);
104 packet_set_callback(transport->packet, smbcli_transport_finish_recv);
105 packet_set_full_request(transport->packet, packet_full_request_nbt);
106 packet_set_error_handler(transport->packet, smbcli_transport_error);
107 packet_set_event_context(transport->packet, transport->socket->event.ctx);
108 packet_set_nofree(transport->packet);
110 smbcli_init_signing(transport);
112 ZERO_STRUCT(transport->called);
114 /* take over event handling from the socket layer - it only
115 handles events up until we are connected */
116 talloc_free(transport->socket->event.fde);
117 transport->socket->event.fde = event_add_fd(transport->socket->event.ctx,
118 transport->socket->sock,
119 socket_get_fd(transport->socket->sock),
120 EVENT_FD_READ,
121 smbcli_transport_event_handler,
122 transport);
124 packet_set_fde(transport->packet, transport->socket->event.fde);
125 packet_set_serialise(transport->packet);
126 talloc_set_destructor(transport, transport_destructor);
128 return transport;
132 mark the transport as dead
134 void smbcli_transport_dead(struct smbcli_transport *transport, NTSTATUS status)
136 smbcli_sock_dead(transport->socket);
138 if (NT_STATUS_EQUAL(NT_STATUS_UNSUCCESSFUL, status)) {
139 status = NT_STATUS_UNEXPECTED_NETWORK_ERROR;
142 /* kill only the first pending receive - this is so that if
143 that async function frees the connection we don't die trying
144 to use old memory. The caller has to cope with only one
145 network error */
146 if (transport->pending_recv) {
147 struct smbcli_request *req = transport->pending_recv;
148 req->state = SMBCLI_REQUEST_ERROR;
149 req->status = status;
150 DLIST_REMOVE(transport->pending_recv, req);
151 if (req->async.fn) {
152 req->async.fn(req);
159 send a session request
161 struct smbcli_request *smbcli_transport_connect_send(struct smbcli_transport *transport,
162 struct nbt_name *calling,
163 struct nbt_name *called)
165 uint8_t *p;
166 struct smbcli_request *req;
167 DATA_BLOB calling_blob, called_blob;
168 TALLOC_CTX *tmp_ctx = talloc_new(transport);
169 NTSTATUS status;
171 status = nbt_name_dup(transport, called, &transport->called);
172 if (!NT_STATUS_IS_OK(status)) goto failed;
174 status = nbt_name_to_blob(tmp_ctx, transport->iconv_convenience, &calling_blob, calling);
175 if (!NT_STATUS_IS_OK(status)) goto failed;
177 status = nbt_name_to_blob(tmp_ctx, transport->iconv_convenience, &called_blob, called);
178 if (!NT_STATUS_IS_OK(status)) goto failed;
180 /* allocate output buffer */
181 req = smbcli_request_setup_nonsmb(transport,
182 NBT_HDR_SIZE +
183 calling_blob.length + called_blob.length);
184 if (req == NULL) goto failed;
186 /* put in the destination name */
187 p = req->out.buffer + NBT_HDR_SIZE;
188 memcpy(p, called_blob.data, called_blob.length);
189 p += called_blob.length;
191 memcpy(p, calling_blob.data, calling_blob.length);
192 p += calling_blob.length;
194 _smb_setlen(req->out.buffer, PTR_DIFF(p, req->out.buffer) - NBT_HDR_SIZE);
195 SCVAL(req->out.buffer,0,0x81);
197 if (!smbcli_request_send(req)) {
198 smbcli_request_destroy(req);
199 goto failed;
202 talloc_free(tmp_ctx);
203 return req;
205 failed:
206 talloc_free(tmp_ctx);
207 return NULL;
211 map a session request error to a NTSTATUS
213 static NTSTATUS map_session_refused_error(uint8_t error)
215 switch (error) {
216 case 0x80:
217 case 0x81:
218 return NT_STATUS_REMOTE_NOT_LISTENING;
219 case 0x82:
220 return NT_STATUS_RESOURCE_NAME_NOT_FOUND;
221 case 0x83:
222 return NT_STATUS_REMOTE_RESOURCES;
224 return NT_STATUS_UNEXPECTED_IO_ERROR;
229 finish a smbcli_transport_connect()
231 NTSTATUS smbcli_transport_connect_recv(struct smbcli_request *req)
233 NTSTATUS status;
235 if (!smbcli_request_receive(req)) {
236 smbcli_request_destroy(req);
237 return NT_STATUS_UNEXPECTED_NETWORK_ERROR;
240 switch (CVAL(req->in.buffer,0)) {
241 case 0x82:
242 status = NT_STATUS_OK;
243 break;
244 case 0x83:
245 status = map_session_refused_error(CVAL(req->in.buffer,4));
246 break;
247 case 0x84:
248 DEBUG(1,("Warning: session retarget not supported\n"));
249 status = NT_STATUS_NOT_SUPPORTED;
250 break;
251 default:
252 status = NT_STATUS_UNEXPECTED_IO_ERROR;
253 break;
256 smbcli_request_destroy(req);
257 return status;
262 send a session request (if needed)
264 bool smbcli_transport_connect(struct smbcli_transport *transport,
265 struct nbt_name *calling,
266 struct nbt_name *called)
268 struct smbcli_request *req;
269 NTSTATUS status;
271 if (transport->socket->port == 445) {
272 return true;
275 req = smbcli_transport_connect_send(transport,
276 calling, called);
277 status = smbcli_transport_connect_recv(req);
278 return NT_STATUS_IS_OK(status);
281 /****************************************************************************
282 get next mid in sequence
283 ****************************************************************************/
284 uint16_t smbcli_transport_next_mid(struct smbcli_transport *transport)
286 uint16_t mid;
287 struct smbcli_request *req;
289 mid = transport->next_mid;
291 again:
292 /* now check to see if this mid is being used by one of the
293 pending requests. This is quite efficient because the list is
294 usually very short */
296 /* the zero mid is reserved for requests that don't have a mid */
297 if (mid == 0) mid = 1;
299 for (req=transport->pending_recv; req; req=req->next) {
300 if (req->mid == mid) {
301 mid++;
302 goto again;
306 transport->next_mid = mid+1;
307 return mid;
310 static void idle_handler(struct tevent_context *ev,
311 struct tevent_timer *te, struct timeval t, void *private_data)
313 struct smbcli_transport *transport = talloc_get_type(private_data,
314 struct smbcli_transport);
315 struct timeval next = timeval_add(&t, 0, transport->idle.period);
316 transport->socket->event.te = event_add_timed(transport->socket->event.ctx,
317 transport,
318 next,
319 idle_handler, transport);
320 transport->idle.func(transport, transport->idle.private_data);
324 setup the idle handler for a transport
325 the period is in microseconds
327 _PUBLIC_ void smbcli_transport_idle_handler(struct smbcli_transport *transport,
328 void (*idle_func)(struct smbcli_transport *, void *),
329 uint64_t period,
330 void *private_data)
332 transport->idle.func = idle_func;
333 transport->idle.private_data = private_data;
334 transport->idle.period = period;
336 if (transport->socket->event.te != NULL) {
337 talloc_free(transport->socket->event.te);
340 transport->socket->event.te = event_add_timed(transport->socket->event.ctx,
341 transport,
342 timeval_current_ofs(0, period),
343 idle_handler, transport);
347 we have a full request in our receive buffer - match it to a pending request
348 and process
350 static NTSTATUS smbcli_transport_finish_recv(void *private_data, DATA_BLOB blob)
352 struct smbcli_transport *transport = talloc_get_type(private_data,
353 struct smbcli_transport);
354 uint8_t *buffer, *hdr, *vwv;
355 int len;
356 uint16_t wct=0, mid = 0, op = 0;
357 struct smbcli_request *req = NULL;
359 buffer = blob.data;
360 len = blob.length;
362 hdr = buffer+NBT_HDR_SIZE;
363 vwv = hdr + HDR_VWV;
365 /* see if it could be an oplock break request */
366 if (smbcli_handle_oplock_break(transport, len, hdr, vwv)) {
367 talloc_free(buffer);
368 return NT_STATUS_OK;
371 /* at this point we need to check for a readbraw reply, as
372 these can be any length */
373 if (transport->readbraw_pending) {
374 transport->readbraw_pending = 0;
376 /* it must match the first entry in the pending queue
377 as the client is not allowed to have outstanding
378 readbraw requests */
379 req = transport->pending_recv;
380 if (!req) goto error;
382 req->in.buffer = buffer;
383 talloc_steal(req, buffer);
384 req->in.size = len;
385 req->in.allocated = req->in.size;
386 goto async;
389 if (len >= MIN_SMB_SIZE) {
390 /* extract the mid for matching to pending requests */
391 mid = SVAL(hdr, HDR_MID);
392 wct = CVAL(hdr, HDR_WCT);
393 op = CVAL(hdr, HDR_COM);
396 /* match the incoming request against the list of pending requests */
397 for (req=transport->pending_recv; req; req=req->next) {
398 if (req->mid == mid) break;
401 /* see if it's a ntcancel reply for the current MID */
402 req = smbcli_handle_ntcancel_reply(req, len, hdr);
404 if (!req) {
405 DEBUG(1,("Discarding unmatched reply with mid %d op %d\n", mid, op));
406 goto error;
409 /* fill in the 'in' portion of the matching request */
410 req->in.buffer = buffer;
411 talloc_steal(req, buffer);
412 req->in.size = len;
413 req->in.allocated = req->in.size;
415 /* handle NBT session replies */
416 if (req->in.size >= 4 && req->in.buffer[0] != 0) {
417 req->status = NT_STATUS_OK;
418 goto async;
421 /* handle non-SMB replies */
422 if (req->in.size < NBT_HDR_SIZE + MIN_SMB_SIZE) {
423 req->state = SMBCLI_REQUEST_ERROR;
424 goto error;
427 if (req->in.size < NBT_HDR_SIZE + MIN_SMB_SIZE + VWV(wct)) {
428 DEBUG(2,("bad reply size for mid %d\n", mid));
429 req->status = NT_STATUS_UNSUCCESSFUL;
430 req->state = SMBCLI_REQUEST_ERROR;
431 goto error;
434 req->in.hdr = hdr;
435 req->in.vwv = vwv;
436 req->in.wct = wct;
437 if (req->in.size >= NBT_HDR_SIZE + MIN_SMB_SIZE + VWV(wct)) {
438 req->in.data = req->in.vwv + VWV(wct) + 2;
439 req->in.data_size = SVAL(req->in.vwv, VWV(wct));
440 if (req->in.size < NBT_HDR_SIZE + MIN_SMB_SIZE + VWV(wct) + req->in.data_size) {
441 DEBUG(3,("bad data size for mid %d\n", mid));
442 /* blergh - w2k3 gives a bogus data size values in some
443 openX replies */
444 req->in.data_size = req->in.size - (NBT_HDR_SIZE + MIN_SMB_SIZE + VWV(wct));
447 req->in.ptr = req->in.data;
448 req->flags2 = SVAL(req->in.hdr, HDR_FLG2);
450 smb_setup_bufinfo(req);
452 if (!(req->flags2 & FLAGS2_32_BIT_ERROR_CODES)) {
453 int eclass = CVAL(req->in.hdr,HDR_RCLS);
454 int code = SVAL(req->in.hdr,HDR_ERR);
455 if (eclass == 0 && code == 0) {
456 transport->error.e.nt_status = NT_STATUS_OK;
457 } else {
458 transport->error.e.nt_status = NT_STATUS_DOS(eclass, code);
460 } else {
461 transport->error.e.nt_status = NT_STATUS(IVAL(req->in.hdr, HDR_RCLS));
464 req->status = transport->error.e.nt_status;
465 if (NT_STATUS_IS_OK(req->status)) {
466 transport->error.etype = ETYPE_NONE;
467 } else {
468 transport->error.etype = ETYPE_SMB;
471 if (!smbcli_request_check_sign_mac(req)) {
472 transport->error.etype = ETYPE_SOCKET;
473 transport->error.e.socket_error = SOCKET_READ_BAD_SIG;
474 req->state = SMBCLI_REQUEST_ERROR;
475 req->status = NT_STATUS_ACCESS_DENIED;
476 goto error;
479 async:
480 /* if this request has an async handler then call that to
481 notify that the reply has been received. This might destroy
482 the request so it must happen last */
484 req->state = SMBCLI_REQUEST_DONE;
486 if (req->recv_helper.fn) {
488 * let the recv helper decide in
489 * what state the request really is
491 req->state = req->recv_helper.fn(req);
493 /* if more parts are needed, wait for them */
494 if (req->state <= SMBCLI_REQUEST_RECV) {
495 return NT_STATUS_OK;
498 DLIST_REMOVE(transport->pending_recv, req);
499 if (req->async.fn) {
500 req->async.fn(req);
502 return NT_STATUS_OK;
504 error:
505 if (req) {
506 DLIST_REMOVE(transport->pending_recv, req);
507 req->state = SMBCLI_REQUEST_ERROR;
508 if (req->async.fn) {
509 req->async.fn(req);
511 } else {
512 talloc_free(buffer);
514 return NT_STATUS_OK;
518 process some read/write requests that are pending
519 return false if the socket is dead
521 _PUBLIC_ bool smbcli_transport_process(struct smbcli_transport *transport)
523 NTSTATUS status;
524 size_t npending;
526 packet_queue_run(transport->packet);
527 if (transport->socket->sock == NULL) {
528 return false;
531 status = socket_pending(transport->socket->sock, &npending);
532 if (NT_STATUS_IS_OK(status) && npending > 0) {
533 packet_recv(transport->packet);
535 if (transport->socket->sock == NULL) {
536 return false;
538 return true;
542 handle timeouts of individual smb requests
544 static void smbcli_timeout_handler(struct tevent_context *ev, struct tevent_timer *te,
545 struct timeval t, void *private_data)
547 struct smbcli_request *req = talloc_get_type(private_data, struct smbcli_request);
549 if (req->state == SMBCLI_REQUEST_RECV) {
550 DLIST_REMOVE(req->transport->pending_recv, req);
552 req->status = NT_STATUS_IO_TIMEOUT;
553 req->state = SMBCLI_REQUEST_ERROR;
554 if (req->async.fn) {
555 req->async.fn(req);
561 destroy a request
563 static int smbcli_request_destructor(struct smbcli_request *req)
565 if (req->state == SMBCLI_REQUEST_RECV) {
566 DLIST_REMOVE(req->transport->pending_recv, req);
568 return 0;
573 put a request into the send queue
575 void smbcli_transport_send(struct smbcli_request *req)
577 DATA_BLOB blob;
578 NTSTATUS status;
580 /* check if the transport is dead */
581 if (req->transport->socket->sock == NULL) {
582 req->state = SMBCLI_REQUEST_ERROR;
583 req->status = NT_STATUS_NET_WRITE_FAULT;
584 return;
587 blob = data_blob_const(req->out.buffer, req->out.size);
588 status = packet_send(req->transport->packet, blob);
589 if (!NT_STATUS_IS_OK(status)) {
590 req->state = SMBCLI_REQUEST_ERROR;
591 req->status = status;
592 return;
595 if (req->one_way_request) {
596 req->state = SMBCLI_REQUEST_DONE;
597 smbcli_request_destroy(req);
598 return;
601 req->state = SMBCLI_REQUEST_RECV;
602 DLIST_ADD(req->transport->pending_recv, req);
604 /* add a timeout */
605 if (req->transport->options.request_timeout) {
606 event_add_timed(req->transport->socket->event.ctx, req,
607 timeval_current_ofs(req->transport->options.request_timeout, 0),
608 smbcli_timeout_handler, req);
611 talloc_set_destructor(req, smbcli_request_destructor);
615 /****************************************************************************
616 Send an SMBecho (async send)
617 *****************************************************************************/
618 _PUBLIC_ struct smbcli_request *smb_raw_echo_send(struct smbcli_transport *transport,
619 struct smb_echo *p)
621 struct smbcli_request *req;
623 req = smbcli_request_setup_transport(transport, SMBecho, 1, p->in.size);
624 if (!req) return NULL;
626 SSVAL(req->out.vwv, VWV(0), p->in.repeat_count);
628 memcpy(req->out.data, p->in.data, p->in.size);
630 ZERO_STRUCT(p->out);
632 if (!smbcli_request_send(req)) {
633 smbcli_request_destroy(req);
634 return NULL;
637 return req;
640 /****************************************************************************
641 raw echo interface (async recv)
642 ****************************************************************************/
643 NTSTATUS smb_raw_echo_recv(struct smbcli_request *req, TALLOC_CTX *mem_ctx,
644 struct smb_echo *p)
646 if (!smbcli_request_receive(req) ||
647 smbcli_request_is_error(req)) {
648 goto failed;
651 SMBCLI_CHECK_WCT(req, 1);
652 p->out.count++;
653 p->out.sequence_number = SVAL(req->in.vwv, VWV(0));
654 p->out.size = req->in.data_size;
655 talloc_free(p->out.data);
656 p->out.data = talloc_array(mem_ctx, uint8_t, p->out.size);
657 NT_STATUS_HAVE_NO_MEMORY(p->out.data);
659 if (!smbcli_raw_pull_data(&req->in.bufinfo, req->in.data, p->out.size, p->out.data)) {
660 req->status = NT_STATUS_BUFFER_TOO_SMALL;
663 if (p->out.count == p->in.repeat_count) {
664 return smbcli_request_destroy(req);
667 return NT_STATUS_OK;
669 failed:
670 return smbcli_request_destroy(req);
673 /****************************************************************************
674 Send a echo (sync interface)
675 *****************************************************************************/
676 NTSTATUS smb_raw_echo(struct smbcli_transport *transport, struct smb_echo *p)
678 struct smbcli_request *req = smb_raw_echo_send(transport, p);
679 return smbcli_request_simple_recv(req);