276 lines
11 KiB
C#
276 lines
11 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading.Tasks;
|
|
using System.Data.SqlServerCe;
|
|
using System.Data;
|
|
using System.IO;
|
|
using Saviour_Backup_System.Properties;
|
|
|
|
namespace Saviour_Backup_System
|
|
{
|
|
class databaseTools
|
|
{
|
|
public static string databaseName = "saviour.sdf";
|
|
public static SqlCeConnection conn = new SqlCeConnection("Data Source = " + databaseName + "; password=12a712d7e6f71ed07822c219318da2c0"); //password is a hash
|
|
private static SqlCeCommand cmd = conn.CreateCommand();
|
|
|
|
private static void copyDatabase() { File.WriteAllBytes(@"" + databaseName, Resources.saviour); } //copy the file from resources
|
|
|
|
/// <summary>
|
|
/// Checks if the database exists, and then pings the connection
|
|
/// </summary>
|
|
public static void init() {
|
|
if (!File.Exists(databaseName)) { //if the database doesnt exists (program hasnt been run before)
|
|
copyDatabase();
|
|
}
|
|
conn.Open();
|
|
conn.Close();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the drive name from an ID
|
|
/// </summary>
|
|
/// <param name="id">ID of the drive to search for</param>
|
|
/// <returns></returns>
|
|
public static string getDriveName(string id) {
|
|
string name = "NONE";
|
|
conn.Open();
|
|
cmd.CommandText = "SELECT Name FROM Drive WHERE ID LIKE ?;";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive_ID", SqlDbType.NText));
|
|
cmd.Parameters["Drive_ID"].Value = id;
|
|
SqlCeDataReader reader = cmd.ExecuteReader();
|
|
while (reader.Read()) { name = reader.GetString(0); }
|
|
conn.Close();
|
|
reader.Close();
|
|
cmd.Parameters.Clear();
|
|
return name;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the backup directory for a drive
|
|
/// </summary>
|
|
/// <param name="id">ID of the drive to search for</param>
|
|
/// <returns></returns>
|
|
public static string getBackupDirectory(string id) {
|
|
string directory = "NONE";
|
|
conn.Open();
|
|
cmd.CommandText = "SELECT Backup_Location FROM Recordset WHERE Drive_ID LIKE ?";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive_ID", SqlDbType.NText));
|
|
cmd.Parameters["Drive_ID"].Value = id;
|
|
try {
|
|
SqlCeDataReader reader = cmd.ExecuteReader();
|
|
while (reader.Read()) { directory = reader.GetString(0); }
|
|
reader.Close();
|
|
} catch { }
|
|
conn.Close();
|
|
cmd.Parameters.Clear();
|
|
return directory;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the creation date for a backup record
|
|
/// </summary>
|
|
/// <param name="id">ID of the drive to search for</param>
|
|
/// <returns></returns>
|
|
public static Int64 getBackupCreationDate(string id)
|
|
{
|
|
Int64 date = 0;
|
|
conn.Open();
|
|
cmd.CommandText = "SELECT Creation_Date FROM Recordset WHERE Drive_ID LIKE ?";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive ID", SqlDbType.NText));
|
|
cmd.Parameters["Drive ID"].Value = id;
|
|
try
|
|
{
|
|
SqlCeDataReader reader = cmd.ExecuteReader();
|
|
while (reader.Read()) { date = reader.GetInt64(0); }
|
|
reader.Close();
|
|
}
|
|
catch { }
|
|
conn.Close();
|
|
cmd.Parameters.Clear();
|
|
return date;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns what drives are set to back up automatically
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public static string[] getAutomaticBackups()
|
|
{
|
|
conn.Open();
|
|
List<string> IDs = new List<string>();
|
|
cmd.CommandText = "SELECT Drive_ID FROM Recordset WHERE Automatic = 1;";
|
|
SqlCeDataReader reader = cmd.ExecuteReader(CommandBehavior.CloseConnection);
|
|
while (reader.Read())
|
|
{
|
|
string DriveID = reader.GetString(0);
|
|
IDs.Add(DriveID);
|
|
}
|
|
reader.Close();
|
|
conn.Close();
|
|
return IDs.ToArray();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the name of the backup
|
|
/// </summary>
|
|
/// <param name="drive">Drive object to search for</param>
|
|
/// <returns></returns>
|
|
public static string getBackupName(DriveInfo drive)
|
|
{
|
|
string name = "";
|
|
conn.Open();
|
|
cmd.CommandText = "SELECT Name FROM Recordset WHERE Drive_ID LIKE ?;";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive ID", SqlDbType.NText));
|
|
cmd.Parameters["Drive ID"].Value = USBTools.calculateDriveID(drive);
|
|
SqlCeDataReader reader = cmd.ExecuteReader();
|
|
while (reader.Read())
|
|
{
|
|
name = reader.GetString(0);
|
|
}
|
|
cmd.Parameters.Clear();
|
|
reader.Close();
|
|
conn.Close();
|
|
return name;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Create a backup record in the database
|
|
/// </summary>
|
|
/// <param name="drive">Drive info object for backup drive</param>
|
|
/// <param name="startDate">Date the backup started (as UNIX timestamp)</param>
|
|
/// <param name="duration">How long the backup took to run (in seconds)</param>
|
|
/// <param name="hash">New hash of the drive</param>
|
|
public static void createBackupRecord(DriveInfo drive, Int64 startDate, Int64 duration, string hash)
|
|
{
|
|
string id = USBTools.calculateDriveID(drive);
|
|
string backupName = getBackupName(drive);
|
|
conn.Open();
|
|
cmd.CommandText = "INSERT INTO Backups VALUES (?,?,?,?,?);";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive ID", SqlDbType.NText));
|
|
cmd.Parameters.Add(new SqlCeParameter("Start Date", SqlDbType.BigInt));
|
|
cmd.Parameters.Add(new SqlCeParameter("Backup Name", SqlDbType.NText));
|
|
cmd.Parameters.Add(new SqlCeParameter("Hash", SqlDbType.NText));
|
|
cmd.Parameters.Add(new SqlCeParameter("Duration", SqlDbType.Int));
|
|
|
|
cmd.Parameters["Drive ID"].Value = id;
|
|
cmd.Parameters["Start Date"].Value = startDate;
|
|
cmd.Parameters["Backup Name"].Value = backupName;
|
|
cmd.Parameters["Hash"].Value = hash;
|
|
cmd.Parameters["Duration"].Value = (int)duration;
|
|
cmd.ExecuteNonQuery();
|
|
cmd.Parameters.Clear();
|
|
conn.Close();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns the most recent hash of a drive
|
|
/// </summary>
|
|
/// <param name="id">ID of drive to search for</param>
|
|
/// <returns></returns>
|
|
public static string getHashofRecentBackup(string id)
|
|
{
|
|
conn.Open();
|
|
string hash = "NONE";
|
|
cmd.CommandText = "SELECT Hash FROM Backups WHERE Drive_ID Like ?;";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive ID", SqlDbType.NText));
|
|
cmd.Parameters["Drive ID"].Value = id;
|
|
SqlCeDataReader reader = cmd.ExecuteReader();
|
|
while (reader.Read()) {
|
|
hash = reader.GetString(0);
|
|
}
|
|
cmd.Parameters.Clear();
|
|
reader.Close();
|
|
conn.Close();
|
|
return hash;
|
|
}
|
|
|
|
|
|
/// <summary>
|
|
/// Returns if the drive is compressed
|
|
/// </summary>
|
|
/// <param name="id">ID of the drive to search for</param>
|
|
/// <returns></returns>
|
|
public static bool isCompression(string id)
|
|
{
|
|
conn.Open();
|
|
bool compression = false;
|
|
cmd.CommandText = "SELECT Compression FROM Recordset WHERE Drive_ID Like ?;";
|
|
cmd.Parameters.Add(new SqlCeParameter("Drive ID", SqlDbType.NText));
|
|
cmd.Parameters["Drive ID"].Value = id;
|
|
SqlCeDataReader reader = cmd.ExecuteReader();
|
|
while (reader.Read()) {
|
|
compression = reader.GetBoolean(0);
|
|
}
|
|
cmd.Parameters.Clear();
|
|
reader.Close();
|
|
conn.Close();
|
|
return compression;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Returns all the drives that have backup records
|
|
/// </summary>
|
|
/// <returns></returns>
|
|
public static DataTable getAllDriveBackups()
|
|
{
|
|
DataTable table = new DataTable();
|
|
conn.Open();
|
|
cmd.CommandText =
|
|
"SELECT Recordset.Name, Recordset.Creation_Date, Recordset.Backup_Location, Drive.Name, Drive.Capacity FROM Recordset, Drive";
|
|
using (SqlCeDataAdapter adapter = new SqlCeDataAdapter(cmd.CommandText, conn)) {
|
|
adapter.Fill(table);
|
|
conn.Close();
|
|
return table;
|
|
}
|
|
}
|
|
|
|
/// <summary>
|
|
/// Deletes a drive record from the database
|
|
/// </summary>
|
|
/// <param name="creationDate">Date of drive creation as UNIX timestamp</param>
|
|
public static void deleteDriveRecord(Int64 creationDate) {
|
|
conn.Open();
|
|
cmd.CommandText = "DELETE FROM RecordSet, Drive WHERE Creation_Date=? AND Recordset.Drive_ID = Drive.ID;";
|
|
cmd.Parameters.Add(new SqlCeParameter("Creation Date", SqlDbType.BigInt));
|
|
cmd.Parameters["Creation Date"].Value = creationDate;
|
|
cmd.ExecuteNonQuery();
|
|
cmd.Parameters.Clear();
|
|
conn.Close();
|
|
}
|
|
|
|
/// <summary>
|
|
/// Update a drive record in the database
|
|
/// </summary>
|
|
/// <param name="backupName">Name of the backup record</param>
|
|
/// <param name="backupLocation">Location to store backup</param>
|
|
/// <param name="automatic">is the backup automatic on insert?</param>
|
|
/// <param name="compression">Will the backup be compressed</param>
|
|
/// <param name="previousBackups">How many previous backups will be stored</param>
|
|
/// <param name="creationDate">Creation date of record</param>
|
|
public static void updateDriveRecord(string backupName, string backupLocation, bool automatic, bool compression, int previousBackups, Int64 creationDate)
|
|
{
|
|
conn.Open();
|
|
cmd.CommandText = "UPDATE Recordset SET Name=?, Backup_Location=?, Automatic=?, Compression=?, Previous_Backups=? WHERE Creation_Date=?;";
|
|
cmd.Parameters.Add(new SqlCeParameter("Backup Name", SqlDbType.NText));
|
|
cmd.Parameters.Add(new SqlCeParameter("Backup Location", SqlDbType.NText));
|
|
cmd.Parameters.Add(new SqlCeParameter("Automatic", SqlDbType.Bit));
|
|
cmd.Parameters.Add(new SqlCeParameter("Compression", SqlDbType.Bit));
|
|
cmd.Parameters.Add(new SqlCeParameter("Previous Backups", SqlDbType.Int));
|
|
cmd.Parameters.Add(new SqlCeParameter("Creation Date", SqlDbType.BigInt));
|
|
|
|
cmd.Parameters["Backup Name"].Value = backupName;
|
|
cmd.Parameters["Backup Location"].Value = backupLocation;
|
|
cmd.Parameters["Automatic"].Value = automatic;
|
|
cmd.Parameters["Compression"].Value = compression;
|
|
cmd.Parameters["Previous Backups"].Value = previousBackups;
|
|
cmd.Parameters["Creation Date"].Value = creationDate;
|
|
|
|
cmd.ExecuteNonQuery();
|
|
cmd.Parameters.Clear();
|
|
conn.Close();
|
|
}
|
|
}
|
|
}
|