private BooleanResult mountCloud(string guid, Config.Clouds cloudSelected, string cloudPath)
{
// Get password from user //
var promptPassword = new PromptPassword();
if (promptPassword.ShowDialog() != DialogResult.OK)
{
// Cancelled
return new BooleanResult() { Success = true };
}
string password = promptPassword.CloudPassword;
// * //
// GET NEXT FREE DRIVE LETTER
string targetDrive = Toolbox.GetNextFreeDriveLetter();
if (targetDrive == null)
{
return new BooleanResult() { Success = false, Message = "ERROR: Cannot find a free drive letter!" };
}
// * //
// Helper result object
BooleanResult res = null;
// Check to see if it is already mounted //
Dictionary<string, string> mounts = EncryptFS.GetAllMountedEncFS();
if (mounts == null)
{
return new BooleanResult() { Success = false, Message = "ERROR: Cannot figure out which EncFS instances are mounted!" };
}
if (mounts.ContainsKey(guid))
{
return new BooleanResult() { Success = false, Message = "This encrypted folder appears to already be mounted!" };
}
// * //
// Get and decrypt user's master key (using user password) //
string masterKey = null;
string encHeader = (string)Registry.GetValue(Config.CURR_USR_REG_DRIVE_ROOT + guid, "encHeader", null);
if (string.IsNullOrEmpty(encHeader))
{
return new BooleanResult() { Success = false, Message = "ERROR: User's header information could not be found!" };
}
masterKey = Toolbox.PasswordDecryptKey(encHeader, password);
// Make sure we got a key back
if (masterKey == null)
{
return new BooleanResult() { Success = false, Message = "ERROR: Failed to decrypt master key!" };
}
// * //
// Mount their freshly-created encrypted drive
res = EncryptFS.MountEncryptedFS(guid, targetDrive, masterKey, "Secure " + cloudSelected.ToString());
if (res == null || !res.Success)
{
return res;
}
// * //
return new BooleanResult() { Success = true };
}
private BooleanResult encryptCloud(Config.Clouds cloudSelected, string cloudPath)
{
// Get new password from user //
var promptPassword = new PromptNewPassword();
if (promptPassword.ShowDialog() != DialogResult.OK)
{
// Cancelled
return new BooleanResult() { Success = true };
}
string password = promptPassword.CloudPassword;
string passwordConf = promptPassword.CloudPasswordConfirm;
// * //
// Sanity checks //
if (password.Length < Config.MIN_PASSWORD_LEN || !string.Equals(password, passwordConf))
{
return new BooleanResult() { Success = false, Message = "Passwords provided must match and be non-zero in length!" };
}
// * //
// GET NEXT FREE DRIVE LETTER
string targetDrive = Toolbox.GetNextFreeDriveLetter();
if (targetDrive == null)
{
return new BooleanResult() { Success = false, Message = "ERROR: Cannot find a free drive letter!" };
}
// * //
// Microsoft OneDrive will cause problems during migration if it is running,
// so terminate it before we start
if (cloudSelected == Config.Clouds.OneDrive)
{
// Ask user permission to close process first
var confirmResult = MessageBox.Show("OneDrive process must be stopped before migration.", "Should I close it?", MessageBoxButtons.YesNo);
if (confirmResult != DialogResult.Yes)
{
return new BooleanResult() { Success = false, Message = "ERROR: Please close OneDrive application before migration." };
}
// Try OneDrive first, then SkyDrive
Process[] processes = Process.GetProcessesByName("OneDrive");
if (processes.Length == 0)
{
processes = Process.GetProcessesByName("SkyDrive");
}
// If we found a OneDrive/SkyDrive process running, attempt to close it, or kill it otherwise
if (processes.Length > 0)
{
processes[0].CloseMainWindow();
processes[0].WaitForExit(5000);
if (!processes[0].HasExited)
{
processes[0].Kill();
processes[0].WaitForExit(5000);
if (!processes[0].HasExited)
{
return new BooleanResult() { Success = false, Message = "ERROR: Could not close OneDrive application!" };
}
}
}
}
else
{
// Tell user to turn of syncing for service
var confirmResult = MessageBox.Show("Please remember to disable file syncronization for " + cloudSelected.ToString(), "Press OK when you're ready.", MessageBoxButtons.OKCancel);
if (confirmResult != DialogResult.OK)
{
return new BooleanResult() { Success = false, Message = "ERROR: Please disable file synchronization before migration." };
}
}
// * //
// Disable while we calcualte stuff
this.Cursor = Cursors.WaitCursor;
g_tabContainer.Controls[1].Enabled = false;
// Progress bar
s_progress.Value = 0;
s_progress.Visible = true;
Application.DoEvents();
s_progress.ProgressBar.Refresh();
// Generate a new GUID to identify this FS
string guid = Guid.NewGuid().ToString();
// Helper result object
BooleanResult res = null;
// Run work-heavy tasks in a separate thread
CancellationTokenSource cts = new CancellationTokenSource();
CancellationToken cancelToken = cts.Token;
var workerThread = Task.Factory.StartNew(() =>
{
// Update UI
this.Invoke((MethodInvoker)delegate
{
s_progress.Value = 25;
l_statusLabel.Text = "Generating encryption key ...";
Application.DoEvents();
s_progress.ProgressBar.Refresh();
});
// Generate master key & protect with user password //
string masterKey = Toolbox.GenerateKey(Config.MASTERKEY_PW_CHAR_COUNT);
string encMasterKey = Toolbox.PasswordEncryptKey(masterKey, password);
// Ensure we got good stuff back
if (masterKey == null)
{
return new BooleanResult() { Success = false, Message = "ERROR: Cannot generate master key!" };
}
if (encMasterKey == null)
{
return new BooleanResult() { Success = false, Message = "ERROR: Cannot encrypt master key!" };
}
Registry.SetValue(Config.CURR_USR_REG_DRIVE_ROOT + guid, "encHeader", encMasterKey);
// * //
// Generate temporary location to hold enc data
string tempFolderName = cloudPath + ".backup-" + Path.GetRandomFileName();
Directory.CreateDirectory(tempFolderName);
// Update UI
this.Invoke((MethodInvoker)delegate
{
s_progress.Value = 50;
l_statusLabel.Text = "Creating EncFS drive";
Application.DoEvents();
s_progress.ProgressBar.Refresh();
});
// Create new EncFS
res = EncryptFS.CreateEncryptedFS(guid, cloudPath, targetDrive, masterKey, "Secure " + cloudSelected.ToString(), true);
if (res == null || !res.Success)
{
return res;
}
// * //
// Update UI
this.Invoke((MethodInvoker)delegate
{
s_progress.Value = 75;
l_statusLabel.Text = "Copying data from Cloud folder to encrypted drive";
Application.DoEvents();
s_progress.ProgressBar.Refresh();
});
// Copy cloud data over
res = EncryptFS.MoveDataFromFolder(cloudPath, tempFolderName);
if (res == null || !res.Success)
{
return res;
}
res = EncryptFS.CopyDataFromFolder(tempFolderName, targetDrive + ":\\");
if (res == null || !res.Success)
{
return res;
}
// * //
return new BooleanResult() { Success = true };
}, TaskCreationOptions.LongRunning);
// When threaded tasks finish, check for errors and continue (if appropriate)
workerThread.ContinueWith((antecedent) =>
{
// Block until we get a result back from previous thread
BooleanResult result = antecedent.Result;
// Check if there was an error in previous thread
if (result == null || !result.Success)
{
ReportEncryptCloudError(result);
return;
}
// Re-enable everything //
this.Cursor = Cursors.Default;
g_tabContainer.Controls[1].Enabled = true;
l_statusLabel.Text = "Successfully moved your cloud folder!";
s_progress.Value = 0;
s_progress.Visible = false;
Application.DoEvents();
// * //
},
cancelToken,
TaskContinuationOptions.OnlyOnRanToCompletion,
TaskScheduler.FromCurrentSynchronizationContext()
);
return new BooleanResult() { Success = true };
}
