Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 1 | // Copyright 2015 The Chromium OS Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
| 4 | |
| 5 | #ifndef BUFFET_WEBSERV_CLIENT_H_ |
| 6 | #define BUFFET_WEBSERV_CLIENT_H_ |
| 7 | |
| 8 | #include <memory> |
| 9 | #include <string> |
| 10 | #include <vector> |
| 11 | |
| 12 | #include <base/memory/weak_ptr.h> |
Alex Vakulenko | e32375b | 2015-09-28 08:55:40 -0700 | [diff] [blame] | 13 | #include <weave/provider/http_server.h> |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 14 | |
| 15 | namespace dbus { |
| 16 | class Bus; |
| 17 | } |
| 18 | |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame^] | 19 | namespace brillo { |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 20 | namespace dbus_utils { |
| 21 | class AsyncEventSequencer; |
| 22 | } |
| 23 | } |
| 24 | |
| 25 | namespace libwebserv { |
| 26 | class ProtocolHandler; |
| 27 | class Request; |
| 28 | class Response; |
| 29 | class Server; |
| 30 | } |
| 31 | |
| 32 | namespace buffet { |
| 33 | |
| 34 | // Wrapper around libwebserv that implements HttpServer interface. |
Alex Vakulenko | e32375b | 2015-09-28 08:55:40 -0700 | [diff] [blame] | 35 | class WebServClient : public weave::provider::HttpServer { |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 36 | public: |
| 37 | WebServClient(const scoped_refptr<dbus::Bus>& bus, |
Alex Vakulenko | 4170585 | 2015-10-13 10:12:06 -0700 | [diff] [blame^] | 38 | brillo::dbus_utils::AsyncEventSequencer* sequencer, |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 39 | const base::Closure& server_available_callback); |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 40 | ~WebServClient() override; |
| 41 | |
| 42 | // HttpServer implementation. |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 43 | void AddHttpRequestHandler(const std::string& path, |
| 44 | const RequestHandlerCallback& callback) override; |
| 45 | void AddHttpsRequestHandler(const std::string& path, |
| 46 | const RequestHandlerCallback& callback) override; |
| 47 | |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 48 | uint16_t GetHttpPort() const override; |
| 49 | uint16_t GetHttpsPort() const override; |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 50 | std::vector<uint8_t> GetHttpsCertificateFingerprint() const override; |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 51 | |
| 52 | private: |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 53 | void OnRequest(const RequestHandlerCallback& callback, |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 54 | std::unique_ptr<libwebserv::Request> request, |
| 55 | std::unique_ptr<libwebserv::Response> response); |
| 56 | |
| 57 | void OnResponse(std::unique_ptr<libwebserv::Response> response, |
| 58 | int status_code, |
| 59 | const std::string& data, |
| 60 | const std::string& mime_type); |
| 61 | |
| 62 | void OnProtocolHandlerConnected( |
| 63 | libwebserv::ProtocolHandler* protocol_handler); |
| 64 | |
| 65 | void OnProtocolHandlerDisconnected( |
| 66 | libwebserv::ProtocolHandler* protocol_handler); |
| 67 | |
| 68 | uint16_t http_port_{0}; |
| 69 | uint16_t https_port_{0}; |
Vitaly Buka | 097cbd7 | 2015-08-13 18:28:14 -0700 | [diff] [blame] | 70 | std::vector<uint8_t> certificate_; |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 71 | |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 72 | std::unique_ptr<libwebserv::Server> web_server_; |
Alex Vakulenko | be39e93 | 2015-10-09 08:10:36 -0700 | [diff] [blame] | 73 | base::Closure server_available_callback_; |
Vitaly Buka | 974647f | 2015-08-02 20:57:43 -0700 | [diff] [blame] | 74 | |
| 75 | base::WeakPtrFactory<WebServClient> weak_ptr_factory_{this}; |
| 76 | DISALLOW_COPY_AND_ASSIGN(WebServClient); |
| 77 | }; |
| 78 | |
| 79 | } // namespace buffet |
| 80 | |
| 81 | #endif // BUFFET_WEBSERV_CLIENT_H_ |