]> git.proxmox.com Git - grub2.git/blob - grub-core/fs/btrfs.c
projects / grub2.git / blob
? search:
summary | shortlog | log | commit | commitdiff | tree
blame | history | raw | HEAD
merge trunk
[grub2.git] / grub-core / fs / btrfs.c
1 /* btrfs.c - B-tree file system. */
2 /*
3 * GRUB -- GRand Unified Bootloader
4 * Copyright (C) 2010 Free Software Foundation, Inc.
5 *
6 * GRUB 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 3 of the License, or
9 * (at your option) any later version.
10 *
11 * GRUB 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 GRUB. If not, see <http://www.gnu.org/licenses/>.
18 */
19
20 #include <grub/err.h>
21 #include <grub/file.h>
22 #include <grub/mm.h>
23 #include <grub/misc.h>
24 #include <grub/disk.h>
25 #include <grub/dl.h>
26 #include <grub/types.h>
27 #include <grub/lib/crc.h>
28 #include <grub/deflate.h>
29
30 GRUB_MOD_LICENSE ("GPLv3+");
31
32 #define GRUB_BTRFS_SIGNATURE "_BHRfS_M"
33
34 typedef grub_uint8_t grub_btrfs_checksum_t[0x20];
35 typedef grub_uint16_t grub_btrfs_uuid_t[8];
36
37 struct grub_btrfs_device
38 {
39 grub_uint64_t device_id;
40 grub_uint8_t dummy[0x62 - 8];
41 } __attribute__ ((packed));
42
43 struct grub_btrfs_superblock
44 {
45 grub_btrfs_checksum_t checksum;
46 grub_btrfs_uuid_t uuid;
47 grub_uint8_t dummy[0x10];
48 grub_uint8_t signature[sizeof (GRUB_BTRFS_SIGNATURE) - 1];
49 grub_uint64_t generation;
50 grub_uint64_t root_tree;
51 grub_uint64_t chunk_tree;
52 grub_uint8_t dummy2[0x20];
53 grub_uint64_t root_dir_objectid;
54 grub_uint8_t dummy3[0x41];
55 struct grub_btrfs_device this_device;
56 char label[0x100];
57 grub_uint8_t dummy4[0x100];
58 grub_uint8_t bootstrap_mapping[0x800];
59 } __attribute__ ((packed));
60
61 struct btrfs_header
62 {
63 grub_btrfs_checksum_t checksum;
64 grub_btrfs_uuid_t uuid;
65 grub_uint8_t dummy[0x30];
66 grub_uint32_t nitems;
67 grub_uint8_t level;
68 } __attribute__ ((packed));
69
70 struct grub_btrfs_device_desc
71 {
72 grub_device_t dev;
73 grub_uint64_t id;
74 };
75
76 struct grub_btrfs_data
77 {
78 struct grub_btrfs_superblock sblock;
79 grub_uint64_t tree;
80 grub_uint64_t inode;
81
82 struct grub_btrfs_device_desc *devices_attached;
83 unsigned n_devices_attached;
84 unsigned n_devices_allocated;
85
86 /* Cached extent data. */
87 grub_uint64_t extstart;
88 grub_uint64_t extend;
89 grub_uint64_t extino;
90 grub_uint64_t exttree;
91 grub_size_t extsize;
92 struct grub_btrfs_extent_data *extent;
93 };
94
95 struct grub_btrfs_key
96 {
97 grub_uint64_t object_id;
98 #define GRUB_BTRFS_ITEM_TYPE_INODE_ITEM 0x01
99 #define GRUB_BTRFS_ITEM_TYPE_DIR_ITEM 0x54
100 #define GRUB_BTRFS_ITEM_TYPE_EXTENT_ITEM 0x6c
101 #define GRUB_BTRFS_ITEM_TYPE_ROOT_ITEM 0x84
102 #define GRUB_BTRFS_ITEM_TYPE_DEVICE 0xd8
103 #define GRUB_BTRFS_ITEM_TYPE_CHUNK 0xe4
104 grub_uint8_t type;
105 grub_uint64_t offset;
106 } __attribute__ ((packed));
107
108 struct grub_btrfs_chunk_item
109 {
110 grub_uint64_t size;
111 grub_uint64_t dummy;
112 grub_uint64_t stripe_length;
113 grub_uint64_t type;
114 #define GRUB_BTRFS_CHUNK_TYPE_BITS_DONTCARE 0x07
115 #define GRUB_BTRFS_CHUNK_TYPE_SINGLE 0x00
116 #define GRUB_BTRFS_CHUNK_TYPE_RAID0 0x08
117 #define GRUB_BTRFS_CHUNK_TYPE_RAID1 0x10
118 #define GRUB_BTRFS_CHUNK_TYPE_DUPLICATED 0x20
119 #define GRUB_BTRFS_CHUNK_TYPE_RAID10 0x40
120 grub_uint8_t dummy2[0xc];
121 grub_uint16_t nstripes;
122 grub_uint16_t nsubstripes;
123 } __attribute__ ((packed));
124
125 struct grub_btrfs_chunk_stripe
126 {
127 grub_uint64_t device_id;
128 grub_uint64_t offset;
129 grub_btrfs_uuid_t device_uuid;
130 } __attribute__ ((packed));
131
132 struct grub_btrfs_leaf_node
133 {
134 struct grub_btrfs_key key;
135 grub_uint32_t offset;
136 grub_uint32_t size;
137 } __attribute__ ((packed));
138
139 struct grub_btrfs_internal_node
140 {
141 struct grub_btrfs_key key;
142 grub_uint64_t addr;
143 grub_uint64_t dummy;
144 } __attribute__ ((packed));
145
146 struct grub_btrfs_dir_item
147 {
148 struct grub_btrfs_key key;
149 grub_uint8_t dummy[8];
150 grub_uint16_t m;
151 grub_uint16_t n;
152 #define GRUB_BTRFS_DIR_ITEM_TYPE_REGULAR 1
153 #define GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY 2
154 #define GRUB_BTRFS_DIR_ITEM_TYPE_SYMLINK 7
155 grub_uint8_t type;
156 char name[0];
157 } __attribute__ ((packed));
158
159 struct grub_btrfs_leaf_descriptor
160 {
161 unsigned depth;
162 unsigned allocated;
163 struct {
164 grub_disk_addr_t addr;
165 unsigned iter;
166 unsigned maxiter;
167 int leaf;
168 } *data;
169 };
170
171 struct grub_btrfs_root_item
172 {
173 grub_uint8_t dummy[0xb0];
174 grub_uint64_t tree;
175 grub_uint64_t inode;
176 };
177
178 struct grub_btrfs_time
179 {
180 grub_int64_t sec;
181 grub_uint32_t nanosec;
182 } __attribute__ ((aligned(4)));
183
184 struct grub_btrfs_inode
185 {
186 grub_uint8_t dummy1[0x10];
187 grub_uint64_t size;
188 grub_uint8_t dummy2[0x70];
189 struct grub_btrfs_time mtime;
190 } __attribute__ ((packed));
191
192 struct grub_btrfs_extent_data
193 {
194 grub_uint64_t dummy;
195 grub_uint64_t size;
196 grub_uint8_t compression;
197 grub_uint8_t encryption;
198 grub_uint16_t encoding;
199 grub_uint8_t type;
200 union
201 {
202 char inl[0];
203 struct
204 {
205 grub_uint64_t laddr;
206 grub_uint64_t compressed_size;
207 grub_uint64_t offset;
208 grub_uint64_t filled;
209 };
210 };
211 } __attribute__ ((packed));
212
213 #define GRUB_BTRFS_EXTENT_INLINE 0
214 #define GRUB_BTRFS_EXTENT_REGULAR 1
215
216 #define GRUB_BTRFS_COMPRESSION_NONE 0
217 #define GRUB_BTRFS_COMPRESSION_ZLIB 1
218
219 #define GRUB_BTRFS_OBJECT_ID_CHUNK 0x100
220
221 static grub_disk_addr_t superblock_sectors[] = { 64 * 2, 64 * 1024 * 2,
222 256 * 1048576 * 2,
223 1048576ULL * 1048576ULL * 2 };
224
225 static grub_err_t
226 grub_btrfs_read_logical (struct grub_btrfs_data *data,
227 grub_disk_addr_t addr, void *buf, grub_size_t size);
228
229 static grub_err_t
230 read_sblock (grub_disk_t disk, struct grub_btrfs_superblock *sb)
231 {
232 unsigned i;
233 grub_err_t err = GRUB_ERR_NONE;
234 for (i = 0; i < ARRAY_SIZE (superblock_sectors); i++)
235 {
236 struct grub_btrfs_superblock sblock;
237 err = grub_disk_read (disk, superblock_sectors[i], 0,
238 sizeof (sblock), &sblock);
239 if (err == GRUB_ERR_OUT_OF_RANGE)
240 break;
241
242 if (grub_memcmp ((char *) sblock.signature, GRUB_BTRFS_SIGNATURE,
243 sizeof (GRUB_BTRFS_SIGNATURE) - 1) != 0)
244 break;
245 if (i == 0 || grub_le_to_cpu64 (sblock.generation)
246 > grub_le_to_cpu64 (sb->generation))
247 grub_memcpy (sb, &sblock, sizeof (sblock));
248 }
249
250 if ((err == GRUB_ERR_OUT_OF_RANGE || !err) && i == 0)
251 return grub_error (GRUB_ERR_BAD_FS, "not a Btrfs filesystem");
252
253 if (err == GRUB_ERR_OUT_OF_RANGE)
254 grub_errno = err = GRUB_ERR_NONE;
255
256 return err;
257 }
258
259 static int
260 key_cmp (const struct grub_btrfs_key *a, const struct grub_btrfs_key *b)
261 {
262 if (grub_cpu_to_le64 (a->object_id) < grub_cpu_to_le64 (b->object_id))
263 return -1;
264 if (grub_cpu_to_le64 (a->object_id) > grub_cpu_to_le64 (b->object_id))
265 return +1;
266
267 if (a->type < b->type)
268 return -1;
269 if (a->type > b->type)
270 return +1;
271
272 if (grub_cpu_to_le64 (a->offset) < grub_cpu_to_le64 (b->offset))
273 return -1;
274 if (grub_cpu_to_le64 (a->offset) > grub_cpu_to_le64 (b->offset))
275 return +1;
276 return 0;
277 }
278
279 static void
280 free_iterator (struct grub_btrfs_leaf_descriptor *desc)
281 {
282 grub_free (desc->data);
283 }
284
285 static grub_err_t
286 save_ref (struct grub_btrfs_leaf_descriptor *desc,
287 grub_disk_addr_t addr, unsigned i, unsigned m, int l)
288 {
289 desc->depth++;
290 if (desc->allocated < desc->depth)
291 {
292 void *newdata;
293 desc->allocated *= 2;
294 newdata = grub_realloc (desc->data, sizeof (desc->data[0])
295 * desc->allocated);
296 if (!newdata)
297 return grub_errno;
298 desc->data = newdata;
299 }
300 desc->data[desc->depth - 1].addr = addr;
301 desc->data[desc->depth - 1].iter = i;
302 desc->data[desc->depth - 1].maxiter = m;
303 desc->data[desc->depth - 1].leaf = l;
304 return GRUB_ERR_NONE;
305 }
306
307 static int
308 next (struct grub_btrfs_data *data,
309 struct grub_btrfs_leaf_descriptor *desc,
310 grub_disk_addr_t *outaddr, grub_size_t *outsize,
311 struct grub_btrfs_key *key_out)
312 {
313 grub_err_t err;
314 struct grub_btrfs_leaf_node leaf;
315
316 for (; desc->depth > 0; desc->depth--)
317 {
318 desc->data[desc->depth - 1].iter++;
319 if (desc->data[desc->depth - 1].iter
320 < desc->data[desc->depth - 1].maxiter)
321 break;
322 }
323 if (desc->depth == 0)
324 return 0;
325 while (!desc->data[desc->depth - 1].leaf)
326 {
327 struct grub_btrfs_internal_node node;
328 struct btrfs_header head;
329
330 err = grub_btrfs_read_logical (data, desc->data[desc->depth - 1].iter
331 * sizeof (node)
332 + sizeof (struct btrfs_header)
333 + desc->data[desc->depth - 1].addr, &node,
334 sizeof (node));
335 if (err)
336 return -err;
337
338 err = grub_btrfs_read_logical (data, grub_le_to_cpu64 (node.addr), &head,
339 sizeof (head));
340 if (err)
341 return -err;
342
343 save_ref (desc, grub_le_to_cpu64 (node.addr), 0,
344 grub_le_to_cpu32 (head.nitems), !head.level);
345 }
346 err = grub_btrfs_read_logical (data, desc->data[desc->depth - 1].iter
347 * sizeof (leaf)
348 + sizeof (struct btrfs_header)
349 + desc->data[desc->depth - 1].addr, &leaf,
350 sizeof (leaf));
351 if (err)
352 return -err;
353 *outsize = grub_le_to_cpu32 (leaf.size);
354 *outaddr = desc->data[desc->depth - 1].addr + sizeof (struct btrfs_header)
355 + grub_le_to_cpu32 (leaf.offset);
356 *key_out = leaf.key;
357 return 1;
358 }
359
360 static grub_err_t
361 lower_bound (struct grub_btrfs_data *data,
362 const struct grub_btrfs_key *key_in,
363 struct grub_btrfs_key *key_out,
364 grub_disk_addr_t root,
365 grub_disk_addr_t *outaddr, grub_size_t *outsize,
366 struct grub_btrfs_leaf_descriptor *desc)
367 {
368 grub_disk_addr_t addr = root;
369 int depth = -1;
370
371 if (desc)
372 {
373 desc->allocated = 16;
374 desc->depth = 0;
375 desc->data = grub_malloc (sizeof (desc->data[0]) * desc->allocated);
376 if (!desc->data)
377 return grub_errno;
378 }
379
380 grub_dprintf ("btrfs",
381 "retrieving %" PRIxGRUB_UINT64_T
382 " %x %" PRIxGRUB_UINT64_T "\n",
383 key_in->object_id, key_in->type, key_in->offset);
384
385 while (1)
386 {
387 grub_err_t err;
388 struct btrfs_header head;
389
390 reiter:
391 depth++;
392 /* FIXME: preread few nodes into buffer. */
393 err = grub_btrfs_read_logical (data, addr, &head, sizeof (head));
394 if (err)
395 return err;
396 addr += sizeof (head);
397 if (head.level)
398 {
399 unsigned i;
400 struct grub_btrfs_internal_node node, node_last;
401 int have_last = 0;
402 grub_memset (&node_last, 0, sizeof (node_last));
403 for (i = 0; i < grub_le_to_cpu32 (head.nitems); i++)
404 {
405 err = grub_btrfs_read_logical (data, addr + i * sizeof (node),
406 &node, sizeof (node));
407 if (err)
408 return err;
409
410 grub_dprintf ("btrfs",
411 "internal node (depth %d) %" PRIxGRUB_UINT64_T
412 " %x %" PRIxGRUB_UINT64_T "\n", depth,
413 node.key.object_id, node.key.type, node.key.offset);
414
415 if (key_cmp (&node.key, key_in) == 0)
416 {
417 err = GRUB_ERR_NONE;
418 if (desc)
419 err = save_ref (desc, addr - sizeof (head), i,
420 grub_le_to_cpu32 (head.nitems), 0);
421 if (err)
422 return err;
423 addr = grub_le_to_cpu64 (node.addr);
424 goto reiter;
425 }
426 if (key_cmp (&node.key, key_in) > 0)
427 break;
428 node_last = node;
429 have_last = 1;
430 }
431 if (have_last)
432 {
433 err = GRUB_ERR_NONE;
434 if (desc)
435 err = save_ref (desc, addr - sizeof (head), i - 1,
436 grub_le_to_cpu32 (head.nitems), 0);
437 if (err)
438 return err;
439 addr = grub_le_to_cpu64 (node_last.addr);
440 goto reiter;
441 }
442 *outsize = 0;
443 *outaddr = 0;
444 grub_memset (key_out, 0, sizeof (*key_out));
445 if (desc)
446 return save_ref (desc, addr - sizeof (head), -1,
447 grub_le_to_cpu32 (head.nitems), 0);
448 return GRUB_ERR_NONE;
449 }
450 {
451 unsigned i;
452 struct grub_btrfs_leaf_node leaf, leaf_last;
453 int have_last = 0;
454 for (i = 0; i < grub_le_to_cpu32 (head.nitems); i++)
455 {
456 err = grub_btrfs_read_logical (data, addr + i * sizeof (leaf),
457 &leaf, sizeof (leaf));
458 if (err)
459 return err;
460
461 grub_dprintf ("btrfs",
462 "leaf (depth %d) %" PRIxGRUB_UINT64_T
463 " %x %" PRIxGRUB_UINT64_T "\n", depth,
464 leaf.key.object_id, leaf.key.type, leaf.key.offset);
465
466 if (key_cmp (&leaf.key, key_in) == 0)
467 {
468 grub_memcpy (key_out, &leaf.key, sizeof(*key_out));
469 *outsize = grub_le_to_cpu32 (leaf.size);
470 *outaddr = addr + grub_le_to_cpu32 (leaf.offset);
471 if (desc)
472 return save_ref (desc, addr - sizeof (head), i,
473 grub_le_to_cpu32 (head.nitems), 1);
474 return GRUB_ERR_NONE;
475 }
476
477 if (key_cmp (&leaf.key, key_in) > 0)
478 break;
479
480 have_last = 1;
481 leaf_last = leaf;
482 }
483
484 if (have_last)
485 {
486 grub_memcpy (key_out, &leaf_last.key, sizeof(*key_out));
487 *outsize = grub_le_to_cpu32 (leaf_last.size);
488 *outaddr = addr + grub_le_to_cpu32 (leaf_last.offset);
489 if (desc)
490 return save_ref (desc, addr - sizeof (head), i - 1,
491 grub_le_to_cpu32 (head.nitems), 1);
492 return GRUB_ERR_NONE;
493 }
494 *outsize = 0;
495 *outaddr = 0;
496 grub_memset (key_out, 0, sizeof (*key_out));
497 if (desc)
498 return save_ref (desc, addr - sizeof (head), -1,
499 grub_le_to_cpu32 (head.nitems), 1);
500 return GRUB_ERR_NONE;
501 }
502 }
503 }
504
505 static grub_device_t
506 find_device (struct grub_btrfs_data *data, grub_uint64_t id,
507 int do_rescan)
508 {
509 grub_device_t dev_found = NULL;
510 auto int hook (const char *name);
511 int hook (const char *name)
512 {
513 grub_device_t dev;
514 grub_err_t err;
515 struct grub_btrfs_superblock sb;
516 dev = grub_device_open (name);
517 if (!dev)
518 return 0;
519 if (!dev->disk)
520 {
521 grub_device_close (dev);
522 return 0;
523 }
524 err = read_sblock (dev->disk, &sb);
525 if (err == GRUB_ERR_BAD_FS)
526 {
527 grub_device_close (dev);
528 grub_errno = GRUB_ERR_NONE;
529 return 0;
530 }
531 if (err)
532 {
533 grub_device_close (dev);
534 grub_print_error ();
535 return 0;
536 }
537 if (grub_memcmp (data->sblock.uuid, sb.uuid, sizeof (sb.uuid)) != 0
538 || sb.this_device.device_id != id)
539 {
540 grub_device_close (dev);
541 return 0;
542 }
543
544 dev_found = dev;
545 return 1;
546 }
547
548 unsigned i;
549
550 for (i = 0; i < data->n_devices_attached; i++)
551 if (id == data->devices_attached[i].id)
552 return data->devices_attached[i].dev;
553 if (do_rescan)
554 grub_device_iterate (hook);
555 if (!dev_found)
556 {
557 grub_error (GRUB_ERR_BAD_FS, "couldn't find a member device");
558 return NULL;
559 }
560 data->n_devices_attached++;
561 if (data->n_devices_attached > data->n_devices_allocated)
562 {
563 void *tmp;
564 data->n_devices_allocated = 2 * data->n_devices_attached + 1;
565 data->devices_attached
566 = grub_realloc (tmp = data->devices_attached,
567 data->n_devices_allocated
568 * sizeof (data->devices_attached[0]));
569 if (!data->devices_attached)
570 {
571 grub_device_close (dev_found);
572 data->devices_attached = tmp;
573 return NULL;
574 }
575 }
576 data->devices_attached[data->n_devices_attached - 1].id = id;
577 data->devices_attached[data->n_devices_attached - 1].dev = dev_found;
578 return dev_found;
579 }
580
581 static grub_err_t
582 grub_btrfs_read_logical (struct grub_btrfs_data *data,
583 grub_disk_addr_t addr,
584 void *buf, grub_size_t size)
585 {
586 while (size > 0)
587 {
588 grub_uint8_t *ptr;
589 struct grub_btrfs_key *key;
590 struct grub_btrfs_chunk_item *chunk;
591 grub_uint64_t csize;
592 grub_err_t err;
593 struct grub_btrfs_key key_out;
594 int challoc = 0;
595 grub_device_t dev;
596 grub_dprintf ("btrfs", "searching for laddr %" PRIxGRUB_UINT64_T "\n",
597 addr);
598 for (ptr = data->sblock.bootstrap_mapping;
599 ptr < data->sblock.bootstrap_mapping
600 + sizeof (data->sblock.bootstrap_mapping)
601 - sizeof (struct grub_btrfs_key);
602 )
603 {
604 key = (struct grub_btrfs_key *) ptr;
605 if (key->type != GRUB_BTRFS_ITEM_TYPE_CHUNK)
606 break;
607 chunk = (struct grub_btrfs_chunk_item *) (key + 1);
608 grub_dprintf ("btrfs", "%" PRIxGRUB_UINT64_T " %" PRIxGRUB_UINT64_T " \n",
609 grub_le_to_cpu64 (key->offset),
610 grub_le_to_cpu64 (chunk->size));
611 if (grub_le_to_cpu64 (key->offset) <= addr
612 && addr < grub_le_to_cpu64 (key->offset)
613 + grub_le_to_cpu64 (chunk->size))
614 goto chunk_found;
615 ptr += sizeof (*key) + sizeof (*chunk)
616 + sizeof (struct grub_btrfs_chunk_stripe)
617 * grub_le_to_cpu16 (chunk->nstripes);
618 }
619 struct grub_btrfs_key key_in;
620 grub_size_t chsize;
621 grub_disk_addr_t chaddr;
622 key_in.object_id = GRUB_BTRFS_OBJECT_ID_CHUNK;
623 key_in.type = GRUB_BTRFS_ITEM_TYPE_CHUNK;
624 key_in.offset = addr;
625 err = lower_bound (data, &key_in, &key_out,
626 grub_le_to_cpu64 (data->sblock.chunk_tree),
627 &chaddr, &chsize, NULL);
628 if (err)
629 return err;
630 key = &key_out;
631 if (key->type != GRUB_BTRFS_ITEM_TYPE_CHUNK
632 || !(grub_le_to_cpu64 (key->offset) <= addr))
633 return grub_error (GRUB_ERR_BAD_FS,
634 "couldn't find the chunk descriptor");
635
636 chunk = grub_malloc (chsize);
637 if (!chunk)
638 return grub_errno;
639
640 challoc = 1;
641 err = grub_btrfs_read_logical (data, chaddr, chunk, chsize);
642 if (err)
643 {
644 grub_free (chunk);
645 return err;
646 }
647
648 chunk_found:
649 {
650 grub_uint32_t stripen;
651 grub_uint64_t stripe_offset;
652 grub_uint64_t off = addr - grub_le_to_cpu64 (key->offset);
653 unsigned redundancy = 1;
654 unsigned i, j;
655
656 if (grub_le_to_cpu64 (chunk->size) <= off)
657 {
658 grub_dprintf ("btrfs", "no chunk\n");
659 return grub_error (GRUB_ERR_BAD_FS,
660 "couldn't find the chunk descriptor");
661 }
662
663 grub_dprintf ("btrfs", "chunk 0x%" PRIxGRUB_UINT64_T
664 "+0x%" PRIxGRUB_UINT64_T
665 " (%d stripes (%d substripes) of %"
666 PRIxGRUB_UINT64_T ")\n",
667 grub_le_to_cpu64 (key->offset),
668 grub_le_to_cpu64 (chunk->size),
669 grub_le_to_cpu16 (chunk->nstripes),
670 grub_le_to_cpu16 (chunk->nsubstripes),
671 grub_le_to_cpu64 (chunk->stripe_length));
672
673 switch (grub_le_to_cpu64 (chunk->type)
674 & ~GRUB_BTRFS_CHUNK_TYPE_BITS_DONTCARE)
675 {
676 case GRUB_BTRFS_CHUNK_TYPE_SINGLE:
677 {
678 grub_uint64_t stripe_length;
679 grub_dprintf ("btrfs", "single\n");
680 stripe_length = grub_divmod64_full (grub_le_to_cpu64 (chunk->size),
681 grub_le_to_cpu16 (chunk->nstripes),
682 NULL);
683 stripen = grub_divmod64_full (off, stripe_length, &stripe_offset);
684 csize = (stripen + 1) * stripe_length - off;
685 break;
686 }
687 case GRUB_BTRFS_CHUNK_TYPE_DUPLICATED:
688 case GRUB_BTRFS_CHUNK_TYPE_RAID1:
689 {
690 grub_dprintf ("btrfs", "RAID1\n");
691 stripen = 0;
692 stripe_offset = off;
693 csize = grub_le_to_cpu64 (chunk->size) - off;
694 redundancy = 2;
695 break;
696 }
697 case GRUB_BTRFS_CHUNK_TYPE_RAID0:
698 {
699 grub_uint64_t middle, high;
700 grub_uint32_t low;
701 grub_dprintf ("btrfs", "RAID0\n");
702 middle = grub_divmod64 (off,
703 grub_le_to_cpu64 (chunk->stripe_length),
704 &low);
705
706 high = grub_divmod64 (middle, grub_le_to_cpu16 (chunk->nstripes),
707 &stripen);
708 stripe_offset = low + grub_le_to_cpu64 (chunk->stripe_length)
709 * high;
710 csize = grub_le_to_cpu64 (chunk->stripe_length) - low;
711 break;
712 }
713 case GRUB_BTRFS_CHUNK_TYPE_RAID10:
714 {
715 grub_uint64_t middle, high;
716 grub_uint32_t low;
717 middle = grub_divmod64 (off,
718 grub_le_to_cpu64 (chunk->stripe_length),
719 &low);
720
721 high = grub_divmod64 (middle,
722 grub_le_to_cpu16 (chunk->nsubstripes),
723 &stripen);
724 stripen *= grub_le_to_cpu16 (chunk->nstripes)
725 / grub_le_to_cpu16 (chunk->nsubstripes);
726 redundancy = grub_le_to_cpu16 (chunk->nstripes)
727 / grub_le_to_cpu16 (chunk->nsubstripes);
728 stripe_offset = low + grub_le_to_cpu64 (chunk->stripe_length)
729 * high;
730 csize = grub_le_to_cpu64 (chunk->stripe_length) - low;
731 break;
732 }
733 default:
734 grub_dprintf ("btrfs", "unsupported RAID\n");
735 return grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET,
736 "unsupported RAID flags %" PRIxGRUB_UINT64_T,
737 grub_le_to_cpu64 (chunk->type));
738 }
739 if (csize == 0)
740 return grub_error (GRUB_ERR_BUG,
741 "couldn't find the chunk descriptor");
742 if ((grub_size_t) csize > size)
743 csize = size;
744
745 for (j = 0; j < 2; j++)
746 {
747 for (i = 0; i < redundancy; i++)
748 {
749 struct grub_btrfs_chunk_stripe *stripe;
750 grub_disk_addr_t paddr;
751
752 stripe = (struct grub_btrfs_chunk_stripe *) (chunk + 1);
753 /* Right now the redundancy handling is easy.
754 With RAID5-like it will be more difficult. */
755 stripe += stripen + i;
756
757 paddr = stripe->offset + stripe_offset;
758
759 grub_dprintf ("btrfs", "chunk 0x%" PRIxGRUB_UINT64_T
760 "+0x%" PRIxGRUB_UINT64_T " (%d stripes (%d substripes) of %"
761 PRIxGRUB_UINT64_T ") stripe %" PRIxGRUB_UINT32_T
762 " maps to 0x%" PRIxGRUB_UINT64_T "\n",
763 grub_le_to_cpu64 (key->offset),
764 grub_le_to_cpu64 (chunk->size),
765 grub_le_to_cpu16 (chunk->nstripes),
766 grub_le_to_cpu16 (chunk->nsubstripes),
767 grub_le_to_cpu64 (chunk->stripe_length),
768 stripen, stripe->offset);
769 grub_dprintf ("btrfs", "reading paddr 0x%" PRIxGRUB_UINT64_T
770 " for laddr 0x%" PRIxGRUB_UINT64_T"\n", paddr,
771 addr);
772
773 dev = find_device (data, stripe->device_id, j);
774 if (!dev)
775 {
776 err = grub_errno;
777 grub_errno = GRUB_ERR_NONE;
778 continue;
779 }
780
781 err = grub_disk_read (dev->disk, paddr >> GRUB_DISK_SECTOR_BITS,
782 paddr & (GRUB_DISK_SECTOR_SIZE - 1),
783 csize, buf);
784 if (!err)
785 break;
786 grub_errno = GRUB_ERR_NONE;
787 }
788 if (i != redundancy)
789 break;
790 }
791 if (err)
792 return grub_errno = err;
793 }
794 size -= csize;
795 buf = (grub_uint8_t *) buf + csize;
796 addr += csize;
797 if (challoc)
798 grub_free (chunk);
799 }
800 return GRUB_ERR_NONE;
801 }
802
803 static struct grub_btrfs_data *
804 grub_btrfs_mount (grub_device_t dev)
805 {
806 struct grub_btrfs_data *data;
807 grub_err_t err;
808
809 if (!dev->disk)
810 {
811 grub_error (GRUB_ERR_BAD_FS, "not BtrFS");
812 return NULL;
813 }
814
815 data = grub_zalloc (sizeof (*data));
816 if (! data)
817 return NULL;
818
819 err = read_sblock (dev->disk, &data->sblock);
820 if (err)
821 {
822 grub_free (data);
823 return NULL;
824 }
825
826 data->n_devices_allocated = 16;
827 data->devices_attached = grub_malloc (sizeof (data->devices_attached[0])
828 * data->n_devices_allocated);
829 if (!data->devices_attached)
830 {
831 grub_free (data);
832 return NULL;
833 }
834 data->n_devices_attached = 1;
835 data->devices_attached[0].dev = dev;
836 data->devices_attached[0].id = data->sblock.this_device.device_id;
837
838 return data;
839 }
840
841 static void
842 grub_btrfs_unmount (struct grub_btrfs_data *data)
843 {
844 unsigned i;
845 /* The device 0 is closed one layer upper. */
846 for (i = 1; i < data->n_devices_attached; i++)
847 grub_device_close (data->devices_attached[i].dev);
848 grub_free (data->devices_attached);
849 grub_free (data->extent);
850 grub_free (data);
851 }
852
853 static grub_err_t
854 grub_btrfs_read_inode (struct grub_btrfs_data *data,
855 struct grub_btrfs_inode *inode, grub_uint64_t num,
856 grub_uint64_t tree)
857 {
858 struct grub_btrfs_key key_in, key_out;
859 grub_disk_addr_t elemaddr;
860 grub_size_t elemsize;
861 grub_err_t err;
862
863 key_in.object_id = num;
864 key_in.type = GRUB_BTRFS_ITEM_TYPE_INODE_ITEM;
865 key_in.offset = 0;
866
867 err = lower_bound (data, &key_in, &key_out, tree,
868 &elemaddr, &elemsize, NULL);
869 if (err)
870 return err;
871 if (num != key_out.object_id
872 || key_out.type != GRUB_BTRFS_ITEM_TYPE_INODE_ITEM)
873 return grub_error (GRUB_ERR_BAD_FS, "inode not found");
874
875 return grub_btrfs_read_logical (data, elemaddr, inode, sizeof (*inode));
876 }
877
878 static grub_ssize_t
879 grub_btrfs_extent_read (struct grub_btrfs_data *data,
880 grub_uint64_t ino, grub_uint64_t tree,
881 grub_off_t pos0, char *buf, grub_size_t len)
882 {
883 grub_off_t pos = pos0;
884 while (len)
885 {
886 grub_size_t csize;
887 grub_err_t err;
888 grub_off_t extoff;
889 if (!data->extent || data->extstart > pos || data->extino != ino
890 || data->exttree != tree || data->extend <= pos)
891 {
892 struct grub_btrfs_key key_in, key_out;
893 grub_disk_addr_t elemaddr;
894 grub_size_t elemsize;
895
896 grub_free (data->extent);
897 key_in.object_id = ino;
898 key_in.type = GRUB_BTRFS_ITEM_TYPE_EXTENT_ITEM;
899 key_in.offset = grub_cpu_to_le64 (pos);
900 err = lower_bound (data, &key_in, &key_out, tree,
901 &elemaddr, &elemsize, NULL);
902 if (err)
903 return -1;
904 if (key_out.object_id != ino
905 || key_out.type != GRUB_BTRFS_ITEM_TYPE_EXTENT_ITEM)
906 {
907 grub_error (GRUB_ERR_BAD_FS, "extent not found");
908 return -1;
909 }
910 data->extstart = grub_le_to_cpu64 (key_out.offset);
911 data->extsize = elemsize;
912 data->extent = grub_malloc (elemsize);
913 data->extino = ino;
914 data->exttree = tree;
915 if (!data->extent)
916 return grub_errno;
917
918 err = grub_btrfs_read_logical (data, elemaddr,
919 data->extent, elemsize);
920 if (err)
921 return err;
922
923 data->extend = data->extstart
924 + grub_le_to_cpu64 (data->extent->size);
925 if (data->extent->type == GRUB_BTRFS_EXTENT_REGULAR
926 && (char *) &data->extent + elemsize
927 >= (char *) &data->extent->filled
928 + sizeof (data->extent->filled))
929 data->extend = data->extstart
930 + grub_le_to_cpu64 (data->extent->filled);
931
932 grub_dprintf ("btrfs", "extent 0x%" PRIxGRUB_UINT64_T "+0x%"
933 PRIxGRUB_UINT64_T " (0x%"
934 PRIxGRUB_UINT64_T ")\n",
935 grub_le_to_cpu64 (key_out.offset),
936 grub_le_to_cpu64 (data->extent->size),
937 grub_le_to_cpu64 (data->extent->filled));
938 if (data->extend <= pos)
939 {
940 grub_error (GRUB_ERR_BAD_FS, "extent not found");
941 return -1;
942 }
943 }
944 csize = data->extend - pos;
945 extoff = pos - data->extstart;
946 if (csize > len)
947 csize = len;
948
949 if (data->extent->encryption)
950 {
951 grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET,
952 "encryption not supported");
953 return -1;
954 }
955
956 if (data->extent->compression != GRUB_BTRFS_COMPRESSION_NONE
957 && data->extent->compression != GRUB_BTRFS_COMPRESSION_ZLIB)
958 {
959 grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET,
960 "compression type 0x%x not supported",
961 data->extent->compression);
962 return -1;
963 }
964
965 if (data->extent->encoding)
966 {
967 grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET,
968 "encoding not supported");
969 return -1;
970 }
971
972 switch (data->extent->type)
973 {
974 case GRUB_BTRFS_EXTENT_INLINE:
975 if (data->extent->compression == GRUB_BTRFS_COMPRESSION_ZLIB)
976 {
977 if (grub_zlib_decompress (data->extent->inl, data->extsize -
978 ((grub_uint8_t *) data->extent->inl
979 - (grub_uint8_t *) data->extent),
980 extoff, buf, csize)
981 != (grub_ssize_t) csize)
982 return -1;
983 }
984 else
985 grub_memcpy (buf, data->extent->inl + extoff, csize);
986 break;
987 case GRUB_BTRFS_EXTENT_REGULAR:
988 if (!data->extent->laddr)
989 {
990 grub_memset (buf, 0, csize);
991 break;
992 }
993 if (data->extent->compression == GRUB_BTRFS_COMPRESSION_ZLIB)
994 {
995 char *tmp;
996 grub_uint64_t zsize;
997 zsize = grub_le_to_cpu64 (data->extent->compressed_size);
998 tmp = grub_malloc (zsize);
999 if (!tmp)
1000 return -1;
1001 err = grub_btrfs_read_logical (data,
1002 grub_le_to_cpu64 (data->extent->laddr),
1003 tmp, zsize);
1004 if (err)
1005 {
1006 grub_free (tmp);
1007 return -1;
1008 }
1009 if (grub_zlib_decompress (tmp, zsize, extoff
1010 + grub_le_to_cpu64 (data->extent->offset),
1011 buf, csize) != (grub_ssize_t) csize)
1012 {
1013 grub_free (tmp);
1014 return -1;
1015 }
1016 grub_free (tmp);
1017 break;
1018 }
1019 err = grub_btrfs_read_logical (data,
1020 grub_le_to_cpu64 (data->extent->laddr)
1021 + grub_le_to_cpu64 (data->extent->offset)
1022 + extoff,
1023 buf, csize);
1024 if (err)
1025 return -1;
1026 break;
1027 default:
1028 grub_error (GRUB_ERR_NOT_IMPLEMENTED_YET,
1029 "unsupported extent type 0x%x", data->extent->type);
1030 return -1;
1031 }
1032 buf += csize;
1033 pos += csize;
1034 len -= csize;
1035 }
1036 return pos - pos0;
1037 }
1038
1039 static grub_err_t
1040 find_path (struct grub_btrfs_data *data,
1041 const char *path, struct grub_btrfs_key *key,
1042 grub_uint64_t *tree, grub_uint8_t *type)
1043 {
1044 const char *slash = path;
1045 grub_err_t err;
1046 grub_disk_addr_t elemaddr;
1047 grub_size_t elemsize;
1048 grub_size_t allocated = 0;
1049 struct grub_btrfs_dir_item *direl = NULL;
1050 struct grub_btrfs_key key_out;
1051 int skip_default;
1052 const char *ctoken;
1053 grub_size_t ctokenlen;
1054 char *path_alloc = NULL;
1055 unsigned symlinks_max = 32;
1056
1057 *type = GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY;
1058 *tree = data->sblock.root_tree;
1059 key->object_id = data->sblock.root_dir_objectid;
1060 key->type = GRUB_BTRFS_ITEM_TYPE_DIR_ITEM;
1061 key->offset = 0;
1062 skip_default = 1;
1063
1064 while (1)
1065 {
1066 if (!skip_default)
1067 {
1068 while (path[0] == '/')
1069 path++;
1070 if (!path[0])
1071 break;
1072 slash = grub_strchr (path, '/');
1073 if (!slash)
1074 slash = path + grub_strlen (path);
1075 ctoken = path;
1076 ctokenlen = slash - path;
1077 }
1078 else
1079 {
1080 ctoken = "default";
1081 ctokenlen = sizeof ("default") - 1;
1082 }
1083
1084 if (*type != GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY)
1085 {
1086 grub_free (path_alloc);
1087 return grub_error (GRUB_ERR_BAD_FILE_TYPE, "not a directory");
1088 }
1089
1090 key->type = GRUB_BTRFS_ITEM_TYPE_DIR_ITEM;
1091 key->offset = grub_cpu_to_le64 (~grub_getcrc32c (1, ctoken, ctokenlen));
1092
1093 err = lower_bound (data, key, &key_out, *tree,
1094 &elemaddr, &elemsize, NULL);
1095 if (err)
1096 {
1097 grub_free (direl);
1098 grub_free (path_alloc);
1099 return err;
1100 }
1101 if (key_cmp (key, &key_out) != 0)
1102 {
1103 grub_free (direl);
1104 grub_free (path_alloc);
1105 return grub_error (GRUB_ERR_FILE_NOT_FOUND, "file not found");
1106 }
1107
1108 struct grub_btrfs_dir_item *cdirel;
1109 if (elemsize > allocated)
1110 {
1111 allocated = 2 * elemsize;
1112 grub_free (direl);
1113 direl = grub_malloc (allocated + 1);
1114 if (!direl)
1115 {
1116 grub_free (path_alloc);
1117 return grub_errno;
1118 }
1119 }
1120
1121 err = grub_btrfs_read_logical (data, elemaddr, direl, elemsize);
1122 if (err)
1123 {
1124 grub_free (direl);
1125 grub_free (path_alloc);
1126 return err;
1127 }
1128
1129 for (cdirel = direl;
1130 (grub_uint8_t *) cdirel - (grub_uint8_t *) direl
1131 < (grub_ssize_t) elemsize;
1132 cdirel = (void *) ((grub_uint8_t *) (direl + 1)
1133 + grub_le_to_cpu16 (cdirel->n)
1134 + grub_le_to_cpu16 (cdirel->m)))
1135 {
1136 if (ctokenlen == grub_le_to_cpu16 (cdirel->n)
1137 && grub_memcmp (cdirel->name, ctoken, ctokenlen) == 0)
1138 break;
1139 }
1140 if ((grub_uint8_t *) cdirel - (grub_uint8_t *) direl
1141 >= (grub_ssize_t) elemsize)
1142 {
1143 grub_free (direl);
1144 grub_free (path_alloc);
1145 return grub_error (GRUB_ERR_FILE_NOT_FOUND, "file not found");
1146 }
1147
1148 if (!skip_default)
1149 path = slash;
1150 skip_default = 0;
1151 if (cdirel->type == GRUB_BTRFS_DIR_ITEM_TYPE_SYMLINK)
1152 {
1153 struct grub_btrfs_inode inode;
1154 char *tmp;
1155 if (--symlinks_max == 0)
1156 {
1157 grub_free (direl);
1158 grub_free (path_alloc);
1159 return grub_error (GRUB_ERR_SYMLINK_LOOP,
1160 "too deep nesting of symlinks");
1161 }
1162
1163 err = grub_btrfs_read_inode (data, &inode,
1164 cdirel->key.object_id, *tree);
1165 if (err)
1166 {
1167 grub_free (direl);
1168 grub_free (path_alloc);
1169 return err;
1170 }
1171 tmp = grub_malloc (grub_le_to_cpu64 (inode.size)
1172 + grub_strlen (path) + 1);
1173 if (!tmp)
1174 {
1175 grub_free (direl);
1176 grub_free (path_alloc);
1177 return grub_errno;
1178 }
1179
1180 if (grub_btrfs_extent_read (data, cdirel->key.object_id,
1181 *tree, 0, tmp,
1182 grub_le_to_cpu64 (inode.size))
1183 != (grub_ssize_t) grub_le_to_cpu64 (inode.size))
1184 {
1185 grub_free (direl);
1186 grub_free (path_alloc);
1187 grub_free (tmp);
1188 return grub_errno;
1189 }
1190 grub_memcpy (tmp + grub_le_to_cpu64 (inode.size), path,
1191 grub_strlen (path) + 1);
1192 grub_free (path_alloc);
1193 path = path_alloc = tmp;
1194 if (path[0] == '/')
1195 {
1196 *type = GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY;
1197 *tree = data->sblock.root_tree;
1198 key->object_id = data->sblock.root_dir_objectid;
1199 key->type = GRUB_BTRFS_ITEM_TYPE_DIR_ITEM;
1200 key->offset = 0;
1201 skip_default = 1;
1202 }
1203 continue;
1204 }
1205 *type = cdirel->type;
1206
1207 switch (cdirel->key.type)
1208 {
1209 case GRUB_BTRFS_ITEM_TYPE_ROOT_ITEM:
1210 {
1211 struct grub_btrfs_root_item ri;
1212 err = lower_bound (data, &cdirel->key, &key_out,
1213 data->sblock.root_tree,
1214 &elemaddr, &elemsize, NULL);
1215 if (err)
1216 {
1217 grub_free (direl);
1218 grub_free (path_alloc);
1219 return err;
1220 }
1221 if (cdirel->key.object_id != key_out.object_id
1222 || cdirel->key.type != key_out.type)
1223 {
1224 grub_free (direl);
1225 grub_free (path_alloc);
1226 return grub_error (GRUB_ERR_FILE_NOT_FOUND, "file not found");
1227 }
1228 err = grub_btrfs_read_logical (data, elemaddr,
1229 &ri, sizeof (ri));
1230 if (err)
1231 {
1232 grub_free (direl);
1233 grub_free (path_alloc);
1234 return err;
1235 }
1236 key->type = GRUB_BTRFS_ITEM_TYPE_DIR_ITEM;
1237 key->offset = 0;
1238 key->object_id = GRUB_BTRFS_OBJECT_ID_CHUNK;
1239 *tree = grub_le_to_cpu64 (ri.tree);
1240 break;
1241 }
1242 case GRUB_BTRFS_ITEM_TYPE_INODE_ITEM:
1243 if (*slash && *type == GRUB_BTRFS_DIR_ITEM_TYPE_REGULAR)
1244 {
1245 grub_free (direl);
1246 grub_free (path_alloc);
1247 return grub_error (GRUB_ERR_FILE_NOT_FOUND, "file not found");
1248 }
1249 *key = cdirel->key;
1250 if (*type == GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY)
1251 key->type = GRUB_BTRFS_ITEM_TYPE_DIR_ITEM;
1252 break;
1253 default:
1254 grub_free (path_alloc);
1255 grub_free (direl);
1256 return grub_error (GRUB_ERR_BAD_FS, "unrecognised object type 0x%x",
1257 cdirel->key.type);
1258 }
1259 }
1260
1261 grub_free (direl);
1262 return GRUB_ERR_NONE;
1263 }
1264
1265 static grub_err_t
1266 grub_btrfs_dir (grub_device_t device, const char *path,
1267 int (*hook) (const char *filename,
1268 const struct grub_dirhook_info *info))
1269 {
1270 struct grub_btrfs_data *data = grub_btrfs_mount (device);
1271 struct grub_btrfs_key key_in, key_out;
1272 grub_err_t err;
1273 grub_disk_addr_t elemaddr;
1274 grub_size_t elemsize;
1275 grub_size_t allocated = 0;
1276 struct grub_btrfs_dir_item *direl = NULL;
1277 struct grub_btrfs_leaf_descriptor desc;
1278 int r = 0;
1279 grub_uint64_t tree;
1280 grub_uint8_t type;
1281
1282 if (!data)
1283 return grub_errno;
1284
1285 err = find_path (data, path, &key_in, &tree, &type);
1286 if (err)
1287 return err;
1288 if (type != GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY)
1289 return grub_error (GRUB_ERR_BAD_FILE_TYPE, "not a directory");
1290
1291 err = lower_bound (data, &key_in, &key_out, tree,
1292 &elemaddr, &elemsize, &desc);
1293 if (err)
1294 return err;
1295 if (key_out.type != GRUB_BTRFS_ITEM_TYPE_DIR_ITEM
1296 || key_out.object_id != key_in.object_id)
1297 {
1298 r = next (data, &desc, &elemaddr, &elemsize, &key_out);
1299 if (r <= 0)
1300 {
1301 free_iterator (&desc);
1302 return -r;
1303 }
1304 }
1305 do
1306 {
1307 struct grub_btrfs_dir_item *cdirel;
1308 if (key_out.type != GRUB_BTRFS_ITEM_TYPE_DIR_ITEM
1309 || key_out.object_id != key_in.object_id)
1310 {
1311 r = 0;
1312 break;
1313 }
1314 if (elemsize > allocated)
1315 {
1316 allocated = 2 * elemsize;
1317 grub_free (direl);
1318 direl = grub_malloc (allocated + 1);
1319 if (!direl)
1320 {
1321 free_iterator (&desc);
1322 return grub_errno;
1323 }
1324 }
1325
1326 err = grub_btrfs_read_logical (data, elemaddr, direl, elemsize);
1327 if (err)
1328 return err;
1329
1330 for (cdirel = direl;
1331 (grub_uint8_t *) cdirel - (grub_uint8_t *) direl
1332 < (grub_ssize_t) elemsize;
1333 cdirel = (void *) ((grub_uint8_t *) (direl + 1)
1334 + grub_le_to_cpu16 (cdirel->n)
1335 + grub_le_to_cpu16 (cdirel->m)))
1336 {
1337 char c;
1338 struct grub_btrfs_inode inode;
1339 struct grub_dirhook_info info;
1340 err = grub_btrfs_read_inode (data, &inode, cdirel->key.object_id,
1341 tree);
1342 grub_memset (&info, 0, sizeof (info));
1343 if (err)
1344 grub_errno = GRUB_ERR_NONE;
1345 else
1346 {
1347 info.mtime = inode.mtime.sec;
1348 info.mtimeset = 1;
1349 }
1350 c = cdirel->name[grub_le_to_cpu16 (cdirel->n)];
1351 cdirel->name[grub_le_to_cpu16 (cdirel->n)] = 0;
1352 info.dir = (cdirel->type == GRUB_BTRFS_DIR_ITEM_TYPE_DIRECTORY);
1353 if (hook (cdirel->name, &info))
1354 goto out;
1355 cdirel->name[grub_le_to_cpu16 (cdirel->n)] = c;
1356 }
1357 r = next (data, &desc, &elemaddr, &elemsize, &key_out);
1358 }
1359 while (r > 0);
1360
1361 out:
1362 grub_free (direl);
1363
1364 free_iterator (&desc);
1365 grub_btrfs_unmount (data);
1366
1367 return -r;
1368 }
1369
1370 static grub_err_t
1371 grub_btrfs_open (struct grub_file *file, const char *name)
1372 {
1373 struct grub_btrfs_data *data = grub_btrfs_mount (file->device);
1374 grub_err_t err;
1375 struct grub_btrfs_inode inode;
1376 grub_uint8_t type;
1377 struct grub_btrfs_key key_in;
1378
1379 if (!data)
1380 return grub_errno;
1381
1382 err = find_path (data, name, &key_in, &data->tree, &type);
1383 if (err)
1384 {
1385 grub_btrfs_unmount (data);
1386 return err;
1387 }
1388 if (type != GRUB_BTRFS_DIR_ITEM_TYPE_REGULAR)
1389 {
1390 grub_btrfs_unmount (data);
1391 return grub_error (GRUB_ERR_BAD_FILE_TYPE, "not a regular file");
1392 }
1393
1394 data->inode = key_in.object_id;
1395 err = grub_btrfs_read_inode (data, &inode, data->inode, data->tree);
1396 if (err)
1397 {
1398 grub_btrfs_unmount (data);
1399 return err;
1400 }
1401
1402 file->data = data;
1403 file->size = grub_le_to_cpu64 (inode.size);
1404
1405 return err;
1406 }
1407
1408 static grub_err_t
1409 grub_btrfs_close (grub_file_t file)
1410 {
1411 grub_btrfs_unmount (file->data);
1412
1413 return GRUB_ERR_NONE;
1414 }
1415
1416 static grub_ssize_t
1417 grub_btrfs_read (grub_file_t file, char *buf, grub_size_t len)
1418 {
1419 struct grub_btrfs_data *data = file->data;
1420
1421 return grub_btrfs_extent_read (data, data->inode,
1422 data->tree, file->offset, buf, len);
1423 }
1424
1425 static grub_err_t
1426 grub_btrfs_uuid (grub_device_t device, char **uuid)
1427 {
1428 struct grub_btrfs_data *data;
1429
1430 *uuid = NULL;
1431
1432 data = grub_btrfs_mount (device);
1433 if (! data)
1434 return grub_errno;
1435
1436 *uuid = grub_xasprintf ("%04x%04x-%04x-%04x-%04x-%04x%04x%04x",
1437 grub_be_to_cpu16 (data->sblock.uuid[0]),
1438 grub_be_to_cpu16 (data->sblock.uuid[1]),
1439 grub_be_to_cpu16 (data->sblock.uuid[2]),
1440 grub_be_to_cpu16 (data->sblock.uuid[3]),
1441 grub_be_to_cpu16 (data->sblock.uuid[4]),
1442 grub_be_to_cpu16 (data->sblock.uuid[5]),
1443 grub_be_to_cpu16 (data->sblock.uuid[6]),
1444 grub_be_to_cpu16 (data->sblock.uuid[7]));
1445
1446 grub_btrfs_unmount (data);
1447
1448 return grub_errno;
1449 }
1450
1451 static grub_err_t
1452 grub_btrfs_label (grub_device_t device, char **label)
1453 {
1454 struct grub_btrfs_data *data;
1455
1456 *label = NULL;
1457
1458 data = grub_btrfs_mount (device);
1459 if (! data)
1460 return grub_errno;
1461
1462 *label = grub_strndup (data->sblock.label, sizeof (data->sblock.label));
1463
1464 grub_btrfs_unmount (data);
1465
1466 return grub_errno;
1467 }
1468
1469 static struct grub_fs grub_btrfs_fs =
1470 {
1471 .name = "btrfs",
1472 .dir = grub_btrfs_dir,
1473 .open = grub_btrfs_open,
1474 .read = grub_btrfs_read,
1475 .close = grub_btrfs_close,
1476 .uuid = grub_btrfs_uuid,
1477 .label = grub_btrfs_label,
1478 #ifdef GRUB_UTIL
1479 .reserved_first_sector = 1,
1480 #endif
1481 };
1482
1483 GRUB_MOD_INIT(btrfs)
1484 {
1485 grub_fs_register (&grub_btrfs_fs);
1486 }
1487
1488 GRUB_MOD_FINI(btrfs)
1489 {
1490 grub_fs_unregister (&grub_btrfs_fs);
1491 }
The GNU GRand Unified Bootloader for Proxmox projects