libsmb: Make cli_ntcreate cancellable
[Samba.git] / source3 / smbd / ipc.c
blobdbb259cedc2b1f52bee6dda668af8c7630f4c888
1 /*
2 Unix SMB/CIFS implementation.
3 Inter-process communication and named pipe handling
4 Copyright (C) Andrew Tridgell 1992-1998
6 SMB Version handling
7 Copyright (C) John H Terpstra 1995-1998
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
23 This file handles the named pipe and mailslot calls
24 in the SMBtrans protocol
27 #include "includes.h"
28 #include "smbd/smbd.h"
29 #include "smbd/globals.h"
30 #include "smbprofile.h"
31 #include "rpc_server/srv_pipe_hnd.h"
33 #define NERR_notsupported 50
35 static void api_no_reply(connection_struct *conn, struct smb_request *req);
37 /*******************************************************************
38 copies parameters and data, as needed, into the smb buffer
40 *both* the data and params sections should be aligned. this
41 is fudged in the rpc pipes by
42 at present, only the data section is. this may be a possible
43 cause of some of the ipc problems being experienced. lkcl26dec97
45 ******************************************************************/
47 static void copy_trans_params_and_data(char *outbuf, int align,
48 char *rparam, int param_offset, int param_len,
49 char *rdata, int data_offset, int data_len)
51 char *copy_into = smb_buf(outbuf);
53 if(param_len < 0)
54 param_len = 0;
56 if(data_len < 0)
57 data_len = 0;
59 DEBUG(5,("copy_trans_params_and_data: params[%d..%d] data[%d..%d] (align %d)\n",
60 param_offset, param_offset + param_len,
61 data_offset , data_offset + data_len,
62 align));
64 *copy_into = '\0';
66 copy_into += 1;
68 if (param_len)
69 memcpy(copy_into, &rparam[param_offset], param_len);
71 copy_into += param_len;
72 if (align) {
73 memset(copy_into, '\0', align);
76 copy_into += align;
78 if (data_len )
79 memcpy(copy_into, &rdata[data_offset], data_len);
82 NTSTATUS nt_status_np_pipe(NTSTATUS status)
84 if (NT_STATUS_EQUAL(status, NT_STATUS_CONNECTION_DISCONNECTED)) {
85 status = NT_STATUS_PIPE_DISCONNECTED;
86 } else if (NT_STATUS_EQUAL(status, NT_STATUS_CONNECTION_RESET)) {
87 status = NT_STATUS_PIPE_BROKEN;
90 return status;
93 /****************************************************************************
94 Send a trans reply.
95 ****************************************************************************/
97 void send_trans_reply(connection_struct *conn,
98 struct smb_request *req,
99 char *rparam, int rparam_len,
100 char *rdata, int rdata_len,
101 bool buffer_too_large)
103 int this_ldata,this_lparam;
104 int tot_data_sent = 0;
105 int tot_param_sent = 0;
106 int align;
108 int ldata = rdata ? rdata_len : 0;
109 int lparam = rparam ? rparam_len : 0;
110 struct smbd_server_connection *sconn = req->sconn;
111 int max_send = sconn->smb1.sessions.max_send;
112 /* HACK: make sure we send at least 128 byte in one go */
113 int hdr_overhead = SMB_BUFFER_SIZE_MIN - 128;
115 if (buffer_too_large)
116 DEBUG(5,("send_trans_reply: buffer %d too large\n", ldata ));
118 this_lparam = MIN(lparam,max_send - hdr_overhead);
119 this_ldata = MIN(ldata,max_send - (hdr_overhead+this_lparam));
121 align = ((this_lparam)%4);
123 reply_outbuf(req, 10, 1+align+this_ldata+this_lparam);
126 * We might have SMBtranss in req which was transferred to the outbuf,
127 * fix that.
129 SCVAL(req->outbuf, smb_com, SMBtrans);
131 copy_trans_params_and_data((char *)req->outbuf, align,
132 rparam, tot_param_sent, this_lparam,
133 rdata, tot_data_sent, this_ldata);
135 SSVAL(req->outbuf,smb_vwv0,lparam);
136 SSVAL(req->outbuf,smb_vwv1,ldata);
137 SSVAL(req->outbuf,smb_vwv3,this_lparam);
138 SSVAL(req->outbuf,smb_vwv4,
139 smb_offset(smb_buf(req->outbuf)+1, req->outbuf));
140 SSVAL(req->outbuf,smb_vwv5,0);
141 SSVAL(req->outbuf,smb_vwv6,this_ldata);
142 SSVAL(req->outbuf,smb_vwv7,
143 smb_offset(smb_buf(req->outbuf)+1+this_lparam+align,
144 req->outbuf));
145 SSVAL(req->outbuf,smb_vwv8,0);
146 SSVAL(req->outbuf,smb_vwv9,0);
148 if (buffer_too_large) {
149 error_packet_set((char *)req->outbuf, ERRDOS, ERRmoredata,
150 STATUS_BUFFER_OVERFLOW, __LINE__, __FILE__);
153 show_msg((char *)req->outbuf);
154 if (!srv_send_smb(sconn, (char *)req->outbuf,
155 true, req->seqnum+1,
156 IS_CONN_ENCRYPTED(conn), &req->pcd)) {
157 exit_server_cleanly("send_trans_reply: srv_send_smb failed.");
160 TALLOC_FREE(req->outbuf);
162 tot_data_sent = this_ldata;
163 tot_param_sent = this_lparam;
165 while (tot_data_sent < ldata || tot_param_sent < lparam)
167 this_lparam = MIN(lparam-tot_param_sent,
168 max_send - hdr_overhead);
169 this_ldata = MIN(ldata -tot_data_sent,
170 max_send - (hdr_overhead+this_lparam));
172 if(this_lparam < 0)
173 this_lparam = 0;
175 if(this_ldata < 0)
176 this_ldata = 0;
178 align = (this_lparam%4);
180 reply_outbuf(req, 10, 1+align+this_ldata+this_lparam);
183 * We might have SMBtranss in req which was transferred to the
184 * outbuf, fix that.
186 SCVAL(req->outbuf, smb_com, SMBtrans);
188 copy_trans_params_and_data((char *)req->outbuf, align,
189 rparam, tot_param_sent, this_lparam,
190 rdata, tot_data_sent, this_ldata);
192 SSVAL(req->outbuf,smb_vwv0,lparam);
193 SSVAL(req->outbuf,smb_vwv1,ldata);
195 SSVAL(req->outbuf,smb_vwv3,this_lparam);
196 SSVAL(req->outbuf,smb_vwv4,
197 smb_offset(smb_buf(req->outbuf)+1,req->outbuf));
198 SSVAL(req->outbuf,smb_vwv5,tot_param_sent);
199 SSVAL(req->outbuf,smb_vwv6,this_ldata);
200 SSVAL(req->outbuf,smb_vwv7,
201 smb_offset(smb_buf(req->outbuf)+1+this_lparam+align,
202 req->outbuf));
203 SSVAL(req->outbuf,smb_vwv8,tot_data_sent);
204 SSVAL(req->outbuf,smb_vwv9,0);
206 if (buffer_too_large) {
207 error_packet_set((char *)req->outbuf,
208 ERRDOS, ERRmoredata,
209 STATUS_BUFFER_OVERFLOW,
210 __LINE__, __FILE__);
213 show_msg((char *)req->outbuf);
214 if (!srv_send_smb(sconn, (char *)req->outbuf,
215 true, req->seqnum+1,
216 IS_CONN_ENCRYPTED(conn), &req->pcd))
217 exit_server_cleanly("send_trans_reply: srv_send_smb "
218 "failed.");
220 tot_data_sent += this_ldata;
221 tot_param_sent += this_lparam;
222 TALLOC_FREE(req->outbuf);
226 /****************************************************************************
227 Start the first part of an RPC reply which began with an SMBtrans request.
228 ****************************************************************************/
230 struct dcerpc_cmd_state {
231 struct fake_file_handle *handle;
232 uint8_t *data;
233 size_t num_data;
234 size_t max_read;
237 static void api_dcerpc_cmd_write_done(struct tevent_req *subreq);
238 static void api_dcerpc_cmd_read_done(struct tevent_req *subreq);
240 static void api_dcerpc_cmd(connection_struct *conn, struct smb_request *req,
241 files_struct *fsp, uint8_t *data, size_t length,
242 size_t max_read)
244 struct tevent_req *subreq;
245 struct dcerpc_cmd_state *state;
246 bool busy;
248 if (!fsp_is_np(fsp)) {
249 api_no_reply(conn, req);
250 return;
254 * Trans requests are only allowed
255 * if no other Trans or Read is active
257 busy = np_read_in_progress(fsp->fake_file_handle);
258 if (busy) {
259 reply_nterror(req, NT_STATUS_PIPE_BUSY);
260 return;
263 state = talloc(req, struct dcerpc_cmd_state);
264 if (state == NULL) {
265 reply_nterror(req, NT_STATUS_NO_MEMORY);
266 return;
268 req->async_priv = state;
270 state->handle = fsp->fake_file_handle;
273 * This memdup severely sucks. But doing it properly essentially means
274 * to rewrite lanman.c, something which I don't really want to do now.
276 state->data = (uint8_t *)talloc_memdup(state, data, length);
277 if (state->data == NULL) {
278 reply_nterror(req, NT_STATUS_NO_MEMORY);
279 return;
281 state->num_data = length;
282 state->max_read = max_read;
284 subreq = np_write_send(state, req->sconn->ev_ctx, state->handle,
285 state->data, length);
286 if (subreq == NULL) {
287 TALLOC_FREE(state);
288 reply_nterror(req, NT_STATUS_NO_MEMORY);
289 return;
291 tevent_req_set_callback(subreq, api_dcerpc_cmd_write_done,
292 talloc_move(conn, &req));
295 static void api_dcerpc_cmd_write_done(struct tevent_req *subreq)
297 struct smb_request *req = tevent_req_callback_data(
298 subreq, struct smb_request);
299 struct dcerpc_cmd_state *state = talloc_get_type_abort(
300 req->async_priv, struct dcerpc_cmd_state);
301 NTSTATUS status;
302 ssize_t nwritten = -1;
304 status = np_write_recv(subreq, &nwritten);
305 TALLOC_FREE(subreq);
306 if (!NT_STATUS_IS_OK(status)) {
307 NTSTATUS old = status;
308 status = nt_status_np_pipe(old);
310 DEBUG(10, ("Could not write to pipe: %s%s%s\n",
311 nt_errstr(old),
312 NT_STATUS_EQUAL(old, status)?"":" => ",
313 NT_STATUS_EQUAL(old, status)?"":nt_errstr(status)));
314 reply_nterror(req, status);
315 goto send;
317 if (nwritten != state->num_data) {
318 status = NT_STATUS_PIPE_NOT_AVAILABLE;
319 DEBUG(10, ("Could not write to pipe: (%d/%d) => %s\n",
320 (int)state->num_data,
321 (int)nwritten, nt_errstr(status)));
322 reply_nterror(req, status);
323 goto send;
326 state->data = talloc_realloc(state, state->data, uint8_t,
327 state->max_read);
328 if (state->data == NULL) {
329 reply_nterror(req, NT_STATUS_NO_MEMORY);
330 goto send;
333 subreq = np_read_send(state, req->sconn->ev_ctx,
334 state->handle, state->data, state->max_read);
335 if (subreq == NULL) {
336 reply_nterror(req, NT_STATUS_NO_MEMORY);
337 goto send;
339 tevent_req_set_callback(subreq, api_dcerpc_cmd_read_done, req);
340 return;
342 send:
343 if (!srv_send_smb(
344 req->sconn, (char *)req->outbuf,
345 true, req->seqnum+1,
346 IS_CONN_ENCRYPTED(req->conn) || req->encrypted,
347 &req->pcd)) {
348 exit_server_cleanly("api_dcerpc_cmd_write_done: "
349 "srv_send_smb failed.");
351 TALLOC_FREE(req);
354 static void api_dcerpc_cmd_read_done(struct tevent_req *subreq)
356 struct smb_request *req = tevent_req_callback_data(
357 subreq, struct smb_request);
358 struct dcerpc_cmd_state *state = talloc_get_type_abort(
359 req->async_priv, struct dcerpc_cmd_state);
360 NTSTATUS status;
361 ssize_t nread;
362 bool is_data_outstanding;
364 status = np_read_recv(subreq, &nread, &is_data_outstanding);
365 TALLOC_FREE(subreq);
367 if (!NT_STATUS_IS_OK(status)) {
368 NTSTATUS old = status;
369 status = nt_status_np_pipe(old);
371 DEBUG(10, ("Could not read from to pipe: %s%s%s\n",
372 nt_errstr(old),
373 NT_STATUS_EQUAL(old, status)?"":" => ",
374 NT_STATUS_EQUAL(old, status)?"":nt_errstr(status)));
375 reply_nterror(req, status);
377 if (!srv_send_smb(req->sconn, (char *)req->outbuf,
378 true, req->seqnum+1,
379 IS_CONN_ENCRYPTED(req->conn)
380 ||req->encrypted, &req->pcd)) {
381 exit_server_cleanly("api_dcerpc_cmd_read_done: "
382 "srv_send_smb failed.");
384 TALLOC_FREE(req);
385 return;
388 send_trans_reply(req->conn, req, NULL, 0, (char *)state->data, nread,
389 is_data_outstanding);
390 TALLOC_FREE(req);
393 /****************************************************************************
394 WaitNamedPipeHandleState
395 ****************************************************************************/
397 static void api_WNPHS(connection_struct *conn, struct smb_request *req,
398 struct files_struct *fsp, char *param, int param_len)
400 if (!param || param_len < 2) {
401 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
402 return;
405 DEBUG(4,("WaitNamedPipeHandleState priority %x\n",
406 (int)SVAL(param,0)));
408 send_trans_reply(conn, req, NULL, 0, NULL, 0, False);
412 /****************************************************************************
413 SetNamedPipeHandleState
414 ****************************************************************************/
416 static void api_SNPHS(connection_struct *conn, struct smb_request *req,
417 struct files_struct *fsp, char *param, int param_len)
419 if (!param || param_len < 2) {
420 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
421 return;
424 DEBUG(4,("SetNamedPipeHandleState to code %x\n", (int)SVAL(param,0)));
426 send_trans_reply(conn, req, NULL, 0, NULL, 0, False);
430 /****************************************************************************
431 When no reply is generated, indicate unsupported.
432 ****************************************************************************/
434 static void api_no_reply(connection_struct *conn, struct smb_request *req)
436 char rparam[4];
438 /* unsupported */
439 SSVAL(rparam,0,NERR_notsupported);
440 SSVAL(rparam,2,0); /* converter word */
442 DEBUG(3,("Unsupported API fd command\n"));
444 /* now send the reply */
445 send_trans_reply(conn, req, rparam, 4, NULL, 0, False);
447 return;
450 /****************************************************************************
451 Handle remote api calls delivered to a named pipe already opened.
452 ****************************************************************************/
454 static void api_fd_reply(connection_struct *conn, uint64_t vuid,
455 struct smb_request *req,
456 uint16 *setup, uint8_t *data, char *params,
457 int suwcnt, int tdscnt, int tpscnt,
458 int mdrcnt, int mprcnt)
460 struct files_struct *fsp;
461 int pnum;
462 int subcommand;
464 DEBUG(5,("api_fd_reply\n"));
466 /* First find out the name of this file. */
467 if (suwcnt != 2) {
468 DEBUG(0,("Unexpected named pipe transaction.\n"));
469 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
470 return;
473 /* Get the file handle and hence the file name. */
475 * NB. The setup array has already been transformed
476 * via SVAL and so is in host byte order.
478 pnum = ((int)setup[1]) & 0xFFFF;
479 subcommand = ((int)setup[0]) & 0xFFFF;
481 fsp = file_fsp(req, pnum);
483 if (!fsp_is_np(fsp)) {
484 if (subcommand == TRANSACT_WAITNAMEDPIPEHANDLESTATE) {
485 /* Win9x does this call with a unicode pipe name, not a pnum. */
486 /* Just return success for now... */
487 DEBUG(3,("Got TRANSACT_WAITNAMEDPIPEHANDLESTATE on text pipe name\n"));
488 send_trans_reply(conn, req, NULL, 0, NULL, 0, False);
489 return;
492 DEBUG(1,("api_fd_reply: INVALID PIPE HANDLE: %x\n", pnum));
493 reply_nterror(req, NT_STATUS_INVALID_HANDLE);
494 return;
497 if (vuid != fsp->vuid) {
498 DEBUG(1, ("Got pipe request (pnum %x) using invalid VUID %llu, "
499 "expected %llu\n", pnum, (unsigned long long)vuid,
500 (unsigned long long)fsp->vuid));
501 reply_nterror(req, NT_STATUS_INVALID_HANDLE);
502 return;
505 DEBUG(3,("Got API command 0x%x on pipe \"%s\" (pnum %x)\n",
506 subcommand, fsp_str_dbg(fsp), pnum));
508 DEBUG(10, ("api_fd_reply: p:%p max_trans_reply: %d\n", fsp, mdrcnt));
510 switch (subcommand) {
511 case TRANSACT_DCERPCCMD: {
512 /* dce/rpc command */
513 api_dcerpc_cmd(conn, req, fsp, (uint8_t *)data, tdscnt,
514 mdrcnt);
515 break;
517 case TRANSACT_WAITNAMEDPIPEHANDLESTATE:
518 /* Wait Named Pipe Handle state */
519 api_WNPHS(conn, req, fsp, params, tpscnt);
520 break;
521 case TRANSACT_SETNAMEDPIPEHANDLESTATE:
522 /* Set Named Pipe Handle state */
523 api_SNPHS(conn, req, fsp, params, tpscnt);
524 break;
525 default:
526 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
527 return;
531 /****************************************************************************
532 Handle named pipe commands.
533 ****************************************************************************/
535 static void named_pipe(connection_struct *conn, uint64_t vuid,
536 struct smb_request *req,
537 const char *name, uint16 *setup,
538 char *data, char *params,
539 int suwcnt, int tdscnt,int tpscnt,
540 int msrcnt, int mdrcnt, int mprcnt)
542 DEBUG(3,("named pipe command on <%s> name\n", name));
544 if (strequal(name,"LANMAN")) {
545 api_reply(conn, vuid, req,
546 data, params,
547 tdscnt, tpscnt,
548 mdrcnt, mprcnt);
549 return;
552 if (strequal(name,"WKSSVC") ||
553 strequal(name,"SRVSVC") ||
554 strequal(name,"WINREG") ||
555 strequal(name,"SAMR") ||
556 strequal(name,"LSARPC")) {
558 DEBUG(4,("named pipe command from Win95 (wow!)\n"));
560 api_fd_reply(conn, vuid, req,
561 setup, (uint8_t *)data, params,
562 suwcnt, tdscnt, tpscnt,
563 mdrcnt, mprcnt);
564 return;
567 if (strlen(name) < 1) {
568 api_fd_reply(conn, vuid, req,
569 setup, (uint8_t *)data,
570 params, suwcnt, tdscnt,
571 tpscnt, mdrcnt, mprcnt);
572 return;
575 if (setup)
576 DEBUG(3,("unknown named pipe: setup 0x%X setup1=%d\n",
577 (int)setup[0],(int)setup[1]));
579 reply_nterror(req, NT_STATUS_NOT_SUPPORTED);
580 return;
583 static void handle_trans(connection_struct *conn, struct smb_request *req,
584 struct trans_state *state)
586 char *local_machine_name;
587 int name_offset = 0;
589 DEBUG(3,("trans <%s> data=%u params=%u setup=%u\n",
590 state->name,(unsigned int)state->total_data,(unsigned int)state->total_param,
591 (unsigned int)state->setup_count));
594 * WinCE wierdness....
597 local_machine_name = talloc_asprintf(state, "\\%s\\",
598 get_local_machine_name());
600 if (local_machine_name == NULL) {
601 reply_nterror(req, NT_STATUS_NO_MEMORY);
602 return;
605 if (strnequal(state->name, local_machine_name,
606 strlen(local_machine_name))) {
607 name_offset = strlen(local_machine_name)-1;
610 if (!strnequal(&state->name[name_offset], "\\PIPE",
611 strlen("\\PIPE"))) {
612 reply_nterror(req, NT_STATUS_NOT_SUPPORTED);
613 return;
616 name_offset += strlen("\\PIPE");
618 /* Win9x weirdness. When talking to a unicode server Win9x
619 only sends \PIPE instead of \PIPE\ */
621 if (state->name[name_offset] == '\\')
622 name_offset++;
624 DEBUG(5,("calling named_pipe\n"));
625 named_pipe(conn, state->vuid, req,
626 state->name+name_offset,
627 state->setup,state->data,
628 state->param,
629 state->setup_count,state->total_data,
630 state->total_param,
631 state->max_setup_return,
632 state->max_data_return,
633 state->max_param_return);
635 if (state->close_on_completion) {
636 struct smbXsrv_tcon *tcon;
637 NTSTATUS status;
639 tcon = conn->tcon;
640 req->conn = NULL;
641 conn = NULL;
644 * TODO: cancel all outstanding requests on the tcon
646 status = smbXsrv_tcon_disconnect(tcon, state->vuid);
647 if (!NT_STATUS_IS_OK(status)) {
648 DEBUG(0, ("handle_trans: "
649 "smbXsrv_tcon_disconnect() failed: %s\n",
650 nt_errstr(status)));
652 * If we hit this case, there is something completely
653 * wrong, so we better disconnect the transport connection.
655 exit_server(__location__ ": smbXsrv_tcon_disconnect failed");
656 return;
659 TALLOC_FREE(tcon);
662 return;
665 /****************************************************************************
666 Reply to a SMBtrans.
667 ****************************************************************************/
669 void reply_trans(struct smb_request *req)
671 connection_struct *conn = req->conn;
672 unsigned int dsoff;
673 unsigned int dscnt;
674 unsigned int psoff;
675 unsigned int pscnt;
676 struct trans_state *state;
677 NTSTATUS result;
679 START_PROFILE(SMBtrans);
681 if (req->wct < 14) {
682 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
683 END_PROFILE(SMBtrans);
684 return;
687 dsoff = SVAL(req->vwv+12, 0);
688 dscnt = SVAL(req->vwv+11, 0);
689 psoff = SVAL(req->vwv+10, 0);
690 pscnt = SVAL(req->vwv+9, 0);
692 result = allow_new_trans(conn->pending_trans, req->mid);
693 if (!NT_STATUS_IS_OK(result)) {
694 DEBUG(2, ("Got invalid trans request: %s\n",
695 nt_errstr(result)));
696 reply_nterror(req, result);
697 END_PROFILE(SMBtrans);
698 return;
701 if ((state = talloc(conn, struct trans_state)) == NULL) {
702 DEBUG(0, ("talloc failed\n"));
703 reply_nterror(req, NT_STATUS_NO_MEMORY);
704 END_PROFILE(SMBtrans);
705 return;
708 state->cmd = SMBtrans;
710 state->mid = req->mid;
711 state->vuid = req->vuid;
712 state->setup_count = CVAL(req->vwv+13, 0);
713 state->setup = NULL;
714 state->total_param = SVAL(req->vwv+0, 0);
715 state->param = NULL;
716 state->total_data = SVAL(req->vwv+1, 0);
717 state->data = NULL;
718 state->max_param_return = SVAL(req->vwv+2, 0);
719 state->max_data_return = SVAL(req->vwv+3, 0);
720 state->max_setup_return = CVAL(req->vwv+4, 0);
721 state->close_on_completion = BITSETW(req->vwv+5, 0);
722 state->one_way = BITSETW(req->vwv+5, 1);
724 srvstr_pull_req_talloc(state, req, &state->name, req->buf,
725 STR_TERMINATE);
727 if ((dscnt > state->total_data) || (pscnt > state->total_param) ||
728 !state->name)
729 goto bad_param;
731 if (state->total_data) {
733 if (trans_oob(state->total_data, 0, dscnt)
734 || trans_oob(smb_len(req->inbuf), dsoff, dscnt)) {
735 goto bad_param;
738 /* Can't use talloc here, the core routines do realloc on the
739 * params and data. Out of paranoia, 100 bytes too many. */
740 state->data = (char *)SMB_MALLOC(state->total_data+100);
741 if (state->data == NULL) {
742 DEBUG(0,("reply_trans: data malloc fail for %u "
743 "bytes !\n", (unsigned int)state->total_data));
744 TALLOC_FREE(state);
745 reply_nterror(req, NT_STATUS_NO_MEMORY);
746 END_PROFILE(SMBtrans);
747 return;
749 /* null-terminate the slack space */
750 memset(&state->data[state->total_data], 0, 100);
752 memcpy(state->data,smb_base(req->inbuf)+dsoff,dscnt);
755 if (state->total_param) {
757 if (trans_oob(state->total_param, 0, pscnt)
758 || trans_oob(smb_len(req->inbuf), psoff, pscnt)) {
759 goto bad_param;
762 /* Can't use talloc here, the core routines do realloc on the
763 * params and data. Out of paranoia, 100 bytes too many */
764 state->param = (char *)SMB_MALLOC(state->total_param+100);
765 if (state->param == NULL) {
766 DEBUG(0,("reply_trans: param malloc fail for %u "
767 "bytes !\n", (unsigned int)state->total_param));
768 SAFE_FREE(state->data);
769 TALLOC_FREE(state);
770 reply_nterror(req, NT_STATUS_NO_MEMORY);
771 END_PROFILE(SMBtrans);
772 return;
774 /* null-terminate the slack space */
775 memset(&state->param[state->total_param], 0, 100);
777 memcpy(state->param,smb_base(req->inbuf)+psoff,pscnt);
780 state->received_data = dscnt;
781 state->received_param = pscnt;
783 if (state->setup_count) {
784 unsigned int i;
787 * No overflow possible here, state->setup_count is an
788 * unsigned int, being filled by a single byte from
789 * CVAL(req->vwv+13, 0) above. The cast in the comparison
790 * below is not necessary, it's here to clarify things. The
791 * validity of req->vwv and req->wct has been checked in
792 * init_smb_request already.
794 if (state->setup_count + 14 > (unsigned int)req->wct) {
795 goto bad_param;
798 if((state->setup = talloc_array(
799 state, uint16, state->setup_count)) == NULL) {
800 DEBUG(0,("reply_trans: setup malloc fail for %u "
801 "bytes !\n", (unsigned int)
802 (state->setup_count * sizeof(uint16))));
803 SAFE_FREE(state->data);
804 SAFE_FREE(state->param);
805 TALLOC_FREE(state);
806 reply_nterror(req, NT_STATUS_NO_MEMORY);
807 END_PROFILE(SMBtrans);
808 return;
811 for (i=0;i<state->setup_count;i++) {
812 state->setup[i] = SVAL(req->vwv + 14 + i, 0);
816 state->received_param = pscnt;
818 if ((state->received_param != state->total_param) ||
819 (state->received_data != state->total_data)) {
820 DLIST_ADD(conn->pending_trans, state);
822 /* We need to send an interim response then receive the rest
823 of the parameter/data bytes */
824 reply_outbuf(req, 0, 0);
825 show_msg((char *)req->outbuf);
826 END_PROFILE(SMBtrans);
827 return;
830 talloc_steal(talloc_tos(), state);
832 handle_trans(conn, req, state);
834 SAFE_FREE(state->data);
835 SAFE_FREE(state->param);
836 TALLOC_FREE(state);
838 END_PROFILE(SMBtrans);
839 return;
841 bad_param:
843 DEBUG(0,("reply_trans: invalid trans parameters\n"));
844 SAFE_FREE(state->data);
845 SAFE_FREE(state->param);
846 TALLOC_FREE(state);
847 END_PROFILE(SMBtrans);
848 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
849 return;
852 /****************************************************************************
853 Reply to a secondary SMBtrans.
854 ****************************************************************************/
856 void reply_transs(struct smb_request *req)
858 connection_struct *conn = req->conn;
859 unsigned int pcnt,poff,dcnt,doff,pdisp,ddisp;
860 struct trans_state *state;
862 START_PROFILE(SMBtranss);
864 show_msg((const char *)req->inbuf);
866 if (req->wct < 8) {
867 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
868 END_PROFILE(SMBtranss);
869 return;
872 for (state = conn->pending_trans; state != NULL;
873 state = state->next) {
874 if (state->mid == req->mid) {
875 break;
879 if ((state == NULL) || (state->cmd != SMBtrans)) {
880 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
881 END_PROFILE(SMBtranss);
882 return;
885 /* Revise total_params and total_data in case they have changed
886 * downwards */
888 if (SVAL(req->vwv+0, 0) < state->total_param)
889 state->total_param = SVAL(req->vwv+0, 0);
890 if (SVAL(req->vwv+1, 0) < state->total_data)
891 state->total_data = SVAL(req->vwv+1, 0);
893 pcnt = SVAL(req->vwv+2, 0);
894 poff = SVAL(req->vwv+3, 0);
895 pdisp = SVAL(req->vwv+4, 0);
897 dcnt = SVAL(req->vwv+5, 0);
898 doff = SVAL(req->vwv+6, 0);
899 ddisp = SVAL(req->vwv+7, 0);
901 state->received_param += pcnt;
902 state->received_data += dcnt;
904 if ((state->received_data > state->total_data) ||
905 (state->received_param > state->total_param))
906 goto bad_param;
908 if (pcnt) {
909 if (trans_oob(state->total_param, pdisp, pcnt)
910 || trans_oob(smb_len(req->inbuf), poff, pcnt)) {
911 goto bad_param;
913 memcpy(state->param+pdisp,smb_base(req->inbuf)+poff,pcnt);
916 if (dcnt) {
917 if (trans_oob(state->total_data, ddisp, dcnt)
918 || trans_oob(smb_len(req->inbuf), doff, dcnt)) {
919 goto bad_param;
921 memcpy(state->data+ddisp, smb_base(req->inbuf)+doff,dcnt);
924 if ((state->received_param < state->total_param) ||
925 (state->received_data < state->total_data)) {
926 END_PROFILE(SMBtranss);
927 return;
930 talloc_steal(talloc_tos(), state);
932 handle_trans(conn, req, state);
934 DLIST_REMOVE(conn->pending_trans, state);
935 SAFE_FREE(state->data);
936 SAFE_FREE(state->param);
937 TALLOC_FREE(state);
939 END_PROFILE(SMBtranss);
940 return;
942 bad_param:
944 DEBUG(0,("reply_transs: invalid trans parameters\n"));
945 DLIST_REMOVE(conn->pending_trans, state);
946 SAFE_FREE(state->data);
947 SAFE_FREE(state->param);
948 TALLOC_FREE(state);
949 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
950 END_PROFILE(SMBtranss);
951 return;