// This is a part of the Active Template Library. // Copyright (C) 1996-2001 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 __ATLSTENCIL_H__ #define __ATLSTENCIL_H__ #pragma once #ifdef _ATL_MIN_CRT #error "_ATL_MIN_CRT cannot be used with atlstencil.h" #endif #include #include #include #ifndef ATL_NO_MLANG #include #endif #ifndef _ATL_NO_DEFAULT_LIBS #pragma comment(lib, "shlwapi.lib") #endif // !_ATL_NO_DEFAULT_LIBS #pragma warning( push ) #pragma warning( disable: 4127 ) #pragma warning(disable: 4511) // copy constructor could not be generated #pragma warning(disable: 4512) // assignment operator could not be generated #pragma warning(disable: 4702) // assignment operator could not be generated namespace ATL { // Token types // These tags are token tags for the standard tag replacer implementation extern __declspec(selectany) const DWORD STENCIL_TEXTTAG = 0x00000000; extern __declspec(selectany) const DWORD STENCIL_REPLACEMENT = 0x00000001; extern __declspec(selectany) const DWORD STENCIL_ITERATORSTART = 0x00000002; extern __declspec(selectany) const DWORD STENCIL_ITERATOREND = 0x00000003; extern __declspec(selectany) const DWORD STENCIL_CONDITIONALSTART = 0x00000004; extern __declspec(selectany) const DWORD STENCIL_CONDITIONALELSE = 0x00000005; extern __declspec(selectany) const DWORD STENCIL_CONDITIONALEND = 0x00000006; extern __declspec(selectany) const DWORD STENCIL_STENCILINCLUDE = 0x00000007; extern __declspec(selectany) const DWORD STENCIL_STATICINCLUDE = 0x00000008; extern __declspec(selectany) const DWORD STENCIL_LOCALE = 0x00000009; extern __declspec(selectany) const DWORD STENCIL_CODEPAGE = 0x0000000a; // The base for user defined token types extern __declspec(selectany) const DWORD STENCIL_USER_TOKEN_BASE = 0x00001000; // Symbols to use in error handling in the stencil processor #define STENCIL_INVALIDINDEX 0xFFFFFFFF #define STENCIL_INVALIDOFFSET 0xFFFFFFFF // error codes #define STENCIL_SUCCESS HTTP_SUCCESS #define STENCL_FAIL HTTP_FAIL #define STENCIL_BASIC_MAP 0 #define STENCIL_ATTR_MAP 1 #ifndef ATL_MAX_METHOD_NAME_LEN #define ATL_MAX_METHOD_NAME_LEN 64 #endif #ifndef ATL_MAX_BLOCK_STACK #define ATL_MAX_BLOCK_STACK 128 #endif template struct CTagReplacerMethodsEx { typedef HTTP_CODE (TBase::*REPLACE_FUNC)(); typedef HTTP_CODE (TBase::*REPLACE_FUNC_EX)(T*); typedef HTTP_CODE (TBase::*PARSE_FUNC)(IAtlMemMgr *, LPCSTR, T**); typedef HTTP_CODE (TBase::*REPLACE_FUNC_EX_V)(void *); typedef HTTP_CODE (TBase::*PARSE_FUNC_V)(IAtlMemMgr *, LPCSTR, void**); static REPLACE_FUNC_EX_V CheckRepl(REPLACE_FUNC p) throw() { return (REPLACE_FUNC_EX_V) p; } static REPLACE_FUNC_EX_V CheckReplEx(REPLACE_FUNC_EX p) throw() { return (REPLACE_FUNC_EX_V) p; } static PARSE_FUNC_V CheckParse(PARSE_FUNC p) throw() { return (PARSE_FUNC_V) p; } }; template struct CTagReplacerMethods { union { HTTP_CODE (TBase::*pfnMethodEx)(void *); HTTP_CODE (TBase::*pfnMethod)(); }; HTTP_CODE (TBase::*pfnParse)(IAtlMemMgr *pMemMgr, LPCSTR, void **); }; #define REPLACEMENT_ENTRY_DEFAULT 0 #define REPLACEMENT_ENTRY_ARGS 1 template struct CTagReplacerMethodEntry { int nType; // REPLACEMENT_ENTRY_* LPCSTR szMethodName; CTagReplacerMethods Methods; }; #define BEGIN_REPLACEMENT_METHOD_MAP(className)\ public:\ void GetReplacementMethodMap(const ATL::CTagReplacerMethodEntry ** ppOut) const throw()\ {\ typedef className __className;\ static const ATL::CTagReplacerMethodEntry methods[] = { #define REPLACEMENT_METHOD_ENTRY(methodName, methodFunc)\ { 0, methodName, { ATL::CTagReplacerMethodsEx<__className, void>::CheckRepl(methodFunc), NULL } }, #define REPLACEMENT_METHOD_ENTRY_EX(methodName, methodFunc, paramType, parseFunc)\ { 1, methodName, { ATL::CTagReplacerMethodsEx<__className, paramType>::CheckReplEx(methodFunc), ATL::CTagReplacerMethodsEx<__className, paramType>::CheckParse(parseFunc) } }, #define REPLACEMENT_METHOD_ENTRY_EX_STR(methodName, methodFunc) \ { 1, methodName, { ATL::CTagReplacerMethodsEx<__className, char>::CheckReplEx(methodFunc), ATL::CTagReplacerMethodsEx<__className, char>::CheckParse(DefaultParseString) } }, #define END_REPLACEMENT_METHOD_MAP()\ { 0, NULL, NULL } };\ *ppOut = methods;\ } #define BEGIN_ATTR_REPLACEMENT_METHOD_MAP(className)\ public:\ void GetAttrReplacementMethodMap(const CTagReplacerMethodEntry ** ppOut) const throw()\ {\ typedef className __className;\ static const ATL::CTagReplacerMethodEntry methods[] = { #define END_ATTR_REPLACEMENT_METHOD_MAP()\ { NULL, NULL, NULL } };\ *ppOut = methods;\ } template class ITagReplacerImpl : public ITagReplacer { protected: IWriteStream *m_pStream; public: typedef HTTP_CODE (T::*REPLACEMENT_METHOD)(); typedef HTTP_CODE (T::*REPLACEMENT_METHOD_EX)(void *pvParam); ITagReplacerImpl() throw() :m_pStream(NULL) { } IWriteStream *SetStream(IWriteStream *pStream) throw() { IWriteStream *pRetStream = m_pStream; m_pStream = pStream; return pRetStream; } // Looks up the replacement method offset. Optionally, it will // look up the replacement method and object offset of an alternate // tag replacer. HTTP_CODE FindReplacementOffset( LPCSTR szMethodName, DWORD *pdwMethodOffset, LPCSTR szHandlerName, DWORD *pdwHandlerOffset, DWORD *pdwMap, void **ppvParam, IAtlMemMgr *pMemMgr) throw(...) { // we at least have to be looking up a method offset if (!pdwMethodOffset || !szMethodName) return HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); *pdwMethodOffset = STENCIL_INVALIDOFFSET; HTTP_CODE hcErr = HTTP_FAIL; T *pT = static_cast(this); char szName[ATL_MAX_METHOD_NAME_LEN]; // if a handler name was supplied, we will try to // find a different object to handle the method if (szHandlerName && *szHandlerName) { if (!pdwHandlerOffset) return HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); hcErr = pT->GetHandlerOffset(szHandlerName, pdwHandlerOffset); // got the alternate handler, now look up the method offset on // the handler. if (!hcErr) { CComPtr spAltTagReplacer; hcErr = pT->GetReplacementObject(*pdwHandlerOffset, &spAltTagReplacer); if (!hcErr) hcErr = spAltTagReplacer->FindReplacementOffset(szMethodName, pdwMethodOffset, NULL, NULL, pdwMap, ppvParam, pMemMgr); return hcErr; } else return hcErr; } strcpy(szName, szMethodName); // check for params char *szLeftPar = strchr(szName, '('); if (szLeftPar) { *szLeftPar = '\0'; szLeftPar++; char *szRightPar = strchr(szLeftPar, ')'); if (!szRightPar) return HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); *szRightPar = '\0'; szMethodName = szName; } // No handler name is specified, so we look up the method name in // T's replacement method map const CTagReplacerMethodEntry *pEntry; pT->GetReplacementMethodMap(&pEntry); hcErr = FindReplacementOffsetInMap(szMethodName, pdwMethodOffset, pEntry); if (hcErr != HTTP_SUCCESS) { pT->GetAttrReplacementMethodMap(&pEntry); hcErr = FindReplacementOffsetInMap(szMethodName, pdwMethodOffset, pEntry); if (hcErr == HTTP_SUCCESS) *pdwMap = STENCIL_ATTR_MAP; } else { *pdwMap = STENCIL_BASIC_MAP; } // This assert will be triggered if arguments are passed to a replacement method that doesn't handle them ATLASSERT( szLeftPar == NULL || (szLeftPar != NULL && (pEntry != NULL && pEntry[*pdwMethodOffset].Methods.pfnParse) != NULL) ); if (hcErr == HTTP_SUCCESS && pEntry && pEntry[*pdwMethodOffset].Methods.pfnParse) hcErr = (pT->*pEntry[*pdwMethodOffset].Methods.pfnParse)(pMemMgr, szLeftPar, ppvParam); return hcErr; } HTTP_CODE FindReplacementOffsetInMap( LPCSTR szMethodName, LPDWORD pdwMethodOffset, const CTagReplacerMethodEntry *pEntry) throw() { if (pEntry == NULL) return HTTP_FAIL; const CTagReplacerMethodEntry *pEntryHead = pEntry; while (pEntry->szMethodName) { if (strcmp(pEntry->szMethodName, szMethodName) == 0) { if (pEntry->Methods.pfnMethod) { *pdwMethodOffset = (DWORD)(pEntry-pEntryHead); return HTTP_SUCCESS; } } pEntry++; } return HTTP_FAIL; } // Used to render a single replacement tag into a stream. // Looks up a pointer to a member function in user code by offseting into the users // replacement map. Much faster than the other overload of this function since // no string compares are performed. HTTP_CODE RenderReplacement(DWORD dwFnOffset, DWORD dwObjOffset, DWORD dwMap, void *pvParam) throw(...) { HTTP_CODE hcErr = HTTP_FAIL; T *pT = static_cast(this); // if we were not passed an object offset, then we assume // that the function at dwFnOffset is in T's replacement // map if (dwObjOffset == STENCIL_INVALIDOFFSET) { // call a function in T's replacement map ATLASSERT(dwFnOffset != STENCIL_INVALIDOFFSET); const CTagReplacerMethodEntry *pEntry; if (dwMap == STENCIL_BASIC_MAP) pT->GetReplacementMethodMap(&pEntry); else pT->GetAttrReplacementMethodMap(&pEntry); if (pEntry) { if (pEntry[dwFnOffset].nType == REPLACEMENT_ENTRY_DEFAULT) { REPLACEMENT_METHOD pfn = NULL; pfn = pEntry[dwFnOffset].Methods.pfnMethod; ATLASSERT(pfn); if (pfn) { hcErr = (pT->*pfn)(); } } else if (pEntry[dwFnOffset].nType == REPLACEMENT_ENTRY_ARGS) { REPLACEMENT_METHOD_EX pfn = NULL; pfn = pEntry[dwFnOffset].Methods.pfnMethodEx; ATLASSERT(pfn); if (pfn) { hcErr = (pT->*pfn)(pvParam); } } else { // unknown entry type ATLASSERT(FALSE); } } } else { // otherwise, we were passed an object offset. The object // offset is a dword ID that T can use to look up the // ITagReplacer* of a tag replacer that will render this // replacement. CComPtr spAltReplacer = NULL; if (!pT->GetReplacementObject(dwObjOffset, &spAltReplacer)) { spAltReplacer->SetStream(m_pStream); hcErr = spAltReplacer->RenderReplacement(dwFnOffset, STENCIL_INVALIDOFFSET, dwMap, pvParam); } } return hcErr; } // Default GetHandlerOffset, does nothing HTTP_CODE GetHandlerOffset(LPCSTR /*szHandlerName*/, DWORD* pdwOffset) throw() { if (pdwOffset) *pdwOffset = 0; return HTTP_FAIL; } // Default GetReplacementObject, does nothing HTTP_CODE GetReplacementObject(DWORD /*dwObjOffset*/, ITagReplacer **ppReplacer) throw() { if (ppReplacer) *ppReplacer = NULL; return HTTP_FAIL; } void GetReplacementMethodMap(const CTagReplacerMethodEntry ** ppOut) const throw() { static const CTagReplacerMethodEntry methods[] = { { NULL, NULL } }; *ppOut = methods; } void GetAttrReplacementMethodMap(const CTagReplacerMethodEntry **ppOut) const throw() { static const CTagReplacerMethodEntry methods[] = { { NULL, NULL } }; *ppOut = methods; } HRESULT GetContext(REFIID, void**) throw() { return E_NOINTERFACE; } HTTP_CODE FreeParam(void * /*pvParam*/) throw() { return HTTP_SUCCESS; } HTTP_CODE DefaultParseString(IAtlMemMgr *pMemMgr, LPCSTR szParams, char **ppParam) throw(...) { ATLASSERT(szParams); size_t nLen = strlen(szParams); if (nLen) { nLen++; *ppParam = (char *) pMemMgr->Allocate(nLen); if (*ppParam) memcpy(*ppParam, szParams, nLen); else return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseUChar(IAtlMemMgr *pMemMgr, LPCSTR szParams, unsigned char **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (unsigned char *) pMemMgr->Allocate(sizeof(unsigned char)); if (*ppParam) { char *szEnd; **ppParam = (unsigned char) strtoul(szParams, &szEnd, 10); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseShort(IAtlMemMgr *pMemMgr, LPCSTR szParams, short **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (short *) pMemMgr->Allocate(sizeof(short)); if (*ppParam) { **ppParam = (short)atoi(szParams); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseUShort(IAtlMemMgr *pMemMgr, LPCSTR szParams, unsigned short **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (unsigned short *) pMemMgr->Allocate(sizeof(short)); if (*ppParam) { char *szEnd; **ppParam = (unsigned short) strtoul(szParams, &szEnd, 10); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseInt(IAtlMemMgr *pMemMgr, LPCSTR szParams, int **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (int *) pMemMgr->Allocate(sizeof(int)); if (*ppParam) { **ppParam = atoi(szParams); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseUInt(IAtlMemMgr *pMemMgr, LPCSTR szParams, unsigned int **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (unsigned int *) pMemMgr->Allocate(sizeof(unsigned int)); if (*ppParam) { char *szEnd; **ppParam = strtoul(szParams, &szEnd, 10); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseInt64(IAtlMemMgr *pMemMgr, LPCSTR szParams, __int64 **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (__int64 *) pMemMgr->Allocate(sizeof(__int64)); if (*ppParam) { **ppParam = _atoi64(szParams); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseUInt64(IAtlMemMgr *pMemMgr, LPCSTR szParams, unsigned __int64 **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (unsigned __int64 *) pMemMgr->Allocate(sizeof(unsigned __int64)); if (*ppParam) { char *szEnd; **ppParam = _strtoui64(szParams, &szEnd, 10); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseBool(IAtlMemMgr *pMemMgr, LPCSTR szParams, bool **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (bool *) pMemMgr->Allocate(sizeof(bool)); if (*ppParam) { if (!_strnicmp(szParams, "true", sizeof("true")-sizeof('\0'))) **ppParam = true; else **ppParam = false; } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseDouble(IAtlMemMgr *pMemMgr, LPCSTR szParams, double **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (double *) pMemMgr->Allocate(sizeof(double)); if (*ppParam) { **ppParam = atof(szParams); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } HTTP_CODE DefaultParseFloat(IAtlMemMgr *pMemMgr, LPCSTR szParams, float **ppParam) throw(...) { ATLASSERT(szParams); *ppParam = (float *) pMemMgr->Allocate(sizeof(float)); if (*ppParam) { **ppParam = (float) atof(szParams); } else { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); } return HTTP_SUCCESS; } }; inline LPSTR SkipSpace(LPSTR sz, WORD nCodePage) { while (_istspace(*sz)) sz = CharNextExA(nCodePage, sz, 0); return sz; } inline LPSTR RSkipSpace(LPSTR pStart, LPSTR sz, WORD nCodePage) throw() { while (_istspace(*sz)) sz = CharPrevExA(nCodePage, pStart, sz, 0); return sz; } // // StencilToken // The stencil class will create an array of these tokens during the parse // phase and use them during rendering to render the stencil struct StencilToken { LPSTR pStart; // Start of fragment to be rendered LPSTR pEnd; // End of fragment to be rendered DWORD type; // Type of token DWORD dwFnOffset; // Offset into the replacement map for the handler function. DWORD dwMap; DWORD dwObjOffset; // An identifier for the caller to use in identifiying the // object that will render this token. CHAR szHandlerName[ATL_MAX_HANDLER_NAME_LEN + 1]; // Name of handler object. CHAR szMethodName[ATL_MAX_METHOD_NAME_LEN + 1]; // Name of handler method. DWORD dwLoopIndex; // Offset into array of StencilTokens of the other loop tag DWORD_PTR dwData; BOOL bDynamicAlloc; }; // Specialization of CElementTraitsBase so you can put a StencilToken safely in // a collection object. template<> class CElementTraits< StencilToken > : public CElementTraitsBase< StencilToken > { public: static ULONG Hash( INARGTYPE t ) throw() { return( ULONG( ULONG_PTR( &t ) ) ); } static bool CompareElements( INARGTYPE element1, INARGTYPE element2 ) throw() { return( (element1.pStart == element2.pStart) && (element1.pEnd == element2.pEnd) ); } static int CompareElementsOrdered( INARGTYPE element1, INARGTYPE element2 ) throw() { if( element1.pStart < element2.pStart ) { return( -1 ); } else if( CompareElements(element1,element2) ) { return( 0 ); } else { ATLASSERT( element1.pStart > element2.pStart ); return( 1 ); } } }; // // Class CStencil // The CStencil class is used to map in a stencil from a file or resource // and parse the stencil into an array of StencilTokens. We then render // the stencil from the array of tokens. This class's parse and render // functions depend on an IReplacementHandlerLookup interface pointer to be // passed so it can retrieve the IReplacementHandler interface pointer of the // handler object that will be called to render replacement tags class CStencil : public IMemoryCacheClient { private: LPSTR m_pBufferStart; // Beginning of CHAR buffer that holds the stencil. // For mapped files this is the beginning of the mapping. LPSTR m_pBufferEnd; // End of CHAR buffer that holds the stencil. CAtlArray m_arrTokens; //An array of tokens. FILETIME m_ftLastModified; // Last modified time (0 for resource) FILETIME m_ftLastChecked; // Last time we retrieved last modified time (0 for resource) HCACHEITEM m_hCacheItem; WORD m_nCodePage; BOOL m_bUseLocaleACP; char m_szDllPath[MAX_PATH+1]; char m_szHandlerName[ATL_MAX_HANDLER_NAME_LEN+1]; // Room for the path, the handler // the '/' and the '\0' #ifdef ATL_DEBUG_STENCILS struct ParseError { char m_szError[256]; LPCSTR m_szPosition; LPCSTR m_szStartLine; LPCSTR m_szEndLine; int m_nLineNumber; bool operator==(const ParseError& that) const throw() { return (m_nLineNumber == that.m_nLineNumber); } }; CSimpleArray m_Errors; class CParseErrorProvider : public ITagReplacerImpl, public CComObjectRootEx { public: BEGIN_COM_MAP(CParseErrorProvider) COM_INTERFACE_ENTRY(ITagReplacer) END_COM_MAP() CSimpleArray *m_pErrors; int m_nCurrentError; CParseErrorProvider() throw() : m_pErrors(NULL), m_nCurrentError(-1) { } void Initialize(CSimpleArray *pErrors) throw() { m_pErrors = pErrors; } HTTP_CODE OnGetNextError() throw() { m_nCurrentError++; if (m_nCurrentError >= m_pErrors->GetSize() || m_nCurrentError < 0 ) { m_nCurrentError = -1; return HTTP_S_FALSE; } else return HTTP_SUCCESS; } HTTP_CODE OnGetErrorLineNumber() throw(...) { if (m_pErrors->GetSize() == 0) return HTTP_SUCCESS; if (m_nCurrentError > m_pErrors->GetSize() || m_nCurrentError < 0) m_nCurrentError = 0; CWriteStreamHelper c(m_pStream); if (!c.Write((*m_pErrors)[m_nCurrentError].m_nLineNumber)) return HTTP_FAIL; return HTTP_SUCCESS; } HTTP_CODE OnGetErrorText() throw(...) { if (m_pErrors->GetSize() == 0) return HTTP_SUCCESS; if (m_nCurrentError > m_pErrors->GetSize() || m_nCurrentError < 0) m_nCurrentError = 0; CWriteStreamHelper c(m_pStream); if (!c.Write((*m_pErrors)[m_nCurrentError].m_szError)) return HTTP_FAIL; return HTTP_SUCCESS; } HTTP_CODE OnGetErrorLine() throw(...) { if (m_pErrors->GetSize() == 0) return HTTP_SUCCESS; if (m_nCurrentError > m_pErrors->GetSize() || m_nCurrentError < 0) m_nCurrentError = 0; m_pStream->WriteStream((*m_pErrors)[m_nCurrentError].m_szStartLine, (int)((*m_pErrors)[m_nCurrentError].m_szEndLine - (*m_pErrors)[m_nCurrentError].m_szStartLine), NULL); return HTTP_SUCCESS; } BEGIN_REPLACEMENT_METHOD_MAP(CParseErrorProvider) REPLACEMENT_METHOD_ENTRY("GetNextError", OnGetNextError) REPLACEMENT_METHOD_ENTRY("GetErrorText", OnGetErrorText) REPLACEMENT_METHOD_ENTRY("GetErrorLine", OnGetErrorLine) REPLACEMENT_METHOD_ENTRY("GetErrorLineNumber", OnGetErrorLineNumber) END_REPLACEMENT_METHOD_MAP() }; #else bool m_bErrorsOccurred; #endif class CSaveThreadLocale { LCID m_locale; public: CSaveThreadLocale() throw() { m_locale = GetThreadLocale(); } ~CSaveThreadLocale() throw() { SetThreadLocale(m_locale); } }; protected: ITagReplacer *m_pReplacer; IAtlMemMgr *m_pMemMgr; static CCRTHeap m_crtHeap; public: enum PARSE_TOKEN_RESULT { INVALID_TOKEN, NORMAL_TOKEN, RESERVED_TOKEN }; CStencil(IAtlMemMgr *pMemMgr=NULL) throw() { m_pBufferStart = NULL; m_pBufferEnd = NULL; m_hCacheItem = NULL; m_ftLastModified.dwLowDateTime = 0; m_ftLastModified.dwHighDateTime = 0; m_ftLastChecked.dwLowDateTime = 0; m_ftLastChecked.dwHighDateTime = 0; m_arrTokens.SetCount(0, 128); m_nCodePage = CP_ACP; m_bUseLocaleACP = TRUE; m_szHandlerName[0] = '\0'; m_szDllPath[0] = '\0'; m_pMemMgr = pMemMgr; if (!pMemMgr) m_pMemMgr = &m_crtHeap; #ifdef ATL_DEBUG_STENCILS // m_pErrorStencil = NULL; #else m_bErrorsOccurred = false; #endif } #ifdef ATL_DEBUG_STENCILS virtual LPCSTR GetErrorStencil() throw() { static LPCSTR s_szString = "

While trying to parse a stencil file, " "the following errors occurred:

\r\n" "{{while GetNextError}}" "\r\n" "\r\n" "\r\n" "\r\n" "
Error type{{GetErrorText}}
Line number{{GetErrorLineNumber}}
Error text
{{GetErrorLine}}
\r\n" "{{endwhile}}" "
Stencil output follows:\r\n" "
"; return s_szString; } bool RenderErrors(IWriteStream *pStream) throw(...) { if (!pStream) return false; LPCSTR szString = GetErrorStencil(); if (szString == NULL) { // assume user doesn't want errors rendered return true; } CComObjectStackEx Errors; Errors.Initialize(&m_Errors); CStencil ErrorStencil; HTTP_CODE hcRet = ErrorStencil.LoadFromString(szString, (DWORD)strlen(szString)); if (hcRet != HTTP_SUCCESS) { return false; } if (!ErrorStencil.ParseReplacements(static_cast(&Errors))) { return false; } ErrorStencil.FinishParseReplacements(); HTTP_CODE hcErrorCode = ErrorStencil.Render(static_cast(&Errors), pStream); if (HTTP_ERROR_CODE(hcErrorCode) >= 400) return false; return true; } bool ParseSuccessful() throw() { return (m_Errors.GetSize() == 0); } bool AddError(LPCSTR szErrorText, LPCSTR szPosition) throw() { int nLineNum = 0; LPCSTR szStartLine = NULL; LPCSTR szPtr = m_pBufferStart; while (szPtr < szPosition) { if (*szPtr == '\n') { szStartLine = szPtr + 1; nLineNum++; } LPSTR szNext = CharNextExA(m_nCodePage, szPtr, 0); if (szNext == szPtr) { break; } szPtr = szNext; } LPCSTR szEndLine = szPtr; while (*szPtr) { if (*szPtr == '\n') break; szEndLine = szPtr; LPSTR szNext = CharNextExA(m_nCodePage, szPtr, 0); if (szNext == szPtr) { break; } szPtr = szNext; } ParseError p; strcpy(p.m_szError, szErrorText); p.m_szPosition = szPosition; p.m_nLineNumber = nLineNum; p.m_szStartLine = szStartLine; p.m_szEndLine = szEndLine; return (m_Errors.Add(p) == TRUE); } #else bool ParseSuccessful() throw() { return m_bErrorsOccurred; } void AddError(LPCSTR /*szErrorText*/, LPCSTR /*szPosition*/) throw() { m_bErrorsOccurred = true; } #endif // Call Uninitialize if you want to re-use an already initialized CStencil void Uninitialize() throw(...) { int nSize = (int) m_arrTokens.GetCount(); for (int nIndex = 0; nIndex < nSize; nIndex++) { if (m_arrTokens[nIndex].bDynamicAlloc) delete [] m_arrTokens[nIndex].pStart; if (m_arrTokens[nIndex].dwData != 0 && m_arrTokens[nIndex].type != STENCIL_LOCALE) m_pMemMgr->Free((void *) m_arrTokens[nIndex].dwData); } m_arrTokens.RemoveAll(); if ((m_ftLastModified.dwLowDateTime || m_ftLastModified.dwHighDateTime) && m_pBufferStart) delete [] m_pBufferStart; m_pBufferStart = NULL; m_pBufferEnd = NULL; } void GetLastModified(FILETIME *pftLastModified) throw() { ATLASSERT(pftLastModified); *pftLastModified = m_ftLastModified; } void GetLastChecked(FILETIME *pftLastChecked) throw() { ATLASSERT(pftLastChecked); *pftLastChecked = m_ftLastChecked; } void SetLastChecked(FILETIME *pftLastChecked) throw() { ATLASSERT(pftLastChecked); m_ftLastChecked = *pftLastChecked; } HCACHEITEM GetCacheItem() throw() { return m_hCacheItem; } void SetCacheItem(HCACHEITEM hCacheItem) throw() { ATLASSERT(m_hCacheItem == NULL); m_hCacheItem = hCacheItem; } void GetHandlerName(LPSTR szDllPath, LPSTR szHandlerName) throw() { strcpy(szDllPath, m_szDllPath); strcpy(szHandlerName, m_szHandlerName); } // Adds a token to the token array, handler name, method name // and handler function offset are optional ATL_NOINLINE DWORD AddToken( LPSTR pStart, LPSTR pEnd, DWORD dwType, LPCSTR szHandlerName = NULL, LPCSTR szMethodName = NULL, DWORD dwFnOffset = STENCIL_INVALIDOFFSET, DWORD dwObjOffset = STENCIL_INVALIDOFFSET, DWORD_PTR dwData = 0, DWORD dwMap = 0, BOOL bDynamicAlloc = 0) throw() { StencilToken t; memset(&t, 0x00, sizeof(t)); t.pStart = pStart; t.pEnd = pEnd; t.type = dwType; t.dwLoopIndex = STENCIL_INVALIDINDEX; t.dwFnOffset = dwFnOffset; t.dwObjOffset = dwObjOffset; t.dwData = dwData; t.dwMap = dwMap; t.bDynamicAlloc = bDynamicAlloc; if (szHandlerName && (sizeof(t.szHandlerName) > strlen(szHandlerName))) strcpy(t.szHandlerName, szHandlerName); else t.szHandlerName[0] = '\0'; if (szMethodName && (sizeof(t.szMethodName) > strlen(szMethodName))) strcpy(t.szMethodName, szMethodName); else t.szMethodName[0] = '\0'; _ATLTRY { return (DWORD) m_arrTokens.Add(t); } _ATLCATCHALL() { return STENCIL_INVALIDINDEX; } } // maps a stencil file into memory HTTP_CODE LoadFile(LPCSTR szFileName) throw() { HRESULT hr = E_FAIL; ULONGLONG dwLen = 0; CAtlFile file; hr = file.Create(CA2CTEX(szFileName), GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING); if (FAILED(hr) || GetFileType(file) != FILE_TYPE_DISK) return HTTP_ERROR(500, ISE_SUBERR_STENCIL_LOAD_FAIL); // couldn't load SRF! if (GetFileTime(file, NULL, NULL, &m_ftLastModified)) { if (SUCCEEDED(file.GetSize(dwLen))) { ATLASSERT(!m_pBufferStart); GetSystemTimeAsFileTime(&m_ftLastChecked); m_pBufferStart = NULL; ATLTRY(m_pBufferStart = new CHAR[(UINT) dwLen]); if (m_pBufferStart == NULL) return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); // out of memory DWORD dwRead; hr = file.Read(m_pBufferStart, (DWORD) dwLen, dwRead); if (FAILED(hr)) { delete [] m_pBufferStart; m_pBufferStart = NULL; return HTTP_ERROR(500, ISE_SUBERR_READFILEFAIL); // ReadFile failed } m_pBufferEnd = m_pBufferStart + dwRead; } } return HTTP_SUCCESS; } // loads a stencil from the specified resource. HTTP_CODE LoadFromResource(HINSTANCE hInstRes, LPCSTR szID, LPCSTR szType = NULL) throw() { if (!szType) szType = (LPCSTR) RT_HTML; HRSRC hRsrc = FindResourceA(hInstRes, szID, szType); HGLOBAL hgResource = NULL; if (hRsrc) { hgResource = LoadResource(hInstRes, hRsrc); if (!hgResource) return HTTP_FAIL; } else return HTTP_FAIL; DWORD dwSize = SizeofResource(hInstRes, hRsrc); m_pBufferStart = (LPSTR)LockResource(hgResource); m_pBufferEnd = m_pBufferStart+dwSize; return HTTP_SUCCESS; } // loads a stencil from the specified resource HTTP_CODE LoadFromResource(HINSTANCE hInstRes, UINT nId, LPCSTR szType = NULL) throw() { return LoadFromResource(hInstRes, MAKEINTRESOURCEA(nId), szType); } // loads a stencil from a string HTTP_CODE LoadFromString(LPCSTR szString, DWORD dwSize) throw() { m_pBufferStart = (LPSTR) szString; m_pBufferEnd = m_pBufferStart+dwSize; return HTTP_SUCCESS; } inline BOOL CheckTag(LPCSTR szTag, DWORD dwTagLen, LPSTR szStart, DWORD dwLen) throw() { if (dwLen < dwTagLen) return FALSE; if (memcmp(szStart, szTag, dwTagLen)) return FALSE; if (_istspace(szStart[dwTagLen]) || szStart[dwTagLen] == '}') return TRUE; return FALSE; } inline void FindTagArgs(LPSTR& szstart, LPSTR& szend, int nKeywordChars) throw() { if (*szstart == '{') szstart += 2; // move past {{ szstart = SkipSpace(szstart, m_nCodePage); // move past whitespace szstart += nKeywordChars; // move past keyword szstart = SkipSpace(szstart, m_nCodePage); // move past whitespace after keyword if (*szend == '}') szend -=2; // chop off }} szend = RSkipSpace(szstart, szend, m_nCodePage); // chop of trailing whitespace } // Cracks the loaded stencil into an array of StencilTokens in preparation for // rendering. LoadStencil must be called prior to calling this function. virtual bool ParseReplacements(ITagReplacer* pReplacer) throw(...) { return ParseReplacements(pReplacer, GetBufferStart(), GetBufferEnd()); } virtual void FinishParseReplacements() throw(...) { DWORD dwSize = (DWORD) m_arrTokens.GetCount(); for (DWORD dwIndex = 0; dwIndex < dwSize; dwIndex++) { StencilToken& token = m_arrTokens[dwIndex]; bool bUnclosedBlock = ((token.type == STENCIL_CONDITIONALSTART || token.type == STENCIL_CONDITIONALELSE || token.type == STENCIL_ITERATORSTART) && token.dwLoopIndex == STENCIL_INVALIDINDEX); if (token.szMethodName[0] && (token.dwFnOffset == STENCIL_INVALIDOFFSET || bUnclosedBlock)) { if (bUnclosedBlock || m_pReplacer->FindReplacementOffset( token.szMethodName, &token.dwFnOffset, token.szHandlerName, &token.dwObjOffset, &token.dwMap, (void **)(&token.dwData), m_pMemMgr) != HTTP_SUCCESS) { if (bUnclosedBlock && token.type == STENCIL_CONDITIONALSTART) AddError("{{if}} without {{endif}}", token.pStart); else if (bUnclosedBlock && token.type == STENCIL_CONDITIONALELSE) AddError("{{else}} without {{endif}}", token.pStart); else if (bUnclosedBlock && token.type == STENCIL_ITERATORSTART) AddError("{{while}} without {{endwhile}}", token.pStart); else { char szBuf[ATL_MAX_METHOD_NAME_LEN+sizeof("Unresolved replacement : ''")]; _snprintf(szBuf, ATL_MAX_METHOD_NAME_LEN+sizeof("Unresolved replacement : ''")-1, "Unresolved replacement : '%s'", token.szMethodName); AddError(szBuf, token.pStart); } // unresolved replacement, convert it to a text token token.type = STENCIL_TEXTTAG; // convert all linked tokens to text tokens as well // this includes: endif, else, endwhile DWORD dwLoopIndex = token.dwLoopIndex; while (dwLoopIndex != dwIndex && dwLoopIndex != STENCIL_INVALIDINDEX) { m_arrTokens[dwLoopIndex].type = STENCIL_TEXTTAG; dwLoopIndex = m_arrTokens[dwLoopIndex].dwLoopIndex; } } } } } virtual bool Parse(ITagReplacer *pReplacer) throw( ... ) { if (ParseReplacements(pReplacer)) { FinishParseReplacements(); return true; } return false; } DWORD CheckTopAndPop(DWORD *pBlockStack, DWORD *pdwTop, DWORD dwToken) throw() { if (*pdwTop == 0) return STENCIL_INVALIDINDEX; if (m_arrTokens[pBlockStack[*pdwTop]].type == dwToken) { *pdwTop = (*pdwTop) - 1; return pBlockStack[(*pdwTop)+1]; } return STENCIL_INVALIDINDEX; } DWORD ParseReplacement( LPSTR szTokenStart, LPSTR szTokenEnd, DWORD * /*pBlockStack*/, DWORD * /*pdwTop*/, DWORD dwTokenType = STENCIL_REPLACEMENT, DWORD dwKeywordLen = 0) throw() { // hold on to the start and end pointers (before removing curlies and whitespace) // this is needed so that we can convert the token to a text token if the method // is not resolved (in FinishParseReplacements) LPSTR szStart = szTokenStart; LPSTR szEnd = szTokenEnd; FindTagArgs(szTokenStart, szTokenEnd, dwKeywordLen); char szMethodName[ATL_MAX_METHOD_NAME_LEN+1]; char szHandlerName[ATL_MAX_HANDLER_NAME_LEN+1]; DWORD dwIndex; //look up the handler name, method name and handler interface if (HTTP_SUCCESS == GetHandlerAndMethodNames(szTokenStart, szTokenEnd, szMethodName, szHandlerName)) dwIndex = AddToken(szStart, szEnd, dwTokenType, szHandlerName, szMethodName, STENCIL_INVALIDINDEX, STENCIL_INVALIDINDEX, 0, STENCIL_BASIC_MAP); else dwIndex = STENCIL_INVALIDINDEX; return dwIndex; } DWORD PushToken(DWORD *pBlockStack, DWORD *pdwTop, DWORD dwIndex) throw() { if (*pdwTop < (ATL_MAX_BLOCK_STACK-1)) { *pdwTop = (*pdwTop) + 1; pBlockStack[*pdwTop] = dwIndex; } else { dwIndex = STENCIL_INVALIDINDEX; } return dwIndex; } DWORD ParseWhile(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { DWORD dwIndex = ParseReplacement(szTokenStart, szTokenEnd, pBlockStack, pdwTop, STENCIL_ITERATORSTART, 5); if (dwIndex == STENCIL_INVALIDINDEX) return dwIndex; return PushToken(pBlockStack, pdwTop, dwIndex); } DWORD ParseEndWhile(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { DWORD dwTopIndex = CheckTopAndPop(pBlockStack, pdwTop, STENCIL_ITERATORSTART); if (dwTopIndex == STENCIL_INVALIDINDEX) { AddError("{{endwhile}} without {{while}}", szTokenStart); return dwTopIndex; } DWORD dwIndex = AddToken(szTokenStart, szTokenEnd, STENCIL_ITERATOREND); m_arrTokens[dwTopIndex].dwLoopIndex = dwIndex; m_arrTokens[dwIndex].dwLoopIndex = dwTopIndex; return dwIndex; } DWORD ParseIf(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { DWORD dwIndex = ParseReplacement(szTokenStart, szTokenEnd, pBlockStack, pdwTop, STENCIL_CONDITIONALSTART, 2); if (dwIndex == STENCIL_INVALIDINDEX) return dwIndex; return PushToken(pBlockStack, pdwTop, dwIndex); } DWORD ParseElse(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { DWORD dwTopIndex = CheckTopAndPop(pBlockStack, pdwTop, STENCIL_CONDITIONALSTART); if (dwTopIndex == STENCIL_INVALIDINDEX) { AddError("{{else}} without {{if}}", szTokenStart); return dwTopIndex; } DWORD dwIndex = AddToken(szTokenStart, szTokenEnd, STENCIL_CONDITIONALELSE); m_arrTokens[dwTopIndex].dwLoopIndex = dwIndex; return PushToken(pBlockStack, pdwTop, dwIndex); } DWORD ParseEndIf(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { DWORD dwTopIndex = CheckTopAndPop(pBlockStack, pdwTop, STENCIL_CONDITIONALSTART); if (dwTopIndex == STENCIL_INVALIDINDEX) { dwTopIndex = CheckTopAndPop(pBlockStack, pdwTop, STENCIL_CONDITIONALELSE); if (dwTopIndex == STENCIL_INVALIDINDEX) { AddError("{{endif}} without {{if}}", szTokenStart); return dwTopIndex; } } DWORD dwIndex = AddToken(szTokenStart, szTokenEnd, STENCIL_CONDITIONALEND); m_arrTokens[dwTopIndex].dwLoopIndex = dwIndex; return dwIndex; } DWORD ParseLocale(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD * /*pBlockStack*/, DWORD * /*pdwTop*/) throw() { LPSTR szstart = szTokenStart; LPSTR szend = szTokenEnd; LCID locale = 0xFFFFFFFF; FindTagArgs(szstart, szend, 6); #ifndef ATL_NO_MLANG if (isdigit(szstart[0])) { locale = (LCID) atoi(szstart); } else { HRESULT hr; CComPtr pML; hr = pML.CoCreateInstance(__uuidof(CMultiLanguage)); if (FAILED(hr)) { ATLTRACE(atlTraceStencil, 0, _T("Couldn't create CMultiLanguage object. check MLANG installation.")); AddError("Couldn't create CMultiLanguage", szTokenStart); } else { CStringW str(szstart, (int)((szend-szstart)+1)); #ifdef __IMultiLanguage2_INTERFACE_DEFINED__ // use IMultiLanguage2 if possible CComPtr spML2; hr = pML.QueryInterface(&spML2); if (FAILED(hr) || !spML2.p) hr = pML->GetLcidFromRfc1766(&locale, CComBSTR(str)); else hr = spML2->GetLcidFromRfc1766(&locale, CComBSTR(str)); #else // __IMultiLanguage2_INTERFACE_DEFINED__ hr = pML->GetLcidFromRfc1766(&locale, CComBSTR(str)); #endif // __IMultiLanguage2_INTERFACE_DEFINED__ if (FAILED(hr)) AddError("Error getting lcid", szTokenStart); } if (FAILED(hr)) locale = 0xFFFFFFFF; } #else locale = (LCID) atoi(szstart); #endif if (m_bUseLocaleACP) { TCHAR szACP[7]; if (GetLocaleInfo(locale, LOCALE_IDEFAULTANSICODEPAGE, szACP, 7) != 0) { m_nCodePage = (WORD) _ttoi(szACP); } else { AddError("GetLocaleInfo failed", szTokenStart); } } DWORD dwCurrentTokenIndex = STENCIL_INVALIDINDEX; if (locale != 0xFFFFFFFF) dwCurrentTokenIndex = AddToken(NULL, NULL, STENCIL_LOCALE, NULL, NULL, STENCIL_INVALIDOFFSET, STENCIL_INVALIDOFFSET, locale); else return STENCIL_INVALIDINDEX; return dwCurrentTokenIndex; } DWORD ParseCodepage(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD * /*pBlockStack*/, DWORD * /*pdwTop*/) throw() { LPSTR szstart = szTokenStart; LPSTR szend = szTokenEnd; WORD nCodePage = 0xFFFF; FindTagArgs(szstart, szend, 8); #ifndef ATL_NO_MLANG if (isdigit(szstart[0])) { nCodePage = (WORD) atoi(szstart); } else { HRESULT hr; CComPtr pML; hr = pML.CoCreateInstance(__uuidof(CMultiLanguage)); if (FAILED(hr)) { ATLTRACE(atlTraceStencil, 0, _T("Couldn't create CMultiLanguage object. check MLANG installation.")); AddError("Couldn't CoCreate CMultiLanguage object", szTokenStart); } else { CStringW str(szstart, (int)((szend-szstart)+1)); MIMECSETINFO info; #ifdef __IMultiLanguage2_INTERFACE_DEFINED__ // use IMultiLanguage2 if possible CComPtr spML2; hr = pML.QueryInterface(&spML2); if (FAILED(hr) || !spML2.p) hr = pML->GetCharsetInfo(CComBSTR(str), &info); else hr = spML2->GetCharsetInfo(CComBSTR(str), &info); #else // __IMultiLanguage2_INTERFACE_DEFINED__ hr = pML->GetCharsetInfo(CComBSTR(str), &info); #endif // __IMultiLanguage2_INTERFACE_DEFINED__ // for most character sets, uiCodePage and uiInternetEncoding // are the same. UTF-8 is the exception that we're concerned about. // for that character set, we want uiInternetEncoding (65001 - UTF-8) // instead of uiCodePage (1200 - UCS-2) if (SUCCEEDED(hr)) nCodePage = (WORD) info.uiInternetEncoding; else AddError("GetCharsetInfo failed", szTokenStart); } if (FAILED(hr)) nCodePage = 0xFFFF; } #else nCodePage = (WORD) atoi(szstart); #endif if (nCodePage != 0xFFFF) m_nCodePage = nCodePage; m_bUseLocaleACP = FALSE; return STENCIL_INVALIDINDEX; } PARSE_TOKEN_RESULT ParseHandler(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD * /*pBlockStack*/, DWORD * /*pdwTop*/) throw() { LPSTR szstart = szTokenStart; LPSTR szend = szTokenEnd; if (m_szHandlerName[0] && m_szDllPath[0]) return RESERVED_TOKEN; // already found the handler and path dll names FindTagArgs(szstart, szend, 7); size_t nlen = (szend-szstart)+1; char szHandlerDllName[MAX_PATH + ATL_MAX_HANDLER_NAME_LEN + 2]; if (nlen < MAX_PATH + ATL_MAX_HANDLER_NAME_LEN + 2) { memcpy(szHandlerDllName, szstart, szend-szstart+1); szHandlerDllName[szend-szstart+1] = '\0'; DWORD dwDllPathLen = MAX_PATH+1; DWORD dwHandlerNameLen = ATL_MAX_HANDLER_NAME_LEN+1; if (!_AtlCrackHandler(szHandlerDllName, m_szDllPath, &dwDllPathLen, m_szHandlerName, &dwHandlerNameLen)) { return INVALID_TOKEN; } } return RESERVED_TOKEN; } virtual PARSE_TOKEN_RESULT ParseToken(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { LPSTR pStart = szTokenStart; pStart += 2; //skip curlies pStart = SkipSpace(pStart, m_nCodePage); DWORD dwLen = (DWORD)(szTokenEnd - szTokenStart); DWORD dwIndex = STENCIL_INVALIDINDEX; PARSE_TOKEN_RESULT ret = RESERVED_TOKEN; if (CheckTag("endwhile", 8, pStart, dwLen)) dwIndex = ParseEndWhile(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (CheckTag("while", 5, pStart, dwLen)) dwIndex = ParseWhile(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (CheckTag("endif", 5, pStart, dwLen)) dwIndex = ParseEndIf(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (CheckTag("else", 4, pStart, dwLen)) dwIndex = ParseElse(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (CheckTag("if", 2, pStart, dwLen)) dwIndex = ParseIf(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (CheckTag("locale", 6, pStart, dwLen)) dwIndex = ParseLocale(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (CheckTag("handler", 7, pStart, dwLen)) { return ParseHandler(szTokenStart, szTokenEnd, pBlockStack, pdwTop); } else if (CheckTag("codepage", 8, pStart, dwLen)) { ParseCodepage(szTokenStart, szTokenEnd, pBlockStack, pdwTop); return RESERVED_TOKEN; } else { dwIndex = ParseReplacement(szTokenStart, szTokenEnd, pBlockStack, pdwTop, STENCIL_REPLACEMENT); if (dwIndex == STENCIL_INVALIDINDEX) return INVALID_TOKEN; ret = NORMAL_TOKEN; } if (dwIndex == STENCIL_INVALIDINDEX) return INVALID_TOKEN; return ret; } virtual bool ParseReplacements(ITagReplacer* pReplacer, LPSTR pStart, LPSTR pEnd) throw() { LPSTR szCurr = pStart; DWORD BlockStack[ATL_MAX_BLOCK_STACK]; DWORD dwTop = 0; m_pReplacer = pReplacer; DWORD dwCurrentTokenIndex = 0; if (!szCurr) { ATLASSERT(FALSE); AddError("NULL parameter to ParseReplacements", NULL); return false; } LPSTR szEnd = pEnd; if (szEnd <= szCurr) { ATLASSERT(FALSE); AddError("Empty or negative string passed to ParseReplacements", NULL); return true; } while(szCurr < szEnd) { //mark the start of this block, then find the end of the block //the end is denoted by an opening curly LPSTR szStart = szCurr; while (szCurr < szEnd && (*szCurr != '{' || szCurr[1] != '{')) { LPSTR szNext = CharNextExA(m_nCodePage, szCurr, 0); if (szNext == szCurr) { // embedded null AddError("Embedded null character in stencil", NULL); return true; } szCurr = szNext; } //special case for the last text block, if there is one if (szCurr >= szEnd) { // add the last token. This is everything after the last // double curly block, which is text. dwCurrentTokenIndex = AddToken(szStart, szEnd-1, STENCIL_TEXTTAG); break; } //if there are any characters between szStart and szCurr inclusive, //copy them to a text token. if (szCurr-1 >= szStart) dwCurrentTokenIndex = AddToken(szStart, szCurr-1, STENCIL_TEXTTAG); //find the end of the tag LPSTR szEndTag; szStart = szCurr; szCurr += 2; // Skip over the two '{' s while (szCurr < szEnd) { if (szCurr[0] == '}' && szCurr[1] == '}') break; else if (szCurr[0] == '{') break; LPSTR szNext = CharNextExA(m_nCodePage, szCurr, 0); if (szNext == szCurr) { // embedded null AddError("Embedded null character in stencil", NULL); return true; } szCurr = szNext; } if (szCurr >= szEnd) { AddError("Unmatched {{", szStart); AddToken(szStart, szCurr-1, STENCIL_TEXTTAG); break; } if (szCurr[0] == '{') { if (szCurr[1] != '{') { szCurr--; } AddError("Mismatched {{", szStart); AddToken(szStart, szCurr-1, STENCIL_TEXTTAG); continue; } szEndTag = CharNextExA(m_nCodePage, szCurr, 0); if (szEndTag == szCurr) { // embedded null AddError("Embedded null character in stencil", NULL); return true; } PARSE_TOKEN_RESULT ret = ParseToken(szStart, szEndTag, BlockStack, &dwTop); if (ret == INVALID_TOKEN) { dwCurrentTokenIndex = AddToken(szStart, szEndTag, STENCIL_TEXTTAG); szCurr = CharNextExA(m_nCodePage, szEndTag, 0); continue; } szCurr = CharNextExA(m_nCodePage, szEndTag, 0); if (szEndTag == szCurr) { // embedded null AddError("Embedded null character in stencil", NULL); return true; } if (ret == RESERVED_TOKEN) { // REVIEW: should these be using CharNextExA if (szCurr < szEnd && *szCurr == '\n') szCurr++; else if ((szCurr+1 < szEnd && *szCurr == '\r' && *(szCurr+1) == '\n')) szCurr += 2; } } return true; } HTTP_CODE GetHandlerAndMethodNames( LPSTR pStart, LPSTR pEnd, LPSTR pszMethodName, LPSTR pszHandlerName) throw() { if (!pszMethodName || !pszHandlerName) { ATLASSERT(FALSE); AddError("Bad parameter", pStart); return HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); } *pszMethodName = '\0'; *pszHandlerName = '\0'; CHAR szMethodString[ATL_MAX_METHOD_NAME_LEN + ATL_MAX_HANDLER_NAME_LEN+1]; //*2 in case the tag is an id.tag and +1 for the dot HTTP_CODE hcErr = HTTP_SUCCESS; // // copy the method string // size_t nMethodLen = (pEnd-pStart)+1; if (nMethodLen >= (ATL_MAX_METHOD_NAME_LEN + ATL_MAX_HANDLER_NAME_LEN+1)) { AddError("Method name too long", pStart); return HTTP_ERROR(500, ISE_SUBERR_LONGMETHODNAME); } memcpy(szMethodString, pStart, nMethodLen); szMethodString[nMethodLen] = '\0'; // // now crack the method string and get the handler // id and function name // LPSTR szParen = strchr(szMethodString, '('); LPSTR szDot = strchr(szMethodString, '.'); if (szDot && (!szParen || (szDot < szParen))) { *szDot = '\0'; szDot++; // copy method name if (strlen(szDot)(pState); // set up locale info CSaveThreadLocale lcidSave; if (_pState) { dwIndex = _pState->dwIndex; // restore the locale if we're restarting rendering if (_pState->locale != CP_ACP) SetThreadLocale(_pState->locale); } pReplacer->SetStream(pWriteStream); while (dwIndex < dwArraySize) { // RenderToken advances dwIndex appropriately for us. dwIndex = RenderToken(dwIndex, pReplacer, pWriteStream, &hcErrorCode, pState); if (dwIndex == STENCIL_INVALIDINDEX || hcErrorCode != HTTP_SUCCESS) break; } if (IsAsyncStatus(hcErrorCode)) { ATLASSERT(_pState); // state is required for async _pState->dwIndex = dwIndex; } // lcidSave destructor will restore the locale info in case it was changed return hcErrorCode; } inline BOOL IsValidIndex(DWORD dwIndex) const throw() { if (dwIndex == STENCIL_INVALIDINDEX) return FALSE; if (dwIndex < GetTokenCount()) return TRUE; else return FALSE; } virtual DWORD RenderToken( DWORD dwIndex, ITagReplacer *pReplacer, IWriteStream *pWriteStream, HTTP_CODE *phcErrorCode, void* pState = NULL) const throw(...) { const StencilToken* pToken = GetToken(dwIndex); DWORD dwNextToken = 0; HTTP_CODE hcErrorCode = HTTP_SUCCESS; if (!pToken) return STENCIL_INVALIDINDEX; switch (pToken->type) { case STENCIL_ITERATORSTART: { HTTP_CODE hcErr = STENCIL_SUCCESS; // A 'while' token has to at least be followed by an endwhile! if (!IsValidIndex(dwIndex+1)) { // This should have been caught at parse time dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_INVALIDINDEX); ATLASSERT(FALSE); break; } // End of loop should be valid if (!IsValidIndex(pToken->dwLoopIndex)) { // This should have been caught at parse time dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_MISMATCHWHILE); ATLASSERT(FALSE); break; } if (pToken->dwFnOffset == STENCIL_INVALIDOFFSET) { // This should have been caught at parse time dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_INVALIDFUNCOFFSET); ATLASSERT(FALSE); break; } DWORD dwLoopIndex = pToken->dwLoopIndex; // points to the end of the loop // Call the replacement method // if it returns HTTP_SUCCESS, enter the loop // if it returns HTTP_S_FALSE, terminate the loop hcErr = pReplacer->RenderReplacement(pToken->dwFnOffset, pToken->dwObjOffset, pToken->dwMap, (void *) pToken->dwData); if (hcErr == HTTP_SUCCESS) { dwNextToken = dwIndex+1; hcErrorCode = HTTP_SUCCESS; } else if (hcErr == HTTP_S_FALSE) { dwNextToken = dwLoopIndex+1; hcErrorCode = HTTP_SUCCESS; } else { dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = hcErr; break; } } break; case STENCIL_ITERATOREND: { dwNextToken = pToken->dwLoopIndex; hcErrorCode = HTTP_SUCCESS; ATLASSERT(GetToken(dwNextToken)->type == STENCIL_ITERATORSTART); } break; case STENCIL_CONDITIONALSTART: { if (pToken->type == STENCIL_CONDITIONALSTART && pToken->dwFnOffset == STENCIL_INVALIDOFFSET) { // This should have been caught at parse time ATLASSERT(FALSE); dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_INVALIDFUNCOFFSET); break; } if (pToken->dwLoopIndex == STENCIL_INVALIDINDEX) { // This should have been caught at parse time ATLASSERT(FALSE); dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_MISMATCHIF); break; } DWORD dwLoopIndex = pToken->dwLoopIndex; // points to the end of the loop HTTP_CODE hcErr; // Call the replacement method. // If it returns HTTP_SUCCESS, we render everything up to // the end of the conditional. // if it returns HTTP_S_FALSE, the condition is not met and we // render the else part if it exists or jump past the endif otherwise hcErr = pReplacer->RenderReplacement(pToken->dwFnOffset, pToken->dwObjOffset, pToken->dwMap, (void *)pToken->dwData); if (hcErr == HTTP_SUCCESS) { dwNextToken = dwIndex+1; hcErrorCode = HTTP_SUCCESS; } else if (hcErr == HTTP_S_FALSE) { dwNextToken = dwLoopIndex+1; hcErrorCode = HTTP_SUCCESS; } else { dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = hcErr; break; } } break; case STENCIL_CONDITIONALELSE: { if (pToken->dwLoopIndex == STENCIL_INVALIDINDEX) { // This should have been caught at parse time ATLASSERT(FALSE); dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_MISMATCHIF); break; } dwNextToken = pToken->dwLoopIndex+1; hcErrorCode = HTTP_SUCCESS; } break; case STENCIL_CONDITIONALEND: { dwNextToken = dwIndex+1; hcErrorCode = HTTP_SUCCESS; } break; case STENCIL_TEXTTAG: { pWriteStream->WriteStream(pToken->pStart, (int)((pToken->pEnd-pToken->pStart)+1), NULL); dwNextToken = dwIndex+1; } break; case STENCIL_REPLACEMENT: { if (pToken->dwFnOffset == STENCIL_INVALIDOFFSET) { // This should have been caught at parse time ATLASSERT(FALSE); dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_INVALIDFUNCOFFSET); break; } hcErrorCode = pReplacer->RenderReplacement(pToken->dwFnOffset, pToken->dwObjOffset, pToken->dwMap, (void *)pToken->dwData); if (IsAsyncContinueStatus(hcErrorCode)) dwNextToken = dwIndex; // call the tag again after we get back else { dwNextToken = dwIndex + 1; // when returned from a handler, these indicate that the handler is done // and that we should move on to the next handler when called back if (hcErrorCode == HTTP_SUCCESS_ASYNC_DONE) hcErrorCode = HTTP_SUCCESS_ASYNC; else if (hcErrorCode == HTTP_SUCCESS_ASYNC_NOFLUSH_DONE) hcErrorCode = HTTP_SUCCESS_ASYNC_NOFLUSH; } } break; case STENCIL_LOCALE: { if (pState) { CStencilState* _pState = reinterpret_cast(pState); _pState->locale = (LCID) pToken->dwData; } SetThreadLocale((LCID) pToken->dwData); dwNextToken = dwIndex + 1; } break; default: { ATLASSERT(FALSE); dwNextToken = STENCIL_INVALIDINDEX; hcErrorCode = HTTP_ERROR(500, ISE_SUBERR_STENCIL_UNEXPECTEDTYPE); break; } } ATLASSERT(dwNextToken != dwIndex || IsAsyncContinueStatus(hcErrorCode)); if (phcErrorCode) *phcErrorCode = hcErrorCode; return dwNextToken; } DWORD GetTokenCount() const throw() { return (DWORD) m_arrTokens.GetCount(); } const StencilToken* GetToken(DWORD dwIndex) const throw() { return &(m_arrTokens[dwIndex]); } StencilToken* GetToken(DWORD dwIndex) throw() { return &(m_arrTokens[dwIndex]); } LPSTR GetBufferStart() const throw() { return m_pBufferStart; } LPSTR GetBufferEnd() const throw() { return m_pBufferEnd; } WORD GetCodePage() const throw() { return m_nCodePage; } // IMemoryCacheClient STDMETHOD(QueryInterface)(REFIID riid, void **ppv) throw() { if (!ppv) return E_POINTER; if (InlineIsEqualGUID(riid, __uuidof(IUnknown)) || InlineIsEqualGUID(riid, __uuidof(IMemoryCacheClient))) { *ppv = static_cast(this); return S_OK; } return E_NOINTERFACE; } STDMETHOD_(ULONG, AddRef)() throw() { return 1; } STDMETHOD_(ULONG, Release)() throw() { return 1; } STDMETHOD(Free)(const void *pData) throw() { if (!pData) return E_UNEXPECTED; ATLASSERT(*((void **) pData) == static_cast(this)); Uninitialize(); delete this; return S_OK; } }; // class CStencil struct StencilIncludeInfo { public: CHAR m_szQueryString[2048]; CHAR m_szFileName[MAX_PATH]; }; class CIncludeServerContext : public CComObjectRootEx, public CWrappedServerContext { public: BEGIN_COM_MAP(CIncludeServerContext) COM_INTERFACE_ENTRY(IHttpServerContext) END_COM_MAP() IWriteStream * m_pStream; const StencilIncludeInfo * m_pIncludeInfo; CIncludeServerContext() throw() { m_pStream = NULL; m_pIncludeInfo = NULL; } void Initialize( IWriteStream *pStream, IHttpServerContext* pServerContext, const StencilIncludeInfo * pIncludeInfo) throw() { m_pStream = pStream; m_spParent = pServerContext; m_pIncludeInfo = pIncludeInfo; } LPCSTR GetRequestMethod() throw() { return "GET"; } LPCSTR GetQueryString() throw() { return m_pIncludeInfo->m_szQueryString; } LPCSTR GetPathTranslated() throw() { return m_pIncludeInfo->m_szFileName; } LPCSTR GetScriptPathTranslated() throw() { return m_pIncludeInfo->m_szFileName; } DWORD GetTotalBytes() throw() { return 0; } DWORD GetAvailableBytes() throw() { return 0; } BYTE *GetAvailableData() throw() { return NULL; } LPCSTR GetContentType() throw() { return 0; } BOOL WriteClient(void *pvBuffer, DWORD *pdwBytes) throw(...) { HRESULT hr = m_pStream->WriteStream((LPCSTR) pvBuffer, *pdwBytes, pdwBytes); return SUCCEEDED(hr); } BOOL ReadClient(void * /*pvBuffer*/, DWORD * /*pdwSize*/) throw() { return FALSE; } BOOL AsyncReadClient(void * /*pvBuffer*/, DWORD * /*pdwSize*/) throw() { return FALSE; } BOOL SendRedirectResponse(LPCSTR /*pszRedirectURL*/) throw() { return FALSE; } BOOL SendResponseHeader( LPCSTR /*pszHeader*/, LPCSTR /*pszStatusCode*/, BOOL /*fKeepConn*/) throw() { return TRUE; } BOOL DoneWithSession(DWORD /*dwHttpStatusCode*/) throw() { return TRUE; } BOOL RequestIOCompletion(PFN_HSE_IO_COMPLETION /*pfn*/, DWORD * /*pdwContext*/) throw() { return FALSE; } }; // class CIncludeServerContext class CIDServerContext : public CComObjectRootEx, public CWrappedServerContext { public: CHttpResponse *m_pResponse; CHttpRequest *m_pRequest; BEGIN_COM_MAP(CIDServerContext) COM_INTERFACE_ENTRY(IHttpServerContext) END_COM_MAP() CIDServerContext() throw() : m_pResponse(NULL), m_pRequest(NULL) { } BOOL Initialize( CHttpResponse *pResponse, CHttpRequest *pRequest) throw() { m_pResponse = pResponse; m_pRequest = pRequest; HRESULT hr = m_pRequest->GetServerContext(&m_spParent); return (SUCCEEDED(hr)); } LPCSTR GetRequestMethod() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetMethodString(); } LPCSTR GetQueryString() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetQueryString(); } LPCSTR GetPathInfo() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetPathInfo(); } LPCSTR GetPathTranslated() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetPathTranslated(); } DWORD GetTotalBytes() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetTotalBytes(); } DWORD GetAvailableBytes() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetAvailableBytes(); } BYTE *GetAvailableData() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetAvailableData(); } LPCSTR GetContentType() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetContentType(); } LPCSTR GetScriptPathTranslated() throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->GetScriptPathTranslated(); } BOOL WriteClient(void *pvBuffer, DWORD *pdwBytes) throw() { ATLASSERT(m_pResponse != NULL); return m_pResponse->WriteLen((LPCSTR)pvBuffer, *pdwBytes); } BOOL ReadClient(void *pvBuffer, DWORD *pdwSize) throw() { ATLASSERT(m_pRequest != NULL); return m_pRequest->ReadData((LPSTR)pvBuffer, pdwSize); } BOOL SendRedirectResponse(LPCSTR pszRedirectURL) throw() { ATLASSERT(m_pResponse != NULL); return m_pResponse->Redirect(pszRedirectURL); } BOOL TransmitFile( HANDLE hFile, PFN_HSE_IO_COMPLETION pfn, void *pContext, LPCSTR szStatusCode, DWORD dwBytesToWrite, DWORD dwOffset, void *pvHead, DWORD dwHeadLen, void *pvTail, DWORD dwTailLen, DWORD dwFlags) throw(...) { ATLASSERT(m_pResponse != NULL); ATLASSERT(m_spParent != NULL); m_pResponse->Flush(); return m_spParent->TransmitFile(hFile, pfn, pContext, szStatusCode, dwBytesToWrite, dwOffset, pvHead, dwHeadLen, pvTail, dwTailLen, dwFlags); } }; // class CIDServerContext // // CHtmlStencil // CHtmlStencil is a specialization of CStencil. CHtmlStencil adds the following // capabilities to CStencil: // // Support for rendering {{include }} tags // The {{include }} tags specify another stencil to be included in-place during // stencil rendering. The {{include }} tag takes a single parameter which is the // URL of the stencil to include. That URL can optionally include parameters. // An example: // {{include mystencil.srf?param1=value1}} // // We also grab the handler name and the name of any subhandlers. A handler must be // the first tag in the .srf file. The syntax for the handler specification is: // {{handler MyDynamicHandler.dll/Default}} // which would cause the MyDynamicHandler.dll to be loaded. Once loaded, the stencil // processor will ask for the IReplacementHandler interface of the object named "Default". // // Additional handlers can be specified after the default handler. An example of an // additional handler would be: // {{subhandler OtherHandler MyOtherHandler.dll/Default}} // would cause the MyOtherHandler.dll to be loaded. Once loaded, the stencil processor will // ask for the IReplacementHandler interface of the object named "Default" and use it in // processing the stencil anywhere it sees a stencil tag of the form // {{OtherHandler.RenderReplacement}} struct CStringPair { CStringA strDllPath; CStringA strHandlerName; CStringPair()throw() { } CStringPair(CStringA &strDllPath_, CStringA &strHandlerName_) throw(...) :strDllPath(strDllPath_), strHandlerName(strHandlerName_) { } }; class CStringPairElementTraits : public CElementTraitsBase< CStringPair > { public: static ULONG Hash( INARGTYPE pair ) throw() { return CStringElementTraits::Hash( pair.strDllPath ); } static bool CompareElements( INARGTYPE pair1, INARGTYPE pair2 ) throw() { return( (pair1.strDllPath == pair2.strDllPath) && (pair1.strHandlerName == pair2.strHandlerName) ); } static int CompareElementsOrdered( INARGTYPE pair1, INARGTYPE pair2 ) throw() { return( pair1.strDllPath.Compare( pair2.strDllPath ) ); } }; class CHtmlStencil : public CStencil { protected: CAtlMap, CStringPairElementTraits > m_arrExtraHandlers; CHAR m_szBaseDir[MAX_PATH]; CComPtr m_spServiceProvider; CComPtr m_spExtension; CComPtr m_spStencilCache; CComPtr m_spDllCache; public: typedef CAtlMap, CStringPairElementTraits > mapType; typedef CStencil baseType; CHtmlStencil(IAtlMemMgr *pMemMgr=NULL) throw() : CStencil(pMemMgr) { } void Initialize(IServiceProvider *pProvider) throw(...) { ATLASSERT(pProvider); if (m_spServiceProvider) m_spServiceProvider.Release(); m_spServiceProvider = pProvider; if (!m_spDllCache) pProvider->QueryService(__uuidof(IDllCache), __uuidof(IDllCache), (void **) &m_spDllCache); if (!m_spExtension) pProvider->QueryInterface(__uuidof(IIsapiExtension), (void **) &m_spExtension); } BOOL GetIncludeInfo(LPCSTR szParamBegin, LPCSTR szParamEnd, StencilIncludeInfo *pInfo) const throw() { LPCSTR szQueryBegin = szParamBegin; while (*szQueryBegin && *szQueryBegin != '?' && *szQueryBegin != '}') { LPSTR szNext = CharNextExA(GetCodePage(), szQueryBegin, 0); if (szNext == szQueryBegin) { return FALSE; } szQueryBegin = szNext; } char szPath[MAX_PATH]; szPath[0] = '\0'; DWORD dwPrefixLen = 0; if (*szParamBegin == '"') { szParamBegin++; } if (!IsFullPathA(szParamBegin)) { if (*szParamBegin != '\\') { strcpy(szPath, m_szBaseDir); } else { LPCSTR szBackslash = strchr(m_szBaseDir, '\\'); if (szBackslash) { strncpy(szPath, m_szBaseDir, szBackslash-m_szBaseDir); szPath[szBackslash-m_szBaseDir] = '\0'; } else { strcpy(szPath, m_szBaseDir); } } dwPrefixLen = (DWORD)strlen(szPath); } if (*szQueryBegin=='?') { size_t nMinus = (*(szQueryBegin-1) == '"') ? 1 : 0; strncat(szPath, szParamBegin, szQueryBegin-szParamBegin-nMinus); szPath[dwPrefixLen + szQueryBegin - szParamBegin] = '\0'; memcpy(pInfo->m_szQueryString, szQueryBegin + 1, szParamEnd - szQueryBegin); pInfo->m_szQueryString[szParamEnd - szQueryBegin] = '\0'; } else { pInfo->m_szQueryString[0] = '\0'; size_t nAdd = (*szParamEnd == '"') ? 0 : 1; strncat(szPath, szParamBegin, szParamEnd - szParamBegin + nAdd); szPath[dwPrefixLen + szParamEnd - szParamBegin + 1] = '\0'; } PathCanonicalizeA(pInfo->m_szFileName, szPath); return HTTP_SUCCESS; } int ParseInclude(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD * /*pBlockStack*/, DWORD * /*pdwTop*/) throw(...) { LPSTR szStart = szTokenStart; LPSTR szEnd = szTokenEnd; FindTagArgs(szStart, szEnd, 7); CHeapPtr szFileNameRelative; CHeapPtr szFileName; ATLTRY(szFileNameRelative.AllocateBytes(max(MAX_PATH, szEnd-szStart+3))) ATLTRY(szFileName.AllocateBytes(((szEnd-szStart)+MAX_PATH+3)*sizeof(TCHAR))) if ((char*)szFileNameRelative == NULL || (TCHAR*)szFileName == NULL) { AddError("Out of memory", szTokenStart); return AddToken(szTokenStart, szTokenEnd, STENCIL_TEXTTAG); } memcpy(szFileNameRelative, szStart, szEnd-szStart + 1); szFileNameRelative[szEnd-szStart + 1] = '\0'; if (!IsFullPathA(szFileNameRelative)) { CHeapPtr szTemp; ATLTRY(szTemp.AllocateBytes((szEnd-szStart)+MAX_PATH+3)); if ((CHAR*)szTemp == NULL) { AddError("Out of memory", szTokenStart); return AddToken(szTokenStart, szTokenEnd, STENCIL_TEXTTAG); } if (*szFileNameRelative != '\\') { strcpy(szTemp, m_szBaseDir); } else { LPCSTR szBackslash = strchr(m_szBaseDir, '\\'); if (szBackslash) { strncpy(szTemp, m_szBaseDir, szBackslash-m_szBaseDir); szTemp[szBackslash-m_szBaseDir] = '\0'; } else { strcpy(szTemp, m_szBaseDir); } } strcat(szTemp, szFileNameRelative); PathCanonicalize(szFileName, CA2CT(szTemp)); } else { _tcscpy(szFileName, CA2CTEX(szFileNameRelative)); } LPTSTR szDot = NULL; LPTSTR szExtra = _tcschr(szFileName, '?'); if (!szExtra) { szExtra = _tcschr(szFileName, '#'); if (!szExtra) { szDot = _tcsrchr(szFileName, '.'); } } if (szExtra != NULL) { // there is some extra information LPTSTR szDotTmp = szFileName; do { szDot = szDotTmp; szDotTmp = _tcschr(szDotTmp+1, '.'); } while (szDotTmp && szDotTmp < szExtra); } if (!szDot || *szDot != '.') { AddError("Unexpected error", szTokenStart); return AddToken(szTokenStart, szTokenEnd, STENCIL_TEXTTAG); } LPTSTR szExtEnd = szDot; while (true) { szExtEnd++; if (!*szExtEnd || *szExtEnd == '/' || *szExtEnd == '\\' || *szExtEnd == '?' || *szExtEnd == '#' || *szExtEnd == '"') break; } if (szDot && szExtEnd-szDot == sizeof(".dll")-sizeof('\0') && !_tcsnicmp(szDot, _T(".dll"), sizeof(".dll")-sizeof('\0'))) { // Do .dll stuff DWORD dwIndex = AddToken(szStart, szEnd, STENCIL_STENCILINCLUDE); StencilIncludeInfo *pInfo = (StencilIncludeInfo *)m_pMemMgr->Allocate(sizeof(StencilIncludeInfo)); if (!pInfo) return -1; GetIncludeInfo(szStart, szEnd, pInfo); // m_arrTokens[dwIndex].pvParam = pInfo; GetToken(dwIndex)->dwData = (DWORD_PTR) pInfo; return dwIndex; } else if (szDot && (size_t)(szExtEnd-szDot) == _tcslen(c_tAtlSRFExtension) && !_tcsnicmp(szDot, c_tAtlSRFExtension, _tcslen(c_tAtlSRFExtension))) { // Do .srf stuff DWORD dwIndex = AddToken(szStart, szEnd, STENCIL_STENCILINCLUDE); StencilIncludeInfo *pInfo = (StencilIncludeInfo *)m_pMemMgr->Allocate(sizeof(StencilIncludeInfo)); if (!pInfo) return -1; GetIncludeInfo(szStart, szEnd, pInfo); // m_arrTokens[dwIndex].pvParam = pInfo; GetToken(dwIndex)->dwData = (DWORD_PTR) pInfo; return dwIndex; } else { // Assume static content CAtlFile file; HRESULT hr = file.Create(szFileName, GENERIC_READ, FILE_SHARE_READ, OPEN_EXISTING); if (FAILED(hr) || GetFileType(file) != FILE_TYPE_DISK) { if (FAILED(hr)) AddError("Could not open included file", szTokenStart); else AddError("Included file is not a disk file", szTokenStart); return AddToken(szTokenStart, szTokenEnd, STENCIL_TEXTTAG); } CAutoVectorPtr szBufferStart; LPSTR szBufferEnd = NULL; ULONGLONG dwLen = 0; if (FAILED(file.GetSize(dwLen))) return AddToken(szTokenStart, szTokenEnd, STENCIL_TEXTTAG); if (!szBufferStart.Allocate((size_t) dwLen)) AtlThrow(E_OUTOFMEMORY); DWORD dwRead; if (FAILED(file.Read(szBufferStart, (DWORD) dwLen, dwRead))) return AddToken(szTokenStart, szTokenEnd, STENCIL_TEXTTAG); szBufferEnd = szBufferStart + dwRead-1; DWORD dwIndex = AddToken(szBufferStart, szBufferEnd, STENCIL_STATICINCLUDE); // m_arrTokens[dwIndex].bDynamicAlloc = TRUE; GetToken(dwIndex)->bDynamicAlloc = TRUE; szBufferStart.Detach(); return dwIndex; } } PARSE_TOKEN_RESULT ParseSubhandler(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD * /*pBlockStack*/, DWORD * /*pdwTop*/) throw() { LPSTR szStart = szTokenStart; LPSTR szEnd = szTokenEnd; // move to the start of the arguments // (the first char past 'subhandler' FindTagArgs(szStart, szEnd, 10); // skip any space to bring us to the start // of the id for the subhandler. szStart = SkipSpace(szStart, GetCodePage()); // id names cannot contain spaces. Mark the // beginning and end if the subhandler id LPCSTR szIdStart = szStart; while (!isspace(*szStart) && *szStart != '}') { LPSTR szNext = CharNextExA(GetCodePage(), szStart, 0); if (szNext == szStart) { // embedded null AddError("Embedded null character in stencil", NULL); return INVALID_TOKEN; } szStart = szNext; } LPCSTR szIdEnd = szStart; // skip space to bring us to the beginning of the // the dllpath/handlername szStart = SkipSpace(szStart, GetCodePage()); // everything up to the end if the tag is // part of the dllpath/handlername LPSTR szHandlerStart = szStart; while (*szStart != '}') { LPSTR szNext = CharNextExA(GetCodePage(), szStart, 0); if (szNext == szStart) { // embedded null AddError("Embedded null character in stencil", NULL); return INVALID_TOKEN; } szStart = szNext; } LPCSTR szHandlerEnd = szStart; _ATLTRY { CStringA strName(szIdStart, (int)(szIdEnd-szIdStart)); CStringA strPath(szHandlerStart, (int)(szHandlerEnd-szHandlerStart)); CStringA strDllPath; CStringA strHandlerName; DWORD dwDllPathLen = MAX_PATH+1; DWORD dwHandlerNameLen = ATL_MAX_HANDLER_NAME_LEN+1; LPSTR szDllPath = strDllPath.GetBuffer(dwDllPathLen); LPSTR szHandlerName = strHandlerName.GetBuffer(dwHandlerNameLen); if (!_AtlCrackHandler(strPath, szDllPath, &dwDllPathLen, szHandlerName, &dwHandlerNameLen)) { strDllPath.ReleaseBuffer(); strHandlerName.ReleaseBuffer(); return INVALID_TOKEN; } strDllPath.ReleaseBuffer(dwDllPathLen); strHandlerName.ReleaseBuffer(dwHandlerNameLen); m_arrExtraHandlers.SetAt(strName, CStringPair(strDllPath, strHandlerName)); } _ATLCATCHALL() { return INVALID_TOKEN; } return RESERVED_TOKEN; } virtual PARSE_TOKEN_RESULT ParseToken(LPSTR szTokenStart, LPSTR szTokenEnd, DWORD *pBlockStack, DWORD *pdwTop) throw() { LPSTR pStart = szTokenStart; pStart += 2; //skip curlies pStart = SkipSpace(pStart, GetCodePage()); DWORD dwLen = (DWORD)(szTokenEnd - szTokenStart); int nIndex = -1; if (CheckTag("include", 7, pStart, dwLen)) nIndex = ParseInclude(szTokenStart, szTokenEnd, pBlockStack, pdwTop); else if (dwLen > 3 && !memcmp("!--", pStart, 3)) return RESERVED_TOKEN; else if (dwLen > 2 && !memcmp("//", pStart, 2)) return RESERVED_TOKEN; else if (CheckTag("subhandler", 10, pStart, dwLen)) { return ParseSubhandler(szTokenStart, szTokenEnd, pBlockStack, pdwTop); } else { return CStencil::ParseToken(szTokenStart, szTokenEnd, pBlockStack, pdwTop); } if (nIndex < 0) return INVALID_TOKEN; return RESERVED_TOKEN; } mapType* GetExtraHandlers() throw() { return &m_arrExtraHandlers; } void SetBaseDirFromFile(LPCSTR szBaseDir) throw() { strcpy(m_szBaseDir, szBaseDir); LPSTR szSlash = strrchr(m_szBaseDir, '\\'); if (szSlash) { szSlash++; *szSlash = '\0'; } else { *m_szBaseDir = '\0'; } } LPCSTR GetBaseDir() throw() { return m_szBaseDir; } // Any value returned from a replacement method other than HTTP_SUCCESS // will stop the rendering of the stencil. DWORD RenderToken( DWORD dwIndex, ITagReplacer* pReplacer, IWriteStream *pWriteStream, HTTP_CODE *phcErrorCode, void* pState = NULL) const throw(...) { DWORD dwNextToken = STENCIL_INVALIDINDEX; HTTP_CODE hcErrorCode = HTTP_SUCCESS; const StencilToken* pToken = GetToken(dwIndex); if (pToken) { if (pToken->type == STENCIL_STENCILINCLUDE) { ATLASSERT(m_spServiceProvider); CComPtr spServerContext; CComPtr spLookup; if (FAILED(pReplacer->GetContext(__uuidof(IHttpServerContext), (VOID**) &spServerContext))) return HTTP_ERROR(500, 0); if (FAILED(pReplacer->GetContext(__uuidof(IHttpRequestLookup), (VOID**) &spLookup))) return HTTP_ERROR(500, 0); hcErrorCode = RenderInclude(m_spServiceProvider, pWriteStream, (StencilIncludeInfo *)pToken->dwData, spServerContext, spLookup, pState); if (hcErrorCode == HTTP_SUCCESS || IsAsyncDoneStatus(hcErrorCode)) dwNextToken = dwIndex+1; else if (IsAsyncContinueStatus(hcErrorCode)) dwNextToken = dwIndex; } else if (pToken->type == STENCIL_STATICINCLUDE) { pWriteStream->WriteStream(pToken->pStart, (int)((pToken->pEnd-pToken->pStart)+1), NULL); dwNextToken = dwIndex+1; } else dwNextToken = baseType::RenderToken(dwIndex, pReplacer, pWriteStream, &hcErrorCode, pState); } if (hcErrorCode == HTTP_SUCCESS_NO_CACHE) { hcErrorCode = HTTP_SUCCESS; CComPtr spContext; HRESULT hr = pReplacer->GetContext(__uuidof(IHttpServerContext), (void **)&spContext); if (hr == S_OK && spContext) { CComQIPtr spControl; spControl = spContext; if (spControl) spControl->Cache(FALSE); } } if (phcErrorCode) *phcErrorCode = hcErrorCode; return dwNextToken; } ATL_NOINLINE HTTP_CODE RenderInclude( IServiceProvider *pServiceProvider, IWriteStream *pWriteStream, const StencilIncludeInfo *pIncludeInfo, IHttpServerContext *pServerContext, IHttpRequestLookup *pLookup, void* pState = NULL) const throw(...) { AtlServerRequest* pRequestInfo = NULL; HTTP_CODE hcErr = HTTP_SUCCESS; CComObjectStackEx serverContext; serverContext.Initialize(pWriteStream, pServerContext, pIncludeInfo); #ifdef _DEBUG bool bAsyncAllowed = false; #endif CStencilState* _pState = reinterpret_cast(pState); if (_pState && _pState->pIncludeInfo) { pRequestInfo = _pState->pIncludeInfo; _pState->pIncludeInfo = NULL; #ifdef _DEBUG bAsyncAllowed = true; #endif } else { if (!m_spStencilCache) m_spServiceProvider->QueryService(__uuidof(IStencilCache), __uuidof(IStencilCache), (void **) &m_spStencilCache); ATLASSERT(m_spStencilCache); if (!m_spDllCache) pServiceProvider->QueryService(__uuidof(IDllCache), __uuidof(IDllCache), (void **) &m_spDllCache); ATLASSERT(m_spDllCache); pRequestInfo = m_spExtension->CreateRequest(); if (pRequestInfo == NULL) return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); pRequestInfo->dwRequestState = ATLSRV_STATE_BEGIN; pRequestInfo->dwRequestType = ATLSRV_REQUEST_STENCIL; pRequestInfo->pDllCache = m_spDllCache; pRequestInfo->pExtension = m_spExtension; pRequestInfo->pServerContext = &serverContext; if (_pState && _pState->pParentInfo) pRequestInfo->pUserData = _pState->pParentInfo->pUserData; // Extract the file extension of the included file by searching // for the first '.' from the right. // Can't use _tcsrchr because we have to use the stencil's codepage LPCSTR szDot = NULL; LPCSTR szMark = pIncludeInfo->m_szFileName; while (*szMark) { if (*szMark == '.') szDot = szMark; LPCSTR szNext = CharNextExA(GetCodePage(), szMark, 0); if (szNext == szMark) { // embedded null return HTTP_FAIL; } szMark = szNext; } if (szDot && _stricmp(szDot, c_AtlSRFExtension) == 0) { hcErr = m_spExtension->LoadDispatchFile(pIncludeInfo->m_szFileName, pRequestInfo); if (hcErr) return hcErr; } else if (szDot && _stricmp(szDot, ".dll") == 0) { // Get the handler name if they used the asdf.dll?Handler=Default notation // REVIEW : case sensitivity on the "Handler"? char szHandlerName[ATL_MAX_HANDLER_NAME_LEN+1] = { '\0' }; LPSTR szStart = strstr(pIncludeInfo->m_szQueryString, "Handler="); if (szStart && ((szStart == pIncludeInfo->m_szQueryString) || ((szStart > pIncludeInfo->m_szQueryString) && (*(szStart-1) == '&')))) { szStart += 8; // Skip past "Handler" and the "=" LPSTR szEnd = strchr(szStart, '&'); if (szEnd) { memcpy(szHandlerName, szStart, min((szEnd-szStart), ATL_MAX_HANDLER_NAME_LEN)); szHandlerName[min((szEnd-szStart), ATL_MAX_HANDLER_NAME_LEN)] = '\0'; } else { strcpy(szHandlerName, szStart); } } else { memcpy(szHandlerName, "Default", sizeof("Default")); } pRequestInfo->dwRequestType = ATLSRV_REQUEST_DLL; hcErr = m_spExtension->LoadRequestHandler(pIncludeInfo->m_szFileName, szHandlerName, pRequestInfo->pServerContext, &pRequestInfo->hInstDll, &pRequestInfo->pHandler); if (hcErr) return hcErr; } DWORD dwStatus; hcErr = pRequestInfo->pHandler->GetFlags(&dwStatus); if (hcErr) return hcErr; if (dwStatus & (ATLSRV_INIT_USEASYNC | ATLSRV_INIT_USEASYNC_EX)) { #ifdef _DEBUG bAsyncAllowed = true; #endif CComObjectNoLock* pNewServerContext = NULL; ATLTRY(pNewServerContext = new CComObjectNoLock); if (pNewServerContext == NULL) return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); pNewServerContext->Initialize(pWriteStream, pServerContext, pIncludeInfo); pNewServerContext->AddRef(); pRequestInfo->pServerContext = pNewServerContext; } _ATLTRY { hcErr = pRequestInfo->pHandler->InitializeChild(pRequestInfo, m_spServiceProvider, pLookup); if (hcErr) return hcErr; } _ATLCATCHALL() { return HTTP_FAIL; } pRequestInfo->pfnHandleRequest = &IRequestHandler::HandleRequest; } if (pRequestInfo) { if (!hcErr) { if (pRequestInfo->pServerContext == NULL) pRequestInfo->pServerContext = &serverContext; _ATLTRY { ATLASSERT(pRequestInfo->pfnHandleRequest != NULL); hcErr = (pRequestInfo->pHandler->*pRequestInfo->pfnHandleRequest)(pRequestInfo, pServiceProvider); } _ATLCATCHALL() { hcErr = HTTP_FAIL; } if (pRequestInfo->pServerContext == &serverContext) pRequestInfo->pServerContext = NULL; #ifdef _DEBUG // must use ATLSRV_INIT_USEASYNC to use ASYNC returns if (IsAsyncStatus(hcErr)) ATLASSERT(bAsyncAllowed); #endif if (IsAsyncStatus(hcErr)) { ATLASSERT(pState); // state is required for async if (IsAsyncContinueStatus(hcErr)) { _pState->pIncludeInfo = pRequestInfo; pRequestInfo->dwRequestState = ATLSRV_STATE_CONTINUE; } else if (IsAsyncDoneStatus(hcErr)) m_spExtension->FreeRequest(pRequestInfo); } else m_spExtension->FreeRequest(pRequestInfo); } } else hcErr = HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); return hcErr; } }; // class CHtmlStencil __declspec(selectany) CCRTHeap CStencil::m_crtHeap; // // CHtmlTagReplacer // This class manages CStencil based objects for HTTP requests. This class will retrieve // CStencil based objects from the stencil cache, store CStencil based objects in the // stencil cache and allocate and initialize CStencil based objects on a per reqeust // basis. Typically, one instance of this class is created for each HTTP request. The // instance is destroyed once the request has been completed. template class CHtmlTagReplacer : public ITagReplacerImpl { protected: typedef StencilType StencilType; CSimpleArray m_hInstHandlers; typedef CAtlMap > mapType; mapType m_Handlers; StencilType *m_pLoadedStencil; WORD m_nCodePage; CComPtr m_spStencilCache; AtlServerRequest m_RequestInfo; public: // public members CHtmlTagReplacer() throw() : m_pLoadedStencil(NULL) { memset(&m_RequestInfo, 0x00, sizeof(m_RequestInfo)); m_nCodePage = CP_THREAD_ACP; } ~CHtmlTagReplacer() throw() { // you should call FreeHandlers before // the object is destructed ATLASSERT(m_hInstHandlers.GetSize() == 0); } HTTP_CODE Initialize(AtlServerRequest *pRequestInfo, IHttpServerContext *pSafeSrvCtx=NULL) throw(...) { CComPtr spServiceProvider; THandler *pT = static_cast(this); HRESULT hr = pT->GetContext(__uuidof(IServiceProvider), (void **)&spServiceProvider); if (FAILED(hr)) return HTTP_FAIL; spServiceProvider->QueryService(__uuidof(IStencilCache), __uuidof(IStencilCache), (void **) &m_spStencilCache); if (!m_spStencilCache) { ATLASSERT(FALSE); return HTTP_FAIL; } m_RequestInfo.cbSize = sizeof(m_RequestInfo); m_RequestInfo.pServerContext = pSafeSrvCtx; m_RequestInfo.pUserData = pRequestInfo->pUserData; return HTTP_SUCCESS; } HTTP_CODE LoadStencilResource( HINSTANCE hInstResource, LPCSTR szResourceID, LPCSTR szResourceType = NULL, LPCSTR szStencilName=NULL) throw(...) { if (!szResourceType) szResourceType = (LPCSTR) (RT_HTML); // look up stencil in cache HTTP_CODE hcErr = HTTP_SUCCESS; // check the cache first StencilType *pStencil = FindCacheStencil(szStencilName ? szStencilName : szResourceID); if (!pStencil) { // create a new stencil pStencil = GetNewCacheStencil(); THandler *pT = static_cast(this); LPCSTR szFileName = pT->m_spServerContext->GetScriptPathTranslated(); if (!szFileName) return HTTP_FAIL; pStencil->SetBaseDirFromFile(szFileName); if (!pStencil) return HTTP_ERROR(500,ISE_SUBERR_OUTOFMEM); // load the stencil and parse its replacements if (HTTP_SUCCESS == pStencil->LoadFromResource(hInstResource, szResourceID, szResourceType)) { _ATLTRY { if (!pStencil->ParseReplacements(static_cast(this), pStencil->GetBufferStart(), pStencil->GetBufferEnd())) return HTTP_ERROR(500, ISE_SUBERR_BADSRF); hcErr = FinishLoadStencil(pStencil, NULL); if (!hcErr) { pStencil->FinishParseReplacements(); } } _ATLCATCHALL() { return HTTP_FAIL; } } else { hcErr = HTTP_FAIL; } // if everything went OK, put the stencil in the stencil cache. if (!hcErr) { hcErr = CacheStencil(szStencilName ? szStencilName : szResourceID, pStencil); } if (pStencil && hcErr) // something went wrong, free the stencil data FreeCacheStencil(pStencil); } else { hcErr = FinishLoadStencil(pStencil); } return hcErr; } HTTP_CODE LoadStencilResource(HINSTANCE hInstResource, UINT nID, LPCSTR szResourceType = NULL) throw(...) { if (!szResourceType) szResourceType = (LPCSTR) RT_HTML; char szName[80]; sprintf(szName, "%p/%u", hInstResource, nID); return LoadStencilResource(hInstResource, MAKEINTRESOURCEA(nID), szResourceType, szName); } HTTP_CODE LoadStencil(LPCSTR szFileName, IHttpRequestLookup * pLookup = NULL) throw(...) { if (!szFileName) { return HTTP_FAIL; } HTTP_CODE hcErr = HTTP_FAIL; // try to find the stencil in the cache StencilType *pStencil = FindCacheStencil(szFileName); if (!pStencil) { // not in cache. Create a new one pStencil = GetNewCacheStencil(); if (!pStencil) { return HTTP_ERROR(500, ISE_SUBERR_OUTOFMEM); // out of memory! } pStencil->SetBaseDirFromFile(szFileName); // finish loading hcErr = pStencil->LoadFile(szFileName); if ( !hcErr ) { _ATLTRY { if (!pStencil->ParseReplacements(static_cast(this), pStencil->GetBufferStart(), pStencil->GetBufferEnd())) return HTTP_ERROR(500, ISE_SUBERR_BADSRF); hcErr = FinishLoadStencil(pStencil, pLookup); if (!hcErr) { pStencil->FinishParseReplacements(); } } _ATLCATCHALL() { return HTTP_FAIL; } } // if everything is OK, cache the stencil if (!hcErr) { hcErr = CacheStencil(szFileName, pStencil); } if (pStencil && hcErr) // something went wrong, free stencil data FreeCacheStencil(pStencil); } else { hcErr = FinishLoadStencil(pStencil, pLookup); } return hcErr; } HTTP_CODE RenderStencil(IWriteStream* pStream, void* pState = NULL) throw(...) { if (!m_pLoadedStencil) return HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); WORD nCodePage = m_pLoadedStencil->GetCodePage(); if (nCodePage != CP_ACP) m_nCodePage = nCodePage; HTTP_CODE hcErr = HTTP_FAIL; hcErr = m_pLoadedStencil->Render(static_cast(this), pStream, pState); if (!IsAsyncStatus(hcErr) && m_pLoadedStencil->GetCacheItem()) m_spStencilCache->ReleaseStencil(m_pLoadedStencil->GetCacheItem()); return hcErr; } HTTP_CODE AddHandler(LPCSTR szName, IRequestHandler *pHandler, HINSTANCE hInstHandler) throw() { _ATLTRY { ATLASSERT(szName); ATLASSERT(pHandler); ATLASSERT(hInstHandler); HTTP_CODE hcErr = HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); if (szName && *szName && pHandler) { if (m_Handlers.SetAt(szName, pHandler)) { if (hInstHandler && m_hInstHandlers.Add(hInstHandler)) hcErr = HTTP_SUCCESS; } } return hcErr; } _ATLCATCHALL() { return HTTP_FAIL; } } void SendHandlerError(LPCSTR szHandlerName) throw() { #ifndef _ATL_STENCIL_SLIENT_ERRORS m_pStream->WriteStream("Handler "); m_pStream->WriteStream(szHandlerName); m_pStream->WriteStream(" was not found"); #endif } //Implementation void FreeHandlers() throw(...) { POSITION pos = m_Handlers.GetStartPosition(); while (pos) { m_Handlers.GetNextValue(pos)->Release(); } m_Handlers.RemoveAll(); int nLen = m_hInstHandlers.GetSize(); THandler *pT = static_cast(this); CComPtr spDllCache; pT->m_spServiceProvider->QueryService(__uuidof(IDllCache), __uuidof(IDllCache), (void **)&spDllCache); for (int i=0; iFree(m_hInstHandlers[i]); } m_hInstHandlers.RemoveAll(); } StencilType* GetNewCacheStencil() throw(...) { StencilType *pStencil = NULL; THandler *pT = static_cast(this); IAtlMemMgr *pMemMgr; if (FAILED(pT->m_spServiceProvider->QueryService(__uuidof(IAtlMemMgr), __uuidof(IAtlMemMgr), (void **)&pMemMgr))) pMemMgr = NULL; ATLTRY(pStencil = new StencilType(pMemMgr)); if (pStencil != NULL) pStencil->Initialize(pT->m_spServiceProvider); return pStencil; } HTTP_CODE CacheStencil( LPCSTR szName, StencilType* pStencilData) throw() { THandler *pT = static_cast(this); HRESULT hr = E_FAIL; HCACHEITEM hCacheItem = NULL; hr = m_spStencilCache->CacheStencil(szName, pStencilData, sizeof(StencilType*), &hCacheItem, pT->m_hInstHandler, static_cast(pStencilData)); if (hr == S_OK && hCacheItem) pStencilData->SetCacheItem(hCacheItem); return (hr == S_OK) ? HTTP_SUCCESS : HTTP_FAIL; } StencilType *FindCacheStencil(LPCSTR szName) throw() { if (!szName || !m_spStencilCache) return NULL; StencilType *pStencilData = NULL; HCACHEITEM hStencil; if (m_spStencilCache->LookupStencil(szName, &hStencil) != S_OK) return NULL; m_spStencilCache->GetStencil(hStencil, reinterpret_cast(&pStencilData)); return pStencilData; } void FreeCacheStencil(StencilType* pStencilData) throw() { CComPtr spCacheClient; if (m_spStencilCache->QueryInterface(__uuidof(IMemoryCacheClient), (void **) &spCacheClient) == S_OK) spCacheClient->Free(pStencilData); } HTTP_CODE GetHandlerOffset(LPCSTR szHandlerName, DWORD* pdwOffset) throw() { if (!pdwOffset) return HTTP_FAIL; mapType::CPair *p = m_Handlers.Lookup(szHandlerName); if (p) { DWORD dwIndex = 0; POSITION pos = m_Handlers.GetStartPosition(); while (pos) { const mapType::CPair *p1 = m_Handlers.GetNext(pos); if (p1 == p) { *pdwOffset = dwIndex; return HTTP_SUCCESS; } dwIndex++; } ATLASSERT(FALSE); return HTTP_SUCCESS; } *pdwOffset = 0; return HTTP_FAIL; } HTTP_CODE GetReplacementObject(DWORD dwObjOffset, ITagReplacer **ppReplacer) throw(...) { HRESULT hr = E_FAIL; POSITION pos = m_Handlers.GetStartPosition(); for (DWORD dwIndex=0; dwIndex < dwObjOffset; dwIndex++) m_Handlers.GetNext(pos); ATLASSERT(pos != NULL); IRequestHandler *pHandler = NULL; pHandler = m_Handlers.GetValueAt(pos); ATLASSERT(pHandler != NULL); hr = pHandler->QueryInterface(__uuidof(ITagReplacer), (void**)ppReplacer); if (hr != S_OK) return HTTP_FAIL; return HTTP_SUCCESS; } // This is where we would actually load any extra request // handlers the HTML stencil might have parsed for us. HTTP_CODE FinishLoadStencil(StencilType *pStencil, IHttpRequestLookup * pLookup = NULL) throw(...) { THandler *pT = static_cast(this); ATLASSERT(pStencil); if (!pStencil) return HTTP_ERROR(500, ISE_SUBERR_UNEXPECTED); // unexpected condition m_pLoadedStencil = pStencil; //load extra handlers if there are any StencilType::mapType *pExtraHandlersMap = pStencil->GetExtraHandlers(); if (pExtraHandlersMap) { POSITION pos = pExtraHandlersMap->GetStartPosition(); CStringA name; CStringPair path; IRequestHandler *pHandler; HINSTANCE hInstHandler; while(pos) { pExtraHandlersMap->GetNextAssoc(pos, name, path); pHandler = NULL; hInstHandler = NULL; HTTP_CODE hcErr = pT->m_spExtension->LoadRequestHandler(path.strDllPath, path.strHandlerName, pT->m_spServerContext, &hInstHandler, &pHandler); if (!hcErr) { _ATLTRY { //map the name to the pointer to request handler m_Handlers.SetAt(name, pHandler); //store HINSTANCE of handler m_hInstHandlers.Add(hInstHandler); } _ATLCATCHALL() { return HTTP_FAIL; } if (pLookup) { hcErr = pHandler->InitializeChild(&m_RequestInfo, pT->m_spServiceProvider, pLookup); if (hcErr != HTTP_SUCCESS) return hcErr; } } else return hcErr; } } return HTTP_SUCCESS; } }; // class CHtmlTagReplacer // CRequestHandlerT // This is the base class for all user request handlers. This class implements // the IReplacementHandler interface whose methods will be called to render HTML // into a stream. The stream will be returned as the HTTP response upon completion // of the HTTP request. template < class THandler, class ThreadModel=CComSingleThreadModel, class TagReplacerType=CHtmlTagReplacer > class CRequestHandlerT : public TagReplacerType, public CComObjectRootEx, public IRequestHandlerImpl { protected: CStencilState m_state; CComObjectStackEx m_SafeSrvCtx; typedef CRequestHandlerT _requestHandler; public: BEGIN_COM_MAP(_requestHandler) COM_INTERFACE_ENTRY(IRequestHandler) COM_INTERFACE_ENTRY(ITagReplacer) END_COM_MAP() // public CRequestHandlerT members CHttpResponse m_HttpResponse; CHttpRequest m_HttpRequest; ATLSRV_REQUESTTYPE m_dwRequestType; AtlServerRequest* m_pRequestInfo; CRequestHandlerT() throw() { m_hInstHandler = NULL; m_dwAsyncFlags = 0; m_pRequestInfo = NULL; } ~CRequestHandlerT() throw() { FreeHandlers(); // free handlers held by CTagReplacer } void ClearResponse() throw() { m_HttpResponse.ClearResponse(); } // Where user initialization should take place HTTP_CODE ValidateAndExchange() throw() { return HTTP_SUCCESS; // continue processing request } // Where user Uninitialization should take place HTTP_CODE Uninitialize(HTTP_CODE hcError) throw() { return hcError; } HTTP_CODE InitializeInternal(AtlServerRequest *pRequestInfo, IServiceProvider *pProvider) throw() { // Initialize our internal references to required services m_pRequestInfo = pRequestInfo; m_state.pParentInfo = pRequestInfo; m_hInstHandler = pRequestInfo->hInstDll; m_spServerContext = pRequestInfo->pServerContext; m_spServiceProvider = pProvider; return HTTP_SUCCESS; } HTTP_CODE InitializeHandler( AtlServerRequest *pRequestInfo, IServiceProvider *pProvider) throw(...) { HTTP_CODE hcErr = HTTP_FAIL; ATLASSERT(pRequestInfo); ATLASSERT(pProvider); THandler* pT = static_cast(this); hcErr = pT->InitializeInternal(pRequestInfo, pProvider); if (!hcErr) { m_HttpResponse.Initialize(m_spServerContext); hcErr = pT->CheckValidRequest(); if (!hcErr) { hcErr = HTTP_FAIL; if (m_HttpRequest.Initialize(m_spServerContext, pT->MaxFormSize(), pT->FormFlags())) { if (m_SafeSrvCtx.Initialize(&m_HttpResponse, &m_HttpRequest)) { hcErr = TagReplacerType::Initialize(pRequestInfo, &m_SafeSrvCtx); if (!hcErr) { hcErr = pT->ValidateAndExchange(); } } } } } return hcErr; } HTTP_CODE InitializeChild( AtlServerRequest *pRequestInfo, IServiceProvider *pProvider, IHttpRequestLookup *pRequestLookup) throw(...) { ATLASSERT(pRequestInfo); ATLASSERT(pProvider); THandler *pT = static_cast(this); HTTP_CODE hcErr = pT->InitializeInternal(pRequestInfo, pProvider); if (hcErr) return hcErr; if (pRequestLookup) { // initialize with the pRequestLookup m_HttpResponse.Initialize(pRequestLookup); // REVIEW: Initialize with the IHttpServerContext if it exists // the only time this is different than the previous call to // initialize is if the user passes a different IHttpServerContext // in pRequestInfo than the one extracted from pRequestLookup. if (m_spServerContext) m_HttpResponse.Initialize(m_spServerContext); hcErr = pT->CheckValidRequest(); if (hcErr) return hcErr; // initialize with the pRequestLookup to chain query parameters m_HttpRequest.Initialize(pRequestLookup); // initialize with the m_spServerContext to get additional query params // if they exist. if (m_spServerContext) m_HttpRequest.Initialize(m_spServerContext); } m_HttpResponse.SetBufferOutput(false); // child cannot buffer // initialize the safe server context // REVIEW: necessary? if (!m_SafeSrvCtx.Initialize(&m_HttpResponse, &m_HttpRequest)) return HTTP_FAIL; hcErr = TagReplacerType::Initialize(pRequestInfo, &m_SafeSrvCtx); if (hcErr) return hcErr; return pT->ValidateAndExchange(); } // HandleRequest is called to perform default processing of HTTP requests. Users // can override this function in their derived classes if they need to perform // specific initialization prior to processing this request or want to change the // way the request is processed. HTTP_CODE HandleRequest( AtlServerRequest *pRequestInfo, IServiceProvider* /*pServiceProvider*/) throw(...) { ATLASSERT(pRequestInfo); THandler *pT = static_cast(this); HTTP_CODE hcErr = HTTP_SUCCESS; if (pRequestInfo->dwRequestState == ATLSRV_STATE_BEGIN) { m_dwRequestType = pRequestInfo->dwRequestType; if (pRequestInfo->dwRequestType==ATLSRV_REQUEST_STENCIL) { LPCSTR szFileName = pRequestInfo->pServerContext->GetScriptPathTranslated(); hcErr = HTTP_FAIL; if (szFileName) hcErr = pT->LoadStencil(szFileName, static_cast(&m_HttpRequest)); } } else if (pRequestInfo->dwRequestState == ATLSRV_STATE_CONTINUE) m_HttpResponse.ClearContent(); #ifdef ATL_DEBUG_STENCILS if (m_pLoadedStencil && !m_pLoadedStencil->ParseSuccessful()) { // An error or series of errors occurred in parsing the stencil m_pLoadedStencil->RenderErrors(static_cast(&m_HttpResponse)); } #endif if (hcErr == HTTP_SUCCESS && m_pLoadedStencil) { // if anything other than HTTP_SUCCESS is returned during // the rendering of replacement tags, we return that value // here. hcErr = pT->RenderStencil(static_cast(&m_HttpResponse), &m_state); if (hcErr == HTTP_SUCCESS && !m_HttpResponse.Flush(TRUE)) hcErr = HTTP_FAIL; } if (IsAsyncFlushStatus(hcErr)) { pRequestInfo->pszBuffer = LPCSTR(m_HttpResponse.m_strContent); pRequestInfo->dwBufferLen = m_HttpResponse.m_strContent.GetLength(); } if (pRequestInfo->dwRequestState == ATLSRV_STATE_BEGIN || IsAsyncDoneStatus(hcErr)) return pT->Uninitialize(hcErr); else if (!IsAsyncStatus(hcErr)) m_HttpResponse.ClearContent(); return hcErr; } HTTP_CODE ServerTransferRequest(LPCSTR szRequest, bool bContinueAfterTransfer=false, int nCodePage = 0, void *pState = NULL) throw(...) { return m_spExtension->TransferRequest( m_pRequestInfo, m_spServiceProvider, static_cast(&m_HttpResponse), static_cast(&m_HttpRequest), szRequest, nCodePage == 0 ? m_nCodePage : nCodePage, bContinueAfterTransfer, pState); } inline DWORD MaxFormSize() throw() { return DEFAULT_MAX_FORM_SIZE; } inline DWORD FormFlags() throw() { return ATL_FORM_FLAG_IGNORE_FILES; } // Override this function to check if the request // is valid. This function is called after m_HttpResponse // has been initialized, so you can use it if you need // to return an error to the client. This is also a // good place to initialize any internal class data needed // to handle the request. CRequestHandlerT::CheckValidRequest // is called after CRequestHandlerT::InitializeInternal is // called, so your override of this method will have access to // m_pRequestInfo (this request's AtlServerRequest structure), // m_hInstHandler (the HINSTANCE of this handler dll), // m_spServerContext (the IHttpServerContext interface for this request), // m_spServiceProvider (the IServiceProvider interface for this request). // You should call CRequestHandlerT::CheckValidRequest in your override // if you override this function. // // Note that m_HttpRequest has not been initialized, so // you cannot use it. This function is intended to // do simple checking throught IHttpServerContext to avoid // expensive initialization of m_HttpRequest. HTTP_CODE CheckValidRequest() throw() { LPCSTR szMethod = NULL; ATLASSERT(m_pRequestInfo); szMethod = m_pRequestInfo->pServerContext->GetRequestMethod(); if (strcmp(szMethod, "GET") && strcmp(szMethod, "POST") && strcmp(szMethod, "HEAD")) return HTTP_NOT_IMPLEMENTED; return HTTP_SUCCESS; } HRESULT GetContext(REFIID riid, void** ppv) throw() { if (!ppv) return E_POINTER; if (InlineIsEqualGUID(riid, __uuidof(IHttpServerContext))) { return m_spServerContext.CopyTo((IHttpServerContext **)ppv); } if (InlineIsEqualGUID(riid, __uuidof(IHttpRequestLookup))) { *ppv = static_cast(&m_HttpRequest); m_HttpRequest.AddRef(); return S_OK; } if (InlineIsEqualGUID(riid, __uuidof(IServiceProvider))) { *ppv = m_spServiceProvider; m_spServiceProvider.p->AddRef(); return S_OK; } return E_NOINTERFACE; } template HRESULT GetContext(Interface** ppInterface) throw() { return GetContext(__uuidof(Interface), reinterpret_cast(ppInterface)); } }; // class CRequestHandlerT } // namespace ATL #pragma warning( pop ) #endif // __ATLSTENCIL_H__