public bool Identify(IFilter imageFilter)
{
Stream stream = imageFilter.GetDataForkStream();
stream.Seek(0, SeekOrigin.Begin);
if (stream.Length < 88)
{
return(false);
}
byte[] hdr = new byte[88];
stream.Read(hdr, 0, 88);
AlcoholHeader header = Marshal.ByteArrayToStructureLittleEndian <AlcoholHeader>(hdr);
return(header.signature.SequenceEqual(alcoholSignature));
}
public bool Identify(IFilter imageFilter)
{
Stream stream = imageFilter.GetDataForkStream();
stream.Seek(0, SeekOrigin.Begin);
if (stream.Length < 88)
{
return(false);
}
byte[] hdr = new byte[88];
stream.Read(hdr, 0, 88);
IntPtr hdrPtr = Marshal.AllocHGlobal(88);
Marshal.Copy(hdr, 0, hdrPtr, 88);
AlcoholHeader header = (AlcoholHeader)Marshal.PtrToStructure(hdrPtr, typeof(AlcoholHeader));
Marshal.FreeHGlobal(hdrPtr);
return(header.signature.SequenceEqual(alcoholSignature));
}
public bool Close()
{
if (!IsWriting)
{
ErrorMessage = "Image is not opened for writing";
return(false);
}
byte sessions = byte.MinValue;
foreach (Track t in writingTracks)
{
if (t.TrackSession > byte.MinValue)
{
sessions = (byte)t.TrackSession;
}
}
AlcoholHeader header = new AlcoholHeader
{
signature = alcoholSignature,
version = new byte[] { 1, 5 },
type = MediaTypeToAlcohol(imageInfo.MediaType),
sessions = sessions,
structuresOffset = (uint)(pfi == null ? 0 : 96),
sessionOffset = (uint)(pfi == null ? 96 : 4196),
unknown1 = new ushort[2],
unknown2 = new uint[2],
unknown3 = new uint[6],
unknown4 = new uint[3]
};
// Alcohol sets this always, Daemon Tool expects this
header.unknown1[0] = 2;
alcSessions = new Dictionary <int, AlcoholSession>();
alcTracks = new Dictionary <int, AlcoholTrack>();
alcToc = new Dictionary <int, Dictionary <int, AlcoholTrack> >();
writingTracks = writingTracks.OrderBy(t => t.TrackSession).ThenBy(t => t.TrackSequence).ToList();
alcTrackExtras = new Dictionary <int, AlcoholTrackExtra>();
long currentTrackOffset = header.sessionOffset + Marshal.SizeOf <AlcoholSession>() * sessions;
FullTOC.CDFullTOC?decodedToc = FullTOC.Decode(fullToc);
long currentExtraOffset = currentTrackOffset;
for (int i = 1; i <= sessions; i++)
{
if (decodedToc.HasValue)
{
currentExtraOffset += Marshal.SizeOf <AlcoholTrack>() *
decodedToc.Value.TrackDescriptors.Count(t => t.SessionNumber == i);
}
else
{
currentExtraOffset += Marshal.SizeOf <AlcoholTrack>() * 3;
currentExtraOffset += Marshal.SizeOf <AlcoholTrack>() *
writingTracks.Count(t => t.TrackSession == i);
if (i < sessions)
{
currentExtraOffset += Marshal.SizeOf <AlcoholTrack>() * 2;
}
}
}
long footerOffset = currentExtraOffset + Marshal.SizeOf <AlcoholTrackExtra>() * writingTracks.Count;
if (bca != null)
{
header.bcaOffset = (uint)footerOffset;
footerOffset += bca.Length;
}
if (isDvd)
{
alcSessions.Add(1,
new AlcoholSession
{
sessionEnd =
(int)(writingTracks[0].TrackEndSector - writingTracks[0].TrackStartSector +
1),
sessionSequence = 1,
allBlocks = 1,
nonTrackBlocks = 3,
firstTrack = 1,
lastTrack = 1,
trackOffset = 4220
});
footerOffset = 4300;
if (bca != null)
{
footerOffset += bca.Length;
}
alcTracks.Add(1,
new AlcoholTrack
{
mode = AlcoholTrackMode.DVD,
adrCtl = 20,
point = 1,
extraOffset =
(uint)(writingTracks[0].TrackEndSector - writingTracks[0].TrackStartSector +
1),
sectorSize = 2048,
files = 1,
footerOffset = (uint)footerOffset,
unknown = new byte[18],
unknown2 = new byte[24]
});
alcToc.Add(1, alcTracks);
}
else
{
for (int i = 1; i <= sessions; i++)
{
Track firstTrack = writingTracks.First(t => t.TrackSession == i);
Track lastTrack = writingTracks.Last(t => t.TrackSession == i);
alcSessions.Add(i,
new AlcoholSession
{
sessionStart = (int)firstTrack.TrackStartSector - 150,
sessionEnd = (int)lastTrack.TrackEndSector + 1,
sessionSequence = (ushort)i,
allBlocks =
(byte)(decodedToc?.TrackDescriptors.Count(t => t.SessionNumber == i) ??
writingTracks.Count(t => t.TrackSession == i) + 3),
nonTrackBlocks =
(byte)(decodedToc?.TrackDescriptors.Count(t => t.SessionNumber == i &&
t.POINT >= 0xA0 &&
t.POINT <= 0xAF) ??
3),
firstTrack = (ushort)firstTrack.TrackSequence,
lastTrack = (ushort)lastTrack.TrackSequence,
trackOffset = (uint)currentTrackOffset
});
Dictionary <int, AlcoholTrack> thisSessionTracks = new Dictionary <int, AlcoholTrack>();
trackFlags.TryGetValue((byte)firstTrack.TrackSequence, out byte firstTrackControl);
trackFlags.TryGetValue((byte)lastTrack.TrackSequence, out byte lastTrackControl);
if (firstTrackControl == 0 && firstTrack.TrackType != TrackType.Audio)
{
firstTrackControl = (byte)CdFlags.DataTrack;
}
if (lastTrackControl == 0 && lastTrack.TrackType != TrackType.Audio)
{
lastTrackControl = (byte)CdFlags.DataTrack;
}
(byte minute, byte second, byte frame)leadinPmsf = LbaToMsf(lastTrack.TrackEndSector + 1);
if (decodedToc.HasValue &&
decodedToc.Value.TrackDescriptors.Any(t => t.SessionNumber == i && t.POINT >= 0xA0 &&
t.POINT <= 0xAF))
{
foreach (FullTOC.TrackDataDescriptor tocTrk in
decodedToc.Value.TrackDescriptors.Where(t => t.SessionNumber == i && t.POINT >= 0xA0 &&
t.POINT <= 0xAF))
{
thisSessionTracks.Add(tocTrk.POINT,
new AlcoholTrack
{
adrCtl = (byte)((tocTrk.ADR << 4) + tocTrk.CONTROL),
tno = tocTrk.TNO,
point = tocTrk.POINT,
min = tocTrk.Min,
sec = tocTrk.Sec,
frame = tocTrk.Frame,
zero = tocTrk.Zero,
pmin = tocTrk.PMIN,
psec = tocTrk.PSEC,
pframe = tocTrk.PFRAME,
mode = AlcoholTrackMode.NoData,
unknown = new byte[18],
unknown2 = new byte[24]
});
currentTrackOffset += Marshal.SizeOf <AlcoholTrack>();
}
}
else
{
thisSessionTracks.Add(0xA0, new AlcoholTrack
{
adrCtl = (byte)((1 << 4) + firstTrackControl),
pmin = (byte)firstTrack.TrackSequence,
mode = AlcoholTrackMode.NoData,
point = 0xA0,
unknown = new byte[18],
unknown2 = new byte[24],
psec = (byte)(imageInfo.MediaType == MediaType.CDI
? 0x10
: writingTracks.Any(t => t.TrackType == TrackType.CdMode2Form1 ||
t.TrackType == TrackType.CdMode2Form2 ||
t.TrackType == TrackType.CdMode2Formless)
? 0x20
: 0)
});
thisSessionTracks.Add(0xA1,
new AlcoholTrack
{
adrCtl = (byte)((1 << 4) + lastTrackControl),
pmin = (byte)lastTrack.TrackSequence,
mode = AlcoholTrackMode.NoData,
point = 0xA1,
unknown = new byte[18],
unknown2 = new byte[24]
});
thisSessionTracks.Add(0xA2,
new AlcoholTrack
{
adrCtl = (byte)((1 << 4) + firstTrackControl),
zero = 0,
pmin = leadinPmsf.minute,
psec = leadinPmsf.second,
pframe = leadinPmsf.frame,
mode = AlcoholTrackMode.NoData,
point = 0xA2,
unknown = new byte[18],
unknown2 = new byte[24]
});
currentTrackOffset += Marshal.SizeOf <AlcoholTrack>() * 3;
}
foreach (Track track in writingTracks.Where(t => t.TrackSession == i).OrderBy(t => t.TrackSequence))
{
AlcoholTrack alcTrk = new AlcoholTrack();
if (decodedToc.HasValue &&
decodedToc.Value.TrackDescriptors.Any(t => t.SessionNumber == i &&
t.POINT == track.TrackSequence))
{
FullTOC.TrackDataDescriptor tocTrk =
decodedToc.Value.TrackDescriptors.First(t => t.SessionNumber == i &&
t.POINT == track.TrackSequence);
alcTrk.adrCtl = (byte)((tocTrk.ADR << 4) + tocTrk.CONTROL);
alcTrk.tno = tocTrk.TNO;
alcTrk.point = tocTrk.POINT;
alcTrk.min = tocTrk.Min;
alcTrk.sec = tocTrk.Sec;
alcTrk.frame = tocTrk.Frame;
alcTrk.zero = tocTrk.Zero;
alcTrk.pmin = tocTrk.PMIN;
alcTrk.psec = tocTrk.PSEC;
alcTrk.pframe = tocTrk.PFRAME;
}
else
{
(byte minute, byte second, byte frame)msf = LbaToMsf(track.TrackStartSector);
trackFlags.TryGetValue((byte)track.TrackSequence, out byte trackControl);
if (trackControl == 0 && track.TrackType != TrackType.Audio)
{
trackControl = (byte)CdFlags.DataTrack;
}
alcTrk.adrCtl = (byte)((1 << 4) + trackControl);
alcTrk.point = (byte)track.TrackSequence;
alcTrk.zero = 0;
alcTrk.pmin = msf.minute;
alcTrk.psec = msf.second;
alcTrk.pframe = msf.frame;
}
alcTrk.mode = TrackTypeToAlcohol(track.TrackType);
alcTrk.subMode = track.TrackSubchannelType != TrackSubchannelType.None
? AlcoholSubchannelMode.Interleaved
: AlcoholSubchannelMode.None;
alcTrk.sectorSize = (ushort)(track.TrackRawBytesPerSector +
(track.TrackSubchannelType != TrackSubchannelType.None ? 96 : 0));
alcTrk.startLba = (uint)track.TrackStartSector;
alcTrk.startOffset = track.TrackFileOffset;
alcTrk.files = 1;
alcTrk.extraOffset = (uint)currentExtraOffset;
alcTrk.footerOffset = (uint)footerOffset;
// Alcohol seems to set that for all CD tracks
// Daemon Tools expect it to be like this
alcTrk.unknown = new byte[]
{
0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
0x00, 0x00, 0x00, 0x00
};
alcTrk.unknown2 = new byte[24];
thisSessionTracks.Add((int)track.TrackSequence, alcTrk);
currentTrackOffset += Marshal.SizeOf <AlcoholTrack>();
currentExtraOffset += Marshal.SizeOf <AlcoholTrackExtra>();
AlcoholTrackExtra trkExtra = new AlcoholTrackExtra
{
sectors = (uint)(track.TrackEndSector - track.TrackStartSector + 1)
};
// When track mode changes there's a mandatory gap, Alcohol needs it
if (track.TrackSequence == firstTrack.TrackSequence)
{
trkExtra.pregap = 150;
}
else if (thisSessionTracks.TryGetValue((int)(track.TrackSequence - 1),
out AlcoholTrack previousTrack) &&
alcTrackExtras.TryGetValue((int)(track.TrackSequence - 1),
out AlcoholTrackExtra previousExtra) &&
previousTrack.mode != alcTrk.mode)
{
previousExtra.sectors -= 150;
trkExtra.pregap = 150;
alcTrackExtras.Remove((int)(track.TrackSequence - 1));
alcTrackExtras.Add((int)(track.TrackSequence - 1), previousExtra);
}
else
{
trkExtra.pregap = 0;
}
alcTrackExtras.Add((int)track.TrackSequence, trkExtra);
}
if (decodedToc.HasValue &&
decodedToc.Value.TrackDescriptors.Any(t => t.SessionNumber == i && t.POINT >= 0xB0))
{
foreach (FullTOC.TrackDataDescriptor tocTrk in
decodedToc.Value.TrackDescriptors.Where(t => t.SessionNumber == i && t.POINT >= 0xB0))
{
thisSessionTracks.Add(tocTrk.POINT,
new AlcoholTrack
{
adrCtl = (byte)((tocTrk.ADR << 4) + tocTrk.CONTROL),
tno = tocTrk.TNO,
point = tocTrk.POINT,
min = tocTrk.Min,
sec = tocTrk.Sec,
frame = tocTrk.Frame,
zero = tocTrk.Zero,
pmin = tocTrk.PMIN,
psec = tocTrk.PSEC,
pframe = tocTrk.PFRAME,
mode = AlcoholTrackMode.NoData,
unknown = new byte[18],
unknown2 = new byte[24]
});
currentTrackOffset += Marshal.SizeOf <AlcoholTrack>();
}
}
else if (i < sessions)
{
(byte minute, byte second, byte frame)leadoutAmsf =
LbaToMsf(writingTracks.First(t => t.TrackSession == i + 1).TrackStartSector - 150);
(byte minute, byte second, byte frame)leadoutPmsf =
LbaToMsf(writingTracks.OrderBy(t => t.TrackSession).ThenBy(t => t.TrackSequence).Last()
.TrackStartSector);
thisSessionTracks.Add(0xB0,
new AlcoholTrack
{
point = 0xB0,
adrCtl = 0x50,
zero = 0,
min = leadoutAmsf.minute,
sec = leadoutAmsf.second,
frame = leadoutAmsf.frame,
pmin = leadoutPmsf.minute,
psec = leadoutPmsf.second,
pframe = leadoutPmsf.frame,
unknown = new byte[18],
unknown2 = new byte[24]
});
thisSessionTracks.Add(0xC0, new AlcoholTrack
{
point = 0xC0,
adrCtl = 0x50,
min = 128,
pmin = 97,
psec = 25,
unknown = new byte[18],
unknown2 = new byte[24]
});
currentTrackOffset += Marshal.SizeOf <AlcoholTrack>() * 2;
}
alcToc.Add(i, thisSessionTracks);
}
}
alcFooter = new AlcoholFooter
{
filenameOffset = (uint)(footerOffset + Marshal.SizeOf <AlcoholFooter>()), widechar = 1
};
byte[] filename = Encoding.Unicode.GetBytes("*.mdf"); // Yup, Alcohol stores no filename but a wildcard.
IntPtr blockPtr;
// Write header
descriptorStream.Seek(0, SeekOrigin.Begin);
byte[] block = new byte[Marshal.SizeOf <AlcoholHeader>()];
blockPtr = System.Runtime.InteropServices.Marshal.AllocHGlobal(Marshal.SizeOf <AlcoholHeader>());
System.Runtime.InteropServices.Marshal.StructureToPtr(header, blockPtr, true);
System.Runtime.InteropServices.Marshal.Copy(blockPtr, block, 0, block.Length);
System.Runtime.InteropServices.Marshal.FreeHGlobal(blockPtr);
descriptorStream.Write(block, 0, block.Length);
// Write DVD structures if pressent
if (header.structuresOffset != 0)
{
if (dmi != null)
{
descriptorStream.Seek(header.structuresOffset, SeekOrigin.Begin);
if (dmi.Length == 2052)
{
descriptorStream.Write(dmi, 0, 2052);
}
else if (dmi.Length == 2048)
{
descriptorStream.Write(new byte[] { 0x08, 0x02, 0x00, 0x00 }, 0, 4);
descriptorStream.Write(dmi, 0, 2048);
}
}
// TODO: Create fake PFI if none present
if (pfi != null)
{
descriptorStream.Seek(header.structuresOffset + 2052, SeekOrigin.Begin);
descriptorStream.Write(pfi, pfi.Length - 2048, 2048);
}
}
// Write sessions
descriptorStream.Seek(header.sessionOffset, SeekOrigin.Begin);
foreach (AlcoholSession session in alcSessions.Values)
{
block = new byte[Marshal.SizeOf <AlcoholSession>()];
blockPtr = System.Runtime.InteropServices.Marshal.AllocHGlobal(Marshal.SizeOf <AlcoholSession>());
System.Runtime.InteropServices.Marshal.StructureToPtr(session, blockPtr, true);
System.Runtime.InteropServices.Marshal.Copy(blockPtr, block, 0, block.Length);
System.Runtime.InteropServices.Marshal.FreeHGlobal(blockPtr);
descriptorStream.Write(block, 0, block.Length);
}
// Write tracks
foreach (KeyValuePair <int, Dictionary <int, AlcoholTrack> > kvp in alcToc)
{
descriptorStream.Seek(alcSessions.First(t => t.Key == kvp.Key).Value.trackOffset, SeekOrigin.Begin);
foreach (AlcoholTrack track in kvp.Value.Values)
{
block = new byte[Marshal.SizeOf <AlcoholTrack>()];
blockPtr = System.Runtime.InteropServices.Marshal.AllocHGlobal(Marshal.SizeOf <AlcoholTrack>());
System.Runtime.InteropServices.Marshal.StructureToPtr(track, blockPtr, true);
System.Runtime.InteropServices.Marshal.Copy(blockPtr, block, 0, block.Length);
System.Runtime.InteropServices.Marshal.FreeHGlobal(blockPtr);
descriptorStream.Write(block, 0, block.Length);
if (isDvd)
{
continue;
}
// Write extra
long position = descriptorStream.Position;
descriptorStream.Seek(track.extraOffset, SeekOrigin.Begin);
if (alcTrackExtras.TryGetValue(track.point, out AlcoholTrackExtra extra))
{
block = new byte[Marshal.SizeOf <AlcoholTrackExtra>()];
blockPtr =
System.Runtime.InteropServices.Marshal.AllocHGlobal(Marshal.SizeOf <AlcoholTrackExtra>());
System.Runtime.InteropServices.Marshal.StructureToPtr(extra, blockPtr, true);
System.Runtime.InteropServices.Marshal.Copy(blockPtr, block, 0, block.Length);
System.Runtime.InteropServices.Marshal.FreeHGlobal(blockPtr);
descriptorStream.Write(block, 0, block.Length);
}
descriptorStream.Seek(position, SeekOrigin.Begin);
}
}
// Write BCA
if (bca != null)
{
descriptorStream.Seek(header.bcaOffset, SeekOrigin.Begin);
descriptorStream.Write(bca, 0, bca.Length);
}
// Write footer
descriptorStream.Seek(footerOffset, SeekOrigin.Begin);
block = new byte[Marshal.SizeOf <AlcoholFooter>()];
blockPtr = System.Runtime.InteropServices.Marshal.AllocHGlobal(Marshal.SizeOf <AlcoholFooter>());
System.Runtime.InteropServices.Marshal.StructureToPtr(alcFooter, blockPtr, true);
System.Runtime.InteropServices.Marshal.Copy(blockPtr, block, 0, block.Length);
System.Runtime.InteropServices.Marshal.FreeHGlobal(blockPtr);
descriptorStream.Write(block, 0, block.Length);
// Write filename
descriptorStream.Write(filename, 0, filename.Length);
// Write filename null termination
descriptorStream.Write(new byte[] { 0, 0 }, 0, 2);
descriptorStream.Flush();
descriptorStream.Close();
imageStream.Flush();
imageStream.Close();
IsWriting = false;
ErrorMessage = "";
return(true);
}