keepass2android/src/keepass2android/Database.cs

/*
This file is part of Keepass2Android, Copyright 2013 Philipp Crocoll. This file is based on Keepassdroid, Copyright Brian Pellin.

  Keepass2Android 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.

  Keepass2Android 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 Keepass2Android.  If not, see <http://www.gnu.org/licenses/>.
  */

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;

using Android.App;
using Android.Content;
using Android.OS;
using Android.Runtime;
using Android.Views;
using Android.Widget;

using KeePassLib;
using KeePassLib.Interfaces;
using KeePassLib.Serialization;

namespace keepass2android
{

	public class Database {
		

		public Dictionary<PwUuid, PwGroup> groups = new Dictionary<PwUuid, PwGroup>(new PwUuidEqualityComparer());
		public Dictionary<PwUuid, PwEntry> entries = new Dictionary<PwUuid, PwEntry>(new PwUuidEqualityComparer());
		public HashSet<PwGroup> dirty = new HashSet<PwGroup>(new PwGroupEqualityFromIdComparer());
		public PwGroup root;
		public PwDatabase pm;
		public IOConnectionInfo mIoc;
		public SearchDbHelper searchHelper;
		
		public DrawableFactory drawFactory = new DrawableFactory();
		
		private bool loaded = false;


		public bool Loaded {
			get { return loaded;}
			set { loaded = value; }
		}

		public bool Open
		{
			get { return Loaded && (!Locked); }
		}

		bool locked;
		public bool Locked
		{
			get
			{
				return locked;
			}
			set
			{
				locked = value;
			}
		}
		
	
		
		public void LoadData (Context ctx, IOConnectionInfo iocInfo, String password, String keyfile, UpdateStatus status)
		{
			mIoc = iocInfo;

			KeePassLib.PwDatabase pwDatabase = new KeePassLib.PwDatabase ();

			KeePassLib.Keys.CompositeKey key = new KeePassLib.Keys.CompositeKey ();
			key.AddUserKey (new KeePassLib.Keys.KcpPassword (password));
			if (!String.IsNullOrEmpty (keyfile)) {

				try { key.AddUserKey(new KeePassLib.Keys.KcpKeyFile(keyfile)); }
				catch(Exception)
				{
					throw new KeyFileException();
				}
			}
			
			pwDatabase.Open(iocInfo, key, status);

			root = pwDatabase.RootGroup;
			populateGlobals(root);


			Loaded = true;
			pm = pwDatabase;
			searchHelper = new SearchDbHelper(ctx);
		}

		bool quickUnlockEnabled = false;
		public bool QuickUnlockEnabled
		{
			get
			{
				return quickUnlockEnabled;
			}
			set
			{
				quickUnlockEnabled = value;
			}
		}

		//KeyLength of QuickUnlock at time of loading the database.
		//This is important to not allow an attacker to set the length to 1 when QuickUnlock is started already.
		public int QuickUnlockKeyLength
		{
			get;
			set;
		}
		
		public PwGroup SearchForText(String str) {
			PwGroup group = searchHelper.searchForText(this, str);
			
			return group;
			
		}

		public PwGroup Search(SearchParameters searchParams)
		{
			return searchHelper.search(this, searchParams);
		}

		
		public PwGroup SearchForExactUrl(String url) {
			PwGroup group = searchHelper.searchForExactUrl(this, url);
			
			return group;
			
		}

		public PwGroup SearchForHost(String url) {
			PwGroup group = searchHelper.searchForHost(this, url);
			
			return group;
			
		}


		public void SaveData(Context ctx)  {
			ISharedPreferences prefs = Android.Preferences.PreferenceManager.GetDefaultSharedPreferences(ctx);
			pm.UseFileTransactions = prefs.GetBoolean(ctx.GetString(Resource.String.UseFileTransactions_key), true);
			pm.Save(null);

		}
		class SaveStatusLogger: IStatusLogger
		{
			#region IStatusLogger implementation
			public void StartLogging (string strOperation, bool bWriteOperationToLog)
			{
			}
			public void EndLogging ()
			{
			}
			public bool SetProgress (uint uPercent)
			{
				Android.Util.Log.Debug("DEBUG", "Progress: " + uPercent+"%");
				return true;
			}
			public bool SetText (string strNewText, LogStatusType lsType)
			{
				Android.Util.Log.Debug("DEBUG", strNewText);
				return true;
			}
			public bool ContinueWork ()
			{
				return true;
			}
			#endregion
		}
		
		private void populateGlobals (PwGroup currentGroup)
		{
			
			var childGroups = currentGroup.Groups;
			var childEntries = currentGroup.Entries;

			foreach (PwEntry e in childEntries) {
				entries [e.Uuid] = e;
			}
			foreach (PwGroup g in childGroups) {
				groups[g.Uuid] = g;
				populateGlobals(g);
			}
		}
		
		public void Clear() {
			groups.Clear();
			entries.Clear();
			dirty.Clear();
			drawFactory.Clear();
			
			root = null;
			pm = null;
			mIoc = null;
			loaded = false;
			locked = false;
		}
		
		public void markAllGroupsAsDirty() {
			foreach ( PwGroup group in groups.Values ) {
				dirty.Add(group);
			}
			

		}
		
		
	}

	/*
		public void LoadData (Context mCtx, string mFileName, string mPass, string mKey, UpdateStatus mStatus)
		{
			KeePassLib.PwDatabase pwDatabase = new KeePassLib.PwDatabase();
			KeePassLib.Serialization.IOConnectionInfo iocInfo = KeePassLib.Serialization.IOConnectionInfo.FromPath("/sdcard/keepass2androidtest.kdbx");
			KeePassLib.Serialization.IOConnectionInfo iocInfoSave = KeePassLib.Serialization.IOConnectionInfo.FromPath("/sdcard/keepass2androidtestSaved.kdbx");
			KeePassLib.Keys.CompositeKey key = new KeePassLib.Keys.CompositeKey();
			key.AddUserKey(new KeePassLib.Keys.KcpPassword("test"));
			
			pwDatabase.Open(iocInfo, key, new LogToButton(this));
			pwDatabase.RootGroup.AddGroup(new KeePassLib.PwGroup(true, true, "generatedFromKp2ANeu", KeePassLib.PwIcon.Folder), true);
			pwDatabase.SaveAs(iocInfoSave,false,new LogToButton(this));
			pwDatabase.Close();
			//KeePassLib.Serialization.KdbxFile f = new KeePassLib.Serialization.KdbxFile(pwDatabase);
		}
*/

}