int _init_wizard_encrypt_pages(
HWND parent,
_wz_sheets *sheets,
_dnode *node
)
{
wchar_t *static_head[ ] =
{
L"# Choice iso-file",
L"# Format Options",
L"# Encryption Settings",
L"# Boot Settings",
L"# Volume Password",
L"# Encryption Progress"
};
HWND hwnd;
dc_conf dcfg;
int k;
int count = 0;
BOOL boot_device = (
_is_boot_device( &node->mnt.info )
);
BOOL force_small = (
boot_device && ( dc_get_conf_flags(&dcfg) == ST_OK ) && ( dcfg.load_flags & DST_SMALL_MEM )
);
while ( sheets[count].id != -1 )
{
HWND hwnd;
sheets[count].hwnd =
CreateDialog(
__hinst, MAKEINTRESOURCE(sheets[count].id), GetDlgItem(parent, IDC_TAB), _tab_proc
);
hwnd = sheets[count].hwnd;
EnumChildWindows( hwnd, __sub_enum, (LPARAM)NULL );
SetWindowText( GetDlgItem( hwnd, IDC_HEAD ), static_head[count] );
SendMessage( GetDlgItem( hwnd, IDC_HEAD ), (UINT)WM_SETFONT, (WPARAM)__font_bold, 0 );
sheets[count].first_tab_hwnd =
(
( sheets[count].first_tab_id != -1 ) ? GetDlgItem( hwnd, sheets[count].first_tab_id ) : HWND_NULL
);
count++;
}
///////////////////////////////////////////////////////////////
hwnd = sheets[WPAGE_ENC_FRMT].hwnd;
///////////////////////////////////////////////////////////////
/////// FORMAT OPTIONS PAGE ///////////////////////////////////
{
HWND h_fs = GetDlgItem(hwnd, IDC_COMBO_FS_LIST);
_sub_class( GetDlgItem(hwnd, IDC_CHECK_QUICK_FORMAT), SUB_STATIC_PROC, HWND_NULL );
_set_check( hwnd, IDC_CHECK_QUICK_FORMAT, FALSE );
for ( k = 0; k < array_num(fs_names); k++ )
{
SendMessage( h_fs, (UINT)CB_ADDSTRING, 0, (LPARAM)fs_names[k] );
}
SendMessage( h_fs, CB_SETCURSEL, 2, 0 );
}
///////////////////////////////////////////////////////////////
hwnd = sheets[WPAGE_ENC_CONF].hwnd;
///////////////////////////////////////////////////////////////
/////// ENCRYPTION SETTINGS PAGE //////////////////////////////
{
HWND h_combo_wipe = GetDlgItem(hwnd, IDC_COMBO_PASSES);
_init_combo( h_combo_wipe, wipe_modes, WP_NONE, FALSE, -1 );
EnableWindow( h_combo_wipe, node->dlg.act_type != ACT_ENCRYPT_CD);
EnableWindow( GetDlgItem(hwnd, IDC_STATIC_PASSES_LIST), node->dlg.act_type != ACT_ENCRYPT_CD );
_init_combo(
GetDlgItem(hwnd, IDC_COMBO_ALGORT), cipher_names, CF_AES, FALSE, -1
);
if ( force_small )
{
EnableWindow( GetDlgItem(hwnd, IDC_COMBO_ALGORT), FALSE );
SendMessage( GetDlgItem(hwnd, IDC_WIZ_CONF_WARNING), (UINT)WM_SETFONT, (WPARAM)__font_bold, 0 );
SetWindowText(
GetDlgItem(hwnd, IDC_WIZ_CONF_WARNING),
L"Your BIOS does not provide enough base memory,\n"
L"you can only use AES to encrypt the boot partition!"
);
}
for ( k = 0; k < array_num(combo_sel); k++ )
{
SendMessage( GetDlgItem(hwnd, combo_sel[k]), CB_SETCURSEL, 0, 0 );
}
}
///////////////////////////////////////////////////////////////
hwnd = sheets[WPAGE_ENC_BOOT].hwnd;
///////////////////////////////////////////////////////////////
/////// BOOT SETTINGS PAGE ////////////////////////////////////
{
int dsk_num = -1;
int rlt = _get_info_install_boot_page( &node->mnt.info, sheets, &dsk_num );
__lists[HENC_WIZARD_BOOT_DEVS] = GetDlgItem(hwnd, IDC_BOOT_DEVS);
_list_devices( __lists[HENC_WIZARD_BOOT_DEVS], TRUE, dsk_num );
SendMessage( GetDlgItem(hwnd, IDC_COMBO_BOOT_INST), (UINT)CB_ADDSTRING, 0, (LPARAM)L"Use external bootloader" );
if ( rlt != ST_OK )
{
SetWindowText( GetDlgItem(hwnd, IDC_WARNING), L"Bootable HDD not found!" );
SendMessage( GetDlgItem(hwnd, IDC_COMBO_BOOT_INST), CB_SETCURSEL, 0, 0 );
SendMessage( GetDlgItem(hwnd, IDC_WARNING), (UINT)WM_SETFONT, (WPARAM)__font_bold, 0 );
EnableWindow( GetDlgItem(hwnd, IDB_BOOT_PREF), TRUE );
} else {
SendMessage( GetDlgItem(hwnd, IDC_COMBO_BOOT_INST), (UINT)CB_ADDSTRING, 0, (LPARAM)L"Install to HDD" );
SendMessage( GetDlgItem(hwnd, IDC_COMBO_BOOT_INST), CB_SETCURSEL, 1, 0 );
}
}
///////////////////////////////////////////////////////////////
hwnd = sheets[WPAGE_ENC_PASS].hwnd;
///////////////////////////////////////////////////////////////
/////// VOLUME PASSWORD PAGE //////////////////////////////////
{
int kbd_layout;
_update_layout( node, -1, &kbd_layout );
_init_combo( GetDlgItem(hwnd, IDC_COMBO_KBLAYOUT), kb_layouts, kbd_layout, FALSE, -1 );
SetWindowText(GetDlgItem( hwnd, IDC_USE_KEYFILES), boot_device ? IDS_USE_KEYFILE : IDS_USE_KEYFILES );
_sub_class( GetDlgItem(hwnd, IDC_CHECK_SHOW), SUB_STATIC_PROC, HWND_NULL );
_set_check( hwnd, IDC_CHECK_SHOW, FALSE );
_sub_class( GetDlgItem(hwnd, IDC_USE_KEYFILES), SUB_STATIC_PROC, HWND_NULL );
_set_check( hwnd, IDC_USE_KEYFILES, FALSE );
SendMessage(
GetDlgItem( hwnd, IDP_BREAKABLE ),
PBM_SETBARCOLOR, 0, _cl( COLOR_BTNSHADOW, DARK_CLR-20 )
);
SendMessage(
GetDlgItem(hwnd, IDP_BREAKABLE),
PBM_SETRANGE, 0, MAKELPARAM(0, 193)
);
SetWindowText( GetDlgItem(hwnd, IDC_HEAD2), L"# Password Rating" );
SendMessage( GetDlgItem(hwnd, IDC_HEAD2), (UINT)WM_SETFONT, (WPARAM)__font_bold, 0 );
SendMessage( GetDlgItem(hwnd, IDE_PASS), EM_LIMITTEXT, MAX_PASSWORD, 0 );
SendMessage( GetDlgItem(hwnd, IDE_CONFIRM), EM_LIMITTEXT, MAX_PASSWORD, 0 );
}
///////////////////////////////////////////////////////////////
hwnd = sheets[WPAGE_ENC_PROGRESS].hwnd;
///////////////////////////////////////////////////////////////
/////// ENCRYPTION PROGRESS PAGE //////////////////////////////
{
_colinfo _progress_iso_crypt_headers[ ] =
{
{ STR_HEAD_NO_ICONS, 100, LVCFMT_LEFT, FALSE },
{ STR_HEAD_NO_ICONS, 120, LVCFMT_LEFT, FALSE },
{ STR_NULL }
};
HWND h_list = GetDlgItem( hwnd, IDC_ISO_PROGRESS );
int rlt = ST_OK;
int j = 0;
ListView_SetBkColor( h_list, GetSysColor(COLOR_BTNFACE) );
_init_list_headers( h_list, _progress_iso_crypt_headers );
while ( wcslen(_act_table_items[j]) > 0 )
{
_list_insert_item( h_list, j, 0, _act_table_items[j], 0 );
if ( j != 2 ) ListView_SetItemText( h_list, j, 1, STR_EMPTY );
j++;
}
SendMessage(
GetDlgItem( hwnd, IDC_PROGRESS_ISO ),
PBM_SETBARCOLOR, 0, _cl(COLOR_BTNSHADOW, DARK_CLR-20)
);
SendMessage(
GetDlgItem( hwnd, IDC_PROGRESS_ISO ),
PBM_SETRANGE, 0, MAKELPARAM(0, PRG_STEP)
);
}
return count;
}
INT_PTR CALLBACK
_benchmark_dlg_proc(
HWND hwnd,
UINT message,
WPARAM wparam,
LPARAM lparam
)
{
switch ( message )
{
case WM_CLOSE :
{
__lists[HBENCHMARK] = HWND_NULL;
EndDialog( hwnd, 0 );
return 0L;
}
break;
case WM_COMMAND :
{
int code = HIWORD(wparam);
int id = LOWORD(wparam);
if ( ( id == IDOK ) || ( id == IDCANCEL ) )
{
EndDialog( hwnd, 0 );
}
if ( id == IDB_REFRESH_TEST )
{
HWND h_button = GetDlgItem( hwnd, IDB_REFRESH_TEST );
SetCursor( __cur_wait );
EnableWindow( h_button, FALSE );
{
bench_item bench[CF_CIPHERS_NUM];
wchar_t s_speed[50];
int cnt;
int lvcount = 0;
int k = 0;
cnt = _benchmark(pv(&bench));
ListView_DeleteAllItems( __lists[HBENCHMARK] );
for ( k = 0; k < cnt; k++ )
{
_list_insert_item( __lists[HBENCHMARK], lvcount, 0, bench[k].alg, 0 );
_list_set_item( __lists[HBENCHMARK], lvcount, 1, STR_EMPTY );
_snwprintf( s_speed, sizeof_w(s_speed), L"%-.2f mb/s", bench[k].speed );
_list_set_item( __lists[HBENCHMARK], lvcount++, 2, s_speed );
}
}
EnableWindow( h_button, TRUE );
SetCursor( __cur_arrow );
}
}
break;
case WM_INITDIALOG :
{
__lists[HBENCHMARK] = GetDlgItem( hwnd, IDC_LIST_BENCHMARK );
_init_list_headers( __lists[HBENCHMARK], _benchmark_headers );
ListView_SetBkColor( __lists[HBENCHMARK], GetSysColor(COLOR_BTNFACE) );
ListView_SetTextBkColor( __lists[HBENCHMARK], GetSysColor(COLOR_BTNFACE) );
ListView_SetExtendedListViewStyle( __lists[HBENCHMARK], LVS_EX_FLATSB | LVS_EX_FULLROWSELECT );
SetForegroundWindow(hwnd);
_sub_class(GetDlgItem(hwnd, IDC_BUTTON), SUB_STATIC_PROC, HWND_NULL);
return 1L;
}
break;
case WM_CTLCOLOREDIT :
{
return (
_ctl_color(wparam, _cl(COLOR_BTNFACE, LGHT_CLR))
);
}
break;
default:
{
int rlt = _draw_proc(message, lparam);
if (rlt != -1)
{
return rlt;
}
}
}
return 0L;
}
static
INT_PTR CALLBACK
_keyfiles_dlg_proc(
HWND hwnd,
UINT message,
WPARAM wparam,
LPARAM lparam
)
{
static int key_list;
static list_entry *head;
switch ( message )
{
case WM_CLOSE :
{
EndDialog( hwnd, 0 );
return 0L;
}
break;
case WM_NOTIFY :
{
if( wparam == IDC_LIST_KEYFILES )
{
if ( ((NMHDR *)lparam)->code == LVN_ITEMCHANGED &&
(((NMLISTVIEW *)lparam)->uNewState & LVIS_FOCUSED ) )
{
HWND h_list = GetDlgItem( hwnd, IDC_LIST_KEYFILES );
EnableWindow(GetDlgItem( hwnd, IDB_REMOVE_ITEM), ListView_GetSelectedCount( h_list ) );
return 1L;
}
if ( ((NM_LISTVIEW *)lparam)->hdr.code == NM_CLICK )
{
HWND h_list = GetDlgItem( hwnd, IDC_LIST_KEYFILES );
EnableWindow( GetDlgItem( hwnd, IDB_REMOVE_ITEM), ListView_GetSelectedCount( h_list ) );
}
}
}
case WM_COMMAND :
{
HWND h_list = GetDlgItem( hwnd, IDC_LIST_KEYFILES );
int code = HIWORD(wparam);
int id = LOWORD(wparam);
switch ( id )
{
case IDB_GENERATE_KEYFILE :
{
wchar_t s_file[MAX_PATH] = { L"keyfile" };
byte keyfile[64];
int rlt;
if ( _save_file_dialog(
hwnd, s_file, countof(s_file), L"Save 64 bytes random keyfile as.."
) )
{
if ( (rlt = dc_get_random(keyfile, sizeof(keyfile))) == ST_OK )
{
rlt = save_file(s_file, keyfile, sizeof(keyfile));
burn(keyfile, sizeof(keyfile));
}
if ( rlt == ST_OK )
{
if ( __msg_q(hwnd,
L"Keyfile \"%s\" successfully created\n\n"
L"Add this file to the keyfiles list?",
s_file
) )
{
_ui_keys_list_refresh(hwnd);
if ( key_list == KEYLIST_EMBEDDED )
{
ListView_DeleteAllItems( h_list );
}
_add_item( h_list, s_file );
_ui_embedded( hwnd, key_list );
}
} else {
__error_s( hwnd, L"Error creating Keyfile", rlt );
}
}
}
break;
case IDB_REMOVE_ITEM :
{
ListView_DeleteItem( h_list, ListView_GetSelectionMark(h_list) );
_ui_keys_list_refresh( hwnd );
_ui_embedded( hwnd, key_list );
}
break;
case IDB_REMOVE_ITEMS :
{
ListView_DeleteAllItems( h_list );
_ui_keys_list_refresh( hwnd );
_ui_embedded( hwnd, key_list );
}
break;
case IDB_ADD_FOLDER :
{
wchar_t path[MAX_PATH];
if ( _folder_choice(hwnd, path, L"Choice folder..") )
{
_ui_keys_list_refresh( hwnd );
_set_trailing_slash( path );
_add_item( h_list, path );
}
}
break;
case IDB_ADD_FILE :
{
wchar_t s_path[MAX_PATH] = { 0 };
if ( _open_file_dialog(hwnd, s_path, countof(s_path), L"Select File..") )
{
if ( key_list == KEYLIST_EMBEDDED )
{
HWND h_file =
CreateFile(
s_path, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL
);
if ( h_file != INVALID_HANDLE_VALUE )
{
if ( GetFileSize( h_file, NULL ) != 64 )
{
__error_s( hwnd, L"Embedded keyfile must be 64 byte size", ST_ERROR );
} else
{
_ui_keys_list_refresh( hwnd );
_add_item( h_list, s_path );
}
CloseHandle( h_file );
}
_ui_embedded( hwnd, key_list );
} else
{
_ui_keys_list_refresh( hwnd );
_add_item( h_list, s_path );
}
}
}
break;
}
if ( id == IDCANCEL )
{
EndDialog(hwnd, 0);
}
if ( id == IDOK )
{
int k = 0;
_keyfiles_wipe(key_list);
for ( ; k < ListView_GetItemCount( h_list ); k ++ )
{
wchar_t item[MAX_PATH];
_get_item_text( h_list, k, 0, item, countof(item) );
if ( wcscmp(item, IDS_EMPTY_LIST) != 0 )
{
_list_key_files *new_node;
if ( (new_node = secure_alloc(sizeof(_list_key_files))) == NULL )
{
__error_s( hwnd, L"Can't allocate memory", ST_NOMEM );
_keyfiles_wipe(key_list);
break;
}
wcsncpy(new_node->path, item, countof(new_node->path));
_insert_tail_list(head, &new_node->next);
}
}
EndDialog(hwnd, 0);
}
}
break;
case WM_INITDIALOG :
{
HWND h_list = GetDlgItem(hwnd, IDC_LIST_KEYFILES);
_list_key_files *key_file;
key_list = (int)lparam;
head = _KEYFILES_HEAD_( key_list );
_init_list_headers( h_list, _keyfiles_headers );
if ( key_file = _first_keyfile( key_list ) )
{
EnableWindow( GetDlgItem(hwnd, IDB_REMOVE_ITEMS), TRUE );
do
{
_list_insert_item( h_list, ListView_GetItemCount(h_list), 0, key_file->path, 0 );
key_file = _next_keyfile( key_file, key_list );
} while ( key_file != NULL );
}
_ui_keys_list_refresh( hwnd );
_ui_embedded( hwnd, key_list );
ListView_SetBkColor( h_list, GetSysColor(COLOR_BTNFACE) );
ListView_SetTextBkColor( h_list, GetSysColor(COLOR_BTNFACE) );
ListView_SetExtendedListViewStyle( h_list, LVS_EX_FLATSB | LVS_EX_FULLROWSELECT );
SetForegroundWindow(hwnd);
return 1L;
}
break;
case WM_CTLCOLOREDIT :
{
return _ctl_color(wparam, _cl(COLOR_BTNFACE, LGHT_CLR));
}
break;
default:
{
int rlt = _draw_proc(message, lparam);
if (rlt != -1) return rlt;
}
}
return 0L;
}
