/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

  • Committer: Teddy Hogeborn
  • Date: 2009-02-05 02:42:39 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090205024239-th0d287ifnl7l3tv
* plugin-runner.c (handle_sigchld): Clarify logic.

Show diffs side-by-side

added added

removed removed

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