215 lines
6.1 KiB
C++
215 lines
6.1 KiB
C++
/*
|
|
Derived from source code of TrueCrypt 7.1a, which is
|
|
Copyright (c) 2008-2012 TrueCrypt Developers Association and which is governed
|
|
by the TrueCrypt License 3.0.
|
|
|
|
Modifications and additions to the original source code (contained in this file)
|
|
and all other portions of this file are Copyright (c) 2013-2017 IDRIX
|
|
and are governed by the Apache License 2.0 the full text of which is
|
|
contained in the file License.txt included in VeraCrypt binary and source
|
|
code distribution packages.
|
|
*/
|
|
|
|
#include "System.h"
|
|
#include "Main/GraphicUserInterface.h"
|
|
#include "NewSecurityTokenKeyfileDialog.h"
|
|
#include "SecurityTokenKeyfilesDialog.h"
|
|
|
|
namespace VeraCrypt
|
|
{
|
|
SecurityTokenKeyfilesDialog::SecurityTokenKeyfilesDialog (wxWindow* parent, bool selectionMode)
|
|
: SecurityTokenKeyfilesDialogBase (parent)
|
|
{
|
|
if (selectionMode)
|
|
SetTitle (LangString["SELECT_TOKEN_KEYFILES"]);
|
|
|
|
list <int> colPermilles;
|
|
|
|
SecurityTokenKeyfileListCtrl->InsertColumn (ColumnSecurityTokenSlotId, LangString["TOKEN_SLOT_ID"], wxLIST_FORMAT_CENTER, 1);
|
|
colPermilles.push_back (102);
|
|
SecurityTokenKeyfileListCtrl->InsertColumn (ColumnSecurityTokenLabel, LangString["TOKEN_NAME"], wxLIST_FORMAT_LEFT, 1);
|
|
colPermilles.push_back (368);
|
|
SecurityTokenKeyfileListCtrl->InsertColumn (ColumnSecurityTokenKeyfileLabel, LangString["TOKEN_DATA_OBJECT_LABEL"], wxLIST_FORMAT_LEFT, 1);
|
|
colPermilles.push_back (529);
|
|
|
|
FillSecurityTokenKeyfileListCtrl();
|
|
|
|
Gui->SetListCtrlWidth (SecurityTokenKeyfileListCtrl, 65);
|
|
Gui->SetListCtrlHeight (SecurityTokenKeyfileListCtrl, 16);
|
|
Gui->SetListCtrlColumnWidths (SecurityTokenKeyfileListCtrl, colPermilles);
|
|
|
|
Fit();
|
|
Layout();
|
|
Center();
|
|
|
|
DeleteButton->Disable();
|
|
ExportButton->Disable();
|
|
OKButton->Disable();
|
|
OKButton->SetDefault();
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::FillSecurityTokenKeyfileListCtrl ()
|
|
{
|
|
wxBusyCursor busy;
|
|
|
|
SecurityTokenKeyfileListCtrl->DeleteAllItems();
|
|
SecurityTokenKeyfileList = Token::GetAvailableKeyfiles(Gui->GetPreferences().EMVSupportEnabled);
|
|
|
|
size_t i = 0;
|
|
foreach (const shared_ptr<TokenKeyfile> key, SecurityTokenKeyfileList)
|
|
{
|
|
vector <wstring> fields (SecurityTokenKeyfileListCtrl->GetColumnCount());
|
|
|
|
fields[ColumnSecurityTokenSlotId] = StringConverter::ToWide ((uint64) key->Token->SlotId);
|
|
fields[ColumnSecurityTokenLabel] = key->Token->Label;
|
|
fields[ColumnSecurityTokenKeyfileLabel] = key->Id;
|
|
|
|
Gui->AppendToListCtrl (SecurityTokenKeyfileListCtrl, fields, 0, key.get());
|
|
}
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::OnDeleteButtonClick (wxCommandEvent& event)
|
|
{
|
|
try
|
|
{
|
|
if (!Gui->AskYesNo (LangString["CONFIRM_SEL_FILES_DELETE"]))
|
|
return;
|
|
|
|
wxBusyCursor busy;
|
|
|
|
foreach (long item, Gui->GetListCtrlSelectedItems (SecurityTokenKeyfileListCtrl))
|
|
{
|
|
SecurityToken::DeleteKeyfile (*reinterpret_cast <SecurityTokenKeyfile *> (SecurityTokenKeyfileListCtrl->GetItemData (item)));
|
|
}
|
|
|
|
FillSecurityTokenKeyfileListCtrl();
|
|
}
|
|
catch (exception &e)
|
|
{
|
|
Gui->ShowError (e);
|
|
}
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::OnExportButtonClick (wxCommandEvent& event)
|
|
{
|
|
try
|
|
{
|
|
foreach (long item, Gui->GetListCtrlSelectedItems (SecurityTokenKeyfileListCtrl))
|
|
{
|
|
TokenKeyfile *keyfile = reinterpret_cast <TokenKeyfile *> (SecurityTokenKeyfileListCtrl->GetItemData (item));
|
|
|
|
FilePathList files = Gui->SelectFiles (this, wxEmptyString, true);
|
|
|
|
if (!files.empty())
|
|
{
|
|
wxBusyCursor busy;
|
|
|
|
vector <byte> keyfileData;
|
|
keyfile->GetKeyfileData (keyfileData);
|
|
|
|
BufferPtr keyfileDataBuf (&keyfileData.front(), keyfileData.size());
|
|
finally_do_arg (BufferPtr, keyfileDataBuf, { finally_arg.Erase(); });
|
|
|
|
File keyfile;
|
|
keyfile.Open (*files.front(), File::CreateWrite);
|
|
keyfile.Write (keyfileDataBuf);
|
|
}
|
|
else
|
|
break;
|
|
|
|
Gui->ShowInfo ("KEYFILE_EXPORTED");
|
|
}
|
|
}
|
|
catch (exception &e)
|
|
{
|
|
Gui->ShowError (e);
|
|
}
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::OnImportButtonClick (wxCommandEvent& event)
|
|
{
|
|
try
|
|
{
|
|
FilePathList keyfilePaths = Gui->SelectFiles (this, LangString["SELECT_KEYFILES"], false);
|
|
|
|
if (keyfilePaths.empty())
|
|
return;
|
|
|
|
FilePath keyfilePath = *keyfilePaths.front();
|
|
|
|
File keyfile;
|
|
keyfile.Open (keyfilePath, File::OpenRead, File::ShareReadWrite, File::PreserveTimestamps);
|
|
|
|
if (keyfile.Length() > 0)
|
|
{
|
|
vector <byte> keyfileData (keyfile.Length());
|
|
BufferPtr keyfileDataBuf (&keyfileData.front(), keyfileData.size());
|
|
|
|
keyfile.ReadCompleteBuffer (keyfileDataBuf);
|
|
finally_do_arg (BufferPtr, keyfileDataBuf, { finally_arg.Erase(); });
|
|
|
|
NewSecurityTokenKeyfileDialog newKeyfileDialog (this, keyfilePath.ToBaseName());
|
|
|
|
if (newKeyfileDialog.ShowModal() == wxID_OK)
|
|
{
|
|
wxBusyCursor busy;
|
|
SecurityToken::CreateKeyfile (newKeyfileDialog.GetSelectedSlotId(), keyfileData, StringConverter::ToSingle (newKeyfileDialog.GetKeyfileName()));
|
|
|
|
FillSecurityTokenKeyfileListCtrl();
|
|
}
|
|
}
|
|
else
|
|
throw InsufficientData (SRC_POS, keyfilePath);
|
|
}
|
|
catch (exception &e)
|
|
{
|
|
Gui->ShowError (e);
|
|
}
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::OnListItemDeselected (wxListEvent& event)
|
|
{
|
|
if (SecurityTokenKeyfileListCtrl->GetSelectedItemCount() == 0)
|
|
{
|
|
DeleteButton->Disable();
|
|
ExportButton->Disable();
|
|
OKButton->Disable();
|
|
}
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::OnListItemSelected(wxListEvent &event)
|
|
{
|
|
if (event.GetItem().GetData() != (wxUIntPtr) nullptr)
|
|
{
|
|
BOOL deletable = true;
|
|
foreach(long
|
|
item, Gui->GetListCtrlSelectedItems(SecurityTokenKeyfileListCtrl))
|
|
{
|
|
TokenKeyfile *keyfile = reinterpret_cast <TokenKeyfile *> (SecurityTokenKeyfileListCtrl->GetItemData(item));
|
|
if (!keyfile->Token->isEditable())
|
|
{
|
|
deletable = false;
|
|
break;
|
|
}
|
|
}
|
|
if (deletable)
|
|
{
|
|
DeleteButton->Enable();
|
|
}
|
|
ExportButton->Enable();
|
|
OKButton->Enable();
|
|
}
|
|
}
|
|
|
|
void SecurityTokenKeyfilesDialog::OnOKButtonClick ()
|
|
{
|
|
foreach (long item, Gui->GetListCtrlSelectedItems (SecurityTokenKeyfileListCtrl))
|
|
{
|
|
TokenKeyfile *key = reinterpret_cast <TokenKeyfile *> (SecurityTokenKeyfileListCtrl->GetItemData(item));
|
|
|
|
SelectedSecurityTokenKeyfilePaths.push_back(*key);
|
|
}
|
|
EndModal (wxID_OK);
|
|
}
|
|
}
|