DWORD
WINAPI
_thread_enc_iso_proc(
LPVOID lparam
)
{
_dnode *node;
dc_open_device( );
if ( (node = pv(lparam)) != NULL )
{
node->dlg.rlt = ST_OK;
node->dlg.rlt = dc_encrypt_cd(
node->dlg.iso.s_iso_src, node->dlg.iso.s_iso_dst, node->dlg.iso.pass, node->dlg.iso.cipher_id, dc_cd_callback, lparam
);
{
secure_free( node->dlg.iso.pass );
SendMessage( GetParent(GetParent(node->dlg.h_page)), WM_CLOSE_DIALOG, 0, 0 );
}
}
//EnterCriticalSection(&crit_sect);
//LeaveCriticalSection(&crit_sect);
dc_close_device( );
return 1L;
}
int WINAPI wWinMain(
HINSTANCE hinst,
HINSTANCE hprev,
LPWSTR cmd_line,
int cmd_show
)
{
int rlt;
int ver;
int app_start = on_app_start( cmd_line );
#ifdef LOG_FILE
_log( L"%0.8X app start", app_start );
#endif
if ( app_start == ST_NEED_EXIT )
{
return 0;
}
if ( _ui_init(hinst) == 0 )
{
__error_s( HWND_DESKTOP, L"Error GUI initialization", ST_OK );
return 0;
}
if ( is_admin( ) != ST_OK )
{
__error_s( HWND_DESKTOP, L"Admin Privileges Required", ST_OK );
return 0;
}
#ifdef _M_IX86
if ( is_wow64( ) != 0 )
{
__error_s( HWND_DESKTOP, L"Please use x64 version of DiskCryptor", ST_OK );
return 0;
}
#endif
if ( dc_is_old_runned( ) != 0 )
{
__error_s(
HWND_DESKTOP,
L"DiskCryptor 0.1-0.4 installed, please completely uninstall it before use this version.", ST_OK
);
return 0;
}
#ifdef LOG_FILE
_log( L"%0.8X driver status", dc_driver_status( ) );
#endif
if ( dc_is_driver_works( ) == FALSE )
{
if ( ( rlt = _drv_action(DA_INSTAL, 0) ) != NO_ERROR )
{
__error_s( HWND_DESKTOP, NULL, rlt );
}
return 0;
}
if ( ( rlt = dc_open_device( ) ) != ST_OK )
{
__error_s( HWND_DESKTOP, L"Can not open DC device", rlt );
return 0;
}
ver = dc_get_version( );
#ifdef LOG_FILE
_log( L"%0.8X dc version", ver );
#endif
if ( ver < DC_DRIVER_VER )
{
if ( ( rlt = _drv_action(DA_UPDATE, ver) ) != NO_ERROR )
{
__error_s( HWND_DESKTOP, NULL, rlt );
}
return 0;
}
if ( ver > DC_DRIVER_VER )
{
__msg_i(
HWND_DESKTOP,
L"DiskCryptor driver v%d detected\n"
L"Please use last program version", ver
);
return 0;
}
{
HWND h_find;
WNDCLASS wc = { 0 };
wc.lpszClassName = DC_CLASS;
wc.lpfnWndProc = &_class_dlg_proc;
wc.cbWndExtra = DLGWINDOWEXTRA;
wc.hIcon = LoadIcon(hinst, MAKEINTRESOURCE(IDI_ICON_TRAY));
dlg_class = RegisterClass(&wc);
#ifdef LOG_FILE
_log( L"%0.8X register class", dlg_class );
#endif
h_find = FindWindow(DC_CLASS, NULL);
#ifdef LOG_FILE
_log( L"%0.8X find window", h_find );
#endif
if ( h_find != NULL )
{
ShowWindow( h_find, SW_SHOW );
SetForegroundWindow( h_find );
#ifdef LOG_FILE
_log( L"show window [ %0.8X ] return", h_find );
#endif
return 0;
}
}
if ( ( rlt = rnd_init( ) ) != ST_OK )
{
__error_s( HWND_DESKTOP, L"Can not initialize RNG", rlt );
return 0;
}
if ( (rlt = dc_load_config(&__config) == NO_ERROR ? ST_OK : ST_ERROR) != ST_OK )
{
__error_s( HWND_DESKTOP, L"Error get config", rlt );
return 0;
}
InitializeCriticalSection( &crit_sect );
#ifdef LOG_FILE
_log( L"initialize critical section" );
#endif
_init_list_head( &__drives );
_init_list_head( &__action );
_init_keyfiles_list( );
#ifdef LOG_FILE
_log( L"init keyfiles list" );
#endif
{
HWND h_dialog;
MSG msg;
__hacc = LoadAccelerators( hinst, MAKEINTRESOURCE(IDR_MAIN_ACCEL) );
#ifdef LOG_FILE
_log( L"before create dialog" );
#endif
h_dialog = CreateDialog( GetModuleHandleA(NULL), MAKEINTRESOURCE(IDD_MAIN_DLG), HWND_DESKTOP, _main_dialog_proc );
#ifdef LOG_FILE
_log( L"%0.8X create dialog", h_dialog );
#endif
rlt = ShowWindow( h_dialog, app_start == ST_AUTORUNNED ? SW_HIDE : SW_SHOW );
#ifdef LOG_FILE
_log( L"%0.8X show window", rlt );
#endif
while ( GetMessage( &msg, NULL, 0, 0 ) )
{
if ( !TranslateAccelerator( h_dialog, __hacc, &msg ) )
{
TranslateMessage( &msg );
DispatchMessage( &msg );
}
}
DestroyAcceleratorTable( __hacc );
}
return TRUE;
}
DWORD
WINAPI
_thread_format_proc(
LPVOID lparam
)
{
int i = 0;
int rlt, wp_mode;
wchar_t device[MAX_PATH];
_dnode *node;
_dact *act;
dc_open_device( );
EnterCriticalSection(&crit_sect);
node = pv(lparam);
act = _create_act_thread(node, -1, -1);
if ( !node || !act )
{
return 0L;
}
wcscpy(device, act->device);
do
{
if ( act->status != ACT_RUNNING )
{
break;
}
if ( i-- == 0 )
{
dc_sync_enc_state(device);
i = 20;
}
wp_mode = act->wp_mode;
LeaveCriticalSection( &crit_sect );
rlt = dc_format_step( device, wp_mode );
EnterCriticalSection( &crit_sect );
if ( rlt == ST_FINISHED )
{
act->status = ACT_STOPPED;
break;
}
if ( ( rlt != ST_OK ) && ( rlt != ST_RW_ERR ) )
{
dc_status st;
dc_get_device_status( device, &st );
__error_s(
HWND_DESKTOP,
L"Format error on volume [%s]", rlt, st.mnt_point
);
act->status = ACT_STOPPED;
break;
}
} while (1);
if ( rlt == ST_FINISHED )
{
_finish_formating( node );
}
LeaveCriticalSection( &crit_sect );
return 1L;
}
static
DWORD WINAPI
_thread_enc_dec_proc(
LPVOID lparam
)
{
BOOL encrypting;
int i = 0;
int rlt, wp_mode;
wchar_t device[MAX_PATH];
_dnode *node;
_dact *act;
dc_open_device( );
EnterCriticalSection( &crit_sect );
node = pv(lparam);
act = _create_act_thread(node, -1, -1);
if ( !node || !act )
{
return 0L;
}
wcscpy( device, act->device );
do
{
if ( act->status != ACT_RUNNING )
{
break;
}
if ( i-- == 0 )
{
dc_sync_enc_state( device );
i = 20;
}
encrypting = act->act != ACT_DECRYPT;
wp_mode = act->wp_mode;
LeaveCriticalSection( &crit_sect );
rlt = encrypting ?
dc_enc_step(device, wp_mode) :
dc_dec_step(device);
EnterCriticalSection(&crit_sect);
if ( rlt == ST_FINISHED )
{
act->status = ACT_STOPPED;
break;
}
if ( rlt == ST_CANCEL )
{
Sleep(5000);
}
if ( ( rlt != ST_OK ) && ( rlt != ST_RW_ERR ) && ( rlt != ST_CANCEL ) )
{
dc_status st;
wchar_t *act_name;
dc_get_device_status( device, &st );
switch ( act->act )
{
case ACT_ENCRYPT: act_name = L"Encryption"; break;
case ACT_DECRYPT: act_name = L"Decryption"; break;
case ACT_REENCRYPT: act_name = L"Reencryption"; break;
}
__error_s(
HWND_DESKTOP,
L"%s error on volume [%s]", rlt, act_name, st.mnt_point
);
act->status = ACT_STOPPED;
break;
}
} while (1);
dc_sync_enc_state(device);
LeaveCriticalSection(&crit_sect);
return 1L;
}
int WINAPI wWinMain(
HINSTANCE hinst,
HINSTANCE hprev,
LPWSTR cmd_line,
int cmd_show
)
{
int d_st;
int resl;
if (dc_is_old_runned() != 0) {
return ST_INCOMPATIBLE;
}
/* open DC driver device */
dc_open_device();
/* get DC sriver status */
d_st = dc_driver_status();
resl = ST_ERROR;
do
{
if (wcscmp(cmd_line, L"-setup") == 0)
{
if (d_st == ST_ERROR) {
resl = dc_install_driver(NULL);
} else
{
resl = dc_update_boot(-1);
if ( (resl != ST_OK) && (resl != ST_BLDR_NOTINST) ) {
break;
}
resl = dc_update_driver();
}
break;
}
if (wcscmp(cmd_line, L"-unins") == 0)
{
if (d_st == ST_ERROR) {
resl = ST_ERROR; break;
}
resl = dc_remove_driver();
break;
}
if (wcscmp(cmd_line, L"-unldr") == 0) {
resl = dc_unset_mbr(-1);
break;
}
if (wcscmp(cmd_line, L"-isboot") == 0) {
ldr_config conf;
resl = dc_get_mbr_config(-1, NULL, &conf);
break;
}
if (wcscmp(cmd_line, L"-isenc") == 0)
{
vol_inf info;
u32 flags;
wchar_t boot_dev[MAX_PATH];
int is_enc = 0;
if (dc_open_device() != ST_OK) {
resl = ST_ERROR; break;
}
if (dc_get_boot_device(boot_dev) != ST_OK) {
boot_dev[0] = 0;
}
if (dc_first_volume(&info) == ST_OK)
{
do
{
flags = info.status.flags;
if ( ((flags & F_SYSTEM) ||
(wcscmp(info.device, boot_dev) == 0)) && (flags & F_ENABLED) )
{
is_enc = 1;
}
} while (dc_next_volume(&info) == ST_OK);
}
dc_close_device();
resl = is_enc != 0 ? ST_ENCRYPTED : ST_OK; break;
}
} while (0);
return resl;
}
