--- suX/src/sux.c	2013/04/09 15:59:44	1.4
+++ suX/src/sux.c	2013/04/10 00:04:11	1.5
@@ -3,7 +3,7 @@
  *  by Michael Pounov <misho@aitbg.com>
  *
  * $Author: misho $
- * $Id: sux.c,v 1.4 2013/04/09 15:59:44 misho Exp $
+ * $Id: sux.c,v 1.5 2013/04/10 00:04:11 misho Exp $
  *
  *************************************************************************/
 #include "global.h"
@@ -48,8 +48,9 @@ Err(const char *fmt, ...)
 static inline void
 DumpProc(const char *txt)
 {
-	Log(0, "%s:: UID:GID=%d:%d Prio=%d Class=%s Name=%s Dir=%s Cmd=%s Script=%s From=%s:%s%s", 
-			txt ? txt : __func__, AIT_GET_I16(&proc.proc_uid), 
+	Log(0, "%s:: uid:gid=%d:%d UID:GID=%d:%d Prio=%d Class=%s Name=%s Dir=%s Cmd=%s "
+			"Script=%s From=%s:%s Get=%s", txt ? txt : __func__, 
+			geteuid(), getegid(), AIT_GET_I16(&proc.proc_uid), 
 			AIT_GET_I16(&proc.proc_gid), AIT_GET_I32(&proc.proc_prio), 
 			AIT_GET_STR(&proc.proc_class), AIT_GET_STR(&proc.proc_name), 
 			AIT_GET_STR(&proc.proc_dir), AIT_GET_STR(&proc.proc_cmd), 
@@ -57,17 +58,14 @@ DumpProc(const char *txt)
 			getenv("REMOTE_PORT"), getenv("REQUEST_URI"));
 }
 
-static void
+static int
 initProg()
 {
-	char d[MAXPATHLEN];
-
 	AIT_SET_I16(&proc.proc_uid, getuid());
 	AIT_SET_I16(&proc.proc_gid, getgid());
 	AIT_SET_I32(&proc.proc_prio, getpriority(PRIO_PROCESS, 0));
 	AIT_INIT_VAL2(&proc.proc_class, string);
-	getcwd(d, sizeof d);
-	AIT_SET_STR(&proc.proc_dir, d);
+	AIT_INIT_VAL2(&proc.proc_dir, string);
 	AIT_INIT_VAL2(&proc.proc_name, string);
 	AIT_INIT_VAL2(&proc.proc_cmd, string);
 
@@ -78,6 +76,7 @@ initProg()
 		lf = stdout;
 
 	openlog(PACKAGE_NAME, LOG_CONS | LOG_PID | LOG_NDELAY, LOG_USER);
+	return 0;
 }
 
 static void
@@ -114,22 +113,38 @@ static inline int
 setUIDGID(char flg, const char *name)
 {
 	struct stat sb;
+	struct passwd *pass;
+	short uid, gid;
 
-	if (stat(name, &sb) == -1) {
-		ESYSERR(0);
-		return -1;
+	if (name) {
+		if (stat(name, &sb) == -1) {
+			ESYSERR(0);
+			return -1;
+		}
+		uid = sb.st_uid;
+		gid = sb.st_gid;
+	} else {
+		pass = getpwnam(getenv("SUX_USER"));
+		if (!pass) {
+			Err("Error:: User %s not found", getenv("SUX_USER"));
+			endpwent();
+			return -1;
+		}
+		uid = pass->pw_uid;
+		gid = pass->pw_gid;
+		endpwent();
 	}
 
 	if (!(flg & SUX_GET_UID))
-		AIT_SET_I16(&proc.proc_uid, sb.st_uid);
+		AIT_SET_I16(&proc.proc_uid, uid);
 	if (!(flg & SUX_GET_GID))
-		AIT_SET_I16(&proc.proc_gid, sb.st_gid);
+		AIT_SET_I16(&proc.proc_gid, gid);
 
 	return 0;
 }
 
 static inline int
-SetClass()
+setClassDir()
 {
 	struct passwd *pass;
 	int ret = 0;
@@ -137,16 +152,17 @@ SetClass()
 	pass = getpwuid(AIT_GET_I16(&proc.proc_uid));
 	if (!pass) {
 		Err("Error:: User with this UID %d not found", AIT_GET_I16(&proc.proc_uid));
-		endpwent();
-		return -1;
-	} else 
+		ret = -1;
+	} else {
 		AIT_SET_STR(&proc.proc_class, pass->pw_class);
+		AIT_SET_STR(&proc.proc_dir, pass->pw_dir);
 
-	if (setusercontext(NULL, pass, AIT_GET_I16(&proc.proc_uid), 
-				LOGIN_SETRESOURCES | LOGIN_SETGROUP | LOGIN_SETLOGIN | 
-				LOGIN_SETPRIORITY)) {
-		Err("Error:: Cant set login class %s", AIT_GET_STR(&proc.proc_class));
-		ret = -1;
+		if (setusercontext(NULL, pass, AIT_GET_I16(&proc.proc_uid), 
+					LOGIN_SETLOGIN | LOGIN_SETGROUP | LOGIN_SETUSER | 
+					LOGIN_SETPRIORITY | LOGIN_SETRESOURCES)) {
+			Err("Error:: Can't set login class %s", AIT_GET_STR(&proc.proc_class));
+			ret = -1;
+		}
 	}
 
 	endpwent();
@@ -164,21 +180,18 @@ LoadCfgData(char flg)
 		Err("Error:: Unknown mode ...");
 		return -1;
 	}
-	if (!strcasecmp(str, "SCRIPT")) {
+	if (!strcasecmp(str, "SCRIPT") && getenv("SUX_USER")) {
 		mode = 1;
-		if (setUIDGID(flg, AIT_GET_STR(&proc.proc_name)) == -1)
+		if (setUIDGID(flg, NULL) == -1)
 			return -1;
 	} else if (!strcasecmp(str, "FILE")) {
 		mode = 2;
 		if (setUIDGID(flg, AIT_GET_STR(&proc.proc_name)) == -1)
 			return -1;
-	} else if (!strcasecmp(str, "DIR") && 
-			(str = cfg_getAttribute(&cfg, "global", "directory"))) {
+	} else if (!strcasecmp(str, "DIR")) {
 		mode = 3;
-		if (!(flg & SUX_GET_DIR))
-			AIT_SET_STR(&proc.proc_dir, str);
-
-		if (setUIDGID(flg, AIT_GET_STR(&proc.proc_dir)) == -1)
+		str = AIT_GET_STR(&proc.proc_dir) ? AIT_GET_STR(&proc.proc_dir) : ".";
+		if (setUIDGID(flg, str) == -1)
 			return -1;
 	} else {
 		Err("Error:: Unknown mode %s", str);
@@ -200,7 +213,8 @@ LoadCfgData(char flg)
 				ELIBERR(cfg);
 				return -1;
 			case 0:
-				cfg_loadAttribute(&cfg, "associate", "default", &proc.proc_cmd, DEFAULT_CMD);
+				cfg_loadAttribute(&cfg, "associate", "default", 
+						&proc.proc_cmd, DEFAULT_CMD);
 		}
 	} else
 		AIT_SET_STR(&proc.proc_cmd, DEFAULT_CMD);
@@ -246,19 +260,17 @@ Run(char **argv, char flg)
 	}
 	array_Destroy(&acmd);
 
-	if (SetClass()) {
+	if (setClassDir()) {
 		if (args)
 			e_free(args);
 		return -1;
 	}
 
 	if (flg & SUX_GET_FORCE) {
-		/*
 		if (setegid(AIT_GET_I16(&proc.proc_gid)) == -1)
 			goto err;
 		if (seteuid(AIT_GET_I16(&proc.proc_uid)) == -1)
 			goto err;
-			*/
 		if (setpriority(PRIO_PROCESS, 0, AIT_GET_I32(&proc.proc_prio)) == -1)
 			goto err;
 	}