1 /*
2  * Copyright (C) 2018 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef IDMAP2_IDMAP2D_IDMAP2SERVICE_H_
18 #define IDMAP2_IDMAP2D_IDMAP2SERVICE_H_
19 
20 #include <android-base/unique_fd.h>
21 #include <android/os/BnIdmap2.h>
22 #include <android/os/FabricatedOverlayInfo.h>
23 #include <binder/BinderService.h>
24 #include <idmap2/ResourceContainer.h>
25 #include <idmap2/Result.h>
26 
27 #include <filesystem>
28 #include <memory>
29 #include <string>
30 #include <vector>
31 
32 namespace android::os {
33 
34 class Idmap2Service : public BinderService<Idmap2Service>, public BnIdmap2 {
35  public:
getServiceName()36   static char const* getServiceName() {
37     return "idmap";
38   }
39 
40   binder::Status getIdmapPath(const std::string& overlay_path, int32_t user_id,
41                               std::string* _aidl_return) override;
42 
43   binder::Status removeIdmap(const std::string& overlay_path, int32_t user_id,
44                              bool* _aidl_return) override;
45 
46   binder::Status verifyIdmap(const std::string& target_path, const std::string& overlay_path,
47                              const std::string& overlay_name, int32_t fulfilled_policies,
48                              bool enforce_overlayable, int32_t user_id,
49                              bool* _aidl_return) override;
50 
51   binder::Status createIdmap(const std::string& target_path, const std::string& overlay_path,
52                              const std::string& overlay_name, int32_t fulfilled_policies,
53                              bool enforce_overlayable, int32_t user_id,
54                              std::optional<std::string>* _aidl_return) override;
55 
56   binder::Status createFabricatedOverlay(
57       const os::FabricatedOverlayInternal& overlay,
58       std::optional<os::FabricatedOverlayInfo>* _aidl_return) override;
59 
60   binder::Status deleteFabricatedOverlay(const std::string& overlay_path,
61                                          bool* _aidl_return) override;
62 
63   binder::Status acquireFabricatedOverlayIterator() override;
64 
65   binder::Status releaseFabricatedOverlayIterator() override;
66 
67   binder::Status nextFabricatedOverlayInfos(
68       std::vector<os::FabricatedOverlayInfo>* _aidl_return) override;
69 
70   binder::Status dumpIdmap(const std::string& overlay_path, std::string* _aidl_return) override;
71 
72  private:
73   // idmap2d is killed after a period of inactivity, so any information stored on this class should
74   // be able to be recalculated if idmap2 dies and restarts.
75   std::unique_ptr<idmap2::TargetResourceContainer> framework_apk_cache_;
76 
77   std::optional<std::filesystem::directory_iterator> frro_iter_;
78 
79   template <typename T>
80   using MaybeUniquePtr = std::variant<std::unique_ptr<T>, T*>;
81 
82   using TargetResourceContainerPtr = MaybeUniquePtr<idmap2::TargetResourceContainer>;
83   idmap2::Result<TargetResourceContainerPtr> GetTargetContainer(const std::string& target_path);
84 
85   template <typename T>
86   WARN_UNUSED static const T* GetPointer(const MaybeUniquePtr<T>& ptr);
87 };
88 
89 template <typename T>
GetPointer(const MaybeUniquePtr<T> & ptr)90 const T* Idmap2Service::GetPointer(const MaybeUniquePtr<T>& ptr) {
91   auto u = std::get_if<T*>(&ptr);
92   if (u != nullptr) {
93     return *u;
94   }
95   return std::get<std::unique_ptr<T>>(ptr).get();
96 }
97 
98 }  // namespace android::os
99 
100 #endif  // IDMAP2_IDMAP2D_IDMAP2SERVICE_H_
101