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