redo fix permissions in c++ for faster changes
diff --git a/fixPermissions.cpp b/fixPermissions.cpp
new file mode 100644
index 0000000..d12dc1a
--- /dev/null
+++ b/fixPermissions.cpp
@@ -0,0 +1,536 @@
+/*
+ Copyright 2012 bigbiff/Dees_Troy TeamWin
+ This file is part of TWRP/TeamWin Recovery Project.
+
+ TWRP is free software: you can redistribute it and/or modify
+ it under the terms of the GNU General Public License as published by
+ the Free Software Foundation, either version 3 of the License, or
+ (at your option) any later version.
+
+ TWRP is distributed in the hope that it will be useful,
+ but WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with TWRP. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include <iostream>
+#include <fstream>
+#include <sstream>
+#include <string>
+#include <vector>
+#include <string.h>
+#include <libgen.h>
+#include <unistd.h>
+#include <sys/stat.h>
+#include <dirent.h>
+#include "gui/rapidxml.hpp"
+#include "fixPermissions.hpp"
+#include "twrp-functions.hpp"
+#include "common.h"
+
+using namespace std;
+using namespace rapidxml;
+
+int fixPermissions::fixPerms(bool enable_debug, bool remove_data_for_missing_apps) {
+ packageFile = "/data/system/packages.xml";
+ debug = enable_debug;
+ remove_data = remove_data_for_missing_apps;
+
+ if (!(TWFunc::Path_Exists(packageFile))) {
+ ui_print("Can't check permissions\n");
+ ui_print("after Factory Reset.\n");
+ ui_print("Please boot rom and try\n");
+ ui_print("again after you reboot into\n");
+ ui_print("recovery.\n");
+ return -1;
+ }
+
+ ui_print("Fixing permissions...\nLoading packages...\n");
+ if ((getPackages()) != 0) {
+ return -1;
+ }
+
+ ui_print("Fixing /system/app perissions...\n");
+ if ((fixSystemApps()) != 0) {
+ return -1;
+ }
+
+ ui_print("Fixing /data/app permisions...\n");
+ if ((fixDataApps()) != 0) {
+ return -1;
+ }
+
+ ui_print("Fixing /data/data permisions...\n");
+ if ((fixDataData()) != 0) {
+ return -1;
+ }
+ ui_print("Done fixing permissions.\n");
+ return 0;
+}
+
+int fixPermissions::pchown(string fn, int puid, int pgid) {
+ LOGI("Fixing %s, uid: %d, gid: %d\n", fn.c_str(), puid, pgid);
+ if (chown(fn.c_str(), puid, pgid) != 0) {
+ LOGE("Unable to chown '%s' %i %i\n", fn.c_str(), puid, pgid);
+ return -1;
+ }
+ return 0;
+}
+
+int fixPermissions::pchmod(string fn, string mode) {
+ long mask = 0;
+ LOGI("Fixing %s, mode: %s\n", fn.c_str(), mode.c_str());
+ for ( std::string::size_type n = 0; n < mode.length(); ++n) {
+ if (n == 0) {
+ if (mode[n] == '0')
+ continue;
+ else if (mode[n] == '1')
+ mask = S_ISVTX;
+ else if (mode[n] == '2')
+ mask = S_ISGID;
+ }
+ else if (n == 1) {
+ if (mode[n] == '7') {
+ mask |= S_IRWXU;
+ }
+ if (mode[n] == '6') {
+ mask |= S_IRUSR;
+ mask |= S_IWUSR;
+ }
+ if (mode[n] == '5') {
+ mask |= S_IRUSR;
+ mask |= S_IXUSR;
+ }
+ if (mode[n] == '4')
+ mask |= S_IRUSR;
+ if (mode[n] == '3') {
+ mask |= S_IWUSR;
+ mask |= S_IRUSR;
+ }
+ if (mode[n] == '2')
+ mask |= S_IWUSR;
+ if (mode[n] == '1')
+ mask |= S_IXUSR;
+ }
+ else if (n == 2) {
+ if (mode[n] == '7') {
+ mask |= S_IRWXG;
+ }
+ if (mode[n] == '6') {
+ mask |= S_IRGRP;
+ mask |= S_IWGRP;
+ }
+ if (mode[n] == '5') {
+ mask |= S_IRGRP;
+ mask |= S_IXGRP;
+ }
+ if (mode[n] == '4')
+ mask |= S_IRGRP;
+ if (mode[n] == '3') {
+ mask |= S_IWGRP;
+ mask |= S_IXGRP;
+ }
+ if (mode[n] == '2')
+ mask |= S_IWGRP;
+ if (mode[n] == '1')
+ mask |= S_IXGRP;
+ }
+ else if (n == 3) {
+ if (mode[n] == '7') {
+ mask |= S_IRWXO;
+ }
+ if (mode[n] == '6') {
+ mask |= S_IROTH;
+ mask |= S_IWOTH;
+ }
+ if (mode[n] == '5') {
+ mask |= S_IROTH;
+ mask |= S_IXOTH;
+ }
+ if (mode[n] == '4')
+ mask |= S_IROTH;
+ if (mode[n] == '3') {
+ mask |= S_IWOTH;
+ mask |= S_IXOTH;
+ }
+ if (mode[n] == '2')
+ mask |= S_IWGRP;
+ if (mode[n] == '1')
+ mask |= S_IXGRP;
+ }
+ }
+
+ if (chmod(fn.c_str(), mask) != 0) {
+ LOGE("Unable to chmod '%s' %l\n", fn.c_str(), mask);
+ return -1;
+ }
+
+ return 0;
+}
+
+int fixPermissions::removeDir(const string path) {
+ DIR *d = opendir(path.c_str());
+ int r = 0;
+ string new_path;
+
+ if (d == NULL) {
+ LOGE("Error opening '%s'\n", path.c_str());
+ return -1;
+ }
+
+ if (d) {
+ struct dirent *p;
+ while (!r && (p = readdir(d))) {
+ if (!strcmp(p->d_name, ".") || !strcmp(p->d_name, ".."))
+ continue;
+
+ new_path = path + "/";
+ new_path.append(p->d_name);
+ if (p->d_type == DT_DIR) {
+ r = removeDir(new_path);
+ if (!r)
+ LOGI("Unable to removeDir '%s'\n", new_path.c_str());
+ } else {
+ r = unlink(new_path.c_str());
+ if (!r)
+ LOGI("Unable to unlink '%s'\n", new_path.c_str());
+ }
+ }
+ closedir(d);
+ }
+ if (!r)
+ r = rmdir(path.c_str());
+ if (!r)
+ LOGI("Unable to rmdir '%s'\n", path.c_str());
+
+ return r;
+}
+
+int fixPermissions::fixSystemApps() {
+ temp = head;
+ while (temp != NULL) {
+ if (TWFunc::Path_Exists(temp->codePath)) {
+ if (temp->appDir.compare("/system/app") == 0) {
+ if (debug) {
+ LOGI("Looking at '%s'\n", temp->codePath.c_str());
+ LOGI("Fixing permissions on '%s'\n", temp->pkgName.c_str());
+ LOGI("Directory: '%s'\n", temp->appDir.c_str());
+ LOGI("Original package owner: %d, group: %d\n", temp->uid, temp->gid);
+ }
+ if (pchown(temp->codePath, 0, 0) != 0)
+ return -1;
+ if (pchmod(temp->codePath, "0644") != 0)
+ return -1;
+ }
+ } else {
+ //Remove data directory since app isn't installed
+ if (remove_data && TWFunc::Path_Exists(temp->dDir) && temp->appDir.size() >= 9 && temp->appDir.substr(0, 9) != "/mnt/asec") {
+ if (debug)
+ LOGI("Looking at '%s', removing data dir: '%s', appDir: '%s'", temp->codePath.c_str(), temp->dDir.c_str(), temp->appDir.c_str());
+ if (removeDir(temp->dDir) != 0) {
+ LOGI("Unable to removeDir '%s'\n", temp->dDir.c_str());
+ return -1;
+ }
+ }
+ }
+ temp = temp->next;
+ }
+ return 0;
+}
+
+int fixPermissions::fixDataApps() {
+ bool fix = false;
+ int new_gid = 0;
+ string perms = "0000";
+
+ temp = head;
+ while (temp != NULL) {
+ if (TWFunc::Path_Exists(temp->codePath)) {
+ if (temp->appDir.compare("/data/app") == 0 || temp->appDir.compare("/sd-ext/app") == 0) {
+ fix = true;
+ new_gid = 1000;
+ perms = "0644";
+ } else if (temp->appDir.compare("/data/app-private") == 0 || temp->appDir.compare("/sd-ext/app-private") == 0) {
+ fix = true;
+ new_gid = temp->gid;
+ perms = "0640";
+ } else
+ fix = false;
+ if (fix) {
+ if (debug) {
+ LOGI("Looking at '%s'\n", temp->codePath.c_str());
+ LOGI("Fixing permissions on '%s'\n", temp->pkgName.c_str());
+ LOGI("Directory: '%s'\n", temp->appDir.c_str());
+ LOGI("Original package owner: %d, group: %d\n", temp->uid, temp->gid);
+ }
+ if (pchown(temp->codePath, 1000, new_gid) != 0)
+ return -1;
+ if (pchmod(temp->codePath, perms) != 0)
+ return -1;
+ }
+ } else {
+ //Remove data directory since app isn't installed
+ if (remove_data && TWFunc::Path_Exists(temp->dDir) && temp->appDir.size() >= 9 && temp->appDir.substr(0, 9) != "/mnt/asec") {
+ if (debug)
+ LOGI("Looking at '%s', removing data dir: '%s', appDir: '%s'", temp->codePath.c_str(), temp->dDir.c_str(), temp->appDir.c_str());
+ if (removeDir(temp->dDir) != 0) {
+ LOGI("Unable to removeDir '%s'\n", temp->dDir.c_str());
+ return -1;
+ }
+ }
+ }
+ temp = temp->next;
+ }
+ return 0;
+}
+
+int fixPermissions::fixAllFiles(string directory, int gid, int uid, string file_perms) {
+ vector <string> files;
+ string file;
+
+ files = listAllFiles(directory);
+ for (unsigned i = 0; i < files.size(); ++i) {
+ file = directory + "/";
+ file.append(files.at(i));
+ if (debug)
+ LOGI("Looking at file '%s'\n", file.c_str());
+ if (pchmod(file, file_perms) != 0)
+ return -1;
+ if (pchown(file, uid, gid) != 0)
+ return -1;
+ }
+ return 0;
+}
+
+int fixPermissions::fixDataData() {
+ string directory;
+
+ temp = head;
+ while (temp != NULL) {
+ if (TWFunc::Path_Exists(temp->dDir)) {
+ vector <string> dataDataDirs = listAllDirectories(temp->dDir);
+ for (unsigned n = 0; n < dataDataDirs.size(); ++n) {
+ directory = temp->dDir + "/";
+ directory.append(dataDataDirs.at(n));
+ if (debug)
+ LOGI("Looking at data directory: '%s'\n", directory.c_str());
+ if (dataDataDirs.at(n) == ".") {
+ if (pchmod(temp->dDir, "0755") != 0)
+ return -1;
+ if (pchown(directory.c_str(), temp->uid, temp->gid) != 0)
+ return -1;
+ if (fixAllFiles(directory, temp->uid, temp->gid, "0755") != 0)
+ return -1;
+ }
+ else if (dataDataDirs.at(n) == "..") {
+ if (debug)
+ LOGI("Skipping ..\n");
+ continue;
+ }
+ else if (dataDataDirs.at(n) == "lib") {
+ if (pchmod(directory.c_str(), "0755") != 0)
+ return -1;
+ if (pchown(directory.c_str(), 1000, 1000) != 0)
+ return -1;
+ if (fixAllFiles(directory, temp->uid, temp->gid, "0755") != 0)
+ return -1;
+ }
+ else if (dataDataDirs.at(n) == "shared_prefs") {
+ if (pchmod(directory.c_str(), "0771") != 0)
+ return -1;
+ if (pchown(directory.c_str(), temp->uid, temp->gid) != 0)
+ return -1;
+ if (fixAllFiles(directory, temp->uid, temp->gid, "0660") != 0)
+ return -1;
+ }
+ else if (dataDataDirs.at(n) == "databases") {
+ if (pchmod(directory.c_str(), "0771") != 0)
+ return -1;
+ if (pchown(directory.c_str(), temp->uid, temp->gid) != 0)
+ return -1;
+ if (fixAllFiles(directory, temp->uid, temp->gid, "0660") != 0)
+ return -1;
+ }
+ else if (dataDataDirs.at(n) == "cache") {
+ if (pchmod(directory.c_str(), "0771") != 0)
+ return -1;
+ if (pchown(directory.c_str(), temp->uid, temp->gid) != 0)
+ return -1;
+ if (fixAllFiles(directory, temp->uid, temp->gid, "0600") != 0)
+ return -1;
+ }
+ else {
+ if (pchmod(directory.c_str(), "0771") != 0)
+ return -1;
+ if (pchown(directory.c_str(), temp->uid, temp->gid) != 0)
+ return -1;
+ if (fixAllFiles(directory, temp->uid, temp->gid, "0755") != 0)
+ return -1;
+ }
+ }
+ }
+ temp = temp->next;
+ }
+ return 0;
+}
+
+vector <string> fixPermissions::listAllDirectories(string path) {
+ DIR *dir = opendir(path.c_str());
+ struct dirent *entry = readdir(dir);
+ vector <string> dirs;
+
+ if (dir == NULL) {
+ LOGE("Error opening '%s'\n", path.c_str());
+ return dirs;
+ }
+
+ while (entry != NULL) {
+ if (entry->d_type == DT_DIR)
+ dirs.push_back(entry->d_name);
+ entry = readdir(dir);
+ }
+ closedir(dir);
+ return dirs;
+}
+
+vector <string> fixPermissions::listAllFiles(string path) {
+ DIR *dir = opendir(path.c_str());
+ struct dirent *entry = readdir(dir);
+ vector <string> files;
+
+ if (dir == NULL) {
+ LOGE("Error opening '%s'\n", path.c_str());
+ return files;
+ }
+
+ while (entry != NULL) {
+ if (entry->d_type == DT_REG)
+ files.push_back(entry->d_name);
+ entry = readdir(dir);
+ }
+ closedir(dir);
+ return files;
+}
+
+int fixPermissions::getPackages() {
+ int len = 0;
+ bool skiploop = false;
+ vector <string> skip;
+ string dataDir;
+ string name;
+ head = NULL;
+
+ skip.push_back("/system/framework/framework-res.apk");
+ skip.push_back("/system/framework/com.htc.resources.apk");
+
+ ifstream xmlFile(packageFile.c_str());
+ xmlFile.seekg(0, ios::end);
+ len = (int) xmlFile.tellg();
+ xmlFile.seekg(0, ios::beg);
+ char xmlBuf[len + 1];
+ xmlFile.read(&xmlBuf[0], len);
+ xmlBuf[len] = '\0';
+ xml_document<> pkgDoc;
+ pkgDoc.parse<parse_full>(&xmlBuf[0]);
+
+ xml_node<> * pkgNode = pkgDoc.first_node("packages");
+ xml_node <> * next = pkgNode->first_node("package");
+
+ if (next == NULL) {
+ LOGE("No packages found to fix.\n");
+ return -1;
+ }
+
+ //Get packages
+ while (next->first_attribute("name") != NULL) {
+ package* temp = new package;
+ for (unsigned n = 0; n < skip.size(); ++n) {
+ if (skip.at(n).compare(next->first_attribute("codePath")->value()) == 0) {
+ skiploop = true;
+ break;
+ }
+ }
+
+ if (skiploop == true) {
+ if (debug)
+ LOGI("Skipping package %s\n", next->first_attribute("codePath")->value());
+ free(temp);
+ next = next->next_sibling();
+ skiploop = false;
+ continue;
+ }
+ name.append((next->first_attribute("name")->value()));
+ temp->pkgName = next->first_attribute("name")->value();
+ if (debug)
+ LOGI("Loading pkg: %s\n", next->first_attribute("name")->value());
+ temp->codePath = next->first_attribute("codePath")->value();
+ temp->app = basename(next->first_attribute("codePath")->value());
+ temp->appDir = dirname(next->first_attribute("codePath")->value());
+ dataDir.append("/data/data/");
+ dataDir.append(name);
+ temp->dDir = dataDir;
+ if ( next->first_attribute("sharedUserId") != NULL) {
+ temp->uid = atoi(next->first_attribute("sharedUserId")->value());
+ temp->gid = atoi(next->first_attribute("sharedUserId")->value());
+ }
+ else {
+ temp->uid = atoi(next->first_attribute("userId")->value());
+ temp->gid = atoi(next->first_attribute("userId")->value());
+ }
+ next = next->next_sibling();
+ name.clear();
+ dataDir.clear();
+ temp->next = head;
+ head = temp;
+ }
+ //Get updated packages
+ next = pkgNode->first_node("updated-package");
+ if (next != NULL) {
+ while (next->first_attribute("name") != NULL) {
+ package* temp = new package;
+ for (unsigned n = 0; n < skip.size(); ++n) {
+ if (skip.at(n).compare(next->first_attribute("codePath")->value()) == 0) {
+ skiploop = true;
+ break;
+ }
+ }
+
+ if (skiploop == true) {
+ if (debug)
+ LOGI("Skipping package %s\n", next->first_attribute("codePath")->value());
+ free(temp);
+ next = next->next_sibling();
+ skiploop = false;
+ continue;
+ }
+ name.append((next->first_attribute("name")->value()));
+ temp->pkgName = next->first_attribute("name")->value();
+ if (debug)
+ LOGI("Loading pkg: %s\n", next->first_attribute("name")->value());
+ temp->codePath = next->first_attribute("codePath")->value();
+ temp->app = basename(next->first_attribute("codePath")->value());
+ temp->appDir = dirname(next->first_attribute("codePath")->value());
+
+ dataDir.append("/data/data/");
+ dataDir.append(name);
+ temp->dDir = dataDir;
+ if ( next->first_attribute("sharedUserId") != NULL) {
+ temp->uid = atoi(next->first_attribute("sharedUserId")->value());
+ temp->gid = atoi(next->first_attribute("sharedUserId")->value());
+ }
+ else {
+ temp->uid = atoi(next->first_attribute("userId")->value());
+ temp->gid = atoi(next->first_attribute("userId")->value());
+ }
+ next = next->next_sibling();
+ name.clear();
+ dataDir.clear();
+ temp->next = head;
+ head = temp;
+ }
+ }
+ return 0;
+}