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.
 
 
 
 
 
 

308 lines
10 KiB

/**MOD+**********************************************************************/
/* Module: tsaxmain.cpp */
/* */
/* Purpose: Implementation of DLL Exports. Header for this module will */
/* be generated in the respective build directory. */
/* */
/* Copyright(C) Microsoft Corporation 1998 */
/* */
/****************************************************************************/
#include "stdafx.h"
#include "atlwarn.h"
BEGIN_EXTERN_C
#define TRC_GROUP TRC_GROUP_UI
#define TRC_FILE "tsaxmain"
#include <atrcapi.h>
END_EXTERN_C
#include "tsaxiids.h"
#include "initguid.h"
#include "mstsax.h"
#ifndef OS_WINCE
#include "mstsax_i.c"
#endif
#include "mstscax.h"
#include "tsaxmod.h"
//
// Version number (property returns this)
//
#ifndef OS_WINCE
#include "ntverp.h"
#else
#include "ceconfig.h" //get build #
#endif
//Unicode wrapper
#include "wraputl.h"
/****************************************************************************/
/* Module object */
/****************************************************************************/
CMsTscAxModule _Module;
/****************************************************************************/
/* Object map */
/****************************************************************************/
BEGIN_OBJECT_MAP(ObjectMap)
OBJECT_ENTRY(CLSID_MsRdpClient3, CMsTscAx)
OBJECT_ENTRY(CLSID_MsRdpClient2, CMsTscAx)
OBJECT_ENTRY(CLSID_MsRdpClient, CMsTscAx)
OBJECT_ENTRY(CLSID_MsTscAx, CMsTscAx)
END_OBJECT_MAP()
#ifdef ECP_TIMEBOMB
//
// Return's true if timebomb test passed otherwise puts up warning
// UI and return's FALSE
//
BOOL CheckTimeBomb()
{
SYSTEMTIME lclTime;
FILETIME lclFileTime;
GetLocalTime(&lclTime);
DCBOOL bTimeBombOk = TRUE;
//
// Simply check that the local date is less than June 30, 2000
//
if(lclTime.wYear < ECP_TIMEBOMB_YEAR)
{
return TRUE;
}
else if (lclTime.wYear == ECP_TIMEBOMB_YEAR)
{
if(lclTime.wMonth < ECP_TIMEBOMB_MONTH)
{
return TRUE;
}
else if(lclTime.wMonth == ECP_TIMEBOMB_MONTH)
{
if(lclTime.wDay < ECP_TIMEBOMB_DAY)
{
return TRUE;
}
}
}
DCTCHAR timeBombStr[256];
if (LoadString(_Module.GetModuleInstance(),
TIMEBOMB_EXPIRED_STR,
timeBombStr,
SIZEOF_TCHARBUFFER(timeBombStr)) != 0)
{
MessageBox(NULL, timeBombStr, NULL,
MB_ICONERROR | MB_OK);
}
//
// If we reach this point the timebomb should trigger
// so put up a messagebox and return FALSE
// so the calling code can disable functionality
//
return FALSE;
}
#endif
#ifdef UNIWRAP
//It's ok to have a global unicode wrapper
//class. All it does is sets up the g_bRunningOnNT
//flag so it can be shared by multiple instances
//also it is only used from DllMain so there
//are no problems with re-entrancy
CUnicodeWrapper g_uwrp;
#endif
#ifdef OS_WINCE
DECLARE_TRACKER_VARS();
#endif
/**PROC+*********************************************************************/
/* Name: DllMain */
/* */
/* Purpose: DLL entry point */
/* */
/**PROC-*********************************************************************/
extern "C"
#ifndef OS_WINCE
BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
#else
BOOL WINAPI DllMain(HANDLE hInstance, DWORD dwReason, LPVOID /*lpReserved*/)
#endif
{
if (dwReason == DLL_PROCESS_ATTACH)
{
#ifdef UNIWRAP
//UNICODE Wrapper intialization has to happen first,
//before anything ELSE. Even DC_BEGIN_FN, which does tracing
g_uwrp.InitializeWrappers();
#endif
TSRNG_Initialize();
CO_StaticInit((HINSTANCE)hInstance);
_Module.Init(ObjectMap, (HINSTANCE)hInstance);
#if ((!defined (OS_WINCE)) || (_WIN32_WCE >= 300) )
DisableThreadLibraryCalls((HINSTANCE)hInstance);
#endif
#ifdef OS_WINCE
CEInitialize();
g_CEConfig = CEGetConfigType(&g_CEUseScanCodes);
#endif
}
else if (dwReason == DLL_PROCESS_DETACH) {
_Module.Term();
CO_StaticTerm();
TSRNG_Shutdown();
#ifdef UNIWRAP
g_uwrp.CleanupWrappers();
#endif
}
return TRUE; // ok
}
/**PROC+*********************************************************************/
/* Name: DllCanUnloadNow */
/* */
/* Purpose: Used to determine whether the DLL can be unloaded by OLE */
/* */
/**PROC-*********************************************************************/
STDAPI DllCanUnloadNow(void)
{
return (_Module.GetLockCount()==0) ? S_OK : S_FALSE;
}
/**PROC+*********************************************************************/
/* Name: DllGetClassObject */
/* */
/* Purpose: Returns a class factory to create an object of the requested */
/* type */
/* */
/**PROC-*********************************************************************/
STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
{
#ifdef ECP_TIMEBOMB
if(!CheckTimeBomb())
{
//
// Timebomb failed, bail out with an error message
//
return E_OUTOFMEMORY;
}
#endif
return _Module.GetClassObject(rclsid, riid, ppv);
}
/**PROC+*********************************************************************/
/* Name: DllRegisterServer */
/* */
/* Purpose: DllRegisterServer - Adds entries to the system registry */
/* */
/**PROC-*********************************************************************/
STDAPI DllRegisterServer(void)
{
// registers object, typelib and all interfaces in typelib
return _Module.RegisterServer(TRUE);
}
/**PROC+*********************************************************************/
/* Name: DllUnregisterServer */
/* */
/* Purpose: DllUnregisterServer - Removes entries from the system registry*/
/* */
/**PROC-*********************************************************************/
STDAPI DllUnregisterServer(void)
{
return _Module.UnregisterServer();
}
/**PROC+*********************************************************************/
/* Name: DllGetTscCtlVer */
/* */
/* Purpose: Returns version of the tsc control */
/* */
/**PROC-*********************************************************************/
STDAPI_(DWORD) DllGetTscCtlVer(void)
{
#ifndef OS_WINCE
return VER_PRODUCTVERSION_DW;
#else
return CE_TSC_BUILDNO;
#endif
}
#ifndef OS_WINCE
#ifdef CRTREPORT_DEBUG_HACK
/**PROC+*********************************************************************/
/* Name: _CrtDbgReport */
/* */
/* Purpose: Redirect all debug reporting to our tracing functions */
/* */
/**PROC-*********************************************************************/
extern "C"
_CRTIMP int __cdecl _CrtDbgReport(int nRptType,
const char * szFile,
int nLine,
const char * szModule,
const char * szFormat,
...)
{
static CHAR bigBuf[2048];
va_list vargs;
HRESULT hr;
DC_BEGIN_FN("AtlTraceXXX");
va_start(vargs, szFormat);
hr = StringCchVPrintfA(bigBuf, sizeof(bigBuf), szFormat, vargs);
va_end( vargs );
#ifdef OS_WINCE
#ifndef _CRT_ASSERT
#define _CRT_ASSERT 2
#endif
#endif
if (_CRT_ASSERT == nRptType)
{
#ifdef UNICODE
TRC_ABORT((TB,_T("AtlAssert. File:%S line:%d - %S"), szFile,
nLine, bigBuf));
#else
TRC_ABORT((TB,_T("AtlAssert. File:%s line:%d - %s"), szFile,
nLine, bigBuf));
#endif
}
else
{
#ifdef UNICODE
TRC_ERR((TB,_T("AtlTrace. File:%S line:%d - %S"), szFile,
nLine, bigBuf));
#else
TRC_ERR((TB,_T("AtlTrace. File:%s line:%d - %s"), szFile,
nLine, bigBuf));
#endif
}
DC_END_FN();
return 0;
}
#endif
#endif //OS_WINCE