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 OHOS_STORAGE_DAEMON_STORAGE_DAEMON_STUB_H 17 #define OHOS_STORAGE_DAEMON_STORAGE_DAEMON_STUB_H 18 19 #include <map> 20 #include <thread> 21 #include "iremote_stub.h" 22 #include "ipc/istorage_daemon.h" 23 #include "utils/storage_statistics_radar.h" 24 25 namespace OHOS { 26 namespace StorageDaemon { 27 constexpr int UID_ROOT = 0; 28 constexpr int UID_STORAGEMANAGER = 1090; 29 class StorageDaemonStub : public IRemoteStub<IStorageDaemon> { 30 public: 31 StorageDaemonStub(); 32 ~StorageDaemonStub(); 33 int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, 34 MessageOption &option) override; 35 36 private: 37 using DaemonInterface = int32_t (StorageDaemonStub::*)(MessageParcel &data, MessageParcel &reply); 38 std::map<uint32_t, DaemonInterface> opToInterfaceMap_; 39 int32_t OnRemoteRequestForBase(uint32_t code, MessageParcel &data, MessageParcel &reply); 40 int32_t OnRemoteRequestForUser(uint32_t code, MessageParcel &data, MessageParcel &reply); 41 int32_t OnRemoteRequestForApp(uint32_t code, MessageParcel &data, MessageParcel &reply); 42 int32_t HandleShutdown(MessageParcel &data, MessageParcel &reply); 43 44 int32_t HandleMount(MessageParcel &data, MessageParcel &reply); 45 int32_t HandleUMount(MessageParcel &data, MessageParcel &reply); 46 int32_t HandleCheck(MessageParcel &data, MessageParcel &reply); 47 int32_t HandleFormat(MessageParcel &data, MessageParcel &reply); 48 int32_t HandlePartition(MessageParcel &data, MessageParcel &reply); 49 int32_t HandleSetVolDesc(MessageParcel &data, MessageParcel &reply); 50 51 int32_t HandleStartUser(MessageParcel &data, MessageParcel &reply); 52 int32_t HandleStopUser(MessageParcel &data, MessageParcel &reply); 53 int32_t HandlePrepareUserDirs(MessageParcel &data, MessageParcel &reply); 54 int32_t HandleDestroyUserDirs(MessageParcel &data, MessageParcel &reply); 55 int32_t HandleCompleteAddUser(MessageParcel &data, MessageParcel &reply); 56 57 // fscrypt api 58 int32_t HandleInitGlobalKey(MessageParcel &data, MessageParcel &reply); 59 int32_t HandleInitGlobalUserKeys(MessageParcel &data, MessageParcel &reply); 60 int32_t HandleGenerateUserKeys(MessageParcel &data, MessageParcel &reply); 61 int32_t HandleDeleteUserKeys(MessageParcel &data, MessageParcel &reply); 62 int32_t HandleUpdateUserAuth(MessageParcel &data, MessageParcel &reply); 63 int32_t HandleActiveUserKey(MessageParcel &data, MessageParcel &reply); 64 int32_t HandleInactiveUserKey(MessageParcel &data, MessageParcel &reply); 65 int32_t HandleUpdateKeyContext(MessageParcel &data, MessageParcel &reply); 66 int32_t HandleMountCryptoPathAgain(MessageParcel &data, MessageParcel &reply); 67 int32_t HandleLockUserScreen(MessageParcel &data, MessageParcel &reply); 68 int32_t HandleUnlockUserScreen(MessageParcel &data, MessageParcel &reply); 69 int32_t HandleGetLockScreenStatus(MessageParcel &data, MessageParcel &reply); 70 int32_t HandleGenerateAppkey(MessageParcel &data, MessageParcel &reply); 71 int32_t HandleDeleteAppkey(MessageParcel &data, MessageParcel &reply); 72 int32_t HandleGetFileEncryptStatus(MessageParcel &data, MessageParcel &reply); 73 74 // app file share api 75 int32_t HandleCreateShareFile(MessageParcel &data, MessageParcel &reply); 76 int32_t HandleDeleteShareFile(MessageParcel &data, MessageParcel &reply); 77 78 int32_t HandleSetBundleQuota(MessageParcel &data, MessageParcel &reply); 79 int32_t HandleGetOccupiedSpace(MessageParcel &data, MessageParcel &reply); 80 int32_t HandleUpdateMemoryPara(MessageParcel &data, MessageParcel &reply); 81 82 int32_t HandleGetBundleStatsForIncrease(MessageParcel &data, MessageParcel &reply); 83 int32_t HandleMountDfsDocs(MessageParcel &data, MessageParcel &reply); 84 int32_t HandleUMountDfsDocs(MessageParcel &data, MessageParcel &reply); 85 int32_t HandleGetUserNeedActiveStatus(MessageParcel &data, MessageParcel &reply); 86 std::mutex mutex_; 87 88 void StorageRadarThd(void); 89 private: 90 std::atomic<bool> stopRadarReport_ { false }; 91 std::condition_variable execRadarReportCon_; 92 std::mutex onRadarReportLock_; 93 std::atomic<bool> isNeedUpdateRadarFile_ { false }; 94 std::thread callRadarStatisticReportThread_; 95 std::map<uint32_t, RadarStatisticInfo> opStatistics_; 96 std::chrono::time_point<std::chrono::system_clock> lastRadarReportTime_; 97 std::map<uint32_t, RadarStatisticInfo>::iterator GetUserStatistics(const uint32_t userId); 98 void GetTempStatistics(std::map<uint32_t, RadarStatisticInfo> &statistics); 99 }; 100 } // StorageDaemon 101 } // OHOS 102 103 #endif // OHOS_STORAGE_DAEMON_STORAGE_DAEMON_STUB_H