diff options
Diffstat (limited to 'source3/client')
| -rw-r--r-- | source3/client/README.smbspool | 17 | ||||
| -rw-r--r-- | source3/client/client.c | 6739 | ||||
| -rw-r--r-- | source3/client/client_proto.h | 53 | ||||
| -rw-r--r-- | source3/client/clitar.c | 1903 | ||||
| -rw-r--r-- | source3/client/clitar_proto.h | 34 | ||||
| -rw-r--r-- | source3/client/dnsbrowse.c | 216 | ||||
| -rw-r--r-- | source3/client/smbspool.c | 888 | ||||
| -rw-r--r-- | source3/client/smbspool_krb5_wrapper.c | 341 | ||||
| -rw-r--r-- | source3/client/tree.c | 812 |
9 files changed, 11003 insertions, 0 deletions
diff --git a/source3/client/README.smbspool b/source3/client/README.smbspool new file mode 100644 index 0000000..f73167a --- /dev/null +++ b/source3/client/README.smbspool @@ -0,0 +1,17 @@ +smbspool +========= + +smbspool is a very small print spooling program that sends a print file to an +SMB printer. The command-line arguments are position-dependent for +compatibility with the CUPS. + +For printing support with Kerberos, CUPS 1.5+ needs a wrapper for the backend +which sets the correct location of the Kerberos credential cache. + +smbspool_krb5_wrapper +====================== + +This tool can be used to print using Kerberos credentials. To get this working +smbspool_krb5_wrapper needs to be the smb backend of CUPS. It needs to be owned +by root and the permissions for the binary need to be 0700. Once +smbspool_krb5_wrapper switched to the user trying to print it executes smbspool. diff --git a/source3/client/client.c b/source3/client/client.c new file mode 100644 index 0000000..7d1ec25 --- /dev/null +++ b/source3/client/client.c @@ -0,0 +1,6739 @@ +/* + 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) Gerald (Jerry) Carter 2004 + Copyright (C) Jeremy Allison 1994-2007 + + 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 3 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, see <http://www.gnu.org/licenses/>. +*/ + +#include "includes.h" +#include "system/filesys.h" +#include "rpc_client/cli_pipe.h" +#include "client/client_proto.h" +#include "client/clitar_proto.h" +#include "../librpc/gen_ndr/ndr_srvsvc_c.h" +#include "../lib/util/select.h" +#include "system/readline.h" +#include "../libcli/smbreadline/smbreadline.h" +#include "../libcli/security/security.h" +#include "system/select.h" +#include "libsmb/libsmb.h" +#include "libsmb/clirap.h" +#include "trans2.h" +#include "libsmb/nmblib.h" +#include "include/ntioctl.h" +#include "../libcli/smb/smbXcli_base.h" +#include "lib/util/time_basic.h" +#include "lib/util/string_wrappers.h" +#include "lib/cmdline/cmdline.h" + +#ifndef REGISTER +#define REGISTER 0 +#endif + +extern int do_smb_browse(void); /* mDNS browsing */ + +static int port = 0; +static char *service; +static char *desthost; +static bool grepable = false; +static bool quiet = false; +static char *cmdstr = NULL; +const char *cmd_ptr = NULL; + +static int io_bufsize = 0; /* we use the default size */ +static int io_timeout = (CLIENT_TIMEOUT/1000); /* Per operation timeout (in seconds). */ + +static int name_type = 0x20; + +static int process_tok(char *tok); +static int cmd_help(void); + +/* 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; + +static bool prompt = true; + +static bool recurse = false; +static bool showacls = false; +bool lowercase = false; +static bool backup_intent = false; + +static struct sockaddr_storage dest_ss; +static char dest_ss_str[INET6_ADDRSTRLEN]; + +#define SEPARATORS " \t\n\r" + +/* timing globals */ +uint64_t get_total_size = 0; +unsigned int get_total_time_ms = 0; +static uint64_t put_total_size = 0; +static unsigned int put_total_time_ms = 0; + +/* totals globals */ +static double dir_total; + +/* root cli_state connection */ + +struct cli_state *cli; + +static char CLI_DIRSEP_CHAR = '\\'; +static char CLI_DIRSEP_STR[] = { '\\', '\0' }; + +/* Accessor functions for directory paths. */ +static char *fileselection; +static const char *client_get_fileselection(void) +{ + if (fileselection) { + return fileselection; + } + return ""; +} + +static const char *client_set_fileselection(const char *new_fs) +{ + SAFE_FREE(fileselection); + if (new_fs) { + fileselection = SMB_STRDUP(new_fs); + } + return client_get_fileselection(); +} + +static char *cwd; +static const char *client_get_cwd(void) +{ + if (cwd) { + return cwd; + } + return CLI_DIRSEP_STR; +} + +static const char *client_set_cwd(const char *new_cwd) +{ + SAFE_FREE(cwd); + if (new_cwd) { + cwd = SMB_STRDUP(new_cwd); + } + return client_get_cwd(); +} + +static char *cur_dir; +const char *client_get_cur_dir(void) +{ + if (cur_dir) { + return cur_dir; + } + return CLI_DIRSEP_STR; +} + +const char *client_set_cur_dir(const char *newdir) +{ + SAFE_FREE(cur_dir); + if (newdir) { + cur_dir = SMB_STRDUP(newdir); + } + return client_get_cur_dir(); +} + +/**************************************************************************** + Put up a yes/no prompt. +****************************************************************************/ + +static bool yesno(const char *p) +{ + char ans[20]; + printf("%s",p); + + if (!fgets(ans,sizeof(ans)-1,stdin)) + return(False); + + if (*ans == 'y' || *ans == 'Y') + return(True); + + return(False); +} + +/**************************************************************************** + 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 ssize_t writefile(int f, char *b, size_t n) +{ + size_t i = 0; + + if (n == 0) { + errno = EINVAL; + return -1; + } + + if (!translation) { + return write(f,b,n); + } + + do { + if (*b == '\r' && (i<(n-1)) && *(b+1) == '\n') { + b++;i++; + } + if (write(f, b, 1) != 1) { + break; + } + b++; + i++; + } while (i < n); + + return (ssize_t)i; +} + +/**************************************************************************** + Read from a file with LF->CR/LF translation if appropriate. Return the + number read. read approx n bytes. +****************************************************************************/ + +static int readfile(uint8_t *b, int n, FILE *f) +{ + int i; + int c; + + if (!translation) + return fread(b,1,n,f); + + i = 0; + while (i < (n - 1)) { + if ((c = getc(f)) == EOF) { + break; + } + + if (c == '\n') { /* change all LFs to CR/LF */ + b[i++] = '\r'; + } + + b[i++] = c; + } + + return(i); +} + +struct push_state { + FILE *f; + off_t nread; +}; + +static size_t push_source(uint8_t *buf, size_t n, void *priv) +{ + struct push_state *state = (struct push_state *)priv; + int result; + + if (feof(state->f)) { + return 0; + } + + result = readfile(buf, n, state->f); + state->nread += result; + return result; +} + +/**************************************************************************** + Send a message. +****************************************************************************/ + +static void send_message(const char *username) +{ + char buf[1600]; + NTSTATUS status; + size_t i; + + d_printf("Type your message, ending it with a Control-D\n"); + + i = 0; + while (i<sizeof(buf)-2) { + int c = fgetc(stdin); + if (c == EOF) { + break; + } + if (c == '\n') { + buf[i++] = '\r'; + } + buf[i++] = c; + } + buf[i] = '\0'; + + status = cli_message(cli, desthost, username, buf); + if (!NT_STATUS_IS_OK(status)) { + d_fprintf(stderr, "cli_message returned %s\n", + nt_errstr(status)); + } +} + +/**************************************************************************** + Check the space on a device. +****************************************************************************/ + +static int do_dskattr(void) +{ + uint64_t total, bsize, avail; + struct cli_state *targetcli = NULL; + char *targetpath = NULL; + TALLOC_CTX *ctx = talloc_tos(); + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + status = cli_resolve_path(ctx, + "", + creds, + cli, + client_get_cur_dir(), &targetcli, + &targetpath); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error in dskattr: %s\n", nt_errstr(status)); + return 1; + } + + status = cli_disk_size(targetcli, targetpath, &bsize, &total, &avail); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error in dskattr: %s\n", nt_errstr(status)); + return 1; + } + + d_printf("\n\t\t%" PRIu64 + " blocks of size %" PRIu64 + ". %" PRIu64 " 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",client_get_cur_dir()); + return 0; +} + +/**************************************************************************** + Ensure name has correct directory separators. +****************************************************************************/ + +static void normalize_name(char *newdir) +{ + if (!(cli->requested_posix_capabilities & CIFS_UNIX_POSIX_PATHNAMES_CAP)) { + string_replace(newdir,'/','\\'); + } +} + +/**************************************************************************** + Local name cleanup before sending to server. SMB1 allows relative pathnames, + but SMB2 does not, so we need to resolve them locally. +****************************************************************************/ + +char *client_clean_name(TALLOC_CTX *ctx, const char *name) +{ + char *newname = NULL; + if (name == NULL) { + return NULL; + } + + /* First ensure any path separators are correct. */ + newname = talloc_strdup(ctx, name); + if (newname == NULL) { + return NULL; + } + normalize_name(newname); + + /* Now remove any relative (..) path components. */ + if (cli->requested_posix_capabilities & CIFS_UNIX_POSIX_PATHNAMES_CAP) { + newname = unix_clean_name(ctx, newname); + } else { + newname = clean_name(ctx, newname); + } + if (newname == NULL) { + return NULL; + } + return newname; +} + +/**************************************************************************** + Change directory - inner section. +****************************************************************************/ + +static int do_cd(const char *new_dir) +{ + char *newdir = NULL; + char *saved_dir = NULL; + char *new_cd = NULL; + char *targetpath = NULL; + struct cli_state *targetcli = NULL; + int ret = 1; + TALLOC_CTX *ctx = talloc_stackframe(); + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + newdir = talloc_strdup(ctx, new_dir); + if (!newdir) { + TALLOC_FREE(ctx); + return 1; + } + + normalize_name(newdir); + + /* Save the current directory in case the new directory is invalid */ + + saved_dir = talloc_strdup(ctx, client_get_cur_dir()); + if (!saved_dir) { + TALLOC_FREE(ctx); + return 1; + } + + if (*newdir == CLI_DIRSEP_CHAR) { + client_set_cur_dir(newdir); + new_cd = newdir; + } else { + new_cd = talloc_asprintf(ctx, "%s%s", + client_get_cur_dir(), + newdir); + if (!new_cd) { + goto out; + } + } + + /* Ensure cur_dir ends in a DIRSEP */ + if ((new_cd[0] != '\0') && (*(new_cd+strlen(new_cd)-1) != CLI_DIRSEP_CHAR)) { + new_cd = talloc_asprintf_append(new_cd, "%s", CLI_DIRSEP_STR); + if (!new_cd) { + goto out; + } + } + client_set_cur_dir(new_cd); + + new_cd = client_clean_name(ctx, new_cd); + client_set_cur_dir(new_cd); + + status = cli_resolve_path(ctx, "", + creds, + cli, new_cd, &targetcli, &targetpath); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cd %s: %s\n", new_cd, nt_errstr(status)); + client_set_cur_dir(saved_dir); + goto out; + } + + if (strequal(targetpath,CLI_DIRSEP_STR )) { + TALLOC_FREE(ctx); + return 0; + } + + + targetpath = talloc_asprintf( + ctx, "%s%s", targetpath, CLI_DIRSEP_STR); + if (!targetpath) { + client_set_cur_dir(saved_dir); + goto out; + } + targetpath = client_clean_name(ctx, targetpath); + if (!targetpath) { + client_set_cur_dir(saved_dir); + goto out; + } + + status = cli_chkpath(targetcli, targetpath); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cd %s: %s\n", new_cd, nt_errstr(status)); + client_set_cur_dir(saved_dir); + goto out; + } + + ret = 0; + +out: + + TALLOC_FREE(ctx); + return ret; +} + +/**************************************************************************** + Change directory. +****************************************************************************/ + +static int cmd_cd(void) +{ + char *buf = NULL; + int rc = 0; + + if (next_token_talloc(talloc_tos(), &cmd_ptr, &buf,NULL)) { + rc = do_cd(buf); + } else { + d_printf("Current directory is %s\n",client_get_cur_dir()); + } + + return rc; +} + +/**************************************************************************** + Change directory. +****************************************************************************/ + +static int cmd_cd_oneup(void) +{ + return do_cd(".."); +} + +/******************************************************************* + Decide if a file should be operated on. +********************************************************************/ + +static bool do_this_one(struct file_info *finfo) +{ + if (!finfo->name) { + return false; + } + + if (finfo->attr & FILE_ATTRIBUTE_DIRECTORY) { + return true; + } + + if (*client_get_fileselection() && + !mask_match(finfo->name,client_get_fileselection(),false)) { + DEBUG(3,("mask_match %s failed\n", finfo->name)); + return false; + } + + if (newer_than && finfo->mtime_ts.tv_sec < newer_than) { + DEBUG(3,("newer_than %s failed\n", finfo->name)); + return false; + } + + if ((archive_level==1 || archive_level==2) && !(finfo->attr & FILE_ATTRIBUTE_ARCHIVE)) { + DEBUG(3,("archive %s failed\n", finfo->name)); + return false; + } + + return true; +} + +/**************************************************************************** + Display info about a file. +****************************************************************************/ + +static NTSTATUS display_finfo(struct cli_state *cli_state, struct file_info *finfo, + const char *dir) +{ + time_t t; + TALLOC_CTX *ctx = talloc_tos(); + NTSTATUS status = NT_STATUS_OK; + + if (!do_this_one(finfo)) { + return NT_STATUS_OK; + } + + t = finfo->mtime_ts.tv_sec; /* the time is assumed to be passed as GMT */ + if (!showacls) { + d_printf(" %-30s%7.7s %8.0f %s", + finfo->name, + attrib_string(talloc_tos(), finfo->attr), + (double)finfo->size, + time_to_asc(t)); + dir_total += finfo->size; + } else { + struct cli_state *targetcli = NULL; + char *targetpath = NULL; + char *afname = NULL; + uint16_t fnum; + struct cli_credentials *creds = samba_cmdline_get_creds(); + + if (ISDOT(finfo->name) || ISDOTDOT(finfo->name)) { + return NT_STATUS_OK; + } + /* create absolute filename for cli_ntcreate() FIXME */ + afname = talloc_asprintf(ctx, + "%s%s%s", + dir, + CLI_DIRSEP_STR, + finfo->name); + if (!afname) { + return NT_STATUS_NO_MEMORY; + } + /* print file meta date header */ + d_printf( "FILENAME:%s\n", finfo->name); + d_printf( "MODE:%s\n", attrib_string(talloc_tos(), finfo->attr)); + d_printf( "SIZE:%.0f\n", (double)finfo->size); + d_printf( "MTIME:%s", time_to_asc(t)); + + status = cli_resolve_path( + ctx, + "", + creds, + cli_state, + afname, + &targetcli, + &targetpath); + if (!NT_STATUS_IS_OK(status)) { + DBG_WARNING("display_finfo() Failed to resolve " + "%s: %s\n", + afname, nt_errstr(status)); + return status; + } + + status = cli_ntcreate( + targetcli, /* cli */ + targetpath, /* fname */ + 0, /* CreatFlags */ + READ_CONTROL_ACCESS, /* DesiredAccess */ + 0, /* FileAttributes */ + FILE_SHARE_READ| + FILE_SHARE_WRITE, /* ShareAccess */ + FILE_OPEN, /* CreateDisposition */ + 0x0, /* CreateOptions */ + 0x0, /* SecurityFlags */ + &fnum, /* pfid */ + NULL); /* cr */ + if (!NT_STATUS_IS_OK(status)) { + DEBUG( 0, ("display_finfo() Failed to open %s: %s\n", + afname, nt_errstr(status))); + } else { + struct security_descriptor *sd = NULL; + status = cli_query_secdesc(targetcli, fnum, + ctx, &sd); + if (!NT_STATUS_IS_OK(status)) { + DEBUG( 0, ("display_finfo() failed to " + "get security descriptor: %s", + nt_errstr(status))); + } else { + display_sec_desc(sd); + } + TALLOC_FREE(sd); + cli_close(targetcli, fnum); + } + TALLOC_FREE(afname); + } + return status; +} + +/**************************************************************************** + Accumulate size of a file. +****************************************************************************/ + +static NTSTATUS do_du(struct cli_state *cli_state, struct file_info *finfo, + const char *dir) +{ + if (do_this_one(finfo)) { + dir_total += finfo->size; + } + return NT_STATUS_OK; +} + +struct do_list_queue_entry { + struct do_list_queue_entry *prev, *next; + char name[]; +}; + +struct do_list_queue { + struct do_list_queue_entry *list; +}; + +static bool do_list_recurse; +static bool do_list_dirs; +static struct do_list_queue *queue = NULL; +static NTSTATUS (*do_list_fn)(struct cli_state *cli_state, struct file_info *, + const char *dir); + +/**************************************************************************** + Functions for do_list_queue. +****************************************************************************/ + +static void reset_do_list_queue(void) +{ + TALLOC_FREE(queue); +} + +static void init_do_list_queue(void) +{ + TALLOC_FREE(queue); + queue = talloc_zero(NULL, struct do_list_queue); +} + +static void add_to_do_list_queue(const char *entry) +{ + struct do_list_queue_entry *e = NULL; + size_t entry_str_len = strlen(entry)+1; + size_t entry_len = offsetof(struct do_list_queue_entry, name); + + entry_len += entry_str_len; + SMB_ASSERT(entry_len >= entry_str_len); + + e = talloc_size(queue, entry_len); + if (e == NULL) { + d_printf("talloc failed for entry %s\n", entry); + return; + } + talloc_set_name_const(e, "struct do_list_queue_entry"); + + memcpy(e->name, entry, entry_str_len); + DLIST_ADD_END(queue->list, e); +} + +static char *do_list_queue_head(void) +{ + return queue->list->name; +} + +static void remove_do_list_queue_head(void) +{ + struct do_list_queue_entry *e = queue->list; + DLIST_REMOVE(queue->list, e); + TALLOC_FREE(e); +} + +static int do_list_queue_empty(void) +{ + return (queue == NULL) || (queue->list == NULL); +} + +/**************************************************************************** + A helper for do_list. +****************************************************************************/ + +struct do_list_helper_state { + const char *mask; + struct cli_state *cli; +}; + +static NTSTATUS do_list_helper( + struct file_info *f, + const char *_mask, + void *private_data) +{ + struct do_list_helper_state *state = private_data; + TALLOC_CTX *ctx = talloc_tos(); + char *dir = NULL; + char *dir_end = NULL; + NTSTATUS status = NT_STATUS_OK; + char *mask2 = NULL; + + /* Work out the directory. */ + dir = talloc_strdup(ctx, state->mask); + if (!dir) { + return NT_STATUS_NO_MEMORY; + } + if ((dir_end = strrchr(dir, CLI_DIRSEP_CHAR)) != NULL) { + *dir_end = '\0'; + } + + if (!(f->attr & FILE_ATTRIBUTE_DIRECTORY)) { + if (do_this_one(f)) { + status = do_list_fn(state->cli, f, dir); + } + TALLOC_FREE(dir); + return status; + } + + if (do_list_dirs && do_this_one(f)) { + status = do_list_fn(state->cli, f, dir); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + } + + if (!do_list_recurse || + (f->name == NULL) || + ISDOT(f->name) || + ISDOTDOT(f->name)) { + return NT_STATUS_OK; + } + + if (!f->name[0]) { + d_printf("Empty dir name returned. Possible server misconfiguration.\n"); + TALLOC_FREE(dir); + return NT_STATUS_UNSUCCESSFUL; + } + + mask2 = talloc_asprintf(ctx, + "%s%c%s%c*", + dir, + CLI_DIRSEP_CHAR, + f->name, + CLI_DIRSEP_CHAR); + if (!mask2) { + TALLOC_FREE(dir); + return NT_STATUS_NO_MEMORY; + } + add_to_do_list_queue(mask2); + TALLOC_FREE(mask2); + + TALLOC_FREE(dir); + return NT_STATUS_OK; +} + +/**************************************************************************** + A wrapper around cli_list that adds recursion. +****************************************************************************/ + +NTSTATUS do_list(const char *mask, + uint32_t attribute, + NTSTATUS (*fn)(struct cli_state *cli_state, struct file_info *, + const char *dir), + bool rec, + bool dirs) +{ + struct do_list_helper_state state = { .cli = cli, }; + static int in_do_list = 0; + TALLOC_CTX *ctx = talloc_tos(); + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS ret_status = NT_STATUS_OK; + NTSTATUS status = NT_STATUS_OK; + + 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; + + init_do_list_queue(); + add_to_do_list_queue(mask); + + while (!do_list_queue_empty()) { + struct cli_state *targetcli = NULL; + char *targetpath = NULL; + + state.mask = do_list_queue_head(); + + /* check for dfs */ + + status = cli_resolve_path( + ctx, + "", + creds, + cli, + state.mask, + &targetcli, + &targetpath); + if (!NT_STATUS_IS_OK(status)) { + d_printf("do_list: [%s] %s\n", state.mask, + nt_errstr(status)); + remove_do_list_queue_head(); + continue; + } + + status = cli_list( + targetcli, + targetpath, + attribute, + do_list_helper, + &state); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s listing %s\n", + nt_errstr(status), targetpath); + ret_status = status; + } + 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] == CLI_DIRSEP_CHAR)) { + save_ch = next_file + + strlen(next_file) - 2; + *save_ch = '\0'; + if (showacls) { + /* cwd is only used if showacls is on */ + client_set_cwd(next_file); + } + } + if (!showacls) /* don't disturbe the showacls output */ + d_printf("\n%s\n",next_file); + if (save_ch) { + *save_ch = CLI_DIRSEP_CHAR; + } + } + TALLOC_FREE(targetpath); + } + + in_do_list = 0; + reset_do_list_queue(); + return ret_status; +} + +/**************************************************************************** + Get a directory listing. +****************************************************************************/ + +static int cmd_dir(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + uint32_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; + char *mask = NULL; + char *buf = NULL; + int rc = 1; + NTSTATUS status; + + dir_total = 0; + mask = talloc_strdup(ctx, client_get_cur_dir()); + if (!mask) { + return 1; + } + + if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + normalize_name(buf); + if (*buf == CLI_DIRSEP_CHAR) { + mask = talloc_strdup(ctx, buf); + } else { + mask = talloc_asprintf_append(mask, "%s", buf); + } + } else { + mask = talloc_asprintf_append(mask, "*"); + } + if (!mask) { + return 1; + } + + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + if (showacls) { + /* cwd is only used if showacls is on */ + client_set_cwd(client_get_cur_dir()); + } + + status = do_list(mask, attribute, display_finfo, recurse, true); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + + rc = do_dskattr(); + + DEBUG(3, ("Total bytes listed: %.0f\n", dir_total)); + + return rc; +} + +/**************************************************************************** + Get a directory listing. +****************************************************************************/ + +static int cmd_du(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + uint32_t attribute = FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; + char *mask = NULL; + char *buf = NULL; + NTSTATUS status; + int rc = 1; + + dir_total = 0; + mask = talloc_strdup(ctx, client_get_cur_dir()); + if (!mask) { + return 1; + } + if ((mask[0] != '\0') && (mask[strlen(mask)-1]!=CLI_DIRSEP_CHAR)) { + mask = talloc_asprintf_append(mask, "%s", CLI_DIRSEP_STR); + if (!mask) { + return 1; + } + } + + if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + normalize_name(buf); + if (*buf == CLI_DIRSEP_CHAR) { + mask = talloc_strdup(ctx, buf); + } else { + mask = talloc_asprintf_append(mask, "%s", buf); + } + } else { + mask = talloc_strdup(ctx, "*"); + } + if (!mask) { + return 1; + } + + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = do_list(mask, attribute, do_du, recurse, true); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + + rc = do_dskattr(); + + d_printf("Total number of bytes: %.0f\n", dir_total); + + return rc; +} + +static int cmd_echo(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *num; + char *data; + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr, &num, NULL) + || !next_token_talloc(ctx, &cmd_ptr, &data, NULL)) { + d_printf("echo <num> <data>\n"); + return 1; + } + + status = cli_echo(cli, atoi(num), data_blob_const(data, strlen(data))); + + if (!NT_STATUS_IS_OK(status)) { + d_printf("echo failed: %s\n", nt_errstr(status)); + return 1; + } + + return 0; +} + +/**************************************************************************** + Get a file from rname to lname +****************************************************************************/ + +static NTSTATUS writefile_sink(char *buf, size_t n, void *priv) +{ + int *pfd = (int *)priv; + ssize_t rc; + + rc = writefile(*pfd, buf, n); + if (rc == -1) { + return map_nt_error_from_unix(errno); + } + return NT_STATUS_OK; +} + +static int do_get(const char *rname, const char *lname_in, bool reget) +{ + TALLOC_CTX *ctx = talloc_tos(); + int handle = 0; + uint16_t fnum; + bool newhandle = false; + struct timespec tp_start; + uint32_t attr; + off_t size; + off_t start = 0; + off_t nread = 0; + int rc = 0; + struct cli_state *targetcli = NULL; + char *targetname = NULL; + char *lname = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + lname = talloc_strdup(ctx, lname_in); + if (!lname) { + return 1; + } + + if (lowercase) { + if (!strlower_m(lname)) { + d_printf("strlower_m %s failed\n", lname); + return 1; + } + } + + status = cli_resolve_path(ctx, "", + creds, + cli, rname, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to open %s: %s\n", rname, nt_errstr(status)); + return 1; + } + + clock_gettime_mono(&tp_start); + + status = cli_open(targetcli, targetname, O_RDONLY, DENY_NONE, &fnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s opening remote file %s\n", nt_errstr(status), + rname); + return 1; + } + + if(!strcmp(lname,"-")) { + handle = fileno(stdout); + } else { + if (reget) { + handle = open(lname, O_WRONLY|O_CREAT, 0644); + if (handle >= 0) { + start = lseek(handle, 0, SEEK_END); + if (start == -1) { + d_printf("Error seeking local file\n"); + close(handle); + return 1; + } + } + } else { + handle = 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; + } + + + status = cli_qfileinfo_basic(targetcli, fnum, &attr, &size, NULL, NULL, + NULL, NULL, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("getattrib: %s\n", nt_errstr(status)); + if (newhandle) { + close(handle); + } + return 1; + } + + DEBUG(1,("getting file %s of size %.0f as %s ", + rname, (double)size, lname)); + + status = cli_pull(targetcli, fnum, start, size, io_bufsize, + writefile_sink, (void *)&handle, &nread); + if (!NT_STATUS_IS_OK(status)) { + d_fprintf(stderr, "parallel_read returned %s\n", + nt_errstr(status)); + if (newhandle) { + close(handle); + } + cli_close(targetcli, fnum); + return 1; + } + + status = cli_close(targetcli, fnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error %s closing remote file\n", nt_errstr(status)); + rc = 1; + } + + if (newhandle) { + close(handle); + } + + if (archive_level >= 2 && (attr & FILE_ATTRIBUTE_ARCHIVE)) { + cli_setatr(cli, rname, attr & ~(uint32_t)FILE_ATTRIBUTE_ARCHIVE, 0); + } + + { + struct timespec tp_end; + int this_time; + + clock_gettime_mono(&tp_end); + this_time = nsec_time_diff(&tp_end,&tp_start)/1000000; + get_total_time_ms += this_time; + get_total_size += nread; + + DEBUG(1,("(%3.1f KiloBytes/sec) (average %3.1f KiloBytes/sec)\n", + nread / (1.024*this_time + 1.0e-4), + get_total_size / (1.024*get_total_time_ms))); + } + + TALLOC_FREE(targetname); + return rc; +} + +/**************************************************************************** + Get a file. +****************************************************************************/ + +static int cmd_get(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *lname = NULL; + char *rname = NULL; + char *fname = NULL; + + rname = talloc_strdup(ctx, client_get_cur_dir()); + if (!rname) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&fname,NULL)) { + d_printf("get <filename> [localname]\n"); + return 1; + } + rname = talloc_asprintf_append(rname, "%s", fname); + if (!rname) { + return 1; + } + rname = client_clean_name(ctx, rname); + if (!rname) { + return 1; + } + + next_token_talloc(ctx, &cmd_ptr,&lname,NULL); + if (!lname) { + lname = fname; + } + + return do_get(rname, lname, false); +} + +/**************************************************************************** + Do an mget operation on one file. +****************************************************************************/ + +static NTSTATUS do_mget(struct cli_state *cli_state, struct file_info *finfo, + const char *dir) +{ + TALLOC_CTX *ctx = talloc_tos(); + const char *client_cwd = NULL; + size_t client_cwd_len; + char *path = NULL; + char *local_path = NULL; + + if (!finfo->name) { + return NT_STATUS_OK; + } + + if (ISDOT(finfo->name) || ISDOTDOT(finfo->name)) { + return NT_STATUS_OK; + } + + if ((finfo->attr & FILE_ATTRIBUTE_DIRECTORY) && !recurse) { + return NT_STATUS_OK; + } + + if (prompt) { + const char *object = (finfo->attr & FILE_ATTRIBUTE_DIRECTORY) ? + "directory" : "file"; + char *quest = NULL; + bool ok; + + quest = talloc_asprintf( + ctx, "Get %s %s? ", object, finfo->name); + if (quest == NULL) { + return NT_STATUS_NO_MEMORY; + } + + ok = yesno(quest); + TALLOC_FREE(quest); + if (!ok) { + return NT_STATUS_OK; + } + } + + path = talloc_asprintf( + ctx, "%s%c%s", dir, CLI_DIRSEP_CHAR, finfo->name); + if (path == NULL) { + return NT_STATUS_NO_MEMORY; + } + path = client_clean_name(ctx, path); + if (path == NULL) { + return NT_STATUS_NO_MEMORY; + } + + /* + * Skip the path prefix if we've done a remote "cd" when + * creating the local path + */ + client_cwd = client_get_cur_dir(); + client_cwd_len = strlen(client_cwd); + + local_path = talloc_strdup(ctx, path + client_cwd_len); + if (local_path == NULL) { + TALLOC_FREE(path); + return NT_STATUS_NO_MEMORY; + } + string_replace(local_path, CLI_DIRSEP_CHAR, '/'); + + if (finfo->attr & FILE_ATTRIBUTE_DIRECTORY) { + int ret = mkdir(local_path, 0777); + + if ((ret == -1) && (errno != EEXIST)) { + return map_nt_error_from_unix(errno); + } + } else { + do_get(path, local_path, false); + } + + return NT_STATUS_OK; +} + +/**************************************************************************** + View the file using the pager. +****************************************************************************/ + +static int cmd_more(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *rname = NULL; + char *fname = NULL; + char *lname = NULL; + char *pager_cmd = NULL; + const char *pager; + int fd; + int rc = 0; + mode_t mask; + + rname = talloc_strdup(ctx, client_get_cur_dir()); + if (!rname) { + return 1; + } + + lname = talloc_asprintf(ctx, "%s/smbmore.XXXXXX",tmpdir()); + if (!lname) { + return 1; + } + mask = umask(S_IRWXO | S_IRWXG); + fd = mkstemp(lname); + umask(mask); + if (fd == -1) { + d_printf("failed to create temporary file for more\n"); + return 1; + } + close(fd); + + if (!next_token_talloc(ctx, &cmd_ptr,&fname,NULL)) { + d_printf("more <filename>\n"); + unlink(lname); + return 1; + } + rname = talloc_asprintf_append(rname, "%s", fname); + if (!rname) { + return 1; + } + rname = client_clean_name(ctx,rname); + if (!rname) { + return 1; + } + + rc = do_get(rname, lname, false); + + pager=getenv("PAGER"); + + pager_cmd = talloc_asprintf(ctx, + "%s %s", + (pager? pager:PAGER), + lname); + if (!pager_cmd) { + return 1; + } + if (system(pager_cmd) == -1) { + d_printf("system command '%s' returned -1\n", + pager_cmd); + } + unlink(lname); + + return rc; +} + +/**************************************************************************** + Do a mget command. +****************************************************************************/ + +static int cmd_mget(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + uint32_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; + char *mget_mask = NULL; + char *buf = NULL; + NTSTATUS status = NT_STATUS_OK; + + if (recurse) { + attribute |= FILE_ATTRIBUTE_DIRECTORY; + } + + while (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + + mget_mask = talloc_strdup(ctx, client_get_cur_dir()); + if (!mget_mask) { + return 1; + } + if (*buf == CLI_DIRSEP_CHAR) { + mget_mask = talloc_strdup(ctx, buf); + } else { + mget_mask = talloc_asprintf_append(mget_mask, + "%s", buf); + } + if (!mget_mask) { + return 1; + } + mget_mask = client_clean_name(ctx, mget_mask); + if (mget_mask == NULL) { + return 1; + } + status = do_list(mget_mask, attribute, do_mget, recurse, true); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + } + + if (mget_mask == NULL) { + d_printf("nothing to mget\n"); + return 0; + } + + if (!*mget_mask) { + mget_mask = talloc_asprintf(ctx, + "%s*", + client_get_cur_dir()); + if (!mget_mask) { + return 1; + } + mget_mask = client_clean_name(ctx, mget_mask); + if (mget_mask == NULL) { + return 1; + } + status = do_list(mget_mask, attribute, do_mget, recurse, true); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + } + + return 0; +} + +/**************************************************************************** + Make a directory of name "name". +****************************************************************************/ + +static bool do_mkdir(const char *name) +{ + TALLOC_CTX *ctx = talloc_tos(); + struct cli_state *targetcli; + char *targetname = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + status = cli_resolve_path(ctx, "", + creds, + cli, name, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("mkdir %s: %s\n", name, nt_errstr(status)); + return false; + } + + status = cli_mkdir(targetcli, targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s making remote directory %s\n", + nt_errstr(status),name); + return false; + } + + return true; +} + +/**************************************************************************** + Show 8.3 name of a file. +****************************************************************************/ + +static bool do_altname(const char *name) +{ + fstring altname; + NTSTATUS status; + + status = cli_qpathinfo_alt_name(cli, name, altname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s getting alt name for %s\n", + nt_errstr(status),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) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + mask = talloc_strdup(ctx, client_get_cur_dir()); + if (!mask) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + if (!recurse) { + d_printf("mkdir <dirname>\n"); + } + return 1; + } + mask = talloc_asprintf_append(mask, "%s", buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + if (recurse) { + char *ddir = NULL; + char *ddir2 = NULL; + struct cli_state *targetcli; + char *targetname = NULL; + char *p = NULL; + char *saveptr; + + ddir2 = talloc_strdup(ctx, ""); + if (!ddir2) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, + &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + + ddir = talloc_strdup(ctx, targetname); + if (!ddir) { + return 1; + } + trim_char(ddir,'.','\0'); + p = strtok_r(ddir, "/\\", &saveptr); + while (p) { + ddir2 = talloc_asprintf_append(ddir2, "%s", p); + if (!ddir2) { + return 1; + } + if (!NT_STATUS_IS_OK(cli_chkpath(targetcli, ddir2))) { + do_mkdir(ddir2); + } + ddir2 = talloc_asprintf_append(ddir2, "%s", CLI_DIRSEP_STR); + if (!ddir2) { + return 1; + } + p = strtok_r(NULL, "/\\", &saveptr); + } + } else { + do_mkdir(mask); + } + + return 0; +} + +/**************************************************************************** + Show alt name. +****************************************************************************/ + +static int cmd_altname(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *name; + char *buf; + + name = talloc_strdup(ctx, client_get_cur_dir()); + if (!name) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) { + d_printf("altname <file>\n"); + return 1; + } + name = talloc_asprintf_append(name, "%s", buf); + if (!name) { + return 1; + } + name = client_clean_name(ctx, name); + if (name == NULL) { + return 1; + } + do_altname(name); + return 0; +} + +static char *attr_str(TALLOC_CTX *mem_ctx, uint32_t attr) +{ + char *attrs = talloc_zero_array(mem_ctx, char, 17); + int i = 0; + + if (!(attr & FILE_ATTRIBUTE_NORMAL)) { + if (attr & FILE_ATTRIBUTE_ENCRYPTED) { + attrs[i++] = 'E'; + } + if (attr & FILE_ATTRIBUTE_NONINDEXED) { + attrs[i++] = 'N'; + } + if (attr & FILE_ATTRIBUTE_OFFLINE) { + attrs[i++] = 'O'; + } + if (attr & FILE_ATTRIBUTE_COMPRESSED) { + attrs[i++] = 'C'; + } + if (attr & FILE_ATTRIBUTE_REPARSE_POINT) { + attrs[i++] = 'r'; + } + if (attr & FILE_ATTRIBUTE_SPARSE) { + attrs[i++] = 's'; + } + if (attr & FILE_ATTRIBUTE_TEMPORARY) { + attrs[i++] = 'T'; + } + if (attr & FILE_ATTRIBUTE_NORMAL) { + attrs[i++] = 'N'; + } + if (attr & FILE_ATTRIBUTE_READONLY) { + attrs[i++] = 'R'; + } + if (attr & FILE_ATTRIBUTE_HIDDEN) { + attrs[i++] = 'H'; + } + if (attr & FILE_ATTRIBUTE_SYSTEM) { + attrs[i++] = 'S'; + } + if (attr & FILE_ATTRIBUTE_DIRECTORY) { + attrs[i++] = 'D'; + } + if (attr & FILE_ATTRIBUTE_ARCHIVE) { + attrs[i++] = 'A'; + } + } + return attrs; +} + +/**************************************************************************** + Show all info we can get +****************************************************************************/ + +static int do_allinfo(const char *name) +{ + fstring altname; + struct timespec b_time, a_time, m_time, c_time; + off_t size; + uint32_t attr; + NTTIME tmp; + uint16_t fnum; + unsigned int num_streams; + struct stream_struct *streams; + int j, num_snapshots; + char **snapshots = NULL; + unsigned int i; + NTSTATUS status; + + status = cli_qpathinfo_alt_name(cli, name, altname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s getting alt name for %s\n", nt_errstr(status), + name); + /* + * Ignore not supported or not implemented, it does not + * hurt if we can't list alternate names. + */ + if (NT_STATUS_EQUAL(status, NT_STATUS_NOT_SUPPORTED) || + NT_STATUS_EQUAL(status, NT_STATUS_NOT_IMPLEMENTED)) { + altname[0] = '\0'; + } else { + return false; + } + } + d_printf("altname: %s\n", altname); + + status = cli_qpathinfo3(cli, name, &b_time, &a_time, &m_time, &c_time, + &size, &attr, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s getting pathinfo for %s\n", nt_errstr(status), + name); + return false; + } + + tmp = full_timespec_to_nt_time(&b_time); + d_printf("create_time: %s\n", nt_time_string(talloc_tos(), tmp)); + + tmp = full_timespec_to_nt_time(&a_time); + d_printf("access_time: %s\n", nt_time_string(talloc_tos(), tmp)); + + tmp = full_timespec_to_nt_time(&m_time); + d_printf("write_time: %s\n", nt_time_string(talloc_tos(), tmp)); + + tmp = full_timespec_to_nt_time(&c_time); + d_printf("change_time: %s\n", nt_time_string(talloc_tos(), tmp)); + + d_printf("attributes: %s (%x)\n", attr_str(talloc_tos(), attr), attr); + + status = cli_qpathinfo_streams(cli, name, talloc_tos(), &num_streams, + &streams); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s getting streams for %s\n", nt_errstr(status), + name); + return false; + } + + for (i=0; i<num_streams; i++) { + d_printf("stream: [%s], %lld bytes\n", streams[i].name, + (unsigned long long)streams[i].size); + } + + if (attr & FILE_ATTRIBUTE_REPARSE_POINT) { + char *subst, *print; + uint32_t flags; + + status = cli_readlink(cli, name, talloc_tos(), &subst, &print, + &flags); + if (!NT_STATUS_IS_OK(status)) { + d_fprintf(stderr, "cli_readlink returned %s\n", + nt_errstr(status)); + } else { + d_printf("symlink: subst=[%s], print=[%s], flags=%x\n", + subst, print, flags); + TALLOC_FREE(subst); + TALLOC_FREE(print); + } + } + + status = cli_ntcreate(cli, name, 0, + SEC_FILE_READ_DATA | SEC_FILE_READ_ATTRIBUTE | + SEC_STD_SYNCHRONIZE, 0, + FILE_SHARE_READ|FILE_SHARE_WRITE + |FILE_SHARE_DELETE, + FILE_OPEN, 0x0, 0x0, &fnum, NULL); + if (!NT_STATUS_IS_OK(status)) { + /* + * Ignore failure, it does not hurt if we can't list + * snapshots + */ + return 0; + } + /* + * In order to get shadow copy data over SMB1 we + * must call twice, once with 'get_names = false' + * to get the size, then again with 'get_names = true' + * to get the data or a Windows server fails to return + * valid info. Samba doesn't have this bug. JRA. + */ + + status = cli_shadow_copy_data(talloc_tos(), cli, fnum, + false, &snapshots, &num_snapshots); + if (!NT_STATUS_IS_OK(status)) { + cli_close(cli, fnum); + return 0; + } + status = cli_shadow_copy_data(talloc_tos(), cli, fnum, + true, &snapshots, &num_snapshots); + if (!NT_STATUS_IS_OK(status)) { + cli_close(cli, fnum); + return 0; + } + + for (j=0; j<num_snapshots; j++) { + char *snap_name; + + d_printf("%s\n", snapshots[j]); + snap_name = talloc_asprintf(talloc_tos(), "%s%s", + snapshots[j], name); + status = cli_qpathinfo3(cli, snap_name, &b_time, &a_time, + &m_time, &c_time, &size, + NULL, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_fprintf(stderr, "pathinfo(%s) failed: %s\n", + snap_name, nt_errstr(status)); + TALLOC_FREE(snap_name); + continue; + } + tmp = unix_timespec_to_nt_time(b_time); + d_printf("create_time: %s\n", nt_time_string(talloc_tos(), tmp)); + tmp = unix_timespec_to_nt_time(a_time); + d_printf("access_time: %s\n", nt_time_string(talloc_tos(), tmp)); + tmp =unix_timespec_to_nt_time(m_time); + d_printf("write_time: %s\n", nt_time_string(talloc_tos(), tmp)); + tmp = unix_timespec_to_nt_time(c_time); + d_printf("change_time: %s\n", nt_time_string(talloc_tos(), tmp)); + d_printf("size: %d\n", (int)size); + } + + TALLOC_FREE(snapshots); + cli_close(cli, fnum); + + return 0; +} + +/**************************************************************************** + Show all info we can get +****************************************************************************/ + +static int cmd_allinfo(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *name; + char *buf; + + name = talloc_strdup(ctx, client_get_cur_dir()); + if (!name) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) { + d_printf("allinfo <file>\n"); + return 1; + } + name = talloc_asprintf_append(name, "%s", buf); + if (!name) { + return 1; + } + name = client_clean_name(ctx, name); + if (name == NULL) { + return 1; + } + do_allinfo(name); + + return 0; +} + +/**************************************************************************** + Put a single file. +****************************************************************************/ + +static int do_put(const char *rname, const char *lname, bool reput) +{ + TALLOC_CTX *ctx = talloc_tos(); + uint16_t fnum; + FILE *f; + off_t start = 0; + int rc = 0; + struct timespec tp_start; + struct cli_state *targetcli; + char *targetname = NULL; + struct push_state state; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + status = cli_resolve_path(ctx, "", + creds, + cli, rname, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to open %s: %s\n", rname, nt_errstr(status)); + return 1; + } + + clock_gettime_mono(&tp_start); + + if (reput) { + status = cli_open(targetcli, targetname, O_RDWR|O_CREAT, DENY_NONE, &fnum); + if (NT_STATUS_IS_OK(status)) { + if (!NT_STATUS_IS_OK(status = cli_qfileinfo_basic( + targetcli, fnum, NULL, + &start, NULL, NULL, + NULL, NULL, NULL))) { + d_printf("getattrib: %s\n", nt_errstr(status)); + return 1; + } + } + } else { + status = cli_open(targetcli, targetname, O_RDWR|O_CREAT|O_TRUNC, DENY_NONE, &fnum); + } + + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s opening remote file %s\n", nt_errstr(status), + 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 = stdin; + /* size of file is not known */ + } else { + f = fopen(lname, "r"); + if (f && reput) { + if (fseek(f, start, SEEK_SET) == -1) { + d_printf("Error seeking local file\n"); + fclose(f); + return 1; + } + } + } + + if (!f) { + d_printf("Error opening local file %s\n",lname); + return 1; + } + + DEBUG(1,("putting file %s as %s ",lname, + rname)); + + setvbuf(f, NULL, _IOFBF, io_bufsize); + + state.f = f; + state.nread = 0; + + status = cli_push(targetcli, fnum, 0, 0, io_bufsize, push_source, + &state); + if (!NT_STATUS_IS_OK(status)) { + d_fprintf(stderr, "cli_push returned %s\n", nt_errstr(status)); + rc = 1; + } + + status = cli_close(targetcli, fnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s closing remote file %s\n", nt_errstr(status), + rname); + if (f != stdin) { + fclose(f); + } + return 1; + } + + if (f != stdin) { + fclose(f); + } + + { + struct timespec tp_end; + int this_time; + + clock_gettime_mono(&tp_end); + this_time = nsec_time_diff(&tp_end,&tp_start)/1000000; + put_total_time_ms += this_time; + put_total_size += state.nread; + + DEBUG(1,("(%3.1f kb/s) (average %3.1f kb/s)\n", + state.nread / (1.024*this_time + 1.0e-4), + put_total_size / (1.024*put_total_time_ms))); + } + + if (f == stdin) { + cli_shutdown(cli); + exit(rc); + } + + return rc; +} + +/**************************************************************************** + Put a file. +****************************************************************************/ + +static int cmd_put(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *lname; + char *rname; + char *buf; + + rname = talloc_strdup(ctx, client_get_cur_dir()); + if (!rname) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&lname,NULL)) { + d_printf("put <filename>\n"); + return 1; + } + + if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + rname = talloc_asprintf_append(rname, "%s", buf); + } else { + rname = talloc_asprintf_append(rname, "%s", lname); + } + if (!rname) { + return 1; + } + + rname = client_clean_name(ctx, rname); + if (!rname) { + return 1; + } + + { + SMB_STRUCT_STAT st; + /* allow '-' to represent stdin + jdblair, 24.jun.98 */ + if (!file_exist_stat(lname, &st, false) && + (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 *l_head) +{ + struct file_list *list, *next; + + for (list = l_head; list; list = next) { + next = list->next; + DLIST_REMOVE(l_head, list); + TALLOC_FREE(list); + } +} + +/**************************************************************************** + 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) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *new_fs = NULL; + next_token_talloc(ctx, &cmd_ptr,&new_fs,NULL) + ; + if (new_fs) { + client_set_fileselection(new_fs); + } else { + client_set_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(TALLOC_CTX *ctx, + 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 (ISDOT(dname) || ISDOTDOT(dname)) { + continue; + } + + path = talloc_asprintf(ctx, "%s/%s", directory, dname); + if (path == NULL) { + 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(ctx, + list, + path, + expression, + false); + } + } else { + d_printf("file_find: cannot stat file %s\n", path); + } + + if (ret == -1) { + TALLOC_FREE(path); + closedir(dir); + return -1; + } + } + entry = talloc_zero(ctx, struct file_list); + if (!entry) { + d_printf("Out of memory in file_find\n"); + closedir(dir); + return -1; + } + entry->file_path = talloc_move(entry, &path); + entry->isdir = isdir; + DLIST_ADD(*list, entry); + } else { + TALLOC_FREE(path); + } + } + + closedir(dir); + return 0; +} + +/**************************************************************************** + mput some files. +****************************************************************************/ + +static int cmd_mput(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *p = NULL; + + while (next_token_talloc(ctx, &cmd_ptr,&p,NULL)) { + int ret; + struct file_list *temp_list; + char *quest, *lname, *rname; + + file_list = NULL; + + ret = file_find(ctx, &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", client_get_cur_dir(), lname) < 0) { + break; + } + normalize_name(rname); + { + char *tmp_rname = + client_clean_name(ctx, rname); + if (tmp_rname == NULL) { + break; + } + SAFE_FREE(rname); + rname = smb_xstrdup(tmp_rname); + TALLOC_FREE(tmp_rname); + if (rname == NULL) { + break; + } + } + if (!NT_STATUS_IS_OK(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", client_get_cur_dir(), lname) < 0) { + break; + } + } + + normalize_name(rname); + + { + char *tmp_rname = client_clean_name(ctx, rname); + if (tmp_rname == NULL) { + break; + } + SAFE_FREE(rname); + rname = smb_xstrdup(tmp_rname); + TALLOC_FREE(tmp_rname); + if (rname == NULL) { + break; + } + } + 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) +{ + if (cli_printjob_del(cli, job)) { + d_printf("Job %d cancelled\n",job); + return 0; + } else { + NTSTATUS status = cli_nt_error(cli); + d_printf("Error cancelling job %d : %s\n", + job, nt_errstr(status)); + return 1; + } +} + +/**************************************************************************** + Cancel a print job. +****************************************************************************/ + +static int cmd_cancel(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf = NULL; + int job; + + if (!next_token_talloc(ctx, &cmd_ptr, &buf,NULL)) { + d_printf("cancel <jobid> ...\n"); + return 1; + } + do { + job = atoi(buf); + do_cancel(job); + } while (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)); + + return 0; +} + +/**************************************************************************** + Print a file. +****************************************************************************/ + +static int cmd_print(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *lname = NULL; + char *rname = NULL; + char *p = NULL; + + if (!next_token_talloc(ctx, &cmd_ptr, &lname,NULL)) { + d_printf("print <filename>\n"); + return 1; + } + + rname = talloc_strdup(ctx, lname); + if (!rname) { + return 1; + } + p = strrchr_m(rname,'/'); + if (p) { + rname = talloc_asprintf(ctx, + "%s-%d", + p+1, + (int)getpid()); + } + if (strequal(lname,"-")) { + rname = talloc_asprintf(ctx, + "stdin-%d", + (int)getpid()); + } + if (!rname) { + return 1; + } + + return do_put(rname, lname, false); +} + +/**************************************************************************** + Show a print queue entry. +****************************************************************************/ + +static void queue_fn(struct print_job_info *p) +{ + d_printf("%-6d %-9d %s\n", (int)p->id, (int)p->size, p->name); +} + +/**************************************************************************** + Show a print queue. +****************************************************************************/ + +static int cmd_queue(void) +{ + cli_print_queue(cli, queue_fn); + return 0; +} + +/**************************************************************************** + Delete some files. +****************************************************************************/ + +static NTSTATUS do_del(struct cli_state *cli_state, struct file_info *finfo, + const char *dir) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + struct cli_state *targetcli = NULL; + char *targetname = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + mask = talloc_asprintf(ctx, + "%s%c%s", + dir, + CLI_DIRSEP_CHAR, + finfo->name); + if (!mask) { + return NT_STATUS_NO_MEMORY; + } + + if (finfo->attr & FILE_ATTRIBUTE_DIRECTORY) { + TALLOC_FREE(mask); + return NT_STATUS_OK; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + goto out; + } + + status = cli_unlink(targetcli, targetname, FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN); +out: + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s deleting remote file %s\n", + nt_errstr(status), mask); + } + TALLOC_FREE(mask); + return status; +} + +/**************************************************************************** + Delete some files. +****************************************************************************/ + +static int cmd_del(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + NTSTATUS status = NT_STATUS_OK; + uint32_t attribute = FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN; + + if (recurse) { + attribute |= FILE_ATTRIBUTE_DIRECTORY; + } + + mask = talloc_strdup(ctx, client_get_cur_dir()); + if (!mask) { + return 1; + } + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("del <filename>\n"); + return 1; + } + mask = talloc_asprintf_append(mask, "%s", buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = do_list(mask,attribute,do_del,false,false); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + return 0; +} + +/**************************************************************************** + Delete some files. +****************************************************************************/ + +static NTSTATUS delete_remote_files_list(struct cli_state *cli_state, + struct file_list *flist) +{ + NTSTATUS status = NT_STATUS_OK; + struct file_list *deltree_list_iter = NULL; + char *targetname = NULL; + struct cli_state *targetcli = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + TALLOC_CTX *ctx = talloc_tos(); + + for (deltree_list_iter = flist; + deltree_list_iter != NULL; + deltree_list_iter = deltree_list_iter->next) { + status = cli_resolve_path(ctx, + "", + creds, + cli_state, + deltree_list_iter->file_path, + &targetcli, + &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("delete_remote_files %s: %s\n", + deltree_list_iter->file_path, + nt_errstr(status)); + return status; + } + if (CLI_DIRSEP_CHAR == '/') { + /* POSIX. */ + status = cli_posix_unlink(targetcli, + targetname); + } else if (deltree_list_iter->isdir) { + status = cli_rmdir(targetcli, + targetname); + } else { + status = cli_unlink(targetcli, + targetname, + FILE_ATTRIBUTE_SYSTEM | + FILE_ATTRIBUTE_HIDDEN); + } + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s deleting remote %s %s\n", + nt_errstr(status), + deltree_list_iter->isdir ? + "directory" : "file", + deltree_list_iter->file_path); + return status; + } + } + return NT_STATUS_OK; +} + +/**************************************************************************** + Save a list of files to delete. +****************************************************************************/ + +static struct file_list *deltree_list_head; + +static NTSTATUS do_deltree_list(struct cli_state *cli_state, + struct file_info *finfo, + const char *dir) +{ + struct file_list **file_list_head_pp = &deltree_list_head; + struct file_list *dt = NULL; + + if (!do_this_one(finfo)) { + return NT_STATUS_OK; + } + + /* skip if this is . or .. */ + if (ISDOT(finfo->name) || ISDOTDOT(finfo->name)) { + return NT_STATUS_OK; + } + + dt = talloc_zero(NULL, struct file_list); + if (dt == NULL) { + return NT_STATUS_NO_MEMORY; + } + + /* create absolute filename for cli_ntcreate() */ + dt->file_path = talloc_asprintf(dt, + "%s%s%s", + dir, + CLI_DIRSEP_STR, + finfo->name); + if (dt->file_path == NULL) { + TALLOC_FREE(dt); + return NT_STATUS_NO_MEMORY; + } + + if (finfo->attr & FILE_ATTRIBUTE_DIRECTORY) { + dt->isdir = true; + } + + DLIST_ADD(*file_list_head_pp, dt); + return NT_STATUS_OK; +} + +static int cmd_deltree(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf = NULL; + NTSTATUS status = NT_STATUS_OK; + struct file_list *deltree_list_norecurse = NULL; + struct file_list *deltree_list_iter = NULL; + uint32_t attribute = FILE_ATTRIBUTE_SYSTEM | + FILE_ATTRIBUTE_HIDDEN | + FILE_ATTRIBUTE_DIRECTORY; + bool ok; + char *mask = talloc_strdup(ctx, client_get_cur_dir()); + if (mask == NULL) { + return 1; + } + ok = next_token_talloc(ctx, &cmd_ptr, &buf, NULL); + if (!ok) { + d_printf("deltree <filename>\n"); + return 1; + } + mask = talloc_asprintf_append(mask, "%s", buf); + if (mask == NULL) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + deltree_list_head = NULL; + + /* + * Get the list of directories to + * delete (in case mask has a wildcard). + */ + status = do_list(mask, attribute, do_deltree_list, false, true); + if (!NT_STATUS_IS_OK(status)) { + goto err; + } + deltree_list_norecurse = deltree_list_head; + deltree_list_head = NULL; + + for (deltree_list_iter = deltree_list_norecurse; + deltree_list_iter != NULL; + deltree_list_iter = deltree_list_iter->next) { + + if (deltree_list_iter->isdir == false) { + char *targetname = NULL; + struct cli_state *targetcli = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + status = cli_resolve_path(ctx, + "", + creds, + cli, + deltree_list_iter->file_path, + &targetcli, + &targetname); + if (!NT_STATUS_IS_OK(status)) { + goto err; + } + /* Just a regular file. */ + if (CLI_DIRSEP_CHAR == '/') { + /* POSIX. */ + status = cli_posix_unlink(targetcli, + targetname); + } else { + status = cli_unlink(targetcli, + targetname, + FILE_ATTRIBUTE_SYSTEM | + FILE_ATTRIBUTE_HIDDEN); + } + if (!NT_STATUS_IS_OK(status)) { + goto err; + } + continue; + } + + /* + * Get the list of files or directories to + * delete in depth order. + */ + status = do_list(deltree_list_iter->file_path, + attribute, + do_deltree_list, + true, + true); + if (!NT_STATUS_IS_OK(status)) { + goto err; + } + status = delete_remote_files_list(cli, deltree_list_head); + free_file_list(deltree_list_head); + deltree_list_head = NULL; + if (!NT_STATUS_IS_OK(status)) { + goto err; + } + } + + free_file_list(deltree_list_norecurse); + free_file_list(deltree_list_head); + return 0; + + err: + + free_file_list(deltree_list_norecurse); + free_file_list(deltree_list_head); + deltree_list_head = NULL; + return 1; +} + + +/**************************************************************************** + Wildcard delete some files. +****************************************************************************/ + +static int cmd_wdel(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + uint32_t attribute; + struct cli_state *targetcli; + char *targetname = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("wdel 0x<attrib> <wcard>\n"); + return 1; + } + + attribute = (uint32_t)strtol(buf, (char **)NULL, 16); + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("wdel 0x<attrib> <wcard>\n"); + return 1; + } + + mask = talloc_asprintf(ctx, "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cmd_wdel %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_unlink(targetcli, targetname, attribute); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s deleting remote files %s\n", nt_errstr(status), + targetname); + } + return 0; +} + +/**************************************************************************** +****************************************************************************/ + +static int cmd_open(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + uint16_t fnum = (uint16_t)-1; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("open <filename>\n"); + return 1; + } + mask = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("open %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_ntcreate(targetcli, targetname, 0, + FILE_READ_DATA|FILE_WRITE_DATA, 0, + FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, + 0x0, 0x0, &fnum, NULL); + if (!NT_STATUS_IS_OK(status)) { + status = cli_ntcreate(targetcli, targetname, 0, + FILE_READ_DATA, 0, + FILE_SHARE_READ|FILE_SHARE_WRITE, FILE_OPEN, + 0x0, 0x0, &fnum, NULL); + if (NT_STATUS_IS_OK(status)) { + d_printf("open file %s: for read/write fnum %d\n", targetname, fnum); + } else { + d_printf("Failed to open file %s. %s\n", + targetname, nt_errstr(status)); + } + } else { + d_printf("open file %s: for read/write fnum %d\n", targetname, fnum); + } + return 0; +} + +static int cmd_posix_encrypt(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + NTSTATUS status = NT_STATUS_UNSUCCESSFUL; + char *domain = NULL; + char *user = NULL; + char *password = NULL; + struct cli_credentials *creds = NULL; + struct cli_credentials *lcreds = NULL; + + if (next_token_talloc(ctx, &cmd_ptr, &domain, NULL)) { + + if (!next_token_talloc(ctx, &cmd_ptr, &user, NULL)) { + d_printf("posix_encrypt domain user password\n"); + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr, &password, NULL)) { + d_printf("posix_encrypt domain user password\n"); + return 1; + } + + lcreds = cli_session_creds_init(ctx, + user, + domain, + NULL, /* realm */ + password, + false, /* use_kerberos */ + false, /* fallback_after_kerberos */ + false, /* use_ccache */ + false); /* password_is_nt_hash */ + if (lcreds == NULL) { + d_printf("cli_session_creds_init() failed.\n"); + return -1; + } + creds = lcreds; + } else { + bool auth_requested = false; + + creds = samba_cmdline_get_creds(); + + auth_requested = cli_credentials_authentication_requested(creds); + if (!auth_requested) { + d_printf("posix_encrypt domain user password\n"); + return 1; + } + } + + status = cli_smb1_setup_encryption(cli, creds); + /* gensec currently references the creds so we can't free them here */ + talloc_unlink(ctx, lcreds); + if (!NT_STATUS_IS_OK(status)) { + d_printf("posix_encrypt failed with error %s\n", nt_errstr(status)); + } else { + bool ok; + + d_printf("encryption on\n"); + ok = cli_credentials_set_smb_encryption(creds, + SMB_ENCRYPTION_REQUIRED, + CRED_SPECIFIED); + SMB_ASSERT(ok); + } + + return 0; +} + +/**************************************************************************** +****************************************************************************/ + +static int cmd_posix_open(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + mode_t mode; + uint16_t fnum; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("posix_open <filename> 0<mode>\n"); + return 1; + } + mask = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("posix_open <filename> 0<mode>\n"); + return 1; + } + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"posix_open\" command can be used.\n"); + return 1; + } + mode = (mode_t)strtol(buf, (char **)NULL, 8); + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("posix_open %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_posix_open(targetcli, targetname, O_CREAT|O_RDWR, mode, + &fnum); + if (!NT_STATUS_IS_OK(status)) { + status = cli_posix_open(targetcli, targetname, + O_CREAT|O_RDONLY, mode, &fnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to open file %s. %s\n", targetname, + nt_errstr(status)); + } else { + d_printf("posix_open file %s: for readonly fnum %d\n", + targetname, fnum); + } + } else { + d_printf("posix_open file %s: for read/write fnum %d\n", + targetname, fnum); + } + + return 0; +} + +static int cmd_posix_mkdir(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + mode_t mode; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("posix_mkdir <filename> 0<mode>\n"); + return 1; + } + mask = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("posix_mkdir <filename> 0<mode>\n"); + return 1; + } + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"posix_mkdir\" command can be used.\n"); + return 1; + } + mode = (mode_t)strtol(buf, (char **)NULL, 8); + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("posix_mkdir %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_posix_mkdir(targetcli, targetname, mode); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to open file %s. %s\n", + targetname, nt_errstr(status)); + } else { + d_printf("posix_mkdir created directory %s\n", targetname); + } + return 0; +} + +static int cmd_posix_unlink(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("posix_unlink <filename>\n"); + return 1; + } + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"posix_unlink\" command can be used.\n"); + return 1; + } + mask = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("posix_unlink %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_posix_unlink(targetcli, targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to unlink file %s. %s\n", + targetname, nt_errstr(status)); + } else { + d_printf("posix_unlink deleted file %s\n", targetname); + } + + return 0; +} + +static int cmd_posix_rmdir(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("posix_rmdir <filename>\n"); + return 1; + } + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"posix_rmdir\" command can be used.\n"); + return 1; + } + mask = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("posix_rmdir %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_posix_rmdir(targetcli, targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to unlink directory %s. %s\n", + targetname, nt_errstr(status)); + } else { + d_printf("posix_rmdir deleted directory %s\n", targetname); + } + + return 0; +} + +static int cmd_close(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf = NULL; + int fnum; + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("close <fnum>\n"); + return 1; + } + + fnum = atoi(buf); + /* We really should use the targetcli here.... */ + status = cli_close(cli, fnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("close %d: %s\n", fnum, nt_errstr(status)); + return 1; + } + return 0; +} + +static int cmd_posix(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + uint16_t major, minor; + uint32_t caplow, caphigh; + char *caps; + NTSTATUS status; + + if (!SERVER_HAS_UNIX_CIFS(cli)) { + d_printf("Server doesn't support UNIX CIFS extensions.\n"); + return 1; + } + + status = cli_unix_extensions_version(cli, &major, &minor, &caplow, + &caphigh); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Can't get UNIX CIFS extensions version from " + "server: %s\n", nt_errstr(status)); + return 1; + } + + d_printf("Server supports CIFS extensions %u.%u\n", (unsigned int)major, (unsigned int)minor); + + caps = talloc_strdup(ctx, ""); + if (!caps) { + return 1; + } + if (caplow & CIFS_UNIX_FCNTL_LOCKS_CAP) { + caps = talloc_asprintf_append(caps, "locks "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_POSIX_ACLS_CAP) { + caps = talloc_asprintf_append(caps, "acls "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_XATTTR_CAP) { + caps = talloc_asprintf_append(caps, "eas "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_POSIX_PATHNAMES_CAP) { + caps = talloc_asprintf_append(caps, "pathnames "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_POSIX_PATH_OPERATIONS_CAP) { + caps = talloc_asprintf_append(caps, "posix_path_operations "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_LARGE_READ_CAP) { + caps = talloc_asprintf_append(caps, "large_read "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_LARGE_WRITE_CAP) { + caps = talloc_asprintf_append(caps, "large_write "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_TRANSPORT_ENCRYPTION_CAP) { + caps = talloc_asprintf_append(caps, "posix_encrypt "); + if (!caps) { + return 1; + } + } + if (caplow & CIFS_UNIX_TRANSPORT_ENCRYPTION_MANDATORY_CAP) { + caps = talloc_asprintf_append(caps, "mandatory_posix_encrypt "); + if (!caps) { + return 1; + } + } + + if (*caps && caps[strlen(caps)-1] == ' ') { + caps[strlen(caps)-1] = '\0'; + } + + d_printf("Server supports CIFS capabilities %s\n", caps); + + status = cli_set_unix_extensions_capabilities(cli, major, minor, + caplow, caphigh); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Can't set UNIX CIFS extensions capabilities. %s.\n", + nt_errstr(status)); + return 1; + } + + if (caplow & CIFS_UNIX_POSIX_PATHNAMES_CAP) { + CLI_DIRSEP_CHAR = '/'; + *CLI_DIRSEP_STR = '/'; + client_set_cur_dir(CLI_DIRSEP_STR); + } + + return 0; +} + +static int cmd_lock(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf = NULL; + uint64_t start, len; + enum brl_type lock_type; + int fnum; + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n"); + return 1; + } + fnum = atoi(buf); + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n"); + return 1; + } + + if (*buf == 'r' || *buf == 'R') { + lock_type = READ_LOCK; + } else if (*buf == 'w' || *buf == 'W') { + lock_type = WRITE_LOCK; + } else { + d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n"); + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n"); + return 1; + } + + start = (uint64_t)strtol(buf, (char **)NULL, 16); + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("lock <fnum> [r|w] <hex-start> <hex-len>\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"lock\" command can be used.\n"); + return 1; + } + + len = (uint64_t)strtol(buf, (char **)NULL, 16); + + status = cli_posix_lock(cli, fnum, start, len, true, lock_type); + if (!NT_STATUS_IS_OK(status)) { + d_printf("lock failed %d: %s\n", fnum, nt_errstr(status)); + } + + return 0; +} + +static int cmd_unlock(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf = NULL; + uint64_t start, len; + int fnum; + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("unlock <fnum> <hex-start> <hex-len>\n"); + return 1; + } + fnum = atoi(buf); + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("unlock <fnum> <hex-start> <hex-len>\n"); + return 1; + } + + start = (uint64_t)strtol(buf, (char **)NULL, 16); + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("unlock <fnum> <hex-start> <hex-len>\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"unlock\" command can be used.\n"); + return 1; + } + + len = (uint64_t)strtol(buf, (char **)NULL, 16); + + status = cli_posix_unlock(cli, fnum, start, len); + if (!NT_STATUS_IS_OK(status)) { + d_printf("unlock failed %d: %s\n", fnum, nt_errstr(status)); + } + + return 0; +} + +static int cmd_posix_whoami(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + NTSTATUS status = NT_STATUS_UNSUCCESSFUL; + uint64_t uid = 0; + uint64_t gid = 0; + uint32_t num_gids = 0; + uint32_t num_sids = 0; + uint64_t *gids = NULL; + struct dom_sid *sids = NULL; + bool guest = false; + uint32_t i; + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"posix_whoami\" command can be used.\n"); + return 1; + } + + status = cli_posix_whoami(cli, + ctx, + &uid, + &gid, + &num_gids, + &gids, + &num_sids, + &sids, + &guest); + + if (!NT_STATUS_IS_OK(status)) { + d_printf("posix_whoami failed with error %s\n", nt_errstr(status)); + return 1; + } + + d_printf("GUEST:%s\n", guest ? "True" : "False"); + d_printf("UID:%" PRIu64 "\n", uid); + d_printf("GID:%" PRIu64 "\n", gid); + d_printf("NUM_GIDS:%" PRIu32 "\n", num_gids); + for (i = 0; i < num_gids; i++) { + d_printf("GIDS[%" PRIu32 "]:%" PRIu64 "\n", i, gids[i]); + } + d_printf("NUM_SIDS:%" PRIu32 "\n", num_sids); + for (i = 0; i < num_sids; i++) { + struct dom_sid_buf buf; + d_printf("SIDS[%" PRIu32 "]:%s\n", + i, + dom_sid_str_buf(&sids[i], &buf)); + } + return 0; +} + + +/**************************************************************************** + Remove a directory. +****************************************************************************/ + +static int cmd_rmdir(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *mask = NULL; + char *buf = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("rmdir <dirname>\n"); + return 1; + } + mask = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!mask) { + return 1; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, mask, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("rmdir %s: %s\n", mask, nt_errstr(status)); + return 1; + } + + status = cli_rmdir(targetcli, targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s removing remote directory file %s\n", + nt_errstr(status), mask); + } + + return 0; +} + +/**************************************************************************** + UNIX hardlink. +****************************************************************************/ + +static int cmd_link(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *oldname = NULL; + char *newname = NULL; + char *buf = NULL; + char *buf2 = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) { + d_printf("link <oldname> <newname>\n"); + return 1; + } + oldname = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!oldname) { + return 1; + } + oldname = client_clean_name(ctx, oldname); + if (oldname == NULL) { + return 1; + } + newname = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf2); + if (!newname) { + return 1; + } + newname = client_clean_name(ctx, newname); + if (newname == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, oldname, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("link %s: %s\n", oldname, nt_errstr(status)); + return 1; + } + + if (!SERVER_HAS_UNIX_CIFS(targetcli)) { + d_printf("Server doesn't support UNIX CIFS calls.\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"link\" command can be used.\n"); + return 1; + } + + status = cli_posix_hardlink(targetcli, targetname, newname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s linking files (%s -> %s)\n", + nt_errstr(status), newname, oldname); + return 1; + } + return 0; +} + +/**************************************************************************** + UNIX readlink. +****************************************************************************/ + +static int cmd_readlink(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *name= NULL; + char *buf = NULL; + char *targetname = NULL; + char *linkname = NULL; + struct cli_state *targetcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("readlink <name>\n"); + return 1; + } + name = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!name) { + return 1; + } + name = client_clean_name(ctx, name); + if (name == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, name, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("readlink %s: %s\n", name, nt_errstr(status)); + return 1; + } + + if (!SERVER_HAS_UNIX_CIFS(targetcli)) { + d_printf("Server doesn't support UNIX CIFS calls.\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"readlink\" command can be used.\n"); + return 1; + } + + status = cli_posix_readlink(targetcli, name, talloc_tos(), &linkname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s readlink on file %s\n", + nt_errstr(status), name); + return 1; + } + + d_printf("%s -> %s\n", name, linkname); + + TALLOC_FREE(linkname); + + return 0; +} + + +/**************************************************************************** + UNIX symlink. +****************************************************************************/ + +static int cmd_symlink(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *link_target = NULL; + char *newname = NULL; + char *buf = NULL; + char *buf2 = NULL; + struct cli_state *newcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) { + d_printf("symlink <link_target> <newname>\n"); + return 1; + } + /* Oldname (link target) must be an untouched blob. */ + link_target = buf; + + if (SERVER_HAS_UNIX_CIFS(cli)) { + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"symlink\" command can be used.\n"); + return 1; + } + newname = talloc_asprintf(ctx, "%s%s", client_get_cur_dir(), + buf2); + if (!newname) { + return 1; + } + newname = client_clean_name(ctx, newname); + if (newname == NULL) { + return 1; + } + /* New name must be present in share namespace. */ + status = cli_resolve_path(ctx, "", + creds, + cli, newname, + &newcli, &newname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("link %s: %s\n", newname, + nt_errstr(status)); + return 1; + } + status = cli_posix_symlink(newcli, link_target, newname); + } else { + status = cli_symlink( + cli, link_target, buf2, + buf2[0] == '\\' ? 0 : SYMLINK_FLAG_RELATIVE); + } + + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s symlinking files (%s -> %s)\n", + nt_errstr(status), newname, link_target); + return 1; + } + + return 0; +} + +/**************************************************************************** + UNIX chmod. +****************************************************************************/ + +static int cmd_chmod(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src = NULL; + char *buf = NULL; + char *buf2 = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + mode_t mode; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) { + d_printf("chmod mode file\n"); + return 1; + } + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf2); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + mode = (mode_t)strtol(buf, NULL, 8); + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("chmod %s: %s\n", src, nt_errstr(status)); + return 1; + } + + if (!SERVER_HAS_UNIX_CIFS(targetcli)) { + d_printf("Server doesn't support UNIX CIFS calls.\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"chmod\" command can be used.\n"); + return 1; + } + + status = cli_posix_chmod(targetcli, targetname, mode); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s chmod file %s 0%o\n", + nt_errstr(status), src, (unsigned int)mode); + return 1; + } + + return 0; +} + +static const char *filetype_to_str(mode_t mode) +{ + if (S_ISREG(mode)) { + return "regular file"; + } else if (S_ISDIR(mode)) { + return "directory"; + } else +#ifdef S_ISCHR + if (S_ISCHR(mode)) { + return "character device"; + } else +#endif +#ifdef S_ISBLK + if (S_ISBLK(mode)) { + return "block device"; + } else +#endif +#ifdef S_ISFIFO + if (S_ISFIFO(mode)) { + return "fifo"; + } else +#endif +#ifdef S_ISLNK + if (S_ISLNK(mode)) { + return "symbolic link"; + } else +#endif +#ifdef S_ISSOCK + if (S_ISSOCK(mode)) { + return "socket"; + } else +#endif + return ""; +} + +static char rwx_to_str(mode_t m, mode_t bt, char ret) +{ + if (m & bt) { + return ret; + } else { + return '-'; + } +} + +static char *unix_mode_to_str(char *s, mode_t m) +{ + char *p = s; + const char *str = filetype_to_str(m); + + switch(str[0]) { + case 'd': + *p++ = 'd'; + break; + case 'c': + *p++ = 'c'; + break; + case 'b': + *p++ = 'b'; + break; + case 'f': + *p++ = 'p'; + break; + case 's': + *p++ = str[1] == 'y' ? 'l' : 's'; + break; + case 'r': + default: + *p++ = '-'; + break; + } + *p++ = rwx_to_str(m, S_IRUSR, 'r'); + *p++ = rwx_to_str(m, S_IWUSR, 'w'); + *p++ = rwx_to_str(m, S_IXUSR, 'x'); + *p++ = rwx_to_str(m, S_IRGRP, 'r'); + *p++ = rwx_to_str(m, S_IWGRP, 'w'); + *p++ = rwx_to_str(m, S_IXGRP, 'x'); + *p++ = rwx_to_str(m, S_IROTH, 'r'); + *p++ = rwx_to_str(m, S_IWOTH, 'w'); + *p++ = rwx_to_str(m, S_IXOTH, 'x'); + *p++ = '\0'; + return s; +} + +/**************************************************************************** + Utility function for UNIX getfacl. +****************************************************************************/ + +static char *perms_to_string(fstring permstr, unsigned char perms) +{ + fstrcpy(permstr, "---"); + if (perms & SMB_POSIX_ACL_READ) { + permstr[0] = 'r'; + } + if (perms & SMB_POSIX_ACL_WRITE) { + permstr[1] = 'w'; + } + if (perms & SMB_POSIX_ACL_EXECUTE) { + permstr[2] = 'x'; + } + return permstr; +} + +/**************************************************************************** + UNIX getfacl. +****************************************************************************/ + +static int cmd_getfacl(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src = NULL; + char *name = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + uint16_t major, minor; + uint32_t caplow, caphigh; + char *retbuf = NULL; + size_t rb_size = 0; + SMB_STRUCT_STAT sbuf; + size_t num_file_acls = 0; + size_t num_dir_acls = 0; + size_t expected_buflen; + uint16_t i; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&name,NULL)) { + d_printf("getfacl filename\n"); + return 1; + } + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + name); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("stat %s: %s\n", src, nt_errstr(status)); + return 1; + } + + if (!SERVER_HAS_UNIX_CIFS(targetcli)) { + d_printf("Server doesn't support UNIX CIFS calls.\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"getfacl\" command can be used.\n"); + return 1; + } + + status = cli_unix_extensions_version(targetcli, &major, &minor, + &caplow, &caphigh); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Can't get UNIX CIFS version from server: %s.\n", + nt_errstr(status)); + return 1; + } + + if (!(caplow & CIFS_UNIX_POSIX_ACLS_CAP)) { + d_printf("This server supports UNIX extensions " + "but doesn't support POSIX ACLs.\n"); + return 1; + } + + status = cli_posix_stat(targetcli, targetname, &sbuf); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s getfacl doing a stat on file %s\n", + nt_errstr(status), src); + return 1; + } + + status = cli_posix_getacl(targetcli, targetname, ctx, &rb_size, &retbuf); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s getfacl file %s\n", + nt_errstr(status), src); + return 1; + } + + /* ToDo : Print out the ACL values. */ + if (rb_size < 6 || SVAL(retbuf,0) != SMB_POSIX_ACL_VERSION) { + d_printf("getfacl file %s, unknown POSIX acl version %u.\n", + src, (unsigned int)CVAL(retbuf,0) ); + return 1; + } + + num_file_acls = SVAL(retbuf,2); + num_dir_acls = SVAL(retbuf,4); + + /* + * No overflow check, num_*_acls comes from a 16-bit value, + * and we expect expected_buflen (size_t) to be of at least 32 + * bit. + */ + expected_buflen = SMB_POSIX_ACL_HEADER_SIZE + + SMB_POSIX_ACL_ENTRY_SIZE*(num_file_acls+num_dir_acls); + + if (rb_size != expected_buflen) { + d_printf("getfacl file %s, incorrect POSIX acl buffer size " + "(should be %zu, was %zu).\n", + src, + expected_buflen, + rb_size); + return 1; + } + + d_printf("# file: %s\n", src); + d_printf("# owner: %u\n# group: %u\n", (unsigned int)sbuf.st_ex_uid, (unsigned int)sbuf.st_ex_gid); + + if (num_file_acls == 0 && num_dir_acls == 0) { + d_printf("No acls found.\n"); + } + + for (i = 0; i < num_file_acls; i++) { + uint32_t uorg; + fstring permstring; + unsigned char tagtype = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)); + unsigned char perms = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+1); + + switch(tagtype) { + case SMB_POSIX_ACL_USER_OBJ: + d_printf("user::"); + break; + case SMB_POSIX_ACL_USER: + uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+2); + d_printf("user:%u:", uorg); + break; + case SMB_POSIX_ACL_GROUP_OBJ: + d_printf("group::"); + break; + case SMB_POSIX_ACL_GROUP: + uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+(i*SMB_POSIX_ACL_ENTRY_SIZE)+2); + d_printf("group:%u:", uorg); + break; + case SMB_POSIX_ACL_MASK: + d_printf("mask::"); + break; + case SMB_POSIX_ACL_OTHER: + d_printf("other::"); + break; + default: + d_printf("getfacl file %s, incorrect POSIX acl tagtype (%u).\n", + src, (unsigned int)tagtype ); + SAFE_FREE(retbuf); + return 1; + } + + d_printf("%s\n", perms_to_string(permstring, perms)); + } + + for (i = 0; i < num_dir_acls; i++) { + uint32_t uorg; + fstring permstring; + unsigned char tagtype = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)); + unsigned char perms = CVAL(retbuf, SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+1); + + switch(tagtype) { + case SMB_POSIX_ACL_USER_OBJ: + d_printf("default:user::"); + break; + case SMB_POSIX_ACL_USER: + uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+2); + d_printf("default:user:%u:", uorg); + break; + case SMB_POSIX_ACL_GROUP_OBJ: + d_printf("default:group::"); + break; + case SMB_POSIX_ACL_GROUP: + uorg = IVAL(retbuf,SMB_POSIX_ACL_HEADER_SIZE+((i+num_file_acls)*SMB_POSIX_ACL_ENTRY_SIZE)+2); + d_printf("default:group:%u:", uorg); + break; + case SMB_POSIX_ACL_MASK: + d_printf("default:mask::"); + break; + case SMB_POSIX_ACL_OTHER: + d_printf("default:other::"); + break; + default: + d_printf("getfacl file %s, incorrect POSIX acl tagtype (%u).\n", + src, (unsigned int)tagtype ); + SAFE_FREE(retbuf); + return 1; + } + + d_printf("%s\n", perms_to_string(permstring, perms)); + } + + return 0; +} + +/**************************************************************************** + Get the EA list of a file +****************************************************************************/ + +static int cmd_geteas(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src = NULL; + char *name = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + NTSTATUS status; + size_t i, num_eas; + struct ea_struct *eas; + struct cli_credentials *creds = samba_cmdline_get_creds(); + + if (!next_token_talloc(ctx, &cmd_ptr,&name,NULL)) { + d_printf("geteas filename\n"); + return 1; + } + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + name); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("stat %s: %s\n", src, nt_errstr(status)); + return 1; + } + + status = cli_get_ea_list_path(targetcli, targetname, talloc_tos(), + &num_eas, &eas); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cli_get_ea_list_path: %s\n", nt_errstr(status)); + return 1; + } + + for (i=0; i<num_eas; i++) { + d_printf("%s (%d) =\n", eas[i].name, (int)eas[i].flags); + dump_data_file(eas[i].value.data, eas[i].value.length, false, + stdout); + d_printf("\n"); + } + + TALLOC_FREE(eas); + + return 0; +} + +/**************************************************************************** + Set an EA of a file +****************************************************************************/ + +static int cmd_setea(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src = NULL; + char *name = NULL; + char *eaname = NULL; + char *eavalue = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr, &name, NULL) + || !next_token_talloc(ctx, &cmd_ptr, &eaname, NULL)) { + d_printf("setea filename eaname value\n"); + return 1; + } + if (!next_token_talloc(ctx, &cmd_ptr, &eavalue, NULL)) { + eavalue = talloc_strdup(ctx, ""); + } + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + name); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("stat %s: %s\n", src, nt_errstr(status)); + return 1; + } + + status = cli_set_ea_path(targetcli, targetname, eaname, eavalue, + strlen(eavalue)); + if (!NT_STATUS_IS_OK(status)) { + d_printf("set_ea %s: %s\n", src, nt_errstr(status)); + return 1; + } + + return 0; +} + +/**************************************************************************** + UNIX stat. +****************************************************************************/ + +static int cmd_stat(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src = NULL; + char *name = NULL; + char *targetname = NULL; + struct cli_state *targetcli; + fstring mode_str; + SMB_STRUCT_STAT sbuf; + struct tm *lt; + time_t tmp_time; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&name,NULL)) { + d_printf("stat file\n"); + return 1; + } + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + name); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("stat %s: %s\n", src, nt_errstr(status)); + return 1; + } + + if (!SERVER_HAS_UNIX_CIFS(targetcli)) { + d_printf("Server doesn't support UNIX CIFS calls.\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"stat\" command can be used.\n"); + return 1; + } + + status = cli_posix_stat(targetcli, targetname, &sbuf); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s stat file %s\n", + nt_errstr(status), src); + return 1; + } + + /* Print out the stat values. */ + d_printf("File: %s\n", src); + d_printf("Size: %-12.0f\tBlocks: %u\t%s\n", + (double)sbuf.st_ex_size, + (unsigned int)sbuf.st_ex_blocks, + filetype_to_str(sbuf.st_ex_mode)); + +#if defined(S_ISCHR) && defined(S_ISBLK) + if (S_ISCHR(sbuf.st_ex_mode) || S_ISBLK(sbuf.st_ex_mode)) { + d_printf("Inode: %.0f\tLinks: %u\tDevice type: %u,%u\n", + (double)sbuf.st_ex_ino, + (unsigned int)sbuf.st_ex_nlink, + unix_dev_major(sbuf.st_ex_rdev), + unix_dev_minor(sbuf.st_ex_rdev)); + } else +#endif + d_printf("Inode: %.0f\tLinks: %u\n", + (double)sbuf.st_ex_ino, + (unsigned int)sbuf.st_ex_nlink); + + d_printf("Access: (0%03o/%s)\tUid: %u\tGid: %u\n", + ((int)sbuf.st_ex_mode & 0777), + unix_mode_to_str(mode_str, sbuf.st_ex_mode), + (unsigned int)sbuf.st_ex_uid, + (unsigned int)sbuf.st_ex_gid); + + tmp_time = convert_timespec_to_time_t(sbuf.st_ex_atime); + lt = localtime(&tmp_time); + if (lt) { + strftime(mode_str, sizeof(mode_str), "%Y-%m-%d %T %z", lt); + } else { + fstrcpy(mode_str, "unknown"); + } + d_printf("Access: %s\n", mode_str); + + tmp_time = convert_timespec_to_time_t(sbuf.st_ex_mtime); + lt = localtime(&tmp_time); + if (lt) { + strftime(mode_str, sizeof(mode_str), "%Y-%m-%d %T %z", lt); + } else { + fstrcpy(mode_str, "unknown"); + } + d_printf("Modify: %s\n", mode_str); + + tmp_time = convert_timespec_to_time_t(sbuf.st_ex_ctime); + lt = localtime(&tmp_time); + if (lt) { + strftime(mode_str, sizeof(mode_str), "%Y-%m-%d %T %z", lt); + } else { + fstrcpy(mode_str, "unknown"); + } + d_printf("Change: %s\n", mode_str); + + return 0; +} + + +/**************************************************************************** + UNIX chown. +****************************************************************************/ + +static int cmd_chown(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src = NULL; + uid_t uid; + gid_t gid; + char *buf, *buf2, *buf3; + struct cli_state *targetcli; + char *targetname = NULL; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf3,NULL)) { + d_printf("chown uid gid file\n"); + return 1; + } + + uid = (uid_t)atoi(buf); + gid = (gid_t)atoi(buf2); + + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf3); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("chown %s: %s\n", src, nt_errstr(status)); + return 1; + } + + if (!SERVER_HAS_UNIX_CIFS(targetcli)) { + d_printf("Server doesn't support UNIX CIFS calls.\n"); + return 1; + } + + if (CLI_DIRSEP_CHAR != '/') { + d_printf("Command \"posix\" must be issued before " + "the \"chown\" command can be used.\n"); + return 1; + } + + status = cli_posix_chown(targetcli, targetname, uid, gid); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s chown file %s uid=%d, gid=%d\n", + nt_errstr(status), src, (int)uid, (int)gid); + return 1; + } + + return 0; +} + +/**************************************************************************** + Rename some file. +****************************************************************************/ + +static int cmd_rename(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src, *dest; + char *buf, *buf2; + struct cli_state *targetcli; + char *targetsrc; + char *targetdest; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + bool replace = false; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) { + d_printf("rename <src> <dest> [-f]\n"); + return 1; + } + + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + dest = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf2); + if (!dest) { + return 1; + } + dest = client_clean_name(ctx, dest); + if (dest == NULL) { + return 1; + } + + if (next_token_talloc(ctx, &cmd_ptr, &buf, NULL) && + strcsequal(buf, "-f")) { + replace = true; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetsrc); + if (!NT_STATUS_IS_OK(status)) { + d_printf("rename %s: %s\n", src, nt_errstr(status)); + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, dest, &targetcli, &targetdest); + if (!NT_STATUS_IS_OK(status)) { + d_printf("rename %s: %s\n", dest, nt_errstr(status)); + return 1; + } + + status = cli_rename(targetcli, targetsrc, targetdest, replace); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s renaming files %s -> %s \n", + nt_errstr(status), + targetsrc, + targetdest); + return 1; + } + + return 0; +} + +struct scopy_timing { + struct timespec tp_start; +}; + +static int scopy_status(off_t written, void *priv) +{ + struct timespec tp_end; + unsigned int scopy_total_time_ms; + struct scopy_timing *st = priv; + + clock_gettime_mono(&tp_end); + scopy_total_time_ms = nsec_time_diff(&tp_end,&st->tp_start)/1000000; + + DEBUG(5,("Copied %jd bytes at an average %3.1f kb/s\n", + (intmax_t)written, written / (1.024*scopy_total_time_ms))); + + return true; +} + +/**************************************************************************** + Server-Side copy some file. +****************************************************************************/ + +static int cmd_scopy(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src, *dest; + char *buf, *buf2; + struct cli_state *targetcli; + char *targetsrc; + char *targetdest; + uint32_t DesiredAccess, ShareAccess, CreateDisposition, CreateOptions; + struct smb_create_returns cr; + uint16_t destfnum = (uint16_t)-1; + uint16_t srcfnum = (uint16_t)-1; + off_t written = 0; + struct scopy_timing st; + int rc = 0; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) { + d_printf("scopy <src> <dest>\n"); + return 1; + } + + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + dest = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf2); + if (!dest) { + return 1; + } + dest = client_clean_name(ctx, dest); + if (dest == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetsrc); + if (!NT_STATUS_IS_OK(status)) { + d_printf("scopy %s: %s\n", src, nt_errstr(status)); + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, dest, &targetcli, &targetdest); + if (!NT_STATUS_IS_OK(status)) { + d_printf("scopy %s: %s\n", dest, nt_errstr(status)); + return 1; + } + + + DesiredAccess = (FILE_READ_DATA|FILE_READ_EA|FILE_READ_ATTRIBUTES| + READ_CONTROL_ACCESS|SYNCHRONIZE_ACCESS); + ShareAccess = FILE_SHARE_READ|FILE_SHARE_DELETE; + CreateDisposition = FILE_OPEN; + CreateOptions = (FILE_SEQUENTIAL_ONLY|FILE_NON_DIRECTORY_FILE| + FILE_OPEN_REPARSE_POINT); + status = cli_ntcreate(targetcli, targetsrc, 0, DesiredAccess, 0, + ShareAccess, CreateDisposition, CreateOptions, 0x0, + &srcfnum, &cr); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to open file %s. %s\n", + targetsrc, nt_errstr(status)); + return 1; + } + + DesiredAccess = (FILE_READ_DATA|FILE_WRITE_DATA|FILE_APPEND_DATA|FILE_READ_EA| + FILE_WRITE_EA|FILE_READ_ATTRIBUTES|FILE_WRITE_ATTRIBUTES| + DELETE_ACCESS|READ_CONTROL_ACCESS|WRITE_DAC_ACCESS|SYNCHRONIZE_ACCESS); + ShareAccess = FILE_SHARE_NONE; + CreateDisposition = FILE_CREATE; + CreateOptions = FILE_SEQUENTIAL_ONLY|FILE_NON_DIRECTORY_FILE; + status = cli_ntcreate(targetcli, targetdest, 0, DesiredAccess, + FILE_ATTRIBUTE_ARCHIVE, ShareAccess, CreateDisposition, + CreateOptions, 0x0, &destfnum, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Failed to create file %s. %s\n", + targetdest, nt_errstr(status)); + cli_close(targetcli, srcfnum); + return 1; + } + + clock_gettime_mono(&st.tp_start); + status = cli_splice(targetcli, targetcli, srcfnum, destfnum, + cr.end_of_file, 0, 0, &written, scopy_status, &st); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s copying file %s -> %s \n", + nt_errstr(status), + targetsrc, + targetdest); + rc = 1; + } + + status = cli_close(targetcli, srcfnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error %s closing remote source file\n", nt_errstr(status)); + rc = 1; + } + status = cli_close(targetcli, destfnum); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error %s closing remote dest file\n", nt_errstr(status)); + rc = 1; + } + + return rc; +} + +/**************************************************************************** + Print the volume name. +****************************************************************************/ + +static int cmd_volume(void) +{ + char *volname; + uint32_t serial_num; + time_t create_date; + NTSTATUS status; + + status = cli_get_fs_volume_info(cli, talloc_tos(), + &volname, &serial_num, + &create_date); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error %s getting volume info\n", nt_errstr(status)); + return 1; + } + + d_printf("Volume: |%s| serial number 0x%x\n", + volname, (unsigned int)serial_num); + return 0; +} + +/**************************************************************************** + Hard link files using the NT call. +****************************************************************************/ + +static int cmd_hardlink(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *src, *dest; + char *buf, *buf2; + struct cli_state *targetcli; + char *targetname; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL) || + !next_token_talloc(ctx, &cmd_ptr,&buf2,NULL)) { + d_printf("hardlink <src> <dest>\n"); + return 1; + } + + src = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (!src) { + return 1; + } + src = client_clean_name(ctx, src); + if (src == NULL) { + return 1; + } + + dest = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf2); + if (!dest) { + return 1; + } + dest = client_clean_name(ctx, dest); + if (dest == NULL) { + return 1; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, src, &targetcli, &targetname); + if (!NT_STATUS_IS_OK(status)) { + d_printf("hardlink %s: %s\n", src, nt_errstr(status)); + return 1; + } + + status = cli_hardlink(targetcli, targetname, dest); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s doing an NT hard link of files\n", + nt_errstr(status)); + 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) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf; + bool ok; + SMB_STRUCT_STAT sbuf; + + ok = next_token_talloc(ctx, &cmd_ptr,&buf,NULL); + if (ok && (sys_stat(buf, &sbuf, false) == 0)) { + newer_than = convert_timespec_to_time_t(sbuf.st_ex_mtime); + DEBUG(1,("Getting files newer than %s", + time_to_asc(newer_than))); + } else { + newer_than = 0; + } + + if (ok && newer_than == 0) { + d_printf("Error setting newer-than time\n"); + return 1; + } + + return 0; +} + +/**************************************************************************** + Watch directory changes +****************************************************************************/ + +static int cmd_notify(void) +{ + TALLOC_CTX *frame = talloc_stackframe(); + char *name, *buf; + NTSTATUS status; + uint16_t fnum; + + name = talloc_strdup(talloc_tos(), client_get_cur_dir()); + if (name == NULL) { + goto fail; + } + if (!next_token_talloc(talloc_tos(), &cmd_ptr, &buf, NULL)) { + goto usage; + } + name = talloc_asprintf_append(name, "%s", buf); + if (name == NULL) { + goto fail; + } + name = client_clean_name(talloc_tos(), name); + if (name == NULL) { + return 1; + } + status = cli_ntcreate( + cli, name, 0, FILE_READ_DATA, 0, + FILE_SHARE_READ|FILE_SHARE_WRITE|FILE_SHARE_DELETE, + FILE_OPEN, 0, 0, &fnum, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Could not open file: %s\n", nt_errstr(status)); + goto fail; + } + + while (1) { + uint32_t i; + uint32_t num_changes = 0; + struct notify_change *changes = NULL; + + status = cli_notify(cli, fnum, 1000, FILE_NOTIFY_CHANGE_ALL, + true, + talloc_tos(), &num_changes, &changes); + if (NT_STATUS_EQUAL(status, NT_STATUS_NOTIFY_ENUM_DIR)) { + printf("NOTIFY_ENUM_DIR\n"); + status = NT_STATUS_OK; + } + if (!NT_STATUS_IS_OK(status)) { + d_printf("notify returned %s\n", + nt_errstr(status)); + goto fail; + } + for (i=0; i<num_changes; i++) { + printf("%4.4x %s\n", changes[i].action, + changes[i].name); + } + TALLOC_FREE(changes); + } +usage: + d_printf("notify <dir name>\n"); +fail: + TALLOC_FREE(frame); + return 1; +} + +/**************************************************************************** + Set the archive level. +****************************************************************************/ + +static int cmd_archive(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf; + + if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + archive_level = atoi(buf); + } else { + d_printf("Archive level is %d\n",archive_level); + } + + return 0; +} + +/**************************************************************************** + Toggle the backup_intent state. +****************************************************************************/ + +static int cmd_backup(void) +{ + backup_intent = !backup_intent; + cli_set_backup_intent(cli, backup_intent); + DEBUG(2,("backup intent is now %s\n",backup_intent?"on":"off")); + return 1; +} + +/**************************************************************************** + 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 case sensitive flag. +****************************************************************************/ + +static int cmd_setcase(void) +{ + bool orig_case_sensitive = cli_set_case_sensitive(cli, false); + + cli_set_case_sensitive(cli, !orig_case_sensitive); + DEBUG(2,("filename case sensitivity is now %s\n",!orig_case_sensitive ? + "on":"off")); + return 0; +} + +/**************************************************************************** + Toggle the showacls flag. +****************************************************************************/ + +static int cmd_showacls(void) +{ + showacls = !showacls; + DEBUG(2,("showacls is now %s\n",showacls?"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 the lcd command. + ****************************************************************************/ + +static int cmd_lcd(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf; + char *d; + + if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + if (chdir(buf) == -1) { + d_printf("chdir to %s failed (%s)\n", + buf, strerror(errno)); + } + } + d = sys_getwd(); + if (!d) { + return 1; + } + DEBUG(2,("the local directory is now %s\n",d)); + SAFE_FREE(d); + return 0; +} + +/**************************************************************************** + Get a file restarting at end of local file. + ****************************************************************************/ + +static int cmd_reget(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *local_name = NULL; + char *remote_name = NULL; + char *fname = NULL; + char *p = NULL; + + remote_name = talloc_strdup(ctx, client_get_cur_dir()); + if (!remote_name) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr, &fname, NULL)) { + d_printf("reget <filename>\n"); + return 1; + } + remote_name = talloc_asprintf_append(remote_name, "%s", fname); + if (!remote_name) { + return 1; + } + remote_name = client_clean_name(ctx,remote_name); + if (!remote_name) { + return 1; + } + + local_name = fname; + next_token_talloc(ctx, &cmd_ptr, &p, NULL); + if (p) { + local_name = p; + } + + return do_get(remote_name, local_name, true); +} + +/**************************************************************************** + Put a file restarting at end of local file. + ****************************************************************************/ + +static int cmd_reput(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *local_name = NULL; + char *remote_name = NULL; + char *buf; + SMB_STRUCT_STAT st; + + remote_name = talloc_strdup(ctx, client_get_cur_dir()); + if (!remote_name) { + return 1; + } + + if (!next_token_talloc(ctx, &cmd_ptr, &local_name, NULL)) { + d_printf("reput <filename>\n"); + return 1; + } + + if (!file_exist_stat(local_name, &st, false)) { + d_printf("%s does not exist\n", local_name); + return 1; + } + + if (next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) { + remote_name = talloc_asprintf_append(remote_name, + "%s", buf); + } else { + remote_name = talloc_asprintf_append(remote_name, + "%s", local_name); + } + if (!remote_name) { + return 1; + } + + remote_name = client_clean_name(ctx, remote_name); + if (!remote_name) { + return 1; + } + + return do_put(remote_name, local_name, true); +} + +/**************************************************************************** + List a share name. + ****************************************************************************/ + +static void browse_fn(const char *name, uint32_t m, + const char *comment, void *state) +{ + const char *typestr = ""; + + switch (m & 7) { + case STYPE_DISKTREE: + typestr = "Disk"; + break; + case STYPE_PRINTQ: + typestr = "Printer"; + break; + case STYPE_DEVICE: + typestr = "Device"; + break; + case STYPE_IPC: + typestr = "IPC"; + break; + } + /* FIXME: If the remote machine returns non-ascii characters + in any of these fields, they can corrupt the output. We + should remove them. */ + if (!grepable) { + d_printf("\t%-15s %-10.10s%s\n", + name,typestr,comment); + } else { + d_printf ("%s|%s|%s\n",typestr,name,comment); + } +} + +static bool browse_host_rpc(bool sort) +{ + NTSTATUS status; + struct rpc_pipe_client *pipe_hnd = NULL; + TALLOC_CTX *frame = talloc_stackframe(); + WERROR werr; + struct srvsvc_NetShareInfoCtr info_ctr; + struct srvsvc_NetShareCtr1 ctr1; + uint32_t resume_handle = 0; + uint32_t total_entries = 0; + uint32_t i; + struct dcerpc_binding_handle *b; + + status = cli_rpc_pipe_open_noauth(cli, &ndr_table_srvsvc, + &pipe_hnd); + + if (!NT_STATUS_IS_OK(status)) { + DEBUG(10, ("Could not connect to srvsvc pipe: %s\n", + nt_errstr(status))); + TALLOC_FREE(frame); + return false; + } + + b = pipe_hnd->binding_handle; + + ZERO_STRUCT(info_ctr); + ZERO_STRUCT(ctr1); + + info_ctr.level = 1; + info_ctr.ctr.ctr1 = &ctr1; + + status = dcerpc_srvsvc_NetShareEnumAll(b, frame, + pipe_hnd->desthost, + &info_ctr, + 0xffffffff, + &total_entries, + &resume_handle, + &werr); + + if (!NT_STATUS_IS_OK(status) || !W_ERROR_IS_OK(werr)) { + TALLOC_FREE(pipe_hnd); + TALLOC_FREE(frame); + return false; + } + + for (i=0; i < info_ctr.ctr.ctr1->count; i++) { + struct srvsvc_NetShareInfo1 info = info_ctr.ctr.ctr1->array[i]; + browse_fn(info.name, info.type, info.comment, NULL); + } + + TALLOC_FREE(pipe_hnd); + TALLOC_FREE(frame); + return true; +} + +/**************************************************************************** + Try and browse available connections on a host. +****************************************************************************/ + +static bool browse_host(bool sort) +{ + int ret; + + if (!grepable) { + d_printf("\n\tSharename Type Comment\n"); + d_printf("\t--------- ---- -------\n"); + } + + if (browse_host_rpc(sort)) { + return true; + } + + if (smbXcli_conn_protocol(cli->conn) > PROTOCOL_NT1) { + return false; + } + + ret = cli_RNetShareEnum(cli, browse_fn, NULL); + if (ret == -1) { + NTSTATUS status = cli_nt_error(cli); + d_printf("Error returning browse list: %s\n", + nt_errstr(status)); + } + + return (ret != -1); +} + +/**************************************************************************** + List a server name. +****************************************************************************/ + +static void server_fn(const char *name, uint32_t m, + const char *comment, void *state) +{ + + if (!grepable){ + d_printf("\t%-16s %s\n", name, comment); + } else { + d_printf("%s|%s|%s\n",(char *)state, name, comment); + } +} + +/**************************************************************************** + Try and browse available connections on a host. +****************************************************************************/ + +static bool list_servers(const char *wk_grp) +{ + fstring state; + + if (!cli->server_domain) + return false; + + if (!grepable) { + d_printf("\n\tServer Comment\n"); + d_printf("\t--------- -------\n"); + }; + fstrcpy( state, "Server" ); + cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_ALL, server_fn, + state); + + if (!grepable) { + d_printf("\n\tWorkgroup Master\n"); + d_printf("\t--------- -------\n"); + }; + + fstrcpy( state, "Workgroup" ); + cli_NetServerEnum(cli, cli->server_domain, SV_TYPE_DOMAIN_ENUM, + server_fn, state); + return true; +} + +/**************************************************************************** + Print or set current VUID +****************************************************************************/ + +static int cmd_vuid(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + d_printf("Current VUID is %d\n", + cli_state_get_uid(cli)); + return 0; + } + + cli_state_set_uid(cli, atoi(buf)); + return 0; +} + +/**************************************************************************** + Setup a new VUID, by issuing a session setup +****************************************************************************/ + +static int cmd_logon(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *l_username, *l_password; + struct cli_credentials *creds = NULL; + NTSTATUS nt_status; + + if (!next_token_talloc(ctx, &cmd_ptr,&l_username,NULL)) { + d_printf("logon <username> [<password>]\n"); + return 0; + } + + if (!next_token_talloc(ctx, &cmd_ptr,&l_password,NULL)) { + char pwd[256] = {0}; + int rc; + + rc = samba_getpass("Password: ", pwd, sizeof(pwd), false, false); + if (rc == 0) { + l_password = talloc_strdup(ctx, pwd); + } + } + if (!l_password) { + return 1; + } + + creds = cli_session_creds_init(ctx, + l_username, + lp_workgroup(), + NULL, /* realm */ + l_password, + false, /* use_kerberos */ + false, /* fallback_after_kerberos */ + false, /* use_ccache */ + false); /* password_is_nt_hash */ + if (creds == NULL) { + d_printf("cli_session_creds_init() failed.\n"); + return -1; + } + nt_status = cli_session_setup_creds(cli, creds); + TALLOC_FREE(creds); + if (!NT_STATUS_IS_OK(nt_status)) { + d_printf("session setup failed: %s\n", nt_errstr(nt_status)); + return -1; + } + + d_printf("Current VUID is %d\n", cli_state_get_uid(cli)); + return 0; +} + +/** + * close the session + */ + +static int cmd_logoff(void) +{ + NTSTATUS status; + + status = cli_ulogoff(cli); + if (!NT_STATUS_IS_OK(status)) { + d_printf("logoff failed: %s\n", nt_errstr(status)); + return -1; + } + + d_printf("logoff successful\n"); + return 0; +} + + +/** + * tree connect (connect to a share) + */ + +static int cmd_tcon(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *sharename; + NTSTATUS status; + + if (!next_token_talloc(ctx, &cmd_ptr, &sharename, NULL)) { + d_printf("tcon <sharename>\n"); + return 0; + } + + if (!sharename) { + return 1; + } + + status = cli_tree_connect(cli, sharename, "?????", NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("tcon failed: %s\n", nt_errstr(status)); + return -1; + } + + d_printf("tcon to %s successful, tid: %u\n", sharename, + cli_state_get_tid(cli)); + + talloc_free(sharename); + + return 0; +} + +/** + * tree disconnect (disconnect from a share) + */ + +static int cmd_tdis(void) +{ + NTSTATUS status; + + status = cli_tdis(cli); + if (!NT_STATUS_IS_OK(status)) { + d_printf("tdis failed: %s\n", nt_errstr(status)); + return -1; + } + + d_printf("tdis successful\n"); + return 0; +} + + +/** + * get or set tid + */ + +static int cmd_tid(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *tid_str; + + if (!next_token_talloc(ctx, &cmd_ptr, &tid_str, NULL)) { + if (cli_state_has_tcon(cli)) { + d_printf("current tid is %d\n", cli_state_get_tid(cli)); + } else { + d_printf("no tcon currently\n"); + } + } else { + uint32_t tid = atoi(tid_str); + if (!cli_state_has_tcon(cli)) { + d_printf("no tcon currently\n"); + } + cli_state_set_tid(cli, tid); + } + + return 0; +} + + +/**************************************************************************** + list active connections +****************************************************************************/ + +static int cmd_list_connect(void) +{ + cli_cm_display(cli); + return 0; +} + +/**************************************************************************** + display the current active client connection +****************************************************************************/ + +static int cmd_show_connect( void ) +{ + TALLOC_CTX *ctx = talloc_tos(); + struct cli_state *targetcli; + char *targetpath; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + status = cli_resolve_path(ctx, "", + creds, + cli, + client_get_cur_dir(), &targetcli, + &targetpath); + if (!NT_STATUS_IS_OK(status)) { + d_printf("showconnect %s: %s\n", cur_dir, nt_errstr(status)); + return 1; + } + + d_printf("//%s/%s\n", smbXcli_conn_remote_name(targetcli->conn), targetcli->share); + return 0; +} + +/** + * cmd_utimes - interactive command to set the four times + * + * Read a filename and four times from the client command line and update + * the file times. A value of -1 for a time means don't change. + */ +static int cmd_utimes(void) +{ + char *buf; + char *fname = NULL; + struct timespec times[4] = {{0}}; + struct timeval_buf tbuf[4]; + int time_count = 0; + int err = 0; + bool ok; + TALLOC_CTX *ctx = talloc_new(NULL); + NTSTATUS status; + + if (ctx == NULL) { + return 1; + } + + ok = next_token_talloc(ctx, &cmd_ptr, &buf, NULL); + if (!ok) { + d_printf("utimes <filename> <create-time> <access-time> " + "<write-time> <change-time>\n"); + d_printf("Dates should be in YY:MM:DD-HH:MM:SS format " + "or -1 for no change\n"); + err = 1; + goto out; + } + + fname = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (fname == NULL) { + err = 1; + goto out; + } + fname = client_clean_name(ctx, fname); + if (fname == NULL) { + err = 1; + goto out; + } + + while (next_token_talloc(ctx, &cmd_ptr, &buf, NULL) && + time_count < 4) { + const char *s = buf; + struct tm tm = {0,}; + time_t t; + char *ret; + + if (strlen(s) == 2 && strcmp(s, "-1") == 0) { + times[time_count] = make_omit_timespec(); + time_count++; + continue; + } + + ret = strptime(s, "%y:%m:%d-%H:%M:%S", &tm); + + if (ret == NULL) { + ret = strptime(s, "%Y:%m:%d-%H:%M:%S", &tm); + } + + /* We could not match all the chars, so print error */ + if (ret == NULL || *ret != 0) { + d_printf("Invalid date format: %s\n", s); + d_printf("utimes <filename> <create-time> " + "<access-time> <write-time> <change-time>\n"); + d_printf("Dates should be in [YY]YY:MM:DD-HH:MM:SS " + "format or -1 for no change\n"); + err = 1; + goto out; + } + + /* Convert tm to a time_t */ + t = mktime(&tm); + times[time_count] = (struct timespec){.tv_sec = t}; + time_count++; + } + + if (time_count < 4) { + d_printf("Insufficient dates: %d\n", time_count); + d_printf("utimes <filename> <create-time> <access-time> " + "<write-time> <change-time>\n"); + d_printf("Dates should be in YY:MM:DD-HH:MM:SS format " + "or -1 for no change\n"); + err = 1; + goto out; + } + + DEBUG(10, ("times\nCreate: %sAccess: %s Write: %sChange: %s\n", + timespec_string_buf(×[0], false, &tbuf[0]), + timespec_string_buf(×[1], false, &tbuf[1]), + timespec_string_buf(×[2], false, &tbuf[2]), + timespec_string_buf(×[3], false, &tbuf[3]))); + + status = cli_setpathinfo_ext( + cli, fname, times[0], times[1], times[2], times[3], + (uint32_t)-1); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cli_setpathinfo_ext failed: %s\n", + nt_errstr(status)); + err = 1; + goto out; + } +out: + talloc_free(ctx); + return err; +} + +/** + * set_remote_attr - set DOS attributes of a remote file + * @filename: path to the file name + * @new_attr: attribute bit mask to use + * @mode: one of ATTR_SET or ATTR_UNSET + * + * Update the file attributes with the one provided. + */ +int set_remote_attr(const char *filename, uint32_t new_attr, int mode) +{ + extern struct cli_state *cli; + uint32_t old_attr; + NTSTATUS status; + + status = cli_getatr(cli, filename, &old_attr, NULL, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cli_getatr failed: %s\n", nt_errstr(status)); + return 1; + } + + if (mode == ATTR_SET) { + new_attr |= old_attr; + } else { + new_attr = old_attr & ~new_attr; + } + + status = cli_setatr(cli, filename, new_attr, 0); + if (!NT_STATUS_IS_OK(status)) { + d_printf("cli_setatr failed: %s\n", nt_errstr(status)); + return 1; + } + + return 0; +} + +/** + * cmd_setmode - interactive command to set DOS attributes + * + * Read a filename and mode from the client command line and update + * the file DOS attributes. + */ +int cmd_setmode(void) +{ + char *buf; + char *fname = NULL; + uint32_t attr[2] = {0}; + int mode = ATTR_SET; + int err = 0; + bool ok; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + ok = next_token_talloc(ctx, &cmd_ptr, &buf, NULL); + if (!ok) { + d_printf("setmode <filename> <[+|-]rsha>\n"); + err = 1; + goto out; + } + + fname = talloc_asprintf(ctx, + "%s%s", + client_get_cur_dir(), + buf); + if (fname == NULL) { + err = 1; + goto out; + } + fname = client_clean_name(ctx, fname); + if (fname == NULL) { + err = 1; + goto out; + } + + while (next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) { + const char *s = buf; + + while (*s) { + switch (*s++) { + case '+': + mode = ATTR_SET; + break; + case '-': + mode = ATTR_UNSET; + break; + case 'r': + attr[mode] |= FILE_ATTRIBUTE_READONLY; + break; + case 'h': + attr[mode] |= FILE_ATTRIBUTE_HIDDEN; + break; + case 's': + attr[mode] |= FILE_ATTRIBUTE_SYSTEM; + break; + case 'a': + attr[mode] |= FILE_ATTRIBUTE_ARCHIVE; + break; + default: + d_printf("setmode <filename> <perm=[+|-]rsha>\n"); + err = 1; + goto out; + } + } + } + + if (attr[ATTR_SET] == 0 && attr[ATTR_UNSET] == 0) { + d_printf("setmode <filename> <[+|-]rsha>\n"); + err = 1; + goto out; + } + + DEBUG(2, ("perm set %d %d\n", attr[ATTR_SET], attr[ATTR_UNSET])); + + /* ignore return value: server might not store DOS attributes */ + set_remote_attr(fname, attr[ATTR_SET], ATTR_SET); + set_remote_attr(fname, attr[ATTR_UNSET], ATTR_UNSET); +out: + talloc_free(ctx); + return err; +} + +/**************************************************************************** + iosize command +***************************************************************************/ + +int cmd_iosize(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf; + int iosize; + struct cli_credentials *creds = samba_cmdline_get_creds(); + bool smb_encrypt = + (cli_credentials_get_smb_encryption(creds) == + SMB_ENCRYPTION_REQUIRED); + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + if (smbXcli_conn_protocol(cli->conn) < PROTOCOL_SMB2_02) { + if (!smb_encrypt) { + d_printf("iosize <n> or iosize 0x<n>. " + "Minimum is 0 (default), " + "max is 16776960 (0xFFFF00)\n"); + } else { + d_printf("iosize <n> or iosize 0x<n>. " + "(Encrypted connection) ," + "Minimum is 0 (default), " + "max is 130048 (0x1FC00)\n"); + } + } else { + d_printf("iosize <n> or iosize 0x<n>.\n"); + } + return 1; + } + + iosize = strtol(buf,NULL,0); + if (smbXcli_conn_protocol(cli->conn) < PROTOCOL_SMB2_02) { + if (smb_encrypt && (iosize < 0 || iosize > 0xFC00)) { + d_printf("iosize out of range for encrypted " + "connection (min = 0 (default), " + "max = 130048 (0x1FC00)\n"); + return 1; + } else if (!smb_encrypt && (iosize < 0 || iosize > 0xFFFF00)) { + d_printf("iosize out of range (min = 0 (default), " + "max = 16776960 (0xFFFF00)\n"); + return 1; + } + } + + io_bufsize = iosize; + d_printf("iosize is now %d\n", io_bufsize); + return 0; +} + +/**************************************************************************** + timeout command +***************************************************************************/ + +static int cmd_timeout(void) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *buf; + + if (!next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + unsigned int old_timeout = cli_set_timeout(cli, 0); + cli_set_timeout(cli, old_timeout); + d_printf("timeout <n> (per-operation timeout " + "in seconds - currently %u).\n", + old_timeout/1000); + return 1; + } + + io_timeout = strtol(buf,NULL,0); + cli_set_timeout(cli, io_timeout*1000); + d_printf("io_timeout per operation is now %d\n", io_timeout); + return 0; +} + + +/**************************************************************************** +history +****************************************************************************/ +static int cmd_history(void) +{ +#if defined(HAVE_LIBREADLINE) && defined(HAVE_HISTORY_LIST) + HIST_ENTRY **hlist; + int i; + + hlist = history_list(); + + for (i = 0; hlist && hlist[i]; i++) { + DEBUG(0, ("%d: %s\n", i, hlist[i]->line)); + } +#else + DEBUG(0,("no history without readline support\n")); +#endif + + return 0; +} + +/* 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}}, + {"allinfo",cmd_allinfo,"<file> show all available info", + {COMPL_REMOTE,COMPL_NONE}}, + {"altname",cmd_altname,"<file> show alt name",{COMPL_REMOTE,COMPL_NONE}}, + {"archive",cmd_archive,"<level>\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}}, + {"backup",cmd_backup,"toggle backup intent state",{COMPL_NONE,COMPL_NONE}}, + {"blocksize",cmd_block,"blocksize <number> (default 20)",{COMPL_NONE,COMPL_NONE}}, + {"cancel",cmd_cancel,"<jobid> cancel a print queue entry",{COMPL_NONE,COMPL_NONE}}, + {"case_sensitive",cmd_setcase,"toggle the case sensitive flag to server",{COMPL_NONE,COMPL_NONE}}, + {"cd",cmd_cd,"[directory] change/report the remote directory",{COMPL_REMOTE,COMPL_NONE}}, + {"chmod",cmd_chmod,"<src> <mode> chmod a file using UNIX permission",{COMPL_REMOTE,COMPL_NONE}}, + {"chown",cmd_chown,"<src> <uid> <gid> chown a file using UNIX uids and gids",{COMPL_REMOTE,COMPL_NONE}}, + {"close",cmd_close,"<fid> close a file given a fid",{COMPL_REMOTE,COMPL_NONE}}, + {"del",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}}, + {"deltree",cmd_deltree,"<mask> recursively delete all matching files and directories",{COMPL_REMOTE,COMPL_NONE}}, + {"dir",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}}, + {"du",cmd_du,"<mask> computes the total size of the current directory",{COMPL_REMOTE,COMPL_NONE}}, + {"echo",cmd_echo,"ping the server",{COMPL_NONE,COMPL_NONE}}, + {"exit",cmd_quit,"logoff the server",{COMPL_NONE,COMPL_NONE}}, + {"get",cmd_get,"<remote name> [local name] get a file",{COMPL_REMOTE,COMPL_LOCAL}}, + {"getfacl",cmd_getfacl,"<file name> get the POSIX ACL on a file (UNIX extensions only)",{COMPL_REMOTE,COMPL_NONE}}, + {"geteas", cmd_geteas, "<file name> get the EA list of a file", + {COMPL_REMOTE, COMPL_NONE}}, + {"hardlink",cmd_hardlink,"<src> <dest> create a Windows hard link",{COMPL_REMOTE,COMPL_REMOTE}}, + {"help",cmd_help,"[command] give help on a command",{COMPL_NONE,COMPL_NONE}}, + {"history",cmd_history,"displays the command history",{COMPL_NONE,COMPL_NONE}}, + {"iosize",cmd_iosize,"iosize <number> (default 64512)",{COMPL_NONE,COMPL_NONE}}, + {"lcd",cmd_lcd,"[directory] change/report the local current working directory",{COMPL_LOCAL,COMPL_NONE}}, + {"link",cmd_link,"<oldname> <newname> create a UNIX hard link",{COMPL_REMOTE,COMPL_REMOTE}}, + {"lock",cmd_lock,"lock <fnum> [r|w] <hex-start> <hex-len> : set a POSIX lock",{COMPL_REMOTE,COMPL_REMOTE}}, + {"lowercase",cmd_lowercase,"toggle lowercasing of filenames for get",{COMPL_NONE,COMPL_NONE}}, + {"ls",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}}, + {"l",cmd_dir,"<mask> list the contents of the current directory",{COMPL_REMOTE,COMPL_NONE}}, + {"mask",cmd_select,"<mask> mask all filenames against this",{COMPL_REMOTE,COMPL_NONE}}, + {"md",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}}, + {"mget",cmd_mget,"<mask> get all the matching files",{COMPL_REMOTE,COMPL_NONE}}, + {"mkdir",cmd_mkdir,"<directory> make a directory",{COMPL_NONE,COMPL_NONE}}, + {"more",cmd_more,"<remote name> view a remote file with your pager",{COMPL_REMOTE,COMPL_NONE}}, + {"mput",cmd_mput,"<mask> put all matching files",{COMPL_REMOTE,COMPL_NONE}}, + {"newer",cmd_newer,"<file> only mget files newer than the specified local file",{COMPL_LOCAL,COMPL_NONE}}, + {"notify",cmd_notify,"<file>Get notified of dir changes",{COMPL_REMOTE,COMPL_NONE}}, + {"open",cmd_open,"<mask> open a file",{COMPL_REMOTE,COMPL_NONE}}, + {"posix", cmd_posix, "turn on all POSIX capabilities", {COMPL_REMOTE,COMPL_NONE}}, + {"posix_encrypt",cmd_posix_encrypt,"<domain> <user> <password> start up transport encryption",{COMPL_REMOTE,COMPL_NONE}}, + {"posix_open",cmd_posix_open,"<name> 0<mode> open_flags mode open a file using POSIX interface",{COMPL_REMOTE,COMPL_NONE}}, + {"posix_mkdir",cmd_posix_mkdir,"<name> 0<mode> creates a directory using POSIX interface",{COMPL_REMOTE,COMPL_NONE}}, + {"posix_rmdir",cmd_posix_rmdir,"<name> removes a directory using POSIX interface",{COMPL_REMOTE,COMPL_NONE}}, + {"posix_unlink",cmd_posix_unlink,"<name> removes a file using POSIX interface",{COMPL_REMOTE,COMPL_NONE}}, + {"posix_whoami",cmd_posix_whoami,"return logged on user information " + "using POSIX interface",{COMPL_REMOTE,COMPL_NONE}}, + {"print",cmd_print,"<file name> print a file",{COMPL_NONE,COMPL_NONE}}, + {"prompt",cmd_prompt,"toggle prompting for filenames for mget and mput",{COMPL_NONE,COMPL_NONE}}, + {"put",cmd_put,"<local name> [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}}, + {"readlink",cmd_readlink,"filename Do a UNIX extensions readlink call on a symlink",{COMPL_REMOTE,COMPL_REMOTE}}, + {"rd",cmd_rmdir,"<directory> remove a directory",{COMPL_NONE,COMPL_NONE}}, + {"recurse",cmd_recurse,"toggle directory recursion for mget and mput",{COMPL_NONE,COMPL_NONE}}, + {"reget",cmd_reget,"<remote name> [local name] get a file restarting at end of local file",{COMPL_REMOTE,COMPL_LOCAL}}, + {"rename",cmd_rename,"<src> <dest> rename some files",{COMPL_REMOTE,COMPL_REMOTE}}, + {"reput",cmd_reput,"<local name> [remote name] put a file restarting at end of remote file",{COMPL_LOCAL,COMPL_REMOTE}}, + {"rm",cmd_del,"<mask> delete all matching files",{COMPL_REMOTE,COMPL_NONE}}, + {"rmdir",cmd_rmdir,"<directory> remove a directory",{COMPL_REMOTE,COMPL_NONE}}, + {"showacls",cmd_showacls,"toggle if ACLs are shown or not",{COMPL_NONE,COMPL_NONE}}, + {"setea", cmd_setea, "<file name> <eaname> <eaval> Set an EA of a file", + {COMPL_REMOTE, COMPL_LOCAL}}, + {"setmode",cmd_setmode,"<file name> <setmode string> change modes of file",{COMPL_REMOTE,COMPL_NONE}}, + {"scopy",cmd_scopy,"<src> <dest> server-side copy file",{COMPL_REMOTE,COMPL_REMOTE}}, + {"stat",cmd_stat,"<file name> Do a UNIX extensions stat call on a file",{COMPL_REMOTE,COMPL_NONE}}, + {"symlink",cmd_symlink,"<oldname> <newname> create a UNIX symlink",{COMPL_REMOTE,COMPL_REMOTE}}, + {"tar",cmd_tar,"tar <c|x>[IXFvbgNan] current directory to/from <file name>",{COMPL_NONE,COMPL_NONE}}, + {"tarmode",cmd_tarmode,"<full|inc|reset|noreset> tar's behaviour towards archive bits",{COMPL_NONE,COMPL_NONE}}, + {"timeout",cmd_timeout,"timeout <number> - set the per-operation timeout in seconds (default 20)",{COMPL_NONE,COMPL_NONE}}, + {"translate",cmd_translate,"toggle text translation for printing",{COMPL_NONE,COMPL_NONE}}, + {"unlock",cmd_unlock,"unlock <fnum> <hex-start> <hex-len> : remove a POSIX lock",{COMPL_REMOTE,COMPL_REMOTE}}, + {"volume",cmd_volume,"print the volume name",{COMPL_NONE,COMPL_NONE}}, + {"vuid",cmd_vuid,"change current vuid",{COMPL_NONE,COMPL_NONE}}, + {"wdel",cmd_wdel,"<attrib> <mask> wildcard delete all matching files",{COMPL_REMOTE,COMPL_NONE}}, + {"logon",cmd_logon,"establish new logon",{COMPL_NONE,COMPL_NONE}}, + {"listconnect",cmd_list_connect,"list open connections",{COMPL_NONE,COMPL_NONE}}, + {"showconnect",cmd_show_connect,"display the current active connection",{COMPL_NONE,COMPL_NONE}}, + {"tcon",cmd_tcon,"connect to a share" ,{COMPL_NONE,COMPL_NONE}}, + {"tdis",cmd_tdis,"disconnect from a share",{COMPL_NONE,COMPL_NONE}}, + {"tid",cmd_tid,"show or set the current tid (tree-id)",{COMPL_NONE,COMPL_NONE}}, + {"utimes", cmd_utimes,"<file name> <create_time> <access_time> <mod_time> " + "<ctime> set times", {COMPL_REMOTE,COMPL_NONE}}, + {"logoff",cmd_logoff,"log off (close the session)",{COMPL_NONE,COMPL_NONE}}, + {"..",cmd_cd_oneup,"change the remote directory (up one level)",{COMPL_REMOTE,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(char *tok) +{ + size_t i = 0, matches = 0; + size_t cmd=0; + size_t 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) +{ + TALLOC_CTX *ctx = talloc_tos(); + int i=0,j; + char *buf; + + if (next_token_talloc(ctx, &cmd_ptr,&buf,NULL)) { + 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(const char *cmd_in) +{ + TALLOC_CTX *ctx = talloc_tos(); + char *cmd = talloc_strdup(ctx, cmd_in); + int rc = 0; + struct cli_credentials *creds = samba_cmdline_get_creds(); + + if (!cmd) { + return 1; + } + /* establish the connection if not already */ + + if (!cli) { + NTSTATUS status; + + status = cli_cm_open(talloc_tos(), NULL, + desthost, + service, + creds, + have_ip ? &dest_ss : NULL, port, + name_type, + &cli); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + cli_set_timeout(cli, io_timeout*1000); + } + + while (cmd[0] != '\0') { + char *line; + char *p; + char *tok; + int i; + + if ((p = strchr_m(cmd, ';')) == 0) { + line = cmd; + cmd += strlen(cmd); + } else { + *p = '\0'; + line = cmd; + cmd = p + 1; + } + + /* and get the first part of the command */ + cmd_ptr = line; + if (!next_token_talloc(ctx, &cmd_ptr,&tok,NULL)) { + 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 + +struct completion_remote { + char *dirmask; + char **matches; + int count, samelen; + const char *text; + int len; +}; + +static NTSTATUS completion_remote_filter(struct file_info *f, + const char *mask, + void *state) +{ + struct completion_remote *info = (struct completion_remote *)state; + + if (info->count >= MAX_COMPLETIONS - 1) { + return NT_STATUS_OK; + } + if (strncmp(info->text, f->name, info->len) != 0) { + return NT_STATUS_OK; + } + if (ISDOT(f->name) || ISDOTDOT(f->name)) { + return NT_STATUS_OK; + } + + if ((info->dirmask[0] == 0) && !(f->attr & FILE_ATTRIBUTE_DIRECTORY)) + info->matches[info->count] = SMB_STRDUP(f->name); + else { + TALLOC_CTX *ctx = talloc_stackframe(); + char *tmp; + + tmp = talloc_strdup(ctx,info->dirmask); + if (!tmp) { + TALLOC_FREE(ctx); + return NT_STATUS_NO_MEMORY; + } + tmp = talloc_asprintf_append(tmp, "%s", f->name); + if (!tmp) { + TALLOC_FREE(ctx); + return NT_STATUS_NO_MEMORY; + } + if (f->attr & FILE_ATTRIBUTE_DIRECTORY) { + tmp = talloc_asprintf_append(tmp, "%s", + CLI_DIRSEP_STR); + } + if (!tmp) { + TALLOC_FREE(ctx); + return NT_STATUS_NO_MEMORY; + } + info->matches[info->count] = SMB_STRDUP(tmp); + TALLOC_FREE(ctx); + } + if (info->matches[info->count] == NULL) { + return NT_STATUS_OK; + } + if (f->attr & 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++; + return NT_STATUS_OK; +} + +static char **remote_completion(const char *text, int len) +{ + TALLOC_CTX *ctx = talloc_stackframe(); + char *dirmask = NULL; + char *targetpath = NULL; + struct cli_state *targetcli = NULL; + int i; + struct completion_remote info = { NULL, NULL, 1, 0, NULL, 0 }; + struct cli_credentials *creds = samba_cmdline_get_creds(); + NTSTATUS status; + + /* can't have non-static initialisation on Sun CC, so do it + at run time here */ + info.samelen = len; + info.text = text; + info.len = len; + + info.matches = SMB_MALLOC_ARRAY(char *,MAX_COMPLETIONS); + if (!info.matches) { + TALLOC_FREE(ctx); + return NULL; + } + + /* + * We're leaving matches[0] free to fill it later with the text to + * display: Either the one single match or the longest common subset + * of the matches. + */ + info.matches[0] = NULL; + info.count = 1; + + for (i = len-1; i >= 0; i--) { + if ((text[i] == '/') || (text[i] == CLI_DIRSEP_CHAR)) { + break; + } + } + + info.text = text+i+1; + info.samelen = info.len = len-i-1; + + if (i > 0) { + info.dirmask = SMB_MALLOC_ARRAY(char, i+2); + if (!info.dirmask) { + goto cleanup; + } + strncpy(info.dirmask, text, i+1); + info.dirmask[i+1] = 0; + dirmask = talloc_asprintf(ctx, + "%s%*s*", + client_get_cur_dir(), + i-1, + text); + } else { + info.dirmask = SMB_STRDUP(""); + if (!info.dirmask) { + goto cleanup; + } + dirmask = talloc_asprintf(ctx, + "%s*", + client_get_cur_dir()); + } + if (!dirmask) { + goto cleanup; + } + dirmask = client_clean_name(ctx, dirmask); + if (dirmask == NULL) { + goto cleanup; + } + + status = cli_resolve_path(ctx, "", + creds, + cli, dirmask, &targetcli, &targetpath); + if (!NT_STATUS_IS_OK(status)) { + goto cleanup; + } + status = cli_list(targetcli, targetpath, FILE_ATTRIBUTE_DIRECTORY | FILE_ATTRIBUTE_SYSTEM | FILE_ATTRIBUTE_HIDDEN, + completion_remote_filter, (void *)&info); + if (!NT_STATUS_IS_OK(status)) { + goto cleanup; + } + + if (info.count == 1) { + /* + * No matches at all, NULL indicates there is nothing + */ + SAFE_FREE(info.matches[0]); + SAFE_FREE(info.matches); + TALLOC_FREE(ctx); + return NULL; + } + + if (info.count == 2) { + /* + * Exactly one match in matches[1], indicate this is the one + * in matches[0]. + */ + info.matches[0] = info.matches[1]; + info.matches[1] = NULL; + info.count -= 1; + TALLOC_FREE(ctx); + return info.matches; + } + + /* + * We got more than one possible match, set the result to the maximum + * common subset + */ + + info.matches[0] = SMB_STRNDUP(info.matches[1], info.samelen); + info.matches[info.count] = NULL; + TALLOC_FREE(ctx); + return info.matches; + +cleanup: + for (i = 0; i < info.count; i++) { + SAFE_FREE(info.matches[i]); + } + SAFE_FREE(info.matches); + SAFE_FREE(info.dirmask); + TALLOC_FREE(ctx); + 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, buf, 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; + size_t i, len, samelen = 0, count=1; + + matches = SMB_MALLOC_ARRAY(char *, 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] = SMB_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] = SMB_STRDUP(matches[1]); + break; + default: + matches[0] = (char *)SMB_MALLOC(samelen+1); + if (!matches[0]) + goto cleanup; + strncpy(matches[0], matches[1], samelen); + matches[0][samelen] = 0; + } + matches[count] = NULL; + return matches; + +cleanup: + for (i = 0; i < count; i++) + free(matches[i]); + + free(matches); + return NULL; + } +} + +static bool finished; + +/**************************************************************************** + Make sure we swallow keepalives during idle time. +****************************************************************************/ + +static void readline_callback(void) +{ + static time_t last_t; + struct timespec now; + time_t t; + NTSTATUS status; + unsigned char garbage[16]; + + clock_gettime_mono(&now); + t = now.tv_sec; + + if (t - last_t < 5) + return; + + last_t = t; + + /* Ping the server to keep the connection alive using SMBecho. */ + memset(garbage, 0xf0, sizeof(garbage)); + status = cli_echo(cli, 1, data_blob_const(garbage, sizeof(garbage))); + if (NT_STATUS_IS_OK(status) || + NT_STATUS_EQUAL(status, NT_STATUS_INVALID_PARAMETER)) { + /* + * Even if server returns NT_STATUS_INVALID_PARAMETER + * it still responded. + * BUG: https://bugzilla.samba.org/show_bug.cgi?id=13007 + */ + return; + } + + if (!cli_state_is_connected(cli)) { + DEBUG(0,("SMBecho failed (%s). The connection is " + "disconnected now\n", nt_errstr(status))); + finished = true; + smb_readline_done(); + } +} + +/**************************************************************************** + Process commands on stdin. +****************************************************************************/ + +static int process_stdin(void) +{ + int rc = 0; + + if (!quiet) { + d_printf("Try \"help\" to get a list of possible commands.\n"); + } + + while (!finished) { + TALLOC_CTX *frame = talloc_stackframe(); + char *tok = NULL; + char *the_prompt = NULL; + char *line = NULL; + int i; + + /* display a prompt */ + if (asprintf(&the_prompt, "smb: %s> ", client_get_cur_dir()) < 0) { + TALLOC_FREE(frame); + break; + } + line = smb_readline(the_prompt, readline_callback, completion_fn); + SAFE_FREE(the_prompt); + if (!line) { + TALLOC_FREE(frame); + break; + } + + /* special case - first char is ! */ + if (*line == '!') { + if (system(line + 1) == -1) { + d_printf("system() command %s failed.\n", + line+1); + } + SAFE_FREE(line); + TALLOC_FREE(frame); + continue; + } + + /* and get the first part of the command */ + cmd_ptr = line; + if (!next_token_talloc(frame, &cmd_ptr,&tok,NULL)) { + TALLOC_FREE(frame); + SAFE_FREE(line); + 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); + } + SAFE_FREE(line); + TALLOC_FREE(frame); + } + return rc; +} + +/**************************************************************************** + Process commands from the client. +****************************************************************************/ + +static int process(const char *base_directory) +{ + int rc = 0; + NTSTATUS status; + struct cli_credentials *creds = samba_cmdline_get_creds(); + + status = cli_cm_open(talloc_tos(), NULL, + desthost, + service, + creds, + have_ip ? &dest_ss : NULL, port, + name_type, &cli); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + + cli_set_timeout(cli, io_timeout*1000); + + if (base_directory && *base_directory) { + rc = do_cd(base_directory); + if (rc) { + cli_shutdown(cli); + return rc; + } + } + + if (cmdstr) { + rc = process_command_string(cmdstr); + } else { + process_stdin(); + } + + cli_shutdown(cli); + return rc; +} + +/**************************************************************************** + Handle a -L query. +****************************************************************************/ + +static int do_host_query(const char *query_host) +{ + NTSTATUS status; + struct cli_credentials *creds = samba_cmdline_get_creds(); + + status = cli_cm_open(talloc_tos(), NULL, + query_host, + "IPC$", + creds, + have_ip ? &dest_ss : NULL, port, + name_type, &cli); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + + cli_set_timeout(cli, io_timeout*1000); + browse_host(true); + + /* Ensure that the host can do IPv4 */ + + if (!interpret_addr(query_host)) { + struct sockaddr_storage ss; + if (interpret_string_addr(&ss, query_host, 0) && + (ss.ss_family != AF_INET)) { + d_printf("%s is an IPv6 address -- no workgroup available\n", + query_host); + return 1; + } + } + + if (lp_client_min_protocol() > PROTOCOL_NT1) { + d_printf("SMB1 disabled -- no workgroup available\n"); + goto out; + } + + if (lp_disable_netbios()) { + d_printf("NetBIOS over TCP disabled -- no workgroup available\n"); + goto out; + } + + if (port != NBT_SMB_PORT || + smbXcli_conn_protocol(cli->conn) > PROTOCOL_NT1) + { + /* + * Workgroups simply don't make sense over anything + * else but port 139 and SMB1. + */ + + cli_shutdown(cli); + d_printf("Reconnecting with SMB1 for workgroup listing.\n"); + lp_set_cmdline("client max protocol", "NT1"); + status = cli_cm_open(talloc_tos(), NULL, + query_host, + "IPC$", + creds, + have_ip ? &dest_ss : NULL, NBT_SMB_PORT, + name_type, &cli); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Unable to connect with SMB1 " + "-- no workgroup available\n"); + return 0; + } + } + + cli_set_timeout(cli, io_timeout*1000); + list_servers(lp_workgroup()); +out: + cli_shutdown(cli); + + return(0); +} + +/**************************************************************************** + Handle a tar operation. +****************************************************************************/ + +static int do_tar_op(const char *base_directory) +{ + struct tar *tar_ctx = tar_get_ctx(); + int ret = 0; + struct cli_credentials *creds = samba_cmdline_get_creds(); + + /* do we already have a connection? */ + if (!cli) { + NTSTATUS status; + + status = cli_cm_open(talloc_tos(), NULL, + desthost, + service, + creds, + have_ip ? &dest_ss : NULL, port, + name_type, &cli); + if (!NT_STATUS_IS_OK(status)) { + ret = 1; + goto out; + } + cli_set_timeout(cli, io_timeout*1000); + } + + recurse = true; + + if (base_directory && *base_directory) { + ret = do_cd(base_directory); + if (ret) { + goto out_cli; + } + } + + ret = tar_process(tar_ctx); + + out_cli: + cli_shutdown(cli); + out: + return ret; +} + +/**************************************************************************** + Handle a message operation. +****************************************************************************/ + +static int do_message_op(struct cli_credentials *creds) +{ + NTSTATUS status; + + if (lp_disable_netbios()) { + d_printf("NetBIOS over TCP disabled.\n"); + return 1; + } + + status = cli_connect_nb(desthost, have_ip ? &dest_ss : NULL, + port ? port : NBT_SMB_PORT, name_type, + lp_netbios_name(), + SMB_SIGNING_OFF, + 0, + &cli); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Connection to %s failed. Error %s\n", desthost, nt_errstr(status)); + return 1; + } + + cli_set_timeout(cli, io_timeout*1000); + send_message(cli_credentials_get_username(creds)); + cli_shutdown(cli); + + return 0; +} + +/**************************************************************************** + main program +****************************************************************************/ + +int main(int argc,char *argv[]) +{ + const char **const_argv = discard_const_p(const char *, argv); + char *base_directory = NULL; + int opt; + char *query_host = NULL; + bool message = false; + static const char *new_name_resolve_order = NULL; + poptContext pc; + char *p; + int rc = 0; + bool tar_opt = false; + bool service_opt = false; + struct tar *tar_ctx = tar_get_ctx(); + bool ok; + + struct poptOption long_options[] = { + POPT_AUTOHELP + + { + .longName = "message", + .shortName = 'M', + .argInfo = POPT_ARG_STRING, + .arg = NULL, + .val = 'M', + .descrip = "Send message", + .argDescrip = "HOST", + }, + { + .longName = "ip-address", + .shortName = 'I', + .argInfo = POPT_ARG_STRING, + .arg = NULL, + .val = 'I', + .descrip = "Use this IP to connect to", + .argDescrip = "IP", + }, + { + .longName = "stderr", + .shortName = 'E', + .argInfo = POPT_ARG_NONE, + .arg = NULL, + .val = 'E', + .descrip = "Write messages to stderr instead of stdout", + }, + { + .longName = "list", + .shortName = 'L', + .argInfo = POPT_ARG_STRING, + .arg = NULL, + .val = 'L', + .descrip = "Get a list of shares available on a host", + .argDescrip = "HOST", + }, + { + .longName = "tar", + .shortName = 'T', + .argInfo = POPT_ARG_STRING, + .arg = NULL, + .val = 'T', + .descrip = "Command line tar", + .argDescrip = "<c|x>IXFvgbNan", + }, + { + .longName = "directory", + .shortName = 'D', + .argInfo = POPT_ARG_STRING, + .arg = NULL, + .val = 'D', + .descrip = "Start from directory", + .argDescrip = "DIR", + }, + { + .longName = "command", + .shortName = 'c', + .argInfo = POPT_ARG_STRING, + .arg = &cmdstr, + .val = 'c', + .descrip = "Execute semicolon separated commands", + }, + { + .longName = "send-buffer", + .shortName = 'b', + .argInfo = POPT_ARG_INT, + .arg = &io_bufsize, + .val = 'b', + .descrip = "Changes the transmit/send buffer", + .argDescrip = "BYTES", + }, + { + .longName = "timeout", + .shortName = 't', + .argInfo = POPT_ARG_INT, + .arg = &io_timeout, + .val = 'b', + .descrip = "Changes the per-operation timeout", + .argDescrip = "SECONDS", + }, + { + .longName = "port", + .shortName = 'p', + .argInfo = POPT_ARG_INT, + .arg = &port, + .val = 'p', + .descrip = "Port to connect to", + .argDescrip = "PORT", + }, + { + .longName = "grepable", + .shortName = 'g', + .argInfo = POPT_ARG_NONE, + .arg = NULL, + .val = 'g', + .descrip = "Produce grepable output", + }, + { + .longName = "quiet", + .shortName = 'q', + .argInfo = POPT_ARG_NONE, + .arg = NULL, + .val = 'q', + .descrip = "Suppress help message", + }, + { + .longName = "browse", + .shortName = 'B', + .argInfo = POPT_ARG_NONE, + .arg = NULL, + .val = 'B', + .descrip = "Browse SMB servers using DNS", + }, + POPT_COMMON_SAMBA + POPT_COMMON_CONNECTION + POPT_COMMON_CREDENTIALS + POPT_LEGACY_S3 + POPT_COMMON_VERSION + POPT_TABLEEND + }; + TALLOC_CTX *frame = talloc_stackframe(); + struct cli_credentials *creds = NULL; + + if (!client_set_cur_dir("\\")) { + exit(ENOMEM); + } + + smb_init_locale(); + + ok = samba_cmdline_init(frame, + SAMBA_CMDLINE_CONFIG_CLIENT, + false /* require_smbconf */); + if (!ok) { + DBG_ERR("Failed to init cmdline parser!\n"); + exit(ENOMEM); + } + lp_set_cmdline("log level", "1"); + + /* skip argv(0) */ + pc = samba_popt_get_context(getprogname(), + argc, + const_argv, + long_options, + 0); + if (pc == NULL) { + DBG_ERR("Failed to setup popt context!\n"); + exit(1); + } + + poptSetOtherOptionHelp(pc, "[OPTIONS] service <password>"); + + creds = samba_cmdline_get_creds(); + while ((opt = poptGetNextOpt(pc)) != -1) { + + /* + * if the tar option has been called previously, now + * we need to eat out the leftovers + */ + /* I see no other way to keep things sane --SSS */ + if (tar_opt == true) { + while (poptPeekArg(pc)) { + poptGetArg(pc); + } + tar_opt = false; + } + + /* if the service has not yet been specified lets see if it is available in the popt stack */ + if (!service_opt && poptPeekArg(pc)) { + service = talloc_strdup(frame, poptGetArg(pc)); + if (!service) { + exit(ENOMEM); + } + service_opt = true; + } + + /* if the service has already been retrieved then check if we have also a password */ + if (service_opt && + cli_credentials_get_password(creds) == NULL && + poptPeekArg(pc)) { + cli_credentials_set_password(creds, + poptGetArg(pc), + CRED_SPECIFIED); + } + + + 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; + desthost = talloc_strdup(frame,poptGetOptArg(pc)); + if (!desthost) { + exit(ENOMEM); + } + if( !port ) + port = NBT_SMB_PORT; + message = true; + break; + case 'I': + { + if (!interpret_string_addr(&dest_ss, poptGetOptArg(pc), 0)) { + exit(1); + } + have_ip = true; + print_sockaddr(dest_ss_str, sizeof(dest_ss_str), &dest_ss); + } + break; + case 'E': + setup_logging("smbclient", DEBUG_STDERR ); + display_set_stderr(); + break; + + case 'L': + query_host = talloc_strdup(frame, poptGetOptArg(pc)); + if (!query_host) { + exit(ENOMEM); + } + break; + case 'T': + /* We must use old option processing for this. Find the + * position of the -T option in the raw argv[]. */ + { + int i; + + for (i = 1; i < argc; i++) { + if (strncmp("-T", argv[i],2)==0) + break; + } + i++; + if (tar_parse_args(tar_ctx, poptGetOptArg(pc), + const_argv + i, argc - i)) { + poptPrintUsage(pc, stderr, 0); + exit(1); + } + } + /* this must be the last option, mark we have parsed it so that we know we have */ + tar_opt = true; + break; + case 'D': + base_directory = talloc_strdup(frame, poptGetOptArg(pc)); + if (!base_directory) { + exit(ENOMEM); + } + break; + case 'g': + grepable=true; + break; + case 'q': + quiet=true; + break; + case 'B': + return(do_smb_browse()); + case POPT_ERROR_BADOPT: + fprintf(stderr, "\nInvalid option %s: %s\n\n", + poptBadOption(pc, 0), poptStrerror(opt)); + poptPrintUsage(pc, stderr, 0); + exit(1); + } + } + + /* We may still have some leftovers after the last popt option has been called */ + if (tar_opt == true) { + while (poptPeekArg(pc)) { + poptGetArg(pc); + } + tar_opt = false; + } + + /* if the service has not yet been specified lets see if it is available in the popt stack */ + if (!service_opt && poptPeekArg(pc)) { + service = talloc_strdup(frame,poptGetArg(pc)); + if (!service) { + exit(ENOMEM); + } + service_opt = true; + } + + /* if the service has already been retrieved then check if we have also a password */ + if (service_opt && + cli_credentials_get_password(creds) == NULL && + poptPeekArg(pc)) { + cli_credentials_set_password(creds, + poptGetArg(pc), + CRED_SPECIFIED); + } + + if (service_opt && service) { + size_t len; + + /* 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); + } + /* Remove trailing slashes */ + len = strlen(service); + while(len > 0 && service[len - 1] == '\\') { + --len; + service[len] = '\0'; + } + } + + if(new_name_resolve_order) + lp_set_cmdline("name resolve order", new_name_resolve_order); + + if (!tar_to_process(tar_ctx) && !query_host && !service && !message) { + poptPrintUsage(pc, stderr, 0); + exit(1); + } + + poptFreeContext(pc); + samba_cmdline_burn(argc, argv); + + DEBUG(3,("Client started (version %s).\n", samba_version_string())); + + if (tar_to_process(tar_ctx)) { + if (cmdstr) + process_command_string(cmdstr); + rc = do_tar_op(base_directory); + } else if (query_host && *query_host) { + char *qhost = query_host; + char *slash; + + while (*qhost == '\\' || *qhost == '/') + qhost++; + + if ((slash = strchr_m(qhost, '/')) + || (slash = strchr_m(qhost, '\\'))) { + *slash = 0; + } + + if ((p=strchr_m(qhost, '#'))) { + *p = 0; + p++; + sscanf(p, "%x", &name_type); + } + + rc = do_host_query(qhost); + } else if (message) { + rc = do_message_op(creds); + } else if (process(base_directory)) { + rc = 1; + } + + TALLOC_FREE(frame); + return rc; +} diff --git a/source3/client/client_proto.h b/source3/client/client_proto.h new file mode 100644 index 0000000..0c83ba9 --- /dev/null +++ b/source3/client/client_proto.h @@ -0,0 +1,53 @@ +/* + * Unix SMB/CIFS implementation. + * collected prototypes header + * + * frozen from "make proto" in May 2008 + * + * Copyright (C) Michael Adam 2008 + * + * 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 3 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, see <http://www.gnu.org/licenses/>. + */ + +#ifndef _CLIENT_PROTO_H_ +#define _CLIENT_PROTO_H_ + +struct cli_state; +struct file_info; + +enum { + ATTR_UNSET, + ATTR_SET, +}; + +/* The following definitions come from client/client.c */ + +const char *client_get_cur_dir(void); +const char *client_set_cur_dir(const char *newdir); +char *client_clean_name(TALLOC_CTX *ctx, const char *name); +NTSTATUS do_list(const char *mask, + uint32_t attribute, + NTSTATUS (*fn)(struct cli_state *cli_state, struct file_info *, + const char *dir), + bool rec, + bool dirs); +int set_remote_attr(const char *filename, uint32_t new_attr, int mode); +int cmd_iosize(void); + +/* The following definitions come from client/dnsbrowse.c */ + +int do_smb_browse(void); +int do_smb_browse(void); + +#endif /* _CLIENT_PROTO_H_ */ diff --git a/source3/client/clitar.c b/source3/client/clitar.c new file mode 100644 index 0000000..a5c7df9 --- /dev/null +++ b/source3/client/clitar.c @@ -0,0 +1,1903 @@ +/* + Unix SMB/CIFS implementation. + Tar backup command extension + Copyright (C) Aurélien Aptel 2013 + + 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 3 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, see <http://www.gnu.org/licenses/>. +*/ + +/** + * # General overview of the tar extension + * + * All tar_xxx() functions work on a `struct tar` which store most of + * the context of the backup process. + * + * The current tar context can be accessed via the global variable + * `tar_ctx`. It's publicly exported as an opaque handle via + * tar_get_ctx(). + * + * A tar context is first configured through tar_parse_args() which + * can be called from either the CLI (in client.c) or the interactive + * session (via the cmd_tar() callback). + * + * Once the configuration is done (successfully), the context is ready + * for processing and tar_to_process() returns true. + * + * The next step is to call tar_process() which dispatch the + * processing to either tar_create() or tar_extract(), depending on + * the context. + * + * ## Archive creation + * + * tar_create() creates an archive using the libarchive API then + * + * - iterates on the requested paths if the context is in inclusion + * mode with tar_create_from_list() + * + * - or iterates on the whole share (starting from the current dir) if + * in exclusion mode or if no specific path were requested + * + * The do_list() function from client.c is used to list recursively + * the share. In particular it takes a DOS path mask (eg. \mydir\*) + * and a callback function which will be called with each file name + * and attributes. The tar callback function is get_file_callback(). + * + * The callback function checks whether the file should be skipped + * according the the configuration via tar_create_skip_path(). If it's + * not skipped it's downloaded and written to the archive in + * tar_get_file(). + * + * ## Archive extraction + * + * tar_extract() opens the archive and iterates on each file in + * it. For each file tar_extract_skip_path() checks whether it should + * be skipped according to the config. If it's not skipped it's + * uploaded on the server in tar_send_file(). + */ + +#include "includes.h" +#include "system/filesys.h" +#include "client/client_proto.h" +#include "client/clitar_proto.h" +#include "libsmb/libsmb.h" + +#ifdef HAVE_LIBARCHIVE + +#include <archive.h> +#include <archive_entry.h> + +/* prepend module name and line number to debug messages */ +#define DBG(a, b) (DEBUG(a, ("tar:%-4d ", __LINE__)), DEBUG(a, b)) + +/* preprocessor magic to stringify __LINE__ (int) */ +#define STR1(x) #x +#define STR2(x) STR1(x) + +/** + * Number of byte in a block unit. + */ +#define TAR_BLOCK_UNIT 512 + +/** + * Default tar block size in TAR_BLOCK_UNIT. + */ +#define TAR_DEFAULT_BLOCK_SIZE 20 + +/** + * Maximum value for the blocksize field + */ +#define TAR_MAX_BLOCK_SIZE 0xffff + +/** + * Size of the buffer used when downloading a file + */ +#define TAR_CLI_READ_SIZE 0xff00 + +#define TAR_DO_LIST_ATTR (FILE_ATTRIBUTE_DIRECTORY \ + | FILE_ATTRIBUTE_SYSTEM \ + | FILE_ATTRIBUTE_HIDDEN) + + +enum tar_operation { + TAR_NO_OPERATION, + TAR_CREATE, /* c flag */ + TAR_EXTRACT, /* x flag */ +}; + +enum tar_selection { + TAR_NO_SELECTION, + TAR_INCLUDE, /* I and F flag, default */ + TAR_EXCLUDE, /* X flag */ +}; + +struct tar { + TALLOC_CTX *talloc_ctx; + + /* in state that needs/can be processed? */ + bool to_process; + + /* flags */ + struct tar_mode { + enum tar_operation operation; /* create, extract */ + enum tar_selection selection; /* include, exclude */ + int blocksize; /* size in TAR_BLOCK_UNIT of a tar file block */ + bool hidden; /* backup hidden file? */ + bool system; /* backup system file? */ + bool incremental; /* backup _only_ archived file? */ + bool reset; /* unset archive bit? */ + bool dry; /* don't write tar file? */ + bool regex; /* XXX: never actually using regex... */ + bool verbose; /* XXX: ignored */ + } mode; + + /* nb of bytes received */ + uint64_t total_size; + + /* path to tar archive name */ + char *tar_path; + + /* list of path to include or exclude */ + char **path_list; + int path_list_size; + + /* archive handle */ + struct archive *archive; + + /* counters */ + uint64_t numdir; + uint64_t numfile; +}; + +/** + * Global context imported in client.c when needed. + * + * Default options. + */ +struct tar tar_ctx = { + .mode.selection = TAR_INCLUDE, + .mode.blocksize = TAR_DEFAULT_BLOCK_SIZE, + .mode.hidden = true, + .mode.system = true, + .mode.incremental = false, + .mode.reset = false, + .mode.dry = false, + .mode.regex = false, + .mode.verbose = false, +}; + +/* tar, local function */ +static int tar_create(struct tar* t); +static int tar_create_from_list(struct tar *t); +static int tar_extract(struct tar *t); +static int tar_read_inclusion_file(struct tar *t, const char* filename); +static int tar_send_file(struct tar *t, struct archive_entry *entry); +static int tar_set_blocksize(struct tar *t, int size); +static int tar_set_newer_than(struct tar *t, const char *filename); +static NTSTATUS tar_add_selection_path(struct tar *t, const char *path); +static void tar_dump(struct tar *t); +static NTSTATUS tar_extract_skip_path(struct tar *t, + struct archive_entry *entry, + bool *_skip); +static TALLOC_CTX *tar_reset_mem_context(struct tar *t); +static void tar_free_mem_context(struct tar *t); +static NTSTATUS tar_create_skip_path(struct tar *t, + const char *fullpath, + const struct file_info *finfo, + bool *_skip); + +static NTSTATUS tar_path_in_list(struct tar *t, const char *path, + bool reverse, bool *_is_in_list); + +static int tar_get_file(struct tar *t, + const char *full_dos_path, + struct file_info *finfo); + +static NTSTATUS get_file_callback(struct cli_state *cli, + struct file_info *finfo, + const char *dir); + +/* utilities */ +static char *fix_unix_path(char *path, bool removeprefix); +static NTSTATUS path_base_name(TALLOC_CTX *ctx, const char *path, char **_base); +static const char* skip_useless_char_in_path(const char *p); +static int make_remote_path(const char *full_path); +static int max_token (const char *str); +static NTSTATUS is_subpath(const char *sub, const char *full, + bool *_subpath_match); + /* + * tar_get_ctx - retrieve global tar context handle + */ +struct tar *tar_get_ctx(void) +{ + return &tar_ctx; +} + +/** + * cmd_block - interactive command to change tar blocksize + * + * Read a size from the client command line and update the current + * blocksize. + */ +int cmd_block(void) +{ + /* XXX: from client.c */ + const extern char *cmd_ptr; + char *buf; + int err = 0; + bool ok; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + ok = next_token_talloc(ctx, &cmd_ptr, &buf, NULL); + if (!ok) { + DBG(0, ("blocksize <n>\n")); + err = 1; + goto out; + } + + ok = tar_set_blocksize(&tar_ctx, atoi(buf)); + if (ok) { + DBG(0, ("invalid blocksize\n")); + err = 1; + goto out; + } + + DBG(2, ("blocksize is now %d\n", tar_ctx.mode.blocksize)); + +out: + talloc_free(ctx); + return err; +} + +/** + * cmd_tarmode - interactive command to change tar behaviour + * + * Read one or more modes from the client command line and update the + * current tar mode. + */ +int cmd_tarmode(void) +{ + const extern char *cmd_ptr; + char *buf; + TALLOC_CTX *ctx; + + struct { + const char *cmd; + bool *p; + bool value; + } table[] = { + {"full", &tar_ctx.mode.incremental, false}, + {"inc", &tar_ctx.mode.incremental, true }, + {"reset", &tar_ctx.mode.reset, true }, + {"noreset", &tar_ctx.mode.reset, false}, + {"system", &tar_ctx.mode.system, true }, + {"nosystem", &tar_ctx.mode.system, false}, + {"hidden", &tar_ctx.mode.hidden, true }, + {"nohidden", &tar_ctx.mode.hidden, false}, + {"verbose", &tar_ctx.mode.verbose, false}, + {"noverbose", &tar_ctx.mode.verbose, true }, + }; + + ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + while (next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) { + size_t i; + for (i = 0; i < ARRAY_SIZE(table); i++) { + if (strequal(table[i].cmd, buf)) { + *table[i].p = table[i].value; + break; + } + } + + if (i == ARRAY_SIZE(table)) + d_printf("tarmode: unrecognised option %s\n", buf); + } + + d_printf("tarmode is now %s, %s, %s, %s, %s\n", + tar_ctx.mode.incremental ? "incremental" : "full", + tar_ctx.mode.system ? "system" : "nosystem", + tar_ctx.mode.hidden ? "hidden" : "nohidden", + tar_ctx.mode.reset ? "reset" : "noreset", + tar_ctx.mode.verbose ? "verbose" : "noverbose"); + + talloc_free(ctx); + return 0; +} + +/** + * cmd_tar - interactive command to start a tar backup/restoration + * + * Check presence of argument, parse them and handle the request. + */ +int cmd_tar(void) +{ + const extern char *cmd_ptr; + const char *flag; + const char **val; + char *buf; + int maxtok = max_token(cmd_ptr); + int i = 0; + int err = 0; + bool ok; + int rc; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + ok = next_token_talloc(ctx, &cmd_ptr, &buf, NULL); + if (!ok) { + d_printf("tar <c|x>[IXFbgvanN] [options] <tar file> [path list]\n"); + err = 1; + goto out; + } + + flag = buf; + val = talloc_array(ctx, const char *, maxtok); + if (val == NULL) { + err = 1; + goto out; + } + + while (next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) { + val[i++] = buf; + } + + rc = tar_parse_args(&tar_ctx, flag, val, i); + if (rc != 0) { + d_printf("parse_args failed\n"); + err = 1; + goto out; + } + + rc = tar_process(&tar_ctx); + if (rc != 0) { + d_printf("tar_process failed\n"); + err = 1; + goto out; + } + +out: + talloc_free(ctx); + return err; +} + + +/** + * tar_parse_args - parse and set tar command line arguments + * @flag: string pointing to tar options + * @val: number of tar arguments + * @valsize: table of arguments after the flags (number of element in val) + * + * tar arguments work in a weird way. For each flag f that takes a + * value v, the user is supposed to type: + * + * on the CLI: + * -Tf1f2f3 v1 v2 v3 TARFILE PATHS... + * + * in the interactive session: + * tar f1f2f3 v1 v2 v3 TARFILE PATHS... + * + * @flag has only flags (eg. "f1f2f3") and @val has the arguments + * (values) following them (eg. ["v1", "v2", "v3", "TARFILE", "PATH1", + * "PATH2"]). + * + * There are only 2 flags that take an arg: b and N. The other flags + * just change the semantic of PATH or TARFILE. + * + * PATH can be a list of included/excluded paths, the path to a file + * containing a list of included/excluded paths to use (F flag). If no + * PATH is provided, the whole share is used (/). + */ +int tar_parse_args(struct tar* t, + const char *flag, + const char **val, + int valsize) +{ + TALLOC_CTX *ctx; + bool do_read_list = false; + /* index of next value to use */ + int ival = 0; + int rc; + + if (t == NULL) { + DBG_WARNING("Invalid tar context\n"); + return 1; + } + + ctx = tar_reset_mem_context(t); + if (ctx == NULL) { + return 1; + } + /* + * Reset back some options - could be from interactive version + * all other modes are left as they are + */ + t->mode.operation = TAR_NO_OPERATION; + t->mode.selection = TAR_NO_SELECTION; + t->mode.dry = false; + t->to_process = false; + t->total_size = 0; + + while (flag[0] != '\0') { + switch(flag[0]) { + /* operation */ + case 'c': + if (t->mode.operation != TAR_NO_OPERATION) { + d_printf("Tar must be followed by only one of c or x.\n"); + return 1; + } + t->mode.operation = TAR_CREATE; + break; + case 'x': + if (t->mode.operation != TAR_NO_OPERATION) { + d_printf("Tar must be followed by only one of c or x.\n"); + return 1; + } + t->mode.operation = TAR_EXTRACT; + break; + + /* selection */ + case 'I': + if (t->mode.selection != TAR_NO_SELECTION) { + d_printf("Only one of I,X,F must be specified\n"); + return 1; + } + t->mode.selection = TAR_INCLUDE; + break; + case 'X': + if (t->mode.selection != TAR_NO_SELECTION) { + d_printf("Only one of I,X,F must be specified\n"); + return 1; + } + t->mode.selection = TAR_EXCLUDE; + break; + case 'F': + if (t->mode.selection != TAR_NO_SELECTION) { + d_printf("Only one of I,X,F must be specified\n"); + return 1; + } + t->mode.selection = TAR_INCLUDE; + do_read_list = true; + break; + + /* blocksize */ + case 'b': + if (ival >= valsize) { + d_printf("Option b must be followed by a blocksize\n"); + return 1; + } + + if (tar_set_blocksize(t, atoi(val[ival]))) { + d_printf("Option b must be followed by a valid blocksize\n"); + return 1; + } + + ival++; + break; + + /* incremental mode */ + case 'g': + t->mode.incremental = true; + break; + + /* newer than */ + case 'N': + if (ival >= valsize) { + d_printf("Option N must be followed by valid file name\n"); + return 1; + } + + if (tar_set_newer_than(t, val[ival])) { + d_printf("Error setting newer-than time\n"); + return 1; + } + + ival++; + break; + + /* reset mode */ + case 'a': + t->mode.reset = true; + break; + + /* verbose */ + case 'v': + t->mode.verbose = true; + break; + + /* regex match */ + case 'r': + t->mode.regex = true; + break; + + /* dry run mode */ + case 'n': + if (t->mode.operation != TAR_CREATE) { + d_printf("n is only meaningful when creating a tar-file\n"); + return 1; + } + + t->mode.dry = true; + d_printf("dry_run set\n"); + break; + + default: + d_printf("Unknown tar option\n"); + return 1; + } + + flag++; + } + + /* no selection given? default selection is include */ + if (t->mode.selection == TAR_NO_SELECTION) { + t->mode.selection = TAR_INCLUDE; + } + + if (valsize - ival < 1) { + d_printf("No tar file given.\n"); + return 1; + } + + /* handle TARFILE */ + t->tar_path = talloc_strdup(ctx, val[ival]); + if (t->tar_path == NULL) { + return 1; + } + ival++; + + /* + * Make sure that dbf points to stderr if we are using stdout for + * tar output + */ + if (t->mode.operation == TAR_CREATE && strequal(t->tar_path, "-")) { + setup_logging("smbclient", DEBUG_STDERR); + } + + /* handle PATHs... */ + + /* flag F -> read file list */ + if (do_read_list) { + if (valsize - ival != 1) { + d_printf("Option F must be followed by exactly one filename.\n"); + return 1; + } + + rc = tar_read_inclusion_file(t, val[ival]); + if (rc != 0) { + return 1; + } + ival++; + /* otherwise store all the PATHs on the command line */ + } else { + int i; + for (i = ival; i < valsize; i++) { + NTSTATUS status; + status = tar_add_selection_path(t, val[i]); + if (!NT_STATUS_IS_OK(status)) { + return 1; + } + } + } + + t->to_process = true; + tar_dump(t); + return 0; +} + +/** + * tar_process - start processing archive + * + * The talloc context of the fields is freed at the end of the call. + */ +int tar_process(struct tar *t) +{ + int rc = 0; + + if (t == NULL) { + DBG_WARNING("Invalid tar context\n"); + return 1; + } + + switch(t->mode.operation) { + case TAR_EXTRACT: + rc = tar_extract(t); + break; + case TAR_CREATE: + rc = tar_create(t); + break; + default: + DBG_WARNING("Invalid tar state\n"); + rc = 1; + } + + t->to_process = false; + tar_free_mem_context(t); + DBG(5, ("tar_process done, err = %d\n", rc)); + return rc; +} + +/** + * tar_create - create archive and fetch files + */ +static int tar_create(struct tar* t) +{ + int r; + int err = 0; + NTSTATUS status; + const char *mask; + struct timespec tp_start, tp_end; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + clock_gettime_mono(&tp_start); + + t->numfile = 0; + t->numdir = 0; + t->archive = archive_write_new(); + + if (!t->mode.dry) { + const int bsize = t->mode.blocksize * TAR_BLOCK_UNIT; + r = archive_write_set_bytes_per_block(t->archive, bsize); + if (r != ARCHIVE_OK) { + d_printf("Can't use a block size of %d bytes", bsize); + err = 1; + goto out; + } + + /* + * Use PAX restricted format which is not the most + * conservative choice but has useful extensions and is widely + * supported + */ + r = archive_write_set_format_pax_restricted(t->archive); + if (r != ARCHIVE_OK) { + d_printf("Can't use pax restricted format: %s\n", + archive_error_string(t->archive)); + err = 1; + goto out; + } + + if (strequal(t->tar_path, "-")) { + r = archive_write_open_fd(t->archive, STDOUT_FILENO); + } else { + r = archive_write_open_filename(t->archive, t->tar_path); + } + + if (r != ARCHIVE_OK) { + d_printf("Can't open %s: %s\n", t->tar_path, + archive_error_string(t->archive)); + err = 1; + goto out_close; + } + } + + /* + * In inclusion mode, iterate on the inclusion list + */ + if (t->mode.selection == TAR_INCLUDE && t->path_list_size > 0) { + if (tar_create_from_list(t)) { + err = 1; + goto out_close; + } + } else { + mask = talloc_asprintf(ctx, "%s\\*", client_get_cur_dir()); + if (mask == NULL) { + err = 1; + goto out_close; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + err = 1; + goto out_close; + } + DBG(5, ("tar_process do_list with mask: %s\n", mask)); + status = do_list(mask, TAR_DO_LIST_ATTR, get_file_callback, true, true); + if (!NT_STATUS_IS_OK(status)) { + DBG(0, ("do_list fail %s\n", nt_errstr(status))); + err = 1; + goto out_close; + } + } + + clock_gettime_mono(&tp_end); + d_printf("tar: dumped %"PRIu64" files and %"PRIu64" directories\n", + t->numfile, t->numdir); + d_printf("Total bytes written: %"PRIu64" (%.1f MiB/s)\n", + t->total_size, + t->total_size/timespec_elapsed2(&tp_start, &tp_end)/1024/1024); + +out_close: + + if (!t->mode.dry) { + r = archive_write_close(t->archive); + if (r != ARCHIVE_OK) { + d_printf("Fatal: %s\n", archive_error_string(t->archive)); + err = 1; + goto out; + } + } +out: +#ifdef HAVE_ARCHIVE_READ_FREE + archive_write_free(t->archive); +#else + archive_write_finish(t->archive); +#endif + talloc_free(ctx); + return err; +} + +/** + * tar_create_from_list - fetch from path list in include mode + */ +static int tar_create_from_list(struct tar *t) +{ + int err = 0; + NTSTATUS status; + char *base; + const char *path, *mask, *start_dir; + int i; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + start_dir = talloc_strdup(ctx, client_get_cur_dir()); + if (start_dir == NULL) { + err = 1; + goto out; + } + + for (i = 0; i < t->path_list_size; i++) { + path = t->path_list[i]; + base = NULL; + status = path_base_name(ctx, path, &base); + if (!NT_STATUS_IS_OK(status)) { + err = 1; + goto out; + } + mask = talloc_asprintf(ctx, "%s\\%s", + client_get_cur_dir(), path); + if (mask == NULL) { + err = 1; + goto out; + } + mask = client_clean_name(ctx, mask); + if (mask == NULL) { + err = 1; + goto out; + } + + DBG(5, ("incl. path='%s', base='%s', mask='%s'\n", + path, base ? base : "NULL", mask)); + + if (base != NULL) { + base = talloc_asprintf(ctx, "%s%s\\", + client_get_cur_dir(), base); + if (base == NULL) { + err = 1; + goto out; + } + base = client_clean_name(ctx, base); + if (base == NULL) { + err = 1; + goto out; + } + + DBG(5, ("cd '%s' before do_list\n", base)); + client_set_cur_dir(base); + } + status = do_list(mask, TAR_DO_LIST_ATTR, get_file_callback, true, true); + if (base != NULL) { + client_set_cur_dir(start_dir); + } + if (!NT_STATUS_IS_OK(status)) { + DBG(0, ("do_list failed on %s (%s)\n", path, nt_errstr(status))); + err = 1; + goto out; + } + } + +out: + talloc_free(ctx); + return err; +} + +/** + * get_file_callback - do_list callback + * + * Callback for client.c do_list(). Called for each file found on the + * share matching do_list mask. Recursively call do_list() with itself + * as callback when the current file is a directory. + */ +static NTSTATUS get_file_callback(struct cli_state *cli, + struct file_info *finfo, + const char *dir) +{ + NTSTATUS status = NT_STATUS_OK; + char *remote_name; + char *old_dir = NULL; + char *new_dir = NULL; + const char *initial_dir = dir; + bool skip = false; + bool isdir; + int rc; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return NT_STATUS_NO_MEMORY; + } + + remote_name = talloc_asprintf(ctx, "%s\\%s", initial_dir, finfo->name); + if (remote_name == NULL) { + status = NT_STATUS_NO_MEMORY; + goto out; + } + remote_name = client_clean_name(ctx, remote_name); + if (remote_name == NULL) { + status = NT_STATUS_NO_MEMORY; + goto out; + } + + if (strequal(finfo->name, "..") || strequal(finfo->name, ".")) { + goto out; + } + + isdir = finfo->attr & FILE_ATTRIBUTE_DIRECTORY; + if (isdir) { + old_dir = talloc_strdup(ctx, initial_dir); + new_dir = talloc_asprintf(ctx, "%s\\", remote_name); + if ((old_dir == NULL) || (new_dir == NULL)) { + status = NT_STATUS_NO_MEMORY; + goto out; + } + } + + status = tar_create_skip_path(&tar_ctx, + isdir ? new_dir : remote_name, + finfo, &skip); + if (!NT_STATUS_IS_OK(status)) { + goto out; + } + + if (skip) { + DBG(5, ("--- %s\n", remote_name)); + status = NT_STATUS_OK; + goto out; + } + + rc = tar_get_file(&tar_ctx, remote_name, finfo); + if (rc != 0) { + status = NT_STATUS_UNSUCCESSFUL; + goto out; + } + +out: + talloc_free(ctx); + return status; +} + +/** + * tar_get_file - fetch a remote file to the local archive + * @full_dos_path: path to the file to fetch + * @finfo: attributes of the file to fetch + */ +static int tar_get_file(struct tar *t, + const char *full_dos_path, + struct file_info *finfo) +{ + extern struct cli_state *cli; + NTSTATUS status; + struct archive_entry *entry; + char *full_unix_path; + char buf[TAR_CLI_READ_SIZE]; + size_t len; + uint64_t off = 0; + uint16_t remote_fd = (uint16_t)-1; + int err = 0, r; + const bool isdir = finfo->attr & FILE_ATTRIBUTE_DIRECTORY; + TALLOC_CTX *ctx = talloc_new(NULL); + + if (ctx == NULL) { + return 1; + } + + DBG(5, ("+++ %s\n", full_dos_path)); + + t->total_size += finfo->size; + + if (t->mode.dry) { + goto out; + } + + if (t->mode.reset) { + /* ignore return value: server might not store DOS attributes */ + set_remote_attr(full_dos_path, FILE_ATTRIBUTE_ARCHIVE, ATTR_UNSET); + } + + full_unix_path = talloc_asprintf(ctx, ".%s", full_dos_path); + if (full_unix_path == NULL) { + err = 1; + goto out; + } + string_replace(full_unix_path, '\\', '/'); + entry = archive_entry_new(); + archive_entry_copy_pathname(entry, full_unix_path); + archive_entry_set_filetype(entry, isdir ? AE_IFDIR : AE_IFREG); + archive_entry_set_atime(entry, + finfo->atime_ts.tv_sec, + finfo->atime_ts.tv_nsec); + archive_entry_set_mtime(entry, + finfo->mtime_ts.tv_sec, + finfo->mtime_ts.tv_nsec); + archive_entry_set_ctime(entry, + finfo->ctime_ts.tv_sec, + finfo->ctime_ts.tv_nsec); + archive_entry_set_perm(entry, isdir ? 0755 : 0644); + /* + * check if we can safely cast unsigned file size to libarchive + * signed size. Very unlikely problem (>9 exabyte file) + */ + if (finfo->size > INT64_MAX) { + d_printf("Remote file %s too big\n", full_dos_path); + goto out_entry; + } + + archive_entry_set_size(entry, (int64_t)finfo->size); + + if (isdir) { + /* It's a directory just write a header */ + r = archive_write_header(t->archive, entry); + if (r != ARCHIVE_OK) { + d_printf("Fatal: %s\n", archive_error_string(t->archive)); + err = 1; + } + if (t->mode.verbose) { + d_printf("a %s\\\n", full_dos_path); + } + DBG(5, ("get_file skip dir %s\n", full_dos_path)); + goto out_entry; + } + + status = cli_open(cli, full_dos_path, O_RDONLY, DENY_NONE, &remote_fd); + if (!NT_STATUS_IS_OK(status)) { + d_printf("%s opening remote file %s\n", + nt_errstr(status), full_dos_path); + goto out_entry; + } + + /* don't make tar file entry until after the file is open */ + r = archive_write_header(t->archive, entry); + if (r != ARCHIVE_OK) { + d_printf("Fatal: %s\n", archive_error_string(t->archive)); + err = 1; + goto out_entry; + } + + if (t->mode.verbose) { + d_printf("a %s\n", full_dos_path); + } + + do { + status = cli_read(cli, remote_fd, buf, off, sizeof(buf), &len); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error reading file %s : %s\n", + full_dos_path, nt_errstr(status)); + err = 1; + goto out_close; + } + + off += len; + + r = archive_write_data(t->archive, buf, len); + if (r < 0) { + d_printf("Fatal: %s\n", archive_error_string(t->archive)); + err = 1; + goto out_close; + } + + } while (off < finfo->size); + t->numfile++; + +out_close: + cli_close(cli, remote_fd); + +out_entry: + archive_entry_free(entry); + +out: + talloc_free(ctx); + return err; +} + +/** + * tar_extract - open archive and send files. + */ +static int tar_extract(struct tar *t) +{ + int err = 0; + int r; + struct archive_entry *entry; + const size_t bsize = t->mode.blocksize * TAR_BLOCK_UNIT; + int rc; + + t->archive = archive_read_new(); + archive_read_support_format_all(t->archive); +#ifdef HAVE_ARCHIVE_READ_SUPPORT_FILTER_ALL + archive_read_support_filter_all(t->archive); +#endif + + if (strequal(t->tar_path, "-")) { + r = archive_read_open_fd(t->archive, STDIN_FILENO, bsize); + } else { + r = archive_read_open_filename(t->archive, t->tar_path, bsize); + } + + if (r != ARCHIVE_OK) { + d_printf("Can't open %s : %s\n", t->tar_path, + archive_error_string(t->archive)); + err = 1; + goto out; + } + + for (;;) { + NTSTATUS status; + bool skip = false; + r = archive_read_next_header(t->archive, &entry); + if (r == ARCHIVE_EOF) { + break; + } + if (r == ARCHIVE_WARN) { + d_printf("Warning: %s\n", archive_error_string(t->archive)); + } + if (r == ARCHIVE_FATAL) { + d_printf("Fatal: %s\n", archive_error_string(t->archive)); + err = 1; + goto out; + } + + status = tar_extract_skip_path(t, entry, &skip); + if (!NT_STATUS_IS_OK(status)) { + err = 1; + goto out; + } + if (skip) { + DBG(5, ("--- %s\n", archive_entry_pathname(entry))); + continue; + } + DBG(5, ("+++ %s\n", archive_entry_pathname(entry))); + + if (t->mode.verbose) { + d_printf("x %s\n", archive_entry_pathname(entry)); + } + + rc = tar_send_file(t, entry); + if (rc != 0) { + err = 1; + goto out; + } + } + +out: +#ifdef HAVE_ARCHIVE_READ_FREE + r = archive_read_free(t->archive); +#else + r = archive_read_finish(t->archive); +#endif + if (r != ARCHIVE_OK) { + d_printf("Can't close %s : %s\n", t->tar_path, + archive_error_string(t->archive)); + err = 1; + } + return err; +} + +/** + * tar_send_file - send @entry to the remote server + * @entry: current archive entry + * + * Handle the creation of the parent directories and transfer the + * entry to a new remote file. + */ +static int tar_send_file(struct tar *t, struct archive_entry *entry) +{ + extern struct cli_state *cli; + char *dos_path; + char *full_path; + NTSTATUS status; + uint16_t remote_fd = (uint16_t) -1; + int err = 0; + int flags = O_RDWR | O_CREAT | O_TRUNC; + mode_t filetype = archive_entry_filetype(entry); + mode_t mode = archive_entry_mode(entry); + time_t mtime = archive_entry_mtime(entry); + int rc; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + dos_path = talloc_strdup(ctx, archive_entry_pathname(entry)); + if (dos_path == NULL) { + err = 1; + goto out; + } + fix_unix_path(dos_path, true); + + full_path = talloc_strdup(ctx, client_get_cur_dir()); + if (full_path == NULL) { + err = 1; + goto out; + } + full_path = talloc_strdup_append(full_path, dos_path); + if (full_path == NULL) { + err = 1; + goto out; + } + full_path = client_clean_name(ctx, full_path); + if (full_path == NULL) { + err = 1; + goto out; + } + + if (filetype != AE_IFREG && filetype != AE_IFDIR) { + d_printf("Skipping non-dir & non-regular file %s\n", full_path); + goto out; + } + + rc = make_remote_path(full_path); + if (rc != 0) { + err = 1; + goto out; + } + + if (filetype == AE_IFDIR) { + goto out; + } + + status = cli_open(cli, full_path, flags, DENY_NONE, &remote_fd); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error opening remote file %s: %s\n", + full_path, nt_errstr(status)); + err = 1; + goto out; + } + + for (;;) { + const void *buf; + size_t len; + off_t off; + int r; + + r = archive_read_data_block(t->archive, &buf, &len, &off); + if (r == ARCHIVE_EOF) { + break; + } + if (r == ARCHIVE_WARN) { + d_printf("Warning: %s\n", archive_error_string(t->archive)); + } + if (r == ARCHIVE_FATAL) { + d_printf("Fatal: %s\n", archive_error_string(t->archive)); + err = 1; + goto close_out; + } + + status = cli_writeall(cli, remote_fd, 0, buf, off, len, NULL); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error writing remote file %s: %s\n", + full_path, nt_errstr(status)); + err = 1; + goto close_out; + } + } + +close_out: + status = cli_close(cli, remote_fd); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error closing remote file %s: %s\n", + full_path, nt_errstr(status)); + err = 1; + } + + status = cli_setatr(cli, full_path, mode, mtime); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Error setting attributes on remote file %s: %s\n", + full_path, nt_errstr(status)); + err = 1; + } + +out: + talloc_free(ctx); + return err; +} + +/** + * tar_add_selection_path - add a path to the path list + * @path: path to add + */ +static NTSTATUS tar_add_selection_path(struct tar *t, const char *path) +{ + const char **list; + TALLOC_CTX *ctx = t->talloc_ctx; + if (!t->path_list) { + t->path_list = str_list_make_empty(ctx); + if (t->path_list == NULL) { + return NT_STATUS_NO_MEMORY; + } + t->path_list_size = 0; + } + + /* cast to silence gcc const-qual warning */ + list = str_list_add((void *)t->path_list, path); + if (list == NULL) { + return NT_STATUS_NO_MEMORY; + } + t->path_list = discard_const_p(char *, list); + t->path_list_size++; + fix_unix_path(t->path_list[t->path_list_size - 1], true); + + return NT_STATUS_OK; +} + +/** + * tar_set_blocksize - set block size in TAR_BLOCK_UNIT + */ +static int tar_set_blocksize(struct tar *t, int size) +{ + if (size <= 0 || size > TAR_MAX_BLOCK_SIZE) { + return 1; + } + + t->mode.blocksize = size; + + return 0; +} + +/** + * tar_set_newer_than - set date threshold of saved files + * @filename: local path to a file + * + * Only files newer than the modification time of @filename will be + * saved. + * + * Note: this function set the global variable newer_than from + * client.c. Thus the time is not a field of the tar structure. See + * cmd_newer() to change its value from an interactive session. + */ +static int tar_set_newer_than(struct tar *t, const char *filename) +{ + extern time_t newer_than; + SMB_STRUCT_STAT stbuf; + int rc; + + rc = sys_stat(filename, &stbuf, false); + if (rc != 0) { + d_printf("Error setting newer-than time\n"); + return 1; + } + + newer_than = convert_timespec_to_time_t(stbuf.st_ex_mtime); + DBG(1, ("Getting files newer than %s\n", time_to_asc(newer_than))); + return 0; +} + +/** + * tar_read_inclusion_file - set path list from file + * @filename: path to the list file + * + * Read and add each line of @filename to the path list. + */ +static int tar_read_inclusion_file(struct tar *t, const char* filename) +{ + char *line; + int err = 0; + int fd = -1; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + fd = open(filename, O_RDONLY); + if (fd < 0) { + d_printf("Can't open inclusion file '%s': %s\n", filename, strerror(errno)); + err = 1; + goto out; + } + + for (line = afdgets(fd, ctx, 0); + line != NULL; + line = afdgets(fd, ctx, 0)) { + NTSTATUS status; + status = tar_add_selection_path(t, line); + if (!NT_STATUS_IS_OK(status)) { + err = 1; + goto out; + } + } + +out: + if (fd != -1) { + close(fd); + } + talloc_free(ctx); + return err; +} + +/** + * tar_path_in_list - check whether @path is in the path list + * @path: path to find + * @reverse: when true also try to find path list element in @path + * @_is_in_list: set if @path is in the path list + * + * Look at each path of the path list and set @_is_in_list if @path is a + * subpath of one of them. + * + * If you want /path to be in the path list (path/a/, path/b/) set + * @reverse to true to try to match the other way around. + */ +static NTSTATUS tar_path_in_list(struct tar *t, const char *path, + bool reverse, bool *_is_in_list) +{ + int i; + const char *p; + const char *pattern; + + if (path == NULL || path[0] == '\0') { + *_is_in_list = false; + return NT_STATUS_OK; + } + + p = skip_useless_char_in_path(path); + + for (i = 0; i < t->path_list_size; i++) { + bool is_in_list; + NTSTATUS status; + + pattern = skip_useless_char_in_path(t->path_list[i]); + status = is_subpath(p, pattern, &is_in_list); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + if (reverse && !is_in_list) { + status = is_subpath(pattern, p, &is_in_list); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + } + if (is_in_list) { + *_is_in_list = true; + return NT_STATUS_OK; + } + } + + *_is_in_list = false; + return NT_STATUS_OK; +} + +/** + * tar_extract_skip_path - check if @entry should be skipped + * @entry: current tar entry + * @_skip: set true if path should be skipped, otherwise false + * + * Skip predicate for tar extraction (archive to server) only. + */ +static NTSTATUS tar_extract_skip_path(struct tar *t, + struct archive_entry *entry, + bool *_skip) +{ + const char *fullpath = archive_entry_pathname(entry); + bool in = true; + + if (t->path_list_size <= 0) { + *_skip = false; + return NT_STATUS_OK; + } + + if (t->mode.regex) { + in = mask_match_list(fullpath, t->path_list, t->path_list_size, true); + } else { + NTSTATUS status = tar_path_in_list(t, fullpath, false, &in); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + } + + if (t->mode.selection == TAR_EXCLUDE) { + *_skip = in; + } else { + *_skip = !in; + } + + return NT_STATUS_OK; +} + +/** + * tar_create_skip_path - check if @fullpath shoud be skipped + * @fullpath: full remote path of the current file + * @finfo: remote file attributes + * @_skip: returned skip not + * + * Skip predicate for tar creation (server to archive) only. + */ +static NTSTATUS tar_create_skip_path(struct tar *t, + const char *fullpath, + const struct file_info *finfo, + bool *_skip) +{ + /* syntaxic sugar */ + const mode_t mode = finfo->attr; + const bool isdir = mode & FILE_ATTRIBUTE_DIRECTORY; + const bool exclude = t->mode.selection == TAR_EXCLUDE; + bool in = true; + + if (!isdir) { + + /* 1. if we don't want X and we have X, skip */ + if (!t->mode.system && (mode & FILE_ATTRIBUTE_SYSTEM)) { + *_skip = true; + return NT_STATUS_OK; + } + + if (!t->mode.hidden && (mode & FILE_ATTRIBUTE_HIDDEN)) { + *_skip = true; + return NT_STATUS_OK; + } + + /* 2. if we only want archive and it's not, skip */ + + if (t->mode.incremental && !(mode & FILE_ATTRIBUTE_ARCHIVE)) { + *_skip = true; + return NT_STATUS_OK; + } + } + + /* 3. is it in the selection list? */ + + /* + * tar_create_from_list() use the include list as a starting + * point, no need to check + */ + if (!exclude) { + *_skip = false; + return NT_STATUS_OK; + } + + /* we are now in exclude mode */ + + /* no matter the selection, no list => include everything */ + if (t->path_list_size <= 0) { + *_skip = false; + return NT_STATUS_OK; + } + + if (t->mode.regex) { + in = mask_match_list(fullpath, t->path_list, t->path_list_size, true); + } else { + bool reverse = isdir && !exclude; + NTSTATUS status = tar_path_in_list(t, fullpath, reverse, &in); + if (!NT_STATUS_IS_OK(status)) { + return status; + } + } + *_skip = in; + + return NT_STATUS_OK; +} + +/** + * tar_to_process - return true if @t is ready to be processed + * + * @t is ready if it properly parsed command line arguments. + */ +bool tar_to_process(struct tar *t) +{ + if (t == NULL) { + DBG_WARNING("Invalid tar context\n"); + return false; + } + return t->to_process; +} + +/** + * skip_useless_char_in_path - skip leading slashes/dots + * + * Skip leading slashes, backslashes and dot-slashes. + */ +static const char* skip_useless_char_in_path(const char *p) +{ + while (p) { + if (*p == '/' || *p == '\\') { + p++; + } + else if (p[0] == '.' && (p[1] == '/' || p[1] == '\\')) { + p += 2; + } + else + return p; + } + return p; +} + +/** + * is_subpath - check if the path @sub is a subpath of @full. + * @sub: path to test + * @full: container path + * @_subpath_match: set true if @sub is a subpath of @full, otherwise false + * + * String comparaison is case-insensitive. + */ +static NTSTATUS is_subpath(const char *sub, const char *full, + bool *_subpath_match) +{ + NTSTATUS status = NT_STATUS_OK; + int len = 0; + char *f, *s; + TALLOC_CTX *tmp_ctx = talloc_new(NULL); + if (tmp_ctx == NULL) { + status = NT_STATUS_NO_MEMORY; + goto out; + } + + f = strlower_talloc(tmp_ctx, full); + if (f == NULL) { + status = NT_STATUS_NO_MEMORY; + goto out_ctx_free; + } + string_replace(f, '\\', '/'); + s = strlower_talloc(tmp_ctx, sub); + if (s == NULL) { + status = NT_STATUS_NO_MEMORY; + goto out_ctx_free; + } + string_replace(s, '\\', '/'); + + /* find the point where sub and full diverge */ + while ((*f != '\0') && (*s != '\0') && (*f == *s)) { + f++; + s++; + len++; + } + + if ((*f == '\0') && (*s == '\0')) { + *_subpath_match = true; /* sub and full match */ + goto out_ctx_free; + } + + if ((*f == '\0') && (len > 0) && (*(f - 1) == '/')) { + /* sub diverges from full at path separator */ + *_subpath_match = true; + goto out_ctx_free; + } + + if ((*s == '\0') && (strcmp(f, "/") == 0)) { + /* full diverges from sub with trailing slash only */ + *_subpath_match = true; + goto out_ctx_free; + } + + if ((*s == '/') && (*f == '\0')) { + /* sub diverges from full with extra path component */ + *_subpath_match = true; + goto out_ctx_free; + } + *_subpath_match = false; + +out_ctx_free: + talloc_free(tmp_ctx); +out: + return status; +} + + +/** + * make_remote_path - recursively make remote dirs + * @full_path: full hierarchy to create + * + * Create @full_path and each parent directories as needed. + */ +static int make_remote_path(const char *full_path) +{ + extern struct cli_state *cli; + char *path; + char *subpath; + char *state; + char *last_backslash; + char *p; + int len; + NTSTATUS status; + int err = 0; + TALLOC_CTX *ctx = talloc_new(NULL); + if (ctx == NULL) { + return 1; + } + + subpath = talloc_strdup(ctx, full_path); + if (subpath == NULL) { + err = 1; + goto out; + } + path = talloc_strdup(ctx, full_path); + if (path == NULL) { + err = 1; + goto out; + } + len = talloc_get_size(path) - 1; + + last_backslash = strrchr_m(path, '\\'); + if (last_backslash == NULL) { + goto out; + } + + *last_backslash = 0; + + subpath[0] = 0; + p = strtok_r(path, "\\", &state); + + while (p != NULL) { + strlcat(subpath, p, len); + status = cli_chkpath(cli, subpath); + if (!NT_STATUS_IS_OK(status)) { + status = cli_mkdir(cli, subpath); + if (!NT_STATUS_IS_OK(status)) { + d_printf("Can't mkdir %s: %s\n", subpath, nt_errstr(status)); + err = 1; + goto out; + } + DBG(3, ("mkdir %s\n", subpath)); + } + + strlcat(subpath, "\\", len); + p = strtok_r(NULL, "/\\", &state); + + } + +out: + talloc_free(ctx); + return err; +} + +/** + * tar_reset_mem_context - reset talloc context associated with @t + * + * At the start of the program the context is NULL so a new one is + * allocated. On the following runs (interactive session only), simply + * free the children. + */ +static TALLOC_CTX *tar_reset_mem_context(struct tar *t) +{ + tar_free_mem_context(t); + t->talloc_ctx = talloc_new(NULL); + return t->talloc_ctx; +} + +/** + * tar_free_mem_context - free talloc context associated with @t + */ +static void tar_free_mem_context(struct tar *t) +{ + if (t->talloc_ctx) { + talloc_free(t->talloc_ctx); + t->talloc_ctx = NULL; + t->path_list_size = 0; + t->path_list = NULL; + t->tar_path = NULL; + } +} + +#define XSET(v) [v] = #v +#define XTABLE(v, t) DBG(2, ("DUMP:%-20.20s = %s\n", #v, t[v])) +#define XBOOL(v) DBG(2, ("DUMP:%-20.20s = %d\n", #v, v ? 1 : 0)) +#define XSTR(v) DBG(2, ("DUMP:%-20.20s = %s\n", #v, v ? v : "NULL")) +#define XINT(v) DBG(2, ("DUMP:%-20.20s = %d\n", #v, v)) +#define XUINT64(v) DBG(2, ("DUMP:%-20.20s = %" PRIu64 "\n", #v, v)) + +/** + * tar_dump - dump tar structure on stdout + */ +static void tar_dump(struct tar *t) +{ + int i; + const char* op[] = { + XSET(TAR_NO_OPERATION), + XSET(TAR_CREATE), + XSET(TAR_EXTRACT), + }; + + const char* sel[] = { + XSET(TAR_NO_SELECTION), + XSET(TAR_INCLUDE), + XSET(TAR_EXCLUDE), + }; + + XBOOL(t->to_process); + XTABLE(t->mode.operation, op); + XTABLE(t->mode.selection, sel); + XINT(t->mode.blocksize); + XBOOL(t->mode.hidden); + XBOOL(t->mode.system); + XBOOL(t->mode.incremental); + XBOOL(t->mode.reset); + XBOOL(t->mode.dry); + XBOOL(t->mode.verbose); + XUINT64(t->total_size); + XSTR(t->tar_path); + XINT(t->path_list_size); + + for (i = 0; t->path_list && t->path_list[i]; i++) { + DBG(2, ("DUMP: t->path_list[%2d] = %s\n", i, t->path_list[i])); + } + + DBG(2, ("DUMP:t->path_list @ %p (%d elem)\n", t->path_list, i)); +} +#undef XSET +#undef XTABLE +#undef XBOOL +#undef XSTR +#undef XINT + +/** + * max_token - return upper limit for the number of token in @str + * + * The result is not exact, the actual number of token might be less + * than what is returned. + */ +static int max_token(const char *str) +{ + const char *s; + int nb = 0; + + if (str == NULL) { + return 0; + } + + s = str; + while (s[0] != '\0') { + if (isspace((int)s[0])) { + nb++; + } + s++; + } + + nb++; + + return nb; +} + +/** + * fix_unix_path - convert @path to a DOS path + * @path: path to convert + * @removeprefix: if true, remove leading ./ or /. + */ +static char *fix_unix_path(char *path, bool do_remove_prefix) +{ + char *from = path, *to = path; + + if (path == NULL || path[0] == '\0') { + return path; + } + + /* remove prefix: + * ./path => path + * /path => path + */ + if (do_remove_prefix) { + /* /path */ + if (path[0] == '/' || path[0] == '\\') { + from += 1; + } + + /* ./path */ + if (path[1] != '\0' && path[0] == '.' && (path[1] == '/' || path[1] == '\\')) { + from += 2; + } + } + + /* replace / with \ */ + while (from[0] != '\0') { + if (from[0] == '/') { + to[0] = '\\'; + } else { + to[0] = from[0]; + } + + from++; + to++; + } + to[0] = '\0'; + + return path; +} + +/** + * path_base_name - return @path basename + * + * If @path doesn't contain any directory separator return NULL. + */ +static NTSTATUS path_base_name(TALLOC_CTX *ctx, const char *path, char **_base) +{ + char *base = NULL; + int last = -1; + int i; + + for (i = 0; path[i]; i++) { + if (path[i] == '\\' || path[i] == '/') { + last = i; + } + } + + if (last >= 0) { + base = talloc_strdup(ctx, path); + if (base == NULL) { + return NT_STATUS_NO_MEMORY; + } + + base[last] = 0; + } + + *_base = base; + return NT_STATUS_OK; +} + +#else + +#define NOT_IMPLEMENTED DEBUG(0, ("tar mode not compiled. build used --without-libarchive\n")) + +int cmd_block(void) +{ + NOT_IMPLEMENTED; + return 1; +} + +int cmd_tarmode(void) +{ + NOT_IMPLEMENTED; + return 1; +} + +int cmd_tar(void) +{ + NOT_IMPLEMENTED; + return 1; +} + +int tar_process(struct tar* tar) +{ + NOT_IMPLEMENTED; + return 1; +} + +int tar_parse_args(struct tar *tar, const char *flag, const char **val, int valsize) +{ + NOT_IMPLEMENTED; + return 1; +} + +bool tar_to_process(struct tar *tar) +{ + return false; +} + +struct tar *tar_get_ctx() +{ + return NULL; +} + +#endif diff --git a/source3/client/clitar_proto.h b/source3/client/clitar_proto.h new file mode 100644 index 0000000..6520476 --- /dev/null +++ b/source3/client/clitar_proto.h @@ -0,0 +1,34 @@ +/* + * Unix SMB/CIFS implementation. + * Tar backup command extension + * Copyright (C) Aurélien Aptel 2013 + * + * 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 3 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, see <http://www.gnu.org/licenses/>. + */ + +#ifndef _CLITAR_PROTO_H_ +#define _CLITAR_PROTO_H_ + +struct tar; + +int cmd_block(void); +int cmd_tarmode(void); +int cmd_setmode(void); +int cmd_tar(void); +int tar_process(struct tar* tar); +int tar_parse_args(struct tar *tar, const char *flag, const char **val, int valsize); +bool tar_to_process(struct tar *tar); +struct tar *tar_get_ctx(void); + +#endif /* _CLITAR_PROTO_H_ */ diff --git a/source3/client/dnsbrowse.c b/source3/client/dnsbrowse.c new file mode 100644 index 0000000..be6eb88 --- /dev/null +++ b/source3/client/dnsbrowse.c @@ -0,0 +1,216 @@ +/* + Unix SMB/CIFS implementation. + DNS-SD browse client + Copyright (C) Rishi Srivatsavai 2007 + + 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 3 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, see <http://www.gnu.org/licenses/>. +*/ + +#include "includes.h" +#include "client/client_proto.h" + +#ifdef WITH_DNSSD_SUPPORT + +#include <dns_sd.h> +#include "system/select.h" + +/* Holds service instances found during DNS browse */ +struct mdns_smbsrv_result +{ + char *serviceName; + char *regType; + char *domain; + uint32_t ifIndex; + struct mdns_smbsrv_result *nextResult; +}; + +/* Maintains state during DNS browse */ +struct mdns_browse_state +{ + struct mdns_smbsrv_result *listhead; /* Browse result list head */ + int browseDone; + +}; + + +static void +do_smb_resolve_reply (DNSServiceRef sdRef, DNSServiceFlags flags, + uint32_t interfaceIndex, DNSServiceErrorType errorCode, + const char *fullname, const char *hosttarget, uint16_t port, + uint16_t txtLen, const unsigned char *txtRecord, void *context) +{ + printf("SMB service available on %s port %u\n", + hosttarget, ntohs(port)); +} + + +static void do_smb_resolve(struct mdns_smbsrv_result *browsesrv) +{ + DNSServiceRef mdns_conn_sdref = NULL; + int mdnsfd; + int fdsetsz; + int ret; + struct timeval tv; + DNSServiceErrorType err; + + TALLOC_CTX * ctx = talloc_tos(); + + err = DNSServiceResolve(&mdns_conn_sdref, 0 /* flags */, + browsesrv->ifIndex, + browsesrv->serviceName, browsesrv->regType, browsesrv->domain, + do_smb_resolve_reply, NULL); + + if (err != kDNSServiceErr_NoError) { + return; + } + + mdnsfd = DNSServiceRefSockFD(mdns_conn_sdref); + for (;;) { + int revents; + + ret = poll_one_fd(mdnsfd, POLLIN|POLLHUP, 1000, &revents); + if (ret <= 0 && errno != EINTR) { + break; + } + + if (revents & (POLLIN|POLLHUP|POLLERR)) { + /* Invoke callback function */ + DNSServiceProcessResult(mdns_conn_sdref); + break; + } + } + + TALLOC_FREE(fdset); + DNSServiceRefDeallocate(mdns_conn_sdref); +} + + +static void +do_smb_browse_reply(DNSServiceRef sdRef, DNSServiceFlags flags, + uint32_t interfaceIndex, DNSServiceErrorType errorCode, + const char *serviceName, const char *regtype, + const char *replyDomain, void *context) +{ + struct mdns_browse_state *bstatep = (struct mdns_browse_state *)context; + struct mdns_smbsrv_result *bresult; + + if (bstatep == NULL) { + return; + } + + if (errorCode != kDNSServiceErr_NoError) { + bstatep->browseDone = 1; + return; + } + + if (flags & kDNSServiceFlagsMoreComing) { + bstatep->browseDone = 0; + } else { + bstatep->browseDone = 1; + } + + if (!(flags & kDNSServiceFlagsAdd)) { + return; + } + + bresult = talloc_array(talloc_tos(), struct mdns_smbsrv_result, 1); + if (bresult == NULL) { + return; + } + + if (bstatep->listhead != NULL) { + bresult->nextResult = bstatep->listhead; + } + + bresult->serviceName = talloc_strdup(talloc_tos(), serviceName); + bresult->regType = talloc_strdup(talloc_tos(), regtype); + bresult->domain = talloc_strdup(talloc_tos(), replyDomain); + bresult->ifIndex = interfaceIndex; + bstatep->listhead = bresult; +} + +int do_smb_browse(void) +{ + int mdnsfd; + int fdsetsz; + int ret; + struct mdns_browse_state bstate; + struct mdns_smbsrv_result *resptr; + struct timeval tv; + DNSServiceRef mdns_conn_sdref = NULL; + DNSServiceErrorType err; + + TALLOC_CTX * ctx = talloc_stackframe(); + + ZERO_STRUCT(bstate); + + err = DNSServiceBrowse(&mdns_conn_sdref, 0, 0, "_smb._tcp", "", + do_smb_browse_reply, &bstate); + + if (err != kDNSServiceErr_NoError) { + d_printf("Error connecting to the Multicast DNS daemon\n"); + TALLOC_FREE(ctx); + return 1; + } + + mdnsfd = DNSServiceRefSockFD(mdns_conn_sdref); + for (;;) { + int revents; + + ret = poll_one_fd(mdnsfd, POLLIN|POLLHUP, 1000, &revents); + if (ret <= 0 && errno != EINTR) { + break; + } + + if (revents & (POLLIN|POLLHUP|POLLERR)) { + /* Invoke callback function */ + if (DNSServiceProcessResult(mdns_conn_sdref)) { + break; + } + if (bstate.browseDone) { + break; + } + } + } + + DNSServiceRefDeallocate(mdns_conn_sdref); + + if (bstate.listhead != NULL) { + resptr = bstate.listhead; + while (resptr != NULL) { + struct mdns_smbsrv_result *oldresptr; + oldresptr = resptr; + + /* Resolve smb service instance */ + do_smb_resolve(resptr); + + resptr = resptr->nextResult; + } + } + + TALLOC_FREE(ctx); + return 0; +} + +#else /* WITH_DNSSD_SUPPORT */ + +int do_smb_browse(void) +{ + d_printf("DNS-SD browsing is not supported on this platform\n"); + return 1; +} + +#endif /* WITH_DNSSD_SUPPORT */ + + diff --git a/source3/client/smbspool.c b/source3/client/smbspool.c new file mode 100644 index 0000000..16a8d44 --- /dev/null +++ b/source3/client/smbspool.c @@ -0,0 +1,888 @@ +/* + Unix SMB/CIFS implementation. + SMB backend for the Common UNIX Printing System ("CUPS") + + Copyright (C) Michael R Sweet 1999 + Copyright (C) Andrew Tridgell 1994-1998 + Copyright (C) Andrew Bartlett 2002 + Copyright (C) Rodrigo Fernandez-Vizarra 2005 + Copyright (C) James Peach 2008 + + 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 3 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, see <http://www.gnu.org/licenses/>. +*/ + +#include "includes.h" +#include "system/filesys.h" +#include "system/passwd.h" +#include "system/kerberos.h" +#include "libsmb/libsmb.h" +#include "lib/param/param.h" +#include "lib/krb5_wrap/krb5_samba.h" + +/* + * Starting with CUPS 1.3, Kerberos support is provided by cupsd including + * the forwarding of user credentials via the authenticated session between + * user and server and the KRB5CCNAME environment variable which will point + * to a temporary file or an in-memory representation depending on the version + * of Kerberos you use. As a result, all of the ticket code that used to + * live here has been removed, and we depend on the user session (if you + * run smbspool by hand) or cupsd to provide the necessary Kerberos info. + * + * Also, the AUTH_USERNAME and AUTH_PASSWORD environment variables provide + * for per-job authentication for non-Kerberized printing. We use those + * if there is no username and password specified in the device URI. + * + * Finally, if we have an authentication failure we return exit code 2 + * which tells CUPS to hold the job for authentication and bug the user + * to get the necessary credentials. + */ + +#define MAX_RETRY_CONNECT 3 + + +/* + * Globals... + */ + + + +/* + * Local functions... + */ + +static int get_exit_code(NTSTATUS nt_status); +static void list_devices(void); +static NTSTATUS +smb_connect(struct cli_state **output_cli, + const char *workgroup, + const char *server, + const int port, + const char *share, + const char *username, + const char *password, + const char *jobusername); +static int smb_print(struct cli_state *, const char *, FILE *); +static char *uri_unescape_alloc(const char *); +#if 0 +static bool smb_encrypt; +#endif + +static const char *auth_info_required; + +/* + * 'main()' - Main entry for SMB backend. + */ + +int /* O - Exit status */ +main(int argc, /* I - Number of command-line arguments */ + char *argv[]) +{ /* I - Command-line arguments */ + int i; /* Looping var */ + int copies; /* Number of copies */ + int port; /* Port number */ + char uri[1024], /* URI */ + *sep, /* Pointer to separator */ + *tmp, *tmp2; /* Temp pointers to do escaping */ + const char *password = NULL; /* Password */ + const char *username = NULL; /* Username */ + char *server, /* Server name */ + *printer;/* Printer name */ + const char *workgroup; /* Workgroup */ + FILE *fp; /* File to print */ + int status = 1; /* Status of LPD job */ + NTSTATUS nt_status = NT_STATUS_UNSUCCESSFUL; + struct cli_state *cli = NULL; /* SMB interface */ + int tries = 0; + const char *dev_uri = NULL; + const char *env = NULL; + const char *config_file = NULL; + TALLOC_CTX *frame = talloc_stackframe(); + const char *print_user = NULL; + const char *print_title = NULL; + const char *print_file = NULL; + const char *print_copies = NULL; + int cmp; + int len; + + if (argc == 1) { + /* + * NEW! In CUPS 1.1 the backends are run with no arguments + * to list the available devices. These can be devices + * served by this backend or any other backends (i.e. you + * can have an SNMP backend that is only used to enumerate + * the available network printers... :) + */ + + list_devices(); + status = 0; + goto done; + } + + /* + * We need at least 5 options if the DEVICE_URI is passed via an env + * variable and printing data comes via stdin. + * We don't accept more than 7 options in total, including optional. + */ + if (argc < 5 || argc > 8) { + fprintf(stderr, +"Usage: %s [DEVICE_URI] job-id user title copies options [file]\n" +" The DEVICE_URI environment variable can also contain the\n" +" destination printer:\n" +"\n" +" smb://[username:password@][workgroup/]server[:port]/printer\n", + argv[0]); + goto done; + } + + /* + * Find out if we have the device_uri in the command line. + * + * If we are started as a CUPS backend argv[0] is normally the + * device_uri! + */ + if (argc == 8) { + /* + * smbspool <uri> <job> <user> <title> <copies> <options> <file> + * 0 1 2 3 4 5 6 7 + */ + + dev_uri = argv[1]; + + print_user = argv[3]; + print_title = argv[4]; + print_copies = argv[5]; + print_file = argv[7]; + } else if (argc == 7) { + int cmp1; + int cmp2; + + /* + * <uri> <job> <user> <title> <copies> <options> <file> + * smbspool <uri> <job> <user> <title> <copies> <options> + * smbspool <job> <user> <title> <copies> <options> <file> | DEVICE_URI + */ + cmp1 = strncmp(argv[0], "smb://", 6); + cmp2 = strncmp(argv[1], "smb://", 6); + + if (cmp1 == 0) { + /* + * <uri> <job> <user> <title> <copies> <options> <file> + * 0 1 2 3 4 5 6 + */ + dev_uri = argv[0]; + + print_user = argv[2]; + print_title = argv[3]; + print_copies = argv[4]; + print_file = argv[6]; + } else if (cmp2 == 0) { + /* + * smbspool <uri> <job> <user> <title> <copies> <options> + * 0 1 2 3 4 5 6 + */ + dev_uri = argv[1]; + + print_user = argv[3]; + print_title = argv[4]; + print_copies = argv[5]; + print_file = NULL; + } else { + /* + * smbspool <job> <user> <title> <copies> <options> <file> | DEVICE_URI + * 0 1 2 3 4 5 6 + */ + print_user = argv[2]; + print_title = argv[3]; + print_copies = argv[4]; + print_file = argv[6]; + } + } else if (argc == 6) { + /* + * <uri> <job> <user> <title> <copies> <options> + * smbspool <job> <user> <title> <copies> <options> | DEVICE_URI + * 0 1 2 3 4 5 + */ + cmp = strncmp(argv[0], "smb://", 6); + if (cmp == 0) { + dev_uri = argv[0]; + } + + print_user = argv[2]; + print_title = argv[3]; + print_copies = argv[4]; + } + + if (print_file != NULL) { + char *endp; + + fp = fopen(print_file, "rb"); + if (fp == NULL) { + fprintf(stderr, + "ERROR: Unable to open print file: %s", + print_file); + goto done; + } + + copies = strtol(print_copies, &endp, 10); + if (print_copies == endp) { + perror("ERROR: Unable to determine number of copies"); + goto done; + } + } else { + fp = stdin; + copies = 1; + } + + /* + * Find the URI ... + * + * The URI in argv[0] is sanitized to remove username/password, so + * use DEVICE_URI if available. Otherwise keep the URI already + * discovered in argv. + */ + env = getenv("DEVICE_URI"); + if (env != NULL && env[0] != '\0') { + dev_uri = env; + } + + if (dev_uri == NULL) { + fprintf(stderr, + "ERROR: No valid device URI has been specified\n"); + goto done; + } + + cmp = strncmp(dev_uri, "smb://", 6); + if (cmp != 0) { + fprintf(stderr, + "ERROR: No valid device URI has been specified\n"); + goto done; + } + len = snprintf(uri, sizeof(uri), "%s", dev_uri); + if (len >= sizeof(uri)) { + fprintf(stderr, + "ERROR: The URI is too long.\n"); + goto done; + } + + auth_info_required = getenv("AUTH_INFO_REQUIRED"); + if (auth_info_required == NULL) { + auth_info_required = "samba"; + } + + /* + * Extract the destination from the URI... + */ + + if ((sep = strrchr_m(uri, '@')) != NULL) { + tmp = uri + 6; + *sep++ = '\0'; + + /* username is in tmp */ + + server = sep; + + /* + * Extract password as needed... + */ + + if ((tmp2 = strchr_m(tmp, ':')) != NULL) { + *tmp2++ = '\0'; + password = uri_unescape_alloc(tmp2); + } + username = uri_unescape_alloc(tmp); + } else { + env = getenv("AUTH_USERNAME"); + if (env != NULL && strlen(env) > 0) { + username = env; + } + + env = getenv("AUTH_PASSWORD"); + if (env != NULL && strlen(env) > 0) { + password = env; + } + + server = uri + 6; + } + + if (password != NULL) { + auth_info_required = "username,password"; + } + + tmp = server; + + if ((sep = strchr_m(tmp, '/')) == NULL) { + fputs("ERROR: Bad URI - need printer name!\n", stderr); + goto done; + } + + *sep++ = '\0'; + tmp2 = sep; + + if ((sep = strchr_m(tmp2, '/')) != NULL) { + /* + * Convert to smb://[username:password@]workgroup/server/printer... + */ + + *sep++ = '\0'; + + workgroup = uri_unescape_alloc(tmp); + server = uri_unescape_alloc(tmp2); + printer = uri_unescape_alloc(sep); + } else { + workgroup = NULL; + server = uri_unescape_alloc(tmp); + printer = uri_unescape_alloc(tmp2); + } + + if ((sep = strrchr_m(server, ':')) != NULL) { + *sep++ = '\0'; + + port = atoi(sep); + } else { + port = 0; + } + + /* + * Setup the SAMBA server state... + */ + + setup_logging("smbspool", DEBUG_STDERR); + + smb_init_locale(); + + config_file = lp_default_path(); + if (!lp_load_client(config_file)) { + fprintf(stderr, + "ERROR: Can't load %s - run testparm to debug it\n", + config_file); + goto done; + } + + if (workgroup == NULL) { + workgroup = lp_workgroup(); + } + + load_interfaces(); + + do { + nt_status = smb_connect(&cli, + workgroup, + server, + port, + printer, + username, + password, + print_user); + if (!NT_STATUS_IS_OK(nt_status)) { + status = get_exit_code(nt_status); + if (status == 2) { + fprintf(stderr, + "DEBUG: Unable to connect to CIFS " + "host: %s", + nt_errstr(nt_status)); + goto done; + } else if (getenv("CLASS") == NULL) { + fprintf(stderr, + "ERROR: Unable to connect to CIFS " + "host: %s. Will retry in 60 " + "seconds...\n", + nt_errstr(nt_status)); + sleep(60); + tries++; + } else { + fprintf(stderr, + "ERROR: Unable to connect to CIFS " + "host: %s. Trying next printer...\n", + nt_errstr(nt_status)); + goto done; + } + } + } while (!NT_STATUS_IS_OK(nt_status) && (tries < MAX_RETRY_CONNECT)); + + if (cli == NULL) { + fprintf(stderr, "ERROR: Unable to connect to CIFS host after (tried %d times)\n", tries); + goto done; + } + + /* + * Now that we are connected to the server, ignore SIGTERM so that we + * can finish out any page data the driver sends (e.g. to eject the + * current page... Only ignore SIGTERM if we are printing data from + * stdin (otherwise you can't cancel raw jobs...) + */ + + if (argc < 7) { + CatchSignal(SIGTERM, SIG_IGN); + } + + /* + * Queue the job... + */ + + for (i = 0; i < copies; i++) { + status = smb_print(cli, print_title, fp); + if (status != 0) { + break; + } + } + + cli_shutdown(cli); + + /* + * Return the queue status... + */ + +done: + + TALLOC_FREE(frame); + return (status); +} + + +/* + * 'get_exit_code()' - Get the backend exit code based on the current error. + */ + +static int +get_exit_code(NTSTATUS nt_status) +{ + size_t i; + + /* List of NTSTATUS errors that are considered + * authentication errors + */ + static const NTSTATUS auth_errors[] = + { + NT_STATUS_ACCESS_DENIED, + NT_STATUS_ACCESS_VIOLATION, + NT_STATUS_ACCOUNT_DISABLED, + NT_STATUS_ACCOUNT_LOCKED_OUT, + NT_STATUS_ACCOUNT_RESTRICTION, + NT_STATUS_DOMAIN_CONTROLLER_NOT_FOUND, + NT_STATUS_INVALID_ACCOUNT_NAME, + NT_STATUS_INVALID_COMPUTER_NAME, + NT_STATUS_INVALID_LOGON_HOURS, + NT_STATUS_INVALID_WORKSTATION, + NT_STATUS_LOGON_FAILURE, + NT_STATUS_NO_SUCH_USER, + NT_STATUS_NO_SUCH_DOMAIN, + NT_STATUS_NO_LOGON_SERVERS, + NT_STATUS_PASSWORD_EXPIRED, + NT_STATUS_PRIVILEGE_NOT_HELD, + NT_STATUS_SHARING_VIOLATION, + NT_STATUS_WRONG_PASSWORD, + }; + + + fprintf(stderr, + "DEBUG: get_exit_code(nt_status=%s [%x])\n", + nt_errstr(nt_status), NT_STATUS_V(nt_status)); + + for (i = 0; i < ARRAY_SIZE(auth_errors); i++) { + if (!NT_STATUS_EQUAL(nt_status, auth_errors[i])) { + continue; + } + + fprintf(stderr, "ATTR: auth-info-required=%s\n", auth_info_required); + + /* + * 2 = authentication required... + */ + + return (2); + + } + + /* + * 1 = fail + */ + + return (1); +} + + +/* + * 'list_devices()' - List the available printers seen on the network... + */ + +static void +list_devices(void) +{ + /* + * Eventually, search the local workgroup for available hosts and printers. + */ + + puts("network smb \"Unknown\" \"Windows Printer via SAMBA\""); +} + + +static NTSTATUS +smb_complete_connection(struct cli_state **output_cli, + const char *myname, + const char *server, + int port, + const char *username, + const char *password, + const char *workgroup, + const char *share, + bool use_kerberos, + bool fallback_after_kerberos) +{ + struct cli_state *cli; /* New connection */ + NTSTATUS nt_status; + struct cli_credentials *creds = NULL; + + /* Start the SMB connection */ + nt_status = cli_start_connection(&cli, myname, server, NULL, port, + SMB_SIGNING_DEFAULT, 0); + if (!NT_STATUS_IS_OK(nt_status)) { + fprintf(stderr, "ERROR: Connection failed: %s\n", nt_errstr(nt_status)); + return nt_status; + } + + creds = cli_session_creds_init(cli, + username, + workgroup, + NULL, /* realm */ + password, + use_kerberos, + fallback_after_kerberos, + false, /* use_ccache */ + false); /* password_is_nt_hash */ + if (creds == NULL) { + fprintf(stderr, "ERROR: cli_session_creds_init failed\n"); + cli_shutdown(cli); + return NT_STATUS_NO_MEMORY; + } + + nt_status = cli_session_setup_creds(cli, creds); + if (!NT_STATUS_IS_OK(nt_status)) { + fprintf(stderr, "ERROR: Session setup failed: %s\n", nt_errstr(nt_status)); + + cli_shutdown(cli); + + return nt_status; + } + + nt_status = cli_tree_connect_creds(cli, share, "?????", creds); + if (!NT_STATUS_IS_OK(nt_status)) { + fprintf(stderr, "ERROR: Tree connect failed (%s)\n", + nt_errstr(nt_status)); + + cli_shutdown(cli); + + return nt_status; + } + + *output_cli = cli; + return NT_STATUS_OK; +} + +static bool kerberos_ccache_is_valid(void) { + krb5_context ctx; + const char *ccache_name = NULL; + krb5_ccache ccache = NULL; + krb5_error_code code; + + code = smb_krb5_init_context_common(&ctx); + if (code != 0) { + DBG_ERR("kerberos init context failed (%s)\n", + error_message(code)); + return false; + } + + ccache_name = krb5_cc_default_name(ctx); + if (ccache_name == NULL) { + DBG_ERR("Failed to get default ccache name\n"); + krb5_free_context(ctx); + return false; + } + + code = krb5_cc_resolve(ctx, ccache_name, &ccache); + if (code != 0) { + DBG_ERR("Failed to resolve ccache name: %s\n", + ccache_name); + krb5_free_context(ctx); + return false; + } else { + krb5_principal default_princ = NULL; + char *princ_name = NULL; + + code = krb5_cc_get_principal(ctx, + ccache, + &default_princ); + if (code != 0) { + DBG_ERR("Failed to get default principal from " + "ccache: %s\n", + ccache_name); + krb5_cc_close(ctx, ccache); + krb5_free_context(ctx); + return false; + } + + code = krb5_unparse_name(ctx, + default_princ, + &princ_name); + if (code == 0) { + fprintf(stderr, + "DEBUG: Try to authenticate as %s\n", + princ_name); + krb5_free_unparsed_name(ctx, princ_name); + } + krb5_free_principal(ctx, default_princ); + } + krb5_cc_close(ctx, ccache); + krb5_free_context(ctx); + + return true; +} + +/* + * 'smb_connect()' - Return a connection to a server. + */ + +static NTSTATUS +smb_connect(struct cli_state **output_cli, + const char *workgroup, /* I - Workgroup */ + const char *server, /* I - Server */ + const int port, /* I - Port */ + const char *share, /* I - Printer */ + const char *username, /* I - Username */ + const char *password, /* I - Password */ + const char *jobusername) /* I - User who issued the print job */ +{ + struct cli_state *cli = NULL; /* New connection */ + char *myname = NULL; /* Client name */ + struct passwd *pwd; + bool use_kerberos = false; + bool fallback_after_kerberos = false; + const char *user = username; + NTSTATUS nt_status; + + /* + * Get the names and addresses of the client and server... + */ + myname = get_myname(talloc_tos()); + if (!myname) { + return NT_STATUS_NO_MEMORY; + } + + + if (strcmp(auth_info_required, "negotiate") == 0) { + if (!kerberos_ccache_is_valid()) { + fprintf(stderr, + "ERROR: No valid Kerberos credential cache found! " + "Using smbspool_krb5_wrapper may help.\n"); + return NT_STATUS_LOGON_FAILURE; + } + user = jobusername; + + use_kerberos = true; + fprintf(stderr, + "DEBUG: Try to connect using Kerberos ...\n"); + } else if (strcmp(auth_info_required, "username,password") == 0) { + if (username == NULL) { + return NT_STATUS_INVALID_ACCOUNT_NAME; + } + + /* Fallback to NTLM */ + fallback_after_kerberos = true; + + fprintf(stderr, + "DEBUG: Try to connect using username/password ...\n"); + } else if (strcmp(auth_info_required, "none") == 0) { + goto anonymous; + } else if (strcmp(auth_info_required, "samba") == 0) { + if (username != NULL) { + fallback_after_kerberos = true; + } else if (kerberos_ccache_is_valid()) { + auth_info_required = "negotiate"; + + user = jobusername; + use_kerberos = true; + } else { + fprintf(stderr, + "DEBUG: This backend requires credentials!\n"); + return NT_STATUS_ACCESS_DENIED; + } + } else { + return NT_STATUS_ACCESS_DENIED; + } + + nt_status = smb_complete_connection(&cli, + myname, + server, + port, + user, + password, + workgroup, + share, + true, /* try kerberos */ + fallback_after_kerberos); + if (NT_STATUS_IS_OK(nt_status)) { + fprintf(stderr, "DEBUG: SMB connection established.\n"); + + *output_cli = cli; + return NT_STATUS_OK; + } + + if (!use_kerberos) { + fprintf(stderr, "ERROR: SMB connection failed!\n"); + return nt_status; + } + + /* give a chance for a passwordless NTLMSSP session setup */ + pwd = getpwuid(geteuid()); + if (pwd == NULL) { + return NT_STATUS_ACCESS_DENIED; + } + + nt_status = smb_complete_connection(&cli, + myname, + server, + port, + pwd->pw_name, + "", + workgroup, + share, + false, false); + if (NT_STATUS_IS_OK(nt_status)) { + fputs("DEBUG: Connected with NTLMSSP...\n", stderr); + + *output_cli = cli; + return NT_STATUS_OK; + } + + /* + * last try. Use anonymous authentication + */ + +anonymous: + nt_status = smb_complete_connection(&cli, + myname, + server, + port, + "", + "", + workgroup, + share, + false, false); + if (NT_STATUS_IS_OK(nt_status)) { + *output_cli = cli; + return NT_STATUS_OK; + } + + return nt_status; +} + + +/* + * 'smb_print()' - Queue a job for printing using the SMB protocol. + */ + +static int /* O - 0 = success, non-0 = failure */ +smb_print(struct cli_state * cli, /* I - SMB connection */ + const char *print_title, /* I - Title/job name */ + FILE * fp) +{ /* I - File to print */ + uint16_t fnum; /* File number */ + int nbytes, /* Number of bytes read */ + tbytes; /* Total bytes read */ + char buffer[8192], /* Buffer for copy */ + *ptr; /* Pointer into title */ + char title[1024] = {0}; + int len; + NTSTATUS nt_status; + + + /* + * Sanitize the title... + */ + len = snprintf(title, sizeof(title), "%s", print_title); + if (len != strlen(print_title)) { + return 2; + } + + for (ptr = title; *ptr; ptr++) { + if (!isalnum((int) *ptr) && !isspace((int) *ptr)) { + *ptr = '_'; + } + } + + /* + * Open the printer device... + */ + + nt_status = cli_open(cli, title, O_RDWR | O_CREAT | O_TRUNC, DENY_NONE, + &fnum); + if (!NT_STATUS_IS_OK(nt_status)) { + fprintf(stderr, "ERROR: %s opening remote spool %s\n", + nt_errstr(nt_status), title); + return get_exit_code(nt_status); + } + + /* + * Copy the file to the printer... + */ + + if (fp != stdin) + rewind(fp); + + tbytes = 0; + + while ((nbytes = fread(buffer, 1, sizeof(buffer), fp)) > 0) { + NTSTATUS status; + + status = cli_writeall(cli, fnum, 0, (uint8_t *)buffer, + tbytes, nbytes, NULL); + if (!NT_STATUS_IS_OK(status)) { + int ret = get_exit_code(status); + fprintf(stderr, "ERROR: Error writing spool: %s\n", + nt_errstr(status)); + fprintf(stderr, "DEBUG: Returning status %d...\n", + ret); + cli_close(cli, fnum); + + return (ret); + } + tbytes += nbytes; + } + + nt_status = cli_close(cli, fnum); + if (!NT_STATUS_IS_OK(nt_status)) { + fprintf(stderr, "ERROR: %s closing remote spool %s\n", + nt_errstr(nt_status), title); + return get_exit_code(nt_status); + } else { + return (0); + } +} + +static char * +uri_unescape_alloc(const char *uritok) +{ + char *ret; + char *end; + ret = (char *) SMB_STRDUP(uritok); + if (!ret) { + return NULL; + } + + end = rfc1738_unescape(ret); + if (end == NULL) { + free(ret); + return NULL; + } + return ret; +} diff --git a/source3/client/smbspool_krb5_wrapper.c b/source3/client/smbspool_krb5_wrapper.c new file mode 100644 index 0000000..9abd1c6 --- /dev/null +++ b/source3/client/smbspool_krb5_wrapper.c @@ -0,0 +1,341 @@ +/* + * Unix SMB/CIFS implementation. + * + * CUPS printing backend helper to execute smbspool + * + * Copyright (C) 2010-2011 Andreas Schneider <asn@samba.org> + * + * 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 3 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, see <http://www.gnu.org/licenses/>. + */ + +#include "includes.h" +#include "system/filesys.h" +#include "system/kerberos.h" +#include "system/passwd.h" + +#include <errno.h> +#include <stdlib.h> +#include <string.h> + +#include <cups/backend.h> + +#include "dynconfig/dynconfig.h" + +#undef calloc + +enum cups_smb_dbglvl_e { + CUPS_SMB_LOG_DEBUG = 0, + CUPS_SMB_LOG_ERROR, +}; +static void cups_smb_debug(enum cups_smb_dbglvl_e lvl, const char *format, ...) + PRINTF_ATTRIBUTE(2, 3); + +#define CUPS_SMB_DEBUG(...) cups_smb_debug(CUPS_SMB_LOG_DEBUG, __VA_ARGS__) +#define CUPS_SMB_ERROR(...) cups_smb_debug(CUPS_SMB_LOG_DEBUG, __VA_ARGS__) + +static void cups_smb_debug(enum cups_smb_dbglvl_e lvl, const char *format, ...) +{ + const char *prefix = "DEBUG"; + char buffer[1024]; + va_list va; + + va_start(va, format); + vsnprintf(buffer, sizeof(buffer), format, va); + va_end(va); + + switch (lvl) { + case CUPS_SMB_LOG_DEBUG: + prefix = "DEBUG"; + break; + case CUPS_SMB_LOG_ERROR: + prefix = "ERROR"; + break; + } + + fprintf(stderr, + "%s: SMBSPOOL_KRB5 - %s\n", + prefix, + buffer); +} + +static bool kerberos_get_default_ccache(char *ccache_buf, size_t len) +{ + krb5_context ctx; + const char *ccache_name = NULL; + char *full_ccache_name = NULL; + krb5_ccache ccache = NULL; + krb5_error_code code; + + code = krb5_init_context(&ctx); + if (code != 0) { + return false; + } + + ccache_name = krb5_cc_default_name(ctx); + if (ccache_name == NULL) { + krb5_free_context(ctx); + return false; + } + + code = krb5_cc_resolve(ctx, ccache_name, &ccache); + if (code != 0) { + krb5_free_context(ctx); + return false; + } + + code = krb5_cc_get_full_name(ctx, ccache, &full_ccache_name); + krb5_cc_close(ctx, ccache); + if (code != 0) { + krb5_free_context(ctx); + return false; + } + + snprintf(ccache_buf, len, "%s", full_ccache_name); + +#ifdef SAMBA4_USES_HEIMDAL + free(full_ccache_name); +#else + krb5_free_string(ctx, full_ccache_name); +#endif + krb5_free_context(ctx); + + return true; +} + +/* + * This is a helper binary to execute smbspool. + * + * It needs to be installed or symlinked as: + * /usr/lib/cups/backend/smb + * + * The permissions of the binary need to be set to 0700 so that it is executed + * as root. The binary switches to the user which is passed via the environment + * variable AUTH_UID, so we can access the kerberos ticket. + */ +int main(int argc, char *argv[]) +{ + char smbspool_cmd[PATH_MAX] = {0}; + struct passwd *pwd; + struct group *g = NULL; + char gen_cc[PATH_MAX] = {0}; + char *env = NULL; + char auth_info_required[256] = {0}; + char device_uri[4096] = {0}; + uid_t uid = (uid_t)-1; + gid_t gid = (gid_t)-1; + gid_t groups[1] = { (gid_t)-1 }; + unsigned long tmp; + bool ok; + int cmp; + int rc; + + env = getenv("DEVICE_URI"); + if (env != NULL && strlen(env) > 2) { + snprintf(device_uri, sizeof(device_uri), "%s", env); + } + + /* We must handle the following values of AUTH_INFO_REQUIRED: + * none: Anonymous/guest printing + * username,password: A username (of the form "username" or "DOMAIN\username") + * and password are required + * negotiate: Kerberos authentication + * NULL (not set): will never happen when called from cupsd + * https://www.cups.org/doc/spec-ipp.html#auth-info-required + * https://github.com/apple/cups/issues/5674 + */ + env = getenv("AUTH_INFO_REQUIRED"); + + /* If not set, then just call smbspool. */ + if (env == NULL || env[0] == 0) { + CUPS_SMB_DEBUG("AUTH_INFO_REQUIRED is not set - " + "executing smbspool"); + /* Pass this printing task to smbspool without Kerberos auth */ + goto smbspool; + } else { + CUPS_SMB_DEBUG("AUTH_INFO_REQUIRED=%s", env); + + /* First test the value of AUTH_INFO_REQUIRED + * against known possible values + */ + cmp = strcmp(env, "none"); + if (cmp == 0) { + CUPS_SMB_DEBUG("Authenticate using none (anonymous) - " + "executing smbspool"); + goto smbspool; + } + + cmp = strcmp(env, "username,password"); + if (cmp == 0) { + CUPS_SMB_DEBUG("Authenticate using username/password - " + "executing smbspool"); + goto smbspool; + } + + /* Now, if 'goto smbspool' still has not happened, + * there are only two variants left: + * 1) AUTH_INFO_REQUIRED is "negotiate" and then + * we have to continue working + * 2) or it is something not known to us, then Kerberos + * authentication is not required, so just also pass + * this task to smbspool + */ + cmp = strcmp(env, "negotiate"); + if (cmp != 0) { + CUPS_SMB_DEBUG("Value of AUTH_INFO_REQUIRED is not known " + "to smbspool_krb5_wrapper, executing smbspool"); + goto smbspool; + } + + snprintf(auth_info_required, + sizeof(auth_info_required), + "%s", + env); + } + + uid = getuid(); + + CUPS_SMB_DEBUG("Started with uid=%d\n", uid); + if (uid != 0) { + goto smbspool; + } + + /* + * AUTH_UID gets only set if we have an incoming connection over the + * CUPS unix domain socket. + */ + env = getenv("AUTH_UID"); + if (env == NULL) { + CUPS_SMB_ERROR("AUTH_UID is not set"); + fprintf(stderr, "ATTR: auth-info-required=negotiate\n"); + return CUPS_BACKEND_AUTH_REQUIRED; + } + + if (strlen(env) > 10) { + CUPS_SMB_ERROR("Invalid AUTH_UID"); + return CUPS_BACKEND_FAILED; + } + + errno = 0; + tmp = strtoul(env, NULL, 10); + if (errno != 0 || tmp >= UINT32_MAX) { + CUPS_SMB_ERROR("Failed to convert AUTH_UID=%s", env); + return CUPS_BACKEND_FAILED; + } + uid = (uid_t)tmp; + + /* If we are printing as the root user, we're done here. */ + if (uid == 0) { + goto smbspool; + } + + pwd = getpwuid(uid); + if (pwd == NULL) { + CUPS_SMB_ERROR("Failed to find system user: %u - %s", + uid, strerror(errno)); + return CUPS_BACKEND_FAILED; + } + gid = pwd->pw_gid; + + rc = setgroups(0, NULL); + if (rc != 0) { + CUPS_SMB_ERROR("Failed to clear groups - %s", + strerror(errno)); + return CUPS_BACKEND_FAILED; + } + + /* + * We need the primary group of the 'lp' user. This is needed to access + * temporary files in /var/spool/cups/. + */ + g = getgrnam("lp"); + if (g == NULL) { + CUPS_SMB_ERROR("Failed to find user 'lp' - %s", + strerror(errno)); + return CUPS_BACKEND_FAILED; + } + + CUPS_SMB_DEBUG("Adding group 'lp' (%u)", g->gr_gid); + groups[0] = g->gr_gid; + rc = setgroups(sizeof(groups), groups); + if (rc != 0) { + CUPS_SMB_ERROR("Failed to set groups for 'lp' - %s", + strerror(errno)); + return CUPS_BACKEND_FAILED; + } + + CUPS_SMB_DEBUG("Switching to gid=%d", gid); + rc = setgid(gid); + if (rc != 0) { + CUPS_SMB_ERROR("Failed to switch to gid=%u - %s", + gid, + strerror(errno)); + return CUPS_BACKEND_FAILED; + } + + CUPS_SMB_DEBUG("Switching to uid=%u", uid); + rc = setuid(uid); + if (rc != 0) { + CUPS_SMB_ERROR("Failed to switch to uid=%u - %s", + uid, + strerror(errno)); + return CUPS_BACKEND_FAILED; + } + + env = getenv("KRB5CCNAME"); + if (env != NULL && env[0] != 0) { + snprintf(gen_cc, sizeof(gen_cc), "%s", env); + CUPS_SMB_DEBUG("User already set KRB5CCNAME [%s] as ccache", + gen_cc); + + goto create_env; + } + + ok = kerberos_get_default_ccache(gen_cc, sizeof(gen_cc)); + if (ok) { + CUPS_SMB_DEBUG("Use default KRB5CCNAME [%s]", + gen_cc); + goto create_env; + } + + /* Fallback to a FILE ccache */ + snprintf(gen_cc, sizeof(gen_cc), "FILE:/tmp/krb5cc_%u", uid); + +create_env: + /* + * Make sure we do not have LD_PRELOAD or other security relevant + * environment variables set. + */ +#ifdef HAVE_CLEARENV + clearenv(); +#else + environ = calloc(3, sizeof(*environ)); +#endif + + CUPS_SMB_DEBUG("Setting KRB5CCNAME to '%s'", gen_cc); + setenv("KRB5CCNAME", gen_cc, 1); + if (device_uri[0] != '\0') { + setenv("DEVICE_URI", device_uri, 1); + } + if (auth_info_required[0] != '\0') { + setenv("AUTH_INFO_REQUIRED", auth_info_required, 1); + } + +smbspool: + snprintf(smbspool_cmd, + sizeof(smbspool_cmd), + "%s/smbspool", + get_dyn_BINDIR()); + + return execv(smbspool_cmd, argv); +} diff --git a/source3/client/tree.c b/source3/client/tree.c new file mode 100644 index 0000000..0e51f61 --- /dev/null +++ b/source3/client/tree.c @@ -0,0 +1,812 @@ +/* + Unix SMB/CIFS implementation. + SMB client GTK+ tree-based application + Copyright (C) Andrew Tridgell 1998 + Copyright (C) Richard Sharpe 2001 + Copyright (C) John Terpstra 2001 + + 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 3 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, see <http://www.gnu.org/licenses/>. +*/ + +/* example-gtk+ application, ripped off from the gtk+ tree.c sample */ + +#include <stdio.h> +#include <errno.h> +#include <gtk/gtk.h> +#include "libsmbclient.h" + +static GtkWidget *clist; + +struct tree_data { + + guint32_t type; /* Type of tree item, an SMBC_TYPE */ + char name[256]; /* May need to change this later */ + +}; + +static void tree_error_message(gchar *message) { + + GtkWidget *dialog, *label, *okay_button; + + /* Create the widgets */ + + dialog = gtk_dialog_new(); + gtk_window_set_modal(GTK_WINDOW(dialog), True); + label = gtk_label_new (message); + okay_button = gtk_button_new_with_label("Okay"); + + /* Ensure that the dialog box is destroyed when the user clicks ok. */ + + gtk_signal_connect_object (GTK_OBJECT (okay_button), "clicked", + GTK_SIGNAL_FUNC (gtk_widget_destroy), dialog); + gtk_container_add (GTK_CONTAINER (GTK_DIALOG(dialog)->action_area), + okay_button); + + /* Add the label, and show everything we've added to the dialog. */ + + gtk_container_add (GTK_CONTAINER (GTK_DIALOG(dialog)->vbox), + label); + gtk_widget_show_all (dialog); +} + +/* + * We are given a widget, and we want to retrieve its URL so we + * can do a directory listing. + * + * We walk back up the tree, picking up pieces until we hit a server or + * workgroup type and return a path from there + */ + +static char *path_string; + +char *get_path(TALLOC_CTX *ctx, GtkWidget *item) +{ + GtkWidget *p = item; + struct tree_data *pd; + char *comps[1024]; /* We keep pointers to the components here */ + int i = 0, j, level,type; + + /* Walk back up the tree, getting the private data */ + + level = GTK_TREE(item->parent)->level; + + /* Pick up this item's component info */ + + pd = (struct tree_data *)gtk_object_get_user_data(GTK_OBJECT(item)); + + comps[i++] = pd->name; + type = pd->type; + + while (level > 0 && type != SMBC_SERVER && type != SMBC_WORKGROUP) { + + /* Find the parent and extract the data etc ... */ + + p = GTK_WIDGET(p->parent); + p = GTK_WIDGET(GTK_TREE(p)->tree_owner); + + pd = (struct tree_data *)gtk_object_get_user_data(GTK_OBJECT(p)); + + level = GTK_TREE(item->parent)->level; + + comps[i++] = pd->name; + type = pd->type; + + } + + /* + * Got a list of comps now, should check that we did not hit a workgroup + * when we got other things as well ... Later + * + * Now, build the path + */ + + TALLOC_FREE(path_string); + path_string = talloc_strdup(ctx, "smb:/"); + + if (path_string) { + for (j = i - 1; j >= 0; j--) { + path_string = talloc_asprintf_append(path_string, "/%s", comps[j]); + } + } + + if (path_string) { + fprintf(stdout, "Path string = %s\n", path_string); + } + + return path_string; + +} + +struct tree_data *make_tree_data(guint32_t type, const char *name) +{ + struct tree_data *p = SMB_MALLOC_P(struct tree_data); + + if (p) { + + p->type = type; + strncpy(p->name, name, sizeof(p->name)); + + } + + return p; + +} + +/* Note that this is called every time the user clicks on an item, + whether it is already selected or not. */ +static void cb_select_child (GtkWidget *root_tree, GtkWidget *child, + GtkWidget *subtree) +{ + gint dh, err, dirlen; + char dirbuf[512]; + struct smbc_dirent *dirp; + struct stat st1; + char *path; + TALLOC_CTX *ctx = talloc_stackframe(); + + g_print ("select_child called for root tree %p, subtree %p, child %p\n", + root_tree, subtree, child); + + /* Now, figure out what it is, and display it in the clist ... */ + + gtk_clist_clear(GTK_CLIST(clist)); /* Clear the CLIST */ + + /* Now, get the private data for the subtree */ + + path = get_path(ctx, child); + if (!path) { + gtk_main_quit(); + TALLOC_FREE(ctx); + return; + } + + if ((dh = smbc_opendir(path)) < 0) { /* Handle error */ + g_print("cb_select_child: Could not open dir %s, %s\n", path, + strerror(errno)); + gtk_main_quit(); + TALLOC_FREE(ctx); + return; + } + + while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, + sizeof(dirbuf))) != 0) { + if (err < 0) { + g_print("cb_select_child: Could not read dir %s, %s\n", path, + strerror(errno)); + gtk_main_quit(); + TALLOC_FREE(ctx); + return; + } + + dirp = (struct smbc_dirent *)dirbuf; + + while (err > 0) { + gchar col1[128], col2[128], col3[128], col4[128]; + gchar *rowdata[4] = {col1, col2, col3, col4}; + + dirlen = dirp->dirlen; + + /* Format each of the items ... */ + + strncpy(col1, dirp->name, 128); + + col2[0] = col3[0] = col4[0] = (char)0; + + switch (dirp->smbc_type) { + + case SMBC_WORKGROUP: + + break; + + case SMBC_SERVER: + + strncpy(col2, (dirp->comment?dirp->comment:""), 128); + + break; + + case SMBC_FILE_SHARE: + + strncpy(col2, (dirp->comment?dirp->comment:""), 128); + + break; + + case SMBC_PRINTER_SHARE: + + strncpy(col2, (dirp->comment?dirp->comment:""), 128); + break; + + case SMBC_COMMS_SHARE: + + break; + + case SMBC_IPC_SHARE: + + break; + + case SMBC_DIR: + case SMBC_FILE: + + /* Get stats on the file/dir and see what we have */ + + if ((strcmp(dirp->name, ".") != 0) && + (strcmp(dirp->name, "..") != 0)) { + char *path1; + + path1 = talloc_asprintf(ctx, + "%s/%s", + path, + dirp->name); + if (!path1) { + gtk_main_quit(); + TALLOC_FREE(ctx); + return; + } + + if (smbc_stat(path1, &st1) < 0) { + if (errno != EBUSY) { + g_print("cb_select_child: Could not stat file %s, %s\n", path1, + strerror(errno)); + gtk_main_quit(); + TALLOC_FREE(ctx); + return; + } else { + strncpy(col2, "Device or resource busy", sizeof(col2)); + } + } + else { + /* Now format each of the relevant things ... */ + + snprintf(col2, sizeof(col2), "%c%c%c%c%c%c%c%c%c(%0X)", + (st1.st_mode&S_IRUSR?'r':'-'), + (st1.st_mode&S_IWUSR?'w':'-'), + (st1.st_mode&S_IXUSR?'x':'-'), + (st1.st_mode&S_IRGRP?'r':'-'), + (st1.st_mode&S_IWGRP?'w':'-'), + (st1.st_mode&S_IXGRP?'x':'-'), + (st1.st_mode&S_IROTH?'r':'-'), + (st1.st_mode&S_IWOTH?'w':'-'), + (st1.st_mode&S_IXOTH?'x':'-'), + st1.st_mode); + snprintf(col3, sizeof(col3), "%u", st1.st_size); + snprintf(col4, sizeof(col4), "%s", ctime(&st1.st_mtime)); + } + } + + break; + + default: + + break; + } + + gtk_clist_append(GTK_CLIST(clist), rowdata); + + (char *)dirp += dirlen; + err -= dirlen; + + } + } + TALLOC_FREE(ctx); +} + +/* Note that this is never called */ +static void cb_unselect_child( GtkWidget *root_tree, + GtkWidget *child, + GtkWidget *subtree ) +{ + g_print ("unselect_child called for root tree %p, subtree %p, child %p\n", + root_tree, subtree, child); +} + +/* for all the GtkItem:: and GtkTreeItem:: signals */ +static void cb_itemsignal( GtkWidget *item, + gchar *signame ) +{ + GtkWidget *real_tree, *aitem, *subtree; + gchar *name; + GtkLabel *label; + gint dh, err, dirlen, level; + char dirbuf[512]; + struct smbc_dirent *dirp; + + label = GTK_LABEL (GTK_BIN (item)->child); + /* Get the text of the label */ + gtk_label_get (label, &name); + + level = GTK_TREE(item->parent)->level; + + /* Get the level of the tree which the item is in */ + g_print ("%s called for item %s->%p, level %d\n", signame, name, + item, GTK_TREE (item->parent)->level); + + real_tree = GTK_TREE_ITEM_SUBTREE(item); /* Get the subtree */ + + if (strncmp(signame, "expand", 6) == 0) { /* Expand called */ + char server[128]; + + if ((dh = smbc_opendir(get_path(item))) < 0) { /* Handle error */ + gchar errmsg[256]; + + g_print("cb_itemsignal: Could not open dir %s, %s\n", get_path(item), + strerror(errno)); + + slprintf(errmsg, sizeof(errmsg), "cb_itemsignal: Could not open dir %s, %s\n", get_path(item), strerror(errno)); + + tree_error_message(errmsg); + + /* gtk_main_quit();*/ + + return; + + } + + while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, + sizeof(dirbuf))) != 0) { + + if (err < 0) { /* An error, report it */ + gchar errmsg[256]; + + g_print("cb_itemsignal: Could not read dir smbc://, %s\n", + strerror(errno)); + + slprintf(errmsg, sizeof(errmsg), "cb_itemsignal: Could not read dir smbc://, %s\n", strerror(errno)); + + tree_error_message(errmsg); + + /* gtk_main_quit();*/ + + return; + + } + + dirp = (struct smbc_dirent *)dirbuf; + + while (err > 0) { + struct tree_data *my_data; + + dirlen = dirp->dirlen; + + my_data = make_tree_data(dirp->smbc_type, dirp->name); + + if (!my_data) { + + g_print("Could not allocate space for tree_data: %s\n", + dirp->name); + + gtk_main_quit(); + return; + + } + + aitem = gtk_tree_item_new_with_label(dirp->name); + + /* Connect all GtkItem:: and GtkTreeItem:: signals */ + gtk_signal_connect (GTK_OBJECT(aitem), "select", + GTK_SIGNAL_FUNC(cb_itemsignal), "select"); + gtk_signal_connect (GTK_OBJECT(aitem), "deselect", + GTK_SIGNAL_FUNC(cb_itemsignal), "deselect"); + gtk_signal_connect (GTK_OBJECT(aitem), "toggle", + GTK_SIGNAL_FUNC(cb_itemsignal), "toggle"); + gtk_signal_connect (GTK_OBJECT(aitem), "expand", + GTK_SIGNAL_FUNC(cb_itemsignal), "expand"); + gtk_signal_connect (GTK_OBJECT(aitem), "collapse", + GTK_SIGNAL_FUNC(cb_itemsignal), "collapse"); + /* Add it to the parent tree */ + gtk_tree_append (GTK_TREE(real_tree), aitem); + + gtk_widget_show (aitem); + + gtk_object_set_user_data(GTK_OBJECT(aitem), (gpointer)my_data); + + fprintf(stdout, "Added: %s, len: %u\n", dirp->name, dirlen); + + if (dirp->smbc_type != SMBC_FILE && + dirp->smbc_type != SMBC_IPC_SHARE && + (strcmp(dirp->name, ".") != 0) && + (strcmp(dirp->name, "..") !=0)){ + + subtree = gtk_tree_new(); + gtk_tree_item_set_subtree(GTK_TREE_ITEM(aitem), subtree); + + gtk_signal_connect(GTK_OBJECT(subtree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), real_tree); + gtk_signal_connect(GTK_OBJECT(subtree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), real_tree); + + } + + (char *)dirp += dirlen; + err -= dirlen; + + } + + } + + smbc_closedir(dh); + + } + else if (strncmp(signame, "collapse", 8) == 0) { + GtkWidget *subtree = gtk_tree_new(); + + gtk_tree_remove_items(GTK_TREE(real_tree), GTK_TREE(real_tree)->children); + + gtk_tree_item_set_subtree(GTK_TREE_ITEM(item), subtree); + + gtk_signal_connect (GTK_OBJECT(subtree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), real_tree); + gtk_signal_connect (GTK_OBJECT(subtree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), real_tree); + + } + +} + +static void cb_selection_changed( GtkWidget *tree ) +{ + GList *i; + + g_print ("selection_change called for tree %p\n", tree); + g_print ("selected objects are:\n"); + + i = GTK_TREE_SELECTION(tree); + while (i){ + gchar *name; + GtkLabel *label; + GtkWidget *item; + + /* Get a GtkWidget pointer from the list node */ + item = GTK_WIDGET (i->data); + label = GTK_LABEL (GTK_BIN (item)->child); + gtk_label_get (label, &name); + g_print ("\t%s on level %d\n", name, GTK_TREE + (item->parent)->level); + i = i->next; + } +} + +/* + * Expand or collapse the whole network ... + */ +static void cb_wholenet(GtkWidget *item, gchar *signame) +{ + GtkWidget *real_tree, *aitem, *subtree; + gchar *name; + GtkLabel *label; + gint dh, err, dirlen; + char dirbuf[512]; + struct smbc_dirent *dirp; + + label = GTK_LABEL (GTK_BIN (item)->child); + gtk_label_get (label, &name); + g_print ("%s called for item %s->%p, level %d\n", signame, name, + item, GTK_TREE (item->parent)->level); + + real_tree = GTK_TREE_ITEM_SUBTREE(item); /* Get the subtree */ + + if (strncmp(signame, "expand", 6) == 0) { /* Expand called */ + + if ((dh = smbc_opendir("smb://")) < 0) { /* Handle error */ + + g_print("cb_wholenet: Could not open dir smbc://, %s\n", + strerror(errno)); + + gtk_main_quit(); + + return; + + } + + while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, + sizeof(dirbuf))) != 0) { + + if (err < 0) { /* An error, report it */ + + g_print("cb_wholenet: Could not read dir smbc://, %s\n", + strerror(errno)); + + gtk_main_quit(); + + return; + + } + + dirp = (struct smbc_dirent *)dirbuf; + + while (err > 0) { + struct tree_data *my_data; + + dirlen = dirp->dirlen; + + my_data = make_tree_data(dirp->smbc_type, dirp->name); + + aitem = gtk_tree_item_new_with_label(dirp->name); + + /* Connect all GtkItem:: and GtkTreeItem:: signals */ + gtk_signal_connect (GTK_OBJECT(aitem), "select", + GTK_SIGNAL_FUNC(cb_itemsignal), "select"); + gtk_signal_connect (GTK_OBJECT(aitem), "deselect", + GTK_SIGNAL_FUNC(cb_itemsignal), "deselect"); + gtk_signal_connect (GTK_OBJECT(aitem), "toggle", + GTK_SIGNAL_FUNC(cb_itemsignal), "toggle"); + gtk_signal_connect (GTK_OBJECT(aitem), "expand", + GTK_SIGNAL_FUNC(cb_itemsignal), "expand"); + gtk_signal_connect (GTK_OBJECT(aitem), "collapse", + GTK_SIGNAL_FUNC(cb_itemsignal), "collapse"); + + gtk_tree_append (GTK_TREE(real_tree), aitem); + /* Show it - this can be done at any time */ + gtk_widget_show (aitem); + + gtk_object_set_user_data(GTK_OBJECT(aitem), (gpointer)my_data); + + fprintf(stdout, "Added: %s, len: %u\n", dirp->name, dirlen); + + subtree = gtk_tree_new(); + + gtk_tree_item_set_subtree(GTK_TREE_ITEM(aitem), subtree); + + gtk_signal_connect(GTK_OBJECT(subtree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), real_tree); + gtk_signal_connect(GTK_OBJECT(subtree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), real_tree); + + (char *)dirp += dirlen; + err -= dirlen; + + } + + } + + smbc_closedir(dh); + + } + else { /* Must be collapse ... FIXME ... */ + GtkWidget *subtree = gtk_tree_new(); + + gtk_tree_remove_items(GTK_TREE(real_tree), GTK_TREE(real_tree)->children); + + gtk_tree_item_set_subtree(GTK_TREE_ITEM(item), subtree); + + gtk_signal_connect (GTK_OBJECT(subtree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), real_tree); + gtk_signal_connect (GTK_OBJECT(subtree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), real_tree); + + + } + +} + +/* Should put up a dialog box to ask the user for username and password */ + +static void +auth_fn(const char *server, const char *share, + char *workgroup, int wgmaxlen, char *username, int unmaxlen, + char *password, int pwmaxlen) +{ + + strncpy(username, "test", unmaxlen); + strncpy(password, "test", pwmaxlen); + +} + +static char *col_titles[] = { + "Name", "Attributes", "Size", "Modification Date", +}; + +int main( int argc, + char *argv[] ) +{ + GtkWidget *window, *scrolled_win, *scrolled_win2, *tree; + GtkWidget *subtree, *item, *main_hbox, *r_pane, *l_pane; + gint err, dh; + gint i; + char dirbuf[512]; + struct smbc_dirent *dirp; + TALLOC_CTX *frame = talloc_stackframe(); + + gtk_init (&argc, &argv); + + /* Init the smbclient library */ + + err = smbc_init(auth_fn, 10); + + /* Print an error response ... */ + + if (err < 0) { + + fprintf(stderr, "smbc_init returned %s (%i)\nDo you have a ~/.smb/smb.conf file?\n", strerror(errno), errno); + exit(1); + + } + + /* a generic toplevel window */ + window = gtk_window_new (GTK_WINDOW_TOPLEVEL); + gtk_widget_set_name(window, "main browser window"); + gtk_signal_connect (GTK_OBJECT(window), "delete_event", + GTK_SIGNAL_FUNC (gtk_main_quit), NULL); + gtk_window_set_title(GTK_WINDOW(window), "The Linux Windows Network Browser"); + gtk_widget_set_usize(GTK_WIDGET(window), 750, -1); + gtk_container_set_border_width (GTK_CONTAINER(window), 5); + + gtk_widget_show (window); + + /* A container for the two panes ... */ + + main_hbox = gtk_hbox_new(FALSE, 1); + gtk_container_border_width(GTK_CONTAINER(main_hbox), 1); + gtk_container_add(GTK_CONTAINER(window), main_hbox); + + gtk_widget_show(main_hbox); + + l_pane = gtk_hpaned_new(); + gtk_paned_gutter_size(GTK_PANED(l_pane), (GTK_PANED(l_pane))->handle_size); + r_pane = gtk_hpaned_new(); + gtk_paned_gutter_size(GTK_PANED(r_pane), (GTK_PANED(r_pane))->handle_size); + gtk_container_add(GTK_CONTAINER(main_hbox), l_pane); + gtk_widget_show(l_pane); + + /* A generic scrolled window */ + scrolled_win = gtk_scrolled_window_new (NULL, NULL); + gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_win), + GTK_POLICY_AUTOMATIC, + GTK_POLICY_AUTOMATIC); + gtk_widget_set_usize (scrolled_win, 150, 200); + gtk_container_add (GTK_CONTAINER(l_pane), scrolled_win); + gtk_widget_show (scrolled_win); + + /* Another generic scrolled window */ + scrolled_win2 = gtk_scrolled_window_new (NULL, NULL); + gtk_scrolled_window_set_policy (GTK_SCROLLED_WINDOW (scrolled_win2), + GTK_POLICY_AUTOMATIC, + GTK_POLICY_AUTOMATIC); + gtk_widget_set_usize (scrolled_win2, 150, 200); + gtk_paned_add2 (GTK_PANED(l_pane), scrolled_win2); + gtk_widget_show (scrolled_win2); + + /* Create the root tree */ + tree = gtk_tree_new(); + g_print ("root tree is %p\n", tree); + /* connect all GtkTree:: signals */ + gtk_signal_connect (GTK_OBJECT(tree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), tree); + gtk_signal_connect (GTK_OBJECT(tree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), tree); + gtk_signal_connect (GTK_OBJECT(tree), "selection_changed", + GTK_SIGNAL_FUNC(cb_selection_changed), tree); + /* Add it to the scrolled window */ + gtk_scrolled_window_add_with_viewport (GTK_SCROLLED_WINDOW(scrolled_win), + tree); + /* Set the selection mode */ + gtk_tree_set_selection_mode (GTK_TREE(tree), + GTK_SELECTION_MULTIPLE); + /* Show it */ + gtk_widget_show (tree); + + /* Now, create a clist and attach it to the second pane */ + + clist = gtk_clist_new_with_titles(4, col_titles); + + gtk_container_add (GTK_CONTAINER(scrolled_win2), clist); + + gtk_widget_show(clist); + + /* Now, build the top level display ... */ + + if ((dh = smbc_opendir("smb:///")) < 0) { + + fprintf(stderr, "Could not list default workgroup: smb:///: %s\n", + strerror(errno)); + + exit(1); + + } + + /* Create a tree item for Whole Network */ + + item = gtk_tree_item_new_with_label ("Whole Network"); + /* Connect all GtkItem:: and GtkTreeItem:: signals */ + gtk_signal_connect (GTK_OBJECT(item), "select", + GTK_SIGNAL_FUNC(cb_itemsignal), "select"); + gtk_signal_connect (GTK_OBJECT(item), "deselect", + GTK_SIGNAL_FUNC(cb_itemsignal), "deselect"); + gtk_signal_connect (GTK_OBJECT(item), "toggle", + GTK_SIGNAL_FUNC(cb_itemsignal), "toggle"); + gtk_signal_connect (GTK_OBJECT(item), "expand", + GTK_SIGNAL_FUNC(cb_wholenet), "expand"); + gtk_signal_connect (GTK_OBJECT(item), "collapse", + GTK_SIGNAL_FUNC(cb_wholenet), "collapse"); + /* Add it to the parent tree */ + gtk_tree_append (GTK_TREE(tree), item); + /* Show it - this can be done at any time */ + gtk_widget_show (item); + + subtree = gtk_tree_new(); /* A subtree for Whole Network */ + + gtk_tree_item_set_subtree(GTK_TREE_ITEM(item), subtree); + + gtk_signal_connect (GTK_OBJECT(subtree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), tree); + gtk_signal_connect (GTK_OBJECT(subtree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), tree); + + /* Now, get the items in smb:/// and add them to the tree */ + + dirp = (struct smbc_dirent *)dirbuf; + + while ((err = smbc_getdents(dh, (struct smbc_dirent *)dirbuf, + sizeof(dirbuf))) != 0) { + + if (err < 0) { /* Handle the error */ + + fprintf(stderr, "Could not read directory for smbc:///: %s\n", + strerror(errno)); + + exit(1); + + } + + fprintf(stdout, "Dir len: %u\n", err); + + while (err > 0) { /* Extract each entry and make a sub-tree */ + struct tree_data *my_data; + int dirlen = dirp->dirlen; + + my_data = make_tree_data(dirp->smbc_type, dirp->name); + + item = gtk_tree_item_new_with_label(dirp->name); + /* Connect all GtkItem:: and GtkTreeItem:: signals */ + gtk_signal_connect (GTK_OBJECT(item), "select", + GTK_SIGNAL_FUNC(cb_itemsignal), "select"); + gtk_signal_connect (GTK_OBJECT(item), "deselect", + GTK_SIGNAL_FUNC(cb_itemsignal), "deselect"); + gtk_signal_connect (GTK_OBJECT(item), "toggle", + GTK_SIGNAL_FUNC(cb_itemsignal), "toggle"); + gtk_signal_connect (GTK_OBJECT(item), "expand", + GTK_SIGNAL_FUNC(cb_itemsignal), "expand"); + gtk_signal_connect (GTK_OBJECT(item), "collapse", + GTK_SIGNAL_FUNC(cb_itemsignal), "collapse"); + /* Add it to the parent tree */ + gtk_tree_append (GTK_TREE(tree), item); + /* Show it - this can be done at any time */ + gtk_widget_show (item); + + gtk_object_set_user_data(GTK_OBJECT(item), (gpointer)my_data); + + fprintf(stdout, "Added: %s, len: %u\n", dirp->name, dirlen); + + subtree = gtk_tree_new(); + + gtk_tree_item_set_subtree(GTK_TREE_ITEM(item), subtree); + + gtk_signal_connect (GTK_OBJECT(subtree), "select_child", + GTK_SIGNAL_FUNC(cb_select_child), tree); + gtk_signal_connect (GTK_OBJECT(subtree), "unselect_child", + GTK_SIGNAL_FUNC(cb_unselect_child), tree); + + (char *)dirp += dirlen; + err -= dirlen; + + } + + } + + smbc_closedir(dh); /* FIXME, check for error :-) */ + + /* Show the window and loop endlessly */ + gtk_main(); + TALLOC_FREE(frame); + return 0; +} +/* example-end */ |