2 Unix SMB/CIFS implementation.
4 Copyright (C) Andrew Tridgell 2003
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 #include "lib/cmdline/popt_common.h"
23 #include "system/filesys.h"
24 #include "system/locale.h"
25 #include "librpc/rpc/dcerpc.h"
26 #include "librpc/rpc/dcerpc_table.h"
28 static const struct dcerpc_interface_call
*find_function(
29 const struct dcerpc_interface_table
*p
,
33 if (isdigit(function
[0])) {
34 i
= strtol(function
, NULL
, 0);
37 for (i
=0;i
<p
->num_calls
;i
++) {
38 if (strcmp(p
->calls
[i
].name
, function
) == 0) {
42 if (i
== p
->num_calls
) {
43 printf("Function '%s' not found\n", function
);
50 static void show_pipes(void)
52 const struct dcerpc_interface_list
*l
;
53 printf("\nYou must specify a pipe\n");
54 printf("known pipes are:\n");
55 for (l
=librpc_dcerpc_pipes();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 static void show_functions(const struct dcerpc_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
= (char *) talloc_realloc(
86 mem_ctx
, result
, char *, total_len
+ num_read
);
88 result
= talloc_size(mem_ctx
, num_read
);
91 memcpy(result
+ total_len
, buf
, num_read
);
93 total_len
+= num_read
;
102 const struct dcerpc_interface_table
*load_iface_from_plugin(const char *plugin
, const char *pipe_name
)
104 const struct dcerpc_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
, "dcerpc_table_%s", pipe_name
);
115 p
= dlsym(handle
, symbol
);
118 printf("%s: Unable to find DCE/RPC interface table for '%s': %s\n", plugin
, pipe_name
, dlerror());
128 int main(int argc
, const char *argv
[])
130 const struct dcerpc_interface_table
*p
= NULL
;
131 const struct dcerpc_interface_call
*f
;
132 const char *pipe_name
, *function
, *inout
, *filename
;
136 struct ndr_pull
*ndr_pull
;
137 struct ndr_print
*ndr_print
;
144 const char *ctx_filename
= NULL
;
145 const char *plugin
= NULL
;
146 BOOL validate
= False
;
147 BOOL dumpdata
= False
;
149 struct poptOption long_options
[] = {
150 {"context-file", 'c', POPT_ARG_STRING
, &ctx_filename
, 0, "In-filename to parse first", "CTX-FILE" },
151 {"validate", 0, POPT_ARG_NONE
, &validate
, 0, "try to validate the data", NULL
},
152 {"dump-data", 0, POPT_ARG_NONE
, &dumpdata
, 0, "dump the hex data", NULL
},
153 {"load-dso", 'l', POPT_ARG_STRING
, &plugin
, 0, "load from shared object file", NULL
},
161 pc
= poptGetContext("ndrdump", argc
, argv
, long_options
, 0);
163 poptSetOtherOptionHelp(
164 pc
, "<pipe|uuid> <function> <inout> [<filename>]");
166 while ((opt
= poptGetNextOpt(pc
)) != -1) {
169 pipe_name
= poptGetArg(pc
);
172 poptPrintUsage(pc
, stderr
, 0);
177 if (plugin
!= NULL
) {
178 p
= load_iface_from_plugin(plugin
, pipe_name
);
182 p
= idl_iface_by_name(pipe_name
);
188 status
= GUID_from_string(pipe_name
, &uuid
);
190 if (NT_STATUS_IS_OK(status
)) {
191 p
= idl_iface_by_uuid(&uuid
);
196 printf("Unknown pipe or UUID '%s'\n", pipe_name
);
200 function
= poptGetArg(pc
);
201 inout
= poptGetArg(pc
);
202 filename
= poptGetArg(pc
);
204 if (!function
|| !inout
) {
205 poptPrintUsage(pc
, stderr
, 0);
210 if (strcmp(inout
, "in") == 0 ||
211 strcmp(inout
, "request") == 0) {
213 } else if (strcmp(inout
, "out") == 0 ||
214 strcmp(inout
, "response") == 0) {
217 printf("Bad inout value '%s'\n", inout
);
221 f
= find_function(p
, function
);
223 mem_ctx
= talloc_init("ndrdump");
225 st
= talloc_zero_size(mem_ctx
, f
->struct_size
);
227 printf("Unable to allocate %d bytes\n", (int)f
->struct_size
);
231 v_st
= talloc_zero_size(mem_ctx
, f
->struct_size
);
233 printf("Unable to allocate %d bytes\n", (int)f
->struct_size
);
238 if (flags
== NDR_IN
) {
239 printf("Context file can only be used for \"out\" packages\n");
243 data
= (uint8_t *)file_load(ctx_filename
, &size
, mem_ctx
);
245 perror(ctx_filename
);
252 ndr_pull
= ndr_pull_init_blob(&blob
, mem_ctx
);
253 ndr_pull
->flags
|= LIBNDR_FLAG_REF_ALLOC
;
255 status
= f
->ndr_pull(ndr_pull
, NDR_IN
, st
);
257 if (ndr_pull
->offset
!= ndr_pull
->data_size
) {
258 printf("WARNING! %d unread bytes while parsing context file\n", ndr_pull
->data_size
- ndr_pull
->offset
);
261 if (!NT_STATUS_IS_OK(status
)) {
262 printf("pull for context file returned %s\n", nt_errstr(status
));
265 memcpy(v_st
, st
, f
->struct_size
);
269 data
= (uint8_t *)file_load(filename
, &size
, mem_ctx
);
271 data
= (uint8_t *)stdin_load(mem_ctx
, &size
);
284 ndr_pull
= ndr_pull_init_blob(&blob
, mem_ctx
);
285 ndr_pull
->flags
|= LIBNDR_FLAG_REF_ALLOC
;
287 status
= f
->ndr_pull(ndr_pull
, flags
, st
);
289 printf("pull returned %s\n", nt_errstr(status
));
291 if (ndr_pull
->offset
!= ndr_pull
->data_size
) {
292 printf("WARNING! %d unread bytes\n", ndr_pull
->data_size
- ndr_pull
->offset
);
293 dump_data(0, ndr_pull
->data
+ndr_pull
->offset
, ndr_pull
->data_size
- ndr_pull
->offset
);
297 printf("%d bytes consumed\n", ndr_pull
->offset
);
298 dump_data(0, blob
.data
, blob
.length
);
301 ndr_print
= talloc_zero(mem_ctx
, struct ndr_print
);
302 ndr_print
->print
= ndr_print_debug_helper
;
303 ndr_print
->depth
= 1;
304 f
->ndr_print(ndr_print
, function
, flags
, st
);
306 if (!NT_STATUS_IS_OK(status
)) {
307 printf("dump FAILED\n");
313 struct ndr_push
*ndr_v_push
;
314 struct ndr_pull
*ndr_v_pull
;
315 struct ndr_print
*ndr_v_print
;
317 ndr_v_push
= ndr_push_init_ctx(mem_ctx
);
319 status
= f
->ndr_push(ndr_v_push
, flags
, st
);
320 if (!NT_STATUS_IS_OK(status
)) {
321 printf("validate push FAILED\n");
325 v_blob
= ndr_push_blob(ndr_v_push
);
328 printf("%ld bytes generated (validate)\n", (long)v_blob
.length
);
329 dump_data(0, v_blob
.data
, v_blob
.length
);
332 ndr_v_pull
= ndr_pull_init_blob(&v_blob
, mem_ctx
);
333 ndr_v_pull
->flags
|= LIBNDR_FLAG_REF_ALLOC
;
335 status
= f
->ndr_pull(ndr_v_pull
, flags
, v_st
);
336 if (!NT_STATUS_IS_OK(status
)) {
337 printf("validate pull FAILED\n");
341 printf("pull returned %s\n", nt_errstr(status
));
343 if (ndr_v_pull
->offset
!= ndr_v_pull
->data_size
) {
344 printf("WARNING! %d unread bytes in validation\n", ndr_v_pull
->data_size
- ndr_v_pull
->offset
);
345 dump_data(0, ndr_v_pull
->data
+ndr_v_pull
->offset
, ndr_v_pull
->data_size
- ndr_v_pull
->offset
);
348 ndr_v_print
= talloc_zero(mem_ctx
, struct ndr_print
);
349 ndr_v_print
->print
= ndr_print_debug_helper
;
350 ndr_v_print
->depth
= 1;
351 f
->ndr_print(ndr_v_print
, function
, flags
, v_st
);
353 if (blob
.length
!= v_blob
.length
) {
354 printf("WARNING! orig bytes:%ld validated pushed bytes:%ld\n", (long)blob
.length
, (long)v_blob
.length
);
357 if (ndr_pull
->offset
!= ndr_v_pull
->offset
) {
358 printf("WARNING! orig pulled bytes:%d validated pulled bytes:%d\n", ndr_pull
->offset
, ndr_v_pull
->offset
);
364 talloc_free(mem_ctx
);