コード例 #1
0
        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;
        }
コード例 #2
0
        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;
            }
            }
        }