public void Defragment(FSRoot root, int size)
{
byte *newBuffer = (byte *)malloc((ulong)size);
uint head = 0;
void FileDefrag(VFile f)
{
memcpy(newBuffer + head, m_Data + f.Offset, f.Size);
f.Offset = head;
head += f.Size;
}
root.RootDir.FileForEach(FileDefrag);
root.RootDir.DirForEach(dir => { dir.FileForEach(FileDefrag); });
Size = (ulong)size;
free(m_Data);
m_Data = newBuffer;
}
public void Execute(string command)
{
string[] args = command.Trim().Split(" ");
switch (args[0])
{
case "exit":
{
Running = false;
break;
}
case "init":
{
ulong size = ulong.Parse(args[1]);
FileStream s = File.Create("disk.vd");
for (ulong i = 0; i < size; i++)
{
s.WriteByte(0);
}
s.Flush();
s.Close();
break;
}
case "read":
{
if (m_Disk != null || m_Root != null)
{
break;
}
m_Disk = new VDisk("disk.vd");
IFormatter formatter = new BinaryFormatter();
try
{
Stream stream = new FileStream("fs", FileMode.Open, FileAccess.ReadWrite, FileShare.None);
m_Root = (FSRoot)formatter.Deserialize(stream);
stream.Flush();
stream.Close();
}
catch (Exception)
{
Console.ForegroundColor = ConsoleColor.Yellow;
Console.WriteLine("File system does not exist. Creating a new file system.");
Console.ForegroundColor = ConsoleColor.White;
m_Root = new FSRoot();
}
m_Disk.Head = m_Root.Head;
m_WorkingDir = m_Root.RootDir;
break;
}
case "write":
{
m_Disk.Write();
IFormatter formatter = new BinaryFormatter();
Stream stream = new FileStream("fs", FileMode.Create, FileAccess.Write, FileShare.None);
formatter.Serialize(stream, m_Root);
stream.Flush();
stream.Close();
break;
}
case "workingDir":
case "pwd":
{
Console.WriteLine(GetWorkingDir());
break;
}
case "changeDir":
case "cd":
{
if (args[1].StartsWith("/"))
{
m_WorkingDir = m_Root.RootDir;
args[1] = args[1].Substring(1);
}
string[] dirs = args[1].Split("/");
foreach (var dir in dirs)
{
if (dir.Trim().Length == 0)
{
break;
}
if (dir == "..")
{
if (m_WorkingDir !.ParentDir != null)
{
m_WorkingDir = m_WorkingDir.ParentDir;
}
}
else
{
VDirectory target = m_WorkingDir.GetSubDirectory(dir);
if (target == null)
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("Not Found: " + dir);
Console.ForegroundColor = ConsoleColor.White;
break;
}
m_WorkingDir = target;
}
}
break;
}
case "moveFileIn":
case "mfi":
case "store":
{
string filePath = args[1];
string name = Path.GetFileName(filePath);
FileInfo info = new FileInfo(filePath);
void * nativeFileHandle = fopen(filePath, "rb");
byte * data = (byte *)malloc((ulong)info.Length);
fread(data, (ulong)info.Length, 1, nativeFileHandle);
fclose(nativeFileHandle);
VFile file = m_Disk.NewFile(name, data, (uint)info.Length);
free(data);
m_WorkingDir.AddFile(file);
break;
}
case "takeFileOut":
case "tfo":
case "retrieve":
{
string name = args[1];
VFile file = m_WorkingDir.GetFile(name);
if (file == null)
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("File : " + name + " does not exist");
Console.ForegroundColor = ConsoleColor.White;
break;
}
byte *fileData = m_Disk.ReadFileNative(file);
void *nativeFileHandle = fopen(name, "wb");
fwrite(fileData, file.Size, 1, nativeFileHandle);
fclose(nativeFileHandle);
break;
}
case "list":
case "ls":
{
Console.ForegroundColor = ConsoleColor.Green;
m_WorkingDir.DirForEach(dir =>
{
Console.WriteLine($"|{"Dir",15} | {dir.Name,40} | {dir.Size + " Bytes",20}|");
});
Console.ForegroundColor = ConsoleColor.Magenta;
m_WorkingDir.FileForEach(f =>
{
Console.WriteLine($"|{"File",15} | {f.Name,40} | {f.Size + " Bytes",20}|");
});
Console.ForegroundColor = ConsoleColor.White;
break;
}
case "makeDir":
case "mkdir":
{
m_WorkingDir.AddSubDirectory(new VDirectory(args[1], m_WorkingDir));
break;
}
case "removeDir":
case "rmdir":
{
VDirectory target = m_WorkingDir.GetSubDirectory(args[1]);
if (target == null)
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("Not Found: " + args[1]);
Console.ForegroundColor = ConsoleColor.White;
break;
}
m_WorkingDir.RemoveSubDirectory(target);
break;
}
case "removeFile":
case "rm":
{
string name = args[1];
VFile file = m_WorkingDir.GetFile(name);
if (file == null)
{
Console.ForegroundColor = ConsoleColor.Red;
Console.WriteLine("File : " + name + " does not exist");
Console.ForegroundColor = ConsoleColor.White;
break;
}
m_WorkingDir.RemoveFile(file);
break;
}
case "defragment":
case "defrag":
{
int size;
if (args.Length >= 2)
{
size = int.Parse(args[1]);
}
else
{
size = (int)m_Disk.Size;
}
m_Disk.Defragment(m_Root, size);
break;
}
case "diskSize":
case "dsize":
{
Console.WriteLine("Size: " + m_Disk.Size);
break;
}
}
}
