#include "tinyxml2.h" #include #include #include #include #include #include //#pragma warning ( disable : 4291 ) using namespace tinyxml2; static const char LINE_FEED = (char)0x0a; // all line endings are normalized to LF static const char LF = LINE_FEED; static const char CARRIAGE_RETURN = (char)0x0d; // CR gets filtered out static const char CR = CARRIAGE_RETURN; static const char SINGLE_QUOTE = '\''; static const char DOUBLE_QUOTE = '\"'; #define DELETE_NODE( node ) { MemPool* pool = node->memPool; node->~XMLNode(); pool->Free( node ); } #define DELETE_ATTRIBUTE( attrib ) { MemPool* pool = attrib->memPool; attrib->~XMLAttribute(); pool->Free( attrib ); } struct Entity { const char* pattern; int length; char value; }; static const int NUM_ENTITIES = 5; static const Entity entities[NUM_ENTITIES] = { { "quot", 4, DOUBLE_QUOTE }, { "amp", 3, '&' }, { "apos", 4, SINGLE_QUOTE }, { "lt", 2, '<' }, { "gt", 2, '>' } }; StrPair::~StrPair() { Reset(); } void StrPair::Reset() { if ( flags & NEEDS_DELETE ) { delete [] start; } flags = 0; start = 0; end = 0; } void StrPair::SetStr( const char* str, int flags ) { Reset(); size_t len = strlen( str ); start = new char[ len+1 ]; memcpy( start, str, len+1 ); end = start + len; this->flags = flags | NEEDS_DELETE; } char* StrPair::ParseText( char* p, const char* endTag, int strFlags ) { TIXMLASSERT( endTag && *endTag ); char* start = p; // fixme: hides a member char endChar = *endTag; int length = strlen( endTag ); // Inner loop of text parsing. while ( *p ) { if ( *p == endChar && strncmp( p, endTag, length ) == 0 ) { Set( start, p, strFlags ); return p + length; } ++p; } return 0; } char* StrPair::ParseName( char* p ) { char* start = p; start = p; if ( !start || !(*start) ) { return 0; } if ( !XMLUtil::IsAlpha( *p ) ) { return 0; } while( *p && ( XMLUtil::IsAlphaNum( (unsigned char) *p ) || *p == '_' || *p == '-' || *p == '.' || *p == ':' )) { ++p; } if ( p > start ) { Set( start, p, 0 ); return p; } return 0; } const char* StrPair::GetStr() { if ( flags & NEEDS_FLUSH ) { *end = 0; flags ^= NEEDS_FLUSH; if ( flags ) { char* p = start; char* q = start; while( p < end ) { if ( (flags & NEEDS_NEWLINE_NORMALIZATION) && *p == CR ) { // CR-LF pair becomes LF // CR alone becomes LF // LF-CR becomes LF if ( *(p+1) == LF ) { p += 2; } else { ++p; } *q++ = LF; } else if ( (flags & NEEDS_NEWLINE_NORMALIZATION) && *p == LF ) { if ( *(p+1) == CR ) { p += 2; } else { ++p; } *q++ = LF; } else if ( (flags & NEEDS_ENTITY_PROCESSING) && *p == '&' ) { int i=0; for( i=0; imemPool = &commentPool; p += xmlHeaderLen; } else if ( XMLUtil::StringEqual( p, commentHeader, commentHeaderLen ) ) { returnNode = new (commentPool.Alloc()) XMLComment( this ); returnNode->memPool = &commentPool; p += commentHeaderLen; } else if ( XMLUtil::StringEqual( p, cdataHeader, cdataHeaderLen ) ) { XMLText* text = new (textPool.Alloc()) XMLText( this ); returnNode = text; returnNode->memPool = &textPool; p += cdataHeaderLen; text->SetCData( true ); } else if ( XMLUtil::StringEqual( p, dtdHeader, dtdHeaderLen ) ) { returnNode = new (commentPool.Alloc()) XMLUnknown( this ); returnNode->memPool = &commentPool; p += dtdHeaderLen; } else if ( XMLUtil::StringEqual( p, elementHeader, elementHeaderLen ) ) { returnNode = new (elementPool.Alloc()) XMLElement( this ); returnNode->memPool = &elementPool; p += elementHeaderLen; } else { returnNode = new (textPool.Alloc()) XMLText( this ); returnNode->memPool = &textPool; p = start; // Back it up, all the text counts. } *node = returnNode; return p; } bool XMLDocument::Accept( XMLVisitor* visitor ) const { if ( visitor->VisitEnter( *this ) ) { for ( const XMLNode* node=FirstChild(); node; node=node->NextSibling() ) { if ( !node->Accept( visitor ) ) break; } } return visitor->VisitExit( *this ); } // --------- XMLNode ----------- // XMLNode::XMLNode( XMLDocument* doc ) : document( doc ), parent( 0 ), firstChild( 0 ), lastChild( 0 ), prev( 0 ), next( 0 ) { } XMLNode::~XMLNode() { ClearChildren(); if ( parent ) { parent->Unlink( this ); } } void XMLNode::SetValue( const char* str, bool staticMem ) { if ( staticMem ) value.SetInternedStr( str ); else value.SetStr( str ); } void XMLNode::ClearChildren() { while( firstChild ) { XMLNode* node = firstChild; Unlink( node ); DELETE_NODE( node ); } firstChild = lastChild = 0; } void XMLNode::Unlink( XMLNode* child ) { TIXMLASSERT( child->parent == this ); if ( child == firstChild ) firstChild = firstChild->next; if ( child == lastChild ) lastChild = lastChild->prev; if ( child->prev ) { child->prev->next = child->next; } if ( child->next ) { child->next->prev = child->prev; } child->parent = 0; } void XMLNode::DeleteChild( XMLNode* node ) { TIXMLASSERT( node->parent == this ); DELETE_NODE( node ); } XMLNode* XMLNode::InsertEndChild( XMLNode* addThis ) { if ( lastChild ) { TIXMLASSERT( firstChild ); TIXMLASSERT( lastChild->next == 0 ); lastChild->next = addThis; addThis->prev = lastChild; lastChild = addThis; addThis->next = 0; } else { TIXMLASSERT( firstChild == 0 ); firstChild = lastChild = addThis; addThis->prev = 0; addThis->next = 0; } addThis->parent = this; return addThis; } XMLNode* XMLNode::InsertFirstChild( XMLNode* addThis ) { if ( firstChild ) { TIXMLASSERT( lastChild ); TIXMLASSERT( firstChild->prev == 0 ); firstChild->prev = addThis; addThis->next = firstChild; firstChild = addThis; addThis->prev = 0; } else { TIXMLASSERT( lastChild == 0 ); firstChild = lastChild = addThis; addThis->prev = 0; addThis->next = 0; } addThis->parent = this; return addThis; } XMLNode* XMLNode::InsertAfterChild( XMLNode* afterThis, XMLNode* addThis ) { TIXMLASSERT( afterThis->parent == this ); if ( afterThis->parent != this ) return 0; if ( afterThis->next == 0 ) { // The last node or the only node. return InsertEndChild( addThis ); } addThis->prev = afterThis; addThis->next = afterThis->next; afterThis->next->prev = addThis; afterThis->next = addThis; addThis->parent = this; return addThis; } const XMLElement* XMLNode::FirstChildElement( const char* value ) const { for( XMLNode* node=firstChild; node; node=node->next ) { XMLElement* element = node->ToElement(); if ( element ) { if ( !value || XMLUtil::StringEqual( element->Name(), value ) ) { return element; } } } return 0; } const XMLElement* XMLNode::LastChildElement( const char* value ) const { for( XMLNode* node=lastChild; node; node=node->prev ) { XMLElement* element = node->ToElement(); if ( element ) { if ( !value || XMLUtil::StringEqual( element->Name(), value ) ) { return element; } } } return 0; } char* XMLNode::ParseDeep( char* p ) { while( p && *p ) { XMLNode* node = 0; p = document->Identify( p, &node ); if ( p && node ) { p = node->ParseDeep( p ); // FIXME: is it the correct closing element? if ( node->IsClosingElement() ) { DELETE_NODE( node ); return p; } this->InsertEndChild( node ); } } return 0; } // --------- XMLText ---------- // char* XMLText::ParseDeep( char* p ) { const char* start = p; if ( this->CData() ) { p = value.ParseText( p, "]]>", StrPair::NEEDS_NEWLINE_NORMALIZATION ); if ( !p ) { document->SetError( XMLDocument::ERROR_PARSING_CDATA, start, 0 ); } return p; } else { p = value.ParseText( p, "<", StrPair::TEXT_ELEMENT ); if ( !p ) { document->SetError( XMLDocument::ERROR_PARSING_TEXT, start, 0 ); } if ( p && *p ) { return p-1; } } return 0; } bool XMLText::Accept( XMLVisitor* visitor ) const { return visitor->Visit( *this ); } // --------- XMLComment ---------- // XMLComment::XMLComment( XMLDocument* doc ) : XMLNode( doc ) { } XMLComment::~XMLComment() { //printf( "~XMLComment\n" ); } char* XMLComment::ParseDeep( char* p ) { // Comment parses as text. const char* start = p; p = value.ParseText( p, "-->", StrPair::COMMENT ); if ( p == 0 ) { document->SetError( XMLDocument::ERROR_PARSING_COMMENT, start, 0 ); } return p; } bool XMLComment::Accept( XMLVisitor* visitor ) const { return visitor->Visit( *this ); } // --------- XMLDeclaration ---------- // XMLDeclaration::XMLDeclaration( XMLDocument* doc ) : XMLNode( doc ) { } XMLDeclaration::~XMLDeclaration() { //printf( "~XMLDeclaration\n" ); } char* XMLDeclaration::ParseDeep( char* p ) { // Declaration parses as text. const char* start = p; p = value.ParseText( p, "?>", StrPair::NEEDS_NEWLINE_NORMALIZATION ); if ( p == 0 ) { document->SetError( XMLDocument::ERROR_PARSING_DECLARATION, start, 0 ); } return p; } bool XMLDeclaration::Accept( XMLVisitor* visitor ) const { return visitor->Visit( *this ); } // --------- XMLUnknown ---------- // XMLUnknown::XMLUnknown( XMLDocument* doc ) : XMLNode( doc ) { } XMLUnknown::~XMLUnknown() { } char* XMLUnknown::ParseDeep( char* p ) { // Unknown parses as text. const char* start = p; p = value.ParseText( p, ">", StrPair::NEEDS_NEWLINE_NORMALIZATION ); if ( !p ) { document->SetError( XMLDocument::ERROR_PARSING_UNKNOWN, start, 0 ); } return p; } bool XMLUnknown::Accept( XMLVisitor* visitor ) const { return visitor->Visit( *this ); } // --------- XMLAttribute ---------- // char* XMLAttribute::ParseDeep( char* p ) { p = name.ParseText( p, "=", StrPair::ATTRIBUTE_NAME ); if ( !p || !*p ) return 0; char endTag[2] = { *p, 0 }; ++p; p = value.ParseText( p, endTag, StrPair::ATTRIBUTE_VALUE ); if ( value.Empty() ) return 0; return p; } void XMLAttribute::SetName( const char* n ) { name.SetStr( n ); } int XMLAttribute::QueryIntAttribute( int* value ) const { if ( TIXML_SSCANF( Value(), "%d", value ) == 1 ) return ATTRIBUTE_SUCCESS; return WRONG_ATTRIBUTE_TYPE; } int XMLAttribute::QueryUnsignedAttribute( unsigned int* value ) const { if ( TIXML_SSCANF( Value(), "%u", value ) == 1 ) return ATTRIBUTE_SUCCESS; return WRONG_ATTRIBUTE_TYPE; } int XMLAttribute::QueryBoolAttribute( bool* value ) const { int ival = -1; QueryIntAttribute( &ival ); if ( ival > 0 || XMLUtil::StringEqual( Value(), "true" ) ) { *value = true; return ATTRIBUTE_SUCCESS; } else if ( ival == 0 || XMLUtil::StringEqual( Value(), "false" ) ) { *value = false; return ATTRIBUTE_SUCCESS; } return WRONG_ATTRIBUTE_TYPE; } int XMLAttribute::QueryDoubleAttribute( double* value ) const { if ( TIXML_SSCANF( Value(), "%lf", value ) == 1 ) return ATTRIBUTE_SUCCESS; return WRONG_ATTRIBUTE_TYPE; } int XMLAttribute::QueryFloatAttribute( float* value ) const { if ( TIXML_SSCANF( Value(), "%f", value ) == 1 ) return ATTRIBUTE_SUCCESS; return WRONG_ATTRIBUTE_TYPE; } void XMLAttribute::SetAttribute( const char* v ) { value.SetStr( v ); } void XMLAttribute::SetAttribute( int v ) { char buf[BUF_SIZE]; TIXML_SNPRINTF( buf, BUF_SIZE-1, "%d", v ); value.SetStr( buf ); } void XMLAttribute::SetAttribute( unsigned v ) { char buf[BUF_SIZE]; TIXML_SNPRINTF( buf, BUF_SIZE-1, "%u", v ); value.SetStr( buf ); } void XMLAttribute::SetAttribute( bool v ) { char buf[BUF_SIZE]; TIXML_SNPRINTF( buf, BUF_SIZE-1, "%d", v ? 1 : 0 ); value.SetStr( buf ); } void XMLAttribute::SetAttribute( double v ) { char buf[BUF_SIZE]; TIXML_SNPRINTF( buf, BUF_SIZE-1, "%f", v ); value.SetStr( buf ); } void XMLAttribute::SetAttribute( float v ) { char buf[BUF_SIZE]; TIXML_SNPRINTF( buf, BUF_SIZE-1, "%f", v ); value.SetStr( buf ); } // --------- XMLElement ---------- // XMLElement::XMLElement( XMLDocument* doc ) : XMLNode( doc ), closing( false ), rootAttribute( 0 ) //lastAttribute( 0 ) { } XMLElement::~XMLElement() { while( rootAttribute ) { XMLAttribute* next = rootAttribute->next; DELETE_ATTRIBUTE( rootAttribute ); rootAttribute = next; } } XMLAttribute* XMLElement::FindAttribute( const char* name ) { XMLAttribute* a = 0; for( a=rootAttribute; a; a = a->next ) { if ( XMLUtil::StringEqual( a->Name(), name ) ) return a; } return 0; } const XMLAttribute* XMLElement::FindAttribute( const char* name ) const { XMLAttribute* a = 0; for( a=rootAttribute; a; a = a->next ) { if ( XMLUtil::StringEqual( a->Name(), name ) ) return a; } return 0; } const char* XMLElement::GetText() const { if ( FirstChild() && FirstChild()->ToText() ) { return FirstChild()->ToText()->Value(); } return 0; } XMLAttribute* XMLElement::FindOrCreateAttribute( const char* name ) { XMLAttribute* attrib = FindAttribute( name ); if ( !attrib ) { attrib = new (document->attributePool.Alloc() ) XMLAttribute( this ); attrib->memPool = &document->attributePool; LinkAttribute( attrib ); attrib->SetName( name ); } return attrib; } void XMLElement::LinkAttribute( XMLAttribute* attrib ) { if ( rootAttribute ) { XMLAttribute* end = rootAttribute; while ( end->next ) end = end->next; end->next = attrib; } else { rootAttribute = attrib; } } void XMLElement::DeleteAttribute( const char* name ) { XMLAttribute* prev = 0; for( XMLAttribute* a=rootAttribute; a; a=a->next ) { if ( XMLUtil::StringEqual( name, a->Name() ) ) { if ( prev ) { prev->next = a->next; } else { rootAttribute = a->next; } DELETE_ATTRIBUTE( a ); break; } prev = a; } } char* XMLElement::ParseAttributes( char* p, bool* closedElement ) { const char* start = p; *closedElement = false; // Read the attributes. while( p ) { p = XMLUtil::SkipWhiteSpace( p ); if ( !p || !(*p) ) { document->SetError( XMLDocument::ERROR_PARSING_ELEMENT, start, Name() ); return 0; } // attribute. if ( XMLUtil::IsAlpha( *p ) ) { XMLAttribute* attrib = new (document->attributePool.Alloc() ) XMLAttribute( this ); attrib->memPool = &document->attributePool; p = attrib->ParseDeep( p ); if ( !p ) { DELETE_ATTRIBUTE( attrib ); document->SetError( XMLDocument::ERROR_PARSING_ATTRIBUTE, start, p ); return 0; } LinkAttribute( attrib ); } // end of the tag else if ( *p == '/' && *(p+1) == '>' ) { if ( closing ) { document->SetError( XMLDocument::ERROR_PARSING_ELEMENT, start, p ); return 0; } *closedElement = true; return p+2; // done; sealed element. } // end of the tag else if ( *p == '>' ) { ++p; break; } else { document->SetError( XMLDocument::ERROR_PARSING_ELEMENT, start, p ); return 0; } } return p; } // // // foobar // char* XMLElement::ParseDeep( char* p ) { // Read the element name. p = XMLUtil::SkipWhiteSpace( p ); if ( !p ) return 0; const char* start = p; // The closing element is the form. It is // parsed just like a regular element then deleted from // the DOM. if ( *p == '/' ) { closing = true; ++p; } p = value.ParseName( p ); if ( value.Empty() ) return 0; bool elementClosed=false; p = ParseAttributes( p, &elementClosed ); if ( !p || !*p || elementClosed || closing ) return p; p = XMLNode::ParseDeep( p ); return p; } bool XMLElement::Accept( XMLVisitor* visitor ) const { if ( visitor->VisitEnter( *this, rootAttribute ) ) { for ( const XMLNode* node=FirstChild(); node; node=node->NextSibling() ) { if ( !node->Accept( visitor ) ) break; } } return visitor->VisitExit( *this ); } // --------- XMLDocument ----------- // XMLDocument::XMLDocument() : XMLNode( 0 ), charBuffer( 0 ) { document = this; // avoid warning about 'this' in initializer list } XMLDocument::~XMLDocument() { ClearChildren(); delete [] charBuffer; #if 0 textPool.Trace( "text" ); elementPool.Trace( "element" ); commentPool.Trace( "comment" ); attributePool.Trace( "attribute" ); #endif TIXMLASSERT( textPool.CurrentAllocs() == 0 ); TIXMLASSERT( elementPool.CurrentAllocs() == 0 ); TIXMLASSERT( commentPool.CurrentAllocs() == 0 ); TIXMLASSERT( attributePool.CurrentAllocs() == 0 ); } void XMLDocument::InitDocument() { errorID = NO_ERROR; errorStr1 = 0; errorStr2 = 0; delete [] charBuffer; charBuffer = 0; } XMLElement* XMLDocument::NewElement( const char* name ) { XMLElement* ele = new (elementPool.Alloc()) XMLElement( this ); ele->memPool = &elementPool; ele->SetName( name ); return ele; } XMLComment* XMLDocument::NewComment( const char* str ) { XMLComment* comment = new (commentPool.Alloc()) XMLComment( this ); comment->memPool = &commentPool; comment->SetValue( str ); return comment; } XMLText* XMLDocument::NewText( const char* str ) { XMLText* text = new (textPool.Alloc()) XMLText( this ); text->memPool = &textPool; text->SetValue( str ); return text; } int XMLDocument::Load( const char* filename ) { ClearChildren(); InitDocument(); FILE* fp = fopen( filename, "rb" ); if ( !fp ) { SetError( ERROR_FILE_NOT_FOUND, filename, 0 ); return errorID; } Load( fp ); fclose( fp ); return errorID; } int XMLDocument::Load( FILE* fp ) { ClearChildren(); InitDocument(); fseek( fp, 0, SEEK_END ); unsigned size = ftell( fp ); fseek( fp, 0, SEEK_SET ); charBuffer = new char[size+1]; fread( charBuffer, size, 1, fp ); charBuffer[size] = 0; ParseDeep( charBuffer ); return errorID; } void XMLDocument::Save( const char* filename ) { FILE* fp = fopen( filename, "w" ); XMLStreamer stream( fp ); Print( &stream ); fclose( fp ); } int XMLDocument::Parse( const char* p ) { ClearChildren(); InitDocument(); if ( !p || !*p ) { return true; // correctly parse an empty string? } size_t len = strlen( p ); charBuffer = new char[ len+1 ]; memcpy( charBuffer, p, len+1 ); ParseDeep( charBuffer ); return errorID; } void XMLDocument::Print( XMLStreamer* streamer ) { XMLStreamer stdStreamer( stdout ); if ( !streamer ) streamer = &stdStreamer; //for( XMLNode* node = firstChild; node; node=node->next ) { // node->Print( streamer ); //} Accept( streamer ); } void XMLDocument::SetError( int error, const char* str1, const char* str2 ) { errorID = error; errorStr1 = str1; errorStr2 = str2; } void XMLDocument::PrintError() const { if ( errorID ) { char buf1[20] = { 0 }; char buf2[20] = { 0 }; if ( errorStr1 ) { strncpy( buf1, errorStr1, 20 ); buf1[19] = 0; } if ( errorStr2 ) { strncpy( buf2, errorStr2, 20 ); buf2[19] = 0; } printf( "XMLDocument error id=%d str1=%s str2=%s\n", errorID, buf1, buf2 ); } } XMLStreamer::XMLStreamer( FILE* file ) : elementJustOpened( false ), firstElement( true ), fp( file ), depth( 0 ), textDepth( -1 ) { for( int i=0; i 0 ) { Print( "%s", p ); } } void XMLStreamer::OpenElement( const char* name ) { if ( elementJustOpened ) { SealElement(); } stack.Push( name ); if ( textDepth < 0 && !firstElement ) { Print( "\n" ); PrintSpace( depth ); } Print( "<%s", name ); elementJustOpened = true; firstElement = false; ++depth; } void XMLStreamer::PushAttribute( const char* name, const char* value ) { TIXMLASSERT( elementJustOpened ); Print( " %s=\"", name ); PrintString( value, false ); Print( "\"" ); } void XMLStreamer::CloseElement() { --depth; const char* name = stack.Pop(); if ( elementJustOpened ) { Print( "/>" ); } else { if ( textDepth < 0 ) { Print( "\n" ); PrintSpace( depth ); } Print( "", name ); } if ( textDepth == depth ) textDepth = -1; if ( depth == 0 ) Print( "\n" ); elementJustOpened = false; } void XMLStreamer::SealElement() { elementJustOpened = false; Print( ">" ); } void XMLStreamer::PushText( const char* text, bool cdata ) { textDepth = depth-1; if ( elementJustOpened ) { SealElement(); } if ( cdata ) { Print( "" ); } else { PrintString( text, true ); } } void XMLStreamer::PushComment( const char* comment ) { if ( elementJustOpened ) { SealElement(); } if ( textDepth < 0 && !firstElement ) { Print( "\n" ); PrintSpace( depth ); } firstElement = false; Print( "", comment ); } void XMLStreamer::PushDeclaration( const char* value ) { if ( elementJustOpened ) { SealElement(); } if ( textDepth < 0 && !firstElement) { Print( "\n" ); PrintSpace( depth ); } firstElement = false; Print( "", value ); } void XMLStreamer::PushUnknown( const char* value ) { if ( elementJustOpened ) { SealElement(); } if ( textDepth < 0 && !firstElement ) { Print( "\n" ); PrintSpace( depth ); } firstElement = false; Print( "", value ); } bool XMLStreamer::VisitEnter( const XMLElement& element, const XMLAttribute* attribute ) { OpenElement( element.Name() ); while ( attribute ) { PushAttribute( attribute->Name(), attribute->Value() ); attribute = attribute->Next(); } return true; } bool XMLStreamer::VisitExit( const XMLElement& element ) { CloseElement(); return true; } bool XMLStreamer::Visit( const XMLText& text ) { PushText( text.Value() ); return true; } bool XMLStreamer::Visit( const XMLComment& comment ) { PushComment( comment.Value() ); return true; } bool XMLStreamer::Visit( const XMLDeclaration& declaration ) { PushDeclaration( declaration.Value() ); return true; } bool XMLStreamer::Visit( const XMLUnknown& unknown ) { PushUnknown( unknown.Value() ); return true; }