pwmc.c

   1 /* vim:tw=78:ts=8:sw=4:set ft=c:  */
   2 /*
   3     Copyright (C) 2007-2008 Ben Kibbey <bjk@luxsci.net>
   4
   5     This program is free software; you can redistribute it and/or modify
   6     it under the terms of the GNU General Public License as published by
   7     the Free Software Foundation; either version 2 of the License, or
   8     (at your option) any later version.
   9
  10     This program is distributed in the hope that it will be useful,
  11     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13     GNU General Public License for more details.
  14
  15     You should have received a copy of the GNU General Public License
  16     along with this program; if not, write to the Free Software
  17     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18 */
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <err.h>
  23 #include <errno.h>
  24 #include <string.h>
  25 #include <libpwmd.h>
  26 #include <assuan.h>
  27
  28 #ifdef HAVE_CONFIG_H
  29 #include <config.h>
  30 #endif
  31
  32 #ifdef HAVE_LOCALE_H
  33 #include <locale.h>
  34 #endif
  35
  36 #include "gettext.h"
  37 #define N_(msgid)       gettext(msgid)
  38
  39 #ifndef MEM_DEBUG
  40 #include "mem.h"
  41 #else
  42 #define xfree           free
  43 #define xrealloc        realloc
  44 #define xmalloc         malloc
  45 #define xstrdup         strdup
  46 #define xcalloc         calloc
  47 #endif
  48
  49 pwm_t *pwm;
  50
  51 static void show_error(gpg_error_t error)
  52 {
  53     fprintf(stderr, "ERR %i %s\n", gpg_err_code(error), pwmd_strerror(error));
  54 }
  55
  56 static void usage(const char *pn)
  57 {
  58     fprintf(stderr, N_(
  59             "Reads PWMD protocol commands from standard input.\n\n"
  60             "Usage: pwmc [-hvX] [-s <socket>] [-P -T -N -D | -p <password>] [-S] \n"
  61             "          [-c <name>] [-d <fd>] [-I <fd>] [filename]\n"
  62             "   -X  disable showing of status messages from the server\n"
  63             "   -c  set the client name\n"
  64             "   -s  socket path (~/.pwmd/socket)\n"
  65             "   -p  password\n"
  66             "   -P  path to the pinentry binary (server default)\n"
  67             "   -T  pinentry tty\n"
  68             "   -N  pinentry terminal type\n"
  69             "   -D  pinentry display\n"
  70             "   -d  redirect command output to the specified file descriptor\n"
  71             "   -I  read inquire data from the specified file descriptor\n"
  72             "   -S  send the SAVE command before exiting\n"
  73             "   -v  version\n"
  74             "   -h  this help text\n"));
  75     exit(EXIT_FAILURE);
  76 }
  77
  78 struct inquire_s {
  79     FILE *fp;
  80     char *data;
  81 };
  82
  83 static gpg_error_t do_inquire(void *data, const char *keyword, gpg_error_t rc,
  84         char **result, size_t *result_len)
  85 {
  86     int c;
  87     static char buf[ASSUAN_LINELENGTH];
  88     char *p;
  89     size_t len = 0;
  90     struct inquire_s *inq = (struct inquire_s *)data;
  91
  92     if (rc) {
  93         memset(buf, 0, sizeof(buf));
  94         return rc;
  95     }
  96
  97     buf[0] = 0;
  98     p = buf;
  99
 100     if (inq->data) {
 101         snprintf(buf, sizeof(buf), "%s", inq->data);
 102         xfree(inq->data);
 103         inq->data = NULL;
 104         len = strlen(buf);
 105         p = buf + len;
 106     }
 107
 108     while ((c = fgetc(inq->fp)) != EOF) {
 109         if (len == sizeof(buf)) {
 110             ungetc(c, inq->fp);
 111             break;
 112         }
 113
 114         *p++ = c;
 115         len++;
 116     }
 117
 118     if (!buf[0]) {
 119         memset(buf, 0, sizeof(buf));
 120         return GPG_ERR_EOF;
 121     }
 122
 123     *result = buf;
 124     *result_len = len;
 125     return 0;
 126 }
 127
 128 static int status_msg_cb(void *data, const char *line)
 129 {
 130     fprintf(stderr, "%s\n", line);
 131     return 0;
 132 }
 133
 134 int main(int argc, char *argv[])
 135 {
 136     int opt;
 137     char *password = NULL;
 138     char *filename = NULL;
 139     char *socketpath = NULL;
 140     char command[ASSUAN_LINELENGTH], *p;
 141     int ret = EXIT_SUCCESS;
 142     gpg_error_t error;
 143     char *result = NULL;
 144     int save = 0;
 145     char *pinentry_path = NULL;
 146     char *display = NULL, *tty = NULL, *ttytype = NULL;
 147     int outfd = STDOUT_FILENO;
 148     FILE *outfp = stdout;
 149     int inquirefd = STDIN_FILENO;
 150     FILE *inquirefp = stdin;
 151     int show_status = 1;
 152     char *clientname = NULL;
 153
 154     setlocale(LC_ALL, "");
 155     bindtextdomain("libpwmd", LOCALEDIR);
 156
 157     while ((opt = getopt(argc, argv, "c:I:XT:N:D:hvP:p:s:Sd:")) != EOF) {
 158         switch (opt) {
 159             case 'c':
 160                 clientname = xstrdup(optarg);
 161                 break;
 162             case 'X':
 163                 show_status = 0;
 164                 break;
 165             case 'T':
 166                 tty = optarg;
 167                 break;
 168             case 'N':
 169                 ttytype = optarg;
 170                 break;
 171             case 'D':
 172                 display = optarg;
 173                 break;
 174             case 'I':
 175                 inquirefd = atoi(optarg);
 176                 inquirefp = fdopen(inquirefd, "r");
 177
 178                 if (!inquirefp) {
 179                     xfree(password);
 180                     err(EXIT_FAILURE, "%i", inquirefd);
 181                 }
 182                 break;
 183             case 'd':
 184                 outfd = atoi(optarg);
 185                 outfp = fdopen(outfd, "w");
 186
 187                 if (!outfp) {
 188                     xfree(password);
 189                     err(EXIT_FAILURE, "%i", outfd);
 190                 }
 191                 break;
 192             case 'S':
 193                 save = 1;
 194                 break;
 195             case 's':
 196                 socketpath = xstrdup(optarg);
 197                 break;
 198             case 'p':
 199                 password = xstrdup(optarg);
 200                 memset(optarg, 0, strlen(optarg));
 201                 break;
 202             case 'P':
 203                 pinentry_path = xstrdup(optarg);
 204                 break;
 205             case 'v':
 206                 xfree(password);
 207                 printf("%s (pwmc)\n%s\n", PACKAGE_STRING, PACKAGE_BUGREPORT);
 208                 exit(EXIT_SUCCESS);
 209             case 'h':
 210             default:
 211                 xfree(password);
 212                 usage(argv[0]);
 213         }
 214     }
 215
 216     filename = argv[optind];
 217     pwmd_init();
 218
 219     if ((pwm = pwmd_connect(socketpath, &error)) == NULL) {
 220         xfree(password);
 221         errx(EXIT_FAILURE, "pwmd_connect(): %s", pwmd_strerror(error));
 222     }
 223
 224     error = pwmd_command(pwm, &result, "OPTION CLIENT NAME=%s", clientname ? clientname : "pwmc");
 225     xfree(clientname);
 226
 227     if (error) {
 228         xfree(password);
 229         errx(EXIT_FAILURE, "pwmd_connect(): %s", pwmd_strerror(error));
 230     }
 231
 232     if (password) {
 233         error = pwmd_setopt(pwm, PWMD_OPTION_PASSWORD, password);
 234
 235         if (error) {
 236             xfree(password);
 237             goto done;
 238         }
 239
 240         xfree(password);
 241     }
 242     else {
 243         if (pinentry_path) {
 244             error = pwmd_command(pwm, &result, "OPTION PATH=%s", pinentry_path);
 245
 246             if (error)
 247                 goto done;
 248         }
 249
 250         if (display) {
 251             error = pwmd_command(pwm, &result, "OPTION DISPLAY=%s", display);
 252
 253             if (error)
 254                 goto done;
 255         }
 256
 257         if (tty) {
 258             error = pwmd_command(pwm, &result, "OPTION TTYNAME=%s", tty);
 259
 260             if (error)
 261                 goto done;
 262         }
 263
 264         if (ttytype) {
 265             error = pwmd_command(pwm, &result, "OPTION TTYTYPE=%s", ttytype);
 266
 267             if (error)
 268                 goto done;
 269         }
 270     }
 271
 272     if (show_status) {
 273         error = pwmd_setopt(pwm, PWMD_OPTION_STATUS_FUNC, status_msg_cb);
 274
 275         if (error)
 276             goto done;
 277     }
 278
 279     if (filename) {
 280         error = pwmd_open(pwm, filename);
 281
 282         if (error)
 283             goto done;
 284     }
 285
 286     p = fgets(command, sizeof(command), stdin);
 287
 288     if (!p)
 289         goto done;
 290
 291     /*
 292      * This is a known INQUIRE command. We use pwmd_inquire() to send the
 293      * data from the do_inquire() callback function.
 294      */
 295     if (strncasecmp(p, "STORE ", 6) == 0) {
 296         struct inquire_s *inq = (struct inquire_s *)malloc(sizeof(struct inquire_s));
 297
 298         if (!inq) {
 299             error = gpg_error_from_errno(ENOMEM);
 300             goto done;
 301         }
 302
 303         inq->data = xstrdup(p+6);
 304         inq->fp = inquirefp;
 305         error = pwmd_inquire(pwm, "STORE", do_inquire, inq);
 306         free(inq);
 307         goto done;
 308     }
 309
 310     if (strcasecmp(p, "BYE") == 0)
 311         goto done;
 312
 313     error = pwmd_command(pwm, &result, command);
 314     memset(command, 0, sizeof(command));
 315
 316     if (error)
 317         goto done;
 318
 319     if (result) {
 320         fwrite(result, 1, strlen(result), outfp);
 321         pwmd_free_result(result);
 322     }
 323
 324 done:
 325     memset(command, 0, sizeof(command));
 326
 327     if (!error && save)
 328         error = pwmd_save(pwm);
 329
 330     if (error) {
 331         show_error(error);
 332         ret = EXIT_FAILURE;
 333     }
 334
 335     pwmd_close(pwm);
 336
 337     if (socketpath)
 338         xfree(socketpath);
 339
 340     exit(ret);
 341 }