weiParser.cpp

Go to the documentation of this file.

/*
    webEngine is the HTML processing library
    Copyright (C) 2009 Andrew Abramov aabramov@ptsecurity.ru

    This file is part of webEngine

    webEngineis free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    webEngineis distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with webEngine.  If not, see <http://www.gnu.org/licenses/>.
*/

#include <time.h>
#include <boost/algorithm/string/predicate.hpp>
#include "weiParser.h"

using namespace boost::algorithm;

#ifndef __DOXYGEN__
weCmpMode iweEntity::compareMode = weCmpDefault;
#endif // __DOXYGEN__

const string iweEntity::Attr(string name)
{
    WeAttrMap::iterator it;

    it = attributes.find(name);
    if (it != attributes.end())
    {
        return attributes.val(it);
    }
    return(*(new string("")));
}

void iweEntity::Attr(string name, string value)
{
    attributes[name] = value;
}

iweEntity* iweEntity::Child(string type)
{
    iweEntity* retval = NULL;
    WeEntityList::iterator  it;

    for(it = chldList.begin(); it != chldList.end(); it++) {
        if((*it)->Name() == type) {
            retval = (*it);
            break;
        }
    }
    return retval;
}

iweEntity* iweEntity::Child(int idx)
{
    if (idx < 0 || idx >= (int)chldList.size()) {
        return(NULL);
    }
    return chldList[idx];
}

#pragma warning(push)
#pragma warning(disable : 4311)
#pragma warning(disable : 4996)
void iweEntity::GenerateId(void)
{
    // Entity format (like GUID): {B46FF8A5-F2E9-4297-9F73-2894AABBB740}
    unsigned int first;
    unsigned int second;
    unsigned int third;
    char buff[40];

    m_entityId = "{";

    first = (unsigned int)this;
    while ((first & 0xF0000000) == 0)
    {
        first += ((unsigned int)rand() << 16);
    }

    second = (unsigned int)time(NULL);
    while ((second & 0xF0000000) == 0)
    {
        second += ((unsigned int)rand() << 16);
    }

    third = ((unsigned int)rand() << 8);
    third ^= first;
    third ^= second;
    sprintf(buff, "%08X-%04X-%04X-9798-A2%08X9F", first, (unsigned short)(second & 0xFFFF), (unsigned short)(second >> 16), third);

    m_entityId += buff;
    m_entityId += "}";
    srand(third);
}
#pragma warning(pop)

iweEntity* iweEntity::FindID(string id)
{
    WeEntityList::iterator  chld;
    iweEntity* retval;

    if (id == m_entityId) {
        return this;
    }
    for (chld = chldList.begin(); chld != chldList.end(); chld++) {
        retval = (*chld)->FindID(id);
        if (retval != NULL) {
            return retval;
        }
    }
    return (NULL);
}

WeEntityList& iweEntity::FindTags(string tag)
{
    WeEntityList* retval = new WeEntityList;
    WeEntityList::iterator  chld;

    if (iequals(tag, entityName)) {
        retval->push_back(this);
    }
    for (chld = chldList.begin(); chld != chldList.end(); chld++) {
        WeEntityList& chlds = (*chld)->FindTags(tag);
        while (chlds.size() > 0) {
            retval->push_back(chlds.back());
            chlds.pop_back();
        }
    }
    return (*retval);
}

void iweEntity::ClearChildren( void )
{
    WeEntityList::iterator  chld;

    for (chld = chldList.begin(); chld != chldList.end(); chld++) {
        delete (*chld); 
    }
    chldList.clear();
}

iweDocument* iweEntity::GetRootDocument( void )
{
    if (entityName == "#document") {
        return dynamic_cast<iweDocument*>(this);
    }
    if (parent == NULL) {
        return (iweDocument*)NULL;
    }
    return parent->GetRootDocument();
}

bool iweEntity::IsParentTag( string tag )
{
    if (iequals(tag, entityName)) {
        return true;
    }
    if (parent == NULL) {
        return false;
    }
    return parent->IsParentTag(tag);
}

---

	©2009 Positive Technologies	Generated on Tue Jul 14 18:27:30 2009 for WebEngine by		1.5.7