First release of open-appsec source code

core/mainloop/coroutine.h

@@ -0,0 +1,63 @@
+// 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__