From c4f805b37748048b62688f205d60165b87d589ef Mon Sep 17 00:00:00 2001
From: Vibhav Pant <vibhavp@gmail.com>
Date: Sun, 27 Aug 2023 00:35:28 +0530
Subject: [PATCH] Add WifiDirectSocket methods.

---
 .../implementation/linux/wifi_direct_socket.h      | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

diff --git a/internal/platform/implementation/linux/wifi_direct_socket.h b/internal/platform/implementation/linux/wifi_direct_socket.h
index 79ec3556..932b3257 100644
--- a/internal/platform/implementation/linux/wifi_direct_socket.h
+++ b/internal/platform/implementation/linux/wifi_direct_socket.h
@@ -9,13 +9,19 @@ namespace nearby {
 namespace linux {
 class WifiDirectSocket : public api::WifiDirectSocket {
 public:
-  WifiDirectSocket(int socket);
+  WifiDirectSocket(int socket)
+      : fd_(sdbus::UnixFd(socket)), output_stream_(fd_), input_stream_(fd_) {}
   ~WifiDirectSocket() = default;
 
-  InputStream &GetInputStream() override;
-  OutputStream &GetOutputStream() override;
+  InputStream &GetInputStream() override { return input_stream_; };
+  OutputStream &GetOutputStream() override { return output_stream_; };
 
-  Exception Close() override;
+  Exception Close() override {
+    input_stream_.Close();
+    output_stream_.Close();
+
+    return Exception{Exception::kSuccess};
+  };
 
 private:
   sdbus::UnixFd fd_;