1 /*
2 ** Copyright 2008, 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 "InstalldNativeService.h"
18 
19 #define ATRACE_TAG ATRACE_TAG_PACKAGE_MANAGER
20 
21 #include <algorithm>
22 #include <errno.h>
23 #include <fstream>
24 #include <fts.h>
25 #include <functional>
26 #include <inttypes.h>
27 #include <regex>
28 #include <stdlib.h>
29 #include <string.h>
30 #include <sys/capability.h>
31 #include <sys/file.h>
32 #include <sys/ioctl.h>
33 #include <sys/mman.h>
34 #include <sys/mount.h>
35 #include <sys/resource.h>
36 #include <sys/stat.h>
37 #include <sys/statvfs.h>
38 #include <sys/types.h>
39 #include <sys/wait.h>
40 #include <sys/xattr.h>
41 #include <unistd.h>
42 
43 #include <android-base/file.h>
44 #include <android-base/logging.h>
45 #include <android-base/parseint.h>
46 #include <android-base/properties.h>
47 #include <android-base/scopeguard.h>
48 #include <android-base/stringprintf.h>
49 #include <android-base/strings.h>
50 #include <android-base/unique_fd.h>
51 #include <cutils/ashmem.h>
52 #include <cutils/fs.h>
53 #include <cutils/properties.h>
54 #include <cutils/sched_policy.h>
55 #include <log/log.h>               // TODO: Move everything to base/logging.
56 #include <logwrap/logwrap.h>
57 #include <private/android_filesystem_config.h>
58 #include <private/android_projectid_config.h>
59 #include <selinux/android.h>
60 #include <system/thread_defs.h>
61 #include <utils/Trace.h>
62 
63 #include "dexopt.h"
64 #include "globals.h"
65 #include "installd_deps.h"
66 #include "otapreopt_utils.h"
67 #include "utils.h"
68 #include "view_compiler.h"
69 
70 #include "CacheTracker.h"
71 #include "CrateManager.h"
72 #include "MatchExtensionGen.h"
73 #include "QuotaUtils.h"
74 
75 #ifndef LOG_TAG
76 #define LOG_TAG "installd"
77 #endif
78 
79 using android::base::ParseUint;
80 using android::base::StringPrintf;
81 using std::endl;
82 
83 namespace android {
84 namespace installd {
85 
86 // An uuid used in unit tests.
87 static constexpr const char* kTestUuid = "TEST";
88 
89 static constexpr const mode_t kRollbackFolderMode = 0700;
90 
91 static constexpr const char* kCpPath = "/system/bin/cp";
92 static constexpr const char* kXattrDefault = "user.default";
93 
94 static constexpr const char* kDataMirrorCePath = "/data_mirror/data_ce";
95 static constexpr const char* kDataMirrorDePath = "/data_mirror/data_de";
96 
97 static constexpr const int MIN_RESTRICTED_HOME_SDK_VERSION = 24; // > M
98 
99 static constexpr const char* PKG_LIB_POSTFIX = "/lib";
100 static constexpr const char* CACHE_DIR_POSTFIX = "/cache";
101 static constexpr const char* CODE_CACHE_DIR_POSTFIX = "/code_cache";
102 
103 // fsverity assumes the page size is always 4096. If not, the feature can not be
104 // enabled.
105 static constexpr int kVerityPageSize = 4096;
106 static constexpr size_t kSha256Size = 32;
107 static constexpr const char* kPropApkVerityMode = "ro.apk_verity.mode";
108 static constexpr const char* kFuseProp = "persist.sys.fuse";
109 
110 /**
111  * Property to control if app data isolation is enabled.
112  */
113 static constexpr const char* kAppDataIsolationEnabledProperty = "persist.zygote.app_data_isolation";
114 static constexpr const char* kMntSdcardfs = "/mnt/runtime/default/";
115 static constexpr const char* kMntFuse = "/mnt/pass_through/0/";
116 
117 static std::atomic<bool> sAppDataIsolationEnabled(false);
118 
119 namespace {
120 
121 constexpr const char* kDump = "android.permission.DUMP";
122 
ok()123 static binder::Status ok() {
124     return binder::Status::ok();
125 }
126 
exception(uint32_t code,const std::string & msg)127 static binder::Status exception(uint32_t code, const std::string& msg) {
128     LOG(ERROR) << msg << " (" << code << ")";
129     return binder::Status::fromExceptionCode(code, String8(msg.c_str()));
130 }
131 
error()132 static binder::Status error() {
133     return binder::Status::fromServiceSpecificError(errno);
134 }
135 
error(const std::string & msg)136 static binder::Status error(const std::string& msg) {
137     PLOG(ERROR) << msg;
138     return binder::Status::fromServiceSpecificError(errno, String8(msg.c_str()));
139 }
140 
error(uint32_t code,const std::string & msg)141 static binder::Status error(uint32_t code, const std::string& msg) {
142     LOG(ERROR) << msg << " (" << code << ")";
143     return binder::Status::fromServiceSpecificError(code, String8(msg.c_str()));
144 }
145 
checkPermission(const char * permission)146 binder::Status checkPermission(const char* permission) {
147     pid_t pid;
148     uid_t uid;
149 
150     if (checkCallingPermission(String16(permission), reinterpret_cast<int32_t*>(&pid),
151             reinterpret_cast<int32_t*>(&uid))) {
152         return ok();
153     } else {
154         return exception(binder::Status::EX_SECURITY,
155                 StringPrintf("UID %d / PID %d lacks permission %s", uid, pid, permission));
156     }
157 }
158 
checkUid(uid_t expectedUid)159 binder::Status checkUid(uid_t expectedUid) {
160     uid_t uid = IPCThreadState::self()->getCallingUid();
161     if (uid == expectedUid || uid == AID_ROOT) {
162         return ok();
163     } else {
164         return exception(binder::Status::EX_SECURITY,
165                 StringPrintf("UID %d is not expected UID %d", uid, expectedUid));
166     }
167 }
168 
checkArgumentUuid(const std::optional<std::string> & uuid)169 binder::Status checkArgumentUuid(const std::optional<std::string>& uuid) {
170     if (!uuid || is_valid_filename(*uuid)) {
171         return ok();
172     } else {
173         return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
174                 StringPrintf("UUID %s is malformed", uuid->c_str()));
175     }
176 }
177 
checkArgumentUuidTestOrNull(const std::optional<std::string> & uuid)178 binder::Status checkArgumentUuidTestOrNull(const std::optional<std::string>& uuid) {
179     if (!uuid || strcmp(uuid->c_str(), kTestUuid) == 0) {
180         return ok();
181     } else {
182         return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
183                 StringPrintf("UUID must be null or \"%s\", got: %s", kTestUuid, uuid->c_str()));
184     }
185 }
186 
checkArgumentPackageName(const std::string & packageName)187 binder::Status checkArgumentPackageName(const std::string& packageName) {
188     if (is_valid_package_name(packageName)) {
189         return ok();
190     } else {
191         return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
192                 StringPrintf("Package name %s is malformed", packageName.c_str()));
193     }
194 }
195 
checkArgumentPath(const std::string & path)196 binder::Status checkArgumentPath(const std::string& path) {
197     if (path.empty()) {
198         return exception(binder::Status::EX_ILLEGAL_ARGUMENT, "Missing path");
199     }
200     if (path[0] != '/') {
201         return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
202                 StringPrintf("Path %s is relative", path.c_str()));
203     }
204     if ((path + '/').find("/../") != std::string::npos) {
205         return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
206                 StringPrintf("Path %s is shady", path.c_str()));
207     }
208     for (const char& c : path) {
209         if (c == '\0' || c == '\n') {
210             return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
211                     StringPrintf("Path %s is malformed", path.c_str()));
212         }
213     }
214     return ok();
215 }
216 
checkArgumentPath(const std::optional<std::string> & path)217 binder::Status checkArgumentPath(const std::optional<std::string>& path) {
218     if (path) {
219         return checkArgumentPath(*path);
220     } else {
221         return ok();
222     }
223 }
224 
225 #define ENFORCE_UID(uid) {                                  \
226     binder::Status status = checkUid((uid));                \
227     if (!status.isOk()) {                                   \
228         return status;                                      \
229     }                                                       \
230 }
231 
232 #define CHECK_ARGUMENT_UUID(uuid) {                         \
233     binder::Status status = checkArgumentUuid((uuid));      \
234     if (!status.isOk()) {                                   \
235         return status;                                      \
236     }                                                       \
237 }
238 
239 #define CHECK_ARGUMENT_UUID_IS_TEST_OR_NULL(uuid) {         \
240     auto status = checkArgumentUuidTestOrNull(uuid);        \
241     if (!status.isOk()) {                                   \
242         return status;                                      \
243     }                                                       \
244 }                                                           \
245 
246 #define CHECK_ARGUMENT_PACKAGE_NAME(packageName) {          \
247     binder::Status status =                                 \
248             checkArgumentPackageName((packageName));        \
249     if (!status.isOk()) {                                   \
250         return status;                                      \
251     }                                                       \
252 }
253 
254 #define CHECK_ARGUMENT_PATH(path) {                         \
255     binder::Status status = checkArgumentPath((path));      \
256     if (!status.isOk()) {                                   \
257         return status;                                      \
258     }                                                       \
259 }
260 
261 #define ASSERT_PAGE_SIZE_4K() {                             \
262     if (getpagesize() != kVerityPageSize) {                 \
263         return error("FSVerity only supports 4K pages");     \
264     }                                                       \
265 }
266 
267 }  // namespace
268 
start()269 status_t InstalldNativeService::start() {
270     IPCThreadState::self()->disableBackgroundScheduling(true);
271     status_t ret = BinderService<InstalldNativeService>::publish();
272     if (ret != android::OK) {
273         return ret;
274     }
275     sp<ProcessState> ps(ProcessState::self());
276     ps->startThreadPool();
277     ps->giveThreadPoolName();
278     sAppDataIsolationEnabled = android::base::GetBoolProperty(
279             kAppDataIsolationEnabledProperty, true);
280     return android::OK;
281 }
282 
dump(int fd,const Vector<String16> &)283 status_t InstalldNativeService::dump(int fd, const Vector<String16> & /* args */) {
284     auto out = std::fstream(StringPrintf("/proc/self/fd/%d", fd));
285     const binder::Status dump_permission = checkPermission(kDump);
286     if (!dump_permission.isOk()) {
287         out << dump_permission.toString8() << endl;
288         return PERMISSION_DENIED;
289     }
290     std::lock_guard<std::recursive_mutex> lock(mLock);
291 
292     out << "installd is happy!" << endl;
293 
294     {
295         std::lock_guard<std::recursive_mutex> lock(mMountsLock);
296         out << endl << "Storage mounts:" << endl;
297         for (const auto& n : mStorageMounts) {
298             out << "    " << n.first << " = " << n.second << endl;
299         }
300     }
301 
302     {
303         std::lock_guard<std::recursive_mutex> lock(mQuotasLock);
304         out << endl << "Per-UID cache quotas:" << endl;
305         for (const auto& n : mCacheQuotas) {
306             out << "    " << n.first << " = " << n.second << endl;
307         }
308     }
309 
310     out << endl;
311     out.flush();
312 
313     return NO_ERROR;
314 }
315 
316 /**
317  * Perform restorecon of the given path, but only perform recursive restorecon
318  * if the label of that top-level file actually changed.  This can save us
319  * significant time by avoiding no-op traversals of large filesystem trees.
320  */
restorecon_app_data_lazy(const std::string & path,const std::string & seInfo,uid_t uid,bool existing)321 static int restorecon_app_data_lazy(const std::string& path, const std::string& seInfo, uid_t uid,
322         bool existing) {
323     int res = 0;
324     char* before = nullptr;
325     char* after = nullptr;
326 
327     // Note that SELINUX_ANDROID_RESTORECON_DATADATA flag is set by
328     // libselinux. Not needed here.
329 
330     if (lgetfilecon(path.c_str(), &before) < 0) {
331         PLOG(ERROR) << "Failed before getfilecon for " << path;
332         goto fail;
333     }
334     if (selinux_android_restorecon_pkgdir(path.c_str(), seInfo.c_str(), uid, 0) < 0) {
335         PLOG(ERROR) << "Failed top-level restorecon for " << path;
336         goto fail;
337     }
338     if (lgetfilecon(path.c_str(), &after) < 0) {
339         PLOG(ERROR) << "Failed after getfilecon for " << path;
340         goto fail;
341     }
342 
343     // If the initial top-level restorecon above changed the label, then go
344     // back and restorecon everything recursively
345     if (strcmp(before, after)) {
346         if (existing) {
347             LOG(DEBUG) << "Detected label change from " << before << " to " << after << " at "
348                     << path << "; running recursive restorecon";
349         }
350         if (selinux_android_restorecon_pkgdir(path.c_str(), seInfo.c_str(), uid,
351                 SELINUX_ANDROID_RESTORECON_RECURSE) < 0) {
352             PLOG(ERROR) << "Failed recursive restorecon for " << path;
353             goto fail;
354         }
355     }
356 
357     goto done;
358 fail:
359     res = -1;
360 done:
361     free(before);
362     free(after);
363     return res;
364 }
365 
restorecon_app_data_lazy(const std::string & parent,const char * name,const std::string & seInfo,uid_t uid,bool existing)366 static int restorecon_app_data_lazy(const std::string& parent, const char* name,
367         const std::string& seInfo, uid_t uid, bool existing) {
368     return restorecon_app_data_lazy(StringPrintf("%s/%s", parent.c_str(), name), seInfo, uid,
369             existing);
370 }
371 
prepare_app_dir(const std::string & path,mode_t target_mode,uid_t uid)372 static int prepare_app_dir(const std::string& path, mode_t target_mode, uid_t uid) {
373     if (fs_prepare_dir_strict(path.c_str(), target_mode, uid, uid) != 0) {
374         PLOG(ERROR) << "Failed to prepare " << path;
375         return -1;
376     }
377     return 0;
378 }
379 
prepare_app_profile_dir(const std::string & packageName,int32_t appId,int32_t userId)380 static bool prepare_app_profile_dir(const std::string& packageName, int32_t appId, int32_t userId) {
381     if (!property_get_bool("dalvik.vm.usejitprofiles", false)) {
382         return true;
383     }
384 
385     int32_t uid = multiuser_get_uid(userId, appId);
386     int shared_app_gid = multiuser_get_shared_gid(userId, appId);
387     if (shared_app_gid == -1) {
388         // TODO(calin): this should no longer be possible but do not continue if we don't get
389         // a valid shared gid.
390         PLOG(WARNING) << "Invalid shared_app_gid for " << packageName;
391         return true;
392     }
393 
394     const std::string profile_dir =
395             create_primary_current_profile_package_dir_path(userId, packageName);
396     // read-write-execute only for the app user.
397     if (fs_prepare_dir_strict(profile_dir.c_str(), 0700, uid, uid) != 0) {
398         PLOG(ERROR) << "Failed to prepare " << profile_dir;
399         return false;
400     }
401     if (selinux_android_restorecon(profile_dir.c_str(), 0)) {
402         PLOG(ERROR) << "Failed to restorecon " << profile_dir;
403         return false;
404     }
405 
406     const std::string ref_profile_path =
407             create_primary_reference_profile_package_dir_path(packageName);
408 
409     // Prepare the reference profile directory. Note that we use the non strict version of
410     // fs_prepare_dir. This will fix the permission and the ownership to the correct values.
411     // This is particularly important given that in O there were some fixes for how the
412     // shared_app_gid is computed.
413     //
414     // Note that by the time we get here we know that we are using a correct uid (otherwise
415     // prepare_app_dir and the above fs_prepare_file_strict which check the uid). So we
416     // are sure that the gid being used belongs to the owning app and not someone else.
417     //
418     // dex2oat/profman runs under the shared app gid and it needs to read/write reference profiles.
419     if (fs_prepare_dir(ref_profile_path.c_str(), 0770, AID_SYSTEM, shared_app_gid) != 0) {
420         PLOG(ERROR) << "Failed to prepare " << ref_profile_path;
421         return false;
422     }
423 
424     return true;
425 }
426 
createAppData(const std::optional<std::string> & uuid,const std::string & packageName,int32_t userId,int32_t flags,int32_t appId,const std::string & seInfo,int32_t targetSdkVersion,int64_t * _aidl_return)427 binder::Status InstalldNativeService::createAppData(const std::optional<std::string>& uuid,
428         const std::string& packageName, int32_t userId, int32_t flags, int32_t appId,
429         const std::string& seInfo, int32_t targetSdkVersion, int64_t* _aidl_return) {
430     ENFORCE_UID(AID_SYSTEM);
431     CHECK_ARGUMENT_UUID(uuid);
432     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
433     std::lock_guard<std::recursive_mutex> lock(mLock);
434 
435     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
436     const char* pkgname = packageName.c_str();
437 
438     // Assume invalid inode unless filled in below
439     if (_aidl_return != nullptr) *_aidl_return = -1;
440 
441     int32_t uid = multiuser_get_uid(userId, appId);
442     int32_t cacheGid = multiuser_get_cache_gid(userId, appId);
443     mode_t targetMode = targetSdkVersion >= MIN_RESTRICTED_HOME_SDK_VERSION ? 0700 : 0751;
444 
445     // If UID doesn't have a specific cache GID, use UID value
446     if (cacheGid == -1) {
447         cacheGid = uid;
448     }
449 
450     if (flags & FLAG_STORAGE_CE) {
451         auto path = create_data_user_ce_package_path(uuid_, userId, pkgname);
452         bool existing = (access(path.c_str(), F_OK) == 0);
453 
454         if (prepare_app_dir(path, targetMode, uid) ||
455                 prepare_app_cache_dir(path, "cache", 02771, uid, cacheGid) ||
456                 prepare_app_cache_dir(path, "code_cache", 02771, uid, cacheGid)) {
457             return error("Failed to prepare " + path);
458         }
459 
460         // Consider restorecon over contents if label changed
461         if (restorecon_app_data_lazy(path, seInfo, uid, existing) ||
462                 restorecon_app_data_lazy(path, "cache", seInfo, uid, existing) ||
463                 restorecon_app_data_lazy(path, "code_cache", seInfo, uid, existing)) {
464             return error("Failed to restorecon " + path);
465         }
466 
467         // Remember inode numbers of cache directories so that we can clear
468         // contents while CE storage is locked
469         if (write_path_inode(path, "cache", kXattrInodeCache) ||
470                 write_path_inode(path, "code_cache", kXattrInodeCodeCache)) {
471             return error("Failed to write_path_inode for " + path);
472         }
473 
474         // And return the CE inode of the top-level data directory so we can
475         // clear contents while CE storage is locked
476         if (_aidl_return != nullptr) {
477             ino_t result;
478             if (get_path_inode(path, &result) != 0) {
479                 return error("Failed to get_path_inode for " + path);
480             }
481             *_aidl_return = static_cast<uint64_t>(result);
482         }
483     }
484     if (flags & FLAG_STORAGE_DE) {
485         auto path = create_data_user_de_package_path(uuid_, userId, pkgname);
486         bool existing = (access(path.c_str(), F_OK) == 0);
487 
488         if (prepare_app_dir(path, targetMode, uid) ||
489                 prepare_app_cache_dir(path, "cache", 02771, uid, cacheGid) ||
490                 prepare_app_cache_dir(path, "code_cache", 02771, uid, cacheGid)) {
491             return error("Failed to prepare " + path);
492         }
493 
494         // Consider restorecon over contents if label changed
495         if (restorecon_app_data_lazy(path, seInfo, uid, existing) ||
496                 restorecon_app_data_lazy(path, "cache", seInfo, uid, existing) ||
497                 restorecon_app_data_lazy(path, "code_cache", seInfo, uid, existing)) {
498             return error("Failed to restorecon " + path);
499         }
500 
501         if (!prepare_app_profile_dir(packageName, appId, userId)) {
502             return error("Failed to prepare profiles for " + packageName);
503         }
504     }
505     return ok();
506 }
507 
508 
createAppData(const android::os::CreateAppDataArgs & args,android::os::CreateAppDataResult * _aidl_return)509 binder::Status InstalldNativeService::createAppData(
510         const android::os::CreateAppDataArgs& args,
511         android::os::CreateAppDataResult* _aidl_return) {
512     ENFORCE_UID(AID_SYSTEM);
513     std::lock_guard<std::recursive_mutex> lock(mLock);
514 
515     int64_t ceDataInode = -1;
516     auto status = createAppData(args.uuid, args.packageName, args.userId, args.flags, args.appId,
517                                 args.seInfo, args.targetSdkVersion, &ceDataInode);
518     _aidl_return->ceDataInode = ceDataInode;
519     _aidl_return->exceptionCode = status.exceptionCode();
520     _aidl_return->exceptionMessage = status.exceptionMessage();
521     return ok();
522 }
523 
createAppDataBatched(const std::vector<android::os::CreateAppDataArgs> & args,std::vector<android::os::CreateAppDataResult> * _aidl_return)524 binder::Status InstalldNativeService::createAppDataBatched(
525         const std::vector<android::os::CreateAppDataArgs>& args,
526         std::vector<android::os::CreateAppDataResult>* _aidl_return) {
527     ENFORCE_UID(AID_SYSTEM);
528     std::lock_guard<std::recursive_mutex> lock(mLock);
529 
530     std::vector<android::os::CreateAppDataResult> results;
531     for (auto arg : args) {
532         android::os::CreateAppDataResult result;
533         createAppData(arg, &result);
534         results.push_back(result);
535     }
536     *_aidl_return = results;
537     return ok();
538 }
539 
migrateAppData(const std::optional<std::string> & uuid,const std::string & packageName,int32_t userId,int32_t flags)540 binder::Status InstalldNativeService::migrateAppData(const std::optional<std::string>& uuid,
541         const std::string& packageName, int32_t userId, int32_t flags) {
542     ENFORCE_UID(AID_SYSTEM);
543     CHECK_ARGUMENT_UUID(uuid);
544     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
545     std::lock_guard<std::recursive_mutex> lock(mLock);
546 
547     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
548     const char* pkgname = packageName.c_str();
549 
550     // This method only exists to upgrade system apps that have requested
551     // forceDeviceEncrypted, so their default storage always lives in a
552     // consistent location.  This only works on non-FBE devices, since we
553     // never want to risk exposing data on a device with real CE/DE storage.
554 
555     auto ce_path = create_data_user_ce_package_path(uuid_, userId, pkgname);
556     auto de_path = create_data_user_de_package_path(uuid_, userId, pkgname);
557 
558     // If neither directory is marked as default, assume CE is default
559     if (getxattr(ce_path.c_str(), kXattrDefault, nullptr, 0) == -1
560             && getxattr(de_path.c_str(), kXattrDefault, nullptr, 0) == -1) {
561         if (setxattr(ce_path.c_str(), kXattrDefault, nullptr, 0, 0) != 0) {
562             return error("Failed to mark default storage " + ce_path);
563         }
564     }
565 
566     // Migrate default data location if needed
567     auto target = (flags & FLAG_STORAGE_DE) ? de_path : ce_path;
568     auto source = (flags & FLAG_STORAGE_DE) ? ce_path : de_path;
569 
570     if (getxattr(target.c_str(), kXattrDefault, nullptr, 0) == -1) {
571         LOG(WARNING) << "Requested default storage " << target
572                 << " is not active; migrating from " << source;
573         if (delete_dir_contents_and_dir(target) != 0) {
574             return error("Failed to delete " + target);
575         }
576         if (rename(source.c_str(), target.c_str()) != 0) {
577             return error("Failed to rename " + source + " to " + target);
578         }
579     }
580 
581     return ok();
582 }
583 
584 
clearAppProfiles(const std::string & packageName,const std::string & profileName)585 binder::Status InstalldNativeService::clearAppProfiles(const std::string& packageName,
586         const std::string& profileName) {
587     ENFORCE_UID(AID_SYSTEM);
588     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
589     std::lock_guard<std::recursive_mutex> lock(mLock);
590 
591     binder::Status res = ok();
592     if (!clear_primary_reference_profile(packageName, profileName)) {
593         res = error("Failed to clear reference profile for " + packageName);
594     }
595     if (!clear_primary_current_profiles(packageName, profileName)) {
596         res = error("Failed to clear current profiles for " + packageName);
597     }
598     return res;
599 }
600 
clearAppData(const std::optional<std::string> & uuid,const std::string & packageName,int32_t userId,int32_t flags,int64_t ceDataInode)601 binder::Status InstalldNativeService::clearAppData(const std::optional<std::string>& uuid,
602         const std::string& packageName, int32_t userId, int32_t flags, int64_t ceDataInode) {
603     ENFORCE_UID(AID_SYSTEM);
604     CHECK_ARGUMENT_UUID(uuid);
605     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
606     std::lock_guard<std::recursive_mutex> lock(mLock);
607 
608     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
609     const char* pkgname = packageName.c_str();
610 
611     binder::Status res = ok();
612     if (flags & FLAG_STORAGE_CE) {
613         auto path = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInode);
614         if (flags & FLAG_CLEAR_CACHE_ONLY) {
615             path = read_path_inode(path, "cache", kXattrInodeCache);
616         } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
617             path = read_path_inode(path, "code_cache", kXattrInodeCodeCache);
618         }
619         if (access(path.c_str(), F_OK) == 0) {
620             if (delete_dir_contents(path) != 0) {
621                 res = error("Failed to delete contents of " + path);
622             } else if ((flags & (FLAG_CLEAR_CACHE_ONLY | FLAG_CLEAR_CODE_CACHE_ONLY)) == 0) {
623                 remove_path_xattr(path, kXattrInodeCache);
624                 remove_path_xattr(path, kXattrInodeCodeCache);
625             }
626         }
627     }
628     if (flags & FLAG_STORAGE_DE) {
629         std::string suffix = "";
630         bool only_cache = false;
631         if (flags & FLAG_CLEAR_CACHE_ONLY) {
632             suffix = CACHE_DIR_POSTFIX;
633             only_cache = true;
634         } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
635             suffix = CODE_CACHE_DIR_POSTFIX;
636             only_cache = true;
637         }
638 
639         auto path = create_data_user_de_package_path(uuid_, userId, pkgname) + suffix;
640         if (access(path.c_str(), F_OK) == 0) {
641             if (delete_dir_contents(path) != 0) {
642                 res = error("Failed to delete contents of " + path);
643             }
644         }
645     }
646     if (flags & FLAG_STORAGE_EXTERNAL) {
647         std::lock_guard<std::recursive_mutex> lock(mMountsLock);
648         for (const auto& n : mStorageMounts) {
649             auto extPath = n.second;
650 
651             if (android::base::GetBoolProperty(kFuseProp, false)) {
652                 std::regex re("^\\/mnt\\/pass_through\\/[0-9]+\\/emulated");
653                 if (std::regex_match(extPath, re)) {
654                     extPath += "/" + std::to_string(userId);
655                 }
656             } else {
657                 if (n.first.compare(0, 14, "/mnt/media_rw/") != 0) {
658                     extPath += StringPrintf("/%d", userId);
659                 } else if (userId != 0) {
660                     // TODO: support devices mounted under secondary users
661                     continue;
662                 }
663             }
664 
665             if (flags & FLAG_CLEAR_CACHE_ONLY) {
666                 // Clear only cached data from shared storage
667                 auto path = StringPrintf("%s/Android/data/%s/cache", extPath.c_str(), pkgname);
668                 if (delete_dir_contents(path, true) != 0) {
669                     res = error("Failed to delete contents of " + path);
670                 }
671             } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
672                 // No code cache on shared storage
673             } else {
674                 // Clear everything on shared storage
675                 auto path = StringPrintf("%s/Android/data/%s", extPath.c_str(), pkgname);
676                 if (delete_dir_contents(path, true) != 0) {
677                     res = error("Failed to delete contents of " + path);
678                 }
679                 path = StringPrintf("%s/Android/media/%s", extPath.c_str(), pkgname);
680                 if (delete_dir_contents(path, true) != 0) {
681                     res = error("Failed to delete contents of " + path);
682                 }
683                 // Note that we explicitly don't delete OBBs - those are only removed on
684                 // app uninstall.
685             }
686         }
687     }
688     return res;
689 }
690 
destroy_app_reference_profile(const std::string & pkgname)691 static int destroy_app_reference_profile(const std::string& pkgname) {
692     return delete_dir_contents_and_dir(
693         create_primary_reference_profile_package_dir_path(pkgname),
694         /*ignore_if_missing*/ true);
695 }
696 
destroy_app_current_profiles(const std::string & pkgname,userid_t userid)697 static int destroy_app_current_profiles(const std::string& pkgname, userid_t userid) {
698     return delete_dir_contents_and_dir(
699         create_primary_current_profile_package_dir_path(userid, pkgname),
700         /*ignore_if_missing*/ true);
701 }
702 
destroyAppProfiles(const std::string & packageName)703 binder::Status InstalldNativeService::destroyAppProfiles(const std::string& packageName) {
704     ENFORCE_UID(AID_SYSTEM);
705     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
706     std::lock_guard<std::recursive_mutex> lock(mLock);
707 
708     binder::Status res = ok();
709     std::vector<userid_t> users = get_known_users(/*volume_uuid*/ nullptr);
710     for (auto user : users) {
711         if (destroy_app_current_profiles(packageName, user) != 0) {
712             res = error("Failed to destroy current profiles for " + packageName);
713         }
714     }
715     if (destroy_app_reference_profile(packageName) != 0) {
716         res = error("Failed to destroy reference profile for " + packageName);
717     }
718     return res;
719 }
720 
destroyAppData(const std::optional<std::string> & uuid,const std::string & packageName,int32_t userId,int32_t flags,int64_t ceDataInode)721 binder::Status InstalldNativeService::destroyAppData(const std::optional<std::string>& uuid,
722         const std::string& packageName, int32_t userId, int32_t flags, int64_t ceDataInode) {
723     ENFORCE_UID(AID_SYSTEM);
724     CHECK_ARGUMENT_UUID(uuid);
725     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
726     std::lock_guard<std::recursive_mutex> lock(mLock);
727 
728     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
729     const char* pkgname = packageName.c_str();
730 
731     binder::Status res = ok();
732     if (flags & FLAG_STORAGE_CE) {
733         auto path = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInode);
734         if (delete_dir_contents_and_dir(path) != 0) {
735             res = error("Failed to delete " + path);
736         }
737     }
738     if (flags & FLAG_STORAGE_DE) {
739         auto path = create_data_user_de_package_path(uuid_, userId, pkgname);
740         if (delete_dir_contents_and_dir(path) != 0) {
741             res = error("Failed to delete " + path);
742         }
743         if ((flags & FLAG_CLEAR_APP_DATA_KEEP_ART_PROFILES) == 0) {
744             destroy_app_current_profiles(packageName, userId);
745             // TODO(calin): If the package is still installed by other users it's probably
746             // beneficial to keep the reference profile around.
747             // Verify if it's ok to do that.
748             destroy_app_reference_profile(packageName);
749         }
750     }
751     if (flags & FLAG_STORAGE_EXTERNAL) {
752         std::lock_guard<std::recursive_mutex> lock(mMountsLock);
753         for (const auto& n : mStorageMounts) {
754             auto extPath = n.second;
755 
756             if (android::base::GetBoolProperty(kFuseProp, false)) {
757                 std::regex re("^\\/mnt\\/pass_through\\/[0-9]+\\/emulated");
758                 if (std::regex_match(extPath, re)) {
759                     extPath += "/" + std::to_string(userId);
760                 }
761             } else {
762                 if (n.first.compare(0, 14, "/mnt/media_rw/") != 0) {
763                     extPath += StringPrintf("/%d", userId);
764                 } else if (userId != 0) {
765                     // TODO: support devices mounted under secondary users
766                     continue;
767                 }
768             }
769 
770             auto path = StringPrintf("%s/Android/data/%s", extPath.c_str(), pkgname);
771             if (delete_dir_contents_and_dir(path, true) != 0) {
772                 res = error("Failed to delete contents of " + path);
773             }
774 
775             path = StringPrintf("%s/Android/media/%s", extPath.c_str(), pkgname);
776             if (delete_dir_contents_and_dir(path, true) != 0) {
777                 res = error("Failed to delete contents of " + path);
778             }
779             path = StringPrintf("%s/Android/obb/%s", extPath.c_str(), pkgname);
780             if (delete_dir_contents_and_dir(path, true) != 0) {
781                 res = error("Failed to delete contents of " + path);
782             }
783         }
784     }
785     return res;
786 }
787 
get_cache_gid(uid_t uid)788 static gid_t get_cache_gid(uid_t uid) {
789     int32_t gid = multiuser_get_cache_gid(multiuser_get_user_id(uid), multiuser_get_app_id(uid));
790     return (gid != -1) ? gid : uid;
791 }
792 
fixupAppData(const std::optional<std::string> & uuid,int32_t flags)793 binder::Status InstalldNativeService::fixupAppData(const std::optional<std::string>& uuid,
794         int32_t flags) {
795     ENFORCE_UID(AID_SYSTEM);
796     CHECK_ARGUMENT_UUID(uuid);
797     std::lock_guard<std::recursive_mutex> lock(mLock);
798 
799     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
800     for (auto user : get_known_users(uuid_)) {
801         ATRACE_BEGIN("fixup user");
802         FTS* fts;
803         FTSENT* p;
804         auto ce_path = create_data_user_ce_path(uuid_, user);
805         auto de_path = create_data_user_de_path(uuid_, user);
806         char *argv[] = { (char*) ce_path.c_str(), (char*) de_path.c_str(), nullptr };
807         if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, nullptr))) {
808             return error("Failed to fts_open");
809         }
810         while ((p = fts_read(fts)) != nullptr) {
811             if (p->fts_info == FTS_D && p->fts_level == 1) {
812                 // Track down inodes of cache directories
813                 uint64_t raw = 0;
814                 ino_t inode_cache = 0;
815                 ino_t inode_code_cache = 0;
816                 if (getxattr(p->fts_path, kXattrInodeCache, &raw, sizeof(raw)) == sizeof(raw)) {
817                     inode_cache = raw;
818                 }
819                 if (getxattr(p->fts_path, kXattrInodeCodeCache, &raw, sizeof(raw)) == sizeof(raw)) {
820                     inode_code_cache = raw;
821                 }
822 
823                 // Figure out expected GID of each child
824                 FTSENT* child = fts_children(fts, 0);
825                 while (child != nullptr) {
826                     if ((child->fts_statp->st_ino == inode_cache)
827                             || (child->fts_statp->st_ino == inode_code_cache)
828                             || !strcmp(child->fts_name, "cache")
829                             || !strcmp(child->fts_name, "code_cache")) {
830                         child->fts_number = get_cache_gid(p->fts_statp->st_uid);
831                     } else {
832                         child->fts_number = p->fts_statp->st_uid;
833                     }
834                     child = child->fts_link;
835                 }
836             } else if (p->fts_level >= 2) {
837                 if (p->fts_level > 2) {
838                     // Inherit GID from parent once we're deeper into tree
839                     p->fts_number = p->fts_parent->fts_number;
840                 }
841 
842                 uid_t uid = p->fts_parent->fts_statp->st_uid;
843                 gid_t cache_gid = get_cache_gid(uid);
844                 gid_t expected = p->fts_number;
845                 gid_t actual = p->fts_statp->st_gid;
846                 if (actual == expected) {
847 #if FIXUP_DEBUG
848                     LOG(DEBUG) << "Ignoring " << p->fts_path << " with expected GID " << expected;
849 #endif
850                     if (!(flags & FLAG_FORCE)) {
851                         fts_set(fts, p, FTS_SKIP);
852                     }
853                 } else if ((actual == uid) || (actual == cache_gid)) {
854                     // Only consider fixing up when current GID belongs to app
855                     if (p->fts_info != FTS_D) {
856                         LOG(INFO) << "Fixing " << p->fts_path << " with unexpected GID " << actual
857                                 << " instead of " << expected;
858                     }
859                     switch (p->fts_info) {
860                     case FTS_DP:
861                         // If we're moving towards cache GID, we need to set S_ISGID
862                         if (expected == cache_gid) {
863                             if (chmod(p->fts_path, 02771) != 0) {
864                                 PLOG(WARNING) << "Failed to chmod " << p->fts_path;
865                             }
866                         }
867                         [[fallthrough]]; // also set GID
868                     case FTS_F:
869                         if (chown(p->fts_path, -1, expected) != 0) {
870                             PLOG(WARNING) << "Failed to chown " << p->fts_path;
871                         }
872                         break;
873                     case FTS_SL:
874                     case FTS_SLNONE:
875                         if (lchown(p->fts_path, -1, expected) != 0) {
876                             PLOG(WARNING) << "Failed to chown " << p->fts_path;
877                         }
878                         break;
879                     }
880                 } else {
881                     // Ignore all other GID transitions, since they're kinda shady
882                     LOG(WARNING) << "Ignoring " << p->fts_path << " with unexpected GID " << actual
883                             << " instead of " << expected;
884                     if (!(flags & FLAG_FORCE)) {
885                         fts_set(fts, p, FTS_SKIP);
886                     }
887                 }
888             }
889         }
890         fts_close(fts);
891         ATRACE_END();
892     }
893     return ok();
894 }
895 
copy_directory_recursive(const char * from,const char * to)896 static int32_t copy_directory_recursive(const char* from, const char* to) {
897     char *argv[] = {
898         (char*) kCpPath,
899         (char*) "-F", /* delete any existing destination file first (--remove-destination) */
900         (char*) "-p", /* preserve timestamps, ownership, and permissions */
901         (char*) "-R", /* recurse into subdirectories (DEST must be a directory) */
902         (char*) "-P", /* Do not follow symlinks [default] */
903         (char*) "-d", /* don't dereference symlinks */
904         (char*) from,
905         (char*) to
906     };
907 
908     LOG(DEBUG) << "Copying " << from << " to " << to;
909     return logwrap_fork_execvp(ARRAY_SIZE(argv), argv, nullptr, false, LOG_ALOG, false, nullptr);
910 }
911 
snapshotAppData(const std::optional<std::string> & volumeUuid,const std::string & packageName,int32_t user,int32_t snapshotId,int32_t storageFlags,int64_t * _aidl_return)912 binder::Status InstalldNativeService::snapshotAppData(
913         const std::optional<std::string>& volumeUuid,
914         const std::string& packageName, int32_t user, int32_t snapshotId,
915         int32_t storageFlags, int64_t* _aidl_return) {
916     ENFORCE_UID(AID_SYSTEM);
917     CHECK_ARGUMENT_UUID_IS_TEST_OR_NULL(volumeUuid);
918     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
919     std::lock_guard<std::recursive_mutex> lock(mLock);
920 
921     const char* volume_uuid = volumeUuid ? volumeUuid->c_str() : nullptr;
922     const char* package_name = packageName.c_str();
923 
924     binder::Status res = ok();
925     // Default result to 0, it will be populated with inode of ce data snapshot
926     // if FLAG_STORAGE_CE has been passed.
927     if (_aidl_return != nullptr) *_aidl_return = 0;
928 
929     bool clear_ce_on_exit = false;
930     bool clear_de_on_exit = false;
931 
932     auto deleter = [&clear_ce_on_exit, &clear_de_on_exit, &volume_uuid, &user, &package_name,
933             &snapshotId] {
934         if (clear_de_on_exit) {
935             auto to = create_data_misc_de_rollback_package_path(volume_uuid, user, snapshotId,
936                     package_name);
937             if (delete_dir_contents(to.c_str(), 1, nullptr) != 0) {
938                 LOG(WARNING) << "Failed to delete app data snapshot: " << to;
939             }
940         }
941 
942         if (clear_ce_on_exit) {
943             auto to = create_data_misc_ce_rollback_package_path(volume_uuid, user, snapshotId,
944                     package_name);
945             if (delete_dir_contents(to.c_str(), 1, nullptr) != 0) {
946                 LOG(WARNING) << "Failed to delete app data snapshot: " << to;
947             }
948         }
949     };
950 
951     auto scope_guard = android::base::make_scope_guard(deleter);
952 
953     if (storageFlags & FLAG_STORAGE_DE) {
954         auto from = create_data_user_de_package_path(volume_uuid, user, package_name);
955         auto to = create_data_misc_de_rollback_path(volume_uuid, user, snapshotId);
956         auto rollback_package_path = create_data_misc_de_rollback_package_path(volume_uuid, user,
957             snapshotId, package_name);
958 
959         int rc = create_dir_if_needed(to.c_str(), kRollbackFolderMode);
960         if (rc != 0) {
961             return error(rc, "Failed to create folder " + to);
962         }
963 
964         rc = delete_dir_contents(rollback_package_path, true /* ignore_if_missing */);
965         if (rc != 0) {
966             return error(rc, "Failed clearing existing snapshot " + rollback_package_path);
967         }
968 
969         // Check if we have data to copy.
970         if (access(from.c_str(), F_OK) == 0) {
971           rc = copy_directory_recursive(from.c_str(), to.c_str());
972         }
973         if (rc != 0) {
974             res = error(rc, "Failed copying " + from + " to " + to);
975             clear_de_on_exit = true;
976             return res;
977         }
978     }
979 
980     // The app may not have any data at all, in which case it's OK to skip here.
981     auto from_ce = create_data_user_ce_package_path(volume_uuid, user, package_name);
982     if (access(from_ce.c_str(), F_OK) != 0) {
983         LOG(INFO) << "Missing source " << from_ce;
984         return ok();
985     }
986 
987     // ce_data_inode is not needed when FLAG_CLEAR_CACHE_ONLY is set.
988     binder::Status clear_cache_result = clearAppData(volumeUuid, packageName, user,
989             storageFlags | FLAG_CLEAR_CACHE_ONLY, 0);
990     if (!clear_cache_result.isOk()) {
991         // It should be fine to continue snapshot if we for some reason failed
992         // to clear cache.
993         LOG(WARNING) << "Failed to clear cache of app " << packageName;
994     }
995 
996     // ce_data_inode is not needed when FLAG_CLEAR_CODE_CACHE_ONLY is set.
997     binder::Status clear_code_cache_result = clearAppData(volumeUuid, packageName, user,
998             storageFlags | FLAG_CLEAR_CODE_CACHE_ONLY, 0);
999     if (!clear_code_cache_result.isOk()) {
1000         // It should be fine to continue snapshot if we for some reason failed
1001         // to clear code_cache.
1002         LOG(WARNING) << "Failed to clear code_cache of app " << packageName;
1003     }
1004 
1005     if (storageFlags & FLAG_STORAGE_CE) {
1006         auto from = create_data_user_ce_package_path(volume_uuid, user, package_name);
1007         auto to = create_data_misc_ce_rollback_path(volume_uuid, user, snapshotId);
1008         auto rollback_package_path = create_data_misc_ce_rollback_package_path(volume_uuid, user,
1009             snapshotId, package_name);
1010 
1011         int rc = create_dir_if_needed(to.c_str(), kRollbackFolderMode);
1012         if (rc != 0) {
1013             return error(rc, "Failed to create folder " + to);
1014         }
1015 
1016         rc = delete_dir_contents(rollback_package_path, true /* ignore_if_missing */);
1017         if (rc != 0) {
1018             return error(rc, "Failed clearing existing snapshot " + rollback_package_path);
1019         }
1020 
1021         rc = copy_directory_recursive(from.c_str(), to.c_str());
1022         if (rc != 0) {
1023             res = error(rc, "Failed copying " + from + " to " + to);
1024             clear_ce_on_exit = true;
1025             return res;
1026         }
1027         if (_aidl_return != nullptr) {
1028             auto ce_snapshot_path = create_data_misc_ce_rollback_package_path(volume_uuid, user,
1029                     snapshotId, package_name);
1030             rc = get_path_inode(ce_snapshot_path, reinterpret_cast<ino_t*>(_aidl_return));
1031             if (rc != 0) {
1032                 res = error(rc, "Failed to get_path_inode for " + ce_snapshot_path);
1033                 clear_ce_on_exit = true;
1034                 return res;
1035             }
1036         }
1037     }
1038 
1039     return res;
1040 }
1041 
restoreAppDataSnapshot(const std::optional<std::string> & volumeUuid,const std::string & packageName,const int32_t appId,const std::string & seInfo,const int32_t user,const int32_t snapshotId,int32_t storageFlags)1042 binder::Status InstalldNativeService::restoreAppDataSnapshot(
1043         const std::optional<std::string>& volumeUuid, const std::string& packageName,
1044         const int32_t appId, const std::string& seInfo, const int32_t user,
1045         const int32_t snapshotId, int32_t storageFlags) {
1046     ENFORCE_UID(AID_SYSTEM);
1047     CHECK_ARGUMENT_UUID_IS_TEST_OR_NULL(volumeUuid);
1048     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1049     std::lock_guard<std::recursive_mutex> lock(mLock);
1050 
1051     const char* volume_uuid = volumeUuid ? volumeUuid->c_str() : nullptr;
1052     const char* package_name = packageName.c_str();
1053 
1054     auto from_ce = create_data_misc_ce_rollback_package_path(volume_uuid,
1055             user, snapshotId, package_name);
1056     auto from_de = create_data_misc_de_rollback_package_path(volume_uuid,
1057             user, snapshotId, package_name);
1058 
1059     const bool needs_ce_rollback = (storageFlags & FLAG_STORAGE_CE) &&
1060         (access(from_ce.c_str(), F_OK) == 0);
1061     const bool needs_de_rollback = (storageFlags & FLAG_STORAGE_DE) &&
1062         (access(from_de.c_str(), F_OK) == 0);
1063 
1064     if (!needs_ce_rollback && !needs_de_rollback) {
1065         return ok();
1066     }
1067 
1068     // We know we're going to rollback one of the CE or DE data, so we clear
1069     // application data first. Note that it's possible that we're asked to
1070     // restore both CE & DE data but that one of the restores fail. Leaving the
1071     // app with no data in those cases is arguably better than leaving the app
1072     // with mismatched / stale data.
1073     LOG(INFO) << "Clearing app data for " << packageName << " to restore snapshot.";
1074     // It's fine to pass 0 as ceDataInode here, because restoreAppDataSnapshot
1075     // can only be called when user unlocks the phone, meaning that CE user data
1076     // is decrypted.
1077     binder::Status res = clearAppData(volumeUuid, packageName, user, storageFlags,
1078             0 /* ceDataInode */);
1079     if (!res.isOk()) {
1080         return res;
1081     }
1082 
1083     if (needs_ce_rollback) {
1084         auto to_ce = create_data_user_ce_path(volume_uuid, user);
1085         int rc = copy_directory_recursive(from_ce.c_str(), to_ce.c_str());
1086         if (rc != 0) {
1087             res = error(rc, "Failed copying " + from_ce + " to " + to_ce);
1088             return res;
1089         }
1090         delete_dir_contents_and_dir(from_ce, true /* ignore_if_missing */);
1091     }
1092 
1093     if (needs_de_rollback) {
1094         auto to_de = create_data_user_de_path(volume_uuid, user);
1095         int rc = copy_directory_recursive(from_de.c_str(), to_de.c_str());
1096         if (rc != 0) {
1097             if (needs_ce_rollback) {
1098                 auto ce_data = create_data_user_ce_package_path(volume_uuid, user, package_name);
1099                 LOG(WARNING) << "de_data rollback failed. Erasing rolled back ce_data " << ce_data;
1100                 if (delete_dir_contents(ce_data.c_str(), 1, nullptr) != 0) {
1101                     LOG(WARNING) << "Failed to delete rolled back ce_data " << ce_data;
1102                 }
1103             }
1104             res = error(rc, "Failed copying " + from_de + " to " + to_de);
1105             return res;
1106         }
1107         delete_dir_contents_and_dir(from_de, true /* ignore_if_missing */);
1108     }
1109 
1110     // Finally, restore the SELinux label on the app data.
1111     return restoreconAppData(volumeUuid, packageName, user, storageFlags, appId, seInfo);
1112 }
1113 
destroyAppDataSnapshot(const std::optional<std::string> & volumeUuid,const std::string & packageName,const int32_t user,const int64_t ceSnapshotInode,const int32_t snapshotId,int32_t storageFlags)1114 binder::Status InstalldNativeService::destroyAppDataSnapshot(
1115         const std::optional<std::string> &volumeUuid, const std::string& packageName,
1116         const int32_t user, const int64_t ceSnapshotInode, const int32_t snapshotId,
1117         int32_t storageFlags) {
1118     ENFORCE_UID(AID_SYSTEM);
1119     CHECK_ARGUMENT_UUID_IS_TEST_OR_NULL(volumeUuid);
1120     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1121     std::lock_guard<std::recursive_mutex> lock(mLock);
1122 
1123     const char* volume_uuid = volumeUuid ? volumeUuid->c_str() : nullptr;
1124     const char* package_name = packageName.c_str();
1125 
1126     if (storageFlags & FLAG_STORAGE_DE) {
1127         auto de_snapshot_path = create_data_misc_de_rollback_package_path(volume_uuid,
1128                 user, snapshotId, package_name);
1129 
1130         int res = delete_dir_contents_and_dir(de_snapshot_path, true /* ignore_if_missing */);
1131         if (res != 0) {
1132             return error(res, "Failed clearing snapshot " + de_snapshot_path);
1133         }
1134     }
1135 
1136     if (storageFlags & FLAG_STORAGE_CE) {
1137         auto ce_snapshot_path = create_data_misc_ce_rollback_package_path(volume_uuid,
1138                 user, snapshotId, package_name, ceSnapshotInode);
1139         int res = delete_dir_contents_and_dir(ce_snapshot_path, true /* ignore_if_missing */);
1140         if (res != 0) {
1141             return error(res, "Failed clearing snapshot " + ce_snapshot_path);
1142         }
1143     }
1144     return ok();
1145 }
1146 
destroyCeSnapshotsNotSpecified(const std::optional<std::string> & volumeUuid,const int32_t user,const std::vector<int32_t> & retainSnapshotIds)1147 binder::Status InstalldNativeService::destroyCeSnapshotsNotSpecified(
1148         const std::optional<std::string> &volumeUuid, const int32_t user,
1149         const std::vector<int32_t>& retainSnapshotIds) {
1150     ENFORCE_UID(AID_SYSTEM);
1151     CHECK_ARGUMENT_UUID_IS_TEST_OR_NULL(volumeUuid);
1152     std::lock_guard<std::recursive_mutex> lock(mLock);
1153 
1154     const char* volume_uuid = volumeUuid ? volumeUuid->c_str() : nullptr;
1155 
1156     auto base_path = create_data_misc_ce_rollback_base_path(volume_uuid, user);
1157 
1158     std::unique_ptr<DIR, decltype(&closedir)> dir(opendir(base_path.c_str()), closedir);
1159     if (!dir) {
1160         return error(-1, "Failed to open rollback base dir " + base_path);
1161     }
1162 
1163     struct dirent* ent;
1164     while ((ent = readdir(dir.get()))) {
1165         if (ent->d_type != DT_DIR) {
1166             continue;
1167         }
1168 
1169         uint snapshot_id;
1170         bool parse_ok = ParseUint(ent->d_name, &snapshot_id);
1171         if (parse_ok &&
1172                 std::find(retainSnapshotIds.begin(), retainSnapshotIds.end(),
1173                           snapshot_id) == retainSnapshotIds.end()) {
1174             auto rollback_path = create_data_misc_ce_rollback_path(
1175                 volume_uuid, user, snapshot_id);
1176             int res = delete_dir_contents_and_dir(rollback_path, true /* ignore_if_missing */);
1177             if (res != 0) {
1178                 return error(res, "Failed clearing snapshot " + rollback_path);
1179             }
1180         }
1181     }
1182     return ok();
1183 }
1184 
moveCompleteApp(const std::optional<std::string> & fromUuid,const std::optional<std::string> & toUuid,const std::string & packageName,int32_t appId,const std::string & seInfo,int32_t targetSdkVersion,const std::string & fromCodePath)1185 binder::Status InstalldNativeService::moveCompleteApp(const std::optional<std::string>& fromUuid,
1186         const std::optional<std::string>& toUuid, const std::string& packageName,
1187         int32_t appId, const std::string& seInfo,
1188         int32_t targetSdkVersion, const std::string& fromCodePath) {
1189     ENFORCE_UID(AID_SYSTEM);
1190     CHECK_ARGUMENT_UUID(fromUuid);
1191     CHECK_ARGUMENT_UUID(toUuid);
1192     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1193     std::lock_guard<std::recursive_mutex> lock(mLock);
1194 
1195     const char* from_uuid = fromUuid ? fromUuid->c_str() : nullptr;
1196     const char* to_uuid = toUuid ? toUuid->c_str() : nullptr;
1197     const char* package_name = packageName.c_str();
1198 
1199     binder::Status res = ok();
1200     std::vector<userid_t> users = get_known_users(from_uuid);
1201 
1202     auto to_app_package_path_parent = create_data_app_path(to_uuid);
1203     auto to_app_package_path = StringPrintf("%s/%s", to_app_package_path_parent.c_str(),
1204                                             android::base::Basename(fromCodePath).c_str());
1205 
1206     // Copy app
1207     {
1208         int rc = copy_directory_recursive(fromCodePath.c_str(), to_app_package_path_parent.c_str());
1209         if (rc != 0) {
1210             res = error(rc, "Failed copying " + fromCodePath + " to " + to_app_package_path);
1211             goto fail;
1212         }
1213 
1214         if (selinux_android_restorecon(to_app_package_path.c_str(), SELINUX_ANDROID_RESTORECON_RECURSE) != 0) {
1215             res = error("Failed to restorecon " + to_app_package_path);
1216             goto fail;
1217         }
1218     }
1219 
1220     // Copy private data for all known users
1221     for (auto user : users) {
1222 
1223         // Data source may not exist for all users; that's okay
1224         auto from_ce = create_data_user_ce_package_path(from_uuid, user, package_name);
1225         if (access(from_ce.c_str(), F_OK) != 0) {
1226             LOG(INFO) << "Missing source " << from_ce;
1227             continue;
1228         }
1229 
1230         if (!createAppData(toUuid, packageName, user, FLAG_STORAGE_CE | FLAG_STORAGE_DE, appId,
1231                 seInfo, targetSdkVersion, nullptr).isOk()) {
1232             res = error("Failed to create package target");
1233             goto fail;
1234         }
1235 
1236         {
1237             auto from = create_data_user_de_package_path(from_uuid, user, package_name);
1238             auto to = create_data_user_de_path(to_uuid, user);
1239 
1240             int rc = copy_directory_recursive(from.c_str(), to.c_str());
1241             if (rc != 0) {
1242                 res = error(rc, "Failed copying " + from + " to " + to);
1243                 goto fail;
1244             }
1245         }
1246         {
1247             auto from = create_data_user_ce_package_path(from_uuid, user, package_name);
1248             auto to = create_data_user_ce_path(to_uuid, user);
1249 
1250             int rc = copy_directory_recursive(from.c_str(), to.c_str());
1251             if (rc != 0) {
1252                 res = error(rc, "Failed copying " + from + " to " + to);
1253                 goto fail;
1254             }
1255         }
1256 
1257         if (!restoreconAppData(toUuid, packageName, user, FLAG_STORAGE_CE | FLAG_STORAGE_DE,
1258                 appId, seInfo).isOk()) {
1259             res = error("Failed to restorecon");
1260             goto fail;
1261         }
1262     }
1263 
1264     // We let the framework scan the new location and persist that before
1265     // deleting the data in the old location; this ordering ensures that
1266     // we can recover from things like battery pulls.
1267     return ok();
1268 
1269 fail:
1270     // Nuke everything we might have already copied
1271     {
1272         if (delete_dir_contents(to_app_package_path.c_str(), 1, nullptr) != 0) {
1273             LOG(WARNING) << "Failed to rollback " << to_app_package_path;
1274         }
1275     }
1276     for (auto user : users) {
1277         {
1278             auto to = create_data_user_de_package_path(to_uuid, user, package_name);
1279             if (delete_dir_contents(to.c_str(), 1, nullptr) != 0) {
1280                 LOG(WARNING) << "Failed to rollback " << to;
1281             }
1282         }
1283         {
1284             auto to = create_data_user_ce_package_path(to_uuid, user, package_name);
1285             if (delete_dir_contents(to.c_str(), 1, nullptr) != 0) {
1286                 LOG(WARNING) << "Failed to rollback " << to;
1287             }
1288         }
1289     }
1290     return res;
1291 }
1292 
createUserData(const std::optional<std::string> & uuid,int32_t userId,int32_t userSerial ATTRIBUTE_UNUSED,int32_t flags)1293 binder::Status InstalldNativeService::createUserData(const std::optional<std::string>& uuid,
1294         int32_t userId, int32_t userSerial ATTRIBUTE_UNUSED, int32_t flags) {
1295     ENFORCE_UID(AID_SYSTEM);
1296     CHECK_ARGUMENT_UUID(uuid);
1297     std::lock_guard<std::recursive_mutex> lock(mLock);
1298 
1299     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1300     if (flags & FLAG_STORAGE_DE) {
1301         if (uuid_ == nullptr) {
1302             if (ensure_config_user_dirs(userId) != 0) {
1303                 return error(StringPrintf("Failed to ensure dirs for %d", userId));
1304             }
1305         }
1306     }
1307 
1308     return ok();
1309 }
1310 
destroyUserData(const std::optional<std::string> & uuid,int32_t userId,int32_t flags)1311 binder::Status InstalldNativeService::destroyUserData(const std::optional<std::string>& uuid,
1312         int32_t userId, int32_t flags) {
1313     ENFORCE_UID(AID_SYSTEM);
1314     CHECK_ARGUMENT_UUID(uuid);
1315     std::lock_guard<std::recursive_mutex> lock(mLock);
1316 
1317     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1318     binder::Status res = ok();
1319     if (flags & FLAG_STORAGE_DE) {
1320         auto path = create_data_user_de_path(uuid_, userId);
1321         if (delete_dir_contents_and_dir(path, true) != 0) {
1322             res = error("Failed to delete " + path);
1323         }
1324         if (uuid_ == nullptr) {
1325             path = create_data_misc_legacy_path(userId);
1326             if (delete_dir_contents_and_dir(path, true) != 0) {
1327                 res = error("Failed to delete " + path);
1328             }
1329             path = create_primary_cur_profile_dir_path(userId);
1330             if (delete_dir_contents_and_dir(path, true) != 0) {
1331                 res = error("Failed to delete " + path);
1332             }
1333         }
1334     }
1335     if (flags & FLAG_STORAGE_CE) {
1336         auto path = create_data_user_ce_path(uuid_, userId);
1337         if (delete_dir_contents_and_dir(path, true) != 0) {
1338             res = error("Failed to delete " + path);
1339         }
1340         path = findDataMediaPath(uuid, userId);
1341         if (delete_dir_contents_and_dir(path, true) != 0) {
1342             res = error("Failed to delete " + path);
1343         }
1344     }
1345     return res;
1346 }
1347 
freeCache(const std::optional<std::string> & uuid,int64_t targetFreeBytes,int64_t cacheReservedBytes,int32_t flags)1348 binder::Status InstalldNativeService::freeCache(const std::optional<std::string>& uuid,
1349         int64_t targetFreeBytes, int64_t cacheReservedBytes, int32_t flags) {
1350     ENFORCE_UID(AID_SYSTEM);
1351     CHECK_ARGUMENT_UUID(uuid);
1352     std::lock_guard<std::recursive_mutex> lock(mLock);
1353 
1354     auto uuidString = uuid.value_or("");
1355     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1356     auto data_path = create_data_path(uuid_);
1357     auto noop = (flags & FLAG_FREE_CACHE_NOOP);
1358 
1359     int64_t free = data_disk_free(data_path);
1360     if (free < 0) {
1361         return error("Failed to determine free space for " + data_path);
1362     }
1363 
1364     int64_t cleared = 0;
1365     int64_t needed = targetFreeBytes - free;
1366     LOG(DEBUG) << "Device " << data_path << " has " << free << " free; requested "
1367             << targetFreeBytes << "; needed " << needed;
1368 
1369     if (free >= targetFreeBytes) {
1370         return ok();
1371     }
1372 
1373     if (flags & FLAG_FREE_CACHE_V2) {
1374         // This new cache strategy fairly removes files from UIDs by deleting
1375         // files from the UIDs which are most over their allocated quota
1376 
1377         // 1. Create trackers for every known UID
1378         ATRACE_BEGIN("create");
1379         std::unordered_map<uid_t, std::shared_ptr<CacheTracker>> trackers;
1380         for (auto user : get_known_users(uuid_)) {
1381             FTS *fts;
1382             FTSENT *p;
1383             auto ce_path = create_data_user_ce_path(uuid_, user);
1384             auto de_path = create_data_user_de_path(uuid_, user);
1385             auto media_path = findDataMediaPath(uuid, user) + "/Android/data/";
1386             char *argv[] = { (char*) ce_path.c_str(), (char*) de_path.c_str(),
1387                     (char*) media_path.c_str(), nullptr };
1388             if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, nullptr))) {
1389                 return error("Failed to fts_open");
1390             }
1391             while ((p = fts_read(fts)) != nullptr) {
1392                 if (p->fts_info == FTS_D && p->fts_level == 1) {
1393                     uid_t uid = p->fts_statp->st_uid;
1394                     if (multiuser_get_app_id(uid) == AID_MEDIA_RW) {
1395                         uid = (multiuser_get_app_id(p->fts_statp->st_gid) - AID_EXT_GID_START)
1396                                 + AID_APP_START;
1397                     }
1398                     auto search = trackers.find(uid);
1399                     if (search != trackers.end()) {
1400                         search->second->addDataPath(p->fts_path);
1401                     } else {
1402                         auto tracker = std::shared_ptr<CacheTracker>(new CacheTracker(
1403                                 multiuser_get_user_id(uid), multiuser_get_app_id(uid), uuidString));
1404                         tracker->addDataPath(p->fts_path);
1405                         {
1406                             std::lock_guard<std::recursive_mutex> lock(mQuotasLock);
1407                             tracker->cacheQuota = mCacheQuotas[uid];
1408                         }
1409                         if (tracker->cacheQuota == 0) {
1410 #if MEASURE_DEBUG
1411                             LOG(WARNING) << "UID " << uid << " has no cache quota; assuming 64MB";
1412 #endif
1413                             tracker->cacheQuota = 67108864;
1414                         }
1415                         trackers[uid] = tracker;
1416                     }
1417                     fts_set(fts, p, FTS_SKIP);
1418                 }
1419             }
1420             fts_close(fts);
1421         }
1422         ATRACE_END();
1423 
1424         // 2. Populate tracker stats and insert into priority queue
1425         ATRACE_BEGIN("populate");
1426         int64_t cacheTotal = 0;
1427         auto cmp = [](std::shared_ptr<CacheTracker> left, std::shared_ptr<CacheTracker> right) {
1428             return (left->getCacheRatio() < right->getCacheRatio());
1429         };
1430         std::priority_queue<std::shared_ptr<CacheTracker>,
1431                 std::vector<std::shared_ptr<CacheTracker>>, decltype(cmp)> queue(cmp);
1432         for (const auto& it : trackers) {
1433             it.second->loadStats();
1434             queue.push(it.second);
1435             cacheTotal += it.second->cacheUsed;
1436         }
1437         ATRACE_END();
1438 
1439         // 3. Bounce across the queue, freeing items from whichever tracker is
1440         // the most over their assigned quota
1441         ATRACE_BEGIN("bounce");
1442         std::shared_ptr<CacheTracker> active;
1443         while (active || !queue.empty()) {
1444             // Only look at apps under quota when explicitly requested
1445             if (active && (active->getCacheRatio() < 10000)
1446                     && !(flags & FLAG_FREE_CACHE_V2_DEFY_QUOTA)) {
1447                 LOG(DEBUG) << "Active ratio " << active->getCacheRatio()
1448                         << " isn't over quota, and defy not requested";
1449                 break;
1450             }
1451 
1452             // Only keep clearing when we haven't pushed into reserved area
1453             if (cacheReservedBytes > 0 && cleared >= (cacheTotal - cacheReservedBytes)) {
1454                 LOG(DEBUG) << "Refusing to clear cached data in reserved space";
1455                 break;
1456             }
1457 
1458             // Find the best tracker to work with; this might involve swapping
1459             // if the active tracker is no longer the most over quota
1460             bool nextBetter = active && !queue.empty()
1461                     && active->getCacheRatio() < queue.top()->getCacheRatio();
1462             if (!active || nextBetter) {
1463                 if (active) {
1464                     // Current tracker still has items, so we'll consider it
1465                     // again later once it bubbles up to surface
1466                     queue.push(active);
1467                 }
1468                 active = queue.top(); queue.pop();
1469                 active->ensureItems();
1470                 continue;
1471             }
1472 
1473             // If no items remain, go find another tracker
1474             if (active->items.empty()) {
1475                 active = nullptr;
1476                 continue;
1477             } else {
1478                 auto item = active->items.back();
1479                 active->items.pop_back();
1480 
1481                 LOG(DEBUG) << "Purging " << item->toString() << " from " << active->toString();
1482                 if (!noop) {
1483                     item->purge();
1484                 }
1485                 active->cacheUsed -= item->size;
1486                 needed -= item->size;
1487                 cleared += item->size;
1488             }
1489 
1490             // Verify that we're actually done before bailing, since sneaky
1491             // apps might be using hardlinks
1492             if (needed <= 0) {
1493                 free = data_disk_free(data_path);
1494                 needed = targetFreeBytes - free;
1495                 if (needed <= 0) {
1496                     break;
1497                 } else {
1498                     LOG(WARNING) << "Expected to be done but still need " << needed;
1499                 }
1500             }
1501         }
1502         ATRACE_END();
1503 
1504     } else {
1505         return error("Legacy cache logic no longer supported");
1506     }
1507 
1508     free = data_disk_free(data_path);
1509     if (free >= targetFreeBytes) {
1510         return ok();
1511     } else {
1512         return error(StringPrintf("Failed to free up %" PRId64 " on %s; final free space %" PRId64,
1513                 targetFreeBytes, data_path.c_str(), free));
1514     }
1515 }
1516 
rmdex(const std::string & codePath,const std::string & instructionSet)1517 binder::Status InstalldNativeService::rmdex(const std::string& codePath,
1518         const std::string& instructionSet) {
1519     ENFORCE_UID(AID_SYSTEM);
1520     CHECK_ARGUMENT_PATH(codePath);
1521     std::lock_guard<std::recursive_mutex> lock(mLock);
1522 
1523     char dex_path[PKG_PATH_MAX];
1524 
1525     const char* path = codePath.c_str();
1526     const char* instruction_set = instructionSet.c_str();
1527 
1528     if (validate_apk_path(path) && validate_system_app_path(path)) {
1529         return error("Invalid path " + codePath);
1530     }
1531 
1532     if (!create_cache_path(dex_path, path, instruction_set)) {
1533         return error("Failed to create cache path for " + codePath);
1534     }
1535 
1536     ALOGV("unlink %s\n", dex_path);
1537     if (unlink(dex_path) < 0) {
1538         // It's ok if we don't have a dalvik cache path. Report error only when the path exists
1539         // but could not be unlinked.
1540         if (errno != ENOENT) {
1541             return error(StringPrintf("Failed to unlink %s", dex_path));
1542         }
1543     }
1544     return ok();
1545 }
1546 
1547 struct stats {
1548     int64_t codeSize;
1549     int64_t dataSize;
1550     int64_t cacheSize;
1551 };
1552 
1553 #if MEASURE_DEBUG
toString(std::vector<int64_t> values)1554 static std::string toString(std::vector<int64_t> values) {
1555     std::stringstream res;
1556     res << "[";
1557     for (size_t i = 0; i < values.size(); i++) {
1558         res << values[i];
1559         if (i < values.size() - 1) {
1560             res << ",";
1561         }
1562     }
1563     res << "]";
1564     return res.str();
1565 }
1566 #endif
1567 
collectQuotaStats(const std::string & uuid,int32_t userId,int32_t appId,struct stats * stats,struct stats * extStats)1568 static void collectQuotaStats(const std::string& uuid, int32_t userId,
1569         int32_t appId, struct stats* stats, struct stats* extStats) {
1570     int64_t space;
1571     uid_t uid = multiuser_get_uid(userId, appId);
1572     if (stats != nullptr) {
1573         if ((space = GetOccupiedSpaceForUid(uuid, uid)) != -1) {
1574             stats->dataSize += space;
1575         }
1576 
1577         int cacheGid = multiuser_get_cache_gid(userId, appId);
1578         if (cacheGid != -1) {
1579             if ((space = GetOccupiedSpaceForGid(uuid, cacheGid)) != -1) {
1580                 stats->cacheSize += space;
1581             }
1582         }
1583 
1584         int sharedGid = multiuser_get_shared_gid(0, appId);
1585         if (sharedGid != -1) {
1586             if ((space = GetOccupiedSpaceForGid(uuid, sharedGid)) != -1) {
1587                 stats->codeSize += space;
1588             }
1589         }
1590     }
1591 
1592     if (extStats != nullptr) {
1593         static const bool supportsSdCardFs = supports_sdcardfs();
1594         space = get_occupied_app_space_external(uuid, userId, appId);
1595 
1596         if (space != -1) {
1597             extStats->dataSize += space;
1598             if (!supportsSdCardFs && stats != nullptr) {
1599                 // On devices without sdcardfs, if internal and external are on
1600                 // the same volume, a uid such as u0_a123 is used for
1601                 // application dirs on both internal and external storage;
1602                 // therefore, substract that amount from internal to make sure
1603                 // we don't count it double.
1604                 stats->dataSize -= space;
1605             }
1606         }
1607 
1608         space = get_occupied_app_cache_space_external(uuid, userId, appId);
1609         if (space != -1) {
1610             extStats->dataSize += space; // cache counts for "data"
1611             extStats->cacheSize += space;
1612             if (!supportsSdCardFs && stats != nullptr) {
1613                 // On devices without sdcardfs, if internal and external are on
1614                 // the same volume, a uid such as u0_a123 is used for both
1615                 // internal and external storage; therefore, substract that
1616                 // amount from internal to make sure we don't count it double.
1617                 stats->dataSize -= space;
1618             }
1619         }
1620 
1621         if (!supportsSdCardFs && stats != nullptr) {
1622             // On devices without sdcardfs, the UID of OBBs on external storage
1623             // matches the regular app UID (eg u0_a123); therefore, to avoid
1624             // OBBs being include in stats->dataSize, compute the OBB size for
1625             // this app, and substract it from the size reported on internal
1626             // storage
1627             long obbProjectId = uid - AID_APP_START + PROJECT_ID_EXT_OBB_START;
1628             int64_t appObbSize = GetOccupiedSpaceForProjectId(uuid, obbProjectId);
1629             stats->dataSize -= appObbSize;
1630         }
1631     }
1632 }
1633 
collectManualStats(const std::string & path,struct stats * stats)1634 static void collectManualStats(const std::string& path, struct stats* stats) {
1635     DIR *d;
1636     int dfd;
1637     struct dirent *de;
1638     struct stat s;
1639 
1640     d = opendir(path.c_str());
1641     if (d == nullptr) {
1642         if (errno != ENOENT) {
1643             PLOG(WARNING) << "Failed to open " << path;
1644         }
1645         return;
1646     }
1647     dfd = dirfd(d);
1648     while ((de = readdir(d))) {
1649         const char *name = de->d_name;
1650 
1651         int64_t size = 0;
1652         if (fstatat(dfd, name, &s, AT_SYMLINK_NOFOLLOW) == 0) {
1653             size = s.st_blocks * 512;
1654         }
1655 
1656         if (de->d_type == DT_DIR) {
1657             if (!strcmp(name, ".")) {
1658                 // Don't recurse, but still count node size
1659             } else if (!strcmp(name, "..")) {
1660                 // Don't recurse or count node size
1661                 continue;
1662             } else {
1663                 // Measure all children nodes
1664                 size = 0;
1665                 calculate_tree_size(StringPrintf("%s/%s", path.c_str(), name), &size);
1666             }
1667 
1668             if (!strcmp(name, "cache") || !strcmp(name, "code_cache")) {
1669                 stats->cacheSize += size;
1670             }
1671         }
1672 
1673         // Legacy symlink isn't owned by app
1674         if (de->d_type == DT_LNK && !strcmp(name, "lib")) {
1675             continue;
1676         }
1677 
1678         // Everything found inside is considered data
1679         stats->dataSize += size;
1680     }
1681     closedir(d);
1682 }
1683 
collectManualStatsForUser(const std::string & path,struct stats * stats,bool exclude_apps=false)1684 static void collectManualStatsForUser(const std::string& path, struct stats* stats,
1685         bool exclude_apps = false) {
1686     DIR *d;
1687     int dfd;
1688     struct dirent *de;
1689     struct stat s;
1690 
1691     d = opendir(path.c_str());
1692     if (d == nullptr) {
1693         if (errno != ENOENT) {
1694             PLOG(WARNING) << "Failed to open " << path;
1695         }
1696         return;
1697     }
1698     dfd = dirfd(d);
1699     while ((de = readdir(d))) {
1700         if (de->d_type == DT_DIR) {
1701             const char *name = de->d_name;
1702             if (fstatat(dfd, name, &s, AT_SYMLINK_NOFOLLOW) != 0) {
1703                 continue;
1704             }
1705             int32_t user_uid = multiuser_get_app_id(s.st_uid);
1706             if (!strcmp(name, ".") || !strcmp(name, "..")) {
1707                 continue;
1708             } else if (exclude_apps && (user_uid >= AID_APP_START && user_uid <= AID_APP_END)) {
1709                 continue;
1710             } else {
1711                 collectManualStats(StringPrintf("%s/%s", path.c_str(), name), stats);
1712             }
1713         }
1714     }
1715     closedir(d);
1716 }
1717 
collectManualExternalStatsForUser(const std::string & path,struct stats * stats)1718 static void collectManualExternalStatsForUser(const std::string& path, struct stats* stats) {
1719     FTS *fts;
1720     FTSENT *p;
1721     char *argv[] = { (char*) path.c_str(), nullptr };
1722     if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, nullptr))) {
1723         PLOG(ERROR) << "Failed to fts_open " << path;
1724         return;
1725     }
1726     while ((p = fts_read(fts)) != nullptr) {
1727         p->fts_number = p->fts_parent->fts_number;
1728         switch (p->fts_info) {
1729         case FTS_D:
1730             if (p->fts_level == 4
1731                     && !strcmp(p->fts_name, "cache")
1732                     && !strcmp(p->fts_parent->fts_parent->fts_name, "data")
1733                     && !strcmp(p->fts_parent->fts_parent->fts_parent->fts_name, "Android")) {
1734                 p->fts_number = 1;
1735             }
1736             [[fallthrough]]; // to count the directory
1737         case FTS_DEFAULT:
1738         case FTS_F:
1739         case FTS_SL:
1740         case FTS_SLNONE:
1741             int64_t size = (p->fts_statp->st_blocks * 512);
1742             if (p->fts_number == 1) {
1743                 stats->cacheSize += size;
1744             }
1745             stats->dataSize += size;
1746             break;
1747         }
1748     }
1749     fts_close(fts);
1750 }
ownsExternalStorage(int32_t appId)1751 static bool ownsExternalStorage(int32_t appId) {
1752     //  Fetch external storage owner appid  and check if it is the same as the
1753     //  current appId whose size is calculated
1754     struct stat s;
1755     auto _picDir = StringPrintf("%s/Pictures", create_data_media_path(nullptr, 0).c_str());
1756     // check if the stat are present
1757     if (stat(_picDir.c_str(), &s) == 0) {
1758         // fetch the appId from the uid of the media app
1759         return ((int32_t)multiuser_get_app_id(s.st_uid) == appId);
1760     }
1761     return false;
1762 }
getAppSize(const std::optional<std::string> & uuid,const std::vector<std::string> & packageNames,int32_t userId,int32_t flags,int32_t appId,const std::vector<int64_t> & ceDataInodes,const std::vector<std::string> & codePaths,std::vector<int64_t> * _aidl_return)1763 binder::Status InstalldNativeService::getAppSize(const std::optional<std::string>& uuid,
1764         const std::vector<std::string>& packageNames, int32_t userId, int32_t flags,
1765         int32_t appId, const std::vector<int64_t>& ceDataInodes,
1766         const std::vector<std::string>& codePaths, std::vector<int64_t>* _aidl_return) {
1767     ENFORCE_UID(AID_SYSTEM);
1768     CHECK_ARGUMENT_UUID(uuid);
1769     for (const auto& packageName : packageNames) {
1770         CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1771     }
1772     for (const auto& codePath : codePaths) {
1773         CHECK_ARGUMENT_PATH(codePath);
1774     }
1775     // NOTE: Locking is relaxed on this method, since it's limited to
1776     // read-only measurements without mutation.
1777 
1778     // When modifying this logic, always verify using tests:
1779     // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetAppSize
1780 
1781 #if MEASURE_DEBUG
1782     LOG(INFO) << "Measuring user " << userId << " app " << appId;
1783 #endif
1784 
1785     // Here's a summary of the common storage locations across the platform,
1786     // and how they're each tagged:
1787     //
1788     // /data/app/com.example                           UID system
1789     // /data/app/com.example/oat                       UID system
1790     // /data/user/0/com.example                        UID u0_a10      GID u0_a10
1791     // /data/user/0/com.example/cache                  UID u0_a10      GID u0_a10_cache
1792     // /data/media/0/foo.txt                           UID u0_media_rw
1793     // /data/media/0/bar.jpg                           UID u0_media_rw GID u0_media_image
1794     // /data/media/0/Android/data/com.example          UID u0_media_rw GID u0_a10_ext
1795     // /data/media/0/Android/data/com.example/cache    UID u0_media_rw GID u0_a10_ext_cache
1796     // /data/media/obb/com.example                     UID system
1797 
1798     struct stats stats;
1799     struct stats extStats;
1800     memset(&stats, 0, sizeof(stats));
1801     memset(&extStats, 0, sizeof(extStats));
1802 
1803     auto uuidString = uuid.value_or("");
1804     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1805 
1806     if (!IsQuotaSupported(uuidString)) {
1807         flags &= ~FLAG_USE_QUOTA;
1808     }
1809 
1810     ATRACE_BEGIN("obb");
1811     for (const auto& packageName : packageNames) {
1812         auto obbCodePath = create_data_media_package_path(uuid_, userId,
1813                 "obb", packageName.c_str());
1814         calculate_tree_size(obbCodePath, &extStats.codeSize);
1815     }
1816     ATRACE_END();
1817     // Calculating the app size of the external storage owning app in a manual way, since
1818     // calculating it through quota apis also includes external media storage in the app storage
1819     // numbers
1820     if (flags & FLAG_USE_QUOTA && appId >= AID_APP_START && !ownsExternalStorage(appId)) {
1821         ATRACE_BEGIN("code");
1822         for (const auto& codePath : codePaths) {
1823             calculate_tree_size(codePath, &stats.codeSize, -1,
1824                     multiuser_get_shared_gid(0, appId));
1825         }
1826         ATRACE_END();
1827 
1828         ATRACE_BEGIN("quota");
1829         collectQuotaStats(uuidString, userId, appId, &stats, &extStats);
1830         ATRACE_END();
1831     } else {
1832         ATRACE_BEGIN("code");
1833         for (const auto& codePath : codePaths) {
1834             calculate_tree_size(codePath, &stats.codeSize);
1835         }
1836         ATRACE_END();
1837 
1838         for (size_t i = 0; i < packageNames.size(); i++) {
1839             const char* pkgname = packageNames[i].c_str();
1840 
1841             ATRACE_BEGIN("data");
1842             auto cePath = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInodes[i]);
1843             collectManualStats(cePath, &stats);
1844             auto dePath = create_data_user_de_package_path(uuid_, userId, pkgname);
1845             collectManualStats(dePath, &stats);
1846             ATRACE_END();
1847 
1848             if (!uuid) {
1849                 ATRACE_BEGIN("profiles");
1850                 calculate_tree_size(
1851                         create_primary_current_profile_package_dir_path(userId, pkgname),
1852                         &stats.dataSize);
1853                 calculate_tree_size(
1854                         create_primary_reference_profile_package_dir_path(pkgname),
1855                         &stats.codeSize);
1856                 ATRACE_END();
1857             }
1858 
1859             ATRACE_BEGIN("external");
1860             auto extPath = create_data_media_package_path(uuid_, userId, "data", pkgname);
1861             collectManualStats(extPath, &extStats);
1862             auto mediaPath = create_data_media_package_path(uuid_, userId, "media", pkgname);
1863             calculate_tree_size(mediaPath, &extStats.dataSize);
1864             ATRACE_END();
1865         }
1866 
1867         if (!uuid) {
1868             ATRACE_BEGIN("dalvik");
1869             int32_t sharedGid = multiuser_get_shared_gid(0, appId);
1870             if (sharedGid != -1) {
1871                 calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize,
1872                         sharedGid, -1);
1873             }
1874             ATRACE_END();
1875         }
1876     }
1877 
1878     std::vector<int64_t> ret;
1879     ret.push_back(stats.codeSize);
1880     ret.push_back(stats.dataSize);
1881     ret.push_back(stats.cacheSize);
1882     ret.push_back(extStats.codeSize);
1883     ret.push_back(extStats.dataSize);
1884     ret.push_back(extStats.cacheSize);
1885 #if MEASURE_DEBUG
1886     LOG(DEBUG) << "Final result " << toString(ret);
1887 #endif
1888     *_aidl_return = ret;
1889     return ok();
1890 }
1891 
1892 struct external_sizes {
1893     int64_t audioSize;
1894     int64_t videoSize;
1895     int64_t imageSize;
1896     int64_t totalSize; // excludes OBBs (Android/obb), but includes app data + cache
1897     int64_t obbSize;
1898 };
1899 
1900 #define PER_USER_RANGE 100000
1901 
getProjectIdForUser(int userId,long projectId)1902 static long getProjectIdForUser(int userId, long projectId) {
1903     return userId * PER_USER_RANGE + projectId;
1904 }
1905 
getExternalSizesForUserWithQuota(const std::string & uuid,int32_t userId,const std::vector<int32_t> & appIds)1906 static external_sizes getExternalSizesForUserWithQuota(const std::string& uuid, int32_t userId, const std::vector<int32_t>& appIds) {
1907     struct external_sizes sizes = {};
1908     int64_t space;
1909 
1910     if (supports_sdcardfs()) {
1911         uid_t uid = multiuser_get_uid(userId, AID_MEDIA_RW);
1912         if ((space = GetOccupiedSpaceForUid(uuid, uid)) != -1) {
1913             sizes.totalSize = space;
1914         }
1915 
1916         gid_t audioGid = multiuser_get_uid(userId, AID_MEDIA_AUDIO);
1917         if ((space = GetOccupiedSpaceForGid(uuid, audioGid)) != -1) {
1918             sizes.audioSize = space;
1919         }
1920 
1921         gid_t videoGid = multiuser_get_uid(userId, AID_MEDIA_VIDEO);
1922         if ((space = GetOccupiedSpaceForGid(uuid, videoGid)) != -1) {
1923             sizes.videoSize = space;
1924         }
1925 
1926         gid_t imageGid = multiuser_get_uid(userId, AID_MEDIA_IMAGE);
1927         if ((space = GetOccupiedSpaceForGid(uuid, imageGid)) != -1) {
1928             sizes.imageSize = space;
1929         }
1930 
1931         if ((space = GetOccupiedSpaceForGid(uuid, AID_MEDIA_OBB)) != -1) {
1932             sizes.obbSize = space;
1933         }
1934     } else {
1935         int64_t totalSize = 0;
1936         long defaultProjectId = getProjectIdForUser(userId, PROJECT_ID_EXT_DEFAULT);
1937         if ((space = GetOccupiedSpaceForProjectId(uuid, defaultProjectId)) != -1) {
1938             // This is all files that are not audio/video/images, excluding
1939             // OBBs and app-private data
1940             totalSize += space;
1941         }
1942 
1943         long audioProjectId = getProjectIdForUser(userId, PROJECT_ID_EXT_MEDIA_AUDIO);
1944         if ((space = GetOccupiedSpaceForProjectId(uuid, audioProjectId)) != -1) {
1945             sizes.audioSize = space;
1946             totalSize += space;
1947         }
1948 
1949         long videoProjectId = getProjectIdForUser(userId, PROJECT_ID_EXT_MEDIA_VIDEO);
1950         if ((space = GetOccupiedSpaceForProjectId(uuid, videoProjectId)) != -1) {
1951             sizes.videoSize = space;
1952             totalSize += space;
1953         }
1954 
1955         long imageProjectId = getProjectIdForUser(userId, PROJECT_ID_EXT_MEDIA_IMAGE);
1956         if ((space = GetOccupiedSpaceForProjectId(uuid, imageProjectId)) != -1) {
1957             sizes.imageSize = space;
1958             totalSize += space;
1959         }
1960 
1961         int64_t totalAppDataSize = 0;
1962         int64_t totalAppCacheSize = 0;
1963         int64_t totalAppObbSize = 0;
1964         for (auto appId : appIds) {
1965             if (appId >= AID_APP_START) {
1966                 // App data
1967                 uid_t uid = multiuser_get_uid(userId, appId);
1968                 long projectId = uid - AID_APP_START + PROJECT_ID_EXT_DATA_START;
1969                 totalAppDataSize += GetOccupiedSpaceForProjectId(uuid, projectId);
1970 
1971                 // App cache
1972                 long cacheProjectId = uid - AID_APP_START + PROJECT_ID_EXT_CACHE_START;
1973                 totalAppCacheSize += GetOccupiedSpaceForProjectId(uuid, cacheProjectId);
1974 
1975                 // App OBBs
1976                 long obbProjectId = uid - AID_APP_START + PROJECT_ID_EXT_OBB_START;
1977                 totalAppObbSize += GetOccupiedSpaceForProjectId(uuid, obbProjectId);
1978             }
1979         }
1980         // Total size should include app data + cache
1981         totalSize += totalAppDataSize;
1982         totalSize += totalAppCacheSize;
1983         sizes.totalSize = totalSize;
1984 
1985         // Only OBB is separate
1986         sizes.obbSize = totalAppObbSize;
1987     }
1988 
1989     return sizes;
1990 }
1991 
getUserSize(const std::optional<std::string> & uuid,int32_t userId,int32_t flags,const std::vector<int32_t> & appIds,std::vector<int64_t> * _aidl_return)1992 binder::Status InstalldNativeService::getUserSize(const std::optional<std::string>& uuid,
1993         int32_t userId, int32_t flags, const std::vector<int32_t>& appIds,
1994         std::vector<int64_t>* _aidl_return) {
1995     ENFORCE_UID(AID_SYSTEM);
1996     CHECK_ARGUMENT_UUID(uuid);
1997     // NOTE: Locking is relaxed on this method, since it's limited to
1998     // read-only measurements without mutation.
1999 
2000     // When modifying this logic, always verify using tests:
2001     // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetUserSize
2002 
2003 #if MEASURE_DEBUG
2004     LOG(INFO) << "Measuring user " << userId;
2005 #endif
2006 
2007     struct stats stats;
2008     struct stats extStats;
2009     memset(&stats, 0, sizeof(stats));
2010     memset(&extStats, 0, sizeof(extStats));
2011 
2012     auto uuidString = uuid.value_or("");
2013     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2014 
2015     if (!IsQuotaSupported(uuidString)) {
2016         flags &= ~FLAG_USE_QUOTA;
2017     }
2018 
2019     if (flags & FLAG_USE_QUOTA) {
2020         ATRACE_BEGIN("code");
2021         calculate_tree_size(create_data_app_path(uuid_), &stats.codeSize, -1, -1, true);
2022         ATRACE_END();
2023 
2024         ATRACE_BEGIN("data");
2025         auto cePath = create_data_user_ce_path(uuid_, userId);
2026         collectManualStatsForUser(cePath, &stats, true);
2027         auto dePath = create_data_user_de_path(uuid_, userId);
2028         collectManualStatsForUser(dePath, &stats, true);
2029         ATRACE_END();
2030 
2031         if (!uuid) {
2032             ATRACE_BEGIN("profile");
2033             auto userProfilePath = create_primary_cur_profile_dir_path(userId);
2034             calculate_tree_size(userProfilePath, &stats.dataSize, -1, -1, true);
2035             auto refProfilePath = create_primary_ref_profile_dir_path();
2036             calculate_tree_size(refProfilePath, &stats.codeSize, -1, -1, true);
2037             ATRACE_END();
2038         }
2039 
2040         ATRACE_BEGIN("external");
2041         auto sizes = getExternalSizesForUserWithQuota(uuidString, userId, appIds);
2042         extStats.dataSize += sizes.totalSize;
2043         extStats.codeSize += sizes.obbSize;
2044         ATRACE_END();
2045 
2046         if (!uuid) {
2047             ATRACE_BEGIN("dalvik");
2048             calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize,
2049                     -1, -1, true);
2050             calculate_tree_size(create_primary_cur_profile_dir_path(userId), &stats.dataSize,
2051                     -1, -1, true);
2052             ATRACE_END();
2053         }
2054         ATRACE_BEGIN("quota");
2055         int64_t dataSize = extStats.dataSize;
2056         for (auto appId : appIds) {
2057             if (appId >= AID_APP_START) {
2058                 collectQuotaStats(uuidString, userId, appId, &stats, &extStats);
2059 #if MEASURE_DEBUG
2060                 // Sleep to make sure we don't lose logs
2061                 usleep(1);
2062 #endif
2063             }
2064         }
2065         extStats.dataSize = dataSize;
2066         ATRACE_END();
2067     } else {
2068         ATRACE_BEGIN("obb");
2069         auto obbPath = create_data_path(uuid_) + "/media/obb";
2070         calculate_tree_size(obbPath, &extStats.codeSize);
2071         ATRACE_END();
2072 
2073         ATRACE_BEGIN("code");
2074         calculate_tree_size(create_data_app_path(uuid_), &stats.codeSize);
2075         ATRACE_END();
2076 
2077         ATRACE_BEGIN("data");
2078         auto cePath = create_data_user_ce_path(uuid_, userId);
2079         collectManualStatsForUser(cePath, &stats);
2080         auto dePath = create_data_user_de_path(uuid_, userId);
2081         collectManualStatsForUser(dePath, &stats);
2082         ATRACE_END();
2083 
2084         if (!uuid) {
2085             ATRACE_BEGIN("profile");
2086             auto userProfilePath = create_primary_cur_profile_dir_path(userId);
2087             calculate_tree_size(userProfilePath, &stats.dataSize);
2088             auto refProfilePath = create_primary_ref_profile_dir_path();
2089             calculate_tree_size(refProfilePath, &stats.codeSize);
2090             ATRACE_END();
2091         }
2092 
2093         ATRACE_BEGIN("external");
2094         auto dataMediaPath = create_data_media_path(uuid_, userId);
2095         collectManualExternalStatsForUser(dataMediaPath, &extStats);
2096 #if MEASURE_DEBUG
2097         LOG(DEBUG) << "Measured external data " << extStats.dataSize << " cache "
2098                 << extStats.cacheSize;
2099 #endif
2100         ATRACE_END();
2101 
2102         if (!uuid) {
2103             ATRACE_BEGIN("dalvik");
2104             calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize);
2105             calculate_tree_size(create_primary_cur_profile_dir_path(userId), &stats.dataSize);
2106             ATRACE_END();
2107         }
2108     }
2109 
2110     std::vector<int64_t> ret;
2111     ret.push_back(stats.codeSize);
2112     ret.push_back(stats.dataSize);
2113     ret.push_back(stats.cacheSize);
2114     ret.push_back(extStats.codeSize);
2115     ret.push_back(extStats.dataSize);
2116     ret.push_back(extStats.cacheSize);
2117 #if MEASURE_DEBUG
2118     LOG(DEBUG) << "Final result " << toString(ret);
2119 #endif
2120     *_aidl_return = ret;
2121     return ok();
2122 }
2123 
getExternalSize(const std::optional<std::string> & uuid,int32_t userId,int32_t flags,const std::vector<int32_t> & appIds,std::vector<int64_t> * _aidl_return)2124 binder::Status InstalldNativeService::getExternalSize(const std::optional<std::string>& uuid,
2125         int32_t userId, int32_t flags, const std::vector<int32_t>& appIds,
2126         std::vector<int64_t>* _aidl_return) {
2127     ENFORCE_UID(AID_SYSTEM);
2128     CHECK_ARGUMENT_UUID(uuid);
2129     // NOTE: Locking is relaxed on this method, since it's limited to
2130     // read-only measurements without mutation.
2131 
2132     // When modifying this logic, always verify using tests:
2133     // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetExternalSize
2134 
2135 #if MEASURE_DEBUG
2136     LOG(INFO) << "Measuring external " << userId;
2137 #endif
2138 
2139     auto uuidString = uuid.value_or("");
2140     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2141 
2142     int64_t totalSize = 0;
2143     int64_t audioSize = 0;
2144     int64_t videoSize = 0;
2145     int64_t imageSize = 0;
2146     int64_t appSize = 0;
2147     int64_t obbSize = 0;
2148 
2149     if (!IsQuotaSupported(uuidString)) {
2150         flags &= ~FLAG_USE_QUOTA;
2151     }
2152 
2153     if (flags & FLAG_USE_QUOTA) {
2154         ATRACE_BEGIN("quota");
2155         auto sizes = getExternalSizesForUserWithQuota(uuidString, userId, appIds);
2156         totalSize = sizes.totalSize;
2157         audioSize = sizes.audioSize;
2158         videoSize = sizes.videoSize;
2159         imageSize = sizes.imageSize;
2160         obbSize = sizes.obbSize;
2161         ATRACE_END();
2162 
2163         ATRACE_BEGIN("apps");
2164         struct stats extStats;
2165         memset(&extStats, 0, sizeof(extStats));
2166         for (auto appId : appIds) {
2167             if (appId >= AID_APP_START) {
2168                 collectQuotaStats(uuidString, userId, appId, nullptr, &extStats);
2169             }
2170         }
2171         appSize = extStats.dataSize;
2172         ATRACE_END();
2173     } else {
2174         ATRACE_BEGIN("manual");
2175         FTS *fts;
2176         FTSENT *p;
2177         auto path = create_data_media_path(uuid_, userId);
2178         char *argv[] = { (char*) path.c_str(), nullptr };
2179         if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, nullptr))) {
2180             return error("Failed to fts_open " + path);
2181         }
2182         while ((p = fts_read(fts)) != nullptr) {
2183             char* ext;
2184             int64_t size = (p->fts_statp->st_blocks * 512);
2185             switch (p->fts_info) {
2186             case FTS_F:
2187                 // Only categorize files not belonging to apps
2188                 if (p->fts_parent->fts_number == 0) {
2189                     ext = strrchr(p->fts_name, '.');
2190                     if (ext != nullptr) {
2191                         switch (MatchExtension(++ext)) {
2192                         case AID_MEDIA_AUDIO: audioSize += size; break;
2193                         case AID_MEDIA_VIDEO: videoSize += size; break;
2194                         case AID_MEDIA_IMAGE: imageSize += size; break;
2195                         }
2196                     }
2197                 }
2198                 [[fallthrough]]; // always count against total
2199             case FTS_D:
2200                 // Ignore data belonging to specific apps
2201                 p->fts_number = p->fts_parent->fts_number;
2202                 if (p->fts_level == 1 && !strcmp(p->fts_name, "Android")) {
2203                     p->fts_number = 1;
2204                 }
2205                 [[fallthrough]]; // always count against total
2206             case FTS_DEFAULT:
2207             case FTS_SL:
2208             case FTS_SLNONE:
2209                 if (p->fts_parent->fts_number == 1) {
2210                     appSize += size;
2211                 }
2212                 totalSize += size;
2213                 break;
2214             }
2215         }
2216         fts_close(fts);
2217         ATRACE_END();
2218 
2219         ATRACE_BEGIN("obb");
2220         auto obbPath = StringPrintf("%s/Android/obb",
2221                 create_data_media_path(uuid_, userId).c_str());
2222         calculate_tree_size(obbPath, &obbSize);
2223         ATRACE_END();
2224     }
2225 
2226     std::vector<int64_t> ret;
2227     ret.push_back(totalSize);
2228     ret.push_back(audioSize);
2229     ret.push_back(videoSize);
2230     ret.push_back(imageSize);
2231     ret.push_back(appSize);
2232     ret.push_back(obbSize);
2233 #if MEASURE_DEBUG
2234     LOG(DEBUG) << "Final result " << toString(ret);
2235 #endif
2236     *_aidl_return = ret;
2237     return ok();
2238 }
2239 
getAppCrates(const std::optional<std::string> & uuid,const std::vector<std::string> & packageNames,int32_t userId,std::optional<std::vector<std::optional<CrateMetadata>>> * _aidl_return)2240 binder::Status InstalldNativeService::getAppCrates(
2241         const std::optional<std::string>& uuid,
2242         const std::vector<std::string>& packageNames, int32_t userId,
2243         std::optional<std::vector<std::optional<CrateMetadata>>>* _aidl_return) {
2244     ENFORCE_UID(AID_SYSTEM);
2245     CHECK_ARGUMENT_UUID(uuid);
2246     for (const auto& packageName : packageNames) {
2247         CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2248     }
2249 #ifdef ENABLE_STORAGE_CRATES
2250     std::lock_guard<std::recursive_mutex> lock(mLock);
2251 
2252     auto retVector = std::vector<std::optional<CrateMetadata>>();
2253     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2254 
2255     std::function<void(CratedFolder, CrateMetadata&&)> onCreateCrate =
2256             [&](CratedFolder cratedFolder, CrateMetadata&& crateMetadata) -> void {
2257         if (cratedFolder == nullptr) {
2258             return;
2259         }
2260         retVector.push_back(std::move(crateMetadata));
2261     };
2262 
2263     for (const auto& packageName : packageNames) {
2264 #if CRATE_DEBUG
2265         LOG(DEBUG) << "packageName = " << packageName;
2266 #endif
2267         auto crateManager = std::make_unique<CrateManager>(uuid_, userId, packageName);
2268         crateManager->traverseAllCrates(onCreateCrate);
2269     }
2270 
2271 #if CRATE_DEBUG
2272     LOG(WARNING) << "retVector.size() =" << retVector.size();
2273     for (auto& item : retVector) {
2274         CrateManager::dump(*item);
2275     }
2276 #endif
2277 
2278     *_aidl_return = std::move(retVector);
2279 #else // ENABLE_STORAGE_CRATES
2280     _aidl_return->reset();
2281 
2282     /* prevent compile warning fail */
2283     if (userId < 0) {
2284         return error();
2285     }
2286 #endif // ENABLE_STORAGE_CRATES
2287     return ok();
2288 }
2289 
getUserCrates(const std::optional<std::string> & uuid,int32_t userId,std::optional<std::vector<std::optional<CrateMetadata>>> * _aidl_return)2290 binder::Status InstalldNativeService::getUserCrates(
2291         const std::optional<std::string>& uuid, int32_t userId,
2292         std::optional<std::vector<std::optional<CrateMetadata>>>* _aidl_return) {
2293     ENFORCE_UID(AID_SYSTEM);
2294     CHECK_ARGUMENT_UUID(uuid);
2295 #ifdef ENABLE_STORAGE_CRATES
2296     std::lock_guard<std::recursive_mutex> lock(mLock);
2297 
2298     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2299     auto retVector = std::vector<std::optional<CrateMetadata>>();
2300 
2301     std::function<void(CratedFolder, CrateMetadata&&)> onCreateCrate =
2302             [&](CratedFolder cratedFolder, CrateMetadata&& crateMetadata) -> void {
2303         if (cratedFolder == nullptr) {
2304             return;
2305         }
2306         retVector.push_back(std::move(crateMetadata));
2307     };
2308 
2309     std::function<void(FTSENT*)> onHandingPackage = [&](FTSENT* packageDir) -> void {
2310         auto crateManager = std::make_unique<CrateManager>(uuid_, userId, packageDir->fts_name);
2311         crateManager->traverseAllCrates(onCreateCrate);
2312     };
2313     CrateManager::traverseAllPackagesForUser(uuid, userId, onHandingPackage);
2314 
2315 #if CRATE_DEBUG
2316     LOG(DEBUG) << "retVector.size() =" << retVector.size();
2317     for (auto& item : retVector) {
2318         CrateManager::dump(*item);
2319     }
2320 #endif
2321 
2322     *_aidl_return = std::move(retVector);
2323 #else // ENABLE_STORAGE_CRATES
2324     _aidl_return->reset();
2325 
2326     /* prevent compile warning fail */
2327     if (userId < 0) {
2328         return error();
2329     }
2330 #endif // ENABLE_STORAGE_CRATES
2331     return ok();
2332 }
2333 
setAppQuota(const std::optional<std::string> & uuid,int32_t userId,int32_t appId,int64_t cacheQuota)2334 binder::Status InstalldNativeService::setAppQuota(const std::optional<std::string>& uuid,
2335         int32_t userId, int32_t appId, int64_t cacheQuota) {
2336     ENFORCE_UID(AID_SYSTEM);
2337     CHECK_ARGUMENT_UUID(uuid);
2338     std::lock_guard<std::recursive_mutex> lock(mQuotasLock);
2339 
2340     int32_t uid = multiuser_get_uid(userId, appId);
2341     mCacheQuotas[uid] = cacheQuota;
2342 
2343     return ok();
2344 }
2345 
2346 // Dumps the contents of a profile file, using pkgname's dex files for pretty
2347 // printing the result.
dumpProfiles(int32_t uid,const std::string & packageName,const std::string & profileName,const std::string & codePath,bool * _aidl_return)2348 binder::Status InstalldNativeService::dumpProfiles(int32_t uid, const std::string& packageName,
2349         const std::string& profileName, const std::string& codePath, bool* _aidl_return) {
2350     ENFORCE_UID(AID_SYSTEM);
2351     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2352     CHECK_ARGUMENT_PATH(codePath);
2353     std::lock_guard<std::recursive_mutex> lock(mLock);
2354 
2355     *_aidl_return = dump_profiles(uid, packageName, profileName, codePath);
2356     return ok();
2357 }
2358 
2359 // Copy the contents of a system profile over the data profile.
copySystemProfile(const std::string & systemProfile,int32_t packageUid,const std::string & packageName,const std::string & profileName,bool * _aidl_return)2360 binder::Status InstalldNativeService::copySystemProfile(const std::string& systemProfile,
2361         int32_t packageUid, const std::string& packageName, const std::string& profileName,
2362         bool* _aidl_return) {
2363     ENFORCE_UID(AID_SYSTEM);
2364     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2365     std::lock_guard<std::recursive_mutex> lock(mLock);
2366     *_aidl_return = copy_system_profile(systemProfile, packageUid, packageName, profileName);
2367     return ok();
2368 }
2369 
2370 // TODO: Consider returning error codes.
mergeProfiles(int32_t uid,const std::string & packageName,const std::string & profileName,int * _aidl_return)2371 binder::Status InstalldNativeService::mergeProfiles(int32_t uid, const std::string& packageName,
2372         const std::string& profileName, int* _aidl_return) {
2373     ENFORCE_UID(AID_SYSTEM);
2374     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2375     std::lock_guard<std::recursive_mutex> lock(mLock);
2376 
2377     *_aidl_return = analyze_primary_profiles(uid, packageName, profileName);
2378     return ok();
2379 }
2380 
createProfileSnapshot(int32_t appId,const std::string & packageName,const std::string & profileName,const std::string & classpath,bool * _aidl_return)2381 binder::Status InstalldNativeService::createProfileSnapshot(int32_t appId,
2382         const std::string& packageName, const std::string& profileName,
2383         const std::string& classpath, bool* _aidl_return) {
2384     ENFORCE_UID(AID_SYSTEM);
2385     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2386     std::lock_guard<std::recursive_mutex> lock(mLock);
2387 
2388     *_aidl_return = create_profile_snapshot(appId, packageName, profileName, classpath);
2389     return ok();
2390 }
2391 
destroyProfileSnapshot(const std::string & packageName,const std::string & profileName)2392 binder::Status InstalldNativeService::destroyProfileSnapshot(const std::string& packageName,
2393         const std::string& profileName) {
2394     ENFORCE_UID(AID_SYSTEM);
2395     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2396     std::lock_guard<std::recursive_mutex> lock(mLock);
2397 
2398     std::string snapshot = create_snapshot_profile_path(packageName, profileName);
2399     if ((unlink(snapshot.c_str()) != 0) && (errno != ENOENT)) {
2400         return error("Failed to destroy profile snapshot for " + packageName + ":" + profileName);
2401     }
2402     return ok();
2403 }
2404 
getCStr(const std::optional<std::string> & data,const char * default_value=nullptr)2405 static const char* getCStr(const std::optional<std::string>& data,
2406         const char* default_value = nullptr) {
2407     return data ? data->c_str() : default_value;
2408 }
dexopt(const std::string & apkPath,int32_t uid,const std::optional<std::string> & packageName,const std::string & instructionSet,int32_t dexoptNeeded,const std::optional<std::string> & outputPath,int32_t dexFlags,const std::string & compilerFilter,const std::optional<std::string> & uuid,const std::optional<std::string> & classLoaderContext,const std::optional<std::string> & seInfo,bool downgrade,int32_t targetSdkVersion,const std::optional<std::string> & profileName,const std::optional<std::string> & dexMetadataPath,const std::optional<std::string> & compilationReason)2409 binder::Status InstalldNativeService::dexopt(const std::string& apkPath, int32_t uid,
2410         const std::optional<std::string>& packageName, const std::string& instructionSet,
2411         int32_t dexoptNeeded, const std::optional<std::string>& outputPath, int32_t dexFlags,
2412         const std::string& compilerFilter, const std::optional<std::string>& uuid,
2413         const std::optional<std::string>& classLoaderContext,
2414         const std::optional<std::string>& seInfo, bool downgrade, int32_t targetSdkVersion,
2415         const std::optional<std::string>& profileName,
2416         const std::optional<std::string>& dexMetadataPath,
2417         const std::optional<std::string>& compilationReason) {
2418     ENFORCE_UID(AID_SYSTEM);
2419     CHECK_ARGUMENT_UUID(uuid);
2420     CHECK_ARGUMENT_PATH(apkPath);
2421     if (packageName && *packageName != "*") {
2422         CHECK_ARGUMENT_PACKAGE_NAME(*packageName);
2423     }
2424     CHECK_ARGUMENT_PATH(outputPath);
2425     CHECK_ARGUMENT_PATH(dexMetadataPath);
2426     std::lock_guard<std::recursive_mutex> lock(mLock);
2427 
2428     const char* oat_dir = getCStr(outputPath);
2429     const char* instruction_set = instructionSet.c_str();
2430     if (oat_dir != nullptr && !createOatDir(oat_dir, instruction_set).isOk()) {
2431         // Can't create oat dir - let dexopt use cache dir.
2432         oat_dir = nullptr;
2433     }
2434 
2435     const char* apk_path = apkPath.c_str();
2436     const char* pkgname = getCStr(packageName, "*");
2437     const char* compiler_filter = compilerFilter.c_str();
2438     const char* volume_uuid = getCStr(uuid);
2439     const char* class_loader_context = getCStr(classLoaderContext);
2440     const char* se_info = getCStr(seInfo);
2441     const char* profile_name = getCStr(profileName);
2442     const char* dm_path = getCStr(dexMetadataPath);
2443     const char* compilation_reason = getCStr(compilationReason);
2444     std::string error_msg;
2445     int res = android::installd::dexopt(apk_path, uid, pkgname, instruction_set, dexoptNeeded,
2446             oat_dir, dexFlags, compiler_filter, volume_uuid, class_loader_context, se_info,
2447             downgrade, targetSdkVersion, profile_name, dm_path, compilation_reason, &error_msg);
2448     return res ? error(res, error_msg) : ok();
2449 }
2450 
compileLayouts(const std::string & apkPath,const std::string & packageName,const std::string & outDexFile,int uid,bool * _aidl_return)2451 binder::Status InstalldNativeService::compileLayouts(const std::string& apkPath,
2452                                                      const std::string& packageName,
2453                                                      const std ::string& outDexFile, int uid,
2454                                                      bool* _aidl_return) {
2455     const char* apk_path = apkPath.c_str();
2456     const char* package_name = packageName.c_str();
2457     const char* out_dex_file = outDexFile.c_str();
2458     *_aidl_return = android::installd::view_compiler(apk_path, package_name, out_dex_file, uid);
2459     return *_aidl_return ? ok() : error("viewcompiler failed");
2460 }
2461 
linkNativeLibraryDirectory(const std::optional<std::string> & uuid,const std::string & packageName,const std::string & nativeLibPath32,int32_t userId)2462 binder::Status InstalldNativeService::linkNativeLibraryDirectory(
2463         const std::optional<std::string>& uuid, const std::string& packageName,
2464         const std::string& nativeLibPath32, int32_t userId) {
2465     ENFORCE_UID(AID_SYSTEM);
2466     CHECK_ARGUMENT_UUID(uuid);
2467     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2468     CHECK_ARGUMENT_PATH(nativeLibPath32);
2469     std::lock_guard<std::recursive_mutex> lock(mLock);
2470 
2471     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2472     const char* pkgname = packageName.c_str();
2473     const char* asecLibDir = nativeLibPath32.c_str();
2474     struct stat s, libStat;
2475     binder::Status res = ok();
2476 
2477     auto _pkgdir = create_data_user_ce_package_path(uuid_, userId, pkgname);
2478     auto _libsymlink = _pkgdir + PKG_LIB_POSTFIX;
2479 
2480     const char* pkgdir = _pkgdir.c_str();
2481     const char* libsymlink = _libsymlink.c_str();
2482 
2483     if (stat(pkgdir, &s) < 0) {
2484         return error("Failed to stat " + _pkgdir);
2485     }
2486 
2487     char *con = nullptr;
2488     if (lgetfilecon(pkgdir, &con) < 0) {
2489         return error("Failed to lgetfilecon " + _pkgdir);
2490     }
2491 
2492     if (chown(pkgdir, AID_INSTALL, AID_INSTALL) < 0) {
2493         res = error("Failed to chown " + _pkgdir);
2494         goto out;
2495     }
2496 
2497     if (chmod(pkgdir, 0700) < 0) {
2498         res = error("Failed to chmod " + _pkgdir);
2499         goto out;
2500     }
2501 
2502     if (lstat(libsymlink, &libStat) < 0) {
2503         if (errno != ENOENT) {
2504             res = error("Failed to stat " + _libsymlink);
2505             goto out;
2506         }
2507     } else {
2508         if (S_ISDIR(libStat.st_mode)) {
2509             if (delete_dir_contents(libsymlink, 1, nullptr) < 0) {
2510                 res = error("Failed to delete " + _libsymlink);
2511                 goto out;
2512             }
2513         } else if (S_ISLNK(libStat.st_mode)) {
2514             if (unlink(libsymlink) < 0) {
2515                 res = error("Failed to unlink " + _libsymlink);
2516                 goto out;
2517             }
2518         }
2519     }
2520 
2521     if (symlink(asecLibDir, libsymlink) < 0) {
2522         res = error("Failed to symlink " + _libsymlink + " to " + nativeLibPath32);
2523         goto out;
2524     }
2525 
2526     if (lsetfilecon(libsymlink, con) < 0) {
2527         res = error("Failed to lsetfilecon " + _libsymlink);
2528         goto out;
2529     }
2530 
2531 out:
2532     free(con);
2533     if (chmod(pkgdir, s.st_mode) < 0) {
2534         auto msg = "Failed to cleanup chmod " + _pkgdir;
2535         if (res.isOk()) {
2536             res = error(msg);
2537         } else {
2538             PLOG(ERROR) << msg;
2539         }
2540     }
2541 
2542     if (chown(pkgdir, s.st_uid, s.st_gid) < 0) {
2543         auto msg = "Failed to cleanup chown " + _pkgdir;
2544         if (res.isOk()) {
2545             res = error(msg);
2546         } else {
2547             PLOG(ERROR) << msg;
2548         }
2549     }
2550 
2551     return res;
2552 }
2553 
restoreconAppData(const std::optional<std::string> & uuid,const std::string & packageName,int32_t userId,int32_t flags,int32_t appId,const std::string & seInfo)2554 binder::Status InstalldNativeService::restoreconAppData(const std::optional<std::string>& uuid,
2555         const std::string& packageName, int32_t userId, int32_t flags, int32_t appId,
2556         const std::string& seInfo) {
2557     ENFORCE_UID(AID_SYSTEM);
2558     CHECK_ARGUMENT_UUID(uuid);
2559     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2560     std::lock_guard<std::recursive_mutex> lock(mLock);
2561 
2562     binder::Status res = ok();
2563 
2564     // SELINUX_ANDROID_RESTORECON_DATADATA flag is set by libselinux. Not needed here.
2565     unsigned int seflags = SELINUX_ANDROID_RESTORECON_RECURSE;
2566     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2567     const char* pkgName = packageName.c_str();
2568     const char* seinfo = seInfo.c_str();
2569 
2570     uid_t uid = multiuser_get_uid(userId, appId);
2571     if (flags & FLAG_STORAGE_CE) {
2572         auto path = create_data_user_ce_package_path(uuid_, userId, pkgName);
2573         if (selinux_android_restorecon_pkgdir(path.c_str(), seinfo, uid, seflags) < 0) {
2574             res = error("restorecon failed for " + path);
2575         }
2576     }
2577     if (flags & FLAG_STORAGE_DE) {
2578         auto path = create_data_user_de_package_path(uuid_, userId, pkgName);
2579         if (selinux_android_restorecon_pkgdir(path.c_str(), seinfo, uid, seflags) < 0) {
2580             res = error("restorecon failed for " + path);
2581         }
2582     }
2583     return res;
2584 }
2585 
createOatDir(const std::string & oatDir,const std::string & instructionSet)2586 binder::Status InstalldNativeService::createOatDir(const std::string& oatDir,
2587         const std::string& instructionSet) {
2588     ENFORCE_UID(AID_SYSTEM);
2589     CHECK_ARGUMENT_PATH(oatDir);
2590     std::lock_guard<std::recursive_mutex> lock(mLock);
2591 
2592     const char* oat_dir = oatDir.c_str();
2593     const char* instruction_set = instructionSet.c_str();
2594     char oat_instr_dir[PKG_PATH_MAX];
2595 
2596     if (validate_apk_path(oat_dir)) {
2597         return error("Invalid path " + oatDir);
2598     }
2599     if (fs_prepare_dir(oat_dir, S_IRWXU | S_IRWXG | S_IXOTH, AID_SYSTEM, AID_INSTALL)) {
2600         return error("Failed to prepare " + oatDir);
2601     }
2602     if (selinux_android_restorecon(oat_dir, 0)) {
2603         return error("Failed to restorecon " + oatDir);
2604     }
2605     snprintf(oat_instr_dir, PKG_PATH_MAX, "%s/%s", oat_dir, instruction_set);
2606     if (fs_prepare_dir(oat_instr_dir, S_IRWXU | S_IRWXG | S_IXOTH, AID_SYSTEM, AID_INSTALL)) {
2607         return error(StringPrintf("Failed to prepare %s", oat_instr_dir));
2608     }
2609     return ok();
2610 }
2611 
rmPackageDir(const std::string & packageDir)2612 binder::Status InstalldNativeService::rmPackageDir(const std::string& packageDir) {
2613     ENFORCE_UID(AID_SYSTEM);
2614     CHECK_ARGUMENT_PATH(packageDir);
2615     std::lock_guard<std::recursive_mutex> lock(mLock);
2616 
2617     if (validate_apk_path(packageDir.c_str())) {
2618         return error("Invalid path " + packageDir);
2619     }
2620     if (rm_package_dir(packageDir) != 0) {
2621         return error("Failed to delete " + packageDir);
2622     }
2623     return ok();
2624 }
2625 
linkFile(const std::string & relativePath,const std::string & fromBase,const std::string & toBase)2626 binder::Status InstalldNativeService::linkFile(const std::string& relativePath,
2627         const std::string& fromBase, const std::string& toBase) {
2628     ENFORCE_UID(AID_SYSTEM);
2629     CHECK_ARGUMENT_PATH(fromBase);
2630     CHECK_ARGUMENT_PATH(toBase);
2631     std::lock_guard<std::recursive_mutex> lock(mLock);
2632 
2633     const char* relative_path = relativePath.c_str();
2634     const char* from_base = fromBase.c_str();
2635     const char* to_base = toBase.c_str();
2636     char from_path[PKG_PATH_MAX];
2637     char to_path[PKG_PATH_MAX];
2638     snprintf(from_path, PKG_PATH_MAX, "%s/%s", from_base, relative_path);
2639     snprintf(to_path, PKG_PATH_MAX, "%s/%s", to_base, relative_path);
2640 
2641     if (validate_apk_path_subdirs(from_path)) {
2642         return error(StringPrintf("Invalid from path %s", from_path));
2643     }
2644 
2645     if (validate_apk_path_subdirs(to_path)) {
2646         return error(StringPrintf("Invalid to path %s", to_path));
2647     }
2648 
2649     if (link(from_path, to_path) < 0) {
2650         return error(StringPrintf("Failed to link from %s to %s", from_path, to_path));
2651     }
2652 
2653     return ok();
2654 }
2655 
moveAb(const std::string & apkPath,const std::string & instructionSet,const std::string & outputPath)2656 binder::Status InstalldNativeService::moveAb(const std::string& apkPath,
2657         const std::string& instructionSet, const std::string& outputPath) {
2658     ENFORCE_UID(AID_SYSTEM);
2659     CHECK_ARGUMENT_PATH(apkPath);
2660     CHECK_ARGUMENT_PATH(outputPath);
2661     std::lock_guard<std::recursive_mutex> lock(mLock);
2662 
2663     const char* apk_path = apkPath.c_str();
2664     const char* instruction_set = instructionSet.c_str();
2665     const char* oat_dir = outputPath.c_str();
2666 
2667     bool success = move_ab(apk_path, instruction_set, oat_dir);
2668     return success ? ok() : error();
2669 }
2670 
deleteOdex(const std::string & apkPath,const std::string & instructionSet,const std::optional<std::string> & outputPath,int64_t * _aidl_return)2671 binder::Status InstalldNativeService::deleteOdex(const std::string& apkPath,
2672         const std::string& instructionSet, const std::optional<std::string>& outputPath,
2673         int64_t* _aidl_return) {
2674     ENFORCE_UID(AID_SYSTEM);
2675     CHECK_ARGUMENT_PATH(apkPath);
2676     CHECK_ARGUMENT_PATH(outputPath);
2677     std::lock_guard<std::recursive_mutex> lock(mLock);
2678 
2679     const char* apk_path = apkPath.c_str();
2680     const char* instruction_set = instructionSet.c_str();
2681     const char* oat_dir = outputPath ? outputPath->c_str() : nullptr;
2682 
2683     *_aidl_return = delete_odex(apk_path, instruction_set, oat_dir);
2684     return *_aidl_return == -1 ? error() : ok();
2685 }
2686 
2687 // This kernel feature is experimental.
2688 // TODO: remove local definition once upstreamed
2689 #ifndef FS_IOC_ENABLE_VERITY
2690 
2691 #define FS_IOC_ENABLE_VERITY           _IO('f', 133)
2692 #define FS_IOC_SET_VERITY_MEASUREMENT  _IOW('f', 134, struct fsverity_measurement)
2693 
2694 #define FS_VERITY_ALG_SHA256           1
2695 
2696 struct fsverity_measurement {
2697     __u16 digest_algorithm;
2698     __u16 digest_size;
2699     __u32 reserved1;
2700     __u64 reserved2[3];
2701     __u8 digest[];
2702 };
2703 
2704 #endif
2705 
installApkVerity(const std::string & filePath,android::base::unique_fd verityInputAshmem,int32_t contentSize)2706 binder::Status InstalldNativeService::installApkVerity(const std::string& filePath,
2707         android::base::unique_fd verityInputAshmem, int32_t contentSize) {
2708     ENFORCE_UID(AID_SYSTEM);
2709     CHECK_ARGUMENT_PATH(filePath);
2710     std::lock_guard<std::recursive_mutex> lock(mLock);
2711 
2712     if (!android::base::GetBoolProperty(kPropApkVerityMode, false)) {
2713         return ok();
2714     }
2715 #ifndef NDEBUG
2716     ASSERT_PAGE_SIZE_4K();
2717 #endif
2718     // TODO: also check fsverity support in the current file system if compiled with DEBUG.
2719     // TODO: change ashmem to some temporary file to support huge apk.
2720     if (!ashmem_valid(verityInputAshmem.get())) {
2721         return error("FD is not an ashmem");
2722     }
2723 
2724     // 1. Seek to the next page boundary beyond the end of the file.
2725     ::android::base::unique_fd wfd(open(filePath.c_str(), O_WRONLY));
2726     if (wfd.get() < 0) {
2727         return error("Failed to open " + filePath);
2728     }
2729     struct stat st;
2730     if (fstat(wfd.get(), &st) < 0) {
2731         return error("Failed to stat " + filePath);
2732     }
2733     // fsverity starts from the block boundary.
2734     off_t padding = kVerityPageSize - st.st_size % kVerityPageSize;
2735     if (padding == kVerityPageSize) {
2736         padding = 0;
2737     }
2738     if (lseek(wfd.get(), st.st_size + padding, SEEK_SET) < 0) {
2739         return error("Failed to lseek " + filePath);
2740     }
2741 
2742     // 2. Write everything in the ashmem to the file.  Note that allocated
2743     //    ashmem size is multiple of page size, which is different from the
2744     //    actual content size.
2745     int shmSize = ashmem_get_size_region(verityInputAshmem.get());
2746     if (shmSize < 0) {
2747         return error("Failed to get ashmem size: " + std::to_string(shmSize));
2748     }
2749     if (contentSize < 0) {
2750         return error("Invalid content size: " + std::to_string(contentSize));
2751     }
2752     if (contentSize > shmSize) {
2753         return error("Content size overflow: " + std::to_string(contentSize) + " > " +
2754                      std::to_string(shmSize));
2755     }
2756     auto data = std::unique_ptr<void, std::function<void (void *)>>(
2757         mmap(nullptr, contentSize, PROT_READ, MAP_SHARED, verityInputAshmem.get(), 0),
2758         [contentSize] (void* ptr) {
2759           if (ptr != MAP_FAILED) {
2760             munmap(ptr, contentSize);
2761           }
2762         });
2763 
2764     if (data.get() == MAP_FAILED) {
2765         return error("Failed to mmap the ashmem");
2766     }
2767     char* cursor = reinterpret_cast<char*>(data.get());
2768     int remaining = contentSize;
2769     while (remaining > 0) {
2770         int ret = TEMP_FAILURE_RETRY(write(wfd.get(), cursor, remaining));
2771         if (ret < 0) {
2772             return error("Failed to write to " + filePath + " (" + std::to_string(remaining) +
2773                          + "/" + std::to_string(contentSize) + ")");
2774         }
2775         cursor += ret;
2776         remaining -= ret;
2777     }
2778     wfd.reset();
2779 
2780     // 3. Enable fsverity (needs readonly fd. Once it's done, the file becomes immutable.
2781     ::android::base::unique_fd rfd(open(filePath.c_str(), O_RDONLY));
2782     if (ioctl(rfd.get(), FS_IOC_ENABLE_VERITY, nullptr) < 0) {
2783         return error("Failed to enable fsverity on " + filePath);
2784     }
2785     return ok();
2786 }
2787 
assertFsverityRootHashMatches(const std::string & filePath,const std::vector<uint8_t> & expectedHash)2788 binder::Status InstalldNativeService::assertFsverityRootHashMatches(const std::string& filePath,
2789         const std::vector<uint8_t>& expectedHash) {
2790     ENFORCE_UID(AID_SYSTEM);
2791     CHECK_ARGUMENT_PATH(filePath);
2792     std::lock_guard<std::recursive_mutex> lock(mLock);
2793 
2794     if (!android::base::GetBoolProperty(kPropApkVerityMode, false)) {
2795         return ok();
2796     }
2797     // TODO: also check fsverity support in the current file system if compiled with DEBUG.
2798     if (expectedHash.size() != kSha256Size) {
2799         return error("verity hash size should be " + std::to_string(kSha256Size) + " but is " +
2800                      std::to_string(expectedHash.size()));
2801     }
2802 
2803     ::android::base::unique_fd fd(open(filePath.c_str(), O_RDONLY));
2804     if (fd.get() < 0) {
2805         return error("Failed to open " + filePath + ": " + strerror(errno));
2806     }
2807 
2808     unsigned int buffer_size = sizeof(fsverity_measurement) + kSha256Size;
2809     std::vector<char> buffer(buffer_size, 0);
2810 
2811     fsverity_measurement* config = reinterpret_cast<fsverity_measurement*>(buffer.data());
2812     config->digest_algorithm = FS_VERITY_ALG_SHA256;
2813     config->digest_size = kSha256Size;
2814     memcpy(config->digest, expectedHash.data(), kSha256Size);
2815     if (ioctl(fd.get(), FS_IOC_SET_VERITY_MEASUREMENT, config) < 0) {
2816         // This includes an expected failure case with no FSVerity setup. It normally happens when
2817         // the apk does not contains the Merkle tree root hash.
2818         return error("Failed to measure fsverity on " + filePath + ": " + strerror(errno));
2819     }
2820     return ok();  // hashes match
2821 }
2822 
reconcileSecondaryDexFile(const std::string & dexPath,const std::string & packageName,int32_t uid,const std::vector<std::string> & isas,const std::optional<std::string> & volumeUuid,int32_t storage_flag,bool * _aidl_return)2823 binder::Status InstalldNativeService::reconcileSecondaryDexFile(
2824         const std::string& dexPath, const std::string& packageName, int32_t uid,
2825         const std::vector<std::string>& isas, const std::optional<std::string>& volumeUuid,
2826         int32_t storage_flag, bool* _aidl_return) {
2827     ENFORCE_UID(AID_SYSTEM);
2828     CHECK_ARGUMENT_UUID(volumeUuid);
2829     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2830     CHECK_ARGUMENT_PATH(dexPath);
2831     std::lock_guard<std::recursive_mutex> lock(mLock);
2832 
2833     bool result = android::installd::reconcile_secondary_dex_file(
2834             dexPath, packageName, uid, isas, volumeUuid, storage_flag, _aidl_return);
2835     return result ? ok() : error();
2836 }
2837 
hashSecondaryDexFile(const std::string & dexPath,const std::string & packageName,int32_t uid,const std::optional<std::string> & volumeUuid,int32_t storageFlag,std::vector<uint8_t> * _aidl_return)2838 binder::Status InstalldNativeService::hashSecondaryDexFile(
2839         const std::string& dexPath, const std::string& packageName, int32_t uid,
2840         const std::optional<std::string>& volumeUuid, int32_t storageFlag,
2841         std::vector<uint8_t>* _aidl_return) {
2842     ENFORCE_UID(AID_SYSTEM);
2843     CHECK_ARGUMENT_UUID(volumeUuid);
2844     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2845     CHECK_ARGUMENT_PATH(dexPath);
2846 
2847     // mLock is not taken here since we will never modify the file system.
2848     // If a file is modified just as we are reading it this may result in an
2849     // anomalous hash, but that's ok.
2850     bool result = android::installd::hash_secondary_dex_file(
2851         dexPath, packageName, uid, volumeUuid, storageFlag, _aidl_return);
2852     return result ? ok() : error();
2853 }
2854 
invalidateMounts()2855 binder::Status InstalldNativeService::invalidateMounts() {
2856     ENFORCE_UID(AID_SYSTEM);
2857     std::lock_guard<std::recursive_mutex> lock(mMountsLock);
2858 
2859     mStorageMounts.clear();
2860 
2861 #if !BYPASS_QUOTA
2862     if (!InvalidateQuotaMounts()) {
2863         return error("Failed to read mounts");
2864     }
2865 #endif
2866 
2867     std::ifstream in("/proc/mounts");
2868     if (!in.is_open()) {
2869         return error("Failed to read mounts");
2870     }
2871 
2872     std::string source;
2873     std::string target;
2874     std::string ignored;
2875     while (!in.eof()) {
2876         std::getline(in, source, ' ');
2877         std::getline(in, target, ' ');
2878         std::getline(in, ignored);
2879 
2880         if (android::base::GetBoolProperty(kFuseProp, false)) {
2881             if (target.find(kMntFuse) == 0) {
2882                 LOG(DEBUG) << "Found storage mount " << source << " at " << target;
2883                 mStorageMounts[source] = target;
2884             }
2885         } else {
2886 #if !BYPASS_SDCARDFS
2887             if (target.find(kMntSdcardfs) == 0) {
2888                 LOG(DEBUG) << "Found storage mount " << source << " at " << target;
2889                 mStorageMounts[source] = target;
2890             }
2891 #endif
2892         }
2893     }
2894     return ok();
2895 }
2896 
2897 // Mount volume's CE and DE storage to mirror
tryMountDataMirror(const std::optional<std::string> & uuid)2898 binder::Status InstalldNativeService::tryMountDataMirror(
2899         const std::optional<std::string>& uuid) {
2900     ENFORCE_UID(AID_SYSTEM);
2901     CHECK_ARGUMENT_UUID(uuid);
2902     if (!sAppDataIsolationEnabled) {
2903         return ok();
2904     }
2905     if (!uuid) {
2906         return error("Should not happen, mounting uuid == null");
2907     }
2908 
2909     const char* uuid_ = uuid->c_str();
2910 
2911     std::string mirrorVolCePath(StringPrintf("%s/%s", kDataMirrorCePath, uuid_));
2912     std::lock_guard<std::recursive_mutex> lock(mLock);
2913     if (fs_prepare_dir(mirrorVolCePath.c_str(), 0711, AID_SYSTEM, AID_SYSTEM) != 0) {
2914         return error("Failed to create CE mirror");
2915     }
2916 
2917     std::string mirrorVolDePath(StringPrintf("%s/%s", kDataMirrorDePath, uuid_));
2918     if (fs_prepare_dir(mirrorVolDePath.c_str(), 0711, AID_SYSTEM, AID_SYSTEM) != 0) {
2919         return error("Failed to create DE mirror");
2920     }
2921 
2922     auto cePath = StringPrintf("%s/user", create_data_path(uuid_).c_str());
2923     auto dePath = StringPrintf("%s/user_de", create_data_path(uuid_).c_str());
2924 
2925     if (access(cePath.c_str(), F_OK) != 0) {
2926         return error("Cannot access CE path: " + cePath);
2927     }
2928     if (access(dePath.c_str(), F_OK) != 0) {
2929         return error("Cannot access DE path: " + dePath);
2930     }
2931 
2932     struct stat ceStat, mirrorCeStat;
2933     if (stat(cePath.c_str(), &ceStat) != 0) {
2934         return error("Failed to stat " + cePath);
2935     }
2936     if (stat(mirrorVolCePath.c_str(), &mirrorCeStat) != 0) {
2937         return error("Failed to stat " + mirrorVolCePath);
2938     }
2939 
2940     if (mirrorCeStat.st_ino == ceStat.st_ino) {
2941         // As it's being called by prepareUserStorage, it can be called multiple times.
2942         // Hence, we if we mount it already, we should skip it.
2943         LOG(WARNING) << "CE dir is mounted already: " + cePath;
2944         return ok();
2945     }
2946 
2947     // Mount CE mirror
2948     if (TEMP_FAILURE_RETRY(mount(cePath.c_str(), mirrorVolCePath.c_str(), NULL,
2949             MS_NOSUID | MS_NODEV | MS_NOATIME | MS_BIND | MS_NOEXEC, nullptr)) == -1) {
2950         return error("Failed to mount " + mirrorVolCePath);
2951     }
2952 
2953     // Mount DE mirror
2954     if (TEMP_FAILURE_RETRY(mount(dePath.c_str(), mirrorVolDePath.c_str(), NULL,
2955             MS_NOSUID | MS_NODEV | MS_NOATIME | MS_BIND | MS_NOEXEC, nullptr)) == -1) {
2956         return error("Failed to mount " + mirrorVolDePath);
2957     }
2958     return ok();
2959 }
2960 
2961 // Unmount volume's CE and DE storage from mirror
onPrivateVolumeRemoved(const std::optional<std::string> & uuid)2962 binder::Status InstalldNativeService::onPrivateVolumeRemoved(
2963         const std::optional<std::string>& uuid) {
2964     ENFORCE_UID(AID_SYSTEM);
2965     CHECK_ARGUMENT_UUID(uuid);
2966     if (!sAppDataIsolationEnabled) {
2967         return ok();
2968     }
2969     if (!uuid) {
2970         // It happens when private volume failed to mount.
2971         LOG(INFO) << "Ignore unmount uuid=null";
2972         return ok();
2973     }
2974     const char* uuid_ = uuid->c_str();
2975 
2976     binder::Status res = ok();
2977 
2978     std::string mirrorCeVolPath(StringPrintf("%s/%s", kDataMirrorCePath, uuid_));
2979     std::string mirrorDeVolPath(StringPrintf("%s/%s", kDataMirrorDePath, uuid_));
2980 
2981     // Unmount CE storage
2982     std::lock_guard<std::recursive_mutex> lock(mLock);
2983     if (TEMP_FAILURE_RETRY(umount(mirrorCeVolPath.c_str())) != 0) {
2984         if (errno != ENOENT) {
2985             res = error(StringPrintf("Failed to umount %s %s", mirrorCeVolPath.c_str(),
2986                                 strerror(errno)));
2987         }
2988     }
2989     if (delete_dir_contents_and_dir(mirrorCeVolPath, true) != 0) {
2990         res = error("Failed to delete " + mirrorCeVolPath);
2991     }
2992 
2993     // Unmount DE storage
2994     if (TEMP_FAILURE_RETRY(umount(mirrorDeVolPath.c_str())) != 0) {
2995         if (errno != ENOENT) {
2996             res = error(StringPrintf("Failed to umount %s %s", mirrorDeVolPath.c_str(),
2997                                 strerror(errno)));
2998         }
2999     }
3000     if (delete_dir_contents_and_dir(mirrorDeVolPath, true) != 0) {
3001         res = error("Failed to delete " + mirrorDeVolPath);
3002     }
3003     return res;
3004 }
3005 
findDataMediaPath(const std::optional<std::string> & uuid,userid_t userid)3006 std::string InstalldNativeService::findDataMediaPath(
3007         const std::optional<std::string>& uuid, userid_t userid) {
3008     std::lock_guard<std::recursive_mutex> lock(mMountsLock);
3009     const char* uuid_ = uuid ? uuid->c_str() : nullptr;
3010     auto path = StringPrintf("%s/media", create_data_path(uuid_).c_str());
3011     auto resolved = mStorageMounts[path];
3012     if (resolved.empty()) {
3013         LOG(WARNING) << "Failed to find storage mount for " << path;
3014         resolved = path;
3015     }
3016     return StringPrintf("%s/%u", resolved.c_str(), userid);
3017 }
3018 
isQuotaSupported(const std::optional<std::string> & uuid,bool * _aidl_return)3019 binder::Status InstalldNativeService::isQuotaSupported(
3020         const std::optional<std::string>& uuid, bool* _aidl_return) {
3021     *_aidl_return = IsQuotaSupported(uuid.value_or(""));
3022     return ok();
3023 }
3024 
prepareAppProfile(const std::string & packageName,int32_t userId,int32_t appId,const std::string & profileName,const std::string & codePath,const std::optional<std::string> & dexMetadata,bool * _aidl_return)3025 binder::Status InstalldNativeService::prepareAppProfile(const std::string& packageName,
3026         int32_t userId, int32_t appId, const std::string& profileName, const std::string& codePath,
3027         const std::optional<std::string>& dexMetadata, bool* _aidl_return) {
3028     ENFORCE_UID(AID_SYSTEM);
3029     CHECK_ARGUMENT_PACKAGE_NAME(packageName);
3030     CHECK_ARGUMENT_PATH(codePath);
3031     std::lock_guard<std::recursive_mutex> lock(mLock);
3032 
3033     *_aidl_return = prepare_app_profile(packageName, userId, appId, profileName, codePath,
3034         dexMetadata);
3035     return ok();
3036 }
3037 
migrateLegacyObbData()3038 binder::Status InstalldNativeService::migrateLegacyObbData() {
3039     ENFORCE_UID(AID_SYSTEM);
3040     // NOTE: The lint warning doesn't apply to the use of system(3) with
3041     // absolute parse and no command line arguments.
3042     if (system("/system/bin/migrate_legacy_obb_data.sh") != 0) { // NOLINT(cert-env33-c)
3043         LOG(ERROR) << "Unable to migrate legacy obb data";
3044     }
3045 
3046     return ok();
3047 }
3048 
3049 }  // namespace installd
3050 }  // namespace android
3051