niceSOFT/libmodsecurity: ModSecurity is an open source, cross platform web application firewall (WAF) engine for Apache, IIS and Nginx. It has a robust event-based programming language which provides protection from a range of attacks against web applications and allows for HTTP traffic monitoring, logging and real-time analysis.

src/parser/driver.h

100 строк3 KB

Eduardo Arias

Simplify initialization of fileName member of Rule instances

04 сен 2024, 16:51

04 сен 2024, 16:512c613fb

/*
 * ModSecurity, http://www.modsecurity.org/
 * Copyright (c) 2015 - 2023 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.
 *
 */

#ifdef __cplusplus
#include <string>
#include <map>
#include <stack>
#include <vector>
#include <list>
#endif


#ifndef SRC_PARSER_DRIVER_H_
#define SRC_PARSER_DRIVER_H_

#include "modsecurity/modsecurity.h"
#include "modsecurity/rules_set.h"
#include "modsecurity/rules_set_properties.h"
#include "modsecurity/audit_log.h"
#include "src/rule_script.h"
#ifndef MS_CPPCHECK_DISABLED_FOR_PARSER
#include "src/parser/seclang-parser.hh"
#endif

using modsecurity::RuleWithOperator;
using modsecurity::RulesSet;


# define YY_DECL \
  yy::seclang_parser::symbol_type yylex(modsecurity::Parser::Driver& driver)

YY_DECL;

namespace modsecurity {
namespace Parser {

#ifdef __cplusplus
class Driver;
#else
typedef struct Driver_t Driver;
#endif


class Driver : public RulesSetProperties {
 public:
    Driver();
    virtual ~Driver();

    int addSecRule(std::unique_ptr<RuleWithActions> rule);
    int addSecAction(std::unique_ptr<RuleWithActions> rule);
    int addSecMarker(const std::string& marker, const std::string &fileName, int lineNumber);
    int addSecRuleScript(std::unique_ptr<RuleScript> rule);

    bool scan_begin();
    void scan_end();
    bool trace_scanning;

    int parseFile(const std::string& f);
    int parse(const std::string& f, const std::string &ref);

    std::string file;

    bool trace_parsing;

    void error(const yy::location& l, const std::string& m);
    void error(const yy::location& l, const std::string& m,
        const std::string& c);

    std::list<yy::location *> loc;

    std::string buffer;
    RuleWithActions *m_lastRule;

    RulesSetPhases m_rulesSetPhases;

    // Retain a list of new'd filenames so that they are available during the lifetime
    // of the Driver object, but so that they will get cleaned up by the Driver
    // destructor. This is to resolve a memory leak of  yy.position.filename in location.hh.
    // Ordinarily other solutions would have been preferable, but location.hh is a
    // bison-generated file, which makes some alternative solutions impractical.
    std::list<std::string> m_filenames;
};


}  // namespace Parser
}  // namespace modsecurity

#endif  // SRC_PARSER_DRIVER_H_