Setup solution

ModernKeePassLib/Keys/KcpUserAccount.cs

@@ -0,0 +1,167 @@
+/*
+  KeePass Password Safe - The Open-Source Password Manager
+  Copyright (C) 2003-2019 Dominik Reichl <dominik.reichl@t-online.de>
+
+  This program is free software; you can redistribute it and/or modify
+  it under the terms of the GNU General Public License as published by
+  the Free Software Foundation; either version 2 of the License, or
+  (at your option) any later version.
+
+  This program is distributed in the hope that it will be useful,
+  but WITHOUT ANY WARRANTY; without even the implied warranty of
+  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+  GNU General Public License for more details.
+
+  You should have received a copy of the GNU General Public License
+  along with this program; if not, write to the Free Software
+  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+*/
+
+using System;
+using System.Diagnostics;
+using System.IO;
+using System.Security;
+
+#if ModernKeePassLib
+using Windows.Storage;
+using System.Security.Cryptography;
+#endif
+
+using ModernKeePassLib.Cryptography;
+using ModernKeePassLib.Native;
+using ModernKeePassLib.Resources;
+using ModernKeePassLib.Security;
+using ModernKeePassLib.Utility;
+
+namespace ModernKeePassLib.Keys
+{
+	/// <summary>
+	/// A user key depending on the currently logged on Windows user account.
+	/// </summary>
+	public sealed class KcpUserAccount : IUserKey
+	{
+		private ProtectedBinary m_pbKeyData = null;
+
+		// Constant initialization vector (unique for KeePass)
+		private static readonly byte[] m_pbEntropy = new byte[] {
+			0xDE, 0x13, 0x5B, 0x5F, 0x18, 0xA3, 0x46, 0x70,
+			0xB2, 0x57, 0x24, 0x29, 0x69, 0x88, 0x98, 0xE6
+		};
+
+		private const string UserKeyFileName = "ProtectedUserKey.bin";
+
+		/// <summary>
+		/// Get key data. Querying this property is fast (it returns a
+		/// reference to a cached <c>ProtectedBinary</c> object).
+		/// If no key data is available, <c>null</c> is returned.
+		/// </summary>
+		public ProtectedBinary KeyData
+		{
+			get { return m_pbKeyData; }
+		}
+
+		/// <summary>
+		/// Construct a user account key.
+		/// </summary>
+		public KcpUserAccount()
+		{
+			if(!CryptoUtil.IsProtectedDataSupported)
+				throw new PlatformNotSupportedException(); // Windows 98/ME
+
+			byte[] pbKey = LoadUserKey(false);
+			if(pbKey == null) pbKey = CreateUserKey();
+			if(pbKey == null) // Should never happen
+			{
+				Debug.Assert(false);
+				throw new SecurityException(KLRes.UserAccountKeyError);
+			}
+
+			m_pbKeyData = new ProtectedBinary(true, pbKey);
+			MemUtil.ZeroByteArray(pbKey);
+		}
+
+		// public void Clear()
+		// {
+		//	m_pbKeyData = null;
+		// }
+
+		private static string GetUserKeyFilePath(bool bCreate)
+		{
+#if ModernKeePassLib
+			string strUserDir = Windows.Storage.ApplicationData.Current.RoamingFolder.Path;
+#else
+			string strUserDir = Environment.GetFolderPath(
+				Environment.SpecialFolder.ApplicationData);
+#endif
+
+			strUserDir = UrlUtil.EnsureTerminatingSeparator(strUserDir, false);
+			strUserDir += PwDefs.ShortProductName;
+
+#if !ModernKeePassLib
+
+			if(bCreate && !Directory.Exists(strUserDir))
+				Directory.CreateDirectory(strUserDir);
+#endif
+
+			strUserDir = UrlUtil.EnsureTerminatingSeparator(strUserDir, false);
+			return (strUserDir + UserKeyFileName);
+		}
+
+		private static byte[] LoadUserKey(bool bThrow)
+		{
+			byte[] pbKey = null;
+
+#if !KeePassLibSD
+			try
+			{
+				string strFilePath = GetUserKeyFilePath(false);
+
+#if ModernKeePassLib
+                var fileStream = StorageFile.GetFileFromPathAsync(strFilePath).GetAwaiter().GetResult().OpenStreamForReadAsync().GetAwaiter().GetResult();
+                var pbProtectedKey = new byte[(int)fileStream.Length];
+                fileStream.Read(pbProtectedKey, 0, (int)fileStream.Length);
+                fileStream.Dispose();
+#else
+				byte[] pbProtectedKey = File.ReadAllBytes(strFilePath);
+#endif
+
+				pbKey = CryptoUtil.UnprotectData(pbProtectedKey, m_pbEntropy,
+					DataProtectionScope.CurrentUser);
+			}
+			catch(Exception)
+			{
+				if(bThrow) throw;
+				pbKey = null;
+			}
+#endif
+
+			return pbKey;
+		}
+
+		private static byte[] CreateUserKey()
+		{
+#if KeePassLibSD
+			return null;
+#else
+			string strFilePath = GetUserKeyFilePath(true);
+
+			byte[] pbRandomKey = CryptoRandom.Instance.GetRandomBytes(64);
+			byte[] pbProtectedKey = CryptoUtil.ProtectData(pbRandomKey,
+				m_pbEntropy, DataProtectionScope.CurrentUser);
+#if ModernKeePassLib
+            var fileStream = StorageFile.GetFileFromPathAsync(strFilePath).GetAwaiter().GetResult().OpenStreamForWriteAsync().GetAwaiter().GetResult();
+            fileStream.Write(pbProtectedKey, 0, (int)fileStream.Length);
+            fileStream.Dispose();
+#else
+			File.WriteAllBytes(strFilePath, pbProtectedKey);
+#endif
+
+			byte[] pbKey = LoadUserKey(true);
+			Debug.Assert(MemUtil.ArraysEqual(pbKey, pbRandomKey));
+
+			MemUtil.ZeroByteArray(pbRandomKey);
+			return pbKey;
+#endif
+		}
+	}
+}