1  // SPDX-License-Identifier: GPL-2.0-or-later
2  /*
3   * ldm - Support for Windows Logical Disk Manager (Dynamic Disks)
4   *
5   * Copyright (C) 2001,2002 Richard Russon <ldm@flatcap.org>
6   * Copyright (c) 2001-2012 Anton Altaparmakov
7   * Copyright (C) 2001,2002 Jakob Kemi <jakob.kemi@telia.com>
8   *
9   * Documentation is available at http://www.linux-ntfs.org/doku.php?id=downloads
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   * ldm_debug/info/error/crit - Output an error message
24   * @f:    A printf format string containing the message
25   * @...:  Variables to substitute into @f
26   *
27   * ldm_debug() writes a DEBUG level message to the syslog but only if the
28   * driver was compiled with debug enabled. Otherwise, the call turns into a NOP.
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  
40  static __printf(3, 4)
_ldm_printk(const char * level,const char * function,const char * fmt,...)41  void _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   * ldm_parse_privhead - Read the LDM Database PRIVHEAD structure
58   * @data:  Raw database PRIVHEAD structure loaded from the device
59   * @ph:    In-memory privhead structure in which to return parsed information
60   *
61   * This parses the LDM database PRIVHEAD structure supplied in @data and
62   * sets up the in-memory privhead structure @ph with the obtained information.
63   *
64   * Return:  'true'   @ph contains the PRIVHEAD data
65   *          'false'  @ph contents are undefined
66   */
ldm_parse_privhead(const u8 * data,struct privhead * ph)67  static 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  	/* Version 2.11 is Win2k/XP and version 2.12 is Vista. */
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) {	/* 1 MiB in sectors. */
94  		/* Warn the user and continue, carefully. */
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   * ldm_parse_tocblock - Read the LDM Database TOCBLOCK structure
114   * @data:  Raw database TOCBLOCK structure loaded from the device
115   * @toc:   In-memory toc structure in which to return parsed information
116   *
117   * This parses the LDM Database TOCBLOCK (table of contents) structure supplied
118   * in @data and sets up the in-memory tocblock structure @toc with the obtained
119   * information.
120   *
121   * N.B.  The *_start and *_size values returned in @toc are not range-checked.
122   *
123   * Return:  'true'   @toc contains the TOCBLOCK data
124   *          'false'  @toc contents are undefined
125   */
ldm_parse_tocblock(const u8 * data,struct tocblock * toc)126  static 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   * ldm_parse_vmdb - Read the LDM Database VMDB structure
161   * @data:  Raw database VMDB structure loaded from the device
162   * @vm:    In-memory vmdb structure in which to return parsed information
163   *
164   * This parses the LDM Database VMDB structure supplied in @data and sets up
165   * the in-memory vmdb structure @vm with the obtained information.
166   *
167   * N.B.  The *_start, *_size and *_seq values will be range-checked later.
168   *
169   * Return:  'true'   @vm contains VMDB info
170   *          'false'  @vm contents are undefined
171   */
ldm_parse_vmdb(const u8 * data,struct vmdb * vm)172  static 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   * ldm_compare_privheads - Compare two privhead objects
204   * @ph1:  First privhead
205   * @ph2:  Second privhead
206   *
207   * This compares the two privhead structures @ph1 and @ph2.
208   *
209   * Return:  'true'   Identical
210   *          'false'  Different
211   */
ldm_compare_privheads(const struct privhead * ph1,const struct privhead * ph2)212  static 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   * ldm_compare_tocblocks - Compare two tocblock objects
228   * @toc1:  First toc
229   * @toc2:  Second toc
230   *
231   * This compares the two tocblock structures @toc1 and @toc2.
232   *
233   * Return:  'true'   Identical
234   *          'false'  Different
235   */
ldm_compare_tocblocks(const struct tocblock * toc1,const struct tocblock * toc2)236  static 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   * ldm_validate_privheads - Compare the primary privhead with its backups
253   * @state: Partition check state including device holding the LDM Database
254   * @ph1:   Memory struct to fill with ph contents
255   *
256   * Read and compare all three privheads from disk.
257   *
258   * The privheads on disk show the size and location of the main disk area and
259   * the configuration area (the database).  The values are range-checked against
260   * @hd, which contains the real size of the disk.
261   *
262   * Return:  'true'   Success
263   *          'false'  Error
264   */
ldm_validate_privheads(struct parsed_partitions * state,struct privhead * ph1)265  static 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  	/* off[1 & 2] are relative to ph[0]->config_start */
286  	ph[0]->config_start = 0;
287  
288  	/* Read and parse privheads */
289  	for (i = 0; i < 3; i++) {
290  		data = read_part_sector(state, ph[0]->config_start + off[i],
291  					&sect);
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); /* Log again */
300  			if (i < 2)
301  				goto out;	/* Already logged */
302  			else
303  				break;	/* FIXME ignore for now, 3rd PH can fail on odd-sized disks */
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  	/* FIXME ignore this for now
327  	if (!ldm_compare_privheads (ph[0], ph[2])) {
328  		ldm_crit ("Primary and backup PRIVHEADs don't match.");
329  		goto out;
330  	}*/
331  	ldm_debug ("Validated PRIVHEADs successfully.");
332  	result = true;
333  out:
334  	kfree (ph[1]);
335  	kfree (ph[2]);
336  	return result;
337  }
338  
339  /**
340   * ldm_validate_tocblocks - Validate the table of contents and its backups
341   * @state: Partition check state including device holding the LDM Database
342   * @base:  Offset, into @state->disk, of the database
343   * @ldb:   Cache of the database structures
344   *
345   * Find and compare the four tables of contents of the LDM Database stored on
346   * @state->disk and return the parsed information into @toc1.
347   *
348   * The offsets and sizes of the configs are range-checked against a privhead.
349   *
350   * Return:  'true'   @toc1 contains validated TOCBLOCK info
351   *          'false'  @toc1 contents are undefined
352   */
ldm_validate_tocblocks(struct parsed_partitions * state,unsigned long base,struct ldmdb * ldb)353  static 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  	 * Try to read and parse all four TOCBLOCKs.
376  	 *
377  	 * Windows Vista LDM v2.12 does not always have all four TOCBLOCKs so
378  	 * skip any that fail as long as we get at least one valid TOCBLOCK.
379  	 */
380  	for (nr_tbs = i = 0; i < 4; i++) {
381  		data = read_part_sector(state, base + off[i], &sect);
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  	/* Range check the TOCBLOCK against a privhead. */
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  	/* Compare all loaded TOCBLOCKs. */
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;
410  err:
411  	kfree(tb[1]);
412  	return result;
413  }
414  
415  /**
416   * ldm_validate_vmdb - Read the VMDB and validate it
417   * @state: Partition check state including device holding the LDM Database
418   * @base:  Offset, into @bdev, of the database
419   * @ldb:   Cache of the database structures
420   *
421   * Find the vmdb of the LDM Database stored on @bdev and return the parsed
422   * information in @ldb.
423   *
424   * Return:  'true'   @ldb contains validated VBDB info
425   *          'false'  @ldb contents are undefined
426   */
ldm_validate_vmdb(struct parsed_partitions * state,unsigned long base,struct ldmdb * ldb)427  static 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, &sect);
442  	if (!data) {
443  		ldm_crit ("Disk read failed.");
444  		return false;
445  	}
446  
447  	if (!ldm_parse_vmdb (data, vm))
448  		goto out;				/* Already logged */
449  
450  	/* Are there uncommitted transactions? */
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  	 * The last_vblkd_seq can be before the end of the vmdb, just make sure
461  	 * it is not out of bounds.
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;
470  out:
471  	put_dev_sector (sect);
472  	return result;
473  }
474  
475  
476  /**
477   * ldm_validate_partition_table - Determine whether bdev might be a dynamic disk
478   * @state: Partition check state including device holding the LDM Database
479   *
480   * This function provides a weak test to decide whether the device is a dynamic
481   * disk or not.  It looks for an MS-DOS-style partition table containing at
482   * least one partition of type 0x42 (formerly SFS, now used by Windows for
483   * dynamic disks).
484   *
485   * N.B.  The only possible error can come from the read_part_sector and that is
486   *       only likely to happen if the underlying device is strange.  If that IS
487   *       the case we should return zero to let someone else try.
488   *
489   * Return:  'true'   @state->disk is a dynamic disk
490   *          'false'  @state->disk is not a dynamic disk, or an error occurred
491   */
ldm_validate_partition_table(struct parsed_partitions * state)492  static 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, &sect);
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  
521  out:
522  	put_dev_sector (sect);
523  	return result;
524  }
525  
526  /**
527   * ldm_get_disk_objid - Search a linked list of vblk's for a given Disk Id
528   * @ldb:  Cache of the database structures
529   *
530   * The LDM Database contains a list of all partitions on all dynamic disks.
531   * The primary PRIVHEAD, at the beginning of the physical disk, tells us
532   * the GUID of this disk.  This function searches for the GUID in a linked
533   * list of vblk's.
534   *
535   * Return:  Pointer, A matching vblk was found
536   *          NULL,    No match, or an error
537   */
ldm_get_disk_objid(const struct ldmdb * ldb)538  static 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   * ldm_create_data_partitions - Create data partitions for this device
555   * @pp:   List of the partitions parsed so far
556   * @ldb:  Cache of the database structures
557   *
558   * The database contains ALL the partitions for ALL disk groups, so we need to
559   * filter out this specific disk. Using the disk's object id, we can find all
560   * the partitions in the database that belong to this disk.
561   *
562   * Add each partition in our database, to the parsed_partitions structure.
563   *
564   * N.B.  This function creates the partitions in the order it finds partition
565   *       objects in the linked list.
566   *
567   * Return:  'true'   Partition created
568   *          'false'  Error, probably a range checking problem
569   */
ldm_create_data_partitions(struct parsed_partitions * pp,const struct ldmdb * ldb)570  static 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  	/* Create the data partitions */
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   * ldm_relative - Calculate the next relative offset
609   * @buffer:  Block of data being worked on
610   * @buflen:  Size of the block of data
611   * @base:    Size of the previous fixed width fields
612   * @offset:  Cumulative size of the previous variable-width fields
613   *
614   * Because many of the VBLK fields are variable-width, it's necessary
615   * to calculate each offset based on the previous one and the length
616   * of the field it pointed to.
617   *
618   * Return:  -1 Error, the calculated offset exceeded the size of the buffer
619   *           n OK, a range-checked offset into buffer
620   */
ldm_relative(const u8 * buffer,int buflen,int base,int offset)621  static 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   * ldm_get_vnum - Convert a variable-width, big endian number, into cpu order
644   * @block:  Pointer to the variable-width number to convert
645   *
646   * Large numbers in the LDM Database are often stored in a packed format.  Each
647   * number is prefixed by a one byte width marker.  All numbers in the database
648   * are stored in big-endian byte order.  This function reads one of these
649   * numbers and returns the result
650   *
651   * N.B.  This function DOES NOT perform any range checking, though the most
652   *       it will read is eight bytes.
653   *
654   * Return:  n A number
655   *          0 Zero, or an error occurred
656   */
ldm_get_vnum(const u8 * block)657  static 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   * ldm_get_vstr - Read a length-prefixed string into a buffer
677   * @block:   Pointer to the length marker
678   * @buffer:  Location to copy string to
679   * @buflen:  Size of the output buffer
680   *
681   * Many of the strings in the LDM Database are not NULL terminated.  Instead
682   * they are prefixed by a one byte length marker.  This function copies one of
683   * these strings into a buffer.
684   *
685   * N.B.  This function DOES NOT perform any range checking on the input.
686   *       If the buffer is too small, the output will be truncated.
687   *
688   * Return:  0, Error and @buffer contents are undefined
689   *          n, String length in characters (excluding NULL)
690   *          buflen-1, String was truncated.
691   */
ldm_get_vstr(const u8 * block,u8 * buffer,int buflen)692  static 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   * ldm_parse_cmp3 - Read a raw VBLK Component object into a vblk structure
711   * @buffer:  Block of data being worked on
712   * @buflen:  Size of the block of data
713   * @vb:      In-memory vblk in which to return information
714   *
715   * Read a raw VBLK Component object (version 3) into a vblk structure.
716   *
717   * Return:  'true'   @vb contains a Component VBLK
718   *          'false'  @vb contents are not defined
719   */
ldm_parse_cmp3(const u8 * buffer,int buflen,struct vblk * vb)720  static 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  		len = r_parent;
740  	}
741  	if (len < 0)
742  		return false;
743  
744  	len += VBLK_SIZE_CMP3;
745  	if (len != get_unaligned_be32(buffer + 0x14))
746  		return false;
747  
748  	comp = &vb->vblk.comp;
749  	ldm_get_vstr (buffer + 0x18 + r_name, comp->state,
750  		sizeof (comp->state));
751  	comp->type      = buffer[0x18 + r_vstate];
752  	comp->children  = ldm_get_vnum (buffer + 0x1D + r_vstate);
753  	comp->parent_id = ldm_get_vnum (buffer + 0x2D + r_child);
754  	comp->chunksize = r_stripe ? ldm_get_vnum (buffer+r_parent+0x2E) : 0;
755  
756  	return true;
757  }
758  
759  /**
760   * ldm_parse_dgr3 - Read a raw VBLK Disk Group object into a vblk structure
761   * @buffer:  Block of data being worked on
762   * @buflen:  Size of the block of data
763   * @vb:      In-memory vblk in which to return information
764   *
765   * Read a raw VBLK Disk Group object (version 3) into a vblk structure.
766   *
767   * Return:  'true'   @vb contains a Disk Group VBLK
768   *          'false'  @vb contents are not defined
769   */
ldm_parse_dgr3(const u8 * buffer,int buflen,struct vblk * vb)770  static int ldm_parse_dgr3 (const u8 *buffer, int buflen, struct vblk *vb)
771  {
772  	int r_objid, r_name, r_diskid, r_id1, r_id2, len;
773  	struct vblk_dgrp *dgrp;
774  
775  	BUG_ON (!buffer || !vb);
776  
777  	r_objid  = ldm_relative (buffer, buflen, 0x18, 0);
778  	r_name   = ldm_relative (buffer, buflen, 0x18, r_objid);
779  	r_diskid = ldm_relative (buffer, buflen, 0x18, r_name);
780  
781  	if (buffer[0x12] & VBLK_FLAG_DGR3_IDS) {
782  		r_id1 = ldm_relative (buffer, buflen, 0x24, r_diskid);
783  		r_id2 = ldm_relative (buffer, buflen, 0x24, r_id1);
784  		len = r_id2;
785  	} else
786  		len = r_diskid;
787  	if (len < 0)
788  		return false;
789  
790  	len += VBLK_SIZE_DGR3;
791  	if (len != get_unaligned_be32(buffer + 0x14))
792  		return false;
793  
794  	dgrp = &vb->vblk.dgrp;
795  	ldm_get_vstr (buffer + 0x18 + r_name, dgrp->disk_id,
796  		sizeof (dgrp->disk_id));
797  	return true;
798  }
799  
800  /**
801   * ldm_parse_dgr4 - Read a raw VBLK Disk Group object into a vblk structure
802   * @buffer:  Block of data being worked on
803   * @buflen:  Size of the block of data
804   * @vb:      In-memory vblk in which to return information
805   *
806   * Read a raw VBLK Disk Group object (version 4) into a vblk structure.
807   *
808   * Return:  'true'   @vb contains a Disk Group VBLK
809   *          'false'  @vb contents are not defined
810   */
ldm_parse_dgr4(const u8 * buffer,int buflen,struct vblk * vb)811  static bool ldm_parse_dgr4 (const u8 *buffer, int buflen, struct vblk *vb)
812  {
813  	char buf[64];
814  	int r_objid, r_name, r_id1, r_id2, len;
815  
816  	BUG_ON (!buffer || !vb);
817  
818  	r_objid  = ldm_relative (buffer, buflen, 0x18, 0);
819  	r_name   = ldm_relative (buffer, buflen, 0x18, r_objid);
820  
821  	if (buffer[0x12] & VBLK_FLAG_DGR4_IDS) {
822  		r_id1 = ldm_relative (buffer, buflen, 0x44, r_name);
823  		r_id2 = ldm_relative (buffer, buflen, 0x44, r_id1);
824  		len = r_id2;
825  	} else
826  		len = r_name;
827  	if (len < 0)
828  		return false;
829  
830  	len += VBLK_SIZE_DGR4;
831  	if (len != get_unaligned_be32(buffer + 0x14))
832  		return false;
833  
834  	ldm_get_vstr (buffer + 0x18 + r_objid, buf, sizeof (buf));
835  	return true;
836  }
837  
838  /**
839   * ldm_parse_dsk3 - Read a raw VBLK Disk object into a vblk structure
840   * @buffer:  Block of data being worked on
841   * @buflen:  Size of the block of data
842   * @vb:      In-memory vblk in which to return information
843   *
844   * Read a raw VBLK Disk object (version 3) into a vblk structure.
845   *
846   * Return:  'true'   @vb contains a Disk VBLK
847   *          'false'  @vb contents are not defined
848   */
ldm_parse_dsk3(const u8 * buffer,int buflen,struct vblk * vb)849  static bool ldm_parse_dsk3 (const u8 *buffer, int buflen, struct vblk *vb)
850  {
851  	int r_objid, r_name, r_diskid, r_altname, len;
852  	struct vblk_disk *disk;
853  
854  	BUG_ON (!buffer || !vb);
855  
856  	r_objid   = ldm_relative (buffer, buflen, 0x18, 0);
857  	r_name    = ldm_relative (buffer, buflen, 0x18, r_objid);
858  	r_diskid  = ldm_relative (buffer, buflen, 0x18, r_name);
859  	r_altname = ldm_relative (buffer, buflen, 0x18, r_diskid);
860  	len = r_altname;
861  	if (len < 0)
862  		return false;
863  
864  	len += VBLK_SIZE_DSK3;
865  	if (len != get_unaligned_be32(buffer + 0x14))
866  		return false;
867  
868  	disk = &vb->vblk.disk;
869  	ldm_get_vstr (buffer + 0x18 + r_diskid, disk->alt_name,
870  		sizeof (disk->alt_name));
871  	if (uuid_parse(buffer + 0x19 + r_name, &disk->disk_id))
872  		return false;
873  
874  	return true;
875  }
876  
877  /**
878   * ldm_parse_dsk4 - Read a raw VBLK Disk object into a vblk structure
879   * @buffer:  Block of data being worked on
880   * @buflen:  Size of the block of data
881   * @vb:      In-memory vblk in which to return information
882   *
883   * Read a raw VBLK Disk object (version 4) into a vblk structure.
884   *
885   * Return:  'true'   @vb contains a Disk VBLK
886   *          'false'  @vb contents are not defined
887   */
ldm_parse_dsk4(const u8 * buffer,int buflen,struct vblk * vb)888  static bool ldm_parse_dsk4 (const u8 *buffer, int buflen, struct vblk *vb)
889  {
890  	int r_objid, r_name, len;
891  	struct vblk_disk *disk;
892  
893  	BUG_ON (!buffer || !vb);
894  
895  	r_objid = ldm_relative (buffer, buflen, 0x18, 0);
896  	r_name  = ldm_relative (buffer, buflen, 0x18, r_objid);
897  	len     = r_name;
898  	if (len < 0)
899  		return false;
900  
901  	len += VBLK_SIZE_DSK4;
902  	if (len != get_unaligned_be32(buffer + 0x14))
903  		return false;
904  
905  	disk = &vb->vblk.disk;
906  	import_uuid(&disk->disk_id, buffer + 0x18 + r_name);
907  	return true;
908  }
909  
910  /**
911   * ldm_parse_prt3 - Read a raw VBLK Partition object into a vblk structure
912   * @buffer:  Block of data being worked on
913   * @buflen:  Size of the block of data
914   * @vb:      In-memory vblk in which to return information
915   *
916   * Read a raw VBLK Partition object (version 3) into a vblk structure.
917   *
918   * Return:  'true'   @vb contains a Partition VBLK
919   *          'false'  @vb contents are not defined
920   */
ldm_parse_prt3(const u8 * buffer,int buflen,struct vblk * vb)921  static bool ldm_parse_prt3(const u8 *buffer, int buflen, struct vblk *vb)
922  {
923  	int r_objid, r_name, r_size, r_parent, r_diskid, r_index, len;
924  	struct vblk_part *part;
925  
926  	BUG_ON(!buffer || !vb);
927  	r_objid = ldm_relative(buffer, buflen, 0x18, 0);
928  	if (r_objid < 0) {
929  		ldm_error("r_objid %d < 0", r_objid);
930  		return false;
931  	}
932  	r_name = ldm_relative(buffer, buflen, 0x18, r_objid);
933  	if (r_name < 0) {
934  		ldm_error("r_name %d < 0", r_name);
935  		return false;
936  	}
937  	r_size = ldm_relative(buffer, buflen, 0x34, r_name);
938  	if (r_size < 0) {
939  		ldm_error("r_size %d < 0", r_size);
940  		return false;
941  	}
942  	r_parent = ldm_relative(buffer, buflen, 0x34, r_size);
943  	if (r_parent < 0) {
944  		ldm_error("r_parent %d < 0", r_parent);
945  		return false;
946  	}
947  	r_diskid = ldm_relative(buffer, buflen, 0x34, r_parent);
948  	if (r_diskid < 0) {
949  		ldm_error("r_diskid %d < 0", r_diskid);
950  		return false;
951  	}
952  	if (buffer[0x12] & VBLK_FLAG_PART_INDEX) {
953  		r_index = ldm_relative(buffer, buflen, 0x34, r_diskid);
954  		if (r_index < 0) {
955  			ldm_error("r_index %d < 0", r_index);
956  			return false;
957  		}
958  		len = r_index;
959  	} else
960  		len = r_diskid;
961  	if (len < 0) {
962  		ldm_error("len %d < 0", len);
963  		return false;
964  	}
965  	len += VBLK_SIZE_PRT3;
966  	if (len > get_unaligned_be32(buffer + 0x14)) {
967  		ldm_error("len %d > BE32(buffer + 0x14) %d", len,
968  				get_unaligned_be32(buffer + 0x14));
969  		return false;
970  	}
971  	part = &vb->vblk.part;
972  	part->start = get_unaligned_be64(buffer + 0x24 + r_name);
973  	part->volume_offset = get_unaligned_be64(buffer + 0x2C + r_name);
974  	part->size = ldm_get_vnum(buffer + 0x34 + r_name);
975  	part->parent_id = ldm_get_vnum(buffer + 0x34 + r_size);
976  	part->disk_id = ldm_get_vnum(buffer + 0x34 + r_parent);
977  	if (vb->flags & VBLK_FLAG_PART_INDEX)
978  		part->partnum = buffer[0x35 + r_diskid];
979  	else
980  		part->partnum = 0;
981  	return true;
982  }
983  
984  /**
985   * ldm_parse_vol5 - Read a raw VBLK Volume object into a vblk structure
986   * @buffer:  Block of data being worked on
987   * @buflen:  Size of the block of data
988   * @vb:      In-memory vblk in which to return information
989   *
990   * Read a raw VBLK Volume object (version 5) into a vblk structure.
991   *
992   * Return:  'true'   @vb contains a Volume VBLK
993   *          'false'  @vb contents are not defined
994   */
ldm_parse_vol5(const u8 * buffer,int buflen,struct vblk * vb)995  static bool ldm_parse_vol5(const u8 *buffer, int buflen, struct vblk *vb)
996  {
997  	int r_objid, r_name, r_vtype, r_disable_drive_letter, r_child, r_size;
998  	int r_id1, r_id2, r_size2, r_drive, len;
999  	struct vblk_volu *volu;
1000  
1001  	BUG_ON(!buffer || !vb);
1002  	r_objid = ldm_relative(buffer, buflen, 0x18, 0);
1003  	if (r_objid < 0) {
1004  		ldm_error("r_objid %d < 0", r_objid);
1005  		return false;
1006  	}
1007  	r_name = ldm_relative(buffer, buflen, 0x18, r_objid);
1008  	if (r_name < 0) {
1009  		ldm_error("r_name %d < 0", r_name);
1010  		return false;
1011  	}
1012  	r_vtype = ldm_relative(buffer, buflen, 0x18, r_name);
1013  	if (r_vtype < 0) {
1014  		ldm_error("r_vtype %d < 0", r_vtype);
1015  		return false;
1016  	}
1017  	r_disable_drive_letter = ldm_relative(buffer, buflen, 0x18, r_vtype);
1018  	if (r_disable_drive_letter < 0) {
1019  		ldm_error("r_disable_drive_letter %d < 0",
1020  				r_disable_drive_letter);
1021  		return false;
1022  	}
1023  	r_child = ldm_relative(buffer, buflen, 0x2D, r_disable_drive_letter);
1024  	if (r_child < 0) {
1025  		ldm_error("r_child %d < 0", r_child);
1026  		return false;
1027  	}
1028  	r_size = ldm_relative(buffer, buflen, 0x3D, r_child);
1029  	if (r_size < 0) {
1030  		ldm_error("r_size %d < 0", r_size);
1031  		return false;
1032  	}
1033  	if (buffer[0x12] & VBLK_FLAG_VOLU_ID1) {
1034  		r_id1 = ldm_relative(buffer, buflen, 0x52, r_size);
1035  		if (r_id1 < 0) {
1036  			ldm_error("r_id1 %d < 0", r_id1);
1037  			return false;
1038  		}
1039  	} else
1040  		r_id1 = r_size;
1041  	if (buffer[0x12] & VBLK_FLAG_VOLU_ID2) {
1042  		r_id2 = ldm_relative(buffer, buflen, 0x52, r_id1);
1043  		if (r_id2 < 0) {
1044  			ldm_error("r_id2 %d < 0", r_id2);
1045  			return false;
1046  		}
1047  	} else
1048  		r_id2 = r_id1;
1049  	if (buffer[0x12] & VBLK_FLAG_VOLU_SIZE) {
1050  		r_size2 = ldm_relative(buffer, buflen, 0x52, r_id2);
1051  		if (r_size2 < 0) {
1052  			ldm_error("r_size2 %d < 0", r_size2);
1053  			return false;
1054  		}
1055  	} else
1056  		r_size2 = r_id2;
1057  	if (buffer[0x12] & VBLK_FLAG_VOLU_DRIVE) {
1058  		r_drive = ldm_relative(buffer, buflen, 0x52, r_size2);
1059  		if (r_drive < 0) {
1060  			ldm_error("r_drive %d < 0", r_drive);
1061  			return false;
1062  		}
1063  	} else
1064  		r_drive = r_size2;
1065  	len = r_drive;
1066  	if (len < 0) {
1067  		ldm_error("len %d < 0", len);
1068  		return false;
1069  	}
1070  	len += VBLK_SIZE_VOL5;
1071  	if (len > get_unaligned_be32(buffer + 0x14)) {
1072  		ldm_error("len %d > BE32(buffer + 0x14) %d", len,
1073  				get_unaligned_be32(buffer + 0x14));
1074  		return false;
1075  	}
1076  	volu = &vb->vblk.volu;
1077  	ldm_get_vstr(buffer + 0x18 + r_name, volu->volume_type,
1078  			sizeof(volu->volume_type));
1079  	memcpy(volu->volume_state, buffer + 0x18 + r_disable_drive_letter,
1080  			sizeof(volu->volume_state));
1081  	volu->size = ldm_get_vnum(buffer + 0x3D + r_child);
1082  	volu->partition_type = buffer[0x41 + r_size];
1083  	memcpy(volu->guid, buffer + 0x42 + r_size, sizeof(volu->guid));
1084  	if (buffer[0x12] & VBLK_FLAG_VOLU_DRIVE) {
1085  		ldm_get_vstr(buffer + 0x52 + r_size, volu->drive_hint,
1086  				sizeof(volu->drive_hint));
1087  	}
1088  	return true;
1089  }
1090  
1091  /**
1092   * ldm_parse_vblk - Read a raw VBLK object into a vblk structure
1093   * @buf:  Block of data being worked on
1094   * @len:  Size of the block of data
1095   * @vb:   In-memory vblk in which to return information
1096   *
1097   * Read a raw VBLK object into a vblk structure.  This function just reads the
1098   * information common to all VBLK types, then delegates the rest of the work to
1099   * helper functions: ldm_parse_*.
1100   *
1101   * Return:  'true'   @vb contains a VBLK
1102   *          'false'  @vb contents are not defined
1103   */
ldm_parse_vblk(const u8 * buf,int len,struct vblk * vb)1104  static bool ldm_parse_vblk (const u8 *buf, int len, struct vblk *vb)
1105  {
1106  	bool result = false;
1107  	int r_objid;
1108  
1109  	BUG_ON (!buf || !vb);
1110  
1111  	r_objid = ldm_relative (buf, len, 0x18, 0);
1112  	if (r_objid < 0) {
1113  		ldm_error ("VBLK header is corrupt.");
1114  		return false;
1115  	}
1116  
1117  	vb->flags  = buf[0x12];
1118  	vb->type   = buf[0x13];
1119  	vb->obj_id = ldm_get_vnum (buf + 0x18);
1120  	ldm_get_vstr (buf+0x18+r_objid, vb->name, sizeof (vb->name));
1121  
1122  	switch (vb->type) {
1123  		case VBLK_CMP3:  result = ldm_parse_cmp3 (buf, len, vb); break;
1124  		case VBLK_DSK3:  result = ldm_parse_dsk3 (buf, len, vb); break;
1125  		case VBLK_DSK4:  result = ldm_parse_dsk4 (buf, len, vb); break;
1126  		case VBLK_DGR3:  result = ldm_parse_dgr3 (buf, len, vb); break;
1127  		case VBLK_DGR4:  result = ldm_parse_dgr4 (buf, len, vb); break;
1128  		case VBLK_PRT3:  result = ldm_parse_prt3 (buf, len, vb); break;
1129  		case VBLK_VOL5:  result = ldm_parse_vol5 (buf, len, vb); break;
1130  	}
1131  
1132  	if (result)
1133  		ldm_debug ("Parsed VBLK 0x%llx (type: 0x%02x) ok.",
1134  			 (unsigned long long) vb->obj_id, vb->type);
1135  	else
1136  		ldm_error ("Failed to parse VBLK 0x%llx (type: 0x%02x).",
1137  			(unsigned long long) vb->obj_id, vb->type);
1138  
1139  	return result;
1140  }
1141  
1142  
1143  /**
1144   * ldm_ldmdb_add - Adds a raw VBLK entry to the ldmdb database
1145   * @data:  Raw VBLK to add to the database
1146   * @len:   Size of the raw VBLK
1147   * @ldb:   Cache of the database structures
1148   *
1149   * The VBLKs are sorted into categories.  Partitions are also sorted by offset.
1150   *
1151   * N.B.  This function does not check the validity of the VBLKs.
1152   *
1153   * Return:  'true'   The VBLK was added
1154   *          'false'  An error occurred
1155   */
ldm_ldmdb_add(u8 * data,int len,struct ldmdb * ldb)1156  static bool ldm_ldmdb_add (u8 *data, int len, struct ldmdb *ldb)
1157  {
1158  	struct vblk *vb;
1159  	struct list_head *item;
1160  
1161  	BUG_ON (!data || !ldb);
1162  
1163  	vb = kmalloc (sizeof (*vb), GFP_KERNEL);
1164  	if (!vb) {
1165  		ldm_crit ("Out of memory.");
1166  		return false;
1167  	}
1168  
1169  	if (!ldm_parse_vblk (data, len, vb)) {
1170  		kfree(vb);
1171  		return false;			/* Already logged */
1172  	}
1173  
1174  	/* Put vblk into the correct list. */
1175  	switch (vb->type) {
1176  	case VBLK_DGR3:
1177  	case VBLK_DGR4:
1178  		list_add (&vb->list, &ldb->v_dgrp);
1179  		break;
1180  	case VBLK_DSK3:
1181  	case VBLK_DSK4:
1182  		list_add (&vb->list, &ldb->v_disk);
1183  		break;
1184  	case VBLK_VOL5:
1185  		list_add (&vb->list, &ldb->v_volu);
1186  		break;
1187  	case VBLK_CMP3:
1188  		list_add (&vb->list, &ldb->v_comp);
1189  		break;
1190  	case VBLK_PRT3:
1191  		/* Sort by the partition's start sector. */
1192  		list_for_each (item, &ldb->v_part) {
1193  			struct vblk *v = list_entry (item, struct vblk, list);
1194  			if ((v->vblk.part.disk_id == vb->vblk.part.disk_id) &&
1195  			    (v->vblk.part.start > vb->vblk.part.start)) {
1196  				list_add_tail (&vb->list, &v->list);
1197  				return true;
1198  			}
1199  		}
1200  		list_add_tail (&vb->list, &ldb->v_part);
1201  		break;
1202  	}
1203  	return true;
1204  }
1205  
1206  /**
1207   * ldm_frag_add - Add a VBLK fragment to a list
1208   * @data:   Raw fragment to be added to the list
1209   * @size:   Size of the raw fragment
1210   * @frags:  Linked list of VBLK fragments
1211   *
1212   * Fragmented VBLKs may not be consecutive in the database, so they are placed
1213   * in a list so they can be pieced together later.
1214   *
1215   * Return:  'true'   Success, the VBLK was added to the list
1216   *          'false'  Error, a problem occurred
1217   */
ldm_frag_add(const u8 * data,int size,struct list_head * frags)1218  static bool ldm_frag_add (const u8 *data, int size, struct list_head *frags)
1219  {
1220  	struct frag *f;
1221  	struct list_head *item;
1222  	int rec, num, group;
1223  
1224  	BUG_ON (!data || !frags);
1225  
1226  	if (size < 2 * VBLK_SIZE_HEAD) {
1227  		ldm_error("Value of size is too small.");
1228  		return false;
1229  	}
1230  
1231  	group = get_unaligned_be32(data + 0x08);
1232  	rec   = get_unaligned_be16(data + 0x0C);
1233  	num   = get_unaligned_be16(data + 0x0E);
1234  	if ((num < 1) || (num > 4)) {
1235  		ldm_error ("A VBLK claims to have %d parts.", num);
1236  		return false;
1237  	}
1238  	if (rec >= num) {
1239  		ldm_error("REC value (%d) exceeds NUM value (%d)", rec, num);
1240  		return false;
1241  	}
1242  
1243  	list_for_each (item, frags) {
1244  		f = list_entry (item, struct frag, list);
1245  		if (f->group == group)
1246  			goto found;
1247  	}
1248  
1249  	f = kmalloc (sizeof (*f) + size*num, GFP_KERNEL);
1250  	if (!f) {
1251  		ldm_crit ("Out of memory.");
1252  		return false;
1253  	}
1254  
1255  	f->group = group;
1256  	f->num   = num;
1257  	f->rec   = rec;
1258  	f->map   = 0xFF << num;
1259  
1260  	list_add_tail (&f->list, frags);
1261  found:
1262  	if (rec >= f->num) {
1263  		ldm_error("REC value (%d) exceeds NUM value (%d)", rec, f->num);
1264  		return false;
1265  	}
1266  	if (f->map & (1 << rec)) {
1267  		ldm_error ("Duplicate VBLK, part %d.", rec);
1268  		f->map &= 0x7F;			/* Mark the group as broken */
1269  		return false;
1270  	}
1271  	f->map |= (1 << rec);
1272  	if (!rec)
1273  		memcpy(f->data, data, VBLK_SIZE_HEAD);
1274  	data += VBLK_SIZE_HEAD;
1275  	size -= VBLK_SIZE_HEAD;
1276  	memcpy(f->data + VBLK_SIZE_HEAD + rec * size, data, size);
1277  	return true;
1278  }
1279  
1280  /**
1281   * ldm_frag_free - Free a linked list of VBLK fragments
1282   * @list:  Linked list of fragments
1283   *
1284   * Free a linked list of VBLK fragments
1285   *
1286   * Return:  none
1287   */
ldm_frag_free(struct list_head * list)1288  static void ldm_frag_free (struct list_head *list)
1289  {
1290  	struct list_head *item, *tmp;
1291  
1292  	BUG_ON (!list);
1293  
1294  	list_for_each_safe (item, tmp, list)
1295  		kfree (list_entry (item, struct frag, list));
1296  }
1297  
1298  /**
1299   * ldm_frag_commit - Validate fragmented VBLKs and add them to the database
1300   * @frags:  Linked list of VBLK fragments
1301   * @ldb:    Cache of the database structures
1302   *
1303   * Now that all the fragmented VBLKs have been collected, they must be added to
1304   * the database for later use.
1305   *
1306   * Return:  'true'   All the fragments we added successfully
1307   *          'false'  One or more of the fragments we invalid
1308   */
ldm_frag_commit(struct list_head * frags,struct ldmdb * ldb)1309  static bool ldm_frag_commit (struct list_head *frags, struct ldmdb *ldb)
1310  {
1311  	struct frag *f;
1312  	struct list_head *item;
1313  
1314  	BUG_ON (!frags || !ldb);
1315  
1316  	list_for_each (item, frags) {
1317  		f = list_entry (item, struct frag, list);
1318  
1319  		if (f->map != 0xFF) {
1320  			ldm_error ("VBLK group %d is incomplete (0x%02x).",
1321  				f->group, f->map);
1322  			return false;
1323  		}
1324  
1325  		if (!ldm_ldmdb_add (f->data, f->num*ldb->vm.vblk_size, ldb))
1326  			return false;		/* Already logged */
1327  	}
1328  	return true;
1329  }
1330  
1331  /**
1332   * ldm_get_vblks - Read the on-disk database of VBLKs into memory
1333   * @state: Partition check state including device holding the LDM Database
1334   * @base:  Offset, into @state->disk, of the database
1335   * @ldb:   Cache of the database structures
1336   *
1337   * To use the information from the VBLKs, they need to be read from the disk,
1338   * unpacked and validated.  We cache them in @ldb according to their type.
1339   *
1340   * Return:  'true'   All the VBLKs were read successfully
1341   *          'false'  An error occurred
1342   */
ldm_get_vblks(struct parsed_partitions * state,unsigned long base,struct ldmdb * ldb)1343  static bool ldm_get_vblks(struct parsed_partitions *state, unsigned long base,
1344  			  struct ldmdb *ldb)
1345  {
1346  	int size, perbuf, skip, finish, s, v, recs;
1347  	u8 *data = NULL;
1348  	Sector sect;
1349  	bool result = false;
1350  	LIST_HEAD (frags);
1351  
1352  	BUG_ON(!state || !ldb);
1353  
1354  	size   = ldb->vm.vblk_size;
1355  	perbuf = 512 / size;
1356  	skip   = ldb->vm.vblk_offset >> 9;		/* Bytes to sectors */
1357  	finish = (size * ldb->vm.last_vblk_seq) >> 9;
1358  
1359  	for (s = skip; s < finish; s++) {		/* For each sector */
1360  		data = read_part_sector(state, base + OFF_VMDB + s, &sect);
1361  		if (!data) {
1362  			ldm_crit ("Disk read failed.");
1363  			goto out;
1364  		}
1365  
1366  		for (v = 0; v < perbuf; v++, data+=size) {  /* For each vblk */
1367  			if (MAGIC_VBLK != get_unaligned_be32(data)) {
1368  				ldm_error ("Expected to find a VBLK.");
1369  				goto out;
1370  			}
1371  
1372  			recs = get_unaligned_be16(data + 0x0E);	/* Number of records */
1373  			if (recs == 1) {
1374  				if (!ldm_ldmdb_add (data, size, ldb))
1375  					goto out;	/* Already logged */
1376  			} else if (recs > 1) {
1377  				if (!ldm_frag_add (data, size, &frags))
1378  					goto out;	/* Already logged */
1379  			}
1380  			/* else Record is not in use, ignore it. */
1381  		}
1382  		put_dev_sector (sect);
1383  		data = NULL;
1384  	}
1385  
1386  	result = ldm_frag_commit (&frags, ldb);	/* Failures, already logged */
1387  out:
1388  	if (data)
1389  		put_dev_sector (sect);
1390  	ldm_frag_free (&frags);
1391  
1392  	return result;
1393  }
1394  
1395  /**
1396   * ldm_free_vblks - Free a linked list of vblk's
1397   * @lh:  Head of a linked list of struct vblk
1398   *
1399   * Free a list of vblk's and free the memory used to maintain the list.
1400   *
1401   * Return:  none
1402   */
ldm_free_vblks(struct list_head * lh)1403  static void ldm_free_vblks (struct list_head *lh)
1404  {
1405  	struct list_head *item, *tmp;
1406  
1407  	BUG_ON (!lh);
1408  
1409  	list_for_each_safe (item, tmp, lh)
1410  		kfree (list_entry (item, struct vblk, list));
1411  }
1412  
1413  
1414  /**
1415   * ldm_partition - Find out whether a device is a dynamic disk and handle it
1416   * @state: Partition check state including device holding the LDM Database
1417   *
1418   * This determines whether the device @bdev is a dynamic disk and if so creates
1419   * the partitions necessary in the gendisk structure pointed to by @hd.
1420   *
1421   * We create a dummy device 1, which contains the LDM database, and then create
1422   * each partition described by the LDM database in sequence as devices 2+. For
1423   * example, if the device is hda, we would have: hda1: LDM database, hda2, hda3,
1424   * and so on: the actual data containing partitions.
1425   *
1426   * Return:  1 Success, @state->disk is a dynamic disk and we handled it
1427   *          0 Success, @state->disk is not a dynamic disk
1428   *         -1 An error occurred before enough information had been read
1429   *            Or @state->disk is a dynamic disk, but it may be corrupted
1430   */
ldm_partition(struct parsed_partitions * state)1431  int ldm_partition(struct parsed_partitions *state)
1432  {
1433  	struct ldmdb  *ldb;
1434  	unsigned long base;
1435  	int result = -1;
1436  
1437  	BUG_ON(!state);
1438  
1439  	/* Look for signs of a Dynamic Disk */
1440  	if (!ldm_validate_partition_table(state))
1441  		return 0;
1442  
1443  	ldb = kmalloc (sizeof (*ldb), GFP_KERNEL);
1444  	if (!ldb) {
1445  		ldm_crit ("Out of memory.");
1446  		goto out;
1447  	}
1448  
1449  	/* Parse and check privheads. */
1450  	if (!ldm_validate_privheads(state, &ldb->ph))
1451  		goto out;		/* Already logged */
1452  
1453  	/* All further references are relative to base (database start). */
1454  	base = ldb->ph.config_start;
1455  
1456  	/* Parse and check tocs and vmdb. */
1457  	if (!ldm_validate_tocblocks(state, base, ldb) ||
1458  	    !ldm_validate_vmdb(state, base, ldb))
1459  	    	goto out;		/* Already logged */
1460  
1461  	/* Initialize vblk lists in ldmdb struct */
1462  	INIT_LIST_HEAD (&ldb->v_dgrp);
1463  	INIT_LIST_HEAD (&ldb->v_disk);
1464  	INIT_LIST_HEAD (&ldb->v_volu);
1465  	INIT_LIST_HEAD (&ldb->v_comp);
1466  	INIT_LIST_HEAD (&ldb->v_part);
1467  
1468  	if (!ldm_get_vblks(state, base, ldb)) {
1469  		ldm_crit ("Failed to read the VBLKs from the database.");
1470  		goto cleanup;
1471  	}
1472  
1473  	/* Finally, create the data partition devices. */
1474  	if (ldm_create_data_partitions(state, ldb)) {
1475  		ldm_debug ("Parsed LDM database successfully.");
1476  		result = 1;
1477  	}
1478  	/* else Already logged */
1479  
1480  cleanup:
1481  	ldm_free_vblks (&ldb->v_dgrp);
1482  	ldm_free_vblks (&ldb->v_disk);
1483  	ldm_free_vblks (&ldb->v_volu);
1484  	ldm_free_vblks (&ldb->v_comp);
1485  	ldm_free_vblks (&ldb->v_part);
1486  out:
1487  	kfree (ldb);
1488  	return result;
1489  }
1490