123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273 |
- #include "MoveStorage.h"
- #include "Utils.h"
- #include "VolumeManager.h"
- #include <android-base/logging.h>
- #include <android-base/properties.h>
- #include <android-base/stringprintf.h>
- #include <hardware_legacy/power.h>
- #include <private/android_filesystem_config.h>
- #include <thread>
- #include <dirent.h>
- #include <sys/wait.h>
- #define CONSTRAIN(amount, low, high) \
- ((amount) < (low) ? (low) : ((amount) > (high) ? (high) : (amount)))
- static const char* kPropBlockingExec = "persist.sys.blocking_exec";
- using android::base::StringPrintf;
- namespace android {
- namespace vold {
- static const int kMoveSucceeded = -100;
- static const int kMoveFailedInternalError = -6;
- static const char* kCpPath = "/system/bin/cp";
- static const char* kRmPath = "/system/bin/rm";
- static const char* kWakeLock = "MoveTask";
- static void notifyProgress(int progress,
- const android::sp<android::os::IVoldTaskListener>& listener) {
- if (listener) {
- android::os::PersistableBundle extras;
- listener->onStatus(progress, extras);
- }
- }
- static bool pushBackContents(const std::string& path, std::vector<std::string>& cmd,
- int searchLevels) {
- if (searchLevels == 0) {
- cmd.emplace_back(path);
- return true;
- }
- auto dirp = std::unique_ptr<DIR, int (*)(DIR*)>(opendir(path.c_str()), closedir);
- if (!dirp) {
- PLOG(ERROR) << "Unable to open directory: " << path;
- return false;
- }
- bool found = false;
- struct dirent* ent;
- while ((ent = readdir(dirp.get())) != NULL) {
- if ((!strcmp(ent->d_name, ".")) || (!strcmp(ent->d_name, ".."))) {
- continue;
- }
- auto subdir = path + "/" + ent->d_name;
- found |= pushBackContents(subdir, cmd, searchLevels - 1);
- }
- return found;
- }
- static status_t execRm(const std::string& path, int startProgress, int stepProgress,
- const android::sp<android::os::IVoldTaskListener>& listener) {
- notifyProgress(startProgress, listener);
- uint64_t expectedBytes = GetTreeBytes(path);
- uint64_t startFreeBytes = GetFreeBytes(path);
- std::vector<std::string> cmd;
- cmd.push_back(kRmPath);
- cmd.push_back("-f");
- cmd.push_back("-R");
- if (!pushBackContents(path, cmd, 2)) {
- LOG(WARNING) << "No contents in " << path;
- return OK;
- }
- if (android::base::GetBoolProperty(kPropBlockingExec, false)) {
- return ForkExecvp(cmd);
- }
- pid_t pid = ForkExecvpAsync(cmd);
- if (pid == -1) return -1;
- int status;
- while (true) {
- if (waitpid(pid, &status, WNOHANG) == pid) {
- if (WIFEXITED(status)) {
- LOG(DEBUG) << "Finished rm with status " << WEXITSTATUS(status);
- return (WEXITSTATUS(status) == 0) ? OK : -1;
- } else {
- break;
- }
- }
- sleep(1);
- uint64_t deltaFreeBytes = GetFreeBytes(path) - startFreeBytes;
- notifyProgress(
- startProgress +
- CONSTRAIN((int)((deltaFreeBytes * stepProgress) / expectedBytes), 0, stepProgress),
- listener);
- }
- return -1;
- }
- static status_t execCp(const std::string& fromPath, const std::string& toPath, int startProgress,
- int stepProgress,
- const android::sp<android::os::IVoldTaskListener>& listener) {
- notifyProgress(startProgress, listener);
- uint64_t expectedBytes = GetTreeBytes(fromPath);
- uint64_t startFreeBytes = GetFreeBytes(toPath);
- if (expectedBytes > startFreeBytes) {
- LOG(ERROR) << "Data size " << expectedBytes << " is too large to fit in free space "
- << startFreeBytes;
- return -1;
- }
- std::vector<std::string> cmd;
- cmd.push_back(kCpPath);
- cmd.push_back("-p");
- cmd.push_back("-R");
- cmd.push_back("-P");
- cmd.push_back("-d");
- if (!pushBackContents(fromPath, cmd, 1)) {
- LOG(WARNING) << "No contents in " << fromPath;
- return OK;
- }
- cmd.push_back(toPath.c_str());
- if (android::base::GetBoolProperty(kPropBlockingExec, false)) {
- return ForkExecvp(cmd);
- }
- pid_t pid = ForkExecvpAsync(cmd);
- if (pid == -1) return -1;
- int status;
- while (true) {
- if (waitpid(pid, &status, WNOHANG) == pid) {
- if (WIFEXITED(status)) {
- LOG(DEBUG) << "Finished cp with status " << WEXITSTATUS(status);
- return (WEXITSTATUS(status) == 0) ? OK : -1;
- } else {
- break;
- }
- }
- sleep(1);
- uint64_t deltaFreeBytes = startFreeBytes - GetFreeBytes(toPath);
- notifyProgress(
- startProgress +
- CONSTRAIN((int)((deltaFreeBytes * stepProgress) / expectedBytes), 0, stepProgress),
- listener);
- }
- return -1;
- }
- static void bringOffline(const std::shared_ptr<VolumeBase>& vol) {
- vol->destroy();
- vol->setSilent(true);
- vol->create();
- vol->setMountFlags(0);
- vol->mount();
- }
- static void bringOnline(const std::shared_ptr<VolumeBase>& vol) {
- vol->destroy();
- vol->setSilent(false);
- vol->create();
- }
- static status_t moveStorageInternal(const std::shared_ptr<VolumeBase>& from,
- const std::shared_ptr<VolumeBase>& to,
- const android::sp<android::os::IVoldTaskListener>& listener) {
- std::string fromPath;
- std::string toPath;
-
- if (from->getType() != VolumeBase::Type::kEmulated) goto fail;
- if (to->getType() != VolumeBase::Type::kEmulated) goto fail;
-
-
- {
- std::lock_guard<std::mutex> lock(VolumeManager::Instance()->getLock());
- bringOffline(from);
- bringOffline(to);
- }
- fromPath = from->getInternalPath();
- toPath = to->getInternalPath();
-
- if (execRm(toPath, 10, 10, listener) != OK) {
- goto fail;
- }
-
- if (execCp(fromPath, toPath, 20, 60, listener) != OK) {
- goto copy_fail;
- }
-
-
- notifyProgress(82, listener);
- {
- std::lock_guard<std::mutex> lock(VolumeManager::Instance()->getLock());
- bringOnline(from);
- bringOnline(to);
- }
-
- if (execRm(fromPath, 85, 15, listener) != OK) {
- goto fail;
- }
- notifyProgress(kMoveSucceeded, listener);
- return OK;
- copy_fail:
-
-
-
- execRm(toPath, 80, 1, listener);
- fail:
-
- {
- std::lock_guard<std::mutex> lock(VolumeManager::Instance()->getLock());
- bringOnline(from);
- bringOnline(to);
- }
-
- notifyProgress(kMoveFailedInternalError, listener);
- return -1;
- }
- void MoveStorage(const std::shared_ptr<VolumeBase>& from, const std::shared_ptr<VolumeBase>& to,
- const android::sp<android::os::IVoldTaskListener>& listener) {
- acquire_wake_lock(PARTIAL_WAKE_LOCK, kWakeLock);
- android::os::PersistableBundle extras;
- status_t res = moveStorageInternal(from, to, listener);
- if (listener) {
- listener->onFinished(res, extras);
- }
- release_wake_lock(kWakeLock);
- }
- }
- }
|