/mandos/release

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

« back to all changes in this revision

Viewing changes to plugins.d/usplash.c

  • Committer: Teddy Hogeborn
  • Date: 2019-03-18 22:29:25 UTC
  • mto: This revision was merged to the branch mainline in revision 382.
  • Revision ID: teddy@recompile.se-20190318222925-jvhek84dgcfgj6g3
mandos-ctl: Refactor tests

* mandos-ctl: Where the clients names "foo" and "barbar" do not refer
              to the actual mock clients in the TestCommand class,
              change all occurrences of these names to "client1" and
              "client2" (or just "client" when only one is used) .
              Also change all test doubles to use correct terminology;
              some things called mocks are actually stubs or spies,
              and rename all true mocks to have "mock" in their names.
              Also eliminate duplicate values in tests; derive values
              from previously defined values whenever possible.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/usplash.c
2
2
/*
3
3
 * Usplash - Read a password from usplash and output it
4
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
 
5
 * Copyright © 2008-2018 Teddy Hogeborn
 
6
 * Copyright © 2008-2018 Björn Påhlsson
 
7
 * 
 
8
 * This file is part of Mandos.
 
9
 * 
 
10
 * Mandos is free software: you can redistribute it and/or modify it
 
11
 * under the terms of the GNU General Public License as published by
 
12
 * the Free Software Foundation, either version 3 of the License, or
 
13
 * (at your option) any later version.
 
14
 * 
 
15
 * Mandos is distributed in the hope that it will be useful, but
14
16
 * WITHOUT ANY WARRANTY; without even the implied warranty of
15
17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16
18
 * General Public License for more details.
17
19
 * 
18
20
 * 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
 * along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
21
22
 * 
22
 
 * Contact the authors at <mandos@fukt.bsnet.se>.
 
23
 * Contact the authors at <mandos@recompile.se>.
23
24
 */
24
25
 
25
 
#define _GNU_SOURCE             /* asprintf() */
 
26
#define _GNU_SOURCE             /* asprintf(), TEMP_FAILURE_RETRY() */
26
27
#include <signal.h>             /* sig_atomic_t, struct sigaction,
27
28
                                   sigemptyset(), sigaddset(), SIGINT,
28
29
                                   SIGHUP, SIGTERM, sigaction(),
31
32
#include <fcntl.h>              /* open(), O_WRONLY, O_RDONLY */
32
33
#include <iso646.h>             /* and, or, not*/
33
34
#include <errno.h>              /* errno, EINTR */
 
35
#include <error.h>
34
36
#include <sys/types.h>          /* size_t, ssize_t, pid_t, DIR, struct
35
37
                                   dirent */
36
38
#include <stddef.h>             /* NULL */
37
 
#include <string.h>             /* strlen(), memcmp() */
38
 
#include <stdio.h>              /* asprintf(), perror() */
 
39
#include <string.h>             /* strlen(), memcmp(), strerror() */
 
40
#include <stdio.h>              /* asprintf(), vasprintf(), vprintf(),
 
41
                                   fprintf() */
39
42
#include <unistd.h>             /* close(), write(), readlink(),
40
43
                                   read(), STDOUT_FILENO, sleep(),
41
44
                                   fork(), setuid(), geteuid(),
42
45
                                   setsid(), chdir(), dup2(),
43
46
                                   STDERR_FILENO, execv() */
44
47
#include <stdlib.h>             /* free(), EXIT_FAILURE, realloc(),
45
 
                                   EXIT_SUCCESS, malloc(), _exit() */
46
 
#include <stdlib.h>             /* getenv() */
 
48
                                   EXIT_SUCCESS, malloc(), _exit(),
 
49
                                   getenv() */
47
50
#include <dirent.h>             /* opendir(), readdir(), closedir() */
48
51
#include <inttypes.h>           /* intmax_t, strtoimax() */
49
52
#include <sys/stat.h>           /* struct stat, lstat(), S_ISLNK */
 
53
#include <sysexits.h>           /* EX_OSERR, EX_UNAVAILABLE */
 
54
#include <argz.h>               /* argz_count(), argz_extract() */
 
55
#include <stdarg.h>             /* va_list, va_start(), ... */
50
56
 
51
57
sig_atomic_t interrupted_by_signal = 0;
52
 
 
53
 
static void termination_handler(__attribute__((unused))int signum){
 
58
int signal_received;
 
59
const char usplash_name[] = "/sbin/usplash";
 
60
 
 
61
/* Function to use when printing errors */
 
62
__attribute__((format (gnu_printf, 3, 4)))
 
63
void error_plus(int status, int errnum, const char *formatstring,
 
64
                ...){
 
65
  va_list ap;
 
66
  char *text;
 
67
  int ret;
 
68
  
 
69
  va_start(ap, formatstring);
 
70
  ret = vasprintf(&text, formatstring, ap);
 
71
  if(ret == -1){
 
72
    fprintf(stderr, "Mandos plugin %s: ",
 
73
            program_invocation_short_name);
 
74
    vfprintf(stderr, formatstring, ap);
 
75
    fprintf(stderr, ": ");
 
76
    fprintf(stderr, "%s\n", strerror(errnum));
 
77
    error(status, errno, "vasprintf while printing error");
 
78
    return;
 
79
  }
 
80
  fprintf(stderr, "Mandos plugin ");
 
81
  error(status, errnum, "%s", text);
 
82
  free(text);
 
83
}
 
84
 
 
85
static void termination_handler(int signum){
 
86
  if(interrupted_by_signal){
 
87
    return;
 
88
  }
54
89
  interrupted_by_signal = 1;
 
90
  signal_received = signum;
55
91
}
56
92
 
57
 
static bool usplash_write(const char *cmd, const char *arg){
 
93
static bool usplash_write(int *fifo_fd_r,
 
94
                          const char *cmd, const char *arg){
58
95
  /* 
59
 
   * usplash_write("TIMEOUT", "15") will write "TIMEOUT 15\0"
60
 
   * usplash_write("PULSATE", NULL) will write "PULSATE\0"
 
96
   * usplash_write(&fd, "TIMEOUT", "15") will write "TIMEOUT 15\0"
 
97
   * usplash_write(&fd, "PULSATE", NULL) will write "PULSATE\0"
61
98
   * SEE ALSO
62
99
   *         usplash_write(8)
63
100
   */
64
101
  int ret;
65
 
  int fifo_fd;
66
 
  do {
67
 
    fifo_fd = open("/dev/.initramfs/usplash_fifo", O_WRONLY);
68
 
    if(fifo_fd == -1 and (errno != EINTR or interrupted_by_signal)){
 
102
  if(*fifo_fd_r == -1){
 
103
    ret = open("/dev/.initramfs/usplash_fifo", O_WRONLY);
 
104
    if(ret == -1){
69
105
      return false;
70
106
    }
71
 
  } while(fifo_fd == -1);
 
107
    *fifo_fd_r = ret;
 
108
  }
72
109
  
73
110
  const char *cmd_line;
74
111
  size_t cmd_line_len;
75
112
  char *cmd_line_alloc = NULL;
76
113
  if(arg == NULL){
77
114
    cmd_line = cmd;
78
 
    cmd_line_len = strlen(cmd);
 
115
    cmd_line_len = strlen(cmd) + 1;
79
116
  } else {
80
117
    do {
81
118
      ret = asprintf(&cmd_line_alloc, "%s %s", cmd, arg);
82
 
      if(ret == -1 and (errno != EINTR or interrupted_by_signal)){
 
119
      if(ret == -1){
83
120
        int e = errno;
84
 
        close(fifo_fd);
 
121
        close(*fifo_fd_r);
85
122
        errno = e;
86
123
        return false;
87
124
      }
92
129
  
93
130
  size_t written = 0;
94
131
  ssize_t sret = 0;
95
 
  while(not interrupted_by_signal and written < cmd_line_len){
96
 
    sret = write(fifo_fd, cmd_line + written,
 
132
  while(written < cmd_line_len){
 
133
    sret = write(*fifo_fd_r, cmd_line + written,
97
134
                 cmd_line_len - written);
98
135
    if(sret == -1){
99
 
      if(errno != EINTR or interrupted_by_signal){
100
 
        int e = errno;
101
 
        close(fifo_fd);
102
 
        free(cmd_line_alloc);
103
 
        errno = e;
104
 
        return false;
105
 
      } else {
106
 
        continue;
107
 
      }
 
136
      int e = errno;
 
137
      close(*fifo_fd_r);
 
138
      free(cmd_line_alloc);
 
139
      errno = e;
 
140
      return false;
108
141
    }
109
142
    written += (size_t)sret;
110
143
  }
111
144
  free(cmd_line_alloc);
112
 
  do {
113
 
    ret = close(fifo_fd);
114
 
    if(ret == -1 and (errno != EINTR or interrupted_by_signal)){
115
 
      return false;
116
 
    }
117
 
  } while(ret == -1);
118
 
  if(interrupted_by_signal){
119
 
    return false;
120
 
  }
 
145
  
121
146
  return true;
122
147
}
123
148
 
 
149
/* Create prompt string */
 
150
char *makeprompt(void){
 
151
  int ret = 0;
 
152
  char *prompt;
 
153
  const char *const cryptsource = getenv("cryptsource");
 
154
  const char *const crypttarget = getenv("crypttarget");
 
155
  const char prompt_start[] = "Enter passphrase to unlock the disk";
 
156
  
 
157
  if(cryptsource == NULL){
 
158
    if(crypttarget == NULL){
 
159
      ret = asprintf(&prompt, "%s: ", prompt_start);
 
160
    } else {
 
161
      ret = asprintf(&prompt, "%s (%s): ", prompt_start,
 
162
                     crypttarget);
 
163
    }
 
164
  } else {
 
165
    if(crypttarget == NULL){
 
166
      ret = asprintf(&prompt, "%s %s: ", prompt_start, cryptsource);
 
167
    } else {
 
168
      ret = asprintf(&prompt, "%s %s (%s): ", prompt_start,
 
169
                     cryptsource, crypttarget);
 
170
    }
 
171
  }
 
172
  if(ret == -1){
 
173
    return NULL;
 
174
  }
 
175
  return prompt;
 
176
}
 
177
 
 
178
pid_t find_usplash(char **cmdline_r, size_t *cmdline_len_r){
 
179
  int ret = 0;
 
180
  ssize_t sret = 0;
 
181
  char *cmdline = NULL;
 
182
  size_t cmdline_len = 0;
 
183
  DIR *proc_dir = opendir("/proc");
 
184
  if(proc_dir == NULL){
 
185
    error_plus(0, errno, "opendir");
 
186
    return -1;
 
187
  }
 
188
  errno = 0;
 
189
  for(struct dirent *proc_ent = readdir(proc_dir);
 
190
      proc_ent != NULL;
 
191
      proc_ent = readdir(proc_dir)){
 
192
    pid_t pid;
 
193
    {
 
194
      intmax_t tmpmax;
 
195
      char *tmp;
 
196
      tmpmax = strtoimax(proc_ent->d_name, &tmp, 10);
 
197
      if(errno != 0 or tmp == proc_ent->d_name or *tmp != '\0'
 
198
         or tmpmax != (pid_t)tmpmax){
 
199
        /* Not a process */
 
200
        errno = 0;
 
201
        continue;
 
202
      }
 
203
      pid = (pid_t)tmpmax;
 
204
    }
 
205
    /* Find the executable name by doing readlink() on the
 
206
       /proc/<pid>/exe link */
 
207
    char exe_target[sizeof(usplash_name)];
 
208
    {
 
209
      /* create file name string */
 
210
      char *exe_link;
 
211
      ret = asprintf(&exe_link, "/proc/%s/exe", proc_ent->d_name);
 
212
      if(ret == -1){
 
213
        error_plus(0, errno, "asprintf");
 
214
        goto fail_find_usplash;
 
215
      }
 
216
      
 
217
      /* Check that it refers to a symlink owned by root:root */
 
218
      struct stat exe_stat;
 
219
      ret = lstat(exe_link, &exe_stat);
 
220
      if(ret == -1){
 
221
        if(errno == ENOENT){
 
222
          free(exe_link);
 
223
          continue;
 
224
        }
 
225
        error_plus(0, errno, "lstat");
 
226
        free(exe_link);
 
227
        goto fail_find_usplash;
 
228
      }
 
229
      if(not S_ISLNK(exe_stat.st_mode)
 
230
         or exe_stat.st_uid != 0
 
231
         or exe_stat.st_gid != 0){
 
232
        free(exe_link);
 
233
        continue;
 
234
      }
 
235
        
 
236
      sret = readlink(exe_link, exe_target, sizeof(exe_target));
 
237
      free(exe_link);
 
238
    }
 
239
    /* Compare executable name */
 
240
    if((sret != ((ssize_t)sizeof(exe_target)-1))
 
241
       or (memcmp(usplash_name, exe_target,
 
242
                  sizeof(exe_target)-1) != 0)){
 
243
      /* Not it */
 
244
      continue;
 
245
    }
 
246
    /* Found usplash */
 
247
    /* Read and save the command line of usplash in "cmdline" */
 
248
    {
 
249
      /* Open /proc/<pid>/cmdline  */
 
250
      int cl_fd;
 
251
      {
 
252
        char *cmdline_filename;
 
253
        ret = asprintf(&cmdline_filename, "/proc/%s/cmdline",
 
254
                       proc_ent->d_name);
 
255
        if(ret == -1){
 
256
          error_plus(0, errno, "asprintf");
 
257
          goto fail_find_usplash;
 
258
        }
 
259
        cl_fd = open(cmdline_filename, O_RDONLY);
 
260
        free(cmdline_filename);
 
261
        if(cl_fd == -1){
 
262
          error_plus(0, errno, "open");
 
263
          goto fail_find_usplash;
 
264
        }
 
265
      }
 
266
      size_t cmdline_allocated = 0;
 
267
      char *tmp;
 
268
      const size_t blocksize = 1024;
 
269
      do {
 
270
        /* Allocate more space? */
 
271
        if(cmdline_len + blocksize > cmdline_allocated){
 
272
          tmp = realloc(cmdline, cmdline_allocated + blocksize);
 
273
          if(tmp == NULL){
 
274
            error_plus(0, errno, "realloc");
 
275
            close(cl_fd);
 
276
            goto fail_find_usplash;
 
277
          }
 
278
          cmdline = tmp;
 
279
          cmdline_allocated += blocksize;
 
280
        }
 
281
        /* Read data */
 
282
        sret = read(cl_fd, cmdline + cmdline_len,
 
283
                    cmdline_allocated - cmdline_len);
 
284
        if(sret == -1){
 
285
          error_plus(0, errno, "read");
 
286
          close(cl_fd);
 
287
          goto fail_find_usplash;
 
288
        }
 
289
        cmdline_len += (size_t)sret;
 
290
      } while(sret != 0);
 
291
      ret = close(cl_fd);
 
292
      if(ret == -1){
 
293
        error_plus(0, errno, "close");
 
294
        goto fail_find_usplash;
 
295
      }
 
296
    }
 
297
    /* Close directory */
 
298
    ret = closedir(proc_dir);
 
299
    if(ret == -1){
 
300
      error_plus(0, errno, "closedir");
 
301
      goto fail_find_usplash;
 
302
    }
 
303
    /* Success */
 
304
    *cmdline_r = cmdline;
 
305
    *cmdline_len_r = cmdline_len;
 
306
    return pid;
 
307
  }
 
308
  
 
309
 fail_find_usplash:
 
310
  
 
311
  free(cmdline);
 
312
  if(proc_dir != NULL){
 
313
    int e = errno;
 
314
    closedir(proc_dir);
 
315
    errno = e;
 
316
  }
 
317
  return 0;
 
318
}
 
319
 
124
320
int main(__attribute__((unused))int argc,
125
321
         __attribute__((unused))char **argv){
126
322
  int ret = 0;
127
323
  ssize_t sret;
128
 
  bool an_error_occured = false;
 
324
  int fifo_fd = -1;
 
325
  int outfifo_fd = -1;
 
326
  char *buf = NULL;
 
327
  size_t buf_len = 0;
 
328
  pid_t usplash_pid = -1;
 
329
  bool usplash_accessed = false;
 
330
  int status = EXIT_FAILURE;    /* Default failure exit status */
129
331
  
130
 
  /* Create prompt string */
131
 
  char *prompt = NULL;
132
 
  {
133
 
    const char *const cryptsource = getenv("cryptsource");
134
 
    const char *const crypttarget = getenv("crypttarget");
135
 
    const char prompt_start[] = "Enter passphrase to unlock the disk";
136
 
    
137
 
    if(cryptsource == NULL){
138
 
      if(crypttarget == NULL){
139
 
        ret = asprintf(&prompt, "%s: ", prompt_start);
140
 
      } else {
141
 
        ret = asprintf(&prompt, "%s (%s): ", prompt_start,
142
 
                       crypttarget);
143
 
      }
144
 
    } else {
145
 
      if(crypttarget == NULL){
146
 
        ret = asprintf(&prompt, "%s %s: ", prompt_start, cryptsource);
147
 
      } else {
148
 
        ret = asprintf(&prompt, "%s %s (%s): ", prompt_start,
149
 
                       cryptsource, crypttarget);
150
 
      }
151
 
    }
152
 
    if(ret == -1){
153
 
      return EXIT_FAILURE;
154
 
    }
 
332
  char *prompt = makeprompt();
 
333
  if(prompt == NULL){
 
334
    status = EX_OSERR;
 
335
    goto failure;
155
336
  }
156
337
  
157
338
  /* Find usplash process */
158
 
  pid_t usplash_pid = 0;
159
339
  char *cmdline = NULL;
160
340
  size_t cmdline_len = 0;
161
 
  const char usplash_name[] = "/sbin/usplash";
162
 
  {
163
 
    DIR *proc_dir = opendir("/proc");
164
 
    if(proc_dir == NULL){
165
 
      free(prompt);
166
 
      perror("opendir");
167
 
      return EXIT_FAILURE;
168
 
    }
169
 
    for(struct dirent *proc_ent = readdir(proc_dir);
170
 
        proc_ent != NULL;
171
 
        proc_ent = readdir(proc_dir)){
172
 
      pid_t pid;
173
 
      {
174
 
        intmax_t tmpmax;
175
 
        char *tmp;
176
 
        errno = 0;
177
 
        tmpmax = strtoimax(proc_ent->d_name, &tmp, 10);
178
 
        if(errno != 0 or tmp == proc_ent->d_name or *tmp != '\0'
179
 
           or tmpmax != (pid_t)tmpmax){
180
 
          /* Not a process */
181
 
          continue;
182
 
        }
183
 
        pid = (pid_t)tmpmax;
184
 
      }
185
 
      /* Find the executable name by doing readlink() on the
186
 
         /proc/<pid>/exe link */
187
 
      char exe_target[sizeof(usplash_name)];
188
 
      {
189
 
        /* create file name string */
190
 
        char *exe_link;
191
 
        ret = asprintf(&exe_link, "/proc/%s/exe", proc_ent->d_name);
192
 
        if(ret == -1){
193
 
          perror("asprintf");
194
 
          free(prompt);
195
 
          closedir(proc_dir);
196
 
          return EXIT_FAILURE;
197
 
        }
198
 
        
199
 
        /* Check that it refers to a symlink owned by root:root */
200
 
        struct stat exe_stat;
201
 
        ret = lstat(exe_link, &exe_stat);
202
 
        if(ret == -1){
203
 
          if(errno == ENOENT){
204
 
            free(exe_link);
205
 
            continue;
206
 
          }
207
 
          perror("lstat");
208
 
          free(exe_link);
209
 
          free(prompt);
210
 
          closedir(proc_dir);
211
 
          return EXIT_FAILURE;
212
 
        }
213
 
        if(not S_ISLNK(exe_stat.st_mode)
214
 
           or exe_stat.st_uid != 0
215
 
           or exe_stat.st_gid != 0){
216
 
          free(exe_link);
217
 
          continue;
218
 
        }
219
 
        
220
 
        sret = readlink(exe_link, exe_target, sizeof(exe_target));
221
 
        free(exe_link);
222
 
      }
223
 
      if((sret == ((ssize_t)sizeof(exe_target)-1))
224
 
         and (memcmp(usplash_name, exe_target,
225
 
                     sizeof(exe_target)-1) == 0)){
226
 
        usplash_pid = pid;
227
 
        /* Read and save the command line of usplash in "cmdline" */
228
 
        {
229
 
          /* Open /proc/<pid>/cmdline  */
230
 
          int cl_fd;
231
 
          {
232
 
            char *cmdline_filename;
233
 
            ret = asprintf(&cmdline_filename, "/proc/%s/cmdline",
234
 
                           proc_ent->d_name);
235
 
            if(ret == -1){
236
 
              perror("asprintf");
237
 
              free(prompt);
238
 
              closedir(proc_dir);
239
 
              return EXIT_FAILURE;
240
 
            }
241
 
            cl_fd = open(cmdline_filename, O_RDONLY);
242
 
            if(cl_fd == -1){
243
 
              perror("open");
244
 
              free(cmdline_filename);
245
 
              free(prompt);
246
 
              closedir(proc_dir);
247
 
              return EXIT_FAILURE;
248
 
            }
249
 
            free(cmdline_filename);
250
 
          }
251
 
          size_t cmdline_allocated = 0;
252
 
          char *tmp;
253
 
          const size_t blocksize = 1024;
254
 
          do {
255
 
            if(cmdline_len + blocksize > cmdline_allocated){
256
 
              tmp = realloc(cmdline, cmdline_allocated + blocksize);
257
 
              if(tmp == NULL){
258
 
                perror("realloc");
259
 
                free(cmdline);
260
 
                free(prompt);
261
 
                closedir(proc_dir);
262
 
                return EXIT_FAILURE;
263
 
              }
264
 
              cmdline = tmp;
265
 
              cmdline_allocated += blocksize;
266
 
            }
267
 
            sret = read(cl_fd, cmdline + cmdline_len,
268
 
                        cmdline_allocated - cmdline_len);
269
 
            if(sret == -1){
270
 
              perror("read");
271
 
              free(cmdline);
272
 
              free(prompt);
273
 
              closedir(proc_dir);
274
 
              return EXIT_FAILURE;
275
 
            }
276
 
            cmdline_len += (size_t)sret;
277
 
          } while(sret != 0);
278
 
          close(cl_fd);
279
 
        }
280
 
        break;
281
 
      }
282
 
    }
283
 
    closedir(proc_dir);
284
 
  }
 
341
  usplash_pid = find_usplash(&cmdline, &cmdline_len);
285
342
  if(usplash_pid == 0){
286
 
    free(prompt);
287
 
    return EXIT_FAILURE;
 
343
    status = EX_UNAVAILABLE;
 
344
    goto failure;
288
345
  }
289
346
  
290
347
  /* Set up the signal handler */
293
350
      new_action = { .sa_handler = termination_handler,
294
351
                     .sa_flags = 0 };
295
352
    sigemptyset(&new_action.sa_mask);
296
 
    sigaddset(&new_action.sa_mask, SIGINT);
297
 
    sigaddset(&new_action.sa_mask, SIGHUP);
298
 
    sigaddset(&new_action.sa_mask, SIGTERM);
 
353
    ret = sigaddset(&new_action.sa_mask, SIGINT);
 
354
    if(ret == -1){
 
355
      error_plus(0, errno, "sigaddset");
 
356
      status = EX_OSERR;
 
357
      goto failure;
 
358
    }
 
359
    ret = sigaddset(&new_action.sa_mask, SIGHUP);
 
360
    if(ret == -1){
 
361
      error_plus(0, errno, "sigaddset");
 
362
      status = EX_OSERR;
 
363
      goto failure;
 
364
    }
 
365
    ret = sigaddset(&new_action.sa_mask, SIGTERM);
 
366
    if(ret == -1){
 
367
      error_plus(0, errno, "sigaddset");
 
368
      status = EX_OSERR;
 
369
      goto failure;
 
370
    }
299
371
    ret = sigaction(SIGINT, NULL, &old_action);
300
372
    if(ret == -1){
301
 
      perror("sigaction");
302
 
      free(prompt);
303
 
      return EXIT_FAILURE;
 
373
      if(errno != EINTR){
 
374
        error_plus(0, errno, "sigaction");
 
375
        status = EX_OSERR;
 
376
      }
 
377
      goto failure;
304
378
    }
305
379
    if(old_action.sa_handler != SIG_IGN){
306
380
      ret = sigaction(SIGINT, &new_action, NULL);
307
381
      if(ret == -1){
308
 
        perror("sigaction");
309
 
        free(prompt);
310
 
        return EXIT_FAILURE;
 
382
        if(errno != EINTR){
 
383
          error_plus(0, errno, "sigaction");
 
384
          status = EX_OSERR;
 
385
        }
 
386
        goto failure;
311
387
      }
312
388
    }
313
389
    ret = sigaction(SIGHUP, NULL, &old_action);
314
390
    if(ret == -1){
315
 
      perror("sigaction");
316
 
      free(prompt);
317
 
      return EXIT_FAILURE;
 
391
      if(errno != EINTR){
 
392
        error_plus(0, errno, "sigaction");
 
393
        status = EX_OSERR;
 
394
      }
 
395
      goto failure;
318
396
    }
319
397
    if(old_action.sa_handler != SIG_IGN){
320
398
      ret = sigaction(SIGHUP, &new_action, NULL);
321
399
      if(ret == -1){
322
 
        perror("sigaction");
323
 
        free(prompt);
324
 
        return EXIT_FAILURE;
 
400
        if(errno != EINTR){
 
401
          error_plus(0, errno, "sigaction");
 
402
          status = EX_OSERR;
 
403
        }
 
404
        goto failure;
325
405
      }
326
406
    }
327
407
    ret = sigaction(SIGTERM, NULL, &old_action);
328
408
    if(ret == -1){
329
 
      perror("sigaction");
330
 
      free(prompt);
331
 
      return EXIT_FAILURE;
 
409
      if(errno != EINTR){
 
410
        error_plus(0, errno, "sigaction");
 
411
        status = EX_OSERR;
 
412
      }
 
413
      goto failure;
332
414
    }
333
415
    if(old_action.sa_handler != SIG_IGN){
334
416
      ret = sigaction(SIGTERM, &new_action, NULL);
335
417
      if(ret == -1){
336
 
        perror("sigaction");
337
 
        free(prompt);
338
 
        return EXIT_FAILURE;
 
418
        if(errno != EINTR){
 
419
          error_plus(0, errno, "sigaction");
 
420
          status = EX_OSERR;
 
421
        }
 
422
        goto failure;
339
423
      }
340
424
    }
341
425
  }
342
426
  
 
427
  usplash_accessed = true;
343
428
  /* Write command to FIFO */
344
 
  if(not interrupted_by_signal){
345
 
    if(not usplash_write("TIMEOUT", "0")
346
 
       and (errno != EINTR)){
347
 
      perror("usplash_write");
348
 
      an_error_occured = true;
349
 
    }
350
 
  }
351
 
  if(not interrupted_by_signal and not an_error_occured){
352
 
    if(not usplash_write("INPUTQUIET", prompt)
353
 
       and (errno != EINTR)){
354
 
      perror("usplash_write");
355
 
      an_error_occured = true;
356
 
    }
357
 
  }
 
429
  if(not usplash_write(&fifo_fd, "TIMEOUT", "0")){
 
430
    if(errno != EINTR){
 
431
      error_plus(0, errno, "usplash_write");
 
432
      status = EX_OSERR;
 
433
    }
 
434
    goto failure;
 
435
  }
 
436
  
 
437
  if(interrupted_by_signal){
 
438
    goto failure;
 
439
  }
 
440
  
 
441
  if(not usplash_write(&fifo_fd, "INPUTQUIET", prompt)){
 
442
    if(errno != EINTR){
 
443
      error_plus(0, errno, "usplash_write");
 
444
      status = EX_OSERR;
 
445
    }
 
446
    goto failure;
 
447
  }
 
448
  
 
449
  if(interrupted_by_signal){
 
450
    goto failure;
 
451
  }
 
452
  
358
453
  free(prompt);
359
 
  
360
 
  /* This is not really a loop; while() is used to be able to "break"
361
 
     out of it; those breaks are marked "Big" */
362
 
  while(not interrupted_by_signal and not an_error_occured){
363
 
    char *buf = NULL;
364
 
    size_t buf_len = 0;
365
 
    
366
 
    /* Open FIFO */
367
 
    int fifo_fd;
368
 
    do {
369
 
      fifo_fd = open("/dev/.initramfs/usplash_outfifo", O_RDONLY);
370
 
      if(fifo_fd == -1){
371
 
        if(errno != EINTR){
372
 
          perror("open");
373
 
          an_error_occured = true;
374
 
          break;
375
 
        }
376
 
        if(interrupted_by_signal){
377
 
          break;
378
 
        }
379
 
      }
380
 
    } while(fifo_fd == -1);
381
 
    if(interrupted_by_signal or an_error_occured){
382
 
      break;                    /* Big */
383
 
    }
384
 
    
385
 
    /* Read from FIFO */
386
 
    size_t buf_allocated = 0;
387
 
    const size_t blocksize = 1024;
388
 
    do {
389
 
      if(buf_len + blocksize > buf_allocated){
390
 
        char *tmp = realloc(buf, buf_allocated + blocksize);
391
 
        if(tmp == NULL){
392
 
          perror("realloc");
393
 
          an_error_occured = true;
394
 
          break;
395
 
        }
396
 
        buf = tmp;
397
 
        buf_allocated += blocksize;
398
 
      }
399
 
      do {
400
 
        sret = read(fifo_fd, buf + buf_len, buf_allocated - buf_len);
401
 
        if(sret == -1){
402
 
          if(errno != EINTR){
403
 
            perror("read");
404
 
            an_error_occured = true;
405
 
            break;
406
 
          }
407
 
          if(interrupted_by_signal){
408
 
            break;
409
 
          }
410
 
        }
411
 
      } while(sret == -1);
412
 
      if(interrupted_by_signal or an_error_occured){
413
 
        break;
414
 
      }
415
 
      
416
 
      buf_len += (size_t)sret;
417
 
    } while(sret != 0);
418
 
    close(fifo_fd);
419
 
    if(interrupted_by_signal or an_error_occured){
420
 
      break;                    /* Big */
421
 
    }
422
 
    
423
 
    if(not usplash_write("TIMEOUT", "15")
424
 
       and (errno != EINTR)){
425
 
        perror("usplash_write");
426
 
        an_error_occured = true;
427
 
    }
428
 
    if(interrupted_by_signal or an_error_occured){
429
 
      break;                    /* Big */
430
 
    }
431
 
    
432
 
    /* Print password to stdout */
433
 
    size_t written = 0;
434
 
    while(written < buf_len){
435
 
      do {
436
 
        sret = write(STDOUT_FILENO, buf + written, buf_len - written);
437
 
        if(sret == -1){
438
 
          if(errno != EINTR){
439
 
            perror("write");
440
 
            an_error_occured = true;
441
 
            break;
442
 
          }
443
 
          if(interrupted_by_signal){
444
 
            break;
445
 
          }
446
 
        }
447
 
      } while(sret == -1);
448
 
      if(interrupted_by_signal or an_error_occured){
449
 
        break;
450
 
      }
451
 
      written += (size_t)sret;
452
 
    }
453
 
    free(buf);
454
 
    if(not interrupted_by_signal and not an_error_occured){
455
 
      free(cmdline);
456
 
      return EXIT_SUCCESS;
457
 
    }
458
 
    break;                      /* Big */
459
 
  }                             /* end of non-loop while() */
460
 
  
461
 
  /* If we got here, an error or interrupt must have happened */
462
 
  
463
 
  /* Create argc and argv for new usplash*/
464
 
  int cmdline_argc = 0;
465
 
  char **cmdline_argv = malloc(sizeof(char *));
466
 
  {
467
 
    size_t position = 0;
468
 
    while(position < cmdline_len){
469
 
      char **tmp = realloc(cmdline_argv,
470
 
                           (sizeof(char *)
471
 
                            * (size_t)(cmdline_argc + 2)));
 
454
  prompt = NULL;
 
455
  
 
456
  /* Read reply from usplash */
 
457
  /* Open FIFO */
 
458
  outfifo_fd = open("/dev/.initramfs/usplash_outfifo", O_RDONLY);
 
459
  if(outfifo_fd == -1){
 
460
    if(errno != EINTR){
 
461
      error_plus(0, errno, "open");
 
462
      status = EX_OSERR;
 
463
    }
 
464
    goto failure;
 
465
  }
 
466
  
 
467
  if(interrupted_by_signal){
 
468
    goto failure;
 
469
  }
 
470
  
 
471
  /* Read from FIFO */
 
472
  size_t buf_allocated = 0;
 
473
  const size_t blocksize = 1024;
 
474
  do {
 
475
    /* Allocate more space */
 
476
    if(buf_len + blocksize > buf_allocated){
 
477
      char *tmp = realloc(buf, buf_allocated + blocksize);
472
478
      if(tmp == NULL){
473
 
        perror("realloc");
474
 
        free(cmdline_argv);
475
 
        return EXIT_FAILURE;
476
 
      }
477
 
      cmdline_argv = tmp;
478
 
      cmdline_argv[cmdline_argc] = cmdline + position;
479
 
      cmdline_argc++;
480
 
      position += strlen(cmdline + position) + 1;
481
 
    }
482
 
    cmdline_argv[cmdline_argc] = NULL;
483
 
  }
 
479
        if(errno != EINTR){
 
480
          error_plus(0, errno, "realloc");
 
481
          status = EX_OSERR;
 
482
        }
 
483
        goto failure;
 
484
      }
 
485
      buf = tmp;
 
486
      buf_allocated += blocksize;
 
487
    }
 
488
    sret = read(outfifo_fd, buf + buf_len,
 
489
                buf_allocated - buf_len);
 
490
    if(sret == -1){
 
491
      if(errno != EINTR){
 
492
        error_plus(0, errno, "read");
 
493
        status = EX_OSERR;
 
494
      }
 
495
      close(outfifo_fd);
 
496
      goto failure;
 
497
    }
 
498
    if(interrupted_by_signal){
 
499
      break;
 
500
    }
 
501
    
 
502
    buf_len += (size_t)sret;
 
503
  } while(sret != 0);
 
504
  ret = close(outfifo_fd);
 
505
  if(ret == -1){
 
506
    if(errno != EINTR){
 
507
      error_plus(0, errno, "close");
 
508
      status = EX_OSERR;
 
509
    }
 
510
    goto failure;
 
511
  }
 
512
  outfifo_fd = -1;
 
513
  
 
514
  if(interrupted_by_signal){
 
515
    goto failure;
 
516
  }
 
517
  
 
518
  if(not usplash_write(&fifo_fd, "TIMEOUT", "15")){
 
519
    if(errno != EINTR){
 
520
      error_plus(0, errno, "usplash_write");
 
521
      status = EX_OSERR;
 
522
    }
 
523
    goto failure;
 
524
  }
 
525
  
 
526
  if(interrupted_by_signal){
 
527
    goto failure;
 
528
  }
 
529
  
 
530
  ret = close(fifo_fd);
 
531
  if(ret == -1){
 
532
    if(errno != EINTR){
 
533
      error_plus(0, errno, "close");
 
534
      status = EX_OSERR;
 
535
    }
 
536
    goto failure;
 
537
  }
 
538
  fifo_fd = -1;
 
539
  
 
540
  /* Print password to stdout */
 
541
  size_t written = 0;
 
542
  while(written < buf_len){
 
543
    do {
 
544
      sret = write(STDOUT_FILENO, buf + written, buf_len - written);
 
545
      if(sret == -1){
 
546
        if(errno != EINTR){
 
547
          error_plus(0, errno, "write");
 
548
          status = EX_OSERR;
 
549
        }
 
550
        goto failure;
 
551
      }
 
552
    } while(sret == -1);
 
553
    
 
554
    if(interrupted_by_signal){
 
555
      goto failure;
 
556
    }
 
557
    written += (size_t)sret;
 
558
  }
 
559
  free(buf);
 
560
  buf = NULL;
 
561
  
 
562
  if(interrupted_by_signal){
 
563
    goto failure;
 
564
  }
 
565
  
 
566
  free(cmdline);
 
567
  return EXIT_SUCCESS;
 
568
  
 
569
 failure:
 
570
  
 
571
  free(buf);
 
572
  
 
573
  free(prompt);
 
574
  
 
575
  /* If usplash was never accessed, we can stop now */
 
576
  if(not usplash_accessed){
 
577
    return status;
 
578
  }
 
579
  
 
580
  /* Close FIFO */
 
581
  if(fifo_fd != -1){
 
582
    ret = close(fifo_fd);
 
583
    if(ret == -1 and errno != EINTR){
 
584
      error_plus(0, errno, "close");
 
585
    }
 
586
    fifo_fd = -1;
 
587
  }
 
588
  
 
589
  /* Close output FIFO */
 
590
  if(outfifo_fd != -1){
 
591
    ret = close(outfifo_fd);
 
592
    if(ret == -1){
 
593
      error_plus(0, errno, "close");
 
594
    }
 
595
  }
 
596
  
 
597
  /* Create argv for new usplash*/
 
598
  char **cmdline_argv = malloc((argz_count(cmdline, cmdline_len) + 1)
 
599
                               * sizeof(char *)); /* Count args */
 
600
  if(cmdline_argv == NULL){
 
601
    error_plus(0, errno, "malloc");
 
602
    return status;
 
603
  }
 
604
  argz_extract(cmdline, cmdline_len, cmdline_argv); /* Create argv */
 
605
  
484
606
  /* Kill old usplash */
485
607
  kill(usplash_pid, SIGTERM);
486
608
  sleep(2);
488
610
    kill(usplash_pid, SIGKILL);
489
611
    sleep(1);
490
612
  }
 
613
  
491
614
  pid_t new_usplash_pid = fork();
492
615
  if(new_usplash_pid == 0){
493
616
    /* Child; will become new usplash process */
496
619
       the real user ID (_mandos) */
497
620
    ret = setuid(geteuid());
498
621
    if(ret == -1){
499
 
      perror("setuid");
 
622
      error_plus(0, errno, "setuid");
500
623
    }
501
624
    
502
625
    setsid();
503
626
    ret = chdir("/");
 
627
    if(ret == -1){
 
628
      error_plus(0, errno, "chdir");
 
629
      _exit(EX_OSERR);
 
630
    }
504
631
/*     if(fork() != 0){ */
505
632
/*       _exit(EXIT_SUCCESS); */
506
633
/*     } */
507
634
    ret = dup2(STDERR_FILENO, STDOUT_FILENO); /* replace our stdout */
508
635
    if(ret == -1){
509
 
      perror("dup2");
510
 
      _exit(EXIT_FAILURE);
 
636
      error_plus(0, errno, "dup2");
 
637
      _exit(EX_OSERR);
511
638
    }
512
639
    
513
640
    execv(usplash_name, cmdline_argv);
514
641
    if(not interrupted_by_signal){
515
 
      perror("execv");
 
642
      error_plus(0, errno, "execv");
516
643
    }
517
644
    free(cmdline);
518
645
    free(cmdline_argv);
519
 
    _exit(EXIT_FAILURE);
 
646
    _exit(EX_OSERR);
520
647
  }
521
648
  free(cmdline);
522
649
  free(cmdline_argv);
523
650
  sleep(2);
524
 
  if(not usplash_write("PULSATE", NULL)
525
 
     and (errno != EINTR)){
526
 
    perror("usplash_write");
527
 
  }
528
 
  
529
 
  return EXIT_FAILURE;
 
651
  if(not usplash_write(&fifo_fd, "PULSATE", NULL)){
 
652
    if(errno != EINTR){
 
653
      error_plus(0, errno, "usplash_write");
 
654
    }
 
655
  }
 
656
  
 
657
  /* Close FIFO (again) */
 
658
  if(fifo_fd != -1){
 
659
    ret = close(fifo_fd);
 
660
    if(ret == -1 and errno != EINTR){
 
661
      error_plus(0, errno, "close");
 
662
    }
 
663
    fifo_fd = -1;
 
664
  }
 
665
  
 
666
  if(interrupted_by_signal){
 
667
    struct sigaction signal_action = { .sa_handler = SIG_DFL };
 
668
    sigemptyset(&signal_action.sa_mask);
 
669
    ret = (int)TEMP_FAILURE_RETRY(sigaction(signal_received,
 
670
                                            &signal_action, NULL));
 
671
    if(ret == -1){
 
672
      error_plus(0, errno, "sigaction");
 
673
    }
 
674
    do {
 
675
      ret = raise(signal_received);
 
676
    } while(ret != 0 and errno == EINTR);
 
677
    if(ret != 0){
 
678
      error_plus(0, errno, "raise");
 
679
      abort();
 
680
    }
 
681
    TEMP_FAILURE_RETRY(pause());
 
682
  }
 
683
  
 
684
  return status;
530
685
}