windows-server-2003/public/sdk/inc/atl30/atlimpl.cpp

// This is a part of the Active Template Library.
// Copyright (C) 1996-1998 Microsoft Corporation
// All rights reserved.
//
// This source code is only intended as a supplement to the
// Active Template Library Reference and related
// electronic documentation provided with the library.
// See these sources for detailed information regarding the
// Active Template Library product.

#ifndef __ATLBASE_H__
        #error atlimpl.cpp requires atlbase.h to be included first
#endif

/////////////////////////////////////////////////////////////////////////////
// Minimize CRT
// Specify DllMain as EntryPoint
// Turn off exception handling
// Define _ATL_MIN_CRT
#ifdef _ATL_MIN_CRT
/////////////////////////////////////////////////////////////////////////////
// Startup Code

#if defined(_WINDLL) || defined(_USRDLL)

// Declare DllMain
extern "C" BOOL WINAPI DllMain(HANDLE hDllHandle, DWORD dwReason, LPVOID lpReserved);

extern "C" BOOL WINAPI _DllMainCRTStartup(HANDLE hDllHandle, DWORD dwReason, LPVOID lpReserved)
{
        return DllMain(hDllHandle, dwReason, lpReserved);
}

#else

// wWinMain is not defined in winbase.h.
extern "C" int WINAPI wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpCmdLine, int nShowCmd);

#define SPACECHAR   _T(' ')
#define DQUOTECHAR  _T('\"')


#ifdef _UNICODE
extern "C" void wWinMainCRTStartup()
#else // _UNICODE
extern "C" void WinMainCRTStartup()
#endif // _UNICODE
{
        LPTSTR lpszCommandLine = ::GetCommandLine();
        if(lpszCommandLine == NULL)
                ::ExitProcess((UINT)-1);

        // Skip past program name (first token in command line).
        // Check for and handle quoted program name.
        if(*lpszCommandLine == DQUOTECHAR)
        {
                // Scan, and skip over, subsequent characters until
                // another double-quote or a null is encountered.
                do
                {
                        lpszCommandLine = ::CharNext(lpszCommandLine);
                }
                while((*lpszCommandLine != DQUOTECHAR) && (*lpszCommandLine != _T('\0')));

                // If we stopped on a double-quote (usual case), skip over it.
                if(*lpszCommandLine == DQUOTECHAR)
                        lpszCommandLine = ::CharNext(lpszCommandLine);
        }
        else
        {
                while(*lpszCommandLine > SPACECHAR)
                        lpszCommandLine = ::CharNext(lpszCommandLine);
        }

        // Skip past any white space preceeding the second token.
        while(*lpszCommandLine && (*lpszCommandLine <= SPACECHAR))
                lpszCommandLine = ::CharNext(lpszCommandLine);

        STARTUPINFO StartupInfo;
        StartupInfo.dwFlags = 0;
        ::GetStartupInfo(&StartupInfo);

        int nRet = _tWinMain(::GetModuleHandle(NULL), NULL, lpszCommandLine,
                (StartupInfo.dwFlags & STARTF_USESHOWWINDOW) ?
                StartupInfo.wShowWindow : SW_SHOWDEFAULT);

        ::ExitProcess((UINT)nRet);
}

#endif // defined(_WINDLL) | defined(_USRDLL)

/////////////////////////////////////////////////////////////////////////////
// Heap Allocation

#ifndef _DEBUG

#ifndef _MERGE_PROXYSTUB
//rpcproxy.h does the same thing as this
int __cdecl _purecall()
{
        ::MessageBox(NULL, 
                _T("Runtime Error!\n\nR6025\n- pure virtual function call\n"), 
                _T("Microsoft Visual C+ ATL"), 
                MB_OK | MB_ICONERROR);
        ::ExitProcess(255);
        return 0;
}
#endif

#if !defined(_M_ALPHA) && !defined(_M_PPC)
//RISC always initializes floating point and always defines _fltused
extern "C" const int _fltused = 0;
#endif

static const int nExtraAlloc = 8;
static const int nOffsetBlock = nExtraAlloc/sizeof(HANDLE);

void* __cdecl malloc(size_t n)
{
        void* pv = NULL;
#ifndef _ATL_NO_MP_HEAP
        if (_Module.m_phHeaps == NULL)
#endif
        {
                pv = (HANDLE*) HeapAlloc(_Module.m_hHeap, 0, n);
        }
#ifndef _ATL_NO_MP_HEAP
        else
        {
                // overallocate to remember the heap handle
                int nHeap = _Module.m_nHeap++;
                HANDLE hHeap = _Module.m_phHeaps[nHeap & _Module.m_dwHeaps];
                HANDLE* pBlock = (HANDLE*) HeapAlloc(hHeap, 0, n + nExtraAlloc);
                if (pBlock != NULL)
                {
                        *pBlock = hHeap;
                        pv = (void*)(pBlock + nOffsetBlock);
                }
                else
                        pv = NULL;
        }
#endif
        return pv;
}

void* __cdecl calloc(size_t n, size_t s)
{
        return malloc(n*s);
}

#pragma prefast(push)
#pragma prefast(suppress:308, "prefast bug 538")
void* __cdecl realloc(void* p, size_t n)
{
        if (p == NULL)
                return malloc(n);
#ifndef _ATL_NO_MP_HEAP
        if (_Module.m_phHeaps == NULL)
#endif
                return HeapReAlloc(_Module.m_hHeap, 0, p, n);
#ifndef _ATL_NO_MP_HEAP
        else
        {
                HANDLE* pHeap = ((HANDLE*)p)-nOffsetBlock;
                pHeap = (HANDLE*) HeapReAlloc(*pHeap, 0, pHeap, n + nExtraAlloc);
                return (pHeap != NULL) ? pHeap + nOffsetBlock : NULL;
        }
#endif
}
#pragma prefast(pop)

void __cdecl free(void* p)
{
        if (p == NULL)
                return;
#ifndef _ATL_NO_MP_HEAP
        if (_Module.m_phHeaps == NULL)
#endif
                HeapFree(_Module.m_hHeap, 0, p);
#ifndef _ATL_NO_MP_HEAP
        else
        {
                HANDLE* pHeap = ((HANDLE*)p)-nOffsetBlock;
                HeapFree(*pHeap, 0, pHeap);
        }
#endif
}

void* __cdecl operator new(size_t n)
{
        return malloc(n);
}

void __cdecl operator delete(void* p)
{
        free(p);
}

#endif  //_DEBUG

#endif //_ATL_MIN_CRT