/***************************************************************************
* 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 <QFileDialog>
# include <QDir>
# include <QStringList>
# include <QCheckBox>
# include <QLineEdit>
# include <QComboBox>
# include <QPushButton>
# include <QMessageBox>
# include <QStringList>
# include "main.h"
# include "PwmConfig.h"
# include "PasswordDlg.h"
# include "lib/FileDialogs.h"
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 ( settings - > value ( " RememberLastKey " , true ) . toBool ( ) & & ! ChangeKeyMode ) {
QString LastKeyType = settings - > value ( " LastKeyType " , " " ) . toString ( ) ;
if ( LastKeyType = = " KeyFile " ) {
setStateKeyFileOnly ( ) ;
Combo_Dirs - > setEditText ( QDir : : cleanPath ( QDir : : current ( ) . absoluteFilePath ( settings - > value ( " LastKeyFile " , " " ) . toString ( ) ) ) ) ;
}
else if ( LastKeyType = = " Composite " ) {
setStateBoth ( ) ;
CheckBox_Both - > setChecked ( true ) ;
Combo_Dirs - > setEditText ( QDir : : cleanPath ( QDir : : current ( ) . absoluteFilePath ( settings - > value ( " LastKeyFile " , " " ) . toString ( ) ) ) ) ;
}
// if(LastKeyType==Password){... is not required because it is already the default state.
}
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 ( ChangeEchoModeDatabaseKey ( ) ) ) ;
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 . ShowPasswordsPasswordDlg ) ChangeEchoModeDatabaseKey ( ) ;
}
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 ( )
{
QString filename = KpxFileDialogs : : openExistingFile ( this , " PasswordDlg " , tr ( " Select a Key File " ) ,
QStringList ( ) < < tr ( " All Files (*) " ) < < tr ( " Key Files (*.key) " ) ) ;
if ( filename ! = QString ( ) ) {
Combo_Dirs - > setEditText ( filename ) ;
}
return ;
}
void CPasswordDialog : : OnButtonBrowse_Set ( )
{
QString filename = KpxFileDialogs : : saveFile ( this , " PasswordDlg " , tr ( " Select a Key File " ) ,
QStringList ( ) < < tr ( " All Files (*) " ) < < tr ( " Key Files (*.key) " ) ,
false ) ;
if ( filename ! = QString ( ) ) {
Combo_Dirs - > setEditText ( filename ) ;
}
return ;
}
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. \n Please 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. \n Please 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. \n Please 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. \n Please 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. \n Please 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. \n Please 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 ( settings - > value ( " RememberLastKey " , true ) . toBool ( ) ) {
QString KeyLocation = keyfile ;
if ( settings - > value ( " SaveRelativePaths " , true ) . toBool ( ) ) {
KeyLocation = KeyLocation . left ( KeyLocation . lastIndexOf ( " / " ) ) ;
KeyLocation = makePathRelative ( KeyLocation , QDir : : currentPath ( ) ) + keyfile . right ( keyfile . length ( ) - keyfile . lastIndexOf ( " / " ) - 1 ) ;
}
settings - > setValue ( " LastKeyFile " , KeyLocation ) ;
if ( KeyType = = PASSWORD )
settings - > setValue ( " LastKeyType " , " Password " ) ;
if ( KeyType = = KEYFILE )
settings - > setValue ( " LastKeyType " , " KeyFile " ) ;
if ( KeyType = = BOTH )
settings - > setValue ( " LastKeyType " , " Composite " ) ;
}
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 : : ChangeEchoModeDatabaseKey ( ) {
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 ) ;
}