// /*************************************************************************** // The Disc Image Chef // ---------------------------------------------------------------------------- // // Filename : Read.cs // Author(s) : Natalia Portillo // // Component : Disk image plugins. // // --[ Description ] ---------------------------------------------------------- // // Reads Apridisk disk images. // // --[ License ] -------------------------------------------------------------- // // This library is free software; you can redistribute it and/or modify // it under the terms of the GNU Lesser General Public License as // published by the Free Software Foundation; either version 2.1 of the // License, or (at your option) any later version. // // This library is distributed in the hope that it will be useful, but // WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU // Lesser General Public License for more details. // // You should have received a copy of the GNU Lesser General Public // License along with this library; if not, see . // // ---------------------------------------------------------------------------- // Copyright © 2011-2018 Natalia Portillo // ****************************************************************************/ using System; using System.IO; using System.Runtime.InteropServices; using DiscImageChef.CommonTypes; using DiscImageChef.CommonTypes.Enums; using DiscImageChef.CommonTypes.Exceptions; using DiscImageChef.CommonTypes.Interfaces; using DiscImageChef.Console; namespace DiscImageChef.DiscImages { public partial class Apridisk { public bool Open(IFilter imageFilter) { Stream stream = imageFilter.GetDataForkStream(); stream.Seek(0, SeekOrigin.Begin); // Skip signature stream.Seek(signature.Length, SeekOrigin.Begin); int totalCylinders = -1; int totalHeads = -1; int maxSector = -1; int recordSize = Marshal.SizeOf(typeof(ApridiskRecord)); // Count cylinders while(stream.Position < stream.Length) { byte[] recB = new byte[recordSize]; stream.Read(recB, 0, recordSize); GCHandle handle = GCHandle.Alloc(recB, GCHandleType.Pinned); ApridiskRecord record = (ApridiskRecord)Marshal.PtrToStructure(handle.AddrOfPinnedObject(), typeof(ApridiskRecord)); handle.Free(); switch(record.type) { // Deleted record, just skip it case RecordType.Deleted: DicConsole.DebugWriteLine("Apridisk plugin", "Found deleted record at {0}", stream.Position); stream.Seek(record.headerSize - recordSize + record.dataSize, SeekOrigin.Current); break; case RecordType.Comment: DicConsole.DebugWriteLine("Apridisk plugin", "Found comment record at {0}", stream.Position); stream.Seek(record.headerSize - recordSize, SeekOrigin.Current); byte[] commentB = new byte[record.dataSize]; stream.Read(commentB, 0, commentB.Length); imageInfo.Comments = StringHandlers.CToString(commentB); DicConsole.DebugWriteLine("Apridisk plugin", "Comment: \"{0}\"", imageInfo.Comments); break; case RecordType.Creator: DicConsole.DebugWriteLine("Apridisk plugin", "Found creator record at {0}", stream.Position); stream.Seek(record.headerSize - recordSize, SeekOrigin.Current); byte[] creatorB = new byte[record.dataSize]; stream.Read(creatorB, 0, creatorB.Length); imageInfo.Creator = StringHandlers.CToString(creatorB); DicConsole.DebugWriteLine("Apridisk plugin", "Creator: \"{0}\"", imageInfo.Creator); break; case RecordType.Sector: if(record.compression != CompressType.Compressed && record.compression != CompressType.Uncompresed) throw new ImageNotSupportedException($"Found record with unknown compression type 0x{(ushort)record.compression:X4} at {stream.Position}"); DicConsole.DebugWriteLine("Apridisk plugin", "Found {4} sector record at {0} for cylinder {1} head {2} sector {3}", stream.Position, record.cylinder, record.head, record.sector, record.compression == CompressType.Compressed ? "compressed" : "uncompressed"); if(record.cylinder > totalCylinders) totalCylinders = record.cylinder; if(record.head > totalHeads) totalHeads = record.head; if(record.sector > maxSector) maxSector = record.sector; stream.Seek(record.headerSize - recordSize + record.dataSize, SeekOrigin.Current); break; default: throw new ImageNotSupportedException($"Found record with unknown type 0x{(uint)record.type:X8} at {stream.Position}"); } } totalCylinders++; totalHeads++; if(totalCylinders <= 0 || totalHeads <= 0) throw new ImageNotSupportedException("No cylinders or heads found"); sectorsData = new byte[totalCylinders][][][]; // Total sectors per track uint[][] spts = new uint[totalCylinders][]; imageInfo.Cylinders = (ushort)totalCylinders; imageInfo.Heads = (byte)totalHeads; DicConsole.DebugWriteLine("Apridisk plugin", "Found {0} cylinders and {1} heads with a maximum sector number of {2}", totalCylinders, totalHeads, maxSector); // Create heads for(int i = 0; i < totalCylinders; i++) { sectorsData[i] = new byte[totalHeads][][]; spts[i] = new uint[totalHeads]; for(int j = 0; j < totalHeads; j++) sectorsData[i][j] = new byte[maxSector + 1][]; } imageInfo.SectorSize = uint.MaxValue; ulong headersizes = 0; // Read sectors stream.Seek(signature.Length, SeekOrigin.Begin); while(stream.Position < stream.Length) { byte[] recB = new byte[recordSize]; stream.Read(recB, 0, recordSize); GCHandle handle = GCHandle.Alloc(recB, GCHandleType.Pinned); ApridiskRecord record = (ApridiskRecord)Marshal.PtrToStructure(handle.AddrOfPinnedObject(), typeof(ApridiskRecord)); handle.Free(); switch(record.type) { // Not sector record, just skip it case RecordType.Deleted: case RecordType.Comment: case RecordType.Creator: stream.Seek(record.headerSize - recordSize + record.dataSize, SeekOrigin.Current); headersizes += record.headerSize + record.dataSize; break; case RecordType.Sector: stream.Seek(record.headerSize - recordSize, SeekOrigin.Current); byte[] data = new byte[record.dataSize]; stream.Read(data, 0, data.Length); spts[record.cylinder][record.head]++; uint realLength = record.dataSize; if(record.compression == CompressType.Compressed) realLength = Decompress(data, out sectorsData[record.cylinder][record.head][record.sector]); else sectorsData[record.cylinder][record.head][record.sector] = data; if(realLength < imageInfo.SectorSize) imageInfo.SectorSize = realLength; headersizes += record.headerSize + record.dataSize; break; } } DicConsole.DebugWriteLine("Apridisk plugin", "Found a minimum of {0} bytes per sector", imageInfo.SectorSize); // Count sectors per track uint spt = uint.MaxValue; for(ushort cyl = 0; cyl < imageInfo.Cylinders; cyl++) { for(ushort head = 0; head < imageInfo.Heads; head++) if(spts[cyl][head] < spt) spt = spts[cyl][head]; } imageInfo.SectorsPerTrack = spt; DicConsole.DebugWriteLine("Apridisk plugin", "Found a minimum of {0} sectors per track", imageInfo.SectorsPerTrack); imageInfo.MediaType = Geometry.GetMediaType(((ushort)imageInfo.Cylinders, (byte)imageInfo.Heads, (ushort)imageInfo.SectorsPerTrack, 512, MediaEncoding.MFM, false)); imageInfo.ImageSize = (ulong)stream.Length - headersizes; imageInfo.CreationTime = imageFilter.GetCreationTime(); imageInfo.LastModificationTime = imageFilter.GetLastWriteTime(); imageInfo.MediaTitle = Path.GetFileNameWithoutExtension(imageFilter.GetFilename()); imageInfo.Sectors = imageInfo.Cylinders * imageInfo.Heads * imageInfo.SectorsPerTrack; imageInfo.XmlMediaType = XmlMediaType.BlockMedia; return true; } public byte[] ReadSector(ulong sectorAddress) { (ushort cylinder, byte head, byte sector) = LbaToChs(sectorAddress); if(cylinder >= sectorsData.Length) throw new ArgumentOutOfRangeException(nameof(sectorAddress), "Sector address not found"); if(head >= sectorsData[cylinder].Length) throw new ArgumentOutOfRangeException(nameof(sectorAddress), "Sector address not found"); if(sector > sectorsData[cylinder][head].Length) throw new ArgumentOutOfRangeException(nameof(sectorAddress), "Sector address not found"); return sectorsData[cylinder][head][sector]; } public byte[] ReadSectors(ulong sectorAddress, uint length) { if(sectorAddress > imageInfo.Sectors - 1) throw new ArgumentOutOfRangeException(nameof(sectorAddress), "Sector address not found"); if(sectorAddress + length > imageInfo.Sectors) throw new ArgumentOutOfRangeException(nameof(length), "Requested more sectors than available"); MemoryStream buffer = new MemoryStream(); for(uint i = 0; i < length; i++) { byte[] sector = ReadSector(sectorAddress + i); buffer.Write(sector, 0, sector.Length); } return buffer.ToArray(); } } }