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