1/*
2 * Copyright (c) 2016-2018, Intel Corporation
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, are permitted provided that the following conditions are met:
6 *
7 * * Redistributions of source code must retain the above copyright notice,
8 * this list of conditions and the following disclaimer.
9 * * Redistributions in binary form must reproduce the above copyright
10 * notice, this list of conditions and the following disclaimer in the
11 * documentation and/or other materials provided with the distribution.
12 * * Neither the name of Intel Corporation nor the names of its contributors
13 * may be used to endorse or promote products derived from this software
14 * without specific prior written permission.
15 *
16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
20 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
21 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
22 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
23 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
24 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
25 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
26 * POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#ifndef SHENGCOMPILE_H
30#define SHENGCOMPILE_H
31
32#include "accel_dfa_build_strat.h"
33#include "rdfa.h"
34#include "util/bytecode_ptr.h"
35#include "util/charreach.h"
36#include "util/flat_containers.h"
37
38#include <memory>
39#include <set>
40
41struct NFA;
42
43namespace ue2 {
44
45class ReportManager;
46struct CompileContext;
47struct raw_dfa;
48
49class sheng_build_strat : public accel_dfa_build_strat {
50public:
51 sheng_build_strat(raw_dfa &rdfa_in, const ReportManager &rm_in,
52 bool only_accel_init_in)
53 : accel_dfa_build_strat(rm_in, only_accel_init_in), rdfa(rdfa_in) {}
54 raw_dfa &get_raw() const override { return rdfa; }
55 std::unique_ptr<raw_report_info> gatherReports(
56 std::vector<u32> &reports /* out */,
57 std::vector<u32> &reports_eod /* out */,
58 u8 *isSingleReport /* out */,
59 ReportID *arbReport /* out */) const override;
60 size_t accelSize(void) const override;
61 u32 max_allowed_offset_accel() const override;
62 u32 max_stop_char() const override;
63 u32 max_floating_stop_char() const override;
64 DfaType getType() const override { return Sheng; }
65
66private:
67 raw_dfa &rdfa;
68};
69
70bytecode_ptr<NFA> shengCompile(raw_dfa &raw, const CompileContext &cc,
71 const ReportManager &rm, bool only_accel_init,
72 std::set<dstate_id_t> *accel_states = nullptr);
73
74struct sheng_escape_info {
75 CharReach outs;
76 CharReach outs2_single;
77 flat_set<std::pair<u8, u8>> outs2;
78 bool outs2_broken = false;
79};
80
81bool has_accel_sheng(const NFA *nfa);
82
83} // namespace ue2
84
85#endif /* SHENGCOMPILE_H */
86