1
 
/*  -*- coding: utf-8 -*- */
 
3
 
 * Usplash - Read a password from usplash and output it
 
5
 
 * Copyright © 2008,2009 Teddy Hogeborn
 
6
 
 * Copyright © 2008,2009 Björn Påhlsson
 
8
 
 * This program is free software: you can redistribute it and/or
 
9
 
 * modify it under the terms of the GNU General Public License as
 
10
 
 * published by the Free Software Foundation, either version 3 of the
 
11
 
 * License, or (at your option) any later version.
 
13
 
 * This program is distributed in the hope that it will be useful, but
 
14
 
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 
15
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 
16
 
 * General Public License for more details.
 
18
 
 * You should have received a copy of the GNU General Public License
 
19
 
 * along with this program.  If not, see
 
20
 
 * <http://www.gnu.org/licenses/>.
 
22
 
 * Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
 
23
 
 * <https://www.fukt.bsnet.se/~teddy/>.
 
26
1
#define _GNU_SOURCE             /* asprintf() */
 
27
2
#include <signal.h>             /* sig_atomic_t, struct sigaction,
 
28
3
                                   sigemptyset(), sigaddset(), SIGINT,
 
 
30
5
                                   SIG_IGN, kill(), SIGKILL */
 
31
6
#include <stdbool.h>            /* bool, false, true */
 
32
7
#include <fcntl.h>              /* open(), O_WRONLY, O_RDONLY */
 
33
 
#include <iso646.h>             /* and, or, not*/
 
34
8
#include <errno.h>              /* errno, EINTR */
 
35
 
#include <sys/types.h>          /* size_t, ssize_t, pid_t, DIR, struct
 
 
9
#include <sys/types.h>          /* size_t, ssize_t, pid_t, DIR,
 
37
11
#include <stddef.h>             /* NULL */
 
38
12
#include <string.h>             /* strlen(), memcmp() */
 
39
 
#include <stdio.h>              /* asprintf(), perror(), sscanf() */
 
 
13
#include <stdio.h>              /* asprintf(), perror() */
 
40
14
#include <unistd.h>             /* close(), write(), readlink(),
 
41
15
                                   read(), STDOUT_FILENO, sleep(),
 
42
16
                                   fork(), setuid(), geteuid(),
 
43
17
                                   setsid(), chdir(), dup2(),
 
44
18
                                   STDERR_FILENO, execv() */
 
45
 
#include <stdlib.h>             /* free(), EXIT_FAILURE, realloc(),
 
46
 
                                   EXIT_SUCCESS, malloc(), _exit() */
 
 
19
#include <stdlib.h>             /* free(), EXIT_FAILURE, strtoul(),
 
 
20
                                   realloc(), EXIT_SUCCESS, malloc(),
 
47
22
#include <stdlib.h>             /* getenv() */
 
48
23
#include <dirent.h>             /* opendir(), readdir(), closedir() */
 
49
 
#include <inttypes.h>           /* intmax_t, SCNdMAX */
 
50
 
#include <sys/stat.h>           /* struct stat, lstat(), S_ISLNK */
 
 
27
#include <iso646.h>             /* and */
 
 
28
#include <sys/wait.h>           /* waitpid(), WIFEXITED(),
 
52
31
sig_atomic_t interrupted_by_signal = 0;
 
 
58
37
static bool usplash_write(const char *cmd, const char *arg){
 
60
 
   * usplash_write("TIMEOUT", "15") will write "TIMEOUT 15\0"
 
61
 
   * usplash_write("PULSATE", NULL) will write "PULSATE\0"
 
 
39
   * usplash_write("TIMEOUT", "15"); -> "TIMEOUT 15\0"
 
 
40
   * usplash_write("PULSATE", NULL); -> "PULSATE\0"
 
 
170
148
    for(struct dirent *proc_ent = readdir(proc_dir);
 
171
149
        proc_ent != NULL;
 
172
150
        proc_ent = readdir(proc_dir)){
 
177
 
        ret = sscanf(proc_ent->d_name, "%" SCNdMAX "%n", &tmpmax,
 
179
 
        if(ret < 1 or tmpmax != (pid_t)tmpmax
 
180
 
           or proc_ent->d_name[numchars] != '\0'){
 
 
151
      pid_t pid = (pid_t) strtoul(proc_ent->d_name, NULL, 10);
 
186
156
      /* Find the executable name by doing readlink() on the
 
187
157
         /proc/<pid>/exe link */
 
188
158
      char exe_target[sizeof(usplash_name)];
 
190
 
        /* create file name string */
 
192
161
        ret = asprintf(&exe_link, "/proc/%s/exe", proc_ent->d_name);
 
 
196
165
          closedir(proc_dir);
 
197
166
          return EXIT_FAILURE;
 
200
 
        /* Check that it refers to a symlink owned by root:root */
 
201
 
        struct stat exe_stat;
 
202
 
        ret = lstat(exe_link, &exe_stat);
 
214
 
        if(not S_ISLNK(exe_stat.st_mode)
 
215
 
           or exe_stat.st_uid != 0
 
216
 
           or exe_stat.st_gid != 0){
 
221
168
        sret = readlink(exe_link, exe_target, sizeof(exe_target));
 
224
174
      if((sret == ((ssize_t)sizeof(exe_target)-1))
 
225
175
         and (memcmp(usplash_name, exe_target,
 
 
457
407
      return EXIT_SUCCESS;
 
460
 
  }                             /* end of non-loop while() */
 
462
412
  /* If we got here, an error or interrupt must have happened */
 
464
 
  /* Create argc and argv for new usplash*/
 
465
414
  int cmdline_argc = 0;
 
466
415
  char **cmdline_argv = malloc(sizeof(char *));
 
 
416
  /* Create argv and argc for new usplash*/
 
468
418
    size_t position = 0;
 
469
419
    while(position < cmdline_len){
 
 
483
433
    cmdline_argv[cmdline_argc] = NULL;
 
485
435
  /* Kill old usplash */
 
486
 
  kill(usplash_pid, SIGTERM);
 
 
436
    kill(usplash_pid, SIGTERM);
 
488
438
  while(kill(usplash_pid, 0) == 0){
 
489
439
    kill(usplash_pid, SIGKILL);