src/commands.c

   1 /* vim:tw=78:ts=8:sw=4:set ft=c:  */
   2 /*
   3     Copyright (C) 2006-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  02110-1301  USA
  18 */
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <err.h>
  23 #include <errno.h>
  24 #include <sys/types.h>
  25 #include <sys/stat.h>
  26 #include <fcntl.h>
  27 #include <ctype.h>
  28 #include <glib.h>
  29 #include <gcrypt.h>
  30 #include <zlib.h>
  31
  32 #ifdef HAVE_CONFIG_H
  33 #include <config.h>
  34 #endif
  35
  36 #ifndef MEM_DEBUG
  37 #include "mem.h"
  38 #endif
  39
  40 #include "xml.h"
  41 #include "common.h"
  42
  43 #ifdef WITH_PINENTRY
  44 #include "pinentry.h"
  45 #endif
  46
  47 #include "pwmd_error.h"
  48 #include "cache.h"
  49 #include "misc.h"
  50 #include "commands.h"
  51
  52 static void *z_alloc(void *data, unsigned items, unsigned size)
  53 {
  54 #ifndef MEM_DEBUG
  55     return gcry_calloc(items, size);
  56 #else
  57     return calloc(items, size);
  58 #endif
  59 }
  60
  61 static void z_free(void *data, void *p)
  62 {
  63 #ifndef MEM_DEBUG
  64     gcry_free(p);
  65 #else
  66     free(p);
  67 #endif
  68 }
  69
  70 static gpg_error_t file_modified(struct client_s *client)
  71 {
  72     struct stat st;
  73     gpg_error_t rc;
  74
  75     if (client->state != STATE_OPEN)
  76         return EPWMD_NO_FILE;
  77
  78     rc = lock_file_mutex(client);
  79
  80     if (rc)
  81         return rc;
  82
  83     if (lstat(client->filename, &st) == 0 && client->mtime) {
  84         if (client->mtime != st.st_mtime)
  85             return EPWMD_FILE_MODIFIED;
  86     }
  87
  88     return 0;
  89 }
  90
  91 static gboolean encrypt_xml(gcry_cipher_hd_t gh, void *outbuf, gsize outsize,
  92         void *inbuf, gsize insize)
  93 {
  94     gpg_error_t rc;
  95
  96     if ((rc = gcry_cipher_encrypt(gh, outbuf, outsize, inbuf, insize))) {
  97         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
  98         return FALSE;
  99     }
 100
 101     return TRUE;
 102 }
 103
 104 static gpg_error_t decrypt_xml(gcry_cipher_hd_t gh, void *outbuf, gsize outsize,
 105         void *inbuf, gsize insize)
 106 {
 107     gpg_error_t rc;
 108
 109     if ((rc = gcry_cipher_decrypt(gh, outbuf, outsize, inbuf, insize)))
 110         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 111
 112     return rc;
 113 }
 114
 115 static gpg_error_t parse_xml(assuan_context_t ctx)
 116 {
 117     struct client_s *client = assuan_get_pointer(ctx);
 118
 119     client->doc = xmlReadMemory(client->xml, client->len, NULL, "UTF-8", XML_PARSE_NOBLANKS);
 120
 121     if (!client->doc)
 122         return EPWMD_LIBXML_ERROR;
 123
 124     return 0;
 125 }
 126
 127 void unlock_file_mutex(struct client_s *client)
 128 {
 129     pth_mutex_t *m;
 130
 131 #ifdef WITH_PINENTRY
 132     if (client->has_lock == FALSE || client->pinentry->status != PINENTRY_NONE)
 133 #else
 134     if (client->has_lock == FALSE)
 135 #endif
 136         return;
 137
 138     CACHE_LOCK(client->ctx);
 139
 140     if (cache_get_mutex(client->md5file, &m) == FALSE) {
 141         CACHE_UNLOCK;
 142         return;
 143     }
 144
 145     CACHE_UNLOCK;
 146     pth_mutex_release(m);
 147     client->has_lock = client->is_lock_cmd = FALSE;
 148 }
 149
 150 gpg_error_t lock_file_mutex(struct client_s *client)
 151 {
 152     pth_mutex_t *m;
 153
 154     if (client->has_lock == TRUE)
 155         return 0;
 156
 157     CACHE_LOCK(client->ctx);
 158
 159     if (cache_get_mutex(client->md5file, &m) == FALSE) {
 160         CACHE_UNLOCK;
 161         return 0;
 162     }
 163
 164     CACHE_UNLOCK;
 165
 166     if (pth_mutex_acquire(m, TRUE, NULL) == FALSE) {
 167         if (errno == EBUSY) {
 168             if (client->ctx) {
 169                 /*
 170                  * If a client disconnects unexpectedly while waiting for a
 171                  * lock, this lets the thread terminate because send_status()
 172                  * will return an error. We can't use an PTH_EVENT_FUNC here
 173                  * because the thread that would call the callback uses it's
 174                  * own stack space which messes up access to the client data
 175                  * (if I understand it correctly).
 176                  */
 177                 while (pth_mutex_acquire(m, TRUE, NULL) == FALSE) {
 178                     gpg_error_t rc = send_status(client->ctx, STATUS_LOCKED);
 179
 180                     if (rc)
 181                         return rc;
 182
 183                     pth_sleep(1);
 184                 }
 185             }
 186             else
 187                 pth_mutex_acquire(m, FALSE, NULL);
 188         }
 189         else {
 190             gint e = errno;
 191             log_write("%s(%i): %s", __FUNCTION__, __LINE__, strerror(errno));
 192             return gpg_error_from_errno(e);
 193         }
 194     }
 195
 196     client->has_lock = TRUE;
 197     return 0;
 198 }
 199
 200 void free_client(struct client_s *client)
 201 {
 202     if (client->doc)
 203         xmlFreeDoc(client->doc);
 204
 205     if (client->xml)
 206         gcry_free(client->xml);
 207
 208     if (client->filename)
 209         g_free(client->filename);
 210
 211     if (client->gh)
 212         gcry_cipher_close(client->gh);
 213 }
 214
 215 void cleanup_client(struct client_s *client)
 216 {
 217     assuan_context_t ctx = client->ctx;
 218     struct client_thread_s *thd = client->thd;
 219 #ifdef WITH_PINENTRY
 220     struct pinentry_s *pin = client->pinentry;
 221 #endif
 222
 223     unlock_file_mutex(client);
 224     CACHE_LOCK(client->ctx);
 225     cache_decr_refcount(client->md5file);
 226
 227     /*
 228      * This may be a new file so don't use a cache slot. save_command() will
 229      * set this to FALSE on success.
 230      */
 231     if (client->new == TRUE)
 232         cache_clear(client->md5file, 1);
 233
 234     free_client(client);
 235     memset(client, 0, sizeof(struct client_s));
 236     client->state = STATE_CONNECTED;
 237     client->ctx = ctx;
 238     client->thd = thd;
 239     client->freed = TRUE;
 240 #ifdef WITH_PINENTRY
 241     client->pinentry = pin;
 242 #endif
 243     CACHE_UNLOCK;
 244 }
 245
 246 gboolean do_decompress(assuan_context_t ctx, gpointer in, gint insize,
 247         gpointer *out, glong *outsize, gint *rc)
 248 {
 249     z_stream z;
 250     gpointer pout;
 251     gz_header h;
 252     gchar buf[17];
 253     gchar str[ASSUAN_LINELENGTH];
 254
 255     z.zalloc = z_alloc;
 256     z.zfree = z_free;
 257     z.next_in = in;
 258     z.avail_in = insize;
 259     z.avail_out = zlib_bufsize;
 260     z.next_out = pout = g_malloc(zlib_bufsize);
 261
 262     if (!pout) {
 263         log_write("%s(%i): %s", __FUNCTION__, __LINE__, strerror(ENOMEM));
 264         *rc = Z_MEM_ERROR;
 265         return FALSE;
 266     }
 267
 268     *rc = inflateInit2(&z, 47);
 269
 270     if (*rc != Z_OK) {
 271         log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 272         g_free(pout);
 273         return FALSE;
 274     }
 275
 276     memset(&h, 0, sizeof(gz_header));
 277     h.comment = (guchar *)buf;
 278     h.comm_max = sizeof(buf);
 279     *rc = inflateGetHeader(&z, &h);
 280
 281     if (*rc != Z_OK) {
 282         log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 283         g_free(pout);
 284         inflateEnd(&z);
 285         return FALSE;
 286     }
 287
 288     *rc = inflate(&z, Z_BLOCK);
 289
 290     if (*rc != Z_OK) {
 291         log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 292         g_free(pout);
 293         inflateEnd(&z);
 294         return FALSE;
 295     }
 296
 297     if (h.comment)
 298         insize = atoi((gchar *)h.comment);
 299
 300     do {
 301         gpointer p;
 302
 303         *rc = inflate(&z, Z_FINISH);
 304
 305         switch (*rc) {
 306             case Z_OK:
 307                 break;
 308             case Z_BUF_ERROR:
 309                 if (!z.avail_out) {
 310                     p = g_realloc(pout, z.total_out + zlib_bufsize);
 311
 312                     if (!p) {
 313                         *rc = Z_MEM_ERROR;
 314                         goto fail;
 315                     }
 316
 317                     pout = p;
 318                     z.next_out = pout + z.total_out;
 319                     z.avail_out = zlib_bufsize;
 320
 321                     if (ctx) {
 322                         *rc = assuan_write_status(ctx, "DECOMPRESS",
 323                                 print_fmt(str, sizeof(str), "%i %i", z.total_out, insize));
 324
 325                         if (*rc)
 326                             goto fail;
 327                     }
 328                 }
 329                 break;
 330             case Z_STREAM_END:
 331                 break;
 332             default:
 333                 goto fail;
 334                 break;
 335         }
 336
 337         pth_yield(NULL);
 338     } while (*rc != Z_STREAM_END);
 339
 340     if (ctx) {
 341         *rc = assuan_write_status(ctx, "DECOMPRESS",
 342                 print_fmt(str, sizeof(str), "%i %i", z.total_out, insize));
 343
 344         if (*rc)
 345             goto fail;
 346     }
 347
 348     *out = pout;
 349     *outsize = z.total_out;
 350     inflateEnd(&z);
 351     *rc = 0;
 352     return TRUE;
 353
 354 fail:
 355     log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 356     g_free(pout);
 357     inflateEnd(&z);
 358     return FALSE;
 359 }
 360
 361 gpg_error_t read_file_header(const gchar *filename, file_header_t *fh)
 362 {
 363     gint fd;
 364     gsize len;
 365
 366     fd = open(filename, O_RDONLY);
 367
 368     if (fd == -1)
 369         return gpg_error_from_errno(errno);
 370
 371     len = pth_read(fd, fh, sizeof(file_header_t));
 372     close(fd);
 373
 374     if (len != sizeof(file_header_t))
 375         return gpg_error_from_errno(errno);
 376
 377     return 0;
 378 }
 379
 380 static gpg_error_t open_command_finalize(assuan_context_t ctx, guchar shakey[],
 381         gboolean cached)
 382 {
 383     struct client_s *client = assuan_get_pointer(ctx);
 384     gpg_error_t rc;
 385     struct stat st;
 386     gint fd;
 387     gint timeout;
 388     gint iter;
 389
 390     if ((fd = open_file(client->filename, &st)) == -1) {
 391         /* New file. */
 392         if (errno == ENOENT) {
 393             if (shakey[0])
 394                 goto update_cache;
 395
 396             goto done;
 397         }
 398
 399         rc = errno;
 400         log_write("%s: %s", client->filename, strerror(errno));
 401         cleanup_client(client);
 402         memset(shakey, 0, sizeof(shakey));
 403         return send_syserror(ctx, rc);
 404     }
 405
 406     rc = try_xml_decrypt(ctx, fd, st, shakey, &iter);
 407     close(fd);
 408
 409     if (rc) {
 410         memset(shakey, 0, sizeof(shakey));
 411         cleanup_client(client);
 412         return send_error(ctx, rc);
 413     }
 414
 415 update_cache:
 416     CACHE_LOCK(client->ctx);
 417
 418     if (cached == FALSE) {
 419         if (cache_update_key(client->md5file, shakey) == FALSE) {
 420             log_write("%s(%i): %s", __FILE__, __LINE__, pwmd_strerror(EPWMD_MAX_SLOTS));
 421             cleanup_client(client);
 422             CACHE_UNLOCK;
 423             return send_error(ctx, EPWMD_MAX_SLOTS);
 424         }
 425
 426         timeout = get_key_file_integer(client->filename, "cache_timeout");
 427         cache_reset_timeout(client->md5file, timeout);
 428     }
 429     else
 430         cache_set_timeout(client->md5file, -2);
 431
 432     CACHE_UNLOCK;
 433
 434 done:
 435     memset(shakey, 0, sizeof(shakey));
 436     rc = parse_xml(ctx);
 437
 438     if (client->xml) {
 439         gcry_free(client->xml);
 440         client->xml = NULL;
 441     }
 442
 443     if (!rc) {
 444         if (client->new == FALSE)
 445             send_status_all(STATUS_CACHE);
 446
 447         client->state = STATE_OPEN;
 448     }
 449
 450     if (!rc && client->new == FALSE &&
 451             iter != get_key_file_integer(client->filename, "iterations")) {
 452         g_key_file_set_integer(keyfileh, client->filename, "iterations", iter);
 453         send_status_all(STATUS_CONFIG);
 454     }
 455
 456     return send_error(ctx, rc);
 457 }
 458
 459 static int open_command(assuan_context_t ctx, char *line)
 460 {
 461     struct stat st;
 462     guchar shakey[gcrykeysize];
 463     gboolean cached = FALSE;
 464     gpg_error_t rc;
 465     struct client_s *client = assuan_get_pointer(ctx);
 466     gchar **req;
 467     gchar *filename = NULL;
 468     file_header_t file_header;
 469
 470     memset(shakey, 0, sizeof(shakey));
 471
 472     if ((req = split_input_line(line, " ", 2)) != NULL)
 473         filename = req[0];
 474
 475     if (!filename || !*filename) {
 476         g_strfreev(req);
 477         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
 478     }
 479
 480     if (valid_filename(filename) == FALSE) {
 481         g_strfreev(req);
 482         return send_error(ctx, EPWMD_INVALID_FILENAME);
 483     }
 484
 485     if (client->state == STATE_OPEN)
 486         cleanup_client(client);
 487
 488     gcry_md_hash_buffer(GCRY_MD_MD5, client->md5file, filename, strlen(filename));
 489     CACHE_LOCK(client->ctx);
 490
 491     if (cache_has_file(client->md5file) == FALSE) {
 492         if (cache_add_file(client->md5file, NULL) == FALSE) {
 493             g_strfreev(req);
 494             CACHE_UNLOCK;
 495             return send_error(ctx, EPWMD_MAX_SLOTS);
 496         }
 497     }
 498
 499     cache_incr_refcount(client->md5file);
 500     CACHE_UNLOCK;
 501     rc = lock_file_mutex(client);
 502
 503     if (rc) {
 504         g_strfreev(req);
 505         return send_error(ctx, rc);
 506     }
 507
 508     client->freed = FALSE;
 509
 510     if ((rc = gcry_cipher_open(&client->gh, GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_CBC, 0))) {
 511         g_strfreev(req);
 512         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 513         cleanup_client(client);
 514         return send_error(ctx, rc);
 515     }
 516
 517     if (lstat(filename, &st) == 0) {
 518         if (!S_ISREG(st.st_mode)) {
 519             log_write("%s: %s", filename, pwmd_strerror(EPWMD_INVALID_FILENAME));
 520             g_strfreev(req);
 521             cleanup_client(client);
 522             return send_error(ctx, EPWMD_INVALID_FILENAME);
 523         }
 524
 525         client->mtime = st.st_mtime;
 526     }
 527
 528     client->filename = g_strdup(filename);
 529
 530     if (!client->filename) {
 531         memset(shakey, 0, sizeof(shakey));
 532         log_write("%s(%i): %s", __FILE__, __LINE__, strerror(ENOMEM));
 533         cleanup_client(client);
 534         g_strfreev(req);
 535         return send_syserror(ctx, ENOMEM);
 536     }
 537
 538 #ifdef WITH_PINENTRY
 539     client->pinentry->filename = g_strdup(client->filename);
 540
 541     if (!client->pinentry->filename) {
 542         memset(shakey, 0, sizeof(shakey));
 543         log_write("%s(%i): %s", __FILE__, __LINE__, strerror(ENOMEM));
 544         cleanup_client(client);
 545         g_strfreev(req);
 546         return send_syserror(ctx, ENOMEM);
 547     }
 548 #endif
 549
 550     /*
 551      * New files don't need a key.
 552      */
 553     if (access(filename, R_OK) != 0) {
 554         if (errno != ENOENT) {
 555             rc = errno;
 556             log_write("%s: %s", filename, strerror(errno));
 557             g_strfreev(req);
 558             cleanup_client(client);
 559             return send_syserror(ctx, rc);
 560         }
 561
 562         if ((client->xml = new_document()) == NULL) {
 563             log_write("%s", strerror(ENOMEM));
 564             g_strfreev(req);
 565             cleanup_client(client);
 566             return send_syserror(ctx, ENOMEM);
 567         }
 568
 569         client->len = xmlStrlen(client->xml);
 570         client->new = TRUE;
 571         client->filename = g_strdup(filename);
 572
 573         if (!client->filename) {
 574             g_strfreev(req);
 575             cleanup_client(client);
 576             log_write("%s(%i): %s", __FILE__, __LINE__, strerror(ENOMEM));
 577             return send_syserror(ctx, ENOMEM);
 578         }
 579
 580         memset(shakey, 0, sizeof(shakey));
 581
 582         if (req[1] && *req[1])
 583             gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, req[1], strlen(req[1]));
 584
 585         g_strfreev(req);
 586         return open_command_finalize(ctx, shakey, cached);
 587     }
 588
 589     rc = read_file_header(filename, &file_header);
 590
 591     if (rc) {
 592         g_strfreev(req);
 593         cleanup_client(client);
 594         return send_error(ctx, rc);
 595     }
 596
 597     if (file_header.iter == -1)
 598         goto done;
 599
 600     CACHE_LOCK(client->ctx);
 601     cached = cache_get_key(client->md5file, shakey);
 602     CACHE_UNLOCK;
 603
 604     if (cached == FALSE) {
 605         /*
 606          * No key specified and no matching filename found in the cache. Use
 607          * pinentry to retrieve the key. Cannot return assuan_process_done()
 608          * here otherwise the command will be interrupted. The event loop in
 609          * client_thread() will poll the file descriptor waiting for it to
 610          * become ready to read a pinentry_key_s which will contain the
 611          * entered key or rc. It will then call open_command_finalize() to
 612          * to finish the command.
 613          */
 614         if (!req[1] || !*req[1]) {
 615 #ifdef WITH_PINENTRY
 616             gboolean b = get_key_file_boolean(filename, "enable_pinentry");
 617
 618             /* From set_pinentry_defaults(). */
 619             if (client->pinentry->enable == FALSE ||
 620                     (client->pinentry->enable == -1 && b == FALSE)) {
 621                 gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, "", 1);
 622                 goto done;
 623             }
 624
 625             g_strfreev(req);
 626             rc = lock_pin_mutex(client);
 627
 628             if (rc) {
 629                 unlock_pin_mutex(client->pinentry);
 630                 cleanup_client(client);
 631                 return send_error(ctx, rc);
 632             }
 633
 634             client->pinentry->which = PINENTRY_OPEN;
 635             rc = pinentry_fork(ctx);
 636
 637             if (rc) {
 638                 unlock_pin_mutex(client->pinentry);
 639                 cleanup_client(client);
 640                 return send_error(ctx, rc);
 641             }
 642
 643             client->pinentry->cb = open_command_finalize;
 644             client->pinentry->status = PINENTRY_INIT;
 645             return 0;
 646 #else
 647             gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, "", 1);
 648             goto done;
 649 #endif
 650         }
 651
 652         gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, req[1], strlen(req[1]));
 653     }
 654
 655 done:
 656     g_strfreev(req);
 657     return open_command_finalize(ctx, shakey, cached);
 658 }
 659
 660 gboolean do_compress(assuan_context_t ctx, gint level, gpointer data,
 661         gint size, gpointer *out, glong *outsize, gint *rc)
 662 {
 663     z_stream z;
 664     gpointer pout, pin;
 665     gz_header h;
 666     gchar buf[17];
 667     gint cmd = Z_NO_FLUSH;
 668     gchar str[ASSUAN_LINELENGTH];
 669
 670     z.zalloc = z_alloc;
 671     z.zfree = z_free;
 672     z.next_in = pin = data;
 673     z.avail_in = size < zlib_bufsize ? size : zlib_bufsize;
 674     z.avail_out = zlib_bufsize;
 675     z.next_out = pout = g_malloc(zlib_bufsize);
 676
 677     if (!pout) {
 678         log_write("%s(%i): %s", __FUNCTION__, __LINE__, strerror(ENOMEM));
 679         *rc = Z_MEM_ERROR;
 680         return FALSE;
 681     }
 682
 683     *rc = deflateInit2(&z, level, Z_DEFLATED, 31, 8, Z_DEFAULT_STRATEGY);
 684
 685     if (*rc != Z_OK) {
 686         log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 687         g_free(pout);
 688         return FALSE;
 689     }
 690
 691     memset(&h, 0, sizeof(gz_header));
 692     g_snprintf(buf, sizeof(buf), "%i", size);
 693     h.comment = (guchar *)buf;
 694     *rc = deflateSetHeader(&z, &h);
 695
 696     if (*rc != Z_OK) {
 697         log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 698         g_free(pout);
 699         deflateEnd(&z);
 700         return FALSE;
 701     }
 702
 703     do {
 704         gpointer p;
 705
 706         *rc = deflate(&z, cmd);
 707
 708         switch (*rc) {
 709             case Z_OK:
 710                 break;
 711             case Z_BUF_ERROR:
 712                 if (!z.avail_out) {
 713                     p = g_realloc(pout, z.total_out + zlib_bufsize);
 714
 715                     if (!p) {
 716                         *rc = Z_MEM_ERROR;
 717                         goto fail;
 718                     }
 719
 720                     pout = p;
 721                     z.next_out = pout + z.total_out;
 722                     z.avail_out = zlib_bufsize;
 723                 }
 724
 725                 if (!z.avail_in && z.total_in < size) {
 726                     if (z.total_in + zlib_bufsize > size)
 727                         z.avail_in = size - z.total_in;
 728                     else
 729                         z.avail_in = zlib_bufsize;
 730
 731                     if (ctx) {
 732                         *rc = assuan_write_status(ctx, "COMPRESS",
 733                                 print_fmt(str, sizeof(str), "%i %i", z.total_in, size));
 734
 735                         if (*rc)
 736                             goto fail;
 737                     }
 738                 }
 739
 740                 if (z.total_in >= size)
 741                     cmd = Z_FINISH;
 742
 743                 break;
 744             case Z_STREAM_END:
 745                 break;
 746             default:
 747                 goto fail;
 748         }
 749
 750         pth_yield(NULL);
 751     } while (*rc != Z_STREAM_END);
 752
 753     if (ctx) {
 754         *rc = assuan_write_status(ctx, "COMPRESS",
 755                 print_fmt(str, sizeof(str), "%i %i", z.total_in, size));
 756
 757         if (*rc)
 758             goto fail;
 759     }
 760
 761     *out = pout;
 762     *outsize = z.total_out;
 763     deflateEnd(&z);
 764     *rc = 0;
 765     return TRUE;
 766
 767 fail:
 768     log_write("%s(%i): %s", __FUNCTION__, __LINE__, z.msg);
 769     g_free(pout);
 770     deflateEnd(&z);
 771     return FALSE;
 772 }
 773
 774 gpg_error_t do_xml_encrypt(struct client_s *client, gcry_cipher_hd_t gh,
 775         const gchar *filename, gpointer data, size_t insize, guchar *shakey,
 776         gint iter)
 777 {
 778     gsize len = insize;
 779     gint fd;
 780     gpointer inbuf;
 781     guchar tkey[gcrykeysize];
 782     gchar *p;
 783     gpg_error_t rc;
 784     guint iter_progress = 0, n_iter = 0, xiter = 0;
 785     gchar tmp[FILENAME_MAX];
 786     struct stat st;
 787     mode_t mode = 0;
 788     file_header_t file_header;
 789     gchar str[ASSUAN_LINELENGTH];
 790
 791     if (iter == -1) {
 792         /*
 793          * cache_file_count() needs both .used == TRUE and a valid key in
 794          * order for it to count as a used cache entry. Fixes CACHE status
 795          * messages.
 796          */
 797         memset(shakey, '!', gcrykeysize);
 798         inbuf = data;
 799         file_header.iter = iter;
 800         goto write_file;
 801     }
 802
 803     if (insize / gcryblocksize) {
 804         len = (insize / gcryblocksize) * gcryblocksize;
 805
 806         if (insize % gcryblocksize)
 807             len += gcryblocksize;
 808     }
 809
 810     /*
 811      * Resize the existing xml buffer to the block size required by gcrypt
 812      * rather than duplicating it and wasting memory.
 813      */
 814     inbuf = gcry_realloc(data, len);
 815
 816     if (!inbuf)
 817         return gpg_error_from_errno(ENOMEM);
 818
 819     insize = len;
 820     gcry_create_nonce(file_header.iv, sizeof(file_header.iv));
 821     memcpy(tkey, shakey, sizeof(tkey));
 822     tkey[0] ^= 1;
 823
 824     if ((rc = gcry_cipher_setkey(gh, tkey, gcrykeysize))) {
 825         memset(tkey, 0, sizeof(tkey));
 826         gcry_free(inbuf);
 827         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 828         return rc;
 829     }
 830
 831     memset(tkey, 0, sizeof(tkey));
 832     file_header.iter = iter;
 833
 834     if (client)
 835         iter_progress = get_key_file_integer(client->filename, "iteration_progress");
 836
 837     if (client && iter_progress && file_header.iter >= iter_progress) {
 838         rc = assuan_write_status(client->ctx, "ENCRYPT",
 839                 print_fmt(str, sizeof(str), "%i %i", 0, file_header.iter));
 840
 841         if (rc) {
 842             gcry_free(inbuf);
 843             return rc;
 844         }
 845     }
 846
 847     while (xiter < file_header.iter) {
 848         if (client && iter_progress > 0 && xiter >= iter_progress) {
 849             if (!(xiter % iter_progress)) {
 850                 rc = assuan_write_status(client->ctx, "ENCRYPT",
 851                         print_fmt(str, sizeof(str), "%i %i", ++n_iter * iter_progress, file_header.iter));
 852
 853                 if (rc) {
 854                     gcry_free(inbuf);
 855                     return rc;
 856                 }
 857             }
 858         }
 859
 860         if ((rc = gcry_cipher_setiv(gh, file_header.iv,
 861                         sizeof(file_header.iv)))) {
 862             gcry_free(inbuf);
 863             log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 864             return rc;
 865         }
 866
 867         if (encrypt_xml(gh, inbuf, insize, NULL, 0)
 868                 == FALSE) {
 869             gcry_free(inbuf);
 870             log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 871             return rc;
 872         }
 873
 874         xiter++;
 875         pth_yield(NULL);
 876     }
 877
 878     if ((rc = gcry_cipher_setiv(gh, file_header.iv,
 879                     sizeof(file_header.iv)))) {
 880         gcry_free(inbuf);
 881         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 882         return rc;
 883     }
 884
 885     if ((rc = gcry_cipher_setkey(gh, shakey, gcrykeysize))) {
 886         gcry_free(inbuf);
 887         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 888         return rc;
 889     }
 890
 891     if (encrypt_xml(gh, inbuf, insize, NULL, 0) == FALSE) {
 892         gcry_free(inbuf);
 893         log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
 894         return rc;
 895     }
 896
 897     if (client && iter_progress && file_header.iter >= iter_progress) {
 898         rc = assuan_write_status(client->ctx, "ENCRYPT",
 899                 print_fmt(str, sizeof(str), "%i %i", file_header.iter, file_header.iter));
 900
 901         if (rc) {
 902             gcry_free(inbuf);
 903             return rc;
 904         }
 905     }
 906
 907 write_file:
 908     if (filename) {
 909         if (lstat(filename, &st) == 0) {
 910             mode = st.st_mode & (S_IRWXU|S_IRWXG|S_IRWXO);
 911
 912             /*
 913              * FIXME What if the file has an ACL?
 914              */
 915             if (!(mode & S_IWUSR)) {
 916                 gcry_free(inbuf);
 917                 return gpg_error_from_errno(EACCES);
 918             }
 919         }
 920         else {
 921             if (errno != ENOENT) {
 922                 rc = errno;
 923                 gcry_free(inbuf);
 924                 return gpg_error_from_errno(rc);
 925             }
 926         }
 927
 928         g_snprintf(tmp, sizeof(tmp), ".%s.tmp", filename);
 929
 930         if ((fd = open(tmp, O_WRONLY|O_CREAT|O_TRUNC, 0600)) == -1) {
 931             rc = errno;
 932             gcry_free(inbuf);
 933             p = strrchr(tmp, '/');
 934             p++;
 935             log_write("%s: %s", p, strerror(rc));
 936             return gpg_error_from_errno(rc);
 937         }
 938     }
 939     else
 940         /*
 941          * xml_import() from command line.
 942          */
 943         fd = STDOUT_FILENO;
 944
 945     len = pth_write(fd, &file_header, sizeof(file_header_t));
 946
 947     if (len != sizeof(file_header)) {
 948         len = errno;
 949
 950         if (filename) {
 951             close(fd);
 952             unlink(tmp);
 953         }
 954
 955         gcry_free(inbuf);
 956         return gpg_error_from_errno(len);
 957     }
 958
 959     len = pth_write(fd, inbuf, insize);
 960
 961     if (len != insize) {
 962         len = errno;
 963
 964         if (filename) {
 965             close(fd);
 966             unlink(tmp);
 967         }
 968
 969         gcry_free(inbuf);
 970         return gpg_error_from_errno(len);
 971     }
 972
 973     if (fsync(fd) == -1) {
 974         len = errno;
 975
 976         if (filename) {
 977             close(fd);
 978             unlink(tmp);
 979         }
 980
 981         gcry_free(inbuf);
 982         return gpg_error_from_errno(len);
 983     }
 984
 985     if (filename) {
 986         close(fd);
 987
 988         if (mode && get_key_file_boolean(filename, "backup") == TRUE) {
 989             gchar tmp2[FILENAME_MAX];
 990
 991             g_snprintf(tmp2, sizeof(tmp2), "%s.backup", filename);
 992
 993             if (rename(filename, tmp2) == -1) {
 994                 unlink(tmp);
 995                 len = errno;
 996                 gcry_free(inbuf);
 997                 return gpg_error_from_errno(len);
 998             }
 999         }
1000
1001         if (rename(tmp, filename) == -1) {
1002             len = errno;
1003             unlink(tmp);
1004             gcry_free(inbuf);
1005             return gpg_error_from_errno(len);
1006         }
1007
1008         if (mode)
1009             chmod(filename, mode);
1010     }
1011
1012     gcry_free(inbuf);
1013     return 0;
1014 }
1015
1016 static gpg_error_t save_command_finalize(assuan_context_t ctx,
1017         guchar shakey[], gboolean cached)
1018 {
1019     struct client_s *client = assuan_get_pointer(ctx);
1020     gpointer xmlbuf;
1021     xmlChar *p;
1022     gint len;
1023     gint iter;
1024     gint timeout;
1025     gpointer outbuf;
1026     glong outsize = 0;
1027     gint zrc;
1028     gpg_error_t rc;
1029     struct stat st;
1030
1031     xmlDocDumpFormatMemory(client->doc, &p, &len, 0);
1032     xmlbuf = p;
1033
1034     iter = get_key_file_integer(client->filename, "compression_level");
1035
1036     if (iter < 0)
1037         iter = 0;
1038
1039     if (do_compress(ctx, iter, xmlbuf, len, &outbuf, &outsize, &zrc) == FALSE) {
1040         memset(shakey, 0, sizeof(shakey));
1041         xmlFree(xmlbuf);
1042
1043         if (zrc == Z_MEM_ERROR) {
1044             return send_syserror(ctx, ENOMEM);
1045         }
1046         else
1047             return send_error(ctx, GPG_ERR_COMPR_ALGO);
1048     }
1049     else {
1050         gcry_free(xmlbuf);
1051         xmlbuf = outbuf;
1052         len = outsize;
1053     }
1054
1055     iter = get_key_file_integer(client->filename, "iterations");
1056     rc = do_xml_encrypt(client, client->gh, client->filename, xmlbuf, len, shakey, iter);
1057
1058     if (rc) {
1059         memset(shakey, 0, sizeof(shakey));
1060         return send_error(ctx, rc);
1061     }
1062
1063     lstat(client->filename, &st);
1064     client->mtime = st.st_mtime;
1065     timeout = get_key_file_integer(client->filename, "cache_timeout");
1066     CACHE_LOCK(client->ctx);
1067
1068     if (cached) {
1069         memset(shakey, 0, sizeof(shakey));
1070         cache_reset_timeout(client->md5file, timeout);
1071         CACHE_UNLOCK;
1072
1073         if (client->new == TRUE)
1074             send_status_all(STATUS_CACHE);
1075
1076         client->new = FALSE;
1077         return send_error(ctx, 0);
1078     }
1079
1080     if (cache_update_key(client->md5file, shakey) == FALSE) {
1081         memset(shakey, 0, sizeof(shakey));
1082         log_write("%s(%i): %s", __FILE__, __LINE__, pwmd_strerror(EPWMD_MAX_SLOTS));
1083         CACHE_UNLOCK;
1084         return send_error(ctx, EPWMD_MAX_SLOTS);
1085     }
1086
1087     client->new = FALSE;
1088     memset(shakey, 0, sizeof(shakey));
1089     cache_reset_timeout(client->md5file, timeout);
1090     CACHE_UNLOCK;
1091     send_status_all(STATUS_CACHE);
1092     return send_error(ctx, 0);
1093 }
1094
1095 static int save_command(assuan_context_t ctx, char *line)
1096 {
1097     gboolean cached = FALSE;
1098     guchar shakey[gcrykeysize];
1099     struct stat st;
1100     struct client_s *client = assuan_get_pointer(ctx);
1101     gpg_error_t rc;
1102
1103     memset(shakey, 0, sizeof(shakey));
1104     rc = file_modified(client);
1105
1106     if (rc)
1107         return send_error(ctx, rc);
1108
1109     rc = lock_file_mutex(client);
1110
1111     if (rc)
1112         return send_error(ctx, rc);
1113
1114     if (lstat(client->filename, &st) == -1 && errno != ENOENT)
1115         return send_syserror(ctx, errno);
1116
1117     if (errno != ENOENT && !S_ISREG(st.st_mode)) {
1118         log_write("%s: %s", client->filename, pwmd_strerror(EPWMD_INVALID_FILENAME));
1119         return send_error(ctx, EPWMD_INVALID_FILENAME);
1120     }
1121
1122     if (get_key_file_integer(client->filename, "iterations") == -1)
1123         goto done;
1124
1125     if (!line || !*line) {
1126         guchar tmp[sizeof(shakey)];
1127         CACHE_LOCK(ctx);
1128
1129         memset(tmp, '!', sizeof(tmp));
1130
1131         if (cache_get_key(client->md5file, shakey) == FALSE ||
1132                 memcmp(shakey, tmp, sizeof(shakey)) == 0) {
1133             CACHE_UNLOCK;
1134 #ifdef WITH_PINENTRY
1135             if (get_key_file_boolean(client->filename, "enable_pinentry") == FALSE) {
1136                 gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, "", 1);
1137                 goto done;
1138             }
1139
1140             lock_pin_mutex(client);
1141             client->pinentry->which = PINENTRY_SAVE;
1142             rc = pinentry_fork(ctx);
1143
1144             if (rc) {
1145                 unlock_pin_mutex(client->pinentry);
1146                 return send_error(ctx, rc);
1147             }
1148
1149             client->pinentry->cb = save_command_finalize;
1150             client->pinentry->status = PINENTRY_INIT;
1151             return 0;
1152 #else
1153             gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, "", 1);
1154             goto done;
1155 #endif
1156         }
1157         else {
1158             CACHE_UNLOCK;
1159             cached = TRUE;
1160         }
1161     }
1162     else {
1163         gcry_md_hash_buffer(GCRY_MD_SHA256, shakey, line, strlen(line));
1164         memset(line, 0, strlen(line));
1165     }
1166
1167 done:
1168     return save_command_finalize(ctx, shakey, cached);
1169 }
1170
1171 static int delete_command(assuan_context_t ctx, char *line)
1172 {
1173     struct client_s *client = assuan_get_pointer(ctx);
1174     gchar **req;
1175     gpg_error_t rc;
1176     xmlNodePtr n;
1177
1178     rc = file_modified(client);
1179
1180     if (rc)
1181         return send_error(ctx, rc);
1182
1183     if (strchr(line, '\t'))
1184         req = split_input_line(line, "\t", -1);
1185     else
1186         req = split_input_line(line, " ", -1);
1187
1188     if (!req || !*req)
1189         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
1190
1191     n = find_account(client->doc, &req, &rc, NULL, 0);
1192
1193     if (!n) {
1194         g_strfreev(req);
1195         return send_error(ctx, rc);
1196     }
1197
1198     /*
1199      * No sub-node defined. Remove the entire node (account).
1200      */
1201     if (!req[1]) {
1202         if (n) {
1203             xmlUnlinkNode(n);
1204             xmlFreeNode(n);
1205         }
1206
1207         g_strfreev(req);
1208         return send_error(ctx, 0);
1209     }
1210
1211     n = find_elements(client->doc, n->children, req+1, &rc, NULL, NULL, NULL, FALSE, 0, NULL);
1212     g_strfreev(req);
1213
1214     if (!n)
1215         return send_error(ctx, rc);
1216
1217     if (n) {
1218         xmlUnlinkNode(n);
1219         xmlFreeNode(n);
1220     }
1221
1222     return send_error(ctx, 0);
1223 }
1224
1225 /*
1226  * Don't return with assuan_process_done() here. This has been called from
1227  * assuan_process_next() and the command should be finished in
1228  * client_thread().
1229  */
1230 static int store_command_finalize(gpointer data, gint assuan_rc, guchar *line,
1231         gsize len)
1232 {
1233     assuan_context_t ctx = data;
1234     struct client_s *client = assuan_get_pointer(ctx);
1235     gchar **req;
1236     xmlNodePtr n;
1237     gpg_error_t rc = file_modified(client);
1238
1239     if (assuan_rc || rc) {
1240         if (line)
1241 #ifndef MEM_DEBUG
1242             xfree(line);
1243 #else
1244             free(line);
1245 #endif
1246         return assuan_rc ? assuan_rc : rc;
1247     }
1248
1249     req = split_input_line((gchar *)line, "\t", 0);
1250 #ifndef MEM_DEBUG
1251     xfree(line);
1252 #else
1253     free(line);
1254 #endif
1255
1256     if (!req || !*req)
1257         return EPWMD_COMMAND_SYNTAX;
1258
1259     if (valid_xml_element((xmlChar *)*req) == FALSE) {
1260         g_strfreev(req);
1261         return EPWMD_INVALID_ELEMENT;
1262     }
1263
1264     if (valid_element_path(req+1, TRUE) == FALSE) {
1265         g_strfreev(req);
1266         return EPWMD_INVALID_ELEMENT;
1267     }
1268
1269 again:
1270     n = find_account(client->doc, &req, &rc, NULL, 0);
1271
1272     if (rc && rc == EPWMD_ELEMENT_NOT_FOUND) {
1273         rc = new_account(client->doc, *req);
1274
1275         if (rc) {
1276             g_strfreev(req);
1277             return rc;
1278         }
1279
1280         goto again;
1281     }
1282
1283     if (!n) {
1284         g_strfreev(req);
1285         return rc;
1286     }
1287
1288     if (req[1]) {
1289         if (!n->children)
1290             create_elements_cb(n, req+1, &rc, NULL);
1291         else
1292             find_elements(client->doc, n->children, req+1, &rc,
1293                     NULL, NULL, create_elements_cb, FALSE, 0, NULL);
1294     }
1295
1296     g_strfreev(req);
1297     client->inquire_status = INQUIRE_DONE;
1298     return rc;
1299 }
1300
1301 static int store_command(assuan_context_t ctx, char *line)
1302 {
1303     struct client_s *client = assuan_get_pointer(ctx);
1304     gpg_error_t rc = file_modified(client);
1305
1306     if (rc)
1307         return send_error(ctx, rc);
1308
1309     rc = assuan_inquire_ext(ctx, "STORE", 0, store_command_finalize, ctx);
1310
1311     if (rc)
1312         return send_error(ctx, rc);
1313
1314     /* Don't return with assuan_process_done() here. This is an INQUIRE. */
1315     client->inquire_status = INQUIRE_BUSY;
1316     return 0;
1317 }
1318
1319 static int get_command(assuan_context_t ctx, char *line)
1320 {
1321     struct client_s *client = assuan_get_pointer(ctx);
1322     gchar **req;
1323     gpg_error_t rc;
1324     xmlNodePtr n;
1325
1326     rc = file_modified(client);
1327
1328     if (rc)
1329         return send_error(ctx, rc);
1330
1331     req = split_input_line(line, "\t", -1);
1332
1333     if (!req || !*req) {
1334         g_strfreev(req);
1335         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
1336     }
1337
1338     n = find_account(client->doc, &req, &rc, NULL, 0);
1339
1340     if (!n) {
1341         g_strfreev(req);
1342         return send_error(ctx, rc);
1343     }
1344
1345     if (req[1])
1346         n = find_elements(client->doc, n->children, req+1, &rc, NULL, NULL, NULL, FALSE, 0, NULL);
1347
1348     g_strfreev(req);
1349
1350     if (rc)
1351         return send_error(ctx, rc);
1352
1353     if (!n || !n->children)
1354         return send_error(ctx, EPWMD_EMPTY_ELEMENT);
1355
1356     n = find_text_node(n->children);
1357
1358     if (!n || !n->content || !*n->content)
1359         return send_error(ctx, EPWMD_EMPTY_ELEMENT);
1360
1361     rc = assuan_send_data(ctx, n->content, xmlStrlen(n->content));
1362     return send_error(ctx, rc);
1363 }
1364
1365 static xmlNodePtr realpath_elements_cb(xmlNodePtr node, gchar **target,
1366         gpg_error_t *rc, gchar **req_orig, void *data)
1367 {
1368     gchar *path = *(gchar **)data;
1369     gchar *tmp = NULL, *result;
1370
1371     if (path) {
1372         g_free(path);
1373         *(gchar **)data = NULL;
1374     }
1375
1376     path = g_strjoinv("\t", target);
1377
1378     if (!path) {
1379         *rc = gpg_error_from_errno(ENOMEM);
1380         return NULL;
1381     }
1382
1383     if (req_orig) {
1384         tmp = g_strjoinv("\t", req_orig);
1385
1386         if (!tmp) {
1387             g_free(path);
1388             *rc = gpg_error_from_errno(ENOMEM);
1389             return NULL;
1390         }
1391     }
1392
1393     if (tmp && *tmp)
1394         result = g_strdup_printf("%s\t%s", path, tmp);
1395     else
1396         result = g_strdup(path);
1397
1398     if (!result) {
1399         *rc = gpg_error_from_errno(ENOMEM);
1400         g_free(path);
1401         g_free(tmp);
1402         return NULL;
1403     }
1404
1405     g_free(path);
1406     g_free(tmp);
1407     *(gchar **)data = result;
1408     return node;
1409 }
1410
1411 static int realpath_command(assuan_context_t ctx, char *line)
1412 {
1413     gpg_error_t rc;
1414     struct client_s *client = assuan_get_pointer(ctx);
1415     gchar **req;
1416     gchar *t;
1417     gint i;
1418     xmlNodePtr n;
1419     GString *string;
1420     gchar *rp = NULL;
1421
1422     rc = file_modified(client);
1423
1424     if (rc)
1425         return send_error(ctx, rc);
1426
1427     if (strchr(line, '\t') != NULL) {
1428         if ((req = split_input_line(line, "\t", 0)) == NULL)
1429             return send_error(ctx, EPWMD_COMMAND_SYNTAX);
1430     }
1431     else {
1432         if ((req = split_input_line(line, " ", 0)) == NULL)
1433             return send_error(ctx, EPWMD_COMMAND_SYNTAX);
1434     }
1435
1436     n = find_account(client->doc, &req, &rc, NULL, 0);
1437
1438     if (!n) {
1439         g_strfreev(req);
1440         return send_error(ctx, rc);
1441     }
1442
1443     rp = g_strjoinv("\t", req);
1444
1445     if (!rp) {
1446         g_strfreev(req);
1447         return send_syserror(ctx, ENOMEM);
1448     }
1449
1450     if (req[1]) {
1451         n = find_elements(client->doc, n->children, req+1, &rc,
1452                 NULL, realpath_elements_cb, NULL, FALSE, 0, &rp);
1453
1454         if (!n) {
1455             g_free(rp);
1456             g_strfreev(req);
1457             return send_error(ctx, rc);
1458         }
1459     }
1460
1461     string = g_string_new(rp);
1462     g_free(rp);
1463     g_strfreev(req);
1464
1465     if (!string)
1466         return send_syserror(ctx, ENOMEM);
1467
1468 again:
1469     for (i = 0, t = string->str + i; *t; t++, i++) {
1470         if ((!i && *t != '!') || (*t == '\t' && *(t+1) && *(t+1) != '!')) {
1471             string = g_string_insert_c(string, !i ? i++ : ++i, '!');
1472             goto again;
1473         }
1474     }
1475
1476     rc = assuan_send_data(ctx, string->str, string->len);
1477     g_string_free(string, TRUE);
1478     return send_error(ctx, rc);
1479 }
1480
1481 static int list_command(assuan_context_t ctx, char *line)
1482 {
1483     struct client_s *client = assuan_get_pointer(ctx);
1484     gpg_error_t rc;
1485     struct element_list_s *elements = NULL;
1486     gchar *tmp;
1487
1488     if (disable_list_and_dump == TRUE)
1489         return send_error(ctx, GPG_ERR_NOT_IMPLEMENTED);
1490
1491     rc = file_modified(client);
1492
1493     if (rc)
1494         return send_error(ctx, rc);
1495
1496     if (!*line) {
1497         GString *str;
1498
1499         rc = list_accounts(client->doc, &str);
1500
1501         if (rc)
1502             return send_error(ctx, rc);
1503
1504         rc = assuan_send_data(ctx, str->str, str->len);
1505         g_string_free(str, TRUE);
1506         return send_error(ctx, rc);
1507     }
1508
1509     elements = g_malloc0(sizeof(struct element_list_s));
1510
1511     if (!elements) {
1512         rc = gpg_err_code_from_errno(ENOMEM);
1513         goto fail;
1514     }
1515
1516     rc = create_path_list(client->doc, elements, line);
1517
1518     if (rc)
1519         goto fail;
1520
1521     if (elements) {
1522         gint total = g_slist_length(elements->list);
1523         gint i;
1524         GString *str;
1525
1526         if (!total) {
1527             rc = EPWMD_EMPTY_ELEMENT;
1528             goto fail;
1529         }
1530
1531         str = g_string_new(NULL);
1532
1533         if (!str) {
1534             rc = gpg_err_code_from_errno(ENOMEM);
1535             goto fail;
1536         }
1537
1538         for (i = 0; i < total; i++) {
1539             tmp = g_slist_nth_data(elements->list, i);
1540             g_string_append_printf(str, "%s%s", tmp, i+1 == total ? "" : "\n");
1541         }
1542
1543         rc = assuan_send_data(ctx, str->str, str->len);
1544         g_string_free(str, TRUE);
1545     }
1546     else
1547         rc = EPWMD_EMPTY_ELEMENT;
1548
1549 fail:
1550     if (elements) {
1551         gint total = g_slist_length(elements->list);
1552         gint i;
1553
1554         for (i = 0; i < total; i++) {
1555             tmp = g_slist_nth_data(elements->list, i);
1556             g_free(tmp);
1557         }
1558
1559         g_slist_free(elements->list);
1560
1561         if (elements->prefix)
1562             g_free(elements->prefix);
1563
1564         g_free(elements);
1565     }
1566
1567     return send_error(ctx, rc);
1568 }
1569
1570 static gpg_error_t add_attribute(xmlNodePtr node, const gchar *name,
1571         const gchar *value)
1572 {
1573     xmlAttrPtr a;
1574
1575     if ((a = xmlHasProp(node, (xmlChar *)name)) == NULL) {
1576         a = xmlNewProp(node, (xmlChar *)name, (xmlChar *)value);
1577
1578         if (!a)
1579             return EPWMD_LIBXML_ERROR;
1580     }
1581     else
1582         xmlNodeSetContent(a->children, (xmlChar *)value);
1583
1584     return 0;
1585 }
1586
1587 /*
1588  * req[0] - element path
1589  */
1590 static int attribute_list(assuan_context_t ctx, gchar **req)
1591 {
1592     struct client_s *client = assuan_get_pointer(ctx);
1593     gchar **attrlist = NULL;
1594     gint i = 0;
1595     gchar **path = NULL;
1596     xmlAttrPtr a;
1597     xmlNodePtr n, an;
1598     gchar *line;
1599     gpg_error_t rc;
1600
1601     if (!req || !req[0])
1602         return EPWMD_COMMAND_SYNTAX;
1603
1604     if ((path = split_input_line(req[0], "\t", 0)) == NULL) {
1605         /*
1606          * The first argument may be only an account.
1607          */
1608         if ((path = split_input_line(req[0], " ", 0)) == NULL)
1609             return EPWMD_COMMAND_SYNTAX;
1610     }
1611
1612     n = find_account(client->doc, &path, &rc, NULL, 0);
1613
1614     if (!n) {
1615         g_strfreev(path);
1616         return rc;
1617     }
1618
1619     if (path[1]) {
1620         n = find_elements(client->doc, n->children, path+1, &rc,
1621                 NULL, NULL, NULL, FALSE, 0, NULL);
1622
1623         if (!n) {
1624             g_strfreev(path);
1625             return rc;
1626         }
1627     }
1628
1629     g_strfreev(path);
1630
1631     for (a = n->properties; a; a = a->next) {
1632         gchar **pa;
1633
1634         if ((pa = g_realloc(attrlist, (i + 2) * sizeof(gchar *))) == NULL) {
1635             if (attrlist)
1636                 g_strfreev(attrlist);
1637
1638             log_write("%s(%i): %s", __FILE__, __LINE__, strerror(ENOMEM));
1639             return gpg_error_from_errno(ENOMEM);
1640         }
1641
1642         attrlist = pa;
1643         an = a->children;
1644         attrlist[i] = g_strdup_printf("%s %s", (gchar *)a->name, (gchar *)an->content);
1645
1646         if (!attrlist[i]) {
1647             g_strfreev(attrlist);
1648             return gpg_error_from_errno(ENOMEM);
1649         }
1650
1651         attrlist[++i] = NULL;
1652     }
1653
1654     if (!attrlist)
1655         return EPWMD_EMPTY_ELEMENT;
1656
1657     line = g_strjoinv("\n", attrlist);
1658
1659     if (!line) {
1660         g_strfreev(attrlist);
1661         return gpg_error_from_errno(ENOMEM);
1662     }
1663
1664     rc = assuan_send_data(ctx, line, strlen(line));
1665     g_free(line);
1666     g_strfreev(attrlist);
1667     return rc;
1668 }
1669
1670 /*
1671  * req[0] - attribute
1672  * req[1] - element path
1673  */
1674 static int attribute_delete(struct client_s *client, gchar **req)
1675 {
1676     xmlAttrPtr a;
1677     xmlNodePtr n;
1678     gchar **path = NULL;
1679     gpg_error_t rc;
1680
1681     if (!req || !req[0] || !req[1])
1682         return EPWMD_COMMAND_SYNTAX;
1683
1684     if ((path = split_input_line(req[1], "\t", 0)) == NULL) {
1685         /*
1686          * The first argument may be only an account.
1687          */
1688         if ((path = split_input_line(req[1], " ", 0)) == NULL)
1689             return EPWMD_COMMAND_SYNTAX;
1690     }
1691
1692     /*
1693      * Don't remove the "name" attribute for the account element. To remove an
1694      * account use DELETE <account>.
1695      */
1696     if (!path[1] && xmlStrEqual((xmlChar *)req[0], (xmlChar *)"name")) {
1697         rc = EPWMD_ATTR_SYNTAX;
1698         goto fail;
1699     }
1700
1701     n = find_account(client->doc, &path, &rc, NULL, 0);
1702
1703     if (!n)
1704         goto fail;
1705
1706     if (path[1]) {
1707         n = find_elements(client->doc, n->children, path+1, &rc,
1708                 NULL, NULL, NULL, FALSE, 0, NULL);
1709
1710         if (!n)
1711             goto fail;
1712     }
1713
1714     g_strfreev(path);
1715
1716     if ((a = xmlHasProp(n, (xmlChar *)req[0])) == NULL)
1717         return EPWMD_ATTR_NOT_FOUND;
1718
1719     if (xmlRemoveProp(a) == -1)
1720         return EPWMD_LIBXML_ERROR;
1721
1722     return 0;
1723
1724 fail:
1725     g_strfreev(path);
1726     return rc;
1727 }
1728
1729 static xmlNodePtr create_element_path(struct client_s *client, gchar ***path,
1730         gpg_error_t *rc)
1731 {
1732     gchar **src = *path;
1733     gchar **src_orig = g_strdupv(src);
1734     xmlNodePtr n = NULL;
1735
1736     *rc = 0;
1737
1738     if (!src_orig) {
1739         *rc = gpg_error_from_errno(ENOMEM);
1740         goto fail;
1741     }
1742
1743 again:
1744     n = find_account(client->doc, &src, rc, NULL, 0);
1745
1746     if (!n) {
1747         if (*rc == EPWMD_ELEMENT_NOT_FOUND) {
1748             *rc = new_account(client->doc, src[0]);
1749
1750             if (*rc)
1751                 goto fail;
1752
1753             goto again;
1754         }
1755         else
1756             goto fail;
1757     }
1758
1759     if (src[1]) {
1760         if (!n->children)
1761             n = create_target_elements_cb(n, src+1, rc, NULL);
1762         else
1763             n = find_elements(client->doc, n->children, src+1, rc,
1764                     NULL, NULL, create_target_elements_cb, FALSE, 0, NULL);
1765
1766         if (!n)
1767             goto fail;
1768
1769         /*
1770          * Reset the position of the element tree now that the elements
1771          * have been created.
1772          */
1773         g_strfreev(src);
1774         src = src_orig;
1775         src_orig = NULL;
1776         n = find_account(client->doc, &src, rc, NULL, 0);
1777
1778         if (!n)
1779             goto fail;
1780
1781         n = find_elements(client->doc, n->children, src+1, rc,
1782                 NULL, NULL, NULL, FALSE, 0, NULL);
1783
1784         if (!n)
1785             goto fail;
1786     }
1787
1788 fail:
1789     if (src_orig)
1790         g_strfreev(src_orig);
1791
1792     *path = src;
1793     return n;
1794 }
1795
1796 /*
1797  * Creates a "target" attribute. When other commands encounter an element with
1798  * this attribute, the element path is modified to the target value. If the
1799  * source element path doesn't exist when using 'ATTR SET target', it is
1800  * created, but the destination element path must exist.
1801  *
1802  * req[0] - source element path
1803  * req[1] - destination element path
1804  */
1805 static gpg_error_t target_attribute(struct client_s *client, gchar **req)
1806 {
1807     gchar **src, **dst, *line = NULL;
1808     gpg_error_t rc;
1809     xmlNodePtr n;
1810
1811     if (!req || !req[0] || !req[1])
1812         return EPWMD_COMMAND_SYNTAX;
1813
1814     if ((src = split_input_line(req[0], "\t", 0)) == NULL) {
1815         /*
1816          * The first argument may be only an account.
1817          */
1818         if ((src = split_input_line(req[0], " ", 0)) == NULL)
1819             return EPWMD_COMMAND_SYNTAX;
1820     }
1821
1822     if (valid_element_path(src, FALSE) == FALSE) {
1823         g_strfreev(src);
1824         return EPWMD_INVALID_ELEMENT;
1825     }
1826
1827     if ((dst = split_input_line(req[1], "\t", 0)) == NULL) {
1828         /*
1829          * The first argument may be only an account.
1830          */
1831         if ((dst = split_input_line(req[1], " ", 0)) == NULL) {
1832             rc = EPWMD_COMMAND_SYNTAX;
1833             goto fail;
1834         }
1835     }
1836
1837     n = find_account(client->doc, &dst, &rc, NULL, 0);
1838
1839     /*
1840      * Make sure the destination element path exists.
1841      */
1842     if (!n)
1843         goto fail;
1844
1845     if (dst[1]) {
1846         n = find_elements(client->doc, n->children, dst+1, &rc,
1847                 NULL, NULL, NULL, FALSE, 0, NULL);
1848
1849         if (!n)
1850             goto fail;
1851     }
1852
1853     n = create_element_path(client, &src, &rc);
1854
1855     if (rc)
1856         goto fail;
1857
1858     line = g_strjoinv("\t", dst);
1859
1860     if (!line) {
1861         rc = gpg_error_from_errno(ENOMEM);
1862         goto fail;
1863     }
1864
1865     rc = add_attribute(n, "target", line);
1866
1867 fail:
1868     g_free(line);
1869     g_strfreev(src);
1870     g_strfreev(dst);
1871     return rc;
1872 }
1873
1874 /*
1875  * req[0] - account name
1876  * req[1] - new name
1877  */
1878 static gpg_error_t name_attribute(struct client_s *client, gchar **req)
1879 {
1880     gpg_error_t rc;
1881     gchar **tmp;
1882     xmlNodePtr n;
1883
1884     tmp = g_strdupv(req);
1885
1886     if (!tmp)
1887         return gpg_error_from_errno(ENOMEM);
1888
1889     n = find_account(client->doc, &tmp, &rc, NULL, 0);
1890     g_strfreev(tmp);
1891
1892     if (!n)
1893         return rc;
1894
1895     if (g_utf8_collate(req[0], req[1]) == 0)
1896         return 0;
1897
1898     /*
1899      * Will not overwrite an existing account.
1900      */
1901     tmp = g_strdupv(req+1);
1902
1903     if (!tmp)
1904         return gpg_error_from_errno(ENOMEM);
1905
1906     n = find_account(client->doc, &tmp, &rc, NULL, 0);
1907     g_strfreev(tmp);
1908
1909     if (rc && rc != EPWMD_ELEMENT_NOT_FOUND)
1910         return rc;
1911
1912     if (n)
1913         return EPWMD_ACCOUNT_EXISTS;
1914
1915     /*
1916      * Whitespace not allowed in account names.
1917      */
1918     if (contains_whitespace(req[1]) == TRUE)
1919         return EPWMD_ATTR_SYNTAX;
1920
1921     tmp = g_strdupv(req);
1922
1923     if (!tmp)
1924         return gpg_error_from_errno(ENOMEM);
1925
1926     n = find_account(client->doc, &tmp, &rc, NULL, 0);
1927     g_strfreev(tmp);
1928
1929     if (!n)
1930         return EPWMD_ELEMENT_NOT_FOUND;
1931
1932     return add_attribute(n, "name", req[1]);
1933 }
1934
1935 /*
1936  * req[0] - attribute
1937  * req[1] - element path
1938  */
1939 static int attribute_get(assuan_context_t ctx, gchar **req)
1940 {
1941     struct client_s *client = assuan_get_pointer(ctx);
1942     xmlNodePtr n;
1943     xmlChar *a;
1944     gchar **path= NULL;
1945     gpg_error_t rc;
1946
1947     if (!req || !req[0] || !req[1])
1948         return EPWMD_COMMAND_SYNTAX;
1949
1950     if (strchr(req[1], '\t')) {
1951         if ((path = split_input_line(req[1], "\t", 0)) == NULL)
1952             return EPWMD_COMMAND_SYNTAX;
1953     }
1954     else {
1955         if ((path = split_input_line(req[1], " ", 0)) == NULL)
1956             return EPWMD_COMMAND_SYNTAX;
1957     }
1958
1959     n = find_account(client->doc, &path, &rc, NULL, 0);
1960
1961     if (!n)
1962         goto fail;
1963
1964     if (path[1]) {
1965         n = find_elements(client->doc, n->children, path+1, &rc,
1966                 NULL, NULL, NULL, FALSE, 0, NULL);
1967
1968         if (!n)
1969             goto fail;
1970     }
1971
1972     g_strfreev(path);
1973
1974     if ((a = xmlGetProp(n, (xmlChar *)req[0])) == NULL)
1975         return EPWMD_ATTR_NOT_FOUND;
1976
1977     rc = assuan_send_data(ctx, a, xmlStrlen(a));
1978     xmlFree(a);
1979     return rc;
1980
1981 fail:
1982     g_strfreev(path);
1983     return rc;
1984 }
1985
1986 /*
1987  * req[0] - attribute
1988  * req[1] - element path
1989  * req[2] - value
1990  */
1991 static int attribute_set(struct client_s *client, gchar **req)
1992 {
1993     gchar **path = NULL;
1994     gpg_error_t rc;
1995     xmlNodePtr n;
1996
1997     if (!req || !req[0] || !req[1] || !req[2])
1998         return EPWMD_COMMAND_SYNTAX;
1999
2000     /*
2001      * Reserved attribute names.
2002      */
2003     if (g_utf8_collate(req[0], "name") == 0) {
2004         /*
2005          * Only reserved for the account element. Not the rest of the
2006          * document.
2007          */
2008         if (strchr(req[1], '\t') == NULL)
2009             return name_attribute(client, req + 1);
2010     }
2011     else if (g_utf8_collate(req[0], "target") == 0)
2012         return target_attribute(client, req + 1);
2013
2014     if ((path = split_input_line(req[1], "\t", 0)) == NULL) {
2015         /*
2016          * The first argument may be only an account.
2017          */
2018         if ((path = split_input_line(req[1], " ", 0)) == NULL)
2019             return EPWMD_COMMAND_SYNTAX;
2020     }
2021
2022     n = find_account(client->doc, &path, &rc, NULL, 0);
2023
2024     if (!n)
2025         goto fail;
2026
2027     if (path[1]) {
2028         n = find_elements(client->doc, n->children, path+1, &rc,
2029                 NULL, NULL, NULL, FALSE, 0, NULL);
2030
2031         if (!n)
2032             goto fail;
2033     }
2034
2035     g_strfreev(path);
2036     return add_attribute(n, req[0], req[2]);
2037
2038 fail:
2039     g_strfreev(path);
2040     return rc;
2041 }
2042
2043 /*
2044  * req[0] - command
2045  * req[1] - attribute name or element path if command is LIST
2046  * req[2] - element path
2047  * req[2] - element path or value
2048  */
2049 static int attr_command(assuan_context_t ctx, char *line)
2050 {
2051     struct client_s *client = assuan_get_pointer(ctx);
2052     gchar **req;
2053     gpg_error_t rc = 0;
2054
2055     rc = file_modified(client);
2056
2057     if (rc)
2058         return send_error(ctx, rc);
2059
2060     req = split_input_line(line, " ", 4);
2061
2062     if (!req || !req[0] || !req[1]) {
2063         g_strfreev(req);
2064         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
2065     }
2066
2067     if (g_ascii_strcasecmp(req[0], "SET") == 0)
2068         rc = attribute_set(client, req+1);
2069     else if (g_ascii_strcasecmp(req[0], "GET") == 0)
2070         rc = attribute_get(ctx, req+1);
2071     else if (g_ascii_strcasecmp(req[0], "DELETE") == 0)
2072         rc = attribute_delete(client, req+1);
2073     else if (g_ascii_strcasecmp(req[0], "LIST") == 0)
2074         rc = attribute_list(ctx, req+1);
2075     else
2076         rc = EPWMD_COMMAND_SYNTAX;
2077
2078     g_strfreev(req);
2079     return send_error(ctx, rc);
2080 }
2081
2082 static int iscached_command(assuan_context_t ctx, char *line)
2083 {
2084     gchar **req = split_input_line(line, " ", 0);
2085     guchar md5file[16];
2086
2087     if (!req || !*req) {
2088         g_strfreev(req);
2089         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
2090     }
2091
2092     gcry_md_hash_buffer(GCRY_MD_MD5, md5file, req[0], strlen(req[0]));
2093     g_strfreev(req);
2094     CACHE_LOCK(ctx);
2095
2096     if (cache_iscached(md5file) == FALSE) {
2097         CACHE_UNLOCK;
2098         return send_error(ctx, EPWMD_CACHE_NOT_FOUND);
2099     }
2100
2101     CACHE_UNLOCK;
2102     return send_error(ctx, 0);
2103 }
2104
2105 static int clearcache_command(assuan_context_t ctx, char *line)
2106 {
2107     struct client_s *client = assuan_get_pointer(ctx);
2108     gchar **req = split_input_line(line, " ", 0);
2109     guchar md5file[16];
2110
2111     CACHE_LOCK(ctx);
2112
2113     if (!req || !*req) {
2114         g_strfreev(req);
2115         cache_clear(client->md5file, 2);
2116         CACHE_UNLOCK;
2117         return send_error(ctx, 0);
2118     }
2119
2120     gcry_md_hash_buffer(GCRY_MD_MD5, md5file, req[0], strlen(req[0]));
2121     g_strfreev(req);
2122
2123     if (cache_clear(md5file, 1) == FALSE) {
2124         CACHE_UNLOCK;
2125         return send_error(ctx, EPWMD_CACHE_NOT_FOUND);
2126     }
2127
2128     CACHE_UNLOCK;
2129     return send_error(ctx, 0);
2130 }
2131
2132 static int cachetimeout_command(assuan_context_t ctx, char *line)
2133 {
2134     guchar md5file[16];
2135     glong timeout;
2136     gchar **req = split_input_line(line, " ", 0);
2137     gchar *p;
2138     struct client_s *client = assuan_get_pointer(ctx);
2139
2140     if (!req || !*req || !req[1]) {
2141         g_strfreev(req);
2142         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
2143     }
2144
2145     errno = 0;
2146     timeout = strtol(req[0], &p, 10);
2147
2148     if (errno != 0 || *p != 0) {
2149         g_strfreev(req);
2150         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
2151     }
2152
2153     gcry_md_hash_buffer(GCRY_MD_MD5, md5file, req[1], strlen(req[1]));
2154     g_strfreev(req);
2155     CACHE_LOCK(client->ctx);
2156
2157     if (cache_set_timeout(md5file, timeout) == FALSE) {
2158         CACHE_UNLOCK;
2159         return send_error(ctx, EPWMD_CACHE_NOT_FOUND);
2160     }
2161
2162     CACHE_UNLOCK;
2163     return send_error(ctx, 0);
2164 }
2165
2166 static int dump_command(assuan_context_t ctx, char *line)
2167 {
2168     xmlChar *xml;
2169     gssize len;
2170     struct client_s *client = assuan_get_pointer(ctx);
2171     gpg_error_t rc;
2172
2173     if (disable_list_and_dump == TRUE)
2174         return send_error(ctx, GPG_ERR_NOT_IMPLEMENTED);
2175
2176     rc = file_modified(client);
2177
2178     if (rc)
2179         return send_error(ctx, rc);
2180
2181     xmlDocDumpFormatMemory(client->doc, &xml, &len, 1);
2182
2183     if (!xml)
2184         return send_syserror(ctx, ENOMEM);
2185
2186     rc = assuan_send_data(ctx, xml, len);
2187     xmlFree(xml);
2188     return send_error(ctx, rc);
2189 }
2190
2191 static int getconfig_command(assuan_context_t ctx, gchar *line)
2192 {
2193     struct client_s *client = assuan_get_pointer(ctx);
2194     gpg_error_t rc = 0;
2195     gchar filename[255]={0}, param[747]={0};
2196     gchar *p, *tmp, *fp = client->filename, *paramp = line;
2197
2198     if (strchr(line, ' ')) {
2199         sscanf(line, " %254[a-zA-Z] %746c", filename, param);
2200         fp = filename;
2201         paramp = param;
2202     }
2203
2204     paramp = g_ascii_strdown(paramp, -1);
2205
2206     if (!paramp)
2207         return send_syserror(ctx, ENOMEM);
2208
2209     p = get_key_file_string(fp ? fp : "global", paramp);
2210     g_free(paramp);
2211
2212     if (!p)
2213         return send_error(ctx, GPG_ERR_NO_VALUE);
2214
2215     tmp = expand_homedir(p);
2216
2217     if (!tmp) {
2218         g_free(p);
2219         return send_syserror(ctx, ENOMEM);
2220     }
2221
2222     g_free(p);
2223     p = tmp;
2224     rc = assuan_send_data(ctx, p, strlen(p));
2225     g_free(p);
2226     return send_error(ctx, rc);
2227 }
2228
2229 static int xpath_command(assuan_context_t ctx, gchar *line)
2230 {
2231     struct client_s *client = assuan_get_pointer(ctx);
2232     gpg_error_t rc;
2233     xmlXPathContextPtr xp;
2234     xmlXPathObjectPtr result;
2235     xmlBufferPtr buf = NULL;
2236     gchar **req = NULL;
2237
2238     if (disable_list_and_dump == TRUE)
2239         return send_error(ctx, GPG_ERR_NOT_IMPLEMENTED);
2240
2241     rc = file_modified(client);
2242
2243     if (rc)
2244         return send_error(ctx, rc);
2245
2246     if (!line || !*line)
2247         return send_error(ctx, EPWMD_COMMAND_SYNTAX);
2248
2249     if ((req = split_input_line(line, "\t", 2)) == NULL) {
2250         if (strv_printf(&req, "%s", line) == FALSE)
2251             return send_syserror(ctx, ENOMEM);
2252     }
2253
2254     xp = xmlXPathNewContext(client->doc);
2255
2256     if (!xp)
2257         return send_error(ctx, EPWMD_LIBXML_ERROR);
2258
2259     result = xmlXPathEvalExpression((xmlChar *)req[0], xp);
2260
2261     if (!result) {
2262         xmlXPathFreeContext(xp);
2263         return send_error(ctx, EPWMD_LIBXML_ERROR);
2264     }
2265
2266     if (xmlXPathNodeSetIsEmpty(result->nodesetval)) {
2267         rc = EPWMD_EMPTY_ELEMENT;
2268         goto fail;
2269     }
2270
2271     rc = recurse_xpath_nodeset(client->doc, result->nodesetval,
2272             (xmlChar *)req[1], &buf);
2273
2274     if (rc)
2275         goto fail;
2276     else if (!req[1] && !xmlBufferLength(buf)) {
2277         rc = EPWMD_EMPTY_ELEMENT;
2278         goto fail;
2279     }
2280     else if (req[1])
2281         goto fail;
2282
2283     rc = assuan_send_data(ctx, xmlBufferContent(buf), xmlBufferLength(buf));
2284
2285 fail:
2286     g_strfreev(req);
2287
2288     if (buf)
2289         xmlBufferFree(buf);
2290
2291     if (result)
2292         xmlXPathFreeObject(result);
2293
2294     if (xp)
2295         xmlXPathFreeContext(xp);
2296
2297     return send_error(ctx, rc);
2298 }
2299
2300 static int import_command_finalize(gpointer data, gint assuan_rc, guchar *line,
2301         gsize len)
2302 {
2303     struct client_s *client = assuan_get_pointer((assuan_context_t)data);
2304     gpg_error_t rc = file_modified(client);
2305     gchar **req, **path = NULL, **path_orig = NULL, *content;
2306     xmlDocPtr doc;
2307     xmlNodePtr n, root, copy;
2308
2309     if (assuan_rc || rc) {
2310         if (line)
2311 #ifndef MEM_DEBUG
2312             xfree(line);
2313 #else
2314             free(line);
2315 #endif
2316         return assuan_rc ? assuan_rc : rc;
2317     }
2318
2319     req = split_input_line((gchar *)line, " ", 2);
2320 #ifndef MEM_DEBUG
2321     xfree(line);
2322 #else
2323     free(line);
2324 #endif
2325
2326     if (!req || !*req)
2327         return EPWMD_COMMAND_SYNTAX;
2328
2329     if ((path = split_input_line(req[0], "\t", 0)) == NULL) {
2330         if ((path = split_input_line(req[0], " ", 0)) == NULL)
2331             return EPWMD_COMMAND_SYNTAX;
2332     }
2333
2334     content = req[1];
2335
2336     if (!content || !*content) {
2337         rc = EPWMD_COMMAND_SYNTAX;
2338         goto fail;
2339     }
2340
2341     if (valid_xml_element((xmlChar *)*path) == FALSE) {
2342         rc = EPWMD_INVALID_ELEMENT;
2343         goto fail;
2344     }
2345
2346     if (valid_element_path(path+1, FALSE) == FALSE) {
2347         rc = EPWMD_INVALID_ELEMENT;
2348         goto fail;
2349     }
2350
2351     doc = xmlReadDoc((xmlChar *)content, NULL, "UTF-8", XML_PARSE_NOBLANKS);
2352
2353     if (!doc) {
2354         rc = EPWMD_LIBXML_ERROR;
2355         goto fail;
2356     }
2357
2358     root = xmlDocGetRootElement(doc);
2359     path_orig = g_strdupv(path);
2360
2361     if (!path_orig) {
2362         xmlFreeDoc(doc);
2363         rc = gpg_error_from_errno(ENOMEM);
2364         goto fail;
2365     }
2366
2367     if (strv_printf(&path, "%s", (gchar *)root->name) == FALSE) {
2368         g_strfreev(path_orig);
2369         xmlFreeDoc(doc);
2370         rc = gpg_error_from_errno(ENOMEM);
2371         goto fail;
2372     }
2373
2374     n = find_account(client->doc, &path, &rc, NULL, 0);
2375
2376     if (rc && rc != EPWMD_ELEMENT_NOT_FOUND) {
2377         g_strfreev(path_orig);
2378         xmlFreeDoc(doc);
2379         goto fail;
2380     }
2381     else if (!rc) {
2382         n = find_elements(client->doc, n->children, path+1, &rc, NULL, NULL, NULL, FALSE, 0, NULL);
2383
2384         if (rc && rc != EPWMD_ELEMENT_NOT_FOUND) {
2385             g_strfreev(path_orig);
2386             xmlFreeDoc(doc);
2387             goto fail;
2388         }
2389         else if (!rc) {
2390             xmlNodePtr parent = n->parent;
2391
2392             xmlUnlinkNode(n);
2393             xmlFreeNode(n);
2394             n = parent;
2395         }
2396     }
2397
2398     g_strfreev(path);
2399     path = path_orig;
2400
2401     if (rc == EPWMD_ELEMENT_NOT_FOUND) {
2402         n = create_element_path(client, &path, &rc);
2403
2404         if (rc) {
2405             xmlFreeDoc(doc);
2406             goto fail;
2407         }
2408     }
2409
2410     copy = xmlCopyNode(root, 1);
2411     n = xmlAddChild(n, copy);
2412     xmlFreeDoc(doc);
2413
2414     if (!n)
2415         rc = EPWMD_LIBXML_ERROR;
2416
2417 fail:
2418     g_strfreev(path);
2419     g_strfreev(req);
2420     client->inquire_status = INQUIRE_DONE;
2421     return rc;
2422 }
2423
2424 static int import_command(assuan_context_t ctx, gchar *line)
2425 {
2426     gpg_error_t rc;
2427     struct client_s *client = assuan_get_pointer(ctx);
2428
2429     rc = file_modified(client);
2430
2431     if (rc)
2432         return send_error(ctx, rc);
2433
2434     rc = assuan_inquire_ext(ctx, "IMPORT", 0, import_command_finalize, ctx);
2435
2436     if (rc)
2437         return send_error(ctx, rc);
2438
2439     /* Don't return with assuan_process_done() here. This is an INQUIRE. */
2440     client->inquire_status = INQUIRE_BUSY;
2441     return 0;
2442 }
2443
2444 static int lock_command(assuan_context_t ctx, gchar *line)
2445 {
2446     gpg_error_t rc;
2447     struct client_s *client = assuan_get_pointer(ctx);
2448
2449     rc = file_modified(client);
2450
2451     if (rc)
2452         return send_error(ctx, rc);
2453
2454     rc = lock_file_mutex(client);
2455
2456     if (!rc)
2457         client->is_lock_cmd = TRUE;
2458
2459     return send_error(ctx, rc);
2460 }
2461
2462 static int unlock_command(assuan_context_t ctx, gchar *line)
2463 {
2464     struct client_s *client = assuan_get_pointer(ctx);
2465     gpg_error_t rc = file_modified(client);
2466
2467     if (rc)
2468         return send_error(ctx, rc);
2469
2470     unlock_file_mutex(client);
2471     return send_error(ctx, 0);
2472 }
2473
2474 static int getpid_command(assuan_context_t ctx, gchar *line)
2475 {
2476     gpg_error_t rc;
2477     gchar buf[32];
2478     pid_t pid = getpid();
2479
2480     print_fmt(buf, sizeof(buf), "%i", pid);
2481     rc = assuan_send_data(ctx, buf, strlen(buf));
2482     return send_error(ctx, rc);
2483 }
2484
2485 void cleanup_assuan(assuan_context_t ctx)
2486 {
2487     struct client_s *cl = assuan_get_pointer(ctx);
2488
2489     if (cl)
2490         cleanup_client(cl);
2491 }
2492
2493 static void reset_notify(assuan_context_t ctx)
2494 {
2495     struct client_s *cl = assuan_get_pointer(ctx);
2496
2497     if (cl)
2498         cleanup_client(cl);
2499 }
2500
2501 static gpg_error_t parse_client_option(assuan_context_t ctx, const gchar *line)
2502 {
2503     gchar name[32] = {0}, value[256] = {0};
2504
2505     if (sscanf(line, " %31[a-zA-Z] = %255c", name, value) != 2)
2506         return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_SYNTAX);
2507
2508     if (g_strcasecmp(name, (gchar *)"NAME") == 0) {
2509         pth_attr_t attr = pth_attr_of(pth_self());
2510
2511         log_write("OPTION CLIENT %s", line);
2512         pth_attr_set(attr, PTH_ATTR_NAME, value);
2513         pth_attr_destroy(attr);
2514     }
2515     else
2516         return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_UNKNOWN_OPTION);
2517
2518     return 0;
2519 }
2520
2521 static int option_handler(assuan_context_t ctx, const gchar *name,
2522         const gchar *value)
2523 {
2524     struct client_s *client = assuan_get_pointer(ctx);
2525
2526     if (!value || !*value)
2527         return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_INV_VALUE);
2528
2529     if (g_strcasecmp(name, (gchar *)"client") == 0)
2530         return parse_client_option(ctx, value);
2531
2532     if (g_strcasecmp(name, (gchar *)"iterations") == 0) {
2533         long n;
2534         gchar *p = NULL;
2535
2536         errno = 0;
2537         n = strtol(value, &p, 10);
2538
2539         if (errno || (p && *p) || n < -1)
2540             return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_INV_VALUE);
2541
2542         g_key_file_set_integer(keyfileh, client->filename ? client->filename : "global", "iterations", n);
2543         send_status_all(STATUS_CONFIG);
2544     }
2545 #ifdef WITH_PINENTRY
2546     else if (g_strcasecmp(name, (gchar *)"ttyname") == 0) {
2547         g_free(client->pinentry->ttyname);
2548         client->pinentry->ttyname = g_strdup(value);
2549     }
2550     else if (g_strcasecmp(name, (gchar *)"ttytype") == 0) {
2551         g_free(client->pinentry->ttytype);
2552         client->pinentry->ttytype = g_strdup(value);
2553     }
2554     else if (g_strcasecmp(name, (gchar *)"display") == 0) {
2555         g_free(client->pinentry->display);
2556         client->pinentry->display = g_strdup(value);
2557     }
2558     else if (g_strcasecmp(name, (gchar *)"path") == 0) {
2559         g_free(client->pinentry->path);
2560         client->pinentry->path = g_strdup(value);
2561     }
2562     else if (g_strcasecmp(name, (gchar *)"title") == 0) {
2563         g_free(client->pinentry->title);
2564         client->pinentry->title = g_strdup(value);
2565     }
2566     else if (g_strcasecmp(name, (gchar *)"prompt") == 0) {
2567         g_free(client->pinentry->prompt);
2568         client->pinentry->prompt = g_strdup(value);
2569     }
2570     else if (g_strcasecmp(name, (gchar *)"desc") == 0) {
2571         g_free(client->pinentry->desc);
2572         client->pinentry->desc = g_strdup(value);
2573     }
2574     /*
2575      * Look at client_thread() to see how this works.
2576      */
2577     else if (g_strcasecmp(name, (gchar *)"timeout") == 0) {
2578         gchar *p = NULL;
2579         gint n = strtol(value, &p, 10);
2580
2581         if (*p || n < 0)
2582             return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_INV_VALUE);
2583
2584         client->pinentry->timeout = n;
2585     }
2586     else if (g_strcasecmp(name, (gchar *)"pinentry") == 0) {
2587         gchar *p = NULL;
2588         gint n = strtol(value, &p, 10);
2589
2590         if (*p || n < 0 || n > 1)
2591             return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_INV_VALUE);
2592
2593         client->pinentry->enable = n == 0 ? FALSE : TRUE;
2594     }
2595 #endif
2596     else
2597         return gpg_err_make(PWMD_ERR_SOURCE, GPG_ERR_UNKNOWN_OPTION);
2598
2599     log_write("OPTION %s=%s", name, value);
2600     return 0;
2601 }
2602
2603 gpg_error_t register_commands(assuan_context_t ctx)
2604 {
2605     static struct {
2606         const gchar *name;
2607         gint (*handler)(assuan_context_t, gchar *line);
2608     } table[] = {
2609         { "OPEN", open_command },
2610         { "SAVE", save_command },
2611         { "LIST", list_command },
2612         { "REALPATH", realpath_command },
2613         { "STORE", store_command },
2614         { "DELETE", delete_command },
2615         { "GET", get_command },
2616         { "ATTR", attr_command },
2617         { "ISCACHED", iscached_command },
2618         { "CLEARCACHE", clearcache_command },
2619         { "CACHETIMEOUT", cachetimeout_command },
2620         { "GETCONFIG", getconfig_command },
2621         { "DUMP", dump_command },
2622         { "XPATH", xpath_command },
2623         { "IMPORT", import_command },
2624         { "LOCK", lock_command },
2625         { "UNLOCK", unlock_command },
2626         { "GETPID", getpid_command },
2627         { "INPUT", NULL },
2628         { "OUTPUT", NULL },
2629         { NULL, NULL }
2630     };
2631     gint i, rc;
2632
2633     for (i=0; table[i].name; i++) {
2634         rc = assuan_register_command (ctx, table[i].name, table[i].handler);
2635
2636         if (rc)
2637             return rc;
2638     }
2639
2640     rc = assuan_register_bye_notify(ctx, cleanup_assuan);
2641
2642     if (rc)
2643         return rc;
2644
2645     rc = assuan_register_option_handler(ctx, option_handler);
2646
2647     if (rc)
2648         return rc;
2649
2650     rc = assuan_register_reset_notify(ctx, reset_notify);
2651
2652     if (rc)
2653         return rc;
2654
2655     return assuan_register_post_cmd_notify(ctx, command_finalize);
2656 }
2657
2658 gpg_error_t try_xml_decrypt(assuan_context_t ctx, gint fd, struct stat st,
2659         guchar *key, gint *dst_iter)
2660 {
2661     guchar *iv;
2662     void *inbuf;
2663     gsize insize, len;
2664     guchar tkey[gcrykeysize];
2665     struct client_s *client = ctx ? assuan_get_pointer(ctx) : NULL;
2666     gcry_cipher_hd_t gh;
2667     guint iter = 0, n_iter = 0;
2668     gint iter_progress = 0;
2669     void *outbuf = NULL;
2670     gint zrc = 0;
2671     glong outsize = 0;
2672     gpg_error_t rc;
2673     file_header_t file_header;
2674     gchar str[ASSUAN_LINELENGTH];
2675
2676     if (!ctx) {
2677         rc = gcry_cipher_open(&gh, GCRY_CIPHER_AES256, GCRY_CIPHER_MODE_CBC, 0);
2678
2679         if (rc)
2680             return rc;
2681     }
2682     else
2683         gh = client->gh;
2684
2685     lseek(fd, 0, SEEK_SET);
2686     insize = st.st_size - sizeof(file_header_t);
2687     iv = gcry_malloc(gcryblocksize);
2688
2689     if (!iv) {
2690         if (!ctx)
2691             gcry_cipher_close(gh);
2692
2693         return gpg_error_from_errno(ENOMEM);
2694     }
2695
2696     len = pth_read(fd, &file_header, sizeof(file_header_t));
2697
2698     if (len != sizeof(file_header_t)) {
2699         len = errno;
2700
2701         if (!ctx)
2702             gcry_cipher_close(gh);
2703
2704         gcry_free(iv);
2705         errno = len;
2706         return gpg_error_from_errno(errno);
2707     }
2708
2709     *dst_iter = file_header.iter;
2710
2711     /* No encryption iterations. This is a plain (gzipped) file. */
2712     if (file_header.iter == -1) {
2713         /*
2714          * cache_file_count() needs both .used == TRUE and a valid key in
2715          * order for it to count as a used cache entry. Fixes CACHE status
2716          * messages.
2717          */
2718         memset(key, '!', gcrykeysize);
2719         insize = st.st_size - sizeof(file_header_t);
2720     }
2721
2722     memcpy(iv, &file_header.iv, sizeof(file_header.iv));
2723     inbuf = gcry_malloc(insize);
2724
2725     if (!inbuf) {
2726         if (!ctx)
2727             gcry_cipher_close(gh);
2728
2729         gcry_free(iv);
2730         return gpg_error_from_errno(ENOMEM);
2731     }
2732
2733     len = pth_read(fd, inbuf, insize);
2734
2735     if (len != insize) {
2736         len = errno;
2737
2738         if (!ctx)
2739             gcry_cipher_close(gh);
2740
2741         gcry_free(iv);
2742         errno = len;
2743         return gpg_error_from_errno(errno);
2744     }
2745
2746     if (file_header.iter == -1)
2747         goto decompress;
2748
2749     if ((rc = gcry_cipher_setiv(gh, iv, gcryblocksize))) {
2750         if (!ctx) {
2751             gcry_cipher_close(gh);
2752             warnx("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2753         }
2754         else
2755             log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2756
2757         gcry_free(inbuf);
2758         gcry_free(iv);
2759         return rc;
2760     }
2761
2762     if ((rc = gcry_cipher_setkey(gh, key, gcrykeysize))) {
2763         if (!ctx) {
2764             gcry_cipher_close(gh);
2765             warnx("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2766         }
2767         else
2768             log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2769
2770         gcry_free(inbuf);
2771         gcry_free(iv);
2772
2773         if (!ctx)
2774             gcry_cipher_close(gh);
2775
2776         return rc;
2777     }
2778
2779     if (client && client->filename)
2780         iter_progress = get_key_file_integer(client->filename, "iteration_progress");
2781
2782     if (ctx && iter_progress > 0 && file_header.iter >= iter_progress) {
2783         rc = assuan_write_status(client->ctx, "DECRYPT",
2784                 print_fmt(str, sizeof(str), "%i %i", 0, file_header.iter));
2785
2786         if (rc) {
2787             gcry_free(inbuf);
2788             gcry_free(iv);
2789             return rc;
2790         }
2791     }
2792
2793     rc = decrypt_xml(gh, inbuf, insize, NULL, 0);
2794
2795     if (rc) {
2796         gcry_free(inbuf);
2797         gcry_free(iv);
2798         return rc;
2799     }
2800
2801     memcpy(tkey, key, sizeof(tkey));
2802     tkey[0] ^= 1;
2803
2804     if ((rc = gcry_cipher_setkey(gh, tkey, gcrykeysize))) {
2805         if (!ctx) {
2806             gcry_cipher_close(gh);
2807             warnx("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2808         }
2809         else
2810             log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2811
2812         memset(tkey, 0, sizeof(tkey));
2813         gcry_free(inbuf);
2814         gcry_free(iv);
2815         return rc;
2816     }
2817
2818     memset(tkey, 0, sizeof(tkey));
2819
2820     while (iter < file_header.iter) {
2821         if (ctx && iter_progress > 0 && iter >= iter_progress) {
2822             if (!(iter % iter_progress)) {
2823                 rc = assuan_write_status(ctx, "DECRYPT",
2824                         print_fmt(str, sizeof(str), "%i %i", ++n_iter * iter_progress, file_header.iter));
2825
2826                 if (rc) {
2827                     gcry_free(inbuf);
2828                     gcry_free(iv);
2829                     return rc;
2830                 }
2831             }
2832         }
2833
2834         if ((rc = gcry_cipher_setiv(gh, iv, gcryblocksize))) {
2835             if (!ctx) {
2836                 gcry_cipher_close(gh);
2837                 warnx("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2838             }
2839             else
2840                 log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2841
2842             gcry_free(inbuf);
2843             gcry_free(iv);
2844             return rc;
2845         }
2846
2847         rc = decrypt_xml(gh, inbuf, insize, NULL, 0);
2848
2849         if (rc) {
2850             if (!ctx) {
2851                 gcry_cipher_close(gh);
2852                 warnx("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2853             }
2854             else
2855                 log_write("%s(%i): %s", __FUNCTION__, __LINE__, gcry_strerror(rc));
2856
2857             gcry_free(inbuf);
2858             gcry_free(iv);
2859             return rc;
2860         }
2861
2862         iter++;
2863         pth_yield(NULL);
2864     }
2865
2866     if (ctx && iter_progress && file_header.iter >= iter_progress) {
2867         rc = assuan_write_status(ctx, "DECRYPT",
2868                 print_fmt(str, sizeof(str), "%i %i", file_header.iter, file_header.iter));
2869
2870         if (rc) {
2871             gcry_free(inbuf);
2872             gcry_free(iv);
2873             return rc;
2874         }
2875     }
2876
2877     gcry_free(iv);
2878
2879 decompress:
2880     if (do_decompress(ctx, inbuf, insize, &outbuf, &outsize, &zrc) == FALSE) {
2881         /*
2882          * Z_DATA_ERROR may be returned if this file hasn't been compressed yet.
2883          */
2884         if (zrc == Z_MEM_ERROR) {
2885             gcry_free(inbuf);
2886             return gpg_error_from_errno(ENOMEM);
2887         }
2888         else if (zrc != Z_DATA_ERROR) {
2889             gcry_free(inbuf);
2890
2891             if (!ctx)
2892                 gcry_cipher_close(gh);
2893
2894             return EPWMD_BADKEY;
2895         }
2896     }
2897     else {
2898         gcry_free(inbuf);
2899         inbuf = outbuf;
2900         insize = outsize;
2901     }
2902
2903     if (g_strncasecmp(inbuf, "<?xml version=\"1.0\"?>", 21) != 0) {
2904         gcry_free(inbuf);
2905
2906         if (!ctx)
2907             gcry_cipher_close(gh);
2908
2909         return EPWMD_BADKEY;
2910     }
2911
2912     if (ctx) {
2913         client->xml = inbuf;
2914         client->len = insize;
2915     }
2916     else {
2917         gcry_cipher_close(gh);
2918         gcry_free(inbuf);
2919     }
2920
2921     return 0;
2922 }
2923
2924 /*
2925  * This is called after every Assuan command.
2926  */
2927 void command_finalize(assuan_context_t ctx, gint rc)
2928 {
2929     struct client_s *client = assuan_get_pointer(ctx);
2930
2931     if (!client->is_lock_cmd)
2932         unlock_file_mutex(client);
2933 }