| // Copyright 2017 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef REMOTING_HOST_FILE_TRANSFER_MESSAGE_HANDLER_H_ |
| #define REMOTING_HOST_FILE_TRANSFER_MESSAGE_HANDLER_H_ |
| |
| #include <cstdint> |
| #include <memory> |
| #include <string> |
| |
| #include "base/optional.h" |
| #include "remoting/host/file_proxy_wrapper.h" |
| #include "remoting/proto/file_transfer.pb.h" |
| #include "remoting/protocol/named_message_pipe_handler.h" |
| |
| namespace remoting { |
| |
| constexpr char kFileTransferDataChannelPrefix[] = "filetransfer-"; |
| |
| class FileTransferMessageHandler : public protocol::NamedMessagePipeHandler { |
| public: |
| FileTransferMessageHandler(const std::string& name, |
| std::unique_ptr<protocol::MessagePipe> pipe, |
| std::unique_ptr<FileProxyWrapper> file_proxy); |
| ~FileTransferMessageHandler() override; |
| |
| // protocol::NamedMessagePipeHandler implementation. |
| void OnConnected() override; |
| void OnIncomingMessage(std::unique_ptr<CompoundBuffer> message) override; |
| void OnDisconnecting() override; |
| |
| private: |
| void SaveResultCallback(base::Optional<protocol::FileTransfer_Error> error); |
| void StartFile(protocol::FileTransfer_Metadata metadata); |
| void CancelAndSendError(protocol::FileTransfer_Error error, |
| const std::string& log_message); |
| |
| std::unique_ptr<FileProxyWrapper> file_proxy_wrapper_; |
| }; |
| |
| } // namespace remoting |
| |
| #endif // REMOTING_HOST_FILE_TRANSFER_MESSAGE_HANDLER_H_ |