* Code cleanup:
[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 <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 int test_part_dos (block_dev_desc_t *dev_desc)
71 {
72         unsigned char buffer[DEFAULT_SECTOR_SIZE];
73
74         if ((dev_desc->block_read(dev_desc->dev, 0, 1, (ulong *) buffer) != 1) ||
75             (buffer[DOS_PART_MAGIC_OFFSET + 0] != 0x55) ||
76             (buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) ) {
77                 return (-1);
78         }
79         return (0);
80 }
81
82 /*  Print a partition that is relative to its Extended partition table
83  */
84 static void print_partition_extended (block_dev_desc_t *dev_desc, int ext_part_sector, int relative,
85                                                            int part_num)
86 {
87         unsigned char buffer[DEFAULT_SECTOR_SIZE];
88         dos_partition_t *pt;
89         int i;
90
91         if (dev_desc->block_read(dev_desc->dev, ext_part_sector, 1, (ulong *) buffer) != 1) {
92                 printf ("** Can't read partition table on %d:%d **\n",
93                         dev_desc->dev, ext_part_sector);
94                 return;
95         }
96         if (buffer[DOS_PART_MAGIC_OFFSET] != 0x55 ||
97                 buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) {
98                 printf ("bad MBR sector signature 0x%02x%02x\n",
99                         buffer[DOS_PART_MAGIC_OFFSET],
100                         buffer[DOS_PART_MAGIC_OFFSET + 1]);
101                 return;
102         }
103
104         /* Print all primary/logical partitions */
105         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
106         for (i = 0; i < 4; i++, pt++) {
107                 /*
108                  * fdisk does not show the extended partitions that
109                  * are not in the MBR
110                  */
111
112                 if ((pt->sys_ind != 0) &&
113                     (ext_part_sector == 0 || !is_extended (pt->sys_ind)) ) {
114                         print_one_part (pt, ext_part_sector, part_num);
115                 }
116
117                 /* Reverse engr the fdisk part# assignment rule! */
118                 if ((ext_part_sector == 0) ||
119                     (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
120                         part_num++;
121                 }
122         }
123
124         /* Follows the extended partitions */
125         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
126         for (i = 0; i < 4; i++, pt++) {
127                 if (is_extended (pt->sys_ind)) {
128                         int lba_start = le32_to_int (pt->start4) + relative;
129
130                         print_partition_extended (dev_desc, lba_start,
131                                                   ext_part_sector == 0  ? lba_start
132                                                                         : relative,
133                                                   part_num);
134                 }
135         }
136
137         return;
138 }
139
140
141 /*  Print a partition that is relative to its Extended partition table
142  */
143 static int get_partition_info_extended (block_dev_desc_t *dev_desc, int ext_part_sector,
144                                  int relative, int part_num,
145                                  int which_part, disk_partition_t *info)
146 {
147         unsigned char buffer[DEFAULT_SECTOR_SIZE];
148         dos_partition_t *pt;
149         int i;
150
151         if (dev_desc->block_read (dev_desc->dev, ext_part_sector, 1, (ulong *) buffer) != 1) {
152                 printf ("** Can't read partition table on %d:%d **\n",
153                         dev_desc->dev, ext_part_sector);
154                 return -1;
155         }
156         if (buffer[DOS_PART_MAGIC_OFFSET] != 0x55 ||
157                 buffer[DOS_PART_MAGIC_OFFSET + 1] != 0xaa) {
158                 printf ("bad MBR sector signature 0x%02x%02x\n",
159                         buffer[DOS_PART_MAGIC_OFFSET],
160                         buffer[DOS_PART_MAGIC_OFFSET + 1]);
161                 return -1;
162         }
163
164         /* Print all primary/logical partitions */
165         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
166         for (i = 0; i < 4; i++, pt++) {
167                 /*
168                  * fdisk does not show the extended partitions that
169                  * are not in the MBR
170                  */
171                 if ((pt->sys_ind != 0) &&
172                     (part_num == which_part) &&
173                     (is_extended(pt->sys_ind) == 0)) {
174                         info->blksz = 512;
175                         info->start = ext_part_sector + le32_to_int (pt->start4);
176                         info->size  = le32_to_int (pt->size4);
177                         switch(dev_desc->if_type) {
178                                 case IF_TYPE_IDE:
179                                 case IF_TYPE_ATAPI:
180                                         sprintf (info->name, "hd%c%d\n", 'a' + dev_desc->dev, part_num);
181                                         break;
182                                 case IF_TYPE_SCSI:
183                                         sprintf (info->name, "sd%c%d\n", 'a' + dev_desc->dev, part_num);
184                                         break;
185                                 case IF_TYPE_USB:
186                                         sprintf (info->name, "usbd%c%d\n", 'a' + dev_desc->dev, part_num);
187                                         break;
188                                 case IF_TYPE_DOC:
189                                         sprintf (info->name, "docd%c%d\n", 'a' + dev_desc->dev, part_num);
190                                         break;
191                                 default:
192                                         sprintf (info->name, "xx%c%d\n", 'a' + dev_desc->dev, part_num);
193                                         break;
194                         }
195                         /* sprintf(info->type, "%d, pt->sys_ind); */
196                         sprintf (info->type, "U-Boot");
197                         return 0;
198                 }
199
200                 /* Reverse engr the fdisk part# assignment rule! */
201                 if ((ext_part_sector == 0) ||
202                     (pt->sys_ind != 0 && !is_extended (pt->sys_ind)) ) {
203                         part_num++;
204                 }
205         }
206
207         /* Follows the extended partitions */
208         pt = (dos_partition_t *) (buffer + DOS_PART_TBL_OFFSET);
209         for (i = 0; i < 4; i++, pt++) {
210                 if (is_extended (pt->sys_ind)) {
211                         int lba_start = le32_to_int (pt->start4) + relative;
212
213                         return get_partition_info_extended (dev_desc, lba_start,
214                                  ext_part_sector == 0 ? lba_start : relative,
215                                  part_num, which_part, info);
216                 }
217         }
218         return -1;
219 }
220
221 void print_part_dos (block_dev_desc_t *dev_desc)
222 {
223         printf ("Partition     Start Sector     Num Sectors     Type\n");
224         print_partition_extended (dev_desc, 0, 0, 1);
225 }
226
227 int get_partition_info_dos (block_dev_desc_t *dev_desc, int part, disk_partition_t * info)
228 {
229         return get_partition_info_extended (dev_desc, 0, 0, 1, part, info);
230 }
231
232
233 #endif  /* (CONFIG_COMMANDS & CFG_CMD_IDE) && CONFIG_DOS_PARTITION */