// /*************************************************************************** // The Disc Image Chef // ---------------------------------------------------------------------------- // // Filename : NDIF.cs // Author(s) : Natalia Portillo // // Component : Disk image plugins. // // --[ Description ] ---------------------------------------------------------- // // Manages Apple New Disk Image Format. // // --[ 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.Collections.Generic; using System.IO; using System.Runtime.InteropServices; using System.Text; using Claunia.RsrcFork; using DiscImageChef.CommonTypes; using DiscImageChef.Console; using DiscImageChef.Filters; using DiscImageChef.DiscImages; using SharpCompress.Compressors; using SharpCompress.Compressors.ADC; namespace DiscImageChef.DiscImages { // TODO: Detect OS X encrypted images // TODO: Check checksum // TODO: Implement segments // TODO: Implement compression public class Ndif : ImagePlugin { #region Internal constants /// /// Resource OSType for NDIF is "bcem" /// const uint NDIF_RESOURCE = 0x6263656D; /// /// Resource ID is always 128? Never found another /// const short NDIF_RESOURCEID = 128; #endregion #region Internal Structures [StructLayout(LayoutKind.Sequential, Pack = 1)] struct ChunkHeader { /// /// Version /// public short version; /// /// Filesystem ID /// public short driver; /// /// Disk image name, Str63 (Pascal string) /// [MarshalAs(UnmanagedType.ByValArray, SizeConst = 64)] public byte[] name; /// /// Sectors in image /// public uint sectors; /// /// Maximum number of sectors per chunk /// public uint maxSectorsPerChunk; /// /// Offset to add to every chunk offset /// public uint dataOffset; /// /// CRC28 of whole image /// public uint crc; /// /// Set to 1 if segmented /// public uint segmented; /// /// Unknown /// public uint p1; /// /// Unknown /// public uint p2; /// /// Unknown, spare? /// [MarshalAs(UnmanagedType.ByValArray, SizeConst = 5)] public uint[] unknown; /// /// Set to 1 by ShrinkWrap if image is encrypted /// public uint encrypted; /// /// Set by ShrinkWrap if image is encrypted, value is the same for same password /// public uint hash; /// /// How many chunks follow the header /// public uint chunks; } [StructLayout(LayoutKind.Sequential, Pack = 1)] struct BlockChunk { /// /// Starting sector, 3 bytes /// public uint sector; /// /// Chunk type /// public byte type; /// /// Offset in start of chunk /// public uint offset; /// /// Length in bytes of chunk /// public uint length; } [StructLayout(LayoutKind.Sequential, Pack = 1)] struct SegmentHeader { /// /// Segment # /// public ushort segment; /// /// How many segments /// public ushort segments; /// /// Seems to be a Guid, changes with different images, same for all segments of same image /// public Guid segmentId; /// /// Seems to be a CRC28 of this segment, unchecked /// public uint crc; } #endregion ChunkHeader header; Dictionary chunks; const byte CHUNK_TYPE_NOCOPY = 0; const byte CHUNK_TYPE_COPY = 2; const byte CHUNK_TYPE_KENCODE = 0x80; const byte CHUNK_TYPE_RLE = 0x81; const byte CHUNK_TYPE_LZH = 0x82; const byte CHUNK_TYPE_ADC = 0x83; /// /// Created by ShrinkWrap 3.5, dunno which version of the StuffIt algorithm it is using /// const byte CHUNK_TYPE_STUFFIT = 0xF0; const byte CHUNK_TYPE_END = 0xFF; const byte CHUNK_TYPE_COMPRESSED_MASK = 0x80; const short DRIVER_OSX = -1; const short DRIVER_HFS = 0; const short DRIVER_PRODOS = 256; const short DRIVER_DOS = 18771; Dictionary sectorCache; Dictionary chunkCache; const uint MAX_CACHE_SIZE = 16777216; const uint SECTOR_SIZE = 512; uint maxCachedSectors = MAX_CACHE_SIZE / SECTOR_SIZE; uint currentChunkCacheSize; uint buffersize; Stream imageStream; public Ndif() { Name = "Apple New Disk Image Format"; PluginUuid = new Guid("5A7FF7D8-491E-458D-8674-5B5EADBECC24"); ImageInfo = new ImageInfo(); ImageInfo.ReadableSectorTags = new List(); ImageInfo.ReadableMediaTags = new List(); ImageInfo.ImageHasPartitions = false; ImageInfo.ImageHasSessions = false; ImageInfo.ImageVersion = null; ImageInfo.ImageApplication = null; ImageInfo.ImageApplicationVersion = null; ImageInfo.ImageCreator = null; ImageInfo.ImageComments = null; ImageInfo.MediaManufacturer = null; ImageInfo.MediaModel = null; ImageInfo.MediaSerialNumber = null; ImageInfo.MediaBarcode = null; ImageInfo.MediaPartNumber = null; ImageInfo.MediaSequence = 0; ImageInfo.LastMediaSequence = 0; ImageInfo.DriveManufacturer = null; ImageInfo.DriveModel = null; ImageInfo.DriveSerialNumber = null; ImageInfo.DriveFirmwareRevision = null; } public override bool IdentifyImage(Filter imageFilter) { if(!imageFilter.HasResourceFork() || imageFilter.GetResourceForkLength() == 0) return false; ResourceFork rsrcFork; try { rsrcFork = new ResourceFork(imageFilter.GetResourceForkStream()); if(!rsrcFork.ContainsKey(NDIF_RESOURCE)) return false; Resource rsrc = rsrcFork.GetResource(NDIF_RESOURCE); if(rsrc.ContainsId(NDIF_RESOURCEID)) return true; } catch(InvalidCastException) { return false; } return false; } public override bool OpenImage(Filter imageFilter) { if(!imageFilter.HasResourceFork() || imageFilter.GetResourceForkLength() == 0) return false; ResourceFork rsrcFork; Resource rsrc; short[] bcems; try { rsrcFork = new ResourceFork(imageFilter.GetResourceForkStream()); if(!rsrcFork.ContainsKey(NDIF_RESOURCE)) return false; rsrc = rsrcFork.GetResource(NDIF_RESOURCE); bcems = rsrc.GetIds(); if(bcems == null || bcems.Length == 0) return false; } catch(InvalidCastException) { return false; } ImageInfo.Sectors = 0; foreach(short id in bcems) { byte[] bcem = rsrc.GetResource(NDIF_RESOURCEID); if(bcem.Length < 128) return false; header = BigEndianMarshal.ByteArrayToStructureBigEndian(bcem); DicConsole.DebugWriteLine("NDIF plugin", "footer.type = {0}", header.version); DicConsole.DebugWriteLine("NDIF plugin", "footer.driver = {0}", header.driver); DicConsole.DebugWriteLine("NDIF plugin", "footer.name = {0}", StringHandlers.PascalToString(header.name, Encoding.GetEncoding("macintosh"))); DicConsole.DebugWriteLine("NDIF plugin", "footer.sectors = {0}", header.sectors); DicConsole.DebugWriteLine("NDIF plugin", "footer.maxSectorsPerChunk = {0}", header.maxSectorsPerChunk); DicConsole.DebugWriteLine("NDIF plugin", "footer.dataOffset = {0}", header.dataOffset); DicConsole.DebugWriteLine("NDIF plugin", "footer.crc = 0x{0:X7}", header.crc); DicConsole.DebugWriteLine("NDIF plugin", "footer.segmented = {0}", header.segmented); DicConsole.DebugWriteLine("NDIF plugin", "footer.p1 = 0x{0:X8}", header.p1); DicConsole.DebugWriteLine("NDIF plugin", "footer.p2 = 0x{0:X8}", header.p2); DicConsole.DebugWriteLine("NDIF plugin", "footer.unknown[0] = 0x{0:X8}", header.unknown[0]); DicConsole.DebugWriteLine("NDIF plugin", "footer.unknown[1] = 0x{0:X8}", header.unknown[1]); DicConsole.DebugWriteLine("NDIF plugin", "footer.unknown[2] = 0x{0:X8}", header.unknown[2]); DicConsole.DebugWriteLine("NDIF plugin", "footer.unknown[3] = 0x{0:X8}", header.unknown[3]); DicConsole.DebugWriteLine("NDIF plugin", "footer.unknown[4] = 0x{0:X8}", header.unknown[4]); DicConsole.DebugWriteLine("NDIF plugin", "footer.encrypted = {0}", header.encrypted); DicConsole.DebugWriteLine("NDIF plugin", "footer.hash = 0x{0:X8}", header.hash); DicConsole.DebugWriteLine("NDIF plugin", "footer.chunks = {0}", header.chunks); // Block chunks and headers chunks = new Dictionary(); BigEndianBitConverter.IsLittleEndian = BitConverter.IsLittleEndian; for(int i = 0; i < header.chunks; i++) { // Obsolete read-only NDIF only prepended the header and then put the image without any kind of block references. // So let's falsify a block chunk BlockChunk bChnk = new BlockChunk(); byte[] sector = new byte[4]; Array.Copy(bcem, 128 + 0 + i * 12, sector, 1, 3); bChnk.sector = BigEndianBitConverter.ToUInt32(sector, 0); bChnk.type = bcem[128 + 3 + i * 12]; bChnk.offset = BigEndianBitConverter.ToUInt32(bcem, 128 + 4 + i * 12); bChnk.length = BigEndianBitConverter.ToUInt32(bcem, 128 + 8 + i * 12); DicConsole.DebugWriteLine("NDIF plugin", "bHdr.chunk[{0}].type = 0x{1:X2}", i, bChnk.type); DicConsole.DebugWriteLine("NDIF plugin", "bHdr.chunk[{0}].sector = {1}", i, bChnk.sector); DicConsole.DebugWriteLine("NDIF plugin", "bHdr.chunk[{0}].offset = {1}", i, bChnk.offset); DicConsole.DebugWriteLine("NDIF plugin", "bHdr.chunk[{0}].length = {1}", i, bChnk.length); if(bChnk.type == CHUNK_TYPE_END) break; bChnk.offset += header.dataOffset; bChnk.sector += (uint)ImageInfo.Sectors; // TODO: Handle compressed chunks if((bChnk.type == CHUNK_TYPE_KENCODE)) throw new ImageNotSupportedException("Chunks compressed with KenCode are not yet supported."); if((bChnk.type == CHUNK_TYPE_RLE)) throw new ImageNotSupportedException("Chunks compressed with RLE are not yet supported."); if((bChnk.type == CHUNK_TYPE_LZH)) throw new ImageNotSupportedException("Chunks compressed with LZH are not yet supported."); if((bChnk.type == CHUNK_TYPE_STUFFIT)) throw new ImageNotSupportedException("Chunks compressed with StuffIt! are not yet supported."); // TODO: Handle compressed chunks if((bChnk.type > CHUNK_TYPE_COPY && bChnk.type < CHUNK_TYPE_KENCODE) || (bChnk.type > CHUNK_TYPE_ADC && bChnk.type < CHUNK_TYPE_STUFFIT) || (bChnk.type > CHUNK_TYPE_STUFFIT && bChnk.type < CHUNK_TYPE_END) || bChnk.type == 1) throw new ImageNotSupportedException(string.Format("Unsupported chunk type 0x{0:X8} found", bChnk.type)); chunks.Add(bChnk.sector, bChnk); } ImageInfo.Sectors += header.sectors; } if(header.segmented > 0) throw new ImageNotSupportedException("Segmented images are not yet supported."); if(header.encrypted > 0) throw new ImageNotSupportedException("Encrypted images are not yet supported."); switch(ImageInfo.Sectors) { case 1440: ImageInfo.MediaType = MediaType.DOS_35_DS_DD_9; break; case 1600: ImageInfo.MediaType = MediaType.AppleSonyDS; break; case 2880: ImageInfo.MediaType = MediaType.DOS_35_HD; break; case 3360: ImageInfo.MediaType = MediaType.DMF; break; default: ImageInfo.MediaType = MediaType.GENERIC_HDD; break; } if(rsrcFork.ContainsKey(0x76657273)) { Resource versRsrc = rsrcFork.GetResource(0x76657273); if(versRsrc != null) { byte[] vers = versRsrc.GetResource(versRsrc.GetIds()[0]); Resources.Version version = new Resources.Version(vers); string major; string minor; string release = null; string dev = null; string pre = null; major = string.Format("{0}", version.MajorVersion); minor = string.Format(".{0}", version.MinorVersion / 10); if(version.MinorVersion % 10 > 0) release = string.Format(".{0}", version.MinorVersion % 10); switch(version.DevStage) { case Resources.Version.DevelopmentStage.Alpha: dev = "a"; break; case Resources.Version.DevelopmentStage.Beta: dev = "b"; break; case Resources.Version.DevelopmentStage.PreAlpha: dev = "d"; break; } if(dev == null && version.PreReleaseVersion > 0) dev = "f"; if(dev != null) pre = string.Format("{0}", version.PreReleaseVersion); ImageInfo.ImageApplicationVersion = string.Format("{0}{1}{2}{3}{4}", major, minor, release, dev, pre); ImageInfo.ImageApplication = version.VersionString; ImageInfo.ImageComments = version.VersionMessage; if(version.MajorVersion == 3) ImageInfo.ImageApplication = "ShrinkWrap™"; else if(version.MajorVersion == 6) ImageInfo.ImageApplication = "DiskCopy"; } } DicConsole.DebugWriteLine("NDIF plugin", "Image application = {0} version {1}", ImageInfo.ImageApplication, ImageInfo.ImageApplicationVersion); sectorCache = new Dictionary(); chunkCache = new Dictionary(); currentChunkCacheSize = 0; imageStream = imageFilter.GetDataForkStream(); buffersize = header.maxSectorsPerChunk * SECTOR_SIZE; ImageInfo.ImageCreationTime = imageFilter.GetCreationTime(); ImageInfo.ImageLastModificationTime = imageFilter.GetLastWriteTime(); ImageInfo.ImageName = StringHandlers.PascalToString(header.name, Encoding.GetEncoding("macintosh")); ImageInfo.SectorSize = SECTOR_SIZE; ImageInfo.XmlMediaType = XmlMediaType.BlockMedia; ImageInfo.ImageSize = ImageInfo.Sectors * SECTOR_SIZE; ImageInfo.ImageApplicationVersion = "6"; ImageInfo.ImageApplication = "Apple DiskCopy"; switch(ImageInfo.MediaType) { case MediaType.AppleSonyDS: ImageInfo.Cylinders = 80; ImageInfo.Heads = 2; ImageInfo.SectorsPerTrack = 10; break; case MediaType.DOS_35_DS_DD_9: ImageInfo.Cylinders = 80; ImageInfo.Heads = 2; ImageInfo.SectorsPerTrack = 9; break; case MediaType.DOS_35_HD: ImageInfo.Cylinders = 80; ImageInfo.Heads = 2; ImageInfo.SectorsPerTrack = 18; break; case MediaType.DMF: ImageInfo.Cylinders = 80; ImageInfo.Heads = 2; ImageInfo.SectorsPerTrack = 21; break; default: ImageInfo.MediaType = MediaType.GENERIC_HDD; ImageInfo.Cylinders = (uint)((ImageInfo.Sectors / 16) / 63); ImageInfo.Heads = 16; ImageInfo.SectorsPerTrack = 63; break; } return true; } public override byte[] ReadSector(ulong sectorAddress) { if(sectorAddress > ImageInfo.Sectors - 1) throw new ArgumentOutOfRangeException(nameof(sectorAddress), string.Format("Sector address {0} not found", sectorAddress)); byte[] sector; if(sectorCache.TryGetValue(sectorAddress, out sector)) return sector; BlockChunk currentChunk = new BlockChunk(); bool chunkFound = false; ulong chunkStartSector = 0; foreach(KeyValuePair kvp in chunks) { if(sectorAddress >= kvp.Key) { currentChunk = kvp.Value; chunkFound = true; chunkStartSector = kvp.Key; } } long relOff = ((long)sectorAddress - (long)chunkStartSector) * SECTOR_SIZE; if(relOff < 0) throw new ArgumentOutOfRangeException(nameof(relOff), string .Format("Got a negative offset for sector {0}. This should not happen.", sectorAddress)); if(!chunkFound) throw new ArgumentOutOfRangeException(nameof(sectorAddress), string.Format("Sector address {0} not found", sectorAddress)); if((currentChunk.type & CHUNK_TYPE_COMPRESSED_MASK) == CHUNK_TYPE_COMPRESSED_MASK) { byte[] buffer; if(!chunkCache.TryGetValue(chunkStartSector, out buffer)) { byte[] cmpBuffer = new byte[currentChunk.length]; imageStream.Seek(currentChunk.offset, SeekOrigin.Begin); imageStream.Read(cmpBuffer, 0, cmpBuffer.Length); MemoryStream cmpMs = new MemoryStream(cmpBuffer); Stream decStream; if(currentChunk.type == CHUNK_TYPE_ADC) decStream = new ADCStream(cmpMs, CompressionMode.Decompress); else throw new ImageNotSupportedException(string.Format("Unsupported chunk type 0x{0:X8} found", currentChunk.type)); byte[] tmpBuffer = new byte[buffersize]; int realSize = decStream.Read(tmpBuffer, 0, (int)buffersize); buffer = new byte[realSize]; Array.Copy(tmpBuffer, 0, buffer, 0, realSize); tmpBuffer = null; if(currentChunkCacheSize + realSize > MAX_CACHE_SIZE) { chunkCache.Clear(); currentChunkCacheSize = 0; } chunkCache.Add(chunkStartSector, buffer); currentChunkCacheSize += (uint)realSize; } sector = new byte[SECTOR_SIZE]; Array.Copy(buffer, relOff, sector, 0, SECTOR_SIZE); if(sectorCache.Count >= maxCachedSectors) sectorCache.Clear(); sectorCache.Add(sectorAddress, sector); return sector; } if(currentChunk.type == CHUNK_TYPE_NOCOPY) { sector = new byte[SECTOR_SIZE]; if(sectorCache.Count >= maxCachedSectors) sectorCache.Clear(); sectorCache.Add(sectorAddress, sector); return sector; } if(currentChunk.type == CHUNK_TYPE_COPY) { imageStream.Seek(currentChunk.offset + relOff, SeekOrigin.Begin); sector = new byte[SECTOR_SIZE]; imageStream.Read(sector, 0, sector.Length); if(sectorCache.Count >= maxCachedSectors) sectorCache.Clear(); sectorCache.Add(sectorAddress, sector); return sector; } throw new ImageNotSupportedException(string.Format("Unsupported chunk type 0x{0:X8} found", currentChunk.type)); } public override byte[] ReadSectors(ulong sectorAddress, uint length) { if(sectorAddress > ImageInfo.Sectors - 1) throw new ArgumentOutOfRangeException(nameof(sectorAddress), string.Format("Sector address {0} not found", sectorAddress)); if(sectorAddress + length > ImageInfo.Sectors) throw new ArgumentOutOfRangeException(nameof(length), "Requested more sectors than available"); MemoryStream ms = new MemoryStream(); for(uint i = 0; i < length; i++) { byte[] sector = ReadSector(sectorAddress + i); ms.Write(sector, 0, sector.Length); } return ms.ToArray(); } public override bool ImageHasPartitions() { return false; } public override ulong GetImageSize() { return ImageInfo.ImageSize; } public override ulong GetSectors() { return ImageInfo.Sectors; } public override uint GetSectorSize() { return ImageInfo.SectorSize; } public override string GetImageFormat() { return "Apple New Disk Image Format"; } public override string GetImageVersion() { return ImageInfo.ImageVersion; } public override string GetImageApplication() { return ImageInfo.ImageApplication; } public override string GetImageApplicationVersion() { return ImageInfo.ImageApplicationVersion; } public override string GetImageCreator() { return ImageInfo.ImageCreator; } public override DateTime GetImageCreationTime() { return ImageInfo.ImageCreationTime; } public override DateTime GetImageLastModificationTime() { return ImageInfo.ImageLastModificationTime; } public override string GetImageName() { return ImageInfo.ImageName; } public override string GetImageComments() { return ImageInfo.ImageComments; } public override MediaType GetMediaType() { return ImageInfo.MediaType; } #region Unsupported features public override byte[] ReadSectorTag(ulong sectorAddress, SectorTagType tag) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorsTag(ulong sectorAddress, uint length, SectorTagType tag) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadDiskTag(MediaTagType tag) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSector(ulong sectorAddress, uint track) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorTag(ulong sectorAddress, uint track, SectorTagType tag) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectors(ulong sectorAddress, uint length, uint track) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorsTag(ulong sectorAddress, uint length, uint track, SectorTagType tag) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorLong(ulong sectorAddress) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorLong(ulong sectorAddress, uint track) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorsLong(ulong sectorAddress, uint length) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override byte[] ReadSectorsLong(ulong sectorAddress, uint length, uint track) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override string GetMediaManufacturer() { return null; } public override string GetMediaModel() { return null; } public override string GetMediaSerialNumber() { return null; } public override string GetMediaBarcode() { return null; } public override string GetMediaPartNumber() { return null; } public override int GetMediaSequence() { return 0; } public override int GetLastDiskSequence() { return 0; } public override string GetDriveManufacturer() { return null; } public override string GetDriveModel() { return null; } public override string GetDriveSerialNumber() { return null; } public override List GetPartitions() { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override List GetTracks() { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override List GetSessionTracks(Session session) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override List GetSessionTracks(ushort session) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override List GetSessions() { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override bool? VerifySector(ulong sectorAddress) { return null; } public override bool? VerifySector(ulong sectorAddress, uint track) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override bool? VerifySectors(ulong sectorAddress, uint length, out List failingLbas, out List unknownLbas) { failingLbas = new List(); unknownLbas = new List(); for(ulong i = 0; i < ImageInfo.Sectors; i++) unknownLbas.Add(i); return null; } public override bool? VerifySectors(ulong sectorAddress, uint length, uint track, out List failingLbas, out List unknownLbas) { throw new FeatureUnsupportedImageException("Feature not supported by image format"); } public override bool? VerifyMediaImage() { return null; } #endregion } }