/mandos/trunk

To get this branch, use:
bzr branch http://bzr.recompile.se/loggerhead/mandos/trunk

« back to all changes in this revision

Viewing changes to plugins.d/usplash.c

merge

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
/*  -*- coding: utf-8 -*- */
2
 
/*
3
 
 * Usplash - Read a password from usplash and output it
4
 
 * 
5
 
 * Copyright © 2008,2009 Teddy Hogeborn
6
 
 * Copyright © 2008,2009 Björn Påhlsson
7
 
 * 
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.
12
 
 * 
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.
17
 
 * 
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/>.
21
 
 * 
22
 
 * Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
23
 
 * <https://www.fukt.bsnet.se/~teddy/>.
24
 
 */
25
 
 
26
1
#define _GNU_SOURCE             /* asprintf() */
27
2
#include <signal.h>             /* sig_atomic_t, struct sigaction,
28
3
                                   sigemptyset(), sigaddset(), SIGINT,
36
11
                                   dirent */
37
12
#include <stddef.h>             /* NULL */
38
13
#include <string.h>             /* strlen(), memcmp() */
39
 
#include <stdio.h>              /* asprintf(), perror(), sscanf() */
 
14
#include <stdio.h>              /* asprintf(), perror() */
40
15
#include <unistd.h>             /* close(), write(), readlink(),
41
16
                                   read(), STDOUT_FILENO, sleep(),
42
17
                                   fork(), setuid(), geteuid(),
43
18
                                   setsid(), chdir(), dup2(),
44
19
                                   STDERR_FILENO, execv() */
45
 
#include <stdlib.h>             /* free(), EXIT_FAILURE, realloc(),
46
 
                                   EXIT_SUCCESS, malloc(), _exit() */
 
20
#include <stdlib.h>             /* free(), EXIT_FAILURE, strtoul(),
 
21
                                   realloc(), EXIT_SUCCESS, malloc(),
 
22
                                   _exit() */
47
23
#include <stdlib.h>             /* getenv() */
48
24
#include <dirent.h>             /* opendir(), readdir(), closedir() */
49
 
#include <inttypes.h>           /* intmax_t, SCNdMAX */
50
25
#include <sys/stat.h>           /* struct stat, lstat(), S_ISLNK */
51
26
 
52
27
sig_atomic_t interrupted_by_signal = 0;
92
67
  }
93
68
  
94
69
  size_t written = 0;
95
 
  ssize_t sret = 0;
96
70
  while(not interrupted_by_signal and written < cmd_line_len){
97
 
    sret = write(fifo_fd, cmd_line + written,
98
 
                 cmd_line_len - written);
99
 
    if(sret == -1){
 
71
    ret = write(fifo_fd, cmd_line + written,
 
72
                cmd_line_len - written);
 
73
    if(ret == -1){
100
74
      if(errno != EINTR or interrupted_by_signal){
101
75
        int e = errno;
102
76
        close(fifo_fd);
107
81
        continue;
108
82
      }
109
83
    }
110
 
    written += (size_t)sret;
 
84
    written += (size_t)ret;
111
85
  }
112
86
  free(cmd_line_alloc);
113
87
  do{
170
144
    for(struct dirent *proc_ent = readdir(proc_dir);
171
145
        proc_ent != NULL;
172
146
        proc_ent = readdir(proc_dir)){
173
 
      pid_t pid;
174
 
      {
175
 
        intmax_t tmpmax;
176
 
        int numchars;
177
 
        ret = sscanf(proc_ent->d_name, "%" SCNdMAX "%n", &tmpmax,
178
 
                     &numchars);
179
 
        if(ret < 1 or tmpmax != (pid_t)tmpmax
180
 
           or proc_ent->d_name[numchars] != '\0'){
181
 
          /* Not a process */
182
 
          continue;
183
 
        }
184
 
        pid = (pid_t)tmpmax;
 
147
      pid_t pid = (pid_t) strtoul(proc_ent->d_name, NULL, 10);
 
148
      if(pid == 0){
 
149
        /* Not a process */
 
150
        continue;
185
151
      }
186
152
      /* Find the executable name by doing readlink() on the
187
153
         /proc/<pid>/exe link */
201
167
        struct stat exe_stat;
202
168
        ret = lstat(exe_link, &exe_stat);
203
169
        if(ret == -1){
204
 
          if(errno == ENOENT){
205
 
            free(exe_link);
206
 
            continue;
207
 
          }
208
170
          perror("lstat");
209
171
          free(exe_link);
210
172
          free(prompt);
220
182
        
221
183
        sret = readlink(exe_link, exe_target, sizeof(exe_target));
222
184
        free(exe_link);
 
185
        if(sret == -1){
 
186
          continue;
 
187
        }
223
188
      }
224
189
      if((sret == ((ssize_t)sizeof(exe_target)-1))
225
190
         and (memcmp(usplash_name, exe_target,
494
459
    /* Child; will become new usplash process */
495
460
    
496
461
    /* Make the effective user ID (root) the only user ID instead of
497
 
       the real user ID (_mandos) */
 
462
       the real user ID (mandos) */
498
463
    ret = setuid(geteuid());
499
464
    if(ret == -1){
500
465
      perror("setuid");