s3: Rename new parameter "ldap ref follow" to "ldap follow referral".
[Samba/gebeck_regimport.git] / source3 / rpc_server / srv_pipe_hnd.c
blob0c858c203e168cd49099bdfe48f94a263ecaeda4
1 /*
2 * Unix SMB/CIFS implementation.
3 * RPC Pipe client / server routines
4 * Copyright (C) Andrew Tridgell 1992-1998,
5 * Largely re-written : 2005
6 * Copyright (C) Jeremy Allison 1998 - 2005
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 3 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, see <http://www.gnu.org/licenses/>.
22 #include "includes.h"
23 #include "librpc/gen_ndr/ndr_named_pipe_auth.h"
25 #undef DBGC_CLASS
26 #define DBGC_CLASS DBGC_RPC_SRV
28 static int pipes_open;
30 static pipes_struct *InternalPipes;
32 /* TODO
33 * the following prototypes are declared here to avoid
34 * code being moved about too much for a patch to be
35 * disrupted / less obvious.
37 * these functions, and associated functions that they
38 * call, should be moved behind a .so module-loading
39 * system _anyway_. so that's the next step...
42 static int close_internal_rpc_pipe_hnd(struct pipes_struct *p);
44 /****************************************************************************
45 Internal Pipe iterator functions.
46 ****************************************************************************/
48 pipes_struct *get_first_internal_pipe(void)
50 return InternalPipes;
53 pipes_struct *get_next_internal_pipe(pipes_struct *p)
55 return p->next;
58 /****************************************************************************
59 Initialise an outgoing packet.
60 ****************************************************************************/
62 static bool pipe_init_outgoing_data(pipes_struct *p)
64 output_data *o_data = &p->out_data;
66 /* Reset the offset counters. */
67 o_data->data_sent_length = 0;
68 o_data->current_pdu_sent = 0;
70 prs_mem_free(&o_data->frag);
72 /* Free any memory in the current return data buffer. */
73 prs_mem_free(&o_data->rdata);
76 * Initialize the outgoing RPC data buffer.
77 * we will use this as the raw data area for replying to rpc requests.
78 */
79 if(!prs_init(&o_data->rdata, 128, p->mem_ctx, MARSHALL)) {
80 DEBUG(0,("pipe_init_outgoing_data: malloc fail.\n"));
81 return False;
84 return True;
87 /****************************************************************************
88 Make an internal namedpipes structure
89 ****************************************************************************/
91 static struct pipes_struct *make_internal_rpc_pipe_p(TALLOC_CTX *mem_ctx,
92 const struct ndr_syntax_id *syntax,
93 const char *client_address,
94 struct auth_serversupplied_info *server_info)
96 pipes_struct *p;
98 DEBUG(4,("Create pipe requested %s\n",
99 get_pipe_name_from_iface(syntax)));
101 p = TALLOC_ZERO_P(mem_ctx, struct pipes_struct);
103 if (!p) {
104 DEBUG(0,("ERROR! no memory for pipes_struct!\n"));
105 return NULL;
108 if ((p->mem_ctx = talloc_init("pipe %s %p",
109 get_pipe_name_from_iface(syntax),
110 p)) == NULL) {
111 DEBUG(0,("open_rpc_pipe_p: talloc_init failed.\n"));
112 TALLOC_FREE(p);
113 return NULL;
116 if (!init_pipe_handle_list(p, syntax)) {
117 DEBUG(0,("open_rpc_pipe_p: init_pipe_handles failed.\n"));
118 talloc_destroy(p->mem_ctx);
119 TALLOC_FREE(p);
120 return NULL;
124 * Initialize the incoming RPC data buffer with one PDU worth of memory.
125 * We cheat here and say we're marshalling, as we intend to add incoming
126 * data directly into the prs_struct and we want it to auto grow. We will
127 * change the type to UNMARSALLING before processing the stream.
130 if(!prs_init(&p->in_data.data, 128, p->mem_ctx, MARSHALL)) {
131 DEBUG(0,("open_rpc_pipe_p: malloc fail for in_data struct.\n"));
132 talloc_destroy(p->mem_ctx);
133 close_policy_by_pipe(p);
134 TALLOC_FREE(p);
135 return NULL;
138 p->server_info = copy_serverinfo(p, server_info);
139 if (p->server_info == NULL) {
140 DEBUG(0, ("open_rpc_pipe_p: copy_serverinfo failed\n"));
141 talloc_destroy(p->mem_ctx);
142 close_policy_by_pipe(p);
143 TALLOC_FREE(p);
144 return NULL;
147 DLIST_ADD(InternalPipes, p);
149 memcpy(p->client_address, client_address, sizeof(p->client_address));
151 p->endian = RPC_LITTLE_ENDIAN;
154 * Initialize the outgoing RPC data buffer with no memory.
156 prs_init_empty(&p->out_data.rdata, p->mem_ctx, MARSHALL);
158 p->syntax = *syntax;
160 DEBUG(4,("Created internal pipe %s (pipes_open=%d)\n",
161 get_pipe_name_from_iface(syntax), pipes_open));
163 talloc_set_destructor(p, close_internal_rpc_pipe_hnd);
165 return p;
168 /****************************************************************************
169 Sets the fault state on incoming packets.
170 ****************************************************************************/
172 static void set_incoming_fault(pipes_struct *p)
174 prs_mem_free(&p->in_data.data);
175 p->in_data.pdu_needed_len = 0;
176 p->in_data.pdu_received_len = 0;
177 p->fault_state = True;
178 DEBUG(10, ("set_incoming_fault: Setting fault state on pipe %s\n",
179 get_pipe_name_from_iface(&p->syntax)));
182 /****************************************************************************
183 Ensures we have at least RPC_HEADER_LEN amount of data in the incoming buffer.
184 ****************************************************************************/
186 static ssize_t fill_rpc_header(pipes_struct *p, char *data, size_t data_to_copy)
188 size_t len_needed_to_complete_hdr = MIN(data_to_copy, RPC_HEADER_LEN - p->in_data.pdu_received_len);
190 DEBUG(10,("fill_rpc_header: data_to_copy = %u, len_needed_to_complete_hdr = %u, receive_len = %u\n",
191 (unsigned int)data_to_copy, (unsigned int)len_needed_to_complete_hdr,
192 (unsigned int)p->in_data.pdu_received_len ));
194 if (p->in_data.current_in_pdu == NULL) {
195 p->in_data.current_in_pdu = talloc_array(p, uint8_t,
196 RPC_HEADER_LEN);
198 if (p->in_data.current_in_pdu == NULL) {
199 DEBUG(0, ("talloc failed\n"));
200 return -1;
203 memcpy((char *)&p->in_data.current_in_pdu[p->in_data.pdu_received_len], data, len_needed_to_complete_hdr);
204 p->in_data.pdu_received_len += len_needed_to_complete_hdr;
206 return (ssize_t)len_needed_to_complete_hdr;
209 /****************************************************************************
210 Unmarshalls a new PDU header. Assumes the raw header data is in current_in_pdu.
211 ****************************************************************************/
213 static ssize_t unmarshall_rpc_header(pipes_struct *p)
216 * Unmarshall the header to determine the needed length.
219 prs_struct rpc_in;
221 if(p->in_data.pdu_received_len != RPC_HEADER_LEN) {
222 DEBUG(0,("unmarshall_rpc_header: assert on rpc header length failed.\n"));
223 set_incoming_fault(p);
224 return -1;
227 prs_init_empty( &rpc_in, p->mem_ctx, UNMARSHALL);
228 prs_set_endian_data( &rpc_in, p->endian);
230 prs_give_memory( &rpc_in, (char *)&p->in_data.current_in_pdu[0],
231 p->in_data.pdu_received_len, False);
234 * Unmarshall the header as this will tell us how much
235 * data we need to read to get the complete pdu.
236 * This also sets the endian flag in rpc_in.
239 if(!smb_io_rpc_hdr("", &p->hdr, &rpc_in, 0)) {
240 DEBUG(0,("unmarshall_rpc_header: failed to unmarshall RPC_HDR.\n"));
241 set_incoming_fault(p);
242 prs_mem_free(&rpc_in);
243 return -1;
247 * Validate the RPC header.
250 if(p->hdr.major != 5 && p->hdr.minor != 0) {
251 DEBUG(0,("unmarshall_rpc_header: invalid major/minor numbers in RPC_HDR.\n"));
252 set_incoming_fault(p);
253 prs_mem_free(&rpc_in);
254 return -1;
258 * If there's not data in the incoming buffer this should be the start of a new RPC.
261 if(prs_offset(&p->in_data.data) == 0) {
264 * AS/U doesn't set FIRST flag in a BIND packet it seems.
267 if ((p->hdr.pkt_type == DCERPC_PKT_REQUEST) && !(p->hdr.flags & DCERPC_PFC_FLAG_FIRST)) {
269 * Ensure that the FIRST flag is set. If not then we have
270 * a stream missmatch.
273 DEBUG(0,("unmarshall_rpc_header: FIRST flag not set in first PDU !\n"));
274 set_incoming_fault(p);
275 prs_mem_free(&rpc_in);
276 return -1;
280 * If this is the first PDU then set the endianness
281 * flag in the pipe. We will need this when parsing all
282 * data in this RPC.
285 p->endian = rpc_in.bigendian_data;
287 DEBUG(5,("unmarshall_rpc_header: using %sendian RPC\n",
288 p->endian == RPC_LITTLE_ENDIAN ? "little-" : "big-" ));
290 } else {
293 * If this is *NOT* the first PDU then check the endianness
294 * flag in the pipe is the same as that in the PDU.
297 if (p->endian != rpc_in.bigendian_data) {
298 DEBUG(0,("unmarshall_rpc_header: FIRST endianness flag (%d) different in next PDU !\n", (int)p->endian));
299 set_incoming_fault(p);
300 prs_mem_free(&rpc_in);
301 return -1;
306 * Ensure that the pdu length is sane.
309 if((p->hdr.frag_len < RPC_HEADER_LEN) || (p->hdr.frag_len > RPC_MAX_PDU_FRAG_LEN)) {
310 DEBUG(0,("unmarshall_rpc_header: assert on frag length failed.\n"));
311 set_incoming_fault(p);
312 prs_mem_free(&rpc_in);
313 return -1;
316 DEBUG(10,("unmarshall_rpc_header: type = %u, flags = %u\n", (unsigned int)p->hdr.pkt_type,
317 (unsigned int)p->hdr.flags ));
319 p->in_data.pdu_needed_len = (uint32)p->hdr.frag_len - RPC_HEADER_LEN;
321 prs_mem_free(&rpc_in);
323 p->in_data.current_in_pdu = TALLOC_REALLOC_ARRAY(
324 p, p->in_data.current_in_pdu, uint8_t, p->hdr.frag_len);
325 if (p->in_data.current_in_pdu == NULL) {
326 DEBUG(0, ("talloc failed\n"));
327 set_incoming_fault(p);
328 return -1;
331 return 0; /* No extra data processed. */
334 /****************************************************************************
335 Call this to free any talloc'ed memory. Do this before and after processing
336 a complete PDU.
337 ****************************************************************************/
339 static void free_pipe_context(pipes_struct *p)
341 if (p->mem_ctx) {
342 DEBUG(3,("free_pipe_context: destroying talloc pool of size "
343 "%lu\n", (unsigned long)talloc_total_size(p->mem_ctx) ));
344 talloc_free_children(p->mem_ctx);
345 } else {
346 p->mem_ctx = talloc_init(
347 "pipe %s %p", get_pipe_name_from_iface(&p->syntax), p);
348 if (p->mem_ctx == NULL) {
349 p->fault_state = True;
354 /****************************************************************************
355 Processes a request pdu. This will do auth processing if needed, and
356 appends the data into the complete stream if the LAST flag is not set.
357 ****************************************************************************/
359 static bool process_request_pdu(pipes_struct *p, prs_struct *rpc_in_p)
361 uint32 ss_padding_len = 0;
362 size_t data_len = p->hdr.frag_len - RPC_HEADER_LEN - RPC_HDR_REQ_LEN -
363 (p->hdr.auth_len ? RPC_HDR_AUTH_LEN : 0) - p->hdr.auth_len;
365 if(!p->pipe_bound) {
366 DEBUG(0,("process_request_pdu: rpc request with no bind.\n"));
367 set_incoming_fault(p);
368 return False;
372 * Check if we need to do authentication processing.
373 * This is only done on requests, not binds.
377 * Read the RPC request header.
380 if(!smb_io_rpc_hdr_req("req", &p->hdr_req, rpc_in_p, 0)) {
381 DEBUG(0,("process_request_pdu: failed to unmarshall RPC_HDR_REQ.\n"));
382 set_incoming_fault(p);
383 return False;
386 switch(p->auth.auth_type) {
387 case PIPE_AUTH_TYPE_NONE:
388 break;
390 case PIPE_AUTH_TYPE_SPNEGO_NTLMSSP:
391 case PIPE_AUTH_TYPE_NTLMSSP:
393 NTSTATUS status;
394 if(!api_pipe_ntlmssp_auth_process(p, rpc_in_p, &ss_padding_len, &status)) {
395 DEBUG(0,("process_request_pdu: failed to do auth processing.\n"));
396 DEBUG(0,("process_request_pdu: error was %s.\n", nt_errstr(status) ));
397 set_incoming_fault(p);
398 return False;
400 break;
403 case PIPE_AUTH_TYPE_SCHANNEL:
404 if (!api_pipe_schannel_process(p, rpc_in_p, &ss_padding_len)) {
405 DEBUG(3,("process_request_pdu: failed to do schannel processing.\n"));
406 set_incoming_fault(p);
407 return False;
409 break;
411 default:
412 DEBUG(0,("process_request_pdu: unknown auth type %u set.\n", (unsigned int)p->auth.auth_type ));
413 set_incoming_fault(p);
414 return False;
417 /* Now we've done the sign/seal we can remove any padding data. */
418 if (data_len > ss_padding_len) {
419 data_len -= ss_padding_len;
423 * Check the data length doesn't go over the 15Mb limit.
424 * increased after observing a bug in the Windows NT 4.0 SP6a
425 * spoolsv.exe when the response to a GETPRINTERDRIVER2 RPC
426 * will not fit in the initial buffer of size 0x1068 --jerry 22/01/2002
429 if(prs_offset(&p->in_data.data) + data_len > MAX_RPC_DATA_SIZE) {
430 DEBUG(0,("process_request_pdu: rpc data buffer too large (%u) + (%u)\n",
431 (unsigned int)prs_data_size(&p->in_data.data), (unsigned int)data_len ));
432 set_incoming_fault(p);
433 return False;
437 * Append the data portion into the buffer and return.
440 if(!prs_append_some_prs_data(&p->in_data.data, rpc_in_p, prs_offset(rpc_in_p), data_len)) {
441 DEBUG(0,("process_request_pdu: Unable to append data size %u to parse buffer of size %u.\n",
442 (unsigned int)data_len, (unsigned int)prs_data_size(&p->in_data.data) ));
443 set_incoming_fault(p);
444 return False;
447 if(p->hdr.flags & DCERPC_PFC_FLAG_LAST) {
448 bool ret = False;
450 * Ok - we finally have a complete RPC stream.
451 * Call the rpc command to process it.
455 * Ensure the internal prs buffer size is *exactly* the same
456 * size as the current offset.
459 if(!prs_set_buffer_size(&p->in_data.data, prs_offset(&p->in_data.data))) {
460 DEBUG(0,("process_request_pdu: Call to prs_set_buffer_size failed!\n"));
461 set_incoming_fault(p);
462 return False;
466 * Set the parse offset to the start of the data and set the
467 * prs_struct to UNMARSHALL.
470 prs_set_offset(&p->in_data.data, 0);
471 prs_switch_type(&p->in_data.data, UNMARSHALL);
474 * Process the complete data stream here.
477 free_pipe_context(p);
479 if(pipe_init_outgoing_data(p)) {
480 ret = api_pipe_request(p);
483 free_pipe_context(p);
486 * We have consumed the whole data stream. Set back to
487 * marshalling and set the offset back to the start of
488 * the buffer to re-use it (we could also do a prs_mem_free()
489 * and then re_init on the next start of PDU. Not sure which
490 * is best here.... JRA.
493 prs_switch_type(&p->in_data.data, MARSHALL);
494 prs_set_offset(&p->in_data.data, 0);
495 return ret;
498 return True;
501 /****************************************************************************
502 Processes a finished PDU stored in current_in_pdu. The RPC_HEADER has
503 already been parsed and stored in p->hdr.
504 ****************************************************************************/
506 static void process_complete_pdu(pipes_struct *p)
508 prs_struct rpc_in;
509 size_t data_len = p->in_data.pdu_received_len - RPC_HEADER_LEN;
510 char *data_p = (char *)&p->in_data.current_in_pdu[RPC_HEADER_LEN];
511 bool reply = False;
513 if(p->fault_state) {
514 DEBUG(10,("process_complete_pdu: pipe %s in fault state.\n",
515 get_pipe_name_from_iface(&p->syntax)));
516 set_incoming_fault(p);
517 setup_fault_pdu(p, NT_STATUS(DCERPC_FAULT_OP_RNG_ERROR));
518 return;
521 prs_init_empty( &rpc_in, p->mem_ctx, UNMARSHALL);
524 * Ensure we're using the corrent endianness for both the
525 * RPC header flags and the raw data we will be reading from.
528 prs_set_endian_data( &rpc_in, p->endian);
529 prs_set_endian_data( &p->in_data.data, p->endian);
531 prs_give_memory( &rpc_in, data_p, (uint32)data_len, False);
533 DEBUG(10,("process_complete_pdu: processing packet type %u\n",
534 (unsigned int)p->hdr.pkt_type ));
536 switch (p->hdr.pkt_type) {
537 case DCERPC_PKT_REQUEST:
538 reply = process_request_pdu(p, &rpc_in);
539 break;
541 case DCERPC_PKT_PING: /* CL request - ignore... */
542 DEBUG(0,("process_complete_pdu: Error. Connectionless packet type %u received on pipe %s.\n",
543 (unsigned int)p->hdr.pkt_type,
544 get_pipe_name_from_iface(&p->syntax)));
545 break;
547 case DCERPC_PKT_RESPONSE: /* No responses here. */
548 DEBUG(0,("process_complete_pdu: Error. DCERPC_PKT_RESPONSE received from client on pipe %s.\n",
549 get_pipe_name_from_iface(&p->syntax)));
550 break;
552 case DCERPC_PKT_FAULT:
553 case DCERPC_PKT_WORKING: /* CL request - reply to a ping when a call in process. */
554 case DCERPC_PKT_NOCALL: /* CL - server reply to a ping call. */
555 case DCERPC_PKT_REJECT:
556 case DCERPC_PKT_ACK:
557 case DCERPC_PKT_CL_CANCEL:
558 case DCERPC_PKT_FACK:
559 case DCERPC_PKT_CANCEL_ACK:
560 DEBUG(0,("process_complete_pdu: Error. Connectionless packet type %u received on pipe %s.\n",
561 (unsigned int)p->hdr.pkt_type,
562 get_pipe_name_from_iface(&p->syntax)));
563 break;
565 case DCERPC_PKT_BIND:
567 * We assume that a pipe bind is only in one pdu.
569 if(pipe_init_outgoing_data(p)) {
570 reply = api_pipe_bind_req(p, &rpc_in);
572 break;
574 case DCERPC_PKT_BIND_ACK:
575 case DCERPC_PKT_BIND_NAK:
576 DEBUG(0,("process_complete_pdu: Error. DCERPC_PKT_BINDACK/DCERPC_PKT_BINDNACK packet type %u received on pipe %s.\n",
577 (unsigned int)p->hdr.pkt_type,
578 get_pipe_name_from_iface(&p->syntax)));
579 break;
582 case DCERPC_PKT_ALTER:
584 * We assume that a pipe bind is only in one pdu.
586 if(pipe_init_outgoing_data(p)) {
587 reply = api_pipe_alter_context(p, &rpc_in);
589 break;
591 case DCERPC_PKT_ALTER_RESP:
592 DEBUG(0,("process_complete_pdu: Error. DCERPC_PKT_ALTER_RESP on pipe %s: Should only be server -> client.\n",
593 get_pipe_name_from_iface(&p->syntax)));
594 break;
596 case DCERPC_PKT_AUTH3:
598 * The third packet in an NTLMSSP auth exchange.
600 if(pipe_init_outgoing_data(p)) {
601 reply = api_pipe_bind_auth3(p, &rpc_in);
603 break;
605 case DCERPC_PKT_SHUTDOWN:
606 DEBUG(0,("process_complete_pdu: Error. DCERPC_PKT_SHUTDOWN on pipe %s: Should only be server -> client.\n",
607 get_pipe_name_from_iface(&p->syntax)));
608 break;
610 case DCERPC_PKT_CO_CANCEL:
611 /* For now just free all client data and continue processing. */
612 DEBUG(3,("process_complete_pdu: DCERPC_PKT_CO_CANCEL. Abandoning rpc call.\n"));
613 /* As we never do asynchronous RPC serving, we can never cancel a
614 call (as far as I know). If we ever did we'd have to send a cancel_ack
615 reply. For now, just free all client data and continue processing. */
616 reply = True;
617 break;
618 #if 0
619 /* Enable this if we're doing async rpc. */
620 /* We must check the call-id matches the outstanding callid. */
621 if(pipe_init_outgoing_data(p)) {
622 /* Send a cancel_ack PDU reply. */
623 /* We should probably check the auth-verifier here. */
624 reply = setup_cancel_ack_reply(p, &rpc_in);
626 break;
627 #endif
629 case DCERPC_PKT_ORPHANED:
630 /* We should probably check the auth-verifier here.
631 For now just free all client data and continue processing. */
632 DEBUG(3,("process_complete_pdu: DCERPC_PKT_ORPHANED. Abandoning rpc call.\n"));
633 reply = True;
634 break;
636 default:
637 DEBUG(0,("process_complete_pdu: Unknown rpc type = %u received.\n", (unsigned int)p->hdr.pkt_type ));
638 break;
641 /* Reset to little endian. Probably don't need this but it won't hurt. */
642 prs_set_endian_data( &p->in_data.data, RPC_LITTLE_ENDIAN);
644 if (!reply) {
645 DEBUG(3,("process_complete_pdu: DCE/RPC fault sent on "
646 "pipe %s\n", get_pipe_name_from_iface(&p->syntax)));
647 set_incoming_fault(p);
648 setup_fault_pdu(p, NT_STATUS(DCERPC_FAULT_OP_RNG_ERROR));
649 prs_mem_free(&rpc_in);
650 } else {
652 * Reset the lengths. We're ready for a new pdu.
654 TALLOC_FREE(p->in_data.current_in_pdu);
655 p->in_data.pdu_needed_len = 0;
656 p->in_data.pdu_received_len = 0;
659 prs_mem_free(&rpc_in);
662 /****************************************************************************
663 Accepts incoming data on an rpc pipe. Processes the data in pdu sized units.
664 ****************************************************************************/
666 static ssize_t process_incoming_data(pipes_struct *p, char *data, size_t n)
668 size_t data_to_copy = MIN(n, RPC_MAX_PDU_FRAG_LEN - p->in_data.pdu_received_len);
670 DEBUG(10,("process_incoming_data: Start: pdu_received_len = %u, pdu_needed_len = %u, incoming data = %u\n",
671 (unsigned int)p->in_data.pdu_received_len, (unsigned int)p->in_data.pdu_needed_len,
672 (unsigned int)n ));
674 if(data_to_copy == 0) {
676 * This is an error - data is being received and there is no
677 * space in the PDU. Free the received data and go into the fault state.
679 DEBUG(0,("process_incoming_data: No space in incoming pdu buffer. Current size = %u \
680 incoming data size = %u\n", (unsigned int)p->in_data.pdu_received_len, (unsigned int)n ));
681 set_incoming_fault(p);
682 return -1;
686 * If we have no data already, wait until we get at least a RPC_HEADER_LEN
687 * number of bytes before we can do anything.
690 if((p->in_data.pdu_needed_len == 0) && (p->in_data.pdu_received_len < RPC_HEADER_LEN)) {
692 * Always return here. If we have more data then the RPC_HEADER
693 * will be processed the next time around the loop.
695 return fill_rpc_header(p, data, data_to_copy);
699 * At this point we know we have at least an RPC_HEADER_LEN amount of data
700 * stored in current_in_pdu.
704 * If pdu_needed_len is zero this is a new pdu.
705 * Unmarshall the header so we know how much more
706 * data we need, then loop again.
709 if(p->in_data.pdu_needed_len == 0) {
710 ssize_t rret = unmarshall_rpc_header(p);
711 if (rret == -1 || p->in_data.pdu_needed_len > 0) {
712 return rret;
714 /* If rret == 0 and pdu_needed_len == 0 here we have a PDU that consists
715 of an RPC_HEADER only. This is a DCERPC_PKT_SHUTDOWN, DCERPC_PKT_CO_CANCEL or DCERPC_PKT_ORPHANED
716 pdu type. Deal with this in process_complete_pdu(). */
720 * Ok - at this point we have a valid RPC_HEADER in p->hdr.
721 * Keep reading until we have a full pdu.
724 data_to_copy = MIN(data_to_copy, p->in_data.pdu_needed_len);
727 * Copy as much of the data as we need into the current_in_pdu buffer.
728 * pdu_needed_len becomes zero when we have a complete pdu.
731 memcpy( (char *)&p->in_data.current_in_pdu[p->in_data.pdu_received_len], data, data_to_copy);
732 p->in_data.pdu_received_len += data_to_copy;
733 p->in_data.pdu_needed_len -= data_to_copy;
736 * Do we have a complete PDU ?
737 * (return the number of bytes handled in the call)
740 if(p->in_data.pdu_needed_len == 0) {
741 process_complete_pdu(p);
742 return data_to_copy;
745 DEBUG(10,("process_incoming_data: not a complete PDU yet. pdu_received_len = %u, pdu_needed_len = %u\n",
746 (unsigned int)p->in_data.pdu_received_len, (unsigned int)p->in_data.pdu_needed_len ));
748 return (ssize_t)data_to_copy;
751 /****************************************************************************
752 Accepts incoming data on an internal rpc pipe.
753 ****************************************************************************/
755 static ssize_t write_to_internal_pipe(struct pipes_struct *p, char *data, size_t n)
757 size_t data_left = n;
759 while(data_left) {
760 ssize_t data_used;
762 DEBUG(10,("write_to_pipe: data_left = %u\n", (unsigned int)data_left ));
764 data_used = process_incoming_data(p, data, data_left);
766 DEBUG(10,("write_to_pipe: data_used = %d\n", (int)data_used ));
768 if(data_used < 0) {
769 return -1;
772 data_left -= data_used;
773 data += data_used;
776 return n;
779 /****************************************************************************
780 Replies to a request to read data from a pipe.
782 Headers are interspersed with the data at PDU intervals. By the time
783 this function is called, the start of the data could possibly have been
784 read by an SMBtrans (file_offset != 0).
786 Calling create_rpc_reply() here is a hack. The data should already
787 have been prepared into arrays of headers + data stream sections.
788 ****************************************************************************/
790 static ssize_t read_from_internal_pipe(struct pipes_struct *p, char *data, size_t n,
791 bool *is_data_outstanding)
793 uint32 pdu_remaining = 0;
794 ssize_t data_returned = 0;
796 if (!p) {
797 DEBUG(0,("read_from_pipe: pipe not open\n"));
798 return -1;
801 DEBUG(6,(" name: %s len: %u\n", get_pipe_name_from_iface(&p->syntax),
802 (unsigned int)n));
805 * We cannot return more than one PDU length per
806 * read request.
810 * This condition should result in the connection being closed.
811 * Netapp filers seem to set it to 0xffff which results in domain
812 * authentications failing. Just ignore it so things work.
815 if(n > RPC_MAX_PDU_FRAG_LEN) {
816 DEBUG(5,("read_from_pipe: too large read (%u) requested on "
817 "pipe %s. We can only service %d sized reads.\n",
818 (unsigned int)n, get_pipe_name_from_iface(&p->syntax),
819 RPC_MAX_PDU_FRAG_LEN ));
820 n = RPC_MAX_PDU_FRAG_LEN;
824 * Determine if there is still data to send in the
825 * pipe PDU buffer. Always send this first. Never
826 * send more than is left in the current PDU. The
827 * client should send a new read request for a new
828 * PDU.
831 pdu_remaining = prs_offset(&p->out_data.frag)
832 - p->out_data.current_pdu_sent;
834 if (pdu_remaining > 0) {
835 data_returned = (ssize_t)MIN(n, pdu_remaining);
837 DEBUG(10,("read_from_pipe: %s: current_pdu_len = %u, "
838 "current_pdu_sent = %u returning %d bytes.\n",
839 get_pipe_name_from_iface(&p->syntax),
840 (unsigned int)prs_offset(&p->out_data.frag),
841 (unsigned int)p->out_data.current_pdu_sent,
842 (int)data_returned));
844 memcpy(data,
845 prs_data_p(&p->out_data.frag)
846 + p->out_data.current_pdu_sent,
847 data_returned);
849 p->out_data.current_pdu_sent += (uint32)data_returned;
850 goto out;
854 * At this point p->current_pdu_len == p->current_pdu_sent (which
855 * may of course be zero if this is the first return fragment.
858 DEBUG(10,("read_from_pipe: %s: fault_state = %d : data_sent_length "
859 "= %u, prs_offset(&p->out_data.rdata) = %u.\n",
860 get_pipe_name_from_iface(&p->syntax), (int)p->fault_state,
861 (unsigned int)p->out_data.data_sent_length,
862 (unsigned int)prs_offset(&p->out_data.rdata) ));
864 if(p->out_data.data_sent_length >= prs_offset(&p->out_data.rdata)) {
866 * We have sent all possible data, return 0.
868 data_returned = 0;
869 goto out;
873 * We need to create a new PDU from the data left in p->rdata.
874 * Create the header/data/footers. This also sets up the fields
875 * p->current_pdu_len, p->current_pdu_sent, p->data_sent_length
876 * and stores the outgoing PDU in p->current_pdu.
879 if(!create_next_pdu(p)) {
880 DEBUG(0,("read_from_pipe: %s: create_next_pdu failed.\n",
881 get_pipe_name_from_iface(&p->syntax)));
882 return -1;
885 data_returned = MIN(n, prs_offset(&p->out_data.frag));
887 memcpy( data, prs_data_p(&p->out_data.frag), (size_t)data_returned);
888 p->out_data.current_pdu_sent += (uint32)data_returned;
890 out:
891 (*is_data_outstanding) = prs_offset(&p->out_data.frag) > n;
893 return data_returned;
896 /****************************************************************************
897 Close an rpc pipe.
898 ****************************************************************************/
900 static int close_internal_rpc_pipe_hnd(struct pipes_struct *p)
902 if (!p) {
903 DEBUG(0,("Invalid pipe in close_internal_rpc_pipe_hnd\n"));
904 return False;
907 prs_mem_free(&p->out_data.frag);
908 prs_mem_free(&p->out_data.rdata);
909 prs_mem_free(&p->in_data.data);
911 if (p->auth.auth_data_free_func) {
912 (*p->auth.auth_data_free_func)(&p->auth);
915 TALLOC_FREE(p->mem_ctx);
917 free_pipe_rpc_context( p->contexts );
919 /* Free the handles database. */
920 close_policy_by_pipe(p);
922 DLIST_REMOVE(InternalPipes, p);
924 ZERO_STRUCTP(p);
926 TALLOC_FREE(p);
928 return True;
931 bool fsp_is_np(struct files_struct *fsp)
933 enum FAKE_FILE_TYPE type;
935 if ((fsp == NULL) || (fsp->fake_file_handle == NULL)) {
936 return false;
939 type = fsp->fake_file_handle->type;
941 return ((type == FAKE_FILE_TYPE_NAMED_PIPE)
942 || (type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY));
945 struct np_proxy_state {
946 struct tevent_queue *read_queue;
947 struct tevent_queue *write_queue;
948 int fd;
950 uint8_t *msg;
951 size_t sent;
954 static int np_proxy_state_destructor(struct np_proxy_state *state)
956 if (state->fd != -1) {
957 close(state->fd);
959 return 0;
962 static struct np_proxy_state *make_external_rpc_pipe_p(TALLOC_CTX *mem_ctx,
963 const char *pipe_name,
964 struct auth_serversupplied_info *server_info)
966 struct np_proxy_state *result;
967 struct sockaddr_un addr;
968 char *socket_path;
969 const char *socket_dir;
971 DATA_BLOB req_blob;
972 struct netr_SamInfo3 *info3;
973 struct named_pipe_auth_req req;
974 DATA_BLOB rep_blob;
975 uint8 rep_buf[20];
976 struct named_pipe_auth_rep rep;
977 enum ndr_err_code ndr_err;
978 NTSTATUS status;
979 ssize_t written;
981 result = talloc(mem_ctx, struct np_proxy_state);
982 if (result == NULL) {
983 DEBUG(0, ("talloc failed\n"));
984 return NULL;
987 result->fd = socket(AF_UNIX, SOCK_STREAM, 0);
988 if (result->fd == -1) {
989 DEBUG(10, ("socket(2) failed: %s\n", strerror(errno)));
990 goto fail;
992 talloc_set_destructor(result, np_proxy_state_destructor);
994 ZERO_STRUCT(addr);
995 addr.sun_family = AF_UNIX;
997 socket_dir = lp_parm_const_string(
998 GLOBAL_SECTION_SNUM, "external_rpc_pipe", "socket_dir",
999 get_dyn_NCALRPCDIR());
1000 if (socket_dir == NULL) {
1001 DEBUG(0, ("externan_rpc_pipe:socket_dir not set\n"));
1002 goto fail;
1005 socket_path = talloc_asprintf(talloc_tos(), "%s/np/%s",
1006 socket_dir, pipe_name);
1007 if (socket_path == NULL) {
1008 DEBUG(0, ("talloc_asprintf failed\n"));
1009 goto fail;
1011 strncpy(addr.sun_path, socket_path, sizeof(addr.sun_path));
1012 TALLOC_FREE(socket_path);
1014 become_root();
1015 if (sys_connect(result->fd, (struct sockaddr *)&addr) == -1) {
1016 unbecome_root();
1017 DEBUG(0, ("connect(%s) failed: %s\n", addr.sun_path,
1018 strerror(errno)));
1019 goto fail;
1021 unbecome_root();
1023 info3 = talloc(talloc_tos(), struct netr_SamInfo3);
1024 if (info3 == NULL) {
1025 DEBUG(0, ("talloc failed\n"));
1026 goto fail;
1029 status = serverinfo_to_SamInfo3(server_info, NULL, 0, info3);
1030 if (!NT_STATUS_IS_OK(status)) {
1031 TALLOC_FREE(info3);
1032 DEBUG(0, ("serverinfo_to_SamInfo3 failed: %s\n",
1033 nt_errstr(status)));
1034 goto fail;
1037 req.level = 1;
1038 req.info.info1 = *info3;
1040 ndr_err = ndr_push_struct_blob(
1041 &req_blob, talloc_tos(), NULL, &req,
1042 (ndr_push_flags_fn_t)ndr_push_named_pipe_auth_req);
1044 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1045 DEBUG(10, ("ndr_push_named_pipe_auth_req failed: %s\n",
1046 ndr_errstr(ndr_err)));
1047 goto fail;
1050 DEBUG(10, ("named_pipe_auth_req(client)[%u]\n", (uint32_t)req_blob.length));
1051 dump_data(10, req_blob.data, req_blob.length);
1053 written = write_data(result->fd, (char *)req_blob.data,
1054 req_blob.length);
1055 if (written == -1) {
1056 DEBUG(3, ("Could not write auth req data to RPC server\n"));
1057 goto fail;
1060 status = read_data(result->fd, (char *)rep_buf, sizeof(rep_buf));
1061 if (!NT_STATUS_IS_OK(status)) {
1062 DEBUG(3, ("Could not read auth result\n"));
1063 goto fail;
1066 rep_blob = data_blob_const(rep_buf, sizeof(rep_buf));
1068 DEBUG(10,("name_pipe_auth_rep(client)[%u]\n", (uint32_t)rep_blob.length));
1069 dump_data(10, rep_blob.data, rep_blob.length);
1071 ndr_err = ndr_pull_struct_blob(
1072 &rep_blob, talloc_tos(), NULL, &rep,
1073 (ndr_pull_flags_fn_t)ndr_pull_named_pipe_auth_rep);
1075 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1076 DEBUG(0, ("ndr_pull_named_pipe_auth_rep failed: %s\n",
1077 ndr_errstr(ndr_err)));
1078 goto fail;
1081 if (rep.length != 16) {
1082 DEBUG(0, ("req invalid length: %u != 16\n",
1083 rep.length));
1084 goto fail;
1087 if (strcmp(NAMED_PIPE_AUTH_MAGIC, rep.magic) != 0) {
1088 DEBUG(0, ("req invalid magic: %s != %s\n",
1089 rep.magic, NAMED_PIPE_AUTH_MAGIC));
1090 goto fail;
1093 if (!NT_STATUS_IS_OK(rep.status)) {
1094 DEBUG(0, ("req failed: %s\n",
1095 nt_errstr(rep.status)));
1096 goto fail;
1099 if (rep.level != 1) {
1100 DEBUG(0, ("req invalid level: %u != 1\n",
1101 rep.level));
1102 goto fail;
1105 result->msg = NULL;
1107 result->read_queue = tevent_queue_create(result, "np_read");
1108 if (result->read_queue == NULL) {
1109 goto fail;
1111 result->write_queue = tevent_queue_create(result, "np_write");
1112 if (result->write_queue == NULL) {
1113 goto fail;
1116 return result;
1118 fail:
1119 TALLOC_FREE(result);
1120 return NULL;
1123 NTSTATUS np_open(TALLOC_CTX *mem_ctx, const char *name,
1124 const char *client_address,
1125 struct auth_serversupplied_info *server_info,
1126 struct fake_file_handle **phandle)
1128 const char **proxy_list;
1129 struct fake_file_handle *handle;
1131 proxy_list = lp_parm_string_list(-1, "np", "proxy", NULL);
1133 handle = talloc(mem_ctx, struct fake_file_handle);
1134 if (handle == NULL) {
1135 return NT_STATUS_NO_MEMORY;
1138 if ((proxy_list != NULL) && str_list_check_ci(proxy_list, name)) {
1139 struct np_proxy_state *p;
1141 p = make_external_rpc_pipe_p(handle, name, server_info);
1143 handle->type = FAKE_FILE_TYPE_NAMED_PIPE_PROXY;
1144 handle->private_data = p;
1145 } else {
1146 struct pipes_struct *p;
1147 struct ndr_syntax_id syntax;
1149 if (!is_known_pipename(name, &syntax)) {
1150 TALLOC_FREE(handle);
1151 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
1154 p = make_internal_rpc_pipe_p(handle, &syntax, client_address,
1155 server_info);
1157 handle->type = FAKE_FILE_TYPE_NAMED_PIPE;
1158 handle->private_data = p;
1161 if (handle->private_data == NULL) {
1162 TALLOC_FREE(handle);
1163 return NT_STATUS_PIPE_NOT_AVAILABLE;
1166 *phandle = handle;
1168 return NT_STATUS_OK;
1171 struct np_write_state {
1172 struct event_context *ev;
1173 struct np_proxy_state *p;
1174 struct iovec iov;
1175 ssize_t nwritten;
1178 static void np_write_done(struct tevent_req *subreq);
1180 struct tevent_req *np_write_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
1181 struct fake_file_handle *handle,
1182 const uint8_t *data, size_t len)
1184 struct tevent_req *req;
1185 struct np_write_state *state;
1186 NTSTATUS status;
1188 DEBUG(6, ("np_write_send: len: %d\n", (int)len));
1189 dump_data(50, data, len);
1191 req = tevent_req_create(mem_ctx, &state, struct np_write_state);
1192 if (req == NULL) {
1193 return NULL;
1196 if (len == 0) {
1197 state->nwritten = 0;
1198 status = NT_STATUS_OK;
1199 goto post_status;
1202 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE) {
1203 struct pipes_struct *p = talloc_get_type_abort(
1204 handle->private_data, struct pipes_struct);
1206 state->nwritten = write_to_internal_pipe(p, (char *)data, len);
1208 status = (state->nwritten >= 0)
1209 ? NT_STATUS_OK : NT_STATUS_UNEXPECTED_IO_ERROR;
1210 goto post_status;
1213 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
1214 struct np_proxy_state *p = talloc_get_type_abort(
1215 handle->private_data, struct np_proxy_state);
1216 struct tevent_req *subreq;
1218 state->ev = ev;
1219 state->p = p;
1220 state->iov.iov_base = CONST_DISCARD(void *, data);
1221 state->iov.iov_len = len;
1223 subreq = writev_send(state, ev, p->write_queue, p->fd,
1224 false, &state->iov, 1);
1225 if (subreq == NULL) {
1226 goto fail;
1228 tevent_req_set_callback(subreq, np_write_done, req);
1229 return req;
1232 status = NT_STATUS_INVALID_HANDLE;
1233 post_status:
1234 if (NT_STATUS_IS_OK(status)) {
1235 tevent_req_done(req);
1236 } else {
1237 tevent_req_nterror(req, status);
1239 return tevent_req_post(req, ev);
1240 fail:
1241 TALLOC_FREE(req);
1242 return NULL;
1245 static void np_write_done(struct tevent_req *subreq)
1247 struct tevent_req *req = tevent_req_callback_data(
1248 subreq, struct tevent_req);
1249 struct np_write_state *state = tevent_req_data(
1250 req, struct np_write_state);
1251 ssize_t received;
1252 int err;
1254 received = writev_recv(subreq, &err);
1255 if (received < 0) {
1256 tevent_req_nterror(req, map_nt_error_from_unix(err));
1257 return;
1259 state->nwritten = received;
1260 tevent_req_done(req);
1263 NTSTATUS np_write_recv(struct tevent_req *req, ssize_t *pnwritten)
1265 struct np_write_state *state = tevent_req_data(
1266 req, struct np_write_state);
1267 NTSTATUS status;
1269 if (tevent_req_is_nterror(req, &status)) {
1270 return status;
1272 *pnwritten = state->nwritten;
1273 return NT_STATUS_OK;
1276 static ssize_t rpc_frag_more_fn(uint8_t *buf, size_t buflen, void *priv)
1278 prs_struct hdr_prs;
1279 struct rpc_hdr_info hdr;
1280 bool ret;
1282 if (buflen > RPC_HEADER_LEN) {
1283 return 0;
1285 prs_init_empty(&hdr_prs, talloc_tos(), UNMARSHALL);
1286 prs_give_memory(&hdr_prs, (char *)buf, RPC_HEADER_LEN, false);
1287 ret = smb_io_rpc_hdr("", &hdr, &hdr_prs, 0);
1288 prs_mem_free(&hdr_prs);
1290 if (!ret) {
1291 return -1;
1294 return (hdr.frag_len - RPC_HEADER_LEN);
1297 struct np_read_state {
1298 struct event_context *ev;
1299 struct np_proxy_state *p;
1300 uint8_t *data;
1301 size_t len;
1303 size_t nread;
1304 bool is_data_outstanding;
1307 static void np_read_trigger(struct tevent_req *req, void *private_data);
1308 static void np_read_done(struct tevent_req *subreq);
1310 struct tevent_req *np_read_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
1311 struct fake_file_handle *handle,
1312 uint8_t *data, size_t len)
1314 struct tevent_req *req;
1315 struct np_read_state *state;
1316 NTSTATUS status;
1318 req = tevent_req_create(mem_ctx, &state, struct np_read_state);
1319 if (req == NULL) {
1320 return NULL;
1323 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE) {
1324 struct pipes_struct *p = talloc_get_type_abort(
1325 handle->private_data, struct pipes_struct);
1327 state->nread = read_from_internal_pipe(
1328 p, (char *)data, len, &state->is_data_outstanding);
1330 status = (state->nread >= 0)
1331 ? NT_STATUS_OK : NT_STATUS_UNEXPECTED_IO_ERROR;
1332 goto post_status;
1335 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
1336 struct np_proxy_state *p = talloc_get_type_abort(
1337 handle->private_data, struct np_proxy_state);
1339 if (p->msg != NULL) {
1340 size_t thistime;
1342 thistime = MIN(talloc_get_size(p->msg) - p->sent,
1343 len);
1345 memcpy(data, p->msg+p->sent, thistime);
1346 state->nread = thistime;
1347 p->sent += thistime;
1349 if (p->sent < talloc_get_size(p->msg)) {
1350 state->is_data_outstanding = true;
1351 } else {
1352 state->is_data_outstanding = false;
1353 TALLOC_FREE(p->msg);
1355 status = NT_STATUS_OK;
1356 goto post_status;
1359 state->ev = ev;
1360 state->p = p;
1361 state->data = data;
1362 state->len = len;
1364 if (!tevent_queue_add(p->read_queue, ev, req, np_read_trigger,
1365 NULL)) {
1366 goto fail;
1368 return req;
1371 status = NT_STATUS_INVALID_HANDLE;
1372 post_status:
1373 if (NT_STATUS_IS_OK(status)) {
1374 tevent_req_done(req);
1375 } else {
1376 tevent_req_nterror(req, status);
1378 return tevent_req_post(req, ev);
1379 fail:
1380 TALLOC_FREE(req);
1381 return NULL;
1384 static void np_read_trigger(struct tevent_req *req, void *private_data)
1386 struct np_read_state *state = tevent_req_data(
1387 req, struct np_read_state);
1388 struct tevent_req *subreq;
1390 subreq = read_packet_send(state, state->ev, state->p->fd,
1391 RPC_HEADER_LEN, rpc_frag_more_fn, NULL);
1392 if (tevent_req_nomem(subreq, req)) {
1393 return;
1395 tevent_req_set_callback(subreq, np_read_done, req);
1398 static void np_read_done(struct tevent_req *subreq)
1400 struct tevent_req *req = tevent_req_callback_data(
1401 subreq, struct tevent_req);
1402 struct np_read_state *state = tevent_req_data(
1403 req, struct np_read_state);
1404 ssize_t received;
1405 size_t thistime;
1406 int err;
1408 received = read_packet_recv(subreq, state->p, &state->p->msg, &err);
1409 TALLOC_FREE(subreq);
1410 if (received == -1) {
1411 tevent_req_nterror(req, map_nt_error_from_unix(err));
1412 return;
1415 thistime = MIN(received, state->len);
1417 memcpy(state->data, state->p->msg, thistime);
1418 state->p->sent = thistime;
1419 state->nread = thistime;
1421 if (state->p->sent < received) {
1422 state->is_data_outstanding = true;
1423 } else {
1424 TALLOC_FREE(state->p->msg);
1425 state->is_data_outstanding = false;
1428 tevent_req_done(req);
1429 return;
1432 NTSTATUS np_read_recv(struct tevent_req *req, ssize_t *nread,
1433 bool *is_data_outstanding)
1435 struct np_read_state *state = tevent_req_data(
1436 req, struct np_read_state);
1437 NTSTATUS status;
1439 if (tevent_req_is_nterror(req, &status)) {
1440 return status;
1442 *nread = state->nread;
1443 *is_data_outstanding = state->is_data_outstanding;
1444 return NT_STATUS_OK;
1448 * Create a new RPC client context which uses a local dispatch function.
1450 NTSTATUS rpc_pipe_open_internal(TALLOC_CTX *mem_ctx,
1451 const struct ndr_syntax_id *abstract_syntax,
1452 NTSTATUS (*dispatch) (struct rpc_pipe_client *cli,
1453 TALLOC_CTX *mem_ctx,
1454 const struct ndr_interface_table *table,
1455 uint32_t opnum, void *r),
1456 struct auth_serversupplied_info *serversupplied_info,
1457 struct rpc_pipe_client **presult)
1459 struct rpc_pipe_client *result;
1461 result = TALLOC_ZERO_P(mem_ctx, struct rpc_pipe_client);
1462 if (result == NULL) {
1463 return NT_STATUS_NO_MEMORY;
1466 result->abstract_syntax = *abstract_syntax;
1467 result->transfer_syntax = ndr_transfer_syntax;
1468 result->dispatch = dispatch;
1470 result->pipes_struct = make_internal_rpc_pipe_p(
1471 result, abstract_syntax, "", serversupplied_info);
1472 if (result->pipes_struct == NULL) {
1473 TALLOC_FREE(result);
1474 return NT_STATUS_NO_MEMORY;
1477 result->max_xmit_frag = -1;
1478 result->max_recv_frag = -1;
1480 *presult = result;
1481 return NT_STATUS_OK;
1484 /*******************************************************************
1485 gets a domain user's groups from their already-calculated NT_USER_TOKEN
1486 ********************************************************************/
1488 static NTSTATUS nt_token_to_group_list(TALLOC_CTX *mem_ctx,
1489 const DOM_SID *domain_sid,
1490 size_t num_sids,
1491 const DOM_SID *sids,
1492 int *numgroups,
1493 struct samr_RidWithAttribute **pgids)
1495 int i;
1497 *numgroups=0;
1498 *pgids = NULL;
1500 for (i=0; i<num_sids; i++) {
1501 struct samr_RidWithAttribute gid;
1502 if (!sid_peek_check_rid(domain_sid, &sids[i], &gid.rid)) {
1503 continue;
1505 gid.attributes = (SE_GROUP_MANDATORY|SE_GROUP_ENABLED_BY_DEFAULT|
1506 SE_GROUP_ENABLED);
1507 ADD_TO_ARRAY(mem_ctx, struct samr_RidWithAttribute,
1508 gid, pgids, numgroups);
1509 if (*pgids == NULL) {
1510 return NT_STATUS_NO_MEMORY;
1513 return NT_STATUS_OK;
1516 /****************************************************************************
1517 inits a netr_SamBaseInfo structure from an auth_serversupplied_info.
1518 *****************************************************************************/
1520 static NTSTATUS serverinfo_to_SamInfo_base(TALLOC_CTX *mem_ctx,
1521 struct auth_serversupplied_info *server_info,
1522 uint8_t *pipe_session_key,
1523 size_t pipe_session_key_len,
1524 struct netr_SamBaseInfo *base)
1526 struct samu *sampw;
1527 struct samr_RidWithAttribute *gids = NULL;
1528 const DOM_SID *user_sid = NULL;
1529 const DOM_SID *group_sid = NULL;
1530 DOM_SID domain_sid;
1531 uint32 user_rid, group_rid;
1532 NTSTATUS status;
1534 int num_gids = 0;
1535 const char *my_name;
1537 struct netr_UserSessionKey user_session_key;
1538 struct netr_LMSessionKey lm_session_key;
1540 NTTIME last_logon, last_logoff, acct_expiry, last_password_change;
1541 NTTIME allow_password_change, force_password_change;
1542 struct samr_RidWithAttributeArray groups;
1543 int i;
1544 struct dom_sid2 *sid = NULL;
1546 ZERO_STRUCT(user_session_key);
1547 ZERO_STRUCT(lm_session_key);
1549 sampw = server_info->sam_account;
1551 user_sid = pdb_get_user_sid(sampw);
1552 group_sid = pdb_get_group_sid(sampw);
1554 if (pipe_session_key && pipe_session_key_len != 16) {
1555 DEBUG(0,("serverinfo_to_SamInfo3: invalid "
1556 "pipe_session_key_len[%zu] != 16\n",
1557 pipe_session_key_len));
1558 return NT_STATUS_INTERNAL_ERROR;
1561 if ((user_sid == NULL) || (group_sid == NULL)) {
1562 DEBUG(1, ("_netr_LogonSamLogon: User without group or user SID\n"));
1563 return NT_STATUS_UNSUCCESSFUL;
1566 sid_copy(&domain_sid, user_sid);
1567 sid_split_rid(&domain_sid, &user_rid);
1569 sid = sid_dup_talloc(mem_ctx, &domain_sid);
1570 if (!sid) {
1571 return NT_STATUS_NO_MEMORY;
1574 if (!sid_peek_check_rid(&domain_sid, group_sid, &group_rid)) {
1575 DEBUG(1, ("_netr_LogonSamLogon: user %s\\%s has user sid "
1576 "%s\n but group sid %s.\n"
1577 "The conflicting domain portions are not "
1578 "supported for NETLOGON calls\n",
1579 pdb_get_domain(sampw),
1580 pdb_get_username(sampw),
1581 sid_string_dbg(user_sid),
1582 sid_string_dbg(group_sid)));
1583 return NT_STATUS_UNSUCCESSFUL;
1586 if(server_info->login_server) {
1587 my_name = server_info->login_server;
1588 } else {
1589 my_name = global_myname();
1592 status = nt_token_to_group_list(mem_ctx, &domain_sid,
1593 server_info->num_sids,
1594 server_info->sids,
1595 &num_gids, &gids);
1597 if (!NT_STATUS_IS_OK(status)) {
1598 return status;
1601 if (server_info->user_session_key.length) {
1602 memcpy(user_session_key.key,
1603 server_info->user_session_key.data,
1604 MIN(sizeof(user_session_key.key),
1605 server_info->user_session_key.length));
1606 if (pipe_session_key) {
1607 arcfour_crypt(user_session_key.key, pipe_session_key, 16);
1610 if (server_info->lm_session_key.length) {
1611 memcpy(lm_session_key.key,
1612 server_info->lm_session_key.data,
1613 MIN(sizeof(lm_session_key.key),
1614 server_info->lm_session_key.length));
1615 if (pipe_session_key) {
1616 arcfour_crypt(lm_session_key.key, pipe_session_key, 8);
1620 groups.count = num_gids;
1621 groups.rids = TALLOC_ARRAY(mem_ctx, struct samr_RidWithAttribute, groups.count);
1622 if (!groups.rids) {
1623 return NT_STATUS_NO_MEMORY;
1626 for (i=0; i < groups.count; i++) {
1627 groups.rids[i].rid = gids[i].rid;
1628 groups.rids[i].attributes = gids[i].attributes;
1631 unix_to_nt_time(&last_logon, pdb_get_logon_time(sampw));
1632 unix_to_nt_time(&last_logoff, get_time_t_max());
1633 unix_to_nt_time(&acct_expiry, get_time_t_max());
1634 unix_to_nt_time(&last_password_change, pdb_get_pass_last_set_time(sampw));
1635 unix_to_nt_time(&allow_password_change, pdb_get_pass_can_change_time(sampw));
1636 unix_to_nt_time(&force_password_change, pdb_get_pass_must_change_time(sampw));
1638 base->last_logon = last_logon;
1639 base->last_logoff = last_logoff;
1640 base->acct_expiry = acct_expiry;
1641 base->last_password_change = last_password_change;
1642 base->allow_password_change = allow_password_change;
1643 base->force_password_change = force_password_change;
1644 base->account_name.string = talloc_strdup(mem_ctx, pdb_get_username(sampw));
1645 base->full_name.string = talloc_strdup(mem_ctx, pdb_get_fullname(sampw));
1646 base->logon_script.string = talloc_strdup(mem_ctx, pdb_get_logon_script(sampw));
1647 base->profile_path.string = talloc_strdup(mem_ctx, pdb_get_profile_path(sampw));
1648 base->home_directory.string = talloc_strdup(mem_ctx, pdb_get_homedir(sampw));
1649 base->home_drive.string = talloc_strdup(mem_ctx, pdb_get_dir_drive(sampw));
1650 base->logon_count = 0; /* ?? */
1651 base->bad_password_count = 0; /* ?? */
1652 base->rid = user_rid;
1653 base->primary_gid = group_rid;
1654 base->groups = groups;
1655 base->user_flags = NETLOGON_EXTRA_SIDS;
1656 base->key = user_session_key;
1657 base->logon_server.string = talloc_strdup(mem_ctx, my_name);
1658 base->domain.string = talloc_strdup(mem_ctx, pdb_get_domain(sampw));
1659 base->domain_sid = sid;
1660 base->LMSessKey = lm_session_key;
1661 base->acct_flags = pdb_get_acct_ctrl(sampw);
1663 ZERO_STRUCT(user_session_key);
1664 ZERO_STRUCT(lm_session_key);
1666 return NT_STATUS_OK;
1669 /****************************************************************************
1670 inits a netr_SamInfo2 structure from an auth_serversupplied_info. sam2 must
1671 already be initialized and is used as the talloc parent for its members.
1672 *****************************************************************************/
1674 NTSTATUS serverinfo_to_SamInfo2(struct auth_serversupplied_info *server_info,
1675 uint8_t *pipe_session_key,
1676 size_t pipe_session_key_len,
1677 struct netr_SamInfo2 *sam2)
1679 NTSTATUS status;
1681 status = serverinfo_to_SamInfo_base(sam2,
1682 server_info,
1683 pipe_session_key,
1684 pipe_session_key_len,
1685 &sam2->base);
1686 if (!NT_STATUS_IS_OK(status)) {
1687 return status;
1690 return NT_STATUS_OK;
1693 /****************************************************************************
1694 inits a netr_SamInfo3 structure from an auth_serversupplied_info. sam3 must
1695 already be initialized and is used as the talloc parent for its members.
1696 *****************************************************************************/
1698 NTSTATUS serverinfo_to_SamInfo3(struct auth_serversupplied_info *server_info,
1699 uint8_t *pipe_session_key,
1700 size_t pipe_session_key_len,
1701 struct netr_SamInfo3 *sam3)
1703 NTSTATUS status;
1705 status = serverinfo_to_SamInfo_base(sam3,
1706 server_info,
1707 pipe_session_key,
1708 pipe_session_key_len,
1709 &sam3->base);
1710 if (!NT_STATUS_IS_OK(status)) {
1711 return status;
1714 sam3->sidcount = 0;
1715 sam3->sids = NULL;
1717 return NT_STATUS_OK;
1720 /****************************************************************************
1721 inits a netr_SamInfo6 structure from an auth_serversupplied_info. sam6 must
1722 already be initialized and is used as the talloc parent for its members.
1723 *****************************************************************************/
1725 NTSTATUS serverinfo_to_SamInfo6(struct auth_serversupplied_info *server_info,
1726 uint8_t *pipe_session_key,
1727 size_t pipe_session_key_len,
1728 struct netr_SamInfo6 *sam6)
1730 NTSTATUS status;
1731 struct pdb_domain_info *dominfo;
1733 if ((pdb_capabilities() & PDB_CAP_ADS) == 0) {
1734 DEBUG(10,("Not adding validation info level 6 "
1735 "without ADS passdb backend\n"));
1736 return NT_STATUS_INVALID_INFO_CLASS;
1739 dominfo = pdb_get_domain_info(sam6);
1740 if (dominfo == NULL) {
1741 return NT_STATUS_NO_MEMORY;
1744 status = serverinfo_to_SamInfo_base(sam6,
1745 server_info,
1746 pipe_session_key,
1747 pipe_session_key_len,
1748 &sam6->base);
1749 if (!NT_STATUS_IS_OK(status)) {
1750 return status;
1753 sam6->sidcount = 0;
1754 sam6->sids = NULL;
1756 sam6->forest.string = talloc_strdup(sam6, dominfo->dns_forest);
1757 if (sam6->forest.string == NULL) {
1758 return NT_STATUS_NO_MEMORY;
1761 sam6->principle.string = talloc_asprintf(sam6, "%s@%s",
1762 pdb_get_username(server_info->sam_account),
1763 dominfo->dns_domain);
1764 if (sam6->principle.string == NULL) {
1765 return NT_STATUS_NO_MEMORY;
1768 return NT_STATUS_OK;