Forward port the change to talloc_init() to make all talloc contexts
[samba.git] / source3 / rpc_server / srv_pipe_hnd.c
1 /* 
2  *  Unix SMB/CIFS implementation.
3  *  RPC Pipe client / server routines
4  *  Copyright (C) Andrew Tridgell              1992-1998,
5  *  Copyright (C) Luke Kenneth Casson Leighton 1996-1998,
6  *  Copyright (C) Jeremy Allison                                    1999.
7  *  
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 2 of the License, or
11  *  (at your option) any later version.
12  *  
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.
17  *  
18  *  You should have received a copy of the GNU General Public License
19  *  along with this program; if not, write to the Free Software
20  *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
21  */
22
23 #include "includes.h"
24
25 #undef DBGC_CLASS
26 #define DBGC_CLASS DBGC_RPC_SRV
27
28 #define PIPE            "\\PIPE\\"
29 #define PIPELEN         strlen(PIPE)
30
31 static smb_np_struct *chain_p;
32 static int pipes_open;
33
34 /*
35  * Sometimes I can't decide if I hate Windows printer driver
36  * writers more than I hate the Windows spooler service driver
37  * writers. This gets around a combination of bugs in the spooler
38  * and the HP 8500 PCL driver that causes a spooler spin. JRA.
39  *
40  * bumped up from 20 -> 64 after viewing traffic from WordPerfect
41  * 2002 running on NT 4.- SP6
42  * bumped up from 64 -> 256 after viewing traffic from con2prt
43  * for lots of printers on a WinNT 4.x SP6 box.
44  */
45  
46 #ifndef MAX_OPEN_SPOOLSS_PIPES
47 #define MAX_OPEN_SPOOLSS_PIPES 256
48 #endif
49 static int current_spoolss_pipes_open;
50
51 static smb_np_struct *Pipes;
52 static pipes_struct *InternalPipes;
53 static struct bitmap *bmap;
54
55 /* TODO
56  * the following prototypes are declared here to avoid
57  * code being moved about too much for a patch to be
58  * disrupted / less obvious.
59  *
60  * these functions, and associated functions that they
61  * call, should be moved behind a .so module-loading
62  * system _anyway_.  so that's the next step...
63  */
64
65 static ssize_t read_from_internal_pipe(void *np_conn, char *data, size_t n,
66                 BOOL *is_data_outstanding);
67 static ssize_t write_to_internal_pipe(void *np_conn, char *data, size_t n);
68 static BOOL close_internal_rpc_pipe_hnd(void *np_conn);
69 static void *make_internal_rpc_pipe_p(char *pipe_name, 
70                               connection_struct *conn, uint16 vuid);
71
72 /****************************************************************************
73  Pipe iterator functions.
74 ****************************************************************************/
75
76 smb_np_struct *get_first_pipe(void)
77 {
78         return Pipes;
79 }
80
81 smb_np_struct *get_next_pipe(smb_np_struct *p)
82 {
83         return p->next;
84 }
85
86 /****************************************************************************
87  Internal Pipe iterator functions.
88 ****************************************************************************/
89
90 pipes_struct *get_first_internal_pipe(void)
91 {
92         return InternalPipes;
93 }
94
95 pipes_struct *get_next_internal_pipe(pipes_struct *p)
96 {
97         return p->next;
98 }
99
100 /* this must be larger than the sum of the open files and directories */
101 static int pipe_handle_offset;
102
103 /****************************************************************************
104  Set the pipe_handle_offset. Called from smbd/files.c
105 ****************************************************************************/
106
107 void set_pipe_handle_offset(int max_open_files)
108 {
109   if(max_open_files < 0x7000)
110     pipe_handle_offset = 0x7000;
111   else
112     pipe_handle_offset = max_open_files + 10; /* For safety. :-) */
113 }
114
115 /****************************************************************************
116  Reset pipe chain handle number.
117 ****************************************************************************/
118
119 void reset_chain_p(void)
120 {
121         chain_p = NULL;
122 }
123
124 /****************************************************************************
125  Initialise pipe handle states.
126 ****************************************************************************/
127
128 void init_rpc_pipe_hnd(void)
129 {
130         bmap = bitmap_allocate(MAX_OPEN_PIPES);
131         if (!bmap)
132                 exit_server("out of memory in init_rpc_pipe_hnd");
133 }
134
135 /****************************************************************************
136  Initialise an outgoing packet.
137 ****************************************************************************/
138
139 static BOOL pipe_init_outgoing_data(pipes_struct *p)
140 {
141         output_data *o_data = &p->out_data;
142
143         /* Reset the offset counters. */
144         o_data->data_sent_length = 0;
145         o_data->current_pdu_len = 0;
146         o_data->current_pdu_sent = 0;
147
148         memset(o_data->current_pdu, '\0', sizeof(o_data->current_pdu));
149
150         /* Free any memory in the current return data buffer. */
151         prs_mem_free(&o_data->rdata);
152
153         /*
154          * Initialize the outgoing RPC data buffer.
155          * we will use this as the raw data area for replying to rpc requests.
156          */     
157         if(!prs_init(&o_data->rdata, MAX_PDU_FRAG_LEN, p->mem_ctx, MARSHALL)) {
158                 DEBUG(0,("pipe_init_outgoing_data: malloc fail.\n"));
159                 return False;
160         }
161
162         return True;
163 }
164
165 /****************************************************************************
166  Find first available pipe slot.
167 ****************************************************************************/
168
169 smb_np_struct *open_rpc_pipe_p(char *pipe_name, 
170                               connection_struct *conn, uint16 vuid)
171 {
172         int i;
173         smb_np_struct *p, *p_it;
174         static int next_pipe;
175         BOOL is_spoolss_pipe = False;
176
177         DEBUG(4,("Open pipe requested %s (pipes_open=%d)\n",
178                  pipe_name, pipes_open));
179
180         if (strstr(pipe_name, "spoolss"))
181                 is_spoolss_pipe = True;
182  
183         if (is_spoolss_pipe && current_spoolss_pipes_open >= MAX_OPEN_SPOOLSS_PIPES) {
184                 DEBUG(10,("open_rpc_pipe_p: spooler bug workaround. Denying open on pipe %s\n",
185                         pipe_name ));
186                 return NULL;
187         }
188
189         /* not repeating pipe numbers makes it easier to track things in 
190            log files and prevents client bugs where pipe numbers are reused
191            over connection restarts */
192         if (next_pipe == 0)
193                 next_pipe = (sys_getpid() ^ time(NULL)) % MAX_OPEN_PIPES;
194
195         i = bitmap_find(bmap, next_pipe);
196
197         if (i == -1) {
198                 DEBUG(0,("ERROR! Out of pipe structures\n"));
199                 return NULL;
200         }
201
202         next_pipe = (i+1) % MAX_OPEN_PIPES;
203
204         for (p = Pipes; p; p = p->next)
205                 DEBUG(5,("open_rpc_pipe_p: name %s pnum=%x\n", p->name, p->pnum));  
206
207         p = (smb_np_struct *)malloc(sizeof(*p));
208
209         if (!p) {
210                 DEBUG(0,("ERROR! no memory for pipes_struct!\n"));
211                 return NULL;
212         }
213
214         ZERO_STRUCTP(p);
215
216         /* add a dso mechanism instead of this, here */
217
218         p->namedpipe_create = make_internal_rpc_pipe_p;
219         p->namedpipe_read = read_from_internal_pipe;
220         p->namedpipe_write = write_to_internal_pipe;
221         p->namedpipe_close = close_internal_rpc_pipe_hnd;
222
223         p->np_state = p->namedpipe_create(pipe_name, conn, vuid);
224
225         if (p->np_state == NULL) {
226                 DEBUG(0,("open_rpc_pipe_p: make_internal_rpc_pipe_p failed.\n"));
227                 SAFE_FREE(p);
228                 return NULL;
229         }
230
231         DLIST_ADD(Pipes, p);
232
233         /*
234          * Initialize the incoming RPC data buffer with one PDU worth of memory.
235          * We cheat here and say we're marshalling, as we intend to add incoming
236          * data directly into the prs_struct and we want it to auto grow. We will
237          * change the type to UNMARSALLING before processing the stream.
238          */
239
240         bitmap_set(bmap, i);
241         i += pipe_handle_offset;
242
243         pipes_open++;
244
245         p->pnum = i;
246
247         p->open = True;
248         p->device_state = 0;
249         p->priority = 0;
250         p->conn = conn;
251         p->vuid  = vuid;
252
253         p->max_trans_reply = 0;
254         
255         fstrcpy(p->name, pipe_name);
256         
257         DEBUG(4,("Opened pipe %s with handle %x (pipes_open=%d)\n",
258                  pipe_name, i, pipes_open));
259         
260         chain_p = p;
261         
262         /* Iterate over p_it as a temp variable, to display all open pipes */ 
263         for (p_it = Pipes; p_it; p_it = p_it->next)
264                 DEBUG(5,("open pipes: name %s pnum=%x\n", p_it->name, p_it->pnum));  
265
266         return chain_p;
267 }
268
269 /****************************************************************************
270  Make an internal namedpipes structure
271 ****************************************************************************/
272
273 static void *make_internal_rpc_pipe_p(char *pipe_name, 
274                               connection_struct *conn, uint16 vuid)
275 {
276         pipes_struct *p;
277         user_struct *vuser = get_valid_user_struct(vuid);
278
279         DEBUG(4,("Create pipe requested %s\n", pipe_name));
280
281         if (!vuser && vuid != UID_FIELD_INVALID) {
282                 DEBUG(0,("ERROR! vuid %d did not map to a valid vuser struct!\n", vuid));
283                 return NULL;
284         }
285
286         p = (pipes_struct *)malloc(sizeof(*p));
287
288         if (!p)
289         {
290                 DEBUG(0,("ERROR! no memory for pipes_struct!\n"));
291                 return NULL;
292         }
293
294         ZERO_STRUCTP(p);
295
296         if ((p->mem_ctx = talloc_init("pipe %s %p", pipe_name, p)) == NULL) {
297                 DEBUG(0,("open_rpc_pipe_p: talloc_init failed.\n"));
298                 SAFE_FREE(p);
299                 return NULL;
300         }
301
302         if (!init_pipe_handle_list(p, pipe_name)) {
303                 DEBUG(0,("open_rpc_pipe_p: init_pipe_handles failed.\n"));
304                 talloc_destroy(p->mem_ctx);
305                 SAFE_FREE(p);
306                 return NULL;
307         }
308
309         /*
310          * Initialize the incoming RPC data buffer with one PDU worth of memory.
311          * We cheat here and say we're marshalling, as we intend to add incoming
312          * data directly into the prs_struct and we want it to auto grow. We will
313          * change the type to UNMARSALLING before processing the stream.
314          */
315
316         if(!prs_init(&p->in_data.data, MAX_PDU_FRAG_LEN, p->mem_ctx, MARSHALL)) {
317                 DEBUG(0,("open_rpc_pipe_p: malloc fail for in_data struct.\n"));
318                 return NULL;
319         }
320
321         DLIST_ADD(InternalPipes, p);
322
323         p->conn = conn;
324
325         /* Ensure the connection isn't idled whilst this pipe is open. */
326         p->conn->num_files_open++;
327
328         p->vuid  = vuid;
329
330         p->ntlmssp_chal_flags = 0;
331         p->ntlmssp_auth_validated = False;
332         p->ntlmssp_auth_requested = False;
333
334         p->pipe_bound = False;
335         p->fault_state = False;
336         p->endian = RPC_LITTLE_ENDIAN;
337
338         ZERO_STRUCT(p->pipe_user);
339
340         p->pipe_user.uid = (uid_t)-1;
341         p->pipe_user.gid = (gid_t)-1;
342         
343         /* Store the session key and NT_TOKEN */
344         if (vuser) {
345                 memcpy(p->session_key, vuser->session_key, sizeof(p->session_key));
346                 p->pipe_user.nt_user_token = dup_nt_token(vuser->nt_user_token);
347         }
348
349         /*
350          * Initialize the incoming RPC struct.
351          */
352
353         p->in_data.pdu_needed_len = 0;
354         p->in_data.pdu_received_len = 0;
355
356         /*
357          * Initialize the outgoing RPC struct.
358          */
359
360         p->out_data.current_pdu_len = 0;
361         p->out_data.current_pdu_sent = 0;
362         p->out_data.data_sent_length = 0;
363
364         /*
365          * Initialize the outgoing RPC data buffer with no memory.
366          */     
367         prs_init(&p->out_data.rdata, 0, p->mem_ctx, MARSHALL);
368         
369         fstrcpy(p->name, pipe_name);
370         
371         DEBUG(4,("Created internal pipe %s (pipes_open=%d)\n",
372                  pipe_name, pipes_open));
373
374         return (void*)p;
375 }
376
377 /****************************************************************************
378  Sets the fault state on incoming packets.
379 ****************************************************************************/
380
381 static void set_incoming_fault(pipes_struct *p)
382 {
383         prs_mem_free(&p->in_data.data);
384         p->in_data.pdu_needed_len = 0;
385         p->in_data.pdu_received_len = 0;
386         p->fault_state = True;
387         DEBUG(10,("set_incoming_fault: Setting fault state on pipe %s : vuid = 0x%x\n",
388                 p->name, p->vuid ));
389 }
390
391 /****************************************************************************
392  Ensures we have at least RPC_HEADER_LEN amount of data in the incoming buffer.
393 ****************************************************************************/
394
395 static ssize_t fill_rpc_header(pipes_struct *p, char *data, size_t data_to_copy)
396 {
397         size_t len_needed_to_complete_hdr = MIN(data_to_copy, RPC_HEADER_LEN - p->in_data.pdu_received_len);
398
399         DEBUG(10,("fill_rpc_header: data_to_copy = %u, len_needed_to_complete_hdr = %u, receive_len = %u\n",
400                         (unsigned int)data_to_copy, (unsigned int)len_needed_to_complete_hdr,
401                         (unsigned int)p->in_data.pdu_received_len ));
402
403         memcpy((char *)&p->in_data.current_in_pdu[p->in_data.pdu_received_len], data, len_needed_to_complete_hdr);
404         p->in_data.pdu_received_len += len_needed_to_complete_hdr;
405
406         return (ssize_t)len_needed_to_complete_hdr;
407 }
408
409 /****************************************************************************
410  Unmarshalls a new PDU header. Assumes the raw header data is in current_in_pdu.
411 ****************************************************************************/
412
413 static ssize_t unmarshall_rpc_header(pipes_struct *p)
414 {
415         /*
416          * Unmarshall the header to determine the needed length.
417          */
418
419         prs_struct rpc_in;
420
421         if(p->in_data.pdu_received_len != RPC_HEADER_LEN) {
422                 DEBUG(0,("unmarshall_rpc_header: assert on rpc header length failed.\n"));
423                 set_incoming_fault(p);
424                 return -1;
425         }
426
427         prs_init( &rpc_in, 0, p->mem_ctx, UNMARSHALL);
428         prs_set_endian_data( &rpc_in, p->endian);
429
430         prs_give_memory( &rpc_in, (char *)&p->in_data.current_in_pdu[0],
431                                         p->in_data.pdu_received_len, False);
432
433         /*
434          * Unmarshall the header as this will tell us how much
435          * data we need to read to get the complete pdu.
436          * This also sets the endian flag in rpc_in.
437          */
438
439         if(!smb_io_rpc_hdr("", &p->hdr, &rpc_in, 0)) {
440                 DEBUG(0,("unmarshall_rpc_header: failed to unmarshall RPC_HDR.\n"));
441                 set_incoming_fault(p);
442                 prs_mem_free(&rpc_in);
443                 return -1;
444         }
445
446         /*
447          * Validate the RPC header.
448          */
449
450         if(p->hdr.major != 5 && p->hdr.minor != 0) {
451                 DEBUG(0,("unmarshall_rpc_header: invalid major/minor numbers in RPC_HDR.\n"));
452                 set_incoming_fault(p);
453                 prs_mem_free(&rpc_in);
454                 return -1;
455         }
456
457         /*
458          * If there's not data in the incoming buffer this should be the start of a new RPC.
459          */
460
461         if(prs_offset(&p->in_data.data) == 0) {
462
463                 /*
464                  * AS/U doesn't set FIRST flag in a BIND packet it seems.
465                  */
466
467                 if ((p->hdr.pkt_type == RPC_REQUEST) && !(p->hdr.flags & RPC_FLG_FIRST)) {
468                         /*
469                          * Ensure that the FIRST flag is set. If not then we have
470                          * a stream missmatch.
471                          */
472
473                         DEBUG(0,("unmarshall_rpc_header: FIRST flag not set in first PDU !\n"));
474                         set_incoming_fault(p);
475                         prs_mem_free(&rpc_in);
476                         return -1;
477                 }
478
479                 /*
480                  * If this is the first PDU then set the endianness
481                  * flag in the pipe. We will need this when parsing all
482                  * data in this RPC.
483                  */
484
485                 p->endian = rpc_in.bigendian_data;
486
487                 DEBUG(5,("unmarshall_rpc_header: using %sendian RPC\n",
488                                 p->endian == RPC_LITTLE_ENDIAN ? "little-" : "big-" ));
489
490         } else {
491
492                 /*
493                  * If this is *NOT* the first PDU then check the endianness
494                  * flag in the pipe is the same as that in the PDU.
495                  */
496
497                 if (p->endian != rpc_in.bigendian_data) {
498                         DEBUG(0,("unmarshall_rpc_header: FIRST endianness flag (%d) different in next PDU !\n", (int)p->endian));
499                         set_incoming_fault(p);
500                         prs_mem_free(&rpc_in);
501                         return -1;
502                 }
503         }
504
505         /*
506          * Ensure that the pdu length is sane.
507          */
508
509         if((p->hdr.frag_len < RPC_HEADER_LEN) || (p->hdr.frag_len > MAX_PDU_FRAG_LEN)) {
510                 DEBUG(0,("unmarshall_rpc_header: assert on frag length failed.\n"));
511                 set_incoming_fault(p);
512                 prs_mem_free(&rpc_in);
513                 return -1;
514         }
515
516         DEBUG(10,("unmarshall_rpc_header: type = %u, flags = %u\n", (unsigned int)p->hdr.pkt_type,
517                         (unsigned int)p->hdr.flags ));
518
519         /*
520          * Adjust for the header we just ate.
521          */
522         p->in_data.pdu_received_len = 0;
523         p->in_data.pdu_needed_len = (uint32)p->hdr.frag_len - RPC_HEADER_LEN;
524
525         /*
526          * Null the data we just ate.
527          */
528
529         memset((char *)&p->in_data.current_in_pdu[0], '\0', RPC_HEADER_LEN);
530
531         prs_mem_free(&rpc_in);
532
533         return 0; /* No extra data processed. */
534 }
535
536 /****************************************************************************
537  Call this to free any talloc'ed memory. Do this before and after processing
538  a complete PDU.
539 ****************************************************************************/
540
541 void free_pipe_context(pipes_struct *p)
542 {
543         if (p->mem_ctx) {
544                 DEBUG(3,("free_pipe_context: destroying talloc pool of size %u\n", talloc_pool_size(p->mem_ctx) ));
545                 talloc_destroy_pool(p->mem_ctx);
546         } else {
547                 p->mem_ctx = talloc_init("pipe %s %p", p->name, p);
548                 if (p->mem_ctx == NULL)
549                         p->fault_state = True;
550         }
551 }
552
553 /****************************************************************************
554  Processes a request pdu. This will do auth processing if needed, and
555  appends the data into the complete stream if the LAST flag is not set.
556 ****************************************************************************/
557
558 static BOOL process_request_pdu(pipes_struct *p, prs_struct *rpc_in_p)
559 {
560         BOOL auth_verify = ((p->ntlmssp_chal_flags & NTLMSSP_NEGOTIATE_SIGN) != 0);
561         size_t data_len = p->hdr.frag_len - RPC_HEADER_LEN - RPC_HDR_REQ_LEN -
562                                 (auth_verify ? RPC_HDR_AUTH_LEN : 0) - p->hdr.auth_len;
563
564         if(!p->pipe_bound) {
565                 DEBUG(0,("process_request_pdu: rpc request with no bind.\n"));
566                 set_incoming_fault(p);
567                 return False;
568         }
569
570         /*
571          * Check if we need to do authentication processing.
572          * This is only done on requests, not binds.
573          */
574
575         /*
576          * Read the RPC request header.
577          */
578
579         if(!smb_io_rpc_hdr_req("req", &p->hdr_req, rpc_in_p, 0)) {
580                 DEBUG(0,("process_request_pdu: failed to unmarshall RPC_HDR_REQ.\n"));
581                 set_incoming_fault(p);
582                 return False;
583         }
584
585         if(p->ntlmssp_auth_validated && !api_pipe_auth_process(p, rpc_in_p)) {
586                 DEBUG(0,("process_request_pdu: failed to do auth processing.\n"));
587                 set_incoming_fault(p);
588                 return False;
589         }
590
591         if (p->ntlmssp_auth_requested && !p->ntlmssp_auth_validated) {
592
593                 /*
594                  * Authentication _was_ requested and it already failed.
595                  */
596
597                 DEBUG(0,("process_request_pdu: RPC request received on pipe %s where \
598 authentication failed. Denying the request.\n", p->name));
599                 set_incoming_fault(p);
600         return False;
601     }
602
603         /*
604          * Check the data length doesn't go over the 15Mb limit.
605          * increased after observing a bug in the Windows NT 4.0 SP6a
606          * spoolsv.exe when the response to a GETPRINTERDRIVER2 RPC
607          * will not fit in the initial buffer of size 0x1068   --jerry 22/01/2002
608          */
609         
610         if(prs_offset(&p->in_data.data) + data_len > 15*1024*1024) {
611                 DEBUG(0,("process_request_pdu: rpc data buffer too large (%u) + (%u)\n",
612                                 (unsigned int)prs_data_size(&p->in_data.data), (unsigned int)data_len ));
613                 set_incoming_fault(p);
614                 return False;
615         }
616
617         /*
618          * Append the data portion into the buffer and return.
619          */
620
621         {
622                 char *data_from = prs_data_p(rpc_in_p) + prs_offset(rpc_in_p);
623
624                 if(!prs_append_data(&p->in_data.data, data_from, data_len)) {
625                         DEBUG(0,("process_request_pdu: Unable to append data size %u to parse buffer of size %u.\n",
626                                         (unsigned int)data_len, (unsigned int)prs_data_size(&p->in_data.data) ));
627                         set_incoming_fault(p);
628                         return False;
629                 }
630
631         }
632
633         if(p->hdr.flags & RPC_FLG_LAST) {
634                 BOOL ret = False;
635                 /*
636                  * Ok - we finally have a complete RPC stream.
637                  * Call the rpc command to process it.
638                  */
639
640                 /*
641                  * Ensure the internal prs buffer size is *exactly* the same
642                  * size as the current offset.
643                  */
644
645                 if(!prs_set_buffer_size(&p->in_data.data, prs_offset(&p->in_data.data)))
646                 {
647                         DEBUG(0,("process_request_pdu: Call to prs_set_buffer_size failed!\n"));
648                         set_incoming_fault(p);
649                         return False;
650                 }
651
652                 /*
653                  * Set the parse offset to the start of the data and set the
654                  * prs_struct to UNMARSHALL.
655                  */
656
657                 prs_set_offset(&p->in_data.data, 0);
658                 prs_switch_type(&p->in_data.data, UNMARSHALL);
659
660                 /*
661                  * Process the complete data stream here.
662                  */
663
664                 free_pipe_context(p);
665
666                 if(pipe_init_outgoing_data(p))
667                         ret = api_pipe_request(p);
668
669                 free_pipe_context(p);
670
671                 /*
672                  * We have consumed the whole data stream. Set back to
673                  * marshalling and set the offset back to the start of
674                  * the buffer to re-use it (we could also do a prs_mem_free()
675                  * and then re_init on the next start of PDU. Not sure which
676                  * is best here.... JRA.
677                  */
678
679                 prs_switch_type(&p->in_data.data, MARSHALL);
680                 prs_set_offset(&p->in_data.data, 0);
681                 return ret;
682         }
683
684         return True;
685 }
686
687 /****************************************************************************
688  Processes a finished PDU stored in current_in_pdu. The RPC_HEADER has
689  already been parsed and stored in p->hdr.
690 ****************************************************************************/
691
692 static ssize_t process_complete_pdu(pipes_struct *p)
693 {
694         prs_struct rpc_in;
695         size_t data_len = p->in_data.pdu_received_len;
696         char *data_p = (char *)&p->in_data.current_in_pdu[0];
697         BOOL reply = False;
698
699         if(p->fault_state) {
700                 DEBUG(10,("process_complete_pdu: pipe %s in fault state.\n",
701                         p->name ));
702                 set_incoming_fault(p);
703                 setup_fault_pdu(p, NT_STATUS(0x1c010002));
704                 return (ssize_t)data_len;
705         }
706
707         prs_init( &rpc_in, 0, p->mem_ctx, UNMARSHALL);
708
709         /*
710          * Ensure we're using the corrent endianness for both the 
711          * RPC header flags and the raw data we will be reading from.
712          */
713
714         prs_set_endian_data( &rpc_in, p->endian);
715         prs_set_endian_data( &p->in_data.data, p->endian);
716
717         prs_give_memory( &rpc_in, data_p, (uint32)data_len, False);
718
719         DEBUG(10,("process_complete_pdu: processing packet type %u\n",
720                         (unsigned int)p->hdr.pkt_type ));
721
722         switch (p->hdr.pkt_type) {
723                 case RPC_BIND:
724                 case RPC_ALTCONT:
725                         /*
726                          * We assume that a pipe bind is only in one pdu.
727                          */
728                         if(pipe_init_outgoing_data(p))
729                                 reply = api_pipe_bind_req(p, &rpc_in);
730                         break;
731                 case RPC_BINDRESP:
732                         /*
733                          * We assume that a pipe bind_resp is only in one pdu.
734                          */
735                         if(pipe_init_outgoing_data(p))
736                                 reply = api_pipe_bind_auth_resp(p, &rpc_in);
737                         break;
738                 case RPC_REQUEST:
739                         reply = process_request_pdu(p, &rpc_in);
740                         break;
741                 default:
742                         DEBUG(0,("process_complete_pdu: Unknown rpc type = %u received.\n", (unsigned int)p->hdr.pkt_type ));
743                         break;
744         }
745
746         /* Reset to little endian. Probably don't need this but it won't hurt. */
747         prs_set_endian_data( &p->in_data.data, RPC_LITTLE_ENDIAN);
748
749         if (!reply) {
750                 DEBUG(3,("process_complete_pdu: DCE/RPC fault sent on pipe %s\n", p->pipe_srv_name));
751                 set_incoming_fault(p);
752                 setup_fault_pdu(p, NT_STATUS(0x1c010002));
753                 prs_mem_free(&rpc_in);
754         } else {
755                 /*
756                  * Reset the lengths. We're ready for a new pdu.
757                  */
758                 p->in_data.pdu_needed_len = 0;
759                 p->in_data.pdu_received_len = 0;
760         }
761
762         prs_mem_free(&rpc_in);
763         return (ssize_t)data_len;
764 }
765
766 /****************************************************************************
767  Accepts incoming data on an rpc pipe. Processes the data in pdu sized units.
768 ****************************************************************************/
769
770 static ssize_t process_incoming_data(pipes_struct *p, char *data, size_t n)
771 {
772         size_t data_to_copy = MIN(n, MAX_PDU_FRAG_LEN - p->in_data.pdu_received_len);
773
774         DEBUG(10,("process_incoming_data: Start: pdu_received_len = %u, pdu_needed_len = %u, incoming data = %u\n",
775                 (unsigned int)p->in_data.pdu_received_len, (unsigned int)p->in_data.pdu_needed_len,
776                 (unsigned int)n ));
777
778         if(data_to_copy == 0) {
779                 /*
780                  * This is an error - data is being received and there is no
781                  * space in the PDU. Free the received data and go into the fault state.
782                  */
783                 DEBUG(0,("process_incoming_data: No space in incoming pdu buffer. Current size = %u \
784 incoming data size = %u\n", (unsigned int)p->in_data.pdu_received_len, (unsigned int)n ));
785                 set_incoming_fault(p);
786                 return -1;
787         }
788
789         /*
790          * If we have no data already, wait until we get at least a RPC_HEADER_LEN
791          * number of bytes before we can do anything.
792          */
793
794         if((p->in_data.pdu_needed_len == 0) && (p->in_data.pdu_received_len < RPC_HEADER_LEN)) {
795                 /*
796                  * Always return here. If we have more data then the RPC_HEADER
797                  * will be processed the next time around the loop.
798                  */
799                 return fill_rpc_header(p, data, data_to_copy);
800         }
801
802         /*
803          * At this point we know we have at least an RPC_HEADER_LEN amount of data
804          * stored in current_in_pdu.
805          */
806
807         /*
808          * If pdu_needed_len is zero this is a new pdu. 
809          * Unmarshall the header so we know how much more
810          * data we need, then loop again.
811          */
812
813         if(p->in_data.pdu_needed_len == 0)
814                 return unmarshall_rpc_header(p);
815
816         /*
817          * Ok - at this point we have a valid RPC_HEADER in p->hdr.
818          * Keep reading until we have a full pdu.
819          */
820
821         data_to_copy = MIN(data_to_copy, p->in_data.pdu_needed_len);
822
823         /*
824          * Copy as much of the data as we need into the current_in_pdu buffer.
825          */
826
827         memcpy( (char *)&p->in_data.current_in_pdu[p->in_data.pdu_received_len], data, data_to_copy);
828         p->in_data.pdu_received_len += data_to_copy;
829
830         /*
831          * Do we have a complete PDU ?
832          */
833
834         if(p->in_data.pdu_received_len == p->in_data.pdu_needed_len)
835                 return process_complete_pdu(p);
836
837         DEBUG(10,("process_incoming_data: not a complete PDU yet. pdu_received_len = %u, pdu_needed_len = %u\n",
838                 (unsigned int)p->in_data.pdu_received_len, (unsigned int)p->in_data.pdu_needed_len ));
839
840         return (ssize_t)data_to_copy;
841
842 }
843
844 /****************************************************************************
845  Accepts incoming data on an rpc pipe.
846 ****************************************************************************/
847
848 ssize_t write_to_pipe(smb_np_struct *p, char *data, size_t n)
849 {
850         DEBUG(6,("write_to_pipe: %x", p->pnum));
851
852         DEBUG(6,(" name: %s open: %s len: %d\n",
853                  p->name, BOOLSTR(p->open), (int)n));
854
855         dump_data(50, data, n);
856
857         return p->namedpipe_write(p->np_state, data, n);
858 }
859
860 /****************************************************************************
861  Accepts incoming data on an internal rpc pipe.
862 ****************************************************************************/
863
864 static ssize_t write_to_internal_pipe(void *np_conn, char *data, size_t n)
865 {
866         pipes_struct *p = (pipes_struct*)np_conn;
867         size_t data_left = n;
868
869         while(data_left) {
870                 ssize_t data_used;
871
872                 DEBUG(10,("write_to_pipe: data_left = %u\n", (unsigned int)data_left ));
873
874                 data_used = process_incoming_data(p, data, data_left);
875
876                 DEBUG(10,("write_to_pipe: data_used = %d\n", (int)data_used ));
877
878                 if(data_used < 0)
879                         return -1;
880
881                 data_left -= data_used;
882                 data += data_used;
883         }       
884
885         return n;
886 }
887
888 /****************************************************************************
889  Replies to a request to read data from a pipe.
890
891  Headers are interspersed with the data at PDU intervals. By the time
892  this function is called, the start of the data could possibly have been
893  read by an SMBtrans (file_offset != 0).
894
895  Calling create_rpc_reply() here is a hack. The data should already
896  have been prepared into arrays of headers + data stream sections.
897 ****************************************************************************/
898
899 ssize_t read_from_pipe(smb_np_struct *p, char *data, size_t n,
900                 BOOL *is_data_outstanding)
901 {
902         if (!p || !p->open) {
903                 DEBUG(0,("read_from_pipe: pipe not open\n"));
904                 return -1;              
905         }
906
907         DEBUG(6,("read_from_pipe: %x", p->pnum));
908
909         return p->namedpipe_read(p->np_state, data, n, is_data_outstanding);
910 }
911
912 /****************************************************************************
913  Replies to a request to read data from a pipe.
914
915  Headers are interspersed with the data at PDU intervals. By the time
916  this function is called, the start of the data could possibly have been
917  read by an SMBtrans (file_offset != 0).
918
919  Calling create_rpc_reply() here is a hack. The data should already
920  have been prepared into arrays of headers + data stream sections.
921 ****************************************************************************/
922
923 static ssize_t read_from_internal_pipe(void *np_conn, char *data, size_t n,
924                 BOOL *is_data_outstanding)
925 {
926         pipes_struct *p = (pipes_struct*)np_conn;
927         uint32 pdu_remaining = 0;
928         ssize_t data_returned = 0;
929
930         if (!p) {
931                 DEBUG(0,("read_from_pipe: pipe not open\n"));
932                 return -1;              
933         }
934
935         DEBUG(6,(" name: %s len: %u\n", p->name, (unsigned int)n));
936
937         /*
938          * We cannot return more than one PDU length per
939          * read request.
940          */
941
942         /*
943          * This condition should result in the connection being closed.  
944          * Netapp filers seem to set it to 0xffff which results in domain
945          * authentications failing.  Just ignore it so things work.
946          */
947
948         if(n > MAX_PDU_FRAG_LEN) {
949                 DEBUG(5,("read_from_pipe: too large read (%u) requested on \
950 pipe %s. We can only service %d sized reads.\n", (unsigned int)n, p->name, MAX_PDU_FRAG_LEN ));
951         }
952
953         /*
954          * Determine if there is still data to send in the
955          * pipe PDU buffer. Always send this first. Never
956          * send more than is left in the current PDU. The
957          * client should send a new read request for a new
958          * PDU.
959          */
960
961         if((pdu_remaining = p->out_data.current_pdu_len - p->out_data.current_pdu_sent) > 0) {
962                 data_returned = (ssize_t)MIN(n, pdu_remaining);
963
964                 DEBUG(10,("read_from_pipe: %s: current_pdu_len = %u, current_pdu_sent = %u \
965 returning %d bytes.\n", p->name, (unsigned int)p->out_data.current_pdu_len, 
966                         (unsigned int)p->out_data.current_pdu_sent, (int)data_returned));
967
968                 memcpy( data, &p->out_data.current_pdu[p->out_data.current_pdu_sent], (size_t)data_returned);
969                 p->out_data.current_pdu_sent += (uint32)data_returned;
970                 goto out;
971         }
972
973         /*
974          * At this point p->current_pdu_len == p->current_pdu_sent (which
975          * may of course be zero if this is the first return fragment.
976          */
977
978         DEBUG(10,("read_from_pipe: %s: fault_state = %d : data_sent_length \
979 = %u, prs_offset(&p->out_data.rdata) = %u.\n",
980                 p->name, (int)p->fault_state, (unsigned int)p->out_data.data_sent_length, (unsigned int)prs_offset(&p->out_data.rdata) ));
981
982         if(p->out_data.data_sent_length >= prs_offset(&p->out_data.rdata)) {
983                 /*
984                  * We have sent all possible data, return 0.
985                  */
986                 data_returned = 0;
987                 goto out;
988         }
989
990         /*
991          * We need to create a new PDU from the data left in p->rdata.
992          * Create the header/data/footers. This also sets up the fields
993          * p->current_pdu_len, p->current_pdu_sent, p->data_sent_length
994          * and stores the outgoing PDU in p->current_pdu.
995          */
996
997         if(!create_next_pdu(p)) {
998                 DEBUG(0,("read_from_pipe: %s: create_next_pdu failed.\n", p->name));
999                 return -1;
1000         }
1001
1002         data_returned = MIN(n, p->out_data.current_pdu_len);
1003
1004         memcpy( data, p->out_data.current_pdu, (size_t)data_returned);
1005         p->out_data.current_pdu_sent += (uint32)data_returned;
1006
1007   out:
1008
1009         (*is_data_outstanding) = p->out_data.current_pdu_len > n;
1010         return data_returned;
1011 }
1012
1013 /****************************************************************************
1014  Wait device state on a pipe. Exactly what this is for is unknown...
1015 ****************************************************************************/
1016
1017 BOOL wait_rpc_pipe_hnd_state(smb_np_struct *p, uint16 priority)
1018 {
1019         if (p == NULL)
1020                 return False;
1021
1022         if (p->open) {
1023                 DEBUG(3,("wait_rpc_pipe_hnd_state: Setting pipe wait state priority=%x on pipe (name=%s)\n",
1024                          priority, p->name));
1025
1026                 p->priority = priority;
1027                 
1028                 return True;
1029         } 
1030
1031         DEBUG(3,("wait_rpc_pipe_hnd_state: Error setting pipe wait state priority=%x (name=%s)\n",
1032                  priority, p->name));
1033         return False;
1034 }
1035
1036
1037 /****************************************************************************
1038  Set device state on a pipe. Exactly what this is for is unknown...
1039 ****************************************************************************/
1040
1041 BOOL set_rpc_pipe_hnd_state(smb_np_struct *p, uint16 device_state)
1042 {
1043         if (p == NULL)
1044                 return False;
1045
1046         if (p->open) {
1047                 DEBUG(3,("set_rpc_pipe_hnd_state: Setting pipe device state=%x on pipe (name=%s)\n",
1048                          device_state, p->name));
1049
1050                 p->device_state = device_state;
1051                 
1052                 return True;
1053         } 
1054
1055         DEBUG(3,("set_rpc_pipe_hnd_state: Error setting pipe device state=%x (name=%s)\n",
1056                  device_state, p->name));
1057         return False;
1058 }
1059
1060
1061 /****************************************************************************
1062  Close an rpc pipe.
1063 ****************************************************************************/
1064
1065 BOOL close_rpc_pipe_hnd(smb_np_struct *p)
1066 {
1067         if (!p) {
1068                 DEBUG(0,("Invalid pipe in close_rpc_pipe_hnd\n"));
1069                 return False;
1070         }
1071
1072         p->namedpipe_close(p->np_state);
1073
1074         bitmap_clear(bmap, p->pnum - pipe_handle_offset);
1075
1076         pipes_open--;
1077
1078         DEBUG(4,("closed pipe name %s pnum=%x (pipes_open=%d)\n", 
1079                  p->name, p->pnum, pipes_open));  
1080
1081         DLIST_REMOVE(Pipes, p);
1082
1083         ZERO_STRUCTP(p);
1084
1085         SAFE_FREE(p);
1086
1087         return True;
1088 }
1089
1090 /****************************************************************************
1091  Close an rpc pipe.
1092 ****************************************************************************/
1093
1094 static BOOL close_internal_rpc_pipe_hnd(void *np_conn)
1095 {
1096         pipes_struct *p = (pipes_struct *)np_conn;
1097         if (!p) {
1098                 DEBUG(0,("Invalid pipe in close_internal_rpc_pipe_hnd\n"));
1099                 return False;
1100         }
1101
1102         prs_mem_free(&p->out_data.rdata);
1103         prs_mem_free(&p->in_data.data);
1104
1105         if (p->mem_ctx)
1106                 talloc_destroy(p->mem_ctx);
1107
1108         /* Free the handles database. */
1109         close_policy_by_pipe(p);
1110
1111         delete_nt_token(&p->pipe_user.nt_user_token);
1112         SAFE_FREE(p->pipe_user.groups);
1113
1114         DLIST_REMOVE(InternalPipes, p);
1115
1116         p->conn->num_files_open--;
1117
1118         ZERO_STRUCTP(p);
1119
1120         SAFE_FREE(p);
1121         
1122         return True;
1123 }
1124
1125 /****************************************************************************
1126  Find an rpc pipe given a pipe handle in a buffer and an offset.
1127 ****************************************************************************/
1128
1129 smb_np_struct *get_rpc_pipe_p(char *buf, int where)
1130 {
1131         int pnum = SVAL(buf,where);
1132
1133         if (chain_p)
1134                 return chain_p;
1135
1136         return get_rpc_pipe(pnum);
1137 }
1138
1139 /****************************************************************************
1140  Find an rpc pipe given a pipe handle.
1141 ****************************************************************************/
1142
1143 smb_np_struct *get_rpc_pipe(int pnum)
1144 {
1145         smb_np_struct *p;
1146
1147         DEBUG(4,("search for pipe pnum=%x\n", pnum));
1148
1149         for (p=Pipes;p;p=p->next)
1150                 DEBUG(5,("pipe name %s pnum=%x (pipes_open=%d)\n", 
1151                           p->name, p->pnum, pipes_open));  
1152
1153         for (p=Pipes;p;p=p->next) {
1154                 if (p->pnum == pnum) {
1155                         chain_p = p;
1156                         return p;
1157                 }
1158         }
1159
1160         return NULL;
1161 }