/* Unix SMB/CIFS implementation. SMB client Copyright (C) Andrew Tridgell 1994-1998 Copyright (C) Simo Sorce 2001-2002 Copyright (C) Jelmer Vernooij 2003 Copyright (C) James J Myers 2003 This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. */ #include "includes.h" #include "../client/client_proto.h" #ifndef REGISTER #define REGISTER 0 #endif struct cli_state *cli; extern BOOL in_client; static int port = 0; pstring cur_dir = "\\"; static pstring cd_path = ""; static pstring service; static pstring desthost; static pstring username; static pstring password; static BOOL use_kerberos; static BOOL got_pass; static char *cmdstr = NULL; static int io_bufsize = 64512; static int name_type = 0x20; static int process_tok(fstring tok); static int cmd_help(void); /* 30 second timeout on most commands */ #define CLIENT_TIMEOUT (30*1000) #define SHORT_TIMEOUT (5*1000) /* value for unused fid field in trans2 secondary request */ #define FID_UNUSED (0xFFFF) time_t newer_than = 0; static int archive_level = 0; static BOOL translation = False; static BOOL have_ip; /* clitar bits insert */ extern int blocksize; extern BOOL tar_inc; extern BOOL tar_reset; /* clitar bits end */ static BOOL prompt = True; static int printmode = 1; static BOOL recurse = False; BOOL lowercase = False; static struct in_addr dest_ip; #define SEPARATORS " \t\n\r" static BOOL abort_mget = True; static pstring fileselection = ""; extern file_info def_finfo; /* timing globals */ SMB_BIG_UINT get_total_size = 0; unsigned int get_total_time_ms = 0; static SMB_BIG_UINT put_total_size = 0; static unsigned int put_total_time_ms = 0; /* totals globals */ static double dir_total; #define USENMB /* some forward declarations */ static struct cli_state *do_connect(const char *server, const char *share); /******************************************************************* Reduce a file name, removing .. elements. ********************************************************************/ void dos_clean_name(char *s) { char *p=NULL; DEBUG(3,("dos_clean_name [%s]\n",s)); /* remove any double slashes */ all_string_sub(s, "\\\\", "\\", 0); while ((p = strstr(s,"\\..\\")) != NULL) { pstring s1; *p = 0; pstrcpy(s1,p+3); if ((p=strrchr_m(s,'\\')) != NULL) *p = 0; else *s = 0; pstrcat(s,s1); } trim_string(s,NULL,"\\.."); all_string_sub(s, "\\.\\", "\\", 0); } /**************************************************************************** write to a local file with CR/LF->LF translation if appropriate. return the number taken from the buffer. This may not equal the number written. ****************************************************************************/ static int writefile(int f, char *b, int n) { int i; if (!translation) { return write(f,b,n); } i = 0; while (i < n) { if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') { b++;i++; } if (write(f, b, 1) != 1) { break; } b++; i++; } return(i); } /**************************************************************************** read from a file with LF->CR/LF translation if appropriate. return the number read. read approx n bytes. ****************************************************************************/ static int readfile(char *b, int n, XFILE *f) { int i; int c; if (!translation) return x_fread(b,1,n,f); i = 0; while (i < (n - 1) && (i < CLI_BUFFER_SIZE)) { if ((c = x_getc(f)) == EOF) { break; } if (c == '\n') { /* change all LFs to CR/LF */ b[i++] = '\r'; } b[i++] = c; } return(i); } /**************************************************************************** send a message ****************************************************************************/ static void send_message(void) { int total_len = 0; int grp_id; if (!cli_message_start(cli, desthost, username, &grp_id)) { d_printf("message start: %s\n", cli_errstr(cli)); return; } d_printf("Connected. Type your message, ending it with a Control-D\n"); while (!feof(stdin) && total_len < 1600) { int maxlen = MIN(1600 - total_len,127); pstring msg; int l=0; int c; ZERO_ARRAY(msg); for (l=0;l= 1600) d_printf("the message was truncated to 1600 bytes\n"); else d_printf("sent %d bytes\n",total_len); if (!cli_message_end(cli, grp_id)) { d_printf("SMBsendend failed (%s)\n",cli_errstr(cli)); return; } } /**************************************************************************** check the space on a device ****************************************************************************/ static int do_dskattr(void) { int total, bsize, avail; if (!cli_dskattr(cli, &bsize, &total, &avail)) { d_printf("Error in dskattr: %s\n",cli_errstr(cli)); return 1; } d_printf("\n\t\t%d blocks of size %d. %d blocks available\n", total, bsize, avail); return 0; } /**************************************************************************** show cd/pwd ****************************************************************************/ static int cmd_pwd(void) { d_printf("Current directory is %s",service); d_printf("%s\n",cur_dir); return 0; } /**************************************************************************** change directory - inner section ****************************************************************************/ static int do_cd(char *newdir) { char *p = newdir; pstring saved_dir; pstring dname; dos_format(newdir); /* Save the current directory in case the new directory is invalid */ pstrcpy(saved_dir, cur_dir); if (*p == '\\') pstrcpy(cur_dir,p); else pstrcat(cur_dir,p); if (*(cur_dir+strlen(cur_dir)-1) != '\\') { pstrcat(cur_dir, "\\"); } dos_clean_name(cur_dir); pstrcpy(dname,cur_dir); pstrcat(cur_dir,"\\"); dos_clean_name(cur_dir); if (!strequal(cur_dir,"\\")) { if (!cli_chkpath(cli, dname)) { d_printf("cd %s: %s\n", dname, cli_errstr(cli)); pstrcpy(cur_dir,saved_dir); } } pstrcpy(cd_path,cur_dir); return 0; } /**************************************************************************** change directory ****************************************************************************/ static int cmd_cd(void) { fstring buf; int rc = 0; if (next_token_nr(NULL,buf,NULL,sizeof(buf))) rc = do_cd(buf); else d_printf("Current directory is %s\n",cur_dir); return rc; } /******************************************************************* decide if a file should be operated on ********************************************************************/ static BOOL do_this_one(file_info *finfo) { if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) return(True); if (*fileselection && !mask_match(cli, finfo->name,fileselection,False)) { DEBUG(3,("mask_match %s failed\n", finfo->name)); return False; } if (newer_than && finfo->mtime < newer_than) { DEBUG(3,("newer_than %s failed\n", finfo->name)); return(False); } if ((archive_level==1 || archive_level==2) && !(finfo->mode & FILE_ATTRIBUTE_ARCHIVE)) { DEBUG(3,("archive %s failed\n", finfo->name)); return(False); } return(True); } /******************************************************************* Return a string representing an attribute for a file. ********************************************************************/ static const char *attrib_string(uint16 mode) { static fstring attrstr; int i, len; const struct { char c; uint16 attr; } attr_strs[] = { {'V', FILE_ATTRIBUTE_VOLUME}, {'D', FILE_ATTRIBUTE_DIRECTORY}, {'A', FILE_ATTRIBUTE_ARCHIVE}, {'H', FILE_ATTRIBUTE_HIDDEN}, {'S', FILE_ATTRIBUTE_SYSTEM}, {'R', FILE_ATTRIBUTE_READONLY}, {'d', FILE_ATTRIBUTE_DEVICE}, {'t', FILE_ATTRIBUTE_TEMPORARY}, {'s', FILE_ATTRIBUTE_SPARSE}, {'r', FILE_ATTRIBUTE_REPARSE_POINT}, {'c', FILE_ATTRIBUTE_COMPRESSED}, {'o', FILE_ATTRIBUTE_OFFLINE}, {'n', FILE_ATTRIBUTE_NONINDEXED}, {'e', FILE_ATTRIBUTE_ENCRYPTED} }; for (len=i=0; imtime; /* the time is assumed to be passed as GMT */ d_printf(" %-30s%7.7s %8.0f %s", finfo->name, attrib_string(finfo->mode), (double)finfo->size, asctime(LocalTime(&t))); dir_total += finfo->size; } } /**************************************************************************** accumulate size of a file ****************************************************************************/ static void do_du(file_info *finfo) { if (do_this_one(finfo)) { dir_total += finfo->size; } } static BOOL do_list_recurse; static BOOL do_list_dirs; static char *do_list_queue = 0; static long do_list_queue_size = 0; static long do_list_queue_start = 0; static long do_list_queue_end = 0; static void (*do_list_fn)(file_info *); /**************************************************************************** functions for do_list_queue ****************************************************************************/ /* * The do_list_queue is a NUL-separated list of strings stored in a * char*. Since this is a FIFO, we keep track of the beginning and * ending locations of the data in the queue. When we overflow, we * double the size of the char*. When the start of the data passes * the midpoint, we move everything back. This is logically more * complex than a linked list, but easier from a memory management * angle. In any memory error condition, do_list_queue is reset. * Functions check to ensure that do_list_queue is non-NULL before * accessing it. */ static void reset_do_list_queue(void) { SAFE_FREE(do_list_queue); do_list_queue_size = 0; do_list_queue_start = 0; do_list_queue_end = 0; } static void init_do_list_queue(void) { reset_do_list_queue(); do_list_queue_size = 1024; do_list_queue = malloc(do_list_queue_size); if (do_list_queue == 0) { d_printf("malloc fail for size %d\n", (int)do_list_queue_size); reset_do_list_queue(); } else { memset(do_list_queue, 0, do_list_queue_size); } } static void adjust_do_list_queue(void) { /* * If the starting point of the queue is more than half way through, * move everything toward the beginning. */ if (do_list_queue && (do_list_queue_start == do_list_queue_end)) { DEBUG(4,("do_list_queue is empty\n")); do_list_queue_start = do_list_queue_end = 0; *do_list_queue = '\0'; } else if (do_list_queue_start > (do_list_queue_size / 2)) { DEBUG(4,("sliding do_list_queue backward\n")); memmove(do_list_queue, do_list_queue + do_list_queue_start, do_list_queue_end - do_list_queue_start); do_list_queue_end -= do_list_queue_start; do_list_queue_start = 0; } } static void add_to_do_list_queue(const char* entry) { char *dlq; long new_end = do_list_queue_end + ((long)strlen(entry)) + 1; while (new_end > do_list_queue_size) { do_list_queue_size *= 2; DEBUG(4,("enlarging do_list_queue to %d\n", (int)do_list_queue_size)); dlq = Realloc(do_list_queue, do_list_queue_size); if (! dlq) { d_printf("failure enlarging do_list_queue to %d bytes\n", (int)do_list_queue_size); reset_do_list_queue(); } else { do_list_queue = dlq; memset(do_list_queue + do_list_queue_size / 2, 0, do_list_queue_size / 2); } } if (do_list_queue) { safe_strcpy(do_list_queue + do_list_queue_end, entry, do_list_queue_size - do_list_queue_end - 1); do_list_queue_end = new_end; DEBUG(4,("added %s to do_list_queue (start=%d, end=%d)\n", entry, (int)do_list_queue_start, (int)do_list_queue_end)); } } static char *do_list_queue_head(void) { return do_list_queue + do_list_queue_start; } static void remove_do_list_queue_head(void) { if (do_list_queue_end > do_list_queue_start) { do_list_queue_start += strlen(do_list_queue_head()) + 1; adjust_do_list_queue(); DEBUG(4,("removed head of do_list_queue (start=%d, end=%d)\n", (int)do_list_queue_start, (int)do_list_queue_end)); } } static int do_list_queue_empty(void) { return (! (do_list_queue && *do_list_queue)); } /**************************************************************************** a helper for do_list ****************************************************************************/ static void do_list_helper(file_info *f, const char *mask, void *state) { if (f->mode & FILE_ATTRIBUTE_DIRECTORY) { if (do_list_dirs && do_this_one(f)) { do_list_fn(f); } if (do_list_recurse && !strequal(f->name,".") && !strequal(f->name,"..")) { pstring mask2; char *p; pstrcpy(mask2, mask); p = strrchr_m(mask2,'\\'); if (!p) return; p[1] = 0; pstrcat(mask2, f->name); pstrcat(mask2,"\\*"); add_to_do_list_queue(mask2); } return; } if (do_this_one(f)) { do_list_fn(f); } } /**************************************************************************** a wrapper around cli_list that adds recursion ****************************************************************************/ void do_list(const char *mask,uint16 attribute,void (*fn)(file_info *),BOOL rec, BOOL dirs) { static int in_do_list = 0; if (in_do_list && rec) { fprintf(stderr, "INTERNAL ERROR: do_list called recursively when the recursive flag is true\n"); exit(1); } in_do_list = 1; do_list_recurse = rec; do_list_dirs = dirs; do_list_fn = fn; if (rec) { init_do_list_queue(); add_to_do_list_queue(mask); while (! do_list_queue_empty()) { /* * Need to copy head so that it doesn't become * invalid inside the call to cli_list. This * would happen if the list were expanded * during the call. * Fix from E. Jay Berkenbilt (ejb@ql.org) */ pstring head; pstrcpy(head, do_list_queue_head()); cli_list(cli, head, attribute, do_list_helper, NULL); remove_do_list_queue_head(); if ((! do_list_queue_empty()) && (fn == display_finfo)) { char* next_file = do_list_queue_head(); char* save_ch = 0; if ((strlen(next_file) >= 2) && (next_file[strlen(next_file) - 1] == '*') && (next_file[strlen(next_file) - 2] == '\\')) { save_ch = next_file + strlen(next_file) - 2; *save_ch = '\0'; } d_printf("\n%s\n",next_file); if (save_ch) { *save_ch = '\\'; } } } } else { if (cli_list(cli, mask, attribute, do_list_helper, NULL) == -1) { d_printf("%s listing %s\n", cli_errstr(cli), mask); } } in_do_list = 0; reset_do_list_queue(); } /**************************************************************************** get a directory listing ****************************************************************************/ static int cmd_dir(void) { uint16 attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; pstring mask; fstring buf; char *p=buf; int rc; dir_total = 0; pstrcpy(mask,cur_dir); if(mask[strlen(mask)-1]!='\\') pstrcat(mask,"\\"); if (next_token_nr(NULL,buf,NULL,sizeof(buf))) { dos_format(p); if (*p == '\\') pstrcpy(mask,p); else pstrcat(mask,p); } else { pstrcat(mask,"*"); } do_list(mask, attribute, display_finfo, recurse, True); rc = do_dskattr(); DEBUG(3, ("Total bytes listed: %.0f\n", dir_total)); return rc; } /**************************************************************************** get a directory listing ****************************************************************************/ static int cmd_du(void) { uint16 attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; pstring mask; fstring buf; char *p=buf; int rc; dir_total = 0; pstrcpy(mask,cur_dir); if(mask[strlen(mask)-1]!='\\') pstrcat(mask,"\\"); if (next_token_nr(NULL,buf,NULL,sizeof(buf))) { dos_format(p); if (*p == '\\') pstrcpy(mask,p); else pstrcat(mask,p); } else { pstrcat(mask,"*"); } do_list(mask, attribute, do_du, recurse, True); rc = do_dskattr(); d_printf("Total number of bytes: %.0f\n", dir_total); return rc; } /**************************************************************************** get a file from rname to lname ****************************************************************************/ static int do_get(char *rname, const char *lname, BOOL reget) { int handle = 0, fnum; BOOL newhandle = False; char *data; struct timeval tp_start; int read_size = io_bufsize; uint16 attr; size_t size; off_t start = 0; off_t nread = 0; int rc = 0; GetTimeOfDay(&tp_start); if (lowercase) { strlower(lname); } fnum = cli_open(cli, rname, O_RDONLY, DENY_NONE); if (fnum == -1) { d_printf("%s opening remote file %s\n",cli_errstr(cli),rname); return 1; } if(!strcmp(lname,"-")) { handle = fileno(stdout); } else { if (reget) { handle = sys_open(lname, O_WRONLY|O_CREAT, 0644); if (handle >= 0) { start = sys_lseek(handle, 0, SEEK_END); if (start == -1) { d_printf("Error seeking local file\n"); return 1; } } } else { handle = sys_open(lname, O_WRONLY|O_CREAT|O_TRUNC, 0644); } newhandle = True; } if (handle < 0) { d_printf("Error opening local file %s\n",lname); return 1; } if (!cli_qfileinfo(cli, fnum, &attr, &size, NULL, NULL, NULL, NULL, NULL) && !cli_getattrE(cli, fnum, &attr, &size, NULL, NULL, NULL)) { d_printf("getattrib: %s\n",cli_errstr(cli)); return 1; } DEBUG(2,("getting file %s of size %.0f as %s ", rname, (double)size, lname)); if(!(data = (char *)malloc(read_size))) { d_printf("malloc fail for size %d\n", read_size); cli_close(cli, fnum); return 1; } while (1) { int n = cli_read(cli, fnum, data, nread + start, read_size); if (n <= 0) break; if (writefile(handle,data, n) != n) { d_printf("Error writing local file\n"); rc = 1; break; } nread += n; } if (nread + start < size) { DEBUG (0, ("Short read when getting file %s. Only got %ld bytes.\n", rname, (long)nread)); rc = 1; } SAFE_FREE(data); if (!cli_close(cli, fnum)) { d_printf("Error %s closing remote file\n",cli_errstr(cli)); rc = 1; } if (newhandle) { close(handle); } if (archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) { cli_setatr(cli, rname, attr & ~(uint16)FILE_ATTRIBUTE_ARCHIVE, 0); } { struct timeval tp_end; int this_time; GetTimeOfDay(&tp_end); this_time = (tp_end.tv_sec - tp_start.tv_sec)*1000 + (tp_end.tv_usec - tp_start.tv_usec)/1000; get_total_time_ms += this_time; get_total_size += nread; DEBUG(2,("(%3.1f kb/s) (average %3.1f kb/s)\n", nread / (1.024*this_time + 1.0e-4), get_total_size / (1.024*get_total_time_ms))); } return rc; } /**************************************************************************** get a file ****************************************************************************/ static int cmd_get(void) { pstring lname; pstring rname; char *p; pstrcpy(rname,cur_dir); pstrcat(rname,"\\"); p = rname + strlen(rname); if (!next_token_nr(NULL,p,NULL,sizeof(rname)-strlen(rname))) { d_printf("get \n"); return 1; } pstrcpy(lname,p); dos_clean_name(rname); next_token_nr(NULL,lname,NULL,sizeof(lname)); return do_get(rname, lname, False); } /**************************************************************************** do a mget operation on one file ****************************************************************************/ static void do_mget(file_info *finfo) { pstring rname; pstring quest; pstring saved_curdir; pstring mget_mask; if (strequal(finfo->name,".") || strequal(finfo->name,"..")) return; if (abort_mget) { d_printf("mget aborted\n"); return; } if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) slprintf(quest,sizeof(pstring)-1, "Get directory %s? ",finfo->name); else slprintf(quest,sizeof(pstring)-1, "Get file %s? ",finfo->name); if (prompt && !yesno(quest)) return; if (!(finfo->mode & FILE_ATTRIBUTE_DIRECTORY)) { pstrcpy(rname,cur_dir); pstrcat(rname,finfo->name); do_get(rname, finfo->name, False); return; } /* handle directories */ pstrcpy(saved_curdir,cur_dir); pstrcat(cur_dir,finfo->name); pstrcat(cur_dir,"\\"); unix_format(finfo->name); if (lowercase) strlower(finfo->name); if (!directory_exist(finfo->name,NULL) && mkdir(finfo->name,0777) != 0) { d_printf("failed to create directory %s\n",finfo->name); pstrcpy(cur_dir,saved_curdir); return; } if (chdir(finfo->name) != 0) { d_printf("failed to chdir to directory %s\n",finfo->name); pstrcpy(cur_dir,saved_curdir); return; } pstrcpy(mget_mask,cur_dir); pstrcat(mget_mask,"*"); do_list(mget_mask, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN | FILE_ATTRIBUTE_DIRECTORY,do_mget,False, True); chdir(".."); pstrcpy(cur_dir,saved_curdir); } /**************************************************************************** view the file using the pager ****************************************************************************/ static int cmd_more(void) { fstring rname,lname,pager_cmd; char *pager; int fd; int rc = 0; fstrcpy(rname,cur_dir); fstrcat(rname,"\\"); slprintf(lname,sizeof(lname)-1, "%s/smbmore.XXXXXX",tmpdir()); fd = smb_mkstemp(lname); if (fd == -1) { d_printf("failed to create temporary file for more\n"); return 1; } close(fd); if (!next_token_nr(NULL,rname+strlen(rname),NULL,sizeof(rname)-strlen(rname))) { d_printf("more \n"); unlink(lname); return 1; } dos_clean_name(rname); rc = do_get(rname, lname, False); pager=getenv("PAGER"); slprintf(pager_cmd,sizeof(pager_cmd)-1, "%s %s",(pager? pager:PAGER), lname); system(pager_cmd); unlink(lname); return rc; } /**************************************************************************** do a mget command ****************************************************************************/ static int cmd_mget(void) { uint16 attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; pstring mget_mask; fstring buf; char *p=buf; *mget_mask = 0; if (recurse) attribute |= FILE_ATTRIBUTE_DIRECTORY; abort_mget = False; while (next_token_nr(NULL,p,NULL,sizeof(buf))) { pstrcpy(mget_mask,cur_dir); if(mget_mask[strlen(mget_mask)-1]!='\\') pstrcat(mget_mask,"\\"); if (*p == '\\') pstrcpy(mget_mask,p); else pstrcat(mget_mask,p); do_list(mget_mask, attribute,do_mget,False,True); } if (!*mget_mask) { pstrcpy(mget_mask,cur_dir); if(mget_mask[strlen(mget_mask)-1]!='\\') pstrcat(mget_mask,"\\"); pstrcat(mget_mask,"*"); do_list(mget_mask, attribute,do_mget,False,True); } return 0; } /**************************************************************************** make a directory of name "name" ****************************************************************************/ static BOOL do_mkdir(char *name) { if (!cli_mkdir(cli, name)) { d_printf("%s making remote directory %s\n", cli_errstr(cli),name); return(False); } return(True); } /**************************************************************************** show 8.3 name of a file ****************************************************************************/ static BOOL do_altname(char *name) { const char *altname; if (!NT_STATUS_IS_OK(cli_qpathinfo_alt_name(cli, name, &altname))) { d_printf("%s getting alt name for %s\n", cli_errstr(cli),name); return(False); } d_printf("%s\n", altname); return(True); } /**************************************************************************** Exit client. ****************************************************************************/ static int cmd_quit(void) { cli_shutdown(cli); exit(0); /* NOTREACHED */ return 0; } /**************************************************************************** make a directory ****************************************************************************/ static int cmd_mkdir(void) { pstring mask; fstring buf; char *p=buf; pstrcpy(mask,cur_dir); if (!next_token_nr(NULL,p,NULL,sizeof(buf))) { if (!recurse) d_printf("mkdir \n"); return 1; } pstrcat(mask,p); if (recurse) { pstring ddir; pstring ddir2; *ddir2 = 0; pstrcpy(ddir,mask); trim_string(ddir,".",NULL); p = strtok(ddir,"/\\"); while (p) { pstrcat(ddir2,p); if (!cli_chkpath(cli, ddir2)) { do_mkdir(ddir2); } pstrcat(ddir2,"\\"); p = strtok(NULL,"/\\"); } } else { do_mkdir(mask); } return 0; } /**************************************************************************** show alt name ****************************************************************************/ static int cmd_altname(void) { pstring name; fstring buf; char *p=buf; pstrcpy(name,cur_dir); if (!next_token_nr(NULL,p,NULL,sizeof(buf))) { d_printf("altname \n"); return 1; } pstrcat(name,p); do_altname(name); return 0; } /**************************************************************************** put a single file ****************************************************************************/ static int do_put(char *rname, char *lname, BOOL reput) { int fnum; XFILE *f; size_t start = 0; off_t nread = 0; char *buf = NULL; int maxwrite = io_bufsize; int rc = 0; struct timeval tp_start; GetTimeOfDay(&tp_start); if (reput) { fnum = cli_open(cli, rname, O_RDWR|O_CREAT, DENY_NONE); if (fnum >= 0) { if (!cli_qfileinfo(cli, fnum, NULL, &start, NULL, NULL, NULL, NULL, NULL) && !cli_getattrE(cli, fnum, NULL, &start, NULL, NULL, NULL)) { d_printf("getattrib: %s\n",cli_errstr(cli)); return 1; } } } else { fnum = cli_open(cli, rname, O_RDWR|O_CREAT|O_TRUNC, DENY_NONE); } if (fnum == -1) { d_printf("%s opening remote file %s\n",cli_errstr(cli),rname); return 1; } /* allow files to be piped into smbclient jdblair 24.jun.98 Note that in this case this function will exit(0) rather than returning. */ if (!strcmp(lname, "-")) { f = x_stdin; /* size of file is not known */ } else { f = x_fopen(lname,O_RDONLY, 0); if (f && reput) { if (x_tseek(f, start, SEEK_SET) == -1) { d_printf("Error seeking local file\n"); return 1; } } } if (!f) { d_printf("Error opening local file %s\n",lname); return 1; } DEBUG(1,("putting file %s as %s ",lname, rname)); buf = (char *)malloc(maxwrite); if (!buf) { d_printf("ERROR: Not enough memory!\n"); return 1; } while (!x_feof(f)) { int n = maxwrite; int ret; if ((n = readfile(buf,n,f)) < 1) { if((n == 0) && x_feof(f)) break; /* Empty local file. */ d_printf("Error reading local file: %s\n", strerror(errno)); rc = 1; break; } ret = cli_write(cli, fnum, 0, buf, nread + start, n); if (n != ret) { d_printf("Error writing file: %s\n", cli_errstr(cli)); rc = 1; break; } nread += n; } if (!cli_close(cli, fnum)) { d_printf("%s closing remote file %s\n",cli_errstr(cli),rname); x_fclose(f); SAFE_FREE(buf); return 1; } if (f != x_stdin) { x_fclose(f); } SAFE_FREE(buf); { struct timeval tp_end; int this_time; GetTimeOfDay(&tp_end); this_time = (tp_end.tv_sec - tp_start.tv_sec)*1000 + (tp_end.tv_usec - tp_start.tv_usec)/1000; put_total_time_ms += this_time; put_total_size += nread; DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n", nread / (1.024*this_time + 1.0e-4), put_total_size / (1.024*put_total_time_ms))); } if (f == x_stdin) { cli_shutdown(cli); exit(0); } return rc; } /**************************************************************************** put a file ****************************************************************************/ static int cmd_put(void) { pstring lname; pstring rname; fstring buf; char *p=buf; pstrcpy(rname,cur_dir); pstrcat(rname,"\\"); if (!next_token_nr(NULL,p,NULL,sizeof(buf))) { d_printf("put \n"); return 1; } pstrcpy(lname,p); if (next_token_nr(NULL,p,NULL,sizeof(buf))) pstrcat(rname,p); else pstrcat(rname,lname); dos_clean_name(rname); { SMB_STRUCT_STAT st; /* allow '-' to represent stdin jdblair, 24.jun.98 */ if (!file_exist(lname,&st) && (strcmp(lname,"-"))) { d_printf("%s does not exist\n",lname); return 1; } } return do_put(rname, lname, False); } /************************************* File list structure *************************************/ static struct file_list { struct file_list *prev, *next; char *file_path; BOOL isdir; } *file_list; /**************************************************************************** Free a file_list structure ****************************************************************************/ static void free_file_list (struct file_list * list) { struct file_list *tmp; while (list) { tmp = list; DLIST_REMOVE(list, list); SAFE_FREE(tmp->file_path); SAFE_FREE(tmp); } } /**************************************************************************** seek in a directory/file list until you get something that doesn't start with the specified name ****************************************************************************/ static BOOL seek_list(struct file_list *list, char *name) { while (list) { trim_string(list->file_path,"./","\n"); if (strncmp(list->file_path, name, strlen(name)) != 0) { return(True); } list = list->next; } return(False); } /**************************************************************************** set the file selection mask ****************************************************************************/ static int cmd_select(void) { pstrcpy(fileselection,""); next_token_nr(NULL,fileselection,NULL,sizeof(fileselection)); return 0; } /**************************************************************************** Recursive file matching function act as find match must be always set to True when calling this function ****************************************************************************/ static int file_find(struct file_list **list, const char *directory, const char *expression, BOOL match) { DIR *dir; struct file_list *entry; struct stat statbuf; int ret; char *path; BOOL isdir; const char *dname; dir = opendir(directory); if (!dir) return -1; while ((dname = readdirname(dir))) { if (!strcmp("..", dname)) continue; if (!strcmp(".", dname)) continue; if (asprintf(&path, "%s/%s", directory, dname) <= 0) { continue; } isdir = False; if (!match || !gen_fnmatch(expression, dname)) { if (recurse) { ret = stat(path, &statbuf); if (ret == 0) { if (S_ISDIR(statbuf.st_mode)) { isdir = True; ret = file_find(list, path, expression, False); } } else { d_printf("file_find: cannot stat file %s\n", path); } if (ret == -1) { SAFE_FREE(path); closedir(dir); return -1; } } entry = (struct file_list *) malloc(sizeof (struct file_list)); if (!entry) { d_printf("Out of memory in file_find\n"); closedir(dir); return -1; } entry->file_path = path; entry->isdir = isdir; DLIST_ADD(*list, entry); } else { SAFE_FREE(path); } } closedir(dir); return 0; } /**************************************************************************** mput some files ****************************************************************************/ static int cmd_mput(void) { fstring buf; char *p=buf; while (next_token_nr(NULL,p,NULL,sizeof(buf))) { int ret; struct file_list *temp_list; char *quest, *lname, *rname; file_list = NULL; ret = file_find(&file_list, ".", p, True); if (ret) { free_file_list(file_list); continue; } quest = NULL; lname = NULL; rname = NULL; for (temp_list = file_list; temp_list; temp_list = temp_list->next) { SAFE_FREE(lname); if (asprintf(&lname, "%s/", temp_list->file_path) <= 0) continue; trim_string(lname, "./", "/"); /* check if it's a directory */ if (temp_list->isdir) { /* if (!recurse) continue; */ SAFE_FREE(quest); if (asprintf(&quest, "Put directory %s? ", lname) < 0) break; if (prompt && !yesno(quest)) { /* No */ /* Skip the directory */ lname[strlen(lname)-1] = '/'; if (!seek_list(temp_list, lname)) break; } else { /* Yes */ SAFE_FREE(rname); if(asprintf(&rname, "%s%s", cur_dir, lname) < 0) break; dos_format(rname); if (!cli_chkpath(cli, rname) && !do_mkdir(rname)) { DEBUG (0, ("Unable to make dir, skipping...")); /* Skip the directory */ lname[strlen(lname)-1] = '/'; if (!seek_list(temp_list, lname)) break; } } continue; } else { SAFE_FREE(quest); if (asprintf(&quest,"Put file %s? ", lname) < 0) break; if (prompt && !yesno(quest)) /* No */ continue; /* Yes */ SAFE_FREE(rname); if (asprintf(&rname, "%s%s", cur_dir, lname) < 0) break; } dos_format(rname); do_put(rname, lname, False); } free_file_list(file_list); SAFE_FREE(quest); SAFE_FREE(lname); SAFE_FREE(rname); } return 0; } /**************************************************************************** cancel a print job ****************************************************************************/ static int do_cancel(int job) { d_printf("REWRITE: print job cancel not implemented\n"); return 1; } /**************************************************************************** cancel a print job ****************************************************************************/ static int cmd_cancel(void) { fstring buf; int job; if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) { d_printf("cancel ...\n"); return 1; } do { job = atoi(buf); do_cancel(job); } while (next_token_nr(NULL,buf,NULL,sizeof(buf))); return 0; } /**************************************************************************** print a file ****************************************************************************/ static int cmd_print(void) { pstring lname; pstring rname; char *p; if (!next_token_nr(NULL,lname,NULL, sizeof(lname))) { d_printf("print \n"); return 1; } pstrcpy(rname,lname); p = strrchr_m(rname,'/'); if (p) { slprintf(rname, sizeof(rname)-1, "%s-%d", p+1, (int)getpid()); } if (strequal(lname,"-")) { slprintf(rname, sizeof(rname)-1, "stdin-%d", (int)getpid()); } return do_put(rname, lname, False); } /**************************************************************************** show a print queue ****************************************************************************/ static int cmd_queue(void) { d_printf("REWRITE: print job queue not implemented\n"); return 0; } /**************************************************************************** delete some files ****************************************************************************/ static void do_del(file_info *finfo) { pstring mask; pstrcpy(mask,cur_dir); pstrcat(mask,finfo->name); if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) return; if (!cli_unlink(cli, mask)) { d_printf("%s deleting remote file %s\n",cli_errstr(cli),mask); } } /**************************************************************************** delete some files ****************************************************************************/ static int cmd_del(void) { pstring mask; fstring buf; uint16 attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; if (recurse) attribute |= FILE_ATTRIBUTE_DIRECTORY; pstrcpy(mask,cur_dir); if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) { d_printf("del \n"); return 1; } pstrcat(mask,buf); do_list(mask, attribute,do_del,False,False); return 0; } /**************************************************************************** delete a whole directory tree ****************************************************************************/ static int cmd_deltree(void) { pstring dname; fstring buf; int ret; pstrcpy(dname,cur_dir); if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) { d_printf("deltree \n"); return 1; } pstrcat(dname,buf); ret = cli_deltree(cli, dname); if (ret == -1) { printf("Failed to delete tree %s - %s\n", dname, cli_errstr(cli)); return -1; } printf("Deleted %d files in %s\n", ret, dname); return 0; } /**************************************************************************** show as much information as possible about a file ****************************************************************************/ static int cmd_allinfo(void) { pstring fname; fstring buf; int ret = 0; TALLOC_CTX *mem_ctx; union smb_fileinfo finfo; NTSTATUS status; pstrcpy(fname,cur_dir); if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) { d_printf("allinfo \n"); return 1; } pstrcat(fname,buf); mem_ctx = talloc_init(fname); /* first a ALL_INFO QPATHINFO */ finfo.generic.level = RAW_FILEINFO_ALL_INFO; finfo.generic.in.fname = fname; status = smb_raw_pathinfo(cli->tree, mem_ctx, &finfo); if (!NT_STATUS_IS_OK(status)) { d_printf("%s - %s\n", fname, nt_errstr(status)); ret = 1; goto done; } d_printf("\tcreate_time: %s\n", nt_time_string(mem_ctx, &finfo.all_info.out.create_time)); d_printf("\taccess_time: %s\n", nt_time_string(mem_ctx, &finfo.all_info.out.access_time)); d_printf("\twrite_time: %s\n", nt_time_string(mem_ctx, &finfo.all_info.out.write_time)); d_printf("\tchange_time: %s\n", nt_time_string(mem_ctx, &finfo.all_info.out.change_time)); d_printf("\tattrib: 0x%x\n", finfo.all_info.out.attrib); d_printf("\talloc_size: %lu\n", (unsigned long)finfo.all_info.out.alloc_size); d_printf("\tsize: %lu\n", (unsigned long)finfo.all_info.out.size); d_printf("\tnlink: %u\n", finfo.all_info.out.nlink); d_printf("\tdelete_pending: %u\n", finfo.all_info.out.delete_pending); d_printf("\tdirectory: %u\n", finfo.all_info.out.directory); d_printf("\tea_size: %u\n", finfo.all_info.out.ea_size); d_printf("\tfname: '%s'\n", finfo.all_info.out.fname.s); /* 8.3 name if any */ finfo.generic.level = RAW_FILEINFO_ALT_NAME_INFO; status = smb_raw_pathinfo(cli->tree, mem_ctx, &finfo); if (NT_STATUS_IS_OK(status)) { d_printf("\talt_name: %s\n", finfo.alt_name_info.out.fname.s); } /* dev/inode if available */ finfo.generic.level = RAW_FILEINFO_INTERNAL_INFORMATION; status = smb_raw_pathinfo(cli->tree, mem_ctx, &finfo); if (NT_STATUS_IS_OK(status)) { d_printf("\tdevice 0x%x\n", finfo.internal_information.out.device); d_printf("\tinode 0x%x\n", finfo.internal_information.out.inode); } /* the EAs, if any */ finfo.generic.level = RAW_FILEINFO_ALL_EAS; status = smb_raw_pathinfo(cli->tree, mem_ctx, &finfo); if (NT_STATUS_IS_OK(status)) { int i; for (i=0;itree, mem_ctx, &finfo); if (NT_STATUS_IS_OK(status)) { int i; for (i=0;itree, mem_ctx, &finfo); if (NT_STATUS_IS_OK(status)) { d_printf("\tcompressed size %ld\n", (long)finfo.compression_info.out.compressed_size); d_printf("\tformat %ld\n", (long)finfo.compression_info.out.format); d_printf("\tunit_shift %ld\n", (long)finfo.compression_info.out.unit_shift); d_printf("\tchunk_shift %ld\n", (long)finfo.compression_info.out.chunk_shift); d_printf("\tcluster_shift %ld\n", (long)finfo.compression_info.out.cluster_shift); } talloc_destroy(mem_ctx); done: return ret; } /**************************************************************************** ****************************************************************************/ static int cmd_open(void) { pstring mask; fstring buf; pstrcpy(mask,cur_dir); if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) { d_printf("open \n"); return 1; } pstrcat(mask,buf); cli_open(cli, mask, O_RDWR, DENY_ALL); return 0; } /**************************************************************************** remove a directory ****************************************************************************/ static int cmd_rmdir(void) { pstring mask; fstring buf; pstrcpy(mask,cur_dir); if (!next_token_nr(NULL,buf,NULL,sizeof(buf))) { d_printf("rmdir \n"); return 1; } pstrcat(mask,buf); if (!cli_rmdir(cli, mask)) { d_printf("%s removing remote directory file %s\n", cli_errstr(cli),mask); } return 0; } /**************************************************************************** UNIX hardlink. ****************************************************************************/ static int cmd_link(void) { pstring src,dest; fstring buf,buf2; if (!(cli->transport->negotiate.capabilities & CAP_UNIX)) { d_printf("Server doesn't support UNIX CIFS calls.\n"); return 1; } pstrcpy(src,cur_dir); pstrcpy(dest,cur_dir); if (!next_token(NULL,buf,NULL,sizeof(buf)) || !next_token(NULL,buf2,NULL, sizeof(buf2))) { d_printf("link \n"); return 1; } pstrcat(src,buf); pstrcat(dest,buf2); if (!cli_unix_hardlink(cli, src, dest)) { d_printf("%s linking files (%s -> %s)\n", cli_errstr(cli), src, dest); return 1; } return 0; } /**************************************************************************** UNIX symlink. ****************************************************************************/ static int cmd_symlink(void) { pstring src,dest; fstring buf,buf2; if (!(cli->transport->negotiate.capabilities & CAP_UNIX)) { d_printf("Server doesn't support UNIX CIFS calls.\n"); return 1; } pstrcpy(src,cur_dir); pstrcpy(dest,cur_dir); if (!next_token(NULL,buf,NULL,sizeof(buf)) || !next_token(NULL,buf2,NULL, sizeof(buf2))) { d_printf("symlink \n"); return 1; } pstrcat(src,buf); pstrcat(dest,buf2); if (!cli_unix_symlink(cli, src, dest)) { d_printf("%s symlinking files (%s -> %s)\n", cli_errstr(cli), src, dest); return 1; } return 0; } /**************************************************************************** UNIX chmod. ****************************************************************************/ static int cmd_chmod(void) { pstring src; mode_t mode; fstring buf, buf2; if (!(cli->transport->negotiate.capabilities & CAP_UNIX)) { d_printf("Server doesn't support UNIX CIFS calls.\n"); return 1; } pstrcpy(src,cur_dir); if (!next_token(NULL,buf,NULL,sizeof(buf)) || !next_token(NULL,buf2,NULL, sizeof(buf2))) { d_printf("chmod mode file\n"); return 1; } mode = (mode_t)strtol(buf, NULL, 8); pstrcat(src,buf2); if (!cli_unix_chmod(cli, src, mode)) { d_printf("%s chmod file %s 0%o\n", cli_errstr(cli), src, (unsigned int)mode); return 1; } return 0; } /**************************************************************************** UNIX chown. ****************************************************************************/ static int cmd_chown(void) { pstring src; uid_t uid; gid_t gid; fstring buf, buf2, buf3; if (!(cli->transport->negotiate.capabilities & CAP_UNIX)) { d_printf("Server doesn't support UNIX CIFS calls.\n"); return 1; } pstrcpy(src,cur_dir); if (!next_token(NULL,buf,NULL,sizeof(buf)) || !next_token(NULL,buf2,NULL, sizeof(buf2)) || !next_token(NULL,buf3,NULL, sizeof(buf3))) { d_printf("chown uid gid file\n"); return 1; } uid = (uid_t)atoi(buf); gid = (gid_t)atoi(buf2); pstrcat(src,buf3); if (!cli_unix_chown(cli, src, uid, gid)) { d_printf("%s chown file %s uid=%d, gid=%d\n", cli_errstr(cli), src, (int)uid, (int)gid); return 1; } return 0; } /**************************************************************************** rename some files ****************************************************************************/ static int cmd_rename(void) { pstring src,dest; fstring buf,buf2; pstrcpy(src,cur_dir); pstrcpy(dest,cur_dir); if (!next_token_nr(NULL,buf,NULL,sizeof(buf)) || !next_token_nr(NULL,buf2,NULL, sizeof(buf2))) { d_printf("rename \n"); return 1; } pstrcat(src,buf); pstrcat(dest,buf2); if (!cli_rename(cli, src, dest)) { d_printf("%s renaming files\n",cli_errstr(cli)); return 1; } return 0; } /**************************************************************************** toggle the prompt flag ****************************************************************************/ static int cmd_prompt(void) { prompt = !prompt; DEBUG(2,("prompting is now %s\n",prompt?"on":"off")); return 1; } /**************************************************************************** set the newer than time ****************************************************************************/ static int cmd_newer(void) { fstring buf; BOOL ok; SMB_STRUCT_STAT sbuf; ok = next_token_nr(NULL,buf,NULL,sizeof(buf)); if (ok && (sys_stat(buf,&sbuf) == 0)) { newer_than = sbuf.st_mtime; DEBUG(1,("Getting files newer than %s", asctime(LocalTime(&newer_than)))); } else { newer_than = 0; } if (ok && newer_than == 0) { d_printf("Error setting newer-than time\n"); return 1; } return 0; } /**************************************************************************** set the archive level ****************************************************************************/ static int cmd_archive(void) { fstring buf; if (next_token_nr(NULL,buf,NULL,sizeof(buf))) { archive_level = atoi(buf); } else d_printf("Archive level is %d\n",archive_level); return 0; } /**************************************************************************** toggle the lowercaseflag ****************************************************************************/ static int cmd_lowercase(void) { lowercase = !lowercase; DEBUG(2,("filename lowercasing is now %s\n",lowercase?"on":"off")); return 0; } /**************************************************************************** toggle the recurse flag ****************************************************************************/ static int cmd_recurse(void) { recurse = !recurse; DEBUG(2,("directory recursion is now %s\n",recurse?"on":"off")); return 0; } /**************************************************************************** toggle the translate flag ****************************************************************************/ static int cmd_translate(void) { translation = !translation; DEBUG(2,("CR/LF<->LF and print text translation now %s\n", translation?"on":"off")); return 0; } /**************************************************************************** do a printmode command ****************************************************************************/ static int cmd_printmode(void) { fstring buf; fstring mode; if (next_token_nr(NULL,buf,NULL,sizeof(buf))) { if (strequal(buf,"text")) { printmode = 0; } else { if (strequal(buf,"graphics")) printmode = 1; else printmode = atoi(buf); } } switch(printmode) { case 0: fstrcpy(mode,"text"); break; case 1: fstrcpy(mode,"graphics"); break; default: slprintf(mode,sizeof(mode)-1,"%d",printmode); break; } DEBUG(2,("the printmode is now %s\n",mode)); return 0; } /**************************************************************************** do the lcd command ****************************************************************************/ static int cmd_lcd(void) { fstring buf; pstring d; if (next_token_nr(NULL,buf,NULL,sizeof(buf))) chdir(buf); DEBUG(2,("the local directory is now %s\n",sys_getwd(d))); return 0; } /**************************************************************************** get a file restarting at end of local file ****************************************************************************/ static int cmd_reget(void) { pstring local_name; pstring remote_name; char *p; pstrcpy(remote_name, cur_dir); pstrcat(remote_name, "\\"); p = remote_name + strlen(remote_name); if (!next_token_nr(NULL, p, NULL, sizeof(remote_name) - strlen(remote_name))) { d_printf("reget \n"); return 1; } pstrcpy(local_name, p); dos_clean_name(remote_name); next_token_nr(NULL, local_name, NULL, sizeof(local_name)); return do_get(remote_name, local_name, True); } /**************************************************************************** put a file restarting at end of local file ****************************************************************************/ static int cmd_reput(void) { pstring local_name; pstring remote_name; fstring buf; char *p = buf; SMB_STRUCT_STAT st; pstrcpy(remote_name, cur_dir); pstrcat(remote_name, "\\"); if (!next_token_nr(NULL, p, NULL, sizeof(buf))) { d_printf("reput \n"); return 1; } pstrcpy(local_name, p); if (!file_exist(local_name, &st)) { d_printf("%s does not exist\n", local_name); return 1; } if (next_token_nr(NULL, p, NULL, sizeof(buf))) pstrcat(remote_name, p); else pstrcat(remote_name, local_name); dos_clean_name(remote_name); return do_put(remote_name, local_name, True); } /**************************************************************************** try and browse available connections on a host ****************************************************************************/ static BOOL browse_host(BOOL sort) { d_printf("REWRITE: host browsing not implemented\n"); return False; } /**************************************************************************** try and browse available connections on a host ****************************************************************************/ static BOOL list_servers(char *wk_grp) { d_printf("REWRITE: list servers not implemented\n"); return False; } /* Some constants for completing filename arguments */ #define COMPL_NONE 0 /* No completions */ #define COMPL_REMOTE 1 /* Complete remote filename */ #define COMPL_LOCAL 2 /* Complete local filename */ /* This defines the commands supported by this client. * NOTE: The "!" must be the last one in the list because it's fn pointer * field is NULL, and NULL in that field is used in process_tok() * (below) to indicate the end of the list. crh */ static struct { const char *name; int (*fn)(void); const char *description; char compl_args[2]; /* Completion argument info */ } commands[] = { {"?",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}}, {"altname",cmd_altname," show alt name",{COMPL_NONE,COMPL_NONE}}, {"allinfo",cmd_allinfo," show all possible info about a file",{COMPL_NONE,COMPL_NONE}}, {"archive",cmd_archive,"\n0=ignore archive bit\n1=only get archive files\n2=only get archive files and reset archive bit\n3=get all files and reset archive bit",{COMPL_NONE,COMPL_NONE}}, {"blocksize",cmd_block,"blocksize (default 20)",{COMPL_NONE,COMPL_NONE}}, {"cancel",cmd_cancel," cancel a print queue entry",{COMPL_NONE,COMPL_NONE}}, {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}}, {"chmod",cmd_chmod," chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_REMOTE}}, {"chown",cmd_chown," chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_REMOTE}}, {"del",cmd_del," delete all matching files",{COMPL_REMOTE,COMPL_NONE}}, {"deltree",cmd_deltree," delete a whole directory tree",{COMPL_REMOTE,COMPL_NONE}}, {"dir",cmd_dir," list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}}, {"du",cmd_du," computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}}, {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}}, {"get",cmd_get," [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}}, {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}}, {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}}, {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}}, {"link",cmd_link," create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}}, {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}}, {"ls",cmd_dir," list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}}, {"mask",cmd_select," mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}}, {"md",cmd_mkdir," make a directory",{COMPL_NONE,COMPL_NONE}}, {"mget",cmd_mget," get all the matching files",{COMPL_REMOTE,COMPL_NONE}}, {"mkdir",cmd_mkdir," make a directory",{COMPL_NONE,COMPL_NONE}}, {"more",cmd_more," view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}}, {"mput",cmd_mput," put all matching files",{COMPL_REMOTE,COMPL_NONE}}, {"newer",cmd_newer," only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}}, {"open",cmd_open," open a file",{COMPL_REMOTE,COMPL_NONE}}, {"print",cmd_print," print a file",{COMPL_NONE,COMPL_NONE}}, {"printmode",cmd_printmode," set the print mode",{COMPL_NONE,COMPL_NONE}}, {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}}, {"put",cmd_put," [remote name] put a file",{COMPL_LOCAL,COMPL_REMOTE}}, {"pwd",cmd_pwd,"show current remote directory (same as 'cd' with no args)",{COMPL_NONE,COMPL_NONE}}, {"q",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}}, {"queue",cmd_queue,"show the print queue",{COMPL_NONE,COMPL_NONE}}, {"quit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}}, {"rd",cmd_rmdir," remove a directory",{COMPL_NONE,COMPL_NONE}}, {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}}, {"reget",cmd_reget," [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}}, {"rename",cmd_rename," rename some files",{COMPL_REMOTE,COMPL_REMOTE}}, {"reput",cmd_reput," [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}}, {"rm",cmd_del," delete all matching files",{COMPL_REMOTE,COMPL_NONE}}, {"rmdir",cmd_rmdir," remove a directory",{COMPL_NONE,COMPL_NONE}}, {"setmode",cmd_setmode,"filename change modes of file",{COMPL_REMOTE,COMPL_NONE}}, {"symlink",cmd_symlink," create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}}, {"tar",cmd_tar,"tar [IXFqbgNan] current directory to/from ",{COMPL_NONE,COMPL_NONE}}, {"tarmode",cmd_tarmode," tar's behaviour towards archive bits",{COMPL_NONE,COMPL_NONE}}, {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}}, /* Yes, this must be here, see crh's comment above. */ {"!",NULL,"run a shell command on the local system",{COMPL_NONE,COMPL_NONE}}, {NULL,NULL,NULL,{COMPL_NONE,COMPL_NONE}} }; /******************************************************************* lookup a command string in the list of commands, including abbreviations ******************************************************************/ static int process_tok(fstring tok) { int i = 0, matches = 0; int cmd=0; int tok_len = strlen(tok); while (commands[i].fn != NULL) { if (strequal(commands[i].name,tok)) { matches = 1; cmd = i; break; } else if (strnequal(commands[i].name, tok, tok_len)) { matches++; cmd = i; } i++; } if (matches == 0) return(-1); else if (matches == 1) return(cmd); else return(-2); } /**************************************************************************** help ****************************************************************************/ static int cmd_help(void) { int i=0,j; fstring buf; if (next_token_nr(NULL,buf,NULL,sizeof(buf))) { if ((i = process_tok(buf)) >= 0) d_printf("HELP %s:\n\t%s\n\n",commands[i].name,commands[i].description); } else { while (commands[i].description) { for (j=0; commands[i].description && (j<5); j++) { d_printf("%-15s",commands[i].name); i++; } d_printf("\n"); } } return 0; } /**************************************************************************** process a -c command string ****************************************************************************/ static int process_command_string(char *cmd) { pstring line; const char *ptr; int rc = 0; /* establish the connection if not already */ if (!cli) { cli = do_connect(desthost, service); if (!cli) return 0; } while (cmd[0] != '\0') { char *p; fstring tok; int i; if ((p = strchr_m(cmd, ';')) == 0) { strncpy(line, cmd, 999); line[1000] = '\0'; cmd += strlen(cmd); } else { if (p - cmd > 999) p = cmd + 999; strncpy(line, cmd, p - cmd); line[p - cmd] = '\0'; cmd = p + 1; } /* and get the first part of the command */ ptr = line; if (!next_token_nr(&ptr,tok,NULL,sizeof(tok))) continue; if ((i = process_tok(tok)) >= 0) { rc = commands[i].fn(); } else if (i == -2) { d_printf("%s: command abbreviation ambiguous\n",tok); } else { d_printf("%s: command not found\n",tok); } } return rc; } #define MAX_COMPLETIONS 100 typedef struct { pstring dirmask; char **matches; int count, samelen; const char *text; int len; } completion_remote_t; static void completion_remote_filter(file_info *f, const char *mask, void *state) { completion_remote_t *info = (completion_remote_t *)state; if ((info->count < MAX_COMPLETIONS - 1) && (strncmp(info->text, f->name, info->len) == 0) && (strcmp(f->name, ".") != 0) && (strcmp(f->name, "..") != 0)) { if ((info->dirmask[0] == 0) && !(f->mode & FILE_ATTRIBUTE_DIRECTORY)) info->matches[info->count] = strdup(f->name); else { pstring tmp; if (info->dirmask[0] != 0) pstrcpy(tmp, info->dirmask); else tmp[0] = 0; pstrcat(tmp, f->name); if (f->mode & FILE_ATTRIBUTE_DIRECTORY) pstrcat(tmp, "/"); info->matches[info->count] = strdup(tmp); } if (info->matches[info->count] == NULL) return; if (f->mode & FILE_ATTRIBUTE_DIRECTORY) smb_readline_ca_char(0); if (info->count == 1) info->samelen = strlen(info->matches[info->count]); else while (strncmp(info->matches[info->count], info->matches[info->count-1], info->samelen) != 0) info->samelen--; info->count++; } } static char **remote_completion(const char *text, int len) { pstring dirmask; int i; completion_remote_t info = { "", NULL, 1, NULL, NULL, NULL }; info.samelen = len; info.text = text; info.len = len; if (len >= PATH_MAX) return(NULL); info.matches = (char **)malloc(sizeof(info.matches[0])*MAX_COMPLETIONS); if (!info.matches) return NULL; info.matches[0] = NULL; for (i = len-1; i >= 0; i--) if ((text[i] == '/') || (text[i] == '\\')) break; info.text = text+i+1; info.samelen = info.len = len-i-1; if (i > 0) { strncpy(info.dirmask, text, i+1); info.dirmask[i+1] = 0; snprintf(dirmask, sizeof(dirmask), "%s%*s*", cur_dir, i-1, text); } else snprintf(dirmask, sizeof(dirmask), "%s*", cur_dir); if (cli_list(cli, dirmask, FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN, completion_remote_filter, &info) < 0) goto cleanup; if (info.count == 2) info.matches[0] = strdup(info.matches[1]); else { info.matches[0] = malloc(info.samelen+1); if (!info.matches[0]) goto cleanup; strncpy(info.matches[0], info.matches[1], info.samelen); info.matches[0][info.samelen] = 0; } info.matches[info.count] = NULL; return info.matches; cleanup: for (i = 0; i < info.count; i++) free(info.matches[i]); free(info.matches); return NULL; } static char **completion_fn(const char *text, int start, int end) { smb_readline_ca_char(' '); if (start) { const char *buf, *sp; int i; char compl_type; buf = smb_readline_get_line_buffer(); if (buf == NULL) return NULL; sp = strchr(buf, ' '); if (sp == NULL) return NULL; for (i = 0; commands[i].name; i++) if ((strncmp(commands[i].name, text, sp - buf) == 0) && (commands[i].name[sp - buf] == 0)) break; if (commands[i].name == NULL) return NULL; while (*sp == ' ') sp++; if (sp == (buf + start)) compl_type = commands[i].compl_args[0]; else compl_type = commands[i].compl_args[1]; if (compl_type == COMPL_REMOTE) return remote_completion(text, end - start); else /* fall back to local filename completion */ return NULL; } else { char **matches; int i, len, samelen, count=1; matches = (char **)malloc(sizeof(matches[0])*MAX_COMPLETIONS); if (!matches) return NULL; matches[0] = NULL; len = strlen(text); for (i=0;commands[i].fn && count < MAX_COMPLETIONS-1;i++) { if (strncmp(text, commands[i].name, len) == 0) { matches[count] = strdup(commands[i].name); if (!matches[count]) goto cleanup; if (count == 1) samelen = strlen(matches[count]); else while (strncmp(matches[count], matches[count-1], samelen) != 0) samelen--; count++; } } switch (count) { case 0: /* should never happen */ case 1: goto cleanup; case 2: matches[0] = strdup(matches[1]); break; default: matches[0] = malloc(samelen+1); if (!matches[0]) goto cleanup; strncpy(matches[0], matches[1], samelen); matches[0][samelen] = 0; } matches[count] = NULL; return matches; cleanup: while (i >= 0) { free(matches[i]); i--; } free(matches); return NULL; } } /**************************************************************************** make sure we swallow keepalives during idle time ****************************************************************************/ static void readline_callback(void) { fd_set fds; struct timeval timeout; static time_t last_t; time_t t; t = time(NULL); if (t - last_t < 5) return; last_t = t; again: if (cli->transport->socket->fd == -1) return; FD_ZERO(&fds); FD_SET(cli->transport->socket->fd, &fds); timeout.tv_sec = 0; timeout.tv_usec = 0; sys_select_intr(cli->transport->socket->fd+1,&fds,NULL,NULL,&timeout); /* We deliberately use cli_swallow_keepalives instead of client_receive_smb as we want to receive session keepalives and then drop them here. */ if (FD_ISSET(cli->transport->socket->fd, &fds)) { if (!cli_request_receive_next(cli->transport)) { d_printf("Lost connection to server\n"); exit(1); } goto again; } if (cli->tree) { cli_chkpath(cli, "\\"); } } /**************************************************************************** process commands on stdin ****************************************************************************/ static void process_stdin(void) { const char *ptr; while (1) { fstring tok; fstring the_prompt; char *cline; pstring line; int i; /* display a prompt */ slprintf(the_prompt, sizeof(the_prompt)-1, "smb: %s> ", cur_dir); cline = smb_readline(the_prompt, readline_callback, completion_fn); if (!cline) break; pstrcpy(line, cline); /* special case - first char is ! */ if (*line == '!') { system(line + 1); continue; } /* and get the first part of the command */ ptr = line; if (!next_token_nr(&ptr,tok,NULL,sizeof(tok))) continue; if ((i = process_tok(tok)) >= 0) { commands[i].fn(); } else if (i == -2) { d_printf("%s: command abbreviation ambiguous\n",tok); } else { d_printf("%s: command not found\n",tok); } } } /***************************************************** return a connection to a server *******************************************************/ static struct cli_state *do_connect(const char *server, const char *share) { struct cli_state *c; struct nmb_name called, calling; const char *server_n; struct in_addr ip; fstring servicename; char *sharename; /* make a copy so we don't modify the global string 'service' */ fstrcpy(servicename, share); sharename = servicename; if (*sharename == '\\') { server = sharename+2; sharename = strchr_m(server,'\\'); if (!sharename) return NULL; *sharename = 0; sharename++; } asprintf(&sharename, "\\\\%s\\%s", server, sharename); server_n = server; zero_ip(&ip); make_nmb_name(&calling, lp_netbios_name(), 0x0); make_nmb_name(&called , server, name_type); again: zero_ip(&ip); if (have_ip) ip = dest_ip; /* have to open a new connection */ if (!(c=cli_state_init()) || !cli_socket_connect(c, server_n, &ip)) { d_printf("Connection to %s failed\n", server_n); return NULL; } if (!cli_transport_establish(c, &calling, &called)) { char *p; d_printf("session request to %s failed (%s)\n", called.name, cli_errstr(c)); cli_shutdown(c); if ((p=strchr_m(called.name, '.'))) { *p = 0; goto again; } if (strcmp(called.name, "*SMBSERVER")) { make_nmb_name(&called , "*SMBSERVER", 0x20); goto again; } return NULL; } DEBUG(4,(" session request ok\n")); if (!cli_negprot(c)) { d_printf("protocol negotiation failed\n"); cli_shutdown(c); return NULL; } if (!got_pass) { const char *pass = getpass("Password: "); if (pass) { pstrcpy(password, pass); } } if (!cli_session_setup(c, username, password, lp_workgroup())) { /* if a password was not supplied then try again with a null username */ if (password[0] || !username[0] || use_kerberos || !cli_session_setup(c, "", "", lp_workgroup())) { d_printf("session setup failed: %s\n", cli_errstr(c)); cli_shutdown(c); return NULL; } d_printf("Anonymous login successful\n"); } DEBUG(4,(" session setup ok\n")); if (!cli_send_tconX(c, sharename, "?????", password)) { d_printf("tree connect failed: %s\n", cli_errstr(c)); cli_shutdown(c); return NULL; } DEBUG(4,(" tconx ok\n")); return c; } /**************************************************************************** process commands from the client ****************************************************************************/ static int process(char *base_directory) { int rc = 0; cli = do_connect(desthost, service); if (!cli) { return 1; } if (*base_directory) do_cd(base_directory); if (cmdstr) { rc = process_command_string(cmdstr); } else { process_stdin(); } cli_shutdown(cli); return rc; } /**************************************************************************** handle a -L query ****************************************************************************/ static int do_host_query(char *query_host) { cli = do_connect(query_host, "IPC$"); if (!cli) return 1; browse_host(True); list_servers(lp_workgroup()); cli_shutdown(cli); return(0); } /**************************************************************************** handle a tar operation ****************************************************************************/ static int do_tar_op(char *base_directory) { int ret; /* do we already have a connection? */ if (!cli) { cli = do_connect(desthost, service); if (!cli) return 1; } recurse=True; if (*base_directory) do_cd(base_directory); ret=process_tar(); cli_shutdown(cli); return(ret); } /**************************************************************************** handle a message operation ****************************************************************************/ static int do_message_op(void) { struct in_addr ip; struct nmb_name called, calling; fstring server_name; char name_type_hex[10]; make_nmb_name(&calling, lp_netbios_name(), 0x0); make_nmb_name(&called , desthost, name_type); fstrcpy(server_name, desthost); snprintf(name_type_hex, sizeof(name_type_hex), "#%X", name_type); fstrcat(server_name, name_type_hex); zero_ip(&ip); if (have_ip) ip = dest_ip; if (!(cli=cli_state_init()) || !cli_socket_connect(cli, server_name, &ip)) { d_printf("Connection to %s failed\n", desthost); return 1; } if (!cli_transport_establish(cli, &calling, &called)) { d_printf("session request failed\n"); cli_shutdown(cli); return 1; } send_message(); cli_shutdown(cli); return 0; } /** * Process "-L hostname" option. * * We don't actually do anything yet -- we just stash the name in a * global variable and do the query when all options have been read. **/ static void remember_query_host(const char *arg, pstring query_host) { char *slash; while (*arg == '\\' || *arg == '/') arg++; pstrcpy(query_host, arg); if ((slash = strchr(query_host, '/')) || (slash = strchr(query_host, '\\'))) { *slash = 0; } } /**************************************************************************** main program ****************************************************************************/ int main(int argc,char *argv[]) { fstring base_directory; int opt; pstring query_host; BOOL message = False; extern char tar_type; pstring term_code; poptContext pc; char *p; int rc = 0; TALLOC_CTX *mem_ctx; struct poptOption long_options[] = { POPT_AUTOHELP { "name-resolve", 'R', POPT_ARG_STRING, NULL, 'R', "Use these name resolution services only", "NAME-RESOLVE-ORDER" }, { "message", 'M', POPT_ARG_STRING, NULL, 'M', "Send message", "HOST" }, { "ip-address", 'I', POPT_ARG_STRING, NULL, 'I', "Use this IP to connect to", "IP" }, { "stderr", 'E', POPT_ARG_NONE, NULL, 'E', "Write messages to stderr instead of stdout" }, { "list", 'L', POPT_ARG_STRING, NULL, 'L', "Get a list of shares available on a host", "HOST" }, { "terminal", 't', POPT_ARG_STRING, NULL, 't', "Terminal I/O code {sjis|euc|jis7|jis8|junet|hex}", "CODE" }, { "max-protocol", 'm', POPT_ARG_STRING, NULL, 'm', "Set the max protocol level", "LEVEL" }, { "tar", 'T', POPT_ARG_STRING, NULL, 'T', "Command line tar", "IXFqgbNan" }, { "directory", 'D', POPT_ARG_STRING, NULL, 'D', "Start from directory", "DIR" }, { "command", 'c', POPT_ARG_STRING, &cmdstr, 'c', "Execute semicolon separated commands" }, { "send-buffer", 'b', POPT_ARG_INT, NULL, 'b', "Changes the transmit/send buffer", "BYTES" }, { "port", 'p', POPT_ARG_INT, &port, 'p', "Port to connect to", "PORT" }, POPT_COMMON_SAMBA POPT_COMMON_CONNECTION POPT_COMMON_CREDENTIALS POPT_TABLEEND }; #ifdef KANJI pstrcpy(term_code, KANJI); #else /* KANJI */ *term_code = 0; #endif /* KANJI */ *query_host = 0; *base_directory = 0; setup_logging(argv[0],True); mem_ctx = talloc_init("client.c/main"); if (!mem_ctx) { d_printf("\nclient.c: Not enough memory\n"); exit(1); } if (!lp_load(dyn_CONFIGFILE,True,False,False)) { fprintf(stderr, "%s: Can't load %s - run testparm to debug it\n", argv[0], dyn_CONFIGFILE); } pc = poptGetContext("smbclient", argc, (const char **) argv, long_options, POPT_CONTEXT_KEEP_FIRST); poptSetOtherOptionHelp(pc, "service "); in_client = True; /* Make sure that we tell lp_load we are */ while ((opt = poptGetNextOpt(pc)) != -1) { switch (opt) { case 'M': /* Messages are sent to NetBIOS name type 0x3 * (Messenger Service). Make sure we default * to port 139 instead of port 445. srl,crh */ name_type = 0x03; pstrcpy(desthost,poptGetOptArg(pc)); if( 0 == port ) port = 139; message = True; break; case 'I': { dest_ip = *interpret_addr2(mem_ctx, poptGetOptArg(pc)); if (is_zero_ip(dest_ip)) exit(1); have_ip = True; } break; case 'E': setup_logging("client", DEBUG_STDERR); break; case 'L': remember_query_host(poptGetOptArg(pc), query_host); break; case 't': pstrcpy(term_code, poptGetOptArg(pc)); break; case 'm': lp_set_cmdline("max protocol", poptGetOptArg(pc)); break; case 'R': lp_set_cmdline("name resolve order", poptGetOptArg(pc)); break; case 'T': if (!tar_parseargs(argc, argv, poptGetOptArg(pc), optind)) { poptPrintUsage(pc, stderr, 0); exit(1); } break; case 'D': fstrcpy(base_directory,poptGetOptArg(pc)); break; case 'b': io_bufsize = MAX(1, atoi(poptGetOptArg(pc))); break; } } poptGetArg(pc); load_interfaces(); if(poptPeekArg(pc)) { pstrcpy(service,poptGetArg(pc)); /* Convert any '/' characters in the service name to '\' characters */ string_replace(service, '/','\\'); if (count_chars(service,'\\') < 3) { d_printf("\n%s: Not enough '\\' characters in service\n",service); poptPrintUsage(pc, stderr, 0); exit(1); } } if (poptPeekArg(pc)) { cmdline_auth_info.got_pass = True; pstrcpy(cmdline_auth_info.password,poptGetArg(pc)); } //init_names(); if (!tar_type && !*query_host && !*service && !message) { poptPrintUsage(pc, stderr, 0); exit(1); } poptFreeContext(pc); pstrcpy(username, cmdline_auth_info.username); pstrcpy(password, cmdline_auth_info.password); use_kerberos = cmdline_auth_info.use_kerberos; got_pass = cmdline_auth_info.got_pass; DEBUG( 3, ( "Client started (version %s).\n", SAMBA_VERSION ) ); talloc_destroy(mem_ctx); if (tar_type) { if (cmdstr) process_command_string(cmdstr); return do_tar_op(base_directory); } if ((p=strchr_m(query_host,'#'))) { *p = 0; p++; sscanf(p, "%x", &name_type); } if (*query_host) { return do_host_query(query_host); } if (message) { return do_message_op(); } if (process(base_directory)) { return 1; } return rc; }