windows-nt/Source/XPSP1/NT/inetsrv/msmq/sdk/samples/mqapitst/mqapitst.cpp
2020-09-26 16:20:57 +08:00

291 lines
7.2 KiB
C++

// Mqapitest.cpp : Defines the class behaviors for the application.
//
//=--------------------------------------------------------------------------=
// Copyright 1997-1999 Microsoft Corporation. All Rights Reserved.
//
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
// PARTICULAR PURPOSE.
//=--------------------------------------------------------------------------=
#include "stdafx.h"
#include "MQApitst.h"
#include "MainFrm.h"
#include "testDoc.h"
#include "testView.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif
/////////////////////////////////////////////////////////////////////////////
// CTestApp
BEGIN_MESSAGE_MAP(CTestApp, CWinApp)
//{{AFX_MSG_MAP(CTestApp)
ON_COMMAND(ID_APP_ABOUT, OnAppAbout)
// NOTE - the ClassWizard will add and remove mapping macros here.
// DO NOT EDIT what you see in these blocks of generated code!
//}}AFX_MSG_MAP
// Standard file based document commands
ON_COMMAND(ID_FILE_NEW, CWinApp::OnFileNew)
ON_COMMAND(ID_FILE_OPEN, CWinApp::OnFileOpen)
END_MESSAGE_MAP()
/////////////////////////////////////////////////////////////////////////////
// CTestApp construction
CTestApp::CTestApp()
{
// TODO: add construction code here,
// Place all significant initialization in InitInstance
}
/////////////////////////////////////////////////////////////////////////////
// The one and only CTestApp object
CTestApp theApp;
/////////////////////////////////////////////////////////////////////////////
// CTestApp initialization
BOOL CTestApp::InitInstance()
{
//
// This application should only be used when DS is enabled.
// if we are in DS-disabled mode we should stop execution.
//
if (!IsDsEnabledLocaly())
{
::MessageBox(NULL,"This application is not designed to run on DS-disabled mode configuration",NULL,MB_OK);
exit(0);
}
//
// Standard initialization
// If you are not using these features and wish to reduce the size
// of your final executable, you should remove from the following
// the specific initialization routines you do not need.
#ifdef _AFXDLL
Enable3dControls(); // Call this when using MFC in a shared DLL
#else
Enable3dControlsStatic(); // Call this when linking to MFC statically
#endif
LoadStdProfileSettings(); // Load standard INI file options (including MRU)
// Register the application's document templates. Document templates
// serve as the connection between documents, frame windows and views.
CSingleDocTemplate* pDocTemplate;
pDocTemplate = new CSingleDocTemplate(
IDR_MAINFRAME,
RUNTIME_CLASS(CTestDoc),
RUNTIME_CLASS(CMainFrame), // main SDI frame window
RUNTIME_CLASS(CTestView));
AddDocTemplate(pDocTemplate);
// Parse command line for standard shell commands, DDE, file open
CCommandLineInfo cmdInfo;
ParseCommandLine(cmdInfo);
// Dispatch commands specified on the command line
if (!ProcessShellCommand(cmdInfo))
return FALSE;
return TRUE;
}
/////////////////////////////////////////////////////////////////////////////
// CAboutDlg dialog used for App About
class CAboutDlg : public CDialog
{
public:
CAboutDlg();
// Dialog Data
//{{AFX_DATA(CAboutDlg)
enum { IDD = IDD_ABOUTBOX };
//}}AFX_DATA
// ClassWizard generated virtual function overrides
//{{AFX_VIRTUAL(CAboutDlg)
protected:
virtual void DoDataExchange(CDataExchange* pDX); // DDX/DDV support
//}}AFX_VIRTUAL
// Implementation
protected:
//{{AFX_MSG(CAboutDlg)
// No message handlers
//}}AFX_MSG
DECLARE_MESSAGE_MAP()
};
CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
{
//{{AFX_DATA_INIT(CAboutDlg)
//}}AFX_DATA_INIT
}
void CAboutDlg::DoDataExchange(CDataExchange* pDX)
{
CDialog::DoDataExchange(pDX);
//{{AFX_DATA_MAP(CAboutDlg)
//}}AFX_DATA_MAP
}
BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
//{{AFX_MSG_MAP(CAboutDlg)
// No message handlers
//}}AFX_MSG_MAP
END_MESSAGE_MAP()
// App command to run the dialog
void CTestApp::OnAppAbout()
{
CAboutDlg aboutDlg;
aboutDlg.DoModal();
}
/////////////////////////////////////////////////////////////////////////////
// CTestApp commands
//
// Pointer to hold the main window pointer.
//
CWnd* pMainView;
extern "C" void PrintToScreen(const TCHAR * Format, ...)
{
TCHAR szText[500];
va_list l;
static len = 0;
//Format the string
va_start(l, Format);
#ifdef UNICODE
vswprintf(szText, Format, l);
#else
vsprintf(szText, Format, l);
#endif
CEdit& rfMainEdit = ((CTestView*)pMainView)->GetEditCtrl();
int i = rfMainEdit.GetLineCount();
((CTestView*)pMainView)->GetEditCtrl().SetSel(INT_MAX,INT_MAX);
((CTestView*)pMainView)->GetEditCtrl().ReplaceSel(szText);
((CTestView*)pMainView)->GetEditCtrl().ReplaceSel(TEXT("\r\n"));
len += lstrlen(szText) + 2;
((CTestView*)pMainView)->GetDocument()->SetModifiedFlag(FALSE);
}
BOOL CTestApp::IsDsEnabledLocaly()
/*++
Routine Description:
The rutine checked if the coomputer is DS -enabled/disabled
Arguments:
None
Return Value:
TRUE - DS - enabled.
FALSE - DS - disabled.
--*/
{
MQPRIVATEPROPS PrivateProps;
QMPROPID aPropId[MAX_VAR];
MQPROPVARIANT aPropVar[MAX_VAR];
DWORD cProp;
HRESULT hr;
//
// Prepare DS-enabled property.
//
cProp = 0;
aPropId[cProp] = PROPID_PC_DS_ENABLED;
aPropVar[cProp].vt = VT_NULL;
++cProp;
//
// Create a PRIVATEPROPS structure.
//
PrivateProps.cProp = cProp;
PrivateProps.aPropID = aPropId;
PrivateProps.aPropVar = aPropVar;
PrivateProps.aStatus = NULL;
//
// Retrieve the information.
//
//
// This code detect DS connection.
// This code is designed to allow the sample to compile both on NT4 and Windows 2000.
// The MQGetPrivateComputerInformation() function can be called directly
// on Windows 2000.
//
HINSTANCE hMqrtLibrary = GetModuleHandle(TEXT("mqrt.dll"));
ASSERT(hMqrtLibrary != NULL);
typedef HRESULT (APIENTRY *MQGetPrivateComputerInformation_ROUTINE)(LPCWSTR , MQPRIVATEPROPS*);
MQGetPrivateComputerInformation_ROUTINE pfMQGetPrivateComputerInformation =
(MQGetPrivateComputerInformation_ROUTINE)GetProcAddress(hMqrtLibrary,
"MQGetPrivateComputerInformation");
if(pfMQGetPrivateComputerInformation == NULL)
{
//
// There is no entry point in the dll matching to this routine
// it must be an old version of mqrt.dll -> product one.
// It will be OK to handle this case as a case of DS-enabled mode.
//
return TRUE;
}
hr = pfMQGetPrivateComputerInformation(
NULL,
&PrivateProps);
if(FAILED(hr))
{
//
// we were not able to determine if DS is enabled or disabled
// notify the user and assume the worst case - (i.e. we are DS-disasbled).
//
AfxMessageBox("Unable to detect DS connection");
return FALSE;
}
if(PrivateProps.aPropVar[0].boolVal == 0)
{
//
// DS-disabled.
//
return FALSE;
}
return TRUE;
}