/* 
   Unix SMB/Netbios implementation.
   Version 2.0
   SMB wrapper functions - frontend
   Copyright (C) Andrew Tridgell 1998
   Copyright (C) Derrell Lipman 2003-2005
   
   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 <sys/types.h>
#include <sys/stat.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <limits.h>
#include <string.h>
#include <libsmbclient.h>
#include "bsd-strlfunc.h"

#ifndef FALSE
# define        FALSE   (0)
# define        TRUE    (! FALSE)
#endif

static void smbsh_usage(void)
{
	printf("smbsh [options] [command [args] ...]\n\n");
        printf(" -p prepend library name\n");
        printf(" -a append library name\n");
        printf(" -n");
	printf(" -W workgroup\n");
	printf(" -U username\n");
	printf(" -P prefix\n");
	printf(" -R resolve order\n");
	printf(" -d debug level\n");
	printf(" -l logfile\n");
	printf(" -L libdir\n");
	exit(0);
}

int main(int argc, char *argv[])
{
	char *p, *u;
	char *libd = ".";
	char line[PATH_MAX], pre[PATH_MAX], post[PATH_MAX];
	int opt;
        int no_ask = 0;
        struct stat statbuf;
	extern char *optarg;
	extern int optind;

        *pre = *post = '\0';

	while ((opt = getopt(argc, argv, "p:a:d:nL:W:U:h")) != -1) {
		switch (opt) {
                case 'p':       /* prepend library before smbwrapper.so */
                        if (*pre != '\0')
                                smbw_strlcat(pre, " ", sizeof(pre));
                        smbw_strlcat(pre, optarg, sizeof(pre));
                        break;
                        
                case 'a':       /* append library after smbwrapper.so */
                        smbw_strlcat(post, " ", sizeof(post));
                        smbw_strlcat(post, optarg, sizeof(post));
                        break;
                        
                case 'd':
                        setenv("DEBUG", optarg, TRUE);
                        break;

                case 'n':       /* don't ask for username/password */
                        no_ask++;
                        break;

		case 'L':
			libd = optarg;
			break;

		case 'W':
			setenv("WORKGROUP", optarg, TRUE);
			break;

		case 'U':
			p = strchr(optarg,'%');
			if (p) {
				*p=0;
				setenv("PASSWORD", p+1, TRUE);
			}
			setenv("USER", optarg, TRUE);
			break;

		case 'h':
		default:
			smbsh_usage();
		}
	}


        if (! no_ask) {
                if (!getenv("USER")) {
                        printf("Username: ");
                        u = fgets(line, sizeof(line)-1, stdin);
                        setenv("USER", u, TRUE);
                }
                
                if (!getenv("PASSWORD")) {
                        p = getpass("Password: ");
                        setenv("PASSWORD", p, TRUE);
                }
        }

        smbw_strlcpy(line, pre, PATH_MAX - strlen(line));
        smbw_strlcat(line, " ", sizeof(line));
        smbw_strlcat(line, libd, sizeof(line));
        smbw_strlcat(line, "/smbwrapper.so", sizeof(line));
        smbw_strlcat(line, post, sizeof(line));
	setenv("LD_PRELOAD", line, TRUE);
        setenv("LD_BIND_NOW", "true", TRUE);

	snprintf(line,sizeof(line)-1,"%s/smbwrapper.32.so", libd);

	if (stat(line, &statbuf) == 0 && S_ISREG(statbuf.st_mode)) {
		snprintf(line, sizeof(line)-1,
                         "%s/smbwrapper.32.so:DEFAULT", libd);
		setenv("_RLD_LIST", line, TRUE);
		snprintf(line, sizeof(line)-1,
                         "%s/smbwrapper.so:DEFAULT", libd);
		setenv("_RLDN32_LIST", line, TRUE);
	} else {
		snprintf(line,sizeof(line)-1,"%s/smbwrapper.so:DEFAULT", libd);
		setenv("_RLD_LIST", line, TRUE);
	}

        if (optind < argc) {
                execvp(argv[optind], &argv[optind]);
        } else {
                char *shellpath = getenv("SHELL");

                setenv("PS1", "smbsh$ ", TRUE);

                if(shellpath) {
			execl(shellpath,"smbsh", NULL);
                } else {
                        setenv("SHELL", "/bin/sh", TRUE);
			execl("/bin/sh", "smbsh", NULL);
                }
	}
	printf("launch failed!\n");
	return 1;
}