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  */
bcpl_strcpy(char * to,char * from)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  */
bstr_print(char * string)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  */
sum_block(struct block_header * header)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  */
print_disk_type(u32 disk_type)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  */
print_part_info(struct partition_block * p)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  */
get_rdisk(struct blk_desc * dev_desc)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 
get_bootcode(struct blk_desc * dev_desc)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  */
part_test_amiga(struct blk_desc * dev_desc)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  */
find_partition(struct blk_desc * dev_desc,int partnum)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  */
part_get_info_amiga(struct blk_desc * dev_desc,int part,disk_partition_t * info)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 
part_print_amiga(struct blk_desc * dev_desc)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
390