/* vi: set sw=4 ts=4: */
/*
 * bare bones sendmail
 *
 * Copyright (C) 2008 by Vladimir Dronnikov <dronnikov@gmail.com>
 *
 * Licensed under GPLv2, see file LICENSE in this tarball for details.
 */
#include "libbb.h"

/*
   Extracted from BB uuencode.c
 */
enum {
	SRC_BUF_SIZE = 45,  /* This *MUST* be a multiple of 3 */
	DST_BUF_SIZE = 4 * ((SRC_BUF_SIZE + 2) / 3),
};

static void uuencode(char *fname, const char *text)
{
#define src_buf text
	int fd;
#define len fd
	char dst_buf[DST_BUF_SIZE + 1];

	if (fname) {
		fd = xopen(fname, O_RDONLY);
		src_buf = bb_common_bufsiz1;
	} else {
		len = strlen(text);
	}

	fflush(stdout); // sync stdio and unistd output
	while (1) {
		size_t size;
		if (fname) {
			size = full_read(fd, (char *)src_buf, SRC_BUF_SIZE);
			if ((ssize_t)size < 0)
				bb_perror_msg_and_die(bb_msg_read_error);
		} else {
			size = len;
			if (len > SRC_BUF_SIZE)
				size = SRC_BUF_SIZE;
		}
		if (!size)
			break;
		// Encode the buffer we just read in
		bb_uuencode(dst_buf, src_buf, size, bb_uuenc_tbl_base64);
		if (fname) {
			xwrite(STDOUT_FILENO, "\n", 1);
		} else {
			src_buf += size;
			len -= size;
		}
		xwrite(STDOUT_FILENO, dst_buf, 4 * ((size + 2) / 3));
	}
	if (ENABLE_FEATURE_CLEAN_UP && fname)
		close(fd);
}

#if ENABLE_FEATURE_SENDMAIL_NETWORK
// generic signal handler
static void signal_handler(int signo)
{
	int err;

	if (SIGALRM == signo)
		bb_error_msg_and_die("timed out");

	// SIGCHLD. reap zombies
	if (wait_any_nohang(&err) > 0)
		if (WIFEXITED(err) && WEXITSTATUS(err))
			bb_error_msg_and_die("child exited (%d)", WEXITSTATUS(err));
}

static pid_t helper_pid;

// read stdin, parses first bytes to a number, i.e. server response
// if code = -1 then just return this number
// if code != -1 then checks whether the number equals the code
// if not equal -> die saying errmsg
static int check(int code, const char *errmsg)
{
	char *answer;

	// read a string and match it against the set of available answers
	fflush(stdout);
	answer = xmalloc_getline(stdin);
	if (answer) {
		int n = atoi(answer);
		if (-1 == code || n == code) {
			free(answer);
			return n;
		}
	}
	// TODO: is there more elegant way to terminate child on program failure?
	if (helper_pid > 0)
		kill(helper_pid, SIGTERM);
	if (!answer)
		answer = (char*)"EOF";
	else
		*strchrnul(answer, '\r') = '\0';
	bb_error_msg_and_die("error at %s: got '%s' instead", errmsg, answer);
}

static int puts_and_check(const char *msg, int code, const char *errmsg)
{
	printf("%s\r\n", msg);
	return check(code, errmsg);
}

// strip argument of bad chars
static char *sane(char *str)
{
	char *s = str;
	char *p = s;
	while (*s) {
		if (isalnum(*s) || '_' == *s || '-' == *s || '.' == *s || '@' == *s) {
			*p++ = *s;
		}
		s++;
	}
	*p = '\0';
	return str;
}
#endif

int sendmail_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
int sendmail_main(int argc, char **argv)
{
	llist_t *recipients = NULL;
	llist_t *bodies = NULL;
	llist_t *attachments = NULL;
	char *from;
	char *notify = NULL;
	const char *subject;
	char *charset = (char*)"utf-8";
#if ENABLE_FEATURE_SENDMAIL_NETWORK
	const char *wsecs = "10";
	const char *server = "127.0.0.1";
	const char *port = NULL;
	const char *opt_user;
	const char *opt_pass;
	unsigned timeout;
#endif
	char *boundary;
	unsigned opts;
	enum {
		OPT_f = 1 << 0,         // sender
		OPT_n = 1 << 2,         // notification
		OPT_s = 1 << 3,         // subject given
		OPT_c = 1 << 6,         // charset
		OPT_d = 1 << 7,         // dry run - no networking
		OPT_w = 1 << 8,         // network timeout
		OPT_h = 1 << 9,         // server
		OPT_p = 1 << 10,        // port
		OPT_U = 1 << 11,        // user specified
		OPT_P = 1 << 12,        // password specified
	};

	// -f must be specified
	// -t, -b, -a may be multiple
	opt_complementary = "f:t::b::a::";
	opts = getopt32(argv,
		"f:t:n::s:b:a:c:" USE_FEATURE_SENDMAIL_NETWORK("dw:h:p:U:P:"),
		&from, &recipients, &notify, &subject, &bodies, &attachments, &charset
		USE_FEATURE_SENDMAIL_NETWORK(, &wsecs, &server, &port, &opt_user, &opt_pass)
	);
	//argc -= optind;
	argv += optind;

	// sanitize user input
	sane(from);
	if (opts & OPT_c)
		sane(charset);

	// establish connection
#if ENABLE_FEATURE_SENDMAIL_NETWORK
	timeout = xatou(wsecs);
	if (!(opts & OPT_d)) {
		// ask password if we need to and while we're still have terminal
		// TODO: get password directly from /dev/tty? or from a secret file?
		if ((opts & (OPT_U+OPT_P)) == OPT_U) {
			if (!isatty(STDIN_FILENO) || !(opt_pass = bb_askpass(0, "Password: "))) {
				bb_error_msg_and_die("no password");
			}
		}
		// set chat timeout
		alarm(timeout);
		// connect to server
		if (argv[0]) {
			// if connection helper given
			// setup vanilla unidirectional pipes interchange
			int idx;
			int pipes[4];
			xpipe(pipes);
			xpipe(pipes+2);
			helper_pid = vfork();
			if (helper_pid < 0)
				bb_perror_msg_and_die("vfork");
			idx = (!helper_pid)*2;
			xdup2(pipes[idx], STDIN_FILENO);
			xdup2(pipes[3-idx], STDOUT_FILENO);
			if (ENABLE_FEATURE_CLEAN_UP)
				for (int i = 4; --i >= 0;)
					if (pipes[i] > STDOUT_FILENO)
						close(pipes[i]);
			// replace child with connection helper
			if (!helper_pid) {
				// child - try to execute connection helper
				BB_EXECVP(argv[0], argv);
				_exit(127);
			}
			// parent - check whether child is alive
			sig_catch(SIGCHLD, signal_handler);
			sig_catch(SIGALRM, signal_handler);
			signal_handler(SIGCHLD);
			// child seems OK -> parent goes on SMTP chat
		} else {
			// no connection helper provided -> make plain connect
			int fd = create_and_connect_stream_or_die(
				server,
				bb_lookup_port(port, "tcp", 25)
			);
			xmove_fd(fd, STDIN_FILENO);
			xdup2(STDIN_FILENO, STDOUT_FILENO);
			// wait for OK
			check(220, "INIT");
		}
		// mail user specified? try modern AUTHentication
		if ((opts & OPT_U)
		 && (334 == puts_and_check("auth login", -1, "auth login"))
		) {
			uuencode(NULL, opt_user);
			puts_and_check("", 334, "AUTH");
			uuencode(NULL, opt_pass);
			puts_and_check("", 235, "AUTH");
		// no mail user specified or modern AUTHentication is not supported?
		} else {
			// fallback to simple HELO authentication
			// fetch domain name (defaults to local)
			const char *domain = strchr(from, '@');
			if (domain)
				domain++;
			else
				domain = "local";
			printf("helo %s\r\n", domain);
			check(250, "HELO");
		}

		// set addresses
		printf("mail from:<%s>\r\n", from);
		check(250, "MAIL FROM");
		for (llist_t *to = recipients; to; to = to->link) {
			printf("rcpt to:<%s>\r\n", sane(to->data));
			check(250, "RCPT TO");
		}
		puts_and_check("data", 354, "DATA");
		// no timeout while sending message
		alarm(0);
	}
#endif

	// now put message
	// put address headers
	printf("From: %s\r\n", from);
	for (llist_t *to = recipients; to; to = to->link) {
		printf("To: %s\r\n", sane(to->data));
	}
	// put encoded subject
	if (opts & OPT_s) {
		printf("Subject: =?%s?B?", charset);
		uuencode(NULL, subject);
		puts("?=\r");
	}
	// put notification
	if (opts & OPT_n) {
		// -n without parameter?
		if (!notify)
			notify = from; // notify sender by default
		printf("Disposition-Notification-To: %s\r\n", sane(notify));
	}
	// put common headers and body start
	//srand(?);
	boundary = xasprintf("%d-%d-%d", rand(), rand(), rand());
	printf(
		"X-Mailer: busybox " BB_VER " sendmail\r\n"
		"Message-ID: <%s>\r\n"
		"Mime-Version: 1.0\r\n"
		"%smultipart/mixed; boundary=\"%s\"\r\n"
		"\r\n"
		"--%s\r\n"
		"%stext/plain; charset=%s\r\n"
		"%s\r\n%s"
		, boundary
		, "Content-Type: "
		, boundary, boundary
		, "Content-Type: "
		, charset
		, "Content-Disposition: inline"
		, "Content-Transfer-Encoding: base64\r\n"
	);
	// put body(ies)
	for (llist_t *f = bodies; f; f = f->link) {
		uuencode(f->data, NULL);
	}
	// put attachment(s)
	for (llist_t *f = attachments; f; f = f->link) {
		printf(
			"\r\n--%s\r\n"
			"%sapplication/octet-stream\r\n"
			"%s; filename=\"%s\"\r\n"
			"%s"
			, boundary
			, "Content-Type: "
			, "Content-Disposition: inline"
			, bb_get_last_path_component_strip(f->data)
			, "Content-Transfer-Encoding: base64\r\n"
		);
		uuencode(f->data, NULL);
	}
	// put terminator
	printf("\r\n--%s--\r\n\r\n", boundary);
	if (ENABLE_FEATURE_CLEAN_UP)
		free(boundary);

#if ENABLE_FEATURE_SENDMAIL_NETWORK
	// end message and say goodbye
	if (!(opts & OPT_d)) {
		alarm(timeout);
		puts_and_check(".", 250, "BODY");
		puts_and_check("quit", 221, "QUIT");
	}
#endif

	return 0;
}