* keyserver.c (keyserver_spawn): Use the full 64-bit keyid in the INFO
[gnupg.git] / g10 / tdbio.c
1 /* tdbio.c
2  * Copyright (C) 1998, 1999, 2000, 2001, 2002 Free Software Foundation, Inc.
3  *
4  * This file is part of GnuPG.
5  *
6  * GnuPG is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License as published by
8  * the Free Software Foundation; either version 2 of the License, or
9  * (at your option) any later version.
10  *
11  * GnuPG is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14  * GNU General Public License for more details.
15  *
16  * You should have received a copy of the GNU General Public License
17  * along with this program; if not, write to the Free Software
18  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA
19  */
20
21 #include <config.h>
22 #include <stdio.h>
23 #include <stdlib.h>
24 #include <string.h>
25 #include <errno.h>
26 #include <assert.h>
27 #include <sys/types.h>
28 #include <sys/stat.h>
29 #include <fcntl.h>
30 #include <unistd.h>
31
32 #include "errors.h"
33 #include "iobuf.h"
34 #include "memory.h"
35 #include "util.h"
36 #include "options.h"
37 #include "main.h"
38 #include "i18n.h"
39 #include "trustdb.h"
40 #include "tdbio.h"
41
42 #if defined(HAVE_DOSISH_SYSTEM)
43 #define ftruncate chsize
44 #endif
45
46 #if defined(HAVE_DOSISH_SYSTEM) || defined(__CYGWIN__)
47 #define MY_O_BINARY  O_BINARY
48 #else
49 #define MY_O_BINARY  0
50 #endif
51
52
53 /****************
54  * Yes, this is a very simple implementation. We should really
55  * use a page aligned buffer and read complete pages.
56  * To implement a simple trannsaction system, this is sufficient.
57  */
58 typedef struct cache_ctrl_struct *CACHE_CTRL;
59 struct cache_ctrl_struct {
60     CACHE_CTRL next;
61     struct {
62         unsigned used:1;
63         unsigned dirty:1;
64     } flags;
65     ulong recno;
66     char data[TRUST_RECORD_LEN];
67 };
68
69 #define MAX_CACHE_ENTRIES_SOFT  200    /* may be increased while in a */
70 #define MAX_CACHE_ENTRIES_HARD  10000  /* transaction to this one */
71 static CACHE_CTRL cache_list;
72 static int cache_entries;
73 static int cache_is_dirty;
74
75 /* a type used to pass infomation to cmp_krec_fpr */
76 struct cmp_krec_fpr_struct {
77     int pubkey_algo;
78     const char *fpr;
79     int fprlen;
80 };
81
82 /* a type used to pass infomation to cmp_[s]dir */
83 struct cmp_xdir_struct {
84     int pubkey_algo;
85     u32 keyid[2];
86 };
87
88
89 static char *db_name;
90 static DOTLOCK lockhandle;
91 static int is_locked;
92 static int  db_fd = -1;
93 static int in_transaction;
94
95 static void open_db(void);
96 static void migrate_from_v2 (void);
97
98
99 \f
100 /*************************************
101  ************* record cache **********
102  *************************************/
103
104 /****************
105  * Get the data from therecord cache and return a
106  * pointer into that cache.  Caller should copy
107  * the return data.  NULL is returned on a cache miss.
108  */
109 static const char *
110 get_record_from_cache( ulong recno )
111 {
112     CACHE_CTRL r;
113
114     for( r = cache_list; r; r = r->next ) {
115         if( r->flags.used && r->recno == recno )
116             return r->data;
117     }
118     return NULL;
119 }
120
121
122 static int
123 write_cache_item( CACHE_CTRL r )
124 {
125     int n;
126
127     if( lseek( db_fd, r->recno * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
128         log_error(_("trustdb rec %lu: lseek failed: %s\n"),
129                                             r->recno, strerror(errno) );
130         return G10ERR_WRITE_FILE;
131     }
132     n = write( db_fd, r->data, TRUST_RECORD_LEN);
133     if( n != TRUST_RECORD_LEN ) {
134         log_error(_("trustdb rec %lu: write failed (n=%d): %s\n"),
135                                             r->recno, n, strerror(errno) );
136         return G10ERR_WRITE_FILE;
137     }
138     r->flags.dirty = 0;
139     return 0;
140 }
141
142 /****************
143  * Put data into the cache.  This function may flush the
144  * some cache entries if there is not enough space available.
145  */
146 int
147 put_record_into_cache( ulong recno, const char *data )
148 {
149     CACHE_CTRL r, unused;
150     int dirty_count = 0;
151     int clean_count = 0;
152
153     /* see whether we already cached this one */
154     for( unused = NULL, r = cache_list; r; r = r->next ) {
155         if( !r->flags.used ) {
156             if( !unused )
157                 unused = r;
158         }
159         else if( r->recno == recno ) {
160             if( !r->flags.dirty ) {
161                 /* Hmmm: should we use a a copy and compare? */
162                 if( memcmp(r->data, data, TRUST_RECORD_LEN ) ) {
163                     r->flags.dirty = 1;
164                     cache_is_dirty = 1;
165                 }
166             }
167             memcpy( r->data, data, TRUST_RECORD_LEN );
168             return 0;
169         }
170         if( r->flags.used ) {
171             if( r->flags.dirty )
172                 dirty_count++;
173             else
174                 clean_count++;
175         }
176     }
177     /* not in the cache: add a new entry */
178     if( unused ) { /* reuse this entry */
179         r = unused;
180         r->flags.used = 1;
181         r->recno = recno;
182         memcpy( r->data, data, TRUST_RECORD_LEN );
183         r->flags.dirty = 1;
184         cache_is_dirty = 1;
185         cache_entries++;
186         return 0;
187     }
188     /* see whether we reached the limit */
189     if( cache_entries < MAX_CACHE_ENTRIES_SOFT ) { /* no */
190         r = m_alloc( sizeof *r );
191         r->flags.used = 1;
192         r->recno = recno;
193         memcpy( r->data, data, TRUST_RECORD_LEN );
194         r->flags.dirty = 1;
195         r->next = cache_list;
196         cache_list = r;
197         cache_is_dirty = 1;
198         cache_entries++;
199         return 0;
200     }
201     /* cache is full: discard some clean entries */
202     if( clean_count ) {
203         int n = clean_count / 3; /* discard a third of the clean entries */
204         if( !n )
205             n = 1;
206         for( unused = NULL, r = cache_list; r; r = r->next ) {
207             if( r->flags.used && !r->flags.dirty ) {
208                 if( !unused )
209                     unused = r;
210                 r->flags.used = 0;
211                 cache_entries--;
212                 if( !--n )
213                     break;
214             }
215         }
216         assert( unused );
217         r = unused;
218         r->flags.used = 1;
219         r->recno = recno;
220         memcpy( r->data, data, TRUST_RECORD_LEN );
221         r->flags.dirty = 1;
222         cache_is_dirty = 1;
223         cache_entries++;
224         return 0;
225     }
226     /* no clean entries: have to flush some dirty entries */
227     if( in_transaction ) {
228         /* but we can't do this while in a transaction
229          * we increase the cache size instead */
230         if( cache_entries < MAX_CACHE_ENTRIES_HARD ) { /* no */
231             if( opt.debug && !(cache_entries % 100) )
232                 log_debug("increasing tdbio cache size\n");
233             r = m_alloc( sizeof *r );
234             r->flags.used = 1;
235             r->recno = recno;
236             memcpy( r->data, data, TRUST_RECORD_LEN );
237             r->flags.dirty = 1;
238             r->next = cache_list;
239             cache_list = r;
240             cache_is_dirty = 1;
241             cache_entries++;
242             return 0;
243         }
244         log_info(_("trustdb transaction too large\n"));
245         return G10ERR_RESOURCE_LIMIT;
246     }
247     if( dirty_count ) {
248         int n = dirty_count / 5; /* discard some dirty entries */
249         if( !n )
250             n = 1;
251         if( !is_locked ) {
252             if( make_dotlock( lockhandle, -1 ) )
253                 log_fatal("can't acquire lock - giving up\n");
254             else
255                 is_locked = 1;
256         }
257         for( unused = NULL, r = cache_list; r; r = r->next ) {
258             if( r->flags.used && r->flags.dirty ) {
259                 int rc = write_cache_item( r );
260                 if( rc )
261                     return rc;
262                 if( !unused )
263                     unused = r;
264                 r->flags.used = 0;
265                 cache_entries--;
266                 if( !--n )
267                     break;
268             }
269         }
270         if( !opt.lock_once ) {
271             if( !release_dotlock( lockhandle ) )
272                 is_locked = 0;
273         }
274         assert( unused );
275         r = unused;
276         r->flags.used = 1;
277         r->recno = recno;
278         memcpy( r->data, data, TRUST_RECORD_LEN );
279         r->flags.dirty = 1;
280         cache_is_dirty = 1;
281         cache_entries++;
282         return 0;
283     }
284     BUG();
285 }
286
287
288 int
289 tdbio_is_dirty()
290 {
291     return cache_is_dirty;
292 }
293
294
295 /****************
296  * Flush the cache.  This cannot be used while in a transaction.
297  */
298 int
299 tdbio_sync()
300 {
301     CACHE_CTRL r;
302     int did_lock = 0;
303
304     if( db_fd == -1 )
305         open_db();
306     if( in_transaction )
307         log_bug("tdbio: syncing while in transaction\n");
308
309     if( !cache_is_dirty )
310         return 0;
311
312     if( !is_locked ) {
313         if( make_dotlock( lockhandle, -1 ) )
314             log_fatal("can't acquire lock - giving up\n");
315         else
316             is_locked = 1;
317         did_lock = 1;
318     }
319     for( r = cache_list; r; r = r->next ) {
320         if( r->flags.used && r->flags.dirty ) {
321             int rc = write_cache_item( r );
322             if( rc )
323                 return rc;
324         }
325     }
326     cache_is_dirty = 0;
327     if( did_lock && !opt.lock_once ) {
328         if( !release_dotlock( lockhandle ) )
329             is_locked = 0;
330     }
331
332     return 0;
333 }
334
335 #if 0
336 /* The transaction code is disabled in the 1.2.x branch, as it is not
337    yet used.  It will be enabled in 1.3.x. */
338
339 /****************
340  * Simple transactions system:
341  * Everything between begin_transaction and end/cancel_transaction
342  * is not immediatly written but at the time of end_transaction.
343  *
344  */
345 int
346 tdbio_begin_transaction()
347 {
348     int rc;
349
350     if( in_transaction )
351         log_bug("tdbio: nested transactions\n");
352     /* flush everything out */
353     rc = tdbio_sync();
354     if( rc )
355         return rc;
356     in_transaction = 1;
357     return 0;
358 }
359
360 int
361 tdbio_end_transaction()
362 {
363     int rc;
364
365     if( !in_transaction )
366         log_bug("tdbio: no active transaction\n");
367     if( !is_locked ) {
368         if( make_dotlock( lockhandle, -1 ) )
369             log_fatal("can't acquire lock - giving up\n");
370         else
371             is_locked = 1;
372     }
373     block_all_signals();
374     in_transaction = 0;
375     rc = tdbio_sync();
376     unblock_all_signals();
377     if( !opt.lock_once ) {
378         if( !release_dotlock( lockhandle ) )
379             is_locked = 0;
380     }
381     return rc;
382 }
383
384 int
385 tdbio_cancel_transaction()
386 {
387     CACHE_CTRL r;
388
389     if( !in_transaction )
390         log_bug("tdbio: no active transaction\n");
391
392     /* remove all dirty marked entries, so that the original ones
393      * are read back the next time */
394     if( cache_is_dirty ) {
395         for( r = cache_list; r; r = r->next ) {
396             if( r->flags.used && r->flags.dirty ) {
397                 r->flags.used = 0;
398                 cache_entries--;
399             }
400         }
401         cache_is_dirty = 0;
402     }
403
404     in_transaction = 0;
405     return 0;
406 }
407 #endif
408
409 \f
410 /********************************************************
411  **************** cached I/O functions ******************
412  ********************************************************/
413
414 static void
415 cleanup(void)
416 {
417     if( is_locked ) {
418         if( !release_dotlock(lockhandle) )
419             is_locked = 0;
420     }
421 }
422
423 /* Caller must sync */
424 int
425 tdbio_update_version_record (void)
426 {
427   TRUSTREC rec;
428   int rc;
429
430   memset( &rec, 0, sizeof rec );
431
432   rc=tdbio_read_record( 0, &rec, RECTYPE_VER);
433   if(rc==0)
434     {
435       rec.r.ver.created     = make_timestamp();
436       rec.r.ver.marginals   = opt.marginals_needed;
437       rec.r.ver.completes   = opt.completes_needed;
438       rec.r.ver.cert_depth  = opt.max_cert_depth;
439       rec.r.ver.trust_model = opt.trust_model;
440       rc=tdbio_write_record(&rec);
441     }
442
443   return rc;
444 }
445
446 static int
447 create_version_record (void)
448 {
449   TRUSTREC rec;
450   int rc;
451   
452   memset( &rec, 0, sizeof rec );
453   rec.r.ver.version     = 3;
454   rec.r.ver.created     = make_timestamp();
455   rec.r.ver.marginals   = opt.marginals_needed;
456   rec.r.ver.completes   = opt.completes_needed;
457   rec.r.ver.cert_depth  = opt.max_cert_depth;
458   if(opt.trust_model==TM_PGP || opt.trust_model==TM_CLASSIC)
459     rec.r.ver.trust_model = opt.trust_model;
460   else
461     rec.r.ver.trust_model = TM_PGP;
462   rec.rectype = RECTYPE_VER;
463   rec.recnum = 0;
464   rc = tdbio_write_record( &rec );
465   if( !rc )
466     tdbio_sync();
467   return rc;
468 }
469
470
471
472 int
473 tdbio_set_dbname( const char *new_dbname, int create )
474 {
475     char *fname;
476     static int initialized = 0;
477
478     if( !initialized ) {
479         atexit( cleanup );
480         initialized = 1;
481     }
482
483     if(new_dbname==NULL)
484       fname=make_filename(opt.homedir,"trustdb" EXTSEP_S "gpg", NULL);
485     else if (*new_dbname != DIRSEP_C )
486       {
487         if (strchr(new_dbname, DIRSEP_C) )
488           fname = make_filename (new_dbname, NULL);
489         else
490           fname = make_filename (opt.homedir, new_dbname, NULL);
491       }
492     else
493       fname = m_strdup (new_dbname);
494
495     if( access( fname, R_OK ) ) {
496         if( errno != ENOENT ) {
497             log_error( _("%s: can't access: %s\n"), fname, strerror(errno) );
498             m_free(fname);
499             return G10ERR_TRUSTDB;
500         }
501         if( create ) {
502             FILE *fp;
503             TRUSTREC rec;
504             int rc;
505             char *p = strrchr( fname, DIRSEP_C );
506             mode_t oldmask;
507
508             assert(p);
509             *p = 0;
510             if( access( fname, F_OK ) ) {
511                 try_make_homedir( fname );
512                 log_fatal( _("%s: directory does not exist!\n"), fname );
513             }
514             *p = DIRSEP_C;
515
516             m_free(db_name);
517             db_name = fname;
518 #ifdef __riscos__
519             if( !lockhandle )
520                 lockhandle = create_dotlock( db_name );
521             if( !lockhandle )
522                 log_fatal( _("%s: can't create lock\n"), db_name );
523             if( make_dotlock( lockhandle, -1 ) )
524                 log_fatal( _("%s: can't make lock\n"), db_name );
525 #endif /* __riscos__ */
526             oldmask=umask(077);
527             fp =fopen( fname, "wb" );
528             umask(oldmask);
529             if( !fp )
530                 log_fatal( _("%s: can't create: %s\n"), fname, strerror(errno) );
531             fclose(fp);
532             db_fd = open( db_name, O_RDWR | MY_O_BINARY );
533             if( db_fd == -1 )
534                 log_fatal( _("%s: can't open: %s\n"), db_name, strerror(errno) );
535
536 #ifndef __riscos__
537             if( !lockhandle )
538                 lockhandle = create_dotlock( db_name );
539             if( !lockhandle )
540                 log_fatal( _("%s: can't create lock\n"), db_name );
541 #endif /* !__riscos__ */
542
543             rc = create_version_record ();
544             if( rc )
545                 log_fatal( _("%s: failed to create version record: %s"),
546                                                    fname, g10_errstr(rc));
547             /* and read again to check that we are okay */
548             if( tdbio_read_record( 0, &rec, RECTYPE_VER ) )
549                 log_fatal( _("%s: invalid trustdb created\n"), db_name );
550
551             if( !opt.quiet )
552                 log_info(_("%s: trustdb created\n"), db_name);
553
554             return 0;
555         }
556     }
557     m_free(db_name);
558     db_name = fname;
559     return 0;
560 }
561
562
563 const char *
564 tdbio_get_dbname()
565 {
566     return db_name;
567 }
568
569
570
571 static void
572 open_db()
573 {
574   byte buf[10];
575   int n;
576   TRUSTREC rec;
577
578   assert( db_fd == -1 );
579
580   if (!lockhandle )
581     lockhandle = create_dotlock( db_name );
582   if (!lockhandle )
583     log_fatal( _("%s: can't create lock\n"), db_name );
584 #ifdef __riscos__
585   if (make_dotlock( lockhandle, -1 ) )
586     log_fatal( _("%s: can't make lock\n"), db_name );
587 #endif /* __riscos__ */
588   db_fd = open (db_name, O_RDWR | MY_O_BINARY );
589   if (db_fd == -1 && errno == EACCES) {
590       db_fd = open (db_name, O_RDONLY | MY_O_BINARY );
591       if (db_fd != -1)
592           log_info (_("NOTE: trustdb not writable\n"));
593   }
594   if ( db_fd == -1 )
595     log_fatal( _("%s: can't open: %s\n"), db_name, strerror(errno) );
596
597   /* check whether we need to do a version migration */
598   do
599     n = read (db_fd, buf, 5);
600   while (n==-1 && errno == EINTR);
601   if (n == 5 && !memcmp (buf, "\x01gpg\x02", 5))
602     {
603       migrate_from_v2 ();
604     }
605   
606   /* read the version record */
607   if (tdbio_read_record (0, &rec, RECTYPE_VER ) )
608     log_fatal( _("%s: invalid trustdb\n"), db_name );
609 }
610
611
612 /****************
613  * Make a hashtable: type 0 = trust hash
614  */
615 static void
616 create_hashtable( TRUSTREC *vr, int type )
617 {
618     TRUSTREC rec;
619     off_t offset;
620     ulong recnum;
621     int i, n, rc;
622
623     offset = lseek( db_fd, 0, SEEK_END );
624     if( offset == -1 )
625         log_fatal("trustdb: lseek to end failed: %s\n", strerror(errno) );
626     recnum = offset / TRUST_RECORD_LEN;
627     assert(recnum); /* this is will never be the first record */
628
629     if( !type )
630         vr->r.ver.trusthashtbl = recnum;
631
632     /* Now write the records */
633     n = (256+ITEMS_PER_HTBL_RECORD-1) / ITEMS_PER_HTBL_RECORD;
634     for(i=0; i < n; i++, recnum++ ) {
635          memset( &rec, 0, sizeof rec );
636          rec.rectype = RECTYPE_HTBL;
637          rec.recnum = recnum;
638          rc = tdbio_write_record( &rec );
639          if( rc )
640              log_fatal( _("%s: failed to create hashtable: %s\n"),
641                                         db_name, g10_errstr(rc));
642     }
643     /* update the version record */
644     rc = tdbio_write_record( vr );
645     if( !rc )
646         rc = tdbio_sync();
647     if( rc )
648         log_fatal( _("%s: error updating version record: %s\n"),
649                                                   db_name, g10_errstr(rc));
650 }
651
652
653 int
654 tdbio_db_matches_options()
655 {
656   static int yes_no = -1;
657
658   if( yes_no == -1 )
659     {
660       TRUSTREC vr;
661       int rc;
662
663       rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
664       if( rc )
665         log_fatal( _("%s: error reading version record: %s\n"),
666                    db_name, g10_errstr(rc) );
667
668       yes_no = vr.r.ver.marginals == opt.marginals_needed
669         && vr.r.ver.completes == opt.completes_needed
670         && vr.r.ver.cert_depth == opt.max_cert_depth
671         && vr.r.ver.trust_model == opt.trust_model;
672     }
673
674   return yes_no;
675 }
676
677 byte
678 tdbio_read_model(void)
679 {
680   TRUSTREC vr;
681   int rc;
682  
683   rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
684   if( rc )
685     log_fatal( _("%s: error reading version record: %s\n"),
686                db_name, g10_errstr(rc) );
687   return vr.r.ver.trust_model;
688 }
689
690 /****************
691  * Return the nextstamp value.
692  */
693 ulong
694 tdbio_read_nextcheck ()
695 {
696     TRUSTREC vr;
697     int rc;
698
699     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
700     if( rc )
701         log_fatal( _("%s: error reading version record: %s\n"),
702                                                     db_name, g10_errstr(rc) );
703     return vr.r.ver.nextcheck;
704 }
705
706 /* Return true when the stamp was actually changed. */
707 int
708 tdbio_write_nextcheck (ulong stamp)
709 {
710     TRUSTREC vr;
711     int rc;
712
713     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
714     if( rc )
715         log_fatal( _("%s: error reading version record: %s\n"),
716                                        db_name, g10_errstr(rc) );
717
718     if (vr.r.ver.nextcheck == stamp)
719       return 0;
720
721     vr.r.ver.nextcheck = stamp;
722     rc = tdbio_write_record( &vr );
723     if( rc )
724         log_fatal( _("%s: error writing version record: %s\n"),
725                                        db_name, g10_errstr(rc) );
726     return 1;
727 }
728
729
730
731 /****************
732  * Return the record number of the trusthash tbl or create a new one.
733  */
734 static ulong
735 get_trusthashrec(void)
736 {
737     static ulong trusthashtbl; /* record number of the trust hashtable */
738
739     if( !trusthashtbl ) {
740         TRUSTREC vr;
741         int rc;
742
743         rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
744         if( rc )
745             log_fatal( _("%s: error reading version record: %s\n"),
746                                             db_name, g10_errstr(rc) );
747         if( !vr.r.ver.trusthashtbl )
748             create_hashtable( &vr, 0 );
749
750         trusthashtbl = vr.r.ver.trusthashtbl;
751     }
752     return trusthashtbl;
753 }
754
755
756
757 /****************
758  * Update a hashtable.
759  * table gives the start of the table, key and keylen is the key,
760  * newrecnum is the record number to insert.
761  */
762 static int
763 upd_hashtable( ulong table, byte *key, int keylen, ulong newrecnum )
764 {
765     TRUSTREC lastrec, rec;
766     ulong hashrec, item;
767     int msb;
768     int level=0;
769     int rc, i;
770
771     hashrec = table;
772   next_level:
773     msb = key[level];
774     hashrec += msb / ITEMS_PER_HTBL_RECORD;
775     rc = tdbio_read_record( hashrec, &rec, RECTYPE_HTBL );
776     if( rc ) {
777         log_error("upd_hashtable: read failed: %s\n",   g10_errstr(rc) );
778         return rc;
779     }
780
781     item = rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
782     if( !item ) { /* insert a new item into the hash table */
783         rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = newrecnum;
784         rc = tdbio_write_record( &rec );
785         if( rc ) {
786             log_error("upd_hashtable: write htbl failed: %s\n",
787                                                             g10_errstr(rc) );
788             return rc;
789         }
790     }
791     else if( item != newrecnum ) {  /* must do an update */
792         lastrec = rec;
793         rc = tdbio_read_record( item, &rec, 0 );
794         if( rc ) {
795             log_error( "upd_hashtable: read item failed: %s\n",
796                                                             g10_errstr(rc) );
797             return rc;
798         }
799
800         if( rec.rectype == RECTYPE_HTBL ) {
801             hashrec = item;
802             level++;
803             if( level >= keylen ) {
804                 log_error( "hashtable has invalid indirections.\n");
805                 return G10ERR_TRUSTDB;
806             }
807             goto next_level;
808         }
809         else if( rec.rectype == RECTYPE_HLST ) { /* extend list */
810             /* see whether the key is already in this list */
811             for(;;) {
812                 for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
813                     if( rec.r.hlst.rnum[i] == newrecnum ) {
814                         return 0; /* okay, already in the list */
815                     }
816                 }
817                 if( rec.r.hlst.next ) {
818                     rc = tdbio_read_record( rec.r.hlst.next,
819                                                        &rec, RECTYPE_HLST);
820                     if( rc ) {
821                         log_error( "upd_hashtable: read hlst failed: %s\n",
822                                                              g10_errstr(rc) );
823                         return rc;
824                     }
825                 }
826                 else
827                     break; /* not there */
828             }
829             /* find the next free entry and put it in */
830             for(;;) {
831                 for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
832                     if( !rec.r.hlst.rnum[i] ) {
833                         rec.r.hlst.rnum[i] = newrecnum;
834                         rc = tdbio_write_record( &rec );
835                         if( rc )
836                             log_error( "upd_hashtable: write hlst failed: %s\n",
837                                                               g10_errstr(rc) );
838                         return rc; /* done */
839                     }
840                 }
841                 if( rec.r.hlst.next ) {
842                     rc = tdbio_read_record( rec.r.hlst.next,
843                                                       &rec, RECTYPE_HLST );
844                     if( rc ) {
845                         log_error( "upd_hashtable: read hlst failed: %s\n",
846                                                              g10_errstr(rc) );
847                         return rc;
848                     }
849                 }
850                 else { /* add a new list record */
851                     rec.r.hlst.next = item = tdbio_new_recnum();
852                     rc = tdbio_write_record( &rec );
853                     if( rc ) {
854                         log_error( "upd_hashtable: write hlst failed: %s\n",
855                                                           g10_errstr(rc) );
856                         return rc;
857                     }
858                     memset( &rec, 0, sizeof rec );
859                     rec.rectype = RECTYPE_HLST;
860                     rec.recnum = item;
861                     rec.r.hlst.rnum[0] = newrecnum;
862                     rc = tdbio_write_record( &rec );
863                     if( rc )
864                         log_error( "upd_hashtable: write ext hlst failed: %s\n",
865                                                           g10_errstr(rc) );
866                     return rc; /* done */
867                 }
868             } /* end loop over hlst slots */
869         }
870         else if( rec.rectype == RECTYPE_TRUST ) { /* insert a list record */
871             if( rec.recnum == newrecnum ) {
872                 return 0;
873             }
874             item = rec.recnum; /* save number of key record */
875             memset( &rec, 0, sizeof rec );
876             rec.rectype = RECTYPE_HLST;
877             rec.recnum = tdbio_new_recnum();
878             rec.r.hlst.rnum[0] = item;       /* old keyrecord */
879             rec.r.hlst.rnum[1] = newrecnum; /* and new one */
880             rc = tdbio_write_record( &rec );
881             if( rc ) {
882                 log_error( "upd_hashtable: write new hlst failed: %s\n",
883                                                   g10_errstr(rc) );
884                 return rc;
885             }
886             /* update the hashtable record */
887             lastrec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = rec.recnum;
888             rc = tdbio_write_record( &lastrec );
889             if( rc )
890                 log_error( "upd_hashtable: update htbl failed: %s\n",
891                                                              g10_errstr(rc) );
892             return rc; /* ready */
893         }
894         else {
895             log_error( "hashtbl %lu: %lu/%d points to an invalid record %lu\n",
896                        table, hashrec, (msb % ITEMS_PER_HTBL_RECORD), item);
897             list_trustdb(NULL);
898             return G10ERR_TRUSTDB;
899         }
900     }
901
902     return 0;
903 }
904
905
906 /****************
907  * Drop an entry from a hashtable
908  * table gives the start of the table, key and keylen is the key,
909  */
910 static int
911 drop_from_hashtable( ulong table, byte *key, int keylen, ulong recnum )
912 {
913     TRUSTREC rec;
914     ulong hashrec, item;
915     int msb;
916     int level=0;
917     int rc, i;
918
919     hashrec = table;
920   next_level:
921     msb = key[level];
922     hashrec += msb / ITEMS_PER_HTBL_RECORD;
923     rc = tdbio_read_record( hashrec, &rec, RECTYPE_HTBL );
924     if( rc ) {
925         log_error("drop_from_hashtable: read failed: %s\n",
926                                                         g10_errstr(rc) );
927         return rc;
928     }
929
930     item = rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
931     if( !item )  /* not found - forget about it  */
932         return 0;
933
934     if( item == recnum ) {  /* tables points direct to the record */
935         rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = 0;
936         rc = tdbio_write_record( &rec );
937         if( rc )
938             log_error("drop_from_hashtable: write htbl failed: %s\n",
939                                                             g10_errstr(rc) );
940         return rc;
941     }
942
943     rc = tdbio_read_record( item, &rec, 0 );
944     if( rc ) {
945         log_error( "drop_from_hashtable: read item failed: %s\n",
946                                                         g10_errstr(rc) );
947         return rc;
948     }
949
950     if( rec.rectype == RECTYPE_HTBL ) {
951         hashrec = item;
952         level++;
953         if( level >= keylen ) {
954             log_error( "hashtable has invalid indirections.\n");
955             return G10ERR_TRUSTDB;
956         }
957         goto next_level;
958     }
959
960     if( rec.rectype == RECTYPE_HLST ) {
961         for(;;) {
962             for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
963                 if( rec.r.hlst.rnum[i] == recnum ) {
964                     rec.r.hlst.rnum[i] = 0; /* drop */
965                     rc = tdbio_write_record( &rec );
966                     if( rc )
967                         log_error("drop_from_hashtable: write htbl failed: %s\n",
968                                                                         g10_errstr(rc) );
969                     return rc;
970                 }
971             }
972             if( rec.r.hlst.next ) {
973                 rc = tdbio_read_record( rec.r.hlst.next,
974                                                    &rec, RECTYPE_HLST);
975                 if( rc ) {
976                     log_error( "drop_from_hashtable: read hlst failed: %s\n",
977                                                          g10_errstr(rc) );
978                     return rc;
979                 }
980             }
981             else
982                 return 0; /* key not in table */
983         }
984     }
985
986     log_error( "hashtbl %lu: %lu/%d points to wrong record %lu\n",
987                     table, hashrec, (msb % ITEMS_PER_HTBL_RECORD), item);
988     return G10ERR_TRUSTDB;
989 }
990
991
992
993 /****************
994  * Lookup a record via the hashtable tablewith key/keylen and return the
995  * result in rec.  cmp() should return if the record is the desired one.
996  * Returns -1 if not found, 0 if found or another errocode
997  */
998 static int
999 lookup_hashtable( ulong table, const byte *key, size_t keylen,
1000                   int (*cmpfnc)(void*, const TRUSTREC *), void *cmpdata,
1001                                                 TRUSTREC *rec )
1002 {
1003     int rc;
1004     ulong hashrec, item;
1005     int msb;
1006     int level=0;
1007
1008     hashrec = table;
1009   next_level:
1010     msb = key[level];
1011     hashrec += msb / ITEMS_PER_HTBL_RECORD;
1012     rc = tdbio_read_record( hashrec, rec, RECTYPE_HTBL );
1013     if( rc ) {
1014         log_error("lookup_hashtable failed: %s\n", g10_errstr(rc) );
1015         return rc;
1016     }
1017
1018     item = rec->r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
1019     if( !item )
1020         return -1; /* not found */
1021
1022     rc = tdbio_read_record( item, rec, 0 );
1023     if( rc ) {
1024         log_error( "hashtable read failed: %s\n", g10_errstr(rc) );
1025         return rc;
1026     }
1027     if( rec->rectype == RECTYPE_HTBL ) {
1028         hashrec = item;
1029         level++;
1030         if( level >= keylen ) {
1031             log_error("hashtable has invalid indirections\n");
1032             return G10ERR_TRUSTDB;
1033         }
1034         goto next_level;
1035     }
1036     else if( rec->rectype == RECTYPE_HLST ) {
1037         for(;;) {
1038             int i;
1039
1040             for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1041                 if( rec->r.hlst.rnum[i] ) {
1042                     TRUSTREC tmp;
1043
1044                     rc = tdbio_read_record( rec->r.hlst.rnum[i], &tmp, 0 );
1045                     if( rc ) {
1046                         log_error( "lookup_hashtable: read item failed: %s\n",
1047                                                               g10_errstr(rc) );
1048                         return rc;
1049                     }
1050                     if( (*cmpfnc)( cmpdata, &tmp ) ) {
1051                         *rec = tmp;
1052                         return 0;
1053                     }
1054                 }
1055             }
1056             if( rec->r.hlst.next ) {
1057                 rc = tdbio_read_record( rec->r.hlst.next, rec, RECTYPE_HLST );
1058                 if( rc ) {
1059                     log_error( "lookup_hashtable: read hlst failed: %s\n",
1060                                                          g10_errstr(rc) );
1061                     return rc;
1062                 }
1063             }
1064             else
1065                 return -1; /* not found */
1066         }
1067     }
1068
1069
1070     if( (*cmpfnc)( cmpdata, rec ) )
1071         return 0; /* really found */
1072
1073     return -1; /* no: not found */
1074 }
1075
1076
1077 /****************
1078  * Update the trust hashtbl or create the table if it does not exist
1079  */
1080 static int
1081 update_trusthashtbl( TRUSTREC *tr )
1082 {
1083     return upd_hashtable( get_trusthashrec(),
1084                           tr->r.trust.fingerprint, 20, tr->recnum );
1085 }
1086
1087
1088
1089 void
1090 tdbio_dump_record( TRUSTREC *rec, FILE *fp  )
1091 {
1092     int i;
1093     ulong rnum = rec->recnum;
1094
1095     fprintf(fp, "rec %5lu, ", rnum );
1096
1097     switch( rec->rectype ) {
1098       case 0: fprintf(fp, "blank\n");
1099         break;
1100       case RECTYPE_VER: fprintf(fp,
1101             "version, td=%lu, f=%lu, m/c/d=%d/%d/%d tm=%d nc=%lu (%s)\n",
1102                                    rec->r.ver.trusthashtbl,
1103                                    rec->r.ver.firstfree,
1104                                    rec->r.ver.marginals,
1105                                    rec->r.ver.completes,
1106                                    rec->r.ver.cert_depth,
1107                                    rec->r.ver.trust_model,
1108                                    rec->r.ver.nextcheck,
1109                                    strtimestamp(rec->r.ver.nextcheck)
1110                                  );
1111         break;
1112       case RECTYPE_FREE: fprintf(fp, "free, next=%lu\n", rec->r.free.next );
1113         break;
1114       case RECTYPE_HTBL:
1115         fprintf(fp, "htbl,");
1116         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ )
1117             fprintf(fp, " %lu", rec->r.htbl.item[i] );
1118         putc('\n', fp);
1119         break;
1120       case RECTYPE_HLST:
1121         fprintf(fp, "hlst, next=%lu,", rec->r.hlst.next );
1122         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ )
1123             fprintf(fp, " %lu", rec->r.hlst.rnum[i] );
1124         putc('\n', fp);
1125         break;
1126       case RECTYPE_TRUST:
1127         fprintf(fp, "trust ");
1128         for(i=0; i < 20; i++ )
1129             fprintf(fp, "%02X", rec->r.trust.fingerprint[i] );
1130         fprintf (fp, ", ot=%d, d=%d, vl=%lu\n", rec->r.trust.ownertrust,
1131                  rec->r.trust.depth, rec->r.trust.validlist);
1132         break;
1133       case RECTYPE_VALID:
1134         fprintf(fp, "valid ");
1135         for(i=0; i < 20; i++ )
1136             fprintf(fp, "%02X", rec->r.valid.namehash[i] );
1137         fprintf (fp, ", v=%d, next=%lu\n", rec->r.valid.validity,
1138                  rec->r.valid.next);
1139         break;
1140       default:
1141         fprintf(fp, "unknown type %d\n", rec->rectype );
1142         break;
1143     }
1144 }
1145
1146 /****************
1147  * read the record with number recnum
1148  * returns: -1 on error, 0 on success
1149  */
1150 int
1151 tdbio_read_record( ulong recnum, TRUSTREC *rec, int expected )
1152 {
1153     byte readbuf[TRUST_RECORD_LEN];
1154     const byte *buf, *p;
1155     int rc = 0;
1156     int n, i;
1157
1158     if( db_fd == -1 )
1159         open_db();
1160     buf = get_record_from_cache( recnum );
1161     if( !buf ) {
1162         if( lseek( db_fd, recnum * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
1163             log_error(_("trustdb: lseek failed: %s\n"), strerror(errno) );
1164             return G10ERR_READ_FILE;
1165         }
1166         n = read( db_fd, readbuf, TRUST_RECORD_LEN);
1167         if( !n ) {
1168             return -1; /* eof */
1169         }
1170         else if( n != TRUST_RECORD_LEN ) {
1171             log_error(_("trustdb: read failed (n=%d): %s\n"), n,
1172                                                         strerror(errno) );
1173             return G10ERR_READ_FILE;
1174         }
1175         buf = readbuf;
1176     }
1177     rec->recnum = recnum;
1178     rec->dirty = 0;
1179     p = buf;
1180     rec->rectype = *p++;
1181     if( expected && rec->rectype != expected ) {
1182         log_error("%lu: read expected rec type %d, got %d\n",
1183                     recnum, expected, rec->rectype );
1184         return G10ERR_TRUSTDB;
1185     }
1186     p++;    /* skip reserved byte */
1187     switch( rec->rectype ) {
1188       case 0:  /* unused (free) record */
1189         break;
1190       case RECTYPE_VER: /* version record */
1191         if( memcmp(buf+1, "gpg", 3 ) ) {
1192             log_error( _("%s: not a trustdb file\n"), db_name );
1193             rc = G10ERR_TRUSTDB;
1194         }
1195         p += 2; /* skip "gpg" */
1196         rec->r.ver.version  = *p++;
1197         rec->r.ver.marginals = *p++;
1198         rec->r.ver.completes = *p++;
1199         rec->r.ver.cert_depth = *p++;
1200         rec->r.ver.trust_model = *p++;
1201         p += 3;
1202         rec->r.ver.created  = buftoulong(p); p += 4;
1203         rec->r.ver.nextcheck = buftoulong(p); p += 4;
1204         p += 4;
1205         p += 4;
1206         rec->r.ver.firstfree =buftoulong(p); p += 4;
1207         p += 4;
1208         rec->r.ver.trusthashtbl =buftoulong(p); p += 4;
1209         if( recnum ) {
1210             log_error( _("%s: version record with recnum %lu\n"), db_name,
1211                                                              (ulong)recnum );
1212             rc = G10ERR_TRUSTDB;
1213         }
1214         else if( rec->r.ver.version != 3 ) {
1215             log_error( _("%s: invalid file version %d\n"), db_name,
1216                                                         rec->r.ver.version );
1217             rc = G10ERR_TRUSTDB;
1218         }
1219         break;
1220       case RECTYPE_FREE:
1221         rec->r.free.next  = buftoulong(p); p += 4;
1222         break;
1223       case RECTYPE_HTBL:
1224         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ ) {
1225             rec->r.htbl.item[i] = buftoulong(p); p += 4;
1226         }
1227         break;
1228       case RECTYPE_HLST:
1229         rec->r.hlst.next = buftoulong(p); p += 4;
1230         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1231             rec->r.hlst.rnum[i] = buftoulong(p); p += 4;
1232         }
1233         break;
1234       case RECTYPE_TRUST:
1235         memcpy( rec->r.trust.fingerprint, p, 20); p+=20;
1236         rec->r.trust.ownertrust = *p++;
1237         rec->r.trust.depth = *p++;
1238         rec->r.trust.min_ownertrust = *p++;
1239         p++;
1240         rec->r.trust.validlist = buftoulong(p); p += 4;
1241         break;
1242       case RECTYPE_VALID:
1243         memcpy( rec->r.valid.namehash, p, 20); p+=20;
1244         rec->r.valid.validity = *p++;
1245         rec->r.valid.next = buftoulong(p); p += 4;
1246         rec->r.valid.full_count = *p++;
1247         rec->r.valid.marginal_count = *p++;
1248         break;
1249       default:
1250         log_error( "%s: invalid record type %d at recnum %lu\n",
1251                                    db_name, rec->rectype, (ulong)recnum );
1252         rc = G10ERR_TRUSTDB;
1253         break;
1254     }
1255
1256     return rc;
1257 }
1258
1259 /****************
1260  * Write the record at RECNUM
1261  */
1262 int
1263 tdbio_write_record( TRUSTREC *rec )
1264 {
1265     byte buf[TRUST_RECORD_LEN], *p;
1266     int rc = 0;
1267     int i;
1268     ulong recnum = rec->recnum;
1269
1270     if( db_fd == -1 )
1271         open_db();
1272
1273     memset(buf, 0, TRUST_RECORD_LEN);
1274     p = buf;
1275     *p++ = rec->rectype; p++;
1276     switch( rec->rectype ) {
1277       case 0:  /* unused record */
1278         break;
1279       case RECTYPE_VER: /* version record */
1280         if( recnum )
1281             BUG();
1282         memcpy(p-1, "gpg", 3 ); p += 2;
1283         *p++ = rec->r.ver.version;
1284         *p++ = rec->r.ver.marginals;
1285         *p++ = rec->r.ver.completes;
1286         *p++ = rec->r.ver.cert_depth;
1287         *p++ = rec->r.ver.trust_model;
1288         p += 3;
1289         ulongtobuf(p, rec->r.ver.created); p += 4;
1290         ulongtobuf(p, rec->r.ver.nextcheck); p += 4;
1291         p += 4;
1292         p += 4;
1293         ulongtobuf(p, rec->r.ver.firstfree ); p += 4;
1294         p += 4;
1295         ulongtobuf(p, rec->r.ver.trusthashtbl ); p += 4;
1296         break;
1297
1298       case RECTYPE_FREE:
1299         ulongtobuf(p, rec->r.free.next); p += 4;
1300         break;
1301
1302
1303       case RECTYPE_HTBL:
1304         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ ) {
1305             ulongtobuf( p, rec->r.htbl.item[i]); p += 4;
1306         }
1307         break;
1308
1309       case RECTYPE_HLST:
1310         ulongtobuf( p, rec->r.hlst.next); p += 4;
1311         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1312             ulongtobuf( p, rec->r.hlst.rnum[i]); p += 4;
1313         }
1314         break;
1315
1316       case RECTYPE_TRUST:
1317         memcpy( p, rec->r.trust.fingerprint, 20); p += 20;
1318         *p++ = rec->r.trust.ownertrust;
1319         *p++ = rec->r.trust.depth;
1320         *p++ = rec->r.trust.min_ownertrust;
1321         p++;
1322         ulongtobuf( p, rec->r.trust.validlist); p += 4;
1323         break;
1324
1325       case RECTYPE_VALID:
1326         memcpy( p, rec->r.valid.namehash, 20); p += 20;
1327         *p++ = rec->r.valid.validity;
1328         ulongtobuf( p, rec->r.valid.next); p += 4;
1329         *p++ = rec->r.valid.full_count;
1330         *p++ = rec->r.valid.marginal_count;
1331         break;
1332
1333       default:
1334         BUG();
1335     }
1336
1337     rc = put_record_into_cache( recnum, buf );
1338     if( rc )
1339         ;
1340     else if( rec->rectype == RECTYPE_TRUST )
1341         rc = update_trusthashtbl( rec );
1342
1343     return rc;
1344 }
1345
1346 int
1347 tdbio_delete_record( ulong recnum )
1348 {
1349     TRUSTREC vr, rec;
1350     int rc;
1351
1352     /* Must read the record fist, so we can drop it from the hash tables */
1353     rc = tdbio_read_record( recnum, &rec, 0 );
1354     if( rc )
1355         ;
1356     else if( rec.rectype == RECTYPE_TRUST ) {
1357          rc = drop_from_hashtable( get_trusthashrec(),
1358                                    rec.r.trust.fingerprint, 20, rec.recnum );
1359     }
1360
1361     if( rc )
1362         return rc;
1363
1364     /* now we can chnage it to a free record */
1365     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
1366     if( rc )
1367         log_fatal( _("%s: error reading version record: %s\n"),
1368                                        db_name, g10_errstr(rc) );
1369
1370     rec.recnum = recnum;
1371     rec.rectype = RECTYPE_FREE;
1372     rec.r.free.next = vr.r.ver.firstfree;
1373     vr.r.ver.firstfree = recnum;
1374     rc = tdbio_write_record( &rec );
1375     if( !rc )
1376         rc = tdbio_write_record( &vr );
1377     return rc;
1378 }
1379
1380 /****************
1381  * create a new record and return its record number
1382  */
1383 ulong
1384 tdbio_new_recnum()
1385 {
1386     off_t offset;
1387     ulong recnum;
1388     TRUSTREC vr, rec;
1389     int rc;
1390
1391     /* look for unused records */
1392     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
1393     if( rc )
1394         log_fatal( _("%s: error reading version record: %s\n"),
1395                                              db_name, g10_errstr(rc) );
1396     if( vr.r.ver.firstfree ) {
1397         recnum = vr.r.ver.firstfree;
1398         rc = tdbio_read_record( recnum, &rec, RECTYPE_FREE );
1399         if( rc ) {
1400             log_error( _("%s: error reading free record: %s\n"),
1401                                                   db_name,  g10_errstr(rc) );
1402             return rc;
1403         }
1404         /* update dir record */
1405         vr.r.ver.firstfree = rec.r.free.next;
1406         rc = tdbio_write_record( &vr );
1407         if( rc ) {
1408             log_error( _("%s: error writing dir record: %s\n"),
1409                                                      db_name, g10_errstr(rc) );
1410             return rc;
1411         }
1412         /*zero out the new record */
1413         memset( &rec, 0, sizeof rec );
1414         rec.rectype = 0; /* unused record */
1415         rec.recnum = recnum;
1416         rc = tdbio_write_record( &rec );
1417         if( rc )
1418             log_fatal(_("%s: failed to zero a record: %s\n"),
1419                                        db_name, g10_errstr(rc));
1420     }
1421     else { /* not found, append a new record */
1422         offset = lseek( db_fd, 0, SEEK_END );
1423         if( offset == -1 )
1424             log_fatal("trustdb: lseek to end failed: %s\n", strerror(errno) );
1425         recnum = offset / TRUST_RECORD_LEN;
1426         assert(recnum); /* this is will never be the first record */
1427         /* we must write a record, so that the next call to this function
1428          * returns another recnum */
1429         memset( &rec, 0, sizeof rec );
1430         rec.rectype = 0; /* unused record */
1431         rec.recnum = recnum;
1432         rc = 0;
1433         if( lseek( db_fd, recnum * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
1434             log_error(_("trustdb rec %lu: lseek failed: %s\n"),
1435                                                 recnum, strerror(errno) );
1436             rc = G10ERR_WRITE_FILE;
1437         }
1438         else {
1439             int n = write( db_fd, &rec, TRUST_RECORD_LEN);
1440             if( n != TRUST_RECORD_LEN ) {
1441                 log_error(_("trustdb rec %lu: write failed (n=%d): %s\n"),
1442                                                  recnum, n, strerror(errno) );
1443                 rc = G10ERR_WRITE_FILE;
1444             }
1445         }
1446
1447         if( rc )
1448             log_fatal(_("%s: failed to append a record: %s\n"),
1449                                     db_name,    g10_errstr(rc));
1450     }
1451     return recnum ;
1452 }
1453
1454
1455
1456 static int
1457 cmp_trec_fpr ( void *fpr, const TRUSTREC *rec )
1458 {
1459     return rec->rectype == RECTYPE_TRUST
1460            && !memcmp( rec->r.trust.fingerprint, fpr, 20);
1461 }
1462
1463
1464 int
1465 tdbio_search_trust_byfpr( const byte *fingerprint, TRUSTREC *rec )
1466 {
1467     int rc;
1468
1469     /* locate the trust record using the hash table */
1470     rc = lookup_hashtable( get_trusthashrec(), fingerprint, 20,
1471                            cmp_trec_fpr, (void*)fingerprint, rec );
1472     return rc;
1473 }
1474
1475 int
1476 tdbio_search_trust_bypk (PKT_public_key *pk, TRUSTREC *rec)
1477 {
1478     byte fingerprint[MAX_FINGERPRINT_LEN];
1479     size_t fingerlen;
1480
1481     fingerprint_from_pk( pk, fingerprint, &fingerlen );
1482     for (; fingerlen < 20; fingerlen++ )
1483       fingerprint[fingerlen] = 0;
1484     return tdbio_search_trust_byfpr (fingerprint, rec);
1485 }
1486
1487
1488
1489 void
1490 tdbio_invalid(void)
1491 {
1492     log_error(_(
1493         "the trustdb is corrupted; please run \"gpg --fix-trustdb\".\n") );
1494     g10_exit(2);
1495 }
1496
1497 /*
1498  * Migrate the trustdb as just up to gpg 1.0.6 (trustdb version 2)
1499  * to the 2.1 version as used with 1.0.6b - This is pretty trivial as needs
1500  * only to scan the tdb and insert new the new trust records.  The old ones are
1501  * obsolte from now on
1502  */
1503 static void
1504 migrate_from_v2 ()
1505 {
1506   TRUSTREC rec;
1507   int i, n;
1508   struct {
1509     ulong keyrecno;
1510     byte  ot;
1511     byte okay;
1512     byte  fpr[20];
1513   } *ottable;
1514   int ottable_size, ottable_used;
1515   byte oldbuf[40];
1516   ulong recno;
1517   int rc, count;
1518
1519   ottable_size = 5;
1520   ottable = m_alloc (ottable_size * sizeof *ottable);
1521   ottable_used = 0;
1522
1523   /* We have some restrictions here.  We can't use the version record
1524    * and we can't use any of the old hashtables because we dropped the
1525    * code.  So we first collect all ownertrusts and then use a second
1526    * pass fo find the associated keys.  We have to do this all without using 
1527    * the regular record read functions.
1528    */
1529
1530   /* get all the ownertrusts */
1531   if (lseek (db_fd, 0, SEEK_SET ) == -1 ) 
1532       log_fatal ("migrate_from_v2: lseek failed: %s\n", strerror (errno));
1533   for (recno=0;;recno++)
1534     {
1535       do
1536         n = read (db_fd, oldbuf, 40);
1537       while (n==-1 && errno == EINTR);
1538       if (!n)
1539         break; /* eof */
1540       if (n != 40)
1541         log_fatal ("migrate_vfrom_v2: read error or short read\n");
1542
1543       if (*oldbuf != 2)
1544         continue;
1545       
1546       /* v2 dir record */
1547       if (ottable_used == ottable_size)
1548         {
1549           ottable_size += 1000;
1550           ottable = m_realloc (ottable, ottable_size * sizeof *ottable);
1551         }
1552       ottable[ottable_used].keyrecno = buftoulong (oldbuf+6);
1553       ottable[ottable_used].ot = oldbuf[18];
1554       ottable[ottable_used].okay = 0;
1555       memset (ottable[ottable_used].fpr,0, 20);
1556       if (ottable[ottable_used].keyrecno && ottable[ottable_used].ot)
1557         ottable_used++;
1558     }
1559   log_info ("found %d ownertrust records\n", ottable_used);
1560
1561   /* Read again and find the fingerprints */
1562   if (lseek (db_fd, 0, SEEK_SET ) == -1 ) 
1563       log_fatal ("migrate_from_v2: lseek failed: %s\n", strerror (errno));
1564   for (recno=0;;recno++)
1565     {
1566       do
1567         n = read (db_fd, oldbuf, 40);
1568       while (n==-1 && errno == EINTR);
1569       if (!n)
1570         break; /* eof */
1571       if (n != 40)
1572         log_fatal ("migrate_from_v2: read error or short read\n");
1573
1574       if (*oldbuf != 3) 
1575         continue;
1576
1577       /* v2 key record */
1578       for (i=0; i < ottable_used; i++)
1579         {
1580           if (ottable[i].keyrecno == recno)
1581             {
1582               memcpy (ottable[i].fpr, oldbuf+20, 20);
1583               ottable[i].okay = 1;
1584               break;
1585             }
1586         }
1587     }
1588
1589   /* got everything - create the v3 trustdb */
1590   if (ftruncate (db_fd, 0))
1591     log_fatal ("can't truncate `%s': %s\n", db_name, strerror (errno) );
1592   if (create_version_record ())
1593     log_fatal ("failed to recreate version record of `%s'\n", db_name);
1594
1595   /* access the hash table, so it is store just after the version record, 
1596    * this is not needed put a dump is more pretty */
1597   get_trusthashrec ();
1598
1599   /* And insert the old ownertrust values */
1600   count = 0;
1601   for (i=0; i < ottable_used; i++)
1602     {
1603       if (!ottable[i].okay)
1604         continue;
1605       
1606       memset (&rec, 0, sizeof rec);
1607       rec.recnum = tdbio_new_recnum ();
1608       rec.rectype = RECTYPE_TRUST;
1609       memcpy(rec.r.trust.fingerprint, ottable[i].fpr, 20);
1610       rec.r.trust.ownertrust = ottable[i].ot;
1611       if (tdbio_write_record (&rec))
1612         log_fatal ("failed to write trust record of `%s'\n", db_name);
1613       count++;
1614     }
1615
1616   revalidation_mark ();
1617   rc = tdbio_sync ();
1618   if (rc)
1619     log_fatal ("failed to sync `%s'\n", db_name);
1620   log_info ("migrated %d version 2 ownertrusts\n", count);
1621   m_free (ottable);
1622 }