| #ifndef DEVICE_GOOGLE_GCE_GCE_UTILS_GCE_VNC_SERVER_TCPSOCKET_H_ |
| #define DEVICE_GOOGLE_GCE_GCE_UTILS_GCE_VNC_SERVER_TCPSOCKET_H_ |
| |
| #include "vnc_utils.h" |
| |
| #include <cstddef> |
| #include <cstdint> |
| #include <mutex> |
| |
| #include <unistd.h> |
| |
| namespace avd { |
| namespace vnc { |
| |
| class ServerSocket; |
| |
| // Recv and Send wait until all data has been received or sent. |
| // Send is thread safe in this regard, Recv is not. |
| class ClientSocket { |
| public: |
| ClientSocket(ClientSocket&& other) : fd_{other.fd_} {} |
| |
| ClientSocket& operator=(ClientSocket&& other) { |
| fd_ = other.fd_; |
| return *this; |
| } |
| |
| ClientSocket(const ClientSocket&) = delete; |
| ClientSocket& operator=(const ClientSocket&) = delete; |
| |
| Message Recv(std::size_t length); |
| ssize_t Send(const std::uint8_t* data, std::size_t size); |
| ssize_t Send(const Message& message); |
| |
| template <std::size_t N> |
| ssize_t Send(const std::uint8_t (&data)[N]) { |
| return Send(data, N); |
| } |
| |
| bool closed() const { return other_side_closed_; } |
| |
| private: |
| friend ServerSocket; |
| explicit ClientSocket(avd::SharedFD fd) : fd_(fd) {} |
| |
| avd::SharedFD fd_; |
| bool other_side_closed_{}; |
| std::mutex send_lock_; |
| }; |
| |
| class ServerSocket { |
| public: |
| explicit ServerSocket(int port); |
| |
| ServerSocket(const ServerSocket&) = delete; |
| ServerSocket& operator=(const ServerSocket&) = delete; |
| |
| ClientSocket Accept(); |
| |
| private: |
| avd::SharedFD fd_; |
| }; |
| |
| } // namespace vnc |
| } // namespace avd |
| |
| #endif |