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