]> git.kernelconcepts.de Git - karo-tx-uboot.git/blob - disk/part.c
disk/part.c: 'usb storage' avoiding overflow when output capacity
[karo-tx-uboot.git] / disk / part.c
1 /*
2  * (C) Copyright 2001
3  * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
4  *
5  * See file CREDITS for list of people who contributed to this
6  * project.
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU General Public License as
10  * published by the Free Software Foundation; either version 2 of
11  * the License, or (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 59 Temple Place, Suite 330, Boston,
21  * MA 02111-1307 USA
22  */
23
24 #include <common.h>
25 #include <command.h>
26 #include <ide.h>
27 #include <part.h>
28
29 #undef  PART_DEBUG
30
31 #ifdef  PART_DEBUG
32 #define PRINTF(fmt,args...)     printf (fmt ,##args)
33 #else
34 #define PRINTF(fmt,args...)
35 #endif
36
37 #if (defined(CONFIG_CMD_IDE) || \
38      defined(CONFIG_CMD_MG_DISK) || \
39      defined(CONFIG_CMD_SATA) || \
40      defined(CONFIG_CMD_SCSI) || \
41      defined(CONFIG_CMD_USB) || \
42      defined(CONFIG_MMC) || \
43      defined(CONFIG_SYSTEMACE) )
44
45 struct block_drvr {
46         char *name;
47         block_dev_desc_t* (*get_dev)(int dev);
48 };
49
50 static const struct block_drvr block_drvr[] = {
51 #if defined(CONFIG_CMD_IDE)
52         { .name = "ide", .get_dev = ide_get_dev, },
53 #endif
54 #if defined(CONFIG_CMD_SATA)
55         {.name = "sata", .get_dev = sata_get_dev, },
56 #endif
57 #if defined(CONFIG_CMD_SCSI)
58         { .name = "scsi", .get_dev = scsi_get_dev, },
59 #endif
60 #if defined(CONFIG_CMD_USB) && defined(CONFIG_USB_STORAGE)
61         { .name = "usb", .get_dev = usb_stor_get_dev, },
62 #endif
63 #if defined(CONFIG_MMC)
64         { .name = "mmc", .get_dev = mmc_get_dev, },
65 #endif
66 #if defined(CONFIG_SYSTEMACE)
67         { .name = "ace", .get_dev = systemace_get_dev, },
68 #endif
69 #if defined(CONFIG_CMD_MG_DISK)
70         { .name = "mgd", .get_dev = mg_disk_get_dev, },
71 #endif
72         { },
73 };
74
75 DECLARE_GLOBAL_DATA_PTR;
76
77 block_dev_desc_t *get_dev(char* ifname, int dev)
78 {
79         const struct block_drvr *drvr = block_drvr;
80         block_dev_desc_t* (*reloc_get_dev)(int dev);
81
82         while (drvr->name) {
83                 reloc_get_dev = drvr->get_dev;
84 #ifndef CONFIG_RELOC_FIXUP_WORKS
85                 reloc_get_dev += gd->reloc_off;
86 #endif
87                 if (strncmp(ifname, drvr->name, strlen(drvr->name)) == 0)
88                         return reloc_get_dev(dev);
89                 drvr++;
90         }
91         return NULL;
92 }
93 #else
94 block_dev_desc_t *get_dev(char* ifname, int dev)
95 {
96         return NULL;
97 }
98 #endif
99
100 #if (defined(CONFIG_CMD_IDE) || \
101      defined(CONFIG_CMD_MG_DISK) || \
102      defined(CONFIG_CMD_SATA) || \
103      defined(CONFIG_CMD_SCSI) || \
104      defined(CONFIG_CMD_USB) || \
105      defined(CONFIG_MMC) || \
106      defined(CONFIG_SYSTEMACE) )
107
108 /* ------------------------------------------------------------------------- */
109 /*
110  * reports device info to the user
111  */
112
113 #ifdef CONFIG_LBA48
114 typedef uint64_t lba512_t;
115 #else
116 typedef lbaint_t lba512_t;
117 #endif
118
119 /*
120  * Overflowless variant of (block_count * mul_by / div_by)
121  * when div_by > mul_by
122  */
123 static lba512_t lba512_muldiv (lba512_t block_count, lba512_t mul_by, lba512_t div_by)
124 {
125         lba512_t bc_quot, bc_rem;
126
127         /* x * m / d == x / d * m + (x % d) * m / d */
128         bc_quot = block_count / div_by;
129         bc_rem  = block_count - div_by * bc_quot;
130         return bc_quot * mul_by + (bc_rem * mul_by) / div_by;
131 }
132
133 void dev_print (block_dev_desc_t *dev_desc)
134 {
135         lba512_t lba512; /* number of blocks if 512bytes block size */
136
137         if (dev_desc->type == DEV_TYPE_UNKNOWN) {
138                 puts ("not available\n");
139                 return;
140         }
141
142         switch (dev_desc->if_type) {
143         case IF_TYPE_SCSI:
144                 printf ("(%d:%d) Vendor: %s Prod.: %s Rev: %s\n",
145                         dev_desc->target,dev_desc->lun,
146                         dev_desc->vendor,
147                         dev_desc->product,
148                         dev_desc->revision);
149                 break;
150         case IF_TYPE_ATAPI:
151         case IF_TYPE_IDE:
152         case IF_TYPE_SATA:
153                 printf ("Model: %s Firm: %s Ser#: %s\n",
154                         dev_desc->vendor,
155                         dev_desc->revision,
156                         dev_desc->product);
157                 break;
158         case IF_TYPE_SD:
159         case IF_TYPE_MMC:
160         case IF_TYPE_USB:
161                 printf ("Vendor: %s Rev: %s Prod: %s\n",
162                         dev_desc->vendor,
163                         dev_desc->revision,
164                         dev_desc->product);
165                 break;
166         case IF_TYPE_DOC:
167                 puts("device type DOC\n");
168                 return;
169         case IF_TYPE_UNKNOWN:
170                 puts("device type unknown\n");
171                 return;
172         default:
173                 printf("Unhandled device type: %i\n", dev_desc->if_type);
174                 return;
175         }
176         puts ("            Type: ");
177         if (dev_desc->removable)
178                 puts ("Removable ");
179         switch (dev_desc->type & 0x1F) {
180         case DEV_TYPE_HARDDISK:
181                 puts ("Hard Disk");
182                 break;
183         case DEV_TYPE_CDROM:
184                 puts ("CD ROM");
185                 break;
186         case DEV_TYPE_OPDISK:
187                 puts ("Optical Device");
188                 break;
189         case DEV_TYPE_TAPE:
190                 puts ("Tape");
191                 break;
192         default:
193                 printf ("# %02X #", dev_desc->type & 0x1F);
194                 break;
195         }
196         puts ("\n");
197         if ((dev_desc->lba * dev_desc->blksz)>0L) {
198                 ulong mb, mb_quot, mb_rem, gb, gb_quot, gb_rem;
199                 lbaint_t lba;
200
201                 lba = dev_desc->lba;
202
203                 lba512 = (lba * (dev_desc->blksz/512));
204                 /* round to 1 digit */
205                 mb = lba512_muldiv(lba512, 10, 2048);   /* 2048 = (1024 * 1024) / 512 MB */
206
207                 mb_quot = mb / 10;
208                 mb_rem  = mb - (10 * mb_quot);
209
210                 gb = mb / 1024;
211                 gb_quot = gb / 10;
212                 gb_rem  = gb - (10 * gb_quot);
213 #ifdef CONFIG_LBA48
214                 if (dev_desc->lba48)
215                         printf ("            Supports 48-bit addressing\n");
216 #endif
217 #if defined(CONFIG_SYS_64BIT_LBA)
218                 printf ("            Capacity: %ld.%ld MB = %ld.%ld GB (%Ld x %ld)\n",
219                         mb_quot, mb_rem,
220                         gb_quot, gb_rem,
221                         lba,
222                         dev_desc->blksz);
223 #else
224                 printf ("            Capacity: %ld.%ld MB = %ld.%ld GB (%ld x %ld)\n",
225                         mb_quot, mb_rem,
226                         gb_quot, gb_rem,
227                         (ulong)lba,
228                         dev_desc->blksz);
229 #endif
230         } else {
231                 puts ("            Capacity: not available\n");
232         }
233 }
234 #endif
235
236 #if (defined(CONFIG_CMD_IDE) || \
237      defined(CONFIG_CMD_MG_DISK) || \
238      defined(CONFIG_CMD_SATA) || \
239      defined(CONFIG_CMD_SCSI) || \
240      defined(CONFIG_CMD_USB) || \
241      defined(CONFIG_MMC)                || \
242      defined(CONFIG_SYSTEMACE) )
243
244 #if defined(CONFIG_MAC_PARTITION) || \
245     defined(CONFIG_DOS_PARTITION) || \
246     defined(CONFIG_ISO_PARTITION) || \
247     defined(CONFIG_AMIGA_PARTITION) || \
248     defined(CONFIG_EFI_PARTITION)
249
250 void init_part (block_dev_desc_t * dev_desc)
251 {
252 #ifdef CONFIG_ISO_PARTITION
253         if (test_part_iso(dev_desc) == 0) {
254                 dev_desc->part_type = PART_TYPE_ISO;
255                 return;
256         }
257 #endif
258
259 #ifdef CONFIG_MAC_PARTITION
260         if (test_part_mac(dev_desc) == 0) {
261                 dev_desc->part_type = PART_TYPE_MAC;
262                 return;
263         }
264 #endif
265
266 /* must be placed before DOS partition detection */
267 #ifdef CONFIG_EFI_PARTITION
268         if (test_part_efi(dev_desc) == 0) {
269                 dev_desc->part_type = PART_TYPE_EFI;
270                 return;
271         }
272 #endif
273
274 #ifdef CONFIG_DOS_PARTITION
275         if (test_part_dos(dev_desc) == 0) {
276                 dev_desc->part_type = PART_TYPE_DOS;
277                 return;
278         }
279 #endif
280
281 #ifdef CONFIG_AMIGA_PARTITION
282         if (test_part_amiga(dev_desc) == 0) {
283             dev_desc->part_type = PART_TYPE_AMIGA;
284             return;
285         }
286 #endif
287 }
288
289
290 int get_partition_info (block_dev_desc_t *dev_desc, int part
291                                         , disk_partition_t *info)
292 {
293         switch (dev_desc->part_type) {
294 #ifdef CONFIG_MAC_PARTITION
295         case PART_TYPE_MAC:
296                 if (get_partition_info_mac(dev_desc,part,info) == 0) {
297                         PRINTF ("## Valid MAC partition found ##\n");
298                         return (0);
299                 }
300                 break;
301 #endif
302
303 #ifdef CONFIG_DOS_PARTITION
304         case PART_TYPE_DOS:
305                 if (get_partition_info_dos(dev_desc,part,info) == 0) {
306                         PRINTF ("## Valid DOS partition found ##\n");
307                         return (0);
308                 }
309                 break;
310 #endif
311
312 #ifdef CONFIG_ISO_PARTITION
313         case PART_TYPE_ISO:
314                 if (get_partition_info_iso(dev_desc,part,info) == 0) {
315                         PRINTF ("## Valid ISO boot partition found ##\n");
316                         return (0);
317                 }
318                 break;
319 #endif
320
321 #ifdef CONFIG_AMIGA_PARTITION
322         case PART_TYPE_AMIGA:
323             if (get_partition_info_amiga(dev_desc, part, info) == 0)
324             {
325                 PRINTF ("## Valid Amiga partition found ##\n");
326                 return (0);
327             }
328             break;
329 #endif
330
331 #ifdef CONFIG_EFI_PARTITION
332         case PART_TYPE_EFI:
333                 if (get_partition_info_efi(dev_desc,part,info) == 0) {
334                         PRINTF ("## Valid EFI partition found ##\n");
335                         return (0);
336                 }
337                 break;
338 #endif
339         default:
340                 break;
341         }
342         return (-1);
343 }
344
345 static void print_part_header (const char *type, block_dev_desc_t * dev_desc)
346 {
347         puts ("\nPartition Map for ");
348         switch (dev_desc->if_type) {
349         case IF_TYPE_IDE:
350                 puts ("IDE");
351                 break;
352         case IF_TYPE_SATA:
353                 puts ("SATA");
354                 break;
355         case IF_TYPE_SCSI:
356                 puts ("SCSI");
357                 break;
358         case IF_TYPE_ATAPI:
359                 puts ("ATAPI");
360                 break;
361         case IF_TYPE_USB:
362                 puts ("USB");
363                 break;
364         case IF_TYPE_DOC:
365                 puts ("DOC");
366                 break;
367         default:
368                 puts ("UNKNOWN");
369                 break;
370         }
371         printf (" device %d  --   Partition Type: %s\n\n",
372                         dev_desc->dev, type);
373 }
374
375 void print_part (block_dev_desc_t * dev_desc)
376 {
377
378                 switch (dev_desc->part_type) {
379 #ifdef CONFIG_MAC_PARTITION
380         case PART_TYPE_MAC:
381                 PRINTF ("## Testing for valid MAC partition ##\n");
382                 print_part_header ("MAC", dev_desc);
383                 print_part_mac (dev_desc);
384                 return;
385 #endif
386 #ifdef CONFIG_DOS_PARTITION
387         case PART_TYPE_DOS:
388                 PRINTF ("## Testing for valid DOS partition ##\n");
389                 print_part_header ("DOS", dev_desc);
390                 print_part_dos (dev_desc);
391                 return;
392 #endif
393
394 #ifdef CONFIG_ISO_PARTITION
395         case PART_TYPE_ISO:
396                 PRINTF ("## Testing for valid ISO Boot partition ##\n");
397                 print_part_header ("ISO", dev_desc);
398                 print_part_iso (dev_desc);
399                 return;
400 #endif
401
402 #ifdef CONFIG_AMIGA_PARTITION
403         case PART_TYPE_AMIGA:
404             PRINTF ("## Testing for a valid Amiga partition ##\n");
405             print_part_header ("AMIGA", dev_desc);
406             print_part_amiga (dev_desc);
407             return;
408 #endif
409
410 #ifdef CONFIG_EFI_PARTITION
411         case PART_TYPE_EFI:
412                 PRINTF ("## Testing for valid EFI partition ##\n");
413                 print_part_header ("EFI", dev_desc);
414                 print_part_efi (dev_desc);
415                 return;
416 #endif
417         }
418         puts ("## Unknown partition table\n");
419 }
420
421
422 #else   /* neither MAC nor DOS nor ISO nor AMIGA nor EFI partition configured */
423 # error neither CONFIG_MAC_PARTITION nor CONFIG_DOS_PARTITION
424 # error nor CONFIG_ISO_PARTITION nor CONFIG_AMIGA_PARTITION
425 # error nor CONFIG_EFI_PARTITION configured!
426 #endif
427
428 #endif