* trustdb.h, trustdb.c (is_disabled), gpgv.c (is_disabled): Rename
[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
336
337 /****************
338  * Simple transactions system:
339  * Everything between begin_transaction and end/cancel_transaction
340  * is not immediatly written but at the time of end_transaction.
341  *
342  */
343 int
344 tdbio_begin_transaction()
345 {
346     int rc;
347
348     if( in_transaction )
349         log_bug("tdbio: nested transactions\n");
350     /* flush everything out */
351     rc = tdbio_sync();
352     if( rc )
353         return rc;
354     in_transaction = 1;
355     return 0;
356 }
357
358 int
359 tdbio_end_transaction()
360 {
361     int rc;
362
363     if( !in_transaction )
364         log_bug("tdbio: no active transaction\n");
365     if( !is_locked ) {
366         if( make_dotlock( lockhandle, -1 ) )
367             log_fatal("can't acquire lock - giving up\n");
368         else
369             is_locked = 1;
370     }
371     block_all_signals();
372     in_transaction = 0;
373     rc = tdbio_sync();
374     unblock_all_signals();
375     if( !opt.lock_once ) {
376         if( !release_dotlock( lockhandle ) )
377             is_locked = 0;
378     }
379     return rc;
380 }
381
382 int
383 tdbio_cancel_transaction()
384 {
385     CACHE_CTRL r;
386
387     if( !in_transaction )
388         log_bug("tdbio: no active transaction\n");
389
390     /* remove all dirty marked entries, so that the original ones
391      * are read back the next time */
392     if( cache_is_dirty ) {
393         for( r = cache_list; r; r = r->next ) {
394             if( r->flags.used && r->flags.dirty ) {
395                 r->flags.used = 0;
396                 cache_entries--;
397             }
398         }
399         cache_is_dirty = 0;
400     }
401
402     in_transaction = 0;
403     return 0;
404 }
405
406
407 \f
408 /********************************************************
409  **************** cached I/O functions ******************
410  ********************************************************/
411
412 static void
413 cleanup(void)
414 {
415     if( is_locked ) {
416         if( !release_dotlock(lockhandle) )
417             is_locked = 0;
418     }
419 }
420
421 /* Caller must sync */
422 int
423 tdbio_update_version_record (void)
424 {
425   TRUSTREC rec;
426   int rc;
427
428   memset( &rec, 0, sizeof rec );
429
430   rc=tdbio_read_record( 0, &rec, RECTYPE_VER);
431   if(rc==0)
432     {
433       rec.r.ver.created     = make_timestamp();
434       rec.r.ver.marginals   = opt.marginals_needed;
435       rec.r.ver.completes   = opt.completes_needed;
436       rec.r.ver.cert_depth  = opt.max_cert_depth;
437       rec.r.ver.trust_model = opt.trust_model;
438       rc=tdbio_write_record(&rec);
439     }
440
441   return rc;
442 }
443
444 static int
445 create_version_record (void)
446 {
447   TRUSTREC rec;
448   int rc;
449   
450   memset( &rec, 0, sizeof rec );
451   rec.r.ver.version     = 3;
452   rec.r.ver.created     = make_timestamp();
453   rec.r.ver.marginals   = opt.marginals_needed;
454   rec.r.ver.completes   = opt.completes_needed;
455   rec.r.ver.cert_depth  = opt.max_cert_depth;
456   if(opt.trust_model==TM_PGP || opt.trust_model==TM_CLASSIC)
457     rec.r.ver.trust_model = opt.trust_model;
458   else
459     rec.r.ver.trust_model = TM_PGP;
460   rec.rectype = RECTYPE_VER;
461   rec.recnum = 0;
462   rc = tdbio_write_record( &rec );
463   if( !rc )
464     tdbio_sync();
465   return rc;
466 }
467
468
469
470 int
471 tdbio_set_dbname( const char *new_dbname, int create )
472 {
473     char *fname;
474     static int initialized = 0;
475
476     if( !initialized ) {
477         atexit( cleanup );
478         initialized = 1;
479     }
480
481     if(new_dbname==NULL)
482       fname=make_filename(opt.homedir,"trustdb" EXTSEP_S "gpg", NULL);
483     else if (*new_dbname != DIRSEP_C )
484       {
485         if (strchr(new_dbname, DIRSEP_C) )
486           fname = make_filename (new_dbname, NULL);
487         else
488           fname = make_filename (opt.homedir, new_dbname, NULL);
489       }
490     else
491       fname = m_strdup (new_dbname);
492
493     if( access( fname, R_OK ) ) {
494         if( errno != ENOENT ) {
495             log_error( _("%s: can't access: %s\n"), fname, strerror(errno) );
496             m_free(fname);
497             return G10ERR_TRUSTDB;
498         }
499         if( create ) {
500             FILE *fp;
501             TRUSTREC rec;
502             int rc;
503             char *p = strrchr( fname, DIRSEP_C );
504             mode_t oldmask;
505
506             assert(p);
507             *p = 0;
508             if( access( fname, F_OK ) ) {
509                 try_make_homedir( fname );
510                 log_fatal( _("%s: directory does not exist!\n"), fname );
511             }
512             *p = DIRSEP_C;
513
514             m_free(db_name);
515             db_name = fname;
516 #ifdef __riscos__
517             if( !lockhandle )
518                 lockhandle = create_dotlock( db_name );
519             if( !lockhandle )
520                 log_fatal( _("%s: can't create lock\n"), db_name );
521             if( make_dotlock( lockhandle, -1 ) )
522                 log_fatal( _("%s: can't make lock\n"), db_name );
523 #endif /* __riscos__ */
524             oldmask=umask(077);
525             fp =fopen( fname, "wb" );
526             umask(oldmask);
527             if( !fp )
528                 log_fatal( _("%s: can't create: %s\n"), fname, strerror(errno) );
529             fclose(fp);
530             db_fd = open( db_name, O_RDWR | MY_O_BINARY );
531             if( db_fd == -1 )
532                 log_fatal( _("%s: can't open: %s\n"), db_name, strerror(errno) );
533
534 #ifndef __riscos__
535             if( !lockhandle )
536                 lockhandle = create_dotlock( db_name );
537             if( !lockhandle )
538                 log_fatal( _("%s: can't create lock\n"), db_name );
539 #endif /* !__riscos__ */
540
541             rc = create_version_record ();
542             if( rc )
543                 log_fatal( _("%s: failed to create version record: %s"),
544                                                    fname, g10_errstr(rc));
545             /* and read again to check that we are okay */
546             if( tdbio_read_record( 0, &rec, RECTYPE_VER ) )
547                 log_fatal( _("%s: invalid trustdb created\n"), db_name );
548
549             if( !opt.quiet )
550                 log_info(_("%s: trustdb created\n"), db_name);
551
552             return 0;
553         }
554     }
555     m_free(db_name);
556     db_name = fname;
557     return 0;
558 }
559
560
561 const char *
562 tdbio_get_dbname()
563 {
564     return db_name;
565 }
566
567
568
569 static void
570 open_db()
571 {
572   byte buf[10];
573   int n;
574   TRUSTREC rec;
575
576   assert( db_fd == -1 );
577
578   if (!lockhandle )
579     lockhandle = create_dotlock( db_name );
580   if (!lockhandle )
581     log_fatal( _("%s: can't create lock\n"), db_name );
582 #ifdef __riscos__
583   if (make_dotlock( lockhandle, -1 ) )
584     log_fatal( _("%s: can't make lock\n"), db_name );
585 #endif /* __riscos__ */
586   db_fd = open (db_name, O_RDWR | MY_O_BINARY );
587   if (db_fd == -1 && errno == EACCES) {
588       db_fd = open (db_name, O_RDONLY | MY_O_BINARY );
589       if (db_fd != -1)
590           log_info (_("NOTE: trustdb not writable\n"));
591   }
592   if ( db_fd == -1 )
593     log_fatal( _("%s: can't open: %s\n"), db_name, strerror(errno) );
594
595   /* check whether we need to do a version migration */
596   do
597     n = read (db_fd, buf, 5);
598   while (n==-1 && errno == EINTR);
599   if (n == 5 && !memcmp (buf, "\x01gpg\x02", 5))
600     {
601       migrate_from_v2 ();
602     }
603   
604   /* read the version record */
605   if (tdbio_read_record (0, &rec, RECTYPE_VER ) )
606     log_fatal( _("%s: invalid trustdb\n"), db_name );
607 }
608
609
610 /****************
611  * Make a hashtable: type 0 = trust hash
612  */
613 static void
614 create_hashtable( TRUSTREC *vr, int type )
615 {
616     TRUSTREC rec;
617     off_t offset;
618     ulong recnum;
619     int i, n, rc;
620
621     offset = lseek( db_fd, 0, SEEK_END );
622     if( offset == -1 )
623         log_fatal("trustdb: lseek to end failed: %s\n", strerror(errno) );
624     recnum = offset / TRUST_RECORD_LEN;
625     assert(recnum); /* this is will never be the first record */
626
627     if( !type )
628         vr->r.ver.trusthashtbl = recnum;
629
630     /* Now write the records */
631     n = (256+ITEMS_PER_HTBL_RECORD-1) / ITEMS_PER_HTBL_RECORD;
632     for(i=0; i < n; i++, recnum++ ) {
633          memset( &rec, 0, sizeof rec );
634          rec.rectype = RECTYPE_HTBL;
635          rec.recnum = recnum;
636          rc = tdbio_write_record( &rec );
637          if( rc )
638              log_fatal( _("%s: failed to create hashtable: %s\n"),
639                                         db_name, g10_errstr(rc));
640     }
641     /* update the version record */
642     rc = tdbio_write_record( vr );
643     if( !rc )
644         rc = tdbio_sync();
645     if( rc )
646         log_fatal( _("%s: error updating version record: %s\n"),
647                                                   db_name, g10_errstr(rc));
648 }
649
650
651 int
652 tdbio_db_matches_options()
653 {
654   static int yes_no = -1;
655
656   if( yes_no == -1 )
657     {
658       TRUSTREC vr;
659       int rc;
660
661       rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
662       if( rc )
663         log_fatal( _("%s: error reading version record: %s\n"),
664                    db_name, g10_errstr(rc) );
665
666       yes_no = vr.r.ver.marginals == opt.marginals_needed
667         && vr.r.ver.completes == opt.completes_needed
668         && vr.r.ver.cert_depth == opt.max_cert_depth
669         && vr.r.ver.trust_model == opt.trust_model;
670     }
671
672   return yes_no;
673 }
674
675 byte
676 tdbio_read_model(void)
677 {
678   TRUSTREC vr;
679   int rc;
680  
681   rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
682   if( rc )
683     log_fatal( _("%s: error reading version record: %s\n"),
684                db_name, g10_errstr(rc) );
685   return vr.r.ver.trust_model;
686 }
687
688 /****************
689  * Return the nextstamp value.
690  */
691 ulong
692 tdbio_read_nextcheck ()
693 {
694     TRUSTREC vr;
695     int rc;
696
697     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
698     if( rc )
699         log_fatal( _("%s: error reading version record: %s\n"),
700                                                     db_name, g10_errstr(rc) );
701     return vr.r.ver.nextcheck;
702 }
703
704 /* Return true when the stamp was actually changed. */
705 int
706 tdbio_write_nextcheck (ulong stamp)
707 {
708     TRUSTREC vr;
709     int rc;
710
711     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
712     if( rc )
713         log_fatal( _("%s: error reading version record: %s\n"),
714                                        db_name, g10_errstr(rc) );
715
716     if (vr.r.ver.nextcheck == stamp)
717       return 0;
718
719     vr.r.ver.nextcheck = stamp;
720     rc = tdbio_write_record( &vr );
721     if( rc )
722         log_fatal( _("%s: error writing version record: %s\n"),
723                                        db_name, g10_errstr(rc) );
724     return 1;
725 }
726
727
728
729 /****************
730  * Return the record number of the trusthash tbl or create a new one.
731  */
732 static ulong
733 get_trusthashrec(void)
734 {
735     static ulong trusthashtbl; /* record number of the trust hashtable */
736
737     if( !trusthashtbl ) {
738         TRUSTREC vr;
739         int rc;
740
741         rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
742         if( rc )
743             log_fatal( _("%s: error reading version record: %s\n"),
744                                             db_name, g10_errstr(rc) );
745         if( !vr.r.ver.trusthashtbl )
746             create_hashtable( &vr, 0 );
747
748         trusthashtbl = vr.r.ver.trusthashtbl;
749     }
750     return trusthashtbl;
751 }
752
753
754
755 /****************
756  * Update a hashtable.
757  * table gives the start of the table, key and keylen is the key,
758  * newrecnum is the record number to insert.
759  */
760 static int
761 upd_hashtable( ulong table, byte *key, int keylen, ulong newrecnum )
762 {
763     TRUSTREC lastrec, rec;
764     ulong hashrec, item;
765     int msb;
766     int level=0;
767     int rc, i;
768
769     hashrec = table;
770   next_level:
771     msb = key[level];
772     hashrec += msb / ITEMS_PER_HTBL_RECORD;
773     rc = tdbio_read_record( hashrec, &rec, RECTYPE_HTBL );
774     if( rc ) {
775         log_error( db_name, "upd_hashtable: read failed: %s\n",
776                                                         g10_errstr(rc) );
777         return rc;
778     }
779
780     item = rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
781     if( !item ) { /* insert a new item into the hash table */
782         rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = newrecnum;
783         rc = tdbio_write_record( &rec );
784         if( rc ) {
785             log_error( db_name, "upd_hashtable: write htbl failed: %s\n",
786                                                             g10_errstr(rc) );
787             return rc;
788         }
789     }
790     else if( item != newrecnum ) {  /* must do an update */
791         lastrec = rec;
792         rc = tdbio_read_record( item, &rec, 0 );
793         if( rc ) {
794             log_error( "upd_hashtable: read item failed: %s\n",
795                                                             g10_errstr(rc) );
796             return rc;
797         }
798
799         if( rec.rectype == RECTYPE_HTBL ) {
800             hashrec = item;
801             level++;
802             if( level >= keylen ) {
803                 log_error( "hashtable has invalid indirections.\n");
804                 return G10ERR_TRUSTDB;
805             }
806             goto next_level;
807         }
808         else if( rec.rectype == RECTYPE_HLST ) { /* extend list */
809             /* see whether the key is already in this list */
810             for(;;) {
811                 for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
812                     if( rec.r.hlst.rnum[i] == newrecnum ) {
813                         return 0; /* okay, already in the list */
814                     }
815                 }
816                 if( rec.r.hlst.next ) {
817                     rc = tdbio_read_record( rec.r.hlst.next,
818                                                        &rec, RECTYPE_HLST);
819                     if( rc ) {
820                         log_error( "upd_hashtable: read hlst failed: %s\n",
821                                                              g10_errstr(rc) );
822                         return rc;
823                     }
824                 }
825                 else
826                     break; /* not there */
827             }
828             /* find the next free entry and put it in */
829             for(;;) {
830                 for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
831                     if( !rec.r.hlst.rnum[i] ) {
832                         rec.r.hlst.rnum[i] = newrecnum;
833                         rc = tdbio_write_record( &rec );
834                         if( rc )
835                             log_error( "upd_hashtable: write hlst failed: %s\n",
836                                                               g10_errstr(rc) );
837                         return rc; /* done */
838                     }
839                 }
840                 if( rec.r.hlst.next ) {
841                     rc = tdbio_read_record( rec.r.hlst.next,
842                                                       &rec, RECTYPE_HLST );
843                     if( rc ) {
844                         log_error( "upd_hashtable: read hlst failed: %s\n",
845                                                              g10_errstr(rc) );
846                         return rc;
847                     }
848                 }
849                 else { /* add a new list record */
850                     rec.r.hlst.next = item = tdbio_new_recnum();
851                     rc = tdbio_write_record( &rec );
852                     if( rc ) {
853                         log_error( "upd_hashtable: write hlst failed: %s\n",
854                                                           g10_errstr(rc) );
855                         return rc;
856                     }
857                     memset( &rec, 0, sizeof rec );
858                     rec.rectype = RECTYPE_HLST;
859                     rec.recnum = item;
860                     rec.r.hlst.rnum[0] = newrecnum;
861                     rc = tdbio_write_record( &rec );
862                     if( rc )
863                         log_error( "upd_hashtable: write ext hlst failed: %s\n",
864                                                           g10_errstr(rc) );
865                     return rc; /* done */
866                 }
867             } /* end loop over hlst slots */
868         }
869         else if( rec.rectype == RECTYPE_TRUST ) { /* insert a list record */
870             if( rec.recnum == newrecnum ) {
871                 return 0;
872             }
873             item = rec.recnum; /* save number of key record */
874             memset( &rec, 0, sizeof rec );
875             rec.rectype = RECTYPE_HLST;
876             rec.recnum = tdbio_new_recnum();
877             rec.r.hlst.rnum[0] = item;       /* old keyrecord */
878             rec.r.hlst.rnum[1] = newrecnum; /* and new one */
879             rc = tdbio_write_record( &rec );
880             if( rc ) {
881                 log_error( "upd_hashtable: write new hlst failed: %s\n",
882                                                   g10_errstr(rc) );
883                 return rc;
884             }
885             /* update the hashtable record */
886             lastrec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = rec.recnum;
887             rc = tdbio_write_record( &lastrec );
888             if( rc )
889                 log_error( "upd_hashtable: update htbl failed: %s\n",
890                                                              g10_errstr(rc) );
891             return rc; /* ready */
892         }
893         else {
894             log_error( "hashtbl %lu: %lu/%d points to an invalid record %lu\n",
895                        table, hashrec, (msb % ITEMS_PER_HTBL_RECORD), item);
896             list_trustdb(NULL);
897             return G10ERR_TRUSTDB;
898         }
899     }
900
901     return 0;
902 }
903
904
905 /****************
906  * Drop an entry from a hashtable
907  * table gives the start of the table, key and keylen is the key,
908  */
909 static int
910 drop_from_hashtable( ulong table, byte *key, int keylen, ulong recnum )
911 {
912     TRUSTREC rec;
913     ulong hashrec, item;
914     int msb;
915     int level=0;
916     int rc, i;
917
918     hashrec = table;
919   next_level:
920     msb = key[level];
921     hashrec += msb / ITEMS_PER_HTBL_RECORD;
922     rc = tdbio_read_record( hashrec, &rec, RECTYPE_HTBL );
923     if( rc ) {
924         log_error( db_name, "drop_from_hashtable: read failed: %s\n",
925                                                         g10_errstr(rc) );
926         return rc;
927     }
928
929     item = rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
930     if( !item )  /* not found - forget about it  */
931         return 0;
932
933     if( item == recnum ) {  /* tables points direct to the record */
934         rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = 0;
935         rc = tdbio_write_record( &rec );
936         if( rc )
937             log_error( db_name, "drop_from_hashtable: write htbl failed: %s\n",
938                                                             g10_errstr(rc) );
939         return rc;
940     }
941
942     rc = tdbio_read_record( item, &rec, 0 );
943     if( rc ) {
944         log_error( "drop_from_hashtable: read item failed: %s\n",
945                                                         g10_errstr(rc) );
946         return rc;
947     }
948
949     if( rec.rectype == RECTYPE_HTBL ) {
950         hashrec = item;
951         level++;
952         if( level >= keylen ) {
953             log_error( "hashtable has invalid indirections.\n");
954             return G10ERR_TRUSTDB;
955         }
956         goto next_level;
957     }
958
959     if( rec.rectype == RECTYPE_HLST ) {
960         for(;;) {
961             for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
962                 if( rec.r.hlst.rnum[i] == recnum ) {
963                     rec.r.hlst.rnum[i] = 0; /* drop */
964                     rc = tdbio_write_record( &rec );
965                     if( rc )
966                         log_error( db_name, "drop_from_hashtable: write htbl failed: %s\n",
967                                                                         g10_errstr(rc) );
968                     return rc;
969                 }
970             }
971             if( rec.r.hlst.next ) {
972                 rc = tdbio_read_record( rec.r.hlst.next,
973                                                    &rec, RECTYPE_HLST);
974                 if( rc ) {
975                     log_error( "drop_from_hashtable: read hlst failed: %s\n",
976                                                          g10_errstr(rc) );
977                     return rc;
978                 }
979             }
980             else
981                 return 0; /* key not in table */
982         }
983     }
984
985     log_error( "hashtbl %lu: %lu/%d points to wrong record %lu\n",
986                     table, hashrec, (msb % ITEMS_PER_HTBL_RECORD), item);
987     return G10ERR_TRUSTDB;
988 }
989
990
991
992 /****************
993  * Lookup a record via the hashtable tablewith key/keylen and return the
994  * result in rec.  cmp() should return if the record is the desired one.
995  * Returns -1 if not found, 0 if found or another errocode
996  */
997 static int
998 lookup_hashtable( ulong table, const byte *key, size_t keylen,
999                   int (*cmpfnc)(void*, const TRUSTREC *), void *cmpdata,
1000                                                 TRUSTREC *rec )
1001 {
1002     int rc;
1003     ulong hashrec, item;
1004     int msb;
1005     int level=0;
1006
1007     hashrec = table;
1008   next_level:
1009     msb = key[level];
1010     hashrec += msb / ITEMS_PER_HTBL_RECORD;
1011     rc = tdbio_read_record( hashrec, rec, RECTYPE_HTBL );
1012     if( rc ) {
1013         log_error( db_name, "lookup_hashtable failed: %s\n", g10_errstr(rc) );
1014         return rc;
1015     }
1016
1017     item = rec->r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
1018     if( !item )
1019         return -1; /* not found */
1020
1021     rc = tdbio_read_record( item, rec, 0 );
1022     if( rc ) {
1023         log_error( db_name, "hashtable read failed: %s\n", g10_errstr(rc) );
1024         return rc;
1025     }
1026     if( rec->rectype == RECTYPE_HTBL ) {
1027         hashrec = item;
1028         level++;
1029         if( level >= keylen ) {
1030             log_error( db_name, "hashtable has invalid indirections\n");
1031             return G10ERR_TRUSTDB;
1032         }
1033         goto next_level;
1034     }
1035     else if( rec->rectype == RECTYPE_HLST ) {
1036         for(;;) {
1037             int i;
1038
1039             for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1040                 if( rec->r.hlst.rnum[i] ) {
1041                     TRUSTREC tmp;
1042
1043                     rc = tdbio_read_record( rec->r.hlst.rnum[i], &tmp, 0 );
1044                     if( rc ) {
1045                         log_error( "lookup_hashtable: read item failed: %s\n",
1046                                                               g10_errstr(rc) );
1047                         return rc;
1048                     }
1049                     if( (*cmpfnc)( cmpdata, &tmp ) ) {
1050                         *rec = tmp;
1051                         return 0;
1052                     }
1053                 }
1054             }
1055             if( rec->r.hlst.next ) {
1056                 rc = tdbio_read_record( rec->r.hlst.next, rec, RECTYPE_HLST );
1057                 if( rc ) {
1058                     log_error( "lookup_hashtable: read hlst failed: %s\n",
1059                                                          g10_errstr(rc) );
1060                     return rc;
1061                 }
1062             }
1063             else
1064                 return -1; /* not found */
1065         }
1066     }
1067
1068
1069     if( (*cmpfnc)( cmpdata, rec ) )
1070         return 0; /* really found */
1071
1072     return -1; /* no: not found */
1073 }
1074
1075
1076 /****************
1077  * Update the trust hashtbl or create the table if it does not exist
1078  */
1079 static int
1080 update_trusthashtbl( TRUSTREC *tr )
1081 {
1082     return upd_hashtable( get_trusthashrec(),
1083                           tr->r.trust.fingerprint, 20, tr->recnum );
1084 }
1085
1086
1087
1088 void
1089 tdbio_dump_record( TRUSTREC *rec, FILE *fp  )
1090 {
1091     int i;
1092     ulong rnum = rec->recnum;
1093
1094     fprintf(fp, "rec %5lu, ", rnum );
1095
1096     switch( rec->rectype ) {
1097       case 0: fprintf(fp, "blank\n");
1098         break;
1099       case RECTYPE_VER: fprintf(fp,
1100             "version, td=%lu, f=%lu, m/c/d=%d/%d/%d tm=%d nc=%lu (%s)\n",
1101                                    rec->r.ver.trusthashtbl,
1102                                    rec->r.ver.firstfree,
1103                                    rec->r.ver.marginals,
1104                                    rec->r.ver.completes,
1105                                    rec->r.ver.cert_depth,
1106                                    rec->r.ver.trust_model,
1107                                    rec->r.ver.nextcheck,
1108                                    strtimestamp(rec->r.ver.nextcheck)
1109                                  );
1110         break;
1111       case RECTYPE_FREE: fprintf(fp, "free, next=%lu\n", rec->r.free.next );
1112         break;
1113       case RECTYPE_HTBL:
1114         fprintf(fp, "htbl,");
1115         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ )
1116             fprintf(fp, " %lu", rec->r.htbl.item[i] );
1117         putc('\n', fp);
1118         break;
1119       case RECTYPE_HLST:
1120         fprintf(fp, "hlst, next=%lu,", rec->r.hlst.next );
1121         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ )
1122             fprintf(fp, " %lu", rec->r.hlst.rnum[i] );
1123         putc('\n', fp);
1124         break;
1125       case RECTYPE_TRUST:
1126         fprintf(fp, "trust ");
1127         for(i=0; i < 20; i++ )
1128             fprintf(fp, "%02X", rec->r.trust.fingerprint[i] );
1129         fprintf (fp, ", ot=%d, d=%d, vl=%lu\n", rec->r.trust.ownertrust,
1130                  rec->r.trust.depth, rec->r.trust.validlist);
1131         break;
1132       case RECTYPE_VALID:
1133         fprintf(fp, "valid ");
1134         for(i=0; i < 20; i++ )
1135             fprintf(fp, "%02X", rec->r.valid.namehash[i] );
1136         fprintf (fp, ", v=%d, next=%lu\n", rec->r.valid.validity,
1137                  rec->r.valid.next);
1138         break;
1139       default:
1140         fprintf(fp, "unknown type %d\n", rec->rectype );
1141         break;
1142     }
1143 }
1144
1145 /****************
1146  * read the record with number recnum
1147  * returns: -1 on error, 0 on success
1148  */
1149 int
1150 tdbio_read_record( ulong recnum, TRUSTREC *rec, int expected )
1151 {
1152     byte readbuf[TRUST_RECORD_LEN];
1153     const byte *buf, *p;
1154     int rc = 0;
1155     int n, i;
1156
1157     if( db_fd == -1 )
1158         open_db();
1159     buf = get_record_from_cache( recnum );
1160     if( !buf ) {
1161         if( lseek( db_fd, recnum * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
1162             log_error(_("trustdb: lseek failed: %s\n"), strerror(errno) );
1163             return G10ERR_READ_FILE;
1164         }
1165         n = read( db_fd, readbuf, TRUST_RECORD_LEN);
1166         if( !n ) {
1167             return -1; /* eof */
1168         }
1169         else if( n != TRUST_RECORD_LEN ) {
1170             log_error(_("trustdb: read failed (n=%d): %s\n"), n,
1171                                                         strerror(errno) );
1172             return G10ERR_READ_FILE;
1173         }
1174         buf = readbuf;
1175     }
1176     rec->recnum = recnum;
1177     rec->dirty = 0;
1178     p = buf;
1179     rec->rectype = *p++;
1180     if( expected && rec->rectype != expected ) {
1181         log_error("%lu: read expected rec type %d, got %d\n",
1182                     recnum, expected, rec->rectype );
1183         return G10ERR_TRUSTDB;
1184     }
1185     p++;    /* skip reserved byte */
1186     switch( rec->rectype ) {
1187       case 0:  /* unused (free) record */
1188         break;
1189       case RECTYPE_VER: /* version record */
1190         if( memcmp(buf+1, "gpg", 3 ) ) {
1191             log_error( _("%s: not a trustdb file\n"), db_name );
1192             rc = G10ERR_TRUSTDB;
1193         }
1194         p += 2; /* skip "gpg" */
1195         rec->r.ver.version  = *p++;
1196         rec->r.ver.marginals = *p++;
1197         rec->r.ver.completes = *p++;
1198         rec->r.ver.cert_depth = *p++;
1199         rec->r.ver.trust_model = *p++;
1200         p += 3;
1201         rec->r.ver.created  = buftoulong(p); p += 4;
1202         rec->r.ver.nextcheck = buftoulong(p); p += 4;
1203         p += 4;
1204         p += 4;
1205         rec->r.ver.firstfree =buftoulong(p); p += 4;
1206         p += 4;
1207         rec->r.ver.trusthashtbl =buftoulong(p); p += 4;
1208         if( recnum ) {
1209             log_error( _("%s: version record with recnum %lu\n"), db_name,
1210                                                              (ulong)recnum );
1211             rc = G10ERR_TRUSTDB;
1212         }
1213         else if( rec->r.ver.version != 3 ) {
1214             log_error( _("%s: invalid file version %d\n"), db_name,
1215                                                         rec->r.ver.version );
1216             rc = G10ERR_TRUSTDB;
1217         }
1218         break;
1219       case RECTYPE_FREE:
1220         rec->r.free.next  = buftoulong(p); p += 4;
1221         break;
1222       case RECTYPE_HTBL:
1223         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ ) {
1224             rec->r.htbl.item[i] = buftoulong(p); p += 4;
1225         }
1226         break;
1227       case RECTYPE_HLST:
1228         rec->r.hlst.next = buftoulong(p); p += 4;
1229         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1230             rec->r.hlst.rnum[i] = buftoulong(p); p += 4;
1231         }
1232         break;
1233       case RECTYPE_TRUST:
1234         memcpy( rec->r.trust.fingerprint, p, 20); p+=20;
1235         rec->r.trust.ownertrust = *p++;
1236         rec->r.trust.depth = *p++;
1237         rec->r.trust.min_ownertrust = *p++;
1238         p++;
1239         rec->r.trust.validlist = buftoulong(p); p += 4;
1240         break;
1241       case RECTYPE_VALID:
1242         memcpy( rec->r.valid.namehash, p, 20); p+=20;
1243         rec->r.valid.validity = *p++;
1244         rec->r.valid.next = buftoulong(p); p += 4;
1245         rec->r.valid.full_count = *p++;
1246         rec->r.valid.marginal_count = *p++;
1247         break;
1248       default:
1249         log_error( "%s: invalid record type %d at recnum %lu\n",
1250                                    db_name, rec->rectype, (ulong)recnum );
1251         rc = G10ERR_TRUSTDB;
1252         break;
1253     }
1254
1255     return rc;
1256 }
1257
1258 /****************
1259  * Write the record at RECNUM
1260  */
1261 int
1262 tdbio_write_record( TRUSTREC *rec )
1263 {
1264     byte buf[TRUST_RECORD_LEN], *p;
1265     int rc = 0;
1266     int i;
1267     ulong recnum = rec->recnum;
1268
1269     if( db_fd == -1 )
1270         open_db();
1271
1272     memset(buf, 0, TRUST_RECORD_LEN);
1273     p = buf;
1274     *p++ = rec->rectype; p++;
1275     switch( rec->rectype ) {
1276       case 0:  /* unused record */
1277         break;
1278       case RECTYPE_VER: /* version record */
1279         if( recnum )
1280             BUG();
1281         memcpy(p-1, "gpg", 3 ); p += 2;
1282         *p++ = rec->r.ver.version;
1283         *p++ = rec->r.ver.marginals;
1284         *p++ = rec->r.ver.completes;
1285         *p++ = rec->r.ver.cert_depth;
1286         *p++ = rec->r.ver.trust_model;
1287         p += 3;
1288         ulongtobuf(p, rec->r.ver.created); p += 4;
1289         ulongtobuf(p, rec->r.ver.nextcheck); p += 4;
1290         p += 4;
1291         p += 4;
1292         ulongtobuf(p, rec->r.ver.firstfree ); p += 4;
1293         p += 4;
1294         ulongtobuf(p, rec->r.ver.trusthashtbl ); p += 4;
1295         break;
1296
1297       case RECTYPE_FREE:
1298         ulongtobuf(p, rec->r.free.next); p += 4;
1299         break;
1300
1301
1302       case RECTYPE_HTBL:
1303         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ ) {
1304             ulongtobuf( p, rec->r.htbl.item[i]); p += 4;
1305         }
1306         break;
1307
1308       case RECTYPE_HLST:
1309         ulongtobuf( p, rec->r.hlst.next); p += 4;
1310         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1311             ulongtobuf( p, rec->r.hlst.rnum[i]); p += 4;
1312         }
1313         break;
1314
1315       case RECTYPE_TRUST:
1316         memcpy( p, rec->r.trust.fingerprint, 20); p += 20;
1317         *p++ = rec->r.trust.ownertrust;
1318         *p++ = rec->r.trust.depth;
1319         *p++ = rec->r.trust.min_ownertrust;
1320         p++;
1321         ulongtobuf( p, rec->r.trust.validlist); p += 4;
1322         break;
1323
1324       case RECTYPE_VALID:
1325         memcpy( p, rec->r.valid.namehash, 20); p += 20;
1326         *p++ = rec->r.valid.validity;
1327         ulongtobuf( p, rec->r.valid.next); p += 4;
1328         *p++ = rec->r.valid.full_count;
1329         *p++ = rec->r.valid.marginal_count;
1330         break;
1331
1332       default:
1333         BUG();
1334     }
1335
1336     rc = put_record_into_cache( recnum, buf );
1337     if( rc )
1338         ;
1339     else if( rec->rectype == RECTYPE_TRUST )
1340         rc = update_trusthashtbl( rec );
1341
1342     return rc;
1343 }
1344
1345 int
1346 tdbio_delete_record( ulong recnum )
1347 {
1348     TRUSTREC vr, rec;
1349     int rc;
1350
1351     /* Must read the record fist, so we can drop it from the hash tables */
1352     rc = tdbio_read_record( recnum, &rec, 0 );
1353     if( rc )
1354         ;
1355     else if( rec.rectype == RECTYPE_TRUST ) {
1356          rc = drop_from_hashtable( get_trusthashrec(),
1357                                    rec.r.trust.fingerprint, 20, rec.recnum );
1358     }
1359
1360     if( rc )
1361         return rc;
1362
1363     /* now we can chnage it to a free record */
1364     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
1365     if( rc )
1366         log_fatal( _("%s: error reading version record: %s\n"),
1367                                        db_name, g10_errstr(rc) );
1368
1369     rec.recnum = recnum;
1370     rec.rectype = RECTYPE_FREE;
1371     rec.r.free.next = vr.r.ver.firstfree;
1372     vr.r.ver.firstfree = recnum;
1373     rc = tdbio_write_record( &rec );
1374     if( !rc )
1375         rc = tdbio_write_record( &vr );
1376     return rc;
1377 }
1378
1379 /****************
1380  * create a new record and return its record number
1381  */
1382 ulong
1383 tdbio_new_recnum()
1384 {
1385     off_t offset;
1386     ulong recnum;
1387     TRUSTREC vr, rec;
1388     int rc;
1389
1390     /* look for unused records */
1391     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
1392     if( rc )
1393         log_fatal( _("%s: error reading version record: %s\n"),
1394                                              db_name, g10_errstr(rc) );
1395     if( vr.r.ver.firstfree ) {
1396         recnum = vr.r.ver.firstfree;
1397         rc = tdbio_read_record( recnum, &rec, RECTYPE_FREE );
1398         if( rc ) {
1399             log_error( _("%s: error reading free record: %s\n"),
1400                                                   db_name,  g10_errstr(rc) );
1401             return rc;
1402         }
1403         /* update dir record */
1404         vr.r.ver.firstfree = rec.r.free.next;
1405         rc = tdbio_write_record( &vr );
1406         if( rc ) {
1407             log_error( _("%s: error writing dir record: %s\n"),
1408                                                      db_name, g10_errstr(rc) );
1409             return rc;
1410         }
1411         /*zero out the new record */
1412         memset( &rec, 0, sizeof rec );
1413         rec.rectype = 0; /* unused record */
1414         rec.recnum = recnum;
1415         rc = tdbio_write_record( &rec );
1416         if( rc )
1417             log_fatal(_("%s: failed to zero a record: %s\n"),
1418                                        db_name, g10_errstr(rc));
1419     }
1420     else { /* not found, append a new record */
1421         offset = lseek( db_fd, 0, SEEK_END );
1422         if( offset == -1 )
1423             log_fatal("trustdb: lseek to end failed: %s\n", strerror(errno) );
1424         recnum = offset / TRUST_RECORD_LEN;
1425         assert(recnum); /* this is will never be the first record */
1426         /* we must write a record, so that the next call to this function
1427          * returns another recnum */
1428         memset( &rec, 0, sizeof rec );
1429         rec.rectype = 0; /* unused record */
1430         rec.recnum = recnum;
1431         rc = 0;
1432         if( lseek( db_fd, recnum * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
1433             log_error(_("trustdb rec %lu: lseek failed: %s\n"),
1434                                                 recnum, strerror(errno) );
1435             rc = G10ERR_WRITE_FILE;
1436         }
1437         else {
1438             int n = write( db_fd, &rec, TRUST_RECORD_LEN);
1439             if( n != TRUST_RECORD_LEN ) {
1440                 log_error(_("trustdb rec %lu: write failed (n=%d): %s\n"),
1441                                                  recnum, n, strerror(errno) );
1442                 rc = G10ERR_WRITE_FILE;
1443             }
1444         }
1445
1446         if( rc )
1447             log_fatal(_("%s: failed to append a record: %s\n"),
1448                                     db_name,    g10_errstr(rc));
1449     }
1450     return recnum ;
1451 }
1452
1453
1454
1455 static int
1456 cmp_trec_fpr ( void *fpr, const TRUSTREC *rec )
1457 {
1458     return rec->rectype == RECTYPE_TRUST
1459            && !memcmp( rec->r.trust.fingerprint, fpr, 20);
1460 }
1461
1462
1463 int
1464 tdbio_search_trust_byfpr( const byte *fingerprint, TRUSTREC *rec )
1465 {
1466     int rc;
1467
1468     /* locate the trust record using the hash table */
1469     rc = lookup_hashtable( get_trusthashrec(), fingerprint, 20,
1470                            cmp_trec_fpr, (void*)fingerprint, rec );
1471     return rc;
1472 }
1473
1474 int
1475 tdbio_search_trust_bypk (PKT_public_key *pk, TRUSTREC *rec)
1476 {
1477     byte fingerprint[MAX_FINGERPRINT_LEN];
1478     size_t fingerlen;
1479
1480     fingerprint_from_pk( pk, fingerprint, &fingerlen );
1481     for (; fingerlen < 20; fingerlen++ )
1482       fingerprint[fingerlen] = 0;
1483     return tdbio_search_trust_byfpr (fingerprint, rec);
1484 }
1485
1486
1487
1488 void
1489 tdbio_invalid(void)
1490 {
1491     log_error(_(
1492         "the trustdb is corrupted; please run \"gpg --fix-trustdb\".\n") );
1493     g10_exit(2);
1494 }
1495
1496 /*
1497  * Migrate the trustdb as just up to gpg 1.0.6 (trustdb version 2)
1498  * to the 2.1 version as used with 1.0.6b - This is pretty trivial as needs
1499  * only to scan the tdb and insert new the new trust records.  The old ones are
1500  * obsolte from now on
1501  */
1502 static void
1503 migrate_from_v2 ()
1504 {
1505   TRUSTREC rec;
1506   int i, n;
1507   struct {
1508     ulong keyrecno;
1509     byte  ot;
1510     byte okay;
1511     byte  fpr[20];
1512   } *ottable;
1513   int ottable_size, ottable_used;
1514   byte oldbuf[40];
1515   ulong recno;
1516   int rc, count;
1517
1518   ottable_size = 5;
1519   ottable = m_alloc (ottable_size * sizeof *ottable);
1520   ottable_used = 0;
1521
1522   /* We have some restrictions here.  We can't use the version record
1523    * and we can't use any of the old hashtables because we dropped the
1524    * code.  So we first collect all ownertrusts and then use a second
1525    * pass fo find the associated keys.  We have to do this all without using 
1526    * the regular record read functions.
1527    */
1528
1529   /* get all the ownertrusts */
1530   if (lseek (db_fd, 0, SEEK_SET ) == -1 ) 
1531       log_fatal ("migrate_from_v2: lseek failed: %s\n", strerror (errno));
1532   for (recno=0;;recno++)
1533     {
1534       do
1535         n = read (db_fd, oldbuf, 40);
1536       while (n==-1 && errno == EINTR);
1537       if (!n)
1538         break; /* eof */
1539       if (n != 40)
1540         log_fatal ("migrate_vfrom_v2: read error or short read\n");
1541
1542       if (*oldbuf != 2)
1543         continue;
1544       
1545       /* v2 dir record */
1546       if (ottable_used == ottable_size)
1547         {
1548           ottable_size += 1000;
1549           ottable = m_realloc (ottable, ottable_size * sizeof *ottable);
1550         }
1551       ottable[ottable_used].keyrecno = buftoulong (oldbuf+6);
1552       ottable[ottable_used].ot = oldbuf[18];
1553       ottable[ottable_used].okay = 0;
1554       memset (ottable[ottable_used].fpr,0, 20);
1555       if (ottable[ottable_used].keyrecno && ottable[ottable_used].ot)
1556         ottable_used++;
1557     }
1558   log_info ("found %d ownertrust records\n", ottable_used);
1559
1560   /* Read again and find the fingerprints */
1561   if (lseek (db_fd, 0, SEEK_SET ) == -1 ) 
1562       log_fatal ("migrate_from_v2: lseek failed: %s\n", strerror (errno));
1563   for (recno=0;;recno++)
1564     {
1565       do
1566         n = read (db_fd, oldbuf, 40);
1567       while (n==-1 && errno == EINTR);
1568       if (!n)
1569         break; /* eof */
1570       if (n != 40)
1571         log_fatal ("migrate_from_v2: read error or short read\n");
1572
1573       if (*oldbuf != 3) 
1574         continue;
1575
1576       /* v2 key record */
1577       for (i=0; i < ottable_used; i++)
1578         {
1579           if (ottable[i].keyrecno == recno)
1580             {
1581               memcpy (ottable[i].fpr, oldbuf+20, 20);
1582               ottable[i].okay = 1;
1583               break;
1584             }
1585         }
1586     }
1587
1588   /* got everything - create the v3 trustdb */
1589   if (ftruncate (db_fd, 0))
1590     log_fatal ("can't truncate `%s': %s\n", db_name, strerror (errno) );
1591   if (create_version_record ())
1592     log_fatal ("failed to recreate version record of `%s'\n", db_name);
1593
1594   /* access the hash table, so it is store just after the version record, 
1595    * this is not needed put a dump is more pretty */
1596   get_trusthashrec ();
1597
1598   /* And insert the old ownertrust values */
1599   count = 0;
1600   for (i=0; i < ottable_used; i++)
1601     {
1602       if (!ottable[i].okay)
1603         continue;
1604       
1605       memset (&rec, 0, sizeof rec);
1606       rec.recnum = tdbio_new_recnum ();
1607       rec.rectype = RECTYPE_TRUST;
1608       memcpy(rec.r.trust.fingerprint, ottable[i].fpr, 20);
1609       rec.r.trust.ownertrust = ottable[i].ot;
1610       if (tdbio_write_record (&rec))
1611         log_fatal ("failed to write trust record of `%s'\n", db_name);
1612       count++;
1613     }
1614
1615   revalidation_mark ();
1616   rc = tdbio_sync ();
1617   if (rc)
1618     log_fatal ("failed to sync `%s'\n", db_name);
1619   log_info ("migrated %d version 2 ownertrusts\n", count);
1620   m_free (ottable);
1621 }
1622
1623
1624