1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183
|
/* Copyright (C) 2015 Wildfire Games.
* This file is part of 0 A.D.
*
* 0 A.D. is 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 2 of the License, or
* (at your option) any later version.
*
* 0 A.D. is 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 0 A.D. If not, see <http://www.gnu.org/licenses/>.
*/
#include "precompiled.h"
#include "RelaxNG.h"
#include "lib/timer.h"
#include "lib/utf8.h"
#include "ps/CLogger.h"
#include "ps/Filesystem.h"
#include <libxml/relaxng.h>
#include <map>
TIMER_ADD_CLIENT(xml_validation);
/*
* libxml2 leaks memory when parsing schemas: https://bugzilla.gnome.org/show_bug.cgi?id=615767
* To minimise that problem, keep a global cache of parsed schemas, so we don't
* leak an indefinitely large amount of memory when repeatedly restarting the simulation.
*/
class RelaxNGSchema;
static std::map<std::string, shared_ptr<RelaxNGSchema> > g_SchemaCache;
static CMutex g_SchemaCacheLock;
void ClearSchemaCache()
{
CScopeLock lock(g_SchemaCacheLock);
g_SchemaCache.clear();
}
static void relaxNGErrorHandler(void* UNUSED(userData), xmlErrorPtr error)
{
// Strip a trailing newline
std::string message = error->message;
if (message.length() > 0 && message[message.length()-1] == '\n')
message.erase(message.length()-1);
LOGERROR("RelaxNGValidator: Validation %s: %s:%d: %s",
error->level == XML_ERR_WARNING ? "warning" : "error",
error->file, error->line, message.c_str());
}
class RelaxNGSchema
{
public:
xmlRelaxNGPtr m_Schema;
RelaxNGSchema(const std::string& grammar)
{
xmlRelaxNGParserCtxtPtr ctxt = xmlRelaxNGNewMemParserCtxt(grammar.c_str(), (int)grammar.size());
m_Schema = xmlRelaxNGParse(ctxt);
xmlRelaxNGFreeParserCtxt(ctxt);
if (m_Schema == NULL)
LOGERROR("RelaxNGValidator: Failed to compile schema");
}
~RelaxNGSchema()
{
if (m_Schema)
xmlRelaxNGFree(m_Schema);
}
};
RelaxNGValidator::RelaxNGValidator() :
m_Schema(NULL)
{
}
RelaxNGValidator::~RelaxNGValidator()
{
}
bool RelaxNGValidator::LoadGrammar(const std::string& grammar)
{
shared_ptr<RelaxNGSchema> schema;
{
CScopeLock lock(g_SchemaCacheLock);
std::map<std::string, shared_ptr<RelaxNGSchema> >::iterator it = g_SchemaCache.find(grammar);
if (it == g_SchemaCache.end())
{
schema = shared_ptr<RelaxNGSchema>(new RelaxNGSchema(grammar));
g_SchemaCache[grammar] = schema;
}
else
{
schema = it->second;
}
}
m_Schema = schema->m_Schema;
if (!m_Schema)
return false;
MD5 hash;
hash.Update((const u8*)grammar.c_str(), grammar.length());
m_Hash = hash;
return true;
}
bool RelaxNGValidator::LoadGrammarFile(const PIVFS& vfs, const VfsPath& grammarPath)
{
CVFSFile file;
if (file.Load(vfs, grammarPath) != PSRETURN_OK)
return false;
return LoadGrammar(file.DecodeUTF8());
}
bool RelaxNGValidator::Validate(const std::wstring& filename, const std::wstring& document) const
{
std::string docutf8 = "<?xml version='1.0' encoding='utf-8'?>" + utf8_from_wstring(document);
return ValidateEncoded(filename, docutf8);
}
bool RelaxNGValidator::ValidateEncoded(const std::wstring& filename, const std::string& document) const
{
TIMER_ACCRUE(xml_validation);
if (!m_Schema)
{
LOGERROR("RelaxNGValidator: No grammar loaded");
return false;
}
xmlDocPtr doc = xmlReadMemory(document.c_str(), (int)document.size(), utf8_from_wstring(filename).c_str(), NULL, XML_PARSE_NONET);
if (doc == NULL)
{
LOGERROR("RelaxNGValidator: Failed to parse document '%s'", utf8_from_wstring(filename).c_str());
return false;
}
bool ret = ValidateEncoded(doc);
xmlFreeDoc(doc);
return ret;
}
bool RelaxNGValidator::ValidateEncoded(xmlDocPtr doc) const
{
xmlRelaxNGValidCtxtPtr ctxt = xmlRelaxNGNewValidCtxt(m_Schema);
xmlRelaxNGSetValidStructuredErrors(ctxt, &relaxNGErrorHandler, NULL);
int ret = xmlRelaxNGValidateDoc(ctxt, doc);
xmlRelaxNGFreeValidCtxt(ctxt);
if (ret == 0)
{
return true;
}
else if (ret > 0)
{
LOGERROR("RelaxNGValidator: Validation failed for '%s'", doc->name);
return false;
}
else
{
LOGERROR("RelaxNGValidator: Internal error %d", ret);
return false;
}
}
bool RelaxNGValidator::CanValidate() const
{
return m_Schema != NULL;
}
|