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 "iremote_stub.h"
21 #include "ipc/istorage_daemon.h"
22 
23 namespace OHOS {
24 namespace StorageDaemon {
25 constexpr int UID_ROOT = 0;
26 constexpr int UID_STORAGEMANAGER = 1090;
27 class StorageDaemonStub : public IRemoteStub<IStorageDaemon> {
28 public:
29     StorageDaemonStub();
30     virtual ~StorageDaemonStub() = default;
31     int32_t OnRemoteRequest(uint32_t code, MessageParcel &data, MessageParcel &reply,
32         MessageOption &option) override;
33 
34 private:
35     using DaemonInterface = int32_t (StorageDaemonStub::*)(MessageParcel &data, MessageParcel &reply);
36     std::map<uint32_t, DaemonInterface> opToInterfaceMap_;
37     int32_t OnRemoteRequestForBase(uint32_t code, MessageParcel &data, MessageParcel &reply);
38     int32_t OnRemoteRequestForUser(uint32_t code, MessageParcel &data, MessageParcel &reply);
39     int32_t OnRemoteRequestForApp(uint32_t code, MessageParcel &data, MessageParcel &reply);
40     int32_t HandleShutdown(MessageParcel &data, MessageParcel &reply);
41 
42     int32_t HandleMount(MessageParcel &data, MessageParcel &reply);
43     int32_t HandleUMount(MessageParcel &data, MessageParcel &reply);
44     int32_t HandleCheck(MessageParcel &data, MessageParcel &reply);
45     int32_t HandleFormat(MessageParcel &data, MessageParcel &reply);
46     int32_t HandlePartition(MessageParcel &data, MessageParcel &reply);
47     int32_t HandleSetVolDesc(MessageParcel &data, MessageParcel &reply);
48 
49     int32_t HandleStartUser(MessageParcel &data, MessageParcel &reply);
50     int32_t HandleStopUser(MessageParcel &data, MessageParcel &reply);
51     int32_t HandlePrepareUserDirs(MessageParcel &data, MessageParcel &reply);
52     int32_t HandleDestroyUserDirs(MessageParcel &data, MessageParcel &reply);
53     int32_t HandleCompleteAddUser(MessageParcel &data, MessageParcel &reply);
54 
55     // fscrypt api
56     int32_t HandleInitGlobalKey(MessageParcel &data, MessageParcel &reply);
57     int32_t HandleInitGlobalUserKeys(MessageParcel &data, MessageParcel &reply);
58     int32_t HandleGenerateUserKeys(MessageParcel &data, MessageParcel &reply);
59     int32_t HandleDeleteUserKeys(MessageParcel &data, MessageParcel &reply);
60     int32_t HandleUpdateUserAuth(MessageParcel &data, MessageParcel &reply);
61     int32_t HandleActiveUserKey(MessageParcel &data, MessageParcel &reply);
62     int32_t HandleInactiveUserKey(MessageParcel &data, MessageParcel &reply);
63     int32_t HandleUpdateKeyContext(MessageParcel &data, MessageParcel &reply);
64     int32_t HandleMountCryptoPathAgain(MessageParcel &data, MessageParcel &reply);
65     int32_t HandleLockUserScreen(MessageParcel &data, MessageParcel &reply);
66     int32_t HandleUnlockUserScreen(MessageParcel &data, MessageParcel &reply);
67     int32_t HandleGetLockScreenStatus(MessageParcel &data, MessageParcel &reply);
68     int32_t HandleGenerateAppkey(MessageParcel &data, MessageParcel &reply);
69     int32_t HandleDeleteAppkey(MessageParcel &data, MessageParcel &reply);
70     int32_t HandleGetFileEncryptStatus(MessageParcel &data, MessageParcel &reply);
71 
72     // app file share api
73     int32_t HandleCreateShareFile(MessageParcel &data, MessageParcel &reply);
74     int32_t HandleDeleteShareFile(MessageParcel &data, MessageParcel &reply);
75 
76     int32_t HandleSetBundleQuota(MessageParcel &data, MessageParcel &reply);
77     int32_t HandleGetOccupiedSpace(MessageParcel &data, MessageParcel &reply);
78     int32_t HandleUpdateMemoryPara(MessageParcel &data, MessageParcel &reply);
79 
80     int32_t HandleGetBundleStatsForIncrease(MessageParcel &data, MessageParcel &reply);
81     int32_t HandleMountDfsDocs(MessageParcel &data, MessageParcel &reply);
82     int32_t HandleUMountDfsDocs(MessageParcel &data, MessageParcel &reply);
83 };
84 } // StorageDaemon
85 } // OHOS
86 
87 #endif // OHOS_STORAGE_DAEMON_STORAGE_DAEMON_STUB_H