111 lines
2.2 KiB
C++
Executable File
111 lines
2.2 KiB
C++
Executable File
#include <string>
|
|
#include <vector>
|
|
#include <stdexcept>
|
|
#include <expat.h>
|
|
#include "xmlassociatedtd.h"
|
|
|
|
XmlAssociateDtd::XmlAssociateDtd(
|
|
const std::string& path,
|
|
const std::string& publicIdentifier,
|
|
size_t size) :
|
|
d(new DtdData())
|
|
{
|
|
d->buffer.reserve(size);
|
|
d->path = path;
|
|
d->publicIdentifier = publicIdentifier;
|
|
d->rootElementSeen = false;
|
|
d->insideDtd = false;
|
|
XML_SetUserData(p, d.get());
|
|
XML_SetElementHandler(p, start, end);
|
|
XML_SetDoctypeDeclHandler(p, startdoctypehandler, enddoctypehandler);
|
|
XML_SetDefaultHandlerExpand(p, defaulthandler);
|
|
}
|
|
|
|
XmlAssociateDtd::~XmlAssociateDtd()
|
|
{
|
|
}
|
|
|
|
void XMLCALL XmlAssociateDtd::defaulthandler(
|
|
void *data,
|
|
const XML_Char *s,
|
|
int len)
|
|
{
|
|
DtdData *d;
|
|
d = (DtdData *)data;
|
|
if (!d->insideDtd)
|
|
d->buffer.append(s, len);
|
|
}
|
|
|
|
void XMLCALL XmlAssociateDtd::start(void *data,
|
|
const XML_Char *el,
|
|
const XML_Char **attr)
|
|
{
|
|
DtdData *d;
|
|
d = (DtdData *)data;
|
|
|
|
if (!d->rootElementSeen)
|
|
{
|
|
if (d->publicIdentifier.empty())
|
|
{
|
|
d->buffer += "<!DOCTYPE ";
|
|
d->buffer += el;
|
|
d->buffer += " SYSTEM \"";
|
|
d->buffer += d->path;
|
|
d->buffer += "\">\n";
|
|
}
|
|
else
|
|
{
|
|
d->buffer += "<!DOCTYPE ";
|
|
d->buffer += el;
|
|
d->buffer += " PUBLIC \"";
|
|
d->buffer += d->publicIdentifier;
|
|
d->buffer += "\" \"";
|
|
d->buffer += d->path;
|
|
d->buffer += "\">\n";
|
|
}
|
|
d->rootElementSeen = true;
|
|
}
|
|
|
|
d->buffer += "<";
|
|
d->buffer += el;
|
|
|
|
while (*attr)
|
|
{
|
|
d->buffer += " ";
|
|
d->buffer += *attr;
|
|
d->buffer += "=\"";
|
|
d->buffer += xmliseAttribute(*(attr + 1));
|
|
d->buffer += "\"";
|
|
attr += 2;
|
|
}
|
|
d->buffer += ">";
|
|
}
|
|
|
|
void XMLCALL XmlAssociateDtd::end(void *data, const XML_Char *el)
|
|
{
|
|
DtdData *d;
|
|
d = (DtdData *)data;
|
|
d->buffer += "</";
|
|
d->buffer += el;
|
|
d->buffer += ">";
|
|
}
|
|
|
|
void XMLCALL XmlAssociateDtd::startdoctypehandler(
|
|
void *data,
|
|
const XML_Char *doctypeName,
|
|
const XML_Char *sysid,
|
|
const XML_Char *pubid,
|
|
int has_internal_subset)
|
|
{
|
|
DtdData *d;
|
|
d = (DtdData *)data;
|
|
d->insideDtd = true;
|
|
}
|
|
|
|
void XMLCALL XmlAssociateDtd::enddoctypehandler(void *data)
|
|
{
|
|
DtdData *d;
|
|
d = (DtdData *)data;
|
|
d->insideDtd = false;
|
|
}
|