core/small_step/cmd_stepper.h

/*
 * SPDX-FileCopyrightText: 2022 The P4 Language Consortium
 *
 * SPDX-License-Identifier: Apache-2.0
 */
 
#ifndef BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_SMALL_STEP_CMD_STEPPER_H_
#define BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_SMALL_STEP_CMD_STEPPER_H_
 
#include <map>
#include <optional>
#include <vector>
 
#include "ir/ir.h"
#include "ir/solver.h"
#include "lib/cstring.h"
 
#include "backends/p4tools/modules/testgen/core/program_info.h"
#include "backends/p4tools/modules/testgen/core/small_step/abstract_stepper.h"
#include "backends/p4tools/modules/testgen/lib/continuation.h"
#include "backends/p4tools/modules/testgen/lib/execution_state.h"
 
namespace P4::P4Tools::P4Testgen {

class CmdStepper : public AbstractStepper {
 public:
    CmdStepper(ExecutionState &state, AbstractSolver &solver, const ProgramInfo &programInfo);
 
    bool preorder(const IR::AssignmentStatement *assign) override;
    bool preorder(const IR::P4Parser *p4parser) override;
    bool preorder(const IR::P4Control *p4control) override;
    bool preorder(const IR::EmptyStatement *empty) override;
    bool preorder(const IR::IfStatement *ifStatement) override;
    bool preorder(const IR::MethodCallStatement *methodCallStatement) override;
    bool preorder(const IR::P4Program *program) override;
    bool preorder(const IR::ParserState *parserState) override;
    bool preorder(const IR::BlockStatement *block) override;
    bool preorder(const IR::ExitStatement *e) override;
    bool preorder(const IR::SwitchStatement *switchStatement) override;
 
 protected:
    const Constraint *startParser(const IR::P4Parser *parser, ExecutionState &state);

    virtual std::optional<const Constraint *> startParserImpl(const IR::P4Parser *parser,
                                                              ExecutionState &state) const = 0;

    virtual void initializeTargetEnvironment(ExecutionState &state) const = 0;

    virtual std::map<Continuation::Exception, Continuation> getExceptionHandlers(
        const IR::P4Parser *parser, Continuation::Body normalContinuation,
        const ExecutionState &state) const = 0;
};
 
}  // namespace P4::P4Tools::P4Testgen
 
#endif /* BACKENDS_P4TOOLS_MODULES_TESTGEN_CORE_SMALL_STEP_CMD_STEPPER_H_ */