* Add support for USB Mass Storage Devices (BBB)
[oweals/u-boot.git] / disk / part_dos.c
1 /*
2  * (C) Copyright 2001
3  * Raymond Lo, lo@routefree.com
4  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
5  *
6  * See file CREDITS for list of people who contributed to this
7  * project.
8  *
9  * This program is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU General Public License as
11  * published by the Free Software Foundation; either version 2 of
12  * the License, or (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
22  * MA 02111-1307 USA
23  */
24
25 /*
26  * Support for harddisk partitions.
27  *
28  * To be compatible with LinuxPPC and Apple we use the standard Apple
29  * SCSI disk partitioning scheme. For more information see:
30  * http://developer.apple.com/techpubs/mac/Devices/Devices-126.html#MARKER-14-92
31  */
32
33 #include <common.h>
34 #include <command.h>
35 #include <ide.h>
36 #include "part_dos.h"
37
38 #if ((CONFIG_COMMANDS & CFG_CMD_IDE)    || \
39      (CONFIG_COMMANDS & CFG_CMD_SCSI)   || \
40      (CONFIG_COMMANDS & CFG_CMD_USB)    ) && defined(CONFIG_DOS_PARTITION)
41
42 /* Convert char[4] in little endian format to the host format integer
43  */
44 static inline int le32_to_int(unsigned char *le32)
45 {
46     return ((le32[3] << 24) +
47             (le32[2] << 16) +
48             (le32[1] << 8) +
49              le32[0]
50            );
51 }
52
53 static inline int is_extended(int part_type)
54 {
55     return (part_type == 0x5 ||
56             part_type == 0xf ||
57             part_type == 0x85);
58 }
59
60 static void print_one_part (dos_partition_t *p, int ext_part_sector, int part_num)
61 {
62         int lba_start = ext_part_sector + le32_to_int (p->start4);
63         int lba_size  = le32_to_int (p->size4);
64
65         printf ("%5d\t\t%10d\t%10d\t%2x%s\n",
66                 part_num, lba_start, lba_size, p->sys_ind,
67                 (is_extended (p->sys_ind) ? " Extd" : ""));
68 }
69
70
71 int test_part_dos (block_dev_desc_t *dev_desc)
72 {
73         unsigned char buffer[DEFAULT_SECTOR_SIZE];
74
75         if ((dev_desc->block_read(dev_desc->dev, 0, 1, (ulong *) buffer) != 1) ||
76             (buffer[DOS_PART_MAGIC_OFFSET + 0] != 0x55) ||
77             (buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) ) {
78                 return (-1);
79         }
80         return (0);
81 }
82
83 /*  Print a partition that is relative to its Extended partition table
84  */
85 static void print_partition_extended (block_dev_desc_t *dev_desc, int ext_part_sector, int relative,
86                                                            int part_num)
87 {
88         unsigned char buffer[DEFAULT_SECTOR_SIZE];
89         dos_partition_t *pt;
90         int i;
91
92         if (dev_desc->block_read(dev_desc->dev, ext_part_sector, 1, (ulong *) buffer) != 1) {
93                 printf ("** Can't read partition table on %d:%d **\n",
94                         dev_desc->dev, ext_part_sector);
95                 return;
96         }
97         if (buffer[DOS_PART_MAGIC_OFFSET] != 0x55 ||
98                 buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) {
99                 printf ("bad MBR sector signature 0x%02x%02x\n",
100                         buffer[DOS_PART_MAGIC_OFFSET],
101                         buffer[DOS_PART_MAGIC_OFFSET + 1]);
102                 return;
103         }
104
105         /* Print all primary/logical partitions */
106         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
107         for (i = 0; i < 4; i++, pt++) {
108                 /*
109                  * fdisk does not show the extended partitions that
110                  * are not in the MBR
111                  */
112
113                 if ((pt->sys_ind != 0) &&
114                     (ext_part_sector == 0 || !is_extended (pt->sys_ind)) ) {
115                         print_one_part (pt, ext_part_sector, part_num);
116                 }
117
118                 /* Reverse engr the fdisk part# assignment rule! */
119                 if ((ext_part_sector == 0) ||
120                     (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
121                         part_num++;
122                 }
123         }
124
125         /* Follows the extended partitions */
126         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
127         for (i = 0; i < 4; i++, pt++) {
128                 if (is_extended (pt->sys_ind)) {
129                         int lba_start = le32_to_int (pt->start4) + relative;
130
131                         print_partition_extended (dev_desc, lba_start,
132                                                   ext_part_sector == 0  ? lba_start
133                                                                         : relative,
134                                                   part_num);
135                 }
136         }
137
138         return;
139 }
140
141
142 /*  Print a partition that is relative to its Extended partition table
143  */
144 static int get_partition_info_extended (block_dev_desc_t *dev_desc, int ext_part_sector,
145                                  int relative, int part_num,
146                                  int which_part, disk_partition_t *info)
147 {
148         unsigned char buffer[DEFAULT_SECTOR_SIZE];
149         dos_partition_t *pt;
150         int i;
151
152         if (dev_desc->block_read (dev_desc->dev, ext_part_sector, 1, (ulong *) buffer) != 1) {
153                 printf ("** Can't read partition table on %d:%d **\n",
154                         dev_desc->dev, ext_part_sector);
155                 return -1;
156         }
157         if (buffer[DOS_PART_MAGIC_OFFSET] != 0x55 ||
158                 buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) {
159                 printf ("bad MBR sector signature 0x%02x%02x\n",
160                         buffer[DOS_PART_MAGIC_OFFSET],
161                         buffer[DOS_PART_MAGIC_OFFSET + 1]);
162                 return -1;
163         }
164
165         /* Print all primary/logical partitions */
166         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
167         for (i = 0; i < 4; i++, pt++) {
168                 /*
169                  * fdisk does not show the extended partitions that
170                  * are not in the MBR
171                  */
172                 if ((pt->sys_ind != 0) &&
173                     (part_num == which_part) &&
174                     (is_extended(pt->sys_ind) == 0)) {
175                         info->blksz = 512;
176                         info->start = ext_part_sector + le32_to_int (pt->start4);
177                         info->size  = le32_to_int (pt->size4);
178                         switch(dev_desc->if_type) {
179                                 case IF_TYPE_IDE:
180                                 case IF_TYPE_ATAPI:
181                                         sprintf (info->name, "hd%c%d\n", 'a' + dev_desc->dev, part_num);
182                                         break;
183                                 case IF_TYPE_SCSI:
184                                         sprintf (info->name, "sd%c%d\n", 'a' + dev_desc->dev, part_num);
185                                         break;
186                                 case IF_TYPE_USB:
187                                         sprintf (info->name, "usbd%c%d\n", 'a' + dev_desc->dev, part_num);
188                                         break;
189                                 case IF_TYPE_DOC:
190                                         sprintf (info->name, "docd%c%d\n", 'a' + dev_desc->dev, part_num);
191                                         break;
192                                 default:
193                                         sprintf (info->name, "xx%c%d\n", 'a' + dev_desc->dev, part_num);
194                                         break;
195                         }
196                         /* sprintf(info->type, "%d, pt->sys_ind); */
197                         sprintf (info->type, "U-Boot");
198                         return 0;
199                 }
200
201                 /* Reverse engr the fdisk part# assignment rule! */
202                 if ((ext_part_sector == 0) ||
203                     (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
204                         part_num++;
205                 }
206         }
207
208         /* Follows the extended partitions */
209         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
210         for (i = 0; i < 4; i++, pt++) {
211                 if (is_extended (pt->sys_ind)) {
212                         int lba_start = le32_to_int (pt->start4) + relative;
213
214                         return get_partition_info_extended (dev_desc, lba_start,
215                                  ext_part_sector == 0 ? lba_start : relative,
216                                  part_num, which_part, info);
217                 }
218         }
219         return -1;
220 }
221
222 void print_part_dos (block_dev_desc_t *dev_desc)
223 {
224         printf ("Partition     Start Sector     Num Sectors     Type\n");
225         print_partition_extended (dev_desc, 0, 0, 1);
226 }
227
228 int get_partition_info_dos (block_dev_desc_t *dev_desc, int part, disk_partition_t * info)
229 {
230         return get_partition_info_extended (dev_desc, 0, 0, 1, part, info);
231 }
232
233
234 #endif  /* (CONFIG_COMMANDS & CFG_CMD_IDE) && CONFIG_DOS_PARTITION */