1 // Copyright 2015 The Android Open Source Project 2 // 3 // Licensed under the Apache License, Version 2.0 (the "License"); 4 // you may not use this file except in compliance with the License. 5 // You may obtain a copy of the License at 6 // 7 // http://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, 11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 // See the License for the specific language governing permissions and 13 // limitations under the License. 14 15 #ifndef BUFFET_SOCKET_STREAM_H_ 16 #define BUFFET_SOCKET_STREAM_H_ 17 18 #include <string> 19 20 #include <base/callback.h> 21 #include <base/macros.h> 22 #include <brillo/streams/stream.h> 23 #include <weave/provider/network.h> 24 #include <weave/stream.h> 25 26 namespace buffet { 27 28 class SocketStream : public weave::Stream { 29 public: SocketStream(brillo::StreamPtr ptr)30 explicit SocketStream(brillo::StreamPtr ptr) : ptr_{std::move(ptr)} {} 31 32 ~SocketStream() override = default; 33 34 void Read(void* buffer, 35 size_t size_to_read, 36 const ReadCallback& callback) override; 37 38 void Write(const void* buffer, 39 size_t size_to_write, 40 const WriteCallback& callback) override; 41 42 void CancelPendingOperations() override; 43 44 static std::unique_ptr<weave::Stream> ConnectBlocking(const std::string& host, 45 uint16_t port); 46 47 static void TlsConnect( 48 std::unique_ptr<weave::Stream> socket, 49 const std::string& host, 50 const weave::provider::Network::OpenSslSocketCallback& callback); 51 52 private: 53 brillo::StreamPtr ptr_; 54 DISALLOW_COPY_AND_ASSIGN(SocketStream); 55 }; 56 57 } // namespace buffet 58 59 #endif // BUFFET_SOCKET_STREAM_H_ 60