/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/askpass-fifo.c

  • Committer: Teddy Hogeborn
  • Date: 2009-09-17 04:16:32 UTC
  • Revision ID: teddy@fukt.bsnet.se-20090917041632-7m1rm83pxguw5zw7
* plugins.d/usplash.c (main): Bug fix: only close FIFO if opened.

Show diffs side-by-side

added added

removed removed

Lines of Context:
2
2
/*
3
3
 * Askpass-FIFO - Read a password from a FIFO and output it
4
4
 * 
5
 
 * Copyright © 2008-2011 Teddy Hogeborn
6
 
 * Copyright © 2008-2011 Björn Påhlsson
 
5
 * Copyright © 2008,2009 Teddy Hogeborn
 
6
 * Copyright © 2008,2009 Björn Påhlsson
7
7
 * 
8
8
 * This program is free software: you can redistribute it and/or
9
9
 * modify it under the terms of the GNU General Public License as
19
19
 * along with this program.  If not, see
20
20
 * <http://www.gnu.org/licenses/>.
21
21
 * 
22
 
 * Contact the authors at <mandos@recompile.se>.
 
22
 * Contact the authors at <mandos@fukt.bsnet.se>.
23
23
 */
24
24
 
25
25
#define _GNU_SOURCE             /* TEMP_FAILURE_RETRY() */
26
26
#include <sys/types.h>          /* ssize_t */
27
27
#include <sys/stat.h>           /* mkfifo(), S_IRUSR, S_IWUSR */
28
28
#include <iso646.h>             /* and */
29
 
#include <errno.h>              /* errno, EACCES, ENOTDIR, ELOOP,
30
 
                                   ENAMETOOLONG, ENOSPC, EROFS,
31
 
                                   ENOENT, EEXIST, EFAULT, EMFILE,
32
 
                                   ENFILE, ENOMEM, EBADF, EINVAL, EIO,
33
 
                                   EISDIR, EFBIG */
34
 
#include <error.h>              /* error() */
35
 
#include <stdio.h>              /* fprintf(), vfprintf(),
36
 
                                   vasprintf() */
 
29
#include <errno.h>              /* errno, EEXIST */
 
30
#include <stdio.h>              /* perror() */
37
31
#include <stdlib.h>             /* EXIT_FAILURE, NULL, size_t, free(),
38
32
                                   realloc(), EXIT_SUCCESS */
39
33
#include <fcntl.h>              /* open(), O_RDONLY */
40
34
#include <unistd.h>             /* read(), close(), write(),
41
35
                                   STDOUT_FILENO */
42
 
#include <sysexits.h>           /* EX_OSERR, EX_OSFILE,
43
 
                                   EX_UNAVAILABLE, EX_IOERR */
44
 
#include <string.h>             /* strerror() */
45
 
#include <stdarg.h>             /* va_list, va_start(), ... */
46
 
 
47
 
 
48
 
/* Function to use when printing errors */
49
 
void error_plus(int status, int errnum, const char *formatstring,
50
 
                ...){
51
 
  va_list ap;
52
 
  char *text;
53
 
  int ret;
54
 
  
55
 
  va_start(ap, formatstring);
56
 
  ret = vasprintf(&text, formatstring, ap);
57
 
  if (ret == -1){
58
 
    fprintf(stderr, "Mandos plugin %s: ",
59
 
            program_invocation_short_name);
60
 
    vfprintf(stderr, formatstring, ap);
61
 
    fprintf(stderr, ": ");
62
 
    fprintf(stderr, "%s\n", strerror(errnum));
63
 
    error(status, errno, "vasprintf while printing error");
64
 
    return;
65
 
  }
66
 
  fprintf(stderr, "Mandos plugin ");
67
 
  error(status, errnum, "%s", text);
68
 
  free(text);
69
 
}
 
36
 
70
37
 
71
38
int main(__attribute__((unused))int argc,
72
39
         __attribute__((unused))char **argv){
76
43
  /* Create FIFO */
77
44
  const char passfifo[] = "/lib/cryptsetup/passfifo";
78
45
  ret = mkfifo(passfifo, S_IRUSR | S_IWUSR);
79
 
  if(ret == -1){
80
 
    int e = errno;
81
 
    switch(e){
82
 
    case EACCES:
83
 
    case ENOTDIR:
84
 
    case ELOOP:
85
 
      error_plus(EX_OSFILE, errno, "mkfifo");
86
 
    case ENAMETOOLONG:
87
 
    case ENOSPC:
88
 
    case EROFS:
89
 
    default:
90
 
      error_plus(EX_OSERR, errno, "mkfifo");
91
 
    case ENOENT:
92
 
      /* no "/lib/cryptsetup"? */
93
 
      error_plus(EX_UNAVAILABLE, errno, "mkfifo");
94
 
    case EEXIST:
95
 
      break;                    /* not an error */
96
 
    }
 
46
  if(ret == -1 and errno != EEXIST){
 
47
    perror("mkfifo");
 
48
    return EXIT_FAILURE;
97
49
  }
98
50
  
99
51
  /* Open FIFO */
100
52
  int fifo_fd = open(passfifo, O_RDONLY);
101
53
  if(fifo_fd == -1){
102
 
    int e = errno;
103
 
    error_plus(0, errno, "open");
104
 
    switch(e){
105
 
    case EACCES:
106
 
    case ENOENT:
107
 
    case EFAULT:
108
 
      return EX_UNAVAILABLE;
109
 
    case ENAMETOOLONG:
110
 
    case EMFILE:
111
 
    case ENFILE:
112
 
    case ENOMEM:
113
 
    default:
114
 
      return EX_OSERR;
115
 
    case ENOTDIR:
116
 
    case ELOOP:
117
 
      return EX_OSFILE;
118
 
    }
 
54
    perror("open");
 
55
    return EXIT_FAILURE;
119
56
  }
120
57
  
121
58
  /* Read from FIFO */
128
65
      if(buf_len + blocksize > buf_allocated){
129
66
        char *tmp = realloc(buf, buf_allocated + blocksize);
130
67
        if(tmp == NULL){
131
 
          error_plus(0, errno, "realloc");
 
68
          perror("realloc");
132
69
          free(buf);
133
 
          return EX_OSERR;
 
70
          return EXIT_FAILURE;
134
71
        }
135
72
        buf = tmp;
136
73
        buf_allocated += blocksize;
137
74
      }
138
75
      sret = read(fifo_fd, buf + buf_len, buf_allocated - buf_len);
139
76
      if(sret == -1){
140
 
        int e = errno;
 
77
        perror("read");
141
78
        free(buf);
142
 
        errno = e;
143
 
        error_plus(0, errno, "read");
144
 
        switch(e){
145
 
        case EBADF:
146
 
        case EFAULT:
147
 
        case EINVAL:
148
 
        default:
149
 
          return EX_OSERR;
150
 
        case EIO:
151
 
          return EX_IOERR;
152
 
        case EISDIR:
153
 
          return EX_UNAVAILABLE;
154
 
        }
 
79
        return EXIT_FAILURE;
155
80
      }
156
81
      buf_len += (size_t)sret;
157
82
    } while(sret != 0);
165
90
  while(written < buf_len){
166
91
    sret = write(STDOUT_FILENO, buf + written, buf_len - written);
167
92
    if(sret == -1){
168
 
      int e = errno;
 
93
      perror("write");
169
94
      free(buf);
170
 
      errno = e;
171
 
      error_plus(0, errno, "write");
172
 
      switch(e){
173
 
      case EBADF:
174
 
      case EFAULT:
175
 
      case EINVAL:
176
 
        return EX_OSFILE;
177
 
      case EFBIG:
178
 
      case EIO:
179
 
      case ENOSPC:
180
 
      default:
181
 
        return EX_IOERR;
182
 
      }
 
95
      return EXIT_FAILURE;
183
96
    }
184
97
    written += (size_t)sret;
185
98
  }
186
99
  free(buf);
187
100
  
188
 
  ret = close(STDOUT_FILENO);
189
 
  if(ret == -1){
190
 
    int e = errno;
191
 
    error_plus(0, errno, "close");
192
 
    switch(e){
193
 
    case EBADF:
194
 
      return EX_OSFILE;
195
 
    case EIO:
196
 
    default:
197
 
      return EX_IOERR;
198
 
    }
199
 
  }
200
101
  return EXIT_SUCCESS;
201
102
}