InstalldNativeService.cpp revision bb54a7df3a42d4beeb5fd99e01369415a7d1c24c
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 <errno.h>
22#include <inttypes.h>
23#include <fstream>
24#include <fts.h>
25#include <regex>
26#include <stdlib.h>
27#include <string.h>
28#include <sys/capability.h>
29#include <sys/file.h>
30#include <sys/resource.h>
31#include <sys/quota.h>
32#include <sys/stat.h>
33#include <sys/statvfs.h>
34#include <sys/types.h>
35#include <sys/wait.h>
36#include <sys/xattr.h>
37#include <unistd.h>
38
39#include <android-base/logging.h>
40#include <android-base/stringprintf.h>
41#include <android-base/strings.h>
42#include <android-base/unique_fd.h>
43#include <cutils/fs.h>
44#include <cutils/properties.h>
45#include <cutils/sched_policy.h>
46#include <log/log.h>               // TODO: Move everything to base/logging.
47#include <logwrap/logwrap.h>
48#include <private/android_filesystem_config.h>
49#include <selinux/android.h>
50#include <system/thread_defs.h>
51#include <utils/Trace.h>
52
53#include "dexopt.h"
54#include "globals.h"
55#include "installd_deps.h"
56#include "otapreopt_utils.h"
57#include "utils.h"
58
59#include "CacheTracker.h"
60#include "MatchExtensionGen.h"
61
62#ifndef LOG_TAG
63#define LOG_TAG "installd"
64#endif
65
66using android::base::StringPrintf;
67using std::endl;
68
69namespace android {
70namespace installd {
71
72static constexpr const char* kCpPath = "/system/bin/cp";
73static constexpr const char* kXattrDefault = "user.default";
74
75static constexpr const int MIN_RESTRICTED_HOME_SDK_VERSION = 24; // > M
76
77static constexpr const char* PKG_LIB_POSTFIX = "/lib";
78static constexpr const char* CACHE_DIR_POSTFIX = "/cache";
79static constexpr const char* CODE_CACHE_DIR_POSTFIX = "/code_cache";
80
81static constexpr const char* IDMAP_PREFIX = "/data/resource-cache/";
82static constexpr const char* IDMAP_SUFFIX = "@idmap";
83
84// NOTE: keep in sync with Installer
85static constexpr int FLAG_CLEAR_CACHE_ONLY = 1 << 8;
86static constexpr int FLAG_CLEAR_CODE_CACHE_ONLY = 1 << 9;
87static constexpr int FLAG_USE_QUOTA = 1 << 12;
88static constexpr int FLAG_FREE_CACHE_V2 = 1 << 13;
89static constexpr int FLAG_FREE_CACHE_V2_DEFY_QUOTA = 1 << 14;
90static constexpr int FLAG_FREE_CACHE_NOOP = 1 << 15;
91static constexpr int FLAG_FORCE = 1 << 16;
92
93namespace {
94
95constexpr const char* kDump = "android.permission.DUMP";
96
97static binder::Status ok() {
98    return binder::Status::ok();
99}
100
101static binder::Status exception(uint32_t code, const std::string& msg) {
102    return binder::Status::fromExceptionCode(code, String8(msg.c_str()));
103}
104
105static binder::Status error() {
106    return binder::Status::fromServiceSpecificError(errno);
107}
108
109static binder::Status error(const std::string& msg) {
110    PLOG(ERROR) << msg;
111    return binder::Status::fromServiceSpecificError(errno, String8(msg.c_str()));
112}
113
114static binder::Status error(uint32_t code, const std::string& msg) {
115    LOG(ERROR) << msg << " (" << code << ")";
116    return binder::Status::fromServiceSpecificError(code, String8(msg.c_str()));
117}
118
119binder::Status checkPermission(const char* permission) {
120    pid_t pid;
121    uid_t uid;
122
123    if (checkCallingPermission(String16(permission), reinterpret_cast<int32_t*>(&pid),
124            reinterpret_cast<int32_t*>(&uid))) {
125        return ok();
126    } else {
127        return exception(binder::Status::EX_SECURITY,
128                StringPrintf("UID %d / PID %d lacks permission %s", uid, pid, permission));
129    }
130}
131
132binder::Status checkUid(uid_t expectedUid) {
133    uid_t uid = IPCThreadState::self()->getCallingUid();
134    if (uid == expectedUid || uid == AID_ROOT) {
135        return ok();
136    } else {
137        return exception(binder::Status::EX_SECURITY,
138                StringPrintf("UID %d is not expected UID %d", uid, expectedUid));
139    }
140}
141
142binder::Status checkArgumentUuid(const std::unique_ptr<std::string>& uuid) {
143    if (!uuid || is_valid_filename(*uuid)) {
144        return ok();
145    } else {
146        return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
147                StringPrintf("UUID %s is malformed", uuid->c_str()));
148    }
149}
150
151binder::Status checkArgumentPackageName(const std::string& packageName) {
152    if (is_valid_package_name(packageName.c_str())) {
153        return ok();
154    } else {
155        return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
156                StringPrintf("Package name %s is malformed", packageName.c_str()));
157    }
158}
159
160#define ENFORCE_UID(uid) {                                  \
161    binder::Status status = checkUid((uid));                \
162    if (!status.isOk()) {                                   \
163        return status;                                      \
164    }                                                       \
165}
166
167#define CHECK_ARGUMENT_UUID(uuid) {                         \
168    binder::Status status = checkArgumentUuid((uuid));      \
169    if (!status.isOk()) {                                   \
170        return status;                                      \
171    }                                                       \
172}
173
174#define CHECK_ARGUMENT_PACKAGE_NAME(packageName) {          \
175    binder::Status status =                                 \
176            checkArgumentPackageName((packageName));        \
177    if (!status.isOk()) {                                   \
178        return status;                                      \
179    }                                                       \
180}
181
182}  // namespace
183
184status_t InstalldNativeService::start() {
185    IPCThreadState::self()->disableBackgroundScheduling(true);
186    status_t ret = BinderService<InstalldNativeService>::publish();
187    if (ret != android::OK) {
188        return ret;
189    }
190    sp<ProcessState> ps(ProcessState::self());
191    ps->startThreadPool();
192    ps->giveThreadPoolName();
193    return android::OK;
194}
195
196status_t InstalldNativeService::dump(int fd, const Vector<String16> & /* args */) {
197    auto out = std::fstream(StringPrintf("/proc/self/fd/%d", fd));
198    const binder::Status dump_permission = checkPermission(kDump);
199    if (!dump_permission.isOk()) {
200        out << dump_permission.toString8() << endl;
201        return PERMISSION_DENIED;
202    }
203    std::lock_guard<std::recursive_mutex> lock(mLock);
204
205    out << "installd is happy!" << endl;
206
207    {
208        std::lock_guard<std::recursive_mutex> lock(mQuotaDevicesLock);
209        out << endl << "Devices with quota support:" << endl;
210        for (const auto& n : mQuotaDevices) {
211            out << "    " << n.first << " = " << n.second << endl;
212        }
213    }
214
215    {
216        std::lock_guard<std::recursive_mutex> lock(mCacheQuotasLock);
217        out << endl << "Per-UID cache quotas:" << endl;
218        for (const auto& n : mCacheQuotas) {
219            out << "    " << n.first << " = " << n.second << endl;
220        }
221    }
222
223    out << endl;
224    out.flush();
225
226    return NO_ERROR;
227}
228
229/**
230 * Perform restorecon of the given path, but only perform recursive restorecon
231 * if the label of that top-level file actually changed.  This can save us
232 * significant time by avoiding no-op traversals of large filesystem trees.
233 */
234static int restorecon_app_data_lazy(const std::string& path, const std::string& seInfo, uid_t uid,
235        bool existing) {
236    int res = 0;
237    char* before = nullptr;
238    char* after = nullptr;
239
240    // Note that SELINUX_ANDROID_RESTORECON_DATADATA flag is set by
241    // libselinux. Not needed here.
242
243    if (lgetfilecon(path.c_str(), &before) < 0) {
244        PLOG(ERROR) << "Failed before getfilecon for " << path;
245        goto fail;
246    }
247    if (selinux_android_restorecon_pkgdir(path.c_str(), seInfo.c_str(), uid, 0) < 0) {
248        PLOG(ERROR) << "Failed top-level restorecon for " << path;
249        goto fail;
250    }
251    if (lgetfilecon(path.c_str(), &after) < 0) {
252        PLOG(ERROR) << "Failed after getfilecon for " << path;
253        goto fail;
254    }
255
256    // If the initial top-level restorecon above changed the label, then go
257    // back and restorecon everything recursively
258    if (strcmp(before, after)) {
259        if (existing) {
260            LOG(DEBUG) << "Detected label change from " << before << " to " << after << " at "
261                    << path << "; running recursive restorecon";
262        }
263        if (selinux_android_restorecon_pkgdir(path.c_str(), seInfo.c_str(), uid,
264                SELINUX_ANDROID_RESTORECON_RECURSE) < 0) {
265            PLOG(ERROR) << "Failed recursive restorecon for " << path;
266            goto fail;
267        }
268    }
269
270    goto done;
271fail:
272    res = -1;
273done:
274    free(before);
275    free(after);
276    return res;
277}
278
279static int restorecon_app_data_lazy(const std::string& parent, const char* name,
280        const std::string& seInfo, uid_t uid, bool existing) {
281    return restorecon_app_data_lazy(StringPrintf("%s/%s", parent.c_str(), name), seInfo, uid,
282            existing);
283}
284
285static int prepare_app_dir(const std::string& path, mode_t target_mode, uid_t uid) {
286    if (fs_prepare_dir_strict(path.c_str(), target_mode, uid, uid) != 0) {
287        PLOG(ERROR) << "Failed to prepare " << path;
288        return -1;
289    }
290    return 0;
291}
292
293/**
294 * Ensure that we have a hard-limit quota to protect against abusive apps;
295 * they should never use more than 90% of blocks or 50% of inodes.
296 */
297static int prepare_app_quota(const std::unique_ptr<std::string>& uuid, const std::string& device,
298        uid_t uid) {
299    if (device.empty()) return 0;
300
301    struct dqblk dq;
302    if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), device.c_str(), uid,
303            reinterpret_cast<char*>(&dq)) != 0) {
304        PLOG(WARNING) << "Failed to find quota for " << uid;
305        return -1;
306    }
307
308    if ((dq.dqb_bhardlimit == 0) || (dq.dqb_ihardlimit == 0)) {
309        auto path = create_data_path(uuid ? uuid->c_str() : nullptr);
310        struct statvfs stat;
311        if (statvfs(path.c_str(), &stat) != 0) {
312            PLOG(WARNING) << "Failed to statvfs " << path;
313            return -1;
314        }
315
316        dq.dqb_valid = QIF_LIMITS;
317        dq.dqb_bhardlimit = (((stat.f_blocks * stat.f_frsize) / 10) * 9) / QIF_DQBLKSIZE;
318        dq.dqb_ihardlimit = (stat.f_files / 2);
319        if (quotactl(QCMD(Q_SETQUOTA, USRQUOTA), device.c_str(), uid,
320                reinterpret_cast<char*>(&dq)) != 0) {
321            PLOG(WARNING) << "Failed to set hard quota for " << uid;
322            return -1;
323        } else {
324            LOG(DEBUG) << "Applied hard quotas for " << uid;
325            return 0;
326        }
327    } else {
328        // Hard quota already set; assume it's reasonable
329        return 0;
330    }
331}
332
333binder::Status InstalldNativeService::createAppData(const std::unique_ptr<std::string>& uuid,
334        const std::string& packageName, int32_t userId, int32_t flags, int32_t appId,
335        const std::string& seInfo, int32_t targetSdkVersion, int64_t* _aidl_return) {
336    ENFORCE_UID(AID_SYSTEM);
337    CHECK_ARGUMENT_UUID(uuid);
338    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
339    std::lock_guard<std::recursive_mutex> lock(mLock);
340
341    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
342    const char* pkgname = packageName.c_str();
343
344    // Assume invalid inode unless filled in below
345    if (_aidl_return != nullptr) *_aidl_return = -1;
346
347    int32_t uid = multiuser_get_uid(userId, appId);
348    int32_t cacheGid = multiuser_get_cache_gid(userId, appId);
349    mode_t targetMode = targetSdkVersion >= MIN_RESTRICTED_HOME_SDK_VERSION ? 0700 : 0751;
350
351    // If UID doesn't have a specific cache GID, use UID value
352    if (cacheGid == -1) {
353        cacheGid = uid;
354    }
355
356    if (flags & FLAG_STORAGE_CE) {
357        auto path = create_data_user_ce_package_path(uuid_, userId, pkgname);
358        bool existing = (access(path.c_str(), F_OK) == 0);
359
360        if (prepare_app_dir(path, targetMode, uid) ||
361                prepare_app_cache_dir(path, "cache", 02771, uid, cacheGid) ||
362                prepare_app_cache_dir(path, "code_cache", 02771, uid, cacheGid)) {
363            return error("Failed to prepare " + path);
364        }
365
366        // Consider restorecon over contents if label changed
367        if (restorecon_app_data_lazy(path, seInfo, uid, existing) ||
368                restorecon_app_data_lazy(path, "cache", seInfo, uid, existing) ||
369                restorecon_app_data_lazy(path, "code_cache", seInfo, uid, existing)) {
370            return error("Failed to restorecon " + path);
371        }
372
373        // Remember inode numbers of cache directories so that we can clear
374        // contents while CE storage is locked
375        if (write_path_inode(path, "cache", kXattrInodeCache) ||
376                write_path_inode(path, "code_cache", kXattrInodeCodeCache)) {
377            return error("Failed to write_path_inode for " + path);
378        }
379
380        // And return the CE inode of the top-level data directory so we can
381        // clear contents while CE storage is locked
382        if ((_aidl_return != nullptr)
383                && get_path_inode(path, reinterpret_cast<ino_t*>(_aidl_return)) != 0) {
384            return error("Failed to get_path_inode for " + path);
385        }
386    }
387    if (flags & FLAG_STORAGE_DE) {
388        auto path = create_data_user_de_package_path(uuid_, userId, pkgname);
389        bool existing = (access(path.c_str(), F_OK) == 0);
390
391        if (prepare_app_dir(path, targetMode, uid) ||
392                prepare_app_cache_dir(path, "cache", 02771, uid, cacheGid) ||
393                prepare_app_cache_dir(path, "code_cache", 02771, uid, cacheGid)) {
394            return error("Failed to prepare " + path);
395        }
396
397        // Consider restorecon over contents if label changed
398        if (restorecon_app_data_lazy(path, seInfo, uid, existing) ||
399                restorecon_app_data_lazy(path, "cache", seInfo, uid, existing) ||
400                restorecon_app_data_lazy(path, "code_cache", seInfo, uid, existing)) {
401            return error("Failed to restorecon " + path);
402        }
403
404        if (prepare_app_quota(uuid, findQuotaDeviceForUuid(uuid), uid)) {
405            return error("Failed to set hard quota " + path);
406        }
407
408        if (property_get_bool("dalvik.vm.usejitprofiles", false)) {
409            const std::string profile_dir =
410                    create_primary_current_profile_package_dir_path(userId, pkgname);
411            // read-write-execute only for the app user.
412            if (fs_prepare_dir_strict(profile_dir.c_str(), 0700, uid, uid) != 0) {
413                return error("Failed to prepare " + profile_dir);
414            }
415            const std::string profile_file = create_current_profile_path(userId, pkgname,
416                    /*is_secondary_dex*/false);
417            // read-write only for the app user.
418            if (fs_prepare_file_strict(profile_file.c_str(), 0600, uid, uid) != 0) {
419                return error("Failed to prepare " + profile_file);
420            }
421            const std::string ref_profile_path =
422                    create_primary_reference_profile_package_dir_path(pkgname);
423            // dex2oat/profman runs under the shared app gid and it needs to read/write reference
424            // profiles.
425            int shared_app_gid = multiuser_get_shared_gid(0, appId);
426            if ((shared_app_gid != -1) && fs_prepare_dir_strict(
427                    ref_profile_path.c_str(), 0700, shared_app_gid, shared_app_gid) != 0) {
428                return error("Failed to prepare " + ref_profile_path);
429            }
430        }
431    }
432    return ok();
433}
434
435binder::Status InstalldNativeService::migrateAppData(const std::unique_ptr<std::string>& uuid,
436        const std::string& packageName, int32_t userId, int32_t flags) {
437    ENFORCE_UID(AID_SYSTEM);
438    CHECK_ARGUMENT_UUID(uuid);
439    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
440    std::lock_guard<std::recursive_mutex> lock(mLock);
441
442    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
443    const char* pkgname = packageName.c_str();
444
445    // This method only exists to upgrade system apps that have requested
446    // forceDeviceEncrypted, so their default storage always lives in a
447    // consistent location.  This only works on non-FBE devices, since we
448    // never want to risk exposing data on a device with real CE/DE storage.
449
450    auto ce_path = create_data_user_ce_package_path(uuid_, userId, pkgname);
451    auto de_path = create_data_user_de_package_path(uuid_, userId, pkgname);
452
453    // If neither directory is marked as default, assume CE is default
454    if (getxattr(ce_path.c_str(), kXattrDefault, nullptr, 0) == -1
455            && getxattr(de_path.c_str(), kXattrDefault, nullptr, 0) == -1) {
456        if (setxattr(ce_path.c_str(), kXattrDefault, nullptr, 0, 0) != 0) {
457            return error("Failed to mark default storage " + ce_path);
458        }
459    }
460
461    // Migrate default data location if needed
462    auto target = (flags & FLAG_STORAGE_DE) ? de_path : ce_path;
463    auto source = (flags & FLAG_STORAGE_DE) ? ce_path : de_path;
464
465    if (getxattr(target.c_str(), kXattrDefault, nullptr, 0) == -1) {
466        LOG(WARNING) << "Requested default storage " << target
467                << " is not active; migrating from " << source;
468        if (delete_dir_contents_and_dir(target) != 0) {
469            return error("Failed to delete " + target);
470        }
471        if (rename(source.c_str(), target.c_str()) != 0) {
472            return error("Failed to rename " + source + " to " + target);
473        }
474    }
475
476    return ok();
477}
478
479
480binder::Status InstalldNativeService::clearAppProfiles(const std::string& packageName) {
481    ENFORCE_UID(AID_SYSTEM);
482    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
483    std::lock_guard<std::recursive_mutex> lock(mLock);
484
485    binder::Status res = ok();
486    if (!clear_primary_reference_profile(packageName)) {
487        res = error("Failed to clear reference profile for " + packageName);
488    }
489    if (!clear_primary_current_profiles(packageName)) {
490        res = error("Failed to clear current profiles for " + packageName);
491    }
492    return res;
493}
494
495binder::Status InstalldNativeService::clearAppData(const std::unique_ptr<std::string>& uuid,
496        const std::string& packageName, int32_t userId, int32_t flags, int64_t ceDataInode) {
497    ENFORCE_UID(AID_SYSTEM);
498    CHECK_ARGUMENT_UUID(uuid);
499    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
500    std::lock_guard<std::recursive_mutex> lock(mLock);
501
502    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
503    const char* pkgname = packageName.c_str();
504
505    binder::Status res = ok();
506    if (flags & FLAG_STORAGE_CE) {
507        auto path = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInode);
508        if (flags & FLAG_CLEAR_CACHE_ONLY) {
509            path = read_path_inode(path, "cache", kXattrInodeCache);
510        } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
511            path = read_path_inode(path, "code_cache", kXattrInodeCodeCache);
512        }
513        if (access(path.c_str(), F_OK) == 0) {
514            if (delete_dir_contents(path) != 0) {
515                res = error("Failed to delete contents of " + path);
516            }
517        }
518    }
519    if (flags & FLAG_STORAGE_DE) {
520        std::string suffix = "";
521        bool only_cache = false;
522        if (flags & FLAG_CLEAR_CACHE_ONLY) {
523            suffix = CACHE_DIR_POSTFIX;
524            only_cache = true;
525        } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
526            suffix = CODE_CACHE_DIR_POSTFIX;
527            only_cache = true;
528        }
529
530        auto path = create_data_user_de_package_path(uuid_, userId, pkgname) + suffix;
531        if (access(path.c_str(), F_OK) == 0) {
532            if (delete_dir_contents(path) != 0) {
533                res = error("Failed to delete contents of " + path);
534            }
535        }
536        if (!only_cache) {
537            if (!clear_primary_current_profile(packageName, userId)) {
538                res = error("Failed to clear current profile for " + packageName);
539            }
540        }
541    }
542    return res;
543}
544
545static int destroy_app_reference_profile(const std::string& pkgname) {
546    return delete_dir_contents_and_dir(
547        create_primary_reference_profile_package_dir_path(pkgname),
548        /*ignore_if_missing*/ true);
549}
550
551static int destroy_app_current_profiles(const std::string& pkgname, userid_t userid) {
552    return delete_dir_contents_and_dir(
553        create_primary_current_profile_package_dir_path(userid, pkgname),
554        /*ignore_if_missing*/ true);
555}
556
557binder::Status InstalldNativeService::destroyAppProfiles(const std::string& packageName) {
558    ENFORCE_UID(AID_SYSTEM);
559    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
560    std::lock_guard<std::recursive_mutex> lock(mLock);
561
562    binder::Status res = ok();
563    std::vector<userid_t> users = get_known_users(/*volume_uuid*/ nullptr);
564    for (auto user : users) {
565        if (destroy_app_current_profiles(packageName, user) != 0) {
566            res = error("Failed to destroy current profiles for " + packageName);
567        }
568    }
569    if (destroy_app_reference_profile(packageName) != 0) {
570        res = error("Failed to destroy reference profile for " + packageName);
571    }
572    return res;
573}
574
575binder::Status InstalldNativeService::destroyAppData(const std::unique_ptr<std::string>& uuid,
576        const std::string& packageName, int32_t userId, int32_t flags, int64_t ceDataInode) {
577    ENFORCE_UID(AID_SYSTEM);
578    CHECK_ARGUMENT_UUID(uuid);
579    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
580    std::lock_guard<std::recursive_mutex> lock(mLock);
581
582    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
583    const char* pkgname = packageName.c_str();
584
585    binder::Status res = ok();
586    if (flags & FLAG_STORAGE_CE) {
587        auto path = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInode);
588        if (delete_dir_contents_and_dir(path) != 0) {
589            res = error("Failed to delete " + path);
590        }
591    }
592    if (flags & FLAG_STORAGE_DE) {
593        auto path = create_data_user_de_package_path(uuid_, userId, pkgname);
594        if (delete_dir_contents_and_dir(path) != 0) {
595            res = error("Failed to delete " + path);
596        }
597        destroy_app_current_profiles(packageName, userId);
598        // TODO(calin): If the package is still installed by other users it's probably
599        // beneficial to keep the reference profile around.
600        // Verify if it's ok to do that.
601        destroy_app_reference_profile(packageName);
602    }
603    return res;
604}
605
606static gid_t get_cache_gid(uid_t uid) {
607    int32_t gid = multiuser_get_cache_gid(multiuser_get_user_id(uid), multiuser_get_app_id(uid));
608    return (gid != -1) ? gid : uid;
609}
610
611binder::Status InstalldNativeService::fixupAppData(const std::unique_ptr<std::string>& uuid,
612        int32_t flags) {
613    ENFORCE_UID(AID_SYSTEM);
614    CHECK_ARGUMENT_UUID(uuid);
615    std::lock_guard<std::recursive_mutex> lock(mLock);
616
617    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
618    for (auto user : get_known_users(uuid_)) {
619        ATRACE_BEGIN("fixup user");
620        FTS* fts;
621        FTSENT* p;
622        auto ce_path = create_data_user_ce_path(uuid_, user);
623        auto de_path = create_data_user_de_path(uuid_, user);
624        char *argv[] = { (char*) ce_path.c_str(), (char*) de_path.c_str(), nullptr };
625        if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, NULL))) {
626            return error("Failed to fts_open");
627        }
628        while ((p = fts_read(fts)) != nullptr) {
629            if (p->fts_info == FTS_D && p->fts_level == 1) {
630                // Track down inodes of cache directories
631                uint64_t raw = 0;
632                ino_t inode_cache = 0;
633                ino_t inode_code_cache = 0;
634                if (getxattr(p->fts_path, kXattrInodeCache, &raw, sizeof(raw)) == sizeof(raw)) {
635                    inode_cache = raw;
636                }
637                if (getxattr(p->fts_path, kXattrInodeCodeCache, &raw, sizeof(raw)) == sizeof(raw)) {
638                    inode_code_cache = raw;
639                }
640
641                // Figure out expected GID of each child
642                FTSENT* child = fts_children(fts, 0);
643                while (child != nullptr) {
644                    if ((child->fts_statp->st_ino == inode_cache)
645                            || (child->fts_statp->st_ino == inode_code_cache)
646                            || !strcmp(child->fts_name, "cache")
647                            || !strcmp(child->fts_name, "code_cache")) {
648                        child->fts_number = get_cache_gid(p->fts_statp->st_uid);
649                    } else {
650                        child->fts_number = p->fts_statp->st_uid;
651                    }
652                    child = child->fts_link;
653                }
654            } else if (p->fts_level >= 2) {
655                if (p->fts_level > 2) {
656                    // Inherit GID from parent once we're deeper into tree
657                    p->fts_number = p->fts_parent->fts_number;
658                }
659
660                uid_t uid = p->fts_parent->fts_statp->st_uid;
661                gid_t cache_gid = get_cache_gid(uid);
662                gid_t expected = p->fts_number;
663                gid_t actual = p->fts_statp->st_gid;
664                if (actual == expected) {
665#if FIXUP_DEBUG
666                    LOG(DEBUG) << "Ignoring " << p->fts_path << " with expected GID " << expected;
667#endif
668                    if (!(flags & FLAG_FORCE)) {
669                        fts_set(fts, p, FTS_SKIP);
670                    }
671                } else if ((actual == uid) || (actual == cache_gid)) {
672                    // Only consider fixing up when current GID belongs to app
673                    if (p->fts_info != FTS_D) {
674                        LOG(INFO) << "Fixing " << p->fts_path << " with unexpected GID " << actual
675                                << " instead of " << expected;
676                    }
677                    switch (p->fts_info) {
678                    case FTS_DP:
679                        // If we're moving towards cache GID, we need to set S_ISGID
680                        if (expected == cache_gid) {
681                            if (chmod(p->fts_path, 02771) != 0) {
682                                PLOG(WARNING) << "Failed to chmod " << p->fts_path;
683                            }
684                        }
685                        // Intentional fall through to also set GID
686                    case FTS_F:
687                        if (chown(p->fts_path, -1, expected) != 0) {
688                            PLOG(WARNING) << "Failed to chown " << p->fts_path;
689                        }
690                        break;
691                    case FTS_SL:
692                    case FTS_SLNONE:
693                        if (lchown(p->fts_path, -1, expected) != 0) {
694                            PLOG(WARNING) << "Failed to chown " << p->fts_path;
695                        }
696                        break;
697                    }
698                } else {
699                    // Ignore all other GID transitions, since they're kinda shady
700                    LOG(WARNING) << "Ignoring " << p->fts_path << " with unexpected GID " << actual
701                            << " instead of " << expected;
702                }
703            }
704        }
705        fts_close(fts);
706        ATRACE_END();
707    }
708    return ok();
709}
710
711binder::Status InstalldNativeService::moveCompleteApp(const std::unique_ptr<std::string>& fromUuid,
712        const std::unique_ptr<std::string>& toUuid, const std::string& packageName,
713        const std::string& dataAppName, int32_t appId, const std::string& seInfo,
714        int32_t targetSdkVersion) {
715    ENFORCE_UID(AID_SYSTEM);
716    CHECK_ARGUMENT_UUID(fromUuid);
717    CHECK_ARGUMENT_UUID(toUuid);
718    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
719    std::lock_guard<std::recursive_mutex> lock(mLock);
720
721    const char* from_uuid = fromUuid ? fromUuid->c_str() : nullptr;
722    const char* to_uuid = toUuid ? toUuid->c_str() : nullptr;
723    const char* package_name = packageName.c_str();
724    const char* data_app_name = dataAppName.c_str();
725
726    binder::Status res = ok();
727    std::vector<userid_t> users = get_known_users(from_uuid);
728
729    // Copy app
730    {
731        auto from = create_data_app_package_path(from_uuid, data_app_name);
732        auto to = create_data_app_package_path(to_uuid, data_app_name);
733        auto to_parent = create_data_app_path(to_uuid);
734
735        char *argv[] = {
736            (char*) kCpPath,
737            (char*) "-F", /* delete any existing destination file first (--remove-destination) */
738            (char*) "-p", /* preserve timestamps, ownership, and permissions */
739            (char*) "-R", /* recurse into subdirectories (DEST must be a directory) */
740            (char*) "-P", /* Do not follow symlinks [default] */
741            (char*) "-d", /* don't dereference symlinks */
742            (char*) from.c_str(),
743            (char*) to_parent.c_str()
744        };
745
746        LOG(DEBUG) << "Copying " << from << " to " << to;
747        int rc = android_fork_execvp(ARRAY_SIZE(argv), argv, NULL, false, true);
748        if (rc != 0) {
749            res = error(rc, "Failed copying " + from + " to " + to);
750            goto fail;
751        }
752
753        if (selinux_android_restorecon(to.c_str(), SELINUX_ANDROID_RESTORECON_RECURSE) != 0) {
754            res = error("Failed to restorecon " + to);
755            goto fail;
756        }
757    }
758
759    // Copy private data for all known users
760    for (auto user : users) {
761
762        // Data source may not exist for all users; that's okay
763        auto from_ce = create_data_user_ce_package_path(from_uuid, user, package_name);
764        if (access(from_ce.c_str(), F_OK) != 0) {
765            LOG(INFO) << "Missing source " << from_ce;
766            continue;
767        }
768
769        if (!createAppData(toUuid, packageName, user, FLAG_STORAGE_CE | FLAG_STORAGE_DE, appId,
770                seInfo, targetSdkVersion, nullptr).isOk()) {
771            res = error("Failed to create package target");
772            goto fail;
773        }
774
775        char *argv[] = {
776            (char*) kCpPath,
777            (char*) "-F", /* delete any existing destination file first (--remove-destination) */
778            (char*) "-p", /* preserve timestamps, ownership, and permissions */
779            (char*) "-R", /* recurse into subdirectories (DEST must be a directory) */
780            (char*) "-P", /* Do not follow symlinks [default] */
781            (char*) "-d", /* don't dereference symlinks */
782            nullptr,
783            nullptr
784        };
785
786        {
787            auto from = create_data_user_de_package_path(from_uuid, user, package_name);
788            auto to = create_data_user_de_path(to_uuid, user);
789            argv[6] = (char*) from.c_str();
790            argv[7] = (char*) to.c_str();
791
792            LOG(DEBUG) << "Copying " << from << " to " << to;
793            int rc = android_fork_execvp(ARRAY_SIZE(argv), argv, NULL, false, true);
794            if (rc != 0) {
795                res = error(rc, "Failed copying " + from + " to " + to);
796                goto fail;
797            }
798        }
799        {
800            auto from = create_data_user_ce_package_path(from_uuid, user, package_name);
801            auto to = create_data_user_ce_path(to_uuid, user);
802            argv[6] = (char*) from.c_str();
803            argv[7] = (char*) to.c_str();
804
805            LOG(DEBUG) << "Copying " << from << " to " << to;
806            int rc = android_fork_execvp(ARRAY_SIZE(argv), argv, NULL, false, true);
807            if (rc != 0) {
808                res = error(rc, "Failed copying " + from + " to " + to);
809                goto fail;
810            }
811        }
812
813        if (!restoreconAppData(toUuid, packageName, user, FLAG_STORAGE_CE | FLAG_STORAGE_DE,
814                appId, seInfo).isOk()) {
815            res = error("Failed to restorecon");
816            goto fail;
817        }
818    }
819
820    // We let the framework scan the new location and persist that before
821    // deleting the data in the old location; this ordering ensures that
822    // we can recover from things like battery pulls.
823    return ok();
824
825fail:
826    // Nuke everything we might have already copied
827    {
828        auto to = create_data_app_package_path(to_uuid, data_app_name);
829        if (delete_dir_contents(to.c_str(), 1, NULL) != 0) {
830            LOG(WARNING) << "Failed to rollback " << to;
831        }
832    }
833    for (auto user : users) {
834        {
835            auto to = create_data_user_de_package_path(to_uuid, user, package_name);
836            if (delete_dir_contents(to.c_str(), 1, NULL) != 0) {
837                LOG(WARNING) << "Failed to rollback " << to;
838            }
839        }
840        {
841            auto to = create_data_user_ce_package_path(to_uuid, user, package_name);
842            if (delete_dir_contents(to.c_str(), 1, NULL) != 0) {
843                LOG(WARNING) << "Failed to rollback " << to;
844            }
845        }
846    }
847    return res;
848}
849
850binder::Status InstalldNativeService::createUserData(const std::unique_ptr<std::string>& uuid,
851        int32_t userId, int32_t userSerial ATTRIBUTE_UNUSED, int32_t flags) {
852    ENFORCE_UID(AID_SYSTEM);
853    CHECK_ARGUMENT_UUID(uuid);
854    std::lock_guard<std::recursive_mutex> lock(mLock);
855
856    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
857    if (flags & FLAG_STORAGE_DE) {
858        if (uuid_ == nullptr) {
859            if (ensure_config_user_dirs(userId) != 0) {
860                return error(StringPrintf("Failed to ensure dirs for %d", userId));
861            }
862        }
863    }
864
865    // Data under /data/media doesn't have an app, but we still want
866    // to limit it to prevent abuse.
867    if (prepare_app_quota(uuid, findQuotaDeviceForUuid(uuid),
868            multiuser_get_uid(userId, AID_MEDIA_RW))) {
869        return error("Failed to set hard quota for media_rw");
870    }
871
872    return ok();
873}
874
875binder::Status InstalldNativeService::destroyUserData(const std::unique_ptr<std::string>& uuid,
876        int32_t userId, int32_t flags) {
877    ENFORCE_UID(AID_SYSTEM);
878    CHECK_ARGUMENT_UUID(uuid);
879    std::lock_guard<std::recursive_mutex> lock(mLock);
880
881    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
882    binder::Status res = ok();
883    if (flags & FLAG_STORAGE_DE) {
884        auto path = create_data_user_de_path(uuid_, userId);
885        if (delete_dir_contents_and_dir(path, true) != 0) {
886            res = error("Failed to delete " + path);
887        }
888        if (uuid_ == nullptr) {
889            path = create_data_misc_legacy_path(userId);
890            if (delete_dir_contents_and_dir(path, true) != 0) {
891                res = error("Failed to delete " + path);
892            }
893            path = create_primary_cur_profile_dir_path(userId);
894            if (delete_dir_contents_and_dir(path, true) != 0) {
895                res = error("Failed to delete " + path);
896            }
897        }
898    }
899    if (flags & FLAG_STORAGE_CE) {
900        auto path = create_data_user_ce_path(uuid_, userId);
901        if (delete_dir_contents_and_dir(path, true) != 0) {
902            res = error("Failed to delete " + path);
903        }
904        path = create_data_media_path(uuid_, userId);
905        if (delete_dir_contents_and_dir(path, true) != 0) {
906            res = error("Failed to delete " + path);
907        }
908    }
909    return res;
910}
911
912/* Try to ensure free_size bytes of storage are available.
913 * Returns 0 on success.
914 * This is rather simple-minded because doing a full LRU would
915 * be potentially memory-intensive, and without atime it would
916 * also require that apps constantly modify file metadata even
917 * when just reading from the cache, which is pretty awful.
918 */
919binder::Status InstalldNativeService::freeCache(const std::unique_ptr<std::string>& uuid,
920        int64_t freeStorageSize, int32_t flags) {
921    ENFORCE_UID(AID_SYSTEM);
922    CHECK_ARGUMENT_UUID(uuid);
923    std::lock_guard<std::recursive_mutex> lock(mLock);
924
925    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
926    auto data_path = create_data_path(uuid_);
927    auto device = findQuotaDeviceForUuid(uuid);
928    auto noop = (flags & FLAG_FREE_CACHE_NOOP);
929
930    int64_t free = data_disk_free(data_path);
931    if (free < 0) {
932        return error("Failed to determine free space for " + data_path);
933    }
934
935    int64_t needed = freeStorageSize - free;
936    LOG(DEBUG) << "Device " << data_path << " has " << free << " free; requested "
937            << freeStorageSize << "; needed " << needed;
938
939    if (free >= freeStorageSize) {
940        return ok();
941    }
942
943    if (flags & FLAG_FREE_CACHE_V2) {
944        // This new cache strategy fairly removes files from UIDs by deleting
945        // files from the UIDs which are most over their allocated quota
946
947        // 1. Create trackers for every known UID
948        ATRACE_BEGIN("create");
949        std::unordered_map<uid_t, std::shared_ptr<CacheTracker>> trackers;
950        for (auto user : get_known_users(uuid_)) {
951            FTS *fts;
952            FTSENT *p;
953            auto ce_path = create_data_user_ce_path(uuid_, user);
954            auto de_path = create_data_user_de_path(uuid_, user);
955            char *argv[] = { (char*) ce_path.c_str(), (char*) de_path.c_str(), nullptr };
956            if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, NULL))) {
957                return error("Failed to fts_open");
958            }
959            while ((p = fts_read(fts)) != NULL) {
960                if (p->fts_info == FTS_D && p->fts_level == 1) {
961                    uid_t uid = p->fts_statp->st_uid;
962                    auto search = trackers.find(uid);
963                    if (search != trackers.end()) {
964                        search->second->addDataPath(p->fts_path);
965                    } else {
966                        auto tracker = std::shared_ptr<CacheTracker>(new CacheTracker(
967                                multiuser_get_user_id(uid), multiuser_get_app_id(uid), device));
968                        tracker->addDataPath(p->fts_path);
969                        {
970                            std::lock_guard<std::recursive_mutex> lock(mCacheQuotasLock);
971                            tracker->cacheQuota = mCacheQuotas[uid];
972                        }
973                        if (tracker->cacheQuota == 0) {
974#if MEASURE_DEBUG
975                            LOG(WARNING) << "UID " << uid << " has no cache quota; assuming 64MB";
976#endif
977                            tracker->cacheQuota = 67108864;
978                        }
979                        trackers[uid] = tracker;
980                    }
981                    fts_set(fts, p, FTS_SKIP);
982                }
983            }
984            fts_close(fts);
985        }
986        ATRACE_END();
987
988        // 2. Populate tracker stats and insert into priority queue
989        ATRACE_BEGIN("populate");
990        auto cmp = [](std::shared_ptr<CacheTracker> left, std::shared_ptr<CacheTracker> right) {
991            return (left->getCacheRatio() < right->getCacheRatio());
992        };
993        std::priority_queue<std::shared_ptr<CacheTracker>,
994                std::vector<std::shared_ptr<CacheTracker>>, decltype(cmp)> queue(cmp);
995        for (const auto& it : trackers) {
996            it.second->loadStats();
997            queue.push(it.second);
998        }
999        ATRACE_END();
1000
1001        // 3. Bounce across the queue, freeing items from whichever tracker is
1002        // the most over their assigned quota
1003        ATRACE_BEGIN("bounce");
1004        std::shared_ptr<CacheTracker> active;
1005        while (active || !queue.empty()) {
1006            // Only look at apps under quota when explicitly requested
1007            if (active && (active->getCacheRatio() < 10000)
1008                    && !(flags & FLAG_FREE_CACHE_V2_DEFY_QUOTA)) {
1009                LOG(DEBUG) << "Active ratio " << active->getCacheRatio()
1010                        << " isn't over quota, and defy not requested";
1011                break;
1012            }
1013
1014            // Find the best tracker to work with; this might involve swapping
1015            // if the active tracker is no longer the most over quota
1016            bool nextBetter = active && !queue.empty()
1017                    && active->getCacheRatio() < queue.top()->getCacheRatio();
1018            if (!active || nextBetter) {
1019                if (active) {
1020                    // Current tracker still has items, so we'll consider it
1021                    // again later once it bubbles up to surface
1022                    queue.push(active);
1023                }
1024                active = queue.top(); queue.pop();
1025                active->ensureItems();
1026                continue;
1027            }
1028
1029            // If no items remain, go find another tracker
1030            if (active->items.empty()) {
1031                active = nullptr;
1032                continue;
1033            } else {
1034                auto item = active->items.back();
1035                active->items.pop_back();
1036
1037                LOG(DEBUG) << "Purging " << item->toString() << " from " << active->toString();
1038                if (!noop) {
1039                    item->purge();
1040                }
1041                active->cacheUsed -= item->size;
1042                needed -= item->size;
1043            }
1044
1045            // Verify that we're actually done before bailing, since sneaky
1046            // apps might be using hardlinks
1047            if (needed <= 0) {
1048                free = data_disk_free(data_path);
1049                needed = freeStorageSize - free;
1050                if (needed <= 0) {
1051                    break;
1052                } else {
1053                    LOG(WARNING) << "Expected to be done but still need " << needed;
1054                }
1055            }
1056        }
1057        ATRACE_END();
1058
1059    } else {
1060        return error("Legacy cache logic no longer supported");
1061    }
1062
1063    free = data_disk_free(data_path);
1064    if (free >= freeStorageSize) {
1065        return ok();
1066    } else {
1067        return error(StringPrintf("Failed to free up %" PRId64 " on %s; final free space %" PRId64,
1068                freeStorageSize, data_path.c_str(), free));
1069    }
1070}
1071
1072binder::Status InstalldNativeService::rmdex(const std::string& codePath,
1073        const std::string& instructionSet) {
1074    ENFORCE_UID(AID_SYSTEM);
1075    std::lock_guard<std::recursive_mutex> lock(mLock);
1076
1077    char dex_path[PKG_PATH_MAX];
1078
1079    const char* path = codePath.c_str();
1080    const char* instruction_set = instructionSet.c_str();
1081
1082    if (validate_apk_path(path) && validate_system_app_path(path)) {
1083        return error("Invalid path " + codePath);
1084    }
1085
1086    if (!create_cache_path(dex_path, path, instruction_set)) {
1087        return error("Failed to create cache path for " + codePath);
1088    }
1089
1090    ALOGV("unlink %s\n", dex_path);
1091    if (unlink(dex_path) < 0) {
1092        return error(StringPrintf("Failed to unlink %s", dex_path));
1093    } else {
1094        return ok();
1095    }
1096}
1097
1098struct stats {
1099    int64_t codeSize;
1100    int64_t dataSize;
1101    int64_t cacheSize;
1102};
1103
1104#if MEASURE_DEBUG
1105static std::string toString(std::vector<int64_t> values) {
1106    std::stringstream res;
1107    res << "[";
1108    for (size_t i = 0; i < values.size(); i++) {
1109        res << values[i];
1110        if (i < values.size() - 1) {
1111            res << ",";
1112        }
1113    }
1114    res << "]";
1115    return res.str();
1116}
1117#endif
1118
1119static void collectQuotaStats(const std::string& device, int32_t userId,
1120        int32_t appId, struct stats* stats, struct stats* extStats) {
1121    if (device.empty()) return;
1122
1123    struct dqblk dq;
1124
1125    if (stats != nullptr) {
1126        uid_t uid = multiuser_get_uid(userId, appId);
1127        if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), device.c_str(), uid,
1128                reinterpret_cast<char*>(&dq)) != 0) {
1129            if (errno != ESRCH) {
1130                PLOG(ERROR) << "Failed to quotactl " << device << " for UID " << uid;
1131            }
1132        } else {
1133#if MEASURE_DEBUG
1134            LOG(DEBUG) << "quotactl() for UID " << uid << " " << dq.dqb_curspace;
1135#endif
1136            stats->dataSize += dq.dqb_curspace;
1137        }
1138
1139        int cacheGid = multiuser_get_cache_gid(userId, appId);
1140        if (cacheGid != -1) {
1141            if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), cacheGid,
1142                    reinterpret_cast<char*>(&dq)) != 0) {
1143                if (errno != ESRCH) {
1144                    PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << cacheGid;
1145                }
1146            } else {
1147#if MEASURE_DEBUG
1148                LOG(DEBUG) << "quotactl() for GID " << cacheGid << " " << dq.dqb_curspace;
1149#endif
1150                stats->cacheSize += dq.dqb_curspace;
1151            }
1152        }
1153
1154        int sharedGid = multiuser_get_shared_gid(0, appId);
1155        if (sharedGid != -1) {
1156            if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), sharedGid,
1157                    reinterpret_cast<char*>(&dq)) != 0) {
1158                if (errno != ESRCH) {
1159                    PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << sharedGid;
1160                }
1161            } else {
1162#if MEASURE_DEBUG
1163                LOG(DEBUG) << "quotactl() for GID " << sharedGid << " " << dq.dqb_curspace;
1164#endif
1165                stats->codeSize += dq.dqb_curspace;
1166            }
1167        }
1168    }
1169
1170    if (extStats != nullptr) {
1171        int extGid = multiuser_get_ext_gid(userId, appId);
1172        if (extGid != -1) {
1173            if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), extGid,
1174                    reinterpret_cast<char*>(&dq)) != 0) {
1175                if (errno != ESRCH) {
1176                    PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << extGid;
1177                }
1178            } else {
1179#if MEASURE_DEBUG
1180                LOG(DEBUG) << "quotactl() for GID " << extGid << " " << dq.dqb_curspace;
1181#endif
1182                extStats->dataSize += dq.dqb_curspace;
1183            }
1184        }
1185
1186        int extCacheGid = multiuser_get_ext_cache_gid(userId, appId);
1187        if (extCacheGid != -1) {
1188            if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), extCacheGid,
1189                    reinterpret_cast<char*>(&dq)) != 0) {
1190                if (errno != ESRCH) {
1191                    PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << extCacheGid;
1192                }
1193            } else {
1194#if MEASURE_DEBUG
1195                LOG(DEBUG) << "quotactl() for GID " << extCacheGid << " " << dq.dqb_curspace;
1196#endif
1197                extStats->dataSize += dq.dqb_curspace;
1198                extStats->cacheSize += dq.dqb_curspace;
1199            }
1200        }
1201    }
1202}
1203
1204static void collectManualStats(const std::string& path, struct stats* stats) {
1205    DIR *d;
1206    int dfd;
1207    struct dirent *de;
1208    struct stat s;
1209
1210    d = opendir(path.c_str());
1211    if (d == nullptr) {
1212        if (errno != ENOENT) {
1213            PLOG(WARNING) << "Failed to open " << path;
1214        }
1215        return;
1216    }
1217    dfd = dirfd(d);
1218    while ((de = readdir(d))) {
1219        const char *name = de->d_name;
1220
1221        int64_t size = 0;
1222        if (fstatat(dfd, name, &s, AT_SYMLINK_NOFOLLOW) == 0) {
1223            size = s.st_blocks * 512;
1224        }
1225
1226        if (de->d_type == DT_DIR) {
1227            if (!strcmp(name, ".")) {
1228                // Don't recurse, but still count node size
1229            } else if (!strcmp(name, "..")) {
1230                // Don't recurse or count node size
1231                continue;
1232            } else {
1233                // Measure all children nodes
1234                size = 0;
1235                calculate_tree_size(StringPrintf("%s/%s", path.c_str(), name), &size);
1236            }
1237
1238            if (!strcmp(name, "cache") || !strcmp(name, "code_cache")) {
1239                stats->cacheSize += size;
1240            }
1241        }
1242
1243        // Legacy symlink isn't owned by app
1244        if (de->d_type == DT_LNK && !strcmp(name, "lib")) {
1245            continue;
1246        }
1247
1248        // Everything found inside is considered data
1249        stats->dataSize += size;
1250    }
1251    closedir(d);
1252}
1253
1254static void collectManualStatsForUser(const std::string& path, struct stats* stats,
1255        bool exclude_apps = false) {
1256    DIR *d;
1257    int dfd;
1258    struct dirent *de;
1259    struct stat s;
1260
1261    d = opendir(path.c_str());
1262    if (d == nullptr) {
1263        if (errno != ENOENT) {
1264            PLOG(WARNING) << "Failed to open " << path;
1265        }
1266        return;
1267    }
1268    dfd = dirfd(d);
1269    while ((de = readdir(d))) {
1270        if (de->d_type == DT_DIR) {
1271            const char *name = de->d_name;
1272            if (fstatat(dfd, name, &s, AT_SYMLINK_NOFOLLOW) != 0) {
1273                continue;
1274            }
1275            int32_t user_uid = multiuser_get_app_id(s.st_uid);
1276            if (!strcmp(name, ".") || !strcmp(name, "..")) {
1277                continue;
1278            } else if (exclude_apps && (user_uid >= AID_APP_START && user_uid <= AID_APP_END)) {
1279                continue;
1280            } else {
1281                collectManualStats(StringPrintf("%s/%s", path.c_str(), name), stats);
1282            }
1283        }
1284    }
1285    closedir(d);
1286}
1287
1288static void collectManualExternalStatsForUser(const std::string& path, struct stats* stats) {
1289    FTS *fts;
1290    FTSENT *p;
1291    char *argv[] = { (char*) path.c_str(), nullptr };
1292    if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, NULL))) {
1293        PLOG(ERROR) << "Failed to fts_open " << path;
1294        return;
1295    }
1296    while ((p = fts_read(fts)) != NULL) {
1297        p->fts_number = p->fts_parent->fts_number;
1298        switch (p->fts_info) {
1299        case FTS_D:
1300            if (p->fts_level == 4
1301                    && !strcmp(p->fts_name, "cache")
1302                    && !strcmp(p->fts_parent->fts_parent->fts_name, "data")
1303                    && !strcmp(p->fts_parent->fts_parent->fts_parent->fts_name, "Android")) {
1304                p->fts_number = 1;
1305            }
1306            // Fall through to count the directory
1307        case FTS_DEFAULT:
1308        case FTS_F:
1309        case FTS_SL:
1310        case FTS_SLNONE:
1311            int64_t size = (p->fts_statp->st_blocks * 512);
1312            if (p->fts_number == 1) {
1313                stats->cacheSize += size;
1314            }
1315            stats->dataSize += size;
1316            break;
1317        }
1318    }
1319    fts_close(fts);
1320}
1321
1322binder::Status InstalldNativeService::getAppSize(const std::unique_ptr<std::string>& uuid,
1323        const std::vector<std::string>& packageNames, int32_t userId, int32_t flags,
1324        int32_t appId, const std::vector<int64_t>& ceDataInodes,
1325        const std::vector<std::string>& codePaths, std::vector<int64_t>* _aidl_return) {
1326    ENFORCE_UID(AID_SYSTEM);
1327    CHECK_ARGUMENT_UUID(uuid);
1328    for (auto packageName : packageNames) {
1329        CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1330    }
1331    // NOTE: Locking is relaxed on this method, since it's limited to
1332    // read-only measurements without mutation.
1333
1334    // When modifying this logic, always verify using tests:
1335    // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetAppSize
1336
1337#if MEASURE_DEBUG
1338    LOG(INFO) << "Measuring user " << userId << " app " << appId;
1339#endif
1340
1341    // Here's a summary of the common storage locations across the platform,
1342    // and how they're each tagged:
1343    //
1344    // /data/app/com.example                           UID system
1345    // /data/app/com.example/oat                       UID system
1346    // /data/user/0/com.example                        UID u0_a10      GID u0_a10
1347    // /data/user/0/com.example/cache                  UID u0_a10      GID u0_a10_cache
1348    // /data/media/0/foo.txt                           UID u0_media_rw
1349    // /data/media/0/bar.jpg                           UID u0_media_rw GID u0_media_image
1350    // /data/media/0/Android/data/com.example          UID u0_media_rw GID u0_a10_ext
1351    // /data/media/0/Android/data/com.example/cache    UID u0_media_rw GID u0_a10_ext_cache
1352    // /data/media/obb/com.example                     UID system
1353
1354    struct stats stats;
1355    struct stats extStats;
1356    memset(&stats, 0, sizeof(stats));
1357    memset(&extStats, 0, sizeof(extStats));
1358
1359    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1360
1361    auto device = findQuotaDeviceForUuid(uuid);
1362    if (device.empty()) {
1363        flags &= ~FLAG_USE_QUOTA;
1364    }
1365
1366    ATRACE_BEGIN("obb");
1367    for (auto packageName : packageNames) {
1368        auto obbCodePath = create_data_media_obb_path(uuid_, packageName.c_str());
1369        calculate_tree_size(obbCodePath, &extStats.codeSize);
1370    }
1371    ATRACE_END();
1372
1373    if (flags & FLAG_USE_QUOTA && appId >= AID_APP_START) {
1374        ATRACE_BEGIN("code");
1375        for (auto codePath : codePaths) {
1376            calculate_tree_size(codePath, &stats.codeSize, -1,
1377                    multiuser_get_shared_gid(0, appId));
1378        }
1379        ATRACE_END();
1380
1381        ATRACE_BEGIN("quota");
1382        collectQuotaStats(device, userId, appId, &stats, &extStats);
1383        ATRACE_END();
1384    } else {
1385        ATRACE_BEGIN("code");
1386        for (auto codePath : codePaths) {
1387            calculate_tree_size(codePath, &stats.codeSize);
1388        }
1389        ATRACE_END();
1390
1391        for (size_t i = 0; i < packageNames.size(); i++) {
1392            const char* pkgname = packageNames[i].c_str();
1393
1394            ATRACE_BEGIN("data");
1395            auto cePath = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInodes[i]);
1396            collectManualStats(cePath, &stats);
1397            auto dePath = create_data_user_de_package_path(uuid_, userId, pkgname);
1398            collectManualStats(dePath, &stats);
1399            ATRACE_END();
1400
1401            if (!uuid) {
1402                ATRACE_BEGIN("profiles");
1403                calculate_tree_size(
1404                        create_primary_current_profile_package_dir_path(userId, pkgname),
1405                        &stats.dataSize);
1406                calculate_tree_size(
1407                        create_primary_reference_profile_package_dir_path(pkgname),
1408                        &stats.codeSize);
1409                ATRACE_END();
1410            }
1411
1412            ATRACE_BEGIN("external");
1413            auto extPath = create_data_media_package_path(uuid_, userId, "data", pkgname);
1414            collectManualStats(extPath, &extStats);
1415            auto mediaPath = create_data_media_package_path(uuid_, userId, "media", pkgname);
1416            calculate_tree_size(mediaPath, &extStats.dataSize);
1417            ATRACE_END();
1418        }
1419
1420        if (!uuid) {
1421            ATRACE_BEGIN("dalvik");
1422            int32_t sharedGid = multiuser_get_shared_gid(0, appId);
1423            if (sharedGid != -1) {
1424                calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize,
1425                        sharedGid, -1);
1426            }
1427            ATRACE_END();
1428        }
1429    }
1430
1431    std::vector<int64_t> ret;
1432    ret.push_back(stats.codeSize);
1433    ret.push_back(stats.dataSize);
1434    ret.push_back(stats.cacheSize);
1435    ret.push_back(extStats.codeSize);
1436    ret.push_back(extStats.dataSize);
1437    ret.push_back(extStats.cacheSize);
1438#if MEASURE_DEBUG
1439    LOG(DEBUG) << "Final result " << toString(ret);
1440#endif
1441    *_aidl_return = ret;
1442    return ok();
1443}
1444
1445binder::Status InstalldNativeService::getUserSize(const std::unique_ptr<std::string>& uuid,
1446        int32_t userId, int32_t flags, const std::vector<int32_t>& appIds,
1447        std::vector<int64_t>* _aidl_return) {
1448    ENFORCE_UID(AID_SYSTEM);
1449    CHECK_ARGUMENT_UUID(uuid);
1450    // NOTE: Locking is relaxed on this method, since it's limited to
1451    // read-only measurements without mutation.
1452
1453    // When modifying this logic, always verify using tests:
1454    // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetUserSize
1455
1456#if MEASURE_DEBUG
1457    LOG(INFO) << "Measuring user " << userId;
1458#endif
1459
1460    struct stats stats;
1461    struct stats extStats;
1462    memset(&stats, 0, sizeof(stats));
1463    memset(&extStats, 0, sizeof(extStats));
1464
1465    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1466
1467    auto device = findQuotaDeviceForUuid(uuid);
1468    if (device.empty()) {
1469        flags &= ~FLAG_USE_QUOTA;
1470    }
1471
1472    if (flags & FLAG_USE_QUOTA) {
1473        struct dqblk dq;
1474
1475        ATRACE_BEGIN("obb");
1476        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), AID_MEDIA_OBB,
1477                reinterpret_cast<char*>(&dq)) != 0) {
1478            if (errno != ESRCH) {
1479                PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << AID_MEDIA_OBB;
1480            }
1481        } else {
1482#if MEASURE_DEBUG
1483            LOG(DEBUG) << "quotactl() for GID " << AID_MEDIA_OBB << " " << dq.dqb_curspace;
1484#endif
1485            extStats.codeSize += dq.dqb_curspace;
1486        }
1487        ATRACE_END();
1488
1489        ATRACE_BEGIN("code");
1490        calculate_tree_size(create_data_app_path(uuid_), &stats.codeSize, -1, -1, true);
1491        ATRACE_END();
1492
1493        ATRACE_BEGIN("data");
1494        auto cePath = create_data_user_ce_path(uuid_, userId);
1495        collectManualStatsForUser(cePath, &stats, true);
1496        auto dePath = create_data_user_de_path(uuid_, userId);
1497        collectManualStatsForUser(dePath, &stats, true);
1498        ATRACE_END();
1499
1500        if (!uuid) {
1501            ATRACE_BEGIN("profile");
1502            auto userProfilePath = create_primary_cur_profile_dir_path(userId);
1503            calculate_tree_size(userProfilePath, &stats.dataSize, -1, -1, true);
1504            auto refProfilePath = create_primary_ref_profile_dir_path();
1505            calculate_tree_size(refProfilePath, &stats.codeSize, -1, -1, true);
1506            ATRACE_END();
1507        }
1508
1509        ATRACE_BEGIN("external");
1510        uid_t uid = multiuser_get_uid(userId, AID_MEDIA_RW);
1511        if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), device.c_str(), uid,
1512                reinterpret_cast<char*>(&dq)) != 0) {
1513            if (errno != ESRCH) {
1514                PLOG(ERROR) << "Failed to quotactl " << device << " for UID " << uid;
1515            }
1516        } else {
1517#if MEASURE_DEBUG
1518            LOG(DEBUG) << "quotactl() for UID " << uid << " " << dq.dqb_curspace;
1519#endif
1520            extStats.dataSize += dq.dqb_curspace;
1521        }
1522        ATRACE_END();
1523
1524        if (!uuid) {
1525            ATRACE_BEGIN("dalvik");
1526            calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize,
1527                    -1, -1, true);
1528            calculate_tree_size(create_primary_cur_profile_dir_path(userId), &stats.dataSize,
1529                    -1, -1, true);
1530            ATRACE_END();
1531        }
1532
1533        ATRACE_BEGIN("quota");
1534        int64_t dataSize = extStats.dataSize;
1535        for (auto appId : appIds) {
1536            if (appId >= AID_APP_START) {
1537                collectQuotaStats(device, userId, appId, &stats, &extStats);
1538
1539#if MEASURE_DEBUG
1540                // Sleep to make sure we don't lose logs
1541                usleep(1);
1542#endif
1543            }
1544        }
1545        extStats.dataSize = dataSize;
1546        ATRACE_END();
1547    } else {
1548        ATRACE_BEGIN("obb");
1549        auto obbPath = create_data_path(uuid_) + "/media/obb";
1550        calculate_tree_size(obbPath, &extStats.codeSize);
1551        ATRACE_END();
1552
1553        ATRACE_BEGIN("code");
1554        calculate_tree_size(create_data_app_path(uuid_), &stats.codeSize);
1555        ATRACE_END();
1556
1557        ATRACE_BEGIN("data");
1558        auto cePath = create_data_user_ce_path(uuid_, userId);
1559        collectManualStatsForUser(cePath, &stats);
1560        auto dePath = create_data_user_de_path(uuid_, userId);
1561        collectManualStatsForUser(dePath, &stats);
1562        ATRACE_END();
1563
1564        if (!uuid) {
1565            ATRACE_BEGIN("profile");
1566            auto userProfilePath = create_primary_cur_profile_dir_path(userId);
1567            calculate_tree_size(userProfilePath, &stats.dataSize);
1568            auto refProfilePath = create_primary_ref_profile_dir_path();
1569            calculate_tree_size(refProfilePath, &stats.codeSize);
1570            ATRACE_END();
1571        }
1572
1573        ATRACE_BEGIN("external");
1574        auto dataMediaPath = create_data_media_path(uuid_, userId);
1575        collectManualExternalStatsForUser(dataMediaPath, &extStats);
1576#if MEASURE_DEBUG
1577        LOG(DEBUG) << "Measured external data " << extStats.dataSize << " cache "
1578                << extStats.cacheSize;
1579#endif
1580        ATRACE_END();
1581
1582        if (!uuid) {
1583            ATRACE_BEGIN("dalvik");
1584            calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize);
1585            calculate_tree_size(create_primary_cur_profile_dir_path(userId), &stats.dataSize);
1586            ATRACE_END();
1587        }
1588    }
1589
1590    std::vector<int64_t> ret;
1591    ret.push_back(stats.codeSize);
1592    ret.push_back(stats.dataSize);
1593    ret.push_back(stats.cacheSize);
1594    ret.push_back(extStats.codeSize);
1595    ret.push_back(extStats.dataSize);
1596    ret.push_back(extStats.cacheSize);
1597#if MEASURE_DEBUG
1598    LOG(DEBUG) << "Final result " << toString(ret);
1599#endif
1600    *_aidl_return = ret;
1601    return ok();
1602}
1603
1604binder::Status InstalldNativeService::getExternalSize(const std::unique_ptr<std::string>& uuid,
1605        int32_t userId, int32_t flags, const std::vector<int32_t>& appIds,
1606        std::vector<int64_t>* _aidl_return) {
1607    ENFORCE_UID(AID_SYSTEM);
1608    CHECK_ARGUMENT_UUID(uuid);
1609    // NOTE: Locking is relaxed on this method, since it's limited to
1610    // read-only measurements without mutation.
1611
1612    // When modifying this logic, always verify using tests:
1613    // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetExternalSize
1614
1615#if MEASURE_DEBUG
1616    LOG(INFO) << "Measuring external " << userId;
1617#endif
1618
1619    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1620
1621    int64_t totalSize = 0;
1622    int64_t audioSize = 0;
1623    int64_t videoSize = 0;
1624    int64_t imageSize = 0;
1625    int64_t appSize = 0;
1626
1627    auto device = findQuotaDeviceForUuid(uuid);
1628    if (device.empty()) {
1629        flags &= ~FLAG_USE_QUOTA;
1630    }
1631
1632    if (flags & FLAG_USE_QUOTA) {
1633        struct dqblk dq;
1634
1635        ATRACE_BEGIN("quota");
1636        uid_t uid = multiuser_get_uid(userId, AID_MEDIA_RW);
1637        if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), device.c_str(), uid,
1638                reinterpret_cast<char*>(&dq)) != 0) {
1639            if (errno != ESRCH) {
1640                PLOG(ERROR) << "Failed to quotactl " << device << " for UID " << uid;
1641            }
1642        } else {
1643#if MEASURE_DEBUG
1644            LOG(DEBUG) << "quotactl() for UID " << uid << " " << dq.dqb_curspace;
1645#endif
1646            totalSize = dq.dqb_curspace;
1647        }
1648
1649        gid_t audioGid = multiuser_get_uid(userId, AID_MEDIA_AUDIO);
1650        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), audioGid,
1651                reinterpret_cast<char*>(&dq)) == 0) {
1652#if MEASURE_DEBUG
1653            LOG(DEBUG) << "quotactl() for GID " << audioGid << " " << dq.dqb_curspace;
1654#endif
1655            audioSize = dq.dqb_curspace;
1656        }
1657        gid_t videoGid = multiuser_get_uid(userId, AID_MEDIA_VIDEO);
1658        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), videoGid,
1659                reinterpret_cast<char*>(&dq)) == 0) {
1660#if MEASURE_DEBUG
1661            LOG(DEBUG) << "quotactl() for GID " << videoGid << " " << dq.dqb_curspace;
1662#endif
1663            videoSize = dq.dqb_curspace;
1664        }
1665        gid_t imageGid = multiuser_get_uid(userId, AID_MEDIA_IMAGE);
1666        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), imageGid,
1667                reinterpret_cast<char*>(&dq)) == 0) {
1668#if MEASURE_DEBUG
1669            LOG(DEBUG) << "quotactl() for GID " << imageGid << " " << dq.dqb_curspace;
1670#endif
1671            imageSize = dq.dqb_curspace;
1672        }
1673        ATRACE_END();
1674
1675        ATRACE_BEGIN("apps");
1676        struct stats extStats;
1677        memset(&extStats, 0, sizeof(extStats));
1678        for (auto appId : appIds) {
1679            if (appId >= AID_APP_START) {
1680                collectQuotaStats(device, userId, appId, nullptr, &extStats);
1681            }
1682        }
1683        appSize = extStats.dataSize + extStats.cacheSize;
1684        ATRACE_END();
1685    } else {
1686        ATRACE_BEGIN("manual");
1687        FTS *fts;
1688        FTSENT *p;
1689        auto path = create_data_media_path(uuid_, userId);
1690        char *argv[] = { (char*) path.c_str(), nullptr };
1691        if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_NOCHDIR | FTS_XDEV, NULL))) {
1692            return error("Failed to fts_open " + path);
1693        }
1694        while ((p = fts_read(fts)) != NULL) {
1695            char* ext;
1696            int64_t size = (p->fts_statp->st_blocks * 512);
1697            switch (p->fts_info) {
1698            case FTS_F:
1699                // Only categorize files not belonging to apps
1700                if (p->fts_parent->fts_number == 0) {
1701                    ext = strrchr(p->fts_name, '.');
1702                    if (ext != nullptr) {
1703                        switch (MatchExtension(++ext)) {
1704                        case AID_MEDIA_AUDIO: audioSize += size; break;
1705                        case AID_MEDIA_VIDEO: videoSize += size; break;
1706                        case AID_MEDIA_IMAGE: imageSize += size; break;
1707                        }
1708                    }
1709                }
1710                // Fall through to always count against total
1711            case FTS_D:
1712                // Ignore data belonging to specific apps
1713                p->fts_number = p->fts_parent->fts_number;
1714                if (p->fts_level == 1 && !strcmp(p->fts_name, "Android")) {
1715                    p->fts_number = 1;
1716                }
1717            case FTS_DEFAULT:
1718            case FTS_SL:
1719            case FTS_SLNONE:
1720                if (p->fts_parent->fts_number == 1) {
1721                    appSize += size;
1722                }
1723                totalSize += size;
1724                break;
1725            }
1726        }
1727        fts_close(fts);
1728        ATRACE_END();
1729    }
1730
1731    std::vector<int64_t> ret;
1732    ret.push_back(totalSize);
1733    ret.push_back(audioSize);
1734    ret.push_back(videoSize);
1735    ret.push_back(imageSize);
1736    ret.push_back(appSize);
1737#if MEASURE_DEBUG
1738    LOG(DEBUG) << "Final result " << toString(ret);
1739#endif
1740    *_aidl_return = ret;
1741    return ok();
1742}
1743
1744binder::Status InstalldNativeService::setAppQuota(const std::unique_ptr<std::string>& uuid,
1745        int32_t userId, int32_t appId, int64_t cacheQuota) {
1746    ENFORCE_UID(AID_SYSTEM);
1747    CHECK_ARGUMENT_UUID(uuid);
1748    std::lock_guard<std::recursive_mutex> lock(mCacheQuotasLock);
1749
1750    int32_t uid = multiuser_get_uid(userId, appId);
1751    mCacheQuotas[uid] = cacheQuota;
1752
1753    return ok();
1754}
1755
1756// Dumps the contents of a profile file, using pkgname's dex files for pretty
1757// printing the result.
1758binder::Status InstalldNativeService::dumpProfiles(int32_t uid, const std::string& packageName,
1759        const std::string& codePaths, bool* _aidl_return) {
1760    ENFORCE_UID(AID_SYSTEM);
1761    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1762    std::lock_guard<std::recursive_mutex> lock(mLock);
1763
1764    const char* pkgname = packageName.c_str();
1765    const char* code_paths = codePaths.c_str();
1766
1767    *_aidl_return = dump_profiles(uid, pkgname, code_paths);
1768    return ok();
1769}
1770
1771// TODO: Consider returning error codes.
1772binder::Status InstalldNativeService::mergeProfiles(int32_t uid, const std::string& packageName,
1773        bool* _aidl_return) {
1774    ENFORCE_UID(AID_SYSTEM);
1775    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1776    std::lock_guard<std::recursive_mutex> lock(mLock);
1777
1778    *_aidl_return = analyze_primary_profiles(uid, packageName);
1779    return ok();
1780}
1781
1782binder::Status InstalldNativeService::dexopt(const std::string& apkPath, int32_t uid,
1783        const std::unique_ptr<std::string>& packageName, const std::string& instructionSet,
1784        int32_t dexoptNeeded, const std::unique_ptr<std::string>& outputPath, int32_t dexFlags,
1785        const std::string& compilerFilter, const std::unique_ptr<std::string>& uuid,
1786        const std::unique_ptr<std::string>& sharedLibraries,
1787        const std::unique_ptr<std::string>& seInfo) {
1788    ENFORCE_UID(AID_SYSTEM);
1789    CHECK_ARGUMENT_UUID(uuid);
1790    if (packageName && *packageName != "*") {
1791        CHECK_ARGUMENT_PACKAGE_NAME(*packageName);
1792    }
1793    std::lock_guard<std::recursive_mutex> lock(mLock);
1794
1795    const char* apk_path = apkPath.c_str();
1796    const char* pkgname = packageName ? packageName->c_str() : "*";
1797    const char* instruction_set = instructionSet.c_str();
1798    const char* oat_dir = outputPath ? outputPath->c_str() : nullptr;
1799    const char* compiler_filter = compilerFilter.c_str();
1800    const char* volume_uuid = uuid ? uuid->c_str() : nullptr;
1801    const char* shared_libraries = sharedLibraries ? sharedLibraries->c_str() : nullptr;
1802    const char* se_info = seInfo ? seInfo->c_str() : nullptr;
1803    int res = android::installd::dexopt(apk_path, uid, pkgname, instruction_set, dexoptNeeded,
1804            oat_dir, dexFlags, compiler_filter, volume_uuid, shared_libraries, se_info);
1805    return res ? error(res, "Failed to dexopt") : ok();
1806}
1807
1808binder::Status InstalldNativeService::markBootComplete(const std::string& instructionSet) {
1809    ENFORCE_UID(AID_SYSTEM);
1810    std::lock_guard<std::recursive_mutex> lock(mLock);
1811
1812    const char* instruction_set = instructionSet.c_str();
1813
1814    char boot_marker_path[PKG_PATH_MAX];
1815    sprintf(boot_marker_path,
1816          "%s/%s/%s/.booting",
1817          android_data_dir.path,
1818          DALVIK_CACHE,
1819          instruction_set);
1820
1821    ALOGV("mark_boot_complete : %s", boot_marker_path);
1822    if (unlink(boot_marker_path) != 0) {
1823        return error(StringPrintf("Failed to unlink %s", boot_marker_path));
1824    }
1825    return ok();
1826}
1827
1828void mkinnerdirs(char* path, int basepos, mode_t mode, int uid, int gid,
1829        struct stat* statbuf)
1830{
1831    while (path[basepos] != 0) {
1832        if (path[basepos] == '/') {
1833            path[basepos] = 0;
1834            if (lstat(path, statbuf) < 0) {
1835                ALOGV("Making directory: %s\n", path);
1836                if (mkdir(path, mode) == 0) {
1837                    chown(path, uid, gid);
1838                } else {
1839                    ALOGW("Unable to make directory %s: %s\n", path, strerror(errno));
1840                }
1841            }
1842            path[basepos] = '/';
1843            basepos++;
1844        }
1845        basepos++;
1846    }
1847}
1848
1849binder::Status InstalldNativeService::linkNativeLibraryDirectory(
1850        const std::unique_ptr<std::string>& uuid, const std::string& packageName,
1851        const std::string& nativeLibPath32, int32_t userId) {
1852    ENFORCE_UID(AID_SYSTEM);
1853    CHECK_ARGUMENT_UUID(uuid);
1854    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1855    std::lock_guard<std::recursive_mutex> lock(mLock);
1856
1857    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1858    const char* pkgname = packageName.c_str();
1859    const char* asecLibDir = nativeLibPath32.c_str();
1860    struct stat s, libStat;
1861    binder::Status res = ok();
1862
1863    auto _pkgdir = create_data_user_ce_package_path(uuid_, userId, pkgname);
1864    auto _libsymlink = _pkgdir + PKG_LIB_POSTFIX;
1865
1866    const char* pkgdir = _pkgdir.c_str();
1867    const char* libsymlink = _libsymlink.c_str();
1868
1869    if (stat(pkgdir, &s) < 0) {
1870        return error("Failed to stat " + _pkgdir);
1871    }
1872
1873    if (chown(pkgdir, AID_INSTALL, AID_INSTALL) < 0) {
1874        return error("Failed to chown " + _pkgdir);
1875    }
1876
1877    if (chmod(pkgdir, 0700) < 0) {
1878        res = error("Failed to chmod " + _pkgdir);
1879        goto out;
1880    }
1881
1882    if (lstat(libsymlink, &libStat) < 0) {
1883        if (errno != ENOENT) {
1884            res = error("Failed to stat " + _libsymlink);
1885            goto out;
1886        }
1887    } else {
1888        if (S_ISDIR(libStat.st_mode)) {
1889            if (delete_dir_contents(libsymlink, 1, NULL) < 0) {
1890                res = error("Failed to delete " + _libsymlink);
1891                goto out;
1892            }
1893        } else if (S_ISLNK(libStat.st_mode)) {
1894            if (unlink(libsymlink) < 0) {
1895                res = error("Failed to unlink " + _libsymlink);
1896                goto out;
1897            }
1898        }
1899    }
1900
1901    if (symlink(asecLibDir, libsymlink) < 0) {
1902        res = error("Failed to symlink " + _libsymlink + " to " + nativeLibPath32);
1903        goto out;
1904    }
1905
1906out:
1907    if (chmod(pkgdir, s.st_mode) < 0) {
1908        auto msg = "Failed to cleanup chmod " + _pkgdir;
1909        if (res.isOk()) {
1910            res = error(msg);
1911        } else {
1912            PLOG(ERROR) << msg;
1913        }
1914    }
1915
1916    if (chown(pkgdir, s.st_uid, s.st_gid) < 0) {
1917        auto msg = "Failed to cleanup chown " + _pkgdir;
1918        if (res.isOk()) {
1919            res = error(msg);
1920        } else {
1921            PLOG(ERROR) << msg;
1922        }
1923    }
1924
1925    return res;
1926}
1927
1928static void run_idmap(const char *target_apk, const char *overlay_apk, int idmap_fd)
1929{
1930    static const char *IDMAP_BIN = "/system/bin/idmap";
1931    static const size_t MAX_INT_LEN = 32;
1932    char idmap_str[MAX_INT_LEN];
1933
1934    snprintf(idmap_str, sizeof(idmap_str), "%d", idmap_fd);
1935
1936    execl(IDMAP_BIN, IDMAP_BIN, "--fd", target_apk, overlay_apk, idmap_str, (char*)NULL);
1937    ALOGE("execl(%s) failed: %s\n", IDMAP_BIN, strerror(errno));
1938}
1939
1940// Transform string /a/b/c.apk to (prefix)/a@b@c.apk@(suffix)
1941// eg /a/b/c.apk to /data/resource-cache/a@b@c.apk@idmap
1942static int flatten_path(const char *prefix, const char *suffix,
1943        const char *overlay_path, char *idmap_path, size_t N)
1944{
1945    if (overlay_path == NULL || idmap_path == NULL) {
1946        return -1;
1947    }
1948    const size_t len_overlay_path = strlen(overlay_path);
1949    // will access overlay_path + 1 further below; requires absolute path
1950    if (len_overlay_path < 2 || *overlay_path != '/') {
1951        return -1;
1952    }
1953    const size_t len_idmap_root = strlen(prefix);
1954    const size_t len_suffix = strlen(suffix);
1955    if (SIZE_MAX - len_idmap_root < len_overlay_path ||
1956            SIZE_MAX - (len_idmap_root + len_overlay_path) < len_suffix) {
1957        // additions below would cause overflow
1958        return -1;
1959    }
1960    if (N < len_idmap_root + len_overlay_path + len_suffix) {
1961        return -1;
1962    }
1963    memset(idmap_path, 0, N);
1964    snprintf(idmap_path, N, "%s%s%s", prefix, overlay_path + 1, suffix);
1965    char *ch = idmap_path + len_idmap_root;
1966    while (*ch != '\0') {
1967        if (*ch == '/') {
1968            *ch = '@';
1969        }
1970        ++ch;
1971    }
1972    return 0;
1973}
1974
1975binder::Status InstalldNativeService::idmap(const std::string& targetApkPath,
1976        const std::string& overlayApkPath, int32_t uid) {
1977    ENFORCE_UID(AID_SYSTEM);
1978    std::lock_guard<std::recursive_mutex> lock(mLock);
1979
1980    const char* target_apk = targetApkPath.c_str();
1981    const char* overlay_apk = overlayApkPath.c_str();
1982    ALOGV("idmap target_apk=%s overlay_apk=%s uid=%d\n", target_apk, overlay_apk, uid);
1983
1984    int idmap_fd = -1;
1985    char idmap_path[PATH_MAX];
1986    struct stat target_apk_stat, overlay_apk_stat, idmap_stat;
1987    bool outdated = false;
1988
1989    if (flatten_path(IDMAP_PREFIX, IDMAP_SUFFIX, overlay_apk,
1990                idmap_path, sizeof(idmap_path)) == -1) {
1991        ALOGE("idmap cannot generate idmap path for overlay %s\n", overlay_apk);
1992        goto fail;
1993    }
1994
1995    if (stat(idmap_path, &idmap_stat) < 0 ||
1996            stat(target_apk, &target_apk_stat) < 0 ||
1997            stat(overlay_apk, &overlay_apk_stat) < 0) {
1998        outdated = true;
1999    } else if (idmap_stat.st_mtime < target_apk_stat.st_mtime ||
2000            idmap_stat.st_mtime < overlay_apk_stat.st_mtime) {
2001        outdated = true;
2002    }
2003
2004    if (outdated) {
2005        unlink(idmap_path);
2006        idmap_fd = open(idmap_path, O_RDWR | O_CREAT | O_EXCL, 0644);
2007    } else {
2008        idmap_fd = open(idmap_path, O_RDWR);
2009    }
2010
2011    if (idmap_fd < 0) {
2012        ALOGE("idmap cannot open '%s' for output: %s\n", idmap_path, strerror(errno));
2013        goto fail;
2014    }
2015    if (fchown(idmap_fd, AID_SYSTEM, uid) < 0) {
2016        ALOGE("idmap cannot chown '%s'\n", idmap_path);
2017        goto fail;
2018    }
2019    if (fchmod(idmap_fd, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH) < 0) {
2020        ALOGE("idmap cannot chmod '%s'\n", idmap_path);
2021        goto fail;
2022    }
2023
2024    if (!outdated) {
2025        close(idmap_fd);
2026        return ok();
2027    }
2028
2029    pid_t pid;
2030    pid = fork();
2031    if (pid == 0) {
2032        /* child -- drop privileges before continuing */
2033        if (setgid(uid) != 0) {
2034            ALOGE("setgid(%d) failed during idmap\n", uid);
2035            exit(1);
2036        }
2037        if (setuid(uid) != 0) {
2038            ALOGE("setuid(%d) failed during idmap\n", uid);
2039            exit(1);
2040        }
2041        if (flock(idmap_fd, LOCK_EX | LOCK_NB) != 0) {
2042            ALOGE("flock(%s) failed during idmap: %s\n", idmap_path, strerror(errno));
2043            exit(1);
2044        }
2045
2046        run_idmap(target_apk, overlay_apk, idmap_fd);
2047        exit(1); /* only if exec call to idmap failed */
2048    } else {
2049        int status = wait_child(pid);
2050        if (status != 0) {
2051            ALOGE("idmap failed, status=0x%04x\n", status);
2052            goto fail;
2053        }
2054    }
2055
2056    close(idmap_fd);
2057    return ok();
2058fail:
2059    if (idmap_fd >= 0) {
2060        close(idmap_fd);
2061        unlink(idmap_path);
2062    }
2063    return error();
2064}
2065
2066binder::Status InstalldNativeService::removeIdmap(const std::string& overlayApkPath) {
2067    const char* overlay_apk = overlayApkPath.c_str();
2068    char idmap_path[PATH_MAX];
2069
2070    if (flatten_path(IDMAP_PREFIX, IDMAP_SUFFIX, overlay_apk,
2071                idmap_path, sizeof(idmap_path)) == -1) {
2072        ALOGE("idmap cannot generate idmap path for overlay %s\n", overlay_apk);
2073        return error();
2074    }
2075    if (unlink(idmap_path) < 0) {
2076        ALOGE("couldn't unlink idmap file %s\n", idmap_path);
2077        return error();
2078    }
2079    return ok();
2080}
2081
2082binder::Status InstalldNativeService::restoreconAppData(const std::unique_ptr<std::string>& uuid,
2083        const std::string& packageName, int32_t userId, int32_t flags, int32_t appId,
2084        const std::string& seInfo) {
2085    ENFORCE_UID(AID_SYSTEM);
2086    CHECK_ARGUMENT_UUID(uuid);
2087    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2088    std::lock_guard<std::recursive_mutex> lock(mLock);
2089
2090    binder::Status res = ok();
2091
2092    // SELINUX_ANDROID_RESTORECON_DATADATA flag is set by libselinux. Not needed here.
2093    unsigned int seflags = SELINUX_ANDROID_RESTORECON_RECURSE;
2094    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
2095    const char* pkgName = packageName.c_str();
2096    const char* seinfo = seInfo.c_str();
2097
2098    uid_t uid = multiuser_get_uid(userId, appId);
2099    if (flags & FLAG_STORAGE_CE) {
2100        auto path = create_data_user_ce_package_path(uuid_, userId, pkgName);
2101        if (selinux_android_restorecon_pkgdir(path.c_str(), seinfo, uid, seflags) < 0) {
2102            res = error("restorecon failed for " + path);
2103        }
2104    }
2105    if (flags & FLAG_STORAGE_DE) {
2106        auto path = create_data_user_de_package_path(uuid_, userId, pkgName);
2107        if (selinux_android_restorecon_pkgdir(path.c_str(), seinfo, uid, seflags) < 0) {
2108            res = error("restorecon failed for " + path);
2109        }
2110    }
2111    return res;
2112}
2113
2114binder::Status InstalldNativeService::createOatDir(const std::string& oatDir,
2115        const std::string& instructionSet) {
2116    ENFORCE_UID(AID_SYSTEM);
2117    std::lock_guard<std::recursive_mutex> lock(mLock);
2118
2119    const char* oat_dir = oatDir.c_str();
2120    const char* instruction_set = instructionSet.c_str();
2121    char oat_instr_dir[PKG_PATH_MAX];
2122
2123    if (validate_apk_path(oat_dir)) {
2124        return error("Invalid path " + oatDir);
2125    }
2126    if (fs_prepare_dir(oat_dir, S_IRWXU | S_IRWXG | S_IXOTH, AID_SYSTEM, AID_INSTALL)) {
2127        return error("Failed to prepare " + oatDir);
2128    }
2129    if (selinux_android_restorecon(oat_dir, 0)) {
2130        return error("Failed to restorecon " + oatDir);
2131    }
2132    snprintf(oat_instr_dir, PKG_PATH_MAX, "%s/%s", oat_dir, instruction_set);
2133    if (fs_prepare_dir(oat_instr_dir, S_IRWXU | S_IRWXG | S_IXOTH, AID_SYSTEM, AID_INSTALL)) {
2134        return error(StringPrintf("Failed to prepare %s", oat_instr_dir));
2135    }
2136    return ok();
2137}
2138
2139binder::Status InstalldNativeService::rmPackageDir(const std::string& packageDir) {
2140    ENFORCE_UID(AID_SYSTEM);
2141    std::lock_guard<std::recursive_mutex> lock(mLock);
2142
2143    if (validate_apk_path(packageDir.c_str())) {
2144        return error("Invalid path " + packageDir);
2145    }
2146    if (delete_dir_contents_and_dir(packageDir) != 0) {
2147        return error("Failed to delete " + packageDir);
2148    }
2149    return ok();
2150}
2151
2152binder::Status InstalldNativeService::linkFile(const std::string& relativePath,
2153        const std::string& fromBase, const std::string& toBase) {
2154    ENFORCE_UID(AID_SYSTEM);
2155    std::lock_guard<std::recursive_mutex> lock(mLock);
2156
2157    const char* relative_path = relativePath.c_str();
2158    const char* from_base = fromBase.c_str();
2159    const char* to_base = toBase.c_str();
2160    char from_path[PKG_PATH_MAX];
2161    char to_path[PKG_PATH_MAX];
2162    snprintf(from_path, PKG_PATH_MAX, "%s/%s", from_base, relative_path);
2163    snprintf(to_path, PKG_PATH_MAX, "%s/%s", to_base, relative_path);
2164
2165    if (validate_apk_path_subdirs(from_path)) {
2166        return error(StringPrintf("Invalid from path %s", from_path));
2167    }
2168
2169    if (validate_apk_path_subdirs(to_path)) {
2170        return error(StringPrintf("Invalid to path %s", to_path));
2171    }
2172
2173    if (link(from_path, to_path) < 0) {
2174        return error(StringPrintf("Failed to link from %s to %s", from_path, to_path));
2175    }
2176
2177    return ok();
2178}
2179
2180binder::Status InstalldNativeService::moveAb(const std::string& apkPath,
2181        const std::string& instructionSet, const std::string& outputPath) {
2182    ENFORCE_UID(AID_SYSTEM);
2183    std::lock_guard<std::recursive_mutex> lock(mLock);
2184
2185    const char* apk_path = apkPath.c_str();
2186    const char* instruction_set = instructionSet.c_str();
2187    const char* oat_dir = outputPath.c_str();
2188
2189    bool success = move_ab(apk_path, instruction_set, oat_dir);
2190    return success ? ok() : error();
2191}
2192
2193binder::Status InstalldNativeService::deleteOdex(const std::string& apkPath,
2194        const std::string& instructionSet, const std::string& outputPath) {
2195    ENFORCE_UID(AID_SYSTEM);
2196    std::lock_guard<std::recursive_mutex> lock(mLock);
2197
2198    const char* apk_path = apkPath.c_str();
2199    const char* instruction_set = instructionSet.c_str();
2200    const char* oat_dir = outputPath.c_str();
2201
2202    bool res = delete_odex(apk_path, instruction_set, oat_dir);
2203    return res ? ok() : error();
2204}
2205
2206binder::Status InstalldNativeService::reconcileSecondaryDexFile(
2207        const std::string& dexPath, const std::string& packageName, int32_t uid,
2208        const std::vector<std::string>& isas, const std::unique_ptr<std::string>& volumeUuid,
2209        int32_t storage_flag, bool* _aidl_return) {
2210    ENFORCE_UID(AID_SYSTEM);
2211    CHECK_ARGUMENT_UUID(volumeUuid);
2212    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
2213
2214    std::lock_guard<std::recursive_mutex> lock(mLock);
2215    bool result = android::installd::reconcile_secondary_dex_file(
2216            dexPath, packageName, uid, isas, volumeUuid, storage_flag, _aidl_return);
2217    return result ? ok() : error();
2218}
2219
2220binder::Status InstalldNativeService::invalidateMounts() {
2221    ENFORCE_UID(AID_SYSTEM);
2222    std::lock_guard<std::recursive_mutex> lock(mQuotaDevicesLock);
2223
2224    mQuotaDevices.clear();
2225
2226    std::ifstream in("/proc/mounts");
2227    if (!in.is_open()) {
2228        return error("Failed to read mounts");
2229    }
2230
2231    std::string source;
2232    std::string target;
2233    std::string ignored;
2234    struct dqblk dq;
2235    while (!in.eof()) {
2236        std::getline(in, source, ' ');
2237        std::getline(in, target, ' ');
2238        std::getline(in, ignored);
2239
2240        if (source.compare(0, 11, "/dev/block/") == 0) {
2241            if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), source.c_str(), 0,
2242                    reinterpret_cast<char*>(&dq)) == 0) {
2243                LOG(DEBUG) << "Found " << source << " with quota";
2244                mQuotaDevices[target] = source;
2245
2246                // ext4 only enables DQUOT_USAGE_ENABLED by default, so we
2247                // need to kick it again to enable DQUOT_LIMITS_ENABLED.
2248                if (quotactl(QCMD(Q_QUOTAON, USRQUOTA), source.c_str(), QFMT_VFS_V1, nullptr) != 0
2249                        && errno != EBUSY) {
2250                    PLOG(ERROR) << "Failed to enable USRQUOTA on " << source;
2251                }
2252                if (quotactl(QCMD(Q_QUOTAON, GRPQUOTA), source.c_str(), QFMT_VFS_V1, nullptr) != 0
2253                        && errno != EBUSY) {
2254                    PLOG(ERROR) << "Failed to enable GRPQUOTA on " << source;
2255                }
2256            }
2257        }
2258    }
2259    return ok();
2260}
2261
2262std::string InstalldNativeService::findQuotaDeviceForUuid(
2263        const std::unique_ptr<std::string>& uuid) {
2264    std::lock_guard<std::recursive_mutex> lock(mQuotaDevicesLock);
2265    auto path = create_data_path(uuid ? uuid->c_str() : nullptr);
2266    return mQuotaDevices[path];
2267}
2268
2269binder::Status InstalldNativeService::isQuotaSupported(
2270        const std::unique_ptr<std::string>& volumeUuid, bool* _aidl_return) {
2271    *_aidl_return = !findQuotaDeviceForUuid(volumeUuid).empty();
2272    return ok();
2273}
2274
2275}  // namespace installd
2276}  // namespace android
2277