137 lines
5.2 KiB
C++
137 lines
5.2 KiB
C++
/******************************************************************************
|
|
* Copyright (C) 2011 Frank Osterfeld <frank.osterfeld@gmail.com> *
|
|
* *
|
|
* 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. For licensing and distribution *
|
|
* details, check the accompanying file 'COPYING'. *
|
|
*****************************************************************************/
|
|
#include "keychain_p.h"
|
|
|
|
#include <QSettings>
|
|
|
|
#include <windows.h>
|
|
#include <wincrypt.h>
|
|
|
|
#include <memory>
|
|
|
|
using namespace QKeychain;
|
|
|
|
QtKeychain::Error Keychain::Private::readEntryImpl( QByteArray* pw,
|
|
const QString& key,
|
|
QString* err ) {
|
|
Q_ASSERT( pw );
|
|
Q_ASSERT( err );
|
|
err->clear();
|
|
|
|
//Use settings member if there, create local settings object if not
|
|
std::auto_ptr<QSettings> local( !settings ? new QSettings( service ) : 0 );
|
|
QSettings* actual = settings ? settings.data() : local.get();
|
|
|
|
QByteArray encrypted = actual->value( key ).toByteArray();
|
|
if ( encrypted.isNull() ) {
|
|
*err = tr("Entry not found");
|
|
return EntryNotFound;
|
|
}
|
|
|
|
DATA_BLOB blob_in, blob_out;
|
|
|
|
blob_in.pbData = reinterpret_cast<BYTE*>( encrypted.data() );
|
|
blob_in.cbData = encrypted.size();
|
|
|
|
const BOOL ret = CryptUnprotectData( &blob_in,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
0,
|
|
&blob_out );
|
|
if ( !ret ) {
|
|
*err = tr("Could not decrypt data");
|
|
return OtherError;
|
|
}
|
|
*pw = QByteArray( reinterpret_cast<char*>( blob_out.pbData ), blob_out.cbData );
|
|
SecureZeroMemory( blob_out.pbData, blob_out.cbData );
|
|
LocalFree( blob_out.pbData );
|
|
return NoError;
|
|
}
|
|
|
|
QtKeychain::Error Keychain::Private::writeEntryImpl( const QString& key,
|
|
const QByteArray& data_,
|
|
QString* err ) {
|
|
Q_ASSERT( err );
|
|
err->clear();
|
|
QByteArray data = data_;
|
|
DATA_BLOB blob_in, blob_out;
|
|
blob_in.pbData = reinterpret_cast<BYTE*>( data.data() );
|
|
blob_in.cbData = data.size();
|
|
const BOOL res = CryptProtectData( &blob_in,
|
|
L"QKeychain-encrypted data",
|
|
NULL,
|
|
NULL,
|
|
NULL,
|
|
0,
|
|
&blob_out );
|
|
if ( !res ) {
|
|
*err = tr("Encryption failed"); //TODO more details available?
|
|
return OtherError;
|
|
}
|
|
|
|
const QByteArray encrypted( reinterpret_cast<char*>( blob_out.pbData ), blob_out.cbData );
|
|
LocalFree( blob_out.pbData );
|
|
|
|
//Use settings member if there, create local settings object if not
|
|
std::auto_ptr<QSettings> local( !settings ? new QSettings( service ) : 0 );
|
|
QSettings* actual = settings ? settings.data() : local.get();
|
|
actual->setValue( key, encrypted );
|
|
actual->sync();
|
|
if ( actual->status() != QSettings::NoError ) {
|
|
*err = actual->status() == QSettings::AccessError
|
|
? tr("Could not store encrypted data in settings: access error")
|
|
: tr("Could not store encrypted data in settings: format error");
|
|
return OtherError;
|
|
}
|
|
|
|
return NoError;
|
|
}
|
|
|
|
QtKeychain::Error Keychain::Private::deleteEntryImpl( const QString& key,
|
|
QString* err ) {
|
|
Q_ASSERT( err );
|
|
err->clear();
|
|
std::auto_ptr<QSettings> local( !settings ? new QSettings( service ) : 0 );
|
|
QSettings* actual = settings ? settings.data() : local.get();
|
|
actual->remove( key );
|
|
actual->sync();
|
|
if ( actual->status() != QSettings::NoError ) {
|
|
*err = actual->status() == QSettings::AccessError
|
|
? tr("Could not delete encrypted data from settings: access error")
|
|
: tr("Could not delete encrypted data from settings: format error");
|
|
return OtherError;
|
|
}
|
|
|
|
return NoError;
|
|
}
|
|
|
|
|
|
QtKeychain::Error Keychain::Private::entryExistsImpl( bool* exists,
|
|
const QString& key,
|
|
QString* err ) {
|
|
Q_ASSERT( exists );
|
|
Q_ASSERT( err );
|
|
err->clear();
|
|
*exists = false;
|
|
std::auto_ptr<QSettings> local( !settings ? new QSettings( service ) : 0 );
|
|
QSettings* actual = settings ? settings.data() : local.get();
|
|
const bool ex = actual->contains( key );
|
|
if ( actual->status() != QSettings::NoError ) {
|
|
*err = actual->status() == QSettings::AccessError
|
|
? tr("Could not read settings: access error")
|
|
: tr("Could not read settings: format error");
|
|
return OtherError;
|
|
}
|
|
|
|
*exists = ex;
|
|
return NoError;
|
|
}
|