/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/splashy.c

  • Committer: Teddy Hogeborn
  • Date: 2016-06-03 17:27:03 UTC
  • Revision ID: teddy@recompile.se-20160603172703-mc6tjor6rhq4xy74
mandos: Bug fix: Do multiprocessing cleanup correctly on exit

* mandos (main): Save module "multiprocessing" and open file "wnull"
                 as scope variables accessible by function cleanup(),
                 since the module and global variable may not be
                 accessible when the cleanup() function is run as
                 scheduled by atexit().

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/splashy.c
1
 
#define _GNU_SOURCE             /* asprintf() */
 
1
/*  -*- coding: utf-8 -*- */
 
2
/*
 
3
 * Splashy - Read a password from splashy and output it
 
4
 * 
 
5
 * Copyright © 2008-2016 Teddy Hogeborn
 
6
 * Copyright © 2008-2016 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 <mandos@recompile.se>.
 
23
 */
 
24
 
 
25
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY(), asprintf() */
2
26
#include <signal.h>             /* sig_atomic_t, struct sigaction,
3
27
                                   sigemptyset(), sigaddset(), SIGINT,
4
28
                                   SIGHUP, SIGTERM, sigaction,
5
29
                                   SIG_IGN, kill(), SIGKILL */
6
30
#include <stddef.h>             /* NULL */
7
31
#include <stdlib.h>             /* getenv() */
8
 
#include <stdio.h>              /* asprintf(), perror() */
9
 
#include <stdlib.h>             /* EXIT_FAILURE, free(), strtoul(),
 
32
#include <stdio.h>              /* asprintf(), vasprintf(), vprintf(),
 
33
                                   fprintf() */
 
34
#include <stdlib.h>             /* EXIT_FAILURE, free(),
10
35
                                   EXIT_SUCCESS */
11
36
#include <sys/types.h>          /* pid_t, DIR, struct dirent,
12
37
                                   ssize_t */
13
38
#include <dirent.h>             /* opendir(), readdir(), closedir() */
 
39
#include <inttypes.h>           /* intmax_t, strtoimax() */
14
40
#include <sys/stat.h>           /* struct stat, lstat(), S_ISLNK */
15
41
#include <iso646.h>             /* not, or, and */
16
42
#include <unistd.h>             /* readlink(), fork(), execl(),
17
43
                                   sleep(), dup2() STDERR_FILENO,
18
 
                                   STDOUT_FILENO, _exit() */
19
 
#include <string.h>             /* memcmp() */
20
 
#include <errno.h>              /* errno */
 
44
                                   STDOUT_FILENO, _exit(),
 
45
                                   pause() */
 
46
#include <string.h>             /* memcmp(), strerror() */
 
47
#include <errno.h>              /* errno, EACCES, ENOTDIR, ELOOP,
 
48
                                   ENOENT, ENAMETOOLONG, EMFILE,
 
49
                                   ENFILE, ENOMEM, ENOEXEC, EINVAL,
 
50
                                   E2BIG, EFAULT, EIO, ETXTBSY,
 
51
                                   EISDIR, ELIBBAD, EPERM, EINTR,
 
52
                                   ECHILD */
 
53
#include <error.h>              /* error() */
21
54
#include <sys/wait.h>           /* waitpid(), WIFEXITED(),
22
55
                                   WEXITSTATUS() */
 
56
#include <sysexits.h>           /* EX_OSERR, EX_OSFILE,
 
57
                                   EX_UNAVAILABLE */
 
58
#include <stdarg.h>             /* va_list, va_start(), ... */
23
59
 
24
60
sig_atomic_t interrupted_by_signal = 0;
25
 
 
26
 
static void termination_handler(__attribute__((unused))int signum){
 
61
int signal_received;
 
62
 
 
63
/* Function to use when printing errors */
 
64
__attribute__((format (gnu_printf, 3, 4)))
 
65
void error_plus(int status, int errnum, const char *formatstring,
 
66
                ...){
 
67
  va_list ap;
 
68
  char *text;
 
69
  int ret;
 
70
  
 
71
  va_start(ap, formatstring);
 
72
  ret = vasprintf(&text, formatstring, ap);
 
73
  if(ret == -1){
 
74
    fprintf(stderr, "Mandos plugin %s: ",
 
75
            program_invocation_short_name);
 
76
    vfprintf(stderr, formatstring, ap);
 
77
    fprintf(stderr, ": ");
 
78
    fprintf(stderr, "%s\n", strerror(errnum));
 
79
    error(status, errno, "vasprintf while printing error");
 
80
    return;
 
81
  }
 
82
  fprintf(stderr, "Mandos plugin ");
 
83
  error(status, errnum, "%s", text);
 
84
  free(text);
 
85
}
 
86
 
 
87
 
 
88
static void termination_handler(int signum){
 
89
  if(interrupted_by_signal){
 
90
    return;
 
91
  }
27
92
  interrupted_by_signal = 1;
 
93
  signal_received = signum;
28
94
}
29
95
 
30
96
int main(__attribute__((unused))int argc,
31
97
         __attribute__((unused))char **argv){
32
98
  int ret = 0;
 
99
  char *prompt = NULL;
 
100
  DIR *proc_dir = NULL;
 
101
  pid_t splashy_pid = 0;
 
102
  pid_t splashy_command_pid = 0;
 
103
  int exitstatus = EXIT_FAILURE;
33
104
  
34
105
  /* Create prompt string */
35
 
  char *prompt = NULL;
36
106
  {
37
107
    const char *const cryptsource = getenv("cryptsource");
38
108
    const char *const crypttarget = getenv("crypttarget");
55
125
      }
56
126
    }
57
127
    if(ret == -1){
58
 
      return EXIT_FAILURE;
 
128
      prompt = NULL;
 
129
      exitstatus = EX_OSERR;
 
130
      goto failure;
59
131
    }
60
132
  }
61
133
  
62
134
  /* Find splashy process */
63
 
  pid_t splashy_pid = 0;
64
135
  {
65
136
    const char splashy_name[] = "/sbin/splashy";
66
 
    DIR *proc_dir = opendir("/proc");
 
137
    proc_dir = opendir("/proc");
67
138
    if(proc_dir == NULL){
68
 
      free(prompt);
69
 
      perror("opendir");
70
 
      return EXIT_FAILURE;
 
139
      int e = errno;
 
140
      error_plus(0, errno, "opendir");
 
141
      switch(e){
 
142
      case EACCES:
 
143
      case ENOTDIR:
 
144
      case ELOOP:
 
145
      case ENOENT:
 
146
      default:
 
147
        exitstatus = EX_OSFILE;
 
148
        break;
 
149
      case ENAMETOOLONG:
 
150
      case EMFILE:
 
151
      case ENFILE:
 
152
      case ENOMEM:
 
153
        exitstatus = EX_OSERR;
 
154
        break;
 
155
      }
 
156
      goto failure;
71
157
    }
72
158
    for(struct dirent *proc_ent = readdir(proc_dir);
73
159
        proc_ent != NULL;
74
160
        proc_ent = readdir(proc_dir)){
75
 
      pid_t pid = (pid_t) strtoul(proc_ent->d_name, NULL, 10);
76
 
      if(pid == 0){
77
 
        /* Not a process */
78
 
        continue;
 
161
      pid_t pid;
 
162
      {
 
163
        intmax_t tmpmax;
 
164
        char *tmp;
 
165
        errno = 0;
 
166
        tmpmax = strtoimax(proc_ent->d_name, &tmp, 10);
 
167
        if(errno != 0 or tmp == proc_ent->d_name or *tmp != '\0'
 
168
           or tmpmax != (pid_t)tmpmax){
 
169
          /* Not a process */
 
170
          continue;
 
171
        }
 
172
        pid = (pid_t)tmpmax;
79
173
      }
80
174
      /* Find the executable name by doing readlink() on the
81
175
         /proc/<pid>/exe link */
85
179
        char *exe_link;
86
180
        ret = asprintf(&exe_link, "/proc/%s/exe", proc_ent->d_name);
87
181
        if(ret == -1){
88
 
          perror("asprintf");
89
 
          free(prompt);
90
 
          closedir(proc_dir);
91
 
          return EXIT_FAILURE;
 
182
          error_plus(0, errno, "asprintf");
 
183
          exitstatus = EX_OSERR;
 
184
          goto failure;
92
185
        }
93
186
        
94
187
        /* Check that it refers to a symlink owned by root:root */
95
188
        struct stat exe_stat;
96
189
        ret = lstat(exe_link, &exe_stat);
97
190
        if(ret == -1){
98
 
          perror("lstat");
 
191
          if(errno == ENOENT){
 
192
            free(exe_link);
 
193
            continue;
 
194
          }
 
195
          int e = errno;
 
196
          error_plus(0, errno, "lstat");
99
197
          free(exe_link);
100
 
          free(prompt);
101
 
          closedir(proc_dir);
102
 
          return EXIT_FAILURE;
 
198
          switch(e){
 
199
          case EACCES:
 
200
          case ENOTDIR:
 
201
          case ELOOP:
 
202
          default:
 
203
            exitstatus = EX_OSFILE;
 
204
            break;
 
205
          case ENAMETOOLONG:
 
206
            exitstatus = EX_OSERR;
 
207
            break;
 
208
          }
 
209
          goto failure;
103
210
        }
104
211
        if(not S_ISLNK(exe_stat.st_mode)
105
212
           or exe_stat.st_uid != 0
119
226
      }
120
227
    }
121
228
    closedir(proc_dir);
 
229
    proc_dir = NULL;
122
230
  }
123
231
  if(splashy_pid == 0){
124
 
    free(prompt);
125
 
    return EXIT_FAILURE;
 
232
    exitstatus = EX_UNAVAILABLE;
 
233
    goto failure;
126
234
  }
127
235
  
128
236
  /* Set up the signal handler */
131
239
      new_action = { .sa_handler = termination_handler,
132
240
                     .sa_flags = 0 };
133
241
    sigemptyset(&new_action.sa_mask);
134
 
    sigaddset(&new_action.sa_mask, SIGINT);
135
 
    sigaddset(&new_action.sa_mask, SIGHUP);
136
 
    sigaddset(&new_action.sa_mask, SIGTERM);
 
242
    ret = sigaddset(&new_action.sa_mask, SIGINT);
 
243
    if(ret == -1){
 
244
      error_plus(0, errno, "sigaddset");
 
245
      exitstatus = EX_OSERR;
 
246
      goto failure;
 
247
    }
 
248
    ret = sigaddset(&new_action.sa_mask, SIGHUP);
 
249
    if(ret == -1){
 
250
      error_plus(0, errno, "sigaddset");
 
251
      exitstatus = EX_OSERR;
 
252
      goto failure;
 
253
    }
 
254
    ret = sigaddset(&new_action.sa_mask, SIGTERM);
 
255
    if(ret == -1){
 
256
      error_plus(0, errno, "sigaddset");
 
257
      exitstatus = EX_OSERR;
 
258
      goto failure;
 
259
    }
137
260
    ret = sigaction(SIGINT, NULL, &old_action);
138
261
    if(ret == -1){
139
 
      perror("sigaction");
140
 
      free(prompt);
141
 
      return EXIT_FAILURE;
 
262
      error_plus(0, errno, "sigaction");
 
263
      exitstatus = EX_OSERR;
 
264
      goto failure;
142
265
    }
143
266
    if(old_action.sa_handler != SIG_IGN){
144
267
      ret = sigaction(SIGINT, &new_action, NULL);
145
268
      if(ret == -1){
146
 
        perror("sigaction");
147
 
        free(prompt);
148
 
        return EXIT_FAILURE;
 
269
        error_plus(0, errno, "sigaction");
 
270
        exitstatus = EX_OSERR;
 
271
        goto failure;
149
272
      }
150
273
    }
151
274
    ret = sigaction(SIGHUP, NULL, &old_action);
152
275
    if(ret == -1){
153
 
      perror("sigaction");
154
 
      free(prompt);
155
 
      return EXIT_FAILURE;
 
276
      error_plus(0, errno, "sigaction");
 
277
      exitstatus = EX_OSERR;
 
278
      goto failure;
156
279
    }
157
280
    if(old_action.sa_handler != SIG_IGN){
158
281
      ret = sigaction(SIGHUP, &new_action, NULL);
159
282
      if(ret == -1){
160
 
        perror("sigaction");
161
 
        free(prompt);
162
 
        return EXIT_FAILURE;
 
283
        error_plus(0, errno, "sigaction");
 
284
        exitstatus = EX_OSERR;
 
285
        goto failure;
163
286
      }
164
287
    }
165
288
    ret = sigaction(SIGTERM, NULL, &old_action);
166
289
    if(ret == -1){
167
 
      perror("sigaction");
168
 
      free(prompt);
169
 
      return EXIT_FAILURE;
 
290
      error_plus(0, errno, "sigaction");
 
291
      exitstatus = EX_OSERR;
 
292
      goto failure;
170
293
    }
171
294
    if(old_action.sa_handler != SIG_IGN){
172
295
      ret = sigaction(SIGTERM, &new_action, NULL);
173
296
      if(ret == -1){
174
 
        perror("sigaction");
175
 
        free(prompt);
176
 
        return EXIT_FAILURE;
 
297
        error_plus(0, errno, "sigaction");
 
298
        exitstatus = EX_OSERR;
 
299
        goto failure;
177
300
      }
178
301
    }
179
302
  }
180
303
  
 
304
  if(interrupted_by_signal){
 
305
    goto failure;
 
306
  }
 
307
  
181
308
  /* Fork off the splashy command to prompt for password */
182
 
  pid_t splashy_command_pid = 0;
183
 
  if(not interrupted_by_signal){
184
 
    splashy_command_pid = fork();
185
 
    if(splashy_command_pid == -1){
186
 
      if(not interrupted_by_signal){
187
 
        perror("fork");
188
 
      }
189
 
      return EXIT_FAILURE;
190
 
    }
191
 
    /* Child */
192
 
    if(splashy_command_pid == 0){
 
309
  splashy_command_pid = fork();
 
310
  if(splashy_command_pid != 0 and interrupted_by_signal){
 
311
    goto failure;
 
312
  }
 
313
  if(splashy_command_pid == -1){
 
314
    error_plus(0, errno, "fork");
 
315
    exitstatus = EX_OSERR;
 
316
    goto failure;
 
317
  }
 
318
  /* Child */
 
319
  if(splashy_command_pid == 0){
 
320
    if(not interrupted_by_signal){
193
321
      const char splashy_command[] = "/sbin/splashy_update";
194
 
      ret = execl(splashy_command, splashy_command, prompt,
195
 
                  (char *)NULL);
196
 
      if(not interrupted_by_signal){
197
 
        perror("execl");
 
322
      execl(splashy_command, splashy_command, prompt, (char *)NULL);
 
323
      int e = errno;
 
324
      error_plus(0, errno, "execl");
 
325
      switch(e){
 
326
      case EACCES:
 
327
      case ENOENT:
 
328
      case ENOEXEC:
 
329
      case EINVAL:
 
330
        _exit(EX_UNAVAILABLE);
 
331
      case ENAMETOOLONG:
 
332
      case E2BIG:
 
333
      case ENOMEM:
 
334
      case EFAULT:
 
335
      case EIO:
 
336
      case EMFILE:
 
337
      case ENFILE:
 
338
      case ETXTBSY:
 
339
      default:
 
340
        _exit(EX_OSERR);
 
341
      case ENOTDIR:
 
342
      case ELOOP:
 
343
      case EISDIR:
 
344
#ifdef ELIBBAD
 
345
      case ELIBBAD:             /* Linux only */
 
346
#endif
 
347
      case EPERM:
 
348
        _exit(EX_OSFILE);
198
349
      }
199
 
      free(prompt);
200
 
      _exit(EXIT_FAILURE);
201
350
    }
 
351
    free(prompt);
 
352
    _exit(EXIT_FAILURE);
202
353
  }
203
354
  
204
355
  /* Parent */
205
356
  free(prompt);
 
357
  prompt = NULL;
 
358
  
 
359
  if(interrupted_by_signal){
 
360
    goto failure;
 
361
  }
206
362
  
207
363
  /* Wait for command to complete */
208
 
  if(not interrupted_by_signal and splashy_command_pid != 0){
 
364
  {
209
365
    int status;
210
 
    ret = waitpid(splashy_command_pid, &status, 0);
 
366
    do {
 
367
      ret = waitpid(splashy_command_pid, &status, 0);
 
368
    } while(ret == -1 and errno == EINTR
 
369
            and not interrupted_by_signal);
 
370
    if(interrupted_by_signal){
 
371
      goto failure;
 
372
    }
211
373
    if(ret == -1){
212
 
      if(errno != EINTR){
213
 
        perror("waitpid");
214
 
      }
 
374
      error_plus(0, errno, "waitpid");
215
375
      if(errno == ECHILD){
216
376
        splashy_command_pid = 0;
217
377
      }
218
378
    } else {
219
379
      /* The child process has exited */
220
380
      splashy_command_pid = 0;
221
 
      if(not interrupted_by_signal and WIFEXITED(status)
222
 
         and WEXITSTATUS(status)==0){
 
381
      if(WIFEXITED(status) and WEXITSTATUS(status) == 0){
223
382
        return EXIT_SUCCESS;
224
383
      }
225
384
    }
226
385
  }
227
 
  kill(splashy_pid, SIGTERM);
228
 
  if(interrupted_by_signal and splashy_command_pid != 0){
229
 
    kill(splashy_command_pid, SIGTERM);
230
 
  }
231
 
  sleep(2);
232
 
  while(kill(splashy_pid, 0) == 0){
233
 
    kill(splashy_pid, SIGKILL);
234
 
    sleep(1);
235
 
  }
236
 
  pid_t new_splashy_pid = fork();
237
 
  if(new_splashy_pid == 0){
238
 
    /* Child; will become new splashy process */
239
 
    
240
 
    /* Make the effective user ID (root) the only user ID instead of
241
 
       the real user ID (mandos) */
242
 
    ret = setuid(geteuid());
243
 
    if(ret == -1){
244
 
      perror("setuid");
245
 
    }
246
 
    
247
 
    setsid();
248
 
    ret = chdir("/");
249
 
/*     if(fork() != 0){ */
250
 
/*       _exit(EXIT_SUCCESS); */
251
 
/*     } */
252
 
    ret = dup2(STDERR_FILENO, STDOUT_FILENO); /* replace our stdout */
253
 
    if(ret == -1){
254
 
      perror("dup2");
255
 
      _exit(EXIT_FAILURE);
256
 
    }
257
 
    
258
 
    execl("/sbin/splashy", "/sbin/splashy", "boot", (char *)NULL);
259
 
    if(not interrupted_by_signal){
260
 
      perror("execl");
261
 
    }
262
 
    _exit(EXIT_FAILURE);
263
 
  }
264
 
  
265
 
  return EXIT_FAILURE;
 
386
  
 
387
 failure:
 
388
  
 
389
  free(prompt);
 
390
  
 
391
  if(proc_dir != NULL){
 
392
    TEMP_FAILURE_RETRY(closedir(proc_dir));
 
393
  }
 
394
  
 
395
  if(splashy_command_pid != 0){
 
396
    TEMP_FAILURE_RETRY(kill(splashy_command_pid, SIGTERM));
 
397
    
 
398
    TEMP_FAILURE_RETRY(kill(splashy_pid, SIGTERM));
 
399
    sleep(2);
 
400
    while(TEMP_FAILURE_RETRY(kill(splashy_pid, 0)) == 0){
 
401
      TEMP_FAILURE_RETRY(kill(splashy_pid, SIGKILL));
 
402
      sleep(1);
 
403
    }
 
404
    pid_t new_splashy_pid = (pid_t)TEMP_FAILURE_RETRY(fork());
 
405
    if(new_splashy_pid == 0){
 
406
      /* Child; will become new splashy process */
 
407
      
 
408
      /* Make the effective user ID (root) the only user ID instead of
 
409
         the real user ID (_mandos) */
 
410
      ret = setuid(geteuid());
 
411
      if(ret == -1){
 
412
        error_plus(0, errno, "setuid");
 
413
      }
 
414
      
 
415
      setsid();
 
416
      ret = chdir("/");
 
417
      if(ret == -1){
 
418
        error_plus(0, errno, "chdir");
 
419
      }
 
420
/*       if(fork() != 0){ */
 
421
/*      _exit(EXIT_SUCCESS); */
 
422
/*       } */
 
423
      ret = dup2(STDERR_FILENO, STDOUT_FILENO); /* replace stdout */
 
424
      if(ret == -1){
 
425
        error_plus(0, errno, "dup2");
 
426
        _exit(EX_OSERR);
 
427
      }
 
428
      
 
429
      execl("/sbin/splashy", "/sbin/splashy", "boot", (char *)NULL);
 
430
      {
 
431
        int e = errno;
 
432
        error_plus(0, errno, "execl");
 
433
        switch(e){
 
434
        case EACCES:
 
435
        case ENOENT:
 
436
        case ENOEXEC:
 
437
        default:
 
438
          _exit(EX_UNAVAILABLE);
 
439
        case ENAMETOOLONG:
 
440
        case E2BIG:
 
441
        case ENOMEM:
 
442
          _exit(EX_OSERR);
 
443
        case ENOTDIR:
 
444
        case ELOOP:
 
445
          _exit(EX_OSFILE);
 
446
        }
 
447
      }
 
448
    }
 
449
  }
 
450
  
 
451
  if(interrupted_by_signal){
 
452
    struct sigaction signal_action;
 
453
    sigemptyset(&signal_action.sa_mask);
 
454
    signal_action.sa_handler = SIG_DFL;
 
455
    ret = (int)TEMP_FAILURE_RETRY(sigaction(signal_received,
 
456
                                            &signal_action, NULL));
 
457
    if(ret == -1){
 
458
      error_plus(0, errno, "sigaction");
 
459
    }
 
460
    do {
 
461
      ret = raise(signal_received);
 
462
    } while(ret != 0 and errno == EINTR);
 
463
    if(ret != 0){
 
464
      error_plus(0, errno, "raise");
 
465
      abort();
 
466
    }
 
467
    TEMP_FAILURE_RETRY(pause());
 
468
  }
 
469
  
 
470
  return exitstatus;
266
471
}