/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 Hogeborn
  • Date: 2009-10-18 17:29:55 UTC
  • Revision ID: teddy@fukt.bsnet.se-20091018172955-qqnsfad2lkpkt35f
* plugins.d/password-prompt.c: Use exit codes from <sysexits.h>.  Do
                               close(STDOUT_FILENO) after writing to
                               check its return code.

Show diffs side-by-side

added added

removed removed

Lines of Context:
plugins.d/password-prompt.c
37
37
#include <stddef.h>             /* NULL, size_t, ssize_t */
38
38
#include <sys/types.h>          /* ssize_t */
39
39
#include <stdlib.h>             /* EXIT_SUCCESS, EXIT_FAILURE,
40
 
                                   getenv() */
 
40
                                   getopt_long, getenv() */
41
41
#include <stdio.h>              /* fprintf(), stderr, getline(),
42
 
                                   stdin, feof(), perror(), fputc()
43
 
                                */
 
42
                                   stdin, feof(), perror(), fputc(),
 
43
                                   getopt_long */
44
44
#include <errno.h>              /* errno, EBADF, ENOTTY, EINVAL,
45
45
                                   EFAULT, EFBIG, EIO, ENOSPC, EINTR
46
46
                                */
86
86
        .doc = "Prefix shown before the prompt", .group = 2 },
87
87
      { .name = "debug", .key = 128,
88
88
        .doc = "Debug mode", .group = 3 },
89
 
      /*
90
 
       * These reproduce what we would get without ARGP_NO_HELP
91
 
       */
92
 
      { .name = "help", .key = '?',
93
 
        .doc = "Give this help list", .group = -1 },
94
 
      { .name = "usage", .key = -3,
95
 
        .doc = "Give a short usage message", .group = -1 },
96
 
      { .name = "version", .key = 'V',
97
 
        .doc = "Print program version", .group = -1 },
98
89
      { .name = NULL }
99
90
    };
100
91
    
101
92
    error_t parse_opt (int key, char *arg, struct argp_state *state){
102
 
      errno = 0;
103
93
      switch (key){
104
94
      case 'p':
105
95
        prefix = arg;
107
97
      case 128:
108
98
        debug = true;
109
99
        break;
110
 
        /*
111
 
         * These reproduce what we would get without ARGP_NO_HELP
112
 
         */
113
 
      case '?':                 /* --help */
114
 
        argp_state_help(state, state->out_stream,
115
 
                        (ARGP_HELP_STD_HELP | ARGP_HELP_EXIT_ERR)
116
 
                        & ~(unsigned int)ARGP_HELP_EXIT_OK);
117
 
      case -3:                  /* --usage */
118
 
        argp_state_help(state, state->out_stream,
119
 
                        ARGP_HELP_USAGE | ARGP_HELP_EXIT_ERR);
120
 
      case 'V':                 /* --version */
121
 
        fprintf(state->out_stream, "%s\n", argp_program_version);
122
 
        exit(argp_err_exit_status);
 
100
      case ARGP_KEY_ARG:
 
101
        argp_usage(state);
 
102
        break;
 
103
      case ARGP_KEY_END:
123
104
        break;
124
105
      default:
125
106
        return ARGP_ERR_UNKNOWN;
126
107
      }
127
 
      return errno;
 
108
      return 0;
128
109
    }
129
110
    
130
111
    struct argp argp = { .options = options, .parser = parse_opt,
131
112
                         .args_doc = "",
132
113
                         .doc = "Mandos password-prompt -- Read and"
133
114
                         " output a password" };
134
 
    ret = argp_parse(&argp, argc, argv,
135
 
                     ARGP_IN_ORDER | ARGP_NO_HELP, NULL, NULL);
136
 
    switch(ret){
137
 
    case 0:
138
 
      break;
139
 
    case ENOMEM:
140
 
    default:
141
 
      errno = ret;
142
 
      perror("argp_parse");
143
 
      return EX_OSERR;
144
 
    case EINVAL:
145
 
      return EX_USAGE;
 
115
    ret = argp_parse(&argp, argc, argv, 0, 0, NULL);
 
116
    if(ret == ARGP_ERR_UNKNOWN){
 
117
      fprintf(stderr, "Unknown error while parsing arguments\n");
 
118
      return EX_SOFTWARE;
146
119
    }
147
120
  }
148
121
  
258
231
      fprintf(stderr, "%s ", prefix);
259
232
    }
260
233
    {
261
 
      const char *cryptsource = getenv("CRYPTTAB_SOURCE");
262
 
      const char *crypttarget = getenv("CRYPTTAB_NAME");
263
 
      /* Before cryptsetup 1.1.0~rc2 */
264
 
      if(cryptsource == NULL){
265
 
        cryptsource = getenv("cryptsource");
266
 
      }
267
 
      if(crypttarget == NULL){
268
 
        crypttarget = getenv("crypttarget");
269
 
      }
270
 
      const char *const prompt1 = "Unlocking the disk";
271
 
      const char *const prompt2 = "Enter passphrase";
 
234
      const char *cryptsource = getenv("cryptsource");
 
235
      const char *crypttarget = getenv("crypttarget");
 
236
      const char *const prompt
 
237
        = "Enter passphrase to unlock the disk";
272
238
      if(cryptsource == NULL){
273
239
        if(crypttarget == NULL){
274
 
          fprintf(stderr, "%s to unlock the disk: ", prompt2);
 
240
          fprintf(stderr, "%s: ", prompt);
275
241
        } else {
276
 
          fprintf(stderr, "%s (%s)\n%s: ", prompt1, crypttarget,
277
 
                  prompt2);
 
242
          fprintf(stderr, "%s (%s): ", prompt, crypttarget);
278
243
        }
279
244
      } else {
280
245
        if(crypttarget == NULL){
281
 
          fprintf(stderr, "%s %s\n%s: ", prompt1, cryptsource,
282
 
                  prompt2);
 
246
          fprintf(stderr, "%s %s: ", prompt, cryptsource);
283
247
        } else {
284
 
          fprintf(stderr, "%s %s (%s)\n%s: ", prompt1, cryptsource,
285
 
                  crypttarget, prompt2);
 
248
          fprintf(stderr, "%s %s (%s): ", prompt, cryptsource,
 
249
                  crypttarget);
286
250
        }
287
251
      }
288
252
    }