keepassx1/src/dialogs/PasswordDlg.cpp

/***************************************************************************
 *   Copyright (C) 2005-2006 by Tarek Saidi                                *
 *   tarek@linux                                                           *
 *                                                                         *
 *   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.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#include "main.h"
#include "PwmConfig.h"
#include "PasswordDlg.h"
#include <QFileDialog>
#include <QDir>
#include <QStringList>
#include <QCheckBox>
#include <QLineEdit>
#include <QComboBox>
#include <QPushButton>
#include <QMessageBox>


CPasswordDialog::CPasswordDialog(QWidget* parent,IDatabase* DB,bool ShowExitButton,bool ChangeKeyMode)
: QDialog(parent)
{
	setupUi(this);
	createBanner(Banner,Icon_Key32x32,tr("Database Key"));
	db=DB;
	QDir media(config.MountDir);
	if(media.exists()){
		QStringList Paths;
		Paths=media.entryList(QStringList()<<"*",QDir::Dirs);
		Paths.erase(Paths.begin()); // delete "."
		Paths.erase(Paths.begin()); // delete ".."
		for(int i=0;i<Paths.count();i++)
			Combo_Dirs->addItem(config.MountDir+Paths[i]);
	}
	
	Combo_Dirs->setEditText(QString());
	if(config.RememberLastKey && !ChangeKeyMode){
		switch(config.LastKeyType){
			//case PASSWORD:	setStatePasswordOnly(); break; //Password-Only is already the default
			case KEYFILE:	setStateKeyFileOnly();
							Combo_Dirs->setEditText(config.LastKeyLocation);
							break;
			case BOTH:		setStateBoth();
							CheckBox_Both->setChecked(true);
							Combo_Dirs->setEditText(config.LastKeyLocation);
							break;
		}
	}
	
	connect( Combo_Dirs, SIGNAL( editTextChanged(const QString&) ),this, SLOT( OnComboTextChanged(const QString&)));
	connect( ButtonCancel, SIGNAL( clicked() ), this, SLOT( OnCancel() ) );
	connect( Edit_Password, SIGNAL( textChanged(const QString&) ), this, SLOT( OnPasswordChanged(const QString&) ) );
	connect( CheckBox_Both, SIGNAL( stateChanged(int) ), this, SLOT( OnCheckBox_BothChanged(int) ) );
	connect( ButtonChangeEchoMode, SIGNAL( clicked() ), this, SLOT( ChangeEchoMode() ) );
	connect( Edit_Password, SIGNAL( returnPressed() ), this, SLOT( OnOK() ) );
	connect( Edit_PasswordRep, SIGNAL( returnPressed() ), this, SLOT( OnOK() ) );
	connect( ButtonExit, SIGNAL( clicked()),this,SLOT(OnButtonExit()));
	
	ButtonExit->setVisible(ShowExitButton);
	Mode_Set=ChangeKeyMode;
	if(!ChangeKeyMode){
		Edit_PasswordRep->hide();
		Label_PasswordRep->hide();
		connect( ButtonOK, SIGNAL( clicked() ), this, SLOT( OnOK() ) );
		connect( ButtonBrowse, SIGNAL( clicked() ), this, SLOT( OnButtonBrowse() ) );
	}else{
		connect( ButtonOK, SIGNAL( clicked() ), this, SLOT( OnOK_Set() ) );
		connect( ButtonBrowse, SIGNAL( clicked() ), this, SLOT( OnButtonBrowse_Set() ) );
	}
	
	if(!config.ShowPasswords)ChangeEchoMode();
}


void CPasswordDialog::setStatePasswordOnly(){
	Combo_Dirs->setEnabled(false);
	ButtonBrowse->setEnabled(false);
	Label_KeyFile->setEnabled(false);
	Label_Password->setEnabled(true);
	Label_PasswordRep->setEnabled(true);
	Edit_Password->setEnabled(true);
	Edit_PasswordRep->setEnabled(true);
	ButtonChangeEchoMode->setEnabled(true);
	KeyType=PASSWORD;
}


void CPasswordDialog::setStateKeyFileOnly(){
	Combo_Dirs->setEnabled(true);
	ButtonBrowse->setEnabled(true);
	Label_KeyFile->setEnabled(true);
	Label_Password->setEnabled(false);
	Label_PasswordRep->setEnabled(false);
	Edit_Password->setEnabled(false);
	Edit_PasswordRep->setEnabled(false);
	ButtonChangeEchoMode->setEnabled(false);
	KeyType=KEYFILE;
}


void CPasswordDialog::setStateBoth(){
	Combo_Dirs->setEnabled(true);
	ButtonBrowse->setEnabled(true);
	Label_KeyFile->setEnabled(true);
	Label_Password->setEnabled(true);
	Label_PasswordRep->setEnabled(true);
	Edit_Password->setEnabled(true);
	Edit_PasswordRep->setEnabled(true);
	ButtonChangeEchoMode->setEnabled(true);
	KeyType=BOTH;
}


void CPasswordDialog::OnButtonBrowse()
{
	QFileDialog FileDlg(this,tr("Select a Key File"),QDir::homePath());
	FileDlg.setFilters(QStringList()<<tr("All Files (*)") << tr("Key Files (*.key)"));
	FileDlg.setFileMode(QFileDialog::ExistingFile);
	if(!FileDlg.exec())return;
	if(!FileDlg.selectedFiles().size())return;	
	QFile file(FileDlg.selectedFiles()[0]);
	if(file.exists()){
		Combo_Dirs->setEditText(FileDlg.selectedFiles()[0]);
		return;
	}
	QMessageBox::warning(this,tr("Error"),tr("Unexpected Error: File does not exist."),tr("OK"),"","",0,0);
}

void CPasswordDialog::OnButtonBrowse_Set()
{
	QFileDialog FileDlg(this,tr("Select a Key File"),QDir::homePath());
	FileDlg.setFilters(QStringList()<<tr("All Files (*)") << tr("Key Files (*.key)"));
	FileDlg.setFileMode(QFileDialog::AnyFile);
	if(!FileDlg.exec())return;
	if(!FileDlg.selectedFiles().size())return;	
	Combo_Dirs->setEditText(FileDlg.selectedFiles()[0]);
}

void CPasswordDialog::OnCancel()
{
	done(0);
}

void CPasswordDialog::OnOK(){
	password=Edit_Password->text();
	keyfile=Combo_Dirs->currentText();

	if(password=="" && keyfile==""){
		QMessageBox::warning(this,tr("Error"),tr("Please enter a Password or select a key file."),tr("OK"),"","",0,0);
		return;
	}

	if(KeyType==BOTH){
		if(password==""){
			QMessageBox::warning(this,tr("Error"),tr("Please enter a Password."),tr("OK"),"","",0,0);
			return;}
		if(keyfile==""){
			QMessageBox::warning(this,tr("Error"),tr("Please choose a key file."),tr("OK"),"","",0,0);
			return;}
	}

	if(KeyType==BOTH || KeyType==KEYFILE){
		QFileInfo fileinfo(keyfile);
		if(!fileinfo.exists()){
			QMessageBox::warning(this,tr("Error"),tr("The selected key file or directory does not exist."),tr("OK"),"","",0,0);
			return;
		}
		if(!fileinfo.isReadable()){
			QMessageBox::warning(this,tr("Error"),tr("The selected key file or directory is not readable.\nPlease check your permissions."),tr("OK"),"","",0,0);
			return;
		}
		if(fileinfo.isDir()){
			if(keyfile.right(1)!="/")keyfile+="/";
			QFile file(keyfile+"pwsafe.key");
			if(!file.exists()){				
				QDir dir(keyfile);
				QStringList files;
				files=dir.entryList(QStringList()<<"*.key",QDir::Files);
				if(!files.size()){
					QMessageBox::warning(this,tr("Error"),tr("The given directory does not contain any key files."),tr("OK"),"","",0,0);
					return;}
				if(files.size()>1){
					QMessageBox::warning(this,tr("Error"),tr("The given directory contains more then one key file.\nPlease specify the key file directly."),tr("OK"),"","",0,0);
					return;}
				QFile file(keyfile+files[0]);
				Q_ASSERT(file.exists());
				if(!QFileInfo(file).isReadable()){
					QMessageBox::warning(this,tr("Error"),tr("The key file found in the given directory is not readable.\nPlease check your permissions."),tr("OK"),"","",0,0);
					return;}				
				keyfile+=files[0];
				}
			else{ /* pwsafe.key exists */
				if(!QFileInfo(file).isReadable()){
					QMessageBox::warning(this,tr("Error"),tr("The key file found in the given directory is not readable.\nPlease check your permissions."),tr("OK"),"","",0,0);
					return;}			
				keyfile+="pwsafe.key";
				}
		}
		else{ /* not a directory */
			QFile file(keyfile);
			if(!file.exists()){
				QMessageBox::warning(this,tr("Error"),tr("Key file could not be found."),tr("OK"),"","",0,0);
				return;}
			if(!QFileInfo(file).isReadable()){
				QMessageBox::warning(this,tr("Error"),tr("Key file is not readable.\nPlease check your permissions."),tr("OK"),"","",0,0);
				return;}
		}

	}
if(doAuth())done(1);
}

void CPasswordDialog::OnOK_Set(){
	password=Edit_Password->text();
	if(password!=Edit_PasswordRep->text()){
		QMessageBox::warning(this,tr("Warning"),tr("Password an password repetition are not equal.\nPlease check your input."),tr("OK"),"","",0,0);
		return;
	}
	keyfile=Combo_Dirs->currentText();
	if(password=="" && keyfile==""){
		QMessageBox::warning(this,tr("Error"),tr("Please enter a password or select a key file."),tr("OK"),"","",0,0);
		return;
	}
	
	if(keyfile!=QString()){	
		QFile file(keyfile);
		if(QFileInfo(file).isDir()){
			if(keyfile.right(1)!="/")keyfile+="/";		
			keyfile+="pwsafe.key";
		}
		if(file.exists()){
			switch(QMessageBox::question(this,tr("File exists."),tr("A file with the selected name already exists, should this file be used as key file or do you want to overwrite it with a newly generated one?"),
					tr("Use"),tr("Overwrite"),tr("Cancel"),0,2)){
							case 0:
								OverwriteKeyFile=false;
								break;
							case 1:
								OverwriteKeyFile=true;
								break;
							case 2:
								return; 
			}
		}		
		IFilePasswordAuth* DbAuth=dynamic_cast<IFilePasswordAuth*>(db);
		if(OverwriteKeyFile){
			if(!DbAuth->createKeyFile(keyfile,32,true)){
				QMessageBox::warning(this,tr("Error"),tr("Key file could not be created.\n%1").arg(db->getError()),tr("OK"),"","",0,0);
				return;
			}		
		}
	}		
	if(doAuth())done(1);
}

bool CPasswordDialog::doAuth(){
	IFilePasswordAuth* DbAuth=dynamic_cast<IFilePasswordAuth*>(db);
	if(password!=QString() && keyfile==QString()){
		DbAuth->authByPwd(password);
	}
	if(password==QString() && keyfile!=QString()){
		if(!DbAuth->authByFile(keyfile))return false;	
	}
	if(password!=QString() && keyfile!=QString()){
		if(!DbAuth->authByFile(keyfile))return false;	
	}	
	
	if(config.RememberLastKey){
		config.LastKeyLocation=keyfile;
		config.LastKeyType=KeyType;
	}
	return true;
	
}

void CPasswordDialog::OnPasswordChanged(const QString &txt){
Edit_PasswordRep->setText("");
if(CheckBox_Both->isChecked() || txt==QString())
	setStateBoth();
else
	setStatePasswordOnly();
}

void CPasswordDialog::OnComboTextChanged(const QString& txt){
if(CheckBox_Both->isChecked() || txt==QString())
	setStateBoth();
else
	setStateKeyFileOnly();
}



void CPasswordDialog::OnCheckBox_BothChanged(int state){
if(state==Qt::Checked)
	setStateBoth();
if(state==Qt::Unchecked){
	if(Edit_Password->text()!=QString() && Combo_Dirs->currentText()!=QString()){
		Combo_Dirs->setEditText(QString());
		setStatePasswordOnly();
	}
	else{
		if(Edit_Password->text()==QString())
			setStateKeyFileOnly();
		else
			setStatePasswordOnly();
	}

}

}

void CPasswordDialog::ChangeEchoMode(){
if(Edit_Password->echoMode()==QLineEdit::Normal){
	Edit_Password->setEchoMode(QLineEdit::Password);
	Edit_PasswordRep->setEchoMode(QLineEdit::Password);}
else{
	Edit_Password->setEchoMode(QLineEdit::Normal);
	Edit_PasswordRep->setEchoMode(QLineEdit::Normal);}
}


void CPasswordDialog::OnButtonExit(){
	done(2);
}