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