Initial revision
[platform/kernel/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 <cmd_disk.h>
37 #include "part_dos.h"
38
39 #if ((CONFIG_COMMANDS & CFG_CMD_IDE) || (CONFIG_COMMANDS & CFG_CMD_SCSI)) && defined(CONFIG_DOS_PARTITION)
40
41 /* Convert char[4] in little endian format to the host format integer
42  */
43 static inline int le32_to_int(unsigned char *le32)
44 {
45     return ((le32[3] << 24) +
46             (le32[2] << 16) +
47             (le32[1] << 8) +
48              le32[0]
49            );
50 }
51
52 static inline int is_extended(int part_type)
53 {
54     return (part_type == 0x5 ||
55             part_type == 0xf ||
56             part_type == 0x85);
57 }
58
59 static void print_one_part (dos_partition_t *p, int ext_part_sector, int part_num)
60 {
61         int lba_start = ext_part_sector + le32_to_int (p->start4);
62         int lba_size  = le32_to_int (p->size4);
63
64         printf ("%5d\t\t%10d\t%10d\t%2x%s\n",
65                 part_num, lba_start, lba_size, p->sys_ind,
66                 (is_extended (p->sys_ind) ? " Extd" : ""));
67 }
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 && part_num == which_part) {
173                         info->blksz = 512;
174                         info->start = ext_part_sector + le32_to_int (pt->start4);
175                         info->size  = le32_to_int (pt->size4);
176                         switch(dev_desc->if_type) {
177                                 case IF_TYPE_IDE:
178                                 case IF_TYPE_ATAPI:
179                                         sprintf (info->name, "hd%c%d\n", 'a' + dev_desc->dev, part_num);
180                                         break;
181                                 case IF_TYPE_SCSI:
182                                         sprintf (info->name, "sd%c%d\n", 'a' + dev_desc->dev, part_num);
183                                         break;
184                                 case IF_TYPE_USB:
185                                         sprintf (info->name, "usbd%c%d\n", 'a' + dev_desc->dev, part_num);
186                                         break;
187                                 case IF_TYPE_DOC:
188                                         sprintf (info->name, "docd%c%d\n", 'a' + dev_desc->dev, part_num);
189                                         break;
190                                 default:
191                                         sprintf (info->name, "xx%c%d\n", 'a' + dev_desc->dev, part_num);
192                                         break;
193                         }
194                         /* sprintf(info->type, "%d, pt->sys_ind); */
195                         sprintf (info->type, "U-Boot");
196                         return 0;
197                 }
198
199                 /* Reverse engr the fdisk part# assignment rule! */
200                 if ((ext_part_sector == 0) ||
201                     (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
202                         part_num++;
203                 }
204         }
205
206         /* Follows the extended partitions */
207         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
208         for (i = 0; i < 4; i++, pt++) {
209                 if (is_extended (pt->sys_ind)) {
210                         int lba_start = le32_to_int (pt->start4) + relative;
211
212                         return get_partition_info_extended (dev_desc, lba_start,
213                                  ext_part_sector == 0 ? lba_start : relative,
214                                  part_num, which_part, info);
215                 }
216         }
217         return -1;
218 }
219
220 void print_part_dos (block_dev_desc_t *dev_desc)
221 {
222         printf ("Partition     Start Sector     Num Sectors     Type\n");
223         print_partition_extended (dev_desc, 0, 0, 1);
224 }
225
226 int get_partition_info_dos (block_dev_desc_t *dev_desc, int part, disk_partition_t * info)
227 {
228         return get_partition_info_extended (dev_desc, 0, 0, 1, part, info);
229 }
230
231
232
233 #endif  /* (CONFIG_COMMANDS & CFG_CMD_IDE) && CONFIG_DOS_PARTITION */