r23134: Set the event context onto the cli_credentials.
[kai/samba-autobuild/.git] / source4 / librpc / rpc / dcerpc_util.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    dcerpc utility functions
5
6    Copyright (C) Andrew Tridgell 2003
7    Copyright (C) Jelmer Vernooij 2004
8    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005
9    Copyright (C) Rafal Szczesniak 2006
10    
11    This program is free software; you can redistribute it and/or modify
12    it under the terms of the GNU General Public License as published by
13    the Free Software Foundation; either version 2 of the License, or
14    (at your option) any later version.
15    
16    This program is distributed in the hope that it will be useful,
17    but WITHOUT ANY WARRANTY; without even the implied warranty of
18    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
19    GNU General Public License for more details.
20    
21    You should have received a copy of the GNU General Public License
22    along with this program; if not, write to the Free Software
23    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
24 */
25
26 #include "includes.h"
27 #include "lib/events/events.h"
28 #include "libcli/composite/composite.h"
29 #include "librpc/gen_ndr/ndr_epmapper_c.h"
30 #include "librpc/gen_ndr/ndr_dcerpc.h"
31 #include "librpc/gen_ndr/ndr_misc.h"
32 #include "auth/credentials/credentials.h"
33
34 /*
35   find a dcerpc call on an interface by name
36 */
37 const struct dcerpc_interface_call *dcerpc_iface_find_call(const struct dcerpc_interface_table *iface,
38                                                            const char *name)
39 {
40         int i;
41         for (i=0;i<iface->num_calls;i++) {
42                 if (strcmp(iface->calls[i].name, name) == 0) {
43                         return &iface->calls[i];
44                 }
45         }
46         return NULL;
47 }
48
49 /* 
50    push a ncacn_packet into a blob, potentially with auth info
51 */
52 NTSTATUS ncacn_push_auth(DATA_BLOB *blob, TALLOC_CTX *mem_ctx, 
53                           struct ncacn_packet *pkt,
54                           struct dcerpc_auth *auth_info)
55 {
56         NTSTATUS status;
57         struct ndr_push *ndr;
58
59         ndr = ndr_push_init_ctx(mem_ctx);
60         if (!ndr) {
61                 return NT_STATUS_NO_MEMORY;
62         }
63
64         if (!(pkt->drep[0] & DCERPC_DREP_LE)) {
65                 ndr->flags |= LIBNDR_FLAG_BIGENDIAN;
66         }
67
68         if (pkt->pfc_flags & DCERPC_PFC_FLAG_OBJECT_UUID) {
69                 ndr->flags |= LIBNDR_FLAG_OBJECT_PRESENT;
70         }
71
72         if (auth_info) {
73                 pkt->auth_length = auth_info->credentials.length;
74         } else {
75                 pkt->auth_length = 0;
76         }
77
78         status = ndr_push_ncacn_packet(ndr, NDR_SCALARS|NDR_BUFFERS, pkt);
79         if (!NT_STATUS_IS_OK(status)) {
80                 return status;
81         }
82
83         if (auth_info) {
84                 status = ndr_push_dcerpc_auth(ndr, NDR_SCALARS|NDR_BUFFERS, auth_info);
85         }
86
87         *blob = ndr_push_blob(ndr);
88
89         /* fill in the frag length */
90         dcerpc_set_frag_length(blob, blob->length);
91
92         return NT_STATUS_OK;
93 }
94
95 #define MAX_PROTSEQ             10
96
97 static const struct {
98         const char *name;
99         enum dcerpc_transport_t transport;
100         int num_protocols;
101         enum epm_protocol protseq[MAX_PROTSEQ];
102 } transports[] = {
103         { "ncacn_np",     NCACN_NP, 3, 
104                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_SMB, EPM_PROTOCOL_NETBIOS }},
105         { "ncacn_ip_tcp", NCACN_IP_TCP, 3, 
106                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_TCP, EPM_PROTOCOL_IP } }, 
107         { "ncacn_http", NCACN_HTTP, 3, 
108                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_HTTP, EPM_PROTOCOL_IP } }, 
109         { "ncadg_ip_udp", NCACN_IP_UDP, 3, 
110                 { EPM_PROTOCOL_NCADG, EPM_PROTOCOL_UDP, EPM_PROTOCOL_IP } },
111         { "ncalrpc", NCALRPC, 2, 
112                 { EPM_PROTOCOL_NCALRPC, EPM_PROTOCOL_PIPE } },
113         { "ncacn_unix_stream", NCACN_UNIX_STREAM, 2, 
114                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_UNIX_DS } },
115         { "ncadg_unix_dgram", NCADG_UNIX_DGRAM, 2, 
116                 { EPM_PROTOCOL_NCADG, EPM_PROTOCOL_UNIX_DS } },
117         { "ncacn_at_dsp", NCACN_AT_DSP, 3, 
118                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_APPLETALK, EPM_PROTOCOL_DSP } },
119         { "ncadg_at_ddp", NCADG_AT_DDP, 3, 
120                 { EPM_PROTOCOL_NCADG, EPM_PROTOCOL_APPLETALK, EPM_PROTOCOL_DDP } },
121         { "ncacn_vns_ssp", NCACN_VNS_SPP, 3, 
122                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_STREETTALK, EPM_PROTOCOL_VINES_SPP } },
123         { "ncacn_vns_ipc", NCACN_VNS_IPC, 3, 
124                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_STREETTALK, EPM_PROTOCOL_VINES_IPC }, },
125         { "ncadg_ipx", NCADG_IPX, 2,
126                 { EPM_PROTOCOL_NCADG, EPM_PROTOCOL_IPX },
127         },
128         { "ncacn_spx", NCACN_SPX, 3,
129                 /* I guess some MS programmer confused the identifier for 
130                  * EPM_PROTOCOL_UUID (0x0D or 13) with the one for 
131                  * EPM_PROTOCOL_SPX (0x13) here. -- jelmer*/
132                 { EPM_PROTOCOL_NCACN, EPM_PROTOCOL_NCALRPC, EPM_PROTOCOL_UUID },
133         },
134 };
135
136 static const struct {
137         const char *name;
138         uint32_t flag;
139 } ncacn_options[] = {
140         {"sign", DCERPC_SIGN},
141         {"seal", DCERPC_SEAL},
142         {"connect", DCERPC_CONNECT},
143         {"spnego", DCERPC_AUTH_SPNEGO},
144         {"ntlm", DCERPC_AUTH_NTLM},
145         {"krb5", DCERPC_AUTH_KRB5},
146         {"validate", DCERPC_DEBUG_VALIDATE_BOTH},
147         {"print", DCERPC_DEBUG_PRINT_BOTH},
148         {"padcheck", DCERPC_DEBUG_PAD_CHECK},
149         {"bigendian", DCERPC_PUSH_BIGENDIAN},
150         {"smb2", DCERPC_SMB2}
151 };
152
153 const char *epm_floor_string(TALLOC_CTX *mem_ctx, struct epm_floor *epm_floor)
154 {
155         struct dcerpc_syntax_id syntax;
156         NTSTATUS status;
157
158         switch(epm_floor->lhs.protocol) {
159                 case EPM_PROTOCOL_UUID:
160                         status = dcerpc_floor_get_lhs_data(epm_floor, &syntax);
161                         if (NT_STATUS_IS_OK(status)) {
162                                 /* lhs is used: UUID */
163                                 char *uuidstr;
164
165                                 if (GUID_equal(&syntax.uuid, &ndr_transfer_syntax.uuid)) {
166                                         return "NDR";
167                                 } 
168
169                                 if (GUID_equal(&syntax.uuid, &ndr64_transfer_syntax.uuid)) {
170                                         return "NDR64";
171                                 } 
172
173                                 uuidstr = GUID_string(mem_ctx, &syntax.uuid);
174
175                                 return talloc_asprintf(mem_ctx, " uuid %s/0x%02x", uuidstr, syntax.if_version);
176                         } else { /* IPX */
177                                 return talloc_asprintf(mem_ctx, "IPX:%s", 
178                                                 data_blob_hex_string(mem_ctx, &epm_floor->rhs.uuid.unknown));
179                         }
180
181                 case EPM_PROTOCOL_NCACN:
182                         return "RPC-C";
183
184                 case EPM_PROTOCOL_NCADG:
185                         return "RPC";
186
187                 case EPM_PROTOCOL_NCALRPC:
188                         return "NCALRPC";
189
190                 case EPM_PROTOCOL_DNET_NSP:
191                         return "DNET/NSP";
192
193                 case EPM_PROTOCOL_IP:
194                         return talloc_asprintf(mem_ctx, "IP:%s", epm_floor->rhs.ip.ipaddr);
195
196                 case EPM_PROTOCOL_PIPE:
197                         return talloc_asprintf(mem_ctx, "PIPE:%s", epm_floor->rhs.pipe.path);
198
199                 case EPM_PROTOCOL_SMB:
200                         return talloc_asprintf(mem_ctx, "SMB:%s", epm_floor->rhs.smb.unc);
201
202                 case EPM_PROTOCOL_UNIX_DS:
203                         return talloc_asprintf(mem_ctx, "Unix:%s", epm_floor->rhs.unix_ds.path);
204
205                 case EPM_PROTOCOL_NETBIOS:
206                         return talloc_asprintf(mem_ctx, "NetBIOS:%s", epm_floor->rhs.netbios.name);
207
208                 case EPM_PROTOCOL_NETBEUI:
209                         return "NETBeui";
210
211                 case EPM_PROTOCOL_SPX:
212                         return "SPX";
213
214                 case EPM_PROTOCOL_NB_IPX:
215                         return "NB_IPX";
216
217                 case EPM_PROTOCOL_HTTP:
218                         return talloc_asprintf(mem_ctx, "HTTP:%d", epm_floor->rhs.http.port);
219
220                 case EPM_PROTOCOL_TCP:
221                         return talloc_asprintf(mem_ctx, "TCP:%d", epm_floor->rhs.tcp.port);
222
223                 case EPM_PROTOCOL_UDP:
224                         return talloc_asprintf(mem_ctx, "UDP:%d", epm_floor->rhs.udp.port);
225
226                 default:
227                         return talloc_asprintf(mem_ctx, "UNK(%02x):", epm_floor->lhs.protocol);
228         }
229 }
230
231
232 /*
233   form a binding string from a binding structure
234 */
235 char *dcerpc_binding_string(TALLOC_CTX *mem_ctx, const struct dcerpc_binding *b)
236 {
237         char *s = talloc_strdup(mem_ctx, "");
238         int i;
239         const char *t_name=NULL;
240
241         for (i=0;i<ARRAY_SIZE(transports);i++) {
242                 if (transports[i].transport == b->transport) {
243                         t_name = transports[i].name;
244                 }
245         }
246         if (!t_name) {
247                 return NULL;
248         }
249
250         if (!GUID_all_zero(&b->object.uuid)) { 
251                 s = talloc_asprintf(s, "%s@",
252                                     GUID_string(mem_ctx, &b->object.uuid));
253         }
254
255         s = talloc_asprintf_append(s, "%s:", t_name);
256         if (!s) return NULL;
257
258         if (b->host) {
259                 s = talloc_asprintf_append(s, "%s", b->host);
260         }
261
262         if (!b->endpoint && !b->options && !b->flags) {
263                 return s;
264         }
265
266         s = talloc_asprintf_append(s, "[");
267
268         if (b->endpoint) {
269                 s = talloc_asprintf_append(s, "%s", b->endpoint);
270         }
271
272         /* this is a *really* inefficent way of dealing with strings,
273            but this is rarely called and the strings are always short,
274            so I don't care */
275         for (i=0;b->options && b->options[i];i++) {
276                 s = talloc_asprintf_append(s, ",%s", b->options[i]);
277                 if (!s) return NULL;
278         }
279
280         for (i=0;i<ARRAY_SIZE(ncacn_options);i++) {
281                 if (b->flags & ncacn_options[i].flag) {
282                         s = talloc_asprintf_append(s, ",%s", ncacn_options[i].name);
283                         if (!s) return NULL;
284                 }
285         }
286
287         s = talloc_asprintf_append(s, "]");
288
289         return s;
290 }
291
292 /*
293   parse a binding string into a dcerpc_binding structure
294 */
295 NTSTATUS dcerpc_parse_binding(TALLOC_CTX *mem_ctx, const char *s, struct dcerpc_binding **b_out)
296 {
297         struct dcerpc_binding *b;
298         char *options, *type;
299         char *p;
300         int i, j, comma_count;
301
302         b = talloc(mem_ctx, struct dcerpc_binding);
303         if (!b) {
304                 return NT_STATUS_NO_MEMORY;
305         }
306
307         p = strchr(s, '@');
308
309         if (p && PTR_DIFF(p, s) == 36) { /* 36 is the length of a UUID */
310                 NTSTATUS status;
311
312                 status = GUID_from_string(s, &b->object.uuid);
313
314                 if (NT_STATUS_IS_ERR(status)) {
315                         DEBUG(0, ("Failed parsing UUID\n"));
316                         return status;
317                 }
318
319                 s = p + 1;
320         } else {
321                 ZERO_STRUCT(b->object);
322         }
323
324         b->object.if_version = 0;
325
326         p = strchr(s, ':');
327         if (!p) {
328                 return NT_STATUS_INVALID_PARAMETER;
329         }
330
331         type = talloc_strndup(mem_ctx, s, PTR_DIFF(p, s));
332         if (!type) {
333                 return NT_STATUS_NO_MEMORY;
334         }
335
336         for (i=0;i<ARRAY_SIZE(transports);i++) {
337                 if (strcasecmp(type, transports[i].name) == 0) {
338                         b->transport = transports[i].transport;
339                         break;
340                 }
341         }
342         if (i==ARRAY_SIZE(transports)) {
343                 DEBUG(0,("Unknown dcerpc transport '%s'\n", type));
344                 return NT_STATUS_INVALID_PARAMETER;
345         }
346         
347         s = p+1;
348
349         p = strchr(s, '[');
350         if (p) {
351                 b->host = talloc_strndup(b, s, PTR_DIFF(p, s));
352                 options = talloc_strdup(mem_ctx, p+1);
353                 if (options[strlen(options)-1] != ']') {
354                         return NT_STATUS_INVALID_PARAMETER;
355                 }
356                 options[strlen(options)-1] = 0;
357         } else {
358                 b->host = talloc_strdup(b, s);
359                 options = NULL;
360         }
361         if (!b->host) {
362                 return NT_STATUS_NO_MEMORY;
363         }
364
365         b->target_hostname = b->host;
366
367         b->options = NULL;
368         b->flags = 0;
369         b->assoc_group_id = 0;
370         b->endpoint = NULL;
371
372         if (!options) {
373                 *b_out = b;
374                 return NT_STATUS_OK;
375         }
376
377         comma_count = count_chars(options, ',');
378
379         b->options = talloc_array(b, const char *, comma_count+2);
380         if (!b->options) {
381                 return NT_STATUS_NO_MEMORY;
382         }
383
384         for (i=0; (p = strchr(options, ',')); i++) {
385                 b->options[i] = talloc_strndup(b, options, PTR_DIFF(p, options));
386                 if (!b->options[i]) {
387                         return NT_STATUS_NO_MEMORY;
388                 }
389                 options = p+1;
390         }
391         b->options[i] = options;
392         b->options[i+1] = NULL;
393
394         /* some options are pre-parsed for convenience */
395         for (i=0;b->options[i];i++) {
396                 for (j=0;j<ARRAY_SIZE(ncacn_options);j++) {
397                         if (strcasecmp(ncacn_options[j].name, b->options[i]) == 0) {
398                                 int k;
399                                 b->flags |= ncacn_options[j].flag;
400                                 for (k=i;b->options[k];k++) {
401                                         b->options[k] = b->options[k+1];
402                                 }
403                                 i--;
404                                 break;
405                         }
406                 }
407         }
408
409         if (b->options[0]) {
410                 /* Endpoint is first option */
411                 b->endpoint = b->options[0];
412                 if (strlen(b->endpoint) == 0) b->endpoint = NULL;
413
414                 for (i=0;b->options[i];i++) {
415                         b->options[i] = b->options[i+1];
416                 }
417         }
418
419         if (b->options[0] == NULL)
420                 b->options = NULL;
421         
422         *b_out = b;
423         return NT_STATUS_OK;
424 }
425
426 NTSTATUS dcerpc_floor_get_lhs_data(struct epm_floor *epm_floor, struct dcerpc_syntax_id *syntax)
427 {
428         TALLOC_CTX *mem_ctx = talloc_init("floor_get_lhs_data");
429         struct ndr_pull *ndr = ndr_pull_init_blob(&epm_floor->lhs.lhs_data, mem_ctx);
430         NTSTATUS status;
431         uint16_t if_version=0;
432
433         ndr->flags |= LIBNDR_FLAG_NOALIGN;
434
435         status = ndr_pull_GUID(ndr, NDR_SCALARS | NDR_BUFFERS, &syntax->uuid);
436         if (NT_STATUS_IS_ERR(status)) {
437                 talloc_free(mem_ctx);
438                 return status;
439         }
440
441         status = ndr_pull_uint16(ndr, NDR_SCALARS, &if_version);
442         syntax->if_version = if_version;
443
444         talloc_free(mem_ctx);
445
446         return status;
447 }
448
449 static DATA_BLOB dcerpc_floor_pack_lhs_data(TALLOC_CTX *mem_ctx, const struct dcerpc_syntax_id *syntax)
450 {
451         struct ndr_push *ndr = ndr_push_init_ctx(mem_ctx);
452
453         ndr->flags |= LIBNDR_FLAG_NOALIGN;
454
455         ndr_push_GUID(ndr, NDR_SCALARS | NDR_BUFFERS, &syntax->uuid);
456         ndr_push_uint16(ndr, NDR_SCALARS, syntax->if_version);
457
458         return ndr_push_blob(ndr);
459 }
460
461 const char *dcerpc_floor_get_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *epm_floor)
462 {
463         switch (epm_floor->lhs.protocol) {
464         case EPM_PROTOCOL_TCP:
465                 if (epm_floor->rhs.tcp.port == 0) return NULL;
466                 return talloc_asprintf(mem_ctx, "%d", epm_floor->rhs.tcp.port);
467                 
468         case EPM_PROTOCOL_UDP:
469                 if (epm_floor->rhs.udp.port == 0) return NULL;
470                 return talloc_asprintf(mem_ctx, "%d", epm_floor->rhs.udp.port);
471
472         case EPM_PROTOCOL_HTTP:
473                 if (epm_floor->rhs.http.port == 0) return NULL;
474                 return talloc_asprintf(mem_ctx, "%d", epm_floor->rhs.http.port);
475
476         case EPM_PROTOCOL_IP:
477                 return talloc_strdup(mem_ctx, epm_floor->rhs.ip.ipaddr);
478
479         case EPM_PROTOCOL_NCACN:
480                 return NULL;
481
482         case EPM_PROTOCOL_NCADG:
483                 return NULL;
484
485         case EPM_PROTOCOL_SMB:
486                 if (strlen(epm_floor->rhs.smb.unc) == 0) return NULL;
487                 return talloc_strdup(mem_ctx, epm_floor->rhs.smb.unc);
488
489         case EPM_PROTOCOL_PIPE:
490                 if (strlen(epm_floor->rhs.pipe.path) == 0) return NULL;
491                 return talloc_strdup(mem_ctx, epm_floor->rhs.pipe.path);
492
493         case EPM_PROTOCOL_NETBIOS:
494                 if (strlen(epm_floor->rhs.netbios.name) == 0) return NULL;
495                 return talloc_strdup(mem_ctx, epm_floor->rhs.netbios.name);
496
497         case EPM_PROTOCOL_NCALRPC:
498                 return NULL;
499                 
500         case EPM_PROTOCOL_VINES_SPP:
501                 return talloc_asprintf(mem_ctx, "%d", epm_floor->rhs.vines_spp.port);
502                 
503         case EPM_PROTOCOL_VINES_IPC:
504                 return talloc_asprintf(mem_ctx, "%d", epm_floor->rhs.vines_ipc.port);
505                 
506         case EPM_PROTOCOL_STREETTALK:
507                 return talloc_strdup(mem_ctx, epm_floor->rhs.streettalk.streettalk);
508                 
509         case EPM_PROTOCOL_UNIX_DS:
510                 if (strlen(epm_floor->rhs.unix_ds.path) == 0) return NULL;
511                 return talloc_strdup(mem_ctx, epm_floor->rhs.unix_ds.path);
512                 
513         case EPM_PROTOCOL_NULL:
514                 return NULL;
515
516         default:
517                 DEBUG(0,("Unsupported lhs protocol %d\n", epm_floor->lhs.protocol));
518                 break;
519         }
520
521         return NULL;
522 }
523
524 static NTSTATUS dcerpc_floor_set_rhs_data(TALLOC_CTX *mem_ctx, struct epm_floor *epm_floor,  const char *data)
525 {
526         switch (epm_floor->lhs.protocol) {
527         case EPM_PROTOCOL_TCP:
528                 epm_floor->rhs.tcp.port = atoi(data);
529                 return NT_STATUS_OK;
530                 
531         case EPM_PROTOCOL_UDP:
532                 epm_floor->rhs.udp.port = atoi(data);
533                 return NT_STATUS_OK;
534
535         case EPM_PROTOCOL_HTTP:
536                 epm_floor->rhs.http.port = atoi(data);
537                 return NT_STATUS_OK;
538
539         case EPM_PROTOCOL_IP:
540                 epm_floor->rhs.ip.ipaddr = talloc_strdup(mem_ctx, data);
541                 NT_STATUS_HAVE_NO_MEMORY(epm_floor->rhs.ip.ipaddr);
542                 return NT_STATUS_OK;
543
544         case EPM_PROTOCOL_NCACN:
545                 epm_floor->rhs.ncacn.minor_version = 0;
546                 return NT_STATUS_OK;
547
548         case EPM_PROTOCOL_NCADG:
549                 epm_floor->rhs.ncadg.minor_version = 0;
550                 return NT_STATUS_OK;
551
552         case EPM_PROTOCOL_SMB:
553                 epm_floor->rhs.smb.unc = talloc_strdup(mem_ctx, data);
554                 NT_STATUS_HAVE_NO_MEMORY(epm_floor->rhs.smb.unc);
555                 return NT_STATUS_OK;
556
557         case EPM_PROTOCOL_PIPE:
558                 epm_floor->rhs.pipe.path = talloc_strdup(mem_ctx, data);
559                 NT_STATUS_HAVE_NO_MEMORY(epm_floor->rhs.pipe.path);
560                 return NT_STATUS_OK;
561
562         case EPM_PROTOCOL_NETBIOS:
563                 epm_floor->rhs.netbios.name = talloc_strdup(mem_ctx, data);
564                 NT_STATUS_HAVE_NO_MEMORY(epm_floor->rhs.netbios.name);
565                 return NT_STATUS_OK;
566
567         case EPM_PROTOCOL_NCALRPC:
568                 return NT_STATUS_OK;
569                 
570         case EPM_PROTOCOL_VINES_SPP:
571                 epm_floor->rhs.vines_spp.port = atoi(data);
572                 return NT_STATUS_OK;
573                 
574         case EPM_PROTOCOL_VINES_IPC:
575                 epm_floor->rhs.vines_ipc.port = atoi(data);
576                 return NT_STATUS_OK;
577                 
578         case EPM_PROTOCOL_STREETTALK:
579                 epm_floor->rhs.streettalk.streettalk = talloc_strdup(mem_ctx, data);
580                 NT_STATUS_HAVE_NO_MEMORY(epm_floor->rhs.streettalk.streettalk);
581                 return NT_STATUS_OK;
582                 
583         case EPM_PROTOCOL_UNIX_DS:
584                 epm_floor->rhs.unix_ds.path = talloc_strdup(mem_ctx, data);
585                 NT_STATUS_HAVE_NO_MEMORY(epm_floor->rhs.unix_ds.path);
586                 return NT_STATUS_OK;
587                 
588         case EPM_PROTOCOL_NULL:
589                 return NT_STATUS_OK;
590
591         default:
592                 DEBUG(0,("Unsupported lhs protocol %d\n", epm_floor->lhs.protocol));
593                 break;
594         }
595
596         return NT_STATUS_NOT_SUPPORTED;
597 }
598
599 enum dcerpc_transport_t dcerpc_transport_by_endpoint_protocol(int prot)
600 {
601         int i;
602
603         /* Find a transport that has 'prot' as 4th protocol */
604         for (i=0;i<ARRAY_SIZE(transports);i++) {
605                 if (transports[i].num_protocols >= 2 && 
606                         transports[i].protseq[1] == prot) {
607                         return transports[i].transport;
608                 }
609         }
610         
611         /* Unknown transport */
612         return (unsigned int)-1;
613 }
614
615 enum dcerpc_transport_t dcerpc_transport_by_tower(struct epm_tower *tower)
616 {
617         int i;
618
619         /* Find a transport that matches this tower */
620         for (i=0;i<ARRAY_SIZE(transports);i++) {
621                 int j;
622                 if (transports[i].num_protocols != tower->num_floors - 2) {
623                         continue; 
624                 }
625
626                 for (j = 0; j < transports[i].num_protocols; j++) {
627                         if (transports[i].protseq[j] != tower->floors[j+2].lhs.protocol) {
628                                 break;
629                         }
630                 }
631
632                 if (j == transports[i].num_protocols) {
633                         return transports[i].transport;
634                 }
635         }
636         
637         /* Unknown transport */
638         return (unsigned int)-1;
639 }
640
641 NTSTATUS dcerpc_binding_from_tower(TALLOC_CTX *mem_ctx, struct epm_tower *tower, struct dcerpc_binding **b_out)
642 {
643         NTSTATUS status;
644         struct dcerpc_binding *binding;
645
646         binding = talloc(mem_ctx, struct dcerpc_binding);
647         NT_STATUS_HAVE_NO_MEMORY(binding);
648
649         ZERO_STRUCT(binding->object);
650         binding->options = NULL;
651         binding->host = NULL;
652         binding->target_hostname = NULL;
653         binding->flags = 0;
654         binding->assoc_group_id = 0;
655
656         binding->transport = dcerpc_transport_by_tower(tower);
657
658         if (binding->transport == (unsigned int)-1) {
659                 return NT_STATUS_NOT_SUPPORTED;
660         }
661
662         if (tower->num_floors < 1) {
663                 return NT_STATUS_OK;
664         }
665
666         /* Set object uuid */
667         status = dcerpc_floor_get_lhs_data(&tower->floors[0], &binding->object);
668         
669         if (!NT_STATUS_IS_OK(status)) {
670                 DEBUG(1, ("Error pulling object uuid and version: %s", nt_errstr(status)));     
671                 return status;
672         }
673
674         /* Ignore floor 1, it contains the NDR version info */
675         
676         binding->options = NULL;
677
678         /* Set endpoint */
679         if (tower->num_floors >= 4) {
680                 binding->endpoint = dcerpc_floor_get_rhs_data(mem_ctx, &tower->floors[3]);
681         } else {
682                 binding->endpoint = NULL;
683         }
684
685         /* Set network address */
686         if (tower->num_floors >= 5) {
687                 binding->host = dcerpc_floor_get_rhs_data(mem_ctx, &tower->floors[4]);
688                 NT_STATUS_HAVE_NO_MEMORY(binding->host);
689                 binding->target_hostname = binding->host;
690         }
691         *b_out = binding;
692         return NT_STATUS_OK;
693 }
694
695 NTSTATUS dcerpc_binding_build_tower(TALLOC_CTX *mem_ctx, struct dcerpc_binding *binding, struct epm_tower *tower)
696 {
697         const enum epm_protocol *protseq = NULL;
698         int num_protocols = -1, i;
699         NTSTATUS status;
700         
701         /* Find transport */
702         for (i=0;i<ARRAY_SIZE(transports);i++) {
703                 if (transports[i].transport == binding->transport) {
704                         protseq = transports[i].protseq;
705                         num_protocols = transports[i].num_protocols;
706                         break;
707                 }
708         }
709
710         if (num_protocols == -1) {
711                 DEBUG(0, ("Unable to find transport with id '%d'\n", binding->transport));
712                 return NT_STATUS_UNSUCCESSFUL;
713         }
714
715         tower->num_floors = 2 + num_protocols;
716         tower->floors = talloc_array(mem_ctx, struct epm_floor, tower->num_floors);
717
718         /* Floor 0 */
719         tower->floors[0].lhs.protocol = EPM_PROTOCOL_UUID;
720
721         tower->floors[0].lhs.lhs_data = dcerpc_floor_pack_lhs_data(mem_ctx, &binding->object);
722
723         tower->floors[0].rhs.uuid.unknown = data_blob_talloc_zero(mem_ctx, 2);
724         
725         /* Floor 1 */
726         tower->floors[1].lhs.protocol = EPM_PROTOCOL_UUID;
727
728         tower->floors[1].lhs.lhs_data = dcerpc_floor_pack_lhs_data(mem_ctx, 
729                                                                 &ndr_transfer_syntax);
730         
731         tower->floors[1].rhs.uuid.unknown = data_blob_talloc_zero(mem_ctx, 2);
732         
733         /* Floor 2 to num_protocols */
734         for (i = 0; i < num_protocols; i++) {
735                 tower->floors[2 + i].lhs.protocol = protseq[i];
736                 tower->floors[2 + i].lhs.lhs_data = data_blob_talloc(mem_ctx, NULL, 0);
737                 ZERO_STRUCT(tower->floors[2 + i].rhs);
738                 dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[2 + i], "");
739         }
740
741         /* The 4th floor contains the endpoint */
742         if (num_protocols >= 2 && binding->endpoint) {
743                 status = dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[3], binding->endpoint);
744                 if (NT_STATUS_IS_ERR(status)) {
745                         return status;
746                 }
747         }
748         
749         /* The 5th contains the network address */
750         if (num_protocols >= 3 && binding->host) {
751                 if (is_ipaddress(binding->host)) {
752                         status = dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[4], 
753                                                            binding->host);
754                 } else {
755                         /* note that we don't attempt to resolve the
756                            name here - when we get a hostname here we
757                            are in the client code, and want to put in
758                            a wildcard all-zeros IP for the server to
759                            fill in */
760                         status = dcerpc_floor_set_rhs_data(mem_ctx, &tower->floors[4], 
761                                                            "0.0.0.0");
762                 }
763                 if (NT_STATUS_IS_ERR(status)) {
764                         return status;
765                 }
766         }
767
768         return NT_STATUS_OK;
769 }
770
771
772 struct epm_map_binding_state {
773         struct dcerpc_binding *binding;
774         const struct dcerpc_interface_table *table;
775         struct dcerpc_pipe *pipe;
776         struct policy_handle handle;
777         struct GUID guid;
778         struct epm_twr_t twr;
779         struct epm_twr_t *twr_r;
780         struct epm_Map r;
781 };
782
783
784 static void continue_epm_recv_binding(struct composite_context *ctx);
785 static void continue_epm_map(struct rpc_request *req);
786
787
788 /*
789   Stage 2 of epm_map_binding: Receive connected rpc pipe and send endpoint
790   mapping rpc request
791 */
792 static void continue_epm_recv_binding(struct composite_context *ctx)
793 {
794         struct rpc_request *map_req;
795
796         struct composite_context *c = talloc_get_type(ctx->async.private_data,
797                                                       struct composite_context);
798         struct epm_map_binding_state *s = talloc_get_type(c->private_data,
799                                                           struct epm_map_binding_state);
800
801         /* receive result of rpc pipe connect request */
802         c->status = dcerpc_pipe_connect_b_recv(ctx, c, &s->pipe);
803         if (!composite_is_ok(c)) return;
804
805         s->pipe->conn->flags |= DCERPC_NDR_REF_ALLOC;
806
807         /* prepare requested binding parameters */
808         s->binding->object         = s->table->syntax_id;
809
810         c->status = dcerpc_binding_build_tower(s->pipe, s->binding, &s->twr.tower);
811         if (!composite_is_ok(c)) return;
812         
813         /* with some nice pretty paper around it of course */
814         s->r.in.object        = &s->guid;
815         s->r.in.map_tower     = &s->twr;
816         s->r.in.entry_handle  = &s->handle;
817         s->r.in.max_towers    = 1;
818         s->r.out.entry_handle = &s->handle;
819
820         /* send request for an endpoint mapping - a rpc request on connected pipe */
821         map_req = dcerpc_epm_Map_send(s->pipe, c, &s->r);
822         if (composite_nomem(map_req, c)) return;
823         
824         composite_continue_rpc(c, map_req, continue_epm_map, c);
825 }
826
827
828 /*
829   Stage 3 of epm_map_binding: Receive endpoint mapping and provide binding details
830 */
831 static void continue_epm_map(struct rpc_request *req)
832 {
833         struct composite_context *c = talloc_get_type(req->async.private_data,
834                                                       struct composite_context);
835         struct epm_map_binding_state *s = talloc_get_type(c->private_data,
836                                                           struct epm_map_binding_state);
837
838         /* receive result of a rpc request */
839         c->status = dcerpc_ndr_request_recv(req);
840         if (!composite_is_ok(c)) return;
841
842         /* check the details */
843         if (s->r.out.result != 0 || *s->r.out.num_towers != 1) {
844                 composite_error(c, NT_STATUS_PORT_UNREACHABLE);
845                 return;
846         }
847         
848         s->twr_r = s->r.out.towers[0].twr;
849         if (s->twr_r == NULL) {
850                 composite_error(c, NT_STATUS_PORT_UNREACHABLE);
851                 return;
852         }
853
854         if (s->twr_r->tower.num_floors != s->twr.tower.num_floors ||
855             s->twr_r->tower.floors[3].lhs.protocol != s->twr.tower.floors[3].lhs.protocol) {
856                 composite_error(c, NT_STATUS_PORT_UNREACHABLE);
857                 return;
858         }
859
860         /* get received endpoint */
861         s->binding->endpoint = talloc_reference(s->binding,
862                                                 dcerpc_floor_get_rhs_data(c, &s->twr_r->tower.floors[3]));
863         if (composite_nomem(s->binding->endpoint, c)) return;
864
865         composite_done(c);
866 }
867
868
869 /*
870   Request for endpoint mapping of dcerpc binding - try to request for endpoint
871   unless there is default one.
872 */
873 struct composite_context *dcerpc_epm_map_binding_send(TALLOC_CTX *mem_ctx,
874                                                       struct dcerpc_binding *binding,
875                                                       const struct dcerpc_interface_table *table,
876                                                       struct event_context *ev)
877 {
878         struct composite_context *c;
879         struct epm_map_binding_state *s;
880         struct composite_context *pipe_connect_req;
881         struct cli_credentials *anon_creds;
882         struct event_context *new_ev = NULL;
883
884         NTSTATUS status;
885         struct dcerpc_binding *epmapper_binding;
886         int i;
887
888         /* Try to find event context in memory context in case passed
889          * event_context (argument) was NULL. If there's none, just
890          * create a new one.
891          */
892         if (ev == NULL) {
893                 ev = event_context_find(mem_ctx);
894                 if (ev == NULL) {
895                         new_ev = event_context_init(mem_ctx);
896                         if (new_ev == NULL) return NULL;
897                         ev = new_ev;
898                 }
899         }
900
901         /* composite context allocation and setup */
902         c = composite_create(mem_ctx, ev);
903         if (c == NULL) {
904                 talloc_free(new_ev);
905                 return NULL;
906         }
907         talloc_steal(c, new_ev);
908
909         s = talloc_zero(c, struct epm_map_binding_state);
910         if (composite_nomem(s, c)) return c;
911         c->private_data = s;
912
913         s->binding = binding;
914         s->table   = table;
915
916         /* anonymous credentials for rpc connection used to get endpoint mapping */
917         anon_creds = cli_credentials_init(mem_ctx);
918         cli_credentials_set_event_context(anon_creds, ev);
919         cli_credentials_set_conf(anon_creds);
920         cli_credentials_set_anonymous(anon_creds);
921
922         /*
923           First, check if there is a default endpoint specified in the IDL
924         */
925         if (table) {
926                 struct dcerpc_binding *default_binding;
927
928                 /* Find one of the default pipes for this interface */
929                 for (i = 0; i < table->endpoints->count; i++) {
930                         status = dcerpc_parse_binding(mem_ctx, table->endpoints->names[i], &default_binding);
931
932                         if (NT_STATUS_IS_OK(status)) {
933                                 if (default_binding->transport == binding->transport && default_binding->endpoint) {
934                                         binding->endpoint = talloc_reference(binding, default_binding->endpoint);
935                                         talloc_free(default_binding);
936
937                                         composite_done(c);
938                                         return c;
939
940                                 } else {
941                                         talloc_free(default_binding);
942                                 }
943                         }
944                 }
945         }
946
947         epmapper_binding = talloc_zero(c, struct dcerpc_binding);
948         if (composite_nomem(epmapper_binding, c)) return c;
949
950         /* basic endpoint mapping data */
951         epmapper_binding->transport             = binding->transport;
952         epmapper_binding->host                  = talloc_reference(epmapper_binding, binding->host);
953         epmapper_binding->target_hostname       = epmapper_binding->host;
954         epmapper_binding->options               = NULL;
955         epmapper_binding->flags                 = 0;
956         epmapper_binding->assoc_group_id        = 0;
957         epmapper_binding->endpoint              = NULL;
958
959         /* initiate rpc pipe connection */
960         pipe_connect_req = dcerpc_pipe_connect_b_send(c, epmapper_binding, &dcerpc_table_epmapper,
961                                                       anon_creds, c->event_ctx);
962         if (composite_nomem(pipe_connect_req, c)) return c;
963         
964         composite_continue(c, pipe_connect_req, continue_epm_recv_binding, c);
965         return c;
966 }
967
968
969 /*
970   Receive result of endpoint mapping request
971  */
972 NTSTATUS dcerpc_epm_map_binding_recv(struct composite_context *c)
973 {
974         NTSTATUS status = composite_wait(c);
975         
976         talloc_free(c);
977         return status;
978 }
979
980
981 /*
982   Get endpoint mapping for rpc connection
983 */
984 NTSTATUS dcerpc_epm_map_binding(TALLOC_CTX *mem_ctx, struct dcerpc_binding *binding,
985                                 const struct dcerpc_interface_table *table, struct event_context *ev)
986 {
987         struct composite_context *c;
988
989         c = dcerpc_epm_map_binding_send(mem_ctx, binding, table, ev);
990         return dcerpc_epm_map_binding_recv(c);
991 }
992
993
994 struct pipe_auth_state {
995         struct dcerpc_pipe *pipe;
996         struct dcerpc_binding *binding;
997         const struct dcerpc_interface_table *table;
998         struct cli_credentials *credentials;
999 };
1000
1001
1002 static void continue_auth_schannel(struct composite_context *ctx);
1003 static void continue_auth(struct composite_context *ctx);
1004 static void continue_auth_none(struct composite_context *ctx);
1005 static void continue_ntlmssp_connection(struct composite_context *ctx);
1006 static void continue_spnego_after_wrong_pass(struct composite_context *ctx);
1007
1008
1009 /*
1010   Stage 2 of pipe_auth: Receive result of schannel bind request
1011 */
1012 static void continue_auth_schannel(struct composite_context *ctx)
1013 {
1014         struct composite_context *c = talloc_get_type(ctx->async.private_data,
1015                                                       struct composite_context);
1016
1017         c->status = dcerpc_bind_auth_schannel_recv(ctx);
1018         if (!composite_is_ok(c)) return;
1019
1020         composite_done(c);
1021 }
1022
1023
1024 /*
1025   Stage 2 of pipe_auth: Receive result of authenticated bind request
1026 */
1027 static void continue_auth(struct composite_context *ctx)
1028 {
1029         struct composite_context *c = talloc_get_type(ctx->async.private_data,
1030                                                       struct composite_context);
1031
1032         c->status = dcerpc_bind_auth_recv(ctx);
1033         if (!composite_is_ok(c)) return;
1034         
1035         composite_done(c);
1036 }
1037 /*
1038   Stage 2 of pipe_auth: Receive result of authenticated bind request, but handle fallbacks:
1039   SPNEGO -> NTLMSSP
1040 */
1041 static void continue_auth_auto(struct composite_context *ctx)
1042 {
1043         struct composite_context *c = talloc_get_type(ctx->async.private_data,
1044                                                       struct composite_context);
1045         struct pipe_auth_state *s = talloc_get_type(c->private_data, struct pipe_auth_state);
1046         struct composite_context *sec_conn_req;
1047
1048         c->status = dcerpc_bind_auth_recv(ctx);
1049         if (NT_STATUS_EQUAL(c->status, NT_STATUS_INVALID_PARAMETER)) {
1050                 /*
1051                  * Retry with NTLMSSP auth as fallback
1052                  * send a request for secondary rpc connection
1053                  */
1054                 sec_conn_req = dcerpc_secondary_connection_send(s->pipe,
1055                                                                 s->binding);
1056                 composite_continue(c, sec_conn_req, continue_ntlmssp_connection, c);
1057                 return;
1058         } else if (NT_STATUS_EQUAL(c->status, NT_STATUS_LOGON_FAILURE)) {
1059                 if (cli_credentials_wrong_password(s->credentials)) {
1060                         /*
1061                          * Retry SPNEGO with a better password
1062                          * send a request for secondary rpc connection
1063                          */
1064                         sec_conn_req = dcerpc_secondary_connection_send(s->pipe,
1065                                                                         s->binding);
1066                         composite_continue(c, sec_conn_req, continue_spnego_after_wrong_pass, c);
1067                         return;
1068                 }
1069         }
1070
1071         if (!composite_is_ok(c)) return;
1072
1073         composite_done(c);
1074 }
1075
1076 /*
1077   Stage 3 of pipe_auth (fallback to NTLMSSP case): Receive secondary
1078   rpc connection (the first one can't be used any more, due to the
1079   bind nak) and perform authenticated bind request
1080 */
1081 static void continue_ntlmssp_connection(struct composite_context *ctx)
1082 {
1083         struct composite_context *c;
1084         struct pipe_auth_state *s;
1085         struct composite_context *auth_req;
1086         struct dcerpc_pipe *p2;
1087
1088         c = talloc_get_type(ctx->async.private_data, struct composite_context);
1089         s = talloc_get_type(c->private_data, struct pipe_auth_state);
1090
1091         /* receive secondary rpc connection */
1092         c->status = dcerpc_secondary_connection_recv(ctx, &p2);
1093         if (!composite_is_ok(c)) return;
1094
1095         talloc_steal(s, p2);
1096         talloc_steal(p2, s->pipe);
1097         s->pipe = p2;
1098
1099         /* initiate a authenticated bind */
1100         auth_req = dcerpc_bind_auth_send(c, s->pipe, s->table,
1101                                          s->credentials, DCERPC_AUTH_TYPE_NTLMSSP,
1102                                          dcerpc_auth_level(s->pipe->conn),
1103                                          s->table->authservices->names[0]);
1104         composite_continue(c, auth_req, continue_auth, c);
1105 }
1106
1107 /*
1108   Stage 3 of pipe_auth (retry on wrong password): Receive secondary
1109   rpc connection (the first one can't be used any more, due to the
1110   bind nak) and perform authenticated bind request
1111 */
1112 static void continue_spnego_after_wrong_pass(struct composite_context *ctx)
1113 {
1114         struct composite_context *c;
1115         struct pipe_auth_state *s;
1116         struct composite_context *auth_req;
1117         struct dcerpc_pipe *p2;
1118
1119         c = talloc_get_type(ctx->async.private_data, struct composite_context);
1120         s = talloc_get_type(c->private_data, struct pipe_auth_state);
1121
1122         /* receive secondary rpc connection */
1123         c->status = dcerpc_secondary_connection_recv(ctx, &p2);
1124         if (!composite_is_ok(c)) return;
1125
1126         talloc_steal(s, p2);
1127         talloc_steal(p2, s->pipe);
1128         s->pipe = p2;
1129
1130         /* initiate a authenticated bind */
1131         auth_req = dcerpc_bind_auth_send(c, s->pipe, s->table,
1132                                          s->credentials, DCERPC_AUTH_TYPE_SPNEGO,
1133                                          dcerpc_auth_level(s->pipe->conn),
1134                                          s->table->authservices->names[0]);
1135         composite_continue(c, auth_req, continue_auth, c);
1136 }
1137
1138
1139 /*
1140   Stage 2 of pipe_auth: Receive result of non-authenticated bind request
1141 */
1142 static void continue_auth_none(struct composite_context *ctx)
1143 {
1144         struct composite_context *c = talloc_get_type(ctx->async.private_data,
1145                                                       struct composite_context);
1146
1147         c->status = dcerpc_bind_auth_none_recv(ctx);
1148         if (!composite_is_ok(c)) return;
1149         
1150         composite_done(c);
1151 }
1152
1153
1154 /*
1155   Request to perform an authenticated bind if required. Authentication
1156   is determined using credentials passed and binding flags.
1157 */
1158 struct composite_context *dcerpc_pipe_auth_send(struct dcerpc_pipe *p, 
1159                                                 struct dcerpc_binding *binding,
1160                                                 const struct dcerpc_interface_table *table,
1161                                                 struct cli_credentials *credentials)
1162 {
1163         struct composite_context *c;
1164         struct pipe_auth_state *s;
1165         struct composite_context *auth_schannel_req;
1166         struct composite_context *auth_req;
1167         struct composite_context *auth_none_req;
1168         struct dcerpc_connection *conn;
1169         uint8_t auth_type;
1170
1171         /* composite context allocation and setup */
1172         c = composite_create(p, p->conn->event_ctx);
1173         if (c == NULL) return NULL;
1174
1175         s = talloc_zero(c, struct pipe_auth_state);
1176         if (composite_nomem(s, c)) return c;
1177         c->private_data = s;
1178
1179         /* store parameters in state structure */
1180         s->binding      = binding;
1181         s->table        = table;
1182         s->credentials  = credentials;
1183         s->pipe         = p;
1184
1185         conn = s->pipe->conn;
1186         conn->flags = binding->flags;
1187         
1188         /* remember the binding string for possible secondary connections */
1189         conn->binding_string = dcerpc_binding_string(p, binding);
1190
1191         if (cli_credentials_is_anonymous(s->credentials)) {
1192                 auth_none_req = dcerpc_bind_auth_none_send(c, s->pipe, s->table);
1193                 composite_continue(c, auth_none_req, continue_auth_none, c);
1194                 return c;
1195         }
1196
1197         if ((binding->flags & DCERPC_SCHANNEL) &&
1198             !cli_credentials_get_netlogon_creds(s->credentials)) {
1199                 /* If we don't already have netlogon credentials for
1200                  * the schannel bind, then we have to get these
1201                  * first */
1202                 auth_schannel_req = dcerpc_bind_auth_schannel_send(c, s->pipe, s->table,
1203                                                                    s->credentials,
1204                                                                    dcerpc_auth_level(conn));
1205                 composite_continue(c, auth_schannel_req, continue_auth_schannel, c);
1206                 return c;
1207         }
1208
1209         /*
1210          * we rely on the already authenticated CIFS connection
1211          * if not doing sign or seal
1212          */
1213         if (conn->transport.transport == NCACN_NP &&
1214             !(s->binding->flags & (DCERPC_SIGN|DCERPC_SEAL))) {
1215                 auth_none_req = dcerpc_bind_auth_none_send(c, s->pipe, s->table);
1216                 composite_continue(c, auth_none_req, continue_auth_none, c);
1217                 return c;
1218         }
1219
1220
1221         /* Perform an authenticated DCE-RPC bind
1222          */
1223         if (!(conn->flags & (DCERPC_SIGN|DCERPC_SEAL))) {
1224                 /*
1225                   we are doing an authenticated connection,
1226                   but not using sign or seal. We must force
1227                   the CONNECT dcerpc auth type as a NONE auth
1228                   type doesn't allow authentication
1229                   information to be passed.
1230                 */
1231                 conn->flags |= DCERPC_CONNECT;
1232         }
1233
1234         if (s->binding->flags & DCERPC_AUTH_SPNEGO) {
1235                 auth_type = DCERPC_AUTH_TYPE_SPNEGO;
1236
1237         } else if (s->binding->flags & DCERPC_AUTH_KRB5) {
1238                 auth_type = DCERPC_AUTH_TYPE_KRB5;
1239
1240         } else if (s->binding->flags & DCERPC_SCHANNEL) {
1241                 auth_type = DCERPC_AUTH_TYPE_SCHANNEL;
1242
1243         } else if (s->binding->flags & DCERPC_AUTH_NTLM) {
1244                 auth_type = DCERPC_AUTH_TYPE_NTLMSSP;
1245
1246         } else {
1247                 /* try SPNEGO with fallback to NTLMSSP */
1248                 auth_req = dcerpc_bind_auth_send(c, s->pipe, s->table,
1249                                                  s->credentials, DCERPC_AUTH_TYPE_SPNEGO,
1250                                                  dcerpc_auth_level(conn),
1251                                                  s->table->authservices->names[0]);
1252                 composite_continue(c, auth_req, continue_auth_auto, c);
1253                 return c;
1254         }
1255
1256         auth_req = dcerpc_bind_auth_send(c, s->pipe, s->table,
1257                                          s->credentials, auth_type,
1258                                          dcerpc_auth_level(conn),
1259                                          s->table->authservices->names[0]);
1260         composite_continue(c, auth_req, continue_auth, c);
1261         return c;
1262 }
1263
1264
1265 /*
1266   Receive result of authenticated bind request on dcerpc pipe
1267
1268   This returns *p, which may be different to the one originally
1269   supllied, as it rebinds to a new pipe due to authentication fallback
1270
1271 */
1272 NTSTATUS dcerpc_pipe_auth_recv(struct composite_context *c, TALLOC_CTX *mem_ctx, 
1273                                struct dcerpc_pipe **p)
1274 {
1275         NTSTATUS status;
1276
1277         struct pipe_auth_state *s = talloc_get_type(c->private_data,
1278                                                     struct pipe_auth_state);
1279         status = composite_wait(c);
1280         if (!NT_STATUS_IS_OK(status)) {
1281                 char *uuid_str = GUID_string(s->pipe, &s->table->syntax_id.uuid);
1282                 DEBUG(0, ("Failed to bind to uuid %s - %s\n", uuid_str, nt_errstr(status)));
1283                 talloc_free(uuid_str);
1284         } else {
1285                 talloc_steal(mem_ctx, s->pipe);
1286                 *p = s->pipe;
1287         }
1288
1289         talloc_free(c);
1290         return status;
1291 }
1292
1293
1294 /* 
1295    Perform an authenticated bind if needed - sync version
1296
1297    This may change *p, as it rebinds to a new pipe due to authentication fallback
1298 */
1299 NTSTATUS dcerpc_pipe_auth(TALLOC_CTX *mem_ctx,
1300                           struct dcerpc_pipe **p, 
1301                           struct dcerpc_binding *binding,
1302                           const struct dcerpc_interface_table *table,
1303                           struct cli_credentials *credentials)
1304 {
1305         struct composite_context *c;
1306
1307         c = dcerpc_pipe_auth_send(*p, binding, table, credentials);
1308         return dcerpc_pipe_auth_recv(c, mem_ctx, p);
1309 }
1310
1311
1312 NTSTATUS dcerpc_generic_session_key(struct dcerpc_connection *c,
1313                                     DATA_BLOB *session_key)
1314 {
1315         /* this took quite a few CPU cycles to find ... */
1316         session_key->data = discard_const_p(unsigned char, "SystemLibraryDTC");
1317         session_key->length = 16;
1318         return NT_STATUS_OK;
1319 }
1320
1321 /*
1322   fetch the user session key - may be default (above) or the SMB session key
1323 */
1324 NTSTATUS dcerpc_fetch_session_key(struct dcerpc_pipe *p,
1325                                   DATA_BLOB *session_key)
1326 {
1327         return p->conn->security_state.session_key(p->conn, session_key);
1328 }
1329
1330
1331 /*
1332   log a rpc packet in a format suitable for ndrdump. This is especially useful
1333   for sealed packets, where ethereal cannot easily see the contents
1334
1335   this triggers on a debug level of >= 10
1336 */
1337 void dcerpc_log_packet(const struct dcerpc_interface_table *ndr,
1338                        uint32_t opnum, uint32_t flags, DATA_BLOB *pkt)
1339 {
1340         const int num_examples = 20;
1341         int i;
1342
1343         if (DEBUGLEVEL < 10) return;
1344
1345         for (i=0;i<num_examples;i++) {
1346                 char *name=NULL;
1347                 asprintf(&name, "%s/rpclog/%s-%u.%d.%s", 
1348                          lp_lockdir(), ndr->name, opnum, i,
1349                          (flags&NDR_IN)?"in":"out");
1350                 if (name == NULL) {
1351                         return;
1352                 }
1353                 if (!file_exist(name)) {
1354                         if (file_save(name, pkt->data, pkt->length)) {
1355                                 DEBUG(10,("Logged rpc packet to %s\n", name));
1356                         }
1357                         free(name);
1358                         break;
1359                 }
1360                 free(name);
1361         }
1362 }
1363
1364
1365
1366 /*
1367   create a secondary context from a primary connection
1368
1369   this uses dcerpc_alter_context() to create a new dcerpc context_id
1370 */
1371 NTSTATUS dcerpc_secondary_context(struct dcerpc_pipe *p, 
1372                                   struct dcerpc_pipe **pp2,
1373                                   const struct dcerpc_interface_table *table)
1374 {
1375         NTSTATUS status;
1376         struct dcerpc_pipe *p2;
1377         
1378         p2 = talloc_zero(p, struct dcerpc_pipe);
1379         if (p2 == NULL) {
1380                 return NT_STATUS_NO_MEMORY;
1381         }
1382         p2->conn = talloc_reference(p2, p->conn);
1383         p2->request_timeout = p->request_timeout;
1384
1385         p2->context_id = ++p->conn->next_context_id;
1386
1387         p2->syntax = table->syntax_id;
1388
1389         p2->transfer_syntax = ndr_transfer_syntax;
1390
1391         p2->binding = talloc_reference(p2, p->binding);
1392
1393         status = dcerpc_alter_context(p2, p2, &p2->syntax, &p2->transfer_syntax);
1394         if (!NT_STATUS_IS_OK(status)) {
1395                 talloc_free(p2);
1396                 return status;
1397         }
1398
1399         *pp2 = p2;
1400
1401         return status;
1402 }