Add missing class to set passwords.

src/com/keepassdroid/database/SetPassword.java

@@ -0,0 +1,102 @@
+/*
+ * Copyright 2009 Brian Pellin.
+ *     
+ * This file is part of KeePassDroid.
+ *
+ *  KeePassDroid 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.
+ *
+ *  KeePassDroid 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 KeePassDroid.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+package com.keepassdroid.database;
+
+import java.io.IOException;
+
+import org.phoneid.keepassj2me.PwManager;
+
+import com.keepassdroid.Database;
+import com.keepassdroid.keepasslib.InvalidKeyFileException;
+
+public class SetPassword extends RunnableOnFinish {
+	
+	private String mPassword;
+	private String mKeyfile;
+	private Database mDb;
+	
+	public SetPassword(Database db, String password, String keyfile, OnFinish finish) {
+		super(finish);
+		
+		mDb = db;
+		mPassword = password;
+		mKeyfile = keyfile;
+	}
+
+	@Override
+	public void run() {
+		PwManager pm = mDb.mPM;
+		
+		byte[] backupKey = new byte[pm.masterKey.length];
+		System.arraycopy(pm.masterKey, 0, backupKey, 0, backupKey.length);
+
+		// Set key
+		try {
+			pm.setMasterKey(mPassword, mKeyfile);
+		} catch (InvalidKeyFileException e) {
+			erase(backupKey);
+			finish(false, e.getMessage());
+			return;
+		} catch (IOException e) {
+			erase(backupKey);
+			finish(false, e.getMessage());
+			return;
+		}
+		
+		// Save Database
+		mFinish = new AfterSave(backupKey, mFinish);
+		SaveDB save = new SaveDB(mDb, mFinish);
+		save.run();
+	}
+	
+	private class AfterSave extends OnFinish {
+		private byte[] mBackup;
+		
+		public AfterSave(byte[] backup, OnFinish finish) {
+			super(finish);
+			
+			mBackup = backup;
+		}
+
+		@Override
+		public void run() {
+			if ( ! mSuccess ) {
+				// Erase the current master key
+				erase(mDb.mPM.masterKey);
+				mDb.mPM.masterKey = mBackup;
+			}
+			
+			super.run();
+		}
+
+	}
+	
+	/** Overwrite the array as soon as we don't need it to avoid keeping the extra data in memory
+	 * @param array
+	 */
+	private void erase(byte[] array) {
+		if ( array == null ) return;
+		
+		for ( int i = 0; i < array.length; i++ ) {
+			array[i] = 0;
+		}
+	}
+
+}