Reconstruct source directory tree.
[lwext4.git] / src / ext4_mbr.c
1 /*
2  * Copyright (c) 2015 Grzegorz Kostka (kostka.grzegorz@gmail.com)
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  *
9  * - Redistributions of source code must retain the above copyright
10  *   notice, this list of conditions and the following disclaimer.
11  * - Redistributions in binary form must reproduce the above copyright
12  *   notice, this list of conditions and the following disclaimer in the
13  *   documentation and/or other materials provided with the distribution.
14  * - The name of the author may not be used to endorse or promote products
15  *   derived from this software without specific prior written permission.
16  *
17  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27  */
28
29 /** @addtogroup lwext4
30  * @{
31  */
32 /**
33  * @file  ext4_mbr.c
34  * @brief Master boot record parser
35  */
36
37 #include "ext4_config.h"
38 #include "ext4_types.h"
39 #include "ext4_debug.h"
40 #include "ext4_mbr.h"
41
42 #include <inttypes.h>
43 #include <string.h>
44
45 #define MBR_SIGNATURE 0xAA55
46
47 #pragma pack(push, 1)
48
49 struct ext4_part_entry {
50         uint8_t status;
51         uint8_t chs1[3];
52         uint8_t type;
53         uint8_t chs2[3];
54         uint32_t first_lba;
55         uint32_t sectors;
56 };
57
58 struct ext4_mbr {
59         uint8_t bootstrap[446];
60         struct ext4_part_entry part_entry[4];
61         uint16_t signature;
62 };
63
64 #pragma pack(pop)
65
66 int ext4_mbr_scan(struct ext4_blockdev *parent, struct ext4_mbr_bdevs *bdevs)
67 {
68         int r;
69         size_t i;
70
71         ext4_dbg(DEBUG_MBR, DBG_INFO "ext4_mbr_scan\n");
72         memset(bdevs, 0, sizeof(struct ext4_mbr_bdevs));
73         r = ext4_block_init(parent);
74         if (r != EOK)
75                 return r;
76
77         r = ext4_block_readbytes(parent, 0, parent->bdif->ph_bbuf, 512);
78         if (r != EOK) {
79                 goto blockdev_fini;
80         }
81
82         const struct ext4_mbr *mbr = (void *)parent->bdif->ph_bbuf;
83
84         if (to_le16(mbr->signature) != MBR_SIGNATURE) {
85                 ext4_dbg(DEBUG_MBR, DBG_ERROR "ext4_mbr_scan: unknown "
86                          "signature: 0x%x\n", to_le16(mbr->signature));
87                 r = ENOENT;
88                 goto blockdev_fini;
89         }
90
91         /*Show bootstrap code*/
92         ext4_dbg(DEBUG_MBR, "mbr_part: bootstrap:");
93         for (i = 0; i < sizeof(mbr->bootstrap); ++i) {
94                 if (!(i & 0xF))
95                                 ext4_dbg(DEBUG_MBR | DEBUG_NOPREFIX, "\n");
96                 ext4_dbg(DEBUG_MBR | DEBUG_NOPREFIX, "%02x, ", mbr->bootstrap[i]);
97         }
98
99         ext4_dbg(DEBUG_MBR | DEBUG_NOPREFIX, "\n\n");
100         for (i = 0; i < 4; ++i) {
101                 const struct ext4_part_entry *pe = &mbr->part_entry[i];
102                 ext4_dbg(DEBUG_MBR, "mbr_part: %d\n", (int)i);
103                 ext4_dbg(DEBUG_MBR, "\tstatus: 0x%x\n", pe->status);
104                 ext4_dbg(DEBUG_MBR, "\ttype 0x%x:\n", pe->type);
105                 ext4_dbg(DEBUG_MBR, "\tfirst_lba: 0x%"PRIx32"\n", pe->first_lba);
106                 ext4_dbg(DEBUG_MBR, "\tsectors: 0x%"PRIx32"\n", pe->sectors);
107
108                 if (!pe->sectors)
109                         continue; /*Empty entry*/
110
111                 if (pe->type != 0x83)
112                         continue; /*Unsupported entry. 0x83 - linux native*/
113
114                 bdevs->partitions[i].bdif = parent->bdif;
115                 bdevs->partitions[i].part_offset =
116                         (uint64_t)pe->first_lba * parent->bdif->ph_bsize;
117                 bdevs->partitions[i].part_size =
118                         (uint64_t)pe->sectors * parent->bdif->ph_bsize;
119         }
120
121         blockdev_fini:
122         ext4_block_fini(parent);
123         return r;
124 }
125
126 /**
127  * @}
128  */
129