libpwmd/exampleclient.c

   1 /* vim:tw=78:ts=8:sw=4:set ft=c:  */
   2 /*
   3     Copyright (C) 2006-2007 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 <pwd.h>
  25 #include <string.h>
  26 #include <limits.h>
  27 #include <fcntl.h>
  28 #include <libpwmd.h>
  29
  30 int main(int argc, char *argv[])
  31 {
  32     pwm_t *pwm;
  33     int error;
  34     int ret;
  35     int opt;
  36     char *socketpath = NULL;
  37     int noagent = 1;
  38     char buf[LINE_MAX];
  39     char *result, *p, *l;
  40
  41     while ((opt = getopt(argc, argv, "hs:a")) != EOF) {
  42         switch (opt) {
  43             case 'a':
  44                 noagent = 0;
  45                 break;
  46             case 's':
  47                 socketpath = optarg;
  48                 break;
  49             case 'h':
  50             default:
  51                 fprintf(stderr, "Usage: %s [-h] [-a] [-s <socketpath>]\n"
  52                                 "    -a   use gpg-agent\n"
  53                                 "    -s   specify an alternate socket path\n"
  54                                 "    -h   this help text\n", argv[0]);
  55                 exit(EXIT_SUCCESS);
  56         }
  57     }
  58
  59     fprintf(stderr, "Connecting to socket...\n");
  60     if ((pwm = pwmd_connect(socketpath, &error)) == NULL)
  61         errx(EXIT_FAILURE, "pwmd_connect(): %s", strerror(error));
  62
  63     fprintf(stderr, "Connected. fd=%i\n", pwm->fd);
  64
  65     if (noagent) {
  66         fprintf(stderr, "Setting password 'test'...\n");
  67
  68         if ((ret = pwmd_command(pwm, &result, &error, PWMD_SETOPT,
  69                     PWMD_OPTION_PASSWORD, "test")) != PWMD_OK) {
  70             pwmd_close(pwm);
  71
  72             if (ret == PWMD_ERROR)
  73                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", strerror(error));
  74             else
  75                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", pwmd_strerror(error));
  76         }
  77     }
  78     else {
  79         fprintf(stderr, "Setting PWMD_OPTION_USEAGENT...\n");
  80         if ((ret = pwmd_command(pwm, &result, &error, PWMD_SETOPT,
  81                     PWMD_OPTION_USEAGENT, 1)) != PWMD_OK) {
  82             pwmd_close(pwm);
  83
  84             if (ret == PWMD_ERROR)
  85                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", strerror(error));
  86             else
  87                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", pwmd_strerror(error));
  88         }
  89
  90         fprintf(stderr, "Setting PWMD_OPTION_TITLE...\n");
  91         if ((ret = pwmd_command(pwm, &result, &error, PWMD_SETOPT,
  92                     PWMD_OPTION_TITLE, "PWMD Client Test")) != PWMD_OK) {
  93             pwmd_close(pwm);
  94
  95             if (ret == PWMD_ERROR)
  96                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", strerror(error));
  97             else
  98                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", pwmd_strerror(error));
  99         }
 100
 101         fprintf(stderr, "Setting PWMD_OPTION_PROMPT...\n");
 102         if ((ret = pwmd_command(pwm, &result, &error, PWMD_SETOPT,
 103                     PWMD_OPTION_PROMPT, "Password please:")) != PWMD_OK) {
 104             pwmd_close(pwm);
 105
 106             if (ret == PWMD_ERROR)
 107                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", strerror(error));
 108             else
 109                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", pwmd_strerror(error));
 110         }
 111
 112         fprintf(stderr, "Setting PWMD_OPTION_DESC...\n");
 113         if ((ret = pwmd_command(pwm, &result, &error, PWMD_SETOPT,
 114                     PWMD_OPTION_DESC, "Enter the password for file \"testfile\".")) != PWMD_OK) {
 115             pwmd_close(pwm);
 116
 117             if (ret == PWMD_ERROR)
 118                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", strerror(error));
 119             else
 120                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", pwmd_strerror(error));
 121         }
 122     }
 123
 124     fprintf(stderr, "Opening 'testfile'...\n");
 125     if ((ret = pwmd_command(pwm, &result, &error, PWMD_OPEN, "testfile")) != PWMD_OK) {
 126         pwmd_close(pwm);
 127
 128         if (ret == PWMD_ERROR)
 129             errx(EXIT_FAILURE, "PWMD_OPEN: %s", strerror(error));
 130         else if (ret == PWMD_PERROR)
 131             errx(EXIT_FAILURE, "PWMD_OPEN: %s", pwmd_strerror(error));
 132         else
 133             errx(EXIT_FAILURE, "PWMD_OPEN: gpg-agent failed. Is GPG_AGENT_INFO set?");
 134     }
 135
 136     fprintf(stderr, "File opened. Creating 'newaccount' and test element...\n");
 137     snprintf(buf, sizeof(buf), "STORE newaccount\tsmtp\thostname\tsomehost.com\n");
 138
 139     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 140         pwmd_close(pwm);
 141
 142         if (ret == PWMD_ERROR)
 143             errx(EXIT_FAILURE, "STORE: %s", strerror(error));
 144         else
 145             errx(EXIT_FAILURE, "STORE: %s", pwmd_strerror(error));
 146     }
 147
 148     fprintf(stderr, "OK. Testing invalid command syntax...\n");
 149     snprintf(buf, sizeof(buf), "GET newaccount\tnop\n");
 150
 151     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 152         if (ret == PWMD_ERROR) {
 153             pwmd_close(pwm);
 154             errx(EXIT_FAILURE, "GET: %s", strerror(error));
 155         }
 156         else
 157             fprintf(stderr, "OK, the error was: %s\n", pwmd_strerror(error));
 158     }
 159
 160     fprintf(stderr, "OK. Creating another element...\n");
 161     snprintf(buf, sizeof(buf), "STORE newaccount\tsmtp\tport\t25\n");
 162     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 163         pwmd_close(pwm);
 164
 165         if (ret == PWMD_ERROR)
 166             errx(EXIT_FAILURE, "STORE: %s", strerror(error));
 167         else
 168             errx(EXIT_FAILURE, "STORE: %s", pwmd_strerror(error));
 169     }
 170
 171     fprintf(stderr, "OK. Testing LIST for account list...\n");
 172     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, "LIST\n")) != PWMD_OK) {
 173         pwmd_close(pwm);
 174
 175         if (ret == PWMD_ERROR)
 176             errx(EXIT_FAILURE, "LIST: %s", strerror(error));
 177         else
 178             errx(EXIT_FAILURE, "LIST: %s", pwmd_strerror(error));
 179     }
 180
 181     l = result;
 182
 183     while ((p = strsep(&result, "\n")))
 184         fprintf(stderr, "%s\n", p);
 185
 186     free(l);
 187
 188     fprintf(stderr, "OK. Testing LIST for specific account 'newaccount'...\n");
 189     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, "LIST newaccount\n")) != PWMD_OK) {
 190         pwmd_close(pwm);
 191
 192         if (ret == PWMD_ERROR)
 193             errx(EXIT_FAILURE, "LIST: %s", strerror(error));
 194         else
 195             errx(EXIT_FAILURE, "LIST: %s", pwmd_strerror(error));
 196     }
 197
 198     l = result;
 199
 200     while ((p = strsep(&result, "\n")))
 201         fprintf(stderr, "%s\n", p);
 202
 203     free(l);
 204     fprintf(stderr, "OK. GET'ing a single element from 'newaccount'...\n");
 205     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND,
 206                     "GET newaccount\tsmtp\thostname\n")) != PWMD_OK) {
 207         pwmd_close(pwm);
 208
 209         if (ret == PWMD_ERROR)
 210             errx(EXIT_FAILURE, "GET: %s", strerror(error));
 211         else
 212             errx(EXIT_FAILURE, "GET: %s", pwmd_strerror(error));
 213     }
 214
 215     fprintf(stderr, "%s\n", result);
 216     free(result);
 217     snprintf(buf, sizeof(buf), "ATTR SET name newaccount namechange\n");
 218     fprintf(stderr, "OK. Changing account 'newaccount' to 'namechange'...\n");
 219     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 220         pwmd_close(pwm);
 221
 222         if (ret == PWMD_ERROR)
 223             errx(EXIT_FAILURE, "ATTR: %s", strerror(error));
 224         else
 225             errx(EXIT_FAILURE, "ATTR: %s", pwmd_strerror(error));
 226     }
 227
 228     fprintf(stderr, "OK. see!\n");
 229     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, "LIST\n")) != PWMD_OK) {
 230         pwmd_close(pwm);
 231
 232         if (ret == PWMD_ERROR)
 233             errx(EXIT_FAILURE, "LIST: %s", strerror(error));
 234         else
 235             errx(EXIT_FAILURE, "LIST: %s", pwmd_strerror(error));
 236     }
 237
 238     l = result;
 239
 240     while ((p = strsep(&result, "\n")))
 241         fprintf(stderr, "%s\n", p);
 242
 243     free(l);
 244
 245     fprintf(stderr, "No really!\n");
 246     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, "LIST namechange\n")) != PWMD_OK) {
 247         pwmd_close(pwm);
 248
 249         if (ret == PWMD_ERROR)
 250             errx(EXIT_FAILURE, "LIST: %s", strerror(error));
 251         else
 252             errx(EXIT_FAILURE, "LIST: %s", pwmd_strerror(error));
 253     }
 254
 255     l = result;
 256
 257     while ((p = strsep(&result, "\n")))
 258         fprintf(stderr, "%s\n", p);
 259
 260     free(l);
 261     fprintf(stderr, "OK. New pop3 tree (STORE)...\n");
 262     snprintf(buf, sizeof(buf), "STORE namechange\tpop3\thostname\tpophost.com\n");
 263     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 264         pwmd_close(pwm);
 265
 266         if (ret == PWMD_ERROR)
 267             errx(EXIT_FAILURE, "STORE: %s", strerror(error));
 268         else
 269             errx(EXIT_FAILURE, "STORE: %s", pwmd_strerror(error));
 270     }
 271
 272     fprintf(stderr, "OK. Testing DELETE of 'namechange\\tsmtp'...\n");
 273     snprintf(buf, sizeof(buf), "DELETE namechange\tsmtp\n");
 274     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 275         pwmd_close(pwm);
 276
 277         if (ret == PWMD_ERROR)
 278             errx(EXIT_FAILURE, "DELETE: %s", strerror(error));
 279         else
 280             errx(EXIT_FAILURE, "DELETE: %s", pwmd_strerror(error));
 281     }
 282
 283     fprintf(stderr, "OK. LIST of account 'namechange'...\n");
 284     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, "LIST namechange\n")) != PWMD_OK) {
 285         pwmd_close(pwm);
 286
 287         if (ret == PWMD_ERROR)
 288             errx(EXIT_FAILURE, "LIST: %s", strerror(error));
 289         else
 290             errx(EXIT_FAILURE, "LIST: %s", pwmd_strerror(error));
 291     }
 292
 293     l = result;
 294
 295     while ((p = strsep(&result, "\n")))
 296         fprintf(stderr, "%s\n", p);
 297
 298     free(l);
 299     fprintf(stderr, "OK. ATTR failure...\n");
 300     snprintf(buf, sizeof(buf), "ATTR GET someattr namechange\n");
 301     if ((ret = pwmd_command(pwm, &result, &error, PWMD_COMMAND, buf)) != PWMD_OK) {
 302         if (ret == PWMD_ERROR)
 303             warnx("ATTR: %s", strerror(error));
 304         else
 305             warnx("ATTR: %s", pwmd_strerror(error));
 306     }
 307
 308     if (noagent == 0) {
 309         fprintf(stderr, "Setting PWMD_OPTION_DESC...\n");
 310         if ((ret = pwmd_command(pwm, &result, &error, PWMD_SETOPT,
 311                     PWMD_OPTION_DESC, "Enter the new password for file \"testfile\".")) != PWMD_OK) {
 312             pwmd_close(pwm);
 313
 314             if (ret == PWMD_ERROR)
 315                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", strerror(error));
 316             else
 317                 errx(EXIT_FAILURE, "PWMD_SETOPT: %s", pwmd_strerror(error));
 318         }
 319     }
 320
 321     fprintf(stderr, "OK. SAVE...\n");
 322     if ((ret = pwmd_command(pwm, &result, &error, PWMD_SAVE)) != PWMD_OK) {
 323         pwmd_close(pwm);
 324
 325         if (ret == PWMD_ERROR)
 326             errx(EXIT_FAILURE, "SAVE: %s", strerror(error));
 327         else if (ret == PWMD_PERROR)
 328             errx(EXIT_FAILURE, "SAVE: %s", pwmd_strerror(error));
 329         else
 330             errx(EXIT_FAILURE, "SAVE: gpg-agent failed. Is GPG_AGENT_INFO set?");
 331     }
 332
 333     pwmd_close(pwm);
 334
 335     fprintf(stderr, "Done! delete the file 'testfile' in your PWMD data directory\n");
 336     exit(EXIT_SUCCESS);
 337 }