mainscnt/mains-frequency-counter-stm32
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use name

Browse Source
This commit is contained in:
Wolfgang Hottgenroth 2021-02-11 12:16:14 +01:00
parent 3e41e4d9e5
commit 0cd894391b
Signed by: wn
GPG Key ID: 6C1E5E531E0D5D7F
1 changed files with 15 additions and 8 deletions

23
sink/sink20169.c

@ -10,6 +10,7 @@
#include <netinet/in.h> #include <netinet/in.h>
#include <string.h> #include <string.h>
#include <getopt.h> #include <getopt.h>
#include <pwd.h>
#include <libconfig.h> #include <libconfig.h>
#include <curl/curl.h> #include <curl/curl.h>
@ -315,8 +316,8 @@ void usage() {
printf(" -v ............... Verbose, writes all logging on stdout too\n"); printf(" -v ............... Verbose, writes all logging on stdout too\n");
printf(" -s FACILITY ...... Sets syslog facility, only LOCAL[0..7]\n"); printf(" -s FACILITY ...... Sets syslog facility, only LOCAL[0..7]\n");
printf(" USER and DAEMON are supported\n"); printf(" USER and DAEMON are supported\n");
printf(" -n UID ........... If started as root drop privileges and become\n"); printf(" -n USER .......... If started as root drop privileges and become\n");
printf(" user with id UID\n"); printf(" USER\n");
printf(" -h ............... This help\n"); printf(" -h ............... This help\n");
} }
@ -327,7 +328,7 @@ int main(int argc, char **argv) {
const char *configFilename = DEFAULT_CONFIG_FILENAME; const char *configFilename = DEFAULT_CONFIG_FILENAME;
uid_t dropPrivilegesToUID = 0; const char dropPrivilegesToUser = NULL;
int c; int c;
while ((c = getopt(argc, argv, "f:vs:hn:")) != -1) { while ((c = getopt(argc, argv, "f:vs:hn:")) != -1) {
@ -342,7 +343,7 @@ int main(int argc, char **argv) {
setfacility(optarg); setfacility(optarg);
break; break;
case 'n': case 'n':
dropPrivilegesToUID = (uid_t) strtol(optarg, NULL, 10); dropPrivilegesToUser = strdup(optarg);
break; break;
case 'h': case 'h':
usage(); usage();
@ -351,10 +352,16 @@ int main(int argc, char **argv) {
} }
} }
if ((getuid() == 0) && (dropPrivilegesToUID != 0)) { if ((getuid() == 0) && (dropPrivilegesToUser != NULL)) {
logmsg(LOG_INFO, "dropping root privileges"); logmsg(LOG_INFO, "dropping root privileges, become %s", dropPrivilegesToUser);
if (setuid(dropPrivilegesToUID) != 0) { passwd *userEntry = getpwnam(dropPrivilegesToUser);
logmsg(LOG_ERR, "unable to drop root privileges"); if (userEntry == NULL) {
logmsg(LOG_ERR, "can not find entry for user %s", dropPrivilegesToUser);
exit(-1);
}
if (setuid(userEntry->pw_uid) != 0) {
logmsg(LOG_ERR, "unable to drop root privileges to %d", userEntry->pw_uid);
exit(-1); exit(-1);
} }
} }