1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-25 23:21:54 +03:00
samba-mirror/source3/client/clitar.c
Aurélien Aptel cbad767ded clitar.c: implement reset mode
Signed-off-by: Aurélien Aptel <aurelien.aptel@gmail.com>
Reviewed-by: David Disseldorp <ddiss@samba.org>
Reviewed-by: Andreas Schneider <asn@samba.org>
2014-02-19 18:22:27 +01:00

1191 lines
31 KiB
C

/*
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/>.
*/
#include "includes.h"
#include "system/filesys.h"
#include "client/client_proto.h"
#include "client/clitar_proto.h"
#include "libsmb/libsmb.h"
#include <archive.h>
#include <archive_entry.h>
#define LEN(x) (sizeof(x)/sizeof((x)[0]))
#define DBG(a, b) (DEBUG(a, ("tar:%-3d ", __LINE__)), DEBUG(a, b))
/**
* Maximum value for the blocksize field
*/
#define TAR_MAX_BLOCK_SIZE 0xffff
/**
* Default tar block size in bytes. Hasn't changed since the first
* commit in 1996...
*
* A more adequate size will be used for better performance unless
* we're dealing with a tape device with a fixed read/write block
* size.
*
* The actual choice is made by libarchive.
*/
#define TAR_DEFAULT_BLOCK_SIZE 20
#define TAR_CLI_READ_SIZE 0xff00
#define TAR_DO_LIST_ATTRIBUTE (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 flag, default */
TAR_INCLUDE_LIST, /* F flag */
TAR_EXCLUDE, /* X flag */
};
enum {
ATTR_UNSET,
ATTR_SET,
};
struct tar {
/* in state that needs/can be processed? */
bool to_process;
/* flags */
struct tar_mode {
enum tar_operation operation; /* create, extract */
enum tar_selection selection; /* inc, inc from file, exclude */
int blocksize; /* size in bytes of a block in the tar file */
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;
} 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;
};
struct tar tar_ctx = {
.mode.selection = TAR_INCLUDE,
.mode.blocksize = TAR_DEFAULT_BLOCK_SIZE,
.mode.hidden = true,
.mode.system = true,
.mode.incremental = false,
.mode.dry = false,
};
static char *fix_unix_path (char *path, bool removeprefix)
{
char *from = path, *to = path;
if (!path || !*path)
return path;
/* remove prefix:
* ./path => path
* /path => path
*/
if (removeprefix) {
/* /path */
if (path[0] == '/' || path[0] == '\\') {
from += 1;
}
/* ./path */
if (path[1] && path[0] == '.' && (path[1] == '/' || path[1] == '\\')) {
from += 2;
}
}
/* replace / with \ */
while (*from) {
if (*from == '/') {
*to = '\\';
} else {
*to = *from;
}
from++; to++;
}
*to = 0;
return path;
}
#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))
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_INCLUDE_LIST),
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
static void tar_add_selection_path(struct tar *t, const char *path)
{
TALLOC_CTX *ctx = talloc_tos();
if(!t->path_list) {
t->path_list = str_list_make_empty(ctx);
t->path_list_size = 0;
}
t->path_list = str_list_add((const char**)t->path_list, path);
t->path_list_size++;
fix_unix_path(t->path_list[t->path_list_size - 1], true);
}
static int tar_set_blocksize(struct tar *t, int size)
{
if (size <= 0 || size > TAR_MAX_BLOCK_SIZE) {
return 0;
}
t->mode.blocksize = size;
return 1;
}
static bool tar_set_newer_than(struct tar *t, const char *filename)
{
extern time_t newer_than;
SMB_STRUCT_STAT stbuf;
if (sys_stat(filename, &stbuf, false) != 0) {
DBG(0, ("Error setting newer-than time\n"));
return 0;
}
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 1;
}
static bool tar_read_inclusion_file (struct tar *t, const char* filename)
{
char *line;
TALLOC_CTX *ctx = talloc_tos();
int fd = open(filename, O_RDONLY);
if (fd < 0) {
DBG(0, ("Can't open inclusion file '%s': %s\n", filename, strerror(errno)));
return 0;
}
while ((line = afdgets(fd, ctx, 0))) {
tar_add_selection_path(t, line);
}
close(fd);
return 1;
}
/* skip leading slashes or dots */
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;
}
/**
* return true if the path @sub is a subpath of @full.
*
* case-insensitive, true if @sub = @full
*/
static bool is_subpath(const char *sub, const char *full)
{
const char *full_copy = full;
while(*full && *sub && (*full == *sub || tolower_m(*full) == tolower_m(*sub) ||
(*full == '\\' && *sub=='/') || (*full == '/' && *sub=='\\'))) {
full++; sub++;
}
/* if full has a trailing slash, it compared equal, so full is an "initial"
string of sub.
*/
if (!*full && full != full_copy && (*(full-1) == '/' || *(full-1) == '\\'))
return true;
/* ignore trailing slash on full */
if (!*sub && (*full == '/' || *full == '\\') && !*(full+1))
return true;
/* check for full is an "initial" string of sub */
if ((*sub == '/' || *sub == '\\') && !*full)
return true;
return *full == *sub;
}
static bool tar_path_in_list(struct tar *t, const char *path)
{
int i;
const char *p = path;
const char *pattern;
if (!p || !p[0])
return false;
p = skip_useless_char_in_path(p);
for (i = 0; i < t->path_list_size; i++) {
pattern = skip_useless_char_in_path(t->path_list[i]);
if (is_subpath(p, pattern)) {
return true;
}
}
return false;
}
static bool tar_extract_skip_path(struct tar *t,
struct archive_entry *entry)
{
const bool skip = true;
const char *fullpath = archive_entry_pathname(entry);
bool in;
in = t->path_list_size > 0 ? tar_path_in_list(t, fullpath) : true;
if (t->mode.selection == TAR_EXCLUDE) {
in = !in;
}
return in ? !skip : skip;
}
static bool tar_create_skip_path(struct tar *t,
const char *fullpath,
const struct file_info *finfo)
{
/* syntaxic sugar */
const bool skip = true;
mode_t mode = finfo->mode;
bool in;
/* 1. if we dont want X and we have X, skip */
if (!t->mode.system && (mode & FILE_ATTRIBUTE_SYSTEM)) {
return skip;
}
if (!t->mode.hidden && (mode & FILE_ATTRIBUTE_HIDDEN)) {
return skip;
}
/* 2. if we only want archive and it's not, skip */
if (t->mode.incremental && !(mode & FILE_ATTRIBUTE_ARCHIVE)) {
return skip;
}
/* 3. is it in the selection list? */
in = t->path_list_size > 0 ? tar_path_in_list(t, fullpath) : true;
/* inverse result if in exclude mode */
if (t->mode.selection == TAR_EXCLUDE) {
in = !in;
}
return in ? !skip : skip;
}
bool tar_to_process (struct tar *t)
{
return t->to_process;
}
/**
* 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;
TALLOC_CTX *ctx = talloc_tos();
if (!next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) {
DBG(0, ("blocksize <n>\n"));
return 1;
}
if(!tar_set_blocksize(&tar_ctx, atoi(buf))) {
DBG(0, ("invalid blocksize\n"));
}
DBG(2, ("blocksize is now %d\n", tar_ctx.mode.blocksize));
return 0;
}
/**
* 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;
int i;
TALLOC_CTX *ctx = talloc_tos();
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, true },
{"noquiet", &tar_ctx.mode.verbose, true },
{"quiet", &tar_ctx.mode.verbose, false},
{"noverbose", &tar_ctx.mode.verbose, false},
};
while (next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) {
for (i = 0; i < LEN(table); i++) {
if (strequal(table[i].cmd, buf)) {
*table[i].p = table[i].value;
break;
}
}
if (i == LEN(table))
DBG(0, ("tarmode: unrecognised option %s\n", buf));
TALLOC_FREE(buf);
}
DBG(0, ("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" : "quiet"));
return 0;
}
/**
* 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.
*/
static void set_remote_attr(const char *filename, uint16 new_attr, int mode)
{
extern struct cli_state *cli;
uint16 old_attr;
NTSTATUS status;
if (!NT_STATUS_IS_OK(cli_getatr(cli, filename, &old_attr, NULL, NULL))) {
/* XXX: debug message */
return;
}
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)) {
DBG(1, ("setatr failed: %s\n", nt_errstr(status)));
}
}
/**
* 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)
{
const extern char *cmd_ptr;
char *buf;
char *fname = NULL;
uint16 attr[2] = {0};
int mode = ATTR_SET;
TALLOC_CTX *ctx = talloc_tos();
if (!next_token_talloc(ctx, &cmd_ptr, &buf, NULL)) {
DBG(0, ("setmode <filename> <[+|-]rsha>\n"));
return 1;
}
fname = talloc_asprintf(ctx,
"%s%s",
client_get_cur_dir(),
buf);
if (!fname) {
return 1;
}
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:
DBG(0, ("setmode <filename> <perm=[+|-]rsha>\n"));
return 1;
}
}
}
if (attr[ATTR_SET] == 0 && attr[ATTR_UNSET] == 0) {
DBG(0, ("setmode <filename> <[+|-]rsha>\n"));
return 1;
}
DBG(2, ("\nperm set %d %d\n", attr[ATTR_SET], attr[ATTR_UNSET]));
set_remote_attr(fname, attr[ATTR_SET], ATTR_SET);
set_remote_attr(fname, attr[ATTR_UNSET], ATTR_UNSET);
return 0;
}
static int make_remote_path(const char *full_path)
{
extern struct cli_state *cli;
TALLOC_CTX *ctx = talloc_tos();
char *path;
char *subpath;
char *state;
char *last_backslash;
char *p;
int len;
NTSTATUS status;
int err = 0;
subpath = talloc_strdup(ctx, full_path);
path = talloc_strdup(ctx, full_path);
len = talloc_get_size(path) - 1;
last_backslash = strrchr_m(path, '\\');
if (!last_backslash) {
goto out;
}
*last_backslash = 0;
subpath[0] = 0;
p = strtok_r(path, "\\", &state);
while (p) {
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)) {
DBG(0, ("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:
return err;
}
static int tar_send_file(struct tar *t, struct archive_entry *entry)
{
extern struct cli_state *cli;
TALLOC_CTX *ctx = talloc_tos();
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 mode = archive_entry_filetype(entry);
dos_path = talloc_strdup(ctx, archive_entry_pathname(entry));
fix_unix_path(dos_path, true);
full_path = talloc_strdup(ctx, client_get_cur_dir());
full_path = talloc_strdup_append(full_path, dos_path);
if (mode != AE_IFREG && mode != AE_IFDIR) {
DBG(0, ("Skipping non-dir & non-regular file %s\n", full_path));
goto out;
}
if (make_remote_path(full_path)) {
err = 1;
goto out;
}
if (mode == AE_IFDIR) {
goto out;
}
status = cli_open(cli, full_path, flags, DENY_NONE, &remote_fd);
if (!NT_STATUS_IS_OK(status)) {
DBG(0, ("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) {
DBG(0, ("Warning: %s\n", archive_error_string(t->archive)));
}
if (r == ARCHIVE_FATAL) {
DBG(0, ("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)) {
DBG(0, ("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)) {
DBG(0, ("Error losing remote file %s: %s\n",
full_path, nt_errstr(status)));
err = 1;
}
out:
return err;
}
static int tar_get_file(struct tar *t, const char *full_dos_path,
struct file_info *finfo)
{
extern struct cli_state *cli;
TALLOC_CTX *ctx = talloc_tos();
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->mode & FILE_ATTRIBUTE_DIRECTORY;
DBG(5, ("+++ %s\n", full_dos_path));
t->total_size += finfo->size;
if (t->mode.dry) {
goto out;
}
if (t->mode.reset) {
set_remote_attr(full_dos_path, FILE_ATTRIBUTE_ARCHIVE, ATTR_UNSET);
}
full_unix_path = talloc_asprintf(ctx, ".%s", full_dos_path);
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) {
DBG(0, ("Remote file %s too big\n", full_dos_path));
goto out_entry;
}
archive_entry_set_size(entry, (int64_t)finfo->size);
r = archive_write_header(t->archive, entry);
if (r != ARCHIVE_OK) {
DBG(0, ("Fatal: %s\n", archive_error_string(t->archive)));
err = 1;
goto out_entry;
}
if (isdir) {
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)) {
DBG(0,("%s opening remote file %s\n",
nt_errstr(status), full_dos_path));
goto out_entry;
}
do {
status = cli_read(cli, remote_fd, buf, off, sizeof(buf), &len);
if (!NT_STATUS_IS_OK(status)) {
DBG(0,("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) {
DBG(0, ("Fatal: %s\n", archive_error_string(t->archive)));
err = 1;
goto out_close;
}
} while (off < finfo->size);
out_close:
cli_close(cli, remote_fd);
out_entry:
archive_entry_free(entry);
out:
return err;
}
static NTSTATUS get_file_callback(struct cli_state *cli,
struct file_info *finfo,
const char *dir)
{
TALLOC_CTX *ctx = talloc_tos();
NTSTATUS err = NT_STATUS_OK;
char *remote_name;
remote_name = talloc_asprintf(ctx, "%s%s",
client_get_cur_dir(), finfo->name);
if (strequal(finfo->name, "..") || strequal(finfo->name, ".")) {
goto out;
}
if (tar_create_skip_path(&tar_ctx, remote_name, finfo)) {
DBG(5, ("--- %s\n", remote_name));
goto out;
}
if (finfo->mode & FILE_ATTRIBUTE_DIRECTORY) {
char *old_dir;
char *new_dir;
char *mask;
old_dir = talloc_strdup(ctx, client_get_cur_dir());
new_dir = talloc_asprintf(ctx, "%s%s\\", client_get_cur_dir(), finfo->name);
mask = talloc_asprintf(ctx, "%s*", new_dir);
if (tar_get_file(&tar_ctx, remote_name, finfo)) {
err = NT_STATUS_UNSUCCESSFUL;
goto out;
}
client_set_cur_dir(new_dir);
do_list(mask, TAR_DO_LIST_ATTRIBUTE, get_file_callback, false, true);
client_set_cur_dir(old_dir);
}
else {
if (tar_get_file(&tar_ctx, remote_name, finfo)) {
err = NT_STATUS_UNSUCCESSFUL;
goto out;
}
}
out:
return err;
}
static int tar_create(struct tar* t)
{
TALLOC_CTX *ctx = talloc_tos();
char *mask = talloc_asprintf(ctx, "%s\\*", client_get_cur_dir());
int r;
int err = 0;
NTSTATUS status;
t->archive = archive_write_new();
if (!t->mode.dry) {
r = archive_write_set_format_pax_restricted(t->archive);
if (r != ARCHIVE_OK) {
DBG(0, ("Can't open %s: %s\n", t->tar_path, archive_error_string(t->archive)));
}
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) {
DBG(0, ("Can't open %s: %s\n", t->tar_path, archive_error_string(t->archive)));
err = 1;
goto out_close;
}
}
DBG(5, ("tar_process do_list with mask: %s\n", mask));
status = do_list(mask, TAR_DO_LIST_ATTRIBUTE, get_file_callback, false, true);
if(!NT_STATUS_IS_OK(status)) {
DBG(0, ("do_list fail %s\n", nt_errstr(status)));
err = 1;
goto out_close;
}
out_close:
DBG(0, ("Total bytes received: %" PRIu64 "\n", t->total_size));
r = archive_write_close(t->archive);
if (r != ARCHIVE_OK) {
DBG(0, ("Fatal: %s\n", archive_error_string(t->archive)));
err = 1;
goto out;
}
out:
archive_write_free(t->archive);
return err;
}
/**
* cmd_tar - interactive command to start a tar backup/restoration
*
* Check presence of argument, parse them and handle the request.
*/
int cmd_tar(void)
{
return 0;
}
static int tar_extract(struct tar *t)
{
int err = 0;
int r;
struct archive_entry *entry;
t->archive = archive_read_new();
archive_read_support_format_all(t->archive);
archive_read_support_filter_all(t->archive);
if (strequal(t->tar_path, "-")) {
r = archive_read_open_fd(t->archive, STDIN_FILENO, t->mode.blocksize);
} else {
r = archive_read_open_filename(t->archive, t->tar_path,
t->mode.blocksize);
}
if (r != ARCHIVE_OK) {
DBG(0, ("Can't open %s : %s\n", t->tar_path,
archive_error_string(t->archive)));
err = 1;
goto out;
}
for (;;) {
r = archive_read_next_header(t->archive, &entry);
if (r == ARCHIVE_EOF) {
break;
}
if (r == ARCHIVE_WARN) {
DBG(0, ("Warning: %s\n", archive_error_string(t->archive)));
}
if (r == ARCHIVE_FATAL) {
DBG(0, ("Fatal: %s\n", archive_error_string(t->archive)));
err = 1;
goto out;
}
if (tar_extract_skip_path(t, entry)) {
DBG(5, ("--- %s\n", archive_entry_pathname(entry)));
continue;
}
DBG(5, ("+++ %s\n", archive_entry_pathname(entry)));
if (tar_send_file(t, entry)) {
err = 1;
goto out;
}
}
out:
r = archive_read_free(t->archive);
if (r != ARCHIVE_OK) {
DBG(0, ("Can't close %s : %s\n", t->tar_path,
archive_error_string(t->archive)));
err = 1;
}
return err;
}
int tar_process(struct tar *t)
{
int rc = 0;
switch(t->mode.operation) {
case TAR_EXTRACT:
rc = tar_extract(t);
break;
case TAR_CREATE:
rc = tar_create(t);
break;
default:
DBG(0, ("Invalid tar state\n"));
rc = 1;
}
DBG(5, ("tar_process done, err = %d\n", rc));
return rc;
}
/**
* 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...
*
* opt 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 = talloc_tos();
/* index of next value to use */
int ival = 0;
/*
* 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) {
switch(*flag++) {
/* operation */
case 'c':
if (t->mode.operation != TAR_NO_OPERATION) {
printf("Tar must be followed by only one of c or x.\n");
return 0;
}
t->mode.operation = TAR_CREATE;
break;
case 'x':
if (t->mode.operation != TAR_NO_OPERATION) {
printf("Tar must be followed by only one of c or x.\n");
return 0;
}
t->mode.operation = TAR_EXTRACT;
break;
/* selection */
case 'I':
if (t->mode.selection != TAR_NO_SELECTION) {
DBG(0,("Only one of I,X,F must be specified\n"));
return 0;
}
t->mode.selection = TAR_INCLUDE;
break;
case 'X':
if (t->mode.selection != TAR_NO_SELECTION) {
DBG(0,("Only one of I,X,F must be specified\n"));
return 0;
}
t->mode.selection = TAR_EXCLUDE;
break;
case 'F':
if (t->mode.selection != TAR_NO_SELECTION) {
DBG(0,("Only one of I,X,F must be specified\n"));
return 0;
}
t->mode.selection = TAR_INCLUDE_LIST;
break;
/* blocksize */
case 'b':
if (ival >= valsize) {
DBG(0, ("Option b must be followed by a blocksize\n"));
return 0;
}
if (!tar_set_blocksize(t, atoi(val[ival]))) {
DBG(0, ("Option b must be followed by a valid blocksize\n"));
return 0;
}
ival++;
break;
/* incremental mode */
case 'g':
t->mode.incremental = true;
break;
/* newer than */
case 'N':
if (ival >= valsize) {
DBG(0, ("Option N must be followed by valid file name\n"));
return 0;
}
if (!tar_set_newer_than(t, val[ival])) {
DBG(0,("Error setting newer-than time\n"));
return 0;
}
ival++;
break;
/* reset mode */
case 'a':
t->mode.reset = true;
break;
/* verbose */
case 'q':
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) {
DBG(0, ("n is only meaningful when creating a tar-file\n"));
return 0;
}
t->mode.dry = true;
DBG(0, ("dry_run set\n"));
break;
default:
DBG(0,("Unknown tar option\n"));
return 0;
}
}
/* no selection given? default selection is include */
if (t->mode.selection == TAR_NO_SELECTION) {
t->mode.selection = TAR_INCLUDE;
}
if (valsize - ival < 1) {
DBG(0, ("No tar file given.\n"));
return 0;
}
/* handle TARFILE */
t->tar_path = talloc_strdup(ctx, val[ival]);
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 (t->mode.selection == TAR_INCLUDE_LIST) {
if (valsize - ival != 1) {
DBG(0,("Option F must be followed by exactly one filename.\n"));
return 0;
}
if (!tar_read_inclusion_file(t, val[ival])) {
return 0;
}
ival++;
}
/* otherwise store all the PATHs on the command line */
else {
int i;
for (i = ival; i < valsize; i++) {
tar_add_selection_path(t, val[i]);
}
}
t->to_process = true;
tar_dump(t);
return 1;
}