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_MANAGER_STORAGE_MANAGER_STUB_H
17 #define OHOS_STORAGE_MANAGER_STORAGE_MANAGER_STUB_H
18 
19 #include <map>
20 #include "iremote_stub.h"
21 #include "istorage_manager.h"
22 
23 namespace OHOS {
24 namespace StorageManager {
25 class StorageManagerStub : public IRemoteStub<IStorageManager> {
26 public:
27     StorageManagerStub();
28     virtual ~StorageManagerStub() = default;
29     int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply, MessageOption &option) override;
30 
31 private:
32     using ManagerInterface = int32_t (StorageManagerStub::*)(MessageParcel &data, MessageParcel &reply);
33     std::map<uint32_t, ManagerInterface> opToInterfaceMap_;
34 
35     int32_t HandlePrepareAddUser(MessageParcel &data, MessageParcel &reply);
36     int32_t HandleRemoveUser(MessageParcel &data, MessageParcel &reply);
37     int32_t HandlePrepareStartUser(MessageParcel &data, MessageParcel &reply);
38     int32_t HandleStopUser(MessageParcel &data, MessageParcel &reply);
39     int32_t HandleCompleteAddUser(MessageParcel &data, MessageParcel &reply);
40     int32_t HandleGetTotal(MessageParcel &data, MessageParcel &reply);
41     int32_t HandleGetFree(MessageParcel &data, MessageParcel &reply);
42     int32_t HandleGetBundleStatus(MessageParcel &data, MessageParcel &reply);
43     int32_t HandleGetSystemSize(MessageParcel &data, MessageParcel &reply);
44     int32_t HandleGetTotalSize(MessageParcel &data, MessageParcel &reply);
45     int32_t HandleGetFreeSize(MessageParcel &data, MessageParcel &reply);
46     int32_t HandleGetCurrUserStorageStats(MessageParcel &data, MessageParcel &reply);
47     int32_t HandleGetUserStorageStats(MessageParcel &data, MessageParcel &reply);
48     int32_t HandleGetUserStorageStatsByType(MessageParcel &data, MessageParcel &reply);
49     int32_t HandleGetCurrentBundleStats(MessageParcel &data, MessageParcel &reply);
50     int32_t HandleNotifyVolumeCreated(MessageParcel &data, MessageParcel &reply);
51     int32_t HandleNotifyVolumeMounted(MessageParcel &data, MessageParcel &reply);
52     int32_t HandleNotifyVolumeStateChanged(MessageParcel &data, MessageParcel &reply);
53     int32_t HandleMount(MessageParcel &data, MessageParcel &reply);
54     int32_t HandleUnmount(MessageParcel &data, MessageParcel &reply);
55     int32_t HandleGetAllVolumes(MessageParcel &data, MessageParcel &reply);
56     int32_t HandleNotifyDiskCreated(MessageParcel &data, MessageParcel &reply);
57     int32_t HandleNotifyDiskDestroyed(MessageParcel &data, MessageParcel &reply);
58     int32_t HandlePartition(MessageParcel &data, MessageParcel &reply);
59     int32_t HandleGetAllDisks(MessageParcel &data, MessageParcel &reply);
60     int32_t HandleGetVolumeByUuid(MessageParcel &data, MessageParcel &reply);
61     int32_t HandleGetVolumeById(MessageParcel &data, MessageParcel &reply);
62     int32_t HandleSetVolDesc(MessageParcel &data, MessageParcel &reply);
63     int32_t HandleFormat(MessageParcel &data, MessageParcel &reply);
64     int32_t HandleGetDiskById(MessageParcel &data, MessageParcel &reply);
65 
66     // fscrypt api
67     int32_t HandleGenerateUserKeys(MessageParcel &data, MessageParcel &reply);
68     int32_t HandleDeleteUserKeys(MessageParcel &data, MessageParcel &reply);
69     int32_t HandleUpdateUserAuth(MessageParcel &data, MessageParcel &reply);
70     int32_t HandleActiveUserKey(MessageParcel &data, MessageParcel &reply);
71     int32_t HandleInactiveUserKey(MessageParcel &data, MessageParcel &reply);
72     int32_t HandleLockUserScreen(MessageParcel &data, MessageParcel &reply);
73     int32_t HandleUnlockUserScreen(MessageParcel &data, MessageParcel &reply);
74     int32_t HandleGetLockScreenStatus(MessageParcel &data, MessageParcel &reply);
75     int32_t HandleUpdateKeyContext(MessageParcel &data, MessageParcel &reply);
76     int32_t HandleGenerateAppkey(MessageParcel &data, MessageParcel &reply);
77     int32_t HandleDeleteAppkey(MessageParcel &data, MessageParcel &reply);
78     int32_t HandleGetFileEncryptStatus(MessageParcel &data, MessageParcel &reply);
79 
80     // app file share api
81     int32_t HandleCreateShareFile(MessageParcel &data, MessageParcel &reply);
82     int32_t HandleDeleteShareFile(MessageParcel &data, MessageParcel &reply);
83 
84     int32_t HandleSetBundleQuota(MessageParcel &data, MessageParcel &reply);
85     int32_t HandleUpdateMemoryPara(MessageParcel &data, MessageParcel &reply);
86 
87     int32_t HandleGetBundleStatsForIncrease(MessageParcel &data, MessageParcel &reply);
88 
89     // dfs service
90     int32_t HandleMountDfsDocs(MessageParcel &data, MessageParcel &reply);
91     int32_t HandleUMountDfsDocs(MessageParcel &data, MessageParcel &reply);
92 };
93 } // StorageManager
94 } // OHOS
95 
96 #endif // OHOS_STORAGE_MANAGER_STORAGE_MANAGER_STUB_H