mirror of
https://github.com/dmolony/DiskBrowser.git
synced 2024-09-27 09:55:28 +00:00
394 lines
15 KiB
Java
Executable File
394 lines
15 KiB
Java
Executable File
package com.bytezone.diskbrowser.prodos;
|
|
|
|
import static com.bytezone.diskbrowser.prodos.ProdosConstants.ENTRY_SIZE;
|
|
|
|
import java.awt.Color;
|
|
import java.time.format.DateTimeFormatter;
|
|
import java.util.ArrayList;
|
|
import java.util.Comparator;
|
|
import java.util.List;
|
|
|
|
import javax.swing.tree.DefaultMutableTreeNode;
|
|
import javax.swing.tree.DefaultTreeModel;
|
|
|
|
import com.bytezone.diskbrowser.applefile.AppleFileSource;
|
|
import com.bytezone.diskbrowser.applefile.BootSector;
|
|
import com.bytezone.diskbrowser.disk.AbstractFormattedDisk;
|
|
import com.bytezone.diskbrowser.disk.AppleDisk;
|
|
import com.bytezone.diskbrowser.disk.DefaultAppleFileSource;
|
|
import com.bytezone.diskbrowser.disk.DefaultSector;
|
|
import com.bytezone.diskbrowser.disk.Disk;
|
|
import com.bytezone.diskbrowser.disk.DiskAddress;
|
|
import com.bytezone.diskbrowser.disk.SectorType;
|
|
import com.bytezone.diskbrowser.gui.DataSource;
|
|
import com.bytezone.diskbrowser.gui.ProdosPreferences;
|
|
import com.bytezone.diskbrowser.utilities.HexFormatter;
|
|
import com.bytezone.diskbrowser.utilities.Utility;
|
|
|
|
// -----------------------------------------------------------------------------------//
|
|
public class ProdosDisk extends AbstractFormattedDisk
|
|
// -----------------------------------------------------------------------------------//
|
|
{
|
|
static ProdosPreferences prodosPreferences; // set by MenuHandler
|
|
|
|
static final DateTimeFormatter df = DateTimeFormatter.ofPattern ("d-LLL-yy");
|
|
static final DateTimeFormatter tf = DateTimeFormatter.ofPattern ("H:mm");
|
|
|
|
final SectorType dosSector = new SectorType ("Bootstrap Loader", Color.lightGray);
|
|
final SectorType catalogSector = new SectorType ("Catalog", new Color (0, 200, 0));
|
|
final SectorType volumeMapSector = new SectorType ("Volume Map", Color.blue);
|
|
final SectorType subcatalogSector = new SectorType ("Subcatalog", Color.magenta);
|
|
final SectorType masterIndexSector = new SectorType ("Master Index", Color.orange);
|
|
final SectorType indexSector = new SectorType ("Index", Color.cyan);
|
|
final SectorType dataSector = new SectorType ("Data", Color.red);
|
|
final SectorType extendedKeySector = new SectorType ("Extended key", Color.gray);
|
|
|
|
private final List<DirectoryHeader> headerEntries = new ArrayList<> ();
|
|
private final DefaultMutableTreeNode volumeNode;
|
|
private VolumeDirectoryHeader volumeDirectoryHeader;
|
|
|
|
private static final boolean debug = false;
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
public static void setProdosPreferences (ProdosPreferences prodosPreferences)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
ProdosDisk.prodosPreferences = prodosPreferences;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
public ProdosDisk (Disk disk)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
super (disk);
|
|
|
|
sectorTypesList.add (dosSector);
|
|
sectorTypesList.add (catalogSector);
|
|
sectorTypesList.add (subcatalogSector);
|
|
sectorTypesList.add (volumeMapSector);
|
|
sectorTypesList.add (masterIndexSector);
|
|
sectorTypesList.add (indexSector);
|
|
sectorTypesList.add (dataSector);
|
|
sectorTypesList.add (extendedKeySector);
|
|
|
|
for (int block = 0; block < 2; block++)
|
|
if (!disk.isBlockEmpty (disk.getDiskAddress (block)))
|
|
sectorTypes[block] = dosSector;
|
|
|
|
DiskAddress da = disk.getDiskAddress (0);
|
|
byte[] buffer = disk.readBlock (da);
|
|
bootSector = new BootSector (disk, buffer, "Prodos", da);
|
|
|
|
DefaultMutableTreeNode root = getCatalogTreeRoot ();
|
|
volumeNode = new DefaultMutableTreeNode ("empty volume node");
|
|
root.add (volumeNode);
|
|
|
|
processDirectoryBlock (2, null, volumeNode);
|
|
makeNodeVisible (volumeNode.getFirstLeaf ());
|
|
|
|
for (DiskAddress da2 : disk)
|
|
{
|
|
int blockNo = da2.getBlockNo ();
|
|
if (freeBlocks.get (blockNo))
|
|
{
|
|
if (!stillAvailable (da2))
|
|
falsePositives++;
|
|
}
|
|
else if (stillAvailable (da2))
|
|
falseNegatives++;
|
|
}
|
|
|
|
if (ProdosDisk.prodosPreferences.sortDirectories)
|
|
{
|
|
sortNodes (volumeNode);
|
|
((DefaultTreeModel) catalogTree.getModel ()).reload ();
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
private void processDirectoryBlock (int block, FileEntry parent,
|
|
DefaultMutableTreeNode parentNode)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
DirectoryHeader localHeader = null;
|
|
SectorType currentSectorType = null;
|
|
|
|
do
|
|
{
|
|
byte[] sectorBuffer = disk.readBlock (block);
|
|
if (!disk.isBlockEmpty (block))
|
|
sectorTypes[block] = currentSectorType;
|
|
|
|
int max = disk.getBlockSize () - ProdosConstants.ENTRY_SIZE;
|
|
for (int ptr = 4, entryNo = 0; ptr < max; ptr += ENTRY_SIZE, entryNo++)
|
|
{
|
|
int storageType = (sectorBuffer[ptr] & 0xF0) >> 4;
|
|
if (storageType == 0) // deleted or unused
|
|
continue;
|
|
|
|
byte[] entry = new byte[ProdosConstants.ENTRY_SIZE];
|
|
System.arraycopy (sectorBuffer, ptr, entry, 0, ProdosConstants.ENTRY_SIZE);
|
|
|
|
switch (storageType)
|
|
{
|
|
case ProdosConstants.VOLUME_HEADER:
|
|
assert headerEntries.size () == 0;
|
|
volumeDirectoryHeader = new VolumeDirectoryHeader (this, entry);
|
|
assert volumeDirectoryHeader.entryLength == ProdosConstants.ENTRY_SIZE;
|
|
headerEntries.add (volumeDirectoryHeader);
|
|
currentSectorType = catalogSector;
|
|
if (!disk.isBlockEmpty (block))
|
|
sectorTypes[block] = currentSectorType;
|
|
for (int i = 0; i < volumeDirectoryHeader.totalBitMapBlocks; i++)
|
|
sectorTypes[volumeDirectoryHeader.bitMapBlock + i] = volumeMapSector;
|
|
parentNode.setUserObject (volumeDirectoryHeader); // populate the empty volume node
|
|
localHeader = volumeDirectoryHeader;
|
|
break;
|
|
|
|
case ProdosConstants.SUBDIRECTORY_HEADER:
|
|
localHeader = new SubDirectoryHeader (this, entry, parent, block);
|
|
headerEntries.add (localHeader);
|
|
currentSectorType = subcatalogSector;
|
|
if (!disk.isBlockEmpty (block))
|
|
sectorTypes[block] = currentSectorType;
|
|
break;
|
|
|
|
case ProdosConstants.SUBDIRECTORY:
|
|
FileEntry fileEntry =
|
|
new FileEntry (this, entry, localHeader, block, entryNo);
|
|
fileEntries.add (fileEntry);
|
|
DefaultMutableTreeNode directoryNode = new DefaultMutableTreeNode (fileEntry);
|
|
directoryNode.setAllowsChildren (true);
|
|
parentNode.add (directoryNode);
|
|
processDirectoryBlock (fileEntry.keyPtr, fileEntry, directoryNode); // Recursion!!
|
|
break;
|
|
|
|
case ProdosConstants.SEEDLING:
|
|
case ProdosConstants.SAPLING:
|
|
case ProdosConstants.TREE:
|
|
case ProdosConstants.PASCAL_ON_PROFILE:
|
|
case ProdosConstants.GSOS_EXTENDED_FILE:
|
|
fileEntry = new FileEntry (this, entry, localHeader, block, entryNo);
|
|
fileEntries.add (fileEntry);
|
|
DefaultMutableTreeNode node = new DefaultMutableTreeNode (fileEntry);
|
|
node.setAllowsChildren (false);
|
|
parentNode.add (node);
|
|
break;
|
|
|
|
default:
|
|
System.out.println ("Unknown storage type : " + storageType);
|
|
System.out.println (HexFormatter.format (entry, 0, entry.length));
|
|
}
|
|
}
|
|
block = Utility.intValue (sectorBuffer[2], sectorBuffer[3]);
|
|
} while (block > 0);
|
|
|
|
// link double hi-res files
|
|
for (AppleFileSource fe : fileEntries)
|
|
{
|
|
String name = fe.getUniqueName ();
|
|
if (name.endsWith (".AUX"))
|
|
{
|
|
String partner1 = name.substring (0, name.length () - 4);
|
|
String partner2 = name.substring (0, name.length () - 4) + ".BIN";
|
|
for (AppleFileSource fe2 : fileEntries)
|
|
if (fe2.getUniqueName ().equals (partner1)
|
|
|| fe2.getUniqueName ().equals (partner2))
|
|
{
|
|
((FileEntry) fe2).link ((FileEntry) fe);
|
|
((FileEntry) fe).link ((FileEntry) fe2);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
public static boolean isCorrectFormat (AppleDisk disk)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
disk.setInterleave (1);
|
|
if (checkFormat (disk))
|
|
return true;
|
|
disk.setInterleave (0);
|
|
return checkFormat (disk);
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
public static boolean checkFormat (AppleDisk disk)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
byte[] buffer = disk.readBlock (2); // Prodos KEY BLOCK
|
|
if (debug)
|
|
{
|
|
System.out.println (HexFormatter.format (buffer));
|
|
System.out.printf ("Entry length : %02X%n", buffer[0x23]);
|
|
System.out.printf ("Entry per block: %02X%n", buffer[0x24]);
|
|
System.out.printf ("Bit map block : %02X%02X%n", buffer[0x27], buffer[0x28]);
|
|
}
|
|
|
|
// check entry length and entries per block
|
|
if (buffer[0x23] != 0x27 || buffer[0x24] != 0x0D)
|
|
return false;
|
|
|
|
int bitMapBlock = Utility.intValue (buffer[0x27], buffer[0x28]);
|
|
if (bitMapBlock < 3 || bitMapBlock > 10)
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
public List<DirectoryHeader> getDirectoryHeaders ()
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
return headerEntries;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
VolumeDirectoryHeader getVolumeDirectoryHeader ()
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
return volumeDirectoryHeader;
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
public DataSource getFile (int fileNo)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
if (fileNo == 0)
|
|
return volumeDirectoryHeader.getDataSource ();
|
|
|
|
return fileEntries.get (fileNo - 1).getDataSource ();
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
@Override
|
|
public AppleFileSource getCatalog ()
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
return new DefaultAppleFileSource ("Catalog", volumeDirectoryHeader.getDataSource (),
|
|
this);
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
@Override
|
|
public DataSource getFormattedSector (DiskAddress da)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
if (da.isZero ())
|
|
return bootSector;
|
|
|
|
byte[] buffer = disk.readBlock (da);
|
|
SectorType type = sectorTypes[da.getBlockNo ()];
|
|
|
|
if (type == catalogSector || type == subcatalogSector)
|
|
return new ProdosCatalogSector (this, disk, buffer, da);
|
|
if (type == volumeMapSector)
|
|
return new ProdosBitMapSector (this, disk, buffer, da);
|
|
if (type == masterIndexSector || type == indexSector)
|
|
return new ProdosIndexSector (getSectorFilename (da), disk, buffer, da);
|
|
if (type == extendedKeySector)
|
|
return new ProdosExtendedKeySector (disk, buffer, da);
|
|
if (type == dosSector)
|
|
return new DefaultSector ("Boot sector", disk, buffer, da);
|
|
|
|
String name = getSectorFilename (da);
|
|
if (name != null)
|
|
return new DefaultSector (name, disk, buffer, da);
|
|
return super.getFormattedSector (da);
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
@Override
|
|
public List<DiskAddress> getFileSectors (int fileNo)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
if (fileNo == 0)
|
|
return volumeDirectoryHeader.getSectors ();
|
|
|
|
return fileEntries.get (fileNo - 1).getSectors ();
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
void sortNodes (DefaultMutableTreeNode node)
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
int totalChildren = node.getChildCount ();
|
|
if (totalChildren == 0)
|
|
return;
|
|
|
|
List<DefaultMutableTreeNode> children = new ArrayList<> (totalChildren);
|
|
for (int i = 0; i < totalChildren; i++)
|
|
{
|
|
DefaultMutableTreeNode child = (DefaultMutableTreeNode) node.getChildAt (i);
|
|
children.add (child);
|
|
if (!child.isLeaf ())
|
|
sortNodes (child);
|
|
}
|
|
|
|
if (totalChildren > 1)
|
|
{
|
|
node.removeAllChildren ();
|
|
children.sort (new NodeComparator ());
|
|
for (DefaultMutableTreeNode child : children)
|
|
node.add (child);
|
|
}
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
@Override
|
|
public String toString ()
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
StringBuilder text = new StringBuilder ();
|
|
|
|
String timeC = volumeDirectoryHeader.created == null ? ""
|
|
: volumeDirectoryHeader.created.format (tf);
|
|
|
|
text.append (String.format ("Disk name : %s%n", getDisplayPath ()));
|
|
text.append (String.format ("Volume name : %s%n", volumeDirectoryHeader.name));
|
|
text.append (String.format ("Creation date : %s%n", timeC));
|
|
text.append (
|
|
String.format ("ProDOS version : %d%n", volumeDirectoryHeader.version));
|
|
text.append (
|
|
String.format ("Min ProDOS version : %d%n", volumeDirectoryHeader.minVersion));
|
|
text.append (
|
|
String.format ("Access rights : %d%n", volumeDirectoryHeader.access));
|
|
text.append (
|
|
String.format ("Entry length : %d%n", volumeDirectoryHeader.entryLength));
|
|
text.append (String.format ("Entries per block : %d%n",
|
|
volumeDirectoryHeader.entriesPerBlock));
|
|
text.append (
|
|
String.format ("File count : %d%n", volumeDirectoryHeader.fileCount));
|
|
text.append (
|
|
String.format ("Bitmap block : %d%n", volumeDirectoryHeader.bitMapBlock));
|
|
text.append (
|
|
String.format ("Total blocks : %d%n", volumeDirectoryHeader.totalBlocks));
|
|
text.append (String.format ("Interleave : %d", disk.getInterleave ()));
|
|
|
|
return text.toString ();
|
|
}
|
|
|
|
// ---------------------------------------------------------------------------------//
|
|
class NodeComparator implements Comparator<DefaultMutableTreeNode>
|
|
// ---------------------------------------------------------------------------------//
|
|
{
|
|
@Override
|
|
public int compare (DefaultMutableTreeNode o1, DefaultMutableTreeNode o2)
|
|
{
|
|
boolean folder1 = o1.getAllowsChildren ();
|
|
boolean folder2 = o2.getAllowsChildren ();
|
|
|
|
if (folder1 && !folder2)
|
|
return -1;
|
|
|
|
if (!folder1 && folder2)
|
|
return 1;
|
|
|
|
String name1 = ((FileEntry) o1.getUserObject ()).name;
|
|
String name2 = ((FileEntry) o2.getUserObject ()).name;
|
|
|
|
return name1.compareTo (name2);
|
|
}
|
|
}
|
|
} |