2 Unix SMB/CIFS implementation.
4 Copyright (C) Andrew Tridgell 2003
5 Copyright (C) Jelmer Vernooij 2006
7 This program is free software; you can redistribute it and/or modify
8 it under the terms of the GNU General Public License as published by
9 the Free Software Foundation; either version 3 of the License, or
10 (at your option) any later version.
12 This program is distributed in the hope that it will be useful,
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 GNU General Public License for more details.
17 You should have received a copy of the GNU General Public License
18 along with this program. If not, see <http://www.gnu.org/licenses/>.
22 #include "system/filesys.h"
23 #include "system/locale.h"
24 #include "librpc/ndr/libndr.h"
25 #include "librpc/ndr/ndr_table.h"
26 #include "lib/cmdline/popt_common.h"
27 #include "param/param.h"
29 static const struct ndr_interface_call
*find_function(
30 const struct ndr_interface_table
*p
,
34 if (isdigit(function
[0])) {
35 i
= strtol(function
, NULL
, 0);
38 for (i
=0;i
<p
->num_calls
;i
++) {
39 if (strcmp(p
->calls
[i
].name
, function
) == 0) {
43 if (i
== p
->num_calls
) {
44 printf("Function '%s' not found\n", function
);
50 _NORETURN_
static void show_pipes(void)
52 const struct ndr_interface_list
*l
;
53 printf("\nYou must specify a pipe\n");
54 printf("known pipes are:\n");
55 for (l
=ndr_table_list();l
;l
=l
->next
) {
56 if(l
->table
->helpstring
) {
57 printf("\t%s - %s\n", l
->table
->name
, l
->table
->helpstring
);
59 printf("\t%s\n", l
->table
->name
);
65 _NORETURN_
static void show_functions(const struct ndr_interface_table
*p
)
68 printf("\nYou must specify a function\n");
69 printf("known functions on '%s' are:\n", p
->name
);
70 for (i
=0;i
<p
->num_calls
;i
++) {
71 printf("\t0x%02x (%2d) %s\n", i
, i
, p
->calls
[i
].name
);
76 static char *stdin_load(TALLOC_CTX
*mem_ctx
, size_t *size
)
78 int num_read
, total_len
= 0;
82 while((num_read
= read(STDIN_FILENO
, buf
, 255)) > 0) {
85 result
= talloc_realloc(
86 mem_ctx
, result
, char, total_len
+ num_read
);
88 result
= talloc_array(mem_ctx
, char, num_read
);
91 memcpy(result
+ total_len
, buf
, num_read
);
93 total_len
+= num_read
;
102 static const struct ndr_interface_table
*load_iface_from_plugin(const char *plugin
, const char *pipe_name
)
104 const struct ndr_interface_table
*p
;
108 handle
= dlopen(plugin
, RTLD_NOW
);
109 if (handle
== NULL
) {
110 printf("%s: Unable to open: %s\n", plugin
, dlerror());
114 symbol
= talloc_asprintf(NULL
, "ndr_table_%s", pipe_name
);
115 p
= (const struct ndr_interface_table
*)dlsym(handle
, symbol
);
118 printf("%s: Unable to find DCE/RPC interface table for '%s': %s\n", plugin
, pipe_name
, dlerror());
128 static void ndrdump_data(uint8_t *d
, uint32_t l
, bool force
)
130 dump_data_file(d
, l
, !force
, stdout
);
133 static NTSTATUS
ndrdump_pull_and_print_pipes(const char *function
,
134 struct ndr_pull
*ndr_pull
,
135 struct ndr_print
*ndr_print
,
136 const struct ndr_interface_call_pipes
*pipes
)
139 enum ndr_err_code ndr_err
;
142 for (i
=0; i
< pipes
->num_pipes
; i
++) {
149 c
= talloc_zero_size(ndr_pull
, pipes
->pipes
[i
].chunk_struct_size
);
150 talloc_set_name(c
, "struct %s", pipes
->pipes
[i
].name
);
152 * Note: the first struct member is always
155 count
= (uint32_t *)c
;
157 n
= talloc_asprintf(c
, "%s: %s[%llu]",
158 function
, pipes
->pipes
[i
].name
,
159 (unsigned long long)idx
);
161 ndr_err
= pipes
->pipes
[i
].ndr_pull(ndr_pull
, NDR_SCALARS
, c
);
162 status
= ndr_map_error2ntstatus(ndr_err
);
164 printf("pull returned %s\n", nt_errstr(status
));
165 if (!NT_STATUS_IS_OK(status
)) {
168 pipes
->pipes
[i
].ndr_print(ndr_print
, n
, c
);
180 int main(int argc
, const char *argv
[])
182 const struct ndr_interface_table
*p
= NULL
;
183 const struct ndr_interface_call
*f
;
184 const char *pipe_name
, *function
, *inout
, *filename
;
188 struct ndr_pull
*ndr_pull
;
189 struct ndr_print
*ndr_print
;
194 enum ndr_err_code ndr_err
;
197 const char *ctx_filename
= NULL
;
198 const char *plugin
= NULL
;
199 bool validate
= false;
200 bool dumpdata
= false;
201 bool assume_ndr64
= false;
203 enum {OPT_CONTEXT_FILE
=1000, OPT_VALIDATE
, OPT_DUMP_DATA
, OPT_LOAD_DSO
, OPT_NDR64
};
204 struct poptOption long_options
[] = {
206 {"context-file", 'c', POPT_ARG_STRING
, NULL
, OPT_CONTEXT_FILE
, "In-filename to parse first", "CTX-FILE" },
207 {"validate", 0, POPT_ARG_NONE
, NULL
, OPT_VALIDATE
, "try to validate the data", NULL
},
208 {"dump-data", 0, POPT_ARG_NONE
, NULL
, OPT_DUMP_DATA
, "dump the hex data", NULL
},
209 {"load-dso", 'l', POPT_ARG_STRING
, NULL
, OPT_LOAD_DSO
, "load from shared object file", NULL
},
210 {"ndr64", 0, POPT_ARG_NONE
, NULL
, OPT_NDR64
, "Assume NDR64 data", NULL
},
215 const struct ndr_interface_call_pipes
*in_pipes
= NULL
;
216 const struct ndr_interface_call_pipes
*out_pipes
= NULL
;
217 uint32_t highest_ofs
;
221 /* Initialise samba stuff */
226 setup_logging("ndrdump", DEBUG_STDOUT
);
228 pc
= poptGetContext("ndrdump", argc
, argv
, long_options
, 0);
230 poptSetOtherOptionHelp(
231 pc
, "<pipe|uuid> <function> <inout> [<filename>]");
233 while ((opt
= poptGetNextOpt(pc
)) != -1) {
235 case OPT_CONTEXT_FILE
:
236 ctx_filename
= poptGetOptArg(pc
);
245 plugin
= poptGetOptArg(pc
);
253 pipe_name
= poptGetArg(pc
);
256 poptPrintUsage(pc
, stderr
, 0);
261 if (plugin
!= NULL
) {
262 p
= load_iface_from_plugin(plugin
, pipe_name
);
265 p
= ndr_table_by_name(pipe_name
);
271 status
= GUID_from_string(pipe_name
, &uuid
);
273 if (NT_STATUS_IS_OK(status
)) {
274 p
= ndr_table_by_uuid(&uuid
);
279 printf("Unknown pipe or UUID '%s'\n", pipe_name
);
283 function
= poptGetArg(pc
);
284 inout
= poptGetArg(pc
);
285 filename
= poptGetArg(pc
);
287 if (!function
|| !inout
) {
288 poptPrintUsage(pc
, stderr
, 0);
293 f
= find_function(p
, function
);
295 if (strcmp(inout
, "in") == 0 ||
296 strcmp(inout
, "request") == 0) {
298 in_pipes
= &f
->in_pipes
;
299 } else if (strcmp(inout
, "out") == 0 ||
300 strcmp(inout
, "response") == 0) {
302 out_pipes
= &f
->out_pipes
;
304 printf("Bad inout value '%s'\n", inout
);
308 mem_ctx
= talloc_init("ndrdump");
310 st
= talloc_zero_size(mem_ctx
, f
->struct_size
);
312 printf("Unable to allocate %d bytes\n", (int)f
->struct_size
);
316 v_st
= talloc_zero_size(mem_ctx
, f
->struct_size
);
318 printf("Unable to allocate %d bytes\n", (int)f
->struct_size
);
323 if (flags
== NDR_IN
) {
324 printf("Context file can only be used for \"out\" packages\n");
328 data
= (uint8_t *)file_load(ctx_filename
, &size
, 0, mem_ctx
);
330 perror(ctx_filename
);
337 ndr_pull
= ndr_pull_init_blob(&blob
, mem_ctx
);
338 ndr_pull
->flags
|= LIBNDR_FLAG_REF_ALLOC
;
340 ndr_pull
->flags
|= LIBNDR_FLAG_NDR64
;
343 ndr_err
= f
->ndr_pull(ndr_pull
, NDR_IN
, st
);
345 if (ndr_pull
->offset
> ndr_pull
->relative_highest_offset
) {
346 highest_ofs
= ndr_pull
->offset
;
348 highest_ofs
= ndr_pull
->relative_highest_offset
;
351 if (highest_ofs
!= ndr_pull
->data_size
) {
352 printf("WARNING! %d unread bytes while parsing context file\n", ndr_pull
->data_size
- highest_ofs
);
355 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err
)) {
356 status
= ndr_map_error2ntstatus(ndr_err
);
357 printf("pull for context file returned %s\n", nt_errstr(status
));
360 memcpy(v_st
, st
, f
->struct_size
);
364 data
= (uint8_t *)file_load(filename
, &size
, 0, mem_ctx
);
366 data
= (uint8_t *)stdin_load(mem_ctx
, &size
);
379 ndr_pull
= ndr_pull_init_blob(&blob
, mem_ctx
);
380 ndr_pull
->flags
|= LIBNDR_FLAG_REF_ALLOC
;
382 ndr_pull
->flags
|= LIBNDR_FLAG_NDR64
;
385 ndr_print
= talloc_zero(mem_ctx
, struct ndr_print
);
386 ndr_print
->print
= ndr_print_printf_helper
;
387 ndr_print
->depth
= 1;
390 status
= ndrdump_pull_and_print_pipes(function
, ndr_pull
, ndr_print
, out_pipes
);
391 if (!NT_STATUS_IS_OK(status
)) {
392 printf("dump FAILED\n");
397 ndr_err
= f
->ndr_pull(ndr_pull
, flags
, st
);
398 status
= ndr_map_error2ntstatus(ndr_err
);
400 printf("pull returned %s\n", nt_errstr(status
));
402 if (ndr_pull
->offset
> ndr_pull
->relative_highest_offset
) {
403 highest_ofs
= ndr_pull
->offset
;
405 highest_ofs
= ndr_pull
->relative_highest_offset
;
408 if (highest_ofs
!= ndr_pull
->data_size
) {
409 printf("WARNING! %d unread bytes\n", ndr_pull
->data_size
- highest_ofs
);
410 ndrdump_data(ndr_pull
->data
+highest_ofs
,
411 ndr_pull
->data_size
- highest_ofs
,
416 printf("%d bytes consumed\n", ndr_pull
->offset
);
417 ndrdump_data(blob
.data
, blob
.length
, dumpdata
);
420 f
->ndr_print(ndr_print
, function
, flags
, st
);
422 if (!NT_STATUS_IS_OK(status
)) {
423 printf("dump FAILED\n");
428 status
= ndrdump_pull_and_print_pipes(function
, ndr_pull
, ndr_print
, in_pipes
);
429 if (!NT_STATUS_IS_OK(status
)) {
430 printf("dump FAILED\n");
437 struct ndr_push
*ndr_v_push
;
438 struct ndr_pull
*ndr_v_pull
;
439 struct ndr_print
*ndr_v_print
;
441 uint8_t byte_a
, byte_b
;
444 ndr_v_push
= ndr_push_init_ctx(mem_ctx
);
446 ndr_err
= f
->ndr_push(ndr_v_push
, flags
, st
);
447 status
= ndr_map_error2ntstatus(ndr_err
);
448 printf("push returned %s\n", nt_errstr(status
));
449 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err
)) {
450 printf("validate push FAILED\n");
454 v_blob
= ndr_push_blob(ndr_v_push
);
457 printf("%ld bytes generated (validate)\n", (long)v_blob
.length
);
458 ndrdump_data(v_blob
.data
, v_blob
.length
, dumpdata
);
461 ndr_v_pull
= ndr_pull_init_blob(&v_blob
, mem_ctx
);
462 ndr_v_pull
->flags
|= LIBNDR_FLAG_REF_ALLOC
;
464 ndr_err
= f
->ndr_pull(ndr_v_pull
, flags
, v_st
);
465 status
= ndr_map_error2ntstatus(ndr_err
);
466 printf("pull returned %s\n", nt_errstr(status
));
467 if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err
)) {
468 printf("validate pull FAILED\n");
473 if (ndr_v_pull
->offset
!= ndr_v_pull
->data_size
) {
474 printf("WARNING! %d unread bytes in validation\n", ndr_v_pull
->data_size
- ndr_v_pull
->offset
);
475 ndrdump_data(ndr_v_pull
->data
+ndr_v_pull
->offset
,
476 ndr_v_pull
->data_size
- ndr_v_pull
->offset
,
480 ndr_v_print
= talloc_zero(mem_ctx
, struct ndr_print
);
481 ndr_v_print
->print
= ndr_print_debug_helper
;
482 ndr_v_print
->depth
= 1;
483 f
->ndr_print(ndr_v_print
, function
, flags
, v_st
);
485 if (blob
.length
!= v_blob
.length
) {
486 printf("WARNING! orig bytes:%llu validated pushed bytes:%llu\n",
487 (unsigned long long)blob
.length
, (unsigned long long)v_blob
.length
);
490 if (ndr_pull
->offset
!= ndr_v_pull
->offset
) {
491 printf("WARNING! orig pulled bytes:%llu validated pulled bytes:%llu\n",
492 (unsigned long long)ndr_pull
->offset
, (unsigned long long)ndr_v_pull
->offset
);
498 for (i
=0; i
< blob
.length
; i
++) {
499 byte_a
= blob
.data
[i
];
501 if (i
== v_blob
.length
) {
507 byte_b
= v_blob
.data
[i
];
509 if (byte_a
!= byte_b
) {
515 printf("WARNING! orig and validated differ at byte 0x%02X (%u)\n", i
, i
);
516 printf("WARNING! orig byte[0x%02X] = 0x%02X validated byte[0x%02X] = 0x%02X\n",
517 i
, byte_a
, i
, byte_b
);
523 talloc_free(mem_ctx
);