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

  • Committer: Teddy Hogeborn
  • Date: 2019-03-12 20:13:34 UTC
  • Revision ID: teddy@recompile.se-20190312201334-my3htrprewjosuw5
mandos-ctl: Refactor

* mandos-ctl: Reorder everything into logical order; put main() first,
              and put every subsequent definition as soon as possible
              after its first use, except superclasses which need to
              be placed before the classes inheriting from them.
              Reorder all tests to match.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/plymouth.c
2
2
/*
3
3
 * Plymouth - Read a password from Plymouth and output it
4
4
 * 
5
 
 * Copyright © 2010-2011 Teddy Hogeborn
6
 
 * Copyright © 2010-2011 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 © 2010-2018 Teddy Hogeborn
 
6
 * Copyright © 2010-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
26
#define _GNU_SOURCE             /* asprintf(), TEMP_FAILURE_RETRY() */
36
37
#include <stddef.h>             /* NULL */
37
38
#include <string.h>             /* strchr(), memcmp() */
38
39
#include <stdio.h>              /* asprintf(), perror(), fopen(),
39
 
                                   fscanf() */
 
40
                                   fscanf(), vasprintf(), fprintf(),
 
41
                                   vfprintf() */
40
42
#include <unistd.h>             /* close(), readlink(), read(),
41
43
                                   fork(), setsid(), chdir(), dup2(),
42
44
                                   STDERR_FILENO, execv(), access() */
50
52
#include <error.h>              /* error() */
51
53
#include <errno.h>              /* TEMP_FAILURE_RETRY */
52
54
#include <argz.h>               /* argz_count(), argz_extract() */
 
55
#include <stdarg.h>             /* va_list, va_start(), ... */
53
56
 
54
57
sig_atomic_t interrupted_by_signal = 0;
55
 
const char plymouth_pid[] = "/dev/.initramfs/plymouth.pid";
 
58
 
 
59
/* Used by Ubuntu 11.04 (Natty Narwahl) */
 
60
const char plymouth_old_old_pid[] = "/dev/.initramfs/plymouth.pid";
 
61
/* Used by Ubuntu 11.10 (Oneiric Ocelot) */
 
62
const char plymouth_old_pid[] = "/run/initramfs/plymouth.pid";
 
63
/* Used by Debian 9 (stretch) */
 
64
const char plymouth_pid[] = "/run/plymouth/pid";
 
65
 
56
66
const char plymouth_path[] = "/bin/plymouth";
57
67
const char plymouthd_path[] = "/sbin/plymouthd";
58
68
const char *plymouthd_default_argv[] = {"/sbin/plymouthd",
59
69
                                        "--mode=boot",
60
70
                                        "--attach-to-session",
61
 
                                        "--pid-file="
62
 
                                        "/dev/.initramfs/"
63
 
                                        "plymouth.pid",
64
71
                                        NULL };
65
72
 
66
73
static void termination_handler(__attribute__((unused))int signum){
70
77
  interrupted_by_signal = 1;
71
78
}
72
79
 
 
80
/* Function to use when printing errors */
 
81
__attribute__((format (gnu_printf, 3, 4)))
 
82
void error_plus(int status, int errnum, const char *formatstring,
 
83
                ...){
 
84
  va_list ap;
 
85
  char *text;
 
86
  int ret;
 
87
  
 
88
  va_start(ap, formatstring);
 
89
  ret = vasprintf(&text, formatstring, ap);
 
90
  if(ret == -1){
 
91
    fprintf(stderr, "Mandos plugin %s: ",
 
92
            program_invocation_short_name);
 
93
    vfprintf(stderr, formatstring, ap);
 
94
    fprintf(stderr, ": ");
 
95
    fprintf(stderr, "%s\n", strerror(errnum));
 
96
    error(status, errno, "vasprintf while printing error");
 
97
    return;
 
98
  }
 
99
  fprintf(stderr, "Mandos plugin ");
 
100
  error(status, errnum, "%s", text);
 
101
  free(text);
 
102
}
 
103
 
73
104
/* Create prompt string */
74
105
char *makeprompt(void){
75
106
  int ret = 0;
109
140
bool become_a_daemon(void){
110
141
  int ret = setuid(geteuid());
111
142
  if(ret == -1){
112
 
    error(0, errno, "setuid");
 
143
    error_plus(0, errno, "setuid");
113
144
  }
114
145
    
115
146
  setsid();
116
147
  ret = chdir("/");
117
148
  if(ret == -1){
118
 
    error(0, errno, "chdir");
 
149
    error_plus(0, errno, "chdir");
119
150
    return false;
120
151
  }
121
152
  ret = dup2(STDERR_FILENO, STDOUT_FILENO); /* replace our stdout */
122
153
  if(ret == -1){
123
 
    error(0, errno, "dup2");
 
154
    error_plus(0, errno, "dup2");
124
155
    return false;
125
156
  }
126
157
  return true;
127
158
}
128
159
 
 
160
__attribute__((nonnull (2, 3)))
129
161
bool exec_and_wait(pid_t *pid_return, const char *path,
130
 
                   const char **argv, bool interruptable,
 
162
                   const char * const *argv, bool interruptable,
131
163
                   bool daemonize){
132
164
  int status;
133
165
  int ret;
134
166
  pid_t pid;
135
167
  pid = fork();
136
168
  if(pid == -1){
137
 
    error(0, errno, "fork");
 
169
    error_plus(0, errno, "fork");
138
170
    return false;
139
171
  }
140
172
  if(pid == 0){
145
177
      }
146
178
    }
147
179
    
148
 
    char **new_argv = NULL;
 
180
    char **new_argv = malloc(sizeof(const char *));
 
181
    if(new_argv == NULL){
 
182
      error_plus(0, errno, "malloc");
 
183
      _exit(EX_OSERR);
 
184
    }
149
185
    char **tmp;
150
186
    int i = 0;
151
 
    for (; argv[i]!=NULL; i++){
152
 
      tmp = realloc(new_argv, sizeof(const char *) * ((size_t)i + 1));
153
 
      if (tmp == NULL){
154
 
        error(0, errno, "realloc");
 
187
    for (; argv[i] != NULL; i++){
 
188
      tmp = realloc(new_argv, sizeof(const char *) * ((size_t)i + 2));
 
189
      if(tmp == NULL){
 
190
        error_plus(0, errno, "realloc");
155
191
        free(new_argv);
156
192
        _exit(EX_OSERR);
157
193
      }
161
197
    new_argv[i] = NULL;
162
198
    
163
199
    execv(path, (char *const *)new_argv);
164
 
    error(0, errno, "execv");
 
200
    error_plus(0, errno, "execv");
165
201
    _exit(EXIT_FAILURE);
166
202
  }
167
203
  if(pid_return != NULL){
176
212
    return false;
177
213
  }
178
214
  if(ret == -1){
179
 
    error(0, errno, "waitpid");
 
215
    error_plus(0, errno, "waitpid");
180
216
    return false;
181
217
  }
182
218
  if(WIFEXITED(status) and (WEXITSTATUS(status) == 0)){
185
221
  return false;
186
222
}
187
223
 
 
224
__attribute__((nonnull))
188
225
int is_plymouth(const struct dirent *proc_entry){
189
226
  int ret;
190
227
  {
191
 
    uintmax_t maxvalue;
 
228
    uintmax_t proc_id;
192
229
    char *tmp;
193
230
    errno = 0;
194
 
    maxvalue = strtoumax(proc_entry->d_name, &tmp, 10);
 
231
    proc_id = strtoumax(proc_entry->d_name, &tmp, 10);
195
232
 
196
233
    if(errno != 0 or *tmp != '\0'
197
 
       or maxvalue != (uintmax_t)((pid_t)maxvalue)){
 
234
       or proc_id != (uintmax_t)((pid_t)proc_id)){
198
235
      return 0;
199
236
    }
200
237
  }
202
239
  char *exe_link;
203
240
  ret = asprintf(&exe_link, "/proc/%s/exe", proc_entry->d_name);
204
241
  if(ret == -1){
205
 
    error(0, errno, "asprintf");
 
242
    error_plus(0, errno, "asprintf");
206
243
    return 0;
207
244
  }
208
245
  
211
248
  if(ret == -1){
212
249
    free(exe_link);
213
250
    if(errno != ENOENT){
214
 
      error(0, errno, "lstat");
 
251
      error_plus(0, errno, "lstat");
215
252
    }
216
253
    return 0;
217
254
  }
235
272
 
236
273
pid_t get_pid(void){
237
274
  int ret;
 
275
  uintmax_t proc_id = 0;
238
276
  FILE *pidfile = fopen(plymouth_pid, "r");
239
 
  uintmax_t maxvalue = 0;
 
277
  /* Try the new pid file location */
240
278
  if(pidfile != NULL){
241
 
    ret = fscanf(pidfile, "%" SCNuMAX, &maxvalue);
 
279
    ret = fscanf(pidfile, "%" SCNuMAX, &proc_id);
242
280
    if(ret != 1){
243
 
      maxvalue = 0;
 
281
      proc_id = 0;
244
282
    }
245
283
    fclose(pidfile);
246
284
  }
247
 
  if(maxvalue == 0){
248
 
    struct dirent **direntries;
 
285
  /* Try the old pid file location */
 
286
  if(proc_id == 0){
 
287
    pidfile = fopen(plymouth_old_pid, "r");
 
288
    if(pidfile != NULL){
 
289
      ret = fscanf(pidfile, "%" SCNuMAX, &proc_id);
 
290
      if(ret != 1){
 
291
        proc_id = 0;
 
292
      }
 
293
      fclose(pidfile);
 
294
    }
 
295
  }
 
296
  /* Try the old old pid file location */
 
297
  if(proc_id == 0){
 
298
    pidfile = fopen(plymouth_old_old_pid, "r");
 
299
    if(pidfile != NULL){
 
300
      ret = fscanf(pidfile, "%" SCNuMAX, &proc_id);
 
301
      if(ret != 1){
 
302
        proc_id = 0;
 
303
      }
 
304
      fclose(pidfile);
 
305
    }
 
306
  }
 
307
  /* Look for a plymouth process */
 
308
  if(proc_id == 0){
 
309
    struct dirent **direntries = NULL;
249
310
    ret = scandir("/proc", &direntries, is_plymouth, alphasort);
250
 
    if (ret == -1){
251
 
      error(0, errno, "scandir");
 
311
    if(ret == -1){
 
312
      error_plus(0, errno, "scandir");
252
313
    }
253
 
    if (ret > 0){
254
 
      ret = sscanf(direntries[0]->d_name, "%" SCNuMAX, &maxvalue);
255
 
      if (ret < 0){
256
 
        error(0, errno, "sscanf");
 
314
    if(ret > 0){
 
315
      for(int i = ret-1; i >= 0; i--){
 
316
        if(proc_id == 0){
 
317
          ret = sscanf(direntries[i]->d_name, "%" SCNuMAX, &proc_id);
 
318
          if(ret < 0){
 
319
            error_plus(0, errno, "sscanf");
 
320
          }
 
321
        }
 
322
        free(direntries[i]);
257
323
      }
258
324
    }
 
325
    /* scandir might preallocate for this variable (man page unclear).
 
326
       even if ret == 0, therefore we need to free it. */
 
327
    free(direntries);
259
328
  }
260
329
  pid_t pid;
261
 
  pid = (pid_t)maxvalue;
262
 
  if((uintmax_t)pid == maxvalue){
 
330
  pid = (pid_t)proc_id;
 
331
  if((uintmax_t)pid == proc_id){
263
332
    return pid;
264
333
  }
265
334
  
266
335
  return 0;
267
336
}
268
337
 
269
 
const char **getargv(pid_t pid){
 
338
char **getargv(pid_t pid){
270
339
  int cl_fd;
271
340
  char *cmdline_filename;
272
341
  ssize_t sret;
275
344
  ret = asprintf(&cmdline_filename, "/proc/%" PRIuMAX "/cmdline",
276
345
                 (uintmax_t)pid);
277
346
  if(ret == -1){
278
 
    error(0, errno, "asprintf");
 
347
    error_plus(0, errno, "asprintf");
279
348
    return NULL;
280
349
  }
281
350
  
283
352
  cl_fd = open(cmdline_filename, O_RDONLY);
284
353
  free(cmdline_filename);
285
354
  if(cl_fd == -1){
286
 
    error(0, errno, "open");
 
355
    error_plus(0, errno, "open");
287
356
    return NULL;
288
357
  }
289
358
  
297
366
    if(cmdline_len + blocksize > cmdline_allocated){
298
367
      tmp = realloc(cmdline, cmdline_allocated + blocksize);
299
368
      if(tmp == NULL){
300
 
        error(0, errno, "realloc");
 
369
        error_plus(0, errno, "realloc");
301
370
        free(cmdline);
302
371
        close(cl_fd);
303
372
        return NULL;
310
379
    sret = read(cl_fd, cmdline + cmdline_len,
311
380
                cmdline_allocated - cmdline_len);
312
381
    if(sret == -1){
313
 
      error(0, errno, "read");
 
382
      error_plus(0, errno, "read");
314
383
      free(cmdline);
315
384
      close(cl_fd);
316
385
      return NULL;
319
388
  } while(sret != 0);
320
389
  ret = close(cl_fd);
321
390
  if(ret == -1){
322
 
    error(0, errno, "close");
 
391
    error_plus(0, errno, "close");
323
392
    free(cmdline);
324
393
    return NULL;
325
394
  }
328
397
  char **argv = malloc((argz_count(cmdline, cmdline_len) + 1)
329
398
                       * sizeof(char *)); /* Get number of args */
330
399
  if(argv == NULL){
331
 
    error(0, errno, "argv = malloc()");
 
400
    error_plus(0, errno, "argv = malloc()");
332
401
    free(cmdline);
333
402
    return NULL;
334
403
  }
335
404
  argz_extract(cmdline, cmdline_len, argv); /* Create argv */
336
 
  return (const char **)argv;
 
405
  return argv;
337
406
}
338
407
 
339
408
int main(__attribute__((unused))int argc,
361
430
        *sig != 0; sig++){
362
431
      ret = sigaddset(&new_action.sa_mask, *sig);
363
432
      if(ret == -1){
364
 
        error(EX_OSERR, errno, "sigaddset");
 
433
        error_plus(EX_OSERR, errno, "sigaddset");
365
434
      }
366
435
      ret = sigaction(*sig, NULL, &old_action);
367
436
      if(ret == -1){
368
 
        error(EX_OSERR, errno, "sigaction");
 
437
        error_plus(EX_OSERR, errno, "sigaction");
369
438
      }
370
439
      if(old_action.sa_handler != SIG_IGN){
371
440
        ret = sigaction(*sig, &new_action, NULL);
372
441
        if(ret == -1){
373
 
          error(EX_OSERR, errno, "sigaction");
 
442
          error_plus(EX_OSERR, errno, "sigaction");
374
443
        }
375
444
      }
376
445
    }
395
464
  ret = asprintf(&prompt_arg, "--prompt=%s", prompt);
396
465
  free(prompt);
397
466
  if(ret == -1){
398
 
    error(EX_OSERR, errno, "asprintf");
 
467
    error_plus(EX_OSERR, errno, "asprintf");
399
468
  }
400
469
  
401
470
  /* plymouth ask-for-password --prompt="$prompt" */
414
483
  }
415
484
  kill_and_wait(plymouth_command_pid);
416
485
  
417
 
  const char **plymouthd_argv;
 
486
  char **plymouthd_argv = NULL;
418
487
  pid_t pid = get_pid();
419
488
  if(pid == 0){
420
 
    error(0, 0, "plymouthd pid not found");
421
 
    plymouthd_argv = plymouthd_default_argv;
 
489
    error_plus(0, 0, "plymouthd pid not found");
422
490
  } else {
423
491
    plymouthd_argv = getargv(pid);
424
492
  }
427
495
                       { plymouth_path, "quit", NULL },
428
496
                       false, false);
429
497
  if(not bret){
 
498
    if(plymouthd_argv != NULL){
 
499
      free(*plymouthd_argv);
 
500
      free(plymouthd_argv);
 
501
    }
430
502
    exit(EXIT_FAILURE);
431
503
  }
432
 
  bret = exec_and_wait(NULL, plymouthd_path, plymouthd_argv,
 
504
  bret = exec_and_wait(NULL, plymouthd_path,
 
505
                       (plymouthd_argv != NULL)
 
506
                       ? (const char * const *)plymouthd_argv
 
507
                       : plymouthd_default_argv,
433
508
                       false, true);
 
509
  if(plymouthd_argv != NULL){
 
510
    free(*plymouthd_argv);
 
511
    free(plymouthd_argv);
 
512
  }
434
513
  if(not bret){
435
514
    exit(EXIT_FAILURE);
436
515
  }