2 Unix SMB/CIFS implementation.
5 Copyright (C) Tim Potter 2003
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 "rpcclient.h"
23 #include "../librpc/gen_ndr/cli_echo.h"
25 static NTSTATUS cmd_echo_add_one(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
26 int argc, const char **argv)
28 uint32 request = 1, response;
32 printf("Usage: %s [num]\n", argv[0]);
37 request = atoi(argv[1]);
39 result = rpccli_echo_AddOne(cli, mem_ctx, request, &response);
41 if (!NT_STATUS_IS_OK(result))
44 printf("%d + 1 = %d\n", request, response);
50 static NTSTATUS cmd_echo_data(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
51 int argc, const char **argv)
55 uint8_t *in_data = NULL, *out_data = NULL;
58 printf("Usage: %s num\n", argv[0]);
63 if ( (in_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
64 printf("Failure to allocate buff of %d bytes\n",
66 result = NT_STATUS_NO_MEMORY;
69 if ( (out_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
70 printf("Failure to allocate buff of %d bytes\n",
72 result = NT_STATUS_NO_MEMORY;
76 for (i = 0; i < size; i++)
77 in_data[i] = i & 0xff;
79 result = rpccli_echo_EchoData(cli, mem_ctx, size, in_data, out_data);
81 if (!NT_STATUS_IS_OK(result))
84 for (i = 0; i < size; i++) {
85 if (in_data[i] != out_data[i]) {
86 printf("mismatch at offset %d, %d != %d\n",
87 i, in_data[i], out_data[i]);
88 result = NT_STATUS_UNSUCCESSFUL;
99 static NTSTATUS cmd_echo_source_data(struct rpc_pipe_client *cli,
100 TALLOC_CTX *mem_ctx, int argc,
105 uint8_t *out_data = NULL;
108 printf("Usage: %s num\n", argv[0]);
112 size = atoi(argv[1]);
113 if ( (out_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
114 printf("Failure to allocate buff of %d bytes\n",
116 result = NT_STATUS_NO_MEMORY;
121 result = rpccli_echo_SourceData(cli, mem_ctx, size, out_data);
123 if (!NT_STATUS_IS_OK(result))
126 for (i = 0; i < size; i++) {
127 if (out_data && out_data[i] != (i & 0xff)) {
128 printf("mismatch at offset %d, %d != %d\n",
129 i, out_data[i], i & 0xff);
130 result = NT_STATUS_UNSUCCESSFUL;
140 static NTSTATUS cmd_echo_sink_data(struct rpc_pipe_client *cli, TALLOC_CTX *mem_ctx,
141 int argc, const char **argv)
145 uint8_t *in_data = NULL;
148 printf("Usage: %s num\n", argv[0]);
152 size = atoi(argv[1]);
153 if ( (in_data = (uint8_t*)SMB_MALLOC(size)) == NULL ) {
154 printf("Failure to allocate buff of %d bytes\n",
156 result = NT_STATUS_NO_MEMORY;
160 for (i = 0; i < size; i++)
161 in_data[i] = i & 0xff;
163 result = rpccli_echo_SinkData(cli, mem_ctx, size, in_data);
165 if (!NT_STATUS_IS_OK(result))
174 /* List of commands exported by this module */
176 struct cmd_set echo_commands[] = {
180 { "echoaddone", RPC_RTYPE_NTSTATUS, cmd_echo_add_one, NULL, &ndr_table_rpcecho.syntax_id, NULL, "Add one to a number", "" },
181 { "echodata", RPC_RTYPE_NTSTATUS, cmd_echo_data, NULL, &ndr_table_rpcecho.syntax_id, NULL, "Echo data", "" },
182 { "sinkdata", RPC_RTYPE_NTSTATUS, cmd_echo_sink_data, NULL, &ndr_table_rpcecho.syntax_id, NULL, "Sink data", "" },
183 { "sourcedata", RPC_RTYPE_NTSTATUS, cmd_echo_source_data, NULL, &ndr_table_rpcecho.syntax_id, NULL, "Source data", "" },