* parse-packet.c (parse_signature): No need to reserve 8 bytes for the
[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( db_name, "upd_hashtable: read failed: %s\n",
778                                                         g10_errstr(rc) );
779         return rc;
780     }
781
782     item = rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
783     if( !item ) { /* insert a new item into the hash table */
784         rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = newrecnum;
785         rc = tdbio_write_record( &rec );
786         if( rc ) {
787             log_error( db_name, "upd_hashtable: write htbl failed: %s\n",
788                                                             g10_errstr(rc) );
789             return rc;
790         }
791     }
792     else if( item != newrecnum ) {  /* must do an update */
793         lastrec = rec;
794         rc = tdbio_read_record( item, &rec, 0 );
795         if( rc ) {
796             log_error( "upd_hashtable: read item failed: %s\n",
797                                                             g10_errstr(rc) );
798             return rc;
799         }
800
801         if( rec.rectype == RECTYPE_HTBL ) {
802             hashrec = item;
803             level++;
804             if( level >= keylen ) {
805                 log_error( "hashtable has invalid indirections.\n");
806                 return G10ERR_TRUSTDB;
807             }
808             goto next_level;
809         }
810         else if( rec.rectype == RECTYPE_HLST ) { /* extend list */
811             /* see whether the key is already in this list */
812             for(;;) {
813                 for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
814                     if( rec.r.hlst.rnum[i] == newrecnum ) {
815                         return 0; /* okay, already in the list */
816                     }
817                 }
818                 if( rec.r.hlst.next ) {
819                     rc = tdbio_read_record( rec.r.hlst.next,
820                                                        &rec, RECTYPE_HLST);
821                     if( rc ) {
822                         log_error( "upd_hashtable: read hlst failed: %s\n",
823                                                              g10_errstr(rc) );
824                         return rc;
825                     }
826                 }
827                 else
828                     break; /* not there */
829             }
830             /* find the next free entry and put it in */
831             for(;;) {
832                 for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
833                     if( !rec.r.hlst.rnum[i] ) {
834                         rec.r.hlst.rnum[i] = newrecnum;
835                         rc = tdbio_write_record( &rec );
836                         if( rc )
837                             log_error( "upd_hashtable: write hlst failed: %s\n",
838                                                               g10_errstr(rc) );
839                         return rc; /* done */
840                     }
841                 }
842                 if( rec.r.hlst.next ) {
843                     rc = tdbio_read_record( rec.r.hlst.next,
844                                                       &rec, RECTYPE_HLST );
845                     if( rc ) {
846                         log_error( "upd_hashtable: read hlst failed: %s\n",
847                                                              g10_errstr(rc) );
848                         return rc;
849                     }
850                 }
851                 else { /* add a new list record */
852                     rec.r.hlst.next = item = tdbio_new_recnum();
853                     rc = tdbio_write_record( &rec );
854                     if( rc ) {
855                         log_error( "upd_hashtable: write hlst failed: %s\n",
856                                                           g10_errstr(rc) );
857                         return rc;
858                     }
859                     memset( &rec, 0, sizeof rec );
860                     rec.rectype = RECTYPE_HLST;
861                     rec.recnum = item;
862                     rec.r.hlst.rnum[0] = newrecnum;
863                     rc = tdbio_write_record( &rec );
864                     if( rc )
865                         log_error( "upd_hashtable: write ext hlst failed: %s\n",
866                                                           g10_errstr(rc) );
867                     return rc; /* done */
868                 }
869             } /* end loop over hlst slots */
870         }
871         else if( rec.rectype == RECTYPE_TRUST ) { /* insert a list record */
872             if( rec.recnum == newrecnum ) {
873                 return 0;
874             }
875             item = rec.recnum; /* save number of key record */
876             memset( &rec, 0, sizeof rec );
877             rec.rectype = RECTYPE_HLST;
878             rec.recnum = tdbio_new_recnum();
879             rec.r.hlst.rnum[0] = item;       /* old keyrecord */
880             rec.r.hlst.rnum[1] = newrecnum; /* and new one */
881             rc = tdbio_write_record( &rec );
882             if( rc ) {
883                 log_error( "upd_hashtable: write new hlst failed: %s\n",
884                                                   g10_errstr(rc) );
885                 return rc;
886             }
887             /* update the hashtable record */
888             lastrec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = rec.recnum;
889             rc = tdbio_write_record( &lastrec );
890             if( rc )
891                 log_error( "upd_hashtable: update htbl failed: %s\n",
892                                                              g10_errstr(rc) );
893             return rc; /* ready */
894         }
895         else {
896             log_error( "hashtbl %lu: %lu/%d points to an invalid record %lu\n",
897                        table, hashrec, (msb % ITEMS_PER_HTBL_RECORD), item);
898             list_trustdb(NULL);
899             return G10ERR_TRUSTDB;
900         }
901     }
902
903     return 0;
904 }
905
906
907 /****************
908  * Drop an entry from a hashtable
909  * table gives the start of the table, key and keylen is the key,
910  */
911 static int
912 drop_from_hashtable( ulong table, byte *key, int keylen, ulong recnum )
913 {
914     TRUSTREC rec;
915     ulong hashrec, item;
916     int msb;
917     int level=0;
918     int rc, i;
919
920     hashrec = table;
921   next_level:
922     msb = key[level];
923     hashrec += msb / ITEMS_PER_HTBL_RECORD;
924     rc = tdbio_read_record( hashrec, &rec, RECTYPE_HTBL );
925     if( rc ) {
926         log_error( db_name, "drop_from_hashtable: read failed: %s\n",
927                                                         g10_errstr(rc) );
928         return rc;
929     }
930
931     item = rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
932     if( !item )  /* not found - forget about it  */
933         return 0;
934
935     if( item == recnum ) {  /* tables points direct to the record */
936         rec.r.htbl.item[msb % ITEMS_PER_HTBL_RECORD] = 0;
937         rc = tdbio_write_record( &rec );
938         if( rc )
939             log_error( db_name, "drop_from_hashtable: write htbl failed: %s\n",
940                                                             g10_errstr(rc) );
941         return rc;
942     }
943
944     rc = tdbio_read_record( item, &rec, 0 );
945     if( rc ) {
946         log_error( "drop_from_hashtable: read item failed: %s\n",
947                                                         g10_errstr(rc) );
948         return rc;
949     }
950
951     if( rec.rectype == RECTYPE_HTBL ) {
952         hashrec = item;
953         level++;
954         if( level >= keylen ) {
955             log_error( "hashtable has invalid indirections.\n");
956             return G10ERR_TRUSTDB;
957         }
958         goto next_level;
959     }
960
961     if( rec.rectype == RECTYPE_HLST ) {
962         for(;;) {
963             for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
964                 if( rec.r.hlst.rnum[i] == recnum ) {
965                     rec.r.hlst.rnum[i] = 0; /* drop */
966                     rc = tdbio_write_record( &rec );
967                     if( rc )
968                         log_error( db_name, "drop_from_hashtable: write htbl failed: %s\n",
969                                                                         g10_errstr(rc) );
970                     return rc;
971                 }
972             }
973             if( rec.r.hlst.next ) {
974                 rc = tdbio_read_record( rec.r.hlst.next,
975                                                    &rec, RECTYPE_HLST);
976                 if( rc ) {
977                     log_error( "drop_from_hashtable: read hlst failed: %s\n",
978                                                          g10_errstr(rc) );
979                     return rc;
980                 }
981             }
982             else
983                 return 0; /* key not in table */
984         }
985     }
986
987     log_error( "hashtbl %lu: %lu/%d points to wrong record %lu\n",
988                     table, hashrec, (msb % ITEMS_PER_HTBL_RECORD), item);
989     return G10ERR_TRUSTDB;
990 }
991
992
993
994 /****************
995  * Lookup a record via the hashtable tablewith key/keylen and return the
996  * result in rec.  cmp() should return if the record is the desired one.
997  * Returns -1 if not found, 0 if found or another errocode
998  */
999 static int
1000 lookup_hashtable( ulong table, const byte *key, size_t keylen,
1001                   int (*cmpfnc)(void*, const TRUSTREC *), void *cmpdata,
1002                                                 TRUSTREC *rec )
1003 {
1004     int rc;
1005     ulong hashrec, item;
1006     int msb;
1007     int level=0;
1008
1009     hashrec = table;
1010   next_level:
1011     msb = key[level];
1012     hashrec += msb / ITEMS_PER_HTBL_RECORD;
1013     rc = tdbio_read_record( hashrec, rec, RECTYPE_HTBL );
1014     if( rc ) {
1015         log_error( db_name, "lookup_hashtable failed: %s\n", g10_errstr(rc) );
1016         return rc;
1017     }
1018
1019     item = rec->r.htbl.item[msb % ITEMS_PER_HTBL_RECORD];
1020     if( !item )
1021         return -1; /* not found */
1022
1023     rc = tdbio_read_record( item, rec, 0 );
1024     if( rc ) {
1025         log_error( db_name, "hashtable read failed: %s\n", g10_errstr(rc) );
1026         return rc;
1027     }
1028     if( rec->rectype == RECTYPE_HTBL ) {
1029         hashrec = item;
1030         level++;
1031         if( level >= keylen ) {
1032             log_error( db_name, "hashtable has invalid indirections\n");
1033             return G10ERR_TRUSTDB;
1034         }
1035         goto next_level;
1036     }
1037     else if( rec->rectype == RECTYPE_HLST ) {
1038         for(;;) {
1039             int i;
1040
1041             for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1042                 if( rec->r.hlst.rnum[i] ) {
1043                     TRUSTREC tmp;
1044
1045                     rc = tdbio_read_record( rec->r.hlst.rnum[i], &tmp, 0 );
1046                     if( rc ) {
1047                         log_error( "lookup_hashtable: read item failed: %s\n",
1048                                                               g10_errstr(rc) );
1049                         return rc;
1050                     }
1051                     if( (*cmpfnc)( cmpdata, &tmp ) ) {
1052                         *rec = tmp;
1053                         return 0;
1054                     }
1055                 }
1056             }
1057             if( rec->r.hlst.next ) {
1058                 rc = tdbio_read_record( rec->r.hlst.next, rec, RECTYPE_HLST );
1059                 if( rc ) {
1060                     log_error( "lookup_hashtable: read hlst failed: %s\n",
1061                                                          g10_errstr(rc) );
1062                     return rc;
1063                 }
1064             }
1065             else
1066                 return -1; /* not found */
1067         }
1068     }
1069
1070
1071     if( (*cmpfnc)( cmpdata, rec ) )
1072         return 0; /* really found */
1073
1074     return -1; /* no: not found */
1075 }
1076
1077
1078 /****************
1079  * Update the trust hashtbl or create the table if it does not exist
1080  */
1081 static int
1082 update_trusthashtbl( TRUSTREC *tr )
1083 {
1084     return upd_hashtable( get_trusthashrec(),
1085                           tr->r.trust.fingerprint, 20, tr->recnum );
1086 }
1087
1088
1089
1090 void
1091 tdbio_dump_record( TRUSTREC *rec, FILE *fp  )
1092 {
1093     int i;
1094     ulong rnum = rec->recnum;
1095
1096     fprintf(fp, "rec %5lu, ", rnum );
1097
1098     switch( rec->rectype ) {
1099       case 0: fprintf(fp, "blank\n");
1100         break;
1101       case RECTYPE_VER: fprintf(fp,
1102             "version, td=%lu, f=%lu, m/c/d=%d/%d/%d tm=%d nc=%lu (%s)\n",
1103                                    rec->r.ver.trusthashtbl,
1104                                    rec->r.ver.firstfree,
1105                                    rec->r.ver.marginals,
1106                                    rec->r.ver.completes,
1107                                    rec->r.ver.cert_depth,
1108                                    rec->r.ver.trust_model,
1109                                    rec->r.ver.nextcheck,
1110                                    strtimestamp(rec->r.ver.nextcheck)
1111                                  );
1112         break;
1113       case RECTYPE_FREE: fprintf(fp, "free, next=%lu\n", rec->r.free.next );
1114         break;
1115       case RECTYPE_HTBL:
1116         fprintf(fp, "htbl,");
1117         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ )
1118             fprintf(fp, " %lu", rec->r.htbl.item[i] );
1119         putc('\n', fp);
1120         break;
1121       case RECTYPE_HLST:
1122         fprintf(fp, "hlst, next=%lu,", rec->r.hlst.next );
1123         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ )
1124             fprintf(fp, " %lu", rec->r.hlst.rnum[i] );
1125         putc('\n', fp);
1126         break;
1127       case RECTYPE_TRUST:
1128         fprintf(fp, "trust ");
1129         for(i=0; i < 20; i++ )
1130             fprintf(fp, "%02X", rec->r.trust.fingerprint[i] );
1131         fprintf (fp, ", ot=%d, d=%d, vl=%lu\n", rec->r.trust.ownertrust,
1132                  rec->r.trust.depth, rec->r.trust.validlist);
1133         break;
1134       case RECTYPE_VALID:
1135         fprintf(fp, "valid ");
1136         for(i=0; i < 20; i++ )
1137             fprintf(fp, "%02X", rec->r.valid.namehash[i] );
1138         fprintf (fp, ", v=%d, next=%lu\n", rec->r.valid.validity,
1139                  rec->r.valid.next);
1140         break;
1141       default:
1142         fprintf(fp, "unknown type %d\n", rec->rectype );
1143         break;
1144     }
1145 }
1146
1147 /****************
1148  * read the record with number recnum
1149  * returns: -1 on error, 0 on success
1150  */
1151 int
1152 tdbio_read_record( ulong recnum, TRUSTREC *rec, int expected )
1153 {
1154     byte readbuf[TRUST_RECORD_LEN];
1155     const byte *buf, *p;
1156     int rc = 0;
1157     int n, i;
1158
1159     if( db_fd == -1 )
1160         open_db();
1161     buf = get_record_from_cache( recnum );
1162     if( !buf ) {
1163         if( lseek( db_fd, recnum * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
1164             log_error(_("trustdb: lseek failed: %s\n"), strerror(errno) );
1165             return G10ERR_READ_FILE;
1166         }
1167         n = read( db_fd, readbuf, TRUST_RECORD_LEN);
1168         if( !n ) {
1169             return -1; /* eof */
1170         }
1171         else if( n != TRUST_RECORD_LEN ) {
1172             log_error(_("trustdb: read failed (n=%d): %s\n"), n,
1173                                                         strerror(errno) );
1174             return G10ERR_READ_FILE;
1175         }
1176         buf = readbuf;
1177     }
1178     rec->recnum = recnum;
1179     rec->dirty = 0;
1180     p = buf;
1181     rec->rectype = *p++;
1182     if( expected && rec->rectype != expected ) {
1183         log_error("%lu: read expected rec type %d, got %d\n",
1184                     recnum, expected, rec->rectype );
1185         return G10ERR_TRUSTDB;
1186     }
1187     p++;    /* skip reserved byte */
1188     switch( rec->rectype ) {
1189       case 0:  /* unused (free) record */
1190         break;
1191       case RECTYPE_VER: /* version record */
1192         if( memcmp(buf+1, "gpg", 3 ) ) {
1193             log_error( _("%s: not a trustdb file\n"), db_name );
1194             rc = G10ERR_TRUSTDB;
1195         }
1196         p += 2; /* skip "gpg" */
1197         rec->r.ver.version  = *p++;
1198         rec->r.ver.marginals = *p++;
1199         rec->r.ver.completes = *p++;
1200         rec->r.ver.cert_depth = *p++;
1201         rec->r.ver.trust_model = *p++;
1202         p += 3;
1203         rec->r.ver.created  = buftoulong(p); p += 4;
1204         rec->r.ver.nextcheck = buftoulong(p); p += 4;
1205         p += 4;
1206         p += 4;
1207         rec->r.ver.firstfree =buftoulong(p); p += 4;
1208         p += 4;
1209         rec->r.ver.trusthashtbl =buftoulong(p); p += 4;
1210         if( recnum ) {
1211             log_error( _("%s: version record with recnum %lu\n"), db_name,
1212                                                              (ulong)recnum );
1213             rc = G10ERR_TRUSTDB;
1214         }
1215         else if( rec->r.ver.version != 3 ) {
1216             log_error( _("%s: invalid file version %d\n"), db_name,
1217                                                         rec->r.ver.version );
1218             rc = G10ERR_TRUSTDB;
1219         }
1220         break;
1221       case RECTYPE_FREE:
1222         rec->r.free.next  = buftoulong(p); p += 4;
1223         break;
1224       case RECTYPE_HTBL:
1225         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ ) {
1226             rec->r.htbl.item[i] = buftoulong(p); p += 4;
1227         }
1228         break;
1229       case RECTYPE_HLST:
1230         rec->r.hlst.next = buftoulong(p); p += 4;
1231         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1232             rec->r.hlst.rnum[i] = buftoulong(p); p += 4;
1233         }
1234         break;
1235       case RECTYPE_TRUST:
1236         memcpy( rec->r.trust.fingerprint, p, 20); p+=20;
1237         rec->r.trust.ownertrust = *p++;
1238         rec->r.trust.depth = *p++;
1239         rec->r.trust.min_ownertrust = *p++;
1240         p++;
1241         rec->r.trust.validlist = buftoulong(p); p += 4;
1242         break;
1243       case RECTYPE_VALID:
1244         memcpy( rec->r.valid.namehash, p, 20); p+=20;
1245         rec->r.valid.validity = *p++;
1246         rec->r.valid.next = buftoulong(p); p += 4;
1247         rec->r.valid.full_count = *p++;
1248         rec->r.valid.marginal_count = *p++;
1249         break;
1250       default:
1251         log_error( "%s: invalid record type %d at recnum %lu\n",
1252                                    db_name, rec->rectype, (ulong)recnum );
1253         rc = G10ERR_TRUSTDB;
1254         break;
1255     }
1256
1257     return rc;
1258 }
1259
1260 /****************
1261  * Write the record at RECNUM
1262  */
1263 int
1264 tdbio_write_record( TRUSTREC *rec )
1265 {
1266     byte buf[TRUST_RECORD_LEN], *p;
1267     int rc = 0;
1268     int i;
1269     ulong recnum = rec->recnum;
1270
1271     if( db_fd == -1 )
1272         open_db();
1273
1274     memset(buf, 0, TRUST_RECORD_LEN);
1275     p = buf;
1276     *p++ = rec->rectype; p++;
1277     switch( rec->rectype ) {
1278       case 0:  /* unused record */
1279         break;
1280       case RECTYPE_VER: /* version record */
1281         if( recnum )
1282             BUG();
1283         memcpy(p-1, "gpg", 3 ); p += 2;
1284         *p++ = rec->r.ver.version;
1285         *p++ = rec->r.ver.marginals;
1286         *p++ = rec->r.ver.completes;
1287         *p++ = rec->r.ver.cert_depth;
1288         *p++ = rec->r.ver.trust_model;
1289         p += 3;
1290         ulongtobuf(p, rec->r.ver.created); p += 4;
1291         ulongtobuf(p, rec->r.ver.nextcheck); p += 4;
1292         p += 4;
1293         p += 4;
1294         ulongtobuf(p, rec->r.ver.firstfree ); p += 4;
1295         p += 4;
1296         ulongtobuf(p, rec->r.ver.trusthashtbl ); p += 4;
1297         break;
1298
1299       case RECTYPE_FREE:
1300         ulongtobuf(p, rec->r.free.next); p += 4;
1301         break;
1302
1303
1304       case RECTYPE_HTBL:
1305         for(i=0; i < ITEMS_PER_HTBL_RECORD; i++ ) {
1306             ulongtobuf( p, rec->r.htbl.item[i]); p += 4;
1307         }
1308         break;
1309
1310       case RECTYPE_HLST:
1311         ulongtobuf( p, rec->r.hlst.next); p += 4;
1312         for(i=0; i < ITEMS_PER_HLST_RECORD; i++ ) {
1313             ulongtobuf( p, rec->r.hlst.rnum[i]); p += 4;
1314         }
1315         break;
1316
1317       case RECTYPE_TRUST:
1318         memcpy( p, rec->r.trust.fingerprint, 20); p += 20;
1319         *p++ = rec->r.trust.ownertrust;
1320         *p++ = rec->r.trust.depth;
1321         *p++ = rec->r.trust.min_ownertrust;
1322         p++;
1323         ulongtobuf( p, rec->r.trust.validlist); p += 4;
1324         break;
1325
1326       case RECTYPE_VALID:
1327         memcpy( p, rec->r.valid.namehash, 20); p += 20;
1328         *p++ = rec->r.valid.validity;
1329         ulongtobuf( p, rec->r.valid.next); p += 4;
1330         *p++ = rec->r.valid.full_count;
1331         *p++ = rec->r.valid.marginal_count;
1332         break;
1333
1334       default:
1335         BUG();
1336     }
1337
1338     rc = put_record_into_cache( recnum, buf );
1339     if( rc )
1340         ;
1341     else if( rec->rectype == RECTYPE_TRUST )
1342         rc = update_trusthashtbl( rec );
1343
1344     return rc;
1345 }
1346
1347 int
1348 tdbio_delete_record( ulong recnum )
1349 {
1350     TRUSTREC vr, rec;
1351     int rc;
1352
1353     /* Must read the record fist, so we can drop it from the hash tables */
1354     rc = tdbio_read_record( recnum, &rec, 0 );
1355     if( rc )
1356         ;
1357     else if( rec.rectype == RECTYPE_TRUST ) {
1358          rc = drop_from_hashtable( get_trusthashrec(),
1359                                    rec.r.trust.fingerprint, 20, rec.recnum );
1360     }
1361
1362     if( rc )
1363         return rc;
1364
1365     /* now we can chnage it to a free record */
1366     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
1367     if( rc )
1368         log_fatal( _("%s: error reading version record: %s\n"),
1369                                        db_name, g10_errstr(rc) );
1370
1371     rec.recnum = recnum;
1372     rec.rectype = RECTYPE_FREE;
1373     rec.r.free.next = vr.r.ver.firstfree;
1374     vr.r.ver.firstfree = recnum;
1375     rc = tdbio_write_record( &rec );
1376     if( !rc )
1377         rc = tdbio_write_record( &vr );
1378     return rc;
1379 }
1380
1381 /****************
1382  * create a new record and return its record number
1383  */
1384 ulong
1385 tdbio_new_recnum()
1386 {
1387     off_t offset;
1388     ulong recnum;
1389     TRUSTREC vr, rec;
1390     int rc;
1391
1392     /* look for unused records */
1393     rc = tdbio_read_record( 0, &vr, RECTYPE_VER );
1394     if( rc )
1395         log_fatal( _("%s: error reading version record: %s\n"),
1396                                              db_name, g10_errstr(rc) );
1397     if( vr.r.ver.firstfree ) {
1398         recnum = vr.r.ver.firstfree;
1399         rc = tdbio_read_record( recnum, &rec, RECTYPE_FREE );
1400         if( rc ) {
1401             log_error( _("%s: error reading free record: %s\n"),
1402                                                   db_name,  g10_errstr(rc) );
1403             return rc;
1404         }
1405         /* update dir record */
1406         vr.r.ver.firstfree = rec.r.free.next;
1407         rc = tdbio_write_record( &vr );
1408         if( rc ) {
1409             log_error( _("%s: error writing dir record: %s\n"),
1410                                                      db_name, g10_errstr(rc) );
1411             return rc;
1412         }
1413         /*zero out the new record */
1414         memset( &rec, 0, sizeof rec );
1415         rec.rectype = 0; /* unused record */
1416         rec.recnum = recnum;
1417         rc = tdbio_write_record( &rec );
1418         if( rc )
1419             log_fatal(_("%s: failed to zero a record: %s\n"),
1420                                        db_name, g10_errstr(rc));
1421     }
1422     else { /* not found, append a new record */
1423         offset = lseek( db_fd, 0, SEEK_END );
1424         if( offset == -1 )
1425             log_fatal("trustdb: lseek to end failed: %s\n", strerror(errno) );
1426         recnum = offset / TRUST_RECORD_LEN;
1427         assert(recnum); /* this is will never be the first record */
1428         /* we must write a record, so that the next call to this function
1429          * returns another recnum */
1430         memset( &rec, 0, sizeof rec );
1431         rec.rectype = 0; /* unused record */
1432         rec.recnum = recnum;
1433         rc = 0;
1434         if( lseek( db_fd, recnum * TRUST_RECORD_LEN, SEEK_SET ) == -1 ) {
1435             log_error(_("trustdb rec %lu: lseek failed: %s\n"),
1436                                                 recnum, strerror(errno) );
1437             rc = G10ERR_WRITE_FILE;
1438         }
1439         else {
1440             int n = write( db_fd, &rec, TRUST_RECORD_LEN);
1441             if( n != TRUST_RECORD_LEN ) {
1442                 log_error(_("trustdb rec %lu: write failed (n=%d): %s\n"),
1443                                                  recnum, n, strerror(errno) );
1444                 rc = G10ERR_WRITE_FILE;
1445             }
1446         }
1447
1448         if( rc )
1449             log_fatal(_("%s: failed to append a record: %s\n"),
1450                                     db_name,    g10_errstr(rc));
1451     }
1452     return recnum ;
1453 }
1454
1455
1456
1457 static int
1458 cmp_trec_fpr ( void *fpr, const TRUSTREC *rec )
1459 {
1460     return rec->rectype == RECTYPE_TRUST
1461            && !memcmp( rec->r.trust.fingerprint, fpr, 20);
1462 }
1463
1464
1465 int
1466 tdbio_search_trust_byfpr( const byte *fingerprint, TRUSTREC *rec )
1467 {
1468     int rc;
1469
1470     /* locate the trust record using the hash table */
1471     rc = lookup_hashtable( get_trusthashrec(), fingerprint, 20,
1472                            cmp_trec_fpr, (void*)fingerprint, rec );
1473     return rc;
1474 }
1475
1476 int
1477 tdbio_search_trust_bypk (PKT_public_key *pk, TRUSTREC *rec)
1478 {
1479     byte fingerprint[MAX_FINGERPRINT_LEN];
1480     size_t fingerlen;
1481
1482     fingerprint_from_pk( pk, fingerprint, &fingerlen );
1483     for (; fingerlen < 20; fingerlen++ )
1484       fingerprint[fingerlen] = 0;
1485     return tdbio_search_trust_byfpr (fingerprint, rec);
1486 }
1487
1488
1489
1490 void
1491 tdbio_invalid(void)
1492 {
1493     log_error(_(
1494         "the trustdb is corrupted; please run \"gpg --fix-trustdb\".\n") );
1495     g10_exit(2);
1496 }
1497
1498 /*
1499  * Migrate the trustdb as just up to gpg 1.0.6 (trustdb version 2)
1500  * to the 2.1 version as used with 1.0.6b - This is pretty trivial as needs
1501  * only to scan the tdb and insert new the new trust records.  The old ones are
1502  * obsolte from now on
1503  */
1504 static void
1505 migrate_from_v2 ()
1506 {
1507   TRUSTREC rec;
1508   int i, n;
1509   struct {
1510     ulong keyrecno;
1511     byte  ot;
1512     byte okay;
1513     byte  fpr[20];
1514   } *ottable;
1515   int ottable_size, ottable_used;
1516   byte oldbuf[40];
1517   ulong recno;
1518   int rc, count;
1519
1520   ottable_size = 5;
1521   ottable = m_alloc (ottable_size * sizeof *ottable);
1522   ottable_used = 0;
1523
1524   /* We have some restrictions here.  We can't use the version record
1525    * and we can't use any of the old hashtables because we dropped the
1526    * code.  So we first collect all ownertrusts and then use a second
1527    * pass fo find the associated keys.  We have to do this all without using 
1528    * the regular record read functions.
1529    */
1530
1531   /* get all the ownertrusts */
1532   if (lseek (db_fd, 0, SEEK_SET ) == -1 ) 
1533       log_fatal ("migrate_from_v2: lseek failed: %s\n", strerror (errno));
1534   for (recno=0;;recno++)
1535     {
1536       do
1537         n = read (db_fd, oldbuf, 40);
1538       while (n==-1 && errno == EINTR);
1539       if (!n)
1540         break; /* eof */
1541       if (n != 40)
1542         log_fatal ("migrate_vfrom_v2: read error or short read\n");
1543
1544       if (*oldbuf != 2)
1545         continue;
1546       
1547       /* v2 dir record */
1548       if (ottable_used == ottable_size)
1549         {
1550           ottable_size += 1000;
1551           ottable = m_realloc (ottable, ottable_size * sizeof *ottable);
1552         }
1553       ottable[ottable_used].keyrecno = buftoulong (oldbuf+6);
1554       ottable[ottable_used].ot = oldbuf[18];
1555       ottable[ottable_used].okay = 0;
1556       memset (ottable[ottable_used].fpr,0, 20);
1557       if (ottable[ottable_used].keyrecno && ottable[ottable_used].ot)
1558         ottable_used++;
1559     }
1560   log_info ("found %d ownertrust records\n", ottable_used);
1561
1562   /* Read again and find the fingerprints */
1563   if (lseek (db_fd, 0, SEEK_SET ) == -1 ) 
1564       log_fatal ("migrate_from_v2: lseek failed: %s\n", strerror (errno));
1565   for (recno=0;;recno++)
1566     {
1567       do
1568         n = read (db_fd, oldbuf, 40);
1569       while (n==-1 && errno == EINTR);
1570       if (!n)
1571         break; /* eof */
1572       if (n != 40)
1573         log_fatal ("migrate_from_v2: read error or short read\n");
1574
1575       if (*oldbuf != 3) 
1576         continue;
1577
1578       /* v2 key record */
1579       for (i=0; i < ottable_used; i++)
1580         {
1581           if (ottable[i].keyrecno == recno)
1582             {
1583               memcpy (ottable[i].fpr, oldbuf+20, 20);
1584               ottable[i].okay = 1;
1585               break;
1586             }
1587         }
1588     }
1589
1590   /* got everything - create the v3 trustdb */
1591   if (ftruncate (db_fd, 0))
1592     log_fatal ("can't truncate `%s': %s\n", db_name, strerror (errno) );
1593   if (create_version_record ())
1594     log_fatal ("failed to recreate version record of `%s'\n", db_name);
1595
1596   /* access the hash table, so it is store just after the version record, 
1597    * this is not needed put a dump is more pretty */
1598   get_trusthashrec ();
1599
1600   /* And insert the old ownertrust values */
1601   count = 0;
1602   for (i=0; i < ottable_used; i++)
1603     {
1604       if (!ottable[i].okay)
1605         continue;
1606       
1607       memset (&rec, 0, sizeof rec);
1608       rec.recnum = tdbio_new_recnum ();
1609       rec.rectype = RECTYPE_TRUST;
1610       memcpy(rec.r.trust.fingerprint, ottable[i].fpr, 20);
1611       rec.r.trust.ownertrust = ottable[i].ot;
1612       if (tdbio_write_record (&rec))
1613         log_fatal ("failed to write trust record of `%s'\n", db_name);
1614       count++;
1615     }
1616
1617   revalidation_mark ();
1618   rc = tdbio_sync ();
1619   if (rc)
1620     log_fatal ("failed to sync `%s'\n", db_name);
1621   log_info ("migrated %d version 2 ownertrusts\n", count);
1622   m_free (ottable);
1623 }
1624
1625
1626