mirror of
https://github.com/openappsec/openappsec.git
synced 2025-06-28 16:41:02 +03:00
64 lines
2.0 KiB
C++
64 lines
2.0 KiB
C++
// Copyright (C) 2022 Check Point Software Technologies Ltd. All rights reserved.
|
|
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#ifndef __COROUTINE_H__
|
|
#define __COROUTINE_H__
|
|
|
|
#include <boost/coroutine2/all.hpp>
|
|
|
|
#include "i_mainloop.h"
|
|
#include "maybe_res.h"
|
|
|
|
class RoutineWrapper
|
|
{
|
|
using pull_type = boost::coroutines2::coroutine<void>::pull_type;
|
|
using push_type = boost::coroutines2::coroutine<void>::push_type;
|
|
|
|
public:
|
|
RoutineWrapper(
|
|
I_MainLoop::RoutineType _pri,
|
|
I_MainLoop::Routine func,
|
|
bool is_primary,
|
|
const std::string &_routine_name
|
|
);
|
|
|
|
~RoutineWrapper();
|
|
RoutineWrapper(const RoutineWrapper &) = delete;
|
|
RoutineWrapper(RoutineWrapper &&) = default;
|
|
RoutineWrapper & operator=(const RoutineWrapper &) = delete;
|
|
RoutineWrapper & operator=(RoutineWrapper &&) = default;
|
|
|
|
bool isPrimary() { return is_primary; }
|
|
const std::string & getRoutineName() const { return routine_name; }
|
|
bool isActive() const;
|
|
bool shouldRun(const I_MainLoop::RoutineType &limit) const;
|
|
void run();
|
|
void yield();
|
|
void halt();
|
|
void resume();
|
|
|
|
private:
|
|
static void invoke(pull_type &pull, I_MainLoop::Routine func);
|
|
static RoutineWrapper *active; // Used by `invoke` to set the value of `pull`
|
|
|
|
I_MainLoop::RoutineType pri;
|
|
// `pull` will hold the object that returns the flow control back to the mainloop.
|
|
pull_type pull;
|
|
push_type routine;
|
|
bool is_primary;
|
|
bool is_halt = false;
|
|
std::string routine_name;
|
|
};
|
|
|
|
#endif // __COROUTINE_H__
|