1/*
2   Unix SMB/CIFS implementation.
3   RPC pipe client
4
5   Copyright (C) Tim Potter 2003
6
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 2 of the License, or
10   (at your option) any later version.
11
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.
16
17   You should have received a copy of the GNU General Public License
18   along with this program; if not, write to the Free Software
19   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20*/
21
22#include "includes.h"
23#include "rpcclient.h"
24
25static NTSTATUS cmd_echo_add_one(struct cli_state *cli, TALLOC_CTX *mem_ctx,
26				 int argc, const char **argv)
27{
28	uint32 request = 1, response;
29	NTSTATUS result;
30
31	if (argc > 2) {
32		printf("Usage: %s [num]\n", argv[0]);
33		return NT_STATUS_OK;
34	}
35
36	if (argc == 2)
37		request = atoi(argv[1]);
38
39	result = cli_echo_add_one(cli, mem_ctx, request, &response);
40
41	if (!NT_STATUS_IS_OK(result))
42		goto done;
43
44	printf("%d + 1 = %d\n", request, response);
45
46done:
47	return result;
48}
49
50static NTSTATUS cmd_echo_data(struct cli_state *cli, TALLOC_CTX *mem_ctx,
51			      int argc, const char **argv)
52{
53	uint32 size, i;
54	NTSTATUS result;
55	char *in_data = NULL, *out_data = NULL;
56
57	if (argc != 2) {
58		printf("Usage: %s num\n", argv[0]);
59		return NT_STATUS_OK;
60	}
61
62	size = atoi(argv[1]);
63	in_data = SMB_MALLOC(size);
64
65	for (i = 0; i < size; i++)
66		in_data[i] = i & 0xff;
67
68	result = cli_echo_data(cli, mem_ctx, size, in_data, &out_data);
69
70	if (!NT_STATUS_IS_OK(result))
71		goto done;
72
73	for (i = 0; i < size; i++) {
74		if (in_data[i] != out_data[i]) {
75			printf("mismatch at offset %d, %d != %d\n",
76			       i, in_data[i], out_data[i]);
77			result = NT_STATUS_UNSUCCESSFUL;
78		}
79	}
80
81done:
82	SAFE_FREE(in_data);
83
84	return result;
85}
86
87static NTSTATUS cmd_echo_source_data(struct cli_state *cli,
88				     TALLOC_CTX *mem_ctx, int argc,
89				     const char **argv)
90{
91	uint32 size, i;
92	NTSTATUS result;
93	char *out_data = NULL;
94
95	if (argc != 2) {
96		printf("Usage: %s num\n", argv[0]);
97		return NT_STATUS_OK;
98	}
99
100	size = atoi(argv[1]);
101
102	result = cli_echo_source_data(cli, mem_ctx, size, &out_data);
103
104	if (!NT_STATUS_IS_OK(result))
105		goto done;
106
107	for (i = 0; i < size; i++) {
108		if (out_data && out_data[i] != (i & 0xff)) {
109			printf("mismatch at offset %d, %d != %d\n",
110			       i, out_data[i], i & 0xff);
111			result = NT_STATUS_UNSUCCESSFUL;
112		}
113	}
114
115done:
116	return result;
117}
118
119static NTSTATUS cmd_echo_sink_data(struct cli_state *cli, TALLOC_CTX *mem_ctx,
120				   int argc, const char **argv)
121{
122	uint32 size, i;
123	NTSTATUS result;
124	char *in_data = NULL;
125
126	if (argc != 2) {
127		printf("Usage: %s num\n", argv[0]);
128		return NT_STATUS_OK;
129	}
130
131	size = atoi(argv[1]);
132	in_data = SMB_MALLOC(size);
133
134	for (i = 0; i < size; i++)
135		in_data[i] = i & 0xff;
136
137	result = cli_echo_sink_data(cli, mem_ctx, size, in_data);
138
139	if (!NT_STATUS_IS_OK(result))
140		goto done;
141
142done:
143	SAFE_FREE(in_data);
144
145	return result;
146}
147
148/* List of commands exported by this module */
149
150struct cmd_set echo_commands[] = {
151
152	{ "ECHO" },
153
154	{ "echoaddone", RPC_RTYPE_NTSTATUS, cmd_echo_add_one,     NULL, PI_ECHO, "Add one to a number", "" },
155	{ "echodata",   RPC_RTYPE_NTSTATUS, cmd_echo_data,        NULL, PI_ECHO, "Echo data",           "" },
156	{ "sinkdata",   RPC_RTYPE_NTSTATUS, cmd_echo_sink_data,   NULL, PI_ECHO, "Sink data",           "" },
157	{ "sourcedata", RPC_RTYPE_NTSTATUS, cmd_echo_source_data, NULL, PI_ECHO, "Source data",         "" },
158	{ NULL }
159};
160