Commit | Line | Data |
---|---|---|
805e021f CE |
1 | /* |
2 | * Copyright 2000, International Business Machines Corporation and others. | |
3 | * All Rights Reserved. | |
4 | * | |
5 | * This software has been released under the terms of the IBM Public | |
6 | * License. For details, see the LICENSE file in the top-level source | |
7 | * directory or online at http://www.openafs.org/dl/license10.html | |
8 | */ | |
9 | ||
10 | #include <afsconfig.h> | |
11 | #include <afs/param.h> | |
12 | #include <afs/stds.h> | |
13 | ||
14 | #include <roken.h> | |
15 | ||
16 | #include <ubik.h> | |
17 | #include <afs/bubasics.h> | |
18 | #include <afs/audit.h> | |
19 | ||
20 | #include "budb_errs.h" | |
21 | #include "database.h" | |
22 | #include "error_macros.h" | |
23 | #include "budb_internal.h" | |
24 | ||
25 | int pollCount; | |
26 | struct memoryDB db; /* really allocate it here */ | |
27 | ||
28 | void | |
29 | db_panic(char *reason) | |
30 | { | |
31 | LogError(0, "db_panic: %s\n", reason); | |
32 | BUDB_EXIT(-1); | |
33 | } | |
34 | ||
35 | afs_int32 | |
36 | InitDB(void) | |
37 | { | |
38 | afs_int32 code; | |
39 | ||
40 | pollCount = 0; | |
41 | ||
42 | memset(&db, 0, sizeof(db)); | |
43 | if ((code = InitDBalloc()) || (code = InitDBhash())) | |
44 | return code; | |
45 | return 0; | |
46 | } | |
47 | ||
48 | /* package up seek and write into one procedure for ease of use */ | |
49 | ||
50 | /* dbwrite | |
51 | * write a portion of the database | |
52 | * entry: | |
53 | * pos - offset into the database (disk address). If this is in the | |
54 | * database header, then buff must be a ptr to a portion of | |
55 | * the in-core header | |
56 | * buff - the information to write | |
57 | * len - size of the write | |
58 | */ | |
59 | ||
60 | afs_int32 | |
61 | dbwrite(struct ubik_trans *ut, afs_int32 pos, void *buff, afs_int32 len) | |
62 | { | |
63 | afs_int32 code = 0; | |
64 | ||
65 | if (((pos < sizeof(db.h)) && (buff != (char *)&db.h + pos)) | |
66 | || (pos >= ntohl(db.h.eofPtr))) { | |
67 | Log("dbwrite: Illegal attempt to write at location 0 or past EOF\n"); | |
68 | ERROR(BUDB_IO); | |
69 | } | |
70 | ||
71 | code = ubik_Seek(ut, 0, pos); | |
72 | if (code) { | |
73 | LogError(code, "dbwrite: ubik_Seek to %d failed\n", pos); | |
74 | ERROR(code); | |
75 | } | |
76 | code = ubik_Write(ut, buff, len); | |
77 | if (code) { | |
78 | LogError(code, "dbwrite: ubik_Write failed\n"); | |
79 | ERROR(code); | |
80 | } | |
81 | ||
82 | error_exit: | |
83 | if (((++pollCount) % 4) == 0) { /* Poll every 4 reads/writes */ | |
84 | #ifndef AFS_PTHREAD_ENV | |
85 | IOMGR_Poll(); | |
86 | #endif | |
87 | pollCount = 0; | |
88 | } | |
89 | return code; | |
90 | } | |
91 | ||
92 | /* same thing for read */ | |
93 | ||
94 | afs_int32 | |
95 | dbread(struct ubik_trans *ut, afs_int32 pos, void *buff, afs_int32 len) | |
96 | { | |
97 | afs_int32 code = 0; | |
98 | ||
99 | if (pos >= ntohl(db.h.eofPtr)) { | |
100 | LogError(0, "dbread: Attempt to read @%d (past EOF)\n", pos); | |
101 | ERROR(BUDB_IO); | |
102 | } | |
103 | ||
104 | code = ubik_Seek(ut, 0, pos); | |
105 | if (code) { | |
106 | LogError(code, "dbread: ubik_Seek to %d failed\n", pos); | |
107 | ERROR(code); | |
108 | } | |
109 | code = ubik_Read(ut, buff, len); | |
110 | if (code) { | |
111 | LogError(code, "dbread: ubik_Read pos %d, buff %"AFS_PTR_FMT | |
112 | ", len %d\n", pos, buff, len); | |
113 | ERROR(code); | |
114 | } | |
115 | ||
116 | error_exit: | |
117 | if (((++pollCount) % 4) == 0) { /* Poll every 4 reads/writes */ | |
118 | #ifndef AFS_PTHREAD_ENV | |
119 | IOMGR_Poll(); | |
120 | #endif | |
121 | pollCount = 0; | |
122 | } | |
123 | return code; | |
124 | } | |
125 | ||
126 | /* Same as dbread excepts it does checking */ | |
127 | afs_int32 | |
128 | cdbread(struct ubik_trans *ut, int type, afs_int32 pos, void *buff, afs_int32 len) | |
129 | { | |
130 | afs_int32 code = 0; | |
131 | ||
132 | code = checkDiskAddress(pos, type, 0, 0); | |
133 | if (code) { | |
134 | LogError(code, "cdbread: Bad Address for block %d (addr 0x%x)\n", | |
135 | type, pos); | |
136 | ERROR(code); | |
137 | } | |
138 | ||
139 | code = ubik_Seek(ut, 0, pos); | |
140 | if (code) { | |
141 | LogError(code, "cdbread: ubik_Seek to 0x%x failed\n", pos); | |
142 | ERROR(code); | |
143 | } | |
144 | code = ubik_Read(ut, buff, len); | |
145 | if (code) { | |
146 | LogError(code, "cdbread: ubik_Read pos 0x%x, buff %"AFS_PTR_FMT | |
147 | ", len %d\n", pos, buff, len); | |
148 | ERROR(code); | |
149 | } | |
150 | ||
151 | error_exit: | |
152 | if (((++pollCount) % 4) == 0) { /* Poll every 4 reads/writes */ | |
153 | #ifndef AFS_PTHREAD_ENV | |
154 | IOMGR_Poll(); | |
155 | #endif | |
156 | pollCount = 0; | |
157 | } | |
158 | return code; | |
159 | } | |
160 | ||
161 | /* check that the database has been initialized. Be careful to fail in a safe | |
162 | manner, to avoid bogusly reinitializing the db. */ | |
163 | ||
164 | /** | |
165 | * reads in db cache from ubik. | |
166 | * | |
167 | * @param[in] ut ubik transaction | |
168 | * @param[in] rock opaque pointer to an int (*) (struct ubik_trans *), which | |
169 | * will be called on rebuilding the database (or NULL to not | |
170 | * rebuild the db) | |
171 | * | |
172 | * @return operation status | |
173 | * @retval 0 success | |
174 | */ | |
175 | static afs_int32 | |
176 | UpdateCache(struct ubik_trans *ut, void *rock) | |
177 | { | |
178 | int (*db_init) (struct ubik_trans *ut) = rock; | |
179 | afs_int32 code; | |
180 | ||
181 | db.h.eofPtr = htonl(sizeof(db.h)); /* for sanity check in dbread */ | |
182 | code = dbread(ut, 0, (char *)&db.h, sizeof(db.h)); | |
183 | if (code) | |
184 | ERROR(code); | |
185 | ||
186 | if ((ntohl(db.h.version) != BUDB_VERSION) | |
187 | || (ntohl(db.h.checkVersion) != BUDB_VERSION)) { | |
188 | ||
189 | if ((ntohl(db.h.version) == 0) || (ntohl(db.h.checkVersion) == 0)) | |
190 | ERROR(BUDB_EMPTY); | |
191 | ||
192 | LogError(0, "DB version should be %d; Initial = %d; Terminal = %d\n", | |
193 | BUDB_VERSION, ntohl(db.h.version), ntohl(db.h.checkVersion)); | |
194 | ERROR(BUDB_IO); | |
195 | } | |
196 | ||
197 | db.readTime = time(0); | |
198 | ht_Reset(&db.volName); | |
199 | ht_Reset(&db.tapeName); | |
200 | ht_Reset(&db.dumpName); | |
201 | ht_Reset(&db.dumpIden); | |
202 | ||
203 | error_exit: | |
204 | if (code) { | |
205 | if ((code == UEOF) || (code == BUDB_EMPTY)) { | |
206 | if (db_init) { | |
207 | LogDebug(0, "No data base - Building new one\n"); | |
208 | ||
209 | /* try to write a good header */ | |
210 | memset(&db.h, 0, sizeof(db.h)); | |
211 | db.h.version = htonl(BUDB_VERSION); | |
212 | db.h.checkVersion = htonl(BUDB_VERSION); | |
213 | db.h.lastUpdate = db.h.lastDumpId = htonl(time(0)); | |
214 | db.h.eofPtr = htonl(sizeof(db.h)); | |
215 | ||
216 | /* text ptrs cleared by bzero */ | |
217 | ht_DBInit(); | |
218 | ||
219 | code = dbwrite(ut, 0, (char *)&db.h, sizeof(db.h)); | |
220 | if (code) | |
221 | code = BUDB_IO; /* return the error code */ | |
222 | else | |
223 | code = db_init(ut); /* initialize the db */ | |
224 | } else { | |
225 | LogDebug(0, "No data base\n"); | |
226 | code = BUDB_EMPTY; | |
227 | } | |
228 | } else { | |
229 | LogDebug(0, "I/O Error\n"); | |
230 | code = BUDB_IO; | |
231 | } | |
232 | } | |
233 | return code; | |
234 | } | |
235 | ||
236 | afs_int32 | |
237 | CheckInit(struct ubik_trans *ut, | |
238 | int (*db_init) (struct ubik_trans *ut)) /* call if rebuilding DB */ | |
239 | { | |
240 | return ubik_CheckCache(ut, UpdateCache, db_init); | |
241 | } |