Merge tag 'efi-2019-10-rc1' of https://gitlab.denx.de/u-boot/custodians/u-boot-efi
[oweals/u-boot.git] / disk / part_amiga.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2001
4  * Hans-Joerg Frieden, Hyperion Entertainment
5  * Hans-JoergF@hyperion-entertainment.com
6  */
7 #include <common.h>
8 #include <command.h>
9 #include <ide.h>
10 #include "part_amiga.h"
11
12 #ifdef CONFIG_HAVE_BLOCK_DEVICE
13
14 #undef AMIGA_DEBUG
15
16 #ifdef AMIGA_DEBUG
17 #define PRINTF(fmt, args...) printf(fmt ,##args)
18 #else
19 #define PRINTF(fmt, args...)
20 #endif
21
22 struct block_header
23 {
24     u32 id;
25     u32 summed_longs;
26     s32 chk_sum;
27 };
28
29 static unsigned char block_buffer[DEFAULT_SECTOR_SIZE];
30 static struct rigid_disk_block rdb = {0};
31 static struct bootcode_block bootcode = {0};
32
33 /*
34  * Copy a bcpl to a c string
35  */
36 static void bcpl_strcpy(char *to, char *from)
37 {
38     int len = *from++;
39
40     while (len)
41     {
42         *to++ = *from++;
43         len--;
44     }
45     *to = 0;
46 }
47
48 /*
49  * Print a BCPL String. BCPL strings start with a byte with the length
50  * of the string, and don't contain a terminating nul character
51  */
52 static void bstr_print(char *string)
53 {
54     int len = *string++;
55     char buffer[256];
56     int i;
57
58     i = 0;
59     while (len)
60     {
61         buffer[i++] = *string++;
62         len--;
63     }
64
65     buffer[i] = 0;
66     printf("%-10s", buffer);
67 }
68
69 /*
70  * Sum a block. The checksum of a block must end up at zero
71  * to be valid. The chk_sum field is selected so that adding
72  * it yields zero.
73  */
74 int sum_block(struct block_header *header)
75 {
76     s32 *block = (s32 *)header;
77     u32 i;
78     s32 sum = 0;
79
80     for (i = 0; i < header->summed_longs; i++)
81         sum += *block++;
82
83     return (sum != 0);
84 }
85
86 /*
87  * Print an AmigaOS disk type. Disk types are a four-byte identifier
88  * describing the file system. They are usually written as a three-letter
89  * word followed by a backslash and a version number. For example,
90  * DOS\0 would be the original file system. SFS\0 would be SmartFileSystem.
91  * DOS\1 is FFS.
92  */
93 static void print_disk_type(u32 disk_type)
94 {
95     char buffer[6];
96     buffer[0] = (disk_type & 0xFF000000)>>24;
97     buffer[1] = (disk_type & 0x00FF0000)>>16;
98     buffer[2] = (disk_type & 0x0000FF00)>>8;
99     buffer[3] = '\\';
100     buffer[4] = (disk_type & 0x000000FF) + '0';
101     buffer[5] = 0;
102     printf("%s", buffer);
103 }
104
105 /*
106  * Print the info contained within the given partition block
107  */
108 static void print_part_info(struct partition_block *p)
109 {
110     struct amiga_part_geometry *g;
111
112     g = (struct amiga_part_geometry *)&(p->environment);
113
114     bstr_print(p->drive_name);
115     printf("%6d\t%6d\t",
116            g->low_cyl * g->block_per_track * g->surfaces ,
117            (g->high_cyl - g->low_cyl + 1) * g->block_per_track * g->surfaces - 1);
118     print_disk_type(g->dos_type);
119     printf("\t%5d\n", g->boot_priority);
120 }
121
122 /*
123  * Search for the Rigid Disk Block. The rigid disk block is required
124  * to be within the first 16 blocks of a drive, needs to have
125  * the ID AMIGA_ID_RDISK ('RDSK') and needs to have a valid
126  * sum-to-zero checksum
127  */
128 struct rigid_disk_block *get_rdisk(struct blk_desc *dev_desc)
129 {
130     int i;
131     int limit;
132     char *s;
133
134     s = env_get("amiga_scanlimit");
135     if (s)
136         limit = simple_strtoul(s, NULL, 10);
137     else
138         limit = AMIGA_BLOCK_LIMIT;
139
140     for (i=0; i<limit; i++)
141     {
142         ulong res = blk_dread(dev_desc, i, 1, (ulong *)block_buffer);
143         if (res == 1)
144         {
145             struct rigid_disk_block *trdb = (struct rigid_disk_block *)block_buffer;
146             if (trdb->id == AMIGA_ID_RDISK)
147             {
148                 PRINTF("Rigid disk block suspect at %d, checking checksum\n",i);
149                 if (sum_block((struct block_header *)block_buffer) == 0)
150                 {
151                     PRINTF("FOUND\n");
152                     memcpy(&rdb, trdb, sizeof(struct rigid_disk_block));
153                     return (struct rigid_disk_block *)&rdb;
154                 }
155             }
156         }
157     }
158     PRINTF("Done scanning, no RDB found\n");
159     return NULL;
160 }
161
162 /*
163  * Search for boot code
164  * Again, the first boot block must be located somewhere in the first 16 blocks, or rooted in the
165  * Ridgid disk block
166  */
167
168 struct bootcode_block *get_bootcode(struct blk_desc *dev_desc)
169 {
170     int i;
171     int limit;
172     char *s;
173
174     s = env_get("amiga_scanlimit");
175     if (s)
176         limit = simple_strtoul(s, NULL, 10);
177     else
178         limit = AMIGA_BLOCK_LIMIT;
179
180     PRINTF("Scanning for BOOT from 0 to %d\n", limit);
181
182     for (i = 0; i < limit; i++)
183     {
184         ulong res = blk_dread(dev_desc, i, 1, (ulong *)block_buffer);
185         if (res == 1)
186         {
187             struct bootcode_block *boot = (struct bootcode_block *)block_buffer;
188             if (boot->id == AMIGA_ID_BOOT)
189             {
190                 PRINTF("BOOT block at %d, checking checksum\n", i);
191                 if (sum_block((struct block_header *)block_buffer) == 0)
192                 {
193                     PRINTF("Found valid bootcode block\n");
194                     memcpy(&bootcode, boot, sizeof(struct bootcode_block));
195                     return &bootcode;
196                 }
197             }
198         }
199     }
200
201     PRINTF("No boot code found on disk\n");
202     return 0;
203 }
204
205 /*
206  * Test if the given partition has an Amiga partition table/Rigid
207  * Disk block
208  */
209 static int part_test_amiga(struct blk_desc *dev_desc)
210 {
211     struct rigid_disk_block *rdb;
212     struct bootcode_block *bootcode;
213
214     PRINTF("part_test_amiga: Testing for an Amiga RDB partition\n");
215
216     rdb = get_rdisk(dev_desc);
217     if (rdb)
218     {
219         bootcode = get_bootcode(dev_desc);
220         if (bootcode)
221             PRINTF("part_test_amiga: bootable Amiga disk\n");
222         else
223             PRINTF("part_test_amiga: non-bootable Amiga disk\n");
224
225         return 0;
226     }
227     else
228     {
229         PRINTF("part_test_amiga: no RDB found\n");
230         return -1;
231     }
232
233 }
234
235 /*
236  * Find partition number partnum on the given drive.
237  */
238 static struct partition_block *find_partition(struct blk_desc *dev_desc,
239                                               int partnum)
240 {
241     struct rigid_disk_block *rdb;
242     struct partition_block *p;
243     u32 block;
244
245     PRINTF("Trying to find partition block %d\n", partnum);
246     rdb = get_rdisk(dev_desc);
247     if (!rdb)
248     {
249         PRINTF("find_partition: no rdb found\n");
250         return NULL;
251     }
252
253     PRINTF("find_partition: Scanning partition list\n");
254
255     block = rdb->partition_list;
256     PRINTF("find_partition: partition list at 0x%x\n", block);
257
258     while (block != 0xFFFFFFFF)
259     {
260         ulong res = blk_dread(dev_desc, block, 1, (ulong *)block_buffer);
261         if (res == 1)
262         {
263             p = (struct partition_block *)block_buffer;
264             if (p->id == AMIGA_ID_PART)
265             {
266                 PRINTF("PART block suspect at 0x%x, checking checksum\n",block);
267                 if (sum_block((struct block_header *)p) == 0)
268                 {
269                     if (partnum == 0) break;
270                     else
271                     {
272                         partnum--;
273                         block = p->next;
274                     }
275                 }
276             } else block = 0xFFFFFFFF;
277         } else block = 0xFFFFFFFF;
278     }
279
280     if (block == 0xFFFFFFFF)
281     {
282         PRINTF("PART block not found\n");
283         return NULL;
284     }
285
286     return (struct partition_block *)block_buffer;
287 }
288
289 /*
290  * Get info about a partition
291  */
292 static int part_get_info_amiga(struct blk_desc *dev_desc, int part,
293                                     disk_partition_t *info)
294 {
295     struct partition_block *p = find_partition(dev_desc, part-1);
296     struct amiga_part_geometry *g;
297     u32 disk_type;
298
299     if (!p) return -1;
300
301     g = (struct amiga_part_geometry *)&(p->environment);
302     info->start = g->low_cyl  * g->block_per_track * g->surfaces;
303     info->size  = (g->high_cyl - g->low_cyl + 1) * g->block_per_track * g->surfaces - 1;
304     info->blksz = rdb.block_bytes;
305     bcpl_strcpy((char *)info->name, p->drive_name);
306
307
308     disk_type = g->dos_type;
309
310     info->type[0] = (disk_type & 0xFF000000)>>24;
311     info->type[1] = (disk_type & 0x00FF0000)>>16;
312     info->type[2] = (disk_type & 0x0000FF00)>>8;
313     info->type[3] = '\\';
314     info->type[4] = (disk_type & 0x000000FF) + '0';
315     info->type[5] = 0;
316
317     return 0;
318 }
319
320 static void part_print_amiga(struct blk_desc *dev_desc)
321 {
322     struct rigid_disk_block *rdb;
323     struct bootcode_block *boot;
324     struct partition_block *p;
325     u32 block;
326     int i = 1;
327
328     rdb = get_rdisk(dev_desc);
329     if (!rdb)
330     {
331         PRINTF("part_print_amiga: no rdb found\n");
332         return;
333     }
334
335     PRINTF("part_print_amiga: Scanning partition list\n");
336
337     block = rdb->partition_list;
338     PRINTF("part_print_amiga: partition list at 0x%x\n", block);
339
340     printf("Summary:  DiskBlockSize: %d\n"
341            "          Cylinders    : %d\n"
342            "          Sectors/Track: %d\n"
343            "          Heads        : %d\n\n",
344            rdb->block_bytes, rdb->cylinders, rdb->sectors,
345            rdb->heads);
346
347     printf("                 First   Num. \n"
348            "Nr.  Part. Name  Block   Block  Type        Boot Priority\n");
349
350     while (block != 0xFFFFFFFF)
351     {
352         ulong res;
353
354         PRINTF("Trying to load block #0x%X\n", block);
355
356         res = blk_dread(dev_desc, block, 1, (ulong *)block_buffer);
357         if (res == 1)
358         {
359             p = (struct partition_block *)block_buffer;
360             if (p->id == AMIGA_ID_PART)
361             {
362                 PRINTF("PART block suspect at 0x%x, checking checksum\n",block);
363                 if (sum_block((struct block_header *)p) == 0)
364                 {
365                     printf("%-4d ", i); i++;
366                     print_part_info(p);
367                     block = p->next;
368                 }
369             } else block = 0xFFFFFFFF;
370         } else block = 0xFFFFFFFF;
371     }
372
373     boot = get_bootcode(dev_desc);
374     if (boot)
375     {
376         printf("Disk is bootable\n");
377     }
378 }
379
380 U_BOOT_PART_TYPE(amiga) = {
381         .name           = "AMIGA",
382         .part_type      = PART_TYPE_AMIGA,
383         .max_entries    = AMIGA_ENTRY_NUMBERS,
384         .get_info       = part_get_info_amiga,
385         .print          = part_print_amiga,
386         .test           = part_test_amiga,
387 };
388
389 #endif