CommandListener.cpp revision 47ef9bcad2399df05df0cef516bacb2cda6431f4
1f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat/*
2f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * Copyright (C) 2008 The Android Open Source Project
3f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *
4f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * Licensed under the Apache License, Version 2.0 (the "License");
5f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * you may not use this file except in compliance with the License.
6f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * You may obtain a copy of the License at
7f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *
8f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *      http://www.apache.org/licenses/LICENSE-2.0
9f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat *
10f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * Unless required by applicable law or agreed to in writing, software
11f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * distributed under the License is distributed on an "AS IS" BASIS,
12f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * See the License for the specific language governing permissions and
14f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat * limitations under the License.
15f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat */
16f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
17f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <stdlib.h>
188575a350ff725842f076500eab161b39e8720938Daichi Hirono#include <sys/mount.h>
19f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sys/socket.h>
208575a350ff725842f076500eab161b39e8720938Daichi Hirono#include <sys/stat.h>
21a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat#include <sys/types.h>
22f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <netinet/in.h>
23f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <arpa/inet.h>
24a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat#include <dirent.h>
25f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <errno.h>
262350c44ff39b4cb2940893964a05f778fc80a436San Mehat#include <fcntl.h>
277929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash#include <fs_mgr.h>
28d1104f75a736210a95ba890473d78e8dfc8b8915Yabin Cui#include <stdio.h>
2937dcda68d334f70e1f7f69a9817def65fe3ee717Olivier Bailly#include <string.h>
305a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey#include <stdint.h>
315a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey#include <inttypes.h>
3247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono#include <ctype.h>
33f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
34d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat#define LOG_TAG "VoldCmdListener"
355a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey
367e128fbe212c64492afa98bfd6d7fab6f1956831Elliott Hughes#include <android-base/stringprintf.h>
375a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey#include <cutils/fs.h>
38f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <cutils/log.h>
39f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
40f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include <sysutils/SocketClient.h>
413ad9072a5d6f6bda32123b367545649364e3c11dKen Sumrall#include <private/android_filesystem_config.h>
42f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
43f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include "CommandListener.h"
44f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat#include "VolumeManager.h"
4536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey#include "VolumeBase.h"
46a2677e4ad01f250b0765f04adf0acfa6627efc98San Mehat#include "ResponseCode.h"
47586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat#include "Process.h"
48d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat#include "Loop.h"
49d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat#include "Devmapper.h"
501d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey#include "MoveTask.h"
51c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey#include "TrimTask.h"
52f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
533fd60b428202a0f5f324fccc67c0c0402b9131baDianne Hackborn#define DUMP_ARGS 0
543fd60b428202a0f5f324fccc67c0c0402b9131baDianne Hackborn
55f1b736bc5605e92e917ab27f5abf3ba839be2270San MehatCommandListener::CommandListener() :
56149aa3eb65a8cb878781206b1476aae110e0e1fdRobert Greenwalt                 FrameworkListener("vold", true) {
57d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    registerCmd(new DumpCmd());
58eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    registerCmd(new VolumeCmd());
59eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    registerCmd(new AsecCmd());
60508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root    registerCmd(new ObbCmd());
61586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    registerCmd(new StorageCmd());
62b87937cdea689594a293979b30b13054e7455deeKen Sumrall    registerCmd(new FstrimCmd());
638575a350ff725842f076500eab161b39e8720938Daichi Hirono    registerCmd(new AppFuseCmd());
64f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
65f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
663fd60b428202a0f5f324fccc67c0c0402b9131baDianne Hackborn#if DUMP_ARGS
673e971277db0d87652af5622c989233e7159ab909Mark Salyzynvoid CommandListener::dumpArgs(int argc, char **argv, int argObscure) {
68d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    char buffer[4096];
69d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    char *p = buffer;
70d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
71d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    memset(buffer, 0, sizeof(buffer));
72d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    int i;
73d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    for (i = 0; i < argc; i++) {
748f869aa1bc685b505c58e97b4e11a9c7491a16f9Ken Sumrall        unsigned int len = strlen(argv[i]) + 1; // Account for space
75d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        if (i == argObscure) {
76d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            len += 2; // Account for {}
77d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        }
78d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        if (((p - buffer) + len) < (sizeof(buffer)-1)) {
79d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            if (i == argObscure) {
80d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat                *p++ = '{';
81d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat                *p++ = '}';
82d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat                *p++ = ' ';
83d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat                continue;
84d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            }
85d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            strcpy(p, argv[i]);
86d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            p+= strlen(argv[i]);
87d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            if (i != (argc -1)) {
88d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat                *p++ = ' ';
89d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat            }
90d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        }
91d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    }
9297ac40e4e6f3ed0bd5d6878d7d8d4a54fcaecb76San Mehat    SLOGD("%s", buffer);
93d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat}
943e971277db0d87652af5622c989233e7159ab909Mark Salyzyn#else
953e971277db0d87652af5622c989233e7159ab909Mark Salyzynvoid CommandListener::dumpArgs(int /*argc*/, char ** /*argv*/, int /*argObscure*/) { }
963e971277db0d87652af5622c989233e7159ab909Mark Salyzyn#endif
97d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
9836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkeyint CommandListener::sendGenericOkFail(SocketClient *cli, int cond) {
9936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    if (!cond) {
10036801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        return cli->sendMsg(ResponseCode::CommandOkay, "Command succeeded", false);
10136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    } else {
10236801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        return cli->sendMsg(ResponseCode::OperationFailed, "Command failed", false);
10336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    }
10436801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey}
10536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
106d9a4e358614a0c5f60cc76c0636ee4bb02004a32San MehatCommandListener::DumpCmd::DumpCmd() :
107d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat                 VoldCommand("dump") {
108d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat}
109d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
110d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehatint CommandListener::DumpCmd::runCommand(SocketClient *cli,
1113e971277db0d87652af5622c989233e7159ab909Mark Salyzyn                                         int /*argc*/, char ** /*argv*/) {
112d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    cli->sendMsg(0, "Dumping loop status", false);
113d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    if (Loop::dumpState(cli)) {
114d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        cli->sendMsg(ResponseCode::CommandOkay, "Loop dump failed", true);
115d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    }
116d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    cli->sendMsg(0, "Dumping DM status", false);
117d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    if (Devmapper::dumpState(cli)) {
118d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        cli->sendMsg(ResponseCode::CommandOkay, "Devmapper dump failed", true);
119d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    }
12096597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat    cli->sendMsg(0, "Dumping mounted filesystems", false);
12196597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat    FILE *fp = fopen("/proc/mounts", "r");
12296597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat    if (fp) {
12396597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat        char line[1024];
12496597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat        while (fgets(line, sizeof(line), fp)) {
12596597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat            line[strlen(line)-1] = '\0';
12696597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat            cli->sendMsg(0, line, false);;
12796597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat        }
12896597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat        fclose(fp);
12996597e8b840ef671fe5279f8bd64fb09a8b38d4cSan Mehat    }
130d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
131d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    cli->sendMsg(ResponseCode::CommandOkay, "dump complete", false);
132d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    return 0;
133d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat}
134d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
135eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan MehatCommandListener::VolumeCmd::VolumeCmd() :
136eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat                 VoldCommand("volume") {
137f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat}
138f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
139eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehatint CommandListener::VolumeCmd::runCommand(SocketClient *cli,
1407929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash                                           int argc, char **argv) {
141d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    dumpArgs(argc, argv, -1);
142d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
143eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    if (argc < 2) {
144eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
145eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        return 0;
146eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    }
147f1b736bc5605e92e917ab27f5abf3ba839be2270San Mehat
148eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    VolumeManager *vm = VolumeManager::Instance();
149c8e04c5a8285de07d2c84bfbda8eda2c14a9457dJeff Sharkey    std::lock_guard<std::mutex> lock(vm->getLock());
150eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat
15136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    // TODO: tease out methods not directly related to volumes
15257df7bf33968d65c23f3d0dc9f30a8ce2625b1d0San Mehat
15336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    std::string cmd(argv[1]);
15436801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    if (cmd == "reset") {
15536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        return sendGenericOkFail(cli, vm->reset());
15636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
15736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    } else if (cmd == "shutdown") {
15836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        return sendGenericOkFail(cli, vm->shutdown());
15936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
160f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey    } else if (cmd == "debug") {
161f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey        return sendGenericOkFail(cli, vm->setDebug(true));
162f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey
16336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    } else if (cmd == "partition" && argc > 3) {
16436801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        // partition [diskId] [public|private|mixed] [ratio]
16536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        std::string id(argv[2]);
16636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        auto disk = vm->findDisk(id);
16736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        if (disk == nullptr) {
16836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown disk", false);
1699caab76c6b5aefdeeb1715a3695491ca793b8c18Ken Sumrall        }
17036801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
17136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        std::string type(argv[3]);
17236801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        if (type == "public") {
17336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return sendGenericOkFail(cli, disk->partitionPublic());
17436801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        } else if (type == "private") {
17536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return sendGenericOkFail(cli, disk->partitionPrivate());
17636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        } else if (type == "mixed") {
17736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            if (argc < 4) {
17836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey                return cli->sendMsg(ResponseCode::CommandSyntaxError, nullptr, false);
17936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            }
18036801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            int frac = atoi(argv[4]);
18136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return sendGenericOkFail(cli, disk->partitionMixed(frac));
18236801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        } else {
18336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return cli->sendMsg(ResponseCode::CommandSyntaxError, nullptr, false);
18457df7bf33968d65c23f3d0dc9f30a8ce2625b1d0San Mehat        }
18536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
18636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    } else if (cmd == "mkdirs" && argc > 2) {
18736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        // mkdirs [path]
18836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        return sendGenericOkFail(cli, vm->mkdirs(argv[2]));
18936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
190bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey    } else if (cmd == "user_added" && argc > 3) {
191bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        // user_added [user] [serial]
192bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        return sendGenericOkFail(cli, vm->onUserAdded(atoi(argv[2]), atoi(argv[3])));
19336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
194bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey    } else if (cmd == "user_removed" && argc > 2) {
195bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        // user_removed [user]
196bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        return sendGenericOkFail(cli, vm->onUserRemoved(atoi(argv[2])));
197bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey
198bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey    } else if (cmd == "user_started" && argc > 2) {
199bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        // user_started [user]
200bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        return sendGenericOkFail(cli, vm->onUserStarted(atoi(argv[2])));
201bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey
202bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey    } else if (cmd == "user_stopped" && argc > 2) {
203bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        // user_stopped [user]
204bd3038df74ace540d46c530a11e3145f922e1b42Jeff Sharkey        return sendGenericOkFail(cli, vm->onUserStopped(atoi(argv[2])));
20536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
20636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    } else if (cmd == "mount" && argc > 2) {
20736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        // mount [volId] [flags] [user]
20836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        std::string id(argv[2]);
20936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        auto vol = vm->findVolume(id);
21036801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        if (vol == nullptr) {
21136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
21257df7bf33968d65c23f3d0dc9f30a8ce2625b1d0San Mehat        }
21336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
214f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey        int mountFlags = (argc > 3) ? atoi(argv[3]) : 0;
215f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey        userid_t mountUserId = (argc > 4) ? atoi(argv[4]) : -1;
21636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
217f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey        vol->setMountFlags(mountFlags);
218f1b996df6f8283aac6953b22bd9e2496d8c30c86Jeff Sharkey        vol->setMountUserId(mountUserId);
21936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
2201bfb375f77c093a8e16bef4ddeab2681ca126d56Jeff Sharkey        int res = vol->mount();
2211bfb375f77c093a8e16bef4ddeab2681ca126d56Jeff Sharkey        if (mountFlags & android::vold::VolumeBase::MountFlags::kPrimary) {
2221bfb375f77c093a8e16bef4ddeab2681ca126d56Jeff Sharkey            vm->setPrimary(vol);
2231bfb375f77c093a8e16bef4ddeab2681ca126d56Jeff Sharkey        }
2241bfb375f77c093a8e16bef4ddeab2681ca126d56Jeff Sharkey        return sendGenericOkFail(cli, res);
22536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
22636801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    } else if (cmd == "unmount" && argc > 2) {
22736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        // unmount [volId]
22836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        std::string id(argv[2]);
22936801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        auto vol = vm->findVolume(id);
23036801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        if (vol == nullptr) {
23136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
232eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
23336801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
23436801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        return sendGenericOkFail(cli, vol->unmount());
23536801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey
236d0640f6358041f7e2657167560b357078db73526Jeff Sharkey    } else if (cmd == "format" && argc > 3) {
237d0640f6358041f7e2657167560b357078db73526Jeff Sharkey        // format [volId] [fsType|auto]
23836801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        std::string id(argv[2]);
239d0640f6358041f7e2657167560b357078db73526Jeff Sharkey        std::string fsType(argv[3]);
24036801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        auto vol = vm->findVolume(id);
24136801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey        if (vol == nullptr) {
24236801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
24371ebe154a5fbbb4b394a439ff0b6b9c84fbd04f5Jeff Sharkey        }
24449e2bce5b74129c26a35e25d4693cbfe98c4688eSan Mehat
245d0640f6358041f7e2657167560b357078db73526Jeff Sharkey        return sendGenericOkFail(cli, vol->format(fsType));
2461d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey
2471d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey    } else if (cmd == "move_storage" && argc > 3) {
2481d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        // move_storage [fromVolId] [toVolId]
2491d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        auto fromVol = vm->findVolume(std::string(argv[2]));
2501d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        auto toVol = vm->findVolume(std::string(argv[3]));
2511d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        if (fromVol == nullptr || toVol == nullptr) {
2521d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey            return cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown volume", false);
2531d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        }
2541d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey
2551d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        (new android::vold::MoveTask(fromVol, toVol))->start();
2561d6fbcc389ecb9f418076e8ab5f4c93a5d911de9Jeff Sharkey        return sendGenericOkFail(cli, 0);
2575a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey
2585a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey    } else if (cmd == "benchmark" && argc > 2) {
2595a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey        // benchmark [volId]
2605a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey        std::string id(argv[2]);
261c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey        nsecs_t res = vm->benchmarkPrivate(id);
2625a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey        return cli->sendMsg(ResponseCode::CommandOkay,
2635a6bfca1638760b87cf64c5ffb48ff3557cc0563Jeff Sharkey                android::base::StringPrintf("%" PRId64, res).c_str(), false);
264bc40cc8f07f69e0e26fc41516e2a83f0a8becbe0Jeff Sharkey
265bc40cc8f07f69e0e26fc41516e2a83f0a8becbe0Jeff Sharkey    } else if (cmd == "forget_partition" && argc > 2) {
266bc40cc8f07f69e0e26fc41516e2a83f0a8becbe0Jeff Sharkey        // forget_partition [partGuid]
267bc40cc8f07f69e0e26fc41516e2a83f0a8becbe0Jeff Sharkey        std::string partGuid(argv[2]);
268bc40cc8f07f69e0e26fc41516e2a83f0a8becbe0Jeff Sharkey        return sendGenericOkFail(cli, vm->forgetPartition(partGuid));
26966270a21df1058434e4d63691221f11ff5387a0fJeff Sharkey
27066270a21df1058434e4d63691221f11ff5387a0fJeff Sharkey    } else if (cmd == "remount_uid" && argc > 3) {
27166270a21df1058434e4d63691221f11ff5387a0fJeff Sharkey        // remount_uid [uid] [none|default|read|write]
27266270a21df1058434e4d63691221f11ff5387a0fJeff Sharkey        uid_t uid = atoi(argv[2]);
27366270a21df1058434e4d63691221f11ff5387a0fJeff Sharkey        std::string mode(argv[3]);
27466270a21df1058434e4d63691221f11ff5387a0fJeff Sharkey        return sendGenericOkFail(cli, vm->remountUid(uid, mode));
275a2677e4ad01f250b0765f04adf0acfa6627efc98San Mehat    }
276a2677e4ad01f250b0765f04adf0acfa6627efc98San Mehat
27736801cccf27152c9eca5aab6ba3527221525110fJeff Sharkey    return cli->sendMsg(ResponseCode::CommandSyntaxError, nullptr, false);
278a2677e4ad01f250b0765f04adf0acfa6627efc98San Mehat}
279a2677e4ad01f250b0765f04adf0acfa6627efc98San Mehat
280586536c60b773e3517531ad8a6cb0de6722c67fcSan MehatCommandListener::StorageCmd::StorageCmd() :
281586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                 VoldCommand("storage") {
282586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat}
283586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
284586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehatint CommandListener::StorageCmd::runCommand(SocketClient *cli,
285586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                                                      int argc, char **argv) {
2867929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash    /* Guarantied to be initialized by vold's main() before the CommandListener is active */
2877929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash    extern struct fstab *fstab;
2887929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash
289d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat    dumpArgs(argc, argv, -1);
290d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat
291586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    if (argc < 2) {
292586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
293586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        return 0;
294586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    }
295586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
2967929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash    if (!strcmp(argv[1], "mountall")) {
2977929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash        if (argc != 2) {
2987929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: mountall", false);
2997929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash            return 0;
3007929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash        }
3017929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash        fs_mgr_mount_all(fstab);
3027929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash        cli->sendMsg(ResponseCode::CommandOkay, "Mountall ran successfully", false);
3037929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash        return 0;
3047929aa73d0c1fa75e8e0fcd4272361ad0ea9b0e6Mohamad Ayyash    }
305586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    if (!strcmp(argv[1], "users")) {
306586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        DIR *dir;
307586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        struct dirent *de;
308586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
309edf7adf21e1c210e5954b1128efb61b62f6da274JP Abgrall        if (argc < 3) {
310edf7adf21e1c210e5954b1128efb61b62f6da274JP Abgrall            cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument: user <mountpoint>", false);
311edf7adf21e1c210e5954b1128efb61b62f6da274JP Abgrall            return 0;
312edf7adf21e1c210e5954b1128efb61b62f6da274JP Abgrall        }
313586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        if (!(dir = opendir("/proc"))) {
314586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            cli->sendMsg(ResponseCode::OperationFailed, "Failed to open /proc", true);
315586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            return 0;
316586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        }
317586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
318586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        while ((de = readdir(dir))) {
319586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            int pid = Process::getPid(de->d_name);
320586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
321586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            if (pid < 0) {
322586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                continue;
323586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            }
324586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
325586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            char processName[255];
326586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            Process::getProcessName(pid, processName, sizeof(processName));
327586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
328586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            if (Process::checkFileDescriptorSymLinks(pid, argv[2]) ||
329586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                Process::checkFileMaps(pid, argv[2]) ||
330586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                Process::checkSymLink(pid, argv[2], "cwd") ||
331586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                Process::checkSymLink(pid, argv[2], "root") ||
332586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                Process::checkSymLink(pid, argv[2], "exe")) {
333586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
334586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                char msg[1024];
335586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                snprintf(msg, sizeof(msg), "%d %s", pid, processName);
336586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat                cli->sendMsg(ResponseCode::StorageUsersListResult, msg, false);
337586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat            }
338586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        }
339586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        closedir(dir);
340586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        cli->sendMsg(ResponseCode::CommandOkay, "Storage user list complete", false);
341586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    } else {
342586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat        cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown storage cmd", false);
343586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    }
344586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat    return 0;
345586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat}
346586536c60b773e3517531ad8a6cb0de6722c67fcSan Mehat
347eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan MehatCommandListener::AsecCmd::AsecCmd() :
348eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat                 VoldCommand("asec") {
349048b0801fcd6fcfbb8fa812284c751181e4821b8San Mehat}
350048b0801fcd6fcfbb8fa812284c751181e4821b8San Mehat
351344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Rootvoid CommandListener::AsecCmd::listAsecsInDirectory(SocketClient *cli, const char *directory) {
352344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    DIR *d = opendir(directory);
353344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
354344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    if (!d) {
355344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        cli->sendMsg(ResponseCode::OperationFailed, "Failed to open asec dir", true);
356344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        return;
357344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    }
358344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
359344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    size_t dirent_len = offsetof(struct dirent, d_name) +
3608c480f73eed963eeca9b7df3e4c4543c6e43b0d7Elliott Hughes            fpathconf(dirfd(d), _PC_NAME_MAX) + 1;
361344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
362344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    struct dirent *dent = (struct dirent *) malloc(dirent_len);
363344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    if (dent == NULL) {
364344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        cli->sendMsg(ResponseCode::OperationFailed, "Failed to allocate memory", true);
365344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        return;
366344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    }
367344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
368344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    struct dirent *result;
369344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
370344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    while (!readdir_r(d, dent, &result) && result != NULL) {
371344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        if (dent->d_name[0] == '.')
372344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            continue;
373344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        if (dent->d_type != DT_REG)
374344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            continue;
375344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        size_t name_len = strlen(dent->d_name);
376344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        if (name_len > 5 && name_len < 260 &&
377344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root                !strcmp(&dent->d_name[name_len - 5], ".asec")) {
378344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            char id[255];
379344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            memset(id, 0, sizeof(id));
3807b0bc8571465666d6cba79bda60b72a97f852c05Kenny Root            strlcpy(id, dent->d_name, name_len - 4);
381344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            cli->sendMsg(ResponseCode::AsecListResult, id, false);
382344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        }
383344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    }
384344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    closedir(d);
385344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
386344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    free(dent);
387344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root}
388344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
389eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehatint CommandListener::AsecCmd::runCommand(SocketClient *cli,
390eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat                                                      int argc, char **argv) {
391eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    if (argc < 2) {
392eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
393048b0801fcd6fcfbb8fa812284c751181e4821b8San Mehat        return 0;
394048b0801fcd6fcfbb8fa812284c751181e4821b8San Mehat    }
395048b0801fcd6fcfbb8fa812284c751181e4821b8San Mehat
396eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    VolumeManager *vm = VolumeManager::Instance();
397eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    int rc = 0;
398a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat
399eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    if (!strcmp(argv[1], "list")) {
400d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
401a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat
4029f18fe7807a4e4089778243dbbd08d154ec15540Jeff Sharkey        listAsecsInDirectory(cli, VolumeManager::SEC_ASECDIR_EXT);
4039f18fe7807a4e4089778243dbbd08d154ec15540Jeff Sharkey        listAsecsInDirectory(cli, VolumeManager::SEC_ASECDIR_INT);
404eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "create")) {
405d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, 5);
406344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        if (argc != 8) {
407eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError,
408344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root                    "Usage: asec create <container-id> <size_mb> <fstype> <key> <ownerUid> "
409344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root                    "<isExternal>", false);
410eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
411a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat        }
412a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat
413a4f48d0f44d09b6a9a3b16f6c0121ffd5123eef3Mateusz Nowak        unsigned long numSectors = (atoi(argv[3]) * (1024 * 1024)) / 512;
414344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        const bool isExternal = (atoi(argv[7]) == 1);
415344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        rc = vm->createAsec(argv[2], numSectors, argv[4], argv[5], atoi(argv[6]), isExternal);
416fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg    } else if (!strcmp(argv[1], "resize")) {
417fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg        dumpArgs(argc, argv, -1);
418fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg        if (argc != 5) {
419fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec resize <container-id> <size_mb> <key>", false);
420fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg            return 0;
421fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg        }
422a4f48d0f44d09b6a9a3b16f6c0121ffd5123eef3Mateusz Nowak        unsigned long numSectors = (atoi(argv[3]) * (1024 * 1024)) / 512;
423fcd34a0ddd45db83b7bc71ff47cba9b789089fddDaniel Rosenberg        rc = vm->resizeAsec(argv[2], numSectors, argv[4]);
424eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "finalize")) {
425d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
426eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        if (argc != 3) {
427eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec finalize <container-id>", false);
428eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
429eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
4308f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        rc = vm->finalizeAsec(argv[2]);
431344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root    } else if (!strcmp(argv[1], "fixperms")) {
432344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        dumpArgs(argc, argv, -1);
433344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        if  (argc != 5) {
434344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec fixperms <container-id> <gid> <filename>", false);
435344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            return 0;
436344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        }
437344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
438344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        char *endptr;
439344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        gid_t gid = (gid_t) strtoul(argv[3], &endptr, 10);
440344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        if (*endptr != '\0') {
441344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec fixperms <container-id> <gid> <filename>", false);
442344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root            return 0;
443344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        }
444344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root
445344ca10856f3d3087a3288ce8f91ad83665d93fbKenny Root        rc = vm->fixupAsecPermissions(argv[2], gid, argv[4]);
446eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "destroy")) {
447d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
4484ba8948dc16463053e21cda5744f519a555080d0San Mehat        if (argc < 3) {
4494ba8948dc16463053e21cda5744f519a555080d0San Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec destroy <container-id> [force]", false);
450eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
451eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
4524ba8948dc16463053e21cda5744f519a555080d0San Mehat        bool force = false;
4534ba8948dc16463053e21cda5744f519a555080d0San Mehat        if (argc > 3 && !strcmp(argv[3], "force")) {
4544ba8948dc16463053e21cda5744f519a555080d0San Mehat            force = true;
4554ba8948dc16463053e21cda5744f519a555080d0San Mehat        }
4568f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        rc = vm->destroyAsec(argv[2], force);
457eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "mount")) {
458d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, 3);
45943ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey        if (argc != 6) {
460eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError,
46143ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey                    "Usage: asec mount <namespace-id> <key> <ownerUid> <ro|rw>", false);
462eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
463eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
46443ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey        bool readOnly = true;
46543ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey        if (!strcmp(argv[5], "rw")) {
46643ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey            readOnly = false;
46743ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey        }
46843ed123d3fc1a3edf3660cd0e2528e971abc399eJeff Sharkey        rc = vm->mountAsec(argv[2], argv[3], atoi(argv[4]), readOnly);
469eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "unmount")) {
470d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
4714ba8948dc16463053e21cda5744f519a555080d0San Mehat        if (argc < 3) {
4724ba8948dc16463053e21cda5744f519a555080d0San Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec unmount <container-id> [force]", false);
473eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
474eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
4754ba8948dc16463053e21cda5744f519a555080d0San Mehat        bool force = false;
4764ba8948dc16463053e21cda5744f519a555080d0San Mehat        if (argc > 3 && !strcmp(argv[3], "force")) {
4774ba8948dc16463053e21cda5744f519a555080d0San Mehat            force = true;
4784ba8948dc16463053e21cda5744f519a555080d0San Mehat        }
4798f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        rc = vm->unmountAsec(argv[2], force);
480eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "rename")) {
481d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
482eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        if (argc != 4) {
483eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError,
484eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat                    "Usage: asec rename <old_id> <new_id>", false);
485eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
486eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
4878f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        rc = vm->renameAsec(argv[2], argv[3]);
488eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat    } else if (!strcmp(argv[1], "path")) {
489d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
490eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        if (argc != 3) {
491eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec path <container-id>", false);
492eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            return 0;
493eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
494eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        char path[255];
495a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat
49688ac2c06539485942bf414efda2d39647fa1a415San Mehat        if (!(rc = vm->getAsecMountPath(argv[2], path, sizeof(path)))) {
497eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat            cli->sendMsg(ResponseCode::AsecPathResult, path, false);
49888ac2c06539485942bf414efda2d39647fa1a415San Mehat            return 0;
499eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        }
500736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn    } else if (!strcmp(argv[1], "fspath")) {
501736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn        dumpArgs(argc, argv, -1);
502736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn        if (argc != 3) {
503736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: asec fspath <container-id>", false);
504736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn            return 0;
505736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn        }
506736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn        char path[255];
507736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn
508736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn        if (!(rc = vm->getAsecFilesystemPath(argv[2], path, sizeof(path)))) {
509736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn            cli->sendMsg(ResponseCode::AsecPathResult, path, false);
510736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn            return 0;
511736910ca99a40b9add4353bf619e778c40938948Dianne Hackborn        }
512a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat    } else {
513d9a4e358614a0c5f60cc76c0636ee4bb02004a32San Mehat        dumpArgs(argc, argv, -1);
514eba65e9d438a05f1c5dfd0f8d31bc463a5d08eeeSan Mehat        cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown asec cmd", false);
515a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat    }
516a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat
5178f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat    if (!rc) {
5188f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        cli->sendMsg(ResponseCode::CommandOkay, "asec operation succeeded", false);
5198f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat    } else {
5208f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        rc = ResponseCode::convertFromErrno();
5218f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat        cli->sendMsg(rc, "asec operation failed", true);
5228f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat    }
5238f2875b29780312f4edda3d831cc8a99e1648dd5San Mehat
524a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat    return 0;
525a19b250bd273455933ca3502cf2c2e0a803aff77San Mehat}
5262350c44ff39b4cb2940893964a05f778fc80a436San Mehat
527508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny RootCommandListener::ObbCmd::ObbCmd() :
528508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root                 VoldCommand("obb") {
529fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root}
530fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root
531508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Rootint CommandListener::ObbCmd::runCommand(SocketClient *cli,
532fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root                                                      int argc, char **argv) {
533fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    if (argc < 2) {
534fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
535fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        return 0;
536fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    }
537fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root
538fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    VolumeManager *vm = VolumeManager::Instance();
539fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    int rc = 0;
540fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root
541508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root    if (!strcmp(argv[1], "list")) {
542508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        dumpArgs(argc, argv, -1);
543508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root
544508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        rc = vm->listMountedObbs(cli);
545508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root    } else if (!strcmp(argv[1], "mount")) {
546fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root            dumpArgs(argc, argv, 3);
547fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root            if (argc != 5) {
548fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root                cli->sendMsg(ResponseCode::CommandSyntaxError,
5496947904a76b69a1db20a3ddd30c0bcd281922fdeJeff Sharkey                        "Usage: obb mount <filename> <key> <ownerGid>", false);
550fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root                return 0;
551fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root            }
552508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root            rc = vm->mountObb(argv[2], argv[3], atoi(argv[4]));
553fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    } else if (!strcmp(argv[1], "unmount")) {
554fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        dumpArgs(argc, argv, -1);
555fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        if (argc < 3) {
556508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: obb unmount <source file> [force]", false);
557fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root            return 0;
558fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        }
559fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        bool force = false;
560fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        if (argc > 3 && !strcmp(argv[3], "force")) {
561fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root            force = true;
562fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        }
563508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        rc = vm->unmountObb(argv[2], force);
564508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root    } else if (!strcmp(argv[1], "path")) {
565508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        dumpArgs(argc, argv, -1);
566508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        if (argc != 3) {
567508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root            cli->sendMsg(ResponseCode::CommandSyntaxError, "Usage: obb path <source file>", false);
568508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root            return 0;
569508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        }
570508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        char path[255];
571508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root
572508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        if (!(rc = vm->getObbMountPath(argv[2], path, sizeof(path)))) {
573508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root            cli->sendMsg(ResponseCode::AsecPathResult, path, false);
574508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root            return 0;
575508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        }
576fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    } else {
577fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        dumpArgs(argc, argv, -1);
578508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        cli->sendMsg(ResponseCode::CommandSyntaxError, "Unknown obb cmd", false);
579fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    }
580fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root
581fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    if (!rc) {
582508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        cli->sendMsg(ResponseCode::CommandOkay, "obb operation succeeded", false);
583fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    } else {
584fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root        rc = ResponseCode::convertFromErrno();
585508c0e1605b795bbb51cb47d955b89f3df26ca94Kenny Root        cli->sendMsg(rc, "obb operation failed", true);
586fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    }
587fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root
588fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root    return 0;
589fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root}
590fb7c4d5a8a1031cf0e493ff182dcf458e5fe8c77Kenny Root
591b87937cdea689594a293979b30b13054e7455deeKen SumrallCommandListener::FstrimCmd::FstrimCmd() :
592b87937cdea689594a293979b30b13054e7455deeKen Sumrall                 VoldCommand("fstrim") {
593b87937cdea689594a293979b30b13054e7455deeKen Sumrall}
594b87937cdea689594a293979b30b13054e7455deeKen Sumrallint CommandListener::FstrimCmd::runCommand(SocketClient *cli,
595b87937cdea689594a293979b30b13054e7455deeKen Sumrall                                                      int argc, char **argv) {
596b87937cdea689594a293979b30b13054e7455deeKen Sumrall    if ((cli->getUid() != 0) && (cli->getUid() != AID_SYSTEM)) {
597b87937cdea689594a293979b30b13054e7455deeKen Sumrall        cli->sendMsg(ResponseCode::CommandNoPermission, "No permission to run fstrim commands", false);
598b87937cdea689594a293979b30b13054e7455deeKen Sumrall        return 0;
599b87937cdea689594a293979b30b13054e7455deeKen Sumrall    }
600b87937cdea689594a293979b30b13054e7455deeKen Sumrall
601b87937cdea689594a293979b30b13054e7455deeKen Sumrall    if (argc < 2) {
602b87937cdea689594a293979b30b13054e7455deeKen Sumrall        cli->sendMsg(ResponseCode::CommandSyntaxError, "Missing Argument", false);
603b87937cdea689594a293979b30b13054e7455deeKen Sumrall        return 0;
604b87937cdea689594a293979b30b13054e7455deeKen Sumrall    }
605b87937cdea689594a293979b30b13054e7455deeKen Sumrall
606c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    VolumeManager *vm = VolumeManager::Instance();
607c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    std::lock_guard<std::mutex> lock(vm->getLock());
608b87937cdea689594a293979b30b13054e7455deeKen Sumrall
609c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    int flags = 0;
610b87937cdea689594a293979b30b13054e7455deeKen Sumrall
611c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    std::string cmd(argv[1]);
612c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    if (cmd == "dotrim") {
613c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey        flags = 0;
614c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    } else if (cmd == "dotrimbench") {
615c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey        flags = android::vold::TrimTask::Flags::kBenchmarkAfter;
616c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    } else if (cmd == "dodtrim") {
617c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey        flags = android::vold::TrimTask::Flags::kDeepTrim;
618c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    } else if (cmd == "dodtrimbench") {
619c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey        flags = android::vold::TrimTask::Flags::kDeepTrim
620c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey                | android::vold::TrimTask::Flags::kBenchmarkAfter;
621c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    }
622b87937cdea689594a293979b30b13054e7455deeKen Sumrall
623c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    (new android::vold::TrimTask(flags))->start();
624c86ab6f538bec63638c168d6c843fe7cf73add3bJeff Sharkey    return sendGenericOkFail(cli, 0);
625b87937cdea689594a293979b30b13054e7455deeKen Sumrall}
6268575a350ff725842f076500eab161b39e8720938Daichi Hirono
62747ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hironostatic size_t kAppFuseMaxMountPointName = 32;
62847ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono
6298575a350ff725842f076500eab161b39e8720938Daichi HironoCommandListener::AppFuseCmd::AppFuseCmd() : VoldCommand("appfuse") {}
6308575a350ff725842f076500eab161b39e8720938Daichi Hirono
6318575a350ff725842f076500eab161b39e8720938Daichi Hironoint CommandListener::AppFuseCmd::runCommand(SocketClient *cli,
6328575a350ff725842f076500eab161b39e8720938Daichi Hirono                                            int argc,
6338575a350ff725842f076500eab161b39e8720938Daichi Hirono                                            char **argv) {
6348575a350ff725842f076500eab161b39e8720938Daichi Hirono    if (argc < 2) {
6358575a350ff725842f076500eab161b39e8720938Daichi Hirono        cli->sendMsg(
6368575a350ff725842f076500eab161b39e8720938Daichi Hirono            ResponseCode::CommandSyntaxError, "Missing argument", false);
6378575a350ff725842f076500eab161b39e8720938Daichi Hirono        return 0;
6388575a350ff725842f076500eab161b39e8720938Daichi Hirono    }
6398575a350ff725842f076500eab161b39e8720938Daichi Hirono
6408575a350ff725842f076500eab161b39e8720938Daichi Hirono    const std::string command(argv[1]);
6418575a350ff725842f076500eab161b39e8720938Daichi Hirono
6428575a350ff725842f076500eab161b39e8720938Daichi Hirono    if (command == "mount" && argc == 4) {
6438575a350ff725842f076500eab161b39e8720938Daichi Hirono        const uid_t uid = atoi(argv[2]);
6448575a350ff725842f076500eab161b39e8720938Daichi Hirono        const std::string name(argv[3]);
64547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono
64647ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        // Check mount point name.
64747ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        bool invalidName = false;
64847ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        if (name.size() > kAppFuseMaxMountPointName) {
64947ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            invalidName = true;
65047ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        }
65147ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        for (size_t i = 0; i < name.size(); i++) {
65247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            if (!isalnum(name[i])) {
65347ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                invalidName = true;
65447ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                break;
65547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            }
65647ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        }
65747ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        if (invalidName) {
65847ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            return cli->sendMsg(
65947ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    ResponseCode::CommandParameterError,
66047ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    "Invalid mount point name.",
66147ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    false);
66247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        }
66347ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono
66447ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        // Create directories.
66547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        char path[PATH_MAX];
66647ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        {
66747ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            snprintf(path, PATH_MAX, "/mnt/appfuse/%d_%s", uid, name.c_str());
66847ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            umount2(path, UMOUNT_NOFOLLOW | MNT_DETACH);
66947ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            const int result = android::vold::PrepareDir(path, 0700, 0, 0);
67047ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            if (result != 0) {
67147ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                return sendGenericOkFail(cli, result);
67247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            }
67347ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        }
67447ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono
67547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        // Open device FD.
6768575a350ff725842f076500eab161b39e8720938Daichi Hirono        const int device_fd = open("/dev/fuse", O_RDWR);
6778575a350ff725842f076500eab161b39e8720938Daichi Hirono        if (device_fd < 0) {
6788575a350ff725842f076500eab161b39e8720938Daichi Hirono            sendGenericOkFail(cli, device_fd);
6798575a350ff725842f076500eab161b39e8720938Daichi Hirono            return 0;
6808575a350ff725842f076500eab161b39e8720938Daichi Hirono        }
6818575a350ff725842f076500eab161b39e8720938Daichi Hirono
68247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        // Mount.
68347ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        {
68447ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            char opts[256];
68547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            snprintf(
68647ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    opts,
68747ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    sizeof(opts),
68847ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    "fd=%i,"
68947ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    "rootmode=40000,"
69047ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    "default_permissions,"
69147ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    "user_id=%d,group_id=%d",
69247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    device_fd,
69347ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    uid,
69447ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    uid);
69547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            // TODO: Make it bound mount in application namespace.
69647ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            // TODO: Add context= option to opts.
69747ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            const int result = mount(
69847ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    "/dev/fuse", path, "fuse",
69947ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                    MS_NOSUID | MS_NODEV | MS_NOEXEC | MS_NOATIME, opts);
70047ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            if (result != 0) {
70147ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                sendGenericOkFail(cli, 1);
70247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono                return 0;
70347ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            }
70447ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono        }
70547ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono
7068575a350ff725842f076500eab161b39e8720938Daichi Hirono        const int result = sendFd(cli, device_fd);
7078575a350ff725842f076500eab161b39e8720938Daichi Hirono        close(device_fd);
7088575a350ff725842f076500eab161b39e8720938Daichi Hirono        return result;
7098575a350ff725842f076500eab161b39e8720938Daichi Hirono    }
7108575a350ff725842f076500eab161b39e8720938Daichi Hirono
71147ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono    return cli->sendMsg(
71247ef9bcad2399df05df0cef516bacb2cda6431f4Daichi Hirono            ResponseCode::CommandSyntaxError,  "Unknown appfuse cmd", false);
7138575a350ff725842f076500eab161b39e8720938Daichi Hirono}
7148575a350ff725842f076500eab161b39e8720938Daichi Hirono
7158575a350ff725842f076500eab161b39e8720938Daichi Hironoint CommandListener::AppFuseCmd::sendFd(SocketClient *cli, int fd) {
7168575a350ff725842f076500eab161b39e8720938Daichi Hirono    struct iovec data;
7178575a350ff725842f076500eab161b39e8720938Daichi Hirono    char dataBuffer[128];
7188575a350ff725842f076500eab161b39e8720938Daichi Hirono    char controlBuffer[CMSG_SPACE(sizeof(int))];
7198575a350ff725842f076500eab161b39e8720938Daichi Hirono    struct msghdr message;
7208575a350ff725842f076500eab161b39e8720938Daichi Hirono
7218575a350ff725842f076500eab161b39e8720938Daichi Hirono    // Message.
7228575a350ff725842f076500eab161b39e8720938Daichi Hirono    memset(&message, 0, sizeof(struct msghdr));
7238575a350ff725842f076500eab161b39e8720938Daichi Hirono    message.msg_iov = &data;
7248575a350ff725842f076500eab161b39e8720938Daichi Hirono    message.msg_iovlen = 1;
7258575a350ff725842f076500eab161b39e8720938Daichi Hirono    message.msg_control = controlBuffer;
7268575a350ff725842f076500eab161b39e8720938Daichi Hirono    message.msg_controllen = CMSG_SPACE(sizeof(int));
7278575a350ff725842f076500eab161b39e8720938Daichi Hirono
7288575a350ff725842f076500eab161b39e8720938Daichi Hirono    // Data.
7298575a350ff725842f076500eab161b39e8720938Daichi Hirono    data.iov_base = dataBuffer;
7308575a350ff725842f076500eab161b39e8720938Daichi Hirono    data.iov_len = snprintf(dataBuffer,
7318575a350ff725842f076500eab161b39e8720938Daichi Hirono                            sizeof(dataBuffer),
7328575a350ff725842f076500eab161b39e8720938Daichi Hirono                            "200 %d AppFuse command succeeded",
7338575a350ff725842f076500eab161b39e8720938Daichi Hirono                            cli->getCmdNum()) + 1;
7348575a350ff725842f076500eab161b39e8720938Daichi Hirono
7358575a350ff725842f076500eab161b39e8720938Daichi Hirono    // Control.
7368575a350ff725842f076500eab161b39e8720938Daichi Hirono    struct cmsghdr* const controlMessage = CMSG_FIRSTHDR(&message);
7378575a350ff725842f076500eab161b39e8720938Daichi Hirono    memset(controlBuffer, 0, CMSG_SPACE(sizeof(int)));
7388575a350ff725842f076500eab161b39e8720938Daichi Hirono    controlMessage->cmsg_level = SOL_SOCKET;
7398575a350ff725842f076500eab161b39e8720938Daichi Hirono    controlMessage->cmsg_type = SCM_RIGHTS;
7408575a350ff725842f076500eab161b39e8720938Daichi Hirono    controlMessage->cmsg_len = CMSG_LEN(sizeof(int));
7418575a350ff725842f076500eab161b39e8720938Daichi Hirono    *((int *) CMSG_DATA(controlMessage)) = fd;
7428575a350ff725842f076500eab161b39e8720938Daichi Hirono
7438575a350ff725842f076500eab161b39e8720938Daichi Hirono    return TEMP_FAILURE_RETRY(sendmsg(cli->getSocket(), &message, 0));
7448575a350ff725842f076500eab161b39e8720938Daichi Hirono}
745