国外知名自动更新软件Updater源代码

// Markup.cpp: implementation of the CMarkup class. // // Markup Release 8.1 // Copyright (C) 1999-2005 First Objective Software, Inc. All rights reserved // Go to www.firstobject.com for the latest CMarkup and EDOM documentation // Use in commercial applications requires written permission // This software is provided "as is", with no warranty. #include "stdafx.h" #include "Markup.h" #ifdef _DEBUG #undef THIS_FILE static char THIS_FILE[]=__FILE__; #define new DEBUG_NEW #endif #ifdef _MBCS #pragma message( "Note: MBCS build (not UTF-8)" ) // For UTF-8, remove _MBCS from project settings C/C++ preprocessor definitions #endif // Defines for Windows CE #ifdef _WIN32_WCE #define _tclen(p) 1 #define _tccpy(p1,p2) *(p1)=*(p2) #endif // Customization #define x_EOL _T("\r\n") // can be \r\n or \n or empty #define x_EOLLEN (sizeof(x_EOL)/sizeof(_TCHAR)-1) // string length of x_EOL #define x_ATTRIBQUOTE _T("\"") // can be double or single quote void CMarkup::operator=( const CMarkup& markup ) { m_iPosParent = markup.m_iPosParent; m_iPos = markup.m_iPos; m_iPosChild = markup.m_iPosChild; m_iPosFree = markup.m_iPosFree; m_iPosDeleted = markup.m_iPosDeleted; m_nNodeType = markup.m_nNodeType; m_nNodeOffset = markup.m_nNodeOffset; m_nNodeLength = markup.m_nNodeLength; m_strDoc = markup.m_strDoc; m_strError = markup.m_strError; m_nFlags = markup.m_nFlags; // Copy used part of the index array m_aPos.RemoveAll(); m_aPos.nSize = m_iPosFree; if ( m_aPos.nSize < 8 ) m_aPos.nSize = 8; m_aPos.nSegs = m_aPos.SegsUsed(); if ( m_aPos.nSegs ) { m_aPos.pSegs = (ElemPos**)(new char[m_aPos.nSegs*sizeof(char*)]); int nSegSize = 1 << m_aPos.PA_SEGBITS; for ( int nSeg=0; nSeg < m_aPos.nSegs; ++nSeg ) { if ( nSeg + 1 == m_aPos.nSegs ) nSegSize = m_aPos.GetSize() - (nSeg << m_aPos.PA_SEGBITS); m_aPos.pSegs[nSeg] = (ElemPos*)(new char[nSegSize*sizeof(ElemPos)]); memcpy( m_aPos.pSegs[nSeg], markup.m_aPos.pSegs[nSeg], nSegSize*sizeof(ElemPos) ); } } // Copy SavedPos map m_mapSavedPos.RemoveAll(); if ( markup.m_mapSavedPos.pTable ) { m_mapSavedPos.AllocMapTable(); for ( int nSlot=0; nSlot < SavedPosMap::SPM_SIZE; ++nSlot ) { SavedPos* pCopySavedPos = markup.m_mapSavedPos.pTable[nSlot]; if ( pCopySavedPos ) { int nCount = 0; while ( pCopySavedPos[nCount].nSavedPosFlags & SavedPosMap::SPM_USED ) { ++nCount; if ( pCopySavedPos[nCount-1].nSavedPosFlags & SavedPosMap::SPM_LAST ) break; } SavedPos* pNewSavedPos = new SavedPos[nCount]; for ( int nCopy=0; nCopy<nCount; ++nCopy ) pNewSavedPos[nCopy] = pCopySavedPos[nCopy]; pNewSavedPos[nCount-1].nSavedPosFlags |= SavedPosMap::SPM_LAST; m_mapSavedPos.pTable[nSlot] = pNewSavedPos; } } } MARKUP_SETDEBUGSTATE; } bool CMarkup::SetDoc( LPCTSTR szDoc ) { // Set document text if ( szDoc ) m_strDoc = szDoc; else m_strDoc.Empty(); m_strError.Empty(); return x_ParseDoc(); }; bool CMarkup::IsWellFormed() { if ( m_aPos.GetSize() && ! (m_aPos[0].nFlags & MNF_ILLFORMED) && m_aPos[0].iElemChild && ! m_aPos[m_aPos[0].iElemChild].iElemNext ) return true; return false; } bool CMarkup::Load( LPCTSTR szFileName ) { if ( ! ReadTextFile(szFileName, m_strDoc, &m_strError, &m_nFlags) ) return false; return x_ParseDoc(); } bool CMarkup::ReadTextFile( LPCTSTR szFileName, CString& strDoc, CString* pstrError, int* pnFlags ) { // Static utility method to load text file into strDoc // // Open file to read binary FILE* fp = _tfopen( szFileName, _T("rb") ); if ( ! fp ) { if ( pstrError ) *pstrError = strerror(errno); return false; } // Set flags to 0 unless flags argument provided int nFlags = pnFlags?*pnFlags:0; _TCHAR szDescBOM[20] = {0}; strDoc.Empty(); // Get file length fseek( fp, 0, SEEK_END ); int nFileByteLen = ftell( fp ); fseek( fp, 0, SEEK_SET ); #if defined(_UNICODE) // convert file to wide char int nWideLen = 0; if ( nFileByteLen ) { char* pBuffer = new char[nFileByteLen]; fread( pBuffer, nFileByteLen, 1, fp ); // For ANSI files, replace CP_UTF8 with CP_ACP in both places nWideLen = MultiByteToWideChar(CP_UTF8,0,pBuffer,nFileByteLen,NULL,0); MultiByteToWideChar(CP_UTF8,0,pBuffer,nFileByteLen,strDoc.GetBuffer(nWideLen),nWideLen); strDoc.ReleaseBuffer( nWideLen ); delete [] pBuffer; } if ( pstrError ) (*pstrError).Format(_T("%s%d bytes to %d wide chars"),szDescBOM,nFileByteLen,nWideLen); #else // read file directly if ( nFileByteLen ) { fread( strDoc.GetBuffer(nFileByteLen), nFileByteLen, 1, fp ); strDoc.ReleaseBuffer( nFileByteLen ); } if ( pstrError ) (*pstrError).Format( _T("%s%d bytes"), szDescBOM, nFileByteLen ); #endif fclose( fp ); if ( pnFlags ) *pnFlags = nFlags; return true; } bool CMarkup::Save( LPCTSTR szFileName ) { return WriteTextFile( szFileName, m_strDoc, &m_strError, &m_nFlags ); } bool CMarkup::WriteTextFile( LPCTSTR szFileName, CString& strDoc, CString* pstrError, int* pnFlags ) { // Static utility method to save strDoc to text file // // Open file to write binary bool bSuccess = true; FILE* fp = _tfopen( szFileName, _T("wb") ); if ( ! fp ) { if ( pstrError ) *pstrError = strerror(errno); return false; } // Set flags to 0 unless flags argument provided int nFlags = pnFlags?*pnFlags:0; _TCHAR szDescBOM[20] = {0}; // Get document length int nDocLength = strDoc.GetLength(); #if defined( _UNICODE ) int nMBLen = 0; if ( nDocLength ) { // For ANSI files, replace CP_UTF8 with CP_ACP in both places nMBLen = WideCharToMultiByte(CP_UTF8,0,strDoc,nDocLength,NULL,0,NULL,NULL); char* pBuffer = new char[nMBLen+1]; WideCharToMultiByte(CP_UTF8,0,strDoc,nDocLength,pBuffer,nMBLen+1,NULL,NULL); bSuccess = ( fwrite( pBuffer, nMBLen, 1, fp ) == 1 ); delete [] pBuffer; } if ( pstrError ) (*pstrError).Format( _T("%d wide chars to %s%d bytes"), nDocLength, szDescBOM, nMBLen ); #else // MBCS or UTF-8 if ( nDocLength ) { CString strDocWrite = strDoc; // reference unless converted nDocLength = strDocWrite.GetLength(); bSuccess = ( fwrite( (LPCTSTR)strDocWrite, nDocLength, 1, fp ) == 1 ); } if ( pstrError ) (*pstrError).Format( _T("%s%d bytes"), szDescBOM, nDocLength ); #endif if ( ! bSuccess && pstrError ) *pstrError = strerror(errno); fclose(fp); if ( pnFlags ) *pnFlags = nFlags; return bSuccess; } bool CMarkup::FindElem( LPCTSTR szName ) { // Change current position only if found // if ( m_aPos.GetSize() ) { int iPos = x_FindElem( m_iPosParent, m_iPos, szName ); if ( iPos ) { // Assign new position x_SetPos( m_aPos[iPos].iElemParent, iPos, 0 ); return true; } } return false; } bool CMarkup::FindChildElem( LPCTSTR szName ) { // Change current child position only if found // // Shorthand: call this with no current main position // means find child under root element if ( ! m_iPos ) FindElem(); int iPosChild = x_FindElem( m_iPos, m_iPosChild, szName ); if ( iPosChild ) { // Assign new position int iPos = m_aPos[iPosChild].iElemParent; x_SetPos( m_aPos[iPos].iElemParent, iPos, iPosChild ); return true; } return false; } CString CMarkup::EscapeText( LPCTSTR szText, int nFlags ) { // Convert text as seen outside XML document to XML friendly // replacing special characters with ampersand escape codes // E.g. convert "6>7" to "6&gt;7" // // &lt; less than // &amp; ampersand // &gt; greater than // // and for attributes: // // &apos; apostrophe or single quote // &quot; double quote // static LPCTSTR szaReplace[] = { _T("&lt;"),_T("&amp;"),_T("&gt;"),_T("&apos;"),_T("&quot;") }; LPCTSTR pFind = (nFlags&MNF_ESCAPEQUOTES)?_T("<&>\'\""):_T("<&>