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 SOFTBUS_AGENT_H
17 #define SOFTBUS_AGENT_H
18 
19 #include <map>
20 #include <mutex>
21 
22 #include "inner_socket.h"
23 #include "network/network_agent_template.h"
24 #include "transport/socket.h"
25 #include "transport/trans_type.h"
26 
27 namespace OHOS {
28 namespace Storage {
29 namespace DistributedFile {
30 class SoftbusAgent final : public NetworkAgentTemplate, public std::enable_shared_from_this<SoftbusAgent> {
31 public:
32     explicit SoftbusAgent(std::weak_ptr<MountPoint> mountPoint);
33     ~SoftbusAgent() = default;
34     void OnSessionOpened(const int32_t sessionId, PeerSocketInfo info);
35     void OnSessionClosed(int32_t sessionId, const std::string peerDeviceId);
36     static bool IsSameAccount(const std::string &networkId);
37     int32_t JudgeNetworkTypeIsWifi(const DeviceInfo &info);
38 protected:
39     void JoinDomain() override;
40     void QuitDomain() override;
41     void StopTopHalf() override;
42     void StopBottomHalf() override;
43     int32_t OpenSession(const DeviceInfo &info, const uint8_t &linkType) override;
44     void CloseSession(std::shared_ptr<BaseSession> session) override;
45 
46 private:
47     std::mutex serverIdMapMutex_;
48     std::map<std::string, int32_t> serverIdMap_;
49     bool IsContinueRetry(const std::string &cid);
50     std::map<std::string, int32_t> OpenSessionRetriedTimesMap_;
51     std::string sessionName_;
52 };
53 } // namespace DistributedFile
54 } // namespace Storage
55 } // namespace OHOS
56 #endif // SOFTBUS_AGENT_H