1
2
3
4
5
6
7
8
9
10
11
12#include <linux/slab.h>
13#include <linux/pagemap.h>
14#include <linux/stringify.h>
15#include <linux/kernel.h>
16#include <linux/uuid.h>
17#include <linux/msdos_partition.h>
18
19#include "ldm.h"
20#include "check.h"
21
22
23
24
25
26
27
28
29
30#ifndef CONFIG_LDM_DEBUG
31#define ldm_debug(...) do {} while (0)
32#else
33#define ldm_debug(f, a...) _ldm_printk (KERN_DEBUG, __func__, f, ##a)
34#endif
35
36#define ldm_crit(f, a...) _ldm_printk (KERN_CRIT, __func__, f, ##a)
37#define ldm_error(f, a...) _ldm_printk (KERN_ERR, __func__, f, ##a)
38#define ldm_info(f, a...) _ldm_printk (KERN_INFO, __func__, f, ##a)
39
40static __printf(3, 4)
41void _ldm_printk(const char *level, const char *function, const char *fmt, ...)
42{
43 struct va_format vaf;
44 va_list args;
45
46 va_start (args, fmt);
47
48 vaf.fmt = fmt;
49 vaf.va = &args;
50
51 printk("%s%s(): %pV\n", level, function, &vaf);
52
53 va_end(args);
54}
55
56
57
58
59
60
61
62
63
64
65
66
67static bool ldm_parse_privhead(const u8 *data, struct privhead *ph)
68{
69 bool is_vista = false;
70
71 BUG_ON(!data || !ph);
72 if (MAGIC_PRIVHEAD != get_unaligned_be64(data)) {
73 ldm_error("Cannot find PRIVHEAD structure. LDM database is"
74 " corrupt. Aborting.");
75 return false;
76 }
77 ph->ver_major = get_unaligned_be16(data + 0x000C);
78 ph->ver_minor = get_unaligned_be16(data + 0x000E);
79 ph->logical_disk_start = get_unaligned_be64(data + 0x011B);
80 ph->logical_disk_size = get_unaligned_be64(data + 0x0123);
81 ph->config_start = get_unaligned_be64(data + 0x012B);
82 ph->config_size = get_unaligned_be64(data + 0x0133);
83
84 if (ph->ver_major == 2 && ph->ver_minor == 12)
85 is_vista = true;
86 if (!is_vista && (ph->ver_major != 2 || ph->ver_minor != 11)) {
87 ldm_error("Expected PRIVHEAD version 2.11 or 2.12, got %d.%d."
88 " Aborting.", ph->ver_major, ph->ver_minor);
89 return false;
90 }
91 ldm_debug("PRIVHEAD version %d.%d (Windows %s).", ph->ver_major,
92 ph->ver_minor, is_vista ? "Vista" : "2000/XP");
93 if (ph->config_size != LDM_DB_SIZE) {
94
95 ldm_info("Database is normally %u bytes, it claims to "
96 "be %llu bytes.", LDM_DB_SIZE,
97 (unsigned long long)ph->config_size);
98 }
99 if ((ph->logical_disk_size == 0) || (ph->logical_disk_start +
100 ph->logical_disk_size > ph->config_start)) {
101 ldm_error("PRIVHEAD disk size doesn't match real disk size");
102 return false;
103 }
104 if (uuid_parse(data + 0x0030, &ph->disk_id)) {
105 ldm_error("PRIVHEAD contains an invalid GUID.");
106 return false;
107 }
108 ldm_debug("Parsed PRIVHEAD successfully.");
109 return true;
110}
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126static bool ldm_parse_tocblock (const u8 *data, struct tocblock *toc)
127{
128 BUG_ON (!data || !toc);
129
130 if (MAGIC_TOCBLOCK != get_unaligned_be64(data)) {
131 ldm_crit ("Cannot find TOCBLOCK, database may be corrupt.");
132 return false;
133 }
134 strncpy (toc->bitmap1_name, data + 0x24, sizeof (toc->bitmap1_name));
135 toc->bitmap1_name[sizeof (toc->bitmap1_name) - 1] = 0;
136 toc->bitmap1_start = get_unaligned_be64(data + 0x2E);
137 toc->bitmap1_size = get_unaligned_be64(data + 0x36);
138
139 if (strncmp (toc->bitmap1_name, TOC_BITMAP1,
140 sizeof (toc->bitmap1_name)) != 0) {
141 ldm_crit ("TOCBLOCK's first bitmap is '%s', should be '%s'.",
142 TOC_BITMAP1, toc->bitmap1_name);
143 return false;
144 }
145 strncpy (toc->bitmap2_name, data + 0x46, sizeof (toc->bitmap2_name));
146 toc->bitmap2_name[sizeof (toc->bitmap2_name) - 1] = 0;
147 toc->bitmap2_start = get_unaligned_be64(data + 0x50);
148 toc->bitmap2_size = get_unaligned_be64(data + 0x58);
149 if (strncmp (toc->bitmap2_name, TOC_BITMAP2,
150 sizeof (toc->bitmap2_name)) != 0) {
151 ldm_crit ("TOCBLOCK's second bitmap is '%s', should be '%s'.",
152 TOC_BITMAP2, toc->bitmap2_name);
153 return false;
154 }
155 ldm_debug ("Parsed TOCBLOCK successfully.");
156 return true;
157}
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172static bool ldm_parse_vmdb (const u8 *data, struct vmdb *vm)
173{
174 BUG_ON (!data || !vm);
175
176 if (MAGIC_VMDB != get_unaligned_be32(data)) {
177 ldm_crit ("Cannot find the VMDB, database may be corrupt.");
178 return false;
179 }
180
181 vm->ver_major = get_unaligned_be16(data + 0x12);
182 vm->ver_minor = get_unaligned_be16(data + 0x14);
183 if ((vm->ver_major != 4) || (vm->ver_minor != 10)) {
184 ldm_error ("Expected VMDB version %d.%d, got %d.%d. "
185 "Aborting.", 4, 10, vm->ver_major, vm->ver_minor);
186 return false;
187 }
188
189 vm->vblk_size = get_unaligned_be32(data + 0x08);
190 if (vm->vblk_size == 0) {
191 ldm_error ("Illegal VBLK size");
192 return false;
193 }
194
195 vm->vblk_offset = get_unaligned_be32(data + 0x0C);
196 vm->last_vblk_seq = get_unaligned_be32(data + 0x04);
197
198 ldm_debug ("Parsed VMDB successfully.");
199 return true;
200}
201
202
203
204
205
206
207
208
209
210
211
212static bool ldm_compare_privheads (const struct privhead *ph1,
213 const struct privhead *ph2)
214{
215 BUG_ON (!ph1 || !ph2);
216
217 return ((ph1->ver_major == ph2->ver_major) &&
218 (ph1->ver_minor == ph2->ver_minor) &&
219 (ph1->logical_disk_start == ph2->logical_disk_start) &&
220 (ph1->logical_disk_size == ph2->logical_disk_size) &&
221 (ph1->config_start == ph2->config_start) &&
222 (ph1->config_size == ph2->config_size) &&
223 uuid_equal(&ph1->disk_id, &ph2->disk_id));
224}
225
226
227
228
229
230
231
232
233
234
235
236static bool ldm_compare_tocblocks (const struct tocblock *toc1,
237 const struct tocblock *toc2)
238{
239 BUG_ON (!toc1 || !toc2);
240
241 return ((toc1->bitmap1_start == toc2->bitmap1_start) &&
242 (toc1->bitmap1_size == toc2->bitmap1_size) &&
243 (toc1->bitmap2_start == toc2->bitmap2_start) &&
244 (toc1->bitmap2_size == toc2->bitmap2_size) &&
245 !strncmp (toc1->bitmap1_name, toc2->bitmap1_name,
246 sizeof (toc1->bitmap1_name)) &&
247 !strncmp (toc1->bitmap2_name, toc2->bitmap2_name,
248 sizeof (toc1->bitmap2_name)));
249}
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265static bool ldm_validate_privheads(struct parsed_partitions *state,
266 struct privhead *ph1)
267{
268 static const int off[3] = { OFF_PRIV1, OFF_PRIV2, OFF_PRIV3 };
269 struct privhead *ph[3] = { ph1 };
270 Sector sect;
271 u8 *data;
272 bool result = false;
273 long num_sects;
274 int i;
275
276 BUG_ON (!state || !ph1);
277
278 ph[1] = kmalloc (sizeof (*ph[1]), GFP_KERNEL);
279 ph[2] = kmalloc (sizeof (*ph[2]), GFP_KERNEL);
280 if (!ph[1] || !ph[2]) {
281 ldm_crit ("Out of memory.");
282 goto out;
283 }
284
285
286 ph[0]->config_start = 0;
287
288
289 for (i = 0; i < 3; i++) {
290 data = read_part_sector(state, ph[0]->config_start + off[i],
291 §);
292 if (!data) {
293 ldm_crit ("Disk read failed.");
294 goto out;
295 }
296 result = ldm_parse_privhead (data, ph[i]);
297 put_dev_sector (sect);
298 if (!result) {
299 ldm_error ("Cannot find PRIVHEAD %d.", i+1);
300 if (i < 2)
301 goto out;
302 else
303 break;
304 }
305 }
306
307 num_sects = get_capacity(state->disk);
308
309 if ((ph[0]->config_start > num_sects) ||
310 ((ph[0]->config_start + ph[0]->config_size) > num_sects)) {
311 ldm_crit ("Database extends beyond the end of the disk.");
312 goto out;
313 }
314
315 if ((ph[0]->logical_disk_start > ph[0]->config_start) ||
316 ((ph[0]->logical_disk_start + ph[0]->logical_disk_size)
317 > ph[0]->config_start)) {
318 ldm_crit ("Disk and database overlap.");
319 goto out;
320 }
321
322 if (!ldm_compare_privheads (ph[0], ph[1])) {
323 ldm_crit ("Primary and backup PRIVHEADs don't match.");
324 goto out;
325 }
326
327
328
329
330
331 ldm_debug ("Validated PRIVHEADs successfully.");
332 result = true;
333out:
334 kfree (ph[1]);
335 kfree (ph[2]);
336 return result;
337}
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353static bool ldm_validate_tocblocks(struct parsed_partitions *state,
354 unsigned long base, struct ldmdb *ldb)
355{
356 static const int off[4] = { OFF_TOCB1, OFF_TOCB2, OFF_TOCB3, OFF_TOCB4};
357 struct tocblock *tb[4];
358 struct privhead *ph;
359 Sector sect;
360 u8 *data;
361 int i, nr_tbs;
362 bool result = false;
363
364 BUG_ON(!state || !ldb);
365 ph = &ldb->ph;
366 tb[0] = &ldb->toc;
367 tb[1] = kmalloc_array(3, sizeof(*tb[1]), GFP_KERNEL);
368 if (!tb[1]) {
369 ldm_crit("Out of memory.");
370 goto err;
371 }
372 tb[2] = (struct tocblock*)((u8*)tb[1] + sizeof(*tb[1]));
373 tb[3] = (struct tocblock*)((u8*)tb[2] + sizeof(*tb[2]));
374
375
376
377
378
379
380 for (nr_tbs = i = 0; i < 4; i++) {
381 data = read_part_sector(state, base + off[i], §);
382 if (!data) {
383 ldm_error("Disk read failed for TOCBLOCK %d.", i);
384 continue;
385 }
386 if (ldm_parse_tocblock(data, tb[nr_tbs]))
387 nr_tbs++;
388 put_dev_sector(sect);
389 }
390 if (!nr_tbs) {
391 ldm_crit("Failed to find a valid TOCBLOCK.");
392 goto err;
393 }
394
395 if (((tb[0]->bitmap1_start + tb[0]->bitmap1_size) > ph->config_size) ||
396 ((tb[0]->bitmap2_start + tb[0]->bitmap2_size) >
397 ph->config_size)) {
398 ldm_crit("The bitmaps are out of range. Giving up.");
399 goto err;
400 }
401
402 for (i = 1; i < nr_tbs; i++) {
403 if (!ldm_compare_tocblocks(tb[0], tb[i])) {
404 ldm_crit("TOCBLOCKs 0 and %d do not match.", i);
405 goto err;
406 }
407 }
408 ldm_debug("Validated %d TOCBLOCKs successfully.", nr_tbs);
409 result = true;
410err:
411 kfree(tb[1]);
412 return result;
413}
414
415
416
417
418
419
420
421
422
423
424
425
426
427static bool ldm_validate_vmdb(struct parsed_partitions *state,
428 unsigned long base, struct ldmdb *ldb)
429{
430 Sector sect;
431 u8 *data;
432 bool result = false;
433 struct vmdb *vm;
434 struct tocblock *toc;
435
436 BUG_ON (!state || !ldb);
437
438 vm = &ldb->vm;
439 toc = &ldb->toc;
440
441 data = read_part_sector(state, base + OFF_VMDB, §);
442 if (!data) {
443 ldm_crit ("Disk read failed.");
444 return false;
445 }
446
447 if (!ldm_parse_vmdb (data, vm))
448 goto out;
449
450
451 if (get_unaligned_be16(data + 0x10) != 0x01) {
452 ldm_crit ("Database is not in a consistent state. Aborting.");
453 goto out;
454 }
455
456 if (vm->vblk_offset != 512)
457 ldm_info ("VBLKs start at offset 0x%04x.", vm->vblk_offset);
458
459
460
461
462
463 if ((vm->vblk_size * vm->last_vblk_seq) > (toc->bitmap1_size << 9)) {
464 ldm_crit ("VMDB exceeds allowed size specified by TOCBLOCK. "
465 "Database is corrupt. Aborting.");
466 goto out;
467 }
468
469 result = true;
470out:
471 put_dev_sector (sect);
472 return result;
473}
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492static bool ldm_validate_partition_table(struct parsed_partitions *state)
493{
494 Sector sect;
495 u8 *data;
496 struct msdos_partition *p;
497 int i;
498 bool result = false;
499
500 BUG_ON(!state);
501
502 data = read_part_sector(state, 0, §);
503 if (!data) {
504 ldm_info ("Disk read failed.");
505 return false;
506 }
507
508 if (*(__le16*) (data + 0x01FE) != cpu_to_le16 (MSDOS_LABEL_MAGIC))
509 goto out;
510
511 p = (struct msdos_partition *)(data + 0x01BE);
512 for (i = 0; i < 4; i++, p++)
513 if (p->sys_ind == LDM_PARTITION) {
514 result = true;
515 break;
516 }
517
518 if (result)
519 ldm_debug ("Found W2K dynamic disk partition type.");
520
521out:
522 put_dev_sector (sect);
523 return result;
524}
525
526
527
528
529
530
531
532
533
534
535
536
537
538static struct vblk * ldm_get_disk_objid (const struct ldmdb *ldb)
539{
540 struct list_head *item;
541
542 BUG_ON (!ldb);
543
544 list_for_each (item, &ldb->v_disk) {
545 struct vblk *v = list_entry (item, struct vblk, list);
546 if (uuid_equal(&v->vblk.disk.disk_id, &ldb->ph.disk_id))
547 return v;
548 }
549
550 return NULL;
551}
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570static bool ldm_create_data_partitions (struct parsed_partitions *pp,
571 const struct ldmdb *ldb)
572{
573 struct list_head *item;
574 struct vblk *vb;
575 struct vblk *disk;
576 struct vblk_part *part;
577 int part_num = 1;
578
579 BUG_ON (!pp || !ldb);
580
581 disk = ldm_get_disk_objid (ldb);
582 if (!disk) {
583 ldm_crit ("Can't find the ID of this disk in the database.");
584 return false;
585 }
586
587 strlcat(pp->pp_buf, " [LDM]", PAGE_SIZE);
588
589
590 list_for_each (item, &ldb->v_part) {
591 vb = list_entry (item, struct vblk, list);
592 part = &vb->vblk.part;
593
594 if (part->disk_id != disk->obj_id)
595 continue;
596
597 put_partition (pp, part_num, ldb->ph.logical_disk_start +
598 part->start, part->size);
599 part_num++;
600 }
601
602 strlcat(pp->pp_buf, "\n", PAGE_SIZE);
603 return true;
604}
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621static int ldm_relative(const u8 *buffer, int buflen, int base, int offset)
622{
623
624 base += offset;
625 if (!buffer || offset < 0 || base > buflen) {
626 if (!buffer)
627 ldm_error("!buffer");
628 if (offset < 0)
629 ldm_error("offset (%d) < 0", offset);
630 if (base > buflen)
631 ldm_error("base (%d) > buflen (%d)", base, buflen);
632 return -1;
633 }
634 if (base + buffer[base] >= buflen) {
635 ldm_error("base (%d) + buffer[base] (%d) >= buflen (%d)", base,
636 buffer[base], buflen);
637 return -1;
638 }
639 return buffer[base] + offset + 1;
640}
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657static u64 ldm_get_vnum (const u8 *block)
658{
659 u64 tmp = 0;
660 u8 length;
661
662 BUG_ON (!block);
663
664 length = *block++;
665
666 if (length && length <= 8)
667 while (length--)
668 tmp = (tmp << 8) | *block++;
669 else
670 ldm_error ("Illegal length %d.", length);
671
672 return tmp;
673}
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692static int ldm_get_vstr (const u8 *block, u8 *buffer, int buflen)
693{
694 int length;
695
696 BUG_ON (!block || !buffer);
697
698 length = block[0];
699 if (length >= buflen) {
700 ldm_error ("Truncating string %d -> %d.", length, buflen);
701 length = buflen - 1;
702 }
703 memcpy (buffer, block + 1, length);
704 buffer[length] = 0;
705 return length;
706}
707
708
709
710
711
712
713
714
715
716
717
718
719
720static bool ldm_parse_cmp3 (const u8 *buffer, int buflen, struct vblk *vb)
721{
722 int r_objid, r_name, r_vstate, r_child, r_parent, r_stripe, r_cols, len;
723 struct vblk_comp *comp;
724
725 BUG_ON (!buffer || !vb);
726
727 r_objid = ldm_relative (buffer, buflen, 0x18, 0);
728 r_name = ldm_relative (buffer, buflen, 0x18, r_objid);
729 r_vstate = ldm_relative (buffer, buflen, 0x18, r_name);
730 r_child = ldm_relative (buffer, buflen, 0x1D, r_vstate);
731 r_parent = ldm_relative (buffer, buflen, 0x2D, r_child);
732
733 if (buffer[0x12] & VBLK_FLAG_COMP_STRIPE) {
734 r_stripe = ldm_relative (buffer, buflen, 0x2E, r_parent);
735 r_cols = ldm_relative (buffer, buflen, 0x2E, r_stripe);
736 len = r_cols;
737 } else {
738 r_stripe = 0;
739 r_cols = 0;
740 len = r_parent;
741 }
742 if (len < 0)
743 return false;
744
745 len += VBLK_SIZE_CMP3;
746 if (len != get_unaligned_be32(buffer + 0x14))
747 return false;
748
749 comp = &vb->vblk.comp;
750 ldm_get_vstr (buffer + 0x18 + r_name, comp->state,
751 sizeof (comp->state));
752 comp->type = buffer[0x18 + r_vstate];
753 comp->children = ldm_get_vnum (buffer + 0x1D + r_vstate);
754 comp->parent_id = ldm_get_vnum (buffer + 0x2D + r_child);
755 comp->chunksize = r_stripe ? ldm_get_vnum (buffer+r_parent+0x2E) : 0;
756
757 return true;
758}
759
760
761
762
763
764
765
766
767
768
769
770
771static int ldm_parse_dgr3 (const u8 *buffer, int buflen, struct vblk *vb)
772{
773 int r_objid, r_name, r_diskid, r_id1, r_id2, len;
774 struct vblk_dgrp *dgrp;
775
776 BUG_ON (!buffer || !vb);
777
778 r_objid = ldm_relative (buffer, buflen, 0x18, 0);
779 r_name = ldm_relative (buffer, buflen, 0x18, r_objid);
780 r_diskid = ldm_relative (buffer, buflen, 0x18, r_name);
781
782 if (buffer[0x12] & VBLK_FLAG_DGR3_IDS) {
783 r_id1 = ldm_relative (buffer, buflen, 0x24, r_diskid);
784 r_id2 = ldm_relative (buffer, buflen, 0x24, r_id1);
785 len = r_id2;
786 } else {
787 r_id1 = 0;
788 r_id2 = 0;
789 len = r_diskid;
790 }
791 if (len < 0)
792 return false;
793
794 len += VBLK_SIZE_DGR3;
795 if (len != get_unaligned_be32(buffer + 0x14))
796 return false;
797
798 dgrp = &vb->vblk.dgrp;
799 ldm_get_vstr (buffer + 0x18 + r_name, dgrp->disk_id,
800 sizeof (dgrp->disk_id));
801 return true;
802}
803
804
805
806
807
808
809
810
811
812
813
814
815static bool ldm_parse_dgr4 (const u8 *buffer, int buflen, struct vblk *vb)
816{
817 char buf[64];
818 int r_objid, r_name, r_id1, r_id2, len;
819
820 BUG_ON (!buffer || !vb);
821
822 r_objid = ldm_relative (buffer, buflen, 0x18, 0);
823 r_name = ldm_relative (buffer, buflen, 0x18, r_objid);
824
825 if (buffer[0x12] & VBLK_FLAG_DGR4_IDS) {
826 r_id1 = ldm_relative (buffer, buflen, 0x44, r_name);
827 r_id2 = ldm_relative (buffer, buflen, 0x44, r_id1);
828 len = r_id2;
829 } else {
830 r_id1 = 0;
831 r_id2 = 0;
832 len = r_name;
833 }
834 if (len < 0)
835 return false;
836
837 len += VBLK_SIZE_DGR4;
838 if (len != get_unaligned_be32(buffer + 0x14))
839 return false;
840
841 ldm_get_vstr (buffer + 0x18 + r_objid, buf, sizeof (buf));
842 return true;
843}
844
845
846
847
848
849
850
851
852
853
854
855
856static bool ldm_parse_dsk3 (const u8 *buffer, int buflen, struct vblk *vb)
857{
858 int r_objid, r_name, r_diskid, r_altname, len;
859 struct vblk_disk *disk;
860
861 BUG_ON (!buffer || !vb);
862
863 r_objid = ldm_relative (buffer, buflen, 0x18, 0);
864 r_name = ldm_relative (buffer, buflen, 0x18, r_objid);
865 r_diskid = ldm_relative (buffer, buflen, 0x18, r_name);
866 r_altname = ldm_relative (buffer, buflen, 0x18, r_diskid);
867 len = r_altname;
868 if (len < 0)
869 return false;
870
871 len += VBLK_SIZE_DSK3;
872 if (len != get_unaligned_be32(buffer + 0x14))
873 return false;
874
875 disk = &vb->vblk.disk;
876 ldm_get_vstr (buffer + 0x18 + r_diskid, disk->alt_name,
877 sizeof (disk->alt_name));
878 if (uuid_parse(buffer + 0x19 + r_name, &disk->disk_id))
879 return false;
880
881 return true;
882}
883
884
885
886
887
888
889
890
891
892
893
894
895static bool ldm_parse_dsk4 (const u8 *buffer, int buflen, struct vblk *vb)
896{
897 int r_objid, r_name, len;
898 struct vblk_disk *disk;
899
900 BUG_ON (!buffer || !vb);
901
902 r_objid = ldm_relative (buffer, buflen, 0x18, 0);
903 r_name = ldm_relative (buffer, buflen, 0x18, r_objid);
904 len = r_name;
905 if (len < 0)
906 return false;
907
908 len += VBLK_SIZE_DSK4;
909 if (len != get_unaligned_be32(buffer + 0x14))
910 return false;
911
912 disk = &vb->vblk.disk;
913 import_uuid(&disk->disk_id, buffer + 0x18 + r_name);
914 return true;
915}
916
917
918
919
920
921
922
923
924
925
926
927
928static bool ldm_parse_prt3(const u8 *buffer, int buflen, struct vblk *vb)
929{
930 int r_objid, r_name, r_size, r_parent, r_diskid, r_index, len;
931 struct vblk_part *part;
932
933 BUG_ON(!buffer || !vb);
934 r_objid = ldm_relative(buffer, buflen, 0x18, 0);
935 if (r_objid < 0) {
936 ldm_error("r_objid %d < 0", r_objid);
937 return false;
938 }
939 r_name = ldm_relative(buffer, buflen, 0x18, r_objid);
940 if (r_name < 0) {
941 ldm_error("r_name %d < 0", r_name);
942 return false;
943 }
944 r_size = ldm_relative(buffer, buflen, 0x34, r_name);
945 if (r_size < 0) {
946 ldm_error("r_size %d < 0", r_size);
947 return false;
948 }
949 r_parent = ldm_relative(buffer, buflen, 0x34, r_size);
950 if (r_parent < 0) {
951 ldm_error("r_parent %d < 0", r_parent);
952 return false;
953 }
954 r_diskid = ldm_relative(buffer, buflen, 0x34, r_parent);
955 if (r_diskid < 0) {
956 ldm_error("r_diskid %d < 0", r_diskid);
957 return false;
958 }
959 if (buffer[0x12] & VBLK_FLAG_PART_INDEX) {
960 r_index = ldm_relative(buffer, buflen, 0x34, r_diskid);
961 if (r_index < 0) {
962 ldm_error("r_index %d < 0", r_index);
963 return false;
964 }
965 len = r_index;
966 } else {
967 r_index = 0;
968 len = r_diskid;
969 }
970 if (len < 0) {
971 ldm_error("len %d < 0", len);
972 return false;
973 }
974 len += VBLK_SIZE_PRT3;
975 if (len > get_unaligned_be32(buffer + 0x14)) {
976 ldm_error("len %d > BE32(buffer + 0x14) %d", len,
977 get_unaligned_be32(buffer + 0x14));
978 return false;
979 }
980 part = &vb->vblk.part;
981 part->start = get_unaligned_be64(buffer + 0x24 + r_name);
982 part->volume_offset = get_unaligned_be64(buffer + 0x2C + r_name);
983 part->size = ldm_get_vnum(buffer + 0x34 + r_name);
984 part->parent_id = ldm_get_vnum(buffer + 0x34 + r_size);
985 part->disk_id = ldm_get_vnum(buffer + 0x34 + r_parent);
986 if (vb->flags & VBLK_FLAG_PART_INDEX)
987 part->partnum = buffer[0x35 + r_diskid];
988 else
989 part->partnum = 0;
990 return true;
991}
992
993
994
995
996
997
998
999
1000
1001
1002
1003
1004static bool ldm_parse_vol5(const u8 *buffer, int buflen, struct vblk *vb)
1005{
1006 int r_objid, r_name, r_vtype, r_disable_drive_letter, r_child, r_size;
1007 int r_id1, r_id2, r_size2, r_drive, len;
1008 struct vblk_volu *volu;
1009
1010 BUG_ON(!buffer || !vb);
1011 r_objid = ldm_relative(buffer, buflen, 0x18, 0);
1012 if (r_objid < 0) {
1013 ldm_error("r_objid %d < 0", r_objid);
1014 return false;
1015 }
1016 r_name = ldm_relative(buffer, buflen, 0x18, r_objid);
1017 if (r_name < 0) {
1018 ldm_error("r_name %d < 0", r_name);
1019 return false;
1020 }
1021 r_vtype = ldm_relative(buffer, buflen, 0x18, r_name);
1022 if (r_vtype < 0) {
1023 ldm_error("r_vtype %d < 0", r_vtype);
1024 return false;
1025 }
1026 r_disable_drive_letter = ldm_relative(buffer, buflen, 0x18, r_vtype);
1027 if (r_disable_drive_letter < 0) {
1028 ldm_error("r_disable_drive_letter %d < 0",
1029 r_disable_drive_letter);
1030 return false;
1031 }
1032 r_child = ldm_relative(buffer, buflen, 0x2D, r_disable_drive_letter);
1033 if (r_child < 0) {
1034 ldm_error("r_child %d < 0", r_child);
1035 return false;
1036 }
1037 r_size = ldm_relative(buffer, buflen, 0x3D, r_child);
1038 if (r_size < 0) {
1039 ldm_error("r_size %d < 0", r_size);
1040 return false;
1041 }
1042 if (buffer[0x12] & VBLK_FLAG_VOLU_ID1) {
1043 r_id1 = ldm_relative(buffer, buflen, 0x52, r_size);
1044 if (r_id1 < 0) {
1045 ldm_error("r_id1 %d < 0", r_id1);
1046 return false;
1047 }
1048 } else
1049 r_id1 = r_size;
1050 if (buffer[0x12] & VBLK_FLAG_VOLU_ID2) {
1051 r_id2 = ldm_relative(buffer, buflen, 0x52, r_id1);
1052 if (r_id2 < 0) {
1053 ldm_error("r_id2 %d < 0", r_id2);
1054 return false;
1055 }
1056 } else
1057 r_id2 = r_id1;
1058 if (buffer[0x12] & VBLK_FLAG_VOLU_SIZE) {
1059 r_size2 = ldm_relative(buffer, buflen, 0x52, r_id2);
1060 if (r_size2 < 0) {
1061 ldm_error("r_size2 %d < 0", r_size2);
1062 return false;
1063 }
1064 } else
1065 r_size2 = r_id2;
1066 if (buffer[0x12] & VBLK_FLAG_VOLU_DRIVE) {
1067 r_drive = ldm_relative(buffer, buflen, 0x52, r_size2);
1068 if (r_drive < 0) {
1069 ldm_error("r_drive %d < 0", r_drive);
1070 return false;
1071 }
1072 } else
1073 r_drive = r_size2;
1074 len = r_drive;
1075 if (len < 0) {
1076 ldm_error("len %d < 0", len);
1077 return false;
1078 }
1079 len += VBLK_SIZE_VOL5;
1080 if (len > get_unaligned_be32(buffer + 0x14)) {
1081 ldm_error("len %d > BE32(buffer + 0x14) %d", len,
1082 get_unaligned_be32(buffer + 0x14));
1083 return false;
1084 }
1085 volu = &vb->vblk.volu;
1086 ldm_get_vstr(buffer + 0x18 + r_name, volu->volume_type,
1087 sizeof(volu->volume_type));
1088 memcpy(volu->volume_state, buffer + 0x18 + r_disable_drive_letter,
1089 sizeof(volu->volume_state));
1090 volu->size = ldm_get_vnum(buffer + 0x3D + r_child);
1091 volu->partition_type = buffer[0x41 + r_size];
1092 memcpy(volu->guid, buffer + 0x42 + r_size, sizeof(volu->guid));
1093 if (buffer[0x12] & VBLK_FLAG_VOLU_DRIVE) {
1094 ldm_get_vstr(buffer + 0x52 + r_size, volu->drive_hint,
1095 sizeof(volu->drive_hint));
1096 }
1097 return true;
1098}
1099
1100
1101
1102
1103
1104
1105
1106
1107
1108
1109
1110
1111
1112
1113static bool ldm_parse_vblk (const u8 *buf, int len, struct vblk *vb)
1114{
1115 bool result = false;
1116 int r_objid;
1117
1118 BUG_ON (!buf || !vb);
1119
1120 r_objid = ldm_relative (buf, len, 0x18, 0);
1121 if (r_objid < 0) {
1122 ldm_error ("VBLK header is corrupt.");
1123 return false;
1124 }
1125
1126 vb->flags = buf[0x12];
1127 vb->type = buf[0x13];
1128 vb->obj_id = ldm_get_vnum (buf + 0x18);
1129 ldm_get_vstr (buf+0x18+r_objid, vb->name, sizeof (vb->name));
1130
1131 switch (vb->type) {
1132 case VBLK_CMP3: result = ldm_parse_cmp3 (buf, len, vb); break;
1133 case VBLK_DSK3: result = ldm_parse_dsk3 (buf, len, vb); break;
1134 case VBLK_DSK4: result = ldm_parse_dsk4 (buf, len, vb); break;
1135 case VBLK_DGR3: result = ldm_parse_dgr3 (buf, len, vb); break;
1136 case VBLK_DGR4: result = ldm_parse_dgr4 (buf, len, vb); break;
1137 case VBLK_PRT3: result = ldm_parse_prt3 (buf, len, vb); break;
1138 case VBLK_VOL5: result = ldm_parse_vol5 (buf, len, vb); break;
1139 }
1140
1141 if (result)
1142 ldm_debug ("Parsed VBLK 0x%llx (type: 0x%02x) ok.",
1143 (unsigned long long) vb->obj_id, vb->type);
1144 else
1145 ldm_error ("Failed to parse VBLK 0x%llx (type: 0x%02x).",
1146 (unsigned long long) vb->obj_id, vb->type);
1147
1148 return result;
1149}
1150
1151
1152
1153
1154
1155
1156
1157
1158
1159
1160
1161
1162
1163
1164
1165static bool ldm_ldmdb_add (u8 *data, int len, struct ldmdb *ldb)
1166{
1167 struct vblk *vb;
1168 struct list_head *item;
1169
1170 BUG_ON (!data || !ldb);
1171
1172 vb = kmalloc (sizeof (*vb), GFP_KERNEL);
1173 if (!vb) {
1174 ldm_crit ("Out of memory.");
1175 return false;
1176 }
1177
1178 if (!ldm_parse_vblk (data, len, vb)) {
1179 kfree(vb);
1180 return false;
1181 }
1182
1183
1184 switch (vb->type) {
1185 case VBLK_DGR3:
1186 case VBLK_DGR4:
1187 list_add (&vb->list, &ldb->v_dgrp);
1188 break;
1189 case VBLK_DSK3:
1190 case VBLK_DSK4:
1191 list_add (&vb->list, &ldb->v_disk);
1192 break;
1193 case VBLK_VOL5:
1194 list_add (&vb->list, &ldb->v_volu);
1195 break;
1196 case VBLK_CMP3:
1197 list_add (&vb->list, &ldb->v_comp);
1198 break;
1199 case VBLK_PRT3:
1200
1201 list_for_each (item, &ldb->v_part) {
1202 struct vblk *v = list_entry (item, struct vblk, list);
1203 if ((v->vblk.part.disk_id == vb->vblk.part.disk_id) &&
1204 (v->vblk.part.start > vb->vblk.part.start)) {
1205 list_add_tail (&vb->list, &v->list);
1206 return true;
1207 }
1208 }
1209 list_add_tail (&vb->list, &ldb->v_part);
1210 break;
1211 }
1212 return true;
1213}
1214
1215
1216
1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227static bool ldm_frag_add (const u8 *data, int size, struct list_head *frags)
1228{
1229 struct frag *f;
1230 struct list_head *item;
1231 int rec, num, group;
1232
1233 BUG_ON (!data || !frags);
1234
1235 if (size < 2 * VBLK_SIZE_HEAD) {
1236 ldm_error("Value of size is too small.");
1237 return false;
1238 }
1239
1240 group = get_unaligned_be32(data + 0x08);
1241 rec = get_unaligned_be16(data + 0x0C);
1242 num = get_unaligned_be16(data + 0x0E);
1243 if ((num < 1) || (num > 4)) {
1244 ldm_error ("A VBLK claims to have %d parts.", num);
1245 return false;
1246 }
1247 if (rec >= num) {
1248 ldm_error("REC value (%d) exceeds NUM value (%d)", rec, num);
1249 return false;
1250 }
1251
1252 list_for_each (item, frags) {
1253 f = list_entry (item, struct frag, list);
1254 if (f->group == group)
1255 goto found;
1256 }
1257
1258 f = kmalloc (sizeof (*f) + size*num, GFP_KERNEL);
1259 if (!f) {
1260 ldm_crit ("Out of memory.");
1261 return false;
1262 }
1263
1264 f->group = group;
1265 f->num = num;
1266 f->rec = rec;
1267 f->map = 0xFF << num;
1268
1269 list_add_tail (&f->list, frags);
1270found:
1271 if (rec >= f->num) {
1272 ldm_error("REC value (%d) exceeds NUM value (%d)", rec, f->num);
1273 return false;
1274 }
1275 if (f->map & (1 << rec)) {
1276 ldm_error ("Duplicate VBLK, part %d.", rec);
1277 f->map &= 0x7F;
1278 return false;
1279 }
1280 f->map |= (1 << rec);
1281 if (!rec)
1282 memcpy(f->data, data, VBLK_SIZE_HEAD);
1283 data += VBLK_SIZE_HEAD;
1284 size -= VBLK_SIZE_HEAD;
1285 memcpy(f->data + VBLK_SIZE_HEAD + rec * size, data, size);
1286 return true;
1287}
1288
1289
1290
1291
1292
1293
1294
1295
1296
1297static void ldm_frag_free (struct list_head *list)
1298{
1299 struct list_head *item, *tmp;
1300
1301 BUG_ON (!list);
1302
1303 list_for_each_safe (item, tmp, list)
1304 kfree (list_entry (item, struct frag, list));
1305}
1306
1307
1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318static bool ldm_frag_commit (struct list_head *frags, struct ldmdb *ldb)
1319{
1320 struct frag *f;
1321 struct list_head *item;
1322
1323 BUG_ON (!frags || !ldb);
1324
1325 list_for_each (item, frags) {
1326 f = list_entry (item, struct frag, list);
1327
1328 if (f->map != 0xFF) {
1329 ldm_error ("VBLK group %d is incomplete (0x%02x).",
1330 f->group, f->map);
1331 return false;
1332 }
1333
1334 if (!ldm_ldmdb_add (f->data, f->num*ldb->vm.vblk_size, ldb))
1335 return false;
1336 }
1337 return true;
1338}
1339
1340
1341
1342
1343
1344
1345
1346
1347
1348
1349
1350
1351
1352static bool ldm_get_vblks(struct parsed_partitions *state, unsigned long base,
1353 struct ldmdb *ldb)
1354{
1355 int size, perbuf, skip, finish, s, v, recs;
1356 u8 *data = NULL;
1357 Sector sect;
1358 bool result = false;
1359 LIST_HEAD (frags);
1360
1361 BUG_ON(!state || !ldb);
1362
1363 size = ldb->vm.vblk_size;
1364 perbuf = 512 / size;
1365 skip = ldb->vm.vblk_offset >> 9;
1366 finish = (size * ldb->vm.last_vblk_seq) >> 9;
1367
1368 for (s = skip; s < finish; s++) {
1369 data = read_part_sector(state, base + OFF_VMDB + s, §);
1370 if (!data) {
1371 ldm_crit ("Disk read failed.");
1372 goto out;
1373 }
1374
1375 for (v = 0; v < perbuf; v++, data+=size) {
1376 if (MAGIC_VBLK != get_unaligned_be32(data)) {
1377 ldm_error ("Expected to find a VBLK.");
1378 goto out;
1379 }
1380
1381 recs = get_unaligned_be16(data + 0x0E);
1382 if (recs == 1) {
1383 if (!ldm_ldmdb_add (data, size, ldb))
1384 goto out;
1385 } else if (recs > 1) {
1386 if (!ldm_frag_add (data, size, &frags))
1387 goto out;
1388 }
1389
1390 }
1391 put_dev_sector (sect);
1392 data = NULL;
1393 }
1394
1395 result = ldm_frag_commit (&frags, ldb);
1396out:
1397 if (data)
1398 put_dev_sector (sect);
1399 ldm_frag_free (&frags);
1400
1401 return result;
1402}
1403
1404
1405
1406
1407
1408
1409
1410
1411
1412static void ldm_free_vblks (struct list_head *lh)
1413{
1414 struct list_head *item, *tmp;
1415
1416 BUG_ON (!lh);
1417
1418 list_for_each_safe (item, tmp, lh)
1419 kfree (list_entry (item, struct vblk, list));
1420}
1421
1422
1423
1424
1425
1426
1427
1428
1429
1430
1431
1432
1433
1434
1435
1436
1437
1438
1439
1440int ldm_partition(struct parsed_partitions *state)
1441{
1442 struct ldmdb *ldb;
1443 unsigned long base;
1444 int result = -1;
1445
1446 BUG_ON(!state);
1447
1448
1449 if (!ldm_validate_partition_table(state))
1450 return 0;
1451
1452 ldb = kmalloc (sizeof (*ldb), GFP_KERNEL);
1453 if (!ldb) {
1454 ldm_crit ("Out of memory.");
1455 goto out;
1456 }
1457
1458
1459 if (!ldm_validate_privheads(state, &ldb->ph))
1460 goto out;
1461
1462
1463 base = ldb->ph.config_start;
1464
1465
1466 if (!ldm_validate_tocblocks(state, base, ldb) ||
1467 !ldm_validate_vmdb(state, base, ldb))
1468 goto out;
1469
1470
1471 INIT_LIST_HEAD (&ldb->v_dgrp);
1472 INIT_LIST_HEAD (&ldb->v_disk);
1473 INIT_LIST_HEAD (&ldb->v_volu);
1474 INIT_LIST_HEAD (&ldb->v_comp);
1475 INIT_LIST_HEAD (&ldb->v_part);
1476
1477 if (!ldm_get_vblks(state, base, ldb)) {
1478 ldm_crit ("Failed to read the VBLKs from the database.");
1479 goto cleanup;
1480 }
1481
1482
1483 if (ldm_create_data_partitions(state, ldb)) {
1484 ldm_debug ("Parsed LDM database successfully.");
1485 result = 1;
1486 }
1487
1488
1489cleanup:
1490 ldm_free_vblks (&ldb->v_dgrp);
1491 ldm_free_vblks (&ldb->v_disk);
1492 ldm_free_vblks (&ldb->v_volu);
1493 ldm_free_vblks (&ldb->v_comp);
1494 ldm_free_vblks (&ldb->v_part);
1495out:
1496 kfree (ldb);
1497 return result;
1498}
1499