1/*
2   Unix SMB/CIFS implementation.
3   SMB client
4   Copyright (C) Andrew Tridgell 1994-1998
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 "includes.h"
22
23#ifndef REGISTER
24#define REGISTER 0
25#endif
26
27extern pstring global_myname;
28
29extern pstring user_socket_options;
30
31
32extern file_info def_finfo;
33
34#define CNV_LANG(s) dos2unix_format(s,False)
35#define CNV_INPUT(s) unix2dos_format(s,True)
36
37static struct cli_state smbcli;
38struct cli_state *smb_cli = &smbcli;
39
40FILE *out_hnd;
41
42static pstring password; /* local copy only, if one is entered */
43
44/****************************************************************************
45initialise smb client structure
46****************************************************************************/
47void rpcclient_init(void)
48{
49	memset((char *)smb_cli, '\0', sizeof(smb_cli));
50	cli_initialise(smb_cli);
51	smb_cli->capabilities |= CAP_NT_SMBS;
52}
53
54/****************************************************************************
55make smb client connection
56****************************************************************************/
57static BOOL rpcclient_connect(struct client_info *info)
58{
59	struct nmb_name calling;
60	struct nmb_name called;
61
62	make_nmb_name(&called , dns_to_netbios_name(info->dest_host ), info->name_type);
63	make_nmb_name(&calling, dns_to_netbios_name(info->myhostname), 0x0);
64
65	if (!cli_establish_connection(smb_cli,
66	                          info->dest_host, &info->dest_ip,
67	                          &calling, &called,
68	                          info->share, info->svc_type,
69	                          False, True))
70	{
71		DEBUG(0,("rpcclient_connect: connection failed\n"));
72		cli_shutdown(smb_cli);
73		return False;
74	}
75
76	return True;
77}
78
79/****************************************************************************
80stop the smb connection(s?)
81****************************************************************************/
82static void rpcclient_stop(void)
83{
84	cli_shutdown(smb_cli);
85}
86
87/****************************************************************************
88  log in as an nt user, log out again.
89****************************************************************************/
90void run_enums_test(int num_ops, struct client_info *cli_info, struct cli_state *cli)
91{
92	pstring cmd;
93	int i;
94
95	/* establish connections.  nothing to stop these being re-established. */
96	rpcclient_connect(cli_info);
97
98	DEBUG(5,("rpcclient_connect: cli->fd:%d\n", cli->fd));
99	if (cli->fd <= 0)
100	{
101		fprintf(out_hnd, "warning: connection could not be established to %s<%02x>\n",
102		                 cli_info->dest_host, cli_info->name_type);
103		return;
104	}
105
106	for (i = 0; i < num_ops; i++)
107	{
108		set_first_token("");
109		cmd_srv_enum_sess(cli_info);
110		set_first_token("");
111		cmd_srv_enum_shares(cli_info);
112		set_first_token("");
113		cmd_srv_enum_files(cli_info);
114
115		if (password[0] != 0)
116		{
117			slprintf(cmd, sizeof(cmd)-1, "1");
118			set_first_token(cmd);
119		}
120		else
121		{
122			set_first_token("");
123		}
124		cmd_srv_enum_conn(cli_info);
125	}
126
127	rpcclient_stop();
128
129}
130
131/****************************************************************************
132  log in as an nt user, log out again.
133****************************************************************************/
134void run_ntlogin_test(int num_ops, struct client_info *cli_info, struct cli_state *cli)
135{
136	pstring cmd;
137	int i;
138
139	/* establish connections.  nothing to stop these being re-established. */
140	rpcclient_connect(cli_info);
141
142	DEBUG(5,("rpcclient_connect: cli->fd:%d\n", cli->fd));
143	if (cli->fd <= 0)
144	{
145		fprintf(out_hnd, "warning: connection could not be established to %s<%02x>\n",
146		                 cli_info->dest_host, cli_info->name_type);
147		return;
148	}
149
150	for (i = 0; i < num_ops; i++)
151	{
152		slprintf(cmd, sizeof(cmd)-1, "%s %s", cli->user_name, password);
153		set_first_token(cmd);
154
155		cmd_netlogon_login_test(cli_info);
156	}
157
158	rpcclient_stop();
159
160}
161
162/****************************************************************************
163  runs n simultaneous functions.
164****************************************************************************/
165static void create_procs(int nprocs, int numops,
166		struct client_info *cli_info, struct cli_state *cli,
167		void (*fn)(int, struct client_info *, struct cli_state *))
168{
169	int i, status;
170
171	for (i=0;i<nprocs;i++)
172	{
173		if (fork() == 0)
174		{
175			pid_t mypid = getpid();
176			sys_srandom(mypid ^ time(NULL));
177			fn(numops, cli_info, cli);
178			fflush(out_hnd);
179			_exit(0);
180		}
181	}
182
183	for (i=0;i<nprocs;i++)
184	{
185		waitpid(0, &status, 0);
186	}
187}
188/****************************************************************************
189usage on the program - OUT OF DATE!
190****************************************************************************/
191static void usage(char *pname)
192{
193  fprintf(out_hnd, "Usage: %s service <password> [-d debuglevel] [-l log] ",
194	   pname);
195
196  fprintf(out_hnd, "\nVersion %s\n",SAMBA_VERSION_STRING);
197  fprintf(out_hnd, "\t-d debuglevel         set the debuglevel\n");
198  fprintf(out_hnd, "\t-l log basename.      Basename for log/debug files\n");
199  fprintf(out_hnd, "\t-n netbios name.      Use this name as my netbios name\n");
200  fprintf(out_hnd, "\t-m max protocol       set the max protocol level\n");
201  fprintf(out_hnd, "\t-I dest IP            use this IP to connect to\n");
202  fprintf(out_hnd, "\t-E                    write messages to stderr instead of stdout\n");
203  fprintf(out_hnd, "\t-U username           set the network username\n");
204  fprintf(out_hnd, "\t-W workgroup          set the workgroup name\n");
205  fprintf(out_hnd, "\t-t terminal code      terminal i/o code {sjis|euc|jis7|jis8|junet|hex}\n");
206  fprintf(out_hnd, "\n");
207}
208
209enum client_action
210{
211	CLIENT_NONE,
212	CLIENT_IPC,
213	CLIENT_SVC
214};
215
216/****************************************************************************
217  main program
218****************************************************************************/
219 int main(int argc,char *argv[])
220{
221	char *pname = argv[0];
222	int opt;
223	extern char *optarg;
224	extern int optind;
225	pstring term_code;
226	BOOL got_pass = False;
227	char *cmd_str="";
228	enum client_action cli_action = CLIENT_NONE;
229	int nprocs = 1;
230	int numops = 100;
231	pstring logfile;
232
233	struct client_info cli_info;
234
235	out_hnd = stdout;
236
237	rpcclient_init();
238
239#ifdef KANJI
240	pstrcpy(term_code, KANJI);
241#else /* KANJI */
242	*term_code = 0;
243#endif /* KANJI */
244
245	if (!lp_load(dyn_CONFIGFILE,True, False, False))
246	{
247		fprintf(stderr, "Can't load %s - run testparm to debug it\n", dyn_CONFIGFILE);
248	}
249
250	DEBUGLEVEL = 0;
251
252	cli_info.put_total_size = 0;
253	cli_info.put_total_time_ms = 0;
254	cli_info.get_total_size = 0;
255	cli_info.get_total_time_ms = 0;
256
257	cli_info.dir_total = 0;
258	cli_info.newer_than = 0;
259	cli_info.archive_level = 0;
260	cli_info.print_mode = 1;
261
262	cli_info.translation = False;
263	cli_info.recurse_dir = False;
264	cli_info.lowercase = False;
265	cli_info.prompt = True;
266	cli_info.abort_mget = True;
267
268	cli_info.dest_ip.s_addr = 0;
269	cli_info.name_type = 0x20;
270
271	pstrcpy(cli_info.cur_dir , "\\");
272	pstrcpy(cli_info.file_sel, "");
273	pstrcpy(cli_info.base_dir, "");
274	pstrcpy(smb_cli->domain, "");
275	pstrcpy(smb_cli->user_name, "");
276	pstrcpy(cli_info.myhostname, "");
277	pstrcpy(cli_info.dest_host, "");
278
279	pstrcpy(cli_info.svc_type, "A:");
280	pstrcpy(cli_info.share, "");
281	pstrcpy(cli_info.service, "");
282
283	ZERO_STRUCT(cli_info.dom.level3_sid);
284	pstrcpy(cli_info.dom.level3_dom, "");
285	ZERO_STRUCT(cli_info.dom.level5_sid);
286	pstrcpy(cli_info.dom.level5_dom, "");
287
288	{
289		int i;
290		for (i=0; i<PI_MAX_PIPES; i++)
291			smb_cli->nt_pipe_fnum[i]   = 0xffff;
292	}
293
294	setup_logging(pname, True);
295
296	if (!get_myname(global_myname))
297	{
298		fprintf(stderr, "Failed to get my hostname.\n");
299	}
300
301	password[0] = 0;
302
303	if (argc < 2)
304	{
305		usage(pname);
306		exit(1);
307	}
308
309	if (*argv[1] != '-')
310	{
311		pstrcpy(cli_info.service, argv[1]);
312		/* Convert any '/' characters in the service name to '\' characters */
313		string_replace( cli_info.service, '/','\\');
314		argc--;
315		argv++;
316
317		DEBUG(1,("service: %s\n", cli_info.service));
318
319		if (count_chars(cli_info.service,'\\') < 3)
320		{
321			usage(pname);
322			printf("\n%s: Not enough '\\' characters in service\n", cli_info.service);
323			exit(1);
324		}
325
326		/*
327		if (count_chars(cli_info.service,'\\') > 3)
328		{
329			usage(pname);
330			printf("\n%s: Too many '\\' characters in service\n", cli_info.service);
331			exit(1);
332		}
333		*/
334
335		if (argc > 1 && (*argv[1] != '-'))
336		{
337			got_pass = True;
338			pstrcpy(password,argv[1]);
339			memset(argv[1],'X',strlen(argv[1]));
340			argc--;
341			argv++;
342		}
343
344		cli_action = CLIENT_SVC;
345	}
346
347	while ((opt = getopt(argc, argv,"s:O:M:S:i:N:o:n:d:l:hI:EB:U:L:t:m:W:T:D:c:")) != EOF)
348	{
349		switch (opt)
350		{
351			case 'm':
352			{
353				/* FIXME ... max_protocol seems to be funny here */
354
355				int max_protocol = 0;
356				max_protocol = interpret_protocol(optarg,max_protocol);
357				fprintf(stderr, "max protocol not currently supported\n");
358				break;
359			}
360
361			case 'O':
362			{
363				pstrcpy(user_socket_options,optarg);
364				break;
365			}
366
367			case 'S':
368			{
369				pstrcpy(cli_info.dest_host,optarg);
370				strupper_m(cli_info.dest_host);
371				cli_action = CLIENT_IPC;
372				break;
373			}
374
375			case 'i':
376			{
377				pstrcpy(scope, optarg);
378				break;
379			}
380
381			case 'U':
382			{
383				char *lp;
384				pstrcpy(smb_cli->user_name,optarg);
385				if ((lp=strchr_m(smb_cli->user_name,'%')))
386				{
387					*lp = 0;
388					pstrcpy(password,lp+1);
389					got_pass = True;
390					memset(strchr_m(optarg,'%')+1,'X',strlen(password));
391				}
392				break;
393			}
394
395			case 'W':
396			{
397				pstrcpy(smb_cli->domain,optarg);
398				break;
399			}
400
401			case 'E':
402			{
403				dbf = x_stderr;
404				break;
405			}
406
407			case 'I':
408			{
409				cli_info.dest_ip = *interpret_addr2(optarg);
410				if (is_zero_ip(cli_info.dest_ip))
411				{
412					exit(1);
413				}
414				break;
415			}
416
417			case 'N':
418			{
419				nprocs = atoi(optarg);
420				break;
421			}
422
423			case 'o':
424			{
425				numops = atoi(optarg);
426				break;
427			}
428
429			case 'n':
430			{
431				fstrcpy(global_myname, optarg);
432				break;
433			}
434
435			case 'd':
436			{
437				if (*optarg == 'A')
438					DEBUGLEVEL = 10000;
439				else
440					DEBUGLEVEL = atoi(optarg);
441				break;
442			}
443
444			case 'l':
445			{
446				slprintf(logfile, sizeof(logfile)-1,
447				         "%s.client",optarg);
448				lp_set_logfile(logfile);
449				break;
450			}
451
452			case 'c':
453			{
454				cmd_str = optarg;
455				got_pass = True;
456				break;
457			}
458
459			case 'h':
460			{
461				usage(pname);
462				exit(0);
463				break;
464			}
465
466			case 's':
467			{
468				pstrcpy(dyn_CONFIGFILE, optarg);
469				break;
470			}
471
472			case 't':
473			{
474				pstrcpy(term_code, optarg);
475				break;
476			}
477
478			default:
479			{
480				usage(pname);
481				exit(1);
482				break;
483			}
484		}
485	}
486
487	if (cli_action == CLIENT_NONE)
488	{
489		usage(pname);
490		exit(1);
491	}
492
493	strupper_m(global_myname);
494	fstrcpy(cli_info.myhostname, global_myname);
495
496	DEBUG(3,("%s client started (version %s)\n",timestring(False),SAMBA_VERSION_STRING));
497
498	if (*smb_cli->domain == 0)
499	{
500		pstrcpy(smb_cli->domain,lp_workgroup());
501	}
502	strupper_m(smb_cli->domain);
503
504	load_interfaces();
505
506	if (cli_action == CLIENT_IPC)
507	{
508		pstrcpy(cli_info.share, "IPC$");
509		pstrcpy(cli_info.svc_type, "IPC");
510	}
511
512	fstrcpy(cli_info.mach_acct, cli_info.myhostname);
513	strupper_m(cli_info.mach_acct);
514	fstrcat(cli_info.mach_acct, "$");
515
516	/* set the password cache info */
517	if (got_pass)
518	{
519		if (password[0] == 0)
520		{
521			pwd_set_nullpwd(&(smb_cli->pwd));
522		}
523		else
524		{
525			pwd_make_lm_nt_16(&(smb_cli->pwd), password); /* generate 16 byte hashes */
526		}
527	}
528	else
529	{
530		char *pwd = getpass("Enter Password:");
531		safe_strcpy(password, pwd, sizeof(password));
532		pwd_make_lm_nt_16(&(smb_cli->pwd), password); /* generate 16 byte hashes */
533	}
534
535	create_procs(nprocs, numops, &cli_info, smb_cli, run_enums_test);
536
537	if (password[0] != 0)
538	{
539		create_procs(nprocs, numops, &cli_info, smb_cli, run_ntlogin_test);
540	}
541
542	fflush(out_hnd);
543
544	return(0);
545}
546