InstalldNativeService.cpp revision 7a9059e8ed297a8e025c843b632c7c8571682b46
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#include <errno.h>
20#include <inttypes.h>
21#include <fstream>
22#include <fts.h>
23#include <regex>
24#include <stdlib.h>
25#include <string.h>
26#include <sys/capability.h>
27#include <sys/file.h>
28#include <sys/resource.h>
29#include <sys/quota.h>
30#include <sys/stat.h>
31#include <sys/statvfs.h>
32#include <sys/types.h>
33#include <sys/wait.h>
34#include <sys/xattr.h>
35#include <unistd.h>
36
37#include <android-base/logging.h>
38#include <android-base/stringprintf.h>
39#include <android-base/strings.h>
40#include <android-base/unique_fd.h>
41#include <cutils/fs.h>
42#include <cutils/properties.h>
43#include <cutils/sched_policy.h>
44#include <log/log.h>               // TODO: Move everything to base/logging.
45#include <logwrap/logwrap.h>
46#include <private/android_filesystem_config.h>
47#include <selinux/android.h>
48#include <system/thread_defs.h>
49
50#include "dexopt.h"
51#include "globals.h"
52#include "installd_deps.h"
53#include "otapreopt_utils.h"
54#include "utils.h"
55
56#include "MatchExtensionGen.h"
57
58#ifndef LOG_TAG
59#define LOG_TAG "installd"
60#endif
61
62using android::base::StringPrintf;
63
64namespace android {
65namespace installd {
66
67static constexpr const char* kCpPath = "/system/bin/cp";
68static constexpr const char* kXattrDefault = "user.default";
69
70static constexpr const char* PKG_LIB_POSTFIX = "/lib";
71static constexpr const char* CACHE_DIR_POSTFIX = "/cache";
72static constexpr const char* CODE_CACHE_DIR_POSTFIX = "/code_cache";
73
74static constexpr const char* IDMAP_PREFIX = "/data/resource-cache/";
75static constexpr const char* IDMAP_SUFFIX = "@idmap";
76
77// NOTE: keep in sync with StorageManager
78static constexpr int FLAG_STORAGE_DE = 1 << 0;
79static constexpr int FLAG_STORAGE_CE = 1 << 1;
80
81// NOTE: keep in sync with Installer
82static constexpr int FLAG_CLEAR_CACHE_ONLY = 1 << 8;
83static constexpr int FLAG_CLEAR_CODE_CACHE_ONLY = 1 << 9;
84static constexpr int FLAG_USE_QUOTA = 1 << 12;
85
86#define MIN_RESTRICTED_HOME_SDK_VERSION 24 // > M
87namespace {
88
89constexpr const char* kDump = "android.permission.DUMP";
90
91static binder::Status ok() {
92    return binder::Status::ok();
93}
94
95static binder::Status exception(uint32_t code, const std::string& msg) {
96    return binder::Status::fromExceptionCode(code, String8(msg.c_str()));
97}
98
99static binder::Status error() {
100    return binder::Status::fromServiceSpecificError(errno);
101}
102
103static binder::Status error(const std::string& msg) {
104    PLOG(ERROR) << msg;
105    return binder::Status::fromServiceSpecificError(errno, String8(msg.c_str()));
106}
107
108static binder::Status error(uint32_t code, const std::string& msg) {
109    LOG(ERROR) << msg << " (" << code << ")";
110    return binder::Status::fromServiceSpecificError(code, String8(msg.c_str()));
111}
112
113binder::Status checkPermission(const char* permission) {
114    pid_t pid;
115    uid_t uid;
116
117    if (checkCallingPermission(String16(permission), reinterpret_cast<int32_t*>(&pid),
118            reinterpret_cast<int32_t*>(&uid))) {
119        return ok();
120    } else {
121        return exception(binder::Status::EX_SECURITY,
122                StringPrintf("UID %d / PID %d lacks permission %s", uid, pid, permission));
123    }
124}
125
126binder::Status checkUid(uid_t expectedUid) {
127    uid_t uid = IPCThreadState::self()->getCallingUid();
128    if (uid == expectedUid || uid == AID_ROOT) {
129        return ok();
130    } else {
131        return exception(binder::Status::EX_SECURITY,
132                StringPrintf("UID %d is not expected UID %d", uid, expectedUid));
133    }
134}
135
136binder::Status checkArgumentUuid(const std::unique_ptr<std::string>& uuid) {
137    if (!uuid || is_valid_filename(*uuid)) {
138        return ok();
139    } else {
140        return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
141                StringPrintf("UUID %s is malformed", uuid->c_str()));
142    }
143}
144
145binder::Status checkArgumentPackageName(const std::string& packageName) {
146    if (is_valid_package_name(packageName.c_str())) {
147        return ok();
148    } else {
149        return exception(binder::Status::EX_ILLEGAL_ARGUMENT,
150                StringPrintf("Package name %s is malformed", packageName.c_str()));
151    }
152}
153
154#define ENFORCE_UID(uid) {                                  \
155    binder::Status status = checkUid((uid));                \
156    if (!status.isOk()) {                                   \
157        return status;                                      \
158    }                                                       \
159}
160
161#define CHECK_ARGUMENT_UUID(uuid) {                         \
162    binder::Status status = checkArgumentUuid((uuid));      \
163    if (!status.isOk()) {                                   \
164        return status;                                      \
165    }                                                       \
166}
167
168#define CHECK_ARGUMENT_PACKAGE_NAME(packageName) {          \
169    binder::Status status =                                 \
170            checkArgumentPackageName((packageName));        \
171    if (!status.isOk()) {                                   \
172        return status;                                      \
173    }                                                       \
174}
175
176}  // namespace
177
178status_t InstalldNativeService::start() {
179    IPCThreadState::self()->disableBackgroundScheduling(true);
180    status_t ret = BinderService<InstalldNativeService>::publish();
181    if (ret != android::OK) {
182        return ret;
183    }
184    sp<ProcessState> ps(ProcessState::self());
185    ps->startThreadPool();
186    ps->giveThreadPoolName();
187    return android::OK;
188}
189
190status_t InstalldNativeService::dump(int fd, const Vector<String16> & /* args */) {
191    const binder::Status dump_permission = checkPermission(kDump);
192    if (!dump_permission.isOk()) {
193        const String8 msg(dump_permission.toString8());
194        write(fd, msg.string(), msg.size());
195        return PERMISSION_DENIED;
196    }
197
198    std::string msg = "installd is happy\n";
199    write(fd, msg.c_str(), strlen(msg.c_str()));
200    return NO_ERROR;
201}
202
203/**
204 * Perform restorecon of the given path, but only perform recursive restorecon
205 * if the label of that top-level file actually changed.  This can save us
206 * significant time by avoiding no-op traversals of large filesystem trees.
207 */
208static int restorecon_app_data_lazy(const std::string& path, const std::string& seInfo, uid_t uid,
209        bool existing) {
210    int res = 0;
211    char* before = nullptr;
212    char* after = nullptr;
213
214    // Note that SELINUX_ANDROID_RESTORECON_DATADATA flag is set by
215    // libselinux. Not needed here.
216
217    if (lgetfilecon(path.c_str(), &before) < 0) {
218        PLOG(ERROR) << "Failed before getfilecon for " << path;
219        goto fail;
220    }
221    if (selinux_android_restorecon_pkgdir(path.c_str(), seInfo.c_str(), uid, 0) < 0) {
222        PLOG(ERROR) << "Failed top-level restorecon for " << path;
223        goto fail;
224    }
225    if (lgetfilecon(path.c_str(), &after) < 0) {
226        PLOG(ERROR) << "Failed after getfilecon for " << path;
227        goto fail;
228    }
229
230    // If the initial top-level restorecon above changed the label, then go
231    // back and restorecon everything recursively
232    if (strcmp(before, after)) {
233        if (existing) {
234            LOG(DEBUG) << "Detected label change from " << before << " to " << after << " at "
235                    << path << "; running recursive restorecon";
236        }
237        if (selinux_android_restorecon_pkgdir(path.c_str(), seInfo.c_str(), uid,
238                SELINUX_ANDROID_RESTORECON_RECURSE) < 0) {
239            PLOG(ERROR) << "Failed recursive restorecon for " << path;
240            goto fail;
241        }
242    }
243
244    goto done;
245fail:
246    res = -1;
247done:
248    free(before);
249    free(after);
250    return res;
251}
252
253static int restorecon_app_data_lazy(const std::string& parent, const char* name,
254        const std::string& seInfo, uid_t uid, bool existing) {
255    return restorecon_app_data_lazy(StringPrintf("%s/%s", parent.c_str(), name), seInfo, uid,
256            existing);
257}
258
259static int prepare_app_dir(const std::string& path, mode_t target_mode, uid_t uid) {
260    if (fs_prepare_dir_strict(path.c_str(), target_mode, uid, uid) != 0) {
261        PLOG(ERROR) << "Failed to prepare " << path;
262        return -1;
263    }
264    return 0;
265}
266
267/**
268 * Prepare an app cache directory, which offers to fix-up the GID and
269 * directory mode flags during a platform upgrade.
270 */
271static int prepare_app_cache_dir(const std::string& parent, const char* name, mode_t target_mode,
272        uid_t uid, gid_t gid) {
273    auto path = StringPrintf("%s/%s", parent.c_str(), name);
274    struct stat st;
275    if (stat(path.c_str(), &st) != 0) {
276        if (errno == ENOENT) {
277            // This is fine, just create it
278            if (fs_prepare_dir_strict(path.c_str(), target_mode, uid, gid) != 0) {
279                PLOG(ERROR) << "Failed to prepare " << path;
280                return -1;
281            } else {
282                return 0;
283            }
284        } else {
285            PLOG(ERROR) << "Failed to stat " << path;
286            return -1;
287        }
288    }
289
290    mode_t actual_mode = st.st_mode & (S_IRWXU | S_IRWXG | S_IRWXO | S_ISGID);
291    if (st.st_uid != uid) {
292        // Mismatched UID is real trouble; we can't recover
293        LOG(ERROR) << "Mismatched UID at " << path << ": found " << st.st_uid
294                << " but expected " << uid;
295        return -1;
296    } else if (st.st_gid == gid && actual_mode == target_mode) {
297        // Everything looks good!
298        return 0;
299    }
300
301    // Directory is owned correctly, but GID or mode mismatch means it's
302    // probably a platform upgrade so we need to fix them
303    FTS *fts;
304    FTSENT *p;
305    char *argv[] = { (char*) path.c_str(), nullptr };
306    if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_XDEV, NULL))) {
307        PLOG(ERROR) << "Failed to fts_open " << path;
308        return -1;
309    }
310    while ((p = fts_read(fts)) != NULL) {
311        switch (p->fts_info) {
312        case FTS_DP:
313            if (chmod(p->fts_accpath, target_mode) != 0) {
314                PLOG(WARNING) << "Failed to chmod " << p->fts_path;
315            }
316            // Intentional fall through to also set GID
317        case FTS_F:
318            if (chown(p->fts_accpath, -1, gid) != 0) {
319                PLOG(WARNING) << "Failed to chown " << p->fts_path;
320            }
321            break;
322        case FTS_SL:
323        case FTS_SLNONE:
324            if (lchown(p->fts_accpath, -1, gid) != 0) {
325                PLOG(WARNING) << "Failed to chown " << p->fts_path;
326            }
327            break;
328        }
329    }
330    fts_close(fts);
331    return 0;
332}
333
334binder::Status InstalldNativeService::createAppData(const std::unique_ptr<std::string>& uuid,
335        const std::string& packageName, int32_t userId, int32_t flags, int32_t appId,
336        const std::string& seInfo, int32_t targetSdkVersion, int64_t* _aidl_return) {
337    ENFORCE_UID(AID_SYSTEM);
338    CHECK_ARGUMENT_UUID(uuid);
339    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
340    std::lock_guard<std::recursive_mutex> lock(mLock);
341
342    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
343    const char* pkgname = packageName.c_str();
344
345    // Assume invalid inode unless filled in below
346    if (_aidl_return != nullptr) *_aidl_return = -1;
347
348    int32_t uid = multiuser_get_uid(userId, appId);
349    int32_t cacheGid = multiuser_get_cache_gid(userId, appId);
350    mode_t targetMode = targetSdkVersion >= MIN_RESTRICTED_HOME_SDK_VERSION ? 0700 : 0751;
351
352    // If UID doesn't have a specific cache GID, use UID value
353    if (cacheGid == -1) {
354        cacheGid = uid;
355    }
356
357    if (flags & FLAG_STORAGE_CE) {
358        auto path = create_data_user_ce_package_path(uuid_, userId, pkgname);
359        bool existing = (access(path.c_str(), F_OK) == 0);
360
361        if (prepare_app_dir(path, targetMode, uid) ||
362                prepare_app_cache_dir(path, "cache", 02771, uid, cacheGid) ||
363                prepare_app_cache_dir(path, "code_cache", 02771, uid, cacheGid)) {
364            return error("Failed to prepare " + path);
365        }
366
367        // Consider restorecon over contents if label changed
368        if (restorecon_app_data_lazy(path, seInfo, uid, existing) ||
369                restorecon_app_data_lazy(path, "cache", seInfo, uid, existing) ||
370                restorecon_app_data_lazy(path, "code_cache", seInfo, uid, existing)) {
371            return error("Failed to restorecon " + path);
372        }
373
374        // Remember inode numbers of cache directories so that we can clear
375        // contents while CE storage is locked
376        if (write_path_inode(path, "cache", kXattrInodeCache) ||
377                write_path_inode(path, "code_cache", kXattrInodeCodeCache)) {
378            return error("Failed to write_path_inode for " + path);
379        }
380
381        // And return the CE inode of the top-level data directory so we can
382        // clear contents while CE storage is locked
383        if ((_aidl_return != nullptr)
384                && get_path_inode(path, reinterpret_cast<ino_t*>(_aidl_return)) != 0) {
385            return error("Failed to get_path_inode for " + path);
386        }
387    }
388    if (flags & FLAG_STORAGE_DE) {
389        auto path = create_data_user_de_package_path(uuid_, userId, pkgname);
390        bool existing = (access(path.c_str(), F_OK) == 0);
391
392        if (prepare_app_dir(path, targetMode, uid) ||
393                prepare_app_cache_dir(path, "cache", 02771, uid, cacheGid) ||
394                prepare_app_cache_dir(path, "code_cache", 02771, uid, cacheGid)) {
395            return error("Failed to prepare " + path);
396        }
397
398        // Consider restorecon over contents if label changed
399        if (restorecon_app_data_lazy(path, seInfo, uid, existing)) {
400            return error("Failed to restorecon " + path);
401        }
402
403        if (property_get_bool("dalvik.vm.usejitprofiles", false)) {
404            const std::string profile_path = create_data_user_profile_package_path(userId, pkgname);
405            // read-write-execute only for the app user.
406            if (fs_prepare_dir_strict(profile_path.c_str(), 0700, uid, uid) != 0) {
407                return error("Failed to prepare " + profile_path);
408            }
409            std::string profile_file = create_primary_profile(profile_path);
410            // read-write only for the app user.
411            if (fs_prepare_file_strict(profile_file.c_str(), 0600, uid, uid) != 0) {
412                return error("Failed to prepare " + profile_path);
413            }
414            const std::string ref_profile_path = create_data_ref_profile_package_path(pkgname);
415            // dex2oat/profman runs under the shared app gid and it needs to read/write reference
416            // profiles.
417            int shared_app_gid = multiuser_get_shared_gid(0, appId);
418            if ((shared_app_gid != -1) && fs_prepare_dir_strict(
419                    ref_profile_path.c_str(), 0700, shared_app_gid, shared_app_gid) != 0) {
420                return error("Failed to prepare " + ref_profile_path);
421            }
422        }
423    }
424    return ok();
425}
426
427binder::Status InstalldNativeService::migrateAppData(const std::unique_ptr<std::string>& uuid,
428        const std::string& packageName, int32_t userId, int32_t flags) {
429    ENFORCE_UID(AID_SYSTEM);
430    CHECK_ARGUMENT_UUID(uuid);
431    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
432    std::lock_guard<std::recursive_mutex> lock(mLock);
433
434    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
435    const char* pkgname = packageName.c_str();
436
437    // This method only exists to upgrade system apps that have requested
438    // forceDeviceEncrypted, so their default storage always lives in a
439    // consistent location.  This only works on non-FBE devices, since we
440    // never want to risk exposing data on a device with real CE/DE storage.
441
442    auto ce_path = create_data_user_ce_package_path(uuid_, userId, pkgname);
443    auto de_path = create_data_user_de_package_path(uuid_, userId, pkgname);
444
445    // If neither directory is marked as default, assume CE is default
446    if (getxattr(ce_path.c_str(), kXattrDefault, nullptr, 0) == -1
447            && getxattr(de_path.c_str(), kXattrDefault, nullptr, 0) == -1) {
448        if (setxattr(ce_path.c_str(), kXattrDefault, nullptr, 0, 0) != 0) {
449            return error("Failed to mark default storage " + ce_path);
450        }
451    }
452
453    // Migrate default data location if needed
454    auto target = (flags & FLAG_STORAGE_DE) ? de_path : ce_path;
455    auto source = (flags & FLAG_STORAGE_DE) ? ce_path : de_path;
456
457    if (getxattr(target.c_str(), kXattrDefault, nullptr, 0) == -1) {
458        LOG(WARNING) << "Requested default storage " << target
459                << " is not active; migrating from " << source;
460        if (delete_dir_contents_and_dir(target) != 0) {
461            return error("Failed to delete " + target);
462        }
463        if (rename(source.c_str(), target.c_str()) != 0) {
464            return error("Failed to rename " + source + " to " + target);
465        }
466    }
467
468    return ok();
469}
470
471
472binder::Status InstalldNativeService::clearAppProfiles(const std::string& packageName) {
473    ENFORCE_UID(AID_SYSTEM);
474    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
475    std::lock_guard<std::recursive_mutex> lock(mLock);
476
477    const char* pkgname = packageName.c_str();
478    binder::Status res = ok();
479    if (!clear_reference_profile(pkgname)) {
480        res = error("Failed to clear reference profile for " + packageName);
481    }
482    if (!clear_current_profiles(pkgname)) {
483        res = error("Failed to clear current profiles for " + packageName);
484    }
485    return res;
486}
487
488binder::Status InstalldNativeService::clearAppData(const std::unique_ptr<std::string>& uuid,
489        const std::string& packageName, int32_t userId, int32_t flags, int64_t ceDataInode) {
490    ENFORCE_UID(AID_SYSTEM);
491    CHECK_ARGUMENT_UUID(uuid);
492    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
493    std::lock_guard<std::recursive_mutex> lock(mLock);
494
495    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
496    const char* pkgname = packageName.c_str();
497
498    binder::Status res = ok();
499    if (flags & FLAG_STORAGE_CE) {
500        auto path = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInode);
501        if (flags & FLAG_CLEAR_CACHE_ONLY) {
502            path = read_path_inode(path, "cache", kXattrInodeCache);
503        } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
504            path = read_path_inode(path, "code_cache", kXattrInodeCodeCache);
505        }
506        if (access(path.c_str(), F_OK) == 0) {
507            if (delete_dir_contents(path) != 0) {
508                res = error("Failed to delete contents of " + path);
509            }
510        }
511    }
512    if (flags & FLAG_STORAGE_DE) {
513        std::string suffix = "";
514        bool only_cache = false;
515        if (flags & FLAG_CLEAR_CACHE_ONLY) {
516            suffix = CACHE_DIR_POSTFIX;
517            only_cache = true;
518        } else if (flags & FLAG_CLEAR_CODE_CACHE_ONLY) {
519            suffix = CODE_CACHE_DIR_POSTFIX;
520            only_cache = true;
521        }
522
523        auto path = create_data_user_de_package_path(uuid_, userId, pkgname) + suffix;
524        if (access(path.c_str(), F_OK) == 0) {
525            if (delete_dir_contents(path) != 0) {
526                res = error("Failed to delete contents of " + path);
527            }
528        }
529        if (!only_cache) {
530            if (!clear_current_profile(pkgname, userId)) {
531                res = error("Failed to clear current profile for " + packageName);
532            }
533        }
534    }
535    return res;
536}
537
538static int destroy_app_reference_profile(const char *pkgname) {
539    return delete_dir_contents_and_dir(
540        create_data_ref_profile_package_path(pkgname),
541        /*ignore_if_missing*/ true);
542}
543
544static int destroy_app_current_profiles(const char *pkgname, userid_t userid) {
545    return delete_dir_contents_and_dir(
546        create_data_user_profile_package_path(userid, pkgname),
547        /*ignore_if_missing*/ true);
548}
549
550binder::Status InstalldNativeService::destroyAppProfiles(const std::string& packageName) {
551    ENFORCE_UID(AID_SYSTEM);
552    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
553    std::lock_guard<std::recursive_mutex> lock(mLock);
554
555    const char* pkgname = packageName.c_str();
556    binder::Status res = ok();
557    std::vector<userid_t> users = get_known_users(/*volume_uuid*/ nullptr);
558    for (auto user : users) {
559        if (destroy_app_current_profiles(pkgname, user) != 0) {
560            res = error("Failed to destroy current profiles for " + packageName);
561        }
562    }
563    if (destroy_app_reference_profile(pkgname) != 0) {
564        res = error("Failed to destroy reference profile for " + packageName);
565    }
566    return res;
567}
568
569binder::Status InstalldNativeService::destroyAppData(const std::unique_ptr<std::string>& uuid,
570        const std::string& packageName, int32_t userId, int32_t flags, int64_t ceDataInode) {
571    ENFORCE_UID(AID_SYSTEM);
572    CHECK_ARGUMENT_UUID(uuid);
573    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
574    std::lock_guard<std::recursive_mutex> lock(mLock);
575
576    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
577    const char* pkgname = packageName.c_str();
578
579    binder::Status res = ok();
580    if (flags & FLAG_STORAGE_CE) {
581        auto path = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInode);
582        if (delete_dir_contents_and_dir(path) != 0) {
583            res = error("Failed to delete " + path);
584        }
585    }
586    if (flags & FLAG_STORAGE_DE) {
587        auto path = create_data_user_de_package_path(uuid_, userId, pkgname);
588        if (delete_dir_contents_and_dir(path) != 0) {
589            res = error("Failed to delete " + path);
590        }
591        destroy_app_current_profiles(pkgname, userId);
592        // TODO(calin): If the package is still installed by other users it's probably
593        // beneficial to keep the reference profile around.
594        // Verify if it's ok to do that.
595        destroy_app_reference_profile(pkgname);
596    }
597    return res;
598}
599
600binder::Status InstalldNativeService::moveCompleteApp(const std::unique_ptr<std::string>& fromUuid,
601        const std::unique_ptr<std::string>& toUuid, const std::string& packageName,
602        const std::string& dataAppName, int32_t appId, const std::string& seInfo,
603        int32_t targetSdkVersion) {
604    ENFORCE_UID(AID_SYSTEM);
605    CHECK_ARGUMENT_UUID(fromUuid);
606    CHECK_ARGUMENT_UUID(toUuid);
607    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
608    std::lock_guard<std::recursive_mutex> lock(mLock);
609
610    const char* from_uuid = fromUuid ? fromUuid->c_str() : nullptr;
611    const char* to_uuid = toUuid ? toUuid->c_str() : nullptr;
612    const char* package_name = packageName.c_str();
613    const char* data_app_name = dataAppName.c_str();
614
615    binder::Status res = ok();
616    std::vector<userid_t> users = get_known_users(from_uuid);
617
618    // Copy app
619    {
620        auto from = create_data_app_package_path(from_uuid, data_app_name);
621        auto to = create_data_app_package_path(to_uuid, data_app_name);
622        auto to_parent = create_data_app_path(to_uuid);
623
624        char *argv[] = {
625            (char*) kCpPath,
626            (char*) "-F", /* delete any existing destination file first (--remove-destination) */
627            (char*) "-p", /* preserve timestamps, ownership, and permissions */
628            (char*) "-R", /* recurse into subdirectories (DEST must be a directory) */
629            (char*) "-P", /* Do not follow symlinks [default] */
630            (char*) "-d", /* don't dereference symlinks */
631            (char*) from.c_str(),
632            (char*) to_parent.c_str()
633        };
634
635        LOG(DEBUG) << "Copying " << from << " to " << to;
636        int rc = android_fork_execvp(ARRAY_SIZE(argv), argv, NULL, false, true);
637        if (rc != 0) {
638            res = error(rc, "Failed copying " + from + " to " + to);
639            goto fail;
640        }
641
642        if (selinux_android_restorecon(to.c_str(), SELINUX_ANDROID_RESTORECON_RECURSE) != 0) {
643            res = error("Failed to restorecon " + to);
644            goto fail;
645        }
646    }
647
648    // Copy private data for all known users
649    for (auto user : users) {
650
651        // Data source may not exist for all users; that's okay
652        auto from_ce = create_data_user_ce_package_path(from_uuid, user, package_name);
653        if (access(from_ce.c_str(), F_OK) != 0) {
654            LOG(INFO) << "Missing source " << from_ce;
655            continue;
656        }
657
658        if (!createAppData(toUuid, packageName, user, FLAG_STORAGE_CE | FLAG_STORAGE_DE, appId,
659                seInfo, targetSdkVersion, nullptr).isOk()) {
660            res = error("Failed to create package target");
661            goto fail;
662        }
663
664        char *argv[] = {
665            (char*) kCpPath,
666            (char*) "-F", /* delete any existing destination file first (--remove-destination) */
667            (char*) "-p", /* preserve timestamps, ownership, and permissions */
668            (char*) "-R", /* recurse into subdirectories (DEST must be a directory) */
669            (char*) "-P", /* Do not follow symlinks [default] */
670            (char*) "-d", /* don't dereference symlinks */
671            nullptr,
672            nullptr
673        };
674
675        {
676            auto from = create_data_user_de_package_path(from_uuid, user, package_name);
677            auto to = create_data_user_de_path(to_uuid, user);
678            argv[6] = (char*) from.c_str();
679            argv[7] = (char*) to.c_str();
680
681            LOG(DEBUG) << "Copying " << from << " to " << to;
682            int rc = android_fork_execvp(ARRAY_SIZE(argv), argv, NULL, false, true);
683            if (rc != 0) {
684                res = error(rc, "Failed copying " + from + " to " + to);
685                goto fail;
686            }
687        }
688        {
689            auto from = create_data_user_ce_package_path(from_uuid, user, package_name);
690            auto to = create_data_user_ce_path(to_uuid, user);
691            argv[6] = (char*) from.c_str();
692            argv[7] = (char*) to.c_str();
693
694            LOG(DEBUG) << "Copying " << from << " to " << to;
695            int rc = android_fork_execvp(ARRAY_SIZE(argv), argv, NULL, false, true);
696            if (rc != 0) {
697                res = error(rc, "Failed copying " + from + " to " + to);
698                goto fail;
699            }
700        }
701
702        if (!restoreconAppData(toUuid, packageName, user, FLAG_STORAGE_CE | FLAG_STORAGE_DE,
703                appId, seInfo).isOk()) {
704            res = error("Failed to restorecon");
705            goto fail;
706        }
707    }
708
709    // We let the framework scan the new location and persist that before
710    // deleting the data in the old location; this ordering ensures that
711    // we can recover from things like battery pulls.
712    return ok();
713
714fail:
715    // Nuke everything we might have already copied
716    {
717        auto to = create_data_app_package_path(to_uuid, data_app_name);
718        if (delete_dir_contents(to.c_str(), 1, NULL) != 0) {
719            LOG(WARNING) << "Failed to rollback " << to;
720        }
721    }
722    for (auto user : users) {
723        {
724            auto to = create_data_user_de_package_path(to_uuid, user, package_name);
725            if (delete_dir_contents(to.c_str(), 1, NULL) != 0) {
726                LOG(WARNING) << "Failed to rollback " << to;
727            }
728        }
729        {
730            auto to = create_data_user_ce_package_path(to_uuid, user, package_name);
731            if (delete_dir_contents(to.c_str(), 1, NULL) != 0) {
732                LOG(WARNING) << "Failed to rollback " << to;
733            }
734        }
735    }
736    return res;
737}
738
739binder::Status InstalldNativeService::createUserData(const std::unique_ptr<std::string>& uuid,
740        int32_t userId, int32_t userSerial ATTRIBUTE_UNUSED, int32_t flags) {
741    ENFORCE_UID(AID_SYSTEM);
742    CHECK_ARGUMENT_UUID(uuid);
743    std::lock_guard<std::recursive_mutex> lock(mLock);
744
745    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
746    if (flags & FLAG_STORAGE_DE) {
747        if (uuid_ == nullptr) {
748            if (ensure_config_user_dirs(userId) != 0) {
749                return error(StringPrintf("Failed to ensure dirs for %d", userId));
750            }
751        }
752    }
753    return ok();
754}
755
756binder::Status InstalldNativeService::destroyUserData(const std::unique_ptr<std::string>& uuid,
757        int32_t userId, int32_t flags) {
758    ENFORCE_UID(AID_SYSTEM);
759    CHECK_ARGUMENT_UUID(uuid);
760    std::lock_guard<std::recursive_mutex> lock(mLock);
761
762    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
763    binder::Status res = ok();
764    if (flags & FLAG_STORAGE_DE) {
765        auto path = create_data_user_de_path(uuid_, userId);
766        if (delete_dir_contents_and_dir(path, true) != 0) {
767            res = error("Failed to delete " + path);
768        }
769        if (uuid_ == nullptr) {
770            path = create_data_misc_legacy_path(userId);
771            if (delete_dir_contents_and_dir(path, true) != 0) {
772                res = error("Failed to delete " + path);
773            }
774            path = create_data_user_profile_path(userId);
775            if (delete_dir_contents_and_dir(path, true) != 0) {
776                res = error("Failed to delete " + path);
777            }
778        }
779    }
780    if (flags & FLAG_STORAGE_CE) {
781        auto path = create_data_user_ce_path(uuid_, userId);
782        if (delete_dir_contents_and_dir(path, true) != 0) {
783            res = error("Failed to delete " + path);
784        }
785        path = create_data_media_path(uuid_, userId);
786        if (delete_dir_contents_and_dir(path, true) != 0) {
787            res = error("Failed to delete " + path);
788        }
789    }
790    return res;
791}
792
793/* Try to ensure free_size bytes of storage are available.
794 * Returns 0 on success.
795 * This is rather simple-minded because doing a full LRU would
796 * be potentially memory-intensive, and without atime it would
797 * also require that apps constantly modify file metadata even
798 * when just reading from the cache, which is pretty awful.
799 */
800binder::Status InstalldNativeService::freeCache(const std::unique_ptr<std::string>& uuid,
801        int64_t freeStorageSize) {
802    ENFORCE_UID(AID_SYSTEM);
803    CHECK_ARGUMENT_UUID(uuid);
804    std::lock_guard<std::recursive_mutex> lock(mLock);
805
806    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
807    cache_t* cache;
808    int64_t avail;
809
810    auto data_path = create_data_path(uuid_);
811
812    avail = data_disk_free(data_path);
813    if (avail < 0) {
814        return error("Failed to determine free space for " + data_path);
815    }
816
817    ALOGI("free_cache(%" PRId64 ") avail %" PRId64 "\n", freeStorageSize, avail);
818    if (avail >= freeStorageSize) {
819        return ok();
820    }
821
822    cache = start_cache_collection();
823
824    auto users = get_known_users(uuid_);
825    for (auto user : users) {
826        add_cache_files(cache, create_data_user_ce_path(uuid_, user));
827        add_cache_files(cache, create_data_user_de_path(uuid_, user));
828        add_cache_files(cache,
829                StringPrintf("%s/Android/data", create_data_media_path(uuid_, user).c_str()));
830    }
831
832    clear_cache_files(data_path, cache, freeStorageSize);
833    finish_cache_collection(cache);
834
835    avail = data_disk_free(data_path);
836    if (avail >= freeStorageSize) {
837        return ok();
838    } else {
839        return error(StringPrintf("Failed to free up %" PRId64 " on %s; final free space %" PRId64,
840                freeStorageSize, data_path.c_str(), avail));
841    }
842}
843
844binder::Status InstalldNativeService::rmdex(const std::string& codePath,
845        const std::string& instructionSet) {
846    ENFORCE_UID(AID_SYSTEM);
847    std::lock_guard<std::recursive_mutex> lock(mLock);
848
849    char dex_path[PKG_PATH_MAX];
850
851    const char* path = codePath.c_str();
852    const char* instruction_set = instructionSet.c_str();
853
854    if (validate_apk_path(path) && validate_system_app_path(path)) {
855        return error("Invalid path " + codePath);
856    }
857
858    if (!create_cache_path(dex_path, path, instruction_set)) {
859        return error("Failed to create cache path for " + codePath);
860    }
861
862    ALOGV("unlink %s\n", dex_path);
863    if (unlink(dex_path) < 0) {
864        return error(StringPrintf("Failed to unlink %s", dex_path));
865    } else {
866        return ok();
867    }
868}
869
870struct stats {
871    int64_t codeSize;
872    int64_t dataSize;
873    int64_t cacheSize;
874};
875
876#if MEASURE_DEBUG
877static std::string toString(std::vector<int64_t> values) {
878    std::stringstream res;
879    res << "[";
880    for (size_t i = 0; i < values.size(); i++) {
881        res << values[i];
882        if (i < values.size() - 1) {
883            res << ",";
884        }
885    }
886    res << "]";
887    return res.str();
888}
889#endif
890
891static std::string findDeviceForUuid(const std::unique_ptr<std::string>& uuid) {
892    auto path = create_data_path(uuid ? uuid->c_str() : nullptr);
893    std::ifstream in("/proc/mounts");
894    if (!in.is_open()) {
895        PLOG(ERROR) << "Failed to read mounts";
896        return "";
897    }
898    std::string source;
899    std::string target;
900    while (!in.eof()) {
901        std::getline(in, source, ' ');
902        std::getline(in, target, ' ');
903        if (target == path) {
904            return source;
905        }
906        // Skip to next line
907        std::getline(in, source);
908    }
909    PLOG(ERROR) << "Failed to resolve block device for " << path;
910    return "";
911}
912
913static void collectQuotaStats(const std::string& device, int32_t userId,
914        int32_t appId, struct stats* stats, struct stats* extStats ATTRIBUTE_UNUSED) {
915    if (device.empty()) return;
916
917    struct dqblk dq;
918
919    uid_t uid = multiuser_get_uid(userId, appId);
920    if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), device.c_str(), uid,
921            reinterpret_cast<char*>(&dq)) != 0) {
922        if (errno != ESRCH) {
923            PLOG(ERROR) << "Failed to quotactl " << device << " for UID " << uid;
924        }
925    } else {
926#if MEASURE_DEBUG
927        LOG(DEBUG) << "quotactl() for UID " << uid << " " << dq.dqb_curspace;
928#endif
929        stats->dataSize += dq.dqb_curspace;
930    }
931
932    int cacheGid = multiuser_get_cache_gid(userId, appId);
933    if (cacheGid != -1) {
934        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), cacheGid,
935                reinterpret_cast<char*>(&dq)) != 0) {
936            if (errno != ESRCH) {
937                PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << cacheGid;
938            }
939        } else {
940#if MEASURE_DEBUG
941        LOG(DEBUG) << "quotactl() for GID " << cacheGid << " " << dq.dqb_curspace;
942#endif
943            stats->cacheSize += dq.dqb_curspace;
944        }
945    }
946
947    int sharedGid = multiuser_get_shared_app_gid(uid);
948    if (sharedGid != -1) {
949        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), sharedGid,
950                reinterpret_cast<char*>(&dq)) != 0) {
951            if (errno != ESRCH) {
952                PLOG(ERROR) << "Failed to quotactl " << device << " for GID " << sharedGid;
953            }
954        } else {
955#if MEASURE_DEBUG
956        LOG(DEBUG) << "quotactl() for GID " << sharedGid << " " << dq.dqb_curspace;
957#endif
958            stats->codeSize += dq.dqb_curspace;
959        }
960    }
961
962#if MEASURE_EXTERNAL
963    // TODO: measure using external GIDs
964#endif
965}
966
967static void collectQuotaStats(const std::unique_ptr<std::string>& uuid, int32_t userId,
968        int32_t appId, struct stats* stats, struct stats* extStats) {
969    collectQuotaStats(findDeviceForUuid(uuid), userId, appId, stats, extStats);
970}
971
972static void collectManualStats(const std::string& path, struct stats* stats) {
973    DIR *d;
974    int dfd;
975    struct dirent *de;
976    struct stat s;
977
978    d = opendir(path.c_str());
979    if (d == nullptr) {
980        if (errno != ENOENT) {
981            PLOG(WARNING) << "Failed to open " << path;
982        }
983        return;
984    }
985    dfd = dirfd(d);
986    while ((de = readdir(d))) {
987        const char *name = de->d_name;
988
989        int64_t size = 0;
990        if (fstatat(dfd, name, &s, AT_SYMLINK_NOFOLLOW) == 0) {
991            size = s.st_blocks * 512;
992        }
993
994        if (de->d_type == DT_DIR) {
995            if (!strcmp(name, ".")) {
996                // Don't recurse, but still count node size
997            } else if (!strcmp(name, "..")) {
998                // Don't recurse or count node size
999                continue;
1000            } else {
1001                // Measure all children nodes
1002                size = 0;
1003                calculate_tree_size(StringPrintf("%s/%s", path.c_str(), name), &size);
1004            }
1005
1006            if (!strcmp(name, "cache") || !strcmp(name, "code_cache")) {
1007                stats->cacheSize += size;
1008            }
1009        }
1010
1011        // Legacy symlink isn't owned by app
1012        if (de->d_type == DT_LNK && !strcmp(name, "lib")) {
1013            continue;
1014        }
1015
1016        // Everything found inside is considered data
1017        stats->dataSize += size;
1018    }
1019    closedir(d);
1020}
1021
1022static void collectManualStatsForUser(const std::string& path, struct stats* stats,
1023        bool exclude_apps = false) {
1024    DIR *d;
1025    int dfd;
1026    struct dirent *de;
1027    struct stat s;
1028
1029    d = opendir(path.c_str());
1030    if (d == nullptr) {
1031        if (errno != ENOENT) {
1032            PLOG(WARNING) << "Failed to open " << path;
1033        }
1034        return;
1035    }
1036    dfd = dirfd(d);
1037    while ((de = readdir(d))) {
1038        if (de->d_type == DT_DIR) {
1039            const char *name = de->d_name;
1040            if (fstatat(dfd, name, &s, AT_SYMLINK_NOFOLLOW) != 0) {
1041                continue;
1042            }
1043            if (!strcmp(name, ".") || !strcmp(name, "..")) {
1044                continue;
1045            } else if (exclude_apps && (s.st_uid >= AID_APP_START && s.st_uid <= AID_APP_END)) {
1046                continue;
1047            } else {
1048                collectManualStats(StringPrintf("%s/%s", path.c_str(), name), stats);
1049            }
1050        }
1051    }
1052    closedir(d);
1053}
1054
1055binder::Status InstalldNativeService::getAppSize(const std::unique_ptr<std::string>& uuid,
1056        const std::vector<std::string>& packageNames, int32_t userId, int32_t flags,
1057        int32_t appId, const std::vector<int64_t>& ceDataInodes,
1058        const std::vector<std::string>& codePaths, std::vector<int64_t>* _aidl_return) {
1059    ENFORCE_UID(AID_SYSTEM);
1060    CHECK_ARGUMENT_UUID(uuid);
1061    for (auto packageName : packageNames) {
1062        CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1063    }
1064    std::lock_guard<std::recursive_mutex> lock(mLock);
1065
1066    // When modifying this logic, always verify using tests:
1067    // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetAppSize
1068
1069#if MEASURE_DEBUG
1070    LOG(INFO) << "Measuring user " << userId << " app " << appId;
1071#endif
1072
1073    // Here's a summary of the common storage locations across the platform,
1074    // and how they're each tagged:
1075    //
1076    // /data/app/com.example                           UID system
1077    // /data/app/com.example/oat                       UID system
1078    // /data/user/0/com.example                        UID u0_a10      GID u0_a10
1079    // /data/user/0/com.example/cache                  UID u0_a10      GID u0_a10_cache
1080    // /data/media/0/foo.txt                           UID u0_media_rw
1081    // /data/media/0/bar.jpg                           UID u0_media_rw GID u0_media_image
1082    // /data/media/0/Android/data/com.example          UID u0_media_rw GID u0_a10_ext
1083    // /data/media/0/Android/data/com.example/cache    UID u0_media_rw GID u0_a10_ext_cache
1084    // /data/media/obb/com.example                     UID system
1085
1086    struct stats stats;
1087    struct stats extStats;
1088    memset(&stats, 0, sizeof(stats));
1089    memset(&extStats, 0, sizeof(extStats));
1090
1091    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1092
1093    for (auto packageName : packageNames) {
1094        auto obbCodePath = create_data_media_obb_path(uuid_, packageName.c_str());
1095        calculate_tree_size(obbCodePath, &extStats.codeSize);
1096    }
1097
1098    if (flags & FLAG_USE_QUOTA && appId >= AID_APP_START) {
1099        for (auto codePath : codePaths) {
1100            calculate_tree_size(codePath, &stats.codeSize, -1,
1101                    multiuser_get_shared_gid(userId, appId));
1102        }
1103
1104        collectQuotaStats(uuid, userId, appId, &stats, &extStats);
1105
1106    } else {
1107        for (auto codePath : codePaths) {
1108            calculate_tree_size(codePath, &stats.codeSize);
1109        }
1110
1111        for (size_t i = 0; i < packageNames.size(); i++) {
1112            const char* pkgname = packageNames[i].c_str();
1113
1114            auto cePath = create_data_user_ce_package_path(uuid_, userId, pkgname, ceDataInodes[i]);
1115            collectManualStats(cePath, &stats);
1116
1117            auto dePath = create_data_user_de_package_path(uuid_, userId, pkgname);
1118            collectManualStats(dePath, &stats);
1119
1120            auto userProfilePath = create_data_user_profile_package_path(userId, pkgname);
1121            calculate_tree_size(userProfilePath, &stats.dataSize);
1122
1123            auto refProfilePath = create_data_ref_profile_package_path(pkgname);
1124            calculate_tree_size(refProfilePath, &stats.codeSize);
1125
1126#if MEASURE_EXTERNAL
1127            auto extPath = create_data_media_package_path(uuid_, userId, pkgname, "data");
1128            collectManualStats(extPath, &extStats);
1129
1130            auto mediaPath = create_data_media_package_path(uuid_, userId, pkgname, "media");
1131            calculate_tree_size(mediaPath, &extStats.dataSize);
1132#endif
1133        }
1134
1135        int32_t sharedGid = multiuser_get_shared_gid(userId, appId);
1136        if (sharedGid != -1) {
1137            calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize,
1138                    sharedGid, -1);
1139        }
1140
1141        calculate_tree_size(create_data_misc_foreign_dex_path(userId), &stats.dataSize,
1142                multiuser_get_uid(userId, appId), -1);
1143    }
1144
1145    std::vector<int64_t> ret;
1146    ret.push_back(stats.codeSize);
1147    ret.push_back(stats.dataSize);
1148    ret.push_back(stats.cacheSize);
1149    ret.push_back(extStats.codeSize);
1150    ret.push_back(extStats.dataSize);
1151    ret.push_back(extStats.cacheSize);
1152#if MEASURE_DEBUG
1153    LOG(DEBUG) << "Final result " << toString(ret);
1154#endif
1155    *_aidl_return = ret;
1156    return ok();
1157}
1158
1159binder::Status InstalldNativeService::getUserSize(const std::unique_ptr<std::string>& uuid,
1160        int32_t userId, int32_t flags, const std::vector<int32_t>& appIds,
1161        std::vector<int64_t>* _aidl_return) {
1162    ENFORCE_UID(AID_SYSTEM);
1163    CHECK_ARGUMENT_UUID(uuid);
1164    std::lock_guard<std::recursive_mutex> lock(mLock);
1165
1166    // When modifying this logic, always verify using tests:
1167    // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetUserSize
1168
1169#if MEASURE_DEBUG
1170    LOG(INFO) << "Measuring user " << userId;
1171#endif
1172
1173    struct stats stats;
1174    struct stats extStats;
1175    memset(&stats, 0, sizeof(stats));
1176    memset(&extStats, 0, sizeof(extStats));
1177
1178    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1179
1180    auto obbPath = create_data_path(uuid_) + "/media/obb";
1181    calculate_tree_size(obbPath, &extStats.codeSize);
1182
1183    if (flags & FLAG_USE_QUOTA) {
1184        calculate_tree_size(create_data_app_path(uuid_), &stats.codeSize, -1, -1, true);
1185
1186        auto cePath = create_data_user_ce_path(uuid_, userId);
1187        collectManualStatsForUser(cePath, &stats, true);
1188
1189        auto dePath = create_data_user_de_path(uuid_, userId);
1190        collectManualStatsForUser(dePath, &stats, true);
1191
1192        auto userProfilePath = create_data_user_profile_path(userId);
1193        calculate_tree_size(userProfilePath, &stats.dataSize, -1, -1, true);
1194
1195        auto refProfilePath = create_data_ref_profile_path();
1196        calculate_tree_size(refProfilePath, &stats.codeSize, -1, -1, true);
1197
1198#if MEASURE_EXTERNAL
1199        // TODO: measure external storage paths
1200#endif
1201
1202        calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize,
1203                -1, -1, true);
1204
1205        calculate_tree_size(create_data_misc_foreign_dex_path(userId), &stats.dataSize,
1206                -1, -1, true);
1207
1208        auto device = findDeviceForUuid(uuid);
1209        for (auto appId : appIds) {
1210            if (appId >= AID_APP_START) {
1211                collectQuotaStats(device, userId, appId, &stats, &extStats);
1212#if MEASURE_DEBUG
1213                // Sleep to make sure we don't lose logs
1214                usleep(1);
1215#endif
1216            }
1217        }
1218    } else {
1219        calculate_tree_size(create_data_app_path(uuid_), &stats.codeSize);
1220
1221        auto cePath = create_data_user_ce_path(uuid_, userId);
1222        collectManualStatsForUser(cePath, &stats);
1223
1224        auto dePath = create_data_user_de_path(uuid_, userId);
1225        collectManualStatsForUser(dePath, &stats);
1226
1227        auto userProfilePath = create_data_user_profile_path(userId);
1228        calculate_tree_size(userProfilePath, &stats.dataSize);
1229
1230        auto refProfilePath = create_data_ref_profile_path();
1231        calculate_tree_size(refProfilePath, &stats.codeSize);
1232
1233#if MEASURE_EXTERNAL
1234        // TODO: measure external storage paths
1235#endif
1236
1237        calculate_tree_size(create_data_dalvik_cache_path(), &stats.codeSize);
1238
1239        calculate_tree_size(create_data_misc_foreign_dex_path(userId), &stats.dataSize);
1240    }
1241
1242    std::vector<int64_t> ret;
1243    ret.push_back(stats.codeSize);
1244    ret.push_back(stats.dataSize);
1245    ret.push_back(stats.cacheSize);
1246    ret.push_back(extStats.codeSize);
1247    ret.push_back(extStats.dataSize);
1248    ret.push_back(extStats.cacheSize);
1249#if MEASURE_DEBUG
1250    LOG(DEBUG) << "Final result " << toString(ret);
1251#endif
1252    *_aidl_return = ret;
1253    return ok();
1254}
1255
1256binder::Status InstalldNativeService::getExternalSize(const std::unique_ptr<std::string>& uuid,
1257        int32_t userId, int32_t flags, std::vector<int64_t>* _aidl_return) {
1258    ENFORCE_UID(AID_SYSTEM);
1259    CHECK_ARGUMENT_UUID(uuid);
1260    std::lock_guard<std::recursive_mutex> lock(mLock);
1261
1262    // When modifying this logic, always verify using tests:
1263    // runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/pm/InstallerTest.java -m testGetExternalSize
1264
1265#if MEASURE_DEBUG
1266    LOG(INFO) << "Measuring external " << userId;
1267#endif
1268
1269    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1270
1271    int64_t totalSize = 0;
1272    int64_t audioSize = 0;
1273    int64_t videoSize = 0;
1274    int64_t imageSize = 0;
1275
1276    if (flags & FLAG_USE_QUOTA) {
1277        struct dqblk dq;
1278
1279        auto device = findDeviceForUuid(uuid);
1280
1281        uid_t uid = multiuser_get_uid(userId, AID_MEDIA_RW);
1282        if (quotactl(QCMD(Q_GETQUOTA, USRQUOTA), device.c_str(), uid,
1283                reinterpret_cast<char*>(&dq)) != 0) {
1284            if (errno != ESRCH) {
1285                PLOG(ERROR) << "Failed to quotactl " << device << " for UID " << uid;
1286            }
1287        } else {
1288#if MEASURE_DEBUG
1289        LOG(DEBUG) << "quotactl() for UID " << uid << " " << dq.dqb_curspace;
1290#endif
1291            totalSize = dq.dqb_curspace;
1292        }
1293
1294        gid_t audioGid = multiuser_get_uid(userId, AID_MEDIA_AUDIO);
1295        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), audioGid,
1296                reinterpret_cast<char*>(&dq)) == 0) {
1297#if MEASURE_DEBUG
1298        LOG(DEBUG) << "quotactl() for GID " << audioGid << " " << dq.dqb_curspace;
1299#endif
1300            audioSize = dq.dqb_curspace;
1301        }
1302        gid_t videoGid = multiuser_get_uid(userId, AID_MEDIA_VIDEO);
1303        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), videoGid,
1304                reinterpret_cast<char*>(&dq)) == 0) {
1305#if MEASURE_DEBUG
1306        LOG(DEBUG) << "quotactl() for GID " << videoGid << " " << dq.dqb_curspace;
1307#endif
1308            videoSize = dq.dqb_curspace;
1309        }
1310        gid_t imageGid = multiuser_get_uid(userId, AID_MEDIA_IMAGE);
1311        if (quotactl(QCMD(Q_GETQUOTA, GRPQUOTA), device.c_str(), imageGid,
1312                reinterpret_cast<char*>(&dq)) == 0) {
1313#if MEASURE_DEBUG
1314        LOG(DEBUG) << "quotactl() for GID " << imageGid << " " << dq.dqb_curspace;
1315#endif
1316            imageSize = dq.dqb_curspace;
1317        }
1318    } else {
1319        FTS *fts;
1320        FTSENT *p;
1321        auto path = create_data_media_path(uuid_, userId);
1322        char *argv[] = { (char*) path.c_str(), nullptr };
1323        if (!(fts = fts_open(argv, FTS_PHYSICAL | FTS_XDEV, NULL))) {
1324            return error("Failed to fts_open " + path);
1325        }
1326        while ((p = fts_read(fts)) != NULL) {
1327            char* ext;
1328            int64_t size = (p->fts_statp->st_blocks * 512);
1329            switch (p->fts_info) {
1330            case FTS_F:
1331                // Only categorize files not belonging to apps
1332                if (p->fts_statp->st_gid < AID_APP_START) {
1333                    ext = strrchr(p->fts_name, '.');
1334                    if (ext != nullptr) {
1335                        switch (MatchExtension(++ext)) {
1336                        case AID_MEDIA_AUDIO: audioSize += size; break;
1337                        case AID_MEDIA_VIDEO: videoSize += size; break;
1338                        case AID_MEDIA_IMAGE: imageSize += size; break;
1339                        }
1340                    }
1341                }
1342                // Fall through to always count against total
1343            case FTS_D:
1344            case FTS_DEFAULT:
1345            case FTS_SL:
1346            case FTS_SLNONE:
1347                totalSize += size;
1348                break;
1349            }
1350        }
1351        fts_close(fts);
1352    }
1353
1354    std::vector<int64_t> ret;
1355    ret.push_back(totalSize);
1356    ret.push_back(audioSize);
1357    ret.push_back(videoSize);
1358    ret.push_back(imageSize);
1359#if MEASURE_DEBUG
1360    LOG(DEBUG) << "Final result " << toString(ret);
1361#endif
1362    *_aidl_return = ret;
1363    return ok();
1364}
1365
1366// Dumps the contents of a profile file, using pkgname's dex files for pretty
1367// printing the result.
1368binder::Status InstalldNativeService::dumpProfiles(int32_t uid, const std::string& packageName,
1369        const std::string& codePaths, bool* _aidl_return) {
1370    ENFORCE_UID(AID_SYSTEM);
1371    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1372    std::lock_guard<std::recursive_mutex> lock(mLock);
1373
1374    const char* pkgname = packageName.c_str();
1375    const char* code_paths = codePaths.c_str();
1376
1377    *_aidl_return = dump_profiles(uid, pkgname, code_paths);
1378    return ok();
1379}
1380
1381// TODO: Consider returning error codes.
1382binder::Status InstalldNativeService::mergeProfiles(int32_t uid, const std::string& packageName,
1383        bool* _aidl_return) {
1384    ENFORCE_UID(AID_SYSTEM);
1385    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1386    std::lock_guard<std::recursive_mutex> lock(mLock);
1387
1388    const char* pkgname = packageName.c_str();
1389    *_aidl_return = analyse_profiles(uid, pkgname);
1390    return ok();
1391}
1392
1393binder::Status InstalldNativeService::dexopt(const std::string& apkPath, int32_t uid,
1394        const std::unique_ptr<std::string>& packageName, const std::string& instructionSet,
1395        int32_t dexoptNeeded, const std::unique_ptr<std::string>& outputPath, int32_t dexFlags,
1396        const std::string& compilerFilter, const std::unique_ptr<std::string>& uuid,
1397        const std::unique_ptr<std::string>& sharedLibraries) {
1398    ENFORCE_UID(AID_SYSTEM);
1399    CHECK_ARGUMENT_UUID(uuid);
1400    if (packageName && *packageName != "*") {
1401        CHECK_ARGUMENT_PACKAGE_NAME(*packageName);
1402    }
1403    std::lock_guard<std::recursive_mutex> lock(mLock);
1404
1405    const char* apk_path = apkPath.c_str();
1406    const char* pkgname = packageName ? packageName->c_str() : "*";
1407    const char* instruction_set = instructionSet.c_str();
1408    const char* oat_dir = outputPath ? outputPath->c_str() : nullptr;
1409    const char* compiler_filter = compilerFilter.c_str();
1410    const char* volume_uuid = uuid ? uuid->c_str() : nullptr;
1411    const char* shared_libraries = sharedLibraries ? sharedLibraries->c_str() : nullptr;
1412
1413    int res = android::installd::dexopt(apk_path, uid, pkgname, instruction_set, dexoptNeeded,
1414            oat_dir, dexFlags, compiler_filter, volume_uuid, shared_libraries);
1415    return res ? error(res, "Failed to dexopt") : ok();
1416}
1417
1418binder::Status InstalldNativeService::markBootComplete(const std::string& instructionSet) {
1419    ENFORCE_UID(AID_SYSTEM);
1420    std::lock_guard<std::recursive_mutex> lock(mLock);
1421
1422    const char* instruction_set = instructionSet.c_str();
1423
1424    char boot_marker_path[PKG_PATH_MAX];
1425    sprintf(boot_marker_path,
1426          "%s/%s/%s/.booting",
1427          android_data_dir.path,
1428          DALVIK_CACHE,
1429          instruction_set);
1430
1431    ALOGV("mark_boot_complete : %s", boot_marker_path);
1432    if (unlink(boot_marker_path) != 0) {
1433        return error(StringPrintf("Failed to unlink %s", boot_marker_path));
1434    }
1435    return ok();
1436}
1437
1438void mkinnerdirs(char* path, int basepos, mode_t mode, int uid, int gid,
1439        struct stat* statbuf)
1440{
1441    while (path[basepos] != 0) {
1442        if (path[basepos] == '/') {
1443            path[basepos] = 0;
1444            if (lstat(path, statbuf) < 0) {
1445                ALOGV("Making directory: %s\n", path);
1446                if (mkdir(path, mode) == 0) {
1447                    chown(path, uid, gid);
1448                } else {
1449                    ALOGW("Unable to make directory %s: %s\n", path, strerror(errno));
1450                }
1451            }
1452            path[basepos] = '/';
1453            basepos++;
1454        }
1455        basepos++;
1456    }
1457}
1458
1459binder::Status InstalldNativeService::linkNativeLibraryDirectory(
1460        const std::unique_ptr<std::string>& uuid, const std::string& packageName,
1461        const std::string& nativeLibPath32, int32_t userId) {
1462    ENFORCE_UID(AID_SYSTEM);
1463    CHECK_ARGUMENT_UUID(uuid);
1464    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1465    std::lock_guard<std::recursive_mutex> lock(mLock);
1466
1467    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1468    const char* pkgname = packageName.c_str();
1469    const char* asecLibDir = nativeLibPath32.c_str();
1470    struct stat s, libStat;
1471    binder::Status res = ok();
1472
1473    auto _pkgdir = create_data_user_ce_package_path(uuid_, userId, pkgname);
1474    auto _libsymlink = _pkgdir + PKG_LIB_POSTFIX;
1475
1476    const char* pkgdir = _pkgdir.c_str();
1477    const char* libsymlink = _libsymlink.c_str();
1478
1479    if (stat(pkgdir, &s) < 0) {
1480        return error("Failed to stat " + _pkgdir);
1481    }
1482
1483    if (chown(pkgdir, AID_INSTALL, AID_INSTALL) < 0) {
1484        return error("Failed to chown " + _pkgdir);
1485    }
1486
1487    if (chmod(pkgdir, 0700) < 0) {
1488        res = error("Failed to chmod " + _pkgdir);
1489        goto out;
1490    }
1491
1492    if (lstat(libsymlink, &libStat) < 0) {
1493        if (errno != ENOENT) {
1494            res = error("Failed to stat " + _libsymlink);
1495            goto out;
1496        }
1497    } else {
1498        if (S_ISDIR(libStat.st_mode)) {
1499            if (delete_dir_contents(libsymlink, 1, NULL) < 0) {
1500                res = error("Failed to delete " + _libsymlink);
1501                goto out;
1502            }
1503        } else if (S_ISLNK(libStat.st_mode)) {
1504            if (unlink(libsymlink) < 0) {
1505                res = error("Failed to unlink " + _libsymlink);
1506                goto out;
1507            }
1508        }
1509    }
1510
1511    if (symlink(asecLibDir, libsymlink) < 0) {
1512        res = error("Failed to symlink " + _libsymlink + " to " + nativeLibPath32);
1513        goto out;
1514    }
1515
1516out:
1517    if (chmod(pkgdir, s.st_mode) < 0) {
1518        auto msg = "Failed to cleanup chmod " + _pkgdir;
1519        if (res.isOk()) {
1520            res = error(msg);
1521        } else {
1522            PLOG(ERROR) << msg;
1523        }
1524    }
1525
1526    if (chown(pkgdir, s.st_uid, s.st_gid) < 0) {
1527        auto msg = "Failed to cleanup chown " + _pkgdir;
1528        if (res.isOk()) {
1529            res = error(msg);
1530        } else {
1531            PLOG(ERROR) << msg;
1532        }
1533    }
1534
1535    return res;
1536}
1537
1538static void run_idmap(const char *target_apk, const char *overlay_apk, int idmap_fd)
1539{
1540    static const char *IDMAP_BIN = "/system/bin/idmap";
1541    static const size_t MAX_INT_LEN = 32;
1542    char idmap_str[MAX_INT_LEN];
1543
1544    snprintf(idmap_str, sizeof(idmap_str), "%d", idmap_fd);
1545
1546    execl(IDMAP_BIN, IDMAP_BIN, "--fd", target_apk, overlay_apk, idmap_str, (char*)NULL);
1547    ALOGE("execl(%s) failed: %s\n", IDMAP_BIN, strerror(errno));
1548}
1549
1550// Transform string /a/b/c.apk to (prefix)/a@b@c.apk@(suffix)
1551// eg /a/b/c.apk to /data/resource-cache/a@b@c.apk@idmap
1552static int flatten_path(const char *prefix, const char *suffix,
1553        const char *overlay_path, char *idmap_path, size_t N)
1554{
1555    if (overlay_path == NULL || idmap_path == NULL) {
1556        return -1;
1557    }
1558    const size_t len_overlay_path = strlen(overlay_path);
1559    // will access overlay_path + 1 further below; requires absolute path
1560    if (len_overlay_path < 2 || *overlay_path != '/') {
1561        return -1;
1562    }
1563    const size_t len_idmap_root = strlen(prefix);
1564    const size_t len_suffix = strlen(suffix);
1565    if (SIZE_MAX - len_idmap_root < len_overlay_path ||
1566            SIZE_MAX - (len_idmap_root + len_overlay_path) < len_suffix) {
1567        // additions below would cause overflow
1568        return -1;
1569    }
1570    if (N < len_idmap_root + len_overlay_path + len_suffix) {
1571        return -1;
1572    }
1573    memset(idmap_path, 0, N);
1574    snprintf(idmap_path, N, "%s%s%s", prefix, overlay_path + 1, suffix);
1575    char *ch = idmap_path + len_idmap_root;
1576    while (*ch != '\0') {
1577        if (*ch == '/') {
1578            *ch = '@';
1579        }
1580        ++ch;
1581    }
1582    return 0;
1583}
1584
1585binder::Status InstalldNativeService::idmap(const std::string& targetApkPath,
1586        const std::string& overlayApkPath, int32_t uid) {
1587    ENFORCE_UID(AID_SYSTEM);
1588    std::lock_guard<std::recursive_mutex> lock(mLock);
1589
1590    const char* target_apk = targetApkPath.c_str();
1591    const char* overlay_apk = overlayApkPath.c_str();
1592    ALOGV("idmap target_apk=%s overlay_apk=%s uid=%d\n", target_apk, overlay_apk, uid);
1593
1594    int idmap_fd = -1;
1595    char idmap_path[PATH_MAX];
1596
1597    if (flatten_path(IDMAP_PREFIX, IDMAP_SUFFIX, overlay_apk,
1598                idmap_path, sizeof(idmap_path)) == -1) {
1599        ALOGE("idmap cannot generate idmap path for overlay %s\n", overlay_apk);
1600        goto fail;
1601    }
1602
1603    unlink(idmap_path);
1604    idmap_fd = open(idmap_path, O_RDWR | O_CREAT | O_EXCL, 0644);
1605    if (idmap_fd < 0) {
1606        ALOGE("idmap cannot open '%s' for output: %s\n", idmap_path, strerror(errno));
1607        goto fail;
1608    }
1609    if (fchown(idmap_fd, AID_SYSTEM, uid) < 0) {
1610        ALOGE("idmap cannot chown '%s'\n", idmap_path);
1611        goto fail;
1612    }
1613    if (fchmod(idmap_fd, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH) < 0) {
1614        ALOGE("idmap cannot chmod '%s'\n", idmap_path);
1615        goto fail;
1616    }
1617
1618    pid_t pid;
1619    pid = fork();
1620    if (pid == 0) {
1621        /* child -- drop privileges before continuing */
1622        if (setgid(uid) != 0) {
1623            ALOGE("setgid(%d) failed during idmap\n", uid);
1624            exit(1);
1625        }
1626        if (setuid(uid) != 0) {
1627            ALOGE("setuid(%d) failed during idmap\n", uid);
1628            exit(1);
1629        }
1630        if (flock(idmap_fd, LOCK_EX | LOCK_NB) != 0) {
1631            ALOGE("flock(%s) failed during idmap: %s\n", idmap_path, strerror(errno));
1632            exit(1);
1633        }
1634
1635        run_idmap(target_apk, overlay_apk, idmap_fd);
1636        exit(1); /* only if exec call to idmap failed */
1637    } else {
1638        int status = wait_child(pid);
1639        if (status != 0) {
1640            ALOGE("idmap failed, status=0x%04x\n", status);
1641            goto fail;
1642        }
1643    }
1644
1645    close(idmap_fd);
1646    return ok();
1647fail:
1648    if (idmap_fd >= 0) {
1649        close(idmap_fd);
1650        unlink(idmap_path);
1651    }
1652    return error();
1653}
1654
1655binder::Status InstalldNativeService::restoreconAppData(const std::unique_ptr<std::string>& uuid,
1656        const std::string& packageName, int32_t userId, int32_t flags, int32_t appId,
1657        const std::string& seInfo) {
1658    ENFORCE_UID(AID_SYSTEM);
1659    CHECK_ARGUMENT_UUID(uuid);
1660    CHECK_ARGUMENT_PACKAGE_NAME(packageName);
1661    std::lock_guard<std::recursive_mutex> lock(mLock);
1662
1663    binder::Status res = ok();
1664
1665    // SELINUX_ANDROID_RESTORECON_DATADATA flag is set by libselinux. Not needed here.
1666    unsigned int seflags = SELINUX_ANDROID_RESTORECON_RECURSE;
1667    const char* uuid_ = uuid ? uuid->c_str() : nullptr;
1668    const char* pkgName = packageName.c_str();
1669    const char* seinfo = seInfo.c_str();
1670
1671    uid_t uid = multiuser_get_uid(userId, appId);
1672    if (flags & FLAG_STORAGE_CE) {
1673        auto path = create_data_user_ce_package_path(uuid_, userId, pkgName);
1674        if (selinux_android_restorecon_pkgdir(path.c_str(), seinfo, uid, seflags) < 0) {
1675            res = error("restorecon failed for " + path);
1676        }
1677    }
1678    if (flags & FLAG_STORAGE_DE) {
1679        auto path = create_data_user_de_package_path(uuid_, userId, pkgName);
1680        if (selinux_android_restorecon_pkgdir(path.c_str(), seinfo, uid, seflags) < 0) {
1681            res = error("restorecon failed for " + path);
1682        }
1683    }
1684    return res;
1685}
1686
1687binder::Status InstalldNativeService::createOatDir(const std::string& oatDir,
1688        const std::string& instructionSet) {
1689    ENFORCE_UID(AID_SYSTEM);
1690    std::lock_guard<std::recursive_mutex> lock(mLock);
1691
1692    const char* oat_dir = oatDir.c_str();
1693    const char* instruction_set = instructionSet.c_str();
1694    char oat_instr_dir[PKG_PATH_MAX];
1695
1696    if (validate_apk_path(oat_dir)) {
1697        return error("Invalid path " + oatDir);
1698    }
1699    if (fs_prepare_dir(oat_dir, S_IRWXU | S_IRWXG | S_IXOTH, AID_SYSTEM, AID_INSTALL)) {
1700        return error("Failed to prepare " + oatDir);
1701    }
1702    if (selinux_android_restorecon(oat_dir, 0)) {
1703        return error("Failed to restorecon " + oatDir);
1704    }
1705    snprintf(oat_instr_dir, PKG_PATH_MAX, "%s/%s", oat_dir, instruction_set);
1706    if (fs_prepare_dir(oat_instr_dir, S_IRWXU | S_IRWXG | S_IXOTH, AID_SYSTEM, AID_INSTALL)) {
1707        return error(StringPrintf("Failed to prepare %s", oat_instr_dir));
1708    }
1709    return ok();
1710}
1711
1712binder::Status InstalldNativeService::rmPackageDir(const std::string& packageDir) {
1713    ENFORCE_UID(AID_SYSTEM);
1714    std::lock_guard<std::recursive_mutex> lock(mLock);
1715
1716    if (validate_apk_path(packageDir.c_str())) {
1717        return error("Invalid path " + packageDir);
1718    }
1719    if (delete_dir_contents_and_dir(packageDir) != 0) {
1720        return error("Failed to delete " + packageDir);
1721    }
1722    return ok();
1723}
1724
1725binder::Status InstalldNativeService::linkFile(const std::string& relativePath,
1726        const std::string& fromBase, const std::string& toBase) {
1727    ENFORCE_UID(AID_SYSTEM);
1728    std::lock_guard<std::recursive_mutex> lock(mLock);
1729
1730    const char* relative_path = relativePath.c_str();
1731    const char* from_base = fromBase.c_str();
1732    const char* to_base = toBase.c_str();
1733    char from_path[PKG_PATH_MAX];
1734    char to_path[PKG_PATH_MAX];
1735    snprintf(from_path, PKG_PATH_MAX, "%s/%s", from_base, relative_path);
1736    snprintf(to_path, PKG_PATH_MAX, "%s/%s", to_base, relative_path);
1737
1738    if (validate_apk_path_subdirs(from_path)) {
1739        return error(StringPrintf("Invalid from path %s", from_path));
1740    }
1741
1742    if (validate_apk_path_subdirs(to_path)) {
1743        return error(StringPrintf("Invalid to path %s", to_path));
1744    }
1745
1746    if (link(from_path, to_path) < 0) {
1747        return error(StringPrintf("Failed to link from %s to %s", from_path, to_path));
1748    }
1749
1750    return ok();
1751}
1752
1753binder::Status InstalldNativeService::moveAb(const std::string& apkPath,
1754        const std::string& instructionSet, const std::string& outputPath) {
1755    ENFORCE_UID(AID_SYSTEM);
1756    std::lock_guard<std::recursive_mutex> lock(mLock);
1757
1758    const char* apk_path = apkPath.c_str();
1759    const char* instruction_set = instructionSet.c_str();
1760    const char* oat_dir = outputPath.c_str();
1761
1762    bool success = move_ab(apk_path, instruction_set, oat_dir);
1763    return success ? ok() : error();
1764}
1765
1766binder::Status InstalldNativeService::deleteOdex(const std::string& apkPath,
1767        const std::string& instructionSet, const std::string& outputPath) {
1768    ENFORCE_UID(AID_SYSTEM);
1769    std::lock_guard<std::recursive_mutex> lock(mLock);
1770
1771    const char* apk_path = apkPath.c_str();
1772    const char* instruction_set = instructionSet.c_str();
1773    const char* oat_dir = outputPath.c_str();
1774
1775    bool res = delete_odex(apk_path, instruction_set, oat_dir);
1776    return res ? ok() : error();
1777}
1778
1779}  // namespace installd
1780}  // namespace android
1781