/* Install MollenOS */
static void InstallMOS(MfsDisk mDisk)
{
/* Open Disk */
SafeFileHandle handle = OpenDisk(mDisk.DeviceId);
if (handle == null)
{
Console.WriteLine("Failed to open drive");
return;
}
/* Load the mbr, modify it */
Console.WriteLine("Setting up BootCode");
Byte[] BootCode = File.ReadAllBytes("Stage1.bin");
/* Setup Mbr */
Byte[] Mbr = ReadDisk(mDisk, handle, 0, 1);
Buffer.BlockCopy(Mbr, 3, BootCode, 3, 49);
/* Setup Flag to OS-DRIVE (0x1) */
BootCode[8] = 0x1;
/* Write bootloader */
Console.WriteLine("Writing BootCode");
WriteDisk(mDisk, handle, 0, BootCode);
/* Get size of stage2-loader */
Console.WriteLine("Loading Stage 2");
FileStream nLoader = new FileStream("ssbl.stm", FileMode.Open);
Byte[] Stage2Data = new Byte[nLoader.Length];
nLoader.Read(Stage2Data, 0, (int)nLoader.Length);
nLoader.Close();
/* Allocate a new buffer */
Console.WriteLine("Writing Stage 2");
Byte[] ReservedBuffer = new Byte[((Stage2Data.Length / mDisk.BytesPerSector) + 1) * mDisk.BytesPerSector];
Stage2Data.CopyTo(ReservedBuffer, 0);
WriteDisk(mDisk, handle, 1, ReservedBuffer);
/* Find MB Ptr */
UInt64 MbSector = BitConverter.ToUInt64(Mbr, 28);
UInt64 MbMirrorSector = BitConverter.ToUInt64(Mbr, 36);
mDisk.BucketSize = BitConverter.ToUInt16(Mbr, 26);
/* Find Root Ptr in MB */
Byte[] Mb = ReadDisk(mDisk, handle, MbSector, 1);
UInt32 RootBucket = BitConverter.ToUInt32(Mb, 12);
/* Setup directories */
String BaseRoot = "Hdd\\";
Console.WriteLine("Creating system file tree");
Console.WriteLine("Root: " + AppDomain.CurrentDomain.BaseDirectory + "Hdd\\");
String[] Dirs = Directory.GetDirectories("Hdd\\", "*", SearchOption.AllDirectories);
foreach (String pDir in Dirs)
{
String RelPath = pDir.Substring(BaseRoot.Length).Replace('\\', '/');
String DirToCreate = pDir.Split(Path.DirectorySeparatorChar).Last();
Console.WriteLine("Creating: " + RelPath + " (" + DirToCreate + ")");
/* Find free entry */
MfsEntry mEntry = CreateRecursive(mDisk, handle, RootBucket, RelPath);
/* Create the entry */
CreateFileEntry(mDisk, handle, (ushort)(MfsEntryFlags.MFS_DIRECTORY | MfsEntryFlags.MFS_SYSTEM),
0xFFFFFFFF, 0, null, DirToCreate, mEntry.DirBucket);
}
/* Cleanup */
handle.Close();
/* Setup files */
Console.WriteLine("Copying system files");
String[] Files = Directory.GetFiles("Hdd\\", "*.*", SearchOption.AllDirectories);
foreach (String pFile in Files)
{
String RelPath = pFile.Substring(BaseRoot.Length).Replace('\\', '/');
Console.WriteLine("Copying: " + RelPath);
/* Create */
WriteToMfs(mDisk, pFile, RelPath);
}
}
/* Format a drive with mfs */
static void Format(MfsDisk mDisk)
{
/* Open Disk */
SafeFileHandle handle = OpenDisk(mDisk.DeviceId);
if (handle == null)
{
Console.WriteLine("Failed to open drive");
return;
}
/* Print */
Console.WriteLine("Calculating metrics");
/* Calc bytes */
UInt64 DriveSizeBytes = mDisk.TotalSectors * mDisk.BytesPerSector;
UInt64 GigaByte = (1024 * 1024 * 1024);
UInt32 BucketSize = 0;
UInt64 Buckets = 0;
UInt64 BucketMapSize = 0;
UInt64 BucketMapSector = 0;
UInt64 ReservedBuckets = 0;
UInt64 MirrorMasterBucketSector = 0;
/* Determine bucket size
* if <1gb = 1 Kb (2 sectors)
* If <64gb = 4 Kb (8 sectors)
* If >64gb = 8 Kb (16 sectors)
* If >256gb = 16 Kb (32 sectors) */
if (DriveSizeBytes >= (256 * GigaByte))
BucketSize = 32;
else if (DriveSizeBytes >= (64 * GigaByte))
BucketSize = 16;
else if (DriveSizeBytes <= GigaByte)
BucketSize = 2;
else
BucketSize = 8;
/* Print */
mDisk.BucketSize = (UInt16)BucketSize;
Console.WriteLine("Bucket Size: " + BucketSize.ToString());
/* Get size of stage2-loader */
FileStream nLoader = new FileStream("ssbl.stm", FileMode.Open);
Byte[] Stage2Data = new Byte[nLoader.Length];
nLoader.Read(Stage2Data, 0, (int)nLoader.Length);
nLoader.Close();
/* Calculate reserved sector count */
Int64 ReservedSectors = 1 + ((Stage2Data.Length / mDisk.BytesPerSector) + 1);
/* Print */
Console.WriteLine("Reserved Sectors: " + ReservedSectors.ToString());
/* Setup Bucket-list
* SectorCount / BucketSize
* Each bucket must point to the next,
* untill we reach the end of buckets
* Position at end of drive */
Buckets = mDisk.TotalSectors / BucketSize;
BucketMapSize = Buckets * 8; /* One bucket descriptor is 8 bytes */
BucketMapSector = (mDisk.TotalSectors - ((BucketMapSize / mDisk.BytesPerSector) + 1));
MirrorMasterBucketSector = BucketMapSector - 1;
/* Reserve an additional for the mirror bucket */
ReservedBuckets = (((BucketMapSize / mDisk.BytesPerSector) + 1) / BucketSize) + 1;
/* Print */
Console.WriteLine("Bucket Count: " + Buckets.ToString());
Console.WriteLine("Bucket Map Size: " + BucketMapSize.ToString());
Console.WriteLine("Bucket Map Sector: " + BucketMapSector.ToString());
Console.WriteLine("Reserved Buckets: " + ReservedBuckets.ToString());
Console.WriteLine("Free BucketCount: " + ((Buckets - ReservedBuckets) - 1).ToString());
/* Step 1 */
Console.WriteLine("Writing BucketTable");
/* Seek to sector BucketMapSector */
UInt64 ValToMove = BucketMapSector * mDisk.BytesPerSector;
int DistHigh = (int)((ValToMove >> 32) & 0xFFFFFFFF);
uint seekres = SetFilePointer(handle, (int)(ValToMove & 0xFFFFFFFF), ref DistHigh, EMoveMethod.Begin);
Byte[] SectorBuffer = new Byte[mDisk.BytesPerSector];
UInt64 Iterator = 0;
UInt64 Max = Buckets;
UInt64 FreeCount = (Buckets - ReservedBuckets) - 1;
UInt32 FreeBlockItr = 0x7FFFFFFF;
while (Iterator < Max)
{
/* Build sector buffer */
for (UInt64 i = 0; i < (mDisk.BytesPerSector / 8); i++)
{
/* Sanity */
if (Iterator >= ((Max - ReservedBuckets) - 1))
{
SectorBuffer[(i * 8)] = 0xFF;
SectorBuffer[(i * 8) + 1] = 0xFF;
SectorBuffer[(i * 8) + 2] = 0xFF;
SectorBuffer[(i * 8) + 3] = 0xFF;
SectorBuffer[(i * 8) + 4] = 0x01;
SectorBuffer[(i * 8) + 5] = 0x00;
SectorBuffer[(i * 8) + 6] = 0x00;
SectorBuffer[(i * 8) + 7] = 0x00;
}
else
{
SectorBuffer[(i * 8)] = 0xFF;
SectorBuffer[(i * 8) + 1] = 0xFF;
SectorBuffer[(i * 8) + 2] = 0xFF;
SectorBuffer[(i * 8) + 3] = 0xFF;
SectorBuffer[(i * 8) + 4] = (Byte)((FreeCount) & 0xFF);
SectorBuffer[(i * 8) + 5] = (Byte)(((FreeCount) >> 8) & 0xFF);
SectorBuffer[(i * 8) + 6] = (Byte)(((FreeCount) >> 16) & 0xFF);
SectorBuffer[(i * 8) + 7] = (Byte)(((FreeCount) >> 24) & 0xFF);
/* Decrease */
FreeCount--;
FreeBlockItr--;
/* Sanity */
if (FreeBlockItr == 0)
FreeBlockItr = 0x7FFFFFFF;
}
/* Go to next */
Iterator++;
}
/* Write sector */
uint bWritten = 0;
WriteFile(handle.DangerousGetHandle(), SectorBuffer, (uint)SectorBuffer.Length, out bWritten, IntPtr.Zero);
}
/* Calculate how many start-buckets we shall skip because of
* reserved sectors */
Int64 SkipBuckets = (ReservedSectors / BucketSize) + 2;
Int64 MasterBucketSector = ReservedSectors + 1;
/* Step 2 */
Console.WriteLine("Writing MasterBucket Original & Mirror");
Console.WriteLine("Bucket Original Sector: " + MasterBucketSector.ToString());
Console.WriteLine("Bucket Mirror Sector: " + MirrorMasterBucketSector.ToString());
/* Setup the MasterBucket structure */
UInt32 BucketStartFree = 0;
while (BucketStartFree < ((MasterBucketSector / BucketSize) + 1))
BucketStartFree++;
Console.WriteLine("First Free Bucket: " + BucketStartFree.ToString());
UInt32 InitialBucketSize = 0;
UInt32 RootIndex = BucketStartFree;
BucketStartFree = AllocateBucket(mDisk, handle, BucketStartFree, 1, out InitialBucketSize);
UInt32 BadBucketIndex = BucketStartFree;
BucketStartFree = AllocateBucket(mDisk, handle, BucketStartFree, 1, out InitialBucketSize);
UInt32 JournalIndex = BucketStartFree;
BucketStartFree = AllocateBucket(mDisk, handle, BucketStartFree, 8, out InitialBucketSize);
Console.WriteLine("First Free Bucket after initial: " + BucketStartFree.ToString());
/* Write it to the two sectors */
Byte[] MasterBucket = new Byte[512];
/* Zero It */
for (int i = 0; i < 512; i++)
MasterBucket[i] = 0;
/* Setup Magic */
MasterBucket[0] = 0x4D;
MasterBucket[1] = 0x46;
MasterBucket[2] = 0x53;
MasterBucket[3] = 0x31;
/* Setup flags */
MasterBucket[4] = 0;
MasterBucket[5] = 0;
MasterBucket[6] = 0;
MasterBucket[7] = 0;
/* Setup free bucket pointer */
MasterBucket[8] = (Byte)(BucketStartFree & 0xFF);
MasterBucket[9] = (Byte)((BucketStartFree >> 8) & 0xFF);
MasterBucket[10] = (Byte)((BucketStartFree >> 16) & 0xFF);
MasterBucket[11] = (Byte)((BucketStartFree >> 24) & 0xFF);
/* Setup Root Index */
MasterBucket[12] = (Byte)(RootIndex & 0xFF);
MasterBucket[13] = (Byte)((RootIndex >> 8) & 0xFF);
MasterBucket[14] = (Byte)((RootIndex >> 16) & 0xFF);
MasterBucket[15] = (Byte)((RootIndex >> 24) & 0xFF);
/* Setup Bad Bucket Index */
MasterBucket[16] = (Byte)(BadBucketIndex & 0xFF);
MasterBucket[17] = (Byte)((BadBucketIndex >> 8) & 0xFF);
MasterBucket[18] = (Byte)((BadBucketIndex >> 16) & 0xFF);
MasterBucket[19] = (Byte)((BadBucketIndex >> 24) & 0xFF);
/* Setup Journal File Index */
MasterBucket[20] = (Byte)(JournalIndex & 0xFF);
MasterBucket[21] = (Byte)((JournalIndex >> 8) & 0xFF);
MasterBucket[22] = (Byte)((JournalIndex >> 16) & 0xFF);
MasterBucket[23] = (Byte)((JournalIndex >> 24) & 0xFF);
/* Setup bucketmap sector */
MasterBucket[24] = (Byte)(BucketMapSector & 0xFF);
MasterBucket[25] = (Byte)((BucketMapSector >> 8) & 0xFF);
MasterBucket[26] = (Byte)((BucketMapSector >> 16) & 0xFF);
MasterBucket[27] = (Byte)((BucketMapSector >> 24) & 0xFF);
MasterBucket[28] = (Byte)((BucketMapSector >> 32) & 0xFF);
MasterBucket[29] = (Byte)((BucketMapSector >> 40) & 0xFF);
MasterBucket[30] = (Byte)((BucketMapSector >> 48) & 0xFF);
MasterBucket[31] = (Byte)((BucketMapSector >> 56) & 0xFF);
/* Setup bucketmap size */
MasterBucket[32] = (Byte)(BucketMapSize & 0xFF);
MasterBucket[33] = (Byte)((BucketMapSize >> 8) & 0xFF);
MasterBucket[34] = (Byte)((BucketMapSize >> 16) & 0xFF);
MasterBucket[35] = (Byte)((BucketMapSize >> 24) & 0xFF);
MasterBucket[36] = (Byte)((BucketMapSize >> 32) & 0xFF);
MasterBucket[37] = (Byte)((BucketMapSize >> 40) & 0xFF);
MasterBucket[38] = (Byte)((BucketMapSize >> 48) & 0xFF);
MasterBucket[39] = (Byte)((BucketMapSize >> 56) & 0xFF);
/* Seek */
WriteDisk(mDisk, handle, (UInt64)MirrorMasterBucketSector, MasterBucket);
WriteDisk(mDisk, handle, (UInt64)MasterBucketSector, MasterBucket);
/* Wipe new sectors */
Byte[] Wipe = new Byte[BucketSize * mDisk.BytesPerSector];
for (int i = 0; i < Wipe.Length; i++)
Wipe[i] = 0;
Console.WriteLine("Wiping Root");
WriteDisk(mDisk, handle, (RootIndex * BucketSize), Wipe);
Console.WriteLine("Wiping Bad Bucket List");
WriteDisk(mDisk, handle, (BadBucketIndex * BucketSize), Wipe);
Wipe = new Byte[(BucketSize * mDisk.BytesPerSector) * 8];
for (int i = 0; i < Wipe.Length; i++)
Wipe[i] = 0;
Console.WriteLine("Wiping Journal");
WriteDisk(mDisk, handle, (JournalIndex * BucketSize), Wipe);
/* Step 3 */
Console.WriteLine("Writing Mbr");
/* Load the mbr, modify it */
Byte[] Mbr = new Byte[mDisk.BytesPerSector];
for (int c = 0; c < mDisk.BytesPerSector; c++)
Mbr[c] = 0;
/* Set magic */
Mbr[3] = 0x4D;
Mbr[4] = 0x46;
Mbr[5] = 0x53;
Mbr[6] = 0x31;
/* Set version */
Mbr[7] = 0x1;
/* Set flags */
Mbr[8] = 0;
/* Disk metrics */
Mbr[9] = 0x80;
Mbr[10] = (Byte)(mDisk.BytesPerSector & 0xFF);
Mbr[11] = (Byte)((mDisk.BytesPerSector >> 8) & 0xFF);
/* Sectors Per Track */
Mbr[12] = (Byte)(mDisk.SectorsPerTrack & 0xFF);
Mbr[13] = (Byte)((mDisk.SectorsPerTrack >> 8) & 0xFF);
/* Heads Per Cylinder */
Mbr[14] = (Byte)(mDisk.TracksPerCylinder & 0xFF);
Mbr[15] = (Byte)((mDisk.TracksPerCylinder >> 8) & 0xFF);
Mbr[16] = (Byte)(mDisk.TotalSectors & 0xFF);
Mbr[17] = (Byte)((mDisk.TotalSectors >> 8) & 0xFF);
Mbr[18] = (Byte)((mDisk.TotalSectors >> 16) & 0xFF);
Mbr[19] = (Byte)((mDisk.TotalSectors >> 24) & 0xFF);
Mbr[20] = (Byte)((mDisk.TotalSectors >> 32) & 0xFF);
Mbr[21] = (Byte)((mDisk.TotalSectors >> 40) & 0xFF);
Mbr[22] = (Byte)((mDisk.TotalSectors >> 48) & 0xFF);
Mbr[23] = (Byte)((mDisk.TotalSectors >> 56) & 0xFF);
Mbr[24] = (Byte)(ReservedSectors & 0xFF);
Mbr[25] = (Byte)((ReservedSectors >> 8) & 0xFF);
Mbr[26] = (Byte)(BucketSize & 0xFF);
Mbr[27] = (Byte)((BucketSize >> 8) & 0xFF);
Mbr[28] = (Byte)(MasterBucketSector & 0xFF);
Mbr[29] = (Byte)((MasterBucketSector >> 8) & 0xFF);
Mbr[30] = (Byte)((MasterBucketSector >> 16) & 0xFF);
Mbr[31] = (Byte)((MasterBucketSector >> 24) & 0xFF);
Mbr[32] = (Byte)((MasterBucketSector >> 32) & 0xFF);
Mbr[33] = (Byte)((MasterBucketSector >> 40) & 0xFF);
Mbr[34] = (Byte)((MasterBucketSector >> 48) & 0xFF);
Mbr[35] = (Byte)((MasterBucketSector >> 56) & 0xFF);
Mbr[36] = (Byte)(MirrorMasterBucketSector & 0xFF);
Mbr[37] = (Byte)((MirrorMasterBucketSector >> 8) & 0xFF);
Mbr[38] = (Byte)((MirrorMasterBucketSector >> 16) & 0xFF);
Mbr[39] = (Byte)((MirrorMasterBucketSector >> 24) & 0xFF);
Mbr[40] = (Byte)((MirrorMasterBucketSector >> 32) & 0xFF);
Mbr[41] = (Byte)((MirrorMasterBucketSector >> 40) & 0xFF);
Mbr[42] = (Byte)((MirrorMasterBucketSector >> 48) & 0xFF);
Mbr[43] = (Byte)((MirrorMasterBucketSector >> 56) & 0xFF);
/* Set volume name */
Mbr[44] = (Byte)'M';
Mbr[45] = (Byte)'o';
Mbr[46] = (Byte)'l';
Mbr[47] = (Byte)'l';
Mbr[48] = (Byte)'e';
Mbr[49] = (Byte)'n';
Mbr[50] = (Byte)'O';
Mbr[51] = (Byte)'S';
/* Write it */
WriteDisk(mDisk, handle, 0, Mbr);
/* Done */
handle.Close();
}
/* Locate next bucket */
static UInt32 GetNextBucket(MfsDisk mDisk, SafeFileHandle hDisk, UInt32 Bucket, out UInt32 BucketLength)
{
/* Calculate Bucket Map */
UInt64 Buckets = mDisk.TotalSectors / mDisk.BucketSize;
UInt64 BucketMapSize = Buckets * 8; /* One bucket descriptor is 8 bytes */
UInt64 BucketMapSector = (mDisk.TotalSectors - ((BucketMapSize / mDisk.BytesPerSector) + 1));
/* Calculate Index */
UInt32 BucketsPerSector = mDisk.BytesPerSector / 8;
UInt32 SectorOffset = Bucket / BucketsPerSector;
UInt32 SectorIndex = Bucket % BucketsPerSector;
/* Read sector */
Byte[] Sector = ReadDisk(mDisk, hDisk, BucketMapSector + SectorOffset, 1);
/* Done */
BucketLength = BitConverter.ToUInt32(Sector, (int)((SectorIndex * 8) + 4));
return BitConverter.ToUInt32(Sector, (int)(SectorIndex * 8));
}
/* Create Recursive */
static MfsEntry CreateRecursive(MfsDisk mDisk, SafeFileHandle hDisk, UInt32 DirBucket, String pPath)
{
/* Sanity, if start with "/" skip */
String mPath = pPath;
if (mPath.StartsWith("/"))
mPath = mPath.Substring(1, mPath.Length - 1);
if (mPath.EndsWith("/"))
mPath = mPath.Substring(0, mPath.Length - 1);
/* Get token */
int iDex = mPath.IndexOf("/");
String LookFor = mPath.Substring(0, iDex == -1 ? mPath.Length : iDex);
/* EoP */
if (String.IsNullOrEmpty(LookFor)
|| iDex == -1)
{
/* List files */
UInt32 IteratorBucket = DirBucket;
UInt32 PrevItrBucket = 0;
int End = 0;
while (End == 0)
{
UInt64 Sector = IteratorBucket * mDisk.BucketSize;
/* Gogo */
Byte[] fBuffer = ReadDisk(mDisk, hDisk, Sector, mDisk.BucketSize);
for (int i = 0; i < (mDisk.BucketSize * mDisk.BytesPerSector); i++)
{
/* EoC? */
if (fBuffer[i] == 0)
{
End = 1;
break;
}
/* Woah, parse */
int Len = 0;
while (fBuffer[i + 76 + Len] != 0)
Len++;
String Name = System.Text.Encoding.UTF8.GetString(fBuffer, i + 76, Len);
UInt16 Flags = BitConverter.ToUInt16(fBuffer, i + 2);
if (Name.ToLower() == LookFor.ToLower())
return null;
/* Next */
i += 1023;
}
/* Sanity */
if (End == 1)
break;
/* Get next bucket */
UInt32 DirBucketLength = 0;
PrevItrBucket = IteratorBucket;
if (End == 0)
IteratorBucket = GetNextBucket(mDisk, hDisk, IteratorBucket, out DirBucketLength);
/* End of list? */
if (IteratorBucket == 0xFFFFFFFF)
End = 1;
}
/* We must reach this point */
MfsEntry nEntry = new MfsEntry();
nEntry.DirBucket = IteratorBucket == 0xFFFFFFFF ? PrevItrBucket : IteratorBucket;
nEntry.DirIndex = 0;
/* Done */
return nEntry;
}
else
{
/* Find LookFor in DirBucket */
UInt32 IteratorBucket = DirBucket;
int End = 0;
while (End == 0)
{
UInt64 Sector = IteratorBucket * mDisk.BucketSize;
/* Gogo */
Byte[] fBuffer = ReadDisk(mDisk, hDisk, Sector, mDisk.BucketSize);
for (int i = 0; i < (mDisk.BucketSize * mDisk.BytesPerSector); i++)
{
/* EoC? */
if (fBuffer[i] == 0)
{
End = 1;
break;
}
/* Woah, parse */
int Len = 0;
while (fBuffer[i + 76 + Len] != 0)
Len++;
String Name = System.Text.Encoding.UTF8.GetString(fBuffer, i + 76, Len);
UInt16 Flags = BitConverter.ToUInt16(fBuffer, i + 2);
if (Name.ToLower() == LookFor.ToLower())
{
/* More sanity */
if (!((MfsEntryFlags)Flags).HasFlag(MfsEntryFlags.MFS_DIRECTORY))
{
Console.WriteLine(LookFor + " is not a directory");
return null;
}
/* Match */
MfsEntry nEntry = new MfsEntry();
nEntry.Name = Name;
nEntry.Size = BitConverter.ToUInt64(fBuffer, i + 60);
nEntry.AllocatedSize = BitConverter.ToUInt64(fBuffer, i + 68);
nEntry.Bucket = BitConverter.ToUInt32(fBuffer, i + 4);
nEntry.BucketLength = BitConverter.ToUInt32(fBuffer, i + 8);
nEntry.DirBucket = IteratorBucket;
nEntry.DirIndex = (uint)i;
/* Sanity */
if (nEntry.Bucket == 0xFFFFFFFF)
{
Byte[] Mbr = ReadDisk(mDisk, hDisk, 0, 1);
/* Find MB Ptr */
UInt64 MbSector = BitConverter.ToUInt64(Mbr, 28);
UInt64 MbMirrorSector = BitConverter.ToUInt64(Mbr, 36);
/* Find Root Ptr in MB */
Byte[] Mb = ReadDisk(mDisk, hDisk, MbSector, 1);
UInt32 FreeBucket = BitConverter.ToUInt32(Mb, 8);
/* Allocate */
nEntry.Bucket = FreeBucket;
UInt32 InitialBucketSize = 0;
UInt32 NextFree = AllocateBucket(mDisk, hDisk, FreeBucket, 1, out InitialBucketSize);
/* Update Mb */
Mb[8] = (Byte)(NextFree & 0xFF);
Mb[9] = (Byte)((NextFree >> 8) & 0xFF);
Mb[10] = (Byte)((NextFree >> 16) & 0xFF);
Mb[11] = (Byte)((NextFree >> 24) & 0xFF);
/* Write Mb */
WriteDisk(mDisk, hDisk, MbSector, Mb);
WriteDisk(mDisk, hDisk, MbMirrorSector, Mb);
/* Update Dir */
fBuffer[i + 4] = (Byte)(nEntry.Bucket & 0xFF);
fBuffer[i + 5] = (Byte)((nEntry.Bucket >> 8) & 0xFF);
fBuffer[i + 6] = (Byte)((nEntry.Bucket >> 16) & 0xFF);
fBuffer[i + 7] = (Byte)((nEntry.Bucket >> 24) & 0xFF);
fBuffer[i + 8] = 0x01;
fBuffer[i + 9] = 0x00;
fBuffer[i + 10] = 0x00;
fBuffer[i + 11] = 0x00;
WriteDisk(mDisk, hDisk, Sector, fBuffer);
/* Wipe bucket */
Byte[] Wipe = new Byte[mDisk.BucketSize * mDisk.BytesPerSector];
for (int g = 0; g < Wipe.Length; g++)
Wipe[g] = 0;
WriteDisk(mDisk, hDisk, (nEntry.Bucket * mDisk.BucketSize), Wipe);
}
/* Done */
return CreateRecursive(mDisk, hDisk, nEntry.Bucket, mPath.Substring(LookFor.Length));
}
/* Next */
i += 1023;
}
/* Get next bucket */
UInt32 DirBucketLength = 0;
if (End == 0)
IteratorBucket = GetNextBucket(mDisk, hDisk, IteratorBucket, out DirBucketLength);
/* CAN not be end of list? */
if (IteratorBucket == 0xFFFFFFFF)
End = 1;
}
}
return null;
}
/* Fill bucketchain with data */
static void FillBucketChain(MfsDisk mDisk, SafeFileHandle hDisk, UInt32 Bucket, UInt32 BucketLength, Byte[] Data)
{
/* Index */
Int64 Index = 0;
UInt32 BucketPtr = Bucket;
UInt32 BucketLengthItr = BucketLength;
/* Iterate */
while (Index < Data.LongLength)
{
/* Alloc Buffer */
Byte[] Buffer = new Byte[(mDisk.BucketSize * mDisk.BytesPerSector) * BucketLengthItr];
/* Copy */
for (int i = 0; i < Buffer.Length; i++)
{
if (Index + i >= Data.LongLength)
Buffer[i] = 0;
else
Buffer[i] = Data[Index + i];
}
/* Inc */
Index += Buffer.Length;
/* Calculate target sector */
UInt64 Sector = mDisk.BucketSize * BucketPtr;
WriteDisk(mDisk, hDisk, Sector, Buffer);
/* Get next bucket */
BucketPtr = GetNextBucket(mDisk, hDisk, BucketPtr, out BucketLengthItr);
/* Sanity */
if (BucketPtr == 0xFFFFFFFF)
break;
}
}
/* Write file */
static void WriteToMfs(MfsDisk mDisk, String pFile, String lPath)
{
/* Open Disk */
SafeFileHandle handle = OpenDisk(mDisk.DeviceId);
if (handle == null)
{
Console.WriteLine("Failed to open drive");
return;
}
/* Read MBR */
Byte[] Mbr = ReadDisk(mDisk, handle, 0, 1);
/* Find MB Ptr */
UInt64 MbSector = BitConverter.ToUInt64(Mbr, 28);
UInt64 MbMirrorSector = BitConverter.ToUInt64(Mbr, 36);
mDisk.BucketSize = BitConverter.ToUInt16(Mbr, 26);
/* Find Root Ptr in MB */
Byte[] Mb = ReadDisk(mDisk, handle, MbSector, 1);
UInt32 FreeBucket = BitConverter.ToUInt32(Mb, 8);
UInt32 RootBucket = BitConverter.ToUInt32(Mb, 12);
/* Load File */
Console.WriteLine("Reading source data");
Byte[] FileData = File.ReadAllBytes(pFile);
/* Can we even write a file there */
MfsEntry nEntry = ListRecursive(mDisk, handle, RootBucket, lPath);
if (nEntry != null)
{
Console.WriteLine("File exists in table, updating");
/* Sanity */
if ((UInt64)FileData.LongLength > nEntry.AllocatedSize)
{
/* Allocate more */
UInt64 NumSectors = ((UInt64)FileData.LongLength - nEntry.AllocatedSize) / mDisk.BytesPerSector;
if ((((UInt64)FileData.LongLength - nEntry.AllocatedSize) % mDisk.BytesPerSector) > 0)
NumSectors++;
UInt64 NumBuckets = NumSectors / mDisk.BucketSize;
if ((NumSectors % mDisk.BucketSize) > 0)
NumBuckets++;
/* Allocate buckets */
UInt32 StartBucket = FreeBucket;
UInt32 InitialBucketSize = 0;
FreeBucket = AllocateBucket(mDisk, handle, FreeBucket, NumBuckets, out InitialBucketSize);
/* Get last bucket in chain */
UInt32 BucketPtr = nEntry.Bucket;
UInt32 BucketPrevPtr = 0;
UInt32 BucketLength = 0;
while (BucketPtr != 0xFFFFFFFF)
{
BucketPrevPtr = BucketPtr;
BucketPtr = GetNextBucket(mDisk, handle, BucketPtr, out BucketLength);
}
/* Update pointer */
SetNextBucket(mDisk, handle, BucketPrevPtr, FreeBucket);
/* Update MB */
Mb[8] = (Byte)(FreeBucket & 0xFF);
Mb[9] = (Byte)((FreeBucket >> 8) & 0xFF);
Mb[10] = (Byte)((FreeBucket >> 16) & 0xFF);
Mb[11] = (Byte)((FreeBucket >> 24) & 0xFF);
/* Write Mb */
WriteDisk(mDisk, handle, MbSector, Mb);
WriteDisk(mDisk, handle, MbMirrorSector, Mb);
/* Adjust */
nEntry.AllocatedSize += (NumBuckets * mDisk.BucketSize * mDisk.BytesPerSector);
}
/* If file size is drastically less, free some buckets */
/* Write Data */
Console.WriteLine("Updating Data");
FillBucketChain(mDisk, handle, nEntry.Bucket, nEntry.BucketLength, FileData);
/* Update entry with new sizes, new dates, new times */
Byte[] fBuffer = ReadDisk(mDisk, handle, (nEntry.DirBucket * mDisk.BucketSize), mDisk.BucketSize);
/* Modify */
fBuffer[nEntry.DirIndex + 4] = (Byte)(nEntry.Bucket & 0xFF);
fBuffer[nEntry.DirIndex + 5] = (Byte)((nEntry.Bucket >> 8) & 0xFF);
fBuffer[nEntry.DirIndex + 6] = (Byte)((nEntry.Bucket >> 16) & 0xFF);
fBuffer[nEntry.DirIndex + 7] = (Byte)((nEntry.Bucket >> 24) & 0xFF);
fBuffer[nEntry.DirIndex + 8] = (Byte)(nEntry.BucketLength & 0xFF);
fBuffer[nEntry.DirIndex + 9] = (Byte)((nEntry.BucketLength >> 8) & 0xFF);
fBuffer[nEntry.DirIndex + 10] = (Byte)((nEntry.BucketLength >> 16) & 0xFF);
fBuffer[nEntry.DirIndex + 11] = (Byte)((nEntry.BucketLength >> 24) & 0xFF);
fBuffer[nEntry.DirIndex + 60] = (Byte)(FileData.LongLength & 0xFF);
fBuffer[nEntry.DirIndex + 61] = (Byte)((FileData.LongLength >> 8) & 0xFF);
fBuffer[nEntry.DirIndex + 62] = (Byte)((FileData.LongLength >> 16) & 0xFF);
fBuffer[nEntry.DirIndex + 63] = (Byte)((FileData.LongLength >> 24) & 0xFF);
fBuffer[nEntry.DirIndex + 64] = (Byte)((FileData.LongLength >> 32) & 0xFF);
fBuffer[nEntry.DirIndex + 65] = (Byte)((FileData.LongLength >> 40) & 0xFF);
fBuffer[nEntry.DirIndex + 66] = (Byte)((FileData.LongLength >> 48) & 0xFF);
fBuffer[nEntry.DirIndex + 67] = (Byte)((FileData.LongLength >> 56) & 0xFF);
fBuffer[nEntry.DirIndex + 68] = (Byte)(nEntry.AllocatedSize & 0xFF);
fBuffer[nEntry.DirIndex + 69] = (Byte)((nEntry.AllocatedSize >> 8) & 0xFF);
fBuffer[nEntry.DirIndex + 70] = (Byte)((nEntry.AllocatedSize >> 16) & 0xFF);
fBuffer[nEntry.DirIndex + 71] = (Byte)((nEntry.AllocatedSize >> 24) & 0xFF);
fBuffer[nEntry.DirIndex + 72] = (Byte)((nEntry.AllocatedSize >> 32) & 0xFF);
fBuffer[nEntry.DirIndex + 73] = (Byte)((nEntry.AllocatedSize >> 40) & 0xFF);
fBuffer[nEntry.DirIndex + 74] = (Byte)((nEntry.AllocatedSize >> 48) & 0xFF);
fBuffer[nEntry.DirIndex + 75] = (Byte)((nEntry.AllocatedSize >> 56) & 0xFF);
/* Write back */
WriteDisk(mDisk, handle, (nEntry.DirBucket * mDisk.BucketSize), fBuffer);
}
else
{
Console.WriteLine("/" + Path.GetFileName(pFile) + " is new, creating");
MfsEntry cInfo = CreateRecursive(mDisk, handle, RootBucket, lPath);
if (cInfo == null)
{
Console.WriteLine("The creation info returned null, somethings wrong");
return;
}
/* Get first free bucket again, could have changed after CreateRecursive */
Mb = ReadDisk(mDisk, handle, MbSector, 1);
FreeBucket = BitConverter.ToUInt32(Mb, 8);
/* Get first free bucket */
UInt64 NumBuckets = (UInt64)(FileData.LongLength / mDisk.BytesPerSector) / mDisk.BucketSize;
if (((FileData.LongLength / mDisk.BytesPerSector) % mDisk.BucketSize) > 0)
NumBuckets++;
/* Allocate a chain */
Console.WriteLine("Allocing " + NumBuckets.ToString() + " buckets, old free ptr " + FreeBucket.ToString());
UInt32 StartBucket = FreeBucket;
UInt32 InitialBucketSize = 0;
FreeBucket = AllocateBucket(mDisk, handle, FreeBucket, NumBuckets, out InitialBucketSize);
Console.WriteLine("Done, new free pointer " + FreeBucket.ToString());
/* Update MB */
Mb[8] = (Byte)(FreeBucket & 0xFF);
Mb[9] = (Byte)((FreeBucket >> 8) & 0xFF);
Mb[10] = (Byte)((FreeBucket >> 16) & 0xFF);
Mb[11] = (Byte)((FreeBucket >> 24) & 0xFF);
/* Write Mb */
WriteDisk(mDisk, handle, MbSector, Mb);
WriteDisk(mDisk, handle, MbMirrorSector, Mb);
/* Create entry */
Console.WriteLine("Creating entry in path");
CreateFileEntry(mDisk, handle, (ushort)(MfsEntryFlags.MFS_FILE | MfsEntryFlags.MFS_SYSTEM | MfsEntryFlags.MFS_SECURITY),
StartBucket, InitialBucketSize, FileData, Path.GetFileName(lPath), cInfo.DirBucket);
Console.WriteLine("Done");
/* Write Data */
Console.WriteLine("Writing Data");
FillBucketChain(mDisk, handle, StartBucket, InitialBucketSize, FileData);
}
/* Done */
Console.WriteLine("File Creation Done");
handle.Close();
}
/* Create entry in Mb */
static void CreateFileEntry(MfsDisk mDisk, SafeFileHandle hDisk, UInt16 Flags, UInt32 Bucket, UInt32 BucketLength, Byte[] Data, String Name, UInt32 DirBucket)
{
/* List files */
UInt32 IteratorBucket = DirBucket;
int End = 0;
while (End == 0)
{
UInt64 Sector = IteratorBucket * mDisk.BucketSize;
/* Gogo */
Byte[] fBuffer = ReadDisk(mDisk, hDisk, Sector, mDisk.BucketSize);
for (int i = 0; i < (mDisk.BucketSize * mDisk.BytesPerSector); i++)
{
/* EoC? */
if (fBuffer[i] == 0)
{
UInt64 NumBuckets = 0;
UInt64 AllocatedSize = 0;
UInt64 DataLen = 0;
if (Data != null)
{
DataLen = (UInt64)Data.LongLength;
NumBuckets = (UInt64)(Data.LongLength / mDisk.BytesPerSector) / mDisk.BucketSize;
if (((Data.LongLength / mDisk.BytesPerSector) % mDisk.BucketSize) > 0)
NumBuckets++;
AllocatedSize = NumBuckets * mDisk.BucketSize * mDisk.BytesPerSector;
}
/* Setup Status */
fBuffer[i] = 0x1;
fBuffer[i + 1] = 0x0;
/* Setup flags */
fBuffer[i + 2] = (Byte)(Flags & 0xFF);
fBuffer[i + 3] = (Byte)((Flags >> 8) & 0xFF);
/* Setup start bucket */
fBuffer[i + 4] = (Byte)(Bucket & 0xFF);
fBuffer[i + 5] = (Byte)((Bucket >> 8) & 0xFF);
fBuffer[i + 6] = (Byte)((Bucket >> 16) & 0xFF);
fBuffer[i + 7] = (Byte)((Bucket >> 24) & 0xFF);
fBuffer[i + 8] = (Byte)(BucketLength & 0xFF);
fBuffer[i + 9] = (Byte)((BucketLength >> 8) & 0xFF);
fBuffer[i + 10] = (Byte)((BucketLength >> 16) & 0xFF);
fBuffer[i + 11] = (Byte)((BucketLength >> 24) & 0xFF);
/* Ignore time and date for now */
fBuffer[i + 60] = (Byte)(DataLen & 0xFF);
fBuffer[i + 61] = (Byte)((DataLen >> 8) & 0xFF);
fBuffer[i + 62] = (Byte)((DataLen >> 16) & 0xFF);
fBuffer[i + 63] = (Byte)((DataLen >> 24) & 0xFF);
fBuffer[i + 64] = (Byte)((DataLen >> 32) & 0xFF);
fBuffer[i + 65] = (Byte)((DataLen >> 40) & 0xFF);
fBuffer[i + 66] = (Byte)((DataLen >> 48) & 0xFF);
fBuffer[i + 67] = (Byte)((DataLen >> 56) & 0xFF);
fBuffer[i + 68] = (Byte)(AllocatedSize & 0xFF);
fBuffer[i + 69] = (Byte)((AllocatedSize >> 8) & 0xFF);
fBuffer[i + 70] = (Byte)((AllocatedSize >> 16) & 0xFF);
fBuffer[i + 71] = (Byte)((AllocatedSize >> 24) & 0xFF);
fBuffer[i + 72] = (Byte)((AllocatedSize >> 32) & 0xFF);
fBuffer[i + 73] = (Byte)((AllocatedSize >> 40) & 0xFF);
fBuffer[i + 74] = (Byte)((AllocatedSize >> 48) & 0xFF);
fBuffer[i + 75] = (Byte)((AllocatedSize >> 56) & 0xFF);
/* Setup Name */
Byte[] NameData = System.Text.Encoding.UTF8.GetBytes(Name);
for (int j = 0; j < NameData.Length; j++)
fBuffer[i + 76 + j] = NameData[j];
/* Don't use rest for now */
WriteDisk(mDisk, hDisk, Sector, fBuffer);
/* Done */
return;
}
/* Next */
i += 1023;
}
/* Get next bucket */
UInt32 DirBucketLength = 0;
UInt32 PrevDirBucket = IteratorBucket;
if (End == 0)
IteratorBucket = GetNextBucket(mDisk, hDisk, IteratorBucket, out DirBucketLength);
/* End of list? */
if (IteratorBucket == 0xFFFFFFFF)
{
/* Read MBR */
Console.WriteLine("Expanding directory");
Byte[] Mbr = ReadDisk(mDisk, hDisk, 0, 1);
/* Find MB Ptr */
UInt64 MbSector = BitConverter.ToUInt64(Mbr, 28);
UInt64 MbMirrorSector = BitConverter.ToUInt64(Mbr, 36);
/* Find Root Ptr in MB */
Byte[] Mb = ReadDisk(mDisk, hDisk, MbSector, 1);
UInt32 FreeBucket = BitConverter.ToUInt32(Mb, 8);
/* Allocate bucket */
Console.WriteLine("New bucket: " + FreeBucket.ToString());
UInt32 NextFreeBucket = AllocateBucket(mDisk, hDisk, FreeBucket, 1, out DirBucketLength);
Console.WriteLine("Next free bucket: " + NextFreeBucket.ToString());
/* Extend directory */
SetNextBucket(mDisk, hDisk, PrevDirBucket, FreeBucket);
/* Update Mb(s) */
Mb[8] = (Byte)(NextFreeBucket & 0xFF);
Mb[9] = (Byte)((NextFreeBucket >> 8) & 0xFF);
Mb[10] = (Byte)((NextFreeBucket >> 16) & 0xFF);
Mb[11] = (Byte)((NextFreeBucket >> 24) & 0xFF);
WriteDisk(mDisk, hDisk, MbSector, Mb);
WriteDisk(mDisk, hDisk, MbMirrorSector, Mb);
/* Wipe the bucket */
Byte[] Wipe = new Byte[mDisk.BucketSize * mDisk.BytesPerSector];
for (int i = 0; i < Wipe.Length; i++)
Wipe[i] = 0;
Console.WriteLine("Wiping new bucket");
WriteDisk(mDisk, hDisk, (FreeBucket * mDisk.BucketSize), Wipe);
/* Update IteratorBucket */
IteratorBucket = FreeBucket;
}
}
}
/* Update bucket ptr */
static void SetNextBucket(MfsDisk mDisk, SafeFileHandle hDisk, UInt32 Bucket, UInt32 NextBucket)
{
/* Calculate Bucket Map */
UInt64 Buckets = mDisk.TotalSectors / mDisk.BucketSize;
UInt64 BucketMapSize = Buckets * 8; /* One bucket descriptor is 8 bytes */
UInt64 BucketMapSector = (mDisk.TotalSectors - ((BucketMapSize / mDisk.BytesPerSector) + 1));
/* Calculate Index */
UInt32 BucketsPerSector = mDisk.BytesPerSector / 8;
UInt32 SectorOffset = Bucket / BucketsPerSector;
UInt32 SectorIndex = Bucket % BucketsPerSector;
/* Read sector */
Byte[] Sector = ReadDisk(mDisk, hDisk, BucketMapSector + SectorOffset, 1);
/* Edit */
Sector[SectorIndex * 8] = (Byte)(NextBucket & 0xFF);
Sector[SectorIndex * 8 + 1] = (Byte)((NextBucket >> 8) & 0xFF);
Sector[SectorIndex * 8 + 2] = (Byte)((NextBucket >> 16) & 0xFF);
Sector[SectorIndex * 8 + 3] = (Byte)((NextBucket >> 24) & 0xFF);
/* Write */
WriteDisk(mDisk, hDisk, BucketMapSector + SectorOffset, Sector);
}
/* Write */
static void WriteDisk(MfsDisk mDisk, SafeFileHandle Disk, UInt64 Sector, Byte[] Buffer)
{
uint BytesWritten = 0;
UInt64 ValToMove = (Sector * mDisk.BytesPerSector);
int DistHigh = (int)((ValToMove >> 32) & 0xFFFFFFFF);
SetFilePointer(Disk, (int)(ValToMove & 0xFFFFFFFF), ref DistHigh, EMoveMethod.Begin);
WriteFile(Disk.DangerousGetHandle(), Buffer, (uint)Buffer.Length, out BytesWritten, IntPtr.Zero);
}
/* Read */
static Byte[] ReadDisk(MfsDisk mDisk, SafeFileHandle Disk, UInt64 Sector, UInt64 SectorCount)
{
/* Alloc buffer */
Byte[] RetBuf = new Byte[SectorCount * mDisk.BytesPerSector];
/* Seek */
UInt64 ValToMove = (Sector * mDisk.BytesPerSector);
int DistHigh = (int)((ValToMove >> 32) & 0xFFFFFFFF);
SetFilePointer(Disk, (int)(ValToMove & 0xFFFFFFFF), ref DistHigh, EMoveMethod.Begin);
/* Read */
uint bRead = 0;
ReadFile(Disk.DangerousGetHandle(), RetBuf, (uint)RetBuf.Length, out bRead, IntPtr.Zero);
/* Done */
return RetBuf;
}
/* Entry */
static void Main(string[] args)
{
/* Print Header */
Console.WriteLine("MFS Utility Software");
Console.WriteLine("Software Capabilities include formatting, read, write to/from MFS.\n");
/* Get a list of physical drives */
Hashtable Drives = new Hashtable();
Console.WriteLine("Available Drives:");
WqlObjectQuery q = new WqlObjectQuery("SELECT * FROM Win32_DiskDrive");
ManagementObjectSearcher res = new ManagementObjectSearcher(q);
int Itr = 0;
foreach (ManagementObject o in res.Get()) {
Console.WriteLine(Itr.ToString() + ". " + o["Caption"] + " (DeviceID = " + o["DeviceID"] + ")");
/* Create Object */
MfsDisk nDisk = new MfsDisk();
nDisk.BytesPerSector = (UInt32)o["BytesPerSector"];
nDisk.DeviceId = (String)o["DeviceID"];
nDisk.TotalSectors = (UInt64)o["TotalSectors"];
nDisk.SectorsPerTrack = (UInt32)o["SectorsPerTrack"];
nDisk.TracksPerCylinder = (UInt32)o["TracksPerCylinder"];
Drives.Add(Itr, nDisk);
Itr++;
}
Console.WriteLine("\nAvailable Commands:");
Console.WriteLine("format <drive>");
Console.WriteLine("write <file> <drive>");
Console.WriteLine("ls <path> <drive>");
Console.WriteLine("install <drive>");
Console.WriteLine("quit");
Console.WriteLine("");
/* Cmd */
while (true)
{
/* Spit */
Console.Write(" $ ");
/* Read */
String Input = Console.ReadLine();
String[] Tokens = Input.Split(new Char[] { ' ' });
/* Lets see */
switch (Tokens[0].ToLower())
{
case "format":
{
/* Parse */
int Option = int.Parse(Tokens[1]);
/* Do the format */
Format((MfsDisk)Drives[Option]);
} break;
case "write":
{
/* Path */
String Path = Tokens[1];
/* Parse */
int Option = int.Parse(Tokens[2]);
/* Gogo */
WriteToMfs((MfsDisk)Drives[Option], Path, "");
} break;
case "ls":
{
/* Path */
String Path = Tokens[1];
/* Parse */
int Option = int.Parse(Tokens[2]);
/* Gogo */
ListDirectory((MfsDisk)Drives[Option], Path);
} break;
case "install":
{
/* Parse */
int Option = int.Parse(Tokens[1]);
/* Gogo */
InstallMOS((MfsDisk)Drives[Option]);
} break;
case "quit":
return;
default:
break;
}
/* Clean */
GC.Collect();
}
}
/* Recursive List */
static MfsEntry ListRecursive(MfsDisk mDisk, SafeFileHandle hDisk, UInt32 DirBucket, String pPath)
{
/* Sanity, if start with "/" skip */
String mPath = pPath;
if (mPath.StartsWith("/"))
mPath = mPath.Substring(1, mPath.Length - 1);
/* Get token */
int iDex = mPath.IndexOf("/");
String LookFor = mPath.Substring(0, iDex == -1 ? mPath.Length : iDex);
/* EoP */
if (String.IsNullOrEmpty(LookFor)
|| LookFor.Contains("."))
{
/* List files */
UInt32 IteratorBucket = DirBucket;
int End = 0;
while (End == 0)
{
UInt64 Sector = IteratorBucket * mDisk.BucketSize;
/* Gogo */
Byte[] fBuffer = ReadDisk(mDisk, hDisk, Sector, mDisk.BucketSize);
for (int i = 0; i < (mDisk.BucketSize * mDisk.BytesPerSector); i++)
{
/* EoC? */
if (fBuffer[i] == 0)
{
End = 1;
break;
}
/* Woah, parse */
int Len = 0;
while (fBuffer[i + 76 + Len] != 0)
Len++;
String Name = System.Text.Encoding.UTF8.GetString(fBuffer, i + 76, Len);
UInt16 Flags = BitConverter.ToUInt16(fBuffer, i + 2);
if (LookFor.Contains(".")
&& Name.ToLower() == LookFor.ToLower())
{
/* Match */
MfsEntry nEntry = new MfsEntry();
nEntry.Name = Name;
nEntry.Size = BitConverter.ToUInt64(fBuffer, i + 60);
nEntry.AllocatedSize = BitConverter.ToUInt64(fBuffer, i + 68);
nEntry.Bucket = BitConverter.ToUInt32(fBuffer, i + 4);
nEntry.BucketLength = BitConverter.ToUInt32(fBuffer, i + 8);
nEntry.DirBucket = IteratorBucket;
nEntry.DirIndex = (uint)i;
/* Done */
return nEntry;
}
else
{
if (((MfsEntryFlags)Flags).HasFlag(MfsEntryFlags.MFS_DIRECTORY))
Console.WriteLine("Dir: " + Name);
else
Console.WriteLine("File: " + Name + " (" + BitConverter.ToUInt64(fBuffer, i + 60).ToString() + " Bytes)");
}
/* Next */
i += 1023;
}
/* Get next bucket */
UInt32 DirBuckLength = 0;
if (End == 0)
IteratorBucket = GetNextBucket(mDisk, hDisk, IteratorBucket, out DirBuckLength);
/* End of list? */
if (IteratorBucket == 0xFFFFFFFF)
End = 1;
}
/* Done */
return null;
}
else
{
/* Find LookFor in DirBucket */
UInt32 IteratorBucket = DirBucket;
int End = 0;
while (End == 0)
{
UInt64 Sector = IteratorBucket * mDisk.BucketSize;
/* Gogo */
Byte[] fBuffer = ReadDisk(mDisk, hDisk, Sector, mDisk.BucketSize);
for (int i = 0; i < (mDisk.BucketSize * mDisk.BytesPerSector); i++)
{
/* EoC? */
if (fBuffer[i] == 0)
{
End = 1;
break;
}
/* Woah, parse */
int Len = 0;
while (fBuffer[i + 76 + Len] != 0)
Len++;
String Name = System.Text.Encoding.UTF8.GetString(fBuffer, i + 76, Len);
UInt16 Flags = BitConverter.ToUInt16(fBuffer, i + 2);
if (Name.ToLower() == LookFor.ToLower())
{
/* More sanity */
if (!((MfsEntryFlags)Flags).HasFlag(MfsEntryFlags.MFS_DIRECTORY))
{
Console.WriteLine(LookFor + " is not a directory");
return null;
}
/* Match */
MfsEntry nEntry = new MfsEntry();
nEntry.Name = Name;
nEntry.Size = BitConverter.ToUInt64(fBuffer, i + 60);
nEntry.AllocatedSize = BitConverter.ToUInt64(fBuffer, i + 68);
nEntry.Bucket = BitConverter.ToUInt32(fBuffer, i + 4);
nEntry.BucketLength = BitConverter.ToUInt32(fBuffer, i + 8);
nEntry.DirBucket = IteratorBucket;
nEntry.DirIndex = (uint)i;
/* Sanity */
if (nEntry.Bucket == 0xFFFFFFFF)
return null;
/* Done */
return ListRecursive(mDisk, hDisk, nEntry.Bucket, mPath.Substring(LookFor.Length));
}
/* Next */
i += 1023;
}
/* Get next bucket */
UInt32 DirBuckLength = 0;
if (End == 0)
IteratorBucket = GetNextBucket(mDisk, hDisk, IteratorBucket, out DirBuckLength);
/* End of list? */
if (IteratorBucket == 0xFFFFFFFF)
End = 1;
}
}
return null;
}
/* Lists root directory */
static void ListDirectory(MfsDisk mDisk, String Path)
{
/* Open Disk */
SafeFileHandle handle = OpenDisk(mDisk.DeviceId);
if (handle == null)
{
Console.WriteLine("Failed to open drive");
return;
}
/* Read MBR */
Byte[] Mbr = ReadDisk(mDisk, handle, 0, 1);
/* Find MB Ptr */
UInt64 MbSector = BitConverter.ToUInt64(Mbr, 28);
mDisk.BucketSize = BitConverter.ToUInt16(Mbr, 26);
/* Find Root Ptr in MB */
Byte[] Mb = ReadDisk(mDisk, handle, MbSector, 1);
UInt32 RootBucket = BitConverter.ToUInt32(Mb, 12);
/* Recurse-Parse Root */
Console.WriteLine("Files in " + Path + ":");
ListRecursive(mDisk, handle, RootBucket, Path);
Console.WriteLine("");
/* Done */
handle.Close();
}
/* Allocate a bucket */
static UInt32 AllocateBucket(MfsDisk mDisk, SafeFileHandle hDisk, UInt32 FreeBucketIndex, UInt64 NumBuckets, out UInt32 InitialSize)
{
/* Calculate Bucket Map */
UInt64 Buckets = mDisk.TotalSectors / mDisk.BucketSize;
UInt64 BucketMapSize = Buckets * 8; /* One bucket descriptor is 8 bytes */
UInt64 BucketMapSector = (mDisk.TotalSectors - ((BucketMapSize / mDisk.BytesPerSector) + 1));
UInt32 BucketsPerSector = mDisk.BytesPerSector / 8;
UInt64 Counter = NumBuckets;
UInt32 BucketPtr = FreeBucketIndex;
UInt32 BucketPrevPtr = 0;
UInt32 FirstFreeSize = 0;
while (Counter > 0)
{
/* Which sector is the next bucket in? */
UInt32 FreeCount = 0;
UInt32 SectorOffset = BucketPtr / BucketsPerSector;
UInt32 SectorIndex = BucketPtr % BucketsPerSector;
/* Read sector */
Byte[] Sector = ReadDisk(mDisk, hDisk, BucketMapSector + SectorOffset, 1);
/* Done */
BucketPrevPtr = BucketPtr;
BucketPtr = BitConverter.ToUInt32(Sector, (int)(SectorIndex * 8));
FreeCount = BitConverter.ToUInt32(Sector, (int)((SectorIndex * 8) + 4));
/* How many buckets? */
if (FreeCount > Counter)
{
/* Calculate next free */
UInt32 NextFreeBucket = BucketPrevPtr + (UInt32)Counter;
UInt32 NextFreeCount = FreeCount - (UInt32)Counter;
if (FirstFreeSize == 0)
FirstFreeSize = (UInt32)Counter;
/* We have to adjust now,
* since we are taking only a chunk
* of the available length */
Sector[SectorIndex * 8] = 0xFF;
Sector[SectorIndex * 8 + 1] = 0xFF;
Sector[SectorIndex * 8 + 2] = 0xFF;
Sector[SectorIndex * 8 + 3] = 0xFF;
Sector[SectorIndex * 8 + 4] = (Byte)(Counter & 0xFF);
Sector[SectorIndex * 8 + 5] = (Byte)((Counter >> 8) & 0xFF);
Sector[SectorIndex * 8 + 6] = (Byte)((Counter >> 16) & 0xFF);
Sector[SectorIndex * 8 + 7] = (Byte)((Counter >> 24) & 0xFF);
/* Write it back */
WriteDisk(mDisk, hDisk, BucketMapSector + SectorOffset, Sector);
/* Setup new block */
SectorOffset = NextFreeBucket / BucketsPerSector;
SectorIndex = NextFreeBucket % BucketsPerSector;
/* Read */
Sector = ReadDisk(mDisk, hDisk, BucketMapSector + SectorOffset, 1);
/* Modify */
Sector[SectorIndex * 8] = (Byte)(BucketPtr & 0xFF);
Sector[SectorIndex * 8 + 1] = (Byte)((BucketPtr >> 8) & 0xFF);
Sector[SectorIndex * 8 + 2] = (Byte)((BucketPtr >> 16) & 0xFF);
Sector[SectorIndex * 8 + 3] = (Byte)((BucketPtr >> 24) & 0xFF);
Sector[SectorIndex * 8 + 4] = (Byte)(NextFreeCount & 0xFF);
Sector[SectorIndex * 8 + 5] = (Byte)((NextFreeCount >> 8) & 0xFF);
Sector[SectorIndex * 8 + 6] = (Byte)((NextFreeCount >> 16) & 0xFF);
Sector[SectorIndex * 8 + 7] = (Byte)((NextFreeCount >> 24) & 0xFF);
/* Write it back */
WriteDisk(mDisk, hDisk, BucketMapSector + SectorOffset, Sector);
/* Done */
InitialSize = FirstFreeSize;
return NextFreeBucket;
}
else
{
/* We can just take the whole cake
* no need to modify it's length */
if (FirstFreeSize == 0)
FirstFreeSize = FreeCount;
/* Next */
Counter -= FreeCount;
}
}
/* Update BucketPrevPtr to 0xFFFFFFFF */
UInt32 _SecOff = BucketPrevPtr / BucketsPerSector;
UInt32 _SecInd = BucketPrevPtr % BucketsPerSector;
/* Read sector */
Byte[] _Sec = ReadDisk(mDisk, hDisk, BucketMapSector + _SecOff, 1);
/* Modify Sector */
_Sec[_SecInd * 8] = 0xFF;
_Sec[_SecInd * 8 + 1] = 0xFF;
_Sec[_SecInd * 8 + 2] = 0xFF;
_Sec[_SecInd * 8 + 3] = 0xFF;
/* Write it back */
WriteDisk(mDisk, hDisk, BucketMapSector + _SecOff, _Sec);
/* Done */
InitialSize = FirstFreeSize;
return BucketPtr;
}
