1 /* Copyright 1998,2001-2003,2007-2009 Alain Knaff.
2 * This file is part of mtools.
3 *
4 * Mtools is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
8 *
9 * Mtools is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License
15 * along with Mtools. If not, see <http://www.gnu.org/licenses/>.
16 */
17 #include "sysincludes.h"
18 #include "vfat.h"
19 #include "dirCache.h"
20 #include "dirCacheP.h"
21 #include <assert.h>
22
23
24 #define BITS_PER_INT (sizeof(unsigned int) * 8)
25
26
rol(uint32_t arg,int shift)27 static __inline__ uint32_t rol(uint32_t arg, int shift)
28 {
29 arg &= 0xffffffff; /* for 64 bit machines */
30 return (arg << shift) | (arg >> (32 - shift));
31 }
32
33
calcHash(wchar_t * name)34 static uint32_t calcHash(wchar_t *name)
35 {
36 uint32_t hash;
37 unsigned int i;
38 wint_t c;
39
40 hash = 0;
41 i = 0;
42 while(*name) {
43 /* rotate it */
44 hash = rol(hash,5); /* a shift of 5 makes sure we spread quickly
45 * over the whole width, moreover, 5 is
46 * prime with 32, which makes sure that
47 * successive letters cannot cover each
48 * other easily */
49 c = towupper((wint_t)*name);
50 hash ^= (c * (c+2)) ^ (i * (i+2));
51 hash &= 0xffffffff;
52 i++;
53 name++;
54 }
55 hash = hash * (hash + 2);
56 /* the following two xors make sure all info is spread evenly over all
57 * bytes. Important if we only keep the low order bits later on */
58 hash ^= (hash & 0xfff) << 12;
59 hash ^= (hash & 0xff000) << 24;
60 return hash;
61 }
62
addBit(unsigned int * bitmap,unsigned int hash,int checkOnly)63 static unsigned int addBit(unsigned int *bitmap,
64 unsigned int hash, int checkOnly)
65 {
66 unsigned int bit;
67 int entry;
68
69 bit = 1u << (hash % BITS_PER_INT);
70 entry = (hash / BITS_PER_INT) % DC_BITMAP_SIZE;
71
72 if(checkOnly)
73 return bitmap[entry] & bit;
74 else {
75 bitmap[entry] |= bit;
76 return 1;
77 }
78 }
79
_addHash(dirCache_t * cache,unsigned int hash,int checkOnly)80 static int _addHash(dirCache_t *cache, unsigned int hash, int checkOnly)
81 {
82 return
83 addBit(cache->bm0, hash, checkOnly) &&
84 addBit(cache->bm1, rol(hash,12), checkOnly) &&
85 addBit(cache->bm2, rol(hash,24), checkOnly);
86 }
87
88
addNameToHash(dirCache_t * cache,wchar_t * name)89 static void addNameToHash(dirCache_t *cache, wchar_t *name)
90 {
91 _addHash(cache, calcHash(name), 0);
92 }
93
hashDce(dirCache_t * cache,dirCacheEntry_t * dce)94 static void hashDce(dirCache_t *cache, dirCacheEntry_t *dce)
95 {
96 if(dce->beginSlot != cache->nrHashed)
97 return;
98 cache->nrHashed = dce->endSlot;
99 if(dce->longName)
100 addNameToHash(cache, dce->longName);
101 addNameToHash(cache, dce->shortName);
102 }
103
isHashed(dirCache_t * cache,wchar_t * name)104 int isHashed(dirCache_t *cache, wchar_t *name)
105 {
106 int ret;
107
108 ret = _addHash(cache, calcHash(name), 1);
109 return ret;
110 }
111
growDirCache(dirCache_t * cache,unsigned int slot)112 int growDirCache(dirCache_t *cache, unsigned int slot)
113 {
114 if((int) slot < 0) {
115 fprintf(stderr, "Bad slot %d\n", slot);
116 exit(1);
117 }
118
119 if( cache->nr_entries <= slot) {
120 unsigned int i;
121
122 cache->entries = realloc(cache->entries,
123 (slot+1) * 2 *
124 sizeof(dirCacheEntry_t *));
125 if(!cache->entries)
126 return -1;
127 for(i= cache->nr_entries; i < (slot+1) * 2; i++) {
128 cache->entries[i] = 0;
129 }
130 cache->nr_entries = (slot+1) * 2;
131 }
132 return 0;
133 }
134
allocDirCache(Stream_t * Stream,unsigned int slot)135 dirCache_t *allocDirCache(Stream_t *Stream, unsigned int slot)
136 {
137 dirCache_t **dcp;
138
139 if((int)slot < 0) {
140 fprintf(stderr, "Bad slot %d\n", slot);
141 exit(1);
142 }
143
144 dcp = getDirCacheP(Stream);
145 if(!*dcp) {
146 *dcp = New(dirCache_t);
147 if(!*dcp)
148 return 0;
149 (*dcp)->entries = NewArray((slot+1)*2+5, dirCacheEntry_t *);
150 if(!(*dcp)->entries) {
151 free(*dcp);
152 return 0;
153 }
154 (*dcp)->nr_entries = (slot+1) * 2;
155 memset( (*dcp)->bm0, 0, sizeof((*dcp)->bm0));
156 memset( (*dcp)->bm1, 0, sizeof((*dcp)->bm1));
157 memset( (*dcp)->bm2, 0, sizeof((*dcp)->bm1));
158 (*dcp)->nrHashed = 0;
159 } else
160 if(growDirCache(*dcp, slot) < 0)
161 return 0;
162 return *dcp;
163 }
164
165 /* Free a range of entries. The range being cleared is always aligned
166 * on the begin of an entry. Entries which are cleared entirely are
167 * free'd. Entries which are cleared half-way (can only happen at end)
168 * are "shortened" by moving its beginSlot
169 * If this was a range representing space after end-mark, return beginning
170 * of range if it had been shortened in its lifetime (which means that end
171 * mark must have moved => may need to be rewritten)
172 */
freeDirCacheRange(dirCache_t * cache,unsigned int beginSlot,unsigned int endSlot)173 static int freeDirCacheRange(dirCache_t *cache,
174 unsigned int beginSlot,
175 unsigned int endSlot)
176 {
177 dirCacheEntry_t *entry;
178 unsigned int clearBegin;
179 unsigned int clearEnd;
180 unsigned int i;
181
182 if(endSlot < beginSlot) {
183 fprintf(stderr, "Bad slots %d %d in free range\n",
184 beginSlot, endSlot);
185 exit(1);
186 }
187
188 while(beginSlot < endSlot) {
189 entry = cache->entries[beginSlot];
190 if(!entry) {
191 beginSlot++;
192 continue;
193 }
194
195 /* Due to the way this is called, we _always_ de-allocate
196 * starting from beginning... */
197 assert(entry->beginSlot == beginSlot);
198
199 clearEnd = entry->endSlot;
200 if(clearEnd > endSlot)
201 clearEnd = endSlot;
202 clearBegin = beginSlot;
203
204 for(i = clearBegin; i <clearEnd; i++)
205 cache->entries[i] = 0;
206
207 entry->beginSlot = clearEnd;
208
209 if(entry->beginSlot == entry->endSlot) {
210 int needWriteEnd = 0;
211 if(entry->endMarkPos != -1 &&
212 entry->endMarkPos < (int) beginSlot)
213 needWriteEnd = 1;
214
215 if(entry->longName)
216 free(entry->longName);
217 if(entry->shortName)
218 free(entry->shortName);
219 free(entry);
220 if(needWriteEnd) {
221 return (int) beginSlot;
222 }
223 }
224
225 beginSlot = clearEnd;
226 }
227 return -1;
228 }
229
allocDirCacheEntry(dirCache_t * cache,unsigned int beginSlot,unsigned int endSlot,dirCacheEntryType_t type)230 static dirCacheEntry_t *allocDirCacheEntry(dirCache_t *cache,
231 unsigned int beginSlot,
232 unsigned int endSlot,
233 dirCacheEntryType_t type)
234 {
235 dirCacheEntry_t *entry;
236 unsigned int i;
237
238 if(growDirCache(cache, endSlot) < 0)
239 return 0;
240
241 entry = New(dirCacheEntry_t);
242 if(!entry)
243 return 0;
244 entry->type = type;
245 entry->longName = 0;
246 entry->shortName = 0;
247 entry->beginSlot = beginSlot;
248 entry->endSlot = endSlot;
249 entry->endMarkPos = -1;
250
251 freeDirCacheRange(cache, beginSlot, endSlot);
252 for(i=beginSlot; i<endSlot; i++) {
253 cache->entries[i] = entry;
254 }
255 return entry;
256 }
257
addUsedEntry(dirCache_t * cache,unsigned int beginSlot,unsigned int endSlot,wchar_t * longName,wchar_t * shortName,struct directory * dir)258 dirCacheEntry_t *addUsedEntry(dirCache_t *cache,
259 unsigned int beginSlot,
260 unsigned int endSlot,
261 wchar_t *longName, wchar_t *shortName,
262 struct directory *dir)
263 {
264 dirCacheEntry_t *entry;
265
266 if(endSlot < beginSlot) {
267 fprintf(stderr,
268 "Bad slots %d %d in add used entry\n",
269 beginSlot, endSlot);
270 exit(1);
271 }
272
273
274 entry = allocDirCacheEntry(cache, beginSlot, endSlot, DCET_USED);
275 if(!entry)
276 return 0;
277
278 entry->beginSlot = beginSlot;
279 entry->endSlot = endSlot;
280 if(longName)
281 entry->longName = wcsdup(longName);
282 entry->shortName = wcsdup(shortName);
283 entry->dir = *dir;
284 hashDce(cache, entry);
285 return entry;
286 }
287
288 /* Try to merge free slot at position "slot" with slot at previous position
289 * After successful merge, both will point to a same DCE (the one which used
290 * to be previous)
291 * Only does something if both of these slots are actually free
292 */
mergeFreeSlots(dirCache_t * cache,unsigned int slot)293 static void mergeFreeSlots(dirCache_t *cache, unsigned int slot)
294 {
295 dirCacheEntry_t *previous, *next;
296 unsigned int i;
297
298 if(slot == 0)
299 return;
300 previous = cache->entries[slot-1];
301 next = cache->entries[slot];
302 if(next && next->type == DCET_FREE &&
303 previous && previous->type == DCET_FREE) {
304 for(i=next->beginSlot; i < next->endSlot; i++)
305 cache->entries[i] = previous;
306 previous->endSlot = next->endSlot;
307 previous->endMarkPos = next->endMarkPos;
308 free(next);
309 }
310 }
311
312 /* Create a free range extending from beginSlot to endSlot, and try to merge
313 * it with any neighboring free ranges */
addFreeEndEntry(dirCache_t * cache,unsigned int beginSlot,unsigned int endSlot,int isAtEnd)314 dirCacheEntry_t *addFreeEndEntry(dirCache_t *cache,
315 unsigned int beginSlot,
316 unsigned int endSlot,
317 int isAtEnd)
318 {
319 dirCacheEntry_t *entry;
320
321 if(beginSlot < cache->nrHashed)
322 cache->nrHashed = beginSlot;
323
324 if(endSlot < beginSlot) {
325 fprintf(stderr, "Bad slots %d %d in add free entry\n",
326 beginSlot, endSlot);
327 exit(1);
328 }
329
330 if(endSlot == beginSlot)
331 return 0;
332 entry = allocDirCacheEntry(cache, beginSlot, endSlot, DCET_FREE);
333 if(isAtEnd)
334 entry->endMarkPos = (int) beginSlot;
335 mergeFreeSlots(cache, beginSlot);
336 mergeFreeSlots(cache, endSlot);
337 return cache->entries[beginSlot];
338 }
339
addFreeEntry(dirCache_t * cache,unsigned int beginSlot,unsigned int endSlot)340 dirCacheEntry_t *addFreeEntry(dirCache_t *cache,
341 unsigned int beginSlot,
342 unsigned int endSlot)
343 {
344 return addFreeEndEntry(cache, beginSlot, endSlot, 0);
345 }
346
addEndEntry(dirCache_t * cache,unsigned int pos)347 dirCacheEntry_t *addEndEntry(dirCache_t *cache, unsigned int pos)
348 {
349 return allocDirCacheEntry(cache, pos, pos+1u, DCET_END);
350 }
351
lookupInDircache(dirCache_t * cache,unsigned int pos)352 dirCacheEntry_t *lookupInDircache(dirCache_t *cache, unsigned int pos)
353 {
354 if(growDirCache(cache, pos+1) < 0)
355 return 0;
356 return cache->entries[pos];
357 }
358
freeDirCache(Stream_t * Stream)359 void freeDirCache(Stream_t *Stream)
360 {
361 dirCache_t *cache, **dcp;
362
363 dcp = getDirCacheP(Stream);
364 cache = *dcp;
365 if(cache) {
366 int n;
367 n=freeDirCacheRange(cache, 0, cache->nr_entries);
368 if(n >= 0)
369 low_level_dir_write_end(Stream, n);
370 free(cache);
371 *dcp = 0;
372 }
373 }
374