Mercurial > irccd
annotate plugins/links/requester.hpp @ 783:d0bd19183743
Plugin links: enable limit to buffer size, closes #942
author | David Demelier <markand@malikania.fr> |
---|---|
date | Mon, 05 Nov 2018 13:47:51 +0100 |
parents | 8c44bbcbbab9 |
children | 8460b4a34191 |
rev | line source |
---|---|
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
1 /* |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
2 * requester.hpp -- convenient HTTP get requester |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
3 * |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
4 * Copyright (c) 2013-2018 David Demelier <markand@malikania.fr> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
5 * |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
6 * Permission to use, copy, modify, and/or distribute this software for any |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
7 * purpose with or without fee is hereby granted, provided that the above |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
8 * copyright notice and this permission notice appear in all copies. |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
9 * |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
17 */ |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
18 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
19 #ifndef IRCCD_REQUESTER_HPP |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
20 #define IRCCD_REQUESTER_HPP |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
21 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
22 /** |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
23 * \file requester.hpp |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
24 * \brief Convenient HTTP get requester. |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
25 */ |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
26 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
27 #include <irccd/sysconfig.hpp> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
28 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
29 #include <boost/beast.hpp> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
30 #include <boost/asio.hpp> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
31 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
32 #if defined(IRCCD_HAVE_SSL) |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
33 # include <boost/asio/ssl.hpp> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
34 #endif |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
35 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
36 #include <cstddef> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
37 #include <memory> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
38 #include <string> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
39 #include <system_error> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
40 #include <variant> |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
41 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
42 #include "uri.hpp" |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
43 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
44 namespace irccd { |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
45 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
46 class server; |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
47 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
48 /** |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
49 * \brief Convenient HTTP get requester. |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
50 */ |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
51 class requester : public std::enable_shared_from_this<requester> { |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
52 private: |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
53 std::variant< |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
54 std::monostate, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
55 boost::asio::ip::tcp::socket |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
56 #if defined(IRCCD_HAVE_SSL) |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
57 , boost::asio::ssl::stream<boost::asio::ip::tcp::socket> |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
58 #endif |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
59 > socket_; |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
60 |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
61 std::size_t level_{0U}; |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
62 std::shared_ptr<server> server_; |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
63 std::string channel_; |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
64 std::string origin_; |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
65 |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
66 uri uri_; |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
67 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
68 #if defined(IRCCD_HAVE_SSL) |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
69 boost::asio::ssl::context ctx_{boost::asio::ssl::context::sslv23}; |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
70 #endif |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
71 |
783
d0bd19183743
Plugin links: enable limit to buffer size, closes #942
David Demelier <markand@malikania.fr>
parents:
773
diff
changeset
|
72 boost::beast::flat_buffer buffer_{163840}; |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
73 boost::beast::http::request<boost::beast::http::empty_body> req_; |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
74 boost::beast::http::response<boost::beast::http::string_body> res_; |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
75 boost::asio::deadline_timer timer_; |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
76 boost::asio::ip::tcp::resolver resolver_; |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
77 |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
78 void notify(const std::string&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
79 void parse(); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
80 void handle_read(const std::error_code&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
81 void read(); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
82 void handle_write(const std::error_code&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
83 void write(); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
84 void handle_handshake(const std::error_code&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
85 void handshake(); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
86 void handle_connect(const std::error_code&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
87 void connect(const boost::asio::ip::tcp::resolver::results_type&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
88 void handle_resolve(const std::error_code&, const boost::asio::ip::tcp::resolver::results_type&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
89 void resolve(); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
90 void handle_timer(const std::error_code&); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
91 void timer(); |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
92 void start(); |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
93 |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
94 requester(boost::asio::io_context&, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
95 std::shared_ptr<server>, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
96 std::string, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
97 std::string, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
98 uri, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
99 std::size_t); |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
100 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
101 public: |
773
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
102 /** |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
103 * Start seeking for a title in the link |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
104 * |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
105 * \param ctx the IO context |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
106 * \param sv the server |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
107 * \param origin the originator |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
108 * \param channel the channel |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
109 * \param message the message text |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
110 */ |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
111 static void run(boost::asio::io_context& ctx, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
112 std::shared_ptr<server> sv, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
113 std::string origin, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
114 std::string channel, |
8c44bbcbbab9
Misc: style, cleanup and update
David Demelier <markand@malikania.fr>
parents:
768
diff
changeset
|
115 std::string message); |
768
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
116 }; |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
117 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
118 } // !irccd |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
119 |
d8bf53170fb6
Plugin links: fix various errors, closes #912 @2h
David Demelier <markand@malikania.fr>
parents:
diff
changeset
|
120 #endif // !IRCCD_REQUESTER_HPP |