Пример #1
0
        /// <summary>
        /// Uses a Microsoft OneDrive Cloud Storage Provider (OneDrive Consumer, OneDrive for Business, Microsoft Graph) to sync the KeePass database
        /// </summary>
        /// <param name="databaseConfig">Configuration of the database to sync</param>
        /// <param name="localKeePassDatabasePath">Path to where the KeePass database to sync resides</param>
        /// <param name="forceSync">Flag to indicate if the sync should always take place</param>
        /// <param name="updateStatus">Action to write status messages to to display in the UI</param>
        /// <returns>True if successful, false if failed</returns>
        public static async Task <bool> SyncUsingOneDriveCloudProvider(Configuration databaseConfig, string localKeePassDatabasePath, bool forceSync, Action <string> updateStatus)
        {
            // Connect to OneDrive
            OneDriveApi oneDriveApi = null;
            bool        retryGettingApiInstance;

            do
            {
                retryGettingApiInstance = false;
                try
                {
                    oneDriveApi = await Utilities.GetOneDriveApi(databaseConfig);
                }
                catch (KoenZomers.OneDrive.Api.Exceptions.TokenRetrievalFailedException e)
                {
                    // Failed to get a OneDrive API instance because retrieving an oAuth token failed. This could be because the oAuth token expired or has been removed. Show the login dialog again so we can get a new token.
                    databaseConfig.RefreshToken = null;
                    retryGettingApiInstance     = true;
                }
                catch (Exception e)
                {
                    // Build the error text to show to the end user
                    var errorMessage = new System.Text.StringBuilder();
                    errorMessage.Append("Failed to connect to ");
                    switch (databaseConfig.CloudStorageType.GetValueOrDefault(CloudStorageType.OneDriveConsumer))
                    {
                    case CloudStorageType.OneDriveConsumer:
                        errorMessage.Append("OneDrive");
                        break;

                    case CloudStorageType.OneDriveForBusiness:
                        errorMessage.Append("OneDrive for Business");
                        break;

                    case CloudStorageType.MicrosoftGraph:
                        errorMessage.Append("Microsoft Graph");
                        break;

                    default:
                        errorMessage.Append("cloud storage provider");
                        break;
                    }
                    errorMessage.AppendLine(":");
                    errorMessage.AppendLine();
                    errorMessage.AppendLine(e.Message);

                    // If there's an inner exception, show its message as well as it typically gives more detail why it went wrong
                    if (e.InnerException != null)
                    {
                        errorMessage.AppendLine(e.InnerException.Message);

                        // Verify if we're offline
                        if (e.InnerException.Message.Contains("remote name could not be resolved"))
                        {
                            // Offline, don't display a modal dialog but use the status bar instead
                            KeePassDatabase.UpdateStatus("Can't connect. Working offline.");
                            return(false);
                        }
                    }

                    MessageBox.Show(errorMessage.ToString(), "Connection failed", MessageBoxButtons.OK, MessageBoxIcon.Exclamation, MessageBoxDefaultButton.Button1);
                }
            } while (retryGettingApiInstance);

            if (oneDriveApi == null)
            {
                switch (databaseConfig.CloudStorageType.GetValueOrDefault(CloudStorageType.OneDriveConsumer))
                {
                case CloudStorageType.OneDriveConsumer: updateStatus("Failed to connect to OneDrive"); break;

                case CloudStorageType.OneDriveForBusiness: updateStatus("Failed to connect to OneDrive for Business"); break;

                case CloudStorageType.MicrosoftGraph: updateStatus("Failed to connect to Microsoft Graph"); break;

                default: updateStatus("Failed to connect to cloud service"); break;
                }

                return(false);
            }

            // Save the RefreshToken in the configuration so we can use it again next time
            databaseConfig.RefreshToken = oneDriveApi.AccessToken.RefreshToken;

            // Verify if we already have retrieved the name of this OneDrive
            if (string.IsNullOrEmpty(databaseConfig.OneDriveName))
            {
                // Fetch details about this OneDrive account
                var oneDriveAccount = await oneDriveApi.GetDrive();

                databaseConfig.OneDriveName = oneDriveAccount.Owner.User.DisplayName;
            }
            Configuration.Save();

            // Check if we have a location on OneDrive to sync with
            if (string.IsNullOrEmpty(databaseConfig.RemoteDatabasePath) && string.IsNullOrEmpty(databaseConfig.RemoteFolderId) && string.IsNullOrEmpty(databaseConfig.RemoteFileName))
            {
                // Ask the user where to store the database on OneDrive
                var oneDriveFilePickerDialog = new Forms.OneDriveFilePickerDialog(oneDriveApi)
                {
                    ExplanationText          = "Select where you want to store the KeePass database. Right click for additional options.",
                    AllowEnteringNewFileName = true,
                    FileName = new System.IO.FileInfo(localKeePassDatabasePath).Name
                };
                await oneDriveFilePickerDialog.LoadFolderItems();

                var result = oneDriveFilePickerDialog.ShowDialog();
                if (result != DialogResult.OK || string.IsNullOrEmpty(oneDriveFilePickerDialog.FileName))
                {
                    return(false);
                }

                databaseConfig.RemoteDatabasePath = (oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference.Path : "") + "/" + oneDriveFilePickerDialog.CurrentOneDriveItem.Name + "/" + oneDriveFilePickerDialog.FileName;
                databaseConfig.RemoteDriveId      = oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference.DriveId : oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference.DriveId != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference.DriveId : null;
                if (oneDriveFilePickerDialog.CurrentOneDriveItem.File != null || (oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem != null && oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.File != null))
                {
                    databaseConfig.RemoteItemId = oneDriveFilePickerDialog.CurrentOneDriveItem.File != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Id;
                }
                else
                {
                    databaseConfig.RemoteFolderId = oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference != null?string.IsNullOrEmpty(oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference.Id) ? oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.Id;
                }
                databaseConfig.RemoteFileName = oneDriveFilePickerDialog.FileName;
                Configuration.Save();
            }

            // Retrieve the metadata of the KeePass database on OneDrive
            OneDriveItem oneDriveItem;

            if (string.IsNullOrEmpty(databaseConfig.RemoteItemId))
            {
                // We don't have the ID of the KeePass file, check if the database is stored on the current user its drive or on a shared drive
                OneDriveItem folder;
                if (string.IsNullOrEmpty(databaseConfig.RemoteDriveId))
                {
                    // KeePass database is on the current user its drive
                    folder = await oneDriveApi.GetItemById(databaseConfig.RemoteFolderId);
                }
                else
                {
                    // KeePass database is on a shared drive
                    folder = await oneDriveApi.GetItemFromDriveById(databaseConfig.RemoteFolderId, databaseConfig.RemoteDriveId);
                }

                // Locate the KeePass file in the folder
                oneDriveItem = await oneDriveApi.GetItemInFolder(folder, databaseConfig.RemoteFileName);

                // Check if the KeePass file has been found
                if (oneDriveItem != null)
                {
                    // Store the direct Id to the KeePass file so we can save several API calls on future syncs
                    databaseConfig.RemoteItemId = oneDriveItem.Id;
                    Configuration.Save();
                }
            }
            else
            {
                // We have the ID of the KeePass file, check if it's on the current user its drive or on a shared drive
                if (string.IsNullOrEmpty(databaseConfig.RemoteDriveId))
                {
                    // KeePass database is on the current user its drive
                    oneDriveItem = await oneDriveApi.GetItemById(databaseConfig.RemoteItemId);
                }
                else
                {
                    // KeePass database is on a shared drive
                    oneDriveItem = await oneDriveApi.GetItemFromDriveById(databaseConfig.RemoteItemId, databaseConfig.RemoteDriveId);
                }
            }

            if (oneDriveItem == null)
            {
                // KeePass database not found on OneDrive
                updateStatus("Database does not exist yet on OneDrive, uploading it now");

                OneDriveItem oneDriveFolder;
                string       fileName;
                if (string.IsNullOrEmpty(databaseConfig.RemoteFolderId))
                {
                    oneDriveFolder = databaseConfig.RemoteDatabasePath.Contains("/") ? await oneDriveApi.GetFolderOrCreate(databaseConfig.RemoteDatabasePath.Remove(databaseConfig.RemoteDatabasePath.LastIndexOf("/", StringComparison.Ordinal))) : await oneDriveApi.GetDriveRoot();

                    fileName = databaseConfig.RemoteDatabasePath.Contains("/") ? databaseConfig.RemoteDatabasePath.Remove(0, databaseConfig.RemoteDatabasePath.LastIndexOf("/", StringComparison.Ordinal) + 1) : databaseConfig.RemoteDatabasePath;
                }
                else
                {
                    oneDriveFolder = databaseConfig.RemoteDriveId == null ? await oneDriveApi.GetItemById(databaseConfig.RemoteFolderId) : await oneDriveApi.GetItemFromDriveById(databaseConfig.RemoteFolderId, databaseConfig.RemoteDriveId);

                    fileName = databaseConfig.RemoteFileName;
                }

                if (oneDriveFolder == null)
                {
                    updateStatus("Unable to upload database to OneDrive. Remote path is invalid.");
                    return(false);
                }

                // Upload the database to OneDrive
                var newUploadResult = await oneDriveApi.UploadFileAs(localKeePassDatabasePath, fileName, oneDriveFolder);

                updateStatus(newUploadResult == null ? "Failed to upload the KeePass database" : "Successfully uploaded the new KeePass database to OneDrive");

                databaseConfig.LocalFileHash = Utilities.GetDatabaseFileHash(localKeePassDatabasePath);
                if (newUploadResult != null)
                {
                    databaseConfig.RemoteItemId  = newUploadResult.Id;
                    databaseConfig.LastCheckedAt = DateTime.Now;
                    databaseConfig.LastSyncedAt  = DateTime.Now;
                    databaseConfig.ETag          = newUploadResult.ETag;
                }
                Configuration.Save();
                return(false);
            }

            // Use the ETag from the OneDrive item to compare it against the local database config etag to see if the content has changed
            // Microsoft Graph API reports back a different ETag when uploading than the file actually gets assigned for some unknown reason. This would cause each sync attempt to sync again as the ETags differ. As a workaround we'll use the CTag which does seem reliable to detect a change to the file.
            if (!forceSync &&
                oneDriveItem.CTag == databaseConfig.ETag)
            {
                updateStatus("Databases are in sync");

                databaseConfig.LastCheckedAt = DateTime.Now;
                Configuration.Save();

                return(false);
            }

            // Download the database from OneDrive
            updateStatus("Downloading KeePass database from OneDrive");

            var temporaryKeePassDatabasePath = System.IO.Path.GetTempFileName();
            var downloadSuccessful           = await oneDriveApi.DownloadItemAndSaveAs(oneDriveItem, temporaryKeePassDatabasePath);

            if (!downloadSuccessful)
            {
                updateStatus("Failed to download the KeePass database from OneDrive");

                return(false);
            }

            // Sync database
            updateStatus("KeePass database downloaded, going to sync");

            // Ensure the database that needs to be synced is still the database currently selected in KeePass to avoid merging the downloaded database with the wrong database in KeePass
            if ((!KoenZomersKeePassOneDriveSyncExt.Host.Database.IOConnectionInfo.Path.StartsWith(AppDomain.CurrentDomain.BaseDirectory) && KoenZomersKeePassOneDriveSyncExt.Host.Database.IOConnectionInfo.Path != localKeePassDatabasePath) ||
                (KoenZomersKeePassOneDriveSyncExt.Host.Database.IOConnectionInfo.Path.StartsWith(AppDomain.CurrentDomain.BaseDirectory) && KoenZomersKeePassOneDriveSyncExt.Host.Database.IOConnectionInfo.Path.Remove(0, AppDomain.CurrentDomain.BaseDirectory.Length) != localKeePassDatabasePath))
            {
                updateStatus("Failed to sync. Please don't switch to another database before done.");

                return(false);
            }

            // Merge the downloaded database with the currently open KeePass database
            var syncSuccessful = KeePassDatabase.MergeDatabases(databaseConfig, temporaryKeePassDatabasePath);

            if (!syncSuccessful)
            {
                updateStatus("Failed to synchronize the KeePass databases");
                return(false);
            }

            // Upload the synced database
            updateStatus("Uploading the new KeePass database to OneDrive");

            var uploadResult = await oneDriveApi.UploadFileAs(temporaryKeePassDatabasePath, oneDriveItem.Name, oneDriveItem.ParentReference.Path.Equals("/drive/root:", StringComparison.CurrentCultureIgnoreCase)?await oneDriveApi.GetDriveRoot() : string.IsNullOrEmpty(databaseConfig.RemoteDriveId)?await oneDriveApi.GetItemById(oneDriveItem.ParentReference.Id) : await oneDriveApi.GetItemFromDriveById(oneDriveItem.ParentReference.Id, oneDriveItem.ParentReference.DriveId));

            if (uploadResult == null)
            {
                updateStatus("Failed to upload the KeePass database");
                return(false);
            }

            // Delete the temporary database used for merging
            System.IO.File.Delete(temporaryKeePassDatabasePath);

            // The ETag changes with every request of the item so we use the CTag instead which only changes when the file changes
            databaseConfig.ETag = uploadResult.CTag;

            return(true);
        }
Пример #2
0
        /// <summary>
        /// Uses a Microsoft OneDrive Cloud Storage Provider (OneDrive Consumer, OneDrive for Business, Microsoft Graph) to sync the KeePass database
        /// </summary>
        /// <param name="databaseConfig">Configuration of the database to sync</param>
        /// <param name="localKeePassDatabasePath">Path to where the KeePass database to sync resides</param>
        /// <param name="forceSync">Flag to indicate if the sync should always take place</param>
        /// <param name="updateStatus">Action to write status messages to to display in the UI</param>
        /// <returns>True if successful, false if failed</returns>
        public static async Task <bool> SyncUsingOneDriveCloudProvider(Configuration databaseConfig, string localKeePassDatabasePath, bool forceSync, Action <string> updateStatus)
        {
            // Connect to OneDrive
            OneDriveApi oneDriveApi = null;
            bool        retryGettingApiInstance;

            do
            {
                retryGettingApiInstance = false;
                try
                {
                    oneDriveApi = await Utilities.GetOneDriveApi(databaseConfig);
                }
                catch (KoenZomers.OneDrive.Api.Exceptions.TokenRetrievalFailedException)
                {
                    // Failed to get a OneDrive API instance because retrieving an oAuth token failed. This could be because the oAuth token expired or has been removed. Show the login dialog again so we can get a new token.
                    databaseConfig.RefreshToken = null;
                    retryGettingApiInstance     = true;
                }
                catch (Exception e)
                {
                    // Build the error text to show to the end user
                    var errorMessage = new System.Text.StringBuilder();
                    errorMessage.Append("Failed to connect to ");
                    switch (databaseConfig.CloudStorageType.GetValueOrDefault(CloudStorageType.OneDriveConsumer))
                    {
                    case CloudStorageType.OneDriveConsumer:
                        errorMessage.Append("OneDrive");
                        break;

                    case CloudStorageType.MicrosoftGraph:
                    case CloudStorageType.MicrosoftGraphDeviceLogin:
                        errorMessage.Append("Microsoft Graph");
                        break;

                    default:
                        errorMessage.Append("cloud storage provider");
                        break;
                    }
                    errorMessage.AppendLine(string.Format(" for database {0}:", databaseConfig.KeePassDatabase.Name));
                    errorMessage.AppendLine();

                    // Revert to the most inner exception to get the best details on what went wrong
                    while (e.InnerException != null)
                    {
                        e = e.InnerException;
                    }

                    errorMessage.AppendLine(e.Message);

                    // Verify if we're offline
                    if (e.Message.Contains("remote name could not be resolved"))
                    {
                        // Offline, don't display a modal dialog but use the status bar instead
                        KeePassDatabase.UpdateStatus("Can't connect. Working offline.");
                        return(false);
                    }

                    MessageBox.Show(errorMessage.ToString(), "Connection failed", MessageBoxButtons.OK, MessageBoxIcon.Exclamation, MessageBoxDefaultButton.Button1);
                }
            } while (retryGettingApiInstance);

            if (oneDriveApi == null)
            {
                switch (databaseConfig.CloudStorageType.GetValueOrDefault(CloudStorageType.OneDriveConsumer))
                {
                case CloudStorageType.OneDriveConsumer: updateStatus(string.Format("Failed to connect to OneDrive for database {0}", databaseConfig.KeePassDatabase.Name)); break;

                case CloudStorageType.MicrosoftGraph:
                case CloudStorageType.MicrosoftGraphDeviceLogin:
                    updateStatus(string.Format("Failed to connect to Microsoft Graph for database {0}", databaseConfig.KeePassDatabase.Name));
                    break;

                default: updateStatus(string.Format("Failed to connect to cloud service for database {0}", databaseConfig.KeePassDatabase.Name)); break;
                }

                return(false);
            }

            // Save the RefreshToken in the configuration so we can use it again next time
            databaseConfig.RefreshToken = oneDriveApi.AccessToken.RefreshToken;

            // Verify if we already have retrieved the name of this OneDrive
            if (string.IsNullOrEmpty(databaseConfig.OneDriveName))
            {
                // Fetch details about this OneDrive account
                var oneDriveAccount = await oneDriveApi.GetDrive();

                databaseConfig.OneDriveName = oneDriveAccount.Owner.User.DisplayName;
            }
            Configuration.Save();

            // Check if we have a location on OneDrive to sync with
            if (string.IsNullOrEmpty(databaseConfig.RemoteDatabasePath) && string.IsNullOrEmpty(databaseConfig.RemoteFolderId) && string.IsNullOrEmpty(databaseConfig.RemoteFileName))
            {
                // Ask the user where to store the database on OneDrive
                var oneDriveFilePickerDialog = new Forms.OneDriveFilePickerDialog(oneDriveApi)
                {
                    ExplanationText          = "Select where you want to store the KeePass database. Right click for additional options.",
                    AllowEnteringNewFileName = true,
                    FileName = new System.IO.FileInfo(localKeePassDatabasePath).Name
                };
                await oneDriveFilePickerDialog.LoadFolderItems();

                var result = oneDriveFilePickerDialog.ShowDialog();
                if (result != DialogResult.OK || string.IsNullOrEmpty(oneDriveFilePickerDialog.FileName))
                {
                    return(false);
                }

                databaseConfig.RemoteDatabasePath = (oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference.Path : "") + "/" + oneDriveFilePickerDialog.CurrentOneDriveItem.Name + "/" + oneDriveFilePickerDialog.FileName;
                databaseConfig.RemoteDriveId      = oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference.DriveId : oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference.DriveId != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.ParentReference.DriveId : null;
                if (oneDriveFilePickerDialog.CurrentOneDriveItem.File != null || (oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem != null && oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.File != null))
                {
                    databaseConfig.RemoteItemId = oneDriveFilePickerDialog.CurrentOneDriveItem.File != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Id;
                }
                else
                {
                    databaseConfig.RemoteFolderId = oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference != null?string.IsNullOrEmpty(oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference.Id) || oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Folder != null ? oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.ParentReference.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.RemoteItem.Id : oneDriveFilePickerDialog.CurrentOneDriveItem.Id;
                }
                databaseConfig.RemoteFileName = oneDriveFilePickerDialog.FileName;
                Configuration.Save();
            }

            // Retrieve the metadata of the KeePass database on OneDrive
            OneDriveItem oneDriveItem;

            if (string.IsNullOrEmpty(databaseConfig.RemoteItemId))
            {
                // We don't have the ID of the KeePass file, check if the database is stored on the current user its drive or on a shared drive
                OneDriveItem folder;
                if (string.IsNullOrEmpty(databaseConfig.RemoteDriveId))
                {
                    // KeePass database is on the current user its drive
                    if (string.IsNullOrEmpty(databaseConfig.RemoteFolderId))
                    {
                        // No direct reference to the folder on the drive available, try locating it by its remote path
                        oneDriveItem = await oneDriveApi.GetItem(databaseConfig.RemoteDatabasePath);
                    }
                    else
                    {
                        // Get the folder in which the KeePass file resides
                        folder = await oneDriveApi.GetItemById(databaseConfig.RemoteFolderId);

                        if (folder == null)
                        {
                            updateStatus(string.Format("Unable to download database {0} from OneDrive. Remote path cannot be found.", databaseConfig.KeePassDatabase.Name));
                            return(false);
                        }

                        // Locate the KeePass file in the folder
                        oneDriveItem = await oneDriveApi.GetItemInFolder(folder, databaseConfig.RemoteFileName);
                    }
                }
                else
                {
                    // KeePass database is on a shared drive or has not been uploaded yet
                    folder = await oneDriveApi.GetItemFromDriveById(databaseConfig.RemoteFolderId, databaseConfig.RemoteDriveId);

                    // Locate the KeePass file in the folder. Will return NULL if the file has not been uploaded yet.
                    oneDriveItem = await oneDriveApi.GetItemInFolder(folder, databaseConfig.RemoteFileName);
                }

                // Check if the KeePass file has been found
                if (oneDriveItem != null)
                {
                    // Store the direct Id to the KeePass file so we can save several API calls on future syncs
                    databaseConfig.RemoteItemId = oneDriveItem.Id;
                    Configuration.Save();
                }
            }
            else
            {
                // We have the ID of the KeePass file, check if it's on the current user its drive or on a shared drive
                if (string.IsNullOrEmpty(databaseConfig.RemoteDriveId))
                {
                    // KeePass database is on the current user its drive
                    oneDriveItem = await oneDriveApi.GetItemById(databaseConfig.RemoteItemId);
                }
                else
                {
                    // KeePass database is on a shared drive
                    oneDriveItem = await oneDriveApi.GetItemFromDriveById(databaseConfig.RemoteItemId, databaseConfig.RemoteDriveId);
                }
            }

            if (oneDriveItem == null)
            {
                // KeePass database not found on OneDrive
                updateStatus(string.Format("Database {0} does not exist yet on OneDrive, uploading it now", databaseConfig.KeePassDatabase.Name));

                OneDriveItem oneDriveFolder;
                string       fileName;
                if (string.IsNullOrEmpty(databaseConfig.RemoteFolderId))
                {
                    oneDriveFolder = databaseConfig.RemoteDatabasePath.Contains("/") ? await oneDriveApi.GetFolderOrCreate(databaseConfig.RemoteDatabasePath.Remove(databaseConfig.RemoteDatabasePath.LastIndexOf("/", StringComparison.Ordinal))) : await oneDriveApi.GetDriveRoot();

                    fileName = databaseConfig.RemoteDatabasePath.Contains("/") ? databaseConfig.RemoteDatabasePath.Remove(0, databaseConfig.RemoteDatabasePath.LastIndexOf("/", StringComparison.Ordinal) + 1) : databaseConfig.RemoteDatabasePath;
                }
                else
                {
                    oneDriveFolder = databaseConfig.RemoteDriveId == null ? await oneDriveApi.GetItemById(databaseConfig.RemoteFolderId) : await oneDriveApi.GetItemFromDriveById(databaseConfig.RemoteFolderId, databaseConfig.RemoteDriveId);

                    fileName = databaseConfig.RemoteFileName;
                }

                if (oneDriveFolder == null)
                {
                    updateStatus(string.Format("Unable to upload database {0} to OneDrive. Remote path is invalid.", databaseConfig.KeePassDatabase.Name));
                    return(false);
                }

                // Upload the database to OneDrive
                var newUploadResult = await oneDriveApi.UploadFileAs(localKeePassDatabasePath, fileName, oneDriveFolder);

                updateStatus(string.Format(newUploadResult == null ? "Failed to upload the KeePass database {0}" : "Successfully uploaded the new KeePass database {0} to OneDrive", databaseConfig.KeePassDatabase.Name));

                databaseConfig.LocalFileHash = Utilities.GetDatabaseFileHash(localKeePassDatabasePath);
                if (newUploadResult != null)
                {
                    databaseConfig.RemoteItemId  = newUploadResult.Id;
                    databaseConfig.LastCheckedAt = DateTime.Now;
                    databaseConfig.LastSyncedAt  = DateTime.Now;
                    databaseConfig.ETag          = newUploadResult.ETag;
                }
                Configuration.Save();
                return(false);
            }

            // Use the ETag from the OneDrive item to compare it against the local database config etag to see if the content has changed
            // Microsoft Graph API reports back a different ETag when uploading than the file actually gets assigned for some unknown reason. This would cause each sync attempt to sync again as the ETags differ. As a workaround we'll use the CTag which does seem reliable to detect a change to the file.
            if (!forceSync &&
                oneDriveItem.CTag == databaseConfig.ETag)
            {
                updateStatus(string.Format("Database {0} is in sync", databaseConfig.KeePassDatabase.Name));

                databaseConfig.LastCheckedAt = DateTime.Now;
                Configuration.Save();

                return(false);
            }

            // Download the database from OneDrive
            updateStatus(string.Format("Downloading KeePass database {0} from OneDrive", databaseConfig.KeePassDatabase.Name));

            var temporaryKeePassDatabasePath = System.IO.Path.GetTempFileName();
            var downloadSuccessful           = await oneDriveApi.DownloadItemAndSaveAs(oneDriveItem, temporaryKeePassDatabasePath);

            if (!downloadSuccessful)
            {
                updateStatus(string.Format("Failed to download the KeePass database {0} from OneDrive", databaseConfig.KeePassDatabase.Name));

                return(false);
            }

            // Sync database
            updateStatus(string.Format("KeePass database {0} downloaded, going to sync", databaseConfig.KeePassDatabase.Name));

            // Merge the downloaded database with the currently open KeePass database
            var syncSuccessful = KeePassDatabase.MergeDatabases(databaseConfig, temporaryKeePassDatabasePath);

            string localDatabaseToUpload;

            if (!syncSuccessful)
            {
                updateStatus(string.Format("Failed to synchronize the KeePass database {0}", databaseConfig.KeePassDatabase.Name));

                var confirm = MessageBox.Show("Unable to merge the databases. Did you just change the master password for this KeePass database? If so and you would like to OVERWRITE the KeePass database stored on your OneDrive with your local database, select Yes, otherwise select No.", "Confirm overwriting your KeePass database", MessageBoxButtons.YesNo, MessageBoxIcon.Exclamation, MessageBoxDefaultButton.Button2);
                if (confirm != DialogResult.Yes)
                {
                    return(false);
                }

                // Upload the local database
                localDatabaseToUpload = databaseConfig.KeePassDatabase.IOConnectionInfo.Path;

                updateStatus(string.Format("Uploading the local KeePass database {0} to OneDrive", databaseConfig.KeePassDatabase.Name));
            }
            else
            {
                // Upload the synced database
                localDatabaseToUpload = temporaryKeePassDatabasePath;

                updateStatus(string.Format("Uploading the merged KeePass database {0} to OneDrive", databaseConfig.KeePassDatabase.Name));
            }

            OneDriveItem uploadResult = null;

            // Due to some issues with the OneDrive API throwing random errors for no reason, retry a few times to upload the database before it will be considered failed
            for (var uploadAttempts = 1; uploadAttempts <= 5; uploadAttempts++)
            {
                try
                {
                    if (!string.IsNullOrEmpty(databaseConfig.RemoteItemId))
                    {
                        // Database is already present on OneDrive, update it
                        uploadResult = await oneDriveApi.UpdateFile(localDatabaseToUpload, oneDriveItem);
                    }
                    else
                    {
                        // Database resides on the user its own OneDrive in the root folder
                        if (oneDriveItem.ParentReference.Path.Equals("/drive/root:", StringComparison.CurrentCultureIgnoreCase))
                        {
                            uploadResult = await oneDriveApi.UploadFileAs(localDatabaseToUpload, oneDriveItem.Name, await oneDriveApi.GetDriveRoot());
                        }
                        else
                        {
                            if (string.IsNullOrEmpty(databaseConfig.RemoteDriveId))
                            {
                                // Database resides on the user its own OneDrive in a folder
                                uploadResult = await oneDriveApi.UploadFileAs(localDatabaseToUpload, oneDriveItem.Name, await oneDriveApi.GetItemById(oneDriveItem.ParentReference.Id));
                            }
                            else
                            {
                                // Database resides on another OneDrive
                                uploadResult = await oneDriveApi.UploadFileAs(localDatabaseToUpload, oneDriveItem.Name, await oneDriveApi.GetItemFromDriveById(oneDriveItem.ParentReference.Id, oneDriveItem.ParentReference.DriveId));
                            }
                        }
                    }

                    // Uploading succeeded
                    break;
                }
                catch (ArgumentNullException e)
                {
                    // If any other exception than the one we will get if the OneDrive API throws the random error, then ensure we pass on the exception instead of swallowing it
                    if (e.ParamName != "oneDriveUploadSession")
                    {
                        throw;
                    }

                    updateStatus(string.Format("Uploading the new KeePass database {1} to OneDrive (attempt {0})", uploadAttempts + 1, databaseConfig.KeePassDatabase.Name));
                }
            }

            if (uploadResult == null)
            {
                updateStatus(string.Format("Failed to upload the KeePass database {0}", databaseConfig.KeePassDatabase.Name));
                return(false);
            }

            // Delete the temporary database used for merging
            System.IO.File.Delete(temporaryKeePassDatabasePath);

            // The ETag changes with every request of the item so we use the CTag instead which only changes when the file changes
            databaseConfig.ETag = uploadResult.CTag;

            return(true);
        }
Пример #3
0
        /// <summary>
        /// Loads the items shared with the user in the "Shared with me" overview
        /// </summary>
        /// <param name="parentItem">OneDriveItem to display the child folders and items of. If not provided or NULL, the root folder will be used.</param>
        public async Task LoadSharedWithMeItems(OneDriveItem parentItem = null)
        {
            SharedWithMePicker.Items.Clear();

            if (parentItem == null)
            {
                // Get the root of the shared with me items
                var itemCollection = await _oneDriveApi.GetSharedWithMe();

                var sortedItemCollection = itemCollection.Collection.OrderBy(i => i.Name).OrderBy(i => i.Folder == null).OrderBy(i => i.RemoteItem.Folder == null);
                SharedWithMeUpButton.Enabled    = false;
                CurrentSharedWithMeOneDriveItem = null;
                GoToSharedWithMeRootTtoolStripMenuItem.Enabled = false;
                SharedWithMePath.Text = string.Empty;

                foreach (var oneDriveItem in sortedItemCollection)
                {
                    var oneDriveListViewItem = new ListViewItem
                    {
                        Text     = oneDriveItem.Name,
                        Tag      = oneDriveItem,
                        ImageKey = (oneDriveItem.RemoteItem != null && oneDriveItem.RemoteItem.Folder != null) || oneDriveItem.Folder != null ? "RemoteFolder" : "File",
                    };

                    OneDriveIdentity sharedInfo = null;
                    if (oneDriveItem.RemoteItem != null && oneDriveItem.RemoteItem.Shared != null && oneDriveItem.RemoteItem.Shared.Owner != null && oneDriveItem.RemoteItem.Shared.Owner.User != null)
                    {
                        sharedInfo = oneDriveItem.RemoteItem.Shared.Owner.User;
                    }
                    else if (oneDriveItem.Shared != null && oneDriveItem.Shared.Owner != null && oneDriveItem.Shared.Owner.User != null)
                    {
                        sharedInfo = oneDriveItem.Shared.Owner.User;
                    }
                    if (sharedInfo != null)
                    {
                        oneDriveListViewItem.ToolTipText = string.Format("Shared by {0} ({1})", sharedInfo.DisplayName, sharedInfo.Id);
                    }
                    else
                    {
                        oneDriveListViewItem.ToolTipText = "Shared by information not available";
                    }
                    if (oneDriveItem.Size > 0)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Size: {0:n0} bytes", oneDriveItem.Size);
                    }
                    else if (oneDriveItem.RemoteItem != null && oneDriveItem.RemoteItem.Size > 0)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Size: {0:n0} bytes", oneDriveItem.RemoteItem.Size);
                    }
                    if (oneDriveItem.CreatedDateTime != null)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Created: {0:d MMMM yyyy HH:mm:ss}", oneDriveItem.CreatedDateTime);
                    }
                    if (oneDriveItem.LastModifiedDateTime != null)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Last modified: {0:d MMMM yyyy HH:mm:ss}", oneDriveItem.LastModifiedDateTime);
                    }

                    SharedWithMePicker.Items.Add(oneDriveListViewItem);
                }
            }
            else
            {
                // Parent folder provided, get its children
                var itemCollection = await _oneDriveApi.GetAllChildrenFromDriveByFolderId(parentItem.RemoteItem != null?parentItem.RemoteItem.ParentReference.DriveId : parentItem.ParentReference.DriveId,
                                                                                          parentItem.RemoteItem != null?parentItem.RemoteItem.ParentReference != null?string.IsNullOrEmpty(parentItem.RemoteItem.ParentReference.Id) || parentItem.RemoteItem.Folder != null?parentItem.RemoteItem.Id : parentItem.RemoteItem.ParentReference.Id : parentItem.RemoteItem.Id : parentItem.Id);

                SharedWithMeUpButton.Tag = parentItem.Name == "root" || parentItem.ParentReference == null || parentItem.ParentReference.Id == null || (parentItem.ParentReference.Path != null && parentItem.ParentReference.Path.EndsWith("root:")) ? null : await _oneDriveApi.GetItemFromDriveById(parentItem.ParentReference.Id, parentItem.ParentReference.DriveId);

                CurrentSharedWithMeOneDriveItem = parentItem;
                GoToSharedWithMeRootTtoolStripMenuItem.Enabled = true;
                SharedWithMeUpButton.Enabled = true;

                if (CurrentSharedWithMeOneDriveItem.ParentReference != null && CurrentSharedWithMeOneDriveItem.ParentReference.Path != null)
                {
                    var rootLocation = CurrentSharedWithMeOneDriveItem.ParentReference.Path.IndexOf("root:");
                    SharedWithMePath.Text = ((rootLocation == -1 ? CurrentSharedWithMeOneDriveItem.ParentReference.Path : CurrentSharedWithMeOneDriveItem.ParentReference.Path.Remove(0, rootLocation + 5).TrimStart(new[] { '/' })) + "/" + CurrentSharedWithMeOneDriveItem.Name).TrimStart(new[] { '/' });
                }
                else
                {
                    SharedWithMePath.Text = CurrentSharedWithMeOneDriveItem.Name;
                }

                foreach (var oneDriveItem in itemCollection)
                {
                    var oneDriveListViewItem = new ListViewItem
                    {
                        Text     = oneDriveItem.Name,
                        Tag      = oneDriveItem,
                        ImageKey = oneDriveItem.Folder != null ? "RemoteFolder" : "File",
                    };

                    OneDriveIdentity sharedInfo = null;
                    if (oneDriveItem.RemoteItem != null && oneDriveItem.RemoteItem.Shared != null && oneDriveItem.RemoteItem.Shared.Owner != null && oneDriveItem.RemoteItem.Shared.Owner.User != null)
                    {
                        sharedInfo = oneDriveItem.RemoteItem.Shared.Owner.User;
                    }
                    else if (oneDriveItem.Shared != null && oneDriveItem.Shared.Owner != null && oneDriveItem.Shared.Owner.User != null)
                    {
                        sharedInfo = oneDriveItem.Shared.Owner.User;
                    }
                    if (sharedInfo != null)
                    {
                        oneDriveListViewItem.ToolTipText = string.Format("Shared by {0} ({1})", sharedInfo.DisplayName, sharedInfo.Id);
                    }
                    else
                    {
                        oneDriveListViewItem.ToolTipText = "Shared by information not available";
                    }
                    if (oneDriveItem.Size > 0)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Size: {0} bytes", oneDriveItem.Size);
                    }
                    else if (oneDriveItem.RemoteItem != null && oneDriveItem.RemoteItem.Size > 0)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Size: {0} bytes", oneDriveItem.RemoteItem.Size);
                    }
                    if (oneDriveItem.CreatedDateTime != null)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Created: {0:d MMMM yyyy HH:mm:ss}", oneDriveItem.CreatedDateTime);
                    }
                    if (oneDriveItem.LastModifiedDateTime != null)
                    {
                        oneDriveListViewItem.ToolTipText += Environment.NewLine + string.Format("Last modified: {0:d MMMM yyyy HH:mm:ss}", oneDriveItem.LastModifiedDateTime);
                    }

                    SharedWithMePicker.Items.Add(oneDriveListViewItem);
                }
            }

            // Define if the OK button should be enabled
            FileNameTextBox_TextChanged(null, null);
        }
        /// <summary>
        /// Loads the items shared with the user in the "Shared with me" overview
        /// </summary>
        /// <param name="parentItem">OneDriveItem to display the child folders and items of. If not provided or NULL, the root folder will be used.</param>
        public async Task LoadSharedWithMeItems(OneDriveItem parentItem = null)
        {
            SharedWithMePicker.Items.Clear();

            if (parentItem == null)
            {
                // Get the root of the shared with me items
                var itemCollection = await _oneDriveApi.GetSharedWithMe();

                SharedWithMeUpButton.Enabled    = false;
                CurrentSharedWithMeOneDriveItem = null;
                GoToSharedWithMeRootTtoolStripMenuItem.Enabled = false;
                SharedWithMePath.Text = string.Empty;

                foreach (var listViewItem in itemCollection.Collection.OrderBy(i => i.Name).OrderBy(i => i.Folder == null).OrderBy(i => i.RemoteItem.Folder == null).Select(oneDriveItem => new ListViewItem
                {
                    Text = oneDriveItem.Name,
                    Tag = oneDriveItem,
                    ImageKey = (oneDriveItem.RemoteItem != null && oneDriveItem.RemoteItem.Folder != null) || oneDriveItem.Folder != null ? "RemoteFolder" : "File"
                }))
                {
                    SharedWithMePicker.Items.Add(listViewItem);
                }
            }
            else
            {
                // Parent folder provided, get its children
                var itemCollection = await _oneDriveApi.GetAllChildrenFromDriveByFolderId(parentItem.RemoteItem != null?parentItem.RemoteItem.ParentReference.DriveId : parentItem.ParentReference.DriveId, parentItem.RemoteItem != null?parentItem.RemoteItem.ParentReference != null?string.IsNullOrEmpty(parentItem.RemoteItem.ParentReference.Id)?parentItem.RemoteItem.Id : parentItem.RemoteItem.ParentReference.Id : parentItem.RemoteItem.Id : parentItem.Id);

                SharedWithMeUpButton.Tag = parentItem.Name == "root" || parentItem.ParentReference.Id == null || (parentItem.ParentReference.Path != null && parentItem.ParentReference.Path.EndsWith("root:")) ? null : await _oneDriveApi.GetItemFromDriveById(parentItem.ParentReference.Id, parentItem.ParentReference.DriveId);

                CurrentSharedWithMeOneDriveItem = parentItem;
                GoToSharedWithMeRootTtoolStripMenuItem.Enabled = true;
                SharedWithMeUpButton.Enabled = CurrentSharedWithMeOneDriveItem.ParentReference != null;

                if (CurrentSharedWithMeOneDriveItem.ParentReference.Path != null)
                {
                    var rootLocation = CurrentSharedWithMeOneDriveItem.ParentReference.Path.IndexOf("root:");
                    SharedWithMePath.Text = ((rootLocation == -1 ? CurrentSharedWithMeOneDriveItem.ParentReference.Path : CurrentSharedWithMeOneDriveItem.ParentReference.Path.Remove(0, rootLocation + 5).TrimStart(new[] { '/' })) + "/" + CurrentSharedWithMeOneDriveItem.Name).TrimStart(new[] { '/' });
                }
                else
                {
                    SharedWithMePath.Text = CurrentSharedWithMeOneDriveItem.Name;
                }

                foreach (var listViewItem in itemCollection.Select(oneDriveItem => new ListViewItem
                {
                    Text = oneDriveItem.Name,
                    Tag = oneDriveItem,
                    ImageKey = oneDriveItem.Folder != null ? "RemoteFolder" : "File"
                }))
                {
                    SharedWithMePicker.Items.Add(listViewItem);
                }
            }

            // Define if the OK button should be enabled
            FileNameTextBox_TextChanged(null, null);
        }