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

First version of a somewhat complete D-Bus server interface.  Also
change user/group name to "_mandos".

* debian/mandos.postinst: Rename old "mandos" user and group to
                          "_mandos"; create "_mandos" user and group
                          if none exist.
* debian/mandos-client.postinst: - '' -

* initramfs-tools-hook: Try "_mandos" before "mandos" as user and
                        group name.

* mandos (_datetime_to_dbus_struct): New; was previously local.
  (Client.started): Renamed to "last_started".  All users changed.
  (Client.started): New; boolean.
  (Client.dbus_object_path): New.
  (Client.check_command): Renamed to "checker_command".  All users
                          changed.
  (Client.__init__): Set and use "self.dbus_object_path".  Set
                     "self.started".
  (Client.start): Update "self.started".  Emit "self.PropertyChanged"
                  signals for both "started" and "last_started".
  (Client.stop): Update "self.started".  Emit "self.PropertyChanged"
                 signal for "started".
  (Client.checker_callback): Take additional "command" argument.  All
                             callers changed. Emit
                             "self.PropertyChanged" signal.
  (Client.bump_timeout): Emit "self.PropertyChanged" signal for
                         "last_checked_ok".
  (Client.start_checker): Emit "self.PropertyChanged" signal for
                          "checker_running".
  (Client.stop_checker): Emit "self.PropertyChanged" signal for
                         "checker_running".
  (Client.still_valid): Bug fix: use "getattr(self, started, False)"
                        instead of "self.started" in case this client
                        object is so new that the "started" attribute
                        has not been created yet.
  (Client.IntervalChanged, Client.CheckerIsRunning, Client.GetChecker,
  Client.GetCreated, Client.GetFingerprint, Client.GetHost,
  Client.GetInterval, Client.GetName, Client.GetStarted,
  Client.GetTimeout, Client.StateChanged, Client.TimeoutChanged):
  Removed; all callers changed.
  (Client.CheckerCompleted): Add "condition" and "command" arguments.
                             All callers changed.
  (Client.GetAllProperties, Client.PropertyChanged): New.
  (Client.StillValid): Renamed to "IsStillValid".
  (Client.StartChecker): Changed to its own function to avoid the
                         return value from "Client.start_checker()".
  (Client.Stop): Changed to its own function to avoid the return value
                 from "Client.stop()".
  (main): Try "_mandos" before "mandos" as user and group name.
          Removed inner function "remove_from_clients".  New inner
          class "MandosServer".

Show diffs side-by-side

added added

removed removed

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