1 /*
2  * Copyright (C) 2019 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 #include "MountRegistry.h"
18 
19 #include <android-base/file.h>
20 #include <android-base/logging.h>
21 #include <android-base/unique_fd.h>
22 #include <gtest/gtest.h>
23 #include <sys/select.h>
24 #include <unistd.h>
25 
26 #include <iterator>
27 #include <optional>
28 #include <thread>
29 
30 #include "incfs.h"
31 #include "path.h"
32 
33 using namespace android::incfs;
34 using namespace std::literals;
35 
36 class MountRegistryTest : public ::testing::Test {
37 protected:
SetUp()38     virtual void SetUp() {}
TearDown()39     virtual void TearDown() {}
40 
41     MountRegistry::Mounts mounts_;
42 
r()43     MountRegistry::Mounts& r() { return mounts_; }
44 };
45 
TEST_F(MountRegistryTest,RootForRoot)46 TEST_F(MountRegistryTest, RootForRoot) {
47     r().addRoot("/root", "/backing");
48     ASSERT_STREQ("/root", r().rootFor("/root").data());
49     ASSERT_STREQ("/root", r().rootFor("/root/1").data());
50     ASSERT_STREQ("/root", r().rootFor("/root/1/2").data());
51     ASSERT_STREQ(nullptr, r().rootFor("/root1/1/2").data());
52     ASSERT_STREQ(nullptr, r().rootFor("/1/root").data());
53     ASSERT_STREQ(nullptr, r().rootFor("root").data());
54 }
55 
TEST_F(MountRegistryTest,OneBind)56 TEST_F(MountRegistryTest, OneBind) {
57     r().addRoot("/root", "/backing");
58     r().addBind("/root/1", "/bind");
59     ASSERT_STREQ("/root", r().rootFor("/root").data());
60     ASSERT_STREQ("/root", r().rootFor("/bind").data());
61     ASSERT_STREQ("/root", r().rootFor("/bind/1").data());
62     ASSERT_STREQ("/root", r().rootFor("/root/1").data());
63     ASSERT_STREQ(nullptr, r().rootFor("/1/bind").data());
64     ASSERT_STREQ(nullptr, r().rootFor("bind").data());
65     ASSERT_STREQ(nullptr, r().rootFor("/bind1").data());
66     ASSERT_STREQ(nullptr, r().rootFor("/.bind").data());
67 }
68 
TEST_F(MountRegistryTest,MultiBind)69 TEST_F(MountRegistryTest, MultiBind) {
70     r().addRoot("/root", "/backing");
71     r().addBind("/root/1", "/bind");
72     r().addBind("/root/2/3", "/bind2");
73     r().addBind("/root/2/3", "/other/bind");
74     ASSERT_STREQ("/root", r().rootFor("/root").data());
75     ASSERT_STREQ("/root", r().rootFor("/bind").data());
76     ASSERT_STREQ("/root", r().rootFor("/bind2").data());
77     ASSERT_STREQ("/root", r().rootFor("/other/bind/dir").data());
78     ASSERT_EQ("/root"s, r().rootAndSubpathFor("/root").first->path);
79     ASSERT_EQ(""s, r().rootAndSubpathFor("/root").second);
80     ASSERT_EQ("/root"s, r().rootAndSubpathFor("/bind").first->path);
81     ASSERT_EQ("1"s, r().rootAndSubpathFor("/bind").second);
82     ASSERT_EQ("/root"s, r().rootAndSubpathFor("/bind2").first->path);
83     ASSERT_EQ("2/3"s, r().rootAndSubpathFor("/bind2").second);
84     ASSERT_EQ("/root"s, r().rootAndSubpathFor("/bind2/blah").first->path);
85     ASSERT_EQ("2/3/blah"s, r().rootAndSubpathFor("/bind2/blah").second);
86     ASSERT_EQ("/root"s, r().rootAndSubpathFor("/other/bind/blah").first->path);
87     ASSERT_EQ("2/3/blah"s, r().rootAndSubpathFor("/other/bind/blah").second);
88 }
89 
TEST_F(MountRegistryTest,MultiRoot)90 TEST_F(MountRegistryTest, MultiRoot) {
91     r().addRoot("/root", "/backing");
92     r().addBind("/root", "/bind");
93     ASSERT_STREQ("/root", r().rootFor("/root").data());
94     ASSERT_STREQ("/root", r().rootFor("/bind").data());
95     ASSERT_STREQ("/root", r().rootFor("/bind/2").data());
96 }
97 
makeFrom(std::string_view str)98 static MountRegistry::Mounts makeFrom(std::string_view str) {
99     TemporaryFile f;
100     EXPECT_TRUE(android::base::WriteFully(f.fd, str.data(), str.size()));
101     EXPECT_EQ(0, lseek(f.fd, 0, SEEK_SET)); // rewind
102 
103     MountRegistry::Mounts m;
104     EXPECT_TRUE(m.loadFrom(f.fd, INCFS_NAME));
105     return std::move(m);
106 }
107 
TEST_F(MountRegistryTest,MultiRootLoad)108 TEST_F(MountRegistryTest, MultiRootLoad) {
109     constexpr char mountsFile[] =
110             R"(4605 34 0:154 / /mnt/installer/0/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,noatime shared:45 master:43 - fuse /dev/fuse rw,lazytime,user_id=0,group_id=0,allow_other
111 4561 35 0:154 / /mnt/androidwritable/0/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,noatime shared:44 master:43 - fuse /dev/fuse rw,lazytime,user_id=0,group_id=0,allow_other
112 4560 99 0:154 / /storage/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,noatime master:43 - fuse /dev/fuse rw,lazytime,user_id=0,group_id=0,allow_other
113 4650 30 0:44 /MyFiles /mnt/pass_through/0/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,relatime shared:31 - 9p media rw,sync,dirsync,access=client,trans=virtio
114 3181 79 0:146 / /data/incremental/MT_data_app_vmdl703/mount rw,nosuid,nodev,noatime shared:46 - incremental-fs /data/incremental/MT_data_app_vmdl703/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0
115 3182 77 0:146 / /var/run/mount/data/mount/data/incremental/MT_data_app_vmdl703/mount rw,nosuid,nodev,noatime shared:46 - incremental-fs /data/incremental/MT_data_app_vmdl703/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0
116 )";
117 
118     auto m = makeFrom(mountsFile);
119 
120     EXPECT_EQ(size_t(1), m.size());
121     EXPECT_STREQ("/data/incremental/MT_data_app_vmdl703/mount",
122                  m.rootFor("/data/incremental/MT_data_app_vmdl703/mount/123/2").data());
123     EXPECT_STREQ("/data/incremental/MT_data_app_vmdl703/mount",
124                  m.rootFor("/var/run/mount/data/mount/data/incremental/MT_data_app_vmdl703/mount/"
125                            "some/thing")
126                          .data());
127 }
128 
TEST_F(MountRegistryTest,MultiRootLoadReversed)129 TEST_F(MountRegistryTest, MultiRootLoadReversed) {
130     constexpr char mountsFile[] =
131             R"(4605 34 0:154 / /mnt/installer/0/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,noatime shared:45 master:43 - fuse /dev/fuse rw,lazytime,user_id=0,group_id=0,allow_other
132 4561 35 0:154 / /mnt/androidwritable/0/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,noatime shared:44 master:43 - fuse /dev/fuse rw,lazytime,user_id=0,group_id=0,allow_other
133 4560 99 0:154 / /storage/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,noatime master:43 - fuse /dev/fuse rw,lazytime,user_id=0,group_id=0,allow_other
134 4650 30 0:44 /MyFiles /mnt/pass_through/0/0000000000000000000000000000CAFEF00D2019 rw,nosuid,nodev,noexec,relatime shared:31 - 9p media rw,sync,dirsync,access=client,trans=virtio
135 3182 77 0:146 / /var/run/mount/data/mount/data/incremental/MT_data_app_vmdl703/mount rw,nosuid,nodev,noatime shared:46 - incremental-fs /data/incremental/MT_data_app_vmdl703/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0
136 3181 79 0:146 / /data/incremental/MT_data_app_vmdl703/mount rw,nosuid,nodev,noatime shared:46 - incremental-fs /data/incremental/MT_data_app_vmdl703/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0
137 )";
138 
139     auto m = makeFrom(mountsFile);
140 
141     EXPECT_EQ(size_t(1), m.size());
142     EXPECT_STREQ("/data/incremental/MT_data_app_vmdl703/mount",
143                  m.rootFor("/data/incremental/MT_data_app_vmdl703/mount/123/2").data());
144     EXPECT_STREQ("/data/incremental/MT_data_app_vmdl703/mount",
145                  m.rootFor("/var/run/mount/data/mount/data/incremental/MT_data_app_vmdl703/mount/"
146                            "some/thing")
147                          .data());
148 }
149 
TEST_F(MountRegistryTest,LoadInvalid)150 TEST_F(MountRegistryTest, LoadInvalid) {
151     constexpr char mountsFile[] =
152             R"(9465 93 0:281 // /data/incremental1 shared:56 - incremental-fs /data/incremental2 rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
153 9529 93 0:281 /st_232_0 /data/app/vmdl1998506227.tmp rw,nosuid,nodev,noatime shared:56 - incremental-fs /data/incremental/MT_data_app_vmdl199/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
154 9657 93 0:282 /st_233_0 /data/app/vmdl2034419270.tmp rw,nosuid,nodev,noatime shared:57 - incremental-fs /data/incremental/MT_data_app_vmdl203/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
155 9721 93 0:283 / /data/incremental/MT_data_app_vmdl154/mount rw,nosuid,nodev,noatime shared:58 - incremental-fs /data/incremental/MT_data_app_vmdl154/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
156 9785 93 0:283 /st_234_0 /data/app/vmdl1545425783.tmp rw,nosuid,nodev,noatime shared:58 - incremental-fs /data/incremental/MT_data_app_vmdl154/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
157 9849 93 0:284 / /data/incremental/MT_data_app_vmdl209/mount rw,nosuid,nodev,noatime shared:59 - incremental-fs /data/incremental/MT_data_app_vmdl209/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
158 9913 93 0:284 /st_235_0 /data/app/vmdl2099748756.tmp rw,nosuid,nodev,noatime shared:59 - incremental-fs /data/incremental/MT_data_app_vmdl209/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
159 4007 93 0:269 /st_240_1 /data/app/~~I499PLubwcOVbJaEFqpHHQ== rw,nosuid,nodev,noatime shared:44 - incremental-fs /data/incremental/MT_data_app_vmdl158/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
160 5285 93 0:270 /st_241_1 /data/app/~~CvgMdGm9eNJpvdq-62Jktg== rw,nosuid,nodev,noatime shared:45 - incremental-fs /data/incremental/MT_data_app_vmdl943/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
161 8786 93 0:271 /st_242_1 /data/app/~~_oNudLuBvqtSE78VoMVY5Q== rw,nosuid,nodev,noatime shared:46 - incremental-fs /data/incremental/MT_data_app_vmdl144/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
162 10358 93 0:272 /st_243_1 /data/app/~~o5-RadxV4DUe-mgPyv9SkQ== rw,nosuid,nodev,noatime shared:47 - incremental-fs /data/incremental/MT_data_app_vmdl642/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
163 10422 93 0:131 /st_244_1 /data/app/~~rTnKswx1F427UguGO9nDRA== rw,nosuid,nodev,noatime shared:48 - incremental-fs /data/incremental/MT_data_app_vmdl336/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
164 10486 93 0:132 /st_245_1 /data/app/~~ZIoVqPDBjLBeajD4thHsYA== rw,nosuid,nodev,noatime shared:49 - incremental-fs /data/incremental/MT_data_app_vmdl993/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
165 10550 93 0:133 /st_246_1 /data/app/~~2qjtCtx5rqPW2Hwlrciu2w== rw,nosuid,nodev,noatime shared:50 - incremental-fs /data/incremental/MT_data_app_vmdl827/backing_store rw,seclabel,read_timeout_ms=10000,readahead=0,report_uid
166 )";
167 
168     auto m = makeFrom(mountsFile);
169     // only two of the mounts in this file are valid
170     EXPECT_EQ(size_t(2), m.size());
171 }
172