keydb_fs.c

   1 /*
   2  * keydb_fs.c - Routines to store and fetch keys in a filesystem hierarchy.
   3  *
   4  * Copyright 2004 Daniel Silverstone <dsilvers@digital-scurf.org>
   5  *
   6  * This program is free software: you can redistribute it and/or modify it
   7  * under the terms of the GNU General Public License as published by the Free
   8  * Software Foundation; version 2 of the License.
   9  *
  10  * This program is distributed in the hope that it will be useful, but WITHOUT
  11  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  13  * more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along with
  16  * this program; if not, write to the Free Software Foundation, Inc., 51
  17  * Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18  */
  19
  20 #include <sys/types.h>
  21 #include <sys/uio.h>
  22 #include <sys/stat.h>
  23 #include <errno.h>
  24 #include <fcntl.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <unistd.h>
  29 #include <limits.h>
  30 #include <dirent.h>
  31
  32 #include "charfuncs.h"
  33 #include "decodekey.h"
  34 #include "keydb.h"
  35 #include "keyid.h"
  36 #include "keystructs.h"
  37 #include "ll.h"
  38 #include "mem.h"
  39 #include "onak-conf.h"
  40 #include "parsekey.h"
  41 #include "log.h"
  42 #include "wordlist.h"
  43
  44 /* Hack: We should really dynamically allocate our path buffers */
  45 #ifndef PATH_MAX
  46 #define PATH_MAX 1024
  47 #endif
  48
  49 static int keydb_lockfile_fd = -1;
  50 static bool keydb_lockfile_readonly;
  51
  52 /*****************************************************************************/
  53
  54 /* Helper functions */
  55
  56 #define FNV_offset_basis 2166136261ul
  57 #define FNV_mixing_prime 16777619ul
  58
  59 static uint32_t calchash(uint8_t * ptr)
  60 {
  61         register uint32_t h = FNV_offset_basis;
  62         register uint32_t p = FNV_mixing_prime;
  63         register uint32_t n = strlen((char *) ptr);
  64         register uint8_t *c = ptr;
  65         while (n--) {
  66                 h *= p;
  67                 h ^= *c++;
  68         }
  69         return h ? h : 1;       /* prevent a hash of zero happening */
  70 }
  71
  72
  73 static void keypath(char *buffer, size_t length, uint64_t _keyid)
  74 {
  75         uint64_t keyid = _keyid << 32;
  76         snprintf(buffer, length, "%s/key/%02X/%02X/%08X/%016" PRIX64,
  77                  config.db_dir, (uint8_t) ((keyid >> 56) & 0xFF),
  78                  (uint8_t) ((keyid >> 48) & 0xFF),
  79                  (uint32_t) (keyid >> 32), _keyid);
  80 }
  81
  82 static void keydir(char *buffer, size_t length, uint64_t _keyid)
  83 {
  84         uint64_t keyid = _keyid << 32;
  85         snprintf(buffer, length, "%s/key/%02X/%02X/%08X", config.db_dir,
  86                  (uint8_t) ((keyid >> 56) & 0xFF),
  87                  (uint8_t) ((keyid >> 48) & 0xFF),
  88                  (uint32_t) (keyid >> 32));
  89 }
  90
  91 static void prove_path_to(uint64_t keyid, char *what)
  92 {
  93         static char buffer[PATH_MAX];
  94         snprintf(buffer, sizeof(buffer), "%s/%s", config.db_dir, what);
  95         mkdir(buffer, 0777);
  96
  97         snprintf(buffer, sizeof(buffer), "%s/%s/%02X", config.db_dir, what,
  98                  (uint8_t) ((keyid >> 24) & 0xFF));
  99         mkdir(buffer, 0777);
 100
 101         snprintf(buffer, sizeof(buffer), "%s/%s/%02X/%02X", config.db_dir,
 102                  what,
 103                  (uint8_t) ((keyid >> 24) & 0xFF),
 104                  (uint8_t) ((keyid >> 16) & 0xFF));
 105         mkdir(buffer, 0777);
 106
 107         snprintf(buffer, sizeof(buffer), "%s/%s/%02X/%02X/%08X", config.db_dir,
 108                  what,
 109                  (uint8_t) ((keyid >> 24) & 0xFF),
 110                  (uint8_t) ((keyid >> 16) & 0xFF), (uint32_t) (keyid));
 111         mkdir(buffer, 0777);
 112 }
 113
 114 static void wordpath(char *buffer, size_t length, char *word, uint32_t hash,
 115                 uint64_t keyid)
 116 {
 117         snprintf(buffer, length, "%s/words/%02X/%02X/%08X/%s/%016" PRIX64,
 118                  config.db_dir, (uint8_t) ((hash >> 24) & 0xFF),
 119                  (uint8_t) ((hash >> 16) & 0xFF), hash, word, keyid);
 120 }
 121
 122 static void worddir(char *buffer, size_t length, char *word, uint32_t hash)
 123 {
 124         snprintf(buffer, length, "%s/words/%02X/%02X/%08X/%s", config.db_dir,
 125                  (uint8_t) ((hash >> 24) & 0xFF),
 126                  (uint8_t) ((hash >> 16) & 0xFF), hash, word);
 127 }
 128
 129 static void subkeypath(char *buffer, size_t length, uint64_t subkey,
 130                 uint64_t keyid)
 131 {
 132         snprintf(buffer, length, "%s/subkeys/%02X/%02X/%08X/%016" PRIX64,
 133                  config.db_dir,
 134                  (uint8_t) ((subkey >> 24) & 0xFF),
 135                  (uint8_t) ((subkey >> 16) & 0xFF),
 136                  (uint32_t) (subkey & 0xFFFFFFFF),
 137                  keyid);
 138 }
 139
 140 static void skshashpath(char *buffer, size_t length,
 141                 const struct skshash *hash)
 142 {
 143         snprintf(buffer, length, "%s/skshash/%02X/%02X/%02X%02X%02X%02X/"
 144                 "%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X",
 145                  config.db_dir,
 146                  hash->hash[0], hash->hash[1],
 147                  hash->hash[0], hash->hash[1], hash->hash[2], hash->hash[3],
 148                  hash->hash[4], hash->hash[5], hash->hash[6], hash->hash[7],
 149                  hash->hash[8], hash->hash[9], hash->hash[10], hash->hash[11],
 150                  hash->hash[12], hash->hash[13], hash->hash[14],
 151                  hash->hash[15]);
 152 }
 153 static void subkeydir(char *buffer, size_t length, uint64_t subkey)
 154 {
 155         snprintf(buffer, length, "%s/subkeys/%02X/%02X/%08X",
 156                  config.db_dir,
 157                  (uint8_t) ((subkey >> 24) & 0xFF),
 158                  (uint8_t) ((subkey >> 16) & 0xFF),
 159                  (uint32_t) (subkey & 0xFFFFFFFF));
 160 }
 161
 162 /*****************************************************************************/
 163
 164 /**
 165  *      initdb - Initialize the key database.
 166  */
 167 static void fs_initdb(bool readonly)
 168 {
 169         char buffer[PATH_MAX];
 170
 171         keydb_lockfile_readonly = readonly;
 172
 173         snprintf(buffer, sizeof(buffer), "%s/.lock", config.db_dir);
 174
 175         if (access(config.db_dir, R_OK | W_OK | X_OK) == -1) {
 176                 if (errno != ENOENT) {
 177                         logthing(LOGTHING_CRITICAL,
 178                                  "Unable to access keydb_fs root of '%s'. (%s)",
 179                                  config.db_dir, strerror(errno));
 180                         exit(1);        /* Lacking rwx on the key dir */
 181                 }
 182                 mkdir(config.db_dir, 0777);
 183                 keydb_lockfile_fd = open(buffer, O_RDWR | O_CREAT, 0600);
 184         }
 185         chdir(config.db_dir);
 186         if (keydb_lockfile_fd == -1)
 187                 keydb_lockfile_fd = open(buffer,
 188                                          (keydb_lockfile_readonly) ?
 189                                          O_RDONLY : O_RDWR);
 190         if (keydb_lockfile_fd == -1)
 191                 keydb_lockfile_fd = open(buffer, O_RDWR | O_CREAT, 0600);
 192         if (keydb_lockfile_fd == -1) {
 193                 logthing(LOGTHING_CRITICAL,
 194                          "Unable to open lockfile '%s'. (%s)",
 195                          buffer, strerror(errno));
 196                 exit(1);        /* Lacking rwx on the key dir */
 197         }
 198 }
 199
 200 /**
 201  *      cleanupdb - De-initialize the key database.
 202  */
 203 static void fs_cleanupdb(void)
 204 {
 205         /* Mmmm nothing to do here? */
 206         close(keydb_lockfile_fd);
 207 }
 208
 209 /**
 210  *      starttrans - Start a transaction.
 211  */
 212 static bool fs_starttrans(void)
 213 {
 214         struct flock lockstruct;
 215         int remaining = 20;
 216         lockstruct.l_type =
 217             F_RDLCK | ((keydb_lockfile_readonly) ? 0 : F_WRLCK);
 218         lockstruct.l_whence = SEEK_SET;
 219         lockstruct.l_start = 0;
 220         lockstruct.l_len = 1;
 221
 222         while (fcntl(keydb_lockfile_fd, F_SETLK, &lockstruct) == -1) {
 223                 if (remaining-- == 0)
 224                         return false;   /* Hope to hell that noodles DTRT */
 225                 usleep(100);
 226         }
 227         return true;
 228 }
 229
 230 /**
 231  *      endtrans - End a transaction.
 232  */
 233 static void fs_endtrans(void)
 234 {
 235         struct flock lockstruct;
 236
 237         lockstruct.l_type = F_UNLCK;
 238         lockstruct.l_whence = SEEK_SET;
 239         lockstruct.l_start = 0;
 240         lockstruct.l_len = 1;
 241         fcntl(keydb_lockfile_fd, F_SETLK, &lockstruct);
 242 }
 243
 244 static uint64_t fs_getfullkeyid(uint64_t keyid)
 245 {
 246         static char buffer[PATH_MAX];
 247         DIR *d = NULL;
 248         struct dirent *de = NULL;
 249         uint64_t ret = 0;
 250
 251         keydir(buffer, sizeof(buffer), keyid);
 252
 253         d = opendir(buffer);
 254         if (d) {
 255                 do {
 256                         de = readdir(d);
 257                         if (de && de->d_name[0] != '.') {
 258                                 ret = strtoull(de->d_name, NULL, 16);
 259                         }
 260                 } while (de && de->d_name[0] == '.');
 261                 closedir(d);
 262         }
 263
 264         if (ret == 0) {
 265                 subkeydir(buffer, sizeof(buffer), keyid);
 266
 267                 d = opendir(buffer);
 268                 if (d) {
 269                         do {
 270                                 de = readdir(d);
 271                                 if (de && de->d_name[0] != '.') {
 272                                         ret = strtoull(de->d_name, NULL, 16);
 273                                 }
 274                         } while (de && de->d_name[0] == '.');
 275                         closedir(d);
 276                 }
 277         }
 278
 279         return ret;
 280 }
 281
 282 /**
 283  *      fetch_key - Given a keyid fetch the key from storage.
 284  *      @keyid: The keyid to fetch.
 285  *      @publickey: A pointer to a structure to return the key in.
 286  *      @intrans: If we're already in a transaction.
 287  */
 288 static int fs_fetch_key(uint64_t keyid, struct openpgp_publickey **publickey,
 289               bool intrans)
 290 {
 291         static char buffer[PATH_MAX];
 292         int ret = 0, fd;
 293         struct openpgp_packet_list *packets = NULL;
 294
 295         if (!intrans)
 296                 fs_starttrans();
 297
 298         if ((keyid >> 32) == 0)
 299                 keyid = fs_getfullkeyid(keyid);
 300
 301         keypath(buffer, sizeof(buffer), keyid);
 302         if ((fd = open(buffer, O_RDONLY)) != -1) {
 303                 /* File is present, load it in... */
 304                 read_openpgp_stream(file_fetchchar, &fd, &packets, 0);
 305                 parse_keys(packets, publickey);
 306                 free_packet_list(packets);
 307                 packets = NULL;
 308                 close(fd);
 309                 ret = 1;
 310         }
 311
 312         if (!intrans)
 313                 fs_endtrans();
 314         return ret;
 315 }
 316
 317 /**
 318  *      store_key - Takes a key and stores it.
 319  *      @publickey: A pointer to the public key to store.
 320  *      @intrans: If we're already in a transaction.
 321  *      @update: If true the key exists and should be updated.
 322  */
 323 static int fs_store_key(struct openpgp_publickey *publickey, bool intrans,
 324               bool update)
 325 {
 326         static char buffer[PATH_MAX];
 327         static char wbuffer[PATH_MAX];
 328         int ret = 0, fd;
 329         struct openpgp_packet_list *packets = NULL;
 330         struct openpgp_packet_list *list_end = NULL;
 331         struct openpgp_publickey *next = NULL;
 332         uint64_t keyid = get_keyid(publickey);
 333         struct ll *wordlist = NULL, *wl = NULL;
 334         struct skshash hash;
 335         uint64_t *subkeyids = NULL;
 336         uint32_t hashid;
 337         int i = 0;
 338
 339
 340         if (!intrans)
 341                 fs_starttrans();
 342
 343         prove_path_to(keyid, "key");
 344         keypath(buffer, sizeof(buffer), keyid);
 345
 346         if ((fd =
 347              open(buffer, O_WRONLY | (update ? O_TRUNC : O_CREAT),
 348                   0644)) != -1) {
 349                 next = publickey->next;
 350                 publickey->next = NULL;
 351                 flatten_publickey(publickey, &packets, &list_end);
 352                 publickey->next = next;
 353
 354                 write_openpgp_stream(file_putchar, &fd, packets);
 355                 close(fd);
 356                 free_packet_list(packets);
 357                 packets = NULL;
 358                 ret = 1;
 359         }
 360
 361         if (ret) {
 362                 wl = wordlist = makewordlistfromkey(wordlist, publickey);
 363                 while (wl) {
 364                         uint32_t hash = calchash((uint8_t *) (wl->object));
 365                         prove_path_to(hash, "words");
 366
 367                         worddir(wbuffer, sizeof(wbuffer), wl->object, hash);
 368                         mkdir(wbuffer, 0777);
 369                         wordpath(wbuffer, sizeof(wbuffer), wl->object, hash,
 370                                 keyid);
 371                         link(buffer, wbuffer);
 372
 373                         wl = wl->next;
 374                 }
 375                 llfree(wordlist, free);
 376
 377                 subkeyids = keysubkeys(publickey);
 378                 i = 0;
 379                 while (subkeyids != NULL && subkeyids[i] != 0) {
 380                         prove_path_to(subkeyids[i], "subkeys");
 381
 382                         subkeydir(wbuffer, sizeof(wbuffer), subkeyids[i]);
 383                         mkdir(wbuffer, 0777);
 384                         subkeypath(wbuffer, sizeof(wbuffer), subkeyids[i],
 385                                 keyid);
 386                         link(buffer, wbuffer);
 387
 388                         i++;
 389                 }
 390                 if (subkeyids != NULL) {
 391                         free(subkeyids);
 392                         subkeyids = NULL;
 393                 }
 394
 395                 get_skshash(publickey, &hash);
 396                 hashid = (hash.hash[0] << 24) + (hash.hash[1] << 16) +
 397                                 (hash.hash[2] << 8) + hash.hash[3];
 398                 prove_path_to(hashid, "skshash");
 399                 skshashpath(wbuffer, sizeof(wbuffer), &hash);
 400                 link(buffer, wbuffer);
 401         }
 402
 403         if (!intrans)
 404                 fs_endtrans();
 405         return ret;
 406 }
 407
 408 /**
 409  *      delete_key - Given a keyid delete the key from storage.
 410  *      @keyid: The keyid to delete.
 411  *      @intrans: If we're already in a transaction.
 412  */
 413 static int fs_delete_key(uint64_t keyid, bool intrans)
 414 {
 415         static char buffer[PATH_MAX];
 416         int ret;
 417         struct openpgp_publickey *pk = NULL;
 418         struct skshash hash;
 419         struct ll *wordlist = NULL, *wl = NULL;
 420         uint64_t *subkeyids = NULL;
 421         int i = 0;
 422
 423         if ((keyid >> 32) == 0)
 424                 keyid = fs_getfullkeyid(keyid);
 425
 426         if (!intrans)
 427                 fs_starttrans();
 428
 429         ret = fs_fetch_key(keyid, &pk, true);
 430
 431         if (ret) {
 432                 logthing(LOGTHING_DEBUG, "Wordlist for key %016" PRIX64,
 433                          keyid);
 434                 wl = wordlist = makewordlistfromkey(wordlist, pk);
 435                 logthing(LOGTHING_DEBUG,
 436                          "Wordlist for key %016" PRIX64 " done", keyid);
 437                 while (wl) {
 438                         uint32_t hash = calchash((uint8_t *) (wl->object));
 439                         prove_path_to(hash, "words");
 440
 441                         wordpath(buffer, sizeof(buffer), wl->object, hash,
 442                                 keyid);
 443                         unlink(buffer);
 444
 445                         wl = wl->next;
 446                 }
 447
 448                 subkeyids = keysubkeys(pk);
 449                 i = 0;
 450                 while (subkeyids != NULL && subkeyids[i] != 0) {
 451                         prove_path_to(subkeyids[i], "subkeys");
 452
 453                         subkeypath(buffer, sizeof(buffer), subkeyids[i],
 454                                 keyid);
 455                         unlink(buffer);
 456
 457                         i++;
 458                 }
 459                 if (subkeyids != NULL) {
 460                         free(subkeyids);
 461                         subkeyids = NULL;
 462                 }
 463
 464                 get_skshash(pk, &hash);
 465                 skshashpath(buffer, sizeof(buffer), &hash);
 466                 unlink(buffer);
 467         }
 468
 469         keypath(buffer, sizeof(buffer), keyid);
 470         unlink(buffer);
 471
 472         if (!intrans)
 473                 fs_endtrans();
 474         return 1;
 475 }
 476
 477 static struct ll *internal_get_key_by_word(char *word, struct ll *mct)
 478 {
 479         struct ll *keys = NULL;
 480         DIR *d = NULL;
 481         char buffer[PATH_MAX];
 482         uint32_t hash = calchash((uint8_t *) (word));
 483         struct dirent *de;
 484
 485         worddir(buffer, sizeof(buffer), word, hash);
 486         d = opendir(buffer);
 487         logthing(LOGTHING_DEBUG, "Scanning for word %s in dir %s", word,
 488                  buffer);
 489         if (d)
 490                 do {
 491                         de = readdir(d);
 492                         if (de && de->d_name[0] != '.') {
 493                                 if ((!mct)
 494                                     || (llfind(mct, de->d_name,
 495                                         (int (*)(const void *, const void *))
 496                                                     strcmp) !=
 497                                         NULL)) {
 498                                         logthing(LOGTHING_DEBUG,
 499                                                  "Found %s // %s", word,
 500                                                  de->d_name);
 501                                         keys =
 502                                             lladd(keys,
 503                                                   strdup(de->d_name));
 504                                 }
 505                         }
 506                 } while (de);
 507         closedir(d);
 508
 509         return keys;
 510 }
 511
 512 /*
 513  *      fetch_key_text - Trys to find the keys that contain the supplied text.
 514  *      @search: The text to search for.
 515  *      @publickey: A pointer to a structure to return the key in.
 516  */
 517 static int fs_fetch_key_text(const char *search,
 518                    struct openpgp_publickey **publickey)
 519 {
 520         struct ll *wordlist = NULL, *wl = NULL;
 521         struct ll *keylist = NULL;
 522         char      *searchtext = NULL;
 523         int addedkeys = 0;
 524
 525         logthing(LOGTHING_DEBUG, "Search was '%s'", search);
 526
 527         searchtext = strdup(search);
 528         wl = wordlist = makewordlist(wordlist, searchtext);
 529
 530         keylist = internal_get_key_by_word(wordlist->object, NULL);
 531
 532         if (!keylist) {
 533                 llfree(wordlist, NULL);
 534                 free(searchtext);
 535                 searchtext = NULL;
 536                 return 0;
 537         }
 538
 539         wl = wl->next;
 540         while (wl) {
 541                 struct ll *nkl =
 542                     internal_get_key_by_word(wl->object, keylist);
 543                 if (!nkl) {
 544                         llfree(wordlist, NULL);
 545                         llfree(keylist, free);
 546                         free(searchtext);
 547                         searchtext = NULL;
 548                         return 0;
 549                 }
 550                 llfree(keylist, free);
 551                 keylist = nkl;
 552                 wl = wl->next;
 553         }
 554
 555         llfree(wordlist, NULL);
 556
 557         /* Now add the keys... */
 558         wl = keylist;
 559         while (wl) {
 560                 logthing(LOGTHING_DEBUG, "Adding key: %s", wl->object);
 561                 addedkeys +=
 562                     fs_fetch_key(strtoull(wl->object, NULL, 16), publickey,
 563                               false);
 564                 if (addedkeys >= config.maxkeys)
 565                         break;
 566                 wl = wl->next;
 567         }
 568
 569         llfree(keylist, free);
 570         free(searchtext);
 571         searchtext = NULL;
 572
 573         return addedkeys;
 574 }
 575
 576 /**
 577  *      fetch_key_skshash - Given an SKS hash fetch the key from storage.
 578  *      @hash: The hash to fetch.
 579  *      @publickey: A pointer to a structure to return the key in.
 580  *      @intrans: If we're already in a transaction.
 581  */
 582 static int fs_fetch_key_skshash(const struct skshash *hash,
 583               struct openpgp_publickey **publickey)
 584 {
 585         static char buffer[PATH_MAX];
 586         int ret = 0, fd;
 587         struct openpgp_packet_list *packets = NULL;
 588
 589         skshashpath(buffer, sizeof(buffer), hash);
 590         if ((fd = open(buffer, O_RDONLY)) != -1) {
 591                 read_openpgp_stream(file_fetchchar, &fd, &packets, 0);
 592                 parse_keys(packets, publickey);
 593                 free_packet_list(packets);
 594                 packets = NULL;
 595                 close(fd);
 596                 ret = 1;
 597         }
 598
 599         return ret;
 600 }
 601
 602 /**
 603  *      iterate_keys - call a function once for each key in the db.
 604  *      @iterfunc: The function to call.
 605  *      @ctx: A context pointer
 606  *
 607  *      Calls iterfunc once for each key in the database. ctx is passed
 608  *      unaltered to iterfunc. This function is intended to aid database dumps
 609  *      and statistic calculations.
 610  *
 611  *      Returns the number of keys we iterated over.
 612  */
 613 static int fs_iterate_keys(void (*iterfunc)(void *ctx,
 614                 struct openpgp_publickey *key), void *ctx)
 615 {
 616         return 0;
 617 }
 618
 619 /*
 620  * Include the basic keydb routines.
 621  */
 622 #define NEED_KEYID2UID 1
 623 #define NEED_GETKEYSIGS 1
 624 #define NEED_UPDATEKEYS 1
 625 #include "keydb.c"
 626
 627 struct dbfuncs keydb_fs_funcs = {
 628         .initdb                 = fs_initdb,
 629         .cleanupdb              = fs_cleanupdb,
 630         .starttrans             = fs_starttrans,
 631         .endtrans               = fs_endtrans,
 632         .fetch_key              = fs_fetch_key,
 633         .fetch_key_text         = fs_fetch_key_text,
 634         .fetch_key_skshash      = fs_fetch_key_skshash,
 635         .store_key              = fs_store_key,
 636         .update_keys            = generic_update_keys,
 637         .delete_key             = fs_delete_key,
 638         .getkeysigs             = generic_getkeysigs,
 639         .cached_getkeysigs      = generic_cached_getkeysigs,
 640         .keyid2uid              = generic_keyid2uid,
 641         .getfullkeyid           = fs_getfullkeyid,
 642         .iterate_keys           = fs_iterate_keys,
 643 };