/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/password-prompt.c

  • Committer: teddy at recompile
  • Date: 2020-12-03 20:30:45 UTC
  • Revision ID: teddy@recompile.se-20201203203045-iqd6nq9y5nwalh1x
Minor fix of a test function

In dracut-module/password-agent, the test function
test_send_password_to_socket_EMSGSIZE() (which tests that the
send_password_to_socket() task function aborts properly when getting
EMSGSIZE when writing to the password socket), part of the test code
is supposed to find a message size which definitely does trigger
EMSGSIZE when send()ing to a socket.  Without a "break" in the proper
place, however, the size given is always exactly 1024 bytes too large.

This is very probably not a problem, since a too large message will
still be too large if it is increased by 1024 bytes, and send(2) in
practice checks the size before reading the buffer.  The biggest issue
would be if some version of send(2) would try to look at the last 1024
bytes of the message buffer before checking the message size; this
would then lead to a buffer over-read when running this test function.
(But even then there would be no security implications since the tests
are not run in the normal operation of the program.)

* dracut-module/password-agent.c
  (test_send_password_to_socket_EMSGSIZE): Break out early when ssret
  < 0 and errno == EMSGSIZE; don't allow loop to increase message_size
  again.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/password-prompt.c
1
 
/*  -*- coding: utf-8 -*- */
 
1
/*  -*- coding: utf-8; mode: c; mode: orgtbl -*- */
2
2
/*
3
 
 * Passprompt - Read a password from the terminal and print it
4
 
 *
5
 
 * Copyright © 2008 Teddy Hogeborn & Björn Påhlsson
6
 
 * 
7
 
 * This program is free software: you can redistribute it and/or
8
 
 * modify it under the terms of the GNU General Public License as
9
 
 * published by the Free Software Foundation, either version 3 of the
10
 
 * License, or (at your option) any later version.
11
 
 * 
12
 
 * This program is distributed in the hope that it will be useful, but
 
3
 * Password-prompt - Read a password from the terminal and print it
 
4
 * 
 
5
 * Copyright © 2008-2019 Teddy Hogeborn
 
6
 * Copyright © 2008-2019 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
13
16
 * WITHOUT ANY WARRANTY; without even the implied warranty of
14
17
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15
18
 * General Public License for more details.
16
19
 * 
17
20
 * You should have received a copy of the GNU General Public License
18
 
 * along with this program.  If not, see
19
 
 * <http://www.gnu.org/licenses/>.
 
21
 * along with Mandos.  If not, see <http://www.gnu.org/licenses/>.
20
22
 * 
21
 
 * Contact the authors at <https://www.fukt.bsnet.se/~belorn/> and
22
 
 * <https://www.fukt.bsnet.se/~teddy/>.
 
23
 * Contact the authors at <mandos@recompile.se>.
23
24
 */
24
25
 
25
 
#define _GNU_SOURCE             /* getline() */
 
26
#define _GNU_SOURCE             /* getline(), asprintf() */
26
27
 
27
 
#include <termios.h>            /* struct termios, tcsetattr(),
 
28
#include <termios.h>            /* struct termios, tcsetattr(),
28
29
                                   TCSAFLUSH, tcgetattr(), ECHO */
29
 
#include <unistd.h>             /* struct termios, tcsetattr(),
30
 
                                   STDIN_FILENO, TCSAFLUSH,
31
 
                                   tcgetattr(), ECHO */
 
30
#include <unistd.h>             /* access(), struct termios,
 
31
                                   tcsetattr(), STDIN_FILENO,
 
32
                                   TCSAFLUSH, tcgetattr(), ECHO,
 
33
                                   readlink() */
32
34
#include <signal.h>             /* sig_atomic_t, raise(), struct
33
35
                                   sigaction, sigemptyset(),
34
36
                                   sigaction(), sigaddset(), SIGINT,
35
 
                                   SIGQUIT, SIGHUP, SIGTERM */
 
37
                                   SIGQUIT, SIGHUP, SIGTERM,
 
38
                                   raise() */
36
39
#include <stddef.h>             /* NULL, size_t, ssize_t */
37
 
#include <sys/types.h>          /* ssize_t */
 
40
#include <sys/types.h>          /* ssize_t, struct dirent, pid_t,
 
41
                                   ssize_t, open() */
38
42
#include <stdlib.h>             /* EXIT_SUCCESS, EXIT_FAILURE,
39
 
                                   getopt_long, getenv() */
 
43
                                   getenv(), free() */
 
44
#include <dirent.h>             /* scandir(), alphasort() */
40
45
#include <stdio.h>              /* fprintf(), stderr, getline(),
41
 
                                   stdin, feof(), perror(), fputc(),
42
 
                                   stdout, getopt_long */
43
 
#include <errno.h>              /* errno, EINVAL */
 
46
                                   stdin, feof(), fputc(), vfprintf(),
 
47
                                   vasprintf() */
 
48
#include <errno.h>              /* errno, EBADF, ENOTTY, EINVAL,
 
49
                                   EFAULT, EFBIG, EIO, ENOSPC, EINTR
 
50
                                */
 
51
#include <error.h>              /* error() */
44
52
#include <iso646.h>             /* or, not */
45
53
#include <stdbool.h>            /* bool, false, true */
46
 
#include <string.h>             /* strlen, rindex, strncmp, strcmp */
 
54
#include <inttypes.h>           /* strtoumax() */
 
55
#include <sys/stat.h>           /* struct stat, lstat(), open() */
 
56
#include <string.h>             /* strlen, rindex, memcmp, strerror()
 
57
                                 */
47
58
#include <argp.h>               /* struct argp_option, struct
48
59
                                   argp_state, struct argp,
49
60
                                   argp_parse(), error_t,
50
61
                                   ARGP_KEY_ARG, ARGP_KEY_END,
51
62
                                   ARGP_ERR_UNKNOWN */
 
63
#include <sysexits.h>           /* EX_SOFTWARE, EX_OSERR,
 
64
                                   EX_UNAVAILABLE, EX_IOERR, EX_OK */
 
65
#include <fcntl.h>              /* open() */
 
66
#include <stdarg.h>             /* va_list, va_start(), ... */
52
67
 
53
 
volatile bool quit_now = false;
 
68
volatile sig_atomic_t quit_now = 0;
 
69
int signal_received;
54
70
bool debug = false;
55
71
const char *argp_program_version = "password-prompt " VERSION;
56
 
const char *argp_program_bug_address = "<mandos@fukt.bsnet.se>";
57
 
 
58
 
static void termination_handler(__attribute__((unused))int signum){
59
 
  quit_now = true;
60
 
}
 
72
const char *argp_program_bug_address = "<mandos@recompile.se>";
 
73
 
 
74
/* Needed for conflict resolution */
 
75
const char plymouth_name[] = "plymouthd";
 
76
 
 
77
/* Function to use when printing errors */
 
78
__attribute__((format (gnu_printf, 3, 4)))
 
79
void error_plus(int status, int errnum, const char *formatstring,
 
80
                ...){
 
81
  va_list ap;
 
82
  char *text;
 
83
  int ret;
 
84
  
 
85
  va_start(ap, formatstring);
 
86
  ret = vasprintf(&text, formatstring, ap);
 
87
  if(ret == -1){
 
88
    fprintf(stderr, "Mandos plugin %s: ",
 
89
            program_invocation_short_name);
 
90
    vfprintf(stderr, formatstring, ap);
 
91
    fprintf(stderr, ": %s\n", strerror(errnum));
 
92
    error(status, errno, "vasprintf while printing error");
 
93
    return;
 
94
  }
 
95
  fprintf(stderr, "Mandos plugin ");
 
96
  error(status, errnum, "%s", text);
 
97
  free(text);
 
98
}
 
99
 
 
100
static void termination_handler(int signum){
 
101
  if(quit_now){
 
102
    return;
 
103
  }
 
104
  quit_now = 1;
 
105
  signal_received = signum;
 
106
}
 
107
 
 
108
bool conflict_detection(void){
 
109
 
 
110
  /* plymouth conflicts with password-prompt since both want to read
 
111
     from the terminal.  Password-prompt will exit if it detects
 
112
     plymouth since plymouth performs the same functionality.
 
113
   */
 
114
  if(access("/run/plymouth/pid", R_OK) == 0){
 
115
    return true;
 
116
  }
 
117
  
 
118
  __attribute__((nonnull))
 
119
  int is_plymouth(const struct dirent *proc_entry){
 
120
    int ret;
 
121
    int cl_fd;
 
122
    {
 
123
      uintmax_t proc_id;
 
124
      char *tmp;
 
125
      errno = 0;
 
126
      proc_id = strtoumax(proc_entry->d_name, &tmp, 10);
 
127
      
 
128
      if(errno != 0 or *tmp != '\0'
 
129
         or proc_id != (uintmax_t)((pid_t)proc_id)){
 
130
        return 0;
 
131
      }
 
132
    }
 
133
    
 
134
    char *cmdline_filename;
 
135
    ret = asprintf(&cmdline_filename, "/proc/%s/cmdline",
 
136
                   proc_entry->d_name);
 
137
    if(ret == -1){
 
138
      error_plus(0, errno, "asprintf");
 
139
      return 0;
 
140
    }
 
141
    
 
142
    /* Open /proc/<pid>/cmdline */
 
143
    cl_fd = open(cmdline_filename, O_RDONLY);
 
144
    free(cmdline_filename);
 
145
    if(cl_fd == -1){
 
146
      if(errno != ENOENT){
 
147
        error_plus(0, errno, "open");
 
148
      }
 
149
      return 0;
 
150
    }
 
151
    
 
152
    char *cmdline = NULL;
 
153
    {
 
154
      size_t cmdline_len = 0;
 
155
      size_t cmdline_allocated = 0;
 
156
      char *tmp;
 
157
      const size_t blocksize = 1024;
 
158
      ssize_t sret;
 
159
      do {
 
160
        /* Allocate more space? */
 
161
        if(cmdline_len + blocksize + 1 > cmdline_allocated){
 
162
          tmp = realloc(cmdline, cmdline_allocated + blocksize + 1);
 
163
          if(tmp == NULL){
 
164
            error_plus(0, errno, "realloc");
 
165
            free(cmdline);
 
166
            close(cl_fd);
 
167
            return 0;
 
168
          }
 
169
          cmdline = tmp;
 
170
          cmdline_allocated += blocksize;
 
171
        }
 
172
        
 
173
        /* Read data */
 
174
        sret = read(cl_fd, cmdline + cmdline_len,
 
175
                    cmdline_allocated - cmdline_len);
 
176
        if(sret == -1){
 
177
          error_plus(0, errno, "read");
 
178
          free(cmdline);
 
179
          close(cl_fd);
 
180
          return 0;
 
181
        }
 
182
        cmdline_len += (size_t)sret;
 
183
      } while(sret != 0);
 
184
      ret = close(cl_fd);
 
185
      if(ret == -1){
 
186
        error_plus(0, errno, "close");
 
187
        free(cmdline);
 
188
        return 0;
 
189
      }
 
190
      cmdline[cmdline_len] = '\0'; /* Make sure it is terminated */
 
191
    }
 
192
    /* we now have cmdline */
 
193
    
 
194
    /* get basename */
 
195
    char *cmdline_base = strrchr(cmdline, '/');
 
196
    if(cmdline_base != NULL){
 
197
      cmdline_base += 1;                /* skip the slash */
 
198
    } else {
 
199
      cmdline_base = cmdline;
 
200
    }
 
201
    
 
202
    if(strcmp(cmdline_base, plymouth_name) != 0){
 
203
      if(debug){
 
204
        fprintf(stderr, "\"%s\" is not \"%s\"\n", cmdline_base,
 
205
                plymouth_name);
 
206
      }
 
207
      free(cmdline);
 
208
      return 0;
 
209
    }
 
210
    if(debug){
 
211
      fprintf(stderr, "\"%s\" equals \"%s\"\n", cmdline_base,
 
212
              plymouth_name);
 
213
    }
 
214
    free(cmdline);
 
215
    return 1;
 
216
  }
 
217
  
 
218
  struct dirent **direntries = NULL;
 
219
  int ret;
 
220
  ret = scandir("/proc", &direntries, is_plymouth, alphasort);
 
221
  if(ret == -1){
 
222
    error_plus(1, errno, "scandir");
 
223
  }
 
224
  {
 
225
    int i = ret;
 
226
    while(i--){
 
227
      free(direntries[i]);
 
228
    }
 
229
  }
 
230
  free(direntries);
 
231
  return ret > 0;
 
232
}
 
233
 
61
234
 
62
235
int main(int argc, char **argv){
63
 
  ssize_t ret;
 
236
  ssize_t sret;
 
237
  int ret;
64
238
  size_t n;
65
239
  struct termios t_new, t_old;
66
240
  char *buffer = NULL;
67
241
  char *prefix = NULL;
 
242
  char *prompt = NULL;
68
243
  int status = EXIT_SUCCESS;
69
244
  struct sigaction old_action,
70
245
    new_action = { .sa_handler = termination_handler,
74
249
      { .name = "prefix", .key = 'p',
75
250
        .arg = "PREFIX", .flags = 0,
76
251
        .doc = "Prefix shown before the prompt", .group = 2 },
 
252
      { .name = "prompt", .key = 129,
 
253
        .arg = "PROMPT", .flags = 0,
 
254
        .doc = "The prompt to show", .group = 2 },
77
255
      { .name = "debug", .key = 128,
78
256
        .doc = "Debug mode", .group = 3 },
 
257
      /*
 
258
       * These reproduce what we would get without ARGP_NO_HELP
 
259
       */
 
260
      { .name = "help", .key = '?',
 
261
        .doc = "Give this help list", .group = -1 },
 
262
      { .name = "usage", .key = -3,
 
263
        .doc = "Give a short usage message", .group = -1 },
 
264
      { .name = "version", .key = 'V',
 
265
        .doc = "Print program version", .group = -1 },
79
266
      { .name = NULL }
80
267
    };
81
 
  
82
 
    error_t parse_opt (int key, char *arg, struct argp_state *state) {
83
 
      /* Get the INPUT argument from `argp_parse', which we know is a
84
 
         pointer to our plugin list pointer. */
85
 
      switch (key) {
86
 
      case 'p':
 
268
    
 
269
    __attribute__((nonnull(3)))
 
270
    error_t parse_opt (int key, char *arg, struct argp_state *state){
 
271
      errno = 0;
 
272
      switch (key){
 
273
      case 'p':                 /* --prefix */
87
274
        prefix = arg;
88
275
        break;
89
 
      case 128:
 
276
      case 128:                 /* --debug */
90
277
        debug = true;
91
278
        break;
92
 
      case ARGP_KEY_ARG:
93
 
        argp_usage (state);
 
279
      case 129:                 /* --prompt */
 
280
        prompt = arg;
94
281
        break;
95
 
      case ARGP_KEY_END:
 
282
        /*
 
283
         * These reproduce what we would get without ARGP_NO_HELP
 
284
         */
 
285
      case '?':                 /* --help */
 
286
        argp_state_help(state, state->out_stream,
 
287
                        (ARGP_HELP_STD_HELP | ARGP_HELP_EXIT_ERR)
 
288
                        & ~(unsigned int)ARGP_HELP_EXIT_OK);
 
289
        __builtin_unreachable();
 
290
      case -3:                  /* --usage */
 
291
        argp_state_help(state, state->out_stream,
 
292
                        ARGP_HELP_USAGE | ARGP_HELP_EXIT_ERR);
 
293
        __builtin_unreachable();
 
294
      case 'V':                 /* --version */
 
295
        fprintf(state->out_stream, "%s\n", argp_program_version);
 
296
        exit(argp_err_exit_status);
96
297
        break;
97
298
      default:
98
299
        return ARGP_ERR_UNKNOWN;
99
300
      }
100
 
      return 0;
 
301
      return errno;
101
302
    }
102
 
  
 
303
    
103
304
    struct argp argp = { .options = options, .parser = parse_opt,
104
305
                         .args_doc = "",
105
306
                         .doc = "Mandos password-prompt -- Read and"
106
307
                         " output a password" };
107
 
    ret = argp_parse (&argp, argc, argv, 0, 0, NULL);
108
 
    if (ret == ARGP_ERR_UNKNOWN){
109
 
      fprintf(stderr, "Unknown error while parsing arguments\n");
110
 
      return EXIT_FAILURE;
 
308
    ret = argp_parse(&argp, argc, argv,
 
309
                     ARGP_IN_ORDER | ARGP_NO_HELP, NULL, NULL);
 
310
    switch(ret){
 
311
    case 0:
 
312
      break;
 
313
    case ENOMEM:
 
314
    default:
 
315
      errno = ret;
 
316
      error_plus(0, errno, "argp_parse");
 
317
      return EX_OSERR;
 
318
    case EINVAL:
 
319
      return EX_USAGE;
111
320
    }
112
321
  }
113
 
    
114
 
  if (debug){
 
322
  
 
323
  if(debug){
115
324
    fprintf(stderr, "Starting %s\n", argv[0]);
116
325
  }
117
 
  if (debug){
 
326
 
 
327
  if(conflict_detection()){
 
328
    if(debug){
 
329
      fprintf(stderr, "Stopping %s because of conflict\n", argv[0]);
 
330
    }
 
331
    return EXIT_FAILURE;
 
332
  }
 
333
  
 
334
  if(debug){
118
335
    fprintf(stderr, "Storing current terminal attributes\n");
119
336
  }
120
337
  
121
 
  if (tcgetattr(STDIN_FILENO, &t_old) != 0){
122
 
    perror("tcgetattr");
123
 
    return EXIT_FAILURE;
 
338
  if(tcgetattr(STDIN_FILENO, &t_old) != 0){
 
339
    int e = errno;
 
340
    error_plus(0, errno, "tcgetattr");
 
341
    switch(e){
 
342
    case EBADF:
 
343
    case ENOTTY:
 
344
      return EX_UNAVAILABLE;
 
345
    default:
 
346
      return EX_OSERR;
 
347
    }
124
348
  }
125
349
  
126
350
  sigemptyset(&new_action.sa_mask);
127
 
  sigaddset(&new_action.sa_mask, SIGINT);
128
 
  sigaddset(&new_action.sa_mask, SIGHUP);
129
 
  sigaddset(&new_action.sa_mask, SIGTERM);
 
351
  ret = sigaddset(&new_action.sa_mask, SIGINT);
 
352
  if(ret == -1){
 
353
    error_plus(0, errno, "sigaddset");
 
354
    return EX_OSERR;
 
355
  }
 
356
  ret = sigaddset(&new_action.sa_mask, SIGHUP);
 
357
  if(ret == -1){
 
358
    error_plus(0, errno, "sigaddset");
 
359
    return EX_OSERR;
 
360
  }
 
361
  ret = sigaddset(&new_action.sa_mask, SIGTERM);
 
362
  if(ret == -1){
 
363
    error_plus(0, errno, "sigaddset");
 
364
    return EX_OSERR;
 
365
  }
 
366
  /* Need to check if the handler is SIG_IGN before handling:
 
367
     | [[info:libc:Initial Signal Actions]] |
 
368
     | [[info:libc:Basic Signal Handling]]  |
 
369
  */
130
370
  ret = sigaction(SIGINT, NULL, &old_action);
131
371
  if(ret == -1){
132
 
    perror("sigaction");
133
 
    return EXIT_FAILURE;
 
372
    error_plus(0, errno, "sigaction");
 
373
    return EX_OSERR;
134
374
  }
135
 
  if (old_action.sa_handler != SIG_IGN){
 
375
  if(old_action.sa_handler != SIG_IGN){
136
376
    ret = sigaction(SIGINT, &new_action, NULL);
137
377
    if(ret == -1){
138
 
      perror("sigaction");
139
 
      return EXIT_FAILURE;
 
378
      error_plus(0, errno, "sigaction");
 
379
      return EX_OSERR;
140
380
    }
141
381
  }
142
382
  ret = sigaction(SIGHUP, NULL, &old_action);
143
383
  if(ret == -1){
144
 
    perror("sigaction");
145
 
    return EXIT_FAILURE;
 
384
    error_plus(0, errno, "sigaction");
 
385
    return EX_OSERR;
146
386
  }
147
 
  if (old_action.sa_handler != SIG_IGN){
 
387
  if(old_action.sa_handler != SIG_IGN){
148
388
    ret = sigaction(SIGHUP, &new_action, NULL);
149
389
    if(ret == -1){
150
 
      perror("sigaction");
151
 
      return EXIT_FAILURE;
 
390
      error_plus(0, errno, "sigaction");
 
391
      return EX_OSERR;
152
392
    }
153
393
  }
154
394
  ret = sigaction(SIGTERM, NULL, &old_action);
155
395
  if(ret == -1){
156
 
    perror("sigaction");
157
 
    return EXIT_FAILURE;
 
396
    error_plus(0, errno, "sigaction");
 
397
    return EX_OSERR;
158
398
  }
159
 
  if (old_action.sa_handler != SIG_IGN){
 
399
  if(old_action.sa_handler != SIG_IGN){
160
400
    ret = sigaction(SIGTERM, &new_action, NULL);
161
401
    if(ret == -1){
162
 
      perror("sigaction");
163
 
      return EXIT_FAILURE;
 
402
      error_plus(0, errno, "sigaction");
 
403
      return EX_OSERR;
164
404
    }
165
405
  }
166
406
  
167
407
  
168
 
  if (debug){
 
408
  if(debug){
169
409
    fprintf(stderr, "Removing echo flag from terminal attributes\n");
170
410
  }
171
411
  
172
412
  t_new = t_old;
173
 
  t_new.c_lflag &= ~ECHO;
174
 
  if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &t_new) != 0){
175
 
    perror("tcsetattr-echo");
176
 
    return EXIT_FAILURE;
 
413
  t_new.c_lflag &= ~(tcflag_t)ECHO;
 
414
  if(tcsetattr(STDIN_FILENO, TCSAFLUSH, &t_new) != 0){
 
415
    int e = errno;
 
416
    error_plus(0, errno, "tcsetattr-echo");
 
417
    switch(e){
 
418
    case EBADF:
 
419
    case ENOTTY:
 
420
      return EX_UNAVAILABLE;
 
421
    case EINVAL:
 
422
    default:
 
423
      return EX_OSERR;
 
424
    }
177
425
  }
178
 
 
179
 
  if (debug){
 
426
  
 
427
  if(debug){
180
428
    fprintf(stderr, "Waiting for input from stdin \n");
181
429
  }
182
430
  while(true){
183
 
    if (quit_now){
 
431
    if(quit_now){
184
432
      if(debug){
185
433
        fprintf(stderr, "Interrupted by signal, exiting.\n");
186
434
      }
191
439
    if(prefix){
192
440
      fprintf(stderr, "%s ", prefix);
193
441
    }
194
 
    {
195
 
      const char *cryptsource = getenv("cryptsource");
196
 
      const char *crypttarget = getenv("crypttarget");
197
 
      const char *const prompt
198
 
        = "Enter passphrase to unlock the disk";
 
442
    if(prompt != NULL){
 
443
      fprintf(stderr, "%s: ", prompt);
 
444
    } else {
 
445
      const char *cryptsource = getenv("CRYPTTAB_SOURCE");
 
446
      const char *crypttarget = getenv("CRYPTTAB_NAME");
 
447
      /* Before cryptsetup 1.1.0~rc2 */
 
448
      if(cryptsource == NULL){
 
449
        cryptsource = getenv("cryptsource");
 
450
      }
 
451
      if(crypttarget == NULL){
 
452
        crypttarget = getenv("crypttarget");
 
453
      }
 
454
      const char *const prompt1 = "Unlocking the disk";
 
455
      const char *const prompt2 = "Enter passphrase";
199
456
      if(cryptsource == NULL){
200
457
        if(crypttarget == NULL){
201
 
          fprintf(stderr, "%s: ", prompt);
 
458
          fprintf(stderr, "%s to unlock the disk: ", prompt2);
202
459
        } else {
203
 
          fprintf(stderr, "%s (%s): ", prompt, crypttarget);
 
460
          fprintf(stderr, "%s (%s)\n%s: ", prompt1, crypttarget,
 
461
                  prompt2);
204
462
        }
205
463
      } else {
206
464
        if(crypttarget == NULL){
207
 
          fprintf(stderr, "%s %s: ", prompt, cryptsource);
 
465
          fprintf(stderr, "%s %s\n%s: ", prompt1, cryptsource,
 
466
                  prompt2);
208
467
        } else {
209
 
          fprintf(stderr, "%s %s (%s): ", prompt, cryptsource,
210
 
                  crypttarget);
 
468
          fprintf(stderr, "%s %s (%s)\n%s: ", prompt1, cryptsource,
 
469
                  crypttarget, prompt2);
211
470
        }
212
471
      }
213
472
    }
214
 
    ret = getline(&buffer, &n, stdin);
215
 
    if (ret > 0){
 
473
    sret = getline(&buffer, &n, stdin);
 
474
    if(sret > 0){
216
475
      status = EXIT_SUCCESS;
217
476
      /* Make n = data size instead of allocated buffer size */
218
 
      n = (size_t)ret;
 
477
      n = (size_t)sret;
219
478
      /* Strip final newline */
220
 
      if(n>0 and buffer[n-1] == '\n'){
 
479
      if(n > 0 and buffer[n-1] == '\n'){
221
480
        buffer[n-1] = '\0';     /* not strictly necessary */
222
481
        n--;
223
482
      }
224
483
      size_t written = 0;
225
484
      while(written < n){
226
 
        ret = write(STDOUT_FILENO, buffer + written, n - written);
227
 
        if(ret < 0){
228
 
          perror("write");
229
 
          status = EXIT_FAILURE;
230
 
          break;
231
 
        }
232
 
        written += (size_t)ret;
 
485
        sret = write(STDOUT_FILENO, buffer + written, n - written);
 
486
        if(sret < 0){
 
487
          int e = errno;
 
488
          error_plus(0, errno, "write");
 
489
          switch(e){
 
490
          case EBADF:
 
491
          case EFAULT:
 
492
          case EINVAL:
 
493
          case EFBIG:
 
494
          case EIO:
 
495
          case ENOSPC:
 
496
          default:
 
497
            status = EX_IOERR;
 
498
            break;
 
499
          case EINTR:
 
500
            status = EXIT_FAILURE;
 
501
            break;
 
502
          }
 
503
          break;
 
504
        }
 
505
        written += (size_t)sret;
 
506
      }
 
507
      sret = close(STDOUT_FILENO);
 
508
      if(sret == -1){
 
509
        int e = errno;
 
510
        error_plus(0, errno, "close");
 
511
        switch(e){
 
512
        case EBADF:
 
513
          status = EX_OSFILE;
 
514
          break;
 
515
        case EIO:
 
516
        default:
 
517
          status = EX_IOERR;
 
518
          break;
 
519
        }
233
520
      }
234
521
      break;
235
522
    }
236
 
    if (ret < 0){
237
 
      if (errno != EINTR and not feof(stdin)){
238
 
        perror("getline");
239
 
        status = EXIT_FAILURE;
240
 
        break;
 
523
    if(sret < 0){
 
524
      int e = errno;
 
525
      if(errno != EINTR){
 
526
        if(not feof(stdin)){
 
527
          error_plus(0, errno, "getline");
 
528
          switch(e){
 
529
          case EBADF:
 
530
            status = EX_UNAVAILABLE;
 
531
            break;
 
532
          case EIO:
 
533
          case EINVAL:
 
534
          default:
 
535
            status = EX_IOERR;
 
536
            break;
 
537
          }
 
538
          break;
 
539
        } else {
 
540
          clearerr(stdin);
 
541
        }
241
542
      }
242
543
    }
243
 
    /* if(ret == 0), then the only sensible thing to do is to retry to
244
 
       read from stdin */
 
544
    /* if(sret == 0), then the only sensible thing to do is to retry
 
545
       to read from stdin */
245
546
    fputc('\n', stderr);
246
547
    if(debug and not quit_now){
247
 
      /* If quit_now is true, we were interrupted by a signal, and
 
548
      /* If quit_now is nonzero, we were interrupted by a signal, and
248
549
         will print that later, so no need to show this too. */
249
550
      fprintf(stderr, "getline() returned 0, retrying.\n");
250
551
    }
251
552
  }
252
 
 
 
553
  
253
554
  free(buffer);
254
555
  
255
 
  if (debug){
 
556
  if(debug){
256
557
    fprintf(stderr, "Restoring terminal attributes\n");
257
558
  }
258
 
  if (tcsetattr(STDIN_FILENO, TCSAFLUSH, &t_old) != 0){
259
 
    perror("tcsetattr+echo");
260
 
  }
261
 
  
262
 
  if (debug){
 
559
  if(tcsetattr(STDIN_FILENO, TCSAFLUSH, &t_old) != 0){
 
560
    error_plus(0, errno, "tcsetattr+echo");
 
561
  }
 
562
  
 
563
  if(quit_now){
 
564
    sigemptyset(&old_action.sa_mask);
 
565
    old_action.sa_handler = SIG_DFL;
 
566
    ret = sigaction(signal_received, &old_action, NULL);
 
567
    if(ret == -1){
 
568
      error_plus(0, errno, "sigaction");
 
569
    }
 
570
    raise(signal_received);
 
571
  }
 
572
  
 
573
  if(debug){
263
574
    fprintf(stderr, "%s is exiting with status %d\n", argv[0],
264
575
            status);
265
576
  }
266
 
  if(status == EXIT_SUCCESS){
 
577
  if(status == EXIT_SUCCESS or status == EX_OK){
267
578
    fputc('\n', stderr);
268
579
  }
269
580