/*
 *  MkUsers by Davide Libenzi ( bulk mail users generator for XMail )
 *  Copyright (C) 1999,..,2004  Davide Libenzi
 *
 *  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 2 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, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *  Davide Libenzi <davidel@xmailserver.org>
 *
 */

#if defined(WIN32)

#include <stdio.h>
#include <io.h>
#include <direct.h>
#include <stdlib.h>
#include <string.h>

#define SYS_SLASH_CHAR              '\\'
#define SYS_SLASH_STR               "\\"
#define SYS_MAX_PATH                256

int SysPathExist(char const *pszPathName)
{
	return (_access(pszPathName, 0) == 0) ? 1 : 0;
}

int SysMakeDir(char const *pszPathName)
{
	return (_mkdir(pszPathName) == 0) ? 1 : 0;
}

int SysErrNo(void)
{
	return errno;
}

char const *SysErrStr(void)
{
	return strerror(errno);
}

#else				// #if defined(WIN32)
#if defined(__LINUX__) || defined(__SOLARIS__) || defined(__BSD__)

#include <sys/stat.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <time.h>
#include <fcntl.h>
#include <unistd.h>
#include <errno.h>
#include <dirent.h>

#define SYS_SLASH_CHAR              '/'
#define SYS_SLASH_STR               "/"
#define SYS_MAX_PATH                256

int SysPathExist(char const *pszPathName)
{
	return (access(pszPathName, 0) == 0) ? 1 : 0;
}

int SysMakeDir(char const *pszPathName)
{
	return (mkdir(pszPathName, 0700) == 0) ? 1 : 0;
}

int SysErrNo(void)
{
	return errno;
}

char const *SysErrStr(void)
{
	return strerror(errno);
}

#else				// #if defined(__LINUX__) || defined(__SOLARIS__)

#error system type not defined !

#endif				// #if defined(__LINUX__) || defined(__SOLARIS__)
#endif				// #if defined(WIN32)

#define MAX_MB_SIZE             10000
#define StrNCpy(t, s, n)        do { strncpy(t, s, n); (t)[(n) - 1] = '\0'; } while (0)
#define StrSNCpy(t, s)          StrNCpy(t, s, sizeof(t))

char *StrCrypt(char const *pszInput, char *pszCrypt)
{
	strcpy(pszCrypt, "");

	for (int ii = 0; pszInput[ii] != '\0'; ii++) {
		unsigned int uChar = (unsigned int) pszInput[ii];
		char szByte[32] = "";

		sprintf(szByte, "%02x", (uChar ^ 101) & 0xff);

		strcat(pszCrypt, szByte);
	}

	return pszCrypt;
}

int CreateUser(char const *pszRootDir, char const *pszDomain,
	       char const *pszUsername, char const *pszPassword,
	       unsigned int uUserId, char const *pszRealName,
	       char const *pszHomePage, unsigned int uMBSize, bool bMaildir, FILE * pUsrFile)
{
	FILE *pTabFile;
	char szPathName[SYS_MAX_PATH] = "";
	char szCryptPwd[256] = "";

	/* Check-create domain directory */
	sprintf(szPathName, "%sdomains" SYS_SLASH_STR "%s", pszRootDir, pszDomain);

	if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
		perror(szPathName);
		return SysErrNo();
	}
	/* Check-create domain/user directory */
	sprintf(szPathName, "%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s", pszRootDir,
		pszDomain, pszUsername);

	if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
		perror(szPathName);
		return SysErrNo();
	}

	if (bMaildir) {
		/* Check-create domain/user/Maildir/(tmp,new,cur) directories */
		sprintf(szPathName,
			"%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s" SYS_SLASH_STR "Maildir",
			pszRootDir, pszDomain, pszUsername);

		if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
			perror(szPathName);
			return SysErrNo();
		}

		sprintf(szPathName,
			"%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s" SYS_SLASH_STR "Maildir"
			SYS_SLASH_STR "tmp", pszRootDir, pszDomain, pszUsername);

		if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
			perror(szPathName);
			return SysErrNo();
		}

		sprintf(szPathName,
			"%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s" SYS_SLASH_STR "Maildir"
			SYS_SLASH_STR "new", pszRootDir, pszDomain, pszUsername);

		if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
			perror(szPathName);
			return SysErrNo();
		}

		sprintf(szPathName,
			"%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s" SYS_SLASH_STR "Maildir"
			SYS_SLASH_STR "cur", pszRootDir, pszDomain, pszUsername);

		if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
			perror(szPathName);
			return SysErrNo();
		}
	} else {
		/* Check-create domain/user/mailbox directory */
		sprintf(szPathName,
			"%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s" SYS_SLASH_STR "mailbox",
			pszRootDir, pszDomain, pszUsername);

		if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
			perror(szPathName);
			return SysErrNo();
		}
	}

	/* Check-create domain/user/mailbox/user.tab file */
	sprintf(szPathName,
		"%sdomains" SYS_SLASH_STR "%s" SYS_SLASH_STR "%s" SYS_SLASH_STR "user.tab",
		pszRootDir, pszDomain, pszUsername);

	if (!SysPathExist(szPathName)) {
		if ((pTabFile = fopen(szPathName, "wt")) == NULL) {
			perror(szPathName);
			return SysErrNo();
		}

		fprintf(pTabFile,
			"\"RealName\"\t\"%s\"\n"
			"\"HomePage\"\t\"%s\"\n"
			"\"MaxMBSize\"\t\"%u\"\n", pszRealName, pszHomePage, uMBSize);

		fclose(pTabFile);
	}
	/* Add user to users file */
	fprintf(pUsrFile,
		"\"%s\"\t"
		"\"%s\"\t"
		"\"%s\"\t"
		"\"%u\"\t"
		"\"%s\"\t"
		"\"U\"\n", pszDomain, pszUsername, StrCrypt(pszPassword, szCryptPwd), uUserId,
		pszUsername);

	return 0;
}

void ShowUsage(char const *pszProgName)
{
	fprintf(stderr,
		"use : %s [-adfursih]\n"
		"          -a numusers     = number of users to create in auto-mode\n"
		"          -d domain       = domain name in auto-mode\n"
		"          -f inputFile    = input file name {stdin}\n"
		"          -u username     = radix user name in auto-mode\n"
		"          -r rootdir      = mail root path {.%s}\n"
		"          -s mboxsize     = mailbox maximum size {%d}\n"
		"          -i useridbase   = base user id {1}\n"
		"          -m              = create Maildir boxes\n"
		"          -h              = show this message\n",
		pszProgName, SYS_SLASH_STR, MAX_MB_SIZE);

}

int main(int argc, char *argv[])
{
	int ii;
	int iNumUsers = 0;
	bool bMaildir = false;
	unsigned int uMBSize = MAX_MB_SIZE;
	unsigned int uUserId = 1;
	FILE *pUsrFile;
	FILE *pInFile = stdin;
	char szRootDir[SYS_MAX_PATH] = "." SYS_SLASH_STR;
	char szInputFile[SYS_MAX_PATH] = "";
	char szPathName[SYS_MAX_PATH] = "";
	char szAutoDomain[256] = "mkuser.net";
	char szAutoUsr[128] = "mkuser";
	char szUsrLine[1024] = "";

	for (ii = 1; ii < argc; ii++) {
		if (argv[ii][0] != '-')
			continue;

		switch (argv[ii][1]) {
		case ('a'):
			if (++ii < argc)
				iNumUsers = atoi(argv[ii]);
			break;

		case ('d'):
			if (++ii < argc)
				StrSNCpy(szAutoDomain, argv[ii]);
			break;

		case ('f'):
			if (++ii < argc)
				StrSNCpy(szInputFile, argv[ii]);
			break;

		case ('u'):
			if (++ii < argc)
				StrSNCpy(szAutoUsr, argv[ii]);
			break;

		case ('r'):
			if (++ii < argc)
				StrSNCpy(szRootDir, argv[ii]);
			break;

		case ('s'):
			if (++ii < argc)
				uMBSize = (unsigned int) atol(argv[ii]);
			break;

		case ('i'):
			if (++ii < argc)
				uUserId = (unsigned int) atol(argv[ii]);
			break;

		case ('m'):
			bMaildir = true;
			break;

		case ('h'):
			ShowUsage(argv[0]);
			return 0;

		default:
			ShowUsage(argv[0]);
			return 1;
		}
	}

	/* Root directory slash termination */
	if (szRootDir[strlen(szRootDir) - 1] != SYS_SLASH_CHAR)
		strcat(szRootDir, SYS_SLASH_STR);

	/* Check-create domains directory */
	sprintf(szPathName, "%sdomains", szRootDir);

	if (!SysPathExist(szPathName) && !SysMakeDir(szPathName)) {
		perror(szPathName);
		return SysErrNo();
	}
	/* Create mailusers.tab file */
	sprintf(szPathName, "%smailusers.tab", szRootDir);

	if (SysPathExist(szPathName)) {
		fprintf(stderr, "%s already exist\n", szPathName);
		return 1;
	}

	if ((pUsrFile = fopen(szPathName, "wt")) == NULL) {
		perror(szPathName);
		return SysErrNo();
	}

	if (iNumUsers == 0) {
		if ((strlen(szInputFile) != 0) && ((pInFile = fopen(szInputFile, "rt")) == NULL)) {
			perror(szPathName);
			fclose(pUsrFile);
			return SysErrNo();
		}
		/* Get input from stdin */
		while (fgets(szUsrLine, sizeof(szUsrLine) - 1, pInFile) != NULL) {
			char *pszDomain;
			char *pszUsername;
			char *pszPassword;
			char *pszRealName;
			char *pszHomePage;

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

			if ((szUsrLine[0] == '#') ||
			    ((pszDomain = strtok(szUsrLine, ";")) == NULL) ||
			    ((pszUsername = strtok(NULL, ";")) == NULL) ||
			    ((pszPassword = strtok(NULL, ";")) == NULL) ||
			    ((pszRealName = strtok(NULL, ";")) == NULL) ||
			    ((pszHomePage = strtok(NULL, ";")) == NULL))
				continue;

			if (CreateUser(szRootDir, pszDomain, pszUsername, pszPassword, uUserId++,
				       pszRealName, pszHomePage, uMBSize, bMaildir,
				       pUsrFile) != 0) {

				fprintf(stderr, "error creating <%s@%s> : %s\n",
					pszUsername, pszDomain, SysErrStr());

			}

		}

		if (pInFile != stdin)
			fclose(pInFile);
	} else {
		/* Automatically generate users */
		for (ii = 1; ii <= iNumUsers; ii++) {
			char szUsername[256] = "";
			char szHomePage[256] = "";

			sprintf(szUsername, "%s%d", szAutoUsr, ii);

			sprintf(szHomePage, "http://www.%s/~%s", szAutoDomain, szUsername);

			if (CreateUser(szRootDir, szAutoDomain, szUsername, szUsername, uUserId++,
				       szUsername, szHomePage, uMBSize, bMaildir, pUsrFile) != 0)
			{

				fprintf(stderr, "error creating <%s@%s> : %s\n",
					szUsername, szAutoDomain, SysErrStr());

			}

		}
	}

	fclose(pUsrFile);

	return 0;
}