WindowsSetupDlg.cpp

//====================================================================================
// Open Computer and Software Inventory Next Generation
// Copyright (C) 2010 OCS Inventory NG Team. All rights reserved.
// Web: http://www.ocsinventory-ng.org

// This code is open source and may be copied and modified as long as the source
// code is always made freely available.
// Please refer to the General Public Licence V2 http://www.gnu.org/ or Licence.txt
//====================================================================================

// WindowsSetupDlg.cpp : implementation file
//

#include "stdafx.h"
#include "ocs_deploy_tool.h"
#include "AgentSettings.h"
#include "WinAgentSettings.h"
#include "WindowsSetupDlg.h"
#include "FileVersion.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CWindowsSetupDlg dialog


CWindowsSetupDlg::CWindowsSetupDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CWindowsSetupDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CWindowsSetupDlg)
		// NOTE: the ClassWizard will add member initialization here
	//}}AFX_DATA_INIT
}


void CWindowsSetupDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CWindowsSetupDlg)
	DDX_Control(pDX, IDC_LIST_FILES, m_ListFiles);
	DDX_Control(pDX, IDC_LIST_PLUGINS, m_ListPlugins);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CWindowsSetupDlg, CDialog)
	//{{AFX_MSG_MAP(CWindowsSetupDlg)
	ON_BN_CLICKED(ID_WIZNEXT, OnWiznext)
	ON_BN_CLICKED(ID_WIZBACK, OnWizback)
	ON_BN_CLICKED(IDC_BUTTON_EXE, OnButtonExe)
	ON_BN_CLICKED(IDC_BUTTON_ADD, OnButtonAdd)
	ON_BN_CLICKED(IDC_BUTTON_REMOVE, OnButtonRemove)
	ON_LBN_DBLCLK(IDC_LIST_FILES, OnDblclkListFiles)
	ON_BN_CLICKED(IDC_BUTTON_CHANGE_DIRECTORY, OnButtonChangeDirectory)
	ON_BN_CLICKED(IDC_CHECK_DEBUG, OnClickVerboseLog)
	ON_BN_CLICKED(IDC_CHECK_NO_SERVICE, OnClickNoService)
	ON_BN_CLICKED(IDC_CHECK_NO_SYSTRAY, OnClickNoSystray)
	ON_BN_CLICKED(IDC_CHECK_LAUNCH_INVENTORY, OnClickLaunchInventory)
	//}}AFX_MSG_MAP
	ON_BN_CLICKED(IDC_BUTTON_ADD_PLUGIN, &CWindowsSetupDlg::OnBnClickedButtonAddPlugin)
	ON_BN_CLICKED(IDC_BUTTON_REMOVE_PLUGIN, &CWindowsSetupDlg::OnBnClickedButtonRemovePlugin)
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CWindowsSetupDlg message handlers

BOOL CWindowsSetupDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// TODO: Add extra initialization here
	try
	{
		CString	csMessage, csSection, csNumber;
		BOOL bNoMoreValue;
		int nIndex;
		
		// Get tool version
		CFileVersion fileVer;
		CString		 csVersion;
		// Get application path	
		if (GetModuleFileName( AfxGetInstanceHandle(), csVersion.GetBuffer( 4*_MAX_PATH+1), 4*_MAX_PATH) == 0)
		{
			csVersion.Empty();
		}
		else
		{
			csVersion.ReleaseBuffer();
			// Open application file to get version from file
			if (fileVer.Open( csVersion))
			{
				csVersion = fileVer.GetProductVersion();
				csVersion.Remove( ' ');
				csVersion.Replace( ',', '.');
				fileVer.Close();
			}
			else
				csVersion.Empty();
		}
		csMessage.FormatMessage( IDS_OCS_DEPLOY_TOOL, csVersion);
		SetDlgItemText( IDC_STATUS, csMessage);
		// Set Agent setup file
		csMessage = AfxGetApp()->GetProfileString( AGENT_SECTION, OPTION_WIN_AGENT_SETUP_FILE, _T( ""));
		SetDlgItemText( IDC_EDIT_EXE, csMessage);
		// Set list of other files to copy
		csSection.Format( _T("%s\\%s"), AGENT_SECTION, OPTION_WIN_AGENT_OTHER_FILES);
		nIndex = 0;
		bNoMoreValue = FALSE;
		while (!bNoMoreValue)
		{
			csNumber.Format( _T( "%d"), nIndex);
			csMessage = AfxGetApp()->GetProfileString( csSection, csNumber, _T( ""));
			if (!csMessage.IsEmpty())
				m_ListFiles.AddString( csMessage);
			else
				bNoMoreValue = TRUE;
			nIndex ++;
		}
		// Set list of plugins to copy
		csSection.Format( _T("%s\\%s"), AGENT_SECTION, WIN_AGENT_PLUGINS_FOLDER);
		nIndex = 0;
		bNoMoreValue = FALSE;
		while (!bNoMoreValue)
		{
			csNumber.Format( _T( "%d"), nIndex);
			csMessage = AfxGetApp()->GetProfileString( csSection, csNumber, _T( ""));
			if (!csMessage.IsEmpty())
				m_ListPlugins.AddString( csMessage);
			else
				bNoMoreValue = TRUE;
			nIndex ++;
		}
		// Set OCS server address
		csMessage = AfxGetApp()->GetProfileString( AGENT_SECTION, OPTION_SERVER_ADDRESS, DEFAULT_SERVER_ADDRESS);
		SetDlgItemText( IDC_EDIT_SERVER_IP, csMessage);
		// Set installer command line options
		csMessage = AfxGetApp()->GetProfileString( AGENT_SECTION, OPTION_WIN_INSTALLER_OPTIONS, _T( ""));
		SetDlgItemText( IDC_EDIT_OPTIONS, csMessage);
		// Check /DEBUG to installer command line options
		if (csMessage.Find( _T( "/DEBUG")) != -1)
			// Option not already set
			CheckDlgButton( IDC_CHECK_DEBUG, TRUE);
		// Check /NO_SERVICE to installer command line options
		if (csMessage.Find( _T( "/NO_SERVICE")) != -1)
			// Option not already set
			CheckDlgButton( IDC_CHECK_NO_SERVICE, TRUE);
		// Check /NO_SYSTRAY to installer command line options
		if (csMessage.Find( _T( "/NO_SYSTRAY")) != -1)
			// Option not already set
			CheckDlgButton( IDC_CHECK_LAUNCH_INVENTORY, TRUE);
		// Check /NOW to installer command line options
		if (csMessage.Find( _T( "/NOW")) != -1)
			// Option not already set
			CheckDlgButton( IDC_CHECK_LAUNCH_INVENTORY, TRUE);
		// Set agent's installation directory
		if (_tcscmp( m_pSettings->GetAgentSetupDirectory(), DEFAULT_WIN_AGENT_DIRECTORY) != 0)
		{
			// Not default setup folder selected, so enable checkbox
			CheckDlgButton( IDC_BUTTON_CHANGE_DIRECTORY, TRUE);
			GetDlgItem( IDC_EDIT_INSTALL)->EnableWindow( TRUE);
		}
		SetDlgItemText( IDC_EDIT_INSTALL, m_pSettings->GetAgentSetupDirectory());
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
	}
	return TRUE;  // return TRUE unless you set the focus to a control
	              // EXCEPTION: OCX Property Pages should return FALSE
}

void CWindowsSetupDlg::OnWiznext() 
{
	// TODO: Add your control notification handler code here
	try
	{
		if (!Save( TRUE))
			return;
		EndDialog( ID_WIZNEXT);
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		return;
	}
}

void CWindowsSetupDlg::OnWizback() 
{
	// TODO: Add your control notification handler code here
	try
	{
		Save();
		EndDialog( ID_WIZBACK);
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		return;
	}
}

void CWindowsSetupDlg::OnCancel() 
{
	// TODO: Add extra cleanup here
	
	CDialog::OnCancel();
}

void CWindowsSetupDlg::OnButtonExe() 
{
	// TODO: Add your control notification handler code here
	LPITEMIDLIST	pMyIdList = NULL;
	LPMALLOC		pIMalloc;

	try
	{
		CFileDialog		dlgOpenFile( TRUE, NULL, NULL, OFN_FILEMUSTEXIST | OFN_HIDEREADONLY, _T( "OCS Inventory NG Agent for Windows Setup|OCS-NG-Windows-Agent-Setup.exe|Executable files|*.exe||"));
		TCHAR			szInitialFolder[4*_MAX_PATH+1];
		CString			csMessage;

		// Get User Desktop path
		if (SHGetMalloc( &pIMalloc) != NOERROR)
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			return;
		}
		if (SHGetSpecialFolderLocation( m_hWnd, CSIDL_DESKTOP, &pMyIdList) != NOERROR)
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			return;
		}
		if (!SHGetPathFromIDList( pMyIdList, szInitialFolder))
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			pIMalloc->Free( pMyIdList);
			return;
		}
		pIMalloc->Free( pMyIdList);

		// Fill in the OPENFILENAME structure to support a template and hook.
		dlgOpenFile.m_ofn.lpstrInitialDir   = szInitialFolder;
		if (!csMessage.LoadString( IDS_SELECT_AGENT_FILE))
			AfxThrowMemoryException();
		dlgOpenFile.m_ofn.lpstrTitle        = csMessage.GetBuffer( csMessage.GetLength());
		if (dlgOpenFile.DoModal() != IDOK)
			return;
		SetDlgItemText( IDC_EDIT_EXE, dlgOpenFile.GetPathName());
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		if (pMyIdList != NULL)
			pIMalloc->Free( pMyIdList);
		return;
	}
}

void CWindowsSetupDlg::OnButtonAdd() 
{
	// TODO: Add your control notification handler code here
	LPITEMIDLIST	pMyIdList = NULL;
	LPMALLOC		pIMalloc;

	try
	{
		CFileDialog		dlgOpenFile( TRUE, NULL, NULL, OFN_FILEMUSTEXIST | OFN_HIDEREADONLY, _T( "All files|*.*||"));
		TCHAR			szInitialFolder[4*_MAX_PATH+1];
		CString			csMessage;

		// Get User Desktop path
		if (SHGetMalloc( &pIMalloc) != NOERROR)
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			return;
		}
		if (SHGetSpecialFolderLocation( m_hWnd, CSIDL_DESKTOP, &pMyIdList) != NOERROR)
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			return;
		}
		if (!SHGetPathFromIDList( pMyIdList, szInitialFolder))
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			pIMalloc->Free( pMyIdList);
			return;
		}
		pIMalloc->Free( pMyIdList);
		pMyIdList = NULL;

		// Fill in the OPENFILENAME structure to support a template and hook.
		dlgOpenFile.m_ofn.lpstrInitialDir   = szInitialFolder;
		if (!csMessage.LoadString( IDS_SELECT_OTHER_FILE))
			AfxThrowMemoryException();
		dlgOpenFile.m_ofn.lpstrTitle        = csMessage.GetBuffer( csMessage.GetLength());
		if (dlgOpenFile.DoModal() != IDOK)
			return;
		m_ListFiles.AddString( dlgOpenFile.GetPathName());
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		if (pMyIdList != NULL)
			pIMalloc->Free( pMyIdList);
		return;
	}
}

void CWindowsSetupDlg::OnButtonRemove() 
{
	// TODO: Add your control notification handler code here
	try
	{
		int		nCurSel;

		if ((nCurSel = m_ListFiles.GetCurSel()) == LB_ERR )
			// No item selected
			return;
		m_ListFiles.DeleteString( nCurSel);
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		return;
	}
}

void CWindowsSetupDlg::OnDblclkListFiles() 
{
	// TODO: Add your control notification handler code here
	OnButtonRemove();
}

void CWindowsSetupDlg::OnClickVerboseLog() 
{
	// TODO: Add your control notification handler code here
	CString csOptions;
	int		nIndex;

	if (IsDlgButtonChecked( IDC_CHECK_DEBUG))
	{
		// Add /DEBUG to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if (csOptions.Find( _T( "/DEBUG")) == -1)
			// Option not already set
			csOptions.Append( _T( " /DEBUG=1"));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
	else
	{
		// Remove /DEBUG to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if ((nIndex = csOptions.Find( _T( "/DEBUG=1"))) != -1)
			// Option not already set
			csOptions.Delete( nIndex, _tcslen( _T( "/DEBUG=1")));
		if ((nIndex = csOptions.Find( _T( "/DEBUG=2"))) != -1)
			// Option not already set
			csOptions.Delete( nIndex, _tcslen( _T( "/DEBUG=2")));
		if ((nIndex = csOptions.Find( _T( "/DEBUG"))) != -1)
			// Option not already set
			csOptions.Delete( nIndex, _tcslen( _T( "/DEBUG")));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
}

void CWindowsSetupDlg::OnClickNoService() 
{
	// TODO: Add your control notification handler code here
	CString csOptions;
	int		nIndex;

	if (IsDlgButtonChecked( IDC_CHECK_NO_SERVICE))
	{
		// Add /NO_SERVICE to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if (csOptions.Find( _T( "/NO_SERVICE")) == -1)
			// Option not already set
			csOptions.Append( _T( " /NO_SERVICE"));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
	else
	{
		// Remove /NO_SERVICE to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if ((nIndex = csOptions.Find( _T( "/NO_SERVICE"))) != -1)
			// Option not already set
			csOptions.Delete( nIndex, _tcslen( _T( "/NO_SERVICE")));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
}

void CWindowsSetupDlg::OnClickNoSystray() 
{
	// TODO: Add your control notification handler code here
	CString csOptions;
	int		nIndex;

	if (IsDlgButtonChecked( IDC_CHECK_NO_SYSTRAY))
	{
		// Add /NO_SYSTRAY to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if (csOptions.Find( _T( "/NO_SYSTRAY")) == -1)
			// Option not already set
			csOptions.Append( _T( " /NO_SYSTRAY"));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
	else
	{
		// Remove /NO_SYSTRAY to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if ((nIndex = csOptions.Find( _T( "/NO_SYSTRAY"))) != -1)
			// Option not already set
			csOptions.Delete( nIndex, _tcslen( _T( "/NO_SYSTRAY")));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
}

void CWindowsSetupDlg::OnClickLaunchInventory() 
{
	// TODO: Add your control notification handler code here
	CString csOptions;
	int		nIndex;

	if (IsDlgButtonChecked( IDC_CHECK_LAUNCH_INVENTORY))
	{
		// Add /NOW to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if (csOptions.Find( _T( "/NOW")) == -1)
			// Option not already set
			csOptions.Append( _T( " /NOW"));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
	else
	{
		// Remove /NOW to installer command line options
		GetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
		if ((nIndex = csOptions.Find( _T( "/NOW"))) != -1)
			// Option not already set
			csOptions.Delete( nIndex, _tcslen( _T( "/NOW")));
		SetDlgItemText( IDC_EDIT_OPTIONS, csOptions);
	}
}

BOOL CWindowsSetupDlg::Save( BOOL bDisplayError)
{
	int nIndex, nCount = 0;
	CString csMessage, csSection, csNumber;
	CStringList *pList;

	// Get Agent setup file
	if (GetDlgItemText( IDC_EDIT_EXE, csMessage) == 0)
	{
		// No data
		bDisplayError && AfxMessageBox( IDS_ERROR_INVALID_VALUE, MB_ICONINFORMATION);
		GetDlgItem( IDC_EDIT_EXE)->SetFocus();
		return FALSE;
	}
	AfxGetApp()->WriteProfileString( AGENT_SECTION, OPTION_WIN_AGENT_SETUP_FILE, csMessage);
	m_pSettings->SetAgentSetupFile( csMessage);
	// Remove registry stored files, before saving new ones
	RemoveOtherFilesFromRegistry();
	// Get list of other files to copy
	pList = m_pSettings->GetAgentOtherFiles();
	csSection.Format( _T("%s\\%s"), AGENT_SECTION, OPTION_WIN_AGENT_OTHER_FILES);
	if (((nCount = m_ListFiles.GetCount()) == LB_ERR) || (nCount == 0))
	{
		// No items in List => NO FILE TO COPY
		pList->RemoveAll();
	}
	else
	{
		// Empty current list
		pList->RemoveAll();
		// Fill in file list with each file from listbox
		for (nIndex = 0; nIndex < nCount; nIndex++)
		{
			m_ListFiles.GetText( nIndex, csMessage);
			pList->AddTail( csMessage);
			csNumber.Format( _T( "%d"), nIndex);
			AfxGetApp()->WriteProfileString( csSection, csNumber, csMessage);
		}
	}
	// Remove registry stored plugins, before saving new ones
	RemovePluginsFilesFromRegistry();
	// Get list of plugins to copy
	pList = m_pSettings->GetPluginFiles();
	csSection.Format( _T("%s\\%s"), AGENT_SECTION, WIN_AGENT_PLUGINS_FOLDER);
	if (((nCount = m_ListPlugins.GetCount()) == LB_ERR) || (nCount == 0))
	{
		// No items in List => NO PLUGIN TO COPY
		pList->RemoveAll();
	}
	else
	{
		// Empty current list
		pList->RemoveAll();
		// Fill in plugin list with each plugin from listbox
		for (nIndex = 0; nIndex < nCount; nIndex++)
		{
			m_ListPlugins.GetText( nIndex, csMessage);
			pList->AddTail( csMessage);
			csNumber.Format( _T( "%d"), nIndex);
			AfxGetApp()->WriteProfileString( csSection, csNumber, csMessage);
		}
	}
	// Get OCS server address
	if (GetDlgItemText( IDC_EDIT_SERVER_IP, csMessage) == 0)
	{
		// No data
		bDisplayError && AfxMessageBox( IDS_ERROR_INVALID_VALUE, MB_ICONINFORMATION);
		GetDlgItem( IDC_EDIT_SERVER_IP)->SetFocus();
		return FALSE;
	}
	AfxGetApp()->WriteProfileString( AGENT_SECTION, OPTION_SERVER_ADDRESS, csMessage);
	m_pSettings->SetServerAddress( csMessage);
	// Get installer command line options
	GetDlgItemText( IDC_EDIT_OPTIONS, csMessage);
	AfxGetApp()->WriteProfileString( AGENT_SECTION, OPTION_WIN_INSTALLER_OPTIONS, csMessage);
	m_pSettings->SetInstallerOptions( csMessage);
	// Get agent's installation directory
	if (GetDlgItemText( IDC_EDIT_INSTALL, csMessage) == 0)
	{
		// No data
		bDisplayError && AfxMessageBox( IDS_ERROR_INVALID_VALUE, MB_ICONINFORMATION);
		GetDlgItem( IDC_EDIT_INSTALL)->SetFocus();
		return FALSE;
	}
	m_pSettings->SetAgentSetupDirectory( csMessage);
	return TRUE;
}

void CWindowsSetupDlg::SetAgentSetting( CAgentSettings *pSettings)
{
	m_pSettings = pSettings;
}

void CWindowsSetupDlg::OnButtonChangeDirectory() 
{
	// TODO: Add your control notification handler code here
	try
	{
		if (IsDlgButtonChecked( IDC_BUTTON_CHANGE_DIRECTORY))
			GetDlgItem( IDC_EDIT_INSTALL)->EnableWindow( TRUE);
		else
		{
			GetDlgItem( IDC_EDIT_INSTALL)->EnableWindow( FALSE);
			SetDlgItemText( IDC_EDIT_INSTALL, DEFAULT_WIN_AGENT_DIRECTORY);
		}
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		return;
	}
}

BOOL CWindowsSetupDlg::RemoveOtherFilesFromRegistry()
{
	HKEY	hKey;
	CString csKey, csValue;
	DWORD	dwErr, dwLength;
	UINT	uIndex = 0;

	csKey.Format( _T( "SOFTWARE\\OCS_DEPLOY_TOOL\\%s\\%s"), AGENT_SECTION, OPTION_WIN_AGENT_OTHER_FILES);
	if ((dwErr = RegOpenKeyEx( HKEY_CURRENT_USER, csKey, 0, KEY_READ|KEY_WRITE, &hKey)) != ERROR_SUCCESS)
		return FALSE;
	while (dwErr == ERROR_SUCCESS)
	{
		csValue.Format( _T( "%u"), uIndex);
		// Check if value exists
		dwErr = REG_SZ;
		dwLength = 0;
		if ((dwErr = RegQueryValueEx( hKey, csValue, NULL, &dwErr, NULL, &dwLength)) == ERROR_SUCCESS)
			// Key exists, delete it
			RegDeleteValue( hKey, csValue);
		// Continue
		uIndex ++;
	}
	RegCloseKey( hKey);
	return TRUE;
}

void CWindowsSetupDlg::OnBnClickedButtonAddPlugin()
{
	// TODO: Add your control notification handler code here
	LPITEMIDLIST	pMyIdList = NULL;
	LPMALLOC		pIMalloc;

	try
	{
		CFileDialog		dlgOpenFile( TRUE, NULL, NULL, OFN_FILEMUSTEXIST | OFN_HIDEREADONLY, _T( "All files|*.*||"));
		TCHAR			szInitialFolder[4*_MAX_PATH+1];
		CString			csMessage;

		// Get User Desktop path
		if (SHGetMalloc( &pIMalloc) != NOERROR)
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			return;
		}
		if (SHGetSpecialFolderLocation( m_hWnd, CSIDL_DESKTOP, &pMyIdList) != NOERROR)
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			return;
		}
		if (!SHGetPathFromIDList( pMyIdList, szInitialFolder))
		{
			AfxMessageBox( IDS_ERROR_DISPLAY_FOLDER, MB_ICONSTOP);
			pIMalloc->Free( pMyIdList);
			return;
		}
		pIMalloc->Free( pMyIdList);
		pMyIdList = NULL;

		// Fill in the OPENFILENAME structure to support a template and hook.
		dlgOpenFile.m_ofn.lpstrInitialDir   = szInitialFolder;
		if (!csMessage.LoadString( IDS_SELECT_PLUGIN_FILE))
			AfxThrowMemoryException();
		dlgOpenFile.m_ofn.lpstrTitle        = csMessage.GetBuffer( csMessage.GetLength());
		if (dlgOpenFile.DoModal() != IDOK)
			return;
		m_ListPlugins.AddString( dlgOpenFile.GetPathName());
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		if (pMyIdList != NULL)
			pIMalloc->Free( pMyIdList);
		return;
	}
}

void CWindowsSetupDlg::OnBnClickedButtonRemovePlugin()
{
	// TODO: Add your control notification handler code here
	try
	{
		int		nCurSel;

		if ((nCurSel = m_ListPlugins.GetCurSel()) == LB_ERR )
			// No item selected
			return;
		m_ListPlugins.DeleteString( nCurSel);
	}
	catch( CException *pEx)
	{
		pEx->ReportError( MB_OK|MB_ICONSTOP);
		pEx->Delete();
		return;
	}
}

BOOL CWindowsSetupDlg::RemovePluginsFilesFromRegistry()
{
	HKEY	hKey;
	CString csKey, csValue;
	DWORD	dwErr, dwLength;
	UINT	uIndex = 0;

	csKey.Format( _T( "SOFTWARE\\OCS_DEPLOY_TOOL\\%s\\%s"), AGENT_SECTION, WIN_AGENT_PLUGINS_FOLDER);
	if ((dwErr = RegOpenKeyEx( HKEY_CURRENT_USER, csKey, 0, KEY_READ|KEY_WRITE, &hKey)) != ERROR_SUCCESS)
		return FALSE;
	while (dwErr == ERROR_SUCCESS)
	{
		csValue.Format( _T( "%u"), uIndex);
		// Check if value exists
		dwErr = REG_SZ;
		dwLength = 0;
		if ((dwErr = RegQueryValueEx( hKey, csValue, NULL, &dwErr, NULL, &dwLength)) == ERROR_SUCCESS)
			// Key exists, delete it
			RegDeleteValue( hKey, csValue);
		// Continue
		uIndex ++;
	}
	RegCloseKey( hKey);
	return TRUE;
}