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