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