Convert np_write to tevent_req
[amitay/samba.git] / source3 / smbd / ipc.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Inter-process communication and named pipe handling
4    Copyright (C) Andrew Tridgell 1992-1998
5
6    SMB Version handling
7    Copyright (C) John H Terpstra 1995-1998
8
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.
13
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.
18
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/>.
21    */
22 /*
23    This file handles the named pipe and mailslot calls
24    in the SMBtrans protocol
25    */
26
27 #include "includes.h"
28 #include "smbd/globals.h"
29
30 #define NERR_notsupported 50
31
32 static void api_no_reply(connection_struct *conn, struct smb_request *req);
33
34 /*******************************************************************
35  copies parameters and data, as needed, into the smb buffer
36
37  *both* the data and params sections should be aligned.  this
38  is fudged in the rpc pipes by 
39  at present, only the data section is.  this may be a possible
40  cause of some of the ipc problems being experienced.  lkcl26dec97
41
42  ******************************************************************/
43
44 static void copy_trans_params_and_data(char *outbuf, int align,
45                                 char *rparam, int param_offset, int param_len,
46                                 char *rdata, int data_offset, int data_len)
47 {
48         char *copy_into = smb_buf(outbuf);
49
50         if(param_len < 0)
51                 param_len = 0;
52
53         if(data_len < 0)
54                 data_len = 0;
55
56         DEBUG(5,("copy_trans_params_and_data: params[%d..%d] data[%d..%d] (align %d)\n",
57                         param_offset, param_offset + param_len,
58                         data_offset , data_offset  + data_len,
59                         align));
60
61         *copy_into = '\0';
62
63         copy_into += 1;
64
65         if (param_len)
66                 memcpy(copy_into, &rparam[param_offset], param_len);
67
68         copy_into += param_len;
69         if (align) {
70                 memset(copy_into, '\0', align);
71         }
72
73         copy_into += align;
74
75         if (data_len )
76                 memcpy(copy_into, &rdata[data_offset], data_len);
77 }
78
79 /****************************************************************************
80  Send a trans reply.
81  ****************************************************************************/
82
83 void send_trans_reply(connection_struct *conn,
84                       struct smb_request *req,
85                       char *rparam, int rparam_len,
86                       char *rdata, int rdata_len,
87                       bool buffer_too_large)
88 {
89         int this_ldata,this_lparam;
90         int tot_data_sent = 0;
91         int tot_param_sent = 0;
92         int align;
93
94         int ldata  = rdata  ? rdata_len : 0;
95         int lparam = rparam ? rparam_len : 0;
96
97         if (buffer_too_large)
98                 DEBUG(5,("send_trans_reply: buffer %d too large\n", ldata ));
99
100         this_lparam = MIN(lparam,max_send - 500); /* hack */
101         this_ldata  = MIN(ldata,max_send - (500+this_lparam));
102
103         align = ((this_lparam)%4);
104
105         reply_outbuf(req, 10, 1+align+this_ldata+this_lparam);
106
107         /*
108          * We might have SMBtranss in req which was transferred to the outbuf,
109          * fix that.
110          */
111         SCVAL(req->outbuf, smb_com, SMBtrans);
112
113         copy_trans_params_and_data((char *)req->outbuf, align,
114                                 rparam, tot_param_sent, this_lparam,
115                                 rdata, tot_data_sent, this_ldata);
116
117         SSVAL(req->outbuf,smb_vwv0,lparam);
118         SSVAL(req->outbuf,smb_vwv1,ldata);
119         SSVAL(req->outbuf,smb_vwv3,this_lparam);
120         SSVAL(req->outbuf,smb_vwv4,
121               smb_offset(smb_buf(req->outbuf)+1, req->outbuf));
122         SSVAL(req->outbuf,smb_vwv5,0);
123         SSVAL(req->outbuf,smb_vwv6,this_ldata);
124         SSVAL(req->outbuf,smb_vwv7,
125               smb_offset(smb_buf(req->outbuf)+1+this_lparam+align,
126                          req->outbuf));
127         SSVAL(req->outbuf,smb_vwv8,0);
128         SSVAL(req->outbuf,smb_vwv9,0);
129
130         if (buffer_too_large) {
131                 error_packet_set((char *)req->outbuf, ERRDOS, ERRmoredata,
132                                  STATUS_BUFFER_OVERFLOW, __LINE__, __FILE__);
133         }
134
135         show_msg((char *)req->outbuf);
136         if (!srv_send_smb(smbd_server_fd(), (char *)req->outbuf,
137                           IS_CONN_ENCRYPTED(conn), &req->pcd)) {
138                 exit_server_cleanly("send_trans_reply: srv_send_smb failed.");
139         }
140
141         TALLOC_FREE(req->outbuf);
142
143         tot_data_sent = this_ldata;
144         tot_param_sent = this_lparam;
145
146         while (tot_data_sent < ldata || tot_param_sent < lparam)
147         {
148                 this_lparam = MIN(lparam-tot_param_sent,
149                                   max_send - 500); /* hack */
150                 this_ldata  = MIN(ldata -tot_data_sent,
151                                   max_send - (500+this_lparam));
152
153                 if(this_lparam < 0)
154                         this_lparam = 0;
155
156                 if(this_ldata < 0)
157                         this_ldata = 0;
158
159                 align = (this_lparam%4);
160
161                 reply_outbuf(req, 10, 1+align+this_ldata+this_lparam);
162
163                 /*
164                  * We might have SMBtranss in req which was transferred to the
165                  * outbuf, fix that.
166                  */
167                 SCVAL(req->outbuf, smb_com, SMBtrans);
168
169                 copy_trans_params_and_data((char *)req->outbuf, align,
170                                            rparam, tot_param_sent, this_lparam,
171                                            rdata, tot_data_sent, this_ldata);
172
173                 SSVAL(req->outbuf,smb_vwv3,this_lparam);
174                 SSVAL(req->outbuf,smb_vwv4,
175                       smb_offset(smb_buf(req->outbuf)+1,req->outbuf));
176                 SSVAL(req->outbuf,smb_vwv5,tot_param_sent);
177                 SSVAL(req->outbuf,smb_vwv6,this_ldata);
178                 SSVAL(req->outbuf,smb_vwv7,
179                       smb_offset(smb_buf(req->outbuf)+1+this_lparam+align,
180                                  req->outbuf));
181                 SSVAL(req->outbuf,smb_vwv8,tot_data_sent);
182                 SSVAL(req->outbuf,smb_vwv9,0);
183
184                 if (buffer_too_large) {
185                         error_packet_set((char *)req->outbuf,
186                                          ERRDOS, ERRmoredata,
187                                          STATUS_BUFFER_OVERFLOW,
188                                          __LINE__, __FILE__);
189                 }
190
191                 show_msg((char *)req->outbuf);
192                 if (!srv_send_smb(smbd_server_fd(), (char *)req->outbuf,
193                                   IS_CONN_ENCRYPTED(conn), &req->pcd))
194                         exit_server_cleanly("send_trans_reply: srv_send_smb "
195                                             "failed.");
196
197                 tot_data_sent  += this_ldata;
198                 tot_param_sent += this_lparam;
199                 TALLOC_FREE(req->outbuf);
200         }
201 }
202
203 /****************************************************************************
204  Start the first part of an RPC reply which began with an SMBtrans request.
205 ****************************************************************************/
206
207 struct dcerpc_cmd_state {
208         struct fake_file_handle *handle;
209         uint8_t *data;
210         size_t num_data;
211         size_t max_read;
212 };
213
214 static void api_dcerpc_cmd_write_done(struct tevent_req *subreq);
215 static void api_dcerpc_cmd_read_done(struct async_req *subreq);
216
217 static void api_dcerpc_cmd(connection_struct *conn, struct smb_request *req,
218                            files_struct *fsp, uint8_t *data, size_t length,
219                            size_t max_read)
220 {
221         struct tevent_req *subreq;
222         struct dcerpc_cmd_state *state;
223
224         if (!fsp_is_np(fsp)) {
225                 api_no_reply(conn, req);
226                 return;
227         }
228
229         state = talloc(req, struct dcerpc_cmd_state);
230         if (state == NULL) {
231                 reply_nterror(req, NT_STATUS_NO_MEMORY);
232                 return;
233         }
234         req->async_priv = state;
235
236         state->handle = fsp->fake_file_handle;
237
238         /*
239          * This memdup severely sucks. But doing it properly essentially means
240          * to rewrite lanman.c, something which I don't really want to do now.
241          */
242         state->data = (uint8_t *)talloc_memdup(state, data, length);
243         if (state->data == NULL) {
244                 reply_nterror(req, NT_STATUS_NO_MEMORY);
245                 return;
246         }
247         state->num_data = length;
248         state->max_read = max_read;
249
250         subreq = np_write_send(state, smbd_event_context(), state->handle,
251                                state->data, length);
252         if (subreq == NULL) {
253                 TALLOC_FREE(state);
254                 reply_nterror(req, NT_STATUS_NO_MEMORY);
255                 return;
256         }
257         tevent_req_set_callback(subreq, api_dcerpc_cmd_write_done,
258                                 talloc_move(conn, &req));
259 }
260
261 static void api_dcerpc_cmd_write_done(struct tevent_req *subreq)
262 {
263         struct smb_request *req = tevent_req_callback_data(
264                 subreq, struct smb_request);
265         struct dcerpc_cmd_state *state = talloc_get_type_abort(
266                 req->async_priv, struct dcerpc_cmd_state);
267         struct async_req *subreq2;
268         NTSTATUS status;
269         ssize_t nwritten = -1;
270
271         status = np_write_recv(subreq, &nwritten);
272         TALLOC_FREE(subreq);
273         if (!NT_STATUS_IS_OK(status) || (nwritten != state->num_data)) {
274                 DEBUG(10, ("Could not write to pipe: %s (%d/%d)\n",
275                            nt_errstr(status), (int)state->num_data,
276                            (int)nwritten));
277                 reply_nterror(req, NT_STATUS_PIPE_NOT_AVAILABLE);
278                 goto send;
279         }
280
281         state->data = TALLOC_REALLOC_ARRAY(state, state->data, uint8_t,
282                                            state->max_read);
283         if (state->data == NULL) {
284                 reply_nterror(req, NT_STATUS_NO_MEMORY);
285                 goto send;
286         }
287
288         subreq2 = np_read_send(req->conn, smbd_event_context(),
289                                state->handle, state->data, state->max_read);
290         if (subreq2 == NULL) {
291                 reply_nterror(req, NT_STATUS_NO_MEMORY);
292                 goto send;
293         }
294
295         subreq2->async.fn = api_dcerpc_cmd_read_done;
296         subreq2->async.priv = req;
297         return;
298
299  send:
300         if (!srv_send_smb(
301                     smbd_server_fd(), (char *)req->outbuf,
302                     IS_CONN_ENCRYPTED(req->conn) || req->encrypted,
303                     &req->pcd)) {
304                 exit_server_cleanly("construct_reply: srv_send_smb failed.");
305         }
306         TALLOC_FREE(req);
307 }
308
309 static void api_dcerpc_cmd_read_done(struct async_req *subreq)
310 {
311         struct smb_request *req = talloc_get_type_abort(
312                 subreq->async.priv, struct smb_request);
313         struct dcerpc_cmd_state *state = talloc_get_type_abort(
314                 req->async_priv, struct dcerpc_cmd_state);
315         NTSTATUS status;
316         ssize_t nread;
317         bool is_data_outstanding;
318
319         status = np_read_recv(subreq, &nread, &is_data_outstanding);
320         TALLOC_FREE(subreq);
321
322         if (!NT_STATUS_IS_OK(status)) {
323                 DEBUG(10, ("Could not read from to pipe: %s\n",
324                            nt_errstr(status)));
325                 reply_nterror(req, status);
326
327                 if (!srv_send_smb(smbd_server_fd(), (char *)req->outbuf,
328                                   IS_CONN_ENCRYPTED(req->conn)
329                                   ||req->encrypted, &req->pcd)) {
330                         exit_server_cleanly("construct_reply: srv_send_smb "
331                                             "failed.");
332                 }
333                 TALLOC_FREE(req);
334                 return;
335         }
336
337         send_trans_reply(req->conn, req, NULL, 0, (char *)state->data, nread,
338                          is_data_outstanding);
339         TALLOC_FREE(req);
340 }
341
342 /****************************************************************************
343  WaitNamedPipeHandleState 
344 ****************************************************************************/
345
346 static void api_WNPHS(connection_struct *conn, struct smb_request *req,
347                       struct files_struct *fsp, char *param, int param_len)
348 {
349         if (!param || param_len < 2) {
350                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
351                 return;
352         }
353
354         DEBUG(4,("WaitNamedPipeHandleState priority %x\n",
355                  (int)SVAL(param,0)));
356
357         send_trans_reply(conn, req, NULL, 0, NULL, 0, False);
358 }
359
360
361 /****************************************************************************
362  SetNamedPipeHandleState 
363 ****************************************************************************/
364
365 static void api_SNPHS(connection_struct *conn, struct smb_request *req,
366                       struct files_struct *fsp, char *param, int param_len)
367 {
368         if (!param || param_len < 2) {
369                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
370                 return;
371         }
372
373         DEBUG(4,("SetNamedPipeHandleState to code %x\n", (int)SVAL(param,0)));
374
375         send_trans_reply(conn, req, NULL, 0, NULL, 0, False);
376 }
377
378
379 /****************************************************************************
380  When no reply is generated, indicate unsupported.
381  ****************************************************************************/
382
383 static void api_no_reply(connection_struct *conn, struct smb_request *req)
384 {
385         char rparam[4];
386
387         /* unsupported */
388         SSVAL(rparam,0,NERR_notsupported);
389         SSVAL(rparam,2,0); /* converter word */
390
391         DEBUG(3,("Unsupported API fd command\n"));
392
393         /* now send the reply */
394         send_trans_reply(conn, req, rparam, 4, NULL, 0, False);
395
396         return;
397 }
398
399 /****************************************************************************
400  Handle remote api calls delivered to a named pipe already opened.
401  ****************************************************************************/
402
403 static void api_fd_reply(connection_struct *conn, uint16 vuid,
404                          struct smb_request *req,
405                          uint16 *setup, uint8_t *data, char *params,
406                          int suwcnt, int tdscnt, int tpscnt,
407                          int mdrcnt, int mprcnt)
408 {
409         struct files_struct *fsp;
410         int pnum;
411         int subcommand;
412
413         DEBUG(5,("api_fd_reply\n"));
414
415         /* First find out the name of this file. */
416         if (suwcnt != 2) {
417                 DEBUG(0,("Unexpected named pipe transaction.\n"));
418                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
419                 return;
420         }
421
422         /* Get the file handle and hence the file name. */
423         /* 
424          * NB. The setup array has already been transformed
425          * via SVAL and so is in host byte order.
426          */
427         pnum = ((int)setup[1]) & 0xFFFF;
428         subcommand = ((int)setup[0]) & 0xFFFF;
429
430         fsp = file_fsp(req, pnum);
431
432         if (!fsp_is_np(fsp)) {
433                 if (subcommand == TRANSACT_WAITNAMEDPIPEHANDLESTATE) {
434                         /* Win9x does this call with a unicode pipe name, not a pnum. */
435                         /* Just return success for now... */
436                         DEBUG(3,("Got TRANSACT_WAITNAMEDPIPEHANDLESTATE on text pipe name\n"));
437                         send_trans_reply(conn, req, NULL, 0, NULL, 0, False);
438                         return;
439                 }
440
441                 DEBUG(1,("api_fd_reply: INVALID PIPE HANDLE: %x\n", pnum));
442                 reply_nterror(req, NT_STATUS_INVALID_HANDLE);
443                 return;
444         }
445
446         if (vuid != fsp->vuid) {
447                 DEBUG(1, ("Got pipe request (pnum %x) using invalid VUID %d, "
448                           "expected %d\n", pnum, vuid, fsp->vuid));
449                 reply_nterror(req, NT_STATUS_INVALID_HANDLE);
450                 return;
451         }
452
453         DEBUG(3,("Got API command 0x%x on pipe \"%s\" (pnum %x)\n",
454                  subcommand, fsp->fsp_name, pnum));
455
456         DEBUG(10, ("api_fd_reply: p:%p max_trans_reply: %d\n", fsp, mdrcnt));
457
458         switch (subcommand) {
459         case TRANSACT_DCERPCCMD: {
460                 /* dce/rpc command */
461                 api_dcerpc_cmd(conn, req, fsp, (uint8_t *)data, tdscnt,
462                                mdrcnt);
463                 break;
464         }
465         case TRANSACT_WAITNAMEDPIPEHANDLESTATE:
466                 /* Wait Named Pipe Handle state */
467                 api_WNPHS(conn, req, fsp, params, tpscnt);
468                 break;
469         case TRANSACT_SETNAMEDPIPEHANDLESTATE:
470                 /* Set Named Pipe Handle state */
471                 api_SNPHS(conn, req, fsp, params, tpscnt);
472                 break;
473         default:
474                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
475                 return;
476         }
477 }
478
479 /****************************************************************************
480  Handle named pipe commands.
481 ****************************************************************************/
482
483 static void named_pipe(connection_struct *conn, uint16 vuid,
484                        struct smb_request *req,
485                        const char *name, uint16 *setup,
486                        char *data, char *params,
487                        int suwcnt, int tdscnt,int tpscnt,
488                        int msrcnt, int mdrcnt, int mprcnt)
489 {
490         DEBUG(3,("named pipe command on <%s> name\n", name));
491
492         if (strequal(name,"LANMAN")) {
493                 api_reply(conn, vuid, req,
494                           data, params,
495                           tdscnt, tpscnt,
496                           mdrcnt, mprcnt);
497                 return;
498         }
499
500         if (strequal(name,"WKSSVC") ||
501             strequal(name,"SRVSVC") ||
502             strequal(name,"WINREG") ||
503             strequal(name,"SAMR") ||
504             strequal(name,"LSARPC")) {
505
506                 DEBUG(4,("named pipe command from Win95 (wow!)\n"));
507
508                 api_fd_reply(conn, vuid, req,
509                              setup, (uint8_t *)data, params,
510                              suwcnt, tdscnt, tpscnt,
511                              mdrcnt, mprcnt);
512                 return;
513         }
514
515         if (strlen(name) < 1) {
516                 api_fd_reply(conn, vuid, req,
517                              setup, (uint8_t *)data,
518                              params, suwcnt, tdscnt,
519                              tpscnt, mdrcnt, mprcnt);
520                 return;
521         }
522
523         if (setup)
524                 DEBUG(3,("unknown named pipe: setup 0x%X setup1=%d\n",
525                          (int)setup[0],(int)setup[1]));
526
527         reply_nterror(req, NT_STATUS_NOT_SUPPORTED);
528         return;
529 }
530
531 static void handle_trans(connection_struct *conn, struct smb_request *req,
532                          struct trans_state *state)
533 {
534         char *local_machine_name;
535         int name_offset = 0;
536
537         DEBUG(3,("trans <%s> data=%u params=%u setup=%u\n",
538                  state->name,(unsigned int)state->total_data,(unsigned int)state->total_param,
539                  (unsigned int)state->setup_count));
540
541         /*
542          * WinCE wierdness....
543          */
544
545         local_machine_name = talloc_asprintf(state, "\\%s\\",
546                                              get_local_machine_name());
547
548         if (local_machine_name == NULL) {
549                 reply_nterror(req, NT_STATUS_NO_MEMORY);
550                 return;
551         }
552
553         if (strnequal(state->name, local_machine_name,
554                       strlen(local_machine_name))) {
555                 name_offset = strlen(local_machine_name)-1;
556         }
557
558         if (!strnequal(&state->name[name_offset], "\\PIPE",
559                        strlen("\\PIPE"))) {
560                 reply_nterror(req, NT_STATUS_NOT_SUPPORTED);
561                 return;
562         }
563
564         name_offset += strlen("\\PIPE");
565
566         /* Win9x weirdness.  When talking to a unicode server Win9x
567            only sends \PIPE instead of \PIPE\ */
568
569         if (state->name[name_offset] == '\\')
570                 name_offset++;
571
572         DEBUG(5,("calling named_pipe\n"));
573         named_pipe(conn, state->vuid, req,
574                    state->name+name_offset,
575                    state->setup,state->data,
576                    state->param,
577                    state->setup_count,state->total_data,
578                    state->total_param,
579                    state->max_setup_return,
580                    state->max_data_return,
581                    state->max_param_return);
582
583         if (state->close_on_completion) {
584                 close_cnum(conn,state->vuid);
585                 req->conn = NULL;
586         }
587
588         return;
589 }
590
591 /****************************************************************************
592  Reply to a SMBtrans.
593  ****************************************************************************/
594
595 void reply_trans(struct smb_request *req)
596 {
597         connection_struct *conn = req->conn;
598         unsigned int dsoff;
599         unsigned int dscnt;
600         unsigned int psoff;
601         unsigned int pscnt;
602         struct trans_state *state;
603         NTSTATUS result;
604
605         START_PROFILE(SMBtrans);
606
607         if (req->wct < 14) {
608                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
609                 END_PROFILE(SMBtrans);
610                 return;
611         }
612
613         dsoff = SVAL(req->vwv+12, 0);
614         dscnt = SVAL(req->vwv+11, 0);
615         psoff = SVAL(req->vwv+10, 0);
616         pscnt = SVAL(req->vwv+9, 0);
617
618         result = allow_new_trans(conn->pending_trans, req->mid);
619         if (!NT_STATUS_IS_OK(result)) {
620                 DEBUG(2, ("Got invalid trans request: %s\n",
621                           nt_errstr(result)));
622                 reply_nterror(req, result);
623                 END_PROFILE(SMBtrans);
624                 return;
625         }
626
627         if ((state = TALLOC_P(conn, struct trans_state)) == NULL) {
628                 DEBUG(0, ("talloc failed\n"));
629                 reply_nterror(req, NT_STATUS_NO_MEMORY);
630                 END_PROFILE(SMBtrans);
631                 return;
632         }
633
634         state->cmd = SMBtrans;
635
636         state->mid = req->mid;
637         state->vuid = req->vuid;
638         state->setup_count = CVAL(req->vwv+13, 0);
639         state->setup = NULL;
640         state->total_param = SVAL(req->vwv+0, 0);
641         state->param = NULL;
642         state->total_data = SVAL(req->vwv+1, 0);
643         state->data = NULL;
644         state->max_param_return = SVAL(req->vwv+2, 0);
645         state->max_data_return = SVAL(req->vwv+3, 0);
646         state->max_setup_return = CVAL(req->vwv+4, 0);
647         state->close_on_completion = BITSETW(req->vwv+5, 0);
648         state->one_way = BITSETW(req->vwv+5, 1);
649
650         srvstr_pull_req_talloc(state, req, &state->name, req->buf,
651                                STR_TERMINATE);
652
653         if ((dscnt > state->total_data) || (pscnt > state->total_param) ||
654                         !state->name)
655                 goto bad_param;
656
657         if (state->total_data)  {
658
659                 if (trans_oob(state->total_data, 0, dscnt)
660                     || trans_oob(smb_len(req->inbuf), dsoff, dscnt)) {
661                         goto bad_param;
662                 }
663
664                 /* Can't use talloc here, the core routines do realloc on the
665                  * params and data. Out of paranoia, 100 bytes too many. */
666                 state->data = (char *)SMB_MALLOC(state->total_data+100);
667                 if (state->data == NULL) {
668                         DEBUG(0,("reply_trans: data malloc fail for %u "
669                                  "bytes !\n", (unsigned int)state->total_data));
670                         TALLOC_FREE(state);
671                         reply_nterror(req, NT_STATUS_NO_MEMORY);
672                         END_PROFILE(SMBtrans);
673                         return;
674                 }
675                 /* null-terminate the slack space */
676                 memset(&state->data[state->total_data], 0, 100);
677
678                 memcpy(state->data,smb_base(req->inbuf)+dsoff,dscnt);
679         }
680
681         if (state->total_param) {
682
683                 if (trans_oob(state->total_param, 0, pscnt)
684                     || trans_oob(smb_len(req->inbuf), psoff, pscnt)) {
685                         goto bad_param;
686                 }
687
688                 /* Can't use talloc here, the core routines do realloc on the
689                  * params and data. Out of paranoia, 100 bytes too many */
690                 state->param = (char *)SMB_MALLOC(state->total_param+100);
691                 if (state->param == NULL) {
692                         DEBUG(0,("reply_trans: param malloc fail for %u "
693                                  "bytes !\n", (unsigned int)state->total_param));
694                         SAFE_FREE(state->data);
695                         TALLOC_FREE(state);
696                         reply_nterror(req, NT_STATUS_NO_MEMORY);
697                         END_PROFILE(SMBtrans);
698                         return;
699                 } 
700                 /* null-terminate the slack space */
701                 memset(&state->param[state->total_param], 0, 100);
702
703                 memcpy(state->param,smb_base(req->inbuf)+psoff,pscnt);
704         }
705
706         state->received_data  = dscnt;
707         state->received_param = pscnt;
708
709         if (state->setup_count) {
710                 unsigned int i;
711
712                 /*
713                  * No overflow possible here, state->setup_count is an
714                  * unsigned int, being filled by a single byte from
715                  * CVAL(req->vwv+13, 0) above. The cast in the comparison
716                  * below is not necessary, it's here to clarify things. The
717                  * validity of req->vwv and req->wct has been checked in
718                  * init_smb_request already.
719                  */
720                 if (state->setup_count + 14 > (unsigned int)req->wct) {
721                         goto bad_param;
722                 }
723
724                 if((state->setup = TALLOC_ARRAY(
725                             state, uint16, state->setup_count)) == NULL) {
726                         DEBUG(0,("reply_trans: setup malloc fail for %u "
727                                  "bytes !\n", (unsigned int)
728                                  (state->setup_count * sizeof(uint16))));
729                         SAFE_FREE(state->data);
730                         SAFE_FREE(state->param);
731                         TALLOC_FREE(state);
732                         reply_nterror(req, NT_STATUS_NO_MEMORY);
733                         END_PROFILE(SMBtrans);
734                         return;
735                 } 
736
737                 for (i=0;i<state->setup_count;i++) {
738                         state->setup[i] = SVAL(req->vwv + 14 + i, 0);
739                 }
740         }
741
742         state->received_param = pscnt;
743
744         if ((state->received_param != state->total_param) ||
745             (state->received_data != state->total_data)) {
746                 DLIST_ADD(conn->pending_trans, state);
747
748                 /* We need to send an interim response then receive the rest
749                    of the parameter/data bytes */
750                 reply_outbuf(req, 0, 0);
751                 show_msg((char *)req->outbuf);
752                 END_PROFILE(SMBtrans);
753                 return;
754         }
755
756         handle_trans(conn, req, state);
757
758         SAFE_FREE(state->data);
759         SAFE_FREE(state->param);
760         TALLOC_FREE(state);
761
762         END_PROFILE(SMBtrans);
763         return;
764
765   bad_param:
766
767         DEBUG(0,("reply_trans: invalid trans parameters\n"));
768         SAFE_FREE(state->data);
769         SAFE_FREE(state->param);
770         TALLOC_FREE(state);
771         END_PROFILE(SMBtrans);
772         reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
773         return;
774 }
775
776 /****************************************************************************
777  Reply to a secondary SMBtrans.
778  ****************************************************************************/
779
780 void reply_transs(struct smb_request *req)
781 {
782         connection_struct *conn = req->conn;
783         unsigned int pcnt,poff,dcnt,doff,pdisp,ddisp;
784         struct trans_state *state;
785
786         START_PROFILE(SMBtranss);
787
788         show_msg((char *)req->inbuf);
789
790         if (req->wct < 8) {
791                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
792                 END_PROFILE(SMBtranss);
793                 return;
794         }
795
796         for (state = conn->pending_trans; state != NULL;
797              state = state->next) {
798                 if (state->mid == req->mid) {
799                         break;
800                 }
801         }
802
803         if ((state == NULL) || (state->cmd != SMBtrans)) {
804                 reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
805                 END_PROFILE(SMBtranss);
806                 return;
807         }
808
809         /* Revise total_params and total_data in case they have changed
810          * downwards */
811
812         if (SVAL(req->vwv+0, 0) < state->total_param)
813                 state->total_param = SVAL(req->vwv+0, 0);
814         if (SVAL(req->vwv+1, 0) < state->total_data)
815                 state->total_data = SVAL(req->vwv+1, 0);
816
817         pcnt = SVAL(req->vwv+2, 0);
818         poff = SVAL(req->vwv+3, 0);
819         pdisp = SVAL(req->vwv+4, 0);
820
821         dcnt = SVAL(req->vwv+5, 0);
822         doff = SVAL(req->vwv+6, 0);
823         ddisp = SVAL(req->vwv+7, 0);
824
825         state->received_param += pcnt;
826         state->received_data += dcnt;
827
828         if ((state->received_data > state->total_data) ||
829             (state->received_param > state->total_param))
830                 goto bad_param;
831
832         if (pcnt) {
833                 if (trans_oob(state->total_param, pdisp, pcnt)
834                     || trans_oob(smb_len(req->inbuf), poff, pcnt)) {
835                         goto bad_param;
836                 }
837                 memcpy(state->param+pdisp,smb_base(req->inbuf)+poff,pcnt);
838         }
839
840         if (dcnt) {
841                 if (trans_oob(state->total_data, ddisp, dcnt)
842                     || trans_oob(smb_len(req->inbuf), doff, dcnt)) {
843                         goto bad_param;
844                 }
845                 memcpy(state->data+ddisp, smb_base(req->inbuf)+doff,dcnt);
846         }
847
848         if ((state->received_param < state->total_param) ||
849             (state->received_data < state->total_data)) {
850                 END_PROFILE(SMBtranss);
851                 return;
852         }
853
854         handle_trans(conn, req, state);
855
856         DLIST_REMOVE(conn->pending_trans, state);
857         SAFE_FREE(state->data);
858         SAFE_FREE(state->param);
859         TALLOC_FREE(state);
860
861         END_PROFILE(SMBtranss);
862         return;
863
864   bad_param:
865
866         DEBUG(0,("reply_transs: invalid trans parameters\n"));
867         DLIST_REMOVE(conn->pending_trans, state);
868         SAFE_FREE(state->data);
869         SAFE_FREE(state->param);
870         TALLOC_FREE(state);
871         reply_nterror(req, NT_STATUS_INVALID_PARAMETER);
872         END_PROFILE(SMBtranss);
873         return;
874 }