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