mirror of
https://github.com/owasp-modsecurity/ModSecurity.git
synced 2025-09-30 19:47:47 +03:00
98 lines
2.3 KiB
C++
98 lines
2.3 KiB
C++
/*
|
|
* ModSecurity, http://www.modsecurity.org/
|
|
* Copyright (c) 2015 Trustwave Holdings, Inc. (http://www.trustwave.com/)
|
|
*
|
|
* You may not use this file except in compliance with
|
|
* the License. You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* If any of the files related to licensing are missing or if you have any
|
|
* other questions related to licensing please contact Trustwave Holdings, Inc.
|
|
* directly using the email address security@modsecurity.org.
|
|
*
|
|
*/
|
|
|
|
#ifndef SRC_OPERATORS_VALIDATE_DTD_H_
|
|
#define SRC_OPERATORS_VALIDATE_DTD_H_
|
|
|
|
#include <stdio.h>
|
|
#include <stdarg.h>
|
|
#include <string.h>
|
|
#include <libxml/xmlschemas.h>
|
|
#include <libxml/xpath.h>
|
|
|
|
#include <string>
|
|
|
|
#include "operators/operator.h"
|
|
|
|
#ifdef __cplusplus
|
|
namespace modsecurity {
|
|
namespace operators {
|
|
|
|
class ValidateDTD : public Operator {
|
|
public:
|
|
/** @ingroup ModSecurity_Operator */
|
|
ValidateDTD(std::string o, std::string p, bool i)
|
|
: Operator(o, p, i),
|
|
m_dtd(NULL) { }
|
|
~ValidateDTD() {
|
|
if (m_dtd != NULL) {
|
|
xmlFreeDtd(m_dtd);
|
|
m_dtd = NULL;
|
|
}
|
|
}
|
|
|
|
bool evaluate(Transaction *transaction, const std::string &str) override;
|
|
bool init(const std::string &file, std::string *error) override;
|
|
|
|
|
|
static void error_runtime(void *ctx, const char *msg, ...) {
|
|
Transaction *t = reinterpret_cast<Transaction *>(ctx);
|
|
char buf[1024];
|
|
std::string s;
|
|
va_list args;
|
|
|
|
va_start(args, msg);
|
|
int len = vsnprintf(buf, sizeof(buf), msg, args);
|
|
va_end(args);
|
|
|
|
if (len > 0) {
|
|
s = "XML Error: " + std::string(buf);
|
|
}
|
|
t->debug(4, s);
|
|
}
|
|
|
|
|
|
static void warn_runtime(void *ctx, const char *msg, ...) {
|
|
Transaction *t = reinterpret_cast<Transaction *>(ctx);
|
|
char buf[1024];
|
|
std::string s;
|
|
va_list args;
|
|
|
|
va_start(args, msg);
|
|
int len = vsnprintf(buf, sizeof(buf), msg, args);
|
|
va_end(args);
|
|
|
|
if (len > 0) {
|
|
s = "XML Warning: " + std::string(buf);
|
|
}
|
|
t->debug(4, s);
|
|
}
|
|
|
|
|
|
static void null_error(void *ctx, const char *msg, ...) {
|
|
}
|
|
|
|
private:
|
|
std::string m_resource;
|
|
xmlDtdPtr m_dtd;
|
|
};
|
|
|
|
} // namespace operators
|
|
} // namespace modsecurity
|
|
#endif
|
|
|
|
|
|
#endif // SRC_OPERATORS_VALIDATE_DTD_H_
|