1 // Copyright 2020 The Pigweed Authors 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); you may not 4 // use this file except in compliance with the License. You may obtain a copy of 5 // the License at 6 // 7 // https://www.apache.org/licenses/LICENSE-2.0 8 // 9 // Unless required by applicable law or agreed to in writing, software 10 // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT 11 // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the 12 // License for the specific language governing permissions and limitations under 13 // the License. 14 #pragma once 15 16 #include <netinet/in.h> 17 18 #include <cstdint> 19 #include <span> 20 21 #include "pw_stream/stream.h" 22 23 namespace pw::stream { 24 25 class SocketStream : public NonSeekableReaderWriter { 26 public: 27 constexpr SocketStream() = default; 28 ~SocketStream()29 ~SocketStream() { Close(); } 30 31 // Listen to the port and return after a client is connected 32 Status Serve(uint16_t port); 33 34 // Connect to a local or remote endpoint. Host must be an IPv4 address. If 35 // host is nullptr then the locahost address is used instead. 36 Status Connect(const char* host, uint16_t port); 37 38 // Close the socket stream and release all resources 39 void Close(); 40 41 private: 42 static constexpr int kInvalidFd = -1; 43 44 Status DoWrite(std::span<const std::byte> data) override; 45 46 StatusWithSize DoRead(ByteSpan dest) override; 47 48 uint16_t listen_port_ = 0; 49 int socket_fd_ = kInvalidFd; 50 int conn_fd_ = kInvalidFd; 51 struct sockaddr_in sockaddr_client_ = {}; 52 }; 53 54 } // namespace pw::stream 55