?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Add queueing to np_read_state, simulate message-type named pipes.
[Samba.git] / source3 / rpc_server / srv_pipe_hnd.c
blob653bab5e5abf36a9e34ede6e2e0aca5389aca3c9
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.
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, see <http://www.gnu.org/licenses/>.
20 */
21
22 #include "includes.h"
23 #include "librpc/gen_ndr/ndr_named_pipe_auth.h"
24
25 #undef DBGC_CLASS
26 #define DBGC_CLASS DBGC_RPC_SRV
27
28 static int pipes_open;
29
30 static pipes_struct *InternalPipes;
31
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.
36 *
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...
40 */
41
42 static int close_internal_rpc_pipe_hnd(struct pipes_struct *p);
43
44 /****************************************************************************
45 Internal Pipe iterator functions.
46 ****************************************************************************/
47
48 pipes_struct *get_first_internal_pipe(void)
49 {
50 return InternalPipes;
51 }
52
53 pipes_struct *get_next_internal_pipe(pipes_struct *p)
54 {
55 return p->next;
56 }
57
58 /****************************************************************************
59 Initialise an outgoing packet.
60 ****************************************************************************/
61
62 static bool pipe_init_outgoing_data(pipes_struct *p)
63 {
64 output_data *o_data = &p->out_data;
65
66 /* Reset the offset counters. */
67 o_data->data_sent_length = 0;
68 o_data->current_pdu_sent = 0;
69
70 prs_mem_free(&o_data->frag);
71
72 /* Free any memory in the current return data buffer. */
73 prs_mem_free(&o_data->rdata);
74
75 /*
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;
82 }
83
84 return True;
85 }
86
87 /****************************************************************************
88 Make an internal namedpipes structure
89 ****************************************************************************/
90
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)
95 {
96 pipes_struct *p;
97
98 DEBUG(4,("Create pipe requested %s\n",
99 get_pipe_name_from_iface(syntax)));
100
101 p = TALLOC_ZERO_P(mem_ctx, struct pipes_struct);
102
103 if (!p) {
104 DEBUG(0,("ERROR! no memory for pipes_struct!\n"));
105 return NULL;
106 }
107
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;
114 }
115
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;
121 }
122
123 /*
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.
128 */
129
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;
136 }
137
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;
145 }
146
147 DLIST_ADD(InternalPipes, p);
148
149 memcpy(p->client_address, client_address, sizeof(p->client_address));
150
151 p->endian = RPC_LITTLE_ENDIAN;
152
153 /*
154 * Initialize the outgoing RPC data buffer with no memory.
155 */
156 prs_init_empty(&p->out_data.rdata, p->mem_ctx, MARSHALL);
157
158 p->syntax = *syntax;
159
160 DEBUG(4,("Created internal pipe %s (pipes_open=%d)\n",
161 get_pipe_name_from_iface(syntax), pipes_open));
162
163 talloc_set_destructor(p, close_internal_rpc_pipe_hnd);
164
165 return p;
166 }
167
168 /****************************************************************************
169 Sets the fault state on incoming packets.
170 ****************************************************************************/
171
172 static void set_incoming_fault(pipes_struct *p)
173 {
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)));
180 }
181
182 /****************************************************************************
183 Ensures we have at least RPC_HEADER_LEN amount of data in the incoming buffer.
184 ****************************************************************************/
185
186 static ssize_t fill_rpc_header(pipes_struct *p, char *data, size_t data_to_copy)
187 {
188 size_t len_needed_to_complete_hdr = MIN(data_to_copy, RPC_HEADER_LEN - p->in_data.pdu_received_len);
189
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 ));
193
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);
197 }
198 if (p->in_data.current_in_pdu == NULL) {
199 DEBUG(0, ("talloc failed\n"));
200 return -1;
201 }
202
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;
205
206 return (ssize_t)len_needed_to_complete_hdr;
207 }
208
209 /****************************************************************************
210 Unmarshalls a new PDU header. Assumes the raw header data is in current_in_pdu.
211 ****************************************************************************/
212
213 static ssize_t unmarshall_rpc_header(pipes_struct *p)
214 {
215 /*
216 * Unmarshall the header to determine the needed length.
217 */
218
219 prs_struct rpc_in;
220
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;
225 }
226
227 prs_init_empty( &rpc_in, p->mem_ctx, UNMARSHALL);
228 prs_set_endian_data( &rpc_in, p->endian);
229
230 prs_give_memory( &rpc_in, (char *)&p->in_data.current_in_pdu[0],
231 p->in_data.pdu_received_len, False);
232
233 /*
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.
237 */
238
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;
244 }
245
246 /*
247 * Validate the RPC header.
248 */
249
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;
255 }
256
257 /*
258 * If there's not data in the incoming buffer this should be the start of a new RPC.
259 */
260
261 if(prs_offset(&p->in_data.data) == 0) {
262
263 /*
264 * AS/U doesn't set FIRST flag in a BIND packet it seems.
265 */
266
267 if ((p->hdr.pkt_type == RPC_REQUEST) && !(p->hdr.flags & RPC_FLG_FIRST)) {
268 /*
269 * Ensure that the FIRST flag is set. If not then we have
270 * a stream missmatch.
271 */
272
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;
277 }
278
279 /*
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.
283 */
284
285 p->endian = rpc_in.bigendian_data;
286
287 DEBUG(5,("unmarshall_rpc_header: using %sendian RPC\n",
288 p->endian == RPC_LITTLE_ENDIAN ? "little-" : "big-" ));
289
290 } else {
291
292 /*
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.
295 */
296
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;
302 }
303 }
304
305 /*
306 * Ensure that the pdu length is sane.
307 */
308
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;
314 }
315
316 DEBUG(10,("unmarshall_rpc_header: type = %u, flags = %u\n", (unsigned int)p->hdr.pkt_type,
317 (unsigned int)p->hdr.flags ));
318
319 p->in_data.pdu_needed_len = (uint32)p->hdr.frag_len - RPC_HEADER_LEN;
320
321 prs_mem_free(&rpc_in);
322
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;
329 }
330
331 return 0; /* No extra data processed. */
332 }
333
334 /****************************************************************************
335 Call this to free any talloc'ed memory. Do this before and after processing
336 a complete PDU.
337 ****************************************************************************/
338
339 static void free_pipe_context(pipes_struct *p)
340 {
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;
350 }
351 }
352 }
353
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 ****************************************************************************/
358
359 static bool process_request_pdu(pipes_struct *p, prs_struct *rpc_in_p)
360 {
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;
364
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;
369 }
370
371 /*
372 * Check if we need to do authentication processing.
373 * This is only done on requests, not binds.
374 */
375
376 /*
377 * Read the RPC request header.
378 */
379
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;
384 }
385
386 switch(p->auth.auth_type) {
387 case PIPE_AUTH_TYPE_NONE:
388 break;
389
390 case PIPE_AUTH_TYPE_SPNEGO_NTLMSSP:
391 case PIPE_AUTH_TYPE_NTLMSSP:
392 {
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;
399 }
400 break;
401 }
402
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;
408 }
409 break;
410
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;
415 }
416
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;
420 }
421
422 /*
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
427 */
428
429 if(prs_offset(&p->in_data.data) + data_len > 15*1024*1024) {
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;
434 }
435
436 /*
437 * Append the data portion into the buffer and return.
438 */
439
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;
445 }
446
447 if(p->hdr.flags & RPC_FLG_LAST) {
448 bool ret = False;
449 /*
450 * Ok - we finally have a complete RPC stream.
451 * Call the rpc command to process it.
452 */
453
454 /*
455 * Ensure the internal prs buffer size is *exactly* the same
456 * size as the current offset.
457 */
458
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;
463 }
464
465 /*
466 * Set the parse offset to the start of the data and set the
467 * prs_struct to UNMARSHALL.
468 */
469
470 prs_set_offset(&p->in_data.data, 0);
471 prs_switch_type(&p->in_data.data, UNMARSHALL);
472
473 /*
474 * Process the complete data stream here.
475 */
476
477 free_pipe_context(p);
478
479 if(pipe_init_outgoing_data(p)) {
480 ret = api_pipe_request(p);
481 }
482
483 free_pipe_context(p);
484
485 /*
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.
491 */
492
493 prs_switch_type(&p->in_data.data, MARSHALL);
494 prs_set_offset(&p->in_data.data, 0);
495 return ret;
496 }
497
498 return True;
499 }
500
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 ****************************************************************************/
505
506 static void process_complete_pdu(pipes_struct *p)
507 {
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;
512
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;
519 }
520
521 prs_init_empty( &rpc_in, p->mem_ctx, UNMARSHALL);
522
523 /*
524 * Ensure we're using the corrent endianness for both the
525 * RPC header flags and the raw data we will be reading from.
526 */
527
528 prs_set_endian_data( &rpc_in, p->endian);
529 prs_set_endian_data( &p->in_data.data, p->endian);
530
531 prs_give_memory( &rpc_in, data_p, (uint32)data_len, False);
532
533 DEBUG(10,("process_complete_pdu: processing packet type %u\n",
534 (unsigned int)p->hdr.pkt_type ));
535
536 switch (p->hdr.pkt_type) {
537 case RPC_REQUEST:
538 reply = process_request_pdu(p, &rpc_in);
539 break;
540
541 case RPC_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;
546
547 case RPC_RESPONSE: /* No responses here. */
548 DEBUG(0,("process_complete_pdu: Error. RPC_RESPONSE received from client on pipe %s.\n",
549 get_pipe_name_from_iface(&p->syntax)));
550 break;
551
552 case RPC_FAULT:
553 case RPC_WORKING: /* CL request - reply to a ping when a call in process. */
554 case RPC_NOCALL: /* CL - server reply to a ping call. */
555 case RPC_REJECT:
556 case RPC_ACK:
557 case RPC_CL_CANCEL:
558 case RPC_FACK:
559 case RPC_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;
564
565 case RPC_BIND:
566 /*
567 * We assume that a pipe bind is only in one pdu.
568 */
569 if(pipe_init_outgoing_data(p)) {
570 reply = api_pipe_bind_req(p, &rpc_in);
571 }
572 break;
573
574 case RPC_BINDACK:
575 case RPC_BINDNACK:
576 DEBUG(0,("process_complete_pdu: Error. RPC_BINDACK/RPC_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;
580
581
582 case RPC_ALTCONT:
583 /*
584 * We assume that a pipe bind is only in one pdu.
585 */
586 if(pipe_init_outgoing_data(p)) {
587 reply = api_pipe_alter_context(p, &rpc_in);
588 }
589 break;
590
591 case RPC_ALTCONTRESP:
592 DEBUG(0,("process_complete_pdu: Error. RPC_ALTCONTRESP on pipe %s: Should only be server -> client.\n",
593 get_pipe_name_from_iface(&p->syntax)));
594 break;
595
596 case RPC_AUTH3:
597 /*
598 * The third packet in an NTLMSSP auth exchange.
599 */
600 if(pipe_init_outgoing_data(p)) {
601 reply = api_pipe_bind_auth3(p, &rpc_in);
602 }
603 break;
604
605 case RPC_SHUTDOWN:
606 DEBUG(0,("process_complete_pdu: Error. RPC_SHUTDOWN on pipe %s: Should only be server -> client.\n",
607 get_pipe_name_from_iface(&p->syntax)));
608 break;
609
610 case RPC_CO_CANCEL:
611 /* For now just free all client data and continue processing. */
612 DEBUG(3,("process_complete_pdu: RPC_ORPHANED. 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);
625 }
626 break;
627 #endif
628
629 case RPC_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: RPC_ORPHANED. Abandoning rpc call.\n"));
633 reply = True;
634 break;
635
636 default:
637 DEBUG(0,("process_complete_pdu: Unknown rpc type = %u received.\n", (unsigned int)p->hdr.pkt_type ));
638 break;
639 }
640
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);
643
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 {
651 /*
652 * Reset the lengths. We're ready for a new pdu.
653 */
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;
657 }
658
659 prs_mem_free(&rpc_in);
660 }
661
662 /****************************************************************************
663 Accepts incoming data on an rpc pipe. Processes the data in pdu sized units.
664 ****************************************************************************/
665
666 static ssize_t process_incoming_data(pipes_struct *p, char *data, size_t n)
667 {
668 size_t data_to_copy = MIN(n, RPC_MAX_PDU_FRAG_LEN - p->in_data.pdu_received_len);
669
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 ));
673
674 if(data_to_copy == 0) {
675 /*
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.
678 */
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;
683 }
684
685 /*
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.
688 */
689
690 if((p->in_data.pdu_needed_len == 0) && (p->in_data.pdu_received_len < RPC_HEADER_LEN)) {
691 /*
692 * Always return here. If we have more data then the RPC_HEADER
693 * will be processed the next time around the loop.
694 */
695 return fill_rpc_header(p, data, data_to_copy);
696 }
697
698 /*
699 * At this point we know we have at least an RPC_HEADER_LEN amount of data
700 * stored in current_in_pdu.
701 */
702
703 /*
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.
707 */
708
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;
713 }
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 RPC_SHUTDOWN, RPC_CO_CANCEL or RPC_ORPHANED
716 pdu type. Deal with this in process_complete_pdu(). */
717 }
718
719 /*
720 * Ok - at this point we have a valid RPC_HEADER in p->hdr.
721 * Keep reading until we have a full pdu.
722 */
723
724 data_to_copy = MIN(data_to_copy, p->in_data.pdu_needed_len);
725
726 /*
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.
729 */
730
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;
734
735 /*
736 * Do we have a complete PDU ?
737 * (return the number of bytes handled in the call)
738 */
739
740 if(p->in_data.pdu_needed_len == 0) {
741 process_complete_pdu(p);
742 return data_to_copy;
743 }
744
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 ));
747
748 return (ssize_t)data_to_copy;
749 }
750
751 /****************************************************************************
752 Accepts incoming data on an internal rpc pipe.
753 ****************************************************************************/
754
755 static ssize_t write_to_internal_pipe(struct pipes_struct *p, char *data, size_t n)
756 {
757 size_t data_left = n;
758
759 while(data_left) {
760 ssize_t data_used;
761
762 DEBUG(10,("write_to_pipe: data_left = %u\n", (unsigned int)data_left ));
763
764 data_used = process_incoming_data(p, data, data_left);
765
766 DEBUG(10,("write_to_pipe: data_used = %d\n", (int)data_used ));
767
768 if(data_used < 0) {
769 return -1;
770 }
771
772 data_left -= data_used;
773 data += data_used;
774 }
775
776 return n;
777 }
778
779 /****************************************************************************
780 Replies to a request to read data from a pipe.
781
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).
785
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 ****************************************************************************/
789
790 static ssize_t read_from_internal_pipe(struct pipes_struct *p, char *data, size_t n,
791 bool *is_data_outstanding)
792 {
793 uint32 pdu_remaining = 0;
794 ssize_t data_returned = 0;
795
796 if (!p) {
797 DEBUG(0,("read_from_pipe: pipe not open\n"));
798 return -1;
799 }
800
801 DEBUG(6,(" name: %s len: %u\n", get_pipe_name_from_iface(&p->syntax),
802 (unsigned int)n));
803
804 /*
805 * We cannot return more than one PDU length per
806 * read request.
807 */
808
809 /*
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.
813 */
814
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;
821 }
822
823 /*
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.
829 */
830
831 pdu_remaining = prs_offset(&p->out_data.frag)
832 - p->out_data.current_pdu_sent;
833
834 if (pdu_remaining > 0) {
835 data_returned = (ssize_t)MIN(n, pdu_remaining);
836
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));
843
844 memcpy(data,
845 prs_data_p(&p->out_data.frag)
846 + p->out_data.current_pdu_sent,
847 data_returned);
848
849 p->out_data.current_pdu_sent += (uint32)data_returned;
850 goto out;
851 }
852
853 /*
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.
856 */
857
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) ));
863
864 if(p->out_data.data_sent_length >= prs_offset(&p->out_data.rdata)) {
865 /*
866 * We have sent all possible data, return 0.
867 */
868 data_returned = 0;
869 goto out;
870 }
871
872 /*
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.
877 */
878
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;
883 }
884
885 data_returned = MIN(n, prs_offset(&p->out_data.frag));
886
887 memcpy( data, prs_data_p(&p->out_data.frag), (size_t)data_returned);
888 p->out_data.current_pdu_sent += (uint32)data_returned;
889
890 out:
891 (*is_data_outstanding) = prs_offset(&p->out_data.frag) > n;
892
893 return data_returned;
894 }
895
896 /****************************************************************************
897 Close an rpc pipe.
898 ****************************************************************************/
899
900 static int close_internal_rpc_pipe_hnd(struct pipes_struct *p)
901 {
902 if (!p) {
903 DEBUG(0,("Invalid pipe in close_internal_rpc_pipe_hnd\n"));
904 return False;
905 }
906
907 prs_mem_free(&p->out_data.frag);
908 prs_mem_free(&p->out_data.rdata);
909 prs_mem_free(&p->in_data.data);
910
911 if (p->auth.auth_data_free_func) {
912 (*p->auth.auth_data_free_func)(&p->auth);
913 }
914
915 TALLOC_FREE(p->mem_ctx);
916
917 free_pipe_rpc_context( p->contexts );
918
919 /* Free the handles database. */
920 close_policy_by_pipe(p);
921
922 DLIST_REMOVE(InternalPipes, p);
923
924 ZERO_STRUCTP(p);
925
926 TALLOC_FREE(p);
927
928 return True;
929 }
930
931 bool fsp_is_np(struct files_struct *fsp)
932 {
933 enum FAKE_FILE_TYPE type;
934
935 if ((fsp == NULL) || (fsp->fake_file_handle == NULL)) {
936 return false;
937 }
938
939 type = fsp->fake_file_handle->type;
940
941 return ((type == FAKE_FILE_TYPE_NAMED_PIPE)
942 || (type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY));
943 }
944
945 struct np_proxy_state {
946 struct async_req_queue *read_queue;
947 int fd;
948
949 uint8_t *msg;
950 size_t sent;
951 };
952
953 static int np_proxy_state_destructor(struct np_proxy_state *state)
954 {
955 if (state->fd != -1) {
956 close(state->fd);
957 }
958 return 0;
959 }
960
961 static struct np_proxy_state *make_external_rpc_pipe_p(TALLOC_CTX *mem_ctx,
962 const char *pipe_name,
963 struct auth_serversupplied_info *server_info)
964 {
965 struct np_proxy_state *result;
966 struct sockaddr_un addr;
967 char *socket_path;
968 const char *socket_dir;
969
970 DATA_BLOB req_blob;
971 struct netr_SamInfo3 *info3;
972 struct named_pipe_auth_req req;
973 DATA_BLOB rep_blob;
974 uint8 rep_buf[20];
975 struct named_pipe_auth_rep rep;
976 enum ndr_err_code ndr_err;
977 NTSTATUS status;
978 ssize_t written;
979
980 result = talloc(mem_ctx, struct np_proxy_state);
981 if (result == NULL) {
982 DEBUG(0, ("talloc failed\n"));
983 return NULL;
984 }
985
986 result->fd = socket(AF_UNIX, SOCK_STREAM, 0);
987 if (result->fd == -1) {
988 DEBUG(10, ("socket(2) failed: %s\n", strerror(errno)));
989 goto fail;
990 }
991 talloc_set_destructor(result, np_proxy_state_destructor);
992
993 ZERO_STRUCT(addr);
994 addr.sun_family = AF_UNIX;
995
996 socket_dir = lp_parm_const_string(
997 GLOBAL_SECTION_SNUM, "external_rpc_pipe", "socket_dir",
998 get_dyn_NCALRPCDIR());
999 if (socket_dir == NULL) {
1000 DEBUG(0, ("externan_rpc_pipe:socket_dir not set\n"));
1001 goto fail;
1002 }
1003
1004 socket_path = talloc_asprintf(talloc_tos(), "%s/np/%s",
1005 socket_dir, pipe_name);
1006 if (socket_path == NULL) {
1007 DEBUG(0, ("talloc_asprintf failed\n"));
1008 goto fail;
1009 }
1010 strncpy(addr.sun_path, socket_path, sizeof(addr.sun_path));
1011 TALLOC_FREE(socket_path);
1012
1013 become_root();
1014 if (sys_connect(result->fd, (struct sockaddr *)&addr) == -1) {
1015 unbecome_root();
1016 DEBUG(0, ("connect(%s) failed: %s\n", addr.sun_path,
1017 strerror(errno)));
1018 goto fail;
1019 }
1020 unbecome_root();
1021
1022 info3 = talloc(talloc_tos(), struct netr_SamInfo3);
1023 if (info3 == NULL) {
1024 DEBUG(0, ("talloc failed\n"));
1025 goto fail;
1026 }
1027
1028 status = serverinfo_to_SamInfo3(server_info, NULL, 0, info3);
1029 if (!NT_STATUS_IS_OK(status)) {
1030 TALLOC_FREE(info3);
1031 DEBUG(0, ("serverinfo_to_SamInfo3 failed: %s\n",
1032 nt_errstr(status)));
1033 goto fail;
1034 }
1035
1036 req.level = 1;
1037 req.info.info1 = *info3;
1038
1039 ndr_err = ndr_push_struct_blob(
1040 &req_blob, talloc_tos(), NULL, &req,
1041 (ndr_push_flags_fn_t)ndr_push_named_pipe_auth_req);
1042
1043 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1044 DEBUG(10, ("ndr_push_named_pipe_auth_req failed: %s\n",
1045 ndr_errstr(ndr_err)));
1046 goto fail;
1047 }
1048
1049 DEBUG(10, ("named_pipe_auth_req(client)[%u]\n", (uint32_t)req_blob.length));
1050 dump_data(10, req_blob.data, req_blob.length);
1051
1052 written = write_data(result->fd, (char *)req_blob.data,
1053 req_blob.length);
1054 if (written == -1) {
1055 DEBUG(3, ("Could not write auth req data to RPC server\n"));
1056 goto fail;
1057 }
1058
1059 status = read_data(result->fd, (char *)rep_buf, sizeof(rep_buf));
1060 if (!NT_STATUS_IS_OK(status)) {
1061 DEBUG(3, ("Could not read auth result\n"));
1062 goto fail;
1063 }
1064
1065 rep_blob = data_blob_const(rep_buf, sizeof(rep_buf));
1066
1067 DEBUG(10,("name_pipe_auth_rep(client)[%u]\n", (uint32_t)rep_blob.length));
1068 dump_data(10, rep_blob.data, rep_blob.length);
1069
1070 ndr_err = ndr_pull_struct_blob(
1071 &rep_blob, talloc_tos(), NULL, &rep,
1072 (ndr_pull_flags_fn_t)ndr_pull_named_pipe_auth_rep);
1073
1074 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
1075 DEBUG(0, ("ndr_pull_named_pipe_auth_rep failed: %s\n",
1076 ndr_errstr(ndr_err)));
1077 goto fail;
1078 }
1079
1080 if (rep.length != 16) {
1081 DEBUG(0, ("req invalid length: %u != 16\n",
1082 rep.length));
1083 goto fail;
1084 }
1085
1086 if (strcmp(NAMED_PIPE_AUTH_MAGIC, rep.magic) != 0) {
1087 DEBUG(0, ("req invalid magic: %s != %s\n",
1088 rep.magic, NAMED_PIPE_AUTH_MAGIC));
1089 goto fail;
1090 }
1091
1092 if (!NT_STATUS_IS_OK(rep.status)) {
1093 DEBUG(0, ("req failed: %s\n",
1094 nt_errstr(rep.status)));
1095 goto fail;
1096 }
1097
1098 if (rep.level != 1) {
1099 DEBUG(0, ("req invalid level: %u != 1\n",
1100 rep.level));
1101 goto fail;
1102 }
1103
1104 result->msg = NULL;
1105
1106 result->read_queue = async_req_queue_init(result);
1107 if (result->read_queue == NULL) {
1108 goto fail;
1109 }
1110
1111 return result;
1112
1113 fail:
1114 TALLOC_FREE(result);
1115 return NULL;
1116 }
1117
1118 NTSTATUS np_open(TALLOC_CTX *mem_ctx, const char *name,
1119 const char *client_address,
1120 struct auth_serversupplied_info *server_info,
1121 struct fake_file_handle **phandle)
1122 {
1123 const char **proxy_list;
1124 struct fake_file_handle *handle;
1125
1126 proxy_list = lp_parm_string_list(-1, "np", "proxy", NULL);
1127
1128 handle = talloc(mem_ctx, struct fake_file_handle);
1129 if (handle == NULL) {
1130 return NT_STATUS_NO_MEMORY;
1131 }
1132
1133 if ((proxy_list != NULL) && str_list_check_ci(proxy_list, name)) {
1134 struct np_proxy_state *p;
1135
1136 p = make_external_rpc_pipe_p(handle, name, server_info);
1137
1138 handle->type = FAKE_FILE_TYPE_NAMED_PIPE_PROXY;
1139 handle->private_data = p;
1140 } else {
1141 struct pipes_struct *p;
1142 struct ndr_syntax_id syntax;
1143
1144 if (!is_known_pipename(name, &syntax)) {
1145 TALLOC_FREE(handle);
1146 return NT_STATUS_OBJECT_NAME_NOT_FOUND;
1147 }
1148
1149 p = make_internal_rpc_pipe_p(handle, &syntax, client_address,
1150 server_info);
1151
1152 handle->type = FAKE_FILE_TYPE_NAMED_PIPE;
1153 handle->private_data = p;
1154 }
1155
1156 if (handle->private_data == NULL) {
1157 TALLOC_FREE(handle);
1158 return NT_STATUS_PIPE_NOT_AVAILABLE;
1159 }
1160
1161 *phandle = handle;
1162
1163 return NT_STATUS_OK;
1164 }
1165
1166 struct np_write_state {
1167 ssize_t nwritten;
1168 };
1169
1170 static void np_write_done(struct async_req *subreq);
1171
1172 struct async_req *np_write_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
1173 struct fake_file_handle *handle,
1174 const uint8_t *data, size_t len)
1175 {
1176 struct async_req *result, *subreq;
1177 struct np_write_state *state;
1178 NTSTATUS status;
1179
1180 DEBUG(6, ("np_write_send: len: %d\n", (int)len));
1181 dump_data(50, data, len);
1182
1183 if (!async_req_setup(mem_ctx, &result, &state,
1184 struct np_write_state)) {
1185 return NULL;
1186 }
1187
1188 if (len == 0) {
1189 state->nwritten = 0;
1190 status = NT_STATUS_OK;
1191 goto post_status;
1192 }
1193
1194 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE) {
1195 struct pipes_struct *p = talloc_get_type_abort(
1196 handle->private_data, struct pipes_struct);
1197
1198 state->nwritten = write_to_internal_pipe(p, (char *)data, len);
1199
1200 status = (state->nwritten >= 0)
1201 ? NT_STATUS_OK : NT_STATUS_UNEXPECTED_IO_ERROR;
1202 goto post_status;
1203 }
1204
1205 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
1206 struct np_proxy_state *p = talloc_get_type_abort(
1207 handle->private_data, struct np_proxy_state);
1208
1209 state->nwritten = len;
1210
1211 subreq = sendall_send(state, ev, p->fd, data, len, 0);
1212 if (subreq == NULL) {
1213 goto fail;
1214 }
1215 subreq->async.fn = np_write_done;
1216 subreq->async.priv = result;
1217 return result;
1218 }
1219
1220 status = NT_STATUS_INVALID_HANDLE;
1221 post_status:
1222 if (async_post_ntstatus(result, ev, status)) {
1223 return result;
1224 }
1225 fail:
1226 TALLOC_FREE(result);
1227 return NULL;
1228 }
1229
1230 static void np_write_done(struct async_req *subreq)
1231 {
1232 struct async_req *req = talloc_get_type_abort(
1233 subreq->async.priv, struct async_req);
1234 int err;
1235 ssize_t ret;
1236
1237 ret = sendall_recv(subreq, &err);
1238 if (ret < 0) {
1239 async_req_nterror(req, map_nt_error_from_unix(err));
1240 return;
1241 }
1242 async_req_done(req);
1243 }
1244
1245 NTSTATUS np_write_recv(struct async_req *req, ssize_t *pnwritten)
1246 {
1247 struct np_write_state *state = talloc_get_type_abort(
1248 req->private_data, struct np_write_state);
1249 NTSTATUS status;
1250
1251 if (async_req_is_nterror(req, &status)) {
1252 return status;
1253 }
1254 *pnwritten = state->nwritten;
1255 return NT_STATUS_OK;
1256 }
1257
1258 static ssize_t rpc_frag_more_fn(uint8_t *buf, size_t buflen, void *priv)
1259 {
1260 prs_struct hdr_prs;
1261 struct rpc_hdr_info hdr;
1262 bool ret;
1263
1264 if (buflen > RPC_HEADER_LEN) {
1265 return 0;
1266 }
1267 prs_init_empty(&hdr_prs, talloc_tos(), UNMARSHALL);
1268 prs_give_memory(&hdr_prs, (char *)buf, RPC_HEADER_LEN, false);
1269 ret = smb_io_rpc_hdr("", &hdr, &hdr_prs, 0);
1270 prs_mem_free(&hdr_prs);
1271
1272 if (!ret) {
1273 return -1;
1274 }
1275
1276 return (hdr.frag_len - RPC_HEADER_LEN);
1277 }
1278
1279 struct np_read_state {
1280 struct event_context *ev;
1281 struct np_proxy_state *p;
1282 uint8_t *data;
1283 size_t len;
1284
1285 size_t nread;
1286 bool is_data_outstanding;
1287 };
1288
1289 static void np_read_trigger(struct async_req *req);
1290 static void np_read_done(struct async_req *subreq);
1291
1292 struct async_req *np_read_send(TALLOC_CTX *mem_ctx, struct event_context *ev,
1293 struct fake_file_handle *handle,
1294 uint8_t *data, size_t len)
1295 {
1296 struct async_req *result;
1297 struct np_read_state *state;
1298 NTSTATUS status;
1299
1300 if (!async_req_setup(mem_ctx, &result, &state,
1301 struct np_read_state)) {
1302 return NULL;
1303 }
1304
1305 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE) {
1306 struct pipes_struct *p = talloc_get_type_abort(
1307 handle->private_data, struct pipes_struct);
1308
1309 state->nread = read_from_internal_pipe(
1310 p, (char *)data, len, &state->is_data_outstanding);
1311
1312 status = (state->nread >= 0)
1313 ? NT_STATUS_OK : NT_STATUS_UNEXPECTED_IO_ERROR;
1314 goto post_status;
1315 }
1316
1317 if (handle->type == FAKE_FILE_TYPE_NAMED_PIPE_PROXY) {
1318 struct np_proxy_state *p = talloc_get_type_abort(
1319 handle->private_data, struct np_proxy_state);
1320
1321 if (p->msg != NULL) {
1322 size_t thistime;
1323
1324 thistime = MIN(talloc_get_size(p->msg) - p->sent,
1325 len);
1326
1327 memcpy(data, p->msg+p->sent, thistime);
1328 state->nread = thistime;
1329 p->sent += thistime;
1330
1331 if (p->sent < talloc_get_size(p->msg)) {
1332 state->is_data_outstanding = true;
1333 } else {
1334 state->is_data_outstanding = false;
1335 TALLOC_FREE(p->msg);
1336 }
1337 status = NT_STATUS_OK;
1338 goto post_status;
1339 }
1340
1341 state->ev = ev;
1342 state->p = p;
1343 state->data = data;
1344 state->len = len;
1345
1346 if (!async_req_enqueue(p->read_queue, ev, result,
1347 np_read_trigger)) {
1348 goto fail;
1349 }
1350 return result;
1351 }
1352
1353 status = NT_STATUS_INVALID_HANDLE;
1354 post_status:
1355 if (async_post_ntstatus(result, ev, status)) {
1356 return result;
1357 }
1358 fail:
1359 TALLOC_FREE(result);
1360 return NULL;
1361 }
1362
1363 static void np_read_trigger(struct async_req *req)
1364 {
1365 struct np_read_state *state = talloc_get_type_abort(
1366 req->private_data, struct np_read_state);
1367 struct async_req *subreq;
1368
1369 subreq = read_pkt_send(state, state->ev, state->p->fd, RPC_HEADER_LEN,
1370 rpc_frag_more_fn, NULL);
1371 if (async_req_nomem(subreq, req)) {
1372 return;
1373 }
1374 subreq->async.fn = np_read_done;
1375 subreq->async.priv = req;
1376 }
1377
1378 static void np_read_done(struct async_req *subreq)
1379 {
1380 struct async_req *req = talloc_get_type_abort(
1381 subreq->async.priv, struct async_req);
1382 struct np_read_state *state = talloc_get_type_abort(
1383 req->private_data, struct np_read_state);
1384 ssize_t received;
1385 size_t thistime;
1386 int err;
1387
1388 received = read_pkt_recv(subreq, state->p, &state->p->msg, &err);
1389 TALLOC_FREE(subreq);
1390 if (received == -1) {
1391 async_req_nterror(req, map_nt_error_from_unix(err));
1392 return;
1393 }
1394
1395 thistime = MIN(received, state->len);
1396
1397 memcpy(state->data, state->p->msg, thistime);
1398 state->p->sent = thistime;
1399 state->nread = thistime;
1400
1401 if (state->p->sent < received) {
1402 state->is_data_outstanding = true;
1403 } else {
1404 TALLOC_FREE(state->p->msg);
1405 state->is_data_outstanding = false;
1406 }
1407
1408 async_req_done(req);
1409 return;
1410 }
1411
1412 NTSTATUS np_read_recv(struct async_req *req, ssize_t *nread,
1413 bool *is_data_outstanding)
1414 {
1415 struct np_read_state *state = talloc_get_type_abort(
1416 req->private_data, struct np_read_state);
1417 NTSTATUS status;
1418
1419 if (async_req_is_nterror(req, &status)) {
1420 return status;
1421 }
1422 *nread = state->nread;
1423 *is_data_outstanding = state->is_data_outstanding;
1424 return NT_STATUS_OK;
1425 }
1426
1427 /**
1428 * Create a new RPC client context which uses a local dispatch function.
1429 */
1430 NTSTATUS rpc_pipe_open_internal(TALLOC_CTX *mem_ctx,
1431 const struct ndr_syntax_id *abstract_syntax,
1432 NTSTATUS (*dispatch) (struct rpc_pipe_client *cli,
1433 TALLOC_CTX *mem_ctx,
1434 const struct ndr_interface_table *table,
1435 uint32_t opnum, void *r),
1436 struct auth_serversupplied_info *serversupplied_info,
1437 struct rpc_pipe_client **presult)
1438 {
1439 struct rpc_pipe_client *result;
1440
1441 result = TALLOC_ZERO_P(mem_ctx, struct rpc_pipe_client);
1442 if (result == NULL) {
1443 return NT_STATUS_NO_MEMORY;
1444 }
1445
1446 result->abstract_syntax = *abstract_syntax;
1447 result->transfer_syntax = ndr_transfer_syntax;
1448 result->dispatch = dispatch;
1449
1450 result->pipes_struct = make_internal_rpc_pipe_p(
1451 result, abstract_syntax, "", serversupplied_info);
1452 if (result->pipes_struct == NULL) {
1453 TALLOC_FREE(result);
1454 return NT_STATUS_NO_MEMORY;
1455 }
1456
1457 result->max_xmit_frag = -1;
1458 result->max_recv_frag = -1;
1459
1460 *presult = result;
1461 return NT_STATUS_OK;
1462 }
Samba GIT mirror