aaru-dps/Aaru
1
0
Fork 0
mirror of https://github.com/aaru-dps/Aaru.git synced 2025-12-16 19:24:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Code restyling.

Browse Source
This commit is contained in:
Natalia Portillo
2020-02-29 18:03:35 +00:00
parent 4ea327f0c6
commit f7e173710e
855 changed files with 43605 additions and 38045 deletions
Download Patch File Download Diff File Expand all files Collapse all files

93
Aaru.Images/QCOW/Write.cs
View File

@@ -46,50 +46,56 @@ namespace Aaru.DiscImages
public partial class Qcow
{
public bool Create(string path, MediaType mediaType, Dictionary<string, string> options, ulong sectors,
uint sectorSize)
uint sectorSize)
{
if(sectorSize != 512)
{
ErrorMessage = "Unsupported sector size";
return false;
}
if(!SupportedMediaTypes.Contains(mediaType))
{
ErrorMessage = $"Unsupport media format {mediaType}";
return false;
}
// TODO: Correct this calculation
if(sectors * sectorSize / 65536 > uint.MaxValue)
if((sectors * sectorSize) / 65536 > uint.MaxValue)
{
ErrorMessage = "Too many sectors for selected cluster size";
return false;
}
imageInfo = new ImageInfo {MediaType = mediaType, SectorSize = sectorSize, Sectors = sectors};
imageInfo = new ImageInfo
{
MediaType = mediaType, SectorSize = sectorSize, Sectors = sectors
};
try { writingStream = new FileStream(path, FileMode.OpenOrCreate, FileAccess.ReadWrite, FileShare.None); }
try
{
writingStream = new FileStream(path, FileMode.OpenOrCreate, FileAccess.ReadWrite, FileShare.None);
}
catch(IOException e)
{
ErrorMessage = $"Could not create new image file, exception {e.Message}";
return false;
}
qHdr = new QCowHeader
{
magic = QCOW_MAGIC,
version = QCOW_VERSION,
size = sectors * sectorSize,
cluster_bits = 12,
l2_bits = 9,
l1_table_offset = (ulong)Marshal.SizeOf<QCowHeader>()
magic = QCOW_MAGIC, version = QCOW_VERSION, size = sectors * sectorSize, cluster_bits = 12,
l2_bits = 9, l1_table_offset = (ulong)Marshal.SizeOf<QCowHeader>()
};
int shift = qHdr.cluster_bits + qHdr.l2_bits;
clusterSize = 1 << qHdr.cluster_bits;
clusterSectors = 1 << (qHdr.cluster_bits - 9);
l1Size = (uint)((qHdr.size + (ulong)(1 << shift) - 1) >> shift);
l1Size = (uint)(((qHdr.size + (ulong)(1 << shift)) - 1) >> shift);
l2Size = 1 << qHdr.l2_bits;
l1Table = new ulong[l1Size];
@@ -102,31 +108,38 @@ namespace Aaru.DiscImages
{
l1Mask <<= 1;
if(c >= 64 - l1Shift) continue;
if(c >= 64 - l1Shift)
continue;
l1Mask += 1;
c++;
}
l2Mask = 0;
for(int i = 0; i < qHdr.l2_bits; i++) l2Mask = (l2Mask << 1) + 1;
for(int i = 0; i < qHdr.l2_bits; i++)
l2Mask = (l2Mask << 1) + 1;
l2Mask <<= qHdr.cluster_bits;
sectorMask = 0;
for(int i = 0; i < qHdr.cluster_bits; i++) sectorMask = (sectorMask << 1) + 1;
byte[] empty = new byte[qHdr.l1_table_offset + l1Size * 8];
for(int i = 0; i < qHdr.cluster_bits; i++)
sectorMask = (sectorMask << 1) + 1;
byte[] empty = new byte[qHdr.l1_table_offset + (l1Size * 8)];
writingStream.Write(empty, 0, empty.Length);
IsWriting = true;
ErrorMessage = null;
return true;
}
public bool WriteMediaTag(byte[] data, MediaTagType tag)
{
ErrorMessage = "Writing media tags is not supported.";
return false;
}
@@ -135,23 +148,27 @@ namespace Aaru.DiscImages
if(!IsWriting)
{
ErrorMessage = "Tried to write on a non-writable image";
return false;
}
if(data.Length != imageInfo.SectorSize)
{
ErrorMessage = "Incorrect data size";
return false;
}
if(sectorAddress >= imageInfo.Sectors)
{
ErrorMessage = "Tried to write past image size";
return false;
}
// Ignore empty sectors
if(ArrayHelpers.ArrayIsNullOrEmpty(data)) return true;
if(ArrayHelpers.ArrayIsNullOrEmpty(data))
return true;
ulong byteAddress = sectorAddress * 512;
@@ -174,7 +191,7 @@ namespace Aaru.DiscImages
ulong l2Off = (byteAddress & l2Mask) >> qHdr.cluster_bits;
writingStream.Seek((long)(l1Table[l1Off] + l2Off * 8), SeekOrigin.Begin);
writingStream.Seek((long)(l1Table[l1Off] + (l2Off * 8)), SeekOrigin.Begin);
byte[] entry = new byte[8];
writingStream.Read(entry, 0, 8);
@@ -185,7 +202,7 @@ namespace Aaru.DiscImages
offset = (ulong)writingStream.Length;
byte[] cluster = new byte[clusterSize];
entry = BigEndianBitConverter.GetBytes(offset);
writingStream.Seek((long)(l1Table[l1Off] + l2Off * 8), SeekOrigin.Begin);
writingStream.Seek((long)(l1Table[l1Off] + (l2Off * 8)), SeekOrigin.Begin);
writingStream.Write(entry, 0, 8);
writingStream.Seek(0, SeekOrigin.End);
writingStream.Write(cluster, 0, cluster.Length);
@@ -195,6 +212,7 @@ namespace Aaru.DiscImages
writingStream.Write(data, 0, data.Length);
ErrorMessage = "";
return true;
}
@@ -204,44 +222,53 @@ namespace Aaru.DiscImages
if(!IsWriting)
{
ErrorMessage = "Tried to write on a non-writable image";
return false;
}
if(data.Length % imageInfo.SectorSize != 0)
{
ErrorMessage = "Incorrect data size";
return false;
}
if(sectorAddress + length > imageInfo.Sectors)
{
ErrorMessage = "Tried to write past image size";
return false;
}
// Ignore empty sectors
if(ArrayHelpers.ArrayIsNullOrEmpty(data)) return true;
if(ArrayHelpers.ArrayIsNullOrEmpty(data))
return true;
for(uint i = 0; i < length; i++)
{
byte[] tmp = new byte[imageInfo.SectorSize];
Array.Copy(data, i * imageInfo.SectorSize, tmp, 0, imageInfo.SectorSize);
if(!WriteSector(tmp, sectorAddress + i)) return false;
if(!WriteSector(tmp, sectorAddress + i))
return false;
}
ErrorMessage = "";
return true;
}
public bool WriteSectorLong(byte[] data, ulong sectorAddress)
{
ErrorMessage = "Writing sectors with tags is not supported.";
return false;
}
public bool WriteSectorsLong(byte[] data, ulong sectorAddress, uint length)
{
ErrorMessage = "Writing sectors with tags is not supported.";
return false;
}
@@ -250,28 +277,31 @@ namespace Aaru.DiscImages
if(!IsWriting)
{
ErrorMessage = "Image is not opened for writing";
return false;
}
qHdr.mtime = (uint)(DateTime.Now - new DateTime(1970, 1, 1, 0, 0, 0)).TotalSeconds;
writingStream.Seek(0, SeekOrigin.Begin);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.magic), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.version), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.magic), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.version), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.backing_file_offset), 0, 8);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.backing_file_size), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.mtime), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.size), 0, 8);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.backing_file_size), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.mtime), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.size), 0, 8);
writingStream.WriteByte(qHdr.cluster_bits);
writingStream.WriteByte(qHdr.l2_bits);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.padding), 0, 2);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.crypt_method), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.padding), 0, 2);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.crypt_method), 0, 4);
writingStream.Write(BigEndianBitConverter.GetBytes(qHdr.l1_table_offset), 0, 8);
writingStream.Seek((long)qHdr.l1_table_offset, SeekOrigin.Begin);
for(long i = 0; i < l1Table.LongLength; i++) l1Table[i] = Swapping.Swap(l1Table[i]);
byte[] l1TableB =
MemoryMarshal.Cast<ulong, byte>(l1Table).ToArray();
for(long i = 0; i < l1Table.LongLength; i++)
l1Table[i] = Swapping.Swap(l1Table[i]);
byte[] l1TableB = MemoryMarshal.Cast<ulong, byte>(l1Table).ToArray();
writingStream.Write(l1TableB, 0, l1TableB.Length);
writingStream.Flush();
@@ -279,6 +309,7 @@ namespace Aaru.DiscImages
IsWriting = false;
ErrorMessage = "";
return true;
}
@@ -289,12 +320,14 @@ namespace Aaru.DiscImages
public bool WriteSectorTag(byte[] data, ulong sectorAddress, SectorTagType tag)
{
ErrorMessage = "Writing sectors with tags is not supported.";
return false;
}
public bool WriteSectorsTag(byte[] data, ulong sectorAddress, uint length, SectorTagType tag)
{
ErrorMessage = "Writing sectors with tags is not supported.";
return false;
}