configwnd.c.orig

一文件过滤与加密,系统监视以及控制的东东,自己看

/******************************************************************/
/*                                                                */
/*  Winpooch : Windows Watchdog                                   */
/*  Copyright (C) 2004-2006  Benoit Blanchon                      */
/*                                                                */
/*  This program is free software; you can redistribute it        */
/*  and/or modify it under the terms of the GNU General Public    */
/*  License as published by the Free Software Foundation; either  */
/*  version 2 of the License, or (at your option) any later       */
/*  version.                                                      */
/*                                                                */
/*  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.  See the GNU General Public License for more         */
/*  details.                                                      */
/*                                                                */
/*  You should have received a copy of the GNU General Public     */
/*  License along with this program; if not, write to the Free    */
/*  Software Foundation, Inc.,                                    */
/*  675 Mass Ave, Cambridge, MA 02139, USA.                       */
/*                                                                */
/******************************************************************/


/******************************************************************/
/* Build configuration                                            */
/******************************************************************/

#define	TRACE_LEVEL		2


/******************************************************************/
/* Includes                                                       */
/******************************************************************/

// module's interface
#include "ConfigWnd.h"

// standard headers
#include <windows.h>
#include <windowsx.h>
#include <commctrl.h>
#include <tchar.h>
#include <stdio.h>
#include <shlwapi.h>

// project's headers
#include "Assert.h"
#include "Config.h"
#include "FilterDefault.h"
#include "FilterFile.h"
#include "Language.h"
#include "LogFile.h"
#include "ProjectInfo.h"
#include "Resources.h"
#include "Scanner.h"
#include "Sounds.h"
#include "SpyServer.h"
#include "Strlcpy.h"
#include "Trace.h"
#include "TrayIcon.h"


/******************************************************************/
/* Internal macros                                                */
/******************************************************************/

#define arraysize(a) (sizeof(a)/sizeof((a)[0]))

#define clear(a) memset(&(a),0,sizeof(a))

#define EnableDlgItem(hDlg,nItem,bEnable) \
  EnableWindow(GetDlgItem(hDlg,nItem),bEnable)

/******************************************************************/
/* Internal data types                                            */
/******************************************************************/

typedef struct {
  UINT		idCtrl ;
  UINT		idCfgInt ;
  UINT		idStr ;
} CONFIG_CHECKBOX ;


/******************************************************************/
/* Internal constants                                             */
/******************************************************************/

#define MAX_FILTER	128

#define WM_UPDATECONFIG	(WM_USER+10)

static CONFIG_CHECKBOX aCheckBoxes[] = {
  { IDC_SPLASH_SCREEN,		CFGINT_SPLASH_SCREEN },
  { IDC_CHECK_UPDATES,		CFGINT_CHECK_FOR_UPDATES },
  { IDC_TRAY_ICON_ANIMATION,	CFGINT_TRAY_ICON_ANIMATION } 
} ;

LPCTSTR g_szConfirmEraseFilters = TEXT("You will lost all existing filters.\nAre-you sure ?") ;


/******************************************************************/
/* Internal functions                                             */
/******************************************************************/

BOOL CALLBACK _ConfigWnd_DlgProc (HWND, UINT, WPARAM, LPARAM) ;

BOOL _ConfigWnd_Browse (HWND hDlg, BOOL bExport, LPTSTR szPath, UINT nMax) ;

BOOL _ConfigWnd_BrowseSound (HWND hDlg, LPTSTR szPath, UINT nMax) ;


/******************************************************************/
/* Exported function : RegisterClass                              */
/******************************************************************/

BOOL ConfigWnd_RegisterClass (HINSTANCE hInstance) 
{

  return TRUE ;
}


/******************************************************************/
/* Exported function : CreateWindow                               */
/******************************************************************/

HWND ConfigWnd_CreateWindow (HINSTANCE hInstance, HWND hwndParent)
{  
  return CreateDialog (hInstance, MAKEINTRESOURCE(DLG_CONFIG),
		       hwndParent, _ConfigWnd_DlgProc) ;
}


/******************************************************************/
/* Internal function : WndProc                                    */
/******************************************************************/

BOOL CALLBACK _ConfigWnd_DlgProc (HWND hDlg, UINT message, WPARAM wParam, LPARAM lParam)
{
  int		i ;
  TCHAR		szBuffer[1024] ;

  switch( message )
    {
    case WM_INITDIALOG:

      // set language list
      for( i=0 ; i<Language_GetLanguageCount() ; i++ )
	SendDlgItemMessage (hDlg, IDC_LANGUAGE, CB_ADDSTRING, 0,
			    (LPARAM)Language_GetLanguage(i)) ;

    case WM_LANGUAGECHANGED:

      // add antivirus
      SendDlgItemMessage (hDlg, IDC_ANTIVIRUS, CB_RESETCONTENT, 0, 0) ; 
      SendDlgItemMessage (hDlg, IDC_ANTIVIRUS, CB_ADDSTRING, 0,
			  (LPARAM) STR_DEF(_ANTIVIRUS_NONE,TEXT("None"))) ;
      SendDlgItemMessage (hDlg, IDC_ANTIVIRUS, CB_ADDSTRING, 0,
			  (LPARAM) STR_DEF(_ANTIVIRUS_CLAMWIN,TEXT("ClamWin"))) ;
      SendDlgItemMessage (hDlg, IDC_ANTIVIRUS, CB_ADDSTRING, 0,
			  (LPARAM) STR_DEF(_ANTIVIRUS_KASPERSKY_WS,
					   TEXT("Kaspersky antivirus for Workstation"))) ;
      SendDlgItemMessage (hDlg, IDC_ANTIVIRUS, CB_ADDSTRING, 0,
			  (LPARAM) STR_DEF(_ANTIVIRUS_BITDEFENDER,
					   TEXT("BitDefender"))) ;

      // set language dependant strings
      if( Language_IsLoaded() )
	{
	  SendDlgItemMessage (hDlg, IDC_LANGUAGE, CB_SELECTSTRING, 0,
			      (LPARAM)STR(_LANGUAGE_NAME_IN_ENGLISH)) ;   
	  SetDlgItemText (hDlg, IDT_CONFIGURATION, STR(_CONFIGURATION)) ;   
	  SetDlgItemText (hDlg, IDT_ANTIVIRUS, STR(_ANTIVIRUS)) ;
	  SetDlgItemText (hDlg, IDT_LANGUAGE, STR(_LANGUAGE)) ;
	  SetDlgItemText (hDlg, IDC_SPLASH_SCREEN, STR(_ENABLE_SPLASH_SCREEN)) ;
	  SetDlgItemText (hDlg, IDC_CHECK_UPDATES, STR(_CHECK_FOR_UPDATES)) ;
	  SetDlgItemText (hDlg, IDT_MAX_LOG_SIZE, STR(_MAX_LOG_FILE_SIZE)) ;
	  SetDlgItemText (hDlg, IDT_KILOBYTES, STR(_KILOBYTES)) ;
	  SetDlgItemText (hDlg, IDT_FILTERS, STR(_FILTERS)) ;
	  SetDlgItemText (hDlg, IDC_IMPORT, STR(_IMPORT)) ;
	  SetDlgItemText (hDlg, IDC_EXPORT, STR(_EXPORT)) ;
	  SetDlgItemText (hDlg, IDC_RESET, STR(_RESET)) ;
	  SetDlgItemText (hDlg, IDT_SOUNDS, STR(_SOUNDS)) ;
	  SetDlgItemText (hDlg, IDT_ALERT, STR(_ALERTING)) ;
	  SetDlgItemText (hDlg, IDT_ASK, STR(_ASKING)) ;
	  SetDlgItemText (hDlg, IDT_VIRUS, STR(_VIRUS_FOUND)) ;
	  SetDlgItemText (hDlg, IDC_TRAY_ICON_ANIMATION, STR(_TRAY_ICON_ANIMATION)) ;
	  SetDlgItemText (hDlg, IDC_NO_SOUND, STR(_NO_SOUND)) ;
	  SetDlgItemText (hDlg, IDC_DEFAULT_SOUNDS, STR(_DEFAULT_SOUNDS)) ;
	  SetDlgItemText (hDlg, IDC_CUSTOM_SOUNDS, STR(_CUSTOM_SOUNDS)) ;
	  SetDlgItemText (hDlg, IDT_SCAN_FILTERS, STR(_SCAN_ONLY_MATCHING_FILES)) ;
	  SetDlgItemText (hDlg, IDT_SCAN_FOLDERS, STR(_BACKGROUND_SCAN_THESE_FOLDERS)) ;
	}

    case WM_UPDATECONFIG:

      // antivirus
      SendDlgItemMessage (hDlg, IDC_ANTIVIRUS, CB_SETCURSEL, 
			  Scanner_GetScanner(), 0) ;

      // background scan folders
      {
	UINT	i, nLength ;
	LPCTSTR * pszArray ;
	HWND	hwndList = GetDlgItem (hDlg, IDC_SCAN_FOLDERS) ;
	LVITEM	lvi = {0} ;

	pszArray = Config_GetStringArray (CFGSAR_SCAN_FOLDERS, &nLength) ;

	ASSERT (hwndList!=NULL) ;
	TRACE_INFO (TEXT("nLength = %d\n"), nLength) ;

	ListView_DeleteAllItems (hwndList) ;
	for( i=0 ; i<nLength ; i++ )
	  {
	    TRACE_INFO (TEXT("%d = %s\n"), i, pszArray[i]) ;
	    lvi.iItem = i ;
	    lvi.mask = LVIF_TEXT ;
	    lvi.pszText = (LPTSTR)pszArray[i] ;
	    ListView_InsertItem (hwndList, &lvi) ;
	  }
      }

      // scan filters
      {
	UINT	i, nLength ;
	LPCTSTR * pszArray ;
	HWND	hwndList = GetDlgItem (hDlg, IDC_SCAN_FILTERS) ;
	LVITEM	lvi = {0} ;

	pszArray = Config_GetStringArray (CFGSAR_SCAN_FILTERS, &nLength) ;

	ASSERT (hwndList!=NULL) ;
	TRACE_INFO (TEXT("nLength = %d\n"), nLength) ;

	ListView_DeleteAllItems (hwndList) ;
	for( i=0 ; i<nLength ; i++ )
	  {
	    TRACE_INFO (TEXT("%d = %s\n"), i, pszArray[i]) ;
	    lvi.iItem = i ;
	    lvi.mask = LVIF_TEXT ;
	    lvi.pszText = (LPTSTR)pszArray[i] ;
	    ListView_InsertItem (hwndList, &lvi) ;
	  }
      }

      // check boxes
      for( i=0 ; i<arraysize(aCheckBoxes) ; i++ )
	CheckDlgButton (hDlg, aCheckBoxes[i].idCtrl, 
			Config_GetInteger(aCheckBoxes[i].idCfgInt)
			? BST_CHECKED : BST_UNCHECKED) ; 

      CheckDlgButton (hDlg, IDC_NO_SOUND, Config_GetInteger(CFGINT_SOUND)==0 ? BST_CHECKED : BST_UNCHECKED) ;
      CheckDlgButton (hDlg, IDC_DEFAULT_SOUNDS, Config_GetInteger(CFGINT_SOUND)==1 ? BST_CHECKED : BST_UNCHECKED) ;
      CheckDlgButton (hDlg, IDC_CUSTOM_SOUNDS, Config_GetInteger(CFGINT_SOUND)==2 ? BST_CHECKED : BST_UNCHECKED) ;

      EnableDlgItem (hDlg, IDC_ALERT_SOUND, Config_GetInteger(CFGINT_SOUND)==2) ;
      EnableDlgItem (hDlg, IDC_ASK_SOUND, Config_GetInteger(CFGINT_SOUND)==2) ;
      EnableDlgItem (hDlg, IDC_VIRUS_SOUND, Config_GetInteger(CFGINT_SOUND)==2) ;
      EnableDlgItem (hDlg, IDC_BROWSE_ALERT_SOUND, Config_GetInteger(CFGINT_SOUND)==2) ;
      EnableDlgItem (hDlg, IDC_BROWSE_ASK_SOUND, Config_GetInteger(CFGINT_SOUND)==2) ;
      EnableDlgItem (hDlg, IDC_BROWSE_VIRUS_SOUND, Config_GetInteger(CFGINT_SOUND)==2) ;

      SetDlgItemText (hDlg, IDC_ALERT_SOUND, Config_GetString(CFGSTR_ALERT_SOUND)) ;
      SetDlgItemText (hDlg, IDC_ASK_SOUND, Config_GetString(CFGSTR_ASK_SOUND)) ;
      SetDlgItemText (hDlg, IDC_VIRUS_SOUND, Config_GetString(CFGSTR_VIRUS_SOUND)) ;     

      // max log size
      SetDlgItemInt (hDlg, IDC_MAX_LOG_SIZE,
		     Config_GetInteger(CFGINT_MAX_LOG_SIZE)>>10,
		     FALSE) ;
      
      return TRUE ;

    case WM_COMMAND:

     switch( wParam )
	{
	case MAKELONG(IDC_LANGUAGE,CBN_SELCHANGE):  //    ======== 'Language' selection changed ========
	  {
	    TCHAR szBuffer[64] ;

	    // language
	    GetDlgItemText (hDlg, IDC_LANGUAGE, szBuffer, 64) ;
	    Language_LoadByName (szBuffer) ;
	  }
	  return TRUE ; //    ======== 'Language' selection changed ========

	case MAKELONG(IDC_ANTIVIRUS,CBN_SELCHANGE): //    ======== 'Antivirus' selection changed ========
	  {
	    int nScanner = SendDlgItemMessage(hDlg,IDC_ANTIVIRUS,CB_GETCURSEL,0,0) ;

	    if( ! Scanner_SetScanner(nScanner) )
	      {
		MessageBox (hDlg, STR_DEF(_ANTIVIRUS_NOT_INSTALLED,
					  TEXT("The antivirus you selected isn't installed")),
			    TEXT(APPLICATION_NAME), MB_ICONERROR) ;
	      }
	  }	  
	  return TRUE ; //    ======== 'Antivirus' selection changed ========

	case MAKELONG(IDC_MAX_LOG_SIZE,EN_KILLFOCUS): //    ======== 'Max log size' changed ========
	  {
	    // max log size
	    UINT nSize = 1024 * GetDlgItemInt (hDlg, IDC_MAX_LOG_SIZE, NULL, FALSE) ;	  
	    Config_SetInteger (CFGINT_MAX_LOG_SIZE, nSize) ;

	    LogFile_ReloadConfig () ;	
	  }
	  return TRUE ; //    ======== 'Max log size' changed ========

	case MAKELONG(IDC_SPLASH_SCREEN,BN_CLICKED):