1 d957951b 2005-02-11 devnull #include <u.h>
  2 d957951b 2005-02-11 devnull #include <libc.h>
  3 d957951b 2005-02-11 devnull #include <bio.h>
  4 d957951b 2005-02-11 devnull #include <ctype.h>
  5 d957951b 2005-02-11 devnull #include <ndb.h>
  6 d957951b 2005-02-11 devnull #include "ndbhf.h"
  7 d957951b 2005-02-11 devnull 
  8 d957951b 2005-02-11 devnull static Ndb*	doopen(char*);
  9 d957951b 2005-02-11 devnull static void	hffree(Ndb*);
 10 d957951b 2005-02-11 devnull 
 11 d957951b 2005-02-11 devnull static char *deffile = "/lib/ndb/local";
 12 d957951b 2005-02-11 devnull 
 13 d957951b 2005-02-11 devnull /*
 14 d957951b 2005-02-11 devnull  *  the database entry in 'file' indicates the list of files
 15 d957951b 2005-02-11 devnull  *  that makeup the database.  Open each one and search in
 16 d957951b 2005-02-11 devnull  *  the same order.
 17 d957951b 2005-02-11 devnull  */
 18 d957951b 2005-02-11 devnull Ndb*
 19 d957951b 2005-02-11 devnull ndbopen(char *file)
 20 d957951b 2005-02-11 devnull {
 21 d957951b 2005-02-11 devnull 	Ndb *db, *first, *last;
 22 d957951b 2005-02-11 devnull 	Ndbs s;
 23 d957951b 2005-02-11 devnull 	Ndbtuple *t, *nt;
 24 d957951b 2005-02-11 devnull 
 25 d957951b 2005-02-11 devnull 	if(file == 0)
 26 d957951b 2005-02-11 devnull 		file = deffile;
 27 d957951b 2005-02-11 devnull 	db = doopen(file);
 28 d957951b 2005-02-11 devnull 	if(db == 0)
 29 d957951b 2005-02-11 devnull 		return 0;
 30 d957951b 2005-02-11 devnull 	first = last = db;
 31 d957951b 2005-02-11 devnull 	t = ndbsearch(db, &s, "database", "");
 32 d957951b 2005-02-11 devnull 	Bseek(&db->b, 0, 0);
 33 d957951b 2005-02-11 devnull 	if(t == 0)
 34 d957951b 2005-02-11 devnull 		return db;
 35 d957951b 2005-02-11 devnull 	for(nt = t; nt; nt = nt->entry){
 36 d957951b 2005-02-11 devnull 		if(strcmp(nt->attr, "file") != 0)
 37 d957951b 2005-02-11 devnull 			continue;
 38 d957951b 2005-02-11 devnull 		if(strcmp(nt->val, file) == 0){
 39 d957951b 2005-02-11 devnull 			/* default file can be reordered in the list */
 40 d957951b 2005-02-11 devnull 			if(first->next == 0)
 41 d957951b 2005-02-11 devnull 				continue;
 42 d957951b 2005-02-11 devnull 			if(strcmp(first->file, file) == 0){
 43 d957951b 2005-02-11 devnull 				db = first;
 44 d957951b 2005-02-11 devnull 				first = first->next;
 45 d957951b 2005-02-11 devnull 				last->next = db;
 46 d957951b 2005-02-11 devnull 				db->next = 0;
 47 d957951b 2005-02-11 devnull 				last = db;
 48 d957951b 2005-02-11 devnull 			}
 49 d957951b 2005-02-11 devnull 			continue;
 50 d957951b 2005-02-11 devnull 		}
 51 d957951b 2005-02-11 devnull 		db = doopen(nt->val);
 52 d957951b 2005-02-11 devnull 		if(db == 0)
 53 d957951b 2005-02-11 devnull 			continue;
 54 d957951b 2005-02-11 devnull 		last->next = db;
 55 d957951b 2005-02-11 devnull 		last = db;
 56 d957951b 2005-02-11 devnull 	}
 57 d957951b 2005-02-11 devnull 	ndbfree(t);
 58 d957951b 2005-02-11 devnull 	return first;
 59 d957951b 2005-02-11 devnull }
 60 d957951b 2005-02-11 devnull 
 61 d957951b 2005-02-11 devnull /*
 62 d957951b 2005-02-11 devnull  *  open a single file
 63 d957951b 2005-02-11 devnull  */
 64 d957951b 2005-02-11 devnull static Ndb*
 65 d957951b 2005-02-11 devnull doopen(char *file)
 66 d957951b 2005-02-11 devnull {
 67 d957951b 2005-02-11 devnull 	Ndb *db;
 68 d957951b 2005-02-11 devnull 
 69 d957951b 2005-02-11 devnull 	db = (Ndb*)malloc(sizeof(Ndb));
 70 d957951b 2005-02-11 devnull 	if(db == 0)
 71 d957951b 2005-02-11 devnull 		return 0;
 72 d957951b 2005-02-11 devnull 	memset(db, 0, sizeof(Ndb));
 73 d957951b 2005-02-11 devnull 	strncpy(db->file, file, sizeof(db->file)-1);
 74 d957951b 2005-02-11 devnull 
 75 d957951b 2005-02-11 devnull 	if(ndbreopen(db) < 0){
 76 d957951b 2005-02-11 devnull 		free(db);
 77 d957951b 2005-02-11 devnull 		return 0;
 78 d957951b 2005-02-11 devnull 	}
 79 d957951b 2005-02-11 devnull 
 80 d957951b 2005-02-11 devnull 	return db;
 81 d957951b 2005-02-11 devnull }
 82 d957951b 2005-02-11 devnull 
 83 d957951b 2005-02-11 devnull /*
 84 d957951b 2005-02-11 devnull  *  dump any cached information, forget the hash tables, and reopen a single file
 85 d957951b 2005-02-11 devnull  */
 86 d957951b 2005-02-11 devnull int
 87 d957951b 2005-02-11 devnull ndbreopen(Ndb *db)
 88 d957951b 2005-02-11 devnull {
 89 d957951b 2005-02-11 devnull 	int fd;
 90 d957951b 2005-02-11 devnull 	Dir *d;
 91 d957951b 2005-02-11 devnull 
 92 d957951b 2005-02-11 devnull 	/* forget what we know about the open files */
 93 d957951b 2005-02-11 devnull 	if(db->mtime){
 94 d957951b 2005-02-11 devnull 		_ndbcacheflush(db);
 95 d957951b 2005-02-11 devnull 		hffree(db);
 96 d957951b 2005-02-11 devnull 		close(Bfildes(&db->b));
 97 d957951b 2005-02-11 devnull 		Bterm(&db->b);
 98 d957951b 2005-02-11 devnull 		db->mtime = 0;
 99 d957951b 2005-02-11 devnull 	}
100 d957951b 2005-02-11 devnull 
101 d957951b 2005-02-11 devnull 	/* try the open again */
102 d957951b 2005-02-11 devnull 	fd = open(db->file, OREAD);
103 d957951b 2005-02-11 devnull 	if(fd < 0)
104 d957951b 2005-02-11 devnull 		return -1;
105 d957951b 2005-02-11 devnull 	d = dirfstat(fd);
106 d957951b 2005-02-11 devnull 	if(d == nil){
107 d957951b 2005-02-11 devnull 		close(fd);
108 d957951b 2005-02-11 devnull 		return -1;
109 d957951b 2005-02-11 devnull 	}
110 d957951b 2005-02-11 devnull 
111 d957951b 2005-02-11 devnull 	db->qid = d->qid;
112 d957951b 2005-02-11 devnull 	db->mtime = d->mtime;
113 d957951b 2005-02-11 devnull 	db->length = d->length;
114 d957951b 2005-02-11 devnull 	Binit(&db->b, fd, OREAD);
115 d957951b 2005-02-11 devnull 	free(d);
116 d957951b 2005-02-11 devnull 	return 0;
117 d957951b 2005-02-11 devnull }
118 d957951b 2005-02-11 devnull 
119 d957951b 2005-02-11 devnull /*
120 d957951b 2005-02-11 devnull  *  close the database files
121 d957951b 2005-02-11 devnull  */
122 d957951b 2005-02-11 devnull void
123 d957951b 2005-02-11 devnull ndbclose(Ndb *db)
124 d957951b 2005-02-11 devnull {
125 d957951b 2005-02-11 devnull 	Ndb *nextdb;
126 d957951b 2005-02-11 devnull 
127 d957951b 2005-02-11 devnull 	for(; db; db = nextdb){
128 d957951b 2005-02-11 devnull 		nextdb = db->next;
129 d957951b 2005-02-11 devnull 		_ndbcacheflush(db);
130 d957951b 2005-02-11 devnull 		hffree(db);
131 d957951b 2005-02-11 devnull 		close(Bfildes(&db->b));
132 d957951b 2005-02-11 devnull 		Bterm(&db->b);
133 d957951b 2005-02-11 devnull 		free(db);
134 d957951b 2005-02-11 devnull 	}
135 d957951b 2005-02-11 devnull }
136 d957951b 2005-02-11 devnull 
137 d957951b 2005-02-11 devnull /*
138 d957951b 2005-02-11 devnull  *  free the hash files belonging to a db
139 d957951b 2005-02-11 devnull  */
140 d957951b 2005-02-11 devnull static void
141 d957951b 2005-02-11 devnull hffree(Ndb *db)
142 d957951b 2005-02-11 devnull {
143 d957951b 2005-02-11 devnull 	Ndbhf *hf, *next;
144 d957951b 2005-02-11 devnull 
145 d957951b 2005-02-11 devnull 	for(hf = db->hf; hf; hf = next){
146 d957951b 2005-02-11 devnull 		next = hf->next;
147 d957951b 2005-02-11 devnull 		close(hf->fd);
148 d957951b 2005-02-11 devnull 		free(hf);
149 d957951b 2005-02-11 devnull 	}
150 d957951b 2005-02-11 devnull 	db->hf = 0;
151 d957951b 2005-02-11 devnull }
152 d957951b 2005-02-11 devnull 
153 d957951b 2005-02-11 devnull /*
154 d957951b 2005-02-11 devnull  *  return true if any part of the database has changed
155 d957951b 2005-02-11 devnull  */
156 d957951b 2005-02-11 devnull int
157 d957951b 2005-02-11 devnull ndbchanged(Ndb *db)
158 d957951b 2005-02-11 devnull {
159 d957951b 2005-02-11 devnull 	Ndb *ndb;
160 d957951b 2005-02-11 devnull 	Dir *d;
161 d957951b 2005-02-11 devnull 
162 d957951b 2005-02-11 devnull 	for(ndb = db; ndb != nil; ndb = ndb->next){
163 d957951b 2005-02-11 devnull 		d = dirfstat(Bfildes(&db->b));
164 d957951b 2005-02-11 devnull 		if(d == nil)
165 d957951b 2005-02-11 devnull 			continue;
166 d957951b 2005-02-11 devnull 		if(ndb->qid.path != d->qid.path
167 d957951b 2005-02-11 devnull 		|| ndb->qid.vers != d->qid.vers){
168 d957951b 2005-02-11 devnull 			free(d);
169 d957951b 2005-02-11 devnull 			return 1;
170 d957951b 2005-02-11 devnull 		}
171 d957951b 2005-02-11 devnull 		free(d);
172 d957951b 2005-02-11 devnull 	}
173 d957951b 2005-02-11 devnull 	return 0;
174 d957951b 2005-02-11 devnull }