Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

454 lines
10 KiB

//============================================================================
// Copyright (C) Microsoft Corporation, 1996 - 1999
//
// File: msgdlg.cpp
//
// History:
// 10/23/96 Abolade Gbadegesin Created.
//
// Implementation of the "Send Message" dialogs.
//============================================================================
#include "stdafx.h"
#include "dialog.h"
#include "rtrstr.h"
extern "C" {
//nclude "dim.h"
//nclude "ras.h"
//nclude "lm.h"
}
#include "msgdlg.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
//----------------------------------------------------------------------------
// Class: CMessageDlg
//
//----------------------------------------------------------------------------
//----------------------------------------------------------------------------
// Function: CMessageDlg::CMessageDlg
//
// Constructor.
//----------------------------------------------------------------------------
CMessageDlg::CMessageDlg(
LPCTSTR pszServerName,
LPCTSTR pszUserName,
LPCTSTR pszComputer,
HANDLE hConnection,
CWnd* pParent
) : CBaseDialog(IDD_DDM_MESSAGE, pParent),
m_fUser(TRUE),
m_sServerName(pszServerName ? pszServerName : c_szEmpty),
m_sUserName(pszUserName ? pszUserName : c_szEmpty),
m_sTarget(pszComputer ? pszComputer : c_szEmpty),
m_hConnection(hConnection)
{
}
CMessageDlg::CMessageDlg(
LPCTSTR pszServerName,
LPCTSTR pszTarget,
CWnd* pParent
) : CBaseDialog(IDD_DDM_MESSAGE, pParent),
m_fUser(FALSE),
m_sServerName(pszServerName ? pszServerName: c_szEmpty),
m_sUserName(c_szEmpty),
m_sTarget(pszTarget ? pszTarget : c_szEmpty)
{
}
//----------------------------------------------------------------------------
// Function: CMessageDlg::DoDataExchange
//
// DDX handler.
//----------------------------------------------------------------------------
VOID
CMessageDlg::DoDataExchange(
CDataExchange* pDX
) {
CBaseDialog::DoDataExchange(pDX);
}
BEGIN_MESSAGE_MAP(CMessageDlg, CBaseDialog)
END_MESSAGE_MAP()
DWORD CMessageDlg::m_dwHelpMap[] =
{
// IDC_DM_TO, HIDC_DM_TO,
// IDC_DM_MESSAGE, HIDC_DM_MESSAGE,
0,0
};
//----------------------------------------------------------------------------
// Function: CMessageDlg::OnInitDialog
//
// Performs dialog initialization.
//----------------------------------------------------------------------------
BOOL
CMessageDlg::OnInitDialog(
) {
CBaseDialog::OnInitDialog();
//
// Set the 'To' text to indicate who the message is going to
//
CString sText;
if (m_fUser) {
//
// We're sending to a client
//
AfxFormatString2(sText, IDS_DM_TO_USER_FORMAT, m_sUserName, m_sTarget);
}
else {
CString stTarget;
// Windows NT Bug : 285468
// Need to adjust for the local machine case. (if we are
// a local machine, then we will get a NULL name).
stTarget = m_sTarget;
if (stTarget.IsEmpty())
{
stTarget.LoadString(IDS_DM_LOCAL_MACHINE);
}
//
// We're sending to all RAS users in a domain or server
//
AfxFormatString1(
sText, IDS_DM_TO_SERVER_FORMAT,
stTarget
);
}
SetDlgItemText(IDC_DM_EDIT_TO, sText);
return FALSE;
}
//----------------------------------------------------------------------------
// Function: CMessageDlg::OnOK
//
//----------------------------------------------------------------------------
VOID
CMessageDlg::OnOK(
) {
AFX_MANAGE_STATE(AfxGetStaticModuleState());
CString sText;
DWORD err = ERROR_SUCCESS;
SPMprServerHandle sphMprServer;
USES_CONVERSION;
GetDlgItemText(IDC_DM_EDIT_MESSAGE, sText);
if (!sText.GetLength() &&
AfxMessageBox(IDS_ERR_NO_TEXT, MB_YESNO|MB_ICONQUESTION) != IDYES)
{
return;
}
if (m_fUser)
{
CWaitCursor wait;
// Need to get a connection to the server (to get the server handle)
err = ::MprAdminServerConnect(T2W((LPTSTR)(LPCTSTR)m_sServerName),
&sphMprServer);
if (err == ERROR_SUCCESS)
err = SendToClient(m_sServerName,
m_sTarget,
sphMprServer,
m_hConnection,
sText);
sphMprServer.Release();
}
else
{
err = SendToServer(m_sServerName, sText);
}
if (err == NERR_Success)
{
CBaseDialog::OnOK();
}
}
/*!--------------------------------------------------------------------------
CMessageDlg::SendToServer
-
Author: KennT
---------------------------------------------------------------------------*/
DWORD
CMessageDlg::SendToServer(
LPCTSTR pszServer,
LPCTSTR pszText,
BOOL* pbCancel
) {
AFX_MANAGE_STATE(AfxGetStaticModuleState());
DWORD dwErr = ERROR_SUCCESS;
CString sText;
TCHAR szServer[MAX_COMPUTERNAME_LENGTH+3];
DWORD i, dwTotal = 0, rc0Count = 0;
RAS_CONNECTION_0 *rc0Table = NULL, *prc0;
WCHAR wszServer[MAX_COMPUTERNAME_LENGTH+3];
HRESULT hr = hrOK;
SPMprServerHandle sphMprServer;
SPMprAdminBuffer spMprBuffer;
Assert(pszServer);
Assert(pszText);
COM_PROTECT_TRY
{
StrCpy(szServer, pszServer);
StrCpyWFromT(wszServer, pszServer);
//
// See if the router is installed on the machine;
//
if (!::MprAdminIsServiceRunning(wszServer))
{
goto Error;
}
//
// Connect to the server
//
{
CWaitCursor wait;
dwErr = ::MprAdminServerConnect(wszServer, &sphMprServer);
}
if (dwErr != NO_ERROR) {
TCHAR szText[2048];
FormatSystemError(HRESULT_FROM_WIN32(dwErr),
szText, DimensionOf(szText), 0,
FSEFLAG_ANYMESSAGE);
AfxFormatString2(sText, IDS_ERR_CONNECT_ERROR, szServer, szText);
AfxMessageBox(sText, MB_OK|MB_ICONINFORMATION);
goto Error;
}
//
// Retrieve an array of the connections on the server
//
{
CWaitCursor wait;
rc0Table = NULL;
dwErr = ::MprAdminConnectionEnum(
sphMprServer,
0,
(BYTE**)&spMprBuffer,
(DWORD)-1,
&rc0Count,
&dwTotal,
NULL
);
rc0Table = (RAS_CONNECTION_0 *) (BYTE *) spMprBuffer;
}
if (dwErr != NO_ERROR) {
TCHAR szText[2048];
FormatSystemError(HRESULT_FROM_WIN32(dwErr),
szText, DimensionOf(szText), 0,
FSEFLAG_ANYMESSAGE);
AfxFormatString2(sText, IDS_ERR_CONNENUM_ERROR, szServer, szText);
AfxMessageBox(sText, MB_OK|MB_ICONINFORMATION);
goto Error;
}
//
// For each one which is a client with RAS_FLAGS_MESSENGER_PRESENT,
// send the message
//
for (i = 0; i < rc0Count; i++)
{
prc0 = rc0Table + i;
if (prc0->dwInterfaceType != ROUTER_IF_TYPE_CLIENT ||
!lstrlenW(prc0->wszRemoteComputer) ||
!(prc0->dwConnectionFlags & RAS_FLAGS_MESSENGER_PRESENT)){
continue;
}
dwErr = SendToClient(pszServer,
prc0->wszRemoteComputer,
sphMprServer,
prc0->hConnection,
pszText);
if (!dwErr) { continue; }
AfxFormatString1(sText, IDS_PROMPT_SERVER_CONTINUE, szServer);
if (AfxMessageBox(sText, MB_YESNO|MB_ICONQUESTION) == IDNO)
{
if (pbCancel) { *pbCancel = TRUE; }
break;
}
}
COM_PROTECT_ERROR_LABEL;
}
COM_PROTECT_CATCH;
if ((dwErr == ERROR_SUCCESS) && !FHrSucceeded(hr))
{
// Assume that the failure was an out of memory
dwErr = ERROR_OUTOFMEMORY;
}
sphMprServer.Release();
return dwErr;
}
/*!--------------------------------------------------------------------------
CMessageDlg::SendToClient
-
Author: KennT
---------------------------------------------------------------------------*/
DWORD CMessageDlg::SendToClient(LPCTSTR pszServerName,
LPCTSTR pszTarget,
MPR_SERVER_HANDLE hMprServer,
HANDLE hConnection,
LPCTSTR pszText)
{
AFX_MANAGE_STATE(AfxGetStaticModuleState());
WCHAR *pswzText = NULL;
CString sText;
DWORD dwErr;
HKEY hkMachine = NULL;
HRESULT hr = hrOK;
RouterVersionInfo verInfo;
USES_CONVERSION;
Assert(pszTarget);
Assert(pszServerName);
Assert(hMprServer);
Assert(hConnection);
COM_PROTECT_TRY
{
// Windows NT Bug : 158746
// Note, if the target machine is NT5, then we can use the
// new APIs
// ------------------------------------------------------------
// setup a default of NT5
// ------------------------------------------------------------
verInfo.dwRouterVersion = 5;
dwErr = ConnectRegistry(pszServerName, &hkMachine);
hr = HRESULT_FROM_WIN32(dwErr);
if (FHrSucceeded(hr) && hkMachine)
{
QueryRouterVersionInfo(hkMachine, &verInfo);
DisconnectRegistry(hkMachine);
}
// For NT4, call the old NetMessageBufferSend
// ------------------------------------------------------------
if (verInfo.dwRouterVersion == 4)
{
CWaitCursor wait;
pswzText = StrDupWFromT(pszText);
dwErr = ::NetMessageBufferSend(
NULL,
T2W((LPTSTR) pszTarget),
NULL,
(BYTE *) pswzText,
CbStrLenW(pswzText));
}
else
{
// For NT5 and up, Use the MprAdminXXX api. This will
// work correctly for the Appletalk case
// --------------------------------------------------------
CWaitCursor wait;
dwErr = ::MprAdminSendUserMessage(hMprServer,
hConnection,
T2W((LPTSTR) pszText));
}
if (dwErr != ERROR_SUCCESS)
{
TCHAR szText[2048];
FormatSystemError(HRESULT_FROM_WIN32(dwErr),
szText, DimensionOf(szText), 0,
FSEFLAG_ANYMESSAGE);
AfxFormatString2(sText, IDS_ERR_SEND_FAILED, pszTarget, szText);
AfxMessageBox(sText, MB_OK|MB_ICONINFORMATION);
}
}
COM_PROTECT_CATCH;
delete pswzText;
if ((dwErr == ERROR_SUCCESS) && !FHrSucceeded(hr))
{
dwErr = ERROR_OUTOFMEMORY;
}
return dwErr;
}