vdc.cpp

   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 }