/*

 * Amanda, The Advanced Maryland Automatic Network Disk Archiver

 * Copyright (c) 1991-1998, 2000 University of Maryland at College Park

 * Copyright (c) 2007-2012 Zmanda, Inc.  All Rights Reserved.

 * Copyright (c) 2013-2016 Carbonite, Inc.  All Rights Reserved.

 * All Rights Reserved.

 *

 * Permission to use, copy, modify, distribute, and sell this software and its

 * documentation for any purpose is hereby granted without fee, provided that

 * the above copyright notice appear in all copies and that both that

 * copyright notice and this permission notice appear in supporting

 * documentation, and that the name of U.M. not be used in advertising or

 * publicity pertaining to distribution of the software without specific,

 * written prior permission.  U.M. makes no representations about the

 * suitability of this software for any purpose.  It is provided "as is"

 * without express or implied warranty.

 *

 * U.M. DISCLAIMS ALL WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL U.M.

 * BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES

 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION

 * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN

 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

 *

 * Authors: the Amanda Development Team.  Its members are listed in a

 * file named AUTHORS, in the root directory of this distribution.

 */

/*

 * $Id: set_commands.c,v 1.3 2006/07/05 13:14:58 martinea Exp $

 *

 * implements the "set" commands in amrecover

 */

#include "amanda.h"

#include "match.h"

#include "amutil.h"

#include "amrecover.h"

#ifdef SAMBA_CLIENT

extern unsigned short samba_extract_method;

#endif /* SAMBA_CLIENT */

/* sets a date, mapping given date into standard form if needed */

int

set_date(

    char *	date)

{

    char *cmd = NULL;

    clear_dir_list();

    cmd = g_strconcat("DATE ", date, NULL);

    if (converse(cmd) == -1)

	exit(1);

    /* if a host/disk/directory is set, then check if that directory

       is still valid at the new date, and if not set directory to

       mount_point */

    if (disk_path != NULL) {

	g_free(cmd);

	cmd = g_strconcat("OISD ", disk_path, NULL);

	if (exchange(cmd) == -1)

	    exit(1);

	if (server_happy())

	{

	    suck_dir_list_from_server();

	}

	else

	{

	    g_printf(_("No index records for cwd on new date\n"));

	    g_printf(_("Setting cwd to mount point\n"));

	    g_free(disk_path);

	    disk_path = g_strdup("/");	/* fake it */

	    clear_dir_list();

	}

    }

    amfree(cmd);

    return 0;

}

void

set_host(

    const char *	host)

{

    char *cmd = NULL;

    struct hostent *hp;

    char **hostp;

    int found_host = 0;

    char *uqhost;

    if (is_extract_list_nonempty())

    {

	g_printf(_("Must clear extract list before changing host\n"));

	return;

    }

    uqhost = unquote_string(host);

    cmd = g_strconcat("HOST ", uqhost, NULL);

    if (converse(cmd) == -1)

	exit(1);

    if (server_happy())

    {

	found_host = 1;

    }

    else

    {

	/*

	 * Try converting the given host to a fully qualified name

	 * and then try each of the aliases.

	 */

	if ((hp = gethostbyname(uqhost)) != NULL) {

	    host = hp->h_name;

	    g_printf(_("Trying host %s ...\n"), host);

	    g_free(cmd);

	    cmd = g_strconcat("HOST ", host, NULL);

	    if (converse(cmd) == -1)

		exit(1);

	    if(server_happy())

	    {

		found_host = 1;

	    }

	    else

	    {

	        for (hostp = hp->h_aliases; (host = *hostp) != NULL; hostp++)

	        {

		    g_printf(_("Trying host %s ...\n"), host);

		    g_free(cmd);

		    cmd = g_strconcat("HOST ", host, NULL);

		    if (converse(cmd) == -1)

		        exit(1);

		    if(server_happy())

		    {

		        found_host = 1;

		        break;

		    }

		}

	    }

	}

    }

    if(found_host)

    {

	g_free(dump_hostname);

	dump_hostname = g_strdup(host);

	amfree(disk_name);

	amfree(mount_point);

	amfree(disk_path);

	clear_dir_list();

    }

    amfree(cmd);

    amfree(uqhost);

}

void

list_host(void)

{

    char *cmd = NULL;

    cmd = g_strdup("LISTHOST");

    if (converse(cmd) == -1)

        exit(1);

    amfree(cmd);

}

void

set_disk(

    char *	dsk,

    char *	mtpt)

{

    char *cmd = NULL;

    char *uqdsk;

    char *uqmtpt = NULL;

    if (is_extract_list_nonempty())

    {

	g_printf(_("Must clear extract list before changing disk\n"));

	return;

    }

    /* if mount point specified, check it is valid */

    if (mtpt != NULL) {

	uqmtpt = unquote_string(mtpt);

	if (*mtpt != '/') {

	    g_printf(_("Mount point \"%s\" invalid - must start with /\n"), uqmtpt);

	    amfree(uqmtpt);

	    return;

	}

    }

    clear_dir_list();

    uqdsk = unquote_string(dsk);

    cmd = g_strconcat("DISK ", uqdsk, NULL);

    if (converse(cmd) == -1)

	exit(1);

    amfree(cmd);

    if (!server_happy()) {

	amfree(uqdsk);

	amfree(uqmtpt);

	return;

    }

    g_free(disk_name);

    disk_name = g_strdup(uqdsk);

    if (mtpt == NULL)

    {

	/* mount point not specified */

	if (*uqdsk == '/')

	{

	    /* disk specified by mount point, hence use it */

	    g_free(mount_point);

	    mount_point = g_strdup(uqdsk);

	}

	else

	{

	    /* device name given, use '/' because nothing better */

	    g_free(mount_point);

	    mount_point = g_strdup("/");

	}

    }

    else

    {

	/* mount point specified */

	g_free(mount_point);

	mount_point = g_strdup(uqmtpt);

    }

    /* set the working directory to the mount point */

    /* there is the possibility that there are no index records for the

       disk for the given date, hence setting the directory to the

       mount point will fail. Preempt this by checking first so we can write

       a more informative message. */

    if (exchange("OISD /") == -1)

	exit(1);

    if (server_happy())

    {

	g_free(disk_path);

	disk_path = g_strdup("/");

	suck_dir_list_from_server();	/* get list of directory contents */

    }

    else

    {

	g_printf(_("No index records for disk for specified date\n"));

	g_printf(_("If date correct, notify system administrator\n"));

	g_free(disk_path);

	disk_path = g_strdup("/");	/* fake it */

	clear_dir_list();

    }

    amfree(uqmtpt);

    amfree(uqdsk);

}

void

list_disk(

    char *	amdevice)

{

    char *cmd = NULL;

    char *uqamdevice;

    if(amdevice) {

	uqamdevice = unquote_string(amdevice);

	cmd = g_strconcat("LISTDISK ", uqamdevice, NULL);

	amfree(uqamdevice);

	if (converse(cmd) == -1)

	    exit(1);

	amfree(cmd);

    }

    else {

	cmd = g_strdup("LISTDISK");

	if (converse(cmd) == -1)

	    exit(1);

	amfree(cmd);

    }

}

void

local_cd(

    char *dir)

{

    char *uqdir = unquote_string(dir);

    if (chdir(uqdir) == -1) {

	perror(uqdir);

    }

    amfree(uqdir);

}

void

cd_glob(

    char *	glob)

{

    char *regex;

    char *regex_path;

    char *s;

    char *uqglob;

    char *path_on_disk = NULL;

    if (disk_name == NULL) {

	g_printf(_("Must select disk before changing directory\n"));

	return;

    }

    uqglob = unquote_string(glob);

    regex = glob_to_regex(uqglob);

    dbprintf(_("cd_glob (%s) -> %s\n"), uqglob, regex);

    if ((s = validate_regexp(regex)) != NULL) {

        g_printf(_("\"%s\" is not a valid shell wildcard pattern: "), glob);

        puts(s);

	amfree(regex);

	amfree(uqglob);

        return;

    }

    /*

     * glob_to_regex() anchors the beginning of the pattern with ^,

     * but we will be tacking it onto the end of the current directory

     * in add_file, so strip that off.  Also, it anchors the end with

     * $, but we need to match a trailing /, add it if it is not there

     */

    regex_path = g_strdup(regex + 1);

    amfree(regex);

    if(regex_path[strlen(regex_path) - 2] != '/' ) {

	regex_path[strlen(regex_path) - 1] = '\0';

	strappend(regex_path, "/$");

    }

    /* convert path (assumed in cwd) to one on disk */

    if (g_str_equal(disk_path, "/"))

        path_on_disk = g_strconcat("/", regex_path, NULL);

    else {

        char *clean_disk_path = clean_regex(disk_path, 0);

        path_on_disk = g_strjoin(NULL, clean_disk_path, "/", regex_path, NULL);

        amfree(clean_disk_path);

    }

    cd_dir(path_on_disk, uqglob);

    amfree(regex_path);

    amfree(path_on_disk);

    amfree(uqglob);

}

void

cd_regex(

    char *	regex)

{

    char *s;

    char *uqregex;

    char *path_on_disk = NULL;

    if (disk_name == NULL) {

	g_printf(_("Must select disk before changing directory\n"));

	return;

    }

    uqregex = unquote_string(regex);

    if ((s = validate_regexp(uqregex)) != NULL) {

	g_printf(_("\"%s\" is not a valid regular expression: "), uqregex);

	amfree(uqregex);

	puts(s);

	return;

    }

    /* convert path (assumed in cwd) to one on disk */

    if (g_str_equal(disk_path, "/"))

        path_on_disk = g_strconcat("/", regex, NULL);

    else {

        char *clean_disk_path = clean_regex(disk_path, 0);

        path_on_disk = g_strjoin(NULL, clean_disk_path, "/", regex, NULL);

        amfree(clean_disk_path);

    }

    cd_dir(path_on_disk, uqregex);

    amfree(path_on_disk);

    amfree(uqregex);

}

void

cd_dir(

    char *	path_on_disk,

    char *	default_dir)

{

    char *path_on_disk_slash = NULL;

    char *dir = NULL;

    int nb_found;

    size_t i;

    DIR_ITEM *ditem;

    path_on_disk_slash = g_strconcat(path_on_disk, "/", NULL);

    nb_found = 0;

    for (ditem=get_dir_list(); ditem!=NULL && nb_found <= 1; 

			       ditem=get_next_dir_item(ditem))

    {

	if (match(path_on_disk, ditem->path)

	    || match(path_on_disk_slash, ditem->path))

	{

	    i = strlen(ditem->path);

	    if((i > 0 && ditem->path[i-1] == '/')

               || (i > 1 && ditem->path[i-2] == '/' && ditem->path[i-1] == '.'))

            {   /* It is a directory */

		char *dir1, *dir2;

		nb_found++;

		g_free(dir);

		dir = g_strdup(ditem->path);

		if(dir[strlen(dir)-1] == '/')

		    dir[strlen(dir)-1] = '\0'; /* remove last / */

		/* remove everything before the last / */

		dir1 = strrchr(dir,'/');

		if (dir1) {

		    dir1++;

		    dir2 = g_strdup(dir1);

		    amfree(dir);

		    dir = dir2;

		}

	    }

	}

    }

    amfree(path_on_disk_slash);

    if(nb_found==0) {

	set_directory(default_dir);

    }

    else if(nb_found==1) {

	set_directory(dir);

    }

    else {

	g_printf(_("Too many directory\n"));

    }

    amfree(dir);

}

void

set_directory(

    char *	dir)

{

    char *cmd = NULL;

    char *new_dir = NULL;

    char *dp, *de;

    char *ldir = NULL;

    /* do nothing if "." */

    if(g_str_equal(dir, ".")) {

	show_directory();		/* say where we are */

	return;

	/*NOTREACHED*/

    }

    if (disk_name == NULL) {

	g_printf(_("Must select disk before setting directory\n"));

	return;

	/*NOTREACHED*/

    }

    ldir = g_strdup(dir);

    clean_pathname(ldir);

    /* convert directory into absolute path relative to disk mount point */

    if (ldir[0] == '/')

    {

	/* absolute path specified, must start with mount point */

	if (g_str_equal(mount_point, "/"))

	{

	    new_dir = g_strdup(ldir);

	}

	else

	{

	    if (strncmp(mount_point, ldir, strlen(mount_point)) != 0)

	    {

		g_printf(_("Invalid directory - Can't cd outside mount point \"%s\"\n"),

		       mount_point);

		amfree(ldir);

		return;

		/*NOTREACHED*/

	    }

	    new_dir = g_strdup(ldir+strlen(mount_point));

	    if (strlen(new_dir) == 0) {

		g_free(new_dir);

		new_dir = g_strdup("/");

					/* i.e. ldir == mount_point */

	    }

	}

    }

    else

    {

	new_dir = g_strdup(disk_path);

	dp = ldir;

	/* strip any leading ..s */

	while (g_str_has_prefix(dp, "../"))

	{

	    de = strrchr(new_dir, '/');	/* always at least 1 */

	    if (de == new_dir)

	    {

		/* at top of disk */

		*(de + 1) = '\0';

		dp = dp + 3;

	    }

	    else

	    {

		*de = '\0';

		dp = dp + 3;

	    }

	}

	if (g_str_equal(dp, "..")) {

	    if (g_str_equal(new_dir, "/")) {

		/* at top of disk */

		g_printf(_("Invalid directory - Can't cd outside mount point \"%s\"\n"),

		       mount_point);

		/*@ignore@*/

		amfree(new_dir);

		/*@end@*/

		amfree(ldir);

		return;

		/*NOTREACHED*/

	    }

	    de = strrchr(new_dir, '/');	/* always at least 1 */

	    if (de == new_dir)

	    {

		/* at top of disk */

		*(de+1) = '\0';

	    }

	    else

	    {

		*de = '\0';

 	    }

	} else {

	    /*@ignore@*/

	    if (!g_str_equal(new_dir, "/")) {

		strappend(new_dir, "/");

	    }

	    strappend(new_dir, ldir);

	    /*@end@*/

	}

    }

    cmd = g_strconcat("OISD ", new_dir, NULL);

    if (exchange(cmd) == -1) {

	exit(1);

	/*NOTREACHED*/

    }

    amfree(cmd);

    if (server_happy())

    {

	g_free(disk_path);

	disk_path = g_strdup(new_dir);

	suck_dir_list_from_server();	/* get list of directory contents */

	show_directory();		/* say where we moved to */

    }

    else

    {

	g_printf(_("Invalid directory - %s\n"), dir);

    }

    /*@ignore@*/

    amfree(new_dir);

    amfree(ldir);

    /*@end@*/

}

/* prints the current working directory */

void

show_directory(void)

{

    if (mount_point == NULL || disk_path == NULL)

        g_printf(_("Must select disk first\n"));

    else if (g_str_equal(mount_point, "/"))

	g_printf("%s\n", disk_path);

    else if (g_str_equal(disk_path, "/"))

	g_printf("%s\n", mount_point);

    else

	g_printf("%s%s\n", mount_point, disk_path);

}

/* set the tape server and device */

void

set_tape(

    char *	tape)

{

    char *uqtape = unquote_string(tape);

    char *tapedev = strchr(uqtape, ':');

    if (tapedev)

    {

	if (tapedev != uqtape) {

	    if((strchr(tapedev+1, ':') == NULL) &&

	       (g_str_has_prefix(uqtape, "null:") ||

		g_str_has_prefix(uqtape, "rait:") ||

		g_str_has_prefix(uqtape, "file:") ||

		g_str_has_prefix(uqtape, "tape:"))) {

		tapedev = uqtape;

	    }

	    else {

		*tapedev = '\0';

		g_free(tape_server_name);

		tape_server_name = g_strdup(uqtape);

		++tapedev;

	    }

	} else { /* reset server_name if start with : */

	    amfree(tape_server_name);

	    ++tapedev;

	}

    } else

	tapedev = uqtape;

    if (tapedev[0])

    {

	if (g_str_equal(tapedev, "default"))

	    amfree(tape_device_name);

	else {

	        g_free(tape_device_name);

	        tape_device_name = g_strdup(tapedev);

	}

    }

    if (tape_device_name)

	g_printf (_("Using tape \"%s\""), tape_device_name);

    else

	g_printf (_("Using default tape"));

    if (tape_server_name)

	g_printf (_(" from server %s.\n"), tape_server_name);

    else

	g_printf (_(".\nTape server unspecified, assumed to be %s.\n"),

		server_name);

    amfree(uqtape);

}

void

set_mode(

    int		mode)

{

#ifdef SAMBA_CLIENT

  if (mode == SAMBA_SMBCLIENT) {

    g_printf (_("SAMBA dumps will be extracted using smbclient\n"));

    samba_extract_method = SAMBA_SMBCLIENT;

  } else {

    if (mode == SAMBA_TAR) {

      g_printf (_("SAMBA dumps will be extracted as TAR dumps\n"));

      samba_extract_method = SAMBA_TAR;

    }

  }

#else

  (void)mode;	/* Quiet unused parameter warning */

#endif /* SAMBA_CLIENT */

}

void

show_mode(void) 

{

#ifdef SAMBA_CLIENT

  g_printf (_("SAMBA dumps are extracted "));

  if (samba_extract_method == SAMBA_TAR) {

    g_printf (_(" as TAR dumps\n"));

  } else {

    g_printf (_("using smbclient\n"));

  }

#endif /* SAMBA_CLIENT */

}