1 /* 2 * Copyright (c) Huawei Technologies Co., Ltd. 2021-2023. All rights reserved. 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 HOOK_SERVICE_H 17 #define HOOK_SERVICE_H 18 19 #include <memory> 20 #include <string> 21 22 #include "hook_common.h" 23 #include "service_entry.h" 24 25 namespace OHOS::Developtools::NativeDaemon { 26 class HookManager; 27 class HookService : public ServiceBase { 28 public: 29 HookService(const ClientConfig& clientConfig, std::shared_ptr<HookManager> hook, bool multipleProcesses = false); 30 ~HookService(); 31 bool ProtocolProc(SocketContext &context, uint32_t pnum, const int8_t *buf, const uint32_t size) override; 32 void CloseSocketFd(pid_t pid); 33 void AddPidInfo(pid_t pid, uid_t uid, gid_t gid); 34 void RemovePidInfo(pid_t pid); 35 private: 36 bool StartService(const std::string& unixSocketName); 37 38 private: 39 std::shared_ptr<ServiceEntry> serviceEntry_{nullptr}; 40 ClientConfig clientConfig_; 41 std::shared_ptr<HookManager> hookMgr_{nullptr}; 42 std::mutex mtx_; 43 std::mutex pidFdMtx_; 44 std::mutex pidInfoMtx_; 45 std::unordered_map<pid_t, int> pidFds_; 46 bool firstProcess_ = true; 47 bool multipleProcesses_ = false; 48 std::unordered_map<pid_t, std::pair<uid_t, gid_t>> pidInfo_; 49 }; 50 } 51 #endif // HOOK_SERVICE_H