~andreas-pokorny/mir/unregitser-fd-handler

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
/*
 * Copyright © 2013 Canonical Ltd.
 *
 * This program is free software: you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 3,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Authored by: Alexandros Frantzis <alexandros.frantzis@canonical.com>
 */

#ifndef MIR_ASIO_MAIN_LOOP_H_
#define MIR_ASIO_MAIN_LOOP_H_

#include "mir/main_loop.h"

#include <boost/asio.hpp>
#include <memory>
#include <vector>
#include <mutex>
#include <utility>
#include <deque>
#include <set>

namespace mir
{

class AsioMainLoop : public MainLoop
{
public:
    AsioMainLoop();
    ~AsioMainLoop() noexcept(true);

    void run();
    void stop();

    void register_signal_handler(
        std::initializer_list<int> signals,
        std::function<void(int)> const& handler);

    void register_fd_handler(
        std::initializer_list<int> fd,
        std::function<void(int)> const& handler);

    std::unique_ptr<time::Alarm> notify_in(std::chrono::milliseconds delay,
                                           std::function<void()> callback) override;
    std::unique_ptr<time::Alarm> notify_at(mir::time::Timestamp time_point,
                                           std::function<void()> callback) override;
    void enqueue(void const* owner, ServerAction const& action);
    void pause_processing_for(void const* owner);
    void resume_processing_for(void const* owner);

private:
    class SignalHandler;
    class FDHandler;
    bool should_process(void const*);
    void process_server_actions();

    boost::asio::io_service io;
    boost::asio::io_service::work work;
    std::vector<std::unique_ptr<SignalHandler>> signal_handlers;
    std::vector<std::unique_ptr<FDHandler>> fd_handlers;
    std::mutex server_actions_mutex;
    std::deque<std::pair<void const*,ServerAction>> server_actions;
    std::set<void const*> do_not_process;
};

}

#endif /* MIR_ASIO_MAIN_LOOP_H */