Merge pull request #1 from kungfux/feature_v11

Version 1.1

Author: kungfux

README.md

@@ -1,11 +1,11 @@
-# EFS (Eat Free Space)
+# ffs-util
 
-EFS is utility to fill up all available free space on the drive.
+FFS (Fill Free Space) is an utility to fill up all available free space on the drive.
 This utility is created to be used for testing purposes (e.g. resilience testing).
 
 ## How it works?
-EFS creates the empty file with defined file length. So, this file will be just a file declaration for file system. Such file can be created very quickly and there is no need to write some data to the file, so hard disk health is not suffering.
+FFS creates empty file with defined file length. So, this file will be just a file declaration for the file system. Such file can be created very quickly and there is no need to write some data to the file, so hard disk health is not suffering.
 
 ### Requirements:
 
-.NET 4.5 is required to build the project.
+.NET Framework 2.0 is required to build the project.

project/App.config

@@ -1,6 +1,5 @@
-<?xml version="1.0" encoding="utf-8" ?>
+<?xml version="1.0"?>
 <configuration>
     <startup> 
-        <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.5" />
-    </startup>
-</configuration>
+    <supportedRuntime version="v2.0.50727"/></startup>
+</configuration>

project/Drive.cs

@@ -0,0 +1,63 @@
+using System;
+using System.IO;
+
+namespace ffs_util
+{
+    internal class DriveSpaceInfo
+    {
+        private enum ESpaceSize
+        {
+            Bytes = 1,
+            KB = 2,
+            MB = 4,
+            GB = 8,
+            TB = 16
+        }
+
+        private string _driveName = "";
+
+
+        public DriveSpaceInfo(string DriveName)
+        {
+            _driveName = DriveName;
+        }
+
+
+        public long GetFreeSpaceAsLong()
+        {
+            try
+            {
+                DriveInfo dinfo = new DriveInfo(_driveName);
+                return dinfo.AvailableFreeSpace;
+            }
+            catch (Exception ex)
+            {
+                Message.Instance.PrintError(ex.Message);
+                Environment.Exit(1);
+                return -1;
+            }
+        }
+
+
+        public string GetFreeSpaceAsString()
+        {
+            long space = GetFreeSpaceAsLong();
+            long[] result = ConvertBytes(space, 1, 16);
+            return string.Format("{0} {1}", result[0].ToString(), (ESpaceSize)result[1]);
+        }
+
+
+        private long[] ConvertBytes(long bytes, int minEnum, int maxEnum)
+        {
+            int value = 1024;
+            int sizeName = 1;
+            double size = bytes;
+            while ((size > value && sizeName < maxEnum) || sizeName < minEnum)
+            {
+                size = size / 1024;
+                sizeName = sizeName * 2;
+            }
+            return new long[] { (long)Math.Round(size, 0), sizeName };
+        }
+    }
+}

project/DriveSpaceInfo.cs

@@ -0,0 +1,73 @@
+using System;
+using System.IO;
+
+namespace ffs_util
+{
+    internal class FillFreeSpace
+    {
+        private const string _pieFile = "pie.ffs";
+
+        /// <summary>
+        /// Fill available free space using temporary file declaration with defined length
+        /// </summary>
+        public void FillFreeSpaceWithEmptyFileOrDelete(string DriveName)
+        {
+            DriveSpaceInfo drive = new DriveSpaceInfo(DriveName);
+
+            string pieFileFullName = DriveName + @":\" + _pieFile;
+
+            if (File.Exists(pieFileFullName))
+            {
+                DeletePieFile(pieFileFullName);
+                return;
+            }
+
+            CreatePieFile(pieFileFullName, drive.GetFreeSpaceAsLong(), drive.GetFreeSpaceAsString());
+        }
+
+
+        private void CreatePieFile(string PieFileFullName, long DriveFreeSpace, string DriveFreeSpaceAsString)
+        {
+            if (DriveFreeSpace <= 0)
+            {
+                Message.Instance.PrintError("There is no free space available. Operation is aborted.");
+                return;
+            }
+
+            Message.Instance.PrintMessage(
+                string.Format("Creating file {0} with size of {1}...",
+                PieFileFullName, DriveFreeSpaceAsString)
+                );
+
+            try
+            {
+                using (var fs = new FileStream(PieFileFullName, FileMode.CreateNew, FileAccess.Write))
+                {
+                    fs.SetLength(DriveFreeSpace);
+                    fs.Close();
+                    Message.Instance.PrintMessage("Done.");
+                }
+            }
+            catch(Exception ex)
+            {
+                Message.Instance.PrintError(ex.Message);
+            }
+        }
+
+
+        private void DeletePieFile(string PieFileFullName)
+        {
+            Message.Instance.PrintMessage("Temporary file already exists. Removing...");
+
+            try
+            {
+                File.Delete(PieFileFullName);
+                Message.Instance.PrintMessage("Done.");
+            }
+            catch (Exception ex)
+            {
+                Message.Instance.PrintError(ex.Message);
+            }
+        }
+    }
+}