1724f482c388c15350609183e158f57d88b629fb
[samba.git] / testprogs / win32 / rpcecho / client.c
1 /* 
2    RPC echo client.
3
4    Copyright (C) Tim Potter 2003
5    
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.
10    
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.
15    
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.
19 */
20
21 #include <stdio.h>
22 #include <stdlib.h>
23 #include <ctype.h>
24 #include "rpcecho.h"
25
26 void main(int argc, char **argv)
27 {
28         RPC_STATUS status;
29         char *binding = NULL;
30         const char *username=NULL;
31         const char *password=NULL;
32         const char *domain=NULL;
33         unsigned sec_options = 0;
34
35         argv += 1;
36         argc -= 1;
37
38         while (argc > 2 && argv[0][0] == '-') {
39                 const char *option;
40
41                 switch (argv[0][1]) {
42                 case 'e':
43                         binding = argv[1];
44                         break;
45                 case 'u':
46                         username = argv[1];
47                         break;
48                 case 'p':
49                         password = argv[1];
50                         break;
51                 case 'd':
52                         domain = argv[1];
53                         break;
54                 case '-':
55                         option = &argv[0][2];
56                         if (strcmp(option, "sign") == 0) {
57                                 if (sec_options == RPC_C_AUTHN_LEVEL_PKT_PRIVACY) {
58                                         printf("You must choose sign or seal, not both\n");
59                                         exit(1);
60                                 }
61                                 sec_options = RPC_C_AUTHN_LEVEL_PKT_INTEGRITY;
62                         } else if (strcmp(option, "seal") == 0) {
63                                 if (sec_options == RPC_C_AUTHN_LEVEL_PKT_INTEGRITY) {
64                                         printf("You must choose sign or seal, not both\n");
65                                         exit(1);
66                                 }
67                                 sec_options = RPC_C_AUTHN_LEVEL_PKT_PRIVACY;
68                         } else {
69                                 printf("Bad security option '%s'\n", option);
70                                 exit(1);
71                         }
72                         argv++;
73                         argc--;
74                         continue;
75                 default:
76                         printf("Bad option -%c\n", argv[0][0]);
77                         exit(1);
78                 }
79                 argv += 2;
80                 argc -= 2;
81         }
82
83         if (argc < 2) {
84                 printf("Usage: client [options] hostname cmd [args]\n\n");
85                 printf("Where hostname is the name of the host to connect to,\n");
86                 printf("and cmd is the command to execute with optional args:\n\n");
87                 printf("\taddone num\tAdd one to num and return the result\n");
88                 printf("\techodata size\tSend an array of size bytes and receive it back\n");
89                 printf("\tsinkdata size\tSend an array of size bytes\n");
90                 printf("\tsourcedata size\tReceive an array of size bytes\n");
91                 printf("\ttest\trun testcall\n");
92                 printf("\noptions:\n");
93                 printf("\t-u username -d domain -p password -e endpoint\n");
94                 printf("\t--sign --seal\n");
95                 printf("\nExamples:\n");
96                 printf("\tclient HOSTNAME addone 3\n");
97                 printf("\tclient 192.168.115.1 addone 3\n");
98                 printf("\tclient -e ncacn_np:HOSTNAME[\\\\pipe\\\\rpcecho] addone 3\n");
99                 printf("\tclient -e ncacn_ip_tcp:192.168.115.1 addone 3\n");
100                 printf("\tclient -e ncacn_ip_tcp:192.168.115.1 -u tridge -d MYDOMAIN -p PASSWORD addone 3\n");
101                 exit(0);
102         }
103
104
105         if (!binding) {
106                 char *network_address = argv[0];
107
108                 argc--;
109                 argv++;
110
111                 status = RpcStringBindingCompose(
112                         NULL, /* uuid */
113                         "ncacn_np",
114                         network_address,
115                         "\\pipe\\rpcecho",
116                         NULL, /* options */
117                         &binding);
118
119                 if (status) {
120                         printf("RpcStringBindingCompose returned %d\n", status);
121                         exit(status);
122                 }
123         }
124
125         printf("Endpoint is %s\n", binding);
126
127         status = RpcBindingFromStringBinding(
128                         binding,
129                         &rpcecho_IfHandle);
130
131         if (status) {
132                 printf("RpcBindingFromStringBinding returned %d\n", status);
133                 exit(status);
134         }
135
136         if (username) {
137                 SEC_WINNT_AUTH_IDENTITY ident = { username, strlen(username),
138                                                   domain, strlen(domain),
139                                                   password, strlen(password),
140                                                   SEC_WINNT_AUTH_IDENTITY_ANSI };
141
142                 status = RpcBindingSetAuthInfo(rpcecho_IfHandle, NULL,
143                                                sec_options,
144                                                RPC_C_AUTHN_WINNT,
145                                                &ident, 0);
146                 if (status) {
147                         printf ("RpcBindingSetAuthInfo failed: 0x%x\n", status);
148                         exit (1);
149                 }
150         }
151
152
153         while (argc > 0) {
154         RpcTryExcept {
155
156                 /* Add one to a number */
157
158                 if (strcmp(argv[0], "addone") == 0) {
159                         int arg, result;
160
161                         if (argc < 2) {
162                                 printf("Usage: addone num\n");
163                                 exit(1);
164                         }
165
166                         arg = atoi(argv[1]);
167
168                         AddOne(arg, &result);
169                         printf("%d + 1 = %d\n", arg, result);
170
171                         argc -= 2;
172                         argv += 2;
173                         continue;
174                 }
175
176                 /* Echo an array */
177
178                 if (strcmp(argv[0], "echodata") == 0) {
179                         int arg, i;
180                         char *indata, *outdata;
181
182                         if (argc < 2) {
183                                 printf("Usage: echo num\n");
184                                 exit(1);
185                         }
186
187                         arg = atoi(argv[1]);
188
189                         if ((indata = malloc(arg)) == NULL) {
190                                 printf("Error allocating %d bytes for input\n", arg);
191                                 exit(1);
192                         }
193
194                         if ((outdata = malloc(arg)) == NULL) {
195                                 printf("Error allocating %d bytes for output\n", arg);
196                                 exit(1);
197                         }
198
199                         for (i = 0; i < arg; i++)
200                                 indata[i] = i & 0xff;
201
202                         EchoData(arg, indata, outdata);
203
204                         printf("echo %d\n", arg);
205
206                         for (i = 0; i < arg; i++) {
207                                 if (indata[i] != outdata[i]) {
208                                         printf("data mismatch at offset %d, %d != %d\n",
209                                                 i, indata[i], outdata[i]);
210                                         exit(0);
211                                 }
212                         }
213
214                         argc -= 2;
215                         argv += 2;
216                         continue;
217                 }
218
219                 if (strcmp(argv[0], "sinkdata") == 0) {
220                         int arg, i;
221                         char *indata;
222
223                         if (argc < 2) {
224                                 printf("Usage: sinkdata num\n");
225                                 exit(1);
226                         }
227
228                         arg = atoi(argv[1]);            
229
230                         if ((indata = malloc(arg)) == NULL) {
231                                 printf("Error allocating %d bytes for input\n", arg);
232                                 exit(1);
233                         }                       
234
235                         for (i = 0; i < arg; i++)
236                                 indata[i] = i & 0xff;
237
238                         SinkData(arg, indata);
239
240                         printf("sinkdata %d\n", arg);
241                         argc -= 2;
242                         argv += 2;
243                         continue;
244                 }
245
246                 if (strcmp(argv[0], "sourcedata") == 0) {
247                         int arg, i;
248                         unsigned char *outdata;
249
250                         if (argc < 2) {
251                                 printf("Usage: sourcedata num\n");
252                                 exit(1);
253                         }
254
255                         arg = atoi(argv[1]);            
256
257                         if ((outdata = malloc(arg)) == NULL) {
258                                 printf("Error allocating %d bytes for output\n", arg);
259                                 exit(1);
260                         }                       
261
262                         SourceData(arg, outdata);
263
264                         printf("sourcedata %d\n", arg);
265
266                         for (i = 0; i < arg; i++) {
267                                 if (outdata[i] != (i & 0xff)) {
268                                         printf("data mismatch at offset %d, %d != %d\n",
269                                                 i, outdata[i], i & 0xff);
270                                 }
271                         }
272
273                         argc -= 2;
274                         argv += 2;
275                         continue;
276                 }
277
278                 if (strcmp(argv[0], "test") == 0) {
279                         printf("no TestCall\n");
280
281                         argc -= 1;
282                         argv += 1;
283                         continue;
284                 }
285
286
287                 if (strcmp(argv[0], "enum") == 0) {
288                         enum echo_Enum1 v = ECHO_ENUM1;
289                         echo_Enum2 e2;
290                         echo_Enum3 e3;
291
292                         e2.e1 = 76;
293                         e2.e2 = ECHO_ENUM1_32;
294                         e3.e1 = ECHO_ENUM2;
295                         
296                         argc -= 1;
297                         argv += 1;
298
299                         echo_TestEnum(&v, &e2, &e3);
300                         
301                         continue;
302                 }
303
304                 if (strcmp(argv[0], "sleep") == 0) {
305                         long arg, result;
306
307                         if (argc < 2) {
308                                 printf("Usage: sleep num\n");
309                                 exit(1);
310                         }
311
312                         arg = atoi(argv[1]);
313
314 //                      result = TestSleep(arg);
315 //                      printf("Slept for %d seconds\n", result);
316                         printf("Sleep disabled (need async code)\n");
317
318                         argc -= 2;
319                         argv += 2;
320                         continue;
321                 }
322
323                 printf("Invalid command '%s'\n", argv[0]);
324                 goto done;
325
326         } RpcExcept(1) {
327                 unsigned long ex;
328
329                 ex = RpcExceptionCode();
330                 printf("Runtime error 0x%x\n", ex);
331         } RpcEndExcept
332                   }
333
334 done:
335
336         status = RpcStringFree(&binding);
337
338         if (status) {
339                 printf("RpcStringFree returned %d\n", status);
340                 exit(status);
341         }
342
343         status = RpcBindingFree(&rpcecho_IfHandle);
344
345         if (status) {
346                 printf("RpcBindingFree returned %d\n", status);
347                 exit(status);
348         }
349
350         exit(0);
351 }