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;
}
int lt_start_simple_server(lt_shmHead *lt_MMHead,char *confFile,int (*on_app_start)(),int (*on_proc_start)(),int (*on_proc_stop)() ){
int iReturn;
/*初始化主进程列表*/
int ltServerSock;
if(on_app_start!=NULL){
iReturn=on_app_start(lt_MMHead); /*一般用来注册函数*/
if(iReturn<0){
fprintf(stderr,"error: can't start on_app_start\n");
exit(-1);
}
}
signal(SIGCHLD, sig_chld);
signal(SIGINT, sig_int);
signal(SIGTERM, sig_int);
signal(SIGPIPE, SIG_IGN);
iReturn=ltLocalServer(confFile);
if(iReturn<0){
fprintf(stderr,"start local server error!\n");
exit(0);
}
fprintf(stderr,"start local server ok!\n");
//printf("start tcp server ok \n");
/*系统主侦听端口*/
ltServerSock=iReturn;
for(;;){
struct sockaddr_in sa;
int dummy = sizeof(sa);
int fd;
fd = accept(ltServerSock,(struct sockaddr *) &sa,&dummy);
if (fd == -1) continue;
switch(fork()) {
case 0:
close(ltServerSock);
signal(SIGINT, ltCatch_fatal);
signal(SIGQUIT, ltCatch_fatal);
signal(SIGILL, ltCatch_fatal);
signal(SIGTRAP, ltCatch_fatal);
signal(SIGIOT, ltCatch_fatal);
signal(SIGFPE, ltCatch_fatal);
signal(SIGTERM, ltCatch_fatal);
signal(SIGALRM, ltCatch_fatal);
alarm(600);
if( on_proc_start!=NULL){
iReturn = on_proc_start(lt_MMHead);
if(iReturn < 0) {
fprintf(stderr," Process %d Init Error \n",getpid());
_exit(0);
}
}
ltTcpDoSRequest(lt_MMHead,fd);
if( on_proc_stop!=NULL){
on_proc_stop(lt_MMHead);
}
_exit(0);
case -1:
fprintf(stderr,"main proc start error: can't start on_app_start\n");
}
close(fd);
}
return 0;
}