Merge with /pub/scm/linux/kernel/git/torvalds/linux-2.6.git
[linux-2.6] / fs / hpfs / anode.c
1 /*
2  *  linux/fs/hpfs/anode.c
3  *
4  *  Mikulas Patocka (mikulas@artax.karlin.mff.cuni.cz), 1998-1999
5  *
6  *  handling HPFS anode tree that contains file allocation info
7  */
8
9 #include "hpfs_fn.h"
10
11 /* Find a sector in allocation tree */
12
13 secno hpfs_bplus_lookup(struct super_block *s, struct inode *inode,
14                    struct bplus_header *btree, unsigned sec,
15                    struct buffer_head *bh)
16 {
17         anode_secno a = -1;
18         struct anode *anode;
19         int i;
20         int c1, c2 = 0;
21         go_down:
22         if (hpfs_sb(s)->sb_chk) if (hpfs_stop_cycles(s, a, &c1, &c2, "hpfs_bplus_lookup")) return -1;
23         if (btree->internal) {
24                 for (i = 0; i < btree->n_used_nodes; i++)
25                         if (btree->u.internal[i].file_secno > sec) {
26                                 a = btree->u.internal[i].down;
27                                 brelse(bh);
28                                 if (!(anode = hpfs_map_anode(s, a, &bh))) return -1;
29                                 btree = &anode->btree;
30                                 goto go_down;
31                         }
32                 hpfs_error(s, "sector %08x not found in internal anode %08x", sec, a);
33                 brelse(bh);
34                 return -1;
35         }
36         for (i = 0; i < btree->n_used_nodes; i++)
37                 if (btree->u.external[i].file_secno <= sec &&
38                     btree->u.external[i].file_secno + btree->u.external[i].length > sec) {
39                         a = btree->u.external[i].disk_secno + sec - btree->u.external[i].file_secno;
40                         if (hpfs_sb(s)->sb_chk) if (hpfs_chk_sectors(s, a, 1, "data")) {
41                                 brelse(bh);
42                                 return -1;
43                         }
44                         if (inode) {
45                                 struct hpfs_inode_info *hpfs_inode = hpfs_i(inode);
46                                 hpfs_inode->i_file_sec = btree->u.external[i].file_secno;
47                                 hpfs_inode->i_disk_sec = btree->u.external[i].disk_secno;
48                                 hpfs_inode->i_n_secs = btree->u.external[i].length;
49                         }
50                         brelse(bh);
51                         return a;
52                 }
53         hpfs_error(s, "sector %08x not found in external anode %08x", sec, a);
54         brelse(bh);
55         return -1;
56 }
57
58 /* Add a sector to tree */
59
60 secno hpfs_add_sector_to_btree(struct super_block *s, secno node, int fnod, unsigned fsecno)
61 {
62         struct bplus_header *btree;
63         struct anode *anode = NULL, *ranode = NULL;
64         struct fnode *fnode;
65         anode_secno a, na = -1, ra, up = -1;
66         secno se;
67         struct buffer_head *bh, *bh1, *bh2;
68         int n;
69         unsigned fs;
70         int c1, c2 = 0;
71         if (fnod) {
72                 if (!(fnode = hpfs_map_fnode(s, node, &bh))) return -1;
73                 btree = &fnode->btree;
74         } else {
75                 if (!(anode = hpfs_map_anode(s, node, &bh))) return -1;
76                 btree = &anode->btree;
77         }
78         a = node;
79         go_down:
80         if ((n = btree->n_used_nodes - 1) < -!!fnod) {
81                 hpfs_error(s, "anode %08x has no entries", a);
82                 brelse(bh);
83                 return -1;
84         }
85         if (btree->internal) {
86                 a = btree->u.internal[n].down;
87                 btree->u.internal[n].file_secno = -1;
88                 mark_buffer_dirty(bh);
89                 brelse(bh);
90                 if (hpfs_sb(s)->sb_chk)
91                         if (hpfs_stop_cycles(s, a, &c1, &c2, "hpfs_add_sector_to_btree #1")) return -1;
92                 if (!(anode = hpfs_map_anode(s, a, &bh))) return -1;
93                 btree = &anode->btree;
94                 goto go_down;
95         }
96         if (n >= 0) {
97                 if (btree->u.external[n].file_secno + btree->u.external[n].length != fsecno) {
98                         hpfs_error(s, "allocated size %08x, trying to add sector %08x, %cnode %08x",
99                                 btree->u.external[n].file_secno + btree->u.external[n].length, fsecno,
100                                 fnod?'f':'a', node);
101                         brelse(bh);
102                         return -1;
103                 }
104                 if (hpfs_alloc_if_possible(s, se = btree->u.external[n].disk_secno + btree->u.external[n].length)) {
105                         btree->u.external[n].length++;
106                         mark_buffer_dirty(bh);
107                         brelse(bh);
108                         return se;
109                 }
110         } else {
111                 if (fsecno) {
112                         hpfs_error(s, "empty file %08x, trying to add sector %08x", node, fsecno);
113                         brelse(bh);
114                         return -1;
115                 }
116                 se = !fnod ? node : (node + 16384) & ~16383;
117         }       
118         if (!(se = hpfs_alloc_sector(s, se, 1, fsecno*ALLOC_M>ALLOC_FWD_MAX ? ALLOC_FWD_MAX : fsecno*ALLOC_M<ALLOC_FWD_MIN ? ALLOC_FWD_MIN : fsecno*ALLOC_M, 1))) {
119                 brelse(bh);
120                 return -1;
121         }
122         fs = n < 0 ? 0 : btree->u.external[n].file_secno + btree->u.external[n].length;
123         if (!btree->n_free_nodes) {
124                 up = a != node ? anode->up : -1;
125                 if (!(anode = hpfs_alloc_anode(s, a, &na, &bh1))) {
126                         brelse(bh);
127                         hpfs_free_sectors(s, se, 1);
128                         return -1;
129                 }
130                 if (a == node && fnod) {
131                         anode->up = node;
132                         anode->btree.fnode_parent = 1;
133                         anode->btree.n_used_nodes = btree->n_used_nodes;
134                         anode->btree.first_free = btree->first_free;
135                         anode->btree.n_free_nodes = 40 - anode->btree.n_used_nodes;
136                         memcpy(&anode->u, &btree->u, btree->n_used_nodes * 12);
137                         btree->internal = 1;
138                         btree->n_free_nodes = 11;
139                         btree->n_used_nodes = 1;
140                         btree->first_free = (char *)&(btree->u.internal[1]) - (char *)btree;
141                         btree->u.internal[0].file_secno = -1;
142                         btree->u.internal[0].down = na;
143                         mark_buffer_dirty(bh);
144                 } else if (!(ranode = hpfs_alloc_anode(s, /*a*/0, &ra, &bh2))) {
145                         brelse(bh);
146                         brelse(bh1);
147                         hpfs_free_sectors(s, se, 1);
148                         hpfs_free_sectors(s, na, 1);
149                         return -1;
150                 }
151                 brelse(bh);
152                 bh = bh1;
153                 btree = &anode->btree;
154         }
155         btree->n_free_nodes--; n = btree->n_used_nodes++;
156         btree->first_free += 12;
157         btree->u.external[n].disk_secno = se;
158         btree->u.external[n].file_secno = fs;
159         btree->u.external[n].length = 1;
160         mark_buffer_dirty(bh);
161         brelse(bh);
162         if ((a == node && fnod) || na == -1) return se;
163         c2 = 0;
164         while (up != -1) {
165                 struct anode *new_anode;
166                 if (hpfs_sb(s)->sb_chk)
167                         if (hpfs_stop_cycles(s, up, &c1, &c2, "hpfs_add_sector_to_btree #2")) return -1;
168                 if (up != node || !fnod) {
169                         if (!(anode = hpfs_map_anode(s, up, &bh))) return -1;
170                         btree = &anode->btree;
171                 } else {
172                         if (!(fnode = hpfs_map_fnode(s, up, &bh))) return -1;
173                         btree = &fnode->btree;
174                 }
175                 if (btree->n_free_nodes) {
176                         btree->n_free_nodes--; n = btree->n_used_nodes++;
177                         btree->first_free += 8;
178                         btree->u.internal[n].file_secno = -1;
179                         btree->u.internal[n].down = na;
180                         btree->u.internal[n-1].file_secno = fs;
181                         mark_buffer_dirty(bh);
182                         brelse(bh);
183                         brelse(bh2);
184                         hpfs_free_sectors(s, ra, 1);
185                         if ((anode = hpfs_map_anode(s, na, &bh))) {
186                                 anode->up = up;
187                                 anode->btree.fnode_parent = up == node && fnod;
188                                 mark_buffer_dirty(bh);
189                                 brelse(bh);
190                         }
191                         return se;
192                 }
193                 up = up != node ? anode->up : -1;
194                 btree->u.internal[btree->n_used_nodes - 1].file_secno = /*fs*/-1;
195                 mark_buffer_dirty(bh);
196                 brelse(bh);
197                 a = na;
198                 if ((new_anode = hpfs_alloc_anode(s, a, &na, &bh))) {
199                         anode = new_anode;
200                         /*anode->up = up != -1 ? up : ra;*/
201                         anode->btree.internal = 1;
202                         anode->btree.n_used_nodes = 1;
203                         anode->btree.n_free_nodes = 59;
204                         anode->btree.first_free = 16;
205                         anode->btree.u.internal[0].down = a;
206                         anode->btree.u.internal[0].file_secno = -1;
207                         mark_buffer_dirty(bh);
208                         brelse(bh);
209                         if ((anode = hpfs_map_anode(s, a, &bh))) {
210                                 anode->up = na;
211                                 mark_buffer_dirty(bh);
212                                 brelse(bh);
213                         }
214                 } else na = a;
215         }
216         if ((anode = hpfs_map_anode(s, na, &bh))) {
217                 anode->up = node;
218                 if (fnod) anode->btree.fnode_parent = 1;
219                 mark_buffer_dirty(bh);
220                 brelse(bh);
221         }
222         if (!fnod) {
223                 if (!(anode = hpfs_map_anode(s, node, &bh))) {
224                         brelse(bh2);
225                         return -1;
226                 }
227                 btree = &anode->btree;
228         } else {
229                 if (!(fnode = hpfs_map_fnode(s, node, &bh))) {
230                         brelse(bh2);
231                         return -1;
232                 }
233                 btree = &fnode->btree;
234         }
235         ranode->up = node;
236         memcpy(&ranode->btree, btree, btree->first_free);
237         if (fnod) ranode->btree.fnode_parent = 1;
238         ranode->btree.n_free_nodes = (ranode->btree.internal ? 60 : 40) - ranode->btree.n_used_nodes;
239         if (ranode->btree.internal) for (n = 0; n < ranode->btree.n_used_nodes; n++) {
240                 struct anode *unode;
241                 if ((unode = hpfs_map_anode(s, ranode->u.internal[n].down, &bh1))) {
242                         unode->up = ra;
243                         unode->btree.fnode_parent = 0;
244                         mark_buffer_dirty(bh1);
245                         brelse(bh1);
246                 }
247         }
248         btree->internal = 1;
249         btree->n_free_nodes = fnod ? 10 : 58;
250         btree->n_used_nodes = 2;
251         btree->first_free = (char *)&btree->u.internal[2] - (char *)btree;
252         btree->u.internal[0].file_secno = fs;
253         btree->u.internal[0].down = ra;
254         btree->u.internal[1].file_secno = -1;
255         btree->u.internal[1].down = na;
256         mark_buffer_dirty(bh);
257         brelse(bh);
258         mark_buffer_dirty(bh2);
259         brelse(bh2);
260         return se;
261 }
262
263 /*
264  * Remove allocation tree. Recursion would look much nicer but
265  * I want to avoid it because it can cause stack overflow.
266  */
267
268 void hpfs_remove_btree(struct super_block *s, struct bplus_header *btree)
269 {
270         struct bplus_header *btree1 = btree;
271         struct anode *anode = NULL;
272         anode_secno ano = 0, oano;
273         struct buffer_head *bh;
274         int level = 0;
275         int pos = 0;
276         int i;
277         int c1, c2 = 0;
278         int d1, d2;
279         go_down:
280         d2 = 0;
281         while (btree1->internal) {
282                 ano = btree1->u.internal[pos].down;
283                 if (level) brelse(bh);
284                 if (hpfs_sb(s)->sb_chk)
285                         if (hpfs_stop_cycles(s, ano, &d1, &d2, "hpfs_remove_btree #1"))
286                                 return;
287                 if (!(anode = hpfs_map_anode(s, ano, &bh))) return;
288                 btree1 = &anode->btree;
289                 level++;
290                 pos = 0;
291         }
292         for (i = 0; i < btree1->n_used_nodes; i++)
293                 hpfs_free_sectors(s, btree1->u.external[i].disk_secno, btree1->u.external[i].length);
294         go_up:
295         if (!level) return;
296         brelse(bh);
297         if (hpfs_sb(s)->sb_chk)
298                 if (hpfs_stop_cycles(s, ano, &c1, &c2, "hpfs_remove_btree #2")) return;
299         hpfs_free_sectors(s, ano, 1);
300         oano = ano;
301         ano = anode->up;
302         if (--level) {
303                 if (!(anode = hpfs_map_anode(s, ano, &bh))) return;
304                 btree1 = &anode->btree;
305         } else btree1 = btree;
306         for (i = 0; i < btree1->n_used_nodes; i++) {
307                 if (btree1->u.internal[i].down == oano) {
308                         if ((pos = i + 1) < btree1->n_used_nodes)
309                                 goto go_down;
310                         else
311                                 goto go_up;
312                 }
313         }
314         hpfs_error(s,
315                    "reference to anode %08x not found in anode %08x "
316                    "(probably bad up pointer)",
317                    oano, level ? ano : -1);
318         if (level)
319                 brelse(bh);
320 }
321
322 /* Just a wrapper around hpfs_bplus_lookup .. used for reading eas */
323
324 static secno anode_lookup(struct super_block *s, anode_secno a, unsigned sec)
325 {
326         struct anode *anode;
327         struct buffer_head *bh;
328         if (!(anode = hpfs_map_anode(s, a, &bh))) return -1;
329         return hpfs_bplus_lookup(s, NULL, &anode->btree, sec, bh);
330 }
331
332 int hpfs_ea_read(struct super_block *s, secno a, int ano, unsigned pos,
333             unsigned len, char *buf)
334 {
335         struct buffer_head *bh;
336         char *data;
337         secno sec;
338         unsigned l;
339         while (len) {
340                 if (ano) {
341                         if ((sec = anode_lookup(s, a, pos >> 9)) == -1)
342                                 return -1;
343                 } else sec = a + (pos >> 9);
344                 if (hpfs_sb(s)->sb_chk) if (hpfs_chk_sectors(s, sec, 1, "ea #1")) return -1;
345                 if (!(data = hpfs_map_sector(s, sec, &bh, (len - 1) >> 9)))
346                         return -1;
347                 l = 0x200 - (pos & 0x1ff); if (l > len) l = len;
348                 memcpy(buf, data + (pos & 0x1ff), l);
349                 brelse(bh);
350                 buf += l; pos += l; len -= l;
351         }
352         return 0;
353 }
354
355 int hpfs_ea_write(struct super_block *s, secno a, int ano, unsigned pos,
356              unsigned len, char *buf)
357 {
358         struct buffer_head *bh;
359         char *data;
360         secno sec;
361         unsigned l;
362         while (len) {
363                 if (ano) {
364                         if ((sec = anode_lookup(s, a, pos >> 9)) == -1)
365                                 return -1;
366                 } else sec = a + (pos >> 9);
367                 if (hpfs_sb(s)->sb_chk) if (hpfs_chk_sectors(s, sec, 1, "ea #2")) return -1;
368                 if (!(data = hpfs_map_sector(s, sec, &bh, (len - 1) >> 9)))
369                         return -1;
370                 l = 0x200 - (pos & 0x1ff); if (l > len) l = len;
371                 memcpy(data + (pos & 0x1ff), buf, l);
372                 mark_buffer_dirty(bh);
373                 brelse(bh);
374                 buf += l; pos += l; len -= l;
375         }
376         return 0;
377 }
378
379 void hpfs_ea_remove(struct super_block *s, secno a, int ano, unsigned len)
380 {
381         struct anode *anode;
382         struct buffer_head *bh;
383         if (ano) {
384                 if (!(anode = hpfs_map_anode(s, a, &bh))) return;
385                 hpfs_remove_btree(s, &anode->btree);
386                 brelse(bh);
387                 hpfs_free_sectors(s, a, 1);
388         } else hpfs_free_sectors(s, a, (len + 511) >> 9);
389 }
390
391 /* Truncate allocation tree. Doesn't join anodes - I hope it doesn't matter */
392
393 void hpfs_truncate_btree(struct super_block *s, secno f, int fno, unsigned secs)
394 {
395         struct fnode *fnode;
396         struct anode *anode;
397         struct buffer_head *bh;
398         struct bplus_header *btree;
399         anode_secno node = f;
400         int i, j, nodes;
401         int c1, c2 = 0;
402         if (fno) {
403                 if (!(fnode = hpfs_map_fnode(s, f, &bh))) return;
404                 btree = &fnode->btree;
405         } else {
406                 if (!(anode = hpfs_map_anode(s, f, &bh))) return;
407                 btree = &anode->btree;
408         }
409         if (!secs) {
410                 hpfs_remove_btree(s, btree);
411                 if (fno) {
412                         btree->n_free_nodes = 8;
413                         btree->n_used_nodes = 0;
414                         btree->first_free = 8;
415                         btree->internal = 0;
416                         mark_buffer_dirty(bh);
417                 } else hpfs_free_sectors(s, f, 1);
418                 brelse(bh);
419                 return;
420         }
421         while (btree->internal) {
422                 nodes = btree->n_used_nodes + btree->n_free_nodes;
423                 for (i = 0; i < btree->n_used_nodes; i++)
424                         if (btree->u.internal[i].file_secno >= secs) goto f;
425                 brelse(bh);
426                 hpfs_error(s, "internal btree %08x doesn't end with -1", node);
427                 return;
428                 f:
429                 for (j = i + 1; j < btree->n_used_nodes; j++)
430                         hpfs_ea_remove(s, btree->u.internal[j].down, 1, 0);
431                 btree->n_used_nodes = i + 1;
432                 btree->n_free_nodes = nodes - btree->n_used_nodes;
433                 btree->first_free = 8 + 8 * btree->n_used_nodes;
434                 mark_buffer_dirty(bh);
435                 if (btree->u.internal[i].file_secno == secs) {
436                         brelse(bh);
437                         return;
438                 }
439                 node = btree->u.internal[i].down;
440                 brelse(bh);
441                 if (hpfs_sb(s)->sb_chk)
442                         if (hpfs_stop_cycles(s, node, &c1, &c2, "hpfs_truncate_btree"))
443                                 return;
444                 if (!(anode = hpfs_map_anode(s, node, &bh))) return;
445                 btree = &anode->btree;
446         }       
447         nodes = btree->n_used_nodes + btree->n_free_nodes;
448         for (i = 0; i < btree->n_used_nodes; i++)
449                 if (btree->u.external[i].file_secno + btree->u.external[i].length >= secs) goto ff;
450         brelse(bh);
451         return;
452         ff:
453         if (secs <= btree->u.external[i].file_secno) {
454                 hpfs_error(s, "there is an allocation error in file %08x, sector %08x", f, secs);
455                 if (i) i--;
456         }
457         else if (btree->u.external[i].file_secno + btree->u.external[i].length > secs) {
458                 hpfs_free_sectors(s, btree->u.external[i].disk_secno + secs -
459                         btree->u.external[i].file_secno, btree->u.external[i].length
460                         - secs + btree->u.external[i].file_secno); /* I hope gcc optimizes this :-) */
461                 btree->u.external[i].length = secs - btree->u.external[i].file_secno;
462         }
463         for (j = i + 1; j < btree->n_used_nodes; j++)
464                 hpfs_free_sectors(s, btree->u.external[j].disk_secno, btree->u.external[j].length);
465         btree->n_used_nodes = i + 1;
466         btree->n_free_nodes = nodes - btree->n_used_nodes;
467         btree->first_free = 8 + 12 * btree->n_used_nodes;
468         mark_buffer_dirty(bh);
469         brelse(bh);
470 }
471
472 /* Remove file or directory and it's eas - note that directory must
473    be empty when this is called. */
474
475 void hpfs_remove_fnode(struct super_block *s, fnode_secno fno)
476 {
477         struct buffer_head *bh;
478         struct fnode *fnode;
479         struct extended_attribute *ea;
480         struct extended_attribute *ea_end;
481         if (!(fnode = hpfs_map_fnode(s, fno, &bh))) return;
482         if (!fnode->dirflag) hpfs_remove_btree(s, &fnode->btree);
483         else hpfs_remove_dtree(s, fnode->u.external[0].disk_secno);
484         ea_end = fnode_end_ea(fnode);
485         for (ea = fnode_ea(fnode); ea < ea_end; ea = next_ea(ea))
486                 if (ea->indirect)
487                         hpfs_ea_remove(s, ea_sec(ea), ea->anode, ea_len(ea));
488         hpfs_ea_ext_remove(s, fnode->ea_secno, fnode->ea_anode, fnode->ea_size_l);
489         brelse(bh);
490         hpfs_free_sectors(s, fno, 1);
491 }