1 /* 2 * Copyright (c) 2021 Huawei Device Co., Ltd. 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 16 #ifndef STREAM_MANAGER_H 17 #define STREAM_MANAGER_H 18 19 #include <memory> 20 #include <string> 21 #include <utility> 22 23 #include "common_inner.h" 24 #include "i_stream.h" 25 #include "i_stream_manager.h" 26 #include "i_stream_msg_manager.h" 27 #include "i_stream_socket.h" 28 #include "session.h" 29 #include "stream_common.h" 30 31 namespace Communication { 32 namespace SoftBus { 33 class StreamManager : public std::enable_shared_from_this<StreamManager>, public IStreamManager { 34 public: 35 class StreamSocketListener : public IStreamSocketListener { 36 public: StreamSocketListener(std::shared_ptr<IStreamManagerListener> streamListener)37 explicit StreamSocketListener(std::shared_ptr<IStreamManagerListener> streamListener) 38 : listener_(streamListener) {} 39 StreamSocketListener() = delete; 40 virtual ~StreamSocketListener() = default; OnStreamReceived(std::unique_ptr<IStream> stream)41 void OnStreamReceived(std::unique_ptr<IStream> stream) override 42 { 43 listener_->OnStreamReceived(std::move(stream)); 44 } OnStreamStatus(int status)45 void OnStreamStatus(int status) override 46 { 47 listener_->OnStreamStatus(status); 48 } 49 OnStreamHdrReceived(std::unique_ptr<char[]> header,int size)50 int OnStreamHdrReceived(std::unique_ptr<char[]> header, int size) override 51 { 52 static_cast<void>(header); 53 static_cast<void>(size); 54 return 0; 55 } 56 OnQosEvent(int32_t eventId,int32_t tvCount,const QosTv * tvList)57 void OnQosEvent(int32_t eventId, int32_t tvCount, const QosTv *tvList) const override 58 { 59 listener_->OnQosEvent(eventId, tvCount, tvList); 60 } 61 62 private: 63 std::shared_ptr<IStreamManagerListener> listener_ = nullptr; 64 }; 65 StreamManager(std::shared_ptr<IStreamManagerListener> streamListener)66 explicit StreamManager(std::shared_ptr<IStreamManagerListener> streamListener) : streamListener_(streamListener) {} 67 StreamManager() = delete; 68 virtual ~StreamManager() = default; 69 70 int CreateStreamClientChannel(IpAndPort &local, IpAndPort remote, Proto protocol, 71 int streamType, const std::string &sessionKey) override; 72 73 int CreateStreamServerChannel(IpAndPort &local, Proto protocol, int streamType, 74 const std::string &sessionKey) override; 75 76 bool DestroyStreamDataChannel() override; 77 78 bool Send(std::unique_ptr<IStream> data) override; 79 80 bool SetOption(int type, const StreamAttr &value) override; 81 StreamAttr GetOption(int type) const override; 82 83 void SetStreamRecvListener(std::shared_ptr<IStreamManagerListener> recvListener) override; 84 bool PrepareEnvironment(const std::string &pkgName) override; 85 void DestroyEnvironment(const std::string &pkgName) override; 86 SetStreamMsgManager(std::shared_ptr<IStreamMsgManager> manager)87 void SetStreamMsgManager(std::shared_ptr<IStreamMsgManager> manager) 88 { 89 msgManager_ = manager; 90 } GetStreamMsgManager()91 std::shared_ptr<IStreamMsgManager> GetStreamMsgManager() 92 { 93 return msgManager_; 94 } 95 96 private: 97 StreamManager(const StreamManager &) = delete; 98 StreamManager(StreamManager &&) = delete; 99 StreamManager &operator=(const StreamManager &) = delete; 100 StreamManager &operator=(StreamManager &&) = delete; 101 102 bool ChangeToOtherChannel(const IStreamSocket &target); 103 bool AddNextFrameFlag(Proto next); 104 105 std::map<Proto, std::shared_ptr<IStreamSocket>> socketMap_; 106 Proto curProtocol_ = VTP; 107 std::shared_ptr<IStreamSocketListener> socketListener_ = nullptr; 108 std::shared_ptr<IStreamMsgManager> msgManager_ = nullptr; 109 std::shared_ptr<IStreamManagerListener> streamListener_ = nullptr; 110 }; 111 } // namespace SoftBus 112 } // namespace Communication 113 114 #endif 115