401 lines
11 KiB
C++
401 lines
11 KiB
C++
//+---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1997.
|
|
//
|
|
// File: S A U I O B J. C P P
|
|
//
|
|
// Contents: Implementation of the LAN ConnectionUI object
|
|
//
|
|
// Notes:
|
|
//
|
|
// Created: tongl 8 Oct 1997
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#include "pch.h"
|
|
#pragma hdrstop
|
|
#include "ncnetcon.h"
|
|
#include "ncras.h"
|
|
#include "sauiobj.h"
|
|
#include "saui.h"
|
|
#include "resource.h"
|
|
#include "lanhelp.h"
|
|
#include "lanui.h"
|
|
#include "ncui.h"
|
|
|
|
//+---------------------------------------------------------------------------
|
|
// INetConnectionUI
|
|
//
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CSharedAccessConnectionUi::SetConnection
|
|
//
|
|
// Purpose: Sets the LAN connection that this UI object will operate upon
|
|
//
|
|
// Arguments:
|
|
// pCon [in] LAN connection object to operate on. Can be NULL.
|
|
//
|
|
// Returns: S_OK if success, OLE error otherwise
|
|
//
|
|
// Author: danielwe 16 Oct 1997
|
|
//
|
|
// Notes:
|
|
//
|
|
STDMETHODIMP CSharedAccessConnectionUi::SetConnection(INetConnection* pCon)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
ReleaseObj(m_pconn);
|
|
m_pconn = pCon;
|
|
AddRefObj(m_pconn);
|
|
|
|
TraceError("CSharedAccessConnectionUi::SetConnection", hr);
|
|
return hr;
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CSharedAccessConnectionUi::Connect
|
|
//
|
|
// Purpose: Tells the connection to connect, optionally displaying UI of
|
|
// connection progress.
|
|
//
|
|
// Arguments:
|
|
// hwndParent [in] Parent window for UI
|
|
// dwFlags [in] Flags affecting how UI is shown
|
|
//
|
|
// Returns: S_OK if success, OLE error otherwise
|
|
//
|
|
// Author: danielwe 16 Oct 1997
|
|
//
|
|
// Notes:
|
|
//
|
|
STDMETHODIMP CSharedAccessConnectionUi::Connect(HWND hwndParent, DWORD dwFlags)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (!m_pconn)
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
}
|
|
else
|
|
{
|
|
CSharedAccessConnectionUiDlg dlg; // we are borrowing the CLanConnectionUiDlg because it works so well for us.
|
|
HWND hwndDlg;
|
|
|
|
NETCON_MEDIATYPE MediaType = NCM_NONE; // assume no ui
|
|
NETCON_PROPERTIES* pProperties;
|
|
hr = m_pconn->GetProperties(&pProperties);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
MediaType = pProperties->MediaType;
|
|
FreeNetconProperties(pProperties);
|
|
}
|
|
else
|
|
{
|
|
hr = S_OK; // ok if prev fails
|
|
}
|
|
|
|
if (!(dwFlags & NCUC_NO_UI))
|
|
{
|
|
// Display UI prior to connect
|
|
//
|
|
|
|
dlg.SetConnection(m_pconn);
|
|
hwndDlg = dlg.Create(hwndParent);
|
|
|
|
if (!hwndDlg)
|
|
{
|
|
hr = E_FAIL;
|
|
}
|
|
}
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = m_pconn->Connect();
|
|
|
|
// Sleep a bit so they can read the text
|
|
Sleep(1000);
|
|
|
|
if (!(dwFlags & NCUC_NO_UI))
|
|
{
|
|
SetDlgItemText(hwndDlg, IDC_TXT_Caption, c_szEmpty);
|
|
Sleep(100);
|
|
|
|
UINT ids = SUCCEEDED(hr) ?
|
|
IDS_SHAREDACCESSUI_CONNECTED :
|
|
IDS_LAN_CONNECT_FAILED;
|
|
|
|
PCWSTR szwResult = SzLoadIds(ids);
|
|
SetDlgItemText(hwndDlg, IDC_TXT_Caption, szwResult);
|
|
|
|
// Sleep a bit so they can read the text
|
|
Sleep(1000);
|
|
|
|
DestroyWindow(hwndDlg);
|
|
|
|
if(E_ACCESSDENIED == hr)
|
|
{
|
|
NcMsgBox(_Module.GetResourceInstance(), NULL, IDS_CONFOLD_WARNING_CAPTION, IDS_SHAREDACCESSUI_ACCESSDENIED, MB_OK | MB_ICONEXCLAMATION);
|
|
hr = S_OK; // handled the error
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
TraceHr(ttidError, FAL, hr, FALSE, "CSharedAccessConnectionUi::Connect");
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP CSharedAccessConnectionUi::Disconnect(HWND hwndParent, DWORD dwFlags)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (!m_pconn)
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
}
|
|
else
|
|
{
|
|
hr = m_pconn->Disconnect();
|
|
if(E_ACCESSDENIED == hr)
|
|
{
|
|
NcMsgBox(_Module.GetResourceInstance(), NULL, IDS_CONFOLD_WARNING_CAPTION, IDS_SHAREDACCESSUI_ACCESSDENIED, MB_OK | MB_ICONEXCLAMATION);
|
|
hr = S_OK; // handled the error
|
|
}
|
|
}
|
|
|
|
TraceHr (ttidError, FAL, hr, FALSE, "CSharedAccessConnectionUi::Disconnect");
|
|
return hr;
|
|
}
|
|
//+---------------------------------------------------------------------------
|
|
// INetConnectionPropertyUi2
|
|
//
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CSharedAccessConnectionUi::AddPages
|
|
//
|
|
// Purpose: Called when our UI object shoud add its pages to a property
|
|
// sheet for the connection UI owned by the shell.
|
|
//
|
|
// Arguments:
|
|
// pfnAddPage [in] Callback function to add the page
|
|
// lParam [in] User-defined paramter required by the callback
|
|
// function.
|
|
//
|
|
// Returns: S_OK if succeeded, otherwise OLE error.
|
|
//
|
|
// Author: danielwe 28 Oct 1997
|
|
//
|
|
// Notes:
|
|
//
|
|
STDMETHODIMP CSharedAccessConnectionUi::AddPages(HWND hwndParent,
|
|
LPFNADDPROPSHEETPAGE pfnAddPage,
|
|
LPARAM lParam)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (!pfnAddPage)
|
|
{
|
|
hr = E_POINTER;
|
|
}
|
|
else if (!m_pconn)
|
|
{
|
|
hr = E_UNEXPECTED;
|
|
}
|
|
else
|
|
{
|
|
NETCON_PROPERTIES* pProperties;
|
|
hr = m_pconn->GetProperties(&pProperties);
|
|
if(SUCCEEDED(hr))
|
|
{
|
|
if (!m_pspSharedAccessPage)
|
|
{
|
|
m_pspSharedAccessPage = new CSharedAccessPage(static_cast<INetConnectionPropertyUi *>(this),
|
|
m_pnc, m_pconn, m_fReadOnly, m_fNeedReboot,
|
|
m_fAccessDenied, g_aHelpIDs_IDD_SHAREDACCESS_GENERAL);
|
|
}
|
|
|
|
if (m_pspSharedAccessPage)
|
|
{
|
|
(VOID) pfnAddPage(m_pspSharedAccessPage->CreatePage(IDD_SHAREDACCESS_GENERAL, 0),
|
|
lParam);
|
|
}
|
|
|
|
|
|
FreeNetconProperties(pProperties);
|
|
}
|
|
}
|
|
|
|
TraceError("CSharedAccessConnectionUi::AddPages(INetConnectionPropertyUi)", hr);
|
|
return hr;
|
|
}
|
|
|
|
STDMETHODIMP
|
|
CSharedAccessConnectionUi::GetIcon (
|
|
DWORD dwSize,
|
|
HICON *phIcon )
|
|
{
|
|
HRESULT hr;
|
|
Assert (phIcon);
|
|
|
|
hr = HrGetIconFromMediaType(dwSize, NCM_SHAREDACCESSHOST_LAN, NCSM_NONE, 7, 0, phIcon);
|
|
|
|
TraceError ("CLanConnectionUi::GetIcon (INetConnectionPropertyUi2)", hr);
|
|
|
|
return hr;
|
|
}
|
|
|
|
|
|
//
|
|
// INetConnectionUiLock
|
|
//
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CSharedAccessConnectionUi::QueryLock
|
|
//
|
|
// Purpose: Causes the UI object to attempt to get the INetCfg write lock.
|
|
//
|
|
// Arguments:
|
|
// ppszwLockHolder [out] Description of component that holds the
|
|
// write lock in the event that it couldn't be
|
|
// obtained.
|
|
//
|
|
// Returns: S_OK if success, S_FALSE if write lock couldn't be obtained,
|
|
// OLE or Win32 error otherwise
|
|
//
|
|
// Author: danielwe 13 Nov 1997
|
|
//
|
|
// Notes:
|
|
//
|
|
STDMETHODIMP CSharedAccessConnectionUi::QueryLock(PWSTR* ppszwLockHolder)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
|
|
if (!ppszwLockHolder)
|
|
{
|
|
hr = E_POINTER;
|
|
}
|
|
else
|
|
{
|
|
INetCfgLock * pnclock;
|
|
|
|
AssertSz(!m_pnc, "We're assuming this is in the property sheet "
|
|
"context and we don't yet have an INetCfg!");
|
|
|
|
*ppszwLockHolder = NULL;
|
|
|
|
// Instantiate an INetCfg
|
|
hr = CoCreateInstance(
|
|
CLSID_CNetCfg,
|
|
NULL,
|
|
CLSCTX_INPROC_SERVER | CLSCTX_NO_CODE_DOWNLOAD,
|
|
IID_INetCfg,
|
|
reinterpret_cast<LPVOID *>(&m_pnc));
|
|
|
|
TraceHr(ttidError, FAL, hr, FALSE, "CoCreateInstance");
|
|
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// Get the locking interface
|
|
hr = m_pnc->QueryInterface(IID_INetCfgLock,
|
|
reinterpret_cast<LPVOID *>(&pnclock));
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
// Attempt to lock the INetCfg for read/write
|
|
hr = pnclock->AcquireWriteLock(0,
|
|
SzLoadIds(IDS_SHAREDACCESSUI_LOCK_DESC), ppszwLockHolder);
|
|
|
|
ReleaseObj(pnclock);
|
|
|
|
if (NETCFG_E_NEED_REBOOT == hr)
|
|
{
|
|
// Can't make any changes because we are pending a reboot.
|
|
m_fReadOnly = TRUE;
|
|
m_fNeedReboot = TRUE;
|
|
hr = S_OK;
|
|
}
|
|
else if(E_ACCESSDENIED == hr)
|
|
{
|
|
// user not logged on as admin
|
|
//
|
|
m_fReadOnly = TRUE;
|
|
m_fAccessDenied = TRUE;
|
|
hr = S_OK;
|
|
}
|
|
else if (S_FALSE == hr)
|
|
{
|
|
// We don't have sufficent rights
|
|
//
|
|
m_fReadOnly = TRUE;
|
|
hr = S_OK;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
TraceError("CSharedAccessConnectionUi::QueryLock", (S_FALSE == hr) ? S_OK : hr);
|
|
return hr;
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CSharedAccessConnectionUiDlg::OnInitDialog
|
|
//
|
|
// Purpose: Handles the WM_INITDIALOG message.
|
|
//
|
|
// Arguments:
|
|
// uMsg []
|
|
// wParam []
|
|
// lParam []
|
|
// bHandled []
|
|
//
|
|
// Returns: TRUE
|
|
//
|
|
// Author: kenwic 19 Sep 2000
|
|
//
|
|
// Notes:
|
|
//
|
|
LRESULT CSharedAccessConnectionUiDlg::OnInitDialog(UINT uMsg, WPARAM wParam,
|
|
LPARAM lParam, BOOL& bHandled)
|
|
{
|
|
HRESULT hr = S_OK;
|
|
NETCON_PROPERTIES* pProps;
|
|
|
|
AssertSz(m_pconn, "No connection object in dialog!");
|
|
|
|
hr = m_pconn->GetProperties(&pProps);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
SetDlgItemText(IDC_TXT_Caption, SzLoadIds(IDS_SHAREDACCESSUI_CONNECTING));
|
|
SetWindowText(pProps->pszwName);
|
|
|
|
HICON hLanIconSmall;
|
|
HICON hLanIconBig;
|
|
|
|
hr = HrGetIconFromMediaType(GetSystemMetrics(SM_CXSMICON), NCM_SHAREDACCESSHOST_LAN, NCSM_NONE, 7, 0, &hLanIconSmall);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
hr = HrGetIconFromMediaType(GetSystemMetrics(SM_CXICON), NCM_SHAREDACCESSHOST_LAN, NCSM_NONE, 7, 0, &hLanIconBig);
|
|
if (SUCCEEDED(hr))
|
|
{
|
|
SetIcon(hLanIconSmall, FALSE);
|
|
SetIcon(hLanIconBig, TRUE);
|
|
|
|
SendDlgItemMessage(IDI_Device_Icon, STM_SETICON, reinterpret_cast<WPARAM>(hLanIconBig), 0);
|
|
|
|
}
|
|
}
|
|
|
|
FreeNetconProperties(pProps);
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|