--- suX/src/sux.c 2013/04/09 07:37:50 1.3 +++ suX/src/sux.c 2013/04/09 15:59:44 1.4 @@ -3,13 +3,13 @@ * by Michael Pounov * * $Author: misho $ - * $Id: sux.c,v 1.3 2013/04/09 07:37:50 misho Exp $ + * $Id: sux.c,v 1.4 2013/04/09 15:59:44 misho Exp $ * *************************************************************************/ #include "global.h" -sl_config cfg; +cfg_root_t cfg; int Verbose; struct tagProc proc; FILE *lf; @@ -20,7 +20,7 @@ Log(int lvl, const char *fmt, ...) { va_list lst, cp; - if (lvl <= Verbose) { + EVERBS(lvl) { va_start(lst, fmt); va_copy(cp, lst); vfprintf(lf, fmt, lst); @@ -45,14 +45,32 @@ Err(const char *fmt, ...) va_end(cp); } +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), + 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), + getenv("PATH_TRANSLATED"), getenv("REMOTE_ADDR"), + getenv("REMOTE_PORT"), getenv("REQUEST_URI")); +} + static void initProg() { - proc.proc_uid = getuid(); - proc.proc_gid = getgid(); - proc.proc_prio = getpriority(PRIO_PROCESS, 0); - getcwd(proc.proc_dir, sizeof proc.proc_dir); + 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_name, string); + AIT_INIT_VAL2(&proc.proc_cmd, string); + #if 0 lf = fopen(DEFAULT_LOG, "a"); if (!lf) @@ -63,6 +81,18 @@ initProg() } static void +endProg() +{ + AIT_FREE_VAL(&proc.proc_uid); + AIT_FREE_VAL(&proc.proc_gid); + AIT_FREE_VAL(&proc.proc_prio); + AIT_FREE_VAL(&proc.proc_class); + AIT_FREE_VAL(&proc.proc_dir); + AIT_FREE_VAL(&proc.proc_name); + AIT_FREE_VAL(&proc.proc_cmd); +} + +static void Usage() { printf( " -= suX =- suExecutor designed for web based applicaions\n" @@ -75,6 +105,7 @@ Usage() "\t-C \t\t\tChroot to directory\n" "\t-c \t\t\tConfig file\n" "\t-l \t\t\tLog file path (default:/var/log/suX.log)\n" + "\t-o\t\t\t\tForce set UID,GID and Priority for program\n" "\t-v\t\t\t\tVerbose, (more -v, more verbosity)\n" "\t-h\t\t\t\tThis help screen!\n\n", PACKAGE_NAME); } @@ -85,14 +116,14 @@ setUIDGID(char flg, const char *name) struct stat sb; if (stat(name, &sb) == -1) { - Err("Error:: %s stat #%d - %s", name, errno, strerror(errno)); + ESYSERR(0); return -1; } - if (!(flg & 1)) - proc.proc_uid = sb.st_uid; - if (!(flg & 2)) - proc.proc_gid = sb.st_gid; + if (!(flg & SUX_GET_UID)) + AIT_SET_I16(&proc.proc_uid, sb.st_uid); + if (!(flg & SUX_GET_GID)) + AIT_SET_I16(&proc.proc_gid, sb.st_gid); return 0; } @@ -100,94 +131,85 @@ setUIDGID(char flg, const char *name) static inline int SetClass() { - login_cap_t *cap; struct passwd *pass; + int ret = 0; - pass = getpwuid(proc.proc_uid); + pass = getpwuid(AIT_GET_I16(&proc.proc_uid)); if (!pass) { - Err("Error:: User with this UID %d not found", proc.proc_uid); + Err("Error:: User with this UID %d not found", AIT_GET_I16(&proc.proc_uid)); endpwent(); return -1; } else - strlcpy(proc.proc_class, pass->pw_class, sizeof proc.proc_class); + AIT_SET_STR(&proc.proc_class, pass->pw_class); - cap = login_getclass(proc.proc_class); - if (!cap) { - Err("Error:: Cant get login class %s", proc.proc_class); - endpwent(); - return -1; + 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(cap, pass, proc.proc_uid, LOGIN_SETALL)) { - Err("Error:: Cant set login class %s", proc.proc_class); - login_close(cap); - endpwent(); - return -1; - } - - login_close(cap); endpwent(); - return 0; + return ret; } static int LoadCfgData(char flg) { - char *str, mode = 0; + const char *str; + char mode = 0; - str = (char*) cfg_GetAttribute(&cfg, CFG("global"), CFG("mode")); + str = cfg_getAttribute(&cfg, "global", "mode"); if (!str) { Err("Error:: Unknown mode ..."); return -1; } if (!strcasecmp(str, "SCRIPT")) { mode = 1; - if (setUIDGID(flg, proc.proc_name) == -1) + if (setUIDGID(flg, AIT_GET_STR(&proc.proc_name)) == -1) return -1; } else if (!strcasecmp(str, "FILE")) { mode = 2; - if (setUIDGID(flg, proc.proc_name) == -1) + if (setUIDGID(flg, AIT_GET_STR(&proc.proc_name)) == -1) return -1; } else if (!strcasecmp(str, "DIR") && - (str = (char*) cfg_GetAttribute(&cfg, CFG("global"), CFG("directory")))) { + (str = cfg_getAttribute(&cfg, "global", "directory"))) { mode = 3; - if (!(flg & 8)) - strlcpy(proc.proc_dir, str, sizeof proc.proc_dir); + if (!(flg & SUX_GET_DIR)) + AIT_SET_STR(&proc.proc_dir, str); - if (setUIDGID(flg, proc.proc_dir) == -1) + if (setUIDGID(flg, AIT_GET_STR(&proc.proc_dir)) == -1) return -1; } else { Err("Error:: Unknown mode %s", str); return -1; } - if (!(flg & 4)) { - str = (char*) cfg_GetAttribute(&cfg, CFG("global"), CFG("priority")); + if (!(flg & SUX_GET_PRIO)) { + str = cfg_getAttribute(&cfg, "global", "priority"); if (str) - proc.proc_prio = strtol(str, NULL, 10); + AIT_SET_I32(&proc.proc_prio, strtol(str, NULL, 10)); } /* find associate extension */ - str = strrchr(proc.proc_name, '.'); + str = strrchr(AIT_GET_STR(&proc.proc_name), '.'); if (str) { str++; str = (*str) ? str : "default"; - switch (cfg_LoadAttribute(&cfg, CFG("associate"), CFG(str), - CFG(proc.proc_cmd), sizeof proc.proc_cmd, NULL)) { + switch (cfg_loadAttribute(&cfg, "associate", str, &proc.proc_cmd, NULL)) { case -1: - Err("Error:: can`t load attribute #%d - %s\n", cfg_GetErrno(), cfg_GetError()); + ELIBERR(cfg); return -1; case 0: - cfg_LoadAttribute(&cfg, CFG("associate"), CFG("default"), - CFG(proc.proc_cmd), sizeof proc.proc_cmd, DEFAULT_CMD); + cfg_loadAttribute(&cfg, "associate", "default", &proc.proc_cmd, DEFAULT_CMD); } } else - strlcpy(proc.proc_cmd, DEFAULT_CMD, sizeof proc.proc_cmd); + AIT_SET_STR(&proc.proc_cmd, DEFAULT_CMD); return 0; } static int -Run(char **argv) +Run(char **argv, char flg) { char **args, *cmd; array_t *acmd, *aarg; @@ -196,66 +218,54 @@ Run(char **argv) if (!argv) return -1; - n = io_arrayMake(proc.proc_cmd, 0, " \t", &acmd); + n = array_Args(AIT_GET_STR(&proc.proc_cmd), 0, " \t", &acmd); if (n < 1) return -1; - if (!(aarg = io_arrayFrom((const char**) argv, 0))) { - io_arrayDestroy(&acmd); + if (!(aarg = array_From((const char***) &argv, 0))) { + array_Destroy(&acmd); return -1; - } else if (*io_arrayGet(acmd, 0, char*) == '!') { - if (io_arrayGrow(acmd, 0)) { - io_arrayDestroy(&aarg); - io_arrayDestroy(&acmd); + } else if (*array(acmd, 0, char*) == '!') { + if (array_Grow(acmd, 0, 0)) { + array_Destroy(&aarg); + array_Destroy(&acmd); return -1; } - cmd = io_arrayGet(aarg, 0, char*); + cmd = array(aarg, 0, char*); } else - cmd = io_arrayGet(acmd, 0, char*); + cmd = array(acmd, 0, char*); - if (io_arrayConcat(acmd, aarg) == -1) { - io_arrayDestroy(&aarg); - io_arrayDestroy(&acmd); + if (array_Concat(acmd, aarg) == -1) { + array_Destroy(&aarg); + array_Destroy(&acmd); return -1; } - io_arrayDestroy(&aarg); - if (!(args = io_arrayTo(acmd))) { - io_arrayDestroy(&acmd); + array_Destroy(&aarg); + if (!(args = array_To(acmd))) { + array_Destroy(&acmd); return -1; } - io_arrayDestroy(&acmd); + array_Destroy(&acmd); if (SetClass()) { if (args) - free(args); + e_free(args); return -1; } - /* - if (setgid(proc.proc_gid) == -1) { - Err("Error:: setgid #%d - %s\n", errno, strerror(errno)); - if (args) - 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; } - if (setuid(proc.proc_uid) == -1) { - Err("Error:: setuid #%d - %s\n", errno, strerror(errno)); - if (args) - free(args); - return -1; - } - if (setpriority(PRIO_PROCESS, 0, proc.proc_prio) == -1) { - Err("Error:: setpriority #%d - %s\n", errno, strerror(errno)); - if (args) - free(args); - return -1; - } - */ - Log(0, "UID:GID=%d:%d Prio=%d Class=%s Name=%s Dir=%s Cmd=%s Script=%s", - proc.proc_uid, proc.proc_gid, proc.proc_prio, proc.proc_class, - proc.proc_name, proc.proc_dir, proc.proc_cmd, getenv("PATH_TRANSLATED")); + DumpProc(__func__); - if (3 <= Verbose) { + EVERBS(3) { char **el = args - 1; while (*++el) Log(3, "args: %s", *el); @@ -264,9 +274,10 @@ Run(char **argv) fflush(lf); execve(cmd, args, environ); +err: + ESYSERR(0); if (args) - free(args); - Err("Error:: in exec() #%d - %s", errno, strerror(errno)); + e_free(args); return -1; } @@ -282,7 +293,7 @@ main(int argc, char **argv) initProg(); strlcpy(szCfg, DEFAULT_CONFIG, sizeof szCfg); - while ((ch = getopt(argc, argv, "hvC:c:u:g:p:d:l:")) != -1) + while ((ch = getopt(argc, argv, "hvoC:c:u:g:p:d:l:")) != -1) switch (ch) { case 'l': f = fopen(optarg, "a"); @@ -295,19 +306,19 @@ main(int argc, char **argv) lf = f; break; case 'd': - strlcpy(proc.proc_dir, optarg, sizeof proc.proc_dir); - flg |= 8; + AIT_SET_STR(&proc.proc_dir, optarg); + flg |= SUX_GET_DIR; break; case 'p': - proc.proc_prio = strtol(optarg, NULL, 0); - flg |= 4; + AIT_SET_I32(&proc.proc_prio, strtol(optarg, NULL, 0)); + flg |= SUX_GET_PRIO; break; case 'g': setgrent(); grp = getgrnam(optarg); if (grp) { - proc.proc_gid = grp->gr_gid; - flg |= 2; + AIT_SET_I16(&proc.proc_gid, grp->gr_gid); + flg |= SUX_GET_GID; } else Err("Error:: Group not found!"); endgrent(); @@ -316,8 +327,8 @@ main(int argc, char **argv) setpwent(); pass = getpwnam(optarg); if (pass) { - proc.proc_uid = pass->pw_uid; - flg |= 1; + AIT_SET_I16(&proc.proc_uid, pass->pw_uid); + flg |= SUX_GET_UID; } else Err("Error:: User not found!"); endpwent(); @@ -332,12 +343,16 @@ main(int argc, char **argv) if ((wrk = strstr(str, optarg))) setenv("PATH_TRANSLATED", str + strlen(optarg), 42); break; + case 'o': + flg |= SUX_GET_FORCE; + break; case 'v': - Verbose++; + e_incVerbose; break; case 'h': default: Usage(); + endProg(); if (fileno(lf) > 2) fclose(lf); return 1; @@ -345,7 +360,7 @@ main(int argc, char **argv) argc -= optind; argv += optind; - if (2 <= Verbose) { + EVERBS(2) { for (pp = argv; *pp; pp++) Log(2, "Args=%s\n", *pp); for (pp = environ; *pp; pp++) @@ -355,36 +370,41 @@ main(int argc, char **argv) if (!argc) { if (!(str = getenv("PATH_TRANSLATED"))) { Usage(); + endProg(); if (fileno(lf) > 2) fclose(lf); return 1; } else - strlcpy(proc.proc_name, str, sizeof proc.proc_name); + AIT_SET_STR(&proc.proc_name, str); } else - strlcpy(proc.proc_name, *argv, sizeof proc.proc_name); + AIT_SET_STR(&proc.proc_name, *argv); Log(2, "Try to load config %s", szCfg); - if (LoadConfig(szCfg, &cfg)) { - Err("Error:: can`t load config #%d - %s\n", cfg_GetErrno(), cfg_GetError()); + if (cfgLoadConfig(szCfg, &cfg)) { + ELIBERR(cfg); + endProg(); if (fileno(lf) > 2) fclose(lf); return 2; } else if (LoadCfgData(flg) == -1) { - UnloadConfig(&cfg); + cfgUnloadConfig(&cfg); + endProg(); if (fileno(lf) > 2) fclose(lf); closelog(); return 3; } - UnloadConfig(&cfg); + cfgUnloadConfig(&cfg); - if (Run(argv) == -1) { + if (Run(argv, flg) == -1) { + endProg(); if (fileno(lf) > 2) fclose(lf); closelog(); return 4; } + endProg(); if (fileno(lf) > 2) fclose(lf); closelog();