src/daemon.c

   1 #include "common.h"
   2 #include <sched.h>
   3
   4 char pidfile[FILENAME_MAX + 1] = "";
   5
   6 void sig_handler_exit(int signo)
   7 {
   8     if(signo) {
   9         error_log_limit_unlimited();
  10         error("Received signal %d. Exiting...", signo);
  11         netdata_exit = 1;
  12     }
  13 }
  14
  15 void sig_handler_logrotate(int signo)
  16 {
  17     if(signo) {
  18         error_log_limit_reset();
  19         info("Received signal %d to re-open the log files", signo);
  20         reopen_all_log_files();
  21     }
  22 }
  23
  24 void sig_handler_save(int signo)
  25 {
  26     if(signo) {
  27         error_log_limit_reset();
  28         info("Received signal %d to save the database...", signo);
  29         rrdset_save_all();
  30     }
  31 }
  32
  33 static void chown_open_file(int fd, uid_t uid, gid_t gid) {
  34     if(fd == -1) return;
  35
  36     struct stat buf;
  37
  38     if(fstat(fd, &buf) == -1) {
  39         error("Cannot fstat() fd %d", fd);
  40         return;
  41     }
  42
  43     if((buf.st_uid != uid || buf.st_gid != gid) && S_ISREG(buf.st_mode)) {
  44         if(fchown(fd, uid, gid) == -1)
  45             error("Cannot fchown() fd %d.", fd);
  46     }
  47 }
  48
  49 int become_user(const char *username, int pid_fd)
  50 {
  51     struct passwd *pw = getpwnam(username);
  52     if(!pw) {
  53         error("User %s is not present.", username);
  54         return -1;
  55     }
  56
  57     uid_t uid = pw->pw_uid;
  58     gid_t gid = pw->pw_gid;
  59
  60     int ngroups = (int)sysconf(_SC_NGROUPS_MAX);
  61     gid_t *supplementary_groups = NULL;
  62     if(ngroups) {
  63         supplementary_groups = mallocz(sizeof(gid_t) * ngroups);
  64         if(getgrouplist(username, gid, supplementary_groups, &ngroups) == -1) {
  65             error("Cannot get supplementary groups of user '%s'.", username);
  66             freez(supplementary_groups);
  67             supplementary_groups = NULL;
  68             ngroups = 0;
  69         }
  70     }
  71
  72     chown_open_file(STDOUT_FILENO, uid, gid);
  73     chown_open_file(STDERR_FILENO, uid, gid);
  74     chown_open_file(stdaccess_fd, uid, gid);
  75     chown_open_file(pid_fd, uid, gid);
  76
  77     if(supplementary_groups && ngroups) {
  78         if(setgroups(ngroups, supplementary_groups) == -1)
  79             error("Cannot set supplementary groups for user '%s'", username);
  80
  81         freez(supplementary_groups);
  82         supplementary_groups = NULL;
  83         ngroups = 0;
  84     }
  85
  86     if(setresgid(gid, gid, gid) != 0) {
  87         error("Cannot switch to user's %s group (gid: %u).", username, gid);
  88         return -1;
  89     }
  90
  91     if(setresuid(uid, uid, uid) != 0) {
  92         error("Cannot switch to user %s (uid: %u).", username, uid);
  93         return -1;
  94     }
  95
  96     if(setgid(gid) != 0) {
  97         error("Cannot switch to user's %s group (gid: %u).", username, gid);
  98         return -1;
  99     }
 100     if(setegid(gid) != 0) {
 101         error("Cannot effectively switch to user's %s group (gid: %u).", username, gid);
 102         return -1;
 103     }
 104     if(setuid(uid) != 0) {
 105         error("Cannot switch to user %s (uid: %u).", username, uid);
 106         return -1;
 107     }
 108     if(seteuid(uid) != 0) {
 109         error("Cannot effectively switch to user %s (uid: %u).", username, uid);
 110         return -1;
 111     }
 112
 113     return(0);
 114 }
 115
 116 void oom_score_adj(int score) {
 117     int done = 0;
 118     int fd = open("/proc/self/oom_score_adj", O_WRONLY);
 119     if(fd != -1) {
 120         char buf[10 + 1];
 121         ssize_t len = snprintfz(buf, 10, "%d", score);
 122         if(write(fd, buf, len) == len) done = 1;
 123         close(fd);
 124     }
 125
 126     if(!done)
 127         error("Cannot adjust my Out-Of-Memory score to %d.", score);
 128     else
 129         info("Adjusted my Out-Of-Memory score to %d.", score);
 130 }
 131
 132 int sched_setscheduler_idle(void) {
 133     const struct sched_param param = {
 134         .sched_priority = 0
 135     };
 136
 137     int i = sched_setscheduler(0, SCHED_IDLE, &param);
 138     if(i != 0)
 139         error("Cannot adjust my scheduling priority to IDLE.");
 140     else
 141         info("Adjusted my scheduling priority to IDLE.");
 142
 143     return i;
 144 }
 145
 146 int become_daemon(int dont_fork, const char *user)
 147 {
 148     if(!dont_fork) {
 149         int i = fork();
 150         if(i == -1) {
 151             perror("cannot fork");
 152             exit(1);
 153         }
 154         if(i != 0) {
 155             exit(0); // the parent
 156         }
 157
 158         // become session leader
 159         if (setsid() < 0) {
 160             perror("Cannot become session leader.");
 161             exit(2);
 162         }
 163
 164         // fork() again
 165         i = fork();
 166         if(i == -1) {
 167             perror("cannot fork");
 168             exit(1);
 169         }
 170         if(i != 0) {
 171             exit(0); // the parent
 172         }
 173     }
 174
 175     // generate our pid file
 176     int pidfd = -1;
 177     if(pidfile[0]) {
 178         pidfd = open(pidfile, O_WRONLY | O_CREAT, 0644);
 179         if(pidfd >= 0) {
 180             if(ftruncate(pidfd, 0) != 0)
 181                 error("Cannot truncate pidfile '%s'.", pidfile);
 182
 183             char b[100];
 184             sprintf(b, "%d\n", getpid());
 185             ssize_t i = write(pidfd, b, strlen(b));
 186             if(i <= 0)
 187                 error("Cannot write pidfile '%s'.", pidfile);
 188         }
 189         else error("Failed to open pidfile '%s'.", pidfile);
 190     }
 191
 192     // Set new file permissions
 193     umask(0002);
 194
 195     // adjust my Out-Of-Memory score
 196     oom_score_adj(1000);
 197
 198     // never become a problem
 199     if(sched_setscheduler_idle() != 0) {
 200         if(nice(19) == -1) error("Cannot lower my CPU priority.");
 201         else info("Set my nice value to 19.");
 202     }
 203
 204     if(user && *user) {
 205         if(become_user(user, pidfd) != 0) {
 206             error("Cannot become user '%s'. Continuing as we are.", user);
 207         }
 208         else info("Successfully became user '%s'.", user);
 209     }
 210
 211     if(pidfd != -1) {
 212         close(pidfd);
 213         pidfd = -1;
 214     }
 215
 216     return(0);
 217 }