1 /* 2 * Copyright (C) 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 <stdio.h> 18 #include <stdlib.h> 19 #include <unistd.h> 20 #include <string.h> 21 #include <signal.h> 22 #include <errno.h> 23 #include <fcntl.h> 24 #include <stdlib.h> 25 #include <poll.h> 26 27 #include <sys/select.h> 28 #include <sys/time.h> 29 #include <sys/types.h> 30 #include <sys/un.h> 31 32 #include "android/os/IVold.h" 33 34 #include <android-base/logging.h> 35 #include <android-base/stringprintf.h> 36 #include <binder/IServiceManager.h> 37 #include <binder/Status.h> 38 39 #include <private/android_filesystem_config.h> 40 41 static void usage(char *progname); 42 43 static android::sp<android::IBinder> getServiceAggressive() { 44 android::sp<android::IBinder> res; 45 auto sm = android::defaultServiceManager(); 46 auto name = android::String16("vold"); 47 for (int i = 0; i < 5000; i++) { 48 res = sm->checkService(name); 49 if (res) { 50 LOG(VERBOSE) << "Waited " << (i * 10) << "ms for vold"; 51 break; 52 } 53 usleep(10000); // 10ms 54 } 55 return res; 56 } 57 58 static void checkStatus(android::binder::Status status) { 59 if (status.isOk()) return; 60 LOG(ERROR) << "Failed: " << status.toString8().string(); 61 exit(ENOTTY); 62 } 63 64 int main(int argc, char** argv) { 65 setenv("ANDROID_LOG_TAGS", "*:v", 1); 66 if (getppid() == 1) { 67 // If init is calling us then it's during boot and we should log to kmsg 68 android::base::InitLogging(argv, &android::base::KernelLogger); 69 } else { 70 android::base::InitLogging(argv, &android::base::StderrLogger); 71 } 72 std::vector<std::string> args(argv + 1, argv + argc); 73 74 if (args.size() > 0 && args[0] == "--wait") { 75 // Just ignore the --wait flag 76 args.erase(args.begin()); 77 } 78 79 if (args.size() < 2) { 80 usage(argv[0]); 81 exit(5); 82 } 83 android::sp<android::IBinder> binder = getServiceAggressive(); 84 if (!binder) { 85 LOG(ERROR) << "Failed to obtain vold Binder"; 86 exit(EINVAL); 87 } 88 auto vold = android::interface_cast<android::os::IVold>(binder); 89 90 if (args[0] == "cryptfs" && args[1] == "enablefilecrypto") { 91 checkStatus(vold->fbeEnable()); 92 } else if (args[0] == "cryptfs" && args[1] == "init_user0") { 93 checkStatus(vold->initUser0()); 94 } else if (args[0] == "cryptfs" && args[1] == "enablecrypto") { 95 int passwordType = android::os::IVold::PASSWORD_TYPE_DEFAULT; 96 int encryptionFlags = android::os::IVold::ENCRYPTION_FLAG_NO_UI; 97 checkStatus(vold->fdeEnable(passwordType, "", encryptionFlags)); 98 } else if (args[0] == "cryptfs" && args[1] == "mountdefaultencrypted") { 99 checkStatus(vold->mountDefaultEncrypted()); 100 } else if (args[0] == "volume" && args[1] == "shutdown") { 101 checkStatus(vold->shutdown()); 102 } else if (args[0] == "cryptfs" && args[1] == "checkEncryption" && args.size() == 3) { 103 checkStatus(vold->checkEncryption(args[2])); 104 } else if (args[0] == "cryptfs" && args[1] == "mountFstab" && args.size() == 3) { 105 checkStatus(vold->mountFstab(args[2])); 106 } else if (args[0] == "cryptfs" && args[1] == "encryptFstab" && args.size() == 3) { 107 checkStatus(vold->encryptFstab(args[2])); 108 } else { 109 LOG(ERROR) << "Raw commands are no longer supported"; 110 exit(EINVAL); 111 } 112 return 0; 113 } 114 115 static void usage(char *progname) { 116 LOG(INFO) << "Usage: " << progname << " [--wait] <system> <subcommand> [args...]"; 117 } 118