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